"T`0@sddlmZddlmZddlmZddlZddlZddlZddl Z ej e Z dZ dZdZdZdZd Zd Zd Zd Zd ZdZdZdZdZe jeZe jeZeeZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)eddddgZ*edddgZ+eddddddgZ,Gd d!d!e-Z.d"d#Z/d$d%Z0dd&d'Z1d(d)Z2d*d+Z3d,d-Z4d.d/Z5d0d1Z6d2d3Z7dS)4)log)util) namedtupleNi <ZIHHIIZBHiIIRTAAttrlengthrta_typedataInterfaceOperstateifname operstate NetlinkHeadertypeflagsseqpidc@seZdZdZdS)NetlinkCreateSocketErrorz5Raised if netlink socket fails during create or bind.N)__name__ __module__ __qualname____doc__r"r"C/usr/lib/python3/dist-packages/cloudinit/sources/helpers/netlink.pyr8s rcCsyHtjtjtjtj}|jtjtf|jdWnAtj k r}zd|}t ||WYdd}~XnXt j d|S)auCreates 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 rz*Exception during netlink socket create: %sNzCreated netlink socket) socketZ AF_NETLINKZSOCK_RAWZ NETLINK_ROUTEZbindosgetpid RTMGRP_LINKZ setblockingerrorrLOGdebug)netlink_socketemsgr"r"r#create_bound_netlink_socket<s    " r.cCs|dk stdt|tks6tdtjt|dt\}}}}}tjd|t |||||S)aGets 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_SIZEstructZunpack NLMSGHDR_FMTMSG_TYPE_OFFSETr)r*r)rZmsg_lenZmsg_typerrrr"r"r#get_netlink_msg_headerRs  "r5cCs|dk stdtj|ggg|\}}}||krLdStjd|jt}|dkrtjd|S)aSelect 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 noneznetlink socket ready for readz,Reading from Netlink socket returned no data)r/selectr)r*ZrecvMAX_SIZEr()r+ZtimeoutZread_set_rr"r"r#read_netlink_socketis $    r9c Cs|dk stdt|tks6td|tksNtdd}}d}y@tjd|d|d}tjd|d|dd}Wntjk rdSYnX||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 lengthrHoffsetr) r/rintRTATTR_START_OFFSETr2Z unpack_fromr(RTA_DATA_START_OFFSETr)rr;rrZ attr_datar"r"r#unpack_rta_attrs  $ r?cCs6|dk stdt|tks6tdd}}t}x|t|krt||}| s|jdkrPt|jtt}||j|7}|jtkrt|j }qI|jt krIt j |j d}|j d}qIW| s|dkrdStjd||t||S)aReads 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_OFFSETrzutf-8z!rta attrs: ifname %s operstate %d)r/r0r=r?r PAD_ALIGNMENTrIFLA_OPERSTATEordr IFLA_IFNAMErZ decode_binarystripr)r*r)rrrr;attrpadlenZinterface_namer"r"r#read_rta_oper_states*   rHcsKtjddfdd}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.Ncs|krdS|dS)NTFr")inamecarrier prevCarrier) existing_nicsrr"r#should_continue_cbs z5wait_for_nic_attach_event..should_continue_cb)r)r*read_netlink_messages RTM_NEWLINKOPER_UP OPER_DOWN)r+rLrMr")rLrr#wait_for_nic_attach_events   rRcsEtjddfdd}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.Ncs |dS)NFr")rIrJrK)rr"r#rMsz5wait_for_nic_detach_event..should_continue_cb)r)r*rN RTM_DELLINKrQ)r+rMr")rr#wait_for_nic_detach_events rTcs|dk stddk s0tdtdksNtdfdd}tjdt|ttgttg|dS) aBlock 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 nonerzinterface name cannot be emptycs6|tko|tk}|r2tjddSdS)NzMedia switch happened on %s.FT)rQrPr)r*)rIrJrKZ isVnetSwitch)rr"r#rMs z=wait_for_media_disconnect_connect..should_continue_cbz1Wait for media disconnect and reconnect to happen) r/r0r)r*rNrOrSrPrQ)r+rrMr")rr#!wait_for_media_disconnect_connects    rUcCs|dkrtdt}t}t}xt|t}|dkrNq0tjdt|||7}tjdt|d} t|} xN| | kr|| d} t| tkrtjdPt | } t| | j krtjdP| j t dt d@} | | } tjd | | j |krJqt | }|dkrutjd |q|dk r|j|krtjd |j|q|j|krq|}|j}||j||sdSqW|| d}q0WdS) 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 socketzLength of data after concat %drz#Data is smaller than netlink headerz*Partial data. Smaller than netlink messagerz"offset to next netlink message: %dz!Failed to read rta attributes: %sz6Ignored netlink event on interface %s. Waiting for %s.) RuntimeErrorbytesrPr9SELECT_TIMEOUTr)r*r0r1r5rrArrHrr)r+Z ifname_filterZ rtm_typesZ operstatesZshould_continue_callbackrrJrKZ recv_datar;ZdatalenZnl_msgZnlheaderrGZinterface_stater"r"r#rN s^                 rN)8Z cloudinitrZloggingr collectionsrr%r6r$r2Z getLoggerrr)r'Z NLMSG_NOOPZ NLMSG_ERRORZ NLMSG_DONErOrSZ RTM_GETLINKZ RTM_SETLINKr7ZRTA_DATA_OFFSETr4rXr3Z IFINFOMSG_FMTZcalcsizer1ZIFINFOMSG_SIZEr=r>rArDrBZ OPER_UNKNOWNZOPER_NOTPRESENTrQZOPER_LOWERLAYERDOWNZ OPER_TESTINGZ OPER_DORMANTrPrrrrVrr.r5r9r?rHrRrTrUrNr"r"r"r#sb          #