CasperSecurity
o
�l�i� � @ s� d dl Z d dlZd dlZd dlmZmZ zd dlmZ d dlm Z m
Z
W n e y/ eZY nw dee e
f de
fdd�Zd ee e
f de
fd
d �Zd e
de
fd d
�Zdede
fdd�Zdee e
f defdd�Zdedede
fdd�Zde
defdd�Zdede
fdd�Zd e
d ede
fd d �Zd e
d ede
fd!d"�Zh d#�Ze�d$d%�e� d& ej �Z d'e
de fd(d)�Z d*Z e�d+�Z!g d,�Z"d'e
de fd-d.�Z#dS )/� N)�Any�Union)�
EllipticCurve)�decode_dss_signature�encode_dss_signature�value�returnc C s* t | t�r
| �d�S t | t�r| S td��)Nzutf-8zExpected a string value)�
isinstance�str�encode�bytes� TypeError)r � r �+/usr/lib/python3/dist-packages/jwt/utils.py� force_bytes s
r �inputc C sB t | t�r
| �d�} t| �d }|dkr | dd| 7 } t�| �S )N�ascii� r � =)r r
r �len�base64�urlsafe_b64decode)r �remr r r �base64url_decode s
r c C s t �| ��dd�S )Nr � )r �urlsafe_b64encode�replace)r r r r �base64url_encode% s r �valc C s0 | dk rt d��t| �}t|�dkrd}t|�S )Nr zMust be a positive integer� )�
ValueError�bytes_from_intr r )r � int_bytesr r r �to_base64url_uint) s r# c C s* t | t�r
| �d�} t| �}tj|dd�S )Nr �big)� byteorder)r r
r r �int�
from_bytes)r �datar r r �from_base64url_uint5 s
r) �num� num_bytesc C s dd| | f }t �|�d��S )Nz%0*x� r )�binascii�a2b_hexr )r* r+ �
padded_hexr r r �number_to_bytes= s r0 �stringc C s t t�| �d�S )N� )r& r- �b2a_hex)r1 r r r �bytes_to_numberB s r4 c C s8 | }d}|dkr|dL }|d7 }|dks| j |ddd�S )Nr � � r$ F)�signed)�to_bytes)r � remaining� byte_lengthr r r r! F s �r! �der_sig�curvec C s2 |j }|d d }t| �\}}t||�t||� S )N� r5 )�key_sizer r0 )r; r< �num_bitsr+ �r�sr r r �der_to_raw_signatureQ s rB �raw_sigc C sT |j }|d d }t| �d| krtd��t| d |� �}t| |d � �}t||�S )Nr= r5 r, zInvalid signature)r> r r r4 r )rC r< r? r+ r@ rA r r r �raw_to_der_signatureZ s
rD > � X509 CRL�
PUBLIC KEY� CERTIFICATE� PRIVATE KEY�
DH PARAMETERS� EC PRIVATE KEY� RSA PUBLIC KEY� DSA PRIVATE KEY� RSA PRIVATE KEY� SSH2 PUBLIC KEY� CERTIFICATE REQUEST� OPENSSH PRIVATE KEY� TRUSTED CERTIFICATE� ENCRYPTED PRIVATE KEY� NEW CERTIFICATE REQUEST� SSH2 ENCRYPTED PRIVATE KEYs ----[- ]BEGIN (� |s, )[- ]----
?
.+?
?
----[- ]END \1[- ]----
?
?�keyc C s t t�| ��S �N)�bool�_PEM_RE�search�rV r r r �
is_pem_format� s r\ s -cert-v01@openssh.coms \A(\S+)[ \t]+(\S+))s ssh-ed25519s ssh-rsas ssh-dsss ecdsa-sha2-nistp256s ecdsa-sha2-nistp384s ecdsa-sha2-nistp521c sP t � fdd�tD ��r
dS t�� �}|r&|�d�}t|tt� d � kr&dS dS )Nc 3 s � | ]}|� v V qd S rW r )�.0� string_valuer[ r r � <genexpr>� s � z is_ssh_key.<locals>.<genexpr>Tr6 F)�any�_SSH_KEY_FORMATS�_SSH_PUBKEY_RC�match�group� _CERT_SUFFIXr )rV �ssh_pubkey_match�key_typer r[ r �
is_ssh_key� s
rh )$r r- �re�typingr r �,cryptography.hazmat.primitives.asymmetric.ecr �/cryptography.hazmat.primitives.asymmetric.utilsr r �ModuleNotFoundErrorr
r r r r r& r# r) r0 r4 r! rB rD �_PEMS�compile�join�DOTALLrY rX r\ re rb ra rh r r r r �<module> sD � ���