 cd136b11a7
			
		
	
	cd136b11a7
	
	
	
		
			
			* dealloc named prepare route to all * add newline * erase from ps manager too * little refactor
		
			
				
	
	
		
			817 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			817 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #pragma once
 | |
| /*
 | |
|  * Copyright (c) 2016 MariaDB Corporation Ab
 | |
|  *
 | |
|  * Use of this software is governed by the Business Source License included
 | |
|  * in the LICENSE.TXT file and at www.mariadb.com/bsl11.
 | |
|  *
 | |
|  * Change Date: 2020-01-01
 | |
|  *
 | |
|  * On the date above, in accordance with the Business Source License, use
 | |
|  * of this software will be governed by version 2 or later of the General
 | |
|  * Public License.
 | |
|  */
 | |
| 
 | |
| #include <maxscale/cdefs.h>
 | |
| #include <maxscale/buffer.h>
 | |
| 
 | |
| MXS_BEGIN_DECLS
 | |
| 
 | |
| #define MXS_QUERY_CLASSIFIER_VERSION {2, 0, 0}
 | |
| 
 | |
| /**
 | |
|  * qc_init_kind_t specifies what kind of initialization should be performed.
 | |
|  */
 | |
| typedef enum qc_init_kind
 | |
| {
 | |
|     QC_INIT_SELF   = 0x01, /*< Initialize/finalize the query classifier itself. */
 | |
|     QC_INIT_PLUGIN = 0x02, /*< Initialize/finalize the plugin. */
 | |
|     QC_INIT_BOTH   = 0x03
 | |
| } qc_init_kind_t;
 | |
| 
 | |
| /**
 | |
|  * qc_sql_mode_t specifies what should be assumed of the statements
 | |
|  * that will be parsed.
 | |
|  */
 | |
| typedef enum qc_sql_mode
 | |
| {
 | |
|     QC_SQL_MODE_DEFAULT, /*< Assume the statements are MariaDB SQL. */
 | |
|     QC_SQL_MODE_ORACLE   /*< Assume the statements are PL/SQL. */
 | |
| } qc_sql_mode_t;
 | |
| 
 | |
| /**
 | |
|  * @c qc_collect_info_t specifies what information should be collected during parsing.
 | |
|  */
 | |
| typedef enum qc_collect_info
 | |
| {
 | |
|     QC_COLLECT_ESSENTIALS = 0x00, /*< Collect only the base minimum. */
 | |
|     QC_COLLECT_TABLES     = 0x01, /*< Collect table names. */
 | |
|     QC_COLLECT_DATABASES  = 0x02, /*< Collect database names. */
 | |
|     QC_COLLECT_FIELDS     = 0x04, /*< Collect field information. */
 | |
|     QC_COLLECT_FUNCTIONS  = 0x08, /*< Collect function information. */
 | |
| 
 | |
|     QC_COLLECT_ALL = (QC_COLLECT_TABLES|QC_COLLECT_DATABASES|QC_COLLECT_FIELDS|QC_COLLECT_FUNCTIONS)
 | |
| } qc_collect_info_t;
 | |
| /**
 | |
|  * qc_query_type_t defines bits that provide information about a
 | |
|  * particular statement.
 | |
|  *
 | |
|  * Note that more than one bit may be set for a single statement.
 | |
|  */
 | |
| typedef enum qc_query_type
 | |
