U hz@sdZddlmZmZmZddlmZmZmZ m Z ddl m Z mZmZddlZeeZddlmZddlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z%d d d d d ddddddddddddddddgZ&e!dZ'e!dZ(e!d Z)e!d!Z*e'e!d"Z+e!d#Z,e!d$Z-e!d%Z.eed&Z/d'Z0d(Z1d)Z2e3ee/Z4d;d*dZ5d+dZ6d,dZ7d-Z8d)Z9d.Z:d/dZ;d0dZd6dZGd7dde?Z@Gd8dde@ZAeAe)ZBeAe)d9d:ZCeAe*d9d:ZDdS) 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. Nrr.ascii)_TRANSLATE_SOURCE isinstancebyteslenAssertionErrorlistritemsrordintrencodeB_EMPTYjoin)mappingsourcetargetkvrFX/var/www/html/timesheet/backend/venv/lib/python3.8/site-packages/passlib/utils/binary.pyr#ns  $   cCst|tS)zr encode using shortened base64 format which omits padding & whitespace. uses default ``+/`` altchars. )r rstrip _BASE64_STRIPdatarFrFrGr&sc Cst|tr:z|d}Wn tk r8ttdYnXt|d@}|dkrPn,|dkrb|t7}n|dkrt|t7}ntdz t |WSt k r}ztt |W5d}~XYnXdS)zq decode from shortened base64 format which omits padding & whitespace. uses default ``+/`` altchars. r34string argument should contain only ASCII charactersrzinvalid base64 inputN) r6rr>UnicodeEncodeErrorr ValueErrorr8 _BASE64_PAD2 _BASE64_PAD1r _BinAsciiError TypeError)rKofferrrFrFrGr's"     s= s==cCst|ddS)z encode using shortened base64 format which omits padding & whitespace. uses custom ``./`` altchars. it is primarily used by Passlib's custom pbkdf2 hashes. +.)r&replacerJrFrFrGr$scCsJt|tr:z|d}Wn tk r8ttdYnXt|ddS)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. r3rLrXrW)r6rr>rOrrPr'rYrJrFrFrGr%s  cCstt|tS)zh wrapper around :func:`base64.b32encode` which strips padding, and returns a native string. )r _b32encoderHB_EQUALrBrFrFrGrsBO)80cCsJt|tr|d}|t}t|d@}|r@|td| 7}t|dS)z wrapper around :func:`base64.b32decode` which handles common mistyped chars. padding optional, ignored if present. r3NT)r6rr> translate_b32_translater8_b32_decode_pad _b32decode)rB remainderrFrFrGrs    c@seZdZdZdZdZdZdZdZdZ d:ddZ e ddZ dd Z d d Zd d ZddZddZddZddZeddZeddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Z d.d/Z!d0d1Z"d2d3Z#d4d5Z$d6d7Z%d8d9Z&dS);r(acProvides 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. NFcCst|tr|d}nt|ts,t|dt|dkr@tdtt|dkrXtd||_ |j |_ t ddt |D}|j |_||_|r|j|_|j|_n|j|_|j|_dS)Nlatin-1charmap@z'charmap must be 64 characters in lengthz-charmap must not contain duplicate characterscss|]\}}||fVqdSNrF).0idxvaluerFrFrG ]sz(Base64Engine.__init__..)r6rr>r7r ExpectedStringErrorr8rPsetbytemap __getitem__ _encode64dict enumerate _decode64big_encode_bytes_big _encode_bytes_decode_bytes_big _decode_bytes_encode_bytes_little_decode_bytes_little)selfrirxlookuprFrFrG__init__Qs$      zBase64Engine.__init__cCs |jdS)zcharmap as unicoderh)rrdecode)rrFrFrGrisszBase64Engine.charmapcCstt|tstdt|ftt|d\}}tr@tt|}ntdd|D}| |||}t t |j |}|S)zencode bytes to base64 string. :arg source: byte string to encode. :returns: byte string containing encoded data. source must be bytes, not %srMcss|]}t|VqdSrk)r<)rlelemrFrFrGrosz,Base64Engine.encode_bytes..) r6r7rTtypedivmodr8r riterrzrrrt)rrBchunkstail next_valuegenoutrFrFrG encode_bytes{s zBase64Engine.encode_bytesccsd}||krh|}|}|}|d@V|d@d>|d?BV|d@d>|d?BV|d?V|d7}q|r|}|dkr|d@V|d?Vn<|dkst|}|d@V|d@d>|d?BV|d?Vd S) z>helper used by encode_bytes() to handle little-endian encodingr?rNrMr4Nr9rrrrrmv1v2v3rFrFrGr}s(       z!Base64Engine._encode_bytes_littleccsd}||krh|}|}|}|d?V|d@d>|d?BV|d@d>|d?BV|d@V|d7}q|r|}|dkr|d?V|d@d>Vn@|dkst|}|d?V|d@d>|d?BV|d@d>Vd S) z;helper used by encode_bytes() to handle big-endian encodingrrNrMrrrrr4NrrrFrFrGrys(      zBase64Engine._encode_bytes_bigc Cst|tstdt|ftt|d\}}|dkr>tdtt|j |}zt | |||WSt k r}ztd|j dfW5d}~XYnXdS)zdecode bytes from base64 string. :arg source: byte string to decode. :returns: byte string containing decoded data. rrr4z(input string length cannot be == 1 mod 4zinvalid character: %rrN)r6r7rTrrr8rPrrrwrr|KeyErrorargs)rrBrrrrVrFrFrG decode_bytess zBase64Engine.decode_bytesc csd}||krh|}|}|}|}||d@d>BV|d?|d@d>BV|d?|d>BV|d7}q|r|}|}||d@d>BV|dkr|}|d?|d@d>BVdS) z>helper used by decode_bytes() to handle little-endian encodingrrMrrNrrr4NrF rrrrrmrrrZv4rFrFrGr~s" z!Base64Engine._decode_bytes_littlec csd}||krh|}|}|}|}|d>|d?BV|d@d>|d?BV|d@d>|BV|d7}q|r|}|}|d>|d?BV|dkr|}|d@d>|d?BVdS) z;helper used by decode_bytes() to handle big-endian encodingrrNrrrMrr4NrFrrFrFrGr{s" zBase64Engine._decode_bytes_bigcsBtfddt|jD}|fddt|jDt|S)z2helper to generate set of valid last chars & bytesc3s|]\}}|@s|VqdSrkrFrlicbitsrFrGroGsz-Base64Engine.__make_padset..c3s|]\}}|@s|VqdSrkrFrrrFrGroHs)rqrvrrupdateri frozenset)rrZpsetrFrrGZ __make_padsetEszBase64Engine.__make_padsetcCs|jr dnd}|||fS)zDmask to clear padding bits, and valid last bytes (for strings 2 % 4)r<rx_Base64Engine__make_padsetrrrFrFrG _padinfo2KszBase64Engine._padinfo2cCs|jr dnd}|||fS)zDmask to clear padding bits, and valid last bytes (for strings 3 % 4)rM0rrrFrFrG _padinfo3RszBase64Engine._padinfo3cCst|d@}|dkr |j\}}n(|dkr4|j\}}n|s@d|fStd|d}||kr`d|fSt|tr|j}||||@}||kstdn2| | ||@}||kstdt rt |g}d|dd|fS) ahelper 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. rMrNFzsource length must != 1 mod 4z%failed to generate valid padding charTN) r8rrrPr6rriindexr9rtrwr r7)rrBrmaskZpadsetlastcmrFrFrGcheck_repair_unusedYs(     z Base64Engine.check_repair_unusedcCs||dS)Nr4)rrrBrFrFrG repair_unusedszBase64Engine.repair_unusedcs<ttstdtftfdd|D}||S)z>encode byte string, first transposing source using offset listrc3s|]}|VqdSrkrFrlrUr\rFrGrosz7Base64Engine.encode_transposed_bytes..)r6r7rTrrr)rrBoffsetstmprFr\rGencode_transposed_bytess z$Base64Engine.encode_transposed_bytescCs<||}dgt|}t||D]\}}|||<q"t|S)zGdecode byte string, then reverse transposition described by offset listN)rr8zipr)rrBrrbufrUcharrFrFrGdecode_transposed_bytess   z$Base64Engine.decode_transposed_bytesc Cst|tstdt|f|j}| d}||d}t||krRtd|f|j}d}z*|rf|nt|D]}|d>||}qnWn"t k rtd|fYnX|r|r||L}n|d|>dM}|S)adecode 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`` rrzsource must be %d charsrzinvalid character in string: %rr4) r6r7rTrrxr8rPrwreversedr) rrBrrxpadcharsrrrrFrFrG _decode_ints&     zBase64Engine._decode_intcCsjt|tstdt|ft|dkr0tdtr<|d}z ||WStk rdtdYnXdS)z(decode single character -> 6 bit integerrr4zsource must be exactly 1 byterinvalid characterN) r6r7rTrr8rPr rwrrrFrFrG decode_int6s   zBase64Engine.decode_int6cCst|tstdt|ft|dkr0td|j}zF|jr\||d||dd>WS||d||dd>WSWntk rtdYnXdS) z'decodes 2 char string -> 12-bit integerrrNzsource must be exactly 2 bytesr4rrrN r6r7rTrr8rPrwrxrrrBrrFrFrG decode_int12s  "zBase64Engine.decode_int12cCst|tstdt|ft|dkr0td|j}z|jr|||d||dd>||dd>||d d >WS||d ||dd>||dd>||dd >WSWntk rtd YnXd S) z'decodes 4 char string -> 24-bit integerrrzsource must be exactly 4 bytesrMrNrr4 rrNrrrFrFrG decode_int24s&   zBase64Engine.decode_int24cCs ||dS)&decode 5 char string -> 30 bit integerrrrFrFrG decode_int30szBase64Engine.decode_int30cCs ||dS)zdecode 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. rjrrrFrFrG decode_int64szBase64Engine.decode_int64csldkstd| d}||7}|jrBt|ddd}|Kn td|d}tt|jfdd|DS)zencode 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))``. rzcaller did not sanitize inputric3s|]}|?d@VqdS)rNrFrrnrFrGro"sz+Base64Engine._encode_int..)r9rxrrrrt)rrnrritrrFrrG _encode_ints    zBase64Engine._encode_intcCs<|dks|dkrtdtr.|j||dS||SdS)z0encodes 6-bit integer -> single hash64 characterrrvalue out of ranger4N)rPr rrrtrrnrFrFrG encode_int6(s zBase64Engine.encode_int6cCsJ|dks|dkrtd|d@|d?d@g}|jr:t|}tt|j|S)z'encodes 12-bit integer -> 2 char stringrirrrrPrxrrrrtrrnrawrFrFrG encode_int121s zBase64Engine.encode_int12cCs^|dks|dkrtd|d@|d?d@|d?d@|d?d@g}|jrNt|}tt|j|S)z'encodes 24-bit integer -> 4 char stringrirrrrrrrrFrFrG encode_int24:s zBase64Engine.encode_int24cCs$|dks|dkrtd||dS)rri?rrrPrrrFrFrG encode_int30DszBase64Engine.encode_int30cCs$|dks|dkrtd||dS)zencode 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. rlrrjrrrFrFrG encode_int64JszBase64Engine.encode_int64)F)'__name__ __module__ __qualname____doc__rrrxrtrwrzr|rpropertyrirr}ryrr~r{rrrrrrrrrrrrrrrrrrrrrFrFrFrGr(sJD " '+#*  )(    cs4eZdZdZdZddZfddZddZZS) r)zs  8 %    !  _