mysqlmanager-pwgen.c 3.85 KB
Newer Older
unknown's avatar
unknown committed
1
/* Copyright (C) 2000 MySQL AB
unknown's avatar
unknown committed
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_PWGEN_VERSION "1.4"
unknown's avatar
unknown committed
18 19 20 21 22 23 24

#include <my_global.h>
#include <m_ctype.h>
#include <my_sys.h>
#include <m_string.h>
#include <mysql_version.h>
#include <errno.h>
25
#include <my_getopt.h>
unknown's avatar
unknown committed
26 27 28 29
#include <md5.h>

const char* outfile=0,*user="root";

30
static struct my_option my_long_options[] =
unknown's avatar
unknown committed
31
{
32 33 34 35 36 37 38 39 40 41
  {"output-file", 'o', "Write the output to the file with the given name",
   (gptr*) &outfile, (gptr*) &outfile, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0,
   0, 0},
  {"user", 'u', "Put given user in the password file", (gptr*) &user,
   (gptr*) &user, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  {"help", '?', "Display this message and exit", 0, 0, 0, GET_NO_ARG, NO_ARG,
   0, 0, 0, 0, 0, 0},
  {"version", 'V', "Display version info", 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}
unknown's avatar
unknown committed
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
};

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_PWGEN_VERSION,
	 MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);
}

void usage()
{
  print_version();
  printf("MySQL AB, by Sasha\n");
  printf("This software comes with ABSOLUTELY NO WARRANTY\n\n");
  printf("Generates a password file to be used by mysqltest.\n\n");
  printf("Usage: %s [OPTIONS]\n", my_progname);
74 75
  my_print_help(my_long_options);
  my_print_variables(my_long_options);
unknown's avatar
unknown committed
76 77
}

78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
static my_bool
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
	       char *argument __attribute__((unused)))
{
  switch (optid) {
  case '?':
    usage();
    exit(0);
  case 'V':
    print_version();
    exit(0);
  }
  return 0;
}


unknown's avatar
unknown committed
94 95
int parse_args(int argc, char** argv)
{
96 97
  int ho_error;

unknown's avatar
unknown committed
98
  if ((ho_error=handle_options(&argc, &argv, my_long_options, get_one_option, NULL)))
99 100
    exit(ho_error);

unknown's avatar
unknown committed
101 102 103 104 105 106 107
  return 0;
}

void get_pass(char* pw, int len)
{
  FILE* fp;
  char* pw_end=pw+len;
108 109 110 111
  /*
    /dev/random is more secure than  rand() because the seed is easy to
    predict, so we resort to rand() only if /dev/random is not available
  */
unknown's avatar
unknown committed
112 113 114 115 116 117
  if ((fp=fopen("/dev/random","r")))
  {
    fread(pw,len,1,fp);
    fclose(fp);
    while (pw<pw_end)
    {
unknown's avatar
unknown committed
118 119
      char tmp= 'a'+((uint)*pw % 26);
      *pw++= tmp;
unknown's avatar
unknown committed
120 121 122 123 124 125
    }
  }
  else
  {
    srand(time(NULL));
    while (pw<pw_end)
unknown's avatar
unknown committed
126 127 128 129
    {
      char tmp= 'a'+((uint)*pw % 26);
      *pw++= tmp;
    }
unknown's avatar
unknown committed
130 131 132 133
  }
  *pw_end=0;
}

134

unknown's avatar
unknown committed
135 136 137 138 139 140 141
int main(int argc, char** argv)
{
  FILE* fp;
  my_MD5_CTX context;
  uchar digest[16];
  char pw[17];
  uint i;
unknown's avatar
unknown committed
142

unknown's avatar
unknown committed
143 144 145 146
  MY_INIT(argv[0]);
  parse_args(argc,argv);
  if (!outfile)
    die("Missing --output-file");
unknown's avatar
unknown committed
147

unknown's avatar
unknown committed
148 149 150 151
  if (!(fp=fopen(outfile,"w")))
    die("Could not open '%s'(errno=%d)",outfile,errno);
  get_pass(pw,sizeof(pw)-1);
  my_MD5Init(&context);
unknown's avatar
unknown committed
152
  my_MD5Update(&context,(uchar*) pw,sizeof(pw)-1);
unknown's avatar
unknown committed
153 154 155 156 157 158 159 160 161
  my_MD5Final(digest,&context);
  fprintf(fp,"%s:",user);
  for (i=0;i<sizeof(digest);i++)
    fprintf(fp,"%02x",digest[i]);
  fprintf(fp,"\n");
  fclose(fp);
  printf("%s\n",pw);
  return 0;
}