File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceRbxCloud.cpython-38.pyc
U
Ӈgg � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z ddl m
Z
mZ e�e�ZdZdd� Zeed �d
d �Zeje d �d
d�Zdd� 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)�sources�subp�util)�
EventScope� EventTypez
/etc/hostsc C s. t jtdd�} | r t�d� dS t�d� 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_text_file� ETC_HOSTS�LOG�debug)Zhosts� r
�F/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceRbxCloud.py�get_manage_etc_hosts s
r )�inc�returnc C s t ttt| ��| ��S �N)�strr �int)�addrr r
r
r � increment_ip# s r )r c C s | t | d�t | d�gS )z�Return 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 � 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 dkrd}| D ]`}z tdd||d |d g� W q tjk
rt } z t�d|d |d |� W 5 d }~X Y qX qd S ) Nz-S)ZfedoraZcentosZrhelz-sz-c�2�source� destinationz&Failed to arping from "%s" to "%s": %s)�namer r ZProcessExecutionErrorr Zwarning)�items�distroZ source_param�item�errorr
r
r �gratuitous_arp= s
��r$ c C s� t t�d�t�d� �} | s dS | D ]�}z(tj|tdddgd�}|rN|W S W q$ tk
r� } z|jtjkrp� W 5 d}~X Y q$ tjk
r� t� t
d |� Y q$X q$t
� d
d � | �� 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�errnoZENOENTZMountFailedError�logexcr r �join)Zdevicesr% �rbx_data�errr
r
r �get_mdR s< ��� � �r0 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
� c sB g | ]:\}� d d� t|��� d �� � 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)�typer2 r4 Zcontrolr6 Zdns_nameserversr
)�.0�ip��netadpr
r �
<listcomp> s �
��z6generate_network_config.<locals>.<listcomp>.<listcomp>r9 )r7 r Z mac_addressZsubnets)�formatr �lower)r8 �ir
r: r r< z s �
��z+generate_network_config.<locals>.<listcomp>)�versionZconfig)� enumerate)Znetadpsr
r
r �generate_network_configq s
��rB c
C s t jt jtj�| d�d�d�}t jtj�| d�dd�}d|ksFd|krVt �td � d
S |� d i �� d �}|� d i �� d
g �}d
}|� d i �� d�r�|d d d }t
|d �}||d d |d d g d�dd� |d D �ddd||dg|d|d�i|t � d�d�}t� d� t� |� |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.json)�fname)�textz user.dataT)rC r Zvmr; z(Failed to load metadata. Invalid format.NZadditionalMetadata�usernameZsshKeysZpasswordZsha512Z_idr )z instance-idzlocal-hostname� public-keysc S s> g | ]6}|d D ](}t |d d �D ]}|d |d��q$qqS )r9 r3 r6 r2 )r r )r )r8 r; r9 �targetr
r
r r< � s
�z+read_user_data_callback.<locals>.<listcomp>Z default_userzALL=(ALL) NOPASSWD:ALLF)r ZgecosZsudoZpasswdZ lock_passwdZssh_authorized_keys)Z
ssh_pwauthZ disable_rootZ system_info�network_configZmanage_etc_hosts)�userdata�metadatar$ �cfgzreturning DATA object:)
r Z load_jsonZload_binary_file�os�pathr- r r, r �getrB r r )Z mount_dirZ meta_dataZ user_datarE Zssh_keys�hashr3 �datar
r
r r) � sT � �
����
��
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 j�| |||� d | _d | _d | _d S r )r �
DataSource�__init__�seedr$ rK )�selfZsys_cfgr! �pathsr
r
r rS � s zDataSourceRbxCloud.__init__c C s t j�| �}d|| jf S )Nz %s [seed=%s])r rR �__str__rT )rU �rootr
r
r rW � 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.
FrI rJ r$ rK T)r0 � userdata_rawrJ r$ rK )rU r. r
r
r � _get_data� s
z DataSourceRbxCloud._get_datac C s
| j d S )NrH �rK �rU r
r
r rH � s z!DataSourceRbxCloud.network_configc C s
| j d S )NrF )rJ r\ r
r
r �get_public_ssh_keys� s z&DataSourceRbxCloud.get_public_ssh_keysc C s | j S r )rY r\ r
r
r �get_userdata_raw� s z#DataSourceRbxCloud.get_userdata_rawc C s | j S r r[ r\ r
r
r �get_config_obj� s z!DataSourceRbxCloud.get_config_objc C s t | j | j� d S r )r$ r! )rU rK Zis_new_instancer
r
r �activate s zDataSourceRbxCloud.activateN)�__name__�
__module__� __qualname__Zdsnamer ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZ BOOT_LEGACYZdefault_update_eventsrS rW rZ �propertyrH r] r^ r_ r` r
r
r
r rQ � s ��
rQ c C s t �| t�S r )r Zlist_from_depends� datasources)Zdependsr
r
r �get_datasource_list s rf )#�__doc__r+ ZloggingrL �os.path�typingZ ipaddressr Z cloudinitr r r Zcloudinit.eventr r Z getLoggerra r r
r r r r ZListr r r$ r0 rB r) rR rQ ZDEP_FILESYSTEMre rf r
r
r
r �<module> s,
!@5
