Preparing an interim release
[gnupg.git] / tools / rfc822parse.c
1 /* rfc822parse.c - Simple mail and MIME parser
2  *      Copyright (C) 1999, 2000 Werner Koch, Duesseldorf
3  *      Copyright (C) 2003, 2004 g10 Code GmbH
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public License
7  * as published by the Free Software Foundation; either version 2.1 of
8  * the License, or (at your option) any later version.
9  * 
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13  * GNU Lesser General Public License for more details.
14  * 
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
18  * 02110-1301, USA.
19  */
20
21
22 /* According to RFC822 binary zeroes are allowed at many places. We do
23  * not handle this correct especially in the field parsing code.  It
24  * should be easy to fix and the API provides a interfaces which
25  * returns the length but in addition makes sure that returned strings
26  * are always ended by a \0.
27  *
28  * Furthermore, the case of field names is changed and thus it is not
29  * always a good idea to use these modified header
30  * lines (e.g. signatures may break).
31  */
32
33 #ifdef HAVE_CONFIG_H
34 #include <config.h>
35 #endif
36
37 #include <stdlib.h>
38 #include <stdio.h>
39 #include <string.h>
40 #include <errno.h>
41 #include <stdarg.h>
42 #include <assert.h>
43
44 #include "rfc822parse.h"
45
46 enum token_type
47   {
48     tSPACE,
49     tATOM,
50     tQUOTED,
51     tDOMAINLIT,
52     tSPECIAL
53   };
54
55 /* For now we directly use our TOKEN as the parse context */
56 typedef struct rfc822parse_field_context *TOKEN;
57 struct rfc822parse_field_context
58 {
59   TOKEN next;
60   enum token_type type;
61   struct {
62     unsigned int cont:1;
63     unsigned int lowered:1;
64   } flags;
65   /*TOKEN owner_pantry; */
66   char data[1];
67 };
68
69 struct hdr_line
70 {
71   struct hdr_line *next;
72   int cont;     /* This is a continuation of the previous line. */
73   unsigned char line[1];
74 };
75
76 typedef struct hdr_line *HDR_LINE;
77
78
79 struct part
80 {
81   struct part *right;     /* The next part. */
82   struct part *down;      /* A contained part. */
83   HDR_LINE hdr_lines;       /* Header lines os that part. */
84   HDR_LINE *hdr_lines_tail; /* Helper for adding lines. */
85   char *boundary;           /* Only used in the first part. */
86 };
87 typedef struct part *part_t;
88
89 struct rfc822parse_context
90 {
91   rfc822parse_cb_t callback;
92   void *callback_value;
93   int callback_error;
94   int in_body;
95   int in_preamble;      /* Wether we are before the first boundary. */
96   part_t parts;         /* The tree of parts. */
97   part_t current_part;  /* Whom we are processing (points into parts). */
98   const char *boundary; /* Current boundary. */
99 };
100
101 static HDR_LINE find_header (rfc822parse_t msg, const char *name,
102                              int which, HDR_LINE * rprev);
103
104
105 static size_t
106 length_sans_trailing_ws (const unsigned char *line, size_t len)
107 {
108   const unsigned char *p, *mark;
109   size_t n;
110   
111   for (mark=NULL, p=line, n=0; n < len; n++, p++)
112     {
113       if (strchr (" \t\r\n", *p ))
114         {
115           if( !mark )
116             mark = p;
117         }
118       else
119         mark = NULL;
120     }
121   
122   if (mark) 
123     return mark - line;
124   return len;
125 }
126
127
128 static void
129 lowercase_string (unsigned char *string)
130 {
131   for (; *string; string++)
132     if (*string >= 'A' && *string <= 'Z')
133       *string = *string - 'A' + 'a';
134 }
135
136 /* Transform a header name into a standard capitalized format; i.e
137    "Content-Type".  Conversion stops at the colon.  As usual we don't
138    use the localized versions of ctype.h.
139  */
140 static void
141 capitalize_header_name (unsigned char *name)
142 {
143   int first = 1;
144
145   for (; *name && *name != ':'; name++)
146     if (*name == '-')
147       first = 1;
148     else if (first)
149       {
150         if (*name >= 'a' && *name <= 'z')
151           *name = *name - 'a' + 'A';
152         first = 0;
153       }
154     else if (*name >= 'A' && *name <= 'Z')
155       *name = *name - 'A' + 'a';
156 }
157
158
159 static char *
160 stpcpy (char *a,const char *b)
161 {
162   while (*b)
163     *a++ = *b++;
164   *a = 0;
165   
166   return (char*)a;
167 }
168
169
170 /* If a callback has been registerd, call it for the event of type
171    EVENT. */
172 static int
173 do_callback (rfc822parse_t msg, rfc822parse_event_t event)
174 {
175   int rc;
176
177   if (!msg->callback || msg->callback_error)
178     return 0;
179   rc = msg->callback (msg->callback_value, event, msg);
180   if (rc)
181     msg->callback_error = rc;
182   return rc;
183 }
184
185 static part_t
186 new_part (void)
187 {
188   part_t part;
189
190   part = calloc (1, sizeof *part);
191   if (part)
192     {
193       part->hdr_lines_tail = &part->hdr_lines;
194     }
195   return part;
196 }
197
198
199 static void
200 release_part (part_t part)
201 {
202   part_t tmp;
203   HDR_LINE hdr, hdr2;
204
205   for (; part; part = tmp)
206     {
207       tmp = part->right;
208       if (part->down)
209         release_part (part->down);
210       for (hdr = part->hdr_lines; hdr; hdr = hdr2)
211         {
212           hdr2 = hdr->next;
213           free (hdr);
214         }
215       free (part->boundary);
216       free (part);
217     }
218 }
219
220
221 static void
222 release_handle_data (rfc822parse_t msg)
223 {
224   release_part (msg->parts);
225   msg->parts = NULL;
226   msg->current_part = NULL;
227   msg->boundary = NULL;
228 }
229
230
231 /* Create a new parsing context for an entire rfc822 message and
232    return it.  CB and CB_VALUE may be given to callback for certain
233    events.  NULL is returned on error with errno set appropriately. */
234 rfc822parse_t
235 rfc822parse_open (rfc822parse_cb_t cb, void *cb_value)
236 {
237   rfc822parse_t msg = calloc (1, sizeof *msg);
238   if (msg)
239     {
240       msg->parts = msg->current_part = new_part ();
241       if (!msg->parts)
242         {
243           free (msg);
244           msg = NULL;
245         }
246       else
247         {
248           msg->callback = cb;
249           msg->callback_value = cb_value;
250           if (do_callback (msg, RFC822PARSE_OPEN))
251             {
252               release_handle_data (msg);
253               free (msg);
254               msg = NULL;
255             }
256         }
257     }
258   return msg;
259 }
260
261
262 void
263 rfc822parse_cancel (rfc822parse_t msg)
264 {
265   if (msg)
266     {
267       do_callback (msg, RFC822PARSE_CANCEL);
268       release_handle_data (msg);
269       free (msg);
270     }
271 }
272
273
274 void
275 rfc822parse_close (rfc822parse_t msg)
276 {
277   if (msg)
278     {
279       do_callback (msg, RFC822PARSE_CLOSE);
280       release_handle_data (msg);
281       free (msg);
282     }
283 }
284
285 static part_t
286 find_parent (part_t tree, part_t target)
287 {
288   part_t part;
289
290   for (part = tree->down; part; part = part->right)
291     {
292       if (part == target)
293         return tree; /* Found. */
294       if (part->down)
295         {
296           part_t tmp = find_parent (part, target);
297           if (tmp)
298             return tmp;
299         }
300     }
301   return NULL;
302 }
303
304 static void
305 set_current_part_to_parent (rfc822parse_t msg)
306 {
307   part_t parent;
308
309   assert (msg->current_part);
310   parent = find_parent (msg->parts, msg->current_part);
311   if (!parent)
312     return; /* Already at the top. */
313
314 #ifndef NDEBUG
315   {
316     part_t part;
317     for (part = parent->down; part; part = part->right)
318       if (part == msg->current_part)
319         break;
320     assert (part);
321   }
322 #endif
323   msg->current_part = parent;
324
325   parent = find_parent (msg->parts, parent);
326   msg->boundary = parent? parent->boundary: NULL;
327 }
328
329
330
331 /****************
332  * We have read in all header lines and are about to receive the body
333  * part.  The delimiter line has already been processed.
334  *
335  * FIXME: we's better return an error in case of memory failures.
336  */
337 static int
338 transition_to_body (rfc822parse_t msg)
339 {
340   rfc822parse_field_t ctx;
341   int rc;
342
343   rc = do_callback (msg, RFC822PARSE_T2BODY);
344   if (!rc)
345     {
346       /* Store the boundary if we have multipart type. */
347       ctx = rfc822parse_parse_field (msg, "Content-Type", -1);
348       if (ctx)
349         {
350           const char *s;
351
352           s = rfc822parse_query_media_type (ctx, NULL);
353           if (s && !strcmp (s,"multipart"))
354             {
355               s = rfc822parse_query_parameter (ctx, "boundary", 0);
356               if (s)
357                 {
358                   assert (!msg->current_part->boundary);
359                   msg->current_part->boundary = malloc (strlen (s) + 1);
360                   if (msg->current_part->boundary) 
361                     {
362                       part_t part;
363                   
364                       strcpy (msg->current_part->boundary, s);
365                       msg->boundary = msg->current_part->boundary;
366                       part = new_part ();
367                       if (!part)
368                         {
369                           int save_errno = errno;
370                           rfc822parse_release_field (ctx);
371                           errno = save_errno;
372                           return -1;
373                         }
374                       rc = do_callback (msg, RFC822PARSE_LEVEL_DOWN);
375                       assert (!msg->current_part->down);
376                       msg->current_part->down = part;
377                       msg->current_part = part;
378                       msg->in_preamble = 1;
379                     }
380                 }
381             }
382           rfc822parse_release_field (ctx);
383         }
384     }
385
386   return rc;
387 }
388
389 /* We have just passed a MIME boundary and need to prepare for new part.
390    headers. */
391 static int
392 transition_to_header (rfc822parse_t msg)
393 {
394   part_t part;
395
396   assert (msg->current_part);
397   assert (!msg->current_part->right);
398
399   part = new_part ();
400   if (!part)
401     return -1;
402
403   msg->current_part->right = part;
404   msg->current_part = part;
405   return 0;
406 }
407
408
409 static int
410 insert_header (rfc822parse_t msg, const unsigned char *line, size_t length)
411 {
412   HDR_LINE hdr;
413
414   assert (msg->current_part);
415   if (!length)
416     {
417       msg->in_body = 1;
418       return transition_to_body (msg);
419     }
420
421   if (!msg->current_part->hdr_lines)
422     do_callback (msg, RFC822PARSE_BEGIN_HEADER);
423
424   length = length_sans_trailing_ws (line, length);
425   hdr = malloc (sizeof (*hdr) + length);
426   if (!hdr)
427     return -1;
428   hdr->next = NULL;
429   hdr->cont = (*line == ' ' || *line == '\t');
430   memcpy (hdr->line, line, length);
431   hdr->line[length] = 0; /* Make it a string. */
432   
433   /* Transform a field name into canonical format. */
434   if (!hdr->cont && strchr (line, ':'))
435      capitalize_header_name (hdr->line);
436
437   *msg->current_part->hdr_lines_tail = hdr;
438   msg->current_part->hdr_lines_tail = &hdr->next;
439
440   /* Lets help the caller to prevent mail loops and issue an event for
441    * every Received header. */
442   if (length >= 9 && !memcmp (line, "Received:", 9))
443      do_callback (msg, RFC822PARSE_RCVD_SEEN);
444   return 0;
445 }
446
447
448 /****************
449  * Note: We handle the body transparent to allow binary zeroes in it.
450  */
451 static int
452 insert_body (rfc822parse_t msg, const unsigned char *line, size_t length)
453 {
454   int rc = 0;
455
456   if (length > 2 && *line == '-' && line[1] == '-' && msg->boundary)
457     {
458       size_t blen = strlen (msg->boundary);
459
460       if (length == blen + 2
461           && !memcmp (line+2, msg->boundary, blen))
462         {
463           rc = do_callback (msg, RFC822PARSE_BOUNDARY);
464           msg->in_body = 0;
465           if (!rc && !msg->in_preamble)
466             rc = transition_to_header (msg);
467           msg->in_preamble = 0;
468         }
469       else if (length == blen + 4
470           && line[length-2] =='-' && line[length-1] == '-'
471           && !memcmp (line+2, msg->boundary, blen))
472         {
473           rc = do_callback (msg, RFC822PARSE_LAST_BOUNDARY);
474           msg->boundary = NULL; /* No current boundary anymore. */
475           set_current_part_to_parent (msg);
476
477           /* Fixme: The next should acctually be sent right before the
478              next boundary, so that we can mark the epilogue. */
479           if (!rc)
480             rc = do_callback (msg, RFC822PARSE_LEVEL_UP);
481         }
482     }
483   if (msg->in_preamble && !rc)
484     rc = do_callback (msg, RFC822PARSE_PREAMBLE);
485
486   return rc;
487 }
488
489 /* Insert the next line into the parser. Return 0 on success or true
490    on error with errno set appropriately. */
491 int
492 rfc822parse_insert (rfc822parse_t msg, const unsigned char *line, size_t length)
493 {
494   return (msg->in_body 
495           ? insert_body (msg, line, length)
496           : insert_header (msg, line, length));
497 }
498
499
500 /* Tell the parser that we have finished the message. */
501 int
502 rfc822parse_finish (rfc822parse_t msg)
503 {
504   return do_callback (msg, RFC822PARSE_FINISH);
505 }
506
507
508
509 /****************
510  * Get a copy of a header line. The line is returned as one long
511  * string with LF to separate the continuation line. Caller must free
512  * the return buffer.  WHICH may be used to enumerate over all lines.
513  * Wildcards are allowed.  This function works on the current headers;
514  * i.e. the regular mail headers or the MIME headers of the current
515  * part.
516  *
517  * WHICH gives the mode:
518  *  -1 := Take the last occurence
519  *   n := Take the n-th  one.
520  * 
521  * Returns a newly allocated buffer or NULL on error.  errno is set in
522  * case of a memory failure or set to 0 if the requested field is not
523  * available.
524  * 
525  * If VALUEOFF is not NULL it will receive the offset of the first non
526  * space character in th value of the line.
527  */
528 char *
529 rfc822parse_get_field (rfc822parse_t msg, const char *name, int which,
530                        size_t *valueoff)
531 {
532   HDR_LINE h, h2;
533   char *buf, *p;
534   size_t n;
535
536   h = find_header (msg, name, which, NULL);
537   if (!h)
538     {
539       errno = 0;
540       return NULL; /* no such field */
541     }
542
543   n = strlen (h->line) + 1;
544   for (h2 = h->next; h2 && h2->cont; h2 = h2->next)
545     n += strlen (h2->line) + 1;
546
547   buf = p = malloc (n);
548   if (buf)
549     {
550       p = stpcpy (p, h->line);
551       *p++ = '\n';
552       for (h2 = h->next; h2 && h2->cont; h2 = h2->next)
553         {
554           p = stpcpy (p, h2->line);
555           *p++ = '\n';
556         }
557       p[-1] = 0;
558     }
559
560   if (valueoff)
561     {
562       p = strchr (buf, ':');
563       if (!p)
564         *valueoff = 0; /* Oops: should never happen. */
565       else
566         {
567           p++;
568           while (*p == ' ' || *p == '\t' || *p == '\r' || *p == '\n')
569             p++;
570           *valueoff = p - buf;
571         }
572     }
573
574   return buf;
575 }
576
577
578 /****************
579  * Enumerate all header.  Caller has to provide the address of a pointer
580  * which has to be initialzed to NULL, the caller should then never change this
581  * pointer until he has closed the enumeration by passing again the address
582  * of the pointer but with msg set to NULL.
583  * The function returns pointers to all the header lines or NULL when
584  * all lines have been enumerated or no headers are available.
585  */
586 const char *
587 rfc822parse_enum_header_lines (rfc822parse_t msg, void **context)
588 {
589   HDR_LINE l;
590
591   if (!msg) /* Close. */
592     return NULL;        
593
594   if (*context == msg || !msg->current_part)
595     return NULL;
596
597   l = *context ? (HDR_LINE) *context : msg->current_part->hdr_lines;
598
599   if (l)
600     {
601       *context = l->next ? (void *) (l->next) : (void *) msg;
602       return l->line;
603     }
604   *context = msg; /* Mark end of list. */
605   return NULL;
606 }
607
608
609
610 /****************
611  * Find a header field.  If the Name does end in an asterisk this is meant
612  * to be a wildcard.
613  *
614  *  which  -1 : Retrieve the last field
615  *         >0 : Retrieve the n-th field
616
617  * RPREV may be used to return the predecessor of the returned field;
618  * which may be NULL for the very first one. It has to be initialzed
619  * to either NULL in which case the search start at the first header line,
620  * or it may point to a headerline, where the search should start
621  */
622 static HDR_LINE
623 find_header (rfc822parse_t msg, const char *name, int which, HDR_LINE *rprev)
624 {
625   HDR_LINE hdr, prev = NULL, mark = NULL;
626   unsigned char *p;
627   size_t namelen, n;
628   int found = 0;
629   int glob = 0;
630
631   if (!msg->current_part)
632     return NULL;
633
634   namelen = strlen (name);
635   if (namelen && name[namelen - 1] == '*')
636     {
637       namelen--;
638       glob = 1;
639     }
640
641   hdr = msg->current_part->hdr_lines;
642   if (rprev && *rprev)
643     {
644       /* spool forward to the requested starting place.
645        * we cannot simply set this as we have to return
646        * the previous list element too */
647       for (; hdr && hdr != *rprev; prev = hdr, hdr = hdr->next)
648         ;
649     }
650
651   for (; hdr; prev = hdr, hdr = hdr->next)
652     {
653       if (hdr->cont)
654         continue;
655       if (!(p = strchr (hdr->line, ':')))
656         continue;               /* invalid header, just skip it. */
657       n = p - hdr->line;
658       if (!n)
659         continue;               /* invalid name */
660       if ((glob ? (namelen <= n) : (namelen == n))
661           && !memcmp (hdr->line, name, namelen))
662         {
663           found++;
664           if (which == -1)
665             mark = hdr;
666           else if (found == which)
667             {
668               if (rprev)
669                 *rprev = prev;
670               return hdr;
671             }
672         }
673     }
674   if (mark && rprev)
675     *rprev = prev;
676   return mark;
677 }
678
679
680
681 static const char *
682 skip_ws (const char *s)
683 {
684   while (*s == ' ' || *s == '\t' || *s == '\r' || *s == '\n')
685     s++;
686   return s;
687 }
688
689
690 static void
691 release_token_list (TOKEN t)
692 {
693   while (t)
694     {
695       TOKEN t2 = t->next;
696       /* fixme: If we have owner_pantry, put the token back to
697        * this pantry so that it can be reused later */
698       free (t);
699       t = t2;
700     }
701 }
702
703
704 static TOKEN
705 new_token (enum token_type type, const char *buf, size_t length)
706 {
707   TOKEN t;
708
709   /* fixme: look through our pantries to find a suitable
710    * token for reuse */
711   t = malloc (sizeof *t + length);
712   if (t)
713     {
714       t->next = NULL;
715       t->type = type;
716       memset (&t->flags, 0, sizeof (t->flags));
717       t->data[0] = 0;
718       if (buf)
719         {
720           memcpy (t->data, buf, length);
721           t->data[length] = 0;  /* Make sure it is a C string. */
722         }
723       else
724         t->data[0] = 0;
725     }
726   return t;
727 }
728
729 static TOKEN
730 append_to_token (TOKEN old, const char *buf, size_t length)
731 {
732   size_t n = strlen (old->data);
733   TOKEN t;
734
735   t = malloc (sizeof *t + n + length);
736   if (t)
737     {
738       t->next = old->next;
739       t->type = old->type;
740       t->flags = old->flags;
741       memcpy (t->data, old->data, n);
742       memcpy (t->data + n, buf, length);
743       t->data[n + length] = 0;
744       old->next = NULL;
745       release_token_list (old);
746     }
747   return t;
748 }
749
750
751
752 /*
753    Parse a field into tokens as defined by rfc822.
754  */
755 static TOKEN
756 parse_field (HDR_LINE hdr)
757 {
758   static const char specials[] = "<>@.,;:\\[]\"()";
759   static const char specials2[] = "<>@.,;:";
760   static const char tspecials[] = "/?=<>@,;:\\[]\"()";
761   static const char tspecials2[] = "/?=<>@.,;:";
762   static struct 
763   {
764     const unsigned char *name;
765     size_t namelen;
766   } tspecial_header[] = {
767     { "Content-Type", 12},
768     { "Content-Transfer-Encoding", 25},
769     { "Content-Disposition", 19},
770     { NULL, 0}
771   };
772   const char *delimiters;
773   const char *delimiters2;
774   const unsigned char *line, *s, *s2;
775   size_t n;
776   int i, invalid = 0;
777   TOKEN t, tok, *tok_tail;
778
779   errno = 0;
780   if (!hdr)
781     return NULL;
782
783   tok = NULL;
784   tok_tail = &tok;
785
786   line = hdr->line;
787   if (!(s = strchr (line, ':')))
788     return NULL; /* oops */
789
790   n = s - line;
791   if (!n)
792     return NULL; /* oops: invalid name */
793
794   delimiters = specials;
795   delimiters2 = specials2;
796   for (i = 0; tspecial_header[i].name; i++)
797     {
798       if (n == tspecial_header[i].namelen
799           && !memcmp (line, tspecial_header[i].name, n))
800         {
801           delimiters = tspecials;
802           delimiters2 = tspecials2;
803           break;
804         }
805     }
806
807   s++; /* Move over the colon. */
808   for (;;)
809     {
810       if (!*s)
811         {
812           if (!hdr->next || !hdr->next->cont)
813             break;
814           hdr = hdr->next;
815           s = hdr->line;
816         }
817
818       if (*s == '(')
819         {
820           int level = 1;
821           int in_quote = 0;
822
823           invalid = 0;
824           for (s++;; s++)
825             {
826               if (!*s)
827                 {
828                   if (!hdr->next || !hdr->next->cont)
829                     break;
830                   hdr = hdr->next;
831                   s = hdr->line;
832                 }
833
834               if (in_quote)
835                 {
836                   if (*s == '\"')
837                     in_quote = 0;
838                   else if (*s == '\\' && s[1])  /* what about continuation? */
839                     s++;
840                 }
841               else if (*s == ')')
842                 {
843                   if (!--level)
844                     break;
845                 }
846               else if (*s == '(')
847                 level++;
848               else if (*s == '\"')
849                 in_quote = 1;
850             }
851           if (!*s)
852             ; /* Actually this is an error, but we don't care about it. */
853           else
854             s++;
855         }
856       else if (*s == '\"' || *s == '[')
857         {
858           /* We do not check for non-allowed nesting of domainliterals */
859           int term = *s == '\"' ? '\"' : ']';
860           invalid = 0;
861           s++;
862           t = NULL;
863
864           for (;;)
865             {
866               for (s2 = s; *s2; s2++)
867                 {
868                   if (*s2 == term)
869                     break;
870                   else if (*s2 == '\\' && s2[1]) /* what about continuation? */
871                     s2++;
872                 }
873               
874               t = (t
875                    ? append_to_token (t, s, s2 - s)
876                    : new_token (term == '\"'? tQUOTED : tDOMAINLIT, s, s2 - s));
877               if (!t)
878                 goto failure;
879                    
880               if (*s2 || !hdr->next || !hdr->next->cont)
881                 break;
882               hdr = hdr->next;
883               s = hdr->line;
884             }
885           *tok_tail = t;
886           tok_tail = &t->next;
887           s = s2;
888           if (*s)
889             s++; /* skip the delimiter */
890         }
891       else if ((s2 = strchr (delimiters2, *s)))
892         { /* Special characters which are not handled above. */
893           invalid = 0;
894           t = new_token (tSPECIAL, s, 1);
895           if (!t)
896             goto failure;
897           *tok_tail = t;
898           tok_tail = &t->next;
899           s++;
900         }
901       else if (*s == ' ' || *s == '\t' || *s == '\r' || *s == '\n')
902         {
903           invalid = 0;
904           s = skip_ws (s + 1);
905         }
906       else if (*s > 0x20 && !(*s & 128))
907         { /* Atom. */
908           invalid = 0;
909           for (s2 = s + 1; *s2 > 0x20
910                && !(*s2 & 128) && !strchr (delimiters, *s2); s2++)
911             ;
912           t = new_token (tATOM, s, s2 - s);
913           if (!t)
914             goto failure;
915           *tok_tail = t;
916           tok_tail = &t->next;
917           s = s2;
918         }
919       else
920         { /* Invalid character. */
921           if (!invalid)
922             { /* For parsing we assume only one space. */
923               t = new_token (tSPACE, NULL, 0);
924               if (!t)
925                 goto failure;
926               *tok_tail = t;
927               tok_tail = &t->next;
928               invalid = 1;
929             }
930           s++;
931         }
932     }
933
934   return tok;
935
936  failure:
937   {
938     int save = errno;
939     release_token_list (tok);
940     errno = save;
941   }
942   return NULL;
943 }
944
945
946
947
948 /****************
949  * Find and parse a header field.
950  * WHICH indicates what to do if there are multiple instance of the same
951  * field (like "Received"); the following value are defined:
952  *  -1 := Take the last occurence
953  *   0 := Reserved
954  *   n := Take the n-th one.
955  * Returns a handle for further operations on the parse context of the field
956  * or NULL if the field was not found.
957  */
958 rfc822parse_field_t
959 rfc822parse_parse_field (rfc822parse_t msg, const char *name, int which)
960 {
961   HDR_LINE hdr;
962
963   if (!which)
964     return NULL;
965
966   hdr = find_header (msg, name, which, NULL);
967   if (!hdr)
968     return NULL;
969   return parse_field (hdr);
970 }
971
972 void
973 rfc822parse_release_field (rfc822parse_field_t ctx)
974 {
975   if (ctx)
976     release_token_list (ctx);
977 }
978
979
980
981 /****************
982  * Check whether T points to a parameter.
983  * A parameter starts with a semicolon and it is assumed that t
984  * points to exactly this one.
985  */
986 static int
987 is_parameter (TOKEN t)
988 {
989   t = t->next;
990   if (!t || t->type != tATOM)
991     return 0;
992   t = t->next;
993   if (!t || !(t->type == tSPECIAL && t->data[0] == '='))
994     return 0;
995   t = t->next;
996   if (!t)
997     return 1; /* We assume that an non existing value is an empty one. */
998   return t->type == tQUOTED || t->type == tATOM;
999 }
1000
1001 /*
1002    Some header (Content-type) have a special syntax where attribute=value
1003    pairs are used after a leading semicolon.  The parse_field code
1004    knows about these fields and changes the parsing to the one defined
1005    in RFC2045.
1006    Returns a pointer to the value which is valid as long as the
1007    parse context is valid; NULL is returned in case that attr is not
1008    defined in the header, a missing value is reppresented by an empty string.
1009  
1010    With LOWER_VALUE set to true, a matching field valuebe be
1011    lowercased.
1012  
1013    Note, that ATTR should be lowercase.
1014  */
1015 const char *
1016 rfc822parse_query_parameter (rfc822parse_field_t ctx, const char *attr,
1017                              int lower_value)
1018 {
1019   TOKEN t, a;
1020
1021   for (t = ctx; t; t = t->next)
1022     {
1023       /* skip to the next semicolon */
1024       for (; t && !(t->type == tSPECIAL && t->data[0] == ';'); t = t->next)
1025         ;
1026       if (!t)
1027         return NULL;
1028       if (is_parameter (t))
1029         { /* Look closer. */
1030           a = t->next; /* We know that this is an atom */
1031           if ( !a->flags.lowered )
1032             {
1033               lowercase_string (a->data);
1034               a->flags.lowered = 1;
1035             }
1036           if (!strcmp (a->data, attr))
1037             { /* found */
1038               t = a->next->next;
1039               /* Either T is now an atom, a quoted string or NULL in
1040                * which case we return an empty string. */
1041
1042               if ( lower_value && t && !t->flags.lowered )
1043                 {
1044                   lowercase_string (t->data);
1045                   t->flags.lowered = 1;
1046                 }
1047               return t ? t->data : "";
1048             }
1049         }
1050     }
1051   return NULL;
1052 }
1053
1054 /****************
1055  * This function may be used for the Content-Type header to figure out
1056  * the media type and subtype.  Note, that the returned strings are
1057  * guaranteed to be lowercase as required by MIME.
1058  *
1059  * Returns: a pointer to the media type and if subtype is not NULL,
1060  *          a pointer to the subtype.
1061  */
1062 const char *
1063 rfc822parse_query_media_type (rfc822parse_field_t ctx, const char **subtype)
1064 {
1065   TOKEN t = ctx;
1066   const char *type;
1067
1068   if (t->type != tATOM)
1069     return NULL;
1070   if (!t->flags.lowered)
1071     {
1072       lowercase_string (t->data);
1073       t->flags.lowered = 1;
1074     }
1075   type = t->data;
1076   t = t->next;
1077   if (!t || t->type != tSPECIAL || t->data[0] != '/')
1078     return NULL;
1079   t = t->next;
1080   if (!t || t->type != tATOM)
1081     return NULL;
1082
1083   if (subtype)
1084     {
1085       if (!t->flags.lowered)
1086         {
1087           lowercase_string (t->data);
1088           t->flags.lowered = 1;
1089         }
1090       *subtype = t->data;
1091     }
1092   return type;
1093 }
1094
1095
1096
1097
1098
1099 #ifdef TESTING
1100
1101 /* Internal debug function to print the structure of the message. */
1102 static void
1103 dump_structure (rfc822parse_t msg, part_t part, int indent)
1104 {
1105   if (!part)
1106     {
1107       printf ("*** Structure of this message:\n");
1108       part = msg->parts;
1109     }
1110
1111   for (; part; part = part->right)
1112     {
1113       rfc822parse_field_t ctx;
1114       part_t save_part; /* ugly hack - we should have a function to
1115                            get part inforation. */
1116       const char *s;
1117       
1118       save_part = msg->current_part;
1119       msg->current_part = part;
1120       ctx = rfc822parse_parse_field (msg, "Content-Type", -1);
1121       msg->current_part = save_part;
1122       if (ctx)
1123         {
1124           const char *s1, *s2;
1125           s1 = rfc822parse_query_media_type (ctx, &s2);
1126           if (s1)
1127             printf ("***   %*s %s/%s", indent*2, "", s1, s2);
1128           else
1129             printf ("***   %*s [not found]", indent*2, "");
1130
1131           s = rfc822parse_query_parameter (ctx, "boundary", 0);
1132           if (s)
1133             printf (" (boundary=\"%s\")", s);
1134           rfc822parse_release_field (ctx);
1135         }
1136       else
1137         printf ("***   %*s text/plain [assumed]", indent*2, "");
1138       putchar('\n');
1139
1140       if (part->down)
1141         dump_structure (msg, part->down, indent + 1);
1142     }
1143   
1144 }
1145
1146
1147
1148 static void
1149 show_param (rfc822parse_field_t ctx, const char *name)
1150 {
1151   const char *s;
1152
1153   if (!ctx)
1154     return;
1155   s = rfc822parse_query_parameter (ctx, name, 0);
1156   if (s)
1157     printf ("***   %s: `%s'\n", name, s);
1158 }
1159
1160
1161
1162 static void
1163 show_event (rfc822parse_event_t event)
1164 {
1165   const char *s;
1166
1167   switch (event)
1168     {
1169     case RFC822PARSE_OPEN: s= "Open"; break;
1170     case RFC822PARSE_CLOSE: s= "Close"; break;
1171     case RFC822PARSE_CANCEL: s= "Cancel"; break;
1172     case RFC822PARSE_T2BODY: s= "T2Body"; break;
1173     case RFC822PARSE_FINISH: s= "Finish"; break;
1174     case RFC822PARSE_RCVD_SEEN: s= "Rcvd_Seen"; break;
1175     case RFC822PARSE_BOUNDARY: s= "Boundary"; break;
1176     case RFC822PARSE_LAST_BOUNDARY: s= "Last_Boundary"; break;
1177     default: s= "***invalid event***"; break;
1178     }
1179   printf ("*** got RFC822 event %s\n", s);
1180 }
1181
1182 static int
1183 msg_cb (void *dummy_arg, rfc822parse_event_t event, rfc822parse_t msg)
1184 {
1185   show_event (event);
1186   if (event == RFC822PARSE_T2BODY)
1187     {
1188       rfc822parse_field_t ctx;
1189       void *ectx;
1190       const char *line;
1191
1192       for (ectx=NULL; (line = rfc822parse_enum_header_lines (msg, &ectx)); )
1193         {
1194           printf ("*** HDR: %s\n", line);
1195         }
1196       rfc822parse_enum_header_lines (NULL, &ectx); /* Close enumerator. */
1197
1198       ctx = rfc822parse_parse_field (msg, "Content-Type", -1);
1199       if (ctx)
1200         {
1201           const char *s1, *s2;
1202           s1 = rfc822parse_query_media_type (ctx, &s2);
1203           if (s1)
1204             printf ("***   media: `%s/%s'\n", s1, s2);
1205           else
1206             printf ("***   media: [not found]\n");
1207           show_param (ctx, "boundary");
1208           show_param (ctx, "protocol");
1209           rfc822parse_release_field (ctx);
1210         }
1211       else
1212         printf ("***   media: text/plain [assumed]\n");
1213       
1214     }
1215
1216
1217   return 0;
1218 }
1219
1220
1221
1222 int
1223 main (int argc, char **argv)
1224 {
1225   char line[5000];
1226   size_t length;
1227   rfc822parse_t msg;
1228
1229   msg = rfc822parse_open (msg_cb, NULL);
1230   if (!msg)
1231     abort ();
1232
1233   while (fgets (line, sizeof (line), stdin))
1234     {
1235       length = strlen (line);
1236       if (length && line[length - 1] == '\n')
1237         line[--length] = 0;
1238       if (length && line[length - 1] == '\r')
1239         line[--length] = 0;
1240       if (rfc822parse_insert (msg, line, length))
1241         abort ();
1242     }
1243
1244   dump_structure (msg, NULL, 0);
1245
1246   rfc822parse_close (msg);
1247   return 0;
1248 }
1249 #endif
1250
1251 /*
1252 Local Variables:
1253 compile-command: "gcc -Wall -g -DTESTING -o rfc822parse rfc822parse.c"
1254 End:
1255 */