File: //lib/python3.6/site-packages/cloudinit/sources/__pycache__/DataSourceScaleway.cpython-36.pyc
3
�q�h�= � @ s: d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZ d dl m
Z
d dl m Z d dl
mZ d dl
mZ d dl
mZmZmZmZ d d lmZmZ d d
lmZ d d lmZm Z d d l m Z d d
l m Z ej!e"�Z#ddgZ$dZ%dZ&dZ'G dd� dej(j)�Z*dd� Z+dd� Z,G dd� dej-�Z.e.ej/ffgZ0dd � Z1dS ) � N)�urlparse)�ConnectionError)�HTTPConnection)� PoolManager)�dmi)�log)�net�sources�
url_helper�util)�
EventScope� EventType)�NoDHCPLeaseError)�EphemeralDHCPv4�EphemeralIPv6Network)�DataSourceHostname)�ProcessExecutionErrorzhttp://169.254.42.42zhttp://[fd00:42::42]� � �
c s* e Zd ZdZ� fdd�Zddd�Z� ZS )�SourceAddressAdapterzF
Adapter for requests to choose the local address to bind to.
c s || _ tt| �jf |� d S )N)�source_address�superr �__init__)�selfr �kwargs)� __class__� �(/usr/lib/python3.6/DataSourceScaleway.pyr , s z SourceAddressAdapter.__init__Fc C s0 t jtjtjdfg }t|||| j|d�| _d S )N� )Z num_pools�maxsize�blockr �socket_options)r Zdefault_socket_options�socketZ
SOL_SOCKETZ SO_REUSEPORTr r Z poolmanager)r Z connectionsr r! r" r r r �init_poolmanager0 s z%SourceAddressAdapter.init_poolmanager)F)�__name__�
__module__� __qualname__�__doc__r r$ �
__classcell__r r )r r r ' s r c C s` y&t j| d|d|dd� d�}tj|j�S t jk
rZ } z|jdkrHdS � W Y dd}~X nX dS )aQ
Retrieve user data or vendor data.
Scaleway user/vendor data API returns HTTP/404 if user/vendor data is not
set.
This function calls `url_helper.readurl` but instead of considering
HTTP/404 as an error that requires a retry, it considers it as empty
user/vendor data.
Also, be aware the user data/vendor API requires the source port to be
below 1024 to ensure the client is root (since non-root users can't bind
ports below 1024). If requests raises ConnectionError (EADDRINUSE), the
caller should retry to call this function on an other port.
Nr c S s |j dkot|jtjj� S )Ni� )�code�
isinstance�cause�requests�
exceptionsr )�_�excr r r �<lambda>X s
z%query_data_api_once.<locals>.<lambda>)�data�timeout�retriesZsessionZ exception_cbi� )r
�readurlr �
decode_binary�contents�UrlErrorr* )� api_addressr3 �requests_session�respr0 r r r �query_data_api_once= s
r< c
C s" �xt dt|d��D �] }y�tjd| |� tj� }d}yRt|�j}|}|d dkr^|dd� }tj |dtj
d�d d } | tj kr�d }W n t k
r� Y nX |j
d
t||fd �� t|||d �}
tjd
| � |
S tjk
�r } z$tjd| | � tjd� | } wW Y dd} ~ X qX qW | �dS )a/ Get user or vendor data.
Handle the retrying logic in case the source port is used.
Scaleway metadata service requires the source port of the client to
be a privileged port (<1024). This is done to ensure that only a
privileged user on the system can access the metadata service.
r r z*Trying to get %s data (bind on port %d)...z0.0.0.0r �[N)�protoz0::zhttp://)r )r3 r: z%s-data downloadedz%Error while trying to get %s data: %s� ���)�range�max�LOG�debugr- ZSessionr �netlocr# � getaddrinfo� IPPROTO_TCP�AF_INET6�
ValueErrorZmountr r< r
r8 �warning�timeZsleep)
Zapi_typer9 r4 r3 Zportr: Z localhostZ url_address�address�
addr_protor2 r0 �last_excr r r �query_data_apie s>
rO c s� e Zd ZdZejejejej hiZ
� fdd�Z dd� Z dd� Z
edd � �Zd
d � Zd d
� Zedd� �Zedd� �Zdd� Zdd� Zd dd�Zedd� �Zedd � �Z� ZS ) �DataSourceScaleway�Scalewayc s� t t| �j|||� tjtj|ddgi �g�| _t| jjdt ��| _
t| jjdt ��| _ t| jjdt
��| _d | _tj| _t| _d | _d | _d | _d| _d| jj� kr�| j| jd 7 _d S )NZ
datasourcerQ r4 r3 �max_waitT�
metadata_urls)r rP r r Z
mergemanydictZget_cfg_by_pathZds_cfg�int�get�DEF_MD_RETRIESr4 �DEF_MD_TIMEOUTr3 �DEF_MD_MAX_WAITrR �_fallback_interfacer �UNSET�_network_config� DS_BASE_URLSrS � userdata_url�vendordata_url�ephemeral_fixed_address�has_ipv4�keys)r Zsys_cfg�distro�paths)r r r r � s zDataSourceScaleway.__init__c C s| t j � }tj|| j| jdd�\}}|rZtjd|� |� d�| _|� d�| _|� d�| _ dS tjd|t
t j � | �� t �dS ) zO
Define metadata_url based upon api-metadata URL availability.
F)�urlsrR r3 Zconnect_synchronouslyz%s is reachablez/conf?format=jsonz/user_data/cloud-initz/vendor_data/cloud-initNz3Unable to reach api-metadata at %s after %s seconds) rK r
Z wait_for_urlrR r3 rC rD � metadata_urlr] r^ rT r )r rd Z
start_timeZ avail_urlr/ r r r �_set_metadata_url� s" z$DataSourceScaleway._set_metadata_urlc C sZ t j| j| j| jd�}tjtj|j ��| _
t d| j | j| j�| _
t d| j| j| j�| _d S )N)r3 r4 z user-dataz vendor-data)r
r5 re r3 r4 �json�loadsr r6 r7 �metadatarO r] Z userdata_rawr^ Zvendordata_raw)r r; r r r �_crawl_metadata� s z"DataSourceScaleway._crawl_metadatac C s> t jd�} | dkrdS tjjd�r&dS tj� }d|kr:dS dS )a
There are three ways to detect if you are on Scaleway:
* check DMI data: not yet implemented by Scaleway, but the check is
made to be future-proof.
* the initrd created the file /var/run/scaleway.
* "scaleway" is in the kernel cmdline.
zsystem-manufacturerrQ Tz/var/run/scalewayZscalewayN)r Z
read_dmi_data�os�path�existsr Z get_cmdline)Z vendor_nameZcmdliner r r � ds_detect� s
z DataSourceScaleway.ds_detectc C s� |dkrt jd|� g S g }x�|D ]~}t|�j}|d dkrH|dd � }tj|d tjd�d d }|tjkr�|dkr�||g7 }q"q"|tjkr"|dkr"||g7 }q"q"W |S )
N�ipv4�ipv6zInvalid IP version : %sr r= r )r> )ro rp r@ ) rC rD r rE r# rF rG ZAF_INETrH )r r> rd Z
filtered_urlsZurlrL rM r r r �_set_urls_on_ip_version� s$
z*DataSourceScaleway._set_urls_on_ip_versionc ! C s0 | j d krtj� | _ | jr�y`t| j| j ��H}tjtj d| j
| j fd� tjtj d| j d� |d | _
d| jd< W d Q R X W n> tttfk
r� } z tjtt|�� d| _W Y d d }~X nX | j�s,yRt| j ��> tjtj d | j
| j fd� tjtj d| j d� d
| jd< W d Q R X W n tk
�r* dS X d S ) Nz3Set api-metadata URL depending on IPv4 availability)�logfunc�msg�func�argszCrawl of metadata service)rr rs rt z
fixed-addressro Z
net_in_useFz3Set api-metadata URL depending on IPv6 availabilityrp T)rY r �find_fallback_nicr` r rb r Zlog_timerC rD rf rS rj r_ ri r r r Zlogexc�strr )r ro �er r r � _get_data
sT
z DataSourceScaleway._get_datac C s� | j dkr tjdtj� tj| _ | j tjkr2| j S | jdkrFtj� | _| jd dk�rFi }i }x�| jd D ]�}|d | j kr�d|d< dd d
�g|d < qjd |j
� kr�|d |d � d
|d � �f7 < n |d � d
|d � �f|d < |d dkrj|d dd�}d |j
� k�r |d |g7 < qj|g|d < qjW ||| j< d|d�| _ n�dd| j d�}ddig}| jd �r�|dd| jd d d| jd d d d d| jd d d �gd �g7 }||d < d!|gd"�| _ tj d#| j � | j S )$z`
Configure networking according to data received from the
metadata API.
Nz5Found None as cached _network_config. Resetting to %sZ
private_ipZ
public_ipsrL TZdhcp4z169.254.42.42/32z
62.210.0.1)�to�via�routesZ addresses�/�netmaskZfamilyZinet6�gatewayz::/0)r{ rz r )�versionZ ethernetsZphysicalz%s)�type�namer� rp Zstaticz::�0)Znetwork�prefixr )r� rL r~ r| �subnetsr )r� �configznetwork_config : %s) r[ rC rJ r rZ rY r rv ri r_ ra rD )r ZnetcfgZip_cfgZipZrouter� r r r �network_configM sV
z!DataSourceScaleway.network_configc C s d S )Nr )r r r r � launch_index� s z DataSourceScaleway.launch_indexc C s
| j d S )N�id)ri )r r r r �get_instance_id� s z"DataSourceScaleway.get_instance_idc C sb dd� | j d D �}d}t|�}x<| j jdg �D ]*}|j|�s@q0|j||d � jdd�� q0W |S )Nc S s g | ] }|d �qS )�keyr )�.0r� r r r �
<listcomp>� s z:DataSourceScaleway.get_public_ssh_keys.<locals>.<listcomp>Zssh_public_keyszAUTHORIZED_KEY=Ztagsr/ � )ri �lenrU �
startswith�append�replace)r Zssh_keysZakeypreZplen�tagr r r �get_public_ssh_keys� s
z&DataSourceScaleway.get_public_ssh_keysFc C s t | jd d�S )NZhostnameF)r ri )r ZfqdnZ
resolve_ipZ
metadata_onlyr r r � get_hostname� s z DataSourceScaleway.get_hostnamec C s d S )Nr )r r r r �availability_zone� s z$DataSourceScaleway.availability_zonec C s d S )Nr )r r r r �region� s zDataSourceScaleway.region)FFF)r% r& r' Zdsnamer ZNETWORKr
ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZdefault_update_eventsr rf rj � staticmethodrn rq ry �propertyr� r� r� r� r� r� r� r) r r )r r rP � s$
@ M
rP c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s r� )2rg rk r# rK Z urllib.parser r- Zrequests.exceptionsr Zurllib3.connectionr Zurllib3.poolmanagerr Z cloudinitr r Zloggingr r r
r Zcloudinit.eventr r
Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr r Zcloudinit.sourcesr Zcloudinit.subpr Z getLoggerr% rC r\ rV rX rW ZadaptersZ HTTPAdapterr r< rO Z
DataSourcerP ZDEP_FILESYSTEMr� r� r r r r �<module> s:
(4 #