|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/uaclient/__pycache__/ |
Upload File : |
o
��Jh�K � @ s d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZmZmZ d dlmZmZmZmZmZmZmZmZm Z d dl m Z m Z d dl m!Z! e�"e�#e$��Z%dd d
�Z&d Z'd Z(d
Z)e �*� Z+G dd� d�Z,de-fdd�Z.ddd�Z/de-fdd�Z0dS )� N)� lru_cache�wraps)�Any�Dict�Optional)�apt� event_logger�
exceptions�http�messages�snap�system�util) � APT_NEWS_URL�BASE_CONTRACT_URL�BASE_LIVEPATCH_URL�BASE_SECURITY_URL�BASE_VULNERABILITY_DATA_URL�CONFIG_DEFAULTS� CONFIG_FIELD_ENVVAR_ALLOWLIST�DEFAULT_CONFIG_FILE�DEFAULT_DATA_DIR)� state_files�user_config_file)� safe_load�name�type)�availableResources�resourceEntitlements�_unset)�
http_proxy� https_proxy�apt_http_proxy�apt_https_proxy�ua_apt_http_proxy�ua_apt_https_proxy�global_apt_http_proxy�global_apt_https_proxy�update_messaging_timer�metering_timer�apt_news� apt_news_url� vulnerability_data_url_prefix�lxd_guest_attach)
� contract_url�data_dir�features�log_file� log_level� security_urlr, �settings_overrides� ua_config�
livepatch_urlc @ s e Zd ZdZdZdZ dIdeeee f dee
j dee ddfd d
�Z e
defd d ��Ze
defd
d��Ze
defdd��Ze
dee fdd��Zejdefdd��Ze
dee fdd��Zejdefdd��Ze
dee fdd��Zejdefdd��Ze
dee fdd ��Zejdefd d ��Ze
defd d ��Zejdefd d ��Ze
edd!�dee fd"d#���Zejdefd$d#��Ze
edd!�dee fd%d&���Zejdefd'd&��Ze
defd(d)��Zejdefd*d)��Ze
defd+d,��Z e jdefd-d,��Z e
de fd.d/��Z e jde fd0d/��Z e
defd1d2��Z e jdefd3d2��Z e
de fd4d5��Z e jde fd6d5��Z e
defd7d8��Z!e!jdefd9d8��Z!e
de
j"fd:d;��Z#e#jde
j"fd<d;��Z#e
d=d>� �Z$e
d?d@� �Z%e
defdAdB��Z&e
dCdD� �Z'dEdF� Z(dGdH� Z)dS )J�UAConfig)r$ r% )r& r' )r" r# N�cfg� user_config�series�returnc
C s� |r d| _ || _d| _n
t� | _ t| j �\| _| _|r || _n,ztj�� | _W n# tyJ } zt j
d|d� t �
d� t� � | _W Y d}~nd}~ww d| jv rctj j i | j�
� �| jd �dd�| _|| _dS )� NzError loading user config)�exc_infozUsing default config valuesr5 T)� optional_type_errors_become_null)�cfg_pathr8 � invalid_keys�get_config_path� parse_configr9 r �read� Exception�LOG�warning�UserConfigData� from_dict�to_dictr: )�selfr8 r9 r: �e� rL �1/usr/lib/python3/dist-packages/uaclient/config.py�__init__V s,
��
�
zUAConfig.__init__c C � | j �dt�S )Nr. )r8 �getr �rJ rL rL rM r. y � zUAConfig.contract_urlc C rO )Nr3 )r8 rP r rQ rL rL rM r3 } rR zUAConfig.security_urlc C rO )Nr6 )r8 rP r rQ rL rL rM r6 � rR zUAConfig.livepatch_urlc C � | j jS �N)r9 r rQ rL rL rM r � � zUAConfig.http_proxy�valuec C � || j _tj �| j � d S rT )r9 r r �write�rJ rV rL rL rM r � � c C rS rT )r9 r! rQ rL rL rM r! � rU zUAConfig.https_proxyc C rW rT )r9 r! r rX rY rL rL rM r! � rZ c C rS rT )r9 r% rQ rL rL rM r% � rU zUAConfig.ua_apt_https_proxyc C rW rT )r9 r% r rX rY rL rL rM r% � rZ c C rS rT )r9 r$ rQ rL rL rM r$ � rU zUAConfig.ua_apt_http_proxyc C rW rT )r9 r$ r rX rY rL rL rM r$ � rZ c C � | j j}|d u r
tS |S rT )r9 r, r �rJ �valrL rL rM r, � � z&UAConfig.vulnerability_data_url_prefixc C rW rT )r9 r, r rX rY rL rL rM r, � rZ )�maxsizec C �: | j j}|r|S | j j}|rt�tjjddd�� |S d S )Nr" r& ��old�new)r9 r& r" �event�infor �WARNING_DEPRECATED_FIELD�format�rJ �
global_val� old_apt_valrL rL rM r& � � ��z UAConfig.global_apt_http_proxyc C �. || j _d | j _tjj�� tj �| j � d S rT )r9 r& r" r7 �fget� cache_clearr rX rY rL rL rM r& � � c C r` )Nr# r' ra )r9 r' r# rd re r rf rg rh rL rL rM r' � rk z UAConfig.global_apt_https_proxyc C rl rT )r9 r' r# r7 rm rn r rX rY rL rL rM r' � ro c C � | j j}|d u r
dS |S )Ni`T )r9 r( r\ rL rL rM r( � r^ z UAConfig.update_messaging_timerc C rW rT )r9 r( r rX rY rL rL rM r( � rZ c C rp )Ni@8 )r9 r) r\ rL rL rM r) � r^ zUAConfig.metering_timerc C rW rT )r9 r) r rX rY rL rL rM r) � rZ c C rp )NF)r9 �poll_for_pro_licenser\ rL rL rM rq � s z UAConfig.poll_for_pro_licensec C rW rT )r9 rq r rX rY rL rL rM rq rZ c C rp )NiX )r9 �polling_error_retry_delayr\ rL rL rM rr s z"UAConfig.polling_error_retry_delayc C rW rT )r9 rr r rX rY rL rL rM rr rZ c C rp )NT)r9 r* r\ rL rL rM r* r^ zUAConfig.apt_newsc C rW rT )r9 r* r rX rY rL rL rM r* ! rZ c C r[ rT )r9 r+ r r\ rL rL rM r+ & r^ zUAConfig.apt_news_urlc C rW rT )r9 r+ r rX rY rL rL rM r+ - rZ c C s | j j}|d u r tjjS |S rT )r9 r- r �LXDGuestAttachEnum�OFFr\ rL rL rM r- 2 s zUAConfig.lxd_guest_attachc C s* |d u rt jj}|| j_t j�| j� d S rT )r rs rt r9 r- rX rY rL rL rM r- 9 s c C rO )Nr/ )r8 rP r rQ rL rL rM r/ @ rR zUAConfig.data_dirc C s8 | j �dd�}ztt|�� �W S ty tj Y S w )Nr2 �DEBUG)r8 rP �getattr�logging�upper�AttributeErrorru )rJ r2 rL rL rM r2 D s
�zUAConfig.log_levelc C s | j �dtd �S )Nr1 )r8 rP r rQ rL rL rM r1 L s zUAConfig.log_filec C s. | j �d�}|rt|t�r|S t�d|� i S )z>Return a dictionary of any features provided in uaclient.conf.r0 zDUnexpected uaclient.conf features value. Expected dict, but found %s)r8 rP �
isinstance�dictrE rF )rJ r0 rL rL rM r0 P s
�zUAConfig.featuresc
C sV dD ]"}t | |�}|d u rt�d|� qt|t�r |dk r$tj||d��q| js+| jr5| j s1| j
r5t� � �t �
d| jt j� t �
d| jt j� t �
d| j t j� t �
d| j
t j� t �
d| jt j� t �
d| jt j� | jsq| jr}t�| j| jtjj� n| j s�| j
r�t�| j | j
tjj� g }t�� r�t�| j| j� | js�t� tj �s�| js�t� tj �r�|� d� ddl m!} dd l"m#} || �j$}t%d
d � |D ��}|r�|�&| j| j� | js�|� |j �s�| js�|� |j �r�|� d � t'|�dk�rd
�(|�}t)t*j+j,|d�� ddl-m.} | j/t0j1j2k�r | | �j3�s t)t*j4� d S t5j6�7t5j8| j/d�� d S )N)r( r) z1No config set for %s, default value will be used.r ��keyrV r
�httpsr )� livepatch)�_enabled_servicesc s s � | ]
}|j d kr|V qdS )r N)r )�.0�entrL rL rM � <genexpr>� s � �z*UAConfig.process_config.<locals>.<genexpr>r z, )�services)� _is_attached)� guest_attach)9rv rE �debugrz �intr �InvalidPosIntConfigValuer& r' r$ r% � InvalidProxyCombinationConfigr
�validate_proxy� PROXY_VALIDATION_APT_HTTP_URL� PROXY_VALIDATION_APT_HTTPS_URLr � PROXY_VALIDATION_SNAP_HTTP_URLr! � PROXY_VALIDATION_SNAP_HTTPS_URLr �setup_apt_proxy�
AptProxyScope�GLOBAL�UACLIENTr �is_snapd_installed�configure_snap_proxy�get_config_option_value�HTTP_PROXY_OPTION�HTTPS_PROXY_OPTION�append�uaclientr �-uaclient.api.u.pro.status.enabled_services.v1r� �enabled_services�any�configure_livepatch_proxy�len�join�printr �!PROXY_DETECTED_BUT_NOT_CONFIGUREDrg �(uaclient.api.u.pro.status.is_attached.v1r� r- r rs rt � is_attached�-WARNING_LXD_GUEST_ATTACH_SET_BUT_NOT_ATTACHEDr �lxd_pro_config_filerX � LXDProConfig)
rJ �proprV �services_with_proxiesr r� r� �livepatch_enabledr� r� rL rL rM �process_config_ s�
��� ����
��
�
� � ��
��
�
��������
�� � �zUAConfig.process_configc C s` | j d urt| j �D ]}t�d|� q
d| jv r.t�d� t�d� t�d� t�d� d S d S )Nz&Ignoring invalid uaclient.conf key: %sr5 z)legacy "ua_config" found in uaclient.confzPlease do the following:z� 1. run `sudo pro config set field=value` for each field/value pair present under "ua_config" in /etc/ubuntu-advantage/uaclient.confzQ 2. Delete "ua_config" and all sub-fields in /etc/ubuntu-advantage/uaclient.conf)r@ �sortedrE rF r8 )rJ � invalid_keyrL rL rM �warn_about_invalid_keys� s
�
���z UAConfig.warn_about_invalid_keys)NNN)*�__name__�
__module__� __qualname__�ua_scoped_proxy_options�global_scoped_proxy_options�&deprecated_global_scoped_proxy_optionsr r �strr r rG rN �propertyr. r3 r6 r �setterr! r% r$ r, r r& r' r� r( r) �boolrq rr r* r+ rs r- r/ r2 r1 r0 r� r� rL rL rL rM r7 K s� ����
�#
vr7 r; c C s t j�d�} | r
| S tS )z4Get config path to be used when loading config dict.�UA_CONFIG_FILE)�os�environrP r )� config_filerL rL rM rA � s rA c C sp t � t�}| s
t� } t�d| � tj�| �r |�t t
� | ��� i }tj �
� D ]Q\}}|�� }|�d�rx|dd� }|�d�rp|dd� }|�d�r^tj�|�rXt t
� |��}ntj|d��d |vri||i|d < q'||d |< q'|tv rx|||< q'|�|� d
|v r�tj�|d
�|d
< d D ]}t�|| �s�tj||| d ��q�t|�� ��t�}|D ]}|� |� q�||fS )
a� Parse known Pro config file
Attempt to find configuration in cwd and fallback to DEFAULT_CONFIG_FILE.
Any missing configuration keys will be set to CONFIG_DEFAULTS.
Values are overridden by any environment variable with prefix 'UA_'.
@param config_path: Fullpath to pro configfile. If unspecified, use
DEFAULT_CONFIG_FILE.
@return: Dict of configuration values.
z%Using client configuration file at %s�ua_� N� features_� �yaml)�filepathr0 r/ )r. r3 r| ) �copyr rA rE r� r� �path�exists�updater r
� load_filer� �items�lower�
startswith�endswithr � InvalidFeatureYamlConfigValuer �
expanduserr
�is_service_url�InvalidURLConfigValue�set�keys�
difference�VALID_UA_CONFIG_KEYS�pop) � config_pathr8 �env_keysr} rV �
field_name�feature_field_namer@ r� rL rL rM rB � sH
��
� rB � override_keyc s � fdd�}|S )a� Decorator used to override function return by config settings.
To identify if we should override the function return, we check
if the config object has the expected override key, we use it
has, we will use the key value as the function return. Otherwise
we will call the function normally.
@param override_key: key to be looked for in the settings_override
entry in the config dict. If that key is present, we will return
its value as the function return.
c s t � �� �fdd��}|S )Nc s@ t � \} }| �di ���t�}|tkr �dkr|d fS |S � � S )Nr4 �
cloud_type)rB rP �UNSET_SETTINGS_OVERRIDE_KEY)r8 �_�value_override)�fr� rL rM �new_fA s
�z>apply_config_settings_override.<locals>.wrapper.<locals>.new_f)r )r� r� �r� )r� rM �wrapper@ s
z/apply_config_settings_override.<locals>.wrapperrL )r� r� rL r� rM � apply_config_settings_override3 s
r� rT )1r� rw r� � functoolsr r �typingr r r r� r r r r
r r r
r �uaclient.defaultsr r r r r r r r r �uaclient.filesr r �
uaclient.yamlr � getLogger� replace_top_level_logger_namer� rE �MERGE_ID_KEY_MAPr� �UA_CONFIGURABLE_KEYSr� �get_event_loggerrd r7 r� rA rB r� rL rL rL rM �<module> s0 (,
� #
?