CasperSecurity
o
t��h�5 �
@ s U d Z ddlZddlZddlmZmZ ddlZddlmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ e�e�Zd Zd
d ged gd
�Zeed< dd� Zdd� Zdede
de deddf
dd�Z dd� Z d d � Z d d � Z !d(d"d#�Z d$e!d%edee fd&d'�Z"dS ))z>LXD: configure lxd with ``lxd init`` and optionally lxd-bridge� N)�List�Tuple)�subp�util)�Cloud)�Config)�
MetaSchema)� PER_INSTANCE�lxdbr0�cc_lxd�ubuntu�lxd)�id�distros� frequency�activate_by_schema_keys�metac C s� g }t | t�s|�dt| �j� d�� t |t�s$|�dt|�j� d�� t |t�s5|�dt|�j� d�� |rV| s;|rV| r@dgng }||rHdgng 7 }|�dd�|�� �� |r_td �|���d
S ) z�Validate user-provided lxd network and bridge config option values.
@raises: ValueError describing invalid values provided.
z/lxd.init config must be a dictionary. found a '�'z1lxd.bridge config must be a dictionary. found a 'z.lxd.preseed config must be a string. found a 'zlxd.initz
lxd.bridgezMUnable to configure LXD. lxd.preseed config can not be provided with key(s): z, z. N)�
isinstance�dict�append�type�__name__�str�join�
ValueError)�init_cfg�
bridge_cfg� preseed_str�errors� incompat_cfg� r! �9/usr/lib/python3/dist-packages/cloudinit/config/cc_lxd.py� supplemental_schema_validation! s<
��
��
�� ���r# c C s� d}g d�}|D ]}| � |�r!|�d|�dd�t| | �f g� qzt�|� W d S tjyT | d dkr9� t�d� t�g d �� d
d � |D �}t�|� Y d S w )
z�Set up lxd if init config is given.
:param init_cfg: dict of lxd init configuration options to provide
to `lxd init --auto`
)�network_address� network_port�storage_backend�storage_create_device�storage_create_loop� storage_pool�trust_password)r
�initz--autoz--%s=%s�_�-r& �lvmzqCloud-init doesn't use thinpool by default on Ubuntu due to LP #1982780. This behavior will change in the future.)�lxc�storage�create�defaultr. zlvm.use_thinpool=falsec S s g | ] }|� d �s|�qS )z--storage-backend)�
startswith)�.0�xr! r! r" �
<listcomp>q s z#handle_init_cfg.<locals>.<listcomp>N)�get�extend�replacer r �ProcessExecutionError�LOG�info)r � init_keys�cmd�k�new_cmdr! r! r" �handle_init_cfgD s(
$� ���rA �name�cfg�cloud�args�returnc
C s� |� d�}|st�d| � d S t|t�s tdt|�j� d���|� di �}|� dd�}|� di �}t|||� t �
d�sZz t � g d �� W n t j yY } zt d
| �|�d }~ww t
||�} t| �r�z|j�| � W n t j y� }
zt�d | |
� W Y d }
~
d S d }
~
ww t � g d �� |r�t j g d
�|d� d S |r�t|� d} |�r6|� dt�} tj�d�r�t �
| �r�t|�}
z t�d| � d�dd� |
�� D ��d }t j dg|d� W n ty� t� td| � Y nw t� d� t�d� t � g d�� d S t |�\}}t | t |�t |�t |�d� |�r"t�dd�|�� t!|� |�r4t�d d�|�� t!|� d S d S |�r?t d | ��d S ) Nr
z8Skipping module named %s, not present or disabled by cfgz*lxd config must be a dictionary. found a 'r r+ �preseed� �bridge)�snap�installr
z#Failed to install lxd from snap: %sz!failed to install packages %s: %s)r
� waitreadyz
--timeout=300)r
r+ z --preseed)�datazdebconf-communicaterB z/etc/default/lxd-bridgezSetting lxd debconf via %s�
c S s g | ]
\}}d ||f �qS )z set %s %sr! )r4 r? �vr! r! r" r6 � s zhandle.<locals>.<listcomp>z Failed to run '%s' for lxd withz Running dpkg-reconfigure for lxd)zdpkg-reconfigurer
z--frontend=noninteractive)�net_name�did_initr1 �attachzCreating lxd bridge: %s� z!Setting up default lxd bridge: %sz*Unable to configure lxd bridge without %s.)"r7 r; �debugr r r r r r# r �whichr: � RuntimeError�get_required_packages�len�distro�install_packages�warningrA �_DEFAULT_NETWORK_NAME�os�path�exists�bridge_to_debconfr �items� Exceptionr �logexc�del_file�
bridge_to_cmd�maybe_cleanup_default�bool�_lxc)rB rC rD rE �lxd_cfgr r r �e�packages�exc�
dconf_commrP �debconfrM �
cmd_create�
cmd_attachr! r! r" �handlex s�
�
��
����
�� � ��� ��
�
� ���rq c C sX i }| � d�dkrd|d< d|d< |S | � d�dkr+d|d< d|d < | � d
�|d< |S | � d�d kr�d|d< | � d
�rB| � d
�|d< | � d �rvd|d
< | � d �|d< | � d�|d< | � d�|d< | � d�|d< | � d�|d< | � dd�|d< | � d�r�d|d< | � d�|d< | � d �|d < | � d d�|d < | � d �r�| � d �|d!< |S td"| � d� ��)#N�mode�none�falsezlxd/setup-bridgerH zlxd/bridge-name�existing�truezlxd/use-existing-bridgerB �new� ipv4_addresszlxd/bridge-ipv4zlxd/bridge-ipv4-address� ipv4_netmaskzlxd/bridge-ipv4-netmask�ipv4_dhcp_firstzlxd/bridge-ipv4-dhcp-first�ipv4_dhcp_lastzlxd/bridge-ipv4-dhcp-last�ipv4_dhcp_leaseszlxd/bridge-ipv4-dhcp-leases�ipv4_natzlxd/bridge-ipv4-nat� ipv6_addresszlxd/bridge-ipv6zlxd/bridge-ipv6-address� ipv6_netmaskzlxd/bridge-ipv6-netmask�ipv6_natzlxd/bridge-ipv6-nat�domainzlxd/bridge-domain�invalid bridge mode "%s")r7 rV )r rn r! r! r" r` � sN )�$�
���
�
�r` c C s� | � d�dkr dS | � dt�}g }dd|ddg}| � d�d kr#d |fS | � d�d
kr3td | � d� ��dd |g}| � d
�r|| � d�r||�d| � d
�| � d�f � | � dd�dkr^|�d� | � d�r{| � d�r{d| � d�| � d�f }|�d| � n|�d� | � d�r�| � d�r�|�d| � d�| � d�f � | � d�dkr�|�d � n|�d � | � d �r�|�d | � d � � | � d d!�}|d!kr�|�d"|� �� ||fS )#Nrr rs )NNrB �networkzattach-profiler2 �eth0ru rw r� r1 rx ry zipv4.address=%s/%sr} rv z
ipv4.nat=truerz r{ z%s-%szipv4.dhcp.ranges=%szipv4.address=noner~ r zipv6.address=%s/%sr� z
ipv6.nat=truezipv6.address=noner� z
dns.domain=%s�mtu���z bridge.mtu=)r7 r\ rV r )r � bridge_namero rp �
dhcp_ranger� r! r! r" re
sX
��
���
��
�
re c C sB dt j�dd�t j�dd�d�}tjdgt| � dg |d � d S )
N�C�HOMEz/root�USER�root)�LC_ALLr� r� r/ z
--force-local)�
update_env)r] �environr7 r �list)r> �envr! r! r" rh I s
�"rh r2 r� c
C s& | t ks|sdS d}d}|rUd}ztdd| |g� t�|| ||� W n" tjyD } z| jdkr2| �t�|| ||� W Y d} ~ nd} ~ ww d}tdd | g� t�|| |� |r�d
}ztd d d
||g� t�||||� W dS tjy� } z| jdkr}| �t�||||� W Y d} ~ dS d} ~ ww dS )a Newer versions of lxc (3.0.1+) create a lxdbr0 network when
'lxd init --auto' is run. Older versions did not.
By removing any that lxd-init created, we simply leave the add/attach
code intact.
https://github.com/canonical/lxd/issues/4649Nz"failed. Assuming it did not exist.z
succeeded.z/Detach of lxd network '%s' from profile '%s' %sr� zdetach-profile� z Deletion of lxd network '%s' %s�deletez+Removal of device '%s' from profile '%s' %s�profile�device�remove)r\ rh r; rT r r: � exit_code)
rP rQ r1 rR r� �nic_name�fail_assume_enoent� succeeded�msgrj r! r! r" rf R s<
��
���rf r r c
C s� g }g }i }d| v r|� | d � |r1d|v r1zt�|�}W n tjttfy0 t�d� Y nw |�dg �D ]}|�d�rE|� |d � q7d|v rTt �
d�sT|� d� d|v rbt �
d�sb|� d � d
|v rpt �
d �sp|� d � |S )
z&identify required packages for installr& �
storage_poolsz|lxd.preseed string value is not YAML. Unable to determine required storage driver packages to support storage_pools config.�driver�zfszzfsutils-linuxr. �lvcreate�lvm2�btrfsz
mkfs.btrfsz btrfs-progs) r �yaml� safe_load� YAMLError� TypeErrorr r; r[ r7 r rU )r r rk �storage_drivers� preseed_cfgr) r! r! r" rW z s0 ��
�
rW )r2 r� )#�__doc__�loggingr] �typingr r r� � cloudinitr r �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar �cloudinit.settingsr � getLoggerr r; r\ r �__annotations__r# rA r r� rq r` re rh rf r rW r! r! r! r" �<module> s4
�# 4d1<
� (