File: //lib/python3.6/site-packages/up2date_client/__pycache__/rhnreg.cpython-36.pyc
3
m��i* � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d d l m Z m
Z
y4d dlZd dlZd d
lmZmZmZmZmZmZ W nD ek
r� d dljZd dljZe Ze Ze Ze Ze Ze Ze!Z"Y nX d dl#Z#e#j$d d d
�Z%e&e%d��se%j#e%_'e%j'Z(dZ)dZ*ddddd�Z+dd� e+j,� D �Z-d dlm.Z. e.j/� Z0ej1� Z2dd� Z3dd � Z4d d � Z5e e6d �d d!�Z7e6d"�d#d$�Z8d6d&d'�Z9d7e6d(�d)d*�Z:d+d,� Z;e d-�d.d/�Z<d8d0d1�Z=d2d3� Z>d4d5� Z?dS )9� N)� up2dateUtils)�
up2dateErrors)� up2dateAuth)� rhnserver)�
up2dateLog)�
statistics)� getPlatform)�ustr�sstr)�ListType� TupleType�
StringType� UnicodeType�DictType�DictionaryTypezrhn-client-toolsT)Zfallback�ugettextz/etc/sysconfig/rhnz /etc/sysconfig/rhn/jwt.tokenZ
shared_pro�shared�solo�admin)zCloudLinux OS Shared ProzCloudLinux OS SharedzCloudLinux OS SolozCloudLinux OS Adminc C s i | ]\}}||�qS � r )�.0�k�vr r � /usr/lib/python3.6/rhnreg.py�
<dictcomp>4 s r )�configc C s t jtd t j�S )N� systemIdPath)�os�access�cfg�R_OKr r r r �
registered< s r! c C s� t jj| �}t j|t j�s dS t j| t j�rLyt j| | d � W n
dS t j| t jt j B t
dd��}t j |d�}z|j t
|�� W d|j� X dS )z; Write a file to disk that is not readable by other users. Fz.saveZ0600� �wNT)r �path�dirnamer �W_OK�F_OK�rename�open�O_WRONLY�O_CREAT�int�fdopen�writer
�close)Z secure_fileZ
file_contentsZdir_name�fdZfd_filer r r �_write_secure_file? s
r1 c C s t td | �}|S )Nr )r1 r )�systemId�resr r r �
writeSystemIdU s r4 )�token�allowTransitionc C sn ddl }ddl}|jjd�s dS dd| g}|r8|jd� |j|�}|j� \}}|jdkrjtj d||f � dS )zH
Execute binary file which we use as hook for jwt token updates
r Nz /opt/cloudlinux/venv/bin/python3z/usr/sbin/cl-pre-jwt-updatez --new-tokenz--allow-transitionz7Pre jwt update hook failed with stdout=%s and stderr=%s)
�
subprocessr r$ �exists�append�Popen� communicate�
returncode�log�log_me)r5 r6 r7 r �cmd�p�stdout�stderrr r r � _execute_pre_jwt_update_hookZ s
rC )r6 c C sj ddl }ddl}|jjd�s dS dg}| r4|jd� |j|�}|j� \}}|jdkrftj d||f � dS )zH
Execute binary file which we use as hook for jwt token updates
r Nz /opt/cloudlinux/venv/bin/python3z /usr/sbin/cl-post-jwt-updatez--allow-transitionz8Post jwt update hook failed with stdout=%s and stderr=%s)
r7 r r$ r8 r9 r: r; r< r= r> )r6 r7 r r? r@ rA rB r r r � _execute_post_jwt_update_hookm s
rD Fc C s� t j� }y|jj| �}W nF tjk
r. dS tjtjtjtj fk
r^ t
j t j
� � dS X t||� tt|� t|� dS )z�
Get a JWT token from CLN and save it to the file
:param systemId: content of file `/etc/sysconfig/rhn/systemid`
:return: None
N)r Z RhnServerZup2dateZ getJWTTokenr �UnknownMethodExceptionZAuthenticationTicketErrorZRhnUuidUniquenessError�CommunicationErrorZ$AuthenticationOrAccountCreationErrorr= Z
log_exception�sys�exc_inforC r1 � JWT_TOKENrD )r2 r6 Z
xmlrpm_server�resultr r r �getAndWriteJWTTokenToFile� s
rK )�human_readablec C sF d}t jj|�sdS t|�� }|j� jd�}| r4|S t| S Q R X d S )Nz/opt/cloudlinux/cl_editionr �
)r r$ r8 r) �read�strip�_human_readable_to_product)rL Zedition_cache_file�fZ raw_editionr r r �getServerEdition� s
rR c C sV y t | �}W nP tjk
rH } z td|j � tjd� W Y d d }~X n tjk
r\ d S X t� }|d }||krxd S |r�tdj t
| t
| d�� td� tjd� |�sRtj j � s�td� td� t
| � dt
| � d �}d
dd �}|j
|�} | d k �rJtj� }
|
| k�r6td || � d
|
� d�� tjd� n|� d|| � d�}t|� d S )Nz%s� �editionz�WARNING: Automatic registration in yum transactions is only available when edition matches the provided license. Your current edition is {current_edition} and your license is {new_edition}.)Zcurrent_editionZ new_editionz0Run clnreg_ks manually to complete registration.a Error: interactive input required for edition migration, but tool is running in non-interactive mode. Please try running the tool again in interactive shell or add `--migrate-silently` flag to accept allquestions and perform the edition migration silently.za edition installed on your server does not match license you are trying to register server with: zh. Migration is required. You may lose access to the services which are not supported by the new edition.� )r r z@The license you are trying to register with allows a maximum of z% hosting accounts which is less than z) users detected on this server. Aborting.zG Also, the license you are trying to register with allows a maximum of zM hosting accounts. Make sure that your system complies with this requirement.)�checkKeyr rF �print�errmsgrG �exitrE rR �format�_product_to_human_readable�stdin�isatty�getr Zcount_server_users�_askConfirmation) �
activationKeyZ
strictEditionZsilentMigrationZlicenseInformation�eZcurrentEditionZlicenseEdition�messageZedition_to_users_limitZlicense_users_limitZusers_on_serverr r r �checkLicenseKey� sB
rc )�confirmationMessagec C s2 t | � td�}|j� dkr.t d� tjd� dS )zS
Prints message and makes sure that client is ready for edition migration.
z Do you want to continue? [N/y]: �yzAborted.rS N)rW �input�lowerrG rY )rd Zresponser r r r_ � s
r_ c C s� | dkr|dkst d��|tj� tj� tj� d�}tj� }|dk rJ||d< |rlx |j� D ]\}} | ||< qXW |rz||d< n| |d< ||d< |dk r�||d< n(tj j
d �r�d
ntj j
d �r�d nd
|d< t j � }
|
j
j|�} | S )ziWrapper for the old xmlrpc to register a system. Activates subscriptions
if a reg num is given.
Nz)username and password usage is deprecated)Z profile_nameZ
os_releaseZ release_nameZ architectureZ system_idr5 �username�passwordrT z /etc/cloudlinux-edition-solor z /etc/cloudlinux-edition-adminr r )�AssertionErrorr Z
getVersionZ getOSReleaseZgetArchr Z getSystemId�itemsr r$ r8 r �RegistrationRhnServer� registrationZ
new_system) rh ri Z profileNamer5 �otherrT Z auth_dictZ
system_id_xml�key�item�s�retr r r �registerSystem� s.
rs c C s t j� }|jj| �}|S )zG
Check the activation key and return it's edition and customer
)r rl rm Z
license_check)r` rq rr r r r rV s rV c G s� g }x�| D ]�}t |�}|tkr(t|�}nZ|tkrDtdd� |D ��}n>|tkr\dd� |D �}n&|tksl|tkr�tdd� |j � D ��}|j
|� q
W t |�dkr�|d S t|�S dS ) u� All the data we gathered from dmi, bios, gudev are in utf-8,
we need to convert characters beyond ord(127) - e.g ® to unicode.
c s s | ]}t |�V qd S )N)�_encode_characters)r �ir r r � <genexpr>4 s z%_encode_characters.<locals>.<genexpr>c S s g | ] }t |��qS r )rt )r ru r r r �
<listcomp>6 s z&_encode_characters.<locals>.<listcomp>c S s g | ]\}}t ||��qS r )rt )r �name�valr r r rw 8 s rS r N) �typer
r r �tupler r r �dictrk r9 �len)�argsrJ rp Z item_typer r r rt * s
rt )F)F)NNNNNN)@r rG Zup2date_clientr r r r r r Zup2date_client.pkgplatformr Zrhn.i18nr r
ZurlparseZ xmlrpclib�typesr r r
r r r � ImportErrorZ urllib.parse�parseZ
xmlrpc.clientZclient�listr{ �bytes�strr| r, Zlong�gettextZ translation�t�hasattrr �_Z SYSID_DIRrI rP rk r[ r ZinitUp2dateConfigr ZinitLogr= r! r1 r4 �boolrC rD rK rR rc r_ rs rV rt r r r r �<module> sj $
>
*