Moved to new module init structure.

[citadel.git] / citadel / serv_spam.c

diff --git a/citadel/serv_spam.c b/citadel/serv_spam.c
index c322f77f5373dfa6f5d903dac0a0b24ce8351697..64ad1f5cba100bdd7d9703c176e19f2422d2c8d9 100644 (file)
--- a/citadel/serv_spam.c
+++ b/citadel/serv_spam.c
@@ -36,12 +36,10 @@
 #include <sys/socket.h>
 #include "citadel.h"
 #include "server.h"
-#include "sysdep_decls.h"
 #include "citserver.h"
 #include "support.h"
 #include "config.h"
 #include "control.h"
-#include "serv_extensions.h"
 #include "room_ops.h"
 #include "user_ops.h"
 #include "policy.h"
@@ -53,45 +51,7 @@
 #include "clientsocket.h"
 
 
-
-/* 
- * This is a scanner I had started writing before deciding to just farm the
- * job out to SpamAssassin.  It *does* work but it's not in use.  We've
- * commented it out so it doesn't even compile.
- */
-#ifdef ___NOT_CURRENTLY_IN_USE___
-/* Scan a message for spam */
-int spam_filter(struct CtdlMessage *msg) {
-       int spam_strings_found = 0;
-       struct spamstrings_t *sptr;
-       char *ptr;
-
-       /* Bail out if there's no message text */
-       if (msg->cm_fields['M'] == NULL) return(0);
-
-
-       /* Scan! */
-       ptr = msg->cm_fields['M'];
-       while (ptr++[0] != 0) {
-               for (sptr = spamstrings; sptr != NULL; sptr = sptr->next) {
-                       if (!strncasecmp(ptr, sptr->string,
-                          strlen(sptr->string))) {
-                               ++spam_strings_found;
-                       }
-               }
-       }
-
-       if (spam_strings_found) {
-               if (msg->cm_fields['0'] != NULL) {
-                       phree(msg->cm_fields['0']);
-               }
-               msg->cm_fields['0'] = strdoop("Unsolicited spam rejected");
-               return(spam_strings_found);
-       }
-
-       return(0);
-}
-#endif
+#include "ctdl_module.h"
 
 
 
@@ -105,6 +65,8 @@ int spam_assassin(struct CtdlMessage *msg) {
        char buf[SIZ];
        int is_spam = 0;
        int sa;
+       char *msgtext;
+       size_t msglen;
 
        /* For users who have authenticated to this server we never want to
         * apply spam filtering, because presumably they're trustworthy.
@@ -117,10 +79,10 @@ int spam_assassin(struct CtdlMessage *msg) {
 
        /* Try them one by one until we get a working one */
         for (sa=0; sa<num_sahosts; ++sa) {
-                extract(buf, sahosts, sa);
-                lprintf(9, "Connecting to SpamAssassin at <%s>\n", buf);
+                extract_token(buf, sahosts, sa, '|', sizeof buf);
+                lprintf(CTDL_INFO, "Connecting to SpamAssassin at <%s>\n", buf);
                 sock = sock_connect(buf, SPAMASSASSIN_PORT, "tcp");
-                if (sock >= 0) lprintf(9, "Connected!\n");
+                if (sock >= 0) lprintf(CTDL_DEBUG, "Connected!\n");
         }
 
        if (sock < 0) {
@@ -131,14 +93,23 @@ int spam_assassin(struct CtdlMessage *msg) {
        }
 
        /* Command */
-       lprintf(9, "Transmitting command\n");
+       lprintf(CTDL_DEBUG, "Transmitting command\n");
        sprintf(buf, "CHECK SPAMC/1.2\r\n\r\n");
        sock_write(sock, buf, strlen(buf));
 
        /* Message */
-       CtdlRedirectOutput(NULL, sock);
-       CtdlOutputPreLoadedMsg(msg, 0L, MT_RFC822, HEADERS_ALL, 0, 1);
-       CtdlRedirectOutput(NULL, -1);
+       CC->redirect_buffer = malloc(SIZ);
+       CC->redirect_len = 0;
+       CC->redirect_alloc = SIZ;
+       CtdlOutputPreLoadedMsg(msg, MT_RFC822, HEADERS_ALL, 0, 1);
+       msgtext = CC->redirect_buffer;
+       msglen = CC->redirect_len;
+       CC->redirect_buffer = NULL;
+       CC->redirect_len = 0;
+       CC->redirect_alloc = 0;
+
+       sock_write(sock, msgtext, msglen);
+       free(msgtext);
 
        /* Close one end of the socket connection; this tells SpamAssassin
         * that we're done.
@@ -146,28 +117,27 @@ int spam_assassin(struct CtdlMessage *msg) {
        sock_shutdown(sock, SHUT_WR);
        
        /* Response */
-       lprintf(9, "Awaiting response\n");
+       lprintf(CTDL_DEBUG, "Awaiting response\n");
         if (sock_gets(sock, buf) < 0) {
                 goto bail;
         }
-        lprintf(9, "<%s\n", buf);
+        lprintf(CTDL_DEBUG, "<%s\n", buf);
        if (strncasecmp(buf, "SPAMD", 5)) {
                goto bail;
        }
         if (sock_gets(sock, buf) < 0) {
                 goto bail;
         }
-        lprintf(9, "<%s\n", buf);
+        lprintf(CTDL_DEBUG, "<%s\n", buf);
        if (!strncasecmp(buf, "Spam: True", 10)) {
                is_spam = 1;
        }
 
        if (is_spam) {
                if (msg->cm_fields['0'] != NULL) {
-                       phree(msg->cm_fields['0']);
+                       free(msg->cm_fields['0']);
                }
-               msg->cm_fields['0'] = strdoop(
-                       "Message rejected by SpamAssassin");
+               msg->cm_fields['0'] = strdup("5.7.1 message rejected by spam filter");
        }
 
 bail:  close(sock);
@@ -176,14 +146,10 @@ bail:     close(sock);
 
 
 
-char *serv_spam_init(void)
+CTDL_MODULE_INIT(spam)
 {
-
-/* (disabled built-in scanner, see above)
-       CtdlRegisterMessageHook(spam_filter, EVT_SMTPSCAN);
- */
-
        CtdlRegisterMessageHook(spam_assassin, EVT_SMTPSCAN);
 
+       /* return our Subversion id for the Log */
         return "$Id$";
 }