← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/session.cc

  • Committer: Jay Pipes
  • Date: 2009-02-21 16:00:06 UTC
  • mto: (907.1.1 trunk-with-temporal)
  • mto: This revision was merged to the branch mainline in revision 908.
  • Revision ID: jpipes@serialcoder-20090221160006-vnk3wt4qbcz62eru
Removes the TIME column type and related time functions.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/session.cc
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
/**
21
 
 * @file Implementation of the Session class and API
22
 
 */
23
20
 
24
 
#include "config.h"
 
21
/*****************************************************************************
 
22
**
 
23
** This file implements classes defined in session.h
 
24
** Especially the classes to handle a result from a select
 
25
**
 
26
*****************************************************************************/
 
27
#include <drizzled/server_includes.h>
25
28
#include <drizzled/session.h>
26
 
#include "drizzled/session_list.h"
27
29
#include <sys/stat.h>
 
30
#include <mysys/mysys_err.h>
28
31
#include <drizzled/error.h>
29
 
#include <drizzled/gettext.h>
30
32
#include <drizzled/query_id.h>
31
33
#include <drizzled/data_home.h>
32
34
#include <drizzled/sql_base.h>
36
38
#include <drizzled/item/return_int.h>
37
39
#include <drizzled/item/empty_string.h>
38
40
#include <drizzled/show.h>
39
 
#include <drizzled/plugin/client.h>
40
 
#include "drizzled/plugin/scheduler.h"
41
 
#include "drizzled/plugin/authentication.h"
42
 
#include "drizzled/plugin/logging.h"
43
 
#include "drizzled/plugin/transactional_storage_engine.h"
44
 
#include "drizzled/probes.h"
45
 
#include "drizzled/table_proto.h"
46
 
#include "drizzled/db.h"
47
 
#include "drizzled/pthread_globals.h"
48
 
#include "drizzled/transaction_services.h"
49
 
 
50
 
#include "plugin/myisam/myisam.h"
51
 
#include "drizzled/internal/iocache.h"
52
 
 
53
 
#include <fcntl.h>
54
 
#include <algorithm>
55
 
#include <climits>
56
 
 
57
 
using namespace std;
58
 
namespace drizzled
59
 
{
60
 
 
61
 
extern "C"
62
 
{
63
 
  unsigned char *get_var_key(user_var_entry *entry, size_t *length, bool );
64
 
  void free_user_var(user_var_entry *entry);
65
 
}
66
 
 
 
41
#include <drizzled/plugin_scheduling.h>
 
42
 
 
43
extern scheduling_st thread_scheduler;
67
44
/*
68
45
  The following is used to initialise Table_ident with a internal
69
46
  table name
74
51
const char * const Session::DEFAULT_WHERE= "field list";
75
52
extern pthread_key_t THR_Session;
76
53
extern pthread_key_t THR_Mem_root;
77
 
extern uint32_t max_used_connections;
78
 
extern atomic<uint32_t> connection_count;
 
54
 
 
55
 
 
56
/*****************************************************************************
 
57
** Instansiate templates
 
58
*****************************************************************************/
 
59
 
 
60
#ifdef HAVE_EXPLICIT_TEMPLATE_INSTANTIATION
 
61
/* Used templates */
 
62
template class List<Key>;
 
63
template class List_iterator<Key>;
 
64
template class List<Key_part_spec>;
 
65
template class List_iterator<Key_part_spec>;
 
66
template class List<Alter_drop>;
 
67
template class List_iterator<Alter_drop>;
 
68
template class List<Alter_column>;
 
69
template class List_iterator<Alter_column>;
 
70
#endif
79
71
 
80
72
 
81
73
/****************************************************************************
82
74
** User variables
83
75
****************************************************************************/
84
 
unsigned char *get_var_key(user_var_entry *entry, size_t *length, bool )
 
76
 
 
77
extern "C" unsigned char *get_var_key(user_var_entry *entry, size_t *length,
 
78
                              bool )
85
79
{
86
80
  *length= entry->name.length;
87
81
  return (unsigned char*) entry->name.str;
88
82
}
89
83
 
90
 
void free_user_var(user_var_entry *entry)
 
84
extern "C" void free_user_var(user_var_entry *entry)
91
85
{
92
 
  delete entry;
 
86
  char *pos= (char*) entry+ALIGN_SIZE(sizeof(*entry));
 
87
  if (entry->value && entry->value != pos)
 
88
    free(entry->value);
 
89
  free((char*) entry);
93
90
}
94
91
 
95
92
bool Key_part_spec::operator==(const Key_part_spec& other) const
99
96
         !strcmp(field_name.str, other.field_name.str);
100
97
}
101
98
 
102
 
Open_tables_state::Open_tables_state(uint64_t version_arg)
103
 
  :version(version_arg), backups_available(false)
 
99
/**
 
100
  Construct an (almost) deep copy of this key. Only those
 
101
  elements that are known to never change are not copied.
 
102
  If out of memory, a partial copy is returned and an error is set
 
103
  in Session.
 
104
*/
 
105
 
 
106
Key::Key(const Key &rhs, MEM_ROOT *mem_root)
 
107
  :type(rhs.type),
 
108
  key_create_info(rhs.key_create_info),
 
109
  columns(rhs.columns, mem_root),
 
110
  name(rhs.name),
 
111
  generated(rhs.generated)
 
112
{
 
113
  list_copy_and_replace_each_value(columns, mem_root);
 
114
}
 
115
 
 
116
/**
 
117
  Construct an (almost) deep copy of this foreign key. Only those
 
118
  elements that are known to never change are not copied.
 
119
  If out of memory, a partial copy is returned and an error is set
 
120
  in Session.
 
121
*/
 
122
 
 
123
Foreign_key::Foreign_key(const Foreign_key &rhs, MEM_ROOT *mem_root)
 
124
  :Key(rhs),
 
125
  ref_table(rhs.ref_table),
 
126
  ref_columns(rhs.ref_columns),
 
127
  delete_opt(rhs.delete_opt),
 
128
  update_opt(rhs.update_opt),
 
129
  match_opt(rhs.match_opt)
 
130
{
 
131
  list_copy_and_replace_each_value(ref_columns, mem_root);
 
132
}
 
133
 
 
134
/*
 
135
  Test if a foreign key (= generated key) is a prefix of the given key
 
136
  (ignoring key name, key type and order of columns)
 
137
 
 
138
  NOTES:
 
139
    This is only used to test if an index for a FOREIGN KEY exists
 
140
 
 
141
  IMPLEMENTATION
 
142
    We only compare field names
 
143
 
 
144
  RETURN
 
145
    0   Generated key is a prefix of other key
 
146
    1   Not equal
 
147
*/
 
148
 
 
149
bool foreign_key_prefix(Key *a, Key *b)
 
150
{
 
151
  /* Ensure that 'a' is the generated key */
 
152
  if (a->generated)
 
153
  {
 
154
    if (b->generated && a->columns.elements > b->columns.elements)
 
155
      std::swap(a, b);                       // Put shorter key in 'a'
 
156
  }
 
157
  else
 
158
  {
 
159
    if (!b->generated)
 
160
      return true;                              // No foreign key
 
161
    std::swap(a, b);                       // Put generated key in 'a'
 
162
  }
 
163
 
 
164
  /* Test if 'a' is a prefix of 'b' */
 
165
  if (a->columns.elements > b->columns.elements)
 
166
    return true;                                // Can't be prefix
 
167
 
 
168
  List_iterator<Key_part_spec> col_it1(a->columns);
 
169
  List_iterator<Key_part_spec> col_it2(b->columns);
 
170
  const Key_part_spec *col1, *col2;
 
171
 
 
172
#ifdef ENABLE_WHEN_INNODB_CAN_HANDLE_SWAPED_FOREIGN_KEY_COLUMNS
 
173
  while ((col1= col_it1++))
 
174
  {
 
175
    bool found= 0;
 
176
    col_it2.rewind();
 
177
    while ((col2= col_it2++))
 
178
    {
 
179
      if (*col1 == *col2)
 
180
      {
 
181
        found= true;
 
182
        break;
 
183
      }
 
184
    }
 
185
    if (!found)
 
186
      return true;                              // Error
 
187
  }
 
188
  return false;                                 // Is prefix
 
189
#else
 
190
  while ((col1= col_it1++))
 
191
  {
 
192
    col2= col_it2++;
 
193
    if (!(*col1 == *col2))
 
194
      return true;
 
195
  }
 
196
  return false;                                 // Is prefix
 
197
#endif
 
198
}
 
199
 
 
200
 
 
201
/*
 
202
  Check if the foreign key options are compatible with columns
 
203
  on which the FK is created.
 
204
 
 
205
  RETURN
 
206
    0   Key valid
 
207
    1   Key invalid
 
208
*/
 
209
bool Foreign_key::validate(List<Create_field> &table_fields)
 
210
{
 
211
  Create_field  *sql_field;
 
212
  Key_part_spec *column;
 
213
  List_iterator<Key_part_spec> cols(columns);
 
214
  List_iterator<Create_field> it(table_fields);
 
215
  while ((column= cols++))
 
216
  {
 
217
    it.rewind();
 
218
    while ((sql_field= it++) &&
 
219
           my_strcasecmp(system_charset_info,
 
220
                         column->field_name.str,
 
221
                         sql_field->field_name)) {}
 
222
    if (!sql_field)
 
223
    {
 
224
      my_error(ER_KEY_COLUMN_DOES_NOT_EXITS, MYF(0), column->field_name.str);
 
225
      return true;
 
226
    }
 
227
    if (type == Key::FOREIGN_KEY && sql_field->vcol_info)
 
228
    {
 
229
      if (delete_opt == FK_OPTION_SET_NULL)
 
230
      {
 
231
        my_error(ER_WRONG_FK_OPTION_FOR_VIRTUAL_COLUMN, MYF(0),
 
232
                 "ON DELETE SET NULL");
 
233
        return true;
 
234
      }
 
235
      if (update_opt == FK_OPTION_SET_NULL)
 
236
      {
 
237
        my_error(ER_WRONG_FK_OPTION_FOR_VIRTUAL_COLUMN, MYF(0),
 
238
                 "ON UPDATE SET NULL");
 
239
        return true;
 
240
      }
 
241
      if (update_opt == FK_OPTION_CASCADE)
 
242
      {
 
243
        my_error(ER_WRONG_FK_OPTION_FOR_VIRTUAL_COLUMN, MYF(0),
 
244
                 "ON UPDATE CASCADE");
 
245
        return true;
 
246
      }
 
247
    }
 
248
  }
 
249
  return false;
 
250
}
 
251
 
 
252
 
 
253
/****************************************************************************
 
254
** Thread specific functions
 
255
****************************************************************************/
 
256
 
 
257
Open_tables_state::Open_tables_state(ulong version_arg)
 
258
  :version(version_arg), state_flags(0U)
104
259
{
105
260
  reset_open_tables_state();
106
261
}
108
263
/*
109
264
  The following functions form part of the C plugin API
110
265
*/
 
266
 
111
267
extern "C" int mysql_tmpfile(const char *prefix)
112
268
{
113
269
  char filename[FN_REFLEN];
114
 
  int fd = internal::create_temp_file(filename, drizzle_tmpdir, prefix, MYF(MY_WME));
 
270
  File fd = create_temp_file(filename, drizzle_tmpdir, prefix,
 
271
                             O_CREAT | O_EXCL | O_RDWR,
 
272
                             MYF(MY_WME));
115
273
  if (fd >= 0) {
116
274
    unlink(filename);
117
275
  }
119
277
  return fd;
120
278
}
121
279
 
 
280
 
 
281
extern "C"
 
282
int session_in_lock_tables(const Session *session)
 
283
{
 
284
  return test(session->in_lock_tables);
 
285
}
 
286
 
 
287
 
122
288
extern "C"
123
289
int session_tablespace_op(const Session *session)
124
290
{
125
291
  return test(session->tablespace_op);
126
292
}
127
293
 
 
294
 
128
295
/**
129
296
   Set the process info field of the Session structure.
130
297
 
145
312
  return session->get_proc_info();
146
313
}
147
314
 
148
 
void **Session::getEngineData(const plugin::MonitoredInTransaction *monitored)
149
 
{
150
 
  return static_cast<void **>(&ha_data[monitored->getId()].ha_ptr);
151
 
}
152
 
 
153
 
ResourceContext *Session::getResourceContext(const plugin::MonitoredInTransaction *monitored,
154
 
                                             size_t index)
155
 
{
156
 
  return &ha_data[monitored->getId()].resource_context[index];
 
315
extern "C"
 
316
void **session_ha_data(const Session *session, const struct handlerton *hton)
 
317
{
 
318
  return (void **) &session->ha_data[hton->slot].ha_ptr;
157
319
}
158
320
 
159
321
extern "C"
174
336
  return (int) session->variables.tx_isolation;
175
337
}
176
338
 
177
 
Session::Session(plugin::Client *client_arg)
178
 
  :
179
 
  Open_tables_state(refresh_version),
180
 
  mem_root(&main_mem_root),
181
 
  lex(&main_lex),
182
 
  query(),
183
 
  client(client_arg),
184
 
  scheduler(NULL),
185
 
  scheduler_arg(NULL),
186
 
  lock_id(&main_lock_id),
187
 
  user_time(0),
188
 
  ha_data(plugin::num_trx_monitored_objects),
189
 
  arg_of_last_insert_id_function(false),
190
 
  first_successful_insert_id_in_prev_stmt(0),
191
 
  first_successful_insert_id_in_cur_stmt(0),
192
 
  limit_found_rows(0),
193
 
  global_read_lock(0),
194
 
  some_tables_deleted(false),
195
 
  no_errors(false),
196
 
  password(false),
197
 
  is_fatal_error(false),
198
 
  transaction_rollback_request(false),
199
 
  is_fatal_sub_stmt_error(0),
200
 
  derived_tables_processing(false),
201
 
  tablespace_op(false),
202
 
  m_lip(NULL),
203
 
  cached_table(0),
204
 
  transaction_message(NULL),
205
 
  statement_message(NULL)
206
 
{
207
 
  memset(process_list_info, 0, PROCESS_LIST_WIDTH);
208
 
  client->setSession(this);
 
339
extern "C"
 
340
void session_inc_row_count(Session *session)
 
341
{
 
342
  session->row_count++;
 
343
}
 
344
 
 
345
/**
 
346
  Clear this diagnostics area.
 
347
 
 
348
  Normally called at the end of a statement.
 
349
*/
 
