Plan 9 from Bell Labs’s /usr/web/sources/contrib/nemo/root/sys/src/cmd/mail2fs/mail2fs.c

Copyright © 2021 Plan 9 Foundation.
Distributed under the MIT License.
Download the Plan 9 distribution.


/*
 * Use upas/fs to convert a plan 9 mail box into
 * a file tree with a plan b mail box (similar to Mh).
 *
 * Implementation is not as clean as it should be,
 * A large part of the code is taken as-is from nedmail, trying to
 * keep it the same, so that bugs are easy to fix both
 * here and in nedmail (should they show up).
 * We should get rid of Strings, for example.
 *
 */

#include "common.h"
#include <ctype.h>
#include <libsec.h>
#include "util.h"

enum {
	Defperm = 0666, /* none needs access for delivery, else could be 0620 */
};

typedef struct Message Message;
typedef struct Ctype Ctype;
typedef struct Cmd Cmd;

struct Message {
	Message	*next;
	Message	*prev;
	Message	*cmd;
	Message	*child;
	Message	*parent;
	String	*path;
	int	id;
	int	len;
	int	fileno;		/* number of directory */
	String	*info;
	char	*from;
	char	*to;
	char	*cc;
	char	*replyto;
	char	*date;
	char	*subject;
	char	*type;
	char	*disposition;
	char	*filename;
	char	*sdigest;
	char	deleted;
	char	stored;
	int	saveit;
};

struct Ctype {
	char	*type;
	char 	*ext;
	int	display;
	char	*plumbdest;
	Ctype	*next;
};

Ctype ctype[] = {
	{ "text/plain",			"txt",	1,	0	},
	{ "text/html",			"htm",	1,	0	},
	{ "text/html",			"html",	1,	0	},
	{ "text/tab-separated-values",	"tsv",	1,	0	},
	{ "text/richtext",		"rtx",	1,	0	},
	{ "text/rtf",			"rtf",	1,	0	},
	{ "text",			"txt",	1,	0	},
	{ "message/rfc822",		"msg",	0,	0	},
	{ "image/bmp",			"bmp",	0,	"image"	},
	{ "image/jpeg",			"jpg",	0,	"image"	},
	{ "image/gif",			"gif",	0,	"image"	},
	{ "image/png",			"png",	0,	"image"	},
	{ "application/pdf",		"pdf",	0,	"postscript"	},
	{ "application/postscript",	"ps",	0,	"postscript"	},
	{ "application/",		0,	0,	0	},
	{ "image/",			0,	0,	0	},
	{ "multipart/",			"mul",	0,	0	},

};

Rune	altspc = L'·';
Rune	altlparen = L'«';
Rune	altrparen = L'»';
Rune	altquote = L'´';
Rune	altamp = L'­';
Rune	altslash = L'.';

enum
{
	NARG=	32,
};

struct Cmd {
	Message	*msgs;
	Message	*(*f)(Cmd*, Message*);
	int	an;
	char	*av[NARG];
	int	delete;
};

Message	top;		/* top-level of upas mailbox */

int	mboxfd;		/* lock/seq fd for plan b mail box */
int	debug;
int	dry;
int	cflag;		/* create */
int	aflag;		/* archive */
int	oflag;		/* old mail */
int	rflag;		/* save complete raw message */
int	mboxisfile;

char*	digests;		/* in-memory copy of digests */
char*	mboxnm;		/* name for mbox in upas */

void
tabs(int n)
{
	static char t[] = "\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t";

	if(n >= sizeof t)
		n = sizeof t - 1;	/* don't run off the end, truncate */
	write(2, t, n);
}

void
printmessage(Message* m, int t)
{
	Message* l;

	tabs(t);
	fprint(2, "message %d file %d %d bytes\n", m->id, m->fileno, m->len);
	if(m->info != nil){
		tabs(t);
		fprint(2, "info %s\n", s_to_c(m->info));
	}
	if(m->from != nil){
		tabs(t);
		fprint(2, "from %s\n", m->from);
	}
	if(m->subject != nil){
		tabs(t);
		fprint(2, "subject %s\n", m->subject);
	}
	if(m->filename != nil){
		tabs(t);
		fprint(2, "file %s\n", m->filename);
	}
	for(l = m->child; l != nil ; l = l->next)
		printmessage(l, t+1);
	fprint(2, "\n");
}

