File: //usr/lib/python3/dist-packages/cloudinit/cmd/devel/__pycache__/hotplug_hook.cpython-38.pyc
U
��0h�- � @ sp 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 dd�dd�Z+dd �d d �Z,e e-d�d d �Z.d!d"� Z/e d#k�rle"� �0� Z1e/e!e1� 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_argumentZadd_subparsersr Z
add_parser)�parserZ
subparsersZ
parser_handle� r% �B/usr/lib/python3/dist-packages/cloudinit/cmd/devel/hotplug_hook.py�
get_parser sT � � � ��� �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__R s
zUeventHandler.__init__c C s
t � �d S r) ��NotImplementedError�r/ r% r% r& �applyY s zUeventHandler.applyc C s
t � �d S r) r1 r3 r% r% r& �config] s zUeventHandler.config��returnc C s
t � �d S r) r1 r3 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- �
ValueErrorr8 � RuntimeErrorr* )r/ Zdetect_presencer% r% r& �detect_hotplugged_devicef s
�z&UeventHandler.detect_hotplugged_devicec C s | � � S r) )r. r3 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: )r/ �resultr% r% r& �update_metadataw s �
��z UeventHandler.update_metadataN)�__name__�
__module__� __qualname__r0 �abc�abstractmethodr4 �propertyr5 �boolr8 r; r< r? 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 )NZaddress)r �os�path�basename�superr0 )r/ r+ r, r- r. r* �� __class__r% r& r0 � 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_configr5 rH rI rJ r, Znetwork_activatorr- Zbring_up_interfacer: �formatZbring_down_interface)r/ Zinterface_nameZ activatorr% r% r& r4 � s �
�
�zNetHandler.applyc C s | j jS r) )r+ Znetwork_configr3 r% r% r& r5 � s zNetHandler.configr6 c s< t � j�}� fdd�|�� D �}t�d� j|� t|�dkS )Nc s g | ]}|� d �� jkr|�qS )Z mac_address)�getr* )�.0Zifacer3 r% r& �
<listcomp>� s �z.NetHandler.device_detected.<locals>.<listcomp>zIfaces with ID=%s : %sr )r
r5 Ziter_interfaces�LOG�debugr* �len)r/ Znetstate�foundr% r3 r& r8 � s
�zNetHandler.device_detected)
r@ rA rB r0 r4 rE r5 rF r8 �
__classcell__r% r% rL r& rG � s
rG 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_PROPERTIES_MAP�KeyErrorr: rN r Zupdate_event_enabledr+ rX r r= )� hotplug_init� subsystemrY �er% r% r& �
is_enabled� s ����r_ �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)rR rS �fetch�get_supported_eventsr r= r_ )r\ r] r+ r% r% r& �initialize_datasource� s
rf )r\ r7 c C s� t | |�}|sd S t| d }t�d|� ||||| jd�}t�� }|jjs\t|||� d S t�� | |jj k r�t|||� t�d� t�
|jj � q\d S )Nr zCreating %s event handler)r+ r, r- r. z>Gathering network configuration again due to IMDS limitations.) rf rZ rR rS Z_write_to_cache�timeZhotplug_retry_settingsZ force_retry� try_hotplugZ sleep_total�sleepZ sleep_period)r\ r, r] �
udevactionr+ Z handler_cls�
event_handler�startr% r% r& �handle_hotplug� s*
� �rm r6 c C s� dddddg}t d�}t|�D ]�\}}t�d| |t|�� zXt�d� |�� |jsft�d � |�� t�d
� |�� t�d � |� � W q�W q t k
r� } z t�d |� t
� |� |}W 5 d }~X Y q X q |�d S )
NrW � � �
� 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� enumeraterR rS rT r? Zskip_hotplug_detectr; r4 r< rg ri )r] rk r+ Z
wait_timesZlast_exceptionZattempt�waitr^ r% r% r& rh � s2 �
rh c C s� | j dd�}|sdS t| d }tj|�tjg��|t� �k}|sZtd|� �tj d� dS t
� | j �}|j
|d kr�td|� d �tj d� d
S |d �|j
� t
j| j �d �t�|�d d
d� t|d
| jd� d
S )Nra rb FrW 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_enabledrX )rd rZ r r= re rO �set�print�sys�stderrr Zread_hotplug_enabled_file�paths�value�appendZ
write_fileZ get_cpath�json�dumpsr rX )r\ r] r+ rY Zhotplug_supportedZhotplug_enabled_filer% r% r& �enable_hotplug sB � �
�
� �r� c C s� t j| tdd�}tg |d�}|�� t�|j� d|jkrLt� |j�
d�� t � d| |j
|jd|krj|jnd d|krz|jnd � |�� z�|j
dkr�zt||j�}W n& tk
r� td � t�d
� Y nX 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 k
�rp 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}rj r, r z9Unable to determine hotplug state. No datasource detectedrW �enabledZdisabledr )r\ r, r] rj r z9Root is required. Try prepending your command with sudo.
r` z)Enabled cloud-init hotplug for subsystem=z*Received fatal exception handling hotplug!zExiting hotplug handler) r ZReportEventStackr" r Zread_cfgr Z
setup_loggingrX r Zupdate_configurationrO rR rS r r] rj r, rf r
ry rz �exitrm rH �getuidr{ �writer� rr Z exceptionZ flush_events)�name�argsZhotplug_reporterr\ r+ r% r% r& � handle_args' sr �
�
��
��
�
�
r� �__main__)N)2r" rC r r ZloggingrH rz rg 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 getLoggerr@ rR r! r' �ABCr( rG ZNETWORKrZ r_ �strrf rm rh rF r� r� �
parse_argsr� r% r% r% r& �<module> sB
62( � #E