3
�Xji4 � @ sD d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ yd dlm Z W n e
k
rd e Z Y nX 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 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� } t� }d}|dkr |S yDtjj|�d d }|jdd�}|dkrN|}n| jd||f � |S tk
r� | jd� | j t
j � � Y nX |S ) z�
Extract the preferred_interface parameter from system_id XML
Returns 'IPv4' or 'IPv6' if specified, otherwise 'IPv4' as default
�IPv4Nr �preferred_interface�IPv6zGInvalid preferred_interface value '%s' in system_id, defaulting to '%s'zGFailed to parse system_id XML, preferred_interface defaulting to 'IPv4')r r ) r �initLogr r � xmlrpclib�loads�get�log_me� Exception�
log_exception�sys�exc_info)�log�systemIdr ZparamsZ
cfg_interfacer r r �getPreferredInterface( s$
r) c
$ C sv t jjd�rd S tj� } y tjjt� �d d d }W n
dS t j
� }| d oZ||k�rrt j � }|j
jt� |�}| d }|d |jd�� }t j|t j�s�yt j|� W n
dS t j|t j�s�dS t jj|�� r�|d }yt j||� W n
dS t j|t jt jB t jB t jB t jB d�}yt j |d �} W n" t k
�rV t j |� � Y nX z| j |� W d | j � X d S )
NZLEAPP_IPU_IN_PROGRESSr Z
os_releaseZchannelOverrider �/z.savei� �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 sf t j� } | jd� ts$| jd� dS tj� td�}tjjt�}tj |tj
�svytj |d� W n | j d| � dS y�tjj
t�r�tjt� tjttjtjB tjB tjB tjB d�}y tj|d� tj|d�}W n tk
r� tj|� � Y nX |� tj||� W d Q R X W n$ tk
�r6 | j d
t � dS X |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� loginInfoi� z'Unable to write pickled loginInfo to %si� r+ Nz Unable to write loginInfo to %sr> zX-RHN-Auth-Expire-OffsetzWrote pickled loginInfo at z with expiration of z seconds.T) r r � log_debugr? r> r r �dirname�pcklAuthFileNamer r. r/ r" r0 �unlinkr r2 r3 r4 r5 r6 �fchmodr7 r# r �json�dump�float)r' �dataZpcklDirr< �pcklAuth�
expireTimer r r �writeCachedLogin� sF
rK 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: %sFr zSUnexpected 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 rG �_updateLoginInfo) r' rI rH r9 ZcidVerZ createdTime�liZ currentTimerJ r r r �readCachedLogin� sH
rQ c C s2 t | �tkr*t t�tkr$tj| � q.| andadS )z,
Update the global var, "loginInfo"
N)�typer r? �update)rP r r r rO � s
rO Fc s� t j� }|jd| � | r,t r,t� r,tS tj|d�}| d krFt� } | sNd S t� |j d� |j
j | t j
� t|�tj� �}t|� t� tr�|j d� ttt�r�td�� � fd
d �tj� D �}|jd |� n|jd
tt�j � tS )Nz login(forceUpdate=%s) invoked)�timeoutzlogging into up2date serverz?successfully retrieved authentication token from up2date server�X-RHN-Server-Id�X-RHN-Auth-User-Id�X-RHN-Auth-Server-Time�X-RHN-Auth-Expire-Offset�X-RHN-Auth-Channelsc s" i | ]\}}|� kr|nd |�qS )z
r )�.0�k�v)�_SAFE_LOGIN_KEYSr r �
" s zlogin..z
logininfo:zlogininfo: )rU rV rW rX rY )r r r@ r? rQ r Z RhnServerr r= r" Zup2date�login�socketZgetfqdn�_get_panel_namer
Zcount_server_usersrO rK �
isinstance�dict� frozenset�itemsrR �__name__)r( � forceUpdaterT r' ZserverrP Zredactedr )r] r r_ � s>
r_ c C s0 t j� }|jd� td| d� ts,tjd��tS )Nz updateLoginInfo() login infoT)rg rT zUnable to authenticate)r r r" r_ r? r ZAuthenticationError)rT r' r r r �updateLoginInfo+ s
rh c C s8 yt a W n tk
r d a Y nX t r*t S t| d� t S )N)rT )r? � NameErrorr_ )rT r r r � getLoginInfo5 s
rj c @ s e Zd ZdS )�_FailedToGetPanelNameN)rf �
__module__� __qualname__r r r r rk B s rk c C sL y
t � }W n8 tk
rB t� s8| jtj� � | jd� t� }Y nX |j� S )Nz?Failed to get panel name via cldetect, using fallback mechanism) � _get_panel_name_via_cldetectrk �_is_cldeploy_runningr$ r% r& r" �_fallback_get_panel_name�lower)r' Z
panel_namer r r ra F s
ra c C st d} d}d }t jj| �r8t| ��}|j� j� }W d Q R X |rpd| }t jj|�rpt|��}d|j� k}W d Q R X |S )Nz/var/lock/cldeploy.lckFz/proc/%s/cmdlineZcldeploy)r r �existsr r �strip)Zlock_file_pathZcldeploy_running�pidr Zpid_cmdline_pathr r r ro S s
ro c C sr ddg \} }}t jj| �s(td| ��tj|tjtjdd�}|j� \}}|jdkrjtddj |�|f ��|j
� 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 rr rk �
subprocess�Popen�PIPEZ communicate�
returncode�joinrs )Zbinary�_�cmdZprocessru rv r r r rn d s
rn c C sd t jjd�rdS t jjd�r dS t jjd�r0dS t jjd�r@dS t jjd �rPd
S t jjd �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 rp w s rp )NFN)N)N)%rE r r% r> r` rx Zup2date_clientr �typesr � ImportErrorrc Zrhnr r r r r r r
r? rB r r) r= rK rQ rO r_ rh rj r# rk ra ro rn rp r r r r � sB
$69,
3