>ijg_ 8UdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddlm Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%erddl&m'Z'dd l(m)Z)dd l*m+Z+m,Z,dd l-m.Z.d Z/e0e1d <e2e3e e)fZ4e#e$e2e$e3e$e e$e)fe1d<de%e2e5fde2fdZ6de2de5fdZ7de2de5fdZ8de2de5fdZ9de2de5fdZ:de2de5fdZ;de2de5fdZde5de#e5e"e5ffdZ?de5de"e#e"e5dffdZ@ dCde5de"e5de"e2de#e5e5ffd ZAde5de2de#e5e2ffd!ZBde5de#e5e"e2ffd"ZCde.deDfd#ZE dDd%e.d&e"eDd'e2ddfd(ZFde eDeDffd)ZGde eDeDffd*ZHde"e eDeDffd+ZIde#eDeDeDffd,ZJd-eDdeDfd.ZKd-eDde eDeeDge0fffd/ZLd0eDde"e#eDeDffd1ZMd2eDdeDfd3ZN dEd6e%e2e.fd7eDdeOfd8ZPe de eDe%eDe#eDeDffffd9ZQd:eDdefd;ZRdZSd=eDddfd?ZTdefd@ZUGdAdBZVdS)Fz 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_c12in_table_c21_c22 in_table_d1 in_table_d2) TYPE_CHECKINGAnyCallableDictListOptionalTupleTypeUnion)MySQLConnectionAbstract) HexLiteral)DEPRECATED_TLS_CIPHERSUITESDEPRECATED_TLS_VERSIONS) StrOrBytesF__MYSQL_DEBUG__ NUMERIC_TYPESbufreturncjt|tr|St|t}}|dkr|dS|dkr6||dd|z zzz }tt jd|dS||dd|z zzz }tt jd|dS)z&Unpacks the given buffer to an integerrrs r4 int2storerCqs6  1uuE =>>> ;tQ  r6cr|dks|dkrtdtjd|ddS)zb Takes an unsigned integer (3 bytes) and packs it as a bytes-object. Returns string. rz%int3store requires 0 <= i <= 16777215r)r;r>s r4 int3storerG|s?  1uuH @AAA ;tQ  ! $$r6cb|dks|dkrtdtjd|S)zb Takes an unsigned integer (4 bytes) and packs it as a bytes-object. Returns string. rz'int4store requires 0 <= i <= 4294967295r)r;r>s r4 int4storerJs6  1uuJBCCC ;tQ  r6cb|dks|dkrtdtjd|S)zZ Takes an unsigned integer (8 bytes) and packs it as string. Returns string. rz!int8store requires 0 <= i <= 2^64r+r;r>s r4 int8storerMs9  1uu(((<=== ;tQ  r6c|dks|dkrtd|dkrt}n1|dkrt}n#|dkrt}n|dkrt}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. rrLz!intstore requires 0 <= i <= 2^64r9rArErI)r<r?rCrGrJrM)r7 formed_strings r4intstorerPs| 1uu(((<===Cxx! e! h! j! ! =  r6c&|dks|dkrtd|dkrtjd|S|dkrdtjd|zS|d kr d tjd |dd zSd tjd|zS)z| Takes an unsigned integer and packs it as bytes, with the information of how much bytes the encoded int takes. rrLzRequires 0 <= i <= 2^64r:rArBrEr)rFr+r;r>s r4lc_intrVs  1uu(((23333ww{4###EzzT1----H}}T1--ac222 V[q)) ))r6sizec.|d|}||d|fS)zi Reads bytes from a buffer. Returns a tuple with buffer less the read bytes, and the bytes. rNr$rWress r4 read_bytesr\s$ af+C J r6c<|ddkr |dddfSdx}}|d}|dkr|}|d|zd|d|dzfS|dkrd}n|dkrd }n|d krd }t|d|dz}|||zdzd||dz||zdzfS) 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). rrRrNrFr*r5)r$r2lsizefsts r4read_lc_stringrfs@ 1v}}ABBFU a&C czzAJLL!3q6A:~#677 czz   SUQY' ( (F "$$ %s519v~7I+I'J KKr6.c"g}dddd}t|}d}||kr||}|dkrdS|dkr|d|d z }n|d kr1|}|||d z||d zz|d |zz }ntd} ||}n#t$rYdSwxYwt||d z||d zz}|||d z|z||z|d zz|d |z|zz }||kt |S) zXReads all length encoded strings from the given buffer Returns a list of bytes r`rFr*)r_rarbrr9NrRrr^)r.appendKeyErrorr5tuple)r$byteslstsizesbuf_lenposfirstr2rds r4read_lc_string_listrpso ')H!! $ $E#hhG C --C C<<4 C<< OOD ! ! ! 1HCC||S1W#'0B$B CDDDq6z! !%LEE   44  cAg#'1B%B!CDDC!GeOfuna6P$P QRRRq5y6))/ --2 ??s B B"!B"endc||td|P ||}n"#t$r}td|d}~wwxYw||dzd|d|fS|t||Std)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 sizezend byte not present in bufferrrz.read_string() needs either end or size (weird))r<indexr\)r$rqrWidxerrs r4 read_stringrv0s {t|ABBB  H))C..CC H H H=>>C G HC!GIIAcE ++ #t$$$ E F FFs- A AA cHt|d|}||d|fS)zMRead an integer from buffer Returns a tuple (truncated buffer, int) rNrcrZs r4read_intrxIs, #af+  C J r6c|std|d}|dkr |dddfS|dkr|ddt|fS|dkr-|ddtjd|dddfS|d kr0|d dtjd |dd d zdfS|d kr-|ddtjd|dddfStd)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.rrRrNr_rFzz"_digest_buffer..os 333! a 333r6c4g|]}dt|dSr)ordrs r4rz"_digest_buffer..ps*4441&#a&&&&&444r6)r,strjoin)r$s r4_digest_bufferrlsV c3  5ww33s333444 7744444 5 55r6abufferprefixlimitc|rF|r|dkrt|d|}nt|}t|dz|zdStt|dS)z2Debug function printing output of _digest_buffer()rz: N)rprint)rrrdigests r4 print_bufferrssz'  -UQYY#GAeG$455FF#G,,F ftmf$%%%%% nW%%&&&&&r6ci}tjdd}tj|s|St |d5}|D]x}|d}t |dkr+|d}|dd  d }|||<y d d d n #1swxYwY|S) zsParse the contents of /etc/os-release file. Returns: A dictionary containing release information. /etcz os-releaseutf-8encoding=r`rr "N ospathrexistsopensplitr.lowerrstripstrip)distroos_release_filefile_objline key_valuekeyvalues r4_parse_os_releasers!  Fgll6<88O 7>>/ * * o 0 0 0 H  D 3I9~~""A,$$&&CaL''--33C88EF3KK                  MsA>Q&&l((**!! ++D1177<<#s  $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ MsA4?:F8H4?7B3>EPS^P[0;  E( 0@@AAAA (((s3xx'''''''(""++-- " "JD$tDzz "dckkTz!!!!! " 4s "A00 B:B BBa_stringczd|D}d|}tjd|}|sdS|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. cVg|]&}t|rdnt|rdn|'S)rr~)rr)rrs r4rz,normalize_unicode_string..WsJ D!!H[->->'HrrDr6r~NFKC)rr normalize)r nstr_listnstrs r4normalize_unicode_stringrDsY&I 779  D   . .D r Kr6r6rpayloadrcLt|tr|St|trt|St|tsF t||S#t $r}t d|d}~wwxYwt|S)z(Initialize a bytearray from the payload.rzpayload must be a str or bytesN)r,r/r-bytesencodeAttributeErrorr<)rrrus r4init_bytearrayris'9%%'3"!!! gu % %H HW^^X^>>?? ? H H H=>>C G H W  s"A55 B?BBcddd}tjdkrxdtjdvrd|d<n7dtjdvrd |d<ntj|d<d tjd |d <nytj|d<tjd kr dtjd|d <n,dtdd|d <|S)z4Return a dict with the platform arch and OS version.r~)archversionnt64rx86_64r32i386zWindows-rrDarwinzmacOS--r`) rrplatform architecture win32_vermachinesystemmac_verrr)plats r4 get_platformrzs=?23N3ND w$ 8(**1- - -#DLL X*,,Q/ / /!DLL#022DL>X%7%9%9!%<>>Y'))V ?   ( (>x'7'9'9!'<>>DOO!hh'9';';AaC'@AADO Kr6fullpathcZt|tstd |dd\}}t j|}t ||}nG#t$rtd|ddttf$r}t||d}~wwxYw|S)a(Import an object from a fully qualified module path. Args: obj (str): A string representing the fully qualified name of the object. Returns: Object: The imported object. Raises: ValueError: If the object can't be imported. .. versionadded:: 8.0.33 z]'fullpath' should be a str representing the fully qualified name of the object to be imported.rzNo callable named ''N) r,rr<rsplit importlib import_modulegetattrrModuleNotFoundError)r module_str callable_strmoduleobjrus r4 import_objectrs h $ $  $   ,#+??3#:#: L(44fl++ FFF:x:::;;E / 0,,,C""+, Js=A$$.B(B##B(cipher_as_ossl tls_versionc|tj|ivr"d|d}tj|t dSdS)aIEmits a warning if a deprecated cipher is being utilized. Args: cipher: Must be ingested as OpenSSL name. tls_versions: TLS version to check the cipher against. Raises: DeprecationWarning: If the cipher is flagged as deprecated according to the OSSA cipher list. z$This connection is using TLS cipher [ which is now deprecated and will be removed in a future release of MySQL Connector/Python.N)rrvalueswarningswarnDeprecationWarning)rrwarn_msgs r4warn_ciphersuites_deprecatedrsh48bIIPPRRRR &> & & &   h 233333 SRr6c\|tvr"d|d}tj|tdSdS)aEmits a warning if a deprecated TLS version is being utilized. Args: tls_versions: TLS version to check. Raises: DeprecationWarning: If the TLS version is flagged as deprecated according to the OSSA cipher list. z%This connection is using TLS version rN)r rrr)rrs r4warn_tls_version_deprecatedrsN--- &K & & &   h 233333 .-r6c*dtdtfd}|S)z Decorator to close the current connection if a read or a write timeout is raised by the method passed via the func parameter. cnx_methodr%c vtjdddtdtdtffd }|S)Ncnxrargskwargsr%c |g|Ri|S#t$r0}t|tr||d}~wwxYw)N) Exceptionr, TimeoutErrorclose)r!r"r#rurs r4handle_cnx_methodzGhandle_read_write_timeout..decorator..handle_cnx_methodsk !z#7777777   c<00 IIKKK  s  A +AA ) functoolswrapsr)rr(s` r4 decoratorz,handle_read_write_timeout..decorators^  $ $ * 36 BE        % $ ! r6)r)r+s r4handle_read_write_timeoutr,s, !h !8 ! ! ! ! r6c\eZdZdZdeddfdZdedefdZded eddffd Zdefd Z xZ S) GenericWrapperz __classcell__)r;s@r4r.r.sFF%%%%%% , , , , , , / /C /D / / / / / /0300000000r6r.)NN)Nr)r6r)WrAr)r rrr0rrrrdecimalrr stringpreprrrrr r r r r rrrrrtypingrrrrrrrrrmysql.connector.abstractsr custom_typesr tls_ciphersrr typesr!r"bool__annotations__r-floatr#rr5r?rCrGrJrMrPrVr\rfrprvrxr|rrrrrrrrrrrr/rrrrrr,r.rYr6r4rMs:                         BAAAAAA$$$$$$MMMMMMMM   Q uT#YU T']D"?2L2L2L2Lj%U%xhuos6J0K'L%%%%T GG G %G 3-G 5%< GGGG2%suUCZ'8>U>uUHSM%9:>>>>46 6s6666EG ' '  '!)# '>A '  ' ' ' '4S>*DcN(HT#s(^$<."E#sC-0""""J S S    TSTT#xt ?T:T5UTTTT811 eCHo1111h"s"s""""L443444444(4S4T4444&8.#0#0#0#0#0#0#0#0#0#0r6