File: //usr/lib/python3.9/site-packages/up2date_client/__pycache__/up2dateAuth.cpython-39.pyc
a
0Q=ji4 � @ sB d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ zd dlm Z W n e
yb e Z Y n0 d dl m
Z
d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ dad Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zd)dd�Z d*dd �Z d+d d �Z G d d � d e �Z d!d"� Z!d#d$� Z"d%d&� Z#d'd(� Z$dS ),� N)� rpcServer)�DictType)�rpclib)�config)� rhnserver)�
up2dateErrors)�
up2dateLog)� up2dateUtils)�
statisticsz /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 s� t �� } t� }d}|du r |S zFtj�|�d d }|�dd�}|dv rN|}n| �d||f � |W S ty� | �d� | j t
� � � Y n0 |S )z�
Extract the preferred_interface parameter from system_id XML
Returns 'IPv4' or 'IPv6' if specified, otherwise 'IPv4' as default
�IPv4Nr �preferred_interface)r ZIPv6zGInvalid preferred_interface value '%s' in system_id, defaulting to '%s'zGFailed to parse system_id XML, preferred_interface defaulting to 'IPv4') r �initLogr r � xmlrpclib�loads�get�log_me� Exception�
log_exception�sys�exc_info)�log�systemIdr �paramsZ
cfg_interfacer r r �getPreferredInterface( s( ��
r) c
C s� t j�d�rd S t�� } z tj�t� �d d d }W n Y dS 0 t �
� }| d �s�||k�r�t � � }|j
�t� |�}| d }|d |�d�� }t �|t j�s�zt �|� W n Y dS 0 t �|t j�s�dS t j�|��r|d }zt �||� W n Y dS 0 t �|t jt jB t jB t jB t jB d�}zt � |d �} W n t �y` t � |� � Y n0 z| � |� W | � � n
| � � 0 d S )
NZLEAPP_IPU_IN_PROGRESSr Z
os_releaseZchannelOverrider �/z.save� �w) r �environr r r
r r r r r Z
getVersionr ZRegistrationRhnServerZ registrationZupgrade_version�rfindr �W_OK�mkdirr �lexists�renamer �O_WRONLY�O_CREAT�O_TRUNC�
O_NOFOLLOW� O_CLOEXEC�fdopenr" r �write)
r �idVerZ systemVer�sZ newSystemIdr �dirZsavePath�fdr r r r �maybeUpdateVersionL sF (
r> c C s~ t �� } | �d� ts$| �d� dS t�� td�}tj�t�}t� |tj
�szzt� |d� W n | � d| � Y dS 0 z�tj�
t�r�t�t� t�ttjtjB tjB tjB tjB d�}z t�|d� t�|d�}W n ty� t�|� � Y n0 |� t�||� W d � n1 �s 0 Y W n$ t�yN | � d
t � Y dS 0 |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� loginInfoi� z'Unable to write pickled loginInfo to %sr+ r, Nz Unable to write loginInfo to %sr? �X-RHN-Auth-Expire-OffsetzWrote pickled loginInfo at z with expiration of z seconds.T) r r � log_debugr@ r? r r �dirname�pcklAuthFileNamer r/ r0 r! r1 �unlinkr r3 r4 r5 r6 r7 �fchmodr8 r" r �json�dump�float)r& �dataZpcklDirr= �pcklAuth�
expireTimer r r �writeCachedLogin� sL
�
�
0�rM 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: %sFr zSUnexpected EOF. Probably an empty file, regenerate auth filer Z system_idzID-%sr@ �X-RHN-Server-Idz#system id version changed: %s vs %sr? rA 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 r rB r r rD r r rG �load�EOFError�
ValueErrorr r r r r r? rI �_updateLoginInfo) r& rK rJ r: ZcidVerZ createdTime�liZ currentTimerL r r r �readCachedLogin� sJ
�
� rT c C s2 t | �tkr*t t�tkr$t�| � q.| andadS )z,
Update the global var, "loginInfo"
N)�typer r@ �update)rS r r r rR � s
rR Fc s� t �� }|�d| � |s(ts(t� r(tS tj|d�}| d krBt� } | sJd S t� |� d� |j
� | t �
� t|�t�� �}t|� t� tr�|� d� ttt�r�td�� � fdd�t�� D �}|�d|� n|�d tt�j � tS )
Nz login(forceUpdate=%s) invoked��timeoutzlogging into up2date serverz?successfully retrieved authentication token from up2date server)rN zX-RHN-Auth-User-IdzX-RHN-Auth-Server-TimerA zX-RHN-Auth-Channelsc s" i | ]\}}||� v r|nd �qS )z
<REDACTED>r )�.0�k�v�Z_SAFE_LOGIN_KEYSr r �
<dictcomp>" s �zlogin.<locals>.<dictcomp>z
logininfo:zlogininfo: <non-dict %s>)r r rB r@ rT r Z RhnServerr r> r! Zup2date�login�socketZgetfqdn�_get_panel_namer
Zcount_server_usersrR rM �
isinstance�dict� frozenset�itemsrU �__name__)r' � forceUpdaterX r& ZserverrS Zredactedr r\ r r^ � s8
�
�r^ c C s0 t �� }|�d� td| d� ts,t�d��tS )Nz updateLoginInfo() login infoT)rf rX zUnable to authenticate)r r r! r^ r@ r ZAuthenticationError)rX r& r r r �updateLoginInfo+ s
rg c C s6 zt a W n ty d a Y n0 t r(t S t| d� t S )NrW )r@ � NameErrorr^ rW r r r � getLoginInfo5 s
ri c @ s e Zd ZdS )�_FailedToGetPanelNameN)re �
__module__� __qualname__r r r r rj B s rj c C sJ z
t � }W n6 ty@ t� s6| jt�� � | �d� t� }Y n0 |�� S )Nz?Failed to get panel name via cldetect, using fallback mechanism) � _get_panel_name_via_cldetectrj �_is_cldeploy_runningr# r$ r% r! �_fallback_get_panel_name�lower)r& Z
panel_namer r r r` F s
r` c C s� d} d}d }t j�| �rLt| �� }|�� �� }W d � n1 sB0 Y |r�d| }t j�|�r�t|�� }d|�� v }W d � n1 s�0 Y |S )Nz/var/lock/cldeploy.lckFz/proc/%s/cmdlineZcldeploy)r r �existsr r �strip)Zlock_file_pathZcldeploy_running�pidr Zpid_cmdline_pathr r r rn S s
*
*rn c C sr ddg \} }}t j�| �s(td| ��tj|tjtjdd�}|�� \}}|jdkrjtdd� |�|f ��|�
� S ) Nz/usr/bin/cldetectz--detect-cp-nameonlyz7Failed to obtain panel name because '%s' does not existT)�stdout�stderrZuniversal_newlinesr z:Failed to obtain panel name using '%s' command; stderr: %s� ) r r rq rj �
subprocess�Popen�PIPEZ communicate�
returncode�joinrr )Zbinary�_�cmdZprocessrt ru r r r rm d s ��
��rm c C sd t j�d�rdS t j�d�r dS t j�d�r0dS t j�d�r@dS t j�d �rPd
S t j�d �r`d S d
S )Nz/usr/local/psa/admin/Zpleskz/usr/local/interworx/Z interworxz#/usr/local/cpanel/whostmgr/docroot/Zcpanelz/usr/local/ispmgr/Zispmgrz/usr/local/directadmin/Z directadminz/usr/local/mgr5/sbin/mgrctlZispmgr5�unknown)r r �isdir�isfiler r r r ro w s ro )NFN)N)N)%rG r r$ r? r_ rw Zup2date_clientr �typesr � ImportErrorrb Zrhnr r r r r r r
r@ rD r r) r> rM rT rR r^ rg ri r" rj r` rn rm ro r r r r �<module> sB
$69,
3