mysqltest.c 165 KB
Newer Older
unknown's avatar
unknown committed
1
/* Copyright (C) 2000 MySQL AB
2

unknown's avatar
unknown committed
3 4 5 6
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.
7

unknown's avatar
unknown committed
8 9 10 11
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
12

unknown's avatar
unknown committed
13 14 15 16 17
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

/* mysqltest test tool
18 19
 * See the manual for more information
 * TODO: document better how mysqltest works
unknown's avatar
unknown committed
20 21 22 23
 *
 * Written by:
 *   Sasha Pachev <sasha@mysql.com>
 *   Matt Wagner  <matt@mysql.com>
24
 *   Monty
25
 *   Jani
unknown's avatar
unknown committed
26 27
 **/

28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
/**********************************************************************
  TODO:

- Do comparison line by line, instead of doing a full comparison of
  the text file.  This will save space as we don't need to keep many
  results in memory.  It will also make it possible to do simple
  'comparison' fixes like accepting the result even if a float differed
  in the last decimals.

- Don't buffer lines from the test that you don't expect to need
  again.

- Change 'read_line' to be faster by using the readline.cc code;
  We can do better than calling feof() for each character!

**********************************************************************/

45
#define MTEST_VERSION "2.6"
46

unknown's avatar
unknown committed
47
#include <my_global.h>
unknown's avatar
unknown committed
48
#include <mysql_embed.h>
49 50 51 52
#include <my_sys.h>
#include <m_string.h>
#include <mysql.h>
#include <mysql_version.h>
unknown's avatar
unknown committed
53
#include <mysqld_error.h>
54 55
#include <m_ctype.h>
#include <my_dir.h>
56
#include <errmsg.h>                       /* Error codes */
unknown's avatar
unknown committed
57
#include <hash.h>
58
#include <my_getopt.h>
unknown's avatar
unknown committed
59 60
#include <stdarg.h>
#include <sys/stat.h>
unknown's avatar
unknown committed
61
#include <violite.h>
unknown's avatar
unknown committed
62
#include "my_regex.h"                     /* Our own version of lib */
unknown's avatar
unknown committed
63
#ifdef HAVE_SYS_WAIT_H
64
#include <sys/wait.h>
unknown's avatar
unknown committed
65 66
#endif
#ifndef WEXITSTATUS
unknown's avatar
unknown committed
67 68 69 70 71
# ifdef __WIN__
#  define WEXITSTATUS(stat_val) (stat_val)
# else
#  define WEXITSTATUS(stat_val) ((unsigned)(stat_val) >> 8)
# endif
unknown's avatar
unknown committed
72
#endif
73 74
/* MAX_QUERY is 256K -- there is a test in sp-big that is >128K */
#define MAX_QUERY     (256*1024)
unknown's avatar
unknown committed
75
#define MAX_VAR_NAME	256
76 77
#define MAX_COLUMNS	256
#define MAX_CONS	128
unknown's avatar
unknown committed
78
#define MAX_INCLUDE_DEPTH 16
79 80
#define INIT_Q_LINES	  1024
#define MIN_VAR_ALLOC	  32
81
#define BLOCK_STACK_DEPTH  32
82
#define MAX_EXPECTED_ERRORS 10
unknown's avatar
unknown committed
83 84
#define QUERY_SEND  1
#define QUERY_REAP  2
85 86 87
#ifndef MYSQL_MANAGER_PORT
#define MYSQL_MANAGER_PORT 23546
#endif
unknown's avatar
unknown committed
88
#define MAX_SERVER_ARGS 64
89

unknown's avatar
unknown committed
90 91 92 93 94 95 96
/*
  Sometimes in a test the client starts before
  the server - to solve the problem, we try again
  after some sleep if connection fails the first
  time
*/
#define CON_RETRY_SLEEP 2
97
#define MAX_CON_TRIES	5
unknown's avatar
unknown committed
98

99
#define SLAVE_POLL_INTERVAL 300000 /* 0.3 of a sec */
100 101
#define DEFAULT_DELIMITER ";"
#define MAX_DELIMITER 16
102

103 104 105
#define RESULT_OK 0
#define RESULT_CONTENT_MISMATCH 1
#define RESULT_LENGTH_MISMATCH 2
106

107
enum {OPT_MANAGER_USER=256,OPT_MANAGER_HOST,OPT_MANAGER_PASSWD,
unknown's avatar
unknown committed
108 109
      OPT_MANAGER_PORT,OPT_MANAGER_WAIT_TIMEOUT, OPT_SKIP_SAFEMALLOC,
      OPT_SSL_SSL, OPT_SSL_KEY, OPT_SSL_CERT, OPT_SSL_CA, OPT_SSL_CAPATH,
110 111
      OPT_SSL_CIPHER,OPT_PS_PROTOCOL,OPT_SP_PROTOCOL,OPT_CURSOR_PROTOCOL,
      OPT_VIEW_PROTOCOL};
unknown's avatar
unknown committed
112

113 114
/* ************************************************************************ */
/*
115 116 117
  The list of error codes to --error are stored in an internal array of
  structs. This struct can hold numeric SQL error codes or SQLSTATE codes
  as strings. The element next to the last active element in the list is
unknown's avatar
unknown committed
118 119
  set to type ERR_EMPTY. When an SQL statement returns an error, we use
  this list to check if this is an expected error.
120
*/
unknown's avatar
unknown committed
121

122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
enum match_err_type
{
  ERR_EMPTY= 0,
  ERR_ERRNO,
  ERR_SQLSTATE
};

typedef struct
{
  enum match_err_type type;
  union
  {
    uint errnum;
    char sqlstate[SQLSTATE_LENGTH+1];  /* \0 terminated string */
  } code;
} match_err;

139 140 141 142 143 144
typedef struct
{
  const char *name;
  long        code;
} st_error;

145 146
static st_error global_error[] =
{
147 148 149 150
#include <mysqld_ername.h>
  { 0, 0 }
};

151 152 153 154
static match_err global_expected_errno[MAX_EXPECTED_ERRORS];
static uint global_expected_errors;

/* ************************************************************************ */
unknown's avatar
unknown committed
155

156
static int record = 0, opt_sleep=0;
unknown's avatar
unknown committed
157
static char *db = 0, *pass=0;
158
const char *user = 0, *host = 0, *unix_sock = 0, *opt_basedir="./";
unknown's avatar
unknown committed
159
const char *opt_include= 0;
160
static int port = 0;
161
static my_bool opt_big_test= 0, opt_compress= 0, silent= 0, verbose = 0;
162 163 164 165 166
static my_bool tty_password= 0;
static my_bool ps_protocol= 0, ps_protocol_enabled= 0;
static my_bool sp_protocol= 0, sp_protocol_enabled= 0;
static my_bool view_protocol= 0, view_protocol_enabled= 0;
static my_bool cursor_protocol= 0, cursor_protocol_enabled= 0;
167
static int parsing_disabled= 0;
168
const char *manager_user="root",*manager_host=0;
169 170 171 172
char *manager_pass=0;
int manager_port=MYSQL_MANAGER_PORT;
int manager_wait_timeout=3;
MYSQL_MANAGER* manager=0;
173

174
static char **default_argv;
unknown's avatar
unknown committed
175
static const char *load_default_groups[]= { "mysqltest","client",0 };
unknown's avatar
unknown committed
176
static char line_buffer[MAX_DELIMITER], *line_buffer_pos= line_buffer;
unknown's avatar
unknown committed
177

178 179 180 181
typedef struct
{
  FILE* file;
  const char *file_name;
182
  uint lineno; /* Current line in file */
183 184 185 186 187
} test_file;

static test_file file_stack[MAX_INCLUDE_DEPTH];
static test_file* cur_file;
static test_file* file_stack_end;
188
uint start_lineno; /* Start line of query */
189

190 191
/* Stores regex substitutions */

192 193
struct st_regex
{
194 195 196
  char* pattern; /* Pattern to be replaced */
  char* replace; /* String or expression to replace the pattern with */
  int icase; /* true if the match is case insensitive */
197 198 199 200
};

struct st_replace_regex
{
201 202 203 204 205 206 207 208 209
  DYNAMIC_ARRAY regex_arr; /* stores a list of st_regex subsitutions */
  
  /* 
    Temporary storage areas for substitutions. To reduce unnessary copying
    and memory freeing/allocation, we pre-allocate two buffers, and alternate
    their use, one for input/one for output, the roles changing on the next
    st_regex substition. At the end of substitutions  buf points to the 
    one containing the final result.
   */
210 211 212 213 214 215 216 217 218
  char* buf;
  char* even_buf;
  uint even_buf_len;
  char* odd_buf;
  uint odd_buf_len;
};

struct st_replace_regex *glob_replace_regex= 0;

unknown's avatar
unknown committed
219
static char TMPDIR[FN_REFLEN];
220 221
static char delimiter[MAX_DELIMITER]= DEFAULT_DELIMITER;
static uint delimiter_length= 1;
unknown's avatar
unknown committed
222

unknown's avatar
unknown committed
223 224 225 226 227 228 229 230 231 232
/* Block stack */
enum block_cmd { cmd_none, cmd_if, cmd_while };
typedef struct
{
  int             line; /* Start line of block */
  my_bool         ok;   /* Should block be executed */
  enum block_cmd  cmd;  /* Command owning the block */
} BLOCK;
static BLOCK block_stack[BLOCK_STACK_DEPTH];
static BLOCK *cur_block, *block_stack_end;
233

unknown's avatar
unknown committed
234
static CHARSET_INFO *charset_info= &my_charset_latin1; /* Default charset */
235
static const char *charset_name= "latin1"; /* Default character set name */
236

237 238 239
static int embedded_server_arg_count=0;
static char *embedded_server_args[MAX_SERVER_ARGS];

240
static my_bool display_result_vertically= FALSE, display_metadata= FALSE;
241

unknown's avatar
unknown committed
242 243 244 245 246 247 248
/* See the timer_output() definition for details */
static char *timer_file = NULL;
static ulonglong timer_start;
static int got_end_timer= FALSE;
static void timer_output(void);
static ulonglong timer_now(void);

unknown's avatar
unknown committed
249 250 251 252 253 254 255 256
/* Precompiled re's */
static my_regex_t ps_re;     /* the query can be run using PS protocol */
static my_regex_t sp_re;     /* the query can be run as a SP */
static my_regex_t view_re;   /* the query can be run as a view*/

static void init_re(void);
static int match_re(my_regex_t *, char *);
static void free_re(void);
257

258 259
static int reg_replace(char** buf_p, int* buf_len_p, char *pattern, char *replace, 
 char *string, int icase);
260

261 262
static const char *embedded_server_groups[]=
{
263 264 265 266 267 268
  "server",
  "embedded",
  "mysqltest_SERVER",
  NullS
};

269 270
DYNAMIC_ARRAY q_lines;

unknown's avatar
unknown committed
271 272
#include "sslopt-vars.h"

273
typedef struct
unknown's avatar
unknown committed
274 275 276 277 278
{
  char file[FN_REFLEN];
  ulong pos;
} MASTER_POS ;

unknown's avatar
unknown committed
279 280 281
struct connection
{
  MYSQL mysql;
282 283
  /* Used when creating views and sp, to avoid implicit commit */
  MYSQL* util_mysql;
unknown's avatar
unknown committed
284
  char *name;
285
  MYSQL_STMT* stmt;
unknown's avatar
unknown committed
286 287
};

unknown's avatar
unknown committed
288 289 290 291
typedef struct
{
  int read_lines,current_line;
} PARSER;
292 293

PARSER parser;
unknown's avatar
unknown committed
294
MASTER_POS master_pos;
unknown's avatar
unknown committed
295 296
/* if set, all results are concated and compared against this file */
const char *result_file = 0;
297

298
typedef struct
299
{
unknown's avatar
unknown committed
300
  char *name;
unknown's avatar
unknown committed
301
  int name_len;
unknown's avatar
unknown committed
302
  char *str_val;
303 304 305 306
  int str_val_len;
  int int_val;
  int alloced_len;
  int int_dirty; /* do not update string if int is updated until first read */
unknown's avatar
unknown committed
307
  int alloced;
308
  char *env_s;
309 310 311 312
} VAR;

VAR var_reg[10];
/*Perl/shell-like variable registers */
unknown's avatar
unknown committed
313
HASH var_hash;
314
my_bool disable_query_log=0, disable_result_log=0, disable_warnings=0;
315
my_bool disable_ps_warnings= 0;
316
my_bool disable_info= 1;			/* By default off */
317
my_bool abort_on_error= 1;
318

unknown's avatar
unknown committed
319 320 321
struct connection cons[MAX_CONS];
struct connection* cur_con, *next_con, *cons_end;

unknown's avatar
unknown committed
322 323 324
  /* Add new commands before Q_UNKNOWN !*/

enum enum_commands {
325
Q_CONNECTION=1,     Q_QUERY,
unknown's avatar
unknown committed
326
Q_CONNECT,	    Q_SLEEP, Q_REAL_SLEEP,
327 328 329 330 331 332
Q_INC,		    Q_DEC,
Q_SOURCE,	    Q_DISCONNECT,
Q_LET,		    Q_ECHO,
Q_WHILE,	    Q_END_BLOCK,
Q_SYSTEM,	    Q_RESULT,
Q_REQUIRE,	    Q_SAVE_MASTER_POS,
333 334 335
Q_SYNC_WITH_MASTER,
Q_SYNC_SLAVE_WITH_MASTER,
Q_ERROR,
336
Q_SEND,		    Q_REAP,
337
Q_DIRTY_CLOSE,	    Q_REPLACE, Q_REPLACE_COLUMN,
338 339
Q_PING,		    Q_EVAL,
Q_RPL_PROBE,	    Q_ENABLE_RPL_PARSE,
340
Q_DISABLE_RPL_PARSE, Q_EVAL_RESULT,
unknown's avatar
unknown committed
341
Q_ENABLE_QUERY_LOG, Q_DISABLE_QUERY_LOG,
unknown's avatar
unknown committed
342
Q_ENABLE_RESULT_LOG, Q_DISABLE_RESULT_LOG,
343
Q_SERVER_START, Q_SERVER_STOP,Q_REQUIRE_MANAGER,
344
Q_WAIT_FOR_SLAVE_TO_STOP,
345
Q_ENABLE_WARNINGS, Q_DISABLE_WARNINGS,
346
Q_ENABLE_PS_WARNINGS, Q_DISABLE_PS_WARNINGS,
347
Q_ENABLE_INFO, Q_DISABLE_INFO,
348
Q_ENABLE_METADATA, Q_DISABLE_METADATA,
349
Q_EXEC, Q_DELIMITER,
350
Q_DISABLE_ABORT_ON_ERROR, Q_ENABLE_ABORT_ON_ERROR,
351
Q_DISPLAY_VERTICAL_RESULTS, Q_DISPLAY_HORIZONTAL_RESULTS,
unknown's avatar
unknown committed
352
Q_QUERY_VERTICAL, Q_QUERY_HORIZONTAL,
unknown's avatar
unknown committed
353
Q_START_TIMER, Q_END_TIMER,
354
Q_CHARACTER_SET, Q_DISABLE_PS_PROTOCOL, Q_ENABLE_PS_PROTOCOL,
unknown's avatar
unknown committed
355
Q_EXIT,
356
Q_DISABLE_RECONNECT, Q_ENABLE_RECONNECT,
unknown's avatar
unknown committed
357
Q_IF,
358
Q_DISABLE_PARSING, Q_ENABLE_PARSING,
359
Q_REPLACE_REGEX,
360

361 362
Q_UNKNOWN,			       /* Unknown command.   */
Q_COMMENT,			       /* Comments, ignored. */
363
Q_COMMENT_WITH_COMMAND
unknown's avatar
unknown committed
364 365
};

366
/* this should really be called command */
367
struct st_query
unknown's avatar
unknown committed
368
{
369
  char *query, *query_buf,*first_argument,*last_argument,*end;
unknown's avatar
unknown committed
370
  int first_word_len;
371
  my_bool abort_on_error, require_file;
372
  match_err expected_errno[MAX_EXPECTED_ERRORS];
unknown's avatar
unknown committed
373
  uint expected_errors;
unknown's avatar
unknown committed
374
  char record_file[FN_REFLEN];
unknown's avatar
unknown committed
375
  enum enum_commands type;
unknown's avatar
unknown committed
376 377
};

378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397
const char *command_names[]=
{
  "connection",
  "query",
  "connect",
  "sleep",
  "real_sleep",
  "inc",
  "dec",
  "source",
  "disconnect",
  "let",
  "echo",
  "while",
  "end",
  "system",
  "result",
  "require",
  "save_master_pos",
  "sync_with_master",
398
  "sync_slave_with_master",
399 400 401 402 403
  "error",
  "send",
  "reap",
  "dirty_close",
  "replace_result",
404
  "replace_column",
405 406 407 408 409 410
  "ping",
  "eval",
  "rpl_probe",
  "enable_rpl_parse",
  "disable_rpl_parse",
  "eval_result",
unknown's avatar
unknown committed
411
  /* Enable/disable that the _query_ is logged to result file */
412 413
  "enable_query_log",
  "disable_query_log",
unknown's avatar
unknown committed
414
  /* Enable/disable that the _result_ from a query is logged to result file */
415 416 417 418 419 420
  "enable_result_log",
  "disable_result_log",
  "server_start",
  "server_stop",
  "require_manager",
  "wait_for_slave_to_stop",
421 422
  "enable_warnings",
  "disable_warnings",
423 424
  "enable_ps_warnings",
  "disable_ps_warnings",
425
  "enable_info",
426
  "disable_info",
427 428
  "enable_metadata",
  "disable_metadata",
429
  "exec",
430
  "delimiter",
431 432
  "disable_abort_on_error",
  "enable_abort_on_error",
433
  "vertical_results",
434
  "horizontal_results",
435
  "query_vertical",
436
  "query_horizontal",
unknown's avatar
unknown committed
437 438
  "start_timer",
  "end_timer",
unknown's avatar
unknown committed
439
  "character_set",
440 441
  "disable_ps_protocol",
  "enable_ps_protocol",
unknown's avatar
unknown committed
442
  "exit",
443 444
  "disable_reconnect",
  "enable_reconnect",
unknown's avatar
unknown committed
445
  "if",
446 447
  "disable_parsing",
  "enable_parsing",
448
  "replace_regex",
unknown's avatar
unknown committed
449
  0
450 451 452
};

TYPELIB command_typelib= {array_elements(command_names),"",
unknown's avatar
unknown committed
453
			  command_names, 0};
454

455
DYNAMIC_STRING ds_res;
unknown's avatar
unknown committed
456
static void die(const char *fmt, ...);
unknown's avatar
unknown committed
457
static void init_var_hash();
458
static VAR* var_from_env(const char *, const char *);
459
static byte* get_var_key(const byte* rec, uint* len, my_bool t);
unknown's avatar
unknown committed
460
static VAR* var_init(VAR* v, const char *name, int name_len, const char *val,
unknown's avatar
unknown committed
461 462 463
		     int val_len);

static void var_free(void* v);
464

465
void dump_result_to_reject_file(const char *record_file, char *buf, int size);
466
void dump_result_to_log_file(const char *record_file, char *buf, int size);
unknown's avatar
unknown committed
467

468
int close_connection(struct st_query*);
unknown's avatar
unknown committed
469
static void set_charset(struct st_query*);
470 471
VAR* var_get(const char *var_name, const char** var_name_end, my_bool raw,
	     my_bool ignore_not_existing);
unknown's avatar
unknown committed
472 473
int eval_expr(VAR* v, const char *p, const char** p_end);
static int read_server_arguments(const char *name);
474

475
/* Definitions for replace result */
unknown's avatar
unknown committed
476 477 478 479 480

typedef struct st_pointer_array {		/* when using array-strings */
  TYPELIB typelib;				/* Pointer to strings */
  byte	*str;					/* Strings is here */
  int7	*flag;					/* Flag about each var. */
481
  uint	array_allocs,max_count,length,max_length;
unknown's avatar
unknown committed
482 483 484 485 486
} POINTER_ARRAY;

struct st_replace;
struct st_replace *init_replace(my_string *from, my_string *to, uint count,
				my_string word_end_chars);
487
void free_replace();
488
static void free_replace_regex();
unknown's avatar
unknown committed
489
static int insert_pointer_name(reg1 POINTER_ARRAY *pa,my_string name);
490 491
static void replace_strings_append(struct st_replace *rep, DYNAMIC_STRING* ds,
                                   const char *from, int len);
unknown's avatar
unknown committed
492
void free_pointer_array(POINTER_ARRAY *pa);
493 494
static void do_eval(DYNAMIC_STRING *query_eval, const char *query,
                    my_bool pass_through_escape_chars);
495
static void str_to_file(const char *fname, char *str, int size);
496 497

#ifdef __WIN__
unknown's avatar
unknown committed
498
static void free_tmp_sh_file();
499 500
static void free_win_path_patterns();
#endif
unknown's avatar
unknown committed
501 502

struct st_replace *glob_replace;
503
static int eval_result = 0;
unknown's avatar
unknown committed
504

505 506 507 508 509 510 511
/* For column replace */
char *replace_column[MAX_COLUMNS];
uint max_replace_column= 0;

static void get_replace_column(struct st_query *q);
static void free_replace_column();

unknown's avatar
unknown committed
512
/* Disable functions that only exist in MySQL 4.0 */
unknown's avatar
SCRUM  
unknown committed
513
#if MYSQL_VERSION_ID < 40000
unknown's avatar
unknown committed
514 515 516
void mysql_enable_rpl_parse(MYSQL* mysql __attribute__((unused))) {}
void mysql_disable_rpl_parse(MYSQL* mysql __attribute__((unused))) {}
int mysql_rpl_parse_enabled(MYSQL* mysql __attribute__((unused))) { return 1; }
517
my_bool mysql_rpl_probe(MYSQL *mysql __attribute__((unused))) { return 1; }
unknown's avatar
unknown committed
518
#endif
519 520
static void replace_dynstr_append_mem(DYNAMIC_STRING *ds, const char *val,
				      int len);
521
static void replace_dynstr_append(DYNAMIC_STRING *ds, const char *val);
522 523 524 525
static void handle_error(const char *query, struct st_query *q,
			 unsigned int err_errno, const char *err_error,
			 const char *err_sqlstate, DYNAMIC_STRING *ds);
static void handle_no_error(struct st_query *q);
526

527 528
static void do_eval(DYNAMIC_STRING* query_eval, const char *query,
                    my_bool pass_through_escape_chars)
529
{
530
  const char *p;
531
  register char c, next_c;
532 533
  register int escaped = 0;
  VAR* v;
unknown's avatar
unknown committed
534
  DBUG_ENTER("do_eval");
unknown's avatar
unknown committed
535

536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
  for (p= query; (c = *p); ++p)
  {
    switch(c) {
    case '$':
      if (escaped)
      {
	escaped = 0;
	dynstr_append_mem(query_eval, p, 1);
      }
      else
      {
	if (!(v = var_get(p, &p, 0, 0)))
	  die("Bad variable in eval");
	dynstr_append_mem(query_eval, v->str_val, v->str_val_len);
      }
      break;
    case '\\':
553
      next_c= *(p+1);
554 555 556 557 558
      if (escaped)
      {
	escaped = 0;
	dynstr_append_mem(query_eval, p, 1);
      }
559 560 561
      else if (next_c == '\\' || next_c == '$')
      {
        /* Set escaped only if next char is \ or $ */
562
	escaped = 1;
563 564 565 566 567 568

        if (pass_through_escape_chars)
        {
          /* The escape char should be added to the output string. */
          dynstr_append_mem(query_eval, p, 1);
        }
569 570 571
      }
      else
	dynstr_append_mem(query_eval, p, 1);
572 573 574 575
      break;
    default:
      dynstr_append_mem(query_eval, p, 1);
      break;
576
    }
577
  }
unknown's avatar
unknown committed
578
  DBUG_VOID_RETURN;
579
}
unknown's avatar
unknown committed
580

581

582 583
static void close_cons()
{
584 585 586
  DBUG_ENTER("close_cons");
  for (--next_con; next_con >= cons; --next_con)
  {
587 588 589
    if (next_con->stmt)
      mysql_stmt_close(next_con->stmt);
    next_con->stmt= 0;
590
    mysql_close(&next_con->mysql);
591 592
    if (next_con->util_mysql)
      mysql_close(next_con->util_mysql);
593 594 595 596 597
    my_free(next_con->name, MYF(MY_ALLOW_ZERO_PTR));
  }
  DBUG_VOID_RETURN;
}

598

599 600
static void close_files()
{
unknown's avatar
unknown committed
601
  DBUG_ENTER("close_files");
602
  for (; cur_file >= file_stack; cur_file--)
603
  {
604 605 606 607 608
    DBUG_PRINT("info", ("file_name: %s", cur_file->file_name));
    if (cur_file->file && cur_file->file != stdin)
      my_fclose(cur_file->file, MYF(0));
    my_free((gptr)cur_file->file_name, MYF(MY_ALLOW_ZERO_PTR));
    cur_file->file_name= 0;
unknown's avatar
unknown committed
609 610
  }
  DBUG_VOID_RETURN;
611 612
}

613

614 615 616 617
static void free_used_memory()
{
  uint i;
  DBUG_ENTER("free_used_memory");
618
#ifndef EMBEDDED_LIBRARY
619 620
  if (manager)
    mysql_manager_close(manager);
621
#endif
622 623
  close_cons();
  close_files();
unknown's avatar
unknown committed
624
  hash_free(&var_hash);
unknown's avatar
unknown committed
625

626 627 628
  for (i=0 ; i < q_lines.elements ; i++)
  {
    struct st_query **q= dynamic_element(&q_lines, i, struct st_query**);
629
    my_free((gptr) (*q)->query_buf,MYF(MY_ALLOW_ZERO_PTR));
630 631
    my_free((gptr) (*q),MYF(0));
  }
632
  for (i=0; i < 10; i++)
unknown's avatar
unknown committed
633 634 635 636
  {
    if (var_reg[i].alloced_len)
      my_free(var_reg[i].str_val, MYF(MY_WME));
  }
637 638
  while (embedded_server_arg_count > 1)
    my_free(embedded_server_args[--embedded_server_arg_count],MYF(0));
639 640
  delete_dynamic(&q_lines);
  dynstr_free(&ds_res);
unknown's avatar
unknown committed
641
  free_replace();
642
  free_replace_column();
643 644
  my_free(pass,MYF(MY_ALLOW_ZERO_PTR));
  free_defaults(default_argv);
unknown's avatar
unknown committed
645
  mysql_server_end();
646
  free_re();
647
#ifdef __WIN__
unknown's avatar
unknown committed
648
  free_tmp_sh_file();
649 650
  free_win_path_patterns();
#endif
651
  DBUG_VOID_RETURN;
652 653
}

654
static void die(const char *fmt, ...)
655 656
{
  va_list args;
657
  DBUG_ENTER("die");
658 659

  /* Print the error message */
660
  va_start(args, fmt);
661 662
  if (fmt)
  {
663 664
    fprintf(stderr, "mysqltest: ");
    if (cur_file && cur_file != file_stack)
665
      fprintf(stderr, "In included file \"%s\": ",
666
              cur_file->file_name);
unknown's avatar
unknown committed
667 668
    if (start_lineno != 0)
      fprintf(stderr, "At line %u: ", start_lineno);
669 670
    vfprintf(stderr, fmt, args);
    fprintf(stderr, "\n");
unknown's avatar
unknown committed
671
    fflush(stderr);
672
  }
673
  va_end(args);
674

675
  /* Dump the result that has been accumulated so far to .log file */
676
  if (result_file && ds_res.length)
677
    dump_result_to_log_file(result_file, ds_res.str, ds_res.length);
678 679

  /* Clean up and exit */
680
  free_used_memory();
681
  my_end(MY_CHECK_ERROR);
unknown's avatar
unknown committed
682 683 684 685

  if (!silent)
    printf("not ok\n");

686 687 688
  exit(1);
}

689 690
/* Note that we will get some memory leaks when calling this! */

691 692
static void abort_not_supported_test()
{
693
  DBUG_ENTER("abort_not_supported_test");
694 695 696
  fprintf(stderr, "This test is not supported by this installation\n");
  if (!silent)
    printf("skipped\n");
697
  free_used_memory();
698
  my_end(MY_CHECK_ERROR);
unknown's avatar
unknown committed
699
  exit(62);
700 701
}

702
static void verbose_msg(const char *fmt, ...)
703 704
{
  va_list args;
705 706 707
  DBUG_ENTER("verbose_msg");
  if (!verbose)
    DBUG_VOID_RETURN;
708 709 710

  va_start(args, fmt);

unknown's avatar
unknown committed
711
  fprintf(stderr, "mysqltest: ");
712
  if (start_lineno != 0)
unknown's avatar
unknown committed
713
    fprintf(stderr, "At line %u: ", start_lineno);
714 715 716
  vfprintf(stderr, fmt, args);
  fprintf(stderr, "\n");
  va_end(args);
717
  DBUG_VOID_RETURN;
718 719
}

unknown's avatar
unknown committed
720

721 722
void init_parser()
{
723 724
  parser.current_line= parser.read_lines= 0;
  memset(&var_reg, 0, sizeof(var_reg));
725
}
unknown's avatar
unknown committed
726 727


728
static int dyn_string_cmp(DYNAMIC_STRING* ds, const char *fname)
unknown's avatar
unknown committed
729 730
{
  MY_STAT stat_info;
731 732
  char *tmp, *res_ptr;
  char eval_file[FN_REFLEN];
unknown's avatar
unknown committed
733
  int res;
734
  uint res_len;
unknown's avatar
unknown committed
735
  int fd;
736
  DYNAMIC_STRING res_ds;
737 738
  DBUG_ENTER("dyn_string_cmp");

739 740 741 742 743 744 745 746 747
  if (!test_if_hard_path(fname))
  {
    strxmov(eval_file, opt_basedir, fname, NullS);
    fn_format(eval_file, eval_file,"","",4);
  }
  else
    fn_format(eval_file, fname,"","",4);

  if (!my_stat(eval_file, &stat_info, MYF(MY_WME)))
748
    die(NullS);
749 750 751 752
  if (!eval_result && (uint) stat_info.st_size != ds->length)
  {
    DBUG_PRINT("info",("Size differs:  result size: %u  file size: %u",
		       ds->length, stat_info.st_size));
753
    DBUG_PRINT("info",("result: '%s'", ds->str));
754
    DBUG_RETURN(RESULT_LENGTH_MISMATCH);
755
  }
756
  if (!(tmp = (char*) my_malloc(stat_info.st_size + 1, MYF(MY_WME))))
757
    die(NullS);
758 759

  if ((fd = my_open(eval_file, O_RDONLY, MYF(MY_WME))) < 0)
760
    die(NullS);
761
  if (my_read(fd, (byte*)tmp, stat_info.st_size, MYF(MY_WME|MY_NABP)))
762
    die(NullS);
763 764 765 766
  tmp[stat_info.st_size] = 0;
  init_dynamic_string(&res_ds, "", 0, 65536);
  if (eval_result)
  {
767
    do_eval(&res_ds, tmp, FALSE);
768
    res_ptr = res_ds.str;
769
    if ((res_len = res_ds.length) != ds->length)
770
    {
771
      res= RESULT_LENGTH_MISMATCH;
772 773 774 775 776 777 778 779
      goto err;
    }
  }
  else
  {
    res_ptr = tmp;
    res_len = stat_info.st_size;
  }
unknown's avatar
unknown committed
780

781 782
  res= (memcmp(res_ptr, ds->str, res_len)) ?
    RESULT_CONTENT_MISMATCH : RESULT_OK;
unknown's avatar
unknown committed
783

784 785 786 787
err:
  if (res && eval_result)
    str_to_file(fn_format(eval_file, fname, "", ".eval",2), res_ptr,
		res_len);
unknown's avatar
unknown committed
788

789 790
  my_free((gptr) tmp, MYF(0));
  my_close(fd, MYF(MY_WME));
791
  dynstr_free(&res_ds);
unknown's avatar
unknown committed
792

793
  DBUG_RETURN(res);
unknown's avatar
unknown committed
794 795
}