| {
 | |
|     QUERY_TYPE_UNKNOWN            = 0x000000, /*< Initial value, can't be tested bitwisely */
 | |
|     QUERY_TYPE_LOCAL_READ         = 0x000001, /*< Read non-database data, execute in MaxScale:any */
 | |
|     QUERY_TYPE_READ               = 0x000002, /*< Read database data:any */
 | |
|     QUERY_TYPE_WRITE              = 0x000004, /*< Master data will be  modified:master */
 | |
|     QUERY_TYPE_MASTER_READ        = 0x000008, /*< Read from the master:master */
 | |
|     QUERY_TYPE_SESSION_WRITE      = 0x000010, /*< Session data will be modified:master or all */
 | |
|     QUERY_TYPE_USERVAR_WRITE      = 0x000020, /*< Write a user variable:master or all */
 | |
|     QUERY_TYPE_USERVAR_READ       = 0x000040, /*< Read a user variable:master or any */
 | |
|     QUERY_TYPE_SYSVAR_READ        = 0x000080, /*< Read a system variable:master or any */
 | |
|     /** Not implemented yet */
 | |
|     //QUERY_TYPE_SYSVAR_WRITE       = 0x000100, /*< Write a system variable:master or all */
 | |
|     QUERY_TYPE_GSYSVAR_READ       = 0x000200, /*< Read global system variable:master or any */
 | |
|     QUERY_TYPE_GSYSVAR_WRITE      = 0x000400, /*< Write global system variable:master or all */
 | |
|     QUERY_TYPE_BEGIN_TRX          = 0x000800, /*< BEGIN or START TRANSACTION */
 | |
|     QUERY_TYPE_ENABLE_AUTOCOMMIT  = 0x001000, /*< SET autocommit=1 */
 | |
|     QUERY_TYPE_DISABLE_AUTOCOMMIT = 0x002000, /*< SET autocommit=0 */
 | |
|     QUERY_TYPE_ROLLBACK           = 0x004000, /*< ROLLBACK */
 | |
|     QUERY_TYPE_COMMIT             = 0x008000, /*< COMMIT */
 | |
|     QUERY_TYPE_PREPARE_NAMED_STMT = 0x010000, /*< Prepared stmt with name from user:all */
 | |
|     QUERY_TYPE_PREPARE_STMT       = 0x020000, /*< Prepared stmt with id provided by server:all */
 | |
|     QUERY_TYPE_EXEC_STMT          = 0x040000, /*< Execute prepared statement:master or any */
 | |
|     QUERY_TYPE_CREATE_TMP_TABLE   = 0x080000, /*< Create temporary table:master (could be all) */
 | |
|     QUERY_TYPE_READ_TMP_TABLE     = 0x100000, /*< Read temporary table:master (could be any) */
 | |
|     QUERY_TYPE_SHOW_DATABASES     = 0x200000, /*< Show list of databases */
 | |
|     QUERY_TYPE_SHOW_TABLES        = 0x400000, /*< Show list of tables */
 | |
|     QUERY_TYPE_DEALLOC_PREPARE    = 0x1000000 /*< Dealloc named prepare stmt:all */
 | |
| } qc_query_type_t;
 | |
| 
 | |
| /**
 | |
|  * qc_query_op_t defines the operations a particular statement can perform.
 | |
|  */
 | |
| typedef enum qc_query_op
 | |
| {
 | |
|     QUERY_OP_UNDEFINED = 0,
 | |
| 
 | |
|     QUERY_OP_ALTER,
 | |
|     QUERY_OP_CALL,
 | |
|     QUERY_OP_CHANGE_DB,
 | |
|     QUERY_OP_CREATE,
 | |
|     QUERY_OP_DELETE,
 | |
|     QUERY_OP_DROP,
 | |
|     QUERY_OP_EXECUTE,
 | |
|     QUERY_OP_EXPLAIN,
 | |
|     QUERY_OP_GRANT,
 | |
|     QUERY_OP_INSERT,
 | |
|     QUERY_OP_LOAD_LOCAL,
 | |
|     QUERY_OP_LOAD,
 | |
|     QUERY_OP_REVOKE,
 | |
|     QUERY_OP_SELECT,
 | |
|     QUERY_OP_SHOW,
 | |
|     QUERY_OP_TRUNCATE,
 | |
|     QUERY_OP_UPDATE,
 | |
| } qc_query_op_t;
 | |
| 
 | |
| /**
 | |
|  * qc_parse_result_t defines the possible outcomes when a statement is parsed.
 | |
|  */
 | |
| typedef enum qc_parse_result
 | |
| {
 | |
|     QC_QUERY_INVALID          = 0, /*< The query was not recognized or could not be parsed. */
 | |
|     QC_QUERY_TOKENIZED        = 1, /*< The query was classified based on tokens; incompletely classified. */
 | |
|     QC_QUERY_PARTIALLY_PARSED = 2, /*< The query was only partially parsed; incompletely classified. */
 | |
|     QC_QUERY_PARSED           = 3  /*< The query was fully parsed; completely classified. */
 | |
| } qc_parse_result_t;
 | |
| 
 | |
| /**
 | |
|  * QC_FIELD_INFO contains information about a field used in a statement.
 | |
|  */
 | |
| typedef struct qc_field_info
 | |
