CasperSecurity
o
쑛h� � @ s� d Z ddlZddlm m mZ ddlmZm Z mZm
Z
ddl m Z ddl
mZ e�e�ZdZed ed d �Zd
Zd Zd ZG d d
� d
ej�Zdd� ZeejffgZdd� Z dS )z;Hetzner Cloud API Documentation
https://docs.hetzner.cloud/� N)�dmi�net�sources�util)�NoDHCPLeaseError)�EphemeralDHCPv4z!http://169.254.169.254/hetzner/v1z /metadataz /userdata)� metadata_url� userdata_url�< � c @ s4 e Zd ZdZdd� Zdd� Zdd� Zedd � �Zd
S ) �DataSourceHetzner�Hetznerc C s� t j�| |||� || _t� | _t�t�|ddgi �t g�| _
| j
d | _ | j
d | _ | j
�
dt�| _| j
�
dt�| _| j
�
dt�| _t j| _t j| _d | _d S )N�
datasourcer
r r �retries�timeout�
wait_retry)r �
DataSource�__init__�distro�dict�metadatar �
mergemanydict�get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg�metadata_address�userdata_address�get�
MD_RETRIESr �
MD_TIMEOUTr �
MD_WAIT_RETRYr �UNSET�_network_config�DSMODE_NETWORK�dsmode�
metadata_full)�self�sys_cfgr �paths� r) �E/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceHetzner.pyr " s ��
zDataSourceHetzner.__init__c
C sP t � \}}|s dS zJt| jt�� dtd igd��1 tj| j| j | j
| j d�}tj | j
| j | j
| j d�}tj| jd | j | j
| j d�}W d � n1 sMw Y W n tyf } zt�d|� � d }~ww t�|�| _|| _|d | jd< |d | jd
< |�d d �| jd < |�d d �| jd < || jd
< |�dd �| _| �� |kr�td��dS )NF�urlz/metadata/instance-id)�iface�connectivity_urls_data)r � sec_betweenr z/private-networkszBailing, DHCP Exception: %sz instance-id�hostnamezlocal-hostname�network-configz public-keyszprivate-networks� vendor_dataz6SMBIOS serial does not match instance ID from metadataT)�get_hcloud_datar r r �find_fallback_nic� BASE_URL_V1� hc_helper�
read_metadatar r r r �
read_userdatar r �LOG�errorr �maybe_b64decode� userdata_rawr% r r �vendordata_raw�get_instance_id� RuntimeError)r& �
on_hetzner�serial�md�ud�pn�er) r) r* � _get_data5 sb
��� ��� �� ��
�zDataSourceHetzner._get_datac C s t �| �� d�S )N�system-serial-number)r � instance_id_matches_system_uuidr= )r& r' r) r) r* �check_instance_idw s �z#DataSourceHetzner.check_instance_idc C sT | j du rt�dtj� tj| _ | j tjkr| j S | jd }|s$td��|| _ | j S )z�Configure the networking. This needs to be done each boot, since
the IP information may have changed due to snapshot and/or
migration.
Nz5Found None as cached _network_config. Resetting to %sr0 z'Unable to get meta-data from server....)r" r8 �warningr r! r r> )r& � _net_configr) r) r* �network_config| s
�
z DataSourceHetzner.network_configN) �__name__�
__module__� __qualname__�dsnamer rE rH �propertyrK r) r) r) r* r s Br c C s@ t �d�} | dkr dS t �d�}|r t�d|� d|fS td��)Nzsystem-manufacturerr
)FNrF z#Running on Hetzner Cloud: serial=%sz+Hetzner Cloud detected, but no serial foundT)r �
read_dmi_datar8 �debugr> )� vendor_namer@ r) r) r* r2 � s
�r2 c C s t �| t�S )N)r �list_from_depends� datasources)�dependsr) r) r* �get_datasource_list� s rW ) �__doc__�logging�!cloudinit.sources.helpers.hetznerr �helpers�hetznerr5 � cloudinitr r r �cloudinit.net.dhcpr �cloudinit.net.ephemeralr � getLoggerrL r8 r4 r r r r r r r2 �DEP_FILESYSTEMrU rW r) r) r) r* �<module> s&
�x
