HOME


Mini Shell 1.0
DIR: /usr/lib/python3/dist-packages/cloudinit/sources/__pycache__/
Upload File :
Current File : //usr/lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceAzure.cpython-312.pyc
�

�fy#�
���UddlZddlZddlZddlZddlZddlZddlZddlmcm	Z
ddlmZddl
mZddlmZmZddlmZmZmZmZddlZddlmZmZmZmZmZddlmZmZddl m!Z!dd	l"m#Z#m$Z$m%Z%dd
l&m'Z'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/dd
l0m1Z1ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?ddl@mAZA	ddlBZBej�eBj�dej�d������ZEeeFd<ej�eN�ZOdZPddiZQdZRdZSdZTdZUGd�de�ZVdZWeeXeFd <gd!�ZYd"�ZZd#�Z[d$eXd%eXd&eeXfd'�Z\d(eXd&eXfd)�Z]e7d&eeXfd*��Z^e7d(eXd&eeXfd+��Z_d`d&eXfd,�Z`d-�Zad.�Zbd/�Zcd&eeXfd0�Zdej��r6d1ZSedd2�Zfef�eOj�d3�d4efzZRneOj�d5�dZWeTd6eRid7d7d8�Zhd6d9d:gd7d;�ieSd<d=�gd>�Zid?ePgZjd@ZkdAZldBZmGdC�dDej��ZodE�ZpdF�ZqdG�ZrdH�ZsdI�ZtdadJ�Zue7dK��Zve7dL��Zwe7eRdMdMfdNeXdOeXdPexdQexfdR��Zye7d`dS��Zze7dT��Z{dU�Z|e7dV��Z}e7eWfdW��Z~e7dX��Ze7dY��Z�e7dZe�d[exd&e�fd\��Z�e7d&e�fd]��Z�e7d`d^��Z�eoZ�eoe�jffgZ�d_�Z�y#eGeHf$r8	ddlIZJeJj�j�j�ZEn#eG$rd�ZEYnwxYwY���wxYw)b�N)�Enum)�Path)�sleep�time)�Any�Dict�List�Optional)�net�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$�)�strlen)�salt�
blowfish_hashc��td��)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 dependency)�ImportError)�_s �C/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceAzure.pyr/r/@s��
�A��
��Azure�instance-idziid-AZURE-NODEz/dev/disk/cloud/azure_resource�ext4z/var/lib/waagentz/dev/sr0c� �eZdZdZdZdZdZdZy)�PPSType�None�PreprovisionedOSDisk�Running�Savable�UnknownN)�__name__�
__module__�__qualname__�NONE�OS_DISK�RUNNING�SAVABLE�UNKNOWN�r4r3r9r9Ws���D�$�G��G��G��Gr4r9z/sys/firmware/acpi/tables/OEM0�PLATFORM_ENTROPY_SOURCE)z"/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�P�|j�D]�}tjd|�s�|j�}t	|�dk\s�9|djd�}t	|�dk\s�\|ddk(s�e|dj|�s�z|djd�}|dcSy	)
z�
    dev.storvsc.1.%pnpinfo:
        classid=32412632-86cb-44a2-9b5c-50d1417354f5
        deviceid=00000000-0001-8899-0000-000000000000
    �pnpinfo���=r�deviceid��.N)�
splitlines�re�search�split�len�
startswith)�
sysctl_outrN�line�fields�columns�compss      r3�"find_storvscid_from_sysctl_pnpinfor\ks����%�%�'�$��
�9�9�Z��&��Z�Z�\�F��6�{�a�� ��)�/�/�#�.����L�A�%���
�j�0���
�-�-�h�7�"�1�I�O�O�C�0�E� ��8�O�$�r4c��|j�D]/}tj||�s�|j�}|dcSy)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
    rN)rQrRrSrT)�camcontrol_out�disk_drvrX�itemss    r3�find_busdev_from_diskra�sC���)�)�+���
