U 4Je@sUdZddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZmZmZmZdd lmZmZmZmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'd Z(e)e*d <Gd ddZ+dS)z,Implements the MySQL Client/Server protocol.N)DecimalDecimalException)AnyDictListOptionalSequenceTupleUnion)utils)get_auth_plugin)PARAMETER_COUNT_AVAILABLE ClientFlag FieldFlag FieldType ServerCmd) DatabaseErrorInterfaceErrorProgrammingError get_exception) ConnAttrsTypeDescriptionType EofPacketType HandShakeType OkPacketType%ParseValueFromBinaryResultPacketTypes SocketTypeStatsPacketType StrOrBytes!SupportedMysqlBinaryProtocolTypes PROTOCOL_VERSIONc@s*eZdZdZeeeeedddZ eeee eeeeeeeee ed ddZ d_ee ee eeeeeeee eeeeeeed ddZeeedddZed`eeeedddZedaeeeedddZedbeeedddZdce ee eeeeeee eeeeeeed ddZeee d d!d"Zeeeeefd d#d$Zeeed d%d&Zeeeed d'd(Zeddeeed*d+d,Zeed d-d.Zedeee e d0d1d2Z!dfe"eed3feee#eeed3feefd4d5d6Z$eeeeeefd7d8d9Z%eeeeee&fd7d:d;Z'edgeeeee(fd=d>d?Z)eeeeeee*e+j,e+j+ffd@dAdBZ-eeeee+j.fd dCdDZ/dhe#eeeee0d3fdEdFdGZ1die"e#eeeee#ee0d3feefdHdIdJZ2eee3eefd dKdLZ4eeeeeefdMdNdOZ5ee*e+j,e+j+feeefdMdPdQZ6ee*e+j.e+j7feeefdMdRdSZ8eeeeedTdUdVZ9djee:e;e:e<eee3eee feee#eeeeeed d]d^Z?dS)k MySQLProtocolzRImplements MySQL client/server protocol Create and parses MySQL packets. ) client_flagsdatabasereturncCs |tj@r|r|ddSdS)z.Prepare database string for handshake responseutf8)rZCONNECT_WITH_DBencode)r$r%r*sz0MySQLProtocol.make_conn_attrs..r3r')sumr7keysvaluesr8r9r))rB attr_nameZconn_attrs_lenZconn_attrs_packetr*rTr+rKs      zMySQLProtocol.make_conn_attrs)r@r$rAr&cCs"t|t|t|dS)z Make a SSL authentication packetsr int4store int2store)r@r$rAr*r*r+ make_auth_sslszMySQLProtocol.make_auth_ssl)commandargumentr&cCst|}|dk r||7}|S)z(Make a MySQL packet containing a commandN)r int1store)r^r_datar*r*r+ make_commands zMySQLProtocol.make_commandr ) statement_idrowsr&cCst|t|S)z0Make a MySQL packet with Fetch Statement command)r r[)rcrdr*r*r+make_stmt_fetchszMySQLProtocol.make_stmt_fetch) r?r-r.r%r@r$r2r0rBr/r&c  Cs"z|d} |p|d}Wn8ttfk rP} ztd| ddW5d} ~ XYnX|sZd}z|d} Wntk r|} YnXt| }td|d tj | }|| ||||| || |7}|| ||7}|td |7}|t j @r||dd 7}|t j@r| dk r||| 7}|S) z0Make a MySQL packet with the Change User commandr1r0rCrDNrEr'r3rFsz2MySQLProtocol.parse_statistics..:r{rqz (:rD) splitrrruintrGrrrepr) rOrerrmsgrxpairsZlblpairvalr;r*r*r+parse_statisticss*  :zMySQLProtocol.parse_statistics.)sockversionrr&c Cs"|}g}d}d}d}|s ||kr$q|} | dr| ddg} |} | drr| | dd|} qL| | ddttd| }n>| ddkr| ddkr|| }d}nd}t| dd}|dkr|dk r||n|dkr|dkrt| |d7}q||fS) 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. NrsrrErr ) recvrwappendr Zread_lc_string_list bytearrayjoinrr) rLrrrrrdeofZrowdatairOZdatasr*r*r+read_text_results6        zMySQLProtocol.read_text_result)rOfieldr&cCs|dtjkrd}d}nL|dtjkr0d}d}n4|dtjtjfkrNd}d}n|dtjkrdd}d}|d tj@rz|}||d t ||d |d fS) z%Parse an integer from a binary packetr @r|dqF| dtjtjtjtjtj fkr| || \}}||qF| dtj tj fkr| || \}}||qF| dtjtjfkr|||\}}||qF| dtjtjtjfkrX||| d\}}||qF| dtjkr||\}}||qFt|\}}z|||WqFtk r||YqFXqFt|S) z(Parse values from a binary result packetrr{rocSsg|] }t|qSr*)r)rRrr*r*r+rbsz6MySQLProtocol._parse_binary_values..rNr )r7 enumeraterrrrrrrrrrFLOATrDECIMALZ NEWDECIMALrrDATErrTIMErr rruUnicodeDecodeErrortuple) rLrrOr@Znull_bitmap_length null_bitmaprXrposrr*r*r+_parse_binary_valuesZsP $      z"MySQLProtocol._parse_binary_values)rcolumnsrr@r&c Csg}d}d}d}|dk rq||kr$q|} | ddkrH|| }d}n&| ddkrnd}||| dd|}|dkr|dk r||n|dkr|dkrt| |d7}q||fS)zwRead MySQL binary protocol result Reads all or given number of binary resultset rows from the socket. Nrrrrir )rrrrr) rLrrrr@rdrrXrrOr*r*r+read_binary_results*     z MySQLProtocol.read_binary_resultc Cs|ddkstdi}zht|ddd\}|d<t|d\}|d<t|d\}|d <|d d}t|d\}|d <Wn,tk r}ztd|W5d}~XYnX|S) z'Parse a MySQL Binary Protocol OK packetrrzFailed parsing Binary OK packetriNrcr{Z num_columnsZ num_paramsr r)rr r|r)rOZok_pktr;r*r*r+parse_binary_prepare_oks  z%MySQLProtocol.parse_binary_prepare_ok)rr&cCsd}d}|dkrX|dkr$d}tj}q|dkr8d}tj}q|dkrLd}tj}qd }tj}nJd }|d krpd }tj}n2|d krd}tj}n|dkrd}tj}n tj}d}t||||fS)z0Prepare an integer for the MySQL binary protocolNriririrrr3irflrzO<| ttj t|q||kr6||dr,tj }ntj }nt |t r`||\} }}| | npt |tr||}| tt||tj }n8t |tr| tt||tj }n t |tr| ttt||t||tj}nt |tr,| td |tj}nt |tjtjfrZ||\} }| | nvt |tjtjfr||\} }| | nH|rt||}| tt||tj }ntd |j j!d | t|t||r|d|kr|||d|}|tt||qt"|t|t"| }|dk r`|t|}n|}|r |dk r|t|7}|d#d d| Dtd7}d}| D]H}||7}|dk r|d|kr||||7}n|d7}|d7}q| D]}||7}q|S)z6Make a MySQL packet with the Statement Execute commandr rrrorEZutf8mb4r'NzTFailed executing prepared statement: data values does not match number of parametersrz&MySQL binary protocol can not handle 'z ' objectscSsg|]}td|qS)B)r8r9)rRbitr*r*r+rsz3MySQLProtocol.make_stmt_execute..r()$r7rlistrrrr r`rZNULLZBLOBSTRINGrrrstrr)Zlc_intbytesrrfloatr8r9rrrrrrrr __class____name__r[r)rLrcrarrrr@rrZiteration_countrrXtypesrZdata_lenZquery_attr_namesrZattr_valrr_flagsrrrOZparameter_countrZa_typeZa_valuer*r*r+make_stmt_executeMs                         zMySQLProtocol.make_stmt_executecCsX|ddkstdtj|dddd\}}|rJ|dd krJ|dd}|d |fS) z&Parse a MySQL AuthSwitchRequest-packetrrz'Failed parsing AuthSwitchRequest packetriNr(rkrprr')rr rtru)rOZ plugin_namer*r*r+parse_auth_switch_requests   z'MySQLProtocol.parse_auth_switch_requestcCs |ddkstd|ddS)z!Parse a MySQL AuthMoreData-packetrr z"Failed parsing AuthMoreData packetriN)r)rOr*r*r+parse_auth_more_datas z"MySQLProtocol.parse_auth_more_data) NNNr=rr>FNNN)r=rr>)N)r ) NNNr=rFNNN)rq)T)r )r')rq)r rq)r*r*rNr'NF)@r __module__ __qualname____doc__ staticmethodrrrrr,rboolr<rrrPrKrr]rbrergrzr r~rrrrrrrrrrrrrrrrrr rrrrrrrrrrrrrrrrr rrrrr*r*r*r+r#Asj' A  =3   )     7 %! ( 2   r#),r rr8decimalrrtypingrrrrrr r rQr Zauthenticationr constantsrrrrrerrorsrrrrrrrrrrrrrrr r"r__annotations__r#r*r*r*r+s$  0