/
usr
/
lib64
/
python3.9
/
site-packages
/
ntp
/
__pycache__
/
Upload File
HOME
a \��dDa � @ s� d dl mZmZ d dlZd dlZd dlZzd dlZd dlZej Z W nJ e y� Z z2ej �d� ej �de � e�d� W Y dZ [ n dZ [ 0 0 dZdZdd � ZG d d � d �ZG d d � d �Zddd�ZdS )� )�print_function�divisionNz1AgentX: can't find Python AgentX Packet library. z%s � � �< c C s t tu r| �� S t| �S d S �N)�str�bytes�next)� generator� r �0/usr/lib64/python3.9/site-packages/ntp/agentx.py�gen_next s r c @ sh e Zd Zddd�Zdd� Zdd � Zd d � Zd d � Zddd�Zddd�Z d dd�Z d dd�Z d dd�Z dS ) � MIBControlNr r c C sH i | _ |d ur|| _ d| _g | _g | _g | _|| _|| _|| _|| _d S �NF) �oidTree�inSetP� setVarbinds� setHandlers� setUndoData�mibRoot� rangeSubid� upperBound�context)�selfr r r r Z mibContextr r r �__init__ s zMIBControl.__init__c C s | j S r )r �r r r r � mib_rootOID/ s zMIBControl.mib_rootOIDc C s | j S r )r r r r r �mib_rangeSubid2 s zMIBControl.mib_rangeSubidc C s | j S r )r r r r r �mib_upperBound5 s zMIBControl.mib_upperBoundc C s | j S r )r r r r r � mib_context8 s zMIBControl.mib_contextc C s� t |tj�rt|j�}| j}|}tj�|d�\}}|d }||� � vrVd d d d�||< |s�||| d< ||| d< |d ur�||| d< d S || d d u r�i || d< || d }q d S )Nr r )�reader�writer�subidsr! r" r# ) � isinstance�ax�OID�tupler# r �ntp�utilZ slicedata�keys)r �oidr! r" ZdynamicZ currentLevelZ remainingOIDZnoder r r �addNode; s& � zMIBControl.addNodeFc C s� t | j| j�}zVt|�\}}}|r,||k}n |j|jk}|rb|d urb|rV||||fW S |||fW S W q ty� |r|Y dS Y dS Y q0 qd S )N)NNNN)NNN)� walkMIBTreer r r r# � StopIteration) r ZnextP� searchoid�returnGenerator�genr+ r! r" Zoidhitr r r � getOID_coreS s zMIBControl.getOID_corec C s | � d||�S )zGet the requested OIDF�r2 �r r/ r0 r r r �getOIDl s zMIBControl.getOIDc C s | � d||�S )z Get the next lexicographical OIDTr3 r4 r r r � getNextOIDp s zMIBControl.getNextOIDc C s, g }t | j| j�}z�t|�\}}}|du r0W qnj|j|jjkrb|jjr\|�|||f� W q�q�W qn8||jkr�|j� � s�||jkr�g W S |�|||f� W q�W q t y� g Y S 0 q|r�|S zLt|�\}}}|du r�W q�n,|j� � s�||jkr�W �q(n|�|||f� W q� t �y$ Y �q(Y q�0 q�|S )z9Get a list of every (optionally the first) OID in a rangeN) r- r r r r# �startZinclude�append�endZisNullr. )r ZoidrangeZ firstOnly�oidsr1 r+ r! r" r r r �getOIDsInRanget s< zMIBControl.getOIDsInRange)Nr r NN)NNN)F)F)F)F) �__name__� __module__� __qualname__r r r r r r, r2 r5 r6 r; r r r r r s � r c @ s� e Zd Zdeddfdd�Zdd� Zdd � Zd d � Zd-d d�Zdd� Z dd� Z edfdd�Z dd� Z d.dd�Z dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� ZdS )/� PacketControlg����MbP?Ni' c s� � �fdd�| _ || _|| _i | _d | _|| _d| _g | _|| _d | _ d| _ d | _ d| _ t j| jt j| jt j| jt j| jt j| jt j| jt j| jt j | j i| _ d S )Nc s t j��| � |�S r )r( r) Zdolog)ZtxtZdbg��debug�logfpr r �<lambda>� � z(PacketControl.__init__.<locals>.<lambda>rD r F) �log�socket�spinGap� packetLog� loopCallback�database� receivedData�receivedPackets�timeout� sessionID�highestTransactionID� lastReception�stillConnectedr% ZPDU_GET� handle_GetPDUZ PDU_GET_NEXT�handle_GetNextPDUZ PDU_GET_BULK�handle_GetBulkPDUZ PDU_TEST_SET�handle_TestSetPDUZPDU_COMMIT_SET�handle_CommitSetPDUZ PDU_UNDO_SET�handle_UndoSetPDUZPDU_CLEANUP_SET�handle_CleanupSetPDU� PDU_RESPONSE�handle_ResponsePDU� pduHandlers)r ZsockZdbaserG rM rB rA r r@ r r � s, �zPacketControl.__init__c C sR | j durdS |rD| j rL| �� | jd ur4| �| � t�| j� qn| �� | j S )NTF)rQ �_dolooprI �time�sleeprG )r Z runforeverr r r �mainloop� s zPacketControl.mainloopc C s� | � � | jr�| j�d�}|j| jkrb| �d| d� t�d|j|j|jdtj d�}| � |d� q|j }|| j v r�| j | |� q| �d| d� q| � � | jd ur�t�� }|| j tkr�| �� d S )Nr z-Received packet with incorrect session ID: %s� TFz(Dropping packet type %i, not implemented� )� packetEaterrL �poprN rE r% � ResponsePDU� transactionID�packetIDZRSPERR_NOT_OPEN� sendPacketZpduTyper[ �checkResponsesrP r] �pingTime�sendPing)r �packet�respZptype� currentTimer r r r\ � s2 � � � zPacketControl._doloopc C s� | � dd� t�dddd| jdd�}| �|d� | �|d�}|j| _t�d| jd d | jd | j� � | j� � | j� � | j� � � }| �|d� | �|� d| _ d S ) NzInitializing new session...r` T� r r zNTPsec SNMP subagentFr )rE r% ZOpenPDUrM rg �waitForResponserN Z RegisterPDUrJ r r r r rQ )r ZopenpktZresponse�registerr r r �initNewSession� s � � z PacketControl.initNewSessionFc C s| | � � | jrj| j�d�}|jtjkr(q|j|jko>|j|jk}|sT|oR|j|jk}|r| � dd� |S qt � | j � q dS )zBWait for a response to a specific packet, dropping everything elser z Received waited for response� N) rb rL rc � __class__r% rd re rf rN rE r] r^ rG )r ZopktZ ignoreSIDrk Zhaveitr r r ro � s � z PacketControl.waitForResponsec C sR t � � }t| j�� �D ]6}| j| \}}}||kr|durD|d|� | j|= qdS )z0Check for expected responses that have timed outN)r] �listrH r* )r rm �keyZ expiration� originalPkt�callbackr r r rh s z PacketControl.checkResponsesc C s� | � � t| j�}|dk r dS zZt�| j�\}}}|s<W dS || _| j�|� |j| jkrb|j| _| � dt |� d� W q tj tj tj fy� } z6|jd tjkr�| �|jtjd� |j| _W Y d}~qd}~0 0 qdS )zDSlurps data from the input buffer and tries to parse packets from it� NzReceived a full packet: %srr �typer )� pollSocket�lenrK r% Z decode_packetrL r8 re rO rE �reprZParseVersionErrorZParsePDUTypeErrorZ ParseError�headerrY �sendErrorResponseZRSPERR_PARSE_ERRORZ remainingData)r Zdatalen�pktZfullPktZ extraData�er r r rb s& � zPacketControl.packetEaterc C sT |� � }| �d|t|�f d� | j�|� |rP|j|j|jf}|||f| j|< d S )Nz#Sending packet (with reply: %s): %srr ) �encoderE r| rF ZsendallrN re rf rH )r rk Z expectsReplyZ replyTimeoutrw Zencoded�indexr r r rg . s �� �zPacketControl.sendPacketc sB � j d }|� _ t�d� j|d�}� fdd�}� j|d|d� d S )N� Tr c s | d u rd� _ d S r )rQ )rl �origr r r rw A s z(PacketControl.sendPing.<locals>.callback�rw )rO r% ZPingPDUrN rg )r �tidr rw r r r rj : s zPacketControl.sendPingc sF �j d }|�_ t�d�j|d||�}� �fdd�� ��|d� � d S )Nr� Tr c s | d u r�j |d� d� d S )NTr� )rg )r r� �� resendNotifyr r r r� N s z.PacketControl.sendNotify.<locals>.resendNotify)rO r% Z NotifyPDUrN rg )r �varbindsr r� r r r� r � sendNotifyG s zPacketControl.sendNotifyc C s: t �|d d |d |d |d d||�}| �|d� d S )N�flagsZ bigEndianZ session_idZtransaction_idZ packet_idr F)r% rd rg )r Z errorHeaderZ errorTypeZ errorIndex�errr r r r~ S s �z PacketControl.sendErrorResponsec C sv d}t � | jgg g d�d }|s"qd|d }|�d�}|rd| �dt|� d� ||7 }t�� | _qqdq| j|7 _dS )z@Reads all currently available data from the socket, non-blockingrD r i zReceived data: %sr� N)�selectrF ZrecvrE r| r] rP rK )r �data�tmp�newdatar r r rz [ s zPacketControl.pollSocketc C s� g }|j D ]r}|j}| j�|�\}}}||ks6|d u rL|�t�tj|�� q ||�}|d u rr|�t�tj|�� q |�|� q t� d| j |j |j dtj d|�} | �| d� d S �NTr F)� oidrangesr7 rJ r5 r8 r% �VarbindZVALUE_NO_SUCH_OBJECTZ VALUE_NULLrd rN re rf � ERR_NOERRORrg ) r rk �binds�oidr�targetr+ r! �_�vbindrl r r r rR p s �zPacketControl.handle_GetPDUc C s� g }|j D ]p}| j�|d�}|s:|�t�tj|j�� q q|d \}}}||�}|d u rlt�||j d�}qq|�|� q qq t� d| j |j |j dtjd|�} | �| d� d S r� )r� rJ r; r8 r% r� �VALUE_END_OF_MIB_VIEWr7 Z SearchRanger9 rd rN re rf r� rg ) r rk r� r� r: r+ r! r� r� rl r r r rS � s& � �z PacketControl.handle_GetNextPDUc C s� g }|j d |j� }|j |jd � }|D ]J}| j�|d�}|sV|�t�tj|j�� q(|d \}}} |�||�� q(|D ]T}| j�|�}|s�|�t�tj|j�� qx|d |j � D ]\}}} |�||�� q�qxt� d| j |j |j dtjd|�} | �| d� d S r� )r� ZnonRepsrJ r; r8 r% r� r� r7 ZmaxRepsrd rN re rf r� rg ) r rk r� ZnonrepsZrepeatsr� r: r+ r! r� rl r r r rT � s& �z PacketControl.handle_GetBulkPDUc C sd | j jrd| j _g | j _g | j _g | j _d }tt|j��D ]�}|j| }| j �|j �\}}}|d u rpt j } q�n|d u r�t j } q�||�}|d|�}|t j kr� q�| j j�|� | j j�|� | j j�|� q:|t j k�r6t �d| j|j|jd||�} | �| d� t|�D ] } | j j| d| j j| � �q d| j _n*t �d| j|j|jdt j d�} | �| d� d S )NT�testr F�clear)rJ r r r r �ranger{ r� r5 r+ r% Z ERR_NO_ACCESSZERR_NOT_WRITABLEr� r8 rd rN re rf rg ) r rk �errorZ bindIndexZvarbindr+ r! r" �undoDatarl �ir r r rU � sH � � �z PacketControl.handle_TestSetPDUc C s� | j js| j j}| j j}tt|��D ]$}|| d|| �}|tjkr$ qJq$|tjkrrt�d| j |j |j d||�}n t�d| j |j |j dtjd�}| � |d� d S )NZcommitTr F) rJ r r r r� r{ r% r� rd rN re rf rg )r rk r� �handlersr� r� rl r r r rV � s � �z!PacketControl.handle_CommitSetPDUc C s� | j j}| j j}| j j}tt|��D ]*}|| d|| || �}|tjkr$ qPq$|tjkrxt�d| j |j |j d||�}n t�d| j |j |j dtjd�}| � |d� d S )NZundoTr F) rJ r r r r� r{ r% r� rd rN re rf rg )r rk r� r� r� r� r� rl r r r rW � s � �z PacketControl.handle_UndoSetPDUc C s@ | j j}| j j}tt|��D ]}|| d|| � q d| j _d S )NZcleanF)rJ r r r� r{ r )r rk r� r� r� r r r rX s z"PacketControl.handle_CleanupSetPDUc C sJ |j |j|jf}|| jv rF| j| \}}}| j|= |d urF|||� n d S r )rN re rf rH )r rk r� rM rv rw r r r rZ s z PacketControl.handle_ResponsePDU)F)N)r< r= r> �defaultTimeoutr r_ r\ rq ro rh rb rg rj r� r~ rz rR rS rT rU rV rW rX rZ r r r r r? � s0 � � .r? r c c sB g }g }| }t |�� �}|�� d}|t|�kr\|rX|�� \}}}}|�� |d7 }q$nd S || }t�|t|� |f �}||| �d�|| �d�fV || �d�} | d u�r4|� ||||f� |� |� t | t �r�| }n2| � }|i k�r|�� \}}}}|�� |d7 }q$t |�� �}|�� d}|| }q$|d7 }q$d S )Nr r r! r" r# ) rt r* �sortr{ rc r% r& r' �getr8 r$ �dict) ZtreeZrootpathZ nodeStackZoidStack�currentZ currentKeysZkeyIDru r+ Zsubsr r r r- sD r- )r )Z __future__r r r� r] �sysZntp.utilr( Zntp.agentx_packetZ agentx_packetr% � ImportErrorr� �stderr�write�exitr� ri r r r? r- r r r r �<module> s, � ~