mysqldump.c 58.8 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
** 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>
33
** Tõnu 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
** 10 Jun 2003: SET NAMES and --no-set-names by Alexander Barkov
unknown's avatar
unknown committed
38
*/
unknown's avatar
unknown committed
39

unknown's avatar
unknown committed
40
#define DUMP_VERSION "10.7"
unknown's avatar
unknown committed
41

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

48
#include "client_priv.h"
unknown's avatar
unknown committed
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
#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

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

unknown's avatar
unknown committed
71 72
static char *add_load_option(char *ptr, const char *object,
			     const char *statement);
73 74
static ulong find_set(TYPELIB *lib, const char *x, uint length,
		      char **err_pos, uint *err_len);
unknown's avatar
unknown committed
75 76

static char *field_escape(char *to,const char *from,uint length);
77 78 79 80
static my_bool  verbose=0,tFlag=0,cFlag=0,dFlag=0,quick= 1, extended_insert= 1,
		lock_tables=1,ignore_errors=0,flush_logs=0,replace=0,
		ignore=0,opt_drop=1,opt_keywords=0,opt_lock=1,opt_compress=0,
                opt_delayed=0,create_options=1,opt_quoted=0,opt_databases=0,
81
                opt_alldbs=0,opt_create_db=0,opt_first_slave=0,opt_set_charset,
unknown's avatar
unknown committed
82 83 84 85 86
		opt_autocommit=0,opt_master_data,opt_disable_keys=1,opt_xml=0,
		opt_delete_master_logs=0, tty_password=0,
		opt_single_transaction=0, opt_comments= 0, opt_compact= 0;

static MYSQL mysql_connection,*sock=0;
87
static char  insert_pat[12 * 1024],*opt_password=0,*current_user=0,
unknown's avatar
unknown committed
88 89
             *current_host=0,*path=0,*fields_terminated=0,
             *lines_terminated=0, *enclosed=0, *opt_enclosed=0, *escaped=0,
unknown's avatar
unknown committed
90
             *where=0,
91
             *opt_compatible_mode_str= 0,
92
             *err_ptr= 0;
93
static char compatible_mode_normal_str[255];
94 95
static ulong opt_compatible_mode= 0;
static uint     opt_mysql_port= 0, err_len= 0;
unknown's avatar
unknown committed
96 97 98 99
static my_string opt_mysql_unix_port=0;
static int   first_error=0;
extern ulong net_buffer_length;
static DYNAMIC_STRING extended_row;
100
#include <sslopt-vars.h>
unknown's avatar
unknown committed
101
FILE  *md_result_file;
102 103 104
#ifdef HAVE_SMEM
static char *shared_memory_base_name=0;
#endif
105
static uint opt_protocol= 0;
unknown's avatar
unknown committed
106
static char *default_charset= (char*) MYSQL_UNIVERSAL_CLIENT_CHARSET;
107
static CHARSET_INFO *charset_info= &my_charset_latin1;
unknown's avatar
unknown committed
108
const char *default_dbug_option="d:t:o,/tmp/mysqldump.trace";
109 110 111 112

const char *compatible_mode_names[]=
{
  "MYSQL323", "MYSQL40", "POSTGRESQL", "ORACLE", "MSSQL", "DB2",
113 114
  "MAXDB", "NO_KEY_OPTIONS", "NO_TABLE_OPTIONS", "NO_FIELD_OPTIONS",
  "ANSI",
115 116
  NullS
};
117 118 119 120 121 122 123 124 125
#define MASK_ANSI_QUOTES \
(\
 (1<<2)  | /* POSTGRESQL */\
 (1<<3)  | /* ORACLE     */\
 (1<<4)  | /* MSSQL      */\
 (1<<5)  | /* DB2        */\
 (1<<6)  | /* MAXDB      */\
 (1<<10)   /* ANSI       */\
)
126 127 128
TYPELIB compatible_mode_typelib= {array_elements(compatible_mode_names) - 1,
				  "", compatible_mode_names};

unknown's avatar
unknown committed
129

