File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceLXD.cpython-310.pyc
o
�Ad�D � @ sV d Z ddlZddlZddlZddlZddlmZmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZ ddlZddlmZ ddlmZ ddlmZ dd lmZ dd
lm Z m Z m Z m Z dd l m!Z! e�"e#�Z$d Z%d
Z&dZ'ddddddd�Z(de)fdd�Z* d<dee
e) de e)e f fdd�Z+G dd� de�Z,G dd � d e�Z-G d d � d e�Z.d e)de/fd d!�Z0G d"d#� d#e j1�Z2de3fd$d%�Z4 &d=d'ej5d(e)d)e3fd*d+�Z6 &d=d'ej5d(e)d)e3dej7fd,d-�Z8G d.d/� d/e�Z9G d0d1� d1�Z:e&e9j;fd2e)d3e9de/fd4d5�Z<e2e j=ffgZ>d6d7� Z?e#d8k�r)ddl@Z@d9ZAe@jBeAd:�ZCeC�D� eEe �Fe<e9j;d;��� dS 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://linuxcontainers.org/lxd/docs/master/dev-lxd
� N)�Flag�auto)�JSONDecodeError)�Any�Dict�List�Optional�Tuple�Union�cast)� HTTPAdapter)�HTTPConnection)�HTTPConnectionPool)�log)�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�rIz
t � dg�\}}W n t jy, } zt�d|� | W Y d }~S d }~ww |�� dv rIt�� d d }|dkrAdS |d krGd
S d S | S ) N�eth0zsystemd-detect-virtzHUnable to run systemd-detect-virt: %s. Rendering default network config.)�kvm�qemu�uname� �ppc64le�enp0s5�s390x�enc9�enp5s0)r �which�ProcessExecutionError�LOG�warning�stripr � system_info)� default_name� virt_type�_�err�arch� r2 �A/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceLXD.py� _get_fallback_interface_name/ s(
��� r4 �nicsc C sF t � }|r t�d|� n t� }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� �physical�dhcpr )�type�control)r9 �name�subnets)�version�config)r r) �debugr4 )r5 � primary_nicr2 r2 r3 �generate_network_configI s$ ��
���rA c �$ e Zd Z� fdd�Zdd� Z� ZS )�SocketHTTPConnectionc s t � �d� || _d S �N� localhost)�super�__init__� socket_path��selfrH �� __class__r2 r3 rG p �
z SocketHTTPConnection.__init__c C s$ t � t jt j�| _| j�| j� d S �N)�socket�AF_UNIX� SOCK_STREAM�sock�connectrH �rJ r2 r2 r3 rS t s z SocketHTTPConnection.connect)�__name__�
__module__� __qualname__rG rS �
__classcell__r2 r2 rK r3 rC o � rC c rB )�SocketConnectionPoolc s || _ t� �d� d S rD )rH rF rG rI rK r2 r3 rG z s z SocketConnectionPool.__init__c C s
t | j�S rN )rC rH rT r2 r2 r3 � _new_conn~ s
z SocketConnectionPool._new_conn)rU rV rW rG r[ rX r2 r2 rK r3 rZ y rY rZ c @ s e Zd Zddd�ZdS )�LXDSocketAdapterNc C s t t�S rN )rZ �LXD_SOCKET_PATH)rJ �url�proxiesr2 r2 r3 �get_connection� s z LXDSocketAdapter.get_connectionrN )rU rV rW r` r2 r2 r2 r3 r\ � s r\ �
metadata_typec
C sx t |t�r|S |du r
i S zt�|�}W n ty+ } z t�dj| |d��|�d}~ww |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})�md_type�valuez:Invalid {md_type} format. Expected YAML but found: {value})�
isinstance�dictr � load_yaml�AttributeErrorr �InvalidMetaDataException�format)ra �metadata_value�parsed_metadata�excr2 r2 r3 �_raw_instance_data_to_dict� s.
�������rm c s� e Zd ZU dZejZeee f e
d< ejZ eee f e
d< ej j
d Z
ee df e
d< dZded d
f� fd d � Zd efd
d�Zd efdd�Zd e fdd�Zd e fdd�Zed efdd��Z� ZS )�
DataSourceLXD�LXD�_network_config�_crawled_metadata)�user.meta-datar r r r .�sensitive_metadata_keysT�ci_pkl_versionr Nc s t � �|� d| _d S )NT)rF � _unpickle�skip_hotplug_detect)rJ rt rK r2 r3 ru � rM zDataSourceLXD._unpicklec C s t � S )z@Check platform environment to report if this datasource may run.)�is_platform_viablerT r2 r2 r3 �_is_platform_viable� s z!DataSourceLXD._is_platform_viablec C s� | � � s t�d� dS tjtjdtd�| _td| j�d��| _ | j�di �}|�di �}|r3td|�}t
| j t �sEt� t�
| j �|g�| _ d| jv rP| jd | _d | jv r^td | jd �| _d
| jv ri| jd
| _d S ) z=Crawl LXD socket API instance data and return True on successz+Not an LXD datasource: No LXD socket found.FzCrawl of metadata service)�logfunc�msg�func� meta-datar> rr r r r T)rx r) r? r �log_time�
read_metadatarq rm �get�metadatard re �
mergemanydictrf � userdata_rawrp �vendordata_raw)rJ r> �
user_metadatar2 r2 r3 � _get_data� s<
� � � �
�
zDataSourceLXD._get_datac C s dj ttd�S )z.Return subplatform details for this datasourcez"LXD socket API v. {ver} ({socket}))�verrO )ri �LXD_SOCKET_API_VERSIONr] rT r2 r2 r3 �_get_subplatform� s �z DataSourceLXD._get_subplatformc C sB t tjd�}|�di �}t|t�st�|�}|�d�| j�d�kS )z%Return True if instance_id unchanged.��
metadata_keysr| z instance-id) r~ � MetaDataKeys� META_DATAr rd re r rf r� )rJ �sys_cfg�response�mdr2 r2 r3 �check_instance_id� s
z DataSourceLXD.check_instance_idc C s� | j tjkr?| jtjkr| �� t| jt�r?| j�d�r(t� d� | jd | _ n| j�d�r?dd� | jd �
� D �}t |�| _ | j tjkrNt� 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 )r9 �nicr2 )�.0�k�vr2 r2 r3 �
<listcomp> s
��z0DataSourceLXD.network_config.<locals>.<listcomp>z8LXD datasource generating network config using fallback.)
rp r �UNSETrq r� rd re r r) r? �itemsrA r )rJ r� r2 r2 r3 �network_config� s(
� �
� z DataSourceLXD.network_config)rU rV rW �dsnamer r� rp r
r �str�__annotations__rq �
DataSourcers r rv �intru �boolrx r� r� r� �propertyre r� rX r2 r2 rK r3 rn � s
��
"rn c C s" t j�t�rt�t �t�j�S dS )z=Return True when this platform appears to have an LXD socket.F)�os�path�existsr] �stat�S_ISSOCK�lstat�st_moder2 r2 r2 r3 rw s rw T�sessionr^ �do_raisec
C sf t | ||�}|jst�d||j|j� i S z|�� W S ty2 } z t� dj
||jd��|�d }~ww )N�Skipping %s on [HTTP:%d]:%szFUnable to process LXD config at {url}. Expected JSON but found: {resp})r^ �resp) � _do_request�okr) r? � status_code�text�jsonr r rh ri )r� r^ r� � url_responserl r2 r2 r3 �_get_json_response s, �
�����r� c C s| t ddd�D ]}| �|�}d|jkr!t�d� t�d|j||� q t�d|j|� |r<|js<t �
dj |j||j 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})�code�router� )
�ranger r� �time�sleepr) r* r? r� r rh ri r� )r� r^ r� �retriesr� r2 r2 r3 r� / s*
�
��r� c @ s0 e Zd Ze� Ze� Ze� Ze� ZeeB eB ZdS )r� N) rU rV rW r �NONE�CONFIG�DEVICESr� �ALLr2 r2 r2 r3 r� L s r� c @ sD e Zd Zefdefdd�Zdejdefdd�Z de
defd d
�Z d S ) �_MetaDataReader� api_versionc C s || _ t�t| j �| _d S rN )r� r � combine_url�LXD_URL� _version_url)rJ r� r2 r2 r3 rG U s z_MetaDataReader.__init__r� r c
C s� di i}t �| jd�}t||�}t|�D ]I}t �t|�}t||dd�}|js1t� d||j
|j � q|� d�d }|j |d |< |t
v r]t
| |vrQ|j |t
| < qt�d||�dd 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� z,Ignoring LXD config %s in favor of %s value.�userz
cloud-initr6 )r r� r� r� �sortedr� r� r� r) r? r� r� �
rpartition�CONFIG_KEY_ALIASESr* �replace) rJ r� r> �
config_url�
config_routes� config_route�config_route_url�config_route_response�cfg_keyr2 r2 r3 �_process_configY s>
�� �� ��z _MetaDataReader._process_configr� c C s� t �� �Q}|�| jt� � d| ji}tj|v r&t� | jd�}t
||�j |d< tj |v r3|�
| �|�� tj|v rLt� | jd�}t||dd�}|rL||d< |W d � S 1 sXw Y d S )N�_metadata_api_versionr| r� Fr� )�requests�Session�mountr� r\ r� r� r� r r� r� r� r� �updater� r� r� )rJ r� r� r� �md_router^ r� r2 r2 r3 �__call__� s"
�
$�z_MetaDataReader.__call__N) rU rV rW r� r� rG r� r� re r� r� r� r2 r2 r2 r3 r� T s 7r� r� 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� r� )r� )r� r� r2 r2 r3 r~ � s #�r~ c C s t �| t�S rN )r �list_from_depends� datasources)�dependsr2 r2 r3 �get_datasource_list� s r� �__main__z*Query LXD metadata and emit a JSON object.)� descriptionr� rN )T)G�__doc__r� rO r� r� �enumr r � json.decoderr �typingr r r r r r
r r� �requests.adaptersr �urllib3.connectionr
�urllib3.connectionpoolr � cloudinitr �loggingr r r r �
cloudinit.netr � getLoggerrU r) r] r� r� r� r� r4 rA rC rZ r\ re rm r� rn r� rw r� r� �Responser� r� r� r� r~ �DEP_FILESYSTEMr� r� �argparser� �ArgumentParser�parser�
parse_args�print�
json_dumpsr2 r2 r2 r3 �<module> s�
$
�
�
�
�&
m���
�����
� Q���
�
*�
