Your IP : 216.73.217.13
U
�Ad�# � @ s( d Z ddlZddlZddlZddlZddlZddlmZmZm Z ddl
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZmZ dd lmZ e�e�Zd
Zd d d �Z G d
d� dej �Z G dd� de �Z de e j fiZ!dd� Z"ee#d�dd�Z$ed�dd�Z%dd� Z&ed k�r$e � �'� Z(e&ee(� dS ) z)Handle reconfiguration on hotplug events.� N)�log� reporting�stages)�
EventScope� EventType)�read_sys_net_safe)�parse_net_config_data)�events)�
DataSource�DataSourceNotFoundException)�Initz hotplug-hookc C s� | st jttd�} t| _| jdddddgd� | jdd d
�}d|_|jd d d
� |jddd
�}|jdddddd� |jddddddgd� | 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 ZhandlezHandle the hotplug event.z-dz --devpath�PATHz Sysfs path to hotplugged device)r �metavarr z-uz --udevactionzSpecify action to take.�add�remove) �argparse�ArgumentParser�NAME�__doc__r � add_argumentZadd_subparsersr Z
add_parser)�parserZ
subparsersZ
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� ��Zeje d�dd ��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( r) r* r! r! r" �__init__J s
zUeventHandler.__init__c C s
t � �d S r% ��NotImplementedError�r+ r! r! r" �applyQ s zUeventHandler.applyc C s
t � �d S r% r- r/ r! r! r" �configU s zUeventHandler.config��returnc C s
t � �d S r% r- r/ r! r! r" �device_detectedZ s z UeventHandler.device_detectedc C sP d }| j dkrd}n | j dkr$d}ntd| j ��|| �� krLtd| j ��d S )Nr Tr FzUnknown action: %sz'Failed to detect %s in updated metadata)r) �
ValueErrorr4 � RuntimeErrorr& )r+ Zdetect_presencer! r! r" �detect_hotplugged_device^ s
�z&UeventHandler.detect_hotplugged_devicec C s | � � S r% )r* r/ r! r! r" �successl s zUeventHandler.successc C s, | j �tjg�}|s(td| j tjf ��|S )Nz&Datasource %s not updated for event %s)r' Z update_metadata_if_supportedr �HOTPLUGr6 )r+ �resultr! r! r" �update_metadatao s �
��z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r, �abc�abstractmethodr0 �propertyr1 �boolr4 r7 r8 r; r! r! r! r" r$ I s
r$ c s> e Zd Z� fdd�Zdd� Zedd� �Zed�dd �Z� Z S )
�
NetHandlerc s* t tj�|�d�}t� �|||||� d S )NZaddress)r �os�path�basename�superr, )r+ r' r( r) r* r& �� __class__r! r" r, | s zNetHandler.__init__c C s| | j jj| jdd� tj�| j�}| j j�� }| j dkrT|�
|�sxt d� | j���n$| j dkrx|�
|�sxt d� | j���d S )NF)Zbring_upr z Failed to bring up device: {}r z Failed to bring down device: {})r' ZdistroZapply_network_configr1 rD rE rF r( Znetwork_activatorr) Zbring_up_interfacer6 �formatZbring_down_interface)r+ Zinterface_nameZ activatorr! r! r" r0 � s �
�
�zNetHandler.applyc C s | j jS r% )r' Znetwork_configr/ r! r! r" r1 � s zNetHandler.configr2 c s< t � j�}� fdd�|�� D �}t�d� j|� t|�dkS )Nc s g | ]}|� d �� jkr|�qS )Z mac_address)�getr& )�.0Zifacer/ r! r" �
<listcomp>� s �z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r r1 Ziter_interfaces�LOG�debugr& �len)r+ Znetstate�foundr! r/ r" r4 � s
�zNetHandler.device_detected)
r< r= r> r, r0 rA r1 rB r4 �
__classcell__r! r! rH r" rC { s
rC r c
C s\ zt | d }W n2 tk
rB } ztd�|��|�W 5 d }~X Y nX tj| j| jtj |d�S )N� z4hotplug-hook: cannot handle events for subsystem: {})r' �cfgZevent_source_type�scope)
�SUBSYSTEM_PROPERTES_MAP�KeyError� ExceptionrJ r Zupdate_event_enabledr' rT r r9 )� hotplug_init� subsystemrU �er! r! r" �
is_enabled� s ����r\ )rY rZ c C sR t �d� | jdd�}|�tjg�s4t �d|� d S t| |�sNt �d|� d S |S )NzFetching datasourceZtrust)�existingz*hotplug not supported for event of type %sz(hotplug not enabled for event of type %s)rN rO ZfetchZget_supported_eventsr r9 r\ )rY rZ r' r! r! r" �initialize_datasource� s
r^ )rY c C s t | |�}|sd S t| d }t�d|� ||||| jd�}dddddg}td �}t|�D ]�\} }
t�d
|| t|�� zZt�d � |�� |j s�t�d � |�
� t�d
� |� � t�d� |� � W �qW qZ tk
�r } z t�d| � t
�|
� | }W 5 d } ~ X Y qZX qZ|�d S )Nr zCreating %s event handler)r' r( r) r* rS � � �
� 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)r^ rV rN rO Z_write_to_cacherX � enumeraterP r; Zskip_hotplug_detectr7 r0 r8 �time�sleep) rY r( rZ �
udevactionr' Z handler_clsZ
event_handlerZ
wait_timesZlast_exceptionZattempt�waitr[ r! r! r" �handle_hotplug� sH
��
rh c C s8 t j| tdd�}tg |d�}|�� t�|j� d|jkrLt� |j�
d�� t � d| |j
|jd|krj|jnd d|krz|jnd � |�� zn|j
dkr�zt||j�}W n& tk
r� td � t�d
� Y nX t|r�d nd � nt||j|j|jd
� W n" tk
�r t �d� � Y nX W 5 Q R X t � d� t�� d S )NT)Zreporting_enabled)Zds_depsZreporterr zh%s called with the following arguments: {hotplug_action: %s, subsystem: %s, udevaction: %s, devpath: %s}rf r( r z9Unable to determine hotplug state. No datasource detectedrS �enabledZdisabled)rY r( rZ rf z*Received fatal exception handling hotplug!zExiting hotplug handler)r ZReportEventStackr r Zread_cfgr Z setupLoggingrT r Zupdate_configurationrK rN rO r rZ rf r( r^ r �print�sys�exitrh rX Z exceptionZ flush_events)�name�argsZhotplug_reporterrY r' r! r! r" � handle_args� sV �
�
���
ro �__main__)N))r r? r rD rk rd Z cloudinitr r r Zcloudinit.eventr r Z
cloudinit.netr Zcloudinit.net.network_stater Zcloudinit.reportingr Zcloudinit.sourcesr
r Zcloudinit.stagesr Z getLoggerr< rN r r# �ABCr$ rC ZNETWORKrV r\ �strr^ rh ro �
parse_argsrn r! r! r! r" �<module> s6
22( �(5