3 m��i*�@s�ddlZddlZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd lmZm Z y4ddlZddlZdd lmZmZmZmZmZmZWnDek r�ddljZddljZeZeZeZeZe Ze Ze!Z"YnXddl#Z#e#j$ddd �Z%e&e%d��se%j#e%_'e%j'Z(dZ)dZ*ddddd�Z+dd�e+j,�D�Z-ddlm.Z.e.j/�Z0ej1�Z2dd�Z3dd�Z4dd�Z5ee6d�d d!�Z7e6d"�d#d$�Z8d6d&d'�Z9d7e6d(�d)d*�Z:d+d,�Z;ed-�d.d/�Zd4d5�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 AdmincCsi|]\}}||�qS�r)�.0�k�vrr�/usr/lib/python3.6/rhnreg.py� 4sr)�configcCstjtdtj�S)N�systemIdPath)�os�access�cfg�R_OKrrrr� registered<sr!cCs�tjj|�}tj|tj�sdStj|tj�rLytj||d�Wn dStj|tjtj Bt dd��}tj|d�}z|jt |��Wd|j�XdS)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)Zsecure_fileZ file_contentsZdir_name�fdZfd_filerrr�_write_secure_file?s r1cCsttd|�}|S)Nr)r1r)�systemId�resrrr� writeSystemIdUsr4)�token�allowTransitioncCsnddl}ddl}|jjd�s dSdd|g}|r8|jd�|j|�}|j�\}}|jdkrjtj d||f�dS)zH Execute binary file which we use as hook for jwt token updates rNz /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) � subprocessrr$�exists�append�Popen�communicate� returncode�log�log_me)r5r6r7r�cmd�p�stdout�stderrrrr�_execute_pre_jwt_update_hookZs rC)r6cCsjddl}ddl}|jjd�s dSdg}|r4|jd�|j|�}|j�\}}|jdkrftj d||f�dS)zH Execute binary file which we use as hook for jwt token updates rNz /opt/cloudlinux/venv/bin/python3z/usr/sbin/cl-post-jwt-updatez--allow-transitionz8Post jwt update hook failed with stdout=%s and stderr=%s) r7rr$r8r9r:r;r<r=r>)r6r7rr?r@rArBrrr�_execute_post_jwt_update_hookms rDFcCs�tj�}y|jj|�}WnFtjk r.dStjtjtjtj fk r^t jtj ��dSXt||�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)rZ RhnServerZup2dateZgetJWTTokenr�UnknownMethodExceptionZAuthenticationTicketErrorZRhnUuidUniquenessError�CommunicationErrorZ$AuthenticationOrAccountCreationErrorr=Z log_exception�sys�exc_inforCr1� JWT_TOKENrD)r2r6Z xmlrpm_server�resultrrr�getAndWriteJWTTokenToFile�s rK)�human_readablec CsFd}tjj|�sdSt|�� }|j�jd�}|r4|St|SQRXdS)Nz/opt/cloudlinux/cl_editionr� )rr$r8r)�read�strip�_human_readable_to_product)rLZedition_cache_file�fZraw_editionrrr�getServerEdition�s rRcCsVyt|�}WnPtjk rH}ztd|j�tjd�WYdd}~Xntjk r\dSXt�}|d}||krxdS|r�tdj t |t |d��td�tjd�|�sRtjj�s�td�td�t |�dt |�d �}d dd�}|j |�} | dk �rJtj�} | | k�r6td||�d | �d��tjd�n|�d||�d�}t|�dS)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_editionZnew_editionz0Run clnreg_ks manually to complete registration.aError: 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.�)rrz@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.)�checkKeyrrF�print�errmsgrG�exitrErR�format�_product_to_human_readable�stdin�isatty�getrZcount_server_users�_askConfirmation)� activationKeyZ strictEditionZsilentMigrationZlicenseInformation�eZcurrentEditionZlicenseEdition�messageZedition_to_users_limitZlicense_users_limitZusers_on_serverrrr�checkLicenseKey�sB rc)�confirmationMessagecCs2t|�td�}|j�dkr.td�tjd�dS)zS Prints message and makes sure that client is ready for edition migration. z Do you want to continue? [N/y]: �yzAborted.rSN)rW�input�lowerrGrY)rdZresponserrrr_�s r_cCs�|tj�tj�tj�d�}tj�}|dk r2||d<|rTx|j�D]\}} | ||<q@W|rb||d<n||d<||d<|dk r�||d<n(tjj d�r�d ntjj d �r�dnd|d<t j�} | jj |�}|S) ziWrapper for the old xmlrpc to register a system. Activates subscriptions if a reg num is given. )Zprofile_nameZ os_releaseZrelease_nameZarchitectureNZ system_idr5�username�passwordrTz/etc/cloudlinux-edition-solorz/etc/cloudlinux-edition-adminrr)rZ getVersionZgetOSReleaseZgetArchrZgetSystemId�itemsrr$r8r�RegistrationRhnServer�registrationZ new_system)rhriZprofileNamer5�otherrTZ auth_dictZ system_id_xml�key�item�s�retrrr�registerSystem�s* rrcCstj�}|jj|�}|S)zG Check the activation key and return it's edition and customer )rrkrlZ license_check)r`rprqrrrrVsrVcGs�g}x�|D]�}t|�}|tkr(t|�}nZ|tkrDtdd�|D��}n>|tkr\dd�|D�}n&|tksl|tkr�tdd�|j �D��}|j |�q Wt|�dkr�|dSt|�SdS) 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. css|]}t|�VqdS)N)�_encode_characters)r�irrr� 4sz%_encode_characters..cSsg|]}t|��qSr)rs)rrtrrr� 6sz&_encode_characters..cSsg|]\}}t||��qSr)rs)r�name�valrrrrv8srSrN)�typer r r�tuplerrr�dictrjr9�len)�argsrJroZ item_typerrrrs*s rs)F)F)NNNNNN)@rrGZup2date_clientrrrrrrZup2date_client.pkgplatformrZrhn.i18nr r ZurlparseZ xmlrpclib�typesrrr rrr�ImportErrorZurllib.parse�parseZ xmlrpc.clientZclient�listrz�bytes�strr{r,Zlong�gettextZtranslation�t�hasattrr�_Z SYSID_DIRrIrPrjr[rZinitUp2dateConfigrZinitLogr=r!r1r4�boolrCrDrKrRrcr_rrrVrsrrrr� sj$ > *