File: //lib/python3/dist-packages/cloudinit/cmd/devel/__pycache__/hotplug_hook.cpython-310.pyc
o
�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"dede#fdd�Z$defdd�Z%dd� Z&ed kr�e � �'� Z(e&ee(� dS 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� | s t 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 �handlezHandle 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_argument�add_subparsersr �
add_parser)�parser�
subparsers�
parser_handle� r& �B/usr/lib/python3/dist-packages/cloudinit/cmd/devel/hotplug_hook.py�
get_parser sF ������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- r. r/ r& r& r' �__init__J s
zUeventHandler.__init__c C � t � �r* ��NotImplementedError�r0 r& r& r' �applyQ � zUeventHandler.applyc C r2 r* r3 r5 r& r& r' �configU s zUeventHandler.config�returnc C r2 r* r3 r5 r& r& r' �device_detectedZ r7 z UeventHandler.device_detectedc C sP d }| j dkr
d}n| j dkrd}ntd| j ��|| �� kr&td| j ��d S )Nr Tr FzUnknown action: %sz'Failed to detect %s in updated metadata)r. �
ValueErrorr: � RuntimeErrorr+ )r0 �detect_presencer& r& r' �detect_hotplugged_device^ s
��z&UeventHandler.detect_hotplugged_devicec C s | � � S r* )r/ r5 r& r& r' �successl 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< )r0 �resultr& r& r' �update_metadatao s �
��z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r1 �abc�abstractmethodr6 �propertyr8 �boolr: r> r? rC r& r& r& r' r) I 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�superr1 )r0 r, r- r. r/ r+ �� __class__r& r' r1 | s zNetHandler.__init__c C s� | j jj| jdd� tj�| j�}| j j�� }| j dkr+|�
|�s)t d� | j���d S | j dkr=|�
|�s?t d� | j���d S d S )NF)�bring_upr z Failed to bring up device: {}r z Failed to bring down device: {})r, �distro�apply_network_configr8 rM rN rO r- �network_activatorr. �bring_up_interfacer< �format�bring_down_interface)r0 �interface_name� activatorr& r& r' r6 � s& �
��
��zNetHandler.applyc C s | j jS r* )r, �network_configr5 r& r& r' r8 � s zNetHandler.configr9 c s< t � j�}� fdd�|�� D �}t�d� j|� t|�dkS )Nc s g | ] }|� d �� jkr|�qS )� mac_address)�getr+ )�.0�ifacer5 r& r' �
<listcomp>� s ��z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r r8 �iter_interfaces�LOG�debugr+ �len)r0 �netstate�foundr& r5 r' r: � s
� zNetHandler.device_detected)
rD rE rF r1 r6 rI r8 rJ r: �
__classcell__r& r& rQ r' rK { s
rK 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_PROPERTES_MAP�KeyError� ExceptionrX r �update_event_enabledr, rj r rA )� hotplug_init� subsystemrl �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)rc rd �fetch�get_supported_eventsr rA rt )rq rr r, r& r& r' �initialize_datasource� s
ry c C s
t | |�}|s d S t| d }t�d|� ||||| jd�}g d�}td�}t|�D ]X\} }
t�d|| t|�� z+t�d� |�� |j sNt�d� |�
� t�d � |� � t�d
� |� � W d S ty� } zt�d | � t
�|
� | }W Y d } ~ q*d } ~ ww |�) Nr zCreating %s event handler)r, r- r. r/ )ri � � �
� 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)ry rm rc rd �_write_to_cachero � enumeratere rC �skip_hotplug_detectr> r6 r? �time�sleep) rq r- rr �
udevactionr, � handler_cls�
event_handler�
wait_times�last_exception�attempt�waitrs r& r& r' �handle_hotplug� sL
��
��r� c C sB t j| tdd�}tg |d�}|�� t�|j� d|jv r&t� |j�
d�� t � d| |j
|jd|v r5|jnd d|v r=|jnd � |�L z6|j
dkrmzt||j�}W n tyc td � t�d
� Y nw t|rid nd � n t||j|j|jd
� W n
ty� 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 detectedri �enabled�disabled)rq r- rr r� z*Received fatal exception handling hotplug!zExiting hotplug handler)r �ReportEventStackr r �read_cfgr � setupLoggingrj r �update_configurationr^ rc rd r rr r� r- ry r �print�sys�exitr� ro � exception� flush_events)�name�args�hotplug_reporterrq r, r& r& r' � handle_args� sZ �
�
� ����
�� �
r� �__main__r* ))r rG r rM r� r� � cloudinitr r r �cloudinit.eventr r �
cloudinit.netr �cloudinit.net.network_stater �cloudinit.reportingr �cloudinit.sourcesr
r �cloudinit.stagesr � getLoggerrD rc r r( �ABCr) rK �NETWORKrm rt �strry r� r� �
parse_argsr� r& r& r& r' �<module> s8
22
(�(5
