mysqltest.c 147 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
unknown's avatar
unknown committed
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 118 119
  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
  set to type ERR_EMPTY. When an SQL statement return 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="./";
159
static int port = 0;
160
static my_bool opt_big_test= 0, opt_compress= 0, silent= 0, verbose = 0;
161 162 163 164 165
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;
166
static int parsing_disabled= 0;
167
const char *manager_user="root",*manager_host=0;
168 169 170 171
char *manager_pass=0;
int manager_port=MYSQL_MANAGER_PORT;
int manager_wait_timeout=3;
MYSQL_MANAGER* manager=0;
172

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

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

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

unknown's avatar
unknown committed
189
static char TMPDIR[FN_REFLEN];
190 191
static char delimiter[MAX_DELIMITER]= DEFAULT_DELIMITER;
static uint delimiter_length= 1;
unknown's avatar
unknown committed
192

unknown's avatar
unknown committed
193 194 195 196 197 198 199 200 201 202
/* 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;
203

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

207 208 209
static int embedded_server_arg_count=0;
static char *embedded_server_args[MAX_SERVER_ARGS];

210
static my_bool display_result_vertically= FALSE, display_metadata= FALSE;
211

unknown's avatar
unknown committed
212 213 214 215 216 217 218
/* 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);

219 220 221 222 223 224 225 226
/* 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);
227

228 229
static const char *embedded_server_groups[]=
{
230 231 232 233 234 235
  "server",
  "embedded",
  "mysqltest_SERVER",
  NullS
};

236 237
DYNAMIC_ARRAY q_lines;

unknown's avatar
unknown committed
238 239
#include "sslopt-vars.h"

240
typedef struct
unknown's avatar
unknown committed
241 242 243 244 245
{
  char file[FN_REFLEN];
  ulong pos;
} MASTER_POS ;

unknown's avatar
unknown committed
246 247 248
struct connection
{
  MYSQL mysql;
249 250
  /* Used when creating views and sp, to avoid implicit commit */
  MYSQL* util_mysql;
unknown's avatar
unknown committed
251
  char *name;
252
  MYSQL_STMT* stmt;
unknown's avatar
unknown committed
253 254
};

unknown's avatar
unknown committed
255 256 257 258
typedef struct
{
  int read_lines,current_line;
} PARSER;
259 260

PARSER parser;
unknown's avatar
unknown committed
261
MASTER_POS master_pos;
unknown's avatar
unknown committed
262 263
/* if set, all results are concated and compared against this file */
const char *result_file = 0;
264

265
typedef struct
266
{
unknown's avatar
unknown committed
267
  char *name;
unknown's avatar
unknown committed
268
  int name_len;
unknown's avatar
unknown committed
269
  char *str_val;
270 271 272 273
  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
274
  int alloced;
275 276 277 278
} VAR;

VAR var_reg[10];
/*Perl/shell-like variable registers */
unknown's avatar
unknown committed
279
HASH var_hash;
280
my_bool disable_query_log=0, disable_result_log=0, disable_warnings=0;
281
my_bool disable_ps_warnings= 0;
282
my_bool disable_info= 1;			/* By default off */
unknown's avatar
unknown committed
283
my_bool abort_on_error= 1;
284

unknown's avatar
unknown committed
285 286 287
struct connection cons[MAX_CONS];
struct connection* cur_con, *next_con, *cons_end;

unknown's avatar
unknown committed
288 289 290
  /* Add new commands before Q_UNKNOWN !*/

enum enum_commands {
291
Q_CONNECTION=1,     Q_QUERY,
unknown's avatar
unknown committed
292
Q_CONNECT,	    Q_SLEEP, Q_REAL_SLEEP,
293 294 295 296 297 298
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,
299 300 301
Q_SYNC_WITH_MASTER,
Q_SYNC_SLAVE_WITH_MASTER,
Q_ERROR,
302
Q_SEND,		    Q_REAP,
303
Q_DIRTY_CLOSE,	    Q_REPLACE, Q_REPLACE_COLUMN,
304 305
Q_PING,		    Q_EVAL,
Q_RPL_PROBE,	    Q_ENABLE_RPL_PARSE,
306
Q_DISABLE_RPL_PARSE, Q_EVAL_RESULT,
unknown's avatar
unknown committed
307
Q_ENABLE_QUERY_LOG, Q_DISABLE_QUERY_LOG,
unknown's avatar
unknown committed
308
Q_ENABLE_RESULT_LOG, Q_DISABLE_RESULT_LOG,
309
Q_SERVER_START, Q_SERVER_STOP,Q_REQUIRE_MANAGER,
310
Q_WAIT_FOR_SLAVE_TO_STOP,
311
Q_ENABLE_WARNINGS, Q_DISABLE_WARNINGS,
312
Q_ENABLE_PS_WARNINGS, Q_DISABLE_PS_WARNINGS,
313
Q_ENABLE_INFO, Q_DISABLE_INFO,
314
Q_ENABLE_METADATA, Q_DISABLE_METADATA,
315
Q_EXEC, Q_DELIMITER,
316
Q_DISABLE_ABORT_ON_ERROR, Q_ENABLE_ABORT_ON_ERROR,
317
Q_DISPLAY_VERTICAL_RESULTS, Q_DISPLAY_HORIZONTAL_RESULTS,
unknown's avatar
unknown committed
318
Q_QUERY_VERTICAL, Q_QUERY_HORIZONTAL,
unknown's avatar
unknown committed
319
Q_START_TIMER, Q_END_TIMER,
320
Q_CHARACTER_SET, Q_DISABLE_PS_PROTOCOL, Q_ENABLE_PS_PROTOCOL,
unknown's avatar
unknown committed
321
Q_EXIT,
322
Q_DISABLE_RECONNECT, Q_ENABLE_RECONNECT,
unknown's avatar
unknown committed
323
Q_IF,
324
Q_DISABLE_PARSING, Q_ENABLE_PARSING,
325

326 327
Q_UNKNOWN,			       /* Unknown command.   */
Q_COMMENT,			       /* Comments, ignored. */
328
Q_COMMENT_WITH_COMMAND
unknown's avatar
unknown committed
329 330
};

331
/* this should really be called command */
332
struct st_query
unknown's avatar
unknown committed
333
{
334
  char *query, *query_buf,*first_argument,*last_argument,*end;
unknown's avatar
unknown committed
335
  int first_word_len;
336
  my_bool abort_on_error, require_file;
337
  match_err expected_errno[MAX_EXPECTED_ERRORS];
unknown's avatar
unknown committed
338
  uint expected_errors;
unknown's avatar
unknown committed
339
  char record_file[FN_REFLEN];
unknown's avatar
unknown committed
340
  enum enum_commands type;
unknown's avatar
unknown committed
341 342
};

343 344 345 346 347
const char *command_names[]=
{
  "connection",
  "query",
  "connect",
unknown's avatar
unknown committed
348 349
  /* the difference between sleep and real_sleep is that sleep will use
     the delay from command line (--sleep) if there is one.
350
     real_sleep always uses delay from mysqltest's command line argument.
unknown's avatar
unknown committed
351 352 353 354
     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
   */
355 356 357 358 359 360 361 362 363 364 365 366 367 368 369
  "sleep",
  "real_sleep",
  "inc",
  "dec",
  "source",
  "disconnect",
  "let",
  "echo",
  "while",
  "end",
  "system",
  "result",
  "require",
  "save_master_pos",
  "sync_with_master",
370
  "sync_slave_with_master",
371 372 373 374 375
  "error",
  "send",
  "reap",
  "dirty_close",
  "replace_result",
376
  "replace_column",
377 378 379 380 381 382 383 384 385 386 387 388 389 390
  "ping",
  "eval",
  "rpl_probe",
  "enable_rpl_parse",
  "disable_rpl_parse",
  "eval_result",
  "enable_query_log",
  "disable_query_log",
  "enable_result_log",
  "disable_result_log",
  "server_start",
  "server_stop",
  "require_manager",
  "wait_for_slave_to_stop",
391 392
  "enable_warnings",
  "disable_warnings",
393 394
  "enable_ps_warnings",
  "disable_ps_warnings",
395
  "enable_info",
396
  "disable_info",
397 398
  "enable_metadata",
  "disable_metadata",
399
  "exec",
400
  "delimiter",
401 402
  "disable_abort_on_error",
  "enable_abort_on_error",
403
  "vertical_results",
404
  "horizontal_results",
405
  "query_vertical",
406
  "query_horizontal",
unknown's avatar
unknown committed
407 408
  "start_timer",
  "end_timer",
unknown's avatar
unknown committed
409
  "character_set",
410 411
  "disable_ps_protocol",
  "enable_ps_protocol",
unknown's avatar
unknown committed
412
  "exit",
413 414
  "disable_reconnect",
  "enable_reconnect",
unknown's avatar
unknown committed
415
  "if",
416 417
  "disable_parsing",
  "enable_parsing",
unknown's avatar
unknown committed
418
  0
419 420 421
};

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

424
DYNAMIC_STRING ds_res;
unknown's avatar
unknown committed
425
static void die(const char *fmt, ...);
unknown's avatar
unknown committed
426
static void init_var_hash();
427
static VAR* var_from_env(const char *, const char *);
428
static byte* get_var_key(const byte* rec, uint* len, my_bool t);
unknown's avatar
unknown committed
429
static VAR* var_init(VAR* v, const char *name, int name_len, const char *val,
unknown's avatar
unknown committed
430 431 432
		     int val_len);

static void var_free(void* v);
433

434
void dump_result_to_reject_file(const char *record_file, char *buf, int size);
unknown's avatar
unknown committed
435

436
int close_connection(struct st_query*);
unknown's avatar
unknown committed
437
static void set_charset(struct st_query*);
438 439
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
440 441
int eval_expr(VAR* v, const char *p, const char** p_end);
static int read_server_arguments(const char *name);
442

443
/* Definitions for replace result */
unknown's avatar
unknown committed
444 445 446 447 448

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. */
449
  uint	array_allocs,max_count,length,max_length;
unknown's avatar
unknown committed
450 451 452 453 454
} POINTER_ARRAY;

struct st_replace;
struct st_replace *init_replace(my_string *from, my_string *to, uint count,
				my_string word_end_chars);
455
void free_replace();
unknown's avatar
unknown committed
456
static int insert_pointer_name(reg1 POINTER_ARRAY *pa,my_string name);
457 458
static void replace_strings_append(struct st_replace *rep, DYNAMIC_STRING* ds,
                                   const char *from, int len);
unknown's avatar
unknown committed
459
void free_pointer_array(POINTER_ARRAY *pa);
460 461
static void do_eval(DYNAMIC_STRING *query_eval, const char *query,
                    my_bool pass_through_escape_chars);
462
static void str_to_file(const char *fname, char *str, int size);
463 464 465 466

#ifdef __WIN__
static void free_win_path_patterns();
#endif
unknown's avatar
unknown committed
467 468

struct st_replace *glob_replace;
469
static int eval_result = 0;
unknown's avatar
unknown committed
470

471 472 473 474 475 476 477
/* 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
478
/* Disable functions that only exist in MySQL 4.0 */
unknown's avatar
SCRUM  
unknown committed
479
#if MYSQL_VERSION_ID < 40000
unknown's avatar
unknown committed
480 481 482
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; }
483
my_bool mysql_rpl_probe(MYSQL *mysql __attribute__((unused))) { return 1; }
unknown's avatar
unknown committed
484
#endif
485 486
static void replace_dynstr_append_mem(DYNAMIC_STRING *ds, const char *val,
				      int len);
487
static void replace_dynstr_append(DYNAMIC_STRING *ds, const char *val);
488 489 490 491
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);
492

493 494
static void do_eval(DYNAMIC_STRING* query_eval, const char *query,
                    my_bool pass_through_escape_chars)
495
{
496
  const char *p;
497
  register char c, next_c;
498 499
  register int escaped = 0;
  VAR* v;
unknown's avatar
unknown committed
500
  DBUG_ENTER("do_eval");
unknown's avatar
unknown committed
501

502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518
  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 '\\':
519
      next_c= *(p+1);
520 521 522 523 524
      if (escaped)
      {
	escaped = 0;
	dynstr_append_mem(query_eval, p, 1);
      }
525 526 527
      else if (next_c == '\\' || next_c == '$')
      {
        /* Set escaped only if next char is \ or $ */
528
	escaped = 1;
529 530 531 532 533 534

        if (pass_through_escape_chars)
        {
          /* The escape char should be added to the output string. */
          dynstr_append_mem(query_eval, p, 1);
        }
535 536 537
      }
      else
	dynstr_append_mem(query_eval, p, 1);
538 539 540 541
      break;
    default:
      dynstr_append_mem(query_eval, p, 1);
      break;
542
    }
543
  }
unknown's avatar
unknown committed
544
  DBUG_VOID_RETURN;
545
}
unknown's avatar
unknown committed
546

547

548 549
static void close_cons()
{
550 551 552
  DBUG_ENTER("close_cons");
  for (--next_con; next_con >= cons; --next_con)
  {
553 554 555
    if (next_con->stmt)
      mysql_stmt_close(next_con->stmt);
    next_con->stmt= 0;
556
    mysql_close(&next_con->mysql);
557 558
    if (next_con->util_mysql)
      mysql_close(next_con->util_mysql);
559 560 561 562 563
    my_free(next_con->name, MYF(MY_ALLOW_ZERO_PTR));
  }
  DBUG_VOID_RETURN;
}

564

565 566
static void close_files()
{
unknown's avatar
unknown committed
567
  DBUG_ENTER("close_files");
568
  for (; cur_file >= file_stack; cur_file--)
569
  {
570 571 572 573 574
    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
575 576
  }
  DBUG_VOID_RETURN;
577 578
}

579

580 581 582 583
static void free_used_memory()
{
  uint i;
  DBUG_ENTER("free_used_memory");
584
#ifndef EMBEDDED_LIBRARY
585 586
  if (manager)
    mysql_manager_close(manager);
587
#endif
588 589
  close_cons();
  close_files();
unknown's avatar
unknown committed
590
  hash_free(&var_hash);
unknown's avatar
unknown committed
591

592 593 594
  for (i=0 ; i < q_lines.elements ; i++)
  {
    struct st_query **q= dynamic_element(&q_lines, i, struct st_query**);
595
    my_free((gptr) (*q)->query_buf,MYF(MY_ALLOW_ZERO_PTR));
596 597
    my_free((gptr) (*q),MYF(0));
  }
598
  for (i=0; i < 10; i++)
unknown's avatar
unknown committed
599 600 601 602
  {
    if (var_reg[i].alloced_len)
      my_free(var_reg[i].str_val, MYF(MY_WME));
  }
603 604
  while (embedded_server_arg_count > 1)
    my_free(embedded_server_args[--embedded_server_arg_count],MYF(0));
605 606
  delete_dynamic(&q_lines);
  dynstr_free(&ds_res);
unknown's avatar
unknown committed
607
  free_replace();
608
  free_replace_column();
609 610
  my_free(pass,MYF(MY_ALLOW_ZERO_PTR));
  free_defaults(default_argv);
unknown's avatar
unknown committed
611
  mysql_server_end();
612
  free_re();
613 614 615
#ifdef __WIN__
  free_win_path_patterns();
#endif
616
  DBUG_VOID_RETURN;
617 618
}

619
static void die(const char *fmt, ...)
620 621
{
  va_list args;
622
  DBUG_ENTER("die");
623
  va_start(args, fmt);
624 625
  if (fmt)
  {
626 627
    fprintf(stderr, "mysqltest: ");
    if (cur_file && cur_file != file_stack)
628
      fprintf(stderr, "In included file \"%s\": ",
629
              cur_file->file_name);
unknown's avatar
unknown committed
630 631
    if (start_lineno != 0)
      fprintf(stderr, "At line %u: ", start_lineno);
632 633
    vfprintf(stderr, fmt, args);
    fprintf(stderr, "\n");
unknown's avatar
unknown committed
634
    fflush(stderr);
635
  }
636
  va_end(args);
637
  free_used_memory();
638
  my_end(MY_CHECK_ERROR);
unknown's avatar
WL#2930  
unknown committed
639 640 641 642

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

643 644 645
  exit(1);
}

646 647
/* Note that we will get some memory leaks when calling this! */

648 649
static void abort_not_supported_test()
{
650
  DBUG_ENTER("abort_not_supported_test");
651 652 653
  fprintf(stderr, "This test is not supported by this installation\n");
  if (!silent)
    printf("skipped\n");
654
  free_used_memory();
655
  my_end(MY_CHECK_ERROR);
unknown's avatar
unknown committed
656
  exit(62);
657 658
}

659
static void verbose_msg(const char *fmt, ...)
660 661
{
  va_list args;
662 663 664
  DBUG_ENTER("verbose_msg");
  if (!verbose)
    DBUG_VOID_RETURN;
665 666 667

  va_start(args, fmt);

unknown's avatar
unknown committed
668
  fprintf(stderr, "mysqltest: ");
669
  if (start_lineno != 0)
unknown's avatar
unknown committed
670
    fprintf(stderr, "At line %u: ", start_lineno);
671 672 673
  vfprintf(stderr, fmt, args);
  fprintf(stderr, "\n");
  va_end(args);
674
  DBUG_VOID_RETURN;
675 676
}

unknown's avatar
unknown committed
677

678 679
void init_parser()
{
680 681
  parser.current_line= parser.read_lines= 0;
  memset(&var_reg, 0, sizeof(var_reg));
682
}
unknown's avatar
unknown committed
683 684


685
static int dyn_string_cmp(DYNAMIC_STRING* ds, const char *fname)
unknown's avatar
unknown committed
686 687
{
  MY_STAT stat_info;
688 689
  char *tmp, *res_ptr;
  char eval_file[FN_REFLEN];
unknown's avatar
unknown committed
690
  int res;
691
  uint res_len;
unknown's avatar
unknown committed
692
  int fd;
693
  DYNAMIC_STRING res_ds;
694 695
  DBUG_ENTER("dyn_string_cmp");

696 697 698 699 700 701 702 703 704
  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)))
705
    die(NullS);
706 707 708 709
  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));
710
    DBUG_PRINT("info",("result: '%s'", ds->str));
711
    DBUG_RETURN(RESULT_LENGTH_MISMATCH);
712
  }
713
  if (!(tmp = (char*) my_malloc(stat_info.st_size + 1, MYF(MY_WME))))
714
    die(NullS);
715 716

  if ((fd = my_open(eval_file, O_RDONLY, MYF(MY_WME))) < 0)
717
    die(NullS);
718
  if (my_read(fd, (byte*)tmp, stat_info.st_size, MYF(MY_WME|MY_NABP)))
719
    die(NullS);
720 721 722 723
  tmp[stat_info.st_size] = 0;
  init_dynamic_string(&res_ds, "", 0, 65536);
  if (eval_result)
  {
724
    do_eval(&res_ds, tmp, FALSE);
725
    res_ptr = res_ds.str;
726
    if ((res_len = res_ds.length) != ds->length)
727
    {
728
      res= RESULT_LENGTH_MISMATCH;
729 730 731 732 733 734 735 736
      goto err;
    }
  }
  else
  {
    res_ptr = tmp;
    res_len = stat_info.st_size;
  }
