o
    w7e                     @   s  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mZmZmZ d dlm  m  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mZ d dlmZ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) d dl*m+Z+ d dl,m-Z- dZ.dgdgddgddgddgdgdgg dg ddgddgdZ/e0e1Z2e3dZ4g dZ5ej6ej7 d  Z8G d!d" d"ej9e j:d#Z;d$e<d%e=fd&d'Z>d$e<fd(d)Z?dej@fd*d+ZAd,d- ZBd.e<d/ee; fd0d1ZC	3d8d4d5ZDd6d7 ZEdS )9    N)StringIO)AnyMappingMutableMappingOptionalType)importer)log)netpersistencessh_utilsubp
temp_utils
type_utilsutil)LinuxNetworking
Networking)hosts)+ALLOW_EC2_MIRRORS_ON_NON_AWS_INSTANCE_TYPES)
activatorsdhcpeninetwork_state	renderers)parse_net_config_data)Rendererallalpinearchdebianubuntufreebsd	dragonflygentoocosnetbsdopenbsd)	almalinuxamazoncentos
cloudlinux	eurolinuxfedoramarinermiraclelinuxopenmandrivaphotonrhelrocky	virtuozzo)opensusezopensuse-leapzopensuse-microoszopensuse-tumbleweedsle_hpcz	sle-microsles	openeulerOpenCloudOS	TencentOS)r   r   r   r!   r#   r%   r&   redhatsuser7   r8   z#^[a-z][a-z]-(?:[a-z]+-)+[0-9][a-z]$)chronyzsystemd-timesyncdntpntpdate-c                   @   sT  e Zd ZU dZdZdZdZdZdZdZ	dZ
d	gZi Zeeeeef f ed
< dZeZee ed< ddddZejZdZdZdZeed< ejej ej!gZ"dd Z#de$ddfddZ%e&j'dd Z(dd Z)e*de+ee,j-  fdd Z.de/fd!d"Z0d#e/fd$d%Z1d&d' Z2dd(d)Z3dd*d+Z4dd,d-Z5e6d.d/ Z7e&j'dd0d1Z8e&j'd2d3 Z9d4d5 Z:dd6d7Z;dd8d9Z<dd;d<Z=dd=d>Z>d?d@ Z?dde@fdAdBZAe&j'ddCdDZBe&j'dEdF ZCdGdH ZDdIdJ ZEe&j'ddKdLZFe&j'dMdN ZGe&j'dOdP ZHdQdR ZIdSdT ZJe6dUdV ZKdWdX ZLdYdZ ZMe*d[d\ ZNd]d^ ZOd_d` ZPdadb ZQdcdd ZRdedf ZSdgdh ZTdidj ZUdkdl ZVddmdnZWdoeXdpe@fdqdrZYdsdt ZZddudvZ[ddxdyZ\ddzd{Z]dd|d}Z^d~d Z_e`dddZae`ddded	edefddZbdedededefddZcdefddZdddeXdedefddZee6dedededededeXfddZfdS )Distrozpython3-pipz/usr/libz
/etc/hostsz/etc/doas.confz"/etc/sudoers.d/90-cloud-init-usersz/etc/hostnamez/usr/share/zoneinfoz	root:rootservicerenderer_configsNnetworking_clsz-Hz-Pz-r)haltpoweroffreboot   Fz/etc/resolv.confosfamilyc                 C   s:   || _ || _|| _|  | _tjtjtjg| _	t
j| _d S N)_paths_cfgnamerC   
networkingr   IscDhclientDhcpcdUdhcpcdhcp_client_priorityiproute2Iproute2net_ops)selfrL   cfgpaths rX   </usr/lib/python3/dist-packages/cloudinit/distros/__init__.py__init__v   s   
zDistro.__init__ci_pkl_versionreturnc                 C   s$   d| j vs	| jj s|  | _dS dS )z)Perform deserialization fixes for Distro.rM   N)__dict__rM   rC   )rU   r[   rX   rX   rY   	_unpickle   s   zDistro._unpicklec                 C      t  rI   NotImplementedError)rU   pkglistrX   rX   rY   install_packages      zDistro.install_packagesc                 C   s   t d| j )=Deprecated. Remove if/when arch and gentoo support renderers.zhLegacy function '_write_network' was called in distro '%s'.
_write_network_config needs implementation.
)ra   rL   )rU   settingsrX   rX   rY   _write_network   s
   zDistro._write_networkc                 C   s6   t | jdd}ztj|dW S  tjy   Y dS w )z=Return the configured network activator for this environment.)networkr   Npriority)r   get_cfg_by_pathrK   r   select_activatorNoActivatorException)rU   rj   rX   rX   rY   network_activator   s   zDistro.network_activatorc                 C   sD   t | jdd }tj|d\}}td|| || j|d}|S )N)rh   r   ri   z-Selected renderer '%s' from priority list: %s)config)	r   rk   rK   r   selectLOGdebugrB   get)rU   rj   rL   
