%PDF- %PDF- Mini Shell
Mini Shell

Direktori : /snap/core18/current/usr/lib/python3/dist-packages/cloudinit/__pycache__/
Current File : //snap/core18/current/usr/lib/python3/dist-packages/cloudinit/__pycache__/util.cpython-36.pyc
3

�Ad�i�
@s�	ddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZ
ddlZddl
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.m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;da<e2j=e>�Z?e
j@d
iZAdejBejCZD�dvZE�dwZFdd�ZGe"��dxdd��ZHe"��dydd��ZI�dzdd�ZJ�d{d d!�ZKd"d#�ZLd$d%�ZMd&d'�ZNGd(d)�d)�ZOGd*d+�d+eP�ZQGd,d-�d-eP�ZRd.d/�ZS�d|d0d1�ZT�d}d2d3�ZU�d~d4d5�ZV�dd7d8�ZW�d�d9d:�ZXdd;�e)d<�d=d>�ZYd?d@�ZZdAdB�Z[dCdD�Z\�d�dFdG�Z]dHdI�Z^e_e`dJ�dKdL�ZadMdN�ZbdEdEde2jcdEfdOdP�Zde"�dQdR��Zee"�dSdT��Zfe"�dUdV��Zge"�dWdX��Zhe"�dYdZ��Zie"�d[d\��Zj�d�d^d_�Zk�d�d`da�Zl�d�dbdc�Zm�d�ddde�Zne"�dfdg��Zodhdi�Zpe"�djdk��Zq�d�dldm�Zr�d�dndo�Zsdpdq�Zt�d�drds�Zu�d�e`d<�dtdu�Zvejwdvdw��Zxejwdxdy��Zydzd{�Zzd|d}�Z{�d�d�d��Z|�d�d�d��Z}de`ffd�d��Z~�d�d�d��Zdd;�e`d<�d�d��Z�dd;�e`d<�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�d̈́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�d]d��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�dEdd�d��d�d�Z��d�d�Z��d��d�d�Z��d�d�Z��d��d�d�Zd��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ːd0�d1�Z̐d2�d3�Z�e?d]f�d4�d5�Zΐd6�d7�Zϐd8�d9�ZАd:�d;�Zѐd<�d=�ZҐd>�d?�ZӐd@�dA�ZԐdB�dC�Z�e?d]f�dD�dE�Z�e_e��dF��dG�dH�Z�e+�dI�Zؐd�e'�dJe�feؐdK��dL�dM�ZِdN�dO�Z�dde
j�j@f�dP�dQ�Zܐd��dS�dT�ZݐdU�dV�Zސd��dW�dX�ZߐdY�dZ�Z�d��d[�d\�Z�d]�d^�Z�d_�d`�Z�d��da�db�Z�d��dd�de�Z�df�dg�Z�d��dh�di�Z�dj�dk�Z�d��dn�do�Z�e#G�dp�dq��dqe�dq�dr�ds�dt�dug���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�1�on�yes�off�0�no�falsecCs"ttttj�jjd�dd���S)N�.�)�tuple�map�int�os�uname�release�split�r.r.�0/usr/lib/python3/dist-packages/cloudinit/util.py�kernel_versionDsr0cCstjddgd|d�}|jj�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)r2�outr.r.r/�get_dpkg_architectureHsr6c	s�ddddd�}i�y�tjddgd|d	�}x:|jj�D],}|jd
�\}}}||kr4|j��||<q4W�fdd�|j�D�}t|�r�tjd
dj	|��WnFtj
k
r�}z(tjd|�tdd�|j�D���WYdd}~XnX�S)N�codenameZdescription�idr,)ZCodenameZDescriptionzDistributor IDZRelease�lsb_releasez--allT)r1r2�: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.)r;�vr.r.r/�	<genexpr>nszlsb_release.<locals>.<genexpr>)rr3�
splitlines�	partitionr4�values�len�LOG�warning�join�ProcessExecutionError�dict)	r2Zfmapr5�line�fnamer�val�missing�errr.)r=r/r9Us((r9�utf-8cCst|t�r|S|j|�S)N)�
isinstance�str�decode)�blob�encodingr.r.r/�
decode_binaryss
rVcCst|t�r|S|j|�S)N)rQ�bytes�encode)�textrUr.r.r/�encode_textzs
rZcCs,t|�}y
|jd�Stk
r&|SXdS)Nzutf-8)rrS�UnicodeDecodeError)�source�decodedr.r.r/�b64d�s

r^cCs"t|t�s|jd�}t|�jd�S)Nzutf-8)rQrWrXrrS)r\r.r.r/�b64e�s

r_cCsP|jdd�}|j�dkrLt|t�rL|j�}|r<|jr<|j}nd}|j|d�S|S)NT)rSrYzutf-8�surrogateescape)Zget_payloadZget_content_maintyperQrWZget_charsetZinput_codecrS)�partZcte_payload�charsetrUr.r.r/�fully_decoded_payload�s
rcc@s&eZdZd	dd�Zdd�Zdd�ZdS)
�SeLinuxGuardFcCs<ytjd�|_Wntk
r*d|_YnX||_||_dS)N�selinux)r�
import_modulere�ImportError�path�	recursive)�selfrhrir.r.r/�__init__�szSeLinuxGuard.__init__cCs|jr|jj�rdSdSdS)NTF)re�is_selinux_enabled)rjr.r.r/�	__enter__�szSeLinuxGuard.__enter__cCs�|js|jj�rdStjj|j�s*dStjj|j�}y"tj|�}|jj||tj	�Wnt
k
rndSXtjd||j
�y|jj||j
d�Wn4t
k
r�}ztjd||j
|�WYdd}~XnXdS)Nz,Restoring selinux mode for %s (recursive=%s))riz,restorecon failed on %s,%s maybe badness? %s)rerlr*rh�lexists�realpath�lstatZmatchpathcon�stat�ST_MODE�OSErrorrF�debugriZ
restoreconrG)rjZ	excp_typeZ
excp_valueZexcp_tracebackrhZstats�er.r.r/�__exit__�s,
zSeLinuxGuard.__exit__N)F)�__name__�
__module__�__qualname__rkrmrvr.r.r.r/rd�s

rdc@seZdZdS)�MountFailedErrorN)rwrxryr.r.r.r/rz�srzc@seZdZdS)�DecompressionErrorN)rwrxryr.r.r.r/r{�sr{cOsttj�}|dkr\y|||�tjd�Wqptk
rXttdtj|��tjd�YqpXntjd|tj|��dS)Nrz&Failed forking and calling callback %s�z(Forked child %s who will run callback %s)	r*�fork�_exit�	Exception�logexcrFr�obj_namert)Zchild_cb�args�kwargsZfidr.r.r/�fork_cb�s
r�cCsBt|t�r|dkSt}|r&t|�|}t|�j�j�|kr>dSdS)NTF)rQ�bool�TRUE_STRINGS�listrR�lowerr4)rM�addons�	check_setr.r.r/�is_true�s
r�cCsBt|t�r|dkSt}|r&t|�|}t|�j�j�|kr>dSdS)NFT)rQr��
FALSE_STRINGSr�rRr�r4)rMr�r�r.r.r/�is_false�s
r�cCs |sdSt|t�r|St||�S)NF)rQr�r�)rMr�r.r.r/�translate_bools

r�� cs8tj���stjtj�dj��fdd�td|�D��S)N�csg|]}�j���qSr.)Zchoice)r;Z_x)�r�select_fromr.r/r>szrand_str.<locals>.<listcomp>r)�randomZSystemRandom�string�
ascii_letters�digitsrH�range)�strlenr�r.)r�r�r/�rand_strsr�cCs.|sd}x tdd�d|}||kr
Pq
W|S)Nr��)r�r)r�)Z
dictionaryZpostfixZnewkeyr.r.r/�
rand_dict_keysr�)�instance_data_file)�returnc Cs�ddlm}m}m}yt|�}Wn2tk
rR}z|jtkr@iS�WYdd}~XnX|r�tj	j
|�r�y||||�}tjd||�WnH|k
r�Yn6|k
r�}ztj
d||t|��WYdd}~XnX|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*rh�existsrFrtrG�repr�	load_yaml)rLr�r�r�r�Zconfig_filerur.r.r/�	read_conf#s8

r�cGstt|��S)N)�sorted�
uniq_merge)�listsr.r.r/�uniq_merge_sortedUsr�cGsJg}x<|D]4}t|t�r4|j�jd�}dd�|D�}|j|�q
Wt|�S)Nr?cSsg|]}|r|�qSr.r.)r;�ar.r.r/r>gszuniq_merge.<locals>.<listcomp>)rQrRr4r-�extend�	uniq_list)r�Z
combined_list�a_listr.r.r/r�as

r�cCslx tj�D]\}}|j||�}q
Wg}x|D]}|tkr,|j|�q,Wx|D]}|j|d�}qLW|j�}|S)Nr�)�FN_REPLACEMENTS�items�replace�
FN_ALLOWED�appendr4)�fnr<r@Zremovalsr.r.r/�clean_filenamels

r�TcCs�yLtjt|��}tjtjddd|���}|r8t|j��S|j�SWdQRXWn8t	k
r�}z|rf|St
t|��|�WYdd}~XnXdS)N�rbr|)�io�BytesIOrZ�
contextlib�closing�gzipZGzipFilerV�readrr{rR)r=�quietrS�bufZghrur.r.r/�decomp_gzipysr�cCs�|sdS|jdd�}|dj�}t|�dkr:|dj�}nd}|sX|dksX|j�dkr\d}|sv|dksv|j�dkrzd}||fS)Nr:r|rr&z-1Znone)NN)r-r4rEr�)Zug_pairZ	ug_parted�u�gr.r.r/�extract_usergroup�sr�)�root_dirr�cCslt�}x`tjtjj|d��D]H}tjj|�s.qtjj|�dd�}|j�}|r|jd�dkr|||<qW|S)Nz*.pyr�r%r|������)	rJ�globr*rhrH�isfile�basenamer4�find)r��entriesrL�modnamer.r.r/�get_modules_from_dir�sr�c	Cs,t|d��}|j|�|j�WdQRXdS)N�w)�open�write�flush)�conpathrYZwfhr.r.r/�write_to_console�s
r�c	
Cs�|rtjj|�|r�d}d}tjj|�rxyt||�d}Wn<tk
rvd}tjj|�d��|rr|j	t
j|�YnX|r�|r�tjj|�|r�|ddkr�|j	||dd��n|j	||�dS)	Nz/dev/consoleFTzFailed to write to /dev/console�
r|r�r�)�sys�stderrr�r*rhr�r�rsr3r�loggingZWARNING)	rYZconsoler�rZ	log_levelZfallback_to_stdoutr�Zwriting_to_console_workedZ
console_errorr.r.r/�	multi_log�s(


r�cCsdtj�kS)NZLinux)�platform�systemr.r.r.r/�is_Linux�sr�cCs$dtj�krdStj�dkr dSdS)NZBSDTZ	DragonFlyF)r�r�r.r.r.r/�is_BSD�s
r�cCst�ddkS)N�variant�freebsd)�system_infor.r.r.r/�
is_FreeBSD�sr�cCst�ddkS)Nr��	dragonfly)r�r.r.r.r/�is_DragonFlyBSD�sr�cCst�ddkS)Nr��netbsd)r�r.r.r.r/�	is_NetBSD�sr�cCst�ddkS)Nr��openbsd)r�r.r.r.r/�
is_OpenBSD�sr�FcCs||kr|St||�S)N)r�)�yobj�keyr�r.r.r/�get_cfg_option_bool�sr�cCs*||kr|S||}t|t�s&t|�}|S)N)rQrR)r�r�r�rMr.r.r/�get_cfg_option_str�s
r�cCstt|||d��S)N)r�)r)r�)r�r�r�r.r.r/�get_cfg_option_intsr�cCs�|sd}tjj|�siSt|�}d}d|kr0d}tj||�}|r�|j�}d|dkr`|d|d<|dj�jd�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\.]+)�namer7z linuxrzred hat enterprise�redhatr)�ID�
VERSION_ID�VERSION_CODENAME)	r*rhr�r��re�match�	groupdictr�rC)Zrelease_fileZredhat_releaseZredhat_regexr��groupr.r.r/�_parse_redhat_releases(r�c	Csnd}d}d}i}d}tjjd�r,ttd��}|s:d}t�}|r�|jdd�}|jdd�}d|ksfd|krptj�}nf|d	kr�|jd
d�}nP|dkr�|r�|jd
d�}n4|jdd�}|s�t	j
d
|jdd��}|r�|j�d}|dkr�d}n�t��rtj
�j�}tj�}nbd}z(ytj�}Wntk
�r*YnXWdd}x|D]}|�r:d}�q:W|�s^tjd�X|S|||fS)Nr�Fz/etc/os-releaseTr�r��sles�suse�photonZPRETTY_NAME�	virtuozzor�z[^ ]+ \((?P<codename>[^)]+)\)�VERSIONr7�rhelr�r|zPUnable to determine distribution, template expansion may have unexpected results)r�r�r�)r*rhr��load_shell_contentr�r��getr��machiner�r�r�r�r�r�r,�distrrFrG)	Zdistro_nameZdistro_versionZflavorZ
os_releaseZos_release_rhelr�r�found�entryr.r.r/�get_linux_distro4sZ




rcCsv|dj�}d}|dkrf|ddj�}|d*kr6|}qr|d+krDd}qr|dkrRd}qr|d,kr`d}qrd}n|d-krr|}|S).Nr��unknown�linuxrr�	almalinux�alpine�arch�centos�
cloudlinux�debian�	eurolinux�fedora�mariner�miraclelinux�	openeuler�opencloudos�openmandrivar�r��rockyr��	tencentosr��ubuntu�	linuxmint�mintr��opensuse�
opensuse-leap�opensuse-microos�opensuse-tumbleweed�sle_hpc�	sle-micror��windows�darwinr�r�r�r�)rrr	r
rrr
rrrrrrr�r�rr�rr�)rrr)rrrrrrr�)rr r�r�r�r�)r�)�infor�r�Z
linux_distr.r.r/�_get_variantrs`r"cCs<tj�tj�tj�tj�ttj��t�d�}t|�|d<|S)N)r�r�r,�pythonr+rr�)r�r�r,Zpython_versionr�r+rr")r!r.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/r>�sz'get_cfg_option_list.<locals>.<listcomp>)rQr�rR)r�r�r�rMZcvalr.r.r/�get_cfg_option_list�s

r$cCs>t|t�r|jd�}|}x |D]}||kr.|S||}qW|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.�/)rQrRr-)r�Zkeypr�Zcur�tokr.r.r/�get_cfg_by_path�s


r'cCs t||�\}}t||�||fS)N)�get_output_cfg�redirect_output)�cfg�mode�outfmt�errfmtr.r.r/�fixup_output�s
r.c
Cs�ttjjd��rtjd�dS|s(tj}|s2tj}dd�}|�rtjd||�|j	dd�\}}|dksn|d	kr�d
}|dkr~d}t
||�}n0|dkr�tj|d
tj
|d�}	|	j}ntd|��|r�tj|j�|j��||k�rtjd||�tj|j�|j��dS|�r�tjd||�|j	dd�\}}|dk�s:|d	k�rXd
}|dk�rLd}t
||�}n2|dk�r~tj|d
tj
|d�}	|	j}ntd|��|�r�tj|j�|j��dS)NZ_CLOUD_INIT_SAVE_STDOUTz5Not redirecting output due to _CLOUD_INIT_SAVE_STDOUTcSs>tjd�ytjd�j}Wntk
r.YnXtj|�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|�>z>>�ab�wb�|T)�shell�stdinZ
preexec_fnz"Invalid type for output format: %sz!Invalid type for error format: %s)r�r*�environr�rFrtr�r3r�r-r��
subprocess�Popen�PIPEr=�	TypeError�dup2�fileno)
r,r-Zo_outZo_errr6r+�argZowithZnew_fp�procr.r.r/r)�s`