unknown's avatar
unknown committed
737

738 739
  res= (memcmp(res_ptr, ds->str, res_len)) ?
    RESULT_CONTENT_MISMATCH : RESULT_OK;
unknown's avatar
unknown committed
740

741 742 743 744
err:
  if (res && eval_result)
    str_to_file(fn_format(eval_file, fname, "", ".eval",2), res_ptr,
		res_len);
unknown's avatar
unknown committed
745

746 747
  my_free((gptr) tmp, MYF(0));
  my_close(fd, MYF(MY_WME));
748
  dynstr_free(&res_ds);
unknown's avatar
unknown committed
749

750
  DBUG_RETURN(res);
unknown's avatar
unknown committed
751 752
}

753 754
/*
  Check the content of ds against content of file fname
unknown's avatar
unknown committed
755

756 757 758 759
  SYNOPSIS
  check_result
  ds - content to be checked
  fname - name of file to check against
unknown's avatar
unknown committed
760
  require_option - if set and check fails, the test will be aborted with the special
761
                   exit code "not supported test"
unknown's avatar
unknown committed
762

763 764 765 766 767
  RETURN VALUES
   error - the function will not return

*/
static void check_result(DYNAMIC_STRING* ds, const char *fname,
768
			my_bool require_option)
unknown's avatar
unknown committed
769
{
770
  int res= dyn_string_cmp(ds, fname);
771
  DBUG_ENTER("check_result");
772 773 774

  if (res && require_option)
    abort_not_supported_test();
775
  switch (res) {
776
  case RESULT_OK:
777
    break; /* ok */
778
  case RESULT_LENGTH_MISMATCH:
779 780
    dump_result_to_reject_file(fname, ds->str, ds->length);
    die("Result length mismatch");
781
    break;
782
  case RESULT_CONTENT_MISMATCH:
783 784
    dump_result_to_reject_file(fname, ds->str, ds->length);
    die("Result content mismatch");
785 786 787 788
    break;
  default: /* impossible */
    die("Unknown error code from dyn_string_cmp()");
  }
789 790

  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
791 792
}

793

794
VAR* var_get(const char *var_name, const char** var_name_end, my_bool raw,
795
	     my_bool ignore_not_existing)
796 797 798
{
  int digit;
  VAR* v;
799 800 801 802
  DBUG_ENTER("var_get");
  DBUG_PRINT("enter",("var_name: %s",var_name));

  if (*var_name != '$')
803
    goto err;
804
  digit = *++var_name - '0';
unknown's avatar
cleanup  
unknown committed
805
  if (digit < 0 || digit >= 10)
806
  {
807
    const char *save_var_name = var_name, *end;
808
    uint length;
unknown's avatar
unknown committed
809
    end = (var_name_end) ? *var_name_end : 0;
810
    while (my_isvar(charset_info,*var_name) && var_name != end)
811
      var_name++;
812 813 814 815
    if (var_name == save_var_name)
    {
      if (ignore_not_existing)
	DBUG_RETURN(0);
unknown's avatar
unknown committed
816
      die("Empty variable");
817
    }
818
    length= (uint) (var_name - save_var_name);
819 820
    if (length >= MAX_VAR_NAME)
      die("Too long variable name: %s", save_var_name);
unknown's avatar
unknown committed
821

822
    if (!(v = (VAR*) hash_search(&var_hash, save_var_name, length)))
823
    {
824
      char buff[MAX_VAR_NAME+1];
unknown's avatar
cleanup  
unknown committed
825
      strmake(buff, save_var_name, length);
826
      v= var_from_env(buff, "");
827
    }
828
    var_name--;					/* Point at last character */
829
  }
830
  else
831
    v = var_reg + digit;
unknown's avatar
unknown committed
832

833 834 835 836
  if (!raw && v->int_dirty)
  {
    sprintf(v->str_val, "%d", v->int_val);
    v->int_dirty = 0;
837
    v->str_val_len = strlen(v->str_val);
838
  }
839
  if (var_name_end)
840
    *var_name_end = var_name  ;
841
  DBUG_RETURN(v);
842 843
err:
  if (var_name_end)
844 845
    *var_name_end = 0;
  die("Unsupported variable name: %s", var_name);
846
  DBUG_RETURN(0);
847 848
}

849
static VAR *var_obtain(const char *name, int len)
unknown's avatar
unknown committed
850 851
{
  VAR* v;
852
  if ((v = (VAR*)hash_search(&var_hash, name, len)))
unknown's avatar
unknown committed
853
    return v;
854
  v = var_init(0, name, len, "", 0);
unknown's avatar
SCRUM  
unknown committed
855
  my_hash_insert(&var_hash, (byte*)v);
unknown's avatar
unknown committed
856 857 858
  return v;
}

unknown's avatar
unknown committed
859 860
int var_set(const char *var_name, const char *var_name_end,
            const char *var_val, const char *var_val_end)
861 862 863
{
  int digit;
  VAR* v;
unknown's avatar
unknown committed
864 865 866 867 868 869
  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)));

870
  if (*var_name++ != '$')
unknown's avatar
unknown committed
871 872 873 874
  {
    var_name--;
    die("Variable name in %s does not start with '$'", var_name);
  }
875
  digit = *var_name - '0';
876
  if (!(digit < 10 && digit >= 0))
unknown's avatar
unknown committed
877 878 879
  {
    v = var_obtain(var_name, (uint) (var_name_end - var_name));
  }
880
  else
unknown's avatar
unknown committed
881
    v = var_reg + digit;
882
  DBUG_RETURN(eval_expr(v, var_val, (const char**)&var_val_end));
883 884
}

unknown's avatar
unknown committed
885

886
int open_file(const char *name)
unknown's avatar
unknown committed
887
{
888
  char buff[FN_REFLEN];
889 890
  DBUG_ENTER("open_file");
  DBUG_PRINT("enter", ("name: %s", name));
891 892 893 894 895 896 897
  if (!test_if_hard_path(name))
  {
    strxmov(buff, opt_basedir, name, NullS);
    name=buff;
  }
  fn_format(buff,name,"","",4);

898
  if (cur_file == file_stack_end)
unknown's avatar
unknown committed
899
    die("Source directives are nesting too deep");
900
  cur_file++;
901 902 903
  if (!(cur_file->file = my_fopen(buff, O_RDONLY | FILE_BINARY, MYF(0))))
  {
    cur_file--;
904
    die("Could not open file %s", buff);
905 906
  }
  cur_file->file_name= my_strdup(buff, MYF(MY_FAE));
907
  cur_file->lineno=1;
908
  DBUG_RETURN(0);
909
}
910

911 912 913 914 915 916 917 918

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

int check_eol_junk(const char *eol)
{
919
  const char *p= eol;
920 921
  DBUG_ENTER("check_eol_junk");
  DBUG_PRINT("enter", ("eol: %s", eol));
922
  /* Remove all spacing chars except new line */
923
  while (*p && my_isspace(charset_info, *p) && (*p != '\n'))
924 925 926
    p++;

  /* Check for extra delimiter */
927
  if (*p && !strncmp(p, delimiter, delimiter_length))
928
    die("Extra delimiter \"%s\" found", delimiter);
929

930 931 932 933 934 935 936
  /* Allow trailing # comment */
  if (*p && *p != '#')
  {
    if (*p == '\n')
      die("Missing delimiter");
    die("End of line junk detected: \"%s\"", p);
  }
937
  DBUG_RETURN(0);
unknown's avatar
unknown committed
938 939
}

unknown's avatar
unknown committed
940

941
/* ugly long name, but we are following the convention */
942
int do_wait_for_slave_to_stop(struct st_query *q __attribute__((unused)))
943 944 945 946
{
  MYSQL* mysql = &cur_con->mysql;
  for (;;)
  {
unknown's avatar
unknown committed
947
    MYSQL_RES *res;
948 949 950
    MYSQL_ROW row;
    int done;
    LINT_INIT(res);
951

unknown's avatar
unknown committed
952 953
    if (mysql_query(mysql,"show status like 'Slave_running'") ||
	!(res=mysql_store_result(mysql)))
954 955 956 957 958 959 960 961 962 963 964
      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
965
    my_sleep(SLAVE_POLL_INTERVAL);
966 967 968 969
  }
  return 0;
}

970
int do_require_manager(struct st_query *query __attribute__((unused)) )
971 972 973 974 975 976
{
  if (!manager)
    abort_not_supported_test();
  return 0;
}

977
#ifndef EMBEDDED_LIBRARY
978
static int do_server_op(struct st_query *q, const char *op)
979
{
980 981
  char *p= q->first_argument;
  char com_buf[256], *com_p;
982 983 984 985
  if (!manager)
  {
    die("Manager is not initialized, manager commands are not possible");
  }
986 987
  com_p= strmov(com_buf,op);
  com_p= strmov(com_p,"_exec ");
988
  if (!*p)
989 990
    die("Missing server name in server_%s", op);
  while (*p && !my_isspace(charset_info, *p))
unknown's avatar
unknown committed
991
   *com_p++= *p++;
992 993 994 995 996 997
  *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);
998 999
  while (!manager->eof)
  {
1000
    if (mysql_manager_fetch_line(manager, com_buf, sizeof(com_buf)))
1001 1002 1003 1004
      die("Error fetching result line: %s(%d)", manager->last_error,
	  manager->last_errno);
  }

1005
  q->last_argument= p;
1006 1007
  return 0;
}
1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018

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");
}

1019
#endif
1020

1021 1022 1023 1024 1025 1026

/*
  Source and execute the given file

  SYNOPSIS
    do_source()
1027
    query	called command
1028 1029 1030 1031 1032 1033 1034 1035

  DESCRIPTION
    source <file_name>

    Open the file <file_name> and execute it

*/

1036
int do_source(struct st_query *query)
unknown's avatar
unknown committed
1037
{
1038
  char *p= query->first_argument, *name;
1039
  if (!*p)
1040
    die("Missing file name in source");
1041
  name= p;
1042
  while (*p && !my_isspace(charset_info,*p))
unknown's avatar
unknown committed
1043
    p++;
1044 1045
  if (*p)
    *p++= 0;
1046 1047 1048 1049 1050
  query->last_argument= p;
  /*
     If this file has already been sourced, dont source it again.
     It's already available in the q_lines cache
  */
1051 1052
  if (parser.current_line < (parser.read_lines - 1))
    return 0;
unknown's avatar
unknown committed
1053 1054 1055
  return open_file(name);
}

1056

1057 1058 1059 1060 1061
/*
  Execute given command.

  SYNOPSIS
    do_exec()
1062
    query	called command
1063 1064

  DESCRIPTION
1065 1066 1067 1068 1069 1070
    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.
1071

1072 1073 1074 1075
  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.
1076 1077
*/

1078
static void do_exec(struct st_query *query)
1079
{
unknown's avatar
unknown committed
1080
  int error;
1081 1082
  char buf[1024];
  FILE *res_file;
1083
  char *cmd= query->first_argument;
1084
  DYNAMIC_STRING ds_cmd;
unknown's avatar
unknown committed
1085
  DBUG_ENTER("do_exec");
1086
  DBUG_PRINT("enter", ("cmd: '%s'", cmd));
1087

unknown's avatar
unknown committed
1088
  while (*cmd && my_isspace(charset_info, *cmd))
1089 1090
    cmd++;
  if (!*cmd)
1091
    die("Missing argument in exec");
1092
  query->last_argument= query->end;
1093

1094 1095
  init_dynamic_string(&ds_cmd, 0, strlen(cmd)+256, 256);
  /* Eval the command, thus replacing all environment variables */
1096
  do_eval(&ds_cmd, cmd, TRUE);
1097 1098
  cmd= ds_cmd.str;

unknown's avatar
unknown committed
1099
  DBUG_PRINT("info", ("Executing '%s' as '%s'",
1100
                      query->first_argument, cmd));
unknown's avatar
unknown committed
1101

1102
  if (!(res_file= popen(cmd, "r")) && query->abort_on_error)
1103
    die("popen(\"%s\", \"r\") failed", query->first_argument);
unknown's avatar
unknown committed
1104

1105
  while (fgets(buf, sizeof(buf), res_file))
1106
  {
1107
    if (disable_result_log)
1108 1109 1110 1111
    {
      buf[strlen(buf)-1]=0;
      DBUG_PRINT("exec_result",("%s", buf));
    }
1112 1113 1114 1115
    else
    {
      replace_dynstr_append(&ds_res, buf);
    }
unknown's avatar
unknown committed
1116 1117 1118
  }
  error= pclose(res_file);
  if (error != 0)
unknown's avatar
patch  
unknown committed
1119
  {
unknown's avatar
unknown committed
1120 1121 1122
    uint status= WEXITSTATUS(error), i;
    my_bool ok= 0;

1123
    if (query->abort_on_error)
1124
      die("command \"%s\" failed", query->first_argument);
unknown's avatar
unknown committed
1125 1126 1127

    DBUG_PRINT("info",
               ("error: %d, status: %d", error, status));
1128
    for (i= 0; i < query->expected_errors; i++)
unknown's avatar
patch  
unknown committed
1129
    {
1130 1131
      DBUG_PRINT("info",
                 ("error: %d, status: %d", error, status));
unknown's avatar
unknown committed
1132
      DBUG_PRINT("info", ("expected error: %d",
1133 1134 1135
                          query->expected_errno[i].code.errnum));
      if ((query->expected_errno[i].type == ERR_ERRNO) &&
          (query->expected_errno[i].code.errnum == status))
1136
      {
unknown's avatar
unknown committed
1137
        ok= 1;
1138
        DBUG_PRINT("info", ("command \"%s\" failed with expected error: %d",
1139
                            query->first_argument, status));
1140
      }
unknown's avatar
patch  
unknown committed
1141
    }
unknown's avatar
unknown committed
1142
    if (!ok)
1143
      die("command \"%s\" failed with wrong error: %d",
1144
          query->first_argument, status);
unknown's avatar
patch  
unknown committed
1145
  }
1146 1147
  else if (query->expected_errno[0].type == ERR_ERRNO &&
           query->expected_errno[0].code.errnum != 0)
unknown's avatar
patch  
unknown committed
1148 1149
  {
    /* Error code we wanted was != 0, i.e. not an expected success */
1150
    die("command \"%s\" succeeded - should have failed with errno %d...",
1151
        query->first_argument, query->expected_errno[0].code.errnum);
unknown's avatar
patch  
unknown committed
1152
  }
unknown's avatar
unknown committed
1153

1154
  free_replace();
1155
  DBUG_VOID_RETURN;
1156 1157
}

1158 1159 1160 1161 1162 1163 1164 1165
/*
  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
1166

1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177

  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
1178

1179 1180 1181
*/

int var_query_set(VAR* var, const char *query, const char** query_end)
1182
{
1183 1184
  char* end = (char*)((query_end && *query_end) ?
		      *query_end : query + strlen(query));
1185 1186 1187 1188
  MYSQL_RES *res;
  MYSQL_ROW row;
  MYSQL* mysql = &cur_con->mysql;
  LINT_INIT(res);
unknown's avatar
unknown committed
1189

1190
  while (end > query && *end != '`')
1191
    --end;
1192
  if (query == end)
1193
    die("Syntax error in query, missing '`'");
1194
  ++query;
1195

1196
  if (mysql_real_query(mysql, query, (int)(end - query)) ||
1197 1198 1199
      !(res = mysql_store_result(mysql)))
  {
    *end = 0;
1200 1201
    die("Error running query '%s': %d: %s", query,
	mysql_errno(mysql) ,mysql_error(mysql));
1202 1203 1204
  }

  if ((row = mysql_fetch_row(res)) && row[0])
1205 1206 1207 1208 1209 1210 1211 1212 1213
  {
    /*
      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;
1214
    MYSQL_FIELD *fields= mysql_fetch_fields(res);
1215 1216 1217 1218 1219 1220

    init_dynamic_string(&result, "", 16384, 65536);
    lengths= mysql_fetch_lengths(res);
    for (i=0; i < mysql_num_fields(res); i++)
    {
      if (row[0])
1221
      {
1222
#ifdef NOT_YET
1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
	/* 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]);
1236
#endif
1237
        /* Add column to tab separated string */
1238
	dynstr_append_mem(&result, row[i], lengths[i]);
1239
      }
1240 1241 1242
      dynstr_append_mem(&result, "\t", 1);
    }
    end= result.str + result.length-1;
1243
    eval_expr(var, result.str, (const char**) &end);
1244 1245
    dynstr_free(&result);
  }
1246
  else
1247
    eval_expr(var, "", 0);
1248 1249 1250 1251

  mysql_free_result(res);
  return 0;
}
1252

