관리-도구
편집 파일: network.cpython-38.pyc
U )�e�J � @ s� d Z ddlmZ ddlZddlZddlZddlZz ddlZW n Y nX ddlm Z m Z ddl m Z m Z mZ dd� Zd d � ZG d d � d e�ZG d d� de�ZG dd� de�ZdS )zGModule implementing low-level socket communication with MySQL servers. � )�dequeN� )� constants�errors)�PY2�init_bytearray� struct_unpackc C s | j st| �S dj| j | jd�S )z_Reformat the IOError error message This function reformats the IOError error message. z{errno} {strerr})�errnoZstrerr)r �str�format�strerror)�err� r �l/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/network.py� _strioerror+ s r c C s� g }t | �}tj}||krZ|�dt�d|� | d|� � | |d� } t | �}|d }q|�t�d|�dd� t�d|� | � |S )z0Prepare a packet for sending to the MySQL server� ����<BNr �<Ir � )�lenr �MAX_PACKET_LENGTH�append�struct�pack)�buf�pktnr�pkts�pllen� maxpktlenr r r �_prepare_packets5 s � ��r c @ s� e Zd ZdZdd� Zedd� �Zedd� �Zdd � Zd d � Z d d � Z dd� Z d%dd�Z e Z d&dd�Zdd� Zdd� Zejdd� dkr�eZeZneZd d � Zd d � Zd d!� Zd'd#d$�ZdS )(�BaseMySQLSocketa Base class for MySQL socket communication This class should not be used directly but overloaded, changing the at least the open_connection()-method. Examples of subclasses are mysql.connector.network.MySQLTCPSocket mysql.connector.network.MySQLUnixSocket c C s* d | _ d | _d| _d| _t� | _d| _d S )N���i )�sock�_connection_timeout�_packet_number�_compressed_packet_numberr � _packet_queueZrecvsize��selfr r r �__init__N s zBaseMySQLSocket.__init__c C s" | j d | _ | j dkr d| _ | j S )z Increments the packet numberr � r )r$ r' r r r �next_packet_numberV s z"BaseMySQLSocket.next_packet_numberc C s" | j d | _ | j dkr d| _ | j S )z'Increments the compressed packet numberr r* r )r% r' r r r � next_compressed_packet_number^ s z-BaseMySQLSocket.next_compressed_packet_numberc C s t �dS )zOpen the socketN��NotImplementedErrorr' r r r �open_connectionf s z BaseMySQLSocket.open_connectionc C s t �dS )z Get the location of the socketNr- r' r r r � get_addressj s zBaseMySQLSocket.get_addressc C s@ z | j �tj� | j �� | `W n tjtfk r: Y nX dS )z&Shut down the socket before closing itN)r" �shutdown�socket� SHUT_RDWR�closer& �error�AttributeErrorr' r r r r1 n s zBaseMySQLSocket.shutdownc C s2 z| j �� | `W n tjtfk r, Y nX dS )zClose the socketN)r" r4 r& r2 r5 r6 r' r r r �close_connectionw s z BaseMySQLSocket.close_connectionNc C s� |dkr| j n|| _t|| j�}|D ]�}z&trB| j�t|�� n | j�|� W q& tk r� } z tj d| � � t |�fd��W 5 d}~X Y q& t k r� tj dd��Y q&X q&dS )z Send packets to the MySQL serverN� �r �values�� �r ) r+ r$ r r r" �sendall�buffer�IOErrorr �OperationalErrorr0 r r6 )r( r � packet_number�compressed_packet_number�packets�packetr r r r � send_plain s �zBaseMySQLSocket.send_plainc C sH |dkr| j n|| _|dkr&| j n|| _| j}t|�}g }tj}||k�r�t||�}tr~t � } |D ] } | | 7 } qft | �} n d� |�} ~t � | dd� �} t�dt| ��dd� t�d| j� d } tr�t | �} |�| | � | dd� } t| �}| j ||k�rxt � | d|� �} t�dt| ��dd� t�d| j� d } t�rNt | �} |�| | � | |d� } t| �}| j q�| �r�t � | �} t�dt| ��dd� t�d| j� t�d|�dd� } t�r�t | �} |�| | � ~ n�t�d|�dd� t�d|� | } t�rt | �} t| �}|d k�rvt � | �} |�t�dt| ��dd� t�d| j� t�d|�dd� | � nRt�d|�dd� t�d| j� t�dd�dd� } t�r�t | �} |�| | � |D ]v} z| j�| � W n^ tk �r } z tjd | �� t|�fd ��W 5 d}~X Y n" tk �r> tjd d��Y nX �q�dS )z+Send compressed packets to the MySQL serverN� � @ r r r r s @ r �2 r8 r9 r; r< )r+ r$ r, r% r r r r r � bytearrayr> �join�zlib�compressr r r r" r= r? r r@ r0 r r6 )r( r rA rB r r Zzpktsr r Ztmpbuf�pktZzbuf�headerZ zip_packetr r r r �send_compressed� s� �� �� �� �� ��� �� �z BaseMySQLSocket.send_compressedc C sJ �zt d�}d}|dk rJ| j�d| �}|s8tjdd��||7 }t|�}q|d | _trzt� dt |dd� d ��d }n t� d|dd� d �d }|}|� t |�� t |�}|dd � }|�r| j�||�}|dkr�|dkr�tjdd��||d � }||8 }q�|W S tk �rD } z tjd | �� t|�fd ��W 5 d }~X Y nX d S ) �%Receive packets from the MySQL serverrF r � �� r< r r � Nr8 r9 )rI r" �recvr �InterfaceErrorr r$ r r � unpack_fromr> �unpack�extend� memoryview� recv_intor? r@ r0 r ) r( rD Z packet_len�chunk� payload_len�restZ packet_view�readr r r r � recv_plain� sD �� �zBaseMySQLSocket.recv_plainc C s� z�t d�}d}|dk rH| j�d| �}|s6tjdd��||7 }t|�}q|d | _td|dd� d �d }|}td�}|dkr�| j�|�}|s�tjdd��||7 }|t|� }qx|| W S t k r� } z tj d | � � t |�fd ��W 5 d }~X Y nX d S ) rP rF r rQ rR r< r r rS r8 r9 N) rI r" rT r rU r r$ r r r? r@ r0 r )r( rN Z header_lenr[ r\ r] �payloadr r r r �recv_py26_plain s2 �z BaseMySQLSocket.recv_py26_plainr � )rb � c C sp |rlt r&t�d|dd� d �d }n t�d|dd� d �d }| j�|d|d � � ||d d� }q dS )zSplit compressed payloadr r r rS rQ N)r r rV rW r& r )r( Z packet_bunch�payload_lengthr r r �_split_zipped_payload5 s �� z%BaseMySQLSocket._split_zipped_payloadc C sh z| j �� }|d | _|W S tk r. Y nX td�}g }�zh| j�d�}|rrt|�dk rr||7 }| j�d�}qL|�r�t|�dk r�tj dd��t d|dd� d �d }|d | _ t d|d d� d �d }t |�}t|�|k �r | j�|t|� �}t|�dk�rtj dd��|| }q�|dk�rJ| � |� | j �� }|d | _|W S |�||f� |d k�rf�q�t d�}| j�d�}|rrt|�dk rr||7 }| j�d�}�qzqrW n> tk �r� } z tjd | �� t| �fd ��W 5 d} ~ X Y nX t d�} |D ]2\}} t�r| t�t| ��7 } n| t�| �7 } �q�| � | � ~ z| j �� }|d | _|W S tk �rb Y nX dS )z0Receive compressed packets from the MySQL serverr rF r � rR r< r r rS rQ rG r8 r9 N)r& �popleftr$ � IndexErrorrI r" rT r r rU r r% r re r r? r@ r0 r r rK � decompressr> ) r( rM rN rC ZabyteZzip_payload_lengthrd Z zip_payloadr[ r �tmpr` r r r �recv_compressedB s| �� � � z BaseMySQLSocket.recv_compressedc C s || _ dS )zSet the connection timeoutN)r# )r( �timeoutr r r �set_connection_timeout� s z&BaseMySQLSocket.set_connection_timeoutFc C s� | j stjdd��z@|r tj}ntj}tj| j ||||dtj|d�| _ | j �� W n� t k rp t� d��Y np tj t fk r� } z tjd| � � t|�fd��W 5 d}~X Y n0 tk r� } zt�t|���W 5 d}~X Y nX dS ) z Switch the socket to use SSLi r< F)�keyfile�certfile�ca_certs� cert_reqs�do_handshake_on_connect� ssl_version�ciphersz&Python installation has no SSL supportr8 r9 N)r" r rU �ssl� CERT_REQUIRED� CERT_NONE� wrap_socket�PROTOCOL_TLSv1� do_handshake� NameErrorZNotSupportedError�SSLErrorr? r0 r r. r )r( �ca�cert�keyZ verify_cert�cipherrq r r r r � switch_to_ssl� s8 �� �z BaseMySQLSocket.switch_to_ssl)NN)NN)FN)�__name__� __module__� __qualname__�__doc__r) �propertyr+ r, r/ r0 r1 r7 rE �sendrO r_ ra �sys� version_inforT re rk rm r� r r r r r E s6 � � U& Or c s2 e Zd ZdZd � fdd� Zdd� Zdd� Z� ZS ) �MySQLUnixSocketzoMySQL socket class using UNIX sockets Opens a connection through the UNIX socket of the MySQL Server. �/tmp/mysql.sockc s t t| ��� || _d S �N)�superr� r) � unix_socket)r( r� �� __class__r r r) � s zMySQLUnixSocket.__init__c C s | j S r� )r� r' r r r r0 � s zMySQLUnixSocket.get_addressc C s� z2t � t jt j�| _| j�| j� | j�| j� W nj tk rn } z t j d| � � t |�fd��W 5 d }~X Y n0 t k r� } zt � t|���W 5 d }~X Y nX d S )Ni� r9 )r2 �AF_UNIX� SOCK_STREAMr" � settimeoutr# �connectr� r? r rU r0 r � Exceptionr )r( r r r r r/ � s �z MySQLUnixSocket.open_connection)r� �r� r� r� r� r) r0 r/ � __classcell__r r r� r r� � s r� c s2 e Zd ZdZd � fdd� Zdd� Zd d � Z� ZS ) �MySQLTCPSocketzXMySQL socket class using TCP/IP Opens a TCP/IP connection to the MySQL Server. � 127.0.0.1�� Fc s* t t| ��� || _|| _|| _d| _d S )Nr )r� r� r) � server_host� server_port� force_ipv6�_family)r( �host�portr� r� r r r) � s zMySQLTCPSocket.__init__c C s d� | j| j�S )Nz{0}:{1})r r� r� r' r r r r0 � s zMySQLTCPSocket.get_addressc C s� dgd }z�t �| j| jdt jt j�}|D ]8}| jrL|d t jkrL|} qdq*|d t jkr*|} qdq*| jr�|d dkr�t � d� | j���|d dkr�|d }W n< t k r� } z t j d| � � t|�fd��W 5 d}~X Y nX |\| _}}}}z0t � | j||�| _| j�| j� | j�|� W nn t k �rZ } z t j d| � � t|�fd��W 5 d}~X Y n2 tk �r� } zt �t|���W 5 d}~X Y nX dS )z7Open the TCP/IP connection to the MySQL server N� r z No IPv6 address found for {0}i� r9 )r2 � getaddrinfor� r� r� �SOL_TCPr� �AF_INET6�AF_INETr rU r r? r0 r r� r" r� r# r� r� r@ r ) r( �addrinfo� addrinfos�infor �socktype�proto�_�sockaddrr r r r/ � sL � � � �z MySQLTCPSocket.open_connection)r� r� Fr� r r r� r r� � s r� )r� � collectionsr r2 r r� rK ru � r r Zcatch23r r r r r �objectr r� r� r r r r �<module> s$ p