U
    L¬÷d„ ã                
   @   sJ
  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZmZ d dlmZ d dl m!Z!m"Z" d dlm#Z#m$Z$ d dl%m&Z& d dl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ d d	l0m1Z1 d d
l2m3Z3m4Z4 d dl2m5Z6 d dl2m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z> d dl?m@Z@ daAe6 BeC¡ZDejEdiZFdejG ejH ZIdZJdZKdd„ ZLe#ƒ d}dd„ƒZMe#ƒ d~dd„ƒZNddd„ZOd€dd„ZPeQeQdœdd„ZRd d!„ ZSG d"d#„ d#ƒZTG d$d%„ d%eUƒZVG d&d'„ d'eUƒZWd(d)„ ZXdd*d+„ZYd‚d,d-„ZZdƒd.d/„Z[d„d1d2„Z\d…d3d4„Z]dd5œe*d6œd7d8„Z^d9d:„ Z_d;d<„ Z`d=d>„ Zad†d@dA„ZbdBdC„ ZcedeedDœdEdF„ZfdGdH„ Zgd?d?de6jhd?fdIdJ„Zie#ƒ dKdL„ ƒZje#ƒ dMdN„ ƒZke#ƒ dOdP„ ƒZle#ƒ dQdR„ ƒZme#ƒ dSdT„ ƒZne#ƒ dUdV„ ƒZod‡dXdY„ZpdˆdZd[„Zqd‰d\d]„ZrdŠd^d_„Zse#ƒ d`da„ ƒZtdbdc„ Zue#ƒ ddde„ ƒZvd‹dfdg„ZwdŒdhdi„Zxdjdk„ Zyddldm„ZzdŽe.e, eednœdodp„Z{ej|dqdr„ ƒZ}ej|dsdt„ ƒZ~dudv„ Zdwdx„ Z€dd{d|„Zdd}d~„Z‚deeffdd€„Zƒd‘d‚dƒ„Z„dd5œeed6œd„d…„Z…dd5œeed6œd†d‡„Z†d’dˆd‰„Z‡d“dŠd‹„ZˆdŒd„ Z‰edŽddd‘gƒZŠd”d’d“„Z‹d•d•d–„ZŒed6œd—d˜„ZŽd™dš„ Zd›dœ„ Zddž„ 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²d³„Z›e#ƒ d´dµ„ ƒZœd¶d·„ Zed6œd¸d¹„Zždd»d¼„ZŸdžd½d¾„Z dŸd¿dÀ„Z¡dÁdÂ„ Z¢dÃdÄ„ Z£dÅdÆ„ Z¤d ededdÇœdÈdÉ„Z¥e
j¦eQdÊœdËdÌ„Z§dÍdÎ„ Z¨dÏdÐ„ Z©dÑdÒ„ Zªd¡dÔdÕ„Z«eeffdÖd×„Z¬dØdÙ„ Z­d¢dÚdÛ„Z®ej|dÜdÝ„ ƒZ¯dÞdß„ Z°d£dàdá„Z±dâdã„ Z²dädå„ Z³d¤dædç„Z´dèdé„ Zµdêdë„ Zdìdí„ Z¶e#ƒ dîdï„ ƒZ·dðdñ„ Z¸dòdó„ Z¹d¥dWdõœeºeddöœd÷dø„Z»dùdú„ Z¼dûdü„ Z½dýdþ„ Z¾edeºdÿœd d„Z¿edeºdœdd„ZÀededdœdd„ZÁededdœdd„ZÂede+ed d	œd
d„ZÃd¦d?dd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/d0„ ZÓeDdWfd1d2„ZÔd3d4„ ZÕd5d6„ ZÖd7d8„ Z×d9d:„ ZØd;d<„ ZÙd«d=d>„ZÚeDdWfd?d@„ZÛededAœdBdC„ZÜe/dDƒZÝd¬e(dEeÝf eÝdFœdGdH„ZÞdIdJ„ ZßddejàjEfdKdL„Zád­dNdO„ZâdPdQ„ Zãd®dRdS„ZädTdU„ Zåd¯dVdW„ZædXdY„ ZçdZd[„ Zèd°d\d]„Zéd±d_d`„Zêdadb„ Zëd²dcdd„Zìdedf„ Zídgdh„ Zîdidj„ Zïd³dmdn„Zðe$G dodp„ dpedpdqdrdsdtgƒƒƒZñddzduœedede-ed eºdvœdwdx„Zòdzdyœededeºdzœd{d|„ZódS (´  é    N)Ú	b64decode)ÚdequeÚ
namedtuple)Úsuppress)ÚEACCESÚENOENT)Ú	lru_cacheÚtotal_ordering)ÚPath)ÚCallableÚDequeÚDictÚListÚMappingÚOptionalÚSequenceÚTypeVar)Úparse)ÚfeaturesÚimporter)Úlog)ÚmergersÚnetÚsafeyamlÚsubpÚ
temp_utilsÚ
type_utilsÚ
url_helperÚversion)ÚCFG_BUILTINÚ_z_-.())ÚtrueÚ1ZonZyes)ZoffÚ0ZnoZfalsec                   C   s"   t ttt ¡ j d¡d d… ƒƒS )NÚ.é   )ÚtupleÚmapÚintÚosÚunameÚreleaseÚsplit© r-   r-   ú0/usr/lib/python3/dist-packages/cloudinit/util.pyÚkernel_versionQ   s    r/   c                 C   s   t j ddgd| 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.
    Zdpkgz--print-architectureT©ÚcaptureÚtarget)r   ÚstdoutÚstrip)r2   Úoutr-   r-   r.   Úget_dpkg_architectureU   s      ÿr6   c           	   
      sÞ   dddddœ}i ‰ z€t j ddgd| d	}|j ¡ D ],}| d
¡\}}}||kr2| ¡ ˆ || < q2‡ fdd„| ¡ D ƒ}t|ƒrt dd 	|¡¡ W nF t j
k
rØ } z&t d|¡ tdd„ | ¡ D ƒƒ‰ W 5 d }~X Y nX ˆ S )NÚcodenameZdescriptionÚidr+   )ZCodenameZDescriptionzDistributor IDZReleaseÚlsb_releasez--allTr0   ú:c                    s   g | ]}|ˆ kr|‘qS r-   r-   )Ú.0Úk©Údatar-   r.   Ú
<listcomp>r   s      zlsb_release.<locals>.<listcomp>z.Missing fields in lsb_release --all output: %sú,z#Unable to get lsb_release --all: %sc                 s   s   | ]}|d fV  qdS )ZUNAVAILABLENr-   ©r;   Úvr-   r-   r.   Ú	<genexpr>{   s     zlsb_release.<locals>.<genexpr>)r   r3   Ú
splitlinesÚ	partitionr4   ÚvaluesÚlenÚLOGÚwarningÚjoinÚProcessExecutionErrorÚdict)	r2   Zfmapr5   ÚlineÚfnamer    ÚvalÚmissingÚerrr-   r=   r.   r9   b   s,    üþ(r9   úutf-8c                 C   s   t | tƒr| S |  |¡S ©N)Ú
isinstanceÚstrÚdecode)ÚblobÚencodingr-   r-   r.   Údecode_binary€   s    
rY   c                 C   s   t | tƒr| S |  |¡S rS   )rT   ÚbytesÚencode)ÚtextrX   r-   r-   r.   Úencode_text‡   s    
r]   )r>   Úreturnc                 C   sH   t | tƒstdt| ƒ ƒ‚zt| ddW S  tjk
rB   |  Y S X dS )z¼base64 decode data

    If data is base64 encoded bytes, return b64decode(data).
    If not, return data unmodified.

    @param data: data as bytes. TypeError is raised if not bytes.
    zdata is '%s', expected bytesT)ZvalidateN)rT   rZ   Ú	TypeErrorÚtyper   ÚbinasciiÚErrorr=   r-   r-   r.   Úmaybe_b64decodeŽ   s    
rc   c                 C   sP   | j dd}|  ¡ dkrLt|tƒrL|  ¡ }|r<|jr<|j}nd}| |d¡S |S )NT©rV   r\   rR   Úsurrogateescape)Zget_payloadZget_content_maintyperT   rZ   Zget_charsetZinput_codecrV   )ÚpartZcte_payloadÚcharsetrX   r-   r-   r.   Úfully_decoded_payloadž   s     ÿ
rh   c                   @   s&   e Zd Zd	dd„Zdd„ Zdd„ ZdS )
ÚSeLinuxGuardFc                 C   s<   zt  d¡| _W n tk
r*   d | _Y nX || _|| _d S )NÚselinux)r   Úimport_modulerj   ÚImportErrorÚpathÚ	recursive)Úselfrm   rn   r-   r-   r.   Ú__init__²   s    zSeLinuxGuard.__init__c                 C   s   | j r| j  ¡ rdS dS d S ©NTF)rj   Úis_selinux_enabled©ro   r-   r-   r.   Ú	__enter__¼   s    zSeLinuxGuard.__enter__c              
   C   sÎ   | j r| j  ¡ sd S tj | j¡s&d S tj | j¡}z"t |¡}| j  ||tj	 ¡ W n t
k
rl   Y d S X t d|| j¡ z| j j|| jd W n4 t
k
rÈ } zt d|| j|¡ W 5 d }~X Y nX d S )Nz,Restoring selinux mode for %s (recursive=%s)©rn   z,restorecon failed on %s,%s maybe badness? %s)rj   rr   r)   rm   ÚlexistsÚrealpathÚlstatZmatchpathconÚstatÚST_MODEÚOSErrorrH   Údebugrn   Z
restoreconrI   )ro   Z	excp_typeZ
excp_valueZexcp_tracebackrm   ZstatsÚer-   r-   r.   Ú__exit__Â   s0    
ýüzSeLinuxGuard.__exit__N)F)Ú__name__Ú
__module__Ú__qualname__rp   rt   r~   r-   r-   r-   r.   ri   ±   s   

ri   c                   @   s   e Zd ZdS )ÚMountFailedErrorN©r   r€   r   r-   r-   r-   r.   r‚   ß   s   r‚   c                   @   s   e Zd ZdS )ÚDecompressionErrorNrƒ   r-   r-   r-   r.   r„   ã   s   r„   c              	   O   st   t  ¡ }|dkr\z| ||Ž t  d¡ W qp tk
rX   ttdt | ¡ƒ t  d¡ Y qpX nt d|t | ¡¡ d S )Nr   z&Failed forking and calling callback %sé   z(Forked child %s who will run callback %s)	r)   ÚforkÚ_exitÚ	ExceptionÚlogexcrH   r   Úobj_namer|   )Zchild_cbÚargsÚkwargsZfidr-   r-   r.   Úfork_cbç   s"    
ýýr   c                 C   sB   t | tƒr| dkS t}|r&t|ƒ| }t| ƒ ¡  ¡ |kr>dS dS rq   )rT   ÚboolÚTRUE_STRINGSÚlistrU   Úlowerr4   ©rO   ÚaddonsZ	check_setr-   r-   r.   Úis_trueü   s    
r”   c                 C   sB   t | tƒr| dkS t}|r&t|ƒ| }t| ƒ ¡  ¡ |kr>dS dS )NFT)rT   rŽ   ÚFALSE_STRINGSr   rU   r‘   r4   r’   r-   r-   r.   Úis_false  s    
r–   c                 C   s    | sdS t | tƒr| S t| |ƒS )NF)rT   rŽ   r”   )rO   r“   r-   r-   r.   Útranslate_bool  s
    
r—   é    c                    s8   t  ¡ ‰ ˆstjtj ‰d ‡ ‡fdd„td| ƒD ƒ¡S )NÚ c                    s   g | ]}ˆ   ˆ¡‘qS r-   )Zchoice)r;   Z_x©ÚrÚselect_fromr-   r.   r?   !  s     zrand_str.<locals>.<listcomp>r   )ÚrandomZSystemRandomÚstringÚascii_lettersÚdigitsrJ   Úrange)Ústrlenrœ   r-   rš   r.   Úrand_str  s    r£   c                 C   s*   |sd}t ddd | }|| krq&q|S )Nr™   é   )r¢   r    )r£   )Z
dictionaryZpostfixZnewkeyr-   r-   r.   Úrand_dict_key$  s    r¥   ©Úinstance_data_file©r^   c             
   C   sì   ddl m}m}m} zt| ƒ}W n: tk
rZ } z|jtkrHi  W Y ¢S ‚ W 5 d}~X Y nX |rÔtj	 
|¡rÔz||| |ƒ}t d|| ¡ W nH |k
rž   Y n6 |k
rÒ } zt d||t|ƒ¡ W 5 d}~X Y nX |dkrài S t|i d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Úerrnor   r)   rm   ÚexistsrH   r|   rI   ÚreprÚ	load_yaml)rN   r§   r©   rª   r«   Zconfig_filer}   r-   r-   r.   Ú	read_conf.  s>    
ýüûr´   c                  G   s   t t| Ž ƒS rS   )ÚsortedÚ
uniq_merge)Úlistsr-   r-   r.   Úuniq_merge_sorted`  s    r¸   c                  G   sF   g }| D ]4}t |tƒr2| ¡  d¡}dd„ |D ƒ}| |¡ qt|ƒS )Nr@   c                 S   s   g | ]}|r|‘qS r-   r-   )r;   Úar-   r-   r.   r?   r  s      zuniq_merge.<locals>.<listcomp>)rT   rU   r4   r,   ÚextendÚ	uniq_list)r·   Zcombined_listÚa_listr-   r-   r.   r¶   l  s    
r¶   c                 C   s`   t  ¡ D ]\}}|  ||¡} qg }| D ]}|tkr&| |¡ q&|D ]}|  |d¡} qB|  ¡ } | S )Nr™   )ÚFN_REPLACEMENTSÚitemsÚreplaceÚ
FN_ALLOWEDÚappendr4   )Úfnr<   rB   Zremovalsr-   r-   r.   Úclean_filenamew  s    rÃ   Tc              
   C   s®   zht  t| ƒ¡}t t d dd|¡¡:}|rFt| ¡ ƒW  5 Q R £ W S | ¡ W  5 Q R £ W S W 5 Q R X W n@ t	k
r¨ } z"|rŠ|  W Y ¢S t
t|ƒƒ|‚W 5 d }~X Y nX d S )NÚrbr…   )ÚioÚBytesIOr]   Ú
contextlibÚclosingÚgzipZGzipFilerY   Úreadrˆ   r„   rU   )r>   ÚquietrV   ÚbufZghr}   r-   r-   r.   Údecomp_gzip„  s    $rÍ   c                 C   s~   | sdS |   dd¡}|d  ¡ }t|ƒdkr:|d  ¡ }nd }|rV|dksV| ¡ dkrZd }|rr|dksr| ¡ dkrvd }||fS )N)NNr:   r…   r   r%   z-1Znone)r,   r4   rG   r‘   )Zug_pairZ	ug_partedÚuÚgr-   r-   r.   Úextract_usergroup”  s    rÐ   )Úroot_dirr^   c                 C   sh   t ƒ }t tj | d¡¡D ]H}tj |¡s,qtj |¡dd… }| ¡ }|r| d¡dkr|||< q|S )Nz*.pyr   éýÿÿÿr$   éÿÿÿÿ)	rL   Úglobr)   rm   rJ   ÚisfileÚbasenamer4   Úfind)rÑ   ÚentriesrN   Úmodnamer-   r-   r.   Úget_modules_from_dir¤  s    
rÚ   c              	   C   s,   t | dƒ}| |¡ | ¡  W 5 Q R X d S )NÚw)ÚopenÚwriteÚflush)Úconpathr\   Zwfhr-   r-   r.   Úwrite_to_console°  s    
rà   c           	      C   sÂ   |rt j | ¡ |rŒd}d}tj |¡rxzt|| ƒ d}W n< tk
rv   d}t j |› d¡ |rr| 	t
j|¡ Y nX |rŒ|sŒt j | ¡ |r¾| d dkr²| 	|| d d… ¡ n| 	|| ¡ d S )Nz/dev/consoleFTzFailed to write to /dev/consoleÚ
rÓ   )ÚsysÚstderrrÝ   r)   rm   r±   rà   r{   r3   r   ÚloggingZWARNING)	r\   Zconsolerã   r   Z	log_levelZfallback_to_stdoutrß   Zwriting_to_console_workedZconsole_errorr-   r-   r.   Ú	multi_log¶  s(    
rå   c                   C   s   dt  ¡ kS )NZLinux©ÚplatformÚsystemr-   r-   r-   r.   Úis_Linuxâ  s    ré   c                   C   s$   dt  ¡ krdS t  ¡ dkr dS dS )NZBSDTZ	DragonFlyFræ   r-   r-   r-   r.   Úis_BSDç  s
    rê   c                   C   s   t ƒ d dkS )NÚvariantÚfreebsd©Úsystem_infor-   r-   r-   r.   Ú
is_FreeBSDð  s    rï   c                   C   s   t ƒ d dkS )Nrë   Ú	dragonflyrí   r-   r-   r-   r.   Úis_DragonFlyBSDõ  s    rñ   c                   C   s   t ƒ d dkS )Nrë   Únetbsdrí   r-   r-   r-   r.   Ú	is_NetBSDú  s    ró   c                   C   s   t ƒ d dkS )Nrë   Úopenbsdrí   r-   r-   r-   r.   Ú
is_OpenBSDÿ  s    rõ   Fc                 C   s   || kr|S t | | ƒS rS   )r—   ©ÚyobjÚkeyr­   r-   r-   r.   Úget_cfg_option_bool  s    rù   c                 C   s*   || kr|S | | }t |tƒs&t|ƒ}|S rS   )rT   rU   )r÷   rø   r­   rO   r-   r-   r.   Úget_cfg_option_str
  s    
rú   c                 C   s   t t| ||dƒS )Nr¬   )r(   rú   rö   r-   r-   r.   Úget_cfg_option_int  s    rû   c                 C   sª   | sd} t j | ¡si S t| ƒ}d}d|kr0d}t ||¡}|r¦| ¡ }d|d kr`|d |d< |d  ¡  d¡d |d< |d d	krŽd
|d< |d |d |d dœS i S )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\.]+)Únamer7   z linuxr   zred hat enterpriseÚredhatr   )ÚIDÚ
VERSION_IDÚVERSION_CODENAME)	r)   rm   r±   r®   ÚreÚmatchÚ	groupdictr‘   rE   )Zrelease_fileZredhat_releaseZredhat_regexr  Úgroupr-   r-   r.   Ú_parse_redhat_release  s,    ÿýr  c            	   	   C   sp  d} d}d}i }d}t j d¡r,ttdƒƒ}|s:d}tƒ }|rê| dd¡} | dd¡}d| ksfd| krpt ¡ }nl| d	ks€| d
krŽ| dd¡}nN| dkr¨|s¨| dd¡}n4| dd¡}|sÜt	 
d| dd¡¡}|rÜ| ¡ d }| dkrèd} n|tƒ rt ¡  ¡ } t ¡ }n^d}z(zt ¡ }W n tk
r0   Y nX W 5 d }|D ]}|r>d}q>|s`t d¡ X |S | ||fS )Nr™   Fú/etc/os-releaseTrþ   rÿ   ÚslesÚsuseÚalpineÚphotonZPRETTY_NAMEÚ	virtuozzor   z[^ ]+ \((?P<codename>[^)]+)\)ZVERSIONr7   Úrhelrý   )r™   r™   r™   r…   zPUnable to determine distribution, template expansion may have unexpected results)r)   rm   r±   Úload_shell_contentr®   r  Úgetrç   Úmachiner  r  r  rê   rè   r‘   r+   rH   rI   Údistrˆ   )	Zdistro_nameZdistro_versionZflavorZ
os_releaseZos_release_rhelr  r  ÚfoundÚentryr-   r-   r.   Úget_linux_distro?  s^    

þ

ÿr  c                 C   sv   | d   ¡ }d}|dkrf| d d   ¡ }|dkr6|}qr|dkrDd}qr|d	krRd
}qr|dkr`d}qrd}n|dkrr|}|S )Nrè   ÚunknownZlinuxr  r   )Z	almalinuxr	  ZarchZcentosZ
cloudlinuxZdebianZ	eurolinuxZfedoraZmarinerZmiraclelinuxZ	openeulerZopencloudosZopenmandrivar
  r  Zrockyr  Z	tencentosr  )ÚubuntuZ	linuxmintZmintr  rý   r  )Zopensusezopensuse-leapzopensuse-microoszopensuse-tumbleweedZsle_hpcz	sle-micror  r  )ZwindowsÚdarwinrì   rò   rô   rð   )r‘   )Úinforè   rë   Z