r)cCsR|rt|�}i}x<|D]4}|rtj|�}|s4tj�}tj|�}|j||�}qW|S)N)�reversedrZdict_extract_mergersZdefault_mergersZ	construct�merge)Zsrcs�reverseZ
merged_cfgr*Zmergers_to_applyZmergerr.r.r/�
mergemanydictOs


rJccs.tj�}ztj|�|VWdtj|�XdS)N)r*�getcwd�chdir)ZndirZcurrr.r.r/rL^s


rLccs&tj|�}z
|VWdtj|�XdS)N)r*r0)Zn_msk�oldr.r.r/r0hs

r0cCsdj||d|d�S)Nz{0:{fill}{align}{size}}�^)�fillZalign�size)�format)rYrOZmax_lenr.r.r/�centerqsrRcCstjd|�tj|�dS)NzRecursively deleting %s)rFrt�shutilZrmtree)rhr.r.r/�del_dirwsrTr��cCsjy.t|||�\}}}||d<||d<||d<dStjk
rd}z|jtjkrRdS�WYdd}~XnXdS)Nz	user-datazvendor-dataz	meta-dataTF)�read_seededr�UrlError�codeZ	NOT_FOUND)rO�base�ext�timeout�md�ud�vdrur.r.r/�read_optional_seed�sr_cCs�i}|sddg}n(tjj|jd�d�tjj|jd�d�g}t|�}dd�|D�}d}x2|D]*}tjjtjj|d��rZtjj|d�}PqZWd}x2|D]*}tjjtjj|d��r�tjj|d�}Pq�W|r�|r�||d	<||d
<n|r�||d	<|S)Nz/var/lib/cloud/data/sslz /var/lib/cloud/instance/data/sslr=ZsslcSs g|]}|rtjj|�r|�qSr.)r*rh�isdir)r;�dr.r.r/r>�sz%fetch_ssl_details.<locals>.<listcomp>zcert.pemzkey.pem�	cert_file�key_file)r*rhrHZ
get_ipath_curZ	get_cpathr�r�)�pathsZssl_detailsZssl_cert_pathsrbrarcr.r.r/�fetch_ssl_details�s0


