a
mi 5 @ s< d Z ddlZddlZddlZddlZddlmZmZ ddlm Z m
Z
zddl m Z m
Z
W n" eyz ddlm Z m
Z
Y n0 ddlZejdddZeed seje_ejZd
d d d
ddddddddddg dfdZg dZG dd d ZG d d d Zd d Zd!d" Zd#d$ Z d/d&d'Z d0d(d)Z d1d*d+Z d2d-d.Z dS )3zj
This module includes the Config and Up2date Config classes use by the
up2date agent to hold config info.
N)idn_ascii_to_punyidn_puny_to_unicode)ustrsstr)urlsplit
urlunsplitzrhn-client-toolsT)fallbackugettext)zUse a HTTP Proxyr )zRemote server URLz)https://xmlrpc.cln.cloudlinux.com/XMLRPC/)z&Remote server URL for access over IPv6z.https://ipv6.xmlrpc.cln.cloudlinux.com/XMLRPC/)zMirror list URLz>https://repo.cloudlinux.com/cloudlinux/mirrorlists/cln-mirrors)z#Whether or not debugging is enabledr )zLocation of system idz/etc/sysconfig/rhn/systemid)z4Override the automatically determined system version )z;HTTP proxy in host:port format, e.g. squid.example.com:3128r
)z'The username for an authenticated proxyr
)z.The password to use for an authenticated proxyr
)z$To use an authenticated proxy or notr )zBNumber of attempts to make at network connections before giving up )z)The CA cert used to verify the ssl serverz'/usr/share/rhn/CLN-ORG-TRUSTED-SSL-CERTzDConfig options that can not be overwritten by a config update action) sslCACert serverURLdisallowConfChanges)Z enableProxyr
serverURLipv6Z mirrorURLdebug systemIdPathZversionOverride httpProxy proxyUser
proxyPasswordZenableProxyAuthZnetworkRetriesr r )r r ZtmpDirc @ sl e Zd ZdZdddZdddZdd 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| d S N)dictfileNameloadselffilename r 9/usr/lib/python3.9/site-packages/up2date_client/config.py__init__B s zConfigFile.__init__c C s |r
|| _ | j d u rd S t| j tjs,d S t| j d}d}| D ]}|ddkrZqD|| }|slqD|d dkr|d d }qDnd}|dd}t |d krqD|d }t
|d }d }|d
} | dkr|d | }|}d }|rr|d }
|d v r t | t
}nTt |
dkrPz t|}W n tyL Y n0 n"|
d dkrdg }nd
d |
D }||f} || jv r| j| } |d ur|| d f} |d ur| d |f} | | j|< qD| d S )Nrr
#r \=r z [comment];)r r c S s g | ]}| r| qS r )strip).0valr r r
z#ConfigFile.load..)r osaccessR_OKopen readlinesfindr' splitlenr r encodelocalegetpreferredencodingint
ValueErrorr close) r r fZ multilineliner2 keyvaluecommentposvaluesZnewvalr r r r H sd
zConfigFile.loadc C sF | j d u rd S t| j tjsTttj| j tjsTttdtj| j d S t| j d d}t | j d t
dd | d | d | d | j
D ]}| j | \}}| td ||f t|tg kr|g}|tv rttjj|}| td
|d tt|f | d q| t| j d | j d S )
Nz%s was not foundz.newwZ0644 zI# Automatically generated Red Hat Update Agent config file, do not edit.
z# Format: 1.0
r
z%s[comment]=%s
z%s=%s
r&
)r r, r- r. pathdirnameprint_r/ chmodr7 writer keysr type FileOptionsmapabspathjoinstrr9 rename)r r: r< r> r= r r r save s,
" zConfigFile.savec C s
|| j v S r r r namer r r __contains__ s zConfigFile.__contains__c C s || v S r r rT r r r has_key s zConfigFile.has_keyc C s
| j S r )r rJ r r r r rJ s zConfigFile.keysc C s dd | j D S )Nc S s g | ] }|d qS )r r )r( ar r r r* r+ z%ConfigFile.values..)r r@ rX r r r r@ s zConfigFile.valuesc C s | j | d S r )r update)r r r r r rZ s zConfigFile.updatec C s || j v r| j | d S d S )Nr rS rT r r r __getitem__ s
zConfigFile.__getitem__c C s0 || j v r| j | }nd}|d |f| j |< d S )N)NNr rS )r rU r= r) r r r __setitem__ s
zConfigFile.__setitem__c C s || j v r| j | d S dS )Nr r
rS rT r r r info s
zConfigFile.info)N)N)__name__
__module__ __qualname____doc__r r rR rV rW rJ r@ rZ r[ r\ r] r r r r r ? s
H#r c @ sv e Zd Zd ddZdd Zdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS ) ConfigNc C s. t | _| jt |r$| j| i | _d S r )r storedrZ Defaultsr runtimer r r r r s
zConfig.__init__c C s || j v rdS || jv r dS dS )NTFre rc rT r r r rV s
zConfig.__contains__c C s || v S r r rT r r r rW s zConfig.has_keyc C s4 t | j }| j D ]}||vr|| q|S r )listre rJ rc appendr retkr r r rJ s
z Config.keysc C s& g }| D ]}|| | q |S r rJ rh r[ ri r r r r@ s z
Config.valuesc C s* g }| D ]}||| |f q |S r rl ri r r r items s z Config.itemsc C s t | S r )r3 rJ rX r r r __len__ s zConfig.__len__c C s || j |< d S r )re r rU r= r r r r\ s zConfig.__setitem__c C s, || j v r| j | S || jv r(| j| S d S r rf rT r r r r[ s
zConfig.__getitem__c C s | j |S r )rc r] rT r r r r] s z Config.infoc C s | j d S r )rc rR rX r r r rR s z Config.savec C s4 | j | | j D ]}|| jvr&q| j|= qd S r )rc r rJ re )r r rk r r r r s
z Config.loadc C s || j |< || jv r | j|= d S r )rc re ro r r r set" s
z
Config.set)N)r^ r_ r` r rV rW rJ r@ rm rn r\ r[ r] rR r rp r r r r rb s
rb c C s^ t } d}| d }|rZ|dd dkr2|dd }|d}tt|d |d< d|}|S )zkreturns proxy string in format hostname:port
hostname is converted to Punycode (RFC3492) if needed
Nr zhttp://:r )initUp2dateConfigr2 rP r rO )cfgproxyZ proxyHostpartsr r r getProxySetting) s
rw c C s: t | }tt|d tt|d |d |d |d fS )z=returns url where hostname is converted to Punycode (RFC3492)r r r% )r r r r r )urlsr r r convert_url_to_puny; s r| c C s& t | trdd | 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| )r( ir r r r* K r+ z$processServerURL..N)
isinstancerg r| )Z serverUrlr r r processServerURLA s
r Fc C s^ t }t|d }t|d }||f}||f}| r6|S ddlm} | }|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)rs r Zup2date_clientr ZgetPreferredInterface) registrationrt Zipv4_urlZipv6_urlZipv4_primary_pairZipv6_primary_pairr Zpreferred_interfacer r r getServerURLPairP s r c C s t | dd 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 | dd 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 sN zt W n ty da Y n0 t dkrJt| a dt d< tj rJdt d< t S )z sV
O
'