Current File : //lib/python3.9/site-packages/passlib/utils/__pycache__/binary.cpython-39.opt-1.pyc
a
f�Wc�z � @ s� d Z ddlmZmZmZ ddlmZmZmZ m
Z ddl m
Z
mZmZ ddlZe�e�ZddlmZ ddlmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z# ddl$m%Z% g d �Z&e!d
�Z'e!d �Z(e!d �Z)e!d
�Z*e'e!d� Z+e!d�Z,e!d�Z-e!d�Z.e ed��Z/dZ0dZ1dZ2e3e e/��Z4d0dd�Z5dd� Z6dd� Z7d Z8dZ9d Z:d d � Z;d d!� Z<d"d#� Z
e5d$d%d&��Z=e2d' Z>d(d)� ZG d*d+� d+e?�Z@G d,d-� d-e@�ZAeAe)�ZBeAe)d.d/�ZCeAe*d.d/�ZDdS )1zC
passlib.utils.binary - binary data encoding/decoding/manipulation
� )�absolute_import�division�print_function)� b64encode� b64decode� b32decode� b32encode)�
b2a_base64�
a2b_base64�ErrorN)�exc) �PY3�
bascii_to_str�irange�imap�iter_byte_chars�join_byte_values�join_byte_elems�
nextgetter�suppress_cause�u�unicode�unicode_or_bytes_types)�memoized_property)� BASE64_CHARS�PADDED_BASE64_CHARS�
AB64_CHARS� HASH64_CHARS� BCRYPT_CHARS� HEX_CHARS�LOWER_HEX_CHARS�UPPER_HEX_CHARS�ALL_BYTE_VALUES�compile_byte_translation� ab64_encode� ab64_decode� b64s_encode� b64s_decoder r � Base64Engine�LazyBase64Engine�h64�h64big�bcrypt64z@ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/z@ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789./z@./0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzz@./ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789�=Z0123456789abcdefABCDEFZ0123456789ABCDEFZ0123456789abcdef� � � � =c C sl |du rt dd� }n tt|��}| �� D ]6\}}t|t�rDt|�}t|t�rX|�d�}|||< q*t �
|�S )a�
return a 256-byte string for translating bytes using specified mapping.
bytes not specified by mapping will be left alone.
:param mapping:
dict mapping input byte (str or int) -> output byte (str or int).
:param source:
optional existing byte translation string to use as base.
(must be 255-length byte string). defaults to identity mapping.
:returns:
255-length byte string for passing to bytes().translate.
N�ascii) �_TRANSLATE_SOURCE�listr �items�
isinstancer �ordr �encode�B_EMPTY�join)�mapping�source�target�k�v� r@ �8/usr/lib/python3.9/site-packages/passlib/utils/binary.pyr# n s
r# c C s t | ��t�S )zr
encode using shortened base64 format which omits padding & whitespace.
uses default ``+/`` altchars.
)r �rstrip�
_BASE64_STRIP��datar@ r@ rA r& � s r& c
C s� t | t�r8z| �d�} W n ty6 ttd���Y n0 t| �d@ }|dkrNn,|dkr`| t7 } n|dkrr| t7 } ntd��z
t | �W S t
y� } ztt |���W Y d}~n
d}~0 0 dS )zq
decode from shortened base64 format which omits padding & whitespace.
uses default ``+/`` altchars.
r2 �4string argument should contain only ASCII characters� r � zinvalid base64 inputN) r6 r r8 �UnicodeEncodeErrorr �
ValueError�len� _BASE64_PAD2� _BASE64_PAD1r
�_BinAsciiError� TypeError)rE �off�errr@ r@ rA r' � s"
r' s =
s ==c C s t | ��dd�S )z�
encode using shortened base64 format which omits padding & whitespace.
uses custom ``./`` altchars.
it is primarily used by Passlib's custom pbkdf2 hashes.
� +� .)r& �replacerD r@ r@ rA r$ � s r$ c C sH t | t�r8z| �d�} W n ty6 ttd���Y n0 t| �dd��S )z�
decode from shortened base64 format which omits padding & whitespace.
uses custom ``./`` altchars, but supports decoding normal ``+/`` altchars as well.
it is primarily used by Passlib's custom pbkdf2 hashes.
r2 rF rS rR )r6 r r8 rI r rJ r' rT rD r@ r@ rA r% � s
r% c C s t t| ��t��S )zh
wrapper around :func:`base64.b32encode` which strips padding,
and returns a native string.
)r �
_b32encoderB �B_EQUAL�r< r@ r@ rA r � s r �B�O)�8�0� c C sJ t | t�r| �d�} | �t�} t| �d@ }|r@| td| � 7 } t| d�S )z�
wrapper around :func:`base64.b32decode`
which handles common mistyped chars.
padding optional, ignored if present.
r2 � NT)r6 r r8 � translate�_b32_translaterK �_b32_decode_pad�
_b32decode)r< Z remainderr@ r@ rA r � s
r c @ s e Zd ZdZdZdZdZdZdZdZ d:dd�Z
e dd� �Z dd � Z
d
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zedd� �Zedd� �Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Z d&d'� Z d(d)� Z d*d+� Z d,d-� Z d.d/� Z!d0d1� Z"d2d3� Z#d4d5� Z$d6d7� Z%d8d9� Z&dS );r( ac Provides routines for encoding/decoding base64 data using
arbitrary character mappings, selectable endianness, etc.
:arg charmap:
A string of 64 unique characters,
which will be used to encode successive 6-bit chunks of data.
A character's position within the string should correspond
to its 6-bit value.
:param big:
Whether the encoding should be big-endian (default False).
.. note::
This class does not currently handle base64's padding characters
in any way what so ever.
Raw Bytes <-> Encoded Bytes
===========================
The following methods convert between raw bytes,
and strings encoded using the engine's specific base64 variant:
.. automethod:: encode_bytes
.. automethod:: decode_bytes
.. automethod:: encode_transposed_bytes
.. automethod:: decode_transposed_bytes
..
.. automethod:: check_repair_unused
.. automethod:: repair_unused
Integers <-> Encoded Bytes
==========================
The following methods allow encoding and decoding
unsigned integers to and from the engine's specific base64 variant.
Endianess is determined by the engine's ``big`` constructor keyword.
.. automethod:: encode_int6
.. automethod:: decode_int6
.. automethod:: encode_int12
.. automethod:: decode_int12
.. automethod:: encode_int24
.. automethod:: decode_int24
.. automethod:: encode_int64
.. automethod:: decode_int64
Informational Attributes
========================
.. attribute:: charmap
unicode string containing list of characters used in encoding;
position in string matches 6bit value of character.
.. attribute:: bytemap
bytes version of :attr:`charmap`
.. attribute:: big
boolean flag indicating this using big-endian encoding.
NFc C s� t |t�r|�d�}nt |t�s,t�|d��t|�dkr@td��tt|��dkrXtd��|| _ |j
| _ t dd� t
|�D ��}|j
| _|| _|r�| j| _| j| _n| j| _| j| _d S )N�latin-1�charmap�@ z'charmap must be 64 characters in lengthz-charmap must not contain duplicate charactersc s s | ]\}}||fV qd S �Nr@ )�.0�idx�valuer@ r@ rA � <genexpr>] r/ z(Base64Engine.__init__.<locals>.<genexpr>)r6 r r8 �bytesr ZExpectedStringErrorrK rJ �set�bytemap� __getitem__� _encode64�dict� enumerate� _decode64�big�_encode_bytes_big�
_encode_bytes�_decode_bytes_big�
_decode_bytes�_encode_bytes_little�_decode_bytes_little)�selfrc rr �lookupr@ r@ rA �__init__Q s$
zBase64Engine.__init__c C s | j �d�S )zcharmap as unicoderb )rl �decode)ry r@ r@ rA rc s s zBase64Engine.charmapc C st t |t�s tdt|�f ��tt|�d�\}}tr@tt|��}ntdd� |D ��}| � |||�}t
t | j |��}|S )z�encode bytes to base64 string.
:arg source: byte string to encode.
:returns: byte string containing encoded data.
� source must be bytes, not %srG c s s | ]}t |�V qd S re )r7 )rf �elemr@ r@ rA ri � r/ z,Base64Engine.encode_bytes.<locals>.<genexpr>)
r6 rj rO �type�divmodrK r
r �iterrt r r rn )ry r< �chunks�tail�
next_value�gen�outr@ r@ rA � encode_bytes{ s
zBase64Engine.encode_bytesc c s� d}||k rh|� }|� }|� }|d@ V |d@ d> |d? B V |d@ d> |d? B V |d? V |d7 }q|r�|� }|dkr�|d@ V |d? V n0|� }|d@ V |d@ d> |d? B V |d? V d S )
z>helper used by encode_bytes() to handle little-endian encodingr �? � rH � rG � � Nr@ �ry r� r� r� rg �v1�v2�v3r@ r@ rA rw � s&
z!Base64Engine._encode_bytes_littlec c s� d}||k rh|� }|� }|� }|d? V |d@ d> |d? B V |d@ d> |d? B V |d@ V |d7 }q|r�|� }|dkr�|d? V |d@ d> V n4|� }|d? V |d@ d> |d? B V |d@ d> V d S )
z;helper used by encode_bytes() to handle big-endian encodingr rH rG r� r� r� r� r� Nr@ r� r@ r@ rA rs � s&
z Base64Engine._encode_bytes_bigc
C s� t |t�s tdt|�f ��tt|�d�\}}|dkr>td��tt| j |��}zt
| � |||��W S t y� } z td|j
d f ��W Y d}~n
d}~0 0 dS )z�decode bytes from base64 string.
:arg source: byte string to decode.
:returns: byte string containing decoded data.
r} r� r� z(input string length cannot be == 1 mod 4zinvalid character: %rr N)r6 rj rO r r� rK rJ r r rq r rv �KeyError�args)ry r< r� r� r� rQ r@ r@ rA � decode_bytes� s
zBase64Engine.decode_bytesc c s� d}||k rh|� }|� }|� }|� }||d@ d> B V |d? |d@ d> B V |d? |d> B V |d7 }q|r�|� }|� }||d@ d> B V |dkr�|� }|d? |d@ d> B V dS ) z>helper used by decode_bytes() to handle little-endian encodingr rG r� rH r� r� r� Nr@ � ry r� r� r� rg r� r� r� Zv4r@ r@ rA rx � s"
z!Base64Engine._decode_bytes_littlec c s� d}||k rh|� }|� }|� }|� }|d> |d? B V |d@ d> |d? B V |d@ d> |B V |d7 }q|r�|� }|� }|d> |d? B V |dkr�|� }|d@ d> |d? B V dS ) z;helper used by decode_bytes() to handle big-endian encodingr rH r� r� rG r� r� Nr@ r� r@ r@ rA ru s"
z Base64Engine._decode_bytes_bigc sB t � fdd�t| j�D ��}|�� fdd�t| j�D �� t|�S )z2helper to generate set of valid last chars & bytesc 3 s | ]\}}|� @ s|V qd S re r@ �rf �i�c��bitsr@ rA ri G r/ z-Base64Engine.__make_padset.<locals>.<genexpr>c 3 s | ]\}}|� @ s|V qd S re r@ r� r� r@ rA ri H r/ )rk rp rl �updaterc � frozenset)ry r� Zpsetr@ r� rA Z
__make_padsetE s zBase64Engine.__make_padsetc C s | j r
dnd}| | �|�fS )zDmask to clear padding bits, and valid last bytes (for strings 2 % 4)r� �< �rr �_Base64Engine__make_padset�ry r� r@ r@ rA � _padinfo2K s zBase64Engine._padinfo2c C s | j r
dnd}| | �|�fS )zDmask to clear padding bits, and valid last bytes (for strings 3 % 4)rG �0 r� r� r@ r@ rA � _padinfo3R s zBase64Engine._padinfo3c C s� t |�d@ }|dkr | j\}}n(|dkr4| j\}}n|s@d|fS td��|d }||v r`d|fS t|t�r�| j}||�|�|@ }n"| �| � |�|@ �}t
r�t |g�}d|dd� | fS )a helper to detect & clear invalid unused bits in last character.
:arg source:
encoded data (as ascii bytes or unicode).
:returns:
`(True, result)` if the string was repaired,
`(False, source)` if the string was ok as-is.
rG rH Fz source length must != 1 mod 4���TN) rK r� r� rJ r6 r rc �indexrn rq r
rj )ry r< r� �maskZpadsetZlast�cmr@ r@ rA �check_repair_unusedY s&
z Base64Engine.check_repair_unusedc C s | � |�d S )Nr� )r� �ry r< r@ r@ rA �
repair_unused� s zBase64Engine.repair_unusedc s< t � t�s tdt� �f ��t� fdd�|D ��}| �|�S )z>encode byte string, first transposing source using offset listr} c 3 s | ]}� | V qd S re r@ �rf rP rW r@ rA ri � r/ z7Base64Engine.encode_transposed_bytes.<locals>.<genexpr>)r6 rj rO r r r� )ry r< �offsets�tmpr@ rW rA �encode_transposed_bytes� s
z$Base64Engine.encode_transposed_bytesc C s< | � |�}dgt|� }t||�D ]\}}|||< q"t|�S )zGdecode byte string, then reverse transposition described by offset listN)r� rK �zipr )ry r< r� r� �bufrP �charr@ r@ rA �decode_transposed_bytes� s
z$Base64Engine.decode_transposed_bytesc C s� t |t�s tdt|�f ��| j}| d }|| d }t|�|krRtd|f ��| j}d}z*|rf|nt|�D ]}|d> ||� }qnW n t y� td|f ��Y n0 |r�|r�||L }n|d|> d M }|S )a� decode base64 string -> integer
:arg source: base64 string to decode.
:arg bits: number of bits in resulting integer.
:raises ValueError:
* if the string contains invalid base64 characters.
* if the string is not long enough - it must be at least
``int(ceil(bits/6))`` in length.
:returns:
a integer in the range ``0 <= n < 2**bits``
r} r� zsource must be %d charsr z invalid character in string: %rr� )
r6 rj rO r rr rK rJ rq �reversedr� ) ry r< r� rr �pad�charsr| r� r� r@ r@ rA � _decode_int� s&
zBase64Engine._decode_intc C sh t |t�s tdt|�f ��t|�dkr0td��tr<|d }z | �|�W S tyb td��Y n0 dS )z(decode single character -> 6 bit integerr} r� z source must be exactly 1 byter �invalid characterN) r6 rj rO r rK rJ r
rq r� r� r@ r@ rA � decode_int6� s
zBase64Engine.decode_int6c C s� t |t�s tdt|�f ��t|�dkr0td��| j}zF| jr\||d �||d �d> W S ||d �||d �d> W S W n ty� td��Y n0 dS ) z'decodes 2 char string -> 12-bit integerr} rH z source must be exactly 2 bytesr� r r� r� N� r6 rj rO r rK rJ rq rr r� �ry r< r| r@ r@ rA � decode_int12� s
" zBase64Engine.decode_int12c C s� t |t�s tdt|�f ��t|�dkr0td��| j}z�| jr|||d �||d �d> ||d �d> ||d �d
> W S ||d �||d �d> ||d �d> ||d �d
> W S W n ty� td ��Y n0 d S )
z'decodes 4 char string -> 24-bit integerr} r� z source must be exactly 4 bytesrG rH r� r� � r � r� Nr� r� r@ r@ rA � decode_int24� s&
����
zBase64Engine.decode_int24c C s | � |d�S )�&decode 5 char string -> 30 bit integer� �r� r� r@ r@ rA � decode_int30� s zBase64Engine.decode_int30c C s | � |d�S )z�decode 11 char base64 string -> 64-bit integer
this format is used primarily by des-crypt & variants to encode
the DES output value used as a checksum.
rd r� r� r@ r@ rA � decode_int64 s zBase64Engine.decode_int64c s\ | d }||7 }| j r2t|d dd�}� |K � n td|d�}tt| j� fdd�|D ���S )z�encode integer into base64 format
:arg value: non-negative integer to encode
:arg bits: number of bits to encode
:returns:
a string of length ``int(ceil(bits/6.0))``.
r� i����r c 3 s | ]}� |? d @ V qdS )r� Nr@ r� �rh r@ rA ri " r/ z+Base64Engine._encode_int.<locals>.<genexpr>)rr r r r rn )ry rh r� r� �itrr@ r� rA � _encode_int s
�zBase64Engine._encode_intc C s< |dk s|dkrt d��tr.| j||d � S | �|�S dS )z0encodes 6-bit integer -> single hash64 characterr r� �value out of ranger� N)rJ r
rl rn �ry rh r@ r@ rA � encode_int6( s
zBase64Engine.encode_int6c C sJ |dk s|dkrt d��|d@ |d? d@ g}| jr:t|�}tt| j|��S )z'encodes 12-bit integer -> 2 char stringr i� r� r� r� �rJ rr r� r r rn �ry rh �rawr@ r@ rA � encode_int121 s zBase64Engine.encode_int12c C s^ |dk s|dkrt d��|d@ |d? d@ |d? d@ |d? d@ g}| jrNt|�}tt| j|��S )z'encodes 24-bit integer -> 4 char stringr i��� r� r� r� r� r� r� r� r@ r@ rA � encode_int24: s �zBase64Engine.encode_int24c C s$ |dk s|dkrt d��| �|d�S )r� r i���?r� r� �rJ r� r� r@ r@ rA � encode_int30D s zBase64Engine.encode_int30c C s$ |dk s|dkrt d��| �|d�S )z�encode 64-bit integer -> 11 char hash64 string
this format is used primarily by des-crypt & variants to encode
the DES output value used as a checksum.
r l ���� r� rd r� r� r@ r@ rA � encode_int64J s zBase64Engine.encode_int64)F)'�__name__�
__module__� __qualname__�__doc__rl rr rn rq rt rv r{ �propertyrc r� rw rs r� rx ru r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r@ r@ r@ rA r( � sJ D
"
'+#*
)(
r( c s4 e Zd ZdZdZdd� Z� fdd�Zdd� Z� ZS ) r) z<Base64Engine which delays initialization until it's accessedNc O s ||f| _ d S re )�
_lazy_opts�ry r� �kwdsr@ r@ rA r{ \ s zLazyBase64Engine.__init__c s. | j \}}tt| �j|i |�� | ` t| _d S re )r� �superr) r{ r( � __class__r� �r� r@ rA �
_lazy_init_ s
zLazyBase64Engine._lazy_initc C s |� d�s| �� t�| |�S )N�_)�
startswithr� �object�__getattribute__)ry �attrr@ r@ rA r� e s
z!LazyBase64Engine.__getattribute__) r� r� r� r� r� r{ r� r� �
__classcell__r@ r@ r� rA r) X s
r) T)rr )N)Er� Z
__future__r r r �base64r r r ra r rU Zbinasciir r
r rN ZloggingZ getLoggerr� �logZpasslibr Zpasslib.utils.compatr
r r r r r r r r r r r Zpasslib.utils.decorr �__all__r r r r r r r! r r" r9 ZB_NULLrV r4 r3 r# r& r' rC rM rL r$ r% r_ r` r� r( r) r* r+ r, r@ r@ r@ rA �<module> sV
8 %
! _
Mini Shell