String*
extendpath(String *dir, char *name)
{
	String *path;

	if(strcmp(s_to_c(dir), ".") == 0)
		path = s_new();
	else {
		path = s_copy(s_to_c(dir));
		s_append(path, "/");
	}
	s_append(path, name);
	return path;
}

String*
file2string(String *dir, char *file)
{
	String *s;
	int fd, n, m;

	s = extendpath(dir, file);
	fd = open(s_to_c(s), OREAD);
	s_grow(s, 512);			/* avoid multiple reads on info files */
	s_reset(s);
	if(fd < 0)
		return s;

	for(;;){
		n = s->end - s->ptr;
		if(n == 0){
			s_grow(s, 128);
			continue;
		}
		m = read(fd, s->ptr, n);
		if(m <= 0)
			break;
		s->ptr += m;
		if(m < n)
			break;
	}
	s_terminate(s);
	close(fd);

	return s;
}

int
lineize(char *s, char **f, int n)
{
	int i;

	for(i = 0; *s && i < n; i++){
		f[i] = s;
		s = strchr(s, '\n');
		if(s == nil)
			break;
		*s++ = 0;
	}
	return i;
}

int
filelen(String *dir, char *file)
{
	String *path;
	Dir *d;
	int rv;

	path = extendpath(dir, file);
	d = dirstat(s_to_c(path));
	if(d == nil){
		s_free(path);
		return -1;
	}
	s_free(path);
	rv = d->length;
	free(d);
	return rv;
}


int dir2message(Message *parent, int reverse);

Message*
file2message(Message *parent, char *name)
{
	Message *m;
	String *path;
	char *f[10];

	m = mallocz(sizeof(Message), 1);
	if(m == nil)
		return nil;
	m->path = path = extendpath(parent->path, name);
	m->fileno = atoi(name);
	m->info = file2string(path, "info");
	lineize(s_to_c(m->info), f, nelem(f));
	m->from = f[0];
	m->to = f[1];
	m->cc = f[2];
	m->replyto = f[3];
	m->date = f[4];
	m->subject = f[5];
	m->type = f[6];
	m->disposition = f[7];
	m->filename = f[8];
	m->len = filelen(path, "raw");
	if(strstr(m->type, "multipart") != nil ||
	 strcmp(m->type, "message/rfc822") == 0)
		dir2message(m, 0);
	m->parent = parent;

	return m;
}

int
dir2message(Message *parent, int reverse)
{
	int i, n, fd, highest, newmsgs;
	Dir *d;
	Message *first, *last, *m;

	fd = open(s_to_c(parent->path), OREAD);
	if(fd < 0)
		return -1;

	/* count current entries */
	first = parent->child;
	highest = newmsgs = 0;
	for(last = parent->child; last != nil && last->next != nil;
	 last = last->next)
		if(last->fileno > highest)
			highest = last->fileno;
	if(last != nil)
		if(last->fileno > highest)
			highest = last->fileno;

	n = dirreadall(fd, &d);
	for(i = 0; i < n; i++){
		if((d[i].qid.type & QTDIR) == 0)
			continue;
		if(atoi(d[i].name) <= highest)
			continue;
		m = file2message(parent, d[i].name);
		if(m == nil)
			break;
		newmsgs++;
		if(reverse){
			m->next = first;
			if(first != nil)
				first->prev = m;
			first = m;
		} else {
			if(first == nil)
				first = m;
			else
				last->next = m;
			m->prev = last;
			last = m;
		}
	}
	free(d);
	close(fd);
	parent->child = first;

	/* renumber */
	for(m = first; m != nil; m = m->next)
		m->id = m->fileno;

	return newmsgs;
}

void
compress(char *p)
{
	char *np;
	int last;

	last = ' ';
	for(np = p; *p; p++)
		if(*p != '
' && (*p != ' ' || last != ' ')){
			last = *p;
			*np++ = last;
		}
	*np = 0;
}

int
printheader(Biobuf* bout, Message* m)
{
	if(m->from == nil || *m->from == 0)
		return 0;
	Bprint(bout, "From: %s\n", m->from);
	if(m->to != nil && *m->to)
		Bprint(bout, "To: %s\n", m->to);
	if(m->cc != nil && *m->cc)
		Bprint(bout, "Cc: %s\n", m->cc);
	if(m->replyto != nil && *m->replyto && m->from != nil && strcmp(m->replyto, m->from))
		Bprint(bout, "Reply-To: %s\n", m->replyto);
	if(m->date != nil && *m->date)
		Bprint(bout, "Date: %s\n", m->date);
	if(m->subject != nil && *m->subject)
		Bprint(bout, "Subject: %s\n", m->subject);
	return 1;
}

