mysqldump.c 48 KB
Newer Older
unknown's avatar
unknown committed
1
/* Copyright (C) 2000 MySQL AB
unknown's avatar
unknown committed
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.
unknown's avatar
unknown committed
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.
unknown's avatar
unknown committed
12

unknown's avatar
unknown committed
13 14 15 16 17 18 19 20
   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 */

/* mysqldump.c  - Dump a tables contents and format to an ASCII file
**
** The author's original notes follow :-
**
unknown's avatar
unknown committed
21 22 23 24 25 26 27 28 29 30 31 32 33
** AUTHOR: Igor Romanenko (igor@frog.kiev.ua)
** DATE:   December 3, 1994
** WARRANTY: None, expressed, impressed, implied
**	    or other
** STATUS: Public domain
** Adapted and optimized for MySQL by
** Michael Widenius, Sinisa Milivojevic, Jani Tolonen
** -w --where added 9/10/98 by Jim Faucette
** slave code by David Saez Padros <david@ols.es>
** master/autocommit code by Brian Aker <brian@tangent.org>
** SSL by
** Andrei Errapart <andreie@no.spam.ee>
** Tnu Samuel  <tonu@please.do.not.remove.this.spam.ee>
unknown's avatar
unknown committed
34 35
** XML by Gary Huntress <ghuntress@mediaone.net> 10/10/01, cleaned up
** and adapted to mysqldump 05/11/01 by Jani Tolonen
36
** Added --single-transaction option 06/06/2002 by Peter Zaitsev
unknown's avatar
unknown committed
37
*/
unknown's avatar
unknown committed
38

39
#define DUMP_VERSION "9.10"
unknown's avatar
unknown committed
40

unknown's avatar
unknown committed
41
#include <my_global.h>
unknown's avatar
unknown committed
42 43 44 45
#include <my_sys.h>
#include <m_string.h>
#include <m_ctype.h>

46
#include "client_priv.h"
unknown's avatar
unknown committed
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
#include "mysql.h"
#include "mysql_version.h"
#include "mysqld_error.h"

/* Exit codes */

#define EX_USAGE 1
#define EX_MYSQLERR 2
#define EX_CONSCHECK 3
#define EX_EOM 4

/* index into 'show fields from table' */

#define SHOW_FIELDNAME  0
#define SHOW_TYPE  1
#define SHOW_NULL  2
#define SHOW_DEFAULT  4
#define SHOW_EXTRA  5
#define QUOTE_CHAR	'`'

67 68 69
/* Size of buffer for dump's select query */
#define QUERY_LENGTH 1536

unknown's avatar
unknown committed
70 71 72 73 74 75 76 77
static char *add_load_option(char *ptr, const char *object,
			     const char *statement);

static char *field_escape(char *to,const char *from,uint length);
static my_bool  verbose=0,tFlag=0,cFlag=0,dFlag=0,quick=0, extended_insert = 0,
		lock_tables=0,ignore_errors=0,flush_logs=0,replace=0,
		ignore=0,opt_drop=0,opt_keywords=0,opt_lock=0,opt_compress=0,
                opt_delayed=0,create_options=0,opt_quoted=0,opt_databases=0,
78
	        opt_alldbs=0,opt_create_db=0,opt_first_slave=0,
79
                opt_autocommit=0,opt_master_data,opt_disable_keys=0,opt_xml=0,
unknown's avatar
unknown committed
80
	        opt_delete_master_logs=0, tty_password=0,
81
		opt_single_transaction=0, opt_comments= 0;
unknown's avatar
unknown committed
82
static MYSQL  mysql_connection,*sock=0;
83
static char  insert_pat[12 * 1024],*opt_password=0,*current_user=0,
unknown's avatar
unknown committed
84 85 86 87 88 89 90 91
             *current_host=0,*path=0,*fields_terminated=0,
             *lines_terminated=0, *enclosed=0, *opt_enclosed=0, *escaped=0,
             *where=0, *default_charset;
static uint     opt_mysql_port=0;
static my_string opt_mysql_unix_port=0;
static int   first_error=0;
extern ulong net_buffer_length;
static DYNAMIC_STRING extended_row;
92
#include <sslopt-vars.h>
unknown's avatar
unknown committed
93
FILE  *md_result_file;
unknown's avatar
unknown committed
94

