File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceConfigDrive.cpython-310.pyc
o
�Ad , � @ s� d dl Z d dlmZ d dlmZmZmZ d dlmZm Z d dl
m Z d dl m
Z
d dlmZ e�e�ZdZd Zd
eiZd Zd Zd
Zedd� eD ��ZG dd� dejej �Z dd� Z dd� Z d dd�Z dd� Z!d dd �Z"e Z#e ej$ffgZ%d d � Z&dS )!� N)�log)�sources�subp�util)�
EventScope� EventType)�eni)�get_ibm_platform)� openstackziid-dsconfigdrive�pass� instance-id)�vfat�iso9660)zconfig-2zCONFIG-2)�sr�cdc c s, � | ]}t d d�D ] }d||f V q qdS )r � z /dev/%s%sN)�range)�.0�z�i� r �I/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceConfigDrive.py� <genexpr> s �* r c sr e Zd ZdZejejejej ej
hiZ � fdd�Z dd� Z
dd� Zdd � Zed
d � �Zed d
� �Zdd� Z� ZS )�DataSourceConfigDrive� ConfigDrivec s\ t t| ��|||� d | _tj�|jd�| _d | _d | _ d | _
t j | _
d | _d | _i | _d S )N� config_drive)�superr �__init__�source�os�path�join�seed_dir�version� ec2_metadata�_network_configr �UNSET� network_json� network_eni�
known_macs�files)�self�sys_cfg�distro�paths�� __class__r r r / s
z DataSourceConfigDrive.__init__c C s0 t j�| �}d|| j| jf }|d| j 7 }|S )Nz%s [%s,ver=%s]z [source=%s])r �
DataSource�__str__�dsmoder# r )r+ �root�mstrr r r r2 ; s z DataSourceConfigDrive.__str__c
C s� d }i }i }| j dfD ]#}tj�|�sq z t|�}|}W n tjy. t�t d|� Y q w |s�| j
� d�}t |d�D ]C}d }t�
� rK|�d�rKd}z tj|t|d�}|}W n# tjya Y n tjyj Y n tjyz t�t d|� Y nw |r nq<|s�d S |� d
i �}t�|tg�}| �|� d �| j� d �|d d
kr�tjnd g�| _| jtjkr�d S t| j�}|d } || kr�t || j | jtjkd� | jtjkr�t � d| | j� d S || _ || _ |� d�| _!|� d�| _"|d | _#| j$�%|� di �� |� d�}
|
| _&zt�'|
�| _(W n t)�y( } zt �*d| � d | _(W Y d } ~ nd } ~ ww |� d�} | | _+zt�'| �| _,W n t)�yU } zt �*d| � d | _,W Y d } ~ nd } ~ ww |� d�| _-|� d�| _.dS )Nz
/config-drivez#Failed reading config drive from %s�datasource_list)�dslistz/dev/cd�cd9660)�mtypezBroken config drive: %sF�metadatar3 r# � r )r- �networkz&%s: not claiming datasource, dsmode=%sz ec2-metadata�userdatar* �
vendordataz"Invalid content in vendor-data: %s� vendordata2z#Invalid content in vendor-data2: %s�network_config� networkdataT)/r" r r �isdir�read_config_driver
� NonReadabler �logexc�LOGr, �get�find_candidate_devs�is_BSD�
startswith�mount_cb�MountFailedError�BrokenMetadata�
mergemanydict�DEFAULT_METADATA�_determine_dsmode�ds_cfgr � DSMODE_PASSr3 �DSMODE_DISABLED�get_previous_iidr. �
on_first_bootr- �debugr r: r$ � userdata_rawr# r* �update�vendordata_pure�convert_vendordata�vendordata_raw�
ValueError�warning�vendordata2_pure�vendordata2_rawr( r' )
r+ �found�md�results�sdirr7 �devr9 �prev_iid�cur_iid�vd�e�vd2r r r � _get_dataA s� �
���
��
� �
��
�� z DataSourceConfigDrive._get_datac C s t �| �� �S �N)r � instance_id_matches_system_uuid�get_instance_id)r+ r, r r r �check_instance_id� s z'DataSourceConfigDrive.check_instance_idc C sv | j d u r8| jd tjfvr t�d� tj| j| jd�| _ | j S | j d ur3t
� | j �| _ t�d� | j S t�d� | j S )Nz(network config provided via network_json)r) z.network config provided via converted eni dataz"no network configuration available) r% r' r r& rF rV r
�convert_net_jsonr) r( r �convert_eni_data�r+ r r r r@ � s
�
�
�z$DataSourceConfigDrive.network_configc C s dS )Nr
r rq r r r �platform� s z DataSourceConfigDrive.platformc 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 rJ )r+ �subplatform_typer r r �_get_subplatform� s z&DataSourceConfigDrive._get_subplatform)�__name__�
__module__� __qualname__�dsnamer �NETWORKr �BOOT_NEW_INSTANCE�BOOT� BOOT_LEGACY�HOTPLUG�supported_update_eventsr r2 rj rn �propertyr@ rr rt �
__classcell__r r r/ r r " s$ �� f
r c C s� t �| �}|jg i f|jg i fg}g }|D ]'\}}}z
||i |��W S t jy< } z |�|� W Y d }~qd }~ww |d �)N���)r
�ConfigDriveReader�read_v2�read_v1rD �append)�
source_dir�reader�finders�excps�functor�args�kwargsrh r r r rC � s
���rC c C s< t j�| �d�d�}z t�|��d�W S ty Y d S w )N�datar �
)r r r! � get_cpathr � load_file�rstrip�IOError)r. �fnamer r r rT � s �rT Tc C s^ t | t�s
tdt| � ��|r&| �dd�}|r&|r&t�d� |�t� |�� t
| �d�� dS )zDPerforms any first-boot actions using data read from a config-drive.z/Config-drive data expected to be a dict; not %sr@ � z-Updating network interfaces from config driver* N) �
isinstance�dict� TypeError�typerG rF r] �apply_network_configr rp �write_injected_files)r� r- r< �net_confr r r rU � s
�
rU c C sx | r8t �dt| �� | �� D ]+\}}|�tj�s tj| }z
tj||dd� W q t y7 t�
t d|� Y qw d S d S )NzWriting %s injected filesi� )�modezFailed writing file: %s) rF rV �len�itemsrJ r �sepr �
write_filer� rE )r* �filename�contentr r r r� � s
��r� c
s
|du rg }| r t D ]}ztj|d� W q
tjy Y q
w g }tD ] }|�t�d| �� q#g � tD ] }� �t�d| �� q4|jdd� � jdd� � � fdd�|D � }� fd d�|D �}t �
d
||� |r�d |v r�t � \}} | |v r�|� | � t �
d | |� |S )
a Return a list of devices that may contain the config drive.
The returned list is sorted by search order where the first item has
should be searched first (highest priority)
config drive v1:
Per documentation, this is "associated as the last available disk on the
instance", and should be VFAT.
Currently, we do not restrict search list to "last available disk"
config drive v2:
Disk should be:
* either vfat or iso9660 formatted
* labeled with 'config-2' or 'CONFIG-2'
N)r zTYPE=%szLABEL=%sT)�reversec s g | ]}|� vr|�qS r r �r �d��by_labelr r �
<listcomp>% s z'find_candidate_devs.<locals>.<listcomp>c s" g | ]
}|� v s
t �|�s|�qS r )r � is_partitionr� r� r r r� ) s ���zdevices=%s dslist=%s�IBMCloudz5IBMCloud device '%s' (%s) removed from candidate list)
�OPTICAL_DEVICESr �find_devs_withr �ProcessExecutionError�FS_TYPES�extend� LABEL_TYPES�sortrF rV r �remove)
�
probe_opticalr7 �device� by_fstype�fs_type�label�
candidates�devices� ibm_platform�ibm_pathr r� r rH � s@ �
�
�rH c C s t �| t�S rk )r �list_from_depends� datasources)�dependsr r r �get_datasource_listF s r� )NT)TN)'r � cloudinitr �loggingr r r �cloudinit.eventr r �
cloudinit.netr �$cloudinit.sources.DataSourceIBMCloudr �cloudinit.sources.helpersr
� getLoggerru rF � DEFAULT_IID� DEFAULT_MODErO r� r� �POSSIBLE_MOUNTS�tupler� � SourceMixinr1 r rC rT rU r� rH �DataSourceConfigDriveNet�DEP_FILESYSTEMr� r� r r r r �<module> s:
� � &
B
