CasperSecurity
o
쑛h�@ � @ s d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dl m Z d dl
m Z m Z m
Z
mZ d dlmZmZmZmZ e�e�ZG dd� de�ZG d d
� d
�ZG d d � d �ZG d
d� 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 dd � d e�Z#dS ) � N)�
NoOptionError�NoSectionError�RawConfigParser)�StringIO)�time)� persistence�settings�
type_utils�util)� CFG_ENV_NAME�
PER_ALWAYS� PER_INSTANCE�PER_ONCEc @ � e Zd ZdS )� LockFailureN��__name__�
__module__� __qualname__� r r �3/usr/lib/python3/dist-packages/cloudinit/helpers.pyr � r c @ r )� DummyLockNr r r r r r r r c @ s4 e Zd Zdd� Zejd dd��Zdd� Zdd � Zd
S ) �DummySemaphoresc C s d S �Nr ��selfr r r �__init__! � zDummySemaphores.__init__Fc c s � t � V d S r )r )r �_name�_freq�_clear_on_failr r r �lock$ s � zDummySemaphores.lockc C � dS �NFr �r r r r r r �has_run( r zDummySemaphores.has_runc C r# )NTr r% r r r �clear+ r zDummySemaphores.clearN�F) r r r r �
contextlib�contextmanagerr" r& r' r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )�FileLockc C �
|| _ d S r )�fn)r r- r r r r 0 �
zFileLock.__init__c C s dt �| �| jf S )Nz<%s using file %r>)r �obj_namer- r r r r �__str__3 s zFileLock.__str__N)r r r r r0 r r r r r+ / s r+ c C s | � dd�S )N�-�_)�replace)�namer r r �canon_sem_name7 s r5 c @ sD e Zd Zdd� Zejddd��Zdd� Zdd � Zd
d � Z d d
� Z
dS )�FileSemaphoresc C r, r )�sem_path)r r7 r r r r < r. zFileSemaphores.__init__Fc c s@ � t |�}z
| �||�V W d S ty |r | �||� � w r )r5 �_acquire� Exceptionr' )r r4 �freq�
clear_on_failr r r r" ? s � �zFileSemaphores.lockc C sL t |�}| �||�}zt�|� W dS ttfy% t�td|� Y dS w )Nz Failed deleting semaphore %sFT)r5 � _get_pathr
�del_file�IOError�OSError�logexc�LOG)r r4 r: �sem_filer r r r' I s ��zFileSemaphores.clearc C sl | � ||�rd S | �||�}dt�� t� f }z t�||� W t |�S ttfy5 t� t
d|� Y d S w )Nz%s: %s
z Failed writing semaphore file %s) r& r< �os�getpidr r
�
write_filer> r? r@ rA r+ )r r4 r: rB �contentsr r r r8 S s ��zFileSemaphores._acquirec C s0 |r|t krdS t|�}| �||�}tj�|�S r$ )r r5 r< rC �path�exists)r r4 r: �cnamerB r r r r& c s
zFileSemaphores.has_runc C s6 | j }|r |tkrtj�||�S tj�|d||f �S )Nz%s.%s)r7 r
rC rG �join)r r4 r: r7 r r r r< m s zFileSemaphores._get_pathNr( ) r r r r r) r* r" r' r8 r& r< r r r r r6 ; s
r6 c @ s& e Zd Zdd� Zdd� Zd dd�ZdS )
�Runnersc C s || _ i | _d S r )�paths�sems)r rL r r r r v �
zRunners.__init__c C sh |t ks|sd S d }|tkr| j�d�}n
|tkr | j�d�}|s#d S || jvr/t|�| j|< | j| S )N�sem)r r
rL � get_ipathr � get_cpathrM r6 )r r: r7 r r r �_get_semz s
zRunners._get_semNFc C s� | � |�}|s
t� }|sg }|�||�r t�d||� dS |�|||��,}|s-td| ��t�d||� t|t�rA|di |��}n||� }d|fW d � S 1 sSw Y d S )Nz%s already ran (freq=%s))FNz Failed to acquire lock for %szRunning %s using lock (%s)Tr ) rR r r& rA �debugr" r �
isinstance�dict) r r4 �functor�argsr: r; rO �lk�resultsr r r �run� s"
$�z Runners.runr$ )r r r r rR rZ r r r r rK u s rK c @ sR e Zd Z ddd�Zdd� Zdd� Zd d
� Zd efd d
�Ze dd� �Z
dS )� ConfigMergerNTc C s( || _ || _|| _|| _|| _d | _d S r )�_paths�_ds�_fns� _base_cfg�_include_vendor�_cfg)r rL �
datasource�additional_fns�base_cfg�include_vendorr r r r � s
zConfigMerger.__init__c C sh g }| j r2z| j �� }|rt|t�r |�|� W |S W |S W |S ty1 t�td| j � Y |S w |S )Nz2Failed loading of datasource config object from %s) r] �get_config_objrT rU �appendr9 r
r@ rA )r �d_cfgs�ds_cfgr r r �_get_datasource_configs� s(
�� ���z$ConfigMerger._get_datasource_configsc C sV g }t tjv r)tjt }z |�t�|�� W |S ty( t�td|� Y |S w |S )Nz%Failed loading of env. config from %s) r rC �environrg r
� read_confr9 r@ rA )r �e_cfgs�e_fnr r r �_get_env_configs� s
��z ConfigMerger._get_env_configsc C s� g }| j s|S dg}| jr|�d� |�d� |D ]9}| j �|�}|rRtj�|�rRz
|�t�|�� W q t yB t
� d|� Y q t yQ t�
t
d|� Y qw q|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)r\ r` rg �
get_ipath_currC rG �isfiler
rl �PermissionErrorrA rS r9 r@ )r �i_cfgs�cc_paths�cc_p�cc_fnr r r �_get_instance_configs� s0
� ���z"ConfigMerger._get_instance_configs�returnc C s� g }| j r%| j D ] }z
|�t�|�� W q ty$ t�td|� Y qw |�| �� � |�| � � � |�| �
� � | j rC|�| j � t� |�S )Nz'Failed loading of configuration from %s)
r^ rg r
rl r9 r@ rA �extendro rz rj r_ �
mergemanydict)r �cfgs�c_fnr r r � _read_cfg� s
��
zConfigMerger._read_cfgc C s | j d u r
| �� | _ | j S r )ra r� r r r r �cfg s
zConfigMerger.cfg)NNNNT) r r r r rj ro rz rU r� �propertyr� r r r r r[ � s
�
r[ 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
rN zContentHandlers.__init__c C �
| � |�S r )�
is_registered)r �itemr r r � __contains__ r. z ContentHandlers.__contains__c C r� r )� _get_handler)r �keyr r r � __getitem__ r. zContentHandlers.__getitem__c C s
|| j v S r �r� �r � content_typer r r r� r. z ContentHandlers.is_registeredFTc C sj t � }|�� D ]}|r|�|� q| �|�s|�|� q|D ]}|| j|< q |r3|| jvr3| j�|� |S r )�set�
list_types�addr� r� r� rg )r �modr� � overwrite�types�tr r r �register s
� zContentHandlers.registerc C s
| j | S r r� r� r r r r� ( r. z ContentHandlers._get_handlerc C s t | j�� �S r )�listr� �itemsr r r r r� + � zContentHandlers.itemsN)FT)
r r r r r� r� r� r� r� r� r r r r r� s
r� c @ st e Zd ZdZddefdd�Zdeddfdd �Zdd
d �Zdd d
�Z ddd�Z
ddd�Z dde fdd�Z
ddd�ZdS )�Paths� N� path_cfgsc
C s || _ |�dd�| _|�dd�| _|�dtj�| _tj� | jd�| _
tj� | j
d�| _ tj� | jd�| _ |�d d
�}tj� |d �| _
i d d �d
d�dd�dd�dd�dd�dd�dd�dd�dd �d d �d d �d d �d!d"�d#d$�d%d&�d'd(�d)d*d+d,d-d.d/d0d1��| _|| _d S )2N� cloud_dirz/var/lib/cloud�docs_dirz/usr/share/doc/cloud-init/�run_dir�instancez
boot-finished�seed�
templates_dirz/etc/cloud/templates/z%s.tmpl� boothooksrp zcloud-config.txt�data�handlers�
instance_data�instance-data.json�instance_data_sensitive� instance-data-sensitive.json�combined_cloud_config�combined-cloud-config.json�network_configznetwork-config.json� instance_idz .instance-id�manual_clean_markerz manual-clean�obj_pklzobj.pkl�scriptsrO �userdatazuser-data.txt.i� userdata_rawz
user-data.txt�
vendordatazvendor-data.txt.i� vendordata2zvendor-data2.txt.izvendor-data2.txtzvendor-data.txtzvendor2-cloud-config.txtzvendor-cloud-config.txtzscripts/vendor�warnings�hotplug.enabled�
.skip-network)�vendordata2_raw�vendordata_rawrq rr �vendor_scriptsr� r� r� )r~ �getr� r� r �DEFAULT_RUN_DIRr� rC rG rJ �
instance_link�
boot_finished�seed_dir� template_tpl�lookupsrb )r r� �ds� template_dirr r r r 2 sx �������� � �
����������
�
!zPaths.__init__�ci_pkl_versionr{ c C sv t | d�st| j| jd�j| _d| jvrd| jd< d| jvr#d| jd< d| jvr-d| jd< d | jvr9d | jd < d
S d
S ) z(Perform deserialization fixes for Paths.r� )r� r� r� r� r� r� r� r� r� N)�hasattrr� r~ rb r� r� )r r� r r r � _unpickleg s"
��
�
�
�zPaths._unpicklec C � | � | j|�S r )r< r� �r r4 r r r rs � r� zPaths.get_ipath_curc C r� r )r< r� r� r r r rQ � r� zPaths.get_cpathc C sf | j sd S | j �� }|d u rd S t|��tjd�}tj�| jd|�}| j �
|�}|r1tj�||�}|S )Nr2 � instances) rb �get_instance_id�strr3 rC �seprG rJ r� r� r� )r r4 �iid�
path_safe_iid�ipath�add_onr r r �
_get_ipath� s
zPaths._get_ipathc C s | � |�}|st�d� d S |S )Nz?No per instance data available, is there an datasource/iid set?)r� rA �warning)r r4 r� r r r rP � s
�zPaths.get_ipath�basec C s |d u r|S t j�|| j| �S r )rC rG rJ r� )r r� r4 r r r r< � s zPaths._get_pathc C r� r )r< r� r� r r r � get_runpath� r� zPaths.get_runpathr )r r r �_ci_pkl_versionrU r �intr� rs rQ r� rP r� r< r� r r r r r� / s 5
r� c @ sX e Zd ZdZdZdZdZdd� Zddd�Zd d
� Z d d � Z
d
d� Z dd� Z ddd�Z
dS )�DefaultingConfigParserr g FNc C sB | j }z
t�| ||�}W |S ty Y |S ty Y |S w 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)� has_section�lower� add_sectionr r� r� r r r r� � s
zDefaultingConfigParser.setc C s" | � ||�rt�| ||� d S d S r )�
has_optionr �
remove_option�r r� r� r r r r� � s �z$DefaultingConfigParser.remove_optionc C � | � ||�s | jS t�| ||�S r )r� � DEF_BOOLEANr �
getbooleanr� r r r r� � � z!DefaultingConfigParser.getbooleanc C r� r )r� � DEF_FLOATr �getfloatr� r r r r� � r� z DefaultingConfigParser.getfloatc C r� r )r� �DEF_INTr �getintr� r r r r� � r� z DefaultingConfigParser.getintc C s< d}t � }| �|� |�� |�� }|r d�||dg�}|S )N� �
)r �write�flush�getvaluerJ )r �headerrF � outputstreamr r r � stringify� s
z DefaultingConfigParser.stringifyr )r r r r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� )$r) �loggingrC � configparserr r r �ior r � cloudinitr r r r
�cloudinit.settingsr r r
r � getLoggerr rA r9 r r r r+ r5 r6 rK r[ r� �CloudInitPickleMixinr� r� r r r r �<module> s*
:-j#