int
printpart(Biobuf* bout, String *s, char *part)
{
	char buf[4096];
	int n, fd, tot;
	String *path;

	path = extendpath(s, part);
	fd = open(s_to_c(path), OREAD);
	s_free(path);
	if(fd < 0){
		fprint(2, "!message disappeared\n");
		return 0;
	}
	tot = 0;
	while((n = read(fd, buf, sizeof(buf)-1)) > 0){
		buf[n] = 0;
		compress(buf);
		n = strlen(buf);
		if(Bwrite(bout, buf, n) <= 0)
			break;
		tot += n;
	}
	close(fd);
	return tot;
}

/*
 * Mail box format:
 * dir/			(mail box)
 *	seq		(file with last msg nb, +l)
 *	digest		(file with mail digests)
 *	yyyymm/		(per month folder)
 *		raw	(raw message; just headers)
 *		text	(text as in reader)
 *		1.x	(first attach)
 *		2.x	(second attach)
 *		3/	(third attach, a mail)
 */

int
mkmdir(char* mdir)
{
	int	fd, seqfd, nattempts;
	char	buf[200];
	char*	s;
	Dir*	d;

	digests = strdup("");
	d = dirstat(mdir);
	s = smprint("%s/seq", mdir);
	if(d == nil){
		if(!cflag)
			sysfatal("%s: not a mail dir", mdir);
		if(debug)
			fprint(2, "create mdir %s\n", mdir);

		fd = create(mdir, OREAD, DMDIR|Defperm|0111);
		if(fd < 0)
			sysfatal("%s: %r", mdir);
		close(fd);

		fd = seqfd = create(s, ORDWR, Defperm|DMEXCL);
		if(fd < 0)
			sysfatal("%s: %r", s);
		fprint(fd, "%11.11d", 0);
		seek(fd, 0, 0);
		free(s);

		s = smprint("%s/digest", mdir);
		fd = create(s, OREAD, Defperm|DMAPPEND|DMEXCL);
		if(fd < 0)
			sysfatal("%s: %r", s);
		close(fd);
	} else {
		if((d->qid.type&QTDIR) == 0)
			sysfatal("%s: not a directory", mdir);
		free(d);
		nattempts = 0;
		while ((seqfd = open(s, ORDWR)) < 0){
			rerrstr(buf, sizeof buf);
			if(strstr(buf, "exclusive") == nil &&
			 strstr(buf, "already open") == nil)
				break;
			if(nattempts++ >= 10)
				sysfatal("%s: can't get lock: %r", mdir);
			sleep(500);
		}
		if (seqfd < 0)
			seqfd = create(s, ORDWR, Defperm|DMEXCL);
		if (seqfd < 0)
			sysfatal("can't create %s: %r", s);
	}
	free(s);
	return seqfd;
}

void
closemdir(int fd)
{
	close(fd);
}

int
newseq(int startover)
{
	int	i, nr;
	char	buf[128];

	if(mboxfd < 0)
		return -1;
	seek(mboxfd, 0, 0);
	nr = read(mboxfd, buf, sizeof(buf)-1);
	if(nr < 0)
		return -1;
	if(nr == 0)
		i = 0;
	else {
		buf[nr] = 0;
		i = atoi(buf);
	}
	i++;

	if(startover && !oflag)		/* start sequencing each month */
		i = 1;			/* unless we are adding old msgs */
	seek(mboxfd, 0, 0);
	fprint(mboxfd, "%11d ", i);
	if(debug)
		fprint(2, "newmsg %d\n", i);
	return i;
}

char*
mksubmsg(int id, char* mdir)
{
	char*	fname;
	int	fd;

	if(aflag)
		fname = smprint("%s/a.%d", mdir, id);
	else
		fname = smprint("%s/%d", mdir, id);
	if(debug)
		fprint(2, "create %s\n", fname);
	fd = create(fname, OREAD, Defperm|0111|DMDIR);
	if(fd < 0){
		fprint(2, "%s: %r\n", fname);
		free(fname);
		fname = nil;
	} else
		close(fd);
	return fname;
}

