U
    ÉÓg¶i ã                   @   sZ
  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m Z  d dl!m"Z" d dl#m$Z$ d dl%m&Z& d dl'm(Z( d d	l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6 d d
l7m8Z8 d dl9Z9d dl:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZE d dlFmGZG d dlHmIZI e+rÒd dlJmKZK daLe MeN¡ZOejPdiZQdejR ejS ZTdZUdZVdd„ ZWe$ƒ dd„ ƒZXe$ƒ dd„ ƒZYdde5eZe[f eZdœdd„Z\dee5eZe[f e[dœdd„Z]e? ^d ¡e[e[d!œd"d#„ƒZ_d$d%„ Z`G d&d'„ d'ƒZaG d(d)„ d)ebƒZcG d*d+„ d+ebƒZdd,d-„ Zedfd.d/„Zfdgd0d1„Zgdhd2d3„Zhdid5d6„Zidjd7d8„Zjdd9œe/d:œd;d<„Zkd=d>„ Zld?d@„ ZmdAdB„ ZndkdDdE„ZodFdG„ ZpeZeqdHœdIdJ„Zre$ƒ dKdL„ ƒZse$ƒ dMdN„ ƒZte$ƒ dOdP„ ƒZue$ƒ dQdR„ ƒZve$ƒ dSdT„ ƒZwe$ƒ dUdV„ ƒZxdldXdY„ZydmdZd[„Zzdnd\d]„Z{dod^d_„Z|e$ƒ d`da„ ƒZ}dbdc„ Z~e$ƒ ddde„ ƒZdpdfdg„Z€dqdhdi„Zdjdk„ Z‚drdldm„Zƒdse4e2 eqdnœdodp„Z„ejdqdr„ ƒZ…ejdsdt„ ƒZ†dudv„ Z‡dwdx„ Zˆdtd{d|„Z‰dud}d~„ZŠdeqffdd€„Z‹dvd‚dƒ„ZŒdd9œeqd:œd„d…„Zdd9œeqd:œd†d‡„ZŽdwdˆd‰„ZdxdŠd‹„ZdŒd„ Z‘edŽddd‘gƒZ’dyd’d“„Z“dzd•d–„Z”e? ^d—¡e•d:œ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dWd±œe5eZej¢f e3e-e£gdf  e•e[d²œd³d´„Z¤ddWd±œe5eZej¢f e3e-e£gdf  e•eZd²œdµd¶„Z¥e$ƒ d·d¸„ ƒZ¦d¹dº„ Z§e•d:œd»d¼„Z¨dd¾d¿„Z©d‚dÀdÁ„ZªdƒdÂdÃ„Z«e/eZe,f e3eZ e1e3eZ  dÄœdÅdÆ„Z¬e/eZe,f dÇœdÈdÉ„Z­d„eZeZdÊœdËdÌ„Z®dÍdÎ„ Z¯dÏdÐ„ Z°dÑdÒ„ Z±d…dÔdÕ„Z²e? ^dÖ¡eqf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£e•dd÷œdødù„ZÁdúdû„ ZÂdüdý„ ZÃeZe£dþœdÿd „ZÄeZe£dœdd„ZÅeZeZdœdd„ZÆeZeZdœdd„ZÇeZe1eZ dœd	d
„ZÈe? ^d¡dŠdCdddœe•e•dœ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"„ ZÑe$ƒ d#d$„ ƒZÒd%d&„ ZÓdŽeZeZe/eZeZf d(œd)d*„ZÔd+d,„ ZÕd-d.„ ZÖd/d0„ Z×eOdWfd1d2„ZØd3d4„ ZÙd5d6„ ZÚd7d8„ ZÛd9d:„ ZÜdd;d<„ZÝeOdWfd=d>„ZÞeZe•d?œd@dA„ZßdBdC„ ZàddejájPfdDdE„ZâddGdH„ZãdIdJ„ Zäd‘dKdL„ZådMdN„ ZædOdP„ ZçdQdR„ ZèdSdT„ Zéd’dUdV„Zêd“dXdY„ZëdZd[„ Zìd”d\d]„Zíd^eqd_œd`da„Zîee0de,df d:œdbdc„ƒZïdS (•  é    N)Ú	b64decode)ÚdequeÚ
namedtuple)ÚcontextmanagerÚsuppress)ÚENOENT)Ú	lru_cache)ÚPath)Ú
ModuleType)ÚIOÚTYPE_CHECKINGÚAnyÚCallableÚDequeÚDictÚ	GeneratorÚListÚMappingÚOptionalÚSequenceÚUnionÚcast)Úparse)ÚfeaturesÚimporterÚmergersÚnetÚperformanceÚsettingsÚsubpÚ
temp_utilsÚ
type_utilsÚ
url_helperÚversion)Úlogexc)ÚCFG_BUILTIN)ÚPathsÚ_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© r4   r4   ú0/usr/lib/python3/dist-packages/cloudinit/util.pyÚkernel_version_   s    r6   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)r   ÚstdoutÚstrip)Úoutr4   r4   r5   Úget_dpkg_architecturec   s    r<   c               
      sÜ   dddddœ} i ‰ z~t j ddgdd	}|j ¡ D ],}| d
