Your IP : 216.73.217.13

Current Path : /proc/thread-self/root/lib/python3/dist-packages/cloudinit/__pycache__/
Upload File :
Current File : //proc/thread-self/root/lib/python3/dist-packages/cloudinit/__pycache__/util.cpython-38.pyc

U

�Ad�i�
@s�	ddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZ
ddlZddl
Z
ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZmZddlmZmZddlmZm Z ddl!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-dd	l.m/Z/m0Z0dd
l.m1Z2ddl.m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;da<e2�=e>�Z?e
j@d
iZAdejBejCZDdZEdZFdd�ZGe"��dpdd��ZHe"��dqdd��ZI�drdd�ZJ�dsdd�ZKdd�ZLdd�ZMd d!�ZNGd"d#�d#�ZOGd$d%�d%eP�ZQGd&d'�d'eP�ZRd(d)�ZS�dtd*d+�ZT�dud,d-�ZU�dvd.d/�ZV�dwd1d2�ZW�dxd3d4�ZXdd5�e)d6�d7d8�ZYd9d:�ZZd;d<�Z[d=d>�Z\�dyd@dA�Z]dBdC�Z^e_e`dD�dEdF�ZadGdH�Zbd?d?de2jcd?fdIdJ�Zde"�dKdL��Zee"�dMdN��Zfe"�dOdP��Zge"�dQdR��Zhe"�dSdT��Zie"�dUdV��Zj�dzdXdY�Zk�d{dZd[�Zl�d|d\d]�Zm�d}d^d_�Zne"�d`da��Zodbdc�Zpe"�ddde��Zq�d~dfdg�Zr�ddhdi�Zsdjdk�Zt�d�dldm�Zu�d�e`d6�dndo�Zvejwdpdq��Zxejwdrds��Zydtdu�Zzdvdw�Z{�d�dzd{�Z|�d�d|d}�Z}de`ffd~d�Z~�d�d�d��Zdd5�e`d6�d�d��Z�dd5�e`d6�d�d��Z��d�d�d��Z��d�d�d��Z�d�d��Z�ed�d�d�d�g�Z��d�d�d��Z��d�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z�d�d��Z��d�d�d��Z��d�d�d��Z��d�d�d��Z��d�d�d��Z��d�d�d��Z��d�d�d��Z�d�d��Z�d�d��Z��d�d�d��Z�e"�d�d���Z�d�d��Z��d�d�d��Z��d�d�d��Z��d�d�d��Z�d�d��Z�d�dÄZ�d�dńZ��d�d�dDŽZ�d�dɄZ�d�d˄Z�d�d̈́Z��d�d�dЄZ�e`ffd�d҄Z�d�dԄZ�d�dքZ�d�d؄Z��d�d�dڄZ�ejwd�d܄�Z�d�dބZ��d�d�d�Z�d�d�Z�d�d�Z��d�d�d�Z�d�d�Z�d�d�Zd�d�Z�e"�d�d��Z�d�d�Z�d�d�Z��d�dWd�e�e�dd��d�d��Z�d�d��Z�d�d��Z�e_e�d��d�d��Z�e_e�d���d�d�Z�e_e_d���d�d�Z�e_e_d���d�d�Z�e_e*e_�d��d�d�Z��d�d?dd�d
��d�d�Z��d
�d�Z��d��d�d�Z��d�d�Z��d��d�d�Zd��d�d�ZÐd�d�ZĐd�d�ZŐd�d�ZƐd�d �Z�e"��d!�d"��ZȐd#�d$�Zɐd��d&�d'�Zʐd(�d)�Zːd*�d+�Z̐d,�d-�Z�e?dWf�d.�d/�Zΐd0�d1�Zϐd2�d3�ZАd4�d5�Zѐd6�d7�ZҐd8�d9�ZӐd:�d;�ZԐd<�d=�Z�e?dWf�d>�d?�Z�e_e��d@��dA�dB�Z�e+�dC�Zؐd�e'�dDe�feؐdE��dF�dG�ZِdH�dI�Z�dde
j�j@f�dJ�dK�Zܐd��dM�dN�ZݐdO�dP�Zސd��dQ�dR�ZߐdS�dT�Z�d��dU�dV�Z�dW�dX�Z�dY�dZ�Z�d��d[�d\�Z�d��d^�d_�Z�d`�da�Z�d��db�dc�Z�dd�de�Z�d��dh�di�Z�e#G�dj�dk��dke�dk�dl�dm�dn�dog���Z�dS(��N)�	b64decode�	b64encode)�deque�
namedtuple)�EACCES�ENOENT)�	lru_cache�total_ordering)�Path)�Callable�Deque�Dict�List�TypeVar)�parse)�features�importer)�log)�mergers�safeyaml�subp�
temp_utils�
type_utils�
url_helper�version)�CFG_BUILTIN�_z_-.())�true�1ZonZyes)Zoff�0ZnoZfalsecCs"tttt��j�d�dd���S)N�.�)�tuple�map�int�os�uname�release�split�r)r)�0/usr/lib/python3/dist-packages/cloudinit/util.py�kernel_versionDsr+cCstjddgd|d�}|j��S)z�Return the sanitized string output by `dpkg --print-architecture`.

    N.B. This function is wrapped in functools.lru_cache, so repeated calls
    won't shell out every time.
    Zdpkgz--print-architectureT��capture�target)r�stdout�strip)r.�outr)r)r*�get_dpkg_architectureHs�r2c	
s�ddddd�}i�z�tjddgd|d	�}|j��D],}|�d
�\}}}||kr2|���||<q2�fdd�|��D�}t|�r�t�d
d�	|��WnFtj
k
r�}z&t�d|�tdd�|��D���W5d}~XYnX�S)N�codenameZdescription�idr')ZCodenameZDescriptionzDistributor IDZRelease�lsb_releasez--allTr,�:csg|]}|�kr|�qSr)r))�.0�k��datar)r*�
<listcomp>eszlsb_release.<locals>.<listcomp>z.Missing fields in lsb_release --all output: %s�,z#Unable to get lsb_release --all: %scss|]}|dfVqdS)ZUNAVAILABLENr)�r7�vr)r)r*�	<genexpr>nszlsb_release.<locals>.<genexpr>)rr/�
splitlines�	partitionr0�values�len�LOG�warning�join�ProcessExecutionError�dict)	r.Zfmapr1�line�fnamer�val�missing�errr)r9r*r5Us,��(r5�utf-8cCst|t�r|S|�|�S�N)�
isinstance�str�decode)�blob�encodingr)r)r*�
decode_binaryss
rUcCst|t�r|S|�|�SrO)rP�bytes�encode)�textrTr)r)r*�encode_textzs
rYcCs2t|�}z|�d�WStk
r,|YSXdS�NrN)rrR�UnicodeDecodeError)�source�decodedr)r)r*�b64d�s
r^cCs"t|t�s|�d�}t|��d�SrZ)rPrVrWrrR)r\r)r)r*�b64e�s

r_cCsP|jdd�}|��dkrLt|t�rL|��}|r<|jr<|j}nd}|�|d�S|S)NT�rRrXrN�surrogateescape)Zget_payloadZget_content_maintyperPrVZget_charsetZinput_codecrR)�partZcte_payload�charsetrTr)r)r*�fully_decoded_payload�s�
rdc@s&eZdZd	dd�Zdd�Zdd�ZdS)
�SeLinuxGuardFcCs<zt�d�|_Wntk
r*d|_YnX||_||_dS)N�selinux)r�
import_modulerf�ImportError�path�	recursive)�selfrirjr)r)r*�__init__�szSeLinuxGuard.__init__cCs|jr|j��rdSdSdS�NTF)rf�is_selinux_enabled�rkr)r)r*�	__enter__�szSeLinuxGuard.__enter__c
Cs�|jr|j��sdStj�|j�s&dStj�|j�}z"t�|�}|j�||tj	�Wnt
k
rlYdSXt�d||j
�z|jj||j
d�Wn4t
k
r�}zt�d||j
|�W5d}~XYnXdS)Nz,Restoring selinux mode for %s (recursive=%s)�rjz,restorecon failed on %s,%s maybe badness? %s)rfrnr%ri�lexists�realpath�lstatZmatchpathcon�stat�ST_MODE�OSErrorrD�debugrjZ
restoreconrE)rkZ	excp_typeZ
excp_valueZexcp_tracebackriZstats�er)r)r*�__exit__�s0
��zSeLinuxGuard.__exit__N)F)�__name__�
__module__�__qualname__rlrprzr)r)r)r*re�s

rec@seZdZdS)�MountFailedErrorN�r{r|r}r)r)r)r*r~�sr~c@seZdZdS)�DecompressionErrorNrr)r)r)r*r��sr�c	Ostt��}|dkr\z|||�t�d�Wqptk
rXttdt�|��t�d�YqpXnt�d|t�|��dS)Nrz&Failed forking and calling callback %s�z(Forked child %s who will run callback %s)	r%�fork�_exit�	Exception�logexcrDr�obj_namerx)Zchild_cb�args�kwargsZfidr)r)r*�fork_cb�s"
��r�cCsBt|t�r|dkSt}|r&t|�|}t|�����|kr>dSdSrm)rP�bool�TRUE_STRINGS�listrQ�lowerr0�rK�addonsZ	check_setr)r)r*�is_true�s
r�cCsBt|t�r|dkSt}|r&t|�|}t|�����|kr>dSdS)NFT)rPr��
FALSE_STRINGSr�rQr�r0r�r)r)r*�is_false�s
r�cCs |sdSt|t�r|St||�S)NF)rPr�r�)rKr�r)r)r*�translate_bools

r�� cs8t����stjtj�d���fdd�td|�D��S)N�csg|]}�����qSr))Zchoice)r7Z_x��r�select_fromr)r*r;szrand_str.<locals>.<listcomp>r)�randomZSystemRandom�string�
ascii_letters�digitsrF�range)�strlenr�r)r�r*�rand_strsr�cCs*|sd}tdd�d|}||krq&q|S)Nr��)r�r)r�)Z
dictionaryZpostfixZnewkeyr)r)r*�
rand_dict_keysr���instance_data_file��returnc
Cs�ddlm}m}m}zt|�}Wn:tk
rZ}z|jtkrHiWY�S�W5d}~XYnX|r�tj	�
|�r�z||||�}t�d||�WnH|k
r�Yn6|k
r�}zt�
d||t|��W5d}~XYnX|dkr�iSt|id�S)z>Read a yaml config with optional template, and convert to dictr)�JinjaLoadError�
NotJinjaError�render_jinja_payload_from_fileNz?Applied instance data in '%s' to configuration loaded from '%s'z:Could not apply Jinja template '%s' to '%s'. Exception: %s��default)Z!cloudinit.handlers.jinja_templater�r�r��	load_file�IOError�errnorr%ri�existsrDrxrE�repr�	load_yaml)rJr�r�r�r�Zconfig_fileryr)r)r*�	read_conf#s>
���r�cGstt|��SrO)�sorted�
uniq_merge)�listsr)r)r*�uniq_merge_sortedUsr�cGsFg}|D]4}t|t�r2|���d�}dd�|D�}|�|�qt|�S)Nr<cSsg|]}|r|�qSr)r))r7�ar)r)r*r;gszuniq_merge.<locals>.<listcomp>)rPrQr0r(�extend�	uniq_list)r�Z
combined_list�a_listr)r)r*r�as
r�cCs`t��D]\}}|�||�}qg}|D]}|tkr&|�|�q&|D]}|�|d�}qB|��}|S)Nr�)�FN_REPLACEMENTS�items�replace�
FN_ALLOWED�appendr0)�fnr8r>Zremovalsr)r)r*�clean_filenamelsr�Tc
Cs�zht�t|��}t�t�ddd|���:}|rFt|���W5QR�WS|��W5QR�WSW5QRXWn@t	k
r�}z"|r�|WY�St
t|��|�W5d}~XYnXdS)N�rbr�)�io�BytesIOrY�
contextlib�closing�gzipZGzipFilerU�readr�r�rQ)r:�quietrR�bufZghryr)r)r*�decomp_gzipys$r�cCs~|sdS|�dd�}|d��}t|�dkr:|d��}nd}|rV|dksV|��dkrZd}|rr|dksr|��dkrvd}||fS)N)NNr6r�rr!z-1Znone)r(r0rCr�)Zug_pairZ	ug_parted�u�gr)r)r*�extract_usergroup�sr�)�root_dirr�cCsht�}t�tj�|d��D]H}tj�|�s,qtj�|�dd�}|��}|r|�d�dkr|||<q|S)Nz*.pyr���r ���)	rH�globr%rirF�isfile�basenamer0�find)r��entriesrJ�modnamer)r)r*�get_modules_from_dir�s
r�c	Cs,t|d��}|�|�|��W5QRXdS)N�w)�open�write�flush)�conpathrXZwfhr)r)r*�write_to_console�s
r�c	Cs�|rtj�|�|r�d}d}tj�|�rxzt||�d}Wn<tk
rvd}tj�|�d��|rr|�	t
j|�YnX|r�|s�tj�|�|r�|ddkr�|�	||dd��n|�	||�dS)Nz/dev/consoleFTzFailed to write to /dev/console�
r�)�sys�stderrr�r%rir�r�rwr/r�loggingZWARNING)	rXZconsoler�rZ	log_levelZfallback_to_stdoutr�Zwriting_to_console_workedZ
console_errorr)r)r*�	multi_log�s(

r�cCsdt��kS)NZLinux��platform�systemr)r)r)r*�is_Linux�sr�cCs$dt��krdSt��dkr dSdS)NZBSDTZ	DragonFlyFr�r)r)r)r*�is_BSD�s
r�cCst�ddkS)N�variant�freebsd��system_infor)r)r)r*�
is_FreeBSD�sr�cCst�ddkS)Nr��	dragonflyr�r)r)r)r*�is_DragonFlyBSD�sr�cCst�ddkS)Nr��netbsdr�r)r)r)r*�	is_NetBSD�sr�cCst�ddkS)Nr��openbsdr�r)r)r)r*�
is_OpenBSD�sr�FcCs||kr|St||�SrO)r���yobj�keyr�r)r)r*�get_cfg_option_bool�sr�cCs*||kr|S||}t|t�s&t|�}|SrO)rPrQ)r�r�r�rKr)r)r*�get_cfg_option_str�s
r�cCstt|||d��S)Nr�)r$r�r�r)r)r*�get_cfg_option_intsr�cCs�|sd}tj�|�siSt|�}d}d|kr0d}t�||�}|r�|��}d|dkr`|d|d<|d���d�d|d<|dd	kr�d
|d<|d|d|dd�SiS)
z�Return a dictionary of distro info fields from /etc/redhat-release.

    Dict keys will align with /etc/os-release keys:
        ID, VERSION_ID, VERSION_CODENAME
    z/etc/redhat-releasezA(?P<name>.+) release (?P<version>[\d\.]+) \((?P<codename>[^)]+)\)Z	Virtuozzoz)(?P<name>.+) release (?P<version>[\d\.]+)�namer3z linuxrzred hat enterprise�redhatr)�ID�
VERSION_ID�VERSION_CODENAME)	r%rir�r��re�match�	groupdictr�rA)Zrelease_fileZredhat_releaseZredhat_regexr��groupr)r)r*�_parse_redhat_releases,��rc		Csfd}d}d}i}d}tj�d�r,ttd��}|s:d}t�}|r�|�dd�}|�dd�}d|ksfd|krpt��}nd|d	kr�|�d
d�}nN|dkr�|s�|�d
d�}n4|�dd�}|s�t	�
d
|�dd��}|r�|��d}|dkr�d}nzt�r�t�
���}t��}n^d}z(zt��}Wntk
�r&YnXW5d}|D]}|�r4d}�q4|�sVt�d�X|S|||fS)Nr�F�/etc/os-releaseTr�r��sles�suse�photonZPRETTY_NAME�	virtuozzor�z[^ ]+ \((?P<codename>[^)]+)\)ZVERSIONr3�rhelr�)r�r�r�r�zPUnable to determine distribution, template expansion may have unexpected results)r%rir��load_shell_contentr�r�getr��machiner�r�rr�r�r�r'rDrE�distr�)	Zdistro_nameZdistro_versionZflavorZ
os_releaseZos_release_rhelr�r�found�entryr)r)r*�get_linux_distro4s^

�

�rcCsv|d��}d}|dkrf|dd��}|dkr6|}qr|dkrDd}qr|d	krRd
}qr|dkr`d}qrd}n|d
krr|}|S)Nr��unknownZlinuxrr)Z	almalinuxZalpineZarchZcentosZ
cloudlinuxZdebianZ	eurolinuxZfedoraZmarinerZmiraclelinuxZ	openeulerZopencloudosZopenmandrivarrZrockyrZ	tencentosr)�ubuntuZ	linuxmintZmintrr�r)Zopensusez
opensuse-leapzopensuse-microoszopensuse-tumbleweedZsle_hpcz	sle-microrr)Zwindows�darwinr�r�r�r�)r�)�infor�r�Z
linux_distr)r)r*�_get_variantrs 	rcCs<t��t��t��t��tt���t�d�}t|�|d<|S)N)r�r�r'�pythonr&rr�)r�r�r'Zpython_versionr�r&rr)rr)r)r*r��s
�r�cCsX||kr|S||dkrgS||}t|t�r@dd�|D�}|St|t�sRt|�}|gS)a�
    Gets the C{key} config option from C{yobj} as a list of strings. If the
    key is present as a single string it will be returned as a list with one
    string arg.

    @param yobj: The configuration object.
    @param key: The configuration key to get.
    @param default: The default to return if key is not found.
    @return: The configuration option as a list of strings or default if key
        is not found.
    NcSsg|]}|�qSr)r)r=r)r)r*r;�sz'get_cfg_option_list.<locals>.<listcomp>)rPr�rQ)r�r�r�rKZcvalr)r)r*�get_cfg_option_list�s