350
 
 
351
void
 
352
Diagnostics_area::reset_diagnostics_area()
 
353
{
 
354
  can_overwrite_status= false;
 
355
  /** Don't take chances in production */
 
356
  m_message[0]= '\0';
 
357
  m_sql_errno= 0;
 
358
  m_server_status= 0;
 
359
  m_affected_rows= 0;
 
360
  m_last_insert_id= 0;
 
361
  m_total_warn_count= 0;
 
362
  is_sent= false;
 
363
  /** Tiny reset in debug mode to see garbage right away */
 
364
  m_status= DA_EMPTY;
 
365
}
 
366
 
 
367
 
 
368
/**
 
369
  Set OK status -- ends commands that do not return a
 
370
  result set, e.g. INSERT/UPDATE/DELETE.
 
371
*/
 
372
 
 
373
void
 
374
Diagnostics_area::set_ok_status(Session *session, ha_rows affected_rows_arg,
 
375
                                uint64_t last_insert_id_arg,
 
376
                                const char *message_arg)
 
377
{
 
378
  assert(! is_set());
 
379
  /*
 
380
    In production, refuse to overwrite an error or a custom response
 
381
    with an OK packet.
 
382
  */
 
383
  if (is_error() || is_disabled())
 
384
    return;
 
385
  /** Only allowed to report success if has not yet reported an error */
 
386
 
 
387
  m_server_status= session->server_status;
 
388
  m_total_warn_count= session->total_warn_count;
 
389
  m_affected_rows= affected_rows_arg;
 
390
  m_last_insert_id= last_insert_id_arg;
 
391
  if (message_arg)
 
392
    strncpy(m_message, message_arg, sizeof(m_message) - 1);
 
393
  else
 
394
    m_message[0]= '\0';
 
395
  m_status= DA_OK;
 
396
}
 
397
 
 
398
 
 
399
/**
 
400
  Set EOF status.
 
401
*/
 
402
 
 
403
void
 
404
Diagnostics_area::set_eof_status(Session *session)
 
405
{
 
406
  /** Only allowed to report eof if has not yet reported an error */
 
407
 
 
408
  assert(! is_set());
 
409
  /*
 
410
    In production, refuse to overwrite an error or a custom response
 
411
    with an EOF packet.
 
412
  */
 
413
  if (is_error() || is_disabled())
 
414
    return;
 
415
 
 
416
  m_server_status= session->server_status;
 
417
  /*
 
418
    If inside a stored procedure, do not return the total
 
419
    number of warnings, since they are not available to the client
 
420
    anyway.
 
421
  */
 
422
  m_total_warn_count= session->total_warn_count;
 
423
 
 
424
  m_status= DA_EOF;
 
425
}
 
426
 
 
427
/**
 
428
  Set ERROR status.
 
429
*/
 
430
 
 
431
void
 
432
Diagnostics_area::set_error_status(Session *,
 
433
                                   uint32_t sql_errno_arg,
 
434
                                   const char *message_arg)
 
435
{
 
436
  /*
 
437
    Only allowed to report error if has not yet reported a success
 
438
    The only exception is when we flush the message to the client,
 
439
    an error can happen during the flush.
 
440
  */
 
441
  assert(! is_set() || can_overwrite_status);
 
442
  /*
 
443
    In production, refuse to overwrite a custom response with an
 
444
    ERROR packet.
 
445
  */
 
446
  if (is_disabled())
 
447
    return;
 
448
 
 
449
  m_sql_errno= sql_errno_arg;
 
450
  strncpy(m_message, message_arg, sizeof(m_message) - 1);
 
451
 
 
452
  m_status= DA_ERROR;
 
453
}
 
454
 
 
455
 
 
456
/**
 
457
  Mark the diagnostics area as 'DISABLED'.
 
458
 
 
459
  This is used in rare cases when the COM_ command at hand sends a response
 
460
  in a custom format. One example is the query cache, another is
 
461
  COM_STMT_PREPARE.
 
462
*/
 
463
 
 
464
void
 
465
Diagnostics_area::disable_status()
 
466
{
 
467
  assert(! is_set());
 
468
  m_status= DA_DISABLED;
 
469
}
 
470
 
 
471
 
 
472
Session::Session()
 
473
   :Statement(&main_lex, &main_mem_root,
 
474
              /* statement id */ 0),
 
475
   Open_tables_state(refresh_version),
 
476
   lock_id(&main_lock_id),
 
477
   user_time(0),
 
478
   arg_of_last_insert_id_function(false),
 
479
   first_successful_insert_id_in_prev_stmt(0),
 
480
   first_successful_insert_id_in_cur_stmt(0),
 
481
   global_read_lock(0),
 
482
   is_fatal_error(0),
 
483
   transaction_rollback_request(0),
 
484
   is_fatal_sub_stmt_error(0),
 
485
   in_lock_tables(0),
 
486
   derived_tables_processing(false),
 
487
   m_lip(NULL),
 
488
   scheduler(0)
 
489
{
 
490
  ulong tmp;
209
491
 
210
492
  /*
211
493
    Pass nominal parameters to init_alloc_root only to ensure that
212
494
    the destructor works OK in case of an error. The main_mem_root
213
495
    will be re-initialized in init_for_queries().
214
496
  */
215
 
  memory::init_sql_alloc(&main_mem_root, memory::ROOT_MIN_BLOCK_SIZE, 0);
216
 
  thread_stack= NULL;
 
497
  init_sql_alloc(&main_mem_root, ALLOC_ROOT_MIN_BLOCK_SIZE, 0);
 
498
  thread_stack= 0;
 
499
  catalog= (char*)"std"; // the only catalog we have for now
 
500
  some_tables_deleted=no_errors=password= 0;
217
501
  count_cuted_fields= CHECK_FIELD_IGNORE;
218
502
  killed= NOT_KILLED;
219
 
  col_access= 0;
220
 
  tmp_table= 0;
221
 
  used_tables= 0;
 
503
  col_access=0;
 
504
  thread_specific_used= false;
 
505
  hash_clear(&handler_tables_hash);
 
506
  tmp_table=0;
 
507
  used_tables=0;
222
508
  cuted_fields= sent_row_count= row_count= 0L;
 
509
  limit_found_rows= 0;
223
510
  row_count_func= -1;
224
511
  statement_id_counter= 0UL;
225
512
  // Must be reset to handle error with Session's created for init of mysqld
231
518
  thread_id= 0;
232
519
  file_id = 0;
233
520
  query_id= 0;
234
 
  warn_query_id= 0;
235
 
  mysys_var= 0;
 
521
  warn_id= 0;
 
522
  db_charset= global_system_variables.collation_database;
 
523
  memset(ha_data, 0, sizeof(ha_data));
 
524
  replication_data= 0;
 
525
  mysys_var=0;
236
526
  dbug_sentry=Session_SENTRY_MAGIC;
237
 
  cleanup_done= abort_on_warning= no_warnings_for_error= false;
 
527
  net.vio= 0;
 
528
  client_capabilities= 0;                       // minimalistic client
 
529
  system_thread= NON_SYSTEM_THREAD;
 
530
  cleanup_done= abort_on_warning= no_warnings_for_error= 0;
 
531
  peer_port= 0;                                 // For SHOW PROCESSLIST
 
532
  transaction.on= 1;
238
533
  pthread_mutex_init(&LOCK_delete, MY_MUTEX_INIT_FAST);
239
534
 
240
535
  /* Variables with default values */
241
536
  proc_info="login";
242
537
  where= Session::DEFAULT_WHERE;
243
 
  command= COM_CONNECT;
244
 
 
245
 
  plugin_sessionvar_init(this);
246
 
  /*
247
 
    variables= global_system_variables above has reset
248
 
    variables.pseudo_thread_id to 0. We need to correct it here to
249
 
    avoid temporary tables replication failure.
250
 
  */
251
 
  variables.pseudo_thread_id= thread_id;
252
 
  server_status= SERVER_STATUS_AUTOCOMMIT;
253
 
  options= session_startup_options;
254
 
 
255
 
  if (variables.max_join_size == HA_POS_ERROR)
256
 
    options |= OPTION_BIG_SELECTS;
257
 
  else
258
 
    options &= ~OPTION_BIG_SELECTS;
259
 
 
260
 
  open_options=ha_open_options;
261
 
  update_lock_default= TL_WRITE;
262
 
  session_tx_isolation= (enum_tx_isolation) variables.tx_isolation;
263
 
  warn_list.empty();
264
 
  memset(warn_count, 0, sizeof(warn_count));
265
 
  total_warn_count= 0;
266
 
  memset(&status_var, 0, sizeof(status_var));
267
 
 
 
538
  server_id = ::server_id;
 
539
  command=COM_CONNECT;
 
540
  *scramble= '\0';
 
541
 
 
542
  init();
268
543
  /* Initialize sub structures */
269
 
  memory::init_sql_alloc(&warn_root, WARN_ALLOC_BLOCK_SIZE, WARN_ALLOC_PREALLOC_SIZE);
 
544
  init_sql_alloc(&warn_root, WARN_ALLOC_BLOCK_SIZE, WARN_ALLOC_PREALLOC_SIZE);
 
545
  user_connect=(USER_CONN *)0;
270
546
  hash_init(&user_vars, system_charset_info, USER_VARS_HASH_SIZE, 0, 0,
271
547
            (hash_get_key) get_var_key,
272
548
            (hash_free_key) free_user_var, 0);
273
549
 
 
550
  /* Protocol */
 
551
  protocol= &protocol_text;                     // Default protocol
 
552
  protocol_text.init(this);
 
553
 
 
554
  const Query_id& local_query_id= Query_id::get_query_id();
 
555
  tablespace_op= false;
 
556
  tmp= sql_rnd();
 
557
  drizzleclient_drizzleclient_randominit(&rand, tmp + (ulong) &rand, tmp + local_query_id.value());
274
558
  substitute_null_with_insert_id = false;
275
559
  thr_lock_info_init(&lock_info); /* safety: will be reset after start */
276
560
  thr_lock_owner_init(&main_lock_id, &lock_info);
278
562
  m_internal_handler= NULL;
279
563
}
280
564
 
281
 
void Session::free_items()
282
 
{
283
 
  Item *next;
284
 
  /* This works because items are allocated with memory::sql_alloc() */
285
 
  for (; free_list; free_list= next)
286
 
  {
287
 
    next= free_list->next;
288
 
    free_list->delete_self();
289
 
  }
290
 
}
291
565
 
292
566
void Session::push_internal_handler(Internal_error_handler *handler)
293
567
{
299
573
  m_internal_handler= handler;
300
574
}
301
575
 
 
576
 
302
577
bool Session::handle_error(uint32_t sql_errno, const char *message,
303
578
                       DRIZZLE_ERROR::enum_warning_level level)
304
579
{
310
585
  return false;                                 // 'false', as per coding style
311
586
}
312
587
 
 
588
 
313
589
void Session::pop_internal_handler()
314
590
{
315
591
  assert(m_internal_handler != NULL);
354
630
}
355
631
#endif
356
632
 
 
633
/*
 
634
  Init common variables that has to be reset on start and on change_user
 
635
*/
 
636
 
 
637
void Session::init(void)
 
638
{
 
639
  pthread_mutex_lock(&LOCK_global_system_variables);
 
640
  plugin_sessionvar_init(this);
 
641
  variables.time_format= date_time_format_copy((Session*) 0,
 
642
                                               variables.time_format);
 
643
  variables.date_format= date_time_format_copy((Session*) 0,
 
644
                                               variables.date_format);
 
645
  variables.datetime_format= date_time_format_copy((Session*) 0,
 
646
                                                   variables.datetime_format);
 
647
  /*
 
648
    variables= global_system_variables above has reset
 
649
    variables.pseudo_thread_id to 0. We need to correct it here to
 
650
    avoid temporary tables replication failure.
 
651
  */
 
652
  variables.pseudo_thread_id= thread_id;
 
653
  pthread_mutex_unlock(&LOCK_global_system_variables);
 
654
  server_status= SERVER_STATUS_AUTOCOMMIT;
 
655
  options= session_startup_options;
 
656
 
 
657
  if (variables.max_join_size == HA_POS_ERROR)
 
658
    options |= OPTION_BIG_SELECTS;
 
659
  else
 
660
    options &= ~OPTION_BIG_SELECTS;
 
661
 
 
662
  transaction.all.modified_non_trans_table= transaction.stmt.modified_non_trans_table= false;
 
663
  open_options=ha_open_options;
 
664
  update_lock_default= TL_WRITE;
 
665
  session_tx_isolation= (enum_tx_isolation) variables.tx_isolation;
 
666
  warn_list.empty();
 
667
  memset(warn_count, 0, sizeof(warn_count));
 
668
  total_warn_count= 0;
 
669
  update_charset();
 
670
  memset(&status_var, 0, sizeof(status_var));
 
671
}
 
672
 
 
673
 
 
674
/*
 
675
  Init Session for query processing.
 
676
  This has to be called once before we call mysql_parse.
 
677
  See also comments in session.h.
 
678
*/
 
679
 
 
680
void Session::init_for_queries()
 
681
{
 
682
  set_time();
 
683
  ha_enable_transaction(this,true);
 
684
 
 
685
  reset_root_defaults(mem_root, variables.query_alloc_block_size,
 
686
                      variables.query_prealloc_size);
 
687
  reset_root_defaults(&transaction.mem_root,
 
688
                      variables.trans_alloc_block_size,
 
689
                      variables.trans_prealloc_size);
 
690
  transaction.xid_state.xid.null();
 
691
  transaction.xid_state.in_session=1;
 
692
}
 
693
 
 
694
 
357
695
/* Do operations that may take a long time */
358
696
 
359
697
void Session::cleanup(void)
360
698
{
361
 
  assert(cleanup_done == false);
 
699
  assert(cleanup_done == 0);
362
700
 
363
701
  killed= KILL_CONNECTION;
364
702
#ifdef ENABLE_WHEN_BINLOG_WILL_BE_ABLE_TO_PREPARE
368
706
  }
369
707
#endif
370
708
  {
371
 
    TransactionServices &transaction_services= TransactionServices::singleton();
372
 
    transaction_services.ha_rollback_trans(this, true);
 
709
    ha_rollback(this);
373
710
    xid_cache_delete(&transaction.xid_state);
374
711
  }
 
712
  if (locked_tables)
 
713
  {
 
714
    lock=locked_tables; locked_tables=0;
 
715
    close_thread_tables(this);
 
716
  }
 
717
  mysql_ha_cleanup(this);
375
718
  hash_free(&user_vars);
376
719
  close_temporary_tables();
 
720
  free((char*) variables.time_format);
 
721
  free((char*) variables.date_format);
 
722
  free((char*) variables.datetime_format);
377
723
 
378
724
  if (global_read_lock)
379
725
    unlock_global_read_lock(this);
380
726
 
381
 
  cleanup_done= true;
 
727
  cleanup_done=1;
 
728
  return;
382
729
}
383
730
 
384
731
Session::~Session()
385
732
{
386
 
  this->checkSentry();
 
733
  Session_CHECK_SENTRY(this);
 
734
  /* Ensure that no one is using Session */
 
735
  pthread_mutex_lock(&LOCK_delete);
 
736
  pthread_mutex_unlock(&LOCK_delete);
387
737
  add_to_status(&global_status_var, &status_var);
388
738
 
389
 
  if (client->isConnected())
390
 
  {
391
 
    if (global_system_variables.log_warnings)
392
 
        errmsg_printf(ERRMSG_LVL_WARN, ER(ER_FORCING_CLOSE),internal::my_progname,
393
 
                      thread_id,
394
 
                      (getSecurityContext().getUser().c_str() ?
395
 
                       getSecurityContext().getUser().c_str() : ""));
396
 
    disconnect(0, false);
397
 
  }
398
 
 
399
739
  /* Close connection */
400
 
  client->close();
401
 
  delete client;
402
 
 
403
 
  if (cleanup_done == false)
 
740
  if (net.vio)
 
741
  {
 
742
    drizzleclient_net_close(&net);
 
743
    drizzleclient_net_end(&net);
 
744
  }
 
745
  if (!cleanup_done)
404
746
    cleanup();
405
747
 
406
 
  plugin::StorageEngine::closeConnection(this);
 
748
  ha_close_connection(this);
407
749
  plugin_sessionvar_cleanup(this);
408
750
 
 
751
  if (db)
 
752
  {
 
753
    free(db);
 
754
    db= NULL;
 
755
  }
409
756
  free_root(&warn_root,MYF(0));
 
757
  free_root(&transaction.mem_root,MYF(0));
410
758
  mysys_var=0;                                  // Safety (shouldn't be needed)
 
759
  pthread_mutex_destroy(&LOCK_delete);
411
760
  dbug_sentry= Session_SENTRY_GONE;
412
761
 
413
762
  free_root(&main_mem_root, MYF(0));
414
763
  pthread_setspecific(THR_Session,  0);
415
 
 
416
 
  plugin::Logging::postEndDo(this);
417
 
 
418
 
  /* Ensure that no one is using Session */
419
 
  pthread_mutex_unlock(&LOCK_delete);
420
 
  pthread_mutex_destroy(&LOCK_delete);
 
764
  return;
421
765
}
422
766
 
 
767
 
423
768
/*
424
769
  Add all status variables to another status variable array
425
770
 
433
778
    If this assumption will change, then we have to explictely add
434
779
    the other variables after the while loop
435
780
*/
436
 
void add_to_status(system_status_var *to_var, system_status_var *from_var)
 
