U
    5i                     @   s   d Z ddlZddlZddlZddlZddlZddlmZ ddlm	Z	m
Z
 ddlmZ ddlmZ ddlmZ edZd	ZG d
d deZG dd deZdd Zdd Zdd Zdd Zd(ddZd)ddZG dd dZddedfd d!Zddedfd"d#ZG d$d% d%ZG d&d' d'eZ dS )*ae  
Functions for creating and restoring url-safe signed JSON objects.

The format used looks like this:

>>> signing.dumps("hello")
'ImhlbGxvIg:1QaUZC:YIye-ze3TTx7gtSv422nZA4sgmk'

There are two components here, separated by a ':'. The first component is a
URLsafe base64 encoded JSON of the object passed to dumps(). The second
component is a base64 encoded hmac/SHA1 hash of "$first_component:$secret"

signing.loads(s) checks the signature and returns the deserialized object.
If the signature fails, a BadSignature exception is raised.

>>> signing.loads("ImhlbGxvIg:1QaUZC:YIye-ze3TTx7gtSv422nZA4sgmk")
'hello'
>>> signing.loads("ImhlbGxvIg:1QaUZC:YIye-ze3TTx7gtSv422nZA4sgmk-modified")
...
BadSignature: Signature failed: ImhlbGxvIg:1QaUZC:YIye-ze3TTx7gtSv422nZA4sgmk-modified

You can optionally compress the JSON prior to base64 encoding it to save
space, using the compress=True argument. This checks if compression actually
helps and only applies compression if the result is a shorter string:

>>> signing.dumps(list(range(1, 20)), compress=True)
'.eJwFwcERACAIwLCF-rCiILN47r-GyZVJsNgkxaFxoDgxcOHGxMKD_T7vhAml:1QaUaL:BA0thEZrp4FQVXIXuOvYJtLJSrQ'

The fact that the string is compressed is signalled by the prefixed '.' at the
start of the base64 JSON.

There are 65 url-safe characters: the 64 used by url-safe base64 and the ':'.
These functions make use of all of them.
    N)settings)constant_time_comparesalted_hmac)force_bytes)import_string)_lazy_re_compilez^[A-z0-9-_=]*$Z>0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzc                   @   s   e Zd ZdZdS )BadSignaturezSignature does not match.N__name__
__module____qualname____doc__ r   r   7/tmp/pip-unpacked-wheel-gcyiijx6/django/core/signing.pyr   4   s   r   c                   @   s   e Zd ZdZdS )SignatureExpiredz3Signature timestamp is older than required max_age.Nr	   r   r   r   r   r   9   s   r   c                 C   sT   | dkrdS | dk rdnd}t | } d}| dkrLt| d\} }t| | }q(|| S )Nr   0- >   )absdivmodBASE62_ALPHABET)ssignencoded	remainderr   r   r   
b62_encode>   s    r   c                 C   sT   | dkrdS d}| d dkr,| dd  } d}d}| D ]}|d t | }q4|| S )Nr   r      r   r   )r   index)r   r   decodeddigitr   r   r   
b62_decodeJ   s    r"   c                 C   s   t | dS )N   =)base64urlsafe_b64encodestrip)r   r   r   r   
b64_encodeW   s    r'   c                 C   s    dt |  d  }t| | S )Nr#      )lenr$   urlsafe_b64decode)r   padr   r   r   
b64_decode[   s    r,   sha1c                 C   s   t t| |||d  S )N	algorithm)r'   r   digestdecode)saltvaluekeyr/   r   r   r   base64_hmac`   s    r5   %django.core.signing.get_cookie_signerc                 C   s$   t tj}ttj}|d| | dS )Ns   django.http.cookiesr2   )r   r   ZSIGNING_BACKENDr   
SECRET_KEY)r2   Signerr4   r   r   r   get_cookie_signerd   s    

r:   c                   @   s    e Zd ZdZdd Zdd ZdS )JSONSerializerzW
    Simple wrapper around json to be used in signing.dumps and
    signing.loads.
    c                 C   s   t j|dddS )N),:)
