shell bypass 403
o
�\�> � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z
ddlm Z dd lm
Z dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZmZ ddlmZ ddl m Z m Z m Z ddl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& ej'j(ej)fej'j*ej+fej'j,ej-fej'j.ej/fej'j0e
j1fej'j2ejfej'j3ej4fej'j5ej6fej'j7ej8fej'j9ej:fej'j;ej<fg Z=ddd�Z>G dd� de?�Z@G dd� de?�ZAdS )z3Implementation of the X protocol for MySQL servers.� N� )�
mysqlx_pb2)�mysqlx_session_pb2)�mysqlx_sql_pb2)�mysqlx_notice_pb2)�mysqlx_datatypes_pb2)�mysqlx_resultset_pb2)�mysqlx_crud_pb2)�mysqlx_expr_pb2)�mysqlx_connection_pb2)�ColumnMetaData)� STRING_TYPES� INT_TYPES)�DbDoc)�InterfaceError�OperationalError�ProgrammingError)�
ExprParser�build_null_scalar�build_string_scalar�build_bool_scalar�build_double_scalar�build_int_scalar�utf-8c C s t | t�r| S | �|�S �N)�
isinstance�bytes�encode)�value�encoding� r �F/usr/local/python-3.10/lib/python3.10/site-packages/mysqlx/protocol.py�encode_to_bytes@ s r" c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �MessageReaderWriterc C s || _ d | _d S r )�_stream�_msg)�selfZ
socket_streamr r r! �__init__E s
z MessageReaderWriter.__init__c C s | j d ur td��|| _ d S )NzMessage push slot is full)r% r �r& �msgr r r! � push_messageI s
z MessageReaderWriter.push_messagec C s" | j d ur
| j }d | _ |S | �� S r )r% �
_read_message)r& �mr r r! � read_messageN s
z MessageReaderWriter.read_messagec C sl | j �d�}t�d|�\}}| j �|d �}tD ]}|d |kr.|d � }|�|� | S qtd�|���)N� �<LBr r zUnknown msg_type: {0})r$ �read�struct�unpack�_SERVER_MESSAGES�ParseFromString�
ValueError�format)r& �hdrZmsg_lenZmsg_type�payloadZ msg_tupler) r r r! r+ U s
�z!MessageReaderWriter._read_messagec C s8 |� � }t�dt|�d |�}| j�d�||g�� d S )Nr/ r � )ZSerializeToStringr1 �pack�lenr$ �sendall�join)r& Zmsg_idr) Zmsg_str�headerr r r! �
write_messageb s z!MessageReaderWriter.write_messageN)�__name__�
__module__� __qualname__r' r* r- r+ r? r r r r! r# D s
r# c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1S )2�Protocolc C s || _ || _d | _d S r )�_reader�_writerZ_message)r& Z
reader_writerr r r! r' i s
zProtocol.__init__c C s$ t �� }| j�tjj|� | j�� S r ) �MySQLxConnectionZCapabilitiesGetrE r? �MySQLx�ClientMessagesZCON_CAPABILITIES_GETrD r- r( r r r! �get_capabilitesn s
�
zProtocol.get_capabilitesc K s\ t �� }|�� D ]\}}| �|�}t j||d�}|jj�|g� q| j�t j
j |� | � � S )N)�namer )
rF ZCapabilitiesSet�items� _create_anyZ
CapabilityZ capabilities�extendrE r? rG rH ZCON_CAPABILITIES_SET�read_ok)r& �kwargsr) �keyr Z
capabilityr r r! �set_capabilitiest s
�zProtocol.set_capabilitiesc C s" t j|d�}| j�tjj|� d S )N)Z mech_name)�
MySQLxSessionZAuthenticateStartrE r? rG rH ZSESS_AUTHENTICATE_START)r& �methodr) r r r! �send_auth_start� s �zProtocol.send_auth_startc C s$ | j �� }t|tj�std��|jS )Nz>Unexpected message encountered during authentication handshake)rD r- r rR �AuthenticateContinuer � auth_datar( r r r! �read_auth_continue� s
zProtocol.read_auth_continuec C s& t jt|�d�}| j�tjj|� d S )N)rV )rR rU r" rE r? rG rH �SESS_AUTHENTICATE_CONTINUE)r& �datar) r r r! �send_auth_continue� s ��zProtocol.send_auth_continuec C s4 | j �� }t|tj�rd S t|tj�rt|j��qr ) rD r- r rR �AuthenticateOkrG �Errorr r) r( r r r! � read_auth_ok� s
�zProtocol.read_auth_okc C sf t |j�}|d g }|jD ]#}|d }||jvr td�|���|j| }| �|d |j �||< q
|S )NrJ z-Unable to find placeholder for parameter: {0}r )r; Z _binding_mapZ _bindingsr r6 �arg_object_to_scalar�
_doc_based)r& � statement�countZscalarsZbindingrJ �posr r r! �get_binding_scalars� s
�
�z Protocol.get_binding_scalarsc C s� |j r
|j�|j� |jr|j�| �|�� |jr#|j |j
_ |j |j
_
|jr-|j�|j� |jr7|j�|j� |jrC|j�|j� d S d S r )Z
_has_whereZcriteria�CopyFromZ _where_exprZ
_has_bindings�argsrM rc Z
_has_limitZ_limit_row_count�limitZ row_countZ
_limit_offset�offsetZ _has_sort�orderZ
_sort_exprZ
_has_group_by�groupingZ _groupingZ _has_havingZgrouping_criteriaZ_having)r& �messager` r r r! �
_apply_filter� s
�zProtocol._apply_filterc C sd t j|jrt jnt jt j|jj|jjd�d�}|j r!|j
� |j � | �
||� | j�tjj|� d S �N�rJ �schema�Z
data_modelZ
collection)�
MySQLxCrudZFindr_ �DOCUMENT�TABLE�
Collection�targetrJ rn Z_has_projection�
projectionrM Z_projection_exprrk rE r? rG rH Z CRUD_FIND)r& �stmt�findr r r! � send_find� s �
�� zProtocol.send_findc C s� t j|jrt jnt jt j|jj|jjd�d�}| � ||� |j
D ]$}t j |j |j
d�}|jd ur=|j�| �|j|j �� |j�|g� q | j�tjj|� d S )Nrm ro )� operation�source)rp ZUpdater_ rq rr rs rt rJ rn rk Z _update_opsZUpdateOperationZ update_typerz r rd �arg_object_to_exprry rM rE r? rG rH Z CRUD_UPDATE)r& r` �updateZ update_opZopexprr r r! � send_update� s* �
��
�
��zProtocol.send_updatec C sP t j|jrt jnt jt j|jj|jjd�d�}| � ||� | j
� t j
j|� d S rl )rp �Deleter_ rq rr rs rt rJ rn rk rE r? rG rH Z CRUD_DELETE)r& rv �deleter r r! � send_delete� s �
�� zProtocol.send_deletec C sL t j|t|�dd�}|D ]}| �|�}|j�|g� q | j�tj j
|� d S )NF)� namespacerv Zcompact_metadata) � MySQLxSQLZ StmtExecuter" rL re rM rE r? rG rH ZSQL_STMT_EXECUTE)r& r� rv re �argr r r r! �send_execute_statement� s �
�z Protocol.send_execute_statementc C s� t j|jrt jnt jt j|jj|jjd�d�}t |d�r0|j
D ]}|j � t
||j ��� g� q |jD ]7}t j�� }t|t�rT|D ]}| �||j �}|j� |g� qAn| �||j �}|j� |g� |j� |g� q3| j�tjj|� d S )Nrm ro �_fields)rp ZInsertr_ rq rr rs rt rJ rn �hasattrr� ru rM r Zparse_table_insert_fieldZ_valuesZTypedRowr �listr{ �field�rowrE r? rG rH Z CRUD_INSERT)r& r` �insertr� r r� �val�objr r r! � send_insert� s6 �
��
�
��zProtocol.send_insertc C st t |t�r tjjt|�d�}tjd|d�}tjd|d�S t |t�r*tjdt|�d�S t |t �r8tjdt
|�d�S d S )N)r � )�typeZv_stringr )r� �scalar) r r
�MySQLxDatatypesZScalar�Stringr" �Any�boolr r r )r& r� r� r� r r r! rL � s
zProtocol._create_anyc C s | � |�}|d ur
td��d S )Nz Expected to close the result)r+ r �r& �rsr) r r r! � close_result s
�zProtocol.close_resultc C s6 | � |�}|d u r d S t|tj�r|S | j�|� d S r )r+ r �MySQLxResultset�RowrD r* r� r r r! �read_row s
zProtocol.read_rowc C s� |j dkr t�� }|�|j� |j�t|j|j|j �� d S |j dkr/t�
� }|�|j� d S |j dkrZt� � }|�|j� |j tj j
krL|jj|_d S |j tj jkr\|jj|_d S d S d S )Nr � � )r� � MySQLxNotice�Warningr4 r8 � _warnings�append�level�coder) ZSessionVariableChangedZSessionStateChanged�paramZ
ROWS_AFFECTEDr Zv_unsigned_intZ_rows_affectedZGENERATED_INSERT_IDZ
_generated_id)r& r) r� Z
warningMsgZ
sessVarMsgZ sessStateMsgr r r! �_process_frame s,
�
���zProtocol._process_framec C s| | j �� }t|tj�rt|j��t|tj�r | � ||� n t|t
j �r&d S t|t j
�r0d|_n
t|t j�r:d|_n |S q)NT)rD r- r rG r\ r r) r� �Framer� r� �
StmtExecuteOkr� � FetchDone�_closed�FetchDoneMoreResultsetsZ_has_more_resultsr� r r r! r+ * s
�zProtocol._read_messagec
C s� g } | � |�}|d u r |S t|tj�r | j�|� |S t|tj�s(td��t|j|j |j
|j |j |j
|j|j|j|j|j|j� }|�|� q)NTzUnexpected msg type)r+ r r� r� rD r* r r r� �catalogrn �tableZoriginal_tablerJ �
original_name�lengthZ collationZfractional_digits�flags� content_typer� )r& r� �columnsr) �colr r r! �get_column_metadata; s(
� � �
�z Protocol.get_column_metadatac C s |d u rt jt jjt� d�S t|t�r t jt jjt|�d�S t|t�r.t jt jjt|�d�S t|t �r>t jt jjt
|�d�S t|t �rnzt ||��
� }|�� r[t jt jjt|�d�W S |W S t jt jjt|�d� Y S t|t�r�t jt jjtt|��d�S td�t|����)N)r� �literalzUnsupported type: {0})�
MySQLxExpr�Expr�LITERALr r r� r r r �floatr r
r �exprZhas_identifierr r �strr r6 r� )r& r �allow_relationalZ
expressionr r r! r{ N sB
�
�
�
�
�
�
�zProtocol.arg_object_to_exprc C s | � ||�jS r )r{ r� )r& r r� r r r! r^ j s z Protocol.arg_object_to_scalarc C s8 | j �� }t|tj�rt|j��t|tj�std��d S )Nz Unexpected message encountered)rD r- r rG r\ r r) �Okr( r r r! rN m s
�zProtocol.read_okc C s t �� }| j�tjj|� d S r )rR �CloserE r? rG rH Z
SESS_CLOSEr( r r r! �
send_closeu s zProtocol.send_closeN)r@ rA rB r' rI rQ rT rW rZ r] rc rk rx r} r� r� r� rL r� r� r� r+ r� r{ r^ rN r� r r r r! rC h s2
rC )r )B�__doc__r1 Zprotobufr rG r rR r r� r r� r r� r r� r rp r
r� r rF �resultr �compatr
r Zdbdocr �errorsr r r r� r r r r r r ZServerMessagesrX rU ZSESS_AUTHENTICATE_OKr[ ZSQL_STMT_EXECUTE_OKr� �ERRORr\ ZNOTICEr� ZRESULTSET_COLUMN_META_DATAZ
RESULTSET_ROWr� ZRESULTSET_FETCH_DONEr� Z$RESULTSET_FETCH_DONE_MORE_RESULTSETSr� ZOKr� ZCONN_CAPABILITIESZ Capabilitiesr3 r" �objectr# rC r r r r! �<module> sN �� � � �
$