CasperSecurity
o
t��h�> � @ s: d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z d dl
m Z e� e
�ZG dd� dej�Zddd�Zdd d
�Zdd d �Zd
d� ZG dd� de�ZeejffeejejffgZdd� Ze
dkr�d dlmZ ej ej d� ed Z e � e �\Z Z!Z"Z#e$de � �� e$de!� �� e$de"� �� e$de#� �� dS dS ) � N)�partial)�dmi� lifecycle�sources�util)�enic sv e Zd ZdZdd� Z� fdd�Zdd� Zdd � Zed
d � �Z d e
fd
d�Z dd� Z dd� Z
dd� Zedd� �Z� ZS )�DataSourceNoCloud�NoCloudc C sV t j�| |||� d | _tj�|jd�tj�|jd�g| _d | _d| _ d | _
d | _ d S )N�nocloud� nocloud-net)�/�file://) r �
DataSource�__init__�seed�os�path�join�seed_dir� seed_dirs�supported_seed_starts�_network_config� _network_eni��self�sys_cfg�distro�paths� r �E/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceNoCloud.pyr s �
zDataSourceNoCloud.__init__c sF t � �� d | jrd| j� d�nd | jtjkr d| j� d� S d S )z@append seed and dsmode info when they contain non-default values� z[seed=�]� z[dsmode=)�super�__str__r �dsmoder �DSMODE_NETWORK�r �� __class__r r r$ & s �� �����zDataSourceNoCloud.__str__c C s| t �d�}|�t �d�� t �d|�� �}|�t �d|�� �� |�t �d| �� tt|�t|�@ �}|jdd� |S )Nz TYPE=vfatz TYPE=iso9660zLABEL=%szLABEL_FATBOOT=%sT)�reverse)r �find_devs_with�extend�upper�lower�list�set�sort)r �label�fslist�
label_list�devlistr r r � _get_devices3 s
z DataSourceNoCloud._get_devicesc C s� d| j d�}g }i ddd d�}z i }t�d�}|r*t||�r*|�d� t|d|i�}W n ty; t�t d� Y d S w zi }t|�rO|�d
� t|d|i�}W n ty` t�t d � Y d S w d dgd
dgd�}| j
D ]'}z tj |fi |��}|�|� t � d|� t||�}W n t
y� Y qmw | j�d�r�|�d� | jd |d d< d | jv r�d| jv r�t|| j�}|�d� dd� } | j�dd�}
|
d u�r6| �|
�D ]b} z4t � d| � z t�| | |�}W n t
y� t �d| |
� Y W q�w t||�}t � d| � |�| � W n, t�y$ } z| jtjk�r� W Y d } ~ q�d } ~ w tj�y5 t�t d| � Y q�w |�s;d S d|d v �r�|d d }
d }| jD ]
}|
�|��rY|} n�qM|�se| �|
� d S t�|
�}
tj|
d d �\}}}}t � d |
� t� |d |g�|d< ||d < ||d
< |�|
� t� |d |g�|d< | � |d �d �g�| _ | j t j k�r�t � d | | j � d S d � |�| _!|d | _"|d | _#|d
| _$|d | _%|d �d!�| _&d"S )#Nr
)� instance-idr% r" )� meta-data� user-data� vendor-data�network-config�system-serial-numberr r8 zUnable to parse dmi dataF�cmdlinez!Unable to parse command line datar9 r: r; )�required�optionalzUsing seeded data from %s�seedfrom�ds_config_seedfrom� ds_configc S s t j| fi |��S �N)r �pathprefix2dict)�mp�datar r r �_pp2d_callback� s z3DataSourceNoCloud._get_data.<locals>._pp2d_callback�fs_label�cidataz Attempting to use data from %sz)device %s with label=%s not a valid seed.zUsing data from %sz(Failed to mount %s when looking for data)�timeoutz Using seeded cache data from %sr% z&%s: not claiming datasource, dsmode=%s�,znetwork-interfacesT)'r% r �
read_dmi_data�load_cmdline_data�append�_merge_new_seed� Exceptionr �logexc�LOGr rD �debug�
ValueError�ds_cfg�getr6 �mount_cb�warning�OSError�errno�ENOENT�MountFailedErrorr �
startswith�_log_unusable_seedfrom� sub_dmi_vars� read_seeded�
mergemanydict�_determine_dsmoder �DSMODE_DISABLEDr r �metadata� userdata_raw�vendordata_rawr r )r �defaults�found�mydata�md�serial� pp2d_kwargsr �seededrG r2 �dev�er@ � seedfound�proto�md_seed�ud�vd�_r r r � _get_data? s� ��
� �
� ��
�
� ��
����
�
�
���
zDataSourceNoCloud._get_datac C s | j s t�� r dnd| _ | j S )N�lxdr
)�_platform_typer �is_lxdr' r r r �
platform_type� s z DataSourceNoCloud.platform_typer@ c C � t �d| | j|� dS )�!Stage-specific level and message.zN%s only uses seeds starting with %s - will try to use %s in the network stage.N)rR �infor �r r@ r r r r^ � s �z(DataSourceNoCloud._log_unusable_seedfromc C s t jS )z=Return unknown when 'cloud-name' key is absent from metadata.)r �METADATA_UNKNOWNr' r r r �_get_cloud_name� s z!DataSourceNoCloud._get_cloud_namec C s$ | j �d�r d}nd}d|| j f S )z/Return the subplatform metadata source details.z/devz config-diskzseed-dirz%s (%s))r r] )r �subplatform_typer r r �_get_subplatform� s z"DataSourceNoCloud._get_subplatformc C s: | � � }|sd S t| d| jg�}t|d�}|sd S ||kS )Nr )�dirs)�get_instance_id�getattrr �_quick_read_instance_id)r r �currentr� �quick_idr r r �check_instance_id� s
z#DataSourceNoCloud.check_instance_idc C s8 | j d u r| jd urtjdddd� t�| j�| _ | j S )Nz$Eni network configuration in NoCloudz24.3z,You can use network v1 or network v2 instead��
deprecated�deprecated_version�
extra_message)r r r � deprecater �convert_eni_datar' r r r �network_config s
�z DataSourceNoCloud.network_config)�__name__�
__module__� __qualname__�dsnamer r$ r6 rv �propertyrz �strr^ r� r� r� r� �
__classcell__r r r( r r s
r c C s� | d u rg } d}i }t |�r||v r|| S | D ].}|d u r qz tj|dgd�}t�|d �}|r<||v r<|| W S W q tyF Y qw d S )Nr7 r8 )r> )rM r rD � load_yamlrT )r� �iid_key�fill�drF rj r r r r� s& � �r� c C s� dt jfdt jfg}|D ]9\}}t|| |�sq d| v r dS | �d�}|r?|�d�r2t j| d< dS |�d�r<t j| d< dS || d< dS dS ) Nz
ds=nocloudzds=nocloud-netr% Tr@ �zhttp://zhttps://zftp://zftps://)r
r F)r � DSMODE_LOCALr& �parse_cmdline_datarV r] )r� r= �pairs�idstrr% r@ r r r rM . s* �
�
�
�rM c C s� |d u rt �� }d| }d| |v sd| |v sdS d}|�� D ]
}|�| �r-|�dd�}q |d �d�}t|�dkrB|dd � }nd}d d
d d �}|D ]+}|dkrSqLz
|�dd�\} }
W n
tyj |} d }
Y nw | |v rs|| } |
|| < qLd
S )Nz %s z %s;Fr" �=� �;r zlocal-hostnamer7 r@ )�h�i�sT)r � get_cmdline�splitr] �lenrP ) �ds_idr� r= �argline�tok�tmp�kvpairs�s2l�item�k�vr r r r� O s8
� �
r� c C s� | � � }|�di �}t|d t�st�|d �}t�| d |g�|d< |�d�r2t�|�d��|d< d|v r<|d |d< d|v rF|d |d< |S )Nr8 r; r9 r: )�copyrV �
isinstance�dictr r� ra )�currm �ret�newmdr r r rO w s
rO c @ s* e Zd Zdd� Zdefdd�Zdd� ZdS ) �DataSourceNoCloudNetc C s t �| |||� d| _d S )Nr� )r r r r r r r r � s
z DataSourceNoCloudNet.__init__r@ c C r{ )r| z6%s only uses seeds starting with %s - %s is not valid.N)rR rX r r~ r r r r^ � s �z+DataSourceNoCloudNet._log_unusable_seedfromc C s� t tjdddd�}dt�� kr|� dS t�t�d�p d��� }|| j �� dfv r8t
� d | � |dkr6|� dS | j �
d
i ��
d i ��
d �rNt
� d
| � dS dS )z�Check dmi and kernel command line for dsname
NoCloud historically used "nocloud-net" as its dsname
for network timeframe (DEP_NETWORK), which supports http(s) urls.
For backwards compatiblity, check for that dsname.
z!The 'nocloud-net' datasource namez24.1zgUse 'nocloud' instead, which uses the seedfrom protocolscheme (http// or file://) to decide how to run.r� r Tr< r" zJMachine is configured by dmi serial number to run on single datasource %s.�
datasourcer r@ zMMachine is configured by system configuration to run on single datasource %s.F)r r r� r �
parse_cmdline�parse_cmdline_or_dmir rL r. r� rR rS r rV )r �log_deprecatedrk r r r � ds_detect� s@ �
��� ��z DataSourceNoCloudNet.ds_detectN)r� r� r� r r� r^ r� r r r r r� � s r� c C s t �| t�S rC )r �list_from_depends� datasources)�dependsr r r �get_datasource_list� s r� �__main__)�argv)�levelr� zseeded: zud: zvd: z network: rC )%rZ �loggingr � functoolsr � cloudinitr r r r �
cloudinit.netr � getLoggerr� rR r r r� rM r� rO r� �DEP_FILESYSTEM� DEP_NETWORKr� r� �sysr� � basicConfig�DEBUGr@ r` rr rs rt �network�printr r r r �<module> s:
!(
E� �