File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceConfigDrive.cpython-38.pyc
U
Ӈg�, � @ s� d dl Z d dlZd dlmZmZmZmZ d dlmZm Z d dl
m Z d dl m
Z
d dlmZ e �e�ZdZdeiZd Zd
Zd Zed d
� eD ��ZG dd� dejej�Z dd� Z dd� Z d dd�Z dd� Z d dd�Z!e Z"e ej#ffgZ$dd � Z%dS ) � N)� lifecycle�sources�subp�util)�
EventScope� EventType)�eni)�get_ibm_platform)� openstackziid-dsconfigdrive� instance-id)ZvfatZiso9660)zconfig-2zCONFIG-2)�srZcdc c s( | ] }t d �D ]}d||f V qqdS )� 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 )�DataSourceConfigDriveZ ConfigDrivec s\ t t| ��|||� d | _tj�|jd�| _d | _d | _ d | _
t j | _
d | _d | _i | _d S )NZ 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% �rootZmstrr r r r, : s z DataSourceConfigDrive.__str__c
C s� d }i }i }| j dfD ]N}tj�|�s(qzt|�}|}W qfW q tjk
rb t�t d|� Y qX q|�s| j
� d�}t |d�D ]�}d }t�
� r�|�d�r�d}ztj|t|d�}|}W nN tjk
r� Y n: tjk
r� Y n& tjk
�r t�t d|� Y nX |r� �qq�|�s d S |� d
i �}t�|tg�}| �|� d �| j� d �|d d
k�rbtjnd g�| _| jtjk�r~d S t| j�}|d } || k�r�t || j | jtjkd� | jtjk�r�t � d| | j� d S || _ || _ |� d�| _!|� d�| _"|d | _#| j$�%|� di �� |� d�}
zt�&|
�| _'W n6 t(k
�rf } zt �)d| � d | _'W 5 d } ~ X Y nX |� d�} zt�&| �| _*W n6 t(k
�r� } zt �)d| � d | _*W 5 d } ~ X Y nX |� d�| _+|� d�| _,dS )Nz
/config-drivez#Failed reading config drive from %sZdatasource_list)�dslistz/dev/cdZcd9660)�mtypezBroken config drive: %sF�metadatar- r � r )r' �networkz&%s: not claiming datasource, dsmode=%sz ec2-metadataZuserdatar$ Z
vendordataz"Invalid content in vendor-data: %sZ vendordata2z#Invalid content in vendor-data2: %s�network_configZ networkdataT)-r r r �isdir�read_config_driver
� NonReadabler �logexc�LOGr& �get�find_candidate_devsZis_BSD�
startswithZmount_cbZMountFailedErrorZBrokenMetadataZ
mergemanydict�DEFAULT_METADATAZ_determine_dsmodeZds_cfgr Z DSMODE_PASSr- ZDSMODE_DISABLED�get_previous_iidr( �
on_first_bootr' �debugr r1 r Z userdata_rawr r$ �updateZconvert_vendordataZvendordata_raw�
ValueError�warningZvendordata2_rawr" r! )
r% �foundZmdZresultsZsdirr/ Zdevr0 Zprev_iidZcur_iidZvd�eZvd2r r r � _get_data@ s�
�
��
� �
z DataSourceConfigDrive._get_datac C s t �| �� �S �N)r Z instance_id_matches_system_uuidZget_instance_id)r% r& r r r �check_instance_id� s z'DataSourceConfigDrive.check_instance_idc C s~ | j d krx| jd tjfkr:t�d� tj| j| jd�| _ n>| j d k rnt
� | j �| _ t�d� t j
dddd� n
t�d� | j S ) Nz(network config provided via network_json)r# z.network config provided via converted eni dataz(Eni network configuration in ConfigDrivez24.3z<You can use openstack's network configuration format instead)Z
deprecatedZdeprecated_versionZ
extra_messagez"no network configuration available)r r! r r r9 r@ r
Zconvert_net_jsonr# r" r �convert_eni_datar Z deprecate�r% r r r r4 � s"
�
�
z$DataSourceConfigDrive.network_configc C s dS )Nr
r rJ r r r �platform� s z DataSourceConfigDrive.platformc C s$ | j �d�rd}nd}d|| j f S )z/Return the subplatform metadata source details.z/devz config-diskzseed-dirz%s (%s))r r< )r% Zsubplatform_typer r r �_get_subplatform� s z&DataSourceConfigDrive._get_subplatform)�__name__�
__module__� __qualname__Zdsnamer ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZHOTPLUGZsupported_update_eventsr r, rF rH �propertyr4 rK rL �
__classcell__r r r) r r ! s" �� d
r c C s� t �| �}|jg i f|jg i fg}g }|D ]J\}}}z|||�W S t jk
rr } z|�|� W 5 d }~X Y q*X q*|d �d S )N���)r
ZConfigDriveReaderZread_v2Zread_v1r7 �append)Z
source_dir�readerZfindersZexcpsZfunctor�args�kwargsrE r r r r6 � s
� r6 c C sB t j�| �d�d�}zt�|��d�W S tk
r< Y d S X d S )N�datar �
)r r r Z get_cpathr Zload_text_file�rstrip�IOError)r( �fnamer r r r> � s
r> Tc C s^ t | t�stdt| � ��|rL| �dd�}|rL|rLt�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 %sr4 � z-Updating network interfaces from config driver$ N) �
isinstance�dict� TypeError�typer: r9 rC Zapply_network_configr rI �write_injected_files)rW r' r3 Znet_confr r r r? � s
�
r? c C sx | rtt �dt| �� | �� D ]V\}}|�tj�s:tj| }ztj||dd� W q t k
rp t�
t d|� Y q X q d S )NzWriting %s injected filesi� )�modezFailed writing file: %s) r9 r@ �len�itemsr< r �sepr Z
write_filerZ r8 )r$ �filenameZcontentr r r ra � s
ra c
s |dkr g }| rBt D ],}ztj|d� W q tjk
r> Y qX qg }tD ]}|�t�d| �� qJg � tD ]}� �t�d| �� ql|jdd� � jdd� � � fdd�|D � }� fd d�|D �}t �
d
||� |�rd |k�rt � \}} | |k�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 | ]}|� kr|�qS r r �r �d�Zby_labelr r �
<listcomp>* s z'find_candidate_devs.<locals>.<listcomp>c s" g | ]}|� kst �|�s|�qS r )r Z is_partitionrh rj r r rk . s
zdevices=%s dslist=%sZIBMCloudz5IBMCloud device '%s' (%s) removed from candidate list)
�OPTICAL_DEVICESr Zfind_devs_withr ZProcessExecutionError�FS_TYPES�extend� LABEL_TYPES�sortr9 r@ r �remove)
Z
probe_opticalr/ ZdeviceZ by_fstypeZfs_typeZlabelZ
candidatesZdevicesZ ibm_platformZibm_pathr rj r r; s>
�
�r; c C s t �| t�S rG )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_listK s rs )NT)TN)&Zloggingr Z cloudinitr r r r Zcloudinit.eventr r Z
cloudinit.netr Z$cloudinit.sources.DataSourceIBMCloudr Zcloudinit.sources.helpersr
Z getLoggerrM r9 Z DEFAULT_IIDr= rm ro ZPOSSIBLE_MOUNTS�tuplerl Z SourceMixinr+ r r6 r> r? ra r; ZDataSourceConfigDriveNetZDEP_FILESYSTEMrr rs r r r r �<module> s8
� � ,
B