¡\}}}|| kr0| ¡ ˆ | | < q0‡ fdd„|  ¡ D ƒ}t|ƒrŽt 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Úidr2   )ZCodenameZDescriptionzDistributor IDZReleaseÚlsb_releasez--allTr7   ú:c                    s   g | ]}|ˆ kr|‘qS r4   r4   )Ú.0Úk©Údatar4   r5   Ú
<listcomp>~   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UNAVAILABLENr4   ©rA   Úvr4   r4   r5   Ú	<genexpr>‡   s     zlsb_release.<locals>.<genexpr>)r   r9   Ú
splitlinesÚ	partitionr:   ÚvaluesÚlenÚLOGÚwarningÚjoinÚProcessExecutionErrorÚdict)Zfmapr;   ÚlineÚfnamer'   ÚvalÚmissingÚerrr4   rC   r5   r?   n   s,    üþ(r?   úutf-8)ÚblobÚreturnc                 C   s   t | tƒr| S | j|dS ©N)Úencoding)Ú
isinstanceÚstrÚdecode)rY   r\   r4   r4   r5   Údecode_binaryŒ   s    r`   )ÚtextrZ   c                 C   s   t | tƒr| S | j|dS r[   )r]   ÚbytesÚencode)ra   r\   r4   r4   r5   Úencode_text‘   s    rd   zBase64 decoding)rD   rZ   c                 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)r]   rb   Ú	TypeErrorÚtyper   ÚbinasciiÚErrorrC   r4   r4   r5   Úmaybe_b64decode–   s    	
ri   c                 C   sP   | j dd}|  ¡ dkrLt|tƒrL|  ¡ }|r<|jr<|j}nd}| |d¡S |S )NT)r_   ra   rX   Úsurrogateescape)Zget_payloadZget_content_maintyper]   rb   Zget_charsetZinput_codecr_   )ÚpartZcte_payloadÚcharsetr\   r4   r4   r5   Úfully_decoded_payload§   s     ÿ
rm   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_modulero   ÚImportErrorÚpathÚ	recursive)Úselfrr   rs   r4   r4   r5   Ú__init__»   s    zSeLinuxGuard.__init__c                 C   s   | j r| j  ¡ rdS dS d S ©NTF)ro   Úis_selinux_enabled)rt   r4   r4   r5   Ú	__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)©rs   z,restorecon failed on %s,%s maybe badness? %s)ro   rw   r0   rr   ÚlexistsÚrealpathÚlstatZmatchpathconÚstatÚST_MODEÚOSErrorrN   Údebugrs   Z
restoreconrO   )rt   Z	excp_typeZ
excp_valueZexcp_tracebackrr   ZstatsÚer4   r4   r5   Ú__exit__Ì   s0    
ýüzSeLinuxGuard.__exit__N)F)Ú__name__Ú
__module__Ú__qualname__ru   rx   r‚   r4   r4   r4   r5   rn   º   s   
rn   c                   @   s   e Zd ZdS )ÚMountFailedErrorN©rƒ   r„   r…   r4   r4   r4   r5   r†   é   s   r†   c                   @   s   e Zd ZdS )ÚDecompressionErrorNr‡   r4   r4   r4   r5   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)	r0   ÚforkÚ_exitÚ	Exceptionr$   rN   r!   Úobj_namer€   )Zchild_cbÚargsÚkwargsZfidr4   r4   r5   Úfork_cbñ   s"    
ýýr   c                 C   sB   t | tƒr| dkS t}|r&t|ƒ| }t| ƒ ¡  ¡ |kr>dS dS rv   )r]   ÚboolÚTRUE_STRINGSÚlistr^   Úlowerr:   ©rU   ÚaddonsZ	check_setr4   r4   r5   Úis_true  s    
r—   c                 C   sB   t | tƒr| dkS t}|r&t|ƒ| }t| ƒ ¡  ¡ |kr>dS dS )NFT)r]   r‘   ÚFALSE_STRINGSr“   r^   r”   r:   r•   r4   r4   r5   Úis_false  s    
r™   c                 C   s    | sdS t | tƒr| S t| |ƒS )NF)r]   r‘   r—   )rU   r–   r4   r4   r5   Útranslate_bool  s
    
rš   é    c                    s6   t  ¡ ‰ ˆstjtj ‰d ‡ ‡fdd„t| ƒD ƒ¡S )NÚ c                    s   g | ]}ˆ   ˆ¡‘qS r4   )Zchoice)rA   Z_x©ÚrÚselect_fromr4   r5   rE   +  s     zrand_str.<locals>.<listcomp>)ÚrandomZSystemRandomÚstringÚascii_lettersÚdigitsrP   Úrange)ÚstrlenrŸ   r4   r   r5   Ú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newkeyr4   r4   r5   Úrand_dict_key.  s    r¨   ©Úinstance_data_file)rZ   c             
   C   sð   ddl m}m}m}m} zt| ƒ}W n tk
r<   i  Y S X |rätj 	|¡räz||| |ƒ}t
 d|| ¡ W nv |k
rœ } zt
 d| |¡ W 5 d}~X Y nH |k
r®   Y n6 |k
râ } zt
 d||t|ƒ¡ W 5 d}~X Y nX t|i dS )z>Read a yaml config with optional template, and convert to dictr   )ÚJinjaLoadErrorÚJinjaSyntaxParsingExceptionÚNotJinjaErrorÚrender_jinja_payload_from_filez?Applied instance data in '%s' to configuration loaded from '%s'z4Failed to render templated yaml config file '%s'. %sNz:Could not apply Jinja template '%s' to '%s'. Exception: %s©Údefault)Z!cloudinit.handlers.jinja_templater«   r¬   r­   r®   Úload_text_fileÚFileNotFoundErrorr0   rr   ÚexistsrN   r€   rO   ÚreprÚ	load_yaml)rT   rª   r«   r¬   r­   r®   Zconfig_filer   r4   r4   r5   Ú	read_conf8  sB    
ýüýûr¶   c                  G   s   t t| Ž ƒS ©N)ÚsortedÚ
uniq_merge)Úlistsr4   r4   r5   Úuniq_merge_sortedl  s    r»   c                  G   sF   g }| D ]4}t |tƒr2| ¡  d¡}dd„ |D ƒ}| |¡ qt|ƒS )NrF   c                 S   s   g | ]}|r|‘qS r4   r4   )rA   Úar4   r4   r5   rE   ~  s      zuniq_merge.<locals>.<listcomp>)r]   r^   r:   r3   ÚextendÚ	uniq_list)rº   Zcombined_listÚa_listr4   r4   r5   r¹   x  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Úappendr:   )ÚfnrB   rH   Zremovalsr4   r4   r5   Úclean_filenameƒ  s    rÆ   Tc                 C   sÌ   z†t  t| ƒ¡n}t d dd|¡R}|rNt| ¡ ƒW  5 Q R £ W  5 Q R £ W S | ¡ W  5 Q R £ W  5 Q R £ W S W 5 Q R X 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ÚBytesIOrd   ÚgzipZGzipFiler`   ÚreadrŒ   rˆ   r^   )rD   Úquietr_   ÚbufZghr   r4   r4   r5   Ú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)r3   r:   rM   r”   )Zug_pairZ	ug_partedÚuÚgr4   r4   r5   Úextract_usergroup   s    rÑ   )Úroot_dirrZ   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+   éÿÿÿÿ)	rR   Úglobr0   rr   rP   ÚisfileÚbasenamer:   Úfind)rÒ   ÚentriesrT   Úmodnamer4   r4   r5   Úget_modules_from_dir°  s    
rÛ   c                   C   s   dt  ¡ kS )a  deprecated: prefer Distro object's `is_linux` property

    Multiple sources of truth is bad, and already know whether we are
    working with Linux from the Distro class. Using Distro offers greater code
    reusablity, cleaner code, and easier maintenance.
    ZLinux©ÚplatformÚsystemr4   r4   r4   r5   Úis_Linux¼  s    rß   c                   C   s$   dt  ¡ krdS t  ¡ dkr dS dS )NZBSDTZ	DragonFlyFrÜ   r4   r4   r4   r5   Úis_BSDÇ  s
    rà   c                   C   s   t ƒ d dkS )NÚvariantÚfreebsd©Úsystem_infor4   r4   r4   r5   Ú
