U
|
£ý°dñM ã @ sr U d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
|
Z
|
ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZ ddlmZm Z m Z m Z m Z m Z m!Z! ddl"m#Z# dd l$m%Z% d
|
Z&e'e(d < e)e*ee#fZ+e e e) e e* e e e e# f e(d < e!e)e,f e)d ddZ-e)e.dddZ/e)e.dddZ0e)e.dddZ1e)e.dddZ2e)e.dddZ3e)e.ddd Z4e)e,dd d Z5e,e)e e,e,f d d d!Z6e,e e,e e, f d d"d#Z7e,e e e e, d$f d d%d&Z8dOe,e e, e e) e e,e,f d'd(d)Z9e,e)e e,e)f d d*d+Z:e,e e,e e) f d d,d-Z;e%e<d d.d/Z=dPe%e e< e)dd1d2d3Z>e e<e<f d4d5d6Z?e e<e<f d4d7d8Z@e e e<e<f d4d9d:ZAe e<e<e<f d4d;d<ZBe<e<d=d>d?ZCe<e e<ee<ge'f f d=d@dAZDe<e e e<e<f dBdCdDZEe<e<dEdFdGZFdQe!e)e%f e<e.dJdKdLZGe
|
e e<e!e<e e<e<f f f d4dMdNZHdS )Rz
|
Utilities.é N)ÚDecimal)Ú lru_cache)Ú in_table_a1Ú in_table_b1Ú in_table_c3Ú in_table_c4Ú in_table_c5Ú in_table_c6Ú in_table_c7Ú in_table_c8Ú in_table_c9Ú in_table_c11Ú in_table_c12Úin_table_c21_c22Ú in_table_d1Ú in_table_d2)ÚCallableÚDictÚListÚOptionalÚTupleÚTypeÚUnioné )Ú
|
HexLiteral)Ú
|
StrOrBytesFÚ__MYSQL_DEBUG__Ú NUMERIC_TYPES)ÚbufÚreturnc C sv t | tr| S t| }|dkr&| d S |dkrR| dd| }tt d|¡d S | dd| }tt d|¡d S )z&Unpacks the given buffer to an integerr r é ó ú<Ié ú<Q)Ú
|
isinstanceÚintÚlenÚstructÚunpack)r ÚlengthÚtmp© r, úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\mysql/connector/utils.pyÚintreadG s
|
r. )Úir c C s( | dk s| dkrt dtt d| ¡S )z^
|
Takes an unsigned byte (1 byte) and packs it as a bytes-object.
|
|
Returns string.
|
r éÿ z int1store requires 0 <= i <= 255ú<B©Ú
|
ValueErrorÚ bytearrayr( Úpack©r/ r, r, r- Ú int1storeU s r7 c C s( | dk s| dkrt dtt d| ¡S )z`
|
Takes an unsigned short (2 bytes) and packs it as a bytes-object.
|
|
Returns string.
|
r éÿÿ z"int2store requires 0 <= i <= 65535ú<Hr2 r6 r, r, r- Ú int2store` s r: c C s0 | dk s| dkrt dtt d| ¡dd
S )zb
|
Takes an unsigned integer (3 bytes) and packs it as a bytes-object.
|
|
Returns string.
|
r éÿÿÿ z%int3store requires 0 <= i <= 16777215r" é r2 r6 r, r, r- Ú int3storek s r= c C s( | dk s| dkrt dtt d| ¡S )zb
|
Takes an unsigned integer (4 bytes) and packs it as a bytes-object.
|
|
Returns string.
|
r ì ÿÿ z'int4store requires 0 <= i <= 4294967295r" r2 r6 r, r, r- Ú int4storev s r? c C s( | dk s| dkrt dtt d| ¡S )zZ
|
Takes an unsigned integer (8 bytes) and packs it as string.
|
|
Returns string.
|
r ì z!int8store requires 0 <= i <= 2^64r$ r2 r6 r, r, r- Ú int8store s rA c C s\ | dk s| dkrt d| dkr&t}n.| dkr4t}n | dkrBt}n| dkrPt}nt}|| S )zÍ
|
Takes an unsigned integers and packs it as a bytes-object.
|
|
This function uses int1store, int2store, int3store,
|
int4store or int8store depending on the integer value.
|
|
returns string.
|
r r@ z!intstore requires 0 <= i <= 2^64r0 r8 r; r> )r3 r7 r: r= r? rA )r/ Z formed_stringr, r, r- Úintstore s rB c C s | dk s| dkrt d| dk r0tt d| ¡S | dkrLdtt d| ¡ S | d krpd
|
tt d | ¡dd
S d tt d| ¡ S )z|
|
Takes an unsigned integer and packs it as bytes,
|
with the information of how much bytes the encoded int takes.
|
r r@ zRequires 0 <= i <= 2^64éû r1 r8 ó ür9 r; ó ýr" r< ó þr$ r2 r6 r, r, r- Úlc_int¦ s rG )r Úsizer c C s | d|
}| |d
|fS )zi
|
Reads bytes from a buffer.
|
|
Returns a tuple with buffer less the read bytes, and the bytes.
|
r Nr, ©r rH Úresr, r, r- Ú
|
read_bytes¸ s rK c C sÀ | d dkr | dd
dfS d }}| d }|dkrX|}| d| d
| d|d
fS |dkrfd}n|dkrtd }n |d
|
krd }t | d|d
}| || d d
| |d || d
fS ) u
|
Takes a buffer and reads a length coded string from the start.
|
|
This is how Length coded strings work
|
|
If the string is 250 bytes long or smaller, then it looks like this:
|
|
<-- 1b -->
|
+----------+-------------------------
|
| length | a string goes here
|
+----------+-------------------------
|
|
If the string is bigger than 250, then it looks like this:
|
|
<- 1b -><- 2/3/8 ->
|
+------+-----------+-------------------------
|
| type | length | a string goes here
|
+------+-----------+-------------------------
|
|
if type == ü:
|
length is code in next 2 bytes
|
elif type == ý:
|
length is code in next 3 bytes
|
elif type == þ:
|
length is code in next 8 bytes
|
|
NULL has a special value. If the buffer starts with û then
|
it's a NULL and we return None as value.
|
|
Returns a tuple (trucated buffer, bytes).
|
r rC r Néú éü é éý r< éþ r# ©r. )r r* ÚlsizeZfstr, r, r- Úread_lc_string s rS .c C s g }dddd}t | }d}||k r
|
| | }|dkr:dS |dkrV| d¡ |d 7 }q |d
|
kr|}| | |d ||d
¡ |d | 7 }q d}z || }W n tk
|
r´ Y dS X t| |d ||d
}| | |d | || |d
¡ |d | | 7 }q t|S ) zXReads all length encoded strings from the given buffer
|
|
Returns a list of bytes
|
rN r< r# )rM rO rP r r0 NrC r rL )r' ÚappendÚKeyErrorr. Útuple)r ZbyteslstÚsizesZbuf_lenÚposÚfirstr* rR r, r, r- Úread_lc_string_list÷ s0
|
|
|
&rZ )r ÚendrH r c
|
C s |dkr|dkrt d|dk rxz| |¡}W n, t k
|
rZ } zt d|W 5 d}~X Y nX | |d d
| d|
fS |dk rt| |S t ddS )zr
|
Reads a string up until a character or for a given size.
|
|
Returns a tuple (trucated buffer, string).
|
Nz&read_string() needs either end or sizez end byte not present in bufferr r z.read_string() needs either end or size (weird))r3 ÚindexrK )r r[ rH ÚidxÚerrr, r, r- Ú read_string s
|
|
r_ c C s t | d|
}| |d
|fS )zMRead an integer from buffer
|
|
Returns a tuple (truncated buffer, int)
|
r NrQ rI r, r, r- Úread_int8 s r` c C sÜ | s t d| d }|dkr,| dd
dfS |dk rH| dd
t|fS |dkrt| dd
t d| dd
¡d fS |d kr¤| d
|
d
t d | dd
|
d ¡d fS |d krÐ| dd
t d| dd
¡d fS t ddS )z
|
Takes a buffer and reads an length code string from the start.
|
|
Returns a tuple with buffer less the integer and the integer read.
|
z Empty buffer.r rC r NrM r< z<xHrO r r" r! rP é z<xQz%Failed reading length encoded integer)r3 r& r( r) )r Zlcbyter, r, r- Ú read_lc_intA s $($rb c C s2 t | ts d dd | D ¡S d dd | D ¡S )z"Debug function for showing buffersÚ c S s g | ]}d |dqS ©z\xZ02xr, ©Ú.0Úcr, r, r- Ú
|
<listcomp>^ s z"_digest_buffer.<locals>.<listcomp>c S s g | ]}d t |dqS rd )Úordre r, r, r- rh _ s )r% ÚstrÚjoin)r r, r, r- Ú_digest_buffer[ s
|
rl é )ÚabufferÚprefixÚlimitr c C sL |r<|r"|dkr"t | d|
}nt | }t|d | n tt | dS )z2Debug function printing output of _digest_buffer()r z: N)rl Úprint)rn ro rp Údigestr, r, r- Ú print_bufferb s rs )r c C s i } t j dd¡}t j |¡s"| S t|ddP}|D ]D}| d¡}t|dkrPq4|d ¡ }|d d ¡ d
|
¡}|| |< q4W 5 Q R X | S ) zsParse the contents of /etc/os-release file.
|
|
Returns:
|
A dictionary containing release information.
|
ú/etcz
|
os-releaseúutf-8©Úencodingú=rN r r Ú
|
ú"©
|
ÚosÚpathrk ÚexistsÚopenÚsplitr' ÚlowerÚrstripÚstrip)ÚdistroZos_release_fileÚfile_objÚlineÚ key_valueÚkeyÚvaluer, r, r- Ú_parse_os_releasep s
|
r c C s i } t j dd¡}t j |¡rt|ddP}|D ]D}| d¡}t|dkrLq0|d ¡ }|d d ¡ d
|
¡}|| |< q0W 5 Q R X | S ) ztParse the contents of /etc/lsb-release file.
|
|
Returns:
|
A dictionary containing release information.
|
rt z lsb-releaseru rv rx rN r r ry rz r{ )r Zlsb_release_filer
r r r r r, r, r- Ú_parse_lsb_release
s
|
r c C s¶ i } t tjddd}ztjd|d}W n tk
|
rH Y W 5 Q R £ dS X | t ¡ ¡ ¡ }|D ]F}|
|
d¡}t |dkr|q`|d d
|
d ¡ ¡ }|d d ¡}|| |< q`W 5 Q R X | S )ztParse the output of the lsb_release command.
|
|
Returns:
|
A dictionary containing release information.
|
Úwru rv )Z lsb_releasez-a)ÚstderrNú:rN r ú Ú_r ú )r r| ÚdevnullÚ
|
subprocessÚ check_outputÚOSErrorÚdecodeÚsysÚgetfilesystemencodingÚ
|
splitlinesr r' Úreplacer r )r r ÚstdoutÚlinesr r r r r, r, r- Ú_parse_lsb_release_command s
|
r c C s t } | r,| dd¡| dd¡| dd¡fS t } | rX| dd¡| dd¡| dd¡fS t } | r| dd¡| d d¡| d
|
d¡fS d S ) aN Tries to determine the name of the Linux OS distribution name.
|
|
First tries to get information from ``/etc/os-release`` file.
|
If fails, tries to get the information of ``/etc/lsb-release`` file.
|
And finally the information of ``lsb-release`` command.
|
|
Returns:
|
A tuple with (`name`, `version`, `codename`)
|
Z
|
distrib_idrc Zdistrib_releaseZdistrib_codenameZdistributor_idÚreleaseZcodenameÚnameZ
|
version_idZversion_codename)rc rc rc )r Úgetr r )r r, r, r- Úlinux_distribution° s&
|
|
|
|
ý
|
|
|
ý
|
|
|
ýr¡ )Ú unicode_strr c C s | rt | d ¡dkrdS dS )zÈGet the readiness direction of the unicode string.
|
|
We assume that the direction is "L-to-R" if the first character does not
|
indicate the direction is "R-to-L" or an "AL" (Arabic Letter).
|
r )ÚRÚALúR-to-LzL-to-R)Ú unicodedataÚ bidirectional)r¢ r, r, r- Ú_get_unicode_read_directionÕ s r¨ c C s@ t | }|dkr8t| d r(t| d s0tddtiS dtiS )at
|
1) The characters in section 5.8 MUST be prohibited.
|
|
2) If a string contains any RandALCat character, the string MUST NOT
|
contain any LCat character.
|
|
3) If a string contains any RandALCat character, a RandALCat
|
character MUST be the first character of the string, and a
|
RandALCat character MUST be the last character of the string.
|
r¥ r éÿÿÿÿzvInvalid unicode Bidirectional sequence, if the first character is RandALCat, the final charactermust be RandALCat too.z7Bidirectional Characters requirement 2 [StringPrep, d2])r¨ r r3 r )r¢ Zread_dirr, r, r- Ú_get_unicode_direction_ruleã s ÿrª )Únormalized_strr c C s¤ t tttttttttt t
|
d }z| t | ¡ W n2 t k
|
rb } z| t|f W Y ¢S d}~X Y nX | D ]6}| ¡ D ](\}}||rt|dkrt||f S qtqhdS )aª Check for Prohibited Output according to rfc4013 profile.
|
|
This profile specifies the following characters as prohibited input:
|
|
- Non-ASCII space characters [StringPrep, C.1.2]
|
- ASCII control characters [StringPrep, C.2.1]
|
- Non-ASCII control characters [StringPrep, C.2.2]
|
- Private Use characters [StringPrep, C.3]
|
- Non-character code points [StringPrep, C.4]
|
- Surrogate code points [StringPrep, C.5]
|
- Inappropriate for plain text characters [StringPrep, C.6]
|
- Inappropriate for canonical representation characters [StringPrep, C.7]
|
- Change display properties or deprecated characters [StringPrep, C.8]
|
- Tagging characters [StringPrep, C.9]
|
|
In addition of checking of Bidirectional Characters [StringPrep, Section 6]
|
and the Unassigned Code Points [StringPrep, A.1].
|
|
Returns:
|
A tuple with ("probited character", "breaked_rule")
|
) z4Space characters that contains the ASCII code pointsz&Space characters non-ASCII code pointsz(Unassigned Code Points [StringPrep, A.1]z.Non-ASCII space characters [StringPrep, C.1.2]z,ASCII control characters [StringPrep, C.2.1]z(Private Use characters [StringPrep, C.3]z+Non-character code points [StringPrep, C.4]z'Surrogate code points [StringPrep, C.5]z9Inappropriate for plain text characters [StringPrep, C.6]zGInappropriate for canonical representation characters [StringPrep, C.7]zDChange display properties or deprecated characters [StringPrep, C.8]z$Tagging characters [StringPrep, C.9]Nr )r r r r r r r r r
|
r r Úupdaterª r3 rj Úitems)r« Úrulesr^ ÚcharZruleÚfuncr, r, r- Ú"validate_normalized_unicode_stringÿ s, ô"r± )Úa_stringr c C s0 dd | D }d |¡}t d|¡}|s,dS |S )a( normalizes a unicode string according to rfc4013
|
|
Normalization of a unicode string according to rfc4013: The SASLprep profile
|
of the "stringprep" algorithm.
|
|
Normalization Unicode equivalence is the specification by the Unicode
|
character encoding standard that some sequences of code points represent
|
essentially the same character.
|
|
This method normalizes using the Normalization Form Compatibility
|
Composition (NFKC), as described in rfc4013 2.2.
|
|
Returns:
|
Normalized unicode string according to rfc4013.
|
c S s( g | ] }t |rd nt|r dn|qS )r rc )r r )rf r¯ r, r, r- rh F s ÿz,normalize_unicode_string.<locals>.<listcomp>rc ÚNFKC)rk r¦ Ú normalize)r² Z nstr_listZnstrr, r, r- Únormalize_unicode_string3 s þ
|
rµ ó ru )Úpayloadrw r c
|
C sr t | tr| S t | tr t| S t | tsjzt| j|dW S tk
|
rh } ztd|W 5 d}~X Y nX t| S )z(Initialize a bytearray from the payload.rv z payload must be a str or bytesN)r% r4 r& ÚbytesÚencodeÚAttributeErrorr3 )r· rw r^ r, r, r- Úinit_bytearrayX s
|
|
|
r» c C s¸ ddd} t jdkrldt ¡ d kr.d| d< n&dt ¡ d krHd | d< n t ¡ | d< d
|
t ¡ d | d < nHt ¡ | d< t ¡ d krdt ¡ d | d < nd t dd
¡| d < | S )z4Return a dict with the platform arch and OS version.rc )ÚarchÚversionÚntÚ64r Úx86_64r¼ Ú32Úi386zWindows-r r½ ÚDarwinzmacOS-ú-rN )
|
r| r ÚplatformÚ architectureÚ win32_verÚmachineÚsystemÚmac_verrk r¡ )Úplatr, r, r- Ú get_platformi s
|
|
|
|
rÌ )NN)Nrm )r¶ ru )IÚ__doc__r| rÅ r( r r r¦ Údecimalr Ú functoolsr Ú
|
stringprepr r r r r r r
|
r r r r r r r Útypingr r r r r r r Z custom_typesr Útypesr r ÚboolÚ__annotations__r& Úfloatr r¸ r. r4 r7 r: r= r? rA rB rG rK rS rZ r_ r` rb rj rl rs r r r r¡ r¨ rª r± rµ r» rÌ r, r, r, r- Ú<module> s @$ ü(
|
5 * ý
|
ü ÿ þ %" þ 4& ÿ
|
þ
|
