o 'h,@sdZddlZddlZddlZddlZddlZddlZddlZddl ZGdddej j Z Gdddej j Z Gdddej j ZGd d d ej j ZGd d d ej j ZGd ddeZGdddeZGdddeZGdddeZejdZejdZejdZejdZejdZejdZejdZejdZejdZejdZ eZ!eded ed!ed"ed#ed$ed%ed!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 _digests4     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|} |jdkrY|jtjjkr7t |jtjj kr@t |jtjj krIt |jtjjkrRttd|jt|j||jkret|j|krlt|j|jkrttt| ||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 %dN)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@s(eZdZefddZddZddZdS)r0cCsZt|tr tj|}||_t|trt|}||_t|tr(tj|}||_ dSr$) r/strr:r! from_textbase64 decodebytesencoder?r1)r"r!r?r1r r r r#Hs      z Key.__init__cCs.t|to|j|jko|j|jko|j|jkSr$)r/r0r!r?r1)r"rrr r r __eq__Ss    z Key.__eq__cCsNd|jdd|jd}|jtkr!|dt|jd7}|d7}|S)Nz