is_FreeBSDÐ  s    rå   c                   C   s   t ƒ d dkS )Nrá   Ú	dragonflyrã   r4   r4   r4   r5   Úis_DragonFlyBSDÕ  s    rç   c                   C   s   t ƒ d dkS )Nrá   Únetbsdrã   r4   r4   r4   r5   Ú	is_NetBSDÚ  s    ré   c                   C   s   t ƒ d dkS )Nrá   Úopenbsdrã   r4   r4   r4   r5   Ú
is_OpenBSDß  s    rë   Fc                 C   s   || kr|S t | | ƒS r·   )rš   ©ÚyobjÚkeyr°   r4   r4   r5   Úget_cfg_option_boolä  s    rï   c                 C   s*   || kr|S | | }t |tƒs&t|ƒ}|S r·   )r]   r^   )rí   rî   r°   rU   r4   r4   r5   Úget_cfg_option_strê  s    
rð   c                 C   s   t t| ||dƒS )Nr¯   )r/   rð   rì   r4   r4   r5   Ú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\.]+)Únamer=   z linuxr   zred hat enterpriseÚredhatr#   )ÚIDÚ
VERSION_IDÚVERSION_CODENAME)	r0   rr   r³   r±   ÚreÚmatchÚ	groupdictr”   rK   )Zrelease_fileZredhat_releaseZredhat_regexrø   Úgroupr4   r4   r5   Ú_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VERSIONr=   Úrhelró   )rœ   rœ   rœ   r‰   zPUnable to determine distribution, template expansion may have unexpected results)r0   rr   r³   Úload_shell_contentr±   rû   ÚgetrÝ   Úmachiner÷   rø   rù   rà   rÞ   r”   r2   rN   rO   ÚdistrŒ   )	Zdistro_nameZdistro_versionZflavorZ
os_releaseZos_release_rhelrø   r  ÚfoundÚentryr4   r4   r5   Ú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aoscZarchZ
azurelinuxZ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_distr4   r4   r5   Ú_get_variant]  s     	r  c                  C   s<   t   ¡ t  ¡ t  ¡ t  ¡ tt  ¡ ƒtƒ dœ} t| ƒ| d< | S )N)rÝ   rÞ   r2   Úpythonr1   r  rá   )rÝ   rÞ   r2   Zpython_versionr“   r1   r	  r  )r  r4   r4   r5   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 r4   r4   rG   r4   r4   r5   rE   ·  s     z'get_cfg_option_list.<locals>.<listcomp>)r]   r“   r^   )rí   rî   r°   rU   Zcvalr4   r4   r5   Ú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.ú/)r]   r^   r3   )rí   Zkeypr°   ZcurÚtokr4   r4   r5   Úget_cfg_by_pathÀ  s    


r  c                 C   s    t | |ƒ\}}t||ƒ ||fS r·   )Úget_output_cfgÚredirect_output)ÚcfgÚmodeÚoutfmtÚerrfmtr4   r4   r5   Ú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sn|d	krŠd
}|dkr~d}t
||ƒ}n:|dkr¸tj|dtj|d}	ttt |	jƒ}ntd|  ƒ‚|rÜt | ¡ | ¡ ¡ || krt d|| ¡ t | ¡ | ¡ ¡ d S |r¸t d||¡ | 	dd¡\}}|dksD|d	krbd
}|dkrVd}t
||ƒ}n<|dkr’tj|dtj|d}	ttt |	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)r0   ÚumaskÚgrpÚgetgrnamÚgr_gidÚKeyErrorÚsetgid)Zgroup_idr4   r4   r5   Ú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—   r0   Úenvironr  rN   r€   Úsysr9   Ústderrr3   ÚopenÚ
subprocessÚPopenÚPIPEr   r   r   r*  re   Údup2Úfileno)
r  r  Zo_outZo_errr"  r  ÚargZowithZnew_fpÚprocr4   r4   r5   r  ë  sd    
ü


ür  )ÚsourcesrZ   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)r6  ÚreverseZ
merged_cfgr  Zmergers_to_applyZmergerr4   r4   r5   Úmergemanydict?  s    *

r:  c              	   c   s.   t  ¡ }zt  | ¡ | V  W 5 t  |¡ X d S r·   )r0   ÚgetcwdÚchdir)ZndirZcurrr4   r4   r5   r<  w  s
    

r<  c              	   c   s&   t  | ¡}z
|V  W 5 t  |¡ X d S r·   )r0   r  )Zn_mskÚoldr4   r4   r5   r    s    

r  c                 C   s   dj | |d|dS )Nz{0:{fill}{align}{size}}ú^)ÚfillZalignÚsize)Úformat)ra   r?  Zmax_lenr4   r4   r5   ÚcenterŠ  s       ÿrB  c                 C   s   t  d| ¡ t | ¡ d S )NzRecursively deleting %s)rN   r€   ÚshutilZrmtree©rr   r4   r4   r5   Údel_dir  s    rE  rœ   é   c           	   
   C   s~   z<t |||d\}}}}|| d< || d< || d< || d< W dS  tjk
rx } z|jtjkrfW Y ¢
dS ‚ W 5 d}~X Y nX dS )	z®
    returns boolean indicating success or failure (presense of files)
    if files are present, populates 'fill' dictionary with 'user-data' and
    'meta-data' entries
    )ÚbaseÚextÚtimeoutú	user-dataúvendor-dataú	meta-dataznetwork-configTFN)Úread_seededr"   ÚUrlErrorÚcodeZ	NOT_FOUND)	r?  rG  rH  rI  ÚmdÚudÚvdÚnetworkr   r4   r4   r5   Úread_optional_seed•  s    