linux_distr-   r-   r.   Ú_get_variant}  s     	r  c                  C   s<   t   ¡ t  ¡ t  ¡ t  ¡ tt  ¡ ƒtƒ dœ} t| ƒ| d< | S )N)rç   rè   r+   Úpythonr*   r  rë   )rç   rè   r+   Zpython_versionr   r*   r  r  )r  r-   r-   r.   rî   µ  s    
úrî   c                 C   sX   || kr|S | | dkrg S | | }t |tƒr@dd„ |D ƒ}|S t |tƒsRt|ƒ}|gS )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.
    Nc                 S   s   g | ]}|‘qS r-   r-   rA   r-   r-   r.   r?   Õ  s     z'get_cfg_option_list.<locals>.<listcomp>)rT   r   rU   )r÷   rø   r­   rO   Zcvalr-   r-   r.   Úget_cfg_option_listÃ  s    

r  c                 C   s>   t |tƒr| d¡}| }|D ]}||kr0|  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.ú/)rT   rU   r,   )r÷   Zkeypr­   ZcurÚtokr-   r-   r.   Úget_cfg_by_pathÞ  s    


r  c                 C   s    t | |ƒ\}}t||ƒ ||fS rS   )Úget_output_cfgÚredirect_output)ÚcfgÚmodeÚoutfmtÚerrfmtr-   r-   r.   Úfixup_outputö  s    
r$  c           
      C   s¤  t tj d¡ƒrt d¡ d S |s(tj}|s2tj}dd„ }| rþt d|| ¡ |  	dd¡\}}|dksl|d	krˆd
}|dkr|d}t
||ƒ}n0|dkr¬tj|dtj|d}	|	j}ntd|  ƒ‚|rÐt | ¡ | ¡ ¡ || krþt d|| ¡ t | ¡ | ¡ ¡ d S |r t d||¡ | 	dd¡\}}|dks6|d	krTd
}|dkrHd}t
||ƒ}n2|dkrztj|dtj|d}	|	j}ntd| ƒ‚|r t | ¡ | ¡ ¡ d S )NZ_CLOUD_INIT_SAVE_STDOUTz5Not redirecting output due to _CLOUD_INIT_SAVE_STDOUTc                  S   s>   t  d¡ zt d¡j} W n tk
r.   Y nX t  | ¡ dS )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)
        é   ZadmN)r)   ÚumaskÚgrpÚgetgrnamÚgr_gidÚKeyErrorÚsetgid)Zgroup_idr-   r-   r.   Úset_subprocess_umask_and_gid  s    

