Google
Edit File: DataSourceAliYun.cpython-39.pyc
a ���i�< � @ s� d dl Z d dlZd dlmZmZ d dlmZmZ d dlmZ d dlm Z d dl m Z m Z d dlmZ d dlmZ d d lmZ d d lmZmZ e�e�Zd ZG d d � d ej�Z dd� Z dd� Z G dd� de �Z e ej ffe ej ej!ffgZ"dd� Z#dS )� N)�List�Union)�dmi�sources)� url_helper)�util)� EventScope� EventType)�NoDHCPLeaseError)�EphemeralIPNetwork)�DataSourceHostname)�aliyun�ec2zAlibaba Cloud ECSc sB e Zd ZU dZdgZdZg Zee e d< dZ dZ dZ e jZeeef e d< d Zed d � �Z� fd d �Zedd�� fdd� Zd:dd�Zdd� Zdd� Zedd� �Zedd� �Z edd � �Z ed d � �Z ed d � �Z ed!d"� �Z d#d$� Z!d%d&� Z"d'd(� Z#d)d*� Z$d+d,� Z%d-d.� Z&d/d0� Z'd1d2� Z(d;d3d4�Z)d<d6d7�Z*d=d8d9�Z+� Z,S )>�DataSourceAliYunZAliYunzhttp://100.100.100.200z 2016-01-01�extended_metadata_versions�� �2 N�_network_configFc C s dS )NzX-aliyun-ecs-metadata-token� ��selfr r �F/usr/lib/python3.9/site-packages/cloudinit/sources/DataSourceAliYun.py�imdsv2_token_put_header+ s z(DataSourceAliYun.imdsv2_token_put_headerc s: t t| ��|||� t�| j�| _| jtj �t j � d S �N) �superr �__init__�copy�deepcopyZdefault_update_eventsr ZNETWORK�addr ZBOOT)r Zsys_cfg�distro�paths�� __class__r r r / s zDataSourceAliYun.__init__)�ci_pkl_version�returnc s t � �|� d S r )r � _unpickle)r r# r! r r r% 4 s zDataSourceAliYun._unpicklec C s* | j �d�}d}|d u r d}d}t||�S )N�hostnameFzlocalhost.localdomainT)�metadata�getr )r ZfqdnZ resolve_ipZ metadata_onlyr& Z is_defaultr r r � get_hostname7 s z DataSourceAliYun.get_hostnamec C s t | j�di ��S )Nz public-keys)�parse_public_keysr' r( r r r r �get_public_ssh_keys? s z$DataSourceAliYun.get_public_ssh_keysc C s t � r| j�� S dS )NZNO_ALIYUN_METADATA)� _is_aliyun�dsname�lowerr r r r �_get_cloud_nameB s z DataSourceAliYun._get_cloud_namec C s | j �� S r )r- r. r r r r �platformG s zDataSourceAliYun.platformc C s dS )Nzlatest/api/tokenr r r r r �api_token_routeL s z DataSourceAliYun.api_token_routec C s dS )NZ21600r r r r r �imdsv2_token_ttl_secondsP s z)DataSourceAliYun.imdsv2_token_ttl_secondsc C s | j | jgS r )r �imdsv2_token_req_headerr r r r �imdsv2_token_redactT s z$DataSourceAliYun.imdsv2_token_redactc C s | j d S )Nz -ttl-seconds)r r r r r r3 X s z(DataSourceAliYun.imdsv2_token_req_headerc C sn | j tjkr| j S i }| jj}| j�d�}t|t�rRt j ||t � | j dd�d�}nt�d|� |S || _ | j S )z@Return a network config dict for rendering ENI or netplan files.�networkZ apply_full_imds_network_configT)Z fallback_nicZfull_network_configz%Metadata 'network' key not valid: %s.)r r �UNSETr �fallback_interfacer' r( � isinstance�dictr Z#convert_ecs_metadata_network_configr Zget_cfg_option_bool�ds_cfg�LOG�warning)r �resultZifaceZnet_mdr r r �network_config\ s" �� z DataSourceAliYun.network_configc C s� g }i }| j }d}|D ]"}d�||�}|�|� |||< qt�d� d}d}| �� } z0tj|| j| j tj | j | j || j dd� \}}W n tjy� Y n0 |r�|r�|| _|| S dS )a= Get an API token for ECS Instance Metadata Service. On ECS. IMDS will always answer an API token, set HttpTokens=optional (default) when create instance will not forcefully use the security-enhanced mode (IMDSv2). https://api.alibabacloud.com/api/Ecs/2014-05-26/RunInstances �PUTz{0}/{1}z Fetching Ecs IMDSv2 API TokenNF) �urlsZmax_wait�timeoutZ status_cb� headers_cb� exception_cb�request_method�headers_redactZconnect_synchronously)r1 �format�appendr; �debugZget_url_params�uhelpZ wait_for_urlZmax_wait_secondsZtimeout_secondsr< � _get_headers�_imds_exception_cbr4 �UrlError� _api_token) r �mdurlsr@ Zurl2baseZurl_pathrD �urlZcur�responseZ url_paramsr r r �_maybe_fetch_api_tokens s< � z'DataSourceAliYun._maybe_fetch_api_tokenc C sJ | j }|�d| j�}| �|�}|r8|| _t�d| j� n t�d� t|�S )N� metadata_urlszUsing metadata source: '%s'z)IMDS's HTTP endpoint is probably disabled) r: r( rR rQ �metadata_addressr; rH r< �bool)r ZmcfgrN rS r r r �wait_for_metadata_service� s z*DataSourceAliYun.wait_for_metadata_servicec C s | � � s i S | j}i }| j}| j}d}| j}z�tj| j| j| j ||dd�|d< tj| j| j| j ||dd�|d< z&tj | j| j| j ||d�}||d< W nB t y� t � td| j� tj| j| j| j |||d�|d< Y n0 W n( t �y t � td | j� i Y S 0 |S ) z�Crawl metadata service when available. @returns: Dictionary of crawled metadata content containing the keys: meta-data, user-data, vendor-data and dynamic. N� user-data)rB rE rC Z item_name� vendor-data)rB rE rC � meta-dataz@Faild read json meta-data from %s fall back directory tree style)rB rE rC Z retrieval_exception_ignore_cbz'Failed reading from metadata address %s)rU r4 � _refresh_stale_aliyun_token_cb�&_skip_or_refresh_stale_aliyun_token_cb�#_skip_json_path_meta_path_aliyun_cbr Zget_instance_data�min_metadata_versionrS rJ Zget_instance_meta_data� Exceptionr Zlogexcr; r Zget_instance_metadata)r ZredactZcrawled_metadataZexc_cbZ exc_cb_udZskip_cbZexe_cb_whole_metar= r r r �crawl_metadata� sp � � � ��� z DataSourceAliYun.crawl_metadatac C s* t |tj�r&|jdkr&t�d� d| _dS )z=Exception handler for Ecs to refresh token if token is stale.i� z+Clearing cached Ecs API token due to expiryNT)r8 rI rL �coder; rH rM �r �msg� exceptionr r r rY � s z/DataSourceAliYun._refresh_stale_aliyun_token_cbc C s |j |vS )z/Returns False if cause.code is in status_codes.)r_ )r Z status_codes�causer r r �_skip_retry_on_codes� s z%DataSourceAliYun._skip_retry_on_codesc C s" | � tj|�}|sdS | �||�S )z^Callback will not retry on SKIP_USERDATA_VENDORDATA_CODES or if no token is available.F)rd r ZSKIP_USERDATA_CODESrY )r ra rb Zretryr r r rZ � s z7DataSourceAliYun._skip_or_refresh_stale_aliyun_token_cbc C s0 t |tj�r$|jdkr$t�d� dS | �||�S )z7Callback will not retry of whole meta_path is not foundi� z&whole meta_path is not found, skippingF)r8 rI rL r_ r; r< rY r` r r r r[ s z4DataSourceAliYun._skip_json_path_meta_path_aliyun_cbc C s� | j | j�� krdS | jr�t�� r0t�d� dS zbt| j | j j ddd��8}| � � | _ t�d|j rld|j � �nd� W d � n1 s�0 Y W q� ty� Y dS 0 n | � � | _ | j r�t| j t�s�dS | j �di �| _| j �d i �| _| j �d i �| _dS ) NFz1FreeBSD doesn't support running dhclient with -sfT)Zipv4Zipv6zCrawled metadata service%s� � rX rV rW )Z cloud_namer- r. �perform_dhcp_setupr Z is_FreeBSDr; rH r r r7 r^ Z_crawled_metadataZ state_msgr r8 r9 r( r' Z userdata_rawZvendordata_raw)r Znetwr r r � _get_data s< � �& �zDataSourceAliYun._get_datac C s� |du r| j }t�d� | j|i}d�| j| j�}ztj||| j dd�}W n6 tj y� } z t� d||� W Y d}~dS d}~0 0 |j S )z�Request new metadata API token. @param seconds: The lifetime of the token in seconds @return: The API token or None if unavailable. Nz!Refreshing Ecs metadata API tokenz{}/{}r? )ZheadersrE rD z/Unable to get API token: %s raised exception %s) r2 r; rH r3 rF rS r1 rI Zreadurlr4 rL r< �contents)r ZsecondsZrequest_headerZ token_urlrP �er r r �_refresh_api_token/ s$ � �z#DataSourceAliYun._refresh_api_tokenrf c C s@ | j | ji}| j|v r|S | js4| �� | _| js4i S | j| jiS )z�Return a dict of headers for accessing a url. If _api_token is unset on AWS, attempt to refresh the token via a PUT and then return the updated token header. )r3 r2 r1 rM rk r )r rO Zrequest_token_headerr r r rJ J s � z DataSourceAliYun._get_headersc C sD t |tj�r@|jr@|jdkr@|jdkr2t�d� n t�d� |�dS )a2 Fail quickly on proper AWS if IMDSv2 rejects API token request Guidance from Amazon is that if IMDSv2 had disabled token requests by returning a 403, or cloud-init malformed requests resulting in other 40X errors, we want the datasource detection to fail quickly without retries as those symptoms will likely not be resolved by retries. Exceptions such as requests.ConnectionError due to IMDS being temporarily unroutable or unavailable will still retry due to the callsite wait_for_url. i� i� zLEcs IMDS endpoint returned a 403 error. HTTP endpoint is disabled. Aborting.z2Fatal error while requesting Ecs IMDSv2 API tokensN)r8 rI rL r_ r; r< r` r r r rK _ s ��z#DataSourceAliYun._imds_exception_cb)FFF)N)rf )N)-�__name__� __module__� __qualname__r- rR r\ r r �str�__annotations__Z url_max_waitZ url_timeoutrM r r6 r r r9 rg �propertyr r �intr% r) r+ r/ r0 r1 r2 r4 r3 r> rQ rU r^ rY rd rZ r[ rh rk rJ rK � __classcell__r r r! r r sN 2@! r c C s t �d�tkS )Nzsystem-product-name)r Z read_dmi_data�ALIYUN_PRODUCTr r r r r, { s r, c C s� g }| � � D ]|\}}t|t�r.|�|�� � q t|t�rD|�|� q t|t�r |�dg �}t|t�rt|�|�� � q t|t�r |�|� q |S )Nz openssh-key) �itemsr8 ro rG �strip�list�extendr9 r( )Z public_keys�keysZ_key_idZkey_body�keyr r r r* s r* c @ s e Zd ZdZdZdS )�DataSourceAliYunLocalay Datasource run at init-local which sets up network to query metadata. 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)rl rm rn �__doc__rg r r r r r{ � s r{ c C s t �| t�S r )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s r~ )$r Zlogging�typingr r Z cloudinitr r r rI r Zcloudinit.eventr r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr Zcloudinit.sourcesr Zcloudinit.sources.helpersr r Z getLoggerrl r; rt Z DataSourcer r, r* r{ ZDEP_FILESYSTEMZ DEP_NETWORKr} r~ r r r r �<module> s, h �