static char*
datedir(void)
{
	Tm *tm;
	static char datebuf[30];

	if(datebuf[0] == 0){
		tm = localtime(time(0));
		seprint(datebuf, datebuf+30, "%.4d%.2d",
			tm->year+1900, tm->mon+1);
	}
	return datebuf;
}

static struct{
	char *m;
	char *n;
} mnames[] = {
	{ "jan", "01"}, {"feb", "02"}, {"mar", "03"}, {"apr", "04"}, {"may", "05"}, {"jun", "06"},
	{ "jul", "07"}, {"aug", "08"}, {"sep", "09"}, {"oct", "10"}, {"nov", "11"}, {"dec", "12"}
};

static char*
mname(char *m)
{
	int i;

	for(i = 0; i < nelem(mnames); i++)
		if(cistrcmp(m, mnames[i].m))
			return mnames[i].n;
	return "00";
}

int
isnumeric(char *s)
{
	while(*s){
		if(!isdigit(*s))
			return 0;
		s++;
	}
	return 1;
}

static char*
mdatedir(Message *m)
{
	int n;
	char *s, *fld[10];
	static char datebuf[30];
	char *month, *year;

	n = 0;
	s= nil;
	if(m->date != nil){
		s = strdup(m->date);
		n = tokenize(s, fld, 10);
	}
	if(n >= 5){
		/* some dates have 19 Apr, some Apr 19 */
		if(strlen(fld[1])<4 && isnumeric(fld[1]))
			month = fld[2];
		else
			month = fld[1];
		month = mname(month);
		if(n > 5)
			year = fld[5];
		else
			year = fld[4];
		seprint(datebuf, datebuf+30, "%.4s%.2s", year, month);
	}else
		fprint(2, "%d fields [%s]\n", n, m->date);

	free(s);
	if(datebuf[0] != 0){
		if(debug)
			fprint(2, "olddir: %s\n", datebuf);
		return datebuf;	/* may be an old one if we failed. that's ok. */
	}else
		return datedir();
}

/*
 * Create directory for the message, including the per-month
 * directory.
 * Date is today or that from the mail if oflag (old mail convert)
 */
char*
mkmsg(Message *m, char* mdir)
{
	int	fd;
	char *d, *fname;
	int newmonth;

	d = smprint("%s/%s", mdir, oflag ? mdatedir(m) : datedir());
	newmonth = 0;
	if(access(d, AEXIST) < 0){
		fd = create(d, OREAD, Defperm|0111|DMDIR);
		if(fd < 0){
			fprint(2, "%s: %s: %r\n", argv0, d);
			free(d);
			return nil;
		}
		close(fd);
		newmonth=1;
	}
	fname = mksubmsg(newseq(newmonth), d);
	free(d);
	return fname;
}


Ctype*
findctype(Message *m)
{
	int n, kid, pid, pfd[2];
	char *p;
	char ftype[128];
	Ctype *a, *cp;
	static Ctype nulltype	= { "", 0, 0, 0 };
	static Ctype bintype 	= { "application/octet-stream", "bin", 0, 0 };

	for(cp = ctype; cp; cp = cp->next)
		if(strncmp(cp->type, m->type, strlen(cp->type)) == 0)
			return cp;

	/*
	 * use file(1) for any unknown mime types
	 */
	if(pipe(pfd) < 0) {
		fprint(2, "%s: out of pipes: %r\n", argv0);
		return &bintype;
	}

	*ftype = 0;
	switch(kid = fork()){
	case -1:
		break;
	case 0:
		close(pfd[1]);
		close(0);
		dup(pfd[0], 0);
		close(1);
		dup(pfd[0], 1);
		execl("/bin/file", "file", "-m",
			s_to_c(extendpath(m->path, "body")), nil);
		sysfatal("no /bin/file: %r");
	default:
		close(pfd[0]);
		n = read(pfd[1], ftype, sizeof(ftype));
		if(n > 0)
			ftype[n] = 0;
		close(pfd[1]);
		while ((pid = waitpid()) != -1 && pid != kid)
			;
		break;
	}

	if(*ftype=='\0' || (p = strchr(ftype, '/')) == nil)
		return &bintype;
	*p++ = 0;

	a = mallocz(sizeof(Ctype), 1);
	a->type = strdup(ftype);
	a->ext = strdup(p);
	a->display = 0;
	a->plumbdest = strdup(ftype);
	for(cp = ctype; cp->next; cp = cp->next)
		continue;
	cp->next = a;
	a->next = nil;
	return a;
}

