o
    x[hê,  ã                   @   sð   d dl Z d dlZd dlmZmZmZmZ d dlmZm	Z	 d dl
mZ d dlmZ d dlmZ e  e¡ZdZdeiZd	Zd
ZdZedd„ eD ƒƒZG dd„ dejejƒZdd„ Zdd„ Zddd„Zdd„ Z ddd„Z!eZ"eej#ffgZ$dd„ Z%dS )é    N)Ú	lifecycleÚsourcesÚsubpÚutil)Ú
EventScopeÚ	EventType)Úeni)Úget_ibm_platform)Ú	openstackziid-dsconfigdriveúinstance-id)ÚvfatÚiso9660)zconfig-2zCONFIG-2)ÚsrÚcdc                 c   s*   � | ]}t d ƒD ]	}d||f V  qqdS )é   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 Zd ZdZejejejej	ej
hiZ‡ fdd„Zdd„ Zdd„ Zdd	„ Zed
d„ ƒZedd„ ƒZdd„ Z‡  ZS )ÚDataSourceConfigDriveÚConfigDrivec                    s\   t t| ƒ |||¡ d | _tj |jd¡| _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| jf }|d| j 7 }|S )Nz%s [%s,ver=%s]z[source=%s])r   Ú
DataSourceÚ__str__Údsmoder"   r   )r*   ÚrootÚmstrr   r   r   r1   :   s   zDataSourceConfigDrive.__str__c              
   C   s¼  d }i }i }| j dfD ]#}tj |¡sqz	t|ƒ}|}W  n tjy.   t t	d|¡ Y qw |s€| j
 d¡}t|d�D ]C}d }t ¡ rK| d¡rKd}ztj|t|d�}|}W n# tjya   Y n tjyj   Y n tjyz   t t	d|¡ Y nw |r nq<|s„d	S | d
i ¡}t |tg¡}|  | d¡| j d¡|d dkr¥tjnd g¡| _| jtjkr²d	S t| jƒ}|d }	||	krËt|| j| jtjkd� | jtjkrÛt	 d| | j¡ d	S || _|| _ | d¡| _!| d¡| _"|d | _#| j$ %| di ¡¡ | d¡}
zt &|
¡| _'W n t(�y% } zt	 )d|¡ d | _'W Y d }~nd }~ww | d¡}zt &|¡| _*W n t(�yO } zt	 )d|¡ d | _*W Y d }~nd }~ww | d¡| _+| d¡| _,dS )Nz/config-drivez#Failed reading config drive from %sÚdatasource_list)Údslistz/dev/cdÚcd9660)ÚmtypezBroken config drive: %sFÚmetadatar2   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_PASSr2   ÚDSMODE_DISABLEDÚget_previous_iidr-   Úon_first_bootr,   Údebugr   r9   r#   Úuserdata_rawr"   r)   ÚupdateÚconvert_vendordataÚvendordata_rawÚ
ValueErrorÚwarningÚvendordata2_rawr'   r&   )r*   ÚfoundÚmdÚresultsÚsdirr6   Údevr8   Úprev_iidÚcur_iidÚvdÚeÚvd2r   r   r   Ú	_get_data@   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   s†   | j d u r@| jd tjfvrt d¡ tj| j| jd�| _ | j S | j	d ur;t
 | j	¡| _ t d¡ tjdddd� | j S t d¡ | j S )	Nz(network config provided via network_json)r(   z.network config provided via converted eni dataz(Eni network configuration in ConfigDrivez24.3z<You can use openstack's network configuration format instead)Ú
deprecatedÚdeprecated_versionÚextra_messagez"no network configuration available)r$   r&   r   r%   rE   rU   r
   Úconvert_net_jsonr(   r'   r   Úconvert_eni_datar   Ú	deprecate©r*   r   r   r   r?   ¨   s$   

ÿ
ó
ü

ÿz$DataSourceConfigDrive.network_configc                 C   s   dS )Nr
   r   rr   r   r   r   Úplatform¿   s   zDataSourceConfigDrive.platformc                 C   s$   | j  d¡r	d}nd}d|| j f S )z/Return the subplatform metadata source details.z/devzconfig-diskzseed-dirz%s (%s))r   rI   )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   r1   rg   rk   Úpropertyr?   rs   ru   Ú__classcell__r   r   r.   r   r   !   s$    üÿ	d

