3
m��i 5 � @ 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;d<d=d>d?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- �
<listcomp>� s z#ConfigFile.load.<locals>.<listcomp>���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 sJ | j d krd S tj| j tj�sTtjtjj| j �tj�sTttdtjj| j � �� d S t| j d d�}tj | j d t
dd�� |j d� |j d� |j d� x�| j j
� D ]�}| j | \}}|j td ||f �� t|�tg �kr�|g}|tkr�ttjj|�}|j td
|d jtt|��f �� |j d � q�W |j� tj| j d | j � d S )
Nz%s was not foundz.new�wZ0644� zI# 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= �chmodrE �writer' �keysr �type� FileOptions�map�abspath�join�strrG �rename)r* rH rJ rL rK r, r, r- �save� s,
"zConfigFile.savec C s
|| j kS )N)r' )r* �namer, r, r- � __contains__� s zConfigFile.__contains__c C s || kS )Nr, )r* ra r, r, r- �has_key� s zConfigFile.has_keyc C s
| j j� S )N)r' rX )r* r, r, r- rX � 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.<locals>.<listcomp>)r' rN )r* r, r, r- rN � s zConfigFile.valuesc C s | j j|� d S )N)r' �update)r* r' r, r, r- re � s zConfigFile.updatec C s || j kr| j | d S d S )Nr )r' )r* ra r, r, r- � __getitem__� s
zConfigFile.__getitem__c C s0 || j kr| j | }nd}|d |f| j |< d S )Nr )NN)r' )r* ra rK r7 r, r, r- � __setitem__� s
zConfigFile.__setitem__c C s || j kr| j | d S dS )Nr r )r' )r* ra r, r, r- �info� s
zConfigFile.info)N)N)�__name__�
__module__� __qualname__�__doc__r. r) r` rb rc rX rN re rf rg rh r, r, r, r- r& ? s
H#r& 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& �storedre �Defaultsr) �runtime)r* r+ r, r, r- r. � s
zConfig.__init__c C s || j krdS || jkr dS dS )NTF)rp rn )r* ra r, r, r- rb � s
zConfig.__contains__c C s || kS )Nr, )r* ra r, r, r- rc � s zConfig.has_keyc C s8 t | jj� �}x$| jj� D ]}||kr|j|� qW |S )N)�listrp rX rn �append)r* �ret�kr, r, r- rX � s
z Config.keysc C s* g }x | j � D ]}|j| j|�� qW |S )N)rX rr rf )r* rs rt r, r, r- rN � s z
Config.valuesc C s. g }x$| j � D ]}|j|| j|�f� qW |S )N)rX rr rf )r* rs rt r, r, r- �items� s z Config.itemsc C s t | j� �S )N)rA rX )r* r, r, r- �__len__ s zConfig.__len__c C s || j |< d S )N)rp )r* ra rK r, r, r- rg s zConfig.__setitem__c C s, || j kr| j | S || jkr(| j| S d S )N)rp rn )r* ra r, r, r- rf s
zConfig.__getitem__c C s | j j|�S )N)rn rh )r* ra r, r, r- rh s z Config.infoc C s | j j� d S )N)rn r` )r* r, r, r- r` s z Config.savec C s8 | j j|� x&| j j� D ]}|| jkr(q| j|= qW d S )N)rn r) rX rp )r* r+ rt r, r, r- r) s
z Config.loadc C s || j |< || jkr | j|= d S )N)rn rp )r* ra rK r, r, r- �set" s
z
Config.set)N)ri rj rk r. rb rc rX rN ru rv rg rf rh r` r) rw r, r, r, r- rm � s
rm 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@ r^ r r] )�cfg�proxyZ proxyHost�partsr, r, r- �getProxySetting) 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_puny; 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 K s z$processServerURL.<locals>.<listcomp>N)�
isinstancerq r� )Z serverUrlr, r, r- �processServerURLA 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)rz r� Zup2date_clientr� ZgetPreferredInterface)� registrationr{ Zipv4_urlZipv6_urlZipv4_primary_pairZipv6_primary_pairr� Zpreferred_interfacer, r, r- �getServerURLPairP 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- � getServerURLw 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<This function is the right way to get at the up2date config.NF�isattyT)r{ � NameErrorrm �sys�stdoutr� )Zcfg_filer, r, r- rz � s
rz )r r )r
r )r r
)r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )F)F)F)r� )!rl r: r� rC ZrequestsZrhn.connectionsr r Zrhn.i18nr r Zurlparser r � ImportErrorZ urllib.parse�gettextZ translation�t�hasattrr rU ro rZ r& rm r~ r� r� r� r� r� rz r, r, r, r- �<module> sV
O
'