← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/session.cc

  • Committer: Brian Aker
  • Date: 2009-05-15 17:06:35 UTC
  • mto: This revision was merged to the branch mainline in revision 1023.
  • Revision ID: brian@gaz-20090515170635-croy1u63a3gqdn9n
Dead convert functions for character sets.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/session.cc
1
 
/* Copyright (C) 2000-2006 MySQL AB
2
 
 
3
 
   This program is free software; you can redistribute it and/or modify
4
 
   it under the terms of the GNU General Public License as published by
5
 
   the Free Software Foundation; version 2 of the License.
6
 
 
7
 
   This program is distributed in the hope that it will be useful,
8
 
   but WITHOUT ANY WARRANTY; without even the implied warranty of
9
 
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10
 
   GNU General Public License for more details.
11
 
 
12
 
   You should have received a copy of the GNU General Public License
13
 
   along with this program; if not, write to the Free Software
14
 
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
15
 
 
16
 
 
17
 
/*****************************************************************************
18
 
**
19
 
** This file implements classes defined in sql_class.h
20
 
** Especially the classes to handle a result from a select
21
 
**
22
 
*****************************************************************************/
 
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 *  Copyright (C) 2008 Sun Microsystems
 
5
 *
 
6
 *  This program is free software; you can redistribute it and/or modify
 
7
 *  it under the terms of the GNU General Public License as published by
 
8
 *  the Free Software Foundation; version 2 of the License.
 
9
 *
 
10
 *  This program is distributed in the hope that it will be useful,
 
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
 *  GNU General Public License for more details.
 
14
 *
 
15
 *  You should have received a copy of the GNU General Public License
 
16
 *  along with this program; if not, write to the Free Software
 
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
18
 */
 
19
 
23
20
#include <drizzled/server_includes.h>
24
 
#include "rpl_rli.h"
25
 
#include "rpl_record.h"
26
 
#include "log_event.h"
 
21
#include <drizzled/session.h>
27
22
#include <sys/stat.h>
28
 
#include <mysys/thr_alarm.h>
29
23
#include <mysys/mysys_err.h>
30
 
#include <drizzled/drizzled_error_messages.h>
 
24
#include <drizzled/error.h>
 
25
#include <drizzled/query_id.h>
 
26
#include <drizzled/data_home.h>
 
27
#include <drizzled/sql_base.h>
 
28
#include <drizzled/lock.h>
 
29
#include <drizzled/item/cache.h>
 
30
#include <drizzled/item/float.h>
 
31
#include <drizzled/item/return_int.h>
 
32
#include <drizzled/item/empty_string.h>
 
33
#include <drizzled/show.h>
 
34
#include <drizzled/scheduling.h>
31
35
 
32
36
/*
33
37
  The following is used to initialise Table_ident with a internal
36
40
char internal_table_name[2]= "*";
37
41
char empty_c_string[1]= {0};    /* used for not defined db */
38
42
 
39
 
const char * const THD::DEFAULT_WHERE= "field list";
40
 
 
 
43
const char * const Session::DEFAULT_WHERE= "field list";
 
44
extern pthread_key_t THR_Session;
 
45
extern pthread_key_t THR_Mem_root;
41
46
 
42
47
/*****************************************************************************
43
48
** Instansiate templates
58
63
/****************************************************************************
59
64
** User variables
60
65
****************************************************************************/
61
 
 
62
66
extern "C" unsigned char *get_var_key(user_var_entry *entry, size_t *length,
63
 
                              bool not_used __attribute__((unused)))
 
67
                              bool )
64
68
{
65
69
  *length= entry->name.length;
66
70
  return (unsigned char*) entry->name.str;
81
85
         !strcmp(field_name.str, other.field_name.str);
82
86
}
83
87
 
84
 
/**
85
 
  Construct an (almost) deep copy of this key. Only those
86
 
  elements that are known to never change are not copied.
87
 
  If out of memory, a partial copy is returned and an error is set
88
 
  in THD.
89
 
*/
90
 
 
91
 
Key::Key(const Key &rhs, MEM_ROOT *mem_root)
92
 
  :type(rhs.type),
93
 
  key_create_info(rhs.key_create_info),
94
 
  columns(rhs.columns, mem_root),
95
 
  name(rhs.name),
96
 
  generated(rhs.generated)
97
 
{
98
 
  list_copy_and_replace_each_value(columns, mem_root);
99
 
}
100
 
 
101
 
/**
102
 
  Construct an (almost) deep copy of this foreign key. Only those
103
 
  elements that are known to never change are not copied.
104
 
  If out of memory, a partial copy is returned and an error is set
105
 
  in THD.
106
 
*/
107
 
 
108
 
Foreign_key::Foreign_key(const Foreign_key &rhs, MEM_ROOT *mem_root)
109
 
  :Key(rhs),
110
 
  ref_table(rhs.ref_table),
111
 
  ref_columns(rhs.ref_columns),
112
 
  delete_opt(rhs.delete_opt),
113
 
  update_opt(rhs.update_opt),
114
 
  match_opt(rhs.match_opt)
115
 
{
116
 
  list_copy_and_replace_each_value(ref_columns, mem_root);
117
 
}
118
 
 
119
 
/*
120
 
  Test if a foreign key (= generated key) is a prefix of the given key
121
 
  (ignoring key name, key type and order of columns)
122
 
 
123
 
  NOTES:
124
 
    This is only used to test if an index for a FOREIGN KEY exists
125
 
 
126
 
  IMPLEMENTATION
127
 
    We only compare field names
128
 
 
129
 
  RETURN
130
 
    0   Generated key is a prefix of other key
131
 
    1   Not equal
132
 
*/
133
 
 
134
 
bool foreign_key_prefix(Key *a, Key *b)
135
 