796 797
/*
  Check the content of ds against content of file fname
unknown's avatar
unknown committed
798

799 800 801 802
  SYNOPSIS
  check_result
  ds - content to be checked
  fname - name of file to check against
unknown's avatar
unknown committed
803 804
  require_option - if set and check fails, the test will be aborted
                   with the special exit code "not supported test"
unknown's avatar
unknown committed
805

806 807 808 809 810
  RETURN VALUES
   error - the function will not return

*/
static void check_result(DYNAMIC_STRING* ds, const char *fname,
811
			my_bool require_option)
unknown's avatar
unknown committed
812
{
813
  int res= dyn_string_cmp(ds, fname);
814
  DBUG_ENTER("check_result");
815 816 817

  if (res && require_option)
    abort_not_supported_test();
818
  switch (res) {
819
  case RESULT_OK:
820
    break; /* ok */
821
  case RESULT_LENGTH_MISMATCH:
822 823
    dump_result_to_reject_file(fname, ds->str, ds->length);
    die("Result length mismatch");
824
    break;
825
  case RESULT_CONTENT_MISMATCH:
826 827
    dump_result_to_reject_file(fname, ds->str, ds->length);
    die("Result content mismatch");
828 829 830 831
    break;
  default: /* impossible */
    die("Unknown error code from dyn_string_cmp()");
  }
832 833

  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
834 835
}

836

837
VAR* var_get(const char *var_name, const char** var_name_end, my_bool raw,
838
	     my_bool ignore_not_existing)
839 840 841
{
  int digit;
  VAR* v;
842 843 844 845
  DBUG_ENTER("var_get");
  DBUG_PRINT("enter",("var_name: %s",var_name));

  if (*var_name != '$')
846
    goto err;
847
  digit = *++var_name - '0';
unknown's avatar
unknown committed
848
  if (digit < 0 || digit >= 10)
849
  {
850
    const char *save_var_name = var_name, *end;
851
    uint length;
unknown's avatar
unknown committed
852
    end = (var_name_end) ? *var_name_end : 0;
853
    while (my_isvar(charset_info,*var_name) && var_name != end)
854
      var_name++;
855 856 857 858
    if (var_name == save_var_name)
    {
      if (ignore_not_existing)
	DBUG_RETURN(0);
unknown's avatar
unknown committed
859
      die("Empty variable");
860
    }
861
    length= (uint) (var_name - save_var_name);
862 863
    if (length >= MAX_VAR_NAME)
      die("Too long variable name: %s", save_var_name);
unknown's avatar
unknown committed
864

865
    if (!(v = (VAR*) hash_search(&var_hash, save_var_name, length)))
866
    {
867
      char buff[MAX_VAR_NAME+1];
unknown's avatar
unknown committed
868
      strmake(buff, save_var_name, length);
869
      v= var_from_env(buff, "");
870
    }
871
    var_name--;					/* Point at last character */
872
  }
873
  else
874
    v = var_reg + digit;
unknown's avatar
unknown committed
875

876 877 878 879
  if (!raw && v->int_dirty)
  {
    sprintf(v->str_val, "%d", v->int_val);
    v->int_dirty = 0;
880
    v->str_val_len = strlen(v->str_val);
881
  }
882
  if (var_name_end)
883
    *var_name_end = var_name  ;
884
  DBUG_RETURN(v);
885 886
err:
  if (var_name_end)
887 888
    *var_name_end = 0;
  die("Unsupported variable name: %s", var_name);
889
  DBUG_RETURN(0);
890 891
}

892
static VAR *var_obtain(const char *name, int len)
unknown's avatar
unknown committed
893 894
{
  VAR* v;
895
  if ((v = (VAR*)hash_search(&var_hash, name, len)))
unknown's avatar
unknown committed
896
    return v;
897
  v = var_init(0, name, len, "", 0);
unknown's avatar
SCRUM  
unknown committed
898
  my_hash_insert(&var_hash, (byte*)v);
unknown's avatar
unknown committed
899 900 901
  return v;
}

unknown's avatar
unknown committed
902 903 904 905 906
/*
  - if variable starts with a $ it is regarded as a local test varable
  - if not it is treated as a environment variable, and the corresponding
  environment variable will be updated
*/
907 908
int var_set(const char *var_name, const char *var_name_end,
            const char *var_val, const char *var_val_end)
909
{
unknown's avatar
unknown committed
910
  int digit, result, env_var= 0;
911
  VAR* v;
912 913 914 915 916 917
  DBUG_ENTER("var_set");
  DBUG_PRINT("enter", ("var_name: '%.*s' = '%.*s' (length: %d)",
                       (int) (var_name_end - var_name), var_name,
                       (int) (var_val_end - var_val), var_val,
                       (int) (var_val_end - var_val)));

unknown's avatar
unknown committed
918 919 920 921 922
  if (*var_name != '$')
    env_var= 1;
  else
    var_name++;

923
  digit = *var_name - '0';
924
  if (!(digit < 10 && digit >= 0))
925 926 927
  {
    v = var_obtain(var_name, (uint) (var_name_end - var_name));
  }
928
  else
929
    v = var_reg + digit;
unknown's avatar
unknown committed
930 931 932 933 934

  result= eval_expr(v, var_val, (const char**) &var_val_end);

  if (env_var)
  {
935
    char buf[1024], *old_env_s= v->env_s;
unknown's avatar
unknown committed
936 937 938 939 940 941
    if (v->int_dirty)
    {
      sprintf(v->str_val, "%d", v->int_val);
      v->int_dirty= 0;
      v->str_val_len= strlen(v->str_val);
    }
942 943 944 945 946
    strxmov(buf, v->name, "=", v->str_val, NullS);
    if (!(v->env_s= my_strdup(buf, MYF(MY_WME))))
      die("Out of memory");
    putenv(v->env_s);
    my_free((gptr)old_env_s, MYF(MY_ALLOW_ZERO_PTR));
unknown's avatar
unknown committed
947 948
  }
  DBUG_RETURN(result);
949 950
}

951

952
int open_file(const char *name)
unknown's avatar
unknown committed
953
{
954
  char buff[FN_REFLEN];
955 956
  DBUG_ENTER("open_file");
  DBUG_PRINT("enter", ("name: %s", name));
957 958 959 960 961 962 963
  if (!test_if_hard_path(name))
  {
    strxmov(buff, opt_basedir, name, NullS);
    name=buff;
  }
  fn_format(buff,name,"","",4);

964
  if (cur_file == file_stack_end)
unknown's avatar
unknown committed
965
    die("Source directives are nesting too deep");
966
  cur_file++;
967 968 969
  if (!(cur_file->file = my_fopen(buff, O_RDONLY | FILE_BINARY, MYF(0))))
  {
    cur_file--;
970
    die("Could not open file %s", buff);
971 972
  }
  cur_file->file_name= my_strdup(buff, MYF(MY_FAE));
973
  cur_file->lineno=1;
974
  DBUG_RETURN(0);
975
}
976

977 978 979 980 981 982 983 984

/*
  Check for unexpected "junk" after the end of query
  This is normally caused by missing delimiters
*/

int check_eol_junk(const char *eol)
{
985
  const char *p= eol;
986 987
  DBUG_ENTER("check_eol_junk");
  DBUG_PRINT("enter", ("eol: %s", eol));
988
  /* Remove all spacing chars except new line */
989
  while (*p && my_isspace(charset_info, *p) && (*p != '\n'))
990 991 992
    p++;

  /* Check for extra delimiter */
993
  if (*p && !strncmp(p, delimiter, delimiter_length))
994
    die("Extra delimiter \"%s\" found", delimiter);
995

996 997 998 999 1000 1001 1002
  /* Allow trailing # comment */
  if (*p && *p != '#')
  {
    if (*p == '\n')
      die("Missing delimiter");
    die("End of line junk detected: \"%s\"", p);
  }
1003
  DBUG_RETURN(0);
unknown's avatar
unknown committed
1004 1005
}

unknown's avatar
unknown committed
1006

1007
/* ugly long name, but we are following the convention */
1008
int do_wait_for_slave_to_stop(struct st_query *q __attribute__((unused)))
1009 1010 1011 1012
{
  MYSQL* mysql = &cur_con->mysql;
  for (;;)
  {
unknown's avatar
unknown committed
1013
    MYSQL_RES *res;
1014 1015 1016
    MYSQL_ROW row;
    int done;
    LINT_INIT(res);
1017

unknown's avatar
unknown committed
1018 1019
    if (mysql_query(mysql,"show status like 'Slave_running'") ||
	!(res=mysql_store_result(mysql)))
1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
      die("Query failed while probing slave for stop: %s",
	  mysql_error(mysql));
    if (!(row=mysql_fetch_row(res)) || !row[1])
    {
      mysql_free_result(res);
      die("Strange result from query while probing slave for stop");
    }
    done = !strcmp(row[1],"OFF");
    mysql_free_result(res);
    if (done)
      break;
unknown's avatar
unknown committed
1031
    my_sleep(SLAVE_POLL_INTERVAL);
1032 1033 1034 1035
  }
  return 0;
}

1036
int do_require_manager(struct st_query *query __attribute__((unused)) )
1037 1038 1039 1040 1041 1042
{
  if (!manager)
    abort_not_supported_test();
  return 0;
}

1043
#ifndef EMBEDDED_LIBRARY
1044
static int do_server_op(struct st_query *q, const char *op)
1045
{
1046 1047
  char *p= q->first_argument;
  char com_buf[256], *com_p;
1048 1049 1050 1051
  if (!manager)
  {
    die("Manager is not initialized, manager commands are not possible");
  }
1052 1053
  com_p= strmov(com_buf,op);
  com_p= strmov(com_p,"_exec ");
1054
  if (!*p)
1055 1056
    die("Missing server name in server_%s", op);
  while (*p && !my_isspace(charset_info, *p))
unknown's avatar
unknown committed
1057
   *com_p++= *p++;
1058 1059 1060 1061 1062 1063
  *com_p++= ' ';
  com_p= int10_to_str(manager_wait_timeout, com_p, 10);
  *com_p++= '\n';
  *com_p= 0;
  if (mysql_manager_command(manager, com_buf, (int)(com_p-com_buf)))
    die("Error in command: %s(%d)", manager->last_error, manager->last_errno);
1064 1065
  while (!manager->eof)
  {
1066
    if (mysql_manager_fetch_line(manager, com_buf, sizeof(com_buf)))
1067 1068 1069 1070
      die("Error fetching result line: %s(%d)", manager->last_error,
	  manager->last_errno);
  }

1071
  q->last_argument= p;
1072 1073
  return 0;
}
1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084

int do_server_start(struct st_query *q)
{
  return do_server_op(q, "start");
}

int do_server_stop(struct st_query *q)
{
  return do_server_op(q, "stop");
}

1085
#endif
1086

1087 1088 1089 1090 1091 1092

/*
  Source and execute the given file

  SYNOPSIS
    do_source()
1093
    query	called command
1094 1095 1096 1097 1098 1099 1100 1101

  DESCRIPTION
    source <file_name>

    Open the file <file_name> and execute it

*/

1102
int do_source(struct st_query *query)
unknown's avatar
unknown committed
1103
{
1104
  char *p= query->first_argument, *name;
1105
  if (!*p)
1106
    die("Missing file name in source");
1107
  name= p;
1108
  while (*p && !my_isspace(charset_info,*p))
unknown's avatar
unknown committed
1109
    p++;
1110 1111
  if (*p)
    *p++= 0;
1112 1113
  query->last_argument= p;
  /*
unknown's avatar
unknown committed
1114 1115
     If this file has already been sourced, don't source it again.
     It's already available in the q_lines cache.
1116
  */
1117 1118
  if (parser.current_line < (parser.read_lines - 1))
    return 0;
unknown's avatar
unknown committed
1119 1120 1121
  return open_file(name);
}

1122
#ifdef __WIN__
unknown's avatar
unknown committed
1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137
/* Variables used for temuprary sh files used for emulating Unix on Windows */
char tmp_sh_name[64], tmp_sh_cmd[70];

static void init_tmp_sh_file()
{
  /* Format a name for the tmp sh file that is unique for this process */
  my_snprintf(tmp_sh_name, sizeof(tmp_sh_name), "tmp_%d.sh", getpid());
  /* Format the command to execute in order to run the script */
  my_snprintf(tmp_sh_cmd, sizeof(tmp_sh_cmd), "sh %s", tmp_sh_name);
}

static void free_tmp_sh_file()
{
  my_delete(tmp_sh_name, MYF(0));
}
1138
#endif
unknown's avatar
unknown committed
1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150

FILE* my_popen(DYNAMIC_STRING* ds_cmd, const char* mode)
{
#ifdef __WIN__
  /* Dump the command into a sh script file and execute with popen */
  str_to_file(tmp_sh_name, ds_cmd->str, ds_cmd->length);
  return popen(tmp_sh_cmd, mode);
#else
  return popen(ds_cmd->str, mode);
#endif
}

1151

1152 1153 1154 1155 1156
/*
  Execute given command.

  SYNOPSIS
    do_exec()
1157
    query	called command
1158 1159

  DESCRIPTION
1160 1161 1162 1163 1164 1165
    exec <command>

    Execute the text between exec and end of line in a subprocess.
    The error code returned from the subprocess is checked against the
    expected error array, previously set with the --error command.
    It can thus be used to execute a command that shall fail.
1166

1167 1168 1169 1170
  NOTE
     Although mysqltest is executed from cygwin shell, the command will be
     executed in "cmd.exe". Thus commands like "rm" etc can NOT be used, use
     system for those commands.
1171 1172
*/

1173
static void do_exec(struct st_query *query)
1174
{
unknown's avatar
unknown committed
1175
  int error;
1176 1177
  char buf[1024];
  FILE *res_file;
1178
  char *cmd= query->first_argument;
1179
  DYNAMIC_STRING ds_cmd;
unknown's avatar
unknown committed
1180
  DBUG_ENTER("do_exec");
1181
  DBUG_PRINT("enter", ("cmd: '%s'", cmd));
1182

unknown's avatar
unknown committed
1183
  while (*cmd && my_isspace(charset_info, *cmd))
1184 1185
    cmd++;
  if (!*cmd)
1186
    die("Missing argument in exec");
1187
  query->last_argument= query->end;
1188

1189 1190
  init_dynamic_string(&ds_cmd, 0, strlen(cmd)+256, 256);
  /* Eval the command, thus replacing all environment variables */
1191
  do_eval(&ds_cmd, cmd, TRUE);
1192 1193
  cmd= ds_cmd.str;

unknown's avatar
unknown committed
1194
  DBUG_PRINT("info", ("Executing '%s' as '%s'",
1195
                      query->first_argument, cmd));
unknown's avatar
unknown committed
1196

unknown's avatar
unknown committed
1197
  if (!(res_file= my_popen(&ds_cmd, "r")) && query->abort_on_error)
1198
    die("popen(\"%s\", \"r\") failed", query->first_argument);
unknown's avatar
unknown committed
1199

1200
  while (fgets(buf, sizeof(buf), res_file))
1201
  {
1202
    if (disable_result_log)
1203 1204 1205 1206
    {
      buf[strlen(buf)-1]=0;
      DBUG_PRINT("exec_result",("%s", buf));
    }
1207 1208 1209 1210
    else
    {
      replace_dynstr_append(&ds_res, buf);
    }
unknown's avatar
unknown committed
1211 1212 1213
  }
  error= pclose(res_file);
  if (error != 0)
unknown's avatar
patch  
unknown committed
1214
  {
unknown's avatar
unknown committed
1215 1216 1217
    uint status= WEXITSTATUS(error), i;
    my_bool ok= 0;

1218
    if (query->abort_on_error)
1219
      die("command \"%s\" failed", query->first_argument);
unknown's avatar
unknown committed
1220 1221 1222

    DBUG_PRINT("info",
               ("error: %d, status: %d", error, status));
1223
    for (i= 0; i < query->expected_errors; i++)
unknown's avatar
patch  
unknown committed
1224
    {
unknown's avatar
unknown committed
1225
      DBUG_PRINT("info", ("expected error: %d",
1226 1227 1228
                          query->expected_errno[i].code.errnum));
      if ((query->expected_errno[i].type == ERR_ERRNO) &&
          (query->expected_errno[i].code.errnum == status))
1229
      {
unknown's avatar
unknown committed
1230
        ok= 1;
1231
        DBUG_PRINT("info", ("command \"%s\" failed with expected error: %d",
1232
                            query->first_argument, status));
1233
      }
unknown's avatar
patch  
unknown committed
1234
    }
unknown's avatar
unknown committed
1235
    if (!ok)
1236
      die("command \"%s\" failed with wrong error: %d",
1237
          query->first_argument, status);
unknown's avatar
patch  
unknown committed
1238
  }
1239 1240
  else if (query->expected_errno[0].type == ERR_ERRNO &&
           query->expected_errno[0].code.errnum != 0)
unknown's avatar
patch  
unknown committed
1241 1242
  {
    /* Error code we wanted was != 0, i.e. not an expected success */
1243
    die("command \"%s\" succeeded - should have failed with errno %d...",
1244
        query->first_argument, query->expected_errno[0].code.errnum);
unknown's avatar
patch  
unknown committed
1245
  }
unknown's avatar
unknown committed
1246

1247
  free_replace();
1248
  DBUG_VOID_RETURN;
1249 1250
}

1251 1252 1253 1254 1255 1256 1257 1258
/*
  Set variable from the result of a query

  SYNOPSIS
    var_query_set()
    var	        variable to set from query
    query       start of query string to execute
    query_end   end of the query string to execute
unknown's avatar
unknown committed
1259

1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270

  DESCRIPTION
    let @<var_name> = `<query>`

    Execute the query and assign the first row of result to var as
    a tab separated strings

    Also assign each column of the result set to
    variable "$<var_name>_<column_name>"
    Thus the tab separated output can be read from $<var_name> and
    and each individual column can be read as $<var_name>_<col_name>
unknown's avatar
unknown committed
1271

1272 1273 1274
*/

int var_query_set(VAR* var, const char *query, const char** query_end)
1275
{
1276 1277
  char* end = (char*)((query_end && *query_end) ?
		      *query_end : query + strlen(query));
1278 1279 1280 1281
  MYSQL_RES *res;
  MYSQL_ROW row;
  MYSQL* mysql = &cur_con->mysql;
  LINT_INIT(res);
unknown's avatar
unknown committed
1282

1283
  while (end > query && *end != '`')
1284
    --end;
1285
  if (query == end)
1286
    die("Syntax error in query, missing '`'");
1287
  ++query;
1288

1289
  if (mysql_real_query(mysql, query, (int)(end - query)) ||
1290 1291 1292
      !(res = mysql_store_result(mysql)))
  {
    *end = 0;
1293 1294
    die("Error running query '%s': %d: %s", query,
	mysql_errno(mysql) ,mysql_error(mysql));
1295 1296 1297
  }

  if ((row = mysql_fetch_row(res)) && row[0])
1298 1299 1300 1301 1302 1303 1304 1305 1306
  {
    /*
      Concatenate all row results with tab in between to allow us to work
      with results from many columns (for example from SHOW VARIABLES)
    */
    DYNAMIC_STRING result;
    uint i;
    ulong *lengths;
    char *end;
1307
    MYSQL_FIELD *fields= mysql_fetch_fields(res);
1308 1309 1310 1311 1312 1313

    init_dynamic_string(&result, "", 16384, 65536);
    lengths= mysql_fetch_lengths(res);
    for (i=0; i < mysql_num_fields(res); i++)
    {
      if (row[0])
1314
      {
1315
#ifdef NOT_YET
1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328
	/* Add to <var_name>_<col_name> */
	uint j;
	char var_col_name[MAX_VAR_NAME];
	uint length= snprintf(var_col_name, MAX_VAR_NAME,
			      "$%s_%s", var->name, fields[i].name);
	/* Convert characters not allowed in variable names to '_' */
	for (j= 1; j < length; j++)
	{
	  if (!my_isvar(charset_info,var_col_name[j]))
	     var_col_name[j]= '_';
        }
	var_set(var_col_name,  var_col_name + length,
		row[i], row[i] + lengths[i]);
1329
#endif
1330
        /* Add column to tab separated string */
1331
	dynstr_append_mem(&result, row[i], lengths[i]);
1332
      }
1333 1334 1335
      dynstr_append_mem(&result, "\t", 1);
    }
    end= result.str + result.length-1;
1336
    eval_expr(var, result.str, (const char**) &end);
1337 1338
    dynstr_free(&result);
  }
1339
  else
1340
    eval_expr(var, "", 0);
1341 1342 1343 1344

  mysql_free_result(res);
  return 0;
}
1345

1346
void var_copy(VAR *dest, VAR *src)
unknown's avatar
unknown committed
1347
{
1348 1349 1350 1351
  dest->int_val= src->int_val;
  dest->int_dirty= src->int_dirty;

  /* Alloc/realloc data for str_val in dest */
unknown's avatar
unknown committed
1352
  if (dest->alloced_len < src->alloced_len &&
1353 1354 1355
      !(dest->str_val= dest->str_val
        ? my_realloc(dest->str_val, src->alloced_len, MYF(MY_WME))
        : my_malloc(src->alloced_len, MYF(MY_WME))))
unknown's avatar
unknown committed
1356
    die("Out of memory");
1357 1358 1359 1360 1361 1362 1363
  else
    dest->alloced_len= src->alloced_len;

  /* Copy str_val data to dest */
  dest->str_val_len= src->str_val_len;
  if (src->str_val_len)
    memcpy(dest->str_val, src->str_val, src->str_val_len);
unknown's avatar
unknown committed
1364 1365
}

1366
int eval_expr(VAR* v, const char *p, const char** p_end)
1367 1368
{
  VAR* vp;
1369
  if (*p == '$')
1370 1371
  {
    if ((vp = var_get(p,p_end,0,0)))
1372
    {
1373 1374
      var_copy(v, vp);
      return 0;
unknown's avatar
unknown committed
1375
    }
1376
  }
1377
  else if (*p == '`')
1378 1379 1380
  {
    return var_query_set(v, p, p_end);
  }
unknown's avatar
unknown committed
1381 1382
  else
    {
1383 1384 1385 1386
      int new_val_len = (p_end && *p_end) ?
	 (int) (*p_end - p) : (int) strlen(p);
      if (new_val_len + 1 >= v->alloced_len)
      {
1387
	v->alloced_len = (new_val_len < MIN_VAR_ALLOC - 1) ?
1388 1389
	  MIN_VAR_ALLOC : new_val_len + 1;
	if (!(v->str_val =
1390
	      v->str_val ? my_realloc(v->str_val, v->alloced_len+1,
1391
				      MYF(MY_WME)) :
1392
	      my_malloc(v->alloced_len+1, MYF(MY_WME))))
1393 1394 1395 1396 1397
	  die("Out of memory");
      }
      v->str_val_len = new_val_len;
      memcpy(v->str_val, p, new_val_len);
      v->str_val[new_val_len] = 0;
unknown's avatar
unknown committed
1398 1399
      v->int_val=atoi(p);
      v->int_dirty=0;
1400 1401
      return 0;
    }
1402

1403 1404 1405 1406
  die("Invalid expr: %s", p);
  return 1;
}

1407

1408
enum enum_operator
1409
{
1410 1411 1412
  DO_DEC,
  DO_INC
};
1413

1414
/*
1415
  Decrease or increase the value of a variable
1416 1417

  SYNOPSIS
1418 1419 1420
    do_modify_var()
    query	called command
    operator    operation to perform on the var
1421 1422 1423

  DESCRIPTION
    dec $var_name
1424
    inc $var_name
1425

1426 1427
*/

unknown's avatar
unknown committed
1428
int do_modify_var(struct st_query *query,
1429
                  enum enum_operator operator)
1430
{
1431
  const char *p= query->first_argument;
1432
  VAR* v;
1433
  if (!*p)
1434
    die("Missing argument to %.*s", query->first_word_len, query->query);
1435
  if (*p != '$')
1436
    die("The argument to %.*s must be a variable (start with $)",
unknown's avatar
unknown committed
1437
        query->first_word_len, query->query);
1438
  v= var_get(p, &p, 1, 0);
1439
  switch (operator) {
1440 1441 1442 1443 1444 1445 1446
  case DO_DEC:
    v->int_val--;
    break;
  case DO_INC:
    v->int_val++;
    break;
  default:
1447
    die("Invalid operator to do_modify_var");
1448 1449 1450
    break;
  }
  v->int_dirty= 1;
1451
  query->last_argument= (char*)++p;
1452 1453 1454
  return 0;
}

1455

unknown's avatar
unknown committed
1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469
/*
  Wrapper for 'system' function

  NOTE
   If mysqltest is executed from cygwin shell, the command will be
   executed in the "windows command interpreter" cmd.exe and we prepend "sh"
   to make it be executed by cygwins "bash". Thus commands like "rm",
   "mkdir" as well as shellscripts can executed by "system" in Windows.

*/

int my_system(DYNAMIC_STRING* ds_cmd)
{
#ifdef __WIN__
unknown's avatar
unknown committed
1470
  /* Dump the command into a sh script file and execute with system */
unknown's avatar
unknown committed
1471
  str_to_file(tmp_sh_name, ds_cmd->str, ds_cmd->length);
unknown's avatar
unknown committed
1472
  return system(tmp_sh_cmd);
unknown's avatar
unknown committed
1473 1474 1475 1476 1477 1478
#else
  return system(ds_cmd->str);
#endif
}


1479 1480 1481 1482 1483 1484 1485 1486 1487 1488
/*

  SYNOPSIS
  do_system
    command	called command

  DESCRIPTION
    system <command>

    Eval the query to expand any $variables in the command.
unknown's avatar
unknown committed
1489
    Execute the command with the "system" command.
1490

unknown's avatar
unknown committed
1491
*/
1492

unknown's avatar
unknown committed
1493
void do_system(struct st_query *command)
unknown's avatar
unknown committed
1494
{
1495
  DYNAMIC_STRING ds_cmd;
unknown's avatar
unknown committed
1496
  DBUG_ENTER("do_system");
1497

1498 1499 1500 1501 1502 1503
  if (strlen(command->first_argument) == 0)
    die("Missing arguments to system, nothing to do!");

  init_dynamic_string(&ds_cmd, 0, strlen(command->first_argument) + 64, 256);

  /* Eval the system command, thus replacing all environment variables */
1504
  do_eval(&ds_cmd, command->first_argument, TRUE);
1505 1506 1507

  DBUG_PRINT("info", ("running system command '%s' as '%s'",
                      command->first_argument, ds_cmd.str));
unknown's avatar
unknown committed
1508
  if (my_system(&ds_cmd))
unknown's avatar
unknown committed
1509
  {
1510 1511
    if (command->abort_on_error)
      die("system command '%s' failed", command->first_argument);
1512

1513 1514 1515 1516
    /* If ! abort_on_error, log message and continue */
    dynstr_append(&ds_res, "system command '");
    replace_dynstr_append(&ds_res, command->first_argument);
    dynstr_append(&ds_res, "' failed\n");
unknown's avatar
unknown committed
1517
  }
1518 1519

  command->last_argument= command->end;
unknown's avatar
unknown committed
1520
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
1521
}
1522

1523 1524 1525

/*
  Print the content between echo and <delimiter> to result file.
1526 1527
  Evaluate all variables in the string before printing, allow
  for variable names to be escaped using \
1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539

  SYNOPSIS
    do_echo()
    q  called command

  DESCRIPTION
    echo text
    Print the text after echo until end of command to result file

    echo $<var_name>
    Print the content of the variable <var_name> to result file

1540 1541 1542 1543 1544 1545
    echo Some text $<var_name>
    Print "Some text" plus the content of the variable <var_name> to
    result file

    echo Some text \$<var_name>
    Print "Some text" plus $<var_name> to result file
1546 1547
*/

1548
int do_echo(struct st_query *command)
1549
{
1550
  DYNAMIC_STRING *ds, ds_echo;
1551

1552
  ds= &ds_res;
1553

1554
  init_dynamic_string(&ds_echo, "", 256, 256);
unknown's avatar
unknown committed
1555
  do_eval(&ds_echo, command->first_argument, FALSE);
1556
  dynstr_append_mem(ds, ds_echo.str, ds_echo.length);
1557
  dynstr_append_mem(ds, "\n", 1);
1558 1559
  dynstr_free(&ds_echo);
  command->last_argument= command->end;
1560 1561 1562
  return 0;
}

1563

1564
int do_sync_with_master2(long offset)
unknown's avatar
unknown committed
1565 1566 1567
{
  MYSQL_RES* res;
  MYSQL_ROW row;
1568
  MYSQL* mysql= &cur_con->mysql;
unknown's avatar
unknown committed
1569
  char query_buf[FN_REFLEN+128];
1570
  int tries= 0;
1571 1572
  int rpl_parse;

1573
  if (!master_pos.file[0])
unknown's avatar
unknown committed
1574
    die("Calling 'sync_with_master' without calling 'save_master_pos'");
1575
  rpl_parse= mysql_rpl_parse_enabled(mysql);
1576
  mysql_disable_rpl_parse(mysql);
unknown's avatar
unknown committed
1577

unknown's avatar
unknown committed
1578
  sprintf(query_buf, "select master_pos_wait('%s', %ld)", master_pos.file,
1579
	  master_pos.pos + offset);
1580 1581 1582

wait_for_position:

1583
  if (mysql_query(mysql, query_buf))
unknown's avatar
unknown committed
1584 1585
    die("failed in %s: %d: %s", query_buf, mysql_errno(mysql),
        mysql_error(mysql));
unknown's avatar
unknown committed
1586

1587
  if (!(res= mysql_store_result(mysql)))
unknown's avatar
unknown committed
1588
    die("mysql_store_result() returned NULL for '%s'", query_buf);
1589
  if (!(row= mysql_fetch_row(res)))
unknown's avatar
unknown committed
1590
    die("empty result in %s", query_buf);
1591
  if (!row[0])
1592 1593 1594 1595 1596 1597
  {
    /*
      It may be that the slave SQL thread has not started yet, though START
      SLAVE has been issued ?
    */
    if (tries++ == 3)
unknown's avatar
unknown committed
1598
      die("could not sync with master ('%s' returned NULL)", query_buf);
1599 1600 1601 1602
    sleep(1); /* So at most we will wait 3 seconds and make 4 tries */
    mysql_free_result(res);
    goto wait_for_position;
  }
unknown's avatar
unknown committed
1603
  mysql_free_result(res);
1604
  if (rpl_parse)
1605
    mysql_enable_rpl_parse(mysql);
unknown's avatar
unknown committed
1606

unknown's avatar
unknown committed
1607 1608 1609
  return 0;
}

