3
\me2 @ s d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m
Z
m Z ej e
ZG dd de jZG dd deZd#d
d Zd d
Zdd Zd$ddZdd Zdd Zdd Zdd Zdd ZG d d d eZ 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)safeyamlsourcessubputilc @ 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__seedospathjoinseed_dir environmentcfgsupported_seed_starts_network_config)selfsys_cfgdistropaths 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 |rt|d\}}} || _d|kr|d r|d | _ |j| t |dkrd 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 Fseedfromz Seed from %s not supported by %s)Ztimeoutz Using seeded cache data from %s,) get_ovf_envr r r
r r read_ovf_environmentr appendtransport_vmware_guestinfotransport_iso9660r lenr
startswithLOGdebugr Z read_seededZ
mergemanydictr metadataZ userdata_rawZvendordata_rawr )r foundmdudZvdZdefaultsZseedfilecontentsr r ZnpnameZ transfuncr Z seedfoundprotoZmd_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|tr&|S |gS d S )Nz public-keys)r)
isinstancelist)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-nethttp://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|ry$tj|
j } t| jd|| < W q tk
r tj d Y qX 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_propertiesitemsbase64Z b64decodeencodesafeload_yaml_or_dictget Exceptionr' r( ) r- Z read_networkpropsr+ r r, Z cfg_propsZmd_propsZ
network_propsZpropvalr6 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 | ts td| tjj| } | jdrB| jdd
} | jdr\| jdd } n| j ddkrndS | svdS d}tj
j d|}t j
|}|j| d k S ) zTest 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_REGEXNrO )r2 str
ValueErrorr
r normpathr& partitionsplitcountenvironrC recompilematch)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|rd| nd D } xV| D ]N}yt j|t|d\}}W n$ t jk
r tj d | wY nX |dk r|S qW d S )
NfstypeZiso9660Z
mountpointFc S s g | ]}t |r|qS r )rZ ).0devr r r
s z%transport_iso9660..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 )rpctoolargcmdstdoutstderrr 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| |krtjtd| d S |stj d d S ytj d
||tS t jk
r } ztjtd| W Y d d }~X nX W Y d d }~X nX d S ) Nvmtoolsdzvmware-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..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_funcretZchildr r r
find_childY s rq c C s t j| }|jjdkr td|jj s0tdd}t|jdd }t|dkrXtdi }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 p s z get_properties..r zNo 'PropertySection'sc S s
| j dkS )NZProperty)rr )rs r r r rt x s keyvalue) r Z parseStringZdocumentElementrr XmlErrorro rq r% Z
attributesZgetNamedItemNSrv ) r- ZdomZenvNsURIZ propSectionsrE Z propElemselemru 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| tS )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 s.
i
)
&E