1253
void var_copy(VAR *dest, VAR *src)
unknown's avatar
unknown committed
1254
{
1255 1256 1257 1258
  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
1259
  if (dest->alloced_len < src->alloced_len &&
1260 1261 1262
      !(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
1263
    die("Out of memory");
1264 1265 1266 1267 1268 1269 1270
  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
1271 1272
}

1273
int eval_expr(VAR* v, const char *p, const char** p_end)
1274 1275
{
  VAR* vp;
1276
  if (*p == '$')
1277 1278
  {
    if ((vp = var_get(p,p_end,0,0)))
1279
    {
1280 1281
      var_copy(v, vp);
      return 0;
unknown's avatar
unknown committed
1282
    }
1283
  }
1284
  else if (*p == '`')
1285 1286 1287
  {
    return var_query_set(v, p, p_end);
  }
unknown's avatar
unknown committed
1288 1289
  else
    {
1290 1291 1292 1293
      int new_val_len = (p_end && *p_end) ?
	 (int) (*p_end - p) : (int) strlen(p);
      if (new_val_len + 1 >= v->alloced_len)
      {
1294
	v->alloced_len = (new_val_len < MIN_VAR_ALLOC - 1) ?
1295 1296
	  MIN_VAR_ALLOC : new_val_len + 1;
	if (!(v->str_val =
1297
	      v->str_val ? my_realloc(v->str_val, v->alloced_len+1,
1298
				      MYF(MY_WME)) :
1299
	      my_malloc(v->alloced_len+1, MYF(MY_WME))))
1300 1301 1302 1303 1304
	  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
1305 1306
      v->int_val=atoi(p);
      v->int_dirty=0;
1307 1308
      return 0;
    }
1309

1310 1311 1312 1313
  die("Invalid expr: %s", p);
  return 1;
}

1314

1315
enum enum_operator
1316
{
1317 1318 1319
  DO_DEC,
  DO_INC
};
1320

1321
/*
1322
  Decrease or increase the value of a variable
1323 1324

  SYNOPSIS
1325 1326 1327 1328
    do_modify_var()
    query	called command
    name        human readable name of operator
    operator    operation to perform on the var
1329 1330 1331

  DESCRIPTION
    dec $var_name
1332
    inc $var_name
1333

1334 1335
*/

1336 1337
int do_modify_var(struct st_query *query, const char *name,
                  enum enum_operator operator)
1338
{
1339
  const char *p= query->first_argument;
1340
  VAR* v;
1341
  if (!*p)
1342
    die("Missing arguments to %s", name);
1343
  if (*p != '$')
1344 1345
    die("First argument to %s must be a variable (start with $)", name);
  v= var_get(p, &p, 1, 0);
1346
  switch (operator) {
1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357
  case DO_DEC:
    v->int_val--;
    break;
  case DO_INC:
    v->int_val++;
    break;
  default:
    die("Invalid operator to do_operator");
    break;
  }
  v->int_dirty= 1;
1358
  query->last_argument= (char*)++p;
1359 1360 1361
  return 0;
}

1362

1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
/*

  SYNOPSIS
  do_system
    command	called command

  DESCRIPTION
    system <command>

    Eval the query to expand any $variables in the command.
    Execute the command withe the "system" command.

  NOTE
   If mysqltest is executed from cygwin shell, the command will be
   executed in cygwin shell. Thus commands like "rm" etc can be used.
 */

int do_system(struct st_query *command)
unknown's avatar
unknown committed
1381
{
1382
  DYNAMIC_STRING ds_cmd;
1383

1384 1385 1386 1387
  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);
1388

1389
  /* Eval the system command, thus replacing all environment variables */
1390
  do_eval(&ds_cmd, command->first_argument, TRUE);
1391 1392 1393 1394

  DBUG_PRINT("info", ("running system command '%s' as '%s'",
                      command->first_argument, ds_cmd.str));
  if (system(ds_cmd.str))
unknown's avatar
unknown committed
1395
  {
1396 1397
    if (command->abort_on_error)
      die("system command '%s' failed", command->first_argument);
1398

1399 1400 1401 1402
    /* 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
1403
  }
1404 1405

  command->last_argument= command->end;
unknown's avatar
unknown committed
1406 1407
  return 0;
}
1408

1409 1410 1411

/*
  Print the content between echo and <delimiter> to result file.
1412 1413
  Evaluate all variables in the string before printing, allow
  for variable names to be escaped using \
1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425

  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

1426 1427 1428 1429 1430 1431
    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
1432 1433
*/

1434
int do_echo(struct st_query *command)
1435
{
1436
  DYNAMIC_STRING *ds, ds_echo;
1437

1438
  ds= &ds_res;
1439

1440
  init_dynamic_string(&ds_echo, "", 256, 256);
unknown's avatar
unknown committed
1441
  do_eval(&ds_echo, command->first_argument, FALSE);
1442
  dynstr_append_mem(ds, ds_echo.str, ds_echo.length);
1443
  dynstr_append_mem(ds, "\n", 1);
1444 1445
  dynstr_free(&ds_echo);
  command->last_argument= command->end;
1446 1447 1448
  return 0;
}

1449

1450
int do_sync_with_master2(long offset)
unknown's avatar
unknown committed
1451 1452 1453
{
  MYSQL_RES* res;
  MYSQL_ROW row;
1454
  MYSQL* mysql= &cur_con->mysql;
unknown's avatar
unknown committed
1455
  char query_buf[FN_REFLEN+128];
1456
  int tries= 0;
1457 1458
  int rpl_parse;

1459
  if (!master_pos.file[0])
unknown's avatar
unknown committed
1460
    die("Calling 'sync_with_master' without calling 'save_master_pos'");
1461
  rpl_parse= mysql_rpl_parse_enabled(mysql);
1462
  mysql_disable_rpl_parse(mysql);
unknown's avatar
unknown committed
1463

unknown's avatar
unknown committed
1464
  sprintf(query_buf, "select master_pos_wait('%s', %ld)", master_pos.file,
1465
	  master_pos.pos + offset);
1466 1467 1468

wait_for_position:

1469
  if (mysql_query(mysql, query_buf))
unknown's avatar
unknown committed
1470 1471
    die("failed in %s: %d: %s", query_buf, mysql_errno(mysql),
        mysql_error(mysql));
unknown's avatar
unknown committed
1472

1473
  if (!(res= mysql_store_result(mysql)))
unknown's avatar
unknown committed
1474
    die("mysql_store_result() returned NULL for '%s'", query_buf);
1475
  if (!(row= mysql_fetch_row(res)))
unknown's avatar
unknown committed
1476
    die("empty result in %s", query_buf);
1477
  if (!row[0])
1478 1479 1480 1481 1482 1483
  {
    /*
      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
1484
      die("could not sync with master ('%s' returned NULL)", query_buf);
1485 1486 1487 1488
    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
1489
  mysql_free_result(res);
1490
  if (rpl_parse)
1491
    mysql_enable_rpl_parse(mysql);
unknown's avatar
unknown committed
1492

unknown's avatar
unknown committed
1493 1494 1495
  return 0;
}

1496
int do_sync_with_master(struct st_query *query)
1497
{
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510
  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);
1511
}
1512

unknown's avatar
unknown committed
1513 1514 1515 1516 1517
int do_save_master_pos()
{
  MYSQL_RES* res;
  MYSQL_ROW row;
  MYSQL* mysql = &cur_con->mysql;
1518
  const char *query;
1519 1520 1521 1522
  int rpl_parse;

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

1524
  if (mysql_query(mysql, query= "show master status"))
1525
    die("failed in show master status: %d: %s",
unknown's avatar
unknown committed
1526 1527
	mysql_errno(mysql), mysql_error(mysql));

1528
  if (!(res = mysql_store_result(mysql)))
unknown's avatar
unknown committed
1529
    die("mysql_store_result() retuned NULL for '%s'", query);
1530
  if (!(row = mysql_fetch_row(res)))
unknown's avatar
unknown committed
1531
    die("empty result in show master status");
1532 1533
  strnmov(master_pos.file, row[0], sizeof(master_pos.file)-1);
  master_pos.pos = strtoul(row[1], (char**) 0, 10);
1534
  mysql_free_result(res);
unknown's avatar
unknown committed
1535

1536
  if (rpl_parse)
1537
    mysql_enable_rpl_parse(mysql);
unknown's avatar
unknown committed
1538

unknown's avatar
unknown committed
1539 1540 1541 1542
  return 0;
}


1543 1544 1545 1546 1547
/*
  Assign the variable <var_name> with <var_val>

  SYNOPSIS
   do_let()
1548
    query	called command
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561

  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
*/

1562
int do_let(struct st_query *query)
1563
{
1564
  char *p= query->first_argument;
1565
  char *var_name, *var_name_end, *var_val_start;
1566 1567

  /* Find <var_name> */
1568
  if (!*p)
1569
    die("Missing arguments to let");
1570 1571
  var_name= p;
  while (*p && (*p != '=') && !my_isspace(charset_info,*p))
1572
    p++;
1573 1574
  var_name_end= p;
  if (var_name+1 == var_name_end)
1575
    die("Missing variable name in let");
1576
  while (my_isspace(charset_info,*p))
1577
    p++;
1578
  if (*p++ != '=')
1579 1580 1581
    die("Missing assignment operator in let");

  /* Find start of <var_val> */
1582
  while (*p && my_isspace(charset_info,*p))
1583
    p++;
1584 1585
  var_val_start= p;
  query->last_argument= query->end;
1586
  /* Assign var_val to var_name */
1587
  return var_set(var_name, var_name_end, var_val_start, query->end);
1588 1589
}

unknown's avatar
unknown committed
1590 1591 1592 1593 1594 1595 1596 1597

/*
  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)
1598
{
unknown's avatar
unknown committed
1599 1600 1601 1602
  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);
1603 1604
}

unknown's avatar
unknown committed
1605

1606
int do_rpl_probe(struct st_query *query __attribute__((unused)))
1607
{
unknown's avatar
unknown committed
1608
  DBUG_ENTER("do_rpl_probe");
1609
  if (mysql_rpl_probe(&cur_con->mysql))
unknown's avatar
unknown committed
1610 1611
    die("Failed in mysql_rpl_probe(): '%s'", mysql_error(&cur_con->mysql));
  DBUG_RETURN(0);
1612 1613
}

unknown's avatar
unknown committed
1614

1615
int do_enable_rpl_parse(struct st_query *query __attribute__((unused)))
1616 1617 1618 1619 1620
{
  mysql_enable_rpl_parse(&cur_con->mysql);
  return 0;
}

unknown's avatar
unknown committed
1621

1622
int do_disable_rpl_parse(struct st_query *query __attribute__((unused)))
1623 1624 1625 1626 1627 1628
{
  mysql_disable_rpl_parse(&cur_con->mysql);
  return 0;
}


1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642
/*
  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

  DESCRIPTION
    sleep <seconds>
    real_sleep

*/

1643
int do_sleep(struct st_query *query, my_bool real_sleep)
unknown's avatar
unknown committed
1644
{
1645 1646 1647 1648 1649 1650
  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
1651
    p++;
1652
  if (!*p)
1653
    die("Missing argument to sleep");
1654 1655 1656 1657 1658 1659 1660 1661 1662
  sleep_start= p;
  /* Check that arg starts with a digit, not handled by my_strtod */
  if (!my_isdigit(charset_info, *sleep_start))
    die("Invalid argument to sleep \"%s\"", query->first_argument);
  sleep_val= my_strtod(sleep_start, &sleep_end, &error);
  if (error)
    die("Invalid argument to sleep \"%s\"", query->first_argument);

  /* Fixed sleep time selected by --sleep option */
1663
  if (opt_sleep && !real_sleep)
1664 1665
    sleep_val= opt_sleep;

1666
  DBUG_PRINT("info", ("sleep_val: %f", sleep_val));
1667 1668
  my_sleep((ulong) (sleep_val * 1000000L));
  query->last_argument= sleep_end;
unknown's avatar
unknown committed
1669
  return 0;
unknown's avatar
unknown committed
1670 1671
}

1672
static void get_file_name(char *filename, struct st_query *q)
1673
{
1674
  char *p= q->first_argument, *name;
1675 1676
  if (!*p)
    die("Missing file name argument");
1677
  name= p;
1678 1679 1680 1681
  while (*p && !my_isspace(charset_info,*p))
    p++;
  if (*p)
    *p++= 0;
1682
  q->last_argument= p;
1683
  strmake(filename, name, FN_REFLEN);
1684 1685
}

1686
static void set_charset(struct st_query *q)
unknown's avatar
unknown committed
1687
{
1688 1689
  char *charset_name= q->first_argument;
  char *p;
unknown's avatar
unknown committed
1690 1691

  if (!charset_name || !*charset_name)
1692
    die("Missing charset name in 'character_set'");
unknown's avatar
unknown committed
1693
  /* Remove end space */
1694 1695 1696 1697 1698
  p= charset_name;
  while (*p && !my_isspace(charset_info,*p))
    p++;
  if(*p)
    *p++= 0;
1699
  q->last_argument= p;
unknown's avatar
unknown committed
1700 1701 1702 1703
  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
1704

1705
static uint get_errcodes(match_err *to,struct st_query *q)
unknown's avatar
unknown committed
1706
{
1707
  char *p= q->first_argument;
1708
  uint count= 0;
1709

1710
  DBUG_ENTER("get_errcodes");
unknown's avatar
unknown committed
1711

unknown's avatar
unknown committed
1712
  if (!*p)
1713
    die("Missing argument in %s", q->query);
unknown's avatar
unknown committed
1714

1715
  do
unknown's avatar
unknown committed
1716
  {
1717 1718 1719
    if (*p == 'S')
    {
      /* SQLSTATE string */
1720 1721 1722 1723 1724 1725 1726
      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;

1727 1728
      to[count].type= ERR_SQLSTATE;
    }
1729 1730 1731 1732 1733
    else if (*p == 'E')
    {
      /* SQL error as string */
      st_error *e= global_error;
      char *start= p++;
unknown's avatar
unknown committed
1734

1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746
      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)
1747
	die("Unknown SQL error '%s'", start);
1748
    }
1749 1750 1751
    else
    {
      long val;
1752 1753

      if (!(p= str2int(p,10,(long) INT_MIN, (long) INT_MAX, &val)))
1754
	die("Invalid argument in %s", q->query);
1755 1756 1757
      to[count].code.errnum= (uint) val;
      to[count].type= ERR_ERRNO;
    }
unknown's avatar
unknown committed
1758
    count++;
1759
  } while (*(p++) == ',');
1760
  q->last_argument= (p - 1);
1761
  to[count].type= ERR_EMPTY;                        /* End of data */
unknown's avatar
unknown committed
1762
  DBUG_RETURN(count);
unknown's avatar
unknown committed
1763 1764
}

unknown's avatar
unknown committed
1765 1766 1767
/*
  Get a string;  Return ptr to end of string
  Strings may be surrounded by " or '
1768 1769

  If string is a '$variable', return the value of the variable.
unknown's avatar
unknown committed
1770 1771 1772
*/


1773
static char *get_string(char **to_ptr, char **from_ptr,
1774
			struct st_query *q)
unknown's avatar
unknown committed
1775 1776
{
  reg1 char c,sep;
1777
  char *to= *to_ptr, *from= *from_ptr, *start=to;
unknown's avatar
unknown committed
1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821
  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)
1822
    die("Wrong string argument in %s", q->query);
unknown's avatar
unknown committed
1823

1824
  while (my_isspace(charset_info,*from))	/* Point to next string */
unknown's avatar
unknown committed
1825 1826
    from++;

1827 1828
  *to =0;				/* End of string marker */
  *to_ptr= to+1;			/* Store pointer to end */
unknown's avatar
unknown committed
1829
  *from_ptr= from;
1830 1831 1832 1833 1834 1835 1836 1837

  /* 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)
    {
1838
      DBUG_PRINT("info",("var: '%s' -> '%s'", start, var->str_val));
1839 1840 1841 1842
      DBUG_RETURN(var->str_val);	/* return found variable value */
    }
  }
  DBUG_RETURN(start);
unknown's avatar
unknown committed
1843 1844 1845 1846 1847 1848 1849
}


/*
  Get arguments for replace. The syntax is:
  replace from to [from to ...]
  Where each argument may be quoted with ' or "
1850 1851
  A argument may also be a variable, in which case the value of the
  variable is replaced.
unknown's avatar
unknown committed
1852 1853 1854 1855 1856
*/

static void get_replace(struct st_query *q)
{
  uint i;
1857
  char *from= q->first_argument;
1858
  char *buff,*start;
unknown's avatar
unknown committed
1859 1860 1861 1862
  char word_end_chars[256],*pos;
  POINTER_ARRAY to_array,from_array;
  DBUG_ENTER("get_replace");

unknown's avatar
unknown committed
1863
  free_replace();
1864

unknown's avatar
unknown committed
1865 1866 1867
  bzero((char*) &to_array,sizeof(to_array));
  bzero((char*) &from_array,sizeof(from_array));
  if (!*from)
1868
    die("Missing argument in %s", q->query);
1869
  start=buff=my_malloc(strlen(from)+1,MYF(MY_WME | MY_FAE));
unknown's avatar
unknown committed
1870 1871 1872
  while (*from)
  {
    char *to=buff;
1873
    to=get_string(&buff, &from, q);
unknown's avatar
unknown committed
1874
    if (!*from)
1875
      die("Wrong number of arguments to replace_result in '%s'", q->query);
unknown's avatar
unknown committed
1876
    insert_pointer_name(&from_array,to);
1877
    to=get_string(&buff, &from, q);
unknown's avatar
unknown committed
1878 1879 1880
    insert_pointer_name(&to_array,to);
  }
  for (i=1,pos=word_end_chars ; i < 256 ; i++)
1881
    if (my_isspace(charset_info,i))
unknown's avatar
unknown committed
1882
      *pos++= i;
1883
  *pos=0;					/* End pointer */
unknown's avatar
unknown committed
1884 1885 1886
  if (!(glob_replace=init_replace((char**) from_array.typelib.type_names,
				  (char**) to_array.typelib.type_names,
				  (uint) from_array.typelib.count,
1887
				  word_end_chars)))
1888
    die("Can't initialize replace from '%s'", q->query);
unknown's avatar
unknown committed
1889 1890
  free_pointer_array(&from_array);
  free_pointer_array(&to_array);
1891
  my_free(start, MYF(0));
1892
  q->last_argument= q->end;
1893
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
1894 1895 1896 1897
}

void free_replace()
{
1898
  DBUG_ENTER("free_replace");
unknown's avatar
unknown committed
1899 1900 1901 1902 1903
  if (glob_replace)
  {
    my_free((char*) glob_replace,MYF(0));
    glob_replace=0;
  }
1904
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
1905 1906
}

unknown's avatar
unknown committed
1907
struct connection * find_connection_by_name(const char *name)
unknown's avatar
unknown committed
1908 1909
{
  struct connection *con;
1910
  for (con= cons; con < next_con; con++)
1911
  {
1912
    if (!strcmp(con->name, name))
1913
    {
unknown's avatar
unknown committed
1914
      return con;
1915 1916
    }
  }
unknown's avatar
unknown committed
1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928
  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
1929 1930
}

1931 1932

int select_connection(struct st_query *query)
1933
{
1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950
  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)
1951
{
1952
  char *p= q->first_argument, *name;
1953
  struct connection *con;
1954 1955 1956
  DBUG_ENTER("close_connection");
  DBUG_PRINT("enter",("name: '%s'",p));

1957
  if (!*p)
unknown's avatar
unknown committed
1958
    die("Missing connection name in disconnect");
1959
  name= p;
1960
  while (*p && !my_isspace(charset_info,*p))
1961
    p++;
1962

1963 1964
  if (*p)
    *p++= 0;
1965 1966
  q->last_argument= p;
  for (con= cons; con < next_con; con++)
1967
  {
1968
    if (!strcmp(con->name, name))
1969
    {
1970
#ifndef EMBEDDED_LIBRARY
unknown's avatar
unknown committed
1971 1972 1973
      if (q->type == Q_DIRTY_CLOSE)
      {
	if (con->mysql.net.vio)
unknown's avatar
unknown committed
1974
	{
unknown's avatar
unknown committed
1975 1976
	  vio_delete(con->mysql.net.vio);
	  con->mysql.net.vio = 0;
unknown's avatar
unknown committed
1977
	}
unknown's avatar
unknown committed
1978
      }
1979
#endif
1980
      mysql_close(&con->mysql);
1981 1982 1983
      if (con->util_mysql)
	mysql_close(con->util_mysql);
      con->util_mysql= 0;
unknown's avatar
unknown committed
1984 1985 1986 1987 1988 1989 1990 1991
      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");
1992 1993 1994
      DBUG_RETURN(0);
    }
  }
1995
  die("connection '%s' not found in connection pool", name);
1996
  DBUG_RETURN(1);				/* Never reached */
1997 1998
}

unknown's avatar
unknown committed
1999

2000 2001
/*
   This one now is a hack - we may want to improve in in the
unknown's avatar
unknown committed
2002 2003 2004
   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
2005 2006 2007 2008

  SYNOPSIS
  safe_get_param
  str - string to get param from
unknown's avatar
unknown committed
2009
  arg - pointer to string where result will be stored
unknown's avatar
unknown committed
2010 2011
  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
2012

unknown's avatar
unknown committed
2013 2014
  RETURNS
  pointer to str after param
unknown's avatar
unknown committed
2015

2016
*/
2017

unknown's avatar
unknown committed
2018
char* safe_get_param(char *str, char** arg, const char *msg)
unknown's avatar
unknown committed
2019
{
2020
  DBUG_ENTER("safe_get_param");
unknown's avatar
unknown committed
2021 2022
  if(!*str)
  {
unknown's avatar
unknown committed
2023
    if (msg)
unknown's avatar
unknown committed
2024 2025 2026 2027
      die(msg);
    *arg= str;
    DBUG_RETURN(str);
  }
2028
  while (*str && my_isspace(charset_info,*str))
unknown's avatar
unknown committed
2029
    str++;
2030
  *arg= str;
unknown's avatar
unknown committed
2031 2032
  while (*str && *str != ',' && *str != ')')
    str++;
unknown's avatar
unknown committed
2033
  if (msg && !*arg)
unknown's avatar
unknown committed
2034
    die(msg);
2035

2036
  *str++= 0;
2037
  DBUG_RETURN(str);
unknown's avatar
unknown committed
2038 2039
}

2040
#ifndef EMBEDDED_LIBRARY
2041 2042 2043 2044 2045 2046 2047 2048 2049 2050
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);

}
2051
#endif
2052

2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075

/*
  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
*/

2076 2077
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
2078
{
2079 2080
  int con_error= 1;
  my_bool reconnect= 1;
unknown's avatar
unknown committed
2081
  int i;
2082
  for (i= 0; i < MAX_CON_TRIES; ++i)
unknown's avatar
unknown committed
2083
  {
2084
    if (mysql_real_connect(mysql, host,user, pass, db, port, sock,
2085
			   CLIENT_MULTI_STATEMENTS | CLIENT_REMEMBER_OPTIONS))
unknown's avatar
unknown committed
2086
    {
2087
      con_error= 0;
unknown's avatar
unknown committed
2088 2089 2090 2091
      break;
    }
    sleep(CON_RETRY_SLEEP);
  }
2092 2093 2094 2095
  /*
   TODO: change this to 0 in future versions, but the 'kill' test relies on
   existing behavior
  */
2096
  mysql_options(mysql, MYSQL_OPT_RECONNECT, (char *)&reconnect);
unknown's avatar
unknown committed
2097 2098 2099
  return con_error;
}

2100

2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128
/*
  Connect to a server and handle connection errors in case when they occur.

  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)
{
2129
  DYNAMIC_STRING *ds;
2130
  my_bool reconnect= 1;
2131 2132
  int error= 0;

2133
  ds= &ds_res;
2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165

  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))
  {
2166 2167
    handle_error("connect", q, mysql_errno(con), mysql_error(con),
		 mysql_sqlstate(con), ds);
2168 2169 2170
    *create_conn= 0;
    goto err;
  }
2171

2172
  handle_no_error(q);
2173

2174 2175 2176 2177
  /*
   TODO: change this to 0 in future versions, but the 'kill' test relies on
   existing behavior
  */
2178
  mysql_options(con, MYSQL_OPT_RECONNECT, (char *)&reconnect);
2179

2180 2181 2182 2183 2184 2185
err:
  free_replace();
  return error;
}


