Use p_new instead of xmalloc()
[apps/madmutt.git] / remailer.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 1999-2000 Thomas Roessler <roessler@does-not-exist.org>
4  *
5  * This file is part of mutt-ng, see http://www.muttng.org/.
6  * It's licensed under the GNU General Public License,
7  * please see the file GPL in the top level source directory.
8  */
9
10 /*
11  * Mixmaster support for Mutt
12  */
13
14 #include <lib-lib/lib-lib.h>
15
16 #include <lib-sys/unix.h>
17 #include <lib-ui/curses.h>
18 #include <lib-ui/menu.h>
19
20 #include "mutt.h"
21 #include "recvattach.h"
22
23 #include "remailer.h"
24
25 #define SW              (option(OPTMBOXPANE)?SidebarWidth:0)
26
27 #define MIX_CAP_COMPRESS  (1 << 0)
28 #define MIX_CAP_MIDDLEMAN (1 << 1)
29 #define MIX_CAP_NEWSPOST  (1 << 2)
30 #define MIX_CAP_NEWSMAIL  (1 << 3)
31
32 /* Mixmaster's maximum chain length.  Don't change this. */
33 #define MAXMIXES 19
34
35 typedef struct remailer_t {
36     int num;
37     int caps;
38     char *shortname;
39     char *addr;
40     char *ver;
41 } remailer_t;
42
43 DO_INIT(remailer_t, remailer);
44 static void remailer_wipe(remailer_t *remailer) {
45     p_delete(&remailer->shortname);
46     p_delete(&remailer->addr);
47     p_delete(&remailer->ver);
48 }
49 DO_NEW(remailer_t, remailer);
50 DO_DELETE(remailer_t, remailer);
51
52 typedef struct mixchain_t {
53     ssize_t cl;
54     int ch[MAXMIXES];
55 } mixchain_t;
56
57 struct coord {
58     short r, c;
59 };
60
61 static int mix_get_caps(const char *capstr)
62 {
63     int caps = 0;
64
65     while (*capstr) {
66         switch (*capstr) {
67           case 'C':
68             caps |= MIX_CAP_COMPRESS;
69             break;
70
71           case 'M':
72             caps |= MIX_CAP_MIDDLEMAN;
73             break;
74
75           case 'N':
76             switch (*++capstr) {
77               case 'm':
78                 caps |= MIX_CAP_NEWSMAIL;
79                 break;
80
81               case 'p':
82                 caps |= MIX_CAP_NEWSPOST;
83                 break;
84
85             }
86         }
87
88         if (*capstr)
89             capstr++;
90     }
91
92     return caps;
93 }
94
95 static void mix_add_entry(remailer_t ***type2_list, remailer_t *entry,
96                           ssize_t *slots, ssize_t *used)
97 {
98     if (*used == *slots) {
99         *slots += 5;
100         p_realloc(type2_list, *slots);
101     }
102
103     (*type2_list)[(*used)++] = entry;
104     if (entry)
105         entry->num = *used;
106 }
107
108 /* parse the type2.list as given by mixmaster -T */
109 static remailer_t **mix_type2_list (ssize_t * l)
110 {
111     FILE *fp;
112     pid_t mm_pid;
113     int devnull;
114
115     char cmd[HUGE_STRING + _POSIX_PATH_MAX];
116     char line[HUGE_STRING];
117     char *t;
118
119     remailer_t **type2_list = NULL, *p;
120     ssize_t slots = 0, used = 0;
121
122     if (!l)
123         return NULL;
124
125     if ((devnull = open ("/dev/null", O_RDWR)) == -1)
126         return NULL;
127
128     snprintf (cmd, sizeof (cmd), "%s -T", Mixmaster);
129
130     if ((mm_pid =
131          mutt_create_filter_fd (cmd, NULL, &fp, NULL, devnull, -1,
132                                 devnull)) == -1) {
133         close (devnull);
134         return NULL;
135     }
136
137     /* first, generate the "random" remailer */
138
139     p = remailer_new();
140     p->shortname = m_strdup("<random>");
141     mix_add_entry (&type2_list, p, &slots, &used);
142
143     while (fgets (line, sizeof (line), fp)) {
144         p = remailer_new();
145
146         if (!(t = strtok (line, " \t\n")))
147             goto problem;
148
149         p->shortname = m_strdup(t);
150
151         if (!(t = strtok (NULL, " \t\n")))
152             goto problem;
153
154         p->addr = m_strdup(t);
155
156         if (!(t = strtok (NULL, " \t\n")))
157             goto problem;
158
159         if (!(t = strtok (NULL, " \t\n")))
160             goto problem;
161
162         p->ver = m_strdup(t);
163
164         if (!(t = strtok (NULL, " \t\n")))
165             goto problem;
166
167         p->caps = mix_get_caps (t);
168
169         mix_add_entry (&type2_list, p, &slots, &used);
170         continue;
171
172 problem:
173         remailer_delete(&p);
174     }
175
176     *l = used;
177
178     mix_add_entry (&type2_list, NULL, &slots, &used);
179     mutt_wait_filter (mm_pid);
180
181     close (devnull);
182
183     return type2_list;
184 }
185
186 static void mix_free_type2_list (remailer_t *** ttlp)
187 {
188     int i;
189     remailer_t **type2_list = *ttlp;
190
191     for (i = 0; type2_list[i]; i++)
192         remailer_delete(&type2_list[i]);
193
194     p_delete(type2_list);
195 }
196
197
198 #define MIX_HOFFSET 2
199 #define MIX_VOFFSET (LINES - 6)
200 #define MIX_MAXROW  (LINES - 3)
201
202
203 static void mix_screen_coordinates (remailer_t ** type2_list,
204                                     struct coord **coordsp,
205                                     mixchain_t * chain, int i)
206 {
207     short c, r, oc;
208     struct coord *coords;
209
210     if (!chain->cl)
211         return;
212
213     p_realloc(coordsp, chain->cl);
214
215     coords = *coordsp;
216
217     if (i) {
218         c =
219             coords[i - 1].c + m_strlen(type2_list[chain->ch[i - 1]]->shortname) + 2;
220         r = coords[i - 1].r;
221     }
222     else {
223         r = MIX_VOFFSET;
224         c = MIX_HOFFSET;
225     }
226
227
228     for (; i < chain->cl; i++) {
229         oc = c;
230         c += m_strlen(type2_list[chain->ch[i]]->shortname) + 2;
231
232         if (c >= COLS) {
233             oc = c = MIX_HOFFSET;
234             r++;
235         }
236
237         coords[i].c = oc;
238         coords[i].r = r;
239
240     }
241
242 }
243
244 static void mix_redraw_ce (remailer_t ** type2_list,
245                            struct coord *coords,
246                            mixchain_t * chain, int i, short selected)
247 {
248     if (!coords || !chain)
249         return;
250
251     if (coords[i].r < MIX_MAXROW) {
252
253         if (selected)
254             SETCOLOR (MT_COLOR_INDICATOR);
255         else
256             SETCOLOR (MT_COLOR_NORMAL);
257
258         mvaddstr (coords[i].r, coords[i].c, type2_list[chain->ch[i]]->shortname);
259         SETCOLOR (MT_COLOR_NORMAL);
260
261         if (i + 1 < chain->cl)
262             addstr (", ");
263     }
264 }
265
266 static void mix_redraw_chain (remailer_t ** type2_list,
267                               struct coord *coords, mixchain_t * chain, int cur)
268 {
269     int i;
270
271     SETCOLOR (MT_COLOR_NORMAL);
272     BKGDSET (MT_COLOR_NORMAL);
273
274     for (i = MIX_VOFFSET; i < MIX_MAXROW; i++) {
275         move (i, 0);
276         clrtoeol ();
277     }
278
279     for (i = 0; i < chain->cl; i++)
280         mix_redraw_ce (type2_list, coords, chain, i, i == cur);
281 }
282
283 static void mix_redraw_head (mixchain_t * chain)
284 {
285     SETCOLOR (MT_COLOR_STATUS);
286     mvprintw (MIX_VOFFSET - 1, 0, "-- Remailer chain [Length: %d]",
287               chain ? chain->cl : 0);
288
289     BKGDSET (MT_COLOR_STATUS);
290     clrtoeol ();
291
292     BKGDSET (MT_COLOR_NORMAL);
293     SETCOLOR (MT_COLOR_NORMAL);
294 }
295
296 static const char *mix_format_caps (remailer_t * r)
297 {
298     static char capbuff[10];
299     char *t = capbuff;
300
301     if (r->caps & MIX_CAP_COMPRESS)
302         *t++ = 'C';
303     else
304         *t++ = ' ';
305
306     if (r->caps & MIX_CAP_MIDDLEMAN)
307         *t++ = 'M';
308     else
309         *t++ = ' ';
310
311     if (r->caps & MIX_CAP_NEWSPOST) {
312         *t++ = 'N';
313         *t++ = 'p';
314     }
315     else {
316         *t++ = ' ';
317         *t++ = ' ';
318     }
319
320     if (r->caps & MIX_CAP_NEWSMAIL) {
321         *t++ = 'N';
322         *t++ = 'm';
323     }
324     else {
325         *t++ = ' ';
326         *t++ = ' ';
327     }
328
329     *t = '\0';
330
331     return capbuff;
332 }
333
334 /*
335  * Format an entry for the remailer menu.
336  * 
337  * %n   number
338  * %c   capabilities
339  * %s   short name
340  * %a   address
341  *
342  */
343 static const char *
344 mix_entry_fmt (char *dest, ssize_t destlen, char op, const char *src,
345                const char *prefix, const char *ifstr,
346                const char *elstr, anytype data, format_flag flags)
347 {
348     char fmt[16];
349     remailer_t *remailer = data.ptr;
350     int optional = (flags & M_FORMAT_OPTIONAL);
351
352     switch (op) {
353       case 'n':
354         if (!optional) {
355             snprintf (fmt, sizeof (fmt), "%%%sd", prefix);
356             snprintf (dest, destlen, fmt, remailer->num);
357         }
358         break;
359       case 'c':
360         if (!optional) {
361             snprintf (fmt, sizeof (fmt), "%%%ss", prefix);
362             snprintf (dest, destlen, fmt, mix_format_caps (remailer));
363         }
364         break;
365       case 's':
366         if (!optional) {
367             snprintf (fmt, sizeof (fmt), "%%%ss", prefix);
368             snprintf (dest, destlen, fmt, NONULL (remailer->shortname));
369         }
370         else if (!remailer->shortname)
371             optional = 0;
372         break;
373       case 'a':
374         if (!optional) {
375             snprintf (fmt, sizeof (fmt), "%%%ss", prefix);
376             snprintf (dest, destlen, fmt, NONULL (remailer->addr));
377         }
378         else if (!remailer->addr)
379             optional = 0;
380         break;
381
382       default:
383         *dest = '\0';
384     }
385
386     if (flags & M_FORMAT_OPTIONAL)
387         m_strformat(dest, destlen, 0, optional ? ifstr : elstr,
388                     mutt_attach_fmt, data, 0);
389     return (src);
390 }
391
392
393
394 static void mix_entry (char *b, ssize_t blen, MUTTMENU * menu, int num)
395 {
396     remailer_t **type2_list = (remailer_t **) menu->data;
397
398     m_strformat(b, blen, COLS - SW, MixEntryFormat, mix_entry_fmt,
399                 type2_list[num],
400                 option(OPTARROWCURSOR) ? M_FORMAT_ARROWCURSOR : 0);
401 }
402
403 static int mix_chain_add (mixchain_t * chain, const char *s,
404                           remailer_t ** type2_list)
405 {
406     int i;
407
408     if (chain->cl >= MAXMIXES)
409         return -1;
410
411     if (!m_strcmp(s, "0") || !ascii_strcasecmp (s, "<random>")) {
412         chain->ch[chain->cl++] = 0;
413         return 0;
414     }
415
416     for (i = 0; type2_list[i]; i++) {
417         if (!ascii_strcasecmp (s, type2_list[i]->shortname)) {
418             chain->ch[chain->cl++] = i;
419             return 0;
420         }
421     }
422
423     /* replace unknown remailers by <random> */
424
425     if (!type2_list[i])
426         chain->ch[chain->cl++] = 0;
427
428     return 0;
429 }
430
431 static struct mapping_t RemailerHelp[] = {
432     {N_("Append"), OP_MIX_APPEND},
433     {N_("Insert"), OP_MIX_INSERT},
434     {N_("Delete"), OP_MIX_DELETE},
435     {N_("Abort"), OP_EXIT},
436     {N_("OK"), OP_MIX_USE},
437     {NULL, OP_NULL}
438 };
439
440
441 void mix_make_chain (string_list_t ** chainp, int *redraw)
442 {
443     string_list_t *p;
444     mixchain_t *chain;
445     int c_cur = 0, c_old = 0;
446     int m_len;
447     short c_redraw = 1;
448
449     remailer_t **type2_list = NULL;
450     ssize_t ttll = 0;
451
452     struct coord *coords = NULL;
453
454     MUTTMENU *menu;
455     char helpstr[STRING];
456     short loop = 1;
457     int op;
458
459     int i, j;
460     const char *t;
461
462     if (!(type2_list = mix_type2_list (&ttll))) {
463         mutt_error _("Can't get mixmaster's type2.list!");
464
465         return;
466     }
467
468     *redraw = REDRAW_FULL;
469
470     chain = p_new(mixchain_t, 1);
471     for (p = *chainp; p; p = p->next)
472         mix_chain_add (chain, (char *) p->data, type2_list);
473
474     string_list_wipe(chainp);
475
476     /* safety check */
477     for (i = 0; i < chain->cl; i++) {
478         if (chain->ch[i] >= ttll)
479             chain->ch[i] = 0;
480     }
481
482     mix_screen_coordinates (type2_list, &coords, chain, 0);
483
484     menu = mutt_new_menu ();
485     menu->menu = MENU_MIX;
486     menu->max = ttll;
487     menu->make_entry = mix_entry;
488     menu->tag = NULL;
489     menu->title = _("Select a remailer chain.");
490     menu->data = type2_list;
491     menu->help =
492         mutt_compile_help (helpstr, sizeof (helpstr), MENU_MIX, RemailerHelp);
493
494     m_len = menu->pagelen = MIX_VOFFSET - menu->offset - 1;
495
496     while (loop) {
497         if (menu->pagelen != m_len) {
498             menu->pagelen = m_len;
499             menu->redraw = REDRAW_FULL;
500         }
501
502         if (c_redraw) {
503             mix_redraw_head (chain);
504             mix_redraw_chain (type2_list, coords, chain, c_cur);
505             c_redraw = 0;
506         }
507         else if (c_cur != c_old) {
508             mix_redraw_ce (type2_list, coords, chain, c_old, 0);
509             mix_redraw_ce (type2_list, coords, chain, c_cur, 1);
510         }
511
512         c_old = c_cur;
513
514         switch ((op = mutt_menuLoop (menu))) {
515           case OP_REDRAW:
516             {
517                 menu_redraw_status (menu);
518                 mix_redraw_head (chain);
519                 mix_screen_coordinates (type2_list, &coords, chain, 0);
520                 mix_redraw_chain (type2_list, coords, chain, c_cur);
521                 menu->pagelen = m_len = MIX_VOFFSET - menu->offset - 1;
522                 break;
523             }
524
525           case OP_EXIT:
526             {
527                 chain->cl = 0;
528                 loop = 0;
529                 break;
530             }
531
532           case OP_MIX_USE:
533             {
534                 if (!chain->cl) {
535                     chain->cl++;
536                     chain->ch[0] = menu->current;
537                     mix_screen_coordinates (type2_list, &coords, chain, c_cur);
538                     c_redraw = 1;
539                 }
540
541                 if (chain->cl && chain->ch[chain->cl - 1] &&
542                     (type2_list[chain->ch[chain->cl - 1]]->caps & MIX_CAP_MIDDLEMAN))
543                 {
544                     mutt_error (_
545                                 ("Error: %s can't be used as the final remailer of a chain."),
546                                 type2_list[chain->ch[chain->cl - 1]]->shortname);
547                 }
548                 else {
549                     loop = 0;
550                 }
551                 break;
552             }
553
554           case OP_GENERIC_SELECT_ENTRY:
555           case OP_MIX_APPEND:
556             {
557                 if (chain->cl < MAXMIXES && c_cur < chain->cl)
558                     c_cur++;
559             }
560             /* fallthrough */
561           case OP_MIX_INSERT:
562             {
563                 if (chain->cl < MAXMIXES) {
564                     chain->cl++;
565                     for (i = chain->cl - 1; i > c_cur; i--)
566                         chain->ch[i] = chain->ch[i - 1];
567
568                     chain->ch[c_cur] = menu->current;
569                     mix_screen_coordinates (type2_list, &coords, chain, c_cur);
570                     c_redraw = 1;
571                 }
572                 else
573                     mutt_error (_("Mixmaster chains are limited to %d elements."),
574                                 MAXMIXES);
575
576                 break;
577             }
578
579           case OP_MIX_DELETE:
580             {
581                 if (chain->cl) {
582                     chain->cl--;
583
584                     for (i = c_cur; i < chain->cl; i++)
585                         chain->ch[i] = chain->ch[i + 1];
586
587                     if (c_cur == chain->cl && c_cur)
588                         c_cur--;
589
590                     mix_screen_coordinates (type2_list, &coords, chain, c_cur);
591                     c_redraw = 1;
592                 }
593                 else {
594                     mutt_error _("The remailer chain is already empty.");
595                 }
596                 break;
597             }
598
599           case OP_MIX_CHAIN_PREV:
600             {
601                 if (c_cur)
602                     c_cur--;
603                 else
604                     mutt_error _("You already have the first chain element selected.");
605
606                 break;
607             }
608
609           case OP_MIX_CHAIN_NEXT:
610             {
611                 if (chain->cl && c_cur < chain->cl - 1)
612                     c_cur++;
613                 else
614                     mutt_error _("You already have the last chain element selected.");
615
616                 break;
617             }
618         }
619     }
620
621     mutt_menuDestroy (&menu);
622
623     /* construct the remailer list */
624
625     if (chain->cl) {
626         for (i = 0; i < chain->cl; i++) {
627             if ((j = chain->ch[i]))
628                 t = type2_list[j]->shortname;
629             else
630                 t = "*";
631
632             *chainp = mutt_add_list (*chainp, t);
633         }
634     }
635
636     mix_free_type2_list (&type2_list);
637     p_delete(&coords);
638     p_delete(&chain);
639 }
640
641 /* some safety checks before piping the message to mixmaster */
642
643 int mix_check_message (HEADER * msg)
644 {
645     const char *fqdn;
646     short need_hostname = 0;
647     address_t *p;
648
649     if (msg->env->cc || msg->env->bcc) {
650         mutt_error _("Mixmaster doesn't accept Cc or Bcc headers.");
651
652         return -1;
653     }
654
655     /* When using mixmaster, we MUST qualify any addresses since
656      * the message will be delivered through remote systems.
657      * 
658      * use_domain won't be respected at this point, hidden_host will.
659      */
660
661     for (p = msg->env->to; p; p = p->next) {
662         if (!p->group && strchr (p->mailbox, '@') == NULL) {
663             need_hostname = 1;
664             break;
665         }
666     }
667
668     if (need_hostname) {
669
670         if (!(fqdn = mutt_fqdn(1))) {
671             mutt_error
672                 _("Please set the hostname variable to a proper value when using mixmaster!");
673             return -1;
674         }
675
676         /* Cc and Bcc are empty at this point. */
677         rfc822_qualify(msg->env->to, fqdn);
678         rfc822_qualify(msg->env->reply_to, fqdn);
679         rfc822_qualify(msg->env->mail_followup_to, fqdn);
680     }
681
682     return 0;
683 }
684
685 int mix_send_message (string_list_t * chain, const char *tempfile)
686 {
687     char cmd[HUGE_STRING];
688     char tmp[HUGE_STRING];
689     char cd_quoted[STRING];
690     int i;
691
692     snprintf (cmd, sizeof (cmd), "cat %s | %s -m ", tempfile, Mixmaster);
693
694     for (i = 0; chain; chain = chain->next, i = 1) {
695         m_strcpy(tmp, sizeof(tmp), cmd);
696         mutt_quote_filename (cd_quoted, sizeof (cd_quoted), (char *) chain->data);
697         snprintf (cmd, sizeof (cmd), "%s%s%s", tmp, i ? "," : " -l ", cd_quoted);
698     }
699
700     if (!option (OPTNOCURSES))
701         mutt_endwin (NULL);
702
703     if ((i = mutt_system (cmd))) {
704         fprintf (stderr, _("Error sending message, child exited %d.\n"), i);
705         if (!option (OPTNOCURSES)) {
706             mutt_any_key_to_continue (NULL);
707             mutt_error _("Error sending message.");
708         }
709     }
710
711     unlink (tempfile);
712     return i;
713 }
714