| {
 | |
|     char* database; /** Present if the field is of the form "a.b.c", NULL otherwise. */
 | |
|     char* table;    /** Present if the field is of the form "a.b", NULL otherwise. */
 | |
|     char* column;   /** Always present. */
 | |
| } QC_FIELD_INFO;
 | |
| 
 | |
| /**
 | |
|  * QC_FUNCTION_INFO contains information about a function used in a statement.
 | |
|  */
 | |
| typedef struct qc_function_info
 | |
| {
 | |
|     char* name;            /** Name of function. */
 | |
|     QC_FIELD_INFO* fields; /** What fields the function accesses. */
 | |
|     uint32_t n_fields;     /** The number of fields in @c fields. */
 | |
| } QC_FUNCTION_INFO;
 | |
| 
 | |
| /**
 | |
|  * Each API function returns @c QC_RESULT_OK if the actual parsing process
 | |
|  * succeeded, and some error code otherwise.
 | |
|  */
 | |
| typedef enum qc_result
 | |
| {
 | |
|     QC_RESULT_OK,
 | |
|     QC_RESULT_ERROR
 | |
| } qc_result_t;
 | |
| 
 | |
| /**
 | |
|  * QUERY_CLASSIFIER defines the object a query classifier plugin must
 | |
|  * implement and return.
 | |
|  *
 | |
|  * To a user of the query classifier functionality, it can in general
 | |
|  * be ignored.
 | |
|  */
 | |
| typedef struct query_classifier
 | |