rcCs>t|t�r|�d�}|}|D]}||kr0|S||}q|S)a�Return the value of the item at path C{keyp} in C{yobj}.

    example:
      get_cfg_by_path({'a': {'b': {'num': 4}}}, 'a/b/num') == 4
      get_cfg_by_path({'a': {'b': {'num': 4}}}, 'c/d') == None

    @param yobj: A dictionary.
    @param keyp: A path inside yobj.  it can be a '/' delimited string,
                 or an iterable.
    @param default: The default to return if the path does not exist.
    @return: The value of the item at keyp."
        is not found.�/)rPrQr()r�Zkeypr�Zcur�tokr)r)r*�get_cfg_by_path�s


rcCs t||�\}}t||�||fSrO)�get_output_cfg�redirect_output)�cfg�mode�outfmt�errfmtr)r)r*�fixup_output�s
r c
Cs�ttj�d��rt�d�dS|s(tj}|s2tj}dd�}|r�t�d||�|�	dd�\}}|dksl|d	kr�d
}|dkr|d}t
||�}n0|dkr�tj|d
tj
|d�}	|	j}ntd|��|r�t�|��|���||kr�t�d||�t�|��|���dS|�r�t�d||�|�	dd�\}}|dk�s6|d	k�rTd
}|dk�rHd}t
||�}n2|dk�rztj|d
tj
|d�}	|	j}ntd|��|�r�t�|��|���dS)NZ_CLOUD_INIT_SAVE_STDOUTz5Not redirecting output due to _CLOUD_INIT_SAVE_STDOUTcSs>t�d�zt�d�j}Wntk
r.YnXt�|�dS)a�Reconfigure umask and group ID to create output files securely.

        This is passed to subprocess.Popen as preexec_fn, so it is executed in
        the context of the newly-created process.  It:

        * sets the umask of the process so created files aren't world-readable
        * if an adm group exists in the system, sets that as the process' GID
          (so that the created file(s) are owned by root:adm)
        �ZadmN)r%�umask�grp�getgrnam�gr_gid�KeyError�setgid)Zgroup_idr)r)r*�set_subprocess_umask_and_gid
