File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceScaleway.cpython-310.pyc
o
�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�} | dkr dS tj�d�rdS t�� }d|v r 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/scaleway�scalewayF)r �
read_dmi_data�os�path�existsr � get_cmdline)� vendor_name�cmdline� 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| �jdi |�� d S )Nr )�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� )� num_pools�maxsize�blockr! �socket_options)r �default_socket_options�socket�
SOL_SOCKET� SO_REUSEPORTr r! � poolmanager)r$ � connectionsr* r+ r, r r r �init_poolmanagerI s � �z%SourceAddressAdapter.init_poolmanager)F)�__name__�
__module__� __qualname__�__doc__r# r3 �
__classcell__r r r& r r @ s r c
C s^ zt j| d|d|dd� d�}t�|j�W S t jy. } z
|jdkr)W Y d}~dS � d}~ww )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 dko
t|jtjj� S )N� )�code�
isinstance�cause�requests�
exceptions�ConnectionError)�_�excr r r �<lambda>q s
z%query_data_api_once.<locals>.<lambda>)�data�timeout�retries�session� exception_cbr9 )r �readurlr �
decode_binary�contents�UrlErrorr: )� api_addressrD �requests_session�resprA r r r �query_data_api_onceV s �
��rO c C s� t dt|d��D ]K}z(t�d| |� t�� }|�dtd|fd�� t|||d�}t�d| � |W S t j
yS } zt� d | |� t �
d
� |}W Y d }~qd }~ww |�) 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! )rD rM z%s-data downloadedz%Error while trying to get %s data: %sr N)�range�max�LOG�debugr= �Session�mountr rO r rK �warning�time�sleep) �api_typerL rE rD �portrM rC rA �last_excr r r �query_data_api~ s. � ��
��r] 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 )N�
datasourcer r r r rE rD )r" r^ r# r �
mergemanydict�get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg�metadata_address�userdata_address�vendordata_address�int�get�DEF_MD_RETRIESrE �DEF_MD_TIMEOUTrD �_fallback_interfacer �UNSET�_network_config)r$ �sys_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)rD rE z user-dataz vendor-data)r rH rd rD rE �json�loadsr rI rJ �metadatar] re � userdata_rawrf �vendordata_raw)r$ rN r r r �_crawl_metadata� s ��
�z"DataSourceScaleway._crawl_metadatac
C s� t � sdS | jd u rt�� | _z)t| j| j�� d�� tjt j
d| j d� W d � W dS 1 s1w Y W dS t yS } zt�
t t|�� W Y d }~dS d }~ww )NF)�tmp_dirzCrawl of metadata service)�logfunc�msg�funcT)r rk r �find_fallback_nicr
ro �get_tmp_exec_pathr �log_timerS rT rv r �logexc�str)r$ �er r r � _get_data� s.
�� � � ���z DataSourceScaleway._get_datac C s� | j du rt�dtj� tj| _ | j tjkr| j S | jdu r#t�� | _dd| j d�}ddig}| jd rY|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 %s�physicalz%s)�type�namer� �dhcp4�ipv6�static�address�netmaskz::�0�gateway)�network�prefixr� )r� r� r� �routes�subnetsr( )�version�config) rm rS rW r rl rk r r{ rs )r$ �netcfgr� r r r �network_config� s6
�
����z!DataSourceScaleway.network_configc C � d S �Nr �r$ r r r � launch_index � z DataSourceScaleway.launch_indexc C s
| j d S )N�id)rs r� 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 )�.0r� r r r �
<listcomp> s z:DataSourceScaleway.get_public_ssh_keys.<locals>.<listcomp>�ssh_public_keyszAUTHORIZED_KEY=�tagsr@ � )rs �lenrh �
startswith�append�replace)r$ �ssh_keys�akeypre�plen�tagr r r �get_public_ssh_keys s
z&DataSourceScaleway.get_public_ssh_keysFc C s t | jd d�S )N�hostnameF)r rs )r$ �fqdn�
resolve_ip�
metadata_onlyr r r � get_hostname s z DataSourceScaleway.get_hostnamec C r� r� r r� r r r �availability_zone! r� z$DataSourceScaleway.availability_zonec C r� r� r r� r r r �region% r� zDataSourceScaleway.region)FFF)r4 r5 r6 �dsnamer
�NETWORKr �BOOT_NEW_INSTANCE�BOOT� BOOT_LEGACY�default_update_eventsr# rv r� �propertyr� r� r� r� r� r� r� r8 r r r& r r^ � s, ��
'
r^ c C s t �| t�S r� )r �list_from_depends� datasources)�dependsr r r �get_datasource_list/ s r� ),rq r r. rX r= �urllib3.connectionr �urllib3.poolmanagerr � cloudinitr r �loggingr r r r �cloudinit.eventr
r �cloudinit.net.dhcpr �cloudinit.net.ephemeralr
�cloudinit.sourcesr � getLoggerr4 rS � DS_BASE_URLrb ri rj r �adapters� HTTPAdapterr rO r] �
DataSourcer^ �DEP_FILESYSTEMr� r� r r r r �<module> s>
�(&
