4 * This module allows Citadel to use SpamAssassin to filter incoming messages
5 * arriving via SMTP. For more information on SpamAssassin, visit
6 * http://www.spamassassin.org (the SpamAssassin project is not in any way
7 * affiliated with the Citadel project).
9 * Copyright (c) 1998-2009 by the citadel.org team
11 * This program is free software; you can redistribute it and/or modify
12 * it under the terms of the GNU General Public License as published by
13 * the Free Software Foundation; either version 3 of the License, or
14 * (at your option) any later version.
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
21 * You should have received a copy of the GNU General Public License
22 * along with this program; if not, write to the Free Software
23 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 #define SPAMASSASSIN_PORT "783"
36 #include <sys/types.h>
38 #if TIME_WITH_SYS_TIME
39 # include <sys/time.h>
43 # include <sys/time.h>
52 #include <sys/socket.h>
53 #include <libcitadel.h>
56 #include "citserver.h"
64 #include "internet_addressing.h"
66 #include "clientsocket.h"
69 #include "ctdl_module.h"
74 * Connect to the SpamAssassin server and scan a message.
76 int spam_assassin(struct CtdlMessage *msg) {
87 /* For users who have authenticated to this server we never want to
88 * apply spam filtering, because presumably they're trustworthy.
90 if (CC->logged_in) return(0);
92 /* See if we have any SpamAssassin hosts configured */
93 num_sahosts = get_hosts(sahosts, "spamassassin");
94 if (num_sahosts < 1) return(0);
96 /* Try them one by one until we get a working one */
97 for (sa=0; sa<num_sahosts; ++sa) {
98 extract_token(buf, sahosts, sa, '|', sizeof buf);
99 CtdlLogPrintf(CTDL_INFO, "Connecting to SpamAssassin at <%s>\n", buf);
100 sock = sock_connect(buf, SPAMASSASSIN_PORT, "tcp");
101 if (sock >= 0) CtdlLogPrintf(CTDL_DEBUG, "Connected!\n");
105 /* If the service isn't running, just pass the mail
106 * through. Potentially throwing away mails isn't good.
111 CCC->sReadBuf = NewStrBuf();
112 CCC->sMigrateBuf = NewStrBuf();
116 CtdlLogPrintf(CTDL_DEBUG, "Transmitting command\n");
117 sprintf(buf, "CHECK SPAMC/1.2\r\n\r\n");
118 sock_write(&sock, buf, strlen(buf));
121 CCC->redirect_buffer = malloc(SIZ);
122 CCC->redirect_len = 0;
123 CCC->redirect_alloc = SIZ;
124 CtdlOutputPreLoadedMsg(msg, MT_RFC822, HEADERS_ALL, 0, 1, 0);
125 msgtext = CC->redirect_buffer;
126 msglen = CC->redirect_len;
127 CCC->redirect_buffer = NULL;
128 CCC->redirect_len = 0;
129 CCC->redirect_alloc = 0;
131 sock_write(&sock, msgtext, msglen);
134 /* Close one end of the socket connection; this tells SpamAssassin
138 sock_shutdown(sock, SHUT_WR);
141 CtdlLogPrintf(CTDL_DEBUG, "Awaiting response\n");
142 if (sock_getln(&sock, buf, sizeof buf) < 0) {
145 CtdlLogPrintf(CTDL_DEBUG, "<%s\n", buf);
146 if (strncasecmp(buf, "SPAMD", 5)) {
149 if (sock_getln(&sock, buf, sizeof buf) < 0) {
152 CtdlLogPrintf(CTDL_DEBUG, "<%s\n", buf);
153 CtdlLogPrintf(CTDL_DEBUG, "c_spam_flag_only setting %d\n", config.c_spam_flag_only);
154 if (config.c_spam_flag_only) {
155 CtdlLogPrintf(CTDL_DEBUG, "flag spam code used");
165 extract_token(sastatus, buf, 1, ' ', sizeof sastatus);
166 extract_token(sascore, buf, 3, ' ', sizeof sascore);
167 extract_token(saoutof, buf, 5, ' ', sizeof saoutof);
169 sprintf(buf,"X-Spam-Level: ");
170 char *cur = buf + 14;
171 for (numscore = atoi(sascore); numscore>0; numscore--)
175 sprintf(cur,"\r\nX-Spam-Status: %s, score=%s required=%s\r\n", sastatus, sascore, saoutof);
176 headerlen = strlen(buf);
177 oldmsgsize = strlen(msg->cm_fields['M']) + 1;
178 newmsgsize = headerlen + oldmsgsize;
180 msg->cm_fields['M'] = realloc(msg->cm_fields['M'], newmsgsize);
182 memmove(msg->cm_fields['M']+headerlen,msg->cm_fields['M'],oldmsgsize);
183 memcpy(msg->cm_fields['M'],buf,headerlen);
186 CtdlLogPrintf(CTDL_DEBUG, "reject spam code used");
187 if (!strncasecmp(buf, "Spam: True", 10)) {
192 if (msg->cm_fields['0'] != NULL) {
193 free(msg->cm_fields['0']);
195 msg->cm_fields['0'] = strdup("message rejected by spam filter");
200 FreeStrBuf(&CCC->sReadBuf);
201 FreeStrBuf(&CCC->sMigrateBuf);
207 CTDL_MODULE_INIT(spam)
211 CtdlRegisterMessageHook(spam_assassin, EVT_SMTPSCAN);
214 /* return our Subversion id for the Log */