1610
int do_sync_with_master(struct st_query *query)
1611
{
1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624
  long offset= 0;
  char *p= query->first_argument;
  const char *offset_start= p;
  if (*offset_start)
  {
    for (; my_isdigit(charset_info, *p); p++)
      offset = offset * 10 + *p - '0';

    if(*p && !my_isspace(charset_info, *p))
      die("Invalid integer argument \"%s\"", offset_start);
    query->last_argument= p;
  }
  return do_sync_with_master2(offset);
1625
}
1626

unknown's avatar
unknown committed
1627 1628 1629 1630
/*
  when ndb binlog is on, this call will wait until last updated epoch
  (locally in the mysqld) has been received into the binlog
*/
unknown's avatar
unknown committed
1631 1632 1633 1634 1635
int do_save_master_pos()
{
  MYSQL_RES* res;
  MYSQL_ROW row;
  MYSQL* mysql = &cur_con->mysql;
1636
  const char *query;
1637 1638 1639 1640
  int rpl_parse;

  rpl_parse = mysql_rpl_parse_enabled(mysql);
  mysql_disable_rpl_parse(mysql);
unknown's avatar
unknown committed
1641

unknown's avatar
unknown committed
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724
#ifdef HAVE_NDB_BINLOG
  /*
     Wait for ndb binlog to be up-to-date with all changes
     done on the local mysql server
  */
  {
    ulong have_ndbcluster;
    if (mysql_query(mysql, query= "show variables like 'have_ndbcluster'"))
      die("At line %u: failed in %s: %d: %s", start_lineno, query,
          mysql_errno(mysql), mysql_error(mysql));
    if (!(res= mysql_store_result(mysql)))
      die("line %u: mysql_store_result() retuned NULL for '%s'", start_lineno,
          query);
    if (!(row= mysql_fetch_row(res)))
      die("line %u: empty result in %s", start_lineno, query);

    have_ndbcluster= strcmp("YES", row[1]) == 0;
    mysql_free_result(res);

    if (have_ndbcluster)
    {
      ulonglong epoch, tmp_epoch= 0;
      int count= 0;

      do
      {
        const char binlog[]= "binlog";
        const char latest_trans_epoch[]=
          "latest_trans_epoch=";
        const char latest_applied_binlog_epoch[]=
          "latest_applied_binlog_epoch=";
        if (count)
          sleep(1);
        if (mysql_query(mysql, query= "show engine ndb status"))
          die("At line %u: failed in '%s': %d: %s", start_lineno, query,
              mysql_errno(mysql), mysql_error(mysql));
        if (!(res= mysql_store_result(mysql)))
          die("line %u: mysql_store_result() retuned NULL for '%s'",
              start_lineno, query);
        while ((row= mysql_fetch_row(res)))
        {
          if (strcmp(row[1], binlog) == 0)
          {
            const char *status= row[2];
            /* latest_trans_epoch */
            if (count == 0)
            {
              while (*status && strncmp(status, latest_trans_epoch,
                                        sizeof(latest_trans_epoch)-1))
                status++;
              if (*status)
              {
                status+= sizeof(latest_trans_epoch)-1;
                epoch= strtoull(status, (char**) 0, 10);
              }
              else
                die("line %u: result does not contain '%s' in '%s'",
                    start_lineno, latest_trans_epoch, query);
            }
            /* latest_applied_binlog_epoch */
            while (*status && strncmp(status, latest_applied_binlog_epoch,
                                      sizeof(latest_applied_binlog_epoch)-1))
              status++;
            if (*status)
            {
              status+= sizeof(latest_applied_binlog_epoch)-1;
              tmp_epoch= strtoull(status, (char**) 0, 10);
            }
            else
              die("line %u: result does not contain '%s' in '%s'",
                  start_lineno, latest_applied_binlog_epoch, query);
            break;
          }
        }
        mysql_free_result(res);
        if (!row)
          die("line %u: result does not contain '%s' in '%s'",
              start_lineno, binlog, query);
        count++;
      } while (tmp_epoch < epoch && count <= 3);
    }
  }
#endif
1725
  if (mysql_query(mysql, query= "show master status"))
1726
    die("failed in show master status: %d: %s",
unknown's avatar
unknown committed
1727 1728
	mysql_errno(mysql), mysql_error(mysql));

1729
  if (!(res = mysql_store_result(mysql)))
unknown's avatar
unknown committed
1730
    die("mysql_store_result() retuned NULL for '%s'", query);
1731
  if (!(row = mysql_fetch_row(res)))
unknown's avatar
unknown committed
1732
    die("empty result in show master status");
1733 1734
  strnmov(master_pos.file, row[0], sizeof(master_pos.file)-1);
  master_pos.pos = strtoul(row[1], (char**) 0, 10);
1735
  mysql_free_result(res);
unknown's avatar
unknown committed
1736

1737
  if (rpl_parse)
1738
    mysql_enable_rpl_parse(mysql);
unknown's avatar
unknown committed
1739

unknown's avatar
unknown committed
1740 1741 1742 1743
  return 0;
}


1744 1745 1746 1747 1748
/*
  Assign the variable <var_name> with <var_val>

  SYNOPSIS
   do_let()
1749
    query	called command
1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762

  DESCRIPTION
    let $<var_name>=<var_val><delimiter>

    <var_name>  - is the string string found between the $ and =
    <var_val>   - is the content between the = and <delimiter>, it may span
                  multiple line and contain any characters except <delimiter>
    <delimiter> - is a string containing of one or more chars, default is ;

  RETURN VALUES
   Program will die if error detected
*/

1763
int do_let(struct st_query *query)
1764
{
1765
  char *p= query->first_argument;
1766
  char *var_name, *var_name_end, *var_val_start;
1767 1768

  /* Find <var_name> */
1769
  if (!*p)
1770
    die("Missing arguments to let");
1771 1772
  var_name= p;
  while (*p && (*p != '=') && !my_isspace(charset_info,*p))
1773
    p++;
1774
  var_name_end= p;
unknown's avatar
unknown committed
1775 1776
  if (var_name == var_name_end ||
      (var_name+1 == var_name_end && *var_name == '$'))
1777
    die("Missing variable name in let");
1778
  while (my_isspace(charset_info,*p))
1779
    p++;
1780
  if (*p++ != '=')
1781 1782 1783
    die("Missing assignment operator in let");

  /* Find start of <var_val> */
1784
  while (*p && my_isspace(charset_info,*p))
1785
    p++;
1786 1787
  var_val_start= p;
  query->last_argument= query->end;
1788
  /* Assign var_val to var_name */
1789
  return var_set(var_name, var_name_end, var_val_start, query->end);
1790 1791
}

1792 1793 1794 1795 1796 1797 1798 1799

/*
  Store an integer (typically the returncode of the last SQL)
  statement in the mysqltest builtin variable $mysql_errno, by
  simulating of a user statement "let $mysql_errno= <integer>"
*/

int var_set_errno(int sql_errno)
1800
{
1801 1802 1803 1804
  const char *var_name= "$mysql_errno";
  char var_val[21];
  uint length= my_sprintf(var_val, (var_val, "%d", sql_errno));
  return var_set(var_name, var_name + 12, var_val, var_val + length);
1805 1806
}

1807

1808
int do_rpl_probe(struct st_query *query __attribute__((unused)))
1809
{
unknown's avatar
unknown committed
1810
  DBUG_ENTER("do_rpl_probe");
1811
  if (mysql_rpl_probe(&cur_con->mysql))
unknown's avatar
unknown committed
1812 1813
    die("Failed in mysql_rpl_probe(): '%s'", mysql_error(&cur_con->mysql));
  DBUG_RETURN(0);
1814 1815
}

1816

1817
int do_enable_rpl_parse(struct st_query *query __attribute__((unused)))
1818 1819 1820 1821 1822
{
  mysql_enable_rpl_parse(&cur_con->mysql);
  return 0;
}

1823

1824
int do_disable_rpl_parse(struct st_query *query __attribute__((unused)))
1825 1826 1827 1828 1829 1830
{
  mysql_disable_rpl_parse(&cur_con->mysql);
  return 0;
}


1831 1832 1833 1834 1835 1836 1837
/*
  Sleep the number of specifed seconds

  SYNOPSIS
   do_sleep()
    q	       called command
    real_sleep  use the value from opt_sleep as number of seconds to sleep
unknown's avatar
unknown committed
1838
                if real_sleep is false
1839 1840 1841

  DESCRIPTION
    sleep <seconds>
unknown's avatar
unknown committed
1842 1843 1844 1845 1846 1847 1848 1849 1850
    real_sleep <seconds>

  The difference between the sleep and real_sleep commands is that sleep
  uses the delay from the --sleep command-line option if there is one.
  (If the --sleep option is not given, the sleep command uses the delay
  specified by its argument.) The real_sleep command always uses the
  delay specified by its argument.  The logic is that sometimes delays are
  cpu-dependent, and --sleep can be used to set this delay.  real_sleep is
  used for cpu-independent delays.
1851 1852
*/

1853
int do_sleep(struct st_query *query, my_bool real_sleep)
unknown's avatar
unknown committed
1854
{
1855 1856 1857 1858 1859 1860
  int error= 0;
  char *p= query->first_argument;
  char *sleep_start, *sleep_end= query->end;
  double sleep_val;

  while (my_isspace(charset_info, *p))
unknown's avatar
unknown committed
1861
    p++;
1862
  if (!*p)
unknown's avatar
unknown committed
1863
    die("Missing argument to %.*s", query->first_word_len, query->query);
1864 1865 1866
  sleep_start= p;
  /* Check that arg starts with a digit, not handled by my_strtod */
  if (!my_isdigit(charset_info, *sleep_start))
unknown's avatar
unknown committed
1867 1868
    die("Invalid argument to %.*s \"%s\"", query->first_word_len, query->query,
		query->first_argument);
1869 1870
  sleep_val= my_strtod(sleep_start, &sleep_end, &error);
  if (error)
unknown's avatar
unknown committed
1871 1872
    die("Invalid argument to %.*s \"%s\"", query->first_word_len, query->query,
		query->first_argument);
1873 1874

  /* Fixed sleep time selected by --sleep option */
1875
  if (opt_sleep && !real_sleep)
1876 1877
    sleep_val= opt_sleep;

1878
  DBUG_PRINT("info", ("sleep_val: %f", sleep_val));
1879 1880
  my_sleep((ulong) (sleep_val * 1000000L));
  query->last_argument= sleep_end;
unknown's avatar
unknown committed
1881
  return 0;
unknown's avatar
unknown committed
1882 1883
}

1884
static void get_file_name(char *filename, struct st_query *q)
1885
{
1886
  char *p= q->first_argument, *name;
1887 1888
  if (!*p)
    die("Missing file name argument");
1889
  name= p;
1890 1891 1892 1893
  while (*p && !my_isspace(charset_info,*p))
    p++;
  if (*p)
    *p++= 0;
1894
  q->last_argument= p;
1895
  strmake(filename, name, FN_REFLEN);
1896 1897
}

1898
static void set_charset(struct st_query *q)
unknown's avatar
unknown committed
1899
{
1900 1901
  char *charset_name= q->first_argument;
  char *p;
unknown's avatar
unknown committed
1902 1903

  if (!charset_name || !*charset_name)
1904
    die("Missing charset name in 'character_set'");
unknown's avatar
unknown committed
1905
  /* Remove end space */
1906 1907 1908 1909 1910
  p= charset_name;
  while (*p && !my_isspace(charset_info,*p))
    p++;
  if(*p)
    *p++= 0;
1911
  q->last_argument= p;
unknown's avatar
unknown committed
1912 1913 1914 1915
  charset_info= get_charset_by_csname(charset_name,MY_CS_PRIMARY,MYF(MY_WME));
  if (!charset_info)
    abort_not_supported_test();
}
unknown's avatar
unknown committed
1916

1917
static uint get_errcodes(match_err *to,struct st_query *q)
unknown's avatar
unknown committed
1918
{
1919
  char *p= q->first_argument;
1920
  uint count= 0;
1921

1922
  DBUG_ENTER("get_errcodes");
1923

unknown's avatar
unknown committed
1924
  if (!*p)
1925
    die("Missing argument in %s", q->query);
1926

1927
  do
1928
  {
1929 1930 1931
    if (*p == 'S')
    {
      /* SQLSTATE string */
1932 1933 1934 1935 1936 1937 1938
      char *end= ++p + SQLSTATE_LENGTH;
      char *to_ptr= to[count].code.sqlstate;

      for (; my_isalnum(charset_info, *p) && p != end; p++)
	*to_ptr++= *p;
      *to_ptr= 0;

1939 1940
      to[count].type= ERR_SQLSTATE;
    }
1941 1942 1943 1944 1945
    else if (*p == 'E')
    {
      /* SQL error as string */
      st_error *e= global_error;
      char *start= p++;
unknown's avatar
unknown committed
1946

1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958
      for (; *p == '_' || my_isalnum(charset_info, *p); p++)
	;
      for (; e->name; e++)
      {
	if (!strncmp(start, e->name, (int) (p - start)))
	{
	  to[count].code.errnum= (uint) e->code;
	  to[count].type= ERR_ERRNO;
	  break;
	}
      }
      if (!e->name)
1959
	die("Unknown SQL error '%s'", start);
1960
    }
1961 1962 1963
    else
    {
      long val;
1964 1965

      if (!(p= str2int(p,10,(long) INT_MIN, (long) INT_MAX, &val)))
1966
	die("Invalid argument in %s", q->query);
1967 1968 1969
      to[count].code.errnum= (uint) val;
      to[count].type= ERR_ERRNO;
    }
unknown's avatar
unknown committed
1970
    count++;
1971
  } while (*(p++) == ',');
1972
  q->last_argument= (p - 1);
1973
  to[count].type= ERR_EMPTY;                        /* End of data */
unknown's avatar
unknown committed
1974
  DBUG_RETURN(count);
unknown's avatar
unknown committed
1975 1976
}

unknown's avatar
unknown committed
1977 1978 1979
/*
  Get a string;  Return ptr to end of string
  Strings may be surrounded by " or '
1980 1981

  If string is a '$variable', return the value of the variable.
unknown's avatar
unknown committed
1982 1983 1984
*/


1985
static char *get_string(char **to_ptr, char **from_ptr,
1986
			struct st_query *q)
unknown's avatar
unknown committed
1987 1988
{
  reg1 char c,sep;
1989
  char *to= *to_ptr, *from= *from_ptr, *start=to;
unknown's avatar
unknown committed
1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033
  DBUG_ENTER("get_string");

  /* Find separator */
  if (*from == '"' || *from == '\'')
    sep= *from++;
  else
    sep=' ';				/* Separated with space */

  for ( ; (c=*from) ; from++)
  {
    if (c == '\\' && from[1])
    {					/* Escaped character */
      /* We can't translate \0 -> ASCII 0 as replace can't handle ASCII 0 */
      switch (*++from) {
      case 'n':
	*to++= '\n';
	break;
      case 't':
	*to++= '\t';
	break;
      case 'r':
	*to++ = '\r';
	break;
      case 'b':
	*to++ = '\b';
	break;
      case 'Z':				/* ^Z must be escaped on Win32 */
	*to++='\032';
	break;
      default:
	*to++ = *from;
	break;
      }
    }
    else if (c == sep)
    {
      if (c == ' ' || c != *++from)
	break;				/* Found end of string */
      *to++=c;				/* Copy duplicated separator */
    }
    else
      *to++=c;
  }
  if (*from != ' ' && *from)
2034
    die("Wrong string argument in %s", q->query);
unknown's avatar
unknown committed
2035

2036
  while (my_isspace(charset_info,*from))	/* Point to next string */
unknown's avatar
unknown committed
2037 2038
    from++;

2039 2040
  *to =0;				/* End of string marker */
  *to_ptr= to+1;			/* Store pointer to end */
unknown's avatar
unknown committed
2041
  *from_ptr= from;
2042 2043 2044 2045 2046 2047 2048 2049

  /* Check if this was a variable */
  if (*start == '$')
  {
    const char *end= to;
    VAR *var=var_get(start, &end, 0, 1);
    if (var && to == (char*) end+1)
    {
2050
      DBUG_PRINT("info",("var: '%s' -> '%s'", start, var->str_val));
2051 2052 2053 2054
      DBUG_RETURN(var->str_val);	/* return found variable value */
    }
  }
  DBUG_RETURN(start);
unknown's avatar
unknown committed
2055 2056
}

2057 2058 2059 2060 2061
/*
  Finds the next (non-escaped) '/' in the expression.
  (If the character '/' is needed, it can be escaped using '\'.)
*/

2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084
#define PARSE_REGEX_ARG \
  while (p < expr_end) \
  {\
    char c= *p;\
    if (c == '/')\
    {\
      if (last_c == '\\')\
      {\
        buf_p[-1]= '/';\
      }\
      else\
      {\
        *buf_p++ = 0;\
        break;\
      }  \
    }  \
    else\
      *buf_p++ = c;\
       \
    last_c= c;\
    p++;\
  }  \

2085 2086 2087 2088 2089 2090 2091
/*
  Initializes the regular substitution expression to be used in the 
  result output of test.

  Returns: st_replace_regex struct with pairs of substitutions
*/
  
2092 2093 2094 2095 2096 2097 2098 2099 2100
static struct st_replace_regex* init_replace_regex(char* expr)
{
  struct st_replace_regex* res;
  char* buf,*expr_end;
  char* p;
  char* buf_p;
  uint expr_len= strlen(expr);
  char last_c = 0;
  struct st_regex reg;
2101 2102 2103 2104
 
  /* my_malloc() will die on fail with MY_FAE */    
  res=(struct st_replace_regex*)my_malloc(
              sizeof(*res)+expr_len ,MYF(MY_FAE+MY_WME));
2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133
  my_init_dynamic_array(&res->regex_arr,sizeof(struct st_regex),128,128);
    
  buf= (char*)res + sizeof(*res);
  expr_end= expr + expr_len;
  p= expr;
  buf_p= buf;  
   
  /* for each regexp substitution statement */
  while (p < expr_end)
  {
    bzero(&reg,sizeof(reg));
    /* find the start of the statement */
    while (p < expr_end)
    {
      if (*p == '/')
        break;
      p++;  
    }  
    
    if (p == expr_end || ++p == expr_end)
    {
      if (res->regex_arr.elements)
        break;
      else  
        goto err;
    }
    /* we found the start */  
    reg.pattern= buf_p;
    
2134
    /* Find first argument -- pattern string to be removed */
2135 2136 2137 2138 2139 2140 2141 2142
    PARSE_REGEX_ARG
    
    if (p == expr_end || ++p == expr_end)
      goto err;
  
    /* buf_p now points to the replacement pattern terminated with \0 */  
    reg.replace= buf_p;  
    
2143
    /* Find second argument -- replace string to replace pattern */
2144 2145 2146 2147 2148 2149 2150 2151
    PARSE_REGEX_ARG
    
    if (p == expr_end)
      goto err;
    
    /* skip the ending '/' in the statement */    
    p++;
    
2152
    /* Check if we should do matching case insensitive */
2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168
    if (p < expr_end && *p == 'i')
      reg.icase= 1;    
    
    /* done parsing the statement, now place it in regex_arr */
    if (insert_dynamic(&res->regex_arr,(gptr) &reg))
      die("Out of memory");
  }  
  res->odd_buf_len= res->even_buf_len= 8192;
  res->even_buf= (char*)my_malloc(res->even_buf_len,MYF(MY_WME+MY_FAE));  
  res->odd_buf= (char*)my_malloc(res->odd_buf_len,MYF(MY_WME+MY_FAE));  
  res->buf= res->even_buf;
        
  return res;  
  
err:
  my_free((gptr)res,0);
2169
  die("Error parsing replace_regex \"%s\"", expr);
2170 2171 2172
  return 0;    
}

2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187
/*  
   Execute all substitutions on val.

   Returns: true if substituition was made, false otherwise
   Side-effect: Sets r->buf to be the buffer with all substitutions done.
   
   IN: 
     struct st_replace_regex* r
     char* val
   Out: 
     struct st_replace_regex* r
     r->buf points at the resulting buffer
     r->even_buf and r->odd_buf might have been reallocated  
     r->even_buf_len and r->odd_buf_len might have been changed
     
2188 2189
  TODO:  at some point figure out if there is a way to do everything
         in one pass 
2190
*/
2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202

static int multi_reg_replace(struct st_replace_regex* r,char* val)
{
  uint i;
  char* in_buf, *out_buf;
  int* buf_len_p;
  
  in_buf= val;
  out_buf= r->even_buf;
  buf_len_p= &r->even_buf_len;
  r->buf= 0;
  
2203
  /* For each substitution, do the replace */
2204 2205 2206 2207 2208 2209 2210
  for (i= 0; i < r->regex_arr.elements; i++)
  {
    struct st_regex re;
    char* save_out_buf= out_buf;
    
    get_dynamic(&r->regex_arr,(gptr)&re,i);
    
2211 2212
    if (!reg_replace(&out_buf, buf_len_p, re.pattern, re.replace,
       in_buf, re.icase))
2213
    {
2214
      /* if the buffer has been reallocated, make adjustements */
2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236
      if (save_out_buf != out_buf)
      {
        if (save_out_buf == r->even_buf)
          r->even_buf= out_buf;
        else
          r->odd_buf= out_buf;  
      }
        
      r->buf= out_buf;
      if (in_buf == val)
        in_buf= r->odd_buf;
        
      swap_variables(char*,in_buf,out_buf);
      
      buf_len_p= (out_buf == r->even_buf) ? &r->even_buf_len :
          &r->odd_buf_len;
    }   
  }
  
  return (r->buf == 0);
}

2237 2238 2239 2240 2241 2242 2243 2244 2245
/*
  Parse the regular expression to be used in all result files
  from now on.
  
  The syntax is --replace_regex /from/to/i /from/to/i ...
  i means case-insensitive match. If omitted, the match is 
  case-sensitive
  
*/
2246 2247 2248 2249 2250 2251 2252 2253 2254
static void get_replace_regex(struct st_query *q)
{
  char *expr= q->first_argument;
  free_replace_regex();
  if (!(glob_replace_regex=init_replace_regex(expr)))
    die("Could not init replace_regex");
  q->last_argument= q->end;  
}

unknown's avatar
unknown committed
2255 2256 2257 2258 2259

/*
  Get arguments for replace. The syntax is:
  replace from to [from to ...]
  Where each argument may be quoted with ' or "
2260 2261
  A argument may also be a variable, in which case the value of the
  variable is replaced.
unknown's avatar
unknown committed
2262 2263 2264 2265 2266
*/

static void get_replace(struct st_query *q)
{
  uint i;
2267
  char *from= q->first_argument;
2268
  char *buff,*start;
unknown's avatar
unknown committed
2269 2270 2271 2272
  char word_end_chars[256],*pos;
  POINTER_ARRAY to_array,from_array;
  DBUG_ENTER("get_replace");

unknown's avatar
unknown committed
2273
  free_replace();
2274

unknown's avatar
unknown committed
2275 2276 2277
  bzero((char*) &to_array,sizeof(to_array));
  bzero((char*) &from_array,sizeof(from_array));
  if (!*from)
2278
    die("Missing argument in %s", q->query);
2279
  start=buff=my_malloc(strlen(from)+1,MYF(MY_WME | MY_FAE));
unknown's avatar
unknown committed
2280 2281 2282
  while (*from)
  {
    char *to=buff;
2283
    to=get_string(&buff, &from, q);
unknown's avatar
unknown committed
2284
    if (!*from)
2285
      die("Wrong number of arguments to replace_result in '%s'", q->query);
unknown's avatar
unknown committed
2286
    insert_pointer_name(&from_array,to);
2287
    to=get_string(&buff, &from, q);
unknown's avatar
unknown committed
2288 2289 2290
    insert_pointer_name(&to_array,to);
  }
  for (i=1,pos=word_end_chars ; i < 256 ; i++)
2291
    if (my_isspace(charset_info,i))
unknown's avatar
unknown committed
2292
      *pos++= i;
2293
  *pos=0;					/* End pointer */
unknown's avatar
unknown committed
2294 2295 2296
  if (!(glob_replace=init_replace((char**) from_array.typelib.type_names,
				  (char**) to_array.typelib.type_names,
				  (uint) from_array.typelib.count,
2297
				  word_end_chars)))
2298
    die("Can't initialize replace from '%s'", q->query);
unknown's avatar
unknown committed
2299 2300
  free_pointer_array(&from_array);
  free_pointer_array(&to_array);
2301
  my_free(start, MYF(0));
2302
  q->last_argument= q->end;
2303
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
2304 2305
}

2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317
static void free_replace_regex()
{
  if (glob_replace_regex)
  {
    my_free(glob_replace_regex->even_buf,MYF(MY_ALLOW_ZERO_PTR));
    my_free(glob_replace_regex->odd_buf,MYF(MY_ALLOW_ZERO_PTR));
    my_free((char*) glob_replace_regex,MYF(0));
    glob_replace_regex=0;
  }
}


unknown's avatar
unknown committed
2318 2319
void free_replace()
{
2320
  DBUG_ENTER("free_replace");
unknown's avatar
unknown committed
2321 2322 2323 2324 2325
  if (glob_replace)
  {
    my_free((char*) glob_replace,MYF(0));
    glob_replace=0;
  }
2326
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
2327 2328
}

unknown's avatar
unknown committed
2329
struct connection * find_connection_by_name(const char *name)
unknown's avatar
unknown committed
2330 2331
{
  struct connection *con;
2332
  for (con= cons; con < next_con; con++)
2333
  {
2334
    if (!strcmp(con->name, name))
2335
    {
unknown's avatar
unknown committed
2336
      return con;
2337 2338
    }
  }
unknown's avatar
unknown committed
2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350
  return 0; /* Connection not found */
}


int select_connection_name(const char *name)
{
  DBUG_ENTER("select_connection2");
  DBUG_PRINT("enter",("name: '%s'", name));

  if (!(cur_con= find_connection_by_name(name)))
    die("connection '%s' not found in connection pool", name);
  DBUG_RETURN(0);
unknown's avatar
unknown committed
2351 2352
}

2353 2354

int select_connection(struct st_query *query)
2355
{
2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372
  char *name;
  char *p= query->first_argument;
  DBUG_ENTER("select_connection");

  if (!*p)
    die("Missing connection name in connect");
  name= p;
  while (*p && !my_isspace(charset_info,*p))
    p++;
  if (*p)
    *p++= 0;
  query->last_argument= p;
  return select_connection_name(name);
}


int close_connection(struct st_query *q)
2373
{
2374
  char *p= q->first_argument, *name;
2375
  struct connection *con;
2376 2377 2378
  DBUG_ENTER("close_connection");
  DBUG_PRINT("enter",("name: '%s'",p));

2379
  if (!*p)
unknown's avatar
unknown committed
2380
    die("Missing connection name in disconnect");
2381
  name= p;
2382
  while (*p && !my_isspace(charset_info,*p))
2383
    p++;
2384

2385 2386
  if (*p)
    *p++= 0;
2387 2388
  q->last_argument= p;
  for (con= cons; con < next_con; con++)
2389
  {
2390
    if (!strcmp(con->name, name))
2391
    {
2392
#ifndef EMBEDDED_LIBRARY
unknown's avatar
unknown committed
2393 2394 2395
      if (q->type == Q_DIRTY_CLOSE)
      {
	if (con->mysql.net.vio)
unknown's avatar
unknown committed
2396
	{
unknown's avatar
unknown committed
2397 2398
	  vio_delete(con->mysql.net.vio);
	  con->mysql.net.vio = 0;
unknown's avatar
unknown committed
2399
	}
unknown's avatar
unknown committed
2400
      }
2401
#endif
2402
      mysql_close(&con->mysql);
2403 2404 2405
      if (con->util_mysql)
	mysql_close(con->util_mysql);
      con->util_mysql= 0;
unknown's avatar
unknown committed
2406 2407 2408 2409 2410 2411 2412 2413
      my_free(con->name, MYF(0));
      /*
         When the connection is closed set name to "closed_connection"
         to make it possible to reuse the connection name.
         The connection slot will not be reused
       */
      if (!(con->name = my_strdup("closed_connection", MYF(MY_WME))))
        die("Out of memory");
2414 2415 2416
      DBUG_RETURN(0);
    }
  }
2417
  die("connection '%s' not found in connection pool", name);
2418
  DBUG_RETURN(1);				/* Never reached */
2419 2420
}

unknown's avatar
unknown committed
2421

2422 2423
/*
   This one now is a hack - we may want to improve in in the
unknown's avatar
unknown committed
2424 2425 2426
   future to handle quotes. For now we assume that anything that is not
   a comma, a space or ) belongs to the argument. space is a chopper, comma or
   ) are delimiters/terminators
unknown's avatar
unknown committed
2427 2428 2429 2430

  SYNOPSIS
  safe_get_param
  str - string to get param from
unknown's avatar
unknown committed
2431
  arg - pointer to string where result will be stored
unknown's avatar
unknown committed
2432 2433
  msg - Message to display if param is not found
       if msg is 0 this param is not required and param may be empty
unknown's avatar
unknown committed
2434

unknown's avatar
unknown committed
2435 2436
  RETURNS
  pointer to str after param
unknown's avatar
unknown committed
2437

2438
*/
2439

unknown's avatar
unknown committed
2440
char* safe_get_param(char *str, char** arg, const char *msg)
unknown's avatar
unknown committed
2441
{
2442
  DBUG_ENTER("safe_get_param");
unknown's avatar
unknown committed
2443 2444
  if(!*str)
  {
unknown's avatar
unknown committed
2445
    if (msg)
unknown's avatar
unknown committed
2446 2447 2448 2449
      die(msg);
    *arg= str;
    DBUG_RETURN(str);
  }
2450
  while (*str && my_isspace(charset_info,*str))
unknown's avatar
unknown committed
2451
    str++;
2452
  *arg= str;
unknown's avatar
unknown committed
2453 2454
  while (*str && *str != ',' && *str != ')')
    str++;
unknown's avatar
unknown committed
2455
  if (msg && !*arg)
unknown's avatar
unknown committed
2456
    die(msg);
2457

2458
  *str++= 0;
2459
  DBUG_RETURN(str);
unknown's avatar
unknown committed
2460 2461
}

2462
#ifndef EMBEDDED_LIBRARY
2463 2464 2465 2466 2467 2468 2469 2470 2471 2472
void init_manager()
{
  if (!(manager=mysql_manager_init(0)))
    die("Failed in mysql_manager_init()");
  if (!mysql_manager_connect(manager,manager_host,manager_user,
			     manager_pass,manager_port))
    die("Could not connect to MySQL manager: %s(%d)",manager->last_error,
	manager->last_errno);

}
2473
#endif
2474

2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497

