diff options
| -rw-r--r-- | tools/ltgrey.c | 138 |
1 files changed, 89 insertions, 49 deletions
diff --git a/tools/ltgrey.c b/tools/ltgrey.c index 75c8438..577cbe9 100644 --- a/tools/ltgrey.c +++ b/tools/ltgrey.c @@ -4,6 +4,7 @@ #include "libltgrey.h" #include "utils.h" #include "qq_exit_codes.h" +#include <list> using namespace std; pid_t mypid; @@ -11,6 +12,68 @@ string progname; #define exeunt exit +typedef const char cc; +typedef cc* str; + +class argParser{ +public: + list<str> argv; + string current_arg; + + argParser(const int _argc, const str _argv[]) + { + for (int ii=0; ii < _argc; ii++) { + argv.push_back(_argv[ii]); + } + } + string shift() { + string rslt = argv.front(); + argv.pop_front(); + return rslt; + } + void next(){ + current_arg = shift(); + } + size_t size() const { + return argv.size(); + } + int prefix(const string longer, const size_t required = 0){ + if (argv.size() < required) { + if (required==1) + cerr << "Option '" << current_arg + << "' requires an argument." << endl; + else + cerr << "Option '" << current_arg + << "' requires " << required << " arguments." << endl; + exeunt(ex_syserr); + } + return current_arg == longer.substr(0, current_arg.length()); + } +}; + +void usage(const string parent_dir){ + cout << +"Usage: ltgrey [options]\n" +"\n" +"Options include\n" +" -scan40 # scan the quarantine directory\n" +" -scanrep # scan the reputation directory\n" +" -copy # no idea\n" +" -verbose # increase the verbosity\n" +" -setrep mid # set the reputation for the given message-id\n" +" -getrep mid # look up the reputation\n" +" -shift ??? # no idea\n" +" -stain ??? # not implemented\n" +" -suffix ??? # no idea\n" +" -help # print this msg (and exit immediately)\n" +"\n" +; + + cout << "parent dir is " << parent_dir << endl; + cout << "box_40.dir is " << box_40.dir << endl; + cout << "box_rep.dir is " << box_rep.dir << endl; +} + int main(int _argc, char** _argv){ mypid = getpid(); @@ -39,64 +102,41 @@ int main(int _argc, char** _argv){ string set_40_mid; string get_rep_mid; string set_rep_mid; - while (argc > 0) { - string arg = argv[0]; argc--; argv++; + argParser ARGS(argc, argv); + + while (ARGS.size()) { + ARGS.next(); if (0){ - } else if (prefix(arg, "-scan40")) { + } else if (ARGS.prefix("-help")) { + usage(foo.parent_dir); + exit(0); + } else if (ARGS.prefix("-scan40")) { scan40mode++; - } else if (prefix(arg, "-scanrep")) { + } else if (ARGS.prefix("-scanrep")) { scanrepmode++; - } else if (prefix(arg, "-copy")) { + } else if (ARGS.prefix("-copy")) { copies++; - } else if (prefix(arg, "-verbose")) { + } else if (ARGS.prefix("-verbose")) { foo.verbosity++; - } else if (prefix(arg, "-dns_mode")) { + } else if (ARGS.prefix("-dns_mode")) { dns_mode++; - } else if (prefix(arg, "-get40")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } - get_40_mid = *argv++; argc--; - } else if (prefix(arg, "-set40")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } - set_40_mid = *argv++; argc--; - } else if (prefix(arg, "-setrep")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } - set_rep_mid = *argv++; argc--; - } else if (prefix(arg, "-getrep")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } - get_rep_mid = *argv++; argc--; - } else if (prefix(arg, "-shift") - || prefix(arg, "-shift")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } + } else if (ARGS.prefix("-get40", 1)) { + get_40_mid = ARGS.shift(); + } else if (ARGS.prefix("-set40", 1)) { + set_40_mid = ARGS.shift(); + } else if (ARGS.prefix("-setrep", 1)) { + set_rep_mid = ARGS.shift(); + } else if (ARGS.prefix("-getrep", 1)) { + get_rep_mid = ARGS.shift(); + } else if (ARGS.prefix("-shift"), 1) { shift = atoi(*argv++); argc--; - } else if (prefix(arg, "-stain")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } + } else if (ARGS.prefix("-stain", 1)) { stain = atoi(*argv++); argc--; - } else if (prefix(arg, "-suffix")) { - if (!argc){ - cerr << "Option '" << arg << "' requires an argument" << endl; - exeunt(ex_syserr); - } - foo.suffix += *argv++; argc--; + if (stain) {} /* FIXME */ + } else if (ARGS.prefix("-suffix", 1)) { + foo.suffix += ARGS.shift(); } else { - cerr << "Unrecognized arg: " << arg << endl; + cerr << "Unrecognized arg: " << ARGS.current_arg << endl; exeunt(ex_syserr); } } |