unknown's avatar
unknown committed
2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209
/*
  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

 */

2210
int do_connect(struct st_query *q)
unknown's avatar
unknown committed
2211
{
2212
  char *con_name, *con_user,*con_pass, *con_host, *con_port_str,
unknown's avatar
unknown committed
2213 2214
    *con_db, *con_sock, *con_options;
  char *con_buf, *p;
2215
  char buff[FN_REFLEN];
unknown's avatar
unknown committed
2216
  int con_port;
unknown's avatar
unknown committed
2217 2218
  bool con_ssl= 0;
  bool con_compress= 0;
2219
  int free_con_sock= 0;
2220 2221
  int error= 0;
  int create_conn= 1;
2222
  VAR *var_port, *var_sock;
unknown's avatar
unknown committed
2223

2224
  DBUG_ENTER("do_connect");
2225
  DBUG_PRINT("enter",("connect: %s", q->first_argument));
unknown's avatar
unknown committed
2226

unknown's avatar
unknown committed
2227 2228 2229 2230 2231
  /* 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;

2232
  if (*p != '(')
2233
    die("Syntax error in connect - expected '(' found '%c'", *p);
unknown's avatar
unknown committed
2234
  p++;
unknown's avatar
unknown committed
2235 2236 2237 2238 2239
  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
2240 2241

  /* Port */
unknown's avatar
unknown committed
2242
  p= safe_get_param(p, &con_port_str, 0);
unknown's avatar
unknown committed
2243
  if (*con_port_str)
unknown's avatar
unknown committed
2244
  {
2245 2246
    if (*con_port_str == '$')
    {
2247
      if (!(var_port= var_get(con_port_str, 0, 0, 0)))
unknown's avatar
unknown committed
2248
        die("Unknown variable '%s'", con_port_str+1);
2249
      con_port= var_port->int_val;
2250 2251
    }
    else
unknown's avatar
unknown committed
2252
    {
2253
      con_port= atoi(con_port_str);
unknown's avatar
unknown committed
2254 2255 2256 2257 2258 2259 2260 2261 2262 2263
      if (con_port == 0)
        die("Illegal argument for port: '%s'", con_port_str);
    }
  }
  else
  {
    con_port= port;
  }

  /* Sock */
unknown's avatar
unknown committed
2264
  p= safe_get_param(p, &con_sock, 0);
unknown's avatar
unknown committed
2265 2266
  if (*con_sock)
  {
2267 2268
    if (*con_sock == '$')
    {
2269
      if (!(var_sock= var_get(con_sock, 0, 0, 0)))
unknown's avatar
unknown committed
2270
        die("Unknown variable '%s'", con_sock+1);
2271
      if (!(con_sock= (char*)my_malloc(var_sock->str_val_len+1, MYF(0))))
unknown's avatar
unknown committed
2272
        die("Out of memory");
2273
      free_con_sock= 1;
2274
      memcpy(con_sock, var_sock->str_val, var_sock->str_val_len);
2275
      con_sock[var_sock->str_val_len]= 0;
2276
    }
unknown's avatar
unknown committed
2277
  }
unknown's avatar
unknown committed
2278 2279 2280 2281 2282 2283
  else
  {
    con_sock= (char*) unix_sock;
  }

  /* Options */
unknown's avatar
unknown committed
2284
  p= safe_get_param(p, &con_options, 0);
unknown's avatar
unknown committed
2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298
  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;
  }
2299 2300
  /* Note: 'p' is pointing into the copy 'con_buf' */
  q->last_argument= q->first_argument + (p - con_buf);
unknown's avatar
unknown committed
2301

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

unknown's avatar
unknown committed
2305 2306 2307
  if (find_connection_by_name(con_name))
    die("Connection %s already exists", con_name);

2308
  if (!mysql_init(&next_con->mysql))
unknown's avatar
unknown committed
2309
    die("Failed on mysql_init()");
unknown's avatar
unknown committed
2310
  if (opt_compress || con_compress)
2311
    mysql_options(&next_con->mysql, MYSQL_OPT_COMPRESS, NullS);
2312
  mysql_options(&next_con->mysql, MYSQL_OPT_LOCAL_INFILE, 0);
unknown's avatar
unknown committed
2313
  mysql_options(&next_con->mysql, MYSQL_SET_CHARSET_NAME, charset_name);
2314

unknown's avatar
unknown committed
2315
#ifdef HAVE_OPENSSL
unknown's avatar
unknown committed
2316
  if (opt_use_ssl || con_ssl)
unknown's avatar
unknown committed
2317 2318 2319
    mysql_ssl_set(&next_con->mysql, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
		  opt_ssl_capath, opt_ssl_cipher);
#endif
2320
  if (con_sock && !free_con_sock && *con_sock && *con_sock != FN_LIBCHAR)
2321
    con_sock=fn_format(buff, con_sock, TMPDIR, "",0);
unknown's avatar
unknown committed
2322
  if (!con_db[0])
2323
    con_db= db;
2324
  /* Special database to allow one to connect without a database name */
unknown's avatar
unknown committed
2325
  if (con_db && !strcmp(con_db,"*NO-ONE*"))
2326
    con_db= 0;
2327 2328
  if (q->abort_on_error)
  {
2329 2330 2331 2332
    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));
2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344
  }
  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++;
  }
2345 2346
  if (free_con_sock)
    my_free(con_sock, MYF(MY_WME));
unknown's avatar
unknown committed
2347
  my_free(con_buf, MYF(MY_WME));
2348
  DBUG_RETURN(error);
unknown's avatar
unknown committed
2349 2350
}

2351

2352
int do_done(struct st_query *q)
2353
{
unknown's avatar
unknown committed
2354
  /* Check if empty block stack */
2355
  if (cur_block == block_stack)
2356 2357 2358
  {
    if (*q->query != '}')
      die("Stray 'end' command - end of block before beginning");
2359
    die("Stray '}' - end of block before beginning");
2360
  }
unknown's avatar
unknown committed
2361 2362 2363

  /* Test if inner block has been executed */
  if (cur_block->ok && cur_block->cmd == cmd_while)
2364
  {
unknown's avatar
unknown committed
2365 2366 2367
    /* Pop block from stack, re-execute outer block */
    cur_block--;
    parser.current_line = cur_block->line;
2368
  }
2369
  else
unknown's avatar
unknown committed
2370
  {
unknown's avatar
unknown committed
2371 2372 2373
    /* Pop block from stack, goto next line */
    cur_block--;
    parser.current_line++;
unknown's avatar
unknown committed
2374
  }
2375 2376 2377
  return 0;
}

unknown's avatar
unknown committed
2378

2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404
/*
  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.

 */

unknown's avatar
unknown committed
2405
int do_block(enum block_cmd cmd, struct st_query* q)
2406
{
2407 2408
  char *p= q->first_argument;
  const char *expr_start, *expr_end;
2409
  VAR v;
2410
  const char *cmd_name= (cmd == cmd_while ? "while" : "if");
2411 2412 2413
  my_bool not_expr= FALSE;
  DBUG_ENTER("do_block");
  DBUG_PRINT("enter", ("%s", cmd_name));
unknown's avatar
unknown committed
2414 2415

  /* Check stack overflow */
2416
  if (cur_block == block_stack_end)
2417
    die("Nesting too deeply");
unknown's avatar
unknown committed
2418 2419 2420 2421 2422 2423

  /* 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)
2424
  {
unknown's avatar
unknown committed
2425 2426 2427 2428
    /* Inner block should be ignored too */
    cur_block++;
    cur_block->cmd= cmd;
    cur_block->ok= FALSE;
2429 2430
    return 0;
  }
unknown's avatar
unknown committed
2431

unknown's avatar
unknown committed
2432
  /* Parse and evaluate test expression */
2433
  expr_start= strchr(p, '(');
2434
  if (!expr_start++)
2435
    die("missing '(' in %s", cmd_name);
2436 2437 2438 2439 2440 2441 2442 2443

  /* Check for !<expr> */
  if (*expr_start == '!')
  {
    not_expr= TRUE;
    expr_start++; /* Step past the '!' */
  }
  /* Find ending ')' */
2444
  expr_end= strrchr(expr_start, ')');
2445
  if (!expr_end)
2446
    die("missing ')' in %s", cmd_name);
2447 2448 2449 2450
  p= (char*)expr_end+1;

  while (*p && my_isspace(charset_info, *p))
    p++;
2451 2452
  if (*p == '{')
    die("Missing newline between %s and '{'", cmd_name);
2453
  if (*p)
2454
    die("Missing '{' after %s. Found \"%s\"", cmd_name, p);
2455

unknown's avatar
unknown committed
2456
  var_init(&v,0,0,0,0);
2457
  eval_expr(&v, expr_start, &expr_end);
unknown's avatar
unknown committed
2458 2459 2460 2461 2462 2463

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

2464 2465 2466 2467 2468
  if (not_expr)
    cur_block->ok = !cur_block->ok;

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

unknown's avatar
unknown committed
2469
  var_free(&v);
2470
  DBUG_VOID_RETURN;
2471 2472
}

unknown's avatar
unknown committed
2473

2474 2475 2476 2477 2478 2479 2480 2481 2482 2483
/*
  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.
*/

2484
int my_getc(FILE *file)
2485 2486 2487
{
  if (line_buffer_pos == line_buffer)
    return fgetc(file);
2488
  return *--line_buffer_pos;
2489 2490 2491 2492
}

void my_ungetc(int c)
{
2493
  *line_buffer_pos++= (char) c;
2494 2495 2496 2497
}


my_bool end_of_query(int c)
2498
{
2499
  uint i;
2500
  char tmp[MAX_DELIMITER];
2501

2502 2503 2504 2505
  if (c != *delimiter)
    return 0;

  for (i= 1; i < delimiter_length &&
2506
	 (c= my_getc(cur_file->file)) == *(delimiter + i);
2507
       i++)
2508 2509 2510
    tmp[i]= c;

  if (i == delimiter_length)
2511 2512 2513
    return 1;					/* Found delimiter */

  /* didn't find delimiter, push back things that we read */
2514 2515 2516
  my_ungetc(c);
  while (i > 1)
    my_ungetc(tmp[--i]);
2517 2518 2519 2520
  return 0;
}


2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544
/*
  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
    This function actually reads several lines an adds them to the
    buffer buf. It will continue to read until it finds what it believes
    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
    found inside strings sorrounded with '"' and '\'' escaped strings.

    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.

*/

2545
int read_line(char *buf, int size)
unknown's avatar
unknown committed
2546 2547
{
  int c;
2548
  char quote;
2549
  char *p= buf, *buf_end= buf + size - 1;
2550
  int no_save= 0;
2551 2552
  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
2553
  DBUG_ENTER("read_line");
2554
  LINT_INIT(quote);
2555

2556
  start_lineno= cur_file->lineno;
2557 2558
  for (; p < buf_end ;)
  {
2559
    no_save= 0;
2560 2561
    c= my_getc(cur_file->file);
    if (feof(cur_file->file))
unknown's avatar
unknown committed
2562
    {
unknown's avatar
unknown committed
2563
  found_eof:
unknown's avatar
unknown committed
2564 2565
      if (cur_file->file != stdin)
      {
2566
	my_fclose(cur_file->file, MYF(0));
2567 2568
        cur_file->file= 0;
      }
2569 2570
      my_free((gptr)cur_file->file_name, MYF(MY_ALLOW_ZERO_PTR));
      cur_file->file_name= 0;
2571
      if (cur_file == file_stack)
2572
      {
2573 2574 2575 2576 2577
        /* We're back at the first file, check if
           all { have matching }
         */
        if (cur_block != block_stack)
          die("Missing end of block");
2578

2579
        DBUG_PRINT("info", ("end of file"));
unknown's avatar
unknown committed
2580
	DBUG_RETURN(1);
2581
      }
2582
      cur_file--;
2583
      start_lineno= cur_file->lineno;
unknown's avatar
unknown committed
2584
      continue;
2585
    }
2586

2587
    if (c == '\n')
2588 2589
    {
      /* Line counting is independent of state */
2590
      cur_file->lineno++;
2591

2592 2593 2594 2595 2596
      /* Convert cr/lf to lf */
      if (p != buf && *(p-1) == '\r')
        *(p-1)= 0;
    }

2597 2598
    switch(state) {
    case R_NORMAL:
2599
      /*  Only accept '{' in the beginning of a line */
2600 2601 2602
      if (end_of_query(c))
      {
	*p= 0;
unknown's avatar
unknown committed
2603
	DBUG_RETURN(0);
2604
      }
2605 2606 2607 2608 2609
      else if (c == '\'' || c == '"' || c == '`')
      {
        quote= c;
	state= R_Q;
      }
2610
      else if (c == '\n')
2611
      {
2612
	state = R_LINE_START;
2613
      }
2614 2615 2616 2617
      break;
    case R_COMMENT:
      if (c == '\n')
      {
2618
	*p= 0;
unknown's avatar
unknown committed
2619
	DBUG_RETURN(0);
2620 2621 2622
      }
      break;
    case R_LINE_START:
2623
      /* Only accept start of comment if this is the first line in query */
2624 2625
      if ((cur_file->lineno == start_lineno) &&
	  (c == '#' || c == '-' || parsing_disabled))
2626 2627 2628
      {
	state = R_COMMENT;
      }
2629
      else if (my_isspace(charset_info, c))
2630 2631
      {
	if (c == '\n')
2632
	  start_lineno= cur_file->lineno; /* Query hasn't started yet */
2633
	no_save= 1;
2634
      }
2635 2636
      else if (c == '}')
      {
2637 2638
	*buf++= '}';
	*buf= 0;
unknown's avatar
unknown committed
2639
	DBUG_RETURN(0);
2640
      }
2641
      else if (end_of_query(c) || c == '{')
2642
      {
2643
	*p= 0;
unknown's avatar
unknown committed
2644
	DBUG_RETURN(0);
2645
      }
2646
      else if (c == '\'' || c == '"' || c == '`')
2647
      {
2648 2649
        quote= c;
	state= R_Q;
2650
      }
2651
      else
2652
	state= R_NORMAL;
2653
      break;
unknown's avatar
unknown committed
2654

2655 2656 2657
    case R_Q:
      if (c == quote)
	state= R_Q_IN_Q;
2658
      else if (c == '\\')
2659
	state= R_SLASH_IN_Q;
2660
      break;
2661
    case R_Q_IN_Q:
2662 2663 2664
      if (end_of_query(c))
      {
	*p= 0;
unknown's avatar
unknown committed
2665
	DBUG_RETURN(0);
2666
      }
2667
      if (c != quote)
2668
	state= R_NORMAL;
2669
      else
2670
	state= R_Q;
2671
      break;
2672 2673
    case R_SLASH_IN_Q:
      state= R_Q;
2674
      break;
2675

unknown's avatar
unknown committed
2676
    }
2677 2678

    if (!no_save)
unknown's avatar
unknown committed
2679 2680 2681 2682 2683
    {
      /* 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
2684 2685
      /* We give up if multibyte character is started but not */
      /* completed before we pass buf_end */
unknown's avatar
unknown committed
2686 2687 2688 2689 2690 2691 2692 2693 2694
      if ((charlen > 1) && (p + charlen) <= buf_end)
      {
	int i;
	char* mb_start = p;

	*p++ = c;

	for (i= 1; i < charlen; i++)
	{
2695
	  if (feof(cur_file->file))
unknown's avatar
unknown committed
2696
	    goto found_eof;	/* FIXME: could we just break here?! */
2697
	  c= my_getc(cur_file->file);
unknown's avatar
unknown committed
2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711
	  *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;
    }
2712
  }
2713
  *p= 0;					/* Always end with \0 */
2714
  DBUG_RETURN(feof(cur_file->file));
unknown's avatar
unknown committed
2715 2716
}

2717 2718 2719 2720
/*
  Create a query from a set of lines

  SYNOPSIS
2721
    read_query()
2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734
    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
2735

2736 2737
static char read_query_buf[MAX_QUERY];

2738
int read_query(struct st_query** q_ptr)
unknown's avatar
unknown committed
2739
{
2740
  char *p= read_query_buf;
2741
  struct st_query* q;
unknown's avatar
unknown committed
2742
  DBUG_ENTER("read_query");
2743

2744 2745
  if (parser.current_line < parser.read_lines)
  {
2746
    get_dynamic(&q_lines, (gptr) q_ptr, parser.current_line) ;
unknown's avatar
unknown committed
2747
    DBUG_RETURN(0);
2748
  }
2749
  if (!(*q_ptr= q= (struct st_query*) my_malloc(sizeof(*q), MYF(MY_WME))) ||
unknown's avatar
unknown committed
2750 2751
      insert_dynamic(&q_lines, (gptr) &q))
    die(NullS);
2752

2753 2754 2755
  q->record_file[0]= 0;
  q->require_file= 0;
  q->first_word_len= 0;
unknown's avatar
unknown committed
2756

2757
  q->type= Q_UNKNOWN;
2758
  q->query_buf= q->query= 0;
2759
  read_query_buf[0]= 0;
2760
  if (read_line(read_query_buf, sizeof(read_query_buf)))
unknown's avatar
unknown committed
2761
  {
2762
    check_eol_junk(read_query_buf);
unknown's avatar
unknown committed
2763
    DBUG_RETURN(1);
unknown's avatar
unknown committed
2764
  }
2765
  
2766
  DBUG_PRINT("info", ("query: %s", read_query_buf));
2767 2768
  if (*p == '#')
  {
2769
    q->type= Q_COMMENT;
unknown's avatar
unknown committed
2770 2771
    /* This goto is to avoid losing the "expected error" info. */
    goto end;
2772
  }
2773 2774 2775 2776 2777 2778 2779 2780
  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
2781
  if (p[0] == '-' && p[1] == '-')
2782
  {
2783 2784
    q->type= Q_COMMENT_WITH_COMMAND;
    p+= 2;					/* To calculate first word */
2785
  }
2786
  else if (!parsing_disabled)
2787
  {
2788
    while (*p && my_isspace(charset_info, *p))
unknown's avatar
unknown committed
2789
      p++ ;
2790
  }
unknown's avatar
unknown committed
2791 2792

end:
2793
  while (*p && my_isspace(charset_info, *p))
unknown's avatar
unknown committed
2794
    p++;
unknown's avatar
unknown committed
2795

2796
  if (!(q->query_buf= q->query= my_strdup(p, MYF(MY_WME))))
2797 2798 2799
    die(NullS);

  /* Calculate first word and first argument */
2800 2801 2802
  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
2803
    p++;
2804 2805
  q->first_argument= p;
  q->end= strend(q->query);
2806
  parser.read_lines++;
unknown's avatar
unknown committed
2807
  DBUG_RETURN(0);
unknown's avatar
unknown committed
2808 2809
}

