3
\�me� � @ s� d Z ddlZddlZddlZddlZddlmZmZmZm Z ej
e �Z dZ
ddgZi ddi d �Zd
d � ZG d d
� d
ej�Zdd� ZeejejffgZdd� ZdS )zb
This file contains code used to gather the user data passed to an
instance on RHEVm and vSphere.
� N)�dmi�sources�subp�utilz/etc/sysconfig/cloud-infoZmodprobeZfloppyi� Z localhost)zblock-device-mappingz instance-idzlocal-hostnameZ placementc C sp | d }| d }yt j|�j� }W nH tk
rj yt j|�j� }W n tk
rd t jtd� dS X Y nX |S )an
Description:
This callback will be applied by util.mount_cb() on the mounted
file.
Deltacloud file name contains deltacloud. Those not using
Deltacloud but instead instrumenting the injection, could
drop deltacloud from the file name.
Input:
mount_dir - Mount directory
Returns:
User Data
z/deltacloud-user-data.txtz/user-data.txtz Failed accessing user data file.N)r � load_file�strip�IOError�logexc�LOG)Z mount_dirZdeltacloud_user_data_fileZuser_data_fileZ user_data� r �(/usr/lib/python3.6/DataSourceAltCloud.py�read_user_data_callback& s r
c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�DataSourceAltCloudZAltCloudc C s" t jj| |||� d | _d| _d S )N�/�file://)r r )r �
DataSource�__init__�seedZsupported_seed_starts)�selfZsys_cfgZdistro�pathsr r r r L s zDataSourceAltCloud.__init__c C s t jj| �}d|| jf S )Nz %s [seed=%s])r r �__str__r )r �rootr r r r Q s zDataSourceAltCloud.__str__c C s� t jjt�rJytjt�j� j� }W n" tk
rD tj t
dt� dS X |S t j d�}|s\dS |j� }|j
d�rrdS |j
d�r�dS dS )at
Description:
Get the type for the cloud back end this instance is running on
by examining the string returned by reading either:
CLOUD_INFO_FILE or
the dmi data.
Input:
None
Returns:
One of the following strings:
'RHEV', 'VSPHERE' or 'UNKNOWN'
z'Unable to access cloud info file at %s.ZUNKNOWNzsystem-product-name�RHEVZVMWARE�VSPHERE)�os�path�exists�CLOUD_INFO_FILEr r r �upperr r r
r Z
read_dmi_data�
startswith)r �
cloud_typeZ system_nameZsys_namer r r �get_cloud_typeU s&
z!DataSourceAltCloud.get_cloud_typec C sb t jd� | j� }t jdt|�� d|kr8| j� rRdS nd|krN| j� rRdS ndS tjt d� dS )a�
Description:
User Data is passed to the launching instance which
is used to perform instance configuration.
Cloud providers expose the user data differently.
It is necessary to determine which cloud provider
the current instance is running on to determine
how to access the user data. Images built with
image factory will contain a CLOUD_INFO_FILE which
contains a string identifying the cloud provider.
Images not built with Imagefactory will try to
determine what the cloud provider is based on system
information.
zInvoked get_data()zcloud_type: %sr Tr FzFailed accessing user data.)r
�debugr! �str�user_data_rhevm�user_data_vspherer r )r r r r r � _get_data~ s
z DataSourceAltCloud._get_datac C s: | j � }t| d�stj| _|dkr(d| _d|j� | jf S )z(Return the subplatform metadata details.�sourcer z/dev/fd0z%s (%s))r! �hasattrr ZMETADATA_UNKNOWNr' �lower)r r r r r �_get_subplatform� s
z#DataSourceAltCloud._get_subplatformc , C s d}y
t � W n0 tjk
r> } ztjtd|� dS d}~X nX d}ytj|dd� W n4 tjtfk
r� } ztjtd|� dS d}~X nX ytj|t �}W nR tk
r� } z|j
t
j kr�� W Y dd}~X n$ tj k
r� tjtd|� Y nX || _
t| _|�rd S dS dS )
a�
RHEVM specific userdata read
If on RHEV-M the user data will be contained on the
floppy device in file <user_data_file>
To access it:
modprobe floppy
Leverage util.mount_cb to:
mkdir <tmp mount dir>
mount /dev/fd0 <tmp mount dir>
The call back passed to util.mount_cb will do:
read <tmp mount dir>/<user_data_file>
NzFailed modprobe: %sFz/dev/fd0� )r ZtimeoutzFailed udevadm_settle: %s
z-Failed to mount %s when looking for user dataT)�modprobe_floppyr ZProcessExecutionErrorr r r
Zudevadm_settle�OSError�mount_cbr
�errno�ENOENT�MountFailedError� userdata_raw�META_DATA_NOT_SUPPORTED�metadata)r �
return_str�eZ
floppy_dev�errr r r r$ � s6
z"DataSourceAltCloud.user_data_rhevmc C s� d}t jd�}x||D ]t}y t j|t�}|r2|| _P W q tk
rd } z|jtjkrT� W Y dd}~X q t jk
r� t j t
d|� Y qX qW || _ t | _
|r�dS dS dS )a�
vSphere specific userdata read
If on vSphere the user data will be contained on the
cdrom device in file <user_data_file>
To access it:
Leverage util.mount_cb to:
mkdir <tmp mount dir>
mount /dev/fd0 <tmp mount dir>
The call back passed to util.mount_cb will do:
read <tmp mount dir>/<user_data_file>
Nz LABEL=CDROMz-Failed to mount %s when looking for user dataTF)r Zfind_devs_withr. r
r' r- r/ r0 r1 r r
r2 r3 r4 )r r5 Z
cdrom_listZ cdrom_devr7 r r r r% � s*
z$DataSourceAltCloud.user_data_vsphereN) �__name__�
__module__� __qualname__Zdsnamer r r! r& r* r$ r% r r r r r H s )' 6r c C s& t j t�\} }tjddjt�| � d S )NzCommand: %s
Output%s� )r �CMD_PROBE_FLOPPYr
r" �join)�outZ_errr 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 r@ )�__doc__r/ Zloggingr Zos.pathZ cloudinitr r r r Z getLoggerr8 r
r r< r3 r
r r r, ZDEP_FILESYSTEMZ DEP_NETWORKr? r@ r r r r �<module>
s$
" F