send.c

   1 #include <ctype.h>
   2 #include <fcntl.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <time.h>
   6 #include <unistd.h>
   7 #include <sys/stat.h>
   8 #include <sys/types.h>
   9 #include "config.h"
  10 #include "mbox.h"
  11 #include "util.h"
  12 #include "send.h"
  13 #include "str.h"
  14
  15 #define USERAGENT               "git://repo.or.cz/mailx.git"
  16 #define MAXLINE                 (1 << 7)
  17
  18 static int isaddr(int c)
  19 {
  20         return isalpha(c) || isdigit(c) || strchr("_.-@", c);
  21 }
  22
  23 static char *cut_addr(char *dst, char *src)
  24 {
  25         while (*src && !isaddr(*src))
  26                 src++;
  27         while (isaddr(*src))
  28                 *dst++ = *src++;
  29         *dst++ = '\0';
  30         return src;
  31 }
  32
  33 static int parse_addr(char **recips, int size, char *s, int len)
  34 {
  35         int n = 0;
  36         char *d = s + len;
  37         while (n < size && s < d && *(s = cut_addr(recips[n], s)))
  38                 if (strchr(recips[n], '@'))
  39                         n++;
  40         return n;
  41 }
  42
  43 static int mail_recips(char **recips, int size, char *s)
  44 {
  45         struct mail mail = {{0}};
  46         char *to, *cc;
  47         int n = 0;
  48         mail_read(&mail, s);
  49         if ((to = mail_hdr(&mail, "To:")))
  50                 n += parse_addr(recips + n, size - n, to + 3, hdr_len(to) - 3);
  51         if ((cc = mail_hdr(&mail, "CC:")))
  52                 n += parse_addr(recips + n, size - n, cc + 3, hdr_len(cc) - 3);
  53         return n;
  54 }
  55
  56 static char *put_from_(char *s)
  57 {
  58         time_t t;
  59         char *logname = getenv("LOGNAME");
  60         time(&t);
  61         s = put_str(s, "From ");
  62         s = put_str(s, logname ? logname : "me");
  63         s += strftime(s, MAXLINE, " %a %b %d %H:%M:%S %Y\n", localtime(&t));
  64         return s;
  65 }
  66
  67 static char *put_date(char *s)
  68 {
  69         time_t t;
  70         time(&t);
  71         s = put_str(s, "Date: ");
  72         s += strftime(s, MAXLINE, "%a, %d %b %Y %H:%M:%S %z\n", localtime(&t));
  73         return s;
  74 }
  75
  76 static char *put_id(char *s)
  77 {
  78         time_t t;
  79         time(&t);
  80         s = put_str(s, "Message-ID: <");
  81         s += strftime(s, MAXLINE, "%Y%d%m%H%M%S", localtime(&t));
  82         s = put_str(s, "@" HOSTNAME ">\n");
  83         return s;
  84 }
  85
  86 static char *put_from(char *s)
  87 {
  88         return FROM ? put_str(s, "From: " FROM "\n") : s;
  89 }
  90
  91 static char *put_to(char *s, char **to, int n)
  92 {
  93         int i;
  94         s = put_str(s, "To: ");
  95         s = put_str(s, to[0]);
  96         for (i = 1; i < n; i++) {
  97                 s = put_str(s, ",\n\t");
  98                 s = put_str(s, to[i]);
  99         }
 100         if (*(s - 1) != '\n')
 101                 s = put_str(s, "\n");
 102         return s;
 103 }
 104
 105 static char *put_subj(char *s, char *subj)
 106 {
 107         s = put_str(s, "Subject: ");
 108         s = put_str(s, subj);
 109         s = put_str(s, "\n");
 110         return s;
 111 }
 112
 113 static char *put_agent(char *s)
 114 {
 115         return put_str(s, "User-Agent: " USERAGENT "\n");
 116 }
 117
 118 void draft_init(struct draft *draft, char **to, int nto, char *subj)
 119 {
 120         char *s = draft->mail;
 121         s = put_from_(s);
 122         s = put_date(s);
 123         s = put_from(s);
 124         if (to && nto)
 125                 s = put_to(s, to, nto);
 126         if (subj)
 127                 s = put_subj(s, subj);
 128         s = put_id(s);
 129         s = put_agent(s);
 130         draft->len = s - draft->mail;
 131 }
 132
 133 static char *hdr_val(char *hdr)
 134 {
 135         hdr = strchr(hdr, ':') + 1;
 136         while (isspace(*hdr))
 137                 hdr++;
 138         return hdr;
 139 }
 140
 141 static char *put_replysubj(char *s, char *subj)
 142 {
 143         subj = hdr_val(subj);
 144         s = put_str(s, "Subject: ");
 145         if (tolower(subj[0]) != 'r' || tolower(subj[1]) != 'e')
 146                 s = put_str(s, "Re: ");
 147         s = put_mem(s, subj, hdr_len(subj));
 148         return s;
 149 }
 150
 151 static char *put_replyto(char *s, char *id, char *ref)
 152 {
 153         s = put_str(s, "In-Reply-To: ");
 154         id = hdr_val(id);
 155         s = put_mem(s, id, hdr_len(id));
 156         s = put_str(s, "References: ");
 157         if (ref) {
 158                 ref = hdr_val(ref);
 159                 s = put_mem(s, ref, hdr_len(ref));
 160                 s = put_str(s, "\t");
 161         }
 162         s = put_mem(s, id, hdr_len(id));
 163         return s;
 164 }
 165
 166 static char *put_reply(char *s, char *from, char *to, char *cc)
 167 {
 168         if (from) {
 169                 from = hdr_val(from);
 170                 s = put_str(s, "To: ");
 171                 s = put_mem(s, from, hdr_len(from));
 172         }
 173         if (to || cc) {
 174                 s = put_str(s, "Cc: ");
 175                 if (to) {
 176                         to = hdr_val(to);
 177                         s = put_mem(s, to, hdr_len(to));
 178                 }
 179                 if (cc) {
 180                         cc = hdr_val(cc);
 181                         if (to) {
 182                                 while (strchr("\r\n ", *(s - 1)))
 183                                         s--;
 184                                 s = put_str(s, ",\n\t");
 185                         }
 186                         s = put_mem(s, cc, hdr_len(cc));
 187                 }
 188         }
 189         return s;
 190 }
 191
 192 static char *quote_body(char *s, int size, struct mail *mail)
 193 {
 194         char *from = mail_hdr(mail, "From:");
 195         char *r = mail->body;
 196         char *rd = r + mail->body_len;
 197         char *sd = s + size - 16;
 198         s = put_str(s, "\n");
 199         if (from) {
 200                 from = hdr_val(from);
 201                 s = put_mem(s, from, hdr_len(from) - 1);
 202                 s = put_str(s, " wrote:\n");
 203         }
 204         while (s < sd && r < rd) {
 205                 char *nl = memchr(r, '\n', rd - r);
 206                 s = put_str(s, "> ");
 207                 nl = nl ? nl + 1 : rd;
 208                 s = put_mem(s, r, MIN(sd - s, nl - r));
 209                 r = nl;
 210         }
 211         *s = '\0';
 212         return s;
 213 }
 214
 215 void draft_reply(struct draft *draft, struct mail *mail)
 216 {
 217         char *s = draft->mail;
 218         char *id_hdr = mail_hdr(mail, "Message-ID:");
 219         char *ref_hdr = mail_hdr(mail, "References:");
 220         char *from_hdr = mail_hdr(mail, "From:");
 221         char *subj_hdr = mail_hdr(mail, "Subject:");
 222         char *to_hdr = mail_hdr(mail, "To:");
 223         char *cc_hdr = mail_hdr(mail, "CC:");
 224
 225         s = put_from_(s);
 226         s = put_date(s);
 227         s = put_from(s);
 228         s = put_reply(s, from_hdr, to_hdr, cc_hdr);
 229         if (subj_hdr)
 230                 s = put_replysubj(s, subj_hdr);
 231         s = put_id(s);
 232         if (id_hdr)
 233                 s = put_replyto(s, id_hdr, ref_hdr);
 234         s = put_agent(s);
 235         s = quote_body(s, sizeof(draft->mail) - (s - draft->mail), mail);
 236         draft->len = s - draft->mail;
 237 }
 238
 239 void draft_edit(struct draft *draft, char *editor)
 240 {
 241         char *edit_argv[] = {NULL, DRAFT, NULL};
 242         int fd;
 243         if ((fd = open(DRAFT, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR)) == -1)
 244                 return;
 245         xwrite(fd, draft->mail, draft->len);
 246         close(fd);
 247         edit_argv[0] = editor;
 248         exec_file(editor, edit_argv);
 249         if ((fd = open(DRAFT, O_RDONLY)) == -1)
 250                 return;
 251         draft->len = xread(fd, draft->mail, sizeof(draft->mail) - 1);
 252         close(fd);
 253         draft->mail[draft->len] = '\0';
 254         unlink(DRAFT);
 255 }
 256
 257 void draft_send(struct draft *draft)
 258 {
 259         char recips[MAXRECP][MAXLINE];
 260         char *send_argv[MAXRECP + 3] = {SENDMAIL, "-i"};
 261         char *beg = strchr(draft->mail, '\n') + 1;
 262         char *end = draft->mail + draft->len;
 263         int i;
 264         for (i = 0; i < MAXRECP; i++)
 265                 send_argv[i + 2] = recips[i];
 266         send_argv[2 + mail_recips(send_argv + 2, MAXRECP, draft->mail)] = NULL;
 267         exec_pipe(SENDMAIL, send_argv, beg, end - beg);
 268 }
 269
 270 static void write_nofrom(int fd, char *s, int len)
 271 {
 272         char *beg = s;
 273         char *d = s + len;
 274         char *r;
 275         while (s < d) {
 276                 if (!strncmp("\nFrom ", s, 6)) {
 277                         write(fd, beg, s - beg + 1);
 278                         write(fd, ">", 1);
 279                         beg = s + 1;
 280                 }
 281                 r = strchr(s + 1, '\n');
 282                 s = r ? r : d;
 283         }
 284         write(fd, beg, s - beg);
 285 }
 286
 287 void draft_save(struct draft *draft, char *path)
 288 {
 289         int fd = open(path, O_WRONLY | O_APPEND | O_CREAT, S_IRUSR | S_IWUSR);
 290         if (fd < -1)
 291                 return;
 292         write_nofrom(fd, draft->mail, draft->len);
 293         write(fd, "\n", 1);
 294         close(fd);
 295 }