
    x[h<                     \   d dl Z d dlZd dlmZmZ d dlmZmZ d dlmZ	 d dlm
Z
 d dlmZmZ d dlmZ d dlmZ d d	lmZ d d
lmZmZ  ej.                  e      ZdZ G d dej6                        Zd Zd Z G d de      Zeej@                  ffeej@                  ejB                  ffgZ"d Z#y)    N)ListUnion)dmisources)
url_helper)util)
EventScope	EventType)NoDHCPLeaseError)EphemeralIPNetwork)DataSourceHostname)aliyunec2zAlibaba Cloud ECSc                   ^    e Zd ZU dZdgZdZg Zee   e	d<   dZ
dZdZej                  Zeeef   e	d<   d	Zed
        Z fdZdeddf fdZd#dZd Zd Zed        Zed        Zed        Zed        Zed        Zed        Z d Z!d Z"d Z#d Z$d Z%d Z&d Z'd Z(d$d Z)d%d!Z*d$d"Z+ xZ,S )&DataSourceAliYunAliYunzhttp://100.100.100.200z
2016-01-01extended_metadata_versions   2   N_network_configFc                      y)NzX-aliyun-ecs-metadata-token selfs    D/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceAliYun.pyimdsv2_token_put_headerz(DataSourceAliYun.imdsv2_token_put_header+   s    ,    c                     t         t        |   |||       t        j                  | j
                        | _        | j
                  t        j                     j                  t        j                         y N)superr   __init__copydeepcopydefault_update_eventsr	   NETWORKaddr
   BOOT)r   sys_cfgdistropaths	__class__s       r   r!   zDataSourceAliYun.__init__/   sQ    .wF%)]]43M3M%N""":#5#56::9>>Jr   ci_pkl_versionreturnc                 $    t         |   |       y r   )r    	_unpickle)r   r,   r+   s     r   r/   zDataSourceAliYun._unpickle4   s    .)r   c                 `    | j                   j                  d      }d}|d}d}t        ||      S )NhostnameFzlocalhost.localdomainT)metadatagetr   )r   fqdn
resolve_ipmetadata_onlyr1   
is_defaults         r   get_hostnamezDataSourceAliYun.get_hostname7   s8    ==$$Z0
.HJ!(J77r   c                 L    t        | j                  j                  di             S )Nzpublic-keys)parse_public_keysr2   r3   r   s    r   get_public_ssh_keysz$DataSourceAliYun.get_public_ssh_keys?   s     !2!2="!EFFr   c                 L    t               r| j                  j                         S y)NNO_ALIYUN_METADATA)
_is_aliyundsnamelowerr   s    r   _get_cloud_namez DataSourceAliYun._get_cloud_nameB   s    <;;$$&&#r   c                 6    | j                   j                         S r   )r?   r@   r   s    r   platformzDataSourceAliYun.platformG   s    {{  ""r   c                      y)Nzlatest/api/tokenr   r   s    r   api_token_routez DataSourceAliYun.api_token_routeL   s    !r   c                      y)N21600r   r   s    r   imdsv2_token_ttl_secondsz)DataSourceAliYun.imdsv2_token_ttl_secondsP   s    r   c                 2    | j                   | j                  gS r   )r   imdsv2_token_req_headerr   s    r   imdsv2_token_redactz$DataSourceAliYun.imdsv2_token_redactT   s    ,,d.J.JKKr   c                      | j                   dz   S )Nz-ttl-seconds)r   r   s    r   rJ   z(DataSourceAliYun.imdsv2_token_req_headerX   s    ++n<<r   c           	         | j                   t        j                  k7  r| j                   S i }| j                  j                  }| j
                  j                  d      }t        |t              r8t        j                  ||t        j                  | j                  dd            }nt        j                  d|       |S || _         | j                   S )z@Return a network config dict for rendering ENI or netplan files.networkapply_full_imds_network_configT)fallback_nicfull_network_configz%Metadata 'network' key not valid: %s.)r   r   UNSETr)   fallback_interfacer2   r3   
isinstancedictr   #convert_ecs_metadata_network_configr   get_cfg_option_boolds_cfgLOGwarning)r   resultifacenet_mds       r   network_configzDataSourceAliYun.network_config\   s     7==0'''..""9-fd#??"$($<$<KK!A4%F KK?HM%###r   c                    g }i }| j                   }d}|D ]*  }dj                  ||      }|j                  |       |||<   , t        j	                  d       d}d}| j                         }		 t        j                  ||	j                  |	j                  t        j                  | j                  | j                  || j                  d	      \  }}|r|r|| _        ||   S y# t        j                  $ r Y &w xY w)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)	urlsmax_waittimeout	status_cb
headers_cbexception_cbrequest_methodheaders_redactconnect_synchronously)rE   formatappendrY   debugget_url_paramsuhelpwait_for_urlmax_wait_secondstimeout_secondsrZ   _get_headers_imds_exception_cbrK   UrlError
_api_token)
r   mdurlsra   url2baseurl_pathrg   urlcurresponse
url_paramss
             r   _maybe_fetch_api_tokenz'DataSourceAliYun._maybe_fetch_api_tokens   s    '' 	 C""31CKKHSM	  			12((*
	!..#44"22++,,!44-#77&+
MC  8&DOC=   ~~ 	 	s   ,A!C C43C4c                    | j                   }|j                  d| j                        }| j                  |      }|r2|| _        t
        j                  d| j                         t        |      S t
        j                  d       t        |      S )Nmetadata_urlszUsing metadata source: '%s'z)IMDS's HTTP endpoint is probably disabled)	rX   r3   r   r}   metadata_addressrY   rl   rZ   bool)r   mcfgrv   r   s       r   wait_for_metadata_servicez*DataSourceAliYun.wait_for_metadata_service   s}    {{/4+=+=>  66v>$4D!II3T5J5JK $%% KKCD$%%r   c           	      N   | j                         si S | j                  }i }| j                  }| j                  }d}| j                  }	 t        j                  | j                  | j                  | j                  ||d      |d<   t        j                  | j                  | j                  | j                  ||d      |d<   	 t        j                  | j                  | j                  | j                  ||      }||d<   |S # t        $ re t        j                  t        d| j                         t        j                   | j                  | j                  | j                  |||      |d<   Y |S w xY w# t        $ r* t        j                  t        d	| j                         i cY S w xY w)
zCrawl metadata service when available.

        @returns: Dictionary of crawled metadata content containing the keys:
          meta-data, user-data, vendor-data and dynamic.
        N	user-data)re   rh   rf   	item_namevendor-data)re   rh   rf   	meta-dataz@Faild read json meta-data from %s fall back directory tree style)re   rh   rf   retrieval_exception_ignore_cbz'Failed reading from metadata address %s)r   rK   _refresh_stale_aliyun_token_cb&_skip_or_refresh_stale_aliyun_token_cb#_skip_json_path_meta_path_aliyun_cbr   get_instance_datamin_metadata_versionr   rr   get_instance_meta_data	Exceptionr   logexcrY   r   get_instance_metadata)r   redactcrawled_metadataexc_cb	exc_cb_udskip_cbexe_cb_whole_metar[   s           r   crawl_metadatazDataSourceAliYun.crawl_metadata   s    --/I))44??	 DD/	,2,D,D))%%,,%&%-[) /5.F.F))%%,,%&'/]+66--))#00#)!2 17 -.  -  5))	 140I0I--))#00#)!'291 -  -  	KK9%%
 I	s2   A8E1 =D   A*E.*E1 -E..E1 10F$#F$c                     t        |t        j                        r+|j                  dk(  rt        j                  d       d| _        y)z=Exception handler for Ecs to refresh token if token is stale.i  z+Clearing cached Ecs API token due to expiryNT)rT   rn   rt   coderY   rl   ru   r   msg	exceptions      r   r   z/DataSourceAliYun._refresh_stale_aliyun_token_cb   s2    i0Y^^s5JIICD"DOr   c                     |j                   |vS )z/Returns False if cause.code is in status_codes.)r   )r   status_codescauses      r   _skip_retry_on_codesz%DataSourceAliYun._skip_retry_on_codes   s    zz--r   c                 l    | j                  t        j                  |      }|sy| j                  ||      S )z^Callback will not retry on SKIP_USERDATA_VENDORDATA_CODES or
        if no token is available.F)r   r   SKIP_USERDATA_CODESr   )r   r   r   retrys       r   r   z7DataSourceAliYun._skip_or_refresh_stale_aliyun_token_cb   s5     ))#*A*A9M223	BBr   c                     t        |t        j                        r%|j                  dk(  rt        j                  d       y| j                  ||      S )z7Callback will not retry of whole meta_path is not foundi  z&whole meta_path is not found, skippingF)rT   rn   rt   r   rY   rZ   r   r   s      r   r   z4DataSourceAliYun._skip_json_path_meta_path_aliyun_cb  s>    i0Y^^s5JKK@A223	BBr   c                 0   | j                   | j                  j                         k7  ry| j                  rt	        j
                         rt        j                  d       y	 t        | j                  | j                  j                  dd      5 }| j                         | _        t        j                  d|j                  rd|j                   nd       d d d        n| j                         | _        | j                  rt        | j                  t               sy| j                  j#                  di       | _        | j                  j#                  d	i       | _        | j                  j#                  d
i       | _        y# 1 sw Y   xY w# t        $ r Y yw xY w)NFz1FreeBSD doesn't support running dhclient with -sfT)ipv4ipv6zCrawled metadata service%s  r   r   r   )
cloud_namer?   r@   perform_dhcp_setupr   
is_FreeBSDrY   rl   r   r)   rS   r   _crawled_metadata	state_msgr   rT   rU   r3   r2   userdata_rawvendordata_raw)r   netws     r   	_get_datazDataSourceAliYun._get_data  sS   ??dkk//11"" 		MN'KKKK22	 

 -1-@-@-BD*II404!DNN+,B
 &*%8%8%:D"%%Z""D.
 ..22;C 2266{BG"4488K/
 
 $ s+    -F	 AE=F	 =FF	 		FFc                 ~   || j                   }t        j                  d       | j                  |i}dj	                  | j
                  | j                        }	 t        j                  ||| j                  d      }|j                  S # t        j                  $ r!}t        j                  d||       Y d}~yd}~ww xY w)zRequest 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`   )headersrh   rg   z/Unable to get API token: %s raised exception %s)rH   rY   rl   rJ   rj   r   rE   rn   readurlrK   rt   rZ   contents)r   secondsrequest_header	token_urlr{   es         r   _refresh_api_tokenz#DataSourceAliYun._refresh_api_token/  s     ?33G		5666@NN4#8#8$:N:NO		}}&#77$	H     ~~ 	KKA9a 		s   #B B<B77B<c                     | j                   | j                  i}| j                  |v r|S | j                  s#| j	                         | _        | j                  si S | j
                  | j                  iS )zReturn 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.
        )rJ   rH   rE   ru   r   r   )r   ry   request_token_headers      r   rr   zDataSourceAliYun._get_headersJ  sm     (($*G*G 
 3&'' #557DO??	,,doo>>r   c                     t        |t        j                        rX|j                  rL|j                  dk\  r=|j                  dk(  rt        j                  d       |t        j                  d       |y)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 tokensT)rT   rn   rt   r   rY   rZ   )r   r   s     r   rs   z#DataSourceAliYun._imds_exception_cb_  sc     i0~~)..C"7>>S(KK?   KKL  r   )FFFr   )r   )-__name__
__module____qualname__r?   r   r   r   r   str__annotations__url_max_waiturl_timeoutru   r   rR   r   r   rU   r   propertyr   r!   intr/   r8   r;   rA   rC   rE   rH   rK   rJ   r^   r}   r   r   r   r   r   r   r   r   rr   rs   __classcell__)r+   s   @r   r   r      sL   F-.M (,.S	. LK J(/OU39%5  - -K
* * *8G$
 # # " "   L L = = $ $,0d&> @.CCB!6?*r   r   c                  :    t        j                  d      t        k(  S )Nzsystem-product-name)r   read_dmi_dataALIYUN_PRODUCTr   r   r   r>   r>   |  s    23~EEr   c                    g }| j                         D ]  \  }}t        |t              r |j                  |j	                                6t        |t
              r|j                  |       Xt        |t              si|j                  dg       }t        |t              r |j                  |j	                                t        |t
              s|j                  |        |S )Nzopenssh-key)	itemsrT   r   rk   striplistextendrU   r3   )public_keyskeys_key_idkey_bodykeys        r   r:   r:     s    D(..0 
!h$KK()$'KK!$',,}b1C#s#CIIK(C&C 
! Kr   c                       e Zd ZdZdZy)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)r   r   r   __doc__r   r   r   r   r   r     s     r   r   c                 6    t        j                  | t              S r   )r   list_from_dependsdatasources)dependss    r   get_datasource_listr     s    $$Wk::r   )$r"   loggingtypingr   r   	cloudinitr   r   r   rn   r   cloudinit.eventr	   r
   cloudinit.net.dhcpr   cloudinit.net.ephemeralr   cloudinit.sourcesr   cloudinit.sources.helpersr   r   	getLoggerr   rY   r   
DataSourcer   r>   r:   r   DEP_FILESYSTEMDEP_NETWORKr   r   r   r   r   <module>r      s       " )  1 / 6 0 1g!$dw)) dNF 	, 	 W3356..0C0CDE;r   