name : netlink.cpython-39.pyc
a

'�Dg�.�@sBddlZddlZddlZddlZddlZddlmZddlmZe�	e
�ZdZdZ
dZdZdZd	ZdZd
ZdZdZe�e�Ze�e�ZeeZd
Zd
ZdZdZdZdZdZ dZ!d
Z"dZ#dZ$edgd��Z%edddg�Z&edgd��Z'Gdd�de(�Z)dd�Z*dd�Z+d-dd �Z,d!d"�Z-d#d$�Z.d%d&�Z/d'd(�Z0d)d*�Z1d+d,�Z2dS).�N)�
namedtuple)�util�����i���<ZIHHIIZBHiII������RTAAttr)�length�rta_type�data�InterfaceOperstate�ifname�	operstate�
NetlinkHeader)r�type�flags�seq�pidc@seZdZdZdS)�NetlinkCreateSocketErrorz5Raised if netlink socket fails during create or bind.N)�__name__�
__module__�__qualname__�__doc__�r r �E/usr/lib/python3.9/site-packages/cloudinit/sources/helpers/netlink.pyr5src
Cs|z4t�tjtjtj�}|�t��tf�|�d�Wn8tj	yl}zd|}t
|�|�WYd}~n
d}~00t�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_ROUTEZbind�os�getpid�RTMGRP_LINKZsetblocking�errorr�LOG�debug)�netlink_socket�e�msgr r r!�create_bound_netlink_socket9s	� 
r,cCs^|dusJd��t|�tks$Jd��t�t|dt��\}}}}}t�d|�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 */
    };
    N�data is nonez+data is smaller than netlink message headerzGot netlink msg of type %d)	�len�
NLMSGHDR_SIZE�struct�unpack�NLMSGHDR_FMT�MSG_TYPE_OFFSETr'r(r)rZmsg_lenZmsg_typerrrr r r!�get_netlink_msg_headerOs
���r4cCs^|dusJd��t�|ggg|�\}}}||vr4dSt�d�|�t�}|durZt�d�|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 = <MAX_SIZE>) from socket,
              if no data read, returns None
    :raises: AssertionError if netlink_socket is None
    N�netlink socket is noneznetlink socket ready for readz,Reading from Netlink socket returned no data)�selectr'r(Zrecv�MAX_SIZEr&)r)�timeoutZread_set�_rr r r!�read_netlink_sockeths



r:cCs�|dusJd��t|t�s"Jd��|tks2Jd��d}}d}z0tjd||d�d}tjd||dd�d}Wntjy�YdS0||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.
    Nr-zoffset is not integerz'rta offset is less than expected lengthr�H)�offsetr)�
isinstance�int�RTATTR_START_OFFSETr0�unpack_fromr&�RTA_DATA_START_OFFSETr)rr<rrZ	attr_datar r r!�unpack_rta_attrs��rBcCs�|dusJd��t|�tks$Jd��d}}t}|t|�kr�t||�}|r�|jdkrVq�t|jtt}||j|7}|jtkr�t|j�}q0|jt	kr0t
�|jd�}|�d�}q0|r�|dur�dSt
�d||�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-z2length of data is smaller than RTATTR_START_OFFSETrzutf-8�z!rta attrs: ifname %s operstate %d)r.r?rBr�
PAD_ALIGNMENTr�IFLA_OPERSTATE�ordr�IFLA_IFNAMErZ
decode_binary�stripr'r(r)rrrr<�attr�padlenZinterface_namer r r!�read_rta_oper_state�s0

��
�

rKcs6t�d�d���fdd�}t|dtgttg|��S)z�Block 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|�vrdS|�dS)NTFr ��iname�carrier�prevCarrier��
existing_nicsrr r!�should_continue_cb�sz5wait_for_nic_attach_event.<locals>.should_continue_cb)r'r(�read_netlink_messages�RTM_NEWLINK�OPER_UP�	OPER_DOWN)r)rQrRr rPr!�wait_for_nic_attach_event�s
	�rWcs2t�d�d��fdd�}t|dtgtg|��S)z�Block 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 rL�rr r!rR�sz5wait_for_nic_detach_event.<locals>.should_continue_cb)r'r(rS�RTM_DELLINKrV)r)rRr rXr!�wait_for_nic_detach_event�s
�rZcsf|dusJd���dus Jd��t��dks4Jd���fdd�}t�d�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.
    Nr5zinterface name is nonerzinterface name cannot be emptycs(|tko|tk}|r$t�d��dSdS)NzMedia switch happened on %s.FT)rVrUr'r()rMrNrOZisVnetSwitchrXr r!rR�s
z=wait_for_media_disconnect_connect.<locals>.should_continue_cbz1Wait for media disconnect and reconnect to happen)r.r'r(rSrTrYrUrV)r)rrRr rXr!�!wait_for_media_disconnect_connect�s

�r[cCsx|durtd��t�}t}t}t|t�}|dur2qt�dt|��||7}t�dt|��d}	t|�}
|	|
k�rf||	d�}t|�tkr�t�d��qft	|�}t|�|j
kr�t�d��qf|j
tdtd@}
|	|
}	t�d	|	�|j|vr�qft
|�}|du�rt�d
|�qf|du�r:|j|k�r:t�d|j|�qf|j|v�rHqf|}|j}||j||�sfdSqf||	d�}qdS)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.)�RuntimeError�bytesrUr:�SELECT_TIMEOUTr'r(r.r/r4rrDrrKrr)r)Z
ifname_filterZ	rtm_typesZ
operstatesZshould_continue_callbackrrNrOZ	recv_datar<ZdatalenZnl_msgZnlheaderrJZinterface_stater r r!rSsh



�

����rS)N)3Zloggingr#r6r"r0�collectionsrZ	cloudinitrZ	getLoggerrr'r%rTrYZRTM_GETLINKZRTM_SETLINKr7r3r^r2Z
IFINFOMSG_FMT�calcsizer/ZIFINFOMSG_SIZEr?rArDrGrEZOPER_UNKNOWNZOPER_NOTPRESENTrVZOPER_LOWERLAYERDOWNZOPER_TESTINGZOPER_DORMANTrUrrrr\rr,r4r:rBrKrWrZr[rSr r r r!�<module>s\


�
% 

© 2025 Cubjrnet7