781
 
 
782
void add_to_status(STATUS_VAR *to_var, STATUS_VAR *from_var)
437
783
{
438
784
  ulong *end= (ulong*) ((unsigned char*) to_var +
439
 
                        offsetof(system_status_var, last_system_status_var) +
 
785
                        offsetof(STATUS_VAR, last_system_status_var) +
440
786
                        sizeof(ulong));
441
787
  ulong *to= (ulong*) to_var, *from= (ulong*) from_var;
442
788
 
456
802
  NOTE
457
803
    This function assumes that all variables are long/ulong.
458
804
*/
459
 
void add_diff_to_status(system_status_var *to_var, system_status_var *from_var,
460
 
                        system_status_var *dec_var)
 
805
 
 
806
void add_diff_to_status(STATUS_VAR *to_var, STATUS_VAR *from_var,
 
807
                        STATUS_VAR *dec_var)
461
808
{
462
 
  ulong *end= (ulong*) ((unsigned char*) to_var + offsetof(system_status_var,
 
809
  ulong *end= (ulong*) ((unsigned char*) to_var + offsetof(STATUS_VAR,
463
810
                                                  last_system_status_var) +
464
811
                        sizeof(ulong));
465
812
  ulong *to= (ulong*) to_var, *from= (ulong*) from_var, *dec= (ulong*) dec_var;
468
815
    *(to++)+= *(from++) - *(dec++);
469
816
}
470
817
 
 
818
 
471
819
void Session::awake(Session::killed_state state_to_set)
472
820
{
473
 
  this->checkSentry();
 
821
  Session_CHECK_SENTRY(this);
474
822
  safe_mutex_assert_owner(&LOCK_delete);
475
823
 
476
824
  killed= state_to_set;
477
825
  if (state_to_set != Session::KILL_QUERY)
478
826
  {
479
 
    scheduler->killSession(this);
480
 
    DRIZZLE_CONNECTION_DONE(thread_id);
 
827
    thread_scheduler.post_kill_notification(this);
481
828
  }
482
829
  if (mysys_var)
483
830
  {
484
831
    pthread_mutex_lock(&mysys_var->mutex);
 
832
    if (!system_thread)         // Don't abort locks
 
833
      mysys_var->abort=1;
485
834
    /*
486
835
      This broadcast could be up in the air if the victim thread
487
836
      exits the cond in the time between read and broadcast, but that is
509
858
    }
510
859
    pthread_mutex_unlock(&mysys_var->mutex);
511
860
  }
 
861
  return;
512
862
}
513
863
 
514
864
/*
515
865
  Remember the location of thread info, the structure needed for
516
 
  memory::sql_alloc() and the structure for the net buffer
 
866
  sql_alloc() and the structure for the net buffer
517
867
*/
518
 
bool Session::storeGlobals()
 
868
 
 
869
bool Session::store_globals()
519
870
{
520
871
  /*
521
872
    Assert that thread_stack is initialized: it's necessary to be able
525
876
 
526
877
  if (pthread_setspecific(THR_Session,  this) ||
527
878
      pthread_setspecific(THR_Mem_root, &mem_root))
528
 
    return true;
529
 
 
 
879
    return 1;
530
880
  mysys_var=my_thread_var;
531
 
 
532
881
  /*
533
882
    Let mysqld define the thread id (not mysys)
534
883
    This allows us to move Session to different threads if needed.
541
890
    created in another thread
542
891
  */
543
892
  thr_lock_info_init(&lock_info);
544
 
  return false;
 
893
  return 0;
545
894
}
546
895
 
 
896
 
547
897
/*
548
 
  Init Session for query processing.
549
 
  This has to be called once before we call mysql_parse.
550
 
  See also comments in session.h.
 
898
  Cleanup after query.
 
899
 
 
900
  SYNOPSIS
 
901
    Session::cleanup_after_query()
 
902
 
 
903
  DESCRIPTION
 
904
    This function is used to reset thread data to its default state.
 
905
 
 
906
  NOTE
 
907
    This function is not suitable for setting thread data to some
 
908
    non-default values, as there is only one replication thread, so
 
909
    different master threads may overwrite data of each other on
 
910
    slave.
551
911
*/
552
912
 
553
 
void Session::prepareForQueries()
554
 
{
555
 
  if (variables.max_join_size == HA_POS_ERROR)
556
 
    options |= OPTION_BIG_SELECTS;
557
 
 
558
 
  version= refresh_version;
559
 
  set_proc_info(NULL);
560
 
  command= COM_SLEEP;
561
 
  set_time();
562
 
 
563
 
  reset_root_defaults(mem_root, variables.query_alloc_block_size,
564
 
                      variables.query_prealloc_size);
565
 
  transaction.xid_state.xid.null();
566
 
  transaction.xid_state.in_session=1;
567
 
}
568
 
 
569
 
bool Session::initGlobals()
570
 
{
571
 
  if (storeGlobals())
572
 
  {
573
 
    disconnect(ER_OUT_OF_RESOURCES, true);
574
 
    statistic_increment(aborted_connects, &LOCK_status);
575
 
    return true;
576
 
  }
577
 
  return false;
578
 
}
579
 
 
580
 
void Session::run()
581
 
{
582
 
  if (initGlobals() || authenticate())
583
 
  {
584
 
    disconnect(0, true);
585
 
    return;
586
 
  }
587
 
 
588
 
  prepareForQueries();
589
 
 
590
 
  while (! client->haveError() && killed != KILL_CONNECTION)
591
 
  {
592
 
    if (! executeStatement())
593
 
      break;
594
 
  }
595
 
 
596
 
  disconnect(0, true);
597
 
}
598
 
 
599
 
bool Session::schedule()
600
 
{
601
 
  scheduler= plugin::Scheduler::getScheduler();
602
 
  assert(scheduler);
603
 
 
604
 
  connection_count.increment();
605
 
 
606
 
  if (connection_count > max_used_connections)
607
 
    max_used_connections= connection_count;
608
 
 
609
 
  thread_id= variables.pseudo_thread_id= global_thread_id++;
610
 
 
611
 
  pthread_mutex_lock(&LOCK_thread_count);
612
 
  getSessionList().push_back(this);
613
 
  pthread_mutex_unlock(&LOCK_thread_count);
614
 
 
615
 
  if (scheduler->addSession(this))
616
 
  {
617
 
    DRIZZLE_CONNECTION_START(thread_id);
618
 
    char error_message_buff[DRIZZLE_ERRMSG_SIZE];
619
 
 
620
 
    killed= Session::KILL_CONNECTION;
621
 
 
622
 
    statistic_increment(aborted_connects, &LOCK_status);
623
 
 
624
 
    /* Can't use my_error() since store_globals has not been called. */
625
 
    /* TODO replace will better error message */
626
 
    snprintf(error_message_buff, sizeof(error_message_buff),
627
 
             ER(ER_CANT_CREATE_THREAD), 1);
628
 
    client->sendError(ER_CANT_CREATE_THREAD, error_message_buff);
629
 
    return true;
630
 
  }
631
 
 
632
 
  return false;
633
 
}
634
 
 
635
 
 
636
 
const char* Session::enter_cond(pthread_cond_t *cond,
637
 
                                pthread_mutex_t* mutex,
638
 
                                const char* msg)
639
 
{
640
 
  const char* old_msg = get_proc_info();
641
 
  safe_mutex_assert_owner(mutex);
642
 
  mysys_var->current_mutex = mutex;
643
 
  mysys_var->current_cond = cond;
644
 
  this->set_proc_info(msg);
645
 
  return old_msg;
646
 
}
647
 
 
648
 
void Session::exit_cond(const char* old_msg)
649
 
{
650
 
  /*
651
 
    Putting the mutex unlock in exit_cond() ensures that
652
 
    mysys_var->current_mutex is always unlocked _before_ mysys_var->mutex is
653
 
    locked (if that would not be the case, you'll get a deadlock if someone
654
 
    does a Session::awake() on you).
655
 
  */
656
 
  pthread_mutex_unlock(mysys_var->current_mutex);
657
 
  pthread_mutex_lock(&mysys_var->mutex);
658
 
  mysys_var->current_mutex = 0;
659
 
  mysys_var->current_cond = 0;
660
 
  this->set_proc_info(old_msg);
661
 
  pthread_mutex_unlock(&mysys_var->mutex);
662
 
}
663
 
 
664
 
bool Session::authenticate()
665
 
{
666
 
  lex_start(this);
667
 
  if (client->authenticate())
668
 
    return false;
669
 
 
670
 
  statistic_increment(aborted_connects, &LOCK_status);
671
 
  return true;
672
 
}
673
 
 
674
 
bool Session::checkUser(const char *passwd, uint32_t passwd_len, const char *in_db)
675
 
{
676
 
  const string passwd_str(passwd, passwd_len);
677
 
  bool is_authenticated=
678
 
    plugin::Authentication::isAuthenticated(getSecurityContext(),
679
 
                                            passwd_str);
680
 
 
681
 
  if (is_authenticated != true)
682
 
  {
683
 
    /* isAuthenticated has pushed the error message */
684
 
    return false;
685
 
  }
686
 
 
687
 
  /* Change database if necessary */
688
 
  if (in_db && in_db[0])
689
 
  {
690
 
    SchemaIdentifier identifier(in_db);
691
 
    if (mysql_change_db(this, identifier))
692
 
    {
693
 
      /* mysql_change_db() has pushed the error message. */
694
 
      return false;
695
 
    }
696
 
  }
697
 
  my_ok();
698
 
  password= test(passwd_len);          // remember for error messages
699
 
 
700
 
  /* Ready to handle queries */
701
 
  return true;
702
 
}
703
 
 
704
 
bool Session::executeStatement()
705
 
{
706
 
  char *l_packet= 0;
707
 
  uint32_t packet_length;
708
 
 
709
 
  enum enum_server_command l_command;
710
 
 
711
 
  /*
712
 
    indicator of uninitialized lex => normal flow of errors handling
713
 
    (see my_message_sql)
714
 
  */
715
 
  lex->current_select= 0;
716
 
  clear_error();
717
 
  main_da.reset_diagnostics_area();
718
 
 
719
 
  if (client->readCommand(&l_packet, &packet_length) == false)
720
 
    return false;
721
 
 
722
 
  if (packet_length == 0)
723
 
    return true;
724
 
 
725
 
  l_command= (enum enum_server_command) (unsigned char) l_packet[0];
726
 
 
727
 
  if (command >= COM_END)
728
 
    command= COM_END;                           // Wrong command
729
 
 
730
 
  assert(packet_length);
731
 
  return ! dispatch_command(l_command, this, l_packet+1, (uint32_t) (packet_length-1));
732
 
}
733
 
 
734
 
bool Session::readAndStoreQuery(const char *in_packet, uint32_t in_packet_length)
735
 
{
736
 
  /* Remove garbage at start and end of query */
737
 
  while (in_packet_length > 0 && my_isspace(charset(), in_packet[0]))
738
 
  {
739
 
    in_packet++;
740
 
    in_packet_length--;
741
 
  }
742
 
  const char *pos= in_packet + in_packet_length; /* Point at end null */
743
 
  while (in_packet_length > 0 &&
744
 
         (pos[-1] == ';' || my_isspace(charset() ,pos[-1])))
745
 
  {
746
 
    pos--;
747
 
    in_packet_length--;
748
 
  }
749
 
 
750
 
  query.assign(in_packet, in_packet + in_packet_length);
751
 
 
752
 
  return true;
753
 
}
754
 
 
755
 
bool Session::endTransaction(enum enum_mysql_completiontype completion)
756
 
{
757
 
  bool do_release= 0;
758
 
  bool result= true;
759
 
  TransactionServices &transaction_services= TransactionServices::singleton();
760
 
 
761
 
  if (transaction.xid_state.xa_state != XA_NOTR)
762
 
  {
763
 
    my_error(ER_XAER_RMFAIL, MYF(0), xa_state_names[transaction.xid_state.xa_state]);
764
 
    return false;
765
 
  }
766
 
  switch (completion)
767
 
  {
768
 
    case COMMIT:
769
 
      /*
770
 
       * We don't use endActiveTransaction() here to ensure that this works
771
 
       * even if there is a problem with the OPTION_AUTO_COMMIT flag
772
 
       * (Which of course should never happen...)
773
 
       */
774
 
      server_status&= ~SERVER_STATUS_IN_TRANS;
775
 
      if (transaction_services.ha_commit_trans(this, true))
776
 
        result= false;
777
 
      options&= ~(OPTION_BEGIN);
778
 
      break;
779
 
    case COMMIT_RELEASE:
780
 
      do_release= 1; /* fall through */
781
 
    case COMMIT_AND_CHAIN:
782
 
      result= endActiveTransaction();
783
 
      if (result == true && completion == COMMIT_AND_CHAIN)
784
 
        result= startTransaction();
785
 
      break;
786
 
    case ROLLBACK_RELEASE:
787
 
      do_release= 1; /* fall through */
788
 
    case ROLLBACK:
789
 
    case ROLLBACK_AND_CHAIN:
790
 
    {
791
 
      server_status&= ~SERVER_STATUS_IN_TRANS;
792
 
      if (transaction_services.ha_rollback_trans(this, true))
793
 
        result= false;
794
 
      options&= ~(OPTION_BEGIN);
795
 
      if (result == true && (completion == ROLLBACK_AND_CHAIN))
796
 
        result= startTransaction();
797
 
      break;
798
 
    }
799
 
    default:
800
 
      my_error(ER_UNKNOWN_COM_ERROR, MYF(0));
801
 
      return false;
802
 
  }
803
 
 
804
 
  if (result == false)
805
 
    my_error(killed_errno(), MYF(0));
806
 
  else if ((result == true) && do_release)
807
 
    killed= Session::KILL_CONNECTION;
808
 
 
809
 
  return result;
810
 
}
811
 
 
812
 
bool Session::endActiveTransaction()
813
 
{
814
 
  bool result= true;
815
 
  TransactionServices &transaction_services= TransactionServices::singleton();
816
 
 
817
 
  if (transaction.xid_state.xa_state != XA_NOTR)
818
 
  {
819
 
    my_error(ER_XAER_RMFAIL, MYF(0), xa_state_names[transaction.xid_state.xa_state]);
820
 
    return false;
821
 
  }
822
 
  if (options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN))
823
 
  {
824
 
    server_status&= ~SERVER_STATUS_IN_TRANS;
825
 
    if (transaction_services.ha_commit_trans(this, true))
826
 
      result= false;
827
 
  }
828
 
  options&= ~(OPTION_BEGIN);
829
 
  return result;
830
 
}
831
 
 
832
 
bool Session::startTransaction(start_transaction_option_t opt)
833
 
{
834
 
  bool result= true;
835
 
 
836
 
  if (! endActiveTransaction())
837
 
  {
838
 
    result= false;
839
 
  }
840
 
  else
841
 
  {
842
 
    options|= OPTION_BEGIN;
843
 
    server_status|= SERVER_STATUS_IN_TRANS;
844
 
 
845
 
    if (plugin::TransactionalStorageEngine::notifyStartTransaction(this, opt))
846
 
    {
847
 
      result= false;
848
 
    }
849
 
  }
850
 
 
851
 
  return result;
852
 
}
853
 
 
854
913
void Session::cleanup_after_query()
855
914
{
856
915
  /*
864
923
  if (first_successful_insert_id_in_cur_stmt > 0)
865
924
  {
866
925
    /* set what LAST_INSERT_ID() will return */
867
 
    first_successful_insert_id_in_prev_stmt= first_successful_insert_id_in_cur_stmt;
 
926
    first_successful_insert_id_in_prev_stmt=
 
927
      first_successful_insert_id_in_cur_stmt;
868
928
    first_successful_insert_id_in_cur_stmt= 0;
869
929
    substitute_null_with_insert_id= true;
870
930
  }
871
 
  arg_of_last_insert_id_function= false;
 
931
  arg_of_last_insert_id_function= 0;
872
932
  /* Free Items that were created during this execution */
873
933
  free_items();
874
934
  /* Reset where. */
875
935
  where= Session::DEFAULT_WHERE;
876
936
}
877
937
 
 
938
 
878
939
/**
879
940
  Create a LEX_STRING in this connection.
880
941
 
886
947
  @return  NULL on failure, or pointer to the LEX_STRING object
887
948
*/
888
949
LEX_STRING *Session::make_lex_string(LEX_STRING *lex_str,
889
 
                                     const std::string &str,
890
 
                                     bool allocate_lex_string)
891
 
{
892
 
  return make_lex_string(lex_str, str.c_str(), str.length(), allocate_lex_string);
893
 
}
894
 
 
895
 
LEX_STRING *Session::make_lex_string(LEX_STRING *lex_str,
896
 
                                     const char* str, uint32_t length,
897
 
                                     bool allocate_lex_string)
 
950
                                 const char* str, uint32_t length,
 
951
                                 bool allocate_lex_string)
898
952
{
899
953
  if (allocate_lex_string)
900
954
    if (!(lex_str= (LEX_STRING *)alloc(sizeof(LEX_STRING))))
905
959
  return lex_str;
906
960
}
907
961
 
 
962
 
 
963
/*
 
964
  Convert a string to another character set
 
965
 
 
966
  SYNOPSIS
 
967
    convert_string()
 
968
    to                          Store new allocated string here
 
969
    to_cs                       New character set for allocated string
 
970
    from                        String to convert
 
971
    from_length                 Length of string to convert
 
972
    from_cs                     Original character set
 
973
 
 
974
  NOTES
 
975
    to will be 0-terminated to make it easy to pass to system funcs
 
976
 
 
977
  RETURN
 
978
    0   ok
 
979
    1   End of memory.
 
980
        In this case to->str will point to 0 and to->length will be 0.
 
981
*/
 
982
 
 
983
bool Session::convert_string(LEX_STRING *to, const CHARSET_INFO * const to_cs,
 
984
                         const char *from, uint32_t from_length,
 
985
                         const CHARSET_INFO * const from_cs)
 
986
{
 
987
  size_t new_length= to_cs->mbmaxlen * from_length;
 
988
  uint32_t dummy_errors;
 
989
  if (!(to->str= (char*) alloc(new_length+1)))
 
990
  {
 
991
    to->length= 0;                              // Safety fix
 
992
    return(1);                          // EOM
 
993
  }
 
994
  to->length= copy_and_convert((char*) to->str, new_length, to_cs,
 
995
                               from, from_length, from_cs, &dummy_errors);
 
996
  to->str[to->length]=0;                        // Safety
 
997
  return(0);
 
998
}
 
999
 
 
1000
 
 
1001
/*
 
1002
  Convert string from source character set to target character set inplace.
 
1003
 
 
1004
  SYNOPSIS
 
1005
    Session::convert_string
 
1006
 
 
1007
  DESCRIPTION
 
1008
    Convert string using convert_buffer - buffer for character set
 
1009
    conversion shared between all protocols.
 
1010
 
 
1011
  RETURN
 
1012
    0   ok
 
1013
   !0   out of memory
 
1014
*/
 
1015
 
 
1016
bool Session::convert_string(String *s, const CHARSET_INFO * const from_cs,
 
1017
                         const CHARSET_INFO * const to_cs)
 
1018
{
 
1019
  uint32_t dummy_errors;
 
1020
  if (convert_buffer.copy(s->ptr(), s->length(), from_cs, to_cs, &dummy_errors))
 
1021
    return true;
 
1022
  /* If convert_buffer >> s copying is more efficient long term */
 
1023
  if (convert_buffer.alloced_length() >= convert_buffer.length() * 2 ||
 
1024
      !s->is_alloced())
 
1025
  {
 
1026
    return s->copy(convert_buffer);
 
1027
  }
 
1028
  s->swap(convert_buffer);
 
1029
  return false;
 
1030
}
 
1031
 
 
1032
 
 
1033
/*
 
1034
  Update some cache variables when character set changes
 
1035
*/
 
1036
 
 
1037
void Session::update_charset()
 
1038
{
 
1039
  uint32_t not_used;
 
1040
  charset_is_system_charset= !String::needs_conversion(0,charset(),
 
1041
                                                       system_charset_info,
 
1042
                                                       &not_used);
 
1043
  charset_is_collation_connection=
 
1044
    !String::needs_conversion(0,charset(),variables.getCollation(),
 
1045
                              &not_used);
 
1046
  charset_is_character_set_filesystem=
 
1047
    !String::needs_conversion(0, charset(),
 
1048
                              variables.character_set_filesystem, &not_used);
 
1049
}
 
1050
 
 
1051
 
 
1052
/* routings to adding tables to list of changed in transaction tables */
 
1053
 
 
1054
inline static void list_include(CHANGED_TableList** prev,
 
1055
                                CHANGED_TableList* curr,
 
1056
                                CHANGED_TableList* new_table)
 
1057
{
 
1058
  if (new_table)
 
1059
  {
 
1060
    *prev = new_table;
 
1061
    (*prev)->next = curr;
 
1062
  }
 
1063
}
 
1064
 
 
1065
/* add table to list of changed in transaction tables */
 
1066
 
 
1067
void Session::add_changed_table(Table *table)
 
1068
{
 
1069
  assert((options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN)) &&
 
1070
              table->file->has_transactions());
 
1071
  add_changed_table(table->s->table_cache_key.str,
 
1072
                    (long) table->s->table_cache_key.length);
 
1073
  return;
 
1074
}
 
1075
 
 
1076
 
 
1077
void Session::add_changed_table(const char *key, long key_length)
 
1078
{
 
1079
  CHANGED_TableList **prev_changed = &transaction.changed_tables;
 
1080
  CHANGED_TableList *curr = transaction.changed_tables;
 
1081
 
 
1082
  for (; curr; prev_changed = &(curr->next), curr = curr->next)
 
1083
  {
 
1084
    int cmp =  (long)curr->key_length - (long)key_length;
 
1085
    if (cmp < 0)
 
1086
    {
 
1087
      list_include(prev_changed, curr, changed_table_dup(key, key_length));
 
1088
      return;
 
1089
    }
 
1090
    else if (cmp == 0)
 
1091
    {
 
1092
      cmp = memcmp(curr->key, key, curr->key_length);
 
1093
      if (cmp < 0)
 
1094
      {
 
1095
        list_include(prev_changed, curr, changed_table_dup(key, key_length));
 
1096
        return;
 
1097
      }
 
1098
      else if (cmp == 0)
 
1099
      {
 
1100
        return;
 
1101
      }
 
1102
    }
 
1103
  }
 
1104
  *prev_changed = changed_table_dup(key, key_length);
 
1105
  return;
 
1106
}
 
1107
 
 
1108
 
 
1109
CHANGED_TableList* Session::changed_table_dup(const char *key, long key_length)
 
1110
{
 
1111
  CHANGED_TableList* new_table =
 
1112
    (CHANGED_TableList*) trans_alloc(ALIGN_SIZE(sizeof(CHANGED_TableList))+
 
1113
                                      key_length + 1);
 
1114
  if (!new_table)
 
1115
  {
 
1116
    my_error(EE_OUTOFMEMORY, MYF(ME_BELL),
 
1117
             ALIGN_SIZE(sizeof(TableList)) + key_length + 1);
 
1118
    killed= KILL_CONNECTION;
 
1119
    return 0;
 
1120
  }
 
1121
 
 
1122
  new_table->key= ((char*)new_table)+ ALIGN_SIZE(sizeof(CHANGED_TableList));
 
1123
  new_table->next = 0;
 
1124
  new_table->key_length = key_length;
 
1125
  ::memcpy(new_table->key, key, key_length);
 
1126
  return new_table;
 
1127
}
 
1128
 
 
1129
 
908
1130
int Session::send_explain_fields(select_result *result)
909
1131
{
910
1132
  List<Item> field_list;
940
1162
  }
941
1163
  item->maybe_null= 1;
942
1164
  field_list.push_back(new Item_empty_string("Extra", 255, cs));
943
 
  return (result->send_fields(field_list));
944
 
}
945
 
 
946
 
void select_result::send_error(uint32_t errcode, const char *err)
947
 
{
948
 
  my_message(errcode, err, MYF(0));
949
 
}
 
1165
  return (result->send_fields(field_list,
 
1166
                              Protocol::SEND_NUM_ROWS | Protocol::SEND_EOF));
 
1167
}
 
1168
 
 
1169
 
 
1170
struct Item_change_record: public ilink
 