s

z5redirect_output.<locals>.set_subprocess_umask_and_gidzRedirecting %s to %s� r��>�>>�ab�wb�|T)�shell�stdinZ
preexec_fnz"Invalid type for output format: %sz!Invalid type for error format: %s)r�r%�environr
rDrxr�r/r�r(r��
subprocess�Popen�PIPEr0�	TypeError�dup2�fileno)
rrZo_outZo_errr(r�argZowithZnew_fp�procr)r)r*r�sd
�

�rcCsN|rt|�}i}|D]4}|rt�|�}|s2t��}t�|�}|�||�}q|SrO)�reversedrZdict_extract_mergersZdefault_mergersZ	construct�merge)Zsrcs�reverseZ
merged_cfgrZmergers_to_applyZmergerr)r)r*�
mergemanydictOs

r=c	cs.t��}zt�|�|VW5t�|�XdSrO)r%�getcwd�chdir)ZndirZcurrr)r)r*r?^s


r?c	cs&t�|�}z
|VW5t�|�XdSrO)r%r")Zn_msk�oldr)r)r*r"hs

r"cCsdj||d|d�S)Nz{0:{fill}{align}{size}}�^)�fillZalign�size)�format)rXrBZmax_lenr)r)r*�centerqs�rEcCst�d|�t�|�dS)NzRecursively deleting %s)rDrx�shutilZrmtree�rir)r)r*�del_dirwsrHr��c
Csrz0t|||�\}}}||d<||d<||d<WdStjk
rl}z|jtjkrZWY�
dS�W5d}~XYnXdS)N�	user-data�vendor-data�	meta-dataTF)�read_seededr�UrlError�codeZ	NOT_FOUND)rB�base�ext�timeout�md�ud�vdryr)r)r*�read_optional_seed�s
rVcCs�i}|sddg}n(tj�|�d�d�tj�|�d�d�g}t|�}dd�|D�}d}|D],}tj�tj�|d��rXtj�|d�}q�qXd}|D],}tj�tj�|d��r�tj�|d�}q�q�|r�|r�||d	<||d
<n|r�||d	<|S)Nz/var/lib/cloud/data/sslz /var/lib/cloud/instance/data/sslr:ZsslcSs g|]}|rtj�|�r|�qSr))r%ri�isdir)r7�dr)r)r*r;�sz%fetch_ssl_details.<locals>.<listcomp>zcert.pemzkey.pem�	cert_file�key_file)r%rirFZ
get_ipath_curZ	get_cpathr�r�)�pathsZssl_detailsZssl_cert_pathsrYrXrZr)r)r*�fetch_ssl_details�s4��
r\c
Cs |}t|�}z\t�dt|�|�t�|�}|dkrBt�d�|}n t||�sbtd|t�	|�f��|}Wn�tj
ttfk
�r}z�d}d}t|d�r�t
|d�r�t
|d�}nt|d�r�t
|d�r�t
|d�}|r�|dj|jd|jd|d	�7}n|d
j|d�7}t�|�W5d}~XYnX|S)NzKAttempting to load yaml from string of length %s with allowed root types %sz-loaded blob returned None, returning default.z2Yaml load allows %s root types, but got %s insteadzFailed loading yaml blobZcontext_markZproblem_markz5. Invalid format at line {line} column {col}: "{err}"r�)rI�colrMz. {err})rM)rUrDrxrCr�loadrPr5rr�Z	YAMLError�
ValueError�hasattr�getattrrDrI�columnrE)rSr�ZallowedZloadedZ	convertedry�msgZmarkr)r)r*r��sJ�


��
��r��
c
CsV|�d�dkr@|�dd|�}|�dd|�}|�dd|�}nTtjrj|ddkrjt�|�jdkrj|d7}d	|d|f}d	|d|f}d	|d|f}tj|||d
�}d}	|�	�r�t
t|j�id�}	tj|||d
�}
d}|
�	�r�|
j}d}ztj|||d
�}
Wn2tj
k
�r.}zt�d|�W5d}~XYnX|
�	��rB|
j}n
t�d
�|	||fS)Nz%srrJrKrLr�rr�z%s%s%s)rR�retriesr�z!Error in vendor-data response: %szError in vendor-data response)r�r�rZ%NOCLOUD_SEED_URL_APPEND_FORWARD_SLASHr�urlparseZqueryrZread_file_or_url�okr�rU�contentsrNrDrx)rPrQrRreZfile_retriesZud_urlZvd_urlZmd_urlZmd_resprSZud_resprTrUZvd_respryr)r)r*rM�sN���


rMcs�tt���dd�}dd�|D�}�fdd�|D�}g}|D]`}z |�ttj��|�|d��Wq:tk
r�}z|jt	kr�t
�d�|�W5d}~XYq:Xq:t|�S)	zRead configuration directory.T�r<cSsg|]}|�d�r|�qS)z.cfg)�endswith�r7�fr)r)r*r;s
zread_conf_d.<locals>.<listcomp>cs&g|]}tj�tj��|��r|�qSr))r%rir�rFrk��confdr)r*r;
sr�z,REDACTED config part %s/%s for non-root userN)
r�r%�listdirr�r�rirFrwr�rrDrEr=)rnr�Zconfs�cfgsr�ryr)rmr*�read_conf_ds(��
�rqc
Cs�t�}i}zt||d�}Wn8tk
rR}z|jtkrBt�d|�W5d}~XYnX|�|�d}d|kr�|d}|r�t|t	�s�t
d|t�|�f��q�t	|��
�}ntj�|�d��r�|�d�}|r�tj�|�r�t||d�}|�|�t|�S)aRead yaml file along with optional ".d" directory, return merged config

    Given a yaml file, load the file as a dictionary. Additionally, if there
    exists a same-named directory with .d extension, read all files from
    that directory in order and return the merged config. The template
    file is optional and will be applied to any applicable jinja file
    in the configs.

    For example, this function can read both /etc/cloud/cloud.cfg and all
    files in /etc/cloud/cloud.cfg.d and merge all configs into a single dict.
    r�z)REDACTED config part %s for non-root userNr�Zconf_dz8Config file %s contains 'conf_d' with non-string type %sz.d)rr�rwr�rrDrEr�rPrQr5rr�r0r%rirWrq�
appendleftr=)Zcfgfiler�rprryrnZ	confd_cfgr)r)r*�read_conf_with_confds2


��

rscCstt|d��S)N��cmdline)r��read_cc_from_cmdlinertr)r)r*�read_conf_from_cmdlineKsrwc	Cs�|dkrt�}d}d}t|�}t|�}t|�}g}|�|�}|dkr�|�|||�}|dkr`|}|�t�||||�����dd��|�|||�}q<d�|�S)Nzcc:Zend_ccrz\nr�)	�get_cmdlinerCr�r�rZunquote�lstripr�rF)	ruZ	tag_beginZtag_endZbegin_lZend_lZclen�tokensZbegin�endr)r)r*rvPs*

