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 1441 1442
  init_dynamic_string(&ds_echo, "", 256, 256);
  do_eval(&ds_echo, command->first_argument);
  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
  return 0;
2472 2473
}

unknown's avatar
unknown committed
2474

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

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

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


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

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

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

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

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


2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545
/*
  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.

*/

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

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

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

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

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

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

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

unknown's avatar
unknown committed
2677
    }
2678 2679

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

	*p++ = c;

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

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

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

2737 2738
static char read_query_buf[MAX_QUERY];

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

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

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

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

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

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

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

2811 2812 2813

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

unknown's avatar
unknown committed
2902 2903 2904

#include <help_start.h>

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

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

2925 2926 2927 2928 2929

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


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

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

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

  return 0;
}


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

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

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

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

unknown's avatar
unknown committed
3073

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

3080

3081
#ifdef __WIN__
3082

3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094
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()
3095
{
3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107
  /* 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++)
3108
  {
3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127
    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++;
    }
3128
  }
3129
  DBUG_VOID_RETURN;
3130 3131
}

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

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

3146 3147 3148 3149 3150 3151 3152 3153 3154 3155
  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 /
*/
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 3198
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);
}
3199

3200 3201

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

3207

3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225
/*
  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__
3226 3227
  else if ((field->type == MYSQL_TYPE_DOUBLE ||
            field->type == MYSQL_TYPE_FLOAT ) &&
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 3258
           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);
  }
}


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

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

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

3284

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

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

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

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

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

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

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

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

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

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


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

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

3364 3365 3366
  for (field_end= field+num_fields ;
       field < field_end ;
       field++)
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 3396
    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);
3397

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

unknown's avatar
unknown committed
3410

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

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


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

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

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

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

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

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

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

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

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

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

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

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


unknown's avatar
WL#2930  
unknown committed
3487

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

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

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

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

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

3542
    }
3543

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

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

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

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

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

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

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

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

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

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

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

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

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


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

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

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

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

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

3654
  DBUG_ENTER("handle_error");
3655 3656 3657

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

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

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

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

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

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

3713
  DBUG_VOID_RETURN;
3714 3715 3716 3717
}


/*
3718
  Handle absence of errors after execution
3719 3720

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

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

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

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

3747
  DBUG_VOID_RETURN;
3748 3749
}

3750 3751

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

3754 3755 3756 3757 3758 3759 3760
  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
3761

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  }

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

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

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

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



3987 3988
/*
  Run query
3989

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

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

3999
*/
4000

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

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

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

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

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

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

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

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

4097
    dynstr_free(&query_str);
4098 4099 4100

  }

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

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

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

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

  /*
4142
    Find out how to run this query
4143

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

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

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

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

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

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

unknown's avatar
unknown committed
4183
    } else {
4184

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

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


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

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

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

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

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


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

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

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

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

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

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

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

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

unknown's avatar
unknown committed
4343

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

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

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

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

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

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


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

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

unknown's avatar
unknown committed
4405

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

4423

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

4771 4772 4773
    parser.current_line += current_line_inc;
  }

4774 4775
  start_lineno= 0;

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

unknown's avatar
WL#2930  
unknown committed
4810
  if (!query_executed && result_file && my_stat(result_file, &res_info, 0))
unknown's avatar
unknown committed
4811
  {
unknown's avatar
WL#2930  
unknown committed
4812 4813 4814 4815 4816 4817 4818 4819
    /*
      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
4820
  }
4821

unknown's avatar
WL#2930  
unknown committed
4822

4823
  dynstr_free(&ds_res);
4824

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

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

4837

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


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

unknown's avatar
unknown committed
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 4910
/****************************************************************************\
 *
 *  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)
  {
4911
    char buf[32], *end;
unknown's avatar
unknown committed
4912
    ulonglong timer= timer_now() - timer_start;
4913 4914
    end= longlong2str(timer, buf, 10);
    str_to_file(timer_file,buf, (int) (end-buf));
unknown's avatar
unknown committed
4915 4916 4917 4918 4919 4920 4921 4922
  }
}

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

unknown's avatar
unknown committed
4923 4924 4925 4926 4927 4928 4929 4930 4931 4932 4933 4934 4935
/****************************************************************************
* 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 {
4936
  bool	 found;
unknown's avatar
unknown committed
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 5041
  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 {
5042 5043
  uint	*bits;				/* Pointer to used sets */
  short next[LAST_CHAR_CODE];		/* Pointer to next sets */
unknown's avatar
unknown committed
5044 5045
  uint	found_len;			/* Best match to date */
  int	found_offset;
5046 5047
  uint	table_offset;
  uint	size_of_bits;			/* For convinience */
unknown's avatar
unknown committed
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 5075
} 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);
5076 5077
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
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 5153
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] == '^')
    {
5154
      internal_set_bit(start_states,states+1);
unknown's avatar
unknown committed
5155 5156 5157 5158 5159 5160 5161 5162
      if (!from[i][2])
      {
	start_states->table_offset=i;
	start_states->found_offset=1;
      }
    }
    else if (from[i][0] == '\\' && from[i][1] == '$')
    {
5163 5164
      internal_set_bit(start_states,states);
      internal_set_bit(word_states,states);
unknown's avatar
unknown committed
5165 5166 5167 5168 5169 5170 5171 5172
      if (!from[i][2] && start_states->table_offset == (uint) ~0)
      {
	start_states->table_offset=i;
	start_states->found_offset=0;
      }
    }
    else
    {
5173
      internal_set_bit(word_states,states);
unknown's avatar
unknown committed
5174
      if (from[i][0] == '\\' && (from[i][1] == 'b' && from[i][2]))
5175
	internal_set_bit(start_states,states+1);
unknown's avatar
unknown committed
5176
      else
5177
	internal_set_bit(start_states,states);
unknown's avatar
unknown committed
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 5282
    }
    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)
5283
	      internal_set_bit(new_set,i+1);		/* To next set */
unknown's avatar
unknown committed
5284
	    else
5285
	      internal_set_bit(new_set,i);
unknown's avatar
unknown committed
5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299 5300 5301
	  }
	}
	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)))
5302
	      internal_clear_bit(new_set,i);
unknown's avatar
unknown committed
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 5450
	    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;
}

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

5457
static void internal_clear_bit(REP_SET *set, uint bit)
unknown's avatar
unknown committed
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 5575
{
  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;
}


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

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

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

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


5628 5629 5630 5631 5632 5633 5634 5635 5636 5637 5638 5639 5640 5641 5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661
/****************************************************************************
 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)
5662
    die("Missing argument in %s", q->query);
5663 5664 5665 5666 5667 5668 5669 5670 5671 5672

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


5686 5687