render_clsrendererrX   rX   rY   _get_renderer   s   zDistro._get_rendererru   c                 C   s   | | d S rI   )render_network_state)rU   r   ru   rX   rX   rY   _write_network_state      zDistro._write_network_statec                 C   s4   t j| jt|}t j|std||f |S )Nz(Invalid timezone %s, no file found at %s)ospathjointz_zone_dirstrisfileIOError)rU   tztz_filerX   rX   rY   _find_tz_file   s   
zDistro._find_tz_filec                 C   s   | j ||S rI   )rK   rs   )rU   opt_namedefaultrX   rX   rY   
get_option   ry   zDistro.get_optionc                 C   s   || j |< d S rI   )rK   )rU   r   valuerX   rX   rY   
set_option   ry   zDistro.set_optionc                 C   s(   |  ||}| || j | | d S rI   )_select_hostname_write_hostnamehostname_conf_fn_apply_hostname)rU   hostnamefqdnwriteable_hostnamerX   rX   rY   set_hostname   s   zDistro.set_hostnamec                   C   s   t  S )z?Wrapper to report whether this distro uses systemd or sysvinit.)uses_systemdrX   rX   rX   rY   r      s   zDistro.uses_systemdc                 C   r_   rI   r`   )rU   commandargspkgsrX   rX   rY   package_command   rd   zDistro.package_commandc                 C   r_   rI   r`   rU   rX   rX   rY   update_package_sources   rd   zDistro.update_package_sourcesc                 C   s   t  d }|dv rdS |S )N   )i386i486i586i686r   )rz   uname)rU   r   rX   rX   rY   get_primary_arch   s   zDistro.get_primary_archc                 C   s"   |  dg }|s|  }t||S )Npackage_mirrors)r   r   _get_arch_package_mirror_info)rU   r   mirror_inforX   rX   rY   r      s   
z$Distro._get_arch_package_mirror_infoc                 C   s   |  |}t||dS )N)data_sourcer   )r   _get_package_mirror_info)rU   r   r   	arch_inforX   rX   rY   get_package_mirror_info   s   
zDistro.get_package_mirror_infoTc                 C   s   |  |}|r| |S dS )re   F)rg   _bring_up_interfaces)rU   rf   bring_up	dev_namesrX   rX   rY   apply_network   s   

zDistro.apply_networkc                 C   sL   | j }td| dd| dg}t|}tj||dd}| j||dS )re   ziapply_network_config is not currently implemented for distribution '%s'.  Attempting to use apply_network
z-# Converted from network_config for distro %sz4# Implementation of _write_network_config is needed.T)headerrender_hwaddressr   )		__class__rq   warningr|   r   r   r   network_state_to_enir   )rU   	netconfigr   distror   nscontentsrX   rX   rY   "_apply_network_from_network_config   s   
z)Distro._apply_network_from_network_configc                 C   s   t  S rI   )r
   generate_fallback_configr   rX   rX   rY   r   
  s   zDistro.generate_fallback_configc                 C   s   z|   }W n ty   | j||d Y S w t||d}| || |r=td | j}|s6td dS |	| dS td dS )a  Apply the network config.

        If bring_up is True, attempt to bring up the passed in devices. If
        devices is None, attempt to bring up devices returned by
        _write_network_config.

        Returns True if any devices failed to come up, otherwise False.
        r   )ru   z/Bringing up newly configured network interfacesz>No network activator found, not bringing up network interfacesTz3Not bringing up newly configured network interfacesF)