�9�9�X�t�$��J�J�L�E���8�O��r4r^�busdev�returnc���|j�D]S}tj||�s�|jd�}t	|�dk(s�:|djd�}|dcSy)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)
    �(rLrO�,rN)rQrRrSrTrU)r^rbrXr`�dev_passs     r3�find_dev_from_busdevrh�sc���)�)�+�#��
�9�9�V�T�"��J�J�s�O�E��5�z�Q�� ��8�>�>�#�.����{�"�#�r4�macc
��t|�dk(r)dj|dd|dd|dd|dd|dd|ddg�}|j�S)	z1Normalize mac address with colons and lower-case.��:rrL����
)rU�join�lower)ris r3�normalize_mac_addressrs�s]��
�3�x�2�~��h�h�
��1�X�s�1�Q�x��Q�q��3�q��8�S��2�Y��B�r�
�K�
���9�9�;�r4c�x�tj�D�cgc]}|ddk(rt|d���c}Scc}w)z-Get Hyper-V NICs as normalized MAC addresses.rL�	hv_netvscrO)r�get_interfacesrs)�ns r3�get_hv_netvsc_macs_normalizedrx�sB��
�#�#�%��
��Q�4�;��	�a��d�#����s�7c�,�tj�D�cgc]}|t|d�k(r|d��}}d|vryt|�dk(r&t	d|�d|��t
j��|dSt	d|�d|��t
j��y	cc}w)
z0Determine the device driver to match on, if any.rOrLruz'Assuming driver for interface with mac=z	 drivers=��logger_funcrz0Unable to specify driver for interface with mac=N)rrvrsrUr(�LOG�debug�warning)ri�i�driverss   r3�determine_device_driver_for_macr��s���
�#�#�%��
��'��!��-�-�	
�!���G��
�g���
�7�|�q����G�
��	�	�	
�
�q�z�����	��K�K��
��+s�Bc���	tj|�jS#tj$r*tj	ddj|��|cYSwxYw)NzFailed to execute: %s� )r�stdout�ProcessExecutionErrorr|r}rq)�cmd�fail_rets  r3�execute_or_debugr��sL����y�y��~�$�$�$���%�%���	�	�)�3�8�8�C�=�9����s�!�:A�Ac� �tddgd��S)N�sysctlzdev.storvsc�)r��r�rGr4r3�get_dev_storvsc_sysctlr��s���X�}�5��C�Cr4c��tgd��S)N)�
camcontrol�devlistz-br�rGr4r3�get_camcontrol_dev_busr��s���;�<�<r4c��tddg�S)Nr�r�r�rGr4r3�get_camcontrol_devr��s���\�9�5�6�6r4c��d}|dkDrd}|dz
}dt|�z}dj||�}t�}t||�}|stjd�yt
�}t�}dj|�}t||�}	|	r&t||	�}
|
�tjd	�y|
Sd
j|�}t||�}	|	r&t||	�}
|
�tjd	�y|
Sy)N�00000000rO�00000001rL�000z{0}-{1}z#Fail to find storvsc id from sysctlz	blkvsc{0}zFail to find /dev/daXz
storvsc{0})
�str�formatr�r\r|r}r�r�rarh)�port_id�g0�g1�g0g1rW�	storvscid�camcontrol_b_outr^�blkvsc�scbusx�devname�storvscs            r3�get_resource_disk_on_freebsdr��s��	�B���{�
���A�+��	��W��	�B����B��#�D�(�)�J�2�:�t�D�I���	�	�7�8��-�/��'�)�N�
�
�
�	�
*�F�
"�#3�V�
<�F�
�&�~�v�>���?��I�I�-�.�����!�!�)�,�G�
"�#3�W�
=�F�
�&�~�v�>���?��I�I�-�.����r4zfreebsd-ufsrOzresource disk is not None�/dev/zresource disk is None�
ephemeral0T)�data_dir�disk_aliases�apply_network_config�&apply_network_config_for_secondary_ips�gpt�d)�
table_type�layout�	overwritezephemeral0.1)�
filesystem�device)�
disk_setup�fs_setup�
datasource�never_destroy_ntfszTemporary Storage�REDACTEDc���eZdZdZej
ejejhiZ	dZ
ejZ
dZd�Zdeddf�fd�Zd	�Zd
�Zededefd��Zedd
ddd�deededededef
d��ZedId��Zdefd�Zed��Zededefd��ZdJ�fd�	Z ed��Z!ed��Z"d�Z#d�Z$ede%efd��Z&d e'de%efd!�Z(de%efd"�Z)d#�Z*d$�Z+dKd%�Z,ed&��Z-e	dLd'ed(ede.fd)��Z/ed*��Z0ed
dd+�d,ed-eddfd.��Z1ed/��Z2ede3jffd0��Z4ed1��Z5edId2��Z6ed3��Z7ed4��Z8ede9fd5��Z:e	dMd6e;jxd7edefd8��Z=edd9�d:ee%edee%efd;��Z>d e'deefd<�Z?d=e'd e'de@fd>�ZAed?��ZBed@e'd e'dee%efdA��ZCdB�ZDedC��ZEeFdD��ZGedE��ZHeFdF��ZIeFdG��ZJed e'defdH��ZK�xZLS)N�DataSourceAzurer5FrOc���tjj||||�tjj|jd�|_i|_d|_tjtj|ti�tg�|_d|_d|_d|_d|_d|_t(|_tjj|j,dd�|_y)N�azureF�data�reported_ready)r�
DataSource�__init__�os�pathrq�seed_dir�cfg�seedr�
mergemanydict�get_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�pathss    r3r�zDataSourceAzure.__init__Es������#�#�D�'�6�5�A������U�^�^�W�=��
������	��(�(�
�
!�
!�'�;��
;�=N�O�
�����
�#���#'�� �*/��'�05��-�$?��!�+-�7�7�<�<��O�O�V�%5�,
��(r4�ci_pkl_versionrcNc����t�|�|�d|_d|_d|_d|_t|_tjj|jjdd�|_
y)NFr�r�)�super�	_unpickler�r�r�r�rr�r�r�rqr�r�r�)r�r��	__class__s  �r3r�zDataSourceAzure._unpickleWs^���
���.�)�#'�� ���
�*/��'�05��-�$?��!�+-�7�7�<�<��J�J� � �&�*:�,
��(r4c�d�tjj|�}|�d|j�d�S)Nz [seed=�])rr��__str__r�)r��roots  r3r�zDataSourceAzure.__str__cs'���!�!�)�)�$�/��!%�t�y�y�1�1r4c���|j�d}n@|jjd�rd}n"|jj�dk(rd}nd}|�d|j�d�S)z/Return the subplatform metadata source details.�unknownz/devzconfig-diskrzseed-dir� (�))r�rVrr)r��subplatform_types  r3�_get_subplatformz DataSourceAzure._get_subplatformgsX���9�9��(��
�Y�Y�
!�
!�&�
)�,��
�Y�Y�_�_�
�&�
(�%��)��,�d�i�i�8�8r4�ephipv4c���|jsy|jD�cgc]}|djd�d��}}d|v|_|j|v|_|jxs|jScc}w)NFr�/z169.254.169.254)�
static_routesrTr�r�r�)r�r��r�routed_networkss    r3�_check_if_primaryz!DataSourceAzure._check_if_primaryss����$�$��7>�7L�7L�M�!�1�Q�4�:�:�c�?�1�-�M��M�+<��*N��'�
�%�%��8�	
�-�

�+�+�
5��4�4�	
��Ns�A2T�)�iface�report_failure_if_not_primary�retry_sleep�timeout_minutesr�r�r�r�c��|j�td��td|�dtj���t
j��t|j|t��|_d}t�}||dzz}tjdd	t�
�5|�=	|jj�}|�t�|z|krt=|�nn|��=|�d|_t-��|d}||j_d|vr
|d|_ tC|�}	|jjD}
|
�td��|jG|
�}td|�d|�d|	�d|
jH�d|
jJ�d|�d|jL�d |jN�d!�t
j��|rA|s?|j#t%jP||	|
jH|
jJ|�"�d
��|cddd�S#t$rQtdt
j ��|j#t%j&t�|z
��d
��Y���t($r$tdt
j*��d|_�t,$rUtd|zt
j*��|j#t%j.t�|z
|��d
��Y��"t0j2$rZ}td|j4�d|j6�d|j8�d|j:��t
j*��Yd}~���d}~wwxYw#1swYyxYw)#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.z,Bringing up ephemeral networking with iface=�: rz)r��
dhcp_log_func�<zobtain-dhcp-leasezobtain dhcp lease��name�description�parentzInterface not found for DHCP)�durationT��	host_onlyzdhclient executable not foundz&Failed to obtain DHCP lease (iface=%s))r��	interfacezCommand failed: cmd=z stderr=z stdout=z exit_code=r�zunknown-245zdhcp context missing ephipv4z!Obtained DHCP lease on interface z
 (primary=z driver=z router=z routes=z lease=z
 imds_routed=z wireserver_routed=r�)r��driver�routerr��lease))r��RuntimeErrorr(rrvr|r}rr�r$rr�ReportEventStackr!�obtain_leaserr~�_report_failurer�$ReportableErrorDhcpInterfaceNotFoundr�errorr�ReportableErrorDhcpLeaserr�r��stderrr��	exit_coderr�r�r�_ephipv4r�r�r�r�r��(ReportableErrorDhcpOnNonPrimaryInterface)r�r�r�r�r�r��
start_time�deadlinerr�r��primarys            r3�_setup_ephemeral_networkingz+DataSourceAzure._setup_ephemeral_networking�s.��*�#�#�/��A��
�	 ��c�(�(�*�
,��	�	�	
�
$3��K�K��%�$
�� ����V�
���"� 4�4��
�
$�
$�$�+�$�
�h	�
�-�,� �4�4�A�A�C�E�\�=�T�V�k�%9�H�%D��+�&��g�-�j�}�+/��(�&�(�(��+�&�E�-2�D�$�$�*���%�,1�-�,@��)�"�5�)�F��.�.�7�7�G���"�#A�B�B��,�,�W�5�G�#�
����N�N��)�)���3�3��9�9�� �I�I�
�"-�W��$�$��C�C�"'�%�&�~�~�&-�&;�&;�#��#�%�	��Qh	�h	��1�
�+�6�C�K�K���(�(��C�C�%)�V�j�%8��#'�	)��7��+�7�S�Y�Y��04�D�,��'��+�@�5�H�$'�I�I���(�(��7�7�%)�V�j�%8�E��#'�	)���1�1��+�"�I�I�!�L�L�!�L�L�!�O�O�
�%(�I�I�
�
����Mh	�h	�sX�M� G<�:!M�DM�<AM�M�BM�M�M�.AM�=M�M�M�Mc�|�d|_d|_|j�y|jj�d|_y)zTeardown ephemeral networking.FN)r�r�r��
clean_network�r�s r3�_teardown_ephemeral_networkingz.DataSourceAzure._teardown_ephemeral_networkings=��+0��'�05��-��#�#�+��� � �.�.�0�#'�� r4c�T�|jduxs|jjduS)z"Check if networking is configured.N)r�r�rs r3�_is_ephemeral_networking_upz+DataSourceAzure._is_ephemeral_networking_up&s5��
�$�$��,�
6��'�'�-�-��5�
�	
r4c	�0�i}|jd}d}ddi}ddddiii}d}i}t|j|�D]�}	|jd�r]t	j
�r"t	j|td	�
�\}}}}nt	j|t�\}}}}||_nt|�\}}}}|}td|ztj��nd}	t|	tj��t|j�xs|du}
|
rdnd}	|j!|��i}|j%�r|j'd��}|s$|�"d}	t|	�t)j*|	��|j-||�}
|
t.j0k7�r�t	j
�r2d}	t|	tj2��t)j*|	��|j%�s2d}	t|	tj2��t)j*|	��|
t.j4k(r|j7�nX|
t.j8k(r|j;�n4|
t.j<k(r|j?�n|jA�|jC�\}}}}|j'd��}|jE|�}|dt.j0jFfvr&|jItKjLd|���|jO|��|xsd|_(|jS||t	jT|d|ig�|d��tW|�}tY|�}t[|�}|r!tjd|�||ddd<|rtjd|�||d d<|rtjd!|�||d d"<|jPdk(r|d#s	t]|||�$�}d%|i|d#<|sXta|�}|rKtjd'�	tcjddjg|ji���|d(<||k(rtd*|ztj��tk�}|r||d d+<|jm�|d d,<|jnd-urh|j%�rX|jq||�.�}	|js|�/�}tjd0|�|r||d d1<|ju�|S|S#t$r"td
|ztj��Y��'tj$r"td|ztj��Y��ZwxYw#t"$rY��twxYw#t^$r)}td&|ztj��Yd}~���d}~wwxYw#t^$rtd)tj��Y���wxYw#t^$rY|SwxYw)2aWalk 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-hostnamer��system_info�default_userr�r��udf)�mtypez!Found provisioning metadata in %srzz$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�T)�report_failurezNo OVF or IMDS availablez%Free BSD is not supported for PPS VMszDHCP failed while in source PPSzextended.compute.ppsType)�key�value)�imds_md�IMDSr)r��files�metadata�userdata_rawz Username retrieved from IMDS: %sz Hostname retrieved from IMDS: %sr"z(Disable password retrieved from IMDS: %s�disable_passwordr!)�username�hostname�
disableSshPwd�ovf-env.xmlz)Failed to construct OVF from IMDS data %szRetrieved userdata from IMDSr#zBad userdata in IMDSzusing files cached in %s�random_seedr6F)r�r��pubkey_infoznegotiating returned %s�public-keys);r��list_possible_azure_dsr�rVr�
is_FreeBSD�mount_cb�load_azure_ds_dirr�r(r|r}r�MountFailedErrorr~�boolr
rr�get_metadata_from_imdsr�InvalidMetaDataException�_determine_pps_typer9rBrrD�_wait_for_pps_running_reuserE�_wait_for_pps_savable_reuserC�_wait_for_pps_os_disk_shutdown�_wait_for_pps_unknown_reuse�_reprovision�_ppstype_from_imdsrrr�"ReportableErrorImdsInvalidMetadata�validate_imds_network_metadatar��updater��_username_from_imds�_hostname_from_imds�_disable_password_from_imdsr#�	Exception�_userdata_from_imds�base64�	b64decoderqrT�_get_random_seed�_iid�_negotiated�!_determine_wireserver_pubkey_info�
_report_ready�_cleanup_markers)r��crawled_data�ddir�
ovf_source�mdr�r#r!�src�msg�requires_imds_metadatar�r�pps_type�imds_ppstype�
imds_username�
imds_hostname�imds_disable_password�contents�e�
imds_userdatar�r+�ssh_keyss                        r3�crawl_metadatazDataSourceAzure.crawl_metadata-s�����{�{�:�&���
���
#���~���|�<�=������)�$�-�-��>�'	B�C�
��>�>�'�*����(�7;�}�}��!2�%�8�4��L�#�u�8<�}�}��!2�8�4��L�#�u�%(�D�M�3D�S�3I�0�B��c�5� �
�'�7�*�D� #�	�	���+'	B�FC�
�

$�C�S�[�[�A�"&�d�m�m�!4�!J�
�d�8J�� 6�"�A��	��,�,�_�,�M����+�+�-��1�1��1�F�G��:�-�,�C�#�C�(��2�2�3�7�7��+�+�C��9���w�|�|�#���� �=��'�����C��6�6�s�;�;��3�3�5�7��'�����C��6�6�s�;�;��7�?�?�*��0�0�2��W�_�_�,��0�0�2��W�_�_�,��3�3�5��0�0�2�+/�+<�+<�+>�(�B��c�5��1�1��1�F�G� �2�2�7�;�L��D�'�,�,�*<�*<�#=�=��$�$��=�=�6�l���	
�+�+�G�+�<��(�&��	������ �.�.��V�W�4E�/F�G� ,�	
�	
�,�G�4�
�+�G�4�
� ;�G� D����I�I�8�-�H�9F�C�
��~�.�v�6���I�I�8�-�H�9F�L��$�%5�6� ��I�I�:�%�
�&�
��$�"�
��9�9���|�G�'<�
�,�*�*�"7���
*7��(A��W�%��/��8�M���	�	�8�9��39�3C�3C����
� 3� 3� 5�6�4�L��0����#�*�T�1�s�y�y�
� �!���6:�L��$�]�3�26�)�)�+��Z� ��/����u�$��)I�)I�)K��@�@���A��K�

(��-�-�+�-�F��
�	�	�3�X�>��>F�L��,�]�;��%�%�'���|���E&�
�'�:�S�@� #�	�	����(�(�
�'�*�S�0�c�i�i���	
��, �	��	��l�
�'�?�!�C� #�	�	����
�� !��+�.�C�K�K�����,�
����
�sf�BT7�V�V(�=5W�1X�7'V�!0V�V�	V%�$V%�(	W�1W�W�$X�X�	X�Xrc�4�t�}|dz}|jsd}nd}d}d}	tj||��S#t$rk}t|�}t�|z
}t
j||��}|js&t|jtj�rd}Yd}~n:d}~wt$r+}t|�}t
j|��}Yd}~nd}~wwxYw|j||��td|zt j"�	�iS)
Ni,�)�max_connection_errors�retry_deadline)�	exceptionr�F)rar�z!Ignoring IMDS metadata due to: %srz)rr�r� fetch_metadata_with_api_fallbackr+r�r�ReportableErrorImdsUrlError�
isinstance�cause�requests�ConnectionError�
ValueError�+ReportableErrorImdsMetadataParsingExceptionrr(r|r~)	r�rr
r`r_�error_string�error_reportrr�s	         r3r3z&DataSourceAzure.get_metadata_from_imds�s���V�
�#�c�)���.�.�$&�!�$(�!�&*��9=��	��8�8�&;�-��
���
	'��u�:�L��v�
�*�H�!�=�=��(��L��2�2�z����X�5�5�8�"'�����	��u�:�L�!�M�M���L��	��	
���\��5G��H��/�,�>����	
��	s"�=�	C#�A!B,�,C#�8!C�C#c�V��tt|�|�tj|_y)z.Reset any cached class attributes to defaults.N)r�r��clear_cached_attrsr�UNSET�_metadata_imds)r��
attr_defaultsr�s  �r3rmz"DataSourceAzure.clear_cached_attrs-s���
�o�t�7�
�F�%�m�m��r4c��tjj�}|�y|j�yt	|jd�j�S)zQCheck platform environment to report if this datasource may
        run.
        TFr()r�ChassisAssetTag�query_systemr�r�exists)r��chassis_tags  r3�	ds_detectzDataSourceAzure.ds_detect2sJ��
�.�.�;�;�=���"���=�=� ���D�M�M�=�1�8�8�:�:r4c��	t�	t	�	tjtjd|j��}	|j�|jr>|jjdk(r%|j j#d�r
t%�t&}t(j*j-|�rBt/d	|ztj�
�tj0|dt2g�|_n(t/d|ztj�
�|d|_|d
d|_tj0|d
t8g�|_|d|_t?|j d|dd��y#t$r!}tjd|�Yd}~���d}~wwxYw#t$r!}tjd|�Yd}~���d}~wwxYw#tj$r+}|j|�Yd}~|j�yd}~wt$r@}tj|�}|j|�Yd}~|j�yd}~wwxYw#|j�wxYw)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: %szCrawl of metadata service)�logfuncrQ�funcF�ubuntur�zXEphemeral resource disk '%s' exists. Merging default Azure cloud ephemeral disk configs.rzr�zdEphemeral resource disk '%s' does not exist. Not merging default Azure cloud ephemeral disk configs.r"rr#r�r!i�)�dirmodeT) r%rBr|r~r'r�log_timer}r\r�ReportableErrorrr�!ReportableErrorUnhandledExceptionr�r�r��get�*maybe_remove_ubuntu_network_config_scripts�RESOURCE_DISK_PATHr�r�rtr(r��#BUILTIN_CLOUD_EPHEMERAL_DISK_CONFIGr�ro�DEFAULT_METADATAr"r#�write_files)r�rYrLr�reportable_error�devpaths      r3�	_get_datazDataSourceAzure._get_dataAs4��	?�� �	C���	2��=�=��	�	�/��(�(��L�
�/�/�1�
�K�K���� � �H�,������ 6�7�6�8�%��
�7�7�>�>�'�"�#�F��� �I�I�	
��)�)��e�$�&I�J��D�H�
$�J��� �I�I�	
�$�E�*�D�H�*�:�6�v�>����*�*�
�*�
%�'7�8�
��
�)��8���	��K�K�
�#�\�'�%:�E�	
���E�	?��K�K�:�A�>�>��	?��
�	C��K�K�>��B�B��	C���%�%�	�� � ��'��
�/�/�1���	�%�G�G��N��� � �!1�2���/�/�1��	��

