← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to client/drizzleadmin.cc

  • Committer: Monty Taylor
  • Date: 2009-01-06 18:46:25 UTC
  • mto: This revision was merged to the branch mainline in revision 762.
  • Revision ID: mordred@inaugust.com-20090106184625-kqu7nsnwjwm5jv4s
Enabled dirty_close.

Show diffs side-by-side

added added

removed removed

Lines of Context:
client/drizzleadmin.cc
32
32
#define SHUTDOWN_DEF_TIMEOUT 3600               /* Wait for shutdown */
33
33
 
34
34
char *host= NULL, *user= NULL, *opt_password= NULL;
35
 
static bool interrupted= false, opt_verbose= false,tty_password= false; 
 
35
static bool interrupted= false, opt_verbose= false,tty_password= false;
36
36
static uint32_t tcp_port= 0, option_wait= 0, option_silent= 0;
37
37
static uint32_t my_end_arg;
38
38
static uint32_t opt_connect_timeout, opt_shutdown_timeout;
41
41
using namespace std;
42
42
 
43
43
/*
44
 
  Forward declarations 
 
44
  Forward declarations
45
45
*/
46
46
static void usage(void);
47
47
static void print_version(void);
76
76
   NO_ARG, 0, 0, 0, 0, 0, 0},
77
77
  {"host", 'h', N_("Connect to host."), (char**) &host, (char**) &host, 0, GET_STR,
78
78
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
79
 
  {"password", 'p',
 
79
  {"password", 'P',
80
80
   N_("Password to use when connecting to server. If password is not given it's asked from the tty."),
81
81
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
82
 
  {"port", 'P', N_("Port number to use for connection or 0 for default to, in "
 
82
  {"port", 'p', N_("Port number to use for connection or 0 for default to, in "
83
83
   "order of preference, drizzle.cnf, $DRIZZLE_TCP_PORT, "
84
84
   "built-in default (" STRINGIFY_ARG(DRIZZLE_PORT) ")."),
85
 
   (char**) &tcp_port,
86
 
   (char**) &tcp_port, 0, GET_UINT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 
85
   0, 0, 0, GET_UINT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
87
86
  {"silent", 's', N_("Silently exit if one can't connect to server."),
88
87
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
89
88
#ifndef DONT_ALLOW_USER_CHANGE
109
108
static const char *load_default_groups[]= { "drizzleadmin","client",0 };
110
109
 
111
110
bool
112
 
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
113
 
               char *argument)
 
111
get_one_option(int optid, const struct my_option *, char *argument)
114
112
{
 
113
  char *endchar= NULL;
 
114
  uint64_t temp_drizzle_port= 0;
115
115
  int error = 0;
116
116
 
117
117
  switch(optid) {
118
118
  case 'p':
 
119
    temp_drizzle_port= (uint64_t) strtoul(argument, &endchar, 10);
 
120
    /* if there is an alpha character this is not a valid port */
 
121
    if (strlen(endchar) != 0)
 
122
    {
 
123
      fprintf(stderr, _("Non-integer value supplied for port.  If you are trying to enter a password please use --password instead.\n"));
 
124
      exit(1);
 
125
    }
 
126
    /* If the port number is > 65535 it is not a valid port
 
127
       This also helps with potential data loss casting unsigned long to a
 
128
       uint32_t. */
 
129
    if ((temp_drizzle_port == 0) || (temp_drizzle_port > 65535))
 
130
    {
 
131
      fprintf(stderr, _("Value supplied for port is not valid.\n"));
 
132
      exit(1);
 
133
    }
 
134
    else
 
135
    {
 
136
      tcp_port= (uint32_t) temp_drizzle_port;
 
137
    }
 
138
    break;
 
139
  case 'P':
119
140
    if (argument)
120
141
    {
121
142
      char *start=argument;
122
 
      free(opt_password);
123
 
      opt_password= my_strdup(argument,MYF(MY_FAE));
124
 
      while (*argument) *argument++= 'x';   /* Destroy argument */
 
143
      if (opt_password)
 
144
        free(opt_password);
 
145
 
 
146
      opt_password= strdup(argument);
 
147
      if (opt_password == NULL)
 
148
      {
 
149
        fprintf(stderr, _("Memory allocation error while copying password. "
 
150
                          "Aborting.\n"));
 
151
        exit(ENOMEM);
 
152
      }
 
153
      while (*argument)
 
154
      {
 
155
        /* Overwriting password with 'x' */
 
156
        *argument++= 'x';
 
157
      }
125
158
      if (*start)
126
 
        start[1]=0; /* Cut length of argument */
 
159
      {
 
160
        /* Cut length of argument */
 
161
        start[1]= 0;
 
162
      }
127
163
      tty_password= 0;
128
164
    }
129
 
    else 
130
 
      tty_password= 1; 
 
165
    else
 
166
      tty_password= 1;
131
167
    break;
132
168
  case 's':
133
169
    option_silent++;
135
171
  case 'V':
136
172
    print_version();
137
173
    exit(0);
138
 
    break;
139
174
  case 'w':
140
175
    if (argument)
141
176
    {
223
258
  free(user);
224
259
  free_defaults(save_argv);
225
260
  my_end(my_end_arg);
226
 
  exit(error ? 1 : 0);
 
261
  return error ? 1 : 0;
227
262
}
228
263
 
229
 
RETSIGTYPE endprog(int signal_number __attribute__((unused)))
 
264
RETSIGTYPE endprog(int)
230
265
{
231
266
  interrupted=1;
232
267
}