Code

read-cache.c cleanup
[git.git] / builtin-mailinfo.c
1 /*
2  * Another stupid program, this one parsing the headers of an
3  * email to figure out authorship and subject
4  */
5 #define _GNU_SOURCE
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <ctype.h>
10 #ifndef NO_ICONV
11 #include <iconv.h>
12 #endif
13 #include "git-compat-util.h"
14 #include "cache.h"
15 #include "builtin.h"
17 static FILE *cmitmsg, *patchfile, *fin, *fout;
19 static int keep_subject = 0;
20 static const char *metainfo_charset = NULL;
21 static char line[1000];
22 static char date[1000];
23 static char name[1000];
24 static char email[1000];
25 static char subject[1000];
27 static enum  {
28         TE_DONTCARE, TE_QP, TE_BASE64,
29 } transfer_encoding;
30 static char charset[256];
32 static char multipart_boundary[1000];
33 static int multipart_boundary_len;
34 static int patch_lines = 0;
36 static char *sanity_check(char *name, char *email)
37 {
38         int len = strlen(name);
39         if (len < 3 || len > 60)
40                 return email;
41         if (strchr(name, '@') || strchr(name, '<') || strchr(name, '>'))
42                 return email;
43         return name;
44 }
46 static int bogus_from(char *line)
47 {
48         /* John Doe <johndoe> */
49         char *bra, *ket, *dst, *cp;
51         /* This is fallback, so do not bother if we already have an
52          * e-mail address.
53          */
54         if (*email)
55                 return 0;
57         bra = strchr(line, '<');
58         if (!bra)
59                 return 0;
60         ket = strchr(bra, '>');
61         if (!ket)
62                 return 0;
64         for (dst = email, cp = bra+1; cp < ket; )
65                 *dst++ = *cp++;
66         *dst = 0;
67         for (cp = line; isspace(*cp); cp++)
68                 ;
69         for (bra--; isspace(*bra); bra--)
70                 *bra = 0;
71         cp = sanity_check(cp, email);
72         strcpy(name, cp);
73         return 1;
74 }
76 static int handle_from(char *in_line)
77 {
78         char line[1000];
79         char *at;
80         char *dst;
82         strcpy(line, in_line);
83         at = strchr(line, '@');
84         if (!at)
85                 return bogus_from(line);
87         /*
88          * If we already have one email, don't take any confusing lines
89          */
90         if (*email && strchr(at+1, '@'))
91                 return 0;
93         /* Pick up the string around '@', possibly delimited with <>
94          * pair; that is the email part.  White them out while copying.
95          */
96         while (at > line) {
97                 char c = at[-1];
98                 if (isspace(c))
99                         break;
100                 if (c == '<') {
101                         at[-1] = ' ';
102                         break;
103                 }
104                 at--;
105         }
106         dst = email;
107         for (;;) {
108                 unsigned char c = *at;
109                 if (!c || c == '>' || isspace(c)) {
110                         if (c == '>')
111                                 *at = ' ';
112                         break;
113                 }
114                 *at++ = ' ';
115                 *dst++ = c;
116         }
117         *dst++ = 0;
119         /* The remainder is name.  It could be "John Doe <john.doe@xz>"
120          * or "john.doe@xz (John Doe)", but we have whited out the
121          * email part, so trim from both ends, possibly removing
122          * the () pair at the end.
123          */
124         at = line + strlen(line);
125         while (at > line) {
126                 unsigned char c = *--at;
127                 if (!isspace(c)) {
128                         at[(c == ')') ? 0 : 1] = 0;
129                         break;
130                 }
131         }
133         at = line;
134         for (;;) {
135                 unsigned char c = *at;
136                 if (!c || !isspace(c)) {
137                         if (c == '(')
138                                 at++;
139                         break;
140                 }
141                 at++;
142         }
143         at = sanity_check(at, email);
144         strcpy(name, at);
145         return 1;
148 static int handle_date(char *line)
150         strcpy(date, line);
151         return 0;
154 static int handle_subject(char *line)
156         strcpy(subject, line);
157         return 0;
160 /* NOTE NOTE NOTE.  We do not claim we do full MIME.  We just attempt
161  * to have enough heuristics to grok MIME encoded patches often found
162  * on our mailing lists.  For example, we do not even treat header lines
163  * case insensitively.
164  */
166 static int slurp_attr(const char *line, const char *name, char *attr)
168         const char *ends, *ap = strcasestr(line, name);
169         size_t sz;
171         if (!ap) {
172                 *attr = 0;
173                 return 0;
174         }
175         ap += strlen(name);
176         if (*ap == '"') {
177                 ap++;
178                 ends = "\"";
179         }
180         else
181                 ends = "; \t";
182         sz = strcspn(ap, ends);
183         memcpy(attr, ap, sz);
184         attr[sz] = 0;
185         return 1;
188 static int handle_subcontent_type(char *line)
190         /* We do not want to mess with boundary.  Note that we do not
191          * handle nested multipart.
192          */
193         if (strcasestr(line, "boundary=")) {
194                 fprintf(stderr, "Not handling nested multipart message.\n");
195                 exit(1);
196         }
197         slurp_attr(line, "charset=", charset);
198         if (*charset) {
199                 int i, c;
200                 for (i = 0; (c = charset[i]) != 0; i++)
201                         charset[i] = tolower(c);
202         }
203         return 0;
206 static int handle_content_type(char *line)
208         *multipart_boundary = 0;
209         if (slurp_attr(line, "boundary=", multipart_boundary + 2)) {
210                 memcpy(multipart_boundary, "--", 2);
211                 multipart_boundary_len = strlen(multipart_boundary);
212         }
213         slurp_attr(line, "charset=", charset);
214         return 0;
217 static int handle_content_transfer_encoding(char *line)
219         if (strcasestr(line, "base64"))
220                 transfer_encoding = TE_BASE64;
221         else if (strcasestr(line, "quoted-printable"))
222                 transfer_encoding = TE_QP;
223         else
224                 transfer_encoding = TE_DONTCARE;
225         return 0;
228 static int is_multipart_boundary(const char *line)
230         return (!memcmp(line, multipart_boundary, multipart_boundary_len));
233 static int eatspace(char *line)
235         int len = strlen(line);
236         while (len > 0 && isspace(line[len-1]))
237                 line[--len] = 0;
238         return len;
241 #define SEEN_FROM 01
242 #define SEEN_DATE 02
243 #define SEEN_SUBJECT 04
244 #define SEEN_BOGUS_UNIX_FROM 010
245 #define SEEN_PREFIX  020
247 /* First lines of body can have From:, Date:, and Subject: or empty */
248 static void handle_inbody_header(int *seen, char *line)
250         if (*seen & SEEN_PREFIX)
251                 return;
252         if (isspace(*line)) {
253                 char *cp;
254                 for (cp = line + 1; *cp; cp++) {
255                         if (!isspace(*cp))
256                                 break;
257                 }
258                 if (!*cp)
259                         return;
260         }
261         if (!memcmp(">From", line, 5) && isspace(line[5])) {
262                 if (!(*seen & SEEN_BOGUS_UNIX_FROM)) {
263                         *seen |= SEEN_BOGUS_UNIX_FROM;
264                         return;
265                 }
266         }
267         if (!memcmp("From:", line, 5) && isspace(line[5])) {
268                 if (!(*seen & SEEN_FROM) && handle_from(line+6)) {
269                         *seen |= SEEN_FROM;
270                         return;
271                 }
272         }
273         if (!memcmp("Date:", line, 5) && isspace(line[5])) {
274                 if (!(*seen & SEEN_DATE)) {
275                         handle_date(line+6);
276                         *seen |= SEEN_DATE;
277                         return;
278                 }
279         }
280         if (!memcmp("Subject:", line, 8) && isspace(line[8])) {
281                 if (!(*seen & SEEN_SUBJECT)) {
282                         handle_subject(line+9);
283                         *seen |= SEEN_SUBJECT;
284                         return;
285                 }
286         }
287         if (!memcmp("[PATCH]", line, 7) && isspace(line[7])) {
288                 if (!(*seen & SEEN_SUBJECT)) {
289                         handle_subject(line);
290                         *seen |= SEEN_SUBJECT;
291                         return;
292                 }
293         }
294         *seen |= SEEN_PREFIX;
297 static char *cleanup_subject(char *subject)
299         if (keep_subject)
300                 return subject;
301         for (;;) {
302                 char *p;
303                 int len, remove;
304                 switch (*subject) {
305                 case 'r': case 'R':
306                         if (!memcmp("e:", subject+1, 2)) {
307                                 subject +=3;
308                                 continue;
309                         }
310                         break;
311                 case ' ': case '\t': case ':':
312                         subject++;
313                         continue;
315                 case '[':
316                         p = strchr(subject, ']');
317                         if (!p) {
318                                 subject++;
319                                 continue;
320                         }
321                         len = strlen(p);
322                         remove = p - subject;
323                         if (remove <= len *2) {
324                                 subject = p+1;
325                                 continue;
326                         }
327                         break;
328                 }
329                 eatspace(subject);
330                 return subject;
331         }
334 static void cleanup_space(char *buf)
336         unsigned char c;
337         while ((c = *buf) != 0) {
338                 buf++;
339                 if (isspace(c)) {
340                         buf[-1] = ' ';
341                         c = *buf;
342                         while (isspace(c)) {
343                                 int len = strlen(buf);
344                                 memmove(buf, buf+1, len);
345                                 c = *buf;
346                         }
347                 }
348         }
351 static void decode_header(char *it);
352 typedef int (*header_fn_t)(char *);
353 struct header_def {
354         const char *name;
355         header_fn_t func;
356         int namelen;
357 };
359 static void check_header(char *line, struct header_def *header)
361         int i;
363         if (header[0].namelen <= 0) {
364                 for (i = 0; header[i].name; i++)
365                         header[i].namelen = strlen(header[i].name);
366         }
367         for (i = 0; header[i].name; i++) {
368                 int len = header[i].namelen;
369                 if (!strncasecmp(line, header[i].name, len) &&
370                     line[len] == ':' && isspace(line[len + 1])) {
371                         /* Unwrap inline B and Q encoding, and optionally
372                          * normalize the meta information to utf8.
373                          */
374                         decode_header(line + len + 2);
375                         header[i].func(line + len + 2);
376                         break;
377                 }
378         }
381 static void check_subheader_line(char *line)
383         static struct header_def header[] = {
384                 { "Content-Type", handle_subcontent_type },
385                 { "Content-Transfer-Encoding",
386                   handle_content_transfer_encoding },
387                 { NULL },
388         };
389         check_header(line, header);
391 static void check_header_line(char *line)
393         static struct header_def header[] = {
394                 { "From", handle_from },
395                 { "Date", handle_date },
396                 { "Subject", handle_subject },
397                 { "Content-Type", handle_content_type },
398                 { "Content-Transfer-Encoding",
399                   handle_content_transfer_encoding },
400                 { NULL },
401         };
402         check_header(line, header);
405 static int is_rfc2822_header(char *line)
407         /*
408          * The section that defines the loosest possible
409          * field name is "3.6.8 Optional fields".
410          *
411          * optional-field = field-name ":" unstructured CRLF
412          * field-name = 1*ftext
413          * ftext = %d33-57 / %59-126
414          */
415         int ch;
416         char *cp = line;
417         while ((ch = *cp++)) {
418                 if (ch == ':')
419                         return cp != line;
420                 if ((33 <= ch && ch <= 57) ||
421                     (59 <= ch && ch <= 126))
422                         continue;
423                 break;
424         }
425         return 0;
428 static int read_one_header_line(char *line, int sz, FILE *in)
430         int ofs = 0;
431         while (ofs < sz) {
432                 int peek, len;
433                 if (fgets(line + ofs, sz - ofs, in) == NULL)
434                         break;
435                 len = eatspace(line + ofs);
436                 if ((len == 0) || !is_rfc2822_header(line)) {
437                         /* Re-add the newline */
438                         line[ofs + len] = '\n';
439                         line[ofs + len + 1] = '\0';
440                         break;
441                 }
442                 ofs += len;
443                 /* Yuck, 2822 header "folding" */
444                 peek = fgetc(in); ungetc(peek, in);
445                 if (peek != ' ' && peek != '\t')
446                         break;
447         }
448         /* Count mbox From headers as headers */
449         if (!ofs && (!memcmp(line, "From ", 5) || !memcmp(line, ">From ", 6)))
450                 ofs = 1;
451         return ofs;
454 static unsigned hexval(int c)
456         if (c >= '0' && c <= '9')
457                 return c - '0';
458         if (c >= 'a' && c <= 'f')
459                 return c - 'a' + 10;
460         if (c >= 'A' && c <= 'F')
461                 return c - 'A' + 10;
462         return ~0;
465 static int decode_q_segment(char *in, char *ot, char *ep, int rfc2047)
467         int c;
468         while ((c = *in++) != 0 && (in <= ep)) {
469                 if (c == '=') {
470                         int d = *in++;
471                         if (d == '\n' || !d)
472                                 break; /* drop trailing newline */
473                         *ot++ = ((hexval(d) << 4) | hexval(*in++));
474                         continue;
475                 }
476                 if (rfc2047 && c == '_') /* rfc2047 4.2 (2) */
477                         c = 0x20;
478                 *ot++ = c;
479         }
480         *ot = 0;
481         return 0;
484 static int decode_b_segment(char *in, char *ot, char *ep)
486         /* Decode in..ep, possibly in-place to ot */
487         int c, pos = 0, acc = 0;
489         while ((c = *in++) != 0 && (in <= ep)) {
490                 if (c == '+')
491                         c = 62;
492                 else if (c == '/')
493                         c = 63;
494                 else if ('A' <= c && c <= 'Z')
495                         c -= 'A';
496                 else if ('a' <= c && c <= 'z')
497                         c -= 'a' - 26;
498                 else if ('0' <= c && c <= '9')
499                         c -= '0' - 52;
500                 else if (c == '=') {
501                         /* padding is almost like (c == 0), except we do
502                          * not output NUL resulting only from it;
503                          * for now we just trust the data.
504                          */
505                         c = 0;
506                 }
507                 else
508                         continue; /* garbage */
509                 switch (pos++) {
510                 case 0:
511                         acc = (c << 2);
512                         break;
513                 case 1:
514                         *ot++ = (acc | (c >> 4));
515                         acc = (c & 15) << 4;
516                         break;
517                 case 2:
518                         *ot++ = (acc | (c >> 2));
519                         acc = (c & 3) << 6;
520                         break;
521                 case 3:
522                         *ot++ = (acc | c);
523                         acc = pos = 0;
524                         break;
525                 }
526         }
527         *ot = 0;
528         return 0;
531 static void convert_to_utf8(char *line, char *charset)
533 #ifndef NO_ICONV
534         char *in, *out;
535         size_t insize, outsize, nrc;
536         char outbuf[4096]; /* cheat */
537         static char latin_one[] = "latin1";
538         char *input_charset = *charset ? charset : latin_one;
539         iconv_t conv = iconv_open(metainfo_charset, input_charset);
541         if (conv == (iconv_t) -1) {
542                 static int warned_latin1_once = 0;
543                 if (input_charset != latin_one) {
544                         fprintf(stderr, "cannot convert from %s to %s\n",
545                                 input_charset, metainfo_charset);
546                         *charset = 0;
547                 }
548                 else if (!warned_latin1_once) {
549                         warned_latin1_once = 1;
550                         fprintf(stderr, "tried to convert from %s to %s, "
551                                 "but your iconv does not work with it.\n",
552                                 input_charset, metainfo_charset);
553                 }
554                 return;
555         }
556         in = line;
557         insize = strlen(in);
558         out = outbuf;
559         outsize = sizeof(outbuf);
560         nrc = iconv(conv, &in, &insize, &out, &outsize);
561         iconv_close(conv);
562         if (nrc == (size_t) -1)
563                 return;
564         *out = 0;
565         strcpy(line, outbuf);
566 #endif
569 static int decode_header_bq(char *it)
571         char *in, *out, *ep, *cp, *sp;
572         char outbuf[1000];
573         int rfc2047 = 0;
575         in = it;
576         out = outbuf;
577         while ((ep = strstr(in, "=?")) != NULL) {
578                 int sz, encoding;
579                 char charset_q[256], piecebuf[256];
580                 rfc2047 = 1;
582                 if (in != ep) {
583                         sz = ep - in;
584                         memcpy(out, in, sz);
585                         out += sz;
586                         in += sz;
587                 }
588                 /* E.g.
589                  * ep : "=?iso-2022-jp?B?GyR...?= foo"
590                  * ep : "=?ISO-8859-1?Q?Foo=FCbar?= baz"
591                  */
592                 ep += 2;
593                 cp = strchr(ep, '?');
594                 if (!cp)
595                         return rfc2047; /* no munging */
596                 for (sp = ep; sp < cp; sp++)
597                         charset_q[sp - ep] = tolower(*sp);
598                 charset_q[cp - ep] = 0;
599                 encoding = cp[1];
600                 if (!encoding || cp[2] != '?')
601                         return rfc2047; /* no munging */
602                 ep = strstr(cp + 3, "?=");
603                 if (!ep)
604                         return rfc2047; /* no munging */
605                 switch (tolower(encoding)) {
606                 default:
607                         return rfc2047; /* no munging */
608                 case 'b':
609                         sz = decode_b_segment(cp + 3, piecebuf, ep);
610                         break;
611                 case 'q':
612                         sz = decode_q_segment(cp + 3, piecebuf, ep, 1);
613                         break;
614                 }
615                 if (sz < 0)
616                         return rfc2047;
617                 if (metainfo_charset)
618                         convert_to_utf8(piecebuf, charset_q);
619                 strcpy(out, piecebuf);
620                 out += strlen(out);
621                 in = ep + 2;
622         }
623         strcpy(out, in);
624         strcpy(it, outbuf);
625         return rfc2047;
628 static void decode_header(char *it)
631         if (decode_header_bq(it))
632                 return;
633         /* otherwise "it" is a straight copy of the input.
634          * This can be binary guck but there is no charset specified.
635          */
636         if (metainfo_charset)
637                 convert_to_utf8(it, "");
640 static void decode_transfer_encoding(char *line)
642         char *ep;
644         switch (transfer_encoding) {
645         case TE_QP:
646                 ep = line + strlen(line);
647                 decode_q_segment(line, line, ep, 0);
648                 break;
649         case TE_BASE64:
650                 ep = line + strlen(line);
651                 decode_b_segment(line, line, ep);
652                 break;
653         case TE_DONTCARE:
654                 break;
655         }
658 static void handle_info(void)
660         char *sub;
662         sub = cleanup_subject(subject);
663         cleanup_space(name);
664         cleanup_space(date);
665         cleanup_space(email);
666         cleanup_space(sub);
668         fprintf(fout, "Author: %s\nEmail: %s\nSubject: %s\nDate: %s\n\n",
669                name, email, sub, date);
672 /* We are inside message body and have read line[] already.
673  * Spit out the commit log.
674  */
675 static int handle_commit_msg(int *seen)
677         if (!cmitmsg)
678                 return 0;
679         do {
680                 if (!memcmp("diff -", line, 6) ||
681                     !memcmp("---", line, 3) ||
682                     !memcmp("Index: ", line, 7))
683                         break;
684                 if ((multipart_boundary[0] && is_multipart_boundary(line))) {
685                         /* We come here when the first part had only
686                          * the commit message without any patch.  We
687                          * pretend we have not seen this line yet, and
688                          * go back to the loop.
689                          */
690                         return 1;
691                 }
693                 /* Unwrap transfer encoding and optionally
694                  * normalize the log message to UTF-8.
695                  */
696                 decode_transfer_encoding(line);
697                 if (metainfo_charset)
698                         convert_to_utf8(line, charset);
700                 handle_inbody_header(seen, line);
701                 if (!(*seen & SEEN_PREFIX))
702                         continue;
704                 fputs(line, cmitmsg);
705         } while (fgets(line, sizeof(line), fin) != NULL);
706         fclose(cmitmsg);
707         cmitmsg = NULL;
708         return 0;
711 /* We have done the commit message and have the first
712  * line of the patch in line[].
713  */
714 static void handle_patch(void)
716         do {
717                 if (multipart_boundary[0] && is_multipart_boundary(line))
718                         break;
719                 /* Only unwrap transfer encoding but otherwise do not
720                  * do anything.  We do *NOT* want UTF-8 conversion
721                  * here; we are dealing with the user payload.
722                  */
723                 decode_transfer_encoding(line);
724                 fputs(line, patchfile);
725                 patch_lines++;
726         } while (fgets(line, sizeof(line), fin) != NULL);
729 /* multipart boundary and transfer encoding are set up for us, and we
730  * are at the end of the sub header.  do equivalent of handle_body up
731  * to the next boundary without closing patchfile --- we will expect
732  * that the first part to contain commit message and a patch, and
733  * handle other parts as pure patches.
734  */
735 static int handle_multipart_one_part(int *seen)
737         int n = 0;
739         while (fgets(line, sizeof(line), fin) != NULL) {
740         again:
741                 n++;
742                 if (is_multipart_boundary(line))
743                         break;
744                 if (handle_commit_msg(seen))
745                         goto again;
746                 handle_patch();
747                 break;
748         }
749         if (n == 0)
750                 return -1;
751         return 0;
754 static void handle_multipart_body(void)
756         int seen = 0;
757         int part_num = 0;
759         /* Skip up to the first boundary */
760         while (fgets(line, sizeof(line), fin) != NULL)
761                 if (is_multipart_boundary(line)) {
762                         part_num = 1;
763                         break;
764                 }
765         if (!part_num)
766                 return;
767         /* We are on boundary line.  Start slurping the subhead. */
768         while (1) {
769                 int hdr = read_one_header_line(line, sizeof(line), fin);
770                 if (!hdr) {
771                         if (handle_multipart_one_part(&seen) < 0)
772                                 return;
773                         /* Reset per part headers */
774                         transfer_encoding = TE_DONTCARE;
775                         charset[0] = 0;
776                 }
777                 else
778                         check_subheader_line(line);
779         }
780         fclose(patchfile);
781         if (!patch_lines) {
782                 fprintf(stderr, "No patch found\n");
783                 exit(1);
784         }
787 /* Non multipart message */
788 static void handle_body(void)
790         int seen = 0;
792         handle_commit_msg(&seen);
793         handle_patch();
794         fclose(patchfile);
795         if (!patch_lines) {
796                 fprintf(stderr, "No patch found\n");
797                 exit(1);
798         }
801 int mailinfo(FILE *in, FILE *out, int ks, const char *encoding,
802              const char *msg, const char *patch)
804         keep_subject = ks;
805         metainfo_charset = encoding;
806         fin = in;
807         fout = out;
809         cmitmsg = fopen(msg, "w");
810         if (!cmitmsg) {
811                 perror(msg);
812                 return -1;
813         }
814         patchfile = fopen(patch, "w");
815         if (!patchfile) {
816                 perror(patch);
817                 fclose(cmitmsg);
818                 return -1;
819         }
820         while (1) {
821                 int hdr = read_one_header_line(line, sizeof(line), fin);
822                 if (!hdr) {
823                         if (multipart_boundary[0])
824                                 handle_multipart_body();
825                         else
826                                 handle_body();
827                         handle_info();
828                         break;
829                 }
830                 check_header_line(line);
831         }
833         return 0;
836 static const char mailinfo_usage[] =
837         "git-mailinfo [-k] [-u | --encoding=<encoding>] msg patch <mail >info";
839 int cmd_mailinfo(int argc, const char **argv, const char *prefix)
841         /* NEEDSWORK: might want to do the optional .git/ directory
842          * discovery
843          */
844         git_config(git_default_config);
846         while (1 < argc && argv[1][0] == '-') {
847                 if (!strcmp(argv[1], "-k"))
848                         keep_subject = 1;
849                 else if (!strcmp(argv[1], "-u"))
850                         metainfo_charset = git_commit_encoding;
851                 else if (!strncmp(argv[1], "--encoding=", 11))
852                         metainfo_charset = argv[1] + 11;
853                 else
854                         usage(mailinfo_usage);
855                 argc--; argv++;
856         }
858         if (argc != 3)
859                 usage(mailinfo_usage);
861         return !!mailinfo(stdin, stdout, keep_subject, metainfo_charset, argv[1], argv[2]);