[EDIT] FILE: rhnreg.cpython-36.pyc
3 �Xj�. � @ 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 $ $ %> *
SAVE
CANCEL