{
136
 
  /* Ensure that 'a' is the generated key */
137
 
  if (a->generated)
138
 
  {
139
 
    if (b->generated && a->columns.elements > b->columns.elements)
140
 
      std::swap(a, b);                       // Put shorter key in 'a'
141
 
  }
142
 
  else
143
 
  {
144
 
    if (!b->generated)
145
 
      return true;                              // No foreign key
146
 
    std::swap(a, b);                       // Put generated key in 'a'
147
 
  }
148
 
 
149
 
  /* Test if 'a' is a prefix of 'b' */
150
 
  if (a->columns.elements > b->columns.elements)
151
 
    return true;                                // Can't be prefix
152
 
 
153
 
  List_iterator<Key_part_spec> col_it1(a->columns);
154
 
  List_iterator<Key_part_spec> col_it2(b->columns);
155
 
  const Key_part_spec *col1, *col2;
156
 
 
157
 
#ifdef ENABLE_WHEN_INNODB_CAN_HANDLE_SWAPED_FOREIGN_KEY_COLUMNS
158
 
  while ((col1= col_it1++))
159
 
  {
160
 
    bool found= 0;
161
 
    col_it2.rewind();
162
 
    while ((col2= col_it2++))
163
 
    {
164
 
      if (*col1 == *col2)
165
 
      {
166
 
        found= true;
167
 
        break;
168
 
      }
169
 
    }
170
 
    if (!found)
171
 
      return true;                              // Error
172
 
  }
173
 
  return false;                                 // Is prefix
174
 
#else
175
 
  while ((col1= col_it1++))
176
 
  {
177
 
    col2= col_it2++;
178
 
    if (!(*col1 == *col2))
179
 
      return true;
180
 
  }
181
 
  return false;                                 // Is prefix
182
 
#endif
183
 
}
184
 
 
185
 
 
186
88
/****************************************************************************
187
89
** Thread specific functions
188
90
****************************************************************************/
200
102
extern "C" int mysql_tmpfile(const char *prefix)
201
103
{
202
104
  char filename[FN_REFLEN];
203
 
  File fd = create_temp_file(filename, mysql_tmpdir, prefix,
204
 
                             O_CREAT | O_EXCL | O_RDWR | O_TEMPORARY,
 
105
  File fd = create_temp_file(filename, drizzle_tmpdir, prefix,
 
106
                             O_CREAT | O_EXCL | O_RDWR,
205
107
                             MYF(MY_WME));
206
108
  if (fd >= 0) {
207
 
    /*
208
 
      This can be removed once the following bug is fixed:
209
 
      Bug #28903  create_temp_file() doesn't honor O_TEMPORARY option
210
 
                  (file not removed) (Unix)
211
 
    */
212
109
    unlink(filename);
213
110
  }
214
111
 
217
114
 
218
115
 
219
116
extern "C"
220
 
int thd_in_lock_tables(const THD *thd)
221
 
{
222
 
  return test(thd->in_lock_tables);
223
 
}
224
 
 
225
 
 
226
 
extern "C"
227
 
int thd_tablespace_op(const THD *thd)
228
 
{
229
 
  return test(thd->tablespace_op);
230
 
}
231
 
 
232
 
 
233
 
extern "C"
234
 
const char *set_thd_proc_info(THD *thd, const char *info,
235
 
                              const char *calling_function __attribute__((unused)),
236
 
                              const char *calling_file __attribute__((unused)),
237
 
                              const unsigned int calling_line __attribute__((unused)))
238
 
{
239
 
  const char *old_info= thd->get_proc_info();
240
 
  thd->set_proc_info(info);
241
 
  return old_info;
242
 
}
243
 
 
244
 
extern "C"
245
 
void **thd_ha_data(const THD *thd, const struct handlerton *hton)
246
 
{
247
 
  return (void **) &thd->ha_data[hton->slot].ha_ptr;
248
 
}
249
 
 
250
 
extern "C"
251
 
int64_t thd_test_options(const THD *thd, int64_t test_options)
252
 
{
253
 
  return thd->options & test_options;
254
 
}
255
 
 
256
 
extern "C"
257
 
int thd_sql_command(const THD *thd)
258
 
{
259
 
  return (int) thd->lex->sql_command;
260
 
}
261
 
 
262
 
extern "C"
263
 
int thd_tx_isolation(const THD *thd)
264
 
{
265
 
  return (int) thd->variables.tx_isolation;
266
 
}
267
 
 
268
 
extern "C"
269
 
void thd_inc_row_count(THD *thd)
270
 
{
271
 
  thd->row_count++;
272
 
}
273
 
 
274
 
/**
275
 
  Clear this diagnostics area. 
276
 
 
277
 
  Normally called at the end of a statement.
278
 
*/
279
 
 
280
 
void
281
 
Diagnostics_area::reset_diagnostics_area()
282
 
{
283
 
  can_overwrite_status= false;
284
 
  /** Don't take chances in production */
285
 
  m_message[0]= '\0';
286
 
  m_sql_errno= 0;
287
 
  m_server_status= 0;
288
 
  m_affected_rows= 0;
289
 
  m_last_insert_id= 0;
290
 
  m_total_warn_count= 0;
291
 
  is_sent= false;
292
 
  /** Tiny reset in debug mode to see garbage right away */
293
 
  m_status= DA_EMPTY;
294
 
}
295
 
 
296
 
 
297
 
/**
298
 
  Set OK status -- ends commands that do not return a
299
 
  result set, e.g. INSERT/UPDATE/DELETE.
300
 
*/
301
 
 
302
 
void
303
 
Diagnostics_area::set_ok_status(THD *thd, ha_rows affected_rows_arg,
304
 
                                uint64_t last_insert_id_arg,
305
 
                                const char *message_arg)
306
 
{
307
 
  assert(! is_set());
308
 
  /*
309
 
    In production, refuse to overwrite an error or a custom response
310
 
    with an OK packet.
311
 
  */
312
 
  if (is_error() || is_disabled())
313
 
    return;
314
 
  /** Only allowed to report success if has not yet reported an error */
315
 
 
316
 
  m_server_status= thd->server_status;
317
 
  m_total_warn_count= thd->total_warn_count;
318
 
  m_affected_rows= affected_rows_arg;
319
 
  m_last_insert_id= last_insert_id_arg;
320
 
  if (message_arg)
321
 
    strmake(m_message, message_arg, sizeof(m_message) - 1);
322
 
  else
323
 
    m_message[0]= '\0';
324
 
  m_status= DA_OK;
325
 
}
326
 
 
327
 
 
328
 
/**
329
 
  Set EOF status.
330
 
*/
331
 
 
332
 
void
333
 
Diagnostics_area::set_eof_status(THD *thd)
334
 
{
335
 
  /** Only allowed to report eof if has not yet reported an error */
336
 
 
337
 
  assert(! is_set());
338
 
  /*
339
 
    In production, refuse to overwrite an error or a custom response
340
 
    with an EOF packet.
341
 
  */
342
 
  if (is_error() || is_disabled())
343
 
    return;
344
 
 
345
 
  m_server_status= thd->server_status;
346
 
  /*
347
 
    If inside a stored procedure, do not return the total
348
 
    number of warnings, since they are not available to the client
349
 
    anyway.
350
 
  */
351
 
  m_total_warn_count= thd->total_warn_count;
352
 
 
353
 
  m_status= DA_EOF;
354
 
}
355
 
 
356
 
/**
357
 
  Set ERROR status.
358
 
*/
359
 
 
360
 
void
361
 
Diagnostics_area::set_error_status(THD *thd __attribute__((unused)),
362
 
                                   uint32_t sql_errno_arg,
363
 
                                   const char *message_arg)
364
 
{
365
 
  /*
366
 
    Only allowed to report error if has not yet reported a success
367
 
    The only exception is when we flush the message to the client,
368
 
    an error can happen during the flush.
369
 
  */
370
 
  assert(! is_set() || can_overwrite_status);
371
 
  /*
372
 
    In production, refuse to overwrite a custom response with an
373
 
    ERROR packet.
374
 
  */
375
 
  if (is_disabled())
376
 
    return;
377
 
 
378
 
  m_sql_errno= sql_errno_arg;
379
 
  strmake(m_message, message_arg, sizeof(m_message) - 1);
380
 
 
381
 
  m_status= DA_ERROR;
382
 
}
383
 
 
384
 
 
385
 
/**
386
 
  Mark the diagnostics area as 'DISABLED'.
387
 
 
388
 
  This is used in rare cases when the COM_ command at hand sends a response
389
 
  in a custom format. One example is the query cache, another is
390
 
  COM_STMT_PREPARE.
391
 
*/
392
 
 
393
 
void
394
 
Diagnostics_area::disable_status()
395
 
{
396
 
  assert(! is_set());
397
 
  m_status= DA_DISABLED;
398
 
}
399
 
 
400
 
 
401
 
THD::THD()
 
117
int session_in_lock_tables(const Session *session)
 
118
{
 
119
  return test(session->in_lock_tables);
 
120
}
 
121
 
 
122
 
 
123
extern "C"
 
124
int session_tablespace_op(const Session *session)
 
125
{
 
126
  return test(session->tablespace_op);
 
127
}
 
128
 
 
129
 
 
130
/**
 
131
   Set the process info field of the Session structure.
 
132
 
 
133
   This function is used by plug-ins. Internally, the
 
134
   Session::set_proc_info() function should be used.
 
135
 
 
136
   @see Session::set_proc_info
 
137
 */
 
138
extern "C" void
 
139
set_session_proc_info(Session *session, const char *info)
 
140
{
 
141
  session->set_proc_info(info);
 
142
}
 
143
 
 
144
extern "C"
 
145
const char *get_session_proc_info(Session *session)
 
146
{
 
147
  return session->get_proc_info();
 
148
}
 
149
 
 
150
extern "C"
 
151
void **session_ha_data(const Session *session, const struct StorageEngine *engine)
 
152
{
 
153
  return (void **) &session->ha_data[engine->slot].ha_ptr;
 
154
}
 
155
 
 
156
extern "C"
 
157
int64_t session_test_options(const Session *session, int64_t test_options)
 
158
{
 
159
  return session->options & test_options;
 
160
}
 
161
 
 
162
extern "C"
 
163
int session_sql_command(const Session *session)
 
164
{
 
165
  return (int) session->lex->sql_command;
 
166
}
 
167
 
 
168
extern "C"
 
169
int session_tx_isolation(const Session *session)
 
170
{
 
171
  return (int) session->variables.tx_isolation;
 
172
}
 
173
 
 
174
extern "C"
 
175
void session_inc_row_count(Session *session)
 
176
{
 
177
  session->row_count++;
 
178
}
 
179
 
 
180
Session::Session(Protocol *protocol_arg)
402
181
   :Statement(&main_lex, &main_mem_root,
403
182
              /* statement id */ 0),
404
 
   Open_tables_state(refresh_version), rli_fake(0),
 
183
   Open_tables_state(refresh_version),
405
184
   lock_id(&main_lock_id),
406
 
   user_time(0), in_sub_stmt(0),
407
 
   binlog_table_maps(0), binlog_flags(0UL),
 
185
   user_time(0),
408
186
   arg_of_last_insert_id_function(false),
409
187
   first_successful_insert_id_in_prev_stmt(0),
410
 
   first_successful_insert_id_in_prev_stmt_for_binlog(0),
411
188
   first_successful_insert_id_in_cur_stmt(0),
412
 
   stmt_depends_on_first_successful_insert_id_in_prev_stmt(false),
413
189
   global_read_lock(0),
414
190
   is_fatal_error(0),
415
191
   transaction_rollback_request(0),
416
192
   is_fatal_sub_stmt_error(0),
417
 
   rand_used(0),
418
 
   time_zone_used(0),
419
193
   in_lock_tables(0),
420
 
   bootstrap(0),
421
194
   derived_tables_processing(false),
422
 
   m_lip(NULL)
 
195
   m_lip(NULL),
 
196
   scheduler(0)
423
197
{
424
 
  ulong tmp;
 
198
  uint64_t tmp;
 
199
 
 
200
  memset(process_list_info, 0, PROCESS_LIST_WIDTH);
425
201
 
426
202
  /*
427
203
    Pass nominal parameters to init_alloc_root only to ensure that
431
207
  init_sql_alloc(&main_mem_root, ALLOC_ROOT_MIN_BLOCK_SIZE, 0);
432
208
  thread_stack= 0;
433
209
  catalog= (char*)"std"; // the only catalog we have for now
434
 
  main_security_ctx.init();
435
 
  security_ctx= &main_security_ctx;
436
210
  some_tables_deleted=no_errors=password= 0;
437
 
  query_start_used= 0;
438
211
  count_cuted_fields= CHECK_FIELD_IGNORE;
439
212
  killed= NOT_KILLED;
440
213
  col_access=0;
441
 
  is_slave_error= thread_specific_used= false;
442
 
  hash_clear(&handler_tables_hash);
443
214
  tmp_table=0;
444
215
  used_tables=0;
445
216
  cuted_fields= sent_row_count= row_count= 0L;
446
217
  limit_found_rows= 0;
447
218
  row_count_func= -1;
448
219
  statement_id_counter= 0UL;
449
 
  // Must be reset to handle error with THD's created for init of mysqld
 
220
  // Must be reset to handle error with Session's created for init of mysqld
450
221
  lex->current_select= 0;
451
222
  start_time=(time_t) 0;
452
223
  start_utime= 0L;
453
224
  utime_after_lock= 0L;
454
 
  current_linfo =  0;
455
 
  slave_thread = 0;
456
225
  memset(&variables, 0, sizeof(variables));
457
226
  thread_id= 0;
458
 
  one_shot_set= 0;
459
227
  file_id = 0;
460
228
  query_id= 0;
461
229
  warn_id= 0;
462
 
  db_charset= global_system_variables.collation_database;
463
230
  memset(ha_data, 0, sizeof(ha_data));
 
231
  replication_data= 0;
464
232
  mysys_var=0;
465
 
  binlog_evt_union.do_union= false;
466
 
  dbug_sentry=THD_SENTRY_MAGIC;
467
 
  net.vio=0;
 
233
  dbug_sentry=Session_SENTRY_MAGIC;
468
234
  client_capabilities= 0;                       // minimalistic client
469
 
  system_thread= NON_SYSTEM_THREAD;
470
 
  cleanup_done= abort_on_warning= no_warnings_for_error= 0;
 
235
  cleanup_done= abort_on_warning= no_warnings_for_error= false;
471
236
  peer_port= 0;                                 // For SHOW PROCESSLIST
472
 
  transaction.m_pending_rows_event= 0;
473
237
  transaction.on= 1;
 
238
  compression= 0;
474
239
  pthread_mutex_init(&LOCK_delete, MY_MUTEX_INIT_FAST);
475
240
 
476
241
  /* Variables with default values */
477
242
  proc_info="login";
478
 
  where= THD::DEFAULT_WHERE;
479
 
  server_id = ::server_id;
480
 
  slave_net = 0;
 
243
  where= Session::DEFAULT_WHERE;
481
244
  command=COM_CONNECT;
482
 
  *scramble= '\0';
483
245
 
484
246
  init();
485
247
  /* Initialize sub structures */
486
248
  init_sql_alloc(&warn_root, WARN_ALLOC_BLOCK_SIZE, WARN_ALLOC_PREALLOC_SIZE);
487
 
  user_connect=(USER_CONN *)0;
488
249
  hash_init(&user_vars, system_charset_info, USER_VARS_HASH_SIZE, 0, 0,
489
250
            (hash_get_key) get_var_key,
490
251
            (hash_free_key) free_user_var, 0);
491
252
 
492
 
  /* For user vars replication*/
493
 
  if (opt_bin_log)
494
 
    my_init_dynamic_array(&user_var_events,
495
 
                          sizeof(BINLOG_USER_VAR_EVENT *), 16, 16);
496
 
  else
497
 
    memset(&user_var_events, 0, sizeof(user_var_events));
498
 
 
499
253
  /* Protocol */
500
 
  protocol= &protocol_text;                     // Default protocol
501
 
  protocol_text.init(this);
 
254
  protocol= protocol_arg;
 
255
  protocol->setSession(this);
502
256
 
 
257
  const Query_id& local_query_id= Query_id::get_query_id();
503
258
  tablespace_op= false;
504
259
  tmp= sql_rnd();
505
 
  randominit(&rand, tmp + (ulong) &rand, tmp + (ulong) ::global_query_id);
 
260
  protocol->setRandom(tmp + (uint64_t) &protocol,
 
261
                      tmp + (uint64_t)local_query_id.value());
506
262
  substitute_null_with_insert_id = false;
507
263
  thr_lock_info_init(&lock_info); /* safety: will be reset after start */
508
264
  thr_lock_owner_init(&main_lock_id, &lock_info);
511
267
}
512
268
 
513
269
 
514
 
void THD::push_internal_handler(Internal_error_handler *handler)
 
270
void Session::push_internal_handler(Internal_error_handler *handler)
515
271
{
516
272
  /*
517
273
    TODO: The current implementation is limited to 1 handler at a time only.
518
 
    THD and sp_rcontext need to be modified to use a common handler stack.
 
274
    Session and sp_rcontext need to be modified to use a common handler stack.
519
275
  */
520
276
  assert(m_internal_handler == NULL);
521
277
  m_internal_handler= handler;
522
278
}
523
279
 
524
280
 
525
 
bool THD::handle_error(uint32_t sql_errno, const char *message,
 
281
bool Session::handle_error(uint32_t sql_errno, const char *message,
526
282
                       DRIZZLE_ERROR::enum_warning_level level)
527
283
{
528
284
  if (m_internal_handler)
534
290
}
535
291
 
536
292
 
537
 
void THD::pop_internal_handler()
 
293
void Session::pop_internal_handler()
538
294
{
539
295
  assert(m_internal_handler != NULL);
540
296
  m_internal_handler= NULL;
541
297
}
542
298
 
543
 
extern "C"
544
 
void *thd_alloc(DRIZZLE_THD thd, unsigned int size)
545
 
{
546
 
  return thd->alloc(size);
547
 
}
548
 
 
549
 
extern "C"
550
 
void *thd_calloc(DRIZZLE_THD thd, unsigned int size)
551
 
{
552
 
  return thd->calloc(size);
553
 
}
554
 
 
555
 
extern "C"
556
 
char *thd_strdup(DRIZZLE_THD thd, const char *str)
557
 
{
558
 
  return thd->strdup(str);
559
 
}
560
 
 
561
 
extern "C"
562
 
char *thd_strmake(DRIZZLE_THD thd, const char *str, unsigned int size)
563
 
{
564
 
  return thd->strmake(str, size);
565
 
}
566
 
 
567
 
extern "C"
568
 
LEX_STRING *thd_make_lex_string(THD *thd, LEX_STRING *lex_str,
569
 
                                const char *str, unsigned int size,
570
 
                                int allocate_lex_string)
571
 
{
572
 
  return thd->make_lex_string(lex_str, str, size,
573
 
                              (bool) allocate_lex_string);
574
 
}
575
 
 
576
 
extern "C"
577
 
void *thd_memdup(DRIZZLE_THD thd, const void* str, unsigned int size)
578
 
{
579
 
  return thd->memdup(str, size);
580
 
}
581
 
 
582
 
extern "C"
583
 
void thd_get_xid(const DRIZZLE_THD thd, DRIZZLE_XID *xid)
584
 
{
585
 
  *xid = *(DRIZZLE_XID *) &thd->transaction.xid_state.xid;
586
 
}
 
299
#if defined(__cplusplus)
 
300
extern "C" {
 
301
#endif
 
302
 
 
303
void *session_alloc(Session *session, unsigned int size)
 
304
{
 
305
  return session->alloc(size);
 
306
}
 
307
 
 
308
void *session_calloc(Session *session, unsigned int size)
 
309
{
 
310
  return session->calloc(size);
 
311
}
 
312
 
 
313
char *session_strdup(Session *session, const char *str)
 
314
{
 
315
  return session->strdup(str);
 
316
}
 
317
 
 
318
char *session_strmake(Session *session, const char *str, unsigned int size)
 
319
{
 
320
  return session->strmake(str, size);
 
321
}
 
322
 
 
323
void *session_memdup(Session *session, const void* str, unsigned int size)
 
324
{
 
325
  return session->memdup(str, size);
 
326
}
 
327
 
 
328
void session_get_xid(const Session *session, DRIZZLE_XID *xid)
 
329
{
 
330
  *xid = *(DRIZZLE_XID *) &session->transaction.xid_state.xid;
 
331
}
 
332
 
 
333
#if defined(__cplusplus)
 
334
}
 
335
#endif
587
336
 
588
337
/*
589
338
  Init common variables that has to be reset on start and on change_user
590
339
*/
591
340
 
592
 
void THD::init(void)
 
341
void Session::init(void)
593
342
{
594
 
  pthread_mutex_lock(&LOCK_global_system_variables);
595
 
  plugin_thdvar_init(this);
596
 
  variables.time_format= date_time_format_copy((THD*) 0,
597
 
                                               variables.time_format);
598
 
  variables.date_format= date_time_format_copy((THD*) 0,
599
 
                                               variables.date_format);
600
 
  variables.datetime_format= date_time_format_copy((THD*) 0,
601
 
                                                   variables.datetime_format);
 
343
  plugin_sessionvar_init(this);
602
344
  /*
603
345
    variables= global_system_variables above has reset
604
346
    variables.pseudo_thread_id to 0. We need to correct it here to
605
347
    avoid temporary tables replication failure.
606
348
  */
607
349
  variables.pseudo_thread_id= thread_id;
608
 
  pthread_mutex_unlock(&LOCK_global_system_variables);
609
350
  server_status= SERVER_STATUS_AUTOCOMMIT;
610
 
  options= thd_startup_options;
 
351
  options= session_startup_options;
611
352
 
612
353
  if (variables.max_join_size == HA_POS_ERROR)
613
354
    options |= OPTION_BIG_SELECTS;
616
357
 
617
358
  transaction.all.modified_non_trans_table= transaction.stmt.modified_non_trans_table= false;
618
359
  open_options=ha_open_options;
619
 
  update_lock_default= (variables.low_priority_updates ?
620
 
                        TL_WRITE_LOW_PRIORITY :
621
 
                        TL_WRITE);
 
360
  update_lock_default= TL_WRITE;
622
361
  session_tx_isolation= (enum_tx_isolation) variables.tx_isolation;
623
362
  warn_list.empty();
624
363
  memset(warn_count, 0, sizeof(warn_count));
625
364
  total_warn_count= 0;
626
365
  update_charset();
627
 
  reset_current_stmt_binlog_row_based();
628
366
  memset(&status_var, 0, sizeof(status_var));
629
367
}
630
368
 
631
369
 
632
370
/*
633
 
  Init THD for query processing.
 
371
  Init Session for query processing.
634
372
  This has to be called once before we call mysql_parse.
635
 
  See also comments in sql_class.h.
 
373
  See also comments in session.h.
636
374
*/
637
375
 
638
 
void THD::init_for_queries()
 
376
void Session::init_for_queries()
639
377
{
640
 
  set_time(); 
 
378
  set_time();
641
379
  ha_enable_transaction(this,true);
642
380
 
643
381
  reset_root_defaults(mem_root, variables.query_alloc_block_size,
646
384
                      variables.trans_alloc_block_size,
647
385
                      variables.trans_prealloc_size);
648
386
  transaction.xid_state.xid.null();
649
 
  transaction.xid_state.in_thd=1;
 
387
  transaction.xid_state.in_session=1;
650
388
}
651
389
 
652
390
 
653
391
/* Do operations that may take a long time */
654
392
 
655
 
void THD::cleanup(void)
 
393
void Session::cleanup(void)
656
394
{
657
 
  assert(cleanup_done == 0);
 
395
  assert(cleanup_done == false);
658
396
 
659
397
  killed= KILL_CONNECTION;
660
398
#ifdef ENABLE_WHEN_BINLOG_WILL_BE_ABLE_TO_PREPARE
672
410
    lock=locked_tables; locked_tables=0;
673
411
    close_thread_tables(this);
674
412
  }
675
 
  mysql_ha_cleanup(this);
676
 
  delete_dynamic(&user_var_events);
677
413
  hash_free(&user_vars);
678
 
  close_temporary_tables(this);
679
 
  free((char*) variables.time_format);
680
 
  free((char*) variables.date_format);
681
 
  free((char*) variables.datetime_format);
682
 
  
 
414
  close_temporary_tables();
 
415
 
683
416
  if (global_read_lock)
684
417
    unlock_global_read_lock(this);
685
418
 
686
 
  cleanup_done=1;
 
419
  cleanup_done= true;
687
420
  return;
688
421
}
689
422
 
690
 
THD::~THD()
 
423
Session::~Session()
691
424
{
692
 
  THD_CHECK_SENTRY(this);
693
 
  /* Ensure that no one is using THD */
694
 
  pthread_mutex_lock(&LOCK_delete);
695
 
  pthread_mutex_unlock(&LOCK_delete);
 
425
  Session_CHECK_SENTRY(this);
696
426
  add_to_status(&global_status_var, &status_var);
697
427
 
 
428
  if (protocol->isConnected())
 
429
  {
 
430
    if (global_system_variables.log_warnings)
 
431
        errmsg_printf(ERRMSG_LVL_WARN, ER(ER_FORCING_CLOSE),my_progname,
 
432
                      thread_id,
 
433
                      (security_ctx.user.c_str() ?
 
434
                       security_ctx.user.c_str() : ""));
 
435
    disconnect(0, false);
 
436
  }
 
437
 
698
438
  /* Close connection */
699
 
  if (net.vio)
700
 
  {
701
 
    net_close(&net);
702
 
    net_end(&net);
703
 
  }
704
 
  if (!cleanup_done)
 
439
  protocol->close();
 
440
  delete protocol;
 
441
 
 
442
  if (cleanup_done == false)
705
443
    cleanup();
706
444
 
707
445
  ha_close_connection(this);
708
 
  plugin_thdvar_cleanup(this);
 
446
  plugin_sessionvar_cleanup(this);
709
447
 
710
 
  main_security_ctx.destroy();
711
448
  if (db)
712
449
  {
713
450
    free(db);
716
453
  free_root(&warn_root,MYF(0));
717
454
  free_root(&transaction.mem_root,MYF(0));
718
455
  mysys_var=0;                                  // Safety (shouldn't be needed)
 
456
  dbug_sentry= Session_SENTRY_GONE;
 
457
 
 
458
  free_root(&main_mem_root, MYF(0));
 
459
  pthread_setspecific(THR_Session,  0);
 
460
 
 
461
 
 
462
  /* Ensure that no one is using Session */
 
463
  pthread_mutex_unlock(&LOCK_delete);
719
464
  pthread_mutex_destroy(&LOCK_delete);
720
 
  dbug_sentry= THD_SENTRY_GONE;
721
 
  if (rli_fake)
722
 
  {
723
 
    delete rli_fake;
724
 
    rli_fake= NULL;
725
 
  }
726
 
  
727
 
  free_root(&main_mem_root, MYF(0));
728
 
  return;
729
465
}
730
466
 
731
467
 
762
498
    to_var       add to this array
763
499
    from_var     from this array
764
500
    dec_var      minus this array
765
 
  
 
501
 
766
502
  NOTE
767
503
    This function assumes that all variables are long/ulong.
768
504
*/
780
516
}
781
517
 
782
518
 
783
 
void THD::awake(THD::killed_state state_to_set)
 
519
void Session::awake(Session::killed_state state_to_set)
784
520
{
785
 
  THD_CHECK_SENTRY(this);
786
 
  safe_mutex_assert_owner(&LOCK_delete); 
 
521
  Session_CHECK_SENTRY(this);
 
522
  safe_mutex_assert_owner(&LOCK_delete);
 
523
  Scheduler &thread_scheduler= get_thread_scheduler();
787
524
 
788
525
  killed= state_to_set;
789
 
  if (state_to_set != THD::KILL_QUERY)
 
526
  if (state_to_set != Session::KILL_QUERY)
790
527
  {
791
 
    thr_alarm_kill(thread_id);
792
 
    if (!slave_thread)
793
 
      thread_scheduler.post_kill_notification(this);
 
528
    thread_scheduler.post_kill_notification(this);
794
529
  }
795
530
  if (mysys_var)
796
531
  {
797
532
    pthread_mutex_lock(&mysys_var->mutex);
798
 
    if (!system_thread)         // Don't abort locks
799
 
      mysys_var->abort=1;
800
533
    /*
801
534
      This broadcast could be up in the air if the victim thread
802
535
      exits the cond in the time between read and broadcast, but that is
813
546
      current_cond and current_mutex are 0), then the victim will not get
814
547
      a signal and it may wait "forever" on the cond (until
815
548
      we issue a second KILL or the status it's waiting for happens).
816
 
      It's true that we have set its thd->killed but it may not
 
549
      It's true that we have set its session->killed but it may not
817
550
      see it immediately and so may have time to reach the cond_wait().
818
551
    */
819
552
    if (mysys_var->current_cond && mysys_var->current_mutex)
831
564
  Remember the location of thread info, the structure needed for
832
565
  sql_alloc() and the structure for the net buffer
833
566
*/
834
 
 
835
 
bool THD::store_globals()
 
567
bool Session::store_globals()
836
568
{
837
569
  /*
838
570
    Assert that thread_stack is initialized: it's necessary to be able
840
572
  */
841
573
  assert(thread_stack);
842
574
 
843
 
  if (my_pthread_setspecific_ptr(THR_THD,  this) ||
844
 
      my_pthread_setspecific_ptr(THR_MALLOC, &mem_root))
 
575
  if (pthread_setspecific(THR_Session,  this) ||
 
576
      pthread_setspecific(THR_Mem_root, &mem_root))
845
577
    return 1;
846
578
  mysys_var=my_thread_var;
 
579
 
847
580
  /*
848
581
    Let mysqld define the thread id (not mysys)
849
 
    This allows us to move THD to different threads if needed.
 
582
    This allows us to move Session to different threads if needed.
850
583
  */
851
584
  mysys_var->id= thread_id;
852
585
  real_id= pthread_self();                      // For debugging
853
586
 
854
587
  /*
855
 
    We have to call thr_lock_info_init() again here as THD may have been
 
588
    We have to call thr_lock_info_init() again here as Session may have been
856
589
    created in another thread
857
590
  */
858
591
  thr_lock_info_init(&lock_info);
859
592
  return 0;
860
593
}
861
594
 
 
595
void Session::prepareForQueries()
 
596
{
 
597
  if (variables.max_join_size == HA_POS_ERROR)
 
598
    options |= OPTION_BIG_SELECTS;
 
599
  if (client_capabilities & CLIENT_COMPRESS)
 
600
  {
 
601
    compression= 1;
 
602
    protocol->enableCompression();
 
603
  }
 
604
 
 
605
  version= refresh_version;
 
606
  set_proc_info(NULL);
 
607
  command= COM_SLEEP;
 
608
  set_time();
 
609
  init_for_queries();
 
610
}
 
611
 
 
612
bool Session::initGlobals()
 
613
{
 
614
  if (store_globals())
 
615
  {
 
616
    disconnect(ER_OUT_OF_RESOURCES, true);
 
617
    statistic_increment(aborted_connects, &LOCK_status);
 
618
    Scheduler &thread_scheduler= get_thread_scheduler();
 
619
    thread_scheduler.end_thread(this, 0);
 
620
    return false;
 
621
  }
 
622
  return true;
 
623
}
 
624
 
 
625
bool Session::authenticate()
 
626
{
 
627
  lex_start(this);
 
628
  if (protocol->authenticate())
 
629
    return true;
 
630
 
 
631
  statistic_increment(aborted_connects, &LOCK_status);
 
632
  return false;
 
633
}
 
634
 
 
635
bool Session::checkUser(const char *passwd, uint32_t passwd_len, const char *in_db)
 
636
{
 
637
  LEX_STRING db_str= { (char *) in_db, in_db ? strlen(in_db) : 0 };
 
638
  bool is_authenticated;
 
639
 
 
640
  /*
 
641
    Clear session->db as it points to something, that will be freed when
 
642
    connection is closed. We don't want to accidentally free a wrong
 
643
    pointer if connect failed. Also in case of 'CHANGE USER' failure,
 
644
    current database will be switched to 'no database selected'.
 
645
  */
 
646
  reset_db(NULL, 0);
 
647
 
 
648
  if (passwd_len != 0 && passwd_len != SCRAMBLE_LENGTH)
 
649
  {
 
650
    my_error(ER_HANDSHAKE_ERROR, MYF(0), security_ctx.ip.c_str());
 
651
    return false;
 
652
  }
 
653
 
 
654
  is_authenticated= authenticate_user(this, passwd);
 
655
 
 
656
  if (is_authenticated != true)
 
657
  {
 
658
    my_error(ER_ACCESS_DENIED_ERROR, MYF(0),
 
659
             security_ctx.user.c_str(),
 
660
             security_ctx.ip.c_str(),
 
661
             passwd_len ? ER(ER_YES) : ER(ER_NO));
 
662
 
 
663
    return false;
 
664
  }
 
665
 
 
666
  security_ctx.skip_grants();
 
667
 
 
668
  /* Change database if necessary */
 
669
  if (in_db && in_db[0])
 
670
  {
 
671
    if (mysql_change_db(this, &db_str, false))
 
672
    {
 
673
      /* mysql_change_db() has pushed the error message. */
 
674
      return false;
 
675
    }
 
676
  }
 
677
  my_ok();
 
678
  password= test(passwd_len);          // remember for error messages
 
679
 
 
680
  /* Ready to handle queries */
 
681
  return true;
 
682
}
 
683
 
 
684
bool Session::executeStatement()
 
685
{
 
686
  char *l_packet= 0;
 
687
  uint32_t packet_length;
 
688
 
 
689
  enum enum_server_command l_command;
 
690
 
 
691
  /*
 
692
    indicator of uninitialized lex => normal flow of errors handling
 
693
    (see my_message_sql)
 
694
  */
 
695
  lex->current_select= 0;
 
696
 
 
697
  if (protocol->readCommand(&l_packet, &packet_length) == false)
 
698
    return false;
 
699
 
 
700
  if (packet_length == 0)
 
701
    return true;
 
702
 
 
703
  l_command= (enum enum_server_command) (unsigned char) l_packet[0];
 
704
 
 
705
  if (command >= COM_END)
 
706
    command= COM_END;                           // Wrong command
 
707
 
 
708
  assert(packet_length);
 
709
  return ! dispatch_command(l_command, this, l_packet+1, (uint32_t) (packet_length-1));
 
710
}
 
711
 
 
712
bool Session::readAndStoreQuery(const char *in_packet, uint32_t in_packet_length)
 
713
{
 
714
  /* Remove garbage at start and end of query */
 
715
  while (in_packet_length > 0 && my_isspace(charset(), in_packet[0]))
 
716
  {
 
717
    in_packet++;
 
718
    in_packet_length--;
 
719
  }
 
720
  const char *pos= in_packet + in_packet_length; /* Point at end null */
 
721
  while (in_packet_length > 0 &&
 
722
         (pos[-1] == ';' || my_isspace(charset() ,pos[-1])))
 
723
  {
 
724
    pos--;
 
725
    in_packet_length--;
 
726
  }
 
727
 
 
728
  /* We must allocate some extra memory for the cached query string */
 
729
  query_length= 0; /* Extra safety: Avoid races */
 
730
  query= (char*) memdup_w_gap((unsigned char*) in_packet, in_packet_length, db_length + 1);
 
731
  if (! query)
 
732
    return false;
 
733
 
 
734
  query[in_packet_length]=0;
 
735
  query_length= in_packet_length;
 
736
 
 
737
  /* Reclaim some memory */
 
738
  packet.shrink(variables.net_buffer_length);
 
739
  convert_buffer.shrink(variables.net_buffer_length);
 
740
 
 
741
  return true;
 
742
}
 
743
 
 
744
bool Session::endTransaction(enum enum_mysql_completiontype completion)
 
745
{
 
746
  bool do_release= 0;
 
747
  bool result= true;
 
748
 
 
749
  if (transaction.xid_state.xa_state != XA_NOTR)
 
750
  {
 
751
    my_error(ER_XAER_RMFAIL, MYF(0), xa_state_names[transaction.xid_state.xa_state]);
 
752
    return false;
 
753
  }
 
754
  switch (completion) 
 
755
  {
 
756
    case COMMIT:
 
757
      /*
 
758
       * We don't use endActiveTransaction() here to ensure that this works
 
759
       * even if there is a problem with the OPTION_AUTO_COMMIT flag
 
760
       * (Which of course should never happen...)
 
761
       */
 
762
      server_status&= ~SERVER_STATUS_IN_TRANS;
 
763
      if (ha_commit(this))
 
764
        result= false;
 
765
      options&= ~(OPTION_BEGIN | OPTION_KEEP_LOG);
 
766
      transaction.all.modified_non_trans_table= false;
 
767
      break;
 
768
    case COMMIT_RELEASE:
 
769
      do_release= 1; /* fall through */
 
770
    case COMMIT_AND_CHAIN:
 
771
      result= endActiveTransaction();
 
772
      if (result == true && completion == COMMIT_AND_CHAIN)
 
773
        result= startTransaction();
 
774
      break;
 
775
    case ROLLBACK_RELEASE:
 
776
      do_release= 1; /* fall through */
 
777
    case ROLLBACK:
 
778
    case ROLLBACK_AND_CHAIN:
 
779
    {
 
780
      server_status&= ~SERVER_STATUS_IN_TRANS;
 
781
      if (ha_rollback(this))
 
782
        result= false;
 
783
      options&= ~(OPTION_BEGIN | OPTION_KEEP_LOG);
 
784
      transaction.all.modified_non_trans_table= false;
 
785
      if (result == true && (completion == ROLLBACK_AND_CHAIN))
 
786
        result= startTransaction();
 
787
      break;
 
788
    }
 
789
    default:
 
790
      my_error(ER_UNKNOWN_COM_ERROR, MYF(0));
 
791
      return false;
 
792
  }
 
793
 
 
794
  if (result == false)
 
795
    my_error(killed_errno(), MYF(0));
 
796
  else if ((result == true) && do_release)
 
797
    killed= Session::KILL_CONNECTION;
 
798
 
 
799
  return result;
 
800
}
 
801
 
 
802
bool Session::endActiveTransaction()
 
803
{
 
804
  bool result= true;
 
805
 
 
806
  if (transaction.xid_state.xa_state != XA_NOTR)
 
807
  {
 
808
    my_error(ER_XAER_RMFAIL, MYF(0), xa_state_names[transaction.xid_state.xa_state]);
 
809
    return false;
 
810
  }
 
811
  if (options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN | OPTION_TABLE_LOCK))
 
812
  {
 
813
    /* Safety if one did "drop table" on locked tables */
 
814
    if (! locked_tables)
 
815
      options&= ~OPTION_TABLE_LOCK;
 
816
    server_status&= ~SERVER_STATUS_IN_TRANS;
 
817
    if (ha_commit(this))
 
818
      result= false;
 
819
  }
 
820
  options&= ~(OPTION_BEGIN | OPTION_KEEP_LOG);
 
821
  transaction.all.modified_non_trans_table= false;
 
822
  return result;
 
823
}
 
824
 
 
825
bool Session::startTransaction()
 
826
{
 
827
  bool result= true;
 
828
 
 
829
  if (locked_tables)
 
830
  {
 
831
    lock= locked_tables;
 
832
    locked_tables= 0;                   // Will be automatically closed
 
833
    close_thread_tables(this);                  // Free tables
 
834
  }
 
835
  if (! endActiveTransaction())
 
836
    result= false;
 
837
  else
 
838
  {
 
839
    options|= OPTION_BEGIN;
 
840
    server_status|= SERVER_STATUS_IN_TRANS;
 
841
    if (lex->start_transaction_opt & DRIZZLE_START_TRANS_OPT_WITH_CONS_SNAPSHOT)
 
842
      if (ha_start_consistent_snapshot(this))
 
843
        result= false;
 
844
  }
 
845
  return result;
 
846
}
862
847
 
863
848
/*
864
849
  Cleanup after query.
865
850
 
866
851
  SYNOPSIS
867
 
    THD::cleanup_after_query()
 
852
    Session::cleanup_after_query()
868
853
 
869
854
  DESCRIPTION
870
855
    This function is used to reset thread data to its default state.
875
860
    different master threads may overwrite data of each other on
876
861
    slave.
877
862
*/
878
 
 
879
 
void THD::cleanup_after_query()
 
863
void Session::cleanup_after_query()
880
864
{
881
865
  /*
882
 
    Reset rand_used so that detection of calls to rand() will save random 
 
866
    Reset rand_used so that detection of calls to rand() will save random
883
867
    seeds if needed by the slave.
884
 
 
885
 
    Do not reset rand_used if inside a stored function or trigger because 
886
 
    only the call to these operations is logged. Thus only the calling 
887
 
    statement needs to detect rand() calls made by its substatements. These
888
 
    substatements must not set rand_used to 0 because it would remove the
889
 
    detection of rand() by the calling statement. 
890
868
  */
891
 
  if (!in_sub_stmt) /* stored functions and triggers are a special case */
892
869
  {
893
870
    /* Forget those values, for next binlogger: */
894
 
    stmt_depends_on_first_successful_insert_id_in_prev_stmt= 0;
895
871
    auto_inc_intervals_in_cur_stmt_for_binlog.empty();
896
 
    rand_used= 0;
897
872
  }
898
873
  if (first_successful_insert_id_in_cur_stmt > 0)
899
874
  {
900
875
    /* set what LAST_INSERT_ID() will return */
901
 
    first_successful_insert_id_in_prev_stmt= 
 
876
    first_successful_insert_id_in_prev_stmt=
902
877
      first_successful_insert_id_in_cur_stmt;
903
878
    first_successful_insert_id_in_cur_stmt= 0;
904
879
    substitute_null_with_insert_id= true;
907
882
  /* Free Items that were created during this execution */
908
883
  free_items();
909
884
  /* Reset where. */
910
 
  where= THD::DEFAULT_WHERE;
 
885
  where= Session::DEFAULT_WHERE;
911
886
}
912
887
 
913
888
 
921
896
                              instead of using lex_str value
922
897
  @return  NULL on failure, or pointer to the LEX_STRING object
923
898
*/
924
 
LEX_STRING *THD::make_lex_string(LEX_STRING *lex_str,
 
899
LEX_STRING *Session::make_lex_string(LEX_STRING *lex_str,
925
900
                                 const char* str, uint32_t length,
926
901
                                 bool allocate_lex_string)
927
902
{
955
930
        In this case to->str will point to 0 and to->length will be 0.
956
931
*/
957
932
 
958
 
bool THD::convert_string(LEX_STRING *to, const CHARSET_INFO * const to_cs,
 
933
bool Session::convert_string(LEX_STRING *to, const CHARSET_INFO * const to_cs,
959
934
                         const char *from, uint32_t from_length,
960
935
                         const CHARSET_INFO * const from_cs)
961
936
{
977
952
  Convert string from source character set to target character set inplace.
978
953
 
979
954
  SYNOPSIS
980
 
    THD::convert_string
 
955
    Session::convert_string
981
956
 
982
957
  DESCRIPTION
983
 
    Convert string using convert_buffer - buffer for character set 
 
958
    Convert string using convert_buffer - buffer for character set
984
959
    conversion shared between all protocols.
985
960
 
986
961
  RETURN
988
963
   !0   out of memory
989
964
*/
990
965
 
991
 
bool THD::convert_string(String *s, const CHARSET_INFO * const from_cs,
 
966
bool Session::convert_string(String *s, const CHARSET_INFO * const from_cs,
992
967
                         const CHARSET_INFO * const to_cs)
993
968
{
994
969
  uint32_t dummy_errors;
1009
984
  Update some cache variables when character set changes
1010
985
*/
1011
986
 
1012
 
void THD::update_charset()
 
987
void Session::update_charset()
1013
988
{
1014
989
  uint32_t not_used;
1015
990
  charset_is_system_charset= !String::needs_conversion(0,charset(),
1016
991
                                                       system_charset_info,
1017
992
                                                       &not_used);
1018
 
  charset_is_collation_connection= 
1019
 
    !String::needs_conversion(0,charset(),variables.collation_connection,
 
993
  charset_is_collation_connection=
 
994
    !String::needs_conversion(0,charset(),variables.getCollation(),
1020
995
                              &not_used);
1021
 
  charset_is_character_set_filesystem= 
 
996
  charset_is_character_set_filesystem=
1022
997
    !String::needs_conversion(0, charset(),
1023
998
                              variables.character_set_filesystem, &not_used);
1024
999
}
1039
1014
 
1040
1015
/* add table to list of changed in transaction tables */
1041
1016
 
1042
 
void THD::add_changed_table(Table *table)
 
1017
void Session::add_changed_table(Table *table)
1043
1018
{
1044
1019
  assert((options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN)) &&
1045
1020
              table->file->has_transactions());
1049
1024
}
1050
1025
 
1051
1026
 
1052
 
void THD::add_changed_table(const char *key, long key_length)
 
1027
void Session::add_changed_table(const char *key, long key_length)
1053
1028
{
1054
1029
  CHANGED_TableList **prev_changed = &transaction.changed_tables;
1055
1030
  CHANGED_TableList *curr = transaction.changed_tables;
1081
1056
}
1082
1057
 
1083
1058
 
1084
 
CHANGED_TableList* THD::changed_table_dup(const char *key, long key_length)
 
1059
CHANGED_TableList* Session::changed_table_dup(const char *key, long key_length)
1085
1060
{
1086
 
  CHANGED_TableList* new_table = 
 
1061
  CHANGED_TableList* new_table =
1087
1062
    (CHANGED_TableList*) trans_alloc(ALIGN_SIZE(sizeof(CHANGED_TableList))+
1088
1063
                                      key_length + 1);
1089
1064
  if (!new_table)
1102
1077
}
1103
1078
 
1104
1079
 
1105
 
int THD::send_explain_fields(select_result *result)
 
1080
int Session::send_explain_fields(select_result *result)
1106
1081
{
1107
1082
  List<Item> field_list;
1108
1083
  Item *item;
1141
1116
                              Protocol::SEND_NUM_ROWS | Protocol::SEND_EOF));
1142
1117
}
1143
1118
 
1144
 
 
1145
 
struct Item_change_record: public ilink
1146
 
{
1147
 
  Item **place;
1148
 
  Item *old_value;
1149
 
  /* Placement new was hidden by `new' in ilink (TODO: check): */
1150
 
  static void *operator new(size_t size __attribute__((unused)),
1151
 
                            void *mem)
1152
 
    { return mem; }
1153
 
  static void operator delete(void *ptr __attribute__((unused)),
1154
 
                              size_t size __attribute__((unused)))
1155
 
    {}
1156
 
  static void operator delete(void *ptr __attribute__((unused)),
1157
 
                              void *mem __attribute__((unused)))
1158
 
    { /* never called */ }
1159
 
};
1160
 
 
1161
 
 
1162
 
/*
1163
 
  Register an item tree tree transformation, performed by the query
1164
 
  optimizer. We need a pointer to runtime_memroot because it may be !=
1165
 
  thd->mem_root (this may no longer be a true statement)
1166
 
*/
1167
 
 
1168
 
void THD::nocheck_register_item_tree_change(Item **place, Item *old_value,
1169
 
                                            MEM_ROOT *runtime_memroot)
1170
 
{
1171
 
  Item_change_record *change;
1172
 
  /*
1173
 
    Now we use one node per change, which adds some memory overhead,
1174
 
    but still is rather fast as we use alloc_root for allocations.
1175
 
    A list of item tree changes of an average query should be short.
1176
 
  */
1177
 
  void *change_mem= alloc_root(runtime_memroot, sizeof(*change));
1178
 
  if (change_mem == 0)
1179
 
  {
1180
 
    /*
1181
 
      OOM, thd->fatal_error() is called by the error handler of the
1182
 
      memroot. Just return.
1183
 
    */
1184
 
    return;
1185
 
  }
1186
 
  change= new (change_mem) Item_change_record;
1187
 
  change->place= place;
1188
 
  change->old_value= old_value;
1189
 
  change_list.append(change);
1190
 
}
1191
 
 
1192
 
 
1193
 
void THD::rollback_item_tree_changes()
1194
 
{
1195
 
  I_List_iterator<Item_change_record> it(change_list);
1196
 
  Item_change_record *change;
1197
 
 
1198
 
  while ((change= it++))
1199
 
    *change->place= change->old_value;
1200
 
  /* We can forget about changes memory: it's allocated in runtime memroot */
1201
 
  change_list.empty();
1202
 
  return;
1203
 
}
1204
 
 
1205
 
 
1206
 
/*****************************************************************************
1207
 
** Functions to provide a interface to select results
1208
 
*****************************************************************************/
1209
 
 
1210
 
select_result::select_result()
1211
 
{
1212
 
  thd=current_thd;
1213
 
}
1214
 
 
1215
 
void select_result::send_error(uint32_t errcode,const char *err)
1216
 
{
1217
 
  my_message(errcode, err, MYF(0));
1218
 
}
1219
 
 
1220
 
 
1221
 
void select_result::cleanup()
1222
 
{
1223
 
  /* do nothing */
1224
 
}
1225
 
 
1226
 
bool select_result::check_simple_select() const
1227
 
{
1228
 
  my_error(ER_SP_BAD_CURSOR_QUERY, MYF(0));
1229
 
  return true;
1230
 
}
1231
 
 
1232
 
 
1233
 
static String default_line_term("\n",default_charset_info);
1234
 
static String default_escaped("\\",default_charset_info);
1235
 
static String default_field_term("\t",default_charset_info);
1236
 
 
1237
 
sql_exchange::sql_exchange(char *name, bool flag,
1238
 
                           enum enum_filetype filetype_arg)
1239
 
  :file_name(name), opt_enclosed(0), dumpfile(flag), skip_lines(0)
1240
 
{
1241
 
  filetype= filetype_arg;
1242
 
  field_term= &default_field_term;
1243
 
  enclosed=   line_start= &my_empty_string;
1244
 
  line_term=  &default_line_term;
1245
 
  escaped=    &default_escaped;
1246
 
  cs= NULL;
1247
 
}
1248
 
 
1249
 
bool select_send::send_fields(List<Item> &list, uint32_t flags)
1250
 
{
1251
 
  bool res;
1252
 
  if (!(res= thd->protocol->send_fields(&list, flags)))
1253
 
    is_result_set_started= 1;
1254
 
  return res;
1255
 
}
1256
 
 
1257
 
void select_send::abort()
1258
 
{
1259
 
  return;
1260
 
}
1261
 
 
1262
 
 
1263
 
/** 
1264
 
  Cleanup an instance of this class for re-use
1265
 
  at next execution of a prepared statement/
1266
 
  stored procedure statement.
1267
 
*/
1268
 
 
1269
 
void select_send::cleanup()
1270
 
{
1271
 
  is_result_set_started= false;
1272
 
}
1273
 
 
1274
 
/* Send data to client. Returns 0 if ok */
1275
 
 
1276
 
bool select_send::send_data(List<Item> &items)
1277
 
{
1278
 
  if (unit->offset_limit_cnt)
1279
 
  {                                             // using limit offset,count
1280
 
    unit->offset_limit_cnt--;
1281
 
    return 0;
1282
 
  }
1283
 
 
1284
 
  /*
1285
 
    We may be passing the control from mysqld to the client: release the
1286
 
    InnoDB adaptive hash S-latch to avoid thread deadlocks if it was reserved
1287
 
    by thd
1288
 
  */
1289
 
  ha_release_temporary_latches(thd);
1290
 
 
1291
 
  List_iterator_fast<Item> li(items);
1292
 
  Protocol *protocol= thd->protocol;
1293
 
  char buff[MAX_FIELD_WIDTH];
1294
 
  String buffer(buff, sizeof(buff), &my_charset_bin);
1295
 
 
1296
 
  protocol->prepare_for_resend();
1297
 
  Item *item;
1298
 
  while ((item=li++))
1299
 
  {
1300
 
    if (item->send(protocol, &buffer))
1301
 
    {
1302
 
      protocol->free();                         // Free used buffer
1303
 
      my_message(ER_OUT_OF_RESOURCES, ER(ER_OUT_OF_RESOURCES), MYF(0));
1304
 
      break;
1305
 
    }
1306
 
  }
1307
 
  thd->sent_row_count++;
1308
 
  if (thd->is_error())
1309
 
  {
1310
 
    protocol->remove_last_row();
1311
 
    return(1);
1312
 
  }
1313
 
  if (thd->vio_ok())
1314
 
    return(protocol->write());
1315
 
  return(0);
1316
 
}
1317
 
 
1318
 
bool select_send::send_eof()
1319
 
{
1320
 
  /* 
1321
 
    We may be passing the control from mysqld to the client: release the
1322
 
    InnoDB adaptive hash S-latch to avoid thread deadlocks if it was reserved
1323
 
    by thd 
1324
 
  */
1325
 
  ha_release_temporary_latches(thd);
1326
 
 
1327
 
  /* Unlock tables before sending packet to gain some speed */
1328
 
  if (thd->lock)
1329
 
  {
1330
 
    mysql_unlock_tables(thd, thd->lock);
1331
 
    thd->lock=0;
1332
 
  }
1333
 
  ::my_eof(thd);
1334
 
  is_result_set_started= 0;
1335
 
  return false;
1336
 
}
1337
 
 
1338
 
 
1339
1119
/************************************************************************
1340
1120
  Handling writing to file
1341
1121
************************************************************************/
1365
1145
      function, SELECT INTO has to have an own SQLCOM.
1366
1146
      TODO: split from SQLCOM_SELECT
1367
1147
    */
1368
 
    ::my_ok(thd,row_count);
 
1148
    session->my_ok(row_count);
1369
1149
  }
1370
1150
  file= -1;
1371
1151
  return error;
1402
1182
 
1403
1183
select_export::~select_export()
1404
1184
{
1405
 
  thd->sent_row_count=row_count;
 
1185
  session->sent_row_count=row_count;
1406
1186
}
1407
1187
 
1408
1188
 
1411
1191
 
1412
1192
  SYNOPSIS
1413
1193
    create_file()
1414
 
    thd                 Thread handle
 
1194
    session                     Thread handle
1415
1195
    path                File name
1416
1196
    exchange            Excange class
1417
1197
    cache               IO cache
1422
1202
*/
1423
1203
 
1424
1204
 
1425
 
static File create_file(THD *thd, char *path, sql_exchange *exchange,
1426
 
                        IO_CACHE *cache)
 
1205
static File create_file(Session *session, char *path, file_exchange *exchange, IO_CACHE *cache)
1427
1206
{
1428
1207
  File file;
1429
1208
  uint32_t option= MY_UNPACK_FILENAME | MY_RELATIVE_PATH;
1434
1213
 
1435
1214
  if (!dirname_length(exchange->file_name))
1436
1215
  {
1437
 
    strxnmov(path, FN_REFLEN-1, mysql_real_data_home, thd->db ? thd->db : "",
1438
 
             NULL);
 
1216
    strcpy(path, drizzle_real_data_home);
 
1217
    if (session->db)
 
1218
      strncat(path, session->db, FN_REFLEN-strlen(drizzle_real_data_home)-1);
1439
1219
    (void) fn_format(path, exchange->file_name, path, "", option);
1440
1220
  }
1441
1221
  else
1442
 
    (void) fn_format(path, exchange->file_name, mysql_real_data_home, "", option);
 
1222
    (void) fn_format(path, exchange->file_name, drizzle_real_data_home, "", option);
1443
1223
 
1444
1224
  if (opt_secure_file_priv &&
1445
1225
      strncmp(opt_secure_file_priv, path, strlen(opt_secure_file_priv)))
1465
1245
  if (init_io_cache(cache, file, 0L, WRITE_CACHE, 0L, 1, MYF(MY_WME)))
1466
1246
  {
1467
1247
    my_close(file, MYF(0));
1468
 
    my_delete(path, MYF(0));  // Delete file on error, it was just created 
 
1248
    my_delete(path, MYF(0));  // Delete file on error, it was just created
1469
1249
    return -1;
1470
1250
  }
1471
1251
  return file;
1473
1253
 
1474
1254
 
1475
1255
int
1476
 
select_export::prepare(List<Item> &list, SELECT_LEX_UNIT *u)
 
1256
select_export::prepare(List<Item> &list, Select_Lex_Unit *u)
1477
1257
{
1478
1258
  bool blob_flag=0;
1479
1259
  bool string_results= false, non_string_results= false;
1480
1260
  unit= u;
1481
 
  if ((uint) strlen(exchange->file_name) + NAME_LEN >= FN_REFLEN)
1482
 
    strmake(path,exchange->file_name,FN_REFLEN-1);
 
1261
  if ((uint32_t) strlen(exchange->file_name) + NAME_LEN >= FN_REFLEN)
 
1262
    strncpy(path,exchange->file_name,FN_REFLEN-1);
1483
1263
 
1484
 
  if ((file= create_file(thd, path, exchange, &cache)) < 0)
1485
 
    return 1;
1486
1264
  /* Check if there is any blobs in data */
1487
1265
  {
1488
1266
    List_iterator_fast<Item> li(list);
1524
1302
      (exchange->opt_enclosed && non_string_results &&
1525
1303
       field_term_length && strchr(NUMERIC_CHARS, field_term_char)))
1526
1304
  {
1527
 
    push_warning(thd, DRIZZLE_ERROR::WARN_LEVEL_WARN,
1528
 
                 ER_AMBIGUOUS_FIELD_TERM, ER(ER_AMBIGUOUS_FIELD_TERM));
1529
 
    is_ambiguous_field_term= true;
 
1305
    my_error(ER_AMBIGUOUS_FIELD_TERM, MYF(0));
 
1306
    return 1;
1530
1307
  }
1531
 
  else
1532
 
    is_ambiguous_field_term= false;
 
1308
 
 
1309
  if ((file= create_file(session, path, exchange, &cache)) < 0)
 
1310
    return 1;
1533
1311
 
1534
1312
  return 0;
1535
1313
}
1603
1381
      {
1604
1382
        char *pos, *start, *end;
1605
1383
        const CHARSET_INFO * const res_charset= res->charset();
1606
 
        const CHARSET_INFO * const character_set_client= thd->variables.
1607
 
                                                            character_set_client;
 
1384
        const CHARSET_INFO * const character_set_client= default_charset_info;
 
1385
 
1608
1386
        bool check_second_byte= (res_charset == &my_charset_bin) &&
1609
1387
                                 character_set_client->
1610
1388
                                 escape_with_backslash_is_dangerous;
1611
1389
        assert(character_set_client->mbmaxlen == 2 ||
1612
 
                    !character_set_client->escape_with_backslash_is_dangerous);
 
1390
               !character_set_client->escape_with_backslash_is_dangerous);
1613
1391
        for (start=pos=(char*) res->ptr(),end=pos+used_length ;
1614
1392
             pos != end ;
1615
1393
             pos++)
1631
1409
            for the clients with character sets big5, cp932, gbk and sjis,
1632
1410
            which can have the escape character (0x5C "\" by default)
1633
1411
            as the second byte of a multi-byte sequence.
1634
 
            
 
1412
 
1635
1413
            If
1636
1414
            - pos[0] is a valid multi-byte head (e.g 0xEE) and
1637
1415
            - pos[1] is 0x00, which will be escaped as "\0",
1638
 
            
 
1416
 
1639
1417
            then we'll get "0xEE + 0x5C + 0x30" in the output file.
1640
 
            
 
1418
 
1641
1419
            If this file is later loaded using this sequence of commands:
1642
 
            
 
1420
 
1643
1421
            mysql> create table t1 (a varchar(128)) character set big5;
1644
1422
            mysql> LOAD DATA INFILE 'dump.txt' INTO Table t1;
1645
 
            
 
1423
 
1646
1424
            then 0x5C will be misinterpreted as the second byte
1647
1425
            of a multi-byte character "0xEE + 0x5C", instead of
1648
1426
            escape character for 0x00.
1649
 
            
 
1427
 
1650
1428
            To avoid this confusion, we'll escape the multi-byte
1651
1429
            head character too, so the sequence "0xEE + 0x00" will be
1652
1430
            dumped as "0x5C + 0xEE + 0x5C + 0x30".
1653
 
            
 
1431
 
1654
1432
            Note, in the condition below we only check if
1655
1433
            mbcharlen is equal to 2, because there are no
1656
1434
            character sets with mbmaxlen longer than 2
1675
1453
                          is_ambiguous_field_sep) ?
1676
1454
                          field_sep_char : escape_char;
1677
1455
            tmp_buff[1]= *pos ? *pos : '0';
1678
 
            if (my_b_write(&cache,(unsigned char*) start,(uint) (pos-start)) ||
 
1456
            if (my_b_write(&cache,(unsigned char*) start,(uint32_t) (pos-start)) ||
1679
1457
                my_b_write(&cache,(unsigned char*) tmp_buff,2))
1680
1458
              goto err;
1681
1459
            start=pos+1;
1682
1460
          }
1683
1461
        }
1684
 
        if (my_b_write(&cache,(unsigned char*) start,(uint) (pos-start)))
 
1462
        if (my_b_write(&cache,(unsigned char*) start,(uint32_t) (pos-start)))
1685
1463
          goto err;
1686
1464
      }
1687
1465
      else if (my_b_write(&cache,(unsigned char*) res->ptr(),used_length))
1735
1513
 
1736
1514
 
1737
1515
int
1738
 
select_dump::prepare(List<Item> &list __attribute__((unused)),
1739
 
                     SELECT_LEX_UNIT *u)
 
1516
select_dump::prepare(List<Item> &, Select_Lex_Unit *u)
1740
1517
{
1741
1518
  unit= u;
1742
 
  return (int) ((file= create_file(thd, path, exchange, &cache)) < 0);
 
1519
  return (int) ((file= create_file(session, path, exchange, &cache)) < 0);
1743
1520
}
1744
1521
 
1745
1522
 
1756
1533
    unit->offset_limit_cnt--;
1757
1534
    return(0);
1758
1535
  }
1759
 
  if (row_count++ > 1) 
 
1536
  if (row_count++ > 1)
1760
1537
  {
1761
1538
    my_message(ER_TOO_MANY_ROWS, ER(ER_TOO_MANY_ROWS), MYF(0));
1762
1539
    goto err;
1919
1696
     sortcmp(val1, val2, cache->collation.collation) < 0);
1920
1697
}
1921
1698
 
1922
 
bool select_exists_subselect::send_data(List<Item> &items __attribute__((unused)))
 
1699
bool select_exists_subselect::send_data(List<Item> &)
1923
1700
{
1924
1701
  Item_exists_subselect *it= (Item_exists_subselect *)item;
1925
1702
  if (unit->offset_limit_cnt)
1933
1710
}
1934
1711
 
1935
1712
 
1936
 
/***************************************************************************
1937
 
  Dump of select to variables
1938
 
***************************************************************************/
1939
 
 
1940
 
int select_dumpvar::prepare(List<Item> &list, SELECT_LEX_UNIT *u)
1941
 
{
1942
 
  unit= u;
1943
 
  
1944
 
  if (var_list.elements != list.elements)
1945
 
  {
1946
 
    my_message(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT,
1947
 
               ER(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT), MYF(0));
1948
 
    return 1;
1949
 
  }               
1950
 
  return 0;
1951
 
}
1952
 
 
1953
 
 
1954
 
bool select_dumpvar::check_simple_select() const
1955
 
{
1956
 
  my_error(ER_SP_BAD_CURSOR_SELECT, MYF(0));
1957
 
  return true;
1958
 
}
1959
 
 
1960
 
 
1961
 
void select_dumpvar::cleanup()
1962
 
{
1963
 
  row_count= 0;
1964
 
}
1965
 
 
1966
 
 
1967
 
void Query_arena::free_items()
1968
 
{
1969
 
  Item *next;
1970
 
  /* This works because items are allocated with sql_alloc() */
1971
 
  for (; free_list; free_list= next)
1972
 
  {
1973
 
    next= free_list->next;
1974
 
    free_list->delete_self();
1975
 
  }
1976
 
  /* Postcondition: free_list is 0 */
1977
 
  return;
1978
 
}
1979
 
 
1980
 
 
1981
1713
/*
1982
1714
  Statement functions
1983
1715
*/
1999
1731
  Don't free mem_root, as mem_root is freed in the end of dispatch_command
2000
1732
  (once for any command).
2001
1733
*/
2002
 
void THD::end_statement()
 
1734
void Session::end_statement()
2003
1735
{
2004
1736
  /* Cleanup SQL processing state to reuse this statement in next query. */
2005
1737
  lex_end(lex);
2006
1738
}
2007
1739
 
2008
1740
 
2009
 
bool THD::copy_db_to(char **p_db, size_t *p_db_length)
 
1741
bool Session::copy_db_to(char **p_db, size_t *p_db_length)
2010
1742
{
2011
1743
  if (db == NULL)
2012
1744
  {
2019
1751
}
2020
1752
 
2021
1753
 
2022
 
bool select_dumpvar::send_data(List<Item> &items)
2023
 
{
2024
 
  List_iterator_fast<my_var> var_li(var_list);
2025
 
  List_iterator<Item> it(items);
2026
 
  Item *item;
2027
 
  my_var *mv;
2028
 
 
2029
 
  if (unit->offset_limit_cnt)
2030
 
  {                                             // using limit offset,count
2031
 
    unit->offset_limit_cnt--;
2032
 
    return(0);
2033
 
  }
2034
 
  if (row_count++) 
2035
 
  {
2036
 
    my_message(ER_TOO_MANY_ROWS, ER(ER_TOO_MANY_ROWS), MYF(0));
2037
 
    return(1);
2038
 
  }
2039
 
  while ((mv= var_li++) && (item= it++))
2040
 
  {
2041
 
    if (mv->local == 0)
2042
 
    {
2043
 
      Item_func_set_user_var *suv= new Item_func_set_user_var(mv->s, item);
2044
 
      suv->fix_fields(thd, 0);
2045
 
      suv->check(0);
2046
 
      suv->update();
2047
 
    }
2048
 
  }
2049
 
  return(thd->is_error());
2050
 
}
2051
 
 
2052
 
bool select_dumpvar::send_eof()
2053
 
{
2054
 
  if (! row_count)
2055
 
    push_warning(thd, DRIZZLE_ERROR::WARN_LEVEL_WARN,
2056
 
                 ER_SP_FETCH_NO_DATA, ER(ER_SP_FETCH_NO_DATA));
2057
 
  /*
2058
 
    In order to remember the value of affected rows for ROW_COUNT()
2059
 
    function, SELECT INTO has to have an own SQLCOM.
2060
 
    TODO: split from SQLCOM_SELECT
2061
 
  */
2062
 
  ::my_ok(thd,row_count);
2063
 
  return 0;
2064
 
}
2065
 
 
2066
1754
/****************************************************************************
2067
 
  TMP_TABLE_PARAM
 
1755
  Tmp_Table_Param
2068
1756
****************************************************************************/
2069
1757
 
2070
 
void TMP_TABLE_PARAM::init()
 
1758
void Tmp_Table_Param::init()
2071
1759
{
2072
1760
  field_count= sum_func_count= func_count= hidden_field_count= 0;
2073
1761
  group_parts= group_length= group_null_parts= 0;
2078
1766
  return;
2079
1767
}
2080
1768
 
2081
 
 
2082
 
void thd_increment_bytes_sent(ulong length)
2083
 
{
2084
 
  THD *thd=current_thd;
2085
 
  if (likely(thd != 0))
2086
 
  { /* current_thd==0 when close_connection() calls net_send_error() */
2087
 
    thd->status_var.bytes_sent+= length;
2088
 
  }
2089
 
}
2090
 
 
2091
 
 
2092
 
void thd_increment_bytes_received(ulong length)
2093
 
{
2094
 
  current_thd->status_var.bytes_received+= length;
2095
 
}
2096
 
 
2097
 
 
2098
 
void thd_increment_net_big_packet_count(ulong length)
2099
 
{
2100
 
  current_thd->status_var.net_big_packet_count+= length;
2101
 
}
2102
 
 
2103
 
void THD::send_kill_message() const
 
1769
void Tmp_Table_Param::cleanup(void)
 
1770
{
 
1771
  /* Fix for Intel compiler */
 
1772
  if (copy_field)
 
1773
  {
 
1774
    delete [] copy_field;
 
1775
    save_copy_field= copy_field= 0;
 
1776
  }
 
1777
}
 
1778
 
 
1779
 
 
1780
void session_increment_bytes_sent(ulong length)
 
1781
{
 
1782
  Session *session=current_session;
 
1783
  if (likely(session != 0))
 
1784
  { /* current_session==0 when disconnect() calls net_send_error() */
 
1785
    session->status_var.bytes_sent+= length;
 
1786
  }
 
1787
}
 
1788
 
 
1789
 
 
1790
void session_increment_bytes_received(ulong length)
 
1791
{
 
1792
  current_session->status_var.bytes_received+= length;
 
1793
}
 
1794
 
 
1795
 
 
1796
void session_increment_net_big_packet_count(ulong length)
 
1797
{
 
1798
  current_session->status_var.net_big_packet_count+= length;
 
1799
}
 
1800
 
 
1801
void Session::send_kill_message() const
2104
1802
{
2105
1803
  int err= killed_errno();
2106
1804
  if (err)
2107
1805
    my_message(err, ER(err), MYF(0));
2108
1806
}
2109
1807
 
2110
 
void THD::set_status_var_init()
 
1808
void Session::set_status_var_init()
2111
1809
{
2112
1810
  memset(&status_var, 0, sizeof(status_var));
2113
1811
}
2114
1812
 
2115
 
 
2116
 
void Security_context::init()
2117
 
{
2118
 
  user= ip= 0;
2119
 
}
2120
 
 
2121
 
 
2122
 
void Security_context::destroy()
2123
 
{
2124
 
  // If not pointer to constant
2125
 
  if (user)
2126
 
  {
2127
 
    free(user);
2128
 
    user= NULL;
2129
 
  }
2130
 
  if (ip)
2131
 
  {
2132
 
    free(ip);
2133
 
    ip= NULL;
2134
 
  }
2135
 
}
2136
 
 
2137
 
 
2138
1813
void Security_context::skip_grants()
2139
1814
{
2140
1815
  /* privileges for the user are unknown everything is allowed */
2149
1824
  access to mysql.proc table to find definitions of stored routines.
2150
1825
****************************************************************************/
2151
1826
 
2152
 
void THD::reset_n_backup_open_tables_state(Open_tables_state *backup)
 
1827
void Session::reset_n_backup_open_tables_state(Open_tables_state *backup)
2153
1828
{
2154
1829
  backup->set_open_tables_state(this);
2155
1830
  reset_open_tables_state();
2158
1833
}
2159
1834
 
2160
1835
 
2161
 
void THD::restore_backup_open_tables_state(Open_tables_state *backup)
 
1836
void Session::restore_backup_open_tables_state(Open_tables_state *backup)
2162
1837
{
2163
1838
  /*
2164
1839
    Before we will throw away current open tables state we want
2171
1846
  return;
2172
1847
}
2173
1848
 
 
1849
 
 
1850
bool Session::set_db(const char *new_db, size_t new_db_len)
 
1851
{
 
1852
  /* Do not reallocate memory if current chunk is big enough. */
 
1853
  if (db && new_db && db_length >= new_db_len)
 
1854
    memcpy(db, new_db, new_db_len+1);
 
1855
  else
 
1856
  {
 
1857
    if (db)
 
1858
      free(db);
 
1859
    if (new_db)
 
1860
    {
 
1861
      db= (char *)malloc(new_db_len + 1);
 
1862
      if (db != NULL)
 
1863
      {
 
1864
        memcpy(db, new_db, new_db_len);
 
1865
        db[new_db_len]= 0;
 
1866
      }
 
1867
    }
 
1868
    else
 
1869
      db= NULL;
 
1870
  }
 
1871
  db_length= db ? new_db_len : 0;
 
1872
  return new_db && !db;
 
1873
}
 
1874
 
 
1875
 
2174
1876
/**
2175
1877
  Check the killed state of a user thread
2176
 
  @param thd  user thread
 
1878
  @param session  user thread
2177
1879
  @retval 0 the user thread is active
2178
1880
  @retval 1 the user thread has been killed
2179
1881
*/
2180
 
extern "C" int thd_killed(const DRIZZLE_THD thd)
 
1882
extern "C" int session_killed(const Session *session)
2181
1883
{
2182
 
  return(thd->killed);
 
1884
  return(session->killed);
2183
1885
}
2184
1886
 
2185
1887
/**
2186
1888
  Return the thread id of a user thread
2187
 
  @param thd user thread
 
1889
  @param session user thread
2188
1890
  @return thread id
2189
1891
*/
2190
 
extern "C" unsigned long thd_get_thread_id(const DRIZZLE_THD thd)
2191
 
{
2192
 
  return((unsigned long)thd->thread_id);
2193
 
}
2194
 
 
2195
 
 
2196
 
#ifdef INNODB_COMPATIBILITY_HOOKS
2197
 
extern "C" const struct charset_info_st *thd_charset(DRIZZLE_THD thd)
2198
 
{
2199
 
  return(thd->charset());
2200
 
}
2201
 
 
2202
 
extern "C" char **thd_query(DRIZZLE_THD thd)
2203
 
{
2204
 
  return(&thd->query);
2205
 
}
2206
 
 
2207
 
extern "C" int thd_slave_thread(const DRIZZLE_THD thd)
2208
 
{
2209
 
  return(thd->slave_thread);
2210
 
}
2211
 
 
2212
 
extern "C" int thd_non_transactional_update(const DRIZZLE_THD thd)
2213
 
{
2214
 
  return(thd->transaction.all.modified_non_trans_table);
2215
 
}
2216
 
 
2217
 
extern "C" int thd_binlog_format(const DRIZZLE_THD thd)
2218
 
{
2219
 
  return (int) thd->variables.binlog_format;
2220
 
}
2221
 
 
2222
 
extern "C" void thd_mark_transaction_to_rollback(DRIZZLE_THD thd, bool all)
2223
 
{
2224
 
  mark_transaction_to_rollback(thd, all);
2225
 
}
2226
 
#endif // INNODB_COMPATIBILITY_HOOKS */
 
1892
extern "C" unsigned long session_get_thread_id(const Session *session)
 
1893
{
 
1894
  return((unsigned long)session->thread_id);
 
1895
}
 
1896
 
 
1897
 
 
1898
extern "C"
 
1899
LEX_STRING *session_make_lex_string(Session *session, LEX_STRING *lex_str,
 
1900
                                const char *str, unsigned int size,
 
1901
                                int allocate_lex_string)
 
1902
{
 
1903
  return session->make_lex_string(lex_str, str, size,
 
1904
                              (bool) allocate_lex_string);
 
1905
}
 
1906
 
 
1907
extern "C" const struct charset_info_st *session_charset(Session *session)
 
1908
{
 
1909
  return(session->charset());
 
1910
}
 
1911
 
 
1912
extern "C" char **session_query(Session *session)
 
1913
{
 
1914
  return(&session->query);
 
1915
}
 
1916
 
 
1917
extern "C" int session_non_transactional_update(const Session *session)
 
1918
{
 
1919
  return(session->transaction.all.modified_non_trans_table);
 
1920
}
 
1921
 
 
1922
extern "C" void session_mark_transaction_to_rollback(Session *session, bool all)
 
1923
{
 
1924
  mark_transaction_to_rollback(session, all);
 
1925
}
2227
1926
 
2228
1927
 
2229
1928
/**
2230
1929
  Mark transaction to rollback and mark error as fatal to a sub-statement.
2231
1930
 
2232
 
  @param  thd   Thread handle
 
1931
  @param  session   Thread handle
2233
1932
  @param  all   true <=> rollback main transaction.
2234
1933
*/
2235
1934
 
2236
 
void mark_transaction_to_rollback(THD *thd, bool all)
2237
 
{
2238
 
  if (thd)
2239
 
  {
2240
 
    thd->is_fatal_sub_stmt_error= true;
2241
 
    thd->transaction_rollback_request= all;
2242
 
  }
2243
 
}
2244
 
/***************************************************************************
2245
 
  Handling of XA id cacheing
2246
 
***************************************************************************/
2247
 
 
2248
 
pthread_mutex_t LOCK_xid_cache;
2249
 
HASH xid_cache;
2250
 
 
2251
 
extern "C" unsigned char *xid_get_hash_key(const unsigned char *, size_t *, bool);
2252
 
extern "C" void xid_free_hash(void *);
2253
 
 
2254
 
unsigned char *xid_get_hash_key(const unsigned char *ptr, size_t *length,
2255
 
                        bool not_used __attribute__((unused)))
2256
 
{
2257
 
  *length=((XID_STATE*)ptr)->xid.key_length();
2258
 
  return ((XID_STATE*)ptr)->xid.key();
2259
 
}
2260
 
 
2261
 
void xid_free_hash(void *ptr)
2262
 
{
2263
 
  if (!((XID_STATE*)ptr)->in_thd)
2264
 
    free((unsigned char*)ptr);
2265
 
}
2266
 
 
2267
 
bool xid_cache_init()
2268
 
{
2269
 
  pthread_mutex_init(&LOCK_xid_cache, MY_MUTEX_INIT_FAST);
2270
 
  return hash_init(&xid_cache, &my_charset_bin, 100, 0, 0,
2271
 
                   xid_get_hash_key, xid_free_hash, 0) != 0;
2272
 
}
2273
 
 
2274
 
void xid_cache_free()
2275
 
{
2276
 
  if (hash_inited(&xid_cache))
2277
 
  {
2278
 
    hash_free(&xid_cache);
2279
 
    pthread_mutex_destroy(&LOCK_xid_cache);
2280
 
  }
2281
 
}
2282
 
 
2283
 
XID_STATE *xid_cache_search(XID *xid)
2284
 
{
2285
 
  pthread_mutex_lock(&LOCK_xid_cache);
2286
 
  XID_STATE *res=(XID_STATE *)hash_search(&xid_cache, xid->key(), xid->key_length());
2287
 
  pthread_mutex_unlock(&LOCK_xid_cache);
2288
 
  return res;
2289
 
}
2290
 
 
2291
 
 
2292
 
bool xid_cache_insert(XID *xid, enum xa_states xa_state)
2293
 
{
2294
 
  XID_STATE *xs;
2295
 
  bool res;
2296
 
  pthread_mutex_lock(&LOCK_xid_cache);
2297
 
  if (hash_search(&xid_cache, xid->key(), xid->key_length()))
2298
 
    res=0;
2299
 
  else if (!(xs=(XID_STATE *)my_malloc(sizeof(*xs), MYF(MY_WME))))
2300
 
    res=1;
2301
 
  else
2302
 
  {
2303
 
    xs->xa_state=xa_state;
2304
 
    xs->xid.set(xid);
2305
 
    xs->in_thd=0;
2306
 
    res=my_hash_insert(&xid_cache, (unsigned char*)xs);
2307
 
  }
2308
 
  pthread_mutex_unlock(&LOCK_xid_cache);
2309
 
  return res;
2310
 
}
2311
 
 
2312
 
 
2313
 
bool xid_cache_insert(XID_STATE *xid_state)
2314
 
{
2315
 
  pthread_mutex_lock(&LOCK_xid_cache);
2316
 
  assert(hash_search(&xid_cache, xid_state->xid.key(),
2317
 
                          xid_state->xid.key_length())==0);
2318
 
  bool res=my_hash_insert(&xid_cache, (unsigned char*)xid_state);
2319
 
  pthread_mutex_unlock(&LOCK_xid_cache);
2320
 
  return res;
2321
 
}
2322
 
 
2323
 
 
2324
 
void xid_cache_delete(XID_STATE *xid_state)
2325
 
{
2326
 
  pthread_mutex_lock(&LOCK_xid_cache);
2327
 
  hash_delete(&xid_cache, (unsigned char *)xid_state);
2328
 
  pthread_mutex_unlock(&LOCK_xid_cache);
2329
 
}
2330
 
 
2331
 
/*
2332
 
  Implementation of interface to write rows to the binary log through the
2333
 
  thread.  The thread is responsible for writing the rows it has
2334
 
  inserted/updated/deleted.
2335
 
*/
2336
 
 
2337
 
 
2338
 
/*
2339
 
  Template member function for ensuring that there is an rows log
2340
 
  event of the apropriate type before proceeding.
2341
 
 
2342
 
  PRE CONDITION:
2343
 
    - Events of type 'RowEventT' have the type code 'type_code'.
2344
 
    
2345
 
  POST CONDITION:
2346
 
    If a non-NULL pointer is returned, the pending event for thread 'thd' will
2347
 
    be an event of type 'RowEventT' (which have the type code 'type_code')
2348
 
    will either empty or have enough space to hold 'needed' bytes.  In
2349
 
    addition, the columns bitmap will be correct for the row, meaning that
2350
 
    the pending event will be flushed if the columns in the event differ from
2351
 
    the columns suppled to the function.
2352
 
 
2353
 
  RETURNS
2354
 
    If no error, a non-NULL pending event (either one which already existed or
2355
 
    the newly created one).
2356
 
    If error, NULL.
2357
 
 */
2358
 
 
2359
 
template <class RowsEventT> Rows_log_event* 
2360
 
THD::binlog_prepare_pending_rows_event(Table* table, uint32_t serv_id,
2361
 
                                       size_t needed,
2362
 
                                       bool is_transactional,
2363
 
                                       RowsEventT *hint __attribute__((unused)))
2364
 
{
2365
 
  /* Pre-conditions */
2366
 
  assert(table->s->table_map_id != UINT32_MAX);
2367
 
 
2368
 
  /* Fetch the type code for the RowsEventT template parameter */
2369
 
  int const type_code= RowsEventT::TYPE_CODE;
2370
 
 
2371
 
  /*
2372
 
    There is no good place to set up the transactional data, so we
2373
 
    have to do it here.
2374
 
  */
2375
 
  if (binlog_setup_trx_data())
2376
 
    return(NULL);
2377
 
 
2378
 
  Rows_log_event* pending= binlog_get_pending_rows_event();
2379
 
 
2380
 
  if (unlikely(pending && !pending->is_valid()))
2381
 
    return(NULL);
2382
 
 
2383
 
  /*
2384
 
    Check if the current event is non-NULL and a write-rows
2385
 
    event. Also check if the table provided is mapped: if it is not,
2386
 
    then we have switched to writing to a new table.
2387
 
    If there is no pending event, we need to create one. If there is a pending
2388
 
    event, but it's not about the same table id, or not of the same type
2389
 
    (between Write, Update and Delete), or not the same affected columns, or
2390
 
    going to be too big, flush this event to disk and create a new pending
2391
 
    event.
2392
 
 
2393
 
    The last test is necessary for the Cluster injector to work
2394
 
    correctly. The reason is that the Cluster can inject two write
2395
 
    rows with different column bitmaps if there is an insert followed
2396
 
    by an update in the same transaction, and these are grouped into a
2397
 
    single epoch/transaction when fed to the injector.
2398
 
 
2399
 
    TODO: Fix the code so that the last test can be removed.
2400
 
  */
2401
 
  if (!pending ||
2402
 
      pending->server_id != serv_id || 
2403
 
      pending->get_table_id() != table->s->table_map_id ||
2404
 
      pending->get_type_code() != type_code || 
2405
 
      pending->get_data_size() + needed > opt_binlog_rows_event_max_size ||
2406
 
      !bitmap_cmp(pending->get_cols(), table->write_set))
2407
 
    {
2408
 
    /* Create a new RowsEventT... */
2409
 
    Rows_log_event* const
2410
 
        ev= new RowsEventT(this, table, table->s->table_map_id,
2411
 
                           is_transactional);
2412
 
    if (unlikely(!ev))
2413
 
      return(NULL);
2414
 
    ev->server_id= serv_id; // I don't like this, it's too easy to forget.
2415
 
    /*
2416
 
      flush the pending event and replace it with the newly created
2417
 
      event...
2418
 
    */
2419
 
    if (unlikely(mysql_bin_log.flush_and_set_pending_rows_event(this, ev)))
2420
 
    {
2421
 
      delete ev;
2422
 
      return(NULL);
2423
 
    }
2424
 
 
2425
 
    return(ev);               /* This is the new pending event */
2426
 
  }
2427
 
  return(pending);        /* This is the current pending event */
2428
 
}
2429
 
 
2430
 
#ifdef HAVE_EXPLICIT_TEMPLATE_INSTANTIATION
2431
 
/*
2432
 
  Instantiate the versions we need, we have -fno-implicit-template as
2433
 
  compiling option.
2434
 
*/
2435
 
template Rows_log_event*
2436
 
THD::binlog_prepare_pending_rows_event(Table*, uint32_t, size_t, bool,
2437
 
                                       Write_rows_log_event*);
2438
 
 
2439
 
template Rows_log_event*
2440
 
THD::binlog_prepare_pending_rows_event(Table*, uint32_t, size_t, bool,
2441
 
                                       Delete_rows_log_event *);
2442
 
 
2443
 
template Rows_log_event* 
2444
 
THD::binlog_prepare_pending_rows_event(Table*, uint32_t, size_t, bool,
2445
 
                                       Update_rows_log_event *);
2446
 
#endif
2447
 
 
2448
 
namespace {
2449
 
  /**
2450
 
     Class to handle temporary allocation of memory for row data.
2451
 
 
2452
 
     The responsibilities of the class is to provide memory for
2453
 
     packing one or two rows of packed data (depending on what
2454
 
     constructor is called).
2455
 
 
2456
 
     In order to make the allocation more efficient for "simple" rows,
2457
 
     i.e., rows that do not contain any blobs, a pointer to the
2458
 
     allocated memory is of memory is stored in the table structure
2459
 
     for simple rows.  If memory for a table containing a blob field
2460
 
     is requested, only memory for that is allocated, and subsequently
2461
 
     released when the object is destroyed.
2462
 
 
2463
 
   */
2464
 
  class Row_data_memory {
2465
 
  public:
2466
 
    /**
2467
 
      Build an object to keep track of a block-local piece of memory
2468
 
      for storing a row of data.
2469
 
 
2470
 
      @param table
2471
 
      Table where the pre-allocated memory is stored.
2472
 
 
2473
 
      @param length
2474
 
      Length of data that is needed, if the record contain blobs.
2475
 
     */
2476
 
    Row_data_memory(Table *table, size_t const len1)
2477
 
      : m_memory(0)
2478
 
    {
2479
 
      m_alloc_checked= false;
2480
 
      allocate_memory(table, len1);
2481
 
      m_ptr[0]= has_memory() ? m_memory : 0;
2482
 
      m_ptr[1]= 0;
2483
 
    }
2484
 
 
2485
 
    Row_data_memory(Table *table, size_t const len1, size_t const len2)
2486
 
      : m_memory(0)
2487
 
    {
2488
 
      m_alloc_checked= false;
2489
 
      allocate_memory(table, len1 + len2);
2490
 
      m_ptr[0]= has_memory() ? m_memory        : 0;
2491
 
      m_ptr[1]= has_memory() ? m_memory + len1 : 0;
2492
 
    }
2493
 
 
2494
 
    ~Row_data_memory()
2495
 
    {
2496
 
      if (m_memory != 0 && m_release_memory_on_destruction)
2497
 
        free((unsigned char*) m_memory);
2498
 
    }
2499
 
 
2500
 
    /**
2501
 
       Is there memory allocated?
2502
 
 
2503
 
       @retval true There is memory allocated
2504
 
       @retval false Memory allocation failed
2505
 
     */
2506
 
    bool has_memory() const {
2507
 
      m_alloc_checked= true;
2508
 
      return m_memory != 0;
2509
 
    }
2510
 
 
2511
 
    unsigned char *slot(uint32_t s)
2512
 
    {
2513
 
      assert(s < sizeof(m_ptr)/sizeof(*m_ptr));
2514
 
      assert(m_ptr[s] != 0);
2515
 
      assert(m_alloc_checked == true);
2516
 
      return m_ptr[s];
2517
 
    }
2518
 
 
2519
 
  private:
2520
 
    void allocate_memory(Table *const table, size_t const total_length)
2521
 
    {
2522
 
      if (table->s->blob_fields == 0)
2523
 
      {
2524
 
        /*
2525
 
          The maximum length of a packed record is less than this
2526
 
          length. We use this value instead of the supplied length
2527
 
          when allocating memory for records, since we don't know how
2528
 
          the memory will be used in future allocations.
2529
 
 
2530
 
          Since table->s->reclength is for unpacked records, we have
2531
 
          to add two bytes for each field, which can potentially be
2532
 
          added to hold the length of a packed field.
2533
 
        */
2534
 
        size_t const maxlen= table->s->reclength + 2 * table->s->fields;
2535
 
 
2536
 
        /*
2537
 
          Allocate memory for two records if memory hasn't been
2538
 
          allocated. We allocate memory for two records so that it can
2539
 
          be used when processing update rows as well.
2540
 
        */
2541
 
        if (table->write_row_record == 0)
2542
 
          table->write_row_record=
2543
 
            (unsigned char *) alloc_root(&table->mem_root, 2 * maxlen);
2544
 
        m_memory= table->write_row_record;
2545
 
        m_release_memory_on_destruction= false;
2546
 
      }
2547
 
      else
2548
 
      {
2549
 
        m_memory= (unsigned char *) my_malloc(total_length, MYF(MY_WME));
2550
 
        m_release_memory_on_destruction= true;
2551
 
      }
2552
 
    }
2553
 
 
2554
 
    mutable bool m_alloc_checked;
2555
 
    bool m_release_memory_on_destruction;
2556
 
    unsigned char *m_memory;
2557
 
    unsigned char *m_ptr[2];
2558
 
  };
2559
 
}
2560
 
 
2561
 
 
2562
 
int THD::binlog_write_row(Table* table, bool is_trans, 
2563
 
                          unsigned char const *record) 
2564
 
{ 
2565
 
  assert(current_stmt_binlog_row_based && mysql_bin_log.is_open());
2566
 
 
2567
 
  /*
2568
 
    Pack records into format for transfer. We are allocating more
2569
 
    memory than needed, but that doesn't matter.
2570
 
  */
2571
 
  Row_data_memory memory(table, table->max_row_length(record));
2572
 
  if (!memory.has_memory())
2573
 
    return HA_ERR_OUT_OF_MEM;
2574
 
 
2575
 
  unsigned char *row_data= memory.slot(0);
2576
 
 
2577
 
  size_t const len= pack_row(table, table->write_set, row_data, record);
2578
 
 
2579
 
  Rows_log_event* const ev=
2580
 
    binlog_prepare_pending_rows_event(table, server_id, len, is_trans,
2581
 
                                      static_cast<Write_rows_log_event*>(0));
2582
 
 
2583
 
  if (unlikely(ev == 0))
2584
 
    return HA_ERR_OUT_OF_MEM;
2585
 
 
2586
 
  return ev->add_row_data(row_data, len);
2587
 
}
2588
 
 
2589
 
int THD::binlog_update_row(Table* table, bool is_trans,
2590
 
                           const unsigned char *before_record,
2591
 
                           const unsigned char *after_record)
2592
 
{ 
2593
 
  assert(current_stmt_binlog_row_based && mysql_bin_log.is_open());
2594
 
 
2595
 
  size_t const before_maxlen = table->max_row_length(before_record);
2596
 
  size_t const after_maxlen  = table->max_row_length(after_record);
2597
 
 
2598
 
  Row_data_memory row_data(table, before_maxlen, after_maxlen);
2599
 
  if (!row_data.has_memory())
2600
 
    return HA_ERR_OUT_OF_MEM;
2601
 
 
2602
 
  unsigned char *before_row= row_data.slot(0);
2603
 
  unsigned char *after_row= row_data.slot(1);
2604
 
 
2605
 
  size_t const before_size= pack_row(table, table->read_set, before_row,
2606
 
                                        before_record);
2607
 
  size_t const after_size= pack_row(table, table->write_set, after_row,
2608
 
                                       after_record);
2609
 
 
2610
 
  Rows_log_event* const ev=
2611
 
    binlog_prepare_pending_rows_event(table, server_id,
2612
 
                                      before_size + after_size, is_trans,
2613
 
                                      static_cast<Update_rows_log_event*>(0));
2614
 
 
2615
 
  if (unlikely(ev == 0))
2616
 
    return HA_ERR_OUT_OF_MEM;
2617
 
 
2618
 
  return
2619
 
    ev->add_row_data(before_row, before_size) ||
2620
 
    ev->add_row_data(after_row, after_size);
2621
 
}
2622
 
 
2623
 
int THD::binlog_delete_row(Table* table, bool is_trans, 
2624
 
                           unsigned char const *record)
2625
 
{ 
2626
 
  assert(current_stmt_binlog_row_based && mysql_bin_log.is_open());
2627
 
 
2628
 
  /* 
2629
 
     Pack records into format for transfer. We are allocating more
2630
 
     memory than needed, but that doesn't matter.
2631
 
  */
2632
 
  Row_data_memory memory(table, table->max_row_length(record));
2633
 
  if (unlikely(!memory.has_memory()))
2634
 
    return HA_ERR_OUT_OF_MEM;
2635
 
 
2636
 
  unsigned char *row_data= memory.slot(0);
2637
 
 
2638
 
  size_t const len= pack_row(table, table->read_set, row_data, record);
2639
 
 
2640
 
  Rows_log_event* const ev=
2641
 
    binlog_prepare_pending_rows_event(table, server_id, len, is_trans,
2642
 
                                      static_cast<Delete_rows_log_event*>(0));
2643
 
 
2644
 
  if (unlikely(ev == 0))
2645
 
    return HA_ERR_OUT_OF_MEM;
2646
 
 
2647
 
  return ev->add_row_data(row_data, len);
2648
 
}
2649
 
 
2650
 
 
2651
 
int THD::binlog_flush_pending_rows_event(bool stmt_end)
2652
 
{
2653
 
  /*
2654
 
    We shall flush the pending event even if we are not in row-based
2655
 
    mode: it might be the case that we left row-based mode before
2656
 
    flushing anything (e.g., if we have explicitly locked tables).
2657
 
   */
2658
 
  if (!mysql_bin_log.is_open())
2659
 
    return(0);
2660
 
 
2661
 
  /*
2662
 
    Mark the event as the last event of a statement if the stmt_end
2663
 
    flag is set.
2664
 
  */
2665
 
  int error= 0;
2666
 
  if (Rows_log_event *pending= binlog_get_pending_rows_event())
2667
 
  {
2668
 
    if (stmt_end)
2669
 
    {
2670
 
      pending->set_flags(Rows_log_event::STMT_END_F);
2671
 
      pending->flags|= LOG_EVENT_UPDATE_TABLE_MAP_VERSION_F;
2672
 
      binlog_table_maps= 0;
2673
 
    }
2674
 
 
2675
 
    error= mysql_bin_log.flush_and_set_pending_rows_event(this, 0);
2676
 
  }
2677
 
 
2678
 
  return(error);
2679
 
}
2680
 
 
2681
 
 
2682
 
/*
2683
 
  Member function that will log query, either row-based or
2684
 
  statement-based depending on the value of the 'current_stmt_binlog_row_based'
2685
 
  the value of the 'qtype' flag.
2686
 
 
2687
 
  This function should be called after the all calls to ha_*_row()
2688
 
  functions have been issued, but before tables are unlocked and
2689
 
  closed.
2690
 
 
2691
 
  OBSERVE
2692
 
    There shall be no writes to any system table after calling
2693
 
    binlog_query(), so these writes has to be moved to before the call
2694
 
    of binlog_query() for correct functioning.
2695
 
 
2696
 
    This is necessesary not only for RBR, but the master might crash
2697
 
    after binlogging the query but before changing the system tables.
2698
 
    This means that the slave and the master are not in the same state
2699
 
    (after the master has restarted), so therefore we have to
2700
 
    eliminate this problem.
2701
 
 
2702
 
  RETURN VALUE
2703
 
    Error code, or 0 if no error.
2704
 
*/
2705
 
int THD::binlog_query(THD::enum_binlog_query_type qtype, char const *query_arg,
2706
 
                      ulong query_len, bool is_trans, bool suppress_use,
2707
 
                      THD::killed_state killed_status_arg)
2708
 
{
2709
 
  assert(query_arg && mysql_bin_log.is_open());
2710
 
 
2711
 
  if (int error= binlog_flush_pending_rows_event(true))
2712
 
    return(error);
2713
 
 
2714
 
  /*
2715
 
    If we are in statement mode and trying to log an unsafe statement,
2716
 
    we should print a warning.
2717
 
  */
2718
 
  if (lex->is_stmt_unsafe() &&
2719
 
      variables.binlog_format == BINLOG_FORMAT_STMT)
2720
 
  {
2721
 
    assert(this->query != NULL);
2722
 
    push_warning(this, DRIZZLE_ERROR::WARN_LEVEL_WARN,
2723
 
                 ER_BINLOG_UNSAFE_STATEMENT,
2724
 
                 ER(ER_BINLOG_UNSAFE_STATEMENT));
2725
 
    if (!(binlog_flags & BINLOG_FLAG_UNSAFE_STMT_PRINTED))
2726
 
    {
2727
 
      char warn_buf[DRIZZLE_ERRMSG_SIZE];
2728
 
      snprintf(warn_buf, DRIZZLE_ERRMSG_SIZE, "%s Statement: %s",
2729
 
               ER(ER_BINLOG_UNSAFE_STATEMENT), this->query);
2730
 
      sql_print_warning(warn_buf);
2731
 
      binlog_flags|= BINLOG_FLAG_UNSAFE_STMT_PRINTED;
2732
 
    }
2733
 
  }
2734
 
 
2735
 
  switch (qtype) {
2736
 
  case THD::ROW_QUERY_TYPE:
2737
 
    if (current_stmt_binlog_row_based)
2738
 
      return(0);
2739
 
    /* Otherwise, we fall through */
2740
 
  case THD::DRIZZLE_QUERY_TYPE:
2741
 
    /*
2742
 
      Using this query type is a conveniece hack, since we have been
2743
 
      moving back and forth between using RBR for replication of
2744
 
      system tables and not using it.
2745
 
 
2746
 
      Make sure to change in check_table_binlog_row_based() according
2747
 
      to how you treat this.
2748
 
    */
2749
 
  case THD::STMT_QUERY_TYPE:
2750
 
    /*
2751
 
      The DRIZZLE_LOG::write() function will set the STMT_END_F flag and
2752
 
      flush the pending rows event if necessary.
2753
 
     */
2754
 
    {
2755
 
      Query_log_event qinfo(this, query_arg, query_len, is_trans, suppress_use,
2756
 
                            killed_status_arg);
2757
 
      qinfo.flags|= LOG_EVENT_UPDATE_TABLE_MAP_VERSION_F;
2758
 
      /*
2759
 
        Binlog table maps will be irrelevant after a Query_log_event
2760
 
        (they are just removed on the slave side) so after the query
2761
 
        log event is written to the binary log, we pretend that no
2762
 
        table maps were written.
2763
 
       */
2764
 
      int error= mysql_bin_log.write(&qinfo);
2765
 
      binlog_table_maps= 0;
2766
 
      return(error);
2767
 
    }
2768
 
    break;
2769
 
 
2770
 
  case THD::QUERY_TYPE_COUNT:
2771
 
  default:
2772
 
    assert(0 <= qtype && qtype < QUERY_TYPE_COUNT);
2773
 
  }
2774
 
  return(0);
2775
 
}
2776
 
 
2777
 
bool Discrete_intervals_list::append(uint64_t start, uint64_t val,
2778
 
                                 uint64_t incr)
2779
 
{
2780
 
  /* first, see if this can be merged with previous */
2781
 
  if ((head == NULL) || tail->merge_if_contiguous(start, val, incr))
2782
 
  {
2783
 
    /* it cannot, so need to add a new interval */
2784
 
    Discrete_interval *new_interval= new Discrete_interval(start, val, incr);
2785
 
    return(append(new_interval));
2786
 
  }
2787
 
  return(0);
2788
 
}
2789
 
 
2790
 
bool Discrete_intervals_list::append(Discrete_interval *new_interval)
2791
 
{
2792
 
  if (unlikely(new_interval == NULL))
2793
 
    return(1);
2794
 
  if (head == NULL)
2795
 
    head= current= new_interval;
2796
 
  else
2797
 
    tail->next= new_interval;
2798
 
  tail= new_interval;
2799
 
  elements++;
2800
 
  return(0);
 
1935
void mark_transaction_to_rollback(Session *session, bool all)
 
1936
{
 
1937
  if (session)
 
1938
  {
 
1939
    session->is_fatal_sub_stmt_error= true;
 
1940
    session->transaction_rollback_request= all;
 
1941
  }
 
1942
}
 
1943
 
 
1944
void Session::disconnect(uint32_t errcode, bool should_lock)
 
1945
{
 
1946
  /* Allow any plugins to cleanup their session variables */
 
1947
  plugin_sessionvar_cleanup(this);
 
1948
 
 
1949
  /* If necessary, log any aborted or unauthorized connections */
 
1950
  if (killed || protocol->wasAborted())
 
1951
    statistic_increment(aborted_threads, &LOCK_status);
 
1952
 
 
1953
  if (protocol->wasAborted())
 
1954
  {
 
1955
    if (! killed && variables.log_warnings > 1)
 
1956
    {
 
1957
      Security_context *sctx= &security_ctx;
 
1958
 
 
1959
      errmsg_printf(ERRMSG_LVL_WARN, ER(ER_NEW_ABORTING_CONNECTION)
 
1960
                  , thread_id
 
1961
                  , (db ? db : "unconnected")
 
1962
                  , sctx->user.empty() == false ? sctx->user.c_str() : "unauthenticated"
 
1963
                  , sctx->ip.c_str()
 
1964
                  , (main_da.is_error() ? main_da.message() : ER(ER_UNKNOWN_ERROR)));
 
1965
    }
 
1966
  }
 
1967
 
 
1968
  /* Close out our connection to the client */
 
1969
  if (should_lock)
 
1970
    (void) pthread_mutex_lock(&LOCK_thread_count);
 
1971
  killed= Session::KILL_CONNECTION;
 
1972
  if (protocol->isConnected())
 
1973
  {
 
1974
    if (errcode)
 
1975
    {
 
1976
      /*my_error(errcode, ER(errcode));*/
 
1977
      protocol->sendError(errcode, ER(errcode)); /* purecov: inspected */
 
1978
    }
 
1979
    protocol->close();
 
1980
  }
 
1981
  if (should_lock)
 
1982
    (void) pthread_mutex_unlock(&LOCK_thread_count);
 
1983
}
 
1984
 
 
1985
/**
 
1986
 Reset Session part responsible for command processing state.
 
1987
 
 
1988
   This needs to be called before execution of every statement
 
1989
   (prepared or conventional).
 
1990
   It is not called by substatements of routines.
 
1991
 
 
1992
  @todo
 
1993
   Make it a method of Session and align its name with the rest of
 
1994
   reset/end/start/init methods.
 
1995
  @todo
 
1996
   Call it after we use Session for queries, not before.
 
1997
*/
 
1998
 
 
1999
void Session::reset_for_next_command()
 
2000
{
 
2001
  free_list= 0;
 
2002
  select_number= 1;
 
2003
  /*
 
2004
    Those two lines below are theoretically unneeded as
 
2005
    Session::cleanup_after_query() should take care of this already.
 
2006
  */
 
2007
  auto_inc_intervals_in_cur_stmt_for_binlog.empty();
 
2008
 
 
2009
  is_fatal_error= 0;
 
2010
  server_status&= ~ (SERVER_MORE_RESULTS_EXISTS |
 
2011
                          SERVER_QUERY_NO_INDEX_USED |
 
2012
                          SERVER_QUERY_NO_GOOD_INDEX_USED);
 
2013
  /*
 
2014
    If in autocommit mode and not in a transaction, reset
 
2015
    OPTION_STATUS_NO_TRANS_UPDATE | OPTION_KEEP_LOG to not get warnings
 
2016
    in ha_rollback_trans() about some tables couldn't be rolled back.
 
2017
  */
 
2018
  if (!(options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN)))
 
2019
  {
 
2020
    options&= ~OPTION_KEEP_LOG;
 
2021
    transaction.all.modified_non_trans_table= false;
 
2022
  }
 
2023
 
 
2024
  clear_error();
 
2025
  main_da.reset_diagnostics_area();
 
2026
  total_warn_count=0;                   // Warnings for this query
 
2027
  sent_row_count= examined_row_count= 0;
 
2028
 
 
2029
  return;
 
2030
}
 
2031
 
 
2032
/*
 
2033
  Close all temporary tables created by 'CREATE TEMPORARY TABLE' for thread
 
2034
  creates one DROP TEMPORARY Table binlog event for each pseudo-thread
 
2035
*/
 
2036
 
 
2037
void Session::close_temporary_tables()
 
2038
{
 
2039
  Table *table;
 
2040
  Table *tmp_next;
 
2041
 
 
2042
  if (!temporary_tables)
 
2043
    return;
 
2044
 
 
2045
  for (table= temporary_tables; table; table= tmp_next)
 
2046
  {
 
2047
    tmp_next= table->next;
 
2048
    close_temporary(table, 1, 1);
 
2049
  }
 
2050
  temporary_tables= 0;
 
2051
 
 
2052
  return;
 
2053
}
 
2054
 
 
2055
 
 
2056
/** Clear most status variables. */
 
2057
extern time_t flush_status_time;
 
2058
extern uint32_t max_used_connections;
 
2059
 
 
2060
void Session::refresh_status()
 
2061
{
 
2062
  pthread_mutex_lock(&LOCK_status);
 
2063
 
 
2064
  /* Add thread's status variabes to global status */
 
2065
  add_to_status(&global_status_var, &status_var);
 
2066
 
 
2067
  /* Reset thread's status variables */
 
2068
  memset(&status_var, 0, sizeof(status_var));
 
2069
 
 
2070
  /* Reset some global variables */
 
2071
  reset_status_vars();
 
2072
 
 
2073
  /* Reset the counters of all key caches (default and named). */
 
2074
  process_key_caches(reset_key_cache_counters);
 
2075
  flush_status_time= time((time_t*) 0);
 
2076
  max_used_connections= 1; /* We set it to one, because we know we exist */
 
2077
  pthread_mutex_unlock(&LOCK_status);
 
2078
}
 
2079
 
 
2080
#define extra_size sizeof(double)
 
2081
 
 
2082
user_var_entry *Session::getVariable(LEX_STRING &name, bool create_if_not_exists)
 
2083
{
 
2084
  user_var_entry *entry= NULL;
 
2085
 
 
2086
  assert(name.length == strlen (name.str));
 
2087
  entry= (user_var_entry*) hash_search(&user_vars, (unsigned char*) name.str, name.length);
 
2088
 
 
2089
  if ((entry == NULL) && create_if_not_exists)
 
2090
  {
 
2091
    uint32_t size=ALIGN_SIZE(sizeof(user_var_entry))+name.length+1+extra_size;
 
2092
    if (!hash_inited(&user_vars))
 
2093
      return 0;
 
2094
    if (!(entry = (user_var_entry*) malloc(size)))
 
2095
      return 0;
 
2096
    entry->name.str=(char*) entry+ ALIGN_SIZE(sizeof(user_var_entry))+
 
2097
      extra_size;
 
2098
    entry->name.length=name.length;
 
2099
    entry->value=0;
 
2100
    entry->length=0;
 
2101
    entry->update_query_id=0;
 
2102
    entry->collation.set(NULL, DERIVATION_IMPLICIT, 0);
 
2103
    entry->unsigned_flag= 0;
 
2104
    /*
 
2105
      If we are here, we were called from a SET or a query which sets a
 
2106
      variable. Imagine it is this:
 
2107
      INSERT INTO t SELECT @a:=10, @a:=@a+1.
 
2108
      Then when we have a Item_func_get_user_var (because of the @a+1) so we
 
2109
      think we have to write the value of @a to the binlog. But before that,
 
2110
      we have a Item_func_set_user_var to create @a (@a:=10), in this we mark
 
2111
      the variable as "already logged" (line below) so that it won't be logged
 
2112
      by Item_func_get_user_var (because that's not necessary).
 
2113
    */
 
2114
    entry->used_query_id= query_id;
 
2115
    entry->type=STRING_RESULT;
 
2116
    memcpy(entry->name.str, name.str, name.length+1);
 
2117
    if (my_hash_insert(&user_vars, (unsigned char*) entry))
 
2118
    {
 
2119
      assert(1);
 
2120
      free((char*) entry);
 
2121
      return 0;
 
2122
    }
 
2123
 
 
2124
  }
 
2125
 
 
2126
  return entry;
2801
2127
}