관리-도구
편집 파일: binary.cpython-39.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 s� |du rt dd� }n&t|t�r,t|�dks0J �tt|��}| �� D ]t\}}t|t�r^t|�}t|t �r~d| kr|dk s�n J �t|t �r�|� d�}t|t�r�t|�dks�J �|||< qDt � |�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� r r. �ascii� )�_TRANSLATE_SOURCE� isinstance�bytes�len�listr �itemsr �ord�intr �encode�B_EMPTY�join)�mapping�source�target�k�v� rE �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��datarE rE rF 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. r3 �4string argument should contain only ASCII characters� r � zinvalid base64 inputN) r6 r r= �UnicodeEncodeErrorr � ValueErrorr8 � _BASE64_PAD2� _BASE64_PAD1r �_BinAsciiError� TypeError)rJ �off�errrE rE rF 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& �replacerI rE rE rF 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. r3 rK rW rV )r6 r r= rN r rO r' rX rI rE rE rF 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 � _b32encoderG �B_EQUAL�rA rE rE rF 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. r3 � NT)r6 r r= � translate�_b32_translater8 �_b32_decode_pad� _b32decode)rA Z remainderrE rE rF 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 �NrE )�.0�idx�valuerE rE rF � <genexpr>] r/ z(Base64Engine.__init__.<locals>.<genexpr>)r6 r r= r7 r ZExpectedStringErrorr8 rO �set�bytemap� __getitem__� _encode64�dict� enumerate� _decode64�big�_encode_bytes_big� _encode_bytes�_decode_bytes_big� _decode_bytes�_encode_bytes_little�_decode_bytes_little)�selfrg ru �lookuprE rE rF �__init__Q s$ zBase64Engine.__init__c C s | j �d�S )zcharmap as unicoderf )ro �decode)r| rE rE rF rg 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 %srL c s s | ]}t |�V qd S ri )r; )rj �elemrE rE rF rm � r/ z,Base64Engine.encode_bytes.<locals>.<genexpr>) r6 r7 rS �type�divmodr8 r r �iterrw r r rq )r| rA �chunks�tail� next_value�gen�outrE rE rF � 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 n<|dks�J �|� }|d@ V |d@ d> |d? B V |d? V d S ) z>helper used by encode_bytes() to handle little-endian encodingr �? � rM � rL � r4 NrE �r| r� r� r� rk �v1�v2�v3rE rE rF rz � 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 n@|dks�J �|� }|d? V |d@ d> |d? B V |d@ d> V d S ) z;helper used by encode_bytes() to handle big-endian encodingr rM rL r� r� r� r� r4 NrE r� rE rE rF rv � 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� r4 z(input string length cannot be == 1 mod 4zinvalid character: %rr N)r6 r7 rS r� r� r8 rO r r rt r ry �KeyError�args)r| rA r� r� r� rU rE rE rF � 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 rL r� rM r� r� r4 NrE � r| r� r� r� rk r� r� r� Zv4rE rE rF r{ � 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 rM r� r� rL r� r4 NrE r� rE rE rF rx 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 ri rE �rj �i�c��bitsrE rF rm G r/ z-Base64Engine.__make_padset.<locals>.<genexpr>c 3 s | ]\}}|� @ s|V qd S ri rE r� r� rE rF rm H r/ )rn rs ro �updaterg � frozenset)r| r� ZpsetrE r� rF 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� �< �ru �_Base64Engine__make_padset�r| r� rE rE rF � _padinfo2K s zBase64Engine._padinfo2c C s | j r dnd}| | �|�fS )zDmask to clear padding bits, and valid last bytes (for strings 3 % 4)rL �0 r� r� rE rE rF � _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}||�|�|@ }||v s�J d��n2| �| � |�|@ �}||v s�J d��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. rL rM Fz source length must != 1 mod 4���z%failed to generate valid padding charTN) r8 r� r� rO r6 r rg �indexrq rt r r7 )r| rA r� �maskZpadsetZlast�cmrE rE rF �check_repair_unusedY s( z Base64Engine.check_repair_unusedc C s | � |�d S )Nr4 )r� �r| rA rE rE rF � 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 ri rE �rj rT r[ rE rF rm � r/ z7Base64Engine.encode_transposed_bytes.<locals>.<genexpr>)r6 r7 rS r� r r� )r| rA �offsets�tmprE r[ rF �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� r8 �zipr )r| rA r� r� �bufrT �charrE rE rF �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: %rr4 ) r6 r7 rS r� ru r8 rO rt �reversedr� ) r| rA r� ru �pad�charsr r� r� rE rE rF � _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� r4 z source must be exactly 1 byter �invalid characterN) r6 r7 rS r� r8 rO r rt r� r� rE rE rF � 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� rM z source must be exactly 2 bytesr4 r r� r� N� r6 r7 rS r� r8 rO rt ru r� �r| rA r rE rE rF � 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 bytesrL rM r� r4 � r � r� Nr� r� rE rE rF � decode_int24� s& ���� zBase64Engine.decode_int24c C s | � |d�S )�&decode 5 char string -> 30 bit integer� �r� r� rE rE rF � 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. rh r� r� rE rE rF � decode_int64 s zBase64Engine.decode_int64c sl � dksJ d��| d }||7 }| j rBt|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 z caller did not sanitize inputr� i����c 3 s | ]}� |? d @ V qdS )r� NrE r� �rl rE rF rm " r/ z+Base64Engine._encode_int.<locals>.<genexpr>)ru r r r rq )r| rl r� r� �itrrE r� rF � _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 ranger4 N)rO r ro rq �r| rl rE rE rF � 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� �rO ru r� r r rq �r| rl �rawrE rE rF � 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� rE rE rF � encode_int24: s �zBase64Engine.encode_int24c C s$ |dk s|dkrt d��| �|d�S )r� r i���?r� r� �rO r� r� rE rE rF � 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� rh r� r� rE rE rF � encode_int64J s zBase64Engine.encode_int64)F)'�__name__� __module__� __qualname__�__doc__ro ru rq rt rw ry r~ �propertyrg r� rz rv r� r{ rx r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rE rE rE rF 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 ri )� _lazy_opts�r| r� �kwdsrE rE rF r~ \ s zLazyBase64Engine.__init__c s. | j \}}tt| �j|i |�� | ` t| _d S ri )r� �superr) r~ r( � __class__r� �r� rE rF � _lazy_init_ s zLazyBase64Engine._lazy_initc C s |� d�s| �� t�| |�S )N�_)� startswithr� �object�__getattribute__)r| �attrrE rE rF r� e s z!LazyBase64Engine.__getattribute__) r� r� r� r� r� r~ r� r� � __classcell__rE rE r� rF r) X s r) T)ru )N)Er� Z __future__r r r �base64r r r re r rY Zbinasciir r r rR 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" r> ZB_NULLrZ r9 r5 r# r& r' rH rQ rP r$ r% rc rd r� r( r) r* r+ r, rE rE rE rF �<module> sV 8 % ! _