3
\me2 @ 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
dddddddddgdZ G dd de Z G dd dej!Z"G dd de"Z#e#ej$ffe"ej$ej%ffgZ&e ej! ddd Z'dS ) N) b64decode)suppress)Enum)AnyListTupleUnion)sources
url_helperutil)find_fallback_nicget_interfaces_by_mac)EphemeralIPNetwork)get_dmi_configget_local_instance_id is_on_akamaizhttp://169.254.169.254zhttp://[fd00:a9fe:a9fe::1])ipv4ipv6z /v1/tokenz /v1/instancez
/v1/user-data)tokenmetadatauserdataTzf2:3) base_urlspathsallow_local_stageallow_init_stage
allow_dhcp
allow_ipv4
allow_ipv6preferred_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 AVAILABLEDEFER r) r) &/usr/lib/python3.6/DataSourceAkamai.pyr - s r c @ s e Zd ZdZdZdd ZdeeedddZe dd d
Z
e dd d Z e dd
dZ e
eeeef ef dddZdeedddZedddZedddZdS )DataSourceAkamaiAkamaiFc C sJ t jd tjj| ||| t | _tjt tj
|ddgi t g| _ d S )Nz Setting up Akamai DataSourceZ
datasourcer, )
LOGdebugr
DataSource__init__dictr r Z
mergemanydictr Zget_cfg_by_pathBUILTIN_DS_CONFIGds_cfg)selfsys_cfgdistror r) r) r* r0 < s
zDataSourceAkamai.__init__) path_nameuse_v6returnc 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
ValueErrorformat)r4 r7 r8 Z version_keyZbase_urlpathr) 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 rt }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).0prefix)macr) r*
s zBDataSourceAkamai._get_network_context_managers..zGFailed to find default interface, attempting DHCP on fallback interfacer FT)r r r r )r ) r? r
r3 itemsanyr- warningr appendr r6 noop)r4 network_context_managersZ
interfacesZ interfaceZpreferred_prefixesinfr) )rF r* _get_network_context_managers sN
z.DataSourceAkamai._get_network_context_managers)r8 r9 c C s4 yt j| jd|dddddddid }|jd
krBtjd |j d S t|}t j| jd
|ddddd|dd}tjt|| _ t j| jd|ddddd|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_methodtimeout sec_betweenretriesheaders 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 %s64F)r
Zreadurlr= coder- r> strjsonloadsr Z userdata_rawr decodebinasciiErrorrJ 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 dS )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| tS )N)r Zlist_from_depends datasources)Zdependsr) r) r* get_datasource_list s rj )(r^ r[ Zloggingbase64r
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* s@
/