← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/archive/archive_reader.cc

  • Committer: Brian Aker
  • Date: 2009-12-08 23:37:35 UTC
  • mfrom: (1192.3.82 pandora-build)
  • mto: This revision was merged to the branch mainline in revision 1241.
  • Revision ID: brian@gaz-20091208233735-6zfhecyxizlw8bub
Merge range fix

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/archive/archive_reader.cc
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2006 MySQL AB
5
 
 *  Copyright (C) 2009 Sun Microsystems, Inc.
 
4
 *  Copyright (c) 2006 MySQL AB
 
5
 *  Copyright (c) 2009 Sun Microsystems, Inc.
6
6
 *
7
7
 *  This program is free software; you can redistribute it and/or modify
8
8
 *  it under the terms of the GNU General Public License as published by
18
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
19
 */
20
20
 
21
 
#include "config.h"
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
 
#include <boost/scoped_ptr.hpp>
29
 
namespace po= boost::program_options;
 
21
#include "drizzled/global.h"
 
22
 
30
23
#include "azio.h"
31
24
#include <string.h>
32
25
#include <assert.h>
33
26
#include <stdio.h>
34
27
#include <stdarg.h>
35
 
#include <fcntl.h>
36
 
#include <memory>
37
 
#include "drizzled/charset_info.h"
38
 
#include "drizzled/internal/m_string.h"
 
28
#include <mystrings/m_ctype.h>
 
29
#include <mystrings/m_string.h>
 
30
#include <mysys/my_getopt.h>
39
31
 
40
32
#define SHOW_VERSION "0.1"
41
33
 
42
 
using namespace drizzled;
 
34
extern "C" bool
 
35
get_one_option(int optid, const struct my_option *opt, char *argument);
43
36
 
44
 
string opt_tmpdir;
 
37
static void get_options(int *argc,char * * *argv);
 
38
static void print_version(void);
 
39
static void usage(void);
 
40
static const char *opt_tmpdir;
 
41
static const char *new_auto_increment;
45
42
uint64_t new_auto_increment_value;
46
 
bool opt_check, 
47
 
  opt_force,
48
 
  opt_backup, 
49
 
  opt_extract_table_message, 
50
 
  opt_silent,
51
 
  opt_quiet,
52
 
  opt_quick,
53
 
  opt_autoincrement;
 
43
static const char *load_default_groups[]= { "archive_reader", 0 };
 
44
static char **default_argv;
 
45
int opt_check, opt_force, opt_quiet, opt_backup= 0, opt_extract_frm;
 
46
int opt_autoincrement;
54
47
 