2810 2811 2812

static struct my_option my_long_options[] =
{
unknown's avatar
unknown committed
2813 2814
  {"help", '?', "Display this help and exit.", 0, 0, 0, GET_NO_ARG, NO_ARG,
   0, 0, 0, 0, 0, 0},
2815
  {"basedir", 'b', "Basedir for tests.", (gptr*) &opt_basedir,
2816
   (gptr*) &opt_basedir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
2817
  {"big-test", 'B', "Define BIG_TEST to 1.", (gptr*) &opt_big_test,
2818
   (gptr*) &opt_big_test, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
2819
  {"compress", 'C', "Use the compressed server/client protocol.",
2820 2821
   (gptr*) &opt_compress, (gptr*) &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0,
   0, 0, 0},
unknown's avatar
unknown committed
2822 2823 2824 2825 2826 2827 2828 2829 2830
  {"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
2831 2832
  {"host", 'h', "Connect to host.", (gptr*) &host, (gptr*) &host, 0,
   GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
2833
  {"manager-host", OPT_MANAGER_HOST, "Undocumented: Used for debugging.",
2834 2835
   (gptr*) &manager_host, (gptr*) &manager_host, 0, GET_STR, REQUIRED_ARG,
   0, 0, 0, 0, 0, 0},
2836
  {"manager-password", OPT_MANAGER_PASSWD, "Undocumented: Used for debugging.",
2837
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
2838
  {"manager-port", OPT_MANAGER_PORT, "Undocumented: Used for debugging.",
2839 2840
   (gptr*) &manager_port, (gptr*) &manager_port, 0, GET_INT, REQUIRED_ARG,
   MYSQL_MANAGER_PORT, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
2841 2842 2843
  {"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},
2844
  {"manager-wait-timeout", OPT_MANAGER_WAIT_TIMEOUT,
2845
   "Undocumented: Used for debugging.", (gptr*) &manager_wait_timeout,
2846 2847 2848 2849
   (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,
2850
   (gptr*) &port, 0, GET_INT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
2851 2852 2853
  {"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},
2854 2855 2856 2857 2858 2859 2860 2861 2862
  {"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},
  {"cursor-protocol", OPT_CURSOR_PROTOCOL, "Use cursors for prepared statment",
   (gptr*) &cursor_protocol, (gptr*) &cursor_protocol, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"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},
2863 2864 2865 2866 2867 2868 2869
  {"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},
2870
  {"server-arg", 'A', "Send enbedded server this as a paramenter.",
2871
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
2872
  {"server-file", 'F', "Read embedded server arguments from file.",
2873 2874 2875
   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},
unknown's avatar
unknown committed
2876
  {"skip-safemalloc", OPT_SKIP_SAFEMALLOC,
2877
   "Don't use the memory allocation checking.", 0, 0, 0, GET_NO_ARG, NO_ARG,
unknown's avatar
unknown committed
2878
   0, 0, 0, 0, 0, 0},
2879
  {"sleep", 'T', "Sleep always this many seconds on sleep commands.",
2880 2881 2882 2883 2884
   (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
2885
#include "sslopt-longopts.h"
2886 2887
  {"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
2888 2889
  {"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},
2890
  {"tmpdir", 't', "Temporary directory where sockets are put.",
2891 2892 2893 2894 2895 2896 2897 2898
   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},
  { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
unknown's avatar
unknown committed
2899 2900
};

unknown's avatar
unknown committed
2901 2902 2903

#include <help_start.h>

unknown's avatar
unknown committed
2904 2905 2906 2907 2908 2909 2910 2911 2912
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();
2913
  printf("MySQL AB, by Sasha, Matt, Monty & Jani\n");
unknown's avatar
unknown committed
2914 2915 2916
  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);
2917
  my_print_help(my_long_options);
unknown's avatar
unknown committed
2918
  printf("  --no-defaults       Don't read default options from any options file.\n");
2919
  my_print_variables(my_long_options);
unknown's avatar
unknown committed
2920 2921
}

unknown's avatar
unknown committed
2922 2923
#include <help_end.h>

2924 2925 2926 2927 2928

static my_bool
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
	       char *argument)
{
2929
  switch(optid) {
2930
  case '#':
unknown's avatar
unknown committed
2931
#ifndef DBUG_OFF
2932
    DBUG_PUSH(argument ? argument : "d:t:S:i:O,/tmp/mysqltest.trace");
unknown's avatar
unknown committed
2933
#endif
2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951
    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);
2952
      DBUG_ASSERT(cur_file == file_stack && cur_file->file == 0);
2953
      if (!(cur_file->file=
2954
            my_fopen(buff, O_RDONLY | FILE_BINARY, MYF(0))))
2955 2956
	die("Could not open %s: errno = %d", buff, errno);
      cur_file->file_name= my_strdup(buff, MYF(MY_FAE));
2957
      cur_file->lineno= 1;
2958 2959
      break;
    }
unknown's avatar
unknown committed
2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972
  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;
    }
2973 2974 2975 2976 2977 2978
  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 */
2979
      tty_password= 0;
2980 2981 2982 2983
    }
    else
      tty_password= 1;
    break;
unknown's avatar
unknown committed
2984
#include <sslopt-case.h>
2985 2986 2987 2988 2989 2990 2991 2992 2993
  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*) "";
    }
2994 2995 2996
    if (embedded_server_arg_count == MAX_SERVER_ARGS-1 ||
        !(embedded_server_args[embedded_server_arg_count++]=
          my_strdup(argument, MYF(MY_FAE))))
2997 2998 2999 3000 3001 3002 3003 3004
    {
      die("Can't use server argument");
    }
    break;
  case 'F':
    if (read_server_arguments(argument))
      die(NullS);
    break;
unknown's avatar
unknown committed
3005 3006 3007 3008 3009
  case OPT_SKIP_SAFEMALLOC:
#ifdef SAFEMALLOC
    sf_malloc_quick=1;
#endif
    break;
3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020
  case 'V':
    print_version();
    exit(0);
  case '?':
    usage();
    exit(1);
  }
  return 0;
}


unknown's avatar
unknown committed
3021 3022 3023
int parse_args(int argc, char **argv)
{
  load_defaults("my",load_default_groups,&argc,&argv);
3024
  default_argv= argv;
3025

3026
  if ((handle_options(&argc, &argv, my_long_options, get_one_option)))
unknown's avatar
unknown committed
3027
    exit(1);
unknown's avatar
unknown committed
3028 3029 3030 3031 3032 3033 3034

  if (argc > 1)
  {
    usage();
    exit(1);
  }
  if (argc == 1)
3035
    db= *argv;
unknown's avatar
unknown committed
3036 3037 3038 3039 3040 3041 3042
  if (tty_password)
    pass=get_tty_password(NullS);

  return 0;
}


unknown's avatar
unknown committed
3043
/*
3044
   Write the content of str into file
unknown's avatar
unknown committed
3045

3046 3047 3048 3049 3050 3051
   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
3052

3053
static void str_to_file(const char *fname, char *str, int size)
unknown's avatar
unknown committed
3054 3055
{
  int fd;
3056 3057 3058 3059
  char buff[FN_REFLEN];
  if (!test_if_hard_path(fname))
  {
    strxmov(buff, opt_basedir, fname, NullS);
unknown's avatar
unknown committed
3060
    fname= buff;
3061 3062
  }
  fn_format(buff,fname,"","",4);
unknown's avatar
unknown committed
3063

unknown's avatar
unknown committed
3064
  if ((fd= my_open(buff, O_WRONLY | O_CREAT | O_TRUNC,
3065
		    MYF(MY_WME | MY_FFNF))) < 0)
3066
    die("Could not open %s: errno = %d", buff, errno);
3067
  if (my_write(fd, (byte*)str, size, MYF(MY_WME|MY_FNABP)))
unknown's avatar
unknown committed
3068 3069 3070 3071
    die("write failed");
  my_close(fd, MYF(0));
}

unknown's avatar
unknown committed
3072

3073
void dump_result_to_reject_file(const char *record_file, char *buf, int size)
unknown's avatar
unknown committed
3074
{
unknown's avatar
unknown committed
3075
  char reject_file[FN_REFLEN];
unknown's avatar
unknown committed
3076
  str_to_file(fn_format(reject_file, record_file,"",".reject",2), buf, size);
unknown's avatar
unknown committed
3077 3078
}

3079

3080
#ifdef __WIN__
3081

3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093
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()
3094
{
3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106
  /* List of string patterns to match in order to find paths */
  const char* paths[] = { "$MYSQL_TEST_DIR", "./test/", 0 };
  int num_paths= 2;
  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++)
3107
  {
3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126
    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++;
    }
3127
  }
3128
  DBUG_VOID_RETURN;
3129 3130
}

3131 3132 3133 3134
static void free_win_path_patterns()
{
  int i= 0;
  for (i=0 ; i < patterns.elements ; i++)
3135
  {
3136 3137
    const char** pattern= dynamic_element(&patterns, i, const char**);
    my_free((gptr) *pattern, MYF(0));
3138
  }
3139
  delete_dynamic(&patterns);
3140 3141
}

3142 3143
/*
  fix_win_paths
unknown's avatar
unknown committed
3144

3145 3146 3147 3148 3149 3150 3151 3152 3153 3154
  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 /
*/
3155

3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197
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));
    /* 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

  if (glob_replace)
    replace_strings_append(glob_replace, ds, val, len);
  else
    dynstr_append_mem(ds, val, len);
}
3198

3199 3200

/* Append zero-terminated string to ds, with optional replace */
3201 3202 3203 3204 3205
static void replace_dynstr_append(DYNAMIC_STRING *ds, const char *val)
{
  replace_dynstr_append_mem(ds, val, strlen(val));
}

3206

3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224
/*
  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__
3225 3226
  else if ((field->type == MYSQL_TYPE_DOUBLE ||
            field->type == MYSQL_TYPE_FLOAT ) &&
3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257
           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);
  }
}


3258 3259
/*
  Append all results to the dynamic string separated with '\t'
3260
  Values may be converted with 'replace_column'
3261 3262 3263 3264 3265
*/

static void append_result(DYNAMIC_STRING *ds, MYSQL_RES *res)
{
  MYSQL_ROW row;
3266
  uint num_fields= mysql_num_fields(res);
3267
  MYSQL_FIELD *fields= mysql_fetch_fields(res);
3268
  ulong *lengths;
3269

3270 3271
  while ((row = mysql_fetch_row(res)))
  {
3272
    uint i;
3273 3274
    lengths = mysql_fetch_lengths(res);
    for (i = 0; i < num_fields; i++)
3275 3276
      append_field(ds, i, &fields[i],
                   (const char*)row[i], lengths[i], !row[i]);
3277 3278
    if (!display_result_vertically)
      dynstr_append_mem(ds, "\n", 1);
3279
  }
3280
  free_replace_column();
3281 3282
}

3283

3284
/*
unknown's avatar
unknown committed
3285
  Append all results from ps execution to the dynamic string separated
3286
  with '\t'. Values may be converted with 'replace_column'
3287
*/
3288

unknown's avatar
unknown committed
3289
static void append_stmt_result(DYNAMIC_STRING *ds, MYSQL_STMT *stmt,
3290
                               MYSQL_FIELD *fields, uint num_fields)
3291
{
3292 3293 3294
  MYSQL_BIND *bind;
  my_bool *is_null;
  ulong *length;
3295
  uint i;
3296

3297 3298 3299 3300 3301 3302 3303
  /* 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));
3304

3305 3306
  /* Allocate data for the result of each field */
  for (i= 0; i < num_fields; i++)
unknown's avatar
unknown committed
3307
  {
3308 3309 3310 3311 3312 3313
    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
3314

unknown's avatar
WL#2930  
unknown committed
3315
    DBUG_PRINT("bind", ("col[%d]: buffer_type: %d, buffer_length: %d",
3316
			i, bind[i].buffer_type, bind[i].buffer_length));
3317
  }
unknown's avatar
unknown committed
3318

3319
  if (mysql_stmt_bind_result(stmt, bind))
unknown's avatar
unknown committed
3320
    die("mysql_stmt_bind_result failed: %d: %s",
3321
	mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
3322

3323
  while (mysql_stmt_fetch(stmt) == 0)
3324
  {
3325 3326 3327
    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);
3328 3329
    if (!display_result_vertically)
      dynstr_append_mem(ds, "\n", 1);
3330 3331
  }

3332 3333
  if (mysql_stmt_fetch(stmt) != MYSQL_NO_DATA)
    die("fetch didn't end with MYSQL_NO_DATA from statement: %d %s",
3334
	mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
3335

3336
  free_replace_column();
unknown's avatar
unknown committed
3337

3338
  for (i= 0; i < num_fields; i++)
3339 3340
  {
    /* Free data for output */
3341
    my_free((gptr)bind[i].buffer, MYF(MY_WME | MY_FAE));
3342 3343 3344 3345 3346
  }
  /* 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
3347 3348 3349
}


3350
/*
3351
  Append metadata for fields to output
3352 3353
*/

3354
static void append_metadata(DYNAMIC_STRING *ds,
unknown's avatar
unknown committed
3355
			    MYSQL_FIELD *field,
3356
			    uint num_fields)
3357
{
3358 3359 3360 3361
  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");
3362

3363 3364 3365
  for (field_end= field+num_fields ;
       field < field_end ;
       field++)
3366
  {
3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395
    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);
3396

3397 3398 3399 3400 3401 3402 3403 3404 3405
    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
3406
  }
3407 3408
}

unknown's avatar
unknown committed
3409

3410 3411 3412
/*
  Append affected row count and other info to output
*/
3413

unknown's avatar
WL#2930  
unknown committed
3414
static void append_info(DYNAMIC_STRING *ds, ulonglong affected_rows,
unknown's avatar
unknown committed
3415
			const char *info)
unknown's avatar
unknown committed
3416
{
3417
  char buf[40];
unknown's avatar
WL#2930  
unknown committed
3418
  sprintf(buf,"affected rows: %llu\n", affected_rows);
3419 3420
  dynstr_append(ds, buf);
  if (info)
unknown's avatar
unknown committed
3421
  {
3422 3423 3424
    dynstr_append(ds, "info: ");
    dynstr_append(ds, info);
    dynstr_append_mem(ds, "\n", 1);
unknown's avatar
unknown committed
3425
  }
3426
}
unknown's avatar
unknown committed
3427 3428


unknown's avatar
unknown committed
3429 3430
/*
   Display the table headings with the names tab separated
3431
*/
unknown's avatar
unknown committed
3432 3433 3434

static void append_table_headings(DYNAMIC_STRING *ds,
				  MYSQL_FIELD *field,
3435 3436 3437 3438
				  uint num_fields)
{
  uint col_idx;
  for (col_idx= 0; col_idx < num_fields; col_idx++)
3439
  {
3440 3441 3442
    if (col_idx)
      dynstr_append_mem(ds, "\t", 1);
    replace_dynstr_append(ds, field[col_idx].name);
3443
  }
3444 3445 3446 3447
  dynstr_append_mem(ds, "\n", 1);
}

/*
unknown's avatar
WL#2930  
unknown committed
3448 3449 3450 3451
  Fetch warnings from server and append to ds

  RETURN VALUE
   Number of warnings appended to ds
3452 3453
*/

unknown's avatar
WL#2930  
unknown committed
3454
static int append_warnings(DYNAMIC_STRING *ds, MYSQL* mysql)
3455 3456
{
  uint count;
3457
  MYSQL_RES *warn_res;
3458 3459 3460
  DBUG_ENTER("append_warnings");

  if (!(count= mysql_warning_count(mysql)))
unknown's avatar
WL#2930  
unknown committed
3461
    DBUG_RETURN(0);
3462 3463 3464 3465 3466 3467 3468

  /*
    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
3469

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

3473
  if (!(warn_res= mysql_store_result(mysql)))
3474 3475
    die("Warning count is %u but didn't get any warnings",
	count);
unknown's avatar
unknown committed
3476

3477 3478 3479
  append_result(ds, warn_res);
  mysql_free_result(warn_res);

3480 3481
  DBUG_PRINT("warnings", ("%s", ds->str));

unknown's avatar
WL#2930  
unknown committed
3482
  DBUG_RETURN(count);
3483 3484 3485
}


unknown's avatar
WL#2930  
unknown committed
3486

3487 3488
/*
  Run query using MySQL C API
unknown's avatar
unknown committed
3489

3490 3491 3492 3493
  SYNPOSIS
  run_query_normal
  mysql - mysql handle
  command - currrent command pointer
3494
  flags -flags indicating wheter to SEND and/or REAP
3495 3496 3497 3498 3499 3500 3501 3502
  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
3503 3504
static void run_query_normal(MYSQL *mysql, struct st_query *command,
			     int flags, char *query, int query_len,
3505
			     DYNAMIC_STRING *ds, DYNAMIC_STRING *ds_warnings)
3506 3507 3508 3509 3510 3511
{
  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
3512

3513 3514
  if (flags & QUERY_SEND)
  {
unknown's avatar
unknown committed
3515
    /*
3516
       Send the query
3517 3518 3519 3520 3521 3522 3523
     */
    if (mysql_send_query(mysql, query, query_len))
    {
      handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		   mysql_sqlstate(mysql), ds);
      goto end;
    }
3524
  }
unknown's avatar
unknown committed
3525

3526 3527
  if (!(flags & QUERY_REAP))
    DBUG_VOID_RETURN;
unknown's avatar
unknown committed
3528

3529
  do
3530
  {
3531
    /*
unknown's avatar
unknown committed
3532
      When  on first result set, call mysql_read_query_result to retrieve
3533 3534 3535
      answer to the query sent earlier
     */
    if ((counter==0) && mysql_read_query_result(mysql))
3536
    {
3537 3538 3539 3540
      handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		   mysql_sqlstate(mysql), ds);
      goto end;

3541
    }
3542

unknown's avatar
unknown committed
3543 3544
    /*
       Store the result. If res is NULL, use mysql_field_count to
3545 3546 3547
       determine if that was expected
     */
    if (!(res= mysql_store_result(mysql)) && mysql_field_count(mysql))
unknown's avatar
unknown committed
3548
    {
3549 3550
      handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		   mysql_sqlstate(mysql), ds);
3551 3552
      goto end;
    }
3553

3554
    if (!disable_result_log)
3555
    {
unknown's avatar
WL#2930  
unknown committed
3556
      ulonglong affected_rows;    /* Ok to be undef if 'disable_info' is set */
unknown's avatar
unknown committed
3557
      LINT_INIT(affected_rows);
unknown's avatar
unknown committed
3558

3559
      if (res)
3560
      {
3561
	MYSQL_FIELD *fields= mysql_fetch_fields(res);
3562
	uint num_fields= mysql_num_fields(res);
3563

3564
	if (display_metadata)
3565
          append_metadata(ds, fields, num_fields);
3566

3567
	if (!display_result_vertically)
3568 3569
	  append_table_headings(ds, fields, num_fields);

3570
	append_result(ds, res);
3571
      }
unknown's avatar
unknown committed
3572

unknown's avatar
unknown committed
3573
      /*
3574
        Need to call mysql_affected_rows() before the "new"
unknown's avatar
unknown committed
3575 3576 3577
        query to find the warnings
      */
      if (!disable_info)
unknown's avatar
WL#2930  
unknown committed
3578
        affected_rows= mysql_affected_rows(mysql);
unknown's avatar
unknown committed
3579

3580 3581 3582 3583 3584
      /*
        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.
      */
3585
      if (!disable_warnings && !mysql_more_results(mysql))
3586
      {
3587
	if (append_warnings(ds_warnings, mysql) || ds_warnings->length)
3588 3589
	{
	  dynstr_append_mem(ds, "Warnings:\n", 10);
3590
	  dynstr_append_mem(ds, ds_warnings->str, ds_warnings->length);
3591
	}
3592
      }
3593

unknown's avatar
unknown committed
3594
      if (!disable_info)
3595
	append_info(ds, affected_rows, mysql_info(mysql));
3596
    }
3597

3598 3599 3600 3601
    if (res)
      mysql_free_result(res);
    counter++;
  } while (!(err= mysql_next_result(mysql)));
3602 3603
  if (err > 0)
  {
3604 3605 3606
    /* We got an error from mysql_next_result, maybe expected */
    handle_error(query, command, mysql_errno(mysql), mysql_error(mysql),
		 mysql_sqlstate(mysql), ds);
3607 3608
    goto end;
  }
3609
  DBUG_ASSERT(err == -1); /* Successful and there are no more results */
3610

3611
  /* If we come here the query is both executed and read successfully */
3612
  handle_no_error(command);
3613

3614
end:
3615
  free_replace();
unknown's avatar
unknown committed
3616 3617 3618 3619 3620 3621

  /*
    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.
  */
3622
  var_set_errno(mysql_errno(mysql));
3623
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
3624 3625 3626
}


3627
/*
3628
  Handle errors which occurred during execution
3629 3630

  SYNOPSIS
3631
    handle_error()
3632 3633
      query - query string
      q     - query context
3634 3635 3636
      err_errno - error number
      err_error - error message
      err_sqlstate - sql state
3637 3638 3639 3640 3641 3642 3643
      ds    - dynamic string which is used for output buffer

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

  RETURN VALUE
3644
    error - function will not return
3645 3646
*/

3647 3648 3649
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)
3650 3651
{
  uint i;
unknown's avatar
unknown committed
3652

3653
  DBUG_ENTER("handle_error");
3654 3655 3656

  if (q->require_file)
    abort_not_supported_test();
unknown's avatar
unknown committed
3657

3658
  if (q->abort_on_error)
3659
    die("query '%s' failed: %d: %s", query, err_errno, err_error);
3660 3661

  for (i= 0 ; (uint) i < q->expected_errors ; i++)
3662
  {
3663 3664 3665 3666
    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)))