recCs |}t|�}y\tjdt|�|�tj|�}|dkrBtjd�|}n t||�sbtd|tj	|�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}tj|�WYdd}~XnX|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 blobZcontext_markZproblem_markz5. Invalid format at line {line} column {col}: "{err}"r|)rK�colrOz. {err})rO)rVrFrtrEr�loadrQrBrr�Z	YAMLError�
ValueError�hasattr�getattrrQrK�columnrG)rTr�ZallowedZloadedZ	convertedru�msgZmarkr.r.r/r��s<



r��
cCsV|jd�dkr@|jdd|�}|jdd|�}|jdd|�}nTtjrj|ddkrjtj|�jdkrj|d7}d	|d|f}d	|d|f}d	|d|f}tj|||d
�}d}	|j	�r�t
t|j�id�}	tj|||d
�}
d}|
j	�r�|
j}d}ytj|||d
�}
Wn2tj
k
�r.}ztjd|�WYdd}~XnX|
j	��rB|
j}n
tjd
�|	||fS)Nz%srz	user-datazvendor-dataz	meta-datar|r%r�z%s%s%s)r[�retries)r�z!Error in vendor-data response: %szError in vendor-data responser�)r�r�rZ%NOCLOUD_SEED_URL_APPEND_FORWARD_SLASHr�urlparseZqueryrZread_file_or_url�okr�rV�contentsrWrFrt)rYrZr[rnZfile_retriesZud_urlZvd_urlZmd_urlZmd_respr\Zud_respr]r^Zvd_resprur.r.r/rV�s<

rVcs�ttj��dd�}dd�|D�}�fdd�|D�}g}xh|D]`}y |jttjj�|�|d��Wq<tk
r�}z|jt	kr�t
jd�|�WYdd}~Xq<Xq<Wt|�S)	zRead configuration directory.T)rIcSsg|]}|jd�r|�qS)z.cfg)�endswith)r;�fr.r.r/r>szread_conf_d.<locals>.<listcomp>cs&g|]}tjjtjj�|��r|�qSr.)r*rhr�rH)r;rs)�confdr.r/r>
s)r�z,REDACTED config part %s/%s for non-root userN)
r�r*�listdirr�r�rhrHrsr�rrFrGrJ)rtr�Zconfs�cfgsr�rur.)rtr/�read_conf_ds

 rwcCs�t�}i}yt||d�}Wn8tk
rR}z|jtkrBtjd|�WYdd}~XnX|j|�d}d|kr�|d}|r�t|t	�s�t
d|tj|�f��q�t	|�j
�}ntjj|�d��r�|�d�}|r�tjj|�r�t||d�}|j|�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�rsr�rrFrGr�rQrRrBrr�r4r*rhr`rw�
appendleftrJ)Zcfgfiler�rvr*rurtZ	confd_cfgr.r.r/�read_conf_with_confds.




rycCstt|d��S)N)�cmdline)r��read_cc_from_cmdline)rzr.r.r/�read_conf_from_cmdlineKsr|c	Cs�|dkrt�}d}d}t|�}t|�}t|�}g}|j|�}x`|dkr�|j|||�}|dkrb|}|jtj||||�j��jdd��|j|||�}q>Wdj|�S)Nzcc:Zend_ccrz\nr�)	�get_cmdlinerEr�r�rZunquote�lstripr�rH)	rzZ	tag_beginZtag_endZbegin_lZend_lZclen�tokensZbegin�endr.r.r/r{Ps$



r{cCs2|jd�}|dks"||ddkr&|S|jdd�S)Nr�rr|�
z
)r�r�)rq�posr.r.r/�dos2unixrs
r��HostnameFqdnInfo�hostname�fqdn�
is_defaultcCs�d}d|kr,|d}t|d|jd�d�}nld|krf|djd�dkrf|d}|dd|jd��}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�)r*Zcloudr�r�r�r�r.r.r/�get_hostname_fqdn�s
r��
/etc/hostscCs�d}yvxpt|�j�D]`}|jd�}|dkr6|d|�}|j�}|sDq|j�}t|�dkrZq||dd�kr|d}PqWWntk
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�#rr�r&r|)r�rBr�r4r-rEr�)r��filenamer�rKZhashpos�toksr.r.r/�get_fqdn_from_hosts�s$