| {
 | |
|     /**
 | |
|      * Called once to setup the query classifier
 | |
|      *
 | |
|      * @param sql_mode  The default sql mode.
 | |
|      * @param args      The value of `query_classifier_args` in the configuration file.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the query classifier could be setup, otherwise
 | |
|      *         some specific error code.
 | |
|      */
 | |
|     int32_t (*qc_setup)(qc_sql_mode_t sql_mode, const char* args);
 | |
| 
 | |
|     /**
 | |
|      * Called once at process startup, after @c qc_setup has successfully
 | |
|      * been called.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the process initialization succeeded.
 | |
|      */
 | |
|     int32_t (*qc_process_init)(void);
 | |
| 
 | |
|     /**
 | |
|      * Called once at process shutdown.
 | |
|      */
 | |
|     void (*qc_process_end)(void);
 | |
| 
 | |
|     /**
 | |
|      * Called once per each thread, except for the thread for which @c qc_process_init
 | |
|      * was called.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the thread initialization succeeded.
 | |
|      */
 | |
|     int32_t (*qc_thread_init)(void);
 | |
| 
 | |
|     /**
 | |
|      * Called once when a thread finishes, except for the thread for which @c qc_process_init
 | |
|      * was called.
 | |
|      */
 | |
|     void (*qc_thread_end)(void);
 | |
| 
 | |
|     /**
 | |
|      * Called to explicitly parse a statement.
 | |
|      *
 | |
|      * @param stmt     The statement to be parsed.
 | |
|      * @param collect  A bitmask of @c qc_collect_info_t values. Specifies what information
 | |
|      *                 should be collected. Only a hint and must not restrict what information
 | |
|      *                 later can be queried.
 | |
|      * @param result   On return, the parse result, if @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_parse)(GWBUF* stmt, uint32_t collect, int32_t* result);
 | |
| 
 | |
|     /**
 | |
|      * Reports the type of the statement.
 | |
|      *
 | |
|      * @param stmt  A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param type  On return, the type mask (combination of @c qc_query_type_t),
 | |
|      *              if @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_type_mask)(GWBUF* stmt, uint32_t* type);
 | |
| 
 | |
|     /**
 | |
|      * Reports the operation of the statement.
 | |
|      *
 | |
|      * @param stmt  A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param type  On return, the operation (one of @c qc_query_op_t), if
 | |
|      *              @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_operation)(GWBUF* stmt, int32_t* op);
 | |
| 
 | |
|     /**
 | |
|      * Reports the name of a created table.
 | |
|      *
 | |
|      * @param stmt  A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param name  On return, the name of the created table, if
 | |
|      *              @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_created_table_name)(GWBUF* stmt, char** name);
 | |
| 
 | |
|     /**
 | |
|      * Reports whether a statement is a "DROP TABLE ..." statement.
 | |
|      *
 | |
|      * @param stmt           A COM_QUERY or COM_STMT_PREPARE packet
 | |
|      * @param is_drop_table  On return, non-zero if the statement is a DROP TABLE
 | |
|      *                       statement, if @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_is_drop_table_query)(GWBUF* stmt, int32_t* is_drop_table);
 | |
| 
 | |
|     /**
 | |
|      * Returns all table names.
 | |
|      *
 | |
|      * @param stmt       A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param fullnames  If non-zero, the full (i.e. qualified) names are returned.
 | |
|      * @param names      On return, the names of the statement, if @c QC_RESULT_OK
 | |
|      *                   is returned.
 | |
|      * @param n_names    On return, how many names were returned, if @c QC_RESULT_OK
 | |
|      *                   is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_table_names)(GWBUF* stmt, int32_t full_names, char*** names, int32_t *n_names);
 | |
| 
 | |
|     /**
 | |
|      * The canonical version of a statement.
 | |
|      *
 | |
|      * @param stmt       A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param canonical  On return, the canonical version of the statement, if @c QC_RESULT_OK
 | |
|      *                   is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_canonical)(GWBUF* stmt, char** canonical);
 | |
| 
 | |
|     /**
 | |
|      * Reports whether the statement has a where clause.
 | |
|      *
 | |
|      * @param stmt        A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param has_clause  On return, non-zero if the statement has a where clause, if
 | |
|      *                    @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_query_has_clause)(GWBUF* stmt, int32_t* has_clause);
 | |
| 
 | |
|     /**
 | |
|      * Reports the database names.
 | |
|      *
 | |
|      * @param stmt   A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param names  On return, the database names, if
 | |
|      *               @c QC_RESULT_OK is returned.
 | |
|      * @param size   On return, the number of names in @names, if
 | |
|      *               @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_database_names)(GWBUF* stmt, char*** names, int32_t* size);
 | |
| 
 | |
|     /**
 | |
|      * Reports the prepare name.
 | |
|      *
 | |
|      * @param stmt  A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param name  On return, the name of a prepare statement, if
 | |
|      *              @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_prepare_name)(GWBUF* stmt, char** name);
 | |
| 
 | |
|     /**
 | |
|      * Reports field information.
 | |
|      *
 | |
|      * @param stmt    A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param infos   On return, array of field infos, if @c QC_RESULT_OK is returned.
 | |
|      * @param n_infos On return, the size of @c infos, if @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_field_info)(GWBUF* stmt, const QC_FIELD_INFO** infos, uint32_t* n_infos);
 | |
| 
 | |
|     /**
 | |
|      * The canonical version of a statement.
 | |
|      *
 | |
|      * @param stmt  A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param infos   On return, array of function infos, if @c QC_RESULT_OK is returned.
 | |
|      * @param n_infos On return, the size of @c infos, if @c QC_RESULT_OK is returned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_function_info)(GWBUF* stmt, const QC_FUNCTION_INFO** infos, uint32_t* n_infos);
 | |
| 
 | |
|     /**
 | |
|      * Return the preparable statement of a PREPARE statement.
 | |
|      *
 | |
|      * @param stmt             A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|      * @param preparable_stmt  On return, the preparable statement (provided @c stmt is a
 | |
|      *                         PREPARE statement), if @c QC_RESULT_OK is returned. Otherwise
 | |
|      *                         NULL.
 | |
|      *
 | |
|      * @attention The returned GWBUF is the property of @c stmt and will be deleted when
 | |
|      *            @c stmt is. If the preparable statement need to be retained beyond the
 | |
|      *            lifetime of @c stmt, it must be cloned.
 | |
|      *
 | |
|      * @return QC_RESULT_OK, if the parsing was not aborted due to resource
 | |
|      *         exhaustion or equivalent.
 | |
|      */
 | |
|     int32_t (*qc_get_preparable_stmt)(GWBUF* stmt, GWBUF** preparable_stmt);
 | |
| 
 | |
|     /**
 | |
|      * Set the version of the server. The version may affect how a statement
 | |
|      * is classified. Note that the server version is maintained separately
 | |
|      * for each thread.
 | |
|      *
 | |
|      * @param version  Version encoded as MariaDB encodes the version, i.e.:
 | |
|      *                 version = major * 10000 + minor * 100 + patch
 | |
|      */
 | |
|     void (*qc_set_server_version)(uint64_t version);
 | |
| 
 | |
|     /**
 | |
|      * Get the thread specific version assumed of the server. If the version has
 | |
|      * not been set, all values are 0.
 | |
|      *
 | |
|      * @param version  The version encoded as MariaDB encodes the version, i.e.:
 | |
|      *                 version = major * 10000 + minor * 100 + patch
 | |
|      */
 | |
|     void (*qc_get_server_version)(uint64_t* version);
 | |
| 
 | |
|     /**
 | |
|      * Gets the sql mode of the *calling* thread.
 | |
|      *
 | |
|      * @param sql_mode  The mode.
 | |
|      *
 | |
|      * @return QC_RESULT_OK
 | |
|      */
 | |
|     int32_t (*qc_get_sql_mode)(qc_sql_mode_t* sql_mode);
 | |
| 
 | |
|     /**
 | |
|      * Sets the sql mode for the *calling* thread.
 | |
|      *
 | |
|      * @param sql_mode  The mode.
 | |
|      *
 | |
|      * @return QC_RESULT_OK if @sql_mode is valid, otherwise QC_RESULT_ERROR.
 | |
|      */
 | |
|     int32_t (*qc_set_sql_mode)(qc_sql_mode_t sql_mode);
 | |
| } QUERY_CLASSIFIER;
 | |
