o j@sUddlmZmZmZddlZddlZddlmZddl m Z ddl m Z ddl mZejdrZddlmZmZdd lmZmZdd lmZmZdd lmZmZmZmZmZd Z nd Z e!ej"j#BdBZ$iZ%eee e$feefe&d<e re%'e j(dfee j)dfee j*dfee j+dfee j,dfee j-dfee j.dfee j/dfee j0dfee j1dfee j2dfei  dde3e4Bde$deefddZ5dedeefddZ6  dde3e4Bdeedej"j#e4BdBde!dBddf ddZ7dS))DictTupleTypeN)GenericPrivateKey) Algorithm)UnsupportedAlgorithm)DNSKEYdnssec) PrivateDSAPrivateDSANSEC3SHA1)PrivateECDSAP256SHA256PrivateECDSAP384SHA384) PrivateED448PrivateED25519) PrivateRSAMD5PrivateRSASHA1PrivateRSASHA1NSEC3SHA1PrivateRSASHA256PrivateRSASHA512TF algorithms algorithmprefixreturncCs6t|}t||f}|r|Stdt|d)zGet Private Key class from Algorithm. *algorithm*, a ``str`` or ``int`` specifying the DNSKEY algorithm. Raises ``UnsupportedAlgorithm`` if the algorithm is unknown. Returns a ``dns.dnssecalgs.GenericPrivateKey`` z algorithm "z" not supported by dnspython)rmakergetrto_text)rrclsr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/dns/dnssecalgs/__init__.pyget_algorithm_cls2s rdnskeycCs^d}|jtjkrtj|jd\}}n|jtjkr)t|jd}|jd|d}t |j|S)zGet Private Key class from DNSKEY. *dnskey*, a ``DNSKEY`` to get Algorithm class for. Raises ``UnsupportedAlgorithm`` if the algorithm is unknown. Returns a ``dns.dnssecalgs.GenericPrivateKey`` Nr) rr PRIVATEDNSdnsname from_wirekey PRIVATEOIDintr)r r_lengthrrrget_algorithm_cls_from_dnskeyFs   r+ algorithm_clsr$oidcCst|ts tdt|}d}|tjkr+|durtdt|tr(t j |}|}n#|tj krB|dur8tdt t|g|}n |rHtd|rNtd|t||f<dS)aRegister Algorithm Private Key class. *algorithm*, a ``str`` or ``int`` specifying the DNSKEY algorithm. *algorithm_cls*: A `GenericPrivateKey` class. *name*, an optional ``dns.name.Name`` or ``str``, for for PRIVATEDNS algorithms. *oid*: an optional BER-encoded `bytes` for PRIVATEOID algorithms. Raises ``ValueError`` if a name or oid is specified incorrectly. zInvalid algorithm classNz'Name required for PRIVATEDNS algorithmsz&OID required for PRIVATEOID algorithmsz,Name only supported for PRIVATEDNS algorithmz+OID only supported for PRIVATEOID algorithm) issubclassr TypeErrorrrr" ValueError isinstancestrr#r$ from_textr'byteslenr)rr,r$r-rrrrregister_algorithm_clsXs&      r6)N)NN)8typingrrr dns._featuresr#dns.namedns.dnssecalgs.baserdns.dnssectypesr dns.exceptionrdns.rdtypes.ANY.DNSKEYr _featureshavedns.dnssecalgs.dsar r dns.dnssecalgs.ecdsar r dns.dnssecalgs.eddsarrdns.dnssecalgs.rsarrrrr_have_cryptographyr4r$NameAlgorithmPrefixr__annotations__updateRSAMD5DSARSASHA1 DSANSEC3SHA1RSASHA1NSEC3SHA1 RSASHA256 RSASHA512ECDSAP256SHA256ECDSAP384SHA384ED25519ED448r(r2rr+r6rrrrsj