many simplifications, cosmetics.
[apps/madmutt.git] / muttlib.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 1996-2000 Michael R. Elkins <me@mutt.org>
4  * Copyright (C) 1999-2000 Thomas Roessler <roessler@does-not-exist.org>
5  *
6  * This file is part of mutt-ng, see http://www.muttng.org/.
7  * It's licensed under the GNU General Public License,
8  * please see the file GPL in the top level source directory.
9  */
10
11 #include <lib-lib/lib-lib.h>
12
13 #include <grp.h>
14 #include <pwd.h>
15
16 #include <lib-ui/curses.h>
17 #include <lib-ui/enter.h>
18 #include <lib-sys/unix.h>
19 #include <imap/imap.h>
20
21 #include "alias.h"
22 #include "mutt.h"
23 #include "attach.h"
24
25 /* Modified by blong to accept a "suggestion" for file name.  If
26  * that file exists, then construct one with unique name but 
27  * keep any extension.  This might fail, I guess.
28  * Renamed to mutt_adv_mktemp so I only have to change where it's
29  * called, and not all possible cases.
30  */
31 void mutt_adv_mktemp (const char* dir, char *s, ssize_t l)
32 {
33     int fd;
34
35     fd = m_tempfd(s, l, m_strisempty(dir) ? NONULL(mod_core.tmpdir) : dir, s);
36     if (fd < 0) {
37         *s = '\0';
38     } else {
39         close(fd);
40         unlink(s);
41     }
42 }
43
44 void mutt_mktemp(char *s)
45 {
46     int fd = m_tempfd(s, _POSIX_PATH_MAX, NONULL(mod_core.tmpdir), NULL);
47     if (fd < 0) {
48         *s = '\0';
49     } else {
50         close(fd);
51         unlink(s);
52     }
53 }
54
55 ssize_t _mutt_expand_path(char *buf, ssize_t len, const char *s, int rx)
56 {
57     char p[_POSIX_PATH_MAX] = "";
58     char tmp[_POSIX_PATH_MAX];
59     const char *tail = "";
60     const address_t *alias;
61
62     switch (*s) {
63       case '~':
64         if (s[1] == '/' || s[1] == '\0') {
65             m_strcpy(p, sizeof(p), mod_core.homedir);
66             tail = s + 1;
67         } else {
68             struct passwd *pw;
69             tail = m_strchrnul(s + 1, '/');
70
71             m_strncpy(tmp, sizeof(tmp), s + 1, tail - s - 1);
72
73             if ((pw = getpwnam(tmp))) {
74                 m_strcpy(p, sizeof(p), pw->pw_dir);
75             } else {
76                 /* user not found! */
77                 tail = s;
78             }
79         }
80         break;
81
82       case '=':
83       case '+':
84         /* if folder = imap[s]://host/: don't append slash */
85         if (imap_is_magic(NONULL(Maildir), NULL) == M_IMAP
86         &&  Maildir[m_strlen(Maildir) - 1] == '/') {
87             m_strcpy(p, sizeof(p), Maildir);
88         } else {
89             snprintf(p, sizeof(p), "%s/", NONULL(Maildir));
90         }
91
92         tail = s + 1;
93         break;
94
95         /* elm compatibility, @ expands alias to user name */
96
97       case '@':
98         if ((alias = alias_lookup(s + 1))) {
99             HEADER h;
100             header_init(&h);
101             h.env = envelope_new();
102             h.env->from = h.env->to = (address_t *)alias;
103             mutt_default_save(p, sizeof (p), &h);
104             h.env->from = h.env->to = NULL;
105             header_wipe(&h);
106
107             if (*p != '@')
108                 return _mutt_expand_path(buf, len, p, rx);
109         }
110         break;
111
112       case '>':
113         m_strcpy(p, sizeof(p), Inbox);
114         tail = s + 1;
115         break;
116
117       case '<':
118         m_strcpy(p, sizeof(p), MAlias.record);
119         tail = s + 1;
120         break;
121
122       case '!':
123         if (s[1] == '!') {
124             m_strcpy(p, sizeof(p), LastFolder);
125             tail = s + 2;
126         } else {
127             m_strcpy(p, sizeof(p), Spoolfile);
128             tail = s + 1;
129         }
130         break;
131
132       case '-':
133         m_strcpy(p, sizeof(p), NONULL(LastFolder));
134         tail = s + 1;
135         break;
136
137       case '^':
138         m_strcpy(p, sizeof(p), NONULL(CurrentFolder));
139         tail = s + 1;
140         break;
141
142       default:
143         *p = '\0';
144         tail = s;
145     }
146
147     if (rx) {
148         char q[_POSIX_PATH_MAX];
149         rx_sanitize_string(q, sizeof(q), p);
150         snprintf(tmp, sizeof(tmp), "%s%s", q, tail);
151     } else {
152         snprintf(tmp, sizeof(tmp), "%s%s", p, tail);
153     }
154
155     return m_strcpy(buf, len, tmp);
156 }
157
158 /* collapse the pathname using ~ or = when possible */
159 void mutt_pretty_mailbox (char *s)
160 {
161   char *p = s, *q = s;
162   ssize_t len;
163   url_scheme_t scheme;
164
165   scheme = url_check_scheme (s);
166
167   if (scheme == U_IMAP || scheme == U_IMAPS) {
168     imap_pretty_mailbox (s);
169     return;
170   }
171
172   /* if s is an url, only collapse path component */
173   if (scheme != U_UNKNOWN) {
174     p = strchr (s, ':') + 1;
175     if (!m_strncmp (p, "//", 2))
176       q = strchr (p + 2, '/');
177     if (!q)
178       q = strchr (p, '\0');
179     p = q;
180   }
181
182   /* first attempt to collapse the pathname */
183   while (*p) {
184     if (*p == '/' && p[1] == '/') {
185       *q++ = '/';
186       p += 2;
187     }
188     else if (p[0] == '/' && p[1] == '.' && p[2] == '/') {
189       *q++ = '/';
190       p += 3;
191     }
192     else
193       *q++ = *p++;
194   }
195   *q = 0;
196
197   if (m_strncmp(s, Maildir, (len = m_strlen(Maildir))) == 0 &&
198       s[len] == '/') {
199     *s++ = '=';
200     memmove (s, s + len, m_strlen(s + len) + 1);
201   }
202   else if (m_strncmp(s, mod_core.homedir, (len = m_strlen(mod_core.homedir))) == 0
203            && s[len] == '/') {
204     *s++ = '~';
205     memmove (s, s + len - 1, m_strlen(s + len - 1) + 1);
206   }
207 }
208
209 /* return 0 on success, -1 on abort, 1 on error */
210 int mutt_check_overwrite (const char *attname, const char *path,
211                           char *fname, ssize_t flen, int *append,
212                           char **directory)
213 {
214   int rc = 0;
215   char tmp[_POSIX_PATH_MAX];
216   struct stat st;
217
218   m_strcpy(fname, flen, path);
219   if (access (fname, F_OK) != 0)
220     return 0;
221   if (stat (fname, &st) != 0)
222     return -1;
223   if (S_ISDIR (st.st_mode)) {
224     if (directory) {
225       switch (mutt_multi_choice
226               (_("File is a directory, save under it? [(y)es, (n)o, (a)ll]"),
227                _("yna"))) {
228       case 3:                  /* all */
229         m_strreplace(directory, fname);
230         break;
231       case 1:                  /* yes */
232         p_delete(directory);
233         break;
234       case -1:                 /* abort */
235         p_delete(directory);
236         return -1;
237       case 2:                  /* no */
238         p_delete(directory);
239         return 1;
240       }
241     }
242     else
243       if ((rc = mutt_yesorno(_("File is a directory, save under it?"),
244                              M_YES)) != M_YES)
245       return (rc == M_NO) ? 1 : -1;
246
247     if (!attname || !attname[0]) {
248       tmp[0] = 0;
249       if (mutt_get_field (_("File under directory: "), tmp, sizeof (tmp),
250                           M_FILE | M_CLEAR) != 0 || !tmp[0])
251         return (-1);
252       mutt_concat_path(fname, flen, path, tmp);
253     }
254     else
255       mutt_concat_path(fname, flen, path, mutt_basename(attname));
256   }
257
258   if (*append == 0 && access (fname, F_OK) == 0) {
259     switch (mutt_multi_choice
260             (_("File exists, (o)verwrite, (a)ppend, or (c)ancel?"), _("oac")))
261     {
262     case -1:                   /* abort */
263       return -1;
264     case 3:                    /* cancel */
265       return 1;
266
267     case 2:                    /* append */
268       *append = M_SAVE_APPEND;
269       break;
270     case 1:                    /* overwrite */
271       *append = M_SAVE_OVERWRITE;
272       break;
273     }
274   }
275   return 0;
276 }
277
278 void mutt_save_path(char *d, ssize_t dsize, address_t *a)
279 {
280     if (a && a->mailbox) {
281         m_strncpy(d, dsize, a->mailbox, strcspn(d, "%@"));
282         m_strtolower(d);
283     } else {
284         *d = '\0';
285     }
286 }
287
288 void mutt_safe_path(char *s, ssize_t l, address_t *a)
289 {
290     mutt_save_path(s, l, a);
291
292     while (*s) {
293         if (*s == '/' || ISSPACE(*s) || !isprint((unsigned char)*s))
294             *s = '_';
295         s++;
296     }
297 }
298
299 /* returns 0 if OK to proceed, -1 to abort, 1 to retry */
300 int mutt_save_confirm (const char *s, struct stat *st)
301 {
302   char tmp[_POSIX_PATH_MAX];
303   int ret = 0;
304   int rc;
305   int magic = 0;
306
307   magic = mx_get_magic (s);
308
309   if (magic == M_POP) {
310     mutt_error _("Can't save message to POP mailbox.");
311
312     return 1;
313   }
314
315 #ifdef USE_NNTP
316   if (magic == M_NNTP) {
317     mutt_error _("Can't save message to newsserver.");
318
319     return 0;
320   }
321 #endif
322
323   if (magic > 0 && !mx_access (s, W_OK)) {
324     if (option (OPTCONFIRMAPPEND) &&
325         (!TrashPath || (m_strcmp(s, TrashPath) != 0))) {
326       /* if we're appending to the trash, there's no point in asking */
327       snprintf (tmp, sizeof (tmp), _("Append messages to %s?"), s);
328       if ((rc = mutt_yesorno (tmp, M_YES)) == M_NO)
329         ret = 1;
330       else if (rc == -1)
331         ret = -1;
332     }
333   }
334
335   if (stat (s, st) != -1) {
336     if (magic == -1) {
337       mutt_error (_("%s is not a mailbox!"), s);
338       return 1;
339     }
340   } else {
341     if (magic != M_IMAP)
342     {
343       st->st_mtime = 0;
344       st->st_atime = 0;
345
346       if (errno == ENOENT) {
347         if (option (OPTCONFIRMCREATE)) {
348           snprintf (tmp, sizeof (tmp), _("Create %s?"), s);
349           if ((rc = mutt_yesorno (tmp, M_YES)) == M_NO)
350             ret = 1;
351           else if (rc == -1)
352             ret = -1;
353         }
354       } else {
355         mutt_perror (s);
356         return 1;
357       }
358     }
359   }
360
361   CLEARLINE (LINES - 1);
362   return (ret);
363 }
364
365 void mutt_sleep (short s)
366 {
367     sleep(MAX(s, SleepTime));
368 }
369
370 const char *mutt_make_version(void)
371 {
372   static char vstring[STRING];
373   snprintf(vstring, sizeof (vstring), "Madmutt/%s", MUTT_VERSION);
374   return vstring;
375 }
376
377 /* return 1 if address lists are strictly identical */
378 static int mutt_cmp_addr (const address_t * a, const address_t * b)
379 {
380   while (a && b) {
381     if (m_strcmp(a->mailbox, b->mailbox) ||
382         m_strcmp(a->personal, b->personal))
383       return (0);
384
385     a = a->next;
386     b = b->next;
387   }
388   if (a || b)
389     return (0);
390
391   return (1);
392 }
393
394 static int mutt_cmp_list (const string_list_t * a, const string_list_t * b)
395 {
396   while (a && b) {
397     if (m_strcmp(a->data, b->data))
398       return (0);
399
400     a = a->next;
401     b = b->next;
402   }
403   if (a || b)
404     return (0);
405
406   return (1);
407 }
408
409 static int mutt_cmp_env (const ENVELOPE * e1, const ENVELOPE * e2)
410 {
411   if (e1 && e2) {
412     if (m_strcmp(e1->message_id, e2->message_id) ||
413         m_strcmp(e1->subject, e2->subject) ||
414         !mutt_cmp_list (e1->references, e2->references) ||
415         !mutt_cmp_addr (e1->from, e2->from) ||
416         !mutt_cmp_addr (e1->sender, e2->sender) ||
417         !mutt_cmp_addr (e1->reply_to, e2->reply_to) ||
418         !mutt_cmp_addr (e1->to, e2->to) ||
419         !mutt_cmp_addr (e1->cc, e2->cc) ||
420         !mutt_cmp_addr (e1->return_path, e2->return_path))
421       return (0);
422     else
423       return (1);
424   }
425   else {
426     if (e1 == NULL && e2 == NULL)
427       return (1);
428     else
429       return (0);
430   }
431 }
432
433 static int mutt_cmp_body (const BODY * b1, const BODY * b2)
434 {
435   if (b1->type != b2->type ||
436       b1->encoding != b2->encoding ||
437       m_strcmp(b1->subtype, b2->subtype) ||
438       m_strcmp(b1->description, b2->description) ||
439       !parameter_equal(b1->parameter, b2->parameter) ||
440       b1->length != b2->length)
441     return (0);
442   return (1);
443 }
444 int mutt_cmp_header (const HEADER * h1, const HEADER * h2) {
445   if (h1 && h2) {
446     if (h1->received != h2->received ||
447         h1->date_sent != h2->date_sent ||
448         h1->content->length != h2->content->length ||
449         h1->lines != h2->lines ||
450         h1->zhours != h2->zhours ||
451         h1->zminutes != h2->zminutes ||
452         h1->zoccident != h2->zoccident ||
453         h1->mime != h2->mime ||
454         !mutt_cmp_env (h1->env, h2->env) ||
455         !mutt_cmp_body (h1->content, h2->content))
456       return (0);
457     else
458       return (1);
459   }
460   else {
461     if (h1 == NULL && h2 == NULL)
462       return (1);
463     else
464       return (0);
465   }
466 }
467
468
469 int mutt_extract_token(BUFFER *dest, BUFFER *tok, int flags)
470 {
471     char ch;
472     char qc = 0;                  /* quote char */
473     char *pc;
474
475     /* reset the destination pointer to the beginning of the buffer */
476     dest->dptr = dest->data;
477
478     tok->dptr = vskipspaces(tok->dptr);
479     while ((ch = *tok->dptr)) {
480         if (!qc) {
481             if ((ISSPACE(ch) && !(flags & M_TOKEN_SPACE))
482             || (ch == '#' && !(flags & M_TOKEN_COMMENT))
483             || (ch == '=' && (flags & M_TOKEN_EQUAL))
484             || ((flags & M_TOKEN_PATTERN) && strchr("~=!|", ch)))
485             {
486                 break;
487             }
488         }
489
490         tok->dptr++;
491
492         if (ch == qc) {
493             qc = 0;                     /* end of quote */
494         } else
495         if (!qc && (ch == '\'' || ch == '"') && !(flags & M_TOKEN_QUOTE)) {
496             qc = ch;
497         } else
498         if (ch == '\\' && qc != '\'') {
499             if (!*tok->dptr)
500                 return -1;              /* premature end of token */
501
502             switch (ch = *tok->dptr++) {
503               case 'c':
504               case 'C':
505                 if (!*tok->dptr)
506                     return -1;          /* premature end of token */
507                 mutt_buffer_addch(dest,
508                                   (toupper((unsigned char)*tok->dptr) - 'A' + 1) & 0x7f);
509                 tok->dptr++;
510                 break;
511               case 'r':
512                 mutt_buffer_addch(dest, '\r');
513                 break;
514               case 'n':
515                 mutt_buffer_addch(dest, '\n');
516                 break;
517               case 't':
518                 mutt_buffer_addch(dest, '\t');
519                 break;
520               case 'f':
521                 mutt_buffer_addch(dest, '\f');
522                 break;
523               case 'e':
524                 mutt_buffer_addch(dest, '\033');
525                 break;
526               default:
527                 if (isdigit((unsigned char)ch)
528                 &&  isdigit((unsigned char)*tok->dptr)
529                 &&  isdigit((unsigned char)*(tok->dptr + 1)))
530                 {
531                     mutt_buffer_addch(dest, (ch << 6) + (*tok->dptr << 3) +
532                                             *(tok->dptr + 1) - 3504);
533                     tok->dptr += 2;
534                 } else {
535                     mutt_buffer_addch(dest, ch);
536                 }
537             }
538         } else
539         if (ch == '^' && (flags & M_TOKEN_CONDENSE)) {
540             if (!*tok->dptr)
541                 return -1;              /* premature end of token */
542             ch = *tok->dptr++;
543             if (ch == '^') {
544                 mutt_buffer_addch(dest, ch);
545             } else
546             if (ch == '[') {
547                 mutt_buffer_addch(dest, '\033');
548             } else
549             if (isalpha((unsigned char)ch)) {
550                 mutt_buffer_addch(dest, toupper((unsigned char)ch) - 'A' + 1);
551             } else {
552                 mutt_buffer_addch(dest, '^');
553                 mutt_buffer_addch(dest, ch);
554             }
555         } else
556         if (ch == '`' && (!qc || qc == '"')) {
557             FILE *fp;
558             pid_t pid;
559             char *cmd, *ptr;
560             ssize_t expnlen;
561             BUFFER expn;
562             int line = 0;
563
564             pc = tok->dptr;
565             do {
566                 if ((pc = strpbrk(pc, "\\`"))) {
567                     /* skip any quoted chars */
568                     if (*pc == '\\')
569                         pc += 2;
570                 }
571             } while (pc && *pc != '`');
572             if (!pc) {
573                 return (-1);
574             }
575
576             cmd = p_dupstr(tok->dptr, pc - tok->dptr);
577             if ((pid = mutt_create_filter(cmd, NULL, &fp, NULL)) < 0) {
578                 p_delete(&cmd);
579                 return -1;
580             }
581             p_delete(&cmd);
582
583             tok->dptr = pc + 1;
584
585             /* read line */
586             p_clear(&expn, 1);
587             expn.data = mutt_read_line(NULL, &expn.dsize, fp, &line);
588             m_fclose(&fp);
589             mutt_wait_filter(pid);
590
591             /* if we got output, make a new string consiting of the shell ouptput
592                plus whatever else was left on the original line */
593             /* BUT: If this is inside a quoted string, directly add output to 
594              * the token */
595             if (expn.data && qc) {
596                 mutt_buffer_addstr(dest, expn.data);
597                 p_delete(&expn.data);
598             } else
599             if (expn.data) {
600                 expnlen = m_strlen(expn.data);
601                 tok->dsize = expnlen + m_strlen(tok->dptr) + 1;
602                 ptr = xmalloc(tok->dsize);
603                 memcpy(ptr, expn.data, expnlen);
604                 m_strcpy(ptr + expnlen, tok->dsize - expnlen, tok->dptr);
605                 if (tok->destroy)
606                     p_delete(&tok->data);
607                 tok->data = ptr;
608                 tok->dptr = ptr;
609                 tok->destroy = 1;       /* mark that the caller should destroy this data */
610                 ptr = NULL;
611                 p_delete(&expn.data);
612             }
613         } else
614         if (ch == '$' && (!qc || qc == '"')
615         && (*tok->dptr == '{' || isalpha((unsigned char)*tok->dptr)))
616         {
617             char *env = NULL, *var = NULL;
618
619             if (*tok->dptr == '{') {
620                 tok->dptr++;
621                 if ((pc = strchr (tok->dptr, '}'))) {
622                     var = p_dupstr(tok->dptr, pc - tok->dptr);
623                     tok->dptr = pc + 1;
624                 }
625             } else {
626                 for (pc = tok->dptr; isalnum((unsigned char)*pc) || *pc == '_';
627                      pc++);
628                 var = p_dupstr(tok->dptr, pc - tok->dptr);
629                 tok->dptr = pc;
630             }
631             if (var) {
632                 char tmp[STRING];
633                 if ((env = getenv (var))
634                 || (mutt_option_value (var, tmp, sizeof (tmp)) && (env = tmp)))
635                 {
636                     mutt_buffer_addstr (dest, env);
637                 }
638             }
639             p_delete(&var);
640         } else {
641             mutt_buffer_addch(dest, ch);
642         }
643     }
644     mutt_buffer_addch(dest, 0);  /* terminate the string */
645     tok->dptr = vskipspaces(tok->dptr);
646     return 0;
647 }