| 
 | |
| /**
 | |
|  * Loads and sets up the default query classifier.
 | |
|  *
 | |
|  * This must be called once during the execution of a process. The query
 | |
|  * classifier functions can only be used if this function first and thereafter
 | |
|  * the @c qc_process_init return true.
 | |
|  *
 | |
|  * MaxScale calls this function, so plugins should not do that.
 | |
|  *
 | |
|  * @param plugin_name  The name of the plugin from which the query classifier
 | |
|  *                     should be loaded.
 | |
|  * @param sql_mode     The default sql mode.
 | |
|  * @param plugin_args  The arguments to be provided to the query classifier.
 | |
|  *
 | |
|  * @return True if the query classifier could be loaded and initialized,
 | |
|  *         false otherwise.
 | |
|  *
 | |
|  * @see qc_end qc_thread_init
 | |
|  */
 | |
| bool qc_setup(const char* plugin_name, qc_sql_mode_t sql_mode, const char* plugin_args);
 | |
| 
 | |
| /**
 | |
|  * Intializes the query classifier.
 | |
|  *
 | |
|  * This function should be called once, provided @c qc_setup returned true,
 | |
|  * before the query classifier functionality is used.
 | |
|  *
 | |
|  * MaxScale calls this functions, so plugins should not do that.
 | |
|  *
 | |
|  * @param kind  What kind of initialization should be performed.
 | |
|  *              Combination of qc_init_kind_t.
 | |
|  *
 | |
|  * @return True, if the process wide initialization could be performed.
 | |
|  *
 | |
|  * @see qc_process_end qc_thread_init
 | |
|  */
 | |
| bool qc_process_init(uint32_t kind);
 | |
| 
 | |
| /**
 | |
|  * Finalizes the query classifier.
 | |
|  *
 | |
|  * A successful call of qc_init() should before program exit be followed
 | |
|  * by a call to this function. MaxScale calls this function, so plugins
 | |
|  * should not do that.
 | |
|  *
 | |
|  * @param kind  What kind of finalization should be performed.
 | |
|  *              Combination of qc_init_kind_t.
 | |
|  *
 | |
|  * @see qc_process_init qc_thread_end
 | |
|  */
 | |
| void qc_process_end(uint32_t kind);
 | |
| 
 | |
| /**
 | |
|  * Loads a particular query classifier.
 | |
|  *
 | |
|  * In general there is no need to use this function, but rely upon qc_init().
 | |
|  * However, if there is a need to use multiple query classifiers concurrently
 | |
|  * then this function provides the means for that. Note that after a query
 | |
|  * classifier has been loaded, it must explicitly be initialized before it
 | |
|  * can be used.
 | |
|  *
 | |
|  * @param plugin_name  The name of the plugin from which the query classifier
 | |
|  *                     should be loaded.
 | |
|  *
 | |
|  * @return A QUERY_CLASSIFIER object if successful, NULL otherwise.
 | |
|  *
 | |
|  * @see qc_unload
 | |
|  */
 | |
| QUERY_CLASSIFIER* qc_load(const char* plugin_name);
 | |
| 
 | |
| /**
 | |
|  * Unloads an explicitly loaded query classifier.
 | |
|  *
 | |
|  * @see qc_load
 | |
|  */
 | |