z5redirect_output.<locals>.set_subprocess_umask_and_gidzRedirecting %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  rH   r|   râ   r3   rã   r,   rÜ   Ú
subprocessÚPopenÚPIPEr4  r_   Údup2Úfileno)
r"  r#  Zo_outZo_errr,  r!  ÚargZowithZnew_fpÚprocr-   r-   r.   r  	  sd    
ü

ür  )Úsourcesr^   c                 C   sR   |rt t| ƒƒ} i }| D ]4}|rt |¡}|s6t ¡ }t |¡}| ||¡}q|S )aF  Merge multiple dicts according to the dict merger rules.

    Dict merger rules can be found in cloud-init documentation. If no mergers
    have been specified, entries will be recursively added, but no values
    get replaced if they already exist. Functionally, this means that the
    highest priority keys must be specified first.

    Example:
    a = {
        "a": 1,
        "b": 2,
        "c": [1, 2, 3],
        "d": {
            "a": 1,
            "b": 2,
        },
    }

    b = {
        "a": 10,
        "c": [4],
        "d": {
            "a": 3,
            "f": 10,
        },
        "e": 20,
    }

    mergemanydict([a, b]) results in:
    {
        'a': 1,
        'b': 2,
        'c': [1, 2, 3],
        'd': {
            'a': 1,
            'b': 2,
            'f': 10,
        },
        'e': 20,
    }
    )r   Úreversedr   Zdict_extract_mergersZdefault_mergersZ	constructÚmerge)r=  ÚreverseZ
merged_cfgr   Zmergers_to_applyZmergerr-   r-   r.   ÚmergemanydictY  s    *

rA  c              	   c   s.   t  ¡ }zt  | ¡ | V  W 5 t  |¡ X d S rS   )r)   ÚgetcwdÚchdir)ZndirZcurrr-   r-   r.   rC  ‘  s
    

rC  c              	   c   s&   t  | ¡}z
|V  W 5 t  |¡ X d S rS   )r)   r&  )Zn_mskÚoldr-   r-   r.   r&  ›  s    

r&  c                 C   s   dj | |d|dS )Nz{0:{fill}{align}{size}}ú^)ÚfillZalignÚsize)Úformat)r\   rF  Zmax_lenr-   r-   r.   Úcenter¤  s       ÿrI  c                 C   s   t  d| ¡ t | ¡ d S )NzRecursively deleting %s)rH   r|   ÚshutilZrmtree©rm   r-   r-   r.   Údel_dirª  s    rL  r™   é   c              
   C   sr   z0t |||ƒ\}}}|| d< || d< || d< W dS  tjk
rl } z|jtjkrZW Y ¢
dS ‚ W 5 d }~X Y nX d S )Nú	user-dataúvendor-dataú	meta-dataTF)Úread_seededr   ÚUrlErrorÚcodeZ	NOT_FOUND)rF  ÚbaseÚextÚtimeoutÚmdÚudÚvdr}   r-   r-   r.   Úread_optional_seed³  s    
rZ  c                 C   sæ   i }| sddg}n(t j |  d¡d¡t j |  d¡d¡g}t|ƒ}dd„ |D ƒ}d }|D ],}t j t j |d¡¡rXt j |d¡} q†qXd }|D ],}t j t j |d¡¡rŽt j |d¡} q¼qŽ|rÖ|rÖ||d	< ||d
< n|râ||d	< |S )Nz/var/lib/cloud/data/sslz /var/lib/cloud/instance/data/sslr>   Zsslc                 S   s    g | ]}|rt j |¡r|‘qS r-   )r)   rm   Úisdir)r;   Údr-   r-   r.   r?   Î  s       z%fetch_ssl_details.<locals>.<listcomp>zcert.pemzkey.pemÚ	cert_fileÚkey_file)r)   rm   rJ   Zget_ipath_curZ	get_cpathr¶   rÕ   )ÚpathsZssl_detailsZssl_cert_pathsr]  r\  r^  r-   r-   r.   Úfetch_ssl_detailsÀ  s4    þþ
r`  c              
   C   s   |}t | ƒ} z\t dt| ƒ|¡ t | ¡}|d krBt d¡ |}n t||ƒsbtd|t 	|¡f ƒ‚|}W n² tj
ttfk
r } zŠd}d }t|dƒr¬t|dƒr¬t|dƒ}nt|dƒrÊt|dƒrÊt|dƒ}|rð|dj|jd |jd |d	7 }n|d
j|d7 }t |¡ W 5 d }~X Y nX |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 insteadzFailed loading yaml blobZcontext_markZproblem_markz5. Invalid format at line {line} column {col}: "{err}"r…   )rM   ÚcolrQ   z. {err})rQ   )rY   rH   r|   rG   r   ÚloadrT   r_   r   rŠ   Z	YAMLErrorÚ
ValueErrorÚhasattrÚgetattrrH  rM   ÚcolumnrI   )rW   r­   ZallowedZloadedZ	convertedr}   ÚmsgZmarkr-   r-   r.   r³   á  sJ    ü


ÿÿ
  ÿÿr³   é
   c              
   C   sV  |   d¡dkr@|  dd| ¡}|  dd| ¡}|  dd| ¡}nTtjrj| d dkrjt | ¡jdkrj| d7 } d	| d|f }d	| d|f }d	| d|f }tj|||d
}d }	| 	¡ rÂt
t|jƒi d}	tj|||d
}
d }|
 	¡ rä|
j}d }ztj|||d
}W n2 tjk
r. } zt d|¡ W 5 d }~X Y nX | 	¡ rB|j}n
t d¡ |	||fS )Nz%sr   rN  rO  rP  rÓ   r  r™   z%s%s%s)rV  Úretriesr¬   z!Error in vendor-data response: %szError in vendor-data response)r×   r¿   r   Z%NOCLOUD_SEED_URL_APPEND_FORWARD_SLASHr   ÚurlparseZqueryr   Zread_file_or_urlÚokr³   rY   ÚcontentsrR  rH   r|   )rT  rU  rV  ri  Zfile_retriesZud_urlZvd_urlZmd_urlZmd_resprW  Zud_resprX  rY  Zvd_respr}   r-   r-   r.   rQ  	  sN      ÿ  ÿ  ÿ


rQ  c                   s¤   t t ˆ ¡dd}dd„ |D ƒ}‡ fdd„|D ƒ}g }|D ]`}z | ttj ˆ |¡|d¡ W q: tk
r˜ } z|jt	krˆt
 dˆ |¡ W 5 d}~X Y q:X q:t|ƒS )	zRead configuration directory.T©r@  c                 S   s   g | ]}|  d ¡r|‘qS )z.cfg)Úendswith©r;   Úfr-   r-   r.   r?   :  s     
 zread_conf_d.<locals>.<listcomp>c                    s&   g | ]}t j t j ˆ |¡¡r|‘qS r-   )r)   rm   rÕ   rJ   ro  ©Úconfdr-   r.   r?   =  s      r¦   z,REDACTED config part %s/%s for non-root userN)rµ   r)   ÚlistdirrÁ   r´   rm   rJ   r{   r°   r   rH   rI   rA  )rr  r§   ZconfsÚcfgsrÂ   r}   r-   rq  r.   Úread_conf_d4  s(    þÿ
  ÿru  c             
   C   sð   t ƒ }i }zt| |d}W n8 tk
rR } z|jtkrBt d| ¡ W 5 d}~X Y nX | |¡ d}d|kr¦|d }|rÂt|t	ƒs˜t
d| t |¡f ƒ‚qÂt	|ƒ ¡ }ntj | › d¡rÂ| › d}|rètj |¡rèt||d}| |¡ t|ƒS )a  Read 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™   Zconf_dz8Config file %s contains 'conf_d' with non-string type %sz.d)r   r´   r{   r°   r   rH   rI   rÁ   rT   rU   r_   r   rŠ   r4   r)   rm   r[  ru  Ú
appendleftrA  )Zcfgfiler§   rt  r   r}   rr  Z	confd_cfgr-   r-   r.   Úread_conf_with_confdR  s2    


ÿÿ