r�cCs
tdkr�t�}d}i}x�|D]�}ydtj|dddtjtj�}g||<x>|D]6\}}}}	}
||jd|	|
df�|j|
d�qHWWqtjtj	fk
r�YqXqW|a|r�t
jd|�y,tj|d�}|ddd}|tkr�d	Sd
Stjtj	fk
�rd	SXdS)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�does-not-exist.example.com.�example.invalid.�!__cloud_init_expected_not_found__rz%s: %szdetected dns redirection: %s�FT)r�r�r�)�_DNS_REDIRECT_IP�set�socketZgetaddrinfoZSOCK_STREAMZAI_CANONNAMEr��addZgaierror�errorrFrt)r�ZbadipsZbadnamesZ
badresultsZiname�resultZ_famZ_stypeZ_protoZcnameZsockaddrZaddrr.r.r/�
is_resolvable�s6


r�cCstj�}|S)N)r�Zgethostname)r�r.r.r/r�sr�cCs*ytj|�dStjk
r$dSXdS)Nr)r��
gethostbyaddrZherror)Zipr.r.r/r�sr�cCs ttjd|ttj|�jfd�S)z5determine if this url is resolvable (existing or ip).zResolving URL: )�logfuncrl�funcr�)�log_timerFrtr�rror�)Zurlr.r.r/�is_resolvable_urls
r�cCs\|dkrdStjd|�x>|D]6}yt|�r<tjd|�|SWqtk
rRYqXqWdS)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')rFrtr�r)Z
candidatesZcandr.r.r/�search_for_mirrors