separatorslatin-1)jsondumpsencode)selfobjr   r   r   rA   o   s    zJSONSerializer.dumpsc                 C   s   t |dS )Nr?   )r@   loadsr1   )rC   datar   r   r   rE   r   s    zJSONSerializer.loadsN)r
   r   r   r   rA   rE   r   r   r   r   r;   j   s   r;   zdjango.core.signingFc                 C   s   t ||dj| ||dS )a  
    Return URL-safe, hmac signed base64 compressed JSON string. If key is
    None, use settings.SECRET_KEY instead. The hmac algorithm is the default
    Signer algorithm.

    If compress is True (not the default), check if compressing using zlib can
    save some space. Prepend a '.' to signify compression. This is included
    in the signature, to protect against zip bombs.

    Salt can be used to namespace the hash, so that a signed string is
    only valid for a given namespace. Leaving this at the default
    value or re-using a salt value across different parts of your
    application without good cause is a security risk.

    The serializer is expected to return a bytestring.
    r7   )
serializercompress)TimestampSignersign_object)rD   r4   r2   rG   rH   r   r   r   rA   v   s    rA   c                 C   s   t ||dj| ||dS )z|
    Reverse of dumps(), raise BadSignature if signature fails.

    The serializer is expected to accept a bytestring.
    r7   )rG   max_age)rI   unsign_object)r   r4   r2   rG   rK   r   r   r   rE      s    rE   c                   @   sH   e Zd ZdddZdd Zdd Zd	d
 ZedfddZefddZ	dS )r9   Nr=   c                 C   sR   |pt j| _|| _t| jr*td| |p@d| jj| jj	f | _
|pJd| _d S )NzJUnsafe Signer separator: %r (cannot be empty or consist of only A-z0-9-_=)z%s.%ssha256)r   r8   r4   sep_SEP_UNSAFEmatch
ValueError	__class__r   r
   r2   r/   )rC   r4   rN   r2   r/   r   r   r   __init__   s    zSigner.__init__c                 C   s   t | jd || j| jdS )NZsignerr.   )r5   r2   r4   r/   rC   r3   r   r   r   	signature   s    zSigner.signaturec                 C   s   d|| j | |f S Nz%s%s%s)rN   rU   rT   r   r   r   r      s    zSigner.signc                 C   sN   | j |krtd| j  || j d\}}t|| |r>|S td| d S )NzNo "%s" found in valuer   zSignature "%s" does not match)rN   r   rsplitr   rU   )rC   Zsigned_valuer3   sigr   r   r   unsign   s    
zSigner.unsignFc                 C   s\   |  |}d}|r:t|}t|t|d k r:|}d}t| }|rRd| }| |S )ae  
        Return URL-safe, hmac signed base64 compressed JSON string.

        If compress is True (not the default), check if compressing using zlib
        can save some space. Prepend a '.' to signify compression. This is
        included in the signature, to protect against zip bombs.

        The serializer is expected to return a bytestring.
        Fr   T.)rA   zlibrH   r)   r'   r1   r   )rC   rD   rG   rH   rF   Zis_compressed
compressedbase64dr   r   r   rJ      s    

zSigner.sign_objectc                 K   sT   | j |f| }|d d dk}|r2|dd  }t|}|rHt|}| |S )Nr      .)rY   rB   r,   r[   
decompressrE   )rC   Z
signed_objrG   kwargsr]   r_   rF   r   r   r   rL      s    
zSigner.unsign_object)Nr=   NN)
r
   r   r   rS   rU   r   rY   r;   rJ   rL   r   r   r   r   r9      s   
r9   c                       s2   e Zd Zdd Z fddZd fdd	Z  ZS )	rI   c                 C   s   t tt S )N)r   inttime)rC   r   r   r   	timestamp   s    zTimestampSigner.timestampc                    s    d|| j |  f }t |S rV   )rN   rc   superr   rT   rR   r   r   r      s    zTimestampSigner.signNc                    sj   t  |}|| jd\}}t|}|dk	rft|tjrB| }t		 | }||krft
d||f |S )zk
        Retrieve original value and check it wasn't signed more
        than max_age seconds ago.
        r   NzSignature age %s > %s seconds)rd   rY   rW   rN   r"   
isinstancedatetime	timedeltatotal_secondsrb   r   )rC   r3   rK   resultrc   Zagere   r   r   rY      s    
zTimestampSigner.unsign)N)r
   r   r   rc   r   rY   __classcell__r   r   re   r   rI      s   rI   )r-   )r6   )!r   r$   rg   r@   rb   r[   Zdjango.confr   Zdjango.utils.cryptor   r   Zdjango.utils.encodingr   Zdjango.utils.module_loadingr   Zdjango.utils.regex_helperr   rO   r   	Exceptionr   r   r   r"   r'   r,   r5   r:   r;   rA   rE   r9   rI   r   r   r   r   <module>   s2   #

	A