CasperSecurity
o
쑛h� � @ s� d dl Z d dlmZ d dlmZmZmZmZmZm Z d dl
m Z d dl m
Z
e �e�ZdZded iZd Zd
Zd
ZG d d � d ej�Zd
d� Zdd� Zddd�ZeejffgZ dS )� N)�
exceptions)�dmi�net�sources�subp�
url_helper�util)�NoDHCPLeaseError)�EphemeralDHCPv4z http://169.254.169.254/api/v1� metadata_urlz /metadata� � c sZ e Zd ZdZdd� Zdeddf� fdd� Zd d
� Zd d � Ze d
d� �Z
e dd� �Z � Z
S )�DataSourceNWCS�NWCSc C s� t j�| |||� || _t� | _t�t�|ddgi �t g�| _
| j
d | _ | j
� dt
�| _| j
� dt�| _| j
� dt�| _t j| _t j| _d | _d S )N�
datasourcer r �retries�timeout�
wait_retry)r �
DataSource�__init__�distro�dict�metadatar �
mergemanydict�get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg�metadata_address�get�
MD_RETRIESr �
MD_TIMEOUTr �
MD_WAIT_RETRYr �UNSET�_network_config�DSMODE_NETWORK�dsmode�
metadata_full)�self�sys_cfgr �paths� r* �B/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceNWCS.pyr s ��
zDataSourceNWCS.__init__�ci_pkl_version�returnNc s" t � �|� | jstj| _d S d S �N)�super� _unpickler# r r" )r' r, �� __class__r* r+ r0 0 s �zDataSourceNWCS._unpicklec C sv | � � }|d u r td��|| _|d | jd< |d | jd< |d | jd< |d | jd< |�dd �| _|�dd �| _d S )
Nzfailed to get metadataz instance-idz public-keys�network�hostnamezlocal-hostname�userdata�
vendordataT)� get_metadata� RuntimeErrorr& r r � userdata_raw�vendordata_raw)r' �mdr* r* r+ � _get_data5 s zDataSourceNWCS._get_datac C s� z4t �d� t| jt�� dtd igd�� t| j| j | j
| j d�W d � W S 1 s-w Y W d S t t
jttjfyM } zt �d|� � d }~ww )Nz#Attempting to get metadata via DHCP�urlz/metadata/instance-id)�iface�connectivity_urls_data�r � sec_betweenr zDHCP failure: %s)�LOG�infor
r r �find_fallback_nic� BASE_URL_V1�
read_metadatar r r r r r �ProcessExecutionErrorr8 r �RequestException�error)r' �er* r* r+ r7 H s4
��� �(� � ��zDataSourceNWCS.get_metadatac C s� t �d� | jtjkr| jS | jd d std��| jd d D ]}t|d �}|r7t �d|d |� ||d< q | jd | _| jS )Nz Attempting network configurationr3 �configz"Unable to get metadata from server� mac_addresszOverriding %s with %s�name) rB �debugr# r r" r r8 �get_interface_namerC )r' �i�
iface_namer* r* r+ �network_confige s
� z DataSourceNWCS.network_configc C s dt �d�kS )Nr zsystem-manufacturer)r �
read_dmi_datar* r* r* r+ � ds_detect{ s zDataSourceNWCS.ds_detect)�__name__�
__module__� __qualname__�dsnamer �intr0 r< r7 �propertyrR � staticmethodrT �
__classcell__r* r* r1 r+ r s
r c C s t �� }| |vr
d S |�| �S r. )r �get_interfaces_by_macr )�mac� macs_to_nicr* r* r+ rO � s
rO c C s t �| t�S r. )r �list_from_depends� datasources)�dependsr* r* r+ �get_datasource_list� s rc � c C s6 t j| |||d�}|�� std| ��t�|j�� �S )Nr@ z unable to read metadata at %s)r �readurl�okr8 r � load_json�contents�decode)r= r rA r �responser* r* r+ rF � s � rF )rd rd r ) �logging�requestsr � cloudinitr r r r r r �cloudinit.net.dhcpr �cloudinit.net.ephemeralr
� getLoggerrU rB rE r r r r! r r rO rc rF �DEP_FILESYSTEMra r* r* r* r+ �<module> s$
�f