�/�/�1�si�
F�
F8�0G%�	F5�F0�0F5�8	G"�G�G"�%I*�8H�	I-�I*�*&I%�I-�%I*�*I-�-I?c��|jrd|jvr|j�St|jd�S)Nr6)r"rGr�rs r3�get_instance_idzDataSourceAzure.get_instance_id�s4���}�}�
�T�]�]� B��9�9�;���4�=�=��/�0�0r4c�>�|jdj|�S)Nr�)r�r)r�r�s  r3�device_name_to_devicez%DataSourceAzure.device_name_to_device�s���{�{�>�*�.�.�t�4�4r4c��	|j|jd�S#ttf$rY|j	�SwxYw)z+
        Retrieve public SSH keys.
        r)�_get_public_keys_from_imdsr"�KeyErrorrh�_get_public_keys_from_ovfrs r3�get_public_ssh_keysz#DataSourceAzure.get_public_ssh_keys�sG��
	��2�2�4�=�=��3H�I�I���*�%�	���-�-�/�/�	�s� �A�Arc�v�	|ddD�cgc]}|d��	}}t	d�|D��r(d}t|tj��t|��dj
t|��}t|tj��|Scc}w#t$rd}t|tj���wxYw)	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.
        �compute�
publicKeys�keyDataz"No SSH keys found in IMDS metadatarzc3�6K�|]}t|�����y�w))rN)�_key_is_openssh_formatted)�.0rs  r3�	<genexpr>z=DataSourceAzure._get_public_keys_from_imds.<locals>.<genexpr>�s����J�#�,��5�5�J�s�zKey(s) not in OpenSSH formatzRetrieved {} keys from IMDS)r�r(r|r}�anyrhr�rU)r�r�
public_keyr[�log_msgs     r3r�z*DataSourceAzure._get_public_keys_from_imds�s���	�#*�)�"4�\�"B����9�%��H���J��J�J�4�G�#�G����C��W�%�%�/�6�6�s�8�}�E����S�Y�Y�?����!���	�:�G�#�G����C��	�s�
B�B�B�B�(B8c���g}	|jd}djt|��}t|tj
��|S#t$r!d}t|tj
��Y|SwxYw)zYGet SSH keys that were fetched from wireserver.

        :returns: List of keys.
        r,zRetrieved {} keys from OVFrzzNo keys available from OVF)r"r�rUr(r|r}r�)r�r[r�s   r3r�z)DataSourceAzure._get_public_keys_from_ovf�sq��
��	D��}�}�]�3�H�2�9�9�#�h�-�H�G�#�G����C�
���	�	D�2�G�#�G����C���		D�s�AA
�
&A4�3A4c��|jS�N�r�rs r3�get_config_objzDataSourceAzure.get_config_obj�s���x�x�r4c�H�tj|j��Sr�)r�instance_id_matches_system_uuidr�)r�r�s  r3�check_instance_idz!DataSourceAzure.check_instance_id�s���6�6�t�7K�7K�7M�N�Nr4c�z�tjj|jj	d�d�}tj�}tjj|�rNtj|�j�}tj|�}|j�||fvr|S|S)Nr�r6)
r�r�rqr��	get_cpathr�query_system_uuidrtr�load_text_file�strip�byte_swap_system_uuidrr)r��previous�
prev_iid_path�system_uuid�
swapped_ids     r3rGzDataSourceAzure._iid�s���������J�J� � ��(�-�
�
��0�0�2��
�7�7�>�>�-�(��*�*�=�9�?�?�A�H�!�7�7��D�J��~�~��K��#<�<����r4c�~�	d}tjddt��5tj|�}ddd�|�d}t|tj��ytd|ztj��y#1swY�HxYw#t$r*}tt|�tj���d}~wwxYw)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.rzz%The preprovisioned nic %s is detached)rrr!r�wait_for_nic_detach_eventr(r|r~r}�AssertionErrorr�r)r��nl_sock�ifnamerQrs     r3�_wait_for_nic_detachz$DataSourceAzure._wait_for_nic_detach�s���	��F��(�(�*�1�(��
D�
!�:�:�7�C��
D��~�2��(�����E�'�;�f�D� #�	�	��
D�
D��"�	�#�C��J�C�I�I�F���	�s3�B	�A=�'B	�B	�=B�B	�		B<�%B7�7B<r��retriesc��t|�D][}|jjj|�r t	d|zt
j��y|dz|ks�Qt|��]t	d|||zfzt
j��y)NzThe link %s is up.rzrOz;The link %s is not up after %f seconds, continuing anyways.)�ranger��
networking�try_set_link_upr(r|�infor)r�r�r�r�rs     r3�wait_for_link_upz DataSourceAzure.wait_for_link_ups����w��	�A��{�{�%�%�5�5�f�=�'�(�6�1�s�x�x����A��� ��k�"�	�
$�M��7�[�0�1�2��H�H�
r4c��|j}tjd|�tj|djt
j�t����tdtj��y)Nz*Creating a marker file to report ready: %sz{pid}: {time}
)�pidrzRSuccessfully created reported ready marker file while in the preprovisioning pool.rz)r�r|r�r�
write_filer�r��getpidrr(r})r�r�s  r3�_create_report_ready_markerz+DataSourceAzure._create_report_ready_markers\���/�/�����=�t�D�����#�*�*�r�y�y�{���*�H�	
�	 �
1��	�	�	
r4)�
create_marker�expect_url_errorr�r�c�V�	|j�d|_|r|j�yy#t$rr}|r3t|t�r#t	dt
j��d|_n3d}t	|t
j��tj|�|�Yd}~��d}~wwxYw)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.rzNz9Failed reporting ready while in the preprovisioning pool.F)
rJrBrdr+r(r|r}r�rrr4rHr�)r�r�r�rrQs     r3�_report_ready_for_ppsz%DataSourceAzure._report_ready_for_pps&s���	G���� �,!�����,�,�.���/�	G� �J�u�h�$?�'�B� #�	�	��
!%��
�P��(�����C��6�6�s�;��F�
��	G�s�-�	B(�A(B#�#B(c��tjd�	g}d}	d}tjddt	|�zt
��5t
j||�}ddd�|j|�td|ztj��|j|�|s*tjd	|�|j|d
d��}|rtjd�y|j���#1swY��xYw#t$r.}tt|�tj��Yd}~yd}~wwxYw)
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.rzz!Checking if %s is the primary nicr)r�r�r�zFound primary nic for this VM.)r|r�rrrUr!r�wait_for_nic_attach_event�appendr(r�r
rr�r�r)r�r��
nics_found�primary_nic_foundr�rs      r3�"_wait_for_hot_attached_primary_nicz2DataSourceAzure._wait_for_hot_attached_primary_nicMs7��	���=�>�3	G��J� %������,�,�.�N��j�/�*�-�
�
�%�>�>����F�
��!�!�&�)�'�/�&�8�c�h�h���%�%�f�-�
)��H�H�@�&�I�(,�(H�(H�$�(*�6;�)I�)�%�%��H�H�=�>���7�7�9�Q�
�
��P�	G�#�C��J�C�I�I�F�F��	G�s6�/D�C:�BD�)D�:D�?D�	D=�$D8�8D=c��	tj�S#tj$r$}td|��tj
���d}~wwxYw)Nz!Failed to create netlink socket: rz)r�create_bound_netlink_socket�NetlinkCreateSocketErrorr(r|r)r�rs  r3�_create_bound_netlink_socketz,DataSourceAzure._create_bound_netlink_socket�sJ��	��6�6�8�8���/�/�	�#�3�E�7�;��I�I�
�
��	�s��A
�A�A
c��|jd��tdtj��t	d�tj��)z4Report ready and wait for host to initiate shutdown.F)r�z"Waiting for host to shutdown VM...rzi�3�)r�r(r|r�rr�ReportableErrorOsDiskPpsFailurers r3r8z.DataSourceAzure._wait_for_pps_os_disk_shutdown�s?��	
�"�"��"�7��0����	
�	�h���4�4�6�6r4c�Z�|j�}	|j�|jj�td��|jj}|j	�t
j
d|�tjddt��5	tj||�ddd�|j�|j!�y#t$r(}td|zt
j��Yd}~�Ud}~wwxYw#1swY�^xYw#|j�wxYw)	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: %srz)r�r�r�r�r�r|r}rrr!r�!wait_for_media_disconnect_connectr�r(r�closer)r�r�r�rYs    r3r6z+DataSourceAzure._wait_for_pps_running_reuse�s���3�3�5��	��(�(�0��+�+�1�1�9�"�#>�?�?��,�,�2�2�E��&�&�(��I�I�5��
��(�(�8�2�(��
�
��=�=�g�u�M�

�
�M�M�O�	
�+�+�-��&��+�A�A�E�$'�I�I������
�
��
�M�M�O�sH�BD�D�C�/D�	D	�!D�?D�D	�	D�D�D�D*c��|j�}	|jd��	|j�|j|�|j|�|j�y#tj$r/}td|ztj��d|_Yd}~�pd}~wwxYw#|j�wxYw)z@Report ready and wait for nic(s) to be hot-attached upon re-use.T)r�zLIgnoring failure while tearing down networking, NIC was likely unplugged: %rrzN)r�r�rrr�r(r|r�r�r�r�r�)r�r�rYs   r3r7z+DataSourceAzure._wait_for_pps_savable_reuse�s����3�3�5��	��&�&��&�=�
0��3�3�5�
�%�%�g�.��3�3�G�<��M�M�O���-�-�
0�'�3�56�7� #����
,0��(�(��

0��
�M�M�O�s3�B-�A(�"B-�(B*�;%B%� B-�%B*�*B-�-B?c��tjj|j�ry|j	�|j�y)z0Report ready if needed for unknown/recovery PPS.N)r�r��isfiler�r�rrs r3r9z+DataSourceAzure._wait_for_pps_unknown_reuse�s7���7�7�>�>�$�:�:�;���"�"�$�	
�+�+�-r4c��d}d}|sf|j�s|dz
}	|jd��tjddt
��5	t
j�}	ddd�|s�ftd	|ztj�
�|S#t$rY��wxYw#t$r|j�Yddd���wxYw#1swY�bxYw)z0Poll IMDs for reprovisiondata XML document data.rNrOr�rzget-reprovision-data-from-imdszget reprovision data from imdsr�z#attempted dhcp %d times after reuserz)
rr
rrrr!r�fetch_reprovision_datar+rr(r|r})r��
dhcp_attempts�reprovision_datas   r3�
_poll_imdszDataSourceAzure._poll_imds�s����
���"��3�3�5���"�
���4�4�Q�4�G��(�(�5�<�(��	
�
�'+�'B�'B�'D�$�
	
�#�&	 �1�M�A��	�	�	
� ���%(����� ���7�7�9��	
�	
���	
�	
�s;�B
�C�
B�
	B�B�C�5C�?C�C�Crr�c���td|j���tj��t	�tj|�}|r|S|j�r;	tdtj��t|j|��d|_y	tdtj��|j�	|jd	�
�t|j|��d|_y#t$r(}td|ztj��Yd}~��d}~wwxYw#t$rY�^wxYw#t$r(}td|ztj��Yd}~yd}~wwxYw)
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: rzz>Using cached ephemeral dhcp context to report failure to Azure)�endpointrTz@Failed to report failure using cached ephemeral dhcp context: %sNz3Using new ephemeral dhcp to report failure to Azurerrz5Failed to report failure using new ephemeral dhcp: %sF)r(�as_encoded_reportr|rr)r�report_failure_to_hostrr}r*r�rHrBrr
r)r�rr��reportedrYs     r3rzDataSourceAzure._report_failuresc��	 �1�%�2I�2I�2K�1L�M��	�	�	
�	���-�-�e�4����O��+�+�-�
�'�1� #�	�	��
)�!�6�6�e��$(�� ��	�#�E��I�I�
�
�/�/�1�
��0�0��0�D�
%��2�2�%�
� $�D����-�
�'�8�:;�<� #�	�	����
��$�
��
���	�#�G�!�K��I�I�
�
�
��
	�sT�!9C9�+D<�D-�D<�9	D*�D%�%D*�-	D9�6D<�8D9�9D<�<	E-�E(�(E-r*r+c��t�tj�	t|j|j
|j|��}d|_d|_|S#t$r$}td|ztj���d}~wwxYw)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.
        )r�r��iso_devr+zQError communicating with Azure fabric; You may experience connectivity issues: %srzNT)r)r�report_success_to_hostr&r�r�r�rBr(r|r~rH)r�r+r�rYs    r3rJzDataSourceAzure._report_readyDs���	���"�"�$�
	�+��2�2��{�{��
�
�'�	�D���
��������	�#�*�,-�.��K�K�
�

��
	�s�-A�	B
�&B�B
c��	|dddS#t$r(}td|ztj��Yd}~yd}~wwxYw)N�extendedr��ppsTypez2Could not retrieve pps configuration from IMDS: %srz)rBr(r|r})r�rrYs   r3r;z"DataSourceAzure._ppstype_from_imdsgsH��	��:�&�y�1�)�<�<���	�#�D�q�H��I�I�
���	�s�

�	>�9�>�ovf_cfgc��tjj|j�rtj
}�na|j
dd�tjjk(s,|j|�tjjk(rtj}n�|j
dd�tjjk(s,|j|�tjjk(rtj}n�|j
d�dusY|j
dd�tjjk(s,|j|�tjjk(rtj}ntj}td|jztj��|S)z@Determine PPS type using OVF, IMDS data, and reprovision marker.�PreprovisionedVMTypeN�PreprovisionedVmTzPPS type: %srz)r�r�r�r�r9rFrrErr;rCrDrBr(r|r�)r�r�rrSs    r3r5z#DataSourceAzure._determine_pps_typeqs,��
�7�7�>�>�$�:�:�;����H��K�K�.��5����9N�9N�N��&�&�w�/�7�?�?�3H�3H�H����H��K�K�.��5����9N�9N�N��&�&�w�/�7�?�?�3H�3H�H����H��K�K�*�+�t�3��{�{�1�4�8����$�$�%��&�&�w�/�7�?�?�3H�3H�H����H��|�|�H���X�^�^�+����	
��r4c��|j�}tjddt��5t	|�\}}}|||d|ifcddd�S#1swYyxYw)zrInitiate the reprovisioning workflow.

        Ephemeral networking is up upon successful reprovisioning.
        zreprovisioning-read-azure-ovfz$read azure ovf during reprovisioningr�r(N)r�rrr!�read_azure_ovf)r�rXrO�udr�s     r3r:zDataSourceAzure._reprovision�s_���?�?�$��
