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^ | dk st dt| tks$t dtjt| 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 */
};
Nz data is nonez+data is smaller than netlink message headerzGot netlink msg of type %d)
AssertionErrorlen
NLMSGHDR_SIZEstructunpack NLMSGHDR_FMTMSG_TYPE_OFFSETr( r) r )r Zmsg_lenZmsg_typer r r r! r! r" get_netlink_msg_headerS s r5 c C s^ | dk st dtj| gg g |\}}}| |kr4dS tjd | jt}|dkrZtjd |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
Nznetlink socket is nonez netlink socket ready for readz,Reading from Netlink socket returned no data)r. selectr( r) ZrecvMAX_SIZEr' )r* ZtimeoutZread_set_r r! r! r" read_netlink_socketl s
r9 c C s | dk st dt|ts"t d|tks2t dd }}d}y0tjd| |dd }tjd| |d dd }W n tjk
r 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.
Nz data is nonezoffset is not integerz'rta offset is less than expected lengthr H)offsetr ) r.
isinstanceintRTATTR_START_OFFSETr1 unpack_fromr' RTA_DATA_START_OFFSETr )r r; r r Z attr_datar! r! r" unpack_rta_attr s
rA c C s | dk st dt| tks$t dd }}t}x|t| krt| |}| sX|jdkrZP t|jt t }||j| 7 }|jtkrt|j }q2|jt
kr2t j |j d}|j
d}q2W | s|dkrdS tjd|| 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.
Nz data is nonez2length of data is smaller than RTATTR_START_OFFSETr zutf-8 z!rta attrs: ifname %s operstate %d)r. r/ r> rA r
PAD_ALIGNMENTr IFLA_OPERSTATEordr IFLA_IFNAMEr Z
decode_binarystripr( r) r )r r r r; attrpadlenZinterface_namer! r! r" read_rta_oper_state s*
rJ 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* rN rO r! )rN r r" wait_for_nic_attach_event s
rT 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! )rK rL rM )r r! r" rO s z5wait_for_nic_detach_event..should_continue_cb)r( r) rP RTM_DELLINKrS )r* rO r! )r r" wait_for_nic_detach_event s
rV c sf | dk st d dk s t dt dks4t d fdd}tjd 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.
Nznetlink socket is nonezinterface name is noner z interface name cannot be emptyc s( |t ko|tk}|r$tjd dS dS )Nz Media switch happened on %s.FT)rS rR r( r) )rK rL rM Z isVnetSwitch)r r! r" rO s
z=wait_for_media_disconnect_connect..should_continue_cbz1Wait for media disconnect and reconnect to happen) r. r/ r( r) rP rQ rU rR rS )r* r rO r! )r r" !wait_for_media_disconnect_connect s
rW 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.) RuntimeErrorbytesrR r9 SELECT_TIMEOUTr( r) r/ r0 r5 r rC r rJ r r )r* Z
ifname_filterZ rtm_typesZ
operstatesZshould_continue_callbackr rL rM Z recv_datar; ZdatalenZnl_msgZnlheaderrI Zinterface_stater! r! r" rP s`
rP )N)7Zloggingr$ r6 r# r1 collectionsr Z cloudinitr Z getLoggerr r( r& Z
NLMSG_NOOPZ NLMSG_ERRORZ
NLMSG_DONErQ rU Z RTM_GETLINKZ RTM_SETLINKr7 ZRTA_DATA_OFFSETr4 rZ r3 Z
IFINFOMSG_FMTcalcsizer0 ZIFINFOMSG_SIZEr> r@ rC rF rD Z OPER_UNKNOWNZOPER_NOTPRESENTrS ZOPER_LOWERLAYERDOWNZ OPER_TESTINGZ OPER_DORMANTrR r r r rX r r- r5 r9 rA rJ rT rV rW rP r! r! r! r" sb
%