0x1949 Team - FAZEMRX - MANAGER
Edit File: cc_wireguard.cpython-310.pyc
o ���f� � @ s U d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddl m Z ddl m Z dd ge d gd �Ze ed < e�e�Zeg d ��ZdZdZdZdefdd�Zdefdd�Zdedefdd�Zde fdd�Z de fd d �Z defd d �Z d d!� Z d"e!d#e ded$e d%df d&d'�Z"dS )(� Wireguard� N)�subp�util)�Cloud)�Config)� MetaSchema)� PER_INSTANCE� cc_wireguard�ubuntu� wireguard)�id�distros� frequency�activate_by_schema_keys�meta)�name� config_path�contenti� � )� � �wg_intc C s� g }t �t| �� ��}|r d�t|��}|�d|� �� t| �� �D ] \}}|dks2|dks2|dkrBt|t �sB|�d|� d|� �� q"|rQt 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 �errors�missingr �key�value� r) �?/usr/lib/python3/dist-packages/cloudinit/config/cc_wireguard.py� supplemental_schema_validation s" ����r+ c C sz t �d| d � zt �d| d � tj| d | d td� W d S ty< } ztd| d � dt� t|�� ��|�d }~ww ) 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 � write_file�WG_CONFIG_FILE_MODE� Exception� RuntimeErrorr$ r" )r �er) r) r* � write_config; s( ��������r5 �cloudc C s� z+t �d| d � |j�dd| d � �� t �d| d � |j�dd| d � �� W dS tjyD } z tdt� t|�� ��|�d}~ww ) 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@%s�restartz0Failed enabling/starting Wireguard interface(s):N) r. r/ �distro�manage_servicer �ProcessExecutionErrorr3 r$ r" )r r6 r4 r) r) r* � enable_wgP s ����r< �wg_readinessprobesc C sZ g }d}| D ]}t |t�s |�d|� d|� �� |d7 }q|r+tdt� 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 )r= r% �pos�cr) r) r* �!readinessprobe_command_validationb s ����rA c C s� g }| D ]1}zt �dt|�� tj|ddd� W q tjy5 } z|�|� d|� �� W Y d}~qd}~ww |rDtdt� 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��capture�shellz: Nz&Failed running readinessprobe command:) r. r/ r" r r; r r3 r$ r )r= r% r@ r4 r) r) r* �readinessprobex s ����rE c C s� dg}t �d�r dS t�� tk r|�d� z| j�� W n ty* t� t d� � w z | j� |� W dS tyB t� t d� � w )z�Install wireguard packages and tools @param cloud: Cloud object @raises: Exception for issues during package installation. zwireguard-tools�wgNr zPackage update failedz!Failed to install wireguard-tools) r �whichr �kernel_version�MIN_KERNEL_VERSIONr r9 �update_package_sourcesr2 �logexcr. �install_packages)r6 �packagesr) r) r* � maybe_install_wireguard_packages� s" � �rN c C s� z$t j dddd�} t�d| j�� �s"t�d� t j dddd� W dS W dS t jy? } zt� tdt � t |�� �� � d}~ww ) zYLoad wireguard kernel module @raises: ProcessExecutionError for issues modprobe �lsmodTrB r z Loading wireguard kernel modulezmodprobe wireguardz Could not load wireguard module:N) r �re�search�stdout�stripr. r/ r; r rK r$ r" )�outr4 r) r) r* � load_wireguard_kernel_module� s ���rU r �cfg�args�returnc C s� d }d|v rt �d� |d }nt �d| � d S t|� t� |d D ]}t|� t|� t||� q#d|v rK|d d urK|d }t|� t|� d S t �d� d S )Nr z!Found Wireguard section in configz<Skipping module named %s, no 'wireguard' configuration found� interfacesrE z+Skipping readinessprobe - no checks defined) r. r/ rN rU r+ r5 r< rA rE )r rV r6 rW � wg_sectionr r= r) r) r* �handle� s* � r[ )#�__doc__�loggingrP � cloudinitr r �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar �cloudinit.settingsr r �__annotations__� getLogger�__name__r. � frozensetr r1 r$ rI �dictr+ r5 r< �listrA rE rN rU r" r[ r) r) r) r* �<module> s4 � "