File: //usr/lib/python3.9/site-packages/dns/dnssecalgs/__pycache__/__init__.cpython-39.pyc
a
�M�e� � @ s� U d dl mZmZmZmZmZ d dlZd dlm Z d dl
m Z d dl m
Z
d dlmZ ej�d�r�d 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 d�dd�Z6eee d�dd�Z7dee4e5f ee eeej#j$e5f ee" dd�dd�Z8dS )� )�Dict�Optional�Tuple�Type�UnionN)�GenericPrivateKey)� Algorithm)�UnsupportedAlgorithm)�DNSKEYZdnssec)�
PrivateDSA�PrivateDSANSEC3SHA1)�PrivateECDSAP256SHA256�PrivateECDSAP384SHA384)� PrivateED448�PrivateED25519)�
PrivateRSAMD5�PrivateRSASHA1�PrivateRSASHA1NSEC3SHA1�PrivateRSASHA256�PrivateRSASHA512TF�
algorithms)� algorithm�prefix�returnc C s6 t �| �} t�| |f�}|r |S tdt �| � ��dS )z�Get 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__.py�get_algorithm_cls. s
�r )�dnskeyr c C s^ d}| j tjkr&tj�| jd�\}}n,| j tjkrRt| jd �}| jd|d � }t | j |�S )z�Get 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 �
PRIVATEDNS�dns�nameZ from_wire�key�
PRIVATEOID�intr )r r �_�lengthr r r � get_algorithm_cls_from_dnskeyB s r* )r �
algorithm_clsr$ �oidr c C s� t |t�std��t�| �} d}| tjkrV|du r:td��t|t�rPt j
� |�}|}nH| tj kr�|du rptd��t
t|�g�| }n|r�td��n |r�td��|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�
isinstance�strr# r$ Z from_textr& �bytes�lenr )r r+ r$ r, r r r r �register_algorithm_clsT s&
r4 )N)NN)9�typingr 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 �<module> sX
�� � � �
