Google
Edit File: 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