mirror of https://github.com/leahneukirchen/mblaze
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
469 lines
9.0 KiB
C
469 lines
9.0 KiB
C
8 years ago
|
#include <sys/stat.h>
|
||
8 years ago
|
#include <sys/types.h>
|
||
|
|
||
8 years ago
|
#include <ctype.h>
|
||
|
#include <errno.h>
|
||
8 years ago
|
#include <fcntl.h>
|
||
8 years ago
|
#include <iconv.h>
|
||
8 years ago
|
#include <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <string.h>
|
||
|
#include <unistd.h>
|
||
8 years ago
|
|
||
|
#include "blaze822.h"
|
||
|
|
||
8 years ago
|
static int rflag;
|
||
|
static int qflag;
|
||
|
static int Hflag;
|
||
|
static int Lflag;
|
||
|
static int tflag;
|
||
8 years ago
|
static int nflag;
|
||
8 years ago
|
static char defaulthflags[] = "from:subject:to:cc:date:";
|
||
|
static char *hflag = defaulthflags;
|
||
|
static char *xflag;
|
||
|
static char *Oflag;
|
||
|
|
||
|
struct message *filters;
|
||
|
|
||
|
static int mimecount;
|
||
|
|
||
8 years ago
|
void
|
||
|
printhdr(char *hdr)
|
||
|
{
|
||
|
int uc = 1;
|
||
|
|
||
8 years ago
|
while (*hdr && *hdr != ':') {
|
||
8 years ago
|
putc(uc ? toupper(*hdr) : *hdr, stdout);
|
||
|
uc = (*hdr == '-');
|
||
|
hdr++;
|
||
|
}
|
||
8 years ago
|
|
||
|
if (*hdr) {
|
||
|
printf("%s\n", hdr);
|
||
|
}
|
||
|
}
|
||
|
|
||
8 years ago
|
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);
|
||
|
}
|
||
|
|
||
8 years ago
|
char *
|
||
|
mimetype(char *ct)
|
||
|
{
|
||
|
char *s;
|
||
|
for (s = ct; *s && *s != ';' && *s != ' ' && *s != '\t'; s++)
|
||
|
;
|
||
|
|
||
|
return strndup(ct, s-ct);
|
||
|
}
|
||
|
|
||
|
char *
|
||
|
tlmimetype(char *ct)
|
||
|
{
|
||
|
char *s;
|
||
|
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, char *, char *, size_t);
|
||
|
|
||
|
mime_action
|
||
|
render_mime(int depth, char *ct, char *body, size_t bodylen)
|
||
|
{
|
||
|
char *mt = mimetype(ct);
|
||
|
char *tlmt = tlmimetype(ct);
|
||
|
|
||
8 years ago
|
char *filename = 0, *fn, *fne;
|
||
8 years ago
|
if (blaze822_mime_parameter(ct, "name", &fn, &fne))
|
||
|
filename = strndup(fn, fne-fn);
|
||
|
|
||
|
mimecount++;
|
||
|
|
||
|
int i;
|
||
|
for (i = 0; i < depth+1; i++)
|
||
|
printf("--- ");
|
||
|
printf("%d: %s size=%zd", mimecount, mt, bodylen);
|
||
8 years ago
|
if (filename) {
|
||
8 years ago
|
printf(" name=%s", filename);
|
||
8 years ago
|
free(filename);
|
||
|
}
|
||
8 years ago
|
|
||
|
char *cmd;
|
||
|
mime_action r = MIME_CONTINUE;
|
||
|
|
||
|
if (filters &&
|
||
|
((cmd = blaze822_chdr(filters, mt)) ||
|
||
|
(cmd = blaze822_chdr(filters, tlmt)))) {
|
||
|
printf(" filter=\"%s\" ---\n", cmd);
|
||
|
FILE *p;
|
||
|
fflush(stdout);
|
||
|
p = popen(cmd, "w");
|
||
|
if (!p) {
|
||
|
perror("popen");
|
||
|
goto nofilter;
|
||
|
}
|
||
8 years ago
|
fwrite(body, 1, bodylen, p);
|
||
8 years ago
|
if (pclose(p) != 0) {
|
||
|
perror("pclose");
|
||
|
goto nofilter;
|
||
|
}
|
||
|
r = MIME_PRUNE;
|
||
|
} else {
|
||
|
nofilter:
|
||
|
printf(" ---\n");
|
||
|
|
||
|
if (strncmp(ct, "text/", 5) == 0) {
|
||
8 years ago
|
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)
|
||
8 years ago
|
fwrite(body, 1, bodylen, stdout);
|
||
8 years ago
|
else
|
||
|
print_u8recode(body, bodylen, charset);
|
||
|
free(charset);
|
||
8 years ago
|
} 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
|
||
|
list_mime(int depth, char *ct, char *body, size_t bodylen)
|
||
|
{
|
||
|
(void) body;
|
||
|
|
||
|
char *mt = mimetype(ct);
|
||
|
char *fn, *fne;
|
||
|
|
||
|
printf("%*.s%d: %s size=%zd", depth*2, "", ++mimecount, mt, bodylen);
|
||
|
if (blaze822_mime_parameter(ct, "name", &fn, &fne)) {
|
||
|
printf(" name=");
|
||
|
fwrite(fn, 1, fne-fn, stdout);
|
||
|
}
|
||
|
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, ct, 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;
|
||
8 years ago
|
}
|
||
|
|
||
|
void
|
||
8 years ago
|
list(char *file)
|
||
8 years ago
|
{
|
||
8 years ago
|
struct message *msg = blaze822_file(file);
|
||
|
if (!msg)
|
||
|
return;
|
||
|
mimecount = 0;
|
||
|
walk_mime(msg, 0, list_mime);
|
||
|
}
|
||
|
|
||
|
static int extract_argc;
|
||
|
static char **extract_argv;
|
||
|
static int extract_stdout;
|
||
|
|
||
|
static int
|
||
|
writefile(char *name, char *buf, ssize_t len)
|
||
|
{
|
||
|
int fd = open(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, char *ct, char *body, size_t bodylen)
|
||
|
{
|
||
|
(void) body;
|
||
|
(void) depth;
|
||
|
|
||
|
char *filename = 0, *fn, *fne;
|
||
|
if (blaze822_mime_parameter(ct, "name", &fn, &fne))
|
||
|
filename = strndup(fn, fne-fn);
|
||
|
|
||
|
mimecount++;
|
||
8 years ago
|
|
||
8 years ago
|
if (extract_argc == 0) {
|
||
|
if (extract_stdout) { // output all parts
|
||
8 years ago
|
fwrite(body, 1, bodylen, stdout);
|
||
8 years ago
|
} 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) {
|
||
8 years ago
|
fwrite(body, 1, bodylen, stdout);
|
||
8 years ago
|
} else {
|
||
|
char buf[255];
|
||
|
if (!filename) {
|
||
|
snprintf(buf, sizeof buf,
|
||
|
"attachment%d", mimecount);
|
||
|
filename = buf;
|
||
|
}
|
||
|
printf("%s\n", filename);
|
||
|
writefile(filename, body, bodylen);
|
||
|
}
|
||
|
} else if (filename && strcmp(a, filename) == 0) {
|
||
|
// extract by name
|
||
|
if (extract_stdout) {
|
||
8 years ago
|
fwrite(body, 1, bodylen, stdout);
|
||
8 years ago
|
} else {
|
||
|
printf("%s\n", filename);
|
||
|
writefile(filename, body, bodylen);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
free(filename);
|
||
|
return MIME_CONTINUE;
|
||
|
}
|
||
|
|
||
|
void
|
||
|
extract(char *file, int argc, char **argv, int use_stdout)
|
||
|
{
|
||
|
struct message *msg = blaze822_file(file);
|
||
8 years ago
|
if (!msg)
|
||
|
return;
|
||
8 years ago
|
mimecount = 0;
|
||
|
extract_argc = argc;
|
||
|
extract_argv = argv;
|
||
|
extract_stdout = use_stdout;
|
||
|
walk_mime(msg, 0, extract_mime);
|
||
|
}
|
||
8 years ago
|
|
||
8 years ago
|
static char *newcur;
|
||
|
|
||
8 years ago
|
void
|
||
|
show(char *file)
|
||
|
{
|
||
|
struct message *msg;
|
||
8 years ago
|
|
||
8 years ago
|
while (*file == ' ' || *file == '\t')
|
||
|
file++;
|
||
|
|
||
|
if (newcur) {
|
||
|
printf("\014\n");
|
||
|
free(newcur);
|
||
|
}
|
||
|
newcur = strdup(file);
|
||
|
|
||
8 years ago
|
if (qflag)
|
||
|
msg = blaze822(file);
|
||
|
else
|
||
|
msg = blaze822_file(file);
|
||
8 years ago
|
if (!msg) {
|
||
|
fprintf(stderr, "show: %s: %s\n", file, strerror(errno));
|
||
8 years ago
|
return;
|
||
8 years ago
|
}
|
||
8 years ago
|
|
||
8 years ago
|
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)
|
||
|
return;
|
||
|
hl = read(fd, header, hl);
|
||
8 years ago
|
fwrite(header, 1, hl, stdout);
|
||
8 years ago
|
} 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) {
|
||
|
printhdr(h);
|
||
|
printf(": %s\n", v);
|
||
|
}
|
||
|
if (n) {
|
||
|
*n = ':';
|
||
|
h = n + 1;
|
||
|
} else {
|
||
|
break;
|
||
|
}
|
||
8 years ago
|
}
|
||
|
}
|
||
|
|
||
8 years ago
|
if (qflag) // no body
|
||
|
goto done;
|
||
|
|
||
8 years ago
|
printf("\n");
|
||
|
|
||
8 years ago
|
if (rflag || !blaze822_check_mime(msg)) { // raw body
|
||
8 years ago
|
fwrite(blaze822_body(msg), 1, blaze822_bodylen(msg), stdout);
|
||
8 years ago
|
goto done;
|
||
|
}
|
||
|
|
||
|
mimecount = 0;
|
||
|
walk_mime(msg, 0, render_mime);
|
||
|
|
||
|
done:
|
||
8 years ago
|
blaze822_free(msg);
|
||
8 years ago
|
}
|
||
|
|
||
|
int
|
||
|
main(int argc, char *argv[])
|
||
|
{
|
||
8 years ago
|
int c;
|
||
8 years ago
|
while ((c = getopt(argc, argv, "h:qrtHLx:O:n")) != -1)
|
||
8 years ago
|
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;
|
||
8 years ago
|
case 'n': nflag = 1; break;
|
||
8 years ago
|
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
|
||
|
blaze822_loop(argc-optind, argv+optind, list);
|
||
|
} else { // show
|
||
|
if (!(qflag || rflag))
|
||
|
filters = blaze822("filters");
|
||
|
blaze822_loop(argc-optind, argv+optind, show);
|
||
8 years ago
|
if (!nflag) // don't set cur
|
||
|
blaze822_seq_setcur(newcur);
|
||
8 years ago
|
}
|
||
8 years ago
|
|
||
|
return 0;
|
||
|
}
|