File: //lib/python3.6/site-packages/cloudinit/sources/__pycache__/DataSourceRbxCloud.cpython-36.pyc
3
A[�dL � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZm Z ddl
m Z m Z ej
e�ZdZdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� ZG dd� dej�ZeejffgZ dd� Z dS )zr
This file contains code used to gather the user data passed to an
instance on rootbox / hyperone cloud platforms
� N)�log)�sources�subp�util)�
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_HOSTS�LOG�debug)Zhosts� r
�(/usr/lib/python3.6/DataSourceRbxCloud.py�get_manage_etc_hosts s
r c C sF | j d�}t|d �d> t|d �d> t|d �d> t|d � S ) N�.r � � � � � � )�split�int)�addr�partsr
r
r �ip2int! s
r c s& dj � fdd�td�d d d� D ��S )Nr c s g | ]}t � |d > ? d@ ��qS )r � )�str)�.0�i)r r
r �
<listcomp>, s zint2ip.<locals>.<listcomp>� r ���)�join�range)r r
)r r �int2ip+ 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_arp/ 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� almalinux�fedora�centos�rhelz-sz-c�2�source� destinationz&Failed to arping from "%s" to "%s": %s)r( r) r* r+ )�namer' r ZProcessExecutionErrorr Zwarning)�items�distroZ source_param�item�errorr
r
r �gratuitous_arp8 s
r4 c C s� t tjd�tjd� �} | 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)�device�callbackZmtypeNz-Failed to mount %s when looking for user dataz0Did not find RbxCloud data, searched devices: %s�,)
�setr Zfind_devs_withZmount_cb�read_user_data_callback�OSError�errno�ENOENTZMountFailedError�logexcr r r# )Zdevicesr5 �rbx_data�errr
r
r �get_mdM s,
r@ c C s ddd� t | �D �d�S )z�Generate network configuration
@param netadps: A list of network adapter settings
@returns: A dict containing network config
r c sB g | ]:\}� d dj t|��� d j� � fdd�� d D �d��qS )Zphysicalzeth{}Z
macaddressc s> g | ]6}d |d � d d d� d d � d d d d��qS ) Zstatic�address�network�netmask�auto�gatewayZdnsZ nameservers)�typerA rC ZcontrolrE Zdns_nameserversr
)r �ip)�netadpr
r r { s
z6generate_network_config.<locals>.<listcomp>.<listcomp>rG )rF r/ Z mac_addressZsubnets)�formatr �lower)r r r
)rH r r v s z+generate_network_config.<locals>.<listcomp>)�version�config)� enumerate)Znetadpsr
r
r �generate_network_configl s rN c
C s t jt jtjj| d�dd�d�}t jtjj| d�dd�}d|ksHd |krXt jtd
� d S |jd i �jd
�}|jd i �jdg �}d }|jd i �jd�r�|d d d }t |d �}||d d |d d g d�dd� |d D �ddd||dg|d|d�i|t
� d�d�}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)�fname�decode)�textz user.dataT)rO r ZvmrH z(Failed to load metadata. Invalid format.NZadditionalMetadata�usernameZsshKeysZpasswordZsha512Z_idr/ )z instance-idzlocal-hostnamez public-keysc S sh g | ]`}|d D ]R}|d d t t|d d �d �t t|d d �d �gD ]}|d |d��qNqqS )rG rB rE r r rA )r- r. )r% r )r rH rG �targetr
r
r r � s
z+read_user_data_callback.<locals>.<listcomp>Z default_userzALL=(ALL) NOPASSWD:ALL)r/ ZgecosZsudoZpasswdZ lock_passwdZssh_authorized_keys)Z
ssh_pwauthZ disable_rootZ system_info�network_configZmanage_etc_hosts)�userdata�metadatar4 �cfgzreturning DATA object:) r Z load_jsonr �os�pathr# r= r �getrN r r )Z mount_dirZ meta_dataZ user_datarR Zssh_keys�hashrB �datar
r
r r9 � sD
r9 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_cfgr1 �pathsr
r
r r_ � s zDataSourceRbxCloud.__init__c C s t jj| �}d|| jf S )Nz %s [seed=%s])r r^ �__str__r` )ra �rootr
r
r rc � 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.
FrU rV r4 rW T)r@ � userdata_rawrV r4 rW )ra r> r
r
r � _get_data� s
z DataSourceRbxCloud._get_datac C s
| j d S )NrT )rW )ra r
r
r rT � s z!DataSourceRbxCloud.network_configc C s
| j d S )Nz public-keys)rV )ra r
r
r �get_public_ssh_keys� s z&DataSourceRbxCloud.get_public_ssh_keysc C s | j S )N)re )ra r
r
r �get_userdata_raw� s z#DataSourceRbxCloud.get_userdata_rawc C s | j S )N)rW )ra r
r
r �get_config_obj� s z!DataSourceRbxCloud.get_config_objc C s t | j | j� d S )N)r4 r1 )ra rW Zis_new_instancer
r
r �activate s zDataSourceRbxCloud.activateN)�__name__�
__module__� __qualname__Zdsnamer ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZdefault_update_eventsr_ rc rf �propertyrT rg rh ri rj 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 rp ) �__doc__r; rX Zos.pathZ cloudinitr Zloggingr r r Zcloudinit.eventr r Z getLoggerrk r r
r r r% r' r4 r@ rN r9 r^ r] ZDEP_FILESYSTEMro rp r
r
r
r �<module> s&
!F3