Google
Edit File: config_handlers.cpython-36.pyc
3 !S�i%" � 2 @ sj d dl Z d dlZd dlZddlmZmZmZmZmZ ddl m Z dZ dddd d d d gZ e � Zd d� Zdd� deddd� eedeeeeeeeeedd� eeeeeeedd� ejdd� dd� ejdejdd� dd� eeeeejddejedd� eddeed�1Zdd� Zd d � ZG d d � d e�Zd d!� Zd"d#� Zd$d%� Zd&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_ENABLEc C s | j � dkS )N�1�TRUE�YES�Y)r r r r )�upper)�value� r �5/usr/libexec/kcare/python/kcarectl/config_handlers.py�bool_converter! s r c C s | j � S )N)�strip)�vr r r �<lambda>' s r c C s | j � S )N)r )r r r r r + s c C s | j d�S )N�/)�rstrip)r r r r r 8 s c C s | pd S )Nr )r r r r r @ s c C s | j d�S )Nr )r )r r r r r B s c C s | j d�S )Nr )r )r r r r r C s c C s | j d�S )Nr )r )r r r r r G s c C s | j d�S )Nr )r )r r r r r H s c C s dd� | j d�D �S )Nc S s g | ]}|j � j� �qS r )r �lower)�.0�ptchr r r � <listcomp>R s z <lambda>.<locals>.<listcomp>�,)�split)r r r r r R s )1�AFTER_UPDATE_COMMAND�AUTO_STICKY_PATCHSET� AUTO_UPDATE�AUTO_UPDATE_DELAY�BEFORE_UPDATE_COMMAND�CHECK_SSL_CERTSr � 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_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�}|j� }|j� x�| j� D ]�\}}d}|d }|d }x^tt|��D ]N}|| j|�sl|| j|�rL|d kr|||= n|d t|� d ||< d}P qLW |s"|j |d t|� d � q"W t j tdj |�� d S )NF�=� z = � T� ) �open�CONFIG� readlines�close�items�range�len� startswith�str�appendr � atomic_write�join) �kwargsZcf�linesZpropr �updatedZprop_eqZprop_sp�ir r r � update_config[ s" rb c C s� i }t jd�}xD| D ]<}|j|�}|r<|j� \}}|sHd }n td| ��|||< qW t|�tt� }|r~tddjt|�� ��x`|j � D ]T\}}t| } |d ks�| d kr�q�y | |� W q� t k r� td||f ��Y q�X q�W t f |� d S )Nz^([^=]+)=([^=]*)$z8Invalid parameter format: %s. Format should be KEY=VALUEzUnknown parameter: %sz, zBad value for %s: %s) �re�compile�match�groups� SystemExit�set�POSSIBLE_CONFIG_OPTIONSr] �sortedrV � Exceptionrb ) �paramsZparams_for_update�pattern�paramre �keyr Zunknown_params�var_name�convertr r r �update_config_from_argsp s, rr c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� FakeSecHeadc C s || _ d| _d S )Nz [asection] )�fp�sechead)�selfrt r r r �__init__� s zFakeSecHead.__init__c C s& | j rz| j S d | _ X n | jj� S d S )N)ru rt �readline)rv r r r rx � s zFakeSecHead.readlinec c s. | j r| j V d | _ x| jD ] }|V q W d S )N)ru rt )rv �liner r r �__iter__� s zFakeSecHead.__iter__N)�__name__� __module__� __qualname__rw rx rz r r r r rs � s rs c s� i �t ddd�d�� y,ttt��} tjr4� j| � n � j| � W n tk rT i S X d � �fdd� }x4dD ],\}}t j |�sl� j d |�}|rl|t j |< qlW x tj� D ]\}}|||d � q�W �S )NrQ )� HTTP_PROXY� HTTPS_PROXY)�defaultsc sJ y� j d| �}W n tk r( |}Y nX |d k rF|r>||�}|�| < d S )N�asection)�getrk )�name�defaultrq r )�cp�resultr r �read_var� s z%get_config_settings.<locals>.read_var�httpr~ �httpsr r� )rq )NN�r� r~ �r� r )r� r� )r rs rR rS r �PY2�readfp� read_filerk r �get_proxy_from_envr� �os�environri rV )r r� �scheme�variable�proxyrp rq r )r� r� r �get_config_settings� s$ r� c C s( t j� t� } tjj| � t j| � d S )N)�_CONFIG_OPTIONS�clearr� r �__dict__�update)Zsettingsr r r � set_settings_from_config_file� s r� c C s� i }xv| j � D ]j\}}|j� }|jd�s*q|jdd�jdd�}ytt|��||< W q tk rv tjj d||� Y qX qW |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-rQ �-�_z(Invalid feature flag header value %s: %s) rV r rY �replace�bool�int� ValueErrorr �kcarelog�error)�headers�flagsZhdr_nameZ hdr_value� param_namer r r � convert_headers_to_feature_flags� s r� c C s t | � tjst� d S )N)�save_feature_flags_cacher r/ � set_feature_flags_from_cache)r� r r r � set_feature_flags_from_headers� s r� )�loggerc C s" t | �}tjtjtj|�d� d S )N)�content)r� r r\ r �FEATURE_FLAGS_CACHE�json�dumps)r� � feature_flagsr r r r� � s r� c C sz t jjtj�sdS ttj��} tj| �}W dQ R X xB|j� D ]6\}}|t krNq<|t krXq<|t j |< t jjd||� q<W dS )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� rR r� �loadrV �FEATURE_FLAGS_WHITELISTr� r r� r r� �info)�fr� ro r r r r r� � s r� )!r� r� rc rQ r r r r r �py23r rS r� rh r� r r� rZ r r ri rb rr �objectrs r� r� r� r� � catch_errors�logwarnr� r� r r r r �<module> s� %