3667
    {
3668
      if (!disable_result_log)
3669
      {
3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683
        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");
3684
      }
3685
      /* OK */
3686
      DBUG_VOID_RETURN;
3687
    }
3688
  }
3689

3690
  DBUG_PRINT("info",("i: %d  expected_errors: %d", i, q->expected_errors));
3691

3692 3693 3694 3695 3696 3697 3698 3699
  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);
  }
3700

3701 3702 3703
  if (i)
  {
    if (q->expected_errno[0].type == ERR_ERRNO)
3704 3705
      die("query '%s' failed with wrong errno %d: '%s', instead of %d...",
          q->query, err_errno, err_error, q->expected_errno[0].code.errnum);
3706
    else
3707 3708 3709
      die("query '%s' failed with wrong sqlstate %s: '%s', instead of %s...",
          q->query, err_sqlstate, err_error,
	  q->expected_errno[0].code.sqlstate);
3710
  }
3711

3712
  DBUG_VOID_RETURN;
3713 3714 3715 3716
}


/*
3717
  Handle absence of errors after execution
3718 3719

  SYNOPSIS
3720
    handle_no_error()
3721 3722 3723
      q - context of query

  RETURN VALUE
3724
    error - function will not return
3725 3726
*/

3727
static void handle_no_error(struct st_query *q)
3728
{
3729
  DBUG_ENTER("handle_no_error");
3730 3731 3732 3733 3734

  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 */
3735 3736
    die("query '%s' succeeded - should have failed with errno %d...",
        q->query, q->expected_errno[0].code.errnum);
3737 3738 3739 3740 3741
  }
  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 */
3742 3743
    die("query '%s' succeeded - should have failed with sqlstate %s...",
        q->query, q->expected_errno[0].code.sqlstate);
3744 3745
  }

3746
  DBUG_VOID_RETURN;
3747 3748
}

3749 3750

/*
3751
  Run query using prepared statement C API
unknown's avatar
unknown committed
3752

3753 3754 3755 3756 3757 3758 3759
  SYNPOSIS
  run_query_stmt
  mysql - mysql handle
  command - currrent command pointer
  query - query string to execute
  query_len - length query string to execute
  ds - output buffer wherte to store result form query
3760

3761 3762
  RETURN VALUE
  error - function will not return
3763 3764
*/

unknown's avatar
unknown committed
3765
static void run_query_stmt(MYSQL *mysql, struct st_query *command,
3766 3767
			   char *query, int query_len, DYNAMIC_STRING *ds,
			   DYNAMIC_STRING *ds_warnings)
3768 3769 3770
{
  MYSQL_RES *res= NULL;     /* Note that here 'res' is meta data result set */
  MYSQL_STMT *stmt;
3771
  DYNAMIC_STRING ds_prepare_warnings;
unknown's avatar
WL#2930  
unknown committed
3772
  DYNAMIC_STRING ds_execute_warnings;
3773
  DBUG_ENTER("run_query_stmt");
3774
  DBUG_PRINT("query", ("'%-.60s'", query));
3775 3776

  /*
3777
    Init a new stmt if it's not alreday one created for this connectoon
3778
  */
3779
  if(!(stmt= cur_con->stmt))
3780
  {
3781 3782 3783
    if (!(stmt= mysql_stmt_init(mysql)))
      die("unable to init stmt structure");
    cur_con->stmt= stmt;
3784 3785
  }

unknown's avatar
unknown committed
3786 3787
  /* Init dynamic strings for warnings */
  if (!disable_warnings)
3788
  {
unknown's avatar
unknown committed
3789 3790
    init_dynamic_string(&ds_prepare_warnings, NULL, 0, 256);
    init_dynamic_string(&ds_execute_warnings, NULL, 0, 256);
3791 3792 3793
  }

  /*
3794
    Prepare the query
3795
  */
3796
  if (mysql_stmt_prepare(stmt, query, query_len))
3797
  {
3798
    handle_error(query, command,  mysql_stmt_errno(stmt),
3799
                 mysql_stmt_error(stmt), mysql_stmt_sqlstate(stmt), ds);
3800
    goto end;
3801 3802 3803
  }

  /*
unknown's avatar
unknown committed
3804 3805
    Get the warnings from mysql_stmt_prepare and keep them in a
    separate string
3806
  */
3807 3808
  if (!disable_warnings)
    append_warnings(&ds_prepare_warnings, mysql);
3809 3810

  /*
3811
    No need to call mysql_stmt_bind_param() because we have no
3812 3813 3814
    parameter markers.
  */

3815 3816
  if (cursor_protocol_enabled)
  {
3817 3818 3819
    /*
      Use cursor when retrieving result
    */
unknown's avatar
WL#2930  
unknown committed
3820
    ulong type= CURSOR_TYPE_READ_ONLY;
3821
    if (mysql_stmt_attr_set(stmt, STMT_ATTR_CURSOR_TYPE, (void*) &type))
unknown's avatar
unknown committed
3822
      die("mysql_stmt_attr_set(STMT_ATTR_CURSOR_TYPE) failed': %d %s",
3823
          mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
3824
  }
3825

3826 3827
  /*
    Execute the query
3828
  */
unknown's avatar
unknown committed
3829
  if (mysql_stmt_execute(stmt))
3830
  {
3831
    handle_error(query, command, mysql_stmt_errno(stmt),
3832
                 mysql_stmt_error(stmt), mysql_stmt_sqlstate(stmt), ds);
3833
    goto end;
3834 3835
  }

3836 3837 3838 3839 3840 3841 3842
  /*
    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);

3843 3844 3845 3846 3847 3848 3849
  /*
    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;
3850
    if (mysql_stmt_attr_set(stmt, STMT_ATTR_UPDATE_MAX_LENGTH, (void*) &one))
unknown's avatar
unknown committed
3851
      die("mysql_stmt_attr_set(STMT_ATTR_UPDATE_MAX_LENGTH) failed': %d %s",
3852
          mysql_stmt_errno(stmt), mysql_stmt_error(stmt));
3853 3854 3855 3856 3857 3858
  }

  /*
    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!
  */
3859
  if (mysql_stmt_store_result(stmt))
3860
  {
3861
    handle_error(query, command, mysql_stmt_errno(stmt),
3862
                 mysql_stmt_error(stmt), mysql_stmt_sqlstate(stmt), ds);
3863
    goto end;
3864
  }
3865

3866
  /* If we got here the statement was both executed and read succeesfully */
3867 3868
  handle_no_error(command);
  if (!disable_result_log)
3869
  {
3870 3871 3872 3873 3874 3875
    /*
      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.
    */
3876
    if ((res= mysql_stmt_result_metadata(stmt)) != NULL)
unknown's avatar
unknown committed
3877
    {
3878 3879 3880
      /* Take the column count from meta info */
      MYSQL_FIELD *fields= mysql_fetch_fields(res);
      uint num_fields= mysql_num_fields(res);
3881

3882
      if (display_metadata)
3883
        append_metadata(ds, fields, num_fields);
3884

3885
      if (!display_result_vertically)
3886
        append_table_headings(ds, fields, num_fields);
3887

unknown's avatar
unknown committed
3888
      append_stmt_result(ds, stmt, fields, num_fields);
3889

3890
      mysql_free_result(res);     /* Free normal result set with meta data */
3891

unknown's avatar
WL#2930  
unknown committed
3892 3893
      /* Clear prepare warnings */
      dynstr_set(&ds_prepare_warnings, NULL);
unknown's avatar
unknown committed
3894 3895
    }
    else
3896
    {
3897 3898 3899
      /*
	This is a query without resultset
      */
3900 3901
    }

3902
    if (!disable_warnings)
3903
    {
unknown's avatar
WL#2930  
unknown committed
3904
      /* Get the warnings from execute */
3905

unknown's avatar
WL#2930  
unknown committed
3906 3907
      /* Append warnings to ds - if there are any */
      if (append_warnings(&ds_execute_warnings, mysql) ||
3908 3909 3910
          ds_execute_warnings.length ||
          ds_prepare_warnings.length ||
          ds_warnings->length)
3911
      {
3912
        dynstr_append_mem(ds, "Warnings:\n", 10);
3913 3914 3915
	if (ds_warnings->length)
	  dynstr_append_mem(ds, ds_warnings->str,
			    ds_warnings->length);
unknown's avatar
WL#2930  
unknown committed
3916 3917 3918 3919 3920 3921
	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);
3922 3923 3924 3925
      }
    }

    if (!disable_info)
unknown's avatar
WL#2930  
unknown committed
3926
      append_info(ds, mysql_affected_rows(mysql), mysql_info(mysql));
3927 3928 3929 3930 3931

  }

end:
  free_replace();
unknown's avatar
unknown committed
3932

3933
  if (!disable_warnings)
unknown's avatar
WL#2930  
unknown committed
3934
  {
3935
    dynstr_free(&ds_prepare_warnings);
unknown's avatar
WL#2930  
unknown committed
3936 3937
    dynstr_free(&ds_execute_warnings);
  }
3938 3939

  /*
3940 3941 3942
    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.
3943
  */
unknown's avatar
unknown committed
3944
  var_set_errno(mysql_stmt_errno(stmt));
unknown's avatar
unknown committed
3945
#ifndef BUG15518_FIXED
3946
  mysql_stmt_close(stmt);
unknown's avatar
unknown committed
3947 3948
  cur_con->stmt= NULL;
#endif
3949
  DBUG_VOID_RETURN;
3950 3951 3952
}


3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985

/*
  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);
}



3986 3987
/*
  Run query
3988

3989 3990 3991
  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
3992

3993 3994 3995 3996
  SYNPOSIS
  run_query
  mysql - mysql handle
  command - currrent command pointer
unknown's avatar
unknown committed
3997

3998
*/
3999

