3
\�meT) � @ s d dl Z d dlZd dlmZmZmZmZ d dlmZm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ e je�Zdd gZd
Zd eiZd Zd
ZeegZdZ dZ dZ eZ e e e e g7 Z G dd� dej ej!�Z"G dd� de"�Z#ddd�Z$e#ej%ffe"ej%ej&ffgZ'dd� Z(dS )� N)�dmi�sources�
url_helper�util)�
EventScope� EventType)�NoDHCPLeaseError)�EphemeralDHCPv4)�DataSourceOracle)� openstackzhttp://[fe80::a9fe:a9fe]zhttp://169.254.169.254ziid-dsopenstackz instance-idzOpenStack NovazOpenStack ComputeZOpenTelekomCloudz
SAP CCloud VMZ HUAWEICLOUDc s� e Zd ZdZejZdZej e
j e
j e
j
e
jhiZ� fdd�Zdd� Zdd� Zd d
� Zed d � �Zd
d� Zdd� Zdd� Z� ZS )�DataSourceOpenStackZ OpenStackFc sH t t| �j|||� d | _tj| j�| _d | _i | _ t
j | _ t
j | _
d S )N)�superr �__init__�metadata_addressr Zfetch_ssl_details�paths� ssl_details�version�filesr �UNSET� ec2_metadata� network_json)�self�sys_cfg�distror )� __class__� �)/usr/lib/python3.6/DataSourceOpenStack.pyr = s z DataSourceOpenStack.__init__c C s" t jj| �}d|| j| jf }|S )Nz%s [%s,ver=%s])r �
DataSource�__str__�dsmoder )r �rootZmstrr r r r F s zDataSourceOpenStack.__str__c C s | j jdt�}dd� |D �}t|�t|�krHtjdtt|�t|� �� t|�rV|}ntjd� t}g }i }x*|D ]"}t j
|d�}|j |� |||< qrW | j � }t
j
� }t j||j|jdd�\} }
| r�tjd || � ntjd
|tt
j
� | �� |j| �| _t| �S ) NZ
metadata_urlsc S s g | ]}t j|�r|�qS r )r Zis_resolvable_url)�.0�xr r r �
<listcomp>M s zADataSourceOpenStack.wait_for_metadata_service.<locals>.<listcomp>z,Removed the following from metadata urls: %sz+Empty metadata url list! using default listr F)�urlsZmax_wait�timeoutZconnect_synchronouslyzUsing metadata source: '%s'z2Giving up on OpenStack md from %s after %s seconds)�ds_cfg�get� DEF_MD_URLS�set�LOG�debug�list�len�warningr Z combine_url�append�get_url_params�timeZ wait_for_urlZmax_wait_seconds�timeout_seconds�intr �bool) r r$ ZfilteredZmd_urlsZurl2baseZurlZmd_url�
url_paramsZ
start_timeZ avail_urlZ _responser r r �wait_for_metadata_serviceK s>
z-DataSourceOpenStack.wait_for_metadata_servicec C s t j| j� �S )N)r Z instance_id_matches_system_uuidZget_instance_id)r r r r r �check_instance_idt s z%DataSourceOpenStack.check_instance_idc C sn | j tjkr| j S tj| jjdd��r2d| _ | j S | jtjkrLtj d� dS tj
d� t j | jdd�| _ | j S )z@Return a network config dict for rendering ENI or netplan files.Zapply_network_configTNz<Unexpected call to network_config when network_json is None.z(network config provided via network_json)Z
known_macs)
�_network_configr r r Zis_falser& r' r r* r. r+ r Zconvert_net_json)r r r r �network_configx s
z"DataSourceOpenStack.network_configc , C s� | j rry2t| j| j�� tjtjd| jd�}W dQ R X W q� t t
j fk
rn } ztj tt
|�� dS d}~X q�X n@y | j� }W n2 t
j k
r� } ztj tt
|�� dS d}~X nX | j|jd�g�| _| jt
jkr�dS |jdi �}tj|tg�}|| _|jd�| _|jd�| _|jd �| _|d
| _| jj|jd i �� |jd �}|| _yt
j |�| _ W n6 t k
�r� } ztj d
|� d| _ W Y dd}~X nX |jd�}|| _ yt
j |�| _!W n6 t k
�r� } ztj d|� d| _!W Y dd}~X nX dS )a
Crawl metadata, parse and persist that data for this instance.
@return: True when metadata discovered indicates OpenStack datasource.
False when unable to contact metadata service or when metadata
format is invalid or disabled.
zCrawl of metadata service)Zlogfunc�msg�funcNFr �metadataz ec2-metadataZ networkdataZuserdatar r Z
vendordataz"Invalid content in vendor-data: %sZ vendordata2z#Invalid content in vendor-data2: %sT)"�perform_dhcp_setupr r Zfallback_interfacer �log_timer* r+ �_crawl_metadatar r �InvalidMetaDataExceptionZlogexc�strZ_determine_dsmoder' r ZDSMODE_DISABLEDZ
mergemanydict�DEFAULT_METADATAr< r r Z userdata_rawr r �updateZvendordata_pureZconvert_vendordataZvendordata_raw�
ValueErrorr. Zvendordata2_pureZvendordata2_raw)r �results�eZmdZvdZvd2r r r � _get_data� sR
z DataSourceOpenStack._get_datac C s� y| j � stjd��W n8 tk
rN } z tjdjt|�d���W Y dd}~X nX | j� }y,tjt j
dt | j g| j
|j|jd�d�}W nr tjk
r� } ztjt|���W Y dd}~X nB tjtfk
r� } z dj| j d �}tj|�|�W Y dd}~X nX |S )
z�Crawl metadata service when available.
@returns: Dictionary with all metadata discovered for this datasource.
@raise: InvalidMetaDataException on unreadable or broken
metadata.
z No active metadata service foundz,IOError contacting metadata service: {error})�errorNz#Crawl of openstack metadata service)r �retriesr% )�args�kwargsz Broken metadata address {addr})Zaddr)r6 r r@ �IOError�formatrA r0 r r> r* r+ �read_metadata_servicer r Z num_retriesr2 r Z NonReadableZBrokenMetadata)r rF r5 �resultr: r r r r? � s2
z#DataSourceOpenStack._crawl_metadatac C sp d| j jd�k}tj� s dS tjd�}|tkr2dS tjd�tkrDdS |rTtj � rTdS tj
d�jd�t krldS dS ) z<Return True when a potential OpenStack platform is detected.ZOracleZdatasource_listTzsystem-product-namezchassis-asset-tag� � product_nameF) r r' r Zis_x86r Z
read_dmi_data�VALID_DMI_PRODUCT_NAMES�VALID_DMI_ASSET_TAGS�oracleZ_is_platform_viableZ get_proc_env�DMI_PRODUCT_NOVA)r Z
accept_oraclerQ r r r � ds_detect� s
z DataSourceOpenStack.ds_detect)�__name__�
__module__� __qualname__Zdsnamer r r8 r= r ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZHOTPLUGZsupported_update_eventsr r r6 r7 �propertyr9 rG r? rV �
__classcell__r r )r r r + s
) 9*r c @ s e Zd ZdZdZdS )�DataSourceOpenStackLocalau Run in init-local using a dhcp discovery prior to metadata crawl.
In init-local, no network is available. This subclass sets up minimal
networking with dhclient on a viable nic so that it can talk to the
metadata service. If the metadata service provides network configuration
then render the network configuration for that instance based on metadata.
TN)rW rX rY �__doc__r= r r r r r\ s r\ � c C s t j| |||d�}|j� S )N)r r% rI )r ZMetadataReaderZread_v2)Zbase_urlr r% rI �readerr r r rN s rN c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list" s ra )Nr^ r^ ))Zloggingr1 Z cloudinitr r r r Zcloudinit.eventr r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr Zcloudinit.sourcesr
rT Zcloudinit.sources.helpersr Z getLoggerrW r* r( Z DEFAULT_IIDrB rU ZDMI_PRODUCT_COMPUTErR ZDMI_ASSET_TAG_OPENTELEKOMZDMI_ASSET_TAG_SAPCCLOUDZDMI_ASSET_TAG_HUAWEICLOUDrS Z SourceMixinr r r\ rN ZDEP_FILESYSTEMZ DEP_NETWORKr` ra r r r r �<module> s:
]