HEX

File: //lib/python3.6/site-packages/cloudinit/sources/helpers/__pycache__/netlink.cpython-36.pyc
3

\�me�.�@sZddlZddlZddlZddlZddlZddlmZddlmZej	e
�ZdZdZ
dZdZdZdZd	Zd
ZdZdZdZd
ZdZdZeje�Zeje�ZeeZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(eddddg�Z)edddg�Z*eddddddg�Z+Gdd �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����RTAAttr�length�rta_type�data�InterfaceOperstate�ifname�	operstate�
NetlinkHeader�type�flags�seq�pidc@seZdZdZdS)�NetlinkCreateSocketErrorz5Raised if netlink socket fails during create or bind.N)�__name__�
__module__�__qualname__�__doc__�r!r!�/usr/lib/python3.6/netlink.pyr9srcCszy4tjtjtjtj�}|jtj�tf�|jd�Wn6tj	k
rj}zd|}t
|�|�WYdd}~XnXtjd�|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_socket=s	
r-cCs^|dk	std��t|�tks$td��tjt|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 */
    };
    Nzdata is nonez+data is smaller than netlink message headerzGot netlink msg of type %d)
�AssertionError�len�
NLMSGHDR_SIZE�struct�unpack�NLMSGHDR_FMT�MSG_TYPE_OFFSETr(r)r)rZmsg_lenZmsg_typerrrr!r!r"�get_netlink_msg_headerSsr5cCs^|dk	std��tj|ggg|�\}}}||kr4dStjd�|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 = <MAX_SIZE>) 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)Zrecv�MAX_SIZEr')r*ZtimeoutZread_set�_rr!r!r"�read_netlink_socketls



r9cCs�|dk	std��t|t�s"td��|tks2td��d}}d}y0tjd||d�d}tjd||dd�d}Wntjk
r�dSX||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.
    Nzdata is nonezoffset is not integerz'rta offset is less than expected lengthr�H)�offsetr)	r.�
isinstance�int�RTATTR_START_OFFSETr1�unpack_fromr'�RTA_DATA_START_OFFSETr)rr;rrZ	attr_datar!r!r"�unpack_rta_attr�s
rAcCs�|dk	std��t|�tks$td��d}}t}x�|t|�kr�t||�}|sX|jdkrZPt|jtt}||j|7}|jtkr�t|j	�}q2|jt
kr2tj|j	d�}|j
d�}q2W|s�|dkr�dStjd||�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.
    Nzdata is nonez2length of data is smaller than RTATTR_START_OFFSETrzutf-8�z!rta attrs: ifname %s operstate %d)r.r/r>rAr�
PAD_ALIGNMENTr�IFLA_OPERSTATE�ordr�IFLA_IFNAMErZ
decode_binary�stripr(r)r)rrrr;�attr�padlenZinterface_namer!r!r"�read_rta_oper_state�s*



rJcs6tjd�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|�krdS|�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*rNrOr!)rNrr"�wait_for_nic_attach_event�s
	rTcs2tjd�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!)rKrLrM)rr!r"rO�sz5wait_for_nic_detach_event.<locals>.should_continue_cb)r(r)rP�RTM_DELLINKrS)r*rOr!)rr"�wait_for_nic_detach_event�s
rVcsf|dk	std���dk	s td��t��dks4td���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 nonerzinterface name cannot be emptycs(|tko|tk}|r$tjd��dSdS)NzMedia switch happened on %s.FT)rSrRr(r))rKrLrMZisVnetSwitch)rr!r"rOs
z=wait_for_media_disconnect_connect.<locals>.should_continue_cbz1Wait for media disconnect and reconnect to happen)	r.r/r(r)rPrQrUrRrS)r*rrOr!)rr"�!wait_for_media_disconnect_connect�s

rWcCs�|dkrtd��t�}t}t}�xZt|t�}|dkr6q"tjdt|��||7}tjdt|��d}	t|�}
�x�|	|
k�rj||	d�}t|�tkr�tjd�Pt	|�}t|�|j
kr�tjd�P|j
tdtd@}
|	|
}	tjd	|	�|j|kr�qnt
|�}|dk�rtjd
|�qn|dk	�r>|j|k�r>tjd|j|�qn|j|k�rLqn|}|j}||j||�sndSqnW||	d�}q"WdS)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�bytesrRr9�SELECT_TIMEOUTr(r)r/r0r5rrCrrJrr)r*Z
ifname_filterZ	rtm_typesZ
operstatesZshould_continue_callbackrrLrMZ	recv_datar;ZdatalenZnl_msgZnlheaderrIZinterface_stater!r!r"rPs`





rP)N)7Zloggingr$r6r#r1�collectionsrZ	cloudinitrZ	getLoggerrr(r&Z
NLMSG_NOOPZNLMSG_ERRORZ
NLMSG_DONErQrUZRTM_GETLINKZRTM_SETLINKr7ZRTA_DATA_OFFSETr4rZr3Z
IFINFOMSG_FMT�calcsizer0ZIFINFOMSG_SIZEr>r@rCrFrDZOPER_UNKNOWNZOPER_NOTPRESENTrSZOPER_LOWERLAYERDOWNZOPER_TESTINGZOPER_DORMANTrRrrrrXrr-r5r9rArJrTrVrWrPr!r!r!r"�<module>sb



%