add mdeliver

1 files changed, 200 insertions, 0 deletions

diff --git a/mdeliver.c b/mdeliver.c
new file mode 100644
index 0000000..938a423
--- /dev/null
+++ b/mdeliver.c
@@ -0,0 +1,200 @@
+#include <sys/time.h>
+#include <sys/types.h>
+
+#include <dirent.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <limits.h>
+#include <search.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "blaze822.h"
+
+/*
+design rationale:
+- only MBOX-RD because it's the only reasonable lossless encoding, and decodes
+  MBOX-O fine
+- date from Date: since From lines are usually crap
+- proper maildir delivery because it's not that hard
+- messages end up in cur/ by default because you are usually importing archives
+- no creation of maildirs, should be a separate tool
+*/
+
+static int Mflag;
+static int nflag;
+static int vflag;
+static char *Xflag;
+
+char *targetdir;
+long delivery;
+
+char host[64];
+void
+gethost() {
+	gethostname(host, sizeof host);
+	// termination not posix guaranteed
+	host[sizeof host - 1] = 0;
+	// replace / and : with _
+	char *s;
+	for (s = host; *s; s++)
+		if (*s == '/' || *s == ':')
+			*s = '-';
+}
+
+int
+deliver(FILE *infile)
+{
+	int outfd;
+	FILE *outfile;
+	char dst[PATH_MAX];
+	char tmp[PATH_MAX];
+	char id[128];
+	struct timeval tv;
+
+	char *line = 0;
+	size_t linelen = 0;
+
+	if (Mflag) {
+		// skip to first "From " line
+		while (1) {
+			errno = 0;
+			ssize_t rd = getline(&line, &linelen, infile);
+			if (rd == -1) {
+				if (errno == 0)
+					// invalid mbox file
+					errno = EINVAL;
+				return -1;
+			}
+
+			if (strncmp("From ", line, 5) == 0)
+				break;
+		}
+	}
+
+	while (!feof(infile)) {
+		delivery++;
+tryagain:
+		gettimeofday(&tv, 0);
+
+		snprintf(id, sizeof id, "%ld.M%06ldP%ldQ%ld.%s",
+			 tv.tv_sec, tv.tv_usec, (long)getpid(), delivery, host);
+		
+		snprintf(tmp, sizeof tmp, "%s/tmp/%s", targetdir, id);
+	
+		outfd = open(tmp, O_CREAT | O_WRONLY | O_EXCL, 0666);
+		if (outfd < 0) {
+			if (errno == EEXIST)
+				goto tryagain;
+			return -1;
+		}
+
+		outfile = fdopen(outfd, "w");
+
+		while (1) {
+			errno = 0;
+			ssize_t rd = getline(&line, &linelen, infile);
+			if (rd == -1) {
+				if (errno != 0)
+					return -1;
+				break;
+			}
+
+			if (Mflag && strncmp("From ", line, 5) == 0)
+				break;
+
+			if (Mflag) {
+				// MBOXRD: strip first > from >>..>>From
+				char *s = line;
+				while (*s == '>')
+					s++;
+				if (strncmp("From ", s, 5) == 0) {
+					line++;
+					rd--;
+				}
+			}
+
+			if (fwrite(line, 1, rd, outfile) != (size_t)rd)
+				return -1;
+		}
+		if (fflush(outfile) == EOF)
+			return -1;
+		if (fsync(outfd) < 0)
+			return -1;
+		if (fclose(outfile) == EOF)
+			return -1;
+
+		char statusflags[5] = { 0 };
+		char *f = statusflags;
+
+		if (Mflag) {
+			struct message *msg = blaze822_file(tmp);
+			time_t date = -1;
+			char *v;
+			
+			if (msg && (v = blaze822_hdr(msg, "date"))) {
+				date = blaze822_date(v);
+				if (date != -1) {
+					const struct timeval times[2] = {
+						{ tv.tv_sec, tv.tv_usec },
+						{ date, 0 }
+					};
+					utimes(tmp, times);
+				}
+			}
+			if (msg && ((v = blaze822_hdr(msg, "status")) ||
+				    (v = blaze822_hdr(msg, "x-status")))) {
+				if (strchr(v, 'F')) *f++ = 'F';
+				if (strchr(v, 'A')) *f++ = 'R';
+				if (strchr(v, 'R') || strchr(v, 'O'))
+					*f++ = 'S';
+				if (strchr(v, 'D')) *f++ = 'T';
+			}
+		}
+		*f = 0;
+
+		snprintf(dst, sizeof dst, "%s/%s/%s:2,%s",
+			 targetdir, nflag ? "new" : "cur", id,
+			 Xflag ? Xflag : statusflags);
+		if (rename(tmp, dst) != 0)
+			return -1;
+
+		if (vflag)
+			printf("%s\n", dst);
+	}
+	return 0;
+}
+
+int
+main(int argc, char *argv[])
+{
+	int c;
+	while ((c = getopt(argc, argv, "MnvX:")) != -1)
+		switch(c) {
+		case 'M': Mflag = 1; break;
+		case 'n': nflag = 1; break;
+		case 'v': vflag = 1; break;
+		case 'X': Xflag = optarg; break;
+		default:
+			// XXX usage
+			exit(1);
+		}
+
+	if (argc != optind+1) {
+		fprintf(stderr, "usage: mdeliver DIR\n");
+		return 1;
+	}
+
+	targetdir = argv[optind];
+
+	gethost();
+
+	if (deliver(stdin) < 0) {
+		perror("deliver");
+		return 2;
+	}
+
+	return 0;
+}