rw  c                 C   s   t t| dƒS )N©Úcmdline)r³   Úread_cc_from_cmdlinerx  r-   r-   r.   Úread_conf_from_cmdline~  s    r{  c           	      C   s¤   | d krt ƒ } d}d}t|ƒ}t|ƒ}t| ƒ}g }|  |¡}|dkrš|  ||| ¡}|dk r`|}| t | || |…  ¡ ¡ dd¡¡ |  ||| ¡}q<d |¡S )Nzcc:Zend_ccr   z\nrá   )	Úget_cmdlinerG   r×   rÁ   r   ZunquoteÚlstripr¿   rJ   )	ry  Z	tag_beginZtag_endZbegin_lZend_lZclenÚtokensZbeginÚendr-   r-   r.   rz  ƒ  s*    

 ÿÿrz  c                 C   s2   |   d¡}|dks"| |d  dkr&| S |  dd¡S )Nrá   r   r…   úz
)r×   r¿   )rl  Úposr-   r-   r.   Údos2unix¥  s    
r‚  ÚHostnameFqdnInfoÚhostnameÚfqdnÚ
is_defaultc                 C   s¤   d}d| kr,| d }t | d| d¡d ƒ}nld| krf| d  d¡dkrf| d }| d d| d¡… }n2|jd|dj}d| krˆ| 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$   r   NT)r…  Úmetadata_only)r‡  )rú   r,   r×   Úget_hostnamer„  rƒ  )r   Zcloudr‡  r†  r…  r„  r-   r-   r.   Úget_hostname_fqdn³  s"     ÿ
ÿ
r‰  ú
/etc/hostsc                 C   s’   d}ztt |ƒ ¡ D ]b}| d¡}|dkr4|d|… }| ¡ }|sBq| ¡ }t|ƒdk rXq| |dd… kr|d } qvqW n tk
rŒ   Y nX |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®   rD   r×   r4   r,   rG   r¯   )r„  Úfilenamer…  rM   ZhashposÚtoksr-   r-   r.   Úget_fqdn_from_hostsÞ  s$    

r  c              
   C   sD  t  | ¡}|j}tdkrÂtƒ }d}i }|D ]‚}z`t |dddtjtj¡}g ||< |D ]6\}}	}
}}||  	d||d f ¡ | 
|d ¡ qTW q* tjtjfk
rª   Y q*X q*|a|rÂt d|¡ z\ttƒ( t |j d¡¡rðW 5 Q R £ W dS W 5 Q R X t |d¡}|d d d }|tkW S  tjtjfk
r>   Y d	S X dS )
aî  determine if a url's network address 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)zdoes-not-exist.example.com.zexample.invalid.Z!__cloud_init_expected_not_found__r   z%s: %szdetected dns redirection: %sz[]Té   F)r   rj  r„  Ú_DNS_REDIRECT_IPÚsetÚsocketZgetaddrinfoZSOCK_STREAMZAI_CANONNAMErÁ   ÚaddZgaierrorÚerrorrH   r|   r   rc  r   Zis_ip_addressZnetlocr4   )ÚurlZ
parsed_urlrü   ZbadipsZbadnamesZ
badresultsZinameÚresultZ_famZ_stypeZ_protoZcnameZsockaddrZaddrr-   r-   r.   Úis_resolvable  sD    
     ÿ

r˜  c                  C   s   t  ¡ } | S rS   )r“  Zgethostname)r„  r-   r-   r.   rˆ  9  s    rˆ  c                 C   s.   zt  | ¡d W S  t jk
r(   Y d S X d S ©Nr   )r“  ÚgethostbyaddrZherror)Zipr-   r-   r.   rš  >  s    rš  c                 C   s   t tjd|  t| fdS )z5determine if this url is resolvable (existing or ip).zResolving URL: )Úlogfuncrg  Úfuncr‹   )Úlog_timerH   r|   r˜  )r–  r-   r-   r.   Úis_resolvable_urlE  s    ürž  c              	   C   s^   | dkrdS t  d| ¡ | D ]<}z"t|ƒr@t  d|¡ |W   S W q tk
rV   Y qX qdS )zc
    Search through a list of mirror urls for one that works
    This needs to return quickly.
    Nz%search for mirror in candidates: '%s'zfound working mirror: '%s')rH   r|   rž  rˆ   )Z
candidatesZcandr-   r-   r.   Úsearch_for_mirrorO  s    rŸ  c               	   C   sD   t tj d¡ƒrdS ttjƒ} t |  ¡ tj	 ¡ ¡ W 5 Q R X dS )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)   r5  r  rÜ   Údevnullr9  r:  râ   r4  )Úfpr-   r-   r.   Úclose_stdinb  s    r¢  Údevicec                 C   sv   g }| st   d¡t   d¡ S |  d¡rL|  d¡}dd„ d| d| fD ƒ}n&| dkr`t   d¡}n| d	krrt   d¡}|S )
Nz/dev/msdosfs/*z/dev/iso9660/*úLABEL=c                 S   s   g | ]}t j |¡r|‘qS r-   ©r)   rm   r±   )r;   Úpr-   r-   r.   r?   x  s   þz*find_devs_with_freebsd.<locals>.<listcomp>z/dev/msdosfs/z/dev/iso9660/ú	TYPE=vfatúTYPE=iso9660)rÔ   Ú
startswithr}  )ÚcriteriaÚoformatÚtagÚno_cacherm   ÚdevlistÚlabelr-   r-   r.   Úfind_devs_with_freebsdp  s    

þ
r°  c                 C   sÈ   g }d }d }| r8|   d¡r$|  d¡}|   d¡r8|  d¡}tjdddgdgd}|j ¡ D ]j}	|sd|r~tjd|	gdd	gd\}
}|rd
| |
krqX|dkr¢d|
kr¢qX|dkr´d|
kr´qX| d|	 ¡ qX|S )Nr¤  zTYPE=Úsysctlú-núhw.disknamesr   ©ZrcsZ	mscdlabelr…   z
label "%s"Úiso9660zISO filesystemÚvfatú/dev/)r©  r}  r   r3   r,   rÁ   )rª  r«  r¬  r­  rm   r®  r¯  Ú_typer5   ÚdevZmscdlabel_outr    r-   r-   r.   Úfind_devs_with_netbsd„  s(    



rº  c                 C   sŠ   t j dddgdgd}g }|j ¡  d¡D ]P}| d¡s:q*|dkrDq*| |d d	… d
 ¡ | d¡s*| |d d	… d ¡ q*dd„ |D ƒS )Nr±  r²  r³  r   r´  r@   r:   zfd0:rÓ   r¹   ÚcdÚic                 S   s   g | ]}d | ‘qS ©r·  r-   ©r;   r¼  r-   r-   r.   r?   «  s     z*find_devs_with_openbsd.<locals>.<listcomp>)r   r3   Úrstripr,   rn  rÁ   r©  )rª  r«  r¬  r­  rm   r5   r®  r  r-   r-   r.   Úfind_devs_with_openbsd  s    

rÀ  c                 C   s€   t j dddgdgd}dd„ t|j ¡ dd	D ƒ}| d
krJdd„ |D ƒ}n(| dkrbdd„ |D ƒ}n| rrt d| ¡ dd„ |D ƒS )Nr±  r²  z
kern.disksr   r´  c                 S   s$   g | ]}|  d ¡s|  d¡s|‘qS )rW  Zvn©r©  r¾  r-   r-   r.   r?   ²  s   
 
þz/find_devs_with_dragonflybsd.<locals>.<listcomp>Trm  r¨  c                 S   s   g | ]}|  d ¡r|‘qS ©)r»  ZacdrÁ  r¾  r-   r-   r.   r?   ¹  s     
 )zLABEL=CONFIG-2r§  c                 S   s   g | ]}|  d ¡s|‘qS rÂ  rÁ  r¾  r-   r-   r.   r?   »  s     
 zUnexpected criteria: %sc                 S   s   g | ]}d | ‘qS r½  r-   r¾  r-   r-   r.   r?   ¾  s     )r   rµ   r3   r,   rH   r|   )rª  r«  r¬  r­  rm   r5   r®  r-   r-   r.   Úfind_devs_with_dragonflybsd®  s    þrÃ  c              
   C   sF  t ƒ rt| ||||ƒS tƒ r,t| ||||ƒS tƒ rBt| ||||ƒS tƒ rXt| ||||ƒS dg}g }| rt| d|  ¡ |r†| d| ¡ |r˜| 	ddg¡ |rª| d| ¡ |r¸| |¡ || }zt
j
|ddgd	\}}	W n: t
jk
r }
 z|
jtkrd
}n‚ W 5 d}
~
X Y nX g }| ¡ 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%sr   r%   r´  r™   N)rï   r°  ró   rº  rõ   rÀ  rñ   rÃ  rÁ   rº   r   rK   r°   r   rD   r4   )rª  r«  r¬  r­  rm   Z
blk_id_cmdÚoptionsÚcmdr5   Ú_errr}   rØ   rM   r-   r-   r.   Úfind_devs_withÁ  sN    
    ÿ
rÊ  c           	      C   sŒ   | dkrg } nt | ƒ} dddg}|r2| ddg¡ | | ¡ tj|ddd	}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Ä  ú-oZfullrÅ  rÆ  Tr¿   )r1   rV   r:   ZDEVNAME)r   rº   r   r3   rD   rE   r  )	ZdevsZdisable_cacherÈ  r5   ÚretrM   r¹  r    r>   r-   r-   r.   rÄ    s    

rÄ  c                 C   s(   g }| D ]}||krqq|  |¡ q|S rS   )rÁ   )Zin_listZout_listr¼  r-   r-   r.   r»      s    r»   c              
   C   s¢   t  d| |¡ t ¡ }z(t| dƒ}t|||d W 5 Q R X W n4 tk
rr } z|sV‚ |jtkrb‚ W 5 d }~X Y nX | 	¡ }t  dt
|ƒ| ¡ |ršt|ƒS |S d S )NzReading from %s (quiet=%s)rÄ   )Úchunk_cbzRead %s bytes from %s)rH   r|   rÅ   rÆ   rÜ   Úpipe_in_outr¯   r°   r   ÚgetvaluerG   rY   )rN   Zread_cbrË   rV   ZofhZifhr}   rl  r-   r-   r.   r®   *  s    
r®   c               
   C   sŠ   t ƒ r\z tdƒ} |  dd¡d d… }W q† tk
rX } zt d|¡ d}W 5 d }~X Y q†X n*ztdƒ ¡ }W n tk
r„   d}Y nX |S )Nz/proc/1/cmdlineú r-  rÓ   z"failed reading /proc/1/cmdline: %sr™   z/proc/cmdline)Úis_containerr®   r¿   rˆ   rH   rI   r4   )rl  ry  r}   r-   r-   r.   Ú_get_cmdline=  s    
rÒ  c                   C   s   dt jkrt jd S tƒ S )NZDEBUG_PROC_CMDLINE)r)   r5  rÒ  r-   r-   r-   r.   r|  P  s    

r|  c               	   C   s:   d} zt | ƒ ¡ }|dkW S  ttfk
r4   Y dS X d S )Nz/proc/sys/crypto/fips_enabledr"   F)r®   r4   r¯   r{   )Z	fips_procrl  r-   r-   r.   Úfips_enabledW  s    
rÓ  é   c                 C   sN   d}|   |¡}t|ƒdkrqBq| |¡ |t|ƒ7 }|r||ƒ q| ¡  |S r™  )rÊ   rG   rÝ   rÞ   )Zin_fhZout_fhZ
chunk_sizerÍ  Zbytes_pipedr>   r-   r-   r.   rÎ  c  s    


rÎ  c                 C   s6   |dkr|dkrd S t  d| ||¡ t | ||¡ d S )N)NrÓ   z%Changing the ownership of %s to %s:%s)rH   r|   r)   Úchown)rN   ÚuidÚgidr-   r-   r.   Ú	chownbyidr  s    rØ  c              
   C   sn   d}d}z$|rt  |¡j}|r*t |¡j}W n0 tk
r\ } ztd| ƒ|‚W 5 d }~X Y nX t| ||ƒ d S )NrÓ   zUnknown user or group: %s)	ÚpwdÚgetpwnamZpw_uidr'  r(  r)  r*  r{   rØ  )rN   Úuserr  rÖ  r×  r}   r-   r-   r.   Úchownbynamez  s     rÜ  c           
      C   st  d d g}| rd| kr|S | d }||kr2|| }nd|kr>|S |d }t |tƒrX||g}t |tƒr’t|ƒdkrz|d |d< t|ƒdkr’|d |d< t |tƒrÄd|kr°|d |d< d|krÄ|d |d< |d dkrÜ|d |d< ddd	g}tt|ƒƒD ]|}|| sqò||  ¡ }d
}|D ]8}	| |	¡rd|	|t|	ƒd …  ¡ f }d} qPq|sfdd| ¡ f }|||< qò|S )NÚoutputÚallr   r…   r•  z&1r/  r.  r2  Fz%s %sT)	rT   rU   r   rG   rL   r¡   r}  r©  r4   )
r   r!  rÌ  ZoutcfgZmodecfgZswlistr¼  rO   r  Úsr-   r-   r.   r    sL    







r  c                 C   s®   g }| rt | tƒs|S |  d¡}|r.| |¡ t| dƒD ]h}|sBq8t d|¡}|sTq8| d¡}| ¡ }t	|ƒdkr~| |¡ q8ddg|dd… kr8| |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>.*)r2   r…   Úteez-ar%   )rT   rL   r  rÁ   r  r  r  r  r,   rG   r   r’  )r   ZlogsZdefault_logÚfmtr  r2   Úpartsr-   r-   r.   Úget_config_logfilesÊ  s&    


rã  c                 G   sB   |r| j |f|žŽ  t ¡ }|dkr(d }| j|f|žd|iŽ d S )N)NNNÚexc_info)rI   râ   rä  r|   )r   rg  r‹   rä  r-   r-   r.   r‰   ä  s    r‰   )Úroutiner^   c                 C   s<   t  |¡}| t| ƒ¡ | ¡ }|d k	r4|d|… S |S d S r™  )ÚhashlibÚnewÚupdater]   Z	hexdigest)rW   rå  ZmlenÚhasherÚdigestr-   r-   r.   Ú	hash_blobö  s    
rë  )rp  r^   c                    s2   t  ¡ }t‡ fdd„dƒD ]}| |¡ q| ¡ S )zˆHash the content of a binary buffer using SHA1.

    @param f: buffered binary stream to hash.
    @return: digested data as bytes.
    c                      s   ˆ   tj¡S rS   )rÊ   rÅ   ÚDEFAULT_BUFFER_SIZEr-   ©rp  r-   r.   Ú<lambda>  ó    zhash_buffer.<locals>.<lambda>rï  )ræ  Zsha1Úiterrè  rê  )rp  ré  Úchunkr-   rí  r.   Úhash_buffer  s    rò  c                 C   s0   zt  | ¡rW dS W n tk
r*   Y dS X d S rq   )rÙ  rÚ  r*  ©rü   r-   r-   r.   Úis_user  s
    

rô  c                 C   s0   zt  | ¡rW dS W n tk
r*   Y dS X d S rq   )r'  r(  r*  ró  r-   r-   r.   Úis_group  s
    

rõ  c                 C   s   t  d| |¡ t | |¡ d S )NzRenaming %s to %s)rH   r|   r)   Úrename©ÚsrcÚdestr-   r-   r.   rö    s    rö  éí  c                 C   s   | D ]}t ||ƒ qd S rS   )Ú
ensure_dir)Údirlistr!  r\  r-   r-   r.   Úensure_dirs#  s    rý  c                 C   sH   t  t| ƒ¡}t|t|ƒƒsDd dd„ |D ƒ¡}td|t|ƒf ƒ‚|S )Nú, c                 S   s   g | ]}t |ƒ‘qS r-   )rU   )r;   Útr-   r-   r.   r?   +  s     zload_json.<locals>.<listcomp>z((%s) root types expected, got %s instead)ÚjsonÚloadsrY   rT   r&   rJ   r_   r`   )r\   Z
root_typesZdecodedZexpected_typesr-   r-   r.   Ú	load_json(  s    
ÿÿr  c                 C   s<   t j | ¡}|t j |¡kr | S t j |¡r0| S t|ƒS dS )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)   rm   Údirnamer[  Úget_non_exist_parent_dir)rm   Zp_pathr-   r-   r.   r  3  s    r  c              	   C   s°   t j | ¡s¢t| ƒ}tt j | ¡dd t  | ¡ W 5 Q R X t| |ƒ |sP|r¬t|||ƒ t	|  
|¡d dd … ƒ}t	|ƒ}|jD ]}| |¡}t|||ƒ q„n
t| |ƒ d S )NTru   r…   )r)   rm   r[  r  ri   r  ÚmakedirsÚchmodrÜ  r
   r,   râ  Zjoinpath)rm   r!  rÛ  r  Znon_existed_parent_dirZsub_relative_dirZsub_pathrf   r-   r-   r.   rû  D  s    


