3
�Xj4. � @ s: d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dlm
Z
d dlm Z d dl m
Z
y$d dlZd dlZd dlZd dlZW n< ek
r� d dljZd dljZd dljZd dljZY nX d dlZejddd �Ze ed
�s�eje_ ej Z d d � Z G d
d� de
j �Z!G dd� d�Z"ddd�Z#dd� Z$dS )� N)�config)�
up2dateLog)�
up2dateErrors)� up2dateUtils)�rpclibzrhn-client-toolsT)Zfallback�ugettextc C s t | � d S )N)�print)�msg� r
� /usr/lib/python3.6/rpcServer.py�stdoutMsgCallback% s r c s0 e Zd Zi Zdd� Z� fdd�Zdd� Z� ZS )� RetryServerc C s
|| _ d S )N)�
serverList)�selfr r
r
r �
addServerList- s zRetryServer.addServerListc
s� t j� | _�x�yt� j||�}W �n� tjk
r< � Y �nj tjk
rt | jj d� t
t d�� t j
d� Y �n2 | jj� }|d kr�| jj� | jr�djdd� | jj� D ��}t
t d�� t
|� � d| j }dt j� d t j� d f }|| j| j< || }|d
| jj� }| jj |� tj| jj� �}|d } |d | _|d | _| j� } | dk�rptj d| ��| j| _ | | _ | jj� | _ | j�s�d| _d| _ wY nX P qW |S )Nz,Error: Server Unavailable. Please try later.� �
c S s g | ]\}}d ||f �qS )z%s:
%sr
)�.0�host�errorr
r
r �
I s z)RetryServer._request1..z>Errors occurred while trying to connect to the remote servers.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�log�superZ_ServerProxy__request� xmlrpclibZFault�httplibZ
BadStatusLine�log_mer �_�sys�exitr �next�resetServerIndex�_error_messages�join�itemsZ_host�exc_info�server�urlparseZurlsplitZ_handler�lowerr ZInvalidRedirectionErrorZ
_orig_handlerZ_typeZ_uriZ_allow_redirect)
r Z
methodnameZparams�retr* Zerror_combined_msgr Z
exception_msgZ parse_res�typ)� __class__r
r � _request10 sT
zRetryServer._request1c C s t jj| j|�S )N)�xmlrpc�clientZ_Methodr0 )r �namer
r
r � __getattr__p s zRetryServer.__getattr__)�__name__�
__module__� __qualname__r& r r0 r4 �
__classcell__r
r
)r/ r r
) s @r
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__w s zServerList.__init__c C s | j | j | _| jS )N)r r: Z serverurl)r r
r
r r* { s zServerList.serverc C s( | j d | _ | j t| j�kr d S | j� S )Nr )r: �lenr r* )r r
r
r r$ s zServerList.nextc C s
d| _ d S )Nr )r: )r r
r
r r% � s zServerList.resetServerIndexN)r5 r6 r7 r; r* r$ r% r
r
r
r r9 v s r9 Fc C s\ t j� }tj� }tjtjj� |d }t|t �r:t
j d��|p@d}|d rTtj � }nd }| s�tj
|d�}tj|d�} | r�x$| D ]}
|
|kr~|j|
� q~W n| }t|�} d } d }
|d r�|d p�d } |d p�d }
t| j� || |
|d �}|j| � |jd
tj� � d d � |jjD �}|�rXtj|tj��sNd
td�|f }|j d| � t
j |��|j |� |S )NZ sslCACertz SSL certificate must be a stringz'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERTZ enableProxy)� registrationZenableProxyAuth� proxyUser�
proxyPassword)�proxyZusernameZpassword�timeoutzX-Up2date-Versionc S s" g | ]}t j |�d dkrd�qS )r r T)r+ )r �ir
r
r r � s z getServer..z%s: %sz ERROR: can not find RHNS CA filez%s) r r r �initUp2dateConfig�urllib3Zdisable_warnings�
exceptionsZSubjectAltNameWarning�
isinstance�listr ZValidationErrorZgetProxySettingZ getServerURLZgetFallbackServerURL�appendr9 r
r* r Z
add_headerr �versionr �os�access�R_OKr! r ZSSLCertificateFileNotFoundZset_trusted_cert)ZserverOverriderA r= r �cfgZcaZ rhns_ca_certZ proxyHostZ
serverUrlsZ fallbackURL�urlr r> r? �sZneed_car r
r
r � getServer� sR
rP c O s� t j� }|jd| jd � tj� }d }d}yt|d �}W n tk
rT d}Y nX |dkrbd}�xLd}d }y| ||�}W �n� tk
r� t j
t d���Y �n� t j
k
�rH } z�|jdtj� d |f � ||k�r2tj� d } t| j�dk�r t j
t| jd ���n$t j
| j�r&t| jd �nt| ���nd}W Y d d } ~ X �n* tjk
�rr td� t j
d��Y �n tjk
�r� tj� d } d }
|
d
| j }
|
d | j }
|
d | j }
|j|
� t j
|
��Y �n� t j k
�rP tj� d } |jd
| j |f � | j dk�r0|jd| j � |jd� t!j"| j#�\} } d}
t$| �dk�rt|jd| � ddl%m&} |j'� t$| �dk�r�|jt d�� t j
| j ��t$| �dk�r*t(|d �t(g �k�r�|d }n|d }t(|�t(g �k�rd|d |d |d |d f }n|}d|| f }
|j|
� t j)|
��|
�sL||k�rHt j
| j ��nd}Y n" t j*k
�rp 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 rC �int�
ValueError�KeyboardInterruptr ZCommunicationErrorr! �socketr r r" r) r< �args�strr ZIncompleteReadr �urllib2Z HTTPError�filename�coder r Z
ProtocolError�errmsgZerrcoderN r Z reportErrorZheaders�abs�up2date_clientrR ZupdateLoginInfo�type�FileNotFoundErrorZ
ResponseError�timeZsleep)�methodr\ �kwargsr rM r- Z
attempt_countZattemptsZfailure�er ZerrCodeZerrMsg�resetrR ZpkgZpkgNamer
r
r �doCall� s�
"
"
rk )NNF)%rJ r" r[ rf Z
xmlrpc.clientr1 rD rc r r r r Zrhnr r r^ r+ r � ImportErrorZ http.clientr2 Zurllib.requestZrequestZ urllib.parse�parse�gettextZ translation�t�hasattrr r! r ZServerr
r9 rP rk r
r
r
r � s@
M
C