ruạṛ
a ��;j�& � 5 @ s| d dl Z d dlZd dlZddlmZmZmZmZmZ ddl m Z dZg d�Ze� Zdd� Zd d � Ze�d �Zd d � Zdd� deddd� eeeedeeeeeeeeedd� eeeeeeeedd� ejdd� dd� ejdejdd� dd� eeeeejddejeeeddeed�4Z dd� Z dd� Z G d d � d e �Z d d � Z!d d!� Z"d"d#� Z#d$d%� Z$ej%ej&d&�d'd(� �Z'ej%ej&d&�d)d*� �Z(dS )+� N� )�config� constants� http_utils� log_utils�utils)� ConfigParserz /etc/sysconfig/kcare/kcare.conf)�USE_CONTENT_FILE_V3�FORCE_JSON_SIG_V3�ENABLE_CRASHREPORTER� KCORE_OUTPUT� KMSG_OUTPUT�SEND_PERF_METRICS� KERNEL_ANOMALY_REPORT_ENABLE�FORCE_DOCTOR_FALLBACKc C s | � � dv S )N)�1�TRUE�YES�Y)�upper��value� r �5/usr/libexec/kcare/python/kcarectl/config_handlers.py�bool_converter# s r c C s dd� | � d�D �S )Nc S s g | ]}|� � r|� � �� �qS r )�strip�lower)�.0�itemr r r � <listcomp>( � z comma_list.<locals>.<listcomp>�,)�splitr r r r � comma_list' s r# z^[a-z][a-z0-9_.+\-]*$c C sN g g }}t | �D ] }t�|�r$|n|�|� q|rJt�d�d�|��� |S )Nz3LIBCARE_LIBS: ignoring invalid library name(s): {0}�, )r# �_LIBCARE_LIB_NAME_RE�match�appendr �logwarn�format�join)r Zvalid�invalid�tokenr r r �libcare_libs_list0 s r- c C s | � � S �N�r ��vr r r �<lambda>@ r r2 c C s | � � S r. r/ r0 r r r r2 D r c C s | � d�S �N�/��rstripr0 r r r r2 S r c C s | pd S r. r r0 r r r r2 \ r c C s | � d�S r3 r5 r0 r r r r2 ^ r c C s | � d�S r3 r5 r0 r r r r2 _ r c C s | � d�S r3 r5 r0 r r r r2 c r c C s | � d�S r3 r5 r0 r r r r2 d r )4�AFTER_UPDATE_COMMAND�AUTO_STICKY_PATCHSET� AUTO_UPDATE�AUTO_UPDATE_DELAY�BEFORE_UPDATE_COMMAND�CHECK_SSL_CERTS� DOCTOR_REPORT_MAX_SIZE_BYTESr r � FORCE_GID� FORCE_IPV4� FORCE_IPV6r � HTTP_TIMEOUT�HTTP_UPLOAD_TIMEOUT�IGNORE_FEATURE_FLAGS�IGNORE_UNKNOWN_KERNELr �KCORE_OUTPUT_SIZE� KDUMPS_DIRr �$KERNEL_ANOMALY_REPORT_MAX_SIZE_BYTESr �LIBCARE_DISABLED� LIBCARE_LIBS�!LIBCARE_PIDLOGS_MAX_TOTAL_SIZE_MB�LIBCARE_SOCKET_TIMEOUT�LIB_AUTO_UPDATE� PATCH_LEVEL� PATCH_METHOD� PATCH_SERVER�PATCH_SERVER_IPV6� PATCH_TYPE�PREFIX�PREV_PATCH_TYPE�REGISTRATION_URL�REGISTRATION_URL_IPV6� PRINT_LEVEL� REPORT_FQDN�SILENCE_ERRORS�STATUS_CHANGE_GAP� STICKY_PATCH�STICKY_PATCHSET� UPDATE_DELAY� UPDATE_POLICY�UPDATE_SYSCTL_CONFIG�USERSPACE_PATCHESr �KERNEL_VERSION_FILE�KCARE_UNAME_FILE�SUCCESS_TIMEOUTr c K s� t t�}|�� }|�� | �� D ]�\}}d}|d }|d }tt|��D ]P}|| �|�sh|| �|�rH|d u rx||= n|d t|� d ||< d} q�qH|s |� |d t|� d � q t � td� |�� d S )NF�=� z = � T� ) �open�CONFIG� readlines�close�items�range�len� startswith�strr' r � atomic_writer* ) �kwargsZcf�linesZpropr �updatedZprop_eqZprop_sp�ir r r � update_configw s" ru c C s� i }t �d�}| D ]<}|�|�}|r:|�� \}}|sFd }n td| ��|||< qt|�tt� }|rztdd�t|�� ��|� � D ]R\}}t| } |d u s�| d u r�q�z | |� W q� t y� td||f ��Y q�0 q�t f i |�� d S )Nz^([^=]+)=([^=]*)$z8Invalid parameter format: %s. Format should be KEY=VALUEzUnknown parameter: %sr$ zBad value for %s: %s) �re�compiler& �groups� SystemExit�set�POSSIBLE_CONFIG_OPTIONSr* �sortedrk � Exceptionru ) �paramsZparams_for_update�pattern�paramr& �keyr Zunknown_params�var_name�convertr r r �update_config_from_args� s, r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� FakeSecHeadc C s || _ d| _d S )Nz [asection] )�fp�sechead)�selfr� r r r �__init__� s zFakeSecHead.__init__c C s. | j r z| j W d | _ S d | _ 0 n | j�� S d S r. )r� r� �readline)r� r r r r� � s � zFakeSecHead.readlinec c s* | j r| j V d | _ | jD ] }|V qd S r. )r� r� )r� �liner r r �__iter__� s zFakeSecHead.__iter__N)�__name__� __module__� __qualname__r� r� r� r r r r r� � s r� c s� i �t ddd�d�� z,ttt��} tjr4� �| � n � �| � W n tyV i Y S 0 d � �fdd� }dD ],\}}t � |�sl� � d|�}|rl|t j |< qlt�� D ]\}}|||d� q��S ) Nrf )� HTTP_PROXY� HTTPS_PROXY)�defaultsc sH z� � d| �}W n ty& |}Y n0 |d urD|r<||�}|�| < d S )N�asection)�getr} )�name�defaultr� r �Zcp�resultr r �read_var� s z%get_config_settings.<locals>.read_var))�httpr� )�httpsr� r� )r� )NN)r r� rg rh r �PY2�readfp� read_filer} r �get_proxy_from_envr� �os�environr{ rk )r r� �scheme�variable�proxyr� r� r r� r �get_config_settings� s$ r� c C s( t �� t� } tj�| � t �| � d S r. )�_CONFIG_OPTIONS�clearr� r �__dict__�update)Zsettingsr r r � set_settings_from_config_file� s r� c C sz i }| � � D ]h\}}|�� }|�d�s(q |�dd��dd�}ztt|��||< W q tyr tj� d||� Y q 0 q |S )ak Checking headers for feature flags which start with 'KC-Flag-' and reformat it to dictionary with keys in upper case and without 'KC-Flag-' prefix and dashes replaced with underscores. For unification all header keys are checked in upper case. For example: 'KC-Flag-Some-Value' -> 'SOME_VALUE' :return: dict {'SOME_VALUE': bool, ...} zKC-FLAG-rf �-�_z(Invalid feature flag header value %s: %s) rk r rn �replace�bool�int� ValueErrorr �kcarelog�error)�headers�flagsZhdr_nameZ hdr_valueZ upper_name� param_namer r r � convert_headers_to_feature_flags� s r� c C s t | � tjst� d S r. )�save_feature_flags_cacher rC � set_feature_flags_from_cache)r� r r r � set_feature_flags_from_headers s r� )�loggerc C s" t | �}tjtjt�|�d� d S )N)�content)r� r rp r �FEATURE_FLAGS_CACHE�json�dumps)r� � feature_flagsr r r r� s r� c C s� t j�tj�sdS ttj��} t�| �}W d � n1 s<0 Y |�� D ]6\}}|t vr`qN|t v rjqN|t j |< t j�d||� qNdS )z� Set global variables using feature flag from cached values received with patchserver headers. Checks that option is allowed by whitelist and update global variable using globals() :return: None Nz$feature flags config override: %s=%s)r� �path�existsr r� rg r� �loadrk �FEATURE_FLAGS_WHITELISTr� r r� r r� �info)�fr� r� r r r r r� s ( r� ))r� r� rv rf r r r r r �py23r �Dict�List�Optional�Setrh r� rz r� r r# rw r% r- r� ro r r r{ ru r� �objectr� r� r� r� r� � catch_errorsr( r� r� r r r r �<module> s� �9 %
cải xoăn