XylotrechusZ
3
`��g�% � @ s< 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� Zd'dd�Zd(dd�Z d)dd �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 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 sB t jjd�rd S tj� } y tjjt� �d d d }W n
dS t j
� }| d oZ||k�r>t j � }|j
jt� |�}| d }|d |jd�� }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|td d
�� W n Y nX d S ) NZLEAPP_IPU_IN_PROGRESSr Z
os_releaseZchannelOverrider �/z.save�w�0600� )r �environ�getr r
r � xmlrpclib�loadsr r
Z
getVersionr � RhnServerZ registrationZupgrade_version�rfindr �W_OK�mkdir�F_OK�renamer �writer �chmod�int) r �idVerZ systemVer�sZ newSystemIdr �dirZsavePathr r r r �maybeUpdateVersion* sB
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 r0 zX-RHN-Auth-Expire-OffsetzWrote pickled loginInfo at z with expiration of z seconds.T)r �initLog� log_debugr1 r0 r r �dirname�pcklAuthFileNamer r% r& r* r+ �log_mer �pickle�dumpr �float)�log�dataZpcklDir�pcklAuth�
expireTimer r r �writeCachedLoginX 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-%sr1 zX-RHN-Server-Idz#system id version changed: %s vs %sr0 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 r3 r4 r r r6 r r r8 �load�EOFError�
ValueErrorr r r! r" r r0 r: �_updateLoginInfo) r; r= r< r, ZcidVerZ createdTime�liZ currentTimer> r r r �readCachedLoginx sH
rF c C s2 t | �tkr*t t�tkr$tj| � q.| andadS )z,
Update the global var, "loginInfo"
N)�typer r1 �update)rE r r r rD � s
rD 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| tj� t|��}|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 r3 r4 r1 rF r r# r ZcapsZ headerFormatZ
add_headerr r/ r7 Zup2date�login�socketZgetfqdn�_get_panel_nameZ capabilitiesZvalidaterD r? ) ZsystemId� forceUpdaterI r; ZserverZ
headerlistZ
headerName�valuerE r r r rJ � s.
rJ c C s0 t j� }|jd� td| d� ts,tjd��tS )Nz updateLoginInfo() login infoT)rM rI zUnable to authenticate)r r3 r7 rJ r1 r ZAuthenticationError)rI r; r r r �updateLoginInfo� s
rO c C s8 yt a W n tk
r d a Y nX t r*t S t| d� t S )N)rI )r1 � NameErrorrJ )rI r r r � getLoginInfo� s
rQ c @ s e Zd ZdS )�_FailedToGetPanelNameN)�__name__�
__module__� __qualname__r r r r rR � s rR 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_cldetectrR �_is_cldeploy_runningZ
log_exception�sys�exc_infor7 �_fallback_get_panel_name�lower)r; Z
panel_namer r r rL � s
rL 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 rW s
rW 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 r\ rR �
subprocess�Popen�PIPEZ communicate�
returncode�joinr] )Zbinary�_�cmdZprocessr_ r` r r r rV s
rV 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 rZ & s rZ )NFN)N)N)$r r8 rX r0 rK rb Zup2date_clientr �typesr � ImportError�dictZrhnr r r r r r r
r1 r6 r r/ r? rF rD rJ rO rQ � ExceptionrR rL rW rV rZ r r r r �<module> s@
. ,
%