ruạṛ
a ���f#( � @ s@ d dl Z d dlZd dlZd dlZd dlZd dlZz d dlZW n eyZ d dlm Z Y n0 d dl Z d dl Z d dl m Z d dlmZ d dlmZ dZddd�ddd�ddd�ddd�ddd�d �Zd d � Zd d � Zdd� Zd#dd�Zd$dd�Zdd� Zd%dd�Zdd� Zdd � Z d d � Z G d d � d ej �Z G d!d"� d"ej �Z!dS )&� N)� HTTPAdapter)�Retry)�configz6/var/lib/oracle-cloud-agent/plugins/osms/loginAuth.pkl�1)�version�value�2)zpackages.runTransactionzpackages.rollBackzpackages.verifyzpackages.extended_profilezpackages.updatec C s dd� t �� D �} dd�| �iS )Nc S s& g | ] \}}d ||d |d f �qS )z %s(%s)=%sr r � )�.0�k�vr r �9/usr/share/oracle-cloud-agent/plugins/osms/osms/server.py� <listcomp>$ � z*get_client_caps_header.<locals>.<listcomp>zX-RHN-Client-Capability�,)� CLIENT_CAPS�items�join)�headersr r r �get_client_caps_header# s r c C s� z�t td��} t�| �}W d � n1 s,0 Y |d }|d }t�t�� �d d d }d|d }||krzW d S |t|d � }t � � |kr�t � d ||� W d S |W S t y� } zt � d |� W Y d }~d S d }~0 0 d S ) N�rb�time� loginInfor � system_idzID-%s�X-RHN-Server-Id�X-RHN-Auth-Expire-Offsetz6Cached loginInfo has expired, created: %s, expire: %s.z#Failed to read cached loginInfo: %s) �open�PICKLE_AUTH_FILENAME�pickle�load� xmlrpclib�loadsr � getSystemId�floatr �logging�debug� Exception)� pickle_auth�dataZ created_time� login_infoZid_verZ cached_id_verZ expire_time�errr r r �read_cached_login( s" ( r+ c C s� z�t � � | d�}tj�t�}t�|tj�sFt�|� t�|t dd�� t td��.}t�tt dd�� t � ||� W d � n1 s�0 Y t �d|d | d � W d S ty� } zt �d |� W Y d }~d S d }~0 0 d S ) N)r r Z0700� �wb�0600z<Wrote cacched loginInfo at %s with expiration of %s seconds.r r Tz$Failed to write cached loginInfo: %sF)r �os�path�dirnamer �access�W_OK�mkdir�chmod�intr r �dumpr$ r% r& )r) r( Zpickle_auth_dirr' r* r r r �write_cached_login@ s( � *� r8 Fc C sj |r d}nd}| | sd S t �� }|s*d S | d rZ| d }| d }|rb|rbd|||f }nd| }d|iS ) N�enableExternalProxy� enableProxy�enableProxyAuth� proxyUser� proxyPasswordzhttp://%s:%s@%sz http://%s�http�r �getProxySetting)�cfg�external�key� proxy_hostZ proxy_user�proxy_password�proxyr r r � get_proxies\ s rG c C sX d }d }d }|rd}nd}| | r6t �� }|r6d| }| d rN| d }| d }|||fS )Nr9 r: zhttp://r; r< r= r? )rA rB � proxy_url�proxy_usernamerE rC rD r r r � get_proxyp s rJ c C s4 | d }|r t |t�r |d }t�|tj�r0|S d S )N� sslCACertr )� isinstance�listr/ r2 �R_OK)rA Zcacertr r r � get_cacert� s rO c C s` t � }| d r| d |d< d|d< |r\g d�}|D ](}||vrJtd| ��t|| �||< q2|S )N�tenantIdz X-Tenant-Idzfollow-redirects=3zX-RHN-Transport-Capability)r zX-RHN-Auth-User-Idz X-RHN-AuthzX-RHN-Auth-Server-Timer z%Missing required login information %s)r r& �str)rA r) r �needed_headers�headerr r r � get_headers� s rT c C sj | j }| j�dd��� dkr&t�|�}| j�dd��� dkrFt�|�}| j�dd��� dkrft�|�}|S )NzContent-Transfer-Encoding� �base64zContent-Encodingzx-zlibzx-gzip) �contentr �get�lowerrV � b64decode�zlib� decompress�gzip)�responserW r r r �decode_response_content� s r_ c K s$ | � � D ]\}}t�d||� qd S )Nzrequest parameter %s: %s)r r$ r% )�kwargsrC r r r r � log_request� s ra c C sF t �d| jj| jj� t �d| jj� t �d| j� t �d| j� d S )Nz request method and url: %s %szrequest headers: %szresponse status code: %szresponse headers: %s)r$ r% �request�method�urlr � status_code)r^ r r r � log_response� s rf c @ s( e Zd Zd dd�Zdd� Zd d d �ZdS ) �RequestsTransportTN�x c K s< || _ || _|| _|| _|| _d | _tjj| fi |�� d S )N) � use_https�verify�proxies�timeoutr � last_responser � SafeTransport�__init__)�selfri rj rk rl r r` r r r ro � s zRequestsTransport.__init__c C s& t dg d�g d�dd�}t|d�}|S )N� )i� i� i� i� )�HEAD�GETZOPTIONS�POST� )�totalZstatus_forcelistZmethod_whitelistZbackoff_factor)Z max_retries)r r )rp Zretry_strategyZadapterr r r � retry_adapter� s � z RequestsTransport.retry_adapterFc C s | j r dnd}d|||f }t�� }|�|| �� � | jrT|rL| j�|�} qX| j} n|} d } t|| j| | j | j | d� |j ||| j| | j | j | d�} t | � z | � � W n> tjy� } z$t�|| jt| �| j��W Y d } ~ n d } ~ 0 0 | | _t| �} | �� \}}|�| � |�� |�� S )N�httpsr> z %s://%s%s)rd rj �authrk rl r )r( rj ry rk rl r )ri �requestsZSession�mountrw r �updatera rj rk rl �postrf Zraise_for_statusZRequestExceptionr � ProtocolErrorre rQ rm r_ Z getparser�feed�close)rp �host�handlerZ request_body�verboser �schemerd �sessionZcombined_headersry r^ �erW �p�ur r r rb � sL �� . zRequestsTransport.request)TNNrh N)FN)�__name__� __module__� __qualname__ro rw rb r r r r rg � s rg c @ s0 e Zd Zd dd�Zdd� Zdd� Zd d d �Zd S )� OsmsServerrh c K sv t �� }t �� d }|�d�}d }|r.t|�}t||t|�|t|�d�| _t j j | || jfi |�� d | _ | � � d S )Nr rx )rj rk rl r )r �initUp2dateConfig� getServerlURL� startswithrO rg rG rT Z transportr � ServerProxyro r) �login)rp rl r` rA Z server_urlri rj r r r ro � s �zOsmsServer.__init__c C s� z| � | ||�W S tjy� } zlz tt|j�d���dk}W n tyT d}Y n0 |s^|�t� d� | j dd� | � | ||�W Y d }~S d }~0 0 d S )NzX-RHN-Fault-Code�" Fzretry on auth errorT)� force_update) �_OsmsServer__request1r r~ �absr6 r rX r& r$ r% r� )rp � methodname�paramsr* Z is_auth_errorr r r Z __request s zOsmsServer.__requestc C sV t j||| j| jd��| jd�}| jj| j| j|| j | j d�}t |�dkrR|d }|S )N)�encodingZ allow_none�xmlcharrefreplace)r� r � r ) r �dumpsZ_OsmsServer__encodingZ_OsmsServer__allow_none�encodeZ_OsmsServer__transportrb Z_OsmsServer__hostZ_OsmsServer__handlerZ_OsmsServer__verboser) �len)rp r� r� rb r^ r r r Z __request1 s$ ��� zOsmsServer.__request1Fc C sf t �� }|sd | _d S |s(t� }|s(d}|r@| j�|�}t|� t| jt�rZ| j� |� n|| _| jS )NT) r r" r) r+ �up2dater� r8 rL �dictr| )rp r� r r) r r r r� * s zOsmsServer.loginN)rh )F)r� r� r� ro Z_OsmsServer__requestr� r� r r r r r� � s r� )F)F)N)"rV r] r$ r/ r r r � ImportErrorZ xmlrpc.client�clientr[ rz Zrequests.adaptersr Z$requests.packages.urllib3.util.retryr �osmsr r r r r+ r8 rG rJ rO rT r_ ra rf rn rg r� r� r r r r �<module> sB � ?
cải xoăn