File: //lib/python3.6/site-packages/up2date_client/__pycache__/rhnreg.cpython-36.pyc
3
/V)j�. � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d d l m
Z
mZ y4d dlZd dlZd d
lmZmZmZmZmZmZ W nD ek
r� d dljZd dlj Ze Ze Ze Ze Ze!Ze!Ze"Z#Y nX d dl$Z$e$j%d d d
�Z&e'e&d��s e&j$e&_(e&j(Z)dZ*dZ+ddddd�Z,dd� e,j-� D �Z.d dlm/Z/ e/j0� Z1ej2� Z3dd� Z4dd � Z5d d � Z6e e7d �d d!�Z8e7d"�d#d$�Z9d6d&d'�Z:d7e7d(�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>5 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 jj| �rJyt j| | d � W n
dS t j| t jt j B t j
B t j B t j B d�}yt j
|d�}W n tk
r� t j|� � Y nX z|jt|�� W d|j� X dS )z; Write a file to disk that is not readable by other users. Fz.savei� �wNT)r �path�dirnamer �W_OK�lexists�rename�open�O_WRONLY�O_CREAT�
O_NOFOLLOW�O_EXCL� O_CLOEXEC�fdopen� Exception�close�writer
)Z secure_fileZ
file_contentsZdir_name�fdZfd_filer r r �_write_secure_file@ s*
r3 c C s t td | �}|S )Nr )r3 r )�systemId�resr r r �
writeSystemIdd s r6 )�token�allowTransitionc C sz ddl }ddl}|jjd�s dS ddg}|r6|jd� |j||jd�}|j| j� d�\}}|j dkrvt
j 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-token-stdinz--allow-transition)�stdin)�inputz7Pre jwt update hook failed with stdout=%s and stderr=%s) �
subprocessr r# �exists�append�Popen�PIPE� communicate�encode�
returncode�log�log_me)r7 r8 r; r �cmd�p�stdout�stderrr r r � _execute_pre_jwt_update_hooki s
rI )r8 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)
r; r r# r< r= r> r@ rB rC rD )r8 r; r rE rF rG rH r r r � _execute_post_jwt_update_hook| s
rJ Fc C s� t j� }y|jj| �}W nF tjk
r. dS tjtjtjtj fk
r^ t
j t j
� � dS X t|t� s�t|�dks�tjd|� r�t|ttf�r�t|�nd}t
jdt|�j|f � dS 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
Ni @ z2\A[A-Za-z0-9_-]+\.[A-Za-z0-9_-]+\.[A-Za-z0-9_-]+\Zzn/azIInvalid JWT token received from server (type=%s, len=%s), skipping update)r Z RhnServerZup2dateZ getJWTTokenr �UnknownMethodExceptionZAuthenticationTicketErrorZRhnUuidUniquenessError�CommunicationErrorZ$AuthenticationOrAccountCreationErrorrC Z
log_exception�sys�exc_info�
isinstance�str�len�re�match�bytesrD �type�__name__rI r3 � JWT_TOKENrJ )r4 r8 Z
xmlrpm_server�resultZ
result_lenr r r �getAndWriteJWTTokenToFile� s*
rY )�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# r< r( �read�strip�_human_readable_to_product)rZ Zedition_cache_file�fZ raw_editionr r r �getServerEdition� s
r` 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 rL �print�errmsgrM �exitrK r` �format�_product_to_human_readabler9 �isatty�getr Zcount_server_users�_askConfirmation) �
activationKeyZ
strictEditionZsilentMigrationZlicenseInformation�eZcurrentEditionZlicenseEdition�messageZedition_to_users_limitZlicense_users_limitZusers_on_serverr r r �checkLicenseKey� sB
rp )�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.ra N)re r: �lowerrM rg )rq Zresponser r r rl s
rl 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_idr7 �username�passwordrb z /etc/cloudlinux-edition-solor z /etc/cloudlinux-edition-adminr r )�AssertionErrorr Z
getVersionZ getOSReleaseZgetArchr Z getSystemId�itemsr r# r< r �RegistrationRhnServer� registrationZ
new_system) rt ru Z profileNamer7 �otherrb Z auth_dictZ
system_id_xml�key�item�s�retr r r �registerSystem
s.
r c C s t j� }|jj| �}|S )zG
Check the activation key and return it's edition and customer
)r rx ry Z
license_check)rm r} r~ r r r rd 9 s rd 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>N s z%_encode_characters.<locals>.<genexpr>c S s g | ] }t |��qS r )r� )r r� r r r �
<listcomp>P s z&_encode_characters.<locals>.<listcomp>c S s g | ]\}}t ||��qS r )r� )r �name�valr r r r� R s ra r N) rU r
r r �tupler r r �dictrw r= rQ )�argsrX r| Z item_typer r r r� D s
r� )F)F)NNNNNN)Ar rR rM 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� rT rP r� �intZlong�gettextZ translation�t�hasattrr �_Z SYSID_DIRrW r^ rw ri r ZinitUp2dateConfigr ZinitLogrC r! r3 r6 �boolrI rJ rY r` rp rl r rd r� r r r r �<module> sl $
$
%>
*