/*
  Connect to a server doing several retries if needed.

  SYNOPSIS
    safe_connect()
      con               - connection structure to be used
      host, user, pass, - connection parameters
      db, port, sock

  NOTE
    This function will try to connect to the given server MAX_CON_TRIES
    times and sleep CON_RETRY_SLEEP seconds between attempts before
    finally giving up. This helps in situation when the client starts
    before the server (which happens sometimes).
    It will ignore any errors during these retries. One should use
    connect_n_handle_errors() if he expects a connection error and wants
    handle as if it was an error from a usual statement.

  RETURN VALUE
    0 - success, non-0 - failure
*/

2498 2499
int safe_connect(MYSQL* mysql, const char *host, const char *user,
		 const char *pass, const char *db, int port, const char *sock)
unknown's avatar
unknown committed
2500
{
2501 2502
  int con_error= 1;
  my_bool reconnect= 1;
unknown's avatar
unknown committed
2503
  int i;
2504
  for (i= 0; i < MAX_CON_TRIES; ++i)
unknown's avatar
unknown committed
2505
  {
2506
    if (mysql_real_connect(mysql, host,user, pass, db, port, sock,
2507
			   CLIENT_MULTI_STATEMENTS | CLIENT_REMEMBER_OPTIONS))
unknown's avatar
unknown committed
2508
    {
2509
      con_error= 0;
unknown's avatar
unknown committed
2510 2511 2512 2513
      break;
    }
    sleep(CON_RETRY_SLEEP);
  }
2514 2515 2516 2517
  /*
   TODO: change this to 0 in future versions, but the 'kill' test relies on
   existing behavior
  */
2518
  mysql_options(mysql, MYSQL_OPT_RECONNECT, (char *)&reconnect);
unknown's avatar
unknown committed
2519 2520 2521
  return con_error;
}

2522

2523
/*
unknown's avatar
unknown committed
2524
  Connect to a server and handle connection errors in case they occur.
2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550

  SYNOPSIS
    connect_n_handle_errors()
      q                 - context of connect "query" (command)
      con               - connection structure to be used
      host, user, pass, - connection parameters
      db, port, sock
      create_conn       - out parameter, set to zero if connection was
                          not established and is not touched otherwise

  DESCRIPTION
    This function will try to establish a connection to server and handle
    possible errors in the same manner as if "connect" was usual SQL-statement
    (If error is expected it will ignore it once it occurs and log the
    "statement" to the query log).
    Unlike safe_connect() it won't do several attempts.

  RETURN VALUE
    0 - success, non-0 - failure
*/

int connect_n_handle_errors(struct st_query *q, MYSQL* con, const char* host,
                            const char* user, const char* pass,
                            const char* db, int port, const char* sock,
                            int* create_conn)
{
2551
  DYNAMIC_STRING *ds;
2552
  my_bool reconnect= 1;
2553 2554
  int error= 0;

2555
  ds= &ds_res;
2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587

  if (!disable_query_log)
  {
    /*
      It is nice to have connect() statement logged in result file
      in this case.
      QQ: Should we do this only if we are expecting an error ?
    */
    char port_buff[22]; /* This should be enough for any int */
    char *port_end;
    dynstr_append_mem(ds, "connect(", 8);
    replace_dynstr_append(ds, host);
    dynstr_append_mem(ds, ",", 1);
    replace_dynstr_append(ds, user);
    dynstr_append_mem(ds, ",", 1);
    replace_dynstr_append(ds, pass);
    dynstr_append_mem(ds, ",", 1);
    if (db)
      replace_dynstr_append(ds, db);
    dynstr_append_mem(ds, ",", 1);
    port_end= int10_to_str(port, port_buff, 10);
    replace_dynstr_append_mem(ds, port_buff, port_end - port_buff);
    dynstr_append_mem(ds, ",", 1);
    if (sock)
      replace_dynstr_append(ds, sock);
    dynstr_append_mem(ds, ")", 1);
    dynstr_append_mem(ds, delimiter, delimiter_length);
    dynstr_append_mem(ds, "\n", 1);
  }
  if (!mysql_real_connect(con, host, user, pass, db, port, sock ? sock: 0,
                          CLIENT_MULTI_STATEMENTS))
  {
2588 2589
    handle_error("connect", q, mysql_errno(con), mysql_error(con),
		 mysql_sqlstate(con), ds);
2590 2591 2592
    *create_conn= 0;
    goto err;
  }
2593

2594
  handle_no_error(q);
2595

2596 2597 2598 2599
  /*
   TODO: change this to 0 in future versions, but the 'kill' test relies on
   existing behavior
  */
2600
  mysql_options(con, MYSQL_OPT_RECONNECT, (char *)&reconnect);
2601

2602 2603
err:
  free_replace();
2604
  free_replace_regex();
2605 2606 2607 2608
  return error;
}


unknown's avatar
unknown committed
2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632
/*
  Open a new connection to MySQL Server with the parameters
  specified

  SYNOPSIS
   do_connect()
    q	       called command

  DESCRIPTION
    connect(<name>,<host>,<user>,<pass>,<db>,[<port>,<sock>[<opts>]]);

      <name> - name of the new connection
      <host> - hostname of server
      <user> - user to connect as
      <pass> - password used when connecting
      <db>   - initial db when connected
      <port> - server port
      <sock> - server socket
      <opts> - options to use for the connection
               SSL - use SSL if available
               COMPRESS - use compression if available

 */

2633
int do_connect(struct st_query *q)
unknown's avatar
unknown committed
2634
{
2635
  char *con_name, *con_user,*con_pass, *con_host, *con_port_str,
unknown's avatar
unknown committed
2636 2637
    *con_db, *con_sock, *con_options;
  char *con_buf, *p;
2638
  char buff[FN_REFLEN];
unknown's avatar
unknown committed
2639
  int con_port;
unknown's avatar
unknown committed
2640 2641
  bool con_ssl= 0;
  bool con_compress= 0;
2642
  int free_con_sock= 0;
2643 2644
  int error= 0;
  int create_conn= 1;
2645
  VAR *var_port, *var_sock;
unknown's avatar
unknown committed
2646

2647
  DBUG_ENTER("do_connect");
2648
  DBUG_PRINT("enter",("connect: %s", q->first_argument));
unknown's avatar
unknown committed
2649

unknown's avatar
unknown committed
2650 2651 2652 2653 2654
  /* Make a copy of query before parsing, safe_get_param will modify */
  if (!(con_buf= my_strdup(q->first_argument, MYF(MY_WME))))
    die("Could not allocate con_buf");
  p= con_buf;

2655
  if (*p != '(')
2656
    die("Syntax error in connect - expected '(' found '%c'", *p);
unknown's avatar
unknown committed
2657
  p++;
unknown's avatar
unknown committed
2658 2659 2660 2661 2662
  p= safe_get_param(p, &con_name, "Missing connection name");
  p= safe_get_param(p, &con_host, "Missing connection host");
  p= safe_get_param(p, &con_user, "Missing connection user");
  p= safe_get_param(p, &con_pass, "Missing connection password");
  p= safe_get_param(p, &con_db, "Missing connection db");
unknown's avatar
unknown committed
2663 2664

  /* Port */
unknown's avatar
unknown committed
2665
  p= safe_get_param(p, &con_port_str, 0);
unknown's avatar
unknown committed
2666
  if (*con_port_str)
unknown's avatar
unknown committed
2667
  {
2668 2669
    if (*con_port_str == '$')
    {
2670
      if (!(var_port= var_get(con_port_str, 0, 0, 0)))
unknown's avatar
unknown committed
2671
        die("Unknown variable '%s'", con_port_str+1);
2672
      con_port= var_port->int_val;
2673 2674
    }
    else
unknown's avatar
unknown committed
2675
    {
2676
      con_port= atoi(con_port_str);
unknown's avatar
unknown committed
2677 2678 2679 2680 2681 2682 2683 2684 2685 2686
      if (con_port == 0)
        die("Illegal argument for port: '%s'", con_port_str);
    }
  }
  else
  {
    con_port= port;
  }

  /* Sock */
unknown's avatar
unknown committed
2687
  p= safe_get_param(p, &con_sock, 0);
unknown's avatar
unknown committed
2688 2689
  if (*con_sock)
  {
2690 2691
    if (*con_sock == '$')
    {
2692
      if (!(var_sock= var_get(con_sock, 0, 0, 0)))
unknown's avatar
unknown committed
2693
        die("Unknown variable '%s'", con_sock+1);
2694
      if (!(con_sock= (char*)my_malloc(var_sock->str_val_len+1, MYF(0))))
unknown's avatar
unknown committed
2695
        die("Out of memory");
2696
      free_con_sock= 1;
2697
      memcpy(con_sock, var_sock->str_val, var_sock->str_val_len);
2698
      con_sock[var_sock->str_val_len]= 0;
2699
    }
unknown's avatar
unknown committed
2700
  }
unknown's avatar
unknown committed
2701 2702 2703 2704 2705 2706
  else
  {
    con_sock= (char*) unix_sock;
  }

  /* Options */
unknown's avatar
unknown committed
2707
  p= safe_get_param(p, &con_options, 0);
unknown's avatar
unknown committed
2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721
  while (*con_options)
  {
    char* str= con_options;
    while (*str && !my_isspace(charset_info, *str))
      str++;
    *str++= 0;
    if (!strcmp(con_options, "SSL"))
      con_ssl= 1;
    else if (!strcmp(con_options, "COMPRESS"))
      con_compress= 1;
    else
      die("Illegal option to connect: %s", con_options);
    con_options= str;
  }
2722 2723
  /* Note: 'p' is pointing into the copy 'con_buf' */
  q->last_argument= q->first_argument + (p - con_buf);
unknown's avatar
unknown committed
2724

2725
  if (next_con == cons_end)
unknown's avatar
unknown committed
2726
    die("Connection limit exhausted - increase MAX_CONS in mysqltest.c");
unknown's avatar
unknown committed
2727

unknown's avatar
unknown committed
2728 2729 2730
  if (find_connection_by_name(con_name))
    die("Connection %s already exists", con_name);

2731
  if (!mysql_init(&next_con->mysql))
unknown's avatar
unknown committed
2732
    die("Failed on mysql_init()");
unknown's avatar
unknown committed
2733
  if (opt_compress || con_compress)
2734
    mysql_options(&next_con->mysql, MYSQL_OPT_COMPRESS, NullS);
2735
  mysql_options(&next_con->mysql, MYSQL_OPT_LOCAL_INFILE, 0);
unknown's avatar
unknown committed
2736
  mysql_options(&next_con->mysql, MYSQL_SET_CHARSET_NAME, charset_name);
2737

unknown's avatar
unknown committed
2738
#ifdef HAVE_OPENSSL
unknown's avatar
unknown committed
2739
  if (opt_use_ssl || con_ssl)
unknown's avatar
unknown committed
2740 2741 2742
    mysql_ssl_set(&next_con->mysql, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
		  opt_ssl_capath, opt_ssl_cipher);
#endif
2743
  if (con_sock && !free_con_sock && *con_sock && *con_sock != FN_LIBCHAR)
2744
    con_sock=fn_format(buff, con_sock, TMPDIR, "",0);
unknown's avatar
unknown committed
2745
  if (!con_db[0])
2746
    con_db= db;
2747
  /* Special database to allow one to connect without a database name */
unknown's avatar
unknown committed
2748
  if (con_db && !strcmp(con_db,"*NO-ONE*"))
2749
    con_db= 0;
2750 2751
  if (q->abort_on_error)
  {
2752 2753 2754 2755
    if (safe_connect(&next_con->mysql, con_host, con_user, con_pass,
		     con_db, con_port, con_sock ? con_sock: 0))
      die("Could not open connection '%s': %d %s", con_name,
          mysql_errno(&next_con->mysql), mysql_error(&next_con->mysql));
2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767
  }
  else
    error= connect_n_handle_errors(q, &next_con->mysql, con_host, con_user,
                                   con_pass, con_db, con_port, con_sock,
                                   &create_conn);

  if (create_conn)
  {
    if (!(next_con->name= my_strdup(con_name, MYF(MY_WME))))
      die(NullS);
    cur_con= next_con++;
  }
2768 2769
  if (free_con_sock)
    my_free(con_sock, MYF(MY_WME));
unknown's avatar
unknown committed
2770
  my_free(con_buf, MYF(MY_WME));
2771
  DBUG_RETURN(error);
unknown's avatar
unknown committed
2772 2773
}

2774

2775
int do_done(struct st_query *q)
2776
{
unknown's avatar
unknown committed
2777
  /* Check if empty block stack */
2778
  if (cur_block == block_stack)
2779 2780 2781
  {
    if (*q->query != '}')
      die("Stray 'end' command - end of block before beginning");
2782
    die("Stray '}' - end of block before beginning");
2783
  }
unknown's avatar
unknown committed
2784 2785 2786

  /* Test if inner block has been executed */
  if (cur_block->ok && cur_block->cmd == cmd_while)
2787
  {
unknown's avatar
unknown committed
2788 2789 2790
    /* Pop block from stack, re-execute outer block */
    cur_block--;
    parser.current_line = cur_block->line;
2791
  }
2792
  else
unknown's avatar
unknown committed
2793
  {
unknown's avatar
unknown committed
2794 2795 2796
    /* Pop block from stack, goto next line */
    cur_block--;
    parser.current_line++;
unknown's avatar
unknown committed
2797
  }
2798 2799 2800
  return 0;
}

unknown's avatar
unknown committed
2801

2802 2803 2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827
/*
  Process start of a "if" or "while" statement

  SYNOPSIS
   do_block()
    cmd        Type of block
    q	       called command

  DESCRIPTION
    if ([!]<expr>)
    {
      <block statements>
    }

    while ([!]<expr>)
    {
      <block statements>
    }

    Evaluates the <expr> and if it evaluates to
    greater than zero executes the following code block.
    A '!' can be used before the <expr> to indicate it should
    be executed if it evaluates to zero.

 */

2828
void do_block(enum block_cmd cmd, struct st_query* q)
2829
{
2830 2831
  char *p= q->first_argument;
  const char *expr_start, *expr_end;
2832
  VAR v;
2833
  const char *cmd_name= (cmd == cmd_while ? "while" : "if");
2834 2835 2836
  my_bool not_expr= FALSE;
  DBUG_ENTER("do_block");
  DBUG_PRINT("enter", ("%s", cmd_name));
unknown's avatar
unknown committed
2837 2838

  /* Check stack overflow */
2839
  if (cur_block == block_stack_end)
2840
    die("Nesting too deeply");
unknown's avatar
unknown committed
2841 2842 2843 2844 2845 2846

  /* Set way to find outer block again, increase line counter */
  cur_block->line= parser.current_line++;

  /* If this block is ignored */
  if (!cur_block->ok)
2847
  {
unknown's avatar
unknown committed
2848 2849 2850 2851
    /* Inner block should be ignored too */
    cur_block++;
    cur_block->cmd= cmd;
    cur_block->ok= FALSE;
2852
    DBUG_VOID_RETURN;
2853
  }
unknown's avatar
unknown committed
2854

unknown's avatar
unknown committed
2855
  /* Parse and evaluate test expression */
2856
  expr_start= strchr(p, '(');
2857
  if (!expr_start++)
2858
    die("missing '(' in %s", cmd_name);
2859 2860 2861 2862 2863 2864 2865 2866

  /* Check for !<expr> */
  if (*expr_start == '!')
  {
    not_expr= TRUE;
    expr_start++; /* Step past the '!' */
  }
  /* Find ending ')' */
2867
  expr_end= strrchr(expr_start, ')');
2868
  if (!expr_end)
2869
    die("missing ')' in %s", cmd_name);
2870 2871 2872 2873
  p= (char*)expr_end+1;

  while (*p && my_isspace(charset_info, *p))
    p++;
2874 2875
  if (*p == '{')
    die("Missing newline between %s and '{'", cmd_name);
2876
  if (*p)
2877
    die("Missing '{' after %s. Found \"%s\"", cmd_name, p);
2878

unknown's avatar
unknown committed
2879
  var_init(&v,0,0,0,0);
2880
  eval_expr(&v, expr_start, &expr_end);
unknown's avatar
unknown committed
2881 2882 2883 2884 2885 2886

  /* Define inner block */
  cur_block++;
  cur_block->cmd= cmd;
  cur_block->ok= (v.int_val ? TRUE : FALSE);

2887 2888 2889 2890 2891
  if (not_expr)
    cur_block->ok = !cur_block->ok;

  DBUG_PRINT("info", ("OK: %d", cur_block->ok));

unknown's avatar
unknown committed
2892
  var_free(&v);
2893
  DBUG_VOID_RETURN;
2894 2895
}

unknown's avatar
unknown committed
2896

2897 2898 2899 2900 2901 2902 2903 2904 2905 2906
/*
  Read characters from line buffer or file. This is needed to allow
  my_ungetc() to buffer MAX_DELIMITER characters for a file

  NOTE:
    This works as long as one doesn't change files (with 'source file_name')
    when there is things pushed into the buffer.  This should however not
    happen for any tests in the test suite.
*/

2907
int my_getc(FILE *file)
2908 2909 2910
{
  if (line_buffer_pos == line_buffer)
    return fgetc(file);
2911
  return *--line_buffer_pos;
2912 2913 2914 2915
}

void my_ungetc(int c)
{
2916
  *line_buffer_pos++= (char) c;
2917 2918 2919 2920
}


my_bool end_of_query(int c)
2921
{
2922
  uint i;
2923
  char tmp[MAX_DELIMITER];
2924

2925 2926 2927 2928
  if (c != *delimiter)
    return 0;

  for (i= 1; i < delimiter_length &&
2929
	 (c= my_getc(cur_file->file)) == *(delimiter + i);
2930
       i++)
2931 2932 2933
    tmp[i]= c;

  if (i == delimiter_length)
2934 2935 2936
    return 1;					/* Found delimiter */

  /* didn't find delimiter, push back things that we read */
2937 2938 2939
  my_ungetc(c);
  while (i > 1)
    my_ungetc(tmp[--i]);
2940 2941 2942 2943
  return 0;
}


2944 2945 2946 2947 2948 2949 2950 2951 2952
/*
  Read one "line" from the file

  SYNOPSIS
    read_line
    buf     buffer for the read line
    size    size of the buffer i.e max size to read

  DESCRIPTION
unknown's avatar
unknown committed
2953 2954
    This function actually reads several lines and adds them to the
    buffer buf. It continues to read until it finds what it believes
2955 2956 2957 2958 2959
    is a complete query.

    Normally that means it will read lines until it reaches the
    "delimiter" that marks end of query. Default delimiter is ';'
    The function should be smart enough not to detect delimiter's
unknown's avatar
unknown committed
2960
    found inside strings surrounded with '"' and '\'' escaped strings.
2961 2962 2963 2964 2965 2966 2967

    If the first line in a query starts with '#' or '-' this line is treated
    as a comment. A comment is always terminated when end of line '\n' is
    reached.

*/

2968
int read_line(char *buf, int size)
unknown's avatar
unknown committed
2969 2970
{
  int c;
2971
  char quote;
2972
  char *p= buf, *buf_end= buf + size - 1;
2973
  int no_save= 0;
2974 2975
  enum {R_NORMAL, R_Q, R_Q_IN_Q, R_SLASH_IN_Q,
	R_COMMENT, R_LINE_START} state= R_LINE_START;
unknown's avatar
unknown committed
2976
  DBUG_ENTER("read_line");
2977
  LINT_INIT(quote);
2978

2979
  start_lineno= cur_file->lineno;
2980 2981
  for (; p < buf_end ;)
  {
2982
    no_save= 0;
2983 2984
    c= my_getc(cur_file->file);
    if (feof(cur_file->file))
unknown's avatar
unknown committed
2985
    {
unknown's avatar
unknown committed
2986
  found_eof:
unknown's avatar
unknown committed
2987 2988
      if (cur_file->file != stdin)
      {
2989
	my_fclose(cur_file->file, MYF(0));
2990 2991
        cur_file->file= 0;
      }
2992 2993
      my_free((gptr)cur_file->file_name, MYF(MY_ALLOW_ZERO_PTR));
      cur_file->file_name= 0;
2994
      if (cur_file == file_stack)
2995
      {
2996 2997 2998 2999 3000
        /* We're back at the first file, check if
           all { have matching }
         */
        if (cur_block != block_stack)
          die("Missing end of block");
3001

3002
        DBUG_PRINT("info", ("end of file"));
unknown's avatar
unknown committed
3003
	DBUG_RETURN(1);
3004
      }
3005
      cur_file--;
3006
      start_lineno= cur_file->lineno;
unknown's avatar
unknown committed
3007
      continue;
3008
    }
3009

3010
    if (c == '\n')
3011 3012
    {
      /* Line counting is independent of state */
3013
      cur_file->lineno++;
3014

3015 3016 3017 3018 3019
      /* Convert cr/lf to lf */
      if (p != buf && *(p-1) == '\r')
        *(p-1)= 0;
    }

3020 3021
    switch(state) {
    case R_NORMAL:
3022
      /*  Only accept '{' in the beginning of a line */
3023 3024 3025
      if (end_of_query(c))
      {
	*p= 0;
unknown's avatar
unknown committed
3026
	DBUG_RETURN(0);
3027
      }
3028 3029 3030 3031 3032
      else if (c == '\'' || c == '"' || c == '`')
      {
        quote= c;
	state= R_Q;
      }
3033
      else if (c == '\n')
3034
      {
3035
	state = R_LINE_START;
3036
      }
3037 3038 3039 3040
      break;
    case R_COMMENT:
      if (c == '\n')
      {
3041
	*p= 0;
unknown's avatar
unknown committed
3042
	DBUG_RETURN(0);
3043 3044 3045
      }
      break;
    case R_LINE_START:
3046
      /* Only accept start of comment if this is the first line in query */
3047 3048
      if ((cur_file->lineno == start_lineno) &&
	  (c == '#' || c == '-' || parsing_disabled))
3049 3050 3051
      {
	state = R_COMMENT;
      }
3052
      else if (my_isspace(charset_info, c))
3053 3054
      {
	if (c == '\n')
3055
	  start_lineno= cur_file->lineno; /* Query hasn't started yet */
3056
	no_save= 1;
3057
      }
3058 3059
      else if (c == '}')
      {
3060 3061
	*buf++= '}';
	*buf= 0;
unknown's avatar
unknown committed
3062
	DBUG_RETURN(0);
3063
      }
3064
      else if (end_of_query(c) || c == '{')
3065
      {
3066
	*p= 0;
unknown's avatar
unknown committed
3067
	DBUG_RETURN(0);
3068
      }
3069
      else if (c == '\'' || c == '"' || c == '`')
3070
      {
3071 3072
        quote= c;
	state= R_Q;
3073
      }
3074
      else
3075
	state= R_NORMAL;
3076
      break;
unknown's avatar
unknown committed
3077

3078 3079 3080
    case R_Q:
      if (c == quote)
	state= R_Q_IN_Q;
3081
      else if (c == '\\')
3082
	state= R_SLASH_IN_Q;
3083
      break;
3084
    case R_Q_IN_Q:
3085 3086 3087
      if (end_of_query(c))
      {
	*p= 0;
unknown's avatar
unknown committed
3088
	DBUG_RETURN(0);
3089
      }
3090
      if (c != quote)
3091
	state= R_NORMAL;
3092
      else
3093
	state= R_Q;
3094
      break;
3095 3096
    case R_SLASH_IN_Q:
      state= R_Q;
3097
      break;
3098

unknown's avatar
unknown committed
3099
    }
3100 3101

    if (!no_save)
unknown's avatar
unknown committed
3102 3103 3104 3105 3106
    {
      /* Could be a multibyte character */
      /* This code is based on the code in "sql_load.cc" */
#ifdef USE_MB
      int charlen = my_mbcharlen(charset_info, c);
unknown's avatar
Merge  
unknown committed
3107 3108
      /* We give up if multibyte character is started but not */
      /* completed before we pass buf_end */
unknown's avatar
unknown committed
3109 3110 3111 3112 3113 3114 3115 3116 3117
      if ((charlen > 1) && (p + charlen) <= buf_end)
      {
	int i;
	char* mb_start = p;

	*p++ = c;

	for (i= 1; i < charlen; i++)
	{
3118
	  if (feof(cur_file->file))
unknown's avatar
unknown committed
3119
	    goto found_eof;	/* FIXME: could we just break here?! */
3120
	  c= my_getc(cur_file->file);
unknown's avatar
unknown committed
3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134
	  *p++ = c;
	}
	if (! my_ismbchar(charset_info, mb_start, p))
	{
	  /* It was not a multiline char, push back the characters */
	  /* We leave first 'c', i.e. pretend it was a normal char */
	  while (p > mb_start)
	    my_ungetc(*--p);
	}
      }
      else
#endif
	*p++= c;
    }
3135
  }
3136
  *p= 0;					/* Always end with \0 */
3137
  DBUG_RETURN(feof(cur_file->file));
unknown's avatar
unknown committed
3138 3139
}

3140 3141 3142 3143
/*
  Create a query from a set of lines

  SYNOPSIS
3144
    read_query()
3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157
    q_ptr pointer where to return the new query

  DESCRIPTION
    Converts lines returned by read_line into a query, this involves
    parsing the first word in the read line to find the query type.


    A -- comment may contain a valid query as the first word after the
    comment start. Thus it's always checked to see if that is the case.
    The advantage with this approach is to be able to execute commands
    terminated by new line '\n' regardless how many "delimiter" it contain.

*/
unknown's avatar
unknown committed
3158

3159 3160
static char read_query_buf[MAX_QUERY];

3161
int read_query(struct st_query** q_ptr)
unknown's avatar
unknown committed
3162
{
3163
  char *p= read_query_buf;
3164
  struct st_query* q;
unknown's avatar
unknown committed
3165
  DBUG_ENTER("read_query");
3166

3167 3168
  if (parser.current_line < parser.read_lines)
  {
3169
    get_dynamic(&q_lines, (gptr) q_ptr, parser.current_line) ;
unknown's avatar
unknown committed
3170
    DBUG_RETURN(0);
3171
  }
3172
  if (!(*q_ptr= q= (struct st_query*) my_malloc(sizeof(*q), MYF(MY_WME))) ||
unknown's avatar
unknown committed
3173 3174
      insert_dynamic(&q_lines, (gptr) &q))
    die(NullS);
3175

3176 3177 3178
  q->record_file[0]= 0;
  q->require_file= 0;
  q->first_word_len= 0;
unknown's avatar
unknown committed
3179

3180
  q->type= Q_UNKNOWN;
3181
  q->query_buf= q->query= 0;
3182
  read_query_buf[0]= 0;
3183
  if (read_line(read_query_buf, sizeof(read_query_buf)))
unknown's avatar
unknown committed
3184
  {
3185
    check_eol_junk(read_query_buf);
unknown's avatar
unknown committed
3186
    DBUG_RETURN(1);
unknown's avatar
unknown committed
3187
  }
3188
  
3189
  DBUG_PRINT("info", ("query: %s", read_query_buf));
3190 3191
  if (*p == '#')
  {
3192
    q->type= Q_COMMENT;
unknown's avatar
unknown committed
3193 3194
    /* This goto is to avoid losing the "expected error" info. */
    goto end;
3195
  }
3196 3197 3198 3199 3200 3201 3202 3203
  if (!parsing_disabled)
  {
    memcpy((gptr) q->expected_errno, (gptr) global_expected_errno,
           sizeof(global_expected_errno));
    q->expected_errors= global_expected_errors;
    q->abort_on_error= (global_expected_errors == 0 && abort_on_error);
  }

unknown's avatar
unknown committed
3204
  if (p[0] == '-' && p[1] == '-')
3205
  {
3206 3207
    q->type= Q_COMMENT_WITH_COMMAND;
    p+= 2;					/* To calculate first word */
3208
  }
3209
  else if (!parsing_disabled)
3210
  {
3211
    while (*p && my_isspace(charset_info, *p))
unknown's avatar
unknown committed
3212
      p++ ;
3213
  }
unknown's avatar
unknown committed
3214 3215

end:
3216
  while (*p && my_isspace(charset_info, *p))
unknown's avatar
unknown committed
3217
    p++;
unknown's avatar
unknown committed
3218

3219
  if (!(q->query_buf= q->query= my_strdup(p, MYF(MY_WME))))
3220 3221 3222
    die(NullS);

  /* Calculate first word and first argument */
3223 3224 3225
  for (p= q->query; *p && !my_isspace(charset_info, *p) ; p++) ;
  q->first_word_len= (uint) (p - q->query);
  while (*p && my_isspace(charset_info, *p))
unknown's avatar
unknown committed
3226
    p++;
3227 3228
  q->first_argument= p;
  q->end= strend(q->query);
3229
  parser.read_lines++;
unknown's avatar
unknown committed
3230
  DBUG_RETURN(0);
unknown's avatar
unknown committed
3231 3232
}

3233 3234 3235

