File: //lib/python3.9/site-packages/up2date_client/__pycache__/up2dateAuth.cpython-39.pyc
a
c8h�* � @ 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)�
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 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 sN t j�d�rd S t�� } z tj�t� �d d d }W n Y dS 0 t �
� }| d �sJ||k�rJt � � }|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 �|td d
�� W n Y n0 d S ) NZLEAPP_IPU_IN_PROGRESSr Z
os_releaseZchannelOverrider �/z.save�w�0600� )r �environr r r
r r r r r
Z
getVersionr ZRegistrationRhnServerZ registrationZupgrade_version�rfindr �W_OK�mkdir�F_OK�renamer �writer �chmod�int) r �idVerZ systemVer�sZ newSystemIdr �dirZsavePathr r r r �maybeUpdateVersionL sB
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 r � log_debugr< r; r r �dirname�pcklAuthFileNamer r0 r1 r5 r6 r! r �pickle�dumpr �float)r& �dataZpcklDir�pcklAuth�
expireTimer r r �writeCachedLoginz s2
�
�rH 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; r> 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 r? r r rA r r rB �load�EOFError�
ValueErrorr r r r r r; rD �_updateLoginInfo) r& rF rE r7 ZcidVerZ createdTime�liZ currentTimerG r r r �readCachedLogin� sJ
�
� rO c C s2 t | �tkr*t t�tkr$t�| � q.| andadS )z,
Update the global var, "loginInfo"
N)�typer r< �update)rN r r r rM � s
rM Fc C s� t �� }|�d| � |s(ts(t� r(tS tj|d�}tj� � }|D ]\}}|�
||� qB| d krft � } | snd S t � |�
d� |j�| t�� 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 r r? r< rO r Z RhnServerr ZcapsZ headerFormatZ
add_headerr r: r! Zup2date�login�socketZgetfqdn�_get_panel_nameZ capabilitiesZvalidaterM rH ) r' � forceUpdaterS r& ZserverZ
headerlistZ
headerName�valuerN r r r rT � s.
rT c C s0 t �� }|�d� td| d� ts,t�d��tS )Nz updateLoginInfo() login infoT)rW rS zUnable to authenticate)r r r! rT r< r ZAuthenticationError)rS r& r r r �updateLoginInfo� s
rY c C s6 zt a W n ty d a Y n0 t r(t S t| d� t S )NrR )r< � NameErrorrT rR r r r � getLoginInfo s
r[ c @ s e Zd ZdS )�_FailedToGetPanelNameN)�__name__�
__module__� __qualname__r r r r r\ s r\ 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_cldetectr\ �_is_cldeploy_runningr# r$ r% r! �_fallback_get_panel_name�lower)r& Z
panel_namer r r rV s
rV 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 ra $ s
*
*ra 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 rd r\ �
subprocess�Popen�PIPEZ communicate�
returncode�joinre )Zbinary�_�cmdZprocessrg rh r r r r` 5 s ��
��r` 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 rb H s rb )NFN)N)N)%r rB r$ r; rU rj Zup2date_clientr �typesr � ImportError�dictZrhnr r r r r r r
r< rA r r) r: rH rO rM rT rY r[ r" r\ rV ra r` rb r r r r �<module> sB
$. ,
%