3
cjZ @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z ddl m
Z
mZ ejeZdZdd Zeed d
d Zeje d d
dZdd Zdd Zdd Zdd Z dd Z G dd de j Z e e j ffgZ!dd Z"dS ) zr
This file contains code used to gather the user data passed to an
instance on rootbox / hyperone cloud platforms
N) IPv4Address)sourcessubputil)
EventScope EventTypez
/etc/hostsc C s. t jtdd} | r tjd dS tjd dS )NT)quietz5/etc/hosts exists - setting manage_etc_hosts to FalseFz=/etc/hosts does not exists - setting manage_etc_hosts to True)r load_file ETC_HOSTSLOGdebug)Zhosts r
(/usr/lib/python3.6/DataSourceRbxCloud.pyget_manage_etc_hosts s
r )increturnc C s t ttt| | S )N)strr int)addrr r
r
r increment_ip# s r )r c C s | t | dt | dgS )zReturn a list of 3 IP addresses: [addr, addr + 2, addr + 3]
@param addr: an object that is passed to IPvAddress
@return: list of strings
)r )r r
r
r
get_three_ips' s r c C s t j dg| S )z
Uses the preferred cloud-init subprocess def of subp.subp
and runs arping. Breaking this to a separate function
for later use in mocking and unittests
Zarping)r )cmdr
r
r _sub_arp4 s r c C s d}|j d
krd}xh| D ]`}y tdd ||d
|d g W q tjk
rv } z tjd |d
|d | W Y d d }~X qX qW d S )Nz-S almalinuxfedoracentos
cloudlinuxrhelz-sz-c2source destinationz&Failed to arping from "%s" to "%s": %s)r r r r r )namer r ZProcessExecutionErrorr Zwarning)itemsdistroZ source_paramitemerrorr
r
r gratuitous_arp= s
r( c C s t tjdtjd } | s dS x| D ]z}y"tj|tdddgd}|rJ|S W q& tk
r| } z|jtjkrl W Y dd}~X q& tjk
r tj t
d | Y q&X q&W t
j d
d j | dS ) zAReturns False (not found or error) or a dictionary with metadata.z
LABEL=CLOUDMDz
LABEL=cloudmdFZvfatZfatZmsdosfs)devicecallbackZmtypeNz-Failed to mount %s when looking for user dataz0Did not find RbxCloud data, searched devices: %s,)
setr Zfind_devs_withZmount_cbread_user_data_callbackOSErrorerrnoENOENTZMountFailedErrorlogexcr r join)Zdevicesr) rbx_dataerrr
r
r get_mdR s,
r5 c C s ddd t | D dS )zGenerate network configuration
@param netadps: A list of network adapter settings
@returns: A dict containing network config
c sB g | ]:\} d dj t| d j fdd d D dqS )Zphysicalzeth{}Z
macaddressc s> g | ]6}d |d d d d d d d d d dqS ) ZstaticaddressnetworknetmaskautogatewayZdnsZ nameservers)typer7 r9 Zcontrolr; Zdns_nameserversr
).0ip)netadpr
r
s
z6generate_network_config...r> )r< r# Z mac_addressZsubnets)formatr lower)r= ir
)r? r r@ { s z+generate_network_config..)versionconfig) enumerate)Znetadpsr
r
r generate_network_configq s rG c
C s t jt jtjj| dddd}t jtjj| ddd}d|ksHd |krXt jtd
d S |jd i jd
}|jd i jdg }d }|jd i jdr|d d d }t |d }||d d |d d g ddd |d D ddd||dg|d|di|t
dd}tj d tj | |S ) a This callback will be applied by util.mount_cb() on the mounted
drive.
@param mount_dir: String representing path of directory where mounted drive
is available
@returns: A dict containing userdata, metadata and cfg based on metadata.
z
cloud.jsonF)fnamedecode)textz user.dataT)rH r Zvmr? z(Failed to load metadata. Invalid format.NZadditionalMetadatausernameZsshKeysZpasswordZsha512Z_idr# )z instance-idzlocal-hostnamez public-keysc S s> g | ]6}|d D ](}t |d d D ]}|d |dq$qqS )r> r8 r; r7 )r! r" )r )r= r? r> targetr
r
r r@ s z+read_user_data_callback..Z default_userzALL=(ALL) NOPASSWD:ALL)r# ZgecosZsudoZpasswdZ lock_passwdZssh_authorized_keys)Z
ssh_pwauthZ disable_rootZ system_infonetwork_configZmanage_etc_hosts)userdatametadatar( cfgzreturning DATA object:) r Z load_jsonr ospathr2 r1 r getrG r r )Z mount_dirZ meta_dataZ user_datarK Zssh_keyshashr8 datar
r
r r- sD
r- c @ sj e Zd ZdZejejejej hiZ
dd Z dd Z dd Z
edd Zd
d Zd d
Zdd Zdd ZdS )DataSourceRbxCloudZRbxCloudc C s t jj| ||| d | _d S )N)r
DataSource__init__seed)selfZsys_cfgr% pathsr
r
r rX s zDataSourceRbxCloud.__init__c C s t jj| }d|| jf S )Nz %s [seed=%s])r rW __str__rY )rZ rootr
r
r r\ s zDataSourceRbxCloud.__str__c C s> t }|dkrdS |d | _|d | _|d | _|d | _dS )zw
Metadata is passed to the launching instance which
is used to perform instance configuration.
FrN rO r( rP T)r5 userdata_rawrO r( rP )rZ r3 r
r
r _get_data s
z DataSourceRbxCloud._get_datac C s
| j d S )NrM )rP )rZ r
r
r rM s z!DataSourceRbxCloud.network_configc C s
| j d S )Nz public-keys)rO )rZ r
r
r get_public_ssh_keys s z&DataSourceRbxCloud.get_public_ssh_keysc C s | j S )N)r^ )rZ r
r
r get_userdata_raw s z#DataSourceRbxCloud.get_userdata_rawc C s | j S )N)rP )rZ r
r
r get_config_obj s z!DataSourceRbxCloud.get_config_objc C s t | j | j d S )N)r( r% )rZ rP Zis_new_instancer
r
r activate s zDataSourceRbxCloud.activateN)__name__
__module__ __qualname__Zdsnamer ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZdefault_update_eventsrX r\ r_ propertyrM r` ra rb rc r
r
r
r rV s
rV c C s t j| tS )N)r Zlist_from_depends datasources)Zdependsr
r
r get_datasource_list s ri )#__doc__r/ ZloggingrQ Zos.pathZtypingZ ipaddressr Z cloudinitr r r Zcloudinit.eventr r Z getLoggerrd r r
r r r r ZListr r r( r5 rG r- rW rV ZDEP_FILESYSTEMrh ri r
r
r
r s*
!B3