1171
{
 
1172
  Item **place;
 
1173
  Item *old_value;
 
1174
  /* Placement new was hidden by `new' in ilink (TODO: check): */
 
1175
  static void *operator new(size_t ,
 
1176
                            void *mem)
 
1177
    { return mem; }
 
1178
  static void operator delete(void *,
 
1179
                              size_t )
 
1180
    {}
 
1181
  static void operator delete(void *,
 
1182
                              void *)
 
1183
    { /* never called */ }
 
1184
};
 
1185
 
 
1186
 
 
1187
/*
 
1188
  Register an item tree tree transformation, performed by the query
 
1189
  optimizer. We need a pointer to runtime_memroot because it may be !=
 
1190
  session->mem_root (this may no longer be a true statement)
 
1191
*/
 
1192
 
 
1193
void Session::nocheck_register_item_tree_change(Item **place, Item *old_value,
 
1194
                                            MEM_ROOT *runtime_memroot)
 
1195
{
 
1196
  Item_change_record *change;
 
1197
  /*
 
1198
    Now we use one node per change, which adds some memory overhead,
 
1199
    but still is rather fast as we use alloc_root for allocations.
 
1200
    A list of item tree changes of an average query should be short.
 
1201
  */
 
1202
  void *change_mem= alloc_root(runtime_memroot, sizeof(*change));
 
1203
  if (change_mem == 0)
 
1204
  {
 
1205
    /*
 
1206
      OOM, session->fatal_error() is called by the error handler of the
 
1207
      memroot. Just return.
 
1208
    */
 
1209
    return;
 
1210
  }
 
1211
  change= new (change_mem) Item_change_record;
 
1212
  change->place= place;
 
1213
  change->old_value= old_value;
 
1214
  change_list.append(change);
 
1215
}
 
1216
 
 
1217
 
 
1218
void Session::rollback_item_tree_changes()
 
1219
{
 
1220
  I_List_iterator<Item_change_record> it(change_list);
 
1221
  Item_change_record *change;
 
1222
 
 
1223
  while ((change= it++))
 
1224
    *change->place= change->old_value;
 
1225
  /* We can forget about changes memory: it's allocated in runtime memroot */
 
1226
  change_list.empty();
 
1227
  return;
 
1228
}
 
1229
 
950
1230
 
951
1231
/************************************************************************
952
1232
  Handling writing to file
957
1237
  my_message(errcode, err, MYF(0));
958
1238
  if (file > 0)
959
1239
  {
960
 
    (void) end_io_cache(cache);
961
 
    (void) internal::my_close(file, MYF(0));
962
 
    (void) internal::my_delete(path, MYF(0));           // Delete file on error
 
1240
    (void) end_io_cache(&cache);
 
1241
    (void) my_close(file,MYF(0));
 
1242
    (void) my_delete(path,MYF(0));              // Delete file on error
963
1243
    file= -1;
964
1244
  }
965
1245
}
967
1247
 
968
1248
bool select_to_file::send_eof()
969
1249
{
970
 
  int error= test(end_io_cache(cache));
971
 
  if (internal::my_close(file, MYF(MY_WME)))
 
1250
  int error= test(end_io_cache(&cache));
 
1251
  if (my_close(file,MYF(MY_WME)))
972
1252
    error= 1;
973
1253
  if (!error)
974
1254
  {
989
1269
  /* In case of error send_eof() may be not called: close the file here. */
990
1270
  if (file >= 0)
991
1271
  {
992
 
    (void) end_io_cache(cache);
993
 
    (void) internal::my_close(file, MYF(0));
 
1272
    (void) end_io_cache(&cache);
 
1273
    (void) my_close(file,MYF(0));
994
1274
    file= -1;
995
1275
  }
996
1276
  path[0]= '\0';
997
1277
  row_count= 0;
998
1278
}
999
1279
 
1000
 
select_to_file::select_to_file(file_exchange *ex)
1001
 
  : exchange(ex),
1002
 
    file(-1),
1003
 
    cache(static_cast<internal::IO_CACHE *>(memory::sql_calloc(sizeof(internal::IO_CACHE)))),
1004
 
    row_count(0L)
1005
 
{
1006
 
  path[0]=0;
1007
 
}
1008
1280
 
1009
1281
select_to_file::~select_to_file()
1010
1282
{
1011
 
  cleanup();
 
1283
  if (file >= 0)
 
1284
  {                                     // This only happens in case of error
 
1285
    (void) end_io_cache(&cache);
 
1286
    (void) my_close(file,MYF(0));
 
1287
    file= -1;
 
1288
  }
1012
1289
}
1013
1290
 
1014
1291
/***************************************************************************
1037
1314
*/
1038
1315
 
1039
1316
 
1040
 
static int create_file(Session *session, char *path, file_exchange *exchange, internal::IO_CACHE *cache)
 
1317
static File create_file(Session *session, char *path, file_exchange *exchange, IO_CACHE *cache)
1041
1318
{
1042
 
  int file;
 
1319
  File file;
1043
1320
  uint32_t option= MY_UNPACK_FILENAME | MY_RELATIVE_PATH;
1044
1321
 
1045
1322
#ifdef DONT_ALLOW_FULL_LOAD_DATA_PATHS
1046
1323
  option|= MY_REPLACE_DIR;                      // Force use of db directory
1047
1324
#endif
1048
1325
 
1049
 
  if (!internal::dirname_length(exchange->file_name))
 
1326
  if (!dirname_length(exchange->file_name))
1050
1327
  {
1051
1328
    strcpy(path, drizzle_real_data_home);
1052
 
    if (! session->db.empty())
1053
 
      strncat(path, session->db.c_str(), FN_REFLEN-strlen(drizzle_real_data_home)-1);
1054
 
    (void) internal::fn_format(path, exchange->file_name, path, "", option);
 
1329
    if (session->db)
 
1330
      strncat(path, session->db, FN_REFLEN-strlen(drizzle_real_data_home)-1);
 
1331
    (void) fn_format(path, exchange->file_name, path, "", option);
1055
1332
  }
1056
1333
  else
1057
 
    (void) internal::fn_format(path, exchange->file_name, drizzle_real_data_home, "", option);
 
1334
    (void) fn_format(path, exchange->file_name, drizzle_real_data_home, "", option);
1058
1335
 
1059
1336
  if (opt_secure_file_priv &&
1060
1337
      strncmp(opt_secure_file_priv, path, strlen(opt_secure_file_priv)))
1070
1347
    return -1;
1071
1348
  }
1072
1349
  /* Create the file world readable */
1073
 
  if ((file= internal::my_create(path, 0666, O_WRONLY|O_EXCL, MYF(MY_WME))) < 0)
 
1350
  if ((file= my_create(path, 0666, O_WRONLY|O_EXCL, MYF(MY_WME))) < 0)
1074
1351
    return file;
 
1352
#ifdef HAVE_FCHMOD
1075
1353
  (void) fchmod(file, 0666);                    // Because of umask()
1076
 
  if (init_io_cache(cache, file, 0L, internal::WRITE_CACHE, 0L, 1, MYF(MY_WME)))
 
1354
#else
 
1355
  (void) chmod(path, 0666);
 
1356
#endif
 
1357
  if (init_io_cache(cache, file, 0L, WRITE_CACHE, 0L, 1, MYF(MY_WME)))
1077
1358
  {
1078
 
    internal::my_close(file, MYF(0));
1079
 
    internal::my_delete(path, MYF(0));  // Delete file on error, it was just created
 
1359
    my_close(file, MYF(0));
 
1360
    my_delete(path, MYF(0));  // Delete file on error, it was just created
1080
1361
    return -1;
1081
1362
  }
1082
1363
  return file;
1137
1418
    return 1;
1138
1419
  }
1139
1420
 
1140
 
  if ((file= create_file(session, path, exchange, cache)) < 0)
 
1421
  if ((file= create_file(session, path, exchange, &cache)) < 0)
1141
1422
    return 1;
1142
1423
 
1143
1424
  return 0;
1144
1425
}
1145
1426
 
 
1427
 
 
1428
#define NEED_ESCAPING(x) ((int) (unsigned char) (x) == escape_char    || \
 
1429
                          (enclosed ? (int) (unsigned char) (x) == field_sep_char      \
 
1430
                                    : (int) (unsigned char) (x) == field_term_char) || \
 
1431
                          (int) (unsigned char) (x) == line_sep_char  || \
 
1432
                          !(x))
 
1433
 
1146
1434
bool select_export::send_data(List<Item> &items)
1147
1435
{
1148
1436
  char buff[MAX_FIELD_WIDTH],null_buff[2],space[MAX_FIELD_WIDTH];
1160
1448
  uint32_t used_length=0,items_left=items.elements;
1161
1449
  List_iterator_fast<Item> li(items);
1162
1450
 
1163
 
  if (my_b_write(cache,(unsigned char*) exchange->line_start->ptr(),
1164
 
                 exchange->line_start->length()))
 
1451
  if (my_b_write(&cache,(unsigned char*) exchange->line_start->ptr(),
 
1452
                 exchange->line_start->length()))
1165
1453
    goto err;
1166
1454
  while ((item=li++))
1167
1455
  {
1171
1459
    res=item->str_result(&tmp);
1172
1460
    if (res && enclosed)
1173
1461
    {
1174
 
      if (my_b_write(cache,(unsigned char*) exchange->enclosed->ptr(),
1175
 
                     exchange->enclosed->length()))
1176
 
        goto err;
 
1462
      if (my_b_write(&cache,(unsigned char*) exchange->enclosed->ptr(),
 
1463
                     exchange->enclosed->length()))
 
1464
        goto err;
1177
1465
    }
1178
1466
    if (!res)
1179
1467
    {                                           // NULL
1180
1468
      if (!fixed_row_size)
1181
1469
      {
1182
 
        if (escape_char != -1)                  // Use \N syntax
1183
 
        {
1184
 
          null_buff[0]=escape_char;
1185
 
          null_buff[1]='N';
1186
 
          if (my_b_write(cache,(unsigned char*) null_buff,2))
1187
 
            goto err;
1188
 
        }
1189
 
        else if (my_b_write(cache,(unsigned char*) "NULL",4))
1190
 
          goto err;
 
1470
        if (escape_char != -1)                  // Use \N syntax
 
1471
        {
 
1472
          null_buff[0]=escape_char;
 
1473
          null_buff[1]='N';
 
1474
          if (my_b_write(&cache,(unsigned char*) null_buff,2))
 
1475
            goto err;
 
1476
        }
 
1477
        else if (my_b_write(&cache,(unsigned char*) "NULL",4))
 
1478
          goto err;
1191
1479
      }
1192
1480
      else
1193
1481
      {
1194
 
        used_length=0;                          // Fill with space
 
1482
        used_length=0;                          // Fill with space
1195
1483
      }
1196
1484
    }
1197
1485
    else
1198
1486
    {
1199
1487
      if (fixed_row_size)
1200
 
        used_length= min(res->length(),item->max_length);
 
1488
        used_length=cmin(res->length(),item->max_length);
1201
1489
      else
1202
 
        used_length= res->length();
1203
 
 
 
1490
        used_length=res->length();
1204
1491
      if ((result_type == STRING_RESULT || is_unsafe_field_sep) &&
1205
 
          escape_char != -1)
 
1492
           escape_char != -1)
1206
1493
      {
1207
1494
        char *pos, *start, *end;
1208
1495
        const CHARSET_INFO * const res_charset= res->charset();
1209
1496
        const CHARSET_INFO * const character_set_client= default_charset_info;
1210
1497
 
1211
1498
        bool check_second_byte= (res_charset == &my_charset_bin) &&
1212
 
          character_set_client->
1213
 
          escape_with_backslash_is_dangerous;
 
1499
                                 character_set_client->
 
1500
                                 escape_with_backslash_is_dangerous;
1214
1501
        assert(character_set_client->mbmaxlen == 2 ||
1215
1502
               !character_set_client->escape_with_backslash_is_dangerous);
1216
 
        for (start=pos=(char*) res->ptr(),end=pos+used_length ;
1217
 
             pos != end ;
1218
 
             pos++)
1219
 
        {
1220
 
          if (use_mb(res_charset))
1221
 
          {
1222
 
            int l;
1223
 
            if ((l=my_ismbchar(res_charset, pos, end)))
1224
 
            {
1225
 
              pos += l-1;
1226
 
              continue;
1227
 
            }
1228
 
          }
 
1503
        for (start=pos=(char*) res->ptr(),end=pos+used_length ;
 
1504
             pos != end ;
 
1505
             pos++)
 
1506
        {
 
1507
#ifdef USE_MB
 
1508
          if (use_mb(res_charset))
 
1509
          {
 
1510
            int l;
 
1511
            if ((l=my_ismbchar(res_charset, pos, end)))
 
1512
            {
 
1513
              pos += l-1;
 
1514
              continue;
 
1515
            }
 
1516
          }
 
1517
#endif
1229
1518
 
1230
1519
          /*
1231
1520
            Special case when dumping BINARY/VARBINARY/BLOB values
1259
1548
            assert before the loop makes that sure.
1260
1549
          */
1261
1550
 
1262
 
          if ((needs_escaping(*pos, enclosed) ||
 
1551
          if ((NEED_ESCAPING(*pos) ||
1263
1552
               (check_second_byte &&
1264
1553
                my_mbcharlen(character_set_client, (unsigned char) *pos) == 2 &&
1265
1554
                pos + 1 < end &&
1266
 
                needs_escaping(pos[1], enclosed))) &&
 
1555
                NEED_ESCAPING(pos[1]))) &&
1267
1556
              /*
1268
 
                Don't escape field_term_char by doubling - doubling is only
1269
 
                valid for ENCLOSED BY characters:
 
1557
               Don't escape field_term_char by doubling - doubling is only
 
1558
               valid for ENCLOSED BY characters:
1270
1559
              */
1271
1560
              (enclosed || !is_ambiguous_field_term ||
1272
1561
               (int) (unsigned char) *pos != field_term_char))
1273
1562
          {
1274
 
            char tmp_buff[2];
 
1563
            char tmp_buff[2];
1275
1564
            tmp_buff[0]= ((int) (unsigned char) *pos == field_sep_char &&
1276
1565
                          is_ambiguous_field_sep) ?
1277
 
              field_sep_char : escape_char;
1278
 
            tmp_buff[1]= *pos ? *pos : '0';
1279
 
            if (my_b_write(cache,(unsigned char*) start,(uint32_t) (pos-start)) ||
1280
 
                my_b_write(cache,(unsigned char*) tmp_buff,2))
1281
 
              goto err;
1282
 
            start=pos+1;
1283
 
          }
1284
 
        }
1285
 
        if (my_b_write(cache,(unsigned char*) start,(uint32_t) (pos-start)))
1286
 
          goto err;
 
1566
                          field_sep_char : escape_char;
 
1567
            tmp_buff[1]= *pos ? *pos : '0';
 
1568
            if (my_b_write(&cache,(unsigned char*) start,(uint32_t) (pos-start)) ||
 
1569
                my_b_write(&cache,(unsigned char*) tmp_buff,2))
 
1570
              goto err;
 
1571
            start=pos+1;
 
1572
          }
 
1573
        }
 
1574
        if (my_b_write(&cache,(unsigned char*) start,(uint32_t) (pos-start)))
 
1575
          goto err;
1287
1576
      }
1288
 
      else if (my_b_write(cache,(unsigned char*) res->ptr(),used_length))
1289
 
        goto err;
 
1577
      else if (my_b_write(&cache,(unsigned char*) res->ptr(),used_length))
 
1578
        goto err;
1290
1579
    }
