3
\�me�3 � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZ ej e
�Z G dd� dej �Z
ddd�Zddd �Zdd
d �Zd d
� ZG dd� de
�Ze
ejffeejejffgZdd� ZdS )� N)�dmi�sources�util)�enic @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d � �Z d d
� Z
dd� Z dd� Z edd� �Z
dS )�DataSourceNoCloudZNoCloudc C sJ t jj| |||� d | _tjj|jd�tjj|jd�g| _d | _d| _ d S )N�nocloudz nocloud-net�/�file://)r r )
r �
DataSource�__init__�seed�os�path�join�seed_dir� seed_dirs�supported_seed_starts)�self�sys_cfg�distro�paths� r �'/usr/lib/python3.6/DataSourceNoCloud.pyr s zDataSourceNoCloud.__init__c C s t jj| �}d|| j| jf S )Nz%s [seed=%s][dsmode=%s])r r
�__str__r �dsmode)r �rootr r r r # s zDataSourceNoCloud.__str__c C s| t jd�}|jt jd�� t jd|j� �}|jt jd|j� �� |jt jd| �� tt|�t|�@ �}|jdd� |S )Nz TYPE=vfatz TYPE=iso9660zLABEL=%szLABEL_FATBOOT=%sT)�reverse)r Zfind_devs_with�extend�upper�lower�list�set�sort)r �labelZfslistZ
label_listZdevlistr r r � _get_devices' s
z DataSourceNoCloud._get_devicesc " C s� d| j d�}g }i ddd d�}y8i }tjd�}|rTt||�rT|jd� t|d|i�}W n tk
rv tjt d� d S X y(i }t|�r�|jd
� t|d|i�}W n tk
r� tjt d � d S X d dgd
dgd�}xZ| j
D ]P}y4tj |f|�}|j|� t j d|� t||�}P W q� t
k
�r* Y q�X q�W | jjd��rZ|jd� | jd |d d< d | jk�r�d| jk�r�t|| j�}|jd� dd� } | jjdd�}
|
d k �r�x�| j|
�D ]�} ynt j d| � ytj| | |�}W n( t
k
�r t jd| |
� �w�Y nX t||�}t j d| � |j| � P W nX tk
�rZ } z| jtjk�rJ� W Y d d } ~ X n& tjk
�r~ tjt d| � Y nX �q�W t|�d k�r�d S d|d k�rL|d d }
d }x"| jD ]}|
j|��r�|}P �q�W |�s�t j d |
| � d S tj|
�}
tj|
d d �\}}}t j d |
� tj |d |g�|d< ||d < ||d
< |j|
� tj |d |g�|d< | j |d jd �g�| _ | j t j k�r�t j d!| | j � d S d"j |�| _!|d | _"|d | _#|d
| _$|d | _%|d jd#�| _&d$S )%Nr )z instance-idr � )z meta-dataz user-dataz vendor-dataznetwork-configzsystem-serial-numberr z meta-datazUnable to parse dmi dataF�cmdlinez!Unable to parse command line dataz user-dataz vendor-dataznetwork-config)�requiredZoptionalzUsing seeded data from %s�seedfromZds_config_seedfromZ ds_configc S s t j| f|�S )N)r �pathprefix2dict)Zmp�datar r r �_pp2d_callbackt s z3DataSourceNoCloud._get_data.<locals>._pp2d_callbackZfs_labelZcidataz 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 datar z Seed from %s not supported by %s)Ztimeoutz Using seeded cache data from %sr z&%s: not claiming datasource, dsmode=%s�,znetwork-interfacesT)'r r Z
read_dmi_data�load_cmdline_data�append�_merge_new_seed� Exceptionr Zlogexc�LOGr r) �debug�
ValueErrorZds_cfg�getr$ Zmount_cbZwarning�OSError�errno�ENOENTZMountFailedError�lenr �
startswithZ sub_dmi_varsZ read_seeded�
mergemanydictZ_determine_dsmoder ZDSMODE_DISABLEDr r ZmetadataZ userdata_rawZvendordata_raw�_network_config� _network_eni)r Zdefaults�foundZmydata�md�serialZ pp2d_kwargsr �seededr+ r# Zdev�er( Z seedfound�protoZmd_seedZudZvdr r r � _get_data3 s�
zDataSourceNoCloud._get_datac C s. t | d�sd | _| js(tj� r"dnd| _| jS )N�_platform_typeZlxdr )�hasattrrD r Zis_lxd)r r r r �
platform_type� s
z DataSourceNoCloud.platform_typec C s t jS )z=Return unknown when 'cloud-name' key is absent from metadata.)r ZMETADATA_UNKNOWN)r r r r �_get_cloud_name� s z!DataSourceNoCloud._get_cloud_namec C s$ | j jd�rd}nd}d|| j f S )z/Return the subplatform metadata source details.z/devz config-diskzseed-dirz%s (%s))r r9 )r Zsubplatform_typer r r �_get_subplatform� s z"DataSourceNoCloud._get_subplatformc C s: | j � }|sd S t| d| jg�}t|d�}|s2d S ||kS )Nr )�dirs)Zget_instance_id�getattrr �_quick_read_instance_id)r r ZcurrentrI Zquick_idr r r �check_instance_id� s
z#DataSourceNoCloud.check_instance_idc C s( | j d kr"| jd k r"tj| j�| _ | j S )N)r; r< r Zconvert_eni_data)r r r r �network_config� s
z DataSourceNoCloud.network_configN)�__name__�
__module__� __qualname__Zdsnamer r r$ rC �propertyrF rG rH rL rM r r r r r s
r c C s� | d kr g } d}i }t |�r,||kr,|| S xb| D ]Z}|d kr@q2y6tj|dgd�}tj|d �}|rt||krt|| S W q2 tk
r� Y q2X q2W d S )Nz instance-idz meta-data)r' )r- r r) � load_yamlr3 )rI Ziid_key�fill�dr* r> r r r rK � s"
rK c C s� dt jfdt jfg}xn|D ]f\}}t|| |�s0qd| kr<dS | jd�}|rv|jd �r`t j| d< q~|jd �r~t j| d< n|| d< dS W d
S )
Nz
ds=nocloudzds=nocloud-netr Tr( �http://�https://�file://r F)rU rV )rW r )r Z DSMODE_LOCALZDSMODE_NETWORK�parse_cmdline_datar4 r9 )rS r& ZpairsZidstrr r( r r r r- s
r- c C s� |d krt j� }d| }d| |kp.d| |ks4dS d}x&|j� D ]}|j| �rB|jdd�}qBW |d jd�}t|�dkr�|dd � }nf }dd d
d �}x`|D ]X}|dkr�q�y|jdd�\} }
W n tk
r� |} d }
Y nX | |kr�|| } |
|| < q�W d S )
Nz %s z %s;Fr% �=� �;zlocal-hostnamez instance-idr( )�h�i�sT)r Z get_cmdline�splitr9 r8 r0 ) Zds_idrS r& Zargline�tokZtmpZkvpairsZs2l�item�k�vr r r rX 3 s4
rX c C s� | j � }|jdi �}t|d t�s0tj|d �}tj| d |g�|d< |jd�rdtj|jd��|d< d|krx|d |d< d|kr�|d |d< |S )Nz meta-dataznetwork-configz user-dataz vendor-data)�copyr4 �
isinstance�dictr rR r: )Zcurr@ �retZnewmdr r r r/ [ s
r/ c @ s e Zd Zdd� Zdd� ZdS )�DataSourceNoCloudNetc C s t j| |||� d| _d S )N�http://�https://)ri rj )r r r )r r r r r r r r n s z DataSourceNoCloudNet.__init__c C s dt j� kS )a NoCloud requires "nocloud-net" as the way to specify
seeding from an http(s) address. This diverges from all other
datasources in that it does a kernel commandline match on something
other than the datasource dsname for only DEP_NETWORK.
z nocloud-net)r Z
parse_cmdline)r r r r � ds_detectr s z DataSourceNoCloudNet.ds_detectN)rN rO rP r rk r r r r rh m s rh c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s rm )N)N)N)r6 Zloggingr
Z cloudinitr r r Z
cloudinit.netr Z getLoggerrN r1 r
r rK r- rX r/ rh ZDEP_FILESYSTEMZ DEP_NETWORKrl rm r r r r �<module> s
i
(