File: //lib64/python3.9/site-packages/ntp/__pycache__/agentx.cpython-39.opt-1.pyc
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,
� ~