130
static struct my_option my_long_options[] =
unknown's avatar
unknown committed
131
{
132 133 134 135 136
  {"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},
  {"add-drop-table", OPT_DROP, "Add a 'drop table' before each create.",
137
   (gptr*) &opt_drop, (gptr*) &opt_drop, 0, GET_BOOL, NO_ARG, 1, 0, 0, 0, 0,
138 139
   0},
  {"add-locks", OPT_LOCKS, "Add locks around insert statements.",
140
   (gptr*) &opt_lock, (gptr*) &opt_lock, 0, GET_BOOL, NO_ARG, 1, 0, 0, 0, 0,
141 142 143 144 145
   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,
146
   "Directory where character sets are.", (gptr*) &charsets_dir,
147
   (gptr*) &charsets_dir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
148
  {"compatible", OPT_COMPATIBLE,
unknown's avatar
unknown committed
149
   "Change the dump to be compatible with a given mode. By default tables are dumped in a format optimized for MySQL. Legal modes are: ansi, mysql323, mysql40, postgresql, oracle, mssql, db2, maxdb, no_key_options, no_table_options, no_field_options. One can use several modes separated by commas. Note: Requires MySQL server version 4.1.0 or higher. This option is ignored with earlier server versions.",
150 151
   (gptr*) &opt_compatible_mode_str, (gptr*) &opt_compatible_mode_str, 0,
   GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
152 153 154 155
  {"compact", OPT_COMPACT,
   "Give less verbose output (useful for debugging). Disables structure comments and header/footer constructs.  Enables options --skip-add-drop-table --no-set-names --skip-disable-keys --skip-lock-tables",
   (gptr*) &opt_compact, (gptr*) &opt_compact, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0,
   0, 0},
156 157
  {"complete-insert", 'c', "Use complete insert statements.", (gptr*) &cFlag,
   (gptr*) &cFlag, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
158 159 160
  {"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},
unknown's avatar
unknown committed
161 162 163 164
  {"create-options", OPT_CREATE_OPTIONS,
   "Include all MySQL specific create options.",
   (gptr*) &create_options, (gptr*) &create_options, 0, GET_BOOL, NO_ARG, 1,
   0, 0, 0, 0, 0},
165 166 167 168
  {"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},
unknown's avatar
unknown committed
169 170 171 172 173 174 175
#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", (gptr*) &default_dbug_option,
   (gptr*) &default_dbug_option, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
#endif
176 177 178 179 180 181
  {"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
182
  {"delete-master-logs", OPT_DELETE_MASTER_LOGS,
unknown's avatar
unknown committed
183
   "Delete logs on master after backup. This automatically enables --first-slave.",
unknown's avatar
unknown committed
184
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
185 186
  {"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,
187
   (gptr*) &opt_disable_keys, 0, GET_BOOL, NO_ARG, 1, 0, 0, 0, 0, 0},
188 189 190
  {"extended-insert", 'e',
   "Allows utilization of the new, much faster INSERT syntax.",
   (gptr*) &extended_insert, (gptr*) &extended_insert, 0, GET_BOOL, NO_ARG,
191
   1, 0, 0, 0, 0, 0},
192 193 194 195 196 197 198 199 200 201
  {"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 ...",
unknown's avatar
unknown committed
202
   (gptr*) &escaped, (gptr*) &escaped, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
203 204 205
  {"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},
unknown's avatar
unknown committed
206 207 208 209
  {"flush-logs", 'F', "Flush logs file in server before starting dump. "
    "Note that if you dump many databases at once (using the option "
    "--databases= or --all-databases), the logs will be flushed for "
    "each database dumped.",
210 211 212 213 214 215 216 217
   (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,
218
   (gptr*) &current_host, 0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
219 220 221 222
  {"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,
223
   (gptr*) &lock_tables, 0, GET_BOOL, NO_ARG, 1, 0, 0, 0, 0, 0},
224
  {"master-data", OPT_MASTER_DATA,
unknown's avatar
unknown committed
225
   "This causes the master position and filename to be appended to your output. This automatically enables --first-slave.",
226 227 228 229 230
   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},
231 232
  {"single-transaction", OPT_TRANSACTION,
   "Dump all tables in single transaction to get consistent snapshot. Mutually exclusive with --lock-tables.",
233
   (gptr*) &opt_single_transaction, (gptr*) &opt_single_transaction, 0,
234
   GET_BOOL, NO_ARG,  0, 0, 0, 0, 0, 0},
235
  {"no-create-db", 'n',
236
   "'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.}.",
237 238 239
   (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.",
240 241 242
   (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},
unknown's avatar
unknown committed
243
  {"no-set-names", 'N',
unknown's avatar
unknown committed
244
   "Deprecated, use --set-charset or --skip-set-charset to enable/disable charset settings instead",
245 246
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"set-charset", OPT_SET_CHARSET,
unknown's avatar
unknown committed
247
   "'SET CHARACTER_SET_CLIENT=default_character_set' will be put in the output",
248 249
   (gptr*) &opt_set_charset, (gptr*) &opt_set_charset, 0, GET_BOOL, NO_ARG, 1,
   0, 0, 0, 0, 0},
250
  {"set-variable", 'O',
unknown's avatar
unknown committed
251
   "Change the value of a variable. Please note that this option is deprecated; you can set variables directly with --variable-name=value.",
252 253
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"opt", OPT_OPTIMIZE,
254
   "Same as --add-drop-table, --add-locks, --create-options, --quick, --extended-insert, --lock-tables, --set-charset, and --disable-keys. Enabled by default, disable with --skip-opt.",
255 256 257 258
   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
259
#ifdef __WIN__
260
  {"pipe", 'W', "Use named pipes to connect to server.", 0, 0, 0, GET_NO_ARG,
261
   NO_ARG, 0, 0, 0, 0, 0, 0},
unknown's avatar
unknown committed
262
#endif
263 264 265
  {"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},
266
  {"protocol", OPT_MYSQL_PROTOCOL, "The protocol of connection (tcp,socket,pipe,memory).",
267
   0, 0, 0, GET_STR,  REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
268
  {"quick", 'q', "Don't buffer query, dump directly to stdout.",
269
   (gptr*) &quick, (gptr*) &quick, 0, GET_BOOL, NO_ARG, 1, 0, 0, 0, 0, 0},
270
  {"quote-names",'Q', "Quote table and column names with backticks (`).",
271
   (gptr*) &opt_quoted, (gptr*) &opt_quoted, 0, GET_BOOL, NO_ARG, 1, 0, 0, 0,
272 273
   0, 0},
  {"result-file", 'r',
unknown's avatar
unknown committed
274
   "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
275
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
276 277
#ifdef HAVE_SMEM
  {"shared_memory_base_name", OPT_SHARED_MEMORY_BASE_NAME,
278
   "Base name of shared memory.", (gptr*) &shared_memory_base_name, (gptr*) &shared_memory_base_name,
279 280
   0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
#endif
281
  {"skip-opt", OPT_SKIP_OPTIMIZATION,
282
   "Disable --opt. Disables --add-drop-table, --add-locks, --create-options, --quick, --extended-insert, --lock-tables, --set-charset, and --disable-keys.",
283
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
284 285 286
  {"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},
287
#include <sslopt-longopts.h>
288 289 290 291 292
  {"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
293
#ifndef DONT_ALLOW_USER_CHANGE
294 295 296
  {"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
297
#endif
298 299 300 301 302 303 304 305
  {"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},
306
  {"max_allowed_packet", OPT_MAX_ALLOWED_PACKET, "",
307
    (gptr*) &max_allowed_packet, (gptr*) &max_allowed_packet, 0,
308 309
    GET_ULONG, REQUIRED_ARG, 24*1024*1024, 4096, 
   (longlong) 2L*1024L*1024L*1024L, MALLOC_OVERHEAD, 1024, 0},
310
  {"net_buffer_length", OPT_NET_BUFFER_LENGTH, "",
311
    (gptr*) &net_buffer_length, (gptr*) &net_buffer_length, 0,
312
    GET_ULONG, REQUIRED_ARG, 1024*1024L-1025, 4096, 16*1024L*1024L,
unknown's avatar
merge  
unknown committed
313
    MALLOC_OVERHEAD-1024, 1024, 0},
314 315 316
  {"comments", 'i', "Write additional information.",
   (gptr*) &opt_comments, (gptr*) &opt_comments, 0, GET_BOOL, NO_ARG,
   1, 0, 0, 0, 0, 0},
317
  {0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
unknown's avatar
unknown committed
318 319 320 321 322
};

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

static void safe_exit(int error);
323
static void write_header(FILE *sql_file, char *db_name);
unknown's avatar
unknown committed
324 325 326 327 328 329 330 331
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
332
static char *quote_name(const char *name, char *buff, my_bool force);
333 334
static const char *check_if_ignore_table(const char *table_name);

unknown's avatar
unknown committed
335 336 337 338 339 340 341 342

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


343 344 345 346 347 348 349 350
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
351 352 353 354 355 356
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");
357
  short_usage_sub();
unknown's avatar
unknown committed
358
  print_defaults("my",load_default_groups);
359 360
  my_print_help(my_long_options);
  my_print_variables(my_long_options);
unknown's avatar
unknown committed
361 362 363
} /* usage */


364 365 366 367 368 369 370
static void short_usage(void)
{
  short_usage_sub();
  printf("For more options, use %s --help\n", my_progname);
}


371
static void write_header(FILE *sql_file, char *db_name)
unknown's avatar
unknown committed
372
{
unknown's avatar
unknown committed
373
  if (opt_xml)
unknown's avatar
unknown committed
374
  {
375 376
    fputs("<?xml version=\"1.0\"?>\n", sql_file);
    fputs("<mysqldump>\n", sql_file);
unknown's avatar
unknown committed
377
  }
unknown's avatar
unknown committed
378
  else if (!opt_compact)
unknown's avatar
unknown committed
379
  {
unknown's avatar
unknown committed
380 381 382 383 384 385 386 387 388 389 390
    if (opt_comments)
    {
      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 :
	      "");
      fputs("-- ------------------------------------------------------\n",
	    sql_file);
      fprintf(sql_file, "-- Server version\t%s\n",
	      mysql_get_server_info(&mysql_connection));
    }
391
    if (opt_set_charset)
392
      fprintf(sql_file,"\n/*!40101 SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT, CHARACTER_SET_CLIENT=%s */;\n",default_charset);
393 394 395
    if (!path)
    {
      fprintf(md_result_file,"\
396 397 398
/*!40014 SET @OLD_UNIQUE_CHECKS=@@UNIQUE_CHECKS, UNIQUE_CHECKS=0 */;\n\
/*!40014 SET @OLD_FOREIGN_KEY_CHECKS=@@FOREIGN_KEY_CHECKS, FOREIGN_KEY_CHECKS=0 */;\n\
");
399 400
    }
    fprintf(sql_file,
unknown's avatar
Merge  
unknown committed
401 402 403
	    "/*!40101 SET @OLD_SQL_MODE=@@SQL_MODE, SQL_MODE=\"%s%s%s\" */;\n",
	    path?"":"NO_AUTO_VALUE_ON_ZERO",compatible_mode_normal_str[0]==0?"":",",
	    compatible_mode_normal_str);
unknown's avatar
unknown committed
404
  }
405
} /* write_header */
unknown's avatar
unknown committed
406

407

unknown's avatar
unknown committed
408 409 410
static void write_footer(FILE *sql_file)
{
  if (opt_xml)
unknown's avatar
unknown committed
411 412
    fputs("</mysqldump>\n", sql_file);
  else if (!opt_compact)
413
  {
414 415 416 417
    fprintf(sql_file,"\n/*!40101 SET SQL_MODE=@OLD_SQL_MODE */;\n");
    if (!path)
    {
      fprintf(md_result_file,"\
418
/*!40014 SET FOREIGN_KEY_CHECKS=@OLD_FOREIGN_KEY_CHECKS */;\n\
unknown's avatar
unknown committed
419
/*!40014 SET UNIQUE_CHECKS=@OLD_UNIQUE_CHECKS */;\n");
420
    }
421
    if (opt_set_charset)
422
      fprintf(sql_file,
unknown's avatar
unknown committed
423 424
	      "/*!40101 SET CHARACTER_SET_CLIENT=@OLD_CHARACTER_SET_CLIENT */;\n");
    fputs("\n", sql_file);
425
  }
unknown's avatar
unknown committed
426
} /* write_footer */
unknown's avatar
unknown committed
427

428

unknown's avatar
unknown committed
429

430 431 432
static my_bool
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
	       char *argument)
unknown's avatar
unknown committed
433
{
unknown's avatar
unknown committed
434
  switch (optid) {
435 436 437 438
  case OPT_MASTER_DATA:
    opt_master_data=1;
    opt_first_slave=1;
    break;
unknown's avatar
unknown committed
439 440 441 442
  case OPT_DELETE_MASTER_LOGS:
    opt_delete_master_logs=1;
    opt_first_slave=1;
    break;
443 444 445 446 447 448 449 450 451 452 453 454 455 456
  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
457
    if (!(md_result_file = my_fopen(argument, O_WRONLY | FILE_BINARY,
458 459 460 461
				    MYF(MY_WME))))
      exit(1);
    break;
  case 'W':
unknown's avatar
unknown committed
462
#ifdef __WIN__
463
    opt_protocol = MYSQL_PROTOCOL_PIPE;
unknown's avatar
unknown committed
464
#endif
465
    break;
466 467 468
  case 'N':
    opt_set_charset= 0;
    break;
469 470 471 472
  case 'T':
    opt_disable_keys=0;
    break;
  case '#':
unknown's avatar
unknown committed
473
    DBUG_PUSH(argument ? argument : default_dbug_option);
474
    break;
475
#include <sslopt-case.h>
476 477 478
  case 'V': print_version(); exit(0);
  case 'X':
    opt_xml = 1;
479 480
    extended_insert= opt_drop= opt_lock= 
      opt_disable_keys= opt_autocommit= opt_create_db= 0;
481 482 483 484 485 486
    break;
  case 'I':
  case '?':
    usage();
    exit(0);
  case (int) OPT_OPTIMIZE:
487
    extended_insert= opt_drop= opt_lock= quick= create_options=
488
      opt_disable_keys= lock_tables= opt_set_charset= 1;
489
    if (opt_single_transaction) lock_tables=0;
490
    break;
491 492
  case (int) OPT_SKIP_OPTIMIZATION:
    extended_insert= opt_drop= opt_lock= quick= create_options=
unknown's avatar
unknown committed
493
      opt_disable_keys= lock_tables= opt_set_charset= 0;
494
    break;
unknown's avatar
unknown committed
495 496 497 498
  case (int) OPT_COMPACT:
  if (opt_compact)
  {
    opt_comments= opt_drop= opt_disable_keys= opt_lock= 0;
499
    opt_set_charset= 0;
unknown's avatar
unknown committed
500
  }
501 502 503
  case (int) OPT_TABLES:
    opt_databases=0;
    break;
504 505 506
  case (int) OPT_COMPATIBLE:
    {  
      char buff[255];
507 508 509
      char *end= compatible_mode_normal_str;
      int i;
      ulong mode;
510 511

      opt_quoted= 1;
512
      opt_set_charset= 0;
513 514 515 516 517 518 519 520 521 522
      opt_compatible_mode_str= argument;
      opt_compatible_mode= find_set(&compatible_mode_typelib,
				    argument, strlen(argument),
				    &err_ptr, &err_len);
      if (err_len)
      {
	strmake(buff, err_ptr, min(sizeof(buff), err_len));
	fprintf(stderr, "Invalid mode to --compatible: %s\n", buff);
	exit(1);
      }
523 524
#if !defined(DBUG_OFF)
      {
unknown's avatar
unknown committed
525
	uint size_for_sql_mode= 0;
526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543
	const char **ptr;
	for (ptr= compatible_mode_names; *ptr; ptr++)
	  size_for_sql_mode+= strlen(*ptr);
	size_for_sql_mode+= sizeof(compatible_mode_names)-1;
	DBUG_ASSERT(sizeof(compatible_mode_normal_str)>=size_for_sql_mode);
      }
#endif
      mode= opt_compatible_mode;
      for (i= 0, mode= opt_compatible_mode; mode; mode>>= 1, i++)
      {
	if (mode & 1)
	{
	  end= strmov(end, compatible_mode_names[i]);
	  end= strmov(end, ",");
	}
      }
      if (end!=compatible_mode_normal_str)
	end[-1]= 0;
544 545 546
      break;
    }
  case (int) OPT_MYSQL_PROTOCOL:
547
    {
548 549 550 551 552 553 554
      if ((opt_protocol= find_type(argument, &sql_protocol_typelib, 0))
	  == ~(ulong) 0)
      {
	fprintf(stderr, "Unknown option to protocol: %s\n", argument);
	exit(1);
      }
      break;
555
    }
unknown's avatar
unknown committed
556
  }
557 558 559 560 561 562 563 564 565 566 567
  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)))
568
    exit(ho_error);
569

unknown's avatar
unknown committed
570 571 572 573 574
  if (opt_delayed)
    opt_lock=0;				/* Can't have lock with delayed */
  if (!path && (enclosed || opt_enclosed || escaped || lines_terminated ||
		fields_terminated))
  {
575 576
    fprintf(stderr,
	    "%s: You must use option --tab with --fields-...\n", my_progname);
unknown's avatar
unknown committed
577 578
    return(1);
  }
579 580
  if (opt_single_transaction)
    lock_tables= 0;
unknown's avatar
unknown committed
581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597
  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);
  }
598 599
  if (strcmp(default_charset, charset_info->csname) &&
      !(charset_info= get_charset_by_csname(default_charset, 
600
  					    MY_CS_PRIMARY, MYF(MY_WME))))
601
    exit(1);
unknown's avatar
unknown committed
602 603
  if ((*argc < 1 && !opt_alldbs) || (*argc > 0 && opt_alldbs))
  {
604
    short_usage();
unknown's avatar
unknown committed
605 606 607
    return 1;
  }
  if (tty_password)
608
    opt_password=get_tty_password(NullS);
unknown's avatar
unknown committed
609 610 611 612 613 614 615 616 617 618
  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
619
  my_printf_error(0,"Got error: %d: %s %s", MYF(0),
unknown's avatar
unknown committed
620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644
		  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)
{
645
  char buff[20+FN_REFLEN];
unknown's avatar
unknown committed
646 647 648
  DBUG_ENTER("dbConnect");
  if (verbose)
  {
649
    fprintf(stderr, "-- Connecting to %s...\n", host ? host : "localhost");
unknown's avatar
unknown committed
650 651 652 653 654 655 656
  }
  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,
657
		  opt_ssl_capath, opt_ssl_cipher);
658 659 660 661 662 663
#endif
  if (opt_protocol)
    mysql_options(&mysql_connection,MYSQL_OPT_PROTOCOL,(char*)&opt_protocol);
#ifdef HAVE_SMEM
  if (shared_memory_base_name)
    mysql_options(&mysql_connection,MYSQL_SHARED_MEMORY_BASE_NAME,shared_memory_base_name);
unknown's avatar
unknown committed
664
#endif
665
  mysql_options(&mysql_connection, MYSQL_SET_CHARSET_NAME, default_charset);
unknown's avatar
unknown committed
666 667 668 669 670 671 672
  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;
  }
673 674 675 676 677 678 679 680 681 682
  sprintf(buff, "/*!40100 SET @@SQL_MODE=\"%s\" */",
	  compatible_mode_normal_str);
  if (mysql_query(sock, buff))
  {
    fprintf(stderr, "%s: Can't set the compatible mode %s (error %s)\n",
	    my_progname, compatible_mode_normal_str, mysql_error(sock));
    mysql_close(sock);
    safe_exit(EX_MYSQLERR);
    return 1;
  }
unknown's avatar
unknown committed
683 684 685 686 687 688 689 690 691 692
  return 0;
} /* dbConnect */


/*
** dbDisconnect -- disconnects from the host.
*/
static void dbDisconnect(char *host)
{
  if (verbose)
693
    fprintf(stderr, "-- Disconnecting from %s...\n", host ? host : "localhost");
unknown's avatar
unknown committed
694 695 696 697 698 699 700 701 702 703 704 705 706
  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
707
  mysql_real_escape_string(&mysql_connection, tmp, pos, length);
unknown's avatar
unknown committed
708 709 710 711 712 713 714 715 716 717 718 719
  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++)
720
    if (!my_isvar(charset_info,*str) && *str != '$')
unknown's avatar
unknown committed
721 722 723 724 725
      return 1;
#endif
  return 0;
} /* test_if_special_chars */

unknown's avatar
unknown committed
726

unknown's avatar
unknown committed
727

unknown's avatar
unknown committed
728
static char *quote_name(const char *name, char *buff, my_bool force)
unknown's avatar
unknown committed
729
{
unknown's avatar
unknown committed
730
  char *to= buff;
731 732
  char qtype= (opt_compatible_mode & MASK_ANSI_QUOTES) ? '\"' : '`';

unknown's avatar
unknown committed
733 734
  if (!force && !opt_quoted && !test_if_special_chars(name))
    return (char*) name;
735
  *to++= qtype;
unknown's avatar
unknown committed
736 737
  while (*name)
  {
738 739
    if (*name == qtype)
      *to++= qtype;
unknown's avatar
unknown committed
740 741
    *to++= *name++;
  }
742 743
  to[0]= qtype;
  to[1]= 0;
unknown's avatar
unknown committed
744 745 746 747
  return buff;
} /* quote_name */


748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764

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


765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856
/*
  Quote and print a string.
  
  SYNOPSIS
    print_quoted_xml()
    output	- output file
    str		- string to print
    len		- its length
    
  DESCRIPTION
    Quote '<' '>' '&' '\"' singns and print a string to the xml_file.
*/

static void print_quoted_xml(FILE *xml_file, const char *str, ulong len)
{
  const char *end;
  
  for (end= str + len; str != end; str++)
  {
    switch (*str) {
    case '<':
      fputs("&lt;", xml_file);
      break;
    case '>':
      fputs("&gt;", xml_file);
      break;
    case '&':
      fputs("&amp;", xml_file);
      break;
    case '\"':
      fputs("&quot;", xml_file);
      break;
    default:
      fputc(*str, xml_file);
      break;
    }
  }
}


/*
  Print xml tag with one attribute.
  
  SYNOPSIS
    print_xml_tag1()
    xml_file	- output file
    sbeg	- line beginning
    stag_atr	- tag and attribute
    sval	- value of attribute
    send	- line ending
    
  DESCRIPTION
    Print tag with one attribute to the xml_file. Format is:
      sbeg<stag_atr="sval">send
  NOTE
    sval MUST be a NULL terminated string.
    sval string will be qouted before output.
*/

static void print_xml_tag1(FILE * xml_file, const char* sbeg,
			   const char* stag_atr, const char* sval,
			   const char* send)
{
  fputs(sbeg, xml_file);
  fputs("<", xml_file);
  fputs(stag_atr, xml_file);
  fputs("\"", xml_file);
  print_quoted_xml(xml_file, sval, strlen(sval));
  fputs("\">", xml_file);
  fputs(send, xml_file);
}


/*
  Print xml tag with many attributes.

  SYNOPSIS
    print_xml_row()
    xml_file	- output file
    row_name	- xml tag name
    tableRes	- query result
    row		- result row
    
  DESCRIPTION
    Print tag with many attribute to the xml_file. Format is:
      \t\t<row_name Atr1="Val1" Atr2="Val2"... />
  NOTE
    All atributes and values will be quoted before output.
*/

static void print_xml_row(FILE *xml_file, const char *row_name,
			  MYSQL_RES *tableRes, MYSQL_ROW *row)
857 858 859 860 861 862 863 864 865
{
  uint i;
  MYSQL_FIELD *field;
  ulong *lengths= mysql_fetch_lengths(tableRes);
  
  fprintf(xml_file, "\t\t<%s", row_name);
  mysql_field_seek(tableRes, 0);
  for (i= 0; (field= mysql_fetch_field(tableRes)); i++)
  {
866
    if ((*row)[i])
867
    {
unknown's avatar
unknown committed
868
      fputc(' ', xml_file);
869
      print_quoted_xml(xml_file, field->name, field->name_length);
870 871
      fputs("=\"", xml_file);
      print_quoted_xml(xml_file, (*row)[i], lengths[i]);
unknown's avatar
unknown committed
872
      fputc('"', xml_file);
873 874 875 876 877
    }
  }
  fputs(" />\n", xml_file);
}

unknown's avatar
unknown committed
878
/*
unknown's avatar
unknown committed
879 880 881 882 883
  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
884
*/
unknown's avatar
unknown committed
885

unknown's avatar
unknown committed
886 887 888 889 890 891
static uint getTableStructure(char *table, char* db)
{
  MYSQL_RES  *tableRes;
  MYSQL_ROW  row;
  my_bool    init=0;
  uint       numFields;
unknown's avatar
unknown committed
892
  char	     *strpos, *result_table, *opt_quoted_table;
unknown's avatar
unknown committed
893
  const char *delayed;
unknown's avatar
unknown committed
894 895
  char	     name_buff[NAME_LEN+3],table_buff[NAME_LEN*2+3];
  char	     table_buff2[NAME_LEN*2+3];
unknown's avatar
unknown committed
896
  FILE       *sql_file = md_result_file;
unknown's avatar
unknown committed
897 898 899 900 901
  DBUG_ENTER("getTableStructure");

  delayed= opt_delayed ? " DELAYED " : "";

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

unknown's avatar
unknown committed
904 905
  sprintf(insert_pat,"SET OPTION SQL_QUOTE_SHOW_CREATE=%d",
	  (opt_quoted || opt_keywords));
unknown's avatar
unknown committed
906
  if (!create_options)
unknown's avatar
unknown committed
907
    strmov(strend(insert_pat), "/*!40102 ,SQL_MODE=concat(@@sql_mode, _utf8 ',NO_KEY_OPTIONS,NO_TABLE_OPTIONS,NO_FIELD_OPTIONS') */");
unknown's avatar
unknown committed
908

unknown's avatar
unknown committed
909 910
  result_table=     quote_name(table, table_buff, 1);
  opt_quoted_table= quote_name(table, table_buff2, 0);
911
  if (!opt_xml && !mysql_query(sock,insert_pat))
unknown's avatar
unknown committed
912
  {
913 914
    /* using SHOW CREATE statement */
    if (!tFlag)
unknown's avatar
unknown committed
915
    {
916 917 918
      /* Make an sql-file, if path was given iow. option -T was given */
      char buff[20+FN_REFLEN];

unknown's avatar
unknown committed
919
      sprintf(buff,"show create table %s", result_table);
920
      if (mysql_query(sock, buff))
unknown's avatar
unknown committed
921
      {
unknown's avatar
unknown committed
922 923
        fprintf(stderr, "%s: Can't get CREATE TABLE for table %s (%s)\n",
		      my_progname, result_table, mysql_error(sock));
924 925
        safe_exit(EX_MYSQLERR);
        DBUG_RETURN(0);
unknown's avatar
unknown committed
926 927
      }

928 929 930
      if (path)
      {
        char filename[FN_REFLEN], tmp_path[FN_REFLEN];
931
        convert_dirname(tmp_path,path,NullS);
932
        sql_file= my_fopen(fn_format(filename, table, tmp_path, ".sql", 4),
933
				 O_WRONLY, MYF(MY_WME));
934 935
        if (!sql_file)			/* If file couldn't be opened */
        {
unknown's avatar
unknown committed
936 937
	  safe_exit(EX_MYSQLERR);
	  DBUG_RETURN(0);
938
        }
939
        write_header(sql_file, db);
940
      }
941
      if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
942
        fprintf(sql_file, "\n--\n-- Table structure for table %s\n--\n\n",
unknown's avatar
unknown committed
943
		result_table);
944
      if (opt_drop)
unknown's avatar
unknown committed
945
        fprintf(sql_file, "DROP TABLE IF EXISTS %s;\n", opt_quoted_table);
946 947 948

      tableRes=mysql_store_result(sock);
      row=mysql_fetch_row(tableRes);
949
      fprintf(sql_file, "%s;\n", row[1]);
950 951
      mysql_free_result(tableRes);
    }
unknown's avatar
unknown committed
952
    sprintf(insert_pat,"show fields from %s", result_table);
953
    if (mysql_query(sock,insert_pat) || !(tableRes=mysql_store_result(sock)))
unknown's avatar
unknown committed
954
    {
unknown's avatar
unknown committed
955 956
      fprintf(stderr, "%s: Can't get info about table: %s\nerror: %s\n",
	      my_progname, result_table, mysql_error(sock));
957 958
      if (path)
	my_fclose(sql_file, MYF(MY_WME));
959 960
      safe_exit(EX_MYSQLERR);
      DBUG_RETURN(0);
unknown's avatar
unknown committed
961
    }
962

unknown's avatar
unknown committed
963
    if (cFlag)
unknown's avatar
unknown committed
964
      sprintf(insert_pat, "INSERT %sINTO %s (", delayed, opt_quoted_table);
965
    else
unknown's avatar
unknown committed
966
    {
unknown's avatar
unknown committed
967 968
      sprintf(insert_pat, "INSERT %sINTO %s VALUES ", delayed,
	      opt_quoted_table);
969 970 971 972 973 974 975 976
      if (!extended_insert)
        strcat(insert_pat,"(");
    }

    strpos=strend(insert_pat);
    while ((row=mysql_fetch_row(tableRes)))
    {
      if (init)
unknown's avatar
unknown committed
977
      {
978
        if (cFlag)
979
	  strpos=strmov(strpos,", ");
unknown's avatar
unknown committed
980
      }
981 982
      init=1;
      if (cFlag)
unknown's avatar
unknown committed
983
        strpos=strmov(strpos,quote_name(row[SHOW_FIELDNAME], name_buff, 0));
unknown's avatar
unknown committed
984
    }
985 986
    numFields = (uint) mysql_num_rows(tableRes);
    mysql_free_result(tableRes);
unknown's avatar
unknown committed
987
  }
988
  else
unknown's avatar
unknown committed
989
  {
unknown's avatar
unknown committed
990 991 992 993
    if (verbose)
      fprintf(stderr,
              "%s: Warning: Can't set SQL_QUOTE_SHOW_CREATE option (%s)\n",
              my_progname, mysql_error(sock));
994

unknown's avatar
unknown committed
995
    sprintf(insert_pat,"show fields from %s", result_table);
996
    if (mysql_query(sock,insert_pat) || !(tableRes=mysql_store_result(sock)))
unknown's avatar
unknown committed
997
    {
unknown's avatar
unknown committed
998 999
      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
1000 1001 1002 1003
      safe_exit(EX_MYSQLERR);
      DBUG_RETURN(0);
    }

1004 1005
    /* Make an sql-file, if path was given iow. option -T was given */
    if (!tFlag)
unknown's avatar
unknown committed
1006
    {
1007
      if (path)
unknown's avatar
unknown committed
1008
      {
1009
        char filename[FN_REFLEN], tmp_path[FN_REFLEN];
1010
        convert_dirname(tmp_path,path,NullS);
1011 1012 1013 1014
        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 */
        {
1015 1016
	  safe_exit(EX_MYSQLERR);
	  DBUG_RETURN(0);
1017
        }
1018
        write_header(sql_file, db);
unknown's avatar
unknown committed
1019
      }
1020
      if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
1021 1022
	fprintf(sql_file, "\n--\n-- Table structure for table %s\n--\n\n",
		result_table);
1023
      if (opt_drop)
1024
        fprintf(sql_file, "DROP TABLE IF EXISTS %s;\n",result_table);
1025 1026 1027
      if (!opt_xml)
	fprintf(sql_file, "CREATE TABLE %s (\n", result_table);
      else
1028
        print_xml_tag1(sql_file, "\t", "table_structure name=", table, "\n");
1029 1030
    }
    if (cFlag)
1031
      sprintf(insert_pat, "INSERT %sINTO %s (", delayed, result_table);
1032 1033
    else
    {
1034
      sprintf(insert_pat, "INSERT %sINTO %s VALUES ", delayed, result_table);
1035 1036
      if (!extended_insert)
        strcat(insert_pat,"(");
unknown's avatar
unknown committed
1037
    }
1038 1039

    strpos=strend(insert_pat);
unknown's avatar
unknown committed
1040 1041
    while ((row=mysql_fetch_row(tableRes)))
    {
1042 1043
      ulong *lengths=mysql_fetch_lengths(tableRes);
      if (init)
unknown's avatar
unknown committed
1044
      {
1045
        if (!opt_xml && !tFlag)
1046
	  fputs(",\n",sql_file);
1047
        if (cFlag)
1048
	  strpos=strmov(strpos,", ");
1049 1050 1051
      }
      init=1;
      if (cFlag)
unknown's avatar
unknown committed
1052
        strpos=strmov(strpos,quote_name(row[SHOW_FIELDNAME], name_buff, 0));
1053 1054
      if (!tFlag)
      {
1055 1056 1057 1058 1059 1060
	if (opt_xml)
	{
	  print_xml_row(sql_file, "field", tableRes, &row);
	  continue;
	}
	
1061
        if (opt_keywords)
1062
	  fprintf(sql_file, "  %s.%s %s", result_table,
unknown's avatar
unknown committed
1063 1064
		  quote_name(row[SHOW_FIELDNAME],name_buff, 0),
		  row[SHOW_TYPE]);
1065
        else
unknown's avatar
unknown committed
1066
	  fprintf(sql_file, "  %s %s", quote_name(row[SHOW_FIELDNAME],
unknown's avatar
unknown committed
1067 1068
						  name_buff, 0),
		  row[SHOW_TYPE]);
1069 1070
        if (row[SHOW_DEFAULT])
        {
unknown's avatar
unknown committed
1071
	  fputs(" DEFAULT ", sql_file);
unknown's avatar
unknown committed
1072
	  unescape(sql_file, row[SHOW_DEFAULT], lengths[SHOW_DEFAULT]);
1073 1074
        }
        if (!row[SHOW_NULL][0])
unknown's avatar
unknown committed
1075
	  fputs(" NOT NULL", sql_file);
1076
        if (row[SHOW_EXTRA][0])
unknown's avatar
unknown committed
1077
	  fprintf(sql_file, " %s",row[SHOW_EXTRA]);
unknown's avatar
unknown committed
1078 1079
      }
    }
1080 1081 1082
    numFields = (uint) mysql_num_rows(tableRes);
    mysql_free_result(tableRes);
    if (!tFlag)
unknown's avatar
unknown committed
1083
    {
1084 1085 1086
      /* 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
1087
      sprintf(buff,"show keys from %s", result_table);
unknown's avatar
unknown committed
1088 1089
      if (mysql_query(sock, buff))
      {
unknown's avatar
unknown committed
1090 1091
        fprintf(stderr, "%s: Can't get keys for table %s (%s)\n",
		my_progname, result_table, mysql_error(sock));
1092
        if (path)
unknown's avatar
unknown committed
1093
	  my_fclose(sql_file, MYF(MY_WME));
1094 1095
        safe_exit(EX_MYSQLERR);
        DBUG_RETURN(0);
unknown's avatar
unknown committed
1096
      }
1097 1098 1099 1100 1101 1102

      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
1103
      {
1104 1105
        if (atoi(row[3]) == 1)
        {
unknown's avatar
unknown committed
1106 1107 1108 1109 1110 1111 1112 1113 1114 1115
	  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;
	  }
1116
        }
unknown's avatar
unknown committed
1117
      }
1118 1119 1120 1121
      mysql_data_seek(tableRes,0);
      keynr=0;
      while ((row=mysql_fetch_row(tableRes)))
      {
1122 1123 1124 1125 1126 1127
	if (opt_xml)
	{
	  print_xml_row(sql_file, "key", tableRes, &row);
	  continue;
	}
        
1128 1129
        if (atoi(row[3]) == 1)
        {
unknown's avatar
unknown committed
1130 1131 1132 1133
	  if (keynr++)
	    putc(')', sql_file);
	  if (atoi(row[1]))       /* Test if duplicate key */
	    /* Duplicate allowed */
unknown's avatar
unknown committed
1134
	    fprintf(sql_file, ",\n  KEY %s (",quote_name(row[2],name_buff,0));
unknown's avatar
unknown committed
1135 1136 1137
	  else if (keynr == primary_key)
	    fputs(",\n  PRIMARY KEY (",sql_file); /* First UNIQUE is primary */
	  else
unknown's avatar
unknown committed
1138 1139
	    fprintf(sql_file, ",\n  UNIQUE %s (",quote_name(row[2],name_buff,
							    0));
1140 1141
        }
        else
unknown's avatar
unknown committed
1142
	  putc(',', sql_file);
unknown's avatar
unknown committed
1143
        fputs(quote_name(row[4], name_buff, 0), sql_file);
1144
        if (row[7])
unknown's avatar
unknown committed
1145
	  fprintf(sql_file, " (%s)",row[7]);      /* Sub key */
1146
      }
1147 1148 1149 1150 1151 1152
      if (!opt_xml)
      {
	if (keynr)
	  putc(')', sql_file);
	fputs("\n)",sql_file);
      }
1153 1154 1155

      /* Get MySQL specific create options */
      if (create_options)
unknown's avatar
unknown committed
1156
      {
1157 1158 1159
	char show_name_buff[FN_REFLEN];
        sprintf(buff,"show table status like %s",
		quote_for_like(table, show_name_buff));
1160 1161
        if (mysql_query(sock, buff))
        {
unknown's avatar
unknown committed
1162 1163 1164 1165
	  if (mysql_errno(sock) != ER_PARSE_ERROR)
	  {					/* If old MySQL version */
	    if (verbose)
	      fprintf(stderr,
unknown's avatar
unknown committed
1166 1167
		      "-- Warning: Couldn't get status information for table %s (%s)\n",
		      result_table,mysql_error(sock));
unknown's avatar
unknown committed
1168
	  }
1169 1170
        }
        else if (!(tableRes=mysql_store_result(sock)) ||
unknown's avatar
unknown committed
1171
		 !(row=mysql_fetch_row(tableRes)))
1172
        {
unknown's avatar
unknown committed
1173
	  fprintf(stderr,
unknown's avatar
unknown committed
1174 1175
		  "Error: Couldn't read status information for table %s (%s)\n",
		  result_table,mysql_error(sock));
1176 1177 1178
        }
        else
        {
1179 1180 1181 1182 1183 1184 1185
	  if (opt_xml)
	  {
	    print_xml_row(sql_file, "options", tableRes, &row);
	  }
	  else
	  {
	    fputs("/*!",sql_file);
unknown's avatar
unknown committed
1186
	    print_value(sql_file,tableRes,row,"engine=","Engine",0);
1187 1188 1189 1190
	    print_value(sql_file,tableRes,row,"","Create_options",0);
	    print_value(sql_file,tableRes,row,"comment=","Comment",1);
	    fputs(" */",sql_file);
	  }
1191 1192
        }
        mysql_free_result(tableRes);		/* Is always safe to free */
unknown's avatar
unknown committed
1193
      }
1194 1195 1196 1197
      if (!opt_xml)
	fputs(";\n", sql_file);
      else
	fputs("\t</table_structure>\n", sql_file);
unknown's avatar
unknown committed
1198 1199 1200 1201 1202 1203 1204 1205
    }
  }
  if (cFlag)
  {
    strpos=strmov(strpos,") VALUES ");
    if (!extended_insert)
      strpos=strmov(strpos,"(");
  }
1206
  if (sql_file != md_result_file)
1207 1208 1209
  {
    fputs("\n", sql_file);
    write_footer(sql_file);
1210
    my_fclose(sql_file, MYF(MY_WME));
1211
  }
unknown's avatar
unknown committed
1212 1213 1214 1215 1216 1217 1218 1219 1220
  DBUG_RETURN(numFields);
} /* getTableStructure */


static char *add_load_option(char *ptr,const char *object,
			     const char *statement)
{
  if (object)
  {
1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
    /* 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
1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271
  }
  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)
{
1272
  char query[QUERY_LENGTH], *end, buff[256],table_buff[NAME_LEN+3];
unknown's avatar
unknown committed
1273
  char *result_table, table_buff2[NAME_LEN*2+3], *opt_quoted_table;
unknown's avatar
unknown committed
1274
  MYSQL_RES	*res;
unknown's avatar
unknown committed
1275 1276
  MYSQL_FIELD	*field;
  MYSQL_ROW	row;
unknown's avatar
unknown committed
1277
  ulong		rownr, row_break, total_length, init_length;
1278
  const char    *table_type;
unknown's avatar
unknown committed
1279

unknown's avatar
unknown committed
1280 1281
  result_table= quote_name(table,table_buff, 1);
  opt_quoted_table= quote_name(table, table_buff2, 0);
1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294

  /* 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
1295 1296 1297
  if (path)
  {
    char filename[FN_REFLEN], tmp_path[FN_REFLEN];
1298
    convert_dirname(tmp_path,path,NullS);
unknown's avatar
unknown committed
1299 1300 1301 1302 1303
    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);
1304 1305
    sprintf(query, "SELECT /*!40001 SQL_NO_CACHE */ * INTO OUTFILE '%s'",
	    filename);
unknown's avatar
unknown committed
1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320
    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
1321
    sprintf(buff," FROM %s", result_table);
unknown's avatar
unknown committed
1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
    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
  {
1333
    if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
1334 1335
      fprintf(md_result_file,"\n--\n-- Dumping data for table %s\n--\n",
	      result_table);
1336
    sprintf(query, "SELECT /*!40001 SQL_NO_CACHE */ * FROM %s",
unknown's avatar
unknown committed
1337
	    result_table);
unknown's avatar
unknown committed
1338 1339
    if (where)
    {
1340
      if (!opt_xml && opt_comments)
unknown's avatar
unknown committed
1341
	fprintf(md_result_file,"-- WHERE:  %s\n",where);
unknown's avatar
unknown committed
1342 1343
      strxmov(strend(query), " WHERE ",where,NullS);
    }
unknown's avatar
unknown committed
1344
    if (!opt_xml && !opt_compact)
unknown's avatar
merge  
unknown committed
1345
      fputs("\n", md_result_file);
unknown's avatar
unknown committed
1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360
    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)
1361
      fprintf(stderr, "-- Retrieving rows...\n");
unknown's avatar
unknown committed
1362 1363
    if (mysql_num_fields(res) != numFields)
    {
unknown's avatar
unknown committed
1364 1365
      fprintf(stderr,"%s: Error in field count for table: %s !  Aborting.\n",
	      my_progname, result_table);
unknown's avatar
unknown committed
1366 1367 1368 1369
      safe_exit(EX_CONSCHECK);
      return;
    }

unknown's avatar
merge  
unknown committed
1370
    if (opt_disable_keys)
unknown's avatar
unknown committed
1371
      fprintf(md_result_file, "\n/*!40000 ALTER TABLE %s DISABLE KEYS */;\n",
unknown's avatar
unknown committed
1372
	      opt_quoted_table);
unknown's avatar
unknown committed
1373
    if (opt_lock)
unknown's avatar
unknown committed
1374
      fprintf(md_result_file,"LOCK TABLES %s WRITE;\n", opt_quoted_table);
unknown's avatar
unknown committed
1375 1376 1377 1378

    total_length=net_buffer_length;		/* Force row break */
    row_break=0;
    rownr=0;
unknown's avatar
unknown committed
1379
    init_length=(uint) strlen(insert_pat)+4;
unknown's avatar
unknown committed
1380
    if (opt_xml)
1381
      print_xml_tag1(md_result_file, "\t", "table_data name=", table, "\n");
unknown's avatar
unknown committed
1382

unknown's avatar
unknown committed
1383 1384 1385
    if (opt_autocommit)
      fprintf(md_result_file, "set autocommit=0;\n");

unknown's avatar
unknown committed
1386 1387 1388 1389 1390
    while ((row=mysql_fetch_row(res)))
    {
      uint i;
      ulong *lengths=mysql_fetch_lengths(res);
      rownr++;
unknown's avatar
unknown committed
1391
      if (!extended_insert && !opt_xml)
unknown's avatar
unknown committed
1392
	fputs(insert_pat,md_result_file);
unknown's avatar
unknown committed
1393 1394
      mysql_field_seek(res,0);

unknown's avatar
merge  
unknown committed
1395
      if (opt_xml)
1396
        fputs("\t<row>\n", md_result_file);
unknown's avatar
merge  
unknown committed
1397

unknown's avatar
unknown committed
1398 1399 1400 1401
      for (i = 0; i < mysql_num_fields(res); i++)
      {
	if (!(field = mysql_fetch_field(res)))
	{
unknown's avatar
unknown committed
1402 1403
	  sprintf(query,"%s: Not enough fields from table %s! Aborting.\n",
		  my_progname, result_table);
unknown's avatar
unknown committed
1404 1405 1406 1407
	  fputs(query,stderr);
	  safe_exit(EX_CONSCHECK);
	  return;
	}
1408
	if (extended_insert)
unknown's avatar
unknown committed
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
	{
	  ulong length = lengths[i];
	  if (i == 0)
	    dynstr_set(&extended_row,"(");
	  else
	    dynstr_append(&extended_row,",");

	  if (row[i])
	  {
	    if (length)
	    {
1420
	      if (!IS_NUM_FIELD(field))
unknown's avatar
unknown committed
1421 1422 1423 1424 1425 1426
	      {
		if (dynstr_realloc(&extended_row,length * 2+2))
		{
		  fputs("Aborting dump (out of memory)",stderr);
		  safe_exit(EX_EOM);
		}
unknown's avatar
unknown committed
1427
		dynstr_append(&extended_row,"'");
unknown's avatar
unknown committed
1428
		extended_row.length +=
unknown's avatar
unknown committed
1429 1430 1431
		  mysql_real_escape_string(&mysql_connection,
					   &extended_row.str[extended_row.length],row[i],length);
		extended_row.str[extended_row.length]='\0';
unknown's avatar
unknown committed
1432
		dynstr_append(&extended_row,"'");
unknown's avatar
unknown committed
1433 1434
	      }
	      else
1435
	      {
unknown's avatar
unknown committed
1436
		/* change any strings ("inf", "-inf", "nan") into NULL */
1437
		char *ptr = row[i];
unknown's avatar
unknown committed
1438 1439
		if (my_isalpha(charset_info, *ptr) || (*ptr == '-' &&
		    my_isalpha(charset_info, ptr[1])))
unknown's avatar
unknown committed
1440 1441 1442 1443 1444 1445
		  dynstr_append(&extended_row, "NULL");
		else
		{
		  if (field->type == FIELD_TYPE_DECIMAL)
		  {
		    /* add " signs around */
unknown's avatar
unknown committed
1446
		    dynstr_append(&extended_row, "'");
unknown's avatar
unknown committed
1447
		    dynstr_append(&extended_row, ptr);
unknown's avatar
unknown committed
1448
		    dynstr_append(&extended_row, "'");
unknown's avatar
unknown committed
1449 1450 1451 1452
		  }
		  else
		    dynstr_append(&extended_row, ptr);
		}
1453
	      }
unknown's avatar
unknown committed
1454 1455
	    }
	    else
unknown's avatar
unknown committed
1456
	      dynstr_append(&extended_row,"''");
unknown's avatar
unknown committed
1457 1458 1459 1460 1461 1462 1463 1464 1465
	  }
	  else if (dynstr_append(&extended_row,"NULL"))
	  {
	    fputs("Aborting dump (out of memory)",stderr);
	    safe_exit(EX_EOM);
	  }
	}
	else
	{
unknown's avatar
unknown committed
1466 1467
	  if (i && !opt_xml)
	    fputc(',', md_result_file);
unknown's avatar
unknown committed
1468 1469
	  if (row[i])
	  {
1470
	    if (!IS_NUM_FIELD(field))
1471
	    {
unknown's avatar
unknown committed
1472
	      if (opt_xml)
1473
	      {
1474 1475
	        print_xml_tag1(md_result_file, "\t\t", "field name=",
			      field->name, "");
1476 1477 1478
		print_quoted_xml(md_result_file, row[i], lengths[i]);
		fputs("</field>\n", md_result_file);
	      }
unknown's avatar
unknown committed
1479 1480
	      else
		unescape(md_result_file, row[i], lengths[i]);
unknown's avatar
unknown committed
1481
	    }
unknown's avatar
unknown committed
1482
	    else
1483
	    {
unknown's avatar
unknown committed
1484
	      /* change any strings ("inf", "-inf", "nan") into NULL */
1485
	      char *ptr = row[i];
unknown's avatar
unknown committed
1486
	      if (opt_xml)
1487
	      {
1488 1489
	        print_xml_tag1(md_result_file, "\t\t", "field name=",
			       field->name, "");
1490 1491 1492 1493
		fputs(!my_isalpha(charset_info, *ptr) ? ptr: "NULL",
		      md_result_file);
		fputs("</field>\n", md_result_file);
	      }
unknown's avatar
unknown committed
1494 1495 1496 1497
	      else if (my_isalpha(charset_info, *ptr) ||
		       (*ptr == '-' && my_isalpha(charset_info, ptr[1])))
	        fputs("NULL", md_result_file);
	      else if (field->type == FIELD_TYPE_DECIMAL)
unknown's avatar
unknown committed
1498 1499
	      {
		/* add " signs around */
unknown's avatar
unknown committed
1500
		fputc('\'', md_result_file);
unknown's avatar
unknown committed
1501
		fputs(ptr, md_result_file);
unknown's avatar
unknown committed
1502
		fputc('\'', md_result_file);
unknown's avatar
unknown committed
1503
	      }
unknown's avatar
unknown committed
1504
	      else
unknown's avatar
unknown committed
1505
		fputs(ptr, md_result_file);
1506
	    }
unknown's avatar
unknown committed
1507 1508 1509 1510
	  }
	}
      }

unknown's avatar
merge  
unknown committed
1511
      if (opt_xml)
1512
        fputs("\t</row>\n", md_result_file);
unknown's avatar
merge  
unknown committed
1513

1514
      if (extended_insert)
unknown's avatar
unknown committed
1515 1516 1517 1518 1519 1520 1521
      {
	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
1522 1523
	  fputc(',',md_result_file);		/* Always row break */
	  fputs(extended_row.str,md_result_file);
unknown's avatar
unknown committed
1524 1525 1526
	}
        else
        {
1527
	  if (row_break)
unknown's avatar
unknown committed
1528
	    fputs(";\n", md_result_file);
unknown's avatar
unknown committed
1529
	  row_break=1;				/* This is first row */
unknown's avatar
unknown committed
1530

1531 1532
          fputs(insert_pat,md_result_file);
          fputs(extended_row.str,md_result_file);
unknown's avatar
unknown committed
1533 1534 1535
	  total_length = row_length+init_length;
        }
      }
unknown's avatar
unknown committed
1536
      else if (!opt_xml)
unknown's avatar
unknown committed
1537
	fputs(");\n", md_result_file);
unknown's avatar
unknown committed
1538
    }
unknown's avatar
unknown committed
1539

unknown's avatar
unknown committed
1540
    /* XML - close table tag and supress regular output */
unknown's avatar
unknown committed
1541
    if (opt_xml)
1542
	fputs("\t</table_data>\n", md_result_file);
unknown's avatar
unknown committed
1543
    else if (extended_insert && row_break)
unknown's avatar
unknown committed
1544 1545
      fputs(";\n", md_result_file);		/* If not empty table */
    fflush(md_result_file);
unknown's avatar
unknown committed
1546 1547
    if (mysql_errno(sock))
    {
unknown's avatar
unknown committed
1548
      sprintf(query,"%s: Error %d: %s when dumping table %s at row: %ld\n",
unknown's avatar
unknown committed
1549 1550 1551
	      my_progname,
	      mysql_errno(sock),
	      mysql_error(sock),
unknown's avatar
unknown committed
1552
	      result_table,
unknown's avatar
unknown committed
1553 1554 1555 1556 1557 1558
	      rownr);
      fputs(query,stderr);
      safe_exit(EX_CONSCHECK);
      return;
    }
    if (opt_lock)
unknown's avatar
unknown committed
1559
      fputs("UNLOCK TABLES;\n", md_result_file);
unknown's avatar
merge  
unknown committed
1560 1561
    if (opt_disable_keys)
      fprintf(md_result_file,"/*!40000 ALTER TABLE %s ENABLE KEYS */;\n",
unknown's avatar
unknown committed
1562
	      opt_quoted_table);
unknown's avatar
unknown committed
1563 1564
    if (opt_autocommit)
      fprintf(md_result_file, "commit;\n");
1565
    mysql_free_result(res);
unknown's avatar
unknown committed
1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581
  }
} /* dumpTable */


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
1582

unknown's avatar
unknown committed
1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620
  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++)
1621
  {
unknown's avatar
unknown committed
1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635
    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
1636
  if (!path && !opt_xml)
unknown's avatar
unknown committed
1637 1638 1639
  {
    if (opt_databases || opt_alldbs)
    {
unknown's avatar
unknown committed
1640 1641 1642
      /*
	length of table name * 2 (if name contain quotas), 2 quotas and 0
      */
unknown's avatar
unknown committed
1643 1644
      char quoted_database_buf[64*2+3];
      char *qdatabase= quote_name(database,quoted_database_buf,opt_quoted);
1645
      if (opt_comments)
unknown's avatar
unknown committed
1646
	fprintf(md_result_file,"\n--\n-- Current Database: %s\n--\n", qdatabase);
unknown's avatar
unknown committed
1647
      if (!opt_create_db)
1648
      {
unknown's avatar
unknown committed
1649
        char qbuf[256];
1650 1651
        MYSQL_ROW row;
        MYSQL_RES *dbinfo;
1652

unknown's avatar
unknown committed
1653
        sprintf(qbuf,"SHOW CREATE DATABASE WITH IF NOT EXISTS %s",
unknown's avatar
unknown committed
1654
		qdatabase);
1655

1656 1657 1658
        if (mysql_query(sock, qbuf) || !(dbinfo = mysql_store_result(sock)))
        {
          /* Old server version, dump generic CREATE DATABASE */
1659
	  fprintf(md_result_file,
unknown's avatar
unknown committed
1660 1661
		  "\nCREATE DATABASE /*!32312 IF NOT EXISTS*/ %s;\n",
		  qdatabase);
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671
	}
	else
        {
	  row = mysql_fetch_row(dbinfo);
	  if (row[1])
	  {
	    fprintf(md_result_file,"\n%s;\n",row[1]);
          }
	}
      }
unknown's avatar
unknown committed
1672
      fprintf(md_result_file,"\nUSE %s;\n", qdatabase);
unknown's avatar
unknown committed
1673 1674
    }
  }
unknown's avatar
unknown committed
1675 1676
  if (extended_insert && init_dynamic_string(&extended_row, "", 1024, 1024))
    exit(EX_EOM);
unknown's avatar
unknown committed
1677 1678 1679 1680
  return 0;
} /* init_dumping */


unknown's avatar
unknown committed
1681

unknown's avatar
unknown committed
1682 1683 1684 1685
static int dump_all_tables_in_db(char *database)
{
  char *table;
  uint numrows;
unknown's avatar
unknown committed
1686
  char table_buff[NAME_LEN*2+3];
unknown's avatar
unknown committed
1687 1688 1689

  if (init_dumping(database))
    return 1;
unknown's avatar
unknown committed
1690
  if (opt_xml)
1691
    print_xml_tag1(md_result_file, "", "database name=", database, "\n");
unknown's avatar
unknown committed
1692 1693 1694 1695 1696 1697
  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
1698
      dynstr_append(&query, quote_name(table, table_buff, 1));
unknown's avatar
unknown committed
1699 1700 1701
      dynstr_append(&query, " READ /*!32311 LOCAL */,");
    }
    if (numrows && mysql_real_query(sock, query.str, query.length-1))
unknown's avatar
unknown committed
1702
      DBerror(sock, "when using LOCK TABLES");
unknown's avatar
unknown committed
1703 1704 1705 1706 1707 1708
            /* 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
1709
      DBerror(sock, "when doing refresh");
unknown's avatar
unknown committed
1710 1711 1712 1713 1714 1715 1716 1717
           /* 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
1718
  if (opt_xml)
1719
    fputs("</database>\n", md_result_file);
unknown's avatar
unknown committed
1720
  if (lock_tables)
1721
    mysql_query(sock,"UNLOCK TABLES");
unknown's avatar
unknown committed
1722 1723 1724 1725 1726 1727 1728 1729
  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
1730
  char table_buff[NAME_LEN*+3];
unknown's avatar
unknown committed
1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741

  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
1742
      dynstr_append(&query, quote_name(table_names[i], table_buff, 1));
unknown's avatar
unknown committed
1743 1744 1745
      dynstr_append(&query, " READ /*!32311 LOCAL */,");
    }
    if (mysql_real_query(sock, query.str, query.length-1))
unknown's avatar
unknown committed
1746
      DBerror(sock, "when doing LOCK TABLES");
unknown's avatar
unknown committed
1747 1748 1749 1750 1751 1752
       /* 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
1753
      DBerror(sock, "when doing refresh");
unknown's avatar
unknown committed
1754 1755
     /* We shall countinue here, if --force was given */
  }
unknown's avatar
unknown committed
1756
  if (opt_xml)
1757
    print_xml_tag1(md_result_file, "", "database name=", db, "\n");
unknown's avatar
unknown committed
1758 1759 1760 1761 1762 1763
  for (; tables > 0 ; tables-- , table_names++)
  {
    numrows = getTableStructure(*table_names, db);
    if (!dFlag && numrows > 0)
      dumpTable(numrows, *table_names);
  }
unknown's avatar
unknown committed
1764
  if (opt_xml)
1765
    fputs("</database>\n", md_result_file);
unknown's avatar
unknown committed
1766
  if (lock_tables)
unknown's avatar
unknown committed
1767
    mysql_query(sock,"UNLOCK TABLES");
unknown's avatar
unknown committed
1768 1769 1770 1771
  return 0;
} /* dump_selected_tables */


1772 1773 1774 1775 1776 1777 1778 1779 1780

static ulong find_set(TYPELIB *lib, const char *x, uint length,
		      char **err_pos, uint *err_len)
{
  const char *end= x + length;
  ulong found= 0;
  uint find;
  char buff[255];

1781
  *err_pos= 0;                  /* No error yet */
1782
  while (end > x && my_isspace(charset_info, end[-1]))
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
    end--;

  *err_len= 0;
  if (x != end)
  {
    const char *start= x;
    for (;;)
    {
      const char *pos= start;
      uint var_len;

      for (; pos != end && *pos != ','; pos++) ;
      var_len= (uint) (pos - start);
      strmake(buff, start, min(sizeof(buff), var_len));
      find= find_type(buff, lib, var_len);
      if (!find)
      {
        *err_pos= (char*) start;
        *err_len= var_len;
      }
      else
        found|= ((longlong) 1 << (find - 1));
      if (pos == end)
        break;
      start= pos + 1;
    }
  }
  return found;
}


unknown's avatar
unknown committed
1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830
/* 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
1831
	  unescape(file,row[0],(uint) strlen(row[0]));
unknown's avatar
unknown committed
1832 1833 1834 1835 1836 1837 1838 1839 1840 1841
	else
	  fputs(row[0], file);
	return;
      }
    }
  }
  return;					/* This shouldn't happen */
} /* print_value */


1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896
/*
  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
1897 1898
int main(int argc, char **argv)
{
unknown's avatar
unknown committed
1899 1900
  MYSQL_ROW row;
  MYSQL_RES *master;
1901
  compatible_mode_normal_str[0]= 0;
unknown's avatar
unknown committed
1902

unknown's avatar
unknown committed
1903 1904 1905 1906 1907 1908
  MY_INIT(argv[0]);
  if (get_options(&argc, &argv))
  {
    my_end(0);
    exit(EX_USAGE);
  }
1909
  if (dbConnect(current_host, current_user, opt_password))
unknown's avatar
unknown committed
1910 1911
    exit(EX_MYSQLERR);
  if (!path)
1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924
    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);
    }
  }
1925
  else if (opt_single_transaction)
1926 1927 1928
  {
    /* There is no sense to start transaction if all tables are locked */ 
    if (mysql_query(sock, "BEGIN"))
1929
    {
1930
      my_printf_error(0, "Error: Couldn't execute 'BEGIN': %s",
1931
                        MYF(0), mysql_error(sock));
1932 1933 1934 1935
      my_end(0);
      return(first_error);
    }    
  }
unknown's avatar
unknown committed
1936 1937 1938
  if (opt_alldbs)
    dump_all_databases();
  else if (argc > 1 && !opt_databases)
1939 1940
  {
    /* Only one database and selected table(s) */
unknown's avatar
unknown committed
1941
    dump_selected_tables(*argv, (argv + 1), (argc - 1));
1942
  }
unknown's avatar
unknown committed
1943
  else
1944 1945
  {
    /* One or more databases, all tables */
unknown's avatar
unknown committed
1946
    dump_databases(argv);
1947
  }
unknown's avatar
unknown committed
1948 1949 1950

  if (opt_first_slave)
  {
1951 1952 1953 1954 1955
    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
1956 1957 1958 1959 1960 1961
    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));
1962
      else
unknown's avatar
unknown committed
1963 1964
      {
	row = mysql_fetch_row(master);
unknown's avatar
unknown committed
1965
	if (row && row[0] && row[1])
1966
	{
1967 1968 1969
	  if (opt_comments)
	    fprintf(md_result_file,
		    "\n--\n-- Position to start replication from\n--\n\n");
unknown's avatar
unknown committed
1970
	  fprintf(md_result_file,
unknown's avatar
unknown committed
1971 1972
		  "CHANGE MASTER TO MASTER_LOG_FILE='%s', \
MASTER_LOG_POS=%s ;\n",row[0],row[1]); 
unknown's avatar
unknown committed
1973 1974 1975 1976
	}
	mysql_free_result(master);
      }
    }
unknown's avatar
unknown committed
1977 1978 1979 1980
    if (mysql_query(sock, "UNLOCK TABLES"))
      my_printf_error(0, "Error: Couldn't execute 'UNLOCK TABLES': %s",
		      MYF(0), mysql_error(sock));
  }
1981
  else if (opt_single_transaction) /* Just to make it beautiful enough */
1982 1983 1984
#ifdef HAVE_SMEM
  my_free(shared_memory_base_name,MYF(MY_ALLOW_ZERO_PTR));
#endif
1985
  {
1986
    /*
1987 1988
      In case we were locking all tables, we did not start transaction
      so there is no need to commit it.
1989 1990
    */

1991 1992 1993 1994
    /* 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",
1995 1996
	      MYF(0), mysql_error(sock));
    }
1997
  }
unknown's avatar
unknown committed
1998
  dbDisconnect(current_host);
1999 2000
  if (!path)
    write_footer(md_result_file);
unknown's avatar
unknown committed
2001 2002
  if (md_result_file != stdout)
    my_fclose(md_result_file, MYF(0));
2003
  my_free(opt_password, MYF(MY_ALLOW_ZERO_PTR));
2004
  if (extended_insert)
unknown's avatar
unknown committed
2005 2006 2007 2008
    dynstr_free(&extended_row);
  my_end(0);
  return(first_error);
} /* main */