shell bypass 403
o
�\�@ � @ s" d Z zddlZdZW n dZY ddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z dd l
mZ dd
lmZmZ dd lmZmZmZ dd lmZmZ d
ZdZG dd� de�Z dd� Z G dd� de�Z G dd� de �Z G dd� de �Z G dd� de�Z!G dd � d e!�Z"G d d � d e!�Z#dS ) z4Implementation of communication for MySQL X servers.� NTF��wraps� )�MySQL41AuthPlugin)�InterfaceError�OperationalError�ProgrammingError��Schema)�Protocol�MessageReaderWriter)�Result� RowResult� DocResult)� SqlStatement� AddStatementz DROP DATABASE IF EXISTS `{0}`z#CREATE DATABASE IF NOT EXISTS `{0}`c @ s@ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zi fd d �Zd
S )� SocketStreamc C s d | _ d| _d S )NF)�_socket�_is_ssl��self� r �H/usr/local/python-3.10/lib/python3.10/site-packages/mysqlx/connection.py�__init__2 s
zSocketStream.__init__c C s6 t |t�rtjntj}t�|tj�| _| j�|� d S �N)�
isinstance�tuple�socket�AF_INET�AF_UNIX� SOCK_STREAMr �connect)r �paramsZs_typer r r r! 6 s zSocketStream.connectc C sb | j d u r td��g }|dkr,| j �|�}|dkr td��|�|� |t|�8 }|dksd�|�S )N� MySQLx Connection not availabler � zUnexpected connection close)r r �recv� RuntimeError�append�len�join)r �count�buf�datar r r �read; s
�
zSocketStream.readc C s" | j d u r td��| j �|� d S )Nr# )r r �sendall�r r, r r r r. G s
zSocketStream.sendallc C s | j sd S | j �� d | _ d S r )r �closer r r r r0 L s
zSocketStream.closec C s2 t s
| �� td��t�tj�}|�� d|v r8z
|�|d � tj|_ W n t
tj fy7 | �� t d��w d|v r\z
|�|d � tj
|_W n t
tj fy[ | �� t d��w d|v r�z|�|d |�dd �� W q� t
tj fy� | �� t d��w d|v r�| �� t d ��|�| j�| _d
| _d S ) Nz'Python installation has no SSL support.zssl-cazInvalid CA certificate.zssl-crlz Invalid CRL.zssl-certzssl-keyz Invalid Client Certificate/Key.z Client Certificate not provided.T)�
SSL_AVAILABLEr0 r& �sslZ
SSLContextZPROTOCOL_SSLv23Zload_default_certsZload_verify_locationsZ
CERT_REQUIREDZ verify_mode�IOErrorZSSLErrorr ZVERIFY_CRL_CHECK_CHAINZ verify_flagsZload_cert_chain�getZ wrap_socketr r )r Zssl_opts�contextr r r �set_sslS sF � �
��
zSocketStream.set_sslN) �__name__�
__module__� __qualname__r r! r- r. r0 r6 r r r r r 1 s r c s t � �� fdd��}|S )Nc s> z � | g|�R i |��W S t jtfy | �� td��w )NzCannot connect to host.)r �errorr& �
disconnectr )r �args�kwargs��funcr r �wrapperx s �z(catch_network_exception.<locals>.wrapperr )r? r@ r r> r �catch_network_exceptionw s rA c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze d
d� �Z
e dd� �Z e dd� �Z e dd� �Z
e dd� �Ze d+dd��Ze dd� �Ze d d � �Ze d d � �Ze d d!� �Ze d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*S ),�
Connectionc C sH |� d�| _|� d�| _|� d�| _d | _|| _t� | _d | _d | _ d S )N�user�password�schema)
r4 �_user� _passwordZ_schema�_active_result�settingsr �stream�
reader_writer�protocol�r rI r r r r � s
zConnection.__init__c C s" | j d ur| j �� d | _ d S d S r )rH � fetch_allr r r r �fetch_active_result� s
�z Connection.fetch_active_resultc C s: d| j v r| j d | j �dd�fS d| j v r| j d S dS )N�host�port�$� r )Z localhostrR )rI r4 r r r r �_connection_params� s
z Connection._connection_paramsc C s< | j �| �� � t| j �| _t| j�| _| �� | �� d S r ) rJ r! rS r rK r rL �_handle_capabilities�
_authenticater r r r r! � s
zConnection.connectc C s� | j �� j}|r|d j�� dks$n | j�dd�r"| �� td��d S t j
dk r:| j�dd�r8| �� t d��d S | j j dd � | j
�| j� d S )
Nr �tlsFz
ssl-enablezSSL not enabled at server.)� � � z=The support for SSL is not available for this Python version.T)rV )rL Zget_capabilitesZ capabilities�name�lowerrI r4 r0 r �sys� version_infor& Zset_capabilitiesrJ r6 r/ r r r rT � s
z Connection._handle_capabilitiesc C sH t | j| j�}| j�|�� � | j�� }| j�|�|�� | j� � d S r )
r rF rG rL Zsend_auth_startZ auth_nameZread_auth_continueZsend_auth_continueZ build_authentication_responseZ read_auth_ok)r Zplugin�
extra_datar r r rU � s
�zConnection._authenticatec G s | j �d||� d S )N�sql)rL �send_execute_statement)r r_ r< r r r �send_sql� s zConnection.send_sqlc C s* | j �|� d }t|t�r|j}t| |�S r )rL � send_insertr r Z_idsr
)r � statementZidsr r r rb � s
zConnection.send_insertc C s" | j �|� |jr
t| �S t| �S r )rL Z send_findZ
_doc_basedr r �r rc r r r �find� s zConnection.findc C � | j �|� t| �S r )rL Z send_deleter
rd r r r �delete� � zConnection.deletec C rf r )rL Z send_updater
rd r r r �update� rh zConnection.updateTc G s | j �|||� t| �S r )rL r` r
)r � namespace�cmdZ
raise_on_failr< r r r �execute_nonquery� � zConnection.execute_nonqueryc G s> | j �d||� t| �}|�� |jdkrtd��|d d S )Nr_ r z
No data found)rL r` r rN r* r )r r_ r< �resultr r r �execute_sql_scalar� s
z Connection.execute_sql_scalarc G s | j �d||� t| �S )NZxplugin)rL r` r )r rk r< r r r �get_row_result� rm zConnection.get_row_resultc C � | j �|�S r )rL �read_row�r rn r r r rr � � zConnection.read_rowc C � | j �|� d S r )rL � close_resultrs r r r rv � s zConnection.close_resultc C rq r )rL �get_column_metadatars r r r rw � rt z Connection.get_column_metadatac C s | j jd uS r )rJ r r r r r �is_open� s zConnection.is_openc C s | � � sd S | j�� d S r )rx rJ r0 r r r r r; � s zConnection.disconnectc C sB | � � sd S | jd ur| j�� | j�� | j�� | j�� d S r )rx rH rN rL Z
send_close�read_okrJ r0 r r r r r0 � s
zConnection.closeN)T)r7 r8 r9 r rO rS r! rT rU rA ra rb re rg ri rl ro rp rr rv rw rx r; r0 r r r r rB � s@
rB c s\ e Zd Z� fdd�Zdd� Z� fdd�Z� fdd�Zd d
� Z� fd d �Z� fd
d�Z � Z
S )� XConnectionc s� t t| ��|� g | _|�dg �| _d|v r*|d r*| j�|�d�|�dd �d�� d| _d| _| � � | jj
dd� dd � d S )
NZroutersrP rQ )rP rQ ���Tc S s | d S )N�priorityr )�xr r r �<lambda> s z&XConnection.__init__.<locals>.<lambda>)�key�reverse) �superrz r �dependent_connections�pop�_routersr' � _cur_router�
_can_failover�_ensure_priorities�sortrM �� __class__r r r s
�zXConnection.__init__c C s� d}d}| j D ]"}|�dd �}|d u r |d7 }||d< n |dkr%tdd��|d8 }qd| k r7t| j �k r?n d S tdd��d S ) Nr �d r| r z(The priorities must be between 0 and 100i� z\You must either assign no priority to any of the routers or give a priority for every routeri� )r� r4 r r( )r Zpriority_countr| �routerZprir r r r� s"
�
��z XConnection._ensure_prioritiesc s� | j s
d| _tt| ��� S | jr| jdu r%d| _d| _| j D ]}d|d< q | jd7 _| j | j d }| j | j d }| jdkrKd| j | jd d< | jt| j �d krXd| _||fS ) NFr{ TZ availabler rP rQ r )r� r� r� rz rS r� r( )r r� rP rQ r� r r rS ) s
z XConnection._connection_paramsc
sr d }| j r&ztt| ��� W S tjy" } z|}W Y d }~nd }~ww | j st| j�dkr4td� |���tdd��)Nr �Cannot connect to host: {0}z(Failed to connect to any of the routers.i� )
r� r� rz r! r r: r( r� r �format)r r: �errr� r r r! @ s ���
zXConnection.connectc C ru r )r� r' )r Z
connectionr r r �bind_connectionM s zXConnection.bind_connectionc �, | j r
| j �� �� | j stt| ��� d S r )r� r� r0 r� rz r r� r r r0 P � �zXConnection.closec r� r )r� r� r; r� rz r r� r r r; U r� zXConnection.disconnect) r7 r8 r9 r r� rS r! r� r0 r; �
__classcell__r r r� r rz s
rz c s( e Zd Z� fdd�Z� fdd�Z� ZS )�NodeConnectionc s t t| ��|� d S r )r� r� r rM r� r r r \ s zNodeConnection.__init__c
s> z
t t| ��� W d S tjy } ztd�|���d }~ww )Nr� )r� r� r! r r: r r� )r r� r� r r r! _ s ��zNodeConnection.connect)r7 r8 r9 r r! r� r r r� r r� [ s r� c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� BaseSessiona� Base functionality for Session classes through the X Protocol.
This class encloses the core functionality to be made available on both
the XSession and NodeSession classes, such functionality includes:
- Accessing available schemas.
- Schema management operations.
- Enabling/disabling warning generation.
- Retrieval of connection information.
Args:
settings (dict): Connection data used to connect to the database.
c C s
|| _ d S r )� _settingsrM r r r r t s
zBaseSession.__init__c C s | j jjd uS r )� _connectionrJ r r r r r rx w � zBaseSession.is_openc C s
t | |�S )z�Retrieves a Schema object from the current session by it's name.
Args:
name (string): The name of the Schema object to be retrieved.
Returns:
mysqlx.Schema: The Schema object with the given name.
r �r rZ r r r �
get_schemaz s
zBaseSession.get_schemac C s( | j j�d�rt| | j jd �S td��)aK Retrieves a Schema object from the current session by the schema
name configured in the connection settings.
Returns:
mysqlx.Schema: The Schema object with the given name at connect
time.
Raises:
ProgrammingError: If default schema not provided.
rE zDefault schema not provided)r� rI r4 r
r r r r r �get_default_schema� s z BaseSession.get_default_schemac C s | j �dt�|�d� dS )z�Drops the schema with the specified name.
Args:
name (string): The name of the Schema object to be retrieved.
r_ TN)r� rl �_DROP_DATABASE_QUERYr� r� r r r � drop_schema� s �zBaseSession.drop_schemac C s | j �dt�|�d� t| |�S )z�Creates a schema on the database and returns the corresponding
object.
Args:
name (string): A string value indicating the schema name.
r_ T)r� rl �_CREATE_DATABASE_QUERYr� r
r� r r r �
create_schema� s �
zBaseSession.create_schemac C � | j �ddd� dS )z4Starts a transaction context on the server.
r_ zSTART TRANSACTIONTN�r� rl r r r r �start_transaction� s z BaseSession.start_transactionc C r� )zXCommits all the operations executed after a call to
startTransaction().
r_ ZCOMMITTNr� r r r r �commit� � zBaseSession.commitc C r� )zYDiscards all the operations executed after a call to
startTransaction().
r_ ZROLLBACKTNr� r r r r �rollback� r� zBaseSession.rollbackc C s | j �� d S r )r� r0 r r r r r0 � r� zBaseSession.closeN)r7 r8 r9 �__doc__r rx r� r� r� r� r� r� r� r0 r r r r r� f s
r� c �( e Zd ZdZ� fdd�Zdd� Z� ZS )�XSessionaW Enables interaction with a X Protocol enabled MySQL Product.
The functionality includes:
- Accessing available schemas.
- Schema management operations.
- Enabling/disabling warning generation.
- Retrieval of connection information.
Args:
settings (dict): Connection data used to connect to the database.
c �* t t| ��|� t| j�| _| j�� d S r )r� r� r rz r� r� r! rM r� r r r � � zXSession.__init__c C s, | � � std��t| j�}| j�|j� |S )Nz$XSession is not connected to a farm.)rx r � NodeSessionr� r� r� )r Znsessr r r �bind_to_default_shard� s
z XSession.bind_to_default_shard)r7 r8 r9 r� r r� r� r r r� r r� � s r� c r� )r� at Enables interaction with a X Protocol enabled MySQL Server.
The functionality includes:
- Accessing available schemas.
- Schema management operations.
- Enabling/disabling warning generation.
- Retrieval of connection information.
- Includes SQL Execution.
Args:
settings (dict): Connection data used to connect to the database.
c r� r )r� r� r r� r� r� r! rM r� r r r � r� zNodeSession.__init__c C s t | j|�S )z}Creates a :class:`mysqlx.SqlStatement` object to allow running the
SQL statement on the target MySQL Server.
)r r� )r r_ r r r r_ � s zNodeSession.sql)r7 r8 r9 r� r r_ r� r r r� r r� � s
r� )$r� r2 r1 r\ r � functoolsr Zauthenticationr �errorsr r r Zcrudr
rL r r rn r
r r rc r r r� r� �objectr rA rB rz r� r� r� r� r r r r �<module> s4 F U W