void
pipecmd(Cmd *c, Biobuf* bout, Message *m, char* part)
{
	int i, fd, nr, kid, pid;
	int pfd[2];
	char *p, *e;
	char cmd[128], buf[512];
	char *av[4];
	String *path;

	path = extendpath(m->path, part);
	fd = open(s_to_c(path), OREAD);
	s_free(path);
	if(fd < 0){	/* compatibility with older upas/fs */
		path = extendpath(m->path, "raw");
		fd = open(s_to_c(path), OREAD);
		s_free(path);
	}
	if(fd < 0){
		fprint(2, "!message disappeared\n");
		return;
	}

	p = cmd;
	e = cmd + sizeof cmd;
	cmd[0] = 0;
	for(i = 1; i < c->an; i++)
		p = seprint(p, e, "%s ", c->av[i]);
	av[0] = "rc";
	av[1] = "-c";
	av[2] = cmd;
	av[3] = 0;
	if(pipe(pfd)<0){
		fprint(2, "%s: pipe: %r\n", argv0);
		return;
	}
	kid = fork();
	switch(kid){
	case -1:
		fprint(2, "%s: fork: %r\n", argv0);
		break;
	case 0:
		dup(fd, 0);
		close(fd);
		close(pfd[0]);
		dup(pfd[1], 1);
		close(pfd[1]);
		exec("/bin/rc", av);
		sysfatal("no /bin/rc: %r");
	default:
		close(fd);
		close(pfd[1]);
		while ((nr = read(pfd[0], buf, sizeof buf)) > 0)
			Bwrite(bout, buf, nr);
		close(pfd[0]);
		while ((pid = waitpid()) != -1 && pid != kid)
			;
		break;
	}
}

int
printhtml(Biobuf* bout, Message *m)
{
	Cmd c;

	c.an = 3;
	c.av[1] = "/bin/htmlfmt";
	c.av[2] = "-a -l 80 -cutf-8";
	pipecmd(&c, bout, m, "body");
	return 0;
}

void
fcopy(char* d, char* s, int append)
{
	int dfd, sfd, nr;
	char buf[1024];

	if(debug)
		fprint(2, "create %s\n", d);
	if(append){
		dfd = open(d, OWRITE);
		if(dfd >= 0)
			seek(dfd, 0, 2);
	}else
		dfd = create(d, OWRITE, 0660 /* TODO 0640 */);
	sfd = open(s, OREAD);
	if(dfd < 0 || sfd < 0){
		fprint(2, "%s: copying %s to %s: %r\n", argv0, s, d);
		goto fail;
	}
	while ((nr = read(sfd, buf, sizeof buf)) > 0)
		if(write(dfd, buf, nr) != nr){
			fprint(2, "%s: copying %s to %s: %r\n", argv0, s, d);
			break;
		}
fail:
	close(dfd);
	close(sfd);
}

char*
importname(char* name)
{
	int nr;
	Rune r;
	char *up, *p9name, *uname;

	uname = name;
	if(name == 0 ||
	 (strchr(name, ' ') == 0 && strchr(name, '(') == 0 &&
	 strchr(name, ')') == 0 && strchr(name, '&') == 0 &&
	 strchr(name, '\'') == 0 && strchr(name, '/') == 0))
		return name;
	p9name = malloc(strlen(name) * 3 + 1);	/* worst case: all blanks + 0 */
	up = p9name;
	while(*name != 0){
		nr = chartorune(&r, name);
		if(r == ' ' || r == '\n' || r == '\r' || r == '\t')
			r = altspc;
		else if(r == '(')
			r = altlparen;
		else if(r == ')')
			r = altrparen;
		else if(r == '&')
			r = altamp;
		else if(r == '\'')
			r = altquote;
		else if(r == '/')
			r = altslash;
		else if(!isalpharune(r)){
			free(p9name);
			free(uname);
			return nil;
		}
		up += runetochar(up, &r);
		name += nr;
	}
	*up = 0;
	free(uname);
	return p9name;
}

