File: //lib/python3.9/site-packages/up2date_client/__pycache__/up2dateAuth.cpython-39.pyc
a
7
fg � @ s� d dl Z d dlZd dlZd dlZd dlmZ zd dlmZ W n eyR e ZY n0 d dl
m Z d dlm Z d dlm
Z
d dlmZ d dlmZ d d lmZ d d
lmZ dad Zd d
� Zdd� Zdd� Zdd� Zdd� Zd dd�Zd dd�Zd dd �ZdS ) � N)� rpcServer)�DictType)�rpclib)�
clientCaps)�config)� rhnserver)�
up2dateErrors)�
up2dateLog)� up2dateUtilsz /var/spool/up2date/loginAuth.pklc C s@ t �� } | d }t�|tj�s"d S t|d�}|�� }|�� |S )N� systemIdPath�r)r �initUp2dateConfig�os�access�R_OK�open�read�close)�cfg�path�f�ret� r �>/usr/lib/python3.9/site-packages/up2date_client/up2dateAuth.py� getSystemId s
r c C s2 t �� } z tj�t� �d d d }W n Y dS 0 t�� }||k�r.t� � }|j
� t� |�}| d }|d |� d�� }t
�|t
j�s�zt
�|� W n Y dS 0 t
�|t
j�s�dS t
�|t
j�r�|d }zt
�||� W n Y dS 0 t|d�}|�|� |�� zt
�|tdd�� W n Y n0 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 �� } | �d� ts$| �d� dS t�� td�}tj�t�}t� |tj
�s�z t� |� t� |t
dd�� W n | �d| � Y dS 0 ttd�}t� tt
d d�� t�||� |�� |d
ttd � }| �d |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. �X-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 �writeCachedLoginR s2
�
�r> c
C sV t �� } | �d� t�ttj�s2| �dt � dS ttd�}zt� |�}W n* t
t fyt | �d� |� � Y dS 0 |� � zNt
j�t� �d d d }d|d d
}||kr�| �d ||f � W dS W n Y n0 |d }|d }t�� }|t|d
� }| �d|d|dt|d
�� ||k�r>| �d||f � dS t|� | �d|� 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. r1 z(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 r2 r3 r r r5 r r r7 �load�EOFError�
ValueErrorr r r r r r. r9 �_updateLoginInfo) r: r< r; r* ZcidVerZ createdTime�liZ currentTimer= r r r �readCachedLoginr sJ
�
� rE c C s2 t | �tkr*t t�tkr$t�| � q.| andadS )z,
Update the global var, "loginInfo"
N)�typer r/ �update)rD r r r rC � s
rC Fc C s� t �� }|�d| � |s(ts(t� r(tS tj|d�}tj� � }|D ]\}}|�
||� qB| d krft � } | snd S t � |�
d� |j�| t�� �}|j�� t|� t� tr�|�
d� |�dt� tS )Nz login(forceUpdate=%s) invoked��timeoutzlogging into up2date serverz?successfully retrieved authentication token from up2date serverz
logininfo:)r r2 r3 r/ rE r r! r ZcapsZ headerFormatZ
add_headerr r- r6 Zup2date�login�socketZgetfqdnZ capabilitiesZvalidaterC r> ) ZsystemId� forceUpdaterI r: ZserverZ
headerlistZ
headerName�valuerD r r r rJ � s.
rJ c C s0 t �� }|�d� td| d� ts,t�d��tS )Nz updateLoginInfo() login infoT)rL rI zUnable to authenticate)r r2 r6 rJ r/ r ZAuthenticationError)rI r: r r r �updateLoginInfo� s
rN c C s6 zt a W n ty d a Y n0 t r(t S t| d� t S )NrH )r/ � NameErrorrJ rH r r r � getLoginInfo� s
rP )NFN)N)N) r r7 r. rK Zup2date_clientr �typesr � ImportError�dictZrhnr r r r r r r
r/ r5 r r- r> rE rC rJ rN rP r r r r �<module> s2
* ,
%