o
    7b                     @   s  d dl Z d dlZd dlZd dlmZmZ zd dlmZ d dlm	Z	m
Z
 W n ey/   eZY nw deeef defddZd	eeef defd
dZd	edefddZdedefddZdeeef defddZdededefddZdedefddZdedefddZdededefddZd ededefd!d"Zh d#Zed$d%e d& ejZd'edefd(d)Zd*Z ed+Z!g d,Z"d'edefd-d.Z#dS )/    N)AnyUnion)EllipticCurve)decode_dss_signatureencode_dss_signaturevaluereturnc                 C   s*   t | tr
| dS t | tr| S td)Nzutf-8zExpected 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 }|dkr| dd|  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 | ddS )Nr       )r   urlsafe_b64encodereplace)r   r   r   r   base64url_encode%   s   r   valc                 C   s0   | dk rt dt| }t|dkrd}t|S )Nr   zMust 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| }tj|dd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    dd| | 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}|dkr|dL }|d7 }|dks| j |ddd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| krtdt| d | }t| |d  }t||S )Nr=   r5   r,   zInvalid 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  fddtD rdS t }|r&|d}t|tt d  kr&dS dS )Nc                 3   s    | ]}| v V  qd 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    		