1291
1580
    if (fixed_row_size)
1292
1581
    {                                           // Fill with space
1293
1582
      if (item->max_length > used_length)
1294
1583
      {
1295
 
        /* QQ:  Fix by adding a my_b_fill() function */
1296
 
        if (!space_inited)
1297
 
        {
1298
 
          space_inited=1;
1299
 
          memset(space, ' ', sizeof(space));
1300
 
        }
1301
 
        uint32_t length=item->max_length-used_length;
1302
 
        for (; length > sizeof(space) ; length-=sizeof(space))
1303
 
        {
1304
 
          if (my_b_write(cache,(unsigned char*) space,sizeof(space)))
1305
 
            goto err;
1306
 
        }
1307
 
        if (my_b_write(cache,(unsigned char*) space,length))
1308
 
          goto err;
 
1584
        /* QQ:  Fix by adding a my_b_fill() function */
 
1585
        if (!space_inited)
 
1586
        {
 
1587
          space_inited=1;
 
1588
          memset(space, ' ', sizeof(space));
 
1589
        }
 
1590
        uint32_t length=item->max_length-used_length;
 
1591
        for (; length > sizeof(space) ; length-=sizeof(space))
 
1592
        {
 
1593
          if (my_b_write(&cache,(unsigned char*) space,sizeof(space)))
 
1594
            goto err;
 
1595
        }
 
1596
        if (my_b_write(&cache,(unsigned char*) space,length))
 
1597
          goto err;
1309
1598
      }
1310
1599
    }
1311
1600
    if (res && enclosed)
1312
1601
    {
1313
 
      if (my_b_write(cache, (unsigned char*) exchange->enclosed->ptr(),
 
1602
      if (my_b_write(&cache, (unsigned char*) exchange->enclosed->ptr(),
1314
1603
                     exchange->enclosed->length()))
1315
1604
        goto err;
1316
1605
    }
1317
1606
    if (--items_left)
1318
1607
    {
1319
 
      if (my_b_write(cache, (unsigned char*) exchange->field_term->ptr(),
 
1608
      if (my_b_write(&cache, (unsigned char*) exchange->field_term->ptr(),
1320
1609
                     field_term_length))
1321
1610
        goto err;
1322
1611
    }
1323
1612
  }
1324
 
  if (my_b_write(cache,(unsigned char*) exchange->line_term->ptr(),
1325
 
                 exchange->line_term->length()))
 
1613
  if (my_b_write(&cache,(unsigned char*) exchange->line_term->ptr(),
 
1614
                 exchange->line_term->length()))
1326
1615
    goto err;
1327
1616
  return(0);
1328
1617
err:
1339
1628
select_dump::prepare(List<Item> &, Select_Lex_Unit *u)
1340
1629
{
1341
1630
  unit= u;
1342
 
  return (int) ((file= create_file(session, path, exchange, cache)) < 0);
 
1631
  return (int) ((file= create_file(session, path, exchange, &cache)) < 0);
1343
1632
}
1344
1633
 
1345
1634
 
1366
1655
    res=item->str_result(&tmp);
1367
1656
    if (!res)                                   // If NULL
1368
1657
    {
1369
 
      if (my_b_write(cache,(unsigned char*) "",1))
 
1658
      if (my_b_write(&cache,(unsigned char*) "",1))
1370
1659
        goto err;
1371
1660
    }
1372
 
    else if (my_b_write(cache,(unsigned char*) res->ptr(),res->length()))
 
1661
    else if (my_b_write(&cache,(unsigned char*) res->ptr(),res->length()))
1373
1662
    {
1374
 
      my_error(ER_ERROR_ON_WRITE, MYF(0), path, errno);
 
1663
      my_error(ER_ERROR_ON_WRITE, MYF(0), path, my_errno);
1375
1664
      goto err;
1376
1665
    }
1377
1666
  }
1412
1701
void select_max_min_finder_subselect::cleanup()
1413
1702
{
1414
1703
  cache= 0;
 
1704
  return;
1415
1705
}
1416
1706
 
1417
1707
 
1531
1821
  return(0);
1532
1822
}
1533
1823
 
 
1824
 
 
1825
/*
 
1826
  Statement functions
 
1827
*/
 
1828
 
 
1829
Statement::Statement(LEX *lex_arg, MEM_ROOT *mem_root_arg, ulong id_arg)
 
1830
  :Query_arena(mem_root_arg),
 
1831
  id(id_arg),
 
1832
  mark_used_columns(MARK_COLUMNS_READ),
 
1833
  lex(lex_arg),
 
1834
  query(0),
 
1835
  query_length(0),
 
1836
  db(NULL),
 
1837
  db_length(0)
 
1838
{
 
1839
}
 
1840
 
 
1841
 
1534
1842
/*
1535
1843
  Don't free mem_root, as mem_root is freed in the end of dispatch_command
1536
1844
  (once for any command).
1541
1849
  lex_end(lex);
1542
1850
}
1543
1851
 
 
1852
 
1544
1853
bool Session::copy_db_to(char **p_db, size_t *p_db_length)
1545
1854
{
1546
 
  if (db.empty())
 
1855
  if (db == NULL)
1547
1856
  {
1548
1857
    my_message(ER_NO_DB_ERROR, ER(ER_NO_DB_ERROR), MYF(0));
1549
1858
    return true;
1550
1859
  }
1551
 
  *p_db= strmake(db.c_str(), db.length());
1552
 
  *p_db_length= db.length();
 
1860
  *p_db= strmake(db, db_length);
 
1861
  *p_db_length= db_length;
1553
1862
  return false;
1554
1863
}
1555
1864
 
 
1865
 
1556
1866
/****************************************************************************
1557
1867
  Tmp_Table_Param
1558
1868
****************************************************************************/
1564
1874
  quick_group= 1;
1565
1875
  table_charset= 0;
1566
1876
  precomputed_group_by= 0;
 
1877
  bit_fields_as_long= 0;
 
1878
  return;
1567
1879
}
1568
1880
 
1569
1881
void Tmp_Table_Param::cleanup(void)
1576
1888
  }
1577
1889
}
1578
1890
 
 
1891
 
 
1892
void session_increment_bytes_sent(ulong length)
 
1893
{
 
1894
  Session *session=current_session;
 
1895
  if (likely(session != 0))
 
1896
  { /* current_session==0 when close_connection() calls net_send_error() */
 
1897
    session->status_var.bytes_sent+= length;
 
1898
  }
 
1899
}
 
1900
 
 
1901
 
 
1902
void session_increment_bytes_received(ulong length)
 
1903
{
 
1904
  current_session->status_var.bytes_received+= length;
 
1905
}
 
1906
 
 
1907
 
 
1908
void session_increment_net_big_packet_count(ulong length)
 
1909
{
 
1910
  current_session->status_var.net_big_packet_count+= length;
 
1911
}
 
1912
 
1579
1913
void Session::send_kill_message() const
1580
1914
{
1581
1915
  int err= killed_errno();
1588
1922
  memset(&status_var, 0, sizeof(status_var));
1589
1923
}
1590
1924
 
 
1925
void Security_context::skip_grants()
 
1926
{
 
1927
  /* privileges for the user are unknown everything is allowed */
 
1928
}
 
1929
 
1591
1930
 
1592
1931
/****************************************************************************
1593
1932
  Handling of open and locked tables states.
1601
1940
{
1602
1941
  backup->set_open_tables_state(this);
1603
1942
  reset_open_tables_state();
1604
 
  backups_available= false;
 
1943
  state_flags|= Open_tables_state::BACKUPS_AVAIL;
 
1944
  return;
1605
1945
}
1606
1946
 
1607
1947
 
1612
1952
    to be sure that it was properly cleaned up.
1613
1953
  */
1614
1954
  assert(open_tables == 0 && temporary_tables == 0 &&
1615
 
              derived_tables == 0 &&
1616
 
              lock == 0);
 
1955
              handler_tables == 0 && derived_tables == 0 &&
 
1956
              lock == 0 && locked_tables == 0);
1617
1957
  set_open_tables_state(backup);
 
1958
  return;
1618
1959
}
1619
1960
 
1620
 
bool Session::set_db(const std::string &new_db)
 
1961
 
 
1962
bool Session::set_db(const char *new_db, size_t new_db_len)
1621
1963
{
1622
1964
  /* Do not reallocate memory if current chunk is big enough. */
1623
 
  if (new_db.length())
1624
 
    db= new_db;
 
1965
  if (db && new_db && db_length >= new_db_len)
 
1966
    memcpy(db, new_db, new_db_len+1);
1625
1967
  else
1626
 
    db.clear();
1627
 
 
1628
 
  return false;
 
1968
  {
 
1969
    if (db)
 
1970
      free(db);
 
1971
    if (new_db)
 
1972
    {
 
1973
      db= (char *)malloc(new_db_len + 1);
 
1974
      if (db != NULL)
 
1975
      {
 
1976
        memcpy(db, new_db, new_db_len);
 
1977
        db[new_db_len]= 0;
 
1978
      }
 
1979
    }
 
1980
    else
 
1981
      db= NULL;
 
1982
  }
 
1983
  db_length= db ? new_db_len : 0;
 
1984
  return new_db && !db;
1629
1985
}
1630
1986
 
1631
1987
 
1632
 
 
1633
 
 
1634
1988
/**
1635
1989
  Check the killed state of a user thread
1636
1990
  @param session  user thread
1643
1997
}
1644
1998
 
1645
1999
/**
1646
 
  Return the session id of a user session
1647
 
  @param pointer to Session object
1648
 
  @return session's id
 
2000
  Return the thread id of a user thread
 
2001
  @param session user thread
 
2002
  @return thread id
1649
2003
*/
1650
2004
extern "C" unsigned long session_get_thread_id(const Session *session)
1651
2005
{
1652
 
  return (unsigned long) session->getSessionId();
1653
 
}
1654
 
 
1655
 
 
1656
 
const struct charset_info_st *session_charset(Session *session)
 
2006
  return((unsigned long)session->thread_id);
 
2007
}
 
2008
 
 
2009
 
 
2010
extern "C"
 
2011
LEX_STRING *session_make_lex_string(Session *session, LEX_STRING *lex_str,
 
2012
                                const char *str, unsigned int size,
 
2013
                                int allocate_lex_string)
 
2014
{
 
2015
  return session->make_lex_string(lex_str, str, size,
 
2016
                              (bool) allocate_lex_string);
 
2017
}
 
2018
 
 
2019
extern "C" const struct charset_info_st *session_charset(Session *session)
1657
2020
{
1658
2021
  return(session->charset());
1659
2022
}
1660
2023
 
1661
 
int session_non_transactional_update(const Session *session)
1662
 
{
1663
 
  return(session->transaction.all.hasModifiedNonTransData());
1664
 
}
1665
 
 
1666
 
void session_mark_transaction_to_rollback(Session *session, bool all)
 
2024
extern "C" char **session_query(Session *session)
 
2025
{
 
2026
  return(&session->query);
 
2027
}
 
2028
 
 
2029
extern "C" int session_non_transactional_update(const Session *session)
 
2030
{
 
2031
  return(session->transaction.all.modified_non_trans_table);
 
2032
}
 
2033
 
 
2034
extern "C" void session_mark_transaction_to_rollback(Session *session, bool all)
1667
2035
{
1668
2036
  mark_transaction_to_rollback(session, all);
1669
2037
}
1670
2038
 
 
2039
 
1671
2040
/**
1672
2041
  Mark transaction to rollback and mark error as fatal to a sub-statement.
1673
2042
 
1674
2043
  @param  session   Thread handle
1675
2044
  @param  all   true <=> rollback main transaction.
1676
2045
*/
 
2046
 
1677
2047
void mark_transaction_to_rollback(Session *session, bool all)
1678
2048
{
1679
2049
  if (session)
1682
2052
    session->transaction_rollback_request= all;
1683
2053
  }
1684
2054
}
1685
 
 
1686
 
void Session::disconnect(uint32_t errcode, bool should_lock)
1687
 