r   c                 C   s‚   t  | ¡}|jg i f|jg i fg}g }|D ]'\}}}z
||i |¤ŽW   S  t jy< } z| |¡ W Y d }~qd }~ww |d ‚)Néÿÿÿÿ)r
   ÚConfigDriveReaderÚread_v2Úread_v1rC   Úappend)Ú
source_dirÚreaderÚfindersÚexcpsÚfunctorÚargsÚkwargsre   r   r   r   rB   Ì   s   


þ€ÿrB   c                 C   s<   t j |  d¡d¡}z	t |¡ d¡W S  ty   Y d S w )NÚdatar   Ú
)r   r   r    Ú	get_cpathr   Úload_text_fileÚrstripÚIOError)r-   Úfnamer   r   r   rS   Û   s   ÿrS   Tc                 C   s^   t | tƒstdt| ƒ ƒ‚|r&|  dd¡}|r&|r&t d¡ | t 	|¡¡ t
|  d¡ƒ dS )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ÚtyperF   rE   r[   Úapply_network_configr   rp   Úwrite_injected_files)rŽ   r,   r;   Únet_confr   r   r   rT   æ   s   

ÿ
rT   c              	   C   sx   | r8t  dt| ƒ¡ |  ¡ D ]+\}}| tj¡stj| }z
tj||dd� W q t	y7   t 
t d|¡ Y qw d S d S )NzWriting %s injected filesi°  )ÚmodezFailed writing file: %s)rE   rU   ÚlenÚitemsrI   r   Úsepr   Ú
write_filer“   rD   )r)   ÚfilenameÚcontentr   r   r   r›   ô   s   
ÿùr›   c           
   	      s
  |du rg }| rt D ]}ztj|d� W q
 tjy   Y q
w g }tD ]}| t d| ¡¡ q#g ‰ tD ]}ˆ  t d| ¡¡ q4|jdd� ˆ jdd� ˆ ‡ fdd„|D ƒ }‡ fd	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   zTYPE=%szLABEL=%sT)Úreversec                    s   g | ]}|ˆ vr|‘qS r   r   ©r   Úd©Úby_labelr   r   Ú
<listcomp>*  s    z'find_candidate_devs.<locals>.<listcomp>c                    s"   g | ]}|ˆ v st  |¡s|‘qS r   )r   Úis_partitionr¥   r§   r   r   r©   .  s    zdevices=%s dslist=%sÚIBMCloudz5IBMCloud device '%s' (%s) removed from candidate list)ÚOPTICAL_DEVICESr   Úfind_devs_withr   ÚProcessExecutionErrorÚFS_TYPESÚextendÚLABEL_TYPESÚsortrE   rU   r	   Úremove)
Úprobe_opticalr6   ÚdeviceÚ	by_fstypeÚfs_typeÚlabelÚ
candidatesÚdevicesÚibm_platformÚibm_pathr   r§   r   rG      s@   ÿ
ÿ

ýrG   c                 C   s   t  | t¡S rh   )r   Úlist_from_dependsÚdatasources)Údependsr   r   r   Úget_datasource_listK  s   rÀ   )NT)TN)&Úloggingr   Ú	cloudinitr   r   r   r   Úcloudinit.eventr   r   Úcloudinit.netr   Ú$cloudinit.sources.DataSourceIBMCloudr	   Úcloudinit.sources.helpersr
   Ú	getLoggerrv   rE   ÚDEFAULT_IIDrN   r¯   r±   ÚPOSSIBLE_MOUNTSÚtupler¬   ÚSourceMixinr0   r   rB   rS   rT   r›   rG   ÚDataSourceConfigDriveNetÚDEP_FILESYSTEMr¾   rÀ   r   r   r   r   Ú<module>   s8   
ÿÿ ,

B
ÿ