mbox.c

   1 #include <ctype.h>
   2 #include <errno.h>
   3 #include <fcntl.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <time.h>
   7 #include <unistd.h>
   8 #include <utime.h>
   9 #include <sys/types.h>
  10 #include <sys/stat.h>
  11 #include "mbox.h"
  12 #include "str.h"
  13 #include "util.h"
  14
  15 #define INCSIZE         (1 << 22)
  16
  17 static void set_atime(char *filename)
  18 {
  19         struct stat st;
  20         struct utimbuf times;
  21         stat(filename, &st);
  22         times.actime = time(NULL);
  23         times.modtime = st.st_mtime;
  24         utime(filename, &times);
  25 }
  26
  27 static char *mail_start(char *r, char *e)
  28 {
  29         char *s = r;
  30         char *t;
  31         while (s < e) {
  32                 if (!strncmp("From ", s, 5) && (s == r || *(s - 1) == '\n'))
  33                         break;
  34                 t = memchr(s + 1, 'F', e - s - 1);
  35                 s = t ? t : e;
  36         }
  37         return s;
  38 }
  39
  40 static char *mail_hdrs(struct mail *mail, char *s)
  41 {
  42         static char stat[] = "Status:";
  43         char *r;
  44         while (s && *s && *s != '\r' && *s != '\n') {
  45                 if (isalpha(*s) && mail->nhdrs < MAXHDRS)
  46                         mail->hdrs[mail->nhdrs++] = s;
  47                 if (!strncmp(stat, s, sizeof(stat) - 1))
  48                         mail->stat_hdr = s;
  49                 if ((r = strchr(s + 1, '\n')))
  50                         s = r + 1;
  51                 else
  52                         s = strchr(s + 1, '\0');
  53         }
  54         mail->hdrs[mail->nhdrs] = s;
  55         return s;
  56 }
  57
  58 char *mail_hdr(struct mail *mail, char *hdr)
  59 {
  60         int i;
  61         for (i = 1; i < mail->nhdrs; i++)
  62                 if (startswith(mail->hdrs[i], hdr))
  63                         return mail->hdrs[i];
  64         return NULL;
  65 }
  66
  67 int hdr_len(char *hdr)
  68 {
  69         char *s = hdr;
  70         while (*s) {
  71                 char *r = strchr(s, '\n');
  72                 if (!r)
  73                         return strchr(s, '\0') - hdr;
  74                 s = r + 1;
  75                 if (!isspace(*s))
  76                         return s - hdr;
  77         }
  78         return 0;
  79 }
  80
  81 static void mail_stat(struct mail *mail)
  82 {
  83         if (mail->stat_hdr) {
  84                 char *s = mail->stat_hdr + strlen("Status:");
  85                 while (*s && *s != '\r' && *s != '\n') {
  86                         switch (*s++) {
  87                         case 'O':
  88                                 mail->stat |= STAT_OLD;
  89                                 break;
  90                         case 'N':
  91                                 mail->stat |= STAT_NEW;
  92                                 break;
  93                         case 'R':
  94                                 mail->stat |= STAT_READ;
  95                                 break;
  96                         }
  97                 }
  98                 mail->orig_stat = mail->stat;
  99         } else {
 100                 mail->stat = STAT_NEW;
 101         }
 102 }
 103
 104 char *mail_read(struct mail *mail, char *s, char *e)
 105 {
 106         char *end;
 107         mail->head = s;
 108         mail->body = mail_hdrs(mail, s);
 109         end = mail_start(mail->body, e);
 110         mail->len = end - mail->head;
 111         mail->body_len = end - mail->body;
 112         mail_stat(mail);
 113         return end;
 114 }
 115
 116 static int read_mails(struct mbox *mbox, char *s, char *e)
 117 {
 118         while (s && *s && mbox->n < MAXMAILS)
 119                 s = mail_read(&mbox->mails[mbox->n++], s, e);
 120         return mbox->n == MAXMAILS;
 121 }
 122
 123 int mbox_inc(struct mbox *mbox)
 124 {
 125         int fd = open(mbox->path, O_RDONLY);
 126         int old_n = mbox->n;
 127         int old_len = mbox->len;
 128         int len = file_size(fd);
 129         if (fd == -1)
 130                 return 0;
 131         if (!mbox->mbox) {
 132                 mbox->size = len + INCSIZE + 1;
 133                 mbox->mbox = malloc(mbox->size);
 134         }
 135         if (!mbox->mbox || len > mbox->size - 1)
 136                 return -1;
 137         if (mbox->len)
 138                 lseek(fd, mbox->len, SEEK_SET);
 139         mbox->len += xread(fd, mbox->mbox + mbox->len, len);
 140         mbox->mbox[mbox->len] = '\0';
 141         close(fd);
 142         set_atime(mbox->path);
 143         if (old_len < mbox->len)
 144                 if (read_mails(mbox, mbox->mbox + old_len,
 145                                 mbox->mbox + mbox->len))
 146                         return -1;
 147         return mbox->n - old_n;
 148 }
 149
 150 struct mbox *mbox_alloc(char *filename)
 151 {
 152         struct mbox *mbox;
 153         mbox = malloc(sizeof(*mbox));
 154         if (!mbox)
 155                 return NULL;
 156         memset(mbox, 0, sizeof(*mbox));
 157         strcpy(mbox->path, filename);
 158         if (mbox_inc(mbox) < 0) {
 159                 mbox_free(mbox);
 160                 return NULL;
 161         }
 162         return mbox;
 163 }
 164
 165 void mbox_free(struct mbox *mbox)
 166 {
 167         free(mbox->mbox);
 168         free(mbox);
 169 }
 170
 171 int mail_head(struct mail *mail, char *buf, int buf_len,
 172                 char **hdrs, int n)
 173 {
 174         int i, j;
 175         char *s = buf;
 176         char *dead = s + buf_len;
 177         for (i = 0; i < n; i++) {
 178                 char *pat = hdrs[i];
 179                 for (j = 0; j < mail->nhdrs; j++) {
 180                         char *hdr = mail->hdrs[j];
 181                         if (startswith(hdr, pat)) {
 182                                 int hdr_len = mail->hdrs[j + 1] - hdr;
 183                                 if (s + hdr_len > dead)
 184                                         hdr_len = dead - s;
 185                                 memcpy(s, hdr, hdr_len);
 186                                 s += hdr_len;
 187                                 break;
 188                         }
 189                 }
 190         }
 191         *s = '\0';
 192         return s - buf;
 193 }
 194
 195 static int mail_changed(struct mail *mail)
 196 {
 197         return mail->orig_stat != mail->stat;
 198 }
 199
 200 void mail_write(struct mail *mail, int fd)
 201 {
 202         char stat[32] = "Status: ";
 203         char *s = strchr(stat, '\0');
 204         if (mail->stat & STAT_READ)
 205                 *s++ = 'R';
 206         if (mail->stat & (STAT_OLD | STAT_READ))
 207                 *s++ = 'O';
 208         if (mail->stat & STAT_NEW)
 209                 *s++ = 'N';
 210         *s++ = '\n';
 211         if (!mail_changed(mail)) {
 212                 xwrite(fd, mail->head, mail->len);
 213         } else {
 214                 char *hdr_beg = mail->body;
 215                 char *hdr_end = mail->body;
 216                 if (mail->stat_hdr) {
 217                         char *nl = strchr(mail->stat_hdr, '\n');
 218                         hdr_beg = mail->stat_hdr;
 219                         hdr_end = nl ? nl + 1 : mail->body;
 220                 }
 221                 xwrite(fd, mail->head, hdr_beg - mail->head);
 222                 xwrite(fd, stat, s - stat);
 223                 if (hdr_end)
 224                         xwrite(fd, hdr_end, mail->head + mail->len - hdr_end);
 225         }
 226 }
 227
 228 static int mbox_changed(struct mbox *mbox)
 229 {
 230         int i;
 231         for (i = 0; i < mbox->n; i++)
 232                 if (mail_changed(&mbox->mails[i]))
 233                         return 1;
 234         return 0;
 235 }
 236
 237 int mbox_write(struct mbox *mbox)
 238 {
 239         int fd;
 240         int i = 0;
 241         if (mbox_inc(mbox) < 0)
 242                 return -1;
 243         if (!mbox_changed(mbox))
 244                 return 0;
 245         fd = open(mbox->path, O_WRONLY | O_TRUNC | O_CREAT, S_IRUSR | S_IWUSR);
 246         while (i < mbox->n) {
 247                 char *beg = mbox->mails[i].head;
 248                 char *end = mbox->mbox + mbox->len;
 249                 while (i < mbox->n && !mail_changed(&mbox->mails[i]))
 250                         i++;
 251                 if (i < mbox->n)
 252                         end = mbox->mails[i].head;
 253                 xwrite(fd, beg, end - beg);
 254                 if (i < mbox->n && !(mbox->mails[i].stat & STAT_DEL))
 255                         mail_write(&mbox->mails[i], fd);
 256                 i++;
 257         }
 258         close(fd);
 259         return 0;
 260 }