Google
Edit File: DataSourceLXD.cpython-39.pyc
a '�Dg�D � @ s: d Z ddlZddlZddlZddlZddlZddlmZmZ ddl m Z ddl m Z m Z mZmZmZmZmZ ddlZddlmZ ddlmZ ddlmZ dd lmZm Z m Z m Z m Z dd l m!Z! e�"e#�Z$d Z%d Z&d Z'ddddddd�Z(e)d�dd�Z*d9eee) e e)e f d�dd�Z+G dd� de�Z,G dd� de�Z-G d d � d e�Z.e)e/d �d d �Z0G d!d"� d"e j1�Z2e3d�d#d$�Z4d:ej5e)e3d&�d'd(�Z6d;ej5e)e3ej7d)�d*d+�Z8G d,d-� d-e�Z9G d.d/� d/�Z:e&e9j;fe)e9e/d0�d1d2�Z<e2e j=ffgZ>d3d4� Z?e#d5k�r6ddl@Z@d6ZAe@jBeAd7�ZCeC�D� eEe�Fe<e9j;d8��� dS )<a� Datasource for LXD, reads /dev/lxd/sock representation of instance data. Notes: * This datasource replaces previous NoCloud datasource for LXD. * Older LXD images may not have updates for cloud-init so NoCloud may still be detected on those images. * Detect LXD datasource when /dev/lxd/sock is an active socket file. * Info on dev-lxd API: https://documentation.ubuntu.com/lxd/en/latest/dev-lxd/ � N)�Flag�auto)�JSONDecodeError)�Any�Dict�List�Optional�Tuple�Union�cast)� HTTPAdapter)�HTTPConnection)�HTTPConnectionPool)� atomic_helper�sources�subp� url_helper�util)�find_fallback_nicz /dev/lxd/sockz1.0z http://lxd� user-data�network-config� vendor-data)�cloud-init.user-datazcloud-init.network-config�cloud-init.vendor-data�user.user-datazuser.network-config�user.vendor-data��returnc C s� d} t �d�r�zt � dg�\}}W n6 t jyX } z t�d|� | W Y d }~S d }~0 0 |�� dv r�t�� d d }|dkr�dS |d kr�d S d S | S ) NZeth0zsystemd-detect-virtzHUnable to run systemd-detect-virt: %s. Rendering default network config.)ZkvmZqemu�uname� Zppc64leZenp0s5Zs390xZenc9Zenp5s0)r �whichZProcessExecutionError�LOG�warning�stripr Z system_info)Z default_nameZ virt_type�_�errZarch� r& �C/usr/lib/python3.9/site-packages/cloudinit/sources/DataSourceLXD.py� _get_fallback_interface_name/ s$ � r( )�nicsr c C sF t � }|rt�d|� nt� }t�d|� dd|ddd�gd�gd �S ) zCReturn network config V1 dict representing instance network config.zCLXD datasource generating network from discovered active device: %szVLXD datasource generating network from systemd-detect-virt platform default device: %s� ZphysicalZdhcpr )�typeZcontrol)r+ �name�subnets)�version�config)r r! �debugr( )r) Z primary_nicr& r& r' �generate_network_configI s$ �� ���r1 c s$ e Zd Z� fdd�Zdd� Z� ZS )�SocketHTTPConnectionc s t � �d� || _d | _d S �NZ localhost)�super�__init__� socket_path�sock��selfr6 �� __class__r& r' r5 p s z SocketHTTPConnection.__init__c C s$ t � t jt j�| _| j�| j� d S �N)�socketZAF_UNIXZ SOCK_STREAMr7 �connectr6 �r9 r& r& r' r> u s z SocketHTTPConnection.connect)�__name__� __module__� __qualname__r5 r> � __classcell__r& r& r: r' r2 o s r2 c s$ e Zd Z� fdd�Zdd� Z� ZS )�SocketConnectionPoolc s || _ t� �d� d S r3 )r6 r4 r5 r8 r: r& r' r5 { s z SocketConnectionPool.__init__c C s t | j�S r< )r2 r6 r? r& r& r' � _new_conn s z SocketConnectionPool._new_conn)r@ rA rB r5 rE rC r& r& r: r' rD z s rD c @ s e Zd Zddd�Zddd�ZdS )�LXDSocketAdapterNc C s t t�S r< )rD �LXD_SOCKET_PATH)r9 �url�proxiesr& r& r' �get_connection� s z LXDSocketAdapter.get_connectionc C s | � |j|�S r< )rJ rH )r9 ZrequestZverifyrI Zcertr& r& r' � get_connection_with_tls_context� s z0LXDSocketAdapter.get_connection_with_tls_context)N)NN)r@ rA rB rJ rK r& r& r& r' rF � s �rF )� metadata_typer c C s� t |t�r|S |du ri S zt�|�}W n: tyb } z"t�dj| |d��|�W Y d}~n d}~0 0 |du r�t�dj| |d���|S )a6 Convert raw instance data from str, bytes, YAML to dict :param metadata_type: string, one of as: meta-data, vendor-data, user-data network-config :param metadata_value: str, bytes or dict representing or instance-data. :raises: InvalidMetaDataError on invalid instance-data content. NzAInvalid {md_type}. Expected str, bytes or dict but found: {value})Zmd_type�valuez:Invalid {md_type} format. Expected YAML but found: {value})� isinstance�dictr � load_yaml�AttributeErrorr �InvalidMetaDataException�format)rL Zmetadata_valueZparsed_metadata�excr& r& r' �_raw_instance_data_to_dict� s* �����rU c s� e Zd ZU dZejZeee f e d< ejZ e eee f e d< ej jd Zee df e d< dZedd �� fd d � Zeed �d d��Zed �dd�Ze d �dd�Ze d �dd�Zeed �dd��Z� Z S )� DataSourceLXDZLXD�_network_config�_crawled_metadata)�user.meta-datar r r r .�sensitive_metadata_keysTN)�ci_pkl_versionr c s t � �|� d| _d S )NT)r4 � _unpickle�skip_hotplug_detect)r9 r[ r: r& r' r\ � s zDataSourceLXD._unpickler c C s t � S )z@Check platform environment to report if this datasource may run.)�is_platform_viabler& r& r& r' � ds_detect� s zDataSourceLXD.ds_detectc C s� t � | _td| j�d��| _| j�di �}|�di �}|rL| j�td|�� d| jv rb| jd | _d| jv r~td| jd �| _d| jv r�| jd | _dS )z=Crawl LXD socket API instance data and return True on success� meta-datar/ rY r r r T) � read_metadatarX rU �get�metadata�updateZ userdata_rawrW Zvendordata_raw)r9 r/ Z user_metadatar& r& r' � _get_data� s& � � � zDataSourceLXD._get_datac C s dj ttd�S )z.Return subplatform details for this datasourcez"LXD socket API v. {ver} ({socket}))Zverr= )rS �LXD_SOCKET_API_VERSIONrG r? r& r& r' �_get_subplatform� s �z DataSourceLXD._get_subplatformc C sB t tjd�}|�di �}t|t�s,t�|�}|�d�| j�d�kS )z%Return True if instance_id unchanged.�� metadata_keysr` z instance-id) ra � MetaDataKeys� META_DATArb rN rO r rP rc )r9 Zsys_cfg�response�mdr& r& r' �check_instance_id� s z DataSourceLXD.check_instance_idc C s� | j tjkr~| jtjkr | �� t| jt�r~| j�d�rPt� d� | jd | _ n.| j�d�r~dd� | jd � � D �}t |�| _ | j tjkr�t� d� t � | _ t t| j �S )z�Network config read from LXD socket config/user.network-config. If none is present, then we generate fallback configuration. r z,LXD datasource using provided network config�devicesc S s g | ]\}}|d dkr|�qS )r+ Znicr& )�.0�k�vr& r& r' � <listcomp>� s �z0DataSourceLXD.network_config.<locals>.<listcomp>z8LXD datasource generating network config using fallback.) rW r �UNSETrX re rN rO rb r! r0 �itemsr1 r )r9 ro r& r& r' �network_config� s( � � �z DataSourceLXD.network_config) r@ rA rB Zdsnamer rt rW r r �str�__annotations__rX r � DataSourcerZ r r] �intr\ � staticmethod�boolr_ re rg rn �propertyrO rv rC r& r& r: r' rV � s �� rV c C s" t j�t�r t�t �t�j�S dS )z=Return True when this platform appears to have an LXD socket.F)�os�path�existsrG �stat�S_ISSOCK�lstat�st_moder& r& r& r' r^ s r^ T)�sessionrH �do_raisec C s� t | ||�}|js0t�d||j|j�d�� i S z |�� W S ty| } z*t � dj ||j�d�d��|�W Y d }~n d }~0 0 d S )N�Skipping %s on [HTTP:%d]:%s�utf-8zFUnable to process LXD config at {url}. Expected JSON but found: {resp})rH �resp) � _do_request�okr! r0 � status_code�content�decodeZjsonr r rR rS )r� rH r� Z url_responserT r& r& r' �_get_json_response s& � ���r� )r� rH r� r c C s� t ddd�D ]:}| �|�}d|jkrBt�d� t�d|j||� q qHq t�d|j|� |r�|js�t � dj |j||j � d �d ���|S ) N� r ���i� g�������?z,[GET] [HTTP:%d] %s, retrying %d more time(s)z[GET] [HTTP:%d] %sz3Invalid HTTP response [{code}] from {route}: {resp}r� )�codeZrouter� )�rangerb r� �time�sleepr! r" r0 r� r rR rS r� r� )r� rH r� Zretriesrl r& r& r' r� - s* � ��r� c @ s0 e Zd Ze� Ze� Ze� Ze� ZeeB eB ZdS )rj N) r@ rA rB r ZNONE�CONFIG�DEVICESrk �ALLr& r& r& r' rj J s rj c @ s@ e Zd Zefed�dd�Zejed�dd�Z e ed�dd �Z d S ) �_MetaDataReader�� api_versionc C s || _ t�t| j �| _d S r< )r� r � combine_url�LXD_URL� _version_url)r9 r� r& r& r' r5 S s z_MetaDataReader.__init__)r� r c C s� di i}t �| jd�}t||�}t|�D ]�}t �t|�}t||dd�}|j�d�}|j slt � d||j |� q(|� d�d } ||d | < | tv r(t| |vr�||t| < q(t �d| | �d d d �� q(|S ) a� Iterate on LXD API config items. Promoting CONFIG_KEY_ALIASES Any CONFIG_KEY_ALIASES which affect cloud-init behavior are promoted as top-level configuration keys: user-data, network-data, vendor-data. LXD's cloud-init.* config keys override any user.* config keys. Log debug messages if any user.* keys are overridden by the related cloud-init.* key. r/ F�r� r� r� �/r� z,Ignoring LXD config %s in favor of %s value.�userz cloud-initr* )r r� r� r� �sortedr� r� r� r� r� r! r0 r� � rpartition�CONFIG_KEY_ALIASESr" �replace) r9 r� r/ Z config_urlZ config_routesZ config_routeZconfig_route_urlZconfig_route_responseZ response_textZcfg_keyr& r& r' �_process_configW s8 � � �z _MetaDataReader._process_config)ri r c C s� t �� ��}|�| jt� � d| ji}tj|v rRt� | jd�}t ||�j � d�|d< tj |v rl|�| �|�� tj|v r�t� | jd�}t||dd�}|r�||d< |W d � S 1 s�0 Y d S )NZ_metadata_api_versionr` r� ro Fr� )�requests�SessionZmountr� rF r� rj rk r r� r� r� r� r� rd r� r� r� )r9 ri r� rm Zmd_routerH ro r& r& r' �__call__� s( ��� z_MetaDataReader.__call__N) r@ rA rB rf rw r5 r� r� rO r� rj r� r& r& r& r' r� R s 6r� )r� ri r c C s t | d�|d�S )a8 Fetch metadata from the /dev/lxd/socket routes. Perform a number of HTTP GETs on known routes on the devlxd socket API. Minimally all containers must respond to <LXD_SOCKET_API_VERSION>/meta-data when the LXD configuration setting `security.devlxd` is true. When `security.devlxd` is false, no /dev/lxd/socket file exists. This datasource will return False from `is_platform_viable` in that case. Perform a GET of <LXD_SOCKET_API_VERSION>/config` and walk all `user.*` configuration keys, storing all keys and values under a dict key LXD_SOCKET_API_VERSION: config {...}. In the presence of the following optional user config keys, create top level aliases: - user.user-data -> user-data - user.vendor-data -> vendor-data - user.network-config -> network-config :param api_version: LXD API version to operated with. :param metadata_keys: Instance of `MetaDataKeys` indicating what keys to fetch. :return: A dict with the following optional keys: meta-data, user-data, vendor-data, network-config, network_mode, devices. Below <LXD_SOCKET_API_VERSION> is a dict representation of all raw configuration keys and values provided to the container surfaced by the socket under the /1.0/config/ route. r� rh )r� )r� ri r& r& r' ra � s #�ra c C s t �| t�S r< )r Zlist_from_depends� datasources)Zdependsr& r& r' �get_datasource_list� s r� �__main__z*Query LXD metadata and emit a JSON object.)� descriptionrh )N)T)T)G�__doc__Zloggingr~ r= r� r� �enumr r Z json.decoderr �typingr r r r r r r r� Zrequests.adaptersr Zurllib3.connectionr Zurllib3.connectionpoolr Z cloudinitr r r r r Z cloudinit.netr Z getLoggerr@ r! rG rf r� r� rw r( r1 r2 rD rF rO rU ry rV r| r^ r� r� ZResponser� rj r� r� ra ZDEP_FILESYSTEMr� r� �argparser� �ArgumentParser�parser� parse_args�printZ json_dumpsr& r& r& r' �<module> s� $ � � � & c �� �� R�� * � �