File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceOpenNebula.cpython-38.pyc
U
Ӈg�> � @ s
d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
m Z m Z m
Z
e�e�ZdZdZdgZdZG dd� de j�ZG d d
� d
e�ZG d d � d e�ZG d
d� d�Zdd� Zdd� Z dd� Z d dd�Z d dd�Z dd� Z eZ!ee j"ffgZ#dd � Z$dS ) � N)�
atomic_helper�net�sources�subp�utilziid-dsopennebulaZnobody�
context.sh)Z
EPOCHREALTIMEZ EPOCHSECONDSZRANDOMZLINENOZSECONDS�_ZSRANDOMZ__vc @ sF e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d � �Z dd
d�Z
dS )�DataSourceOpenNebulaZ
OpenNebulac C s4 t j�| |||� d | _tj�|jd�| _d | _d S )NZ
opennebula) r �
DataSource�__init__�seed�os�path�join�seed_dir�network)�selfZsys_cfg�distro�paths� r �H/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceOpenNebula.pyr / s z DataSourceOpenNebula.__init__c C s t j�| �}d|| j| jf S )Nz%s [seed=%s][dsmode=%s])r r
�__str__r �dsmode)r �rootr r r r 5 s z DataSourceOpenNebula.__str__c
C s� dt i}d }d }t}d| jkr*| j�d�}| jg}|�t� � |D ]�}zLtj� | j�rht
|| j |d�}n(|� d�r�t
jt
|| j d�}t�||�}W n\ tk
r� Y qBY nF tk
r� } z|�W 5 d }~X Y n" tjk
r� t�d|� Y nX |rB|}t�d|� �q qB|�sdS |d } t�| |g�} | �|�d
�| j�d �g�| _| jtjk�r\dS || _ |�d �| _ | | _ |�d
�| _ dS )Nz instance-id� parseuser��asuserz/dev)r r z%s was not mountablezfound datasource in %sF�metadataZDSMODEr �network-interfaces�userdataT) � DEFAULT_IID�DEFAULT_PARSEUSERZds_cfg�getr �extend�find_candidate_devsr
r �isdir�read_context_disk_dirr �
startswith� functools�partialr Zmount_cb�NonContextDiskDir�BrokenContextDiskDirZMountFailedError�LOG�warning�debugZ
mergemanydictZ_determine_dsmoder r ZDSMODE_DISABLEDr r r Z userdata_raw)
r �defaults�resultsr r Z
candidatesZcdevZpartially_applied_func�excZmdr r r � _get_data9 s^
�
� � z DataSourceOpenNebula._get_datac C s$ | j | jkrd}nd}d|| jf S )z/Return the subplatform metadata source details.zseed-dirz config-diskz%s (%s))r r )r Zsubplatform_typer r r �_get_subplatformv s z%DataSourceOpenNebula._get_subplatformc C s | j d k r| j S d S d S �N)r )r r r r �network_config~ s
z#DataSourceOpenNebula.network_configFc C s. |d kr | j tjkrd}nd}tj�| ||�S )NTF)r r ZDSMODE_NETWORKr
� get_hostname)r ZfqdnZ
resolve_ipZ
metadata_onlyr r r r6 � s
z!DataSourceOpenNebula.get_hostnameN)FFF) �__name__�
__module__� __qualname__Zdsnamer r r2 r3 �propertyr5 r6 r r r r r + s =
r c @ s e Zd ZdS )r* N�r7 r8 r9 r r r r r* � s r* c @ s e Zd ZdS )r+ Nr; r r r r r+ � s r+ c @ sp e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z ddd�Z
dd� ZdS ) �OpenNebulaNetworkNc C s| || _ |d krt|�}t�dd� t|�� dd� d�D ��| _i | _|�� D ],\}}t� d|�}|rJ|�
d�| j|� � < qJd S )Nc S s g | ]}|�qS r r ��.0�kr r r �
<listcomp>� s �z.OpenNebulaNetwork.__init__.<locals>.<listcomp>c S s t �| d �S )N� )r Znatural_sort_key)r? r r r �<lambda>� � z,OpenNebulaNetwork.__init__.<locals>.<lambda>)�keyz
^(.+)_MAC$rA ) �context�get_physical_nics_by_mac� collections� OrderedDict�sorted�items�ifaces�context_devname�re�match�group�lower)r rE r Zsystem_nics_by_macr? �v�mr r r r � s ��� zOpenNebulaNetwork.__init__c C s" d� dd� |�d�dd � D ��S )N�.c S s g | ]}t t|d ���qS )� )�str�int)r> �cr r r r@ � s z,OpenNebulaNetwork.mac2ip.<locals>.<listcomp>�:� )r �split)r �macr r r �mac2ip� s zOpenNebulaNetwork.mac2ipc C s\ i }| � |dd��� }|�| j�dd��� � |r:||d< | � |dd��� }|rX||d< |S )N�dns� ZDNS� addresses�
search_domain�search)� get_fieldrZ r# rE r" )r �dev� nameserversr] r` r r r �get_nameservers� s z!OpenNebulaNetwork.get_nameserversc C s | � |d�S )N�mtu�rb �r rc r r r �get_mtu� s zOpenNebulaNetwork.get_mtuc C s | � |d| �|��S )NZip)rb r\ )r rc r[ r r r �get_ip� s zOpenNebulaNetwork.get_ipc C s< g }| � |d�}|r |�|� | � |d�}|r8|�|� |S )N�ip6�ip6_ula)rb �append)r rc �
addresses6rk rl r r r �get_ip6� s
zOpenNebulaNetwork.get_ip6c C s | � |dd�S )NZip6_prefix_lengthZ64rg rh r r r �get_ip6_prefix� s z OpenNebulaNetwork.get_ip6_prefixc C s | � |d�S )N�gatewayrg rh r r r � get_gateway� s z OpenNebulaNetwork.get_gatewayc C s | � |d| � |d��S )NZ ip6_gateway�gateway6rg rh r r r � get_gateway6� s
�z OpenNebulaNetwork.get_gateway6c C s | � |dd�S )N�maskz
255.255.255.0rg rh r r r �get_mask� s zOpenNebulaNetwork.get_maskc C s* | j �d�||f��� �}|dkr&|S |S )z�return the field name in context for device dev.
context stores <dev>_<NAME> (example: eth0_DOMAIN).
an empty string for value will return default.r )Nr^ )rE r" r �upper)r rc �name�default�valr r r rb � s ��� zOpenNebulaNetwork.get_fieldc s6 i }d|d< i |d< i }| j �� D �]\}}|�� }| j�||�}i }d|i|d< g |d< | �|�}tt�|��}|d � | �
||�d | � | � |�} | r�| � |�� |d �
� fdd �| D �� | �|�}
|
r�|
|d
< | �|�} | r�| |d < | �|�} | �r| |d < | �|�}
|
�r |
|d
< |||< q"||d< |S )NrY �version� ethernetsZ
macaddressrN r_ �/c s g | ]}|d � �qS )r} r )r> �i�Zprefix6r r r@ s z.OpenNebulaNetwork.gen_conf.<locals>.<listcomp>Zgateway4rs rd rf )rK rJ rP rL r" rv rU r Zipv4_mask_to_net_prefixrm rj ro rp r# rr rt re ri )r Znetconfr| r[ rc Zc_devZdevconfru �prefixrn rq rs rd rf r r r �gen_conf� sD
�
zOpenNebulaNetwork.gen_conf)N)N)r7 r8 r9 r r\ re ri rj ro rp rr rt rv rb r� r r r r r< � s
r< c C s@ g } dD ]2}t �|�}|�� |D ]}|| kr"| �|� q"q| S )zE
Return a list of devices that may contain the context disk.
)z
LABEL=CONTEXTz LABEL=CDROMz TYPE=iso9660)r Zfind_devs_with�sortrm )Zcombined�f�devs�dr r r r$ & s
r$ c C s
dd| gS )NZsudoz-ur ��userr r r �switch_user_cmd5 s r� c C s
t �d�S )zDprint the shell environment variables within delimiters to be parsedz�
printf "%s\0" _start_
[ $0 != 'sh' ] && set -o posix
set
[ $0 != 'sh' ] && set +o posix
printf "%s\0" _start_
)�textwrap�dedentr r r r �
varprinter9 s �r� c
C s d| � � krt�d� t� d| t� d }g }|dk rBt|�}|�ddg� tj||d�j}i }|�d �d
d� }|d |d }}t d
d� t
�|�D ��} t dd� t
�|�D ��}
t | � � ��
t |
� � ��D ]6} | tkr�q�|
�| �} | dk r�| | �| �kr�| || < q�|S )a< run content and return environment variables which changed
WARNING: the special variable _start_ is used to delimit content
a context.sh that defines this variable might break in unexpected
ways
compatible with posix shells such as dash and ash and any shell
which supports `set -o posix`
s _start_ zYUser defined _start_ variable in context.sh, this may breakcloud-init in unexpected ways.z{
%s
:
} > /dev/null
�
NZshz-e)�dataz_start_ rA r rY c s s | ]}|j d dd�V qdS ��=rA )�maxsplitN�rZ �r> Zvariabler r r � <genexpr>} s z%parse_shell_config.<locals>.<genexpr>c s s | ]}|j d dd�V qdS r� r� r� r r r r� � s )�encoder, r- r� r� r# r �stdoutrZ �dict�shlex�set�keys�union�
EXCLUDED_VARSr" )
�contentr Zbcmd�cmd�outputZretZsectionsZbeforeZafterZpre_envZpost_envrD �valuer r r �parse_shell_configF sB �������
r� c
C s� i }t D ]&}tj�| |�}tj�|�r|||< q|sDtd| df ��i }di d�}d|k�r8|dk r�zt�|� W n4 tk
r� } zt dj
|d��|�W 5 d}~X Y nX z(tj�| d�} t � | �}
t
|
|d�}W nd tjk
�r } zt d | �|�W 5 d}~X Y n2 tk
�r4 } ztd
| �|�W 5 d}~X Y nX ntd ��|�sJ|S ||d < d} d
|k�rfd
} nd|k�rtd} | �r�|�| ��� } dd� | D �|d d< dD ]&}
|
|k�r�||
|d d< �qʐq�d|k�r�|d |d< nd|k�r�|d |d< d|k�rV|�d|�d��}|dk�rVzt�|d �|d< W n tk
�rT t�d� Y nX dd� |D �}|�r�t||�}|�� |d < |S ) z�
read_context_disk_dir(source_dir):
read source_dir and return a tuple with metadata dict and user-data
string populated. If not a valid dir, raise a NonContextDiskDir
z%s: %szno files foundN)r r r z'configured user '{user}' does not existr� r z Error processing context.sh: %sz Error reading context.sh: %szMissing context.shr ZSSH_KEYZSSH_PUBLIC_KEYc S s" g | ]}t |�r|�d �s|�qS )�#)�lenr' )r> �liner r r r@ � s
z)read_context_disk_dir.<locals>.<listcomp>z public-keys)Z SET_HOSTNAMEZHOSTNAMEZ PUBLIC_IPZ IP_PUBLICZETH0_IPzlocal-hostnameZ USER_DATAr ZUSERDATAZUSERDATA_ENCODINGZUSER_DATA_ENCODING�base64z"Failed base64 decoding of userdatac S s g | ]}t �d |�r|�qS )z
^ETH\d+_IP.*$)rM rN r= r r r r@ � s r )�CONTEXT_DISK_FILESr
r r �isfiler* �pwd�getpwnam�KeyErrorr+ �formatr Zload_text_filer� r ZProcessExecutionError�IOErrorr" �
splitlinesr Zb64d� TypeErrorr, r- r< r� )Z
source_dirr r �foundZaf�fnrE r0 �er r� Z ssh_key_var�linesr? �encodingZ ipaddr_keysZonetr r r r& � s�
���
����
�
�
r& c s" t �� }t� fdd�|�� D ��S )Nc s$ g | ] \}}� j �|�r||f�qS r )Z
networkingZ is_physical)r> rR �n�r r r r@ � s z,get_physical_nics_by_mac.<locals>.<listcomp>)r Zget_interfaces_by_macr� rJ )r r� r r� r rF � s �rF c C s t �| t�S r4 )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s r� )N)N)%rG r( Zloggingr
r� rM r� r� Z cloudinitr r r r r Z getLoggerr7 r, r r! r� r� r
r � Exceptionr* r+ r< r$ r� r� r� r& rF ZDataSourceOpenNebulaNetZDEP_FILESYSTEMr� r� r r r r �<module> s8
c
G
`
