a
{)j+ ã @ 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 || _ | ¡ }| d¡r$|dd… }|| _tt|ƒ| _t |¡| _ t
dd¡| _ d| _
| ¡ 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.9/site-packages/isc/rndc.pyÚ__init__$ s
z
rndc.__init__c C s t | j|dd ƒ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 ƒ }| ¡ D ]ô\}}|r$|dkr$q|t dt|ƒ¡| d¡ 7 }t|ƒtkrp|t ddt|ƒ¡| d¡ 7 }qt|ƒtkr˜|t ddt|ƒ¡| 7 }qt|ƒt krÀ|t ddt|ƒ¡| 7 }qt|ƒt krò|
|¡}|t ddt|ƒ¡| 7 }qt dt|ƒ ƒ‚q|S )NÚ_authÚBÚasciiz>BIé é 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ÚvÚsdr r r Z__serialize_dict; s$ "
ÿzrndc.__serialize_dictc O s0 | j d7 _ tt ¡ ƒ}t|i |¤Ž}tƒ }tƒ |d< tƒ |d< t| j ƒ|d d< t|ƒ|d d< t|d ƒ|d d< | jd ur| j|d d< ||d < | j|d
d }t | j || j
¡ ¡ }t
|¡}| jd kræt d
|¡|d d< n"tt d| j| j |¡ƒ|d d< | |¡}t dt|ƒd d¡| }|S )Nr( r% Ú_ctrlZ_serZ_timé< Z_expÚ_noncer T©r5 r
Z22sÚhmd5ZB88sÚhshaú>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 ur |d d | j kr dS | jdkr8|d d }n|d d dd … }t|ƒtkrb| d ¡}|d
d t|ƒd 7 }t |¡}| j|d d
}t
| j || j ¡
¡ }||kS )Nr: r< Fr
r% r> r? r( r' ú=rA Tr= )r r r r1 Údecoder. r r r2 rD rE r r rF )r rK rM Z remote_hashZmy_msgZmy_hashr r r Z __verify_msgj s
zrndc.__verify_msgc O sÔ | j |i |¤Ž}| j |¡}|t|ƒkr0tdƒ‚| j d¡}t|ƒdkrPtdƒ‚t d|¡\}}|dkrttd| ƒ‚|d8 }| j |tj ¡}t|ƒ|kr tdƒ‚t
|ƒt kr´t |ƒ}|
|¡}| |¡sÐtd ƒ‚|S )
NzCannot send the messageé zCan't read response headerr@ r( zWrong message version %drA 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 rH rI rK ÚsentÚheaderÚlengthÚversionr4 r r r Z __commandy s(
zrndc.__commandc C s2 t | j¡| _ d | _| jdd}|d d | _d S )NZnullr r: r< )rR Zcreate_connectionr r r" )r rK r r r Z__connect_login– s zrndc.__connect_loginc C sê d}|| }|d7 }|||| … d¡}||7 }|| }|d7 }t d|||d … ¡d }|d7 }|||| … }||7 }||d … }|dkrœ|||fS |dkrÚtƒ } t|ƒdkrÐ| |¡\}
} }| | |
< qª|| |fS td| ƒ‚d S )Nr r( r' z>IrA r) zUnknown element type %d)rO r, rU r r. Ú_rndc__parse_elementr3 ) r ÚinputÚposZlabellenÚlabelr Zdatalenr4 ÚrestrJ ZilabelÚvaluer r r Z__parse_elementœ s*
zrndc.__parse_elementc C s4 t ƒ }d }t|ƒdkr0| |¡\}}}|||< q
|S )Nr )r r. r\ )r r] r6 Zhdatar_ ra r r r Z__parse_message¶ s
zrndc.__parse_messageN)F)Ú__name__Ú
__module__Ú __qualname__Ú__doc__rG r r$ r2 rQ rW r" r r\ rV r r r r r s" û
r ) Ú collectionsr rC r, r rD r r rR Úobjectr r r r r Ú s