rû  c              	   c   s(   z
| V  W 5 | r"d| g}t   |¡ X d S )NÚumount)r   )r  Z
umount_cmdr-   r-   r.   Ú	unmounter[  s
    
r  c               	   C   s  i } zât j d¡r$tdƒ ¡ }d}nt d¡}|j ¡ }d}d}|D ]}zV|dkrh| ¡ \}}}}	}
}n4t 	||¡}| 
d¡}| 
d¡}| 
d¡}| 
d¡}	W n tk
r¶   Y qDY nX | d	d
¡}|||	dœ| |< qDt d| |¡ W n$ ttfk
r
   ttdƒ Y nX | S )Nz/proc/mountsr<  Úmountz*^(/dev/[\S]+) on (/.*) \((.+), .+, (.+)\)$r…   r%   rŒ  r  z\040r-  )ÚfstypeÚ
mountpointÚoptszFetched %s mounts from %szFailed fetching mount points)r)   rm   r±   r®   rD   r   r3   r,   r  Úsearchr  rˆ   r¿   rH   r|   r¯   r{   r‰   )ÚmountedZ
mount_locsÚmethodr5   ZmountreZmpliner¹  Zmpr
  r  Z_freqZ_passnoÚmr-   r-   r.   Úmountse  s:    





ýr  c                 C   s&  t |tƒr|g}n:t |ttfƒr*t|ƒ}n"|dkr8d}ntdjt|ƒdƒ‚tƒ rb|dkr´dg}nRtƒ r®|dkrzdddg}t	|ƒD ](\}}|dkršd||< |d	kr‚d||< q‚nd
g}t
ƒ }t ¡ X}	d}
tj | ¡|krð|tj | ¡ d }nÊd}|D ]¨}d}zTdddg}|r | d|g¡ | | ¡ | |	¡ tj||d |	}
|	}W  q¢W qø ttfk
rž } z&|rŠt d| |d |¡|¡ |}W 5 d}~X Y qøX qø|sºtd| |	|f ƒ‚| d¡sÎ|d7 }t|
ƒ< |dkrì||ƒ}n
|||ƒ}|W  5 Q R £ W  5 Q R £ S Q R X W 5 Q R X dS )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ÚufsZcd9660Zmsdosrµ  )r¶  Zmsdosfsr™   Fr  r	  rË  Zroz-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  )rT   rU   r   r&   r_   rH  r`   ré   rê   Ú	enumerater  r   Ztempdirr)   rm   rw   rº   rÁ   r   r¯   r{   rH   r|   rJ   r‚   rn  r  )r£  Úcallbackr>   ZmtypeZupdate_env_for_mountZ	log_errorZmtypesÚindexr  Ztmpdr  r  Zfailure_reasonZmountcmdÚexcrÌ  r-   r-   r.   Úmount_cb  s€    

ÿÿ



ù	ÿÿ



r  c                   C   s
   t  t¡S rS   )Úobj_copyÚdeepcopyr   r-   r-   r-   r.   Úget_builtin_cfgë  s    r  c                 C   s   t  d| ¡ tj | ¡S )NzTesting if a link exists for %s)rH   r|   r)   rm   ÚislinkrK  r-   r-   r.   Úis_linkð  s    r  c                 C   sh   t  d|| ¡ |rXtj |¡rXtj tj |¡dtdƒ ¡}t | |¡ t 	||¡ d S t | |¡ d S )Nz$Creating symbolic link from %r => %rZtmpr¤   )
rH   r|   r)   rm   rv   rJ   r  r£   Úsymlinkr¿   )ÚsourceÚlinkZforceZtmp_linkr-   r-   r.   Úsym_linkõ  s    r!  c              
   C   sP   t  d| ¡ zt | ¡ W n0 tk
rJ } z|jtkr:|‚W 5 d }~X Y nX d S )NzAttempting to remove %s)rH   r|   r)   Úunlinkr{   r°   r   )rm   r}   r-   r-   r.   Údel_file  s    
r#  c                 C   s   t  d| |¡ t | |¡ d S )NzCopying %s to %s)rH   r|   rJ  Úcopyr÷  r-   r-   r.   r$  
  s    r$  c                  C   s2   zt  dt  ¡ ¡} W n tk
r,   d} Y nX | S )Nz%a, %d %b %Y %H:%M:%S %zú??)ÚtimeÚstrftimeÚgmtimerˆ   )Ztsr-   r-   r.   Útime_rfc2822  s
    
r)  c                     s˜   ddl ‰ ddl‰ d} G ‡ fdd„dˆ jƒ}ˆ  ˆ j d¡¡}ˆ  ¡ }ˆ  |¡|_|ƒ }| 	d|  ˆ  
|¡ˆ  
|¡dd¡dkrŒ|j|jd	  S td
ƒ‚dS )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
    r   Nó    c                       s    e Zd Zd” jfd” jfgZdS )zboottime.<locals>.timevalÚtv_secÚtv_usecN)r   r€   r   Zc_int64Z_fields_r-   ©Úctypesr-   r.   Útimeval%  s   r/  Úcs   kern.boottimerÓ   g    €„.Az/Unable to retrieve kern.boottime on this system)r.  Zctypes.utilZ	StructureZCDLLÚutilZfind_libraryZc_size_tZsizeofÚvalueZsysctlbynameZbyrefr+  r,  ÚRuntimeError)Z
NULL_BYTESr/  ZlibcrG  rÌ   r-   r-  r.   Úboottime  s(    	ûùÿ
r4  c                  C   st   d} d}zDt j d¡r4d}tdƒ}|rJ| ¡ d } nd}tt ¡ tƒ  ƒ} W n" tk
rn   t	t
d| ƒ Y nX | S )Nr%  r  z/proc/uptimer   r.  z&Unable to read uptime using method: %s)r)   rm   r±   r®   r,   rU   r&  r4  rˆ   r‰   rH   )Z
uptime_strr  rl  r-   r-   r.   Úuptime:  s    r5  c                 C   s   t | |dd d d S )Nr0  )Úomoder!  ©Ú
write_file)rm   Úcontentr-   r-   r.   Úappend_fileM  s    r:  é¤  )Úpreserve_mode)r!  r<  r^   c                C   s   t | dd||d d S )Nr™   r0  )r9  r6  r!  r<  r7  )rm   r!  r<  r-   r-   r.   Úensure_fileQ  s        ÿr=  c              	   C   s*   z
t | ƒW S  ttfk
r$   Y d S X d S rS   )r(   rc  r_   )Zpossible_intr-   r-   r.   Úsafe_intY  s    
r>  c                 C   s   t | ƒdd…  dd¡}t |ƒdd…  dd¡}t|dd… |dd… ddd…  dƒ}t|dd… |dd… ddd…  dƒ}||k||k  S )aC  Compare two file modes in octal.

    If mode1 is less restrictive than mode2 return 1
    If mode1 is more restrictive than mode2 return -1
    If mode1 is same as mode2, return 0

    The comparison starts from the permission of the
    set of users in "others" and then works up to the
    permission of "user" set.
    r%   NrŒ  r#   rÒ   rÓ   r¤   )ÚoctÚrjustr(   )Zmode1Zmode2Z	mode1_octZ	mode2_octZm1Zm2r-   r-   r.   Úcompare_permission`  s
    ((rA  c              	   C   s4   t |ƒ}| r0|r0t| ƒ t | |¡ W 5 Q R X d S rS   )r>  ri   r)   r  )rm   r!  Z	real_moder-   r-   r.   r  w  s    