rv   ra   r   r   rx   rq   rr   rn   r   bring_up_all_interfaces)rU   r   r   ru   r   rn   rX   rX   rY   apply_network_config  s*   



zDistro.apply_network_configc                 C   r_   rI   r`   )rU   localeout_fnrX   rX   rY   apply_locale2  rd   zDistro.apply_localec                 C   r_   rI   r`   )rU   r   rX   rX   rY   set_timezone6  rd   zDistro.set_timezonec                 C   s   dS )Nz	127.0.0.1rX   r   rX   rX   rY   _get_localhost_ip:  s   zDistro._get_localhost_ipc                 C   r_   rI   r`   r   rX   rX   rY   
get_locale=  s   zDistro.get_localec                 C   r_   rI   r`   )rU   filenamer   rX   rX   rY   _read_hostname@  rd   zDistro._read_hostnamec                 C   r_   rI   r`   )rU   r   r   rX   rX   rY   r   D  rd   zDistro._write_hostnamec                 C   r_   rI   r`   r   rX   rX   rY   _read_system_hostnameH  rd   zDistro._read_system_hostnamec                 C   sF   t d| z
td|g W d S  tjy"   tt d| Y d S w )Nz2Non-persistently setting the system hostname to %sr   z;Failed to non-persistently adjust the system hostname to %s)rq   rr   r   ProcessExecutionErrorr   logexc)rU   r   rX   rX   rY   r   L  s   
zDistro._apply_hostnamec                 C   s&   t | jd| jr|r|S |s|S |S )Nprefer_fqdn_over_hostname)r   get_cfg_option_boolrK   prefer_fqdn)rU   r   r   rX   rX   rY   r   ]  s   
zDistro._select_hostnamec                 C   s6   g }| D ]}|t vrtd||t |  q|S )Nz&No distributions found for osfamily {})
OSFAMILIES
ValueErrorformatextend)family_listdistrosfamilyrX   rX   rY   expand_osfamilyk  s   zDistro.expand_osfamilyc           
   
   C   s  |}|  ||}|rtj|r| |}nd }|  \}}g }|r&||kr+|| |r5||kr:||kr:|| |rK|rK||krKtd|| d S t	dd |D }t
d|t| |D ]}	z| ||	 W q_ tyz   ttd||	 Y q_w ||v r| | d S d S )Nz6%s differs from %s, assuming user maintained hostname.c                 S   s   g | ]}|r|qS rX   rX   ).0frX   rX   rY   
<listcomp>      z*Distro.update_hostname.<locals>.<listcomp>z/Attempting to update hostname to %s in %s filesz!Failed to write hostname %s to %s)r   rz   r{   existsr   r   appendrq   infosetrr   lenr   r   r   r   r   )
rU   r   r   prev_hostname_fnapplying_hostnameprev_hostnamesys_fnsys_hostnameupdate_filesfnrX   rX   rY   update_hostnamev  sH   

zDistro.update_hostnamec                 C   sz  d}t j| jrtt| j}ntd}tjdd}| 	 }|
|}d}|s5|||| d}ndd}|D ]*}d }	g }
t|dkrI|d }	t|dkrU|dd  }
|	d urc|	|krc||
v rcd}q9|rt|}|||g || |D ] }t|dkr|||d  qxt|dkr|j|g|R   qx|rt }|r|d	|  |d	|  tj| j| d
d d S d S )N addedbaseFTrG   r      z%s
i  mode)rz   r{   r   hosts_fnr   	HostsConfr   	load_filemake_headerr   	get_entry	add_entryr   listr   del_entriesr   write
write_filegetvalue)rU   r   r   r   ehlocal_ip	prev_infoneed_changeentry
entry_fqdnentry_aliasesnew_entriesr   rX   rX   rY   update_etc_hosts  sP   


zDistro.update_etc_hostsc                 C   s   | j stt| _ | j S )z7Allow distro to determine the preferred ntp client list)_preferred_ntp_clientsr   PREFERRED_NTP_CLIENTSr   rX   rX   rY   preferred_ntp_clients  s   
zDistro.preferred_ntp_clientsc                 C   s   t )re   r`   )rU   device_namerX   rX   rY   _bring_up_interface  s   zDistro._bring_up_interfacec                 C   s0   d}|D ]}|  |s|d7 }q|dkrdS dS )re   r   rG   TF)r   )rU   device_names	am_faileddrX   rX   rY   r     s   
zDistro._bring_up_interfacesc                 C   s
   |  dS )Ndefault_user)r   r   rX   rX   rY   get_default_user  s   
zDistro.get_default_userc                 K   sv  t |rtd| dS d|v r|d}nd}d|g}d|g}t  r/|d |d ddd	d
ddddddd
}dddd}dg}|d}	|	rt|	t	rV|	
d}	t|	trgt jd| dddd dd |	D }	d|	|d< |d }
|
r|	|
 |r|	r|	D ]}t |s| | td!|| qd"| v rt	|d" |d"< t| D ]D\}}||v r|rt|t	r||| |g ||v r||| d#g q||| |g q||v r|r|||  |||  q|d$s|d%r
|d& |d& n
|d' |d' td(| z
tj||d) W dS  ty: } z	t td*| |d}~ww )+z
        Add a user to the system using standard GNU tools

        This should be overriden on distros where useradd is not desirable or
        not available.
        z!User %s already exists, skipping.Ncreate_groupsTuseradd--extrausersz	--commentz--homez--gidz--uidz--groupsz
--passwordz--shellz--expiredatez
--inactivez--selinux-user)
gecoshomedirprimary_groupuidgroupspasswdshell
expiredateinactiveselinux_userz--no-user-groupz--systemz--no-log-init)no_user_groupsystemno_log_initr  r  ,z	The user z) has a 'groups' config value of type dict22.3z=Use a comma-delimited string or array instead: group1,group2.
deprecateddeprecated_versionextra_messagec                 S   s   g | ]}|  qS rX   strip)r   grX   rX   rY   r   =  r   z#Distro.add_user.<locals>.<listcomp>r
  z created group '%s' for user '%s'r  REDACTEDno_create_homer  z-Mz-mzAdding user %s	logstringzFailed to create user %s)r   is_userrq   r   popsystem_is_snappyr   rs   
isinstancer~   splitdict	deprecater|   is_groupcreate_grouprr   keyssorteditemsr   r   	Exceptionr   )rU   rL   kwargsr  useradd_cmdlog_useradd_cmduseradd_optsuseradd_flagsredact_optsr  r
  groupkeyvalerX   rX   rY   add_user  s   
















zDistro.add_userc              
   K   s   | d}| dd}g d}|r|d || td| ztj||dd\}}td	|| t|}| d
d}	W |	S  tyU }
 z	ttd| |
d}
~
ww )zD
        Add a snappy user to the system using snappy tools
        snapuserknownF)snapzcreate-userz--sudoerz--jsonz--knownzAdding snap user %sT)r!  capturez snap create-user returned: %s:%susernameNzFailed to create snap user %s)	rs   r   rq   rr   r   r   	load_jsonr.  r   )rU   rL   r/  r:  r;  create_user_cmdouterrjobjr>  r8  rX   rX   rY   add_snap_usero  s(   




zDistro.add_snap_userc                 K   s  d|v r| j |fi |S | j|fi | d|v r&|d r&| ||d  d|v r8|d r8| j||d dd |ddrC| | d|v rS|d rS| ||d  d|v rv|d rd| ||d  n|d d	u rvtjd
| dddd d|v r|d }t	|t
r|g}nt	|trt| }|durt	|tttfstdt| g }nt|pg }tt|| d|v r|dg }|std||d  dS |d }tj}|d|}|d|}tjt|||d dS )a  
        Creates or partially updates the ``name`` user in the system.

        This defers the actual user creation to ``self.add_user`` or
        ``self.add_snap_user``, and most of the keys in ``kwargs`` will be
        processed there if and only if the user does not already exist.

        Once the existence of the ``name`` user has been ensured, this method
        then processes these keys (for both just-created and pre-existing
        users):

        * ``plain_text_passwd``
        * ``hashed_passwd``
        * ``lock_passwd``
        * ``doas``
        * ``sudo``
        * ``ssh_authorized_keys``
        * ``ssh_redirect_user``
        r:  plain_text_passwdhashed_passwdT)hashedlock_passwddoassudoFzThe value of 'false' in user z's 'sudo' configr  zUse 'null' instead.r  ssh_authorized_keysNzZInvalid type '%s' detected for 'ssh_authorized_keys', expected list, string, dict, or set.ssh_redirect_usercloud_public_ssh_keysz^Unable to disable SSH logins for %s given ssh_redirect_user: %s. No cloud public-keys present.z$USERz$DISABLE_USER)options)rD  r9  
set_passwdrs   rH  write_doas_ruleswrite_sudo_rulesr   r(  r%  r~   r'  r   valuestupler   rq   r   typer   setup_user_keysDISABLE_USER_OPTSreplace)rU   rL   r/  r+  
cloud_keysredirect_userdisable_optionrX   rX   rY   create_user  sj   




zDistro.create_userc              
   C   s   dd|gdd|gf}zt dd |D }W n ty/ } ztd|dd	 |D f |d
}~ww zt| W d
S  tyM } z	ttd| |d
}~ww )zL
        Lock the password of a user, i.e., disable password logins
        r  z-lusermodz--lockc                 s   s"    | ]}t |d  r|V  qdS )r   N)r   which)r   toolrX   rX   rY   	<genexpr>       z%Distro.lock_passwd.<locals>.<genexpr>zBUnable to lock user account '%s'. No tools available.   Tried: %s.c                 S   s   g | ]}|d  qS )r   rX   r   crX   rX   rY   r     r   z&Distro.lock_passwd.<locals>.<listcomp>Nz&Failed to disable password for user %s)nextStopIterationRuntimeErrorr   r.  r   r   rq   )rU   rL   
lock_toolscmdr8  rX   rX   rY   rH    s(   zDistro.lock_passwdc              
   C   sB   zt  dd|g W d S  ty  } z	ttd| |d }~ww )Nr  z--expirezFailed to set 'expire' for %s)r   r.  r   r   rq   )rU   userr8  rX   rX   rY   expire_passwd  s   zDistro.expire_passwdc              
   C   sf   d||f }dg}|r| d ztj||d| d W dS  ty2 } z	ttd| |d }~ww )Nz%s:%schpasswd-ezchpasswd for %sr   zFailed to set password for %sT)r   r   r.  r   r   rq   )rU   rh  r  rG  pass_stringrg  r8  rX   rX   rY   rO    s   
zDistro.set_passwdplist_inrG  c                 C   s<   d dd |D d }dg|rdgng  }t|| d S )Nr   c                 s   s"    | ]\}}d  ||gV  qdS ):Nr|   )r   rL   passwordrX   rX   rY   r_    r`  z"Distro.chpasswd.<locals>.<genexpr>rj  rk  )r|   r   )rU   rm  rG  payloadrg  rX   rX   rY   rj    s   zDistro.chpasswdc                 C   st   d}t d|| t||}|r3t d|d |d|kr(t d dS t d|d dS t d	 dS )
Nz^(?:permit|deny)(?:\s+(?:nolog|nopass|persist|keepenv|setenv \{[^}]+\})+)*\s+([a-zA-Z0-9_]+)+(?:\s+as\s+[a-zA-Z0-9_]+)*(?:\s+cmd\s+[^\s]+(?:\s+args\s+[^\s]+(?:\s*[^\s]+)*)*)*\s*$z3Checking if user '%s' is referenced in doas rule %rz!User '%s' referenced in doas rulerG   z'Correct user is referenced in doas ruleTz.Incorrect user '%s' is referenced in doas ruleFz/doas rule does not appear to reference any user)rq   rr   researchr5  )rU   rh  rulerule_patternvalid_matchrX   rX   rY   is_doas_rule_valid!  s(   	


zDistro.is_doas_rule_validc           
   
   C   s"  |s| j }|D ]}| ||sd||f }t|  d S qdd| g}|D ]	}|d|  q&d|}|d7 }tj|sit	
 |g}zt	j|d|dd W d S  tyh }	 z	t	td| |	d }	~	ww |t	|vrz	t	|| W d S  ty }	 z	t	td	| |	d }	~	ww d S )
NzHInvalid doas rule %r for user '%s', not writing any doas rules for user!r   z# cloud-init User rules for %s%sr      r   zFailed to write doas file %sz Failed to append to doas file %s)doas_fnrw  rq   errorr   r|   rz   r{   r   r   r   r   r   r   r   append_file)
rU   rh  rules	doas_filert  msglinescontentr   r8  rX   rX   rY   rP  A  sH   

zDistro.write_doas_rules/etc/sudoersc              
   C   s6  d}d}t j|rt|}d}d}| D ]'}| }td|}|s&q|	d }|s0qt j
|}||kr>d} nq|sz<|s^ddtjddd	| dg}	d
|	}t||d ndtjddd	| dg}	d
|	}t|| td|| W n ty }
 z	ttd| |
d }
~
ww t|d d S )Nr   FTz^[#|@]includedir\s+(.*)$rG   z?# See sudoers(5) for more information on "#include" directives:r   r   z#includedir %sr   ry  zAdded '#includedir %s' to %szFailed to write %si  )rz   r{   r   r   r   
splitlinesr  rr  rs  r5  abspathr   r|   r   r|  rq   rr   r   r   
ensure_dir)rU   r{   	sudo_basesudoers_contentsbase_existsfound_includelineinclude_matchincluded_dirr  r8  rX   rX   rY   ensure_sudo_dire  sX   




zDistro.ensure_sudo_dirc           
   
   C   sH  |s| j }dd| g}t|ttfr!|D ]}|d||f  qnt|tr0|d||f  nd}t|t| d	|}|d7 }| 
tj| tj|s|t |g}zt|d	|d W d S  ty{ }	 z	ttd| |	d }	~	ww |t|vrz	t|| W d S  ty }	 z	ttd| |	d }	~	ww d S )	Nr   z# User rules for %sz%s %sz1Can not create sudoers rule addition with type %rr   ry  zFailed to write sudoers file %sz#Failed to append to sudoers file %s)ci_sudoers_fnr%  r   rS  r   r~   	TypeErrorr   obj_namer|   r  rz   r{   dirnamer   r   r   r   r   r   rq   r   r|  )
rU   rh  r}  	sudo_filer  rt  r  r  r   r8  rX   rX   rY   rQ    sL   

zDistro.write_sudo_rulesc                 C   s   d|g}t  r|d |sg }t |rtd| nzt| td| W n ty:   t 	td| Y nw t
|dkrd|D ]"}t |sRtd|| qCtdd	d
||g td|| qCd S d S )Ngroupaddr  z(Skipping creation of existing group '%s'zCreated new group %szFailed to create group %sr   zCUnable to add group member '%s' to group '%s'; user does not exist.r\  z-az-GzAdded user '%s' to group '%s')r   r$  r   r)  rq   r   r   r   r.  r   r   r"  )rU   rL   membersgroup_add_cmdmemberrX   rX   rY   r*    s4   



zDistro.create_groupc             
   C   sl   d| j | g}z|dkrdt| }W n ty' } ztd|f |d }~ww ||g }|r4|| |S )Nshutdownnowz+%dz?power_state[delay] must be 'now' or '+m' (minutes). found '%s'.)shutdown_options_mapintr   r  r   )rU   r   delaymessager   r8  r   rX   rX   rY   shutdown_command  s&   

zDistro.shutdown_commandc                 C   s2   | j }|  sd|v r|dg}tj|d|dS dS )zX
        Reload systemd startup daemon.
        May raise ProcessExecutionError
        	systemctlzdaemon-reloadTr=  rcsN)init_cmdr   r   )clsr  r  rg  rX   rX   rY   reload_init  s
   zDistro.reload_init)r  action
extra_argsc             	   G   s   | j }|  sd|v r*dg}d|gd|gd|gd|gd|gd|gd|gd	|gd
}n|dg|dg|dg|dg|dg|dg|dg|d	gd
}t|t||  }tj|d|dS )z
        Perform the requested action on a service. This handles the common
        'systemctl' and 'service' cases and may be overridden in subclasses
        as necessary.
        May raise ProcessExecutionError
        r  stopstartenabledisablerestartzreload-or-restartzreload-or-try-restartstatus)r  r  r  r  r  reloadz
try-reloadr  Tr  )r  r   r   r   )r  r  rA   r  r  r  cmdsrg  rX   rX   rY   manage_service  s.   

