add mmime

1 files changed, 261 insertions, 0 deletions

diff --git a/mmime.c b/mmime.c
new file mode 100644
index 0000000..93377fc
--- /dev/null
+++ b/mmime.c
@@ -0,0 +1,261 @@
+#include <sys/mman.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
+#include <dirent.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <limits.h>
+#include <pwd.h>
+#include <search.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <unistd.h>
+
+#include "blaze822.h"
+
+int gen_b64(uint8_t *s, off_t size)
+{
+	static char *b64 =
+	    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+
+	off_t i;
+	int l;
+	uint32_t v;
+	for (i = 0, l = 0; i+2 < size; i += 3) {
+		v = (s[i] << 16) | (s[i+1] << 8) | s[i+2];
+		putc_unlocked(b64[(v & 0xfc0000) >> 18], stdout);
+		putc_unlocked(b64[(v & 0x03f000) >> 12], stdout);
+		putc_unlocked(b64[(v & 0x000fc0) >>  6], stdout);
+		putc_unlocked(b64[(v & 0x3f)], stdout);
+		l += 4;
+		if (l > 72) {
+			l = 0;
+			printf("\n");
+		}
+	}
+	if (size - i == 2) { // 2 bytes left, XXX=
+		v = (s[size - 2] << 16) | (s[size - 1] << 8);
+		putc_unlocked(b64[(v & 0xfc0000) >> 18], stdout);
+		putc_unlocked(b64[(v & 0x03f000) >> 12], stdout);
+		putc_unlocked(b64[(v & 0x000fc0) >>  6], stdout);
+		putc_unlocked('=', stdout);
+	} else if (size - i == 1) { // 1 byte left, XX==
+		v = s[size - 1] << 16;
+		putc_unlocked(b64[(v & 0xfc0000) >> 18], stdout);
+		putc_unlocked(b64[(v & 0x03f000) >> 12], stdout);
+		putc_unlocked('=', stdout);
+		putc_unlocked('=', stdout);
+	}
+	printf("\n");
+	return 0;
+}
+
+int gen_qp(uint8_t *s, off_t size)
+{
+	off_t i;
+	int linelen = 0;
+	char prev = 0;
+
+	for (i = 0; i < size; i++) {
+		if ((s[i] > 126) ||
+		    (s[i] < 32 && s[i] != '\n' && s[i] != '\t') ||
+		    (s[i] == '=')) {
+			printf("=%02X", s[i]);
+			linelen += 3;
+			prev = s[i];
+		} else if (s[i] == '\n') {
+			if (i > 1 && (prev == ' ' || prev == '\t'))
+				puts("=");
+			putc_unlocked('\n', stdout);
+			linelen = 0;
+			prev = 0;
+		} else {
+			putc_unlocked(s[i], stdout);
+			linelen++;
+			prev = s[i];
+		}
+		
+		if (linelen >= 75) {
+			linelen = 0;
+			prev = '\n';
+			puts("=");
+		}
+	}
+	if (linelen > 0)
+		puts("=");
+	return 0;
+}
+
+static const char *
+basenam(const char *s)
+{
+        char *r = strrchr(s, '/');
+        return r ? r + 1 : s;
+}
+
+int gen_file(char *file, char *ct)
+{
+	int fd = open(file, O_RDONLY);
+	if (!fd)
+		return 0;
+
+	struct stat st;
+
+	fstat(fd, &st);
+	uint8_t *content = mmap(0, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
+	close(fd);
+
+	if (content == MAP_FAILED)
+		return -1;
+	
+	off_t bithigh = 0;
+	off_t bitlow = 0;
+	off_t linelen = 0;
+	off_t maxlinelen = 0;
+	off_t i;
+	for (i = 0; i < st.st_size; i++) {
+		if (content[i] == '\n') {
+			if (maxlinelen < linelen)
+				maxlinelen = linelen;
+			linelen = 0;
+		} else {
+			linelen++;
+		}
+		if (content[i] != '\t' && content[i] != '\n' && content[i] < 32)
+			bitlow++;
+		if (content[i] > 127)
+			bithigh++;
+	}
+
+	printf("Content-Disposition: attachment; filename=\"%s\"\n",
+	       basenam(file));
+	if (bitlow == 0 && bithigh == 0 &&
+	    maxlinelen <= 72 && content[st.st_size-1] == '\n') {
+		if (!ct)
+			ct = "text/plain";
+		printf("Content-Type: %s\n", ct);
+		printf("Content-Transfer-Encoding: 7bit\n\n");
+		fwrite(content, 1, st.st_size, stdout);
+		return 0;
+	} else if (bitlow == 0 && bithigh == 0) {
+		if (!ct)
+			ct = "text/plain";
+		printf("Content-Type: %s\n", ct);
+		printf("Content-Transfer-Encoding: quoted-printable\n\n");
+		return gen_qp(content, st.st_size);
+	} else if (bitlow > st.st_size/10 || bithigh > st.st_size/4) {
+		if (!ct)
+			ct = "application/binary";
+		printf("Content-Type: %s\n", ct);
+		printf("Content-Transfer-Encoding: base64\n\n");
+		return gen_b64(content, st.st_size);
+	} else {
+		if (!ct)
+			ct = "text/plain";
+		printf("Content-Type: %s\n", ct);
+		printf("Content-Transfer-Encoding: quoted-printable\n\n");
+		return gen_qp(content, st.st_size);
+	}
+}
+
+int
+gen_mixed(int argc, char *argv[])
+{
+	char sep[100];
+	snprintf(sep, sizeof sep, "----_=_%08lx%08lx%08lx_=_",
+	    lrand48(), lrand48(), lrand48());
+	int i;
+
+	printf("Content-Type: multipart/mixed; boundary=\"%s\"\n", sep);
+	printf("\n");
+	printf("This is a multipart message in MIME format.\n\n");
+	for (i = 0; i < argc; i++) {
+		printf("--%s\n", sep);
+		gen_file(argv[i], 0);
+	}
+	printf("--%s--\n", sep);
+
+	return 0;
+}
+
+int
+gen_build()
+{
+	char sep[100];
+	snprintf(sep, sizeof sep, "----_=_%08lx%08lx%08lx_=_",
+	    lrand48(), lrand48(), lrand48());
+
+	char *line = 0;
+	size_t linelen = 0;
+	int inheader = 1;
+	int intext = 0;
+
+	while (1) {
+		int read = getline(&line, &linelen, stdin);
+		if (read == -1) {
+			if (feof(stdin))
+				break;
+			else
+				exit(1);
+		}
+		if (inheader) {
+			if (line[0] == '\n') {
+				inheader = 0;
+				printf("MIME-Version: 1.0\n");
+				printf("Content-Type: multipart/mixed; boundary=\"%s\"\n", sep);
+				printf("\n");
+				printf("This is a multipart message in MIME format.\n\n");
+			} else {
+				printf("%s", line);
+			}
+			continue;
+		}
+
+		if (line[0] == '#') {
+			char *f = strchr(line, ' ');
+			*f = 0;
+			if (strchr(line, '/')) {
+				printf("--%s\n", sep);
+				if (line[read-1] == '\n')
+					line[read-1] = 0;
+				gen_file(f+1, (char *)line+1);
+				intext = 0;
+				continue;
+			}
+		}
+
+		if (!intext) {
+			printf("--%s\n", sep);
+			printf("Content-Type: text/plain\n");
+			printf("Content-Disposition: inline\n");
+			printf("Content-Transfer-Encoding: quoted-printable\n\n");
+			
+			intext = 1;
+		}
+
+		gen_qp((uint8_t *)line, strlen(line));
+	}
+	printf("--%s--\n", sep);
+
+	free(line);
+	return 0;
+}
+
+int
+main(int argc, char *argv[])
+{
+	srand48(time(0) ^ getpid());
+
+	if (argc == 1)
+		return gen_build();
+
+	printf("MIME-Version: 1.0\n");
+	if (argc >= 2)
+		return gen_mixed(argc-1, argv+1);
+	else
+		return gen_file(argv[1], 0);
+}