~drizzle-trunk/drizzle/development

/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 *  Copyright (C) 2008 Sun Microsystems

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; version 2 of the License.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

/**

  @file Handling of MySQL SQL variables

  @details

  To add a new variable, one has to do the following:

  - Use one of the 'sys_var... classes from set_var.h or write a specific

    one for the variable type.

  - Define it in the 'variable definition list' in this file.

  - If the variable is thread specific, add it to 'system_variables' struct.

    If not, add it to mysqld.cc and an declaration in 'mysql_priv.h'

  - If the variable should be changed from the command line, add a definition

    of it in the my_option structure list in mysqld.cc

  - Don't forget to initialize new fields in global_system_variables and

    max_system_variables!

  @todo

    Add full support for the variable character_set (for 4.1)

  @note

    Be careful with var->save_result: sys_var::check() only updates

    uint64_t_value; so other members of the union are garbage then; to use

    them you must first assign a value to them (in specific ::check() for

    example).

*/

#include "config.h"

#include "drizzled/my_getopt.h"

#include <drizzled/error.h>

#include <drizzled/gettext.h>

#include <drizzled/tztime.h>

#include <drizzled/data_home.h>

#include <drizzled/set_var.h>

#include <drizzled/session.h>

#include <drizzled/sql_base.h>

#include <drizzled/lock.h>

#include <drizzled/item/uint.h>

#include <drizzled/item/null.h>

#include <drizzled/item/float.h>

#include <drizzled/plugin.h>

#include "drizzled/version.h"

#include "drizzled/strfunc.h"

#include "drizzled/internal/m_string.h"

#include "drizzled/pthread_globals.h"

#include "drizzled/charset.h"

#include "drizzled/transaction_services.h"

#include <map>

#include <algorithm>

using namespace std;

namespace drizzled

