File: //usr/lib/python3/dist-packages/cloudinit/config/__pycache__/cc_wireguard.cpython-38.pyc
U
Ӈg� � @ s U d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
dd ge
d
gd �Ze ed < e�e�Zed
ddg�ZdZdZdZed�dd�Zed�dd�Zeed�dd�Ze d�d d �Z e d�d d �Z ed �d!d"�Z d#d$� Z e!e ee dd%�d&d'�Z"dS )(Z Wireguard� N)�subp�util)�Cloud)�Config)�
MetaSchema)� PER_INSTANCEZ cc_wireguardZubuntu� wireguard)�idZdistrosZ frequencyZactivate_by_schema_keys�meta�name� config_path�contenti� �
)� � )�wg_intc C s� g }t �t| �� ��}|r8d�t|��}|�d|� �� t| �� �D ]@\}}|dksd|dksd|dkrDt|t �sD|�d|� d|� �� qD|r�t
dt � t �|�� ���d S )
aR Validate user-provided wg:interfaces option values.
This function supplements flexible jsonschema validation with specific
value checks to aid in triage of invalid user-provided configuration.
@param wg_int: Dict of configuration value under 'wg:interfaces'.
@raises: ValueError describing invalid values provided.
z, z%Missing required wg:interfaces keys: r r r
z$Expected a string for wg:interfaces:�. Found z*Invalid wireguard interface configuration:N) �REQUIRED_WG_INT_KEYS�
difference�set�keys�join�sorted�append�items�
isinstance�str�
ValueError�NL)r �errorsZmissingr �key�value� r" �?/usr/lib/python3/dist-packages/cloudinit/config/cc_wireguard.py� supplemental_schema_validation s
��r$ c
C s� t �d| d � z,t �d| d � tj| d | d td� W nD tk
r� } z&td| d � dt� t|�� ��|�W 5 d }~X Y nX d S )
z�Writing user-provided configuration into Wireguard
interface configuration file.
@param wg_int: Dict of configuration value under 'wg:interfaces'.
@raises: RuntimeError for issues writing of configuration file.
z"Configuring Wireguard interface %sr z#Writing wireguard config to file %sr r
)�modez-Failure writing Wireguard configuration file �:N) �LOG�debugr Z
write_file�WG_CONFIG_FILE_MODE� Exception� RuntimeErrorr r )r �er" r" r# � write_config; s �
��r- )r �cloudc
C s� zTt �d| d � |j�dd| d � �� t �d| d � |j�dd| d � �� W n< tjk
r� } z tdt� t|�� ��|�W 5 d}~X Y nX dS ) z�Enable and start Wireguard interface
@param wg_int: Dict of configuration value under 'wg:interfaces'.
@raises: RuntimeError for issues enabling WG interface.
z Enabling wg-quick@%s at bootr �enablez wg-quick@z!Bringing up interface wg-quick@%sZrestartz0Failed enabling/starting Wireguard interface(s):N) r' r( �distroZmanage_servicer �ProcessExecutionErrorr+ r r )r r. r, r" r" r# � enable_wgP s ��r2 )�wg_readinessprobesc C sZ g }d}| D ],}t |t�s |�d|� d|� �� |d7 }q |rVtdt� t�|�� ���dS )z�Basic validation of user-provided probes
@param wg_readinessprobes: List of readinessprobe probe(s).
@raises: ValueError of wrong datatype provided for probes.
r z(Expected a string for readinessprobe at r � z Invalid readinessProbe commands:N)r r r r r r )r3 r �pos�cr" r" r# �!readinessprobe_command_validationb s
�
�r7 c C s� g }| D ]b}z$t �dt|�� tj|ddd� W q tjk
rh } z|�|� d|� �� W 5 d}~X Y qX q|r�tdt� t�|�� ���dS )z�Execute provided readiness probe(s)
@param wg_readinessprobes: List of readinessprobe probe(s).
@raises: ProcessExecutionError for issues during execution of probes.
z Running readinessprobe: '%s'T�Zcapture�shellz: Nz&Failed running readinessprobe command:) r' r( r r r1 r r+ r r )r3 r r6 r, r" r" r# �readinessprobex s (�r: )r. c C s� dg}t �d�rdS t�� tk r*|�d� z| j�� W n" tk
rZ t� t
d� � Y nX z| j� |� W n" tk
r� t� t
d� � Y nX dS )z�Install wireguard packages and tools
@param cloud: Cloud object
@raises: Exception for issues during package
installation.
zwireguard-toolsZwgNr zPackage update failedz!Failed to install wireguard-tools) r Zwhichr Zkernel_version�MIN_KERNEL_VERSIONr r0 Zupdate_package_sourcesr* �logexcr' Zinstall_packages)r. Zpackagesr" r" r# � maybe_install_wireguard_packages� s
r= c
C s� z@t j dddd�} t�d| j�� �s>t�d� t j dddd� W n@ t jk
r� } z t� tdt
� t |�� �� � W 5 d}~X Y nX dS ) zYLoad wireguard kernel module
@raises: ProcessExecutionError for issues modprobe
ZlsmodTr8 r z Loading wireguard kernel modulezmodprobe wireguardz Could not load wireguard module:N) r �re�search�stdout�stripr' r( r1 r r< r r )�outr, r" r" r# � load_wireguard_kernel_module� s
rC )r �cfgr. �args�returnc C s� d }d|kr t �d� |d }nt �d| � d S t|� t� |d D ] }t|� t|� t||� qFd|kr�|d d k r�|d }t|� t|� n
t �d� d S )Nr z!Found Wireguard section in configz<Skipping module named %s, no 'wireguard' configuration foundZ
interfacesr: z+Skipping readinessprobe - no checks defined) r' r( r= rC r$ r- r2 r7 r: )r rD r. rE Z
wg_sectionr r3 r" r" r# �handle� s.
� �
�
rG )#�__doc__Zloggingr> Z cloudinitr r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar Zcloudinit.settingsr r
�__annotations__Z getLogger�__name__r' � frozensetr r) r r; �dictr$ r- r2 �listr7 r: r= rC r rG r" r" r" r# �<module> s2 