set40 appears to be working

1 files changed, 11 insertions, 1 deletions

diff --git a/tools/ltgrey.c b/tools/ltgrey.c
index 7a63b04..bb43b80 100644
--- a/tools/ltgrey.c
+++ b/tools/ltgrey.c
@@ -30,6 +30,7 @@ int main(int _argc, char** _argv){
   int stain(0);
   int dns_mode(0);
   string get40_mid;
+  string set40_mid;
   while (argc > 0) {
     string arg = argv[0]; argc--; argv++;
     if (prefix(arg, "-scan")) {
@@ -46,7 +47,12 @@ int main(int _argc, char** _argv){
         exeunt(ex_syserr);
       }
       get40_mid = *argv++;  argc--;
-
+    } else if (prefix(arg, "-set40")) {
+      if (!argc){
+        cerr << "Option '" << arg << "' requires an argument" << endl;
+        exeunt(ex_syserr);
+      }
+      set40_mid = *argv++;  argc--;
     } else if (prefix(arg, "-shift")
         || prefix(arg, "-shift")) {
       if (!argc){
@@ -88,6 +94,10 @@ int main(int _argc, char** _argv){
     return 0;
   }
 
+  if (set40_mid.length()){
+    return foo.set40(set40_mid, shift);
+  }
+
   if (scanmode) {
     string dirname = parent_dir + "/quarante";
     scan(foo.progid, dirname, copies);