static struct my_option my_long_options[] =
{
unknown's avatar
unknown committed
3236 3237
  {"help", '?', "Display this help and exit.", 0, 0, 0, GET_NO_ARG, NO_ARG,
   0, 0, 0, 0, 0, 0},
3238
  {"basedir", 'b', "Basedir for tests.", (gptr*) &opt_basedir,
3239
   (gptr*) &opt_basedir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
3240
  {"big-test", 'B', "Define BIG_TEST to 1.", (gptr*) &opt_big_test,
3241
   (gptr*) &opt_big_test, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
3242
  {"compress", 'C', "Use the compressed server/client protocol.",
3243 3244
   (gptr*) &opt_compress, (gptr*) &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0,
   0, 0, 0},
unknown's avatar
unknown committed
3245
  {"cursor-protocol", OPT_CURSOR_PROTOCOL, "Use cursors for prepared statements.",
unknown's avatar
unknown committed
3246 3247
   (gptr*) &cursor_protocol, (gptr*) &cursor_protocol, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3248 3249 3250 3251 3252 3253 3254 3255 3256
  {"database", 'D', "Database to use.", (gptr*) &db, (gptr*) &db, 0,
   GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
#ifdef DBUG_OFF
  {"debug", '#', "This is a non-debug version. Catch this and exit",
   0,0, 0, GET_DISABLED, OPT_ARG, 0, 0, 0, 0, 0, 0},
#else
  {"debug", '#', "Output debug log. Often this is 'd:t:o,filename'.",
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
#endif
3257 3258
  {"host", 'h', "Connect to host.", (gptr*) &host, (gptr*) &host, 0,
   GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3259 3260
  {"include", 'i', "Include SQL before each test case.", (gptr*) &opt_include,
   (gptr*) &opt_include, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
3261
  {"manager-host", OPT_MANAGER_HOST, "Undocumented: Used for debugging.",
3262 3263
   (gptr*) &manager_host, (gptr*) &manager_host, 0, GET_STR, REQUIRED_ARG,
   0, 0, 0, 0, 0, 0},
3264
  {"manager-password", OPT_MANAGER_PASSWD, "Undocumented: Used for debugging.",
3265
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
3266
  {"manager-port", OPT_MANAGER_PORT, "Undocumented: Used for debugging.",
3267 3268
   (gptr*) &manager_port, (gptr*) &manager_port, 0, GET_INT, REQUIRED_ARG,
   MYSQL_MANAGER_PORT, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3269 3270 3271
  {"manager-user", OPT_MANAGER_USER, "Undocumented: Used for debugging.",
   (gptr*) &manager_user, (gptr*) &manager_user, 0, GET_STR, REQUIRED_ARG, 0,
   0, 0, 0, 0, 0},
3272
  {"manager-wait-timeout", OPT_MANAGER_WAIT_TIMEOUT,
3273
   "Undocumented: Used for debugging.", (gptr*) &manager_wait_timeout,
3274 3275 3276 3277
   (gptr*) &manager_wait_timeout, 0, GET_INT, REQUIRED_ARG, 3, 0, 0, 0, 0, 0},
  {"password", 'p', "Password to use when connecting to server.",
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
  {"port", 'P', "Port number to use for connection.", (gptr*) &port,
3278
   (gptr*) &port, 0, GET_INT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
3279 3280 3281
  {"ps-protocol", OPT_PS_PROTOCOL, "Use prepared statements protocol for communication",
   (gptr*) &ps_protocol, (gptr*) &ps_protocol, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
3282 3283 3284 3285 3286 3287 3288
  {"quiet", 's', "Suppress all normal output.", (gptr*) &silent,
   (gptr*) &silent, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"record", 'r', "Record output of test_file into result file.",
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"result-file", 'R', "Read/Store result from/in this file.",
   (gptr*) &result_file, (gptr*) &result_file, 0, GET_STR, REQUIRED_ARG,
   0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3289
  {"server-arg", 'A', "Send option value to embedded server as a parameter.",
3290
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
3291
  {"server-file", 'F', "Read embedded server arguments from file.",
3292 3293 3294
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"silent", 's', "Suppress all normal output. Synonym for --quiet.",
   (gptr*) &silent, (gptr*) &silent, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
3295
  {"skip-safemalloc", OPT_SKIP_SAFEMALLOC,
3296
   "Don't use the memory allocation checking.", 0, 0, 0, GET_NO_ARG, NO_ARG,
3297
   0, 0, 0, 0, 0, 0},
3298
  {"sleep", 'T', "Sleep always this many seconds on sleep commands.",
3299 3300 3301 3302 3303
   (gptr*) &opt_sleep, (gptr*) &opt_sleep, 0, GET_INT, REQUIRED_ARG, 0, 0, 0,
   0, 0, 0},
  {"socket", 'S', "Socket file to use for connection.",
   (gptr*) &unix_sock, (gptr*) &unix_sock, 0, GET_STR, REQUIRED_ARG, 0, 0, 0,
   0, 0, 0},
unknown's avatar
unknown committed
3304 3305 3306
  {"sp-protocol", OPT_SP_PROTOCOL, "Use stored procedures for select",
   (gptr*) &sp_protocol, (gptr*) &sp_protocol, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3307
#include "sslopt-longopts.h"
3308 3309
  {"test-file", 'x', "Read test from/in this file (default stdin).",
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3310 3311
  {"timer-file", 'm', "File where the timing in micro seconds is stored.",
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
3312
  {"tmpdir", 't', "Temporary directory where sockets are put.",
3313 3314 3315 3316 3317 3318 3319
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"user", 'u', "User for login.", (gptr*) &user, (gptr*) &user, 0, GET_STR,
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"verbose", 'v', "Write more.", (gptr*) &verbose, (gptr*) &verbose, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"version", 'V', "Output version information and exit.",
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
3320 3321 3322
  {"view-protocol", OPT_VIEW_PROTOCOL, "Use views for select",
   (gptr*) &view_protocol, (gptr*) &view_protocol, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
3323
  { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
unknown's avatar
unknown committed
3324 3325
};

unknown's avatar
unknown committed
3326 3327 3328

#include <help_start.h>

unknown's avatar
unknown committed
3329 3330 3331 3332 3333 3334 3335 3336 3337
static void print_version(void)
{
  printf("%s  Ver %s Distrib %s, for %s (%s)\n",my_progname,MTEST_VERSION,
	 MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);
}

void usage()
{
  print_version();
3338
  printf("MySQL AB, by Sasha, Matt, Monty & Jani\n");
unknown's avatar
unknown committed
3339 3340 3341
  printf("This software comes with ABSOLUTELY NO WARRANTY\n\n");
  printf("Runs a test against the mysql server and compares output with a results file.\n\n");
  printf("Usage: %s [OPTIONS] [database] < test_file\n", my_progname);
3342
  my_print_help(my_long_options);
unknown's avatar
unknown committed
3343
  printf("  --no-defaults       Don't read default options from any options file.\n");
3344
  my_print_variables(my_long_options);
unknown's avatar
unknown committed
3345 3346
}

unknown's avatar
unknown committed
3347 3348
#include <help_end.h>

3349 3350 3351 3352 3353

static my_bool
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
	       char *argument)
{
3354
  switch(optid) {
3355
  case '#':
unknown's avatar
unknown committed
3356
#ifndef DBUG_OFF
3357
    DBUG_PUSH(argument ? argument : "d:t:S:i:O,/tmp/mysqltest.trace");
unknown's avatar
unknown committed
3358
#endif
3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376
    break;
  case 'r':
    record = 1;
    break;
  case (int)OPT_MANAGER_PASSWD:
    my_free(manager_pass,MYF(MY_ALLOW_ZERO_PTR));
    manager_pass=my_strdup(argument, MYF(MY_FAE));
    while (*argument) *argument++= 'x';		/* Destroy argument */
    break;
  case 'x':
    {
      char buff[FN_REFLEN];
      if (!test_if_hard_path(argument))
      {
	strxmov(buff, opt_basedir, argument, NullS);
	argument= buff;
      }
      fn_format(buff, argument, "", "", 4);
3377
      DBUG_ASSERT(cur_file == file_stack && cur_file->file == 0);
3378
      if (!(cur_file->file=
3379
            my_fopen(buff, O_RDONLY | FILE_BINARY, MYF(0))))
3380 3381
	die("Could not open %s: errno = %d", buff, errno);
      cur_file->file_name= my_strdup(buff, MYF(MY_FAE));
3382
      cur_file->lineno= 1;
3383 3384
      break;
    }
unknown's avatar
unknown committed
3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397
  case 'm':
    {
      static char buff[FN_REFLEN];
      if (!test_if_hard_path(argument))
      {
	strxmov(buff, opt_basedir, argument, NullS);
	argument= buff;
      }
      fn_format(buff, argument, "", "", 4);
      timer_file= buff;
      unlink(timer_file);	     /* Ignore error, may not exist */
      break;
    }
3398 3399 3400 3401 3402 3403
  case 'p':
    if (argument)
    {
      my_free(pass, MYF(MY_ALLOW_ZERO_PTR));
      pass= my_strdup(argument, MYF(MY_FAE));
      while (*argument) *argument++= 'x';		/* Destroy argument */
3404
      tty_password= 0;
3405 3406 3407 3408
    }
    else
      tty_password= 1;
    break;
unknown's avatar
unknown committed
3409
#include <sslopt-case.h>
3410 3411 3412 3413 3414 3415 3416 3417 3418
  case 't':
    strnmov(TMPDIR, argument, sizeof(TMPDIR));
    break;
  case 'A':
    if (!embedded_server_arg_count)
    {
      embedded_server_arg_count=1;
      embedded_server_args[0]= (char*) "";
    }
3419 3420 3421
    if (embedded_server_arg_count == MAX_SERVER_ARGS-1 ||
        !(embedded_server_args[embedded_server_arg_count++]=
          my_strdup(argument, MYF(MY_FAE))))
3422 3423 3424 3425 3426 3427 3428 3429
    {
      die("Can't use server argument");
    }
    break;
  case 'F':
    if (read_server_arguments(argument))
      die(NullS);
    break;
3430 3431 3432 3433 3434
  case OPT_SKIP_SAFEMALLOC:
#ifdef SAFEMALLOC
    sf_malloc_quick=1;
#endif
    break;
3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445
  case 'V':
    print_version();
    exit(0);
  case '?':
    usage();
    exit(1);
  }
  return 0;
}


unknown's avatar
unknown committed
3446 3447 3448
int parse_args(int argc, char **argv)
{
  load_defaults("my",load_default_groups,&argc,&argv);
3449
  default_argv= argv;
3450

3451
  if ((handle_options(&argc, &argv, my_long_options, get_one_option)))
unknown's avatar
unknown committed
3452
    exit(1);
unknown's avatar
unknown committed
3453 3454 3455 3456 3457 3458 3459

  if (argc > 1)
  {
    usage();
    exit(1);
  }
  if (argc == 1)
3460
    db= *argv;
unknown's avatar
unknown committed
3461 3462 3463 3464 3465 3466 3467
  if (tty_password)
    pass=get_tty_password(NullS);

  return 0;
}


unknown's avatar
unknown committed
3468
/*
3469
   Write the content of str into file
unknown's avatar
unknown committed
3470

3471 3472 3473 3474 3475 3476
   SYNOPSIS
   str_to_file
   fname - name of file to truncate/create and write to
   str - content to write to file
   size - size of content witten to file
*/
unknown's avatar
unknown committed
3477

3478
static void str_to_file(const char *fname, char *str, int size)
unknown's avatar
unknown committed
3479 3480
{
  int fd;
3481 3482 3483 3484
  char buff[FN_REFLEN];
  if (!test_if_hard_path(fname))
  {
    strxmov(buff, opt_basedir, fname, NullS);
unknown's avatar
unknown committed
3485
    fname= buff;
3486 3487
  }
  fn_format(buff,fname,"","",4);
unknown's avatar
unknown committed
3488

unknown's avatar
unknown committed
3489
  if ((fd= my_open(buff, O_WRONLY | O_CREAT | O_TRUNC,
3490
		    MYF(MY_WME | MY_FFNF))) < 0)
3491
    die("Could not open %s: errno = %d", buff, errno);
3492
  if (my_write(fd, (byte*)str, size, MYF(MY_WME|MY_FNABP)))
unknown's avatar
unknown committed
3493 3494 3495 3496
    die("write failed");
  my_close(fd, MYF(0));
}

unknown's avatar
unknown committed
3497

3498
void dump_result_to_reject_file(const char *record_file, char *buf, int size)
unknown's avatar
unknown committed
3499
{
unknown's avatar
unknown committed
3500
  char reject_file[FN_REFLEN];
unknown's avatar
unknown committed
3501
  str_to_file(fn_format(reject_file, record_file,"",".reject",2), buf, size);
unknown's avatar
unknown committed
3502 3503
}

3504 3505 3506 3507 3508 3509
void dump_result_to_log_file(const char *record_file, char *buf, int size)
{
  char log_file[FN_REFLEN];
  str_to_file(fn_format(log_file, record_file,"",".log",2), buf, size);
}

3510 3511 3512 3513 3514 3515 3516
static void check_regerr(my_regex_t* r, int err)
{
  char err_buf[1024];

  if (err)
  {
    my_regerror(err,r,err_buf,sizeof(err_buf));
3517
    die("Regex error: %s\n", err_buf);
3518 3519 3520
  }  
}

3521 3522 3523 3524
/* 
  auxiluary macro used by reg_replace
  makes sure the result buffer has sufficient length
*/  
3525 3526 3527 3528 3529 3530 3531 3532
#define SECURE_REG_BUF   if (buf_len < need_buf_len)\
  {\
    int off= res_p - buf;\
    buf= (char*)my_realloc(buf,need_buf_len,MYF(MY_WME+MY_FAE));\
    res_p= buf + off;\
    buf_len= need_buf_len;\
  }\

3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546
/*
  Performs a regex substitution
  
  IN:
  
    buf_p - result buffer pointer. Will change if reallocated
    buf_len_p - result buffer length. Will change if the buffer is reallocated
    pattern - regexp pattern to match
    replace - replacement expression
    string - the string to perform substituions in
    icase - flag, if set to 1 the match is case insensitive
 */  
static int reg_replace(char** buf_p, int* buf_len_p, char *pattern, 
  char *replace, char *string, int icase)
3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560
{
  my_regex_t r;
  my_regmatch_t* subs;
  char* buf_end, *replace_end;
  char* buf= *buf_p;
  int len;
  int buf_len,need_buf_len;
  int cflags= REG_EXTENDED;
  int err_code;
  char* res_p,*str_p,*str_end;
  
  buf_len= *buf_len_p;  
  len= strlen(string);
  str_end= string + len;
3561 3562 3563 3564
  
  /* start with a buffer of a reasonable size that hopefully will not 
     need to be reallocated
   */
3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587
  need_buf_len= len * 2 + 1;
  res_p= buf;

  SECURE_REG_BUF    
  
  buf_end = buf + buf_len;
  
  if (icase)
    cflags |= REG_ICASE;
    
  if ((err_code=my_regcomp(&r,pattern,cflags,&my_charset_latin1)))
  {
    check_regerr(&r,err_code);
    return 1;
  }  
  
  subs= (my_regmatch_t*)my_malloc(sizeof(my_regmatch_t) * (r.re_nsub+1),
     MYF(MY_WME+MY_FAE));
  
  *res_p= 0;
  str_p= string;
  replace_end= replace + strlen(replace);
  
3588
  /* for each pattern match instance perform a replacement */
3589 3590
  while (!err_code)
  {
3591 3592
    /* find the match */
    err_code= my_regexec(&r,str_p, r.re_nsub+1, subs, 
3593 3594
      (str_p == string) ? REG_NOTBOL : 0);
    
3595
    /* if regular expression error (eg. bad syntax, or out of memory) */  
3596 3597 3598 3599 3600 3601 3602
    if (err_code && err_code != REG_NOMATCH)
    {
      check_regerr(&r,err_code);
      my_regfree(&r);
      return 1;
    }
    
3603
    /* if match found */
3604 3605 3606 3607 3608
    if (!err_code)
    {
      char* expr_p= replace;
      int c;
      
3609 3610 3611 3612
      /* 
        we need at least what we have so far in the buffer + the part
        before this match
      */
3613 3614
      need_buf_len= (res_p - buf) + subs[0].rm_so;
      
3615
      /* on this pass, calculate the memory for the result buffer */
3616 3617 3618 3619 3620 3621 3622 3623 3624 3625
      while (expr_p < replace_end)
      {
        int back_ref_num= -1;
        c= *expr_p;
               
        if (c == '\\' && expr_p + 1 < replace_end)
        {
          back_ref_num= expr_p[1] - '0';
        }
        
3626
        /* found a valid back_ref (eg. \1)*/
3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643
        if (back_ref_num >= 0 && back_ref_num <= (int)r.re_nsub)
        {
          int start_off,end_off;
          if ((start_off=subs[back_ref_num].rm_so) > -1 && 
                   (end_off=subs[back_ref_num].rm_eo) > -1)
          {
             need_buf_len += (end_off - start_off);    
          }  
          expr_p += 2;
        }
        else
        {
          expr_p++;
          need_buf_len++;
        }
      }
      need_buf_len++;
3644 3645 3646 3647
      /* 
        now that we know the size of the buffer, 
        make sure it is big enough
      */  
3648 3649
      SECURE_REG_BUF
      
3650
      /* copy the pre-match part */
3651 3652 3653 3654 3655 3656 3657 3658
      if (subs[0].rm_so)
      {
        memcpy(res_p,str_p,subs[0].rm_so);
        res_p += subs[0].rm_so;
      }
        
      expr_p= replace;
      
3659
      /* copy the match and expand back_refs */
3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687
      while (expr_p < replace_end)
      {
        int back_ref_num= -1;
        c= *expr_p;
        
        if (c == '\\' && expr_p + 1 < replace_end)
        {
          back_ref_num= expr_p[1] - '0';
        }
        
        if (back_ref_num >= 0 && back_ref_num <= (int)r.re_nsub)
        {
          int start_off,end_off;
          if ((start_off=subs[back_ref_num].rm_so) > -1 && 
                   (end_off=subs[back_ref_num].rm_eo) > -1)
          {
             int block_len= end_off - start_off;
             memcpy(res_p,str_p + start_off, block_len);
             res_p += block_len; 
          }  
          expr_p += 2;
        }
        else
        {
          *res_p++ = *expr_p++;
        }
      } 
      
3688
      /* handle the post-match part */
3689 3690 3691 3692 3693 3694 3695 3696 3697 3698 3699 3700
      if (subs[0].rm_so == subs[0].rm_eo)
      {
        if (str_p + subs[0].rm_so >= str_end)
          break;
        str_p += subs[0].rm_eo ;
        *res_p++ = *str_p++; 
      }    
      else
      {
        str_p += subs[0].rm_eo;
      }  
    }
3701
    else /* no match this time, just copy the string as is */
3702 3703 3704 3705 3706 3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717
    {
      int left_in_str= str_end-str_p;
      need_buf_len= (res_p-buf) + left_in_str;
      SECURE_REG_BUF
      memcpy(res_p,str_p,left_in_str);
      res_p += left_in_str;
      str_p= str_end;
    }
  }      
  my_regfree(&r);   
  *res_p= 0;
  *buf_p= buf;
  *buf_len_p= buf_len; 
  return 0;
}

3718

3719
#ifdef __WIN__
3720

3721 3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732
DYNAMIC_ARRAY patterns;

/*
  init_win_path_patterns

  DESCRIPTION
   Setup string patterns that will be used to detect filenames that
   needs to be converted from Win to Unix format

*/

static void init_win_path_patterns()
3733
{
3734
  /* List of string patterns to match in order to find paths */
3735 3736 3737 3738
  const char* paths[] = { "$MYSQL_TEST_DIR",
                          "$MYSQL_TMP_DIR",
                          "./test/", 0 };
  int num_paths= 3;
3739 3740 3741 3742 3743 3744 3745 3746 3747
  int i;
  char* p;

  DBUG_ENTER("init_win_path_patterns");

  my_init_dynamic_array(&patterns, sizeof(const char*), 16, 16);

  /* Loop through all paths in the array */
  for (i= 0; i < num_paths; i++)
3748
  {
3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767
    VAR* v;
    if (*(paths[i]) == '$')
    {
      v= var_get(paths[i], 0, 0, 0);
      p= my_strdup(v->str_val, MYF(MY_FAE));
    }
    else
      p= my_strdup(paths[i], MYF(MY_FAE));

    if (insert_dynamic(&patterns, (gptr) &p))
        die(NullS);

    DBUG_PRINT("info", ("p: %s", p));
    while (*p)
    {
      if (*p == '/')
        *p='\\';
      p++;
    }
3768
  }
3769
  DBUG_VOID_RETURN;
3770 3771
}

3772 3773
static void free_win_path_patterns()
{
unknown's avatar
unknown committed
3774
  uint i= 0;
3775 3776 3777 3778 3779 3780 3781
  for (i=0 ; i < patterns.elements ; i++)
  {
    const char** pattern= dynamic_element(&patterns, i, const char**);
    my_free((gptr) *pattern, MYF(0));
  }
  delete_dynamic(&patterns);
}
unknown's avatar
unknown committed
3782

3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795
/*
  fix_win_paths

  DESCRIPTION
   Search the string 'val' for the patterns that are known to be
   strings that contain filenames. Convert all \ to / in the
   filenames that are found.

   Ex:
   val = 'Error "c:\mysql\mysql-test\var\test\t1.frm" didn't exist'
          => $MYSQL_TEST_DIR is found by strstr
          => all \ from c:\mysql\m... until next space is converted into /
*/
3796

3797 3798 3799 3800 3801 3802 3803 3804 3805 3806
static void fix_win_paths(const char* val, int len)
{
  uint i;
  char *p;

  DBUG_ENTER("fix_win_paths");
  for (i= 0; i < patterns.elements; i++)
  {
    const char** pattern= dynamic_element(&patterns, i, const char**);
    DBUG_PRINT("info", ("pattern: %s", *pattern));
3807
    if (strlen(*pattern) == 0) continue;
3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822 3823 3824 3825 3826 3827 3828 3829 3830 3831 3832 3833 3834
    /* Search for the path in string */
    while ((p= strstr(val, *pattern)))
    {
      DBUG_PRINT("info", ("Found %s in val p: %s", *pattern, p));

      while (*p && !my_isspace(charset_info, *p))
      {
        if (*p == '\\')
          *p= '/';
        p++;
      }
      DBUG_PRINT("info", ("Converted \\ to /, p: %s", p));
    }
  }
  DBUG_PRINT("exit", (" val: %s, len: %d", val, len));
  DBUG_VOID_RETURN;
}
#endif

/* Append the string to ds, with optional replace */
static void replace_dynstr_append_mem(DYNAMIC_STRING *ds,
                                      const char *val,  int len)
{
#ifdef __WIN__
  fix_win_paths(val, len);
#endif

3835 3836
  if (glob_replace_regex)
  {
unknown's avatar
unknown committed
3837
    if (!multi_reg_replace(glob_replace_regex, (char*)val))
3838 3839
    {
      val= glob_replace_regex->buf;
3840
      len= strlen(val);
3841
    }
3842
  }
3843

3844 3845 3846 3847
  if (glob_replace)
    replace_strings_append(glob_replace, ds, val, len);
  else
    dynstr_append_mem(ds, val, len);
3848 3849
}

unknown's avatar
unknown committed
3850

3851 3852 3853 3854 3855 3856
/* Append zero-terminated string to ds, with optional replace */
static void replace_dynstr_append(DYNAMIC_STRING *ds, const char *val)
{
  replace_dynstr_append_mem(ds, val, strlen(val));
}

3857

3858 3859 3860 3861 3862 3863 3864 3865 3866 3867 3868 3869 3870 3871 3872 3873 3874 3875
/*
  Append the result for one field to the dynamic string ds
*/

static void append_field(DYNAMIC_STRING *ds, uint col_idx, MYSQL_FIELD* field,
                         const char* val, ulonglong len, bool is_null)
{
  if (col_idx < max_replace_column && replace_column[col_idx])
  {
    val= replace_column[col_idx];
    len= strlen(val);
  }
  else if (is_null)
  {
    val= "NULL";
    len= 4;
  }
#ifdef __WIN__
3876 3877
  else if ((field->type == MYSQL_TYPE_DOUBLE ||
            field->type == MYSQL_TYPE_FLOAT ) &&
3878 3879 3880 3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908
           field->decimals >= 31)
  {
    /* Convert 1.2e+018 to 1.2e+18 and 1.2e-018 to 1.2e-18 */
    char *start= strchr(val, 'e');
    if (start && strlen(start) >= 5 &&
        (start[1] == '-' || start[1] == '+') && start[2] == '0')
    {
      start+=2; /* Now points at first '0' */
      /* Move all chars after the first '0' one step left */
      memmove(start, start + 1, strlen(start));
      len--;
    }
  }
#endif

  if (!display_result_vertically)
  {
    if (col_idx)
      dynstr_append_mem(ds, "\t", 1);
    replace_dynstr_append_mem(ds, val, (int)len);
  }
  else
  {
    dynstr_append(ds, field->name);
    dynstr_append_mem(ds, "\t", 1);
    replace_dynstr_append_mem(ds, val, (int)len);
    dynstr_append_mem(ds, "\n", 1);
  }
}


3909 3910
/*
  Append all results to the dynamic string separated with '\t'
3911
  Values may be converted with 'replace_column'
3912 3913 3914 3915 3916
*/

static void append_result(DYNAMIC_STRING *ds, MYSQL_RES *res)
{
  MYSQL_ROW row;
3917
  uint num_fields= mysql_num_fields(res);
3918
  MYSQL_FIELD *fields= mysql_fetch_fields(res);
3919
  ulong *lengths;
3920

3921 3922
  while ((row = mysql_fetch_row(res)))
  {
3923
    uint i;
3924 3925
    lengths = mysql_fetch_lengths(res);
    for (i = 0; i < num_fields; i++)
3926 3927
      append_field(ds, i, &fields[i],
                   (const char*)row[i], lengths[i], !row[i]);
3928 3929
    if (!display_result_vertically)
      dynstr_append_mem(ds, "\n", 1);
3930
  }
3931
  free_replace_column();
3932 3933
}

3934

3935
/*
unknown's avatar
unknown committed
3936
  Append all results from ps execution to the dynamic string separated
3937
  with '\t'. Values may be converted with 'replace_column'
3938
*/
3939

unknown's avatar
unknown committed
3940
static void append_stmt_result(DYNAMIC_STRING *ds, MYSQL_STMT *stmt,
3941
                               MYSQL_FIELD *fields, uint num_fields)
3942
{
3943 3944 3945
  MYSQL_BIND *bind;
  my_bool *is_null;
  ulong *length;
3946
  uint i;
3947

3948 3949 3950 3951 3952 3953 3954
  /* Allocate array with bind structs, lengths and NULL flags */
  bind= (MYSQL_BIND*) my_malloc(num_fields * sizeof(MYSQL_BIND),
				MYF(MY_WME | MY_FAE | MY_ZEROFILL));
  length= (ulong*) my_malloc(num_fields * sizeof(ulong),
			     MYF(MY_WME | MY_FAE));
  is_null= (my_bool*) my_malloc(num_fields * sizeof(my_bool),
				MYF(MY_WME | MY_FAE));
3955

3956 3957
  /* Allocate data for the result of each field */
  for (i= 0; i < num_fields; i++)
unknown's avatar
unknown committed
3958
  {
3959 3960 3961 3962 3963 3964
    uint max_length= fields[i].max_length + 1;
    bind[i].buffer_type= MYSQL_TYPE_STRING;
    bind[i].buffer= (char *)my_malloc(max_length, MYF(MY_WME | MY_FAE));
    bind[i].buffer_length= max_length;
    bind[i].is_null= &is_null[i];
    bind[i].length= &length[i];
unknown's avatar
unknown committed
3965

unknown's avatar
unknown committed
3966
    DBUG_PRINT("bind", ("col[%d]: buffer_type: %d, buffer_length: %d",
3967
			i, bind[i].buffer_type, bind[i].buffer_length));
3968
  }
unknown's avatar
unknown committed
3969

3970
  if (mysql_stmt_bind_result(stmt, bind))
unknown's avatar
unknown committed
3971
    die("mysql_stmt_bind_result failed: %d: %s",
3972
	mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
3973

3974
  while (mysql_stmt_fetch(stmt) == 0)
3975
  {
3976 3977 3978
    for (i= 0; i < num_fields; i++)
      append_field(ds, i, &fields[i], (const char *) bind[i].buffer,
                   *bind[i].length, *bind[i].is_null);
3979 3980
    if (!display_result_vertically)
      dynstr_append_mem(ds, "\n", 1);
3981 3982
  }

3983 3984
  if (mysql_stmt_fetch(stmt) != MYSQL_NO_DATA)
    die("fetch didn't end with MYSQL_NO_DATA from statement: %d %s",
3985
	mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
3986

3987
  free_replace_column();
3988

3989
  for (i= 0; i < num_fields; i++)
3990 3991
  {
    /* Free data for output */
3992
    my_free((gptr)bind[i].buffer, MYF(MY_WME | MY_FAE));
3993 3994 3995 3996 3997
  }
  /* Free array with bind structs, lengths and NULL flags */
  my_free((gptr)bind    , MYF(MY_WME | MY_FAE));
  my_free((gptr)length  , MYF(MY_WME | MY_FAE));
  my_free((gptr)is_null , MYF(MY_WME | MY_FAE));
unknown's avatar
unknown committed
3998 3999 4000
}


4001
/*
4002
  Append metadata for fields to output
4003 4004
*/

4005
static void append_metadata(DYNAMIC_STRING *ds,
unknown's avatar
unknown committed
4006
			    MYSQL_FIELD *field,
4007
			    uint num_fields)
4008
{
4009 4010 4011 4012
  MYSQL_FIELD *field_end;
  dynstr_append(ds,"Catalog\tDatabase\tTable\tTable_alias\tColumn\t"
                "Column_alias\tType\tLength\tMax length\tIs_null\t"
                "Flags\tDecimals\tCharsetnr\n");
4013

4014 4015 4016
  for (field_end= field+num_fields ;
       field < field_end ;
       field++)
4017
  {
4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046
    char buff[22];
    dynstr_append_mem(ds, field->catalog,
                          field->catalog_length);
    dynstr_append_mem(ds, "\t", 1);
    dynstr_append_mem(ds, field->db, field->db_length);
    dynstr_append_mem(ds, "\t", 1);
    dynstr_append_mem(ds, field->org_table,
                          field->org_table_length);
    dynstr_append_mem(ds, "\t", 1);
    dynstr_append_mem(ds, field->table,
                          field->table_length);
    dynstr_append_mem(ds, "\t", 1);
    dynstr_append_mem(ds, field->org_name,
                          field->org_name_length);
    dynstr_append_mem(ds, "\t", 1);
    dynstr_append_mem(ds, field->name, field->name_length);
    dynstr_append_mem(ds, "\t", 1);
    int10_to_str((int) field->type, buff, 10);
    dynstr_append(ds, buff);
    dynstr_append_mem(ds, "\t", 1);
    longlong10_to_str((unsigned int) field->length, buff, 10);
    dynstr_append(ds, buff);
    dynstr_append_mem(ds, "\t", 1);
    longlong10_to_str((unsigned int) field->max_length, buff, 10);
    dynstr_append(ds, buff);
    dynstr_append_mem(ds, "\t", 1);
    dynstr_append_mem(ds, (char*) (IS_NOT_NULL(field->flags) ?
                                   "N" : "Y"), 1);
    dynstr_append_mem(ds, "\t", 1);
4047

4048 4049 4050 4051 4052 4053 4054 4055 4056
    int10_to_str((int) field->flags, buff, 10);
    dynstr_append(ds, buff);
    dynstr_append_mem(ds, "\t", 1);
    int10_to_str((int) field->decimals, buff, 10);
    dynstr_append(ds, buff);
    dynstr_append_mem(ds, "\t", 1);
    int10_to_str((int) field->charsetnr, buff, 10);
    dynstr_append(ds, buff);
    dynstr_append_mem(ds, "\n", 1);
unknown's avatar
unknown committed
4057
  }
4058 4059
}

unknown's avatar
unknown committed
4060

4061 4062 4063
/*
  Append affected row count and other info to output
*/
4064

unknown's avatar
unknown committed
4065
static void append_info(DYNAMIC_STRING *ds, ulonglong affected_rows,
unknown's avatar
unknown committed
4066
			const char *info)
unknown's avatar
unknown committed
4067
{
4068
  char buf[40];
unknown's avatar
unknown committed
4069
  sprintf(buf,"affected rows: %llu\n", affected_rows);
4070 4071
  dynstr_append(ds, buf);
  if (info)
unknown's avatar
unknown committed
4072
  {
4073 4074 4075
    dynstr_append(ds, "info: ");
    dynstr_append(ds, info);
    dynstr_append_mem(ds, "\n", 1);
unknown's avatar
unknown committed
4076
  }
4077
}
unknown's avatar
unknown committed
4078 4079


unknown's avatar
unknown committed
4080 4081
/*
   Display the table headings with the names tab separated
4082
*/
unknown's avatar
unknown committed
4083 4084 4085

static void append_table_headings(DYNAMIC_STRING *ds,
				  MYSQL_FIELD *field,
4086 4087 4088 4089
				  uint num_fields)
{
  uint col_idx;
  for (col_idx= 0; col_idx < num_fields; col_idx++)
4090
  {
4091 4092 4093
    if (col_idx)
      dynstr_append_mem(ds, "\t", 1);
    replace_dynstr_append(ds, field[col_idx].name);
4094
  }
4095 4096 4097 4098
  dynstr_append_mem(ds, "\n", 1);
}

/*
unknown's avatar
unknown committed
4099 4100 4101 4102
  Fetch warnings from server and append to ds

  RETURN VALUE
   Number of warnings appended to ds
4103 4104
*/

unknown's avatar
unknown committed
4105
static int append_warnings(DYNAMIC_STRING *ds, MYSQL* mysql)
4106 4107
{
  uint count;
4108
  MYSQL_RES *warn_res;
4109 4110 4111
  DBUG_ENTER("append_warnings");

  if (!(count= mysql_warning_count(mysql)))
unknown's avatar
unknown committed
4112
    DBUG_RETURN(0);
4113 4114 4115 4116 4117 4118 4119

  /*
    If one day we will support execution of multi-statements
    through PS API we should not issue SHOW WARNINGS until
    we have not read all results...
  */
  DBUG_ASSERT(!mysql_more_results(mysql));
unknown's avatar
unknown committed
4120

4121 4122
  if (mysql_real_query(mysql, "SHOW WARNINGS", 13))
    die("Error running query \"SHOW WARNINGS\": %s", mysql_error(mysql));
unknown's avatar
unknown committed
4123

4124
  if (!(warn_res= mysql_store_result(mysql)))
4125 4126
    die("Warning count is %u but didn't get any warnings",
	count);
unknown's avatar
unknown committed
4127

4128 4129 4130
  append_result(ds, warn_res);
  mysql_free_result(warn_res);

4131 4132
  DBUG_PRINT("warnings", ("%s", ds->str));

unknown's avatar
unknown committed
4133
  DBUG_RETURN(count);
4134 4135 4136
}


unknown's avatar
unknown committed
4137

4138 4139
/*
  Run query using MySQL C API
unknown's avatar
unknown committed
4140

4141 4142 4143 4144
  SYNPOSIS
  run_query_normal
  mysql - mysql handle
  command - currrent command pointer
4145
  flags -flags indicating wheter to SEND and/or REAP
4146 4147 4148 4149 4150 4151 4152 4153
  query - query string to execute
  query_len - length query string to execute
  ds - output buffer wherte to store result form query

  RETURN VALUE
  error - function will not return
*/

unknown's avatar
unknown committed
4154 4155
static void run_query_normal(MYSQL *mysql, struct st_query *command,
			     int flags, char *query, int query_len,
4156
			     DYNAMIC_STRING *ds, DYNAMIC_STRING *ds_warnings)
4157 4158 4159 4160 4161 4162
{
  MYSQL_RES *res= 0;
  int err= 0, counter= 0;
  DBUG_ENTER("run_query_normal");
  DBUG_PRINT("enter",("flags: %d", flags));
  DBUG_PRINT("enter", ("query: '%-.60s'", query));
unknown's avatar
unknown committed
4163

4164 4165
  if (flags & QUERY_SEND)
  {
unknown's avatar
unknown committed
4166
    /*
4167
       Send the query
4168 4169 4170 4171 4172 4173 4174
     */
    if (mysql_send_query(mysql, query, query_len))
    {
      handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		   mysql_sqlstate(mysql), ds);
      goto end;
    }
4175
  }
unknown's avatar
unknown committed
4176

4177 4178
  if (!(flags & QUERY_REAP))
    DBUG_VOID_RETURN;
unknown's avatar
unknown committed
4179

4180
  do
4181
  {
4182
    /*
unknown's avatar
unknown committed
4183
      When  on first result set, call mysql_read_query_result to retrieve
4184 4185 4186
      answer to the query sent earlier
     */
    if ((counter==0) && mysql_read_query_result(mysql))
4187
    {
4188 4189 4190 4191
      handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		   mysql_sqlstate(mysql), ds);
      goto end;

4192
    }
4193

unknown's avatar
unknown committed
4194 4195
    /*
       Store the result. If res is NULL, use mysql_field_count to
4196 4197 4198
       determine if that was expected
     */
    if (!(res= mysql_store_result(mysql)) && mysql_field_count(mysql))
unknown's avatar
unknown committed
4199
    {
4200 4201
      handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		   mysql_sqlstate(mysql), ds);
4202 4203
      goto end;
    }
4204

4205
    if (!disable_result_log)
4206
    {
unknown's avatar
unknown committed
4207
      ulonglong affected_rows;    /* Ok to be undef if 'disable_info' is set */
unknown's avatar
unknown committed
4208
      LINT_INIT(affected_rows);
unknown's avatar
unknown committed
4209

4210
      if (res)
4211
      {
4212
	MYSQL_FIELD *fields= mysql_fetch_fields(res);
4213
	uint num_fields= mysql_num_fields(res);
4214

4215
	if (display_metadata)
4216
          append_metadata(ds, fields, num_fields);
4217

4218
	if (!display_result_vertically)
4219 4220
	  append_table_headings(ds, fields, num_fields);

4221
	append_result(ds, res);
4222
      }
unknown's avatar
unknown committed
4223

unknown's avatar
unknown committed
4224
      /*
4225
        Need to call mysql_affected_rows() before the "new"
unknown's avatar
unknown committed
4226 4227 4228
        query to find the warnings
      */
      if (!disable_info)
unknown's avatar
unknown committed
4229
        affected_rows= mysql_affected_rows(mysql);
unknown's avatar
unknown committed
4230

4231 4232 4233 4234 4235
      /*
        Add all warnings to the result. We can't do this if we are in
        the middle of processing results from multi-statement, because
        this will break protocol.
      */
4236
      if (!disable_warnings && !mysql_more_results(mysql))
4237
      {
4238
	if (append_warnings(ds_warnings, mysql) || ds_warnings->length)
4239 4240
	{
	  dynstr_append_mem(ds, "Warnings:\n", 10);
4241
	  dynstr_append_mem(ds, ds_warnings->str, ds_warnings->length);
4242
	}
4243
      }
4244

unknown's avatar
unknown committed
4245
      if (!disable_info)
4246
	append_info(ds, affected_rows, mysql_info(mysql));
4247
    }
4248

4249 4250 4251 4252
    if (res)
      mysql_free_result(res);
    counter++;
  } while (!(err= mysql_next_result(mysql)));
4253 4254
  if (err > 0)
  {
4255 4256 4257
    /* We got an error from mysql_next_result, maybe expected */
    handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		 mysql_sqlstate(mysql), ds);
4258 4259
    goto end;
  }
4260
  DBUG_ASSERT(err == -1); /* Successful and there are no more results */
4261

4262
  /* If we come here the query is both executed and read successfully */
4263
  handle_no_error(command);
4264

4265
end:
4266
  free_replace();
4267
  free_replace_regex();
4268 4269 4270 4271 4272 4273

  /*
    We save the return code (mysql_errno(mysql)) from the last call sent
    to the server into the mysqltest builtin variable $mysql_errno. This
    variable then can be used from the test case itself.
  */
4274
  var_set_errno(mysql_errno(mysql));
4275
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
4276 4277 4278
}


4279
/*
4280
  Handle errors which occurred during execution
4281 4282

  SYNOPSIS
4283
    handle_error()
4284 4285
      query - query string
      q     - query context
4286 4287 4288
      err_errno - error number
      err_error - error message
      err_sqlstate - sql state
4289 4290 4291 4292 4293 4294 4295
      ds    - dynamic string which is used for output buffer

  NOTE
    If there is an unexpected error this function will abort mysqltest
    immediately.

  RETURN VALUE
4296
    error - function will not return
4297 4298
*/

4299 4300 4301
static void handle_error(const char *query, struct st_query *q,
			 unsigned int err_errno, const char *err_error,
			 const char *err_sqlstate, DYNAMIC_STRING *ds)
4302 4303
{
  uint i;
unknown's avatar
unknown committed
4304

4305
  DBUG_ENTER("handle_error");
4306 4307

  if (q->require_file)
unknown's avatar
unknown committed
4308 4309 4310 4311 4312 4313 4314 4315 4316
  {
    /*
      The query after a "--require" failed. This is fine as long the server
      returned a valid reponse. Don't allow 2013 or 2006 to trigger an
      abort_not_supported_test
     */
    if (err_errno == CR_SERVER_LOST ||
        err_errno == CR_SERVER_GONE_ERROR)
      die("require query '%s' failed: %d: %s", query, err_errno, err_error);
4317
    abort_not_supported_test();
unknown's avatar
unknown committed
4318
  }
unknown's avatar
unknown committed
4319

4320
  if (q->abort_on_error)
4321
    die("query '%s' failed: %d: %s", query, err_errno, err_error);
4322 4323

  for (i= 0 ; (uint) i < q->expected_errors ; i++)
4324
  {
4325 4326 4327 4328
    if (((q->expected_errno[i].type == ERR_ERRNO) &&
         (q->expected_errno[i].code.errnum == err_errno)) ||
        ((q->expected_errno[i].type == ERR_SQLSTATE) &&
         (strcmp(q->expected_errno[i].code.sqlstate, err_sqlstate) == 0)))
4329
    {
4330
      if (!disable_result_log)
4331
      {
4332 4333 4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345
        if (q->expected_errors == 1)
        {
          /* Only log error if there is one possible error */
          dynstr_append_mem(ds, "ERROR ", 6);
          replace_dynstr_append(ds, err_sqlstate);
          dynstr_append_mem(ds, ": ", 2);
          replace_dynstr_append(ds, err_error);
          dynstr_append_mem(ds,"\n",1);
        }
        /* Don't log error if we may not get an error */
        else if (q->expected_errno[0].type == ERR_SQLSTATE ||
                 (q->expected_errno[0].type == ERR_ERRNO &&
                  q->expected_errno[0].code.errnum != 0))
          dynstr_append(ds,"Got one of the listed errors\n");
4346
      }
4347
      /* OK */
4348
      DBUG_VOID_RETURN;
4349
    }
4350
  }
4351

4352
  DBUG_PRINT("info",("i: %d  expected_errors: %d", i, q->expected_errors));
4353

4354 4355 4356 4357 4358 4359 4360 4361
  if (!disable_result_log)
  {
    dynstr_append_mem(ds, "ERROR ",6);
    replace_dynstr_append(ds, err_sqlstate);
    dynstr_append_mem(ds, ": ", 2);
    replace_dynstr_append(ds, err_error);
    dynstr_append_mem(ds, "\n", 1);
  }
4362

4363 4364 4365
  if (i)
  {
    if (q->expected_errno[0].type == ERR_ERRNO)
4366 4367
      die("query '%s' failed with wrong errno %d: '%s', instead of %d...",
          q->query, err_errno, err_error, q->expected_errno[0].code.errnum);
4368
    else
4369 4370 4371
      die("query '%s' failed with wrong sqlstate %s: '%s', instead of %s...",
          q->query, err_sqlstate, err_error,
	  q->expected_errno[0].code.sqlstate);
4372
  }
4373

4374
  DBUG_VOID_RETURN;
4375 4376 4377 4378
}


/*
4379
  Handle absence of errors after execution
4380 4381

  SYNOPSIS
4382
    handle_no_error()
4383 4384 4385
      q - context of query

  RETURN VALUE
4386
    error - function will not return
4387 4388
*/

4389
static void handle_no_error(struct st_query *q)
4390
{
4391
  DBUG_ENTER("handle_no_error");
4392 4393 4394 4395 4396

  if (q->expected_errno[0].type == ERR_ERRNO &&
      q->expected_errno[0].code.errnum != 0)
  {
    /* Error code we wanted was != 0, i.e. not an expected success */
4397 4398
    die("query '%s' succeeded - should have failed with errno %d...",
        q->query, q->expected_errno[0].code.errnum);
4399 4400 4401 4402 4403
  }
  else if (q->expected_errno[0].type == ERR_SQLSTATE &&
           strcmp(q->expected_errno[0].code.sqlstate,"00000") != 0)
  {
    /* SQLSTATE we wanted was != "00000", i.e. not an expected success */
4404 4405
    die("query '%s' succeeded - should have failed with sqlstate %s...",
        q->query, q->expected_errno[0].code.sqlstate);
4406 4407
  }

4408
  DBUG_VOID_RETURN;
4409 4410
}

4411 4412

/*
4413
  Run query using prepared statement C API
unknown's avatar
unknown committed
4414

4415 4416 4417 4418 4419 4420
  SYNPOSIS
  run_query_stmt
  mysql - mysql handle
  command - currrent command pointer
  query - query string to execute
  query_len - length query string to execute
unknown's avatar
unknown committed
4421
  ds - output buffer where to store result form query
4422

4423 4424
  RETURN VALUE
  error - function will not return
4425 4426
*/

unknown's avatar
unknown committed
4427
static void run_query_stmt(MYSQL *mysql, struct st_query *command,
4428 4429
			   char *query, int query_len, DYNAMIC_STRING *ds,
			   DYNAMIC_STRING *ds_warnings)
4430 4431 4432
{
  MYSQL_RES *res= NULL;     /* Note that here 'res' is meta data result set */
  MYSQL_STMT *stmt;
4433
  DYNAMIC_STRING ds_prepare_warnings;
unknown's avatar
unknown committed
4434
  DYNAMIC_STRING ds_execute_warnings;
4435
  DBUG_ENTER("run_query_stmt");
4436
  DBUG_PRINT("query", ("'%-.60s'", query));
4437 4438

  /*
unknown's avatar
unknown committed
4439
    Init a new stmt if it's not already one created for this connection
4440
  */
4441
  if(!(stmt= cur_con->stmt))
4442
  {
4443 4444 4445
    if (!(stmt= mysql_stmt_init(mysql)))
      die("unable to init stmt structure");
    cur_con->stmt= stmt;
4446 4447
  }

unknown's avatar
unknown committed
4448 4449
  /* Init dynamic strings for warnings */
  if (!disable_warnings)
4450
  {
unknown's avatar
unknown committed
4451 4452
    init_dynamic_string(&ds_prepare_warnings, NULL, 0, 256);
    init_dynamic_string(&ds_execute_warnings, NULL, 0, 256);
4453 4454 4455
  }

  /*
4456
    Prepare the query
4457
  */
4458
  if (mysql_stmt_prepare(stmt, query, query_len))
4459
  {
4460
    handle_error(query, command,  mysql_stmt_errno(stmt),
4461
                 mysql_stmt_error(stmt), mysql_stmt_sqlstate(stmt), ds);
4462
    goto end;
4463 4464 4465
  }

  /*
unknown's avatar
unknown committed
4466 4467
    Get the warnings from mysql_stmt_prepare and keep them in a
    separate string
4468
  */
4469 4470
  if (!disable_warnings)
    append_warnings(&ds_prepare_warnings, mysql);
4471 4472

  /*
4473
    No need to call mysql_stmt_bind_param() because we have no
4474 4475 4476
    parameter markers.
  */

4477 4478
  if (cursor_protocol_enabled)
  {
4479 4480 4481
    /*
      Use cursor when retrieving result
    */
unknown's avatar
unknown committed
4482
    ulong type= CURSOR_TYPE_READ_ONLY;
4483
    if (mysql_stmt_attr_set(stmt, STMT_ATTR_CURSOR_TYPE, (void*) &type))
unknown's avatar
unknown committed
4484
      die("mysql_stmt_attr_set(STMT_ATTR_CURSOR_TYPE) failed': %d %s",
4485
          mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
4486
  }
4487

4488 4489
  /*
    Execute the query
4490
  */
unknown's avatar
unknown committed
4491
  if (mysql_stmt_execute(stmt))
4492
  {
4493
    handle_error(query, command, mysql_stmt_errno(stmt),
4494
                 mysql_stmt_error(stmt), mysql_stmt_sqlstate(stmt), ds);
4495
    goto end;
4496 4497
  }

4498 4499 4500 4501 4502 4503 4504
  /*
    When running in cursor_protocol get the warnings from execute here
    and keep them in a separate string for later.
  */
  if (cursor_protocol_enabled && !disable_warnings)
    append_warnings(&ds_execute_warnings, mysql);

4505 4506 4507 4508 4509 4510 4511
  /*
    We instruct that we want to update the "max_length" field in
     mysql_stmt_store_result(), this is our only way to know how much
     buffer to allocate for result data
  */
  {
    my_bool one= 1;
4512
    if (mysql_stmt_attr_set(stmt, STMT_ATTR_UPDATE_MAX_LENGTH, (void*) &one))
unknown's avatar
unknown committed
4513
      die("mysql_stmt_attr_set(STMT_ATTR_UPDATE_MAX_LENGTH) failed': %d %s",
4514
          mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
4515 4516 4517 4518 4519 4520
  }

  /*
    If we got here the statement succeeded and was expected to do so,
    get data. Note that this can still give errors found during execution!
  */
4521
  if (mysql_stmt_store_result(stmt))
4522
  {
4523
    handle_error(query, command, mysql_stmt_errno(stmt),
4524
                 mysql_stmt_error(stmt), mysql_stmt_sqlstate(stmt), ds);
4525
    goto end;
4526
  }
4527

unknown's avatar
unknown committed
4528
  /* If we got here the statement was both executed and read successfully */
4529 4530
  handle_no_error(command);
  if (!disable_result_log)
4531
  {
4532 4533 4534 4535 4536 4537
    /*
      Not all statements creates a result set. If there is one we can
      now create another normal result set that contains the meta
      data. This set can be handled almost like any other non prepared
      statement result set.
    */
4538
    if ((res= mysql_stmt_result_metadata(stmt)) != NULL)
unknown's avatar
unknown committed
4539
    {
4540 4541 4542
      /* Take the column count from meta info */
      MYSQL_FIELD *fields= mysql_fetch_fields(res);
      uint num_fields= mysql_num_fields(res);
4543

4544
      if (display_metadata)
4545
        append_metadata(ds, fields, num_fields);
4546

4547
      if (!display_result_vertically)
4548
        append_table_headings(ds, fields, num_fields);
4549

unknown's avatar
unknown committed
4550
      append_stmt_result(ds, stmt, fields, num_fields);
4551

4552
      mysql_free_result(res);     /* Free normal result set with meta data */
4553

unknown's avatar
unknown committed
4554 4555
      /* Clear prepare warnings */
      dynstr_set(&ds_prepare_warnings, NULL);
unknown's avatar
unknown committed
4556 4557
    }
    else
4558
    {
4559 4560 4561
      /*
	This is a query without resultset
      */
4562 4563
    }

4564
    if (!disable_warnings)
4565
    {
unknown's avatar
unknown committed
4566
      /* Get the warnings from execute */
4567

unknown's avatar
unknown committed
4568 4569
      /* Append warnings to ds - if there are any */
      if (append_warnings(&ds_execute_warnings, mysql) ||
4570 4571 4572
          ds_execute_warnings.length ||
          ds_prepare_warnings.length ||
          ds_warnings->length)
4573
      {
4574
        dynstr_append_mem(ds, "Warnings:\n", 10);
4575 4576 4577
	if (ds_warnings->length)
	  dynstr_append_mem(ds, ds_warnings->str,
			    ds_warnings->length);
unknown's avatar
unknown committed
4578 4579 4580 4581 4582 4583
	if (ds_prepare_warnings.length)
	  dynstr_append_mem(ds, ds_prepare_warnings.str,
			    ds_prepare_warnings.length);
	if (ds_execute_warnings.length)
	  dynstr_append_mem(ds, ds_execute_warnings.str,
			    ds_execute_warnings.length);
4584 4585 4586 4587
      }
    }

    if (!disable_info)
unknown's avatar
unknown committed
4588
      append_info(ds, mysql_affected_rows(mysql), mysql_info(mysql));
4589 4590 4591 4592 4593

  }

end:
  free_replace();
4594
  free_replace_regex();
unknown's avatar
unknown committed
4595 4596 4597 4598 4599 4600 4601 4602 4603 4604 4605 4606 4607
  
  if (!disable_warnings)
  {
    dynstr_free(&ds_prepare_warnings);
    dynstr_free(&ds_execute_warnings);
  }

  /*
    We save the return code (mysql_stmt_errno(stmt)) from the last call sent
    to the server into the mysqltest builtin variable $mysql_errno. This
    variable then can be used from the test case itself.
  */
  
4608
  var_set_errno(mysql_stmt_errno(stmt));
unknown's avatar
unknown committed
4609
#ifndef BUG15518_FIXED
4610
  mysql_stmt_close(stmt);
unknown's avatar
unknown committed
4611 4612
  cur_con->stmt= NULL;
#endif
4613
  DBUG_VOID_RETURN;
4614 4615 4616
}


4617 4618 4619 4620 4621 4622 4623 4624 4625 4626 4627 4628 4629 4630 4631 4632 4633 4634 4635 4636 4637 4638 4639 4640 4641 4642 4643 4644 4645 4646 4647 4648 4649

/*
  Create a util connection if one does not already exists
  and use that to run the query
  This is done to avoid implict commit when creating/dropping objects such
  as view, sp etc.
*/

static int util_query(MYSQL* org_mysql, const char* query){

  MYSQL* mysql;
  DBUG_ENTER("util_query");

  if(!(mysql= cur_con->util_mysql))
  {
    DBUG_PRINT("info", ("Creating util_mysql"));
    if (!(mysql= mysql_init(mysql)))
      die("Failed in mysql_init()");

    if (safe_connect(mysql, org_mysql->host, org_mysql->user,
		     org_mysql->passwd, org_mysql->db, org_mysql->port,
		     org_mysql->unix_socket))
      die("Could not open util connection: %d %s",
	  mysql_errno(mysql), mysql_error(mysql));

    cur_con->util_mysql= mysql;
  }

  return mysql_query(mysql, query);
}



4650 4651
/*
  Run query
4652

4653 4654 4655
  flags control the phased/stages of query execution to be performed
  if QUERY_SEND bit is on, the query will be sent. If QUERY_REAP is on
  the result will be read - for regular query, both bits must be on
4656

4657 4658 4659 4660
  SYNPOSIS
  run_query
  mysql - mysql handle
  command - currrent command pointer
unknown's avatar
unknown committed
4661

4662
*/
4663

4664
static void run_query(MYSQL *mysql, struct st_query *command, int flags)
4665
{
4666
  DYNAMIC_STRING *ds;
4667
  DYNAMIC_STRING ds_result;
4668
  DYNAMIC_STRING ds_warnings;
4669 4670 4671
  DYNAMIC_STRING eval_query;
  char *query;
  int query_len;
4672 4673
  my_bool view_created= 0, sp_created= 0;
  my_bool complete_query= ((flags & QUERY_SEND) && (flags & QUERY_REAP));
4674

4675 4676
  init_dynamic_string(&ds_warnings, NULL, 0, 256);

4677 4678 4679 4680
  /*
    Evaluate query if this is an eval command
   */
  if (command->type == Q_EVAL)
4681
  {
4682
    init_dynamic_string(&eval_query, "", 16384, 65536);
4683
    do_eval(&eval_query, command->query, FALSE);
4684 4685
    query = eval_query.str;
    query_len = eval_query.length;
4686 4687 4688
  }
  else
  {
4689 4690
    query = command->query;
    query_len = strlen(query);
4691 4692
  }

4693
  /*
unknown's avatar
unknown committed
4694
    When command->record_file is set the output of _this_ query
4695 4696 4697 4698 4699
    should be compared with an already existing file
    Create a temporary dynamic string to contain the output from
    this query.
   */
  if (command->record_file[0])
4700
  {
4701 4702
    init_dynamic_string(&ds_result, "", 16384, 65536);
    ds= &ds_result;
4703 4704 4705 4706
  }
  else
    ds= &ds_res;

unknown's avatar
unknown committed
4707 4708
  /*
     Log the query into the output buffer
4709
  */
4710
  if (!disable_query_log && (flags & QUERY_SEND))
4711
  {
4712
    replace_dynstr_append_mem(ds, query, query_len);
4713 4714 4715 4716
    dynstr_append_mem(ds, delimiter, delimiter_length);
    dynstr_append_mem(ds, "\n", 1);
  }

4717 4718 4719
  if (view_protocol_enabled &&
      complete_query &&
      match_re(&view_re, query))
4720
  {
4721
    /*
4722
       Create the query as a view.
unknown's avatar
unknown committed
4723
       Use replace since view can exist from a failed mysqltest run
4724
    */
unknown's avatar
unknown committed
4725 4726 4727
    DYNAMIC_STRING query_str;
    init_dynamic_string(&query_str,
			"CREATE OR REPLACE VIEW mysqltest_tmp_v AS ",
4728 4729
			query_len+64, 256);
    dynstr_append_mem(&query_str, query, query_len);
4730
    if (util_query(mysql, query_str.str))
4731 4732 4733 4734 4735
    {
      /*
	Failed to create the view, this is not fatal
	just run the query the normal way
       */
unknown's avatar
unknown committed
4736
      DBUG_PRINT("view_create_error",
4737 4738
		 ("Failed to create view '%s': %d: %s", query_str.str,
		  mysql_errno(mysql), mysql_error(mysql)));
4739

unknown's avatar
unknown committed
4740 4741 4742
      /* Log error to create view */
      verbose_msg("Failed to create view '%s' %d: %s", query_str.str,
		  mysql_errno(mysql), mysql_error(mysql));
4743 4744 4745 4746 4747 4748 4749
    }
    else
    {
      /*
	Yes, it was possible to create this query as a view
       */
      view_created= 1;
unknown's avatar
unknown committed
4750
      query= (char*)"SELECT * FROM mysqltest_tmp_v";
4751
      query_len = strlen(query);
4752

4753 4754 4755 4756
      /*
	 Collect warnings from create of the view that should otherwise
         have been produced when the SELECT was executed
      */
4757
      append_warnings(&ds_warnings, cur_con->util_mysql);
4758 4759
    }

4760
    dynstr_free(&query_str);
4761 4762 4763

  }

4764 4765 4766
  if (sp_protocol_enabled &&
      complete_query &&
      match_re(&sp_re, query))
4767
  {
4768
    /*
4769
      Create the query as a stored procedure
unknown's avatar
unknown committed
4770
      Drop first since sp can exist from a failed mysqltest run
4771
    */
4772
    DYNAMIC_STRING query_str;
unknown's avatar
unknown committed
4773
    init_dynamic_string(&query_str,
4774
			"DROP PROCEDURE IF EXISTS mysqltest_tmp_sp;",
4775
			query_len+64, 256);
4776
    util_query(mysql, query_str.str);
4777 4778
    dynstr_set(&query_str, "CREATE PROCEDURE mysqltest_tmp_sp()\n");
    dynstr_append_mem(&query_str, query, query_len);
4779
    if (util_query(mysql, query_str.str))
4780
    {
4781
      /*
unknown's avatar
unknown committed
4782
	Failed to create the stored procedure for this query,
4783 4784 4785 4786 4787
	this is not fatal just run the query the normal way
      */
      DBUG_PRINT("sp_create_error",
		 ("Failed to create sp '%s': %d: %s", query_str.str,
		  mysql_errno(mysql), mysql_error(mysql)));
unknown's avatar
unknown committed
4788 4789 4790 4791

      /* Log error to create sp */
      verbose_msg("Failed to create sp '%s' %d: %s", query_str.str,
		  mysql_errno(mysql), mysql_error(mysql));
4792

4793
    }
4794
    else
4795
    {
4796
      sp_created= 1;
unknown's avatar
unknown committed
4797 4798

      query= (char*)"CALL mysqltest_tmp_sp()";
4799
      query_len = strlen(query);
4800
    }
4801
    dynstr_free(&query_str);
4802 4803 4804
  }

  /*
4805
    Find out how to run this query
4806

unknown's avatar
unknown committed
4807
    Always run with normal C API if it's not a complete
4808
    SEND + REAP
4809

4810
    If it is a '?' in the query it may be a SQL level prepared
4811
    statement already and we can't do it twice
4812
  */
4813
  if (ps_protocol_enabled &&
4814 4815
      complete_query &&
      match_re(&ps_re, query))
4816
    run_query_stmt(mysql, command, query, query_len, ds, &ds_warnings);
4817
  else
4818 4819
    run_query_normal(mysql, command, flags, query, query_len,
		     ds, &ds_warnings);
unknown's avatar
unknown committed
4820

4821
  if (sp_created)
4822
  {
4823
    if (util_query(mysql, "DROP PROCEDURE mysqltest_tmp_sp "))
4824
      die("Failed to drop sp: %d: %s", mysql_errno(mysql), mysql_error(mysql));
4825 4826
  }

4827
  if (view_created)
4828
  {
4829
    if (util_query(mysql, "DROP VIEW mysqltest_tmp_v "))
unknown's avatar
unknown committed
4830
      die("Failed to drop view: %d: %s",
4831
	  mysql_errno(mysql), mysql_error(mysql));
4832 4833
  }

unknown's avatar
unknown committed
4834
  if (command->record_file[0])
4835 4836
  {

unknown's avatar
unknown committed
4837 4838
    /* A result file was specified for _this_ query  */
    if (record)
4839
    {
unknown's avatar
unknown committed
4840 4841 4842 4843 4844
      /*
	 Recording in progress
         Dump the output from _this_ query to the specified record_file
      */
      str_to_file(command->record_file, ds->str, ds->length);
4845

unknown's avatar
unknown committed
4846
    } else {
4847

unknown's avatar
unknown committed
4848 4849 4850 4851 4852
      /*
	The output from _this_ query should be checked against an already
	existing file which has been specified using --require or --result
      */
      check_result(ds, command->record_file, command->require_file);
4853 4854 4855
    }
  }

4856
  dynstr_free(&ds_warnings);
4857 4858
  if (ds == &ds_result)
    dynstr_free(&ds_result);
4859
  if (command->type == Q_EVAL)
4860 4861 4862 4863 4864
    dynstr_free(&eval_query);
}


/****************************************************************************\
4865
 *  Functions to detect different SQL statements
4866 4867
\****************************************************************************/

4868
static char *re_eprint(int err)
4869
{
4870
  static char epbuf[100];
unknown's avatar
unknown committed
4871
  size_t len= my_regerror(REG_ITOA|err, (my_regex_t *)NULL,
4872 4873 4874
			  epbuf, sizeof(epbuf));
  assert(len <= sizeof(epbuf));
  return(epbuf);
4875 4876
}

4877
static void init_re_comp(my_regex_t *re, const char* str)
4878
{
4879 4880 4881
  int err= my_regcomp(re, str, (REG_EXTENDED | REG_ICASE | REG_NOSUB),
                      &my_charset_latin1);
  if (err)
4882
  {
4883 4884
    char erbuf[100];
    int len= my_regerror(err, re, erbuf, sizeof(erbuf));
unknown's avatar
unknown committed
4885
    die("error %s, %d/%d `%s'\n",
4886
	re_eprint(err), len, (int)sizeof(erbuf), erbuf);
4887 4888 4889
  }
}

4890
static void init_re(void)
4891
{
unknown's avatar
unknown committed
4892 4893
  /*
     Filter for queries that can be run using the
4894 4895
     MySQL Prepared Statements C API
  */
4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907 4908 4909
  const char *ps_re_str =
    "^("
    "[[:space:]]*REPLACE[[:space:]]|"
    "[[:space:]]*INSERT[[:space:]]|"
    "[[:space:]]*UPDATE[[:space:]]|"
    "[[:space:]]*DELETE[[:space:]]|"
    "[[:space:]]*SELECT[[:space:]]|"
    "[[:space:]]*CREATE[[:space:]]+TABLE[[:space:]]|"
    "[[:space:]]*DO[[:space:]]|"
    "[[:space:]]*SET[[:space:]]+OPTION[[:space:]]|"
    "[[:space:]]*DELETE[[:space:]]+MULTI[[:space:]]|"
    "[[:space:]]*UPDATE[[:space:]]+MULTI[[:space:]]|"
    "[[:space:]]*INSERT[[:space:]]+SELECT[[:space:]])";

unknown's avatar
unknown committed
4910 4911
  /*
     Filter for queries that can be run using the
4912 4913 4914 4915
     Stored procedures
  */
  const char *sp_re_str =ps_re_str;

unknown's avatar
unknown committed
4916
  /*
4917 4918 4919 4920 4921 4922 4923 4924 4925
     Filter for queries that can be run as views
  */
  const char *view_re_str =
    "^("
    "[[:space:]]*SELECT[[:space:]])";

  init_re_comp(&ps_re, ps_re_str);
  init_re_comp(&sp_re, sp_re_str);
  init_re_comp(&view_re, view_re_str);
4926 4927 4928
}


4929
static int match_re(my_regex_t *re, char *str)
4930
{
4931
  int err= my_regexec(re, str, (size_t)0, NULL, 0);
4932 4933 4934 4935 4936

  if (err == 0)
    return 1;
  else if (err == REG_NOMATCH)
    return 0;
4937

4938 4939
  {
    char erbuf[100];
4940 4941 4942
    int len= my_regerror(err, re, erbuf, sizeof(erbuf));
    die("error %s, %d/%d `%s'\n",
	re_eprint(err), len, (int)sizeof(erbuf), erbuf);
4943
  }
4944
  return 0;
4945 4946
}

4947
static void free_re(void)
4948
{
unknown's avatar
unknown committed
4949
  my_regfree(&ps_re);
4950 4951
  my_regfree(&sp_re);
  my_regfree(&view_re);
unknown's avatar
unknown committed
4952
  my_regex_end();
4953 4954 4955 4956
}

/****************************************************************************/

4957
void get_query_type(struct st_query* q)
4958
{
4959 4960
  char save;
  uint type;
unknown's avatar
unknown committed
4961 4962
  DBUG_ENTER("get_query_type");

4963
  if (!parsing_disabled && *q->query == '}')
4964 4965
  {
    q->type = Q_END_BLOCK;
unknown's avatar
unknown committed
4966
    DBUG_VOID_RETURN;
4967 4968
  }
  if (q->type != Q_COMMENT_WITH_COMMAND)
4969
    q->type= parsing_disabled ? Q_COMMENT : Q_QUERY;
4970

4971 4972
  save=q->query[q->first_word_len];
  q->query[q->first_word_len]=0;
4973
  type=find_type(q->query, &command_typelib, 1+2);
4974
  q->query[q->first_word_len]=save;
4975
  if (type > 0)
4976
  {
unknown's avatar
unknown committed
4977
    q->type=(enum enum_commands) type;		/* Found command */
4978 4979
    /*
      If queries are disabled, only recognize
unknown's avatar
unknown committed
4980
      --enable_parsing and --disable_parsing
4981 4982 4983 4984 4985
    */
    if (parsing_disabled && q->type != Q_ENABLE_PARSING &&
        q->type != Q_DISABLE_PARSING)
      q->type= Q_COMMENT;
  }
4986
  else if (q->type == Q_COMMENT_WITH_COMMAND &&
unknown's avatar
unknown committed
4987
	   q->first_word_len &&
4988 4989 4990 4991 4992 4993 4994 4995 4996 4997 4998 4999 5000 5001 5002
           q->query[q->first_word_len-1] == ';')
  {
    /*
       Detect comment with command using extra delimiter
       Ex --disable_query_log;
                             ^ Extra delimiter causing the command
                               to be skipped
    */
    save= q->query[q->first_word_len-1];
    q->query[q->first_word_len-1]= 0;
    type= find_type(q->query, &command_typelib, 1+2);
    q->query[q->first_word_len-1]= save;
    if (type > 0)
      die("Extra delimiter \";\" found");
  }
unknown's avatar
unknown committed
5003
  DBUG_VOID_RETURN;
5004
}
unknown's avatar
unknown committed
5005

unknown's avatar
unknown committed
5006

unknown's avatar
unknown committed
5007
static byte *get_var_key(const byte* var, uint* len,
unknown's avatar
unknown committed
5008
			 my_bool __attribute__((unused)) t)
unknown's avatar
unknown committed
5009 5010 5011 5012 5013 5014 5015
{
  register char* key;
  key = ((VAR*)var)->name;
  *len = ((VAR*)var)->name_len;
  return (byte*)key;
}

unknown's avatar
unknown committed
5016
static VAR *var_init(VAR *v, const char *name, int name_len, const char *val,
unknown's avatar
unknown committed
5017 5018 5019
		     int val_len)
{
  int val_alloc_len;
unknown's avatar
unknown committed
5020
  VAR *tmp_var;
5021
  if (!name_len && name)
unknown's avatar
unknown committed
5022
    name_len = strlen(name);
5023
  if (!val_len && val)
unknown's avatar
unknown committed
5024 5025
    val_len = strlen(val) ;
  val_alloc_len = val_len + 16; /* room to grow */
5026
  if (!(tmp_var=v) && !(tmp_var = (VAR*)my_malloc(sizeof(*tmp_var)
unknown's avatar
unknown committed
5027
						 + name_len+1, MYF(MY_WME))))
unknown's avatar
unknown committed
5028
    die("Out of memory");
unknown's avatar
unknown committed
5029

5030
  tmp_var->name = (name) ? (char*) tmp_var + sizeof(*tmp_var) : 0;
5031
  tmp_var->alloced = (v == 0);
5032

5033
  if (!(tmp_var->str_val = my_malloc(val_alloc_len+1, MYF(MY_WME))))
5034
    die("Out of memory");
unknown's avatar
unknown committed
5035

unknown's avatar
unknown committed
5036
  memcpy(tmp_var->name, name, name_len);
5037
  if (val)
unknown's avatar
unknown committed
5038 5039 5040 5041
  {
    memcpy(tmp_var->str_val, val, val_len);
    tmp_var->str_val[val_len]= 0;
  }
unknown's avatar
unknown committed
5042 5043 5044
  tmp_var->name_len = name_len;
  tmp_var->str_val_len = val_len;
  tmp_var->alloced_len = val_alloc_len;
5045
  tmp_var->int_val = (val) ? atoi(val) : 0;
unknown's avatar
unknown committed
5046
  tmp_var->int_dirty = 0;
5047
  tmp_var->env_s = 0;
unknown's avatar
unknown committed
5048 5049 5050
  return tmp_var;
}

unknown's avatar
unknown committed
5051
static void var_free(void *v)
unknown's avatar
unknown committed
5052
{
unknown's avatar
unknown committed
5053
  my_free(((VAR*) v)->str_val, MYF(MY_WME));
unknown's avatar
unknown committed
5054 5055
  if (((VAR*)v)->alloced)
   my_free((char*) v, MYF(MY_WME));
unknown's avatar
unknown committed
5056 5057 5058
}


5059
static VAR* var_from_env(const char *name, const char *def_val)
unknown's avatar
unknown committed
5060
{
unknown's avatar
unknown committed
5061 5062
  const char *tmp;
  VAR *v;
5063
  if (!(tmp = getenv(name)))
unknown's avatar
unknown committed
5064
    tmp = def_val;
unknown's avatar
unknown committed
5065

5066
  v = var_init(0, name, strlen(name), tmp, strlen(tmp));
unknown's avatar
SCRUM  
unknown committed
5067
  my_hash_insert(&var_hash, (byte*)v);
5068
  return v;
5069
}
unknown's avatar
unknown committed
5070

5071

unknown's avatar
unknown committed
5072
static void init_var_hash(MYSQL *mysql)
unknown's avatar
unknown committed
5073
{
unknown's avatar
unknown committed
5074
  VAR *v;
5075
  DBUG_ENTER("init_var_hash");
unknown's avatar
unknown committed
5076
  if (hash_init(&var_hash, charset_info,
unknown's avatar
unknown committed
5077
                1024, 0, 0, get_var_key, var_free, MYF(0)))
unknown's avatar
unknown committed
5078
    die("Variable hash initialization failed");
5079 5080
  my_hash_insert(&var_hash, (byte*) var_init(0,"BIG_TEST", 0,
                                             (opt_big_test) ? "1" : "0", 0));
unknown's avatar
unknown committed
5081
  v= var_init(0,"MAX_TABLES", 0, (sizeof(ulong) == 4) ? "31" : "62",0);
unknown's avatar
SCRUM  
unknown committed
5082
  my_hash_insert(&var_hash, (byte*) v);
unknown's avatar
unknown committed
5083
  v= var_init(0,"SERVER_VERSION", 0, mysql_get_server_info(mysql), 0);
5084
  my_hash_insert(&var_hash, (byte*) v);  v= var_init(0,"DB", 2, db, 0);
unknown's avatar
unknown committed
5085
  my_hash_insert(&var_hash, (byte*) v);
5086
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
5087
}
5088

5089

unknown's avatar
unknown committed
5090
int main(int argc, char **argv)
unknown's avatar
unknown committed
5091
{
unknown's avatar
unknown committed
5092
  struct st_query *q;
5093 5094
  my_bool require_file=0, q_send_flag=0, abort_flag= 0,
          query_executed= 0;
5095
  char save_file[FN_REFLEN];
5096
  MY_STAT res_info;
5097 5098
  MY_INIT(argv[0]);

unknown's avatar
unknown committed
5099 5100 5101
  /* Use all time until exit if no explicit 'start_timer' */
  timer_start= timer_now();

5102
  save_file[0]=0;
5103
  TMPDIR[0]=0;
5104 5105

  /* Init cons */
unknown's avatar
unknown committed
5106 5107 5108 5109
  memset(cons, 0, sizeof(cons));
  cons_end = cons + MAX_CONS;
  next_con = cons + 1;
  cur_con = cons;
unknown's avatar
unknown committed
5110

5111
  /* Init file stack */
unknown's avatar
unknown committed
5112
  memset(file_stack, 0, sizeof(file_stack));
5113 5114
  file_stack_end= file_stack + MAX_INCLUDE_DEPTH - 1;
  cur_file= file_stack;
unknown's avatar
unknown committed
5115

5116
  /* Init block stack */
5117
  memset(block_stack, 0, sizeof(block_stack));
5118
  block_stack_end= block_stack + BLOCK_STACK_DEPTH - 1;
unknown's avatar
unknown committed
5119 5120 5121 5122
  cur_block= block_stack;
  cur_block->ok= TRUE; /* Outer block should always be executed */
  cur_block->cmd= cmd_none;

5123 5124 5125 5126 5127
  my_init_dynamic_array(&q_lines, sizeof(struct st_query*), INIT_Q_LINES,
		     INIT_Q_LINES);

  memset(&master_pos, 0, sizeof(master_pos));

5128
  init_dynamic_string(&ds_res, "", 0, 65536);
unknown's avatar
unknown committed
5129
  parse_args(argc, argv);
5130 5131

  DBUG_PRINT("info",("result_file: '%s'", result_file ? result_file : ""));
5132 5133 5134
  if (mysql_server_init(embedded_server_arg_count,
			embedded_server_args,
			(char**) embedded_server_groups))
unknown's avatar
unknown committed
5135
    die("Can't initialize MySQL server");
5136
  if (cur_file == file_stack && cur_file->file == 0)
5137
  {
5138 5139
    cur_file->file= stdin;
    cur_file->file_name= my_strdup("<stdin>", MYF(MY_WME));
5140
    cur_file->lineno= 1;
5141
  }
5142 5143 5144
#ifndef EMBEDDED_LIBRARY
  if (manager_host)
    init_manager();
5145
#endif
5146 5147 5148 5149 5150 5151 5152
  init_re();
  ps_protocol_enabled= ps_protocol;
  sp_protocol_enabled= sp_protocol;
  view_protocol_enabled= view_protocol;
  cursor_protocol_enabled= cursor_protocol;
  /* Cursor protcol implies ps protocol */
  if (cursor_protocol_enabled)
5153
    ps_protocol_enabled= 1;
5154

5155
  if (!( mysql_init(&cur_con->mysql)))
unknown's avatar
unknown committed
5156
    die("Failed in mysql_init()");
unknown's avatar
unknown committed
5157 5158
  if (opt_compress)
    mysql_options(&cur_con->mysql,MYSQL_OPT_COMPRESS,NullS);
5159
  mysql_options(&cur_con->mysql, MYSQL_OPT_LOCAL_INFILE, 0);
unknown's avatar
unknown committed
5160
  mysql_options(&cur_con->mysql, MYSQL_SET_CHARSET_NAME, charset_name);
unknown's avatar
unknown committed
5161

unknown's avatar
unknown committed
5162 5163 5164 5165 5166
#ifdef HAVE_OPENSSL
  if (opt_use_ssl)
    mysql_ssl_set(&cur_con->mysql, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
		  opt_ssl_capath, opt_ssl_cipher);
#endif
5167

unknown's avatar
unknown committed
5168
  if (!(cur_con->name = my_strdup("default", MYF(MY_WME))))
unknown's avatar
unknown committed
5169
    die("Out of memory");
unknown's avatar
unknown committed
5170

unknown's avatar
unknown committed
5171
  if (safe_connect(&cur_con->mysql, host, user, pass, db, port, unix_sock))
5172 5173
    die("Could not open connection '%s': %d %s", cur_con->name,
	mysql_errno(&cur_con->mysql), mysql_error(&cur_con->mysql));
unknown's avatar
unknown committed
5174

unknown's avatar
unknown committed
5175 5176
  init_var_hash(&cur_con->mysql);

5177
#ifdef __WIN__
5178
  init_tmp_sh_file();
5179 5180 5181
  init_win_path_patterns();
#endif

5182 5183 5184 5185 5186
  /*
    Initialize $mysql_errno with -1, so we can
    - distinguish it from valid values ( >= 0 ) and
    - detect if there was never a command sent to the server
  */
5187 5188
  var_set_errno(-1);

unknown's avatar
unknown committed
5189 5190 5191 5192 5193
  if (opt_include)
  {
    open_file(opt_include);
  }

unknown's avatar
unknown committed
5194
  while (!abort_flag && !read_query(&q))
5195 5196 5197 5198
  {
    int current_line_inc = 1, processed = 0;
    if (q->type == Q_UNKNOWN || q->type == Q_COMMENT_WITH_COMMAND)
      get_query_type(q);
unknown's avatar
unknown committed
5199
    if (cur_block->ok)
unknown's avatar
unknown committed
5200
    {
5201
      q->last_argument= q->first_argument;
5202 5203
      processed = 1;
      switch (q->type) {
5204
      case Q_CONNECT:
5205
        do_connect(q);
5206
        break;
5207
      case Q_CONNECTION: select_connection(q); break;
unknown's avatar
unknown committed
5208
      case Q_DISCONNECT:
5209
      case Q_DIRTY_CLOSE:
unknown's avatar
unknown committed
5210
	close_connection(q); break;
5211
      case Q_RPL_PROBE: do_rpl_probe(q); break;
5212
      case Q_ENABLE_RPL_PARSE:	 do_enable_rpl_parse(q); break;
unknown's avatar
unknown committed
5213
      case Q_DISABLE_RPL_PARSE:  do_disable_rpl_parse(q); break;
5214
      case Q_ENABLE_QUERY_LOG:   disable_query_log=0; break;
unknown's avatar
unknown committed
5215
      case Q_DISABLE_QUERY_LOG:  disable_query_log=1; break;
5216 5217
      case Q_ENABLE_ABORT_ON_ERROR:  abort_on_error=1; break;
      case Q_DISABLE_ABORT_ON_ERROR: abort_on_error=0; break;
unknown's avatar
unknown committed
5218 5219
      case Q_ENABLE_RESULT_LOG:  disable_result_log=0; break;
      case Q_DISABLE_RESULT_LOG: disable_result_log=1; break;
5220 5221
      case Q_ENABLE_WARNINGS:    disable_warnings=0; break;
      case Q_DISABLE_WARNINGS:   disable_warnings=1; break;
5222 5223
      case Q_ENABLE_PS_WARNINGS:    disable_ps_warnings=0; break;
      case Q_DISABLE_PS_WARNINGS:   disable_ps_warnings=1; break;
5224 5225
      case Q_ENABLE_INFO:        disable_info=0; break;
      case Q_DISABLE_INFO:       disable_info=1; break;
5226
      case Q_ENABLE_METADATA:    display_metadata=1; break;
5227
      case Q_DISABLE_METADATA:   display_metadata=0; break;
5228
      case Q_SOURCE: do_source(q); break;
5229 5230
      case Q_SLEEP: do_sleep(q, 0); break;
      case Q_REAL_SLEEP: do_sleep(q, 1); break;
5231
      case Q_WAIT_FOR_SLAVE_TO_STOP: do_wait_for_slave_to_stop(q); break;
5232
      case Q_REQUIRE_MANAGER: do_require_manager(q); break;
5233
#ifndef EMBEDDED_LIBRARY
5234 5235
      case Q_SERVER_START: do_server_start(q); break;
      case Q_SERVER_STOP: do_server_stop(q); break;
5236
#endif
unknown's avatar
unknown committed
5237 5238
      case Q_INC: do_modify_var(q, DO_INC); break;
      case Q_DEC: do_modify_var(q, DO_DEC); break;
5239
      case Q_ECHO: do_echo(q); query_executed= 1; break;
5240
      case Q_SYSTEM: do_system(q); break;
5241 5242 5243
      case Q_DELIMITER:
	strmake(delimiter, q->first_argument, sizeof(delimiter) - 1);
	delimiter_length= strlen(delimiter);
5244
        q->last_argument= q->first_argument+delimiter_length;
5245
	break;
5246 5247 5248 5249 5250 5251
      case Q_DISPLAY_VERTICAL_RESULTS:
        display_result_vertically= TRUE;
        break;
      case Q_DISPLAY_HORIZONTAL_RESULTS:
	display_result_vertically= FALSE;
        break;
5252
      case Q_LET: do_let(q); break;
5253
      case Q_EVAL_RESULT:
5254
        eval_result = 1; break;
5255
      case Q_EVAL:
5256
	if (q->query == q->query_buf)
unknown's avatar
unknown committed
5257
        {
unknown's avatar
unknown committed
5258
	  q->query= q->first_argument;
unknown's avatar
unknown committed
5259 5260
          q->first_word_len= 0;
        }
5261
	/* fall through */
5262
      case Q_QUERY_VERTICAL:
5263
      case Q_QUERY_HORIZONTAL:
5264 5265
      {
	my_bool old_display_result_vertically= display_result_vertically;
unknown's avatar
unknown committed
5266
	/* fix up query pointer if this is first iteration for this line */
5267 5268
	if (q->query == q->query_buf)
	  q->query += q->first_word_len + 1;
5269
	display_result_vertically= (q->type==Q_QUERY_VERTICAL);
5270 5271 5272 5273 5274 5275
	if (save_file[0])
	{
	  strmov(q->record_file,save_file);
	  q->require_file=require_file;
	  save_file[0]=0;
	}
5276
	run_query(&cur_con->mysql, q, QUERY_REAP|QUERY_SEND);
5277
	display_result_vertically= old_display_result_vertically;
5278
        q->last_argument= q->end;
5279
        query_executed= 1;
5280 5281
	break;
      }
5282
      case Q_QUERY:
5283
      case Q_REAP:
5284
      {
5285 5286 5287 5288 5289
	/*
	  We read the result always regardless of the mode for both full
	  query and read-result only (reap)
	*/
	int flags = QUERY_REAP;
5290
	if (q->type != Q_REAP) /* for a full query, enable the send stage */
unknown's avatar
unknown committed
5291
	  flags |= QUERY_SEND;
unknown's avatar
unknown committed
5292 5293 5294 5295 5296
	if (q_send_flag)
	{
	  flags= QUERY_SEND;
	  q_send_flag=0;
	}
5297
	if (save_file[0])
5298
	{
5299 5300 5301
	  strmov(q->record_file,save_file);
	  q->require_file=require_file;
	  save_file[0]=0;
5302
	}
5303
	run_query(&cur_con->mysql, q, flags);
5304
	query_executed= 1;
5305
        q->last_argument= q->end;
unknown's avatar
unknown committed
5306
	break;
5307
      }
unknown's avatar
unknown committed
5308
      case Q_SEND:
unknown's avatar
unknown committed
5309 5310
	if (!q->query[q->first_word_len])
	{
unknown's avatar
unknown committed
5311
	  /* This happens when we use 'send' on its own line */
unknown's avatar
unknown committed
5312 5313 5314
	  q_send_flag=1;
	  break;
	}
unknown's avatar
unknown committed
5315
	/* fix up query pointer if this is first iteration for this line */
unknown's avatar
unknown committed
5316
	if (q->query == q->query_buf)
5317
	  q->query += q->first_word_len;
5318
	/*
unknown's avatar
unknown committed
5319
	  run_query() can execute a query partially, depending on the flags.
5320 5321 5322
	  QUERY_SEND flag without QUERY_REAP tells it to just send the
	  query and read the result some time later when reap instruction
	  is given on this connection.
5323
	 */
5324
	run_query(&cur_con->mysql, q, QUERY_SEND);
5325
	query_executed= 1;
5326
        q->last_argument= q->end;
unknown's avatar
unknown committed
5327
	break;
5328 5329 5330 5331
      case Q_RESULT:
	get_file_name(save_file,q);
	require_file=0;
	break;
unknown's avatar
unknown committed
5332
      case Q_ERROR:
5333
        global_expected_errors=get_errcodes(global_expected_errno,q);
unknown's avatar
unknown committed
5334
	break;
5335 5336 5337 5338
      case Q_REQUIRE:
	get_file_name(save_file,q);
	require_file=1;
	break;
unknown's avatar
unknown committed
5339 5340 5341
      case Q_REPLACE:
	get_replace(q);
	break;
5342 5343 5344 5345
      case Q_REPLACE_REGEX:
        get_replace_regex(q);
        break;

5346 5347 5348
      case Q_REPLACE_COLUMN:
	get_replace_column(q);
	break;
5349 5350
      case Q_SAVE_MASTER_POS: do_save_master_pos(); break;
      case Q_SYNC_WITH_MASTER: do_sync_with_master(q); break;
5351 5352 5353 5354
      case Q_SYNC_SLAVE_WITH_MASTER:
      {
	do_save_master_pos();
	if (*q->first_argument)
5355
	  select_connection(q);
5356
	else
5357 5358
	  select_connection_name("slave");
	do_sync_with_master2(0);
5359 5360
	break;
      }
5361
      case Q_COMMENT:				/* Ignore row */
5362
      case Q_COMMENT_WITH_COMMAND:
5363
        q->last_argument= q->end;
5364
	break;
5365 5366 5367
      case Q_PING:
	(void) mysql_ping(&cur_con->mysql);
	break;
5368
      case Q_EXEC:
unknown's avatar
unknown committed
5369
	do_exec(q);
5370
	query_executed= 1;
5371
	break;
unknown's avatar
unknown committed
5372 5373 5374 5375 5376 5377 5378 5379 5380
      case Q_START_TIMER:
	/* Overwrite possible earlier start of timer */
	timer_start= timer_now();
	break;
      case Q_END_TIMER:
	/* End timer before ending mysqltest */
	timer_output();
	got_end_timer= TRUE;
	break;
5381
      case Q_CHARACTER_SET:
unknown's avatar
unknown committed
5382 5383
	set_charset(q);
	break;
5384 5385 5386 5387 5388 5389
      case Q_DISABLE_PS_PROTOCOL:
        ps_protocol_enabled= 0;
        break;
      case Q_ENABLE_PS_PROTOCOL:
        ps_protocol_enabled= ps_protocol;
        break;
5390
      case Q_DISABLE_RECONNECT:
5391 5392 5393
      {
        my_bool reconnect= 0;
        mysql_options(&cur_con->mysql, MYSQL_OPT_RECONNECT, (char *)&reconnect);
5394
        break;
5395
      }
5396
      case Q_ENABLE_RECONNECT:
5397 5398 5399
      {
        my_bool reconnect= 1;
        mysql_options(&cur_con->mysql, MYSQL_OPT_RECONNECT, (char *)&reconnect);
5400
        break;
5401
      }
5402 5403 5404 5405 5406
      case Q_DISABLE_PARSING:
        parsing_disabled++;
        break;
      case Q_ENABLE_PARSING:
        /*
unknown's avatar
unknown committed
5407
          Ensure we don't get parsing_disabled < 0 as this would accidentally
5408 5409 5410 5411 5412
          disable code we don't want to have disabled
        */
        if (parsing_disabled > 0)
          parsing_disabled--;
        break;
5413

unknown's avatar
unknown committed
5414 5415 5416
      case Q_EXIT:
        abort_flag= 1;
        break;
5417 5418 5419 5420

      default:
        processed= 0;
        break;
5421 5422
      }
    }
5423

5424 5425
    if (!processed)
    {
5426
      current_line_inc= 0;
unknown's avatar
unknown committed
5427
      switch (q->type) {
unknown's avatar
unknown committed
5428 5429
      case Q_WHILE: do_block(cmd_while, q); break;
      case Q_IF: do_block(cmd_if, q); break;
5430 5431 5432
      case Q_END_BLOCK: do_done(q); break;
      default: current_line_inc = 1; break;
      }
unknown's avatar
unknown committed
5433
    }
5434 5435
    else
      check_eol_junk(q->last_argument);
unknown's avatar
unknown committed
5436

5437 5438 5439 5440 5441 5442 5443 5444 5445
    if (q->type != Q_ERROR)
    {
      /*
        As soon as any non "error" command has been executed,
        the array with expected errors should be cleared
      */
      global_expected_errors= 0;
      bzero((gptr) global_expected_errno, sizeof(global_expected_errno));
    }
unknown's avatar
unknown committed
5446

5447 5448 5449
    parser.current_line += current_line_inc;
  }

5450 5451
  start_lineno= 0;

5452
  /*
unknown's avatar
unknown committed
5453 5454 5455
    The whole test has been executed _sucessfully_.
    Time to compare result or save it to record file.
    The entire output from test is now kept in ds_res.
unknown's avatar
unknown committed
5456
  */
5457
  if (ds_res.length)
5458
  {
5459 5460
    if (result_file)
    {
5461 5462 5463 5464 5465
      if (record)
      {
	/* Dump the output from test to result file */
	str_to_file(result_file, ds_res.str, ds_res.length);
      }
5466
      else
5467
      {
unknown's avatar
unknown committed
5468 5469 5470 5471
	/* Check that the output from test is equal to result file
	   - detect missing result file
	   - detect zero size result file
	 */
5472 5473
	check_result(&ds_res, result_file, 0);
      }
5474
    }
5475
    else
5476
    {
unknown's avatar
unknown committed
5477
      /* No result_file specified to compare with, print to stdout */
5478 5479
      printf("%s", ds_res.str);
    }
5480
  }
unknown's avatar
unknown committed
5481
  else
unknown's avatar
unknown committed
5482
  {
unknown's avatar
unknown committed
5483
    die("The test didn't produce any output");
unknown's avatar
unknown committed
5484
  }
5485

unknown's avatar
unknown committed
5486
  if (!query_executed && result_file && my_stat(result_file, &res_info, 0))
unknown's avatar
unknown committed
5487
  {
unknown's avatar
unknown committed
5488 5489 5490
    /*
      my_stat() successful on result file. Check if we have not run a
      single query, but we do have a result file that contains data.
5491 5492
      Note that we don't care, if my_stat() fails. For example, for a
      non-existing or non-readable file, we assume it's fine to have
unknown's avatar
unknown committed
5493 5494 5495
      no query output from the test file, e.g. regarded as no error.
    */
    die("No queries executed but result file found!");
unknown's avatar
unknown committed
5496
  }
5497

unknown's avatar
unknown committed
5498

5499
  dynstr_free(&ds_res);
5500

unknown's avatar
unknown committed
5501 5502
  if (!got_end_timer)
    timer_output();				/* No end_timer cmd, end it */
5503
  free_used_memory();
5504
  my_end(MY_CHECK_ERROR);
5505 5506 5507 5508 5509 5510

  /* Yes, if we got this far the test has suceeded! Sakila smiles */
  if (!silent)
    printf("ok\n");
  exit(0);
  return 0;				/* Keep compiler happy */
unknown's avatar
unknown committed
5511
}
unknown's avatar
unknown committed
5512

5513

5514 5515 5516 5517 5518 5519
/*
  Read arguments for embedded server and put them into
  embedded_server_args_count and embedded_server_args[]
*/


unknown's avatar
unknown committed
5520
static int read_server_arguments(const char *name)
5521 5522 5523 5524 5525 5526 5527 5528 5529 5530 5531 5532 5533 5534 5535 5536
{
  char argument[1024],buff[FN_REFLEN], *str=0;
  FILE *file;

  if (!test_if_hard_path(name))
  {
    strxmov(buff, opt_basedir, name, NullS);
    name=buff;
  }
  fn_format(buff,name,"","",4);

  if (!embedded_server_arg_count)
  {
    embedded_server_arg_count=1;
    embedded_server_args[0]= (char*) "";		/* Progname */
  }
unknown's avatar
unknown committed
5537
  if (!(file=my_fopen(buff, O_RDONLY | FILE_BINARY, MYF(MY_WME))))
5538 5539 5540 5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551 5552 5553 5554 5555 5556 5557 5558
    return 1;
  while (embedded_server_arg_count < MAX_SERVER_ARGS &&
	 (str=fgets(argument,sizeof(argument), file)))
  {
    *(strend(str)-1)=0;				/* Remove end newline */
    if (!(embedded_server_args[embedded_server_arg_count]=
	  (char*) my_strdup(str,MYF(MY_WME))))
    {
      my_fclose(file,MYF(0));
      return 1;
    }
    embedded_server_arg_count++;
  }
  my_fclose(file,MYF(0));
  if (str)
  {
    fprintf(stderr,"Too many arguments in option file: %s\n",name);
    return 1;
  }
  return 0;
}
unknown's avatar
unknown committed
5559

unknown's avatar
unknown committed
5560 5561 5562 5563 5564 5565 5566 5567 5568 5569 5570 5571 5572 5573 5574 5575 5576 5577 5578 5579 5580 5581 5582 5583 5584 5585 5586
/****************************************************************************\
 *
 *  A primitive timer that give results in milliseconds if the
 *  --timer-file=<filename> is given. The timer result is written
 *  to that file when the result is available. To not confuse
 *  mysql-test-run with an old obsolete result, we remove the file
 *  before executing any commands. The time we measure is
 *
 *    - If no explicit 'start_timer' or 'end_timer' is given in the
 *      test case, the timer measure how long we execute in mysqltest.
 *
 *    - If only 'start_timer' is given we measure how long we execute
 *      from that point until we terminate mysqltest.
 *
 *    - If only 'end_timer' is given we measure how long we execute
 *      from that we enter mysqltest to the 'end_timer' is command is
 *      executed.
 *
 *    - If both 'start_timer' and 'end_timer' are given we measure
 *      the time between executing the two commands.
 *
\****************************************************************************/

static void timer_output(void)
{
  if (timer_file)
  {
5587
    char buf[32], *end;
unknown's avatar
unknown committed
5588
    ulonglong timer= timer_now() - timer_start;
5589 5590
    end= longlong2str(timer, buf, 10);
    str_to_file(timer_file,buf, (int) (end-buf));
unknown's avatar
unknown committed
5591 5592 5593 5594 5595 5596 5597 5598
  }
}

static ulonglong timer_now(void)
{
  return my_getsystime() / 10000;
}

unknown's avatar
unknown committed
5599 5600 5601 5602 5603 5604 5605 5606 5607 5608 5609 5610 5611
/****************************************************************************
* Handle replacement of strings
****************************************************************************/

#define PC_MALLOC		256	/* Bytes for pointers */
#define PS_MALLOC		512	/* Bytes for data */

#define SPACE_CHAR	256
#define START_OF_LINE	257
#define END_OF_LINE	258
#define LAST_CHAR_CODE	259

typedef struct st_replace {
5612
  bool	 found;
unknown's avatar
unknown committed
5613 5614 5615 5616 5617 5618 5619 5620 5621 5622 5623 5624 5625 5626 5627 5628 5629 5630 5631 5632 5633 5634 5635 5636 5637 5638 5639 5640 5641 5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663 5664 5665 5666 5667 5668 5669 5670 5671 5672 5673 5674 5675 5676 5677 5678 5679 5680 5681 5682 5683 5684 5685 5686 5687 5688 5689 5690 5691 5692 5693 5694 5695 5696 5697 5698 5699 5700 5701 5702 5703 5704 5705 5706 5707 5708 5709 5710 5711 5712 5713 5714 5715 5716 5717
  struct st_replace *next[256];
} REPLACE;

typedef struct st_replace_found {
  bool found;
  char *replace_string;
  uint to_offset;
  int from_offset;
} REPLACE_STRING;

#ifndef WORD_BIT
#define WORD_BIT (8*sizeof(uint))
#endif


static int insert_pointer_name(reg1 POINTER_ARRAY *pa,my_string name)
{
  uint i,length,old_count;
  byte *new_pos;
  const char **new_array;
  DBUG_ENTER("insert_pointer_name");

  if (! pa->typelib.count)
  {
    if (!(pa->typelib.type_names=(const char **)
	  my_malloc(((PC_MALLOC-MALLOC_OVERHEAD)/
		     (sizeof(my_string)+sizeof(*pa->flag))*
		     (sizeof(my_string)+sizeof(*pa->flag))),MYF(MY_WME))))
      DBUG_RETURN(-1);
    if (!(pa->str= (byte*) my_malloc((uint) (PS_MALLOC-MALLOC_OVERHEAD),
				     MYF(MY_WME))))
    {
      my_free((gptr) pa->typelib.type_names,MYF(0));
      DBUG_RETURN (-1);
    }
    pa->max_count=(PC_MALLOC-MALLOC_OVERHEAD)/(sizeof(byte*)+
					       sizeof(*pa->flag));
    pa->flag= (int7*) (pa->typelib.type_names+pa->max_count);
    pa->length=0;
    pa->max_length=PS_MALLOC-MALLOC_OVERHEAD;
    pa->array_allocs=1;
  }
  length=(uint) strlen(name)+1;
  if (pa->length+length >= pa->max_length)
  {
    if (!(new_pos= (byte*) my_realloc((gptr) pa->str,
				      (uint) (pa->max_length+PS_MALLOC),
				      MYF(MY_WME))))
      DBUG_RETURN(1);
    if (new_pos != pa->str)
    {
      my_ptrdiff_t diff=PTR_BYTE_DIFF(new_pos,pa->str);
      for (i=0 ; i < pa->typelib.count ; i++)
	pa->typelib.type_names[i]= ADD_TO_PTR(pa->typelib.type_names[i],diff,
					      char*);
      pa->str=new_pos;
    }
    pa->max_length+=PS_MALLOC;
  }
  if (pa->typelib.count >= pa->max_count-1)
  {
    int len;
    pa->array_allocs++;
    len=(PC_MALLOC*pa->array_allocs - MALLOC_OVERHEAD);
    if (!(new_array=(const char **) my_realloc((gptr) pa->typelib.type_names,
					       (uint) len/
					 (sizeof(byte*)+sizeof(*pa->flag))*
					 (sizeof(byte*)+sizeof(*pa->flag)),
					 MYF(MY_WME))))
      DBUG_RETURN(1);
    pa->typelib.type_names=new_array;
    old_count=pa->max_count;
    pa->max_count=len/(sizeof(byte*) + sizeof(*pa->flag));
    pa->flag= (int7*) (pa->typelib.type_names+pa->max_count);
    memcpy((byte*) pa->flag,(my_string) (pa->typelib.type_names+old_count),
	   old_count*sizeof(*pa->flag));
  }
  pa->flag[pa->typelib.count]=0;			/* Reset flag */
  pa->typelib.type_names[pa->typelib.count++]= pa->str+pa->length;
  pa->typelib.type_names[pa->typelib.count]= NullS;	/* Put end-mark */
  VOID(strmov(pa->str+pa->length,name));
  pa->length+=length;
  DBUG_RETURN(0);
} /* insert_pointer_name */


	/* free pointer array */

void free_pointer_array(POINTER_ARRAY *pa)
{
  if (pa->typelib.count)
  {
    pa->typelib.count=0;
    my_free((gptr) pa->typelib.type_names,MYF(0));
    pa->typelib.type_names=0;
    my_free((gptr) pa->str,MYF(0));
  }
} /* free_pointer_array */


	/* Code for replace rutines */

#define SET_MALLOC_HUNC 64

typedef struct st_rep_set {
5718 5719
  uint	*bits;				/* Pointer to used sets */
  short next[LAST_CHAR_CODE];		/* Pointer to next sets */
unknown's avatar
unknown committed
5720 5721
  uint	found_len;			/* Best match to date */
  int	found_offset;
5722 5723
  uint	table_offset;
  uint	size_of_bits;			/* For convinience */
unknown's avatar
unknown committed
5724 5725 5726 5727 5728 5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742 5743 5744 5745 5746 5747 5748 5749 5750 5751
} REP_SET;

typedef struct st_rep_sets {
  uint		count;			/* Number of sets */
  uint		extra;			/* Extra sets in buffer */
  uint		invisible;		/* Sets not chown */
  uint		size_of_bits;
  REP_SET	*set,*set_buffer;
  uint		*bit_buffer;
} REP_SETS;

typedef struct st_found_set {
  uint table_offset;
  int found_offset;
} FOUND_SET;

typedef struct st_follow {
  int chr;
  uint table_offset;
  uint len;
} FOLLOWS;


static int init_sets(REP_SETS *sets,uint states);
static REP_SET *make_new_set(REP_SETS *sets);
static void make_sets_invisible(REP_SETS *sets);
static void free_last_set(REP_SETS *sets);
static void free_sets(REP_SETS *sets);
5752 5753
static void internal_set_bit(REP_SET *set, uint bit);
static void internal_clear_bit(REP_SET *set, uint bit);
unknown's avatar
unknown committed
5754 5755 5756 5757 5758 5759 5760 5761 5762 5763 5764 5765 5766 5767 5768 5769 5770 5771 5772 5773 5774 5775 5776 5777 5778 5779 5780 5781 5782 5783 5784 5785 5786 5787 5788 5789 5790 5791 5792 5793 5794 5795 5796 5797 5798 5799 5800 5801 5802 5803 5804 5805 5806 5807 5808 5809 5810 5811 5812 5813 5814 5815 5816 5817 5818 5819 5820 5821 5822 5823 5824 5825 5826 5827 5828 5829
static void or_bits(REP_SET *to,REP_SET *from);
static void copy_bits(REP_SET *to,REP_SET *from);
static int cmp_bits(REP_SET *set1,REP_SET *set2);
static int get_next_bit(REP_SET *set,uint lastpos);
static int find_set(REP_SETS *sets,REP_SET *find);
static int find_found(FOUND_SET *found_set,uint table_offset,
			  int found_offset);
static uint start_at_word(my_string pos);
static uint end_of_word(my_string pos);
static uint replace_len(my_string pos);

static uint found_sets=0;


	/* Init a replace structure for further calls */

REPLACE *init_replace(my_string *from, my_string *to,uint count,
		      my_string word_end_chars)
{
  uint i,j,states,set_nr,len,result_len,max_length,found_end,bits_set,bit_nr;
  int used_sets,chr,default_state;
  char used_chars[LAST_CHAR_CODE],is_word_end[256];
  my_string pos,to_pos,*to_array;
  REP_SETS sets;
  REP_SET *set,*start_states,*word_states,*new_set;
  FOLLOWS *follow,*follow_ptr;
  REPLACE *replace;
  FOUND_SET *found_set;
  REPLACE_STRING *rep_str;
  DBUG_ENTER("init_replace");

  /* Count number of states */
  for (i=result_len=max_length=0 , states=2 ; i < count ; i++)
  {
    len=replace_len(from[i]);
    if (!len)
    {
      errno=EINVAL;
      my_message(0,"No to-string for last from-string",MYF(ME_BELL));
      DBUG_RETURN(0);
    }
    states+=len+1;
    result_len+=(uint) strlen(to[i])+1;
    if (len > max_length)
      max_length=len;
  }
  bzero((char*) is_word_end,sizeof(is_word_end));
  for (i=0 ; word_end_chars[i] ; i++)
    is_word_end[(uchar) word_end_chars[i]]=1;

  if (init_sets(&sets,states))
    DBUG_RETURN(0);
  found_sets=0;
  if (!(found_set= (FOUND_SET*) my_malloc(sizeof(FOUND_SET)*max_length*count,
					  MYF(MY_WME))))
  {
    free_sets(&sets);
    DBUG_RETURN(0);
  }
  VOID(make_new_set(&sets));			/* Set starting set */
  make_sets_invisible(&sets);			/* Hide previus sets */
  used_sets=-1;
  word_states=make_new_set(&sets);		/* Start of new word */
  start_states=make_new_set(&sets);		/* This is first state */
  if (!(follow=(FOLLOWS*) my_malloc((states+2)*sizeof(FOLLOWS),MYF(MY_WME))))
  {
    free_sets(&sets);
    my_free((gptr) found_set,MYF(0));
    DBUG_RETURN(0);
  }

	/* Init follow_ptr[] */
  for (i=0, states=1, follow_ptr=follow+1 ; i < count ; i++)
  {
    if (from[i][0] == '\\' && from[i][1] == '^')
    {
5830
      internal_set_bit(start_states,states+1);
unknown's avatar
unknown committed
5831 5832 5833 5834 5835 5836 5837 5838
      if (!from[i][2])
      {
	start_states->table_offset=i;
	start_states->found_offset=1;
      }
    }
    else if (from[i][0] == '\\' && from[i][1] == '$')
    {
5839 5840
      internal_set_bit(start_states,states);
      internal_set_bit(word_states,states);
unknown's avatar
unknown committed
5841 5842 5843 5844 5845 5846 5847 5848
      if (!from[i][2] && start_states->table_offset == (uint) ~0)
      {
	start_states->table_offset=i;
	start_states->found_offset=0;
      }
    }
    else
    {
5849
      internal_set_bit(word_states,states);
unknown's avatar
unknown committed
5850
      if (from[i][0] == '\\' && (from[i][1] == 'b' && from[i][2]))
5851
	internal_set_bit(start_states,states+1);
unknown's avatar
unknown committed
5852
      else
5853
	internal_set_bit(start_states,states);
unknown's avatar
unknown committed
5854 5855 5856 5857 5858 5859 5860 5861 5862 5863 5864 5865 5866 5867 5868 5869 5870 5871 5872 5873 5874 5875 5876 5877 5878 5879 5880 5881 5882 5883 5884 5885 5886 5887 5888 5889 5890 5891 5892 5893 5894 5895 5896 5897 5898 5899 5900 5901 5902 5903 5904 5905 5906 5907 5908 5909 5910 5911 5912 5913 5914 5915 5916 5917 5918 5919 5920 5921 5922 5923 5924 5925 5926 5927 5928 5929 5930 5931 5932 5933 5934 5935 5936 5937 5938 5939 5940 5941 5942 5943 5944 5945 5946 5947 5948 5949 5950 5951 5952 5953 5954 5955 5956 5957 5958
    }
    for (pos=from[i], len=0; *pos ; pos++)
    {
      if (*pos == '\\' && *(pos+1))
      {
	pos++;
	switch (*pos) {
	case 'b':
	  follow_ptr->chr = SPACE_CHAR;
	  break;
	case '^':
	  follow_ptr->chr = START_OF_LINE;
	  break;
	case '$':
	  follow_ptr->chr = END_OF_LINE;
	  break;
	case 'r':
	  follow_ptr->chr = '\r';
	  break;
	case 't':
	  follow_ptr->chr = '\t';
	  break;
	case 'v':
	  follow_ptr->chr = '\v';
	  break;
	default:
	  follow_ptr->chr = (uchar) *pos;
	  break;
	}
      }
      else
	follow_ptr->chr= (uchar) *pos;
      follow_ptr->table_offset=i;
      follow_ptr->len= ++len;
      follow_ptr++;
    }
    follow_ptr->chr=0;
    follow_ptr->table_offset=i;
    follow_ptr->len=len;
    follow_ptr++;
    states+=(uint) len+1;
  }


  for (set_nr=0,pos=0 ; set_nr < sets.count ; set_nr++)
  {
    set=sets.set+set_nr;
    default_state= 0;				/* Start from beginning */

    /* If end of found-string not found or start-set with current set */

    for (i= (uint) ~0; (i=get_next_bit(set,i)) ;)
    {
      if (!follow[i].chr)
      {
	if (! default_state)
	  default_state= find_found(found_set,set->table_offset,
				    set->found_offset+1);
      }
    }
    copy_bits(sets.set+used_sets,set);		/* Save set for changes */
    if (!default_state)
      or_bits(sets.set+used_sets,sets.set);	/* Can restart from start */

    /* Find all chars that follows current sets */
    bzero((char*) used_chars,sizeof(used_chars));
    for (i= (uint) ~0; (i=get_next_bit(sets.set+used_sets,i)) ;)
    {
      used_chars[follow[i].chr]=1;
      if ((follow[i].chr == SPACE_CHAR && !follow[i+1].chr &&
	   follow[i].len > 1) || follow[i].chr == END_OF_LINE)
	used_chars[0]=1;
    }

    /* Mark word_chars used if \b is in state */
    if (used_chars[SPACE_CHAR])
      for (pos= word_end_chars ; *pos ; pos++)
	used_chars[(int) (uchar) *pos] = 1;

    /* Handle other used characters */
    for (chr= 0 ; chr < 256 ; chr++)
    {
      if (! used_chars[chr])
	set->next[chr]= chr ? default_state : -1;
      else
      {
	new_set=make_new_set(&sets);
	set=sets.set+set_nr;			/* if realloc */
	new_set->table_offset=set->table_offset;
	new_set->found_len=set->found_len;
	new_set->found_offset=set->found_offset+1;
	found_end=0;

	for (i= (uint) ~0 ; (i=get_next_bit(sets.set+used_sets,i)) ; )
	{
	  if (!follow[i].chr || follow[i].chr == chr ||
	      (follow[i].chr == SPACE_CHAR &&
	       (is_word_end[chr] ||
		(!chr && follow[i].len > 1 && ! follow[i+1].chr))) ||
	      (follow[i].chr == END_OF_LINE && ! chr))
	  {
	    if ((! chr || (follow[i].chr && !follow[i+1].chr)) &&
		follow[i].len > found_end)
	      found_end=follow[i].len;
	    if (chr && follow[i].chr)
5959
	      internal_set_bit(new_set,i+1);		/* To next set */
unknown's avatar
unknown committed
5960
	    else
5961
	      internal_set_bit(new_set,i);
unknown's avatar
unknown committed
5962 5963 5964 5965 5966 5967 5968 5969 5970 5971 5972 5973 5974 5975 5976 5977
	  }
	}
	if (found_end)
	{
	  new_set->found_len=0;			/* Set for testing if first */
	  bits_set=0;
	  for (i= (uint) ~0; (i=get_next_bit(new_set,i)) ;)
	  {
	    if ((follow[i].chr == SPACE_CHAR ||
		 follow[i].chr == END_OF_LINE) && ! chr)
	      bit_nr=i+1;
	    else
	      bit_nr=i;
	    if (follow[bit_nr-1].len < found_end ||
		(new_set->found_len &&
		 (chr == 0 || !follow[bit_nr].chr)))
5978
	      internal_clear_bit(new_set,i);
unknown's avatar
unknown committed
5979 5980 5981 5982 5983 5984 5985 5986 5987 5988 5989 5990 5991 5992 5993 5994 5995 5996 5997 5998 5999 6000 6001 6002 6003 6004 6005 6006 6007 6008 6009 6010 6011 6012 6013 6014 6015 6016 6017 6018 6019 6020 6021 6022 6023 6024 6025 6026 6027 6028 6029 6030 6031 6032 6033 6034 6035 6036 6037 6038 6039 6040 6041 6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057 6058 6059 6060 6061 6062 6063 6064 6065 6066 6067 6068 6069 6070 6071 6072 6073 6074 6075 6076 6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089 6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117 6118 6119 6120 6121 6122 6123 6124 6125 6126
	    else
	    {
	      if (chr == 0 || !follow[bit_nr].chr)
	      {					/* best match  */
		new_set->table_offset=follow[bit_nr].table_offset;
		if (chr || (follow[i].chr == SPACE_CHAR ||
			    follow[i].chr == END_OF_LINE))
		  new_set->found_offset=found_end;	/* New match */
		new_set->found_len=found_end;
	      }
	      bits_set++;
	    }
	  }
	  if (bits_set == 1)
	  {
	    set->next[chr] = find_found(found_set,
					new_set->table_offset,
					new_set->found_offset);
	    free_last_set(&sets);
	  }
	  else
	    set->next[chr] = find_set(&sets,new_set);
	}
	else
	  set->next[chr] = find_set(&sets,new_set);
      }
    }
  }

	/* Alloc replace structure for the replace-state-machine */

  if ((replace=(REPLACE*) my_malloc(sizeof(REPLACE)*(sets.count)+
				    sizeof(REPLACE_STRING)*(found_sets+1)+
				    sizeof(my_string)*count+result_len,
				    MYF(MY_WME | MY_ZEROFILL))))
  {
    rep_str=(REPLACE_STRING*) (replace+sets.count);
    to_array=(my_string*) (rep_str+found_sets+1);
    to_pos=(my_string) (to_array+count);
    for (i=0 ; i < count ; i++)
    {
      to_array[i]=to_pos;
      to_pos=strmov(to_pos,to[i])+1;
    }
    rep_str[0].found=1;
    rep_str[0].replace_string=0;
    for (i=1 ; i <= found_sets ; i++)
    {
      pos=from[found_set[i-1].table_offset];
      rep_str[i].found= !bcmp(pos,"\\^",3) ? 2 : 1;
      rep_str[i].replace_string=to_array[found_set[i-1].table_offset];
      rep_str[i].to_offset=found_set[i-1].found_offset-start_at_word(pos);
      rep_str[i].from_offset=found_set[i-1].found_offset-replace_len(pos)+
	end_of_word(pos);
    }
    for (i=0 ; i < sets.count ; i++)
    {
      for (j=0 ; j < 256 ; j++)
	if (sets.set[i].next[j] >= 0)
	  replace[i].next[j]=replace+sets.set[i].next[j];
	else
	  replace[i].next[j]=(REPLACE*) (rep_str+(-sets.set[i].next[j]-1));
    }
  }
  my_free((gptr) follow,MYF(0));
  free_sets(&sets);
  my_free((gptr) found_set,MYF(0));
  DBUG_PRINT("exit",("Replace table has %d states",sets.count));
  DBUG_RETURN(replace);
}


static int init_sets(REP_SETS *sets,uint states)
{
  bzero((char*) sets,sizeof(*sets));
  sets->size_of_bits=((states+7)/8);
  if (!(sets->set_buffer=(REP_SET*) my_malloc(sizeof(REP_SET)*SET_MALLOC_HUNC,
					      MYF(MY_WME))))
    return 1;
  if (!(sets->bit_buffer=(uint*) my_malloc(sizeof(uint)*sets->size_of_bits*
					   SET_MALLOC_HUNC,MYF(MY_WME))))
  {
    my_free((gptr) sets->set,MYF(0));
    return 1;
  }
  return 0;
}

	/* Make help sets invisible for nicer codeing */

static void make_sets_invisible(REP_SETS *sets)
{
  sets->invisible=sets->count;
  sets->set+=sets->count;
  sets->count=0;
}

static REP_SET *make_new_set(REP_SETS *sets)
{
  uint i,count,*bit_buffer;
  REP_SET *set;
  if (sets->extra)
  {
    sets->extra--;
    set=sets->set+ sets->count++;
    bzero((char*) set->bits,sizeof(uint)*sets->size_of_bits);
    bzero((char*) &set->next[0],sizeof(set->next[0])*LAST_CHAR_CODE);
    set->found_offset=0;
    set->found_len=0;
    set->table_offset= (uint) ~0;
    set->size_of_bits=sets->size_of_bits;
    return set;
  }
  count=sets->count+sets->invisible+SET_MALLOC_HUNC;
  if (!(set=(REP_SET*) my_realloc((gptr) sets->set_buffer,
				   sizeof(REP_SET)*count,
				  MYF(MY_WME))))
    return 0;
  sets->set_buffer=set;
  sets->set=set+sets->invisible;
  if (!(bit_buffer=(uint*) my_realloc((gptr) sets->bit_buffer,
				      (sizeof(uint)*sets->size_of_bits)*count,
				      MYF(MY_WME))))
    return 0;
  sets->bit_buffer=bit_buffer;
  for (i=0 ; i < count ; i++)
  {
    sets->set_buffer[i].bits=bit_buffer;
    bit_buffer+=sets->size_of_bits;
  }
  sets->extra=SET_MALLOC_HUNC;
  return make_new_set(sets);
}

static void free_last_set(REP_SETS *sets)
{
  sets->count--;
  sets->extra++;
  return;
}

static void free_sets(REP_SETS *sets)
{
  my_free((gptr)sets->set_buffer,MYF(0));
  my_free((gptr)sets->bit_buffer,MYF(0));
  return;
}

6127
static void internal_set_bit(REP_SET *set, uint bit)
unknown's avatar
unknown committed
6128 6129 6130 6131 6132
{
  set->bits[bit / WORD_BIT] |= 1 << (bit % WORD_BIT);
  return;
}

6133
static void internal_clear_bit(REP_SET *set, uint bit)
unknown's avatar
unknown committed
6134 6135 6136 6137 6138 6139 6140 6141 6142 6143 6144 6145 6146 6147 6148 6149 6150 6151 6152 6153 6154 6155 6156 6157 6158 6159 6160 6161 6162 6163 6164 6165 6166 6167 6168 6169 6170 6171 6172 6173 6174 6175 6176 6177 6178 6179 6180 6181 6182 6183 6184
{
  set->bits[bit / WORD_BIT] &= ~ (1 << (bit % WORD_BIT));
  return;
}


static void or_bits(REP_SET *to,REP_SET *from)
{
  reg1 uint i;
  for (i=0 ; i < to->size_of_bits ; i++)
    to->bits[i]|=from->bits[i];
  return;
}

static void copy_bits(REP_SET *to,REP_SET *from)
{
  memcpy((byte*) to->bits,(byte*) from->bits,
	 (size_t) (sizeof(uint) * to->size_of_bits));
}

static int cmp_bits(REP_SET *set1,REP_SET *set2)
{
  return bcmp((byte*) set1->bits,(byte*) set2->bits,
	      sizeof(uint) * set1->size_of_bits);
}


	/* Get next set bit from set. */

static int get_next_bit(REP_SET *set,uint lastpos)
{
  uint pos,*start,*end,bits;

  start=set->bits+ ((lastpos+1) / WORD_BIT);
  end=set->bits + set->size_of_bits;
  bits=start[0] & ~((1 << ((lastpos+1) % WORD_BIT)) -1);

  while (! bits && ++start < end)
    bits=start[0];
  if (!bits)
    return 0;
  pos=(uint) (start-set->bits)*WORD_BIT;
  while (! (bits & 1))
  {
    bits>>=1;
    pos++;
  }
  return pos;
}

	/* find if there is a same set in sets. If there is, use it and
unknown's avatar
unknown committed
6185
	   free given set, else put in given set in sets and return its
unknown's avatar
unknown committed
6186 6187 6188 6189 6190 6191 6192 6193 6194 6195 6196 6197 6198 6199 6200 6201 6202 6203
	   position */

static int find_set(REP_SETS *sets,REP_SET *find)
{
  uint i;
  for (i=0 ; i < sets->count-1 ; i++)
  {
    if (!cmp_bits(sets->set+i,find))
    {
      free_last_set(sets);
      return i;
    }
  }
  return i;				/* return new postion */
}

	/* find if there is a found_set with same table_offset & found_offset
	   If there is return offset to it, else add new offset and return pos.
unknown's avatar
unknown committed
6204
	   Pos returned is -offset-2 in found_set_structure because it is
unknown's avatar
unknown committed
6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218 6219 6220 6221 6222 6223 6224 6225 6226 6227 6228 6229 6230 6231 6232 6233 6234 6235 6236 6237 6238 6239 6240 6241 6242 6243 6244 6245 6246 6247 6248 6249 6250 6251
	   saved in set->next and set->next[] >= 0 points to next set and
	   set->next[] == -1 is reserved for end without replaces.
	   */

static int find_found(FOUND_SET *found_set,uint table_offset, int found_offset)
{
  int i;
  for (i=0 ; (uint) i < found_sets ; i++)
    if (found_set[i].table_offset == table_offset &&
	found_set[i].found_offset == found_offset)
      return -i-2;
  found_set[i].table_offset=table_offset;
  found_set[i].found_offset=found_offset;
  found_sets++;
  return -i-2;				/* return new postion */
}

	/* Return 1 if regexp starts with \b or ends with \b*/

static uint start_at_word(my_string pos)
{
  return (((!bcmp(pos,"\\b",2) && pos[2]) || !bcmp(pos,"\\^",2)) ? 1 : 0);
}

static uint end_of_word(my_string pos)
{
  my_string end=strend(pos);
  return ((end > pos+2 && !bcmp(end-2,"\\b",2)) ||
	  (end >= pos+2 && !bcmp(end-2,"\\$",2))) ?
	    1 : 0;
}


static uint replace_len(my_string str)
{
  uint len=0;
  while (*str)
  {
    if (str[0] == '\\' && str[1])
      str++;
    str++;
    len++;
  }
  return len;
}


6252 6253 6254
/* Replace strings while appending to ds */
void replace_strings_append(REPLACE *rep, DYNAMIC_STRING* ds,
                            const char *str, int len)
unknown's avatar
unknown committed
6255 6256 6257
{
  reg1 REPLACE *rep_pos;
  reg2 REPLACE_STRING *rep_str;
6258 6259
  const char *start, *from;
  DBUG_ENTER("replace_strings_append");
unknown's avatar
unknown committed
6260

6261
  start= from= str;
unknown's avatar
unknown committed
6262
  rep_pos=rep+1;
6263
  for (;;)
unknown's avatar
unknown committed
6264
  {
6265 6266
    /* Loop through states */
    DBUG_PRINT("info", ("Looping through states"));
unknown's avatar
unknown committed
6267
    while (!rep_pos->found)
6268 6269 6270
      rep_pos= rep_pos->next[(uchar) *from++];

    /* Does this state contain a string to be replaced */
unknown's avatar
unknown committed
6271 6272
    if (!(rep_str = ((REPLACE_STRING*) rep_pos))->replace_string)
    {
6273 6274 6275 6276
      /* No match found */
      dynstr_append_mem(ds, start, from - start - 1);
      DBUG_PRINT("exit", ("Found no more string to replace, appended: %s", start));
      DBUG_VOID_RETURN;
unknown's avatar
unknown committed
6277
    }
6278 6279 6280 6281 6282 6283 6284 6285 6286 6287 6288 6289 6290

    /* Found a string that needs to be replaced */
    DBUG_PRINT("info", ("found: %d, to_offset: %d, from_offset: %d, string: %s",
                        rep_str->found, rep_str->to_offset,
                        rep_str->from_offset, rep_str->replace_string));

    /* Append part of original string before replace string */
    dynstr_append_mem(ds, start, (from - rep_str->to_offset) - start);

    /* Append replace string */
    dynstr_append_mem(ds, rep_str->replace_string,
                      strlen(rep_str->replace_string));

unknown's avatar
unknown committed
6291
    if (!*(from-=rep_str->from_offset) && rep_pos->found != 2)
6292 6293 6294 6295 6296 6297 6298
    {
      /* End of from string */
      DBUG_PRINT("exit", ("Found end of from string"));
      DBUG_VOID_RETURN;
    }
    DBUG_ASSERT(from <= str+len);
    start= from;
unknown's avatar
unknown committed
6299 6300 6301 6302 6303
    rep_pos=rep;
  }
}


6304 6305 6306 6307 6308 6309 6310 6311 6312 6313 6314 6315 6316 6317 6318 6319 6320 6321 6322 6323 6324 6325 6326 6327 6328 6329 6330 6331 6332 6333 6334 6335 6336 6337
/****************************************************************************
 Replace results for a column
*****************************************************************************/

static void free_replace_column()
{
  uint i;
  for (i=0 ; i < max_replace_column ; i++)
  {
    if (replace_column[i])
    {
      my_free(replace_column[i], 0);
      replace_column[i]= 0;
    }
  }
  max_replace_column= 0;
}

/*
  Get arguments for replace_columns. The syntax is:
  replace-column column_number to_string [column_number to_string ...]
  Where each argument may be quoted with ' or "
  A argument may also be a variable, in which case the value of the
  variable is replaced.
*/

static void get_replace_column(struct st_query *q)
{
  char *from=q->first_argument;
  char *buff,*start;
  DBUG_ENTER("get_replace_columns");

  free_replace_column();
  if (!*from)
6338
    die("Missing argument in %s", q->query);
6339 6340 6341 6342 6343 6344 6345 6346 6347 6348

  /* Allocate a buffer for results */
  start=buff=my_malloc(strlen(from)+1,MYF(MY_WME | MY_FAE));
  while (*from)
  {
    char *to;
    uint column_number;

    to= get_string(&buff, &from, q);
    if (!(column_number= atoi(to)) || column_number > MAX_COLUMNS)
6349
      die("Wrong column number to replace_column in '%s'", q->query);
6350
    if (!*from)
6351
      die("Wrong number of arguments to replace_column in '%s'", q->query);
6352 6353 6354 6355 6356 6357
    to= get_string(&buff, &from, q);
    my_free(replace_column[column_number-1], MY_ALLOW_ZERO_PTR);
    replace_column[column_number-1]= my_strdup(to, MYF(MY_WME | MY_FAE));
    set_if_bigger(max_replace_column, column_number);
  }
  my_free(start, MYF(0));
6358
  q->last_argument= q->end;
6359
}
unknown's avatar
unknown committed
6360 6361


6362 6363