3
\�me? � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z ej
e�ZdZdZdgZG dd� de
j�ZG dd � d e�ZG d
d � d e�ZG d d
� d
�Zdd� Zdd� Zddd�Zddd�Z dd� Z eZ ee
j ffgZ dd� Z!dS ) � N)�
atomic_helper�net�sources�subp�utilziid-dsopennebulaZnobodyz
context.shc @ sF e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d � �Z dd
d�Z
dS )�DataSourceOpenNebulaZ
OpenNebulac C s. t jj| |||� d | _tjj|jd�| _d S )NZ
opennebula)r �
DataSource�__init__�seed�os�path�join�seed_dir)�selfZsys_cfg�distro�paths� r �*/usr/lib/python3.6/DataSourceOpenNebula.pyr $ s z DataSourceOpenNebula.__init__c C s t jj| �}d|| j| jf S )Nz%s [seed=%s][dsmode=%s])r r �__str__r
�dsmode)r �rootr r r r ) s z DataSourceOpenNebula.__str__c
C s� dt i}d }d }t}d| jkr*| jjd�}| jg}|jt� � x�|D ]�}yLtjj | j�rjt
|| j |d�}n(|j d�r�t
jt
|| j d�}tj||�}W nZ tk
r� wDY nF tk
r� } z
|�W Y d d }~X n" tjk
r� tjd|� Y nX |rD|}tjd|� P qDW |�sdS |d } tj| |g�} | j|jd
�| jjd �g�| _| jtjk�rZdS || _ |jd �| _ | | _ |jd
�| _ dS )Nz instance-id� parseuser)�asuserz/dev)r r z%s was not mountablezfound datasource in %sF�metadataZDSMODEr znetwork-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
�networkr Z userdata_raw)
r Zdefaults�resultsr
r Z
candidatesZcdevZpartially_applied_func�excZmdr r r � _get_data- sT
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_subplatformj s z%DataSourceOpenNebula._get_subplatformc C s | j d k r| j S d S d S )N)r* )r r r r �network_configr s
z#DataSourceOpenNebula.network_configFc C s. |d kr | j tjkrd}nd}tjj| ||�S )NTF)r r ZDSMODE_NETWORKr � get_hostname)r ZfqdnZ
resolve_ipZ
metadata_onlyr r r r0 y s
z!DataSourceOpenNebula.get_hostnameN)FFF) �__name__�
__module__� __qualname__Zdsnamer r r- r. �propertyr/ r0 r r r r r s = r c @ s e Zd ZdS )r% N)r1 r2 r3 r r r r r% � s r% c @ s e Zd ZdS )r& N)r1 r2 r3 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|�}tjdd� t|j� dd� d�D ��| _i | _x8|j� D ],\}}tj d|�}|rL|j
d�| j|j � < qLW d S )Nc S s g | ]}|�qS r r )�.0�kr r r �
<listcomp>� s z.OpenNebulaNetwork.__init__.<locals>.<listcomp>c S s t j| d �S )N� )r Znatural_sort_key)r7 r r r �<lambda>� s z,OpenNebulaNetwork.__init__.<locals>.<lambda>)�keyz
^(.+)_MAC$r9 ) �context�get_physical_nics_by_mac� collections� OrderedDict�sorted�items�ifaces�context_devname�re�match�group�lower)r r< r Zsystem_nics_by_macr7 �v�mr r r r � s zOpenNebulaNetwork.__init__c C s" dj dd� |jd�dd � D ��S )N�.c S s g | ]}t t|d ���qS )� )�str�int)r6 �cr r r r8 � s z,OpenNebulaNetwork.mac2ip.<locals>.<listcomp>�:� )r
�split)r �macr r r �mac2ip� s zOpenNebulaNetwork.mac2ipc C s\ i }| j |dd�j� }|j| jjdd�j� � |r:||d< | j |dd�j� }|rX||d< |S )N�dns� ZDNS� addresses�
search_domain�search)� get_fieldrQ r r< r )r �dev� nameserversrT rW r r r �get_nameservers� s z!OpenNebulaNetwork.get_nameserversc C s | j |d�S )N�mtu)rY )r rZ r r r �get_mtu� s zOpenNebulaNetwork.get_mtuc C s | j |d| j|��S )NZip)rY rS )r rZ rR r r r �get_ip� s zOpenNebulaNetwork.get_ipc C s< g }| j |d�}|r |j|� | j |d�}|r8|j|� |S )N�ip6�ip6_ula)rY �append)r rZ �
addresses6r` ra r r r �get_ip6� s
zOpenNebulaNetwork.get_ip6c C s | j |dd�S )NZip6_prefix_lengthZ64)rY )r rZ r r r �get_ip6_prefix� s z OpenNebulaNetwork.get_ip6_prefixc C s | j |d�S )N�gateway)rY )r rZ r r r � get_gateway� s z OpenNebulaNetwork.get_gatewayc C s | j |d| j |d��S )NZ ip6_gateway�gateway6)rY )r rZ r r r � get_gateway6� s z OpenNebulaNetwork.get_gateway6c C s | j |dd�S )N�maskz
255.255.255.0)rY )r rZ r r r �get_mask� s zOpenNebulaNetwork.get_maskc C s* | j jdj||f�j� �}|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.�_NrU )NrU )r< r r
�upper)r rZ �name�default�valr r r rY � s
zOpenNebulaNetwork.get_fieldc s< i }d|d< i |d< i }�x| j j� D �]\}}|j� }| jj||�}i }d|i|d< g |d< | j|�}ttj|��}|d j | j
||�d | � | j |�} | r�| j |�� |d j
� fdd �| D �� | j|�}
|
r�|
|d
< | j|�} | r�| |d < | j|�} | �r | |d < | j|�}
|
�r$|
|d
< |||< q&W ||d< |S )NrP �version� ethernetsZ
macaddressrE rV �/c s g | ]}|d � �qS )rs r )r6 �i)�prefix6r r r8 � s z.OpenNebulaNetwork.gen_conf.<locals>.<listcomp>Zgateway4rh r[ r] )rB rA rG rC r rk rL r Zipv4_mask_to_net_prefixrb r_ rd re r rg ri r\ r^ )r Znetconfrr rR rZ Zc_devZdevconfrj �prefixrc rf rh r[ r] r )ru r �gen_conf� sB
zOpenNebulaNetwork.gen_conf)N)N)r1 r2 r3 r rS r\ r^ r_ rd re rg ri rk rY rw r r r r r5 � s
r5 c C sH g } x>dD ]6}t j|�}|j� x |D ]}|| kr&| j|� q&W q
W | S )zE
Return a list of devices that may contain the context disk.
�
LABEL=CONTEXT� LABEL=CDROM� TYPE=iso9660)rx ry rz )r Zfind_devs_with�sortrb )Zcombined�f�devs�dr r r r s
r c C s
dd| gS )NZsudoz-ur )�userr r r �switch_user_cmd) s r� c C s� t |t�r|g}n|d kr"ddg}|d kr.t}dd� tjd D �}|}|d krV|}g }djd�}d d
� }d | ||� d | d ||� d } g }
|d k r�||�}
|
j|� tj|
| d
�\} } d}
i }i }d }| dd � } x�| jd�D ]�}yT|jdd�\}}||k�r|||< n,||
k�r>||k�s6|j |�|k�r>|||< W q� t
k
�r� |dk�r^� |d k�rn|}n||k�r||}Y q�X q�W |S ) N�bashz-ec S s g | ] }d | �qS )z${!%s*}r )r6 �xr r r r8 ; s z&parse_shell_config.<locals>.<listcomp>rl �
�__v="";rU c S s dj dddj | � dddf�S )Nr� zprintf "%s\0" _start_zfor __v in %s; do� z% printf "%s=%s\0" "$__v" "${!__v}";�donerU )r
)Zvlistr r r �
varprinterI s z&parse_shell_config.<locals>.varprinterz
unset IFS
z{
%s
:
} > /dev/null
)�data�
EPOCHREALTIME� EPOCHSECONDS�RANDOM�LINENO�SECONDS�SRANDOM�__vr r9 � �=Z_start_)r� rU )r� r� r� r� r� rl r� r� ���) �
isinstancerL r� �stringZ
ascii_lettersr
r r rQ r �
ValueError)�contentZkeylistr� r Zswitch_user_cbZallvarsZ
keylist_inZsetupr� Zbcmd�cmd�outputZ_errorZexcludedZpreset�ret�target�liner; rp r r r �parse_shell_config- sb
(
r� c ' C s� i }x.t D ]&}tjj| |�}tjj|�r
|||< q
W |sHtd| df ��i }di d�}d|k�r<|dk r�ytj|� W n4 tk
r� } zt dj
|d��|�W Y dd}~X nX y(tjj| d�} t j | �}
t
|
|d�}W nd tjk
�r } zt d | �|�W Y dd}~X n2 tk
�r8 } ztd
| �|�W Y dd}~X nX ntd ��|�sN|S ||d < d} d
|k�rjd
} nd|k�rxd} | �r�|j| �j� } dd� | D �|d d< x*d!D ]"}
|
|k�r�||
|d d< P �q�W d|k�r�|d |d< nd|k�r�|d |d< d|k�rZ|jd|jd ��}|d k�rZytj|d �|d< W n tk
�rX tjd � Y nX d d� |D �}|�r�t||�}|j� |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 z
context.shz'configured user '{user}' does not exist)r )r z Error processing context.sh: %sz Error reading context.sh: %szMissing context.shr ZSSH_KEYZSSH_PUBLIC_KEYc S s$ g | ] }t |�r|jd � r|�qS )�#)�lenr" )r6 r� r r r r8 � s z)read_context_disk_dir.<locals>.<listcomp>z public-keys� SET_HOSTNAME�HOSTNAME� PUBLIC_IP� IP_PUBLIC�ETH0_IPzlocal-hostnameZ USER_DATAr ZUSERDATAZUSERDATA_ENCODINGZUSER_DATA_ENCODING�base64z"Failed base64 decoding of userdatac S s g | ]}t jd |�r|�qS )z
^ETH\d+_IP.*$)rD rE )r6 r7 r r r r8 � s znetwork-interfaces)r� r� r� r� r� )�CONTEXT_DISK_FILESr r r
�isfiler% �pwd�getpwnam�KeyErrorr& �formatr Z load_filer� r ZProcessExecutionError�IOErrorr �
splitlinesr Zb64d� TypeErrorr' r( r5 rw )Z
source_dirr r �foundZaf�fnr< r+ �er r� Z ssh_key_var�linesr7 �encodingZ ipaddr_keysZonetr r r r! � s~
r! c s" t j� }t� fdd�|j� D ��S )Nc s$ g | ] \}}� j j|�r||f�qS r )Z
networkingZ is_physical)r6 rI �n)r r r r8 � s z,get_physical_nics_by_mac.<locals>.<listcomp>)r Zget_interfaces_by_mac�dictrA )r r} r )r r r= � s r= c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list s r� )NNNN)N)"r> r# Zloggingr r� rD r� Z cloudinitr r r r r Z getLoggerr1 r' r r r� r r � Exceptionr% r& r5 r r� r� r! r= ZDataSourceOpenNebulaNetZDEP_FILESYSTEMr� r� r r r r �<module> s0
b
j
`