r�c
CsDttjjd��rdSttj��}tj|j�tj	j��WdQRXdS)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*r>r�r��devnullrCrDr�r=)�fpr.r.r/�close_stdin+sr��devicecCsvg}|stjd�tjd�S|jd�rL|jd�}dd�d|d|gD�}n&|dkr`tjd�}n|d	krrtjd�}|S)
Nz/dev/msdosfs/*z/dev/iso9660/*zLABEL=cSsg|]}tjj|�r|�qSr.)r*rhr�)r;�pr.r.r/r>Bsz*find_devs_with_freebsd.<locals>.<listcomp>z
/dev/msdosfs/z
/dev/iso9660/z	TYPE=vfatzTYPE=iso9660)r��
startswithr~)�criteria�oformat�tag�no_cacherh�devlist�labelr.r.r/�find_devs_with_freebsd9s


r�cCs�g}d}d}|r8|jd�r$|jd�}|jd�r8|jd�}tjdddgdgd�}xx|jj�D]j}	|sf|r�tjd|	gdd	gd�\}
}|r�d
||
kr�qZ|dkr�d|
kr�qZ|d
kr�d|
kr�qZ|jd|	�qZW|S)NzLABEL=zTYPE=�sysctlz-nzhw.disknamesr)�rcsZ	mscdlabelr|z
label "%s"�iso9660zISO filesystem�vfatz/dev/)r�r~rr3r-r�)r�r�r�r�rhr�r��_typer5�devZ
mscdlabel_outrr.r.r/�find_devs_with_netbsdMs(



r�cCs�tjdddgdgd�}g}xd|jj�jd�D]P}|jd�s<q,|dkrFq,|j|dd�d
�|jd�s,|j|dd�d�q,Wd
d�|D�S)Nr�z-nzhw.disknamesr)r�r?r:zfd0:r|r��cd�icSsg|]}d|�qS)z/dev/r.)r;r�r.r.r/r>tsz*find_devs_with_openbsd.<locals>.<listcomp>r�r�)rr3�rstripr-rrr�r�)r�r�r�r�rhr5r�rr.r.r/�find_devs_with_openbsdfs

r�cCs�tjdddgdgd�}dd�t|jj�dd	�D�}|d
krJdd�|D�}n(|dkrbdd�|D�}n|rrtjd|�dd�|D�S)Nr�z-nz
kern.disksr)r�cSs(g|] }|jd�r|jd�r|�qS)r\Zvn)r�)r;r�r.r.r/r>|sz/find_devs_with_dragonflybsd.<locals>.<listcomp>T)rIzTYPE=iso9660cSs$g|]}|jd�s|jd�r|�qS)r��acd)r�)r;r�r.r.r/r>�s�LABEL=CONFIG-2�	TYPE=vfatcSs$g|]}|jd�p|jd�s|�qS)r�r�)r�)r;r�r.r.r/r>�szUnexpected criteria: %scSsg|]}d|�qS)z/dev/r.)r;r�r.r.r/r>�s)r�r�)rr�r3r-rFrt)r�r�r�r�rhr5r�r.r.r/�find_devs_with_dragonflybsdws
r�c
CsJt�rt|||||�St�r,t|||||�St�rBt|||||�St�rXt|||||�Sdg}g}|rt|jd|�|r�|jd|�|r�|j	ddg�|r�|jd|�|r�|j|�||}yt
j
|ddgd	�\}}	Wn:t
jk
�r}
z|
jt
k�rd
}n�WYdd}
~
XnXg}x*|j�D]}|j�}|�r$|j|��q$W|S)z�
    find devices matching given criteria (via blkid)
    criteria can be *one* of:
      TYPE=<filesystem>
      LABEL=<label>
      UUID=<uuid>
    �blkidz-t%sz-s%sz-cz	/dev/nullz-o%srr&)r�r�N)r�r�r�r�r�r�r�r�r�r�rrIr�rrBr4)
r�r�r�r�rhZ
blk_id_cmdZoptions�cmdr5�_errrur�rKr.r.r/�find_devs_with�sD

r�c	Cs�|dkrg}nt|�}dddg}|r2|jddg�|j|�tj|ddd	�}i}x:|jj�D],}|jd
�\}}}t|�||<|||d<q\W|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�z-oZfullz-cz	/dev/nullTr�)r1rSr:ZDEVNAME)r�r�rr3rBrCr�)	ZdevsZ
disable_cacher�r5�retrKr�rr=r.r.r/r��s

r�c	Cs.tjd||�t|d��}|j|�SQRXdS)NzPeeking at %s (max_bytes=%s)r�)rFrtr�r�)rLZ	max_bytes�ifhr.r.r/�	peek_file�sr�cCs,g}x"|D]}||krq
q
|j|�q
W|S)N)r�)Zin_listZout_listr�r.r.r/r��s
r�cCs�tjd||�tj�}y(t|d��}t|||d�WdQRXWn4tk
rr}z|sV�|jtkrb�WYdd}~XnX|j	�}tjdt
|�|�|r�t|�S|SdS)NzReading from %s (quiet=%s)r�)�chunk_cbzRead %s bytes from %s)rFrtr�r�r��pipe_in_outr�r�r�getvaluerErV)rLZread_cbr�rSZofhr�rurqr.r.r/r��s
r�cCs�t�r\y td�}|jdd�dd�}Wq�tk
rX}ztjd|�d}WYdd}~Xq�Xn*ytd�j�}Wntk
r�d}YnX|S)	Nz/proc/1/cmdline�r7r|z"failed reading /proc/1/cmdline: %sr�z
/proc/cmdliner�)�is_containerr�r�rrFrGr4)rqrzrur.r.r/�_get_cmdlines
r�cCsdtjkrtjdSt�S)NZDEBUG_PROC_CMDLINE)r*r>r�r.r.r.r/r}%s

r}�cCsRd}x@|j|�}t|�dkr Pq|j|�|t|�7}|r||�qW|j�|S)Nr)r�rEr�r�)Zin_fhZout_fhZ
chunk_sizer�Zbytes_pipedr=r.r.r/r�,s

r�cCs6|dkr|dkrdStjd|||�tj|||�dS)Nr|z%Changing the ownership of %s to %s:%sr�)Nr�r�)Nr�)rFrtr*�chown)rL�uid�gidr.r.r/�	chownbyid;sr�cCsnd}d}y$|rtj|�j}|r*tj|�j}Wn0tk
r\}ztd|�|�WYdd}~XnXt|||�dS)Nr|zUnknown user or group: %sr�r�)	�pwd�getpwnamZpw_uidr1r2r3r4rsr�)rL�userr�r�r�rur.r.r/�chownbynameCs r�c
Cszddg}|sd|kr|S|d}||kr4||}nd|kr@|S|d}t|t�rZ||g}t|t�r�t|�dkr||d|d<t|�dkr�|d|d<t|t�r�d|kr�|d|d<d|kr�|d|d<|ddkr�|d|d<ddd	g}x�tt|��D]|}||�sq�||j�}d
}x<|D]4}	|j|	��rd|	|t|	�d�j�f}d}P�qW|�sjdd|j�f}|||<q�W|S)
N�output�allrr|r�z&1z>>r8r;Fz%s %sT)	rQrRr�rErJr�r~r�r4)
r*r+r�ZoutcfgZmodecfgZswlistr�rMr�sr.r.r/r(YsL






r(cCs�g}|st|t�r|S|jd�}|r2|j|�xvt|d�D]h}|sHq>tjd|�}|sZq>|jd�}|j�}t	|�dkr�|j|�q>ddg|dd�kr>|j|d�q>Wt
t|��S)	z�Return a list of log file paths from the configuration dictionary.

    @param cfg: The cloud-init merged configuration dictionary.
    Zdef_log_fileNz (?P<type>\||>+)\s*(?P<target>.*)r2r|�teez-ar&)rQrJr�r�r(r�r�r�r-rEr�r�)r*ZlogsZdefault_log�fmtr�r2�partsr.r.r/�get_config_logfiles�s&


r�cGsB|r|j|f|��tj�}|dkr(d}|j|f|�d|i�dS)N�exc_info)NNN)rGr�r�rt)rrlr�r�r.r.r/r��sr�cCs<tj|�}|jt|��|j�}|dk	r4|d|�S|SdS)Nr)�hashlib�new�updaterZZ	hexdigest)rTZroutineZmlenZhasherZdigestr.r.r/�	hash_blob�s
r�cCs,ytj|�rdSWntk
r&dSXdS)NTF)r�r�r4)r�r.r.r/�is_user�s

r�cCs,ytj|�rdSWntk
r&dSXdS)NTF)r1r2r4)r�r.r.r/�is_group�s

r�cCstjd||�tj||�dS)NzRenaming %s to %s)rFrtr*�rename)�src�destr.r.r/r��sr���cCsx|D]}t||�qWdS)N)�
ensure_dir)�dirlistr+rar.r.r/�ensure_dirs�s
r�cCsHtjt|��}t|t|��sDdjdd�|D��}td|t|�f��|S)Nz, cSsg|]}t|��qSr.)rR)r;�tr.r.r/r>�szload_json.<locals>.<listcomp>z((%s) root types expected, got %s instead)�json�loadsrVrQr'rHrB�type)rYZ
root_typesr]Zexpected_typesr.r.r/�	load_json�sr�cCs2ydjt|��Stk
r,djt|��SXdS)z1Handler for types which aren't json serializable.z
ci-b64:{0}z)Warning: redacted unserializable type {0}N)rQr_�AttributeErrorr�)Z_objr.r.r/�json_serialize_default�sr�cCstj|dddtd�S)z%Return data in nicely formatted json.r|Tr?�: )�indentZ	sort_keysZ
separatorsr�)r?r�)r��dumpsr�)r=r.r.r/�
json_dumps�sr�cCs<tjj|�}|tjj|�kr |Stjj|�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*rh�dirnamer`�get_non_exist_parent_dir)rhZp_pathr.r.r/r�sr�cCs�tjj|�s�t|�}ttjj|�dd��tj|�WdQRXt||�|sP|r�t|||�t	|j
|�ddd��}t	|�}x0|jD]}|j|�}t|||�q�Wn
t||�dS)NT)rir|)
r*rhr`r�rdr��makedirs�chmodr�r
r-r�Zjoinpath)rhr+r�r�Znon_existed_parent_dirZsub_relative_dirZsub_pathrar.r.r/r�s

r�ccs(z
|VWd|r"d|g}tj|�XdS)N�umount)r)r�Z
umount_cmdr.r.r/�	unmounter+s

r�c
Csi}y�tjjd�r$td�j�}d}ntjd�}|jj�}d}d}x�|D]�}yV|dkrj|j�\}}}}	}
}n4tj	||�}|j
d�}|j
d�}|j
d�}|j
d�}	Wntk
r�wFYnX|jd	d
�}|||	d�||<qFWt
jd||�Wn$ttfk
�rtt
d
�YnX|S)Nz/proc/mountsrF�mountz*^(/dev/[\S]+) on (/.*) \((.+), .+, (.+)\)$r|r&r�r�z\040r7)�fstype�
mountpoint�optszFetched %s mounts from %szFailed fetching mount points)r*rhr�r�rBrr3r-r��searchr�rr�rFrtr�rsr�)
�mounted�
mount_locs�methodr5ZmountreZmpliner�Zmpr�r�Z_freqZ_passno�mr.r.r/�mounts5s8





rcCst|t�r|g}n:t|ttf�r*t|�}n"|dkr8d}ntdjt|�d���t�rb|dkr�dg}nVt�r�|dkrzdddg}x<t	|�D](\}}|dkr�d||<|dkr�d||<q�Wndg}t
�}tj���:}d}	t
jj|�|kr�|t
jj|�d
}
n�d}x�|D]�}d}
yNdddg}|�r&|jd|g�|j|�|j|�tj||d�|}	|}
PWnDttfk
�r�}
z"tjd||dj|�|
�|
}WYdd}
~
XnX�q�W|
�s�td|||f��|
jd��s�|
d7}
t|	��$|dk�r�||
�}n
||
|�}|SQRXWdQRXdS)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��msdosfsr�Fr�r�z-oZroz-t)Z
update_envzbFailed to mount device: '%s' with type: '%s' using mount command: '%s', which caused exception: %sr7z#Failed mounting %s to %s due to: %sr%)r�r)rQrRr�r'rBrQr�r�r��	enumeraterrZtempdirr*rhror�r�rr�rsrFrtrHrzrrr�)r��callbackr=ZmtypeZupdate_env_for_mountZmtypes�indexr�Ztmpdr�r�Zfailure_reasonZmountcmd�excr�r.r.r/�mount_cb]st










r
cCs
tjt�S)N)�obj_copy�deepcopyrr.r.r.r/�get_builtin_cfg�sr
cCstjd|�tjj|�S)NzTesting if a link exists for %s)rFrtr*rh�islink)rhr.r.r/�is_link�srcCshtjd||�|rXtjj|�rXtjjtjj|�dtd��}tj||�tj	||�dStj||�dS)Nz$Creating symbolic link from %r => %rZtmpr�)
rFrtr*rhrnrHr�r��symlinkr�)r\�linkZforceZtmp_linkr.r.r/�sym_link�srcCsPtjd|�ytj|�Wn0tk
rJ}z|jtkr:|�WYdd}~XnXdS)NzAttempting to remove %s)rFrtr*�unlinkrsr�r)rhrur.r.r/�del_file�s
rcCstjd||�tj||�dS)NzCopying %s to %s)rFrtrS�copy)r�r�r.r.r/r�srcCs2ytjdtj��}Wntk
r,d}YnX|S)Nz%a, %d %b %Y %H:%M:%S %zz??)�timeZstrftimeZgmtimer)Ztsr.r.r/�time_rfc2822�s

rcs�ddl�ddl�d}G�fdd�d�j�}�j�jjd��}�j�}�j|�|_|�}|j	d|�j
|��j
|�dd�dkr�|j|jd	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)rwrxryZc_int64Z_fields_r.)�ctypesr.r/�timeval�sr�cs
kern.boottimer|g��.Az/Unable to retrieve kern.boottime on this systemr�)rZctypes.utilZ	StructureZCDLL�utilZfind_libraryZc_size_tZsizeof�valueZsysctlbynameZbyrefrr�RuntimeError)Z
NULL_BYTESrZlibcrPr�r.)rr/�boottime�s"	r!cCstd}d}yDtjjd�r4d}td�}|rJ|j�d}nd}ttj�t��}Wn"tk
rnt	t
d|�YnX|S)Nz??rz/proc/uptimerrz&Unable to read uptime using method: %s)r*rhr�r�r-rRrr!rr�rF)Z
uptime_strrrqr.r.r/�uptimesr"cCst||ddd�dS)Nr9)�omoder+)�
write_file)rh�contentr.r.r/�append_filesr&�)�
preserve_mode)r+r(r�cCst|dd||d�dS)Nr�r9)r%r#r+r()r$)rhr+r(r.r.r/�ensure_filesr)cCs&yt|�Sttfk
r dSXdS)N)r)rhrB)Zpossible_intr.r.r/�safe_int#sr*c
Cs4t|�}|r0|r0t|��tj||�WdQRXdS)N)r*rdr*r�)rhr+Z	real_moder.r.r/r�*s
r�)�grp_namer�cCs:d}ytj|�j}Wn tk
r4tjd|�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 namer�)r1r2r3r4rFrt)r+r�r.r.r/�get_group_id1sr,)rhr�cCstjtj|�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.
    )rq�S_IMODEr*�st_mode)rhr.r.r/�get_permissions?sr/cCstj|�}tj|j�jS)zw
    Returns the owner of the file/folder pointed by the path.

    @param path: The full path of the file/folder.
    )r*rqr��getpwuid�st_uidZpw_name)rh�str.r.r/�	get_ownerJs
r3cCstj|�}tj|j�jS)zw
    Returns the group of the file/folder pointed by the path.

    @param path: The full path of the file/folder.
    )r*rqr1�getgrgid�st_gid�gr_name)rhr2r.r.r/�	get_groupTs
r7)�usernamer�cCsNg}x&tj�D]}||jkr|j|j�qWtj|�j}|jtj|�j�|S)zp
    Returns a list of all groups to which the user belongs

    @param username: the user we want to check
    )	r1ZgetgrallZgr_memr�r6r�r�Zpw_gidr4)r8�groupsr�r�r.r.r/�get_user_groups^s
r:r:)�ensure_dir_existsr�r�c!Cs�|r&yt|�}Wntk
r$YnX|r@ttjj|�||d�d|j�krZt|�}d}nt|�}d}yd|}	Wnt	k
r�d|}	YnXt
jd|||	t|�|�t
|d��.t||��}
|
j|�|
j�Wd	QRXWd	QRXt||�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��brWZ
charactersz%oz%rzWriting to %s - %s: [%s] %s %s)rhN)r/rsr�r*rhr�r�rZrVrBrFrtrErdr�r�r�r�)r�r%r+r#r(r;r�r�Z
write_typeZmode_rZfhr.r.r/r$ns8
r$cCsDx>tj|�D]0}tjj||�}tjj|�r4t|�qt|�qWdS)z�
    Deletes all contents of a directory without deleting the directory itself.

    @param dirname: The directory whose contents should be deleted.
    N)r*rurhrHr`rTr)r�ZnodeZ
