JezK
Edit File: ospfclient.cpython-39.opt-1.pyc
a ��sgq� � @ sB d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m Z dZ e� e �ZdZd ZdZdZdZe� e�ZdZdZd Zd Zd Zd Zd ZdZ dZ dZ edededefedefeddedd� fedede de de di Z dZ d Z!d Z"d Z#d Z$d Z%d!Z&d"Z'd#Z(d$Z)d%Z*e!d&e"d'e#d(d)edd� fe$d*d)edd� fe%d+e&d,e'd-e(d.e)d/e*d0i Z+d Z,d1Z-d2Z.d3Z/d4Z0d5Z1d6Z2d7Z3d8Z4d9Z5d:Z6e,d;e-d<e.d=e/d>e0d?e1d@e2dAe3dBe4dCe5dDe6dEi Z7i Z8e8�9e � e8�9e+� dFdG� e8�:� D �Z;dHdG� e8�:� D �Z<dIdG� e<�:� D �Z=dJdK� Z>dLdM� Z?G dNdO� dOe@�ZAG dPdQ� dQe@�ZBG dRdS� dSe@�ZCd ZDdZEdZFd ZGd ZHd ZId ZJd ZKdZLdZMd ZNd ZOdTdU� ZPd ZQdZRdZSd ZTd ZUd ZVd ZWd ZXdVdW� ZYd ZZdZ[dZ\d Z]d Z^d Z_d Z`d ZadZbdZcdXdY� ZdG dZd[� d[�ZeG d\d]� d]�ZfG d^d_� d_ef�Zgdgd`da�Zhdbdc� Ziddde� Zjekdfk�r>ej� Zle�mel� dS )h� N)�Event�Lock)� ip_addressz>BBHLz>HBB� � z >HBBIILHH� � � � � � � � )ZREGISTER_OPAQUETYPE�BBxx)ZUNREGISTER_OPAQUETYPEr ZREGISTER_EVENTZ SYNC_LSDBZORIGINATE_REQUEST�>II)ZDELETE_REQUESTz>IBBxBL)�MSG_SYNC_REACHABLE� )� MSG_SYNC_ISMr )� MSG_SYNC_NSMr )�MSG_SYNC_ROUTER_IDr � � � � � � � � � � )ZREPLYZbxxx)Z READY_NOTIFYz>BBxxIZLSA_UPDATE_NOTIFY�>IIBxxxZLSA_DELETE_NOTIFY)ZNEW_IFr )ZDEL_IF�>I)Z ISM_CHANGEr )Z NSM_CHANGEz>IIIBxxx)ZREACHABLE_CHANGEz>HH)ZROUTER_ID_CHANGEr! ���������������i����i����i����i����i����� OSPF_API_OK�OSPF_API_NOSUCHINTERFACE�OSPF_API_NOSUCHAREA�OSPF_API_NOSUCHLSA�OSPF_API_ILLEGALLSATYPE�OSPF_API_OPAQUETYPEINUSE� OSPF_API_OPAQUETYPENOTREGISTERED�OSPF_API_NOTREADY�OSPF_API_NOMEMORY�OSPF_API_ERROR�OSPF_API_UNDEFc C s i | ]\}}||d �qS )r � ��.0�k�vr2 r2 �/usr/lib/frr/ospfclient.py� <dictcomp>� � r8 c C s i | ]\}}||d �qS )r r2 r3 r2 r2 r7 r8 � r9 c C s i | ]\}}|t �|��qS r2 )�struct�calcsizer3 r2 r2 r7 r8 � r9 c C s t �| t| ��S �N)�msg_name�get�str)�mtr2 r2 r7 � api_msgname� s rA c C s t �| t| ��S r< )� msg_errnamer> r? )�ecoder2 r2 r7 � api_errname� s rD c @ s e Zd ZdS )�APIErrorN��__name__� __module__� __qualname__r2 r2 r2 r7 rE � s rE c @ s e Zd ZdS )� MsgTypeErrorNrF r2 r2 r2 r7 rJ � s rJ c @ s e Zd ZdS )� SeqNumErrorNrF r2 r2 r2 r7 rK � s rK c C s@ t dtdtdtdtdtdtdtdtd t d t d i }|� | t | ��S ) Nz LSA:ROUTERz LSA:NETWORKz LSA:SUMMARYzLSA:ASBR_SUMMARYzLSA:AS_EXTERNALzLSA:GROUP_MEMBERz LSA:AS_NSSAzLSA:EXTERNAL_ATTRIBUTESzLSA:OPAQUE_LINKzLSA:OPAQUE_AREAz LSA:OPAQUE_AS) �LSA_TYPE_ROUTER�LSA_TYPE_NETWORK�LSA_TYPE_SUMMARY�LSA_TYPE_ASBR_SUMMARY�LSA_TYPE_AS_EXTERNAL�LSA_TYPE_GROUP_MEMBER�LSA_TYPE_AS_NSSA� LSA_TYPE_EXTERNAL_ATTRIBUTES�LSA_TYPE_OPAQUE_LINK�LSA_TYPE_OPAQUE_AREA�LSA_TYPE_OPAQUE_ASr> r? )�lsa_type�namesr2 r2 r7 � lsa_typename� s � rY c C s4 t dtdtdtdtdtdtdtdi}|�| t | ��S ) N�ISM_DEPENDUPON�ISM_DOWN� ISM_LOOPBACK� ISM_WAITING�ISM_POINTTOPOINT� ISM_DROTHER� ISM_BACKUP�ISM_DR) rZ r[ r\ r] r^ r_ r` ra r> r? ��staterX r2 r2 r7 �ism_name� s � rd c C s< t dtdtdtdtdtdtdtdtd t d i }|� | t | ��S ) N�NSM_DEPENDUPON� NSM_DELETED�NSM_DOWN� NSM_ATTEMPT�NSM_INIT� NSM_TWOWAY� NSM_EXSTART� NSM_EXCHANGE� NSM_LOADING�NSM_FULL) re rf rg rh ri rj rk rl rm rn r> r? rb r2 r2 r7 �nsm_name� s � ro c @ s e Zd ZdZdd� Zdd� ZdS )� WithNothingz<An object that does nothing when used with `with` statement.c � s d S r< r2 ��selfr2 r2 r7 � __aenter__ s zWithNothing.__aenter__c � s d S r< r2 )rr �args�kwargsr2 r2 r7 � __aexit__ s zWithNothing.__aexit__N)rG rH rI �__doc__rs rv r2 r2 r2 r7 rp s rp c @ s� e Zd Zdd� Zedd� �Zd-dd�Zd d � Zd d � Ze d d� �Z dd� Z dd� Z ed.dd��Z dd� Zdd� Zd/dd �Zd d � Zed d � �Zed!d"� �Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� ZdS )0� OspfApiClientc C s d� | j�S )NzOspfApiClient({}))�format�serverrq r2 r2 r7 �__str__ s zOspfApiClient.__str__c C s� t � t jt jd�}z�|�t jt jd� |�d| f� t � t jt jd�}z0|�t jt jd� |�d| d f� ||fW W S ty� |�� � Y n0 W n ty� |�� � Y n0 d S )Nr r r ) �socketZAF_INETZ SOCK_STREAMZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRZbind� Exception�close)�port�s1�s2r2 r2 r7 �_get_bound_sockets s z OspfApiClient._get_bound_sockets� localhostNc C s� d| _ d| _d| _d| _d | _ | _| _|| _|dur<|nt� | _ t � | _ d}d}t ||d d�D ]�}z0t �d| ||d � | �|�\| _| _W q�W qd ty� } zF|jtjks�||kr�t �d| ||� � t �d | ||d � W Y d}~qdd}~0 0 qddS ) af A client connection to OSPF Daemon using the OSPF API The client object is not created in a connected state. To connect to the server the `connect` method should be called. If an error is encountered when sending messages to the server an exception will be raised and the connection will be closed. When this happens `connect` may be called again to restore the connection. Args: server: hostname or IP address of server default is "localhost" handlers: dict of message handlers, the key is the API message type, the value is a function. The functions signature is: `handler(msg_type, msg, msg_extra, *params)`, where `msg` is the message data after the API header, `*params` will be the unpacked message values, and msg_extra are any bytes beyond the fixed parameters of the message. Raises: Will raise exceptions for failures with various `socket` modules functions such as `socket.socket`, `socket.setsockopt`, `socket.bind`. r Ni � i�� r z%s: binding to ports %s, %sr z %s: binding port %s error %sz%s: ports %s, %s in use.)�_seq�_s�_as�_ls�_ar�_r�_wrz �dict�handlersr � write_lock�range�logging�debugr� �OSError�errnoZ EADDRINUSE�warning)rr rz r� Z PORTSTARTZPORTENDr �errorr2 r2 r7 �__init__) s* ,zOspfApiClient.__init__c � s� t �d| � t�� }| j�� zNt �d| � |�| j| jdf�I d H t �d| � |� | j�I d H \| _ }W n" t y� | � � I d H � Y n0 t �d| � tj | jd�I d H \| _| _tj | j d�I d H \| _}d| _d S )Nz%s: connect to OSPF APIz$%s: connecting sync socket to serveri/ z!%s: accepting connect from serverz %s: success)Zsockr )r� r� �asyncio�get_event_loopr� ZlistenZ sock_connectr� rz Z sock_acceptr� r} � _close_lockedZopen_connectionr� r� r� r� )rr �loop�_r2 r2 r7 �_connect_lockedW s z OspfApiClient._connect_lockedc � sL | j 4 I d H �$ | �� I d H W d �I d H qH1 I d H s>0 Y d S r< )r� r� rq r2 r2 r7 �connectl s zOspfApiClient.connectc C s | j dkS )z!True if the connection is closed.r )r� rq r2 r2 r7 �closedp s zOspfApiClient.closedc � s� t �d| � | jrP| jr:| j�� | j�� I d H d | _n | j�� d | _d | _| jrl| j�� d | _d | _| j r�| j �� d | _ d| _ d S )Nz %s: closingr ) r� r� r� r� r~ Z wait_closedr� r� r� r� r� rq r2 r2 r7 r� u s" zOspfApiClient._close_lockedc � sL | j 4 I d H �$ | �� I d H W d �I d H qH1 I d H s>0 Y d S r< )r� r� rq r2 r2 r7 r~ � s zOspfApiClient.closer" c � s� z�| � t�I dH }t�t|�\}}}}|dkr<td�|���|dkrRt�d|� n||krjt d�||���|r~| � |�I dH nd}||fW S t j y� t �Y n0 dS )aW Read an OSPF API message from the socket `r` Args: r: socket to read msg from expseq: sequence number to expect or -1 for any. Raises: Will raise exceptions for failures with various `socket` modules, Additionally may raise SeqNumError if unexpected seqnum is received. Nr z'received unexpected OSPF API version {}r" z&_msg_read: got seq: 0x%x on async read� rx {} != {}r9 ) Z readexactly�FMT_APIMSGHDR_SIZEr: �unpack� FMT_APIMSGHDRr} ry r� r� rK r� ZIncompleteReadError�EOFError)�rZexpseq�mhr6 r@ �l�seq�msgr2 r2 r7 � _msg_read� s zOspfApiClient._msg_readc � s t �| jd�I dH S )z�Read a message from the async notify channel. Raises: May raise exceptions for failures with various `socket` modules. r" N)rx r� r� rq r2 r2 r7 �msg_read� s zOspfApiClient.msg_readc � s t �d| t|�| j� t�td|t|�| j�}| j}|d | _z�| j4 I dH �N | j � || � | j � � I dH t � | j|�I dH \}}W d �I dH q�1 I dH s�0 Y |tkr�td�t|�tt����t�tt |�d W S t�y | �� I dH � Y n0 dS )a Send a message to OSPF API and wait for error code reply. Args: mt: the messaage type mp: the message payload Returns: error: an OSPF_API_XXX error code, 0 for OK. Raises: Raises SeqNumError if the synchronous reply is the wrong sequence number; MsgTypeError if the synchronous reply is not MSG_REPLY. Also, may raise exceptions for failures with various `socket` modules, The connection will be closed. z SEND: %s: sending %s seq 0x%xr Nr� r )r� r� rA r� r: �packr� �lenr� r� �writeZdrainrx r� r� � MSG_REPLYrJ ry r� �msg_fmtr} r~ )rr r@ �mpr� r� r2 r2 r7 �msg_send� s" B�zOspfApiClient.msg_send� c � s2 | � ||�I dH }|r.td�t|�t|����dS )a� Send a message to OSPF API and wait for error code reply. Args: mt: the messaage type mp: the message payload Raises: APIError if the server replies with an error. Also may raise exceptions for failures with various `socket` modules, as well as MsgTypeError if the synchronous reply is incorrect. The connection will be closed for these non-API error exceptions. Nz {} error {})r� rE ry rA rD )rr r@ r� rC r2 r2 r7 �msg_send_raises� s z OspfApiClient.msg_send_raisesc � s� |t vrt�d| |� d S t | }t| }t�||d |� �}||d � }|| jvrjt�d| t|�� d S t�d| t|�� | j| |||g|�R � I d H d S )Nz#RECV: %s: unknown async msg type %sz%RECV: %s: no handlers for msg type %sz RECV: %s: calling handler for %s)r� r� r� �msg_sizer: r� r� rA )rr r@ r� �fmtZsz�tup�extrar2 r2 r7 �handle_async_msg� s �z OspfApiClient.handle_async_msgc G s& | sdS d}| D ]}|d|> O }q|S )z.Return a 16 bit mask for each LSA type passed.i�� r r r2 )� lsa_types�mask�tr2 r2 r7 � lsa_type_mask� s zOspfApiClient.lsa_type_maskc C s@ t j|� }t|�}t|r"d�|�nd }tj||| |g|�R � S )z�Return an LSA filter. Return the filter message bytes based on `origin` the `areas` list and the LSAs types in the `lsa_types` list. z{}Ir )rx r� r� �FMT_LSA_FILTERry r: r� )�originZareasr� r� Znarear� r2 r2 r7 � lsa_filter s zOspfApiClient.lsa_filterc � sN t �d| � t�tg g �}| �t|�I dH t �d| � | �t|�I dH dS )zHRegister for all LSA notifications and request an LSDB synchronoization.z SEND: %s: request LSDB eventsNzSEND: %s: request LSDB sync)r� r� rx r� �LSAF_ORIGIN_ANYr� �MSG_REGISTER_EVENT� MSG_SYNC_LSDB)rr r� r2 r2 r7 � req_lsdb_sync s zOspfApiClient.req_lsdb_syncc � s t �d| � | �t�I dH dS )z(Request a dump of all reachable routers.z#SEND: %s: request reachable changesN)r� r� r� r rq r2 r2 r7 �req_reachable_routers s z#OspfApiClient.req_reachable_routersc � s t �d| � | �t�I dH dS )z;Request a dump of the current ISM states of all interfaces.z SEND: %s: request ISM changesN)r� r� r� r rq r2 r2 r7 �req_ism_states" s z OspfApiClient.req_ism_statesc � s t �d| � | �t�I dH dS )�:Request a dump of the current NSM states of all neighbors.z SEND: %s: request NSM changesN)r� r� r� r rq r2 r2 r7 �req_nsm_states' s z OspfApiClient.req_nsm_statesc � s t �d| � | �t�I dH dS )r� z SEND: %s: request router ID syncN)r� r� r� r rq r2 r2 r7 �req_router_id_sync, s z OspfApiClient.req_router_id_sync)r� N)r" )r� )rG rH rI r{ � staticmethodr� r� r� r� �propertyr� r� r~ r� r� r� r� r� r� r� r� r� r� r� r� r2 r2 r2 r7 rx s2 . & rx c s� e Zd ZdZd6� fdd� Zdd� Zdd � Zed d � �Zed d � �Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd7d d �Zd!d"� Zd8d$d%�Zd&d'� Zd9d(d)�Zd*d+� Zd:d,d-�Zd;d.d/�Zd<d0d1�Zd=d2d3�Zd>d4d5�Z � Z S )?�OspfOpaqueClientaP A client connection to OSPF Daemon for manipulating Opaque LSA data. The client object is not created in a connected state. To connect to the server the `connect` method should be called. If an error is encountered when sending messages to the server an exception will be raised and the connection will be closed. When this happens `connect` may be called again to restore the connection. Args: server: hostname or IP address of server default is "localhost" wait_ready: if True then wait for OSPF to signal ready, in newer versions FRR ospfd is always ready so this overhead can be skipped. default is False. Raises: Will raise exceptions for failures with various `socket` modules functions such as `socket.socket`, `socket.setsockopt`, `socket.bind`. r� Fc s� t | jt| jt| jt| jt| jt| j t | j t | j i}|rB| j|t< t� �||� || _|r`t� nt� | _ti ti ti i| _td�| _d | _ i | _ d | _ i | _ t � | _!d | _"i | _#i | _$d | _%i | _&d | _'d S )Nr )(�MSG_LSA_UPDATE_NOTIFY�_lsa_change_msg�MSG_LSA_DELETE_NOTIFY� MSG_NEW_IF�_if_msg� MSG_DEL_IF�MSG_ISM_CHANGE�_if_change_msg�MSG_NSM_CHANGE�_nbr_change_msg�MSG_REACHABLE_CHANGE�_reachable_msg�MSG_ROUTER_ID_CHANGE�_router_id_msg� _ready_msg�MSG_READY_NOTIFY�superr� � wait_readyr rp � ready_lockrT rU rV � ready_cond�ip� router_id�router_id_change_cbZ lsid_seq_num� lsa_change_cb�opaque_change_cb�set�reachable_routers�reachable_change_cb�if_area� ism_states� ism_change_cb� nsm_states� nsm_change_cb)rr rz r� r� �� __class__r2 r7 r� F s< � � zOspfOpaqueClient.__init__c � s� | j 4 I d H �t | j| �|�}t�dt|�|� t}t�t | ||�}| � ||�I d H | j snd| j| |< W d �I d H q�1 I d H s�0 Y d S )Nzregister %s opaque-type %sT) r� r� r> r� r� rY �MSG_REGISTER_OPAQUETYPEr: r� r� r� r� )rr rW �otype�condr@ r� r2 r2 r7 �_register_opaque_datao s z&OspfOpaqueClient._register_opaque_datac � s| zXt �d� | �� I d H \}}|tv r:| �||�I d H q t|�}t �d|t|�� q W n tyv t � d� Y dS 0 d S )Nz entering async msg handling loopz#ignoring unexpected msg: %s len: %s�3Got EOF from OSPF API server on async notify socketr ) r� r� r� � amsg_infor� rA r� r� r� �info)rr r@ r� Zmtsr2 r2 r7 �_handle_msg_loop� s � z!OspfOpaqueClient._handle_msg_loopc C s( |d> |B }dd| |dddt t|� fS )N� r )�FMT_LSA_HEADER_SIZEr� )rW r� �oidr� Zlsidr2 r2 r7 � _opaque_args� s z OspfOpaqueClient._opaque_argsc C s* t jtgt�| |||��R � }||7 }|S r< )r: r� �FMT_LSA_HEADERr� r� )rW r� r� r� �lsar2 r2 r7 �_make_opaque_lsa� s ��z!OspfOpaqueClient._make_opaque_lsac � s� |t krd�t|��}n |tkr0d�t|��}nd}t�d| t|�||� | j4 I d H �4 | j| � |�}d| j| |< W d �I d H q�1 I d H s�0 Y |du r�t� dt|�|� n|r�|D ] } | � � q�d S )Nz ifaddr {}zarea {}r z-RECV: %s ready notify for %s opaque-type %s%sTz.RECV: dup ready received for %s opaque-type %s) rT ry r� rU r� r� rY r� r� r> r� r� ) rr r@ r� r� rW r� �addr�er� �evtr2 r2 r7 r� � s0 � 8�zOspfOpaqueClient._ready_msgc � s@ |t kr|\}}n|d d }}t�dt|�t|�t|�� d S )Nr z RECV: %s ifaddr %s areaid %s)r� r� r� rA r� )rr r@ r� r� rt �ifaddr�aidr2 r2 r7 r� � s �zOspfOpaqueClient._if_msgc � sV t |�}t |�}t�dt|�||t|�� || j|< || j|< | jrR| �|||� d S )Nz%RECV: %s ifaddr %s areaid %s state %s)r� r� r� rA rd r� r� r� )rr r@ r� r� r r rc r2 r2 r7 r� � s � z OspfOpaqueClient._if_change_msgc � st t |�}t |�}t |�}t�dt|�|||t|�� || jvrHi | j|< || j| ||f< | jrp| �||||� d S )Nz3RECV: %s ifaddr %s nbraddr %s router_id %s state %s)r� r� r� rA ro r� r� )rr r@ r� r� r Znbraddrr� rc r2 r2 r7 r� � s � z OspfOpaqueClient._nbr_change_msgc � s� |\}} } } } } } } | d? d@ }|t kr.d}nd}t�d|t| �t|�| | |� | |f}t| t }||d � }|| jv r�| j| ||||||� | jr�| �||||||� d S )Nr� � �update�deletez=RECV: LSA %s msg for LSA %s in area %s seq 0x%x len %s age %s)r� r� r� r� r� r� r� r� )rr r@ r� r� r r Zis_selfZ ls_headerZlsa_ager� rW Zls_idZls_seqZls_lenr� �ts�idxZ pre_lsa_sizer� r2 r2 r7 r� � s< � � z OspfOpaqueClient._lsa_change_msgc � s� t �d�|| �|�}dd� |D �}t�dt|�|d |� ||d � � | jt|d |� �O _| jt||d � �8 _t�dt|�| j� | jr�t�dt|�� | �|d |� ||d � �I d H d S )Nz>{}Ic S s g | ] }t |��qS r2 )r� )r4 �xr2 r2 r7 � <listcomp> r9 z3OspfOpaqueClient._reachable_msg.<locals>.<listcomp>z RECV: %s added %s removed %szRECV: %s new set %s�RECV: %s calling callback) r: r� ry r� r� rA r� r� r� )rr r@ r� r� ZnaddZnremoveZ router_idsr2 r2 r7 r� s �z OspfOpaqueClient._reachable_msgc � sr t |�}t�dt|�|� | j}||kr,d S || _t�dt|�||� | jrnt�dt|�� | �||�I d H d S )NzRECV: %s router ID %sz,RECV: %s new router ID %s older router ID %sr )r� r� r� rA r� r� )rr r@ r� r� r� Z old_router_idr2 r2 r7 r� ( s �z OspfOpaqueClient._router_id_msgc � s� |t krt|�d }}n |tkr0dt|� }}nd\}}t}tjt| ||gt�||||��R � } | |7 } | � || �I dH dS )a� Add an instance of opaque data. Add an instance of opaque data. This call will register for the given LSA and opaque type if not already done. Args: addr: depends on lsa_type, LINK => ifaddr, AREA => area ID, AS => ignored lsa_type: LSA_TYPE_OPAQUE_{LINK,AREA,AS} otype: (octet) opaque type oid: (3 octets) ID of this opaque data data: the opaque data Raises: See `msg_send_raises` r )r r N) rT �intrU �MSG_ORIGINATE_REQUESTr: r� r� r� r� r� ) rr r� rW r� r� �datar r r@ r� r2 r2 r7 �add_opaque_data; s ��z OspfOpaqueClient.add_opaque_datar c � s6 t }t�t| t|�||||�}| �||�I dH dS )a7 Delete an instance of opaque data. Delete an instance of opaque data. This call will register for the given LSA and opaque type if not already done. Args: addr: depends on lsa_type, LINK => ifaddr, AREA => area ID, AS => ignored lsa_type: LSA_TYPE_OPAQUE_{LINK,AREA,AS} otype: (octet) opaque type. oid: (3 octets) ID of this opaque data flags: (octet) optional flags (e.g., OSPF_API_DEL_ZERO_LEN_LSA, defaults to no flags) Raises: See `msg_send_raises` N)�MSG_DELETE_REQUESTr: r� r� r r� )rr r� rW r� r� �flagsr@ r� r2 r2 r7 �delete_opaque_data^ s z#OspfOpaqueClient.delete_opaque_datac � sV | j 4 I dH �. | j�|i ��|�duW d �I dH S 1 I dH sH0 Y dS )aX Determine if an (lsa_type, otype) tuple has been registered with FRR This determines if the type has been registered, but not necessarily if it is ready, if that is required use the `wait_opaque_ready` metheod. Args: lsa_type: LSA_TYPE_OPAQUE_{LINK,AREA,AS} otype: (octet) opaque type. N)r� r� r> )rr rW r� r2 r2 r7 � is_registereds s z OspfOpaqueClient.is_registeredNc � sV |r|| j ||f< n,||f| j v r@t�dt|�|� | j ||f= | �||�I dH dS )a� Register intent to advertise opaque data. The application should wait for the async notificaiton that the server is ready to advertise the given opaque data type. The API currently only allows a single "owner" of each unique (lsa_type,otype). To wait call `wait_opaque_ready` Args: lsa_type: LSA_TYPE_OPAQUE_{LINK,AREA,AS} otype: (octet) opaque type. callback: if given, callback will be called when changes are received for LSA of the given (lsa_type, otype). The callbacks signature is: `callback(msg_type, ifaddr, area_id, lsa_header, data, lsa)` Args: msg_type: MSG_LSA_UPDATE_NOTIFY or MSG_LSA_DELETE_NOTIFY ifaddr: integer identifying an interface (by IP address) area_id: integer identifying an area lsa_header: the LSA header as an unpacked tuple (fmt: ">HBBIILHH") data: the opaque data that follows the LSA header lsa: the octets of the full lsa Raises: See `msg_send_raises` z=OSPFCLIENT: register: removing callback for %s opaque-type %sN)r� r� r� rY r� �rr rW r� �callbackr2 r2 r7 �register_opaque_data� s � z%OspfOpaqueClient.register_opaque_datac � s� | j 4 I d H �| | j| �|�}|du r>W d �I d H d S t�dt|�|� |sfg }| j| |< t� }|�|� W d �I d H q�1 I d H s�0 Y |�� I d H t�dt|�|� d S )NTz#waiting for ready %s opaque-type %szREADY for %s opaque-type %s) r� r� r> r� r� rY r �append�wait)rr rW r� r� r� r2 r2 r7 �wait_opaque_ready� s �4z"OspfOpaqueClient.wait_opaque_readyc � s* | � |||�I dH | �||�I dH dS )a� Register intent to advertise opaque data and wait for ready. The API currently only allows a single "owner" of each unique (lsa_type,otype). Args: lsa_type: LSA_TYPE_OPAQUE_{LINK,AREA,AS} otype: (octet) opaque type. callback: if given, callback will be called when changes are received for LSA of the given (lsa_type, otype). The callbacks signature is: `callback(msg_type, ifaddr, area_id, lsa_header, data, lsa)` Args: msg_type: MSG_LSA_UPDATE_NOTIFY or MSG_LSA_DELETE_NOTIFY ifaddr: integer identifying an interface (by IP address) area_id: integer identifying an area lsa_header: the LSA header as an unpacked tuple (fmt: ">HBBIILHH") data: the opaque data that follows the LSA header lsa: the octets of the full lsa Raises: See `msg_send_raises` N)r r r r2 r2 r7 �register_opaque_data_wait� s z*OspfOpaqueClient.register_opaque_data_waitc � sF ||f| j v r| j ||f= t}t�t| ||�}| �||�I dH dS )aB Unregister intent to advertise opaque data. This will also cause the server to flush/delete all opaque data of the given (lsa_type,otype). Args: lsa_type: LSA_TYPE_OPAQUE_{LINK,AREA,AS} otype: (octet) opaque type. Raises: See `msg_send_raises` N)r� �MSG_UNREGISTER_OPAQUETYPEr: r� r� r� )rr rW r� r@ r� r2 r2 r7 �unregister_opaque_data� s z'OspfOpaqueClient.unregister_opaque_datac � s || _ | �� I dH dS )a� Monitor changes to LSAs. Args: callback: if given, callback will be called when changes are received for any LSA. The callback signature is: `callback(msg_type, ifaddr, area_id, lsa_header, extra, lsa)` Args: msg_type: MSG_LSA_UPDATE_NOTIFY or MSG_LSA_DELETE_NOTIFY ifaddr: integer identifying an interface (by IP address) area_id: integer identifying an area lsa_header: the LSA header as an unpacked tuple (fmt: ">HBBIILHH") extra: the octets that follow the LSA header lsa: the octets of the full lsa N)r� r� �rr r r2 r2 r7 � monitor_lsa� s z OspfOpaqueClient.monitor_lsac � s || _ | �� I dH dS )a* Monitor the set of reachable routers. The property `reachable_routers` contains the set() of reachable router IDs as integers. This set is updated prior to calling the `callback` Args: callback: callback will be called when the set of reachable routers changes. The callback signature is: `callback(added, removed)` Args: added: list of integer router IDs being added removed: list of integer router IDs being removed N)r� r� r r2 r2 r7 �monitor_reachable s z"OspfOpaqueClient.monitor_reachablec � s || _ | �� I dH dS )a� Monitor the state of OSPF enabled interfaces. Args: callback: callback will be called when an interface changes state. The callback signature is: `callback(ifaddr, area_id, state)` Args: ifaddr: integer identifying an interface (by IP address) area_id: integer identifying an area state: ISM_* N)r� r� r r2 r2 r7 � monitor_ism s z OspfOpaqueClient.monitor_ismc � s || _ | �� I dH dS )a Monitor the state of OSPF neighbors. Args: callback: callback will be called when a neighbor changes state. The callback signature is: `callback(ifaddr, nbr_addr, router_id, state)` Args: ifaddr: integer identifying an interface (by IP address) nbr_addr: integer identifying neighbor by IP address router_id: integer identifying neighbor router ID state: NSM_* N)r� r� r r2 r2 r7 � monitor_nsm( s z OspfOpaqueClient.monitor_nsmc � s || _ | �� I dH dS )a� Monitor the OSPF router ID. The property `router_id` contains the OSPF urouter ID. This value is updated prior to calling the `callback` Args: callback: callback will be called when the router ID changes. The callback signature is: `callback(new_router_id, old_router_id)` Args: new_router_id: the new router ID old_router_id: the old router ID N)r� r� r r2 r2 r7 �monitor_router_id: s z"OspfOpaqueClient.monitor_router_id)r� F)r )N)N)N)N)N)N)N) rG rH rI rw r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r � __classcell__r2 r2 r� r7 r� 2 s6 ) (# ) r� c c s4 | r| D ] }|V qnt d�}|s$q0|�� V qdS )z"Get next action from list or STDINr N)�input�strip)Z action_list�actionr2 r2 r7 � next_actionQ s r% c � s* t | j�}|�� I d H �z�tjd dkr:t�|�� � nt�� �|�� � |� � I d H |� � I d H |� � I d H |� � I d H t | j�D �]�}|�d�}|�d�}|�� dkr�t|�d��}t�d|� t�|�I d H t�d|� q�t|�d��}|dk�rtd �}n8|�d�}ztt|��}W n t�yD t|�}Y n0 ||t|�d��t|�d��g} |�| d | d �I d H �s�|�| d | d �I d H |�� d k�r6zt�|�d��} W n t �y� d } Y n0 t�d t | �� t | �d } | �r| dd| 7 } t�dt | �� |j g | �| �R � I d H q�d } t |�dk�rtzt|�d��} W n t �yr d } Y n0 |j g | �| �R � I d H q�| j�r�| j �r�W d S W n8 t!�y� } z tj"d| dd� W Y d } ~ d S d } ~ 0 0 z"t�d� t�d�I d H �q�W n t#�y$ t�d� Y d S 0 d S )Nr r �,Fr zwaiting %s secondszwait complete: %s secondsr r r �addr9 zopaque data is %s octetsr � zopaque padding to %s octetsz async_main: unexpected error: %sT��exc_infozSleeping forever�x r� )$r� rz r� �sys� version_infor� Z create_taskr� r� r� r� r� r� r% �actions�split�pop�casefoldr r� r� �sleepr� � ValueErrorr r �bytes�fromhex� IndexErrorr� r r �exitr} r� r� )rt �cr$ r� ZwhatZstimeZltyper� ZavalZoargs�b�mod�fr� r2 r2 r7 � async_main^ sv r< c G sL t �| �}|jdddd� |jdddd� |jd d d d� |jd d ddd� |jdddd� |�� } | jrptjntj}tj|d� | j �d� t� d� d}zPt j d dkr�t�t| ��}n.t�� }z|�t| ��}W |�� n |�� 0 W nR t�y t� d� Y n@ t�y< } z tj d|dd � W Y d }~nd }~0 0 t� d � |S ) Nz--logtagZCLIENTz tag to identify log messages)�default�helpz--exit� store_truezExit after commands)r$ r> z--serverr� zOSPF API serverz-vz --verbosez be verboser. �*z?WAIT,SEC|(ADD|DEL),LSATYPE,[ADDR,],OTYPE,OID,[HEXDATA|DEL_FLAG])�nargsr> z3%(asctime)s %(levelname)s: {}: %(name)s %(message)s)�levelry zospfclient: startingr r r z+Exiting, received KeyboardInterrupt in mainz Exiting, unexpected exception %sTr) zospfclient: clean exit)�argparse�ArgumentParser� add_argument� parse_args�verboser� �DEBUG�INFOZ basicConfigry Zlogtagr� r, r- r� �runr< r� Zrun_until_completer~ �KeyboardInterruptr} )rt ZaprB �statusr� r� r2 r2 r7 �main� sB ��� & rM �__main__)N)nrC r� r� r� r| r: r, r r � ipaddressr r� r� r; r� r� ZLSAF_ORIGIN_NON_SELFZLSAF_ORIGIN_SELFr� r� r� r� r r� r� r r r r r r Z smsg_infoZOSPF_API_DEL_ZERO_LEN_LSAr� r� r� r� r� r� r� r� r� r� r� r' r( r) r* r+ r, r- r. r/ r0 r1 rB Zmsg_infor �itemsr= r� r� rA rD r} rE rJ rK ZLSA_TYPE_UNKNOWNrL rM rN rO rP rQ rR rS rT rU rV rY rZ r[ r\ r] r^ r_ r` ra rd re rf rg rh ri rj rk rl rm rn ro rp rx r� r% r< rM rG � exit_statusr7 r2 r2 r2 r7 �<module> s �� � # # L,