
­­­­­­­­­­­­­­­­­­
<!DOCTYPE html>
<html>
3
cjZ                 @   s   d Z ddlZddlZddlZddlZddlZddlmZ ddlm	Z	m
Z
mZ ddlmZmZ ejeZdZdd Zeed	d
dZeje dddZdd Zdd Zdd Zdd Zdd ZG dd de	jZee	j ffgZ!dd Z"dS )zr
This file contains code used to gather the user data passed to an
instance on rootbox / hyperone cloud platforms
    N)IPv4Address)sourcessubputil)
EventScope	EventTypez
/etc/hostsc              C   s.   t jtdd} | r tjd dS tjd dS )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_HOSTSLOGdebug)Zhosts r   (/usr/lib/python3.6/DataSourceRbxCloud.pyget_manage_etc_hosts   s    

r   )increturnc             C   s   t ttt| | S )N)strr   int)addrr   r   r   r   increment_ip#   s    r   )r   c             C   s   | t | dt | dgS )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 dg|  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
    Zarping)r   )cmdr   r   r   _sub_arp4   s    r   c             C   s   d}|j dkrd}xh| D ]`}ytdd	||d
 |d g W q tjk
rv } ztjd|d
 |d | W Y d d }~X qX qW d S )Nz-S	almalinuxfedoracentos
cloudlinuxrhelz-sz-c2sourcedestinationz&Failed to arping from "%s" to "%s": %s)r   r   r   r   r   )namer   r   ZProcessExecutionErrorr   Zwarning)itemsdistroZsource_paramitemerrorr   r   r   gratuitous_arp=   s    

r(   c              C   s   t tjdtjd } | s dS x| D ]z}y"tj|tdddgd}|rJ|S W q& tk
r| } z|jtjkrl W Y dd}~X q& tjk
r   tj	t
d	| Y q&X q&W t
jd
dj|  dS )zAReturns False (not found or error) or a dictionary with metadata.zLABEL=CLOUDMDzLABEL=cloudmdFZvfatZfatZmsdosfs)devicecallbackZmtypeNz-Failed to mount %s when looking for user dataz0Did not find RbxCloud data, searched devices: %s,)setr   Zfind_devs_withZmount_cbread_user_data_callbackOSErrorerrnoENOENTZMountFailedErrorlogexcr   r   join)Zdevicesr)   rbx_dataerrr   r   r   get_mdR   s,    
r5   c             C   s   ddd t | D dS )zGenerate network configuration

    @param netadps: A list of network adapter settings

    @returns: A dict containing network config
       c                sB   g | ]:\} d dj t| d j  fdd d D dqS )Zphysicalzeth{}Z
macaddressc          	      s>   g | ]6}d |d  d d d d d  d d d dqS )	ZstaticaddressnetworknetmaskautogatewayZdnsZnameservers)typer7   r9   Zcontrolr;   Zdns_nameserversr   ).0ip)netadpr   r   
<listcomp>   s   
z6generate_network_config.<locals>.<listcomp>.<listcomp>r>   )r<   r#   Zmac_addressZsubnets)formatr   lower)r=   ir   )r?   r   r@   {   s   z+generate_network_config.<locals>.<listcomp>)versionconfig)	enumerate)Znetadpsr   r   r   generate_network_configq   s    rG   c             C   s  t jt jtjj| dddd}t jtjj| ddd}d|ksHd	|krXt jtd
 dS |jdi jd}|jdi jdg }d}|jdi jdr|d d d }t	|d	 }||d d |d d g ddd |d	 D ddd||dg|d|di|t
 dd}tjd tj| |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)fnamedecode)textz	user.dataT)rH   r   Zvmr?   z(Failed to load metadata. Invalid format.NZadditionalMetadatausernameZsshKeysZpasswordZsha512Z_idr#   )zinstance-idzlocal-hostnamezpublic-keysc             S   s>   g | ]6}|d  D ](}t |d d D ]}|d |dq$qqS )r>   r8   r;   r7   )r!   r"   )r   )r=   r?   r>   targetr   r   r   r@      s   z+read_user_data_callback.<locals>.<listcomp>Zdefault_userzALL=(ALL) NOPASSWD:ALL)r#   ZgecosZsudoZpasswdZlock_passwdZssh_authorized_keys)Z
ssh_pwauthZdisable_rootZsystem_infonetwork_configZmanage_etc_hosts)userdatametadatar(   cfgzreturning DATA object:)r   Z	load_jsonr	   ospathr2   r1   r   getrG   r   r   )Z	mount_dirZ	meta_dataZ	user_datarK   Zssh_keyshashr8   datar   r   r   r-      sD    	




r-   c               @   sj   e Zd ZdZejejejej	hiZ
dd Zdd Zdd Zedd	 Zd
d Zdd Zdd Zdd ZdS )DataSourceRbxCloudZRbxCloudc             C   s   t jj| ||| d | _d S )N)r   
DataSource__init__seed)selfZsys_cfgr%   pathsr   r   r   rX      s    zDataSourceRbxCloud.__init__c             C   s   t jj| }d|| jf S )Nz%s [seed=%s])r   rW   __str__rY   )rZ   rootr   r   r   r\      s    zDataSourceRbxCloud.__str__c             C   s>   t  }|dkrdS |d | _|d | _|d | _|d | _dS )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   Zis_new_instancer   r   r   activate  s    zDataSourceRbxCloud.activateN)__name__
__module____qualname__Zdsnamer   ZNETWORKr   ZBOOT_NEW_INSTANCEZBOOTZBOOT_LEGACYZdefault_update_eventsrX   r\   r_   propertyrM   r`   ra   rb   rc   r   r   r   r   rV      s   
rV   c             C   s   t j| tS )N)r   Zlist_from_dependsdatasources)Zdependsr   r   r   get_datasource_list  s    ri   )#__doc__r/   ZloggingrQ   Zos.pathZtypingZ	ipaddressr   Z	cloudinitr   r   r   Zcloudinit.eventr   r   Z	getLoggerrd   r   r
   r   r   r   r   ZListr   r   r(   r5   rG   r-   rW   rV   ZDEP_FILESYSTEMrh   ri   r   r   r   r   <module>   s*   
		!B3