{

namespace internal

{

extern bool timed_mutexes;

}

extern plugin::StorageEngine *myisam_engine;

extern bool timed_mutexes;

extern struct my_option my_long_options[];

extern const CHARSET_INFO *character_set_filesystem;

extern size_t my_thread_stack_size;

class sys_var_pluginvar;

static DYNAMIC_ARRAY fixed_show_vars;

typedef map<string, sys_var *> SystemVariableMap;

static SystemVariableMap system_variable_map;

extern char *opt_drizzle_tmpdir;

extern TYPELIB tx_isolation_typelib;

const char *bool_type_names[]= { "OFF", "ON", NULL };

TYPELIB bool_typelib=

{

  array_elements(bool_type_names)-1, "", bool_type_names, NULL

};

static bool set_option_bit(Session *session, set_var *var);

static bool set_option_autocommit(Session *session, set_var *var);

static int  check_pseudo_thread_id(Session *session, set_var *var);

static int check_tx_isolation(Session *session, set_var *var);

static void fix_tx_isolation(Session *session, sql_var_t type);

static int check_completion_type(Session *session, set_var *var);

static void fix_completion_type(Session *session, sql_var_t type);

static void fix_max_join_size(Session *session, sql_var_t type);

static void fix_session_mem_root(Session *session, sql_var_t type);

static void fix_server_id(Session *session, sql_var_t type);

static bool get_unsigned32(Session *session, set_var *var);

static bool get_unsigned64(Session *session, set_var *var);

bool throw_bounds_warning(Session *session, bool fixed, bool unsignd,

                          const std::string &name, int64_t val);

static unsigned char *get_error_count(Session *session);

static unsigned char *get_warning_count(Session *session);

static unsigned char *get_tmpdir(Session *session);

/*

  Variable definition list

  These are variables that can be set from the command line, in

  alphabetic order.

  The variables are linked into the list. A variable is added to

  it in the constructor (see sys_var class for details).

*/

static sys_var_chain vars = { NULL, NULL };

static sys_var_session_uint64_t

sys_auto_increment_increment(&vars, "auto_increment_increment",

                             &system_variables::auto_increment_increment);

static sys_var_session_uint64_t

sys_auto_increment_offset(&vars, "auto_increment_offset",

                          &system_variables::auto_increment_offset);

static sys_var_const_str       sys_basedir(&vars, "basedir", drizzle_home);

static sys_var_session_uint64_t	sys_bulk_insert_buff_size(&vars, "bulk_insert_buffer_size",

                                                          &system_variables::bulk_insert_buff_size);

static sys_var_session_uint32_t	sys_completion_type(&vars, "completion_type",

                                                    &system_variables::completion_type,

                                                    check_completion_type,

                                                    fix_completion_type);

static sys_var_collation_sv

sys_collation_server(&vars, "collation_server", &system_variables::collation_server, &default_charset_info);

static sys_var_const_str       sys_datadir(&vars, "datadir", drizzle_real_data_home);

static sys_var_session_uint64_t	sys_join_buffer_size(&vars, "join_buffer_size",

                                                     &system_variables::join_buff_size);

static sys_var_session_uint32_t	sys_max_allowed_packet(&vars, "max_allowed_packet",

                                                       &system_variables::max_allowed_packet);

static sys_var_uint64_t_ptr	sys_max_connect_errors(&vars, "max_connect_errors",

                                               &max_connect_errors);

static sys_var_session_uint64_t	sys_max_error_count(&vars, "max_error_count",

                                                  &system_variables::max_error_count);

static sys_var_session_uint64_t	sys_max_heap_table_size(&vars, "max_heap_table_size",

                                                        &system_variables::max_heap_table_size);

static sys_var_session_uint64_t sys_pseudo_thread_id(&vars, "pseudo_thread_id",

                                              &system_variables::pseudo_thread_id,

                                              0, check_pseudo_thread_id);

static sys_var_session_ha_rows	sys_max_join_size(&vars, "max_join_size",

                                                  &system_variables::max_join_size,

                                                  fix_max_join_size);

static sys_var_session_uint64_t	sys_max_seeks_for_key(&vars, "max_seeks_for_key",

                                                      &system_variables::max_seeks_for_key);

static sys_var_session_uint64_t   sys_max_length_for_sort_data(&vars, "max_length_for_sort_data",

                                                               &system_variables::max_length_for_sort_data);

static sys_var_session_size_t	sys_max_sort_length(&vars, "max_sort_length",

                                                    &system_variables::max_sort_length);

static sys_var_uint64_t_ptr	sys_max_write_lock_count(&vars, "max_write_lock_count",

                                                 &max_write_lock_count);

static sys_var_session_uint64_t sys_min_examined_row_limit(&vars, "min_examined_row_limit",

                                                           &system_variables::min_examined_row_limit);

/* these two cannot be static */

static sys_var_session_bool sys_optimizer_prune_level(&vars, "optimizer_prune_level",

                                                      &system_variables::optimizer_prune_level);

static sys_var_session_uint32_t sys_optimizer_search_depth(&vars, "optimizer_search_depth",

                                                           &system_variables::optimizer_search_depth);

static sys_var_session_uint64_t sys_preload_buff_size(&vars, "preload_buffer_size",

                                                      &system_variables::preload_buff_size);

static sys_var_session_uint32_t sys_read_buff_size(&vars, "read_buffer_size",

                                                   &system_variables::read_buff_size);

static sys_var_session_uint32_t	sys_read_rnd_buff_size(&vars, "read_rnd_buffer_size",

                                                       &system_variables::read_rnd_buff_size);

static sys_var_session_uint32_t	sys_div_precincrement(&vars, "div_precision_increment",

                                                      &system_variables::div_precincrement);

static sys_var_session_size_t	sys_range_alloc_block_size(&vars, "range_alloc_block_size",

                                                           &system_variables::range_alloc_block_size);

static sys_var_session_uint32_t	sys_query_alloc_block_size(&vars, "query_alloc_block_size",

                                                           &system_variables::query_alloc_block_size,

                                                           false, fix_session_mem_root);

static sys_var_session_uint32_t	sys_query_prealloc_size(&vars, "query_prealloc_size",

                                                        &system_variables::query_prealloc_size,

                                                        false, fix_session_mem_root);

static sys_var_readonly sys_tmpdir(&vars, "tmpdir", OPT_GLOBAL, SHOW_CHAR, get_tmpdir);

static sys_var_const_str_ptr sys_secure_file_priv(&vars, "secure_file_priv",

                                             &opt_secure_file_priv);

static sys_var_uint32_t_ptr  sys_server_id(&vars, "server_id", &server_id,

                                           fix_server_id);

static sys_var_session_size_t	sys_sort_buffer(&vars, "sort_buffer_size",

                                                &system_variables::sortbuff_size);

static sys_var_session_storage_engine sys_storage_engine(&vars, "storage_engine",

				       &system_variables::storage_engine);

static sys_var_const_str	sys_system_time_zone(&vars, "system_time_zone",

                                             system_time_zone);

static sys_var_size_t_ptr	sys_table_def_size(&vars, "table_definition_cache",

                                             &table_def_size);

static sys_var_uint64_t_ptr	sys_table_cache_size(&vars, "table_open_cache",

					     &table_cache_size);

static sys_var_uint64_t_ptr	sys_table_lock_wait_timeout(&vars, "table_lock_wait_timeout",

                                                    &table_lock_wait_timeout);

static sys_var_session_enum	sys_tx_isolation(&vars, "tx_isolation",

                                             &system_variables::tx_isolation,

                                             &tx_isolation_typelib,

                                             fix_tx_isolation,

                                             check_tx_isolation);

static sys_var_session_uint64_t	sys_tmp_table_size(&vars, "tmp_table_size",

					   &system_variables::tmp_table_size);

static sys_var_bool_ptr  sys_timed_mutexes(&vars, "timed_mutexes", &internal::timed_mutexes);

static sys_var_const_str  sys_version(&vars, "version", version().c_str());

static sys_var_const_str	sys_version_comment(&vars, "version_comment",

                                            COMPILATION_COMMENT);

static sys_var_const_str	sys_version_compile_machine(&vars, "version_compile_machine",

                                                      HOST_CPU);

static sys_var_const_str	sys_version_compile_os(&vars, "version_compile_os",

                                                 HOST_OS);

static sys_var_const_str	sys_version_compile_vendor(&vars, "version_compile_vendor",

                                                 HOST_VENDOR);

/* Variables that are bits in Session */

sys_var_session_bit sys_autocommit(&vars, "autocommit", 0,

                               set_option_autocommit,

                               OPTION_NOT_AUTOCOMMIT,

                               1);

static sys_var_session_bit	sys_big_selects(&vars, "sql_big_selects", 0,

					set_option_bit,

					OPTION_BIG_SELECTS);

static sys_var_session_bit	sys_sql_warnings(&vars, "sql_warnings", 0,

					 set_option_bit,

					 OPTION_WARNINGS);

static sys_var_session_bit	sys_sql_notes(&vars, "sql_notes", 0,

					 set_option_bit,

					 OPTION_SQL_NOTES);

static sys_var_session_bit	sys_buffer_results(&vars, "sql_buffer_result", 0,

					   set_option_bit,

					   OPTION_BUFFER_RESULT);

static sys_var_session_bit	sys_foreign_key_checks(&vars, "foreign_key_checks", 0,

					       set_option_bit,

					       OPTION_NO_FOREIGN_KEY_CHECKS, 1);

static sys_var_session_bit	sys_unique_checks(&vars, "unique_checks", 0,

					  set_option_bit,

					  OPTION_RELAXED_UNIQUE_CHECKS, 1);

/* Local state variables */

static sys_var_session_ha_rows	sys_select_limit(&vars, "sql_select_limit",

						 &system_variables::select_limit);

static sys_var_timestamp sys_timestamp(&vars, "timestamp");

static sys_var_last_insert_id

sys_last_insert_id(&vars, "last_insert_id");

/*

  identity is an alias for last_insert_id(), so that we are compatible

  with Sybase

*/

static sys_var_last_insert_id sys_identity(&vars, "identity");

static sys_var_session_lc_time_names sys_lc_time_names(&vars, "lc_time_names");

/*

  We want statements referring explicitly to @@session.insert_id to be

  unsafe, because insert_id is modified internally by the slave sql

  thread when NULL values are inserted in an AUTO_INCREMENT column.

  This modification interfers with the value of the

  @@session.insert_id variable if @@session.insert_id is referred

  explicitly by an insert statement (as is seen by executing "SET

  @@session.insert_id=0; CREATE TABLE t (a INT, b INT KEY

  AUTO_INCREMENT); INSERT INTO t(a) VALUES (@@session.insert_id);" in

  statement-based logging mode: t will be different on master and

  slave).

*/

static sys_var_readonly sys_error_count(&vars, "error_count",

                                        OPT_SESSION,

                                        SHOW_INT,

                                        get_error_count);

static sys_var_readonly sys_warning_count(&vars, "warning_count",

                                          OPT_SESSION,

                                          SHOW_INT,

                                          get_warning_count);

sys_var_session_uint64_t sys_group_concat_max_len(&vars, "group_concat_max_len",

                                                  &system_variables::group_concat_max_len);

sys_var_session_time_zone sys_time_zone(&vars, "time_zone");

/* Global read-only variable containing hostname */

static sys_var_const_str        sys_hostname(&vars, "hostname", glob_hostname);

/*

  Additional variables (not derived from sys_var class, not accessible as

  @@varname in SELECT or SET). Sorted in alphabetical order to facilitate

  maintenance - SHOW VARIABLES will sort its output.

  TODO: remove this list completely

*/

#define FIXED_VARS_SIZE (sizeof(fixed_vars) / sizeof(drizzle_show_var))

static drizzle_show_var fixed_vars[]= {

  {"back_log",                (char*) &back_log,                SHOW_INT},

  {"language",                language,                         SHOW_CHAR},

  {"pid_file",                (char*) pidfile_name,             SHOW_CHAR},

  {"plugin_dir",              (char*) opt_plugin_dir,           SHOW_CHAR},

  {"thread_stack",            (char*) &my_thread_stack_size,    SHOW_INT},

};

bool sys_var::check(Session *, set_var *var)

{

  var->save_result.uint64_t_value= var->value->val_int();

  return 0;

}

bool sys_var_str::check(Session *session, set_var *var)

{

  int res;

  if (!check_func)

    return 0;

  if ((res=(*check_func)(session, var)) < 0)

    my_error(ER_WRONG_VALUE_FOR_VAR, MYF(0), getName().c_str(), var->value->str_value.ptr());

  return res;

}

/*

  Functions to check and update variables

*/

/**

  Set the OPTION_BIG_SELECTS flag if max_join_size == HA_POS_ERROR.

*/

static void fix_max_join_size(Session *session, sql_var_t type)

{

  if (type != OPT_GLOBAL)

  {

    if (session->variables.max_join_size == HA_POS_ERROR)

      session->options|= OPTION_BIG_SELECTS;

    else

      session->options&= ~OPTION_BIG_SELECTS;

  }

}

/**

  Can't change the 'next' tx_isolation while we are already in

  a transaction

*/

static int check_tx_isolation(Session *session, set_var *var)

{

  if (var->type == OPT_DEFAULT && (session->server_status & SERVER_STATUS_IN_TRANS))

  {

    my_error(ER_CANT_CHANGE_TX_ISOLATION, MYF(0));

    return 1;

  }

  return 0;

}

/*

  If one doesn't use the SESSION modifier, the isolation level

  is only active for the next command.

*/

static void fix_tx_isolation(Session *session, sql_var_t type)

{

  if (type == OPT_SESSION)

    session->session_tx_isolation= ((enum_tx_isolation)

                                    session->variables.tx_isolation);

}

static void fix_completion_type(Session *, sql_var_t) {}

static int check_completion_type(Session *, set_var *var)

{

  int64_t val= var->value->val_int();

  if (val < 0 || val > 2)

  {

    char buf[64];

    my_error(ER_WRONG_VALUE_FOR_VAR, MYF(0), var->var->getName().c_str(), internal::llstr(val, buf));

    return 1;

  }

  return 0;

}

static void fix_session_mem_root(Session *session, sql_var_t type)

{

  if (type != OPT_GLOBAL)

    reset_root_defaults(session->mem_root,

                        session->variables.query_alloc_block_size,

                        session->variables.query_prealloc_size);

}

static void fix_server_id(Session *, sql_var_t)

{

}

bool throw_bounds_warning(Session *session, bool fixed, bool unsignd,

                          const std::string &name, int64_t val)

{

  if (fixed)

  {

    char buf[22];

    if (unsignd)

      internal::ullstr((uint64_t) val, buf);

    else

      internal::llstr(val, buf);

    push_warning_printf(session, DRIZZLE_ERROR::WARN_LEVEL_ERROR,

                        ER_TRUNCATED_WRONG_VALUE,

                        ER(ER_TRUNCATED_WRONG_VALUE), name.c_str(), buf);

  }

  return false;

}

uint64_t fix_unsigned(Session *session, uint64_t num,

                              const struct my_option *option_limits)

{

  bool fixed= false;

  uint64_t out= getopt_ull_limit_value(num, option_limits, &fixed);

  throw_bounds_warning(session, fixed, true, option_limits->name, (int64_t) num);

  return out;

}

static size_t fix_size_t(Session *session, size_t num,

                           const struct my_option *option_limits)

{

  bool fixed= false;

  size_t out= (size_t)getopt_ull_limit_value(num, option_limits, &fixed);

  throw_bounds_warning(session, fixed, true, option_limits->name, (int64_t) num);

  return out;

}

static bool get_unsigned32(Session *session, set_var *var)

{

  if (var->value->unsigned_flag)

    var->save_result.uint32_t_value= 

      static_cast<uint32_t>(var->value->val_int());

  else

  {

    int64_t v= var->value->val_int();

    if (v > UINT32_MAX)

      throw_bounds_warning(session, true, true,var->var->getName().c_str(), v);

    var->save_result.uint32_t_value= 

      static_cast<uint32_t>((v > UINT32_MAX) ? UINT32_MAX : (v < 0) ? 0 : v);

  }

  return false;

}

static bool get_unsigned64(Session *, set_var *var)

{

  if (var->value->unsigned_flag)

      var->save_result.uint64_t_value=(uint64_t) var->value->val_int();

  else

  {

    int64_t v= var->value->val_int();

      var->save_result.uint64_t_value= (uint64_t) ((v < 0) ? 0 : v);

  }

  return 0;

}

static bool get_size_t(Session *, set_var *var)

{

  if (var->value->unsigned_flag)

    var->save_result.size_t_value= (size_t) var->value->val_int();

  else

  {

    ssize_t v= (ssize_t)var->value->val_int();

    var->save_result.size_t_value= (size_t) ((v < 0) ? 0 : v);

  }

  return 0;

}

bool sys_var_uint32_t_ptr::check(Session *, set_var *var)

{

  var->save_result.uint32_t_value= (uint32_t)var->value->val_int();

  return 0;

}

bool sys_var_uint32_t_ptr::update(Session *session, set_var *var)

{

  uint32_t tmp= var->save_result.uint32_t_value;

  pthread_mutex_lock(&LOCK_global_system_variables);

  if (option_limits)

  {

    uint32_t newvalue= (uint32_t) fix_unsigned(session, tmp, option_limits);

    if(newvalue==tmp)

      *value= newvalue;

  }

  else

    *value= (uint32_t) tmp;

  pthread_mutex_unlock(&LOCK_global_system_variables);

  return 0;

}

void sys_var_uint32_t_ptr::set_default(Session *, sql_var_t)

{

  bool not_used;

  pthread_mutex_lock(&LOCK_global_system_variables);

  *value= (uint32_t)getopt_ull_limit_value((uint32_t) option_limits->def_value,

                                           option_limits, &not_used);

  pthread_mutex_unlock(&LOCK_global_system_variables);

}

bool sys_var_uint64_t_ptr::update(Session *session, set_var *var)

{

  uint64_t tmp= var->save_result.uint64_t_value;

  pthread_mutex_lock(&LOCK_global_system_variables);

  if (option_limits)

  {

    uint64_t newvalue= (uint64_t) fix_unsigned(session, tmp, option_limits);

    if(newvalue==tmp)

      *value= newvalue;

  }

  else

    *value= (uint64_t) tmp;

  pthread_mutex_unlock(&LOCK_global_system_variables);

  return 0;

}

void sys_var_uint64_t_ptr::set_default(Session *, sql_var_t)

{

  bool not_used;

  pthread_mutex_lock(&LOCK_global_system_variables);

  *value= getopt_ull_limit_value((uint64_t) option_limits->def_value,

                                 option_limits, &not_used);

  pthread_mutex_unlock(&LOCK_global_system_variables);

}

bool sys_var_size_t_ptr::update(Session *session, set_var *var)

{

  size_t tmp= var->save_result.size_t_value;

  pthread_mutex_lock(&LOCK_global_system_variables);

  if (option_limits)

    *value= fix_size_t(session, tmp, option_limits);

  else

    *value= tmp;

  pthread_mutex_unlock(&LOCK_global_system_variables);

  return 0;

}

void sys_var_size_t_ptr::set_default(Session *, sql_var_t)

{

  bool not_used;

  pthread_mutex_lock(&LOCK_global_system_variables);

  *value= (size_t)getopt_ull_limit_value((size_t) option_limits->def_value,

                                         option_limits, &not_used);

  pthread_mutex_unlock(&LOCK_global_system_variables);

}

bool sys_var_bool_ptr::update(Session *, set_var *var)

{

  *value= (bool) var->save_result.uint32_t_value;

  return 0;

}

void sys_var_bool_ptr::set_default(Session *, sql_var_t)

{

  *value= (bool) option_limits->def_value;

}

/*

  32 bit types for session variables

*/

bool sys_var_session_uint32_t::check(Session *session, set_var *var)

{

  return (get_unsigned32(session, var) ||

          (check_func && (*check_func)(session, var)));

}

bool sys_var_session_uint32_t::update(Session *session, set_var *var)

{

  uint64_t tmp= (uint64_t) var->save_result.uint32_t_value;

  /* Don't use bigger value than given with --maximum-variable-name=.. */

  if ((uint32_t) tmp > max_system_variables.*offset)

  {

    throw_bounds_warning(session, true, true, getName(), (int64_t) tmp);

    tmp= max_system_variables.*offset;

  }

  if (option_limits)

    tmp= (uint32_t) fix_unsigned(session, tmp, option_limits);

  else if (tmp > UINT32_MAX)

  {

    tmp= UINT32_MAX;

    throw_bounds_warning(session, true, true, getName(), (int64_t) var->save_result.uint64_t_value);

  }

  if (var->type == OPT_GLOBAL)