File manager - Edit - /usr/lib/python3/dist-packages/cloudinit/sources/helpers/vmware/imc/__pycache__/guestcust_util.cpython-310.pyc
Back
o o@�d�O � @ 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 �e�Zd 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/� Z/d0d1� Z0d2d3� Z1 d[d6d7�Z2d8d9� Z3d:d;� Z4d<d=� Z5d>d?� Z6d@dA� Z7dBdC� Z8dDdE� Z9dFdG� Z:dHdI� Z;dJdK� Z<dLdM� Z=dNdO� Z>dPdQ� Z?dRdS� Z@dTdU� ZAdVdW� ZBdS )\� N)�safeyaml�subp�util� )�Config)�PostCustomScript�PreCustomScript)� ConfigFile)�NicConfigurator)�PasswordConfigurator)�GuestCustErrorEnum)�GuestCustEventEnum)�GuestCustStateEnumz/var/log/cloud-init.log�queryNicsSupported� connected�/var/run/vmware-imc� deployPkgzenable-custom-scriptsc C s� | sd S d}d}z"t �d| � tjd| gdgd�\}}|r&|�� }W ||fS W ||fS tyI } zt �d� t �|� W Y d }~||fS d }~ww )N� z Error sending the RPC commandzSending RPC command: %szvmware-rpctoolr )�rcszFailed to send RPC command)�logger�debugr �rstrip� Exception� exception)�rpc�out�err�e� r �U/usr/lib/python3/dist-packages/cloudinit/sources/helpers/vmware/imc/guestcust_util.py�send_rpc( s$ �� ��r c C s< d}|r t d | }nt }d| ||f }t|�\}}||fS )Nr �@z deployPkg.update.state %d %d %s)�CLOUDINIT_LOG_FILEr )� custstate� custerror� errormessage�messager r r r r r �set_customization_status> s r' c C sV | sdS d}t j�| �sdS t| d��}|�|�}W d � |S 1 s$w 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 �r)�os�path�exists�open�read)� nicsfilepath� NICS_SIZE�fp�nicsr r r �get_nics_to_enableK s ��r2 c C s� | s t �d� d S d}d}d}td|�D ]O}t �d|� ttjtj| �\}}|s0t � || � q|t kr<t �d� d S td|�D ]!}ttjtj | �\}}|r]|t kr]t �d|� d S t � |� qAqt �d|� 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 �warning�ranger r' r �GUESTCUST_STATE_RUNNINGr �GUESTCUST_EVENT_ENABLE_NICS�time�sleep�QUERY_NICS_SUPPORTED�GUESTCUST_EVENT_QUERY_NICS�NICS_STATUS_CONNECTED�info)r1 �enableNicsWaitRetries�enableNicsWaitCount�enableNicsWaitSeconds�attemptr �_err�countr r r � enable_nics` sD � � � �rD c C s� t �d�s t�d� |S ddd| |g}zt � |�}W n2 t jyL } z%|jdkr3t�dd�|�|� n t�d||j� t�|� |W Y d }~S d }~ww |}t � d |j �}|ri|� d �� � }t�d | ||� |S t�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 �ProcessExecutionError� exit_code�join�errorr �re�match�stdout�group�strip)�section�key� defaultVal�cmdr r �retValue�mr r r �get_tools_config� sB � � �� ��rZ c C s | r | j r d| }t|�S d S )Nz info-set guestinfo.gc.status %s)�post_gc_statusr )rE �gcMsgr r r r � set_gc_status� s r] c C s t S �N)� IMC_DIR_PATHr r r r �get_imc_dir_path� s r` c C s� i d d i f\}}}}t |d� t|�\}}t| |�} | r%t||||�s%dS t| |�s,dS zt�d� t|dd|j�|d< W n t yZ } zt dt | �t j |� W Y d } ~ dS d } ~ ww t|� ttjtj� t |d� ||||fS )N�Started)NNNNz#Preparing the Network configurationT�networkz%Error preparing Network Configuration� Successful)r] �)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 �$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�cfg�is_special_customizationr r r r �get_data_from_imc_cust_cfg� sB � ����� rz c C s� t | d� d\}}}| j}|r�tj�t� |�}tj�|�s)tdd| tj | � dS zt � |�}W n t yM } ztdt |�tj | � W Y d }~dS d }~ww z t�d� t�|�}W n tjyw } ztdt |�tj| � W Y d }~nd }~ww | j}|r�tj�t� |�}tj�|�s�tdd | tj | � dS z t � |��d d �}W n t y� } ztd t |�tj | � W Y d }~dS d }~ww ttjtj� t | d � |||fS )Nra �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 filerc )r] �meta_data_namer) r* rM r` r+ rj r � GUESTCUST_EVENT_CUSTOMIZE_FAILEDr � load_filer rk r r r �load� YAMLErrorr �!GUESTCUST_ERROR_WRONG_META_FORMAT�user_data_name�replacer' r rn ro ) rr ru rv rw �md_file�md_pathr �ud_file�ud_pathr r r �#get_data_from_imc_raw_data_cust_cfg� s� ���� ��� ���� � r� c C s^ i i }}| j r| jr| j d | j |d< n| j |d< | jr#| j|d< | jr+| j|d< ||fS )N�.zlocal-hostname�timezonez instance-id)� host_name� domain_namer� � instance_id)rr ru rx r r r rd 1 s � rd TFc C s( t | j|�}|�||�}t|| j| j�S r^ )r r1 �generate�get_v1_network_config� name_servers� dns_suffixes)rr �use_system_devices� configureri �nicConfigurator� nics_cfg_listr r r rh A s �rh c C s( | }|s|r|� d||d�� d|d�S )N� nameserver)�type�address�searchr )�versionrE )�append)r� � nameserversr� � config_listr r r r� L s � r� c C s: t j�| d�}t j�|�rt�d|� tt|�� d S d S )Nznics.txtz %s file found, to connect nics)r) r* rM r+ r r rD r2 )rs � nics_filer r r rm W s �rm c C s t �| dd� S )N� disable_vmware_customizationT�r �get_cfg_option_bool)�sys_cfgr r r �is_vmware_cust_enabled^ s �r� c C s t �| dd�S )N�allow_raw_dataTr� )�ds_cfgr r r �is_raw_data_cust_enabledd s r� c C sB t | �}tjtjdtd|fd�}|rt�d|� |S t�d� d S )N�3Waiting for VMware customization configuration filezcust.cfg)�logfunc�msg�func�argsz3Found VMware customization configuration file at %sz0No VMware customization configuration file found)�get_max_wait_from_cfgr �log_timer r �wait_for_cust_cfg_file)r� �max_wait�cust_cfg_file_pathr r r �get_cust_cfg_fileh s �� r� � r3 c C s^ d}||krd}||k r-t j�|| �}t j�|�r|S t�d� t�|� ||7 }||k s d S )Nr r r� )r) r* rM �isfiler r r8 r9 )�filename�maxwait�naplen�dirpath�waited� fileFullPathr r r r� � s �r� c C sj d}d}|}| s |S z t | �||��}W n ty$ t�d||� Y nw |dk r3t�d|||� |}|S )N� �vmware_cust_file_max_waitz Failed to get '%s', using %sr z/Invalid value '%s' for '%s', using '%s' instead)�intrF � ValueErrorr r4 )r� �default_max_wait�max_wait_cfg_optionr� r r r r� � s. ���r� c C s$ |j }t|tj�| d��}|o| S )N�data)� marker_id�check_marker_existsr) r* rM )rp rr �product_marker�has_marker_filer r r re � s � re c C s2 | sdS t j�|d| d �}t j�|�rdS 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* rM r+ )�markerid� marker_dir� markerfiler r r r� � s r� c C s |j }|r t| |�s dS dS )NFT)r� �create_marker_file)rp rr r� r r r rg � s rg c C sX zt |jtj�| d�� W dS ty+ } ztdt|�tj |� W Y d }~dS d }~ww )Nr� zError creating marker filesFT) �setup_marker_filesr� r) r* rM r rj rk r r~ )rp rr r r r r r� � s ����r� c C s` t �d� tj�|d| d �}t�|�D ]}|�d�r&t�tj�||�� qt |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* rM �listdir� startswithr �del_filer, �close)r� r� � marker_file�fnamer r r r� � s �r� c C st d}d}d}d}|j }|rt|�}|rt|||�}t||�}|r0|r0tj�| d�} t|||| �}|r8|o7|o7|S |S )NFz per-instance)�custom_script_name� check_custom_script_enablement�do_pre_custom_script�do_password_customizationr) r* rM �do_post_custom_script) rq rr rs rt �is_pre_custom_successful� is_password_custom_successful�is_post_custom_successful�is_custom_script_enabled� custom_script� ccScriptsDirr r r rf � s0 � ���rf c C sT z t ||�}|�� W dS ty) } ztdt|�tj| � W Y d }~dS d }~ww )Nz(Error executing pre-customization scriptFT)r �executer rj rk r r~ )rr r� rs �precustr r r r r� s ����r� c C sV z t |||�}|�� W dS ty* } ztdt|�tj| � W Y d }~dS d }~ww )Nz)Error executing post-customization scriptFT)r r� r rj rk r r~ )rr r� rs r� �postcustr r r r r� s ����r� c C sP d}d}| j rt�d� d}ttt|�}|�� dkr$tddtj | � |S d}|S )NF�falsez=Set default value to true due to customization configuration.�truez-Custom script is disabled by VM Administratorz'Error checking custom script enablementT) �default_run_post_scriptr r rZ �IMC_TOOLS_CONF_GROUPNAME�"IMC_TOOLS_CONF_ENABLE_CUST_SCRIPTS�lowerrj r � GUESTCUST_ERROR_SCRIPT_DISABLED)rr r� � default_value�custom_script_enablementr r r r� $ s, �� ��r� c C s� t �d� t� }| j}z| j}|s|r |�|||� W dS t �d� W dS tyA } ztdt|�t j | � W Y d }~dS d }~ww )Nz Applying password customizationz Changing password is not neededz%Error applying password configurationFT) r r r �admin_password�reset_passwordr� r rj rk r r~ )rr rt �pwdConfigurator� admin_pwd� reset_pwdr r r r r� = s( � ����r� c C s t t| ��S r^ )r r )�cfg_filer r r �parse_cust_cfgR s r� c C s: d\}}| j rd}t�d� ||fS d}t�d� ||fS )N)FFTz!raw cloudinit data cust cfg foundzvmware cust cfg found)r} r r )rr �is_vmware_cust_cfg�is_raw_data_cust_cfgr r r �get_cust_cfg_typeV s � r� c C s6 d} d}| D ]}t ||�}|rt�d|� dS qdS )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-toolszlibdeployPkgPlugin.soz$Found the customization plugin at %sTF)� search_filer r )� search_paths� cust_pluginr* �cust_plugin_pathr r r �is_cust_plugin_availablea s ��r� c C s@ | r|sd S t �| �D ]\}}}||v r t j�||� S q d S r^ )r) �walkr* rM )r� r� �root�_dirs�filesr r r r� u s �r� c C s* t �td| |� ttj|� t|| � dS )zS Set customization status to the underlying VMware Virtualization Platform z%s: %sN)r �logexcr r' r r6 r] )�prefixrN �eventrr r r r rj � s rj r^ )TFNr{ )r� r3 r )C�loggingr) rO r8 � cloudinitr r r rE r �config_custom_scriptr r � config_filer � config_nicr � config_passwdr �guestcust_errorr �guestcust_eventr �guestcust_stater � getLogger�__name__r r" r: r< r_ r� r� r r' r2 rD rZ r] r` rz r� rd rh r� rm r� r� r� r� r� re r� rg r� r� rf r� r� r� r� r� r� r� r� rj r r r r �<module> sn *2(B � �
| ver. 1.4 |
Github
|
.
| PHP 8.2.9 | Generation time: 0.02 |
proxy
|
phpinfo
|
Settings