Your IP : 216.73.216.31
U
�Ad�A � @ s d dl Z d dlZd dlmZmZmZ d dlmZ d dlmZ d dl m
Z d dl m Z m
Z
mZ d dlmZmZmZmZ e �e�ZG dd � d e�ZG d
d � d �ZG d d
� d
�ZG dd� d�Zdd� Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� de j!�Z"G d d � d e�Z#dS ) � N)�
NoOptionError�NoSectionError�RawConfigParser)�StringIO)�time)�log)� persistence�
type_utils�util)� CFG_ENV_NAME�
PER_ALWAYS� PER_INSTANCE�PER_ONCEc @ s e Zd ZdS )� LockFailureN��__name__�
__module__� __qualname__� r r �3/usr/lib/python3/dist-packages/cloudinit/helpers.pyr s r c @ s e Zd ZdS )� DummyLockNr r r r r r s r c @ s< e Zd Zdd� Zejd
dd��Zdd� Zdd � Zd
d � Z d S )�DummySemaphoresc C s d S �Nr ��selfr r r �__init__! s zDummySemaphores.__init__Fc c s t � V d S r )r )r �_name�_freqZ_clear_on_failr r r �lock$ s zDummySemaphores.lockc C s dS )NFr �r r r r r r �has_run( s zDummySemaphores.has_runc C s dS )NTr r r r r �clear+ s zDummySemaphores.clearc C s d S r r r r r r � clear_all. s zDummySemaphores.clear_allN)F)
r r r r �
contextlib�contextmanagerr r r! r" r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )�FileLockc C s
|| _ d S r )�fn)r r& r r r r 3 s zFileLock.__init__c C s dt �| �| jf S )Nz<%s using file %r>)r Zobj_namer& r r r r �__str__6 s zFileLock.__str__N)r r r r r' r r r r r% 2 s r% c C s | � dd�S )N�-�_)�replace)�namer r r �canon_sem_name: s r, c @ sL e Zd Zdd� Zejddd��Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dS )�FileSemaphoresc C s
|| _ d S r )�sem_path)r r. r r r r ? s zFileSemaphores.__init__Fc c sF t |�}z| �||�V W n& tk
r@ |r:| �||� � Y nX d S r )r, �_acquire� Exceptionr! )r r+ �freq�
clear_on_failr r r r B s zFileSemaphores.lockc C sP t |�}| �||�}zt�|� W n( ttfk
rJ t�td|� Y dS X dS )Nz Failed deleting semaphore %sFT)r, � _get_pathr
Zdel_file�IOError�OSError�logexc�LOG)r r+ r1 �sem_filer r r r! L s zFileSemaphores.clearc C s> zt �| j� W n( ttfk
r8 t �td| j� Y nX d S )Nz&Failed deleting semaphore directory %s)r
Zdel_dirr. r4 r5 r6 r7 r r r r r" V s �zFileSemaphores.clear_allc C sp | � ||�rd S | �||�}dt�� t� f }zt�||� W n( ttfk
rf t� t
d|� Y d S X t |�S )Nz%s: %s
z Failed writing semaphore file %s) r r3 �os�getpidr r
Z
write_filer4 r5 r6 r7 r% )r r+ r1 r8 �contentsr r r r/ ^ s zFileSemaphores._acquirec C sf |r |t krdS t|�}| �||�}tj�|�r4dS ||krbtj�| �||��rbt�d||� dS dS )NFTz�%s has run without canonicalized name [%s].
likely the migrator has not yet run. It will run next boot.
run manually with: cloud-init single --name=migrator)r r, r3 r9 �path�existsr7 �warning)r r+ r1 Zcnamer8 r r r r n s �zFileSemaphores.has_runc C s: | j }|r|tkr tj�||�S tj�|d||f �S d S )Nz%s.%s)r. r
r9 r< �join)r r+ r1 r. r r r r3 � s zFileSemaphores._get_pathN)F) r r r r r# r$ r r! r" r/ r r3 r r r r r- > s
r- c @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�Runnersc C s || _ i | _d S r )�paths�sems)r rA r r r r � s zRunners.__init__c C sh |t ks |sd S d }|tkr*| j�d�}n|tkr>| j�d�}|sFd S || jkr^t|�| j|< | j| S )N�sem)r r
rA � get_ipathr � get_cpathrB r- )r r1 r. r r r �_get_sem� s
zRunners._get_semNFc
C s� | � |�}|st� }|s g }|�||�r:t�d||� dS |�|||��X}|s\td| ��n@t�d||� t|t�r�|f |�}n||� }d|fW 5 Q R � S W 5 Q R X d S )Nz%s already ran (freq=%s))FNz Failed to acquire lock for %szRunning %s using lock (%s)T) rF r r r7 �debugr r �
isinstance�dict) r r+ Zfunctor�argsr1 r2 rC ZlkZresultsr r r �run� s
z Runners.run)NF)r r r r rF rK r r r r r@ � s r@ c @ sH e Zd Zddd�Zdd� Zdd� Zd d
� Zed �d d
�Ze dd� �Z
dS )� ConfigMergerNTc C s( || _ || _|| _|| _|| _d | _d S r )�_paths�_ds�_fns� _base_cfg�_include_vendor�_cfg)r rA �
datasourceZadditional_fnsZbase_cfgZinclude_vendorr r r r � s zConfigMerger.__init__c C sZ g }| j rVz&| j �� }|r.t|t�r.|�|� W n$ tk
rT t�td| j � Y nX |S )Nz2Failed loading of datasource config object from %s) rN Zget_config_objrH rI �appendr0 r
r6 r7 )r Zd_cfgsZds_cfgr r r �_get_datasource_configs� s
�
z$ConfigMerger._get_datasource_configsc C sT g }t tjkrPtjt }z|�t�|�� W n" tk
rN t�td|� Y nX |S )Nz%Failed loading of env. config from %s) r r9 �environrT r
� read_confr0 r6 r7 )r Ze_cfgsZe_fnr r r �_get_env_configs� s
z ConfigMerger._get_env_configsc C s� g }| j s|S dg}| jr.|�d� |�d� |D ]v}| j �|�}|r2tj�|�r2z|�t�|�� W q2 t k
r� t
� d|� Y q2 t k
r� t�
t
d|� Y q2X q2|S )N� cloud_config�vendor2_cloud_config�vendor_cloud_configz5Skipped loading cloud-config from %s due to non-root.z&Failed loading of cloud-config from %s)rM rQ rT �
get_ipath_curr9 r< �isfiler
rW �PermissionErrorr7 rG r0 r6 )r Zi_cfgsZcc_pathsZcc_pZcc_fnr r r �_get_instance_configs� s0
� � z"ConfigMerger._get_instance_configs)�returnc C s� g }| j rN| j D ]<}z|�t�|�� W q tk
rJ t�td|� Y qX q|�| �� � |�| � � � |�| �
� � | j r�|�| j � t� |�S )Nz'Failed loading of configuration from %s)
rO rT r
rW r0 r6 r7 �extendrX r_ rU rP Z
mergemanydict)r �cfgsZc_fnr r r � _read_cfg s"
� zConfigMerger._read_cfgc C s | j d kr| �� | _ | j S r )rR rc r r r r �cfg s
zConfigMerger.cfg)NNNNT) r r r r rU rX r_ rI rc �propertyrd r r r r rL � s �
rL c @ sF e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd
d� Zdd� Z dS )�ContentHandlersc C s i | _ g | _d S r )�
registered� initializedr r r r r ( s zContentHandlers.__init__c C s
| � |�S r )�
is_registered)r �itemr r r � __contains__, s z ContentHandlers.__contains__c C s
| � |�S r )� _get_handler)r �keyr r r � __getitem__/ s zContentHandlers.__getitem__c C s
|| j kS r �rg �r Z content_typer r r ri 2 s z ContentHandlers.is_registeredFTc C sj t � }|�� D ](}|r"|�|� q| �|�s|�|� q|D ]}|| j|< q<|rf|| jkrf| j�|� |S r )�setZ
list_types�addri rg rh rT )r �modrh Z overwrite�types�tr r r �register5 s
zContentHandlers.registerc C s
| j | S r ro rp r r r rl C s z ContentHandlers._get_handlerc C s t | j�� �S r )�listrg �itemsr r r r rx F s zContentHandlers.itemsN)FT)
r r r r rk rn ri rv rl rx r r r r rf ' s
rf c @ sr e Zd ZdZded�dd�Zedd�dd�Zdd d
�Zdd d �Z dd
d�Z
ddd�Z de d�dd�Z
d dd�ZdS ) �Paths� N)� path_cfgsc C s� || _ |�dd�| _|�dd�| _tj�| jd�| _tj�| jd�| _tj�| jd�| _ |�dd �}tj�|d
�| _
d d d
ddddddddddddddd d d d d �| _ || _ d S )!N� cloud_dirz/var/lib/cloud�run_dirz/run/cloud-init�instancez
boot-finishedZseedZ
templates_dirz/etc/cloud/templates/z%s.tmpl� boothookszcloud-config.txt�data�handlers�instance-data.json� instance-data-sensitive.jsonz .instance-idz manual-cleanzobj.pkl�scriptsrC zuser-data.txt.iz
user-data.txtzvendor-data.txt.izvendor-data2.txt.izvendor-data2.txtzvendor-data.txtzvendor2-cloud-config.txtzvendor-cloud-config.txtzscripts/vendor�warnings)r rY r� r� �
instance_data�instance_data_sensitiveZ instance_idZmanual_clean_markerZobj_pklr� rC ZuserdataZ userdata_rawZ
vendordataZ vendordata2Zvendordata2_rawZvendordata_rawrZ r[ Zvendor_scriptsr� )
rb �getr| r} r9 r< r? �
instance_linkZ
boot_finishedZseed_dirZ template_tpl�lookupsrS )r r{ �dsZ template_dirr r r r M sJ � �� zPaths.__init__)�ci_pkl_versionr` c C sJ t | d�s t| j| jd�j| _d| jkr2d| jd< d| jkrFd| jd< dS )z(Perform deserialization fixes for Paths.r} )r{ r� r� r� r� r� N)�hasattrry rb rS r} r� )r r� r r r � _unpickle{ s
�
��zPaths._unpicklec C s | � | j|�S r )r3 r� �r r+ r r r r\ � s zPaths.get_ipath_curc C s | � | j|�S r )r3 r| r� r r r rE � s zPaths.get_cpathc C sf | j s
d S | j �� }|d kr d S t|��tjd�}tj�| jd|�}| j �
|�}|rbtj�||�}|S )Nr) Z instances) rS Zget_instance_id�strr* r9 �sepr< r? r| r� r� )r r+ ZiidZ
path_safe_iid�ipathZadd_onr r r �
_get_ipath� s
zPaths._get_ipathc C s$ | � |�}|s t�d� d S |S d S )Nz?No per instance data available, is there an datasource/iid set?)r� r7 r> )r r+ r� r r r rD � s
�zPaths.get_ipath)�basec C s |d kr |S t j�|| j| �S r )r9 r< r? r� )r r� r+ r r r r3 � s zPaths._get_pathc C s | � | j|�S r )r3 r} r� r r r � get_runpath� s zPaths.get_runpath)N)N)N)N)N)N)N)r r r Z_ci_pkl_versionrI r �intr� r\ rE r� rD r� r3 r� r r r r ry J s .
ry c @ sX e Zd ZdZdZdZdZdd� Zddd�Zd d
� Z d d � Z
d
d� Z dd� Z ddd�Z
dS )�DefaultingConfigParserr g FNc C sD | j }zt�| ||�}W n& tk
r, Y n tk
r> Y nX |S r )�DEF_BASEr r� r r �r �section�option�valuer r r r� � s zDefaultingConfigParser.getc C s4 | � |�s |�� dkr | �|� t�| |||� d S )N�default)Z has_section�lowerZ add_sectionr rq r� r r r rq � s
zDefaultingConfigParser.setc C s | � ||�rt�| ||� d S r )�
has_optionr �
remove_option�r r� r� r r r r� � s z$DefaultingConfigParser.remove_optionc C s | � ||�s| jS t�| ||�S r )r� � DEF_BOOLEANr �
getbooleanr� r r r r� � s z!DefaultingConfigParser.getbooleanc C s | � ||�s| jS t�| ||�S r )r� � DEF_FLOATr �getfloatr� r r r r� � s z DefaultingConfigParser.getfloatc C s | � ||�s| jS t�| ||�S r )r� �DEF_INTr �getintr� r r r r� � s z DefaultingConfigParser.getintc C s< d}t � }| �|� |�� |�� }|r8d�||dg�}|S )N� �
)r �write�flush�getvaluer? )r �headerr; Z outputstreamr r r � stringify� s
z DefaultingConfigParser.stringify)N)N)r r r r� r� r� r� r� rq r� r� r� r� r� r r r r r� � s
r� )$r# r9 � configparserr r r �ior r Z cloudinitr Zloggingr r r
Zcloudinit.settingsr r r
r Z getLoggerr r7 r0 r r r r% r, r- r@ rL rf ZCloudInitPickleMixinry r� r r r r �<module> s&
R-j#|