3
\me. @ sZ d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ e j e
Z dZ dZ
dZdZdZdZd Zd
Zd Zd ZdZd
ZdZdZejeZejeZ ee Z dZ dZ dZ dZ!d Z"dZ#dZ$dZ%dZ&dZ'dZ(eddddgZ)edddgZ*edddd d d gZ+G d d d e,Z-d!d" Z.d#d$ Z/d3d%d&Z0d'd( Z1d)d* Z2d+d, Z3d-d. Z4d/d0 Z5d1d2 Z6dS )4 N)
namedtuple)util i < ZIHHIIZBHiII RTAAttrlengthrta_typedataInterfaceOperstateifname operstate
NetlinkHeadertypeflagsseqpidc @ s e Zd ZdZdS )NetlinkCreateSocketErrorz5Raised if netlink socket fails during create or bind.N)__name__
__module__ __qualname____doc__ r! r! /usr/lib/python3.6/netlink.pyr 9 s r c C sz y4t j t jt jt j} | jtj tf | jd W n6 t j k
rj } zd| }t
||W Y dd}~X nX t j d | S )au Creates netlink socket and bind on netlink group to catch interface
down/up events. The socket will bound only on RTMGRP_LINK (which only
includes RTM_NEWLINK/RTM_DELLINK/RTM_GETLINK events). The socket is set to
non-blocking mode since we're only receiving messages.
:returns: netlink socket in non-blocking mode
:raises: NetlinkCreateSocketError
r z*Exception during netlink socket create: %sNzCreated netlink socket)
socketZ
AF_NETLINKZSOCK_RAWZ
NETLINK_ROUTEZbindosgetpid RTMGRP_LINKZ setblockingerrorr LOGdebug)netlink_socketemsgr! r! r" create_bound_netlink_socket= s
r- c C s: t jt| dt \}}}}}tjd| t|||||S )a Gets netlink message type and length
:param: data read from netlink socket
:returns: netlink message type
:raises: AssertionError if data is None or data is not >= NLMSGHDR_SIZE
struct nlmsghdr {
__u32 nlmsg_len; /* Length of message including header */
__u16 nlmsg_type; /* Type of message content */
__u16 nlmsg_flags; /* Additional flags */
__u32 nlmsg_seq; /* Sequence number */
__u32 nlmsg_pid; /* Sender port ID */
};
NzGot netlink msg of type %d)structunpack NLMSGHDR_FMTMSG_TYPE_OFFSETr( r) r )r Zmsg_lenZmsg_typer r r r! r! r" get_netlink_msg_headerS s r2 c C sN t j | gg g |\}}}| |kr$dS tjd | jt}|dkrJtjd |S )a Select and read from the netlink socket if ready.
:param: netlink_socket: specify which socket object to read from
:param: timeout: specify a timeout value (integer) to wait while reading,
if none, it will block indefinitely until socket ready for read
:returns: string of data read (max length = ) from socket,
if no data read, returns None
:raises: AssertionError if netlink_socket is None
Nz netlink socket ready for readz,Reading from Netlink socket returned no data)selectr( r) ZrecvMAX_SIZEr' )r* ZtimeoutZread_set_r r! r! r" read_netlink_socketl s
r6 c C st d }}d}y0t jd| |dd }t jd| |d dd }W n t jk
rR dS X | |t || }t|||S )a( Unpack a single rta attribute.
:param: data: string of data read from netlink socket
:param: offset: starting offset of RTA Attribute
:return: RTAAttr object with length, type and data. On error, return None.
:raises: AssertionError if data is None or offset is not integer.
r NH)offsetr )r. unpack_fromr' RTA_DATA_START_OFFSETr )r r8 r r Z attr_datar! r! r" unpack_rta_attr s
r; c C s d }}t }x|t| krt| |}| s4|jdkr6P t|jt t }||j| 7 }|jtkrlt|j}q|jt krt
j |jd}|j d}qW | s|dkrdS t
jd|| t||S )a Reads Interface name and operational state from RTA Data.
:param: data: string of data read from netlink socket
:returns: InterfaceOperstate object containing if_name and oper_state.
None if data does not contain valid IFLA_OPERSTATE and
IFLA_IFNAME messages.
:raises: AssertionError if data is None or length of data is
smaller than RTATTR_START_OFFSET.
Nr zutf-8 z!rta attrs: ifname %s operstate %d)RTATTR_START_OFFSETlenr; r
PAD_ALIGNMENTr IFLA_OPERSTATEordr IFLA_IFNAMEr Z
decode_binarystripr( r) r )r r r r8 attrpadlenZinterface_namer! r! r" read_rta_oper_state s$
rF c s6 t jd d fdd}t| dtgttg| S )zBlock until a single nic is attached.
:param: netlink_socket: netlink_socket to receive events
:param: existing_nics: List of existing nics so that we can skip them.
:raises: AssertionError if netlink_socket is none.
z!Preparing to wait for nic attach.Nc s | kr dS | dS )NTFr! )inamecarrier prevCarrier)
existing_nicsr r! r" should_continue_cb s z5wait_for_nic_attach_event..should_continue_cb)r( r) read_netlink_messages RTM_NEWLINKOPER_UP OPER_DOWN)r* rJ rK r! )rJ r r" wait_for_nic_attach_event s
rP c s2 t jd d fdd}t| dtgtg| S )zBlock until a single nic is detached and its operational state is down.
:param: netlink_socket: netlink_socket to receive events.
z!Preparing to wait for nic detach.Nc s | dS )NFr! )rG rH rI )r r! r" rK s z5wait_for_nic_detach_event..should_continue_cb)r( r) rL RTM_DELLINKrO )r* rK r! )r r" wait_for_nic_detach_event s
rR c s2 fdd}t jd t| ttgttg| dS )a Block until media disconnect and connect has happened on an interface.
Listens on netlink socket to receive netlink events and when the carrier
changes from 0 to 1, it considers event has happened and
return from this function
:param: netlink_socket: netlink_socket to receive events
:param: ifname: Interface name to lookout for netlink events
:raises: AssertionError if netlink_socket is None or ifname is None.
c s( |t ko|tk}|r$tjd dS dS )Nz Media switch happened on %s.FT)rO rN r( r) )rG rH rI Z isVnetSwitch)r r! r" rK s
z=wait_for_media_disconnect_connect..should_continue_cbz1Wait for media disconnect and reconnect to happenN)r( r) rL rM rQ rN rO )r* r rK r! )r r" !wait_for_media_disconnect_connect s
rS c C s | dkrt dt }t}t}xZt| t}|dkr6q"tjdt| ||7 }tjdt| d} t|}
x| |
k rj|| d } t| tk rtjd P t | } t| | j
k rtjd P | j
t d t d @ }
| |
} tjd | | j |krqnt
| }|dkrtjd
| qn|dk r>|j|kr>tjd |j| qn|j|krLqn|}|j}||j||sndS qnW || d }q"W dS ) a Reads from the netlink socket until the condition specified by
the continuation callback is met.
:param: netlink_socket: netlink_socket to receive events.
:param: ifname_filter: if not None, will only listen for this interface.
:param: rtm_types: Type of netlink events to listen for.
:param: operstates: Operational states to listen.
:param: should_continue_callback: Specifies when to stop listening.
NzNetlink socket is nonezread %d bytes from socketz Length of data after concat %dr z#Data is smaller than netlink headerz*Partial data. Smaller than netlink messager z"offset to next netlink message: %dz!Failed to read rta attributes: %sz6Ignored netlink event on interface %s. Waiting for %s.) RuntimeErrorbytesrN r6 SELECT_TIMEOUTr( r) r>
NLMSGHDR_SIZEr2 r r? r rF r r )r* Z
ifname_filterZ rtm_typesZ
operstatesZshould_continue_callbackr rH rI Z recv_datar8 ZdatalenZnl_msgZnlheaderrE Zinterface_stater! r! r" rL s`
rL )N)7Zloggingr$ r3 r# r. collectionsr Z cloudinitr Z getLoggerr r( r& Z
NLMSG_NOOPZ NLMSG_ERRORZ
NLMSG_DONErM rQ Z RTM_GETLINKZ RTM_SETLINKr4 ZRTA_DATA_OFFSETr1 rV r0 Z
IFINFOMSG_FMTcalcsizerW ZIFINFOMSG_SIZEr= r: r? rB r@ Z OPER_UNKNOWNZOPER_NOTPRESENTrO ZOPER_LOWERLAYERDOWNZ OPER_TESTINGZ OPER_DORMANTrN r r r rT r r- r2 r6 r; rF rP rR rS rL r! r! r! r" sb
%