o h8-@sdZddlZddlZddlZddlZddlZddlZddlZddl Zddl ZGdddej j Z Gdddej j ZGdddej j ZGd d d ej j ZGd d d ej j ZGd ddeZGdddeZGdddeZGdddeZejdZejdZejdZejdZejdZejdZejdZejdZejdZ ejdZ!eZ"eded ed!ed"ed#ed$ed%e d!ed"e!d&i Z#Gd'd(d(Z$Gd)d*d*Z%Gd+d,d,Z&d:d-d.Z'd/d0Z(d;d2d3Z) 1d 65535 bytesz!HH) get_contextr&structpacklen original_idr! to_digestabler:r;r< time_signedfudgeother ValueErrorr1error) wirer6rdatatime request_macctxmultifirst upper_time lower_time time_encoded other_lenr r r _digests6      rcCs4|rt|}|tdt||||SdS)zIf this is the first message in a multi-message sequence, start a new context. @rtype: dns.tsig.HMACTSig or dns.tsig.GSSTSig object rhN)rjr&rkrlrm)r6rWrzryr r r _maybe_start_digests  rFc Cs:t|||||||}|}|j||d}|t|||fS)a~Return a (tsig_rdata, mac, ctx) tuple containing the HMAC TSIG rdata for the input parameters, the HMAC MAC calculated by applying the TSIG signature algorithm, and the TSIG digest context. @rtype: (string, dns.tsig.HMACTSig or dns.tsig.GSSTSig object) @raises ValueError: I{other_data} is too long @raises NotImplementedError: I{algorithm} is not supported )rprW)rr(replacer) rur6rvrwrxryrzrWtsigr r r r(s r(c Cstd|dd\} | dkrtjj| d8} |ddtd| |d|} |jdkrZ|jtjjkr7t |jtjj kr@t |jtjj krIt |jtjjkrRttd|jt|j||jkrft|j|krmt|j|jkrutt| ||d|||}||jt||j|S)aFValidate the specified TSIG rdata against the other input parameters. @raises FormError: The TSIG is badly formed. @raises BadTime: There is too much time skew between the client and the server. @raises BadSignature: The TSIG signature did not validate @rtype: dns.tsig.HMACTSig or dns.tsig.GSSTSig objectrh rrJzunknown TSIG error code N)rkunpackr: exception FormErrorrlrtrcodeBADSIGrBADKEYrBADTIMErBADTRUNCrrabsrprqrr!r r1rrr,rWr) rur6ownerrvnowrx tsig_startryrzadcountnew_wirer r r validates0 $    rcCs"|jtkr t|jSt|j|jS)zReturns an HMAC context for the specified key. @rtype: HMAC context @raises NotImplementedError: I{algorithm} is not supported )r1r2rr?rF)r6r r r rj<s  rjc@sJeZdZefdejjeBdeeBdejjeBfddZ ddZ dd Z d S) r0r!r?r1cCsZt|tr tj|}||_t|trt|}||_t|tr(tj|}||_ dSr$) r/strr:r! from_textbase64 decodebytesencoder?r1)r"r!r?r1r r r r#Js      z Key.__init__cCs.t|to|j|jko|j|jko|j|jkSr$)r/r0r!r?r1)r"rrr r r __eq__Zs    z Key.__eq__cCsNd|jdd|jd}|jtkr!|dt|jd7}|d7}|S)Nz