G)f?<dZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZdd lmZmZmZmZmZdd lmZmZmZm Z dd l!m!Z!dd l"m#Z#m$Z$dd l%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.e rddl/m0Z0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8GddZ9y)z,Implements the MySQL Client/Server protocol.) annotationsN)deque)DecimalDecimalException) TYPE_CHECKINGAnyDequeDictListOptionalSequenceTupleUnion)utils)PARAMETER_COUNT_AVAILABLE ClientFlag FieldFlag FieldType ServerCmd) DatabaseErrorInterfaceErrorProgrammingError get_exception)logger)MySQLAuthPluginget_auth_plugin)"MySQLCachingSHA2PasswordAuthPlugin)BinaryProtocolTypeDescriptionType EofPacketType HandShakeType OkPacketTypeStatsPacketType StrOrBytes) MySQLSocket -i@c >eZdZdZed#dZed$dZed$dZed%dZed&dZ e d' d(d Z ede d e ddddddf d)d Z ee d e f d*d Zed+d,d Zed-d.dZed/dZed0dZed1dZed2d3dZd4dZed5d6dZ d- d7dZe d8dZe d9dZe d: d;dZe ddZ d? d@dZedAdZedBdZ e dCdZ!e dDd Z"edEd!Z# dF dGd"Z$y)H MySQLProtocolzRImplements MySQL client/server protocol Create and parses MySQL packets. c:|dtk(s td|ddS)a Parse a MySQL auth more data packet. Args: pkt: Packet representing an `auth more data` response. Returns: auth_data: Authentication method data (see [1]). Raises: InterfaceError: If packet's status tag doesn't match `protocol.EXCHANGE_FURTHER_STATUS`. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_auth_more_data.html z"Failed parsing AuthMoreData packetN)EXCHANGE_FURTHER_STATUSr)pkts U/var/www/html/flask-app/venv/lib/python3.12/site-packages/mysql/connector/protocol.pyparse_auth_more_dataz"MySQLProtocol.parse_auth_more_data[s($1v00 !EF F12wc|dtk7r tdtj|ddd\}}|r |ddk(r|dd}|j |fS) a|Parse a MySQL auth switch request packet. Args: pkt: Packet representing an `auth switch request` response. Returns: plugin_name: Name of the client authentication plugin to switch to. plugin_provided_data: Plugin provided data (see [1]). Raises: InterfaceError: If packet's status tag doesn't match `protocol.AUTH_SWITCH_STATUS`. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/ latest/page_protocol_connection_phase_packets_protocol_ auth_switch_request.html r/z'Failed parsing AuthSwitchRequest packetr0Nendr)AUTH_SWITCH_STATUSrr read_stringdecode)r2 plugin_names r3parse_auth_switch_requestz'MySQLProtocol.parse_auth_switch_requestqsf( q6' ' !JK K ,,SW'B[ 3r7a<cr(C!!#S((r5ctj|ddd\}}|tk7r tdtj|d\}}|j |fS)aParse a MySQL auth next factor packet. Args: pkt: Packet representing an `auth next factor` response. Returns: plugin_name: Name of the client authentication plugin. plugin_provided_data: Initial authentication data for that client plugin (see [1]). Raises: InterfaceError: If packet's packet type doesn't match `protocol.MFA_STATUS`. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_auth_ next_factor_request.html r/Nrz.Failed parsing AuthNextFactor packet (invalid)r7r8)rread_int MFA_STATUSrr<r=)r2statusr>s r3parse_auth_next_factorz$MySQLProtocol.parse_auth_next_factors]*nnSWa0 V Z  !QR R ,,Sg>[!!#S((r5c VD] }| d|<tfdDtjztjz}t j d|g}D]}|j t j dt||j |j|j t j dt||j |jdj|S)azEncode the connection attributes. Args: conn_attrs: Connection attributes. Returns: serialized_conn_attrs: Serialized connection attributes as per [1]. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_handshake_response.html c3RK|]}t|t|z ywN)len).0x conn_attrss r3 z0MySQLProtocol.make_conn_attrs..s#@AZ]++@s$'?# EI  $ $V[[s9~%F G  $ $Y%5%5%7 8  $ $V[[s:i;P7Q%R S  $ $Z %:%A%A%C D  E xx)**r5cV|tjzr|r|jdzSdS)aPrepare database string for handshake response. Args: client_flags: Integer representing client capabilities flags. database: Initial database name for the connection. Returns: serialized_database: Serialized database name as per [1]. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_handshake_response.html r7)rCONNECT_WITH_DBrU) client_flagsdatabases r3connect_with_dbzMySQLProtocol.connect_with_dbs4"j888X OO  '  r5NFcz|sdt|||fS|i} t|||||}|j|fi|} | t d|j |t jzr!tjt| | zn| dz} | |fS#ttf$r} t d| | d} ~ wwxYw)aPrepare the first authentication response. Args: auth_data: Authorization data from initial handshake. username: Account's username. password: Account's password. client_flags: Integer representing client capabilities flags. auth_plugin: Authorization plugin name. auth_plugin_class: Authorization plugin class (has higher precedence than the authorization plugin name). ssl_enabled: Whether SSL is enabled or not. plugin_config: Custom configuration to be passed to the auth plugin when invoked. The parameters defined here will override the ones defined in the auth plugin itself. Returns: auth_response: Authorization plugin response. auth_strategy: Authorization plugin instance created based on the provided `auth_plugin` and `auth_plugin_class` parameters. Raises: InterfaceError: If authentication fails or when got a NULL auth response. r7) ssl_enabledNzFailed authentication: z8Got NULL auth response while authenticating with plugin ) rr auth_response TypeErrorrnamerSECURE_CONNECTIONr int1storerI) auth_datausernamepasswordr] auth_pluginauth_plugin_classra plugin_config auth_strategyrberrs r3auth_plugin_first_responsez(MySQLProtocol.auth_plugin_first_responsesF>(    M KKOK9JK( M8M77 S]SM   ',,-/ j::: OOC . /- ?(  m++>* K #:3%!@As J Ks(BB:&B55B:rc ^|j} g}| rtjdtjd| tdd|j d tdd |xs|d}tjd || rB|jtjd t| d tj| n9d }|jtjd|t| d |||| tj|d|||||| | \}}|j||jtj||| r%|jtjd||t j"zr"|j|jdz|t j$zr&| $|jtj'| dj)||fS#t t f$r}td|d dd}~wwxYw)aMake a MySQL Authentication packet. Args: handshake: Initial handshake. username: Account's username. password: Account's password. database: Initial database name for the connection charset: Client charset (see [2]), only the lower 8-bits. client_flags: Integer representing client capabilities flags. max_allowed_packet: Maximum packet size. auth_plugin: Authorization plugin name. auth_plugin_class: Authorization plugin class (has higher precedence than the authorization plugin name). conn_attrs: Connection attributes. is_change_user_request: Whether is a `change user request` operation or not. ssl_enabled: Whether SSL is enabled or not. plugin_config: Custom configuration to be passed to the auth plugin when invoked. The parameters defined here will override the one defined in the auth plugin itself. Returns: handshake_response: Handshake response as per [1]. auth_strategy: Authorization plugin instance created based on the provided `auth_plugin` and `auth_plugin_class`. Raises: ProgrammingError: Handshake misses authentication info. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_handshake_response.html [2]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_basic_character_set.html#a_protocol_character_set zGot a `change user` requestzStarting authorization phaseNzGot a NULL handshakergz$Handshake misses authentication inforjz-Handshake misses authentication plugin info ()z#The provided initial strategy is %srNsxxxxxxxxxxxxxxxxxxxxxxxz ? *00 0  # #K$6$6$87$B C :22 2 8N  # #M$A$A*$M Nxx()=88o8$ "?uAF  s+ HH,H''H,cdjtj|tj|tj|dgS)aMake a SSL authentication packet (see [1]). Args: charset: Client charset (see [2]), only the lower 8-bits. client_flags: Integer representing client capabilities flags. max_allowed_packet: Maximum packet size. Returns: ssl_request_pkt: SSL connection request packet. References: [1]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_connection_phase_packets_protocol_ssl_request.html [2]: https://dev.mysql.com/doc/dev/mysql-server/latest/ page_protocol_basic_character_set.html#a_protocol_character_set r5srVr int4store int2store)r|r]r}s r3 make_auth_sslzMySQLProtocol.make_auth_sslsB0xx - 23(    r5c>tj|}||S||zS)z(Make a MySQL packet containing a command)rrf)commandargumentdatas r3 make_commandzMySQLProtocol.make_commands&w''tqAJ z?. .6s:6GH$$4#57 271B1B 12JG2 .-. MM2F1RL A ! "    N   ),-F)G)N)N)P %&}}]]%BC  *66 64D3r+a/0"D$JDE]F"~"'_ *00 0f$-F)G)R)R*14V,]+/4/@/@W/U,]+!$]!3!:!:7!CC !8C % 2K*N r5c|ddk(s tdi} tjd|ddd|d<tj|dd\}|d<tj|\}|d <tjd |dd\|d <|d <|dd}|r2tj |\}|d <|d j d|d <|S#t$r}td|d}~wwxYw)zParse a MySQL OK-packetr/rz#Failed parsing OK packet (invalid).rr0 field_countN affected_rows insert_idz->v-F *FIk* eVAa[1 -(/*ABZF050D0DV0L- *-(1*(=(D(DW(M *% G !<=3 F GsB3C C& C!!C&c tj|ddd}|S#tjtf$r}t d|d}~wwxYw)z=Parse a MySQL packet with the number of columns in result setr/NrzFailed parsing column count)rrrRerrorrr)rcountrns r3parse_column_countz MySQLProtocol.parse_column_countsQ I%%fQRj1!4EL j) I !>?S H IsA  AA c tj|dd\}}tj|\}}tj|\}}tj|\}}tj|\}}tj|\}} tjd|\}}}}}|j ||dddd|tjz||f S#tj$r t ddwxYw)zParse a MySQL column-packet.r/Nz ==( (. 3}}Z8H s"s6{a'7 ) )'{O%a[M || 3 )s 2 3sA((B ; BB c $d}i}dg}d}|r|ddjd}n|jd}|D]T} |jddDcgc]}|jc}\}}|j d} t |||<V|Scc}w#t$r} t|| d} ~ wwxYw#t tf$rQ t|j d||<n/#t$r#} t|d |d t|d | d} ~ wwxYwYwxYw) zParse the statistics packetz)Failed getting COM_STATISTICS informationr5r/Ns :r)rz (:rq) splitstriprrr=intrwrrrepr) r with_headererrmsgrpairslblpairvvalrns r3parse_statisticszMySQLProtocol.parse_statisticsMs8=.0 12J$$[1ELL-E SD 6/3zz$/BC!AGGICS **W%C Ss8C S D 6$V,#5 6j) SS&szz'':;CH'S(F82cU!DI;a)HIsRS Ss_B B #B;B/ B B, B''B,/D?CD D &DD  DDc|}g}d}d}d} |s||k(r ||fS|j} | jdr| ddg} |j} | jdr6| j| dd|j} | jdr6| j| ddtjdj | }nG| ddk(r| ddkr|j | }d}n#d}tjt| dd}|||j|n| | t| |dz }6) zRead MySQL text result Reads all or given number of rows from the socket. Returns a tuple with 2 elements: a list with all rows and the EOF packet. Nrsr/r5r(r) recvrrTrread_lc_string_listrVrbytesr) rsockversionrrreofrowdatairdatass r3read_text_resultzMySQLProtocol.read_text_resultjsS  a5j*Sy)YY[F  1 ''8LL,!YY[F''8 VABZ(33CHHUODc!fQi!mnnV,33E&*4EF{w2 G$#F++ FA-r5c|dtjk(rd}d}n_|dtjk(rd}d}nD|dtjtjfvrd}d}n|dtj k(rd}d}|d t jzrj}|d tj|d |d fS) z%Parse an integer from a binary packetr>: :GF 1X++ +GF 8i(( (mmoGvwwq8H!I!!LMMr5c|dtjk(rd}d}nd}d}||dtj||d|dfS)z)Parse a float/double from a binary packetrr >vw O  e$q""## #'">">vuQx"P e$qY^^+"&"9"9&"A e$qY---qR"'"6"6v"> e$"'"6"6v">)MM%,,w"78G% )NV}YER*)MM%()sJ; KKKc6g}d}d}d} | ||fS||k(r ||fSt|j} | ddk(r|j| }d}n | ddk(rd}|j|| dd|}|||j |n| | t | |dz })zwRead MySQL binary protocol result Reads all or given number of binary resultset rows from the socket. Nrr/r(r0r)rrrr%rTr) rrcolumnsrr|rrrQrrs r3read_binary_resultz MySQLProtocol.read_binary_result-s c{Ezc{499;'FayCnnV,a227F12JP{v1 F##F++ FA!r5c\|ddk(s tdi} tj|ddd\}|d<tj|d\}|d<tj|d\}|d <|d d}tj|d\}|d <|S#t$r}td|d}~wwxYw) z(Parse a MySQL Binary Protocol OK packet.r/rzFailed parsing Binary OK packetr0Nrr) num_columns num_paramsrr)rrrAr)rok_pktrns r3parse_binary_prepare_okz%MySQLProtocol.parse_binary_prepare_okOsayA~ !BC C M-2^^F12J-J *FF>*,1NN61,E )FF=)+0>>&!+D (FF<(ABZF.3nnVQ.G +FF?+  M !BC L MsA8B B+ B&&B+cd}d}|dkr[|dk\rd}tj}n|dk\rd}tj}n|dk\rd}tj}nod }tj}n\d }|d krd }tj}nB|d krd}tj}n*|dkrd}tj}ntj}d}t j ||||fS)z0Prepare an integer for the MySQL binary protocolNriririrrr*rNirtlrz++4 4%44 4 ( 4t  '     O D"@@B(A6(A (A(AT5A75A 5A5AnTT.0!;?FJ',E E +E  E  E 9 E E DE !%E  E r5r-):rl __future__rrrR collectionsrdecimalrrtypingrrr r r r r rrrFr constantsrrrrrerrorsrrrrrpluginsrrplugins.caching_sha2_passwordrrXrr r!r"r#r$r%networkr&rr;r1 OK_STATUSrB ERR_STATUSrnror-rir5r3r{s:3" -   SR5M$    '^ ^ r5