��rvcCs2|�d�}|dks"||ddkr&|S|�dd�S)Nr�rr��
z
)r�r�)rh�posr)r)r*�dos2unixrs
r~�HostnameFqdnInfo�hostname�fqdn�
is_defaultcCs�d}d|kr,|d}t|d|�d�d�}nld|krf|d�d�dkrf|d}|dd|�d��}n2|jd|d�j}d|kr�|d}n|j|d	�\}}t|||�S)
a�Get hostname and fqdn from config if present and fallback to cloud.

    @param cfg: Dictionary of merged user-data configuration (from init.cfg).
    @param cloud: Cloud instance from init.cloudify().
    @param metadata_only: Boolean, set True to only query cloud meta-data,
        returning None if not present in meta-data.
    @return: a namedtuple of
        <hostname>, <fqdn>, <is_default> (str, str, bool).
        Values can be none when
        metadata_only is True and no cfg or metadata provides hostname info.
        is_default is a bool and
        it's true only if hostname is localhost and was
        returned by util.get_hostname() as a default.
        This is used to differentiate with a user-defined
        localhost hostname.
    Fr�r�r rNT)r��
metadata_only)r�)r�r(r��get_hostnamer�r)rZcloudr�r�r�r�r)r)r*�get_hostname_fqdn�s"�
�
r��
/etc/hostscCs�d}ztt|���D]b}|�d�}|dkr4|d|�}|��}|sBq|��}t|�dkrXq||dd�kr|d}qvqWntk
r�YnX|S)a�
    For each host a single line should be present with
      the following information:

        IP_address canonical_hostname [aliases...]

      Fields of the entry are separated by any number of  blanks  and/or  tab
      characters.  Text  from a "#" character until the end of the line is a
      comment, and is ignored. Host  names  may  contain  only  alphanumeric
      characters, minus signs ("-"), and periods (".").  They must begin with
      an  alphabetic  character  and  end  with  an  alphanumeric  character.
      Optional aliases provide for name changes, alternate spellings, shorter
      hostnames, or generic hostnames (for example, localhost).
    N�#r�r!r�)r�r@r�r0r(rCr�)r��filenamer�rIZhashpos�toksr)r)r*�get_fqdn_from_hosts�s$

