관리-도구
편집 파일: netlink.cpython-39.opt-1.pyc
a '�Dg�. � @ sB d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ e � e �Z dZ dZ dZdZdZd ZdZd Zd Zd Ze�e�Ze�e�Zee Zd Zd ZdZ dZ d Z dZ dZ dZ!d Z"dZ#dZ$edg d��Z%edddg�Z&edg d��Z'G dd� de(�Z)dd � Z*d d � Z+d-d d �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 @ s e Zd ZdZdS )�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.pyr 5 s r c C s| z4t � t jt jt j�} | �t�� tf� | �d� W n8 t j yl } z d| }t |�|�W Y d}~n d}~0 0 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_socket9 s � r, c C s: 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 */ }; NzGot netlink msg of type %d)�struct�unpack� NLMSGHDR_FMT�MSG_TYPE_OFFSETr' r( r )r Zmsg_lenZmsg_typer r r r r r! �get_netlink_msg_headerO s � r1 c C sN t � | gg g |�\}}}| |vr$dS t�d� | �t�}|du rJt�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 Nz netlink socket ready for readz,Reading from Netlink socket returned no data)�selectr' r( Zrecv�MAX_SIZEr& )r) �timeoutZread_set�_r r r r! �read_netlink_socketh s r6 c C st d }}d}z0t jd| |d�d }t jd| |d d�d }W n t jyR Y dS 0 | |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. r N�H)�offsetr )r- � unpack_fromr& �RTA_DATA_START_OFFSETr )r r8 r r Z attr_datar r r! �unpack_rta_attr s r; c C s� d }}t }|t| �kr�t| |�}|r�|jdkr2q�t|jt t }||j| 7 }|jtkrht|j�}q |jt kr t � |jd�}|� d�}q |r�|du r�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 zutf-8� z!rta attrs: ifname %s operstate %d)�RTATTR_START_OFFSET�lenr; r � PAD_ALIGNMENTr �IFLA_OPERSTATE�ordr � IFLA_IFNAMEr Z decode_binary�stripr' r( r )r r r r8 �attr�padlenZinterface_namer r r! �read_rta_oper_state� s& � rF 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 | � v r 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) rL rM r rK r! �wait_for_nic_attach_event� s �rR 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 rG �r r r! rM � s z5wait_for_nic_detach_event.<locals>.should_continue_cb)r' r( rN � RTM_DELLINKrQ )r) rM r rS r! �wait_for_nic_detach_event� s �rU c s2 � 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. c s( |t ko|tk}|r$t�d� � dS dS )Nz Media switch happened on %s.FT)rQ rP r' r( )rH rI rJ Z isVnetSwitchrS r r! rM � s z=wait_for_media_disconnect_connect.<locals>.should_continue_cbz1Wait for media disconnect and reconnect to happenN)r' r( rN rO rT rP rQ )r) r rM r rS r! �!wait_for_media_disconnect_connect� s �rV c C sx | du rt d��t� }t}t}t| t�}|du r2q 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 |vr�qft | �}|du �rt�d |� qf|du�r:|j|k�r:t�d |j|� qf|j|v�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�bytesrP r6 �SELECT_TIMEOUTr' r( r> � NLMSGHDR_SIZEr1 r r? r rF r r )r) Z ifname_filterZ rtm_typesZ operstatesZshould_continue_callbackr rI rJ Z recv_datar8 ZdatalenZnl_msgZnlheaderrE Zinterface_stater r r! rN sh � ��� �rN )N)3Zloggingr# r2 r" r- � collectionsr Z cloudinitr Z getLoggerr r' r% rO rT Z RTM_GETLINKZ RTM_SETLINKr3 r0 rY r/ Z IFINFOMSG_FMT�calcsizerZ ZIFINFOMSG_SIZEr= r: r? rB r@ Z OPER_UNKNOWNZOPER_NOTPRESENTrQ ZOPER_LOWERLAYERDOWNZ OPER_TESTINGZ OPER_DORMANTrP r r r rW r r, r1 r6 r; rF rR rU rV rN r r r r! �<module> s\ � %