관리-도구
편집 파일: utils.cpython-38.pyc
U )�e�# � @ s� d Z ddlmZ dZddlZddlmZ dd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd(d d �Zd d � Zd!d"� Zd#d$� Zd)d&d'�ZdS )*z Utilities � )�print_functionFN� )� struct_unpackc C s� zvt | t�r| W S t| �}|dkr,| d W S |dkrT| dd| }td|�d W S | dd| }td|�d W S W n � Y nX dS ) z&Unpacks the given buffer to an integerr r � � �<I� �<QN)� isinstance�int�lenr )�buf�length�tmp� r �j/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysql/connector/utils.py�intread# s r c C s. | dk s| dkrt d��ntt�d| ��S dS )z^ Takes an unsigned byte (1 byte) and packs it as a bytes-object. Returns string. r � z int1store requires 0 <= i <= 255�<BN�� ValueError� bytearray�struct�pack��ir r r � int1store5 s r c C s. | dk s| dkrt d��ntt�d| ��S dS )z` Takes an unsigned short (2 bytes) and packs it as a bytes-object. Returns string. r � z"int2store requires 0 <= i <= 65535�<HNr r r r r � int2storeA s r c C s6 | dk s| dkrt d��ntt�d| �dd� �S dS )zb Takes an unsigned integer (3 bytes) and packs it as a bytes-object. Returns string. r �� z%int3store requires 0 <= i <= 16777215r � Nr r r r r � int3storeM s r" c C s. | dk s| dkrt d��ntt�d| ��S dS )zb Takes an unsigned integer (4 bytes) and packs it as a bytes-object. Returns string. r � �� z'int4store requires 0 <= i <= 4294967295r Nr r r r r � int4storeY s r$ c C s. | dk s| dkrt d��ntt�d| ��S dS )zZ Takes an unsigned integer (8 bytes) and packs it as string. Returns string. r � z!int8store requires 0 <= i <= 2^64r Nr r r r r � int8storee s r& c C s\ | dk s| dkrt d��| dkr&t}n.| dkr4t}n | dkrBt}n| dkrPt}nt}|| �S )z� Takes an unsigned integers and packs it as a bytes-object. This function uses int1store, int2store, int3store, int4store or int8store depending on the integer value. returns string. r r% z!intstore requires 0 <= i <= 2^64r r r r# )r r r r" r$ r& )r Z formed_stringr r r �intstoreq s r' c C s� | dk s| dkrt d��| dk r0tt�d| ��S | dkrLdtt�d| �� S | d krpd tt�d | �dd � � S d tt�d| �� S dS )z| Takes an unsigned integer and packs it as bytes, with the information of how much bytes the encoded int takes. r r% zRequires 0 <= i <= 2^64� r r � �r r � �r r! � �r Nr r r r r �lc_int� s r, c C s | d|� }| |d� |fS )zi Reads bytes from a buffer. Returns a tuple with buffer less the read bytes, and the bytes. r Nr �r �size�resr r r � read_bytes� s r0 c C s� | d dkr | dd� dfS d }}| d }|dkrX|}| d| d� | d|d � fS |dkrfd}n |dkrrd }|d kr~d }t | d|d � �}| || d d� | |d || d � fS ) u� Takes a buffer and reads a length coded string from the start. This is how Length coded strings work If the string is 250 bytes long or smaller, then it looks like this: <-- 1b --> +----------+------------------------- | length | a string goes here +----------+------------------------- If the string is bigger than 250, then it looks like this: <- 1b -><- 2/3/8 -> +------+-----------+------------------------- | type | length | a string goes here +------+-----------+------------------------- if type == ü: length is code in next 2 bytes elif type == ý: length is code in next 3 bytes elif type == þ: length is code in next 8 bytes NULL has a special value. If the buffer starts with û then it's a NULL and we return None as value. Returns a tuple (trucated buffer, bytes). r r( r N� � � � r! � r �r )r r �lsizeZfstr r r �read_lc_string� s r8 c C s g }dddd�}t | �}d}||k �r | | }|dkr:dS |dkrV|�d� |d 7 }q |d kr�|}|�| |d ||d � � |d | 7 }q d}z || }W n tk r� Y dS X t| |d ||d � �}|�| |d | || |d � � |d | | 7 }q t|�S ) zXReads all length encoded strings from the given buffer Returns a list of bytes r3 r! r )r2 r4 r5 r r Nr( r r1 )r �append�KeyErrorr �tuple)r Zbyteslst�sizes�buf_len�pos�firstr r7 r r r �read_lc_string_list� s4 �r@ c C s� |dkr|dkrt d��|dk rhz| �|�}W n t k rJ t d��Y nX | |d d� | d|� fS |dk rzt| |�S t d��dS )zr Reads a string up until a character or for a given size. Returns a tuple (trucated buffer, string). Nz&read_string() needs either end or sizez end byte not present in bufferr r z.read_string() needs either end or size (weird))r �indexr0 )r �endr. �idxr r r � read_string s rD c C s4 zt | d|� �}W n � Y nX | |d� |fS )zMRead an integer from buffer Returns a tuple (truncated buffer, int) r Nr6 r- r r r �read_int s rE c C s� | s t d��| d }|dkr,| dd� dfS |dk rH| dd� t|�fS |dkrr| dd� td| dd� �d fS |d kr�| d d� td | dd � d �d fS |d kr�| dd� td| dd� �d fS t d��dS )z� Takes a buffer and reads an length code string from the start. Returns a tuple with buffer less the integer and the integer read. z Empty buffer.r r( r Nr2 r! z<xHr4 r r r r5 � z<xQz%Failed reading length encoded integer)r r r )r Zlcbyter r r � read_lc_int( s "&"rG c C s2 t | t�s d�dd� | D ��S d�dd� | D ��S )z"Debug function for showing buffers� c S s g | ] }d | �qS �z\x%02xr ��.0�cr r r � <listcomp>F s z"_digest_buffer.<locals>.<listcomp>c S s g | ]}d t |� �qS rI )�ordrJ r r r rM G s )r �str�join)r r r r �_digest_bufferC s rQ � c C sL |r<|r"|dkr"t | d|� �}nt | �}t|d | � n tt | �� dS )z2Debug function printing output of _digest_buffer()r z: N)rQ �print)Zabuffer�prefix�limit�digestr r r � print_bufferI s rW )NN)NrR )�__doc__� __future__r Z__MYSQL_DEBUG__r Zcatch23r r r r r" r$ r& r' r, r0 r8 r@ rD rE rG rQ rW r r r r �<module> s( 5)