3
\meO @ s d dl Z d dlZd dlZd dlZd dlmZmZmZ ddlm Z ddl
m Z m Z ddl
mZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ e jeZd Z d
Z dZ dZ dZ dZ!dd Z"dXddZ#dd Z$dd Z%dd Z&d d Z'd d Z(d d! Z)d"d# Z*d$d% Z+dYd(d)Z,dZd*d+Z-d,d- Z.d.d/ Z/d0d1 Z0d2d3 Z1d[d6d7Z2d8d9 Z3d:d; Z4dd? Z6d@dA Z7dBdC Z8dDdE Z9dFdG Z:dHdI Z;dJdK ZdPdQ Z?dRdS Z@dTdU ZAdVdW ZBdS )\ N)safeyamlsubputil )Config)PostCustomScriptPreCustomScript)
ConfigFile)NicConfigurator)PasswordConfigurator)GuestCustErrorEnum)GuestCustEventEnum)GuestCustStateEnumz/var/log/cloud-init.logZqueryNicsSupportedZ connected/var/run/vmware-imcZ deployPkgzenable-custom-scriptsc C s | sd S d}d}y4t jd| tjd| gdgd\}}|rB|j }W n6 tk
rz } zt jd t j| W Y d d }~X nX ||fS )N z Error sending the RPC commandzSending RPC command: %szvmware-rpctoolr )rcszFailed to send RPC command)loggerdebugr rstrip Exception exception)rpcouterre r $/usr/lib/python3.6/guestcust_util.pysend_rpc( s
r c C s< d}|rt d | }nt }d| ||f }t|\}}||fS )Nr @z deployPkg.update.state %d %d %s)CLOUDINIT_LOG_FILEr )Z custstateZ custerrorZ errormessagemessager r r r r r set_customization_status> s r! c C s@ | sdS d}t jj| s dS t| d}|j|}W dQ R X |S )zReads the NICS from the specified file path and returns the content
@param nicsfilepath: Absolute file path to the NICS.txt file.
Ni r)ospathexistsopenread)Z nicsfilepathZ NICS_SIZEfpnicsr r r get_nics_to_enableK s r* c C s | st jd d S d}d}d}xtd|D ]}t jd| ttjtj| \}}|sbt j
|| q*|t krxt jd d S xLtd|D ]>}ttjtj | \}}|r|t
krt jd| d S t j
| qW q*W t jd| d S ) Nz
No Nics found r r z(Trying to connect interfaces, attempt %dz-NICS connection status query is not supportedz NICS are connected on %d secondz2Can't connect network interfaces after %d attempts)r warningranger r! r GUESTCUST_STATE_RUNNINGr
ZGUESTCUST_EVENT_ENABLE_NICStimesleepQUERY_NICS_SUPPORTEDZGUESTCUST_EVENT_QUERY_NICSNICS_STATUS_CONNECTEDinfo)r) ZenableNicsWaitRetriesZenableNicsWaitCountZenableNicsWaitSecondsZattemptr Z_errcountr r r enable_nics` s<
r5 c C s t jdstjd |S ddd| |g}yt j |}W n\ t jk
r } z>|jdkrhtjddj|| ntjd||j tj| |S d }~X nX |}t j
d
|j }|r|j d j
}tjd | || ntjd
| || |S )a$ Return the value of [section] key from VMTools configuration.
@param section: String of section to read from VMTools config
@returns: String value from key in [section] or defaultVal if
[section] is not present or vmware-toolbox-cmd is
not installed.
zvmware-toolbox-cmdz9vmware-toolbox-cmd not installed, returning default valueconfiggetE zRvmware-toolbox-cmd returned 69 (unavailable) for cmd: %s. Return default value: %s zFailed running %s[%s]Nz ([^=]+)=(.*) z Get tools config: [%s] %s = %sz sh
*2(B