← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/archive/archive_reader.cc

  • Committer: Stewart Smith
  • Date: 2010-11-03 03:27:09 UTC
  • mto: (1902.1.1 build) (1910.1.2 build)
  • mto: This revision was merged to the branch mainline in revision 1903.
  • Revision ID: stewart@flamingspork.com-20101103032709-oyvfrc6eb8fzj0mr
fix docs warning: docs/unlock.rst:2: (WARNING/2) Title underline too short.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/archive/archive_reader.cc
19
19
 */
20
20
 
21
21
#include "config.h"
22
 
 
 
22
#include <iostream>
 
23
#include <string>
 
24
#include <fstream>
 
25
#include <drizzled/configmake.h>
 
26
using namespace std;
 
27
#include <boost/program_options.hpp>
 
28
namespace po= boost::program_options;
23
29
#include "azio.h"
24
30
#include <string.h>
25
31
#include <assert.h>
28
34
#include <fcntl.h>
29
35
#include "drizzled/charset_info.h"
30
36
#include "drizzled/internal/m_string.h"
31
 
#include "drizzled/option.h"
32
37
 
33
38
#define SHOW_VERSION "0.1"
34
39
 
35
40
using namespace drizzled;
36
41
 
37
 
int get_one_option(int optid, const struct option *opt, char *argument);
38
 
 
39
 
static void get_options(int *argc,char * * *argv);
40
 
static void print_version(void);
41
 
static void usage(void);
42
 
static const char *opt_tmpdir;
43
 
static const char *new_auto_increment;
 
42
string opt_tmpdir;
44
43
uint64_t new_auto_increment_value;
45
 
static const char *load_default_groups[]= { "archive_reader", 0 };
46
 
static char **default_argv;
47
 
int opt_check, opt_force, opt_quiet, opt_backup= 0, opt_extract_table_message;
48
 
int opt_autoincrement;
 
44
bool opt_check, 
 
45
  opt_force,
 
46
  opt_backup, 
 
47
  opt_extract_table_message, 
 
48
  opt_silent,
 
49
  opt_quiet,
 
50
  opt_quick,
 
51
  opt_autoincrement;
49
52
 
