* replace rewrite_ctdl_sieve_config()s use of malloc/realloc/strlen/sprintf by StrBuf...
[citadel.git] / citadel / modules / sieve / serv_sieve.c
1 /*
2  * $Id$
3  *
4  * This module glues libSieve to the Citadel server in order to implement
5  * the Sieve mailbox filtering language (RFC 3028).
6  *
7  * Copyright (c) 1987-2010 by the citadel.org team
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 3 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
22  */
23
24 #include "sysdep.h"
25 #include <stdlib.h>
26 #include <unistd.h>
27 #include <stdio.h>
28 #include <fcntl.h>
29 #include <ctype.h>
30 #include <pwd.h>
31 #include <errno.h>
32 #include <sys/types.h>
33
34 #if TIME_WITH_SYS_TIME
35 # include <sys/time.h>
36 # include <time.h>
37 #else
38 # if HAVE_SYS_TIME_H
39 #  include <sys/time.h>
40 # else
41 #  include <time.h>
42 # endif
43 #endif
44
45 #include <sys/wait.h>
46 #include <string.h>
47 #include <limits.h>
48 #include <libcitadel.h>
49 #include "citadel.h"
50 #include "server.h"
51 #include "citserver.h"
52 #include "support.h"
53 #include "config.h"
54 #include "policy.h"
55 #include "database.h"
56 #include "msgbase.h"
57 #include "internet_addressing.h"
58 #include "ctdl_module.h"
59 #include "serv_sieve.h"
60
61 struct RoomProcList *sieve_list = NULL;
62 char *msiv_extensions = NULL;
63
64
65 /*
66  * Callback function to send libSieve trace messages to Citadel log facility
67  */
68 int ctdl_debug(sieve2_context_t *s, void *my)
69 {
70         CtdlLogPrintf(CTDL_DEBUG, "Sieve: %s\n", sieve2_getvalue_string(s, "message"));
71         return SIEVE2_OK;
72 }
73
74
75 /*
76  * Callback function to log script parsing errors
77  */
78 int ctdl_errparse(sieve2_context_t *s, void *my)
79 {
80         CtdlLogPrintf(CTDL_WARNING, "Error in script, line %d: %s\n",
81                 sieve2_getvalue_int(s, "lineno"),
82                 sieve2_getvalue_string(s, "message")
83         );
84         return SIEVE2_OK;
85 }
86
87
88 /*
89  * Callback function to log script execution errors
90  */
91 int ctdl_errexec(sieve2_context_t *s, void *my)
92 {
93         CtdlLogPrintf(CTDL_WARNING, "Error executing script: %s\n",
94                 sieve2_getvalue_string(s, "message")
95         );
96         return SIEVE2_OK;
97 }
98
99
100 /*
101  * Callback function to redirect a message to a different folder
102  */
103 int ctdl_redirect(sieve2_context_t *s, void *my)
104 {
105         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
106         struct CtdlMessage *msg = NULL;
107         struct recptypes *valid = NULL;
108         char recp[256];
109
110         safestrncpy(recp, sieve2_getvalue_string(s, "address"), sizeof recp);
111
112         CtdlLogPrintf(CTDL_DEBUG, "Action is REDIRECT, recipient <%s>\n", recp);
113
114         valid = validate_recipients(recp, NULL, 0);
115         if (valid == NULL) {
116                 CtdlLogPrintf(CTDL_WARNING, "REDIRECT failed: bad recipient <%s>\n", recp);
117                 return SIEVE2_ERROR_BADARGS;
118         }
119         if (valid->num_error > 0) {
120                 CtdlLogPrintf(CTDL_WARNING, "REDIRECT failed: bad recipient <%s>\n", recp);
121                 free_recipients(valid);
122                 return SIEVE2_ERROR_BADARGS;
123         }
124
125         msg = CtdlFetchMessage(cs->msgnum, 1);
126         if (msg == NULL) {
127                 CtdlLogPrintf(CTDL_WARNING, "REDIRECT failed: unable to fetch msg %ld\n", cs->msgnum);
128                 free_recipients(valid);
129                 return SIEVE2_ERROR_BADARGS;
130         }
131
132         CtdlSubmitMsg(msg, valid, NULL, 0);
133         cs->cancel_implicit_keep = 1;
134         free_recipients(valid);
135         CtdlFreeMessage(msg);
136         return SIEVE2_OK;
137 }
138
139
140 /*
141  * Callback function to indicate that a message *will* be kept in the inbox
142  */
143 int ctdl_keep(sieve2_context_t *s, void *my)
144 {
145         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
146         
147         CtdlLogPrintf(CTDL_DEBUG, "Action is KEEP\n");
148
149         cs->keep = 1;
150         cs->cancel_implicit_keep = 1;
151         return SIEVE2_OK;
152 }
153
154
155 /*
156  * Callback function to file a message into a different mailbox
157  */
158 int ctdl_fileinto(sieve2_context_t *s, void *my)
159 {
160         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
161         const char *dest_folder = sieve2_getvalue_string(s, "mailbox");
162         int c;
163         char foldername[256];
164         char original_room_name[ROOMNAMELEN];
165
166         CtdlLogPrintf(CTDL_DEBUG, "Action is FILEINTO, destination is <%s>\n", dest_folder);
167
168         /* FILEINTO 'INBOX' is the same thing as KEEP */
169         if ( (!strcasecmp(dest_folder, "INBOX")) || (!strcasecmp(dest_folder, MAILROOM)) ) {
170                 cs->keep = 1;
171                 cs->cancel_implicit_keep = 1;
172                 return SIEVE2_OK;
173         }
174
175         /* Remember what room we came from */
176         safestrncpy(original_room_name, CC->room.QRname, sizeof original_room_name);
177
178         /* First try a mailbox name match (check personal mail folders first) */
179         snprintf(foldername, sizeof foldername, "%010ld.%s", cs->usernum, dest_folder);
180         c = CtdlGetRoom(&CC->room, foldername);
181
182         /* Then a regular room name match (public and private rooms) */
183         if (c != 0) {
184                 safestrncpy(foldername, dest_folder, sizeof foldername);
185                 c = CtdlGetRoom(&CC->room, foldername);
186         }
187
188         if (c != 0) {
189                 CtdlLogPrintf(CTDL_WARNING, "FILEINTO failed: target <%s> does not exist\n", dest_folder);
190                 return SIEVE2_ERROR_BADARGS;
191         }
192
193         /* Yes, we actually have to go there */
194         CtdlUserGoto(NULL, 0, 0, NULL, NULL);
195
196         c = CtdlSaveMsgPointersInRoom(NULL, &cs->msgnum, 1, 0, NULL, 0);
197
198         /* Go back to the room we came from */
199         if (strcasecmp(original_room_name, CC->room.QRname)) {
200                 CtdlUserGoto(original_room_name, 0, 0, NULL, NULL);
201         }
202
203         if (c == 0) {
204                 cs->cancel_implicit_keep = 1;
205                 return SIEVE2_OK;
206         }
207         else {
208                 return SIEVE2_ERROR_BADARGS;
209         }
210 }
211
212
213 /*
214  * Callback function to indicate that a message should be discarded.
215  */
216 int ctdl_discard(sieve2_context_t *s, void *my)
217 {
218         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
219
220         CtdlLogPrintf(CTDL_DEBUG, "Action is DISCARD\n");
221
222         /* Cancel the implicit keep.  That's all there is to it. */
223         cs->cancel_implicit_keep = 1;
224         return SIEVE2_OK;
225 }
226
227
228
229 /*
230  * Callback function to indicate that a message should be rejected
231  */
232 int ctdl_reject(sieve2_context_t *s, void *my)
233 {
234         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
235         char *reject_text = NULL;
236
237         CtdlLogPrintf(CTDL_DEBUG, "Action is REJECT\n");
238
239         /* If we don't know who sent the message, do a DISCARD instead. */
240         if (IsEmptyStr(cs->sender)) {
241                 CtdlLogPrintf(CTDL_INFO, "Unknown sender.  Doing DISCARD instead of REJECT.\n");
242                 return ctdl_discard(s, my);
243         }
244
245         /* Assemble the reject message. */
246         reject_text = malloc(strlen(sieve2_getvalue_string(s, "message")) + 1024);
247         if (reject_text == NULL) {
248                 return SIEVE2_ERROR_FAIL;
249         }
250
251         sprintf(reject_text, 
252                 "Content-type: text/plain\n"
253                 "\n"
254                 "The message was refused by the recipient's mail filtering program.\n"
255                 "The reason given was as follows:\n"
256                 "\n"
257                 "%s\n"
258                 "\n"
259         ,
260                 sieve2_getvalue_string(s, "message")
261         );
262
263         quickie_message(        /* This delivers the message */
264                 NULL,
265                 cs->envelope_to,
266                 cs->sender,
267                 NULL,
268                 reject_text,
269                 FMT_RFC822,
270                 "Delivery status notification"
271         );
272
273         free(reject_text);
274         cs->cancel_implicit_keep = 1;
275         return SIEVE2_OK;
276 }
277
278
279
280 /*
281  * Callback function to indicate that a vacation message should be generated
282  */
283 int ctdl_vacation(sieve2_context_t *s, void *my)
284 {
285         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
286         struct sdm_vacation *vptr;
287         int days = 1;
288         const char *message;
289         char *vacamsg_text = NULL;
290         char vacamsg_subject[1024];
291
292         CtdlLogPrintf(CTDL_DEBUG, "Action is VACATION\n");
293
294         message = sieve2_getvalue_string(s, "message");
295         if (message == NULL) return SIEVE2_ERROR_BADARGS;
296
297         if (sieve2_getvalue_string(s, "subject") != NULL) {
298                 safestrncpy(vacamsg_subject, sieve2_getvalue_string(s, "subject"), sizeof vacamsg_subject);
299         }
300         else {
301                 snprintf(vacamsg_subject, sizeof vacamsg_subject, "Re: %s", cs->subject);
302         }
303
304         days = sieve2_getvalue_int(s, "days");
305         if (days < 1) days = 1;
306         if (days > MAX_VACATION) days = MAX_VACATION;
307
308         /* Check to see whether we've already alerted this sender that we're on vacation. */
309         for (vptr = cs->u->first_vacation; vptr != NULL; vptr = vptr->next) {
310                 if (!strcasecmp(vptr->fromaddr, cs->sender)) {
311                         if ( (time(NULL) - vptr->timestamp) < (days * 86400) ) {
312                                 CtdlLogPrintf(CTDL_DEBUG, "Already alerted <%s> recently.\n", cs->sender);
313                                 return SIEVE2_OK;
314                         }
315                 }
316         }
317
318         /* Assemble the reject message. */
319         vacamsg_text = malloc(strlen(message) + 1024);
320         if (vacamsg_text == NULL) {
321                 return SIEVE2_ERROR_FAIL;
322         }
323
324         sprintf(vacamsg_text, 
325                 "Content-type: text/plain\n"
326                 "\n"
327                 "%s\n"
328                 "\n"
329         ,
330                 message
331         );
332
333         quickie_message(        /* This delivers the message */
334                 NULL,
335                 cs->envelope_to,
336                 cs->sender,
337                 NULL,
338                 vacamsg_text,
339                 FMT_RFC822,
340                 vacamsg_subject
341         );
342
343         free(vacamsg_text);
344
345         /* Now update the list to reflect the fact that we've alerted this sender.
346          * If they're already in the list, just update the timestamp.
347          */
348         for (vptr = cs->u->first_vacation; vptr != NULL; vptr = vptr->next) {
349                 if (!strcasecmp(vptr->fromaddr, cs->sender)) {
350                         vptr->timestamp = time(NULL);
351                         return SIEVE2_OK;
352                 }
353         }
354
355         /* If we get to this point, create a new record.
356          */
357         vptr = malloc(sizeof(struct sdm_vacation));
358         memset(vptr, 0, sizeof(struct sdm_vacation));
359         vptr->timestamp = time(NULL);
360         safestrncpy(vptr->fromaddr, cs->sender, sizeof vptr->fromaddr);
361         vptr->next = cs->u->first_vacation;
362         cs->u->first_vacation = vptr;
363
364         return SIEVE2_OK;
365 }
366
367
368 /*
369  * Callback function to parse addresses per local system convention
370  * It is disabled because we don't support subaddresses.
371  */
372 #if 0
373 int ctdl_getsubaddress(sieve2_context_t *s, void *my)
374 {
375         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
376
377         /* libSieve does not take ownership of the memory used here.  But, since we
378          * are just pointing to locations inside a struct which we are going to free
379          * later, we're ok.
380          */
381         sieve2_setvalue_string(s, "user", cs->recp_user);
382         sieve2_setvalue_string(s, "detail", "");
383         sieve2_setvalue_string(s, "localpart", cs->recp_user);
384         sieve2_setvalue_string(s, "domain", cs->recp_node);
385         return SIEVE2_OK;
386 }
387 #endif
388
389
390 /*
391  * Callback function to parse message envelope
392  */
393 int ctdl_getenvelope(sieve2_context_t *s, void *my)
394 {
395         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
396
397         CtdlLogPrintf(CTDL_DEBUG, "Action is GETENVELOPE\nEnvFrom: %s\n  EnvTo: %s\n",
398                 cs->envelope_from, cs->envelope_to);
399
400         if (cs->envelope_from != NULL) {
401                 if ((cs->envelope_from[0] != '@')&&(cs->envelope_from[strlen(cs->envelope_from)-1] != '@')) {
402                         sieve2_setvalue_string(s, "from", cs->envelope_from);
403                 }
404                 else {
405                         sieve2_setvalue_string(s, "from", "invalid_envelope_from@example.org");
406                 }
407         }
408         else {
409                 sieve2_setvalue_string(s, "from", "null_envelope_from@example.org");
410         }
411
412
413         if (cs->envelope_to != NULL) {
414                 if ((cs->envelope_to[0] != '@') && (cs->envelope_to[strlen(cs->envelope_to)-1] != '@')) {
415                         sieve2_setvalue_string(s, "to", cs->envelope_to);
416                 }
417                 else {
418                         sieve2_setvalue_string(s, "to", "invalid_envelope_to@example.org");
419                 }
420         }
421         else {
422                 sieve2_setvalue_string(s, "to", "null_envelope_to@example.org");
423         }
424
425         return SIEVE2_OK;
426 }
427
428
429 /*
430  * Callback function to fetch message body
431  * (Uncomment the code if we implement this extension)
432  *
433 int ctdl_getbody(sieve2_context_t *s, void *my)
434 {
435         return SIEVE2_ERROR_UNSUPPORTED;
436 }
437  *
438  */
439
440
441 /*
442  * Callback function to fetch message size
443  */
444 int ctdl_getsize(sieve2_context_t *s, void *my)
445 {
446         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
447         struct MetaData smi;
448
449         GetMetaData(&smi, cs->msgnum);
450         
451         if (smi.meta_rfc822_length > 0L) {
452                 sieve2_setvalue_int(s, "size", (int)smi.meta_rfc822_length);
453                 return SIEVE2_OK;
454         }
455
456         return SIEVE2_ERROR_UNSUPPORTED;
457 }
458
459
460 /*
461  * Callback function to retrieve the sieve script
462  */
463 int ctdl_getscript(sieve2_context_t *s, void *my) {
464         struct sdm_script *sptr;
465         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
466
467         for (sptr=cs->u->first_script; sptr!=NULL; sptr=sptr->next) {
468                 if (sptr->script_active > 0) {
469                         CtdlLogPrintf(CTDL_DEBUG, "ctdl_getscript() is using script '%s'\n", sptr->script_name);
470                         sieve2_setvalue_string(s, "script", sptr->script_content);
471                         return SIEVE2_OK;
472                 }
473         }
474                 
475         CtdlLogPrintf(CTDL_DEBUG, "ctdl_getscript() found no active script\n");
476         return SIEVE2_ERROR_GETSCRIPT;
477 }
478
479 /*
480  * Callback function to retrieve message headers
481  */
482 int ctdl_getheaders(sieve2_context_t *s, void *my) {
483
484         struct ctdl_sieve *cs = (struct ctdl_sieve *)my;
485
486         CtdlLogPrintf(CTDL_DEBUG, "ctdl_getheaders() was called\n");
487         sieve2_setvalue_string(s, "allheaders", cs->rfc822headers);
488         return SIEVE2_OK;
489 }
490
491
492
493 /*
494  * Add a room to the list of those rooms which potentially require sieve processing
495  */
496 void sieve_queue_room(struct ctdlroom *which_room) {
497         struct RoomProcList *ptr;
498
499         ptr = (struct RoomProcList *) malloc(sizeof (struct RoomProcList));
500         if (ptr == NULL) return;
501
502         safestrncpy(ptr->name, which_room->QRname, sizeof ptr->name);
503         begin_critical_section(S_SIEVELIST);
504         ptr->next = sieve_list;
505         sieve_list = ptr;
506         end_critical_section(S_SIEVELIST);
507         CtdlLogPrintf(CTDL_DEBUG, "<%s> queued for Sieve processing\n", which_room->QRname);
508 }
509
510
511
512 /*
513  * Perform sieve processing for one message (called by sieve_do_room() for each message)
514  */
515 void sieve_do_msg(long msgnum, void *userdata) {
516         struct sdm_userdata *u = (struct sdm_userdata *) userdata;
517         sieve2_context_t *sieve2_context;
518         struct ctdl_sieve my;
519         int res;
520         struct CtdlMessage *msg;
521         int i;
522         size_t headers_len = 0;
523         int len = 0;
524
525         if (u == NULL)
526         {
527                 CtdlLogPrintf(CTDL_EMERG, "Can't process message <%ld> without userdata!\n", msgnum);
528                 return;
529         }
530
531         sieve2_context = u->sieve2_context;
532
533         CtdlLogPrintf(CTDL_DEBUG, "Performing sieve processing on msg <%ld>\n", msgnum);
534
535         /*
536          * Make sure you include message body so you can get those second-level headers ;)
537          */
538         msg = CtdlFetchMessage(msgnum, 1);
539         if (msg == NULL) return;
540
541         /*
542          * Grab the message headers so we can feed them to libSieve.
543          * Use HEADERS_ONLY rather than HEADERS_FAST in order to include second-level headers.
544          */
545         CC->redirect_buffer = malloc(SIZ);
546         CC->redirect_len = 0;
547         CC->redirect_alloc = SIZ;
548         CtdlOutputPreLoadedMsg(msg, MT_RFC822, HEADERS_ONLY, 0, 1, 0);
549         my.rfc822headers = CC->redirect_buffer;
550         headers_len = CC->redirect_len;
551         CC->redirect_buffer = NULL;
552         CC->redirect_len = 0;
553         CC->redirect_alloc = 0;
554
555         /*
556          * libSieve clobbers the stack if it encounters badly formed
557          * headers.  Sanitize our headers by stripping nonprintable
558          * characters.
559          */
560         for (i=0; i<headers_len; ++i) {
561                 if (!isascii(my.rfc822headers[i])) {
562                         my.rfc822headers[i] = '_';
563                 }
564         }
565
566         my.keep = 0;                            /* Set to 1 to declare an *explicit* keep */
567         my.cancel_implicit_keep = 0;            /* Some actions will cancel the implicit keep */
568         my.usernum = atol(CC->room.QRname);     /* Keep track of the owner of the room's namespace */
569         my.msgnum = msgnum;                     /* Keep track of the message number in our local store */
570         my.u = u;                               /* Hand off a pointer to the rest of this info */
571
572         /* Keep track of the recipient so we can do handling based on it later */
573         process_rfc822_addr(msg->cm_fields['R'], my.recp_user, my.recp_node, my.recp_name);
574
575         /* Keep track of the sender so we can use it for REJECT and VACATION responses */
576         if (msg->cm_fields['F'] != NULL) {
577                 safestrncpy(my.sender, msg->cm_fields['F'], sizeof my.sender);
578         }
579         else if ( (msg->cm_fields['A'] != NULL) && (msg->cm_fields['N'] != NULL) ) {
580                 snprintf(my.sender, sizeof my.sender, "%s@%s", msg->cm_fields['A'], msg->cm_fields['N']);
581         }
582         else if (msg->cm_fields['A'] != NULL) {
583                 safestrncpy(my.sender, msg->cm_fields['A'], sizeof my.sender);
584         }
585         else {
586                 strcpy(my.sender, "");
587         }
588
589         /* Keep track of the subject so we can use it for VACATION responses */
590         if (msg->cm_fields['U'] != NULL) {
591                 safestrncpy(my.subject, msg->cm_fields['U'], sizeof my.subject);
592         }
593         else {
594                 strcpy(my.subject, "");
595         }
596
597         /* Keep track of the envelope-from address (use body-from if not found) */
598         if (msg->cm_fields['P'] != NULL) {
599                 safestrncpy(my.envelope_from, msg->cm_fields['P'], sizeof my.envelope_from);
600                 stripallbut(my.envelope_from, '<', '>');
601         }
602         else if (msg->cm_fields['F'] != NULL) {
603                 safestrncpy(my.envelope_from, msg->cm_fields['F'], sizeof my.envelope_from);
604                 stripallbut(my.envelope_from, '<', '>');
605         }
606         else {
607                 strcpy(my.envelope_from, "");
608         }
609
610         len = strlen(my.envelope_from);
611         for (i=0; i<len; ++i) {
612                 if (isspace(my.envelope_from[i])) my.envelope_from[i] = '_';
613         }
614         if (haschar(my.envelope_from, '@') == 0) {
615                 strcat(my.envelope_from, "@");
616                 strcat(my.envelope_from, config.c_fqdn);
617         }
618
619         /* Keep track of the envelope-to address (use body-to if not found) */
620         if (msg->cm_fields['V'] != NULL) {
621                 safestrncpy(my.envelope_to, msg->cm_fields['V'], sizeof my.envelope_to);
622                 stripallbut(my.envelope_to, '<', '>');
623         }
624         else if (msg->cm_fields['R'] != NULL) {
625                 safestrncpy(my.envelope_to, msg->cm_fields['R'], sizeof my.envelope_to);
626                 if (msg->cm_fields['D'] != NULL) {
627                         strcat(my.envelope_to, "@");
628                         strcat(my.envelope_to, msg->cm_fields['D']);
629                 }
630                 stripallbut(my.envelope_to, '<', '>');
631         }
632         else {
633                 strcpy(my.envelope_to, "");
634         }
635
636         len = strlen(my.envelope_to);
637         for (i=0; i<len; ++i) {
638                 if (isspace(my.envelope_to[i])) my.envelope_to[i] = '_';
639         }
640         if (haschar(my.envelope_to, '@') == 0) {
641                 strcat(my.envelope_to, "@");
642                 strcat(my.envelope_to, config.c_fqdn);
643         }
644
645         CtdlFreeMessage(msg);
646
647         sieve2_setvalue_string(sieve2_context, "allheaders", my.rfc822headers);
648         
649         CtdlLogPrintf(CTDL_DEBUG, "Calling sieve2_execute()\n");
650         res = sieve2_execute(sieve2_context, &my);
651         if (res != SIEVE2_OK) {
652                 CtdlLogPrintf(CTDL_CRIT, "sieve2_execute() returned %d: %s\n", res, sieve2_errstr(res));
653         }
654
655         free(my.rfc822headers);
656         my.rfc822headers = NULL;
657
658         /*
659          * Delete the message from the inbox unless either we were told not to, or
660          * if no other action was successfully taken.
661          */
662         if ( (!my.keep) && (my.cancel_implicit_keep) ) {
663                 CtdlLogPrintf(CTDL_DEBUG, "keep is 0 -- deleting message from inbox\n");
664                 CtdlDeleteMessages(CC->room.QRname, &msgnum, 1, "");
665         }
666
667         CtdlLogPrintf(CTDL_DEBUG, "Completed sieve processing on msg <%ld>\n", msgnum);
668         u->lastproc = msgnum;
669
670         return;
671 }
672
673
674
675 /*
676  * Given the on-disk representation of our Sieve config, load
677  * it into an in-memory data structure.
678  */
679 void parse_sieve_config(char *conf, struct sdm_userdata *u) {
680         char *ptr;
681         char *c, *vacrec;
682         char keyword[256];
683         struct sdm_script *sptr;
684         struct sdm_vacation *vptr;
685
686         ptr = conf;
687         while (c = ptr, ptr = bmstrcasestr(ptr, CTDLSIEVECONFIGSEPARATOR), ptr != NULL) {
688                 *ptr = 0;
689                 ptr += strlen(CTDLSIEVECONFIGSEPARATOR);
690
691                 extract_token(keyword, c, 0, '|', sizeof keyword);
692
693                 if (!strcasecmp(keyword, "lastproc")) {
694                         u->lastproc = extract_long(c, 1);
695                 }
696
697                 else if (!strcasecmp(keyword, "script")) {
698                         sptr = malloc(sizeof(struct sdm_script));
699                         extract_token(sptr->script_name, c, 1, '|', sizeof sptr->script_name);
700                         sptr->script_active = extract_int(c, 2);
701                         remove_token(c, 0, '|');
702                         remove_token(c, 0, '|');
703                         remove_token(c, 0, '|');
704                         sptr->script_content = strdup(c);
705                         sptr->next = u->first_script;
706                         u->first_script = sptr;
707                 }
708
709                 else if (!strcasecmp(keyword, "vacation")) {
710
711                         if (c != NULL) while (vacrec=c, c=strchr(c, '\n'), (c != NULL)) {
712
713                                 *c = 0;
714                                 ++c;
715
716                                 if (strncasecmp(vacrec, "vacation|", 9)) {
717                                         vptr = malloc(sizeof(struct sdm_vacation));
718                                         extract_token(vptr->fromaddr, vacrec, 0, '|', sizeof vptr->fromaddr);
719                                         vptr->timestamp = extract_long(vacrec, 1);
720                                         vptr->next = u->first_vacation;
721                                         u->first_vacation = vptr;
722                                 }
723                         }
724                 }
725
726                 /* ignore unknown keywords */
727         }
728 }
729
730 /*
731  * We found the Sieve configuration for this user.
732  * Now do something with it.
733  */
734 void get_sieve_config_backend(long msgnum, void *userdata) {
735         struct sdm_userdata *u = (struct sdm_userdata *) userdata;
736         struct CtdlMessage *msg;
737         char *conf;
738
739         u->config_msgnum = msgnum;
740         msg = CtdlFetchMessage(msgnum, 1);
741         if (msg == NULL) {
742                 u->config_msgnum = (-1) ;
743                 return;
744         }
745
746         conf = msg->cm_fields['M'];
747         msg->cm_fields['M'] = NULL;
748         CtdlFreeMessage(msg);
749
750         if (conf != NULL) {
751                 parse_sieve_config(conf, u);
752                 free(conf);
753         }
754
755 }
756
757
758 /* 
759  * Write our citadel sieve config back to disk
760  * 
761  * (Set yes_write_to_disk to nonzero to make it actually write the config;
762  * otherwise it just frees the data structures.)
763  */
764 void rewrite_ctdl_sieve_config(struct sdm_userdata *u, int yes_write_to_disk) {
765         StrBuf *text;
766         struct sdm_script *sptr;
767         struct sdm_vacation *vptr;
768         
769         text = NewStrBufPlain(NULL, SIZ);
770         StrBufPrintf(text,
771                      "Content-type: application/x-citadel-sieve-config\n"
772                      "\n"
773                      CTDLSIEVECONFIGSEPARATOR
774                      "lastproc|%ld"
775                      CTDLSIEVECONFIGSEPARATOR
776                      ,
777                      u->lastproc
778                 );
779
780         while (u->first_script != NULL) {
781                 StrBufAppendPrintf(text,
782                                    "script|%s|%d|%s" CTDLSIEVECONFIGSEPARATOR,
783                                    u->first_script->script_name,
784                                    u->first_script->script_active,
785                                    u->first_script->script_content
786                         );
787                 sptr = u->first_script;
788                 u->first_script = u->first_script->next;
789                 free(sptr->script_content);
790                 free(sptr);
791         }
792
793         if (u->first_vacation != NULL) {
794
795                 StrBufAppendPrintf(text, "vacation|\n");
796                 while (u->first_vacation != NULL) {
797                         if ( (time(NULL) - u->first_vacation->timestamp) < (MAX_VACATION * 86400)) {
798                                 StrBufAppendPrintf(text, "%s|%ld\n",
799                                                    u->first_vacation->fromaddr,
800                                                    u->first_vacation->timestamp
801                                         );
802                         }
803                         vptr = u->first_vacation;
804                         u->first_vacation = u->first_vacation->next;
805                         free(vptr);
806                 }
807                 StrBufAppendPrintf(text, CTDLSIEVECONFIGSEPARATOR);
808         }
809
810         if (yes_write_to_disk)
811         {
812                 /* Save the config */
813                 quickie_message("Citadel", NULL, NULL, u->config_roomname,
814                                 ChrPtr(text),
815                                 4,
816                                 "Sieve configuration"
817                 );
818                 
819                 /* And delete the old one */
820                 if (u->config_msgnum > 0) {
821                         CtdlDeleteMessages(u->config_roomname, &u->config_msgnum, 1, "");
822                 }
823         }
824
825         FreeStrBuf (&text);
826
827 }
828
829
830 /*
831  * This is our callback registration table for libSieve.
832  */
833 sieve2_callback_t ctdl_sieve_callbacks[] = {
834         { SIEVE2_ACTION_REJECT,         ctdl_reject             },
835         { SIEVE2_ACTION_VACATION,       ctdl_vacation           },
836         { SIEVE2_ERRCALL_PARSE,         ctdl_errparse           },
837         { SIEVE2_ERRCALL_RUNTIME,       ctdl_errexec            },
838         { SIEVE2_ACTION_FILEINTO,       ctdl_fileinto           },
839         { SIEVE2_ACTION_REDIRECT,       ctdl_redirect           },
840         { SIEVE2_ACTION_DISCARD,        ctdl_discard            },
841         { SIEVE2_ACTION_KEEP,           ctdl_keep               },
842         { SIEVE2_SCRIPT_GETSCRIPT,      ctdl_getscript          },
843         { SIEVE2_DEBUG_TRACE,           ctdl_debug              },
844         { SIEVE2_MESSAGE_GETALLHEADERS, ctdl_getheaders         },
845         { SIEVE2_MESSAGE_GETSIZE,       ctdl_getsize            },
846         { SIEVE2_MESSAGE_GETENVELOPE,   ctdl_getenvelope        },
847 /*
848  * These actions are unsupported by Citadel so we don't declare them.
849  *
850         { SIEVE2_ACTION_NOTIFY,         ctdl_notify             },
851         { SIEVE2_MESSAGE_GETSUBADDRESS, ctdl_getsubaddress      },
852         { SIEVE2_MESSAGE_GETBODY,       ctdl_getbody            },
853  *
854  */
855         { 0 }
856 };
857
858
859 /*
860  * Perform sieve processing for a single room
861  */
862 void sieve_do_room(char *roomname) {
863         
864         struct sdm_userdata u;
865         sieve2_context_t *sieve2_context = NULL;        /* Context for sieve parser */
866         int res;                                        /* Return code from libsieve calls */
867         long orig_lastproc = 0;
868
869         memset(&u, 0, sizeof u);
870
871         /* See if the user who owns this 'mailbox' has any Sieve scripts that
872          * require execution.
873          */
874         snprintf(u.config_roomname, sizeof u.config_roomname, "%010ld.%s", atol(roomname), USERCONFIGROOM);
875         if (CtdlGetRoom(&CC->room, u.config_roomname) != 0) {
876                 CtdlLogPrintf(CTDL_DEBUG, "<%s> does not exist.  No processing is required.\n", u.config_roomname);
877                 return;
878         }
879
880         /*
881          * Find the sieve scripts and control record and do something
882          */
883         u.config_msgnum = (-1);
884         CtdlForEachMessage(MSGS_LAST, 1, NULL, SIEVECONFIG, NULL,
885                 get_sieve_config_backend, (void *)&u );
886
887         if (u.config_msgnum < 0) {
888                 CtdlLogPrintf(CTDL_DEBUG, "No Sieve rules exist.  No processing is required.\n");
889                 return;
890         }
891
892         CtdlLogPrintf(CTDL_DEBUG, "Rules found.  Performing Sieve processing for <%s>\n", roomname);
893
894         if (CtdlGetRoom(&CC->room, roomname) != 0) {
895                 CtdlLogPrintf(CTDL_CRIT, "ERROR: cannot load <%s>\n", roomname);
896                 return;
897         }
898
899         /* Initialize the Sieve parser */
900         
901         res = sieve2_alloc(&sieve2_context);
902         if (res != SIEVE2_OK) {
903                 CtdlLogPrintf(CTDL_CRIT, "sieve2_alloc() returned %d: %s\n", res, sieve2_errstr(res));
904                 return;
905         }
906
907         res = sieve2_callbacks(sieve2_context, ctdl_sieve_callbacks);
908         if (res != SIEVE2_OK) {
909                 CtdlLogPrintf(CTDL_CRIT, "sieve2_callbacks() returned %d: %s\n", res, sieve2_errstr(res));
910                 goto BAIL;
911         }
912
913         /* Validate the script */
914
915         struct ctdl_sieve my;           /* dummy ctdl_sieve struct just to pass "u" slong */
916         memset(&my, 0, sizeof my);
917         my.u = &u;
918         res = sieve2_validate(sieve2_context, &my);
919         if (res != SIEVE2_OK) {
920                 CtdlLogPrintf(CTDL_CRIT, "sieve2_validate() returned %d: %s\n", res, sieve2_errstr(res));
921                 goto BAIL;
922         }
923
924         /* Do something useful */
925         u.sieve2_context = sieve2_context;
926         orig_lastproc = u.lastproc;
927         CtdlForEachMessage(MSGS_GT, u.lastproc, NULL, NULL, NULL,
928                 sieve_do_msg,
929                 (void *) &u
930         );
931
932 BAIL:
933         res = sieve2_free(&sieve2_context);
934         if (res != SIEVE2_OK) {
935                 CtdlLogPrintf(CTDL_CRIT, "sieve2_free() returned %d: %s\n", res, sieve2_errstr(res));
936         }
937
938         /* Rewrite the config if we have to */
939         rewrite_ctdl_sieve_config(&u, (u.lastproc > orig_lastproc) ) ;
940 }
941
942
943 /*
944  * Perform sieve processing for all rooms which require it
945  */
946 void perform_sieve_processing(void) {
947         struct RoomProcList *ptr = NULL;
948
949         if (sieve_list != NULL) {
950                 CtdlLogPrintf(CTDL_DEBUG, "Begin Sieve processing\n");
951                 while (sieve_list != NULL) {
952                         char spoolroomname[ROOMNAMELEN];
953                         safestrncpy(spoolroomname, sieve_list->name, sizeof spoolroomname);
954                         begin_critical_section(S_SIEVELIST);
955
956                         /* pop this record off the list */
957                         ptr = sieve_list;
958                         sieve_list = sieve_list->next;
959                         free(ptr);
960
961                         /* invalidate any duplicate entries to prevent double processing */
962                         for (ptr=sieve_list; ptr!=NULL; ptr=ptr->next) {
963                                 if (!strcasecmp(ptr->name, spoolroomname)) {
964                                         ptr->name[0] = 0;
965                                 }
966                         }
967
968                         end_critical_section(S_SIEVELIST);
969                         if (spoolroomname[0] != 0) {
970                                 sieve_do_room(spoolroomname);
971                         }
972                 }
973         }
974 }
975
976
977 void msiv_load(struct sdm_userdata *u) {
978         char hold_rm[ROOMNAMELEN];
979
980         strcpy(hold_rm, CC->room.QRname);       /* save current room */
981
982         /* Take a spin through the user's personal address book */
983         if (CtdlGetRoom(&CC->room, USERCONFIGROOM) == 0) {
984         
985                 u->config_msgnum = (-1);
986                 strcpy(u->config_roomname, CC->room.QRname);
987                 CtdlForEachMessage(MSGS_LAST, 1, NULL, SIEVECONFIG, NULL,
988                         get_sieve_config_backend, (void *)u );
989
990         }
991
992         if (strcmp(CC->room.QRname, hold_rm)) {
993                 CtdlGetRoom(&CC->room, hold_rm);    /* return to saved room */
994         }
995 }
996
997 void msiv_store(struct sdm_userdata *u, int yes_write_to_disk) {
998 /*
999  * Initialise the sieve configs last processed message number.
1000  * We don't need to get the highest message number for the users inbox since the systems
1001  * highest message number will be higher than that and loer than this scripts message number
1002  * This prevents this new script from processing any old messages in the inbox.
1003  * Most importantly it will prevent vacation messages being sent to lots of old messages
1004  * in the inbox.
1005  */
1006         u->lastproc = CtdlGetCurrentMessageNumber();
1007         rewrite_ctdl_sieve_config(u, yes_write_to_disk);
1008 }
1009
1010
1011 /*
1012  * Select the active script.
1013  * (Set script_name to an empty string to disable all scripts)
1014  * 
1015  * Returns 0 on success or nonzero for error.
1016  */
1017 int msiv_setactive(struct sdm_userdata *u, char *script_name) {
1018         int ok = 0;
1019         struct sdm_script *s;
1020
1021         /* First see if the supplied value is ok */
1022
1023         if (IsEmptyStr(script_name)) {
1024                 ok = 1;
1025         }
1026         else {
1027                 for (s=u->first_script; s!=NULL; s=s->next) {
1028                         if (!strcasecmp(s->script_name, script_name)) {
1029                                 ok = 1;
1030                         }
1031                 }
1032         }
1033
1034         if (!ok) return(-1);
1035
1036         /* Now set the active script */
1037         for (s=u->first_script; s!=NULL; s=s->next) {
1038                 if (!strcasecmp(s->script_name, script_name)) {
1039                         s->script_active = 1;
1040                 }
1041                 else {
1042                         s->script_active = 0;
1043                 }
1044         }
1045         
1046         return(0);
1047 }
1048
1049
1050 /*
1051  * Fetch a script by name.
1052  *
1053  * Returns NULL if the named script was not found, or a pointer to the script
1054  * if it was found.   NOTE: the caller does *not* own the memory returned by
1055  * this function.  Copy it if you need to keep it.
1056  */
1057 char *msiv_getscript(struct sdm_userdata *u, char *script_name) {
1058         struct sdm_script *s;
1059
1060         for (s=u->first_script; s!=NULL; s=s->next) {
1061                 if (!strcasecmp(s->script_name, script_name)) {
1062                         if (s->script_content != NULL) {
1063                                 return (s->script_content);
1064                         }
1065                 }
1066         }
1067
1068         return(NULL);
1069 }
1070
1071
1072 /*
1073  * Delete a script by name.
1074  *
1075  * Returns 0 if the script was deleted.
1076  *       1 if the script was not found.
1077  *       2 if the script cannot be deleted because it is active.
1078  */
1079 int msiv_deletescript(struct sdm_userdata *u, char *script_name) {
1080         struct sdm_script *s = NULL;
1081         struct sdm_script *script_to_delete = NULL;
1082
1083         for (s=u->first_script; s!=NULL; s=s->next) {
1084                 if (!strcasecmp(s->script_name, script_name)) {
1085                         script_to_delete = s;
1086                         if (s->script_active) {
1087                                 return(2);
1088                         }
1089                 }
1090         }
1091
1092         if (script_to_delete == NULL) return(1);
1093
1094         if (u->first_script == script_to_delete) {
1095                 u->first_script = u->first_script->next;
1096         }
1097         else for (s=u->first_script; s!=NULL; s=s->next) {
1098                 if (s->next == script_to_delete) {
1099                         s->next = s->next->next;
1100                 }
1101         }
1102
1103         free(script_to_delete->script_content);
1104         free(script_to_delete);
1105         return(0);
1106 }
1107
1108
1109 /*
1110  * Add or replace a new script.  
1111  * NOTE: after this function returns, "u" owns the memory that "script_content"
1112  * was pointing to.
1113  */
1114 void msiv_putscript(struct sdm_userdata *u, char *script_name, char *script_content) {
1115         int replaced = 0;
1116         struct sdm_script *s, *sptr;
1117
1118         for (s=u->first_script; s!=NULL; s=s->next) {
1119                 if (!strcasecmp(s->script_name, script_name)) {
1120                         if (s->script_content != NULL) {
1121                                 free(s->script_content);
1122                         }
1123                         s->script_content = script_content;
1124                         replaced = 1;
1125                 }
1126         }
1127
1128         if (replaced == 0) {
1129                 sptr = malloc(sizeof(struct sdm_script));
1130                 safestrncpy(sptr->script_name, script_name, sizeof sptr->script_name);
1131                 sptr->script_content = script_content;
1132                 sptr->script_active = 0;
1133                 sptr->next = u->first_script;
1134                 u->first_script = sptr;
1135         }
1136 }
1137
1138
1139
1140 /*
1141  * Citadel protocol to manage sieve scripts.
1142  * This is basically a simplified (read: doesn't resemble IMAP) version
1143  * of the 'managesieve' protocol.
1144  */
1145 void cmd_msiv(char *argbuf) {
1146         char subcmd[256];
1147         struct sdm_userdata u;
1148         char script_name[256];
1149         char *script_content = NULL;
1150         struct sdm_script *s;
1151         int i;
1152         int changes_made = 0;
1153
1154         memset(&u, 0, sizeof(struct sdm_userdata));
1155
1156         if (CtdlAccessCheck(ac_logged_in)) return;
1157         extract_token(subcmd, argbuf, 0, '|', sizeof subcmd);
1158         msiv_load(&u);
1159
1160         if (!strcasecmp(subcmd, "putscript")) {
1161                 extract_token(script_name, argbuf, 1, '|', sizeof script_name);
1162                 if (!IsEmptyStr(script_name)) {
1163                         cprintf("%d Transmit script now\n", SEND_LISTING);
1164                         script_content = CtdlReadMessageBody(HKEY("000"), config.c_maxmsglen, NULL, 0, 0);
1165                         msiv_putscript(&u, script_name, script_content);
1166                         changes_made = 1;
1167                 }
1168                 else {
1169                         cprintf("%d Invalid script name.\n", ERROR + ILLEGAL_VALUE);
1170                 }
1171         }       
1172         
1173         else if (!strcasecmp(subcmd, "listscripts")) {
1174                 cprintf("%d Scripts:\n", LISTING_FOLLOWS);
1175                 for (s=u.first_script; s!=NULL; s=s->next) {
1176                         if (s->script_content != NULL) {
1177                                 cprintf("%s|%d|\n", s->script_name, s->script_active);
1178                         }
1179                 }
1180                 cprintf("000\n");
1181         }
1182
1183         else if (!strcasecmp(subcmd, "setactive")) {
1184                 extract_token(script_name, argbuf, 1, '|', sizeof script_name);
1185                 if (msiv_setactive(&u, script_name) == 0) {
1186                         cprintf("%d ok\n", CIT_OK);
1187                         changes_made = 1;
1188                 }
1189                 else {
1190                         cprintf("%d Script '%s' does not exist.\n",
1191                                 ERROR + ILLEGAL_VALUE,
1192                                 script_name
1193                         );
1194                 }
1195         }
1196
1197         else if (!strcasecmp(subcmd, "getscript")) {
1198                 extract_token(script_name, argbuf, 1, '|', sizeof script_name);
1199                 script_content = msiv_getscript(&u, script_name);
1200                 if (script_content != NULL) {
1201                         int script_len;
1202
1203                         cprintf("%d Script:\n", LISTING_FOLLOWS);
1204                         script_len = strlen(script_content);
1205                         client_write(script_content, script_len);
1206                         if (script_content[script_len-1] != '\n') {
1207                                 cprintf("\n");
1208                         }
1209                         cprintf("000\n");
1210                 }
1211                 else {
1212                         cprintf("%d Invalid script name.\n", ERROR + ILLEGAL_VALUE);
1213                 }
1214         }
1215
1216         else if (!strcasecmp(subcmd, "deletescript")) {
1217                 extract_token(script_name, argbuf, 1, '|', sizeof script_name);
1218                 i = msiv_deletescript(&u, script_name);
1219                 if (i == 0) {
1220                         cprintf("%d ok\n", CIT_OK);
1221                         changes_made = 1;
1222                 }
1223                 else if (i == 1) {
1224                         cprintf("%d Script '%s' does not exist.\n",
1225                                 ERROR + ILLEGAL_VALUE,
1226                                 script_name
1227                         );
1228                 }
1229                 else if (i == 2) {
1230                         cprintf("%d Script '%s' is active and cannot be deleted.\n",
1231                                 ERROR + ILLEGAL_VALUE,
1232                                 script_name
1233                         );
1234                 }
1235                 else {
1236                         cprintf("%d unknown error\n", ERROR);
1237                 }
1238         }
1239
1240         else {
1241                 cprintf("%d Invalid subcommand\n", ERROR + CMD_NOT_SUPPORTED);
1242         }
1243
1244         msiv_store(&u, changes_made);
1245 }
1246
1247
1248
1249 void ctdl_sieve_init(void) {
1250         char *cred = NULL;
1251         sieve2_context_t *sieve2_context = NULL;
1252         int res;
1253
1254         /*
1255          *      We don't really care about dumping the entire credits to the log
1256          *      every time the server is initialized.  The documentation will suffice
1257          *      for that purpose.  We are making a call to sieve2_credits() in order
1258          *      to demonstrate that we have successfully linked in to libsieve.
1259          */
1260         cred = strdup(sieve2_credits());
1261         if (cred == NULL) return;
1262
1263         if (strlen(cred) > 60) {
1264                 strcpy(&cred[55], "...");
1265         }
1266
1267         CtdlLogPrintf(CTDL_INFO, "%s\n",cred);
1268         free(cred);
1269
1270         /* Briefly initialize a Sieve parser instance just so we can list the
1271          * extensions that are available.
1272          */
1273         res = sieve2_alloc(&sieve2_context);
1274         if (res != SIEVE2_OK) {
1275                 CtdlLogPrintf(CTDL_CRIT, "sieve2_alloc() returned %d: %s\n", res, sieve2_errstr(res));
1276                 return;
1277         }
1278
1279         res = sieve2_callbacks(sieve2_context, ctdl_sieve_callbacks);
1280         if (res != SIEVE2_OK) {
1281                 CtdlLogPrintf(CTDL_CRIT, "sieve2_callbacks() returned %d: %s\n", res, sieve2_errstr(res));
1282                 goto BAIL;
1283         }
1284
1285         msiv_extensions = strdup(sieve2_listextensions(sieve2_context));
1286         CtdlLogPrintf(CTDL_INFO, "Extensions: %s\n", msiv_extensions);
1287
1288 BAIL:   res = sieve2_free(&sieve2_context);
1289         if (res != SIEVE2_OK) {
1290                 CtdlLogPrintf(CTDL_CRIT, "sieve2_free() returned %d: %s\n", res, sieve2_errstr(res));
1291         }
1292
1293 }
1294
1295 int serv_sieve_room(struct ctdlroom *room)
1296 {
1297         if (!strcasecmp(&room->QRname[11], MAILROOM)) {
1298                 sieve_queue_room(room);
1299         }
1300         return 0;
1301 }
1302
1303 CTDL_MODULE_INIT(sieve)
1304 {
1305         if (!threading)
1306         {
1307
1308                 ctdl_sieve_init();
1309                 CtdlRegisterProtoHook(cmd_msiv, "MSIV", "Manage Sieve scripts");
1310                 CtdlRegisterRoomHook(serv_sieve_room);
1311                 CtdlRegisterSessionHook(perform_sieve_processing, EVT_HOUSE);
1312         }
1313         
1314         /* return our Subversion id for the Log */
1315         return "$Id$";
1316 }
1317