a
Me @ s U d dl mZmZmZmZmZ d dlZd dlm Z d dl
m Z d dl m
Z
d dlmZ ejdrd dlmZmZ d d lmZmZ d d
lmZmZ d d lm Z m Z m Z m Z m Z d Z!nd
Z!eee"ej#j$f Z%i Z&eee e%f ee f e'd< e!rje&(e j)dfe e j*dfee j+dfe e j,dfee j-dfe e j.dfe e j/dfe e j0dfee j1dfee j2dfee j3dfei dee4e5f e%ee dddZ6eee dddZ7dee4e5f ee eeej#j$e5f ee" ddddZ8dS ) )DictOptionalTupleTypeUnionN)GenericPrivateKey) Algorithm)UnsupportedAlgorithm)DNSKEYZdnssec)
PrivateDSAPrivateDSANSEC3SHA1)PrivateECDSAP256SHA256PrivateECDSAP384SHA384) PrivateED448PrivateED25519)
PrivateRSAMD5PrivateRSASHA1PrivateRSASHA1NSEC3SHA1PrivateRSASHA256PrivateRSASHA512TF
algorithms) algorithmprefixreturnc C s6 t | } t| |f}|r |S tdt | dS )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 "%s" not supported by dnspythonN)r maker getr Zto_text)r r cls r ;/usr/lib/python3.9/site-packages/dns/dnssecalgs/__init__.pyget_algorithm_cls. s
r )dnskeyr c C s^ d}| j tjkr&tj| jd\}}n,| j tjkrRt| 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 )
r r
PRIVATEDNSdnsnameZ from_wirekey
PRIVATEOIDintr )r r _lengthr r r get_algorithm_cls_from_dnskeyB s r* )r
algorithm_clsr$ oidr c C s t |tstdt| } d}| tjkrV|du r:tdt|trPt j
|}|}nH| tj kr|du rptdt
t|g| }n|rtdn |rtd|t| |f< dS )a Register 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 TypeErrorr r r"
ValueError
isinstancestrr# r$ Z from_textr& byteslenr )r r+ r$ r, r r r r register_algorithm_clsT s&
r4 )N)NN)9typingr r r r r Zdns.namer# Zdns.dnssecalgs.baser Zdns.dnssectypesr Z
dns.exceptionr Zdns.rdtypes.ANY.DNSKEYr
Z _featuresZhaveZdns.dnssecalgs.dsar r Zdns.dnssecalgs.ecdsar
r Zdns.dnssecalgs.eddsar r Zdns.dnssecalgs.rsar r r r r Z_have_cryptographyr2 r$ NameZAlgorithmPrefixr __annotations__updateZRSAMD5ZDSAZRSASHA1Z DSANSEC3SHA1ZRSASHA1NSEC3SHA1Z RSASHA256Z RSASHA512ZECDSAP256SHA256ZECDSAP384SHA384ZED25519ZED448r' r1 r r* r4 r r r r sX