�
$�
$�0�>�$�
�	<�
)��2�K�B��C���C�-��!:�;�
	<�	<�	<�s�A�Ar�c���d}	|j|�|S#ttf$rO|jdd�}dj	|�t|�nd�}t
|tj��Y|SwxYw)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 OVFrrz)	r�r�rhrr�rUr(r|r})r�r�rr+r�s     r3rIz1DataSourceAzure._determine_wireserver_pubkey_info�s���,0��	D��+�+�G�4����
�*�%�	D��'�'�*�d�3�K�:�A�A�$/�$;��K� ���G�
$�G����C���
	D�s��AA5�4A5c�B�tj|j�y)zCleanup any marker files.N)r�del_filer�rs r3rKz DataSourceAzure._cleanup_markers�s���
�
�d�6�6�7r4c���|jj�}	t|||jj	t
d���t
�y#t
�wxYw)NF)�is_new_instance�
preserve_ntfs)r��
get_ipath_cur�address_ephemeral_resizer�r�DS_CFG_KEY_PRESERVE_NTFSr))r�r�r��instance_dirs    r3�activatezDataSourceAzure.activate�sO���z�z�/�/�1��	"�$�� /�"�k�k�o�o�.F��N�
�
 �!���
 �!�s�,A�Ac�x�|jjdi�jdi�jd�S)Nrr��platformFaultDomain�r"rrs r3�availability_zonez!DataSourceAzure.availability_zone�s4��
�M�M���f�b�)�
�S��B�
�
�S�&�
'�	
r4c���|jrl|jtjk7rO|jj	d�r4	t|jd|jj	d���S	t�S#t$r)}tjdt|��Yd}~�8d}~wwxYw#t$r*}tjdt|��Yd}~iSd}~wwxYw)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)rorrnr�r�6generate_network_config_from_instance_network_metadatarBr|rr��-_generate_network_config_from_fallback_config)r�rYs  r3�_generate_network_configz(DataSourceAzure._generate_network_config�s���

����#�#�w�}�}�4������ 6�7�
�M��'�'�	�2�;?�;�;�?�?�@�<���	O�@�B�B���
��	�	�5���F����
���	O��I�I�E�s�1�v�N�N��	��	O�s0�2B�:	B9�	B6�
B1�1B6�9	C,�C'�'C,c��|jr)|jtjk7r|jS|j�|_|jS)z,Provide network configuration v2 dictionary.)r�rrnrrs r3�network_configzDataSourceAzure.network_config�sG�����D�$8�$8�G�M�M�$I��'�'�'�#�<�<�>����#�#�#r4c�x�|jjdi�jdi�jd�S)Nrr��locationr�rs r3�regionzDataSourceAzure.region�s1���}�}� � ���,�0�0��B�?�C�C�J�O�Or4c�z�t�}	|d}|dD�cgc]}t|d���}}|D�cgc]	}||vs�|��}}|sytd|�d	|��t
j��|jr|jjsytj|jj�}|rt|t�syt|�}||vr!td
|�d	|��t
j��ycc}w#t$r0td|j	d�zt
j��YywxYwcc}w)z=Validate IMDS network config and report telemetry for errors.r�r��
macAddressz6IMDS network metadata has incomplete configuration: %rrzFTz8IMDS network metadata is missing configuration for NICs r�z-IMDS network metadata is missing primary NIC )
rxrsr�r(rr|r~r�r�r�get_interface_macrdr�)	r�r�
local_macsrr�	imds_macs�m�missing_macs�primary_macs	         r3r=z.DataSourceAzure.validate_imds_network_metadata�s>��3�4�
�	�$�Y�/�N�(��4���&�a��o�6��I��$.�D�a��)�1C��D��D�����^�
-����	
��'�'�t�/G�/G�/M�/M���+�+�D�,D�,D�,J�,J�K���*�[�#�">��+�K�8���,�&�#���0��K�K�
���O���	�#�H��+�+�i�(�)��K�K�
�
�
	��Es+�C<�C7�C<�	D8�D8�7C<�<6D5�4D5)rcN)rGr�)r�g�������?)F)Mr?r@rA�dsnamer�NETWORKr�BOOT_NEW_INSTANCE�BOOT�default_update_eventsrHrrnro�_ci_pkl_versionr��intr�r�r�r"rr2r�r
r�r
rrr\rr3rmrvr�r�r�r	r��dictr�r�r�r�rGr��floatr�r�r�r��socketr�r8r6r7r9�bytesr�rr}rrJr;r9r5r:rIrKr��propertyr�rrrr=�
__classcell__)r�s@r3r�r�9s}���
�F�����'�'��N�N�
����K��]�]�N��O�
�$

��

��

�2�
9�!�
�)=�
�$�
�!�
�*!� $�.2�� �
O���}�O�(,�	O�
�O��
O�
�O�!�O�b!�(�!�(�
�T�
�!�O�!�O�b!�+�T�+�d�+�!�+�Z,�
!�;�!�;�!�J�!�J�X1�
5�!�	0�T�#�Y�	0�!�	0��$��4��9��6�4��9�� �O��"!��!��>!�DG����$'��<A��!��&!�

�!�

�!�#�!&�	$/��$/��	$/�

�$/�!�$/�L!�6G�!�6G�p!��f�m�m��!��!�	7�!�	7�!�".�!�".�H!��!��*!�	.�!�	.�!� �E� �!� �:!�?D�?��+�+�?�8<�?�	
�?�!�?�B!�48� �&�t�C�y�1� �	�$�s�)�	� �!� �D�$��8�C�=���4��$��7��:!�<�!�<�!����%)��	�$�s�)�	��!��$8�!�
�!�
��
��
�!��!��<�$��$��P��P�!�-�d�-�t�-�!�-r4r�c�8�	|dddS#t$rYywxYw)Nr��	osProfile�
adminUsername�r���	imds_datas r3r?r?'s.�����#�K�0��A�A��������

