o hQo@sdZddlZddlZddlmZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZmZdd l mZmZd ZGd d d eZdS) z,Implements the MySQL Client/Server protocol N)Decimal) FieldFlag ServerCmd FieldType ClientFlag)errorsutils)get_auth_plugin)PY2 struct_unpack) DatabaseError get_exception c@seZdZdZddZddZ   dDd d Z  dEd dZdFddZdGddZ   dHddZ ddZ ddZ ddZ dIddZdd ZdJd"d#ZdGd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZdId.d/ZdKd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z < =dLd>d?Zd@dAZdBdCZdS)M MySQLProtocolzRImplements MySQL client/server protocol Create and parses MySQL packets. cCs |tj@r|r|ddSdS)z.Prepare database string for handshake responseutf8)rCONNECT_WITH_DBencode)self client_flagsdatabaserJ/var/www/html/env/lib/python3.10/site-packages/mysql/connector/protocol.py_connect_with_db4szMySQLProtocol._connect_with_dbc Cs|sdSzt||||||d}|} Wnttjfy.} z tdt| d} ~ ww|tj@rBt | } t d| | } | S| d} | S)z#Prepare the authentication responser)usernamepasswordr ssl_enabledzFailed authentication: {0}Nz   zMySQLProtocol.make_authcCs"t|t|t|dS)z Make a SSL authentication packets)r int4store int1store)rr:rr;rrr make_auth_sslvszMySQLProtocol.make_auth_sslcCst|}|dur ||7}|S)z(Make a MySQL packet containing a commandN)r r@)rcommandargumentdatarrr make_command~s zMySQLProtocol.make_commandrcCst|t|S)z0Make a MySQL packet with Fetch Statement command)r r?)r statement_idrowsrrrmake_stmt_fetchszMySQLProtocol.make_stmt_fetchc Csz |d} |p |d}Wnttfy"} ztd| d} ~ ww|s'd}z|d} Wn ty9|} Ynwtdjt | dt j | } | | |||||| |7} | | ||7} | td |7} |tj@rs| |dd 7} | S) z0Make a MySQL packet with the Change User commandr(r'r0Nr1rz 2z2MySQLProtocol.parse_statistics..:rrWz{0} ({1}:{2}).)splitrr r\intrr!) rr= with_headererrmsgr`pairspairlblvalrrrparse_statistics's,    zMySQLProtocol.parse_statisticsc Cs"g}d}d}d} |s||kr ||fS|}|drS|ddg} |}|dr?| |dd|}|ds-| |ddttd| }n|ddkrg|ddkrg||}d}n d}t|dd}|dur|dur||n |dur|durt||d 7}q ) 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. NrTsrhr1r}r) recvr_appendr read_lc_string_list bytearrayjoinrr) rsockversionrurGeofrowdatair=datasrrrread_text_resultBs:      zMySQLProtocol.read_text_resultcCs|dtjkr d}d}n&|dtjkrd}d}n|dtjtjfvr'd}d}n |dtjkr2d}d}|d tj@r=|}||d t ||d |d fS) z%Parse an integer from a binary packetrbhrrrhqrUrNr) rTINYSHORTINT24LONGLONGLONGrUNSIGNEDupperr )rr=fieldformat_lengthrrr_parse_binary_integergs"z#MySQLProtocol._parse_binary_integercCsB|dtjkr d}d}nd}d}||dt||d|dfS)z)Parse a float/double from a binary packetrrUdrhfNr)rDOUBLEr )rr=rrrrrr_parse_binary_float{s "z!MySQLProtocol._parse_binary_floatc Cs|d}d}|dkr tjtd|ddd|d|dd}n6|dkrVd}|d kr7td |d |dd}tjtd|ddd|d|d|d |d |d|d}||dd|fS)z&Parse a timestamp from a binary packetrNrhHrr)yearmonthdayr IrUrL)rrrhourminutesecond microsecond)datetimedater )rr=rrvaluemcsrrr_parse_binary_timestamps, z%MySQLProtocol._parse_binary_timestampcCs|d}|d|d}d}|dkrtd|ddd}td|ddd}|ddkr2|d9}tj||d||d |dd }||dd|fS) z'Parse a time value from a binary packetrrrUrNrLrVrr)daysseconds microsecondsminuteshours)r r timedelta)rr=rrrDrrtmprrr_parse_binary_times z MySQLProtocol._parse_binary_timec Cslt|ddd}dd|d|D}||d}g}t|D]\}}|t|ddd|dd>@r=|dq!|dtjtjtjtjtj fvr[| ||\}} || q!|dtj tj fvrs| ||\}} || q!|dtjtjtjfvr|||\}} || q!|dtjkr|||\}} || q!t|\}} || |q!t|S) z(Parse values from a binary result packetrrrUcSsg|]}t|qSr)r)rrrrrrrz6MySQLProtocol._parse_binary_values..rNr)r$ enumeraterrrrrrrrrrFLOATrDATETIMEDATE TIMESTAMPrTIMErr rpr\tuple) rfieldsr=r:null_bitmap_length null_bitmapvaluesposrrrrr_parse_binary_valuess8 $     z"MySQLProtocol._parse_binary_valuesc Csg}d}d}d} |dur ||fS||kr ||fS|} | ddkr-|| }d}n| ddkr@d}||| dd|}|durN|durN||n |durZ|durZt| |d7}q )zwRead MySQL binary protocol result Reads all or given number of binary resultset rows from the socket. NrTrhr}rLr)rrrrr) rrcolumnsrur:rGrrrr=rrrread_binary_results0     z MySQLProtocol.read_binary_resultcCs|ddks tdi}z5t|ddd\}|d<t|d\}|d<t|d\}|d <|d d}t|d\}|d <W|StyNtdw) z'Parse a MySQL Binary Protocol OK packetrhrzFailed parsing Binary OK packetrLNrFr num_columns num_paramsrrm)rr r read_intrq)rr=ok_pktrrrparse_binary_prepare_oks    z%MySQLProtocol.parse_binary_prepare_okcCsd}d}|dkr,|dkrd}tj}n?|dkrd}tj}n5|dkr&d}tj}n+d }tj}n%d }|d kr8d }tj}n|d krBd}tj}n|dkrLd}tj}ntj}d}t||||fS)z0Prepare an integer for the MySQL binary protocolNriririrrBirlrQ)rrrrrr%r&)rrrzr{rrrr_prepare_binary_integers6z%MySQLProtocol._prepare_binary_integercCst|tjr tj}nt|tjrtj}ntdt|j t |j t |j }t|tjrQ|t |j t |jt |j}|jdkrQ|t|j7}t t||}||fS)aPrepare a timestamp object for the MySQL binary protocol This method prepares a timestamp of type datetime.datetime or datetime.date for sending over the MySQL binary protocol. A tuple is returned with the prepared value and field type as elements. Raises ValueError when the argument value is of invalid type. Returns a tuple. z2Argument must a datetime.datetime or datetime.dater) isinstancerrrrrrqr int2storerr@rrrrrrr?r$)rrrzpackedrrr_prepare_binary_timestamps,         z'MySQLProtocol._prepare_binary_timestampc Cst|tjtjfs tdtj}d}d}d}t|tjrO|jdkr#d}t|j d\}}t|d\}} |t t |jt |t |t | 7}|j}n|t dt |jt |jt |j7}|j}|rt|t |7}t ||}t t||}||fS)aPrepare a time object for the MySQL binary protocol This method prepares a time object of type datetime.timedelta or datetime.time for sending over the MySQL binary protocol. A tuple is returned with the prepared value and field type as elements. Raises ValueError when the argument value is of invalid type. Returns a tuple. z3Argument must a datetime.timedelta or datetime.timerNr1ri<)rrrtimerqrrrdivmodrr r?absr@rrrrrr$) rrrznegativerrr remainderminssecsrrr_prepare_binary_time@sD       z"MySQLProtocol._prepare_binary_timecCst|t||}|S)zMPrepare long data for prepared statements Returns a string. )r r?r)r statementparamrDr=rrr_prepare_stmt_send_long_datamsz*MySQLProtocol._prepare_stmt_send_long_datarrc Csd}dgt|dd}g} g} d} |dkrd}|duri}|r[|r[t|t|kr2td t|D]#\} } || }d}|durb|| dd| d>O<| ttjt|q6| |vrt|| drptj }ntj }nt |t r| |\} }}| | nt |trtr| tt||n||}| tt||tj}nt |tr| tt||tj }ntrt |tr||}| tt||tj}nlt |tr| ttt||t||tj}nLt |tr| td |tj}n9t |tjtjfr+||\} }| | n"t |tjtjfrB| |\} }| | n t!d j"|j#j$d | t|t|q6t%|t|t%|d&d d|Dtd}| D]}||7}q{| D]}||7}q|S)z6Make a MySQL packet with the Statement Execute commandrrrrUr1utf8mb4rNzTFailed executing prepared statement: data values does not match number of parametersrz:MySQL binary protocol can not handle '{classname}' objects) classnamecSsg|]}td|qS)r)r%r&)rbitrrrrsz3MySQLProtocol.make_stmt_execute..)'r$rr rrr r@rNULLBLOBSTRINGrrrr"r lc_intrVARCHARbytesunicoderDECIMALfloatr%r&rrrrrrrr6r! __class____name__r?r)rrFrD parametersr{long_data_usedr:iteration_countrrtypesrrrxrrzr=a_typea_valuerrrmake_stmt_executexs                zMySQLProtocol.make_stmt_executecCsZ|ddks tdtj|dddd\}}|r&|dd kr&|dd}|d |fS) z&Parse a MySQL AuthSwitchRequest-packetrhr}z'Failed parsing AuthSwitchRequest packetrLNrrNrVrr)rr r r[r\)rr= plugin_namerrrparse_auth_switch_requests  z'MySQLProtocol.parse_auth_switch_requestcCs"|ddks td|ddS)z!Parse a MySQL AuthMoreData-packetrhrz"Failed parsing AuthMoreData packetrLN)rr )rr=rrrparse_auth_more_datas  z"MySQLProtocol.parse_auth_more_data)NNNr.rr/FN)r.rr/)N)r)NNNr.rFN)rW)T)rrW)rrrNr) r __module__ __qualname____doc__rr-r>rArErHrJrgrsrvr|rrrrrrrrrrrrrrrr r rrrrr.sR %   $0   %   !"-  \ r)r r%rdecimalr constantsrrrrrr authenticationr catch23r r r rrZobjectrrrrrs