node_fullpathr.r.r/�delete_dir_contents�s

r=r��createdcCs6tj�}t|�}|d|j�|f7}|dt�7}|S)Nz %s by cloud-init v. %sz on %s)rZversion_stringrR�titler)Zcomment_charrYZci_ver�headerr.r.r/�make_header�s
rAcGstjjtjj|f|���S)N)r*rh�abspathrH)rYrdr.r.r/�abs_join�srCcCs�t|ttf�s tdtj|���d}|r0|d7}dd}d}x�|D]�}t|ttf�r�g}x&|D]}|jdt|�jd|��q^Wd	|d
j	|�f}|d7}qBt|t�r�d	||f}|d7}qB|dkr�qBtdtj|�|f��qBWt
jd
|�|S)Nz8Input to shellify was type '%s'. Expected list or tuple.r�z
#!/bin/sh
z%s%s%s%s�'�\rz'%s'z%s%s
r7r|zCUnable to shellify type '%s'. Expected list, string, tuple. Got: %szShellified %s commands.)rDrErDrD)rQr'r�rBrr�r�rRr�rHrFrt)ZcmdlistZ
add_headerr%ZescapedZ	cmds_mader�Zfixedrsr.r.r/�shellify�s4




rFcCsB|r|j|�r|t|�d�}|r>|j|�r>|dt|��}|S)N)r�rErr)rK�prefix�suffixr.r.r/�strip_prefix_suffix�s
rIcCs@tj|d�dkrdSytj|�Wntjk
r:dSXdS)NrFT)r�whichrI)r�r.r.r/�_cmd_exits_zero�srKcCstdddg�S)Nzsystemd-detect-virtz--quietz--container)rKr.r.r.r/�_is_container_systemd	srLcCs
tdg�S)Nzlxc-is-container)rKr.r.r.r/�_is_container_old_lxc	srMcCsDt�s
dSdddg}tj|d�dkr*dStj|�\}}|j�dkS)NFr�z-qnzsecurity.jail.jailedrr)r�rrJr4)r�r5rr.r.r/�_is_container_freebsd	s
rNcCs�tttf}x|D]}|�rdSqWy$td�}d|kr8dSd|krDdSWnttfk
r^YnXtjjd�r~tjjd�r~dSyHt	d�j
�}x6|D].}|jd�r�|j�j
d	d�\}}|d
kr�dSq�WWnttfk
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:r:r"F)rLrNrM�get_proc_envr�rsr*rhr`r�rBr�r4r-)Zchecks�helperZpid1env�linesrKZ_keyrMr.r.r/r�	s4


