3
\�me�2 � @ s d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m
Z
m Z ej e
�ZG dd� de j�ZG dd� de�Zd#d
d �Zd d
� Zdd� Zd$dd�Zdd� Zdd� Zdd� Zdd� Zdd � ZG d d � d e�Z ee j ffee j e j fffZ d d � Z d!d"� Z!dS )%zZCloud-Init DataSource for OVF
This module provides a cloud-init datasource for OVF data.
� N)�minidom)�safeyaml�sources�subp�utilc @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z dS )�
DataSourceOVFZOVFc C sF t jj| |||� d | _tjj|jd�| _d | _i | _ d| _
d | _ d S )NZovf�/�file://)r r ) r �
DataSource�__init__�seed�os�path�join�seed_dir� environment�cfg�supported_seed_starts�_network_config)�self�sys_cfg�distro�paths� r �#/usr/lib/python3.6/DataSourceOVF.pyr s zDataSourceOVF.__init__c C s t jj| �}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^tjj| jj|�}t|�\}}} || _|j|� npdt fdt
fg}
d } x |
D ]\} } | � }|rxP qxW |r�t|d�\}}} || _d|kr�|d r�|d | _ |j| � t |�dkr�d S d
|ko�|d
�rp|d
}
d }x"| j
D ]}|
j|��r|}P �qW |�s8tjd |
| � d S tj|
d d �\}}}tjd
|
� tj||g�}|j|
� tj||g�}dj|�| _|| _|| _|| _| | _dS )N� z instance-idz iid-dsovfzcom.vmware.guestInfoZisoTznetwork-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�udZvdZdefaultsZseedfile�contentsr r Znp�nameZ transfuncr Z seedfound�protoZmd_seedr r r � _get_data+ s^
zDataSourceOVF._get_datac C s
d| j S )Nzovf (%s))r )r r r r �_get_subplatformm s z DataSourceOVF._get_subplatformc C s0 d| j krg S | j d }t|t�r&|S |gS d S )Nz public-keys)r) �
isinstance�list)r Zpksr r r �get_public_ssh_keysp s
z!DataSourceOVF.get_public_ssh_keysc C s | j S )N)r )r r r r �get_config_obj| s z DataSourceOVF.get_config_objc C s | j S )N)r )r r r r �network_config s z DataSourceOVF.network_configN) �__name__�
__module__� __qualname__Zdsnamer r r0 r1 r4 r5 �propertyr6 r r r r r s B r c @ s e Zd Zdd� ZdS )�DataSourceOVFNetc C s, t j| |||� tjj|jd�| _d| _d S )Nzovf-net�http://�https://)r< r= )r r r
r r r r )r r r r r r r r � s zDataSourceOVFNet.__init__N)r7 r8 r9 r r r r r r; � s r; Fc C s t | �}i }i }d }dg}ddddg}dg}x�|j� D ]�\} }
| dkrJd} | |kr\|
|| < q6| |krn|
|| < q6| |kr�|r�y$tj|
j� �} t| �jd�|| < W q� tk
r� tj d � Y q�X q6| d
kr6ytj|
j� �}W q6 tk
r� |
j� }Y q6X q6W |||fS ) NZpasswordr zlocal-hostnamez public-keysz instance-idznetwork-configZhostnameZnetworkz%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�valr6 r r r r! � s4
r! c C sf d}x\|D ]T}t jj| |�}t jj|�r
ytj|�}||fS tk
r\ tjtd|� Y q
X q
W dS ) N� ovf-env.xml� ovf_env.xml� OVF_ENV.XML� OVF-ENV.XMLzFailed loading ovf file %sF)rG rH rI rJ )NF) r
r r �isfiler Z load_filerD �logexcr' )�dirnameZ env_namesZfnameZfull_fnr- r r r r � s
r c C s� | sdS t | t�s td| ��tjj| �} | jd�rB| jd�d
} | jd�r\| jd�d } n| j d�dkrndS | svdS d}tj
j d|�}t j
|�}|j| �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���rO )r2 �str�
ValueErrorr
r �normpathr& � partition�split�count�environrC �re�compile�match)ZdevnameZ
default_regexZ
devname_regexZcdmatchr r r �maybe_cdrom_device� s"
rZ Tc
C s� t j� }xT|j� D ]H\}}|d }|dkr0| r0qt|�s:q|d }t|�\}}|dk r|S qW | rhd}nd }dd� t j|r�d| nd �D �} xV| D ]N}yt j|t|d�\}}W n$ t jk
r� tj d |� w�Y nX |dk r�|S q�W d S )
N�fstypeZiso9660Z
mountpointFc S s g | ]}t |�r|�qS r )rZ )�.0�devr r r �
<listcomp>� s z%transport_iso9660.<locals>.<listcomp>zTYPE=%s)�mtypez%s not mountable as iso9660)
r �mountsr? rZ r Zfind_devs_withZmount_cbZMountFailedErrorr' r( )
Z require_isor` r] �infor[ ZmpZ_fnamer- r_ Zdevsr r r r$ � s2
r$ c C s | |g}t j |�\}}|||fS )N)r )�rpctool�arg�cmd�stdout�stderrr r r �exec_vmware_rpctool s rg c C s" | d|g}t j |�\}}|||fS )Nz--cmd)r )rb rc rd re rf r r r �
exec_vmtoolsd s
rh c C s d \} }t jd�}t jd�}|r6|t } }tjd|� |rT|sH|t } }tjd|� | sftjd� d S dd� }y
|| |�S t jk
�r } z�|r�| |kr�tjtd|� d S |s�tj d � d S ytj d
� ||t�S t jk
�r } ztjtd|� W Y d d }~X nX W Y d d }~X nX d S ) N�vmtoolsdzvmware-rpctoolz discovered vmware-rpctool: %szdiscovered vmtoolsd: %szno rpctool discoveredc S s| t jd| � y(|| d�\}}}|r&|S t jd|� d S tjk
rv } z$|jdkrbt jd| |j� |�W Y d d }~X nX d S )Nz query guestinfo.ovfEnv with %szinfo-get guestinfo.ovfEnvz!cmd %s exited 0 with empty stdoutrN z%s exited with code %d)r' ra r( r �ProcessExecutionErrorZ exit_codeZwarning)rb �
rpctool_fnrd re �_�errorr r r �query_guestinfo+ s
z3transport_vmware_guestinfo.<locals>.query_guestinfoz+vmtoolsd failed to get guestinfo.ovfEnv: %sz$vmtoolsd fallback option not presentzfallback to vmtoolsd)NN)
r Zwhichrg r' r( rh rj r rL ra )rb rk ri Zvmware_rpctoolrn rm r r r r# s>
.r# c C s6 g }| j � s|S x | jD ]}||�r|j|� qW |S )N)�
hasChildNodesZ
childNodesr" )ZnodeZ filter_func�retZchildr r r �
find_childY s rq c C s� t j| �}|jjdkr td��|jj� s0td��d}t|jdd� �}t|�dkrXtd��i }t|d d d� �}x4|D ],}|jj |d
�j
}|jj |d �j
}|||< qtW |S ) NZ EnvironmentzNo Environment NodezNo Child Nodesz)http://schemas.dmtf.org/ovf/environment/1c S s
| j dkS )NZPropertySection)� localName)�nr r r �<lambda>p s z get_properties.<locals>.<lambda>r zNo 'PropertySection'sc S s
| j dkS )NZProperty)rr )rs r r r rt x s �key�value) r Z parseStringZdocumentElementrr �XmlErrorro rq r% Z
attributesZgetNamedItemNSrv ) r- ZdomZenvNsURIZ propSectionsrE Z propElems�elemru rF r r r r> c s$
r> c @ s e Zd ZdS )rw N)r7 r8 r9 r r r r rw � s rw c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s rz c C s | si S t j| �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 rB � s rB )F)T)"�__doc__r@ Zloggingr
rW Zxml.domr Z cloudinitr r r r Z getLoggerr7 r' r
r r; r! r rZ r$ rg rh r# rq r> rD rw ZDEP_FILESYSTEMZ DEP_NETWORKry rz rB r r r r �<module> s.
i
)
&E