r�c
Cstdkr�t�}d}i}|D]�}z`t�|dddtjtj�}g||<|D]6\}}}}	}
||�d|	|
df�|�|
d�qDWqtjtj	fk
r�YqXq|a|r�t
�d|�z0t�|d�}|ddd}|tkr�WdSWdStjtj	fk
�rYdSXdS)	a�determine if a url is resolvable, return a boolean
    This also attempts to be resilent against dns redirection.

    Note, that normal nsswitch resolution is used here.  So in order
    to avoid any utilization of 'search' entries in /etc/resolv.conf
    we have to append '.'.

    The top level 'invalid' domain is invalid per RFC.  And example.com
    should also not exist.  The '__cloud_init_expected_not_found__' entry will
    be resolved inside the search list.
    N)zdoes-not-exist.example.com.zexample.invalid.Z!__cloud_init_expected_not_found__rz%s: %szdetected dns redirection: %s�FT)�_DNS_REDIRECT_IP�set�socketZgetaddrinfoZSOCK_STREAMZAI_CANONNAMEr��addZgaierror�errorrDrx)r�ZbadipsZbadnamesZ
badresultsZiname�resultZ_famZ_stypeZ_protoZcnameZsockaddrZaddrr)r)r*�
is_resolvable�s>
�r�cCst��}|SrO)r�Zgethostname)r�r)r)r*r�sr�cCs.zt�|�dWStjk
r(YdSXdS�Nr)r��
gethostbyaddrZherror)Zipr)r)r*r�sr�cCs ttjd|tt�|�jfd�S)z5determine if this url is resolvable (existing or ip).zResolving URL: )�logfuncrc�funcr�)�log_timerDrxr�rrfr�)Zurlr)r)r*�is_resolvable_urls�r�c	Cs^|dkrdSt�d|�|D]<}z"t|�r@t�d|�|WSWqtk
rVYqXqdS)zc
    Search through a list of mirror urls for one that works
    This needs to return quickly.
    Nz%search for mirror in candidates: '%s'zfound working mirror: '%s')rDrxr�r�)Z
candidatesZcandr)r)r*�search_for_mirrorsr�c	CsDttj�d��rdSttj��}t�|��tj	���W5QRXdS)z�
    reopen stdin as /dev/null so even subprocesses or other os level things get
    /dev/null as input.

    if _CLOUD_INIT_SAVE_STDIN is set in environment to a non empty and true
    value then input will not be closed (useful for debugging).
    Z_CLOUD_INIT_SAVE_STDINN)
r�r%r1r
r��devnullr6r7r�r0)�fpr)r)r*�close_stdin+sr��devicecCsvg}|st�d�t�d�S|�d�rL|�d�}dd�d|d|fD�}n&|dkr`t�d�}n|d	krrt�d�}|S)
Nz/dev/msdosfs/*z/dev/iso9660/*�LABEL=cSsg|]}tj�|�r|�qSr)�r%rir�)r7�pr)r)r*r;As�z*find_devs_with_freebsd.<locals>.<listcomp>z
/dev/msdosfs/z
/dev/iso9660/�	TYPE=vfat�TYPE=iso9660)r��
startswithry)�criteria�oformat�tag�no_cacheri�devlist�labelr)r)r*�find_devs_with_freebsd9s

�
r�cCs�g}d}d}|r8|�d�r$|�d�}|�d�r8|�d�}tjdddgdgd�}|j��D]j}	|sd|r~tjd|	gdd	gd�\}
}|r�d
||
kr�qX|dkr�d|
kr�qX|d
kr�d|
kr�qX|�d|	�qX|S)Nr�zTYPE=�sysctl�-n�hw.disknamesr�ZrcsZ	mscdlabelr�z
label "%s"�iso9660zISO filesystem�vfat�/dev/)r�ryrr/r(r�)r�r�r�r�rir�r��_typer1�devZ
mscdlabel_outrr)r)r*�find_devs_with_netbsdMs(



r�cCs�tjdddgdgd�}g}|j���d�D]P}|�d�s:q*|dkrDq*|�|dd	�d
�|�d�s*|�|dd	�d�q*d
d�|D�S)Nr�r�r�rr�r<r6zfd0:r�r��cd�icSsg|]}d|�qS�r�r)�r7r�r)r)r*r;tsz*find_devs_with_openbsd.<locals>.<listcomp>)rr/�rstripr(rjr�r�)r�r�r�r�rir1r�rr)r)r*�find_devs_with_openbsdfs

r�cCs�tjdddgdgd�}dd�t|j��dd	�D�}|d
krJdd�|D�}n(|dkrbd
d�|D�}n|rrt�d|�dd�|D�S)Nr�r�z
kern.disksrr�cSs$g|]}|�d�s|�d�s|�qS)rSZvn�r�r�r)r)r*r;{s

�z/find_devs_with_dragonflybsd.<locals>.<listcomp>Trir�cSs$g|]}|�d�s|�d�r|�qS�r�Zacdr�r�r)r)r*r;�s

)zLABEL=CONFIG-2r�cSs$g|]}|�d�s|�d�s|�qSr�r�r�r)r)r*r;�s

�zUnexpected criteria: %scSsg|]}d|�qSr�r)r�r)r)r*r;�s)rr�r/r(rDrx)r�r�r�r�rir1r�r)r)r*�find_devs_with_dragonflybsdws���r�c

CsFt�rt|||||�St�r,t|||||�St�rBt|||||�St�rXt|||||�Sdg}g}|rt|�d|�|r�|�d|�|r�|�	ddg�|r�|�d|�|r�|�|�||}zt
j
|ddgd	�\}}	Wn:t
jk
�r}
z|
jt
k�rd
}n�W5d}
~
XYnXg}|��D]}|��}|�r"|�|��q"|S)z�
    find devices matching given criteria (via blkid)
    criteria can be *one* of:
      TYPE=<filesystem>
      LABEL=<label>
      UUID=<uuid>
    �blkidz-t%sz-s%s�-c�	/dev/nullz-o%srr!r�r�N)r�r�r�r�r�r�r�r�r�r�rrGr�rr@r0)
r�r�r�r�riZ
blk_id_cmdZoptions�cmdr1�_errryr�rIr)r)r*�find_devs_with�sN
�
r�c	Cs�|dkrg}nt|�}dddg}|r2|�ddg�|�|�tj|ddd	�}i}|j��D],}|�d
�\}}}t|�||<|||d<qZ|S)z�Get all device tags details from blkid.

    @param devs: Optional list of device paths you wish to query.
    @param disable_cache: Bool, set True to start with clean cache.

    @return: Dict of key value pairs of info for the device.
    Nr��-oZfullr�r�Tr�)r-rRr6ZDEVNAME)r�r�rr/r@rAr	)	ZdevsZ
disable_cacher�r1�retrIr�rr:r)r)r*r��s

r�c
Cs:t�d||�t|d��}|�|�W5QR�SQRXdS)NzPeeking at %s (max_bytes=%s)r�)rDrxr�r�)rJZ	max_bytes�ifhr)r)r*�	peek_file�sr�cCs(g}|D]}||krqq|�|�q|SrO)r�)Zin_listZout_listr�r)r)r*r��sr�c
Cs�t�d||�t��}z(t|d��}t|||d�W5QRXWn4tk
rr}z|sV�|jtkrb�W5d}~XYnX|�	�}t�dt
|�|�|r�t|�S|SdS)NzReading from %s (quiet=%s)r�)�chunk_cbzRead %s bytes from %s)rDrxr�r�r��pipe_in_outr�r�r�getvaluerCrU)rJZread_cbr�rRZofhr�ryrhr)r)r*r��s
r�c
Cs�t�r\z td�}|�dd�dd�}Wq�tk
rX}zt�d|�d}W5d}~XYq�Xn*ztd���}Wntk
r�d}YnX|S)Nz/proc/1/cmdline�r)r�z"failed reading /proc/1/cmdline: %sr�z
/proc/cmdline)�is_containerr�r�r�rDrEr0)rhruryr)r)r*�_get_cmdlines
r�cCsdtjkrtjdSt�S)NZDEBUG_PROC_CMDLINE)r%r1r�r)r)r)r*rx%s

rx�cCsNd}|�|�}t|�dkrqBq|�|�|t|�7}|r||�q|��|Sr�)r�rCr�r�)Zin_fhZout_fhZ
chunk_sizer�Zbytes_pipedr:r)r)r*r�,s


r�cCs6|dkr|dkrdSt�d|||�t�|||�dS)N)Nr�z%Changing the ownership of %s to %s:%s)rDrxr%�chown)rJ�uid�gidr)r)r*�	chownbyid;sr�c
Csnd}d}z$|rt�|�j}|r*t�|�j}Wn0tk
r\}ztd|�|�W5d}~XYnXt|||�dS)Nr�zUnknown user or group: %s)	�pwd�getpwnam�pw_uidr#r$r%r&rwr�)rJ�userrr�r�ryr)r)r*�chownbynameCs r�c
Cstddg}|rd|kr|S|d}||kr2||}nd|kr>|S|d}t|t�rX||g}t|t�r�t|�dkrz|d|d<t|�dkr�|d|d<t|t�r�d|kr�|d|d<d|kr�|d|d<|ddkr�|d|d<ddd	g}tt|��D]|}||�sq�||��}d
}|D]8}	|�|	��rd|	|t|	�d���f}d}�qP�q|�sfdd|��f}|||<q�|S)
N�output�allrr�r�z&1r+r*r.Fz%s %sT)	rPrQr�rCrHr�ryr�r0)
rrr�ZoutcfgZmodecfgZswlistr�rKr
�sr)r)r*rYsL







rcCs�g}|rt|t�s|S|�d�}|r.|�|�t|d�D]h}|sBq8t�d|�}|sTq8|�d�}|��}t	|�dkr~|�|�q8ddg|dd�kr8|�|d�q8t
t|��S)	z�Return a list of log file paths from the configuration dictionary.

    @param cfg: The cloud-init merged configuration dictionary.
    Zdef_log_fileNz (?P<type>\||>+)\s*(?P<target>.*)r.r��teez-ar!)rPrHr
r�rr�r�rr(rCr�r�)rZlogsZdefault_log�fmtr�r.�partsr)r)r*�get_config_logfiles�s&


r�cGsB|r|j|f|��t��}|dkr(d}|j|f|�d|i�dS)N)NNN�exc_info)rEr�r�rx)rrcr�r�r)r)r*r��sr�cCs<t�|�}|�t|��|��}|dk	r4|d|�S|SdSr�)�hashlib�new�updaterYZ	hexdigest)rSZroutineZmlenZhasherZdigestr)r)r*�	hash_blob�s
r�cCs0zt�|�rWdSWntk
r*YdSXdSrm)r�r�r&�r�r)r)r*�is_user�s


r�cCs0zt�|�rWdSWntk
r*YdSXdSrm)r#r$r&r�r)r)r*�is_group�s


r�cCst�d||�t�||�dS)NzRenaming %s to %s)rDrxr%�rename��src�destr)r)r*r��sr���cCs|D]}t||�qdSrO)�
ensure_dir)�dirlistrrXr)r)r*�ensure_dirs�sr�cCsHt�t|��}t|t|��sDd�dd�|D��}td|t|�f��|S)Nz, cSsg|]}t|��qSr))rQ)r7�tr)r)r*r;�szload_json.<locals>.<listcomp>z((%s) root types expected, got %s instead)�json�loadsrUrPr"rFr5�type)rXZ
root_typesr]Zexpected_typesr)r)r*�	load_json�s
��r�cCs8zd�t|��WStk
r2d�t|��YSXdS)z1Handler for types which aren't json serializable.z
ci-b64:{0}z)Warning: redacted unserializable type {0}N)rDr_�AttributeErrorr�)Z_objr)r)r*�json_serialize_default�sr�cCstj|dddtd�S)z%Return data in nicely formatted json.r�T)r<z: )�indentZ	sort_keysZ
separatorsr�)r��dumpsr�r9r)r)r*�
json_dumps�s�r�cCs<tj�|�}|tj�|�kr |Stj�|�r0|St|�SdS)z�Get the last directory in a path that does not exist.

    Example: when path=/usr/a/b and /usr/a does not exis but /usr does,
    return /usr/a
    N)r%ri�dirnamerW�get_non_exist_parent_dir)riZp_pathr)r)r*r�sr�c	Cs�tj�|�s�t|�}ttj�|�dd��t�|�W5QRXt||�|sP|r�t|||�t	|�
|�ddd��}t	|�}|jD]}|�|�}t|||�q�n
t||�dS)NTrqr�)
r%rirWr�rer��makedirs�chmodr�r
r(r�Zjoinpath)rirr�rZnon_existed_parent_dirZsub_relative_dirZsub_pathrbr)r)r*r�s


r�c	cs(z
|VW5|r"d|g}t�|�XdS)N�umount)r)r�Z
umount_cmdr)r)r*�	unmounter+s

r�c
	Csi}z�tj�d�r$td���}d}nt�d�}|j��}d}d}|D]�}zV|dkrh|��\}}}}	}
}n4t�	||�}|�
d�}|�
d�}|�
d�}|�
d�}	Wntk
r�YqDYnX|�d	d
�}|||	d�||<qDt
�d||�Wn$ttfk
�r
tt
d
�YnX|S)Nz/proc/mountsr9�mountz*^(/dev/[\S]+) on (/.*) \((.+), .+, (.+)\)$r�r!r�r�z\040r))�fstype�
mountpoint�optszFetched %s mounts from %szFailed fetching mount points)r%rir�r�r@rr/r(r��searchrr�r�rDrxr�rwr�)
�mounted�
mount_locs�methodr1ZmountreZmpliner�ZmprrZ_freqZ_passno�mr)r)r*�mounts5s:





�r	c
Cs t|t�r|g}n:t|ttf�r*t|�}n"|dkr8d}ntdjt|�d���t�rb|dkr�dg}nRt�r�|dkrzdddg}t	|�D](\}}|dkr�d||<|d	kr�d||<q�nd
g}t
�}t����R}d}	t
j�|�|kr�|t
j�|�d}
n�d}|D]�}d}
zTd
ddg}|�r |�d|g�|�|�|�|�tj||d�|}	|}
W�q�Wq�ttfk
�r�}
z t�d||d�|�|
�|
}W5d}
~
XYq�Xq�|
�s�td|||f��|
�d��s�|
d7}
t|	��<|dk�r�||
�}n
||
|�}|W5QR�W5QR�SQRXW5QRXdS)a2
    Mount the device, call method 'callback' passing the directory
    in which it was mounted, then unmount.  Return whatever 'callback'
    returned.  If data != None, also pass data to callback.

    mtype is a filesystem type.  it may be a list, string (a single fsname)
    or a list of fsnames.
    Nz6Unsupported type provided for mtype parameter: {_type})r��auto�ufsZcd9660Zmsdosr�)r�Zmsdosfsr�Frrr�Zroz-t)Z
update_envzbFailed to mount device: '%s' with type: '%s' using mount command: '%s', which caused exception: %sr)z#Failed mounting %s to %s due to: %sr)rPrQr�r"r5rDr�r�r��	enumerater	rZtempdirr%rirsr�r�rr�rwrDrxrFr~rjr�)r��callbackr:ZmtypeZupdate_env_for_mountZmtypes�indexrZtmpdr�rZfailure_reasonZmountcmd�excr�r)r)r*�mount_cb]s~

��



�	��



rcCs
t�t�SrO)�obj_copy�deepcopyrr)r)r)r*�get_builtin_cfg�srcCst�d|�tj�|�S)NzTesting if a link exists for %s)rDrxr%ri�islinkrGr)r)r*�is_link�srcCsht�d||�|rXtj�|�rXtj�tj�|�dtd��}t�||�t�	||�dSt�||�dS)Nz$Creating symbolic link from %r => %rZtmpr�)
rDrxr%rirrrFr�r��symlinkr�)r\�linkZforceZtmp_linkr)r)r*�sym_link�src
CsPt�d|�zt�|�Wn0tk
rJ}z|jtkr:|�W5d}~XYnXdS)NzAttempting to remove %s)rDrxr%�unlinkrwr�r)riryr)r)r*�del_file�s
rcCst�d||�t�||�dS)NzCopying %s to %s)rDrxrF�copyr�r)r)r*r�srcCs2zt�dt���}Wntk
r,d}YnX|S)Nz%a, %d %b %Y %H:%M:%S %z�??)�time�strftime�gmtimer�)Ztsr)r)r*�time_rfc2822�s

r cs�ddl�ddl�d}G�fdd�d�j�}���j�d��}���}��|�|_|�}|�	d|��
|���
|�dd�dkr�|j|jd	St
d
��dS)z�Use sysctlbyname(3) via ctypes to find kern.boottime

    kern.boottime is of type struct timeval. Here we create a
    private class to easier unpack it.

    @return boottime: float to be compatible with linux
    rN�cs eZdZd�jfd�jfgZdS)zboottime.<locals>.timeval�tv_sec�tv_usecN)r{r|r}Zc_int64Z_fields_r)��ctypesr)r*�timeval�sr&�cs
kern.boottimer�g��.Az/Unable to retrieve kern.boottime on this system)r%Zctypes.utilZ	StructureZCDLL�utilZfind_libraryZc_size_tZsizeof�valueZsysctlbynameZbyrefr"r#�RuntimeError)Z
NULL_BYTESr&ZlibcrCr�r)r$r*�boottime�s(	���
r+cCstd}d}zDtj�d�r4d}td�}|rJ|��d}nd}tt��t��}Wn"tk
rnt	t
d|�YnX|S)Nrrz/proc/uptimerr%z&Unable to read uptime using method: %s)r%rir�r�r(rQrr+r�r�rD)Z
uptime_strrrhr)r)r*�uptimesr,cCst||ddd�dS)Nr,)�omoder��
write_file)ri�contentr)r)r*�append_filesr1�)�
preserve_mode)rr3r�cCst|dd||d�dS)Nr�r,)r0r-rr3r.)rirr3r)r)r*�ensure_files�r4c	Cs*z
t|�WSttfk
r$YdSXdSrO)r$r_r5)Zpossible_intr)r)r*�safe_int#s
r5c	Cs4t|�}|r0|r0t|��t�||�W5QRXdSrO)r5rer%r�)rirZ	real_moder)r)r*r�*s
r�)�grp_namer�cCs:d}zt�|�j}Wn tk
r4t�d|�YnX|S)zt
    Returns the group id of a group name, or -1 if no group exists

    @param grp_name: the name of the group
    r�z"Group %s is not a valid group name)r#r$r%r&rDrx)r6r�r)r)r*�get_group_id1sr7)rir�cCst�t�|�j�S)z�
    Returns the octal permissions of the file/folder pointed by the path,
    encoded as an int.

    @param path: The full path of the file/folder.
    )ru�S_IMODEr%�st_moderGr)r)r*�get_permissions?sr:cCst�|�}t�|j�jS)zw
    Returns the owner of the file/folder pointed by the path.

    @param path: The full path of the file/folder.
    )r%rur��getpwuid�st_uid�pw_name�ri�str)r)r*�	get_ownerJs
r@cCst�|�}t�|j�jS)zw
    Returns the group of the file/folder pointed by the path.

    @param path: The full path of the file/folder.
    )r%rur#�getgrgid�st_gid�gr_namer>r)r)r*�	get_groupTs
rD)�usernamer�cCsJg}t��D]}||jkr|�|j�qt�|�j}|�t�|�j�|S)zp
    Returns a list of all groups to which the user belongs

    @param username: the user we want to check
    )	r#ZgetgrallZgr_memr�rCr�r��pw_gidrA)rE�groupsrr�r)r)r*�get_user_groups^s
rHr-)�ensure_dir_existsr�rc
Cs�|r&zt|�}Wntk
r$YnX|r@ttj�|�||d�d|��krZt|�}d}nt|�}d}zd|}	Wnt	k
r�d|}	YnXt
�d|||	t|�|�t
|d��.t||��}
|
�|�|
��W5QRXW5QRXt||�d	S)
a)
    Writes a file with the given content and sets the file mode as specified.
    Restores the SELinux context if possible.

    @param filename: The full path of the file to write.
    @param content: The content to write to the file.
    @param mode: The filesystem mode to set on the file.
    @param omode: The open mode used when opening the file (w, wb, a, etc.)
    @param preserve_mode: If True and `filename` exists, preserve `filename`s
                          current mode instead of applying `mode`.
    @param ensure_dir_exists: If True (the default), ensure that the directory
                              containing `filename` exists before writing to
                              the file.
    @param user: The user to set on the file.
    @param group: The group to set on the file.
    )r�r�brVZ
charactersz%oz%rzWriting to %s - %s: [%s] %s %srGN)r:rwr�r%rir�r�rYrUr5rDrxrCrer�r�r�r�)r�r0rr-r3rIr�rZ
write_typeZmode_rZfhr)r)r*r/ns:�
r/cCs@t�|�D]0}tj�||�}tj�|�r2t|�q
t|�q
dS)z�
    Deletes all contents of a directory without deleting the directory itself.

    @param dirname: The directory whose contents should be deleted.
    N)r%rorirFrWrHr)r�ZnodeZ
node_fullpathr)r)r*�delete_dir_contents�s

rKr��createdcCs6t��}t|�}|d|��|f7}|dt�7}|S)Nz %s by cloud-init v. %sz on %s)rZversion_stringrQ�titler )Zcomment_charrPZci_ver�headerr)r)r*�make_header�s
rOcGstj�tjj|f|���SrO)r%ri�abspathrF)rPr[r)r)r*�abs_join�srQc	Cs�t|ttf�s tdt�|���d}|r0|d7}dd}d}|D]�}t|ttf�r�g}|D]}|�dt|��d|��qZd	|d
�	|�f}|d7}q@t|t�r�d	||f}|d7}q@|dkr�q@tdt�|�|f��q@t
�d
|�|S)Nz8Input to shellify was type '%s'. Expected list or tuple.r�z
#!/bin/sh
z%s%s%s%s)�'�\rRrRrz'%s'rRz%s%s
r)r�zCUnable to shellify type '%s'. Expected list, string, tuple. Got: %szShellified %s commands.)rPr"r�r5rr�r�rQr�rFrDrx)ZcmdlistZ
add_headerr0ZescapedZ	cmds_mader�Zfixedrlr)r)r*�shellify�s<��


��rTcCsB|r|�|�r|t|�d�}|r>|�|�r>|dt|��}|SrO)r�rCrj)rI�prefix�suffixr)r)r*�strip_prefix_suffix�s
rWcCsBt�|d�dkrdSzt�|�Wntjk
r<YdSXdS)NrFT)r�whichrG)r�r)r)r*�_cmd_exits_zero�srYcCstdddg�S)Nzsystemd-detect-virtz--quietz--container�rYr)r)r)r*�_is_container_systemd	sr[cCs
tdg�S)Nzlxc-is-containerrZr)r)r)r*�_is_container_old_lxc	sr\cCsDt�s
dSdddg}t�|d�dkr*dSt�|�\}}|��dkS)NFr�z-qnzsecurity.jail.jailedrr)r�rrXr0)r�r1rr)r)r*�_is_container_freebsd	s
r]c	Cs�tttf}|D]}|�rdSqz(td�}d|kr8WdSd|krFWdSWnttfk
r`YnXtj�d�r~tj�d�s~dSzHt	d��
�}|D]2}|�d�r�|���
d	d�\}}|d
kr�WdSq�Wnttfk
r�YnXdS)zH
    Checks to see if this code running in a container of some sort
    Tr�Z	containerZLIBVIRT_LXC_UUIDz/proc/vzz/proc/bcz/proc/self/statuszVxID:r6rF)r[r]r\�get_proc_envr�rwr%rirWr�r@r�r0r()Zchecks�helperZpid1env�linesrIZ_keyrKr)r)r*r�	s6�

r�cCstj�d�S)z2Check to see if we are running in a lxd container.z
/dev/lxd/sockr�r)r)r)r*�is_lxd@	srar�c	Cs�tj�dt|�d�}zt|dd�}Wnttfk
r@iYSXi}d\}}|rfd\}}|�||�}|�|�D]&}|szqp|�|d�\}	}
|	rp|
||	<qp|S)aH
    Return the environment in a dict that a given process id was started with.

    @param encoding: if true, then decoding will be done with
                     .decode(encoding, errors) and text will be returned.
                     if false then binary will be returned.
    @param errors:   only used if encoding is true.z/procr1Fr`)r!�=)r��=r�)	r%rirFrQr�r�rwrRr()�pidrT�errorsr�rh�envZnullZequalrr�rKr)r)r*r^E	s"

r^c	CsPi}|��D]>}z|�dd�\}}Wntk
r@|}d}YnX|||<q|S)Nrcr�T)r(r_)Zkvstringr�rr�rKr)r)r*�keyval_str_to_dictc	s

rgcCs&|�d�r|dd�}tj�d|�S)Nr�rIz/sys/class/block/%s/partition)r�r%rir�)r�r)r)r*�is_partitiono	s
rhcCs�t|t�s|g}g}|D]�}t|t�r2|�|�qt|ttf�r�t|�dksXt|�dkr`td��t|�dkr�|dr�|�|t|��q|�|d�qtd��q|S)Nr�r!z Invalid package & version tuple.rzInvalid package type.)rPr�rQr�r"rCr*)Zversion_fmtZpkgsZpkglist�pkgr)r)r*�expand_package_listv	s 



rjc	Cs�dd�|�d�D�}d}d}d}d}t|�D�]D\}	}
|
��}t|�dkrh|�d|	dt|�|
�dS|d}d	d�|�d�D�}
t|
�t|�kr�q,tt|
�t|��}|
d
|�|d
|�kr�q,|dk	r�t|�t|
�kr�q,z|�d�}	Wn,tk
�r|�d|	d|
�YdSXz||	d}||	d
}Wn,tk
�r`|�d|	d|
�YdSX|}|
}|d}q,|�r�|�r�|�r�|�r�|�r�||||fSn|�r�|�r�|�r�|||fSdS)zRReturn the mount information for PATH given the lines from
    /proc/$$/mountinfo.cSsg|]}|r|�qSr)r)�r7ryr)r)r*r;�	sz$parse_mount_info.<locals>.<listcomp>rNrdz$Line %d has two few columns (%d): %sr�r�cSsg|]}|r|�qSr)r)rkr)r)r*r;�	sr�-z,Did not find column named '-' in line %d: %sr!z/Too few columns after '-' column in line %d: %srI)r(rrCrx�minrr_�
IndexError)riZmountinfo_linesr�get_mnt_optsZ
path_elements�devpth�fs_typeZmatch_mount_pointZmatch_mount_point_elementsr�rIr��mount_pointZmount_point_elements�xZ
mount_optionsr)r)r*�parse_mount_info�	sp�
���
�


rtcCsBtd���D]0}|��dd�\}}}||kr|||fSqdS)z<On older kernels there's no /proc/$$/mountinfo, so use mtab.�	/etc/mtabNr�)r�r@r()rirIrprrrqr)r)r*�
parse_mtab�	s
rvcCs�|�d�}t|�dkr|dS|ddkr�|dd�}t�dddg�\}}|�d	�D]0}|��}t|�d
krT|d
|krT|d}q�qTt|�St�d|�dS)Nrr�r!)r�ZgptrrIZglabel�statusz-sr�rz)Unexpected input in find_freebsd_part: %s)r(rCrrQrDrE)�fs�splittedZtarget_labelrbr��labelsr�r)r)r*�find_freebsd_part�	s
r{cCs:|�d�}t|�dkr*|ddkr*|dSt�d|�dS)Nrr�r�r�r!z.Unexpected input in find_dragonflybsd_part: %s)r(rCrDrE)rxryr)r)r*�find_dragonflybsd_part�	s
r|cCsHd}|�d�D]4}|��}t|�dkrtj�|d|�r|}qDq|S)Nr�r!r�)r(rCr%rir�)ri�mnt_list�
path_foundrIr�r)r)r*�get_path_dev_freebsd
s rcCsztjdd|gddgd�\}}t|�rPt�ddg�\}}t||�}|dkrLdS|}|��}t|d�}d||d|dfS)Nrz-prr�r�r�r!)rrCrr(r{)rir�rMr}r~r�Z
label_partr)r)r*�get_mount_info_freebsd
s
r�c
Cs�tj�d�st�d�dSzt�dd|g�\}}Wn8tjk
rj}zt�d||�WY�dSd}~XYnXt|�rxdSd}|�	d�D]B}t
�||�r�||kr�d|kr�|�	�d	}t�d
||�|Sq�dS)Nz/dev/zfsz"Cannot get zpool info, no /dev/zfs�zpoolrwz$Unable to get zpool status of %s: %sz.*(ONLINE).*r��staterzfound zpool "%s" on disk %s)r%rir�rDrxrrGrErCr(r�r)r�ZzpoolstatusrMr�rIZdiskr)r)r*�get_device_info_from_zpool
s 
r�cCs�t�dg�\}}|��}d}t�r&d}|D]�}t�||�}|s@q*|�d�}|�d�}|�d�}	|	dkrp|�d�}	t�d|||	�t�d	|�}
|
s�t�r�|	d
kr�t	|�S||kr*||	|fSq*dS)NrzH^(/dev/[\S]+|.*zroot\S*?) on (/[\S]*) (?=(?:type)[\s]+([\S]+)|\(([^,]*))zA^(/dev/[\S]+|\S*?) on (/[\S]*) (?=(?:type)[\s]+([\S]+)|\(([^,]*))r�r!r�r�z?found line in mount -> devpth: %s, mount_point: %s, fs_type: %sz^(/dev/.+)p([0-9])$Zzfs)
rr@r�r�rrrDrxr�r�)riZmountoutputr�rZregexrIrrprrrqZdevmr)r)r*�parse_mount-
s8��



�r�cCsRdt��}tj�|�r2t|���}t||||�Stj�d�rFt|�St|�SdS)Nz/proc/%s/mountinforu)	r%�getpidrir�r�r@rtrvr�)rirroZmountinfo_pathr`r)r)r*�get_mount_info[
sr�)�optr�cCs t|dd��^}}||�d�kS)NT�ror<�r�r()rir�rZmnt_optsr)r)r*�
has_mount_opt�
sr��T.)r�r�cCs�|dkrg}|dkri}t��}d}|rLztt��}Wntk
rJYnXz|||�}W5t��|}d}	|dk	r�ztt��|}	Wntk
r�YnXd|}
|r�t|	t�r�|
d|	7}
n|
d7}
z|||
�Wntk
r�YnXX|S)Nz took %0.3f secondsz (%0.2f)z (N/A))r�floatr,r_rPr�)r�rcr�r�r�Z
get_uptime�startZustartZdeltaZudeltaZtmsgr�r)r)r*r��
s<
r�cCs(|�dd�}t|�dkr|S|dfSdS)Nr r�)�rsplitrC)Zdottedr�r)r)r*�expand_dotted_devname�
sr�cCs�|dkrg}|dkrg}g}i}||D]d}zt|||ddd�||<Wq(tk
r�}z"|jtkrh�||krz|�|�W5d}~XYq(Xq(t|�r�tdjd�|�d���|S)NF)r�rRzMissing required files: {files}r<)�files)	r�r�r�rr�rCr_rDrF)rPZrequiredZoptionalZdelimrLr�rlryr)r)r*�pathprefix2dict�
s&
�r��
/proc/meminfoc		Cs�ddddd�}dddd	�}i}t|���D]v}z|��\}}}Wn$tk
rd|��\}}d
}YnX|r�t|�||||<q*||kr*t|�|||||<q*|S)Nr��r��@)ZkBZmB�BZgBZtotalZfreeZ	available)z	MemTotal:zMemFree:z
MemAvailable:r�)r�r@r(r_r$)	Zmeminfo�raw�mpliersZkmapr�rIr�r)Zunitr)r)r*�read_meminfo�
s"�
r�c
Cs�|}|�d�r|dd�}n|�d�r2|dd�}dddd	d
d�}|}d}|D]$}|�|�rN|}|dt|��}qNzt|�}Wn0tk
r�}ztd
|�|�W5d}~XYnX|dkr�td|��t|||�S)a}Convert human string or integer to size in bytes

    In the original implementation, SI prefixes parse to IEC values
    (1KB=1024B). Later, support for parsing IEC prefixes was added,
    also parsing to IEC values (1KiB=1024B). To maintain backwards
    compatibility for the long-used implementation, no fix is provided for SI
    prefixes (to make 1KB=1000B may now violate user expectations).

    Future prospective callers of this function should consider implementing a
    new function with more standard expectations (1KB=1000B and 1KiB=1024B)

    Examples:
    10M => 10485760
    10MB => 10485760
    10MiB => 10485760
    ZiBN���r�r�r�r�r�r�l)r��K�M�Gr�rz'%s' is not valid input.z'%s': cannot be negative)rjrCr�r_r$)rCZsize_inr�ZnumZmplierrryr)r)r*�human2bytes�
s&


 r�cCs<|dkrt��d}|dkp6|ddko6|dd�dk}|S)z$Return True if platform is x86-basedNr�Zx86_64rr�r!Z86)r%r&)Z
uname_archZx86_arch_matchr)r)r*�is_x86s�r�cCs,tjdd�dkr"t�t�|��St�|�S)Nr!)r!�)r��version_info�emailZmessage_from_filer��StringIO�message_from_string)r�r)r)r*r�(sr�c	Cs�tjddg|dd�}t�}|j��D]\}z|�dd�\}}}Wntk
rVYq$YnX|�d�sl|�d�r$|�t�	dd	|��q$|S)
Nz
dpkg-queryz--listT)r.r-r!�hiZiiz:.*r�)
rr�r/r@r(r_r�r�r��sub)r.r1Z	pkgs_instrIr�rirr)r)r*�get_installed_packages.s
r�c
Cs�d}z.tt|dd��}|�dd���dkr0WdSWn0tk
rb}zt�d||�W5d}~XYnXt�}d|krvdStd	dd�}d|��kr�dStj	�
d
�r�dSdS)NrT�r�r�r�zubuntu-corez!Unexpected error loading '%s': %sz
snap_core=z/etc/system-image/channel.iniz/etc/system-image/config.d/F)r	r�r
r�r_rDrErxr%rirW)ZorpathZorinforyrur0r)r)r*�system_is_snappy=s 
 r�cCs�d}|��D]}|�d�r|dd�}q,q|dkr8dS|�d�rF|S|�d�rdd|td�d�S|�d�r�d|td�d���S|�d�r�d	|td�d���}tj�|�r�|St|�}|r�|d
S|Sd|S)Nzroot=rIr�r�z/dev/disk/by-label/zUUID=z/dev/disk/by-uuid/z	PARTUUID=z/dev/disk/by-partuuid/r)r(r�rCr�r%rir�r�)rur
rZ
disks_pathZresultsr)r)r*�rootdev_from_cmdlineUs.




�r�cCsFdd�}i}||�D],}|�dd�\}}|s0|}|s8|r|||<q|S)z�Given shell like syntax (key=value
key2=value2
) in content
    return the data in dictionary form.  If 'add_empty' is True
    then add entries in to the returned dictionary for 'VAR='
    variables.  Set their value to empty_val.cSstj|dd�S)NT)Zcomments)�shlexr()rSr)r)r*�_shlex_splitzsz(load_shell_content.<locals>._shlex_splitrcr�)r()r0Z	add_emptyZ	empty_valr�r:rIr�r)r)r)r*r	ts
r	��?cCs�t|�}d}|tdd�|D��8}t|�dkrBt�d|||�gS|dkrZt�d|||�|||krhq|t�|�||7}qt�d|||�|S)NrcSsg|]}tj�|�r|�qSr)r�rkr)r)r*r;�sz"wait_for_files.<locals>.<listcomp>z)%sAll files appeared after %s seconds: %sz6%sWaiting up to %s seconds for the following files: %sz*%sStill missing files after %s seconds: %s)r�rCrDrxr�sleep)ZflistZmaxwaitZnaplenZlog_preZneedZwaitedr)r)r*�wait_for_files�s:��

�r�cCs&t|dd�}|d�d�}|ddkS)z1Check whether the given mount point is mounted rwTr�r�r<rZrwr�)rrr�Z
mount_optsr)r)r*�mount_is_read_write�sr�cCsXt�d�sdSddg}|r:tj�|�r*dS|�d|g�|rN|�d|g�t�|�S)zAInvoke udevadm settle with optional exists and timeout parametersZudevadmNZsettlez--exit-if-exists=%sz--timeout=%s)rrXr%rir�r�)r�rRZ
settle_cmdr)r)r*�udevadm_settle�s
r�c
Cs�d}zJtd|dd�}|rLt�dt|��}|r>t|�d��}nt�d||�Wn0tk
r~}zt�d||�W5d	}~XYnX|S)
z-
    Return the parent pid of a process.
    rz
/proc/%s/statTr�z ^\d+ \(.+\) [RSDZTtWXxKPI] (\d+)r�z6Unable to match parent pid of process pid=%s input: %sz Failed to load /proc/%s/stat. %sN)	r�r�rrQr$rrDrEr�)rdZppidrhrryr)r)r*�
get_proc_ppid�s� r�r��	Error:
{}cCs&t|�|�tjd�|r"t�|�|S)a
    Print error to stderr and return or exit

    @param msg: message to print
    @param rc: return code (default: 1)
    @param fmt: format string for putting message in (default: 'Error:
 {}')
    @param sys_exit: exit when called (default: false)
    )�file)�printrDr�r��exit)rcZrcr�Zsys_exitr)r)r*r��s	
r�cs^eZdZd�fdd�	Zeed�dd��Zdd�Zd	d
�Zdd�Z	d
d�Z
ed�dd�Z�Z
S)�Versionr�cstt|��|||||�S)zPDefault of -1 allows us to tiebreak in favor of the most specific
        number)�superr��__new__)�cls�major�minor�patch�rev��	__class__r)r*r��szVersion.__new__)rcCs|ttt|�d����S�Nr )r�r#r$r()r�rr)r)r*�from_str�szVersion.from_strcCsd|�|�kS)Nr�)�_compare_version�rk�otherr)r)r*�__gt__�szVersion.__gt__cCs0|j|jko.|j|jko.|j|jko.|j|jkSrO�r�r�r�r�r�r)r)r*�__eq__�s
�
�
�zVersion.__eq__ccs6|j|j|j|jfD]}|dkr,t|�Vqq2qdS)z)Iterate over the version (drop sentinels)r�N)r�r�r�r�rQ)rk�nr)r)r*�__iter__�szVersion.__iter__cCs
d�|�Sr�)rFror)r)r*�__str__szVersion.__str__r�cCsP||krdS|j|jkrdS|j|jkr,dS|j|jkr<dS|j|jkrLdSdS)z�
        return values:
            1: self > v2
            -1: self < v2
            0: self == v2

        to break a tie between 3.1.N and 3.1, always treat the more
        specific number as larger
        rr�r�r�r�r)r)r*r�s
zVersion._compare_version)r�r�r�r�)r{r|r}r��classmethodrQr�r�r�r�r�r$r��
__classcell__r)r)r�r*r��sr�r�r�r�r�)N)N)rN)rN)N)N)N)r�N)N)TT)F)N)r)N)N)N)NN)F)r�r�rI)N)r�r�rIrdr)N)N)F)r�)Nr�NFN)Nr�NFN)Nr�NFN)Nr�NFN)Nr�NFN)NF)NFT)r�N)NN)NN)N)r�)NNN)NNN)F)r2)r2r-F)r�rL)T)NN)rNr�)NNF)r�F)N)N)FN)r�r�)NN)r�r�F)�r�rrr�r�r#r�r�r�r�r%�os.pathr�r�r�r�r�rFr�rur�r2r�r�base64rr�collectionsrrr�rr�	functoolsrr	�pathlibr
�typingrrr
rrZurllibrZ	cloudinitrrrr�rrrrrrrZcloudinit.settingsrr�Z	getLoggerr{rD�sepr�r�r�r�r�r�r+r2r5rUrYr^r_rdrer�r~r�r�r�r�r�r�r�r�r�r�r�r�r�rQrHr�r��DEBUGr�r�r�r�r�r�r�r�r�r�rrrr�rrr rr=�contextmanagerr?r"rErHrVr\r�rMrqrsrwrvr~rr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rxr�r�r�rr�r�r�r�r�r�r�r�r�r�r�r�r�r	rrrrrr r+r,r1r$r�r4r5r�r7r:r@rDrHr/rKrOrQrTrWrYr[r\r]r�rar^rgrhrjrtrvr{r|rr�r�r�r�r�r�r�r�rir�r�r�r�r�r�r�r�r	r�r�r�r�r�r�r)r)r)r*�<module>s$	
�
.
2
�
,





	(
=8

Q
	
	
!(+,"�+(/
�
�
�
�
�
@

:
	)�
X	
"���

��=%

-N	
.'
��.+