�	�c�2�	|ddS#t$rYywxYw)Nr��userDatarr s r3rCrC.s)�����#�J�/�/������s�
�	�c�8�	|dddS#t$rYywxYw)Nr�r�computerNamerr s r3r@r@5s.�����#�K�0��@�@������r"c�>�	|ddddk(S#t$rYywxYw)Nr�r�disablePasswordAuthentication�truerr s r3rArA<s;����i� ��-�.M�N��
�	
������s�
�	�c��d|j�vrytj�}	|j|�}|j
duS#t$rYywxYw)z?
    Validate whether or not the key is OpenSSH-formatted.
    z
FN)r�r
�AuthKeyLineParser�parse�	TypeError�keytype)r�parser�akls   r3r�r�Fs[��
�������
�
'�
'�
)�F���l�l�3����;�;�d�"�"������s�A�	A�Ac��dD]|}g}td|�D]c}||zt|�z}tjj	|�s�4|j|tjj
|�f��e|s�z|cSgS)N)z-part�pr�rO)r�r�r�r�rtr��realpath)r��maxnum�suff�found�pnum�ppaths      r3�_partitions_on_devicer9Ws��"������!�V�$�	>�D��d�N�S��Y�.�E��w�w�~�~�e�$����d�B�G�G�$4�$4�U�$;�<�=�	>���L���Ir4c��tjdd��}tjd|�tj
j
|�|vS)Nz	TYPE=ntfsT)�no_cachezntfs_devices found = %s)r�find_devs_withr|r}r�r�r3)r��ntfs_devicess  r3�_has_ntfs_filesystemr>ds=���&�&�{�T�B�L��I�I�'��6�
�7�7���G�$��4�4r4c
�n�|r$ddjt��dt�d�}d|fStjj|�sdd|zfStjd|tjj|��t|�}t|�dk(rdd|zfSt|�d	kDr,d
|�ddj|D�cgc]}|d
��	c}���}d|fSt|�d	k(r	|d
\}}n|d\}}t|�sd|�d|�d|�d�}d|fStd��}d|�d|�d|�d�}tjddt��5}		t!j"||dddid��}
|
dk7rKd!|
z|	_tj+d"djt�t�d|d#|
zzfcddd�S	ddd�d|d$zfScc}w#t j$$rI}d|	_dt)|�vrd|dzfcYd}~cddd�Sd|d|�d |��zfcYd}~cddd�Sd}~wwxYw#1swY�txYw)%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 (rPz), skipping checksFzdevice %s does not existzResolving realpath of %s -> %srzdevice %s was not partitionedrLzdevice z had 3 or more partitions: r�rOz
partition r�z) on device z was not ntfs formattedc��tddg�}ttj|�D�cgc]}|j	�|vs�|��c}�Scc}w)Nzdataloss_warning_readme.txtzsystem volume information)�setrUr��listdirrr)�mp�ignored�fs   r3�count_filesz+can_dev_be_reformatted.<locals>.count_files�sE���
*�,G�H�
���r�z�z�"�~�J�!�����'�1I�A�J�K�K��Js�A
�A
z was ntfs formattedzmount-ntfs-and-countr��ntfs�LANG�C)r�update_env_for_mount�	log_errorzcannot mount ntfszunknown filesystem type 'ntfs'Tz^ but this system cannot mount NTFS, assuming there are no important files. Formatting allowed.Nz but mount of z	 failed: zmounted 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.)rqr�r�r�r�rtr|r}r3r9rUr>r"rrr!rr/r1r�r�r~)r�r�rQ�
partitionsr2�	cand_part�	cand_pathrF�bmsg�evt�
file_countrYs            r3�can_dev_be_reformattedrRks���&��H�H�[�!�$�
���c�z��
�7�7�>�>�'�"��0�7�:�:�:��I�I�(�'�2�7�7�3C�3C�G�3L��'�w�/�J�
�:��!���5��?�?�?�	�Z��1�	���H�H�J�/�q�a��d�/�0�
���c�z��	�Z��A�	�)�!�}��	�9�)�!�}��	�9��	�*����
��
�c�z�� �L�!�L�	����D�
�	 �	 �
#�*� �
�#I�

�	P�������&,�c�]���J�(��?�<�z�I�C�O��K�K�'�����%�(�
��$�!;�j�!H�H�H�G#I�#I�4�5#I�J��M�M�M�M��E0��T�$�$�	P�1�C�O�/�3�q�6�9��� /�/���!#I�#I�0�$�)�Q�!O�O�O�O�1#I�#I��	P��#I�#I�sU�G�	H+�G�'AH+�H(�H#�9H(�:H+�H#�H(�H+�#H(�(H+�+H4Fr�r�r�r�c��tjj|�std|ztj
��ytd|ztj
��d}d}|rd\}}nt
||�\}}tjd||�|sydD]�}tjj|dd	|z�}d
|�d|�d�}tjj|�r-	tj|�tjd
|��|tjd|���y#t$r!}	tjd||	�Yd}	~	��d}	~	wwxYw)Nz,Ephemeral resource disk '%s' does not exist.rzz$Ephemeral resource disk '%s' exists.F)TzFirst instance boot.zreformattable=%s: %s)r��mounts�sem�config_zMarker "z" for module "�"z%s removed.z%s: remove failed! (%s)z%s did not exist.)r�r�rtr(r|r}rRrq�unlink�FileNotFoundErrorr~)
r�r�r�r��resultrQ�mod�sempathrOrYs
          r3r�r��s'���7�7�>�>�'�"��:�W�D��	�	�	
