Edit File: rhnreg.cpython-36.pyc
3 ��cg � @ s> d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d d lm Z d d lmZ d d lmZ d d lmZmZ d d lmZ y4d dlZd dlZd dlmZmZm Z m Z m Z m Z W nF e k �r6 d dl!j"Zd dl#j$Ze%Ze&Ze'Z e(Z e)Z e)Z e*Z+Y nX yd dl,m-Z- W n e k �rb dZ-Y nX d dl.Z.e.j/ddd�Z0e1e0d��s�e0j.e0_2e0j2Z3dZ4de4 Z5de4 Z6dZ7dZ8dZ9dZ:dZ;d d lm<Z< e<j=� Z>e j?� Z@d d � ZAd d � ZBd!d"� ZCd#d$� ZDd%d&� ZEd'd(� ZFd)d*� ZGd+d,� ZHd-d.� ZId/d0� ZJd1d2� ZKd3d4� ZLd5d6� ZMd7d8� ZNd9d:� ZOd;d<� ZPd=d>� ZQd?d@� ZRdAdB� ZSG dCdD� dD�ZTdidEdF�ZUdGdH� ZVdIdJ� ZWdddddi fdKdL�ZXdMdN� ZYdOdP� ZZdQdR� Z[dSdT� Z\dUdV� Z]dWdX� Z^djdYdZ�Z_G d[d\� d\�Z`d]d^� Zad_d`� Zbdkdadb�Zcdcdd� Zde� dek�r.dfdg� Zen d dhlfmeZe dS )l� N)� up2dateUtils)� up2dateErrors)� up2dateAuth)� rhnserver)�pkgUtils)� up2dateLog)�rhnreg_constants)�hardware)� convertPackagesFromHashToList)� getPlatform)�ustr�sstr)� raise_with_tb)�ListType� TupleType� StringType� UnicodeType�DictType�DictionaryType)�supportzrhn-client-toolsT)Zfallback�ugettextz/etc/sysconfig/rhnz%s/rhn_register_remindz%s/hw-activation-codez/etc/pki/consumer/cert.pemz$/etc/sysconfig/rhn/cl-rollout-ca.pemz!/etc/sysconfig/rhn/cl-rollout.pemz%/etc/sysconfig/rhn/cl-rollout-key.pemz /etc/sysconfig/rhn/jwt.token)�configc C s� t jdt jt jB �r�t jdt j�rZt jdt jt jB �rLt jd� t jd� q�ttd�� n|t jdt jt jB �rzt jd� n ttd �� d } t j| t jt jB �s�t jd t jt jB �r�d } t jd | �}|r�t jd | � d S )Nz/usr/sbin/rhnsdz%/usr/lib/systemd/system/rhnsd.servicez/usr/bin/systemctlz+/usr/bin/systemctl enable rhnsd > /dev/nullz*/usr/bin/systemctl start rhnsd > /dev/nullz,Warning: unable to enable rhnsd with systemdz/sbin/chkconfigz$/sbin/chkconfig rhnsd on > /dev/nullz.Warning: unable to enable rhnsd with chkconfigz /sbin/servicez/usr/sbin/servicez%s rhnsd status > /dev/nullz%s rhnsd start > /dev/null)�os�access�R_OK�X_OK�system�print�_)Z service_pathZrc� r � /usr/lib/python3.6/rhnreg.py� startRhnsdK s r! c C s� t d p d} tj| tj�s i S t| d�}|j� }i }x`|D ]X}|j� }|dkrPq:y|jd�\}}W n" tk r� t t j |�� Y nX |j� ||< q:W |S )NZ oemInfoFilez/etc/sysconfig/rhn/oeminfo�r� �:) �cfgr r r �open� readlines�strip�split� ValueErrorr r ZOemInfoFileError)Z configFile�fd�L�info�i�key�valuer r r � getOemInfoe s r1 c C s* t jtt j�r"t jt�} | jdkS dS dS )z@ Returns true if system is registred using subscription manager r FN)r r � RHSM_FILEr �stat�st_size)Zstatinfor r r �rhsm_registered| s r5 c C s t jtd t j�S )N� systemIdPath)r r r% r r r r r � registered� s r7 c C s$ t jtt j�s ttd�} | j� d S )Nzw+)r r � REMIND_FILEr r& �close)r+ r r r � createSystemRegisterRemindFile� s r: c C s t jtt j�rt jt� d S )N)r r r8 r �unlinkr r r r � removeSystemRegisterRemindFile� 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�renamer& �O_WRONLY�O_CREAT�int�fdopen�writer r9 )Z secure_fileZ file_contentsZdir_namer+ Zfd_filer r r �_write_secure_file� s rI c C s" t td | �}|rt� t� |S )Nr6 )rI r% r< �updateRhsmStatus)�systemId�resr r r � writeSystemId� s rM c C sx d} t j� }|d krd S y6tj|�}|j� }|j| �d j}t|dd � �}W n$ ttj fk rr t j d� d S X |S )Nz'//member[name='system_id']/value/stringr � z@systemID file doesn't have system_id field or the file is broken) r � getSystemId�libxml2ZparseDocZxpathNewContextZ xpathEvalZcontentrF � IndexErrorZ parserError�log�log_me)Z xpath_strrK �result�context� system_idr r r �extract_system_id� s rW c C sn t j� }y|jj| �}W nF tjk r. dS tjtjtjtj fk r^ t j t j � � dS X tt|� 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 � RhnServer�up2dateZ getJWTTokenr �UnknownMethodException�AuthenticationTicketError�RhnUuidUniquenessError�CommunicationError�$AuthenticationOrAccountCreationErrorrR � log_exception�sys�exc_inforI � JWT_TOKEN)rK Z xmlrpm_serverrT r r r �getAndWriteJWTTokenToFile� s rc c ; C s� d }| dkr�yt jt� W n ttfk r2 Y nX yt jt� W n ttfk rZ Y nX yt jt� W n ttfk r� Y nX dS tj� }y|j j | �}W nF t j k r� dS t j t jt jt jfk r� tjtj� � dS X x,|D ]$}||j� kr�tjdj|�� dS q�W ytj|d �}W n tk �rH tjd� dS X ytj|d �}W n tk �r| tjd� dS X ytj|d �}W n tk �r� tjd� dS X tt|� tt|� tt|� dS ) z� Get ca.crt, rollout.crt and rollout.key from CLN and write them to files :param systemId: registration ID :return: None � cert_base64� ca_base64� key_base64NzCLN did not return "{}".z/CLN returned incorrectly encoded CA certificatez,CLN returned incorrectly encoded rollout keyz4CLN returned incorrectly encoded rollout certificate)rd re rf ) r �remove� ROLLOUT_KEY�OSError�IOError� ROLLOUT_CERT�ROLLOUT_CA_CERTr rX rY ZgetCLNCertificater rZ r[ r\ r] r^ rR r_ r` ra �keysrS �format�base64Z b64decode� TypeErrorrI )rK rm �srT r/ Zca_certZ rollout_keyZ rollout_certr r r �"prepareGradualRolloutCertsIfNeeded� sh rr c C s t t| d �S )z=Returns True if the write is successful or False if it fails.� )rI � HW_CODE_FILE)�hw_activation_coder r r � writeHWCode% s rv c C sp y"t jjd�r t� \} }| |fS W n tk r6 Y nX t� \} }| dk rR| |fS t� \} }| dk rl| |fS dS )a� This function returns the UUID and virtualization type of this system, if it is a guest. Otherwise, it returns None. To figure this out, we'll use a number of heuristics (list in order of precedence): 1. Check /proc/xen/xsd_port. If exists, we know the system is a host; exit. 2. Check SMBIOS. If vendor='Xen' and UUID is non-zero, we know the system is a fully-virt guest; exit. 3. Check /sys/hypervisor/uuid. If exists and is non-zero, we know the system is a para-virt guest; exit. 4. If non of the above checks worked; we know we have a non-xen-enabled system; exit. z/proc/xen/xsd_portN)NN)r r? �exists�get_fully_virt_inforj �get_para_virt_info)�uuid� virt_typer r r � get_virt_info) s r| c C sV y<t dd�} | j� }| j� |j� jdd�jd�}d}||fS tk rP Y nX dS ) z� This function checks /sys/hypervisor/uuid to see if the system is a para-virt guest. It returns a (uuid, virt_type) tuple. z/sys/hypervisor/uuidr" �-r# z ZparaN)NN)r& �readr9 �lower�replace�rstriprj )Z uuid_filerz r{ r r r ry U s ry c C s@ t j� } t j� }| j� dkr8|j� jdd�}d}||fS dS dS )z� This function looks in the SMBIOS area to determine if this is a fully-virt guest. It returns a (uuid, virt_type) tuple. Zxenr} r# ZfullyN)NN)r Z dmi_vendorZdmi_system_uuidr r� )Zvendorrz r{ r r r rx g s rx c C s t d| �} t| �td�kS )Nz0x%sr )�eval�long)rz r r r � _is_host_uuidu s r� c C s t j� } | jj� S )N)r rX � registrationZwelcome_message)rq r r r � welcomeTexty s r� c C s t j� } | jj� d S )N)r rX � capabilitiesZvalidate)rq r r r �getCaps s r� c C s t j� }|jj| |�S )N)r rX r� Z reserve_user)�username�passwordrq r r r � reserveUser� s r� c @ sf e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dS )�RegistrationResultNc C sB || _ || _|| _|| _|| _t|�dkr2|| _nd | _|| _d S )Nr )� _systemId� _channels�_failedChannels� _systemSlots�_failedSystemSlots�len�_universalActivationKey�rawDict)�selfrK �channelsZfailedChannels� systemSlotsZfailedSystemSlotsZuniversalActivationKeyr� r r r �__init__� s zRegistrationResult.__init__c C s | j S )N)r� )r� r r r rO � s z RegistrationResult.getSystemIdc C s | j S )N)r� )r� r r r � getChannels� s z RegistrationResult.getChannelsc C s | j S )N)r� )r� r r r �getFailedChannels� s z$RegistrationResult.getFailedChannelsc C s | j S )N)r� )r� r r r �getSystemSlots� s z!RegistrationResult.getSystemSlotsc s � fdd�� j D �S )Nc s g | ]}� j |��qS r )�_getSlotDescription)�.0rq )r� r r � <listcomp>� s z@RegistrationResult.getSystemSlotDescriptions.<locals>.<listcomp>)r� )r� r )r� r �getSystemSlotDescriptions� s z,RegistrationResult.getSystemSlotDescriptionsc s � fdd�� j D �S )Nc s g | ]}� j |��qS r )�_getFailedSlotDescription)r� rq )r� r r r� � s zFRegistrationResult.getFailedSystemSlotDescriptions.<locals>.<listcomp>)r� )r� r )r� r � getFailedSystemSlotDescriptions� s z2RegistrationResult.getFailedSystemSlotDescriptionsc C s | j S )z5Returns None if no universal activation key was used.)r� )r� r r r �getUniversalActivationKey� s z,RegistrationResult.getUniversalActivationKeyc C s t | j�dkot | j�dkS )z�Returns True if the system was subscribed to at least one channel and was given any type of system slot so it will get updates. In other words, returns True if the system will be getting at least basic updates. r )r� r� r� )r� r r r �hasBaseAndUpdates� s z$RegistrationResult.hasBaseAndUpdatesc C s& |dkrt jd t j S | j|�S d S )N�virtualization_host� )r �VIRTZ VIRT_FAILEDr� )r� �slotr r r r� � s z,RegistrationResult._getFailedSlotDescriptionc C s$ |dkrt jS |dkr t jS |S d S )NZenterprise_entitledr� )r Z MANAGEMENTr� )r� r� r r r r� � s z&RegistrationResult._getSlotDescription)N)�__name__� __module__� __qualname__r� rO r� r� r� r� r� r� r� r� r� r r r r r� � s r� c C s� |t j� t j� t j� d�}|r<x |j� D ]\}}|||< q(W |rJ||d< n| |d< ||d< |dk rl||d< n(tjjd�r|dntjjd �r�d nd |d< td r�t t j � �|d < t j � } | jj|�} | S )ziWrapper for the old xmlrpc to register a system. Activates subscriptions if a reg num is given. )Z profile_nameZ os_releaseZ release_nameZ architecture�tokenr� r� N�editionz /etc/cloudlinux-edition-soloZsoloz /etc/cloudlinux-edition-adminZadminZshared�supportsSMBIOS�smbios)r � getVersion� getOSRelease�getArch�itemsr r? rw r% �_encode_charactersr � get_smbiosr rX r� Z new_system) r� r� � profileNamer� �otherr� Z auth_dictr/ �itemrq �retr r r �registerSystem� s( r� c C sl y,t j� } | j| dddd�}t j|dd�}W n t jk rB d S X y |j� W n t jk rf Y nX d S )Nz com.redhat.SubscriptionManagerz/EntitlementStatusF)Z introspectz0com.redhat.SubscriptionManager.EntitlementStatus)Zdbus_interface)�dbusZ SystemBusZProxyObjectClassZ InterfaceZ DBusExceptionZ check_status)ZbusZ validity_objZvalidity_ifacer r r rJ � s rJ c C s� t j� }tj� }tj� }tj� }d }y|jj| ||||�}W n@ tj k r| t j � d }|j dkrvt tj|j�� n� Y nX |S )N� �c )r rX r r� r� Z getReleaser� Zavailable_eus_channels� xmlrpclibZFaultr` ra Z faultCoder r Z DelayErrorZ faultString)r� r� rq Z server_archZserver_versionZserver_releaseZavailableChannels�fr r r �getAvailableChannels s r� c C s$ |dkr i }|r6| dks t �|dks(t �|dk sZt �n$| dk sBt �|dk sNt �|dksZt �x |j� D ]}|dksdt �qdW td r�ttj� �|d< tj� }|r�|jj |t j � t j � t j � ||�}n$|jj|t j � t j � t j � | ||�}tjd | � t|d |d |d |d |d |d |d�} | S )a� Uses the new xmlrpcs to register a system. Returns a dict instead of just system id. The main differences between this and registerSystem and that this doesn't do activation and does child channel subscriptions if possible. See the documentation for the xmlrpc handlers in backend for more detail. If nothing is going to be in other, it can be {} or None. New in RHEL 5. N�registration_number�org_id� virt_uuidr{ �channelr� r� z Returned: %srV r� Zfailed_channelsZ system_slotsZfailed_system_slotsZuniversal_activation_key)r� )r� r� r� r{ r� )�AssertionErrorrm r% r� r r� r rX r� Znew_system_activation_keyr r� r� r� Znew_system_user_passrR � log_debugr� ) r� r� r� ZpackagesZ activationKeyr� r/ rq r- rT r r r �registerSystem2 sN r� c C s t d S )NZ supportsEUS)r% r r r r �server_supports_eusW s r� c sF dd� � t j� }|jjdd�s0� fdd�|D �}|jj| t|�� d S )Nc S s | d dkrd| kr| d= | S )N�classZNETINFOZip6addrr )�xr r r �remove_ip6addr[ s z$sendHardware.<locals>.remove_ip6addrZipv6r� c s g | ] }� |��qS r r )r� r. )r� r r r� a s z sendHardware.<locals>.<listcomp>)r rX r� � hasCapabilityr� Zadd_hw_profiler� )rK Z hardwareListrq r )r� r � sendHardwareZ s r� c C s0 t j� }|jjdd�s t|�}|jj| |� d S )Nz xmlrpc.packages.extended_profile� )r rX r� r� r r� Z add_packages)rK Z packageListrq r r r � sendPackagesd s r� c C s t d k rt j� d S )N)r Zrefresh)rK r r r � sendVirtInfok s r� c C s t j� }t|jj| � � d S )N)r rX r r� Z list_packages)rK rq r r r � listPackageso s r� c C s� t j | �\}}}}}}|dks&|dkrDd| } t j | �\}}}}}}|d krVtjd��|dksn|dksn|dkrrd}t j||||||f�} | S ) zzRaises up2dateErrors.InvalidProtocolError if the server url has a protocol specified and it's not http or https. Nr# zhttps://�https�httpzCYou specified an invalid protocol. Only https and http are allowed.�/z/XMLRPC)r� r� )�urlparser ZInvalidProtocolErrorZ urlunparse)�serverZprotocol�hostr? Z parametersZqueryZfragmentIdentifierr r r �makeNiceServerUrls s r� c C s dS )zdReturns 'hosted' if the url points to a known hosted server. Otherwise returns 'satellite'. Z satelliter )Z serverUrlr r r � getServerType� s r� c @ sB e Zd ZdZdZi i fdd�Zdd� Zdd� Zd d � Zd d � Z d S )�ActivationResultr r� c C s || _ || _|| _|| _dS )zschannels and systemSlots are dicts where the key/value pairs are label (string) / quantity (int). N)�_status�_regNumr� r� )r� ZstatusZregistrationNumberr� r� r r r r� � s zActivationResult.__init__c C s | j S )N)r� )r� r r r � getStatus� s zActivationResult.getStatusc C s | j S )N)r� )r� r r r �getRegistrationNumber� s z&ActivationResult.getRegistrationNumberc C s | j S )z7Returns a dict- the key/value pairs are label/quantity.)r� )r� r r r �getChannelsActivated� s z%ActivationResult.getChannelsActivatedc C s | j S )z7Returns a dict- the key/value pairs are label/quantity.)r� )r� r r r �getSystemSlotsActivated� s z(ActivationResult.getSystemSlotsActivatedN) r� r� r� � ACTIVATED_NOW� ALREADY_USEDr� r� r� r� r� r r r r r� � s r� 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)r� )r� r. r r r � <genexpr>� s z%_encode_characters.<locals>.<genexpr>c S s g | ] }t |��qS r )r� )r� r. r r r r� � s z&_encode_characters.<locals>.<listcomp>c S s g | ]\}}t ||��qS r )r� )r� �name�valr r r r� � s r� r N) �typer r r �tupler r r �dictr� �appendr� )�argsrT r� Z item_typer r r r� � s r� c C s� d }d }d }yt j� }t|�}W n$ tjd� tjtj� � Y nX |d k r�y.t| ||�}|j � t j krz|j � }t |� W n> tjk r� tjd� Y n tjk r� tjd� Y nX |S )NzMThere was an error while reading the hardware info from the bios. Traceback: z<There are are no entitlements associated with this hardware.z,The hardware id was not recognized as valid.)r Zget_hal_system_and_smbiosr� rR rS r_ r` ra �activateHardwareInfor� r� r� r� rv r ZNotEntitlingErrorr� Z InvalidRegistrationNumberError)Zloginr� ZactivateHWResult� hardwareInforu r r r �_activate_hardware� s* r� c C s� i }|rd|i}t j� }|jj| |||�}|d }|d }tjd| � |dkr\ttj|�S |dkrpttj|�S d| } t j | ��dS ) z�Tries to activate an entitlement linked to the hardware info that we read from the bios. Returns an ActivationResult. Can raise: Invalid number. Hardware info is not entitling. Communication errors, etc r� Z status_coder� z Server returned status code %sr r� zNThe server returned unknown status code %s while activating the hardware info.N) r rX r� Zactivate_hardware_inforR r� r� r� r� r r] ) r� r� r� ZorgIdr� r� rT Z statusCodeZregNum�messager r r r� � s r� c C sd t jdt jt jB �rVddlm} m} | dg|||d�}tdd� |jj � |j j � � n t j d� d S )Nz/usr/sbin/rhn_checkr )�Popen�PIPE)�stdin�stdout�stderrc S s t j| �S )N)rR rS )r� r r r �<lambda> s z$spawnRhnCheckForUI.<locals>.<lambda>z Warning: unable to run rhn_check) r r r r � subprocessr� r� �mapr� r' r� rR rS )r� r� �pr r r �spawnRhnCheckForUI s r� Zdebc C s dS )z.On Debian no extra action for plugin is neededr� r )r� r r r r r r � pluginEnable s r� )r� )NNNNNN)N)N)gr r` r� ro rP Zup2date_clientr r r r r r r r Z up2date_client.rhnPackageInfor Zup2date_client.pkgplatformr Zrhn.i18nr r Zrhn.tbr r� r� �typesr r r r r r � ImportErrorZ urllib.parse�parseZ xmlrpc.clientZclient�listr� �bytes�strr� rF r� Zvirtualizationr �gettextZ translation�t�hasattrr r Z SYSID_DIRr8 rt r2 rl rk rh rb r ZinitUp2dateConfigr% ZinitLogrR r! r1 r5 r7 r: r< rI rM rW rc rr rv r| ry rx r� r� r� r� r� r� rJ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zup2date_client.pmPluginr r r r �<module> s� $ E,? "; "