r�cCstjjd�S)z2Check to see if we are running in a lxd container.z
/dev/lxd/sock)r*rhr�r.r.r.r/�is_lxd@	srRr�cCs�tjjdt|�d�}yt|dd�}Wnttfk
r<iSXi}d
\}}|rbd\}}|j||�}x4|j|�D]&}|sxqn|j|d	�\}	}
|	rn|
||	<qnW|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/procr>F)rSr�=r��=r|)rrS)r�rT)	r*rhrHrRr�r�rsrSr-)�pidrU�errorsr�rq�envZnullZequalr&r�rMr.r.r/rOE	s"rOcCsTi}xJ|j�D]>}y|jdd�\}}Wntk
rB|}d}YnX|||<qW|S)NrTr|T)r-rh)Zkvstringr�r&r�rMr.r.r/�keyval_str_to_dictc	s
rXcCs&|jd�r|dd�}tjjd|�S)Nz/dev/rUz/sys/class/block/%s/partition)r�r*rhr�)r�r.r.r/�is_partitiono	s
rYcCs�t|t�s|g}g}x�|D]�}t|t�r4|j|�qt|ttf�r�t|�dksZt|�dkrbtd��t|�dkr�|dr�|j|t|��q|j|d�qtd��qW|S)Nr|r&z Invalid package & version tuple.rzInvalid package type.)rQr�rRr�r'rEr )Zversion_fmtZpkgsZpkglist�pkgr.r.r/�expand_package_listv	s 



r[cCs�dd�|jd�D�}d}d}d}d}�xHt|�D�]:\}	}
|
j�}t|�dkrj|jd|	dt|�|
�dS|d}d	d�|jd�D�}
t|
�t|�kr�q0tt|
�t|��}|
d
|�|d
|�kr�q0|dk	r�t|�t|
�kr�q0y|jd�}	Wn(tk
�r|jd|	d|
�dSXy||	d}||	d
}Wn(tk
�rZ|jd|	d|
�dSX|}|
}|d}q0W|�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.)r;rur.r.r/r>�	sz$parse_mount_info.<locals>.<listcomp>r%Nrmz$Line %d has two few columns (%d): %sr|r�cSsg|]}|r|�qSr.r.)r;rur.r.r/r>�	sr�-z,Did not find column named '-' in line %d: %sr&z/Too few columns after '-' column in line %d: %srU)r-rrErt�minrrh�
IndexError)rhZmountinfo_linesr�get_mnt_optsZ
path_elements�devpth�fs_typeZmatch_mount_pointZmatch_mount_point_elementsr�rKr��mount_pointZmount_point_elements�xZ
mount_optionsr.r.r/�parse_mount_info�	sX