95
static struct my_option my_long_options[] =
unknown's avatar
unknown committed
96
{
97 98 99 100
  {"all-databases", 'A',
   "Dump all the databases. This will be same as --databases with all databases selected.",
   (gptr*) &opt_alldbs, (gptr*) &opt_alldbs, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,
   0, 0},
101 102 103
  {"all", 'a', "Include all MySQL specific create options.",
   (gptr*) &create_options, (gptr*) &create_options, 0, GET_BOOL, NO_ARG, 0,
   0, 0, 0, 0, 0},
104 105 106 107 108 109 110 111 112 113 114 115
  {"add-drop-table", OPT_DROP, "Add a 'drop table' before each create.",
   (gptr*) &opt_drop, (gptr*) &opt_drop, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0,
   0},
  {"add-locks", OPT_LOCKS, "Add locks around insert statements.",
   (gptr*) &opt_lock, (gptr*) &opt_lock, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0,
   0},
  {"allow-keywords", OPT_KEYWORDS,
   "Allow creation of column names that are keywords.", (gptr*) &opt_keywords,
   (gptr*) &opt_keywords, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"character-sets-dir", OPT_CHARSETS_DIR,
   "Directory where character sets are", (gptr*) &charsets_dir,
   (gptr*) &charsets_dir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
116 117
  {"complete-insert", 'c', "Use complete insert statements.", (gptr*) &cFlag,
   (gptr*) &cFlag, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
  {"compress", 'C', "Use compression in server/client protocol.",
   (gptr*) &opt_compress, (gptr*) &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0,
   0, 0, 0},
  {"databases", 'B',
   "To dump several databases. Note the difference in usage; In this case no tables are given. All name arguments are regarded as databasenames. 'USE db_name;' will be included in the output.",
   (gptr*) &opt_databases, (gptr*) &opt_databases, 0, GET_BOOL, NO_ARG, 0, 0,
   0, 0, 0, 0},
  {"debug", '#', "Output debug log. Often this is 'd:t:o,filename'.",
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
  {"default-character-set", OPT_DEFAULT_CHARSET,
   "Set the default character set.", (gptr*) &default_charset,
   (gptr*) &default_charset, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"delayed-insert", OPT_DELAYED, "Insert rows with INSERT DELAYED.",
   (gptr*) &opt_delayed, (gptr*) &opt_delayed, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,
   0, 0},
unknown's avatar
unknown committed
133 134 135
  {"delete-master-logs", OPT_DELETE_MASTER_LOGS,
   "Delete logs on master after backup. This will automagically enable --first-slave.",
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156
  {"disable-keys", 'K',
   "'/*!40000 ALTER TABLE tb_name DISABLE KEYS */; and '/*!40000 ALTER TABLE tb_name ENABLE KEYS */; will be put in the output.", (gptr*) &opt_disable_keys,
   (gptr*) &opt_disable_keys, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"extended-insert", 'e',
   "Allows utilization of the new, much faster INSERT syntax.",
   (gptr*) &extended_insert, (gptr*) &extended_insert, 0, GET_BOOL, NO_ARG,
   0, 0, 0, 0, 0, 0},
  {"fields-terminated-by", OPT_FTB,
   "Fields in the textfile are terminated by ...", (gptr*) &fields_terminated,
   (gptr*) &fields_terminated, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"fields-enclosed-by", OPT_ENC,
   "Fields in the importfile are enclosed by ...", (gptr*) &enclosed,
   (gptr*) &enclosed, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0 ,0, 0},
  {"fields-optionally-enclosed-by", OPT_O_ENC,
   "Fields in the i.file are opt. enclosed by ...", (gptr*) &opt_enclosed,
   (gptr*) &opt_enclosed, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0 ,0, 0},
  {"fields-escaped-by", OPT_ESC, "Fields in the i.file are escaped by ...",
   (gptr*) &escaped, (gptr*) &escaped, 0, GET_STR, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"first-slave", 'x', "Locks all tables across all databases.",
   (gptr*) &opt_first_slave, (gptr*) &opt_first_slave, 0, GET_BOOL, NO_ARG,
   0, 0, 0, 0, 0, 0},
157
  {"flush-logs", 'F', "Flush logs file in server before starting dump. Note that if you use this option in combination with the --all-databases (or -A) option, the logs will be flushed for each database dumped.",
158 159 160 161 162 163 164 165
   (gptr*) &flush_logs, (gptr*) &flush_logs, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,
   0, 0},
  {"force", 'f', "Continue even if we get an sql-error.",
   (gptr*) &ignore_errors, (gptr*) &ignore_errors, 0, GET_BOOL, NO_ARG,
   0, 0, 0, 0, 0, 0},
  {"help", '?', "Display this help message and exit.", 0, 0, 0, GET_NO_ARG,
   NO_ARG, 0, 0, 0, 0, 0, 0},
  {"host", 'h', "Connect to host.", (gptr*) &current_host,
166
   (gptr*) &current_host, 0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
167 168 169 170 171 172 173 174 175 176 177 178
  {"lines-terminated-by", OPT_LTB, "Lines in the i.file are terminated by ...",
   (gptr*) &lines_terminated, (gptr*) &lines_terminated, 0, GET_STR,
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"lock-tables", 'l', "Lock all tables for read.", (gptr*) &lock_tables,
   (gptr*) &lock_tables, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"master-data", OPT_MASTER_DATA,
   "This will cause the master position and filename to be appended to your output. This will automagically enable --first-slave.",
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"no-autocommit", OPT_AUTOCOMMIT,
   "Wrap tables with autocommit/commit statements.",
   (gptr*) &opt_autocommit, (gptr*) &opt_autocommit, 0, GET_BOOL, NO_ARG,
   0, 0, 0, 0, 0, 0},
179 180 181
  {"single-transaction", OPT_TRANSACTION,
   "Dump all tables in single transaction to get consistent snapshot. Mutually exclusive with --lock-tables.",
   (gptr*) &opt_single_transaction, (gptr*) &opt_single_transaction, 0, GET_BOOL, NO_ARG,
182
   0, 0, 0, 0, 0, 0},
183 184 185 186 187
  {"no-create-db", 'n',
   "'CREATE DATABASE /*!32312 IF NOT EXISTS*/ db_name;' will not be put in the output. The above line will be added otherwise, if --databases or --all-databases option was given.}",
   (gptr*) &opt_create_db, (gptr*) &opt_create_db, 0, GET_BOOL, NO_ARG, 0, 0,
   0, 0, 0, 0},
  {"no-create-info", 't', "Don't write table creation info.",
188 189 190
   (gptr*) &tFlag, (gptr*) &tFlag, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"no-data", 'd', "No row information.", (gptr*) &dFlag, (gptr*) &dFlag, 0,
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
191
  {"set-variable", 'O',
unknown's avatar
unknown committed
192
   "Change the value of a variable. Please note that this option is deprecated; you can set variables directly with --variable-name=value.",
193 194 195 196 197 198 199
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"opt", OPT_OPTIMIZE,
   "Same as --add-drop-table --add-locks --all --quick --extended-insert --lock-tables --disable-keys",
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"password", 'p',
   "Password to use when connecting to server. If password is not given it's solicited on the tty.",
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
200
#ifdef __WIN__
201 202
  {"pipe", 'W', "Use named pipes to connect to server", 0, 0, 0, GET_NO_ARG,
   NO_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
203
#endif
204 205 206
  {"port", 'P', "Port number to use for connection.", (gptr*) &opt_mysql_port,
   (gptr*) &opt_mysql_port, 0, GET_UINT, REQUIRED_ARG, MYSQL_PORT, 0, 0, 0, 0,
   0},
207 208 209 210 211 212
  {"quick", 'q', "Don't buffer query, dump directly to stdout.",
   (gptr*) &quick, (gptr*) &quick, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"quote-names",'Q', "Quote table and column names with a `",
   (gptr*) &opt_quoted, (gptr*) &opt_quoted, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,
   0, 0},
  {"result-file", 'r',
unknown's avatar
unknown committed
213
   "Direct output to a given file. This option should be used in MSDOS, because it prevents new line '\\n' from being converted to '\\r\\n' (carriage return + line feed).",
unknown's avatar
unknown committed
214
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
215 216 217
  {"socket", 'S', "Socket file to use for connection.",
   (gptr*) &opt_mysql_unix_port, (gptr*) &opt_mysql_unix_port, 0, GET_STR,
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
218
#include <sslopt-longopts.h>
219 220 221 222 223
  {"tab",'T',
   "Creates tab separated textfile for each table to given path. (creates .sql and .txt files). NOTE: This only works if mysqldump is run on the same machine as the mysqld daemon.",
   (gptr*) &path, (gptr*) &path, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"tables", OPT_TABLES, "Overrides option --databases (-B).",
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
224
#ifndef DONT_ALLOW_USER_CHANGE
225 226 227
  {"user", 'u', "User for login if not current user.",
   (gptr*) &current_user, (gptr*) &current_user, 0, GET_STR, REQUIRED_ARG,
   0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
228
#endif
229 230 231 232 233 234 235 236
  {"verbose", 'v', "Print info about the various stages.",
   (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},
  {"where", 'w', "Dump only selected records; QUOTES mandatory!",
   (gptr*) &where, (gptr*) &where, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"xml", 'X', "Dump a database as well formed XML.", 0, 0, 0, GET_NO_ARG,
   NO_ARG, 0, 0, 0, 0, 0, 0},
237
  {"max_allowed_packet", OPT_MAX_ALLOWED_PACKET, "",
238
    (gptr*) &max_allowed_packet, (gptr*) &max_allowed_packet, 0,
239 240
    GET_ULONG, REQUIRED_ARG, 24*1024*1024, 4096, 
   (longlong) 2L*1024L*1024L*1024L, MALLOC_OVERHEAD, 1024, 0},
241
  {"net_buffer_length", OPT_NET_BUFFER_LENGTH, "",
242
    (gptr*) &net_buffer_length, (gptr*) &net_buffer_length, 0,
243
    GET_ULONG, REQUIRED_ARG, 1024*1024L-1025, 4096, 16*1024L*1024L,
unknown's avatar
merge  
unknown committed
244
    MALLOC_OVERHEAD-1024, 1024, 0},
245 246 247
  {"comments", 'i', "Write additional information.",
   (gptr*) &opt_comments, (gptr*) &opt_comments, 0, GET_BOOL, NO_ARG,
   1, 0, 0, 0, 0, 0},
248
  {0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
unknown's avatar
unknown committed
249 250 251 252 253
};

static const char *load_default_groups[]= { "mysqldump","client",0 };

static void safe_exit(int error);
254
static void write_header(FILE *sql_file, char *db_name);
unknown's avatar
unknown committed
255 256 257 258 259 260 261 262
static void print_value(FILE *file, MYSQL_RES  *result, MYSQL_ROW row,
			const char *prefix,const char *name,
			int string_value);
static int dump_selected_tables(char *db, char **table_names, int tables);
static int dump_all_tables_in_db(char *db);
static int init_dumping(char *);
static int dump_databases(char **);
static int dump_all_databases();
unknown's avatar
unknown committed
263
static char *quote_name(const char *name, char *buff, my_bool force);
unknown's avatar
unknown committed
264
static void print_quoted_xml(FILE *output, char *fname, char *str, uint len);
265 266
static const char *check_if_ignore_table(const char *table_name);

unknown's avatar
unknown committed
267 268 269 270 271 272 273 274

static void print_version(void)
{
  printf("%s  Ver %s Distrib %s, for %s (%s)\n",my_progname,DUMP_VERSION,
   MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);
} /* print_version */


275 276 277 278 279 280 281 282
static void short_usage_sub(void)
{
  printf("Usage: %s [OPTIONS] database [tables]\n", my_progname);
  printf("OR     %s [OPTIONS] --databases [OPTIONS] DB1 [DB2 DB3...]\n",
	 my_progname);
  printf("OR     %s [OPTIONS] --all-databases [OPTIONS]\n", my_progname);
}

unknown's avatar
unknown committed
283 284 285 286 287 288
static void usage(void)
{
  print_version();
  puts("By Igor Romanenko, Monty, Jani & Sinisa");
  puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,\nand you are welcome to modify and redistribute it under the GPL license\n");
  puts("Dumping definition and data mysql database or table");
289
  short_usage_sub();
unknown's avatar
unknown committed
290
  print_defaults("my",load_default_groups);
291 292
  my_print_help(my_long_options);
  my_print_variables(my_long_options);
unknown's avatar
unknown committed
293 294 295
} /* usage */


296 297 298 299 300 301 302
static void short_usage(void)
{
  short_usage_sub();
  printf("For more options, use %s --help\n", my_progname);
}


303
static void write_header(FILE *sql_file, char *db_name)
unknown's avatar
unknown committed
304
{
unknown's avatar
unknown committed
305
  if (opt_xml)
unknown's avatar
unknown committed
306
  {
unknown's avatar
unknown committed
307
    fprintf(sql_file,"<?xml version=\"1.0\"?>\n");
unknown's avatar
unknown committed
308 309
    fprintf(sql_file,"<mysqldump>\n");
  }
310
  else if (opt_comments)
unknown's avatar
unknown committed
311 312 313 314
  {
    fprintf(sql_file, "-- MySQL dump %s\n--\n", DUMP_VERSION);
    fprintf(sql_file, "-- Host: %s    Database: %s\n",
	    current_host ? current_host : "localhost", db_name ? db_name : "");
315
    fputs("-- ------------------------------------------------------\n",
unknown's avatar
unknown committed
316 317 318 319
	  sql_file);
    fprintf(sql_file, "-- Server version\t%s\n",
	    mysql_get_server_info(&mysql_connection));
  }
unknown's avatar
unknown committed
320
  return;
321
} /* write_header */
unknown's avatar
unknown committed
322

unknown's avatar
unknown committed
323 324 325 326 327 328
static void write_footer(FILE *sql_file)
{
  if (opt_xml)
    fprintf(sql_file,"</mysqldump>");
  fputs("\n", sql_file);
} /* write_footer */
unknown's avatar
unknown committed
329

unknown's avatar
unknown committed
330

331 332 333
static my_bool
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
	       char *argument)
unknown's avatar
unknown committed
334
{
unknown's avatar
unknown committed
335
  switch (optid) {
336 337 338 339
  case OPT_MASTER_DATA:
    opt_master_data=1;
    opt_first_slave=1;
    break;
unknown's avatar
unknown committed
340 341 342 343
  case OPT_DELETE_MASTER_LOGS:
    opt_delete_master_logs=1;
    opt_first_slave=1;
    break;
344 345 346 347 348 349 350 351 352 353 354 355 356 357
  case 'p':
    if (argument)
    {
      char *start=argument;
      my_free(opt_password,MYF(MY_ALLOW_ZERO_PTR));
      opt_password=my_strdup(argument,MYF(MY_FAE));
      while (*argument) *argument++= 'x';		/* Destroy argument */
      if (*start)
	start[1]=0;				/* Cut length of argument */
    }
    else
      tty_password=1;
    break;
  case 'r':
unknown's avatar
unknown committed
358
    if (!(md_result_file = my_fopen(argument, O_WRONLY | FILE_BINARY,
359 360 361 362
				    MYF(MY_WME))))
      exit(1);
    break;
  case 'W':
unknown's avatar
unknown committed
363
#ifdef __WIN__
364
    opt_mysql_unix_port=MYSQL_NAMEDPIPE;
unknown's avatar
unknown committed
365
#endif
366 367 368 369 370 371 372
    break;
  case 'T':
    opt_disable_keys=0;
    break;
  case '#':
    DBUG_PUSH(argument ? argument : "d:t:o");
    break;
373
#include <sslopt-case.h>
374 375 376 377 378 379 380 381 382 383
  case 'V': print_version(); exit(0);
  case 'X':
    opt_xml = 1;
    opt_disable_keys=0;
    break;
  case 'I':
  case '?':
    usage();
    exit(0);
  case (int) OPT_OPTIMIZE:
384 385 386
    extended_insert=opt_drop=opt_lock=quick=create_options=opt_disable_keys=
    lock_tables=1;
    if (opt_single_transaction) lock_tables=0;
387 388 389 390
    break;
  case (int) OPT_TABLES:
    opt_databases=0;
    break;
unknown's avatar
unknown committed
391
  }
392 393 394 395 396 397 398 399 400 401 402 403
  return 0;
}


static int get_options(int *argc, char ***argv)
{
  int ho_error;

  md_result_file= stdout;
  load_defaults("my",load_default_groups,argc,argv);

  if ((ho_error=handle_options(argc, argv, my_long_options, get_one_option)))
404
    exit(ho_error);
405

unknown's avatar
unknown committed
406 407 408 409 410
  if (opt_delayed)
    opt_lock=0;				/* Can't have lock with delayed */
  if (!path && (enclosed || opt_enclosed || escaped || lines_terminated ||
		fields_terminated))
  {
411 412
    fprintf(stderr,
	    "%s: You must use option --tab with --fields-...\n", my_progname);
unknown's avatar
unknown committed
413 414
    return(1);
  }
415 416 417 418 419 420
  
  if (opt_single_transaction && lock_tables) 
  {
    fprintf(stderr, "%s: You can't use --lock-tables and --single-transaction at the same time.\n", my_progname);
    return(1);    
  }
unknown's avatar
unknown committed
421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445

  if (enclosed && opt_enclosed)
  {
    fprintf(stderr, "%s: You can't use ..enclosed.. and ..optionally-enclosed.. at the same time.\n", my_progname);
    return(1);
  }
  if (replace && ignore)
  {
    fprintf(stderr, "%s: You can't use --ignore (-i) and --replace (-r) at the same time.\n",my_progname);
    return(1);
  }
  if ((opt_databases || opt_alldbs) && path)
  {
    fprintf(stderr,
	    "%s: --databases or --all-databases can't be used with --tab.\n",
	    my_progname);
    return(1);
  }
  if (default_charset)
  {
    if (set_default_charset_by_name(default_charset, MYF(MY_WME)))
      exit(1);
  }
  if ((*argc < 1 && !opt_alldbs) || (*argc > 0 && opt_alldbs))
  {
446
    short_usage();
unknown's avatar
unknown committed
447 448 449
    return 1;
  }
  if (tty_password)
450
    opt_password=get_tty_password(NullS);
unknown's avatar
unknown committed
451 452 453 454 455 456 457 458 459 460
  return(0);
} /* get_options */


/*
** DBerror -- prints mysql error message and exits the program.
*/
static void DBerror(MYSQL *mysql, const char *when)
{
  DBUG_ENTER("DBerror");
unknown's avatar
unknown committed
461
  my_printf_error(0,"Got error: %d: %s %s", MYF(0),
unknown's avatar
unknown committed
462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489
		  mysql_errno(mysql), mysql_error(mysql), when);
  safe_exit(EX_MYSQLERR);
  DBUG_VOID_RETURN;
} /* DBerror */


static void safe_exit(int error)
{
  if (!first_error)
    first_error= error;
  if (ignore_errors)
    return;
  if (sock)
    mysql_close(sock);
  exit(error);
}
/* safe_exit */


/*
** dbConnect -- connects to the host and selects DB.
**        Also checks whether the tablename is a valid table name.
*/
static int dbConnect(char *host, char *user,char *passwd)
{
  DBUG_ENTER("dbConnect");
  if (verbose)
  {
490
    fprintf(stderr, "-- Connecting to %s...\n", host ? host : "localhost");
unknown's avatar
unknown committed
491 492 493 494 495 496 497
  }
  mysql_init(&mysql_connection);
  if (opt_compress)
    mysql_options(&mysql_connection,MYSQL_OPT_COMPRESS,NullS);
#ifdef HAVE_OPENSSL
  if (opt_use_ssl)
    mysql_ssl_set(&mysql_connection, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
498
		  opt_ssl_capath, opt_ssl_cipher);
unknown's avatar
unknown committed
499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516
#endif
  if (!(sock= mysql_real_connect(&mysql_connection,host,user,passwd,
         NULL,opt_mysql_port,opt_mysql_unix_port,
         0)))
  {
    DBerror(&mysql_connection, "when trying to connect");
    return 1;
  }
  return 0;
} /* dbConnect */


/*
** dbDisconnect -- disconnects from the host.
*/
static void dbDisconnect(char *host)
{
  if (verbose)
517
    fprintf(stderr, "-- Disconnecting from %s...\n", host ? host : "localhost");
unknown's avatar
unknown committed
518 519 520 521 522 523 524 525 526 527 528 529 530
  mysql_close(sock);
} /* dbDisconnect */


static void unescape(FILE *file,char *pos,uint length)
{
  char *tmp;
  DBUG_ENTER("unescape");
  if (!(tmp=(char*) my_malloc(length*2+1, MYF(MY_WME))))
  {
    ignore_errors=0;				/* Fatal error */
    safe_exit(EX_MYSQLERR);			/* Force exit */
  }
unknown's avatar
unknown committed
531
  mysql_real_escape_string(&mysql_connection, tmp, pos, length);
unknown's avatar
unknown committed
532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549
  fputc('\'', file);
  fputs(tmp, file);
  fputc('\'', file);
  my_free(tmp, MYF(MY_WME));
  DBUG_VOID_RETURN;
} /* unescape */


static my_bool test_if_special_chars(const char *str)
{
#if MYSQL_VERSION_ID >= 32300
  for ( ; *str ; str++)
    if (!isvar(*str) && *str != '$')
      return 1;
#endif
  return 0;
} /* test_if_special_chars */

unknown's avatar
unknown committed
550 551

static char *quote_name(const char *name, char *buff, my_bool force)
unknown's avatar
unknown committed
552
{
unknown's avatar
unknown committed
553 554 555 556 557 558 559 560 561 562 563 564
  char *to= buff;
  if (!force && !opt_quoted && !test_if_special_chars(name))
    return (char*) name;
  *to++= QUOTE_CHAR;
  while (*name)
  {
    if (*name == QUOTE_CHAR)
      *to= QUOTE_CHAR;
    *to++= *name++;
  }
  to[0]=QUOTE_CHAR;
  to[1]=0;
unknown's avatar
unknown committed
565 566 567 568
  return buff;
} /* quote_name */


569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585

static char *quote_for_like(const char *name, char *buff)
{
  char *to= buff;
  *to++= '\'';
  while (*name)
  {
    if (*name == '\'' || *name == '_' || *name == '\\' || *name == '%')
      *to++= '\\';
    *to++= *name++;
  }
  to[0]= '\'';
  to[1]= 0;
  return buff;
}


unknown's avatar
unknown committed
586
/*
unknown's avatar
unknown committed
587 588 589 590 591
  getStructure -- retrievs database structure, prints out corresponding
  CREATE statement and fills out insert_pat.

  RETURN
    number of fields in table, 0 if error
unknown's avatar
unknown committed
592
*/
unknown's avatar
unknown committed
593

unknown's avatar
unknown committed
594 595 596 597 598 599
static uint getTableStructure(char *table, char* db)
{
  MYSQL_RES  *tableRes;
  MYSQL_ROW  row;
  my_bool    init=0;
  uint       numFields;
unknown's avatar
unknown committed
600
  char	     *strpos, *result_table, *opt_quoted_table;
unknown's avatar
unknown committed
601
  const char *delayed;
unknown's avatar
unknown committed
602 603
  char	     name_buff[NAME_LEN+3],table_buff[NAME_LEN*2+3];
  char	     table_buff2[NAME_LEN*2+3];
unknown's avatar
unknown committed
604
  FILE       *sql_file = md_result_file;
unknown's avatar
unknown committed
605 606 607 608 609
  DBUG_ENTER("getTableStructure");

  delayed= opt_delayed ? " DELAYED " : "";

  if (verbose)
610
    fprintf(stderr, "-- Retrieving table structure for table %s...\n", table);
unknown's avatar
unknown committed
611

612 613
  sprintf(insert_pat,"SET OPTION SQL_QUOTE_SHOW_CREATE=%d",
	  (opt_quoted || opt_keywords));
unknown's avatar
unknown committed
614 615
  result_table=     quote_name(table, table_buff, 1);
  opt_quoted_table= quote_name(table, table_buff2, 0);
unknown's avatar
unknown committed
616
  if (!mysql_query(sock,insert_pat))
unknown's avatar
unknown committed
617
  {
618 619
    /* using SHOW CREATE statement */
    if (!tFlag)
unknown's avatar
unknown committed
620
    {
621 622 623
      /* Make an sql-file, if path was given iow. option -T was given */
      char buff[20+FN_REFLEN];

unknown's avatar
unknown committed
624
      sprintf(buff,"show create table %s", result_table);
625
      if (mysql_query(sock, buff))
unknown's avatar
unknown committed
626
      {
unknown's avatar
unknown committed
627 628
        fprintf(stderr, "%s: Can't get CREATE TABLE for table %s (%s)\n",
		      my_progname, result_table, mysql_error(sock));
629 630
        safe_exit(EX_MYSQLERR);
        DBUG_RETURN(0);
unknown's avatar
unknown committed
631 632
      }

633 634 635
      if (path)
      {
        char filename[FN_REFLEN], tmp_path[FN_REFLEN];
636
        convert_dirname(tmp_path,path,NullS);
637
        sql_file= my_fopen(fn_format(filename, table, tmp_path, ".sql", 4),
638
				 O_WRONLY, MYF(MY_WME));
639 640
        if (!sql_file)			/* If file couldn't be opened */
        {
unknown's avatar
unknown committed
641 642
	  safe_exit(EX_MYSQLERR);
	  DBUG_RETURN(0);
643
        }
644
        write_header(sql_file, db);
645
      }
646
      if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
647 648
	fprintf(sql_file, "\n--\n-- Table structure for table %s\n--\n\n",
		result_table);
649
      if (opt_drop)
unknown's avatar
unknown committed
650
        fprintf(sql_file, "DROP TABLE IF EXISTS %s;\n", opt_quoted_table);
651 652 653

      tableRes=mysql_store_result(sock);
      row=mysql_fetch_row(tableRes);
unknown's avatar
unknown committed
654 655
      if (!opt_xml)
	fprintf(sql_file, "%s;\n", row[1]);
656 657
      mysql_free_result(tableRes);
    }
unknown's avatar
unknown committed
658
    sprintf(insert_pat,"show fields from %s", result_table);
659
    if (mysql_query(sock,insert_pat) || !(tableRes=mysql_store_result(sock)))
unknown's avatar
unknown committed
660
    {
unknown's avatar
unknown committed
661 662
      fprintf(stderr, "%s: Can't get info about table: %s\nerror: %s\n",
	      my_progname, result_table, mysql_error(sock));
663 664
      if (path)
	my_fclose(sql_file, MYF(MY_WME));
665 666
      safe_exit(EX_MYSQLERR);
      DBUG_RETURN(0);
unknown's avatar
unknown committed
667
    }
668

unknown's avatar
unknown committed
669
    if (cFlag)
unknown's avatar
unknown committed
670
      sprintf(insert_pat, "INSERT %sINTO %s (", delayed, opt_quoted_table);
671
    else
unknown's avatar
unknown committed
672
    {
unknown's avatar
unknown committed
673 674
      sprintf(insert_pat, "INSERT %sINTO %s VALUES ", delayed,
	      opt_quoted_table);
675 676 677 678 679 680 681 682
      if (!extended_insert)
        strcat(insert_pat,"(");
    }

    strpos=strend(insert_pat);
    while ((row=mysql_fetch_row(tableRes)))
    {
      if (init)
unknown's avatar
unknown committed
683
      {
684
        if (cFlag)
685
	  strpos=strmov(strpos,", ");
unknown's avatar
unknown committed
686
      }
687 688
      init=1;
      if (cFlag)
unknown's avatar
unknown committed
689
        strpos=strmov(strpos,quote_name(row[SHOW_FIELDNAME], name_buff, 0));
unknown's avatar
unknown committed
690
    }
691 692
    numFields = (uint) mysql_num_rows(tableRes);
    mysql_free_result(tableRes);
unknown's avatar
unknown committed
693
  }
694
  else
unknown's avatar
unknown committed
695
  {
696 697 698
  /*  fprintf(stderr, "%s: Can't set SQL_QUOTE_SHOW_CREATE option (%s)\n",
      my_progname, mysql_error(sock)); */

unknown's avatar
unknown committed
699
    sprintf(insert_pat,"show fields from %s", result_table);
700
    if (mysql_query(sock,insert_pat) || !(tableRes=mysql_store_result(sock)))
unknown's avatar
unknown committed
701
    {
unknown's avatar
unknown committed
702 703
      fprintf(stderr, "%s: Can't get info about table: %s\nerror: %s\n",
		    my_progname, result_table, mysql_error(sock));
unknown's avatar
unknown committed
704 705 706 707
      safe_exit(EX_MYSQLERR);
      DBUG_RETURN(0);
    }

708 709
    /* Make an sql-file, if path was given iow. option -T was given */
    if (!tFlag)
unknown's avatar
unknown committed
710
    {
711
      if (path)
unknown's avatar
unknown committed
712
      {
713
        char filename[FN_REFLEN], tmp_path[FN_REFLEN];
714
        convert_dirname(tmp_path,path,NullS);
715 716 717 718
        sql_file= my_fopen(fn_format(filename, table, tmp_path, ".sql", 4),
				 O_WRONLY, MYF(MY_WME));
        if (!sql_file)			/* If file couldn't be opened */
        {
719 720
	  safe_exit(EX_MYSQLERR);
	  DBUG_RETURN(0);
721
        }
722
        write_header(sql_file, db);
unknown's avatar
unknown committed
723
      }
724
      if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
725 726
	fprintf(sql_file, "\n--\n-- Table structure for table %s\n--\n\n",
		result_table);
727
      if (opt_drop)
728 729
        fprintf(sql_file, "DROP TABLE IF EXISTS %s;\n",result_table);
      fprintf(sql_file, "CREATE TABLE %s (\n", result_table);
730 731
    }
    if (cFlag)
732
      sprintf(insert_pat, "INSERT %sINTO %s (", delayed, result_table);
733 734
    else
    {
735
      sprintf(insert_pat, "INSERT %sINTO %s VALUES ", delayed, result_table);
736 737
      if (!extended_insert)
        strcat(insert_pat,"(");
unknown's avatar
unknown committed
738
    }
739 740

    strpos=strend(insert_pat);
unknown's avatar
unknown committed
741 742
    while ((row=mysql_fetch_row(tableRes)))
    {
743 744
      ulong *lengths=mysql_fetch_lengths(tableRes);
      if (init)
unknown's avatar
unknown committed
745
      {
746
        if (!tFlag)
747
	  fputs(",\n",sql_file);
748
        if (cFlag)
749
	  strpos=strmov(strpos,", ");
750 751 752
      }
      init=1;
      if (cFlag)
unknown's avatar
unknown committed
753
        strpos=strmov(strpos,quote_name(row[SHOW_FIELDNAME], name_buff, 0));
754 755 756
      if (!tFlag)
      {
        if (opt_keywords)
757
	  fprintf(sql_file, "  %s.%s %s", result_table,
unknown's avatar
unknown committed
758 759
		  quote_name(row[SHOW_FIELDNAME],name_buff, 0),
		  row[SHOW_TYPE]);
760
        else
unknown's avatar
unknown committed
761
	  fprintf(sql_file, "  %s %s", quote_name(row[SHOW_FIELDNAME],
unknown's avatar
unknown committed
762 763
						  name_buff, 0),
		  row[SHOW_TYPE]);
764 765
        if (row[SHOW_DEFAULT])
        {
unknown's avatar
unknown committed
766
	  fputs(" DEFAULT ", sql_file);
unknown's avatar
unknown committed
767
	  unescape(sql_file, row[SHOW_DEFAULT], lengths[SHOW_DEFAULT]);
768 769
        }
        if (!row[SHOW_NULL][0])
unknown's avatar
unknown committed
770
	  fputs(" NOT NULL", sql_file);
771
        if (row[SHOW_EXTRA][0])
unknown's avatar
unknown committed
772
	  fprintf(sql_file, " %s",row[SHOW_EXTRA]);
unknown's avatar
unknown committed
773 774
      }
    }
775 776 777
    numFields = (uint) mysql_num_rows(tableRes);
    mysql_free_result(tableRes);
    if (!tFlag)
unknown's avatar
unknown committed
778
    {
779 780 781
      /* Make an sql-file, if path was given iow. option -T was given */
      char buff[20+FN_REFLEN];
      uint keynr,primary_key;
unknown's avatar
unknown committed
782
      sprintf(buff,"show keys from %s", result_table);
unknown's avatar
unknown committed
783 784
      if (mysql_query(sock, buff))
      {
unknown's avatar
unknown committed
785 786
        fprintf(stderr, "%s: Can't get keys for table %s (%s)\n",
		my_progname, result_table, mysql_error(sock));
787
        if (path)
unknown's avatar
unknown committed
788
	  my_fclose(sql_file, MYF(MY_WME));
789 790
        safe_exit(EX_MYSQLERR);
        DBUG_RETURN(0);
unknown's avatar
unknown committed
791
      }
792 793 794 795 796 797

      tableRes=mysql_store_result(sock);
      /* Find first which key is primary key */
      keynr=0;
      primary_key=INT_MAX;
      while ((row=mysql_fetch_row(tableRes)))
unknown's avatar
unknown committed
798
      {
799 800
        if (atoi(row[3]) == 1)
        {
unknown's avatar
unknown committed
801 802 803 804 805 806 807 808 809 810
	  keynr++;
#ifdef FORCE_PRIMARY_KEY
	  if (atoi(row[1]) == 0 && primary_key == INT_MAX)
	    primary_key=keynr;
#endif
	  if (!strcmp(row[2],"PRIMARY"))
	  {
	    primary_key=keynr;
	    break;
	  }
811
        }
unknown's avatar
unknown committed
812
      }
813 814 815 816 817 818
      mysql_data_seek(tableRes,0);
      keynr=0;
      while ((row=mysql_fetch_row(tableRes)))
      {
        if (atoi(row[3]) == 1)
        {
unknown's avatar
unknown committed
819 820 821 822
	  if (keynr++)
	    putc(')', sql_file);
	  if (atoi(row[1]))       /* Test if duplicate key */
	    /* Duplicate allowed */
unknown's avatar
unknown committed
823
	    fprintf(sql_file, ",\n  KEY %s (",quote_name(row[2],name_buff,0));
unknown's avatar
unknown committed
824 825 826
	  else if (keynr == primary_key)
	    fputs(",\n  PRIMARY KEY (",sql_file); /* First UNIQUE is primary */
	  else
unknown's avatar
unknown committed
827
	    fprintf(sql_file, ",\n  UNIQUE %s (",quote_name(row[2],name_buff,0));
828 829
        }
        else
unknown's avatar
unknown committed
830
	  putc(',', sql_file);
unknown's avatar
unknown committed
831
        fputs(quote_name(row[4], name_buff, 0), sql_file);
832
        if (row[7])
unknown's avatar
unknown committed
833
	  fprintf(sql_file, " (%s)",row[7]);      /* Sub key */
834 835 836 837 838 839 840
      }
      if (keynr)
        putc(')', sql_file);
      fputs("\n)",sql_file);

      /* Get MySQL specific create options */
      if (create_options)
unknown's avatar
unknown committed
841
      {
842 843 844
	char show_name_buff[FN_REFLEN];
        sprintf(buff,"show table status like %s",
		quote_for_like(table, show_name_buff));
845 846
        if (mysql_query(sock, buff))
        {
unknown's avatar
unknown committed
847 848 849 850
	  if (mysql_errno(sock) != ER_PARSE_ERROR)
	  {					/* If old MySQL version */
	    if (verbose)
	      fprintf(stderr,
unknown's avatar
unknown committed
851 852
		      "-- Warning: Couldn't get status information for table %s (%s)\n",
		      result_table,mysql_error(sock));
unknown's avatar
unknown committed
853
	  }
854 855
        }
        else if (!(tableRes=mysql_store_result(sock)) ||
unknown's avatar
unknown committed
856
		 !(row=mysql_fetch_row(tableRes)))
857
        {
unknown's avatar
unknown committed
858
	  fprintf(stderr,
unknown's avatar
unknown committed
859 860
		  "Error: Couldn't read status information for table %s (%s)\n",
		  result_table,mysql_error(sock));
861 862 863
        }
        else
        {
unknown's avatar
unknown committed
864 865 866 867 868
	  fputs("/*!",sql_file);
	  print_value(sql_file,tableRes,row,"type=","Type",0);
	  print_value(sql_file,tableRes,row,"","Create_options",0);
	  print_value(sql_file,tableRes,row,"comment=","Comment",1);
	  fputs(" */",sql_file);
869 870
        }
        mysql_free_result(tableRes);		/* Is always safe to free */
unknown's avatar
unknown committed
871
      }
872
      fputs(";\n", sql_file);
unknown's avatar
unknown committed
873 874 875 876 877 878 879 880
    }
  }
  if (cFlag)
  {
    strpos=strmov(strpos,") VALUES ");
    if (!extended_insert)
      strpos=strmov(strpos,"(");
  }
881 882
  if (sql_file != md_result_file)
    my_fclose(sql_file, MYF(MY_WME));
unknown's avatar
unknown committed
883 884 885 886 887 888 889 890 891
  DBUG_RETURN(numFields);
} /* getTableStructure */


static char *add_load_option(char *ptr,const char *object,
			     const char *statement)
{
  if (object)
  {
892 893 894 895 896 897 898 899 900 901
    /* Don't escape hex constants */
    if (object[0] == '0' && (object[1] == 'x' || object[1] == 'X'))
      ptr= strxmov(ptr," ",statement," ",object,NullS);
    else
    {
      /* char constant; escape */
      ptr= strxmov(ptr," ",statement," '",NullS);
      ptr= field_escape(ptr,object,(uint) strlen(object));
      *ptr++= '\'';
    }
unknown's avatar
unknown committed
902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942
  }
  return ptr;
} /* add_load_option */


/*
** Allow the user to specify field terminator strings like:
** "'", "\", "\\" (escaped backslash), "\t" (tab), "\n" (newline)
** This is done by doubleing ' and add a end -\ if needed to avoid
** syntax errors from the SQL parser.
*/

static char *field_escape(char *to,const char *from,uint length)
{
  const char *end;
  uint end_backslashes=0;

  for (end= from+length; from != end; from++)
  {
    *to++= *from;
    if (*from == '\\')
      end_backslashes^=1;    /* find odd number of backslashes */
    else
    {
      if (*from == '\'' && !end_backslashes)
	*to++= *from;      /* We want a duplicate of "'" for MySQL */
      end_backslashes=0;
    }
  }
  /* Add missing backslashes if user has specified odd number of backs.*/
  if (end_backslashes)
    *to++= '\\';
  return to;
} /* field_escape */


/*
** dumpTable saves database contents as a series of INSERT statements.
*/
static void dumpTable(uint numFields, char *table)
{
943
  char query[QUERY_LENGTH], *end, buff[256],table_buff[NAME_LEN+3];
unknown's avatar
unknown committed
944
  char *result_table, table_buff2[NAME_LEN*2+3], *opt_quoted_table;
unknown's avatar
unknown committed
945
  MYSQL_RES	*res;
unknown's avatar
unknown committed
946 947
  MYSQL_FIELD	*field;
  MYSQL_ROW	row;
unknown's avatar
unknown committed
948
  ulong		rownr, row_break, total_length, init_length;
949
  const char    *table_type;
unknown's avatar
unknown committed
950

unknown's avatar
unknown committed
951 952
  result_table= quote_name(table,table_buff, 1);
  opt_quoted_table= quote_name(table, table_buff2, 0);
953 954 955 956 957 958 959 960 961 962 963 964 965

  /* Check table type */
  if ((table_type= check_if_ignore_table(table)))
  {
    if (verbose)
      fprintf(stderr,
	      "-- Skipping data for table '%s' because it's of type %s\n",
	      table, table_type);
    return;
  }

  if (verbose)
    fprintf(stderr, "-- Sending SELECT query...\n");
unknown's avatar
unknown committed
966 967 968
  if (path)
  {
    char filename[FN_REFLEN], tmp_path[FN_REFLEN];
969
    convert_dirname(tmp_path,path,NullS);
unknown's avatar
unknown committed
970 971 972 973 974
    my_load_path(tmp_path, tmp_path, NULL);
    fn_format(filename, table, tmp_path, ".txt", 4);
    my_delete(filename, MYF(0)); /* 'INTO OUTFILE' doesn't work, if
				    filename wasn't deleted */
    to_unix_path(filename);
975 976
    sprintf(query, "SELECT /*!40001 SQL_NO_CACHE */ * INTO OUTFILE '%s'",
	    filename);
unknown's avatar
unknown committed
977 978 979 980 981 982 983 984 985 986 987 988 989 990 991
    end= strend(query);
    if (replace)
      end= strmov(end, " REPLACE");
    if (ignore)
      end= strmov(end, " IGNORE");

    if (fields_terminated || enclosed || opt_enclosed || escaped)
      end= strmov(end, " FIELDS");
    end= add_load_option(end, fields_terminated, " TERMINATED BY");
    end= add_load_option(end, enclosed, " ENCLOSED BY");
    end= add_load_option(end, opt_enclosed, " OPTIONALLY ENCLOSED BY");
    end= add_load_option(end, escaped, " ESCAPED BY");
    end= add_load_option(end, lines_terminated, " LINES TERMINATED BY");
    *end= '\0';

unknown's avatar
unknown committed
992
    sprintf(buff," FROM %s", result_table);
unknown's avatar
unknown committed
993 994 995 996 997 998 999 1000 1001 1002 1003
    end= strmov(end,buff);
    if (where)
      end= strxmov(end, " WHERE ",where,NullS);
    if (mysql_query(sock, query))
    {
      DBerror(sock, "when executing 'SELECT INTO OUTFILE'");
      return;
    }
  }
  else
  {
1004
    if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
1005 1006
      fprintf(md_result_file,"\n--\n-- Dumping data for table %s\n--\n",
	      result_table);
1007
    sprintf(query, "SELECT /*!40001 SQL_NO_CACHE */ * FROM %s",
unknown's avatar
unknown committed
1008
	    result_table);
unknown's avatar
unknown committed
1009 1010
    if (where)
    {
1011
      if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
1012
	fprintf(md_result_file,"-- WHERE:  %s\n",where);
unknown's avatar
unknown committed
1013 1014
      strxmov(strend(query), " WHERE ",where,NullS);
    }
unknown's avatar
unknown committed
1015
    if (!opt_xml)
unknown's avatar
merge  
unknown committed
1016
      fputs("\n", md_result_file);
unknown's avatar
unknown committed
1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031
    if (mysql_query(sock, query))
    {
      DBerror(sock, "when retrieving data from server");
      return;
    }
    if (quick)
      res=mysql_use_result(sock);
    else
      res=mysql_store_result(sock);
    if (!res)
    {
      DBerror(sock, "when retrieving data from server");
      return;
    }
    if (verbose)
1032
      fprintf(stderr, "-- Retrieving rows...\n");
unknown's avatar
unknown committed
1033 1034
    if (mysql_num_fields(res) != numFields)
    {
unknown's avatar
unknown committed
1035 1036
      fprintf(stderr,"%s: Error in field count for table: %s !  Aborting.\n",
	      my_progname, result_table);
unknown's avatar
unknown committed
1037 1038 1039 1040
      safe_exit(EX_CONSCHECK);
      return;
    }

unknown's avatar
merge  
unknown committed
1041
    if (opt_disable_keys)
unknown's avatar
unknown committed
1042
      fprintf(md_result_file, "\n/*!40000 ALTER TABLE %s DISABLE KEYS */;\n",
unknown's avatar
unknown committed
1043
	      opt_quoted_table);
unknown's avatar
unknown committed
1044
    if (opt_lock)
unknown's avatar
unknown committed
1045
      fprintf(md_result_file,"LOCK TABLES %s WRITE;\n", opt_quoted_table);
unknown's avatar
unknown committed
1046 1047 1048 1049

    total_length=net_buffer_length;		/* Force row break */
    row_break=0;
    rownr=0;
unknown's avatar
unknown committed
1050
    init_length=(uint) strlen(insert_pat)+4;
unknown's avatar
unknown committed
1051
    if (opt_xml)
unknown's avatar
unknown committed
1052
      fprintf(md_result_file, "\t<table name=\"%s\">\n", table);
unknown's avatar
unknown committed
1053

unknown's avatar
unknown committed
1054 1055 1056
    if (opt_autocommit)
      fprintf(md_result_file, "set autocommit=0;\n");

unknown's avatar
unknown committed
1057 1058 1059 1060 1061
    while ((row=mysql_fetch_row(res)))
    {
      uint i;
      ulong *lengths=mysql_fetch_lengths(res);
      rownr++;
unknown's avatar
unknown committed
1062
      if (!extended_insert && !opt_xml)
unknown's avatar
unknown committed
1063
	fputs(insert_pat,md_result_file);
unknown's avatar
unknown committed
1064 1065
      mysql_field_seek(res,0);

unknown's avatar
merge  
unknown committed
1066 1067 1068
      if (opt_xml)
        fprintf(md_result_file, "\t<row>\n");

unknown's avatar
unknown committed
1069 1070 1071 1072
      for (i = 0; i < mysql_num_fields(res); i++)
      {
	if (!(field = mysql_fetch_field(res)))
	{
unknown's avatar
unknown committed
1073 1074
	  sprintf(query,"%s: Not enough fields from table %s! Aborting.\n",
		  my_progname, result_table);
unknown's avatar
unknown committed
1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090
	  fputs(query,stderr);
	  safe_exit(EX_CONSCHECK);
	  return;
	}
	if (extended_insert)
	{
	  ulong length = lengths[i];
	  if (i == 0)
	    dynstr_set(&extended_row,"(");
	  else
	    dynstr_append(&extended_row,",");

	  if (row[i])
	  {
	    if (length)
	    {
1091
	      if (!IS_NUM_FIELD(field))
unknown's avatar
unknown committed
1092 1093 1094 1095 1096 1097 1098
	      {
		if (dynstr_realloc(&extended_row,length * 2+2))
		{
		  fputs("Aborting dump (out of memory)",stderr);
		  safe_exit(EX_EOM);
		}
		dynstr_append(&extended_row,"\'");
unknown's avatar
unknown committed
1099
		extended_row.length +=
unknown's avatar
unknown committed
1100 1101 1102 1103 1104 1105
		  mysql_real_escape_string(&mysql_connection,
					   &extended_row.str[extended_row.length],row[i],length);
		extended_row.str[extended_row.length]='\0';
		dynstr_append(&extended_row,"\'");
	      }
	      else
1106
	      {
unknown's avatar
unknown committed
1107
		/* change any strings ("inf", "-inf", "nan") into NULL */
1108
		char *ptr = row[i];
1109
		if (isalpha(*ptr) || (*ptr == '-' && isalpha(ptr[1])))
unknown's avatar
unknown committed
1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122
		  dynstr_append(&extended_row, "NULL");
		else
		{
		  if (field->type == FIELD_TYPE_DECIMAL)
		  {
		    /* add " signs around */
		    dynstr_append(&extended_row, "\"");
		    dynstr_append(&extended_row, ptr);
		    dynstr_append(&extended_row, "\"");
		  }
		  else
		    dynstr_append(&extended_row, ptr);
		}
1123
	      }
unknown's avatar
unknown committed
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135
	    }
	    else
	      dynstr_append(&extended_row,"\'\'");
	  }
	  else if (dynstr_append(&extended_row,"NULL"))
	  {
	    fputs("Aborting dump (out of memory)",stderr);
	    safe_exit(EX_EOM);
	  }
	}
	else
	{
unknown's avatar
unknown committed
1136 1137
	  if (i && !opt_xml)
	    fputc(',', md_result_file);
unknown's avatar
unknown committed
1138 1139
	  if (row[i])
	  {
1140
	    if (!IS_NUM_FIELD(field))
1141
	    {
unknown's avatar
unknown committed
1142 1143 1144 1145 1146
	      if (opt_xml)
		print_quoted_xml(md_result_file, field->name, row[i],
				 lengths[i]);
	      else
		unescape(md_result_file, row[i], lengths[i]);
unknown's avatar
unknown committed
1147
	    }
unknown's avatar
unknown committed
1148
	    else
1149
	    {
unknown's avatar
unknown committed
1150
	      /* change any strings ("inf", "-inf", "nan") into NULL */
1151
	      char *ptr = row[i];
unknown's avatar
unknown committed
1152
	      if (opt_xml)
unknown's avatar
unknown committed
1153 1154
		fprintf(md_result_file, "\t\t<field name=\"%s\">%s</field>\n",
			field->name,!isalpha(*ptr) ?ptr: "NULL");
1155
	      else if (isalpha(*ptr) || (*ptr == '-' && isalpha(ptr[1])))
unknown's avatar
unknown committed
1156
	        fputs("NULL", md_result_file);
unknown's avatar
unknown committed
1157
	      else
unknown's avatar
unknown committed
1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168
	      {
		if (field->type == FIELD_TYPE_DECIMAL)
		{
		  /* add " signs around */
		  fputs("\"", md_result_file);
		  fputs(ptr, md_result_file);
		  fputs("\"", md_result_file);
		}
		else
		  fputs(ptr, md_result_file);
	      }
1169
	    }
unknown's avatar
unknown committed
1170 1171 1172
	  }
	  else
	  {
unknown's avatar
unknown committed
1173
	    if (opt_xml)
unknown's avatar
unknown committed
1174 1175
	      fprintf(md_result_file, "\t\t<field name=\"%s\">%s</field>\n",
		      field->name, "NULL");
unknown's avatar
unknown committed
1176 1177
	    else
	      fputs("NULL", md_result_file);
unknown's avatar
unknown committed
1178 1179 1180 1181
	  }
	}
      }

unknown's avatar
merge  
unknown committed
1182 1183 1184
      if (opt_xml)
        fprintf(md_result_file, "\t</row>\n");

unknown's avatar
unknown committed
1185 1186 1187 1188 1189 1190 1191 1192
      if (extended_insert)
      {
	ulong row_length;
	dynstr_append(&extended_row,")");
        row_length = 2 + extended_row.length;
        if (total_length + row_length < net_buffer_length)
        {
	  total_length += row_length;
unknown's avatar
unknown committed
1193 1194
	  fputc(',',md_result_file);		/* Always row break */
	  fputs(extended_row.str,md_result_file);
unknown's avatar
unknown committed
1195 1196 1197
	}
        else
        {
unknown's avatar
unknown committed
1198
	  if (row_break && !opt_xml)
unknown's avatar
unknown committed
1199
	    fputs(";\n", md_result_file);
unknown's avatar
unknown committed
1200
	  row_break=1;				/* This is first row */
unknown's avatar
unknown committed
1201

unknown's avatar
unknown committed
1202 1203 1204 1205 1206
	  if (!opt_xml)
	  {
	    fputs(insert_pat,md_result_file);
	    fputs(extended_row.str,md_result_file);
	  }
unknown's avatar
unknown committed
1207 1208 1209
	  total_length = row_length+init_length;
        }
      }
unknown's avatar
unknown committed
1210
      else if (!opt_xml)
unknown's avatar
unknown committed
1211
	fputs(");\n", md_result_file);
unknown's avatar
unknown committed
1212
    }
unknown's avatar
unknown committed
1213

unknown's avatar
unknown committed
1214
    /* XML - close table tag and supress regular output */
unknown's avatar
unknown committed
1215
    if (opt_xml)
unknown's avatar
unknown committed
1216
	fprintf(md_result_file, "\t</table>\n");
unknown's avatar
unknown committed
1217
    else if (extended_insert && row_break)
unknown's avatar
unknown committed
1218 1219
      fputs(";\n", md_result_file);		/* If not empty table */
    fflush(md_result_file);
unknown's avatar
unknown committed
1220 1221
    if (mysql_errno(sock))
    {
unknown's avatar
unknown committed
1222
      sprintf(query,"%s: Error %d: %s when dumping table %s at row: %ld\n",
unknown's avatar
unknown committed
1223 1224 1225
	      my_progname,
	      mysql_errno(sock),
	      mysql_error(sock),
unknown's avatar
unknown committed
1226
	      result_table,
unknown's avatar
unknown committed
1227 1228 1229 1230 1231 1232
	      rownr);
      fputs(query,stderr);
      safe_exit(EX_CONSCHECK);
      return;
    }
    if (opt_lock)
unknown's avatar
unknown committed
1233
      fputs("UNLOCK TABLES;\n", md_result_file);
unknown's avatar
merge  
unknown committed
1234 1235
    if (opt_disable_keys)
      fprintf(md_result_file,"/*!40000 ALTER TABLE %s ENABLE KEYS */;\n",
unknown's avatar
unknown committed
1236
	      opt_quoted_table);
unknown's avatar
unknown committed
1237 1238
    if (opt_autocommit)
      fprintf(md_result_file, "commit;\n");
1239
    mysql_free_result(res);
unknown's avatar
unknown committed
1240 1241 1242 1243
  }
} /* dumpTable */


unknown's avatar
unknown committed
1244 1245 1246
static void print_quoted_xml(FILE *output, char *fname, char *str, uint len)
{
  const char *end;
unknown's avatar
unknown committed
1247

unknown's avatar
unknown committed
1248
  fprintf(output, "\t\t<field name=\"%s\">", fname);
unknown's avatar
unknown committed
1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261
  for (end = str + len; str != end; str++)
  {
    if (*str == '<')
      fputs("&lt;", output);
    else if (*str == '>')
      fputs("&gt;", output);
    else if (*str == '&')
      fputs("&amp;", output);
    else if (*str == '\"')
      fputs("&quot;", output);
    else
      fputc(*str, output);
  }
unknown's avatar
unknown committed
1262
  fprintf(output, "</field>\n");
unknown's avatar
unknown committed
1263 1264
}

unknown's avatar
unknown committed
1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276
static char *getTableName(int reset)
{
  static MYSQL_RES *res = NULL;
  MYSQL_ROW    row;

  if (!res)
  {
    if (!(res = mysql_list_tables(sock,NullS)))
      return(NULL);
  }
  if ((row = mysql_fetch_row(res)))
    return((char*) row[0]);
unknown's avatar
unknown committed
1277

unknown's avatar
unknown committed
1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
  if (reset)
    mysql_data_seek(res,0);      /* We want to read again */
  else
  {
    mysql_free_result(res);
    res = NULL;
  }
  return(NULL);
} /* getTableName */


static int dump_all_databases()
{
  MYSQL_ROW row;
  MYSQL_RES *tableres;
  int result=0;

  if (mysql_query(sock, "SHOW DATABASES") ||
      !(tableres = mysql_store_result(sock)))
  {
    my_printf_error(0, "Error: Couldn't execute 'SHOW DATABASES': %s",
		    MYF(0), mysql_error(sock));
    return 1;
  }
  while ((row = mysql_fetch_row(tableres)))
  {
    if (dump_all_tables_in_db(row[0]))
      result=1;
  }
  return result;
}
/* dump_all_databases */


static int dump_databases(char **db_names)
{
  int result=0;
  for ( ; *db_names ; db_names++)
1316
  {
unknown's avatar
unknown committed
1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330
    if (dump_all_tables_in_db(*db_names))
      result=1;
  }
  return result;
} /* dump_databases */


static int init_dumping(char *database)
{
  if (mysql_select_db(sock, database))
  {
    DBerror(sock, "when selecting the database");
    return 1;			/* If --force */
  }
unknown's avatar
unknown committed
1331
  if (!path && !opt_xml)
unknown's avatar
unknown committed
1332 1333 1334
  {
    if (opt_databases || opt_alldbs)
    {
unknown's avatar
unknown committed
1335 1336 1337
      /* length of table name * 2 (if name contain quotas), 2 quotas and 0 */
      char quoted_database_buf[64*2+3];
      char *qdatabase= quote_name(database,quoted_database_buf,opt_quoted);
1338 1339
      if (opt_comments)
	fprintf(md_result_file,"\n--\n-- Current Database: %s\n--\n", database);
unknown's avatar
unknown committed
1340
      if (!opt_create_db)
unknown's avatar
unknown committed
1341
	fprintf(md_result_file,"\nCREATE DATABASE /*!32312 IF NOT EXISTS*/ %s;\n",
unknown's avatar
unknown committed
1342
		qdatabase);
unknown's avatar
unknown committed
1343
      fprintf(md_result_file,"\nUSE %s;\n", database);
unknown's avatar
unknown committed
1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356
    }
  }
  if (extended_insert)
    if (init_dynamic_string(&extended_row, "", 1024, 1024))
      exit(EX_EOM);
  return 0;
} /* init_dumping */


static int dump_all_tables_in_db(char *database)
{
  char *table;
  uint numrows;
unknown's avatar
unknown committed
1357
  char table_buff[NAME_LEN*2+3];
unknown's avatar
unknown committed
1358 1359 1360

  if (init_dumping(database))
    return 1;
unknown's avatar
unknown committed
1361 1362
  if (opt_xml)
    fprintf(md_result_file, "<database name=\"%s\">\n", database);
unknown's avatar
unknown committed
1363 1364 1365 1366 1367 1368
  if (lock_tables)
  {
    DYNAMIC_STRING query;
    init_dynamic_string(&query, "LOCK TABLES ", 256, 1024);
    for (numrows=0 ; (table = getTableName(1)) ; numrows++)
    {
unknown's avatar
unknown committed
1369
      dynstr_append(&query, quote_name(table, table_buff, 1));
unknown's avatar
unknown committed
1370 1371 1372
      dynstr_append(&query, " READ /*!32311 LOCAL */,");
    }
    if (numrows && mysql_real_query(sock, query.str, query.length-1))
unknown's avatar
unknown committed
1373
      DBerror(sock, "when using LOCK TABLES");
unknown's avatar
unknown committed
1374 1375 1376 1377 1378 1379
            /* We shall continue here, if --force was given */
    dynstr_free(&query);
  }
  if (flush_logs)
  {
    if (mysql_refresh(sock, REFRESH_LOG))
unknown's avatar
unknown committed
1380
      DBerror(sock, "when doing refresh");
unknown's avatar
unknown committed
1381 1382 1383 1384 1385 1386 1387 1388
           /* We shall continue here, if --force was given */
  }
  while ((table = getTableName(0)))
  {
    numrows = getTableStructure(table, database);
    if (!dFlag && numrows > 0)
      dumpTable(numrows,table);
  }
unknown's avatar
unknown committed
1389 1390
  if (opt_xml)
    fprintf(md_result_file, "</database>\n");
unknown's avatar
unknown committed
1391
  if (lock_tables)
unknown's avatar
unknown committed
1392
    mysql_query(sock,"UNLOCK TABLES");
unknown's avatar
unknown committed
1393 1394 1395 1396 1397 1398 1399 1400
  return 0;
} /* dump_all_tables_in_db */



static int dump_selected_tables(char *db, char **table_names, int tables)
{
  uint numrows;
unknown's avatar
unknown committed
1401
  char table_buff[NAME_LEN*+3];
unknown's avatar
unknown committed
1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412

  if (init_dumping(db))
    return 1;
  if (lock_tables)
  {
    DYNAMIC_STRING query;
    int i;

    init_dynamic_string(&query, "LOCK TABLES ", 256, 1024);
    for (i=0 ; i < tables ; i++)
    {
unknown's avatar
unknown committed
1413
      dynstr_append(&query, quote_name(table_names[i], table_buff, 1));
unknown's avatar
unknown committed
1414 1415 1416
      dynstr_append(&query, " READ /*!32311 LOCAL */,");
    }
    if (mysql_real_query(sock, query.str, query.length-1))
unknown's avatar
unknown committed
1417
      DBerror(sock, "when doing LOCK TABLES");
unknown's avatar
unknown committed
1418 1419 1420 1421 1422 1423
       /* We shall countinue here, if --force was given */
    dynstr_free(&query);
  }
  if (flush_logs)
  {
    if (mysql_refresh(sock, REFRESH_LOG))
unknown's avatar
unknown committed
1424
      DBerror(sock, "when doing refresh");
unknown's avatar
unknown committed
1425 1426
     /* We shall countinue here, if --force was given */
  }
unknown's avatar
unknown committed
1427 1428
  if (opt_xml)
    fprintf(md_result_file, "<database name=\"%s\">\n", db);
unknown's avatar
unknown committed
1429 1430 1431 1432 1433 1434
  for (; tables > 0 ; tables-- , table_names++)
  {
    numrows = getTableStructure(*table_names, db);
    if (!dFlag && numrows > 0)
      dumpTable(numrows, *table_names);
  }
unknown's avatar
unknown committed
1435 1436
  if (opt_xml)
    fprintf(md_result_file, "</database>\n");
unknown's avatar
unknown committed
1437
  if (lock_tables)
unknown's avatar
unknown committed
1438
    mysql_query(sock,"UNLOCK TABLES");
unknown's avatar
unknown committed
1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459
  return 0;
} /* dump_selected_tables */


/* Print a value with a prefix on file */
static void print_value(FILE *file, MYSQL_RES  *result, MYSQL_ROW row,
			const char *prefix, const char *name,
			int string_value)
{
  MYSQL_FIELD	*field;
  mysql_field_seek(result, 0);

  for ( ; (field = mysql_fetch_field(result)) ; row++)
  {
    if (!strcmp(field->name,name))
    {
      if (row[0] && row[0][0] && strcmp(row[0],"0")) /* Skip default */
      {
	fputc(' ',file);
	fputs(prefix, file);
	if (string_value)
unknown's avatar
unknown committed
1460
	  unescape(file,row[0],(uint) strlen(row[0]));
unknown's avatar
unknown committed
1461 1462 1463 1464 1465 1466 1467 1468 1469 1470
	else
	  fputs(row[0], file);
	return;
      }
    }
  }
  return;					/* This shouldn't happen */
} /* print_value */


1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525
/*
  Check if we the table is one of the table types that should be ignored:
  MRG_ISAM, MRG_MYISAM

  SYNOPSIS
    check_if_ignore_table()
    table_name			Table name to check

  GLOBAL VARIABLES
    sock			MySQL socket
    verbose			Write warning messages

  RETURN
    0	Table should be backuped
    #	Type of table (that should be skipped)
*/

static const char *check_if_ignore_table(const char *table_name)
{
  char buff[FN_REFLEN+80], show_name_buff[FN_REFLEN];
  MYSQL_RES *res;
  MYSQL_ROW row;
  const char *result= 0;

  sprintf(buff,"show table status like %s",
	  quote_for_like(table_name, show_name_buff));
  if (mysql_query(sock, buff))
  {
    if (mysql_errno(sock) != ER_PARSE_ERROR)
    {					/* If old MySQL version */
      if (verbose)
	fprintf(stderr,
		"-- Warning: Couldn't get status information for table %s (%s)\n",
		table_name,mysql_error(sock));
      return 0;					/* assume table is ok */
    }
  }
  if (!(res= mysql_store_result(sock)) ||
      !(row= mysql_fetch_row(res)))
  {
    fprintf(stderr,
	    "Error: Couldn't read status information for table %s (%s)\n",
	    table_name, mysql_error(sock));
    if (res)
      mysql_free_result(res);
    return 0;					/* assume table is ok */
  }
  if (strcmp(row[1], (result= "MRG_MyISAM")) &&
      strcmp(row[1], (result= "MRG_ISAM")))
    result= 0;
  mysql_free_result(res);  
  return result;
}


unknown's avatar
unknown committed
1526 1527
int main(int argc, char **argv)
{
unknown's avatar
unknown committed
1528 1529 1530
  MYSQL_ROW row;
  MYSQL_RES *master;

unknown's avatar
unknown committed
1531 1532 1533 1534 1535 1536
  MY_INIT(argv[0]);
  if (get_options(&argc, &argv))
  {
    my_end(0);
    exit(EX_USAGE);
  }
1537
  if (dbConnect(current_host, current_user, opt_password))
unknown's avatar
unknown committed
1538 1539
    exit(EX_MYSQLERR);
  if (!path)
1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552
    write_header(md_result_file, *argv);

  if (opt_first_slave)
  {
    lock_tables=0;				/* No other locks needed */
    if (mysql_query(sock, "FLUSH TABLES WITH READ LOCK"))
    {
      my_printf_error(0, "Error: Couldn't execute 'FLUSH TABLES WITH READ LOCK': %s",
                      MYF(0), mysql_error(sock));
      my_end(0);
      return(first_error);
    }
  }
1553
  else if (opt_single_transaction)
1554 1555 1556
  {
    /* There is no sense to start transaction if all tables are locked */ 
    if (mysql_query(sock, "BEGIN"))
1557
    {
1558
      my_printf_error(0, "Error: Couldn't execute 'BEGIN': %s",
1559
                        MYF(0), mysql_error(sock));
1560 1561 1562 1563
      my_end(0);
      return(first_error);
    }    
  }
unknown's avatar
unknown committed
1564 1565 1566
  if (opt_alldbs)
    dump_all_databases();
  else if (argc > 1 && !opt_databases)
1567 1568
  {
    /* Only one database and selected table(s) */
unknown's avatar
unknown committed
1569
    dump_selected_tables(*argv, (argv + 1), (argc - 1));
1570
  }
unknown's avatar
unknown committed
1571
  else
1572 1573
  {
    /* One or more databases, all tables */
unknown's avatar
unknown committed
1574
    dump_databases(argv);
1575
  }
unknown's avatar
unknown committed
1576 1577 1578

  if (opt_first_slave)
  {
1579 1580 1581 1582 1583
    if (opt_delete_master_logs && mysql_query(sock, "FLUSH MASTER"))
    {
      my_printf_error(0, "Error: Couldn't execute 'FLUSH MASTER': %s",
		      MYF(0), mysql_error(sock));
    }
unknown's avatar
unknown committed
1584 1585 1586 1587 1588 1589
    if (opt_master_data)
    {
      if (mysql_query(sock, "SHOW MASTER STATUS") ||
	  !(master = mysql_store_result(sock)))
	my_printf_error(0, "Error: Couldn't execute 'SHOW MASTER STATUS': %s",
			MYF(0), mysql_error(sock));
1590
      else
unknown's avatar
unknown committed
1591 1592
      {
	row = mysql_fetch_row(master);
unknown's avatar
unknown committed
1593
	if (row && row[0] && row[1])
1594
	{
1595 1596 1597
	  if (opt_comments)
	    fprintf(md_result_file,
		    "\n--\n-- Position to start replication from\n--\n\n");
unknown's avatar
unknown committed
1598
	  fprintf(md_result_file,
unknown's avatar
unknown committed
1599 1600
		  "CHANGE MASTER TO MASTER_LOG_FILE='%s', \
MASTER_LOG_POS=%s ;\n",row[0],row[1]); 
unknown's avatar
unknown committed
1601 1602 1603 1604
	}
	mysql_free_result(master);
      }
    }
unknown's avatar
unknown committed
1605 1606 1607 1608
    if (mysql_query(sock, "UNLOCK TABLES"))
      my_printf_error(0, "Error: Couldn't execute 'UNLOCK TABLES': %s",
		      MYF(0), mysql_error(sock));
  }
1609 1610
  else if (opt_single_transaction) /* Just to make it beautiful enough */
  {
1611
    /*
1612 1613
      In case we were locking all tables, we did not start transaction
      so there is no need to commit it.
1614 1615
    */

1616 1617 1618 1619 1620 1621 1622
    /* This should just free locks as we did not change anything */
    if (mysql_query(sock, "COMMIT"))
    {
      my_printf_error(0, "Error: Couldn't execute 'COMMIT': %s",
  	      MYF(0), mysql_error(sock));
    }		      
  }
unknown's avatar
unknown committed
1623
  dbDisconnect(current_host);
unknown's avatar
unknown committed
1624
  write_footer(md_result_file);
unknown's avatar
unknown committed
1625 1626
  if (md_result_file != stdout)
    my_fclose(md_result_file, MYF(0));
1627
  my_free(opt_password, MYF(MY_ALLOW_ZERO_PTR));
unknown's avatar
unknown committed
1628 1629 1630 1631 1632
  if (extended_insert)
    dynstr_free(&extended_row);
  my_end(0);
  return(first_error);
} /* main */