Your IP : 216.73.217.13
U
�Ad�% � @ s* d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl m
Z
mZmZmZ d dlmZmZ d dlmZ d d lmZ d d
lmZ e �e�Z d Z e d e d
e d d�Z dZ dZ dd� Z!G dd� dej"j#�Z$dd� Z%dd� Z&G dd� dej'�Z(e(ej)ffgZ*d d � Z+dS ) � N)�HTTPConnection)� PoolManager)�dmi)�log)�net�sources�
url_helper�util)�
EventScope� EventType)�NoDHCPLeaseError)�EphemeralDHCPv4)�DataSourceHostnamezhttp://169.254.42.42z/conf?format=jsonz/user_data/cloud-initz/vendor_data/cloud-init)� metadata_url� userdata_url�vendordata_url� �
c C s> t �d�} | dkrdS tj�d�r&dS t�� }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-manufacturer�ScalewayTz/var/run/scalewayZscalewayF)r Z
read_dmi_data�os�path�existsr Z get_cmdline)Z vendor_nameZcmdline� r �F/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceScaleway.py� on_scaleway) s
r 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__r r r E 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_poolmanagerI s ��z%SourceAddressAdapter.init_poolmanager)F)�__name__�
__module__� __qualname__�__doc__r r) �
__classcell__r r r" r r @ s r c
C sh z(t j| d|d|dd� d�}t�|j�W S t jk
rb } z|jdkrPW Y �
dS � W 5 d}~X Y 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 )N� )�code�
isinstance�cause�requests�
exceptions�ConnectionError)�_�excr r r �<lambda>q s
z%query_data_api_once.<locals>.<lambda>)�data�timeout�retriesZsessionZ exception_cbr/ )r �readurlr �
decode_binary�contents�UrlErrorr0 )� api_addressr: �requests_session�respr7 r r r �query_data_api_onceV s �
rC c C s� t dt|d��D ]�}zPt�d| |� t�� }|�dtd|fd�� t|||d�}t�d| � |W S t j
k
r� } z(t� d | |� t �
d
� |}W Y �qW 5 d }~X Y qX q|�d S ) 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$ � z*Trying to get %s data (bind on port %d)...zhttp://z0.0.0.0)r )r: rA z%s-data downloadedz%Error while trying to get %s data: %sr N)�range�max�LOG�debugr3 ZSessionZmountr rC r r? �warning�time�sleep) Zapi_typer@ r; r: ZportrA r9 r7 �last_excr r r �query_data_api~ s2 � � �
rM c s� e Zd ZdZejejejej hiZ
� fdd�Z dd� Z dd� Z
edd � �Zed
d � �Zd d
� Zdd� Zddd�Zedd� �Zedd� �Z� ZS )�DataSourceScalewayr c s� t t| ��|||� t�t�|ddgi �tg�| _| jd | _| jd | _ | jd | _
t | j� dt
��| _t | j� dt��| _d | _tj| _d S )NZ
datasourcer r r r r; r: )r rN r r Z
mergemanydictZget_cfg_by_path�BUILTIN_DS_CONFIGZds_cfg�metadata_address�userdata_address�vendordata_address�int�get�DEF_MD_RETRIESr; �DEF_MD_TIMEOUTr: �_fallback_interfacer �UNSET�_network_config)r Zsys_cfg�distro�pathsr" r r r � s �� zDataSourceScaleway.__init__c C sZ t j| j| j| jd�}t�t�|j ��| _
t d| j | j| j�| _
t d| j| j| j�| _d S )N)r: r; z user-dataz vendor-data)r r< rP r: r; �json�loadsr r= r> �metadatarM rQ Z userdata_rawrR Zvendordata_raw)r rB r r r �_crawl_metadata� s$ � � �z"DataSourceScaleway._crawl_metadatac
C s� t � s
dS | jd kr t�� | _z8t| j| j�� d�� tjt j
d| j d� W 5 Q R X W n8 t k
r� } zt�
t t|�� W Y �dS d }~X Y nX dS )NF)Ztmp_dirzCrawl of metadata service)Zlogfunc�msg�funcT)r rW r �find_fallback_nicr
rZ Zget_tmp_exec_pathr Zlog_timerG rH r_ r Zlogexc�str)r �er r r � _get_data� s$
��z DataSourceScaleway._get_datac C s� | j dkr t�dtj� tj| _ | j tjkr2| j S | jdkrFt�� | _dd| j d�}ddig}| jd r�|d d| jd d
d| jd d d d
d| jd d d�gd�g7 }||d< d|gd�| _ | j S )z`
Configure networking according to data received from the
metadata API.
Nz5Found None as cached _network_config. Resetting to %sZphysicalz%s)�type�namerf Zdhcp4Zipv6Zstatic�address�netmaskz::�0�gateway)Znetwork�prefixrk )rf rh ri Zroutes�subnetsr$ )�versionZconfig) rY rG rI r rX rW r rb r^ )r Znetcfgrm r r r �network_config� s6
�
����z!DataSourceScaleway.network_configc C s d S r r �r r r r � launch_index s z DataSourceScaleway.launch_indexc C s
| j d S )N�id)r^ rp r r r �get_instance_id s z"DataSourceScaleway.get_instance_idc C s^ dd� | j d D �}d}t|�}| j �dg �D ]*}|�|�s>q.|�|d |� �dd�� q.|S )Nc S s g | ] }|d �qS )�keyr )�.0rt r r r �
<listcomp> s z:DataSourceScaleway.get_public_ssh_keys.<locals>.<listcomp>Zssh_public_keyszAUTHORIZED_KEY=Ztagsr6 � )r^ �lenrT �
startswith�append�replace)r Zssh_keysZakeypre�plen�tagr r r �get_public_ssh_keys s
z&DataSourceScaleway.get_public_ssh_keysFc C s t | jd d�S )NZhostnameF)r r^ )r ZfqdnZ
resolve_ipZ
metadata_onlyr r r � get_hostname s z DataSourceScaleway.get_hostnamec C s d S r r rp r r r �availability_zone! s z$DataSourceScaleway.availability_zonec C s d S r r rp r r r �region% s zDataSourceScaleway.region)FFF)r* r+ r, Zdsnamer
ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZdefault_update_eventsr r_ re �propertyro rq rs r~ r r� r� r. r r r" r rN � s* ��
'
rN c C s t �| t�S r )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list/ s r� ),r\ r r( rJ r3 Zurllib3.connectionr Zurllib3.poolmanagerr Z cloudinitr r Zloggingr r r r Zcloudinit.eventr
r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr
Zcloudinit.sourcesr Z getLoggerr* rG Z DS_BASE_URLrO rU rV r ZadaptersZ HTTPAdapterr rC rM Z
DataSourcerN ZDEP_FILESYSTEMr� r� r r r r �<module> s<
�(&