char*
mfname(Message* m, Ctype* cp)
{
	char *s, *r;

	if(m->filename == nil || *m->filename == 0)
		s = smprint("%d", m->id);
	else
		s = strdup(m->filename);
	s = importname(s);
	if(s == nil)
		s = smprint("%d", m->id);
	if(cp->ext != nil)
		if(strstr(s, cp->ext) != s + strlen(s) - strlen(cp->ext)){
			r = smprint("%s.%s",s, cp->ext);
			free(s);
			return r;
		}

	return s;
}

static void
addnl(char *f)
{
	int fd;

	fd = open(f, OWRITE);
	if(fd < 0)
		return;
	seek(fd, 0, 2);
	write(fd, "\n\n", 2);
	close(fd);
}

static char*
xtabs(int n)
{
	static char t[20];

	memset(t, '\t', sizeof(t));
	t[n] = 0;
	return t;
}
/*
 * Main conversion routine.
 * Convert message m into a directory.
 * It may recur for compound msgs, bout
 * is the message text, nil in the first call.
 */
int
msg2fs(Biobuf* bout, Message* m, char* mdir)
{
	char *dfname, *f, *fname, *msgdir;
	Ctype *cp;
	Message *nm;
	String *s;
	static int nest = 0;

	nest++;
	msgdir = nil;
	if(m->parent == &top || strcmp(m->type, "message/rfc822") == 0){
		if(m->parent == &top)
			msgdir = mkmsg(m, mdir);
		else
			msgdir = mksubmsg(m->id, mdir);
		if(msgdir == nil){
			nest--;
			return -1;
		}
		mdir = msgdir;
		fname = smprint("%s/text", mdir);
		if(debug){
			if(m->parent == &top)
				fprint(2, "%stoplevel\n", xtabs(nest));
			fprint(2, "%screate %s\n", xtabs(nest), fname);
		}
		if(oflag)
			print("%s\n", fname);
		close(create(fname, OWRITE, 0666 /* TODO 0640 */));
		bout = Bopen(fname, OWRITE);
		if(bout == nil){
			fprint(2, "%s: %s: %r\n", argv0, fname);
			free(fname);
			free(msgdir);
			nest--;
			return -1;
		}
		free(fname);

		dfname = smprint("%s/raw", mdir);
		fname = smprint("%s/unixheader", s_to_c(m->path));
		fcopy(dfname, fname, 0);
		free(fname);
		/* save entire original in raw or save just headers? */
		if(rflag){
			fname = smprint("%s/raw", s_to_c(m->path));
			fcopy(dfname, fname, 1);
			free(fname);
		}else{
			fname = smprint("%s/rawheader", s_to_c(m->path));
			fcopy(dfname, fname, 1);
			addnl(dfname);
			free(fname);
		}
		free(dfname);
	}
	if(bout != nil && printheader(bout, m) > 0)
		Bprint(bout, "\n");
	cp = findctype(m);
	if(cp == nil)
		fprint(2, "nil ctype\n");
	else if(cp->display){
		if(debug)
			fprint(2, "%sbody\n", xtabs(nest));
		if(strcmp(m->type, "text/html") == 0)
			printhtml(bout, m);
		else
			printpart(bout, m->path, "body");
		/* experiment: save also inline attachments */
		if(m->saveit)
			goto Save;
	} else if(strcmp(m->type, "multipart/alternative") == 0){
		if(debug){
			fprint(2, "%smultipart/alternative\n", xtabs(nest));
			for(nm = m->child; nm != nil; nm = nm->next)
				fprint(2, "%s->%s\n", xtabs(nest+1), nm->type);
		}
		/* all parts should be equivalent, according to mime. however,
		 * Apple mail dares to place attachs into a part that is multipart,
		 * which is not equivalent to the first text part we find. So, we must
		 * prefer a multipart subpart to any other plain text or displayed part.
		 */
		for(nm = m->child; nm != nil; nm = nm->next)
			if(strncmp(nm->type, "multipart/", 10) == 0)
				break;
		if(nm == nil)
			for(nm = m->child; nm != nil; nm = nm->next){
				cp = findctype(nm);
				if(cp->ext != nil && strncmp(cp->ext, "txt", 3) == 0)
					break;
			}
		if(nm == nil)
			for(nm = m->child; nm != nil; nm = nm->next){
				cp = findctype(nm);
				if(cp->display)
					break;
			}
		if(nm != nil)
			msg2fs(bout, nm, mdir);
	} else if(strncmp(m->type, "multipart/", 10) == 0){
		if(debug)
			fprint(2, "%smultipart\n", xtabs(nest));
		nm = m->child;
		if(nm != nil){
			/* always print first part */
			msg2fs(bout, nm, mdir);

			for(nm = nm->next; nm != nil; nm = nm->next){
				fname = smprint("%s", s_to_c(nm->path));
				s = s_copy(fname);
				free(fname);
				cp = findctype(nm);
				if(nm->type != nil &&
				 strcmp(nm->type, "message/rfc822") == 0){
					Bprint(bout, "\n— %d/\n", nm->id);
					msg2fs(bout, nm, mdir);
				} else {
					f = mfname(nm, cp);
					Bprint(bout,"\n!— %s\n\n",f);
					free(f);
					/* experiment: save also inline attachments */
					nm->saveit = 1;
					msg2fs(bout, nm, mdir);
				}
				s_free(s);
			}
		}
	} else if(strcmp(m->type, "message/rfc822") == 0){
		if(debug)
			fprint(2, "%smessage/rfc822\n", xtabs(nest));
		msg2fs(bout, m->child, mdir);
	} else {
	Save:
		if(debug)
			fprint(2, "%sattach\n", xtabs(nest));
		f = mfname(m, cp);
		dfname = smprint("%s/%s", mdir, f);
		free(f);
		if(cp->ext)
			fname = smprint("%s/body.%s", s_to_c(m->path), cp->ext);
		else
			fname = smprint("%s/body", s_to_c(m->path));
		fcopy(dfname, fname, 0);
		free(dfname);
		free(fname);
	}
	free(msgdir);
	if(m->parent == &top || strcmp(m->type, "message/rfc822") == 0)
		Bterm(bout);
	nest--;
	return 0;
}

