U
    ˹h                     @   s(  d dl Z d dlZd dlZzd dlmZ d9ddZW n* ek
r\   d dlm	Z d:ddZY nX d;ddZ
d	d
 Zdd Zdd Zdd Zdd Zdd Zdd Zdddddddddd d!d"d#d$d%d&hZed'd(d)d* eD  d+ Zeed,d-d.Zd/Zed0Zd1d2d3d4d5d6gZeed,d7d8ZdS )<    N)int_to_bytesc                 C   s   t | |p
d S N)_long_to_bytes)n	blocksize r   0./venv/lib/python3.8/site-packages/jose/utils.pylong_to_bytes
   s    r	   )int_to_stringc                 C   s@   t | }|dkr|S t||ks$t|t| }d| | S d S )Nr       )r   lenAssertionError)r   r   retZpaddingr   r   r   r	      s    c                 C   s   t t| |dS )N   =)base64urlsafe_b64encoder	   strip)datasizer   r   r   long_to_base64   s    r   c                 C   s   t ddd | D dS )N c                 S   s   g | ]}d | qS )z%02xr   ).0Zbyter   r   r   
<listcomp>   s     z#int_arr_to_long.<locals>.<listcomp>   )intjoin)Zarrr   r   r   int_arr_to_long   s    r   c                 C   s>   t | tr| d} tt| d }ttdt	| |S )Nasciis   ==z%sB)

isinstancestrencoder   urlsafe_b64decodebytesr   structunpackr   )r   Z_dr   r   r   base64_to_long"   s    

r%   c                 C   s@   ||  d }tt|d }|d| }t|}|dS )a  Helper method for calculating an access token
    hash, as described in http://openid.net/specs/openid-connect-core-1_0.html#CodeIDToken

    Its value is the base64url encoding of the left-most half of the hash of the octets
    of the ASCII representation of the access_token value, where the hash algorithm
    used is the hash algorithm used in the alg Header Parameter of the ID Token's JOSE
    Header. For instance, if the alg is RS256, hash the access_token value with SHA-256,
    then take the left-most 128 bits and base64url encode them. The at_hash value is a
    case sensitive string.

    Args:
        access_token (str): An access token string.
        hash_alg (callable): A callable returning a hash object, e.g. hashlib.sha256

    utf-8   N)r    Zdigestr   r   base64url_encodedecode)Zaccess_tokenZhash_algZhash_digestZcut_atZ	truncatedZat_hashr   r   r   calculate_at_hash+   s
    r*   c                 C   s.   t | d }|dkr$| dd|  7 } t| S )zwHelper method to base64url_decode a string.

    Args:
        input (str): A base64url_encoded string to decode.

       r   r   )r   r   r!   )inputZremr   r   r   base64url_decodeB   s    r-   c                 C   s   t | ddS )zwHelper method to base64url_encode a string.

    Args:
        input (str): A base64url_encoded string to encode.

    r       )r   r   replace)r,   r   r   r   r(   Q   s    r(   c                 C   s   | j d d d | j S )zHelper method to determine the total number of seconds
    from a timedelta.

    Args:
        delta (timedelta): A timedelta to convert to seconds.
       <   )ZdaysZseconds)Zdeltar   r   r   timedelta_total_seconds[   s    r2   c                 C   s<   t | tr| S t | tr$| ddS tdt|  ddS )zCoerce **s** to bytes.r&   strictznot expecting type ''N)r   r"   r   r    	TypeErrortype)sr   r   r   ensure_binarye   s
    

r8   s   CERTIFICATEs   TRUSTED CERTIFICATEs   PRIVATE KEYs
   PUBLIC KEYs   ENCRYPTED PRIVATE KEYs   OPENSSH PRIVATE KEYs   DSA PRIVATE KEYs   RSA PRIVATE KEYs   RSA PUBLIC KEYs   EC PRIVATE KEYs   DH PARAMETERSs   NEW CERTIFICATE REQUESTs   CERTIFICATE REQUESTs   SSH2 PUBLIC KEYs   SSH2 ENCRYPTED PRIVATE KEYs   X509 CRLs   ----[- ]BEGIN (   |c                 c   s   | ]}t |V  qd S r   )reescape)r   Zpemr   r   r   	<genexpr>   s     r<   s	   )[- ]----)keyreturnc                 C   s   t t| S r   )bool_PEM_REsearchr=   r   r   r   is_pem_format   s    rC   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 }|rL|d}t|tt d  krLdS dS )Nc                 3   s   | ]}| kV  qd S r   r   )r   Zstring_valuerB   r   r   r<      s     zis_ssh_key.<locals>.<genexpr>T   F)any_SSH_KEY_FORMATS_SSH_PUBKEY_RCmatchgroup_CERT_SUFFIXr   )r=   Zssh_pubkey_matchZkey_typer   rB   r   
is_ssh_key   s    

rK   )r   )r   )r   )r   r:   r#   Zcryptography.utilsr   r   r	   ImportErrorZecdsa.ecdsar
   r   r   r%   r*   r-   r(   r2   r8   Z_PEMScompiler   r@   r"   r?   rC   rJ   rG   rF   rK   r   r   r   r   <module>   s^   

	