r  )Úgrp_namer^   c                 C   s:   d}zt  | ¡j}W n  tk
r4   t d| ¡ Y nX |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*  rH   r|   )rB  r×  r-   r-   r.   Úget_group_id~  s    rC  )rm   r^   c                 C   s   t  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.
    )ry   ÚS_IMODEr)   Úst_moderK  r-   r-   r.   Úget_permissionsŒ  s    rF  c                 C   s   t  | ¡}t |j¡jS )zw
    Returns the owner of the file/folder pointed by the path.

    @param path: The full path of the file/folder.
    )r)   ry   rÙ  ÚgetpwuidÚst_uidZpw_name©rm   Ústr-   r-   r.   Ú	get_owner—  s    
rK  c                 C   s   t  | ¡}t |j¡jS )zw
    Returns the group of the file/folder pointed by the path.

    @param path: The full path of the file/folder.
    )r)   ry   r'  ÚgetgrgidÚst_gidÚgr_namerI  r-   r-   r.   Ú	get_group¡  s    
rO  )Úusernamer^   c                 C   sJ   g }t  ¡ D ]}| |jkr| |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'  ZgetgrallZgr_memrÁ   rN  rÙ  rÚ  Zpw_gidrL  )rP  Úgroupsr  r×  r-   r-   r.   Úget_user_groups«  s    
rR  r1  )Úensure_dir_existsrÛ  r  c             
   C   sô   |r&zt | ƒ}W n tk
r$   Y nX |r@ttj | ¡||d d| ¡ krZt|ƒ}d}nt|ƒ}d}zd| }	W n t	k
rŽ   d| }	Y nX t
 d| ||	t|ƒ|¡ t| d. t| |ƒ}
|
 |¡ |
 ¡  W 5 Q R X W 5 Q R X t| |ƒ 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  ÚbrZ   Z
charactersz%oz%rzWriting to %s - %s: [%s] %s %srK  N)rF  r{   rû  r)   rm   r  r‘   r]   rY   r_   rH   r|   rG   ri   rÜ   rÝ   rÞ   r  )r  r9  r!  r6  r<  rS  rÛ  r  Z
write_typeZmode_rZfhr-   r-   r.   r8  »  s:    ú
r8  c                 C   s@   t  | ¡D ]0}t j | |¡}t j |¡r2t|ƒ q
t|ƒ q
dS )z™
    Deletes all contents of a directory without deleting the directory itself.

    @param dirname: The directory whose contents should be deleted.
    N)r)   rs  rm   rJ   r[  rL  r#  )r  ZnodeZnode_fullpathr-   r-   r.   Údelete_dir_contentsø  s
    
rU  r‹  Úcreatedc                 C   s6   t  ¡ }t| ƒ}|d| ¡ |f 7 }|dtƒ  7 }|S )Nz %s by cloud-init v. %sz on %s)r   Zversion_stringrU   Útitler)  )Zcomment_charrT  Zci_verÚheaderr-   r-   r.   Úmake_header	  s
    rY  c                 G   s   t j t jj| f|žŽ ¡S rS   )r)   rm   ÚabspathrJ   )rT  r_  r-   r-   r.   Úabs_join	  s    r[  c              	   C   sè   t | ttfƒs tdt | ¡ ƒ‚d}|r0|d7 }dd }d}| D ]–}t |ttfƒr–g }|D ]}| dt|ƒ d|¡ ¡ qZd	|d
 	|¡f }|d7 }q@t |tƒr¶d	||f }|d7 }q@|d krÀq@td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)ú'ú\r\  r\  r   z'%s'r\  z%s%s
r-  r…   zCUnable to shellify type '%s'. Expected list, string, tuple. Got: %szShellified %s commands.)rT   r&   r   r_   r   rŠ   rÁ   rU   r¿   rJ   rH   r|   )ZcmdlistZ
add_headerr9  ZescapedZ	cmds_mader‹   Zfixedrp  r-   r-   r.   Úshellify	  s<    ÿÿ


ÿÿr^  c                 C   sB   |r|   |¡r| t|ƒd … } |r>|  |¡r>| d t|ƒ … } | S rS   )r©  rG   rn  )rM   ÚprefixÚsuffixr-   r-   r.   Ústrip_prefix_suffix;	  s
    ra  c                 C   sB   t  | d ¡d krdS zt   | ¡ W n t jk
r<   Y dS X dS )Nr   FT)r   ÚwhichrK   )rÈ  r-   r-   r.   Ú_cmd_exits_zeroC	  s    rc  c                   C   s   t dddgƒS )Nzsystemd-detect-virtz--quietz--container©rc  r-   r-   r-   r.   Ú_is_container_systemdM	  s    re  c                   C   s
   t dgƒS )Nzlxc-is-containerrd  r-   r-   r-   r.   Ú_is_container_old_lxcQ	  s    rf  c                  C   sD   t ƒ s
dS dddg} t | d ¡d kr*dS t | ¡\}}| ¡ dkS )NFr±  z-qnzsecurity.jail.jailedr   r"   )rï   r   rb  r4   )rÈ  r5   r    r-   r-   r.   Ú_is_container_freebsdU	  s    
rg  c               	   C   sä   t ttf} | D ]}|ƒ r dS qz(tdƒ}d|kr8W dS d|krFW dS W n ttfk
r`   Y nX tj d¡r~tj d¡s~dS zHt	dƒ 
¡ }|D ]2}| d¡r| ¡  d	d¡\}}|d
kr W dS qW n ttfk
rÞ   Y nX dS )zH
    Checks to see if this code running in a container of some sort
    Tr…   Z	containerZLIBVIRT_LXC_UUIDz/proc/vzz/proc/bcz/proc/self/statuszVxID:r:   r#   F)re  rg  rf  Úget_proc_envr¯   r{   r)   rm   r[  r®   rD   r©  r4   r,   )ZchecksÚhelperZpid1envÚlinesrM   Z_keyrO   r-   r-   r.   rÑ  _	  s6    ý

rÑ  c                   C   s   t j d¡S )z2Check to see if we are running in a lxd container.z/dev/lxd/sockr¥  r-   r-   r-   r.   Úis_lxd	  s    rk  r¿   c              	   C   sœ   t j dt| ƒd¡}zt|dd}W n ttfk
r@   i  Y S X i }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/procr5  Frd   )r*  ó   =)rÐ  ú=r…   )	r)   rm   rJ   rU   r®   r¯   r{   rV   r,   )ÚpidrX   ÚerrorsrÂ   rl  ÚenvZnullZequalr  rü   rO   r-   r-   r.   rh  ’	  s"    

rh  c              	   C   sP   i }|   ¡ D ]>}z|  dd¡\}}W n tk
r@   |}d}Y nX |||< q|S )Nrm  r…   T)r,   rc  )ZkvstringrÌ  r  rø   rO   r-   r-   r.   Úkeyval_str_to_dict°	  s    

rq  c                 C   s&   |   d¡r| dd … } tj d|  ¡S )Nr·  rM  z/sys/class/block/%s/partition)r©  r)   rm   rÕ   )r£  r-   r-   r.   Úis_partition¼	  s    
rr  c                 C   s¦   t |tƒs|g}g }|D ]ˆ}t |tƒr2| |¡ qt |ttfƒr˜t|ƒdk sXt|ƒdkr`tdƒ‚t|ƒdkrˆ|d rˆ| | t|ƒ ¡ q| |d ¡ qtdƒ‚q|S )Nr…   r%   z Invalid package & version tuple.r   zInvalid package type.)rT   r   rU   rÁ   r&   rG   r3  )Zversion_fmtZpkgsZpkglistÚpkgr-   r-   r.   Úexpand_package_listÃ	  s     



rt  c              	   C   sÀ  dd„ |   d¡D ƒ}d}d}d}d}t|ƒD ]D\}	}
|
  ¡ }t|ƒdk rh| d|	d t|ƒ|