rT  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/sslrD   Zsslc                 S   s    g | ]}|rt j |¡r|‘qS r4   )r0   rr   Úisdir)rA   Údr4   r4   r5   rE   ¶  s       z%fetch_ssl_details.<locals>.<listcomp>zcert.pemzkey.pemÚ	cert_fileÚkey_file)r0   rr   rP   Zget_ipath_curÚ	get_cpathr¹   rÖ   )ÚpathsZssl_detailsZssl_cert_pathsrW  rV  rX  r4   r4   r5   Ú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‰   )rS   ÚcolrW   z. {err})rW   )r`   rN   r€   rM   ÚyamlZ	safe_loadr]   re   r!   r   Z	YAMLErrorÚ
ValueErrorÚhasattrÚgetattrrA  rS   ÚcolumnrO   )rY   r°   ZallowedZloadedZ	convertedr   ÚmsgZmarkr4   r4   r5   rµ   É  sJ    ü


ÿÿ
  ÿÿrµ   é
   c              
   C   sX  |   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 }d }tj|||d
}d }	| 	¡ rÂ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   rJ  rK  rL  rÔ   r  rœ   z%s%s%s)rI  Ú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µ   ÚcontentsrN  rN   r€   )rG  rH  rI  rd  Zud_urlZvd_urlZmd_urlrS  Zmd_resprP  Zud_resprQ  rR  Zvd_respr   r4   r4   r5   rM  ñ  sP      ÿ  ÿ  ÿ


rM  c                   s¼   t t ˆ ¡dd}dd„ |D ƒ}‡ fdd„|D ƒ}g }|D ]x}tj ˆ |¡}z| t||d¡ W q: tk
r‚   t 	d|¡ Y q: t
k
r° } zt 	d||¡ W 5 d	}~X Y q:X q:t|ƒS )
zRead configuration directory.T©r9  c                 S   s   g | ]}|  d ¡r|‘qS )z.cfg)Úendswith©rA   Úfr4   r4   r5   rE   #  s     
 zread_conf_d.<locals>.<listcomp>c                    s&   g | ]}t j t j ˆ |¡¡r|‘qS r4   )r0   rr   rÖ   rP   rj  ©Úconfdr4   r5   rE   &  s      r©   ú1REDACTED config part %s, insufficient permissionsúError accessing file %s: [%s]N)r¸   r0   Úlistdirrr   rP   rÄ   r¶   ÚPermissionErrorrN   rO   r   r:  )rm  rª   ZconfsÚcfgsrÅ   rr   r   r4   rl  r5   Úread_conf_d  s*    þÿ ÿ"rs  c             
   C   s  t ƒ }i }zt| |d}W nN tk
r:   t d| ¡ Y n: tk
rh } zt 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©   rn  ro  Nrœ   Zconf_dz8Config file %s contains 'conf_d' with non-string type %sz.d)r   r¶   rq  rN   rO   r   rÄ   r]   r^   re   r!   r   r:   r0   rr   rU  rs  Ú
appendleftr:  )Zcfgfilerª   rr  r  r   rm  Z	confd_cfgr4   r4   r5   Úread_conf_with_confd=  s:     ÿ 

ÿÿ

ru  c                 C   s   t t| dƒS )N©Úcmdline)rµ   Úread_cc_from_cmdlinerv  r4   r4   r5   Úread_conf_from_cmdlinel  s    ry  c           	      C   s®   | d krt ƒ } d| › } d}d}t|ƒ}t|ƒ}t| ƒ}g }|  |¡}|dkr¤|  ||| ¡}|dk rj|}| t | || |…  ¡ ¡ dd¡¡ |  ||| ¡}qFd |¡S )Nr#  z cc:Zend_ccr   z\nÚ
)	Úget_cmdlinerM   rØ   rÄ   r   ZunquoteÚlstriprÂ   rP   )	rw  Z	tag_beginZtag_endZbegin_lZend_lZclenÚtokensZbeginÚendr4   r4   r5   rx  q  s,    


 ÿÿrx  c                 C   s2   |   d¡}|dks"| |d  dkr&| S |  dd¡S )Nrz  r   r‰   úz
)rØ   rÂ   )rg  Úposr4   r4   r5   Ú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ð   r3   rØ   Úget_hostnamerƒ  r‚  )r  Zcloudr†  r…  r„  rƒ  r4   r4   r5   Ú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±   rJ   rØ   r:   r3   rM   ÚIOError)rƒ  Úfilenamer„  rS   ZhashposÚtoksr4   r4   r5   Úget_fqdn_from_hostsÍ  s$    

r  zResolving URLc              
   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   re  rƒ  Ú_DNS_REDIRECT_IPÚsetÚsocketZgetaddrinfoZSOCK_STREAMZAI_CANONNAMErÄ   ÚaddZgaierrorÚerrorrN   r€   r   r^  r   Zis_ip_addressZnetlocr:   )ÚurlZ
parsed_urlrò   ZbadipsZbadnamesZ
badresultsZinameÚresultZ_famZ_stypeZ_protoZcnameZsockaddrZaddrr4   r4   r5   Úis_resolvableõ  sD    
     ÿ

r˜  c                  C   s   t  ¡ } | S r·   )r“  Zgethostname)rƒ  r4   r4   r5   r‡  )  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ipr4   r4   r5   rš  .  s    rš  c                 C   s   t | ƒS )z5determine if this url is resolvable (existing or ip).)r˜  )r–  r4   r4   r5   Úis_resolvable_url5  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')rN   r€   r›  rŒ   )Z
candidatesZcandr4   r4   r5   Úsearch_for_mirror:  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 r4   ©r0   rr   r³   )rA   Úpr4   r4   r5   rE   U  s   þz*find_devs_with_freebsd.<locals>.<listcomp>z/dev/msdosfs/z/dev/iso9660/ú	TYPE=vfatúTYPE=iso9660)rÕ   Ú
startswithr|  )ÚcriteriaÚoformatÚtagÚno_cacherr   ÚdevlistÚlabelr4   r4   r5   Úfind_devs_with_freebsdM  s    

þ
rª  c                 C   sÌ   g }d }d }d}| r<|   d¡r(|  d¡}|   d¡r<|  d¡}tjdddgdgd}	|	j ¡ D ]j}
|sh|r‚tjd	|
gdd
gd\}}|r”d| |kr”q\|dkr¦d|kr¦q\|dkr¸d|kr¸q\| d|
 ¡ q\|S )Nrœ   rž  zTYPE=Úsysctlú-núhw.disknamesr   ©ZrcsZ	mscdlabelr‰   z
label "%s"Úiso9660zISO filesystemÚvfatú/dev/)r£  r|  r   r9   r3   rÄ   )r¤  r¥  r¦  r§  rr   r¨  r©  Ú_typeZmscdlabel_outr;   Údevr'   r4   r4   r5   Úfind_devs_with_netbsda  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®  rF   r@   zfd0:rÔ   r¼   ÚcdÚic                 S   s   g | ]}d | ‘qS ©r±  r4   ©rA   r¶  r4   r4   r5   rE   ‰  s     z*find_devs_with_openbsd.<locals>.<listcomp>)r   r9   Úrstripr3   ri  rÄ   r£  )r¤  r¥  r¦  r§  rr   r;   r¨  r  r4   r4   r5   Ú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 )rP  Zvn©r£  r¸  r4   r4   r5   rE     s   
 
þz/find_devs_with_dragonflybsd.<locals>.<listcomp>Trh  r¢  c                 S   s   g | ]}|  d ¡r|‘qS ©)rµ  Zacdr»  r¸  r4   r4   r5   rE   —  s     
 )zLABEL=CONFIG-2r¡  c                 S   s   g | ]}|  d ¡s|‘qS r¼  r»  r¸  r4   r4   r5   rE   ™  s     
 zUnexpected criteria: %sc                 S   s   g | ]}d | ‘qS r·  r4   r¸  r4   r4   r5   rE   œ  s     )r   r¸   r9   r3   rN   r€   )r¤  r¥  r¦  r§  rr   r;   r¨  r4   r4   r5   Ú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   rQ   Úerrnor   rJ   r:   )r¤  r¥  r¦  r§  rr   Z
blk_id_cmdÚoptionsÚcmdr;   Ú_errr   rÙ   rS   r4   r4   r5   Ú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Â   )r8   r_   r@   ZDEVNAME)r“   r½   r   r9   rJ   rK   r  )	ZdevsZdisable_cacherÃ  r;   ÚretrS   r³  r'   rD   r4   r4   r5   r¾  ß  s    

r¾  c                 C   s(   g }| D ]}||krqq|  |¡ q|S r·   )rÄ   )Zin_listZout_listr¶  r4   r4   r5   r¾   þ  s    r¾   ©Úread_cbrÌ   )rT   rÉ  rÌ   rZ   c                C   s¦   t jddd}t d| |¡ |f t ¡ R}z(t| dƒ}t|||d W 5 Q R X W n tk
rn   |sj‚ Y nX | 	¡ }W 5 Q R X W 5 Q R X t dt
|ƒ| |j¡ |S )Nrœ   Úskip)Zlog_modezReading from %s (quiet=%s)rÇ   )Úchunk_cbzReading %s bytes from %s%s)r   ZTimedrN   r€   rÈ   rÉ   r.  Úpipe_in_outr²   ÚgetvaluerM   Úoutput)rT   rÉ  rÌ   ZtimerZofhZifhrg  r4   r4   r5   Úload_binary_file  s    
rÏ  c                C   s   t t| ||dƒS )NrÈ  )r`   rÏ  )rT   rÉ  rÌ   r4   r4   r5   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Œ   rN   rO   r:   )rg  rw  r   r4   r4   r5   Ú_get_cmdline&  s    
rÒ  c                   C   s   dt jkrt jd S tƒ S )NZDEBUG_PROC_CMDLINE)r0   r+  rÒ  r4   r4   r4   r5   r{  9  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±   r:   rŒ  r   )Z	fips_procrg  r4   r4   r5   Úfips_enabled@  s    
rÓ  é   c                 C   sN   d}|   |¡}t|ƒdkrqBq| |¡ |t|ƒ7 }|r||ƒ q| ¡  |S r™  )rË   rM   ÚwriteÚflush)Zin_fhZout_fhZ
chunk_sizerË  Zbytes_pipedrD   r4   r4   r5   rÌ  L  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)rN   r€   r0   Úchown)rT   ÚuidÚgidr4   r4   r5   Ú	chownbyid[  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Ú  )rT   Úuserrú   rØ  rÙ  r   r4   r4   r5   Úchownbynamec  s     rÞ  )r  r  rZ   c                 C   sj  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|ƒD ]v\}}|süqî| ¡ }d}	|D ]8}
| |
¡rd|
|t|
ƒd…  ¡ f }d}	 qFq|	s\dd| ¡ f }|||< qî|S )aŸ  Get the output configuration for a given mode.

    The output config is a dictionary that specifies how to deal with stdout
    and stderr for the cloud-init modules. It is a (frustratingly) flexible
    format that can take multiple forms such as:
        output: { all: "| tee -a /var/log/cloud-init-output.log" }
    or
        output:
            init:
                output: "> /var/log/cloud-init.out"
                error: "> /var/log/cloud-init.err"
            config: "tee -a /var/log/cloud-config.log"
            final:
                - ">> /var/log/cloud-final.out"
                - "/var/log/cloud-final.err"

    Mode can be one of the configuration stages. If you pass a
    non-existent mode, it will assume the "all" mode configuration if
    defined.

    Stderr can be specified as &1 to indicate that it should
    be the same as stdout.

    If a file is specified with no redirection, it will default to
    appending to the file.

    If not overridden, output is provided in
    '/etc/cloud/config/cloud.cfg.d/05_logging.cfg' and defaults to:
        {"all": "| tee -a /var/log/cloud-init-output.log"}

    :param cfg: The base configuration that may or may not contain the
        'output' configuration dictionary
    :param mode: The mode to get the output configuration for.
    :return: A list of two strings (or Nones), the first for stdout for the
        specified mode and the second for stderr.
    NrÎ  Úallr   r‰   r•  z&1r%  r$  r(  Fz%s %sT)	r]   r^   r“   rM   rR   Ú	enumerater|  r£  r:   )r  r  rÇ  ZoutcfgZmodecfgZswlistr¶  rž   rU   r  Úsr4   r4   r5   r  p  sL    '






