3
\me? @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z ej
eZdZdZdgZG dd de
jZG dd d eZG d
d d eZG d d
d
Zdd Zdd ZdddZdddZ dd Z eZ ee
j ffgZ dd Z!dS ) N)
atomic_helpernetsourcessubputilziid-dsopennebulaZnobodyz
context.shc @ sF e Zd ZdZdd Zdd Zdd Zdd Zed
d Z dd
dZ
dS )DataSourceOpenNebulaZ
OpenNebulac C s. t jj| ||| d | _tjj|jd| _d S )NZ
opennebula)r
DataSource__init__seedospathjoinseed_dir)selfZsys_cfgdistropaths 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 | jrjt
|| j |d}n(|j drt
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| _| jtjkrZdS || _ |jd | _ | | _ |jd
| _ dS )Nz instance-id parseuser)asuserz/dev)r r z%s was not mountablezfound datasource in %sFmetadataZDSMODEr znetwork-interfacesuserdataT) DEFAULT_IIDDEFAULT_PARSEUSERZds_cfggetr extendfind_candidate_devsr r isdirread_context_disk_dirr
startswith functoolspartialr Zmount_cbNonContextDiskDirBrokenContextDiskDirZMountFailedErrorLOGwarningdebugZ
mergemanydictZ_determine_dsmoder r ZDSMODE_DISABLEDr
networkr Z userdata_raw)
r Zdefaultsresultsr
r Z
candidatesZcdevZpartially_applied_funcexcZmdr 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 ZdddZdd Zdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dddZ
dd ZdS ) OpenNebulaNetworkNc C s || _ |d krt|}tjdd t|j dd dD | _i | _x8|j D ],\}}tj d|}|rL|j
d| j|j < qLW d S )Nc S s g | ]}|qS r r ).0kr r r
s z.OpenNebulaNetwork.__init__..c S s t j| d S )N )r Znatural_sort_key)r7 r r r s z,OpenNebulaNetwork.__init__..)keyz
^(.+)_MAC$r9 ) contextget_physical_nics_by_mac collections OrderedDictsorteditemsifacescontext_devnamerematchgrouplower)r r< r Zsystem_nics_by_macr7 vmr r r r s zOpenNebulaNetwork.__init__c C s" dj dd |jddd D S )N.c S s g | ]}t t|d qS ) )strint)r6 cr r r r8 s z,OpenNebulaNetwork.mac2ip..: )r
split)r macr r r mac2ip s zOpenNebulaNetwork.mac2ipc C s\ i }| j |ddj }|j| jjddj |r:||d< | j |ddj }|rX||d< |S )Ndns ZDNS addresses
search_domainsearch) get_fieldrQ r r< r )r dev nameserversrT rW r r r get_nameservers s z!OpenNebulaNetwork.get_nameserversc C s | j |dS )Nmtu)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 )Nip6ip6_ula)rY append)r rZ
addresses6r` ra r r r get_ip6 s
zOpenNebulaNetwork.get_ip6c C s | j |ddS )NZip6_prefix_lengthZ64)rY )r rZ r r r get_ip6_prefix s z OpenNebulaNetwork.get_ip6_prefixc C s | j |dS )Ngateway)rY )r rZ r r r get_gateway s z OpenNebulaNetwork.get_gatewayc C s | j |d| j |dS )NZ ip6_gatewaygateway6)rY )r rZ r r r get_gateway6 s z OpenNebulaNetwork.get_gateway6c C s | j |ddS )Nmaskz
255.255.255.0)rY )r rZ r r r get_mask s zOpenNebulaNetwork.get_maskc C s* | j jdj||fj }|dkr&|S |S )zreturn the field name in context for device dev.
context stores _ (example: eth0_DOMAIN).
an empty string for value will return default._NrU )NrU )r< r r
upper)r rZ namedefaultvalr 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..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_withsortrb )Zcombinedfdevsdr r r r s
r c C s
dd| gS )NZsudoz-ur )userr r r switch_user_cmd) s r c C s t |tr|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| jdD ]}yT|jdd\}}||kr|||< n,||
kr>||ks6|j ||kr>|||< W q t
k
r |dkr^ |d krn|}n||kr||}Y qX qW |S ) Nbashz-ec S s g | ] }d | qS )z${!%s*}r )r6 xr r r r8 ; s z&parse_shell_config..rl
__v="";rU c S s dj dddj | dddfS )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..varprinterz
unset IFS
z{
%s
:
} > /dev/null
)data
EPOCHREALTIME EPOCHSECONDSRANDOMLINENOSECONDSSRANDOM__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 ZbcmdcmdoutputZ_errorZexcludedZpresetrettargetliner; 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|kr<|dk rytj| 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
|krjd
} nd|krxd} | r|j| j } dd | D |d d< x*d!D ]"}
|
|kr||
|d d< P qW d|kr|d |d< nd|kr|d |d< d|krZ|jd|jd }|d krZytj|d |d< W n tk
rX tjd Y nX d d |D }|rt||}|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..z public-keys SET_HOSTNAMEHOSTNAME PUBLIC_IP IP_PUBLICETH0_IPzlocal-hostnameZ USER_DATAr ZUSERDATAZUSERDATA_ENCODINGZUSER_DATA_ENCODINGbase64z"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% pwdgetpwnamKeyErrorr& formatr Z load_filer r ZProcessExecutionErrorIOErrorr
splitlinesr Zb64d TypeErrorr' r( r5 rw )Z
source_dirr r foundZaffnr< r+ er r Z ssh_key_varlinesr7 encodingZ ipaddr_keysZonetr r r r! s~
r! c s" t j }t fdd|j D S )Nc s$ g | ] \}} j j|r||fqS r )Z
networkingZ is_physical)r6 rI n)r r r r8 s z,get_physical_nics_by_mac..)r Zget_interfaces_by_macdictrA )r r} r )r r r= s r= c C s t j| tS )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 s0
b
j
`