Commit 5ae1ff70 authored by Chuck Bell's avatar Chuck Bell

WL#5710 : Enable and disable plugins

This patch fixes line endings in the mysql_plugin.c file.
parent e73ee41c
/* Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved. /* Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved.
This program is free software; you can redistribute it and/or modify This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by it under the terms of the GNU General Public License as published by
the Free Software Foundation; version 2 of the License. the Free Software Foundation; version 2 of the License.
This program is distributed in the hope that it will be useful, This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details. GNU General Public License for more details.
You should have received a copy of the GNU General Public License You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/ */
#include <m_string.h> #include <m_string.h>
#include <mysql.h> #include <mysql.h>
#include <my_getopt.h> #include <my_getopt.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <my_global.h> #include <my_global.h>
#include <stdio.h> #include <stdio.h>
#include <string.h> #include <string.h>
#define SHOW_VERSION "1.0.0" #define SHOW_VERSION "1.0.0"
#define PRINT_VERSION do { printf("%s Ver %s Distrib %s\n", \ #define PRINT_VERSION do { printf("%s Ver %s Distrib %s\n", \
my_progname, SHOW_VERSION, MYSQL_SERVER_VERSION); \ my_progname, SHOW_VERSION, MYSQL_SERVER_VERSION); \
} while(0) } while(0)
/* Global variables. */ /* Global variables. */
static uint my_end_arg= 0; static uint my_end_arg= 0;
static uint opt_verbose=0; static uint opt_verbose=0;
static uint opt_no_defaults= 0; static uint opt_no_defaults= 0;
static uint opt_print_defaults= 0; static uint opt_print_defaults= 0;
static char *opt_datadir=0, *opt_basedir=0, static char *opt_datadir=0, *opt_basedir=0,
*opt_plugin_dir=0, *opt_plugin_ini=0; *opt_plugin_dir=0, *opt_plugin_ini=0;
static char bootstrap[FN_REFLEN]; static char bootstrap[FN_REFLEN];
/* plugin struct */ /* plugin struct */
struct st_plugin struct st_plugin
{ {
const char *name; /* plugin name */ const char *name; /* plugin name */
const char *so_name; /* plugin so (library) name */ const char *so_name; /* plugin so (library) name */
const char *components[16]; /* components to load */ const char *components[16]; /* components to load */
} plugin_data; } plugin_data;
/* Options */ /* Options */
static struct my_option my_long_options[] = static struct my_option my_long_options[] =
{ {
{"help", '?', "Display this help and exit.", 0, 0, 0, GET_NO_ARG, NO_ARG, {"help", '?', "Display this help and exit.", 0, 0, 0, GET_NO_ARG, NO_ARG,
0, 0, 0, 0, 0, 0}, 0, 0, 0, 0, 0, 0},
{"basedir", 'b', "The basedir for the server.", {"basedir", 'b', "The basedir for the server.",
0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"datadir", 'd', "The datadir for the server.", {"datadir", 'd', "The datadir for the server.",
0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"plugin-dir", 'p', "The plugin dir for the server.", {"plugin-dir", 'p', "The plugin dir for the server.",
0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"plugin-ini", 'i', "Read plugin information from configuration file " {"plugin-ini", 'i', "Read plugin information from configuration file "
"specified instead of from <plugin-dir>/<plugin_name>.ini.", "specified instead of from <plugin-dir>/<plugin_name>.ini.",
0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
{"no-defaults", 'n', "Do not read values from configuration file.", {"no-defaults", 'n', "Do not read values from configuration file.",
0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
{"print-defaults", 'P', "Show default values from configuration file.", {"print-defaults", 'P', "Show default values from configuration file.",
0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
{"verbose", 'v', {"verbose", 'v',
"More verbose output; you can use this multiple times to get even more " "More verbose output; you can use this multiple times to get even more "
"verbose output.", "verbose output.",
0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
{"version", 'V', "Output version information and exit.", 0, 0, 0, GET_NO_ARG, {"version", 'V', "Output version information and exit.", 0, 0, 0, GET_NO_ARG,
NO_ARG, 0, 0, 0, 0, 0, 0}, NO_ARG, 0, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0} {0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
}; };
/* Methods */ /* Methods */
static int process_options(int argc, char *argv[], char *operation); static int process_options(int argc, char *argv[], char *operation);
static int check_access(); static int check_access();
static int find_tool(const char *tool_name, char *tool_path); static int find_tool(const char *tool_name, char *tool_path);
static int find_plugin(char *tp_path); static int find_plugin(char *tp_path);
static int build_bootstrap_file(char *operation, char *bootstrap); static int build_bootstrap_file(char *operation, char *bootstrap);
static int dump_bootstrap_file(char *bootstrap_file); static int dump_bootstrap_file(char *bootstrap_file);
static int bootstrap_server(char *server_path, char *bootstrap_file); static int bootstrap_server(char *server_path, char *bootstrap_file);
int main(int argc,char *argv[]) int main(int argc,char *argv[])
{ {
int error= 0; int error= 0;
char tp_path[FN_REFLEN]; char tp_path[FN_REFLEN];
char server_path[FN_REFLEN]; char server_path[FN_REFLEN];
char operation[16]; char operation[16];
MY_INIT(argv[0]); MY_INIT(argv[0]);
plugin_data.name= 0; // initialize name plugin_data.name= 0; // initialize name
/* /*
The following operations comprise the method for enabling or disabling The following operations comprise the method for enabling or disabling
a plugin. We begin by processing the command options then check the a plugin. We begin by processing the command options then check the
directories specified for --datadir, --basedir, --plugin-dir, and directories specified for --datadir, --basedir, --plugin-dir, and
--plugin-ini (if specified). If the directories are Ok, we then look --plugin-ini (if specified). If the directories are Ok, we then look
for the mysqld executable and the plugin soname. Finally, we build a for the mysqld executable and the plugin soname. Finally, we build a
bootstrap command file for use in bootstraping the server. bootstrap command file for use in bootstraping the server.
If any step fails, the method issues an error message and the tool exits. If any step fails, the method issues an error message and the tool exits.
1) Parse, execute, and verify command options. 1) Parse, execute, and verify command options.
2) Check access to directories. 2) Check access to directories.
3) Look for mysqld executable. 3) Look for mysqld executable.
4) Look for the plugin. 4) Look for the plugin.
5) Build a bootstrap file with commands to enable or disable plugin. 5) Build a bootstrap file with commands to enable or disable plugin.
*/ */
if ((error= process_options(argc, argv, operation)) || if ((error= process_options(argc, argv, operation)) ||
(error= check_access()) || (error= check_access()) ||
(error= find_tool("mysqld" FN_EXEEXT, server_path)) || (error= find_tool("mysqld" FN_EXEEXT, server_path)) ||
(error= find_plugin(tp_path)) || (error= find_plugin(tp_path)) ||
(error= build_bootstrap_file(operation, bootstrap))) (error= build_bootstrap_file(operation, bootstrap)))
goto exit; goto exit;
/* Dump the bootstrap file if --verbose specified. */ /* Dump the bootstrap file if --verbose specified. */
if (opt_verbose && ((error= dump_bootstrap_file(bootstrap)))) if (opt_verbose && ((error= dump_bootstrap_file(bootstrap))))
goto exit; goto exit;
/* Start the server in bootstrap mode and execute bootstrap commands */ /* Start the server in bootstrap mode and execute bootstrap commands */
error= bootstrap_server(server_path, bootstrap); error= bootstrap_server(server_path, bootstrap);
exit: exit:
/* Remove file */ /* Remove file */
my_delete(bootstrap, MYF(0)); my_delete(bootstrap, MYF(0));
if (opt_verbose && error == 0) if (opt_verbose && error == 0)
{ {
printf("# Operation succeeded.\n"); printf("# Operation succeeded.\n");
} }
my_end(my_end_arg); my_end(my_end_arg);
exit(error ? 1 : 0); exit(error ? 1 : 0);
return 0; /* No compiler warnings */ return 0; /* No compiler warnings */
} }
/** /**
Get a temporary file name. Get a temporary file name.
@param[out] filename The file name of the temporary file @param[out] filename The file name of the temporary file
@param[in] ext An extension for the file (optional) @param[in] ext An extension for the file (optional)
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int make_tempfile(char *filename, const char *ext) static int make_tempfile(char *filename, const char *ext)
{ {
int fd= 0; int fd= 0;
if ((fd=create_temp_file(filename, NullS, ext, O_CREAT | O_WRONLY, if ((fd=create_temp_file(filename, NullS, ext, O_CREAT | O_WRONLY,
MYF(MY_WME))) < 0) MYF(MY_WME))) < 0)
{ {
fprintf(stderr, "ERROR: Cannot generate temporary file. Error code: %d.\n", fprintf(stderr, "ERROR: Cannot generate temporary file. Error code: %d.\n",
fd); fd);
return 1; return 1;
} }
my_close(fd, MYF(0)); my_close(fd, MYF(0));
return 0; return 0;
} }
/** /**
Get the value of an option from a string read from my_print_defaults output. Get the value of an option from a string read from my_print_defaults output.
@param[in] line The line (string) read from the file @param[in] line The line (string) read from the file
@param[in] item The option to search for (e.g. --datadir) @param[in] item The option to search for (e.g. --datadir)
@returns NULL if not found, string containing value if found @returns NULL if not found, string containing value if found
*/ */
static char *get_value(char *line, const char *item) static char *get_value(char *line, const char *item)
{ {
char *destination= 0; char *destination= 0;
int item_len= (int)strlen(item); int item_len= (int)strlen(item);
int line_len = (int)strlen(line); int line_len = (int)strlen(line);
if ((strncasecmp(line, item, item_len) == 0)) if ((strncasecmp(line, item, item_len) == 0))
{ {
int start= 0; int start= 0;
char *s= 0; char *s= 0;
s = line + item_len + 1; s = line + item_len + 1;
destination= my_strndup(s, line_len - start, MYF(MY_FAE)); destination= my_strndup(s, line_len - start, MYF(MY_FAE));
destination[line_len - item_len - 2]= 0; destination[line_len - item_len - 2]= 0;
} }
return destination; return destination;
} }
/** /**
Run a command in a shell. Run a command in a shell.
This function will attempt to execute the command specified by using the This function will attempt to execute the command specified by using the
popen() method to open a shell and execute the command passed and store the popen() method to open a shell and execute the command passed and store the
output in a result file. If the --verbose option was specified, it will open output in a result file. If the --verbose option was specified, it will open
the result file and print the contents to stdout. the result file and print the contents to stdout.
@param[in] cmd The command to execute. @param[in] cmd The command to execute.
@param[in] mode The mode for popen() (e.g. "r", "w", "rw") @param[in] mode The mode for popen() (e.g. "r", "w", "rw")
@return int error code or 0 for success. @return int error code or 0 for success.
*/ */
static int run_command(char* cmd, const char *mode) static int run_command(char* cmd, const char *mode)
{ {
char buf[512]= {0}; char buf[512]= {0};
FILE *res_file; FILE *res_file;
int error; int error;
if (!(res_file= popen(cmd, mode))) if (!(res_file= popen(cmd, mode)))
return -1; return -1;
if (opt_verbose) if (opt_verbose)
{ {
while (fgets(buf, sizeof(buf), res_file)) while (fgets(buf, sizeof(buf), res_file))
{ {
fprintf(stdout, "%s", buf); fprintf(stdout, "%s", buf);
} }
} }
error= pclose(res_file); error= pclose(res_file);
return error; return error;
} }
#ifdef __WIN__ #ifdef __WIN__
/** /**
Check to see if there are spaces in a path. Check to see if there are spaces in a path.
@param[in] path The Windows path to examine. @param[in] path The Windows path to examine.
@retval int spaces found = 1, no spaces = 0 @retval int spaces found = 1, no spaces = 0
*/ */
static int has_spaces(const char *path) static int has_spaces(const char *path)
{ {
if (strchr(path, ' ') != NULL) if (strchr(path, ' ') != NULL)
return 1; return 1;
return 0; return 0;
} }
/** /**
Convert a Unix path to a Windows path. Convert a Unix path to a Windows path.
@param[in] path The Windows path to examine. @param[in] path The Windows path to examine.
@returns string containing path with / changed to \\ @returns string containing path with / changed to \\
*/ */
static char *convert_path(const char *argument) static char *convert_path(const char *argument)
{ {
/* Convert / to \\ to make Windows paths */ /* Convert / to \\ to make Windows paths */
char *winfilename= my_strdup(argument, MYF(MY_FAE)); char *winfilename= my_strdup(argument, MYF(MY_FAE));
char *pos, *end; char *pos, *end;
int length= strlen(argument); int length= strlen(argument);
for (pos= winfilename, end= pos+length ; pos < end ; pos++) for (pos= winfilename, end= pos+length ; pos < end ; pos++)
{ {
if (*pos == '/') if (*pos == '/')
{ {
*pos= '\\'; *pos= '\\';
} }
} }
return winfilename; return winfilename;
} }
/** /**
Add quotes if the path has spaces in it. Add quotes if the path has spaces in it.
@param[in] path The Windows path to examine. @param[in] path The Windows path to examine.
@returns string containing excaped quotes if spaces found in path @returns string containing excaped quotes if spaces found in path
*/ */
static char *add_quotes(const char *path) static char *add_quotes(const char *path)
{ {
char windows_cmd_friendly[FN_REFLEN]; char windows_cmd_friendly[FN_REFLEN];
if (has_spaces(path)) if (has_spaces(path))
snprintf(windows_cmd_friendly, sizeof(windows_cmd_friendly), snprintf(windows_cmd_friendly, sizeof(windows_cmd_friendly),
"\"%s\"", path); "\"%s\"", path);
else else
snprintf(windows_cmd_friendly, sizeof(windows_cmd_friendly), snprintf(windows_cmd_friendly, sizeof(windows_cmd_friendly),
"%s", path); "%s", path);
return my_strdup(windows_cmd_friendly, MYF(MY_FAE)); return my_strdup(windows_cmd_friendly, MYF(MY_FAE));
} }
#endif #endif
/** /**
Get the default values from the my.cnf file. Get the default values from the my.cnf file.
This method gets the default values for the following parameters: This method gets the default values for the following parameters:
--datadir --datadir
--basedir --basedir
--plugin-dir --plugin-dir
--plugin-ini --plugin-ini
These values are used if the user has not specified a value. These values are used if the user has not specified a value.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int get_default_values() static int get_default_values()
{ {
char tool_path[FN_REFLEN]; char tool_path[FN_REFLEN];
char defaults_cmd[FN_REFLEN]; char defaults_cmd[FN_REFLEN];
char defaults_file[FN_REFLEN]; char defaults_file[FN_REFLEN];
char line[FN_REFLEN]; char line[FN_REFLEN];
int error= 0; int error= 0;
int ret= 0; int ret= 0;
FILE *file= 0; FILE *file= 0;
if ((error= find_tool("my_print_defaults" FN_EXEEXT, tool_path))) if ((error= find_tool("my_print_defaults" FN_EXEEXT, tool_path)))
goto exit; goto exit;
else else
{ {
if ((error= make_tempfile(defaults_file, "txt"))) if ((error= make_tempfile(defaults_file, "txt")))
goto exit; goto exit;
#ifdef __WIN__ #ifdef __WIN__
{ {
char *format_str= 0; char *format_str= 0;
if (has_spaces(tool_path) || has_spaces(defaults_file)) if (has_spaces(tool_path) || has_spaces(defaults_file))
format_str = "\"%s mysqld > %s\""; format_str = "\"%s mysqld > %s\"";
else else
format_str = "%s mysqld > %s"; format_str = "%s mysqld > %s";
snprintf(defaults_cmd, sizeof(defaults_cmd), format_str, snprintf(defaults_cmd, sizeof(defaults_cmd), format_str,
add_quotes(tool_path), add_quotes(defaults_file)); add_quotes(tool_path), add_quotes(defaults_file));
} }
#else #else
snprintf(defaults_cmd, sizeof(defaults_cmd), snprintf(defaults_cmd, sizeof(defaults_cmd),
"%s mysqld > %s", tool_path, defaults_file); "%s mysqld > %s", tool_path, defaults_file);
#endif #endif
/* Execute the command */ /* Execute the command */
if (opt_verbose) if (opt_verbose)
{ {
printf("# Command: %s\n", defaults_cmd); printf("# Command: %s\n", defaults_cmd);
} }
error= run_command(defaults_cmd, "r"); error= run_command(defaults_cmd, "r");
if (error) if (error)
{ {
fprintf(stderr, "ERROR: my_print_defaults failed. Error code: %d.\n", fprintf(stderr, "ERROR: my_print_defaults failed. Error code: %d.\n",
ret); ret);
goto exit; goto exit;
} }
/* Now open the file and read the defaults we want. */ /* Now open the file and read the defaults we want. */
file= fopen(defaults_file, "r"); file= fopen(defaults_file, "r");
while (fgets(line, FN_REFLEN, file) != NULL) while (fgets(line, FN_REFLEN, file) != NULL)
{ {
char *value= 0; char *value= 0;
if ((opt_datadir == 0) && ((value= get_value(line, "--datadir")))) if ((opt_datadir == 0) && ((value= get_value(line, "--datadir"))))
{ {
opt_datadir= my_strdup(value, MYF(MY_FAE)); opt_datadir= my_strdup(value, MYF(MY_FAE));
} }
if ((opt_basedir == 0) && ((value= get_value(line, "--basedir")))) if ((opt_basedir == 0) && ((value= get_value(line, "--basedir"))))
{ {
opt_basedir= my_strdup(value, MYF(MY_FAE)); opt_basedir= my_strdup(value, MYF(MY_FAE));
} }
if ((opt_plugin_dir == 0) && ((value= get_value(line, "--plugin_dir")))) if ((opt_plugin_dir == 0) && ((value= get_value(line, "--plugin_dir"))))
{ {
opt_plugin_dir= my_strdup(value, MYF(MY_FAE)); opt_plugin_dir= my_strdup(value, MYF(MY_FAE));
} }
if ((opt_plugin_ini == 0) && ((value= get_value(line, "--plugin_ini")))) if ((opt_plugin_ini == 0) && ((value= get_value(line, "--plugin_ini"))))
{ {
opt_plugin_ini= my_strdup(value, MYF(MY_FAE)); opt_plugin_ini= my_strdup(value, MYF(MY_FAE));
} }
} }
} }
exit: exit:
if (file) if (file)
{ {
fclose(file); fclose(file);
/* Remove file */ /* Remove file */
my_delete(defaults_file, MYF(0)); my_delete(defaults_file, MYF(0));
} }
return error; return error;
} }
/** /**
Print usage. Print usage.
*/ */
static void usage(void) static void usage(void)
{ {
PRINT_VERSION; PRINT_VERSION;
puts("Copyright (c) 2011, Oracle and/or its affiliates. " puts("Copyright (c) 2011, Oracle and/or its affiliates. "
"All rights reserved.\n"); "All rights reserved.\n");
puts("Enable or disable plugins."); puts("Enable or disable plugins.");
printf("\nUsage: %s [options] <plugin> ENABLE|DISABLE\n\nOptions:\n", printf("\nUsage: %s [options] <plugin> ENABLE|DISABLE\n\nOptions:\n",
my_progname); my_progname);
my_print_help(my_long_options); my_print_help(my_long_options);
puts("\n"); puts("\n");
} }
/** /**
Print the default values as read from the my.cnf file. Print the default values as read from the my.cnf file.
This method displays the default values for the following parameters: This method displays the default values for the following parameters:
--datadir --datadir
--basedir --basedir
--plugin-dir --plugin-dir
--plugin-ini --plugin-ini
*/ */
static void print_default_values(void) static void print_default_values(void)
{ {
printf("%s would have been started with the following arguments:\n", printf("%s would have been started with the following arguments:\n",
my_progname); my_progname);
get_default_values(); get_default_values();
if (opt_datadir) if (opt_datadir)
{ {
printf("--datadir=%s ", opt_datadir); printf("--datadir=%s ", opt_datadir);
} }
if (opt_basedir) if (opt_basedir)
{ {
printf("--basedir=%s ", opt_basedir); printf("--basedir=%s ", opt_basedir);
} }
if (opt_plugin_dir) if (opt_plugin_dir)
{ {
printf("--plugin_dir=%s ", opt_plugin_dir); printf("--plugin_dir=%s ", opt_plugin_dir);
} }
if (opt_plugin_ini) if (opt_plugin_ini)
{ {
printf("--plugin_ini=%s ", opt_plugin_ini); printf("--plugin_ini=%s ", opt_plugin_ini);
} }
printf("\n"); printf("\n");
} }
/** /**
Process the arguments and identify an option and store its value. Process the arguments and identify an option and store its value.
@param[in] optid The single character shortcut for the argument. @param[in] optid The single character shortcut for the argument.
@param[in] my_option Structure of legal options. @param[in] my_option Structure of legal options.
@param[in] argument The argument value to process. @param[in] argument The argument value to process.
*/ */
static my_bool static my_bool
get_one_option(int optid, get_one_option(int optid,
const struct my_option *opt __attribute__((unused)), const struct my_option *opt __attribute__((unused)),
char *argument) char *argument)
{ {
switch(optid) { switch(optid) {
case 'n': case 'n':
opt_no_defaults++; opt_no_defaults++;
break; break;
case 'P': case 'P':
opt_print_defaults++; opt_print_defaults++;
print_default_values(); print_default_values();
break; break;
case 'v': case 'v':
opt_verbose++; opt_verbose++;
break; break;
case 'V': case 'V':
PRINT_VERSION; PRINT_VERSION;
exit(0); exit(0);
break; break;
case '?': case '?':
case 'I': /* Info */ case 'I': /* Info */
usage(); usage();
exit(0); exit(0);
case 'd': case 'd':
opt_datadir= my_strdup(argument, MYF(MY_FAE)); opt_datadir= my_strdup(argument, MYF(MY_FAE));
break; break;
case 'b': case 'b':
opt_basedir= my_strdup(argument, MYF(MY_FAE)); opt_basedir= my_strdup(argument, MYF(MY_FAE));
break; break;
case 'p': case 'p':
opt_plugin_dir= my_strdup(argument, MYF(MY_FAE)); opt_plugin_dir= my_strdup(argument, MYF(MY_FAE));
break; break;
case 'i': case 'i':
opt_plugin_ini= my_strdup(argument, MYF(MY_FAE)); opt_plugin_ini= my_strdup(argument, MYF(MY_FAE));
break; break;
} }
return 0; return 0;
} }
/** /**
Check to see if a file exists. Check to see if a file exists.
@param[in] filename File to locate. @param[in] filename File to locate.
@retval int file not found = 1, file found = 0 @retval int file not found = 1, file found = 0
*/ */
static int file_exists(char * filename) static int file_exists(char * filename)
{ {
struct stat buf; struct stat buf;
int i = stat (filename, &buf); int i = stat (filename, &buf);
/* File found */ /* File found */
if (i == 0) if (i == 0)
{ {
return 1; return 1;
} }
return 0; return 0;
} }
/** /**
Search a specific path and sub directory for a file name. Search a specific path and sub directory for a file name.
@param[in] base_path Original path to use. @param[in] base_path Original path to use.
@param[in] tool_name Name of the tool to locate. @param[in] tool_name Name of the tool to locate.
@param[in] subdir The sub directory to search. @param[in] subdir The sub directory to search.
@param[out] tool_path If tool found, return complete path. @param[out] tool_path If tool found, return complete path.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int search_dir(const char * base_path, const char *tool_name, static int search_dir(const char * base_path, const char *tool_name,
const char *subdir, char *tool_path) const char *subdir, char *tool_path)
{ {
char new_path[FN_REFLEN]; char new_path[FN_REFLEN];
strcpy(new_path, base_path); strcpy(new_path, base_path);
strcat(new_path, subdir); strcat(new_path, subdir);
fn_format(new_path, new_path, "", tool_name, MY_UNPACK_FILENAME); fn_format(new_path, new_path, "", tool_name, MY_UNPACK_FILENAME);
if (file_exists(new_path)) if (file_exists(new_path))
{ {
strcpy(tool_path, new_path); strcpy(tool_path, new_path);
return 1; return 1;
} }
return 0; return 0;
} }
/** /**
Search known common paths and sub directories for a file name. Search known common paths and sub directories for a file name.
@param[in] base_path Original path to use. @param[in] base_path Original path to use.
@param[in] tool_name Name of the tool to locate. @param[in] tool_name Name of the tool to locate.
@param[out] tool_path If tool found, return complete path. @param[out] tool_path If tool found, return complete path.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int search_paths(const char *base_path, const char *tool_name, static int search_paths(const char *base_path, const char *tool_name,
char *tool_path) char *tool_path)
{ {
int i= 0; int i= 0;
static const char *paths[]= { static const char *paths[]= {
"", "/share/", "/scripts/", "/bin/", "/sbin/", "/libexec/", "", "/share/", "/scripts/", "/bin/", "/sbin/", "/libexec/",
"/mysql/", "/sql/", "/mysql/", "/sql/",
}; };
for (i = 0 ; i < (int)array_elements(paths); i++) for (i = 0 ; i < (int)array_elements(paths); i++)
{ {
if (search_dir(base_path, tool_name, paths[i], tool_path)) if (search_dir(base_path, tool_name, paths[i], tool_path))
{ {
return 1; return 1;
} }
} }
return 0; return 0;
} }
/** /**
Read the plugin ini file. Read the plugin ini file.
This function attempts to read the plugin config file from the plugin_dir This function attempts to read the plugin config file from the plugin_dir
path saving the data in the the st_plugin structure. If the file is not path saving the data in the the st_plugin structure. If the file is not
found or the file cannot be read, an error is generated. found or the file cannot be read, an error is generated.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int load_plugin_data(char *plugin_name, char *config_file) static int load_plugin_data(char *plugin_name, char *config_file)
{ {
FILE *file_ptr; FILE *file_ptr;
char path[FN_REFLEN]; char path[FN_REFLEN];
char line[1024]; char line[1024];
char *reason= 0; char *reason= 0;
char *res; char *res;
int i= -1; int i= -1;
if (opt_plugin_ini == 0) if (opt_plugin_ini == 0)
{ {
fn_format(path, config_file, opt_plugin_dir, "", MYF(0)); fn_format(path, config_file, opt_plugin_dir, "", MYF(0));
opt_plugin_ini= my_strdup(path, MYF(MY_FAE)); opt_plugin_ini= my_strdup(path, MYF(MY_FAE));
} }
if (!file_exists(opt_plugin_ini)) if (!file_exists(opt_plugin_ini))
{ {
reason= (char *)"File does not exist."; reason= (char *)"File does not exist.";
goto error; goto error;
} }
file_ptr= fopen(opt_plugin_ini, "r"); file_ptr= fopen(opt_plugin_ini, "r");
if (file_ptr == NULL) if (file_ptr == NULL)
{ {
reason= (char *)"Cannot open file."; reason= (char *)"Cannot open file.";
goto error; goto error;
} }
/* save name */ /* save name */
plugin_data.name= my_strdup(plugin_name, MYF(MY_WME)); plugin_data.name= my_strdup(plugin_name, MYF(MY_WME));
/* Read plugin components */ /* Read plugin components */
while (i < 16) while (i < 16)
{ {
res= fgets(line, sizeof(line), file_ptr); res= fgets(line, sizeof(line), file_ptr);
/* strip /n */ /* strip /n */
if (line[strlen(line)-1] == '\n') if (line[strlen(line)-1] == '\n')
{ {
line[strlen(line)-1]= '\0'; line[strlen(line)-1]= '\0';
} }
if (res == NULL) if (res == NULL)
{ {
if (i < 1) if (i < 1)
{ {
reason= (char *)"Bad format in plugin configuration file."; reason= (char *)"Bad format in plugin configuration file.";
fclose(file_ptr); fclose(file_ptr);
goto error; goto error;
} }
break; break;
} }
if ((line[0] == '#') || (line[0] == '\n')) // skip comment and blank lines if ((line[0] == '#') || (line[0] == '\n')) // skip comment and blank lines
{ {
continue; continue;
} }
if (i == -1) // if first pass, read this line as so_name if (i == -1) // if first pass, read this line as so_name
{ {
/* save so_name */ /* save so_name */
plugin_data.so_name= my_strdup(line, MYF(MY_WME)); plugin_data.so_name= my_strdup(line, MYF(MY_WME));
/* Add proper file extension for soname */ /* Add proper file extension for soname */
strcat((char *)plugin_data.so_name, FN_SOEXT); strcat((char *)plugin_data.so_name, FN_SOEXT);
i++; i++;
} }
else else
{ {
if (strlen(line) > 0) if (strlen(line) > 0)
{ {
plugin_data.components[i]= my_strdup(line, MYF(MY_WME)); plugin_data.components[i]= my_strdup(line, MYF(MY_WME));
i++; i++;
} }
else else
{ {
plugin_data.components[i]= NULL; plugin_data.components[i]= NULL;
} }
} }
} }
fclose(file_ptr); fclose(file_ptr);
return 0; return 0;
error: error:
fprintf(stderr, "ERROR: Cannot read plugin config file %s. %s\n", fprintf(stderr, "ERROR: Cannot read plugin config file %s. %s\n",
plugin_name, reason); plugin_name, reason);
return 1; return 1;
} }
/** /**
Check the options for validity. Check the options for validity.
This function checks the arguments for validity issuing the appropriate This function checks the arguments for validity issuing the appropriate
error message if arguments are missing or invalid. On success, @operation error message if arguments are missing or invalid. On success, @operation
is set to either "ENABLE" or "DISABLE". is set to either "ENABLE" or "DISABLE".
@param[in] argc The number of arguments. @param[in] argc The number of arguments.
@param[in] argv The arguments. @param[in] argv The arguments.
@param[out] operation The operation chosen (enable|disable) @param[out] operation The operation chosen (enable|disable)
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int check_options(int argc, char **argv, char *operation) static int check_options(int argc, char **argv, char *operation)
{ {
int i= 0; // loop counter int i= 0; // loop counter
int num_found= 0; // number of options found (shortcut loop) int num_found= 0; // number of options found (shortcut loop)
char config_file[FN_REFLEN]; // configuration file name char config_file[FN_REFLEN]; // configuration file name
char plugin_name[FN_REFLEN]; // plugin name char plugin_name[FN_REFLEN]; // plugin name
/* Form prefix strings for the options. */ /* Form prefix strings for the options. */
const char *basedir_prefix = "--basedir="; const char *basedir_prefix = "--basedir=";
int basedir_len= strlen(basedir_prefix); int basedir_len= strlen(basedir_prefix);
const char *datadir_prefix = "--datadir="; const char *datadir_prefix = "--datadir=";
int datadir_len= strlen(datadir_prefix); int datadir_len= strlen(datadir_prefix);
const char *plugin_dir_prefix = "--plugin_dir="; const char *plugin_dir_prefix = "--plugin_dir=";
int plugin_dir_len= strlen(plugin_dir_prefix); int plugin_dir_len= strlen(plugin_dir_prefix);
strcpy(plugin_name, ""); strcpy(plugin_name, "");
for (i = 0; i < argc && num_found < 5; i++) for (i = 0; i < argc && num_found < 5; i++)
{ {
if (!argv[i]) if (!argv[i])
{ {
continue; continue;
} }
if ((strcasecmp(argv[i], "ENABLE") == 0) || if ((strcasecmp(argv[i], "ENABLE") == 0) ||
(strcasecmp(argv[i], "DISABLE") == 0)) (strcasecmp(argv[i], "DISABLE") == 0))
{ {
strcpy(operation, argv[i]); strcpy(operation, argv[i]);
num_found++; num_found++;
} }
else if ((strncasecmp(argv[i], basedir_prefix, basedir_len) == 0) && else if ((strncasecmp(argv[i], basedir_prefix, basedir_len) == 0) &&
!opt_basedir) !opt_basedir)
{ {
opt_basedir= my_strndup(argv[i]+basedir_len, opt_basedir= my_strndup(argv[i]+basedir_len,
strlen(argv[i])-basedir_len, MYF(MY_FAE)); strlen(argv[i])-basedir_len, MYF(MY_FAE));
num_found++; num_found++;
} }
else if ((strncasecmp(argv[i], datadir_prefix, datadir_len) == 0) && else if ((strncasecmp(argv[i], datadir_prefix, datadir_len) == 0) &&
!opt_datadir) !opt_datadir)
{ {
opt_datadir= my_strndup(argv[i]+datadir_len, opt_datadir= my_strndup(argv[i]+datadir_len,
strlen(argv[i])-datadir_len, MYF(MY_FAE)); strlen(argv[i])-datadir_len, MYF(MY_FAE));
num_found++; num_found++;
} }
else if ((strncasecmp(argv[i], plugin_dir_prefix, plugin_dir_len) == 0) && else if ((strncasecmp(argv[i], plugin_dir_prefix, plugin_dir_len) == 0) &&
!opt_plugin_dir) !opt_plugin_dir)
{ {
opt_plugin_dir= my_strndup(argv[i]+plugin_dir_len, opt_plugin_dir= my_strndup(argv[i]+plugin_dir_len,
strlen(argv[i])-plugin_dir_len, MYF(MY_FAE)); strlen(argv[i])-plugin_dir_len, MYF(MY_FAE));
num_found++; num_found++;
} }
/* read the plugin config file and check for match against argument */ /* read the plugin config file and check for match against argument */
else else
{ {
strcpy(plugin_name, argv[i]); strcpy(plugin_name, argv[i]);
strcpy(config_file, argv[i]); strcpy(config_file, argv[i]);
strcat(config_file, ".ini"); strcat(config_file, ".ini");
} }
} }
if (!opt_basedir) if (!opt_basedir)
{ {
fprintf(stderr, "ERROR: Missing --basedir option.\n"); fprintf(stderr, "ERROR: Missing --basedir option.\n");
return 1; return 1;
} }
if (!opt_datadir) if (!opt_datadir)
{ {
fprintf(stderr, "ERROR: Missing --datadir option.\n"); fprintf(stderr, "ERROR: Missing --datadir option.\n");
return 1; return 1;
} }
if (!opt_plugin_dir) if (!opt_plugin_dir)
{ {
fprintf(stderr, "ERROR: Missing --plugin_dir option.\n"); fprintf(stderr, "ERROR: Missing --plugin_dir option.\n");
return 1; return 1;
} }
/* If a plugin was specified, read the config file. */ /* If a plugin was specified, read the config file. */
else if (strlen(plugin_name) > 0) else if (strlen(plugin_name) > 0)
{ {
if (load_plugin_data(plugin_name, config_file)) if (load_plugin_data(plugin_name, config_file))
{ {
return 1; return 1;
} }
if (strcasecmp(plugin_data.name, plugin_name) != 0) if (strcasecmp(plugin_data.name, plugin_name) != 0)
{ {
fprintf(stderr, "ERROR: plugin name requested does not match config " fprintf(stderr, "ERROR: plugin name requested does not match config "
"file data.\n"); "file data.\n");
return 1; return 1;
} }
} }
else else
{ {
fprintf(stderr, "ERROR: No plugin specified.\n"); fprintf(stderr, "ERROR: No plugin specified.\n");
return 1; return 1;
} }
if ((strlen(operation) == 0)) if ((strlen(operation) == 0))
{ {
fprintf(stderr, "ERROR: missing operation. Please specify either " fprintf(stderr, "ERROR: missing operation. Please specify either "
"'<plugin> ENABLE' or '<plugin> DISABLE'.\n"); "'<plugin> ENABLE' or '<plugin> DISABLE'.\n");
return 1; return 1;
} }
return 0; return 0;
} }
/** /**
Parse, execute, and verify command options. Parse, execute, and verify command options.
This method handles all of the option processing including the optional This method handles all of the option processing including the optional
features for displaying data (--print-defaults, --help ,etc.) that do not features for displaying data (--print-defaults, --help ,etc.) that do not
result in an attempt to ENABLE or DISABLE of a plugin. result in an attempt to ENABLE or DISABLE of a plugin.
@param[in] arc Count of arguments @param[in] arc Count of arguments
@param[in] argv Array of arguments @param[in] argv Array of arguments
@param[out] operation Operation (ENABLE or DISABLE) @param[out] operation Operation (ENABLE or DISABLE)
@retval int error = 1, success = 0, exit program = -1 @retval int error = 1, success = 0, exit program = -1
*/ */
static int process_options(int argc, char *argv[], char *operation) static int process_options(int argc, char *argv[], char *operation)
{ {
int error= 0; int error= 0;
int i= 0; int i= 0;
/* Parse and execute command-line options */ /* Parse and execute command-line options */
if ((error= handle_options(&argc, &argv, my_long_options, get_one_option))) if ((error= handle_options(&argc, &argv, my_long_options, get_one_option)))
goto exit; goto exit;
/* If the print defaults option used, exit. */ /* If the print defaults option used, exit. */
if (opt_print_defaults) if (opt_print_defaults)
{ {
error= -1; error= -1;
goto exit; goto exit;
} }
/* Add a trailing directory separator if not present */ /* Add a trailing directory separator if not present */
if (opt_basedir) if (opt_basedir)
{ {
i= (int)strlength(opt_basedir); i= (int)strlength(opt_basedir);
if (opt_basedir[i-1] != FN_LIBCHAR || opt_basedir[i-1] != FN_LIBCHAR2) if (opt_basedir[i-1] != FN_LIBCHAR || opt_basedir[i-1] != FN_LIBCHAR2)
#ifdef __WIN__ #ifdef __WIN__
if (opt_basedir[i-1] != '/') if (opt_basedir[i-1] != '/')
strcat(opt_basedir, "//"); strcat(opt_basedir, "//");
#else #else
strcat(opt_basedir, FN_DIRSEP); strcat(opt_basedir, FN_DIRSEP);
#endif #endif
} }
/* /*
If the user did not specify the option to skip loading defaults from a If the user did not specify the option to skip loading defaults from a
config file and the required options are not present or there was an error config file and the required options are not present or there was an error
generated when the defaults were read from the file, exit. generated when the defaults were read from the file, exit.
*/ */
if (!opt_no_defaults && ((error= get_default_values()))) if (!opt_no_defaults && ((error= get_default_values())))
{ {
error= -1; error= -1;
goto exit; goto exit;
} }
/* /*
Check to ensure required options are present and validate the operation. Check to ensure required options are present and validate the operation.
Note: this method also validates the plugin specified by attempting to Note: this method also validates the plugin specified by attempting to
read a configuration file named <plugin_name>.ini from the --plugin-dir read a configuration file named <plugin_name>.ini from the --plugin-dir
or --plugin-ini location if the --plugin-ini option presented. or --plugin-ini location if the --plugin-ini option presented.
*/ */
strcpy(operation, ""); strcpy(operation, "");
if ((error = check_options(argc, argv, operation))) if ((error = check_options(argc, argv, operation)))
{ {
goto exit; goto exit;
} }
if (opt_verbose) if (opt_verbose)
{ {
printf("# basedir = %s\n", opt_basedir); printf("# basedir = %s\n", opt_basedir);
printf("# plugin_dir = %s\n", opt_plugin_dir); printf("# plugin_dir = %s\n", opt_plugin_dir);
printf("# datadir = %s\n", opt_datadir); printf("# datadir = %s\n", opt_datadir);
printf("# plugin_ini = %s\n", opt_plugin_ini); printf("# plugin_ini = %s\n", opt_plugin_ini);
} }
exit: exit:
return error; return error;
} }
/** /**
Check access Check access
This method checks to ensure all of the directories (opt_basedir, This method checks to ensure all of the directories (opt_basedir,
opt_plugin_dir, opt_datadir, and opt_plugin_ini) are accessible by opt_plugin_dir, opt_datadir, and opt_plugin_ini) are accessible by
the user. the user.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int check_access() static int check_access()
{ {
int error= 0; int error= 0;
if ((error= my_access(opt_basedir, F_OK))) if ((error= my_access(opt_basedir, F_OK)))
{ {
fprintf(stderr, "ERROR: Cannot access basedir at '%s'.\n", fprintf(stderr, "ERROR: Cannot access basedir at '%s'.\n",
opt_basedir); opt_basedir);
goto exit; goto exit;
} }
if ((error= my_access(opt_plugin_dir, F_OK))) if ((error= my_access(opt_plugin_dir, F_OK)))
{ {
fprintf(stderr, "ERROR: Cannot access plugin_dir at '%s'.\n", fprintf(stderr, "ERROR: Cannot access plugin_dir at '%s'.\n",
opt_plugin_dir); opt_plugin_dir);
goto exit; goto exit;
} }
if ((error= my_access(opt_datadir, F_OK))) if ((error= my_access(opt_datadir, F_OK)))
{ {
fprintf(stderr, "ERROR: Cannot access datadir at '%s'.\n", fprintf(stderr, "ERROR: Cannot access datadir at '%s'.\n",
opt_datadir); opt_datadir);
goto exit; goto exit;
} }
if ((error= my_access(opt_plugin_ini, F_OK))) if ((error= my_access(opt_plugin_ini, F_OK)))
{ {
fprintf(stderr, "ERROR: Cannot access plugin config file at '%s'.\n", fprintf(stderr, "ERROR: Cannot access plugin config file at '%s'.\n",
opt_plugin_ini); opt_plugin_ini);
goto exit; goto exit;
} }
exit: exit:
return error; return error;
} }
/** /**
Locate the tool and form tool path. Locate the tool and form tool path.
@param[in] tool_name Name of the tool to locate. @param[in] tool_name Name of the tool to locate.
@param[out] tool_path If tool found, return complete path. @param[out] tool_path If tool found, return complete path.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int find_tool(const char *tool_name, char *tool_path) static int find_tool(const char *tool_name, char *tool_path)
{ {
int i= 0; int i= 0;
const char *paths[]= { const char *paths[]= {
opt_basedir, "/usr", "/usr/local/mysql", "/usr/sbin", "/usr/share", opt_basedir, "/usr", "/usr/local/mysql", "/usr/sbin", "/usr/share",
}; };
for (i= 0; i < (int)array_elements(paths); i++) for (i= 0; i < (int)array_elements(paths); i++)
{ {
if (paths[i] && (search_paths(paths[i], tool_name, tool_path))) if (paths[i] && (search_paths(paths[i], tool_name, tool_path)))
goto found; goto found;
} }
fprintf(stderr, "WARNING: Cannot find %s.\n", tool_name); fprintf(stderr, "WARNING: Cannot find %s.\n", tool_name);
return 1; return 1;
found: found:
if (opt_verbose) if (opt_verbose)
printf("# Found tool '%s' as '%s'.\n", tool_name, tool_path); printf("# Found tool '%s' as '%s'.\n", tool_name, tool_path);
return 0; return 0;
} }
/** /**
Find the plugin library. Find the plugin library.
This function attempts to use the @c plugin_dir option passed on the This function attempts to use the @c plugin_dir option passed on the
command line to locate the plugin. command line to locate the plugin.
@param[out] tp_path The actual path to plugin with FN_SOEXT applied. @param[out] tp_path The actual path to plugin with FN_SOEXT applied.
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int find_plugin(char *tp_path) static int find_plugin(char *tp_path)
{ {
/* Check for existance of plugin */ /* Check for existance of plugin */
fn_format(tp_path, plugin_data.so_name, opt_plugin_dir, "", MYF(0)); fn_format(tp_path, plugin_data.so_name, opt_plugin_dir, "", MYF(0));
if (!file_exists(tp_path)) if (!file_exists(tp_path))
{ {
fprintf(stderr, "ERROR: The plugin library is missing or in a different" fprintf(stderr, "ERROR: The plugin library is missing or in a different"
" location.\n"); " location.\n");
return 1; return 1;
} }
else if (opt_verbose) else if (opt_verbose)
{ {
printf("# Found plugin '%s' as '%s'\n", plugin_data.name, tp_path); printf("# Found plugin '%s' as '%s'\n", plugin_data.name, tp_path);
} }
return 0; return 0;
} }
/** /**
Build the boostrap file. Build the boostrap file.
Create a new file and populate it with SQL commands to ENABLE or DISABLE Create a new file and populate it with SQL commands to ENABLE or DISABLE
the plugin via INSERT and DELETE operations on the mysql.plugin table. the plugin via INSERT and DELETE operations on the mysql.plugin table.
param[in] operation The type of operation (ENABLE or DISABLE) param[in] operation The type of operation (ENABLE or DISABLE)
param[out] bootstrap A FILE* pointer param[out] bootstrap A FILE* pointer
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int build_bootstrap_file(char *operation, char *bootstrap) static int build_bootstrap_file(char *operation, char *bootstrap)
{ {
int error= 0; int error= 0;
FILE *file= 0; FILE *file= 0;
/* /*
Perform plugin operation : ENABLE or DISABLE Perform plugin operation : ENABLE or DISABLE
The following creates a temporary bootstrap file and populates it with The following creates a temporary bootstrap file and populates it with
the appropriate SQL commands for the operation. For ENABLE, INSERT the appropriate SQL commands for the operation. For ENABLE, INSERT
statements are created. For DISABLE, DELETE statements are created. The statements are created. For DISABLE, DELETE statements are created. The
values for these statements are derived from the plugin_data read from the values for these statements are derived from the plugin_data read from the
<plugin_name>.ini configuration file. Once the file is built, a call to <plugin_name>.ini configuration file. Once the file is built, a call to
mysqld is made in read only, bootstrap modes to read the SQL statements mysqld is made in read only, bootstrap modes to read the SQL statements
and execute them. and execute them.
*/ */
if ((error= make_tempfile(bootstrap, "sql"))) if ((error= make_tempfile(bootstrap, "sql")))
{ {
/* Fail if we cannot create a temporary file for the bootstrap commands. */ /* Fail if we cannot create a temporary file for the bootstrap commands. */
fprintf(stderr, "ERROR: Cannot create bootstrap file.\n"); fprintf(stderr, "ERROR: Cannot create bootstrap file.\n");
goto exit; goto exit;
} }
if ((file= fopen(bootstrap, "w+")) == NULL) if ((file= fopen(bootstrap, "w+")) == NULL)
{ {
fprintf(stderr, "ERROR: Cannot open bootstrap file for writing.\n"); fprintf(stderr, "ERROR: Cannot open bootstrap file for writing.\n");
error= 1; error= 1;
goto exit; goto exit;
} }
if (strcasecmp(operation, "enable") == 0) if (strcasecmp(operation, "enable") == 0)
{ {
int i= 0; int i= 0;
fprintf(file, "INSERT IGNORE INTO mysql.plugin VALUES "); fprintf(file, "INSERT IGNORE INTO mysql.plugin VALUES ");
for (i= 0; i < (int)array_elements(plugin_data.components); i++) for (i= 0; i < (int)array_elements(plugin_data.components); i++)
{ {
/* stop when we read the end of the symbol list - marked with NULL */ /* stop when we read the end of the symbol list - marked with NULL */
if (plugin_data.components[i] == NULL) if (plugin_data.components[i] == NULL)
{ {
break; break;
} }
if (i > 0) if (i > 0)
{ {
fprintf(file, ", "); fprintf(file, ", ");
} }
fprintf(file, "('%s','%s')", fprintf(file, "('%s','%s')",
plugin_data.components[i], plugin_data.so_name); plugin_data.components[i], plugin_data.so_name);
} }
fprintf(file, ";\n"); fprintf(file, ";\n");
if (opt_verbose) if (opt_verbose)
{ {
printf("# Enabling %s...\n", plugin_data.name); printf("# Enabling %s...\n", plugin_data.name);
} }
} }
else else
{ {
fprintf(file, fprintf(file,
"DELETE FROM mysql.plugin WHERE name = '%s';", plugin_data.name); "DELETE FROM mysql.plugin WHERE name = '%s';", plugin_data.name);
if (opt_verbose) if (opt_verbose)
{ {
printf("# Disabling %s...\n", plugin_data.name); printf("# Disabling %s...\n", plugin_data.name);
} }
} }
exit: exit:
fclose(file); fclose(file);
return error; return error;
} }
/** /**
Dump bootstrap file. Dump bootstrap file.
Read the contents of the bootstrap file and print it out. Read the contents of the bootstrap file and print it out.
@param[in] bootstrap_file Name of bootstrap file to read @param[in] bootstrap_file Name of bootstrap file to read
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int dump_bootstrap_file(char *bootstrap_file) static int dump_bootstrap_file(char *bootstrap_file)
{ {
char *ret= 0; char *ret= 0;
int error= 0; int error= 0;
char query_str[512]; char query_str[512];
FILE *file= 0; FILE *file= 0;
if ((file= fopen(bootstrap_file, "r")) == NULL) if ((file= fopen(bootstrap_file, "r")) == NULL)
{ {
fprintf(stderr, "ERROR: Cannot open bootstrap file for reading.\n"); fprintf(stderr, "ERROR: Cannot open bootstrap file for reading.\n");
error= 1; error= 1;
goto exit; goto exit;
} }
ret= fgets(query_str, 512, file); ret= fgets(query_str, 512, file);
if (ret == 0) if (ret == 0)
{ {
fprintf(stderr, "ERROR: Cannot read bootstrap file.\n"); fprintf(stderr, "ERROR: Cannot read bootstrap file.\n");
error= 1; error= 1;
goto exit; goto exit;
} }
printf("# Query: %s", query_str); printf("# Query: %s", query_str);
exit: exit:
if (file) if (file)
{ {
fclose(file); fclose(file);
} }
return error; return error;
} }
/** /**
Bootstrap the server Bootstrap the server
Create a command line sequence to launch mysqld in bootstrap mode. This Create a command line sequence to launch mysqld in bootstrap mode. This
will allow mysqld to launch a minimal server instance to read and will allow mysqld to launch a minimal server instance to read and
execute SQL commands from a file piped in (the boostrap file). We use execute SQL commands from a file piped in (the boostrap file). We use
the --no-defaults option to skip reading values from the config file. the --no-defaults option to skip reading values from the config file.
The bootstrap mode skips loading of plugins and many other subsystems. The bootstrap mode skips loading of plugins and many other subsystems.
This allows the mysql_plugin tool to insert the correct rows into the This allows the mysql_plugin tool to insert the correct rows into the
mysql.plugin table (for ENABLE) or delete the rows (for DISABLE). Once mysql.plugin table (for ENABLE) or delete the rows (for DISABLE). Once
the server is launched in normal mode, the plugin will be loaded the server is launched in normal mode, the plugin will be loaded
(for ENABLE) or not loaded (for DISABLE). In this way, we avoid the (for ENABLE) or not loaded (for DISABLE). In this way, we avoid the
(sometimes) complicated LOAD PLUGIN commands. (sometimes) complicated LOAD PLUGIN commands.
@param[in] server_path Path to server executable @param[in] server_path Path to server executable
@param[in] bootstrap_file Name of bootstrap file to read @param[in] bootstrap_file Name of bootstrap file to read
@retval int error = 1, success = 0 @retval int error = 1, success = 0
*/ */
static int bootstrap_server(char *server_path, char *bootstrap_file) static int bootstrap_server(char *server_path, char *bootstrap_file)
{ {
char bootstrap_cmd[FN_REFLEN]; char bootstrap_cmd[FN_REFLEN];
int error= 0; int error= 0;
int ret= 0; int ret= 0;
#ifdef __WIN__ #ifdef __WIN__
char *format_str= 0; char *format_str= 0;
char *verbose_str= ""; char *verbose_str= "";
if (opt_verbose) if (opt_verbose)
strcat(verbose_str, "--console"); strcat(verbose_str, "--console");
if (has_spaces(opt_datadir) || has_spaces(opt_basedir) || if (has_spaces(opt_datadir) || has_spaces(opt_basedir) ||
has_spaces(bootstrap_file)) has_spaces(bootstrap_file))
format_str= "\"%s %s --bootstrap --datadir=%s --basedir=%s < %s\""; format_str= "\"%s %s --bootstrap --datadir=%s --basedir=%s < %s\"";
else else
format_str= "%s %s --bootstrap --datadir=%s --basedir=%s < %s"; format_str= "%s %s --bootstrap --datadir=%s --basedir=%s < %s";
snprintf(bootstrap_cmd, sizeof(bootstrap_cmd), format_str, snprintf(bootstrap_cmd, sizeof(bootstrap_cmd), format_str,
add_quotes(convert_path(server_path)), verbose_str, add_quotes(convert_path(server_path)), verbose_str,
add_quotes(opt_datadir), add_quotes(opt_basedir), add_quotes(opt_datadir), add_quotes(opt_basedir),
add_quotes(bootstrap_file)); add_quotes(bootstrap_file));
#else #else
snprintf(bootstrap_cmd, sizeof(bootstrap_cmd), snprintf(bootstrap_cmd, sizeof(bootstrap_cmd),
"%s --no-defaults --bootstrap --datadir=%s --basedir=%s" "%s --no-defaults --bootstrap --datadir=%s --basedir=%s"
" < %s", server_path, opt_datadir, opt_basedir, bootstrap_file); " < %s", server_path, opt_datadir, opt_basedir, bootstrap_file);
#endif #endif
/* Execute the command */ /* Execute the command */
if (opt_verbose) if (opt_verbose)
{ {
printf("# Command: %s\n", bootstrap_cmd); printf("# Command: %s\n", bootstrap_cmd);
} }
error= run_command(bootstrap_cmd, "r"); error= run_command(bootstrap_cmd, "r");
if (error) if (error)
fprintf(stderr, fprintf(stderr,
"ERROR: Unexpected result from bootstrap. Error code: %d.\n", "ERROR: Unexpected result from bootstrap. Error code: %d.\n",
ret); ret);
return error; return error;
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment