diff options
| -rw-r--r-- | Makefile | 3 | ||||
| -rw-r--r-- | mmime.c | 261 |
2 files changed, 263 insertions, 1 deletions
diff --git a/Makefile b/Makefile index 57de9ef..58237ef 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,6 @@ CFLAGS=-g -O1 -Wall -Wno-switch -Wextra -fstack-protector-strong -D_FORTIFY_SOURCE=2 -ALL = mscan mthread mhdr mshow mlist mseq msort +ALL = mscan mthread mhdr mshow mlist mseq msort mmime all: $(ALL) @@ -11,6 +11,7 @@ mshow: mshow.o blaze822.o seq.o rfc2045.o rfc2047.c mlist: mlist.o mseq: mseq.o seq.o msort: msort.o blaze822.o seq.o +mmime: mmime.o clean: FRC -rm -f $(ALL) *.o 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); +} |