¡  dS |d }d	d„ |  d¡D ƒ}t|ƒt|ƒkr–q,tt|ƒt|ƒƒ}|d
|… |d
|… krÂq,|dk	rÜt|ƒt|ƒkrÜq,z| d¡}	W n, tk
r   | d|	d |
¡ Y  dS X z||	d  }||	d  }W n, tk
r`   | d|	d |
¡ Y  dS X |}|}|d }q,|r |r¼|r¼|r¼|r¼||||fS n|r¼|r¼|r¼|||fS dS )zRReturn the mount information for PATH given the lines from
    /proc/$$/mountinfo.c                 S   s   g | ]}|r|‘qS r-   r-   ©r;   r}   r-   r-   r.   r?   â	  s      z$parse_mount_info.<locals>.<listcomp>r  Nrh  z$Line %d has two few columns (%d): %sr…   r  c                 S   s   g | ]}|r|‘qS r-   r-   ru  r-   r-   r.   r?   ÷	  s      r   ú-z,Did not find column named '-' in line %d: %sr%   z/Too few columns after '-' column in line %d: %srM  )r,   r  rG   r|   Úminr  rc  Ú
IndexError)rm   Zmountinfo_linesr   Úget_mnt_optsÚpath_elementsÚdevpthÚfs_typeÚmatch_mount_pointÚmatch_mount_point_elementsr¼  rM   râ  Úmount_pointÚmount_point_elementsÚxÚmount_optionsr-   r-   r.   Úparse_mount_infoÞ	  sp       ÿ
ÿþ  ÿ
  ÿ


rƒ  c                 C   sB   t dƒ ¡ D ]0}| ¡ dd… \}}}|| kr|||f  S qdS )z<On older kernels there's no /proc/$$/mountinfo, so use mtab.ú	/etc/mtabNrŒ  )r®   rD   r,   )rm   rM   r{  r  r|  r-   r-   r.   Ú
parse_mtab,
  s
    r…  c                 C   s²   |   d¡}t|ƒdkr|d S t|ƒdkr2|d S |d dkr¢| dd … }t dd	d
g¡\}}|  d¡D ]0}|  ¡ }t|ƒdkrh|d |krh|d } qšqht|ƒS t d| ¡ d S )Nr  r…   r   rŒ  r%   )r¯  Zgptr  rM  ZglabelÚstatusz-srá   z)Unexpected input in find_freebsd_part: %s)r,   rG   r   rU   rH   rI   )ZfsZsplittedZtarget_labelrf   rÉ  Úlabelsr¾   r-   r-   r.   Úfind_freebsd_part5
  s    
rˆ  c                 C   sH   d }|  d¡D ]4}|  ¡ }t|ƒdkrtj |d |  ¡r|} qDq|S )Nrá   r%   r…   )r,   rG   r)   rm   r±   )rm   Úmnt_listÚ
path_foundrM   r¾   r-   r-   r.   Úget_path_dev_freebsdH
  s     r‹  c                 C   sl   t j dd| gddgd\}}t|ƒrPt   ddg¡\}}t| |ƒ}|d krLd S |}| ¡ }t|d ƒ}d| S )Nr	  ú-pr   r…   r´  r·  )r   rG   r‹  r,   rˆ  )rm   r—  rQ   r‰  rŠ  rÌ  Z
label_partr-   r-   r.   Úget_freebsd_devpthR
  s    
r  c              
   C   sÎ   t j d¡st d¡ d S zt dd| g¡\}}W n8 tjk
rj } zt d| |¡ W Y ¢d S d }~X Y nX t|ƒrxd S d}| 	d¡D ]B}t
 ||¡r†| |kr†d|kr†| 	¡ d	 }t d
| |¡ |  S q†d S )Nz/dev/zfsz"Cannot get zpool info, no /dev/zfsÚzpoolr†  z$Unable to get zpool status of %s: %sz.*(ONLINE).*rá   Ústater   zfound zpool "%s" on disk %s)r)   rm   r±   rH   r|   r   rK   rI   rG   r,   r  r  )rŽ  ZzpoolstatusrQ   r›   rM   Zdiskr-   r-   r.   Úget_device_info_from_zpool`
  s     
