File: //lib/python3.6/site-packages/up2date_client/__pycache__/rhnreg.cpython-36.pyc
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� $
,?
;
"