File manager - Edit - /usr/lib/python3.6/site-packages/cloudinit/sources/helpers/vmware/imc/__pycache__/guestcust_util.cpython-36.pyc
Back
3 \�me�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 je�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� Z1d[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.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)�logger�debugr �rstrip� Exception� exception)�rpc�out�err�e� r �$/usr/lib/python3.6/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 jj| �s dS t| d��}|j|�}W dQ R X |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)Z nicsfilepathZ NICS_SIZE�fp�nicsr r r �get_nics_to_enableK s r* c C s� | st jd� d S d}d}d}x�td|�D ]�}t jd|� ttjtj| �\}}|sbt j || � q*|t krxt jd� d S xLtd|�D ]>}ttjtj | �\}}|r�|t kr�t jd|� d S t j |� q�W 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 �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_nics` s< r5 c C s� t jd�stjd� |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 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 Zwhichr r ZProcessExecutionErrorZ exit_code�join�errorr �re�match�stdout�group�strip)Zsection�keyZ defaultVal�cmdr r ZretValue�mr r r �get_tools_config� s6 rE 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 rF c C s t S )N)� IMC_DIR_PATHr r r r �get_imc_dir_path� s rH c C s� i d d i f\}}}}t |d� t|�\}}t| |�} | rJt||||�sJdS t| |�sXdS y"tjd� t|dd|j�|d< W n4 t k r� } zt dt | �t j |� d S d } ~ X nX t|� ttjtj� t |d� ||||fS ) N�Startedz#Preparing the Network configurationTZnetworkz%Error preparing Network Configuration� Successful)NNNN)NNNN)NNNN)rF �)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� s6 r_ c - C s� t | d� d \}}}| j}|�r|tjjt� |�}tjj|�sTtdd| tj | � dS yt j |�}W n4 t k r� } ztdt |�tj | � dS d }~X nX ytjd� tj|�}W n: tjk � r� } ztdt |�tj| � W Y d d }~X nX | j}|�r|tjjt� |�}tjj|��s.tdd| tj | � dS yt j |�jd d �}W n6 t k �rz } ztd t |�tj | � dS d }~X nX ttjtj� t | d � |||fS )NrI z,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 filerJ )NNN)NNN)NNN)NNN)NNN)rF �meta_data_namer# r$ r; rH r% rQ r � GUESTCUST_EVENT_CUSTOMIZE_FAILEDr Z load_filer rR r r r �loadZ YAMLErrorr Z!GUESTCUST_ERROR_WRONG_META_FORMATZuser_data_name�replacer! r rT rU ) rX r[ r\ r] Zmd_fileZmd_pathr Zud_fileZud_pathr r r �#get_data_from_imc_raw_data_cust_cfg� sl re 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_namerg Z instance_id)rX r[ r^ r r r rK 1 s rK TFc C s( t | j|�}|j||�}t|| j| j�S )N)r r) Zgenerate�get_v1_network_configZ name_serversZ dns_suffixes)rX Zuse_system_devices� configurerP ZnicConfigurator� nics_cfg_listr r r rO A s rO c C s( | }|s |r |j d||d�� d|d�S )NZ nameserver)�typeZaddress�searchr )�versionr6 )�append)rj Z nameserversrl Z config_listr r r rh L s rh c C s6 t jj| d�}t jj|�r2tjd|� tt|�� d S )Nznics.txtz %s file found, to connect nics)r# r$ r; r% r r r5 r* )rY Z nics_filer r r rS W s rS c C s t j| dd� S )NZ disable_vmware_customizationT)r �get_cfg_option_bool)Zsys_cfgr r r �is_vmware_cust_enabled^ s rp c C s t j| dd�S )NZallow_raw_dataT)r ro )�ds_cfgr r r �is_raw_data_cust_enabledd s rr c C sB t | �}tjtjdtd|fd�}|r4tjd|� |S tjd� d S )Nz3Waiting for VMware customization configuration filezcust.cfg)Zlogfunc�msg�func�argsz3Found VMware customization configuration file at %sz0No VMware customization configuration file found)�get_max_wait_from_cfgr Zlog_timer r �wait_for_cust_cfg_file)rq �max_waitZcust_cfg_file_pathr r r �get_cust_cfg_fileh s ry � r+ c C s\ d}||krd}xF||k rVt jj|| �}t jj|�r8|S tjd� tj|� ||7 }qW d S )Nr r z3Waiting for VMware customization configuration file)r# r$ r; �isfiler r r/ r0 )�filenameZmaxwaitZnaplen�dirpathZwaitedZ fileFullPathr r r rw � s rw c C sl d}d}|}| s|S yt | j||��}W n" tk rJ tjd||� Y nX |dk rhtjd|||� |}|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, )rq Zdefault_max_waitZmax_wait_cfg_optionrx r r r rv � s( rv c C s$ |j }t|tjj| d��}|o"| S )N�data)� marker_id�check_marker_existsr# r$ r; )rV rX �product_markerZhas_marker_filer r r rL � s rL c C s2 | sdS t jj|d| d �}t jj|�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. Fz .markerfile-z.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)rV rX r� r r r rN � s rN c C sT yt |jtjj| d�� W n4 tk rN } ztdt|�tj |� dS d }~X nX dS )Nr� zError creating marker filesFT) �setup_marker_filesr� r# r$ r; r rQ rR r rb )rV rX r r r r r� � s r� c C sd t jd� tjj|d| d �}x0tj|�D ]"}|jd�r,tjtjj||�� q,W t |d�j � 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 creationz .markerfile-z.txtz .markerfile�wN) r r r# r$ r; �listdir� startswithr Zdel_filer& �close)r� r� Z marker_fileZfnamer r r r� � s r� c C st d}d}d}d}|j }|r2t|�}|r2t|||�}t||�}|r`|r`tjj| 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) rW rX rY rZ Zis_pre_custom_successfulZ is_password_custom_successfulZis_post_custom_successful�is_custom_script_enabled� custom_script� ccScriptsDirr r r rM � s( rM c C sP yt ||�}|j� W n4 tk rJ } ztdt|�tj| � dS d }~X nX dS )Nz(Error executing pre-customization scriptFT)r �executer rQ rR r rb )rX r� rY Zprecustr r r r r� s r� c C sR yt |||�}|j� W n4 tk rL } ztdt|�tj| � dS d }~X nX dS )Nz)Error executing post-customization scriptFT)r r� r rQ rR r rb )rX r� rY r� Zpostcustr r r r r� s r� c C sN d}d}| j r tjd� d}ttt|�}|j� 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 rE �IMC_TOOLS_CONF_GROUPNAME�"IMC_TOOLS_CONF_ENABLE_CUST_SCRIPTS�lowerrQ r Z GUESTCUST_ERROR_SCRIPT_DISABLED)rX r� Z default_valueZcustom_script_enablementr r r r� $ s$ r� c C s| t jd� t� }| j}y,| j}|s&|r6|j|||� n t jd� W n4 tk rv } ztdt|�t j | � dS d }~X nX dS )Nz Applying password customizationz Changing password is not neededz%Error applying password configurationFT) r r r Zadmin_passwordZreset_passwordri r rQ rR r rb )rX rZ ZpwdConfiguratorZ admin_pwdZ reset_pwdr r r r r� = s r� c C s t t| ��S )N)r r )Zcfg_filer r r �parse_cust_cfgR s r� c C s4 d\}}| j r d}tjd� nd}tjd� ||fS )NFTz!raw cloudinit data cust cfg foundzvmware cust cfg found)FF)ra r r )rX Zis_vmware_cust_cfgZis_raw_data_cust_cfgr r r �get_cust_cfg_typeV s r� c C s8 d } d}x*| D ]"}t ||�}|rtjd |� d S qW d S ) N�/usr/lib/vmware-tools�/usr/lib64/vmware-tools�/usr/lib/open-vm-tools�/usr/lib64/open-vm-tools�'/usr/lib/x86_64-linux-gnu/open-vm-tools�(/usr/lib/aarch64-linux-gnu/open-vm-tools�%/usr/lib/i386-linux-gnu/open-vm-toolszlibdeployPkgPlugin.soz$Found the customization plugin at %sTF)r� r� r� r� r� r� r� )� search_filer r )Z search_pathsZ cust_pluginr$ Zcust_plugin_pathr r r �is_cust_plugin_availablea s r� c C sD | s | rd S x.t j| �D ] \}}}||kr t jj||�S q W d S )N)r# �walkr$ r; )r} r| �rootZ_dirs�filesr r r r� v s r� c C s* t jtd| |� ttj|� t|| � dS )zS Set customization status to the underlying VMware Virtualization Platform z%s: %sN)r Zlogexcr r! r r. rF )�prefixr< ZeventrX r r r rQ � s rQ )N)TFN)NNN)rz r+ r )CZloggingr# r= r/ 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 rG r� r� r r! r* r5 rE rF rH r_ re rK rO rh rS rp rr ry rw rv rL r� rN r� r� rM r� r� r� r� r� r� r� r� rQ r r r r �<module> sh *2(B
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings