| Server IP : 138.197.176.125 / Your IP : 216.73.217.55 Web Server : Apache/2.4.41 (Ubuntu) System : Linux SuiteCRM-8 5.4.0-216-generic #236-Ubuntu SMP Fri Apr 11 19:53:21 UTC 2025 x86_64 User : root ( 0) PHP Version : 8.3.19 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /lib/python3/dist-packages/cloudinit/sources/__pycache__/ |
Upload File : |
U
Ӈg . � @ s� U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm m Z
d dl m Z d dl
mZ d dlmZmZmZ d dlmZmZmZmZ d dlZd dlmZmZmZm Z m Z m Z d dl m Z m!Z! d dl"m#Z# d d l$m%Z%m&Z&m'Z' d d
l(m)Z)m*Z* d d l+m,Z, d d l-m.Z.m/Z/m0Z0m1Z1 d d
l2m3Z3 d dl4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZA d dlBmCZC z2d dlDZDejEeDjDde jFdd�� �d�ZGeeHd< W nP eIeJfk
�r� zd dlKZLeLjMjNjMZGW n eIk
�r� dd� ZGY nX Y nX e�OeP�ZQdZRddiZSdZTdZUdZVd ZWG d d � d e �ZXd ZYeeZ eHd < d!d"d#d$gZ[d%d&� Z\d'd(� Z]eZeZeeZ d)�d*d+�Z^eZeZd,�d-d.�Z_e9eeZ d/�d0d1��Z`e9eZeeZ d,�d2d3��Zad|eZd/�d4d5�Zbd6d7� Zcd8d9� Zdd:d;� ZeeeZ d/�d<d=�Zfe �g� �r&d>ZUefd?�Zhehdk �reQ�id@� dAeh ZTn
eQ�idB� dZYeVdCeTidDdDdE�ZjdCdFdGgdDdH�ieUdIdJ�gdK�ZkdLeRgZldMZmdNZnG dOdP� dPejo�ZpdQdR� ZqdSdT� ZrdUdV� ZsdWdX� ZtdYdZ� Zud}d[d\�Zve9d]d^� �Zwe9d_d`� �Zxe9eTdadafeZeZeyeydb�dcdd��Zze9d~dedf��Z{e9dgdh� �Z|didj� Z}e9dkdl� �Z~e9eYfdmdn��Ze9dodp� �Z�e9dqdr� �Z�e9e�eye�ds�dtdu��Z�e9e�d/�dvdw��Z�e9ddxdy��Z�epZ�epej�ffgZ�dzd{� Z�dS )�� N)�Enum)�Path)� monotonic�sleep�time)�Any�Dict�List�Optional)�net� performance�sources�ssh_util�subp�util)�
EventScope� EventType)�
device_driver)�NoDHCPLeaseError�NoDHCPLeaseInterfaceError� NoDHCPLeaseMissingDhclientError)�EphemeralDHCPv4�EphemeralIPv4Network)�events)�errors�identity�imds�kvp)�netlink)
�DEFAULT_WIRESERVER_ENDPOINT�NonAzureDataSource� OvfEnvXml�azure_ds_reporter�azure_ds_telemetry_reporter�build_minimal_ovf� dhcp_log_cb�get_boot_telemetry�get_metadata_from_fabric�get_system_info�report_diagnostic_event�report_dmesg_to_kvp�report_failure_to_fabric)�UrlErrorz$6$� )Zstrlen)Zsalt�
blowfish_hashc C s t d��dS )z�Raise when called so that importing this module doesn't throw
ImportError when ds_detect() returns false. In this case, crypt
and passlib are not needed.
z/crypt and passlib not found, missing dependencyN)� ImportError)�_� r1 �C/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceAzure.pyr. @ s ��Azure� instance-idziid-AZURE-NODEz /dev/disk/cloud/azure_resourceZext4z/var/lib/waagentz/dev/sr0c @ s e Zd ZdZdZdZdZdZdS )�PPSType�NoneZPreprovisionedOSDiskZRunningZSavableZUnknownN)�__name__�
__module__� __qualname__�NONE�OS_DISK�RUNNING�SAVABLE�UNKNOWNr1 r1 r1 r2 r5 W s
r5 z /sys/firmware/acpi/tables/OEM0�PLATFORM_ENTROPY_SOURCEz"/etc/netplan/90-hotplug-azure.yamlz /usr/local/sbin/ephemeral_eth.shz+/etc/udev/rules.d/10-net-device-added.rulesz#/run/network/interfaces.ephemeral.dc C s� | � � D ]r}t�d|�r|�� }t|�dkr|d �d�}t|�dkr|d dkr|d �|�r|d �d�}|d S qd S )
z�
dev.storvsc.1.%pnpinfo:
classid=32412632-86cb-44a2-9b5c-50d1417354f5
deviceid=00000000-0001-8899-0000-000000000000
Zpnpinfo� � �=r �deviceid� �.N)�
splitlines�re�search�split�len�
startswith)�
sysctl_outrC �lineZfields�columns�compsr1 r1 r2 �"find_storvscid_from_sysctl_pnpinfok s
�
� �rP c C s2 | � � D ]$}t�||�r|�� }|d S qdS )z�
scbus0 on ata0 bus 0
scbus1 on ata1 bus 0
scbus2 on blkvsc0 bus 0
scbus3 on blkvsc1 bus 0
scbus4 on storvsc2 bus 0
scbus5 on storvsc3 bus 0
scbus-1 on xpt0 bus 0
r N)rF rG rH rI )�camcontrol_outZdisk_drvrM �itemsr1 r1 r2 �find_busdev_from_disk� s
rS )rQ �busdev�returnc C sN | � � D ]@}t�||�r|�d�}t|�dkr|d �d�}|d S qdS )z�
<Msft Virtual CD/ROM 1.0> at scbus1 target 0 lun 0 (cd0,pass0)
<Msft Virtual Disk 1.0> at scbus2 target 0 lun 0 (da0,pass1)
<Msft Virtual Disk 1.0> at scbus3 target 1 lun 0 (da1,pass2)
�(rA rD �,r N)rF rG rH rI rJ )rQ rT rM rR Zdev_passr1 r1 r2 �find_dev_from_busdev� s
rX )�macrU c
C sZ t | �dkrRd�| dd� | dd� | dd� | dd� | dd� | dd� g�} | �� S ) z1Normalize mac address with colons and lower-case.� �:r rA � � � �
)rJ �join�lower�rY r1 r1 r2 �normalize_mac_address� s
>�rc �rU c C s dd� t �� D �S )z-Get Hyper-V NICs as normalized MAC addresses.c S s$ g | ] }|d dkrt |d ��qS )rA � hv_netvscrD �rc )�.0�nr1 r1 r2 �
<listcomp>� s �z1get_hv_netvsc_macs_normalized.<locals>.<listcomp>)r �get_interfacesr1 r1 r1 r2 � get_hv_netvsc_macs_normalized� s �rk c sf � fdd�t �� D �}d|kr"dS t|�dkrLtd� |f tjd� |d S td� |f tjd� d S )
z0Determine the device driver to match on, if any.c s$ g | ] }� t |d �kr|d �qS )rD rA rf �rg �irb r1 r2 ri � s �z3determine_device_driver_for_mac.<locals>.<listcomp>re rD z4Assuming driver for interface with mac=%s drivers=%r�Z logger_funcr z=Unable to specify driver for interface with mac=%s drivers=%rN)r rj rJ r) �LOG�debug�warning)rY Zdriversr1 rb r2 � determine_device_driver_for_mac� s(
� ����rr c C s@ zt � | �jW S t jk
r: t�dd�| �� | Y S X d S )NzFailed to execute: %s� )r �stdout�ProcessExecutionErrorro rp r` )�cmd�fail_retr1 r1 r2 �execute_or_debug� s
rx c C s t ddgdd�S )NZsysctlz dev.storvsc� )rw �rx r1 r1 r1 r2 �get_dev_storvsc_sysctl� s r{ c C s t dddg�S )N�
camcontrol�devlistz-brz r1 r1 r1 r2 �get_camcontrol_dev_bus� s r~ c C s t ddg�S )Nr| r} rz r1 r1 r1 r2 �get_camcontrol_dev� s r c C s� d}| dkrd}| d } dt | � }d�||�}t� }t||�}|sRt�d� d S t� }t� }d�|�}t||�} | r�t || �}
|
d kr�t�d � d S |
S d
�|�} t|| �} | r�t || �}
|
d kr�t�d � d S |
S d S ) NZ00000000rD Z00000001rA Z000z{0}-{1}z#Fail to find storvsc id from sysctlz blkvsc{0}zFail to find /dev/daXz
storvsc{0})
�str�formatr{ rP ro rp r~ r rS rX ) Zport_idZg0Zg1Zg0g1rL Z storvscidZcamcontrol_b_outrQ ZblkvscZscbusxZdevnameZstorvscr1 r1 r2 � get_resource_disk_on_freebsd� s<
r� z freebsd-ufsrD zresource disk is not None�/dev/zresource disk is NoneZ
ephemeral0T)�data_dir� disk_aliases�apply_network_config�&apply_network_config_for_secondary_ipsZgpt�d )Z
table_typeZlayoutZ overwritez ephemeral0.1)Z
filesystemZdevice)�
disk_setupZfs_setupZ
datasourceZnever_destroy_ntfsZREDACTEDc s e Zd ZdZejejejhiZ dZ
e j Z
dZdd� Zedd�� fdd � Zd
d � Zd d
� Zeeed�dd��Zeddddd�ee eeeed�dd��Zedd�dd��Zed�dd�Z dd�d d �Z ed d � �Z eee d �d!d"��Z du� fd$d%� Z!ed&d'� �Z"ed(d)� �Z#d*d+� Z$d,d-� Z%ee&e d�d.d/��Z'e(e&e d0�d1d2�Z)e&e d�d3d4�Z*d5d6� Z+d7d8� Z,dvd9d:�Z-ed;d<� �Z.edweee/d?�d@dA��Z0edBdC� �Z1edddD�eeddE�dFdG��Z2edHdI� �Z3ee4j4d�dJdK��Z5edLdM� �Z6edd�dNdO��Z7edPdQ� �Z8edRdS� �Z9ee:d�dTdU��Z;edxe<j=eedV�dWdX��Z>eddY�ee&e ee&e dZ�d[d\��Z?e(ee d0�d]d^�Z@e(e(eAd_�d`da�ZBedbdc� �ZCee(e(ee&e dd�dedf��ZDdgdh� ZEedidj� �ZFeGdkdl� �ZHedmdn� �ZIeGdodp� �ZJeGdqdr� �ZKee(ed0�dsdt��ZL� ZMS )y�DataSourceAzurer3 FrD c C s� t j�| |||� tj�|jd�| _i | _d | _t �
t � |t i �t
g�| _d | _d | _d | _d| _d| _t| _tj�|jdd�| _d S )NZazureF�data�reported_ready)r
�
DataSource�__init__�os�pathr` �seed_dir�cfg�seedr �
mergemanydictZget_cfg_by_path� DS_CFG_PATH�BUILTIN_DS_CONFIG�ds_cfg�_iso_dev�_network_config�_ephemeral_dhcp_ctx�_route_configured_for_imds� _route_configured_for_wireserverr �_wireserver_endpoint� cloud_dir�_reported_ready_marker_file)�self�sys_cfg�distro�pathsr1 r1 r2 r� D s$ � �zDataSourceAzure.__init__N)�ci_pkl_versionrU c sD t � �|� d | _d | _d| _d| _t| _tj �
| j j dd�| _
d S )NFr� r� )�super� _unpickler� r� r� r� r r� r� r� r` r� r� r� )r� r� �� __class__r1 r2 r� V s �zDataSourceAzure._unpicklec C s t j�| �}d|| jf S )Nz %s [seed=%s])r
r� �__str__r� )r� �rootr1 r1 r2 r� b s zDataSourceAzure.__str__c C sH | j dkrd}n*| j �d�r"d}n| j �� dkr6d}nd}d|| j f S )z/Return the subplatform metadata source details.N�unknownz/devz config-diskr zseed-dirz%s (%s))r� rK ra )r� Zsubplatform_typer1 r1 r2 �_get_subplatformf s
z DataSourceAzure._get_subplatform)�ephipv4rU c C s< |j s
dS dd� |j D �}d|k| _| j|k| _| jp:| jS )NFc S s g | ]}|d � d�d �qS )r �/)rI )rg �rr1 r1 r2 ri x s z5DataSourceAzure._check_if_primary.<locals>.<listcomp>z169.254.169.254)�
static_routesr� r� r� )r� r� Zrouted_networksr1 r1 r2 �_check_if_primaryr s
��z!DataSourceAzure._check_if_primaryT� )�iface� report_failure_if_not_primary� retry_sleep�timeout_minutes)r� r� r� r� rU c C st | j dk rtd��td|t�� f tjd� t| j|t d�| _ d}t
� }||d }t j ddt
d ��� |dk�r�z| j �� }W n� tk
r� td
tjd� | jtjt
� | d �d d
� Y n� tk
r� tdtjd� d| _ � Y n� tk
�r* td| tjd� | jtjt
� | |d�d d
� Y nH tjk
�rp } z&td|j|j|j |j f tjd� W 5 d}~X Y nX |dk�r�t
� | |k �r�t |� qh�q�qh|dk�r�d| _ t� �|d }|| j _ d|k�r�|d | _ t!|�} | j j"}
|
dk�r�td��| �#|
�} td|| | |
j$|
j%|| j&| j'f tjd� |�rZ| �sZ| jtj(|| |
j$|
j%|d�d d
� | W 5 Q R � S Q R X dS )a� Setup ephemeral networking.
Keep retrying DHCP up to specified number of minutes. This does
not kill dhclient, so the timeout in practice may be up to
timeout_minutes + the system-configured timeout for dhclient.
:param timeout_minutes: Number of minutes to keep retrying for.
:raises NoDHCPLeaseError: If unable to obtain DHCP lease.
:returns: True if NIC is determined to be primary.
Nz/Bringing up networking when already configured.z2Bringing up ephemeral networking with iface=%s: %rrn )r� Z
dhcp_log_func�< zobtain-dhcp-leasezobtain dhcp lease��name� description�parentz Interface not found for DHCP)�durationT�� host_onlyz dhclient executable not foundz&Failed to obtain DHCP lease (iface=%s))r� � interfacez7Command failed: cmd=%r stderr=%r stdout=%r exit_code=%sr� z unknown-245z dhcp context missing ephipv4z{Obtained DHCP lease on interface %r (primary=%r driver=%r router=%r routes=%r lease=%r imds_routed=%r wireserver_routed=%r))r� �driver�routerr� �lease))r� � RuntimeErrorr) r rj ro rp r r� r% r r �ReportEventStackr" Z obtain_leaser rq �_report_failurer Z$ReportableErrorDhcpInterfaceNotFoundr �errorr ZReportableErrorDhcpLeaser ru rv �stderrrt � exit_coder r� r� r Z_ephipv4r� r� r� r� r� Z(ReportableErrorDhcpOnNonPrimaryInterface) r� r� r� r� r� r� �
start_timeZdeadliner� r� r� Zprimaryr1 r1 r2 �_setup_ephemeral_networking� s�
�
��� �
���
�� ��
���
��
� ��
z+DataSourceAzure._setup_ephemeral_networkingrd c C s. d| _ d| _| jdkrdS | j�� d| _dS )z Teardown ephemeral networking.FN)r� r� r� Z
clean_network�r� r1 r1 r2 � _teardown_ephemeral_networking s
z.DataSourceAzure._teardown_ephemeral_networkingc C s | j dkp| j jdk S )z"Check if networking is configured.N)r� r� r� r1 r1 r2 �_is_ephemeral_networking_up% s
�z+DataSourceAzure._is_ephemeral_networking_upc
C s� z6ddddg}t � |�\}}td|||f tjd� W n� t jk
r� } zdt|jt�rtt�d� t �
� }| � |� n6td|j |j
|j|jf tjd� t �|�}| � |� W 5 d }~X Y nX d S )
ab Check if azure-proxy-agent is ready for communication with WS/IMDS.
If ProvisionGuestProxyAgent is true, query azure-proxy-agent status,
waiting up to 120 seconds for the proxy to negotiate with Wireserver
and configure an eBPF proxy. Once azure-proxy-agent is ready,
it will exit with code 0 and cloud-init can then expect to be able to
communicate with these services.
Fail deployment if azure-proxy-agent is not found or otherwise returns
an error.
For more information, check out:
https://github.com/azure/guestproxyagent
zazure-proxy-agentz--statusz--waitZ120z1Executing %s resulted in stderr=%r with stdout=%rrn zGFailed to activate Azure Guest Proxy Agent: azure-proxy-agent not foundzgFailed to activate Azure Guest Proxy Agent: status check failed cmd=%r stderr=%r stdout=%r exit_code=%sN)r r) ro rp ru �
isinstance�reason�FileNotFoundErrorr� r Z!ReportableErrorProxyAgentNotFoundr� rv r� rt r� Z&ReportableErrorProxyAgentStatusFailure)r� rv �out�errr� Z report_error�reportable_errorr1 r1 r2 � _check_azure_proxy_agent_status, sB ���
� �� �
�z/DataSourceAzure._check_azure_proxy_agent_statusc
C s
i }| j d }d}ddi}ddddiii}d}i }t| j|�D ]�}zz|�d�r�t�� rptj|td d
�\}}}}nt�|t�\}}}}|| _nt|�\}}}}|}t d | t
j d � W �q(W q> t k
r� t d
| t
j d � Y q>Y q> tj
k
�r t d| t
j d � Y q>Y q>X q>d} t | t
jd � t| j��p:|dk}
|
�rFdnd} z| j| d� W n tk
�rp Y nX i } | �� �r�|�d��r�| �� | jdd�} | �s�|dk�r�d} t | � t�| ��| �|| �}
|
tjk�r�t�� �rd} t | t
jd � t�| ��| �� �s*d} t | t
jd � t�| ��|
tj k�r@| � � n4|
tj k�rV| � � n |
tj k�rl| �!� n| �"� | �#� \}}}}|�d��r�| �� | jdd�} | �$| �}|dtjj%fk�r�| �&t'j(d|d�� | j)| d� |�p�d | _*|�+||t�,|d | ig�|d �� t-| �}t.| �}t/| �}|�rHt
� d |� ||d d d< |�rft
� d |� ||d! d< |�r�t
� d"|� ||d! d#< | j*d k�r�|d$ �s�z t0|||d%�}d&|i|d$< W n6 t1k
�r� } zt d'| t
j d � W 5 d}~X Y nX |�sPt2| �}|�rPt
� d(� z t3�4d�5|�6� ��|d)< W n$ t1k
�rN t d*t
jd � Y nX ||k�rlt d+| t
j d � t7� }|�r�||d! d,< | �8� |d! d-< | j9d.k�r| �� �r| j:|| d/�}z| j;|d0�}W n t1k
�r� Y n(X t
� d1|� |�r�||d! d2< | �<� |S )3a Walk all instance metadata sources returning a dict on success.
@return: A dictionary of any metadata content for this instance.
@raise: InvalidMetaDataException when the expected metadata service is
unavailable, broken or disabled.
r� N�local-hostnamery � system_info� default_userr� r� �udf)�mtypez!Found provisioning metadata in %srn z$Did not find Azure data source in %sz%s was not mountablez�Unable to find provisioning media, falling back to IMDS metadata. Be aware that IMDS metadata does not support admin passwords or custom-data (user-data only).� r� �r� �ProvisionGuestProxyAgentT)�report_failurezNo OVF or IMDS availablez%Free BSD is not supported for PPS VMsz DHCP failed while in source PPSzextended.compute.ppsType)�key�value)�imds_mdZIMDSr )r� �files�metadata� userdata_rawz Username retrieved from IMDS: %sz Hostname retrieved from IMDS: %sr� z(Disable password retrieved from IMDS: %sZdisable_passwordr� )�username�hostnameZ
disableSshPwd� ovf-env.xmlz)Failed to construct OVF from IMDS data %sz Retrieved userdata from IMDSr� zBad userdata in IMDSzusing files cached in %sZ random_seedr4 F)r� r� �� pubkey_infoznegotiating returned %s� public-keys)=r� �list_possible_azure_dsr� rK r �
is_FreeBSD�mount_cb�load_azure_ds_dirr� r) ro rp r �MountFailedErrorrq �boolr� r r� �getr� �get_metadata_from_imdsr
�InvalidMetaDataException�_determine_pps_typer5 r: r� r< �_wait_for_pps_running_reuser= �_wait_for_pps_savable_reuser; � _wait_for_pps_os_disk_shutdown�_wait_for_pps_unknown_reuse� _reprovision�_ppstype_from_imdsr� r� r Z"ReportableErrorImdsInvalidMetadata� validate_imds_network_metadatar� �updater� �_username_from_imds�_hostname_from_imds�_disable_password_from_imdsr$ � Exception�_userdata_from_imds�base64Z b64decoder` rI �_get_random_seed�_iid� _negotiated�!_determine_wireserver_pubkey_info�
_report_ready�_cleanup_markers)r� � crawled_dataZddirZ
ovf_source�mdr� r� r� �src�msgZrequires_imds_metadatar� r� �pps_typeZ imds_ppstypeZ
imds_usernameZ
imds_hostnameZimds_disable_password�contents�eZ
imds_userdatar� r� �ssh_keysr1 r1 r2 �crawl_metadatab sF
� � � � � �
�� �� �����
� �
� � z DataSourceAzure.crawl_metadata)r� rU c
C s� t � }|d }| jsd}nd }d }d }ztj||d�W S tk
r� } z<t|�}t � | }tj||d�}| js�t|j t
j �r�d}W 5 d }~X Y n6 t k
r� } zt|�}tj
|d�}W 5 d }~X Y nX | j|| d� td| tjd � i S )
Ni, � )�max_connection_errors�retry_deadline)� exceptionr� F)r r� z!Ignoring IMDS metadata due to: %srn )r r� r Z fetch_metadata_with_api_fallbackr, r� r ZReportableErrorImdsUrlErrorr� �cause�requests�ConnectionError�
ValueErrorZ+ReportableErrorImdsMetadataParsingExceptionr� r) ro rq ) r� r� r� r r Z error_stringZ error_reportr� r� r1 r1 r2 r� < sF �
� ���z&DataSourceAzure.get_metadata_from_imdsr1 c s t t| ��|� tj| _dS )z.Reset any cached class attributes to defaults.N)r� r� �clear_cached_attrsr
�UNSET�_metadata_imds)r� Z
attr_defaultsr� r1 r2 r j s z"DataSourceAzure.clear_cached_attrsc C s4 t j�� }|dk rdS | jdkr$dS t| jd��� S )zQCheck platform environment to report if this datasource may
run.
NTFr� )r ZChassisAssetTagZ query_systemr� r �exists)r� Z chassis_tagr1 r1 r2 � ds_detecto s
zDataSourceAzure.ds_detectc
C s� z
t � W n. tk
r8 } zt�d|� W 5 d}~X Y nX z
t� W n. tk
rr } zt�d|� W 5 d}~X Y nX z�z | �� }W nx tjk
r� } z| � |� W Y �W �VdS d}~X Y nB tk
� r� } z"t�
|�}| � |� W Y �W �dS d}~X Y nX W 5 | �� X | j �r4| j j dk�r4| j
�d��r4t� t}tj�|��rntd| tjd� t�|d tg�| _n td
| tjd� |d | _|d d | _t�|d tg�| _ |d
| _ t | j
d |d dd� dS )z�Crawl and process datasource metadata caching metadata as attrs.
@return: True on success, False on error, invalid or disabled
datasource.
z Failed to get boot telemetry: %sNz$Failed to get system information: %sFZubuntur� zXEphemeral resource disk '%s' exists. Merging default Azure cloud ephemeral disk configs.rn r� zdEphemeral resource disk '%s' does not exist. Not merging default Azure cloud ephemeral disk configs.r� r r� r� r� i� )�dirmodeT) r& r� ro rq r( r� r r �ReportableErrorr� Z!ReportableErrorUnhandledExceptionr� r� r� r� �*maybe_remove_ubuntu_network_config_scripts�RESOURCE_DISK_PATHr� r� r r) rp r r� �#BUILTIN_CLOUD_EPHEMERAL_DISK_CONFIGr� r �DEFAULT_METADATAr� r� � write_files)r� r r r� r� �devpathr1 r1 r2 � _get_data~ sl
�
�
���
���
�
�zDataSourceAzure._get_datac C s& | j rd| j kr| �� S t| j d �S )Nr4 )r� r r� r� r1 r1 r2 �get_instance_id� s z DataSourceAzure.get_instance_idc C s | j d �|�S )Nr� )r� r� )r� r� r1 r1 r2 �device_name_to_device� s z%DataSourceAzure.device_name_to_devicec C s4 z| � | jd �W S ttfk
r* Y nX | �� S )z+
Retrieve public SSH keys.
r )�_get_public_keys_from_imdsr� �KeyErrorr �_get_public_keys_from_ovfr� r1 r1 r2 �get_public_ssh_keys� s
z#DataSourceAzure.get_public_ssh_keys)r� rU c C s� zdd� |d d D �}W n( t k
rB d}t|tjd� � Y nX tdd� |D ��rpd }t|tjd� t|��d
�t|��}t|tjd� |S ) z�Get SSH keys from IMDS metadata.
:raises KeyError: if IMDS metadata is malformed/missing.
:raises ValueError: if key format is not supported.
:returns: List of keys.
c S s g | ] }|d �qS )ZkeyDatar1 )rg Z
public_keyr1 r1 r2 ri � s �z>DataSourceAzure._get_public_keys_from_imds.<locals>.<listcomp>�computeZ
publicKeysz"No SSH keys found in IMDS metadatarn c s s | ]}t |d � V qdS ))r� N)�_key_is_openssh_formatted)rg r� r1 r1 r2 � <genexpr>� s z=DataSourceAzure._get_public_keys_from_imds.<locals>.<genexpr>z Key(s) not in OpenSSH formatzRetrieved {} keys from IMDS)r* r) ro rp �anyr r� rJ )r� r� r �log_msgr1 r1 r2 r) � s
�
z*DataSourceAzure._get_public_keys_from_imdsc C sZ g }z*| j d }d�t|��}t|tjd� W n& tk
rT d}t|tjd� Y nX |S )zYGet SSH keys that were fetched from wireserver.
:returns: List of keys.
r� zRetrieved {} keys from OVFrn zNo keys available from OVF)r� r� rJ r) ro rp r* )r� r r1 r1 r1 r2 r+ � s
z)DataSourceAzure._get_public_keys_from_ovfc C s | j S �N�r� r� r1 r1 r2 �get_config_obj s z DataSourceAzure.get_config_objc C s t �| �� �S r2 )r
Z instance_id_matches_system_uuidr' )r� r� r1 r1 r2 �check_instance_id s z!DataSourceAzure.check_instance_idc C sZ t j�| j�d�d�}t�� }t j�|�rVt� |��
� }t� |�}|� � ||fkrV|S |S )Nr� r4 )
r� r� r` r� Z get_cpathr Zquery_system_uuidr r Zload_text_file�stripZbyte_swap_system_uuidra )r� ZpreviousZ
prev_iid_pathZ system_uuidZ
swapped_idr1 r1 r2 r
s
�
zDataSourceAzure._iidc
C s� z\d}t jddtd�� t�|�}W 5 Q R X |dkrHd}t|tjd� ntd| tjd� W n6 t k
r� } ztt
|�tj d� � W 5 d}~X Y nX dS )z�Use the netlink socket provided to wait for nic detach event.
NOTE: The function doesn't close the socket. The caller owns closing
the socket and disposing it safely.
Nzwait-for-nic-detachzwait for nic detachr� zHPreprovisioned nic not detached as expected. Proceeding without failing.rn z%The preprovisioned nic %s is detached) r r� r" r Zwait_for_nic_detach_eventr) ro rq rp �AssertionErrorr� r� )r� �nl_sock�ifnamer r� r1 r1 r2 �_wait_for_nic_detach s& ���
z$DataSourceAzure._wait_for_nic_detachr� 皙�����?)r9 �retriesr� c C sd t |�D ]<}| jj�|�r0td| tjd� q`|d |k rt|� qtd||| f tjd� d S )NzThe link %s is up.rn rD z;The link %s is not up after %f seconds, continuing anyways.)�ranger� Z
networkingZtry_set_link_upr) ro �infor )r� r9 r< r� rm r1 r1 r2 �wait_for_link_up> s �
��z DataSourceAzure.wait_for_link_upc C s@ | j }t�d|� t�|djt�� t� d�� t dtj
d� d S )Nz*Creating a marker file to report ready: %sz{pid}: {time}
)�pidr zRSuccessfully created reported ready marker file while in the preprovisioning pool.rn ) r� ro r> r �
write_filer� r� �getpidr r) rp )r� r� r1 r1 r2 �_create_report_ready_markerR s ��z+DataSourceAzure._create_report_ready_marker)�
create_marker�expect_url_error)rD rE rU c
C s� z | � � W nd tk
rp } zF|rBt|t�rBtdtjd� d| _n d}t|tjd� t �
|�|�W 5 d}~X Y nX d| _ |r�| � � dS )z�Report ready for PPS, creating the marker file upon completion.
:raises sources.InvalidMetaDataException: On error reporting ready.
z,Ignoring http call failure, it was expected.rn Nz9Failed reporting ready while in the preprovisioning pool.F)
r r� r� r, r) ro rp r� r� r
r� r rC )r� rD rE r� r r1 r1 r2 �_report_ready_for_pps_ s �� z%DataSourceAzure._report_ready_for_ppsc
C s� t �d� z�g }d}d}tjddt|� td�� t�||�}W 5 Q R X |�|� t d| t jd� | �
|� |s�t �d |� | j |d
dd �}|r�t �d � q�q| � � qW n4 t
k
r� } zt t|�t jd� W 5 d}~X Y nX dS )
z6Wait until the primary nic for the vm is hot-attached.z*Waiting for primary nic to be hot-attachedFNzwait-for-nic-attachz4wait for nic attach after %d nics have been attachedr� zDetected nic %s attached.rn z!Checking if %s is the primary nicr� )r� r� r� z Found primary nic for this VM.)ro r> r r� rJ r" r Zwait_for_nic_attach_event�appendr) r? r� r� r7 r� r� )r� r8 Z
nics_foundZprimary_nic_foundr9 r� r1 r1 r2 �"_wait_for_hot_attached_primary_nic� sF
�� �
�
�
z2DataSourceAzure._wait_for_hot_attached_primary_nicc
C sJ z
t �� W S t jk
rD } ztd|� �tjd� � W 5 d }~X Y nX d S )Nz!Failed to create netlink socket: rn )r Zcreate_bound_netlink_socketZNetlinkCreateSocketErrorr) ro r� )r� r� r1 r1 r2 � _create_bound_netlink_socket� s
�z,DataSourceAzure._create_bound_netlink_socketc C s. | j dd� tdtjd� td� t�� �dS )z4Report ready and wait for host to initiate shutdown.F)rD z"Waiting for host to shutdown VM...rn i�3�N)rF r) ro r> r r Z ReportableErrorOsDiskPpsFailurer� r1 r1 r2 r� � s �z.DataSourceAzure._wait_for_pps_os_disk_shutdownc C s� | � � }z�| jdks | jjdkr(td��| jj}| �� t�d|� tj ddt
d��L zt � ||� W n4 t
k
r� } ztd| tjd� W 5 d}~X Y nX W 5 Q R X W 5 |�� X | �� dS ) z9Report ready and wait for nic link to switch upon re-use.Nzmissing ephemeral contextz#Wait for vnetswitch to happen on %sz!wait-for-media-disconnect-connectzwait for vnet switchr� z'Error while waiting for vnet switch: %srn )rI �closer� r� r� rF ro rp r r� r" r Z!wait_for_media_disconnect_connectr7 r) r� r� )r� r8 r� r r1 r1 r2 r� � s6 �
����&
z+DataSourceAzure._wait_for_pps_running_reusec
C s� | � � }zn| jdd� z | �� W n< tjk
r^ } z td| tjd� d| _ W 5 d}~X Y nX | �
|� | � |� W 5 |�� X dS )z@Report ready and wait for nic(s) to be hot-attached upon re-use.T)rE zLIgnoring failure while tearing down networking, NIC was likely unplugged: %rrn N) rI rJ rF r� r ru r) ro r> r� r: rH )r� r8 r r1 r1 r2 r� � s ��
z+DataSourceAzure._wait_for_pps_savable_reusec C s& t j�| j�rdS | �� | �� dS )z0Report ready if needed for unknown/recovery PPS.N)r� r� �isfiler� rF r� r� r1 r1 r2 r� s z+DataSourceAzure._wait_for_pps_unknown_reusec C s� d}d}|s�| � � sF|d7 }z| jdd� W n tk
rD Y qY nX tjddtd��> z t�� }W n* tk
r� | � � Y W 5 Q R � qY nX W 5 Q R X qt
d | t j d
� |S ) z0Poll IMDs for reprovisiondata XML document data.r NrD r� r� z get-reprovision-data-from-imdsz get reprovision data from imdsr� z#attempted dhcp %d times after reusern )
r� r� r r r� r" r Zfetch_reprovision_datar, r� r) ro rp )r� Z
dhcp_attemptsZreprovision_datar1 r1 r2 �
_poll_imds s0
� �zDataSourceAzure._poll_imds)r� r� rU c
C s( t d|�� � �tjd� t� t�|�}|r0|S | �� r�z(t dtjd� t | j
|d� d| _ W dS t k
r� } zt d| tjd� W 5 d}~X Y nX zVt dtjd� | �
� z| jd d
� W n tk
r� Y nX t | j
|d� d| _ W dS t k
�r" } zt d | tjd� W 5 d}~X Y nX d S )
a� Report failure to Azure host and fabric.
For errors that may be recoverable (e.g. DHCP), host_only provides a
mechanism to report the failure that can be updated later with success.
DHCP will not be attempted if host_only=True and networking is down.
@param error: Error to report.
@param host_only: Only report to host (error may be recoverable).
@return: The success status of sending the failure signal.
z#Azure datasource failure occurred: rn z>Using cached ephemeral dhcp context to report failure to Azure)�endpointr� Tz@Failed to report failure using cached ephemeral dhcp context: %sNz3Using new ephemeral dhcp to report failure to Azurer� r� z5Failed to report failure using new ephemeral dhcp: %sF)r) Zas_encoded_reportro r� r* r Zreport_failure_to_hostr� rp r+ r� r r� r� r� r )r� r� r� Zreportedr r1 r1 r2 r� ; sb �
� ���� ��z DataSourceAzure._report_failurer� )r� rU c
C sp t � t�� zt| j| j| j|d�}W n6 tk
r^ } ztd| t j
d� � W 5 d}~X Y nX d| _d| _ |S )z�Tells the fabric provisioning has completed.
:param pubkey_info: Fingerprints of keys to request from Wireserver.
:raises Exception: if failed to report.
:returns: List of SSH keys, if requested.
)rM r� Ziso_devr� zQError communicating with Azure fabric; You may experience connectivity issues: %srn NT) r* r Zreport_success_to_hostr' r� r� r� r� r) ro rq r )r� r� r� r r1 r1 r2 r } s( �
��z DataSourceAzure._report_readyc
C sR z|d d d W S t k
rL } z td| tjd� W Y �d S d }~X Y nX d S )NZextendedr- ZppsTypez2Could not retrieve pps configuration from IMDS: %srn )r� r) ro rp )r� r� r r1 r1 r2 r� � s �z"DataSourceAzure._ppstype_from_imds)�ovf_cfgr� rU c C s� t j�| j�rtj}n�|�dd�tjjks<| � |�tjjkrDtj}np|�dd�tj
jksj| � |�tj
jkrrtj
}nB|�d�dks�|�dd�tj jks�| � |�tj jkr�tj }ntj }t
d|j tjd� |S )z@Determine PPS type using OVF, IMDS data, and reprovision marker.�PreprovisionedVMTypeN�PreprovisionedVmTz PPS type: %srn )r� r� rK r� r5 r> r� r= r� r� r; r< r: r) ro r> )r� rN r� r r1 r1 r2 r� � s6 ���� �
��� �z#DataSourceAzure._determine_pps_typec
C sN | � � }tjddtd��, t|�\}}}|||d|ifW 5 Q R � S Q R X dS )zrInitiate the reprovisioning workflow.
Ephemeral networking is up upon successful reprovisioning.
z reprovisioning-read-azure-ovfz$read azure ovf during reprovisioningr� r� N)rL r r� r" �read_azure_ovf)r� r
r �udr� r1 r1 r2 r� � s �z DataSourceAzure._reprovision)r� r� rU c C sd d}z| � |� W nL ttfk
r^ |�dd�}d�|dk rFt|�nd�}t|tjd� Y nX |S )z�Determine the fingerprints we need to retrieve from Wireserver.
:return: List of keys to request from Wireserver, if any, else None.
N�_pubkeysz"Retrieved {} fingerprints from OVFr rn ) r) r* r r� r� rJ r) ro rp )r� r� r� r� r1 r1 r1 r2 r � s �z1DataSourceAzure._determine_wireserver_pubkey_infoc C s t �| j� dS )zCleanup any marker files.N)r �del_filer� r� r1 r1 r2 r � s z DataSourceAzure._cleanup_markersc C s4 | j �� }z t||| j�td�d� W 5 t� X d S )NF)�is_new_instance�
preserve_ntfs)r� Z
get_ipath_curr* �address_ephemeral_resizer� r� �DS_CFG_KEY_PRESERVE_NTFS)r� r� rU � instance_dirr1 r1 r2 �activate� s
�
zDataSourceAzure.activatec C s | j �di ��di ��d�S )Nr r- ZplatformFaultDomain�r� r� r� r1 r1 r2 �availability_zone� s ���z!DataSourceAzure.availability_zonec
C s� | j rn| j tjkrn| j�d�rnz t| j d | j�d�d�W S tk
rl } zt�dt |�� W 5 d}~X Y nX zt
� W S tk
r� } zt�dt |�� W 5 d}~X Y nX i S )z:Generate network configuration according to configuration.r� �networkr� )r� z?Failed generating network config from IMDS network metadata: %sNz-Failed generating fallback network config: %s) r r
r r� r� �6generate_network_config_from_instance_network_metadatar� ro r� r� �-_generate_network_config_from_fallback_config)r� r r1 r1 r2 �_generate_network_config s. �
�
����"z(DataSourceAzure._generate_network_configc C s( | j r| j tjkr| j S | �� | _ | j S )z,Provide network configuration v2 dictionary.)r� r
r r` r� r1 r1 r2 �network_config! s
z DataSourceAzure.network_configc C s | j �di ��di ��d�S )Nr r- �locationr[ r� r1 r1 r2 �region+ s zDataSourceAzure.regionc s� t � }z |d }dd� |d D �� W n. tk
rR td|�d� tjd� Y dS X � fdd�|D �}|snd S td
||f tjd� | jr�| jjs�dS t� | jj�}|r�t
|t �s�dS t |�}||kr�td ||f tjd� dS ) z=Validate IMDS network config and report telemetry for errors.r] c S s g | ]}t |d ��qS )�
macAddressrf rl r1 r1 r2 ri 6 s �zBDataSourceAzure.validate_imds_network_metadata.<locals>.<listcomp>r� z6IMDS network metadata has incomplete configuration: %rrn Fc s g | ]}|� kr|�qS r1 r1 )rg �m�Z imds_macsr1 r2 ri B s Tz>IMDS network metadata is missing configuration for NICs %r: %rz3IMDS network metadata is missing primary NIC %r: %r)
rk r* r) r� ro rq r� r� r Zget_interface_macr� r� rc )r� r� Z
local_macsra Z missing_macsZ primary_macr1 rf r2 r� / sJ �
������z.DataSourceAzure.validate_imds_network_metadata)r1 )N)r� r; )F)Nr7 r8 r9 Zdsnamer ZNETWORKr ZBOOT_NEW_INSTANCE�BOOTZdefault_update_eventsr r
r r Z_ci_pkl_versionr� �intr� r� r� r# r r� r� r
r� r� r� r� r� r r r� r r r& r' r( r r, �dictr) r+ r4 r5 r r: �floatr? rC rF rH �socketrI r� r� r� r� �bytesrL r r r� r r� r5 r� r� r r rZ �propertyr\ r` ra rc r� �
__classcell__r1 r1 r� r2 r� 8 s� �� ��
6
Z-
H
� �
��&
8
$
� �A�
�"
�
r� c C s. z| d d d W S t k
r( Y d S X d S )Nr- � osProfileZ
adminUsername�r* �Z imds_datar1 r1 r2 r� ` s r� c C s* z| d d W S t k
r$ Y d S X d S )Nr- ZuserDatarp rq r1 r1 r2 r g s r c C s. z| d d d W S t k
r( Y d S X d S )Nr- ro Z computerNamerp rq r1 r1 r2 r� n s r� c C s2 z| d d d dkW S t k
r, Y d S X d S )Nr- ro Z disablePasswordAuthentication�truerp rq r1 r1 r2 r� u s ��r� c C sH d| � � krdS t�� }z|�| �}W n tk
r< Y dS X |jdk S )z?
Validate whether or not the key is OpenSSH-formatted.
z
FN)r6 r ZAuthKeyLineParser�parse� TypeErrorZkeytype)r� �parserZaklr1 r1 r2 r. s r. c C s` dD ]V}g }t d|�D ]6}| | t|� }tj�|�r|�|tj�|�f� q|r| S qg S )N)z-part�pry rD )r= r� r� r� r rG �realpath)r% ZmaxnumZsuff�foundZpnumZppathr1 r1 r2 �_partitions_on_device� s
ry c C s* t jddd�}t�d|� tj�| �|kS )Nz TYPE=ntfsT)Zno_cachezntfs_devices found = %s)r �find_devs_withro rp r� r� rw )r% Z ntfs_devicesr1 r1 r2 �_has_ntfs_filesystem� s r{ c C s |r dd� t�tf }d|fS tj�| �s6dd| fS t�d| tj�| �� t | �}t
|�dkrldd| fS t
|�dkr�d | d
� d d � |D ��f }d|fS t
|�dkr�|d
\}}n |d \}}t |�s�d||| f }d|fS t dd� �}d||| f }t
jddtd���}z tj||dddidd�} W n~ tjk
�r� }
z\d|_dt|
�k�rrd|d f W Y �6W 5 Q R � S d|d ||
f f W Y �W 5 Q R � S d }
~
X Y nX | dk�r�d | |_t�d d� t�t� d|d | fW 5 Q R � S W 5 Q R X d|d! fS )"a� Determine if the ephemeral drive at devpath should be reformatted.
A fresh ephemeral disk is formatted by Azure and will:
a.) have a partition table (dos or gpt)
b.) have 1 partition that is ntfs formatted, or
have 2 partitions with the second partition ntfs formatted.
(larger instances with >2TB ephemeral disk have gpt, and will
have a microsoft reserved partition as part 1. LP: #1686514)
c.) the ntfs partition will have no files other than possibly
'dataloss_warning_readme.txt'
User can indicate that NTFS should never be destroyed by setting
DS_CFG_KEY_PRESERVE_NTFS in dscfg.
If data is found on NTFS, user is warned to set DS_CFG_KEY_PRESERVE_NTFS
to make sure cloud-init does not accidentally wipe their data.
If cloud-init cannot mount the disk to check for data, destruction
will be allowed, unless the dscfg key is set.z:config says to never destroy NTFS (%s.%s), skipping checksrE Fzdevice %s does not existz Resolving realpath of %s -> %sr z device %s was not partitionedrA z&device %s had 3 or more partitions: %srs c S s g | ] }|d �qS )rD r1 )rg rv r1 r1 r2 ri � s z*can_dev_be_reformatted.<locals>.<listcomp>rD z5partition %s (%s) on device %s was not ntfs formattedc s( t ddg�� t� fdd�t�| �D ��S )Nzdataloss_warning_readme.txtzsystem volume informationc s g | ]}|� � � kr|�qS r1 )ra )rg �f�Zignoredr1 r2 ri � s z?can_dev_be_reformatted.<locals>.count_files.<locals>.<listcomp>)�setrJ r� �listdir)Zmpr1 r} r2 � count_files� s �z+can_dev_be_reformatted.<locals>.count_filesz1partition %s (%s) on device %s was ntfs formattedzmount-ntfs-and-countr� ZntfsZLANG�C)r� Zupdate_env_for_mountZ log_errorzcannot mount ntfsz unknown filesystem type 'ntfs'Tz^ but this system cannot mount NTFS, assuming there are no important files. Formatting allowed.z but mount of %s failed: %sNz mounted and counted %d fileszxit looks like you're using NTFS on the ephemeral disk, to ensure that filesystem does not get wiped, set %s.%s in configz but had %d files on it.z3 and had no important files. Safe for reformatting.)r` r� rX r� r� r ro rp rw ry rJ r{ r# r r� r" r r� r� r� r� rq ) r% rV r Z
partitionsZ cand_partZ cand_pathr� �bmsgZevtZ
file_countr r1 r1 r2 �can_dev_be_reformatted� s� �
� � �
���
�6
�&r� F)rY r% rU rV c
C s t j�|�s"td| tjd� d S td| tjd� d}d }|rJd\}}nt||�\}}t�d||� |snd S dD ]�}t j�| dd | �}d
||f }t j�|�r�zt �|� t�d |� W q� t k
r� } zt�
d || � W 5 d } ~ X Y q�X qrt�d
|� qrd S )Nz,Ephemeral resource disk '%s' does not exist.rn z$Ephemeral resource disk '%s' exists.F)TzFirst instance boot.zreformattable=%s: %s)r� ZmountsZsemZconfig_zMarker "%s" for module "%s"z %s removed.z%s: remove failed! (%s)z%s did not exist.) r� r� r r) ro rp r� r` �unlinkr� rq )
rY r% rU rV �resultr �modZsempathr� r r1 r1 r2 rW s: ��
"rW c C sj dd� }| sd S |si }t �| |� |�� D ]8\}}tj�| |�}d|krT|||�}t j||dd� q,d S )Nc S sh z>t �| �}|�� D ] }d|jkr|jtkrt|_qt �|�W S tk
rb t� d|� | Y S X dS )z>Azure provides the UserPassword in plain text. So we redact itZ UserPasswordz#failed to redact userpassword in %sN)
�ETZ
fromstring�iter�tag�text�DEF_PASSWD_REDACTIONZtostringr� ro Zcritical)Zcnt�fnamer� �elemr1 r1 r2 �_redact_password@ s
�� z%write_files.<locals>._redact_passwordr� i� )�filename�content�mode)r Z
ensure_dirrR r� r� r` rA )Zdatadirr� r r� r� r� r� r1 r1 r2 r$ > s
r$ c C s t �| �}i }i }|jpd}|jr,|j|d< |jr<|j|d< |jdk rT|j |d< n|jrbd|d< i }|jrv|j|d< |jr�d|d < t|jkr�t |j�|d
< |r�d |i|d < |j
|d
< t d|j
t j
d� |j|d< t d|j t j
d� |j|d< t d|j t j
d� |||fS )z�Parse OVF XML contents.
:return: Tuple of metadata, configuration, userdata dicts.
:raises NonAzureDataSource: if XML is not in Azure's format.
:raises errors.ReportableError: if XML is unparsable or invalid.
ry r� rS NZ
ssh_pwauthTr� FZ lock_passwdZ
hashed_passwdr� r� rP zPreprovisionedVm: %srn rO zPreprovisionedVMType: %sr� z ProvisionGuestProxyAgent: %s)r! Z
parse_textZ custom_datar� Z public_keysZdisable_ssh_password_auth�passwordr� r� � encrypt_passZpreprovisioned_vmr) ro r> Zpreprovisioned_vm_typeZprovision_guest_proxy_agent)r
Zovf_envr r� rR Zdefuserr1 r1 r2 rQ [ sJ
�
�
�rQ c C s t | �S r2 )r. )r� r1 r1 r2 r� � s r� c C sT z.t | �� }|�d� W 5 Q R � W dS Q R X W n tk
rN t�d| � Y nX dS )z;Return boolean indicating path to cdrom device has content.i Tz cdrom (%s) is not configuredF)�open�read�IOErrorro rp )� cdrom_dev�fpr1 r1 r2 �_check_freebsd_cdrom� s
r� c C s( | dkr dS t j| dd�}t�|��� S )zIReturn content random seed file if available, otherwise,
return None.NT)�quiet)r Zload_binary_filer Z b64encode�decode)�sourcer� r1 r1 r2 r � s
r c c sT | V t V t�� r(d}t|�rF|V n dD ]}t�d| �E d H q,|rP|V d S )Nz/dev/cd0)Ziso9660r� zTYPE=%s)� DEFAULT_PROVISIONING_ISO_DEVr r� r� rz )r� Z cache_dirr� Zfstyper1 r1 r2 r� � s r� c
C st t j�| d�}t j�|�s"td��t�d��$ t|d��}|�� }W 5 Q R X W 5 Q R X t |�\}}}|||d|ifS )Nr� zNo ovf-env file foundzReading ovf-env.xml�rb)
r� r� r` rK r r ZTimedr� r� rQ )Z
source_dirZovf_filer� r
r rR r� r1 r1 r2 r� � s r� )�network_metadatar� rU c C s� di d�}t | d �D �]j\}}d}dj|d�}d|d d i}|d
krPd|d < d |dd
�}dD ]�} |�| i ��dg �}
|
s�t�d| |� q`d }| dkr�d} nd} |
r�d |d< ||d< |s�q`|
dd� D ]N} || d d
�d| �}
| d }|�d�s�g |d< |d �dj||
d �� q�q`|�rr|�rrt|d �}|�d |�� i|d �� t |�}|�rd||d d!< ||d" |< qt�d#|||� q|S )$z�Convert imds network metadata dictionary to network v2 configuration.
:param: network_metadata: Dict of "network" key from instance metadata.
:return: Dictionary containing network version 2 standard configuration.
rA )�version� ethernetsr� Fzeth{idx})�idxz route-metricrD r� r zuse-dnsT)Zdhcp4zdhcp4-overrides�dhcp6)�ipv4Zipv6Z ipAddressz No %s addresses found for: %rr� Z24Z128r� zdhcp6-overridesNZsubnet�prefixZprivateIpAddress� addressesz
{ip}/{prefix})Zipr� rd Z
macaddress)�matchzset-namer� r� r� z<No configuration for: %s (dev_config=%r) (has_ip_address=%r))
� enumerater� r� ro rp rG rc r� ra rr )r� r� Z netconfigr� ZintfZhas_ip_addressZnicnameZ
dhcp_overrideZ
dev_configZ addr_typer� Zdefault_prefixZaddrZ netPrefixZ privateIprY r� r1 r1 r2 r^ � sj
� �
� � �r^ c C s t jdd�} | dkri S | S )ztGenerate fallback network config.
@return: Dictionary containing network version 2 standard configuration.
T)Z
config_driverN)r Zgenerate_fallback_configr3 r1 r1 r2 r_ & s r_ c C sh | st } d}| D ]R}tj�|�r|s@t�dtjjtj jg� d}tj�
|�rXt � |� qt �
|� qdS )a( Remove Azure-specific ubuntu network config for non-primary nics.
@param paths: List of networking scripts or directories to remove when
present.
In certain supported ubuntu images, static udev rules or netplan yaml
config is delivered in the base ubuntu image to support dhcp on any
additional interfaces which get attached by a customer at some point
after initial boot. Since the Azure datasource can now regenerate
network configuration as metadata reports these new devices, we no longer
want the udev rules or netplan's 90-hotplug-azure.yaml to configure
networking on eth1 or greater as it might collide with cloud-init's
configuration.
Remove the any existing extended network scripts if the datasource is
enabled to write network per-boot.
Fz|Removing Ubuntu extended network scripts because cloud-init updates Azure network configuration on the following events: %s.TN)� UBUNTU_EXTENDED_NETWORK_SCRIPTSr� r� r ro r> r rg r� Z BOOT_LEGACY�isdirr Zdel_dirrT )r� Zloggedr� r1 r1 r2 r 2 s � r c C s t �| t�S r2 )r
Zlist_from_depends� datasources)Zdependsr1 r1 r2 �get_datasource_listb s r� )N)r- )N)N)�r � functoolsZloggingr� �os.pathrG rk Zxml.etree.ElementTreeZetreeZ ElementTreer� �enumr �pathlibr r r r �typingr r r r
r Z cloudinitr r r
r r r Zcloudinit.eventr r Z
cloudinit.netr Zcloudinit.net.dhcpr r r Zcloudinit.net.ephemeralr r Zcloudinit.reportingr Zcloudinit.sources.azurer r r r Zcloudinit.sources.helpersr Z cloudinit.sources.helpers.azurer r r! r" r# r$ r% r& r' r( r) r* r+ Zcloudinit.url_helperr, Zcrypt�partialZrand_strr. �__annotations__r/ �AttributeErrorZ passlib.hashZpasslib�hashZ sha512_cryptZ getLoggerr7 ro ZDS_NAMEr# r! Z
DEFAULT_FSZAGENT_SEED_DIRr� r5 r? r� r� rP rS rX rc rk rr rx r{ r~ r r� r� Zres_diskrp r� r" r� rX r� r� r� r� r r� r� r. ry r{ r� r� rW r$ rQ r� r� r r� r� ri r^ r_ r ZDataSourceAzureNetZDEP_FILESYSTEMr� r� r1 r1 r1 r2 �<module> s < �
�
+
� ��
� 2
m��+
7
�P &
