관리-도구
편집 파일: protocol.cpython-38.pyc
U )�e(m � @ s� d Z ddlZddlZddlmZ ddlmZmZmZm Z m Z ddl m Z m Z ddlmZ ddlmZmZ dd l mZ G d d � d e�ZdS ) z,Implements the MySQL Client/Server protocol � N)�Decimal� )� FieldFlag� ServerCmd� FieldType� ClientFlag�MAX_MYSQL_TABLE_COLUMNS)�errors�utils)�get_auth_plugin)�PY2� struct_unpack)� get_exceptionc @ s� e Zd ZdZdd� Zdd� ZdBd d �ZdCd d�ZdDdd�ZdEdd�Z dd� Z dd� Z dd� Z dFdd�Z d d � ZdGd d �ZdHd"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� ZdId.d/�Zd0d1� Zd2d3� Zd4d5� Zd6d7� Zd8d9� ZdJd<d=�Z d>d?� Z d@dA� Z dS )K� MySQLProtocolzRImplements MySQL client/server protocol Create and parses MySQL packets. c C s |t j@ r |r |�d�d S dS )z.Prepare database string for handshake response�utf8� )r ZCONNECT_WITH_DB�encode)�self� client_flags�database� r �m/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/protocol.py�_connect_with_db- s z MySQLProtocol._connect_with_dbc C s� |sdS z"t |�|||||d�}|�� } W n< ttjfk rf } zt�d�t| ����W 5 d} ~ X Y nX |tj @ r�t | �} t � d| �| } n| d } | S )z#Prepare the authentication responser )�username�passwordr � ssl_enabledzFailed authentication: {0}Nz<B) r � auth_response� TypeErrorr �InterfaceError�ProgrammingError�format�strr �SECURE_CONNECTION�len�struct�pack) r r r r r � auth_plugin� auth_datar �authZplugin_auth_response�excZresplenr r r r �_auth_response3 s( � � z MySQLProtocol._auth_responseN�! r � @Fc C s� z|d } | p|d } W n6 t tfk rN } zt�d�| ���W 5 d} ~ X Y nX |sXd}z|�d�} W n tk r~ |} Y nX t�djdt | �d �|||| �} | | � ||||| | |�7 } | | � ||�7 } |t j @ r�| | �d�d 7 } | S ) z"Make a MySQL Authentication packetr' r& �*Handshake misses authentication info ({0})N� r z<IIB{filler}{usrlen}sxZxxxxxxxxxxxxxxxxxxxxxxx)Zfiller�usrlenr )r �KeyErrorr r r r �AttributeErrorr$ r% r# r* r r � PLUGIN_AUTH)r � handshaker r r �charsetr �max_allowed_packetr r& r' r) �username_bytes�packetr r r � make_authJ s@ � � � � zMySQLProtocol.make_authc C s" t �|�t �|� t �|� d S )z Make a SSL authentication packets )r � int4store� int1store)r r4 r r5 r r r � make_auth_sslo s ���zMySQLProtocol.make_auth_sslc C s t �|�}|dk r||7 }|S )z(Make a MySQL packet containing a commandN)r r: )r �command�argument�datar r r � make_commandw s zMySQLProtocol.make_commandc C s� z|d } |p|d }W n6 t tfk rN } zt�d�| ���W 5 d} ~ X Y nX |sXd}z|�d�} W n tk r~ |} Y nX t�djt | �d�t j | �} | | � |||||| |�7 } | | � ||�7 } | t�d |�7 } |tj@ r�| |�d�d 7 } | S ) z0Make a MySQL packet with the Change User commandr' r& r- Nr. r z <B{usrlen}sx)r/ z<Hr )r r0 r r r r r1 r$ r% r# r Z CHANGE_USERr* r r r2 ) r r3 r r r r4 r r r& r' r) r6 r7 r r r �make_change_user~ s8 � � � z MySQLProtocol.make_change_userc C sf i }t d|dd� �d |d< tj|dd� dd�\}|d< t d |dd � �\|d < }}|d < |d < }}|d �� |d< |d d� }t�|| �}d}|tj@ r�|r�td|d �nd} |d| � }|| d� }|d dkr�|dd� }|tj@ �rFd|k�r |d � d��r d| }|d< ntj|dd�\}|d< |d �d�|d< nd|d< || |d< ||d< |S )z Parse a MySQL Handshake-packet�<xxxxBr � �protocolNr ��endZserver_version_originalz<I8sx2sBH2sBxxxxxxxxxx� Zserver_threadidr4 Z server_statusr. � � ���z5.5.8r& �utf-8Zmysql_native_passwordr' � capabilities) r r � read_string�decodeZintreadr r" �minr2 � startswith) r r7 �resZ auth_data1Z capabilities1Z capabilities2Zauth_data_lengthrK Z auth_data2�sizer r r �parse_handshake� sL � � � � z MySQLProtocol.parse_handshakec C s� |d dkst �d��i }z�td|dd� �d |d< t�|dd� �\}|d< t�|�\}|d < td |dd� �\|d < |d < |dd� }|r�t�|�\}|d < |d �d�|d < W n tk r� t �d��Y nX |S )zParse a MySQL OK-packet� r z#Failed parsing OK packet (invalid).rA rB Z field_countNZ affected_rowsZ insert_idz<HH� status_flag� warning_countZinfo_msgrJ zFailed parsing OK packet.)r r r r � read_lc_int�read_lc_stringrM � ValueError)r r7 Z ok_packetr r r �parse_ok� s" � zMySQLProtocol.parse_okc C sT z*t �|dd� �d }|tkr&W dS |W S tjtfk rN t�d��Y nX dS )z=Parse a MySQL packet with the number of columns in result setrS Nr zFailed parsing column count)r rV r r$ �errorrX r r )r r7 �countr r r �parse_column_count� s z MySQLProtocol.parse_column_countrJ c C s� t �|dd� �\}}t �|�\}}t �|�\}}t �|�\}}t �|�\}}t �|�\}}ztd|�\}}}}}W n tjk r� t�d��Y nX |�|�|dddd| tj @ |fS )zParse a MySQL column-packetrS Nz <xHIBHBxxz!Failed parsing column information) r rW r r$ rZ r r rM r ZNOT_NULL)r r7 r4 �_�name� field_type�flagsr r r � parse_column� s. � � �zMySQLProtocol.parse_columnc C s� |d dkr| � |�S d}i }ztd|�}W n tjk rL t�|��Y nX |d dkrft|�dkspt�|��|d |d < |d |d < |S ) zParse a MySQL EOF-packetrS r zFailed parsing EOF packet.z<xxxBBHHr � � � rU � rT )rY r r$ rZ r r r# )r r7 �err_msgrP �unpackedr r r � parse_eof s zMySQLProtocol.parse_eofTc C s� d}i }|r |dd� � d�}n |� d�}|D ]�}z dd� |� dd�D �\}}W n t�|��Y nX |�d �}zt|�||< W q. zt|�d ��||< W n t�d �|||���Y nX Y q.X q.|S ) zParse the statistics packetz)Failed getting COM_STATISTICS informationrS Ns c S s g | ] }|� � �qS r )�strip)�.0�vr r r � <listcomp>% s z2MySQLProtocol.parse_statistics.<locals>.<listcomp>� :rd rJ z{0} ({1}:{2}).)�splitr r rM �intr r ) r r7 Z with_header�errmsgrP �pairs�pairZlbl�valr r r �parse_statistics s* �z MySQLProtocol.parse_statisticsr c C sV g }d}d}d}|dk}|s$||kr(�qN|� � } | �d�r�| dd� g} |� � } | �d�rv| �| dd� � |� � } qP| �| dd� � t�td��| ��}nj| d dkr�| d dk r�| �| �}d}nB|r�| d dkr�| d d kr�| �| �}d}nd}t�| dd� �}|dk�r(|dk �r(|�|� n |dk�rD|dk�rDt | ��|d 7 }q||fS ) z�Read 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. Nr )rB � rB s ���rS r. rb ru rc r ) �recvrO �appendr Zread_lc_string_list� bytearray�joinrh rY r ) r �sock�versionr[ �rows�eofZrowdata�iZeof57r7 Zdatasr r r �read_text_result5 s< z MySQLProtocol.read_text_resultc C s� |d t jkrd}d}nL|d t jkr0d}d}n4|d t jt jfkrNd}d}n|d t jkrdd}d}|d tj@ rz|�� }||d � t ||d |� �d fS ) z%Parse an integer from a binary packetr �b�hrd r~ rS �qrH ru Nr ) r �TINY�SHORT�INT24�LONG�LONGLONGr ZUNSIGNED�upperr )r r7 �field�format_�lengthr r r �_parse_binary_integer_ s z#MySQLProtocol._parse_binary_integerc C sB |d t jkrd}d}nd}d}||d� t||d|� �d fS )z)Parse a float/double from a binary packetr rH �drS �fNr )r �DOUBLEr )r r7 r� r� r� r r r �_parse_binary_floats s z!MySQLProtocol._parse_binary_floatc C s� |d }d}|dkr@t jtd|dd� �d |d |d d�}nl|dkr�d}|d krntd |d |d � �d }t j td|dd� �d |d |d |d |d |d |d�}||d d� |fS )z&Parse a timestamp from a binary packetr NrS �Hr re )�year�month�dayru � �IrH rB � )r� r� r� �hour�minute�second� microsecond)�datetime�dater )r r7 r� r� �value�mcsr r r �_parse_binary_timestamp~ s, �� z%MySQLProtocol._parse_binary_timestampc C s� |d }|d|d � }d}|dkr:t d|dd� �d }t d|dd� �d }|d dkrd|d9 }tj||d ||d |d d �}||d d� |fS ) z'Parse a time value from a binary packetr r rH r� NrB rI ru r� )�days�seconds� microseconds�minutes�hours)r r� � timedelta)r r7 r� r� r> r� r� �tmpr r r �_parse_binary_time� s �z MySQLProtocol._parse_binary_timec C sn t |�d d d }dd� |d|� D �}||d� }g }t|�D �] \}}|t|d d � d|d d > @ r~|�d� qBqB|d tjtjtjtjtj fkr�| � ||�\}}|�|� qB|d tj tj fkr�| � ||�\}}|�|� qB|d tjtjtjfk�r | �||�\}}|�|� qB|d tjk�rL| �||�\}}|�|� qBt�|�\}}|�|� qBt|�S ) z(Parse values from a binary result packetru rd rH c S s g | ] }t |��qS r )ro )rj r~ r r r rl � s z6MySQLProtocol._parse_binary_values.<locals>.<listcomp>r Nr )r# � enumeratero rw r r� r� r� r� r� r� r� �FLOATr� �DATETIME�DATE� TIMESTAMPr� �TIMEr� r rW �tuple) r �fieldsr7 Znull_bitmap_length� null_bitmap�values�posr� r� r r r �_parse_binary_values� s: $ � � z"MySQLProtocol._parse_binary_valuesc C s� g }d}d}d}|dk rq�||kr$q�|� � }|d dkrH| �|�}d}n$|d dkrld}| �||dd� �}|dkr�|dk r�|�|� n|dkr�|dkr�t|��|d7 }q||fS )zwRead MySQL binary protocol result Reads all or given number of binary resultset rows from the socket. Nr rS rb rB r )rv rh r� rw r ) r rz �columnsr[ r| r} r� r~ r7 r r r �read_binary_result� s* z MySQLProtocol.read_binary_resultc C s� |d dkst �d��i }zht�|dd� d�\}|d< t�|d�\}|d< t�|d�\}|d < |d d� }t�|d�\}|d < W n tk r� t �d��Y nX |S ) z'Parse a MySQL Binary Protocol OK packetrS r z Failed parsing Binary OK packetrB N� statement_idrd Z num_columnsZ num_paramsr rU )r r r Zread_intrX )r r7 Zok_pktr r r �parse_binary_prepare_ok� s z%MySQLProtocol.parse_binary_prepare_okc C s� d}d}|dk rX|dkr$d}t j}q�|dkr8d}t j}q�|dkrLd}t j}q�d }t j}nJd }|d krpd }t j}n2|d kr�d}t j}n |dkr�d}t j}n t j}d}t�||�||fS )z0Prepare an integer for the MySQL binary protocolNr i����r� i ���r� i �r~ r� � � �Bi�� r� l �� r� �Q)r r� r� r� r� r$ r% )r r� r_ r` r� r r r �_prepare_binary_integer� s6 z%MySQLProtocol._prepare_binary_integerc C s� t |tj�rtj}n t |tj�r(tj}ntd��t�|j �t� |j � t� |j � }t |tj�r�|t� |j � t� |j� t� |j� }|jdkr�|t�|j�7 }t� t|��| }||fS )a� Prepare 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 )� isinstancer� r r� r� r� rX r � int2storer� r: r� r� r� r� r� r� r9 r# )r r� r_ �packedr r r �_prepare_binary_timestamp s, � � � � � z'MySQLProtocol._prepare_binary_timestampc C s t |tjtjf�std��tj}d}d}d}t |tj�r�|jdk rFd}t|j d�\}}t|d�\}} |t � t |j��t � |� t � |� t � | � 7 }|j}n8|t � d�t � |j� t � |j� t � |j� 7 }|j}|r�|t � |�7 }t � |�| }t � t|��| }||fS )a� Prepare 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.timer Nr. r i �< )r� r� r� �timerX r r� r� �divmodr� r r9 �absr: r� r� r� r� r� r# ) r r� r_ �negativer� r� r� � remainderZminsZsecsr r r �_prepare_binary_time8 sD � ��� � � �z"MySQLProtocol._prepare_binary_timec C s t �|�t �|� | }|S )zMPrepare long data for prepared statements Returns a string. )r r9 r� )r Z statement�paramr> r7 r r r � _prepare_stmt_send_long_datae s ���z*MySQLProtocol._prepare_stmt_send_long_datar r c C s* d}dgt |�d d }g } g } d} |dkr2i }|�r�|�r�t |�t |�krXt�d��t|�D �]^\} } || }d}|dkr�|| d d| d > O < | �t�tj�t�|� � q`�n�| |kr�|| d r�tj }ntj }�n�t |t ��r | � |�\} }}| �| � �n�t |t��rdt�r8| �t�t |��| � n"|�|�}| �t�t |��| � tj}�nBt |t��r�| �t�t |��| � tj }�nt�r�t |t��r�|�|�}| �t�t |��| � tj}n�t |t��r| �t�t t|��|���t|��|� � tj}n�t |t��r4| �t�d|�� tj}nrt |tjtj f��rb| � |�\} }| �| � nDt |tj tj f��r�| � |�\} }| �| � nt�!d j"|j#j$d ���| �t�|�t�|� � q`t�%|�t�|� t�%|� d�&d d � |D �� t�d� }| D ]}||7 }�q| D ]}||7 }�q|S ) z6Make a MySQL packet with the Statement Execute commandr r ru rH r. NzTFailed executing prepared statement: data values does not match number of parametersr� z:MySQL binary protocol can not handle '{classname}' objects)� classnamec S s g | ]}t �d |��qS )r� )r$ r% )rj �bitr r r rl � s z3MySQLProtocol.make_stmt_execute.<locals>.<listcomp>)'r# r r r� rw r r: r ZNULLZBLOB�STRINGr� ro r� r! r Zlc_intr ZVARCHAR�bytes�unicoder ZDECIMAL�floatr$ r% r� r� r� r� r� r� r� r r � __class__�__name__r9 ry )r r� r> � parametersr` Zlong_data_usedr4 Ziteration_countr� r� �typesr� r� r] r� r_ r7 �a_typeZa_valuer r r �make_stmt_executep s� � � � � � � � �� � �� ������ z MySQLProtocol.make_stmt_executec C sZ |d dkst �d��tj|dd� dd�\}}|rL|d d krL|dd� }|�d �|fS ) z&Parse a MySQL AuthSwitchRequest-packetrS rb z'Failed parsing AuthSwitchRequest packetrB Nr rD rI r r )r r r rL rM )r r7 Z plugin_namer r r �parse_auth_switch_request� s � z'MySQLProtocol.parse_auth_switch_requestc C s" |d dkst �d��|dd� S )z!Parse a MySQL AuthMoreData-packetrS r z"Failed parsing AuthMoreData packetrB N)r r )r r7 r r r �parse_auth_more_data� s �z"MySQLProtocol.parse_auth_more_data)NNNr+ r r, FN)r+ r r, )N)NNNr+ r FN)rJ )T)r )r )r r r Nr ) r� � __module__� __qualname__�__doc__r r* r8 r; r? r@ rR rY r\ ra rh rt r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r ' s\ � % � � $, * !"- � Z r )r� r$ r� �decimalr � constantsr r r r r � r r Zauthenticationr Zcatch23r r r �objectr r r r r �<module> s