/*
 * Compute signature for mail.
 * Msgs with same signature are considered dups and not
 * added to the mail box.
 * Returns 0 if message is already there.
 */
static char sdigest[MD5dlen*2+1];

char*
lastsigned(void)
{
	return sdigest;
}

int
sign(Message* m)
{
	int fd, i, n;
	char *edigest, *odigests, *p;
	uchar buf[256], digest[MD5dlen];
	DigestState *s;
	String *path;

	s = md5((uchar*)"mail2fs", 7, nil, nil);

	path = extendpath(m->path, "subject");
	fd = open(s_to_c(path), OREAD);
	s_free(path);
	if(fd < 0)
		goto fail;
	n = read(fd, buf, sizeof(buf));
	if(n < 0)
		goto fail;
	if(n > 0)
		md5(buf, n, nil, s);
	close(fd);
	path = extendpath(m->path, "body");
	fd = open(s_to_c(path), OREAD);
	s_free(path);
	if(fd < 0)
		goto fail;
	while((n = read(fd, buf, 256)) > 0)
		md5(buf, n, nil, s);
	md5((uchar*)"mail2fs", 7, digest, s);
	close(fd);
	p = sdigest;
	edigest = sdigest + sizeof(sdigest);
	for(i = 0; i < MD5dlen; i++)
		p = seprint(p, edigest, "%02x", digest[i]);
	*p = 0;
	if(strstr(digests, sdigest)){
		if(debug)
			fprint(2, "%s: %s: message already in digests\n",
				argv0, s_to_c(m->path));
		return 0;
	}
	odigests = digests;
	digests = smprint("%s%s\n", odigests, sdigest);
	if(digests == nil)
		sysfatal("not enough memory");
	if(debug)
		fprint(2, "%s: %s: digest %s\n",
			argv0, s_to_c(m->path), sdigest);
	free(odigests);
	return 1;
fail:
	close(fd);
	fprint(2, "!message disappeared\n");
	md5((uchar*)"mail2fs", 7, digest, s);
	return 1;
}

void
upasfs(char* mbox)
{
	int kid, pid;

	rfork(RFNAMEG);
	unmount(nil, "/mail/fs");
	kid = fork();
	switch(kid){
	case -1:
		break;
	case 0:
		execl("/bin/upas/fs", "upas/fs", "-p", "-f", mbox, nil);
		sysfatal("no upas/fs: %r");
	default:
		while ((pid = waitpid()) != -1 && pid != kid)
			;
		break;
	}
}


