관리-도구
편집 파일: guestcust_util.cpython-39.opt-1.pyc
a '�Dg�O � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZ ddl m Z ddl m Z m Z ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ e �e�Z d Z d Z dZ dZ dZ!dZ"dd� Z#dXdd�Z$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/� Z0d0d1� Z1d2d3� Z2d[d6d7�Z3d8d9� Z4d:d;� Z5d<d=� Z6d>d?� Z7d@dA� Z8dBdC� Z9dDdE� Z:dFdG� Z;dHdI� Z<dJdK� Z=dLdM� Z>dNdO� Z?dPdQ� Z@dRdS� ZAdTdU� ZBdVdW� ZCdS )\� N)� performance�subp�util� )�Config)�PostCustomScript�PreCustomScript)� 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}z4t �d| � tjd| gdgd�\}}|rB|�� }W n8 ty| } z t �d� t �|� W Y d }~n d }~0 0 ||fS )N� z Error sending the RPC commandzSending RPC command: %szvmware-rpctoolr )ZrcszFailed to send RPC command)�logger�debugr �rstrip� Exception� exception)�rpc�out�err�e� r �W/usr/lib/python3.9/site-packages/cloudinit/sources/helpers/vmware/imc/guestcust_util.py�send_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 errormessage�messager r r r r r �set_customization_status@ s r c C s� | sdS d}t j�| �s dS t�d| � ���D t| d��}|�|�}W d � n1 sX0 Y W d � n1 sv0 Y |S )z�Reads the NICS from the specified file path and returns the content @param nicsfilepath: Absolute file path to the NICS.txt file. Ni zReading �r)�os�path�existsr �Timed�open�read)Z nicsfilepathZ NICS_SIZE�fp�nicsr r r �get_nics_to_enableM s �Fr* c C s� | st �d� d S d}d}d}t|�D ]�}t �d|� ttjtj| �\}}|s^t � || � q&|t krvt �d� d S t|�D ]B}ttjtj | �\}}|r�|t kr�t �d|� d S t � |� q~q&t �d|� d S )Nz No Nics found� 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 �warning�ranger r r �GUESTCUST_STATE_RUNNINGr ZGUESTCUST_EVENT_ENABLE_NICS�time�sleep�QUERY_NICS_SUPPORTEDZGUESTCUST_EVENT_QUERY_NICS�NICS_STATUS_CONNECTED�info)r) ZenableNicsWaitRetriesZenableNicsWaitCountZenableNicsWaitSecondsZattemptr Z_err�countr r r � enable_nicsd sB � � �r5 c C s� t �d�st�d� |S ddd| |g}zt � |�}W nd t jy� } zJ|jdkrft�dd�|�|� nt�d||j� t�|� |W Y d }~S d }~0 0 |}t � d |j �}|r�|� d �� � }t�d | ||� nt�d | ||� |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 value�config�get�E 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<Tools config: [%s] %s is not found, return default value: %s)r �whichr r ZProcessExecutionErrorZ exit_code�join�errorr �re�match�stdout�group�strip)�section�keyZ defaultVal�cmdr r ZretValue�mr r r �get_tools_config� s< � � �rG c C s | r| j rd| }t|�S d S )Nz info-set guestinfo.gc.status %s)Zpost_gc_statusr )r6 ZgcMsgr r r r � set_gc_status� s rH c C s t S �N)� IMC_DIR_PATHr r r r �get_imc_dir_path� s rK c C s� i d d i f\}}}}t |d� t|�\}}t| |�} | rJt||||�sJdS t| |�sXdS z"t�d� t|dd|j�|d< W n: t y� } z"t dt | �t j |� W Y d } ~ dS d } ~ 0 0 t|� ttjtj� t |d� ||||fS )N�Started)NNNNz#Preparing the Network configurationT�networkz%Error preparing Network Configuration� Successful)rH �)get_non_network_data_from_vmware_cust_cfg� check_markers�do_special_customization�recheck_markersr r �%get_network_data_from_vmware_cust_cfg�osfamilyr �set_cust_error_status�strr Z$GUESTCUST_EVENT_NETWORK_SETUP_FAILED� connect_nicsr r �GUESTCUST_STATE_DONEr �GUESTCUST_ERROR_SUCCESS) � cloud_dir� scripts_cpath�cust_cfg� cust_cfg_dir�distro�md�ud�vd�cfgZis_special_customizationr r r r �get_data_from_imc_cust_cfg� s> � � �� rc c C s� t | d� d\}}}| j}|�r�tj�t� |�}tj�|�sTtdd| tj | � dS zt � |�}W n: t y� } z"tdt |�tj | � W Y d }~dS d }~0 0 zt�d� t�|�}W n: tjy� } z tdt |�tj| � W Y d }~n d }~0 0 | j}|�r�tj�t� |�}tj�|��s4tdd | tj | � dS zt � |��d d �}W n< t �y� } z"td t |�tj | � W Y d }~dS d }~0 0 ttjtj� t | d � |||fS )NrL )NNNz,Error locating the cloud-init meta data filez Meta data file is not found: %sz'Error loading cloud-init meta data filez'Validating if meta data is valid or notz&Error parsing the cloud-init meta dataz+Error locating the cloud-init userdata filez Userdata file is not found: %s� r z&Error loading cloud-init userdata filerN )rH �meta_data_namer" r# r< rK r$ rU r � GUESTCUST_EVENT_CUSTOMIZE_FAILEDr Zload_text_filer rV r r �yamlZ safe_loadZ YAMLErrorr Z!GUESTCUST_ERROR_WRONG_META_FORMATZuser_data_name�replacer r rX rY ) r\ r_ r` ra Zmd_fileZmd_pathr Zud_fileZud_pathr r r �#get_data_from_imc_raw_data_cust_cfg� sx �� ���� ri c C s^ i i }}| j r6| jr,| j d | j |d< n | j |d< | jrF| j|d< | jrV| j|d< ||fS )N�.zlocal-hostname�timezonez instance-id)Z host_nameZ domain_namerk Z instance_id)r\ r_ rb r r r rO 5 s � rO TFc C s( t | j|�}|�||�}t|| j| j�S rI )r r) Zgenerate�get_v1_network_configZ name_serversZ dns_suffixes)r\ Zuse_system_devices� configurerT ZnicConfigurator� nics_cfg_listr r r rS E s �rS c C s( | }|s |r |� d||d�� d|d�S )NZ nameserver)�type�address�searchr )�versionr6 )�append)rn Z nameserversrq Z config_listr r r rl P s �rl c C s6 t j�| d�}t j�|�r2t�d|� tt|�� d S )Nznics.txtz %s file found, to connect nics)r" r# r< r$ r r r5 r* )r] Z nics_filer r r rW [ s rW c C s t �| dd� S )NZ disable_vmware_customizationT�r Zget_cfg_option_bool)Zsys_cfgr r r �is_vmware_cust_enabledb s �ru c C s t �| dd�S )NZallow_raw_dataTrt )�ds_cfgr r r �is_raw_data_cust_enabledh s rw c C s^ t | �}t�d�� td|�}W d � n1 s20 Y |rPt�d|� |S t�d� d S )N�3Waiting for VMware customization configuration filezcust.cfgz3Found VMware customization configuration file at %sz0No VMware customization configuration file found)�get_max_wait_from_cfgr r% �wait_for_cust_cfg_filer r )rv �max_waitZcust_cfg_file_pathr r r �get_cust_cfg_filel s �(� r| � r+ c C sX d}||krd}||k rTt j�|| �}t j�|�r6|S t�d� t�|� ||7 }qd S )Nr r rx )r" r# r< �isfiler r r/ r0 )�filenameZmaxwaitZnaplen�dirpathZwaitedZ fileFullPathr r r rz � s rz c C sj d}d}|}| s|S zt | �||��}W n tyH t�d||� Y n0 |dk rft�d|||� |}|S )N� Zvmware_cust_file_max_waitz Failed to get '%s', using %sr z/Invalid value '%s' for '%s', using '%s' instead)�intr7 � ValueErrorr r, )rv Zdefault_max_waitZmax_wait_cfg_optionr{ r r r ry � s, � �ry c C s$ |j }t|tj�| d��}|o"| S )N�data)� marker_id�check_marker_existsr" r# r< )rZ r\ �product_markerZhas_marker_filer r r rP � s �rP c C s2 | sdS t j�|d| d �}t j�|�r.dS dS )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. F� .markerfile-�.txtT)r" r# r< r$ )Zmarkerid� marker_dirZ markerfiler r r r� � s r� c C s |j }|rt| |�sdS dS )NFT)r� �create_marker_file)rZ r\ r� r r r rR � s rR c C sZ zt |jtj�| d�� W n: tyT } z"tdt|�tj |� W Y d }~dS d }~0 0 dS )Nr� zError creating marker filesFT) �setup_marker_filesr� r" r# r< r rU rV r rf )rZ r\ r r r r r� � s �r� c C s` t �d� tj�|d| d �}t�|�D ]"}|�d�r*t�tj�||�� q*t |d�� � dS )a# Create a new marker file. Marker files are unique to a full customization workflow in VMware environment. @param marker_id: is an unique string representing a particular product marker. @param: marker_dir: The directory in which markers exist. zHandle marker creationr� r� z .markerfile�wN) r r r" r# r< �listdir� startswithr Zdel_filer& �close)r� r� Z marker_file�fnamer r r r� � s r� c C st d}d}d}d}|j }|r2t|�}|r2t|||�}t||�}|r`|r`tj�| d�} t|||| �}|rp|on|on|S |S )NFz per-instance)Zcustom_script_name� check_custom_script_enablement�do_pre_custom_script�do_password_customizationr" r# r< �do_post_custom_script) r[ r\ r] r^ Zis_pre_custom_successfulZ is_password_custom_successfulZis_post_custom_successful�is_custom_script_enabled� custom_script� ccScriptsDirr r r rQ � s0 � ���rQ c C sV zt ||�}|�� W n: tyP } z"tdt|�tj| � W Y d }~dS d }~0 0 dS )Nz(Error executing pre-customization scriptFT)r �executer rU rV r rf )r\ r� r] Zprecustr r r r r� s �r� c C sX zt |||�}|�� W n: tyR } z"tdt|�tj| � W Y d }~dS d }~0 0 dS )Nz)Error executing post-customization scriptFT)r r� r rU rV r rf )r\ r� r] r� Zpostcustr r r r r� s �r� c C sN d}d}| j r t�d� d}ttt|�}|�� dkrFtddtj | � nd}|S )NFZfalsez=Set default value to true due to customization configuration.�truez-Custom script is disabled by VM Administratorz'Error checking custom script enablementT) Zdefault_run_post_scriptr r rG �IMC_TOOLS_CONF_GROUPNAME�"IMC_TOOLS_CONF_ENABLE_CUST_SCRIPTS�lowerrU r Z GUESTCUST_ERROR_SCRIPT_DISABLED)r\ r� Z default_valueZcustom_script_enablementr r r r� & s* �� �r� c C s� t �d� t� }| j}z,| j}|s&|r6|�|||� n t �d� W n: ty| } z"tdt|�t j | � W Y d }~dS d }~0 0 dS )Nz Applying password customizationz Changing password is not neededz%Error applying password configurationFT) r r r Zadmin_passwordZreset_passwordrm r rU rV r rf )r\ r^ ZpwdConfiguratorZ admin_pwdZ reset_pwdr r r r r� ? s" �r� c C s t t| ��S rI )r r )Zcfg_filer r r �parse_cust_cfgT s r� c C s4 d\}}| j r d}t�d� nd}t�d� ||fS )N)FFTz!raw cloudinit data cust cfg foundzvmware cust cfg found)re r r )r\ Zis_vmware_cust_cfgZis_raw_data_cust_cfgr r r �get_cust_cfg_typeX s r� c C s6 d} d}| D ]$}t ||�}|r t�d|� dS q dS )N)z/usr/lib/vmware-toolsz/usr/lib64/vmware-toolsz/usr/lib/open-vm-toolsz/usr/lib64/open-vm-toolsz'/usr/lib/x86_64-linux-gnu/open-vm-toolsz(/usr/lib/aarch64-linux-gnu/open-vm-toolsz%/usr/lib/i386-linux-gnu/open-vm-toolszlibdeployPkgPlugin.soz$Found the customization plugin at %sTF)� search_filer r )Z search_pathsZ cust_pluginr# Zcust_plugin_pathr r r �is_cust_plugin_availablec s �r� c C s@ | r|s d S t �| �D ]$\}}}||v rt j�||� S qd S rI )r" �walkr# r< )r� r �rootZ_dirs�filesr r r r� x s r� c C s* t �td| |� ttj|� t|| � dS )zS Set customization status to the underlying VMware Virtualization Platform z%s: %sN)r Zlogexcr r r r. rH )�prefixr= Zeventr\ r r r rU � s rU )N)TFN)NNN)r} r+ r )DZloggingr" r> r/ rg Z cloudinitr r r r6 r Zconfig_custom_scriptr r Z config_filer Z config_nicr Z config_passwdr Zguestcust_errorr Zguestcust_eventr Zguestcust_stater Z getLogger�__name__r r r1 r2 rJ r� r� r r r* r5 rG rH rK rc ri rO rS rl rW ru rw r| rz ry rP r� rR r� r� rQ r� r� r� r� r� r� r� r� rU r r r r �<module> sn *2(B � �