3
7&j+ @ sX d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZG dd de Z
dS ) ) OrderedDictNc @ sl e Zd ZdZdddddddZd d
Zd d Zd ddZdd Zdd Z dd Z
dd Z dd Z dd Z
d S ) rndcz RNDC protocol client library )md5Zsha1Zsha224Zsha256Zsha384Zsha512c C sb || _ |j }|jdr$|dd }|| _tt|| _tj|| _ t
j dd| _ d| _
| j dS )zCreates 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 )hostlower
startswithalgogetattrhashlibhlalgobase64 b64decodesecretrandomZrandintsernonce_rndc__connect_login)selfr r r r /usr/lib/python3.6/rndc.py__init__$ s
z
rndc.__init__c C s t | j|dd S )zCall 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_authBasciiz>BIr z#Cannot serialize element of type %s) bytearrayitemsstructpacklenencoder strbytesr _rndc__serialize_dictNotImplementedError)r data ignore_authrvkvZsdr 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 krtjd
||d d< n"ttjd| j| j ||d d< | j|}tjdt|d d| }|S )Nr r& _ctrlZ_serZ_tim< Z_exp_noncer! T)r5 r
Z22shmd5ZB88shshaz>II )r inttimer r0 r r2 hmacnewr r digestr Z b64encoder r, r- r* _rndc__algosr. ) r argskwargsZnowr4 dmsghashbhashr 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|dkrLtdtjd|\}}|dkrptd| |d8 }| jj|tj }t||krtdt
|t krt |}| j
|}| j|std |S )
NzCannot send the message zCan't read response headerz>IIr zWrong message version %dr> zCan't read response datazAuthentication failure)_rndc__prep_messagesocketsendr. IOErrorZrecvr, unpackr3 Z MSG_WAITALLr r0 r* _rndc__parse_message_rndc__verify_msg) r rE rF rH ZsentheaderZlengthversionr4 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 |dkrt } x(t|dkr| j|\}
} }| | |
< qW || |fS td| d S )Nr r r( z>Ir> r) zUnknown element type %d)rL r, rR r r. _rndc__parse_elementr3 ) r inputposZlabellenlabelr Zdatalenr4 restrG Zilabelvaluer 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 s