File: //lib/python3/dist-packages/cloudinit/cmd/devel/__pycache__/hotplug_hook.cpython-310.pyc
o
x[h�- � @ 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� |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)�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/ r0 r1 r2 r) r) r* �__init__R s
zUeventHandler.__init__c C � t � �r- ��NotImplementedError�r3 r) r) r* �applyY � zUeventHandler.applyc C r5 r- r6 r8 r) r) r* �config] s zUeventHandler.config�returnc C r5 r- r6 r8 r) r) r* �device_detectedb r: 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)r1 �
ValueErrorr= � RuntimeErrorr. )r3 �detect_presencer) r) r* �detect_hotplugged_devicef s
��z&UeventHandler.detect_hotplugged_devicec C s | � � S r- )r2 r8 r) r) r* �successt 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? )r3 �resultr) r) r* �update_metadataw s �
��z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r4 �abc�abstractmethodr9 �propertyr; �boolr= rA rB rF 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�superr4 )r3 r/ r0 r1 r2 r. �� __class__r) r* r4 � 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_configr; rP rQ rR r0 �network_activatorr1 �bring_up_interfacer? �format�bring_down_interface)r3 �interface_name� activatorr) r) r* r9 � s& �
��
��zNetHandler.applyc C s | j jS r- )r/ �network_configr8 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�ifacer8 r) r* �
<listcomp>� s
�z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r
r; �iter_interfaces�LOG�debugr. �len)r3 �netstate�foundr) r8 r* r= � s
� zNetHandler.device_detected)
rG rH rI r4 r9 rL r; rM r= �
__classcell__r) r) rT r* rN � s
rN 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? r[ r �update_event_enabledr/ rm r rD )� hotplug_init� subsystemro �er) r) r* �
is_enabled� s$ ������rv rs rt 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)rf rg �fetch�get_supported_eventsr rD rv )rs rt r/ r) r) r* �initialize_datasource� s
r| 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/ r0 r1 r2 z>Gathering network configuration again due to IMDS limitations.) r| rp rf rg �_write_to_cache�time�hotplug_retry_settings� force_retry� try_hotplug� sleep_total�sleep� sleep_period)rs r0 rt �
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)rl � � �
� 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� enumeraterf rg rh rF �skip_hotplug_detectrA r9 rB r~ r� )rt r� r/ �
wait_times�last_exception�attempt�waitru 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 )Nrw rx Frl 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_enabledrm )rz rp r rD r{ ra �set�print�sys�stderrr �read_hotplug_enabled_file�paths�value�append�
write_file� get_cpath�json�dumpsr rm )rs rt r/ ro �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� r0 r z9Unable to determine hotplug state. No datasource detectedrl �enabled�disabledr )rs r0 rt r� r z9Root is required. Try prepending your command with sudo.
)rs rt z)Enabled cloud-init hotplug for subsystem=z*Received fatal exception handling hotplug!zExiting hotplug handler) r �ReportEventStackr" r �read_cfgr �
setup_loggingrm r �update_configurationra rf rg r rt r� r0 r| r
r� r� �exitr� rP �getuidr� �writer� r� � exception� flush_events)�name�args�hotplug_reporterrs r/ r) r) r* � handle_args' sx �
�
� ��
� �
�
���
�� �
) r� �__main__r- )r< N)2r" rJ r r� �loggingrP 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 � getLoggerrG rf r! r+ �ABCr, rN �NETWORKrp rv �strr| r� r� rM r� r� �
parse_argsr� r) r) r) r* �<module> sD
62
(�
#E