r  )r  c           
      C   sô   g }g }| rt | tƒs|S |  d¡}|r2| |¡ t| dƒD ]h}|sFq<t d|¡}|sXq<| d¡}| ¡ }t	|ƒdkr‚| |¡ q<ddg|dd… kr<| |d ¡ q<|D ]8}t
 
|› d	¡D ]"}	tj |	¡r¾|	|kr¾| |	¡ q¾qªtt|| ƒƒS )
a=  Return a list of log file paths from the configuration dictionary.

    Obtains the paths from the 'def_log_file' and 'output' configuration
    defined in the base configuration.

    If not provided in base configuration, 'def_log_file' is specified in
    'cloudinit/settings.py' and defaults to:
        /var/log/cloud-init.log
    If not overridden, output is provided in
    '/etc/cloud/config/cloud.cfg.d/05_logging.cfg' and defaults to:
        {"all": "| tee -a /var/log/cloud-init-output.log"}

    @param cfg: The cloud-init merged configuration dictionary.
    Zdef_log_fileNz (?P<type>\||>+)\s*(?P<target>.*)Útargetr‰   Úteez-ar,   Ú*)r]   rR   r  rÄ   r  r÷   rø   rú   r3   rM   rÕ   r0   rr   rÖ   r“   r’  )
r  ZlogsZrotated_logsZdefault_logZfmtrø   râ  ÚpartsZlogfileZrotated_logfiler4   r4   r5   Úget_config_logfilesÒ  s0    


ræ  )ÚroutinerZ   c                 C   s<   t  |¡}| t| ƒ¡ | ¡ }|d k	r4|d|… S |S d S r™  )ÚhashlibÚnewÚupdaterd   Z	hexdigest)rY   rç  ZmlenZhasherZdigestr4   r4   r5   Ú	hash_blob   s    
rë  c                 C   s0   zt  | ¡rW dS W n tk
r*   Y dS X d S rv   )rÛ  rÜ  r   ©rò   r4   r4   r5   Úis_user  s
    

rí  c                 C   s0   zt  | ¡rW dS W n tk
r*   Y dS X d S rv   )r  r  r   rì  r4   r4   r5   Úis_group  s
    

