src/utils/test_view_mail.c

   1 #define  _BSD_SOURCE
   2
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include <unistd.h>
   6 #include <errno.h>
   7 #include <fcntl.h>
   8 #include <time.h>
   9 #include <ctype.h>
  10
  11 #include <sys/types.h>
  12 #include <sys/stat.h>
  13
  14 #include "string_utils.h"
  15 #include "file_utils.h"
  16 #include "rfc5322.h"
  17 #include "rfc2047.h"
  18 #include "rfc822.h"
  19 #include "mime.h"
  20 #include "mail.h"
  21
  22 static void plain_body(char *line)
  23 {
  24         printf("%s", line);
  25 }
  26
  27
  28 static void qp_body(char *line)
  29 {
  30         char buf[1024];
  31
  32         rfc2047_decode_quoted_printable(buf, line, sizeof(buf) - 1);
  33         printf("%s", buf);
  34 }
  35
  36 static void parse_mail_body(FILE *fh, struct mail_header *hdr)
  37 {
  38         char buf[1024];
  39         void (*decode_body)(char *line);
  40
  41         switch (hdr->mime.transfer_encoding) {
  42         case MIME_ENCODING_QUOTED_PRINTABLE:
  43                 decode_body = qp_body;
  44                 break;
  45         default:
  46                 decode_body = plain_body;
  47                 break;
  48         }
  49
  50         (void) fseek(fh, hdr->body_offset, SEEK_SET);
  51
  52         while (fgets(buf, sizeof(buf), fh)) {
  53                 decode_body(buf);
  54         }
  55 }
  56
  57 static void parse_mail_header(FILE *fh, struct mail_header *hdr)
  58 {
  59         char *crlf = NULL;
  60         char buf[1024];
  61
  62         /* Scan all lines */
  63         while (fgets(buf, sizeof(buf), fh)) {
  64                 if (!strncmp(buf, "From:", 5)) {
  65                         rfc5322_extract_address(hdr->from, buf, sizeof(hdr->from));
  66                         crlf = strrchr(buf, '\n');
  67                         if (crlf)
  68                                 *crlf = '\0';
  69                 } else if (!strncmp(buf, "Subject:", 8)) {
  70                         rfc5322_extract_subject(hdr->subject, buf, sizeof(hdr->subject));
  71                 } else if (!strncmp(buf, "Date:", 5)) {
  72                         hdr->date = rfc822_parsedt(buf);
  73                 } else if (!strncmp(buf, "Content-Type:", 13)) {
  74                         mime_content_type(buf, &hdr->mime);
  75                 } else if (!strncmp(buf, "Content-Disposition:", 20)) {
  76                         hdr->mime.disposition = mime_content_disposition(buf);
  77                 } else if (!strncmp(buf, "Content-Transfer-Encoding:", 26)) {
  78                         hdr->mime.transfer_encoding = mime_content_transfer_encoding(buf);
  79                 } else {
  80                         if (rfc822_check_header_end(buf)) {
  81                                 break;
  82                         }
  83                         /*printf("[SKIP] ------- %s", buf); */
  84                 }
  85         }
  86
  87         hdr->body_offset = ftell(fh);
  88 }
  89
  90 static void init_mail_header(struct mail_header *hdr)
  91 {
  92         memset(hdr->from, 0, sizeof(hdr->from));
  93         memset(hdr->subject, 0, sizeof(hdr->subject));
  94
  95         hdr->body_offset = 0L;
  96
  97         /* MIME */
  98         hdr->mime.type = MIME_TYPE_UNKNOWN;
  99         hdr->mime.disposition = MIME_DISPOSITION_UNKNOWN;
 100         hdr->mime.transfer_encoding = MIME_ENCODING_UNKNOWN;
 101 }
 102
 103 void usage(const char *appname)
 104 {
 105         fprintf(stderr, "Usage:\n%s [-h] -f <file>\n", appname);
 106         fprintf(stderr, "Parameters:\n");
 107         fprintf(stderr, " -h Parse header only\n"
 108                         " -f <dir> Parse file\n");
 109 }
 110
 111 int main(int argc, char **argv)
 112 {
 113         FILE *fh;
 114         struct mail_header mail_hdr;
 115
 116         int opt;
 117
 118         char *file = NULL;
 119         int hdr_only = 0;
 120
 121
 122         while ((opt = getopt(argc, argv, "f:h")) != -1) {
 123                 switch (opt) {
 124                 case 'f':
 125                         file = optarg;
 126                         break;
 127                 case 'h':
 128                         hdr_only = 1;
 129                         break;
 130                 default:
 131                         fprintf(stderr, "Unknown parameter.\n");
 132                         usage(argv[0]);
 133                         return -1;
 134                 }
 135         }
 136
 137         if (file == NULL) {
 138                 usage(argv[0]);
 139                 return -1;
 140         }
 141
 142         if (file_type(file))
 143                 return -1;
 144
 145         fh = fopen(file, "r");
 146         if (!fh) {
 147                 return -1;
 148         }
 149
 150         init_mail_header(&mail_hdr);
 151
 152         parse_mail_header(fh, &mail_hdr);
 153
 154         fprintf(stderr, "-------------------------------------------\n");
 155         fprintf(stderr, "Test file: %s\n", file);
 156         fprintf(stderr, "-------------------------------------------\n");
 157         fprintf(stderr, "Date   : %s\n", rfc822_mkdt(mail_hdr.date));
 158         fprintf(stderr, "From   : %s\n", mail_hdr.from);
 159         fprintf(stderr, "Subject: %s\n", mail_hdr.subject);
 160
 161         debug_mime_header(&mail_hdr.mime);
 162
 163         fprintf(stderr, "-------------------------------------------\n\n");
 164
 165         if (!hdr_only) {
 166                 /* Message Body */
 167                 parse_mail_body(fh, &mail_hdr);
 168         }
 169
 170         fclose(fh);
 171
 172         return 0;
 173 }
 174