CasperSecurity
o
쑛hI, � @ st d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
m Z ddl m
Z
ddlmZmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZmZ dd l m Z e� e �Z d Z!d$d
d�Z"G dd� dej#�Z$G dd� de$�Z%de%ej&fiZ'dd� Z(de de)fdd�Z*de ddfdd �Z+d%d d �Z,de de-fd d �Z.d!d"� Z/e d#kr�e"� �0� Z1e/e!e1� dS dS )&z)Handle reconfiguration on hotplug events.� N)� reporting�stages�util)�install_hotplug)�
EventScope� EventType)�loggers)�read_sys_net_safe)�parse_net_config_data)�events)�
DataSource�DataSourceNotFoundException)�Initz hotplug-hookc C s� | s t jttd�} t| _| jdddddgd� | jdd d
�}d|_|jd d d
� |jddd
�}|jdddddd� |jdddddgd� |jddd
� | S )a Build or extend an arg parser for hotplug-hook utility.
@param parser: Optional existing ArgumentParser instance representing the
subcommand which will be extended to support the args of this utility.
@returns: ArgumentParser with proper argument configuration.
)�prog� descriptionz-sz --subsystemTzsubsystem to act on�net)�required�help�choiceszHotplug Action�hotplug_action)�title�dest�queryz0Query if hotplug is enabled for given subsystem.)r �handlezHandle the hotplug event.z-dz --devpath�PATHz Sysfs path to hotplugged device)r �metavarr z-uz --udevactionzSpecify action to take.�add�enablez%Enable hotplug for a given subsystem.) �argparse�ArgumentParser�NAME�__doc__r � add_argument�add_subparsersr �
add_parser)�parser�
subparsers�
parser_handle� r( �B/usr/lib/python3/dist-packages/cloudinit/cmd/devel/hotplug_hook.py�
get_parser sL �������r* c @ s` e Zd Zdd� Zejdd� �Zeejdd� ��Zejde fdd ��Z
d
d � Z d d
� Z dd� Z
dS )�
UeventHandlerc C s" || _ || _|| _|| _|| _d S �N)�id�
datasource�devpath�action�
success_fn)�selfr- r. r/ r0 r1 r( r( r) �__init__R s
zUeventHandler.__init__c C � t � �r, ��NotImplementedError�r2 r( r( r) �applyY � zUeventHandler.applyc C r4 r, r5 r7 r( r( r) �config] s zUeventHandler.config�returnc C r4 r, r5 r7 r( r( r) �device_detectedb r9 z UeventHandler.device_detectedc C s@ d }| j dkr
d}ntd| j ��|| �� kr td| j ��d S )Nr TzUnknown action: %sz'Failed to detect %s in updated metadata)r0 �
ValueErrorr< � RuntimeErrorr- )r2 �detect_presencer( r( r) �detect_hotplugged_devicef s
��z&UeventHandler.detect_hotplugged_devicec C s | � � S r, )r1 r7 r( r( r) �successr s zUeventHandler.successc C s, | j �tjg�}|std| j tjf ��|S )Nz&Datasource %s not updated for event %s)r. � update_metadata_if_supportedr �HOTPLUGr> )r2 �resultr( r( r) �update_metadatau s �
��z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r3 �abc�abstractmethodr8 �propertyr: �boolr< r@ rA rE r( r( r( r) r+ Q s
r+ c s> e Zd Z� fdd�Zdd� Zedd� �Zdefdd �Z� Z S )
�
NetHandlerc s* t tj�|�d�}t� �|||||� d S )N�address)r �os�path�basename�superr3 )r2 r. r/ r0 r1 r- �� __class__r( r) r3 � s zNetHandler.__init__c C sZ | j jj| jdd� tj�| j�}| j j�� }| j dkr)|�
|�s+t d� | j���d S d S )NF)�bring_upr z Failed to bring up device: {})
r. �distro�apply_network_configr: rO rP rQ r/ �network_activatorr0 �bring_up_interfacer> �format)r2 �interface_name� activatorr( r( r) r8 � s �
��zNetHandler.applyc C s | j jS r, )r. �network_configr7 r( r( r) r: � s zNetHandler.configr; c s< t � j�}� fdd�|�� D �}t�d� j|� t|�dkS )Nc s g | ] }|� d �� jkr|�qS )� mac_address)�getr- )�.0�ifacer7 r( r) �
<listcomp>� s
�z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r
r: �iter_interfaces�LOG�debugr- �len)r2 �netstate�foundr( r7 r) r< � s
� zNetHandler.device_detected)
rF rG rH r3 r8 rK r: rL r< �
__classcell__r( r( rS r) rM � s
rM r c
C sR zt | d }W n ty } ztd�|��|�d }~ww tj| j| jtj |d�S )N� z4hotplug-hook: cannot handle events for subsystem: {})r. �cfg�event_source_type�scope)
�SUBSYSTEM_PROPERTIES_MAP�KeyErrorr> rZ r �update_event_enabledr. rk r rC )� hotplug_init� subsystemrm �er( r( r) �
is_enabled� s$ ������rt rq rr c C sR t �d� | jdd�}|�tjg�st �d|� d S t| |�s't �d|� d S |S )NzFetching datasource�trust��existingz*hotplug not supported for event of type %sz(hotplug not enabled for event of type %s)rd re �fetch�get_supported_eventsr rC rt )rq rr r. r( r( r) �initialize_datasource� s
rz r; c C s� t | |�}|s d S t| d }t�d|� ||||| jd�}t�� }|jjs.t|||� d S t�� | |jj k rVt|||� t�d� t�
|jj � t�� | |jj k s8d S d S )Nr zCreating %s event handler)r. r/ r0 r1 z>Gathering network configuration again due to IMDS limitations.) rz rn rd re �_write_to_cache�time�hotplug_retry_settings� force_retry� try_hotplug� sleep_total�sleep� sleep_period)rq r/ rr �
udevactionr. � handler_cls�
event_handler�startr( r( r) �handle_hotplug� s,
� � �r� c C s� g d�}t d�}t|�D ]X\}}t�d| |t|�� z+t�d� |�� |js0t�d� |�� t�d� |�� t�d� |� � W d S t yd } zt�d|� t
� |� |}W Y d }~q d }~ww |�) N)rj � � �
� z#Bug while processing hotplug event.z!subsystem=%s update attempt %s/%szRefreshing metadataz$Detecting device in updated metadatazApplying config changezUpdating cachez,Exception while processing hotplug event. %s) � Exception� enumeraterd re rf rE �skip_hotplug_detectr@ r8 rA r| r� )rr r� r. �
wait_times�last_exception�attempt�waitrs r( r( r) r � s6 �
��r c C s� | j dd�}|s
dS t| d }tj|�tjg��|t� �v }|s-td|� �tj d� dS t
� | j �}|j
|d v rGtd|� d �tj d� d
S |d �|j
� t
j| j �d �t�|�d d
d� t|d
| jd� d
S )Nru rv Frj z#hotplug not supported for event of )�file�scopesz)Not installing hotplug for event of type z. Reason: Already done.Tzhotplug.enabled�wi� )�omode�mode)�network_hotplug_enabledrk )rx rn r rC ry r_ �set�print�sys�stderrr �read_hotplug_enabled_file�paths�value�append�
write_file� get_cpath�json�dumpsr rk )rq rr r. rm �hotplug_supported�hotplug_enabled_filer( r( r) �enable_hotplug� s< � �
�
��r� c C s� t j| tdd�}tg |d�}|�� t�|j� d|jv r&t� |j�
d�� t � d| |j
|jd|v r5|jnd d|v r=|jnd � |�w za|j
dkrmzt||j�}W n tyc td � t�d
� Y nw t|rid nd � n6|j
d
kr~t||j|j|jd� n%t�� dkr�tj�d� t�d
� t||jd�s�t�d
� td|j� �� W n
t y� t � d� � w W d � n1 s�w Y t � d� t� � d S )NT)�reporting_enabled)�ds_deps�reporterr zh%s called with the following arguments: {hotplug_action: %s, subsystem: %s, udevaction: %s, devpath: %s}r� r/ r z9Unable to determine hotplug state. No datasource detectedrj �enabled�disabledr )rq r/ rr r� r z9Root is required. Try prepending your command with sudo.
)rq rr z)Enabled cloud-init hotplug for subsystem=z*Received fatal exception handling hotplug!zExiting hotplug handler) r �ReportEventStackr! r �read_cfgr �
setup_loggingrk r �update_configurationr_ rd re r rr r� r/ rz r
r� r� �exitr� rO �getuidr� �writer� r� � exception� flush_events)�name�args�hotplug_reporterrq r. r( r( r) � handle_args sx �
�
� ��
� �
�
���
�� �
) r� �__main__r, )r; N)2r! rI r r� �loggingrO r� r| � cloudinitr r r �#cloudinit.config.cc_install_hotplugr �cloudinit.eventr r �
cloudinit.logr �
cloudinit.netr �cloudinit.net.network_stater
�cloudinit.reportingr �cloudinit.sourcesr r
�cloudinit.stagesr � getLoggerrF rd r r* �ABCr+ rM �NETWORKrn rt �strrz r� r rL r� r� �
parse_argsr� r( r( r( r) �<module> sD
60
#�
#E