4000
static void run_query(MYSQL *mysql, struct st_query *command, int flags)
4001
{
4002
  DYNAMIC_STRING *ds;
4003
  DYNAMIC_STRING ds_result;
4004
  DYNAMIC_STRING ds_warnings;
4005 4006 4007
  DYNAMIC_STRING eval_query;
  char *query;
  int query_len;
4008 4009
  my_bool view_created= 0, sp_created= 0;
  my_bool complete_query= ((flags & QUERY_SEND) && (flags & QUERY_REAP));
4010

4011 4012
  init_dynamic_string(&ds_warnings, NULL, 0, 256);

4013 4014 4015 4016
  /*
    Evaluate query if this is an eval command
   */
  if (command->type == Q_EVAL)
4017
  {
4018
    init_dynamic_string(&eval_query, "", 16384, 65536);
4019
    do_eval(&eval_query, command->query, FALSE);
4020 4021
    query = eval_query.str;
    query_len = eval_query.length;
4022 4023 4024
  }
  else
  {
4025 4026
    query = command->query;
    query_len = strlen(query);
4027 4028
  }

4029
  /*
unknown's avatar
unknown committed
4030
    When command->record_file is set the output of _this_ query
4031 4032 4033 4034 4035
    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])
4036
  {
4037 4038
    init_dynamic_string(&ds_result, "", 16384, 65536);
    ds= &ds_result;
4039 4040 4041 4042
  }
  else
    ds= &ds_res;

unknown's avatar
unknown committed
4043 4044
  /*
     Log the query into the output buffer
4045
  */
4046
  if (!disable_query_log && (flags & QUERY_SEND))
4047
  {
4048
    replace_dynstr_append_mem(ds, query, query_len);
4049 4050 4051 4052
    dynstr_append_mem(ds, delimiter, delimiter_length);
    dynstr_append_mem(ds, "\n", 1);
  }

4053 4054 4055
  if (view_protocol_enabled &&
      complete_query &&
      match_re(&view_re, query))
4056
  {
4057
    /*
4058
       Create the query as a view.
unknown's avatar
unknown committed
4059
       Use replace since view can exist from a failed mysqltest run
4060
    */
unknown's avatar
unknown committed
4061 4062 4063
    DYNAMIC_STRING query_str;
    init_dynamic_string(&query_str,
			"CREATE OR REPLACE VIEW mysqltest_tmp_v AS ",
4064 4065
			query_len+64, 256);
    dynstr_append_mem(&query_str, query, query_len);
4066
    if (util_query(mysql, query_str.str))
4067 4068 4069 4070 4071
    {
      /*
	Failed to create the view, this is not fatal
	just run the query the normal way
       */
unknown's avatar
unknown committed
4072
      DBUG_PRINT("view_create_error",
4073 4074
		 ("Failed to create view '%s': %d: %s", query_str.str,
		  mysql_errno(mysql), mysql_error(mysql)));
4075

unknown's avatar
WL#2930  
unknown committed
4076 4077 4078
      /* Log error to create view */
      verbose_msg("Failed to create view '%s' %d: %s", query_str.str,
		  mysql_errno(mysql), mysql_error(mysql));
4079 4080 4081 4082 4083 4084 4085
    }
    else
    {
      /*
	Yes, it was possible to create this query as a view
       */
      view_created= 1;
unknown's avatar
unknown committed
4086
      query= (char*)"SELECT * FROM mysqltest_tmp_v";
4087
      query_len = strlen(query);
4088

4089 4090 4091 4092
      /*
	 Collect warnings from create of the view that should otherwise
         have been produced when the SELECT was executed
      */
4093
      append_warnings(&ds_warnings, cur_con->util_mysql);
4094 4095
    }

4096
    dynstr_free(&query_str);
4097 4098 4099

  }

4100 4101 4102
  if (sp_protocol_enabled &&
      complete_query &&
      match_re(&sp_re, query))
4103
  {
4104
    /*
4105
      Create the query as a stored procedure
unknown's avatar
unknown committed
4106
      Drop first since sp can exist from a failed mysqltest run
4107
    */
4108
    DYNAMIC_STRING query_str;
unknown's avatar
unknown committed
4109
    init_dynamic_string(&query_str,
4110
			"DROP PROCEDURE IF EXISTS mysqltest_tmp_sp;",
4111
			query_len+64, 256);
4112
    util_query(mysql, query_str.str);
4113 4114
    dynstr_set(&query_str, "CREATE PROCEDURE mysqltest_tmp_sp()\n");
    dynstr_append_mem(&query_str, query, query_len);
4115
    if (util_query(mysql, query_str.str))
4116
    {
4117
      /*
unknown's avatar
unknown committed
4118
	Failed to create the stored procedure for this query,
4119 4120 4121 4122 4123
	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
WL#2930  
unknown committed
4124 4125 4126 4127

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

4129
    }
4130
    else
4131
    {
4132
      sp_created= 1;
unknown's avatar
unknown committed
4133 4134

      query= (char*)"CALL mysqltest_tmp_sp()";
4135
      query_len = strlen(query);
4136
    }
4137
    dynstr_free(&query_str);
4138 4139 4140
  }

  /*
4141
    Find out how to run this query
4142

unknown's avatar
unknown committed
4143
    Always run with normal C API if it's not a complete
4144
    SEND + REAP
4145

4146
    If it is a '?' in the query it may be a SQL level prepared
4147
    statement already and we can't do it twice
4148
  */
4149
  if (ps_protocol_enabled &&
4150 4151
      complete_query &&
      match_re(&ps_re, query))
4152
    run_query_stmt(mysql, command, query, query_len, ds, &ds_warnings);
4153
  else
4154 4155
    run_query_normal(mysql, command, flags, query, query_len,
		     ds, &ds_warnings);
unknown's avatar
unknown committed
4156

4157
  if (sp_created)
4158
  {
4159
    if (util_query(mysql, "DROP PROCEDURE mysqltest_tmp_sp "))
4160
      die("Failed to drop sp: %d: %s", mysql_errno(mysql), mysql_error(mysql));
4161 4162
  }

4163
  if (view_created)
4164
  {
4165
    if (util_query(mysql, "DROP VIEW mysqltest_tmp_v "))
unknown's avatar
unknown committed
4166
      die("Failed to drop view: %d: %s",
4167
	  mysql_errno(mysql), mysql_error(mysql));
4168 4169
  }

unknown's avatar
unknown committed
4170
  if (command->record_file[0])
4171 4172
  {

unknown's avatar
unknown committed
4173 4174
    /* A result file was specified for _this_ query  */
    if (record)
4175
    {
unknown's avatar
unknown committed
4176 4177 4178 4179 4180
      /*
	 Recording in progress
         Dump the output from _this_ query to the specified record_file
      */
      str_to_file(command->record_file, ds->str, ds->length);
4181

unknown's avatar
unknown committed
4182
    } else {
4183

unknown's avatar
unknown committed
4184 4185 4186 4187 4188
      /*
	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);
4189 4190 4191
    }
  }

4192
  dynstr_free(&ds_warnings);
4193 4194
  if (ds == &ds_result)
    dynstr_free(&ds_result);
4195
  if (command->type == Q_EVAL)
4196 4197 4198 4199 4200
    dynstr_free(&eval_query);
}


/****************************************************************************\
4201
 *  Functions to detect different SQL statements
4202 4203
\****************************************************************************/

4204
static char *re_eprint(int err)
4205
{
4206
  static char epbuf[100];
unknown's avatar
unknown committed
4207
  size_t len= my_regerror(REG_ITOA|err, (my_regex_t *)NULL,
4208 4209 4210
			  epbuf, sizeof(epbuf));
  assert(len <= sizeof(epbuf));
  return(epbuf);
4211 4212
}

4213
static void init_re_comp(my_regex_t *re, const char* str)
4214
{
4215 4216 4217
  int err= my_regcomp(re, str, (REG_EXTENDED | REG_ICASE | REG_NOSUB),
                      &my_charset_latin1);
  if (err)
4218
  {
4219 4220
    char erbuf[100];
    int len= my_regerror(err, re, erbuf, sizeof(erbuf));
unknown's avatar
unknown committed
4221
    die("error %s, %d/%d `%s'\n",
4222
	re_eprint(err), len, (int)sizeof(erbuf), erbuf);
4223 4224 4225
  }
}

4226
static void init_re(void)
4227
{
unknown's avatar
unknown committed
4228 4229
  /*
     Filter for queries that can be run using the
4230 4231
     MySQL Prepared Statements C API
  */
4232 4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244 4245
  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
4246 4247
  /*
     Filter for queries that can be run using the
4248 4249 4250 4251
     Stored procedures
  */
  const char *sp_re_str =ps_re_str;

unknown's avatar
unknown committed
4252
  /*
4253 4254 4255 4256 4257 4258 4259 4260 4261
     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);
4262 4263 4264
}


4265
static int match_re(my_regex_t *re, char *str)
4266
{
4267
  int err= my_regexec(re, str, (size_t)0, NULL, 0);
4268 4269 4270 4271 4272

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

4274 4275
  {
    char erbuf[100];
4276 4277 4278
    int len= my_regerror(err, re, erbuf, sizeof(erbuf));
    die("error %s, %d/%d `%s'\n",
	re_eprint(err), len, (int)sizeof(erbuf), erbuf);
4279
  }
4280
  return 0;
4281 4282
}

4283
static void free_re(void)
4284
{
unknown's avatar
unknown committed
4285
  my_regfree(&ps_re);
4286 4287
  my_regfree(&sp_re);
  my_regfree(&view_re);
unknown's avatar
unknown committed
4288
  my_regex_end();
4289 4290 4291 4292
}

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

4293
void get_query_type(struct st_query* q)
4294
{
4295 4296
  char save;
  uint type;
unknown's avatar
unknown committed
4297 4298
  DBUG_ENTER("get_query_type");

4299
  if (!parsing_disabled && *q->query == '}')
4300 4301
  {
    q->type = Q_END_BLOCK;
unknown's avatar
unknown committed
4302
    DBUG_VOID_RETURN;
4303 4304
  }
  if (q->type != Q_COMMENT_WITH_COMMAND)
4305
    q->type= parsing_disabled ? Q_COMMENT : Q_QUERY;
4306

4307 4308
  save=q->query[q->first_word_len];
  q->query[q->first_word_len]=0;
4309
  type=find_type(q->query, &command_typelib, 1+2);
4310
  q->query[q->first_word_len]=save;
4311
  if (type > 0)
4312
  {
unknown's avatar
unknown committed
4313
    q->type=(enum enum_commands) type;		/* Found command */
4314 4315 4316 4317 4318 4319 4320 4321
    /*
      If queries are disabled, only recognize
      --enable-queries and --disable-queries
    */
    if (parsing_disabled && q->type != Q_ENABLE_PARSING &&
        q->type != Q_DISABLE_PARSING)
      q->type= Q_COMMENT;
  }
4322
  else if (q->type == Q_COMMENT_WITH_COMMAND &&
unknown's avatar
unknown committed
4323
	   q->first_word_len &&
4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338
           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
4339
  DBUG_VOID_RETURN;
4340
}
unknown's avatar
unknown committed
4341

unknown's avatar
unknown committed
4342

unknown's avatar
unknown committed
4343
static byte *get_var_key(const byte* var, uint* len,
unknown's avatar
unknown committed
4344
			 my_bool __attribute__((unused)) t)
unknown's avatar
unknown committed
4345 4346 4347 4348 4349 4350 4351
{
  register char* key;
  key = ((VAR*)var)->name;
  *len = ((VAR*)var)->name_len;
  return (byte*)key;
}

unknown's avatar
unknown committed
4352
static VAR *var_init(VAR *v, const char *name, int name_len, const char *val,
unknown's avatar
unknown committed
4353 4354 4355
		     int val_len)
{
  int val_alloc_len;
unknown's avatar
unknown committed
4356
  VAR *tmp_var;
4357
  if (!name_len && name)
unknown's avatar
unknown committed
4358
    name_len = strlen(name);
4359
  if (!val_len && val)
unknown's avatar
unknown committed
4360 4361
    val_len = strlen(val) ;
  val_alloc_len = val_len + 16; /* room to grow */
4362
  if (!(tmp_var=v) && !(tmp_var = (VAR*)my_malloc(sizeof(*tmp_var)
unknown's avatar
unknown committed
4363
						 + name_len+1, MYF(MY_WME))))
unknown's avatar
unknown committed
4364
    die("Out of memory");
unknown's avatar
unknown committed
4365

4366
  tmp_var->name = (name) ? (char*) tmp_var + sizeof(*tmp_var) : 0;
4367
  tmp_var->alloced = (v == 0);
4368

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

4372 4373
  if (name)
    strmake(tmp_var->name, name, name_len);
4374
  if (val)
4375
    strmake(tmp_var->str_val, val, val_len);
unknown's avatar
unknown committed
4376 4377 4378
  tmp_var->name_len = name_len;
  tmp_var->str_val_len = val_len;
  tmp_var->alloced_len = val_alloc_len;
4379
  tmp_var->int_val = (val) ? atoi(val) : 0;
unknown's avatar
unknown committed
4380 4381 4382 4383
  tmp_var->int_dirty = 0;
  return tmp_var;
}

unknown's avatar
unknown committed
4384
static void var_free(void *v)
unknown's avatar
unknown committed
4385
{
unknown's avatar
unknown committed
4386
  my_free(((VAR*) v)->str_val, MYF(MY_WME));
unknown's avatar
unknown committed
4387 4388
  if (((VAR*)v)->alloced)
   my_free((char*) v, MYF(MY_WME));
unknown's avatar
unknown committed
4389 4390 4391
}


4392
static VAR* var_from_env(const char *name, const char *def_val)
unknown's avatar
unknown committed
4393
{
unknown's avatar
unknown committed
4394 4395
  const char *tmp;
  VAR *v;
4396
  if (!(tmp = getenv(name)))
unknown's avatar
unknown committed
4397
    tmp = def_val;
unknown's avatar
unknown committed
4398

unknown's avatar
unknown committed
4399
  v = var_init(0, name, strlen(name), tmp, strlen(tmp));
unknown's avatar
SCRUM  
unknown committed
4400
  my_hash_insert(&var_hash, (byte*)v);
4401
  return v;
4402
}
unknown's avatar
unknown committed
4403

unknown's avatar
unknown committed
4404

unknown's avatar
unknown committed
4405
static void init_var_hash(MYSQL *mysql)
unknown's avatar
unknown committed
4406
{
unknown's avatar
unknown committed
4407
  VAR *v;
unknown's avatar
unknown committed
4408
  DBUG_ENTER("init_var_hash");
unknown's avatar
unknown committed
4409
  if (hash_init(&var_hash, charset_info,
unknown's avatar
unknown committed
4410
                1024, 0, 0, get_var_key, var_free, MYF(0)))
unknown's avatar
unknown committed
4411
    die("Variable hash initialization failed");
4412 4413
  my_hash_insert(&var_hash, (byte*) var_init(0,"BIG_TEST", 0,
                                             (opt_big_test) ? "1" : "0", 0));
unknown's avatar
unknown committed
4414
  v= var_init(0,"MAX_TABLES", 0, (sizeof(ulong) == 4) ? "31" : "62",0);
unknown's avatar
SCRUM  
unknown committed
4415
  my_hash_insert(&var_hash, (byte*) v);
unknown's avatar
unknown committed
4416
  v= var_init(0,"SERVER_VERSION", 0, mysql_get_server_info(mysql), 0);
4417
  my_hash_insert(&var_hash, (byte*) v);  v= var_init(0,"DB", 2, db, 0);
unknown's avatar
unknown committed
4418
  my_hash_insert(&var_hash, (byte*) v);
unknown's avatar
unknown committed
4419
  DBUG_VOID_RETURN;
unknown's avatar
unknown committed
4420
}
4421

4422

unknown's avatar
unknown committed
4423
int main(int argc, char **argv)
unknown's avatar
unknown committed
4424
{
unknown's avatar
unknown committed
4425
  struct st_query *q;
4426 4427
  my_bool require_file=0, q_send_flag=0, abort_flag= 0,
          query_executed= 0;
4428
  char save_file[FN_REFLEN];
4429
  MY_STAT res_info;
4430 4431
  MY_INIT(argv[0]);

unknown's avatar
unknown committed
4432 4433 4434
  /* Use all time until exit if no explicit 'start_timer' */
  timer_start= timer_now();

4435
  save_file[0]=0;
4436
  TMPDIR[0]=0;
4437 4438

  /* Init cons */
unknown's avatar
unknown committed
4439 4440 4441 4442
  memset(cons, 0, sizeof(cons));
  cons_end = cons + MAX_CONS;
  next_con = cons + 1;
  cur_con = cons;
unknown's avatar
unknown committed
4443

4444
  /* Init file stack */
unknown's avatar
unknown committed
4445
  memset(file_stack, 0, sizeof(file_stack));
4446 4447
  file_stack_end= file_stack + MAX_INCLUDE_DEPTH - 1;
  cur_file= file_stack;
4448

4449
  /* Init block stack */
4450
  memset(block_stack, 0, sizeof(block_stack));
4451
  block_stack_end= block_stack + BLOCK_STACK_DEPTH - 1;
unknown's avatar
unknown committed
4452 4453 4454 4455
  cur_block= block_stack;
  cur_block->ok= TRUE; /* Outer block should always be executed */
  cur_block->cmd= cmd_none;

4456 4457 4458 4459 4460
  my_init_dynamic_array(&q_lines, sizeof(struct st_query*), INIT_Q_LINES,
		     INIT_Q_LINES);

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

4461
  init_dynamic_string(&ds_res, "", 0, 65536);
unknown's avatar
unknown committed
4462
  parse_args(argc, argv);
4463 4464

  DBUG_PRINT("info",("result_file: '%s'", result_file ? result_file : ""));
4465 4466 4467
  if (mysql_server_init(embedded_server_arg_count,
			embedded_server_args,
			(char**) embedded_server_groups))
unknown's avatar
unknown committed
4468
    die("Can't initialize MySQL server");
4469
  if (cur_file == file_stack && cur_file->file == 0)
4470
  {
4471 4472
    cur_file->file= stdin;
    cur_file->file_name= my_strdup("<stdin>", MYF(MY_WME));
4473
    cur_file->lineno= 1;
4474
  }
4475 4476 4477
#ifndef EMBEDDED_LIBRARY
  if (manager_host)
    init_manager();
4478
#endif
4479 4480 4481 4482 4483 4484 4485
  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)
4486
    ps_protocol_enabled= 1;
4487

4488
  if (!( mysql_init(&cur_con->mysql)))
unknown's avatar
unknown committed
4489
    die("Failed in mysql_init()");
unknown's avatar
unknown committed
4490 4491
  if (opt_compress)
    mysql_options(&cur_con->mysql,MYSQL_OPT_COMPRESS,NullS);
4492
  mysql_options(&cur_con->mysql, MYSQL_OPT_LOCAL_INFILE, 0);
unknown's avatar
unknown committed
4493
  mysql_options(&cur_con->mysql, MYSQL_SET_CHARSET_NAME, charset_name);
unknown's avatar
unknown committed
4494

unknown's avatar
unknown committed
4495 4496 4497 4498 4499
#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
4500

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

unknown's avatar
unknown committed
4504
  if (safe_connect(&cur_con->mysql, host, user, pass, db, port, unix_sock))
4505 4506
    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
4507

unknown's avatar
unknown committed
4508 4509
  init_var_hash(&cur_con->mysql);

4510 4511 4512 4513
#ifdef __WIN__
  init_win_path_patterns();
#endif

unknown's avatar
unknown committed
4514 4515 4516 4517 4518
  /*
    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
  */
4519 4520
  var_set_errno(-1);

unknown's avatar
unknown committed
4521
  while (!abort_flag && !read_query(&q))
4522 4523 4524 4525
  {
    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
4526
    if (cur_block->ok)
unknown's avatar
unknown committed
4527
    {
4528
      q->last_argument= q->first_argument;
4529 4530
      processed = 1;
      switch (q->type) {
4531
      case Q_CONNECT:
4532
        do_connect(q);
4533
        break;
4534
      case Q_CONNECTION: select_connection(q); break;
unknown's avatar
unknown committed
4535
      case Q_DISCONNECT:
4536
      case Q_DIRTY_CLOSE:
unknown's avatar
unknown committed
4537
	close_connection(q); break;
4538
      case Q_RPL_PROBE: do_rpl_probe(q); break;
4539
      case Q_ENABLE_RPL_PARSE:	 do_enable_rpl_parse(q); break;
unknown's avatar
unknown committed
4540
      case Q_DISABLE_RPL_PARSE:  do_disable_rpl_parse(q); break;
4541
      case Q_ENABLE_QUERY_LOG:   disable_query_log=0; break;
unknown's avatar
unknown committed
4542
      case Q_DISABLE_QUERY_LOG:  disable_query_log=1; break;
unknown's avatar
unknown committed
4543 4544
      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
4545 4546
      case Q_ENABLE_RESULT_LOG:  disable_result_log=0; break;
      case Q_DISABLE_RESULT_LOG: disable_result_log=1; break;
4547 4548
      case Q_ENABLE_WARNINGS:    disable_warnings=0; break;
      case Q_DISABLE_WARNINGS:   disable_warnings=1; break;
4549 4550
      case Q_ENABLE_PS_WARNINGS:    disable_ps_warnings=0; break;
      case Q_DISABLE_PS_WARNINGS:   disable_ps_warnings=1; break;
4551 4552
      case Q_ENABLE_INFO:        disable_info=0; break;
      case Q_DISABLE_INFO:       disable_info=1; break;
4553
      case Q_ENABLE_METADATA:    display_metadata=1; break;
4554
      case Q_DISABLE_METADATA:   display_metadata=0; break;
4555
      case Q_SOURCE: do_source(q); break;
4556 4557
      case Q_SLEEP: do_sleep(q, 0); break;
      case Q_REAL_SLEEP: do_sleep(q, 1); break;
4558
      case Q_WAIT_FOR_SLAVE_TO_STOP: do_wait_for_slave_to_stop(q); break;
4559
      case Q_REQUIRE_MANAGER: do_require_manager(q); break;
4560
#ifndef EMBEDDED_LIBRARY
4561 4562
      case Q_SERVER_START: do_server_start(q); break;
      case Q_SERVER_STOP: do_server_stop(q); break;
4563
#endif
4564 4565
      case Q_INC: do_modify_var(q, "inc", DO_INC); break;
      case Q_DEC: do_modify_var(q, "dec", DO_DEC); break;
4566
      case Q_ECHO: do_echo(q); query_executed= 1; break;
4567
      case Q_SYSTEM: do_system(q); break;
4568 4569 4570
      case Q_DELIMITER:
	strmake(delimiter, q->first_argument, sizeof(delimiter) - 1);
	delimiter_length= strlen(delimiter);
4571
        q->last_argument= q->first_argument+delimiter_length;
4572
	break;
4573 4574 4575 4576 4577 4578
      case Q_DISPLAY_VERTICAL_RESULTS:
        display_result_vertically= TRUE;
        break;
      case Q_DISPLAY_HORIZONTAL_RESULTS:
	display_result_vertically= FALSE;
        break;
4579
      case Q_LET: do_let(q); break;
4580
      case Q_EVAL_RESULT:
4581
        eval_result = 1; break;
4582
      case Q_EVAL:
4583
	if (q->query == q->query_buf)
unknown's avatar
unknown committed
4584
        {
unknown's avatar
unknown committed
4585
	  q->query= q->first_argument;
unknown's avatar
unknown committed
4586 4587
          q->first_word_len= 0;
        }
4588
	/* fall through */
4589
      case Q_QUERY_VERTICAL:
4590
      case Q_QUERY_HORIZONTAL:
4591 4592 4593 4594 4595
      {
	my_bool old_display_result_vertically= display_result_vertically;
	/* fix up query pointer if this is * first iteration for this line */
	if (q->query == q->query_buf)
	  q->query += q->first_word_len + 1;
4596
	display_result_vertically= (q->type==Q_QUERY_VERTICAL);
4597 4598 4599 4600 4601 4602
	if (save_file[0])
	{
	  strmov(q->record_file,save_file);
	  q->require_file=require_file;
	  save_file[0]=0;
	}
4603
	run_query(&cur_con->mysql, q, QUERY_REAP|QUERY_SEND);
4604
	display_result_vertically= old_display_result_vertically;
4605
        q->last_argument= q->end;
4606
        query_executed= 1;
4607 4608
	break;
      }
4609
      case Q_QUERY:
4610
      case Q_REAP:
4611
      {
4612 4613 4614 4615 4616
	/*
	  We read the result always regardless of the mode for both full
	  query and read-result only (reap)
	*/
	int flags = QUERY_REAP;
4617
	if (q->type != Q_REAP) /* for a full query, enable the send stage */
unknown's avatar
unknown committed
4618
	  flags |= QUERY_SEND;
unknown's avatar
unknown committed
4619 4620 4621 4622 4623
	if (q_send_flag)
	{
	  flags= QUERY_SEND;
	  q_send_flag=0;
	}
4624
	if (save_file[0])
4625
	{
4626 4627 4628
	  strmov(q->record_file,save_file);
	  q->require_file=require_file;
	  save_file[0]=0;
4629
	}
4630
	run_query(&cur_con->mysql, q, flags);
4631
	query_executed= 1;
4632
        q->last_argument= q->end;
unknown's avatar
unknown committed
4633
	break;
4634
      }
unknown's avatar
unknown committed
4635
      case Q_SEND:
unknown's avatar
unknown committed
4636 4637 4638 4639 4640 4641 4642 4643
	if (!q->query[q->first_word_len])
	{
	  /* This happens when we use 'send' on it's own line */
	  q_send_flag=1;
	  break;
	}
	/* fix up query pointer if this is * first iteration for this line */
	if (q->query == q->query_buf)
4644
	  q->query += q->first_word_len;
4645 4646 4647 4648 4649
	/*
	  run_query() can execute a query partially, depending on the flags
	  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.
4650
	 */
4651
	run_query(&cur_con->mysql, q, QUERY_SEND);
4652
	query_executed= 1;
4653
        q->last_argument= q->end;
unknown's avatar
unknown committed
4654
	break;
4655 4656 4657 4658
      case Q_RESULT:
	get_file_name(save_file,q);
	require_file=0;
	break;
unknown's avatar
unknown committed
4659
      case Q_ERROR:
4660
        global_expected_errors=get_errcodes(global_expected_errno,q);
unknown's avatar
unknown committed
4661
	break;
4662 4663 4664 4665
      case Q_REQUIRE:
	get_file_name(save_file,q);
	require_file=1;
	break;
unknown's avatar
unknown committed
4666 4667 4668
      case Q_REPLACE:
	get_replace(q);
	break;
4669 4670 4671
      case Q_REPLACE_COLUMN:
	get_replace_column(q);
	break;
4672 4673
      case Q_SAVE_MASTER_POS: do_save_master_pos(); break;
      case Q_SYNC_WITH_MASTER: do_sync_with_master(q); break;
4674 4675 4676 4677
      case Q_SYNC_SLAVE_WITH_MASTER:
      {
	do_save_master_pos();
	if (*q->first_argument)
4678
	  select_connection(q);
4679
	else
4680 4681
	  select_connection_name("slave");
	do_sync_with_master2(0);
4682 4683
	break;
      }
4684
      case Q_COMMENT:				/* Ignore row */
4685
      case Q_COMMENT_WITH_COMMAND:
4686
        q->last_argument= q->end;
4687
	break;
4688 4689 4690
      case Q_PING:
	(void) mysql_ping(&cur_con->mysql);
	break;
4691
      case Q_EXEC:
unknown's avatar
unknown committed
4692
	do_exec(q);
4693
	query_executed= 1;
4694
	break;
unknown's avatar
unknown committed
4695 4696 4697 4698 4699 4700 4701 4702 4703
      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;
4704
      case Q_CHARACTER_SET:
unknown's avatar
unknown committed
4705 4706
	set_charset(q);
	break;
4707 4708 4709 4710 4711 4712
      case Q_DISABLE_PS_PROTOCOL:
        ps_protocol_enabled= 0;
        break;
      case Q_ENABLE_PS_PROTOCOL:
        ps_protocol_enabled= ps_protocol;
        break;
4713
      case Q_DISABLE_RECONNECT:
4714 4715 4716
      {
        my_bool reconnect= 0;
        mysql_options(&cur_con->mysql, MYSQL_OPT_RECONNECT, (char *)&reconnect);
4717
        break;
4718
      }
4719
      case Q_ENABLE_RECONNECT:
4720 4721 4722
      {
        my_bool reconnect= 1;
        mysql_options(&cur_con->mysql, MYSQL_OPT_RECONNECT, (char *)&reconnect);
4723
        break;
4724
      }
4725 4726 4727 4728 4729 4730 4731 4732 4733 4734 4735
      case Q_DISABLE_PARSING:
        parsing_disabled++;
        break;
      case Q_ENABLE_PARSING:
        /*
          Ensure we don't get parsing_disabled < 0 as this would accidently
          disable code we don't want to have disabled
        */
        if (parsing_disabled > 0)
          parsing_disabled--;
        break;
4736

unknown's avatar
unknown committed
4737 4738 4739
      case Q_EXIT:
        abort_flag= 1;
        break;
4740 4741 4742 4743

      default:
        processed= 0;
        break;
4744 4745
      }
    }
4746

4747 4748
    if (!processed)
    {
4749
      current_line_inc= 0;
unknown's avatar
unknown committed
4750
      switch (q->type) {
unknown's avatar
unknown committed
4751 4752
      case Q_WHILE: do_block(cmd_while, q); break;
      case Q_IF: do_block(cmd_if, q); break;
4753 4754 4755
      case Q_END_BLOCK: do_done(q); break;
      default: current_line_inc = 1; break;
      }
unknown's avatar
unknown committed
4756
    }
4757 4758
    else
      check_eol_junk(q->last_argument);
unknown's avatar
unknown committed
4759

4760 4761 4762 4763 4764 4765 4766 4767 4768
    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
4769

4770 4771 4772
    parser.current_line += current_line_inc;
  }

4773 4774
  start_lineno= 0;

4775
  /*
unknown's avatar
WL#2930  
unknown committed
4776
    The whole test has been executed _sucessfully_
4777 4778
    Time to compare result or save it to record file
    The entire output from test is now kept in ds_res
unknown's avatar
WL#2930  
unknown committed
4779
  */
4780
  if (ds_res.length)
4781
  {
4782 4783
    if (result_file)
    {
4784 4785 4786 4787 4788
      if (record)
      {
	/* Dump the output from test to result file */
	str_to_file(result_file, ds_res.str, ds_res.length);
      }
4789
      else
4790
      {
unknown's avatar
WL#2930  
unknown committed
4791 4792 4793 4794
	/* Check that the output from test is equal to result file
	   - detect missing result file
	   - detect zero size result file
	 */
4795 4796
	check_result(&ds_res, result_file, 0);
      }
4797
    }
4798
    else
4799
    {
unknown's avatar
WL#2930  
unknown committed
4800
      /* No result_file specified to compare with, print to stdout */
4801 4802
      printf("%s", ds_res.str);
    }
4803
  }
unknown's avatar
unknown committed
4804
  else
unknown's avatar
unknown committed
4805
  {
unknown's avatar
WL#2930  
unknown committed
4806
    die("The test didn't produce any output");
unknown's avatar
unknown committed
4807
  }
4808

unknown's avatar
WL#2930  
unknown committed
4809
  if (!query_executed && result_file && my_stat(result_file, &res_info, 0))
unknown's avatar
unknown committed
4810
  {
unknown's avatar
WL#2930  
unknown committed
4811 4812 4813 4814 4815 4816 4817 4818
    /*
      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.
      Note that we don't care, if my_stat() fails. For example for
      non-existing or non-readable file we assume it's fine to have
      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
4819
  }
4820

unknown's avatar
WL#2930  
unknown committed
4821

4822
  dynstr_free(&ds_res);
4823

unknown's avatar
unknown committed
4824 4825
  if (!got_end_timer)
    timer_output();				/* No end_timer cmd, end it */
4826
  free_used_memory();
4827
  my_end(MY_CHECK_ERROR);
4828 4829 4830 4831 4832 4833

  /* 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
4834
}
unknown's avatar
unknown committed
4835

4836

4837 4838 4839 4840 4841 4842
/*
  Read arguments for embedded server and put them into
  embedded_server_args_count and embedded_server_args[]
*/


unknown's avatar
unknown committed
4843
static int read_server_arguments(const char *name)
4844 4845 4846 4847 4848 4849 4850 4851 4852 4853 4854 4855 4856 4857 4858 4859
{
  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
4860
  if (!(file=my_fopen(buff, O_RDONLY | FILE_BINARY, MYF(MY_WME))))
4861 4862 4863 4864 4865 4866 4867 4868 4869 4870 4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881
    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
4882

unknown's avatar
unknown committed
4883 4884 4885 4886 4887 4888 4889 4890 4891 4892 4893 4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907 4908 4909
/****************************************************************************\
 *
 *  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)
  {
4910
    char buf[32], *end;
unknown's avatar
unknown committed
4911
    ulonglong timer= timer_now() - timer_start;
4912 4913
    end= longlong2str(timer, buf, 10);
    str_to_file(timer_file,buf, (int) (end-buf));
unknown's avatar
unknown committed
4914 4915 4916 4917 4918 4919 4920 4921
  }
}

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

unknown's avatar
unknown committed
4922 4923 4924 4925 4926 4927 4928 4929 4930 4931 4932 4933 4934
/****************************************************************************
* 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 {
4935
  bool	 found;
unknown's avatar
unknown committed
4936 4937 4938 4939 4940 4941 4942 4943 4944 4945 4946 4947 4948 4949 4950 4951 4952 4953 4954 4955 4956 4957 4958 4959 4960 4961 4962 4963 4964 4965 4966 4967 4968 4969 4970 4971 4972 4973 4974 4975 4976 4977 4978 4979 4980 4981 4982 4983 4984 4985 4986 4987 4988 4989 4990 4991 4992 4993 4994 4995 4996 4997 4998 4999 5000 5001 5002 5003 5004 5005 5006 5007 5008 5009 5010 5011 5012 5013 5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033 5034 5035 5036 5037 5038 5039 5040
  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 {
5041 5042
  uint	*bits;				/* Pointer to used sets */
  short next[LAST_CHAR_CODE];		/* Pointer to next sets */
unknown's avatar
unknown committed
5043 5044
  uint	found_len;			/* Best match to date */
  int	found_offset;
5045 5046
  uint	table_offset;
  uint	size_of_bits;			/* For convinience */
unknown's avatar
unknown committed
5047 5048 5049 5050 5051 5052 5053 5054 5055 5056 5057 5058 5059 5060 5061 5062 5063 5064 5065 5066 5067 5068 5069 5070 5071 5072 5073 5074
} 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);
5075 5076
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
5077 5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090 5091 5092 5093 5094 5095 5096 5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115 5116 5117 5118 5119 5120 5121 5122 5123 5124 5125 5126 5127 5128 5129 5130 5131 5132 5133 5134 5135 5136 5137 5138 5139 5140 5141 5142 5143 5144 5145 5146 5147 5148 5149 5150 5151 5152
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] == '^')
    {
5153
      internal_set_bit(start_states,states+1);
unknown's avatar
unknown committed
5154 5155 5156 5157 5158 5159 5160 5161
      if (!from[i][2])
      {
	start_states->table_offset=i;
	start_states->found_offset=1;
      }
    }
    else if (from[i][0] == '\\' && from[i][1] == '$')
    {
5162 5163
      internal_set_bit(start_states,states);
      internal_set_bit(word_states,states);
unknown's avatar
unknown committed
5164 5165 5166 5167 5168 5169 5170 5171
      if (!from[i][2] && start_states->table_offset == (uint) ~0)
      {
	start_states->table_offset=i;
	start_states->found_offset=0;
      }
    }
    else
    {
5172
      internal_set_bit(word_states,states);
unknown's avatar
unknown committed
5173
      if (from[i][0] == '\\' && (from[i][1] == 'b' && from[i][2]))
5174
	internal_set_bit(start_states,states+1);
unknown's avatar
unknown committed
5175
      else
5176
	internal_set_bit(start_states,states);
unknown's avatar
unknown committed
5177 5178 5179 5180 5181 5182 5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198 5199 5200 5201 5202 5203 5204 5205 5206 5207 5208 5209 5210 5211 5212 5213 5214 5215 5216 5217 5218 5219 5220 5221 5222 5223 5224 5225 5226 5227 5228 5229 5230 5231 5232 5233 5234 5235 5236 5237 5238 5239 5240 5241 5242 5243 5244 5245 5246 5247 5248 5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262 5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281
    }
    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)
5282
	      internal_set_bit(new_set,i+1);		/* To next set */
unknown's avatar
unknown committed
5283
	    else
5284
	      internal_set_bit(new_set,i);
unknown's avatar
unknown committed
5285 5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299 5300
	  }
	}
	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)))