char*
openmbox(char* mbox)
{
	char*	d;
	int	fd;

	mboxnm = strrchr(mbox, '/');
	if (mboxnm != nil)
		mboxnm++;
	else
		mboxnm = mbox;
	d = smprint("/mail/fs/%s", mboxnm);
	upasfs(mbox); /* run upas/fs on mbox file to produce a tree */

	fd = open("/mail/fs/ctl", OWRITE);
	if(fd < 0)
		sysfatal("/mail/fs/ctl: %r");
	if(fprint(fd, "open %s\n", mbox) < 0){
		close(fd);
		sysfatal("%s: open: %r", mbox);
	}
	close(fd);
	return d;
}

void
closembox(void)
{
	int	fd, some;
	char *s, *e;
	char	buf[1024];
	Message *l;

	fd = open("/mail/fs/ctl", OWRITE);
	if(fd < 0)
		sysfatal("/mail/fs/ctl: %r");
	e = buf + sizeof buf;
	s = seprint(buf, e, "delete %s", mboxnm);
	some = 0;
	for(l = top.child; l != nil; l = l->next)
		if(l->deleted){
			if(e - s < 20){
				if(write(fd, buf, s-buf) != s-buf)
					fprint(2, "%s: delete: %r\n", mboxnm);
				s = seprint(buf, e, "delete %s", mboxnm);
				some = 0;
			}
			some++;
			s = seprint(s, e, " %d", l->id);
		}
	if(some && write(fd, buf, s-buf) != s-buf)
		fprint(2, "%s: delete: %r\n", mboxnm);
	close(fd);
}

int
mbox2fs(char* mbox, char* mdir)
{
	int	n, sfd;
	char*	fname;
	Message *l;
	String *ddir, *s;

	if(chdir(mbox) < 0){
		fprint(2, "%s: %s: not a directory: %r\n", argv0, mbox);
		return -1;
	}
	top.path = s_copy(mbox);
	ddir = s_copy(mdir);
	s = file2string(ddir, "digest");
	digests = strdup(s_to_c(s));
	s_free(s);
	s_free(ddir);
	n = dir2message(&top, 0);
	if(n < 0) {
		fprint(2, "%s: dirmessage failed\n", argv0);
		return -1;
	}
	fname = smprint("%s/digest", mdir);
	sfd = open(fname, OWRITE);
	free(fname);
	if(sfd >= 0)
		seek(sfd, 0, 2);
	for(l = top.child; l != nil; l = l->next){
		if(debug)
			fprint(2, "converting %s/%d\n", mbox, l->id);
		if(sign(l))
			if(msg2fs(nil, l, mdir) < 0){
				fprint(2, "mbox2fs: msg2fs errors\n");
				return -1;
			} else {
				if(sfd >= 0)
					fprint(sfd, "%s\n", lastsigned());
			}
		if(!dry)
			l->deleted = 1;
	}
	if(sfd >= 0)
		close(sfd);
	return 0;
}

static void
usage(void)
{
	fprint(2, "usage: %s [-acDnor] [-d mdir] [mbox]\n", argv0);
	exits("usage");
}

void
main(int argc, char*argv[])
{
	char *mbox = "mbox";
	char *mdir, *top;
	Ctype *cp;

	top = smprint("/mail/box/%s", getuser());
	mdir = smprint("%s/msgs", top);
	ARGBEGIN{
	case 'a':
		aflag++;
		break;
	case 'c':
		cflag++;
		break;
	case 'd':
		free(mdir);
		mdir = EARGF(usage());
		mdir = cleanpath(mdir, top);
		break;
	case 'D':
		debug++;
		break;
	case 'n':
		dry++;
		break;
	case 'o':
		oflag++;
		break;
	case 'r':
		rflag++;
		break;
	default:
		usage();
	}ARGEND;
	switch(argc){
	case 1:
		mbox = cleanpath(argv[0], top);
		break;
	case 0:
		mbox = cleanpath(mbox, top);
		break;
	default:
		usage();
	}

	for(cp = ctype; cp < ctype + nelem(ctype) - 1; cp++)
		cp->next = cp + 1;
	mbox = openmbox(mbox);
	mboxfd = mkmdir(mdir);
	mbox2fs(mbox, mdir);
	closembox();
	closemdir(mboxfd);
	exits(nil);
}

Bell Labs OSI certified Powered by Plan 9

(Return to Plan 9 Home Page)

Copyright © 2021 Plan 9 Foundation. All Rights Reserved.
Comments to [email protected].