#define DUMP_VERSION "10.13"
#include <hash.h>
#include <m_ctype.h>
#include <m_string.h>
#include <my_global.h>
#include <my_sys.h>
#include <my_user.h>
#include <stdarg.h>
#include <welcome_copyright_notice.h>
#include <list>
#include <map>
#include <string>
#include "ProcMetrics.h"
#include "ProcOdbc.h"
#include "client_priv.h"
#include "my_default.h"
#include "mysql.h"
#include "mysql_version.h"
#include "mysqld_error.h"
#define EX_USAGE 1
#define EX_MYSQLERR 2
#define EX_CONSCHECK 3
#define EX_EOM 4
#define EX_EOF 5
#define EX_ILLEGAL_TABLE 6
#define SHOW_FIELDNAME 0
#define SHOW_TYPE 1
#define SHOW_NULL 2
#define SHOW_DEFAULT 4
#define SHOW_EXTRA 5
#define QUERY_LENGTH 1536
#define COMMENT_LENGTH 2048
#define IGNORE_NONE 0x00
#define IGNORE_DATA 0x01
#define MAX_FIELDS 4000
static DMODBC::ProcOdbc *sg_procOdbc;
static DMODBC::ProcMetrics *sg_procMetrics;
static struct DMODBC::MysqlInfo sg_mysqlInfo;
static std::map<std::string, std::string> sg_createSuccessTables;
static std::list<std::string> sg_createFailedTables;
static bool sg_firstdumptables;
static pthread_key_t sg_connectionkey;
static void add_load_option(DYNAMIC_STRING *str, const char *option,
const char *option_value);
static char *alloc_query_str(size_t size);
static void field_escape(DYNAMIC_STRING *in, const char *from);
#define MYSQL_OPT_MASTER_DATA_EFFECTIVE_SQL 1
#define MYSQL_OPT_MASTER_DATA_COMMENTED_SQL 2
#define MYSQL_OPT_SLAVE_DATA_EFFECTIVE_SQL 1
#define MYSQL_OPT_SLAVE_DATA_COMMENTED_SQL 2
my_bool verbose = 0;
char *opt_ignore_error = 0;
my_bool opt_force = 0;
my_bool opt_comments = 1;
my_bool opt_comments_used = 1;
char *path = 0;
my_bool opt_xml = 0;
class MySQLContext {
public:
MySQLContext();
void init_mysql_context();
void write_header(FILE *sql_file, char *db_name);
void write_footer(FILE *sql_file);
my_bool get_one_option(int optid, const char *argument);
int get_options(int *argc, char ***argv);
int fetch_db_collation(const char *db_name, char *db_cl_name, int db_cl_size);
void free_resources();
int mainproc(int argc, char **argv);
int switch_character_set_results(MYSQL *mysql, const char *cs_name);
int connect_to_db(char *host, char *user, char *passwd);
my_bool do_ignore_error();
void maybe_exit(int error);
void dbDisconnect(char *host);
void unescape(FILE *file, char *pos, size_t length, DYNAMIC_STRING *output);
void unescapestr(char *pos, size_t length, DYNAMIC_STRING *output);
char *quote_name(const char *name, char *buff, my_bool force);
uint dump_events_for_db(char *db);
uint dump_routines_for_db(char *db);
uint get_table_structure(char *table, char *db, char *table_type,
char *ignore_flag, my_bool real_columns[]);
void dump_trigger_old(FILE *sql_file, MYSQL_RES *show_triggers_rs,
MYSQL_ROW *show_trigger_row, const char *table_name);
int dump_trigger(FILE *sql_file, MYSQL_RES *show_create_trigger_rs,
const char *db_name, const char *db_cl_name);
int dump_triggers_for_table(char *table_name, char *db_name);
void dump_table(char *table, char *db);
char *getTableName(int reset);
int dump_tablespaces_for_tables(char *db, char **table_names, int tables);
int dump_tablespaces(char *ts_where);
int is_ndbinfo(MYSQL *mysql, const char *dbname);
int dump_all_databases();
int dump_databases(char **db_names);
int init_dumping(char *database, int init_func);
int init_dumping_tables(char *qdatabase);
int init_dumping_views(char *qdatabase);
int dump_all_tables_in_db(char *database);
my_bool dump_all_views_in_db(char *database);
char *get_actual_table_name(const char *old_table_name, MEM_ROOT *root);
int dump_selected_tables(char *db, char **table_names, int tables);
int dump_tablespaces_for_databases(char **databases);
int switch_db_collation(FILE *sql_file, const char *db_name,
const char *delimiter, const char *current_db_cl_name,
const char *required_db_cl_name, int *db_cl_altered);
int restore_db_collation(FILE *sql_file, const char *db_name,
const char *delimiter, const char *db_cl_name);
char *primary_key_fields(const char *table_name);
char check_if_ignore_table(const char *table_name, char *table_type);
int dump_all_tablespaces();
my_bool get_view_structure(char *table, char *db);
int do_show_master_status(MYSQL *mysql_con);
int do_stop_slave_sql(MYSQL *mysql_con);
int add_stop_slave(void);
int add_slave_statements(void);
int do_show_slave_status(MYSQL *mysql_con);
int do_start_slave_sql(MYSQL *mysql_con);
int do_flush_tables_read_lock(MYSQL *mysql_con);
int do_unlock_tables(MYSQL *mysql_con);
int get_bin_log_name(MYSQL *mysql_con, char *buff_log_name, uint buff_len);
int purge_bin_logs_to(MYSQL *mysql_con, char *log_name);
int start_transaction(MYSQL *mysql_con);
ulong find_set(TYPELIB *lib, const char *x, size_t length, char **err_pos,
uint *err_len);
void print_value(FILE *file, MYSQL_RES *result, MYSQL_ROW row,
const char *prefix, const char *name, int string_value);
my_bool process_set_gtid_purged(MYSQL *mysql_con);
private:
my_bool opt_no_create_info;
my_bool opt_no_data;
my_bool quick;
my_bool extended_insert;
my_bool lock_tables;
my_bool flush_logs;
my_bool flush_privileges;
my_bool opt_drop;
my_bool opt_keywords;
my_bool opt_lock;
my_bool opt_compress;
my_bool create_options;
my_bool opt_quoted;
my_bool opt_databases;
my_bool opt_alldbs;
my_bool opt_create_db;
my_bool opt_lock_all_tables;
my_bool opt_set_charset;
my_bool opt_dump_date;
my_bool opt_autocommit;
my_bool opt_disable_keys;
my_bool opt_delete_master_logs;
my_bool tty_password;
my_bool opt_single_transaction;
my_bool opt_compact;
my_bool opt_hex_blob;
my_bool opt_order_by_primary;
my_bool opt_ignore;
my_bool opt_complete_insert;
my_bool opt_drop_database;
my_bool opt_replace_into;
my_bool opt_dump_triggers;
my_bool opt_routines;
my_bool opt_tz_utc;
my_bool opt_slave_apply;
my_bool opt_include_master_host_port;
my_bool opt_events;
my_bool opt_alltspcs;
my_bool opt_notspcs;
my_bool opt_drop_trigger;
my_bool opt_secure_auth;
my_bool insert_pat_inited;
my_bool debug_info_flag;
my_bool debug_check_flag;
ulong opt_max_allowed_packet;
ulong opt_net_buffer_length;
MYSQL mysql_connection;
MYSQL *mysql;
DYNAMIC_STRING insert_pat;
char *opt_password;
char *current_user;
char *current_host;
char *fields_terminated;
char *lines_terminated;
char *enclosed;
char *opt_enclosed;
char *escaped;
char *where;
char *order_by;
char *opt_compatible_mode_str;
char *err_ptr;
char *log_error_file;
char **defaults_argv;
char compatible_mode_normal_str[255];
my_bool server_supports_switching_charsets;
ulong opt_compatible_mode;
uint opt_enable_cleartext_plugin;
my_bool using_opt_enable_cleartext_plugin;
uint opt_mysql_port;
uint opt_master_data;
uint opt_slave_data;
uint my_end_arg;
char *opt_mysql_unix_port;
char *opt_bind_addr;
int first_error;
};
MySQLContext::MySQLContext() { init_mysql_context(); }
void MySQLContext::init_mysql_context() {
verbose = 0;
opt_no_create_info = 0;
opt_no_data = 0;
quick = 1;
extended_insert = 1;
lock_tables = 1;
opt_force = 0;
flush_logs = 0;
flush_privileges = 0;
opt_drop = 1;
opt_keywords = 0;
opt_lock = 1;
opt_compress = 0;
create_options = 1;
opt_quoted = 0;
opt_databases = 0;
opt_alldbs = 0;
opt_create_db = 0;
opt_lock_all_tables = 0;
opt_set_charset = 1;
opt_dump_date = 1;
opt_autocommit = 0;
opt_disable_keys = 1;
opt_delete_master_logs = 0;
tty_password = 0;
opt_single_transaction = 0;
opt_compact = 0;
opt_hex_blob = 0;
opt_order_by_primary = 0;
opt_ignore = 0;
opt_complete_insert = 0;
opt_drop_database = 0;
opt_replace_into = 0;
opt_dump_triggers = 0;
opt_routines = 0;
opt_tz_utc = 1;
opt_slave_apply = 0;
opt_include_master_host_port = 0;
opt_events = 0;
opt_alltspcs = 0;
opt_notspcs = 0;
opt_drop_trigger = 0;
opt_secure_auth = TRUE;
insert_pat_inited = 0;
debug_info_flag = 0;
debug_check_flag = 0;
opt_max_allowed_packet = 24*1024*1024;
opt_net_buffer_length = 1024*1024L-1025;
mysql = 0;
opt_password = 0;
current_user = 0;
current_host = 0;
fields_terminated = 0;
lines_terminated = 0;
enclosed = 0;
opt_enclosed = 0;
escaped = 0;
where = 0;
order_by = 0;
opt_compatible_mode_str = 0;
err_ptr = 0;
log_error_file = NULL;
defaults_argv = 0;
server_supports_switching_charsets = TRUE;
opt_compatible_mode = 0;
opt_enable_cleartext_plugin = 0;
using_opt_enable_cleartext_plugin = 0;
opt_mysql_port = 0;
opt_master_data = 0;
opt_slave_data = 0;
my_end_arg = 0;
opt_mysql_unix_port = 0;
opt_bind_addr = NULL;
first_error = 0;
}
static MySQLContext *create_connection_thread_context() {
MySQLContext *ret = new MySQLContext();
pthread_setspecific(sg_connectionkey, ret);
return ret;
}
static MySQLContext *get_connection_thread_context() {
return (MySQLContext *)pthread_getspecific(sg_connectionkey);
}
static void release_connection_thread_context() {
MySQLContext *context = (MySQLContext *)pthread_getspecific(sg_connectionkey);
if (context) {
delete context;
}
}
#include <caching_sha2_passwordopt-vars.h>
#include <sslopt-vars.h>
FILE *md_result_file = 0;
FILE *stderror_file = 0;
const char *set_gtid_purged_mode_names[] = {"OFF", "AUTO", "ON", NullS};
static TYPELIB set_gtid_purged_mode_typelib = {
array_elements(set_gtid_purged_mode_names) - 1, "",
set_gtid_purged_mode_names, NULL};
static enum enum_set_gtid_purged_mode {
SET_GTID_PURGED_OFF = 0,
SET_GTID_PURGED_AUTO = 1,
SET_GTID_PURGED_ON = 2
} opt_set_gtid_purged_mode = SET_GTID_PURGED_AUTO;
#if defined(_WIN32) && !defined(EMBEDDED_LIBRARY)
static char *shared_memory_base_name = 0;
#endif
static uint opt_protocol = 0;
static char *opt_plugin_dir = 0, *opt_default_auth = 0;
DYNAMIC_ARRAY ignore_error;
static int parse_ignore_error();
Dynamic_string wrapper functions. In this file use these
wrappers, they will terminate the process if there is
an allocation failure.
*/
static void init_dynamic_string_checked(DYNAMIC_STRING *str,
const char *init_str, size_t init_alloc,
size_t alloc_increment);
static void dynstr_append_checked(DYNAMIC_STRING *dest, const char *src);
static void dynstr_set_checked(DYNAMIC_STRING *str, const char *init_str);
static void dynstr_append_mem_checked(DYNAMIC_STRING *str, const char *append,
size_t length);
static void dynstr_realloc_checked(DYNAMIC_STRING *str, size_t additional_size);
Constant for detection of default value of default_charset.
If default_charset is equal to mysql_universal_client_charset, then
it is the default value which assigned at the very beginning of main().
*/
static const char *mysql_universal_client_charset = "utf8";
static char *default_charset;
static CHARSET_INFO *charset_info = &my_charset_latin1;
const char *default_dbug_option = "d:t:o,/tmp/mysqldump.trace";
my_bool seen_views = 0;
const char *compatible_mode_names[] = {"MYSQL323",
"MYSQL40",
"POSTGRESQL",
"ORACLE",
"MSSQL",
"DB2",
"MAXDB",
"NO_KEY_OPTIONS",
"NO_TABLE_OPTIONS",
"NO_FIELD_OPTIONS",
"ANSI",
NullS};
#define MASK_ANSI_QUOTES \
((1 << 2) | \
(1 << 3) | \
(1 << 4) | \
(1 << 5) | \
(1 << 6) | \
(1 << 10) \
)
TYPELIB compatible_mode_typelib = {array_elements(compatible_mode_names) - 1,
"", compatible_mode_names, NULL};
HASH ignore_table;
static const char *load_default_groups[] = {"mysqldump", "client", 0};
static void die(int error, const char *reason, ...);
static void maybe_die(int error, const char *reason, ...);
static void print_comment(FILE *sql_file, my_bool is_error, const char *format,
...);
static char const *fix_identifier_with_newline(char const *object_name,
my_bool *freemem);
Print the supplied message if in verbose mode
SYNOPSIS
verbose_msg()
fmt format specifier
... variable number of parameters
*/
static void verbose_msg(const char *fmt, ...) {
va_list args;
DBUG_ENTER("verbose_msg");
if (!verbose) DBUG_VOID_RETURN;
va_start(args, fmt);
vfprintf(stderr, fmt, args);
va_end(args);
fflush(stderr);
DBUG_VOID_RETURN;
}
exit with message if ferror(file)
SYNOPSIS
check_io()
file - checked file
*/
void check_io(FILE *file) {
if (ferror(file) || errno == 5) die(EX_EOF, "Got errno %d on write", errno);
}
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);
}
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);
}
void MySQLContext::write_header(FILE *sql_file, char *db_name) {
if (opt_xml) {
fputs("<?xml version=\"1.0\"?>\n", sql_file);
Schema reference. Allows use of xsi:nil for NULL values and
xsi:type to define an element's data type.
*/
fputs("<mysqldump ", sql_file);
fputs("xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"", sql_file);
fputs(">\n", sql_file);
check_io(sql_file);
} else if (!opt_compact) {
my_bool freemem = FALSE;
char const *text = fix_identifier_with_newline(db_name, &freemem);
print_comment(
sql_file, 0, "-- MySQL dump %s Distrib %s, for %s (%s)\n--\n",
DUMP_VERSION, MYSQL_SERVER_VERSION, SYSTEM_TYPE, MACHINE_TYPE);
print_comment(sql_file, 0, "-- Host: %s Database: %s\n",
current_host ? current_host : "localhost", text);
if (freemem) my_free((void *)text);
print_comment(
sql_file, 0,
"-- ------------------------------------------------------\n");
print_comment(sql_file, 0, "-- Server version\t%s\n",
mysql_get_server_info(&mysql_connection));
if (opt_set_charset)
fprintf(
sql_file,
"\n/*!40101 SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT */;"
"\n/*!40101 SET @OLD_CHARACTER_SET_RESULTS=@@CHARACTER_SET_RESULTS "
"*/;"
"\n/*!40101 SET @OLD_COLLATION_CONNECTION=@@COLLATION_CONNECTION */;"
"\n/*!40101 SET NAMES %s */;\n",
default_charset);
if (opt_tz_utc) {
fprintf(sql_file, "/*!40103 SET @OLD_TIME_ZONE=@@TIME_ZONE */;\n");
fprintf(sql_file, "/*!40103 SET TIME_ZONE='+00:00' */;\n");
}
if (!path) {
fprintf(md_result_file,
"\
/*!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\
");
}
fprintf(sql_file,
"/*!40101 SET @OLD_SQL_MODE=@@SQL_MODE, SQL_MODE='%s%s%s' */;\n"
"/*!40111 SET @OLD_SQL_NOTES=@@SQL_NOTES, SQL_NOTES=0 */;\n",
path ? "" : "NO_AUTO_VALUE_ON_ZERO",
compatible_mode_normal_str[0] == 0 ? "" : ",",
compatible_mode_normal_str);
check_io(sql_file);
}
}
void MySQLContext::write_footer(FILE *sql_file) {
if (opt_xml) {
fputs("</mysqldump>\n", sql_file);
check_io(sql_file);
} else if (!opt_compact) {
if (opt_tz_utc)
fprintf(sql_file, "/*!40103 SET TIME_ZONE=@OLD_TIME_ZONE */;\n");
fprintf(sql_file, "\n/*!40101 SET SQL_MODE=@OLD_SQL_MODE */;\n");
if (!path) {
fprintf(md_result_file,
"\
/*!40014 SET FOREIGN_KEY_CHECKS=@OLD_FOREIGN_KEY_CHECKS */;\n\
/*!40014 SET UNIQUE_CHECKS=@OLD_UNIQUE_CHECKS */;\n");
}
if (opt_set_charset)
fprintf(
sql_file,
"/*!40101 SET CHARACTER_SET_CLIENT=@OLD_CHARACTER_SET_CLIENT */;\n"
"/*!40101 SET CHARACTER_SET_RESULTS=@OLD_CHARACTER_SET_RESULTS */;\n"
"/*!40101 SET COLLATION_CONNECTION=@OLD_COLLATION_CONNECTION */;\n");
fprintf(sql_file, "/*!40111 SET SQL_NOTES=@OLD_SQL_NOTES */;\n");
fputs("\n", sql_file);
if (opt_dump_date) {
char time_str[20];
get_date(time_str, GETDATE_DATE_TIME, 0);
print_comment(sql_file, 0, "-- Dump completed on %s\n", time_str);
} else
print_comment(sql_file, 0, "-- Dump completed\n");
check_io(sql_file);
}
}
uchar *get_table_key(const char *entry, size_t *length,
my_bool not_used MY_ATTRIBUTE((unused))) {
*length = strlen(entry);
return (uchar *)entry;
}
my_bool MySQLContext::get_one_option(int optid, const char *argument) {
switch (optid) {
case (int)OPT_COMPATIBLE: {
char buff[255];
char *end = compatible_mode_normal_str;
int i;
ulong mode;
uint err_len;
opt_quoted = 1;
opt_set_charset = 1;
opt_compatible_mode_str = (char *)argument;
opt_compatible_mode = find_set(&compatible_mode_typelib, argument,
strlen(argument), &err_ptr, &err_len);
if (err_len) {
strmake(buff, err_ptr, MY_MIN(sizeof(buff) - 1, err_len));
fprintf(stderr, "Invalid mode to --compatible: %s\n", buff);
exit(1);
}
#if !defined(DBUG_OFF)
{
size_t size_for_sql_mode = 0;
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 = my_stpcpy(end, compatible_mode_names[i]);
end = my_stpcpy(end, ",");
}
}
if (end != compatible_mode_normal_str) end[-1] = 0;
Set charset to the default compiled value if it hasn't
been reset yet by --default-character-set=xxx.
*/
break;
}
}
return 0;
}
int MySQLContext::get_options(int *argc, char ***argv) {
if (mysql_get_option(NULL, MYSQL_OPT_MAX_ALLOWED_PACKET,
&opt_max_allowed_packet) ||
mysql_get_option(NULL, MYSQL_OPT_NET_BUFFER_LENGTH,
&opt_max_allowed_packet)) {
exit(1);
}
md_result_file = stdout;
my_getopt_use_args_separator = TRUE;
if (load_defaults("my", load_default_groups, argc, argv)) return 1;
my_getopt_use_args_separator = FALSE;
defaults_argv = *argv;
if (my_hash_init(&ignore_table, charset_info, 16, 0, 0,
(my_hash_get_key)get_table_key, my_free, 0,
PSI_NOT_INSTRUMENTED))
return (EX_EOM);
if (my_hash_insert(&ignore_table,
(uchar *)my_strdup(PSI_NOT_INSTRUMENTED,
"mysql.apply_status", MYF(MY_WME))) ||
my_hash_insert(&ignore_table,
(uchar *)my_strdup(PSI_NOT_INSTRUMENTED, "mysql.schema",
MYF(MY_WME))) ||
my_hash_insert(&ignore_table,
(uchar *)my_strdup(PSI_NOT_INSTRUMENTED,
"mysql.general_log", MYF(MY_WME))) ||
my_hash_insert(&ignore_table,
(uchar *)my_strdup(PSI_NOT_INSTRUMENTED, "mysql.slow_log",
MYF(MY_WME))))
return (EX_EOM);
get_one_option(OPT_COMPATIBLE, "postgresql");
if (mysql_options(NULL, MYSQL_OPT_MAX_ALLOWED_PACKET,
&opt_max_allowed_packet) ||
mysql_options(NULL, MYSQL_OPT_NET_BUFFER_LENGTH,
&opt_net_buffer_length)) {
exit(1);
}
if (debug_info_flag) my_end_arg = MY_CHECK_ERROR | MY_GIVE_INFO;
if (debug_check_flag) my_end_arg = MY_CHECK_ERROR;
if (!path && (enclosed || opt_enclosed || escaped || lines_terminated ||
fields_terminated)) {
fprintf(stderr, "%s: You must use option --tab with --fields-...\n",
my_progname);
return (EX_USAGE);
}
if (opt_slave_data) {
opt_lock_all_tables = !opt_single_transaction;
opt_master_data = 0;
opt_delete_master_logs = 0;
}
if (opt_delete_master_logs && !opt_master_data)
opt_master_data = MYSQL_OPT_MASTER_DATA_COMMENTED_SQL;
if (opt_single_transaction && opt_lock_all_tables) {
fprintf(stderr,
"%s: You can't use --single-transaction and "
"--lock-all-tables at the same time.\n",
my_progname);
return (EX_USAGE);
}
if (opt_master_data) {
opt_lock_all_tables = !opt_single_transaction;
opt_slave_data = 0;
}
if (opt_single_transaction || opt_lock_all_tables) lock_tables = 0;
if (enclosed && opt_enclosed) {
fprintf(stderr,
"%s: You can't use ..enclosed.. and ..optionally-enclosed.. at the "
"same time.\n",
my_progname);
return (EX_USAGE);
}
if ((opt_databases || opt_alldbs) && path) {
fprintf(stderr,
"%s: --databases or --all-databases can't be used with --tab.\n",
my_progname);
return (EX_USAGE);
}
if (strcmp(default_charset, charset_info->csname) &&
!(charset_info =
get_charset_by_csname(default_charset, MY_CS_PRIMARY, MYF(MY_WME))))
exit(1);
return (0);
}
** DB_error -- prints mysql error message and exits the program.
*/
static void DB_error(MYSQL *mysql_arg, const char *when) {
DBUG_ENTER("DB_error");
maybe_die(EX_MYSQLERR, "Got error: %d: %s %s", mysql_errno(mysql_arg),
mysql_error(mysql_arg), when);
DBUG_VOID_RETURN;
}
Prints out an error message and kills the process.
SYNOPSIS
die()
error_num - process return value
fmt_reason - a format string for use by my_vsnprintf.
... - variable arguments for above fmt_reason string
DESCRIPTION
This call prints out the formatted error message to stderr and then
terminates the process.
*/
static void die(int error_num, const char *fmt_reason, ...) {
char buffer[1000];
va_list args;
va_start(args, fmt_reason);
my_vsnprintf(buffer, sizeof(buffer), fmt_reason, args);
va_end(args);
fprintf(stderr, "%s: %s\n", my_progname, buffer);
fflush(stderr);
opt_force = 0;
if (opt_ignore_error) my_free(opt_ignore_error);
opt_ignore_error = 0;
MySQLContext *context = get_connection_thread_context();
context->maybe_exit(error_num);
}
Prints out an error message and maybe kills the process.
SYNOPSIS
maybe_die()
error_num - process return value
fmt_reason - a format string for use by my_vsnprintf.
... - variable arguments for above fmt_reason string
DESCRIPTION
This call prints out the formatted error message to stderr and then
terminates the process, unless the --force command line option is used.
This call should be used for non-fatal errors (such as database
errors) that the code may still be able to continue to the next unit
of work.
*/
static void maybe_die(int error_num, const char *fmt_reason, ...) {
char buffer[1000];
va_list args;
va_start(args, fmt_reason);
my_vsnprintf(buffer, sizeof(buffer), fmt_reason, args);
va_end(args);
fprintf(stderr, "%s: %s\n", my_progname, buffer);
fflush(stderr);
MySQLContext *context = get_connection_thread_context();
if (context) {
context->maybe_exit(error_num);
}
}
Sends a query to server, optionally reads result, prints error message if
some.
SYNOPSIS
mysql_query_with_error_report()
mysql_con connection to use
res if non zero, result will be put there with
mysql_store_result()
query query to send to server
RETURN VALUES
0 query sending and (if res!=0) result reading went ok
1 error
*/
static int mysql_query_with_error_report(MYSQL *mysql_con, MYSQL_RES **res,
const char *query) {
if (mysql_query(mysql_con, query) ||
(res && !((*res) = mysql_store_result(mysql_con)))) {
maybe_die(EX_MYSQLERR, "Couldn't execute '%s': %s (%d)", query,
mysql_error(mysql_con), mysql_errno(mysql_con));
return 1;
}
return 0;
}
static int mysql_query_without_exit(MYSQL *mysql_con, MYSQL_RES **res,
const char *query) {
if (mysql_query(mysql_con, query) ||
(res && !((*res) = mysql_store_result(mysql_con)))) {
printf("Couldn't execute '%s': %s (%d)\n", query, mysql_error(mysql_con),
mysql_errno(mysql_con));
return 1;
}
return 0;
}
int MySQLContext::fetch_db_collation(const char *db_name, char *db_cl_name,
int db_cl_size) {
my_bool err_status = FALSE;
char query[QUERY_LENGTH];
MYSQL_RES *db_cl_res;
MYSQL_ROW db_cl_row;
char quoted_database_buf[NAME_LEN * 2 + 3];
char *qdatabase = quote_name(db_name, quoted_database_buf, 1);
my_snprintf(query, sizeof(query), "use %s", qdatabase);
if (mysql_query_with_error_report(mysql, NULL, query)) return 1;
if (mysql_query_with_error_report(mysql, &db_cl_res,
"select @@collation_database"))
return 1;
do {
if (mysql_num_rows(db_cl_res) != 1) {
err_status = TRUE;
break;
}
if (!(db_cl_row = mysql_fetch_row(db_cl_res))) {
err_status = TRUE;
break;
}
strncpy(db_cl_name, db_cl_row[0], db_cl_size - 1);
db_cl_name[db_cl_size - 1] = 0;
} while (FALSE);
mysql_free_result(db_cl_res);
return err_status ? 1 : 0;
}
static char *my_case_str(const char *str, size_t str_len, const char *token,
size_t token_len) {
my_match_t match;
uint status = my_charset_latin1.coll->instr(&my_charset_latin1, str, str_len,
token, token_len, &match, 1);
return status ? (char *)str + match.end : NULL;
}
int MySQLContext::switch_db_collation(FILE *sql_file, const char *db_name,
const char *delimiter,
const char *current_db_cl_name,
const char *required_db_cl_name,
int *db_cl_altered) {
if (strcmp(current_db_cl_name, required_db_cl_name) != 0) {
char quoted_db_buf[NAME_LEN * 2 + 3];
char *quoted_db_name = quote_name(db_name, quoted_db_buf, FALSE);
CHARSET_INFO *db_cl = get_charset_by_name(required_db_cl_name, MYF(0));
if (!db_cl) return 1;
fprintf(sql_file, "ALTER DATABASE %s CHARACTER SET %s COLLATE %s %s\n",
(const char *)quoted_db_name, (const char *)db_cl->csname,
(const char *)db_cl->name, (const char *)delimiter);
*db_cl_altered = 1;
return 0;
}
*db_cl_altered = 0;
return 0;
}
int MySQLContext::restore_db_collation(FILE *sql_file, const char *db_name,
const char *delimiter,
const char *db_cl_name) {
char quoted_db_buf[NAME_LEN * 2 + 3];
char *quoted_db_name = quote_name(db_name, quoted_db_buf, FALSE);
CHARSET_INFO *db_cl = get_charset_by_name(db_cl_name, MYF(0));
if (!db_cl) return 1;
fprintf(sql_file, "ALTER DATABASE %s CHARACTER SET %s COLLATE %s %s\n",
(const char *)quoted_db_name, (const char *)db_cl->csname,
(const char *)db_cl->name, (const char *)delimiter);
return 0;
}
static void switch_cs_variables(FILE *sql_file, const char *delimiter,
const char *character_set_client,
const char *character_set_results,
const char *collation_connection) {
fprintf(sql_file,
"/*!50003 SET @saved_cs_client = @@character_set_client */ %s\n"
"/*!50003 SET @saved_cs_results = @@character_set_results */ %s\n"
"/*!50003 SET @saved_col_connection = @@collation_connection */ %s\n"
"/*!50003 SET character_set_client = %s */ %s\n"
"/*!50003 SET character_set_results = %s */ %s\n"
"/*!50003 SET collation_connection = %s */ %s\n",
(const char *)delimiter, (const char *)delimiter,
(const char *)delimiter,
(const char *)character_set_client, (const char *)delimiter,
(const char *)character_set_results, (const char *)delimiter,
(const char *)collation_connection, (const char *)delimiter);
}
static void restore_cs_variables(FILE *sql_file, const char *delimiter) {
fprintf(sql_file,
"/*!50003 SET character_set_client = @saved_cs_client */ %s\n"
"/*!50003 SET character_set_results = @saved_cs_results */ %s\n"
"/*!50003 SET collation_connection = @saved_col_connection */ %s\n",
(const char *)delimiter, (const char *)delimiter,
(const char *)delimiter);
}
This function will remove specific sql mode.
@param[in] sql_mode Original sql mode from where input mode needs to
be removed.
@param[in] replace_mode sql mode which needs to be removed from original
sql mode.
@param[in] replace_len length of sql mode which needs to be removed.
@retval 1 replace_mode is not present
0 replace_mode is removed successfully
*/
static int remove_sql_mode(char *sql_mode, const char *replace_mode,
size_t replace_len) {
char *start = strstr(sql_mode, replace_mode);
if (!start) return 1;
if (strlen(start) == replace_len) {
if (start == sql_mode)
*start = 0;
else
start[-1] = 0;
} else {
const char *next = start + replace_len + 1;
memmove(start, next, strlen(next) + 1);
}
return 0;
}
static void switch_sql_mode(FILE *sql_file, const char *delimiter,
const char *sql_mode) {
fprintf(sql_file,
"/*!50003 SET @saved_sql_mode = @@sql_mode */ %s\n"
"/*!50003 SET sql_mode = '%s' */ %s\n",
(const char *)delimiter,
(const char *)sql_mode, (const char *)delimiter);
}
static void restore_sql_mode(FILE *sql_file, const char *delimiter) {
fprintf(sql_file,
"/*!50003 SET sql_mode = @saved_sql_mode */ %s\n",
(const char *)delimiter);
}
static void switch_time_zone(FILE *sql_file, const char *delimiter,
const char *time_zone) {
fprintf(sql_file,
"/*!50003 SET @saved_time_zone = @@time_zone */ %s\n"
"/*!50003 SET time_zone = '%s' */ %s\n",
(const char *)delimiter,
(const char *)time_zone, (const char *)delimiter);
}
static void restore_time_zone(FILE *sql_file, const char *delimiter) {
fprintf(sql_file,
"/*!50003 SET time_zone = @saved_time_zone */ %s\n",
(const char *)delimiter);
}
Switch charset for results to some specified charset. If the server does not
support character_set_results variable, nothing can be done here. As for
whether something should be done here, future new callers of this function
should be aware that the server lacking the facility of switching charsets is
treated as success.
@note If the server lacks support, then nothing is changed and no error
condition is returned.
@returns whether there was an error or not
*/
int MySQLContext::switch_character_set_results(MYSQL *mysql,
const char *cs_name) {
char query_buffer[QUERY_LENGTH];
size_t query_length;
if (!server_supports_switching_charsets) return FALSE;
query_length = my_snprintf(query_buffer, sizeof(query_buffer),
"SET SESSION character_set_results = '%s'",
(const char *)cs_name);
return mysql_real_query(mysql, query_buffer, (ulong)query_length);
}
Rewrite statement, enclosing DEFINER clause in version-specific comment.
This function parses any CREATE statement and encloses DEFINER-clause in
version-specific comment:
input query: CREATE DEFINER=a@b FUNCTION ...
rewritten query: CREATE * / / *!50020 DEFINER=a@b * / / *!50003 FUNCTION ...
@note This function will go away when WL#3995 is implemented.
@param[in] stmt_str CREATE statement string.
@param[in] stmt_length Length of the stmt_str.
@param[in] definer_version_str Minimal MySQL version number when
DEFINER clause is supported in the
given statement.
@param[in] definer_version_length Length of definer_version_str.
@param[in] stmt_version_str Minimal MySQL version number when the
given statement is supported.
@param[in] stmt_version_length Length of stmt_version_str.
@param[in] keyword_str Keyword to look for after CREATE.
@param[in] keyword_length Length of keyword_str.
@return pointer to the new allocated query string.
*/
static char *cover_definer_clause(const char *stmt_str, size_t stmt_length,
const char *definer_version_str,
size_t definer_version_length,
const char *stmt_version_str,
size_t stmt_version_length,
const char *keyword_str,
size_t keyword_length) {
char *definer_begin =
my_case_str(stmt_str, stmt_length, C_STRING_WITH_LEN(" DEFINER"));
char *definer_end = NULL;
char *query_str = NULL;
char *query_ptr;
LEX_CSTRING comment = {C_STRING_WITH_LEN("*/ /*!")};
if (!definer_begin) return NULL;
definer_end = my_case_str(definer_begin, strlen(definer_begin), keyword_str,
keyword_length);
if (!definer_end) return NULL;
Allocate memory for new query string: original string
from SHOW statement and version-specific comments.
*/
query_str = alloc_query_str(stmt_length + 23);
query_ptr = my_stpncpy(query_str, stmt_str, definer_begin - stmt_str);
query_ptr = my_stpncpy(query_ptr, comment.str, comment.length + 1);
query_ptr =
my_stpncpy(query_ptr, definer_version_str, definer_version_length);
query_ptr = my_stpncpy(query_ptr, definer_begin, definer_end - definer_begin);
query_ptr = my_stpncpy(query_ptr, comment.str, comment.length + 1);
query_ptr = my_stpncpy(query_ptr, stmt_version_str, stmt_version_length);
query_ptr = strxmov(query_ptr, definer_end, NullS);
return query_str;
}
Open a new .sql file to dump the table or view into
SYNOPSIS
open_sql_file_for_table
name name of the table or view
flags flags (as per "man 2 open")
RETURN VALUES
0 Failed to open file
> 0 Handle of the open file
*/
static FILE *open_sql_file_for_table(const char *table, int flags) {
FILE *res;
char filename[FN_REFLEN], tmp_path[FN_REFLEN];
convert_dirname(tmp_path, path, NullS);
res = my_fopen(fn_format(filename, table, tmp_path, ".sql", 4), flags,
MYF(MY_WME));
return res;
}
void MySQLContext::free_resources() {
if (md_result_file && md_result_file != stdout)
my_fclose(md_result_file, MYF(0));
my_free(opt_password);
if (my_hash_inited(&ignore_table)) my_hash_free(&ignore_table);
if (insert_pat_inited) dynstr_free(&insert_pat);
if (defaults_argv) free_defaults(defaults_argv);
if (opt_ignore_error) my_free(opt_ignore_error);
delete_dynamic(&ignore_error);
my_end(my_end_arg);
pthread_key_delete(sg_connectionkey);
}
Parse the list of error numbers to be ignored and store into a dynamic
array.
@return Operation status
@retval 0 Success
@retval >0 Failure
*/
static int parse_ignore_error() {
const char *search = ",";
char *token;
uint my_err;
DBUG_ENTER("parse_ignore_error");
if (my_init_dynamic_array(&ignore_error, PSI_NOT_INSTRUMENTED, sizeof(uint),
NULL, 12, 12))
goto error;
token = strtok(opt_ignore_error, search);
while (token != NULL) {
my_err = atoi(token);
if (my_err != 0) {
if (insert_dynamic(&ignore_error, &my_err)) goto error;
}
token = strtok(NULL, search);
}
DBUG_RETURN(0);
error:
DBUG_RETURN(EX_EOM);
}
Check if the last error should be ignored.
@retval 1 yes
0 no
*/
my_bool MySQLContext::do_ignore_error() {
uint i, last_errno, *my_err;
my_bool found = 0;
DBUG_ENTER("do_ignore_error");
last_errno = mysql_errno(mysql);
if (last_errno == 0) goto done;
for (i = 0; i < ignore_error.elements; i++) {
my_err = dynamic_element(&ignore_error, i, uint *);
if (last_errno == *my_err) {
found = 1;
break;
}
}
done:
DBUG_RETURN(found);
}
void MySQLContext::maybe_exit(int error) {
if (!first_error) first_error = error;
Return if --force is used; else return only if the
last error number is in the list of error numbers
specified using --ignore-error option.
*/
if (opt_force || (opt_ignore_error && do_ignore_error())) return;
if (mysql) mysql_close(mysql);
free_resources();
exit(error);
}
db_connect -- connects to the host and selects DB.
*/
int MySQLContext::connect_to_db(char *host, char *user, char *passwd) {
char buff[20 + FN_REFLEN];
DBUG_ENTER("connect_to_db");
verbose_msg("-- Connecting to %s...\n", host ? host : "localhost");
mysql_init(&mysql_connection);
if (opt_compress) mysql_options(&mysql_connection, MYSQL_OPT_COMPRESS, NullS);
SSL_SET_OPTIONS(&mysql_connection);
if (opt_protocol)
mysql_options(&mysql_connection, MYSQL_OPT_PROTOCOL, (char *)&opt_protocol);
if (opt_bind_addr)
mysql_options(&mysql_connection, MYSQL_OPT_BIND, opt_bind_addr);
#if defined(_WIN32) && !defined(EMBEDDED_LIBRARY)
if (shared_memory_base_name)
mysql_options(&mysql_connection, MYSQL_SHARED_MEMORY_BASE_NAME,
shared_memory_base_name);
#endif
mysql_options(&mysql_connection, MYSQL_SET_CHARSET_NAME, default_charset);
if (opt_plugin_dir && *opt_plugin_dir)
mysql_options(&mysql_connection, MYSQL_PLUGIN_DIR, opt_plugin_dir);
if (opt_default_auth && *opt_default_auth)
mysql_options(&mysql_connection, MYSQL_DEFAULT_AUTH, opt_default_auth);
if (using_opt_enable_cleartext_plugin)
mysql_options(&mysql_connection, MYSQL_ENABLE_CLEARTEXT_PLUGIN,
(char *)&opt_enable_cleartext_plugin);
mysql_options(&mysql_connection, MYSQL_OPT_CONNECT_ATTR_RESET, 0);
mysql_options4(&mysql_connection, MYSQL_OPT_CONNECT_ATTR_ADD, "program_name",
"mysqldump");
set_server_public_key(&mysql_connection);
set_get_server_public_key_option(&mysql_connection);
if (!(mysql = mysql_real_connect(&mysql_connection, host, user, passwd, NULL,
opt_mysql_port, opt_mysql_unix_port, 0))) {
DB_error(&mysql_connection, "when trying to connect");
DBUG_RETURN(1);
}
if ((mysql_get_server_version(&mysql_connection) < 40100) ||
(opt_compatible_mode & 3)) {
opt_set_charset = 0;
server_supports_switching_charsets = FALSE;
}
As we're going to set SQL_MODE, it would be lost on reconnect, so we
cannot reconnect.
*/
mysql->reconnect = 0;
my_snprintf(buff, sizeof(buff), "/*!40100 SET @@SQL_MODE='%s' */",
compatible_mode_normal_str);
if (mysql_query_without_exit(mysql, 0, buff)) {
printf("cannot set sql_mode to %s, set sql_mode to ansi and retry",
compatible_mode_normal_str);
my_snprintf(buff, sizeof(buff), "/*!40100 SET @@SQL_MODE='ANSI' */");
if (mysql_query_with_error_report(mysql, 0, buff)) {
DBUG_RETURN(1);
}
}
set time_zone to UTC to allow dumping date types between servers with
different time zone settings
*/
if (opt_tz_utc) {
my_snprintf(buff, sizeof(buff), "/*!40103 SET TIME_ZONE='+00:00' */");
if (mysql_query_with_error_report(mysql, 0, buff)) DBUG_RETURN(1);
}
DBUG_RETURN(0);
}
** dbDisconnect -- disconnects from the host.
*/
void MySQLContext::dbDisconnect(char *host) {
verbose_msg("-- Disconnecting from %s...\n", host ? host : "localhost");
mysql_close(mysql);
}
void MySQLContext::unescape(FILE *file, char *pos, size_t length,
DYNAMIC_STRING *output) {
char *tmp;
DBUG_ENTER("unescape");
if (!(tmp = (char *)my_malloc(PSI_NOT_INSTRUMENTED, length * 2 + 1,
MYF(MY_WME))))
die(EX_MYSQLERR, "Couldn't allocate memory");
mysql_real_escape_string_quote(&mysql_connection, tmp, pos, (ulong)length,
'\'');
fputc('\'', file);
fputs(tmp, file);
fputc('\'', file);
if (output) {
dynstr_append_checked(output, "\'");
dynstr_append_checked(output, tmp);
dynstr_append_checked(output, "\'");
}
check_io(file);
my_free(tmp);
DBUG_VOID_RETURN;
}
void MySQLContext::unescapestr(char *pos, size_t length,
DYNAMIC_STRING *output) {
char *tmp;
DBUG_ENTER("unescape");
if (!(tmp = (char *)my_malloc(PSI_NOT_INSTRUMENTED, length * 2 + 1,
MYF(MY_WME))))
die(EX_MYSQLERR, "Couldn't allocate memory");
mysql_real_escape_string_quote(&mysql_connection, tmp, pos, (ulong)length,
'\'');
if (output) {
dynstr_append_checked(output, "\'");
dynstr_append_checked(output, tmp);
dynstr_append_checked(output, "\'");
}
my_free(tmp);
DBUG_VOID_RETURN;
}
static my_bool test_if_special_chars(const char *str) {
for (; *str; str++)
if (!my_isvar(charset_info, *str) && *str != '$') return 1;
return 0;
}
quote_name(name, buff, force)
Quotes char string, taking into account compatible mode
Args
name Unquoted string containing that which will be quoted
buff The buffer that contains the quoted value, also returned
force Flag to make it ignore 'test_if_special_chars'
Returns
buff quoted string
*/
char *MySQLContext::quote_name(const char *name, char *buff, my_bool force) {
char *to = buff;
char qtype = (opt_compatible_mode & MASK_ANSI_QUOTES) ? '\"' : '`';
if (!force && !opt_quoted && !test_if_special_chars(name))
return (char *)name;
*to++ = qtype;
while (*name) {
if (*name == qtype) *to++ = qtype;
*to++ = *name++;
}
to[0] = qtype;
to[1] = 0;
return buff;
}
Quote a table name so it can be used in "SHOW TABLES LIKE <tabname>"
SYNOPSIS
quote_for_like()
name name of the table
buff quoted name of the table
DESCRIPTION
Quote \, _, ' and % characters
Note: Because MySQL uses the C escape syntax in strings
(for example, '\n' to represent newline), you must double
any '\' that you use in your LIKE strings. For example, to
search for '\n', specify it as '\\n'. To search for '\', specify
it as '\\\\' (the backslashes are stripped once by the parser
and another time when the pattern match is done, leaving a
single backslash to be matched).
Example: "t\1" => "t\\\\1"
*/
static char *quote_for_like(const char *name, char *buff) {
char *to = buff;
*to++ = '\'';
while (*name) {
if (*name == '\\') {
*to++ = '\\';
*to++ = '\\';
*to++ = '\\';
} else if (*name == '\'' || *name == '_' || *name == '%')
*to++ = '\\';
*to++ = *name++;
}
to[0] = '\'';
to[1] = 0;
return buff;
}
Quote and print a string.
@param xml_file - Output file.
@param str - String to print.
@param len - Its length.
@param is_attribute_name - A check for attribute name or value.
@description
Quote '<' '>' '&' '\"' chars and print a string to the xml_file.
*/
static void print_quoted_xml(FILE *xml_file, const char *str, size_t len,
my_bool is_attribute_name) {
const char *end;
for (end = str + len; str != end; str++) {
switch (*str) {
case '<':
fputs("<", xml_file);
break;
case '>':
fputs(">", xml_file);
break;
case '&':
fputs("&", xml_file);
break;
case '\"':
fputs(""", xml_file);
break;
case ' ':
if (is_attribute_name) {
fputs("_", xml_file);
break;
}
default:
fputc(*str, xml_file);
break;
}
}
check_io(xml_file);
}
Print xml tag. Optionally add attribute(s).
SYNOPSIS
print_xml_tag(xml_file, sbeg, send, tag_name, first_attribute_name,
..., attribute_name_n, attribute_value_n, NullS)
xml_file - output file
sbeg - line beginning
line_end - line ending
tag_name - XML tag name.
first_attribute_name - tag and first attribute
first_attribute_value - (Implied) value of first attribute
attribute_name_n - attribute n
attribute_value_n - value of attribute n
DESCRIPTION
Print XML tag with any number of attribute="value" pairs to the xml_file.
Format is:
sbeg<tag_name first_attribute_name="first_attribute_value" ...
attribute_name_n="attribute_value_n">send
NOTE
Additional arguments must be present in attribute/value pairs.
The last argument should be the null character pointer.
All attribute_value arguments MUST be NULL terminated strings.
All attribute_value arguments will be quoted before output.
*/
static void print_xml_tag(FILE *xml_file, const char *sbeg,
const char *line_end, const char *tag_name,
const char *first_attribute_name, ...) {
va_list arg_list;
const char *attribute_name, *attribute_value;
fputs(sbeg, xml_file);
fputc('<', xml_file);
fputs(tag_name, xml_file);
va_start(arg_list, first_attribute_name);
attribute_name = first_attribute_name;
while (attribute_name != NullS) {
attribute_value = va_arg(arg_list, char *);
DBUG_ASSERT(attribute_value != NullS);
fputc(' ', xml_file);
fputs(attribute_name, xml_file);
fputc('\"', xml_file);
print_quoted_xml(xml_file, attribute_value, strlen(attribute_value), 0);
fputc('\"', xml_file);
attribute_name = va_arg(arg_list, char *);
}
va_end(arg_list);
fputc('>', xml_file);
fputs(line_end, xml_file);
check_io(xml_file);
}
Print xml tag with for a field that is null
SYNOPSIS
print_xml_null_tag()
xml_file - output file
sbeg - line beginning
stag_atr - tag and attribute
sval - value of attribute
line_end - line ending
DESCRIPTION
Print tag with one attribute to the xml_file. Format is:
<stag_atr="sval" xsi:nil="true"/>
NOTE
sval MUST be a NULL terminated string.
sval string will be qouted before output.
*/
static void print_xml_null_tag(FILE *xml_file, const char *sbeg,
const char *stag_atr, const char *sval,
const char *line_end) {
fputs(sbeg, xml_file);
fputs("<", xml_file);
fputs(stag_atr, xml_file);
fputs("\"", xml_file);
print_quoted_xml(xml_file, sval, strlen(sval), 0);
fputs("\" xsi:nil=\"true\" />", xml_file);
fputs(line_end, xml_file);
check_io(xml_file);
}
Print xml CDATA section.
@param xml_file - output file
@param str - string to print
@param len - length of the string
@note
This function also takes care of the presence of '[[>'
string in the str. If found, the CDATA section is broken
into two CDATA sections, <![CDATA[]]]]> and <![CDATA[>]].
*/
static void print_xml_cdata(FILE *xml_file, const char *str, ulong len) {
const char *end;
fputs("<![CDATA[\n", xml_file);
for (end = str + len; str != end; str++) {
switch (*str) {
case ']':
if ((*(str + 1) == ']') && (*(str + 2) == '>')) {
fputs("]]]]><![CDATA[>", xml_file);
str += 2;
continue;
}
default:
fputc(*str, xml_file);
break;
}
}
fputs("\n]]>\n", xml_file);
check_io(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
str_create - create statement header string
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,
const char *str_create) {
uint i;
char *create_stmt_ptr = NULL;
ulong create_stmt_len = 0;
MYSQL_FIELD *field;
ulong *lengths = mysql_fetch_lengths(tableRes);
fprintf(xml_file, "\t\t<%s", row_name);
check_io(xml_file);
mysql_field_seek(tableRes, 0);
for (i = 0; (field = mysql_fetch_field(tableRes)); i++) {
if ((*row)[i]) {
if ((str_create) && (strcmp(str_create, field->name) == 0)) {
create_stmt_ptr = (*row)[i];
create_stmt_len = lengths[i];
} else {
fputc(' ', xml_file);
print_quoted_xml(xml_file, field->name, field->name_length, 1);
fputs("=\"", xml_file);
print_quoted_xml(xml_file, (*row)[i], lengths[i], 0);
fputc('"', xml_file);
check_io(xml_file);
}
}
}
if (create_stmt_len) {
fputs(">\n", xml_file);
print_xml_cdata(xml_file, create_stmt_ptr, create_stmt_len);
fprintf(xml_file, "\t\t</%s>\n", row_name);
} else
fputs(" />\n", xml_file);
check_io(xml_file);
}
Print xml comments.
@param xml_file - output file
@param len - length of comment message
@param comment_string - comment message
@description
Print the comment message in the format:
"<!-- \n comment string \n -->\n"
@note
Any occurrence of continuous hyphens will be
squeezed to a single hyphen.
*/
static void print_xml_comment(FILE *xml_file, size_t len,
const char *comment_string) {
const char *end;
fputs("<!-- ", xml_file);
for (end = comment_string + len; comment_string != end; comment_string++) {
The string "--" (double-hyphen) MUST NOT occur within xml comments.
*/
switch (*comment_string) {
case '-':
if (*(comment_string + 1) == '-')
break;
default:
fputc(*comment_string, xml_file);
break;
}
}
fputs(" -->\n", xml_file);
check_io(xml_file);
}
static void print_comment(FILE *sql_file, my_bool is_error, const char *format,
...) {
static char comment_buff[COMMENT_LENGTH];
va_list args;
if (!is_error && !opt_comments) return;
va_start(args, format);
my_vsnprintf(comment_buff, COMMENT_LENGTH, format, args);
va_end(args);
if (!opt_xml) {
fputs(comment_buff, sql_file);
check_io(sql_file);
return;
}
print_xml_comment(sql_file, strlen(comment_buff), comment_buff);
}
@brief Accepts object names and prefixes them with "-- " wherever
end-of-line character ('\n') is found.
@param[in] object_name object name list (concatenated string)
@param[out] freemem should buffer be released after usage
@return
@retval pointer to a string with prefixed objects
*/
static char const *fix_identifier_with_newline(char const *object_name,
my_bool *freemem) {
const size_t PREFIX_LENGTH = 3;
static char storage[NAME_LEN + 1];
static char *buffer = storage;
static size_t buffer_size = sizeof(storage) - 1;
size_t index = 0;
size_t required_size = 0;
*freemem = FALSE;
while (object_name && *object_name) {
++required_size;
if (*object_name == '\n') required_size += PREFIX_LENGTH;
if (required_size > buffer_size) {
buffer_size = COMMENT_LENGTH * (1 + required_size / COMMENT_LENGTH);
if (*freemem) {
buffer = (char *)my_realloc(PSI_NOT_INSTRUMENTED, buffer,
buffer_size + 1, MYF(MY_WME));
if (!buffer) exit(1);
} else {
buffer = (char *)my_malloc(PSI_NOT_INSTRUMENTED, buffer_size + 1,
MYF(MY_WME));
if (!buffer) exit(1);
strncpy(buffer, storage, index);
*freemem = TRUE;
}
}
buffer[index] = *object_name;
++index;
if (*object_name == '\n') {
strcpy(buffer + index, "-- ");
index += PREFIX_LENGTH;
}
++object_name;
}
buffer[index] = '\0';
return buffer;
}
create_delimiter
Generate a new (null-terminated) string that does not exist in query
and is therefore suitable for use as a query delimiter. Store this
delimiter in delimiter_buff .
This is quite simple in that it doesn't even try to parse statements as an
interpreter would. It merely returns a string that is not in the query, which
is much more than adequate for constructing a delimiter.
RETURN
ptr to the delimiter on Success
NULL on Failure
*/
static char *create_delimiter(char *query, char *delimiter_buff,
int delimiter_max_size) {
int proposed_length;
char *presence;
delimiter_buff[0] = ';';
for (proposed_length = 2; proposed_length < delimiter_max_size;
delimiter_max_size++) {
delimiter_buff[proposed_length - 1] = ';';
delimiter_buff[proposed_length] = '\0';
presence = strstr(query, delimiter_buff);
if (presence == NULL) {
return delimiter_buff;
}
}
return NULL;
}
dump_events_for_db
-- retrieves list of events for a given db, and prints out
the CREATE EVENT statement into the output (the dump).
RETURN
0 Success
1 Error
*/
uint MySQLContext::dump_events_for_db(char *db) {
char query_buff[QUERY_LENGTH];
char db_name_buff[NAME_LEN * 2 + 3], name_buff[NAME_LEN * 2 + 3];
char *event_name;
char delimiter[QUERY_LENGTH];
FILE *sql_file = md_result_file;
MYSQL_RES *event_res, *event_list_res;
MYSQL_ROW row, event_list_row;
char db_cl_name[MY_CS_NAME_SIZE];
int db_cl_altered = FALSE;
my_bool freemem = FALSE;
char const *text = fix_identifier_with_newline(db, &freemem);
DBUG_ENTER("dump_events_for_db");
DBUG_PRINT("enter", ("db: '%s'", db));
mysql_real_escape_string_quote(mysql, db_name_buff, db, (ulong)strlen(db),
'\'');
print_comment(sql_file, 0, "\n--\n-- Dumping events for database '%s'\n--\n",
text);
if (freemem) my_free((void *)text);
not using "mysql_query_with_error_report" because we may have not
enough privileges to lock mysql.events.
*/
if (lock_tables) mysql_query(mysql, "LOCK TABLES mysql.event READ");
if (mysql_query_with_error_report(mysql, &event_list_res, "show events"))
DBUG_RETURN(0);
strcpy(delimiter, ";");
if (mysql_num_rows(event_list_res) > 0) {
if (opt_xml)
fputs("\t<events>\n", sql_file);
else {
fprintf(sql_file, "/*!50106 SET @save_time_zone= @@TIME_ZONE */ ;\n");
if (fetch_db_collation(db_name_buff, db_cl_name, sizeof(db_cl_name)))
DBUG_RETURN(1);
}
if (switch_character_set_results(mysql, "binary")) DBUG_RETURN(1);
while ((event_list_row = mysql_fetch_row(event_list_res)) != NULL) {
event_name = quote_name(event_list_row[1], name_buff, 0);
DBUG_PRINT("info", ("retrieving CREATE EVENT for %s", name_buff));
my_snprintf(query_buff, sizeof(query_buff), "SHOW CREATE EVENT %s",
event_name);
if (mysql_query_with_error_report(mysql, &event_res, query_buff))
DBUG_RETURN(1);
while ((row = mysql_fetch_row(event_res)) != NULL) {
if (opt_xml) {
print_xml_row(sql_file, "event", event_res, &row, "Create Event");
continue;
}
if the user has EXECUTE privilege he can see event names, but not the
event body!
*/
if (strlen(row[3]) != 0) {
char *query_str;
if (opt_drop)
fprintf(sql_file, "/*!50106 DROP EVENT IF EXISTS %s */%s\n",
event_name, delimiter);
if (create_delimiter(row[3], delimiter, sizeof(delimiter)) == NULL) {
fprintf(stderr,
"%s: Warning: Can't create delimiter for event '%s'\n",
my_progname, event_name);
DBUG_RETURN(1);
}
fprintf(sql_file, "DELIMITER %s\n", delimiter);
if (mysql_num_fields(event_res) >= 7) {
if (switch_db_collation(sql_file, db_name_buff, delimiter,
db_cl_name, row[6], &db_cl_altered)) {
DBUG_RETURN(1);
}
switch_cs_variables(sql_file, delimiter,
row[4],
row[4],
row[5]);
} else {
mysqldump is being run against the server, that does not
provide character set information in SHOW CREATE
statements.
NOTE: the dump may be incorrect, since character set
information is required in order to restore event properly.
*/
fprintf(sql_file,
"--\n"
"-- WARNING: old server version. "
"The following dump may be incomplete.\n"
"--\n");
}
remove_sql_mode(row[1], C_STRING_WITH_LEN("NO_AUTO_CREATE_USER"));
switch_sql_mode(sql_file, delimiter, row[1]);
switch_time_zone(sql_file, delimiter, row[2]);
query_str = cover_definer_clause(
row[3], strlen(row[3]), C_STRING_WITH_LEN("50117"),
C_STRING_WITH_LEN("50106"), C_STRING_WITH_LEN(" EVENT"));
fprintf(sql_file, "/*!50106 %s */ %s\n",
(const char *)(query_str != NULL ? query_str : row[3]),
(const char *)delimiter);
my_free(query_str);
restore_time_zone(sql_file, delimiter);
restore_sql_mode(sql_file, delimiter);
if (mysql_num_fields(event_res) >= 7) {
restore_cs_variables(sql_file, delimiter);
if (db_cl_altered) {
if (restore_db_collation(sql_file, db_name_buff, delimiter,
db_cl_name))
DBUG_RETURN(1);
}
}
}
}
mysql_free_result(event_res);
}
if (opt_xml) {
fputs("\t</events>\n", sql_file);
check_io(sql_file);
} else {
fprintf(sql_file, "DELIMITER ;\n");
fprintf(sql_file, "/*!50106 SET TIME_ZONE= @save_time_zone */ ;\n");
}
if (switch_character_set_results(mysql, default_charset)) DBUG_RETURN(1);
}
mysql_free_result(event_list_res);
if (lock_tables)
(void)mysql_query_with_error_report(mysql, 0, "UNLOCK TABLES");
DBUG_RETURN(0);
}
Print hex value for blob data.
SYNOPSIS
print_blob_as_hex()
output_file - output file
str - string to print
len - its length
DESCRIPTION
Print hex value for blob data.
*/
static void print_blob_as_hex(FILE *output_file, const char *str, ulong len) {
const char *ptr = str, *end = ptr + len;
for (; ptr < end; ptr++) fprintf(output_file, "%02X", *((uchar *)ptr));
check_io(output_file);
}
static void append_blob_as_hex(DYNAMIC_STRING *output, const char *str,
ulong len) {
static char tmp[256];
const char *ptr = str, *end = ptr + len;
for (; ptr < end; ptr++) {
sprintf(tmp, "%02X", *((uchar *)ptr));
dynstr_append_checked(output, tmp);
}
}
dump_routines_for_db
-- retrieves list of routines for a given db, and prints out
the CREATE PROCEDURE definition into the output (the dump).
This function has logic to print the appropriate syntax depending on whether
this is a procedure or functions
RETURN
0 Success
1 Error
*/
uint MySQLContext::dump_routines_for_db(char *db) {
char query_buff[QUERY_LENGTH];
const char *routine_type[] = {"FUNCTION", "PROCEDURE"};
char db_name_buff[NAME_LEN * 2 + 3], name_buff[NAME_LEN * 2 + 3];
char *routine_name;
int i;
FILE *sql_file = md_result_file;
MYSQL_RES *routine_res, *routine_list_res;
MYSQL_ROW row, routine_list_row;
std::string sql;
char db_cl_name[MY_CS_NAME_SIZE];
my_bool freemem = FALSE;
char const *text = fix_identifier_with_newline(db, &freemem);
struct DMODBC::DBError dbError;
bool isfunction;
DBUG_ENTER("dump_routines_for_db");
DBUG_PRINT("enter", ("db: '%s'", db));
mysql_real_escape_string_quote(mysql, db_name_buff, db, (ulong)strlen(db),
'\'');
print_comment(sql_file, 0,
"\n--\n-- Dumping routines for database '%s'\n--\n", text);
if (freemem) my_free((void *)text);
not using "mysql_query_with_error_report" because we may have not
enough privileges to lock mysql.proc.
*/
if (lock_tables) mysql_query(mysql, "LOCK TABLES mysql.proc READ");
if (fetch_db_collation(db_name_buff, db_cl_name, sizeof(db_cl_name)))
DBUG_RETURN(1);
if (switch_character_set_results(mysql, "binary")) DBUG_RETURN(1);
if (opt_xml) fputs("\t<routines>\n", sql_file);
for (i = 0; i <= 1; i++) {
my_snprintf(query_buff, sizeof(query_buff),
"SHOW %s STATUS WHERE Db = '%s'", routine_type[i],
db_name_buff);
if (mysql_query_with_error_report(mysql, &routine_list_res, query_buff))
DBUG_RETURN(1);
if (mysql_num_rows(routine_list_res)) {
while ((routine_list_row = mysql_fetch_row(routine_list_res))) {
routine_name = quote_name(routine_list_row[1], name_buff, 0);
DBUG_PRINT("info",
("retrieving CREATE %s for %s", routine_type[i], name_buff));
my_snprintf(query_buff, sizeof(query_buff), "SHOW CREATE %s %s",
routine_type[i], routine_name);
if (mysql_query_with_error_report(mysql, &routine_res, query_buff))
DBUG_RETURN(1);
while ((row = mysql_fetch_row(routine_res))) {
if the user has EXECUTE privilege he see routine names, but NOT the
routine body of other routines that are not the creator of!
*/
DBUG_PRINT("info",
("length of body for %s row[2] '%s' is %zu", routine_name,
row[2] ? row[2] : "(null)", row[2] ? strlen(row[2]) : 0));
if (row[2] == NULL) {
my_bool freemem = FALSE;
char const *text =
fix_identifier_with_newline(current_user, &freemem);
print_comment(sql_file, 1, "\n-- insufficient privileges to %s\n",
query_buff);
print_comment(sql_file, 1,
"-- does %s have permissions on mysql.proc?\n\n",
text);
if (freemem) my_free((void *)text);
maybe_die(EX_MYSQLERR, "%s has insufficent privileges to %s!",
current_user, query_buff);
} else if (strlen(row[2])) {
sql = sg_procOdbc->procProcedureFunctionSql((const char *)row[2],
isfunction);
if (!sql.empty()) {
if (isfunction) {
sg_procMetrics->addFunctioncount(1);
sg_procMetrics->startSql("function", "function", sql.c_str());
} else {
sg_procMetrics->addProcedurecount(1);
sg_procMetrics->startSql("procedure", "procedure", sql.c_str());
}
if (sg_procOdbc->executeGeneralSql(sql.c_str(), &dbError)) {
if (isfunction) {
sg_procMetrics->endSql("function", "function", sql.c_str(),
false, "", false);
} else {
sg_procMetrics->endSql("procedure", "procedure", sql.c_str(),
false, "", false);
}
} else {
if (isfunction) {
sg_procMetrics->endSql("function", "function", sql.c_str(),
true, dbError.error.c_str(), false);
} else {
sg_procMetrics->endSql("procedure", "procedure", sql.c_str(),
true, dbError.error.c_str(), false);
}
}
}
}
}
mysql_free_result(routine_res);
}
}
mysql_free_result(routine_list_res);
}
if (opt_xml) {
fputs("\t</routines>\n", sql_file);
check_io(sql_file);
}
if (switch_character_set_results(mysql, default_charset)) DBUG_RETURN(1);
if (lock_tables)
(void)mysql_query_with_error_report(mysql, 0, "UNLOCK TABLES");
DBUG_RETURN(0);
}
static inline my_bool general_log_or_slow_log_tables(const char *db,
const char *table) {
return (!my_strcasecmp(charset_info, db, "mysql")) &&
(!my_strcasecmp(charset_info, table, "general_log") ||
!my_strcasecmp(charset_info, table, "slow_log"));
}
slave_master_info,slave_relay_log_info and gtid_executed tables under
mysql database
*/
static inline my_bool replication_metadata_tables(const char *db,
const char *table) {
return (!my_strcasecmp(charset_info, db, "mysql")) &&
(!my_strcasecmp(charset_info, table, "slave_master_info") ||
!my_strcasecmp(charset_info, table, "slave_relay_log_info") ||
!my_strcasecmp(charset_info, table, "gtid_executed"));
}
get_table_structure -- retrievs database structure, prints out corresponding
CREATE statement and fills out insert_pat if the table is the type we will
be dumping.
ARGS
table - table name
db - db name
table_type - table type, e.g. "MyISAM" or "InnoDB", but also "VIEW"
ignore_flag - what we must particularly ignore - see IGNORE_ defines above
real_columns- Contains one byte per column, 0 means unused, 1 is used
Generated columns are marked as unused
RETURN
number of fields in table, 0 if error
*/
uint MySQLContext::get_table_structure(char *table, char *db, char *table_type,
char *ignore_flag,
my_bool real_columns[]) {
my_bool init = 0, write_data, complete_insert;
my_ulonglong num_fields;
char *result_table, *opt_quoted_table;
const char *insert_option;
char name_buff[NAME_LEN + 3], table_buff[NAME_LEN * 2 + 3];
char table_buff2[NAME_LEN * 2 + 3], query_buff[QUERY_LENGTH];
const char *show_fields_stmt =
"SELECT `COLUMN_NAME` AS `Field`, "
"`COLUMN_TYPE` AS `Type`, "
"`IS_NULLABLE` AS `Null`, "
"`COLUMN_KEY` AS `Key`, "
"`COLUMN_DEFAULT` AS `Default`, "
"`EXTRA` AS `Extra`, "
"`COLUMN_COMMENT` AS `Comment` "
"FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE "
"TABLE_SCHEMA = '%s' AND TABLE_NAME = '%s'";
const char *show_increment =
"SELECT `COLUMN_NAME` AS `Field`"
"FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE "
"TABLE_SCHEMA = '%s' AND TABLE_NAME = '%s' AND EXTRA='auto_increment'";
FILE *sql_file = md_result_file;
size_t len;
my_bool is_log_table;
my_bool is_replication_metadata_table;
unsigned int colno;
MYSQL_RES *result;
MYSQL_ROW row;
std::string incrementkey;
DBUG_ENTER("get_table_structure");
DBUG_PRINT("enter", ("db: %s table: %s", db, table));
*ignore_flag = check_if_ignore_table(table, table_type);
complete_insert = 0;
if ((write_data = !(*ignore_flag & IGNORE_DATA))) {
complete_insert = opt_complete_insert;
if (!insert_pat_inited) {
insert_pat_inited = 1;
init_dynamic_string_checked(&insert_pat, "", 1024, 1024);
} else
dynstr_set_checked(&insert_pat, "");
}
insert_option = (opt_ignore ? " IGNORE " : "");
verbose_msg("-- Retrieving table structure for table %s...\n", table);
len =
my_snprintf(query_buff, sizeof(query_buff),
"SET SQL_QUOTE_SHOW_CREATE=%d", (opt_quoted || opt_keywords));
if (!create_options)
my_stpcpy(query_buff + len,
"/*!40102 ,SQL_MODE=concat(@@sql_mode, _utf8 "
"',NO_KEY_OPTIONS,NO_TABLE_OPTIONS,NO_FIELD_OPTIONS') */");
result_table = quote_name(table, table_buff, 1);
opt_quoted_table = quote_name(table, table_buff2, 0);
if (opt_order_by_primary) order_by = primary_key_fields(result_table);
char buff4ik[20 + FN_REFLEN];
my_snprintf(buff4ik, sizeof(buff4ik), show_increment, db, table);
if (!mysql_query_with_error_report(mysql, &result, buff4ik)) {
if (result->row_count > 0) {
row = mysql_fetch_row(result);
incrementkey = "\"";
incrementkey += row[0];
incrementkey += "\"";
}
mysql_free_result(result);
}
if (!opt_xml && !mysql_query_with_error_report(mysql, 0, query_buff)) {
if (!opt_no_create_info) {
char buff[20 + FN_REFLEN];
MYSQL_FIELD *field;
my_bool freemem = FALSE;
char const *text;
my_snprintf(buff, sizeof(buff), "show create table %s", result_table);
if (switch_character_set_results(mysql, "binary") ||
mysql_query_with_error_report(mysql, &result, buff) ||
switch_character_set_results(mysql, default_charset))
DBUG_RETURN(0);
if (path) {
if (!(sql_file = open_sql_file_for_table(table, O_WRONLY)))
DBUG_RETURN(0);
write_header(sql_file, db);
}
text = fix_identifier_with_newline(result_table, &freemem);
if (freemem) my_free((void *)text);
field = mysql_fetch_field_direct(result, 0);
if (strcmp(field->name, "View") == 0) {
char *scv_buff = NULL;
my_ulonglong n_cols;
verbose_msg("-- It's a view, create dummy view\n");
if ((row = mysql_fetch_row(result)) && (scv_buff = row[1]))
scv_buff = my_strdup(PSI_NOT_INSTRUMENTED, scv_buff, MYF(0));
mysql_free_result(result);
Create a table with the same name as the view and with columns of
the same name in order to satisfy views that depend on this view.
The table will be removed when the actual view is created.
The properties of each column, are not preserved in this temporary
table, because they are not necessary.
This will not be necessary once we can determine dependencies
between views and can simply dump them in the appropriate order.
*/
my_snprintf(query_buff, sizeof(query_buff), "SHOW FIELDS FROM %s",
result_table);
if (switch_character_set_results(mysql, "binary") ||
mysql_query_with_error_report(mysql, &result, query_buff) ||
switch_character_set_results(mysql, default_charset)) {
View references invalid or privileged table/col/fun (err 1356),
so we cannot create a stand-in table. Be defensive and dump
a comment with the view's 'show create' statement. (Bug #17371)
*/
if (mysql_errno(mysql) == ER_VIEW_INVALID)
fprintf(sql_file, "\n-- failed on view %s: %s\n\n", result_table,
scv_buff ? scv_buff : "");
my_free(scv_buff);
DBUG_RETURN(0);
} else
my_free(scv_buff);
n_cols = mysql_num_rows(result);
if (0 != n_cols) {
The actual formula is based on the column names and how the .FRM
files are stored and is too volatile to be repeated here.
Thus we simply warn the user if the columns exceed a limit we
know works most of the time.
*/
if (n_cols >= 1000)
fprintf(stderr,
"-- Warning: Creating a stand-in table for view %s may"
" fail when replaying the dump file produced because "
"of the number of columns exceeding 1000. Exercise "
"caution when replaying the produced dump file.\n",
table);
Get first row, following loop will prepend comma - keeps from
having to know if the row being printed is last to determine if
there should be a _trailing_ comma.
*/
row = mysql_fetch_row(result);
A temporary view is created to resolve the view interdependencies.
This temporary view is dropped when the actual view is created.
*/
}
mysql_free_result(result);
if (path) my_fclose(sql_file, MYF(MY_WME));
seen_views = 1;
DBUG_RETURN(0);
}
row = mysql_fetch_row(result);
is_log_table = general_log_or_slow_log_tables(db, table);
is_replication_metadata_table = replication_metadata_tables(db, table);
if (is_log_table || is_replication_metadata_table)
row[1] += 13;
if (sg_firstdumptables) {
sg_procMetrics->addTablecount(1);
}
std::list<std::string> createsqls =
sg_procOdbc->procCreateTableSql(row[1], incrementkey.c_str());
bool hastable = false;
struct DMODBC::DBError dbError;
for (std::list<std::string>::iterator iter = createsqls.begin();
iter != createsqls.end(); iter++) {
sg_procMetrics->startSql(table, "table", iter->c_str());
if (sg_procOdbc->executeCreateTableSql(iter->c_str(), &dbError)) {
sg_procMetrics->endSql(table, "table", iter->c_str(), false, "", false);
if (!hastable) {
if (iter->find("TABLE ") != std::string::npos) {
sg_createSuccessTables[table] = table;
hastable = true;
}
}
} else {
sg_procMetrics->endSql(table, "table", iter->c_str(), true,
dbError.error.c_str(), false);
if (!hastable) {
if (iter->find("TABLE ") != std::string::npos) {
if (!dbError.alreadyExistError) {
sg_createFailedTables.push_back(table);
hastable = true;
}
break;
}
}
}
}
check_io(sql_file);
mysql_free_result(result);
}
my_snprintf(query_buff, sizeof(query_buff), "show fields from %s",
result_table);
if (mysql_query_with_error_report(mysql, &result, query_buff)) {
if (path) my_fclose(sql_file, MYF(MY_WME));
DBUG_RETURN(0);
}
if (write_data && !complete_insert) {
If data contents of table are to be written and complete_insert
is false (column list not required in INSERT statement), scan the
column list for generated columns, as presence of any generated column
will require that an explicit list of columns is printed.
*/
while ((row = mysql_fetch_row(result))) {
complete_insert |= strcmp(row[SHOW_EXTRA], "STORED GENERATED") == 0 ||
strcmp(row[SHOW_EXTRA], "VIRTUAL GENERATED") == 0;
}
mysql_free_result(result);
if (mysql_query_with_error_report(mysql, &result, query_buff)) {
if (path) my_fclose(sql_file, MYF(MY_WME));
DBUG_RETURN(0);
}
}
If write_data is true, then we build up insert statements for
the table's data. Note: in subsequent lines of code, this test
will have to be performed each time we are appending to
insert_pat.
*/
if (write_data) {
if (opt_replace_into)
dynstr_append_checked(&insert_pat, "REPLACE ");
else
dynstr_append_checked(&insert_pat, "INSERT ");
dynstr_append_checked(&insert_pat, insert_option);
dynstr_append_checked(&insert_pat, "INTO ");
dynstr_append_checked(&insert_pat, opt_quoted_table);
if (complete_insert) {
dynstr_append_checked(&insert_pat, " (");
} else {
dynstr_append_checked(&insert_pat, " VALUES ");
if (!extended_insert) dynstr_append_checked(&insert_pat, "(");
}
}
colno = 0;
while ((row = mysql_fetch_row(result))) {
real_columns[colno] = strcmp(row[SHOW_EXTRA], "STORED GENERATED") != 0 &&
strcmp(row[SHOW_EXTRA], "VIRTUAL GENERATED") != 0;
if (real_columns[colno++] && complete_insert) {
if (init) {
dynstr_append_checked(&insert_pat, ", ");
}
init = 1;
dynstr_append_checked(&insert_pat,
quote_name(row[SHOW_FIELDNAME], name_buff, 0));
}
}
num_fields = mysql_num_rows(result);
mysql_free_result(result);
} else {
verbose_msg("%s: Warning: Can't set SQL_QUOTE_SHOW_CREATE option (%s)\n",
my_progname, mysql_error(mysql));
my_snprintf(query_buff, sizeof(query_buff), show_fields_stmt, db, table);
if (mysql_query_with_error_report(mysql, &result, query_buff))
DBUG_RETURN(0);
if (write_data && !complete_insert) {
If data contents of table are to be written and complete_insert
is false (column list not required in INSERT statement), scan the
column list for generated columns, as presence of any generated column
will require that an explicit list of columns is printed.
*/
while ((row = mysql_fetch_row(result))) {
complete_insert |= strcmp(row[SHOW_EXTRA], "STORED GENERATED") == 0 ||
strcmp(row[SHOW_EXTRA], "VIRTUAL GENERATED") == 0;
}
mysql_free_result(result);
if (mysql_query_with_error_report(mysql, &result, query_buff)) {
if (path) my_fclose(sql_file, MYF(MY_WME));
DBUG_RETURN(0);
}
}
if (!opt_no_create_info) {
my_bool freemem = FALSE;
char const *text;
if (path) {
if (!(sql_file = open_sql_file_for_table(table, O_WRONLY)))
DBUG_RETURN(0);
write_header(sql_file, db);
}
text = fix_identifier_with_newline(result_table, &freemem);
print_comment(sql_file, 0,
"\n--\n-- Table structure for table %s\n--\n\n", text);
if (freemem) my_free((void *)text);
if (opt_drop)
fprintf(sql_file, "DROP TABLE IF EXISTS %s;\n", result_table);
if (!opt_xml)
fprintf(sql_file, "CREATE TABLE %s (\n", result_table);
else
print_xml_tag(sql_file, "\t", "\n", "table_structure", "name=", table,
NullS);
check_io(sql_file);
}
if (write_data) {
if (opt_replace_into)
dynstr_append_checked(&insert_pat, "REPLACE ");
else
dynstr_append_checked(&insert_pat, "INSERT ");
dynstr_append_checked(&insert_pat, insert_option);
dynstr_append_checked(&insert_pat, "INTO ");
dynstr_append_checked(&insert_pat, result_table);
if (complete_insert)
dynstr_append_checked(&insert_pat, " (");
else {
dynstr_append_checked(&insert_pat, " VALUES ");
if (!extended_insert) dynstr_append_checked(&insert_pat, "(");
}
}
colno = 0;
while ((row = mysql_fetch_row(result))) {
ulong *lengths = mysql_fetch_lengths(result);
real_columns[colno] = strcmp(row[SHOW_EXTRA], "STORED GENERATED") != 0 &&
strcmp(row[SHOW_EXTRA], "VIRTUAL GENERATED") != 0;
if (!real_columns[colno++]) continue;
if (init) {
if (!opt_xml && !opt_no_create_info) {
fputs(",\n", sql_file);
check_io(sql_file);
}
if (complete_insert) dynstr_append_checked(&insert_pat, ", ");
}
init = 1;
if (complete_insert)
dynstr_append_checked(&insert_pat,
quote_name(row[SHOW_FIELDNAME], name_buff, 0));
if (!opt_no_create_info) {
if (opt_xml) {
print_xml_row(sql_file, "field", result, &row, NullS);
continue;
}
if (opt_keywords)
fprintf(sql_file, " %s.%s %s", result_table,
quote_name(row[SHOW_FIELDNAME], name_buff, 0),
row[SHOW_TYPE]);
else
fprintf(sql_file, " %s %s",
quote_name(row[SHOW_FIELDNAME], name_buff, 0),
row[SHOW_TYPE]);
if (row[SHOW_DEFAULT]) {
fputs(" DEFAULT ", sql_file);
unescape(sql_file, row[SHOW_DEFAULT], lengths[SHOW_DEFAULT], NULL);
}
if (!row[SHOW_NULL][0]) fputs(" NOT NULL", sql_file);
if (row[SHOW_EXTRA][0]) fprintf(sql_file, " %s", row[SHOW_EXTRA]);
check_io(sql_file);
}
}
num_fields = mysql_num_rows(result);
mysql_free_result(result);
if (!opt_no_create_info) {
char buff[20 + FN_REFLEN];
uint keynr, primary_key;
my_snprintf(buff, sizeof(buff), "show keys from %s", result_table);
if (mysql_query_with_error_report(mysql, &result, buff)) {
if (mysql_errno(mysql) == ER_WRONG_OBJECT) {
fputs("\t\t<options Comment=\"view\" />\n", sql_file);
goto continue_xml;
}
fprintf(stderr, "%s: Can't get keys for table %s (%s)\n", my_progname,
result_table, mysql_error(mysql));
if (path) my_fclose(sql_file, MYF(MY_WME));
DBUG_RETURN(0);
}
keynr = 0;
primary_key = INT_MAX;
while ((row = mysql_fetch_row(result))) {
if (atoi(row[3]) == 1) {
keynr++;
if (!strcmp(row[2], "PRIMARY")) {
primary_key = keynr;
break;
}
}
}
mysql_data_seek(result, 0);
keynr = 0;
while ((row = mysql_fetch_row(result))) {
if (opt_xml) {
print_xml_row(sql_file, "key", result, &row, NullS);
continue;
}
if (atoi(row[3]) == 1) {
if (keynr++) putc(')', sql_file);
if (atoi(row[1]))
fprintf(sql_file, ",\n KEY %s (",
quote_name(row[2], name_buff, 0));
else if (keynr == primary_key)
fputs(",\n PRIMARY KEY (", sql_file);
else
fprintf(sql_file, ",\n UNIQUE %s (",
quote_name(row[2], name_buff, 0));
} else
putc(',', sql_file);
fputs(quote_name(row[4], name_buff, 0), sql_file);
if (row[7]) fprintf(sql_file, " (%s)", row[7]);
check_io(sql_file);
}
mysql_free_result(result);
if (!opt_xml) {
if (keynr) putc(')', sql_file);
fputs("\n)", sql_file);
check_io(sql_file);
}
if (create_options) {
char show_name_buff[NAME_LEN * 2 + 2 + 24];
my_snprintf(buff, sizeof(buff), "show table status like %s",
quote_for_like(table, show_name_buff));
if (mysql_query_with_error_report(mysql, &result, buff)) {
if (mysql_errno(mysql) != ER_PARSE_ERROR) {
verbose_msg(
"-- Warning: Couldn't get status information for "
"table %s (%s)\n",
result_table, mysql_error(mysql));
}
} else if (!(row = mysql_fetch_row(result))) {
fprintf(stderr,
"Error: Couldn't read status information for table %s (%s)\n",
result_table, mysql_error(mysql));
} else {
if (opt_xml)
print_xml_row(sql_file, "options", result, &row, NullS);
else {
fputs("/*!", sql_file);
print_value(sql_file, result, row, "engine=", "Engine", 0);
print_value(sql_file, result, row, "", "Create_options", 0);
print_value(sql_file, result, row, "comment=", "Comment", 1);
fputs(" */", sql_file);
check_io(sql_file);
}
}
mysql_free_result(result);
}
continue_xml:
if (!opt_xml)
fputs(";\n", sql_file);
else
fputs("\t</table_structure>\n", sql_file);
check_io(sql_file);
}
}
if (complete_insert) {
dynstr_append_checked(&insert_pat, ") VALUES ");
if (!extended_insert) dynstr_append_checked(&insert_pat, "(");
}
if (sql_file != md_result_file) {
fputs("\n", sql_file);
write_footer(sql_file);
my_fclose(sql_file, MYF(MY_WME));
}
DBUG_RETURN((uint)num_fields);
}
void MySQLContext::dump_trigger_old(FILE *sql_file, MYSQL_RES *show_triggers_rs,
MYSQL_ROW *show_trigger_row,
const char *table_name) {
char quoted_table_name_buf[NAME_LEN * 2 + 3];
char *quoted_table_name = quote_name(table_name, quoted_table_name_buf, 1);
char name_buff[NAME_LEN * 4 + 3];
const char *xml_msg =
"\nWarning! mysqldump being run against old server "
"that does not\nsupport 'SHOW CREATE TRIGGERS' "
"statement. Skipping..\n";
DBUG_ENTER("dump_trigger_old");
if (opt_xml) {
print_xml_comment(sql_file, strlen(xml_msg), xml_msg);
check_io(sql_file);
DBUG_VOID_RETURN;
}
fprintf(sql_file,
"--\n"
"-- WARNING: old server version. "
"The following dump may be incomplete.\n"
"--\n");
if (opt_compact)
fprintf(sql_file, "/*!50003 SET @OLD_SQL_MODE=@@SQL_MODE*/;\n");
if (opt_drop_trigger)
fprintf(sql_file, "/*!50032 DROP TRIGGER IF EXISTS %s */;\n",
(*show_trigger_row)[0]);
fprintf(sql_file,
"DELIMITER ;;\n"
"/*!50003 SET SESSION SQL_MODE=\"%s\" */;;\n"
"/*!50003 CREATE */ ",
(*show_trigger_row)[6]);
if (mysql_num_fields(show_triggers_rs) > 7) {
mysqldump can be run against the server, that does not support
definer in triggers (there is no DEFINER column in SHOW TRIGGERS
output). So, we should check if we have this column before
accessing it.
*/
size_t user_name_len;
char user_name_str[USERNAME_LENGTH + 1];
char quoted_user_name_str[USERNAME_LENGTH * 2 + 3];
size_t host_name_len;
char host_name_str[HOSTNAME_LENGTH + 1];
char quoted_host_name_str[HOSTNAME_LENGTH * 2 + 3];
parse_user((*show_trigger_row)[7], strlen((*show_trigger_row)[7]),
user_name_str, &user_name_len, host_name_str, &host_name_len);
fprintf(sql_file, "/*!50017 DEFINER=%s@%s */ ",
quote_name(user_name_str, quoted_user_name_str, FALSE),
quote_name(host_name_str, quoted_host_name_str, FALSE));
}
fprintf(sql_file,
"/*!50003 TRIGGER %s %s %s ON %s FOR EACH ROW%s%s */;;\n"
"DELIMITER ;\n",
quote_name((*show_trigger_row)[0], name_buff, 0),
(*show_trigger_row)[4],
(*show_trigger_row)[1],
quoted_table_name,
(strchr(" \t\n\r", *((*show_trigger_row)[3]))) ? "" : " ",
(*show_trigger_row)[3] );
if (opt_compact)
fprintf(sql_file, "/*!50003 SET SESSION SQL_MODE=@OLD_SQL_MODE */;\n");
DBUG_VOID_RETURN;
}
int MySQLContext::dump_trigger(FILE *sql_file,
MYSQL_RES *show_create_trigger_rs,
const char *db_name, const char *db_cl_name) {
MYSQL_ROW row;
char *query_str;
int db_cl_altered = FALSE;
DBUG_ENTER("dump_trigger");
while ((row = mysql_fetch_row(show_create_trigger_rs))) {
if (opt_xml) {
print_xml_row(sql_file, "trigger", show_create_trigger_rs, &row,
"SQL Original Statement");
check_io(sql_file);
continue;
}
query_str = cover_definer_clause(
row[2], strlen(row[2]), C_STRING_WITH_LEN("50017"),
C_STRING_WITH_LEN("50003"), C_STRING_WITH_LEN(" TRIGGER"));
if (switch_db_collation(sql_file, db_name, ";", db_cl_name, row[5],
&db_cl_altered))
DBUG_RETURN(TRUE);
switch_cs_variables(sql_file, ";", row[3],
row[3],
row[4]);
remove_sql_mode(row[1], C_STRING_WITH_LEN("NO_AUTO_CREATE_USER"));
switch_sql_mode(sql_file, ";", row[1]);
if (opt_drop_trigger)
fprintf(sql_file, "/*!50032 DROP TRIGGER IF EXISTS %s */;\n", row[0]);
fprintf(sql_file,
"DELIMITER ;;\n"
"/*!50003 %s */;;\n"
"DELIMITER ;\n",
(const char *)(query_str != NULL ? query_str : row[2]));
restore_sql_mode(sql_file, ";");
restore_cs_variables(sql_file, ";");
if (db_cl_altered) {
if (restore_db_collation(sql_file, db_name, ";", db_cl_name))
DBUG_RETURN(TRUE);
}
my_free(query_str);
}
DBUG_RETURN(FALSE);
}
Dump the triggers for a given table.
This should be called after the tables have been dumped in case a trigger
depends on the existence of a table.
@param[in] table_name
@param[in] db_name
@return Error status.
@retval TRUE error has occurred.
@retval FALSE operation succeed.
*/
int MySQLContext::dump_triggers_for_table(char *table_name, char *db_name) {
char name_buff[NAME_LEN * 4 + 3];
char query_buff[QUERY_LENGTH];
uint old_opt_compatible_mode = opt_compatible_mode;
MYSQL_RES *show_triggers_rs;
MYSQL_ROW row;
FILE *sql_file = md_result_file;
char db_cl_name[MY_CS_NAME_SIZE];
int ret = TRUE;
DBUG_ENTER("dump_triggers_for_table");
DBUG_PRINT("enter", ("db: %s, table_name: %s", db_name, table_name));
if (path &&
!(sql_file = open_sql_file_for_table(table_name, O_WRONLY | O_APPEND)))
DBUG_RETURN(1);
opt_compatible_mode &= ~MASK_ANSI_QUOTES;
if (switch_character_set_results(mysql, "binary")) goto done;
if (fetch_db_collation(db_name, db_cl_name, sizeof(db_cl_name))) goto done;
my_snprintf(query_buff, sizeof(query_buff), "SHOW TRIGGERS LIKE %s",
quote_for_like(table_name, name_buff));
if (mysql_query_with_error_report(mysql, &show_triggers_rs, query_buff))
goto done;
if (!mysql_num_rows(show_triggers_rs)) goto skip;
if (opt_xml)
print_xml_tag(sql_file, "\t", "\n", "triggers", "name=", table_name, NullS);
while ((row = mysql_fetch_row(show_triggers_rs))) {
my_snprintf(query_buff, sizeof(query_buff), "SHOW CREATE TRIGGER %s",
quote_name(row[0], name_buff, TRUE));
if (mysql_query(mysql, query_buff)) {
mysqldump is being run against old server, that does not support
SHOW CREATE TRIGGER statement. We should use SHOW TRIGGERS output.
NOTE: the dump may be incorrect, as old SHOW TRIGGERS does not
provide all the necessary information to restore trigger properly.
*/
dump_trigger_old(sql_file, show_triggers_rs, &row, table_name);
} else {
MYSQL_RES *show_create_trigger_rs = mysql_store_result(mysql);
if (!show_create_trigger_rs ||
dump_trigger(sql_file, show_create_trigger_rs, db_name, db_cl_name))
goto done;
mysql_free_result(show_create_trigger_rs);
}
}
if (opt_xml) {
fputs("\t</triggers>\n", sql_file);
check_io(sql_file);
}
skip:
mysql_free_result(show_triggers_rs);
if (switch_character_set_results(mysql, default_charset)) goto done;
make sure to set back opt_compatible mode to
original value
*/
opt_compatible_mode = old_opt_compatible_mode;
ret = FALSE;
done:
if (path) my_fclose(sql_file, MYF(0));
DBUG_RETURN(ret);
}
static void add_load_option(DYNAMIC_STRING *str, const char *option,
const char *option_value) {
if (!option_value) {
return;
}
dynstr_append_checked(str, option);
if (strncmp(option_value, "0x", sizeof("0x") - 1) == 0) {
dynstr_append_checked(str, option_value);
} else {
field_escape(str, option_value);
}
}
Allow the user to specify field terminator strings like:
"'", "\", "\\" (escaped backslash), "\t" (tab), "\n" (newline)
This is done by doubling ' and add a end -\ if needed to avoid
syntax errors from the SQL parser.
*/
static void field_escape(DYNAMIC_STRING *in, const char *from) {
uint end_backslashes = 0;
dynstr_append_checked(in, "'");
while (*from) {
dynstr_append_mem_checked(in, from, 1);
if (*from == '\\')
end_backslashes ^= 1;
else {
if (*from == '\'' && !end_backslashes) {
dynstr_append_checked(in, "\'");
}
end_backslashes = 0;
}
from++;
}
if (end_backslashes) dynstr_append_checked(in, "\\");
dynstr_append_checked(in, "'");
}
static char *alloc_query_str(size_t size) {
char *query;
if (!(query = (char *)my_malloc(PSI_NOT_INSTRUMENTED, size, MYF(MY_WME))))
die(EX_MYSQLERR, "Couldn't allocate a query string.");
return query;
}
SYNOPSIS
dump_table()
dump_table saves database contents as a series of INSERT statements.
ARGS
table - table name
db - db name
RETURNS
void
*/
void MySQLContext::dump_table(char *table, char *db) {
char ignore_flag;
char buf[200], table_buff[NAME_LEN + 3];
DYNAMIC_STRING query_string;
DYNAMIC_STRING extended_row;
DYNAMIC_STRING insert_row;
char table_type[NAME_LEN];
char *result_table, table_buff2[NAME_LEN * 2 + 3];
int error = 0;
ulong rownr, row_break;
size_t total_length, init_length;
uint num_fields;
MYSQL_RES *res;
MYSQL_FIELD *field;
MYSQL_ROW row;
struct DMODBC::DBError dbError;
my_bool real_columns[MAX_FIELDS];
DBUG_ENTER("dump_table");
Make sure you get the create table info before the following check for
--no-data flag below. Otherwise, the create table info won't be printed.
*/
num_fields =
get_table_structure(table, db, table_type, &ignore_flag, real_columns);
The "table" could be a view. If so, we don't do anything here.
*/
if (strcmp(table_type, "VIEW") == 0) DBUG_VOID_RETURN;
We don't dump data fo`r replication metadata tables.
*/
if (replication_metadata_tables(db, table)) DBUG_VOID_RETURN;
if (opt_no_data) {
verbose_msg("-- Skipping dump data for table '%s', --no-data was used\n",
table);
DBUG_VOID_RETURN;
}
DBUG_PRINT("info",
("ignore_flag: %x num_fields: %d", (int)ignore_flag, num_fields));
If the table type is a merge table or any type that has to be
_completely_ ignored and no data dumped
*/
if (ignore_flag & IGNORE_DATA) {
verbose_msg(
"-- Warning: Skipping data for table '%s' because "
"it's of type %s\n",
table, table_type);
DBUG_VOID_RETURN;
}
if (num_fields == 0) {
verbose_msg("-- Skipping dump data for table '%s', it has no fields\n",
table);
DBUG_VOID_RETURN;
}
result_table = quote_name(table, table_buff, 1);
quote_name(table, table_buff2, 0);
verbose_msg("-- Sending SELECT query...\n");
init_dynamic_string_checked(&query_string, "", 1024, 1024);
if (extended_insert)
init_dynamic_string_checked(&extended_row, "", 1024, 1024);
init_dynamic_string_checked(&insert_row, "", 1024, 1024);
if (path) {
char filename[FN_REFLEN], tmp_path[FN_REFLEN];
Convert the path to native os format
and resolve to the full filepath.
*/
convert_dirname(tmp_path, path, NullS);
my_load_path(tmp_path, tmp_path, NULL);
fn_format(filename, table, tmp_path, ".txt", MYF(MY_UNPACK_FILENAME));
my_delete(filename, MYF(0));
to_unix_path(filename);
dynstr_append_checked(&query_string,
"SELECT /*!40001 SQL_NO_CACHE */ * INTO OUTFILE '");
dynstr_append_checked(&query_string, filename);
dynstr_append_checked(&query_string, "'");
dynstr_append_checked(&query_string, " /*!50138 CHARACTER SET ");
dynstr_append_checked(&query_string,
default_charset == mysql_universal_client_charset
? my_charset_bin.name
:
default_charset);
dynstr_append_checked(&query_string, " */");
if (fields_terminated || enclosed || opt_enclosed || escaped)
dynstr_append_checked(&query_string, " FIELDS");
add_load_option(&query_string, " TERMINATED BY ", fields_terminated);
add_load_option(&query_string, " ENCLOSED BY ", enclosed);
add_load_option(&query_string, " OPTIONALLY ENCLOSED BY ", opt_enclosed);
add_load_option(&query_string, " ESCAPED BY ", escaped);
add_load_option(&query_string, " LINES TERMINATED BY ", lines_terminated);
dynstr_append_checked(&query_string, " FROM ");
dynstr_append_checked(&query_string, result_table);
if (where) {
dynstr_append_checked(&query_string, " WHERE ");
dynstr_append_checked(&query_string, where);
} else if ((!my_strcasecmp(charset_info, db, "mysql")) &&
(!my_strcasecmp(charset_info, table, "proc")) && opt_alldbs) {
dynstr_append_checked(&query_string, " WHERE db != 'sys'");
}
if (order_by) {
dynstr_append_checked(&query_string, " ORDER BY ");
dynstr_append_checked(&query_string, order_by);
}
if (mysql_real_query(mysql, query_string.str, (ulong)query_string.length)) {
DB_error(mysql, "when executing 'SELECT INTO OUTFILE'");
dynstr_free(&query_string);
DBUG_VOID_RETURN;
}
} else {
my_bool freemem = FALSE;
char const *text = fix_identifier_with_newline(result_table, &freemem);
print_comment(md_result_file, 0, "\n--\n-- Dumping data for table %s\n--\n",
text);
if (freemem) my_free((void *)text);
dynstr_append_checked(&query_string,
"SELECT /*!40001 SQL_NO_CACHE */ * FROM ");
dynstr_append_checked(&query_string, result_table);
if (where) {
freemem = FALSE;
text = fix_identifier_with_newline(where, &freemem);
print_comment(md_result_file, 0, "-- WHERE: %s\n", text);
if (freemem) my_free((void *)text);
dynstr_append_checked(&query_string, " WHERE ");
dynstr_append_checked(&query_string, where);
}
If table is mysql.proc then do not dump routines which belong
to sys schema
*/
else if ((!my_strcasecmp(charset_info, db, "mysql")) &&
(!my_strcasecmp(charset_info, table, "proc")) && opt_alldbs) {
dynstr_append_checked(&query_string, " WHERE db != 'sys'");
}
if (order_by) {
freemem = FALSE;
text = fix_identifier_with_newline(order_by, &freemem);
print_comment(md_result_file, 0, "-- ORDER BY: %s\n", text);
if (freemem) my_free((void *)text);
dynstr_append_checked(&query_string, " ORDER BY ");
dynstr_append_checked(&query_string, order_by);
}
if (!opt_xml && !opt_compact) {
fputs("\n", md_result_file);
check_io(md_result_file);
}
if (mysql_query_with_error_report(mysql, 0, query_string.str)) {
DB_error(mysql, "when retrieving data from server");
goto err;
}
if (quick)
res = mysql_use_result(mysql);
else
res = mysql_store_result(mysql);
if (!res) {
DB_error(mysql, "when retrieving data from server");
goto err;
}
verbose_msg("-- Retrieving rows...\n");
if (mysql_num_fields(res) != num_fields) {
fprintf(stderr, "%s: Error in field count for table: %s ! Aborting.\n",
my_progname, result_table);
error = EX_CONSCHECK;
goto err;
}
total_length = opt_net_buffer_length;
row_break = 0;
rownr = 0;
init_length = (uint)insert_pat.length + 4;
if (opt_xml)
print_xml_tag(md_result_file, "\t", "\n", "table_data", "name=", table,
NullS);
while ((row = mysql_fetch_row(res))) {
uint i;
ulong *lengths = mysql_fetch_lengths(res);
rownr++;
mysql_field_seek(res, 0);
if (opt_xml) {
fputs("\t<row>\n", md_result_file);
check_io(md_result_file);
}
for (i = 0; i < mysql_num_fields(res); i++) {
int is_blob;
ulong length = lengths[i];
if (!(field = mysql_fetch_field(res)))
die(EX_CONSCHECK, "Not enough fields from table %s! Aborting.\n",
result_table);
if (!real_columns[i]) continue;
63 is my_charset_bin. If charsetnr is not 63,
we have not a BLOB but a TEXT column.
*/
is_blob =
(field->charsetnr == 63 && (field->type == MYSQL_TYPE_BIT ||
field->type == MYSQL_TYPE_STRING ||
field->type == MYSQL_TYPE_VAR_STRING ||
field->type == MYSQL_TYPE_VARCHAR ||
field->type == MYSQL_TYPE_BLOB ||
field->type == MYSQL_TYPE_LONG_BLOB ||
field->type == MYSQL_TYPE_MEDIUM_BLOB ||
field->type == MYSQL_TYPE_TINY_BLOB ||
field->type == MYSQL_TYPE_GEOMETRY))
? 1
: 0;
if (extended_insert && !opt_xml) {
if (i == 0)
dynstr_set_checked(&extended_row, "(");
else
dynstr_append_checked(&extended_row, ",");
if (row[i]) {
if (length) {
if (!(field->flags & NUM_FLAG)) {
"length * 2 + 4 + 1" is OK for both HEX non-HEX modes
*/
dynstr_realloc_checked(&extended_row, length * 2 + 4 + 1);
if (opt_hex_blob && is_blob) {
dynstr_append_checked(&extended_row, "0x");
extended_row.length += mysql_hex_string(
extended_row.str + extended_row.length, row[i], length);
DBUG_ASSERT(extended_row.length + 1 <=
extended_row.max_length);
DBUG_ASSERT(extended_row.str[extended_row.length] == '\0');
} else {
if (is_blob) {
dynstr_append_checked(&extended_row, "'\\x");
extended_row.length += mysql_hex_string(
extended_row.str + extended_row.length, row[i], length);
dynstr_append_checked(&extended_row, "'");
} else {
dynstr_append_checked(&extended_row, "'");
extended_row.length += mysql_real_escape_string_quote(
&mysql_connection,
&extended_row.str[extended_row.length], row[i], length,
'\'');
extended_row.str[extended_row.length] = '\0';
dynstr_append_checked(&extended_row, "'");
}
}
} else {
char *ptr = row[i];
if (my_isalpha(charset_info, *ptr) ||
(*ptr == '-' && my_isalpha(charset_info, ptr[1])))
dynstr_append_checked(&extended_row, "NULL");
else {
if (field->type == MYSQL_TYPE_DECIMAL) {
dynstr_append_checked(&extended_row, "'");
dynstr_append_checked(&extended_row, ptr);
dynstr_append_checked(&extended_row, "'");
} else
dynstr_append_checked(&extended_row, ptr);
}
}
} else
dynstr_append_checked(&extended_row, "''");
} else
dynstr_append_checked(&extended_row, "NULL");
} else {
if (i && !opt_xml) {
dynstr_append_checked(&insert_row, ",");
}
if (row[i]) {
if (!(field->flags & NUM_FLAG)) {
if (opt_xml) {
if (opt_hex_blob && is_blob && length) {
print_xml_tag(md_result_file, "\t\t", "", "field",
"name=", field->name,
"xsi:type=", "xs:hexBinary", NullS);
print_blob_as_hex(md_result_file, row[i], length);
} else {
print_xml_tag(md_result_file, "\t\t", "", "field",
"name=", field->name, NullS);
print_quoted_xml(md_result_file, row[i], length, 0);
}
fputs("</field>\n", md_result_file);
} else if (opt_hex_blob && is_blob && length) {
dynstr_append_checked(&insert_row, "0x");
append_blob_as_hex(&insert_row, row[i], length);
} else {
if (is_blob) {
}
unescapestr(row[i], length, &insert_row);
}
} else {
char *ptr = row[i];
if (opt_xml) {
print_xml_tag(md_result_file, "\t\t", "", "field",
"name=", field->name, NullS);
fputs(!my_isalpha(charset_info, *ptr) ? ptr : "NULL",
md_result_file);
fputs("</field>\n", md_result_file);
} else if (my_isalpha(charset_info, *ptr) ||
(*ptr == '-' && my_isalpha(charset_info, ptr[1]))) {
dynstr_append_checked(&insert_row, "NULL ");
} else if (field->type == MYSQL_TYPE_DECIMAL) {
dynstr_append_checked(&insert_row, "\'");
dynstr_append_checked(&insert_row, ptr);
dynstr_append_checked(&insert_row, "\'");
} else {
dynstr_append_checked(&insert_row, ptr);
}
}
} else {
if (!opt_xml) {
dynstr_append_checked(&insert_row, "NULL");
}
}
}
}
if (opt_xml) {
fputs("\t</row>\n", md_result_file);
check_io(md_result_file);
}
if (extended_insert) {
size_t row_length;
dynstr_append_checked(&extended_row, ")");
row_length = 2 + extended_row.length;
if (total_length + row_length < opt_net_buffer_length) {
total_length += row_length;
dynstr_append_checked(&insert_row, ",");
dynstr_append_checked(&insert_row, extended_row.str);
} else {
if (row_break) {
dynstr_append_checked(&insert_row, ";\n");
if (sg_createSuccessTables.find(table) !=
sg_createSuccessTables.end()) {
sg_procMetrics->startSql(table, "table", insert_row.str);
if (sg_procOdbc->executeInsertSql(insert_row.str, &dbError)) {
sg_procMetrics->endSql(table, "table", insert_row.str, false,
"", true);
} else {
sg_procMetrics->endSql(table, "table", insert_row.str, true,
dbError.error.c_str(), true);
}
}
dynstr_set_checked(&insert_row, "");
}
row_break = 1;
dynstr_append_checked(&insert_row, insert_pat.str);
dynstr_append_checked(&insert_row, extended_row.str);
total_length = row_length + init_length;
}
} else if (!opt_xml) {
dynstr_append_checked(&insert_row, ");\n");
}
}
if (opt_xml)
fputs("\t</table_data>\n", md_result_file);
else if (extended_insert && row_break) {
dynstr_append_checked(&insert_row, ";\n");
}
if (strlen(insert_row.str) > 0) {
if (sg_createSuccessTables.find(table) != sg_createSuccessTables.end()) {
sg_procMetrics->startSql(table, "table", insert_row.str);
if (sg_procOdbc->executeInsertSql(insert_row.str, &dbError)) {
sg_procMetrics->endSql(table, "table", insert_row.str, false, "", true);
} else {
sg_procMetrics->endSql(table, "table", insert_row.str, true,
dbError.error.c_str(), true);
}
}
}
if (mysql_errno(mysql)) {
my_snprintf(buf, sizeof(buf),
"%s: Error %d: %s when dumping table %s at row: %ld\n",
my_progname, mysql_errno(mysql), mysql_error(mysql),
result_table, rownr);
fputs(buf, stderr);
error = EX_CONSCHECK;
goto err;
}
if (opt_disable_keys) {
}
if (opt_lock) {
}
if (opt_autocommit) {
}
mysql_free_result(res);
}
dynstr_free(&query_string);
if (extended_insert) dynstr_free(&extended_row);
dynstr_free(&insert_row);
DBUG_VOID_RETURN;
err:
dynstr_free(&query_string);
if (extended_insert) dynstr_free(&extended_row);
dynstr_free(&insert_row);
maybe_exit(error);
DBUG_VOID_RETURN;
}
char *MySQLContext::getTableName(int reset) {
static MYSQL_RES *res = NULL;
MYSQL_ROW row;
if (!res) {
if (!(res = mysql_list_tables(mysql, NullS))) return (NULL);
}
if ((row = mysql_fetch_row(res))) return ((char *)row[0]);
if (reset)
mysql_data_seek(res, 0);
else {
mysql_free_result(res);
res = NULL;
}
return (NULL);
}
dump all logfile groups and tablespaces
*/
int MySQLContext::dump_all_tablespaces() { return dump_tablespaces(NULL); }
int MySQLContext::dump_tablespaces_for_tables(char *db, char **table_names,
int tables) {
DYNAMIC_STRING where;
int r;
int i;
char name_buff[NAME_LEN * 2 + 3];
mysql_real_escape_string_quote(mysql, name_buff, db, (ulong)strlen(db), '\'');
init_dynamic_string_checked(&where,
" AND TABLESPACE_NAME IN ("
"SELECT DISTINCT TABLESPACE_NAME FROM"
" INFORMATION_SCHEMA.PARTITIONS"
" WHERE"
" TABLE_SCHEMA='",
256, 1024);
dynstr_append_checked(&where, name_buff);
dynstr_append_checked(&where, "' AND TABLE_NAME IN (");
for (i = 0; i < tables; i++) {
mysql_real_escape_string_quote(mysql, name_buff, table_names[i],
(ulong)strlen(table_names[i]), '\'');
dynstr_append_checked(&where, "'");
dynstr_append_checked(&where, name_buff);
dynstr_append_checked(&where, "',");
}
dynstr_trunc(&where, 1);
dynstr_append_checked(&where, "))");
DBUG_PRINT("info", ("Dump TS for Tables where: %s", where.str));
r = dump_tablespaces(where.str);
dynstr_free(&where);
return r;
}
int MySQLContext::dump_tablespaces_for_databases(char **databases) {
DYNAMIC_STRING where;
int r;
int i;
init_dynamic_string_checked(&where,
" AND TABLESPACE_NAME IN ("
"SELECT DISTINCT TABLESPACE_NAME FROM"
" INFORMATION_SCHEMA.PARTITIONS"
" WHERE"
" TABLE_SCHEMA IN (",
256, 1024);
for (i = 0; databases[i] != NULL; i++) {
char db_name_buff[NAME_LEN * 2 + 3];
mysql_real_escape_string_quote(mysql, db_name_buff, databases[i],
(ulong)strlen(databases[i]), '\'');
dynstr_append_checked(&where, "'");
dynstr_append_checked(&where, db_name_buff);
dynstr_append_checked(&where, "',");
}
dynstr_trunc(&where, 1);
dynstr_append_checked(&where, "))");
DBUG_PRINT("info", ("Dump TS for DBs where: %s", where.str));
r = dump_tablespaces(where.str);
dynstr_free(&where);
return r;
}
int MySQLContext::dump_tablespaces(char *ts_where) {
MYSQL_ROW row;
MYSQL_RES *tableres;
char buf[FN_REFLEN];
DYNAMIC_STRING sqlbuf;
int first = 0;
The following are used for parsing the EXTRA field
*/
char extra_format[] = "UNDO_BUFFER_SIZE=";
char *ubs;
char *endsemi;
DBUG_ENTER("dump_tablespaces");
init_dynamic_string_checked(&sqlbuf,
"SELECT LOGFILE_GROUP_NAME,"
" FILE_NAME,"
" TOTAL_EXTENTS,"
" INITIAL_SIZE,"
" ENGINE,"
" EXTRA"
" FROM INFORMATION_SCHEMA.FILES"
" WHERE FILE_TYPE = 'UNDO LOG'"
" AND FILE_NAME IS NOT NULL"
" AND LOGFILE_GROUP_NAME IS NOT NULL",
256, 1024);
if (ts_where) {
dynstr_append_checked(&sqlbuf,
" AND LOGFILE_GROUP_NAME IN ("
"SELECT DISTINCT LOGFILE_GROUP_NAME"
" FROM INFORMATION_SCHEMA.FILES"
" WHERE FILE_TYPE = 'DATAFILE'");
dynstr_append_checked(&sqlbuf, ts_where);
dynstr_append_checked(&sqlbuf, ")");
}
dynstr_append_checked(&sqlbuf,
" GROUP BY LOGFILE_GROUP_NAME, FILE_NAME"
", ENGINE, TOTAL_EXTENTS, INITIAL_SIZE"
" ORDER BY LOGFILE_GROUP_NAME");
if (mysql_query(mysql, sqlbuf.str) ||
!(tableres = mysql_store_result(mysql))) {
dynstr_free(&sqlbuf);
if (mysql_errno(mysql) == ER_BAD_TABLE_ERROR ||
mysql_errno(mysql) == ER_BAD_DB_ERROR ||
mysql_errno(mysql) == ER_UNKNOWN_TABLE) {
fprintf(md_result_file,
"\n--\n-- Not dumping tablespaces as no INFORMATION_SCHEMA.FILES"
" table on this server\n--\n");
check_io(md_result_file);
DBUG_RETURN(0);
}
my_printf_error(0, "Error: '%s' when trying to dump tablespaces", MYF(0),
mysql_error(mysql));
DBUG_RETURN(1);
}
buf[0] = 0;
while ((row = mysql_fetch_row(tableres))) {
if (strcmp(buf, row[0]) != 0) first = 1;
if (first) {
print_comment(md_result_file, 0, "\n--\n-- Logfile group: %s\n--\n",
row[0]);
fprintf(md_result_file, "\nCREATE");
} else {
fprintf(md_result_file, "\nALTER");
}
fprintf(md_result_file,
" LOGFILE GROUP %s\n"
" ADD UNDOFILE '%s'\n",
row[0], row[1]);
if (first) {
ubs = strstr(row[5], extra_format);
if (!ubs) break;
ubs += strlen(extra_format);
endsemi = strstr(ubs, ";");
if (endsemi) endsemi[0] = '\0';
fprintf(md_result_file, " UNDO_BUFFER_SIZE %s\n", ubs);
}
fprintf(md_result_file,
" INITIAL_SIZE %s\n"
" ENGINE=%s;\n",
row[3], row[4]);
check_io(md_result_file);
if (first) {
first = 0;
strxmov(buf, row[0], NullS);
}
}
dynstr_free(&sqlbuf);
mysql_free_result(tableres);
init_dynamic_string_checked(&sqlbuf,
"SELECT DISTINCT TABLESPACE_NAME,"
" FILE_NAME,"
" LOGFILE_GROUP_NAME,"
" EXTENT_SIZE,"
" INITIAL_SIZE,"
" ENGINE"
" FROM INFORMATION_SCHEMA.FILES"
" WHERE FILE_TYPE = 'DATAFILE'",
256, 1024);
if (ts_where) dynstr_append_checked(&sqlbuf, ts_where);
dynstr_append_checked(&sqlbuf,
" ORDER BY TABLESPACE_NAME, LOGFILE_GROUP_NAME");
if (mysql_query_with_error_report(mysql, &tableres, sqlbuf.str)) {
dynstr_free(&sqlbuf);
DBUG_RETURN(1);
}
buf[0] = 0;
while ((row = mysql_fetch_row(tableres))) {
if (strcmp(buf, row[0]) != 0) first = 1;
if (first) {
print_comment(md_result_file, 0, "\n--\n-- Tablespace: %s\n--\n", row[0]);
fprintf(md_result_file, "\nCREATE");
} else {
fprintf(md_result_file, "\nALTER");
}
fprintf(md_result_file,
" TABLESPACE %s\n"
" ADD DATAFILE '%s'\n",
row[0], row[1]);
if (first) {
fprintf(md_result_file,
" USE LOGFILE GROUP %s\n"
" EXTENT_SIZE %s\n",
row[2], row[3]);
}
fprintf(md_result_file,
" INITIAL_SIZE %s\n"
" ENGINE=%s;\n",
row[4], row[5]);
check_io(md_result_file);
if (first) {
first = 0;
strxmov(buf, row[0], NullS);
}
}
mysql_free_result(tableres);
dynstr_free(&sqlbuf);
DBUG_RETURN(0);
}
int MySQLContext::is_ndbinfo(MYSQL *mysql, const char *dbname) {
static int checked_ndbinfo = 0;
static int have_ndbinfo = 0;
if (!checked_ndbinfo) {
MYSQL_RES *res;
MYSQL_ROW row;
char buf[32], query[64];
my_snprintf(query, sizeof(query), "SHOW VARIABLES LIKE %s",
quote_for_like("ndbinfo_version", buf));
checked_ndbinfo = 1;
if (mysql_query_with_error_report(mysql, &res, query)) return 0;
if (!(row = mysql_fetch_row(res))) {
mysql_free_result(res);
return 0;
}
have_ndbinfo = 1;
mysql_free_result(res);
}
if (!have_ndbinfo) return 0;
if (my_strcasecmp(&my_charset_latin1, dbname, "ndbinfo") == 0) return 1;
return 0;
}
int MySQLContext::dump_all_databases() {
MYSQL_ROW row;
MYSQL_RES *tableres;
int result = 0;
if (mysql_query_with_error_report(mysql, &tableres, "SHOW DATABASES"))
return 1;
while ((row = mysql_fetch_row(tableres))) {
if (mysql_get_server_version(mysql) >= FIRST_INFORMATION_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, row[0], INFORMATION_SCHEMA_DB_NAME))
continue;
if (mysql_get_server_version(mysql) >= FIRST_PERFORMANCE_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, row[0], PERFORMANCE_SCHEMA_DB_NAME))
continue;
if (mysql_get_server_version(mysql) >= FIRST_SYS_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, row[0], SYS_SCHEMA_DB_NAME))
continue;
if (is_ndbinfo(mysql, row[0])) continue;
if (dump_all_tables_in_db(row[0])) result = 1;
}
mysql_free_result(tableres);
if (seen_views) {
if (mysql_query(mysql, "SHOW DATABASES") ||
!(tableres = mysql_store_result(mysql))) {
my_printf_error(0, "Error: Couldn't execute 'SHOW DATABASES': %s", MYF(0),
mysql_error(mysql));
return 1;
}
while ((row = mysql_fetch_row(tableres))) {
if (mysql_get_server_version(mysql) >= FIRST_INFORMATION_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, row[0],
INFORMATION_SCHEMA_DB_NAME))
continue;
if (mysql_get_server_version(mysql) >= FIRST_PERFORMANCE_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, row[0],
PERFORMANCE_SCHEMA_DB_NAME))
continue;
if (mysql_get_server_version(mysql) >= FIRST_SYS_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, row[0], SYS_SCHEMA_DB_NAME))
continue;
if (is_ndbinfo(mysql, row[0])) continue;
if (dump_all_views_in_db(row[0])) result = 1;
}
mysql_free_result(tableres);
}
return result;
}
int MySQLContext::dump_databases(char **db_names) {
int result = 0;
char **db;
DBUG_ENTER("dump_databases");
for (db = db_names; *db; db++) {
if (dump_all_tables_in_db(*db)) result = 1;
}
if (!result && seen_views) {
for (db = db_names; *db; db++) {
if (dump_all_views_in_db(*db)) result = 1;
}
}
DBUG_RETURN(result);
}
View Specific database initalization.
SYNOPSIS
init_dumping_views
qdatabase quoted name of the database
RETURN VALUES
0 Success.
1 Failure.
*/
int MySQLContext::init_dumping_views(char *qdatabase MY_ATTRIBUTE((unused))) {
return 0;
}
Table Specific database initalization.
SYNOPSIS
init_dumping_tables
qdatabase quoted name of the database
RETURN VALUES
0 Success.
1 Failure.
*/
int MySQLContext::init_dumping_tables(char *qdatabase) {
DBUG_ENTER("init_dumping_tables");
if (!opt_create_db) {
char qbuf[256];
MYSQL_ROW row;
MYSQL_RES *dbinfo;
my_snprintf(qbuf, sizeof(qbuf), "SHOW CREATE DATABASE IF NOT EXISTS %s",
qdatabase);
if (mysql_query(mysql, qbuf) || !(dbinfo = mysql_store_result(mysql))) {
if (opt_drop_database)
fprintf(md_result_file, "\n/*!40000 DROP DATABASE IF EXISTS %s*/;\n",
qdatabase);
fprintf(md_result_file,
"\nCREATE DATABASE /*!32312 IF NOT EXISTS*/ %s;\n", qdatabase);
} else {
if (opt_drop_database)
fprintf(md_result_file, "\n/*!40000 DROP DATABASE IF EXISTS %s*/;\n",
qdatabase);
row = mysql_fetch_row(dbinfo);
if (row[1]) {
fprintf(md_result_file, "\n%s;\n", row[1]);
}
mysql_free_result(dbinfo);
}
}
DBUG_RETURN(0);
}
int MySQLContext::init_dumping(char *database, int init_func) {
if (is_ndbinfo(mysql, database)) {
verbose_msg("-- Skipping dump of ndbinfo database\n");
return 0;
}
if (mysql_select_db(mysql, database)) {
DB_error(mysql, "when selecting the database");
return 1;
}
if (!path && !opt_xml) {
if (opt_databases || opt_alldbs) {
length of table name * 2 (if name contains quotes), 2 quotes and 0
*/
char quoted_database_buf[NAME_LEN * 2 + 3];
char *qdatabase = quote_name(database, quoted_database_buf, opt_quoted);
my_bool freemem = FALSE;
char const *text = fix_identifier_with_newline(qdatabase, &freemem);
print_comment(md_result_file, 0, "\n--\n-- Current Database: %s\n--\n",
text);
if (freemem) my_free((void *)text);
switch (init_func) {
case 1:
init_dumping_tables(qdatabase);
break;
case 2:
init_dumping_views(qdatabase);
break;
}
fprintf(md_result_file, "\nUSE %s;\n", qdatabase);
check_io(md_result_file);
}
}
return 0;
}
my_bool include_table(const uchar *hash_key, size_t len) {
return !my_hash_search(&ignore_table, hash_key, len);
}
int MySQLContext::dump_all_tables_in_db(char *database) {
char *table;
uint numrows;
char table_buff[NAME_LEN * 2 + 3];
char hash_key[2 * NAME_LEN + 2];
char *afterdot;
my_bool general_log_table_exists = 0, slow_log_table_exists = 0;
int using_mysql_db = !my_strcasecmp(charset_info, database, "mysql");
my_bool real_columns[MAX_FIELDS];
DBUG_ENTER("dump_all_tables_in_db");
afterdot = my_stpcpy(hash_key, database);
*afterdot++ = '.';
if (init_dumping(database, 1)) DBUG_RETURN(1);
if (opt_xml)
print_xml_tag(md_result_file, "", "\n", "database", "name=", database,
NullS);
if (lock_tables) {
DYNAMIC_STRING query;
init_dynamic_string_checked(&query, "LOCK TABLES ", 256, 1024);
for (numrows = 0; (table = getTableName(1));) {
char *end = my_stpcpy(afterdot, table);
if (include_table((uchar *)hash_key, end - hash_key)) {
numrows++;
dynstr_append_checked(&query, quote_name(table, table_buff, 1));
dynstr_append_checked(&query, " READ /*!32311 LOCAL */,");
}
}
if (numrows &&
mysql_real_query(mysql, query.str, (ulong)(query.length - 1)))
DB_error(mysql, "when using LOCK TABLES");
dynstr_free(&query);
}
if (flush_logs) {
if (mysql_refresh(mysql, REFRESH_LOG))
DB_error(mysql, "when doing refresh");
else
verbose_msg("-- dump_all_tables_in_db : logs flushed successfully!\n");
}
if (opt_single_transaction && mysql_get_server_version(mysql) >= 50500) {
verbose_msg("-- Setting savepoint...\n");
if (mysql_query_with_error_report(mysql, 0, "SAVEPOINT sp")) DBUG_RETURN(1);
}
std::list<std::string> alltables;
while ((table = getTableName(0))) {
alltables.push_back(table);
}
sg_firstdumptables = true;
while (alltables.size() > 0) {
for (std::list<std::string>::iterator iter = alltables.begin();
iter != alltables.end(); iter++) {
table = (char *)iter->c_str();
sg_procMetrics->startTable(table);
char *end = my_stpcpy(afterdot, table);
if (include_table((uchar *)hash_key, end - hash_key)) {
dump_table(table, database);
my_free(order_by);
order_by = 0;
if (opt_dump_triggers && mysql_get_server_version(mysql) >= 50009) {
if (dump_triggers_for_table(table, database)) {
if (path) my_fclose(md_result_file, MYF(MY_WME));
maybe_exit(EX_MYSQLERR);
}
}
ROLLBACK TO SAVEPOINT in --single-transaction mode to release metadata
lock on table which was already dumped. This allows to avoid blocking
concurrent DDL on this table without sacrificing correctness, as we
won't access table second time and dumps created by
--single-transaction mode have validity point at the start of
transaction anyway. Note that this doesn't make --single-transaction
mode with concurrent DDL safe in general case. It just improves
situation for people for whom it might be working.
*/
if (opt_single_transaction &&
mysql_get_server_version(mysql) >= 50500) {
verbose_msg("-- Rolling back to savepoint sp...\n");
if (mysql_query_with_error_report(mysql, 0,
"ROLLBACK TO SAVEPOINT sp"))
maybe_exit(EX_MYSQLERR);
}
} else {
If general_log and slow_log exists in the 'mysql' database,
we should dump the table structure. But we cannot
call get_table_structure() here as 'LOCK TABLES' query got executed
above on the session and that 'LOCK TABLES' query does not contain
'general_log' and 'slow_log' tables. (you cannot acquire lock
on log tables). Hence mark the existence of these log tables here and
after 'UNLOCK TABLES' query is executed on the session, get the table
structure from server and dump it in the file.
*/
if (using_mysql_db) {
if (!my_strcasecmp(charset_info, table, "general_log"))
general_log_table_exists = 1;
else if (!my_strcasecmp(charset_info, table, "slow_log"))
slow_log_table_exists = 1;
}
}
sg_procMetrics->endTable(table);
}
alltables = sg_createFailedTables;
sg_createFailedTables = std::list<std::string>();
sg_firstdumptables = false;
}
if (opt_single_transaction && mysql_get_server_version(mysql) >= 50500) {
verbose_msg("-- Releasing savepoint...\n");
if (mysql_query_with_error_report(mysql, 0, "RELEASE SAVEPOINT sp"))
DBUG_RETURN(1);
}
if (opt_events && mysql_get_server_version(mysql) >= 50106) {
DBUG_PRINT("info", ("Dumping events for database %s", database));
dump_events_for_db(database);
}
if (opt_routines && mysql_get_server_version(mysql) >= 50009) {
DBUG_PRINT("info", ("Dumping routines for database %s", database));
dump_routines_for_db(database);
}
if (opt_xml) {
fputs("</database>\n", md_result_file);
check_io(md_result_file);
}
if (lock_tables)
(void)mysql_query_with_error_report(mysql, 0, "UNLOCK TABLES");
if (using_mysql_db) {
char table_type[NAME_LEN];
char ignore_flag;
if (general_log_table_exists) {
if (!get_table_structure((char *)"general_log", database, table_type,
&ignore_flag, real_columns))
verbose_msg(
"-- Warning: get_table_structure() failed with some internal "
"error for 'general_log' table\n");
}
if (slow_log_table_exists) {
if (!get_table_structure((char *)"slow_log", database, table_type,
&ignore_flag, real_columns))
verbose_msg(
"-- Warning: get_table_structure() failed with some internal "
"error for 'slow_log' table\n");
}
}
if (flush_privileges && using_mysql_db) {
fprintf(md_result_file, "\n--\n-- Flush Grant Tables \n--\n");
fprintf(md_result_file, "\n/*! FLUSH PRIVILEGES */;\n");
}
DBUG_RETURN(0);
}
dump structure of views of database
SYNOPSIS
dump_all_views_in_db()
database database name
RETURN
0 OK
1 ERROR
*/
my_bool MySQLContext::dump_all_views_in_db(char *database) {
char *table;
uint numrows;
char table_buff[NAME_LEN * 2 + 3];
char hash_key[2 * NAME_LEN + 2];
char *afterdot;
afterdot = my_stpcpy(hash_key, database);
*afterdot++ = '.';
if (init_dumping(database, 2)) return 1;
if (opt_xml)
print_xml_tag(md_result_file, "", "\n", "database", "name=", database,
NullS);
if (lock_tables) {
DYNAMIC_STRING query;
init_dynamic_string_checked(&query, "LOCK TABLES ", 256, 1024);
for (numrows = 0; (table = getTableName(1));) {
char *end = my_stpcpy(afterdot, table);
if (include_table((uchar *)hash_key, end - hash_key)) {
numrows++;
dynstr_append_checked(&query, quote_name(table, table_buff, 1));
dynstr_append_checked(&query, " READ /*!32311 LOCAL */,");
}
}
if (numrows &&
mysql_real_query(mysql, query.str, (ulong)(query.length - 1)))
DB_error(mysql, "when using LOCK TABLES");
dynstr_free(&query);
}
if (flush_logs) {
if (mysql_refresh(mysql, REFRESH_LOG))
DB_error(mysql, "when doing refresh");
else
verbose_msg("-- dump_all_views_in_db : logs flushed successfully!\n");
}
while ((table = getTableName(0))) {
char *end = my_stpcpy(afterdot, table);
if (include_table((uchar *)hash_key, end - hash_key))
get_view_structure(table, database);
}
if (opt_xml) {
fputs("</database>\n", md_result_file);
check_io(md_result_file);
}
if (lock_tables)
(void)mysql_query_with_error_report(mysql, 0, "UNLOCK TABLES");
return 0;
}
get_actual_table_name -- executes a SHOW TABLES LIKE '%s' to get the actual
table name from the server for the table name given on the command line.
we do this because the table name given on the command line may be a
different case (e.g. T1 vs t1)
RETURN
pointer to the table name
0 if error
*/
char *MySQLContext::get_actual_table_name(const char *old_table_name,
MEM_ROOT *root) {
char *name = 0;
MYSQL_RES *table_res;
MYSQL_ROW row;
char query[50 + 2 * NAME_LEN];
char show_name_buff[FN_REFLEN];
DBUG_ENTER("get_actual_table_name");
DBUG_ASSERT(2 * sizeof(old_table_name) < sizeof(show_name_buff));
my_snprintf(query, sizeof(query), "SHOW TABLES LIKE %s",
quote_for_like(old_table_name, show_name_buff));
if (mysql_query_with_error_report(mysql, 0, query)) DBUG_RETURN(NullS);
if ((table_res = mysql_store_result(mysql))) {
my_ulonglong num_rows = mysql_num_rows(table_res);
if (num_rows > 0) {
ulong *lengths;
Return first row
TODO: Return all matching rows
*/
row = mysql_fetch_row(table_res);
lengths = mysql_fetch_lengths(table_res);
name = strmake_root(root, row[0], lengths[0]);
}
mysql_free_result(table_res);
}
DBUG_PRINT("exit", ("new_table_name: %s", name));
DBUG_RETURN(name);
}
int MySQLContext::dump_selected_tables(char *db, char **table_names,
int tables) {
char table_buff[NAME_LEN * 2 + 3];
DYNAMIC_STRING lock_tables_query;
MEM_ROOT root;
char **dump_tables, **pos, **end;
DBUG_ENTER("dump_selected_tables");
if (init_dumping(db, 1)) DBUG_RETURN(1);
init_alloc_root(PSI_NOT_INSTRUMENTED, &root, 8192, 0);
if (!(dump_tables = pos =
(char **)alloc_root(&root, tables * sizeof(char *))))
die(EX_EOM, "alloc_root failure.");
init_dynamic_string_checked(&lock_tables_query, "LOCK TABLES ", 256, 1024);
for (; tables > 0; tables--, table_names++) {
if ((*pos = get_actual_table_name(*table_names, &root))) {
if (lock_tables) {
dynstr_append_checked(&lock_tables_query,
quote_name(*pos, table_buff, 1));
dynstr_append_checked(&lock_tables_query, " READ /*!32311 LOCAL */,");
}
pos++;
} else {
if (!opt_force) {
dynstr_free(&lock_tables_query);
free_root(&root, MYF(0));
}
maybe_die(EX_ILLEGAL_TABLE, "Couldn't find table: \"%s\"", *table_names);
}
}
end = pos;
if (lock_tables &&
!(mysql_get_server_version(mysql) >= FIRST_INFORMATION_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, db, INFORMATION_SCHEMA_DB_NAME)) &&
!(mysql_get_server_version(mysql) >= FIRST_PERFORMANCE_SCHEMA_VERSION &&
!my_strcasecmp(&my_charset_latin1, db, PERFORMANCE_SCHEMA_DB_NAME))) {
if (mysql_real_query(mysql, lock_tables_query.str,
(ulong)(lock_tables_query.length - 1))) {
if (!opt_force) {
dynstr_free(&lock_tables_query);
free_root(&root, MYF(0));
}
DB_error(mysql, "when doing LOCK TABLES");
}
}
dynstr_free(&lock_tables_query);
if (flush_logs) {
if (mysql_refresh(mysql, REFRESH_LOG)) {
if (!opt_force) free_root(&root, MYF(0));
DB_error(mysql, "when doing refresh");
}
else
verbose_msg("-- dump_selected_tables : logs flushed successfully!\n");
}
if (opt_xml)
print_xml_tag(md_result_file, "", "\n", "database", "name=", db, NullS);
if (opt_single_transaction && mysql_get_server_version(mysql) >= 50500) {
verbose_msg("-- Setting savepoint...\n");
if (mysql_query_with_error_report(mysql, 0, "SAVEPOINT sp")) DBUG_RETURN(1);
}
for (pos = dump_tables; pos < end; pos++) {
DBUG_PRINT("info", ("Dumping table %s", *pos));
dump_table(*pos, db);
if (opt_dump_triggers && mysql_get_server_version(mysql) >= 50009) {
if (dump_triggers_for_table(*pos, db)) {
if (path) my_fclose(md_result_file, MYF(MY_WME));
maybe_exit(EX_MYSQLERR);
}
}
ROLLBACK TO SAVEPOINT in --single-transaction mode to release metadata
lock on table which was already dumped. This allows to avoid blocking
concurrent DDL on this table without sacrificing correctness, as we
won't access table second time and dumps created by --single-transaction
mode have validity point at the start of transaction anyway.
Note that this doesn't make --single-transaction mode with concurrent
DDL safe in general case. It just improves situation for people for whom
it might be working.
*/
if (opt_single_transaction && mysql_get_server_version(mysql) >= 50500) {
verbose_msg("-- Rolling back to savepoint sp...\n");
if (mysql_query_with_error_report(mysql, 0, "ROLLBACK TO SAVEPOINT sp"))
maybe_exit(EX_MYSQLERR);
}
}
if (opt_single_transaction && mysql_get_server_version(mysql) >= 50500) {
verbose_msg("-- Releasing savepoint...\n");
if (mysql_query_with_error_report(mysql, 0, "RELEASE SAVEPOINT sp"))
DBUG_RETURN(1);
}
if (seen_views) {
for (pos = dump_tables; pos < end; pos++) get_view_structure(*pos, db);
}
if (opt_events && mysql_get_server_version(mysql) >= 50106) {
DBUG_PRINT("info", ("Dumping events for database %s", db));
dump_events_for_db(db);
}
if (opt_routines && mysql_get_server_version(mysql) >= 50009) {
DBUG_PRINT("info", ("Dumping routines for database %s", db));
dump_routines_for_db(db);
}
free_root(&root, MYF(0));
my_free(order_by);
order_by = 0;
if (opt_xml) {
fputs("</database>\n", md_result_file);
check_io(md_result_file);
}
if (lock_tables)
(void)mysql_query_with_error_report(mysql, 0, "UNLOCK TABLES");
DBUG_RETURN(0);
}
int MySQLContext::do_show_master_status(MYSQL *mysql_con) {
MYSQL_ROW row;
MYSQL_RES *master;
const char *comment_prefix =
(opt_master_data == MYSQL_OPT_MASTER_DATA_COMMENTED_SQL) ? "-- " : "";
if (mysql_query_with_error_report(mysql_con, &master, "SHOW MASTER STATUS")) {
return 1;
} else {
row = mysql_fetch_row(master);
if (row && row[0] && row[1]) {
print_comment(md_result_file, 0,
"\n--\n-- Position to start replication or point-in-time "
"recovery from\n--\n\n");
fprintf(md_result_file,
"%sCHANGE MASTER TO MASTER_LOG_FILE='%s', MASTER_LOG_POS=%s;\n",
comment_prefix, row[0], row[1]);
check_io(md_result_file);
} else if (!opt_force) {
my_printf_error(0, "Error: Binlogging on server not active", MYF(0));
mysql_free_result(master);
maybe_exit(EX_MYSQLERR);
return 1;
}
mysql_free_result(master);
}
return 0;
}
int MySQLContext::do_stop_slave_sql(MYSQL *mysql_con) {
MYSQL_RES *slave;
if (mysql_query_with_error_report(mysql_con, &slave, "SHOW SLAVE STATUS"))
return (1);
else {
MYSQL_ROW row = mysql_fetch_row(slave);
if (row && row[11]) {
if (!strcmp(row[11], "No")) {
mysql_free_result(slave);
return (0);
}
}
}
mysql_free_result(slave);
if (mysql_query_with_error_report(mysql_con, 0, "STOP SLAVE SQL_THREAD"))
return (1);
return (0);
}
int MySQLContext::add_stop_slave(void) {
if (opt_comments)
fprintf(md_result_file,
"\n--\n-- stop slave statement to make a recovery dump)\n--\n\n");
fprintf(md_result_file, "STOP SLAVE;\n");
return (0);
}
int MySQLContext::add_slave_statements(void) {
if (opt_comments)
fprintf(md_result_file,
"\n--\n-- start slave statement to make a recovery dump)\n--\n\n");
fprintf(md_result_file, "START SLAVE;\n");
return (0);
}
int MySQLContext::do_show_slave_status(MYSQL *mysql_con) {
MYSQL_RES *slave = NULL;
const char *comment_prefix =
(opt_slave_data == MYSQL_OPT_SLAVE_DATA_COMMENTED_SQL) ? "-- " : "";
if (mysql_query_with_error_report(mysql_con, &slave, "SHOW SLAVE STATUS")) {
if (!opt_force) {
my_printf_error(0, "Error: Slave not set up", MYF(0));
}
mysql_free_result(slave);
return 1;
} else {
const int n_master_host = 1;
const int n_master_port = 3;
const int n_master_log_file = 9;
const int n_master_log_pos = 21;
const int n_channel_name = 55;
MYSQL_ROW row = mysql_fetch_row(slave);
while (row) {
if (row[n_master_log_file] && row[n_master_log_pos]) {
if (opt_comments)
fprintf(md_result_file,
"\n--\n-- Position to start replication or point-in-time "
"recovery from (the master of this slave)\n--\n\n");
fprintf(md_result_file, "%sCHANGE MASTER TO ", comment_prefix);
if (opt_include_master_host_port) {
if (row[n_master_host])
fprintf(md_result_file, "MASTER_HOST='%s', ", row[n_master_host]);
if (row[n_master_port])
fprintf(md_result_file, "MASTER_PORT=%s, ", row[n_master_port]);
}
fprintf(md_result_file, "MASTER_LOG_FILE='%s', MASTER_LOG_POS=%s",
row[n_master_log_file], row[n_master_log_pos]);
if (slave->row_count > 1)
fprintf(md_result_file, " FOR CHANNEL '%s'", row[n_channel_name]);
fprintf(md_result_file, ";\n");
}
row = mysql_fetch_row(slave);
}
check_io(md_result_file);
mysql_free_result(slave);
}
return 0;
}
int MySQLContext::do_start_slave_sql(MYSQL *mysql_con) {
MYSQL_RES *slave;
if (mysql_query_with_error_report(mysql_con, &slave, "SHOW SLAVE STATUS"))
return (1);
else {
MYSQL_ROW row = mysql_fetch_row(slave);
if (row && row[11]) {
if (!strcmp(row[11], "Yes")) {
mysql_free_result(slave);
return (0);
}
}
}
mysql_free_result(slave);
if (mysql_query_with_error_report(mysql_con, 0, "START SLAVE")) {
my_printf_error(0, "Error: Unable to start slave", MYF(0));
return 1;
}
return (0);
}
int MySQLContext::do_flush_tables_read_lock(MYSQL *mysql_con) {
We do first a FLUSH TABLES. If a long update is running, the FLUSH TABLES
will wait but will not stall the whole mysqld, and when the long update is
done the FLUSH TABLES WITH READ LOCK will start and succeed quickly. So,
FLUSH TABLES is to lower the probability of a stage where both mysqldump
and most client connections are stalled. Of course, if a second long
update starts between the two FLUSHes, we have that bad stall.
*/
return (mysql_query_with_error_report(
mysql_con, 0,
((opt_master_data != 0) ? "FLUSH /*!40101 LOCAL */ TABLES"
: "FLUSH TABLES")) ||
mysql_query_with_error_report(mysql_con, 0,
"FLUSH TABLES WITH READ LOCK"));
}
int MySQLContext::do_unlock_tables(MYSQL *mysql_con) {
return mysql_query_with_error_report(mysql_con, 0, "UNLOCK TABLES");
}
int MySQLContext::get_bin_log_name(MYSQL *mysql_con, char *buff_log_name,
uint buff_len) {
MYSQL_RES *res;
MYSQL_ROW row;
if (mysql_query(mysql_con, "SHOW MASTER STATUS") ||
!(res = mysql_store_result(mysql)))
return 1;
if (!(row = mysql_fetch_row(res))) {
mysql_free_result(res);
return 1;
}
Only one row is returned, and the first column is the name of the
active log.
*/
strmake(buff_log_name, row[0], buff_len - 1);
mysql_free_result(res);
return 0;
}
int MySQLContext::purge_bin_logs_to(MYSQL *mysql_con, char *log_name) {
DYNAMIC_STRING str;
int err;
init_dynamic_string_checked(&str, "PURGE BINARY LOGS TO '", 1024, 1024);
dynstr_append_checked(&str, log_name);
dynstr_append_checked(&str, "'");
err = mysql_query_with_error_report(mysql_con, 0, str.str);
dynstr_free(&str);
return err;
}
int MySQLContext::start_transaction(MYSQL *mysql_con) {
verbose_msg("-- Starting transaction...\n");
We use BEGIN for old servers. --single-transaction --master-data will fail
on old servers, but that's ok as it was already silently broken (it didn't
do a consistent read, so better tell people frankly, with the error).
We want the first consistent read to be used for all tables to dump so we
need the REPEATABLE READ level (not anything lower, for example READ
COMMITTED would give one new consistent read per dumped table).
*/
if ((mysql_get_server_version(mysql_con) < 40100) && opt_master_data) {
fprintf(stderr,
"-- %s: the combination of --single-transaction and "
"--master-data requires a MySQL server version of at least 4.1 "
"(current server's version is %s). %s\n",
opt_force ? "Warning" : "Error",
mysql_con->server_version ? mysql_con->server_version : "unknown",
opt_force ? "Continuing due to --force, backup may not be "
"consistent across all tables!"
: "Aborting.");
if (!opt_force) exit(EX_MYSQLERR);
}
return (
mysql_query_with_error_report(mysql_con, 0,
"SET SESSION TRANSACTION ISOLATION "
"LEVEL REPEATABLE READ") ||
mysql_query_with_error_report(mysql_con, 0,
"START TRANSACTION "
"/*!40100 WITH CONSISTENT SNAPSHOT */"));
}
ulong MySQLContext::find_set(TYPELIB *lib, const char *x, size_t length,
char **err_pos, uint *err_len) {
const char *end = x + length;
ulong found = 0;
uint find;
char buff[255];
*err_pos = 0;
while (end > x && my_isspace(charset_info, end[-1])) 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, MY_MIN(sizeof(buff) - 1, var_len));
find = find_type(buff, lib, FIND_TYPE_BASIC);
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;
}
void MySQLContext::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"))
{
fputc(' ', file);
fputs(prefix, file);
if (string_value)
unescape(file, row[0], strlen(row[0]), NULL);
else
fputs(row[0], file);
check_io(file);
return;
}
}
}
return;
}
SYNOPSIS
Check if the table is one of the table types that should be ignored:
MRG_ISAM, MRG_MYISAM.
If the table should be altogether ignored, it returns a TRUE, FALSE if it
should not be ignored.
ARGS
check_if_ignore_table()
table_name Table name to check
table_type Type of table
GLOBAL VARIABLES
mysql MySQL connection
verbose Write warning messages
RETURN
char (bit value) See IGNORE_ values at top
*/
char MySQLContext::check_if_ignore_table(const char *table_name,
char *table_type) {
char result = IGNORE_NONE;
char buff[FN_REFLEN + 80], show_name_buff[FN_REFLEN];
MYSQL_RES *res = NULL;
MYSQL_ROW row;
DBUG_ENTER("check_if_ignore_table");
DBUG_ASSERT(2 * sizeof(table_name) < sizeof(show_name_buff));
my_snprintf(buff, sizeof(buff), "show table status like %s",
quote_for_like(table_name, show_name_buff));
if (mysql_query_with_error_report(mysql, &res, buff)) {
if (mysql_errno(mysql) != ER_PARSE_ERROR) {
verbose_msg(
"-- Warning: Couldn't get status information for "
"table %s (%s)\n",
table_name, mysql_error(mysql));
DBUG_RETURN(result);
}
}
if (!(row = mysql_fetch_row(res))) {
fprintf(stderr,
"Error: Couldn't read status information for table %s (%s)\n",
table_name, mysql_error(mysql));
mysql_free_result(res);
DBUG_RETURN(result);
}
if (!(row[1]))
strmake(table_type, "VIEW", NAME_LEN - 1);
else {
strmake(table_type, row[1], NAME_LEN - 1);
if (!opt_no_data &&
(!my_strcasecmp(&my_charset_latin1, table_type, "MRG_MyISAM") ||
!strcmp(table_type, "MRG_ISAM") || !strcmp(table_type, "FEDERATED")))
result = IGNORE_DATA;
}
mysql_free_result(res);
DBUG_RETURN(result);
}
Get string of comma-separated primary key field names
SYNOPSIS
char *primary_key_fields(const char *table_name)
RETURNS pointer to allocated buffer (must be freed by caller)
table_name quoted table name
DESCRIPTION
Use SHOW KEYS FROM table_name, allocate a buffer to hold the
field names, and then build that string and return the pointer
to that buffer.
Returns NULL if there is no PRIMARY or UNIQUE key on the table,
or if there is some failure. It is better to continue to dump
the table unsorted, rather than exit without dumping the data.
*/
char *MySQLContext::primary_key_fields(const char *table_name) {
MYSQL_RES *res = NULL;
MYSQL_ROW row;
char show_keys_buff[15 + NAME_LEN * 2 + 3];
size_t result_length = 0;
char *result = 0;
char buff[NAME_LEN * 2 + 3];
char *quoted_field;
my_snprintf(show_keys_buff, sizeof(show_keys_buff), "SHOW KEYS FROM %s",
table_name);
if (mysql_query(mysql, show_keys_buff) ||
!(res = mysql_store_result(mysql))) {
fprintf(stderr,
"Warning: Couldn't read keys from table %s;"
" records are NOT sorted (%s)\n",
table_name, mysql_error(mysql));
goto cleanup;
}
* Figure out the length of the ORDER BY clause result.
* Note that SHOW KEYS is ordered: a PRIMARY key is always the first
* row, and UNIQUE keys come before others. So we only need to check
* the first key, not all keys.
*/
if ((row = mysql_fetch_row(res)) && atoi(row[1]) == 0) {
do {
quoted_field = quote_name(row[4], buff, 0);
result_length += strlen(quoted_field) + 1;
} while ((row = mysql_fetch_row(res)) && atoi(row[3]) > 1);
}
if (result_length) {
char *end;
result = (char *)my_malloc(PSI_NOT_INSTRUMENTED, result_length + 10,
MYF(MY_WME));
if (!result) {
fprintf(stderr, "Error: Not enough memory to store ORDER BY clause\n");
goto cleanup;
}
mysql_data_seek(res, 0);
row = mysql_fetch_row(res);
quoted_field = quote_name(row[4], buff, 0);
end = my_stpcpy(result, quoted_field);
while ((row = mysql_fetch_row(res)) && atoi(row[3]) > 1) {
quoted_field = quote_name(row[4], buff, 0);
end = strxmov(end, ",", quoted_field, NullS);
}
}
cleanup:
if (res) mysql_free_result(res);
return result;
}
Replace a substring
SYNOPSIS
replace
ds_str The string to search and perform the replace in
search_str The string to search for
search_len Length of the string to search for
replace_str The string to replace with
replace_len Length of the string to replace with
RETURN
0 String replaced
1 Could not find search_str in str
*/
static int replace(DYNAMIC_STRING *ds_str, const char *search_str,
size_t search_len, const char *replace_str,
size_t replace_len) {
DYNAMIC_STRING ds_tmp;
const char *start = strstr(ds_str->str, search_str);
if (!start) return 1;
init_dynamic_string_checked(&ds_tmp, "", ds_str->length + replace_len, 256);
dynstr_append_mem_checked(&ds_tmp, ds_str->str, start - ds_str->str);
dynstr_append_mem_checked(&ds_tmp, replace_str, replace_len);
dynstr_append_checked(&ds_tmp, start + search_len);
dynstr_set_checked(ds_str, ds_tmp.str);
dynstr_free(&ds_tmp);
return 0;
}
This function sets the session binlog in the dump file.
When --set-gtid-purged is used, this function is called to
disable the session binlog and at the end of the dump, to restore
the session binlog.
@note: md_result_file should have been opened, before
this function is called.
@param[in] flag If FALSE, disable binlog.
If TRUE and binlog disabled previously,
restore the session binlog.
*/
static void set_session_binlog(my_bool flag) {
static my_bool is_binlog_disabled = FALSE;
if (!flag && !is_binlog_disabled) {
fprintf(md_result_file,
"SET @MYSQLDUMP_TEMP_LOG_BIN = @@SESSION.SQL_LOG_BIN;\n");
fprintf(md_result_file, "SET @@SESSION.SQL_LOG_BIN= 0;\n");
is_binlog_disabled = 1;
} else if (flag && is_binlog_disabled) {
fprintf(md_result_file,
"SET @@SESSION.SQL_LOG_BIN = @MYSQLDUMP_TEMP_LOG_BIN;\n");
is_binlog_disabled = 0;
}
}
This function gets the GTID_EXECUTED sets from the
server and assigns those sets to GTID_PURGED in the
dump file.
@param[in] mysql_con connection to the server
@retval FALSE succesfully printed GTID_PURGED sets
in the dump file.
@retval TRUE failed.
*/
static my_bool add_set_gtid_purged(MYSQL *mysql_con) {
MYSQL_RES *gtid_purged_res;
MYSQL_ROW gtid_set;
ulonglong num_sets, idx;
if (mysql_query_with_error_report(mysql_con, >id_purged_res,
"SELECT @@GLOBAL.GTID_EXECUTED"))
return TRUE;
if ((num_sets = mysql_num_rows(gtid_purged_res)) > 0) {
if (opt_comments)
fprintf(md_result_file,
"\n--\n-- GTID state at the beginning of the backup \n--\n\n");
fprintf(md_result_file, "SET @@GLOBAL.GTID_PURGED='");
for (idx = 0; idx < num_sets - 1; idx++) {
gtid_set = mysql_fetch_row(gtid_purged_res);
fprintf(md_result_file, "%s,", (char *)gtid_set[0]);
}
gtid_set = mysql_fetch_row(gtid_purged_res);
fprintf(md_result_file, "%s';\n", (char *)gtid_set[0]);
}
mysql_free_result(gtid_purged_res);
return FALSE;
}
This function processes the opt_set_gtid_purged option.
This function also calls set_session_binlog() function before
setting the SET @@GLOBAL.GTID_PURGED in the output.
@param[in] mysql_con the connection to the server
@retval FALSE successful according to the value
of opt_set_gtid_purged.
@retval TRUE fail.
*/
my_bool MySQLContext::process_set_gtid_purged(MYSQL *mysql_con) {
MYSQL_RES *gtid_mode_res;
MYSQL_ROW gtid_mode_row;
char *gtid_mode_val = 0;
char buf[32], query[64];
if (opt_set_gtid_purged_mode == SET_GTID_PURGED_OFF)
return FALSE;
Check if the server has the knowledge of GTIDs(pre mysql-5.6)
or if the gtid_mode is ON or OFF.
*/
my_snprintf(query, sizeof(query), "SHOW VARIABLES LIKE %s",
quote_for_like("gtid_mode", buf));
if (mysql_query_with_error_report(mysql_con, >id_mode_res, query))
return TRUE;
gtid_mode_row = mysql_fetch_row(gtid_mode_res);
gtid_mode_row is NULL for pre 5.6 versions. For versions >= 5.6,
get the gtid_mode value from the second column.
*/
gtid_mode_val = gtid_mode_row ? (char *)gtid_mode_row[1] : NULL;
if (gtid_mode_val && strcmp(gtid_mode_val, "OFF")) {
For any gtid_mode !=OFF and irrespective of --set-gtid-purged
being AUTO or ON, add GTID_PURGED in the output.
*/
if (opt_databases || !opt_alldbs || !opt_dump_triggers || !opt_routines ||
!opt_events) {
fprintf(stderr,
"Warning: A partial dump from a server that has GTIDs will "
"by default include the GTIDs of all transactions, even "
"those that changed suppressed parts of the database. If "
"you don't want to restore GTIDs, pass "
"--set-gtid-purged=OFF. To make a complete dump, pass "
"--all-databases --triggers --routines --events. \n");
}
set_session_binlog(FALSE);
if (add_set_gtid_purged(mysql_con)) {
mysql_free_result(gtid_mode_res);
return TRUE;
}
} else
{
if (opt_set_gtid_purged_mode == SET_GTID_PURGED_ON) {
fprintf(stderr, "Error: Server has GTIDs disabled.\n");
mysql_free_result(gtid_mode_res);
return TRUE;
}
}
mysql_free_result(gtid_mode_res);
return FALSE;
}
Getting VIEW structure
SYNOPSIS
get_view_structure()
table view name
db db name
RETURN
0 OK
1 ERROR
*/
my_bool MySQLContext::get_view_structure(char *table, char *db) {
MYSQL_RES *table_res;
MYSQL_ROW row;
MYSQL_FIELD *field;
char *result_table;
char table_buff[NAME_LEN * 2 + 3];
char table_buff2[NAME_LEN * 2 + 3];
char query[QUERY_LENGTH];
FILE *sql_file = md_result_file;
my_bool freemem = FALSE;
char const *text;
std::string createviewsql;
DBUG_ENTER("get_view_structure");
if (opt_no_create_info)
DBUG_RETURN(0);
verbose_msg("-- Retrieving view structure for table %s...\n", table);
result_table = quote_name(table, table_buff, 1);
quote_name(table, table_buff2, 0);
if (switch_character_set_results(mysql, "binary")) DBUG_RETURN(1);
my_snprintf(query, sizeof(query), "SHOW CREATE TABLE %s", result_table);
if (mysql_query_with_error_report(mysql, &table_res, query)) {
switch_character_set_results(mysql, default_charset);
DBUG_RETURN(0);
}
field = mysql_fetch_field_direct(table_res, 0);
if (strcmp(field->name, "View") != 0) {
switch_character_set_results(mysql, default_charset);
verbose_msg("-- It's base table, skipped\n");
mysql_free_result(table_res);
DBUG_RETURN(0);
}
if (path) {
if (!(sql_file = open_sql_file_for_table(table, O_WRONLY))) {
mysql_free_result(table_res);
DBUG_RETURN(1);
}
write_header(sql_file, db);
}
text = fix_identifier_with_newline(result_table, &freemem);
print_comment(sql_file, 0,
"\n--\n-- Final view structure for view %s\n--\n\n", text);
if (freemem) my_free((void *)text);
verbose_msg("-- Dropping the temporary view structure created\n");
my_snprintf(query, sizeof(query),
"SELECT CHECK_OPTION, DEFINER, SECURITY_TYPE, "
" CHARACTER_SET_CLIENT, COLLATION_CONNECTION "
"FROM information_schema.views "
"WHERE table_name=\"%s\" AND table_schema=\"%s\"",
table, db);
if (mysql_query(mysql, query)) {
Use the raw output from SHOW CREATE TABLE if
information_schema query fails.
*/
row = mysql_fetch_row(table_res);
createviewsql = row[1];
mysql_free_result(table_res);
} else {
char *ptr;
ulong *lengths;
char search_buf[256], replace_buf[256];
ulong search_len, replace_len;
DYNAMIC_STRING ds_view;
row = mysql_fetch_row(table_res);
lengths = mysql_fetch_lengths(table_res);
init_dynamic_string_checked(&ds_view, row[1], lengths[1] + 1, 1024);
mysql_free_result(table_res);
if (!(table_res = mysql_store_result(mysql)) ||
!(row = mysql_fetch_row(table_res))) {
if (table_res) mysql_free_result(table_res);
dynstr_free(&ds_view);
DB_error(
mysql,
"when trying to save the result of SHOW CREATE TABLE in ds_view.");
DBUG_RETURN(1);
}
lengths = mysql_fetch_lengths(table_res);
"WITH %s CHECK OPTION" is available from 5.0.2
Surround it with !50002 comments
*/
if (strcmp(row[0], "NONE")) {
ptr = search_buf;
search_len =
(ulong)(strxmov(ptr, "WITH ", row[0], " CHECK OPTION", NullS) - ptr);
ptr = replace_buf;
replace_len = (ulong)(
strxmov(ptr, "*/\n/*!50002 WITH ", row[0], " CHECK OPTION", NullS) -
ptr);
replace(&ds_view, search_buf, search_len, replace_buf, replace_len);
}
"DEFINER=%s SQL SECURITY %s" is available from 5.0.13
Surround it with !50013 comments
*/
{
size_t user_name_len;
char user_name_str[USERNAME_LENGTH + 1];
char quoted_user_name_str[USERNAME_LENGTH * 2 + 3];
size_t host_name_len;
char host_name_str[HOSTNAME_LENGTH + 1];
char quoted_host_name_str[HOSTNAME_LENGTH * 2 + 3];
parse_user(row[1], lengths[1], user_name_str, &user_name_len,
host_name_str, &host_name_len);
ptr = search_buf;
search_len = (ulong)(
strxmov(ptr, "DEFINER=",
quote_name(user_name_str, quoted_user_name_str, FALSE), "@",
quote_name(host_name_str, quoted_host_name_str, FALSE),
" SQL SECURITY ", row[2], NullS) -
ptr);
ptr = replace_buf;
replace_len = (ulong)(
strxmov(ptr, "*/\n/*!50013 DEFINER=",
quote_name(user_name_str, quoted_user_name_str, FALSE), "@",
quote_name(host_name_str, quoted_host_name_str, FALSE),
" SQL SECURITY ", row[2], " */\n/*!50001", NullS) -
ptr);
replace(&ds_view, search_buf, search_len, replace_buf, replace_len);
}
createviewsql = "CREATE ";
createviewsql += ds_view.str;
check_io(sql_file);
mysql_free_result(table_res);
dynstr_free(&ds_view);
}
struct DMODBC::DBError dbError;
sg_procMetrics->startSql(table, "view", createviewsql.c_str());
sg_procMetrics->addViewcount(1);
if (sg_procOdbc->executeCreateViewSql(createviewsql.c_str(), &dbError)) {
sg_procMetrics->endSql(table, "view", createviewsql.c_str(), false, "", false);
} else {
sg_procMetrics->endSql(table, "view", createviewsql.c_str(), true,
dbError.error.c_str(), false);
}
if (switch_character_set_results(mysql, default_charset)) DBUG_RETURN(1);
if (sql_file != md_result_file) {
fputs("\n", sql_file);
write_footer(sql_file);
my_fclose(sql_file, MYF(MY_WME));
}
DBUG_RETURN(0);
}
The following functions are wrappers for the dynamic string functions
and if they fail, the wrappers will terminate the current process.
*/
#define DYNAMIC_STR_ERROR_MSG "Couldn't perform DYNAMIC_STRING operation"
static void init_dynamic_string_checked(DYNAMIC_STRING *str,
const char *init_str, size_t init_alloc,
size_t alloc_increment) {
if (init_dynamic_string(str, init_str, init_alloc, alloc_increment))
die(EX_MYSQLERR, DYNAMIC_STR_ERROR_MSG);
}
static void dynstr_append_checked(DYNAMIC_STRING *dest, const char *src) {
if (dynstr_append(dest, src)) die(EX_MYSQLERR, DYNAMIC_STR_ERROR_MSG);
}
static void dynstr_set_checked(DYNAMIC_STRING *str, const char *init_str) {
if (dynstr_set(str, init_str)) die(EX_MYSQLERR, DYNAMIC_STR_ERROR_MSG);
}
static void dynstr_append_mem_checked(DYNAMIC_STRING *str, const char *append,
size_t length) {
if (dynstr_append_mem(str, append, length))
die(EX_MYSQLERR, DYNAMIC_STR_ERROR_MSG);
}
static void dynstr_realloc_checked(DYNAMIC_STRING *str,
size_t additional_size) {
if (dynstr_realloc(str, additional_size))
die(EX_MYSQLERR, DYNAMIC_STR_ERROR_MSG);
}
int main(int argc, char **argv) {
MySQLContext mySQLContext;
return mySQLContext.mainproc(argc, argv);
}
int MySQLContext::mainproc(int argc, char **argv) {
sg_procOdbc = new DMODBC::ProcOdbc("opengauss");
if (!sg_procOdbc->connect()) {
return 1;
}
if (pthread_key_create(&sg_connectionkey, 0) != 0) {
printf("cannot create connection thread local key");
return 1;
}
sg_mysqlInfo = DMODBC::ProcOdbc::getMysqlInfo();
printf("mysql addr: %s:%d\n", sg_mysqlInfo.host.c_str(), sg_mysqlInfo.port);
printf("mysql username: %s\n", sg_mysqlInfo.username.c_str());
printf("mysql database: %s\n", sg_mysqlInfo.database.c_str());
sg_procOdbc->setLoglevel(sg_mysqlInfo.loglevel);
sg_firstdumptables = true;
char srcaddr[256];
char dstaddr[256];
const char *metricsDbtype = getenv("metrics_dbtype");
if (metricsDbtype == NULL) {
metricsDbtype = "mysql";
}
char metricds[256];
sprintf(metricds, "metrics_%s", metricsDbtype);
sprintf(srcaddr, "%s:%d", sg_mysqlInfo.host.c_str(), sg_mysqlInfo.port);
sprintf(dstaddr, "%s:%d", sg_mysqlInfo.dsthost.c_str(), sg_mysqlInfo.dstport);
sg_procMetrics = new DMODBC::ProcMetrics(
metricds, srcaddr, dstaddr, sg_mysqlInfo.database.c_str(),
sg_mysqlInfo.dstdatabase.c_str(), 1000000, metricsDbtype);
char bin_log_name[FN_REFLEN];
int md_result_fd;
MY_INIT("mysqldump");
compatible_mode_normal_str[0] = 0;
default_charset = "utf8";
memset(&ignore_table, 0, sizeof(ignore_table));
argc = sg_procOdbc->getCurrentArgc();
argv = sg_procOdbc->getCurrentArgv();
get_options(&argc, &argv);
md_result_file = stdout;
current_host = (char *)sg_mysqlInfo.host.c_str();
current_user = (char *)sg_mysqlInfo.username.c_str();
opt_password = my_strdup(PSI_NOT_INSTRUMENTED, sg_mysqlInfo.password.c_str(),
MYF(MY_WME));
opt_mysql_port = sg_mysqlInfo.port;
opt_alldbs = 0;
Disable comments in xml mode if 'comments' option is not explicitly used.
*/
if (opt_xml && !opt_comments_used) opt_comments = 0;
if (log_error_file) {
if (!(stderror_file = freopen(log_error_file, "a+", stderr))) {
free_resources();
exit(EX_MYSQLERR);
}
}
if (connect_to_db(current_host, current_user, opt_password)) {
free_resources();
exit(EX_MYSQLERR);
}
if (!path) write_header(md_result_file, *argv);
if (opt_slave_data && do_stop_slave_sql(mysql)) goto err;
if ((opt_lock_all_tables || opt_master_data ||
(opt_single_transaction && flush_logs)) &&
do_flush_tables_read_lock(mysql))
goto err;
Flush logs before starting transaction since
this causes implicit commit starting mysql-5.5.
*/
if (opt_lock_all_tables || opt_master_data ||
(opt_single_transaction && flush_logs) || opt_delete_master_logs) {
if (flush_logs || opt_delete_master_logs) {
if (mysql_refresh(mysql, REFRESH_LOG)) {
DB_error(mysql, "when doing refresh");
goto err;
}
verbose_msg("-- main : logs flushed successfully!\n");
}
flush_logs = 0;
}
if (opt_delete_master_logs) {
if (get_bin_log_name(mysql, bin_log_name, sizeof(bin_log_name))) goto err;
}
if (opt_single_transaction && start_transaction(mysql)) goto err;
if (opt_slave_apply && add_stop_slave()) goto err;
*/
if (process_set_gtid_purged(mysql)) goto err;
if (opt_master_data && do_show_master_status(mysql)) goto err;
if (opt_slave_data && do_show_slave_status(mysql)) goto err;
if (opt_single_transaction &&
do_unlock_tables(mysql))
goto err;
mysql_query_with_error_report(mysql, 0, "set net_read_timeout = 3600");
mysql_query_with_error_report(mysql, 0, "set net_write_timeout = 3600");
sg_procMetrics->startDB();
dump_all_tables_in_db((char *)sg_mysqlInfo.database.c_str());
dump_all_views_in_db((char *)sg_mysqlInfo.database.c_str());
dump_routines_for_db((char *)sg_mysqlInfo.database.c_str());
sg_procMetrics->endDB();
if (opt_slave_data && do_start_slave_sql(mysql)) goto err;
if --set-gtid-purged, restore binlog at the end of the session
if required.
*/
set_session_binlog(TRUE);
if (opt_slave_apply && add_slave_statements()) goto err;
if (md_result_file) md_result_fd = my_fileno(md_result_file);
Ensure dumped data flushed.
First we will flush the file stream data to kernel buffers with fflush().
Second we will flush the kernel buffers data to physical disk file with
my_sync(), this will make sure the data succeessfully dumped to disk file.
fsync() fails with EINVAL if stdout is not redirected to any file, hence
MY_IGNORE_BADFD is passed to ingnore that error.
*/
if (md_result_file &&
(fflush(md_result_file) || my_sync(md_result_fd, MYF(MY_IGNORE_BADFD)))) {
if (!first_error) first_error = EX_MYSQLERR;
goto err;
}
if (opt_delete_master_logs && purge_bin_logs_to(mysql, bin_log_name))
goto err;
#if defined(_WIN32) && !defined(EMBEDDED_LIBRARY)
my_free(shared_memory_base_name);
#endif
No reason to explicitely COMMIT the transaction, neither to explicitely
UNLOCK TABLES: these will be automatically be done by the server when we
disconnect now. Saves some code here, some network trips, adds nothing to
server.
*/
err:
dbDisconnect(current_host);
if (!path) write_footer(md_result_file);
free_resources();
if (stderror_file) fclose(stderror_file);
sg_procOdbc->disConnect();
delete sg_procOdbc;
delete sg_procMetrics;
return (first_error);
}