From abcd53ddce872e3f331d8a6d7c1ff44c070b91b0 Mon Sep 17 00:00:00 2001
From: John Denker <jsd@av8n.com>
Date: Fri, 20 Jul 2012 09:20:36 -0700
Subject: greylist will now zap its program-group

---
 tools/greylist.c    | 46 +++++++++++++++++++++++++++++++++++-----------
 tools/hi-test2.conf |  2 +-
 tools/makefile      |  2 +-
 3 files changed, 37 insertions(+), 13 deletions(-)

(limited to 'tools')

diff --git a/tools/greylist.c b/tools/greylist.c
index 525fc9b..910b40c 100644
--- a/tools/greylist.c
+++ b/tools/greylist.c
@@ -11,7 +11,8 @@
 #include <fstream>              /* for ofstream() */
 #include <fcntl.h>              /* for creat() */
 #include <sys/time.h>           /* for gettimeofday() */
-#include <sstream>
+#include <sstream>              /* for stringstream */
+#include <signal.h>             /* for kill(), SIGUSR1 */
 
 // requires apt-get install libboost-filesystem-dev:
 #include <boost/filesystem.hpp>
@@ -43,10 +44,32 @@ void dump(const string var){
 }
 
 
-   //       int stat(const char *path, struct stat *buf);
-   //       int fstat(int fd, struct stat *buf);
-   //       int lstat(const char *path, struct stat *buf);
 
+////////////////
+// little utility to help with argument parsing:
+//
+int prefix(const string shorter, const string longer){
+  return shorter == longer.substr(0, shorter.length());
+}
+
+void exeunt(const int sts){
+  if (sts == sa_good) exit(sts);
+
+  const char* foo = getenv("HI_Q_GROUP");
+  if (!foo) exit(sts);
+
+// No point in signalling ourself:
+  sighandler_t rslt = signal(SIGUSR1, SIG_IGN);
+  if (rslt == SIG_ERR) {
+    cerr << "error setting signal" << endl;
+  }
+  int k = kill(-atoi(foo), SIGUSR1);
+  if (k) {
+    cerr << "kill failed on group " << atoi(foo) << " ... ";
+    perror(0);
+  }
+  exit(sts);
+}
 
 class whatsit{
 public:
@@ -161,7 +184,8 @@ int main(int _argc, char** _argv){
       return 0;
     }
     else {
-      cerr << "Unrecognized arg, ignored: " << arg << endl;
+      cerr << "Unrecognized arg: " << arg << endl;
+      exeunt(sa_syserr);
     }
   }
 
@@ -172,7 +196,7 @@ int whatsit::doit(){
   char* ipvar = getenv("TCPREMOTEIP");
   if (!ipvar) {
     cerr << progname << ": TCPREMOTEIP not set???" << endl;
-    exit(sa_syserr);
+    exeunt(sa_syserr);
   }
   ipbase = ipvar;
   char* hostvar = getenv("TCPREMOTEHOST");
@@ -194,7 +218,7 @@ int whatsit::doit(){
         << ": mkdir failed for '"
         << dirname << "' : ";
       perror(0);
-      exit(sa_syserr);
+      exeunt(sa_syserr);
     }
   }
 
@@ -216,7 +240,7 @@ int whatsit::doit(){
     }
     close(fd);
     update("new customer", now, now);
-    return(bug_bait_grey);
+    exeunt(bug_bait_grey);
   }
 // here if stat succeeded
   mod_age = now.tv_sec - ipstat.st_mtime;
@@ -224,16 +248,16 @@ int whatsit::doit(){
   timeval mod_orig = {ipstat.st_mtime, 0};
   if (mod_age < minimum_age) {
     update("early bird", mod_orig, now);
-    return(bug_bait_grey);
+    exeunt(bug_bait_grey);
   }
   if (mod_age - ac_age < minimum_age    // early bird, or completely unused
     && mod_age > probation) {           // did not diligently resubmit
     update("disprobation", now, now);
-    return(bug_bait_grey);
+    exeunt(bug_bait_grey);
   }
   if (ac_age > maximum_age) {
     update("too old, starting over", now, now);
-    return(bug_bait_grey);
+    exeunt(bug_bait_grey);
   }
 // if all checks are passed, must be OK:
   update("returning customer", mod_orig, now);
diff --git a/tools/hi-test2.conf b/tools/hi-test2.conf
index c7312fb..3c1422c 100644
--- a/tools/hi-test2.conf
+++ b/tools/hi-test2.conf
@@ -1,3 +1,3 @@
 grey    hi-test x0 -snooze 10
-gray    greylist  x2 -snooze 10
+gray    greylist
 qq      hi-test x1 -snooze 1 -exit 3
diff --git a/tools/makefile b/tools/makefile
index 3e599b8..cf62473 100644
--- a/tools/makefile
+++ b/tools/makefile
@@ -44,7 +44,7 @@ logmark:
 	logger -t jsd -p mail.info =========================
 
 todo: 
-	echo zap -- kill program group \
+	echo zap penalize greylist status of spam       \
 	extended error codes from skrewt, greylist       \
 	extended error codes [-x] from spamc
 
-- 
cgit v1.2.3