3
%T�f�6 � @ sf d dl mZ d dl mZ d dlmZmZ d dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m Z d dl
mZ d dlZd dlZd dlZd dlZd dlmZ d dlmZ d Zed
�Zed �Zed �Zed
�Zed�Z ed�Z ed�Z ed�Z ed�Z ed�Z!ed�Z"ed�Z#ed�Z$G dd� dej%�Z&G dd� dej'j(�Z)dd � Z*G d d � d e+�Z,d d � Z-d!d"� Z.dS )#� )�absolute_import)�unicode_literals)�_�loggerN)�copy)�PRIO_PLUGINCONFIG)�ustr)�
up2dateErrorsz_spacewalk.jsonz7CloudLinux Network based repositories will be disabled.z4CloudLinux Network channel support will be disabled.z@There was an error communicating with CloudLinux Network server.z=This system is not registered with CloudLinux Network server.z.This system is not subscribed to any channels.z SystemId could not be acquired.z%You can use rhn_register to register.z@This system is receiving updates from CloudLinux Network server.z�For security reasons packages from CloudLinux Network based repositories can be verified only with locally installed gpg keys. GPG key '%s' has been rejected.z.Package profile information could not be sent.z=Missing required login information for CloudLinux Network: %sz'Leapp upgrade is running - using cache.z>Spacewalk plugin has to be run under with the root privileges.c sZ e Zd ZdZ� fdd�Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z dd� Z
dd� Z � Z S )� SpacewalkZ spacewalkc s� t t| �j||� || _|| _tjj| jjj t
�| _ d| _ i | _
t| jj�| _| j| j�| _d| jj� kr�| jjd�}x |D ]\}}| jj||t� q|W tjj� s�tjt� d| j_| jjs�d S tj d� | j � d S )NF�mainz initialized Spacewalk plugin) �superr
�__init__�base�cli�os�path�join�confZ
persistdir�STORED_CHANNELS_NAME�stored_channels_path�connected_to_spacewalk� up2date_cfgr Z read_config�parser�sections�items�
_set_valuer �dnf�utilZ am_i_rootr �warning� MUST_BE_ROOT�enabled�debug�activate_channels)�selfr r �options�key�value)� __class__� � /usr/lib/python3.6/spacewalk.pyr
; s(
zSpacewalk.__init__c C s | j js d S d| jj_d S )NT)r r r ZdemandsZ root_user)r# r( r( r) �configR s zSpacewalk.configc C s t jd� d S )Nz$/usr/sbin/clnreg_ks --strict-edition)r �system)r# r( r( r) �clnregW s zSpacewalk.clnregTc C s� i }d }d}d }d }| j � }|s(|}�n�tjjd�rFtjt� |}�n�tjj � | _
t | j
�}| j
d f}d}xl|s�ytj j
| jjd�}d}W qn tjk
r� } z*|dkr�| j� d}wntjdtt| � d S d } ~ X qnX qnW |� s�tjdtt� | ji � d S ytjj| jjd�}
W n� tjk
�rF } ztjdtt| � d S d } ~ X nX tjk
�rt tjdt t � | ji � d S tj k
�r� tjd t t t t� d S X d| _!tj"t#� x,|
D ]$} | d
�r�t$| j%� �|| d < �q�W | j|� | j&j'} x�|j%� D ]�\}
}|j(|
�}d }|�r |j(d
�}t)| j�}|
| j*j+� k�rf| j*j%|
�}x |D ]\}}|j,||t-� �qJW t.|| j&j||j|||||j/|j0d � �}| j1|� �q�W tj2|� d S )
Nr z/etc/cln_leapp_in_progressZuseNoSSLForPackagesF)�timeoutTz%s
%s
%sz%s
%sz %s %s
%s
%s�version�label) r �proxyr- � sslcacert�
force_http�cached_version�
login_info�gpgcheckr )3�_read_channels_filer r �isfiler r �LEAPP_IN_PROGRESS�up2date_clientr* ZinitUp2dateConfigr �get_ssl_ca_certZ up2dateAuthZ getLoginInfor r- r �RhnServerExceptionr, �error�COMMUNICATION_ERROR� RHN_DISABLED�NOT_REGISTERED_ERROR�_write_channels_fileZ
rhnChannelZgetChannelDetailsZCommunicationErrorZNoChannelsError�NOT_SUBSCRIBED_ERROR�CHANNELS_DISABLEDZNoSystemIdError�NO_SYSTEM_ID_ERROR�USE_RHNREGISTERr �info�UPDATES_FROM_SPACEWALK�dictr r �repos�getr r r r r �
SpacewalkRepor5 r �addr! )r# Z
networkingZenabled_channelsr1 r2 Z proxy_urlr4 Zcached_channelsZ clnreg_tried�eZ svrChannels�channelrH Z
channel_idZ channel_dictZcached_channelr3 r r$ r% r& �repor( r( r) r" Z s�
zSpacewalk.activate_channelsc C s� | j js dS | jsdS | jd dkr8| j� }tjj|� ytjj| j j d� W n4 t
j k
r� } zt j
dtt|� W Y dd}~X nX dS )z, Update system's profile after transaction. NZwriteChangesToLog� )r- z%s
%s
%s)r r r r �_make_package_deltar9 ZrhnPackageInfoZlogDeltaPackagesZupdatePackageProfiler- r r; r r< r= �PROFILE_NOT_SENT)r# �deltarL r( r( r) � transaction� s zSpacewalk.transactionc C s� y.t | jd��}|j� }tj|�}|S Q R X W nX ttfk
rb } z|jtjkrR� W Y d d }~X n& tj j
k
r� } zW Y d d }~X nX i S )N�r) �openr �read�json�loads�FileNotFoundError�IOError�errno�ENOENT�decoderZJSONDecodeError)r# �
channels_fileZcontentZchannelsrL r( r( r) r6 � s
z Spacewalk._read_channels_filec C sf y,t | jd��}tj||dd� W d Q R X W n4 ttfk
r` } z|jtjkrP� W Y d d }~X nX d S )N�w� )�indent)rU r rW �dumprY rZ r[ r\ )r# �varr^ rL r( r( r) r@ � s z Spacewalk._write_channels_filec C s. dd� | j jjD �dd� | j jjD �d�}|S )Nc S s$ g | ] }|j |j|j|j|jf�qS r( )�namer. �release�epoch�arch)�.0�pr( r( r) �
<listcomp>� s z1Spacewalk._make_package_delta.<locals>.<listcomp>c S s$ g | ] }|j |j|j|j|jf�qS r( )rd r. re rf rg )rh ri r( r( r) rj � s )ZaddedZremoved)r rS Z install_setZ
remove_set)r# rR r( r( r) rP � s
z Spacewalk._make_package_delta)T)
�__name__�
__module__� __qualname__rd r
r* r, r" rS r6 r@ rP �
__classcell__r( r( )r' r) r
7 s
Yr
c sD e Zd ZdZdddddgZ� fdd�Zd d
� Zd� fd d
� Z� ZS )rJ zB
Repository object for Spacewalk. Uses up2date libraries.
zX-RHN-Server-IdzX-RHN-Auth-User-Idz
X-RHN-AuthzX-RHN-Auth-Server-TimezX-RHN-Auth-Expire-Offsetc sT t t� �jt|d �|jd�� t|d �� _� fdd�|d D �� _|jd�� _|jd�� _yt |d �� _
W n< t k
r� } z t j
ttjj|�� g � _
W Y d d }~X nX |d
|jd �kr�d � _|jd
�� _d� _d� _d � _d� _|jd�� _|jd�� _|jd�� _|jd��r$� j� n� j � t � d��rP� j � }|�rP� j |� d S )Nr/ r rd c s g | ]}|d � j �qS )z /GET-REQ/)�id)rh �url)r# r( r) rj � s z*SpacewalkRepo.__init__.<locals>.<listcomp>rp r1 r0 Z gpg_key_urlr. r3 rO r4 r r- r5 r2 r �set_http_headers) r rJ r
r rI rd Zbaseurlr1 r0 �get_gpg_key_urlsZgpgkey�InvalidGpgKeyLocationr r �GPG_KEY_REJECTEDr Zi18nZucdZmetadata_expirer4 Z keepaliveZ bandwidthZretriesZthrottler- r5 r2 �enable�disable�hasattr�create_http_headersrq )r# rM ZoptsrL � http_headers)r' )r# r) r
� s8
zSpacewalkRepo.__init__c C s� g }| j s|S xb| jD ]X}|| j kr8t| }tjj|��| j | dkrV|jd| � q|jd|| j | f � qW | js�|jd� |S )N� z*%s:
X-libcurl-Empty-Header-Workaround: *z%s: %sz.X-RHN-Transport-Capability: follow-redirects=3)Nrz )r4 �needed_headers�MISSING_HEADERr �ErrorZ RepoError�appendr2 )r# ry �headerr< r( r( r) rx s
z!SpacewalkRepo.create_http_headersTc s0 t t| �j||�}| j� }|r,|jtj|� |S )N)r rJ �_handle_new_remoterx Zsetopt�librepoZLRO_HTTPHEADER)r# ZdestdirZ mirror_setupZhandlery )r' r( r) r� & s
z SpacewalkRepo._handle_new_remote)T) rk rl rm �__doc__r{ r
rx r� rn r( r( )r' r) rJ � s %rJ c C s* | j � }x |D ]}t|�st|��qW |S )a
Parse the key urls and validate them.
key_url_string is a space seperated list of gpg key urls that must be
located in /etc/pkg/rpm-gpg/.
Return a list of strings containing the key urls.
Raises InvalidGpgKeyLocation if any of the key urls are invalid.
)�split�is_valid_gpg_key_urlrs )Zkey_url_stringZkey_urls�key_urlr( r( r) rr 2 s
rr c @ s e Zd ZdS )rs N)rk rl rm r( r( r( r) rs A s rs c C sP | j d�}t|�dkrdS |\}}|j� dkr2dS tjj|�}|jd�sLdS dS )Nz://� F�filez/etc/pki/rpm-gpg/T)r� �len�lowerr r �normpath�
startswith)r� Z proto_split�protor r( r( r) r� D s
r� c C s4 d| ko| d st �| d }t|�tkr0|d S |S )NZ sslCACertr )ZBadSslCaCertConfig�type�list)r Zca_certsr( r( r) r: R s r: )/Z
__future__r r Zdnfpluginscorer r r Zdnf.exceptionsr[ rW r� r r Zdnf.conf.configr Zup2date_client.up2dateAuthr9 Zup2date_client.configZup2date_client.rhnChannelZ up2date_client.rhnPackageInfoZrhn.i18nr r r r> rB r= r? rA rC rD rF rt rQ r| r8 r ZPluginr
rN ZReporJ rr � Exceptionrs r� r: r( r( r( r) �<module> sJ -O