summaryrefslogtreecommitdiffstats
path: root/libmysqld/lib_sql.cc
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-04 18:00:34 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-04 18:00:34 +0000
commit3f619478f796eddbba6e39502fe941b285dd97b1 (patch)
treee2c7b5777f728320e5b5542b6213fd3591ba51e2 /libmysqld/lib_sql.cc
parentInitial commit. (diff)
downloadmariadb-3f619478f796eddbba6e39502fe941b285dd97b1.tar.xz
mariadb-3f619478f796eddbba6e39502fe941b285dd97b1.zip
Adding upstream version 1:10.11.6.upstream/1%10.11.6upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'libmysqld/lib_sql.cc')
-rw-r--r--libmysqld/lib_sql.cc1449
1 files changed, 1449 insertions, 0 deletions
diff --git a/libmysqld/lib_sql.cc b/libmysqld/lib_sql.cc
new file mode 100644
index 00000000..d4edd242
--- /dev/null
+++ b/libmysqld/lib_sql.cc
@@ -0,0 +1,1449 @@
+/*
+ * Copyright (c) 2000
+ * SWsoft company
+ *
+ * Modifications copyright (c) 2001, 2013. Oracle and/or its affiliates.
+ * All rights reserved.
+ *
+ * This material is provided "as is", with absolutely no warranty expressed
+ * or implied. Any use is at your own risk.
+ *
+ * Permission to use or copy this software for any purpose is hereby granted
+ * without fee, provided the above notices are retained on all copies.
+ * Permission to modify the code and to distribute modified code is granted,
+ * provided the above notices are retained, and a notice that the code was
+ * modified is included with the above copyright notice.
+ *
+
+ This code was modified by the MySQL team
+*/
+
+/*
+ The following is needed to not cause conflicts when we include mysqld.cc
+*/
+
+#define main main1
+#define mysql_unix_port mysql_inix_port1
+#define mysql_port mysql_port1
+
+extern "C"
+{
+ extern unsigned long max_allowed_packet, net_buffer_length;
+}
+
+#include "../sql/mysqld.cc"
+
+C_MODE_START
+
+#include <mysql.h>
+#undef ER
+#include "errmsg.h"
+#include "embedded_priv.h"
+
+extern unsigned int mysql_server_last_errno;
+extern char mysql_server_last_error[MYSQL_ERRMSG_SIZE];
+static my_bool emb_read_query_result(MYSQL *mysql);
+static void free_embedded_thd(MYSQL *mysql);
+static bool embedded_print_errors= 0;
+
+extern "C" void unireg_clear(int exit_code)
+{
+ DBUG_ENTER("unireg_clear");
+ embedded_print_errors= 0;
+ clean_up(!opt_help && !exit_code); /* purecov: inspected */
+ clean_up_mutexes();
+ my_end(opt_endinfo ? MY_CHECK_ERROR | MY_GIVE_INFO : 0);
+ DBUG_VOID_RETURN;
+}
+
+
+static my_bool mysql_embedded_init= 0;
+
+/*
+ Wrapper error handler for embedded server to call client/server error
+ handler based on whether thread is in client/server context
+*/
+
+static void embedded_error_handler(uint error, const char *str, myf MyFlags)
+{
+ DBUG_ENTER("embedded_error_handler");
+
+ /*
+ If current_thd is NULL, it means restore_global has been called and
+ thread is in client context, then call client error handler else call
+ server error handler.
+ */
+ DBUG_RETURN(current_thd ? my_message_sql(error, str, MyFlags):
+ my_message_stderr(error, str, MyFlags));
+}
+
+/*
+ Reads error information from the MYSQL_DATA and puts
+ it into proper MYSQL members
+
+ SYNOPSIS
+ embedded_get_error()
+ mysql connection handler
+ data query result
+
+ NOTES
+ after that function error information will be accessible
+ with usual functions like mysql_error()
+ data is my_free-d in this function
+ most of the data is stored in data->embedded_info structure
+*/
+
+void embedded_get_error(MYSQL *mysql, MYSQL_DATA *data)
+{
+ NET *net= &mysql->net;
+ struct embedded_query_result *ei= data->embedded_info;
+ net->last_errno= ei->last_errno;
+ strmake_buf(net->last_error, ei->info);
+ memcpy(net->sqlstate, ei->sqlstate, sizeof(net->sqlstate));
+ mysql->server_status= ei->server_status;
+ my_free(data);
+}
+
+static my_bool
+emb_advanced_command(MYSQL *mysql, enum enum_server_command command,
+ const uchar *header, ulong header_length,
+ const uchar *arg, ulong arg_length, my_bool skip_check,
+ MYSQL_STMT *stmt)
+{
+ my_bool result= 1;
+ THD *thd=(THD *) mysql->thd;
+ NET *net= &mysql->net;
+ my_bool stmt_skip= stmt ? stmt->state != MYSQL_STMT_INIT_DONE : FALSE;
+
+ if (thd && thd->killed != NOT_KILLED)
+ {
+ if (thd->killed < KILL_CONNECTION)
+ thd->killed= NOT_KILLED;
+ else
+ {
+ free_embedded_thd(mysql);
+ thd= 0;
+ }
+ }
+
+ if (!thd)
+ {
+ /* Do "reconnect" if possible */
+ if (mysql_reconnect(mysql) || stmt_skip)
+ return 1;
+ thd= (THD *) mysql->thd;
+ }
+
+ thd->clear_data_list();
+ /* Check that we are calling the client functions in right order */
+ if (mysql->status != MYSQL_STATUS_READY)
+ {
+ set_mysql_error(mysql, CR_COMMANDS_OUT_OF_SYNC, unknown_sqlstate);
+ result= 1;
+ goto end;
+ }
+
+ /* Clear result variables */
+ thd->clear_error(1);
+ mysql->affected_rows= ~(my_ulonglong) 0;
+ mysql->field_count= 0;
+ net_clear_error(net);
+ thd->current_stmt= stmt;
+
+ thd->thread_stack= (char*) &thd;
+ thd->store_globals(); // Fix if more than one connect
+ /*
+ We have to call free_old_query before we start to fill mysql->fields
+ for new query. In the case of embedded server we collect field data
+ during query execution (not during data retrieval as it is in remote
+ client). So we have to call free_old_query here
+ */
+ free_old_query(mysql);
+
+ thd->extra_length= arg_length;
+ thd->extra_data= (char *)arg;
+ if (header)
+ {
+ arg= header;
+ arg_length= header_length;
+ }
+
+ result= dispatch_command(command, thd, (char *) arg, arg_length);
+ thd->cur_data= 0;
+ thd->mysys_var= NULL;
+
+ if (!skip_check)
+ result= thd->is_error() ? -1 : 0;
+
+ thd->mysys_var= 0;
+
+end:
+ thd->reset_globals();
+ return result;
+}
+
+static void emb_flush_use_result(MYSQL *mysql, my_bool)
+{
+ THD *thd= (THD*) mysql->thd;
+ if (thd->cur_data)
+ {
+ free_rows(thd->cur_data);
+ thd->cur_data= 0;
+ }
+ else if (thd->first_data)
+ {
+ MYSQL_DATA *data= thd->first_data;
+ thd->first_data= data->embedded_info->next;
+ free_rows(data);
+ }
+}
+
+
+/*
+ reads dataset from the next query result
+
+ SYNOPSIS
+ emb_read_rows()
+ mysql connection handle
+ other parameters are not used
+
+ NOTES
+ It just gets next MYSQL_DATA from the result's queue
+
+ RETURN
+ pointer to MYSQL_DATA with the coming recordset
+*/
+
+static MYSQL_DATA *
+emb_read_rows(MYSQL *mysql, MYSQL_FIELD *mysql_fields __attribute__((unused)),
+ unsigned int fields __attribute__((unused)))
+{
+ MYSQL_DATA *result= ((THD*)mysql->thd)->cur_data;
+ ((THD*)mysql->thd)->cur_data= 0;
+ if (result->embedded_info->last_errno)
+ {
+ embedded_get_error(mysql, result);
+ return NULL;
+ }
+ *result->embedded_info->prev_ptr= NULL;
+ return result;
+}
+
+
+static MYSQL_FIELD *emb_list_fields(MYSQL *mysql)
+{
+ MYSQL_DATA *res;
+ if (emb_read_query_result(mysql))
+ return 0;
+ res= ((THD*) mysql->thd)->cur_data;
+ ((THD*) mysql->thd)->cur_data= 0;
+ mysql->field_alloc= res->alloc;
+ my_free(res);
+ mysql->status= MYSQL_STATUS_READY;
+ return mysql->fields;
+}
+
+static my_bool emb_read_prepare_result(MYSQL *mysql, MYSQL_STMT *stmt)
+{
+ THD *thd= (THD*) mysql->thd;
+ MYSQL_DATA *res;
+
+ stmt->stmt_id= thd->client_stmt_id;
+ stmt->param_count= thd->client_param_count;
+ stmt->field_count= 0;
+ mysql->warning_count= thd->get_stmt_da()->current_statement_warn_count();
+
+ if (thd->first_data)
+ {
+ if (emb_read_query_result(mysql))
+ return 1;
+ stmt->field_count= mysql->field_count;
+ mysql->status= MYSQL_STATUS_READY;
+ res= thd->cur_data;
+ thd->cur_data= NULL;
+ if (!(mysql->server_status & SERVER_STATUS_AUTOCOMMIT))
+ mysql->server_status|= SERVER_STATUS_IN_TRANS;
+
+ stmt->fields= mysql->fields;
+ stmt->mem_root= res->alloc;
+ mysql->fields= NULL;
+ my_free(res);
+ }
+
+ return 0;
+}
+
+/**************************************************************************
+ Get column lengths of the current row
+ If one uses mysql_use_result, res->lengths contains the length information,
+ else the lengths are calculated from the offset between pointers.
+**************************************************************************/
+
+static void emb_fetch_lengths(ulong *to, MYSQL_ROW column,
+ unsigned int field_count)
+{
+ MYSQL_ROW end;
+
+ for (end=column + field_count; column != end ; column++,to++)
+ *to= *column ? *(uint *)((*column) - sizeof(uint)) : 0;
+}
+
+static my_bool emb_read_query_result(MYSQL *mysql)
+{
+ THD *thd= (THD*) mysql->thd;
+ MYSQL_DATA *res= thd->first_data;
+ DBUG_ASSERT(!thd->cur_data);
+ thd->first_data= res->embedded_info->next;
+ if (res->embedded_info->last_errno &&
+ !res->embedded_info->fields_list)
+ {
+ embedded_get_error(mysql, res);
+ return 1;
+ }
+
+ mysql->warning_count= res->embedded_info->warning_count;
+ mysql->server_status= res->embedded_info->server_status;
+ mysql->field_count= res->fields;
+ if (!(mysql->fields= res->embedded_info->fields_list))
+ {
+ mysql->affected_rows= res->embedded_info->affected_rows;
+ mysql->insert_id= res->embedded_info->insert_id;
+ }
+ net_clear_error(&mysql->net);
+ mysql->info= 0;
+
+ if (res->embedded_info->info[0])
+ {
+ strmake(mysql->info_buffer, res->embedded_info->info, MYSQL_ERRMSG_SIZE-1);
+ mysql->info= mysql->info_buffer;
+ }
+
+ if (res->embedded_info->fields_list)
+ {
+ mysql->status=MYSQL_STATUS_GET_RESULT;
+ thd->cur_data= res;
+ }
+ else
+ my_free(res);
+
+ return 0;
+}
+
+static int emb_stmt_execute(MYSQL_STMT *stmt)
+{
+ DBUG_ENTER("emb_stmt_execute");
+ uchar header[9];
+ THD *thd;
+ my_bool res;
+
+ if (stmt->param_count && !stmt->bind_param_done)
+ {
+ set_stmt_error(stmt, CR_PARAMS_NOT_BOUND, unknown_sqlstate, NULL);
+ DBUG_RETURN(1);
+ }
+
+ int4store(header, stmt->stmt_id);
+ header[4]= (uchar) stmt->flags;
+ header[5]= header[6]= header[7]= header[8]= 0; // safety
+ thd= (THD*)stmt->mysql->thd;
+ thd->client_param_count= stmt->param_count;
+ thd->client_params= stmt->params;
+
+ res= MY_TEST(emb_advanced_command(stmt->mysql, COM_STMT_EXECUTE, 0, 0,
+ header, sizeof(header), 1, stmt) ||
+ emb_read_query_result(stmt->mysql));
+ stmt->affected_rows= stmt->mysql->affected_rows;
+ stmt->insert_id= stmt->mysql->insert_id;
+ stmt->server_status= stmt->mysql->server_status;
+ if (res)
+ {
+ NET *net= &stmt->mysql->net;
+ set_stmt_errmsg(stmt, net);
+ DBUG_RETURN(1);
+ }
+ else if (stmt->mysql->status == MYSQL_STATUS_GET_RESULT)
+ stmt->mysql->status= MYSQL_STATUS_STATEMENT_GET_RESULT;
+ DBUG_RETURN(0);
+}
+
+int emb_read_binary_rows(MYSQL_STMT *stmt)
+{
+ MYSQL_DATA *data;
+ if (!(data= emb_read_rows(stmt->mysql, 0, 0)))
+ {
+ set_stmt_errmsg(stmt, &stmt->mysql->net);
+ return 1;
+ }
+ stmt->result= *data;
+ my_free(data);
+ set_stmt_errmsg(stmt, &stmt->mysql->net);
+ return 0;
+}
+
+int emb_read_rows_from_cursor(MYSQL_STMT *stmt)
+{
+ MYSQL *mysql= stmt->mysql;
+ THD *thd= (THD*) mysql->thd;
+ MYSQL_DATA *res= thd->first_data;
+ DBUG_ASSERT(!thd->first_data->embedded_info->next);
+ thd->first_data= 0;
+ if (res->embedded_info->last_errno)
+ {
+ embedded_get_error(mysql, res);
+ set_stmt_errmsg(stmt, &mysql->net);
+ return 1;
+ }
+
+ thd->cur_data= res;
+ mysql->warning_count= res->embedded_info->warning_count;
+ mysql->server_status= res->embedded_info->server_status;
+ net_clear_error(&mysql->net);
+
+ return emb_read_binary_rows(stmt);
+}
+
+int emb_unbuffered_fetch(MYSQL *mysql, char **row)
+{
+ THD *thd= (THD*) mysql->thd;
+ MYSQL_DATA *data= thd->cur_data;
+ if (data && data->embedded_info->last_errno)
+ {
+ embedded_get_error(mysql, data);
+ thd->cur_data= 0;
+ return 1;
+ }
+ if (!data || !data->data)
+ {
+ *row= NULL;
+ if (data)
+ {
+ thd->cur_data= thd->first_data;
+ thd->first_data= data->embedded_info->next;
+ free_rows(data);
+ }
+ }
+ else
+ {
+ *row= (char *)data->data->data;
+ data->data= data->data->next;
+ }
+ return 0;
+}
+
+static void free_embedded_thd(MYSQL *mysql)
+{
+ THD *thd= (THD*)mysql->thd;
+ server_threads.erase(thd);
+ thd->clear_data_list();
+ thd->store_globals();
+ delete thd;
+ set_current_thd(nullptr);
+ mysql->thd=0;
+}
+
+static const char * emb_read_statistics(MYSQL *mysql)
+{
+ THD *thd= (THD*)mysql->thd;
+ return thd->is_error() ? thd->get_stmt_da()->message() : "";
+}
+
+
+static MYSQL_RES * emb_store_result(MYSQL *mysql)
+{
+ return mysql_store_result(mysql);
+}
+
+static int emb_read_change_user_result(MYSQL *mysql)
+{
+ mysql->net.read_pos= (uchar*)""; // fake an OK packet
+ return mysql_errno(mysql) ? (int)packet_error : 1 /* length of the OK packet */;
+}
+
+static void emb_on_close_free(MYSQL *mysql)
+{
+ my_free(mysql->info_buffer);
+ mysql->info_buffer= 0;
+ if (mysql->thd)
+ {
+ free_embedded_thd(mysql);
+ mysql->thd= 0;
+ }
+}
+
+MYSQL_METHODS embedded_methods=
+{
+ emb_read_query_result,
+ emb_advanced_command,
+ emb_read_rows,
+ emb_store_result,
+ emb_fetch_lengths,
+ emb_flush_use_result,
+ emb_read_change_user_result,
+ emb_on_close_free,
+ emb_list_fields,
+ emb_read_prepare_result,
+ emb_stmt_execute,
+ emb_read_binary_rows,
+ emb_unbuffered_fetch,
+ emb_read_statistics,
+ emb_read_query_result,
+ emb_read_rows_from_cursor
+};
+
+/*
+ Make a copy of array and the strings array points to
+*/
+
+char **copy_arguments(int argc, char **argv)
+{
+ size_t length= 0;
+ char **from, **res, **end= argv+argc;
+
+ for (from=argv ; from != end ; from++)
+ length+= strlen(*from);
+
+ if ((res= (char**) my_malloc(PSI_NOT_INSTRUMENTED,
+ sizeof(argv)*(argc+1)+length+argc, MYF(MY_WME))))
+ {
+ char **to= res, *to_str= (char*) (res+argc+1);
+ for (from=argv ; from != end ;)
+ {
+ *to++= to_str;
+ to_str= strmov(to_str, *from++)+1;
+ }
+ *to= 0; // Last ptr should be null
+ }
+ return res;
+}
+
+char ** copy_arguments_ptr= 0;
+
+int init_embedded_server(int argc, char **argv, char **groups)
+{
+ /*
+ This mess is to allow people to call the init function without
+ having to mess with a fake argv
+ */
+ int *argcp;
+ char ***argvp;
+ char *fake_argv[] = { (char *)"", 0 };
+ const char *fake_groups[] = { "server", "embedded", 0 };
+ my_bool acl_error;
+
+ DBUG_ASSERT(mysql_embedded_init == 0);
+ embedded_print_errors= 1;
+ if (my_thread_init())
+ return 1;
+
+ if (init_early_variables())
+ return 1;
+
+ if (!argc)
+ {
+ argc= 1;
+ argv= fake_argv;
+ }
+ argcp= &argc;
+ argvp= &argv;
+
+ if (!groups)
+ groups= (char**) fake_groups;
+
+ if (!my_progname)
+ my_progname= (char *)"mysql_embedded";
+
+ /*
+ Perform basic logger initialization logger. Should be called after
+ MY_INIT, as it initializes mutexes. Log tables are inited later.
+ */
+ logger.init_base();
+
+ orig_argc= *argcp;
+ orig_argv= *argvp;
+ if (load_defaults("my", (const char **)groups, argcp, argvp))
+ return 1;
+ defaults_argc= *argcp;
+ defaults_argv= *argvp;
+ remaining_argc= *argcp;
+ remaining_argv= *argvp;
+
+ /* Must be initialized early for comparison of options name */
+ system_charset_info= &my_charset_utf8mb3_general_ci;
+ sys_var_init();
+
+ int ho_error= handle_early_options();
+ if (ho_error != 0)
+ return 1;
+
+ my_timer_init(&sys_timer_info);
+
+ if (init_common_variables())
+ {
+ mysql_server_end();
+ return 1;
+ }
+
+ mysql_data_home= mysql_real_data_home;
+ mysql_data_home_len= mysql_real_data_home_len;
+
+ /* Get default temporary directory */
+ opt_mysql_tmpdir=getenv("TMPDIR"); /* Use this if possible */
+#if defined(_WIN32)
+ if (!opt_mysql_tmpdir)
+ opt_mysql_tmpdir=getenv("TEMP");
+ if (!opt_mysql_tmpdir)
+ opt_mysql_tmpdir=getenv("TMP");
+#endif
+ if (!opt_mysql_tmpdir || !opt_mysql_tmpdir[0])
+ opt_mysql_tmpdir= const_cast<char*>(DEFAULT_TMPDIR); /* purecov: inspected*/
+
+ init_ssl();
+ umask(((~my_umask) & 0666));
+ if (init_server_components())
+ {
+ mysql_server_end();
+ return 1;
+ }
+
+ /*
+ set error_handler_hook to embedded_error_handler wrapper.
+ */
+ error_handler_hook= embedded_error_handler;
+
+ acl_error= 0;
+#ifndef NO_EMBEDDED_ACCESS_CHECKS
+ if (!(acl_error= acl_init(opt_noacl)) &&
+ !opt_noacl)
+ (void) grant_init();
+#endif
+ if (acl_error || my_tz_init((THD *)0, default_tz_name, opt_bootstrap))
+ {
+ mysql_server_end();
+ return 1;
+ }
+
+ init_max_user_conn();
+ init_update_queries();
+
+#ifdef HAVE_DLOPEN
+#ifndef NO_EMBEDDED_ACCESS_CHECKS
+ if (!opt_noacl)
+#endif
+ udf_init();
+#endif
+
+ (void) thr_setconcurrency(concurrency); // 10 by default
+
+ if (flush_time && flush_time != ~(ulong) 0L)
+ start_handle_manager();
+
+ // FIXME initialize binlog_filter and rpl_filter if not already done
+ // corresponding delete is in clean_up()
+ if(!binlog_filter) binlog_filter = new Rpl_filter;
+ if(!global_rpl_filter) global_rpl_filter = new Rpl_filter;
+
+ if (opt_init_file)
+ {
+ if (read_init_file(opt_init_file))
+ {
+ mysql_server_end();
+ return 1;
+ }
+ }
+
+ if (ddl_log_execute_recovery() > 0)
+ {
+ mysql_server_end();
+ return 1;
+ }
+ mysql_embedded_init= 1;
+ return 0;
+}
+
+void end_embedded_server()
+{
+ if (mysql_embedded_init)
+ {
+ my_free(copy_arguments_ptr);
+ copy_arguments_ptr=0;
+ clean_up(0);
+ clean_up_mutexes();
+ mysql_embedded_init= 0;
+ }
+}
+
+
+void init_embedded_mysql(MYSQL *mysql, ulong client_flag)
+{
+ THD *thd = (THD *)mysql->thd;
+ thd->mysql= mysql;
+ mysql->server_version= server_version;
+ mysql->client_flag= client_flag;
+ init_alloc_root(PSI_NOT_INSTRUMENTED, &mysql->field_alloc, 8192, 0, MYF(0));
+}
+
+/**
+ @brief Initialize a new THD for a connection in the embedded server
+
+ @param client_flag Client capabilities which this thread supports
+ @return pointer to the created THD object
+
+ @todo
+ This function copies code from several places in the server, including
+ create_new_thread(), and prepare_new_connection_state(). This should
+ be refactored to avoid code duplication.
+*/
+void *create_embedded_thd(ulong client_flag)
+{
+ THD * thd= new THD(next_thread_id());
+
+ thd->thread_stack= (char*) &thd;
+ thd->store_globals();
+ lex_start(thd);
+
+ /* TODO - add init_connect command execution */
+
+ if (thd->variables.max_join_size == HA_POS_ERROR)
+ thd->variables.option_bits |= OPTION_BIG_SELECTS;
+ thd->proc_info=0; // Remove 'login'
+ thd->set_command(COM_SLEEP);
+ thd->set_time();
+ thd->init_for_queries();
+ thd->client_capabilities= client_flag | MARIADB_CLIENT_EXTENDED_METADATA;
+ thd->real_id= pthread_self();
+
+ thd->db= null_clex_str;
+#ifndef NO_EMBEDDED_ACCESS_CHECKS
+ thd->security_ctx->db_access= DB_ACLS;
+ thd->security_ctx->master_access= ALL_KNOWN_ACL;
+#endif
+ thd->cur_data= 0;
+ thd->first_data= 0;
+ thd->data_tail= &thd->first_data;
+ bzero((char*) &thd->net, sizeof(thd->net));
+ server_threads.insert(thd);
+ thd->mysys_var= 0;
+ thd->reset_globals();
+ return thd;
+}
+
+
+#ifdef NO_EMBEDDED_ACCESS_CHECKS
+static void
+emb_transfer_connect_attrs(MYSQL *mysql)
+{
+#ifdef HAVE_PSI_THREAD_INTERFACE
+ if (mysql->options.extension &&
+ mysql->options.extension->connection_attributes_length)
+ {
+ uchar *buf, *ptr;
+ THD *thd= (THD*)mysql->thd;
+ size_t length= mysql->options.extension->connection_attributes_length;
+
+ /* 9 = max length of the serialized length */
+ ptr= buf= (uchar *) my_alloca(length + 9);
+ send_client_connect_attrs(mysql, buf);
+ net_field_length_ll(&ptr);
+ PSI_CALL_set_thread_connect_attrs((char *) ptr, length, thd->charset());
+ my_afree(buf);
+ }
+#endif
+}
+
+
+int check_embedded_connection(MYSQL *mysql, const char *db)
+{
+ int result;
+ LEX_CSTRING db_str = { db, safe_strlen(db) };
+ THD *thd= (THD*)mysql->thd;
+
+ /* the server does the same as the client */
+ mysql->server_capabilities= mysql->client_flag;
+
+ thd_init_client_charset(thd, mysql->charset->number);
+ thd->update_charset();
+ Security_context *sctx= thd->security_ctx;
+ sctx->host_or_ip= sctx->host= (char*) my_localhost;
+ strmake_buf(sctx->priv_host, (char*) my_localhost);
+ strmake_buf(sctx->priv_user, mysql->user);
+ sctx->user= my_strdup(PSI_NOT_INSTRUMENTED, mysql->user, MYF(0));
+ sctx->proxy_user[0]= 0;
+ sctx->master_access= GLOBAL_ACLS; // Full rights
+ emb_transfer_connect_attrs(mysql);
+ /* Change database if necessary */
+ if (!(result= (db && db[0] && mysql_change_db(thd, &db_str, FALSE))))
+ my_ok(thd);
+ thd->protocol->end_statement();
+ emb_read_query_result(mysql);
+ return result;
+}
+
+#else
+int check_embedded_connection(MYSQL *mysql, const char *db)
+{
+ /*
+ we emulate a COM_CHANGE_USER user here,
+ it's easier than to emulate the complete 3-way handshake
+ */
+ char *buf, *end;
+ NET *net= &mysql->net;
+ THD *thd= (THD*)mysql->thd;
+ Security_context *sctx= thd->security_ctx;
+ size_t connect_attrs_len=
+ (mysql->options.extension) ?
+ mysql->options.extension->connection_attributes_length : 0;
+
+ buf= (char *)my_alloca(USERNAME_LENGTH + SCRAMBLE_LENGTH + 1 +
+ 2*NAME_LEN + 2 +
+ connect_attrs_len + 2);
+ if (mysql->options.client_ip)
+ {
+ sctx->host= my_strdup(mysql->options.client_ip, MYF(0));
+ sctx->ip= my_strdup(sctx->host, MYF(0));
+ }
+ else
+ sctx->host= (char*)my_localhost;
+ sctx->host_or_ip= sctx->host;
+
+ if (acl_check_host(sctx->host, sctx->ip))
+ goto err;
+
+ /* construct a COM_CHANGE_USER packet */
+ end= strmake(buf, mysql->user, USERNAME_LENGTH) + 1;
+
+ memset(thd->scramble, 55, SCRAMBLE_LENGTH); // dummy scramble
+ thd->scramble[SCRAMBLE_LENGTH]= 0;
+ //strcpy(mysql->scramble, thd->scramble);
+
+ if (mysql->passwd && mysql->passwd[0])
+ {
+ *end++= SCRAMBLE_LENGTH;
+ scramble(end, thd->scramble, mysql->passwd);
+ end+= SCRAMBLE_LENGTH;
+ }
+ else
+ *end++= 0;
+
+ end= strmake(end, db ? db : "", NAME_LEN) + 1;
+
+ int2store(end, (ushort) mysql->charset->number);
+ end+= 2;
+
+ // There is no pluging compatibility in the embedded server
+ //end= strmake(end, "mysql_native_password", NAME_LEN) + 1;
+
+ /* the server does the same as the client */
+ mysql->server_capabilities= mysql->client_flag;
+
+ end= (char *) send_client_connect_attrs(mysql, (uchar *) end);
+
+ /* acl_authenticate() takes the data from thd->net->read_pos */
+ thd->net.read_pos= (uchar*)buf;
+
+ if (acl_authenticate(thd, (uint) (end - buf)))
+ {
+ my_free(thd->security_ctx->user);
+ goto err;
+ }
+ my_afree(buf);
+ return 0;
+
+err:
+ strmake_buf(net->last_error, thd->get_stmt_da()->message());
+ memcpy(net->sqlstate,
+ mysql_errno_to_sqlstate(thd->get_stmt_da()->sql_errno()),
+ sizeof(net->sqlstate)-1);
+ my_afree(buf);
+ return 1;
+}
+#endif
+
+C_MODE_END
+
+void THD::clear_data_list()
+{
+ while (first_data)
+ {
+ MYSQL_DATA *data= first_data;
+ first_data= data->embedded_info->next;
+ free_rows(data);
+ }
+ data_tail= &first_data;
+ free_rows(cur_data);
+ cur_data= 0;
+}
+
+
+static char *dup_str_aux(MEM_ROOT *root, const char *from, uint length,
+ CHARSET_INFO *fromcs, CHARSET_INFO *tocs)
+{
+ uint32 dummy32;
+ uint dummy_err;
+ char *result;
+
+ /* 'tocs' is set 0 when client issues SET character_set_results=NULL */
+ if (tocs && String::needs_conversion(0, fromcs, tocs, &dummy32))
+ {
+ uint new_len= (tocs->mbmaxlen * length) / fromcs->mbminlen + 1;
+ result= (char *)alloc_root(root, new_len);
+ length= copy_and_convert(result, new_len,
+ tocs, from, length, fromcs, &dummy_err);
+ }
+ else
+ {
+ result= (char *)alloc_root(root, length + 1);
+ memcpy(result, from, length);
+ }
+
+ result[length]= 0;
+ return result;
+}
+
+
+static char *dup_str_aux(MEM_ROOT *root, const LEX_CSTRING &from,
+ CHARSET_INFO *fromcs, CHARSET_INFO *tocs)
+{
+ return dup_str_aux(root, from.str, (uint) from.length, fromcs, tocs);
+}
+
+
+/*
+ creates new result and hooks it to the list
+
+ SYNOPSIS
+ alloc_new_dataset()
+
+ NOTES
+ allocs the MYSQL_DATA + embedded_query_result couple
+ to store the next query result,
+ links these two and attach it to the THD::data_tail
+
+ RETURN
+ pointer to the newly created query result
+*/
+
+MYSQL_DATA *THD::alloc_new_dataset()
+{
+ MYSQL_DATA *data;
+ struct embedded_query_result *emb_data;
+ if (!my_multi_malloc(PSI_NOT_INSTRUMENTED, MYF(MY_WME | MY_ZEROFILL),
+ &data, sizeof(*data), &emb_data, sizeof(*emb_data), NULL))
+ return NULL;
+
+ emb_data->prev_ptr= &data->data;
+ cur_data= data;
+ *data_tail= data;
+ data_tail= &emb_data->next;
+ data->embedded_info= emb_data;
+ return data;
+}
+
+
+/**
+ Stores server_status and warning_count in the current
+ query result structures.
+
+ @param thd current thread
+
+ @note Should be called after we get the recordset-result.
+*/
+
+static
+bool
+write_eof_packet(THD *thd, uint server_status, uint statement_warn_count)
+{
+ if (!thd->mysql) // bootstrap file handling
+ return FALSE;
+ /*
+ The following test should never be true, but it's better to do it
+ because if 'is_fatal_error' is set the server is not going to execute
+ other queries (see the if test in dispatch_command / COM_QUERY)
+ */
+ if (thd->is_fatal_error)
+ thd->server_status&= ~SERVER_MORE_RESULTS_EXISTS;
+ thd->cur_data->embedded_info->server_status= server_status;
+ /*
+ Don't send warn count during SP execution, as the warn_list
+ is cleared between substatements, and mysqltest gets confused
+ */
+ thd->cur_data->embedded_info->warning_count=
+ (thd->spcont ? 0 : MY_MIN(statement_warn_count, 65535));
+ return FALSE;
+}
+
+
+/*
+ allocs new query result and initialises Protocol::alloc
+
+ SYNOPSIS
+ Protocol::begin_dataset()
+
+ RETURN
+ 0 if success
+ 1 if memory allocation failed
+*/
+
+bool Protocol::begin_dataset()
+{
+ MYSQL_DATA *data= thd->alloc_new_dataset();
+ if (!data)
+ return 1;
+ alloc= &data->alloc;
+ /* Assume rowlength < 8192 */
+ init_alloc_root(PSI_NOT_INSTRUMENTED, alloc, 8192, 0, MYF(0));
+ alloc->min_malloc= sizeof(MYSQL_ROWS);
+ return 0;
+}
+
+
+bool Protocol::begin_dataset(THD *thd, uint numfields)
+{
+ if (begin_dataset())
+ return true;
+ MYSQL_DATA *data= thd->cur_data;
+ data->fields= field_count= numfields;
+ if (!(data->embedded_info->fields_list=
+ (MYSQL_FIELD*)alloc_root(&data->alloc, sizeof(MYSQL_FIELD)*field_count)))
+ return true;
+ return false;
+}
+
+
+/*
+ remove last row of current recordset
+
+ SYNOPSIS
+ Protocol_text::remove_last_row()
+
+ NOTES
+ does the loop from the beginning of the current recordset to
+ the last record and cuts it off.
+ Not supposed to be frequently called.
+*/
+
+void Protocol_text::remove_last_row()
+{
+ MYSQL_DATA *data= thd->cur_data;
+ MYSQL_ROWS **last_row_hook= &data->data;
+ my_ulonglong count= data->rows;
+ DBUG_ENTER("Protocol_text::remove_last_row");
+ while (--count)
+ last_row_hook= &(*last_row_hook)->next;
+
+ *last_row_hook= 0;
+ data->embedded_info->prev_ptr= last_row_hook;
+ data->rows--;
+
+ DBUG_VOID_RETURN;
+}
+
+
+
+static MARIADB_CONST_STRING ma_const_string_copy_root(MEM_ROOT *memroot,
+ const char *str,
+ size_t length)
+{
+ MARIADB_CONST_STRING res;
+ if (!str || !(res.str= strmake_root(memroot, str, length)))
+ return null_clex_str;
+ res.length= length;
+ return res;
+}
+
+
+class Client_field_extension: public Sql_alloc,
+ public MARIADB_FIELD_EXTENSION
+{
+public:
+ Client_field_extension()
+ {
+ memset((void*) this, 0, sizeof(*this));
+ }
+ void copy_extended_metadata(MEM_ROOT *memroot,
+ const Send_field_extended_metadata &src)
+ {
+ for (uint i= 0; i <= MARIADB_FIELD_ATTR_LAST; i++)
+ {
+ LEX_CSTRING attr= src.attr(i);
+ metadata[i]= ma_const_string_copy_root(memroot, attr.str, attr.length);
+ }
+ }
+};
+
+
+bool Protocol_text::store_field_metadata(const THD * thd,
+ const Send_field &server_field,
+ CHARSET_INFO *charset_for_protocol,
+ uint pos)
+{
+ CHARSET_INFO *cs= system_charset_info;
+ CHARSET_INFO *thd_cs= thd->variables.character_set_results;
+ MYSQL_DATA *data= thd->cur_data;
+ MEM_ROOT *field_alloc= &data->alloc;
+ MYSQL_FIELD *client_field= &thd->cur_data->embedded_info->fields_list[pos];
+ DBUG_ASSERT(server_field.is_sane());
+
+ client_field->db= dup_str_aux(field_alloc, server_field.db_name,
+ cs, thd_cs);
+ client_field->table= dup_str_aux(field_alloc, server_field.table_name,
+ cs, thd_cs);
+ client_field->name= dup_str_aux(field_alloc, server_field.col_name,
+ cs, thd_cs);
+ client_field->org_table= dup_str_aux(field_alloc, server_field.org_table_name,
+ cs, thd_cs);
+ client_field->org_name= dup_str_aux(field_alloc, server_field.org_col_name,
+ cs, thd_cs);
+ if (charset_for_protocol == &my_charset_bin || thd_cs == NULL)
+ {
+ /* No conversion */
+ client_field->charsetnr= charset_for_protocol->
+ get_id(MY_COLLATION_ID_TYPE_COMPAT_100800);
+ client_field->length= server_field.length;
+ }
+ else
+ {
+ /* With conversion */
+ client_field->charsetnr= thd_cs->
+ get_id(MY_COLLATION_ID_TYPE_COMPAT_100800);
+ client_field->length= server_field.max_octet_length(charset_for_protocol,
+ thd_cs);
+ }
+ client_field->type= server_field.type_handler()->type_code_for_protocol();
+ client_field->flags= (uint16) server_field.flags;
+ client_field->decimals= server_field.decimals;
+
+ client_field->db_length= (uint)strlen(client_field->db);
+ client_field->table_length= (uint)strlen(client_field->table);
+ client_field->name_length= (uint)strlen(client_field->name);
+ client_field->org_name_length= (uint)strlen(client_field->org_name);
+ client_field->org_table_length= (uint)strlen(client_field->org_table);
+
+ client_field->catalog= dup_str_aux(field_alloc, "def", 3, cs, thd_cs);
+ client_field->catalog_length= 3;
+
+ if (server_field.has_extended_metadata())
+ {
+ Client_field_extension *ext= new (field_alloc) Client_field_extension();
+ if ((client_field->extension= static_cast<MARIADB_FIELD_EXTENSION*>(ext)))
+ ext->copy_extended_metadata(field_alloc, server_field);
+ }
+ else
+ {
+ client_field->extension= NULL;
+ }
+
+ if (IS_NUM(client_field->type))
+ client_field->flags|= NUM_FLAG;
+
+ client_field->max_length= 0;
+ client_field->def= 0;
+ return false;
+}
+
+
+bool Protocol::send_result_set_metadata(List<Item> *list, uint flags)
+{
+ List_iterator_fast<Item> it(*list);
+ Item *item;
+ Protocol_text prot(thd);
+ DBUG_ENTER("send_result_set_metadata");
+
+ if (!thd->mysql) // bootstrap file handling
+ DBUG_RETURN(0);
+
+ if (begin_dataset(thd, list->elements))
+ goto err;
+
+ for (uint pos= 0 ; (item= it++); pos++)
+ {
+ if (prot.store_item_metadata(thd, item, pos))
+ goto err;
+ }
+
+ if (flags & SEND_EOF)
+ write_eof_packet(thd, thd->server_status,
+ thd->get_stmt_da()->current_statement_warn_count());
+
+ DBUG_RETURN(prepare_for_send(list->elements));
+ err:
+ my_error(ER_OUT_OF_RESOURCES, MYF(0)); /* purecov: inspected */
+ DBUG_RETURN(1); /* purecov: inspected */
+}
+
+
+static void
+list_fields_send_default(THD *thd, Protocol *p, Field *fld, uint pos)
+{
+ char buff[80];
+ String tmp(buff, sizeof(buff), default_charset_info), *res;
+ MYSQL_FIELD *client_field= &thd->cur_data->embedded_info->fields_list[pos];
+
+ if (fld->is_null() || !(res= fld->val_str(&tmp)))
+ {
+ client_field->def_length= 0;
+ client_field->def= strmake_root(&thd->cur_data->alloc, "", 0);
+ }
+ else
+ {
+ client_field->def_length= res->length();
+ client_field->def= strmake_root(&thd->cur_data->alloc, res->ptr(),
+ client_field->def_length);
+ }
+}
+
+
+bool Protocol::send_list_fields(List<Field> *list, const TABLE_LIST *table_list)
+{
+ DBUG_ENTER("send_result_set_metadata");
+ Protocol_text prot(thd);
+ List_iterator_fast<Field> it(*list);
+ Field *fld;
+
+ if (!thd->mysql) // bootstrap file handling
+ DBUG_RETURN(0);
+
+ if (begin_dataset(thd, list->elements))
+ goto err;
+
+ for (uint pos= 0 ; (fld= it++); pos++)
+ {
+ if (prot.store_field_metadata_for_list_fields(thd, fld, table_list, pos))
+ goto err;
+ list_fields_send_default(thd, this, fld, pos);
+ }
+
+ DBUG_RETURN(prepare_for_send(list->elements));
+err:
+ my_error(ER_OUT_OF_RESOURCES, MYF(0));
+ DBUG_RETURN(1);
+}
+
+
+bool Protocol::write()
+{
+ if (!thd->mysql) // bootstrap file handling
+ return false;
+
+ *next_field= 0;
+ return false;
+}
+
+bool Protocol_binary::write()
+{
+ MYSQL_ROWS *cur;
+ MYSQL_DATA *data= thd->cur_data;
+
+ data->rows++;
+ if (!(cur= (MYSQL_ROWS *)alloc_root(alloc,
+ sizeof(MYSQL_ROWS)+packet->length())))
+ {
+ my_error(ER_OUT_OF_RESOURCES,MYF(0));
+ return true;
+ }
+ cur->data= (MYSQL_ROW)(((char *)cur) + sizeof(MYSQL_ROWS));
+ memcpy(cur->data, packet->ptr()+1, packet->length()-1);
+ cur->length= packet->length(); /* To allow us to do sanity checks */
+
+ *data->embedded_info->prev_ptr= cur;
+ data->embedded_info->prev_ptr= &cur->next;
+ cur->next= 0;
+
+ return false;
+}
+
+
+/**
+ Embedded library implementation of OK response.
+
+ This function is used by the server to write 'OK' packet to
+ the "network" when the server is compiled as an embedded library.
+ Since there is no network in the embedded configuration,
+ a different implementation is necessary.
+ Instead of marshalling response parameters to a network representation
+ and then writing it to the socket, here we simply copy the data to the
+ corresponding client-side connection structures.
+
+ @sa Server implementation of net_send_ok in protocol.cc for
+ description of the arguments.
+
+ @return
+ @retval TRUE An error occurred
+ @retval FALSE Success
+*/
+
+bool Protocol::net_send_ok(THD *thd,
+ uint server_status, uint statement_warn_count,
+ ulonglong affected_rows, ulonglong id, const char *message,
+ bool)
+{
+ DBUG_ENTER("emb_net_send_ok");
+ MYSQL_DATA *data;
+ MYSQL *mysql= thd->mysql;
+
+ if (!mysql) // bootstrap file handling
+ DBUG_RETURN(FALSE);
+ if (!(data= thd->alloc_new_dataset()))
+ DBUG_RETURN(TRUE);
+ data->embedded_info->affected_rows= affected_rows;
+ data->embedded_info->insert_id= id;
+ if (message)
+ strmake_buf(data->embedded_info->info, message);
+
+ bool error= write_eof_packet(thd, server_status, statement_warn_count);
+ thd->cur_data= 0;
+ DBUG_RETURN(error);
+}
+
+
+/**
+ Embedded library implementation of EOF response.
+
+ @sa net_send_ok
+
+ @return
+ @retval TRUE An error occurred
+ @retval FALSE Success
+*/
+
+bool
+Protocol::net_send_eof(THD *thd, uint server_status, uint statement_warn_count)
+{
+ bool error= write_eof_packet(thd, server_status, statement_warn_count);
+ thd->cur_data= 0;
+ return error;
+}
+
+
+bool Protocol::net_send_error_packet(THD *thd, uint sql_errno, const char *err,
+ const char *sqlstate)
+{
+ uint error;
+ char converted_err[MYSQL_ERRMSG_SIZE];
+ MYSQL_DATA *data= thd->cur_data;
+ struct embedded_query_result *ei;
+
+ if (!thd->mysql) // bootstrap file handling
+ {
+ fprintf(stderr, "ERROR: %d %s\n", sql_errno, err);
+ return TRUE;
+ }
+
+ if (!data)
+ data= thd->alloc_new_dataset();
+
+ ei= data->embedded_info;
+ ei->last_errno= sql_errno;
+ convert_error_message(converted_err, sizeof(converted_err),
+ thd->variables.character_set_results,
+ err, strlen(err),
+ system_charset_info, &error);
+ /* Converted error message is always null-terminated. */
+ strmake_buf(ei->info, converted_err);
+ strmov(ei->sqlstate, sqlstate);
+ ei->server_status= thd->server_status;
+ thd->cur_data= 0;
+ return FALSE;
+}
+
+
+void Protocol_text::prepare_for_resend()
+{
+ MYSQL_ROWS *cur;
+ MYSQL_DATA *data= thd->cur_data;
+ DBUG_ENTER("send_data");
+
+ if (!thd->mysql) // bootstrap file handling
+ DBUG_VOID_RETURN;
+
+ data->rows++;
+ if (!(cur= (MYSQL_ROWS *)alloc_root(alloc, sizeof(MYSQL_ROWS)+(field_count + 1) * sizeof(char *))))
+ {
+ my_error(ER_OUT_OF_RESOURCES,MYF(0));
+ DBUG_VOID_RETURN;
+ }
+ cur->data= (MYSQL_ROW)(((char *)cur) + sizeof(MYSQL_ROWS));
+
+ *data->embedded_info->prev_ptr= cur;
+ data->embedded_info->prev_ptr= &cur->next;
+ next_field=cur->data;
+ next_mysql_field= data->embedded_info->fields_list;
+#ifndef DBUG_OFF
+ field_pos= 0;
+#endif
+
+ DBUG_VOID_RETURN;
+}
+
+bool Protocol_text::store_null()
+{
+ *(next_field++)= NULL;
+ ++next_mysql_field;
+ return false;
+}
+
+
+bool Protocol::net_store_data(const uchar *from, size_t length)
+{
+ char *field_buf;
+ if (!thd->mysql) // bootstrap file handling
+ return FALSE;
+
+ if (!(field_buf= (char*) alloc_root(alloc, length + sizeof(uint) + 1)))
+ return TRUE;
+ *(uint *)field_buf= (uint)length;
+ *next_field= field_buf + sizeof(uint);
+ memcpy((uchar*) *next_field, from, length);
+ (*next_field)[length]= 0;
+ if (next_mysql_field->max_length < length)
+ next_mysql_field->max_length=(ulong)length;
+ ++next_field;
+ ++next_mysql_field;
+ return FALSE;
+}
+
+
+bool Protocol::net_store_data_cs(const uchar *from, size_t length,
+ CHARSET_INFO *from_cs, CHARSET_INFO *to_cs)
+{
+ size_t conv_length= length * to_cs->mbmaxlen / from_cs->mbminlen;
+ uint dummy_error;
+ char *field_buf;
+ if (!thd->mysql) // bootstrap file handling
+ return false;
+
+ if (!(field_buf= (char*) alloc_root(alloc, conv_length + sizeof(uint) + 1)))
+ return true;
+ *next_field= field_buf + sizeof(uint);
+ length= copy_and_convert(*next_field, conv_length, to_cs,
+ (const char*) from, length, from_cs, &dummy_error);
+ *(uint *) field_buf= (uint)length;
+ (*next_field)[length]= 0;
+ if (next_mysql_field->max_length < length)
+ next_mysql_field->max_length= (ulong)length;
+ ++next_field;
+ ++next_mysql_field;
+ return false;
+}
+
+#if defined(_MSC_VER) && _MSC_VER < 1400
+#define vsnprintf _vsnprintf
+#endif
+
+int vprint_msg_to_log(enum loglevel level __attribute__((unused)),
+ const char *format, va_list argsi)
+{
+ vsnprintf(mysql_server_last_error, sizeof(mysql_server_last_error),
+ format, argsi);
+ mysql_server_last_errno= CR_UNKNOWN_ERROR;
+ if (embedded_print_errors && level == ERROR_LEVEL)
+ {
+ /* The following is for testing when someone removes the above test */
+ const char *tag= (level == ERROR_LEVEL ? "ERROR" :
+ level == WARNING_LEVEL ? "Warning" :
+ "Note");
+ fprintf(stderr,"Got %s: \"%s\" errno: %d\n",
+ tag, mysql_server_last_error, mysql_server_last_errno);
+ }
+ return 0;
+}
+