rdcCsBx<td�j�D],}|j�dd�\}}}||kr|||fSqWdS)z<On older kernels there's no /proc/$$/mountinfo, so use mtab.z	/etc/mtabNr�)r�rBr-)rhrKr`rbrar.r.r/�
parse_mtab�	s
recCs�|jd�}t|�dkr|dS|ddkr�|dd�}tjdd	d
g�\}}x<|jd�D].}|j�}t|�dkrV|d|krV|d}PqVWt|�Stjd
|�dS)Nr%r�r&r��gptrrUZglabel�statusz-sr�rz)Unexpected input in find_freebsd_part: %s)r�rfr)r-rErrRrFrG)�fs�splittedZtarget_labelrar�Zlabelsr�r.r.r/�find_freebsd_part�	s
rjcCs:|jd�}t|�dkr*|ddkr*|dStjd|�dS)Nr%r�r|r�r&z.Unexpected input in find_dragonflybsd_part: %s)r-rErFrG)rhrir.r.r/�find_dragonflybsd_part�	s
rkcCsJd}x@|jd�D]2}|j�}t|�dkrtjj|d|�r|}PqW|S)Nr�r&r|)r-rEr*rhr�)rh�mnt_list�
path_foundrKr�r.r.r/�get_path_dev_freebsd
s rncCsztjdd|gddgd�\}}t|�rPtjddg�\}}t||�}|dkrLdS|}|j�}t|d�}d||d|dfS)Nr�z-prr|)r�z/dev/r&)rrErnr-rj)rhr�rOrlrmr�Z
label_partr.r.r/�get_mount_info_freebsd
s
rocCs�tjjd�stjd�dSytjdd|g�\}}Wn0tjk
rb}ztjd||�dSd}~XnXt|�rpdSd}xL|j	d�D]>}t
j||�r�||kr�d|kr�|j	�d	}tjd
||�|Sq�WdS)Nz/dev/zfsz"Cannot get zpool info, no /dev/zfs�zpoolrgz$Unable to get zpool status of %s: %sz.*(ONLINE).*r��staterzfound zpool "%s" on disk %s)r*rhr�rFrtrrIrGrEr-r�r�)rpZzpoolstatusrOr�rKZdiskr.r.r/�get_device_info_from_zpool
s 
rrcCs�tjdg�\}}|j�}d}t�r&d}x�|D]�}tj||�}|sBq,|jd�}|jd�}|jd�}	|	dkrr|jd�}	tjd|||	�tjd	|�}
|
r�t�r�|	d
kr�t	|�S||kr,||	|fSq,WdS)Nr�zH^(/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)
rrBr�r�r�r�rFrtr�ro)rhZmountoutputr�r�ZregexrKrr`rbraZdevmr.r.r/�parse_mount-
s2




rscCsRdtj�}tjj|�r2t|�j�}t||||�Stjjd�rFt|�St|�SdS)Nz/proc/%s/mountinfoz	/etc/mtab)	r*�getpidrhr�r�rBrdrers)rhrr_Zmountinfo_pathrQr.r.r/�get_mount_info[
sru)�optr�cCs t|dd��^}}||jd�kS)NT)r_r?)rur-)rhrvrZmnt_optsr.r.r/�
has_mount_opt�
srw�T.)r�r�c$Cs�|dkrg}|dkri}tj�}d}|rLytt��}Wntk
rJYnXz|||�}Wdtj�|}	d}
|dk	r�ytt��|}
Wntk
r�YnXd|	}|r�t|
t�r�|d|
7}n|d7}y|||�Wntk
r�YnXX|S)Nz took %0.3f secondsz (%0.2f)z (N/A))r�floatr"rhrQr)r�rlr�r�r�Z
get_uptime�startZustartr�ZdeltaZudeltaZtmsgr.r.r/r��
s<
r�cCs(|jdd�}t|�dkr|S|dfSdS)Nr%r|)�rsplitrE)Zdottedr�r.r.r/�expand_dotted_devname�
sr|cCs�|dkrg}|dkrg}g}i}xp||D]d}yt|||ddd�||<Wq*tk
r�}z$|jtkrj�||kr||j|�WYdd}~Xq*Xq*Wt|�r�tdjdj|�d���|S)NF)r�rSzMissing required files: {files}r?)�files)	r�r�r�rr�rErhrQrH)rYZrequiredZoptionalZdelimrNr�rsrur.r.r/�pathprefix2dict�
s$
 r~�
/proc/meminfoc	
Cs�dd
ddd�}ddd	d
�}i}x�t|�j�D]v}y|j�\}}}Wn$tk
rf|j�\}}d}YnX|r�t|�||||<q,||kr,t|�|||||<q,W|S)Nr&rm�r|�)ZkBZmB�BZgBZtotalZfreeZ	available)z	MemTotal:zMemFree:z
MemAvailable:r�iii@)r�rBr-rhr))	Zmeminfo�raw�mpliersZkmapr�rKr�rZunitr.r.r/�read_meminfo�
s 
r�cCs�|}|jd�r|dd�}n|jd�r2|dd�}dddddd
�}|}d}x,|D]$}|j|�rP|}|dt|��}qPWyt|�}Wn0tk
r�}ztd|�|�WYdd}~XnX|dkr�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
    ZiBNr&r�r|rmr�r��()r��K�M�Grxrz'%s' is not valid input.z'%s': cannot be negative���r�iii@l)rrrEryrhr))rPZsize_inr�ZnumZmplierrrur.r.r/�human2bytes�
s&



 r�cCs<|dkrtj�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"tjtj|��Stj|�S)Nr&�)r&r�)r��version_info�emailZmessage_from_filer��StringIO�message_from_string)r�r.r.r/r�(sr�cCs�tjddg|dd�}t�}xh|jj�D]Z}y|jdd�\}}}Wntk
rVw&YnX|jd�sl|jd�r&|jtj	dd	|��q&W|S)
Nz
dpkg-queryz--listT)r2r1r&�hiZiiz:.*r�)
rr�r3rBr-rhr�r�r��sub)r2r5Z	pkgs_instrKrqrZrr.r.r/�get_installed_packages.sr�cCs�d}y,tt|dd��}|jdd�j�dkr.dSWn0tk
r`}ztjd||�WYdd}~XnXt�}d|krtdStd	dd�}d|j�kr�dStj	j
d
�r�dSdS)Nz/etc/os-releaseT)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�rhrFrGr}r*rhr`)ZorpathZorinforurzr%r.r.r/�system_is_snappy=s  r�cCs�d}x(|j�D]}|jd�r|dd�}PqW|dkr:dS|jd�rH|S|jd�rfd|td�d�S|jd�r�d|td�d�j�S|jd�r�d	|td�d�j�}tjj|�r�|St|�}|r�|d
S|Sd|S)Nzroot=rUz/dev/zLABEL=z/dev/disk/by-label/zUUID=z/dev/disk/by-uuid/z	PARTUUID=z/dev/disk/by-partuuid/r)r-r�rEr�r*rhr�r�)rzrr&Z
disks_pathZresultsr.r.r/�rootdev_from_cmdlineUs,




r�cCsJdd�}i}x8||�D],}|jdd�\}}|s2|}|s:|r|||<qW|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-)rTr.r.r/�_shlex_splitzsz(load_shell_content.<locals>._shlex_splitrTr|)r-)r%Z	add_emptyZ	empty_valr�r=rKr�rr.r.r/r�tsr���?cCs�t|�}d}xr|tdd�|D��8}t|�dkrDtjd|||�gS|dkr\tjd|||�|||krjPtj|�||7}qWtjd|||�|S)NrcSsg|]}tjj|�r|�qSr.)r*rhr�)r;rsr.r.r/r>�sz"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�rErFrtrZsleep)ZflistZmaxwaitZnaplenZlog_preZneedZwaitedr.r.r/�wait_for_files�s0
r�cCs&t|dd�}|djd�}|ddkS)z1Check whether the given mount point is mounted rwT)r_r|r?rZrwr�)rur-)rbr�Z
mount_optsr.r.r/�mount_is_read_write�sr�cCsXtjd�sdSddg}|r:tjj|�r*dS|jd|g�|rN|jd|g�tj|�S)zAInvoke udevadm settle with optional exists and timeout parametersZudevadmNZsettlez--exit-if-exists=%sz--timeout=%s)rrJr*rhr�r�)r�r[Z
settle_cmdr.r.r/�udevadm_settle�s
r�cCs�d}yJtd|dd�}|rLtjdt|��}|r>t|jd��}ntjd||�Wn0tk
r~}ztjd||�WYd	d	}~XnX|S)
z-
    Return the parent pid of a process.
    rz
/proc/%s/statT)r�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�r�rRr)r�rFrGr�)rUZppidrqrrur.r.r/�
get_proc_ppid�s
 r�r|�	Error:
{}cCs&t|j|�tjd�|r"tj|�|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)�printrQr�r��exit)rlZrcr�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|�j|||||�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|jd����S)Nr%)r�r(r)r-)r�rr.r.r/�from_str�szVersion.from_strcCsd|j|�kS)Nr|)�_compare_version)rj�otherr.r.r/�__gt__�szVersion.__gt__cCs0|j|jko.|j|jko.|j|jko.|j|jkS)N)r�r�r�r�)rjr�r.r.r/�__eq__�szVersion.__eq__ccs8x2|j|j|j|jfD]}|dkr.t|�VqPqWdS)z)Iterate over the version (drop sentinels)r|Nr�)r�r�r�r�rR)rj�nr.r.r/�__iter__�szVersion.__iter__cCs
dj|�S)Nr%)rH)rjr.r.r/�__str__szVersion.__str__)r�cCsP||krdS|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�)rjr�r.r.r/r�s
zVersion._compare_versionr�r�r�r�)r�r�r�r�)rwrxryr��classmethodrRr�r�r�r�r�r)r��
__classcell__r.r.)r�r/r��sr�r�r�r�r�)rrrr )r!r"r#r$)N)N)rP)rP)N)N)N)r�N)N)TT)F)N)r)N)N)N)NN)F)r�r�rU)N)r�r�rUrmr)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)r')r'r:F)r�r>)T)NN)rPr�)NNF)rF)N)N)FN)r�r�)NN)r|r�F)�r�rrr�r�r1r�r�r�r�r*�os.pathr�r�r�r�r�rSr�rqr�r?r�r�base64rr�collectionsrrr�rr�	functoolsrr	Zpathlibr
Ztypingrrr
rrZurllibrZ	cloudinitrrrr�rrrrrrrZcloudinit.settingsrr�Z	getLoggerrwrF�sepr�r�r�r�r�r�r0r6r9rVrZr^r_rcrdrrzr{r�r�r�r�r�r�r�r�r�r�r�r�rRrJr�r��DEBUGr�r�r�r�r�r�r�r�r�r�r�rr"r�r$r'r.r)rJ�contextmanagerrLr0rRrTr_rer�rVrwryr|r{r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r}r�r�r�r(r�r�r�r�r�r�r�r�r�r�r�r�r�rr
r
rrrrr!r"r&r)r�r)r*r�r,r/r3r7r:r$r=rArCrFrIrKrLrMrNr�rRrOrXrYr[rdrerjrkrnrorrrsrurwrxr�r|rhr~r�r�r�r�r�r�r�r�r�r�r�r�r�r�r.r.r.r/�<module>s�$	


.
2
&		(>8Q
		
!(+,"+(/
?
:
)W	#

4%

.N	
.'
'+
Zerion Mini Shell 1.0