5301
	      internal_clear_bit(new_set,i);
unknown's avatar
unknown committed
5302 5303 5304 5305 5306 5307 5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323 5324 5325 5326 5327 5328 5329 5330 5331 5332 5333 5334 5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346 5347 5348 5349 5350 5351 5352 5353 5354 5355 5356 5357 5358 5359 5360 5361 5362 5363 5364 5365 5366 5367 5368 5369 5370 5371 5372 5373 5374 5375 5376 5377 5378 5379 5380 5381 5382 5383 5384 5385 5386 5387 5388 5389 5390 5391 5392 5393 5394 5395 5396 5397 5398 5399 5400 5401 5402 5403 5404 5405 5406 5407 5408 5409 5410 5411 5412 5413 5414 5415 5416 5417 5418 5419 5420 5421 5422 5423 5424 5425 5426 5427 5428 5429 5430 5431 5432 5433 5434 5435 5436 5437 5438 5439 5440 5441 5442 5443 5444 5445 5446 5447 5448 5449
	    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;
}

5450
static void internal_set_bit(REP_SET *set, uint bit)
unknown's avatar
unknown committed
5451 5452 5453 5454 5455
{
  set->bits[bit / WORD_BIT] |= 1 << (bit % WORD_BIT);
  return;
}

5456
static void internal_clear_bit(REP_SET *set, uint bit)
unknown's avatar
unknown committed
5457 5458 5459 5460 5461 5462 5463 5464 5465 5466 5467 5468 5469 5470 5471 5472 5473 5474 5475 5476 5477 5478 5479 5480 5481 5482 5483 5484 5485 5486 5487 5488 5489 5490 5491 5492 5493 5494 5495 5496 5497 5498 5499 5500 5501 5502 5503 5504 5505 5506 5507 5508 5509 5510 5511 5512 5513 5514 5515 5516 5517 5518 5519 5520 5521 5522 5523 5524 5525 5526 5527 5528 5529 5530 5531 5532 5533 5534 5535 5536 5537 5538 5539 5540 5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551 5552 5553 5554 5555 5556 5557 5558 5559 5560 5561 5562 5563 5564 5565 5566 5567 5568 5569 5570 5571 5572 5573 5574
{
  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
	   free given set, else put in given set in sets and return it's
	   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.
	   Pos returned is -offset-2 in found_set_structure because it's is
	   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;
}


5575 5576 5577
/* 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
5578 5579 5580
{
  reg1 REPLACE *rep_pos;
  reg2 REPLACE_STRING *rep_str;
5581 5582
  const char *start, *from;
  DBUG_ENTER("replace_strings_append");
unknown's avatar
unknown committed
5583

5584
  start= from= str;
unknown's avatar
unknown committed
5585
  rep_pos=rep+1;
5586
  for (;;)
unknown's avatar
unknown committed
5587
  {
5588 5589
    /* Loop through states */
    DBUG_PRINT("info", ("Looping through states"));
unknown's avatar
unknown committed
5590
    while (!rep_pos->found)
5591 5592 5593
      rep_pos= rep_pos->next[(uchar) *from++];

    /* Does this state contain a string to be replaced */
unknown's avatar
unknown committed
5594 5595
    if (!(rep_str = ((REPLACE_STRING*) rep_pos))->replace_string)
    {
5596 5597 5598 5599
      /* 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
5600
    }
5601 5602 5603 5604 5605 5606 5607 5608 5609 5610 5611 5612 5613

    /* 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
5614
    if (!*(from-=rep_str->from_offset) && rep_pos->found != 2)
5615 5616 5617 5618 5619 5620 5621
    {
      /* 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
5622 5623 5624 5625 5626
    rep_pos=rep;
  }
}


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
/****************************************************************************
 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)
5661
    die("Missing argument in %s", q->query);
5662 5663 5664 5665 5666 5667 5668 5669 5670 5671

  /* 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)
5672
      die("Wrong column number to replace_column in '%s'", q->query);
5673
    if (!*from)
5674
      die("Wrong number of arguments to replace_column in '%s'", q->query);
5675 5676 5677 5678 5679 5680
    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));
5681
  q->last_argument= q->end;
5682
}
unknown's avatar
unknown committed
5683 5684


5685 5686