rî  c                 C   s   t  d| |¡ t | |¡ d S )NzRenaming %s to %s)rN   r€   r0   Úrename©ÚsrcÚdestr4   r4   r5   rï    s    rï  éí  c                 C   s   | D ]}t ||ƒ qd S r·   )Ú
ensure_dir)Údirlistr  rV  r4   r4   r5   Úensure_dirs!  s    rö  zLoading jsonc                 C   sH   t  t| ƒ¡}t|t|ƒƒsDd dd„ |D ƒ¡}td|t|ƒf ƒ‚|S )Nú, c                 S   s   g | ]}t |ƒ‘qS r4   )r^   )rA   Útr4   r4   r5   rE   *  s     zload_json.<locals>.<listcomp>z((%s) root types expected, got %s instead)ÚjsonÚloadsr`   r]   r-   rP   re   rf   )ra   Z
root_typesZdecodedZexpected_typesr4   r4   r5   Ú	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)r0   rr   ÚdirnamerU  Úget_non_exist_parent_dir)rr   Zp_pathr4   r4   r5   rý  2  s    rý  c           
   	   C   sÄ   t j | ¡s¶t| ƒ}t j | ¡}d}|dkr0d}t||d t  | ¡ W 5 Q R X t| |ƒ |sd|rÀt|||ƒ t	|  
|¡d dd … ƒ}t	|ƒ}|jD ]}	| |	¡}t|||ƒ q˜n
t| |ƒ d S )NTr  Fry   r‰   )r0   rr   rU  rý  rü  rn   ÚmakedirsÚchmodrÞ  r	   r3   rå  Zjoinpath)
rr   r  rÝ  rú   Znon_existed_parent_dirZdir_nameZselinux_recursiveZsub_relative_dirZsub_pathrk   r4   r4   r5   rô  C  s"    


rô  c              	   c   s(   z
| V  W 5 | r"d| g}t   |¡ X d S )NÚumount)r   )r   Z
umount_cmdr4   r4   r5   Ú	unmounter^  s
    
r  c               	   C   s"  i } zôt j d¡r$tdƒ ¡ }d}nt d¡}|j ¡ }d}t d¡}|D ]œ}|dkr~| 	¡ }t
|ƒdkrlqJ|\}}}	}
}}nL| |¡}|d ksJt
| ¡ ƒdk r¢qJ| d¡}| d¡}| d	¡}	| d¡}
| d
d¡}|	||
dœ| |< qJt d| |¡ W n$ ttfk
r   ttdƒ Y nX | S )Nz/proc/mountsr5  Úmountz*^(/dev/[\S]+) on (/.*) \((.+), .+, (.+)\)$é   r  r‰   r,   r‹  z\040r#  )ÚfstypeÚ
mountpointÚoptszFetched %s mounts from %szFailed fetching mount points)r0   rr   r³   r±   rJ   r   r9   r÷   Úcompiler3   rM   ÚsearchÚgroupsrú   rÂ   rN   r€   rŒ  r   r$   )ÚmountedZ
mount_locsÚmethodr;   ZmountreZmplineZwordsr³  Zmpr  r  Z_freqZ_passnoÚmr4   r4   r5   Úmountsh  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  )r]   r^   r“   r-   re   rA  rf   rß   rà   rà  r  r    Ztempdirr0   rr   r{   r½   rÄ   r   rŒ  r   rN   r€   rP   r†   ri  r  )r  ÚcallbackrD   ZmtypeZupdate_env_for_mountZ	log_errorZmtypesÚindexr
  Ztmpdr   r  Zfailure_reasonZmountcmdÚexcrÇ  r4   r4   r5   Úmount_cb’  s€    

ÿÿ



ù	ÿÿ



r  c                   C   s
   t  t¡S r·   )Úobj_copyÚdeepcopyr%   r4   r4   r4   r5   Úget_builtin_cfgð  s    r  c                 C   s   t  d| ¡ tj | ¡S )NzTesting if a link exists for %s)rN   r€   r0   rr   ÚislinkrD  r4   r4   r5   Ú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§   )
rN   r€   r0   rr   rz   rP   rü  r¦   ÚsymlinkrÂ   )ÚsourceÚlinkZforceZtmp_linkr4   r4   r5   Úsym_linkú  s    r  c                 C   s4   t  d| ¡ zt | ¡ W n tk
r.   Y nX d S )NzAttempting to remove %s)rN   r€   r0   Úunlinkr²   rD  r4   r4   r5   Údel_file  s
    r  c                 C   s   t  d| |¡ t | |¡ d S )NzCopying %s to %s)rN   r€   rC  Úcopyrð  r4   r4   r5   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tsr4   r4   r5   Útime_rfc2822  s
    
r$  c                     s¼   ddl ‰ ddl‰ G ‡ fdd„dˆ jƒ} ˆ  ˆ j d¡¡}ˆ  ¡ }ˆ  | ¡|_dt	ƒ sXdndg}ˆ j
d	 |Ž }| ƒ }| |ˆ  
t|ƒ¡ˆ  |¡ˆ  |¡dd¡d
kr°|j|jd  S tdƒ‚dS )aÛ  Use sysctl(3) via ctypes to find kern.boottime

    kern.boottime is of type struct timeval. Here we create a
    private class to easier unpack it.
    Use sysctl(3) (or sysctl(2) on OpenBSD) because sysctlbyname(3) does not
    exist on OpenBSD. That complicates retrieval on NetBSD, which #defines
    KERN_BOOTTIME as 83 instead of 21.
    21 on NetBSD is KERN_OBOOTTIME, the kern.boottime up until NetBSD 5.0

    @return boottime: float to be compatible with linux
    r   Nc                       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_r4   ©Úctypesr4   r5   Útimeval+  s   r)  Úcr‰   é   éS   r,   rÔ   g    €„.Az/Unable to retrieve kern.boottime on this system)r(  Zctypes.utilZ	StructureZCDLLÚutilZfind_libraryZc_size_tZsizeofÚvalueré   Zc_intr«  rM   Zbyrefr%  r&  ÚRuntimeError)r)  Zlibcr@  Z
mib_valuesZmibrÍ   r4   r'  r5   Úboottime  s0    þúøÿr0  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)r0   rr   r³   r±   r3   r^   r!  r0  rŒ   r$   rN   )Z
uptime_strr  rg  r4   r4   r5   ÚuptimeF  s    r1  c                 C   s   t | |dd d d S )Nr&  )Úomoder  ©Ú
write_file)rr   Úcontentr4   r4   r5   Úappend_fileY  s    r6  é¤  )Úpreserve_mode)r  r8  rZ   c                C   s   t | dd||d d S )Nrœ   r&  )r5  r2  r  r8  r3  )rr   r  r8  r4   r4   r5   Úensure_file]  s        ÿr9  c              	   C   s*   z
t | ƒW S  ttfk
r$   Y d S X d S r·   )r/   r^  re   )Zpossible_intr4   r4   r5   Úsafe_inte  s    
r:  c              	   C   s4   t |ƒ}| r0|r0t| ƒ t | |¡ W 5 Q R X d S r·   )r:  rn   r0   rÿ  )rr   r  Z	real_moder4   r4   r5   rÿ  l  s    
rÿ  )Úgrp_namerZ   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   rN   r€   )r;  rÙ  r4   r4   r5   Úget_group_ids  s    r<  )rr   rZ   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.
    )r}   ÚS_IMODEr0   Úst_moderD  r4   r4   r5   Úget_permissions  s    r?  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.
    )r0   r}   rÛ  ÚgetpwuidÚst_uidZpw_name©rr   Ústr4   r4   r5   Ú	get_ownerŒ  s    
rD  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.
    )r0   r}   r  ÚgetgrgidÚst_gidÚgr_namerB  r4   r4   r5   Ú	get_group–  s    
rH  )ÚusernamerZ   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Ä   rG  rÛ  rÜ  Zpw_gidrE  )rI  r	  rú   rÙ  r4   r4   r5   Úget_user_groups   s    
rJ  zWriting filer'  )Úensure_dir_existsrÝ  rú   )r8  rK  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ú   Úbrb   Z
charactersz%oz%rzWriting to %s - %s: [%s] %s %srD  N)r?  r   rô  r0   rr   rü  r”   rd   r`   re   rN   r€   rM   rn   r.  rÕ  rÖ  rÿ  )r  r5  r  r2  r8  rK  rÝ  rú   Z
write_typeZmode_rZfhr4   r4   r5   r4  °  s:    ú
r4  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)r0   rp  rr   rP   rU  rE  r  )rü  ZnodeZnode_fullpathr4   r4   r5   Údelete_dir_contentsî  s
    
