mirror of
https://github.com/leahneukirchen/mblaze
synced 2024-11-03 15:40:32 +00:00
669 lines
13 KiB
C
669 lines
13 KiB
C
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <fnmatch.h>
|
|
#include <iconv.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#include "blaze822.h"
|
|
|
|
static int rflag;
|
|
static int Rflag;
|
|
static int qflag;
|
|
static int Hflag;
|
|
static int Lflag;
|
|
static int tflag;
|
|
static int nflag;
|
|
static char defaulthflags[] = "from:subject:to:cc:date:";
|
|
static char *hflag = defaulthflags;
|
|
static char *xflag;
|
|
static char *Oflag;
|
|
|
|
struct message *filters;
|
|
|
|
static int mimecount;
|
|
|
|
void
|
|
printhdr(char *hdr)
|
|
{
|
|
int uc = 1;
|
|
|
|
while (*hdr && *hdr != ':') {
|
|
putc(uc ? toupper(*hdr) : *hdr, stdout);
|
|
uc = (*hdr == '-');
|
|
hdr++;
|
|
}
|
|
|
|
if (*hdr) {
|
|
printf("%s\n", hdr);
|
|
}
|
|
}
|
|
|
|
void
|
|
print_u8recode(char *body, size_t bodylen, char *srcenc)
|
|
{
|
|
iconv_t ic;
|
|
|
|
ic = iconv_open("UTF-8", srcenc);
|
|
if (ic == (iconv_t)-1) {
|
|
printf("unsupported encoding: %s\n", srcenc);
|
|
return;
|
|
}
|
|
|
|
char final_char = 0;
|
|
|
|
char buf[4096];
|
|
while (bodylen > 0) {
|
|
char *bufptr = buf;
|
|
size_t buflen = sizeof buf;
|
|
size_t r = iconv(ic, &body, &bodylen, &bufptr, &buflen);
|
|
|
|
if (bufptr != buf) {
|
|
fwrite(buf, 1, bufptr-buf, stdout);
|
|
final_char = bufptr[-1];
|
|
}
|
|
|
|
if (r != (size_t)-1) { // done, flush iconv
|
|
bufptr = buf;
|
|
buflen = sizeof buf;
|
|
r = iconv(ic, 0, 0, &bufptr, &buflen);
|
|
if (bufptr != buf) {
|
|
fwrite(buf, 1, bufptr-buf, stdout);
|
|
final_char = bufptr[-1];
|
|
}
|
|
if (r != (size_t)-1)
|
|
break;
|
|
}
|
|
|
|
if (r == (size_t)-1 && errno != E2BIG) {
|
|
perror("iconv");
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (final_char != '\n')
|
|
printf("\n");
|
|
|
|
iconv_close(ic);
|
|
}
|
|
|
|
char *
|
|
mimetype(char *ct)
|
|
{
|
|
char *s;
|
|
|
|
if (!ct)
|
|
return 0;
|
|
for (s = ct; *s && *s != ';' && *s != ' ' && *s != '\t'; s++)
|
|
;
|
|
|
|
return strndup(ct, s-ct);
|
|
}
|
|
|
|
char *
|
|
tlmimetype(char *ct)
|
|
{
|
|
char *s;
|
|
|
|
if (!ct)
|
|
return 0;
|
|
for (s = ct; *s && *s != ';' && *s != ' ' && *s != '\t' && *s != '/'; s++)
|
|
;
|
|
|
|
return strndup(ct, s-ct);
|
|
}
|
|
|
|
typedef enum {
|
|
MIME_CONTINUE,
|
|
MIME_STOP,
|
|
MIME_PRUNE,
|
|
} mime_action;
|
|
|
|
typedef mime_action (*mime_callback)(int, struct message *, char *, size_t);
|
|
|
|
char *
|
|
mime_filename(struct message *msg)
|
|
{
|
|
char *filename = 0, *fn, *fne, *v;
|
|
|
|
if ((v = blaze822_hdr(msg, "content-disposition"))) {
|
|
if (blaze822_mime_parameter(v, "filename", &fn, &fne))
|
|
filename = strndup(fn, fne-fn);
|
|
} else if ((v = blaze822_hdr(msg, "content-type"))) {
|
|
if (blaze822_mime_parameter(v, "name", &fn, &fne))
|
|
filename = strndup(fn, fne-fn);
|
|
}
|
|
|
|
return filename;
|
|
}
|
|
|
|
mime_action
|
|
render_mime(int depth, struct message *msg, char *body, size_t bodylen)
|
|
{
|
|
char *ct = blaze822_hdr(msg, "content-type");
|
|
if (!ct)
|
|
ct = "text/x-unknown";
|
|
char *mt = mimetype(ct);
|
|
char *tlmt = tlmimetype(ct);
|
|
char *filename = mime_filename(msg);
|
|
|
|
mimecount++;
|
|
|
|
int i;
|
|
for (i = 0; i < depth+1; i++)
|
|
printf("--- ");
|
|
printf("%d: %s size=%zd", mimecount, mt, bodylen);
|
|
if (filename) {
|
|
printf(" name=\"%s\"", filename);
|
|
free(filename);
|
|
}
|
|
|
|
char *cmd;
|
|
mime_action r = MIME_CONTINUE;
|
|
|
|
if (filters &&
|
|
((cmd = blaze822_chdr(filters, mt)) ||
|
|
(cmd = blaze822_chdr(filters, tlmt)))) {
|
|
char *charset = 0, *cs, *cse;
|
|
if (blaze822_mime_parameter(ct, "charset", &cs, &cse)) {
|
|
charset = strndup(cs, cse-cs);
|
|
printf(" charset=\"%s\"", charset);
|
|
setenv("PIPE_CHARSET", charset, 1);
|
|
free(charset);
|
|
}
|
|
printf(" filter=\"%s\" ---\n", cmd);
|
|
FILE *p;
|
|
fflush(stdout);
|
|
p = popen(cmd, "w");
|
|
if (!p) {
|
|
perror("popen");
|
|
goto nofilter;
|
|
}
|
|
fwrite(body, 1, bodylen, p);
|
|
if (pclose(p) != 0) {
|
|
perror("pclose");
|
|
goto nofilter;
|
|
}
|
|
r = MIME_PRUNE;
|
|
} else {
|
|
nofilter:
|
|
printf(" ---\n");
|
|
|
|
if (strncmp(ct, "text/", 5) == 0) {
|
|
char *charset = 0, *cs, *cse;
|
|
if (blaze822_mime_parameter(ct, "charset", &cs, &cse))
|
|
charset = strndup(cs, cse-cs);
|
|
if (!charset ||
|
|
strcasecmp(charset, "utf-8") == 0 ||
|
|
strcasecmp(charset, "utf8") == 0 ||
|
|
strcasecmp(charset, "us-ascii") == 0)
|
|
fwrite(body, 1, bodylen, stdout);
|
|
else
|
|
print_u8recode(body, bodylen, charset);
|
|
free(charset);
|
|
} else if (strncmp(ct, "message/rfc822", 14) == 0) {
|
|
struct message *imsg = blaze822_mem(body, bodylen);
|
|
char *h = 0;
|
|
if (imsg) {
|
|
while ((h = blaze822_next_header(imsg, h)))
|
|
printf("%s\n", h);
|
|
printf("\n");
|
|
}
|
|
} else if (strncmp(ct, "multipart/", 10) == 0) {
|
|
;
|
|
} else {
|
|
printf("no filter or default handler\n");
|
|
}
|
|
}
|
|
|
|
free(mt);
|
|
free(tlmt);
|
|
|
|
return r;
|
|
}
|
|
|
|
mime_action
|
|
reply_mime(int depth, struct message *msg, char *body, size_t bodylen)
|
|
{
|
|
(void) depth;
|
|
|
|
char *ct = blaze822_hdr(msg, "content-type");
|
|
char *mt = mimetype(ct);
|
|
char *tlmt = tlmimetype(ct);
|
|
|
|
if (!ct || strncmp(ct, "text/plain", 10) == 0) {
|
|
char *charset = 0, *cs, *cse;
|
|
if (blaze822_mime_parameter(ct, "charset", &cs, &cse))
|
|
charset = strndup(cs, cse-cs);
|
|
if (!charset ||
|
|
strcasecmp(charset, "utf-8") == 0 ||
|
|
strcasecmp(charset, "utf8") == 0 ||
|
|
strcasecmp(charset, "us-ascii") == 0)
|
|
fwrite(body, 1, bodylen, stdout);
|
|
else
|
|
print_u8recode(body, bodylen, charset);
|
|
free(charset);
|
|
}
|
|
|
|
free(mt);
|
|
free(tlmt);
|
|
|
|
return MIME_CONTINUE;
|
|
}
|
|
|
|
mime_action
|
|
list_mime(int depth, struct message *msg, char *body, size_t bodylen)
|
|
{
|
|
(void) body;
|
|
|
|
char *ct = blaze822_hdr(msg, "content-type");
|
|
if (!ct)
|
|
ct = "text/x-unknown";
|
|
char *mt = mimetype(ct);
|
|
char *filename = mime_filename(msg);
|
|
|
|
printf(" %*.s%d: %s size=%zd", depth*2, "", ++mimecount, mt, bodylen);
|
|
if (filename) {
|
|
printf(" name=\"%s\"", filename);
|
|
free(filename);
|
|
}
|
|
printf("\n");
|
|
|
|
return MIME_CONTINUE;
|
|
}
|
|
|
|
mime_action
|
|
walk_mime(struct message *msg, int depth, mime_callback visit)
|
|
{
|
|
char *ct, *body, *bodychunk;
|
|
size_t bodylen;
|
|
|
|
mime_action r = MIME_CONTINUE;
|
|
|
|
if (blaze822_mime_body(msg, &ct, &body, &bodylen, &bodychunk)) {
|
|
|
|
mime_action r = visit(depth, msg, body, bodylen);
|
|
|
|
if (r == MIME_CONTINUE) {
|
|
if (strncmp(ct, "multipart/", 10) == 0) {
|
|
struct message *imsg = 0;
|
|
while (blaze822_multipart(msg, &imsg)) {
|
|
r = walk_mime(imsg, depth+1, visit);
|
|
if (r == MIME_STOP)
|
|
break;
|
|
}
|
|
} else if (strncmp(ct, "message/rfc822", 14) == 0) {
|
|
struct message *imsg = blaze822_mem(body, bodylen);
|
|
if (imsg)
|
|
walk_mime(imsg, depth+1, visit);
|
|
}
|
|
}
|
|
|
|
free(bodychunk);
|
|
}
|
|
|
|
return r;
|
|
}
|
|
|
|
void
|
|
list(char *file)
|
|
{
|
|
struct message *msg = blaze822_file(file);
|
|
if (!msg)
|
|
return;
|
|
mimecount = 0;
|
|
printf("%s\n", file);
|
|
walk_mime(msg, 0, list_mime);
|
|
}
|
|
|
|
void
|
|
reply(char *file)
|
|
{
|
|
struct message *msg = blaze822_file(file);
|
|
if (!msg)
|
|
return;
|
|
walk_mime(msg, 0, reply_mime);
|
|
}
|
|
|
|
static int extract_argc;
|
|
static char **extract_argv;
|
|
static int extract_stdout;
|
|
|
|
|
|
static const char *
|
|
basenam(const char *s)
|
|
{
|
|
char *r = strrchr(s, '/');
|
|
return r ? r + 1 : s;
|
|
}
|
|
|
|
static int
|
|
writefile(char *name, char *buf, ssize_t len)
|
|
{
|
|
int fd = open(basenam(name), O_CREAT | O_EXCL | O_WRONLY, 0666);
|
|
if (fd == -1) {
|
|
perror("open");
|
|
return -1;
|
|
}
|
|
if (write(fd, buf, len) != len) {
|
|
// XXX partial write
|
|
perror("write");
|
|
return -1;
|
|
}
|
|
close(fd);
|
|
return 0;
|
|
}
|
|
|
|
mime_action
|
|
extract_mime(int depth, struct message *msg, char *body, size_t bodylen)
|
|
{
|
|
(void) body;
|
|
(void) depth;
|
|
|
|
char *filename = mime_filename(msg);
|
|
|
|
mimecount++;
|
|
|
|
if (extract_argc == 0) {
|
|
if (extract_stdout) { // output all parts
|
|
fwrite(body, 1, bodylen, stdout);
|
|
} else { // extract all named attachments
|
|
if (filename) {
|
|
printf("%s\n", filename);
|
|
writefile(filename, body, bodylen);
|
|
}
|
|
}
|
|
} else {
|
|
int i;
|
|
for (i = 0; i < extract_argc; i++) {
|
|
char *a = extract_argv[i];
|
|
char *b;
|
|
errno = 0;
|
|
long d = strtol(a, &b, 10);
|
|
if (errno == 0 && !*b && d == mimecount) {
|
|
// extract by id
|
|
if (extract_stdout) {
|
|
fwrite(body, 1, bodylen, stdout);
|
|
} else {
|
|
char buf[255];
|
|
char *bufptr;
|
|
if (filename) {
|
|
bufptr = filename;
|
|
} else {
|
|
snprintf(buf, sizeof buf,
|
|
"attachment%d", mimecount);
|
|
bufptr = buf;
|
|
}
|
|
printf("%s\n", bufptr);
|
|
writefile(bufptr, body, bodylen);
|
|
}
|
|
} else if (filename &&
|
|
fnmatch(a, filename, FNM_PATHNAME) == 0) {
|
|
// extract by name
|
|
if (extract_stdout) {
|
|
fwrite(body, 1, bodylen, stdout);
|
|
} else {
|
|
printf("%s\n", filename);
|
|
writefile(filename, body, bodylen);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
free(filename);
|
|
return MIME_CONTINUE;
|
|
}
|
|
|
|
void
|
|
extract_cb(char *file)
|
|
{
|
|
struct message *msg = blaze822_file(file);
|
|
if (!msg)
|
|
return;
|
|
mimecount = 0;
|
|
walk_mime(msg, 0, extract_mime);
|
|
}
|
|
|
|
void
|
|
extract(char *file, int argc, char **argv, int use_stdout)
|
|
{
|
|
extract_argc = argc;
|
|
extract_argv = argv;
|
|
extract_stdout = use_stdout;
|
|
blaze822_loop1(file, extract_cb);
|
|
}
|
|
|
|
static char *newcur;
|
|
|
|
static void
|
|
print_date_header(char *v)
|
|
{
|
|
static time_t now = -1;
|
|
if (now == -1) {
|
|
setenv("TZ", "", 1);
|
|
tzset();
|
|
now = time(0);
|
|
}
|
|
|
|
printf("Date: %s", v);
|
|
|
|
time_t t = blaze822_date(v);
|
|
if (t == -1) {
|
|
printf(" (invalid)");
|
|
} else {
|
|
printf(" (");
|
|
time_t d = t < now ? now - t : t - now;
|
|
|
|
int l;
|
|
if (d > 60*60*24*7*52) l = 'y';
|
|
else if (d > 60*60*24*7) l = 'w';
|
|
else if (d > 60*60*24) l = 'd';
|
|
else if (d > 60*60) l = 'h';
|
|
else if (d > 60) l = 'm';
|
|
else l = 's';
|
|
int p = 3;
|
|
|
|
int z;
|
|
switch(l) {
|
|
case 'y':
|
|
z = d / (60*60*24*7*52);
|
|
d = d % (60*60*24*7*52);
|
|
if (z > 0) {
|
|
printf("%d year%s", z, z>1 ? "s" : "");
|
|
if (!--p) break;
|
|
printf(", ");
|
|
}
|
|
case 'w':
|
|
z = d / (60*60*24*7);
|
|
d = d % (60*60*24*7);
|
|
if (z > 0) {
|
|
printf("%d week%s", z, z>1 ? "s" : "");
|
|
if (!--p) break;
|
|
printf(", ");
|
|
}
|
|
case 'd':
|
|
z = d / (60*60*24);
|
|
d = d % (60*60*24);
|
|
if (z > 0) {
|
|
printf("%d day%s", z, z>1 ? "s" : "");
|
|
if (!--p) break;
|
|
printf(", ");
|
|
}
|
|
case 'h':
|
|
z = d / (60*60);
|
|
d = d % (60*60);
|
|
if (z > 0) {
|
|
printf("%d hour%s", z, z>1 ? "s" : "");
|
|
if (!--p) break;
|
|
printf(", ");
|
|
}
|
|
case 'm':
|
|
z = d / (60);
|
|
d = d % (60);
|
|
if (z > 0) {
|
|
printf("%d minute%s", z, z>1 ? "s" : "");
|
|
if (!--p) break;
|
|
printf(", ");
|
|
}
|
|
case 's':
|
|
z = d;
|
|
printf("%d second%s", z, z>1 ? "s" : "");
|
|
}
|
|
|
|
if (t < now)
|
|
printf(" ago)");
|
|
else
|
|
printf(" in the future)");
|
|
}
|
|
printf("\n");
|
|
}
|
|
|
|
static void
|
|
print_decode_header(char *h, char *v)
|
|
{
|
|
char d[4096];
|
|
blaze822_decode_rfc2047(d, v, sizeof d, "UTF-8");
|
|
printhdr(h);
|
|
printf(": %s\n", d);
|
|
}
|
|
|
|
void
|
|
show(char *file)
|
|
{
|
|
struct message *msg;
|
|
|
|
while (*file == ' ' || *file == '\t')
|
|
file++;
|
|
|
|
if (newcur) {
|
|
printf("\014\n");
|
|
free(newcur);
|
|
}
|
|
newcur = strdup(file);
|
|
|
|
if (qflag)
|
|
msg = blaze822(file);
|
|
else
|
|
msg = blaze822_file(file);
|
|
if (!msg) {
|
|
fprintf(stderr, "mshow: %s: %s\n", file, strerror(errno));
|
|
return;
|
|
}
|
|
|
|
if (Hflag) { // raw headers
|
|
size_t hl = blaze822_headerlen(msg);
|
|
char *header = malloc(hl);
|
|
if (!header)
|
|
return;
|
|
int fd = open(file, O_RDONLY);
|
|
if (fd == -1) {
|
|
free(header);
|
|
return;
|
|
}
|
|
hl = read(fd, header, hl);
|
|
fwrite(header, 1, hl, stdout);
|
|
} else if (Lflag) { // all headers
|
|
char *h = 0;
|
|
while ((h = blaze822_next_header(msg, h))) {
|
|
char d[4096];
|
|
blaze822_decode_rfc2047(d, h, sizeof d, "UTF-8");
|
|
printhdr(d);
|
|
}
|
|
} else { // selected headers
|
|
char *h = hflag;
|
|
char *v;
|
|
while (*h) {
|
|
char *n = strchr(h, ':');
|
|
if (n)
|
|
*n = 0;
|
|
v = blaze822_chdr(msg, h);
|
|
if (v) {
|
|
if (strcmp("date", h) == 0)
|
|
print_date_header(v);
|
|
else
|
|
print_decode_header(h, v);
|
|
}
|
|
if (n) {
|
|
*n = ':';
|
|
h = n + 1;
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (qflag) // no body
|
|
goto done;
|
|
|
|
printf("\n");
|
|
|
|
if (rflag || !blaze822_check_mime(msg)) { // raw body
|
|
fwrite(blaze822_body(msg), 1, blaze822_bodylen(msg), stdout);
|
|
goto done;
|
|
}
|
|
|
|
mimecount = 0;
|
|
walk_mime(msg, 0, render_mime);
|
|
|
|
done:
|
|
blaze822_free(msg);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
int c;
|
|
while ((c = getopt(argc, argv, "h:qrtHLx:O:Rn")) != -1)
|
|
switch(c) {
|
|
case 'h': hflag = optarg; break;
|
|
case 'q': qflag = 1; break;
|
|
case 'r': rflag = 1; break;
|
|
case 'H': Hflag = 1; break;
|
|
case 'L': Lflag = 1; break;
|
|
case 't': tflag = 1; break;
|
|
case 'x': xflag = optarg; break;
|
|
case 'O': Oflag = optarg; break;
|
|
case 'R': Rflag = 1; break;
|
|
case 'n': nflag = 1; break;
|
|
default:
|
|
// XXX usage
|
|
exit(1);
|
|
}
|
|
|
|
if (xflag) { // extract
|
|
extract(xflag, argc-optind, argv+optind, 0);
|
|
} else if (Oflag) { // extract to stdout
|
|
extract(Oflag, argc-optind, argv+optind, 1);
|
|
} else if (tflag) { // list
|
|
if (argc == optind && isatty(0))
|
|
blaze822_loop1(".", list);
|
|
else
|
|
blaze822_loop(argc-optind, argv+optind, list);
|
|
} else if (Rflag) { // render for reply
|
|
blaze822_loop(argc-optind, argv+optind, reply);
|
|
} else { // show
|
|
if (!(qflag || rflag)) {
|
|
char *f = getenv("MAILFILTER");
|
|
if (!f)
|
|
f = blaze822_home_file(".santoku/filter");
|
|
if (f)
|
|
filters = blaze822(f);
|
|
}
|
|
if (argc == optind && isatty(0))
|
|
blaze822_loop1(".", show);
|
|
else
|
|
blaze822_loop(argc-optind, argv+optind, show);
|
|
if (!nflag) // don't set cur
|
|
blaze822_seq_setcur(newcur);
|
|
}
|
|
|
|
return 0;
|
|
}
|