50
53
int main(int argc, char *argv[])
51
54
{
 
55
try
 
56
{
 
57
  po::options_description commandline_options("Options used only in command line");
 
58
  commandline_options.add_options()
 
59
  ("force,f",po::value<bool>(&opt_force)->default_value(false)->zero_tokens(),
 
60
  "Restart with -r if there are any errors in the table")
 
61
  ("help,?","Display this help and exit")
 
62
  ("version,V","Print version and exit")
 
63
  ("no-defaults", po::value<bool>()->default_value(false)->zero_tokens(),
 
64
  "Configuration file defaults are not used if no-defaults is set")
 
65
  ;
 
66
 
 
67
  po::options_description archive_reader_options("Options specific to the archive reader");
 
68
  archive_reader_options.add_options()
 
69
  ("tmpdir,t",po::value<string>(&opt_tmpdir)->default_value(""),
 
70
  "Path for temporary files.") 
 
71
  ("set-auto-increment,A",po::value<uint64_t>(&new_auto_increment_value)->default_value(0),
 
72
  "Force auto_increment to start at this or higher value. If no value is given, then sets the next auto_increment value to the highest used value for the auto key + 1.")
 
73
  ("silent,s",po::value<bool>(&opt_silent)->default_value(false)->zero_tokens(),
 
74
  "Only print errors. One can use two -s to make archive_reader very silent.")
 
75
  ("quick,q",po::value<bool>(&opt_quick)->default_value(false)->zero_tokens(),
 
76
  "Faster repair but not modifying the data")
 
77
  ("repair,r",po::value<bool>(&opt_quick)->default_value(false)->zero_tokens(),
 
78
  "Repair a damaged Archive version 3 or above file.")
 
79
  ("back-up,b",po::value<bool>(&opt_backup)->default_value(false)->zero_tokens(),
 
80
  "Make a backup of an archive table.")
 
81
  ("check,c",po::value<bool>(&opt_check)->default_value(false)->zero_tokens(),
 
82
  "Check table for errors")
 
83
  ("extract-table-message,e",po::value<bool>(&opt_extract_table_message)->default_value(false)->zero_tokens(),
 
84
  "Extract the table protobuf message.")
 
85
  ;
 
86
 
52
87
  unsigned int ret;
53
88
  azio_stream reader_handle;
54
89
 
55
 
  internal::my_init();
56
 
  MY_INIT(argv[0]);
57
 
  get_options(&argc, &argv);
58
 
 
 
90
  std::string system_config_dir_archive_reader(SYSCONFDIR); 
 
91
  system_config_dir_archive_reader.append("/drizzle/archive_reader.cnf");
 
92
 
 
93
  std::string user_config_dir((getenv("XDG_CONFIG_HOME")? getenv("XDG_CONFIG_HOME"):"~/.config"));
 
94
  
 
95
  po::options_description long_options("Allowed Options");
 
96
  long_options.add(commandline_options).add(archive_reader_options);
 
97
 
 
98
  po::variables_map vm;
 
99
  po::store(po::parse_command_line(argc,argv,long_options),vm);
 
100
 
 
101
  if (! vm["no-defaults"].as<bool>())
 
102
  {
 
103
    std::string user_config_dir_archive_reader(user_config_dir);
 
104
    user_config_dir_archive_reader.append("/drizzle/archive_reader.cnf");
 
105
  
 
106
    ifstream user_archive_reader_ifs(user_config_dir_archive_reader.c_str());
 
107
    po::store(parse_config_file(user_archive_reader_ifs, archive_reader_options), vm);
 
108
 
 
109
    ifstream system_archive_reader_ifs(system_config_dir_archive_reader.c_str());
 
110
    store(parse_config_file(system_archive_reader_ifs, archive_reader_options), vm);
 
111
 
 
112
  }
 
113
  po::notify(vm);
 
114
 
 
115
  if (vm.count("force") || vm.count("quiet") || vm.count("tmpdir"))
 
116
    cout << "Not implemented yet";
 
117
 
 
118
  if (vm.count("version"))
 
119
  {
 
120
    printf("%s  Ver %s, for %s-%s (%s)\n", internal::my_progname, SHOW_VERSION,
 
121
        HOST_VENDOR, HOST_OS, HOST_CPU);
 
122
    exit(0);
 
123
  }
 
124
  
 
125
  if (vm.count("set-auto-increment"))
 
126
  {
 
127
    opt_autoincrement= true;
 
128
  }
 
129
 
 
130
  if (vm.count("help") || argc == 0)
 
131
  {
 
132
    printf("%s  Ver %s, for %s-%s (%s)\n", internal::my_progname, SHOW_VERSION,
 
133
        HOST_VENDOR, HOST_OS, HOST_CPU);
 
134
    puts("This software comes with ABSOLUTELY NO WARRANTY. This is free "
 
135
         "software,\n"
 
136
         "and you are welcome to modify and redistribute it under the GPL "
 
137
         "license\n");
 
138
    puts("Read and modify Archive files directly\n");
 
139
    printf("Usage: %s [OPTIONS] file_to_be_looked_at [file_for_backup]\n", internal::my_progname);
 
140
    cout << long_options << endl;    
 
141
    exit(0); 
 
142
  }
 
143
  
59
144
  if (argc < 1)
60
145
  {
61
146
    printf("No file specified. \n");
62
 
    return 0;
 
147
    return -1;
63
148
  }
64
149
 
65
150
  if (!(ret= azopen(&reader_handle, argv[0], O_RDONLY, AZ_METHOD_BLOCK)))
66
151
  {
67
152
    printf("Could not open Archive file\n");
68
 
    return 0;
 
153
    return -1;
69
154
  }
70
155
 
71
156
  if (opt_autoincrement)
255
340
  azclose(&reader_handle);
256
341
 
257
342
  internal::my_end();
258
 
  return 0;
259
 
}
260
 
 
261
 
int get_one_option(int optid, const struct option *opt, char *argument)
262
 
{
263
 
  (void)opt;
264
 
  switch (optid) {
265
 
  case 'b':
266
 
    opt_backup= 1;
267
 
    break;
268
 
  case 'c':
269
 
    opt_check= 1;
270
 
    break;
271
 
  case 'e':
272
 
    opt_extract_table_message= 1;
273
 
    break;
274
 
  case 'f':
275
 
    opt_force= 1;
276
 
    printf("Not implemented yet\n");
277
 
    break;
278
 
  case 'q':
279
 
    opt_quiet= 1;
280
 
    printf("Not implemented yet\n");
281
 
    break;
282
 
  case 'V':
283
 
    print_version();
284
 
    exit(0);
285
 
  case 't':
286
 
    printf("Not implemented yet\n");
287
 
    break;
288
 
  case 'A':
289
 
    opt_autoincrement= 1;
290
 
    if (argument)
291
 
      new_auto_increment_value= strtoull(argument, NULL, 0);
292
 
    else
293
 
      new_auto_increment_value= 0;
294
 
    break;
295
 
  case '?':
296
 
    usage();
297
 
    exit(0);
298
 
  }
299
 
  return 0;
300
 
}
301
 
 
302
 
static struct option my_long_options[] =
303
 
{
304
 
  {"backup", 'b',
305
 
   "Make a backup of an archive table.",
306
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
307
 
  {"check", 'c', "Check table for errors.",
308
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
309
 
  {"extract-table-message", 'e',
310
 
   "Extract the table protobuf message.",
311
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
312
 
  {"force", 'f',
313
 
   "Restart with -r if there are any errors in the table.",
314
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
315
 
  {"help", '?',
316
 
   "Display this help and exit.",
317
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
318
 
  {"quick", 'q', "Faster repair by not modifying the data file.",
319
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
320
 
  {"repair", 'r', "Repair a damaged Archive version 3 or above file.",
321
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
322
 
  {"set-auto-increment", 'A',
323
 
   "Force auto_increment to start at this or higher value. If no value is given, then sets the next auto_increment value to the highest used value for the auto key + 1.",
324
 
   (char**) &new_auto_increment,
325
 
   (char**) &new_auto_increment,
326
 
   0, GET_ULL, OPT_ARG, 0, 0, 0, 0, 0, 0},
327
 
  {"silent", 's',
328
 
   "Only print errors. One can use two -s to make archive_reader very silent.",
329
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
330
 
  {"tmpdir", 't',
331
 
   "Path for temporary files.",
332
 
   (char**) &opt_tmpdir,
333
 
   0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
334
 
  {"version", 'V',
335
 
   "Print version and exit.",
336
 
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
337
 
  { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
338
 
};
339
 
 
340
 
static void usage(void)
341
 
{
342
 
  print_version();
343
 
  puts("Copyright (C) 2007 MySQL AB");
344
 
  puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,\
345
 
       \nand you are welcome to modify and redistribute it under the GPL \
346
 
       license\n");
347
 
  puts("Read and modify Archive files directly\n");
348
 
  printf("Usage: %s [OPTIONS] file_to_be_looked_at [file_for_backup]\n", internal::my_progname);
349
 
  internal::print_defaults("drizzle", load_default_groups);
350
 
  my_print_help(my_long_options);
351
 
}
352
 
 
353
 
static void print_version(void)
354
 
{
355
 
  printf("%s  Ver %s, for %s-%s (%s)\n", internal::my_progname, SHOW_VERSION,
356
 
         HOST_VENDOR, HOST_OS, HOST_CPU);
357
 
}
358
 
 
359
 
static void get_options(int *argc, char ***argv)
360
 
{
361
 
  internal::load_defaults("drizzle", load_default_groups, argc, argv);
362
 
  default_argv= *argv;
363
 
 
364
 
  handle_options(argc, argv, my_long_options, get_one_option);
365
 
 
366
 
  if (*argc == 0)
 
343
 
 
344
}
 
345
  catch(exception &e)
367
346
  {
368
 
    usage();
369
 
    exit(-1);
 
347
    cerr<<"Error"<<e.what()<<endl;
370
348
  }
 
349
  return 0;
 
350
}
371
351
 
372
 
  return;
373
 
} /* get options */