File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceRbxCloud.cpython-310.pyc
o
�AdU � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm
Z
m Z m Z ddl
mZmZ e �e�ZdZdd � Zd
ed efd d
�Zd eje fdd�Zdd� Zdd� Zdd� Z dd� Z dd� Z G dd� de
j �Z e e
j!ffgZ"d d � Z#dS ) zr
This file contains code used to gather the user data passed to an
instance on rootbox / hyperone cloud platforms
� N)� IPv4Address)�log)�sources�subp�util)�
EventScope� EventTypez
/etc/hostsc C s. t jtdd�} | rt�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_file� ETC_HOSTS�LOG�debug)�hosts� 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 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
�arping)r )�cmdr r r �_sub_arp4 s r c C s| d}|j dv r d}| D ]0}ztdd||d |d g� W q tjy; } zt�d|d |d |� W Y d }~q d }~ww d S ) Nz-S)�fedora�centos�rhelz-sz-c�2�source� destinationz&Failed to arping from "%s" to "%s": %s)�namer r �ProcessExecutionErrorr �warning)�items�distro� source_param�item�errorr r r �gratuitous_arp= s$
�����r- c C s� t t�d�t�d� �} | sdS | D ]>}ztj|tg d�d�}|r&|W S W q ty@ } z
|jtjkr6� W Y d}~qd}~w tjyP t� t
d|� Y qw t
� dd � | �� dS )
zAReturns False (not found or error) or a dictionary with metadata.z
LABEL=CLOUDMDz
LABEL=cloudmdF)�vfat�fat�msdosfs)�device�callback�mtypeNz-Failed to mount %s when looking for user dataz0Did not find RbxCloud data, searched devices: %s�,)
�setr �find_devs_with�mount_cb�read_user_data_callback�OSError�errno�ENOENT�MountFailedError�logexcr r
�join)�devicesr1 �rbx_data�errr r r �get_mdR s> ���
� ����
�rB 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 )�physicalzeth{}�
macaddressc s> g | ]}d |d � d d d� d d � d d d d��qS ) �static�address�network�netmask�auto�gateway�dns� nameservers)�typerG rI �controlrK �dns_nameserversr )�.0�ip��netadpr r �
<listcomp> s �
���z6generate_network_config.<locals>.<listcomp>.<listcomp>rR )rN r% � mac_address�subnets)�formatr �lower)rQ �ir rS r rU z s �
���z+generate_network_config.<locals>.<listcomp>)�version�config)� enumerate)�netadpsr r r �generate_network_configq s
��r_ c
C s t jt jtj�| d�dd�d�}t jtj�| d�dd�}d|vs$d |vr,t �td
� d S |�d i ��d
�}|�d i ��dg �}d }|�d i ��d�rR|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.jsonF)�fname�decode)�textz user.dataT)r` r �vmrT z(Failed to load metadata. Invalid format.N�additionalMetadata�username�sshKeys�password�sha512�_idr% )z instance-idzlocal-hostname� public-keysc S s> g | ]}|d D ]}t |d d �D ] }|d |d��qqqS )rR rH rK rG )r# r$ )r )rQ rT rR �targetr r r rU � s �� �
�z+read_user_data_callback.<locals>.<listcomp>� default_userzALL=(ALL) NOPASSWD:ALL)r% �gecos�sudo�passwd� lock_passwd�ssh_authorized_keys)�
ssh_pwauth� disable_root� system_info�network_config�manage_etc_hosts)�userdata�metadatar- �cfgzreturning DATA object:) r � load_jsonr
�os�pathr> r= r �getr_ r r
)� mount_dir� meta_data� user_datare �ssh_keys�hashrH �datar r r r8 � sV ���
����
��
r8 c @ sj e Zd ZdZejejejej hiZ
dd� Z dd� Z dd� Z
edd � �Zd
d � Zd d
� Zdd� Zdd� ZdS )�DataSourceRbxCloud�RbxCloudc C s t j�| |||� d | _d S r )r �
DataSource�__init__�seed)�self�sys_cfgr) �pathsr r r r� � s
zDataSourceRbxCloud.__init__c C s t j�| �}d|| jf S )Nz %s [seed=%s])r r� �__str__r� )r� �rootr r r r� � s zDataSourceRbxCloud.__str__c C s> t � }|du r dS |d | _|d | _|d | _|d | _dS )zw
Metadata is passed to the launching instance which
is used to perform instance configuration.
Frw rx r- ry T)rB � userdata_rawrx r- ry )r� r@ r r r � _get_data� s
z DataSourceRbxCloud._get_datac C �
| j d S )Nru �ry �r� r r r ru � s
z!DataSourceRbxCloud.network_configc C r� )Nrj )rx r� r r r �get_public_ssh_keys� s
z&DataSourceRbxCloud.get_public_ssh_keysc C � | j S r )r� r� r r r �get_userdata_raw� � z#DataSourceRbxCloud.get_userdata_rawc C r� r r� r� r r r �get_config_obj� r� z!DataSourceRbxCloud.get_config_objc C s t | j | j� d S r )r- r) )r� ry �is_new_instancer r r �activate s zDataSourceRbxCloud.activateN)�__name__�
__module__� __qualname__�dsnamer �NETWORKr �BOOT_NEW_INSTANCE�BOOT� BOOT_LEGACY�default_update_eventsr� r� r� �propertyru r� r� r� r� r r r r r� � s" ��
r� c C s t �| t�S r )r �list_from_depends� datasources)�dependsr r r �get_datasource_list s r� )$�__doc__r: r{ �os.path�typing� ipaddressr � cloudinitr �loggingr r r �cloudinit.eventr r � getLoggerr� r r r r r r �Listr r r- rB r_ r8 r� r� �DEP_FILESYSTEMr� r� r r r r �<module> s.
!B
3