{
1688
 
  /* Allow any plugins to cleanup their session variables */
1689
 
  plugin_sessionvar_cleanup(this);
1690
 
 
1691
 
  /* If necessary, log any aborted or unauthorized connections */
1692
 
  if (killed || client->wasAborted())
1693
 
    statistic_increment(aborted_threads, &LOCK_status);
1694
 
 
1695
 
  if (client->wasAborted())
1696
 
  {
1697
 
    if (! killed && variables.log_warnings > 1)
1698
 
    {
1699
 
      SecurityContext *sctx= &security_ctx;
1700
 
 
1701
 
      errmsg_printf(ERRMSG_LVL_WARN, ER(ER_NEW_ABORTING_CONNECTION)
1702
 
                  , thread_id
1703
 
                  , (db.empty() ? "unconnected" : db.c_str())
1704
 
                  , sctx->getUser().empty() == false ? sctx->getUser().c_str() : "unauthenticated"
1705
 
                  , sctx->getIp().c_str()
1706
 
                  , (main_da.is_error() ? main_da.message() : ER(ER_UNKNOWN_ERROR)));
1707
 
    }
1708
 
  }
1709
 
 
1710
 
  /* Close out our connection to the client */
 
2055
/***************************************************************************
 
2056
  Handling of XA id cacheing
 
2057
***************************************************************************/
 
2058
 
 
2059
pthread_mutex_t LOCK_xid_cache;
 
2060
HASH xid_cache;
 
2061
 
 
2062
extern "C" unsigned char *xid_get_hash_key(const unsigned char *, size_t *, bool);
 
2063
extern "C" void xid_free_hash(void *);
 
2064
 
 
2065
unsigned char *xid_get_hash_key(const unsigned char *ptr, size_t *length,
 
2066
                        bool )
 
2067
{
 
2068
  *length=((XID_STATE*)ptr)->xid.key_length();
 
2069
  return ((XID_STATE*)ptr)->xid.key();
 
2070
}
 
2071
 
 
2072
void xid_free_hash(void *ptr)
 
2073
{
 
2074
  if (!((XID_STATE*)ptr)->in_session)
 
2075
    free((unsigned char*)ptr);
 
2076
}
 
2077
 
 
2078
bool xid_cache_init()
 
2079
{
 
2080
  pthread_mutex_init(&LOCK_xid_cache, MY_MUTEX_INIT_FAST);
 
2081
  return hash_init(&xid_cache, &my_charset_bin, 100, 0, 0,
 
2082
                   xid_get_hash_key, xid_free_hash, 0) != 0;
 
2083
}
 
2084
 
 
2085
void xid_cache_free()
 
2086
{
 
2087
  if (hash_inited(&xid_cache))
 
2088
  {
 
2089
    hash_free(&xid_cache);
 
2090
    pthread_mutex_destroy(&LOCK_xid_cache);
 
2091
  }
 
2092
}
 
2093
 
 
2094
XID_STATE *xid_cache_search(XID *xid)
 
2095
{
 
2096
  pthread_mutex_lock(&LOCK_xid_cache);
 
2097
  XID_STATE *res=(XID_STATE *)hash_search(&xid_cache, xid->key(), xid->key_length());
 
2098
  pthread_mutex_unlock(&LOCK_xid_cache);
 
2099
  return res;
 
2100
}
 
2101
 
 
2102
 
 
2103
bool xid_cache_insert(XID *xid, enum xa_states xa_state)
 
2104
{
 
2105
  XID_STATE *xs;
 
2106
  bool res;
 
2107
  pthread_mutex_lock(&LOCK_xid_cache);
 
2108
  if (hash_search(&xid_cache, xid->key(), xid->key_length()))
 
2109
    res=0;
 
2110
  else if (!(xs=(XID_STATE *)malloc(sizeof(*xs))))
 
2111
    res=1;
 
2112
  else
 
2113
  {
 
2114
    xs->xa_state=xa_state;
 
2115
    xs->xid.set(xid);
 
2116
    xs->in_session=0;
 
2117
    res=my_hash_insert(&xid_cache, (unsigned char*)xs);
 
2118
  }
 
2119
  pthread_mutex_unlock(&LOCK_xid_cache);
 
2120
  return res;
 
2121
}
 
2122
 
 
2123
 
 
2124
bool xid_cache_insert(XID_STATE *xid_state)
 
2125
{
 
2126
  pthread_mutex_lock(&LOCK_xid_cache);
 
2127
  assert(hash_search(&xid_cache, xid_state->xid.key(),
 
2128
                          xid_state->xid.key_length())==0);
 
2129
  bool res=my_hash_insert(&xid_cache, (unsigned char*)xid_state);
 
2130
  pthread_mutex_unlock(&LOCK_xid_cache);
 
2131
  return res;
 
2132
}
 
2133
 
 
2134
 
 
2135
void xid_cache_delete(XID_STATE *xid_state)
 
2136
{
 
2137
  pthread_mutex_lock(&LOCK_xid_cache);
 
2138
  hash_delete(&xid_cache, (unsigned char *)xid_state);
 
2139
  pthread_mutex_unlock(&LOCK_xid_cache);
 
2140
}
 
2141
 
 
2142
namespace {
 
2143
  /**
 
2144
     Class to handle temporary allocation of memory for row data.
 
2145
 
 
2146
     The responsibilities of the class is to provide memory for
 
2147
     packing one or two rows of packed data (depending on what
 
2148
     constructor is called).
 
2149
 
 
2150
     In order to make the allocation more efficient for "simple" rows,
 
2151
     i.e., rows that do not contain any blobs, a pointer to the
 
2152
     allocated memory is of memory is stored in the table structure
 
2153
     for simple rows.  If memory for a table containing a blob field
 
2154
     is requested, only memory for that is allocated, and subsequently
 
2155
     released when the object is destroyed.
 
2156
 
 
2157
   */
 
2158
  class Row_data_memory {
 
2159
  public:
 
2160
    /**
 
2161
      Build an object to keep track of a block-local piece of memory
 
2162
      for storing a row of data.
 
2163
 
 
2164
      @param table
 
2165
      Table where the pre-allocated memory is stored.
 
2166
 
 
2167
      @param length
 
2168
      Length of data that is needed, if the record contain blobs.
 
2169
     */
 
2170
    Row_data_memory(Table *table, size_t const len1)
 
2171
      : m_memory(0)
 
2172
    {
 
2173
      m_alloc_checked= false;
 
2174
      allocate_memory(table, len1);
 
2175
      m_ptr[0]= has_memory() ? m_memory : 0;
 
2176
      m_ptr[1]= 0;
 
2177
    }
 
2178
 
 
2179
    Row_data_memory(Table *table, size_t const len1, size_t const len2)
 
2180
      : m_memory(0)
 
2181
    {
 
2182
      m_alloc_checked= false;
 
2183
      allocate_memory(table, len1 + len2);
 
2184
      m_ptr[0]= has_memory() ? m_memory        : 0;
 
2185
      m_ptr[1]= has_memory() ? m_memory + len1 : 0;
 
2186
    }
 
2187
 
 
2188
    ~Row_data_memory()
 
2189
    {
 
2190
      if (m_memory != 0 && m_release_memory_on_destruction)
 
2191
        free((unsigned char*) m_memory);
 
2192
    }
 
2193
 
 
2194
    /**
 
2195
       Is there memory allocated?
 
2196
 
 
2197
       @retval true There is memory allocated
 
2198
       @retval false Memory allocation failed
 
2199
     */
 
2200
    bool has_memory() const {
 
2201
      m_alloc_checked= true;
 
2202
      return m_memory != 0;
 
2203
    }
 
2204
 
 
2205
    unsigned char *slot(uint32_t s)
 
2206
    {
 
2207
      assert(s < sizeof(m_ptr)/sizeof(*m_ptr));
 
2208
      assert(m_ptr[s] != 0);
 
2209
      assert(m_alloc_checked == true);
 
2210
      return m_ptr[s];
 
2211
    }
 
2212
 
 
2213
  private:
 
2214
    void allocate_memory(Table *const table, size_t const total_length)
 
2215
    {
 
2216
      if (table->s->blob_fields == 0)
 
2217
      {
 
2218
        /*
 
2219
          The maximum length of a packed record is less than this
 
2220
          length. We use this value instead of the supplied length
 
2221
          when allocating memory for records, since we don't know how
 
2222
          the memory will be used in future allocations.
 
2223
 
 
2224
          Since table->s->reclength is for unpacked records, we have
 
2225
          to add two bytes for each field, which can potentially be
 
2226
          added to hold the length of a packed field.
 
2227
        */
 
2228
        size_t const maxlen= table->s->reclength + 2 * table->s->fields;
 
2229
 
 
2230
        /*
 
2231
          Allocate memory for two records if memory hasn't been
 
2232
          allocated. We allocate memory for two records so that it can
 
2233
          be used when processing update rows as well.
 
2234
        */
 
2235
        if (table->write_row_record == 0)
 
2236
          table->write_row_record=
 
2237
            (unsigned char *) alloc_root(&table->mem_root, 2 * maxlen);
 
2238
        m_memory= table->write_row_record;
 
2239
        m_release_memory_on_destruction= false;
 
2240
      }
 
2241
      else
 
2242
      {
 
2243
        m_memory= (unsigned char *) malloc(total_length);
 
2244
        m_release_memory_on_destruction= true;
 
2245
      }
 
2246
    }
 
2247
 
 
2248
    mutable bool m_alloc_checked;
 
2249
    bool m_release_memory_on_destruction;
 
2250
    unsigned char *m_memory;
 
2251
    unsigned char *m_ptr[2];
 
2252
  };
 
2253
}
 
2254
 
 
2255
/**
 
2256
  Close a connection.
 
2257
 
 
2258
  @param session                Thread handle
 
2259
  @param errcode        Error code to print to console
 
2260
  @param should_lock 1 if we have have to lock LOCK_thread_count
 
2261
 
 
2262
  @note
 
2263
    For the connection that is doing shutdown, this is called twice
 
2264
*/
 
2265
void Session::close_connection(uint32_t errcode, bool should_lock)
 
2266
{
 
2267
  st_vio *vio;
1711
2268
  if (should_lock)
1712
2269
    (void) pthread_mutex_lock(&LOCK_thread_count);
1713
2270
  killed= Session::KILL_CONNECTION;
1714
 
  if (client->isConnected())
 
2271
  if ((vio= net.vio) != 0)
1715
2272
  {
1716
2273
    if (errcode)
1717
 
    {
1718
 
      /*my_error(errcode, ER(errcode));*/
1719
 
      client->sendError(errcode, ER(errcode));
1720
 
    }
1721
 
    client->close();
 
2274
      net_send_error(this, errcode, ER(errcode)); /* purecov: inspected */
 
2275
    drizzleclient_net_close(&net);              /* vio is freed in delete session */
1722
2276
  }
1723
2277
  if (should_lock)
1724
2278
    (void) pthread_mutex_unlock(&LOCK_thread_count);
1725
2279
}
1726
2280
 
 
2281
 
 
2282
 
 
2283
/**
 
2284
 Reset Session part responsible for command processing state.
 
2285
 
 
2286
   This needs to be called before execution of every statement
 
2287
   (prepared or conventional).
 
2288
   It is not called by substatements of routines.
 
2289
 
 
2290
  @todo
 
2291
   Make it a method of Session and align its name with the rest of
 
2292
   reset/end/start/init methods.
 
2293
  @todo
 
2294
   Call it after we use Session for queries, not before.
 
2295
*/
 
2296
 
1727
2297
void Session::reset_for_next_command()
1728
2298
{
1729
2299
  free_list= 0;
1734
2304
  */
1735
2305
  auto_inc_intervals_in_cur_stmt_for_binlog.empty();
1736
2306
 
1737
 
  is_fatal_error= false;
 
2307
  is_fatal_error= 0;
1738
2308
  server_status&= ~ (SERVER_MORE_RESULTS_EXISTS |
1739
2309
                          SERVER_QUERY_NO_INDEX_USED |
1740
2310
                          SERVER_QUERY_NO_GOOD_INDEX_USED);
 
2311
  /*
 
2312
    If in autocommit mode and not in a transaction, reset
 
2313
    OPTION_STATUS_NO_TRANS_UPDATE | OPTION_KEEP_LOG to not get warnings
 
2314
    in ha_rollback_trans() about some tables couldn't be rolled back.
 
2315
  */
 
2316
  if (!(options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN)))
 
2317
  {
 
2318
    options&= ~OPTION_KEEP_LOG;
 
2319
    transaction.all.modified_non_trans_table= false;
 
2320
  }
 
2321
  thread_specific_used= false;
1741
2322
 
1742
2323
  clear_error();
1743
2324
  main_da.reset_diagnostics_area();
1744
2325
  total_warn_count=0;                   // Warnings for this query
1745
2326
  sent_row_count= examined_row_count= 0;
 
2327
 
 
2328
  return;
 
2329
}
 
2330
 
 
2331
 
 
2332
/**
 
2333
  return true if the table was created explicitly.
 
2334
*/
 
2335
inline bool is_user_table(Table * table)
 
2336
{
 
2337
  const char *name= table->s->table_name.str;
 
2338
  return strncmp(name, TMP_FILE_PREFIX, TMP_FILE_PREFIX_LENGTH);
1746
2339
}
1747
2340
 
1748
2341
/*
1749
2342
  Close all temporary tables created by 'CREATE TEMPORARY TABLE' for thread
 
2343
  creates one DROP TEMPORARY Table binlog event for each pseudo-thread
1750
2344
*/
1751
2345
 
1752
2346
void Session::close_temporary_tables()
1754
2348
  Table *table;
1755
2349
  Table *tmp_next;
1756
2350
 
1757
 
  if (not temporary_tables)
 
2351
  if (!temporary_tables)
1758
2352
    return;
1759
2353
 
1760
2354
  for (table= temporary_tables; table; table= tmp_next)
1761
2355
  {
1762
2356
    tmp_next= table->next;
1763
 
    nukeTable(table);
1764
 
  }
1765
 
  temporary_tables= NULL;
1766
 
}
1767
 
 
1768
 
/*
1769
 
  unlink from session->temporary tables and close temporary table
1770
 
*/
1771
 
 
1772
 
void Session::close_temporary_table(Table *table)
1773
 
{
1774
 
  if (table->prev)
1775
 
  {
1776
 
    table->prev->next= table->next;
1777
 
    if (table->prev->next)
1778
 
      table->next->prev= table->prev;
1779
 
  }
1780
 
  else
1781
 
  {
1782
 
    /* removing the item from the list */
1783
 
    assert(table == temporary_tables);
1784
 
    /*
1785
 
      slave must reset its temporary list pointer to zero to exclude
1786
 
      passing non-zero value to end_slave via rli->save_temporary_tables
1787
 
      when no temp tables opened, see an invariant below.
1788
 
    */
1789
 
    temporary_tables= table->next;
1790
 
    if (temporary_tables)
1791
 
      table->next->prev= NULL;
1792
 
  }
1793
 
  nukeTable(table);
1794
 
}
1795
 
 
1796
 