r  c                 C   sÐ  t   dg¡\}}d}dd„ |  d¡D ƒ}d}d}d}d}	| ¡ D ],}
t ||
¡}|sXq@| d¡}| d¡}d	d„ | d¡D ƒ}t|ƒt|ƒkr’q@tt|ƒt|ƒƒ}|d
|… |d
|… kr¾q@|	dk	rØt|	ƒt|ƒkrØq@|}|}	| d¡}| d¡}|dk	rd | d¡ 	d¡ 	¡  d¡¡}t
 d||||¡ t d|¡}|s\tƒ r\|dkr\t|ƒ}|}|| kr@ qpq@|r€|| kr„dS |r°|rÌ|rÌ|rÌ|rÌ||||fS n|rÌ|rÌ|rÌ|||fS dS )zƒReturn the mount information for PATH given the lines ``mount(1)``
    This function is compatible with ``util.parse_mount_info()``r	  ze^(?P<devpth>[\S]+?) on (?P<mountpoint>[\S]+?) (\(|type )(?P<type>[^,\(\) ]+)( \()?(?P<options>.*?)\)$c                 S   s   g | ]}|r|‘qS r-   r-   ru  r-   r-   r.   r?   
  s      zparse_mount.<locals>.<listcomp>r  Nr{  r  c                 S   s   g | ]}|r|‘qS r-   r-   ru  r-   r-   r.   r?   Œ
  s      r   r`   rÇ  r@   rþ  zNfound line in mount -> devpth: %s, mount_point: %s, fs_type: %s, options: '%s'z^(/dev/.+)[sp]([0-9])$)ZzfsZnfs)r   r,   rD   r  r  r  rG   rw  rJ   r4   rH   r|   rï   r  )rm   ry  ZmountoutputrÉ  Zregexrz  r{  r  r}  r~  rM   r  r€  r  r|  r‚  ZdevmZmatch_devpthr-   r-   r.   Úparse_mountt
  sl    ÿ


ÿþ


ÿúr‘  c                 C   sT   dt  ¡  }t j |¡r2t|ƒ ¡ }t| |||ƒS t j d¡rFt| ƒS t| |ƒS d S )Nz/proc/%s/mountinfor„  )	r)   Úgetpidrm   r±   r®   rD   rƒ  r…  r‘  )rm   r   ry  Zmountinfo_pathrj  r-   r-   r.   Úget_mount_infoË
  s    r“  )Úoptr^   c                 C   s    t | dd^ }}|| d¡kS )NT©ry  r@   ©r“  r,   )rm   r”  r    Zmnt_optsr-   r-   r.   Úhas_mount_optò
  s    r—  ÚT.)rœ  r^   c                 C   sô   |d krg }|d kri }t   ¡ }d }|rLzttƒ ƒ}W n tk
rJ   Y nX z|||Ž}W 5 t   ¡ | }d }	|d k	rœzttƒ ƒ| }	W n tk
rš   Y nX d| }
|rÈt|	tƒrÀ|
d|	 7 }
n|
d7 }
z| ||
 ƒ W n tk
rì   Y nX X |S )Nz took %0.3f secondsz (%0.2f)z (N/A))r&  Úfloatr5  rc  rT   rˆ   )r›  rg  rœ  r‹   rŒ   Z
get_uptimeÚstartZustartZdeltaZudeltaZtmsgrÌ  r-   r-   r.   r  ú
  s<    
r  c                 C   s(   |   dd¡}t|ƒdkr|S | d fS d S )Nr$   r…   )ÚrsplitrG   )ZdottedrŽ  r-   r-   r.   Úexpand_dotted_devname(  s    rœ  c                 C   s°   |d krg }|d krg }g }i }|| D ]d}zt | | | ddd||< W q( tk
rŠ } z"|jtkrh‚ ||krz| |¡ W 5 d }~X Y q(X q(t|ƒr¬tdjd |¡dƒ‚|S )NF)rË   rV   zMissing required files: {files}r@   )Úfiles)	r®   r¯   r°   r   rÁ   rG   rc  rH  rJ   )rT  ZrequiredZoptionalZdelimrP   rÌ  rp  r}   r-   r-   r.   Úpathprefix2dict0  s&    
ÿrž  ú/proc/meminfoc           	   	   C   s¦   dddddœ}dddd	œ}i }t | ƒ ¡ D ]v}z| ¡ \}}}W n$ tk
rd   | ¡ \}}d
}Y nX |r€t|ƒ||  ||< q*||kr*t|ƒ||  ||| < q*|S )NrÔ  é   r…   é   @)ZkBZmBÚBZgBZtotalZfreeZ	available)z	MemTotal:zMemFree:zMemAvailable:r¢  )r®   rD   r,   rc  r(   )	ZmeminfoÚrawÚmpliersZkmaprÌ  rM   rø   r2  Zunitr-   r-   r.   Úread_meminfoK  s"    ý
r¥  c              
   C   sÖ   | }|   d¡r| dd… } n|   d¡r2| dd… } dddd	d
dœ}| }d}|D ]$}|   |¡rN|}| dt|ƒ … }qNzt|ƒ}W n0 tk
r° } ztd| ƒ|‚W 5 d}~X Y nX |dk rÆtd| ƒ‚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
    ZiBNéþÿÿÿr¢  rÓ   r…   rÔ  r   r¡  l        )r¢  ÚKÚMÚGr˜  r   z'%s' is not valid input.z'%s': cannot be negative)rn  rG   r™  rc  r(   )rG  Zsize_inr¤  ZnumZmplierr  r}   r-   r-   r.   Úhuman2bytesc  s&    


 rª  c                 C   s<   | dkrt  ¡ d } | dkp6| d dko6| dd… dk}|S )z$Return True if platform is x86-basedNr  Zx86_64r   r¼  r%   Z86)r)   r*   )Z
uname_archZx86_arch_matchr-   r-   r.   Úis_x86Ž  s    ÿr«  c                 C   s,   t jd d… dk r"t t | ¡¡S t | ¡S )Nr%   )r%   é   )râ   Úversion_infoÚemailZmessage_from_filerÅ   ÚStringIOÚmessage_from_string)rž   r-   r-   r.   r°  ˜  s    r°  c              	   C   s|   t j ddg| dd}tƒ }|j ¡ D ]R}z| d d¡\}}}W n tk
rV   Y q$Y nX | d¡r$| t 	dd|¡¡ q$|S )	Nz
dpkg-queryz--listT)r2   r1   r%   )ÚhiZiiz:.*r™   )
r   r’  r3   rD   r,   rc  r©  r”  r  Úsub)r2   r5   Z	pkgs_instrM   r  rs  r    r-   r-   r.   Úget_installed_packagesž  s    

r³  c               
   C   s¦   d} z.t t| ddƒ}| dd¡ ¡ dkr0W dS W n0 tk
rb } zt d| |¡ W 5 d }~X Y nX tƒ }d|krvdS td	dd}d| ¡ kr’dS tj	 
d
¡r¢dS dS )Nr  T©rË   rþ   r™   zubuntu-corez!Unexpected error loading '%s': %sz
snap_core=z/etc/system-image/channel.iniz/etc/system-image/config.d/F)r  r®   r  r‘   rc  rH   rI   r|  r)   rm   r[  )ZorpathZorinfor}   ry  r9  r-   r-   r.   Úsystem_is_snappy­  s     
 rµ  c                 C   sØ   d }|   ¡ D ]}| d¡r|dd … } q,q|d kr8d S | d¡rF|S | d¡rdd|tdƒd …  S | d¡r†d|tdƒd …  ¡  S | d¡rÐd	|tdƒd …  ¡  }tj |¡r¸|S t|ƒ}|rÌ|d
 S |S d| S )Nzroot=rM  r·  r¤  z/dev/disk/by-label/zUUID=z/dev/disk/by-uuid/z	PARTUUID=z/dev/disk/by-partuuid/r   )r,   r©  rG   r‘   r)   rm   r±   rÊ  )ry  r  r  Z
disks_pathZresultsr-   r-   r.   Úrootdev_from_cmdlineÅ  s.    




ÿr¶  c                 C   sF   dd„ }i }|| ƒD ],}|  dd¡\}}|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.c                 S   s   t j| ddS )NT)Zcomments)Úshlexr,   )rW   r-   r-   r.   Ú_shlex_splitê  s    z(load_shell_content.<locals>._shlex_splitrm  r…   )r,   )r9  Z	add_emptyZ	empty_valr¸  r>   rM   rø   r2  r-   r-   r.   r  ä  s    
r  ç      à?c                 C   s   t | ƒ}d}|t dd„ |D ƒƒ8 }t|ƒdkrBt d||| ¡ g S |dkrZt d||| ¡ || |krhq|t |¡ ||7 }qt d|||¡ |S )Nr   c                 S   s   g | ]}t j |¡r|‘qS r-   r¥  ro  r-   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’  rG   rH   r|   r&  Úsleep)ZflistZmaxwaitZnaplenZlog_preZneedZwaitedr-   r-   r.   Úwait_for_filesø  s:    üü

   ÿr»  c                 C   s&   t | dd}|d  d¡}|d dkS )z1Check whether the given mount point is mounted rwTr•  rÓ   r@   r   Zrwr–  )r  r—  Z
mount_optsr-   r-   r.   Úmount_is_read_write  s    r¼  c                 C   sX   t  d¡sdS ddg}| r:tj | ¡r*dS | d|  g¡ |rN| d| g¡ t   |¡S )zAInvoke udevadm settle with optional exists and timeout parametersZudevadmNZsettlez--exit-if-exists=%sz--timeout=%s)r   rb  r)   rm   r±   rº   )r±   rV  Z
settle_cmdr-   r-   r.   Úudevadm_settle  s    
r½  c              
   C   s„   d}zJt d|  dd}|rLt dt|ƒ¡}|r>t| d¡ƒ}nt d| |¡ W n0 tk
r~ } zt d| |¡ W 5 d	}~X Y nX |S )
zH
    Return the parent pid of a process by parsing /proc/$pid/stat.
    r   z/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  r  rU   r(   r  rH   rI   r¯   )rn  Úppidrl  r  r}   r-   r-   r.   Úget_proc_ppid_linux0  s    ý r¿  c                 C   s&   t   dddt| ƒg¡\}}t| ¡ ƒS )z;
    Return the parent pid of a process by checking ps
    Zpsz-oppid=rŒ  )r   rU   r(   r4   )rn  r¾  r    r-   r-   r.   Úget_proc_ppid_psG  s    rÀ  c                 C   s   t ƒ rt| ƒS t| ƒS dS )z-
    Return the parent pid of a process.
    N)ré   r¿  rÀ  )rn  r-   r-   r.   Úget_proc_ppidO  s    rÁ  r…   ú	Error:
{}c                 C   s&   t | | ¡tjd |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)ÚprintrH  râ   rã   Úexit)rg  Zrcrá  Zsys_exitr-   r-   r.   r•  Y  s    	
r•  c                       s^   e Zd Zd‡ fdd„	Zeedœdd„ƒZdd„ Zd	d
„ Zdd„ Z	dd„ Z
edœdd„Z‡  ZS )ÚVersionrÓ   c                    s   t t| ƒ | ||||¡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È  j  s    zVersion.__new__)r   c                 C   s   | t tt| d¡ƒƒŽ S ©Nr$   )r   r'   r(   r,   )rÉ  r   r-   r-   r.   Úfrom_stro  s    zVersion.from_strc                 C   s   d|   |¡kS )Nr…   )Ú_compare_version©ro   Úotherr-   r-   r.   Ú__gt__s  s    zVersion.__gt__c                 C   s0   | j |j ko.| j|jko.| j|jko.| j|jkS rS   ©rÊ  rË  rÌ  rÍ  rÓ  r-   r-   r.   Ú__eq__v  s    
ÿ
þ
üzVersion.__eq__c                 c   s6   | j | j| j| jfD ]}|dkr,t|ƒV  q q2qdS )z)Iterate over the version (drop sentinels)rÓ   N)rÊ  rË  rÌ  rÍ  rU   )ro   Únr-   r-   r.   Ú__iter__~  s    zVersion.__iter__c                 C   s
   d  | ¡S rÐ  )rJ   rs   r-   r-   r.   Ú__str__†  s    zVersion.__str__r¨   c                 C   sP   | |krdS | j |j krdS | j|jkr,dS | j|jkr<dS | j|jkrLdS dS )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
        r   r…   rÓ   rÖ  rÓ  r-   r-   r.   rÒ  ‰  s    
zVersion._compare_version)rÓ   rÓ   rÓ   rÓ   )r   r€   r   rÈ  ÚclassmethodrU   rÑ  rÕ  r×  rÙ  rÚ  r(   rÒ  Ú__classcell__r-   r-   rÎ  r.   rÆ  h  s   rÆ  rÊ  rË  rÌ  rÍ  )Úextra_messageÚschedule)Ú
deprecatedÚdeprecated_versionrÝ  rÞ  c           	      C   s¦   t tdƒstƒ t_|pd}t| | | t|ƒ ƒ}t |¡}t|j| |j	ƒ}|tjkr¢tj 
|¡ | › d|› d|› d|›  ¡ }t tdƒr˜t |¡ n
t |¡ dS )a  Mark a "thing" as deprecated. Deduplicated deprecations are
    logged.

    @param deprecated: Noun to be deprecated. Write this as the start
        of a sentence, with no period. Version and extra message will
        be appended.
    @param deprecated_version: The version in which the thing was
        deprecated
    @param extra_message: A remedy for the user's problem. A good
        message will be actionable and specific (i.e., don't use a
        generic "Use updated key." if the user used a deprecated key).
        End the string with a period.
    @param schedule: Manually set the deprecation schedule. Defaults to
        5 years. Leave a comment explaining your reason for deviation if
        setting this value.

    Note: uses keyword-only arguments to improve legibility
    Ú_logr™   z is deprecated in z  and scheduled to be removed in z. rß  N)rd  Ú	deprecater’  rá  ÚhashrU   rÆ  rÑ  rÊ  rË  r”  r¿  rH   rß  rI   )	rß  rà  rÝ  rÞ  ÚmessageZdedupr   Zversion_removedZdeprecate_msgr-   r-   r.   râ     s    


ÿ
râ  )rÞ  ©rà  rÝ  rÞ  c                    s   ‡ ‡‡fdd„}|S )a~  Mark a "thing" as deprecated. Deduplicated deprecations are
    logged.

    @param deprecated_version: The version in which the thing was
        deprecated
    @param extra_message: A remedy for the user's problem. A good
        message will be actionable and specific (i.e., don't use a
        generic "Use updated key." if the user used a deprecated key).
        End the string with a period.
    @param schedule: Manually set the deprecation schedule. Defaults to
        5 years. Leave a comment explaining your reason for deviation if
        setting this value.

    Note: uses keyword-only arguments to improve legibility
    c                    s    t  ˆ ¡‡‡‡ ‡fdd„ƒ}|S )Nc                     s    ˆ| |Ž}t ˆ ˆjˆˆd |S )N)rà  rß  rÝ  rÞ  )râ  r   )r‹   rŒ   r5   )rà  rÝ  rœ  rÞ  r-   r.   Ú	decoratorà  s    
üz2deprecate_call.<locals>.wrapper.<locals>.decorator)Ú	functoolsÚwraps)rœ  ræ  rå  )rœ  r.   Úwrapperß  s    zdeprecate_call.<locals>.wrapperr-   )rà  rÝ  rÞ  ré  r-   rå  r.   Údeprecate_callÌ  s    rê  )N)N)rR   )rR   )N)N)N)r˜   N)N)TT)F)N)r   )N)N)N)NN)F)r™   r™   rM  )N)r™   r™   rM  rh  r   )N)N)F)rŠ  )Nr£  NFN)Nr£  NFN)Nr£  NFN)Nr£  NFN)Nr£  NFN)NF)NFT)rÔ  N)NN)NN)N)rú  )NNN)NNNT)F)r;  )r;  r1  F)r‹  rV  )T)NN)rR   r¿   )F)NNF)rŸ  F)N)N)FN)r¹  r™   )NN)r…   rÂ  F)ôra   rÇ   r$  r  r®  rç  rÔ   r'  rÉ   ræ  rÅ   r   r)   Úos.pathrç   rÙ  r   r  r·  rJ  r“  ry   rž   r6  râ   r&  Úbase64r   Úcollectionsr   r   r   r°   r   r   r   r	   Úpathlibr
   Útypingr   r   r   r   r   r   r   r   Zurllibr   Z	cloudinitr   r   r   rä   r   r   r   r   r   r   r   r   Zcloudinit.settingsr   r‘  Z	getLoggerr   rH   Úsepr½   r    rŸ   rÀ   r   r•   r/   r6   r9   rY   r]   rZ   rc   rh   ri   rˆ   r‚   r„   r   r”   r–   r—   r£   r¥   r´   r¸   r¶   rÃ   rÍ   rÐ   rU   rL   rÚ   rà   ÚDEBUGrå   ré   rê   rï   rñ   ró   rõ   rù   rú   rû   r  r  r  rî   r  r  r$  r  rA  ÚcontextmanagerrC  r&  rI  rL  rZ  r`  r³   rQ  ru  rw  r{  rz  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ë  ÚBufferedIOBaserò  rô  rõ  rö  rý  r  r  rû  r  r  r  r  r  r!  r#  r)  r4  r5  r:  r(   r=  r>  rA  r  rC  rF  rK  rO  rR  r8  rU  rY  r[  r^  ra  rc  re  rf  rg  rÑ  rk  rh  rq  rr  rt  rƒ  r…  rˆ  r‹  r  r  r‘  r“  r—  r˜  r  rœ  rm   rž  r¥  rª  r«  r°  r³  rµ  r¶  r  r»  r¼  r½  r¿  rÀ  rÁ  r•  rÆ  râ  rê  r-   r-   r-   r.   Ú<module>   sB  (
(

 ÿ.
2ú
,





	(
=8
P8
	
	!(+,"þ+(3
        ÿ
        ÿ
        ÿ
        ÿ
        ÿ
@

:
	+   ú
^	
"ÿÿ þ

  û÷=%

-N	
W'
  úù.+

0;ûû-ÿ  ÿ