zDistro.manage_servicelayoutmodelvariantrN  c                 C   s(   |   rtdd||||g d S t )N	localectlzset-x11-keymap)r   r   ra   )rU   r  r  r  rN  rX   rX   rY   
set_keymap  s   zDistro.set_keymapc                 C   s.   t jdd}t|ds|S tj| jddS )NT)	needs_exenoexecz
cloud-initclouddir)r   get_tmp_ancestorr   has_mount_optrz   r{   r|   usr_lib_exec)rU   tmp_dirrX   rX   rY   get_tmp_exec_path.  s   zDistro.get_tmp_exec_pathr   r   rh  cwdc              	   K   s>   |rd| dnd}t j dd|d|d d| gfi |S )	a`  
        Perform a command as the requested user. Behaves like subp()

        Note: We pass `PATH` to the user env by using `env`. This could be
        probably simplified after bionic EOL by using
        `su --whitelist-environment=PATH ...`, more info on:
        https://lore.kernel.org/all/20180815110445.4qefy5zx5gfgbqly@ws.net.home/T/
        zcd z && r   sur?   z-czenv PATH=$PATH  )r   r|   )rU   r   rh  r  r/  	directoryrX   rX   rY   do_as4  s   	zDistro.do_asr{   
lease_filepid_file	interfaceconfig_filec              	   C   s,   | ddd|d|ddg	|rd||g S |g S )Nz-1z-vz-lfz-pfz-sfz	/bin/truez-cfrX   )r{   r  r  r  r  rX   rX   rY   build_dhclient_cmdI  s   	

zDistro.build_dhclient_cmdrI   )NN)T)F)r  )r   )g__name__
__module____qualname__pip_package_namer  r   rz  r  r   r}   default_ownerr  rB   r   r~   r   r   __annotations__r   r   rC   r   r   r  rR   rS   rT   _ci_pkl_versionr   resolve_conf_fnr   rN   rO   rP   rQ   rZ   r  r^   abcabstractmethodrc   rg   propertyr   r   NetworkActivatorrn   r   rv   rx   r   r   r   r   staticmethodr   r   r   r   r   r   r   r   r   boolr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r9  rD  r[  rH  ri  rO  r   rj  rw  rP  r  rQ  r*  r  classmethodr  r  r  r  r  r  rX   rX   rX   rY   r@   ]   s   
 










%




A+

ta


 
$
1
) 
$r@   )	metaclassurltransformationsc                 C   s   zt j| }W n
 ty   Y dS w |j}|du rdS |D ]}||}|du r,|   S q|}|jdur;d||j}t j|j|dS )a  
    Apply transformations to a URL's hostname, return transformed URL.

    This is a separate function because unwrapping and rewrapping only the
    hostname portion of a URL is complex.

    :param url:
        The URL to operate on.
    :param transformations:
        A list of ``(str) -> Optional[str]`` functions, which will be applied
        in order to the hostname portion of the URL.  If any function
        (regardless of ordering) returns None, ``url`` will be returned without
        any modification.

    :return:
        A string whose value is ``url`` with the hostname ``transformations``
        applied, or ``None`` if ``url`` is unparseable.
    Nz{}:{})netloc)	urllibparseurlsplitr   r   portr   
urlunsplit_replace)r  r  partsnew_hostnametransformation
new_netlocrX   rX   rY   &_apply_hostname_transformations_to_url^  s"   
r  c                    s2   t d  dd dd  fdddd g}t| |S )aH  
    Given a mirror URL, replace or remove any invalid URI characters.

    This performs the following actions on the URL's hostname:
      * Checks if it is an IP address, returning the URL immediately if it is
      * Converts it to its IDN form (see below for details)
      * Replaces any non-Letters/Digits/Hyphen (LDH) characters in it with
        hyphens
      * Removes any leading/trailing hyphens from each domain name label

    Before we replace any invalid domain name characters, we first need to
    ensure that any valid non-ASCII characters in the hostname will not be
    replaced, by ensuring the hostname is in its Internationalized domain name
    (IDN) representation (see RFC 5890).  This conversion has to be applied to
    the whole hostname (rather than just the substitution variables), because
    the Punycode algorithm used by IDNA transcodes each part of the hostname as
    a whole string (rather than encoding individual characters).  It cannot be
    applied to the whole URL, because (a) the Punycode algorithm expects to
    operate on domain names so doesn't output a valid URL, and (b) non-ASCII
    characters in non-hostname parts of the URL aren't encoded via Punycode.

    To put this in RFC 5890's terminology: before we remove or replace any
    characters from our domain name (which we do to ensure that each label is a
    valid LDH Label), we first ensure each label is in its A-label form.

    (Note that Python's builtin idna encoding is actually IDNA2003, not
    IDNA2008.  This changes the specifics of how some characters are encoded to
    ASCII, but doesn't affect the logic here.)

    :param url:
        The URL to operate on.

    :return:
        A sanitized version of the URL, which will have been IDNA encoded if
        necessary, or ``None`` if the generated string is not a parseable URL.
    .c                 S   s   t | rd S | S rI   )r
   is_ip_addressr   rX   rX   rY   <lambda>  s    z&_sanitize_mirror_url.<locals>.<lambda>c                 S   s   |  ddS )Nidnaascii)encodedecoder  rX   rX   rY   r    s    c                    s   d  fdd| D S )Nr   c                 3   s     | ]}| v r
|nd V  qdS r?   NrX   ra  acceptable_charsrX   rY   r_    s    
9_sanitize_mirror_url.<locals>.<lambda>.<locals>.<genexpr>ro  r  r  rX   rY   r    s    c                 S   s   d dd | dD S )Nr  c                 s   s    | ]}| d V  qdS r  r  )r   partrX   rX   rY   r_    s    

r  )r|   r&  r  rX   rX   rY   r    s   
 )LDH_ASCII_CHARSr  )r  r  rX   r  rY   _sanitize_mirror_url  s   &

r  c              
   C   s"  | si } i }|r1|j r1|j |d< t|j r1|j dd }tr&d| |d< n|jdkr1d| |d< |r;|jr;|j|d< i }| di  D ]\}}|||< qE| d	i  D ]2\}}g }	|D ]}
z|
| }W n	 tyo   Y q^w t	|}|d ur}|	
| q^||	}|r|||< qVtd
| |S )Navailability_zoner   rx  
ec2_regionec2regionfailsafers  zfiltered distro mirror info: %s)r  
_EC2_AZ_REmatchr   platform_typer  rs   r-  KeyErrorr  r   rq   rr   )r   r   mirror_filtersubstr  resultsrL   mirror
searchlistmirrorstmplfoundrX   rX   rY   r     sD   






r   c                 C   s8   d }| D ]}| d}||v r|  S d|v r|}q|S )Narchesr   )rs   )r   r   r   itemr  rX   rX   rY   r     s   
r   rL   r\   c                 C   sH   t | dtgdg\}}|std| |f t |d }t|d}|S )Nr   r@   z1No distribution found for distro %s (searched %s)r   )r   find_moduler  ImportErrorimport_modulegetattr)rL   locslooked_locsmodr  rX   rX   rY   fetch  s   
r  /etc/timezone/etc/localtimec                 C   sj   t |t|  d  |r3|r3tj|}|stj|s-|r%t | t	|| d S t 
|| d S )Nr   )r   r   r~   rstriprz   r{   islinkr   del_filesymlinkcopy)r   r   tz_conftz_localr  rX   rX   rY   set_etc_timezone  s   
r  c                  C   s.   zt d} t| jW S  ty   Y dS w )Nz/run/systemd/systemF)rz   lstatstatS_ISDIRst_moder.  )resrX   rX   rY   r      s   
r   )Nr  r  )Fr  rz   rr  r  stringurllib.parser  ior   typingr   r   r   r   r   cloudinit.net.netops.iproute2r
   netopsrR   	cloudinitr   r	   loggingr   r   r   r   r   r   cloudinit.distros.networkingr   r   cloudinit.distros.parsersr   cloudinit.featuresr   cloudinit.netr   r   r   r   r   cloudinit.net.network_stater   cloudinit.net.rendererr   ALL_DISTROSr   	getLoggerr  rq   compiler  r   ascii_lettersdigitsr  CloudInitPickleMixinABCMetar@   r~   r   r  r  search_for_mirrorr   r   r  r  r   rX   rX   rX   rY   <module>   sj   $		
$
        	+<
3