�	��2�W�<��	�	�	
�
�F�
�C��4����,�W�m�D�����I�I�$�f�c�2���'�
1���'�'�,�,�|�U�I��O�D��07��=��
�7�7�>�>�'�"�
@��	�	�'�"��	�	�-��.�
�I�I�)�4�0�
1���	%�
@����5�t�Q�?�?��
@�s�$+D)�)	E�2E�Ec��d�}|sy|si}tj||�|j�D]J\}}tjj||�}d|vr	|||�}tj||d���Ly)Nc�,�	tj|�}|j�D]0}d|jvs�|jt
k7s�&t
|_�2tj|�S#t$rtjd|�|cYSwxYw)z>Azure provides the UserPassword in plain text. So we redact it�UserPasswordz#failed to redact userpassword in %s)
�ET�
fromstring�iter�tag�text�DEF_PASSWD_REDACTION�tostringrBr|�critical)�cnt�fnamer��elems    r3�_redact_passwordz%write_files.<locals>._redact_passwords~��	��=�=��%�D��	�	��
5��"�d�h�h�.��	�	�%9�9� 4�D�I�
5��;�;�t�$�$���	��L�L�>��F��J�	�s�6A/�A/�
!A/�/!B�Br(i�)�filename�content�mode)r�
ensure_dirr`r�r�rqr�)�datadirr!r{rkr�rmris       r3r�r�sx��
�������O�O�G�W�%�����E�
��g������W�d�+���D� �&�w��6�G������e�D�	Er4c�4�tj|�}i}i}|jxsd}|jr|j|d<|jr|j|d<|j
�|j
|d<n|jrd|d<i}|jr|j|d<|jr0d|d<t|jk7rt|j�|d	<|rd
|i|d<|j|d<td
|jztj��|j|d<td|jztj��|j|d<td|jztj��|||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.
    r�rr��
ssh_pwauthTr�F�lock_passwd�
hashed_passwdrrr�zPreprovisionedVm: %srzr�zPreprovisionedVMType: %s�ProvisionGuestProxyAgentzProvisionGuestProxyAgent: %s)r �
parse_text�custom_datar&�public_keys�disable_ssh_password_auth�passwordr%re�encrypt_pass�preprovisioned_vmr(r|r��preprovisioned_vm_type�provision_guest_proxy_agent)rX�ovf_envrOr�r��defusers      r3r�r�"s����"�"�8�,�G��B�
�C�	�	�	�	"��B����&�/�/�������!�-�-��J���(�(�4� '� A� A�A��L��	�	�	� ��L���G����!�*�*�������!&��
���7�#3�#3�3�'3�G�4D�4D�'E�G�O�$��,�g�6��M��%�7�7�C������!:�!:�:��H�H��
#*�"@�"@�C����"�W�%C�%C�C��H�H��
'.�&I�&I�C�"�#��&��)L�)L�L��H�H��
��C�=�r4c��t|�Sr�)r/)rzs r3r{r{Zs
����"�"r4c��	t|�5}|jd�	ddd�y#1swYyxYw#t$rtj	d|�YywxYw)z;Return boolean indicating path to cdrom device has content.iNTzcdrom (%s) is not configuredF)�open�read�IOErrorr|r})�	cdrom_dev�fps  r3�_check_freebsd_cdromr�^sZ��=�
�)�_�	���G�G�D�M��	�	�	�
���=��	�	�0�)�<��=�s#�5�)�5�2�5�5�A�Ac�|�|�ytj|d��}tj|�j	�S)zIReturn content random seed file if available, otherwise,
    return None.NT)�quiet)r�load_binary_filerD�	b64encode�decode)�sourcer�s  r3rFrFjs:���~��� � ��t�4�D����D�!�(�(�*�*r4c#��K�|��t��tj�rd}t|�r,|��n'dD]"}tjd|z�Ed{����$|r|��yy7��w)Nz/dev/cd0)�iso9660rzTYPE=%s)�DEFAULT_PROVISIONING_ISO_DEVrr.r�r<)r��	cache_dirr��fstypes    r3r-r-sg����
�J�
&�&������	��	�*��O�(�	?�F��*�*�9�v�+=�>�>�>�	?�����
?�s�AA&�A$�A&c�&�tjj|d�}tjj|�st	d��t|d�5}|j
�}ddd�t�\}}}|||d|ifS#1swY� xYw)Nr(zNo ovf-env file found�rb)r�r�rqr�rr�r�r�)�
source_dir�ovf_filer�rXrOr�r�s       r3r0r0�s����w�w�|�|�J�
�6�H�
�7�7�>�>�(�#� �!8�9�9�	
�h��	����7�7�9���!��*�K�B��C���C�-��2�3�3�	��s�B�B�network_metadatar�c	���did�}t|d�D�]Z\}}d}dj|��}d|dzd	zi}d
|dd�}dD]�}	|j|	i�jd
g�}
|
stj	d|	|��?d
}|	dk(rd}nd}|
r
d
|d<||d<|s�Z|
ddD]]}||	ddjd|�}
|d}|jd�sg|d<|djdj||
����_��|rS|rQt
|d�}|jd|j�i|d��t|�}|r||dd <||d!|<��Ctj	d"|||���]|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.
    rL)�version�	ethernetsr�Fzeth{idx})�idxzroute-metricrOr�T)�dhcp4zdhcp4-overrides�dhcp6)�ipv4�ipv6�	ipAddresszNo %s addresses found for: %rr��24�128r�zdhcp6-overridesN�subnetr�prefix�privateIpAddress�	addressesz
{ip}/{prefix})�ipr�r�
macaddress)�matchzset-namer�r�r�z<No configuration for: %s (dev_config=%r) (has_ip_address=%r))
�	enumerater�rr|r}r�rsr>rrr�)r�r��	netconfigr��intf�has_ip_address�nicname�
dhcp_override�
dev_config�	addr_typer��default_prefix�addr�	netPrefix�	privateIprir�s                 r3r�r��s���-.�B� ?�I��/��<�=�=
�	��T����#�#��#�,��'�#��'�S��9�
��,��&
�
�
*� 	�I�����B�/�3�3�K��D�I���	�	�9�9�d�K��!�N��F�"�!%��!&���*.�J�w�'�5B�J�0�1�9��!�!�"�
�

�� ��O�H�5�a�8�<�<��n��	�!�!3�4�	�!�~�~�k�2�.0�J�{�+��;�'�.�.�#�*�*�i�	�*�J��

�- 	�B�.�'��\�(:�;�C����'�����5�7�K�
�5�S�9�F��06�
�7�#�H�-�.8�I�k�"�7�+���	�	�J����		
�q=
�|�r4c�:�tjd��}|�iS|S)ztGenerate fallback network config.

    @return: Dictionary containing network version 2 standard configuration.
    T)�
config_driver)r�generate_fallback_configr�s r3rr�s#���
&�
&�T�
:�C�
�{��	��Jr4c��|st}d}|D]�}tjj|�s�#|sJtjdtjjtjjg�d}tjj|�rtj|���tj|���y)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�rtr|r�rrr�BOOT_LEGACY�isdirr�del_dirr�)r��loggedr�s   r3r�r��s���&�/��
�F��
$��
�7�7�>�>�$������-��^�^�)�)�9�+@�+@�+F�+F�G�	����w�w�}�}�T�"����T�"��
�
�d�#�
$r4c�6�tj|t�Sr�)r�list_from_depends�datasources)�dependss r3�get_datasource_listr�&s���$�$�W�k�:�:r4r�)r,)�rD�	functools�loggingr��os.pathrRr�xml.etree.ElementTree�etree�ElementTreer`�enumr�pathlibrrr�typingrrr	r
rf�	cloudinitrrr
rr�cloudinit.eventrr�
cloudinit.netr�cloudinit.net.dhcprrr�cloudinit.net.ephemeralrr�cloudinit.reportingr�cloudinit.sources.azurerrrr�cloudinit.sources.helpersr�cloudinit.sources.helpers.azurerrr r!r"r#r$r%r&r'r(r)r*�cloudinit.url_helperr+�crypt�partial�rand_strr/�__annotations__r1�AttributeError�passlib.hash�passlib�hash�sha512_crypt�	getLoggerr?r|�DS_NAMEr�r��
DEFAULT_FS�AGENT_SEED_DIRr�r9rHr�r�r\rarhrsrxr�r�r�r�r�r�r.�res_diskr}r�r�r�r��DEF_EPHEMERAL_LABELrer�r�r?rCr@rAr�r9r>rRr2r�r�r�r{r�rFr-r0rr�rr��DataSourceAzureNet�DEP_FILESYSTEMr�r�rGr4r3�<module>r�sw�����	��	�
�"�"����,�,��8�8�1�'���
J�&�?�?�-�����*���*��*�*�
���C�
��
�
�R� 8�9�:��M�3��(�g����!��
��!�#3�4��6��
�
�#��)���d��*J���#��I�#���,�&���c��h�s�m��"�s��s����t�C�y���������#�����4�C��D�=�7�'�X�c�]�'�V�4�?�?���J�+�A�.�H����	�	�-�.�$�x�/���	�	�)�*�"���!�#5�6� �.2�	��	���e��
��!+�n�E�F�	'�#��W�%��/��)��"��k�g�(�(�k�\'����#�"
��5��5��jN��jN�Z�&�!��	(��(�
�(��(��	(��(�V�E��E�8�4��4�n#������3�+��+�(�����
4��
4��J��J�-1�J�
�	J��J�Z��t�����"$��"$�L%���w�-�-�/�0���;��[?	�^�$��
�����1�1�6�6�
���	�	�	���s6�6;J#�#K"�+$K�K"�K�K"�K�K"�!K"