ruạṛ
3 U��Z(X � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d l m Z d d l mZ d d lmZmZ d d lmZ y4d dlZd dlZd d lmZmZmZmZmZm Z W nF e k �r d dl j Zd dl j!Ze"Ze#Ze$Ze%Ze&Ze&Z e'Z(Y nX yd dl)m*Z* W n e k �rF dZ*Y nX d dl+Z+e+j,ddd�Z-e.e-d��sre-j+e-_/e-j/Z0dZ1de1 Z2de1 Z3dZ4d dlm5Z5 e5j6� Z7ej8� Z9dd� Z:dd� Z;d d � Z<d d � Z=d d!� Z>d"d#� Z?d$d%� Z@d&d'� ZAd(d)� ZBd*d+� ZCd,d-� ZDd.d/� ZEd0d1� ZFd2d3� ZGd4d5� ZHd6d7� ZIG d8d9� d9�ZJd^d:d;�ZKd<d=� ZLd>d?� ZMdddddi fd@dA�ZNdBdC� ZOdDdE� ZPdFdG� ZQdHdI� ZRdJdK� ZSdLdM� ZTd_dNdO�ZUG dPdQ� dQ�ZVdRdS� ZWdTdU� ZXd`dVdW�ZYdXdY� ZZe� dZk�r�d[d\� Z[n d d]l\m[Z[ dS )a� N)� up2dateUtils)� up2dateErrors)� 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.pem)�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� startRhnsdB 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 � getOemInfo\ s r0 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_registereds s r4 c C s t jtd t j�S )N� systemIdPath)r r r$ r r r r r � registered{ s r6 c C s$ t jtt j�s ttd�} | j� d S )Nzw+)r r � REMIND_FILEr r% �close)r* r r r � createSystemRegisterRemindFile~ s r9 c C s t jtt j�rt jt� d S )N)r r r7 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 r8 )Z secure_fileZ file_contentsZdir_namer* Zfd_filer r r �_write_secure_file� s rH c C s" t td | �}|rt� t� |S )Nr5 )rH r$ r; �updateRhsmStatus)�systemId�resr r r � writeSystemId� s rL c C s t t| d �S )z=Returns True if the write is successful or False if it fails.� )rH � HW_CODE_FILE)�hw_activation_coder r r � writeHWCode� s rP 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_info�IOError�get_para_virt_info)�uuid� virt_typer r r � get_virt_info� s rW 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% �readr8 �lower�replace�rstriprS )Z uuid_filerU rV r r r rT � s rT 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. ZxenrX r" ZfullyN)NN)r Z dmi_vendorZdmi_system_uuidrZ r[ )ZvendorrU rV r r r rR � s rR c C s t d| �} t| �td�kS )Nz0x%sr )�eval�long)rU r r r � _is_host_uuid� s r_ c C s t j� } | jj� S )N)r � RhnServer� registrationZwelcome_message)�sr r r � welcomeText� s rc c C s t j� } | jj� d S )N)r r` � capabilitiesZvalidate)rb r r r �getCaps s re c C s t j� }|jj| |�S )N)r r` ra Z reserve_user)�username�passwordrb r r r � reserveUser s rh 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)�selfrJ �channelsZfailedChannels� systemSlotsZfailedSystemSlotsZuniversalActivationKeyrq r r r �__init__ s zRegistrationResult.__init__c C s | j S )N)rj )rr r r r � getSystemId s z RegistrationResult.getSystemIdc C s | j S )N)rk )rr r r r � getChannels! s z RegistrationResult.getChannelsc C s | j S )N)rl )rr r r r �getFailedChannels$ s z$RegistrationResult.getFailedChannelsc C s | j S )N)rm )rr r r r �getSystemSlots' s z!RegistrationResult.getSystemSlotsc s � fdd�� j D �S )Nc s g | ]}� j |��qS r )�_getSlotDescription)�.0rb )rr r r � <listcomp>+ s z@RegistrationResult.getSystemSlotDescriptions.<locals>.<listcomp>)rm )rr r )rr r �getSystemSlotDescriptions* s z,RegistrationResult.getSystemSlotDescriptionsc s � fdd�� j D �S )Nc s g | ]}� j |��qS r )�_getFailedSlotDescription)r{ rb )rr r r r| . s zFRegistrationResult.getFailedSystemSlotDescriptions.<locals>.<listcomp>)rn )rr r )rr r � getFailedSystemSlotDescriptions- s z2RegistrationResult.getFailedSystemSlotDescriptionsc C s | j S )z5Returns None if no universal activation key was used.)rp )rr r r r �getUniversalActivationKey0 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 )ro rk rm )rr r r r �hasBaseAndUpdates4 s z$RegistrationResult.hasBaseAndUpdatesc C s& |dkrt jd t j S | j|�S d S )N�virtualization_host� )r �VIRTZ VIRT_FAILEDrz )rr �slotr r r r~ ? s z,RegistrationResult._getFailedSlotDescriptionc C s$ |dkrt jS |dkr t jS |S d S )NZenterprise_entitledr� )r Z MANAGEMENTr� )rr r� r r r rz E s z&RegistrationResult._getSlotDescription)N)�__name__� __module__� __qualname__ru rv rw rx ry r} r r� r� r~ rz r r r r ri s ri c C s� |t j� t j� t j� d�}|r<x |j� D ]\}}|||< q(W |rJ||d< n| |d< ||d< td rrttj� �|d< t j � }|j j |�} | 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�tokenrf rg �supportsSMBIOS�smbios) r � getVersion� getOSRelease�getArch�itemsr$ �_encode_charactersr � get_smbiosr r` ra Z new_system) rf rg � profileNamer� �otherZ auth_dictr. �itemrb �retr r r �registerSystemN 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 rI k s rI 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 r` r r� r� Z getReleasera Zavailable_eus_channels� xmlrpclibZFault�sys�exc_infoZ faultCoder r Z DelayErrorZ faultString)rf rg rb 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_uuidrV �channelr� r� z Returned: %sZ system_idrs Zfailed_channelsZ system_slotsZfailed_system_slotsZuniversal_activation_key)rq )r� r� r� rV r� )�AssertionError�keysr$ r� r r� r r` ra Znew_system_activation_keyr r� r� r� Znew_system_user_pass�log� log_debugri ) rf rg r� ZpackagesZ activationKeyr� r. rb r, �resultr r r �registerSystem2� sN r� c C s t d S )NZ supportsEUS)r$ r r r r �server_supports_eus� 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| � s z sendHardware.<locals>.<listcomp>)r r` rd � hasCapabilityra Zadd_hw_profiler� )rJ Z hardwareListrb r )r� r � sendHardware� s r� c C s0 t j� }|jjdd�s t|�}|jj| |� d S )Nz xmlrpc.packages.extended_profile� )r r` rd r� r ra Z add_packages)rJ Z packageListrb r r r � sendPackages� s r� c C s t d k rt j� d S )N)r Zrefresh)rJ r r r � sendVirtInfo� s r� c C s t j� }t|jj| � � d S )N)r r` r ra Z list_packages)rJ rb r r r � listPackages� 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 �makeNiceServerUrl� 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�_regNumrk rm )rr ZstatusZregistrationNumberrs rt r r r ru s zActivationResult.__init__c C s | j S )N)r� )rr r r r � getStatus s zActivationResult.getStatusc C s | j S )N)r� )rr r r r �getRegistrationNumber! s z&ActivationResult.getRegistrationNumberc C s | j S )z7Returns a dict- the key/value pairs are label/quantity.)rk )rr r r r �getChannelsActivated$ s z%ActivationResult.getChannelsActivatedc C s | j S )z7Returns a dict- the key/value pairs are label/quantity.)rm )rr r r r �getSystemSlotsActivated( s z(ActivationResult.getSystemSlotsActivatedN) r� r� r� � ACTIVATED_NOW� ALREADY_USEDru 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>6 s z%_encode_characters.<locals>.<genexpr>c S s g | ] }t |��qS r )r� )r{ r- r r r r| 8 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� �appendro )�argsr� 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� r� �log_meZ log_exceptionr� r� �activateHardwareInfor� r� r� r� rP r ZNotEntitlingErrorr� Z InvalidRegistrationNumberError)Zloginrg ZactivateHWResult� hardwareInforO r r r �_activate_hardwareB 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 r` ra Zactivate_hardware_infor� r� r� r� r� r ZCommunicationError) rf rg r� ZorgIdr� r� r� 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)r� r� )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� r� r� )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� )NNNNN)N)N)]r r� r� Zup2date_clientr 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� rE r^ Zvirtualizationr �gettextZ translation�t�hasattrr r Z SYSID_DIRr7 rN r1 r ZinitUp2dateConfigr$ ZinitLogr� r r0 r4 r6 r9 r; rH rL rP rW rT rR r_ rc re rh ri r� rI r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zup2date_client.pmPluginr r r r �<module> s� $ ,? ; "
cải xoăn