File: //usr/lib/python3.6/site-packages/up2date_client/__pycache__/up2dateAuth.cpython-36.pyc
3
U��Z" � @ s� d dl Z d dlZd dlZyd dlmZ W n ek
r@ eZY nX d dlmZ d dl m
Z
d dl m Z d dl m Z d dl m
Z
d dl mZ d d l mZ dad
Zd d � Zd
d� Zdd� Zdd� Zdd� Zd dd�Zd dd�Zd dd�ZdS ) � N)�DictType)�rpclib)�
clientCaps)�config)� rhnserver)�
up2dateErrors)�
up2dateLog)� up2dateUtilsz /var/spool/up2date/loginAuth.pklc C s@ t j� } | d }tj|tj�s"d S t|d�}|j� }|j� |S )N� systemIdPath�r)r �initUp2dateConfig�os�access�R_OK�open�read�close)�cfg�path�f�ret� r �!/usr/lib/python3.6/up2dateAuth.py� getSystemId s
r c C s& t j� } y tjjt� �d d d }W n
dS tj� }||k�r"tj � }|j
j t� |�}| d }|d |j d�� }t
j|t
j�s�yt
j|� W n
dS t
j|t
j�s�dS t
j|t
j�r�|d }yt
j||� W n
dS t|d�}|j|� |j� yt
j|tdd�� W n Y nX d S ) Nr Z
os_releaser
�/z.save�w�0600� )r r r � xmlrpclib�loadsr r Z
getVersionr � RhnServerZ registrationZupgrade_version�rfindr
r �W_OK�mkdir�F_OK�renamer �writer �chmod�int) r �idVerZ systemVer�sZ newSystemIdr �dirZsavePathr r r r �maybeUpdateVersion& s>
r, c C s� t j� } | jd� ts$| jd� dS tj� td�}tjjt�}tj |tj
�s�y tj |� tj |t
dd�� W n | jd| � dS ttd�}tj tt
d d�� tj||� |j� |d
ttd � }| jd |d
d
|d� dS )z�
Pickle loginInfo to a file
Returns:
True -- wrote loginInfo to a pickle file
False -- did _not_ write loginInfo to a pickle file
zwriteCachedLogin() invokedz1writeCachedLogin() loginInfo is None, so bailing.F)�time� loginInfoZ0700r z'Unable to write pickled loginInfo to %s�wbr r- zX-RHN-Auth-Expire-OffsetzWrote pickled loginInfo at z with expiration of z seconds.T)r �initLog� log_debugr. r- r
r �dirname�pcklAuthFileNamer r" r# r' r( �log_mer �pickle�dumpr �float)�log�dataZpcklDir�pcklAuth�
expireTimer r r �writeCachedLoginP s.
r< c C sT t j� } | jd� tjttj�s2| jdt � dS ttd�}ytj |�}W n* t
t fk
rt | jd� |j � dS X |j � yLt
jjt� �d d d }d|d d
}||kr�| jd ||f � dS W n Y nX |d }|d }tj� }|t|d
� }| jd|d|dt|d
�� ||k�r<| jd||f � dS t|� | jd|� dS )zb
Read pickle info from a file
Caches authorization info for connecting to the server.
zreadCachedLogin invokedz'Unable to read pickled loginInfo at: %sF�rbzSUnexpected EOF. Probably an empty file, regenerate auth filer Z system_idzID-%sr. zX-RHN-Server-Idz#system id version changed: %s vs %sr- zX-RHN-Auth-Expire-Offsetz(Checking pickled loginInfo, currentTime=z
, createTime=z, expire-offset=z9Pickled loginInfo has expired, created = %s, expire = %s.z<readCachedLogin(): using pickled loginInfo set to expire at T)r r0 r1 r
r r3 r r r5 �load�EOFError�
ValueErrorr r r r r r- r7 �_updateLoginInfo) r8 r: r9 r) ZcidVerZ createdTime�liZ currentTimer; r r r �readCachedLoginp sH
rC c C s2 t | �tkr*t t�tkr$tj| � q.| andadS )z,
Update the global var, "loginInfo"
N)�typer r. �update)rB r r r rA � s
rA Fc C s� t j� }|jd| � | r,t r,t� r,tS tj|d�}tjj � }x |D ]\}}|j
||� qHW | d krnt � } | svd S t � |j
d� |jj| �}|jj� t|� t� tr�|j
d� |jdt� tS )Nz login(forceUpdate=%s) invoked)�timeoutzlogging into up2date serverz?successfully retrieved authentication token from up2date serverz
logininfo:)r r0 r1 r. rC r r r ZcapsZ headerFormatZ
add_headerr r, r4 Zup2date�loginZ capabilitiesZvalidaterA r< ) ZsystemId� forceUpdaterF r8 ZserverZ
headerlistZ
headerName�valuerB r r r rG � s.
rG c C s0 t j� }|jd� td| d� ts,tjd��tS )Nz updateLoginInfo() login infoT)rH rF zUnable to authenticate)r r0 r4 rG r. r ZAuthenticationError)rF r8 r r r �updateLoginInfo� s
rJ c C s8 yt a W n tk
r d a Y nX t r*t S t| d� t S )N)rF )r. � NameErrorrG )rF r r r � getLoginInfo� s
rL )NFN)N)N)r
r5 r- �typesr � ImportError�dictZrhnr Zup2date_clientr r r r r r r. r3 r r, r< rC rA rG rJ rL r r r r �<module> s.
* ,
%