| void qc_unload(QUERY_CLASSIFIER* classifier);
 | |
| 
 | |
| /**
 | |
|  * Performs thread initialization needed by the query classifier. Should
 | |
|  * be called in every thread, except the one where qc_process_init()
 | |
|  * was called.
 | |
|  *
 | |
|  * MaxScale calls this function, so plugins should not do that.
 | |
|  *
 | |
|  * @param kind  What kind of initialization should be performed.
 | |
|  *              Combination of qc_init_kind_t.
 | |
|  *
 | |
|  * @return True if the initialization succeeded, false otherwise.
 | |
|  *
 | |
|  * @see qc_thread_end
 | |
|  */
 | |
| bool qc_thread_init(uint32_t kind);
 | |
| 
 | |
| /**
 | |
|  * Performs thread finalization needed by the query classifier.
 | |
|  * A successful call to qc_thread_init() should at some point be followed
 | |
|  * by a call to this function.
 | |
|  *
 | |
|  * MaxScale calls this function, so plugins should not do that.
 | |
|  *
 | |
|  * @param kind  What kind of finalization should be performed.
 | |
|  *              Combination of qc_init_kind_t.
 | |
|  *
 | |
|  * @see qc_thread_init
 | |
|  */
 | |
| void qc_thread_end(uint32_t kind);
 | |
| 
 | |
| /**
 | |
|  * Parses the statement in the provided buffer and returns a value specifying
 | |
|  * to what extent the statement could be parsed.
 | |
|  *
 | |
|  * There is no need to call this function explicitly before calling any of
 | |
|  * the other functions; e.g. qc_get_type_mask(). When some particular property of
 | |
|  * a statement is asked for, the statement will be parsed if it has not been
 | |
|  * parsed yet. Also, if the statement in the provided buffer has been parsed
 | |
|  * already then this function will only return the result of that parsing;
 | |
|  * the statement will not be parsed again.
 | |
|  *
 | |
|  * @param stmt     A buffer containing an COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  * @param collect  A bitmask of @c qc_collect_info_t values. Specifies what information
 | |
|  *                 should be collected.
 | |
|  *
 | |
|  *                 Note that this is merely a hint and does not restrict what
 | |
|  *                 information can be queried for. If necessary, the statement
 | |
|  *                 will transparently be reparsed.
 | |
|  *
 | |
|  * @return To what extent the statement could be parsed.
 | |
|  */
 | |
| qc_parse_result_t qc_parse(GWBUF* stmt, uint32_t collect);
 | |
| 
 | |
| /**
 | |
|  * Returns information about affected fields.
 | |
|  *
 | |
|  * @param stmt     A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  * @param infos    Pointer to pointer that after the call will point to an
 | |
|  *                 array of QC_FIELD_INFO:s.
 | |
|  * @param n_infos  Pointer to size_t variable where the number of items
 | |
|  *                 in @c infos will be returned.
 | |
|  *
 | |
|  * @note The returned array belongs to the GWBUF and remains valid for as
 | |
|  *       long as the GWBUF is valid. If the data is needed for longer than
 | |
|  *       that, it must be copied.
 | |
|  */
 | |
| void qc_get_field_info(GWBUF* stmt, const QC_FIELD_INFO** infos, size_t* n_infos);
 | |
| 
 | |
| /**
 | |
|  * Returns information about function usage.
 | |
|  *
 | |
|  * @param stmt     A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  * @param infos    Pointer to pointer that after the call will point to an
 | |
|  *                 array of QC_FUNCTION_INFO:s.
 | |
|  * @param n_infos  Pointer to size_t variable where the number of items
 | |
|  *                 in @c infos will be returned.
 | |
|  *
 | |
|  * @note The returned array belongs to the GWBUF and remains valid for as
 | |
|  *       long as the GWBUF is valid. If the data is needed for longer than
 | |
|  *       that, it must be copied.
 | |
|  *
 | |
|  * @note For each function, only the fields that any invocation of it directly
 | |
|  *       accesses will be returned. For instance:
 | |
|  *
 | |
|  *           select length(a), length(concat(b, length(a))) from t
 | |
|  *
 | |
|  *       will for @length return the field @a and for @c concat the field @b.
 | |
|  */
 | |
| void qc_get_function_info(GWBUF* stmt, const QC_FUNCTION_INFO** infos, size_t* n_infos);
 | |
| 
 | |