/*
1797
 
  Close and drop a temporary table
1798
 
 
1799
 
  NOTE
1800
 
  This dosn't unlink table from session->temporary
1801
 
  If this is needed, use close_temporary_table()
1802
 
*/
1803
 
 
1804
 
void Session::nukeTable(Table *table)
1805
 
{
1806
 
  plugin::StorageEngine *table_type= table->s->db_type();
1807
 
 
1808
 
  table->free_io_cache();
1809
 
  table->closefrm(false);
1810
 
 
1811
 
  TableIdentifier identifier(table->s->getSchemaName(), table->s->table_name.str, table->s->path.str);
1812
 
  rm_temporary_table(table_type, identifier);
1813
 
 
1814
 
  table->s->free_table_share();
1815
 
 
1816
 
  /* This makes me sad, but we're allocating it via malloc */
1817
 
  free(table);
1818
 
}
 
2357
    close_temporary(table, 1, 1);
 
2358
  }
 
2359
  temporary_tables= 0;
 
2360
 
 
2361
  return;
 
2362
}
 
2363
 
1819
2364
 
1820
2365
/** Clear most status variables. */
1821
2366
extern time_t flush_status_time;
1835
2380
  reset_status_vars();
1836
2381
 
1837
2382
  /* Reset the counters of all key caches (default and named). */
1838
 
  reset_key_cache_counters();
 
2383
  process_key_caches(reset_key_cache_counters);
1839
2384
  flush_status_time= time((time_t*) 0);
1840
2385
  max_used_connections= 1; /* We set it to one, because we know we exist */
1841
2386
  pthread_mutex_unlock(&LOCK_status);
1842
2387
}
1843
 
 
1844
 
user_var_entry *Session::getVariable(LEX_STRING &name, bool create_if_not_exists)
1845
 
{
1846
 
  user_var_entry *entry= NULL;
1847
 
 
1848
 
  entry= (user_var_entry*) hash_search(&user_vars, (unsigned char*) name.str, name.length);
1849
 
 
1850
 
  if ((entry == NULL) && create_if_not_exists)
1851
 
  {
1852
 
    if (!hash_inited(&user_vars))
1853
 
      return NULL;
1854
 
    entry= new (nothrow) user_var_entry(name.str, query_id);
1855
 
 
1856
 
    if (entry == NULL)
1857
 
      return NULL;
1858
 
 
1859
 
    if (my_hash_insert(&user_vars, (unsigned char*) entry))
1860
 
    {
1861
 
      assert(1);
1862
 
      free((char*) entry);
1863
 
      return 0;
1864
 
    }
1865
 
 
1866
 
  }
1867
 
 
1868
 
  return entry;
1869
 
}
1870
 
 
1871
 
void Session::mark_temp_tables_as_free_for_reuse()
1872
 
{
1873
 
  for (Table *table= temporary_tables ; table ; table= table->next)
1874
 
  {
1875
 
    if (table->query_id == query_id)
1876
 
    {
1877
 
      table->query_id= 0;
1878
 
      table->cursor->ha_reset();
1879
 
    }
1880
 
  }
1881
 
}
1882
 
 
1883
 
void Session::mark_used_tables_as_free_for_reuse(Table *table)
1884
 
{
1885
 
  for (; table ; table= table->next)
1886
 
  {
1887
 
    if (table->query_id == query_id)
1888
 
    {
1889
 
      table->query_id= 0;
1890
 
      table->cursor->ha_reset();
1891
 
    }
1892
 
  }
1893
 
}
1894
 
 
1895
 
/*
1896
 
  Unlocks tables and frees derived tables.
1897
 
  Put all normal tables used by thread in free list.
1898
 
 
1899
 
  It will only close/mark as free for reuse tables opened by this
1900
 
  substatement, it will also check if we are closing tables after
1901
 
  execution of complete query (i.e. we are on upper level) and will
1902
 
  leave prelocked mode if needed.
1903
 
*/
1904
 
void Session::close_thread_tables()
1905
 
{
1906
 
  Table *table;
1907
 
 
1908
 
  /*
1909
 
    We are assuming here that session->derived_tables contains ONLY derived
1910
 
    tables for this substatement. i.e. instead of approach which uses
1911
 
    query_id matching for determining which of the derived tables belong
1912
 
    to this substatement we rely on the ability of substatements to
1913
 
    save/restore session->derived_tables during their execution.
1914
 
 
1915
 
    TODO: Probably even better approach is to simply associate list of
1916
 
          derived tables with (sub-)statement instead of thread and destroy
1917
 
          them at the end of its execution.
1918
 
  */
1919
 
  if (derived_tables)
1920
 
  {
1921
 
    Table *next;
1922
 
    /*
1923
 
      Close all derived tables generated in queries like
1924
 
      SELECT * FROM (SELECT * FROM t1)
1925
 
    */
1926
 
    for (table= derived_tables ; table ; table= next)
1927
 
    {
1928
 
      next= table->next;
1929
 
      table->free_tmp_table(this);
1930
 
    }
1931
 
    derived_tables= 0;
1932
 
  }
1933
 
 
1934
 
  /*
1935
 
    Mark all temporary tables used by this statement as free for reuse.
1936
 
  */
1937
 
  mark_temp_tables_as_free_for_reuse();
1938
 
  /*
1939
 
    Let us commit transaction for statement. Since in 5.0 we only have
1940
 
    one statement transaction and don't allow several nested statement
1941
 
    transactions this call will do nothing if we are inside of stored
1942
 
    function or trigger (i.e. statement transaction is already active and
1943
 
    does not belong to statement for which we do close_thread_tables()).
1944
 
    TODO: This should be fixed in later releases.
1945
 
   */
1946
 
  if (backups_available == false)
1947
 
  {
1948
 
    TransactionServices &transaction_services= TransactionServices::singleton();
1949
 
    main_da.can_overwrite_status= true;
1950
 
    transaction_services.ha_autocommit_or_rollback(this, is_error());
1951
 
    main_da.can_overwrite_status= false;
1952
 
    transaction.stmt.reset();
1953
 
  }
1954
 
 
1955
 
  if (lock)
1956
 
  {
1957
 
    /*
1958
 
      For RBR we flush the pending event just before we unlock all the
1959
 
      tables.  This means that we are at the end of a topmost
1960
 
      statement, so we ensure that the STMT_END_F flag is set on the
1961
 
      pending event.  For statements that are *inside* stored
1962
 
      functions, the pending event will not be flushed: that will be
1963
 
      handled either before writing a query log event (inside
1964
 
      binlog_query()) or when preparing a pending event.
1965
 
     */
1966
 
    mysql_unlock_tables(this, lock);
1967
 
    lock= 0;
1968
 
  }
1969
 
  /*
1970
 
    Note that we need to hold LOCK_open while changing the
1971
 
    open_tables list. Another thread may work on it.
1972
 
    (See: remove_table_from_cache(), mysql_wait_completed_table())
1973
 
    Closing a MERGE child before the parent would be fatal if the
1974
 
    other thread tries to abort the MERGE lock in between.
1975
 
  */
1976
 
  if (open_tables)
1977
 
    close_open_tables();
1978
 
}
1979
 
 
1980
 
void Session::close_tables_for_reopen(TableList **tables)
1981
 
{
1982
 
  /*
1983
 
    If table list consists only from tables from prelocking set, table list
1984
 
    for new attempt should be empty, so we have to update list's root pointer.
1985
 
  */
1986
 
  if (lex->first_not_own_table() == *tables)
1987
 
    *tables= 0;
1988
 
  lex->chop_off_not_own_tables();
1989
 
  for (TableList *tmp= *tables; tmp; tmp= tmp->next_global)
1990
 
    tmp->table= 0;
1991
 
  close_thread_tables();
1992
 
}
1993
 
 
1994
 
bool Session::openTablesLock(TableList *tables)
1995
 
{
1996
 
  uint32_t counter;
1997
 
  bool need_reopen;
1998
 
 
1999
 
  for ( ; ; )
2000
 
  {
2001
 
    if (open_tables_from_list(&tables, &counter))
2002
 
      return true;
2003
 
 
2004
 
    if (not lock_tables(tables, counter, &need_reopen))
2005
 
      break;
2006
 
    if (not need_reopen)
2007
 
      return true;
2008
 
    close_tables_for_reopen(&tables);
2009
 
  }
2010
 
  if ((mysql_handle_derived(lex, &mysql_derived_prepare) ||
2011
 
       (fill_derived_tables() &&
2012
 
        mysql_handle_derived(lex, &mysql_derived_filling))))
2013
 
    return true;
2014
 
 
2015
 
  return false;
2016
 
}
2017
 
 
2018
 
bool Session::openTables(TableList *tables, uint32_t flags)
2019
 
{
2020
 
  uint32_t counter;
2021
 
  bool ret= fill_derived_tables();
2022
 
  assert(ret == false);
2023
 
  if (open_tables_from_list(&tables, &counter, flags) ||
2024
 
      mysql_handle_derived(lex, &mysql_derived_prepare))
2025
 
    return true;
2026
 
  return false;
2027
 
}
2028
 
 
2029
 
bool Session::rm_temporary_table(TableIdentifier &identifier)
2030
 
{
2031
 
  if (plugin::StorageEngine::dropTable(*this, identifier))
2032
 
  {
2033
 
    errmsg_printf(ERRMSG_LVL_WARN, _("Could not remove temporary table: '%s', error: %d"),
2034
 
                  identifier.getSQLPath().c_str(), errno);
2035
 
    dumpTemporaryTableNames("rm_temporary_table()");
2036
 
 
2037
 
    return true;
2038
 
  }
2039
 
 
2040
 
  return false;
2041
 
}
2042
 
 
2043
 
bool Session::rm_temporary_table(plugin::StorageEngine *base, TableIdentifier &identifier)
2044
 
{
2045
 
  assert(base);
2046
 
 
2047
 
  if (plugin::StorageEngine::dropTable(*this, *base, identifier))
2048
 
  {
2049
 
    errmsg_printf(ERRMSG_LVL_WARN, _("Could not remove temporary table: '%s', error: %d"),
2050
 
                  identifier.getSQLPath().c_str(), errno);
2051
 
    dumpTemporaryTableNames("rm_temporary_table()");
2052
 
 
2053
 
    return true;
2054
 
  }
2055
 
 
2056
 
  return false;
2057
 
}
2058
 
 
2059
 
/**
2060
 
  @note this will be removed, I am looking through Hudson to see if it is finding
2061
 
  any tables that are missed during cleanup.
2062
 
*/
2063
 
void Session::dumpTemporaryTableNames(const char *foo)
2064
 
{
2065
 
  Table *table;
2066
 
 
2067
 
  if (not temporary_tables)
2068
 
    return;
2069
 
 
2070
 
  cerr << "Begin Run: " << foo << "\n";
2071
 
  for (table= temporary_tables; table; table= table->next)
2072
 
  {
2073
 
    bool have_proto= false;
2074
 
 
2075
 
    message::Table *proto= table->s->getTableProto();
2076
 
    if (table->s->getTableProto())
2077
 
      have_proto= true;
2078
 
 
2079
 
    const char *answer= have_proto ? "true" : "false";
2080
 
 
2081
 
    if (have_proto)
2082
 
    {
2083
 
      cerr << "\tTable Name " << table->s->getSchemaName() << "." << table->s->table_name.str << " : " << answer << "\n";
2084
 
      cerr << "\t\t Proto " << proto->schema() << " " << proto->name() << "\n";
2085
 
    }
2086
 
    else
2087
 
      cerr << "\tTabl;e Name " << table->s->getSchemaName() << "." << table->s->table_name.str << " : " << answer << "\n";
2088
 
  }
2089
 
}
2090
 
 
2091
 
bool Session::storeTableMessage(TableIdentifier &identifier, message::Table &table_message)
2092
 
{
2093
 
  table_message_cache.insert(make_pair(identifier.getPath(), table_message));
2094
 
 
2095
 
  return true;
2096
 
}
2097
 
 
2098
 
bool Session::removeTableMessage(TableIdentifier &identifier)
2099
 
{
2100
 
  TableMessageCache::iterator iter;
2101
 
 
2102
 
  iter= table_message_cache.find(identifier.getPath());
2103
 
 
2104
 
  if (iter == table_message_cache.end())
2105
 
    return false;
2106
 
 
2107
 
  table_message_cache.erase(iter);
2108
 
 
2109
 
  return true;
2110
 
}
2111
 
 
2112
 
bool Session::getTableMessage(TableIdentifier &identifier, message::Table &table_message)
2113
 
{
2114
 
  TableMessageCache::iterator iter;
2115
 
 
2116
 
  iter= table_message_cache.find(identifier.getPath());
2117
 
 
2118
 
  if (iter == table_message_cache.end())
2119
 
    return false;
2120
 
 
2121
 
  table_message.CopyFrom(((*iter).second));
2122
 
 
2123
 
  return true;
2124
 
}
2125
 
 
2126
 
bool Session::doesTableMessageExist(TableIdentifier &identifier)
2127
 
{
2128
 
  TableMessageCache::iterator iter;
2129
 
 
2130
 
  iter= table_message_cache.find(identifier.getPath());
2131
 
 
2132
 
  if (iter == table_message_cache.end())
2133
 
  {
2134
 
    return false;
2135
 
  }
2136
 
 
2137
 
  return true;
2138
 
}
2139
 
 
2140
 
bool Session::renameTableMessage(TableIdentifier &from, TableIdentifier &to)
2141
 
{
2142
 
  TableMessageCache::iterator iter;
2143
 
 
2144
 
  table_message_cache[to.getPath()]= table_message_cache[from.getPath()];
2145
 
 
2146
 
  iter= table_message_cache.find(to.getPath());
2147
 
 
2148
 
  if (iter == table_message_cache.end())
2149
 
  {
2150
 
    return false;
2151
 
  }
2152
 
 
2153
 
  (*iter).second.set_schema(to.getSchemaName());
2154
 
  (*iter).second.set_name(to.getTableName());
2155
 
 
2156
 
  return true;
2157
 
}
2158
 
 
2159
 
} /* namespace drizzled */