File: //usr/lib/python3.9/site-packages/cloudinit/cmd/devel/__pycache__/hotplug_hook.cpython-39.pyc
a
'�Dgm+ � @ s` 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(e e)d�dd�Z*e d�dd�Z+e e,d �d d �Z-d d � Z.e d!k�r\e"� �/� Z0e.e!e0� 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� | st jttd�} t| _| jdddddgd� | jdd d
�}d|_|jd d d
� |jddd
�}|jdddddd� |jddddddgd� |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�remove�enablez%Enable hotplug for a given subsystem.) �argparse�ArgumentParser�NAME�__doc__r � add_argument�add_subparsersr �
add_parser)�parserZ
subparsersZ
parser_handle� r' �D/usr/lib/python3.9/site-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/ r0 r' r' r( �__init__R s
zUeventHandler.__init__c C s
t � �d S r+ ��NotImplementedError�r1 r' r' r( �applyY s zUeventHandler.applyc C s
t � �d S r+ r3 r5 r' r' r( �config] s zUeventHandler.config��returnc C s
t � �d S r+ r3 r5 r' r' r( �device_detectedb 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/ �
ValueErrorr: � RuntimeErrorr, )r1 Zdetect_presencer' r' r( �detect_hotplugged_devicef s
�z&UeventHandler.detect_hotplugged_devicec C s | � � S r+ )r0 r5 r' r' r( �successt 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 �HOTPLUGr< )r1 �resultr' r' r( �update_metadataw s �
��z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r2 �abc�abstractmethodr6 �propertyr7 �boolr: r= r> rA r' r' r' r( r* Q s
r* c s> e Zd Z� fdd�Zdd� Zedd� �Zed�dd �Z� Z S )
�
NetHandlerc s* t tj�|�d�}t� �|||||� d S )N�address)r �os�path�basename�superr2 )r1 r- r. r/ r0 r, �� __class__r' r( r2 � 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_configr7 rK rL rM r. Znetwork_activatorr/ Zbring_up_interfacer< �formatZbring_down_interface)r1 Zinterface_nameZ activatorr' r' r( r6 � s �
�
�zNetHandler.applyc C s | j jS r+ )r- Znetwork_configr5 r' r' r( r7 � s zNetHandler.configr8 c s< t � j�}� fdd�|�� D �}t�d� j|� t|�dkS )Nc s g | ]}|� d �� jkr|�qS )Z mac_address)�getr, )�.0Zifacer5 r' r( �
<listcomp>� s �z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r
r7 Ziter_interfaces�LOG�debugr, �len)r1 Znetstate�foundr' r5 r( r: � s
�zNetHandler.device_detected)
rB rC rD r2 r6 rG r7 rH r: �
__classcell__r' r' rO r( rI � s
rI r c
C s^ zt | d }W n4 tyD } z td�|��|�W Y d }~n
d }~0 0 tj| j| jtj |d�S )N� z4hotplug-hook: cannot handle events for subsystem: {})r- �cfgZevent_source_type�scope)
�SUBSYSTEM_PROPERTIES_MAP�KeyErrorr< rQ r Zupdate_event_enabledr- r[ r r? )� hotplug_init� subsystemr\ �er' r' r( �
is_enabled� s ����rb �r_ r` c C sR t �d� | jdd�}|�tjg�s4t �d|� d S t| |�sNt �d|� d S |S )NzFetching datasource�trust��existingz*hotplug not supported for event of type %sz(hotplug not enabled for event of type %s)rU rV �fetch�get_supported_eventsr r? rb )r_ r` r- r' r' r( �initialize_datasource� s
ri )r_ c C s t | |�}|sd S t| d }t�d|� ||||| jd�}g d�}td�}t|�D ]�\} }
t�d|| t|�� zZt�d� |�� |j s�t�d� |�
� t�d � |� � t�d
� |� � W �qW qT t�y
} z&t�d | � t
�|
� | }W Y d } ~ qTd } ~ 0 0 qT|�d S ) Nr zCreating %s event handler)r- r. r/ r0 )rZ � � �
� 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)ri r] rU rV Z_write_to_cache� Exception� enumeraterW rA Zskip_hotplug_detectr= r6 r> �time�sleep) r_ r. r` �
udevactionr- Z handler_clsZ
event_handlerZ
wait_timesZlast_exceptionZattempt�waitra r' r' r( �handle_hotplug� sH
��
rt )r_ r9 c C s� | j dd�}|sdS t| d }tj|�tjg��|t� �v }|sZtd|� �tj d� dS t
� | j �}|j
|d v r�td|� d �tj d� d
S |d �|j
� t
j| j �d �t�|�d d
d� t|d
| jd� d
S )Nrd re FrZ z#hotplug not supported for event of )�fileZscopesz)Not installing hotplug for event of type z. Reason: Already done.Tzhotplug.enabled�wi� )Zomode�mode)Znetwork_hotplug_enabledr[ )rg r] r r? rh rR �set�print�sys�stderrr Zread_hotplug_enabled_file�paths�value�appendZ
write_fileZ get_cpath�json�dumpsr r[ )r_ r` r- r\ Zhotplug_supportedZhotplug_enabled_filer' r' r( �enable_hotplug� s< � �
�
��r� c C s� t j| tdd�}tg |d�}|�� t�|j� d|jv rLt� |j�
d�� t � d| |j
|jd|v rj|jnd d|v rz|jnd � |�� z�|j
dkr�zt||j�}W n$ ty� td � t�d
� Y n0 t|r�d nd � np|j
d
kr�t||j|j|jd� nNt�� dk�r tj�d� t�d
� t||jd��s:t�d
� td|j� �� W n t �yl t � d� � Y n0 W d � n1 �s�0 Y 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}rr r. r z9Unable to determine hotplug state. No datasource detectedrZ ZenabledZdisabledr )r_ r. r` rr r z9Root is required. Try prepending your command with sudo.
rc z)Enabled cloud-init hotplug for subsystem=z*Received fatal exception handling hotplug!zExiting hotplug handler) r ZReportEventStackr" r Zread_cfgr Z
setup_loggingr[ r Zupdate_configurationrR rU rV r r` rr r. ri r
ry rz �exitrt rK �getuidr{ �writer� rn � exceptionZ flush_events)�name�argsZhotplug_reporterr_ r- r' r' r( � handle_args sn �
�
� �
��
�
��
(
r� �__main__)N)1r" rE r r ZloggingrK rz rp Z cloudinitr r r Z#cloudinit.config.cc_install_hotplugr Zcloudinit.eventr r Z
cloudinit.logr Z
cloudinit.netr Zcloudinit.net.network_stater
Zcloudinit.reportingr Zcloudinit.sourcesr r
Zcloudinit.stagesr Z getLoggerrB rU r! r) �ABCr* rI ZNETWORKr] rb �strri rt rH r� r� �
parse_argsr� r' r' r' r( �<module> s>
62(
�(#E