3
\me# @ s0 d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
ddl m Z m
Z
ddlmZ ddlmZ ddlmZ ddlmZmZ dd lmZ ejeZd
Z d d d Z G d
d dej Z G dd de Z de e j!fiZ"dd Z#ee$dddZ%edddZ&dd Z'ed kr,e j( Z)e'e e) dS ) z)Handle reconfiguration on hotplug events. N)log reportingstages)
EventScope EventType)read_sys_net_safe)parse_net_config_data)events)
DataSourceDataSourceNotFoundException)Initz hotplug-hookc C s | st 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 onnet)requiredhelpchoiceszHotplug Actionhotplug_action)titledestqueryz0Query if hotplug is enabled for given subsystem.)r ZhandlezHandle the hotplug event.z-dz --devpathPATHz Sysfs path to hotplugged device)r metavarr z-uz --udevactionzSpecify action to take.addremove) argparseArgumentParserNAME__doc__r add_argumentZadd_subparsersr Z
add_parser)parserZ
subparsersZ
parser_handle r! "/usr/lib/python3.6/hotplug_hook.py
get_parser s:
r# c @ s` e Zd Zdd Zejdd Zeejdd Zeje ddd Z
d
d Z d d
Z dd Z
dS )
UeventHandlerc C s" || _ || _|| _|| _|| _d S )N)id
datasourcedevpathaction
success_fn)selfr% r& r' r( r) r! r! r" __init__K s
zUeventHandler.__init__c C s
t d S )N)NotImplementedError)r* r! r! r" applyR s zUeventHandler.applyc C s
t d S )N)r, )r* r! r! r" configV s zUeventHandler.config)returnc C s
t d S )N)r, )r* r! r! r" device_detected[ s z UeventHandler.device_detectedc C sP d }| j dkrd}n | j dkr$d}ntd| j || j krLtd| j d S )Nr Tr FzUnknown action: %sz'Failed to detect %s in updated metadata)r(
ValueErrorr0 RuntimeErrorr% )r* Zdetect_presencer! r! r" detect_hotplugged_device_ s
z&UeventHandler.detect_hotplugged_devicec C s | j S )N)r) )r* r! r! r" successm s zUeventHandler.successc C s, | j jtjg}|s(td| j tjf |S )Nz&Datasource %s not updated for event %s)r& Z update_metadata_if_supportedr HOTPLUGr2 )r* resultr! r! r" update_metadatap s
z UeventHandler.update_metadataN)__name__
__module__ __qualname__r+ abcabstractmethodr- propertyr. boolr0 r3 r4 r7 r! r! r! r" r$ J s r$ c s> e Zd Z fddZdd Zedd Zeddd Z Z S )
NetHandlerc s* t tjj|d}t j||||| d S )NZaddress)r ospathbasenamesuperr+ )r* r& r' r( r) r% ) __class__r! r" r+ } s zNetHandler.__init__c C s| | j jj| jdd tjj| j}| j jj }| j dkrT|j
|sxt dj | jn$| j dkrx|j
|sxt dj | jd S )NF)Zbring_upr z Failed to bring up device: {}r z Failed to bring down device: {})r& ZdistroZapply_network_configr. r@ rA rB r' Znetwork_activatorr( Zbring_up_interfacer2 formatZbring_down_interface)r* Zinterface_nameZ activatorr! r! r" r- s
zNetHandler.applyc C s | j jS )N)r& Znetwork_config)r* r! r! r" r. s zNetHandler.config)r/ c s< t j} fdd|j D }tjd j| t|dkS )Nc s g | ]}|j d jkr|qS )Z mac_address)getr% ).0Ziface)r* r! r"
s z.NetHandler.device_detected..zIfaces with ID=%s : %sr )r r. Ziter_interfacesLOGdebugr% len)r* Znetstatefoundr! )r* r" r0 s
zNetHandler.device_detected)
r8 r9 r: r+ r- r= r. r> r0
__classcell__r! r! )rD r" r? | s r? r c C s\ yt | d }W n2 tk
rB } ztdj||W Y d d }~X nX tj| j| jtj |dS )N z4hotplug-hook: cannot handle events for subsystem: {})r& cfgZevent_source_typescope)
SUBSYSTEM_PROPERTES_MAPKeyErrorr2 rE r Zupdate_event_enabledr& rO r r5 ) hotplug_init subsystemrP er! r! r"
is_enabled s rV )rS rT c C sR t jd | jdd}|jtjgs4t jd| d S t| |sNt jd| d S |S )NzFetching datasourceZtrust)Zexistingz*hotplug not supported for event of type %sz(hotplug not enabled for event of type %s)rI rJ ZfetchZget_supported_eventsr r5 rV )rS rT r& r! r! r" initialize_datasource s
rW )rS c C s t | |}|sd S t| d }tjd| ||||| jd}dddddg}td }xt|D ]\} }
tjd
|| t| yTtjd |j |j stjd |j
tjd
|j tjd |j P W q\ tk
r
} z tjd| t
j|
| }W Y d d } ~ X q\X q\W |d S )Nr zCreating %s event handler)r& r' r( r) rN
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)rW rQ rI rJ Z_write_to_cache Exception enumeraterK r7 Zskip_hotplug_detectr3 r- r4 timeZsleep) rS r' rT
udevactionr& Z handler_clsZ
event_handlerZ
wait_timesZlast_exceptionZattemptwaitrU r! r! r" handle_hotplug sD
ra c C s8 t j| tdd}tg |d}|j tj|j d|jkrLtj |jj
d t j d| |j
|jd|krj|jnd d|krz|jnd | yn|j
dkryt||j}W n& tk
r td tjd
Y nX t|rd nd nt||j|j|jd
W n" tk
r t jd Y nX W d Q R X t j d tj d S )NT)Zreporting_enabled)Zds_depsZreporterr 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 detectedrN ZenabledZdisabled)rS r' rT r_ z*Received fatal exception handling hotplug!zExiting hotplug handler)r ZReportEventStackr r Zread_cfgr Z
setup_loggingrO r Zupdate_configurationrF rI rJ r rT r_ r' rW r printsysexitra r\ Z exceptionZ flush_events)nameargsZhotplug_reporterrS r& r! r! r" handle_args sF
rg __main__)N)*r r; r Zloggingr@ rc r^ Z cloudinitr r r Zcloudinit.eventr r Z
cloudinit.netr Zcloudinit.net.network_stater Zcloudinit.reportingr Zcloudinit.sourcesr
r Zcloudinit.stagesr Z getLoggerr8 rI r r# ABCr$ r? ZNETWORKrQ rV strrW ra rg
parse_argsrf r! r! r! r" s4
22((5