mysqlmanagerc.c 4.74 KB
Newer Older
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
1
/* Copyright (C) 2000 MySQL AB
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

17
#define MANAGER_CLIENT_VERSION "1.4"
18

19
#include <my_global.h>
20 21 22
#include <mysql.h>
#include <mysql_version.h>
#include <mysqld_error.h>
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi committed
23 24
#include <my_sys.h>
#include <m_string.h>
25
#include <my_getopt.h>
26 27 28 29 30
#include <stdarg.h>
#include <sys/stat.h>
#include <unistd.h>

#ifndef MYSQL_MANAGER_PORT
31
#define MYSQL_MANAGER_PORT 9305
32 33 34 35 36 37
#endif

static void die(const char* fmt, ...);

const char* user="root",*host="localhost";
char* pass=0;
38
my_bool quiet=0;
39 40 41 42 43 44
uint port=MYSQL_MANAGER_PORT;
static const char *load_default_groups[]= { "mysqlmanagerc",0 };
char** default_argv;
MYSQL_MANAGER *manager;
FILE* fp, *fp_out;

45
static struct my_option my_long_options[] =
46
{
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
  {"host", 'h', "Connect to host.", (gptr*) &host, (gptr*) &host, 0,
   GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"user", 'u', "User for login.", (gptr*) &user, (gptr*) &user, 0,
   GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"password", 'p', "Password to use when connecting to server.",
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
  {"port", 'P', "Port number to use for connection.", (gptr*) &port,
   (gptr*) &port, 0, GET_UINT, REQUIRED_ARG, MYSQL_MANAGER_PORT, 0, 0, 0, 0,
   0},
  {"help", '?', "Display this help and exit.", 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, NO_ARG, 0, 0, 0, 0, 0, 0},
  {"quiet", 'q', "Suppress all normal output.", (gptr*) &quiet, (gptr*) &quiet,
   0, GET_BOOL, 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}
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
};

static void die(const char* fmt, ...)
{
  va_list args;
  DBUG_ENTER("die");
  va_start(args, fmt);
  if (fmt)
  {
    fprintf(stderr, "%s: ", my_progname);
    vfprintf(stderr, fmt, args);
    fprintf(stderr, "\n");
    fflush(stderr);
  }
  va_end(args);
  exit(1);
}

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

void usage()
{
  print_version();
91
  printf("MySQL AB, by Sasha\n");
92
  printf("This software comes with ABSOLUTELY NO WARRANTY\n\n");
93
  printf("Command-line client for MySQL manager daemon.\n\n");
94
  printf("Usage: %s [OPTIONS] < command_file\n", my_progname);
95
  my_print_help(my_long_options);
jani@hynda.(none)'s avatar
jani@hynda.(none) committed
96
  printf("  --no-defaults         Don't read default options from any options file.\n");
97
  my_print_variables(my_long_options);
98
}
99 100 101 102 103


static my_bool
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
	       char *argument)
104 105
{
  my_bool tty_password=0;
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
  
  switch (optid) {
  case 'p':
    if (argument)
    {
      my_free(pass, MYF(MY_ALLOW_ZERO_PTR));
      pass= my_strdup(argument, MYF(MY_FAE));
      while (*argument) *argument++= 'x';	/* Destroy argument */
    }
    else
      tty_password=1;
    break;
  case 'V':
    print_version();
    exit(0);
  case '?':
    usage();
    exit(0);				
  }
  return 0;
}


int parse_args(int argc, char **argv)
{
  int ho_error;
132 133 134

  load_defaults("my",load_default_groups,&argc,&argv);
  default_argv= argv;
135

rburnett@build.mysql.com's avatar
rburnett@build.mysql.com committed
136
  if ((ho_error=handle_options(&argc, &argv, my_long_options, get_one_option, NULL)))
137 138
    exit(ho_error);

139
  return 0;
140
}
141 142


143 144 145 146 147 148 149 150 151
int main(int argc, char** argv)
{
  MY_INIT(argv[0]);
  fp=stdin;
  fp_out=stdout;
  parse_args(argc,argv);
  if (!(manager=mysql_manager_init(0)))
    die("Failed in mysql_manager_init()");
  if (!mysql_manager_connect(manager,host,user,pass,port))
152
    die("Could not connect to MySQL manager: %s (%d)",manager->last_error,
153 154 155
	manager->last_errno);
  for (;!feof(fp);)
  {
156
    char buf[4096];
157 158
    if (!fgets(buf,sizeof(buf),fp))
      break;
159 160
    if (!quiet)
      fprintf(fp_out,"<<%s",buf);
161
    if (mysql_manager_command(manager,buf,strlen(buf)))
162
      die("Error in command: %s (%d)",manager->last_error,manager->last_errno);
163 164 165
    while (!manager->eof)
    {
      if (mysql_manager_fetch_line(manager,buf,sizeof(buf)))
166
	die("Error fetching result line: %s (%d)", manager->last_error,
167
	    manager->last_errno);
168 169
      if (!quiet)
        fprintf(fp_out,">>%s\n",buf);
170 171 172 173 174
    }
  }
  mysql_manager_close(manager);
  return 0;
}