Your IP : 216.73.217.13
U
�Ad�) � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z e� e
�ZG dd� de j�ZG dd� de�Zd d
d �Zd d
� Zdd� Zd dd�Zdd� Zdd� Zdd� ZG dd� de�Zee jffee je j fffZ dd � Z d d � Z dS )!� N)�minidom)�log)�safeyaml�sources�subp�utilc @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z dS )�
DataSourceOVFZOVFc C sF t j�| |||� d | _tj�|jd�| _d | _i | _ d| _
d | _ d S )NZovf)�/zfile://) r �
DataSource�__init__�seed�os�path�join�seed_dir� environment�cfg�supported_seed_starts�_network_config��selfZsys_cfgZdistro�paths� r �A/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceOVF.pyr s zDataSourceOVF.__init__c C s t j�| �}d|| jf S )Nz %s [seed=%s])r r
�__str__r )r �rootr r r r # s zDataSourceOVF.__str__c C s� g }i }d}d}ddi}t | jj�\}}|r^tj�| jj|�}t|�\}}} || _|�|� nndt fdt
fg}
d } |
D ]\} } | � }|rv q�qv|r�t|d�\}}} || _d|kr�|d r�|d | _ |�| � t |�dkr�d S d
|k�rp|d
�rp|d
}
d }| j
D ] }|
�|��r|} �q �q|�s8t�d |
| � d S tj|
d d �\}}}t�d
|
� t�||g�}|�|
� t�||g�}d�|�| _|| _|| _|| _| | _dS )N� � instance-idz iid-dsovfzcom.vmware.guestInfoZisoT�network-configr F�seedfromz Seed from %s not supported by %s)Ztimeoutz Using seeded cache data from %s�,)� get_ovf_envr r r
r r �read_ovf_environmentr �append�transport_vmware_guestinfo�transport_iso9660r �lenr �
startswith�LOG�debugr Z read_seededZ
mergemanydictr �metadataZ userdata_rawZvendordata_rawr )r �found�md�udZvd�defaultsZseedfile�contentsr r Znp�nameZ transfuncr Z seedfound�protoZmd_seedr r r � _get_data' sd � �
zDataSourceOVF._get_datac C s
d| j S )Nzovf (%s))r �r r r r �_get_subplatformi s z DataSourceOVF._get_subplatformc C s0 d| j krg S | j d }t|t�r&|S |gS d S )N� public-keys)r* �
isinstance�list)r Zpksr r r �get_public_ssh_keysl s
z!DataSourceOVF.get_public_ssh_keysc C s | j S �N)r r3 r r r �get_config_objx s z DataSourceOVF.get_config_objc C s | j S r9 )r r3 r r r �network_config{ s z DataSourceOVF.network_configN) �__name__�
__module__� __qualname__Zdsnamer r r2 r4 r8 r: �propertyr; r r r r r s B r c @ s e Zd Zdd� ZdS )�DataSourceOVFNetc C s, t �| |||� tj�|jd�| _d| _d S )Nzovf-net)zhttp://zhttps://)r r r
r r r r r r r r r � s zDataSourceOVFNet.__init__N)r< r= r> r r r r r r@ � s r@ Fc C s t | �}i }i }d }dg}ddddg}dg}|�� D ]�\} }
| dkrHd} | |krZ|
|| < q4| |krl|
|| < q4| |kr�|r�z$t�|
�� �} t| ��d�|| < W q� tk
r� t� d � Y q�X q4| d
kr4zt�|
�� �}W q4 tk
r� |
�� }Y q4X q4|||fS ) NZpasswordr zlocal-hostnamer5 r r ZhostnameZnetworkz%Ignore network-config in wrong formatz user-data)
�get_properties�items�base64Z b64decode�encode�safeload_yaml_or_dict�get� Exceptionr( r) ) r/ Z read_network�propsr, r r- Z cfg_propsZmd_propsZ
network_propsZprop�valr; r r r r" � s4
r" c C sh d}|D ]Z}t j�| |�}t j�|�rzt�|�}||fW S tk
r` t�td|� Y qX qdS )N)z ovf-env.xmlz ovf_env.xmlz OVF_ENV.XMLz OVF-ENV.XMLzFailed loading ovf file %s)NF) r
r r �isfiler Z load_filerG Zlogexcr( )�dirnameZ env_names�fnameZfull_fnr/ r r r r! � s
r! c C s� | sdS t | t�s td| ��tj�| �} | �d�rB| �d�d } | �d�r\| �d�d } n| � d�dkrndS | svdS d}tj
� d|�}t �
|�}|�| �d k S )
z�Test if devname matches known list of devices which may contain iso9660
filesystems.
Be helpful in accepting either knames (with no leading /dev/) or full path
names, but do not allow paths outside of /dev/, like /dev/foo/bar/xxx.
Fz Unexpected input for devname: %sz/dev/���r r z^(sr[0-9]+|hd[a-z]|xvd.*)ZCLOUD_INIT_CDROM_DEV_REGEXN)r6 �str�
ValueErrorr
r �normpathr' � partition�split�count�environrF �re�compile�match)ZdevnameZ
default_regexZ
devname_regexZcdmatchr r r �maybe_cdrom_device� s"
rX Tc
C s� t �� }|�� D ]L\}}|d }|dkr.| r.qt|�s8q|d }t|�\}}|dk r| S q| rhd}nd }dd� t �|r�d| nd �D �} | D ]T}zt j|t|d�\}}W n& t jk
r� t� d |� Y q�Y nX |dk r�| S q�d S )
N�fstypeZiso9660Z
mountpointFc S s g | ]}t |�r|�qS r )rX )�.0�devr r r �
<listcomp>� s �z%transport_iso9660.<locals>.<listcomp>zTYPE=%s)�mtypez%s not mountable as iso9660)
r �mountsrB rX r! Zfind_devs_withZmount_cbZMountFailedErrorr( r) )
Z require_isor^ r[ �inforY ZmpZ_fnamer/ r] Zdevsr r r r% � s4
�
r% c
C s� d} d }t �| �s|S | dg}z*t � |�\}}|r8|W S t�d||� W nH t jk
r� } z(|jdkr�t�d| |j� t�|� W 5 d }~X Y nX |S )Nzvmware-rpctoolzinfo-get guestinfo.ovfEnvz%cmd %s exited 0 with empty stdout: %s� z%s exited with code %d)r Zwhichr( r) ZProcessExecutionErrorZ exit_codeZwarning)ZrpctoolZ not_found�cmd�outZ_err�er r r r$ s
r$ c C s2 g }| � � s|S | jD ]}||�r|�|� q|S r9 )�
hasChildNodesZ
childNodesr# )ZnodeZ filter_funcZretZchildr r r �
find_child s
re c C s� t �| �}|jjdkr td��|j�� s0td��d}t|jdd� �}t|�dkrXtd��i }t|d d d� �}|D ],}|j� |d
�j
}|j� |d �j
}|||< qr|S ) NZ EnvironmentzNo Environment NodezNo Child Nodesz)http://schemas.dmtf.org/ovf/environment/1c S s
| j dkS )NZPropertySection�� localName��nr r r �<lambda>/ � z get_properties.<locals>.<lambda>r zNo 'PropertySection'sc S s
| j dkS )NZPropertyrf rh r r r rj 7 rk �key�value) r Z parseStringZdocumentElementrg �XmlErrorrd re r& Z
attributesZgetNamedItemNSrm ) r/ ZdomZenvNsURIZ propSectionsrH Z propElems�elemrl rI r r r rA ! s,
� �
rA c @ s e Zd ZdS )rn N)r< r= r> r r r r rn B s rn c C s t �| t�S r9 )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_listN s rq c C s | si S t �| �S )z�
The meta data could be JSON or YAML. Since YAML is a strict superset of
JSON, we will unmarshal the data as YAML. If data is None then a new
dictionary is returned.
)r �load)�datar r r rE R s rE )F)T) rC r
rU Zxml.domr Z cloudinitr Zloggingr r r r Z getLoggerr< r( r
r r@ r" r! rX r% r$ re rA rG rn ZDEP_FILESYSTEMZ DEP_NETWORKrp rq rE r r r r �<module> s*
j
)
'
!
