"T`s@sddlZddlZddlZddlZddlmZddlmZddlmZ ddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!m"Z"m#Z#m$Z$m%Z%e j&e'Z(dZ)dZ*dZ+Gddde j,Z-Gddde-Z.ddZ/dddddZ0dd dd!d"Z1dddd#d$Z2d%d&Z3d'd(Z4d)d*Z5d+d,Z6dd-d.Z7d/d0Z8d1d2Z9d3d4Z:d5d6Z;Gd7d8d8e<Z=e-e j>ffe.e j>e j?fffZ@d9d:ZAd;d<ZBd=d>ZCd?d@ZDdAdBZEdCdDZFdEdFZGdS)GN)minidom)dmi)log)safeyaml)sources)subp)util)Config)PreCustomScriptPostCustomScript) ConfigFile)NicConfigurator)PasswordConfigurator)GuestCustErrorEnum)GuestCustEventEnum)GuestCustStateEnum) enable_nicsget_nics_to_enableset_customization_statusget_tools_config set_gc_statusZ deployPkgzenable-custom-scriptsz/var/run/vmware-imcc@speZdZdZddZddZddZdd Zd d Zd d Z e ddZ dS) DataSourceOVFZOVFcCstjj||||d|_tjj|jd|_d|_i|_ d|_ d|_ d|_ d|_ d|_d|_dS)NZovf/file://TF)rr)r DataSource__init__seedospathjoinseed_dir environmentcfgsupported_seed_startsvmware_customization_supported_network_config_vmware_nics_to_enable_vmware_cust_conf_vmware_cust_found)selfsys_cfgdistropathsr-A/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceOVF.pyr:s        zDataSourceOVF.__init__cCs#tjj|}d||jfS)Nz %s [seed=%s])rr__str__r)r)rootr-r-r.r/GszDataSourceOVF.__str__c-fCs g}i}d}d}d}d}ddi}t|jj\}} tjd} | dkrptjd|rtjj |jj|} t | \}}} | |_ |j | n| rd| j krtjd|jstjdnd;} d }d}x4| D],}t||}|rtjd|PqW|rt|j}tjdtjdddtdd|f}n tjdd}|rDtjj|}t|}t||_tjd|yt|j\}}}WqQtk r@}z!td|tj||jWYdd}~XqQXn tjd|stj|jddrtjdd}t |o|}|rt!|jdtjdyCt"||\}}} }|r||_#n|j$j%|_#Wnt&j'k r@}z!td|t(j)||jWYdd}~XnAt*k r}z!td|tj||jWYdd}~XnXd|_+|j d tj,|t-t.j/t(j0t!|jd!n|rd|_1y,t!|jdt2|j\}}} t3||_1|jj4}t5|tjj |jj6d"}|oM| }|jj7}|r|rd#}|jj8rtjd$d%}t9t:t;|}|j d%krd&}tj|t-t.j<t(j=t>|tjj |jj?d'd(} WnAt*k rH}z!td)|tj||jWYdd}~XnX|r|ryt@||}!|!jAWnAt*k r}z!td*|tj||jWYdd}~XnXy2tjd+tB|jdd|j$jC|_#WnAt*k r+}z!td,|tjD||jWYdd}~XnX|rDtjd-tE}"|jjF}#yB|jjG}$|#so|$r|"jH|#|$|j$n tjd.WnAt*k r}z!td/|tj||jWYdd}~XnX|rDy tI||| }%|%jAWnAt*k rC}z!td0|tj||jWYdd}~XnX|ry&tJ|tjj |jj6d"WnAt*k r}z!td1|tj||jWYdd}~XnXd|_+|j d tj,tjj|tK|j1t-t.j/t(j0t!|jd!nvd2tLfd3tMfg}&d}'x$|&D]\}'}(|(} | r;Pq;W| rt | \}}} | |_ |j |'tN|d4krd5Sd6|kr_ |d6r_ |d6})d5}*x'|jOD]}+|)jP|+r|+}*PqW|*s tjd7|)|d5StjQ|)d8d\},}}tjd9|)tjR||,g}|j |)tjR||g}d:j ||_S||_T||_U||_V| |_WdS)Z seedfoundprotoZmd_seedr-r-r. _get_dataKs                         !                                    zDataSourceOVF._get_datacCs7tjdj}|dkr,d|jSd|jS)Nzsystem-product-namer2z vmware (%s)zovf (%s))rrArFr)r)rfr-r-r._get_subplatforms  zDataSourceOVF._get_subplatformcCs>d|jkrgS|jd}t|tr3|S|gSdS)Nz public-keys)ra isinstancelist)r)Zpksr-r-r.get_public_ssh_keyss  z!DataSourceOVF.get_public_ssh_keyscCs|jS)N)r")r)r-r-r.get_config_objszDataSourceOVF.get_config_objcCs|jS)N)r%)r)r-r-r.network_configszDataSourceOVF.network_configN) __name__ __module__ __qualname__Zdsnamerr/rnrorrrspropertyrtr-r-r-r.r6s   6  rc@seZdZddZdS)DataSourceOVFNetcCsGtj||||tjj|jd|_d|_d|_dS)Nzovf-nethttp://https://F)rzr{)rrrrrr r#r$)r)r*r+r,r-r-r.rs zDataSourceOVFNet.__init__N)rurvrwrr-r-r-r.rys ryc Csd}d}|}|s|Syt|j||}Wn%tk r_tjd||YnX|dkrtjd||||}|S)NZvmware_cust_file_max_waitzFailed to get '%s', using %srz/Invalid value '%s' for '%s', using '%s' instead)intget ValueErrorrBwarning)r"Zdefault_max_waitZmax_wait_cfg_optionrgr-r-r.rHs     rHcCsld}x_||krgtjj||}tjj|r@|Stjdtj|||7}q WdS)Nrz,Waiting for VMware Customization Config File)rrrisfilerBrCtimeZsleep)filenameZmaxwaitZnaplendirpathZwaitedZ fileFullPathr-r-r.rIs  rITFcCs:t|j|}|j||}t||j|jS)N)r nicsZgenerateget_network_configZ name_serversZ dns_suffixes)configZuse_system_devicesr[rZZnicConfiguratorZ nics_cfg_listr-r-r.rYs rYcCsA|}|s|r1|jddd|d|iddd|iS)NtypeZ nameserverZaddresssearchversionr)rE)rZ nameserversrZ config_listr-r-r.rs   rcCsyi}i}d}|jrL|jr?|jd|j|dzlocal-hostnamez public-keysz instance-idZhostnamez user-data)get_propertiesitemsbase64Z b64decodeencoderQ) repropsrcr"rdZ cfg_propsZmd_propsZpropvalr-r-r.rDs&         rDc Csd}xz|D]r}tjj||}tjj|r ytj|}||fSWq tk r~tjtd|Yq Xq WdS) N ovf-env.xml ovf_env.xml OVF_ENV.XML OVF-ENV.XMLzFailed loading ovf file %sF)rrrr)NF) rrrrr load_filerQZlogexcrB)rJZ env_namesfnameZfull_fnrer-r-r.r@s  r@cCs|s dSt|ts)td|tjj|}|jdr]|jdd }|jdr|jdd }n|j ddkrdS|sdSd}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/rrrz^(sr[0-9]+|hd[a-z]|xvd.*)ZCLOUD_INIT_CDROM_DEV_REGEXNr)rpstrrrrnormpathr` partitionsplitcountenvironr~recompilematch)ZdevnameZ default_regexZ devname_regexZcdmatchr-r-r.maybe_cdrom_devices"rc Cs6tj}xt|jD]f\}}|d}|dkrD|rDqt|sSq|d}t|\}}|dk r|SqW|rd}nd}ddtj|rd|ndD} xk| D]c}y"tj|td|\}}Wn(tjk rtj d |wYnX|dk r|SqWdS) NfstypeZiso9660Z mountpointFcSs"g|]}t|r|qSr-)r).0devr-r-r. Rs z%transport_iso9660..zTYPE=%smtypez%s not mountable as iso9660) rmountsrrr@Zfind_devs_withZmount_cbZMountFailedErrorrBrC) Z require_isorrinforZmpZ_fnamererZdevsr-r-r.r^=s2       # " r^cCsd}d}tj|s|S|dg}y6tj|\}}|rM|Stjd||WnZtjk r}z7|jdkrtjd||jtj|WYdd}~XnX|S)Nzvmware-rpctoolzinfo-get guestinfo.ovfEnvz%cmd %s exited 0 with empty stdout: %srz%s exited with code %d)rZwhichrBrCZProcessExecutionErrorZ exit_coder)ZrpctoolZ not_foundcmdoutZ_errrjr-r-r.r]bs  r]cCsGg}|js|Sx*|jD]}||r |j|q W|S)N) hasChildNodesZ childNodesrE)ZnodeZ filter_funcZretZchildr-r-r. find_childts  rc Cstj|}|jjdkr-td|jjsHtdd}t|jdd}t|dkrtdi}t|dd d}xH|D]@}|jj |d j }|jj |d j }|||sz get_properties..rzNo 'PropertySection'scSs |jdkS)NZProperty)r)rr-r-r.rskeyvalue) rZ parseStringZdocumentElementrXmlErrorrrr_Z attributesZgetNamedItemNSr) reZdomZenvNsURIZ propSectionsrZ propElemselemrrr-r-r.r~s$      rcCsX| s| rdSx?tj|D].\}}}||kr"tjj||Sq"WdS)N)rwalkrr)rrr0Z_dirsfilesr-r-r.rGs  rGc@seZdZdS)rN)rurvrwr-r-r-r.rs rcCstj|tS)N)rZlist_from_depends datasources)Zdependsr-r-r.get_datasource_listsrcCsA|s dStjj|d|d}tjj|r=dSdS)ag Check the existence of a marker file. Presence of marker file determines whether a certain code path is to be executed. It is needed for partial guest customization in VMware. @param markerid: is an unique string representing a particular product marker. @param: marker_dir: The directory in which markers exist. Fz .markerfile-z.txtT)rrrexists)markerid marker_dir markerfiler-r-r.rTs rTcCstjdtjj|d|d}xBtj|D]1}|jdr:tjtjj||q:Wt |dj dS)a" Create a new marker file. Marker files are unique to a full customization workflow in VMware environment. @param markerid: is an unique string representing a particular product marker. @param: marker_dir: The directory in which markers exist. zHandle marker creationz .markerfile-z.txtz .markerfilewN) rBrCrrrlistdirr`rZdel_fileopenclose)rrrrr-r-r.r\s  r\cCsStjd||ttj|t||tjtj j ||dS)z Raise error and send customization status to the underlying VMware Virtualization Platform. Also, cleanup the imc directory. z%s: %sN) rBrCrrrWrrrRrrrJ)prefixerrorZeventZ config_fileconfr-r-r.rMs rMcCstjd||i}d}d}ttj|}d|krP|d}|rqtj|jdd}||i|fS)z Load the cloud-init meta data, user data, cfg and network from the given files @return: 4-tuple of configuration metadata, userdata, cfg={}, network @raises: FileNotFoundError if md_path or ud_path are absent z+load meta data from: %s: user data from: %sNrk r1)rBrCsafeload_yaml_or_dictrrreplace)rhrircrdrkr-r-r.rPs    rPcCs|s iStj|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. )rload)r:r-r-r.rsrcCsd}d}d}|j}|rtjjt|}tjj|sXtd||j}|rtjjt|}tjj|std|n=tjjtd}tjj|stj d|d}|||fS)a\ collect all the other imc files. metadata is preferred to nics.txt configuration data. If metadata file exists because it is specified in customization configuration, then metadata is required and userdata is optional. @return a 3-tuple containing desired configuration file paths if present Expected returns: 1. user provided metadata and userdata (md_path, ud_path, None) 2. user provided metadata (md_path, None, None) 3. user-provided network config (None, None, nics_path) 4. No config found (None, None, None) Nzmeta data file is not found: %szuser data file is not found: %sznics.txtz%s does not exist.) Zmeta_data_namerrrVMWARE_IMC_DIRrrLZuser_data_namerBrC)Z cust_confrhriZ nics_pathZmd_fileZud_filer-r-r.rKs(    rK)HrrrrZxml.domrZ cloudinitrrZloggingrrrrZ+cloudinit.sources.helpers.vmware.imc.configr Z9cloudinit.sources.helpers.vmware.imc.config_custom_scriptr r Z0cloudinit.sources.helpers.vmware.imc.config_filer Z/cloudinit.sources.helpers.vmware.imc.config_nicr Z2cloudinit.sources.helpers.vmware.imc.config_passwdrZ4cloudinit.sources.helpers.vmware.imc.guestcust_errorrZ4cloudinit.sources.helpers.vmware.imc.guestcust_eventrrNZ4cloudinit.sources.helpers.vmware.imc.guestcust_staterZ3cloudinit.sources.helpers.vmware.imc.guestcust_utilrrrrrZ getLoggerrurBrUrVrrrryrHrIrYrrSrDr@rr^r]rrrGrQrZDEP_FILESYSTEMZ DEP_NETWORKrrrTr\rMrPrrKr-r-r-r. sf    (e      )%