Current File : //lib/python3.6/site-packages/isc/__pycache__/rndc.cpython-36.pyc
3
���f+ � @ sX d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZG dd� de �Z
dS )� )� OrderedDictNc @ sl e Zd ZdZddddddd�Zd d
� Zd d � Zd dd�Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
d S ) �rndcz RNDC protocol client library� � � � � � )�md5Zsha1Zsha224Zsha256Zsha384Zsha512c C sb || _ |j� }|jd�r$|dd� }|| _tt|�| _tj|�| _ t
j dd�| _ d| _
| j� dS )z�Creates a persistent connection to RNDC and logs in
host - (ip, port) tuple
algo - HMAC algorithm: one of md5, sha1, sha224, sha256, sha384, sha512
(with optional prefix 'hmac-')
secret - HMAC secret, base64 encodedzhmac-� Nr � � i )�host�lower�
startswith�algo�getattr�hashlib�hlalgo�base64� b64decode�secret�randomZrandint�ser�nonce�_rndc__connect_login)�selfr r r � r �/usr/lib/python3.6/rndc.py�__init__$ s
z
rndc.__init__c C s t | j|d�d �S )z�Call a RNDC command, all parsing is done on the server side
cmd - a complete string with a command (eg 'reload zone example.com')
)�type�_data)�dict�_rndc__command)r �cmdr r r �call5 s z rndc.callFc C s t � }�x |j� D ]�\}}|r(|dkr(q|tjdt|��|jd� 7 }t|�tkrt|tjddt|��|jd� 7 }qt|�tkr�|tjddt|��| 7 }qt|�t kr�|tjddt|��| 7 }qt|�t kr�| j
|�}|tjddt|��| 7 }qt dt|� ��qW |S )N�_auth�B�asciiz>BIr � z#Cannot serialize element of type %s) � bytearray�items�struct�pack�len�encoder �str�bytesr �_rndc__serialize_dict�NotImplementedError)r �data� ignore_auth�rv�k�vZsdr r r Z__serialize_dict; s" "
zrndc.__serialize_dictc O s, | j d7 _ ttj� �}t||�}t� }t� |d< t� |d< t| j �|d d< t|�|d d< t|d �|d d< | jd k r�| j|d d< ||d < | j|d
d �}tj| j || j
�j � }t j
|�}| jd kr�tjd
|�|d d< n"ttjd| j| j |��|d d< | j|�}tjdt|�d d�| }|S )Nr r& �_ctrlZ_serZ_tim�< Z_exp�_noncer! T)r5 r
Z22s�hmd5ZB88s�hshaz>II� )r �int�timer r0 r r2 �hmac�newr r �digestr Z b64encoder r, r- r* � _rndc__algosr. ) r �args�kwargsZnowr4 �d�msg�hash�bhashr r r Z__prep_messageO s,
zrndc.__prep_messagec C s� | j d k r |d d | j kr dS | jdkr8|d d }n|d d dd � }t|�tkrb|jd �}|d
d t|�d 7 }tj|�}| j|d d
�}t j
| j || j �j
� }||kS )Nr9 r; Fr
r& r< r= r r( �=r> T)r5 )r r r r1 �decoder. r r r2 rA rB r r rC )r rH rJ Z remote_hashZmy_msgZmy_hashr r r Z __verify_msgj s
zrndc.__verify_msgc O s� | j ||�}| jj|�}|t|�kr,td��| jjd�}t|�dkrLtd��tjd|�\}}|dkrptd| ��|d8 }| jj|tj �}t|�|kr�td��t
|�t kr�t |�}| j
|�}| j|�s�td ��|S )
NzCannot send the message� zCan't read response headerz>IIr zWrong message version %dr> zCan't read response datazAuthentication failure)�_rndc__prep_message�socket�sendr. �IOErrorZrecvr, �unpackr3 Z MSG_WAITALLr r0 r* �_rndc__parse_message�_rndc__verify_msg) r rE rF rH Zsent�headerZlength�versionr4 r r r Z __commandy s(
zrndc.__commandc C s2 t j| j�| _ d | _| jdd�}|d d | _d S )NZnull)r r9 r; )rO Zcreate_connectionr r r# )r rH r r r Z__connect_login� s zrndc.__connect_loginc C s� d}|| }|d7 }|||| � j d�}||7 }|| }|d7 }tjd|||d � �d }|d7 }|||| � }||7 }||d � }|dkr�|||fS |dkr�t� } x(t|�dkr�| j|�\}
} }| | |
< q�W || |fS td| ��d S )Nr r r( z>Ir> r) zUnknown element type %d)rL r, rR r r. �_rndc__parse_elementr3 ) r �input�posZlabellen�labelr Zdatalenr4 �restrG Zilabel�valuer r r Z__parse_element� s*
zrndc.__parse_elementc C s8 t � }d }x(t|�dkr2| j|�\}}}|||< q W |S )Nr )r r. rW )r rX r6 ZhdatarZ r\ r r r Z__parse_message� s zrndc.__parse_messageN)F)�__name__�
__module__� __qualname__�__doc__rD r r% r2 rN rT r# r rW rS r r r r r s
r ) � collectionsr r@ r, r rA r r rO �objectr r r r r �<module> s
Mini Shell