rM  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_stringr^   Útitler$  )Zcomment_charrG  Zci_verÚheaderr4   r4   r5   Úmake_headerü  s
    rQ  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)ú'ú\rR  rR  r   z'%s'rR  z%s%s
r#  r‰   zCUnable to shellify type '%s'. Expected list, string, tuple. Got: %szShellified %s commands.)r]   r-   r“   re   r!   r   rÄ   r^   rÂ   rP   rN   r€   )ZcmdlistZ
add_headerr5  ZescapedZ	cmds_maderŽ   Zfixedrk  r4   r4   r5   Úshellify	  s<    ÿÿ


ÿÿrT  c                 C   sB   |r|   |¡r| t|ƒd … } |r>|  |¡r>| d t|ƒ … } | S r·   )r£  rM   ri  )rS   ÚprefixÚsuffixr4   r4   r5   Ústrip_prefix_suffix-	  s
    rW  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   ÚwhichrQ   )rÃ  r4   r4   r5   Ú_cmd_exits_zero5	  s    rY  c                   C   s   t dddgƒS )Nzsystemd-detect-virtz--quietz--container©rY  r4   r4   r4   r5   Ú_is_container_systemd?	  s    r[  c                   C   s
   t dgƒS )Nzlxc-is-containerrZ  r4   r4   r4   r5   Ú_is_container_old_lxcC	  s    r\  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   rX  r:   )rÃ  r;   r'   r4   r4   r5   Ú_is_container_freebsdG	  s    
r]  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)r[  r]  r\  Úget_proc_envrŒ  r   r0   rr   rU  r±   rJ   r£  r:   r3   )ZchecksÚhelperZpid1envÚlinesrS   Z_keyrU   r4   r4   r5   rÑ  Q	  s6    ý

rÑ  c                   C   s   t j d¡S )z2Check to see if we are running in a lxd container.z/dev/lxd/sockrŸ  r4   r4   r4   r5   Úis_lxd	  s    ra  rÂ   )r\   ÚerrorsrZ   c              	   C   sŒ   t j dt| ƒd¡}zt|ƒ}W n ttfk
r<   i  Y S X i }d\}}| ||¡}| |¡D ]&}|sjq`| |d¡\}	}
|	r`|
||	< q`|S )zû
    Return the environment in a dict that a given process id was started with.

    @param encoding: decoding will be done with .decode(encoding, errors) and
    text will be returned.
    @param errors: passed through .decode(encoding, errors).
    z/procr+  )rÐ  ú=r‰   )	r0   rr   rP   r^   rÏ  rŒ  r   r_   r3   )Úpidr\   rb  rÅ   rg  ÚenvZnullZequalr  rò   rU   r4   r4   r5   r^  „	  s    


r^  c              	   C   sP   i }|   ¡ D ]>}z|  dd¡\}}W n tk
r@   |}d}Y nX |||< q|S )Nrc  r‰   T)r3   r^  )ZkvstringrÇ  r  rî   rU   r4   r4   r5   Úkeyval_str_to_dict£	  s    

rf  c                 C   s&   |   d¡r| dd … } tj d|  ¡S )Nr±  rF  z/sys/class/block/%s/partition)r£  r0   rr   rÖ   )r  r4   r4   r5   Úis_partition¯	  s    
rg  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.)r]   r“   r^   rÄ   r-   rM   r/  )Zversion_fmtZpkgsZpkglistÚpkgr4   r4   r5   Úexpand_package_list¶	  s     



ri  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 r4   r4   ©rA   r   r4   r4   r5   rE   Õ	  s      z$parse_mount_info.<locals>.<listcomp>r  Nrc  z$Line %d has two few columns (%d): %sr‰   r  c                 S   s   g | ]}|r|‘qS r4   r4   rj  r4   r4   r5   rE   ê	  s      r   ú-z,Did not find column named '-' in line %d: %sr,   z/Too few columns after '-' column in line %d: %srF  )r3   rà  rM   r€   Úminr  r^  Ú
IndexError)rr   Zmountinfo_linesÚlogÚget_mnt_optsÚpath_elementsÚdevpthÚfs_typeÚmatch_mount_pointÚmatch_mount_point_elementsr¶  rS   rå  Úmount_pointÚmount_point_elementsÚxÚmount_optionsr4   r4   r5   Úparse_mount_infoÑ	  sp       ÿ
ÿþ  ÿ
  ÿ


ry  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±   rJ   r3   )rr   rS   rq  ru  rr  r4   r4   r5   Ú
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gptZgptidr  ZufsidrF  ZglabelZstatusz-srz  z)Unexpected input in find_freebsd_part: %s)r3   rM   r   r^   rN   rO   )ZfsZsplittedZtarget_labelrk   rÄ  ÚlabelsrÁ   r4   r4   r5   Úfind_freebsd_part(
  s    
r}  c                 C   sH   d }|  d¡D ]4}|  ¡ }t|ƒdkrtj |d |  ¡r|} qDq|S )Nrz  r,   r‰   )r3   rM   r0   rr   r³   )rr   Úmnt_listÚ
path_foundrS   rÁ   r4   r4   r5   Úget_path_dev_freebsd;
  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  z-pr   r‰   r®  r±  )r   rM   r€  r3   r}  )rr   r—  rW   r~  r  rÇ  Z
label_partr4   r4   r5   Úget_freebsd_devpthE
  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 r4   r4   rj  r4   r4   r5   rE   `
  s      zparse_mount.<locals>.<listcomp>r  Nrq  r  c                 S   s   g | ]}|r|‘qS r4   r4   rj  r4   r4   r5   rE   k
  s      r   rf   rÂ  rF   r÷  zNfound line in mount -> devpth: %s, mount_point: %s, fs_type: %s, options: '%s'z^(/dev/.+)[sp]([0-9])$)ZzfsZnfs)r   r3   rJ   r÷   r  rú   rM   rl  rP   r:   rN   r€   rå   r  )rr   ro  ZmountoutputrÄ  Zregexrp  rq  ru  rs  rt  rS   r  rv  rw  rr  rx  ZdevmZmatch_devpthr4   r4   r5   Úparse_mountS
  sl    ÿ


ÿþ


