Your IP : 216.73.217.13
U
�Ad/ � @ s^ d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z e�
e �Z dZ
dZdZdZdZd Zd
Zd Zd Zd
ZdZdZdZdZe�e�Z e�e�Z e e Z dZ dZ dZ!dZ"d Z#dZ$dZ%dZ&dZ'dZ(dZ)eddddg�Z*edddg�Z+eddd d d d g�Z,G d d!� d!e-�Z.d"d#� Z/d$d%� Z0d4d&d'�Z1d(d)� Z2d*d+� Z3d,d-� Z4d.d/� Z5d0d1� Z6d2d3� Z7dS )5� N)�
namedtuple)�log)�util� � � � � � � i�� � �< ZIHHIIZBHiII� � � �RTAAttr�length�rta_type�data�InterfaceOperstate�ifname� operstate�
NetlinkHeader�type�flags�seq�pidc @ s e Zd ZdZdS )�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.pyr 9 s r c
C sz z4t � t jt jt j�} | �t�� tf� | �d� W n6 t j k
rj } zd| }t
|�|�W 5 d}~X Y nX t � 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_ROUTEZbind�os�getpid� RTMGRP_LINKZ setblocking�errorr �LOG�debug)�netlink_socket�e�msgr" r" r# �create_bound_netlink_socket= s �
r. c C s^ | dk st d��t| �tks$t d��t�t| dt� �\}}}}}t�d|� 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 */
};
N� data is nonez+data is smaller than netlink message headerzGot netlink msg of type %d)
�AssertionError�len�
NLMSGHDR_SIZE�structZunpack� NLMSGHDR_FMT�MSG_TYPE_OFFSETr) r* r )r Zmsg_lenZmsg_typer r r r" r" r# �get_netlink_msg_headerS s
��
� r6 c C s^ | dk st d��t�| gg g |�\}}}| |kr4dS t�d� | �t�}|dkrZt�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 nonez netlink socket ready for readz,Reading from Netlink socket returned no data)r0 �selectr) r* Zrecv�MAX_SIZEr( )r+ ZtimeoutZread_set�_r r" r" r# �read_netlink_socketl s
r; c C s� | dk st d��t|�tks$t d��|tks4t d��d }}d}z0tjd| |d�d }tjd| |d d�d }W n tjk
r� Y 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.
Nr/ zoffset is not integerz'rta offset is less than expected lengthr �H)�offsetr ) r0 r �int�RTATTR_START_OFFSETr3 Z 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 d��t| �tks$t d��d }}t}|t| �kr�t| |�}|r�|jdkrVq�t|jt t }||j| 7 }|jtkr�t|j �}q0|jt
kr0t � |j d�}|�
d�}q0|r�|dkr�dS t�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_OFFSETr zutf-8� z!rta attrs: ifname %s operstate %d)r0 r1 r? rA r �
PAD_ALIGNMENTr �IFLA_OPERSTATE�ordr � IFLA_IFNAMEr Z
decode_binary�stripr) r* r )r r r r= �attr�padlenZinterface_namer" r" r# �read_rta_oper_state� s0
��
�
rJ c s6 t �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.Nc s | � kr dS | �dS )NTFr" ��iname�carrier� prevCarrier��
existing_nicsr r" r# �should_continue_cb� s z5wait_for_nic_attach_event.<locals>.should_continue_cb)r) r* �read_netlink_messages� RTM_NEWLINK�OPER_UP� OPER_DOWN)r+ rP rQ r" rO r# �wait_for_nic_attach_event� s
�rV c s2 t �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.Nc s | � dS )NFr" rK �r r" r# rQ � s z5wait_for_nic_detach_event.<locals>.should_continue_cb)r) r* rR � RTM_DELLINKrU )r+ rQ r" rW r# �wait_for_nic_detach_event� s
�rY c sf | dk st d��� dk s t d��t� �dks4t d��� 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.
Nr7 zinterface name is noner z interface name cannot be emptyc s( |t ko|tk}|r$t�d� � dS dS )Nz Media switch happened on %s.FT)rU rT r) r* )rL rM rN Z isVnetSwitchrW r" r# rQ s
z=wait_for_media_disconnect_connect.<locals>.should_continue_cbz1Wait for media disconnect and reconnect to happen) r0 r1 r) r* rR rS rX rT rU )r+ r rQ r" rW r# �!wait_for_media_disconnect_connect� s
�rZ c C sx | dkrt d��t� }t}t}t| t�}|dkr2q t�dt|�� ||7 }t�dt|�� d} t|�}
| |
k �rf|| d� } t| �tk r�t�d� �qft | �} t| �| j
k r�t�d� �qf| j
t d t d @ }
| |
} t�d | � | j |kr�qft
| �}|dk�rt�d
|� qf|dk �r:|j|k�r:t�d |j|� qf|j|k�rHqf|}|j}||j||�sfdS qf|| d� }q 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.)� RuntimeError�bytesrT r; �SELECT_TIMEOUTr) r* r1 r2 r6 r rC r rJ r r )r+ Z
ifname_filterZ rtm_typesZ
operstatesZshould_continue_callbackr rM rN Z recv_datar= ZdatalenZnl_msgZnlheaderrI Zinterface_stater" r" r# rR sl
�
��� �rR )N)8r% r8 r$ r3 � collectionsr Z cloudinitr Zloggingr Z getLoggerr r) r' Z
NLMSG_NOOPZ NLMSG_ERRORZ
NLMSG_DONErS rX Z RTM_GETLINKZ RTM_SETLINKr9 ZRTA_DATA_OFFSETr5 r] r4 Z
IFINFOMSG_FMTZcalcsizer2 ZIFINFOMSG_SIZEr? r@ rC rF rD Z OPER_UNKNOWNZOPER_NOTPRESENTrU ZOPER_LOWERLAYERDOWNZ OPER_TESTINGZ OPER_DORMANTrT r r r r[ r r. r6 r; rA rJ rV rY rZ rR r" r" r" r# �<module> sf
�
%