3
\�me�2 � @ s. d dl Z d dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
mZ d dlmZmZmZ d dlmZmZ d dlmZ d d lmZmZmZ ej e �Z d
d d �d
ddd�ddddddgd�Z G dd� de �Z G dd� dej!�Z"G dd� de"�Z#e#ej$ffe"ej$ej%ffgZ&e ej! d�dd �Z'dS ) � N)� b64decode)�suppress)�Enum)�Any�List�Tuple�Union)�sources�
url_helper�util)�find_fallback_nic�get_interfaces_by_mac)�EphemeralIPNetwork)�get_dmi_config�get_local_instance_id� is_on_akamaizhttp://169.254.169.254zhttp://[fd00:a9fe:a9fe::1])�ipv4�ipv6z /v1/tokenz /v1/instancez
/v1/user-data)�token�metadata�userdataTzf2:3)� base_urls�paths�allow_local_stage�allow_init_stage�
allow_dhcp�
allow_ipv4�
allow_ipv6�preferred_mac_prefixesc @ s e Zd ZdZdZdZdZdS )�MetadataAvailabilityResultzj
Used to indicate how this instance should behave based on the availability
of metadata to it
r � � N)�__name__�
__module__� __qualname__�__doc__�
NOT_AVAILABLE� AVAILABLE�DEFER� r) r) �&/usr/lib/python3.6/DataSourceAkamai.pyr - s r c @ s� e Zd ZdZdZdd� Zdeeed�dd�Ze d�d d
�Z
e d�d d �Z e d�d
d�Z e
eeeef ef d�dd�Zdeed�dd�Zed�dd�Zed�dd�ZdS )�DataSourceAkamai�AkamaiFc C sJ t jd� tjj| |||� t� | _tjt � tj
|ddgi �t g�| _ d S )Nz Setting up Akamai DataSourceZ
datasourcer, )
�LOG�debugr �
DataSource�__init__�dictr r Z
mergemanydictr Zget_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg)�self�sys_cfg�distror r) r) r* r0 < s
zDataSourceAkamai.__init__)� path_name�use_v6�returnc C s\ || j d kr tdj|���d}|s0| j d r4d}| j d | }| j d | }dj||�S )z�
Looks up the path for a given name and returns a full url for it. If
use_v6 is passed in, the IPv6 base url is used; otherwise the IPv4 url
is used unless IPv4 is not allowed in ds_cfg
r zUnknown path name {}r r r r z{}{})r3 �
ValueError�format)r4 r7 r8 Z version_keyZbase_url�pathr) r) r* �
_build_urlN s zDataSourceAkamai._build_url)r9 c C sZ | j d r| j d s0| j d r@| j d r@tjd� tjS | jrN| j� S | j� S dS )z�
Returns whether metadata should be retrieved at this stage, at the next
stage, or never, in the form of a MetadataAvailabilityResult.
r r r r z*Configuration prohibits fetching metadata.N)r3 r- �infor r& � local_stage�_should_fetch_data_local�_should_fetch_data_network)r4 r) r) r* �_should_fetch_data` s
z#DataSourceAkamai._should_fetch_datac C sH | j d stjd� tjS | j d rB| j d rBtjd� tjS tjS )z�
Returns whether metadata should be retrieved during the local stage, or
if it should wait for the init stage.
r z)Configuration prohibits local stage setupr r z9Configuration does not allow for ephemeral network setup.)r3 r- r> r r( r' )r4 r) r) r* r@ t s
z)DataSourceAkamai._should_fetch_data_localc C s | j d stjd� tjS tjS )zS
Returns whether metadata should be fetched during the init stage.
r z1Configuration does not allow for init stage setup)r3 r- r> r r( r' )r4 r) r) r* rA � s
z+DataSourceAkamai._should_fetch_data_networkc s� g }| j r�t� }d}| jd }x0|j� D ]$\� }t� fdd�|D ��r(|}P q(W |dkrhtjd� t� }g }| jd r�|jt | j
|ddd �df� | jd
r�| jd r�|jt | j
|dd �df� n4| jd r�|jt � df� | jd
r�|jt � df� |S )
z�
Returns a list of context managers which should be tried when setting
up a network context. If we're running in init mode, this return a
noop since networking should already be configured.
Nr c s g | ]}� j |��qS r) )�
startswith)�.0�prefix)�macr) r* �
<listcomp>� s zBDataSourceAkamai._get_network_context_managers.<locals>.<listcomp>zGFailed to find default interface, attempting DHCP on fallback interfacer FT)r r r r )r ) r? r
r3 �items�anyr- �warningr �appendr r6 �noop)r4 �network_context_managersZ
interfacesZ interfaceZpreferred_prefixes�infr) )rF r* � _get_network_context_managers� sN
z.DataSourceAkamai._get_network_context_managers)r8 r9 c C s4 y�t j| jd|d�ddddddid �}|jd
krBtjd |j� d S t|�}t j| jd
|d�dddd|d�d�}tjt|��| _ t j| jd|d�dddd|id�}t|�| _
yt | j
�j � | _
W n0 t
jk
r� } ztjd|� W Y dd}~X nX W n< t jk
�r. } z tjd|�rdnd|� dS d}~X nX d S )z�
Runs through the sequence of requests necessary to retrieve our
metadata and user data, creating a token for use in doing so, capturing
the results.
r )r8 ZPUT� r! � z Metadata-Token-Expiry-SecondsZ300)Zrequest_method�timeout� sec_between�retries�headers�� z-Fetching token returned %s; not fetching dataTr zapplication/json)ZAcceptzMetadata-Token)rR rS rT rU r zMetadata-Tokenz*Failed to base64 decode userdata due to %sNz1Failed to retrieve metadata using IPv%s due to %s�6�4F)r
Zreadurlr= �coder- r> �str�json�loadsr Z userdata_rawr �decode�binascii�ErrorrJ ZUrlError)r4 r8 Ztoken_responser r r �er) r) r* �_fetch_metadata� sP
" z DataSourceAkamai._fetch_metadatac C s� t jd� t� s t jd� dS t� }d|i| _| j� }|tjkrf|tj krXt jd� dS t jd� dS | j
� }xP|D ]>\}}|�, | j |d�}|r�| jj d |�| jd< P W d
Q R X qtW t j
d � dS ) zW
Overrides _get_data in the DataSource class to actually retrieve data
z#Getting data from Akamai DataSourcez#Not running on Akamai, not running.Fz instance-idz5Metadata is not available, returning local data only.TzFConfigured not to fetch data at this stage; waiting for a later stage.)r8 �idNzHFailed to contact metadata service, falling back to local metadata only.)r- r. r r> r r rB r r' r& rO ra �getrJ )r4 Zlocal_instance_idZ availabilityrM Zmanagerr8 �doner) r) r* � _get_data' s6
zDataSourceAkamai._get_datac C s t j| j� d�S )zj
A local-only check to see if the instance id matches the id we see on
the system
zsystem-serial-number)r Z instance_id_matches_system_uuidZget_instance_id)r4 r5 r) r) r* �check_instance_id[ s z"DataSourceAkamai.check_instance_idN)F)F)r" r# r$ Zdsnamer? r0 rZ �boolr= r rB r@ rA r r r r r rO ra re rf r) r) r) r* r+ 8 s
"MD4r+ c @ s e Zd ZdZdZdS )�DataSourceAkamaiLocalz�
A subclass of DataSourceAkamai that runs the same functions, but during the
init-local stage. This allows configuring networking via cloud-init, as
networking hasn't been configured yet.
TN)r" r# r$ r% r? r) r) r) r* rh e s rh )r9 c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr) r) r* �get_datasource_list s rj )(r^ r[ Zlogging�base64r �
contextlibr rL �enumr Ztypingr r r r Z cloudinitr r
r Z
cloudinit.netr r
Zcloudinit.net.ephemeralr Z cloudinit.sources.helpers.akamair r r Z getLoggerr" r- r2 r r/ r+ rh ZDEP_FILESYSTEMZ DEP_NETWORKri rj r) r) r) r* �<module> s@
/