3 e�$g�+ � @ sZ d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl m Z d dl m Z d d l mZ y$d dlZd dlZd dlZd dlZW n< ek r� d dljZd dljZd dljZd dljZY nX d dlZej d d d �Z e e d ��s e je _ e j Z dd� Z!G dd� de j"�Z#G dd� d�Z$ddd�Z%ddd�Z&dd� Z'dS ) � N)�config)� clientCaps)� up2dateLog)� up2dateErrors)� up2dateUtils)�SSL)�rpclib)� raise_with_tbzrhn-client-toolsT)Zfallback�ugettextc C s t | � d S )N)�print)�msg� r � /usr/lib/python3.6/rpcServer.py�stdoutMsgCallback$ s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RetryServerc C s || _ d S )N)� serverList)�selfr r r r � addServerList) s zRetryServer.addServerListc C s� t j� | _�xry| j||�}W �nX tjk r: � Y �n@ tjk rR � Y �n( tj k r� | jj d� t t d�� t jd � Y n� | jj� }|d kr�| jj� � d| j }|dt j� d t j� d f }|d| jj� }| jj |� tj| jj� �}|d }|d | _|d | _|j� }|d k�rDttjd | �� | j| _|| _| jj� | _ | j�sld | _d| _ wY nX P qW |S )Nz,Error: Server Unavailable. Please try later.� z!An error occurred talking to %s: z%s %s r z Trying the next serverURL: %s � �http�httpsz%Redirected to unsupported protocol %sz/RPC2���)r r ) r �initLog�logZ_requestr ZInvalidRedirectionError� xmlrpclibZFault�httplibZ BadStatusLine�log_mer �_�sys�exitr �next�resetServerIndexZ_host�exc_info�server�urlparseZurlsplitZ_handler�lowerr Z _orig_handlerZ_typeZ_uriZ_allow_redirect)r Z methodnameZparams�retr$ r Z parse_res�typr r r � _request1, sN zRetryServer._request1c C s t jj| j|�S )N)r r Z_Methodr) )r �namer r r � __getattr__^ s zRetryServer.__getattr__N)�__name__� __module__� __qualname__r r) r+ r r r r r ( s 2r c @ s0 e Zd Zg fdd�Zdd� Zdd� Zdd� Zd S ) � ServerListc C s || _ d| _d S )Nr )r �index)r Z serverlistr r r �__init__e s zServerList.__init__c C s | j | j | _| jS )N)r r0 Z serverurl)r r r r r$ i s zServerList.serverc C s( | j d | _ | j t| j�kr d S | j� S )Nr )r0 �lenr r$ )r r r r r! n s zServerList.nextc C s d| _ d S )Nr )r0 )r r r r r" t s zServerList.resetServerIndexN)r, r- r. r1 r$ r! r" r r r r r/ d s r/ c C sF t j� }|d kr:|d }|d ks0t|j� �dkr4d}|g}t| ||�S )NZ clnServerURLr z&http://cln.cloudlinux.com/clweb/xmlrpc)r �initUp2dateConfigr2 �strip� getServer)�refreshCallback�serverOverride�timeout�cfgr r r � getCLNServerw s r: c C s� t j� }tj� }|d }t|t�s(|g}|p0dg}|d rDtj� }nd }|sVtj� }n|}t|�} d } d } |d r�|d p|d } |d p�d } d } xFdD ]>} | t j kr�t j | s�q�t j | j d �d } | j d �d } P q�W t | j � | || | |d�}|j| � |jdtj� � | �r|j| � dd� |jjD �}|�r�xP|D ]H}t j|t j��srdtd�|f }|jd| � tj|��|j|� �q6W tj � tj j � }x |D ]\}}|j||� �q�W |S )NZ sslCACertz'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERTZ enableProxyZenableProxyAuth� proxyUser� proxyPassword�LANGUAGE�LC_ALL� LC_MESSAGES�LANG�:r �.)r6 �proxyZusernameZpasswordr8 zX-Up2date-Versionc S s" g | ]}t j |�d dkrd�qS )r r T)r% )�.0�ir r r � <listcomp>� s z getServer.<locals>.<listcomp>z%s: %sz ERROR: can not find RHNS CA filez%s)r= r> r? r@ ) r r r r3 � isinstance�listZgetProxySettingZ getServerlURLr/ �os�environ�splitr r$ r Z add_headerr �versionZsetlangr �access�R_OKr r r ZSSLCertificateFileNotFoundZadd_trusted_certr Z loadLocalCapsZcapsZ headerFormat)r6 r7 r8 r r9 ZcaZ rhns_ca_certsZ proxyHostZ serverUrlsr r; r< Zlang�env�sZneed_caZ rhns_ca_certr Z headerlistZ headerName�valuer r r r5 � sd r5 c O s� t j� }|jd| jd � tj� }d }d}yt|d �}W n tk rT d}Y nX |dkrbd}�xPd}d }y| ||�}W �n� tk r� t t j t d��� Y �n� t jtjfk �r4 |jdtj� d |f � ||k�r*tj� d } t| j�dk�rt t j | jd �� nt t j | jd �� nd}Y �nB tjk �rb td� t t j d�� Y �n tjk �r� tj� d } d } | d | j } | d | j } | d | j } |j| � t t j | �� Y �n� t j k �rP tj� d } |jd | j |f � | j!dk�r$|jd| j" � |jd� t#j$| j%�\} } d} t&| �dk�rh|jd| � ddl'm(} |j)� t&| �dk�r�|jt d�� t t j | j �� t&| �dk�r&t*|d �t*g �k�r�|d }n|d }t*|�t*g �k�r�d|d |d |d |d f }n|}d|| f } |j| � t t j+| �� | �sL||k�rHt t j | j �� nd}Y n& t j,k �rt t t j d�� Y nX |d k�r�P nd}|�r�t-j.d � |d }||krft j d ��qfW |S ) Nz rpcServer: Calling XMLRPC %sZ _Method__namer ZnetworkRetriesr z Connection aborted by the userz(A socket error occurred: %s, attempt #%szhttplib.IncompleteReadz An HTTP error occurred: zURL: %s zStatus Code: %s zError Message: %s z,A protocol error occurred: %s , attempt #%s,i� zCould not find URL, %sz)Check server name and/or URL, then retry �" z'Auth token timeout occurred errmsg: %s)� up2dateAuth�3 z.Server has refused connection due to high load� z %s-%s-%s.%sr � zFile Not Found: %s %sz Broken response from the server.� z0The data returned from the server was incomplete)/r r Z log_debug�__dict__r r3 �int� ValueError�KeyboardInterruptr r ZCommunicationErrorr �socket�errorr Z socket_errorr r r# r2 �argsr ZIncompleteReadr �urllib2Z HTTPError�filename�coder r Z ProtocolError�errmsgZerrcodeZurlr Z reportErrorZheaders�abs�up2date_clientrS ZupdateLoginInfo�type�FileNotFoundErrorZ ResponseError�timeZsleep)�methodr^ �kwargsr r9 r' Z attempt_countZattemptsZfailure�er ZerrCodeZerrMsg�resetrS ZpkgZpkgNamer r r �doCall� s� " rl )NNN)NNN)(rI r r\ rg rd r r r r r Zrhnr r Zrhn.tbr r r_ r% r � ImportErrorZ http.clientZclientZurllib.requestZrequestZ urllib.parse�parseZ xmlrpc.client�gettextZ translation�t�hasattrr r r ZServerr r/ r: r5 rl r r r r �<module> s@ < L