| /**
 | |
|  * Returns the statement, with literals replaced with question marks.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return A statement in its canonical form, or NULL if a memory
 | |
|  *         allocation fails. The string must be freed by the caller.
 | |
|  */
 | |
| char* qc_get_canonical(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns the name of the created table.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return The name of the created table or NULL if the statement
 | |
|  *         does not create a table or a memory allocation failed.
 | |
|  *         The string must be freed by the caller.
 | |
|  */
 | |
| char* qc_get_created_table_name(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns the databases accessed by the statement. Note that a
 | |
|  * possible default database is not returned.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  * @param size  Pointer to integer where the number of databases
 | |
|  *              is stored.
 | |
|  *
 | |
|  * @return Array of strings or NULL if a memory allocation fails.
 | |
|  *
 | |
|  * @note The returned array and the strings pointed to @b must be freed
 | |
|  *       by the caller.
 | |
|  */
 | |
| char** qc_get_database_names(GWBUF* stmt, int* size);
 | |
| 
 | |
| /**
 | |
|  * Returns the operation of the statement.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return The operation of the statement.
 | |
|  */
 | |
| qc_query_op_t qc_get_operation(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns the name of the prepared statement, if the statement
 | |
|  * is a PREPARE or EXECUTE statement.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return The name of the prepared statement, if the statement
 | |
|  *         is a PREPARE or EXECUTE statement; otherwise NULL.
 | |
|  *
 | |
|  * @note The returned string @b must be freed by the caller.
 | |
|  *
 | |
|  * @note Even though a COM_STMT_PREPARE can be given to the query
 | |
|  *       classifier for parsing, this function will in that case
 | |
|  *       return NULL since the id of the statement is provided by
 | |
|  *       the server.
 | |
|  */
 | |
| char* qc_get_prepare_name(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns the preparable statement of a PREPARE statment. Other query classifier
 | |
|  * functions can then be used on the returned statement to find out information
 | |
|  * about the preparable statement. The returned @c GWBUF should not be used for
 | |
|  * anything else but for obtaining information about the preparable statement.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return The preparable statement, if @stmt was a COM_QUERY PREPARE statement,
 | |
|  *         or NULL.
 | |
|  *
 | |
|  * @attention If the packet was a COM_STMT_PREPARE, then this function will
 | |
|  *            return NULL and the actual properties of the query can be obtained
 | |
|  *            by calling any of the qc-functions directly on the GWBUF containing
 | |
|  *            the COM_STMT_PREPARE. However, the type mask will contain the
 | |
|  *            bit @c QUERY_TYPE_PREPARE_STMT.
 | |
|  *
 | |
|  * @attention The returned @c GWBUF is the property of @c stmt and will be
 | |
|  *            deleted along with it.
 | |
|  */
 | |
| GWBUF* qc_get_preparable_stmt(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Gets the sql mode of the *calling* thread.
 | |
|  *
 | |
|  * @return The mode.
 | |
|  */
 | |
| qc_sql_mode_t qc_get_sql_mode();
 | |
| 
 | |
| /**
 | |
|  * Returns the tables accessed by the statement.
 | |
|  *
 | |
|  * @param stmt       A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  * @param tblsize    Pointer to integer where the number of tables is stored.
 | |
|  * @param fullnames  If true, a table names will include the database name
 | |
|  *                   as well (if explicitly referred to in the statement).
 | |
|  *
 | |
|  * @return Array of strings or NULL if a memory allocation fails.
 | |
|  *
 | |
|  * @note The returned array and the strings pointed to @b must be freed
 | |
|  *       by the caller.
 | |
|  */
 | |
| char** qc_get_table_names(GWBUF* stmt, int* size, bool fullnames);
 | |
| 
 | |
| 
 | |
| /**
 | |
|  * Returns a bitmask specifying the type(s) of the statement. The result
 | |
|  * should be tested against specific qc_query_type_t values* using the
 | |
|  * bitwise & operator, never using the == operator.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return A bitmask with the type(s) the query.
 | |
|  *
 | |
|  * @see qc_query_is_type
 | |
|  */
 | |
| uint32_t qc_get_type_mask(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns the type bitmask of transaction related statements.
 | |
|  *
 | |
|  * If the statement starts a transaction, ends a transaction or
 | |
|  * changes the autocommit state, the returned bitmap will be a
 | |
|  * combination of:
 | |
|  *
 | |
|  *    QUERY_TYPE_BEGIN_TRX
 | |
|  *    QUERY_TYPE_COMMIT
 | |
|  *    QUERY_TYPE_ROLLBACK
 | |
|  *    QUERY_TYPE_ENABLE_AUTOCOMMIT
 | |
|  *    QUERY_TYPE_DISABLE_AUTOCOMMIT
 | |
|  *    QUERY_TYPE_READ  (explicitly read only transaction)
 | |
|  *    QUERY_TYPE_WRITE (explicitly read write transaction)
 | |
|  *
 | |
|  * Otherwise the result will be 0.
 | |
|  *
 | |
|  * @param stmt A COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return The relevant type bits if the statement is transaction
 | |
|  *         related, otherwise 0.
 | |
|  */
 | |
| uint32_t qc_get_trx_type_mask(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns whether the statement is a DROP TABLE statement.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return True if the statement is a DROP TABLE statement, false otherwise.
 | |
|  *
 | |
|  * @todo This function is far too specific.
 | |
|  */
 | |
| bool qc_is_drop_table_query(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Returns the string representation of a query operation.
 | |
|  *
 | |
|  * @param op  A query operation.
 | |
|  *
 | |
|  * @return The corresponding string.
 | |
|  *
 | |
|  * @note The returned string is statically allocated and must *not* be freed.
 | |
|  */
 | |
| const char* qc_op_to_string(qc_query_op_t op);
 | |
| 
 | |
| /**
 | |
|  * Returns whether the typemask contains a particular type.
 | |
|  *
 | |
|  * @param typemask  A bitmask of query types.
 | |
|  * @param type      A particular qc_query_type_t value.
 | |
|  *
 | |
|  * @return True, if the type is in the mask.
 | |
|  */
 | |
| static inline bool qc_query_is_type(uint32_t typemask, qc_query_type_t type)
 | |
| {
 | |
|     return (typemask & (uint32_t)type) == (uint32_t)type;
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Returns whether the statement has a WHERE or a USING clause.
 | |
|  *
 | |
|  * @param stmt  A buffer containing a COM_QUERY or COM_STMT_PREPARE packet.
 | |
|  *
 | |
|  * @return True, if the statement has a WHERE or USING clause, false
 | |
|  *         otherwise.
 | |
|  */
 | |
| bool qc_query_has_clause(GWBUF* stmt);
 | |
| 
 | |
| /**
 | |
|  * Sets the sql mode for the *calling* thread.
 | |
|  *
 | |
|  * @param sql_mode  The mode.
 | |
|  */
 | |
| void qc_set_sql_mode(qc_sql_mode_t sql_mode);
 | |
| 
 | |
| /**
 | |
|  * Returns the string representation of a query type.
 | |
|  *
 | |
|  * @param type  A query type (not a bitmask of several).
 | |
|  *
 | |
|  * @return The corresponding string.
 | |
|  *
 | |
|  * @note The returned string is statically allocated and must @b not be freed.
 | |
|  */
 | |
| const char* qc_type_to_string(qc_query_type_t type);
 | |
| 
 | |
| /**
 | |
|  * Returns a string representation of a type bitmask.
 | |
|  *
 | |
|  * @param typemask  A bit mask of query types.
 | |
|  *
 | |
|  * @return The corresponding string or NULL if the allocation fails.
 | |
|  *
 | |
|  * @note The returned string is dynamically allocated and @b must be freed.
 | |
|  */
 | |
| char* qc_typemask_to_string(uint32_t typemask);
 | |
| 
 | |
| /**
 | |
|  * Set the version of the server. The version may affect how a statement
 | |
|  * is classified. Note that the server version is maintained separately
 | |
|  * for each thread.
 | |
|  *
 | |
|  * @param version  Version encoded as MariaDB encodes the version, i.e.:
 | |
|  *                 version = major * 10000 + minor * 100 + patch
 | |
|  */
 | |
| void qc_set_server_version(uint64_t version);
 | |
| 
 | |
| /**
 | |
|  * Get the thread specific version assumed of the server. If the version has
 | |
|  * not been set, all values are 0.
 | |
|  *
 | |
|  * @return The version as MariaDB encodes the version, i.e:
 | |
|  *         version = major * 10000 + minor * 100 + patch
 | |
|  */
 | |
| uint64_t qc_get_server_version();
 | |
| 
 | |
| MXS_END_DECLS
 |