3
�XjS; � @ sB d Z ddlZddlZddlZddlZddlmZmZ ddlm Z m
Z
yddl m Z m
Z
W n$ ek
r| ddlm Z m
Z
Y nX ddlZejddd�Zeed �s�eje_ejZd9d:d;dd?d@dAdBdCdDdEd d d d!gfd"�Zd#d d$gZG d%d&� d&�ZG d'd(� d(�Zd)d*� Zd+d,� Zd-d.� Z dFd0d1�Z dGd2d3�Z dHd4d5�Z dId7d8�Z dS )Jzj
This module includes the Config and Up2date Config classes use by the
up2date agent to hold config info.
� N)�idn_ascii_to_puny�idn_puny_to_unicode)�ustr�sstr)�urlsplit�
urlunsplitzrhn-client-toolsT)Zfallback�ugettext�Use a HTTP Proxy�Remote server URL�)https://xmlrpc.cln.cloudlinux.com/XMLRPC/�&Remote server URL for access over IPv6�.https://ipv6.xmlrpc.cln.cloudlinux.com/XMLRPC/�Mirror list URL�>https://repo.cloudlinux.com/cloudlinux/mirrorlists/cln-mirrors�#Whether or not debugging is enabled�Location of system id�/etc/sysconfig/rhn/systemid�4Override the automatically determined system version� �;HTTP proxy in host:port format, e.g. squid.example.com:3128�'The username for an authenticated proxy�.The password to use for an authenticated proxy�$To use an authenticated proxy or not�BNumber of attempts to make at network connections before giving up� �)The CA cert used to verify the ssl server�'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERTzDConfig options that can not be overwritten by a config update action� sslCACert� serverURL�disallowConfChanges)Z enableProxyr �
serverURLipv6Z mirrorURL�debug� systemIdPathZversionOverride� httpProxy� proxyUser�
proxyPasswordZenableProxyAuthZnetworkRetriesr r r" ZtmpDirc @ sl e Zd ZdZddd�Zddd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
ConfigFilez;class for handling persistent config options for the clientNc C s i | _ || _| jr| j� d S )N)�dict�fileName�load)�self�filename� r, � /usr/lib/python3.6/config.py�__init__B s zConfigFile.__init__c C s� |r
|| _ | j d krd S tj| j tj�s,d S t| j d�}d}�x�|j� D �]�}|jd�dkr^qH||j� }|spqH|d dkr�|d d� j� }qHnd}|jdd�}t |�dkr�qH|d j� }t
|d j� �}d }|jd �} | dkr�|d | � }|}d }|�rx|jd
�}
|dk�r"t |j t
j� ��}nVt |
�dk�rVy t|�}W n tk
�rR Y nX n"|
d dk�rjg }nd
d� |
D �}||f} || jk�r�| j| } |d k �r�|| d f} |d k �r�| d |f} | | j|< qHW |j� d S )N�rr �#r r �\�=� z [comment]�;r$ r% c S s g | ]}|j � r|j � �qS r, )�strip)�.0�valr, r, r- �
� s z#ConfigFile.load..�����r9 r9 )r$ r% )r( �os�access�R_OK�open� readlines�findr5 �split�lenr r �encode�locale�getpreferredencoding�int�
ValueErrorr' �close) r* r+ �fZ multiline�liner@ �key�value�comment�pos�valuesZnewvalr, r, r- r) H sd
zConfigFile.loadc C s | j d krd S tj| j tj�sTtjtjj| j �tj�sTttdtjj| j � �� d S x~| jj � D ]p}| j| \}}t
|�t
g �kr�|n|g}xB||gt |� D ].}dt |�ks�dt |�kr�t
d|| j f ��q�W q`W tj| j d tjtjB tjB tjB tjB d�}y tj|d� tj|d�}W n" tk
�r@ tj|� � Y nX |jd� |jd � |jd
� x�| jj � D ]�}| j| \} }
|jtd || f �� t
|
�t
g �k�r�|
g}
|tk�r�ttjj |
�}
|jtd |d
j tt |
��f �� |jd� �qlW |j� tj | j d | j � d S )Nz%s was not found�
�
z[refusing to write config option %r to %s: a key, comment, or value contains a CR/LF newlinez.newi� �wzI# Automatically generated Red Hat Update Agent config file, do not edit.
z# Format: 1.0
r z%s[comment]=%s
z%s=%s
r4 ) r( r: r; r< �path�dirname�print�_r' �keys�type�list�strrF r= �O_WRONLY�O_CREAT�O_TRUNC�
O_NOFOLLOW� O_CLOEXEC�fchmod�fdopen� ExceptionrG �writer � FileOptions�map�abspath�join�rename) r* Z_keyZ_commentZ_valueZ_valuesZ_field�fdrH rJ rL rK r, r, r- �save� sL
"zConfigFile.savec C s
|| j kS )N)r' )r* �namer, r, r- � __contains__� s zConfigFile.__contains__c C s || kS )Nr, )r* rj r, r, r- �has_key� s zConfigFile.has_keyc C s
| j j� S )N)r' rV )r* r, r, r- rV � s zConfigFile.keysc C s dd� | j j� D �S )Nc S s g | ] }|d �qS )r r, )r6 �ar, r, r- r8 � s z%ConfigFile.values..)r' rN )r* r, r, r- rN � s zConfigFile.valuesc C s | j j|� d S )N)r' �update)r* r' r, r, r- rn � s zConfigFile.updatec C s || j kr| j | d S d S )Nr )r' )r* rj r, r, r- � __getitem__� s
zConfigFile.__getitem__c C s0 || j kr| j | }nd}|d |f| j |< d S )Nr )NN)r' )r* rj rK r7 r, r, r- � __setitem__� s
zConfigFile.__setitem__c C s || j kr| j | d S dS )Nr r )r' )r* rj r, r, r- �info� s
zConfigFile.info)N)N)�__name__�
__module__� __qualname__�__doc__r. r) ri rk rl rV rN rn ro rp rq r, r, r, r- r& ? s
HBr& c @ sv e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �ConfigNc C s. t � | _| jjt� |r$| jj|� i | _d S )N)r& �storedrn �Defaultsr) �runtime)r* r+ r, r, r- r. � s
zConfig.__init__c C s || j krdS || jkr dS dS )NTF)ry rw )r* rj r, r, r- rk s
zConfig.__contains__c C s || kS )Nr, )r* rj r, r, r- rl
s zConfig.has_keyc C s8 t | jj� �}x$| jj� D ]}||kr|j|� qW |S )N)rX ry rV rw �append)r* �ret�kr, r, r- rV s
z Config.keysc C s* g }x | j � D ]}|j| j|�� qW |S )N)rV rz ro )r* r{ r| r, r, r- rN s z
Config.valuesc C s. g }x$| j � D ]}|j|| j|�f� qW |S )N)rV rz ro )r* r{ r| r, r, r- �items s z Config.itemsc C s t | j� �S )N)rA rV )r* r, r, r- �__len__! s zConfig.__len__c C s || j |< d S )N)ry )r* rj rK r, r, r- rp $ s zConfig.__setitem__c C s, || j kr| j | S || jkr(| j| S d S )N)ry rw )r* rj r, r, r- ro ( s
zConfig.__getitem__c C s | j j|�S )N)rw rq )r* rj r, r, r- rq 1 s z Config.infoc C s | j j� d S )N)rw ri )r* r, r, r- ri 4 s z Config.savec C s8 | j j|� x&| j j� D ]}|| jkr(q| j|= qW d S )N)rw r) rV ry )r* r+ r| r, r, r- r) 7 s
z Config.loadc C s || j |< || jkr | j|= d S )N)rw ry )r* rj rK r, r, r- �setA s
z
Config.set)N)rr rs rt r. rk rl rV rN r} r~ rp ro rq ri r) r r, r, r, r- rv � s
rv c C s^ t � } d}| d }|rZ|dd� dkr2|dd� }|jd�}tt|d ��|d< dj|�}|S )zkreturns proxy string in format hostname:port
hostname is converted to Punycode (RFC3492) if needed
Nr# � zhttp://�:r )�initUp2dateConfigr@ rY r rf )�cfg�proxyZ proxyHost�partsr, r, r- �getProxySettingH s
r� c C s: t | �}tt|d tt|d ��|d |d |d f��S )z=returns url where hostname is converted to Punycode (RFC3492)r r r3 � � )r r r r r )Zurl�sr, r, r- �convert_url_to_punyZ s r� c C s& t | t�rdd� | D �S t| �gS dS )z�
Internal function to process server URL to Punycode format.
Processes both single URLs and lists of URLs.
:param serverUrl: URL or list of URLs to process.
:return: List of processed URLs in Punycode format.
c S s g | ] }t |��qS r, )r� )r6 �ir, r, r- r8 j s z$processServerURL..N)�
isinstancerX r� )Z serverUrlr, r, r- �processServerURL` s
r� Fc C s^ t � }t|d �}t|d �}||f}||f}| r6|S ddlm} |j� }|dkrZ|rZ|S |S )a�
Return a pair of server URLs (primary and fallback) based on the preferred interface.
:param registration: If True, it indicates that the URLs are being fetched for registration purposes.
If that is the case, the function will always use IPv4 URLs with IPv6 as fallback,
regardless of the preferred_interface setting.
Assumed false by default, expected to be specified explicitly when needed.
:return: Pair of server URL configs, first is the primary, second is the fallback.
r r r )� up2dateAuthZIPv6)r� r� Zup2date_clientr� ZgetPreferredInterface)� registrationr� Zipv4_urlZipv6_urlZipv4_primary_pairZipv6_primary_pairr� Zpreferred_interfacer, r, r- �getServerURLPairo s r� c C s t | d�d S )az
Return the primary server URL from config based on preferred_interface.
If preferred_interface=IPv6 in system_id, returns serverURLipv6 if available.
Otherwise returns normal serverURL.
Note: the config may have one value or a list of values, but this function always returns a list.
:return: List of server URLs with hostnames converted to Punycode.
)r� r )r� )r� r, r, r- � getServerURL� s
r� c C s t | d�d S )z�
Determine the fallback server URL from system_id.
The fallback server URL is the one that is *not* set as preferred_interface.
If preferred_interface is IPv6, it returns serverURL (i.e. IPv4 host), otherwise serverURLipv6.
)r� r )r� )r� r, r, r- �getFallbackServerURL� s r� �/etc/sysconfig/rhn/up2datec C sP yt W n tk
r da Y nX t dkrLt| �a dt d< tjj� rLdt d< t S )z sV
;O
'