ÿúr‚  c                 C   sT   dt  ¡  }t j |¡r2t|ƒ ¡ }t| |||ƒS t j d¡rFt| ƒS t| |ƒS d S )Nz/proc/%s/mountinforz  )	r0   Úgetpidrr   r³   r±   rJ   ry  r{  r‚  )rr   rn  ro  Zmountinfo_pathr`  r4   r4   r5   Úget_mount_infoª
  s    r„  )ÚoptrZ   c                 C   s    t | dd^ }}|| d¡kS )NT©ro  rF   ©r„  r3   )rr   r…  r'   Zmnt_optsr4   r4   r5   Úhas_mount_optÑ
  s    rˆ  c                 C   s(   |   dd¡}t|ƒdkr|S | d fS d S )Nr+   r‰   )ÚrsplitrM   )ZdottedrŽ  r4   r4   r5   Úexpand_dotted_devnameÖ
  s    rŠ  c              	   C   s”   |d krg }|d krg }g }i }|| D ]H}zt | | | dd||< W q( tk
rn   ||krj| |¡ Y q(X q(t|ƒrtdjd |¡dƒ‚|S )NF©rÌ   zMissing required files: {files}rF   )Úfiles)rÏ  r²   rÄ   rM   r^  rA  rP   )rG  ZrequiredZoptionalZdelimrV   rÇ  rk  r4   r4   r5   Úpathprefix2dictÞ
  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±   rJ   r3   r^  r/   )	ZmeminfoÚrawÚmpliersZkmaprÇ  rS   rî   r.  Zunitr4   r4   r5   Úread_meminfoö
  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ÚGÚTr   z'%s' is not valid input.z'%s': cannot be negative)ri  rM   Úfloatr^  r/   )r@  Zsize_inr“  ZnumZmplierr  r   r4   r4   r5   Úhuman2bytes  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)r0   r1   )Z
uname_archZx86_arch_matchr4   r4   r5   Úis_x869  s    ÿrœ  c                 C   s
   t  | ¡S r·   )ÚemailÚmessage_from_string)r¡   r4   r4   r5   rž  C  s    rž  c               	   C   sz   t j ddgdd} tƒ }| j ¡ D ]R}z| d d¡\}}}W n tk
rT   Y q"Y nX | d¡r"| t 	dd|¡¡ q"|S )	Nz
dpkg-queryz--listTr7   r,   )ÚhiZiiz:.*rœ   )
r   r’  r9   rJ   r3   r^  r£  r”  r÷   Úsub)r;   Z	pkgs_instrS   Ústaterh  r'   r4   r4   r5   Úget_installed_packagesG  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ü   Tr‹  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”   r^  rN   rO   r{  r0   rr   rU  )ZorpathZorinfor   rw  r5  r4   r4   r5   Úsystem_is_snappyV  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=rF  r±  rž  z/dev/disk/by-label/zUUID=z/dev/disk/by-uuid/z	PARTUUID=z/dev/disk/by-partuuid/r   )r3   r£  rM   r”   r0   rr   r³   rÅ  )rw  r  r  Z
disks_pathZresultsr4   r4   r5   Úrootdev_from_cmdlinen  s.    




ÿr¤  c                 C   sF   dd„ }i }|| ƒD ],}|  dd¡\}}|s0|}|s8|r|||< q|S )zéGiven shell like syntax (key=value\nkey2=value2\n) 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)Úshlexr3   )rY   r4   r4   r5   Ú_shlex_split“  s    z(load_shell_content.<locals>._shlex_splitrc  r‰   )r3   )r5  Z	add_emptyZ	empty_valr¦  rD   rS   rî   r.  r4   r4   r5   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 r4   rŸ  rj  r4   r4   r5   rE   ¥  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’  rM   rN   r€   r!  Úsleep)ZflistZmaxwaitZnaplenZlog_preZneedZwaitedr4   r4   r5   Ú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Ô   rF   r   Zrwr‡  )ru  r—  Z
mount_optsr4   r4   r5   Ú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   rX  r0   rr   r³   r½   )r³   rI  Z
settle_cmdr4   r4   r5   Úudevadm_settleÇ  s    
r«  r&   )rZ  rZ   c              
   C   s’   dg i}zt  t|  d¡dd¡}W nX tk
rJ   t d|  d¡¡ Y nD t jk
r| } zt dt	j
|¡ W 5 d }~X Y nX d|krŽg |d< |S )NZscopeszhotplug.enabledFr‹  zFile not found: %sz>Ignoring contents of %s because it is not decodable. Error: %s)rù  rú  r±   rY  r²   rN   r€   ZJSONDecodeErrorrO   r   ZHOTPLUG_ENABLED_FILE)rZ  r5  r   r4   r4   r5   Úread_hotplug_enabled_fileÙ  s     ÿýr¬  c                   c   s
   dV  dS )ztContext manager that does nothing.

    Note: In python-3.7+, this can be substituted by contextlib.nullcontext
    Nr4   r4   r4   r4   r5   Únullcontextí  s    r­  )rX   )rX   )N)N)N)r›   N)N)TT)F)N)r   )N)N)N)NN)F)rœ   rœ   rF  )N)rœ   rœ   rF  rc  )N)N)F)r‰  )Nr  NFN)Nr  NFN)Nr  NFN)Nr  NFN)Nr  NFN)NF)rÔ  N)NN)NN)N)ró  )NNN)NNNT)F)r7  )r7  r'  F)rŠ  rN  )T)NN)rX   rÂ   )F)rŽ  F)N)FN)r§  rœ   )NN)ðrg   Ú
contextlibr  r  r  rÕ   r  rÊ   rè  rÈ   rù  Zloggingr0   Úos.pathrÝ   rÛ  r    r÷   r¥  rC  r“  r}   r¡   r/  r,  r!  Úbase64r   Úcollectionsr   r   r   r   rÁ  r   Ú	functoolsr   Úpathlibr	   Útypesr
   Útypingr   r   r   r   r   r   r   r   r   r   r   r   r   Zurllibr   r]  Z	cloudinitr   r   r   r   r   r   r   r    r!   r"   r#   Zcloudinit.log.log_utilr$   Zcloudinit.settingsr%   Zcloudinit.helpersr&   r‘  Z	getLoggerrƒ   rN   ÚseprÀ   r£   r¢   rÃ   r’   r˜   r6   r<   r?   r^   rb   r`   rd   Ztimedri   rm   rn   rŒ   r†   rˆ   r   r—   r™   rš   r¦   r¨   r¶   r»   r¹   rÆ   rÎ   rÑ   rR   rÛ   rß   rà   rå   rç   ré   rë   rï   rð   rñ   rû   r	  r  rä   r  r  r  r  r:  r<  r  rB  rE  rT  r[  rµ   rM  rs  ru  ry  rx  r  r‚  rˆ  r  r‘   r˜  r‡  rš  r›  rœ  rª  r´  rº  r½  rÅ  r¾  r¾   ÚPathLiker/   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$  r0  r1  r6  r9  r:  rÿ  r<  r?  rD  rH  rJ  r4  rM  rQ  rT  rW  rY  r[  r\  r]  rÑ  ra  r^  rf  rg  ri  ry  r{  r}  r€  r  r‚  r„  rˆ  rŠ  rr   r  r”  r›  rœ  rž  r¢  r£  r¤  r  r©  rª  r«  r¬  r­  r4   r4   r4   r5   Ú<module>   s:  <4
 ÿ


/
4






	(
=:
T8
	
!(, /#þ+(3        ÿ
        ÿ
        ÿ
        ÿ
        ÿ
@üûüû	

 
þb.
	-   ú
^
*ÿÿ þ


  û÷ù=%

-  ÿ 
þN	
W'+