55
48
int main(int argc, char *argv[])
56
49
{
57
 
try
58
 
{
59
 
  po::options_description commandline_options("Options used only in command line");
60
 
  commandline_options.add_options()
61
 
  ("force,f",po::value<bool>(&opt_force)->default_value(false)->zero_tokens(),
62
 
  "Restart with -r if there are any errors in the table")
63
 
  ("help,?","Display this help and exit")
64
 
  ("version,V","Print version and exit")
65
 
  ("no-defaults", po::value<bool>()->default_value(false)->zero_tokens(),
66
 
  "Configuration file defaults are not used if no-defaults is set")
67
 
  ;
68
 
 
69
 
  po::options_description archive_reader_options("Options specific to the archive reader");
70
 
  archive_reader_options.add_options()
71
 
  ("tmpdir,t",po::value<string>(&opt_tmpdir)->default_value(""),
72
 
  "Path for temporary files.") 
73
 
  ("set-auto-increment,A",po::value<uint64_t>(&new_auto_increment_value)->default_value(0),
74
 
  "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.")
75
 
  ("silent,s",po::value<bool>(&opt_silent)->default_value(false)->zero_tokens(),
76
 
  "Only print errors. One can use two -s to make archive_reader very silent.")
77
 
  ("quick,q",po::value<bool>(&opt_quick)->default_value(false)->zero_tokens(),
78
 
  "Faster repair but not modifying the data")
79
 
  ("repair,r",po::value<bool>(&opt_quick)->default_value(false)->zero_tokens(),
80
 
  "Repair a damaged Archive version 3 or above file.")
81
 
  ("back-up,b",po::value<bool>(&opt_backup)->default_value(false)->zero_tokens(),
82
 
  "Make a backup of an archive table.")
83
 
  ("check,c",po::value<bool>(&opt_check)->default_value(false)->zero_tokens(),
84
 
  "Check table for errors")
85
 
  ("extract-table-message,e",po::value<bool>(&opt_extract_table_message)->default_value(false)->zero_tokens(),
86
 
  "Extract the table protobuf message.")
87
 
  ;
88
 
 
89
50
  unsigned int ret;
90
 
  boost::scoped_ptr<azio_stream> reader_handle_ap(new azio_stream);
91
 
  azio_stream &reader_handle= *reader_handle_ap.get();
92
 
 
93
 
  std::string system_config_dir_archive_reader(SYSCONFDIR); 
94
 
  system_config_dir_archive_reader.append("/drizzle/archive_reader.cnf");
95
 
 
96
 
  std::string user_config_dir((getenv("XDG_CONFIG_HOME")? getenv("XDG_CONFIG_HOME"):"~/.config"));
97
 
 
98
 
  if (user_config_dir.compare(0, 2, "~/") == 0)
99
 
  {
100
 
    char *homedir;
101
 
    homedir= getenv("HOME");
102
 
    if (homedir != NULL)
103
 
      user_config_dir.replace(0, 1, homedir);
104
 
  }
105
 
  
106
 
  po::options_description long_options("Allowed Options");
107
 
  long_options.add(commandline_options).add(archive_reader_options);
108
 
 
109
 
  po::variables_map vm;
110
 
  po::store(po::parse_command_line(argc,argv,long_options),vm);
111
 
 
112
 
  if (! vm["no-defaults"].as<bool>())
113
 
  {
114
 
    std::string user_config_dir_archive_reader(user_config_dir);
115
 
    user_config_dir_archive_reader.append("/drizzle/archive_reader.cnf");
116
 
  
117
 
    ifstream user_archive_reader_ifs(user_config_dir_archive_reader.c_str());
118
 
    po::store(parse_config_file(user_archive_reader_ifs, archive_reader_options), vm);
119
 
 
120
 
    ifstream system_archive_reader_ifs(system_config_dir_archive_reader.c_str());
121
 
    store(parse_config_file(system_archive_reader_ifs, archive_reader_options), vm);
122
 
 
123
 
  }
124
 
  po::notify(vm);
125
 
 
126
 
  if (vm.count("force") || vm.count("quiet") || vm.count("tmpdir"))
127
 
    cout << "Not implemented yet";
128
 
 
129
 
  if (vm.count("version"))
130
 
  {
131
 
    printf("%s  Ver %s, for %s-%s (%s)\n", internal::my_progname, SHOW_VERSION,
132
 
        HOST_VENDOR, HOST_OS, HOST_CPU);
133
 
    exit(0);
134
 
  }
135
 
  
136
 
  if (vm.count("set-auto-increment"))
137
 
  {
138
 
    opt_autoincrement= true;
139
 
  }
140
 
 
141
 
  if (vm.count("help") || argc == 0)
142
 
  {
143
 
    printf("%s  Ver %s, for %s-%s (%s)\n", internal::my_progname, SHOW_VERSION,
144
 
        HOST_VENDOR, HOST_OS, HOST_CPU);
145
 
    puts("This software comes with ABSOLUTELY NO WARRANTY. This is free "
146
 
         "software,\n"
147
 
         "and you are welcome to modify and redistribute it under the GPL "
148
 
         "license\n");
149
 
    puts("Read and modify Archive files directly\n");
150
 
    printf("Usage: %s [OPTIONS] file_to_be_looked_at [file_for_backup]\n", internal::my_progname);
151
 
    cout << long_options << endl;    
152
 
    exit(0); 
153
 
  }
154
 
  
 
51
  azio_stream reader_handle;
 
52
 
 
53
  my_init();
 
54
  MY_INIT(argv[0]);
 
55
  get_options(&argc, &argv);
 
56
 
155
57
  if (argc < 1)
156
58
  {
157
59
    printf("No file specified. \n");
158
 
    return -1;
 
60
    return 0;
159
61
  }
160
62
 
161
63
  if (!(ret= azopen(&reader_handle, argv[0], O_RDONLY, AZ_METHOD_BLOCK)))
162
64
  {
163
65
    printf("Could not open Archive file\n");
164
 
    return -1;
 
66
    return 0;
165
67
  }
166
68
 
167
69
  if (opt_autoincrement)
168
70
  {
169
 
    boost::scoped_ptr<azio_stream> writer_handle_ap(new azio_stream);
170
 
    azio_stream &writer_handle= *writer_handle_ap.get();
 
71
    azio_stream writer_handle;
171
72
 
172
73
    if (new_auto_increment_value)
173
74
    {
208
109
    printf("\tLongest Row %u\n", reader_handle.longest_row);
209
110
    printf("\tShortest Row %u\n", reader_handle.shortest_row);
210
111
    printf("\tState %s\n", ( reader_handle.dirty ? "dirty" : "clean"));
211
 
    printf("\tTable protobuf message stored at %u\n",
212
 
           reader_handle.frm_start_pos);
 
112
    printf("\tFRM stored at %u\n", reader_handle.frm_start_pos);
213
113
    printf("\tComment stored at %u\n", reader_handle.comment_start_pos);
214
114
    printf("\tData starts at %u\n", (unsigned int)reader_handle.start);
215
115
    if (reader_handle.frm_start_pos)
216
 
      printf("\tTable proto message length %u\n", reader_handle.frm_length);
 
116
      printf("\tFRM length %u\n", reader_handle.frm_length);
217
117
    if (reader_handle.comment_start_pos)
218
118
    {
219
119
      char *comment =
264
164
    uint64_t row_count= 0;
265
165
    char *buffer;
266
166
 
267
 
    boost::scoped_ptr<azio_stream> writer_handle_ap(new azio_stream);
268
 
    azio_stream &writer_handle= *writer_handle_ap.get();
 
167
    azio_stream writer_handle;
269
168
 
270
169
    buffer= (char *)malloc(reader_handle.longest_row);
271
170
    if (buffer == NULL)
331
230
    azclose(&writer_handle);
332
231
  }
333
232
 
334
 
  if (opt_extract_table_message)
 
233
  if (opt_extract_frm)
335
234
  {
336
 
    int frm_file;
 
235
    File frm_file;
337
236
    char *ptr;
338
 
    frm_file= internal::my_open(argv[1], O_CREAT|O_RDWR, MYF(0));
 
237
    frm_file= my_open(argv[1], O_CREAT|O_RDWR, MYF(0));
339
238
    ptr= (char *)malloc(sizeof(char) * reader_handle.frm_length);
340
239
    if (ptr == NULL)
341
240
    {
343
242
      goto end;
344
243
    }
345
244
    azread_frm(&reader_handle, ptr);
346
 
    internal::my_write(frm_file, (unsigned char*) ptr, reader_handle.frm_length, MYF(0));
347
 
    internal::my_close(frm_file, MYF(0));
 
245
    my_write(frm_file, (unsigned char*) ptr, reader_handle.frm_length, MYF(0));
 
246
    my_close(frm_file, MYF(0));
348
247
    free(ptr);
349
248
  }
350
249
 
352
251
  printf("\n");
353
252
  azclose(&reader_handle);
354
253
 
355
 
  internal::my_end();
356
 
 
357
 
}
358
 
  catch(exception &e)
 
254
  my_end();
 
255
  return 0;
 
256
}
 
257
 
 
258
bool get_one_option(int optid, const struct my_option *opt, char *argument)
 
259
{
 
260
  (void)opt;
 
261
  switch (optid) {
 
262
  case 'b':
 
263
    opt_backup= 1;
 
264
    break;
 
265
  case 'c':
 
266
    opt_check= 1;
 
267
    break;
 
268
  case 'e':
 
269
    opt_extract_frm= 1;
 
270
    break;
 
271
  case 'f':
 
272
    opt_force= 1;
 
273
    printf("Not implemented yet\n");
 
274
    break;
 
275
  case 'q':
 
276
    opt_quiet= 1;
 
277
    printf("Not implemented yet\n");
 
278
    break;
 
279
  case 'V':
 
280
    print_version();
 
281
    exit(0);
 
282
  case 't':
 
283
    printf("Not implemented yet\n");
 
284
    break;
 
285
  case 'A':
 
286
    opt_autoincrement= 1;
 
287
    if (argument)
 
288
      new_auto_increment_value= strtoull(argument, NULL, 0);
 
289
    else
 
290
      new_auto_increment_value= 0;
 
291
    break;
 
292
  case '?':
 
293
    usage();
 
294
    exit(0);
 
295
  }
 
296
  return 0;
 
297
}
 
298
 
 
299
static struct my_option my_long_options[] =
 
300
{
 
301
  {"backup", 'b',
 
302
   "Make a backup of an archive table.",
 
303
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
304
  {"check", 'c', "Check table for errors.",
 
305
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
306
  {"extract-frm", 'e',
 
307
   "Extract the frm file.",
 
308
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
309
  {"force", 'f',
 
310
   "Restart with -r if there are any errors in the table.",
 
311
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
312
  {"help", '?',
 
313
   "Display this help and exit.",
 
314
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
315
  {"quick", 'q', "Faster repair by not modifying the data file.",
 
316
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
317
  {"repair", 'r', "Repair a damaged Archive version 3 or above file.",
 
318
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
319
  {"set-auto-increment", 'A',
 
320
   "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.",
 
321
   (char**) &new_auto_increment,
 
322
   (char**) &new_auto_increment,
 
323
   0, GET_ULL, OPT_ARG, 0, 0, 0, 0, 0, 0},
 
324
  {"silent", 's',
 
325
   "Only print errors. One can use two -s to make archive_reader very silent.",
 
326
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
327
  {"tmpdir", 't',
 
328
   "Path for temporary files.",
 
329
   (char**) &opt_tmpdir,
 
330
   0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 
331
  {"version", 'V',
 
332
   "Print version and exit.",
 
333
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 
334
  { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
 
335
};
 
336
 
 
337
static void usage(void)
 
338
{
 
339
  print_version();
 
340
  puts("Copyright (C) 2007 MySQL AB");
 
341
  puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,\
 
342
       \nand you are welcome to modify and redistribute it under the GPL \
 
343
       license\n");
 
344
  puts("Read and modify Archive files directly\n");
 
345
  printf("Usage: %s [OPTIONS] file_to_be_looked_at [file_for_backup]\n", my_progname);
 
346
  print_defaults("drizzle", load_default_groups);
 
347
  my_print_help(my_long_options);
 
348
}
 
349
 
 
350
static void print_version(void)
 
351
{
 
352
  printf("%s  Ver %s, for %s-%s (%s)\n", my_progname, SHOW_VERSION,
 
353
         HOST_VENDOR, HOST_OS, HOST_CPU);
 
354
}
 
355
 
 
356
static void get_options(int *argc, char ***argv)
 
357
{
 
358
  load_defaults("drizzle", load_default_groups, argc, argv);
 
359
  default_argv= *argv;
 
360
 
 
361
  handle_options(argc, argv, my_long_options, get_one_option);
 
362
 
 
363
  if (*argc == 0)
359
364
  {
360
 
    cerr<<"Error"<<e.what()<<endl;
 
365
    usage();
 
366
    exit(-1);
361
367
  }
362
 
  return 0;
363
 
}
364
368
 
 
369
  return;
 
370
} /* get options */