src/cmd/faces/plumb.c - plan9 - Git at Google

 #include <u.h>
 #include <libc.h>
 #include <draw.h>
 #include <plumb.h>
 #include <regexp.h>
 #include <bio.h>
 #include <9pclient.h>
 #include "faces.h"

 static int		showfd = -1;
 static int		seefd = -1;
 static int		logfd = -1;
 static char	*user;
 static char	*logtag;

 char		**maildirs;
 int		nmaildirs;

 void
 initplumb(void)
 {
 	showfd = plumbopen("send", OWRITE);
 	seefd = plumbopen("seemail", OREAD);

 	if(seefd < 0){
 		logfd = open(unsharp("#9/log/mail"), OREAD);
 		seek(logfd, 0LL, 2);
 		user = getenv("user");
 		if(user == nil){
 			fprint(2, "faces: can't find user name: %r\n");
 			exits("$user");
 		}
 		logtag = emalloc(32+strlen(user)+1);
 		sprint(logtag, " delivered %s From ", user);
 	}
 }

 void
 addmaildir(char *dir)
 {
 	maildirs = erealloc(maildirs, (nmaildirs+1)*sizeof(char*));
 	maildirs[nmaildirs++] = dir;
 }

 char*
 attr(Face *f)
 {
 	static char buf[128];

 	if(f->str[Sdigest]){
 		snprint(buf, sizeof buf, "digest=%s", f->str[Sdigest]);
 		return buf;
 	}
 	return nil;
 }

 void
 showmail(Face *f)
 {
 	Plumbmsg pm;
 	Plumbattr a;
 	char *s;

 	if(showfd<0 || f->str[Sshow]==nil || f->str[Sshow][0]=='\0')
 		return;
 	s = emalloc(strlen("/mail/fs")+1+strlen(f->str[Sshow]));
 	sprint(s,"/mail/fs/%s",f->str[Sshow]);
 	pm.src = "faces";
 	pm.dst = "showmail";
 	pm.wdir = "/mail/fs";
 	pm.type = "text";
 	a.name = "digest";
 	a.value = f->str[Sdigest];
 	a.next = nil;
 	pm.attr = &a;
 	pm.ndata = strlen(s);
 	pm.data = s;
 	plumbsend(showfd,&pm);
 }

 char*
 value(Plumbattr *attr, char *key, char *def)
 {
 	char *v;

 	v = plumblookup(attr, key);
 	if(v)
 		return v;
 	return def;
 }

 void
 setname(Face *f, char *sender)
 {
 	char *at, *bang;
 	char *p;

 	/* works with UTF-8, although it's written as ASCII */
 	for(p=sender; *p!='\0'; p++)
 		*p = tolower(*p);
 	f->str[Suser] = sender;
 	at = strchr(sender, '@');
 	if(at){
 		*at++ = '\0';
 		f->str[Sdomain] = estrdup(at);
 		return;
 	}
 	bang = strchr(sender, '!');
 	if(bang){
 		*bang++ = '\0';
 		f->str[Suser] = estrdup(bang);
 		f->str[Sdomain] = sender;
 		return;
 	}
 }

 int
 getc(void)
 {
 	static uchar buf[512];
 	static int nbuf = 0;
 	static int i = 0;

 	while(i == nbuf){
 		i = 0;
 		nbuf = read(logfd, buf, sizeof buf);
 		if(nbuf == 0){
 			sleep(15000);
 			continue;
 		}
 		if(nbuf < 0)
 			return -1;
 	}
 	return buf[i++];
 }

 char*
 getline(char *buf, int n)
 {
 	int i, c;

 	for(i=0; i<n-1; i++){
 		c = getc();
 		if(c <= 0)
 			return nil;
 		if(c == '\n')
 			break;
 		buf[i] = c;
 	}
 	buf[i] = '\0';
 	return buf;
 }

 static char* months[] = {
 	"jan", "feb", "mar", "apr",
 	"may", "jun", "jul", "aug",
 	"sep", "oct", "nov", "dec"
 };

 static int
 getmon(char *s)
 {
 	int i;

 	for(i=0; i<nelem(months); i++)
 		if(cistrcmp(months[i], s) == 0)
 			return i;
 	return -1;
 }

 /* Fri Jul 23 14:05:14 EDT 1999 */
 ulong
 parsedatev(char **a)
 {
 	char *p;
 	Tm tm;

 	memset(&tm, 0, sizeof tm);
 	if((tm.mon=getmon(a[1])) == -1)
 		goto Err;
 	tm.mday = strtol(a[2], &p, 10);
 	if(*p != '\0')
 		goto Err;
 	tm.hour = strtol(a[3], &p, 10);
 	if(*p != ':')
 		goto Err;
 	tm.min = strtol(p+1, &p, 10);
 	if(*p != ':')
 		goto Err;
 	tm.sec = strtol(p+1, &p, 10);
 	if(*p != '\0')
 		goto Err;
 	if(strlen(a[4]) != 3)
 		goto Err;
 	strcpy(tm.zone, a[4]);
 	if(strlen(a[5]) != 4)
 		goto Err;
 	tm.year = strtol(a[5], &p, 10);
 	if(*p != '\0')
 		goto Err;
 	tm.year -= 1900;
 	return tm2sec(&tm);
 Err:
 	return time(0);
 }

 ulong
 parsedate(char *s)
 {
 	char *f[10];
 	int nf;

 	nf = getfields(s, f, nelem(f), 1, " ");
 	if(nf < 6)
 		return time(0);
 	return parsedatev(f);
 }

 /* achille Jul 23 14:05:15 delivered jmk From ms.com!bub Fri Jul 23 14:05:14 EDT 1999 (plan9.bell-labs.com!jmk) 1352 */
 /* achille Oct 26 13:45:42 remote local!rsc From rsc Sat Oct 26 13:45:41 EDT 2002 (rsc) 170 */
 int
 parselog(char *s, char **sender, ulong *xtime)
 {
 	char *f[20];
 	int nf;

 	nf = getfields(s, f, nelem(f), 1, " ");
 	if(nf < 14)
 		return 0;
 	if(strcmp(f[4], "delivered") == 0 && strcmp(f[5], user) == 0)
 		goto Found;
 	if(strcmp(f[4], "remote") == 0 && strncmp(f[5], "local!", 6) == 0 && strcmp(f[5]+6, user) == 0)
 		goto Found;
 	return 0;

 Found:
 	*sender = estrdup(f[7]);
 	*xtime = parsedatev(&f[8]);
 	return 1;
 }

 int
 logrecv(char **sender, ulong *xtime)
 {
 	char buf[4096];

 	for(;;){
 		if(getline(buf, sizeof buf) == nil)
 			return 0;
 		if(parselog(buf, sender, xtime))
 			return 1;
 	}
 	return -1;
 }

 char*
 tweakdate(char *d)
 {
 	char e[8];

 	/* d, date = "Mon Aug  2 23:46:55 EDT 1999" */

 	if(strlen(d) < strlen("Mon Aug  2 23:46:55 EDT 1999"))
 		return estrdup("");
 	if(strncmp(date, d, 4+4+3) == 0)
 		snprint(e, sizeof e, "%.5s", d+4+4+3);	/* 23:46 */
 	else
 		snprint(e, sizeof e, "%.6s", d+4);	/* Aug  2 */
 	return estrdup(e);
 }

 Face*
 nextface(void)
 {
 	int i;
 	Face *f;
 	Plumbmsg *m;
 	char *t, *senderp, *showmailp, *digestp;
 	ulong xtime;

 	f = emalloc(sizeof(Face));
 	for(;;){
 		if(seefd >= 0){
 			m = plumbrecv(seefd);
 			if(m == nil)
 				killall("error on seemail plumb port");
 			t = value(m->attr, "mailtype", "");
 			if(strcmp(t, "delete") == 0)
 				delete(m->data, value(m->attr, "digest", nil));
 			else if(strcmp(t, "new") != 0)
 				fprint(2, "faces: unknown plumb message type %s\n", t);
 			else for(i=0; i<nmaildirs; i++) {
 				if(strncmp(m->data,"/mail/fs/",strlen("/mail/fs/")) == 0)
 					m->data += strlen("/mail/fs/");
 				if(strncmp(m->data, maildirs[i], strlen(maildirs[i])) == 0)
 					goto Found;
 			}
 			plumbfree(m);
 			continue;

 		Found:
 			xtime = parsedate(value(m->attr, "date", date));
 			digestp = value(m->attr, "digest", nil);
 			if(alreadyseen(digestp)){
 				/* duplicate upas/fs can send duplicate messages */
 				plumbfree(m);
 				continue;
 			}
 			senderp = estrdup(value(m->attr, "sender", "???"));
 			showmailp = estrdup(m->data);
 			if(digestp)
 				digestp = estrdup(digestp);
 			plumbfree(m);
 		}else{
 			if(logrecv(&senderp, &xtime) <= 0)
 				killall("error reading log file");
 			showmailp = estrdup("");
 			digestp = nil;
 		}
 		setname(f, senderp);
 		f->time = xtime;
 		f->tm = *localtime(xtime);
 		f->str[Sshow] = showmailp;
 		f->str[Sdigest] = digestp;
 		return f;
 	}
 	return nil;
 }

 char*
 iline(char *data, char **pp)
 {
 	char *p;

 	for(p=data; *p!='\0' && *p!='\n'; p++)
 		;
 	if(*p == '\n')
 		*p++ = '\0';
 	*pp = p;
 	return data;
 }

 Face*
 dirface(char *dir, char *num)
 {
 	Face *f;
 	char *from, *date;
 	char buf[1024],  *info, *p, *digest;
 	int n;
 	ulong len;
 	CFid *fid;

 #if 0
 	/*
 	 * loadmbox leaves us in maildir, so we needn't
 	 * walk /mail/fs/mbox for each face; this makes startup
 	 * a fair bit quicker.
 	 */
 	if(getwd(pwd, sizeof pwd) != nil && strcmp(pwd, dir) == 0)
 		sprint(buf, "%s/info", num);
 	else
 		sprint(buf, "%s/%s/info", dir, num);
 #endif
 	sprint(buf, "%s/%s/info", dir, num);
 	len = fsdirlen(upasfs, buf);
 	if(len <= 0)
 		return nil;
 	fid = fsopen(upasfs,buf, OREAD);
 	if(fid == nil)
 		return nil;
 	info = emalloc(len+1);
 	n = fsreadn(fid, info, len);
 	fsclose(fid);
 	if(n < 0){
 		free(info);
 		return nil;
 	}
 	info[n] = '\0';
 	f = emalloc(sizeof(Face));
 	from = iline(info, &p);	/* from */
 	iline(p, &p);	/* to */
 	iline(p, &p);	/* cc */
 	iline(p, &p);	/* replyto */
 	date = iline(p, &p);	/* date */
 	setname(f, estrdup(from));
 	f->time = parsedate(date);
 	f->tm = *localtime(f->time);
 	sprint(buf, "%s/%s", dir, num);
 	f->str[Sshow] = estrdup(buf);
 	iline(p, &p);	/* subject */
 	iline(p, &p);	/* mime content type */
 	iline(p, &p);	/* mime disposition */
 	iline(p, &p);	/* filename */
 	digest = iline(p, &p);	/* digest */
 	f->str[Sdigest] = estrdup(digest);
 	free(info);
 	return f;
 }
	#include <u.h>
	#include <libc.h>
	#include <draw.h>
	#include <plumb.h>
	#include <regexp.h>
	#include <bio.h>
	#include <9pclient.h>
	#include "faces.h"

	static int showfd = -1;
	static int seefd = -1;
	static int logfd = -1;
	static char *user;
	static char *logtag;

	char **maildirs;
	int nmaildirs;

	void
	initplumb(void)
	{
	showfd = plumbopen("send", OWRITE);
	seefd = plumbopen("seemail", OREAD);

	if(seefd < 0){
	logfd = open(unsharp("#9/log/mail"), OREAD);
	seek(logfd, 0LL, 2);
	user = getenv("user");
	if(user == nil){
	fprint(2, "faces: can't find user name: %r\n");
	exits("$user");
	}
	logtag = emalloc(32+strlen(user)+1);
	sprint(logtag, " delivered %s From ", user);
	}
	}

	void
	addmaildir(char *dir)
	{
	maildirs = erealloc(maildirs, (nmaildirs+1)sizeof(char));
	maildirs[nmaildirs++] = dir;
	}

	char*
	attr(Face *f)
	{
	static char buf[128];

	if(f->str[Sdigest]){
	snprint(buf, sizeof buf, "digest=%s", f->str[Sdigest]);
	return buf;
	}
	return nil;
	}

	void
	showmail(Face *f)
	{
	Plumbmsg pm;
	Plumbattr a;
	char *s;

	if(showfd<0 \|\| f->str[Sshow]==nil \|\| f->str[Sshow][0]=='\0')
	return;
	s = emalloc(strlen("/mail/fs")+1+strlen(f->str[Sshow]));
	sprint(s,"/mail/fs/%s",f->str[Sshow]);
	pm.src = "faces";
	pm.dst = "showmail";
	pm.wdir = "/mail/fs";
	pm.type = "text";
	a.name = "digest";
	a.value = f->str[Sdigest];
	a.next = nil;
	pm.attr = &a;
	pm.ndata = strlen(s);
	pm.data = s;
	plumbsend(showfd,&pm);
	}

	char*
	value(Plumbattr attr, char key, char *def)
	{
	char *v;

	v = plumblookup(attr, key);
	if(v)
	return v;
	return def;
	}

	void
	setname(Face f, char sender)
	{
	char at, bang;
	char *p;

	/* works with UTF-8, although it's written as ASCII */
	for(p=sender; *p!='\0'; p++)
	p = tolower(p);
	f->str[Suser] = sender;
	at = strchr(sender, '@');
	if(at){
	*at++ = '\0';
	f->str[Sdomain] = estrdup(at);
	return;
	}
	bang = strchr(sender, '!');
	if(bang){
	*bang++ = '\0';
	f->str[Suser] = estrdup(bang);
	f->str[Sdomain] = sender;
	return;
	}
	}

	int
	getc(void)
	{
	static uchar buf[512];
	static int nbuf = 0;
	static int i = 0;

	while(i == nbuf){
	i = 0;
	nbuf = read(logfd, buf, sizeof buf);
	if(nbuf == 0){
	sleep(15000);
	continue;
	}
	if(nbuf < 0)
	return -1;
	}
	return buf[i++];
	}

	char*
	getline(char *buf, int n)
	{
	int i, c;

	for(i=0; i<n-1; i++){
	c = getc();
	if(c <= 0)
	return nil;
	if(c == '\n')
	break;
	buf[i] = c;
	}
	buf[i] = '\0';
	return buf;
	}

	static char* months[] = {
	"jan", "feb", "mar", "apr",
	"may", "jun", "jul", "aug",
	"sep", "oct", "nov", "dec"
	};

	static int
	getmon(char *s)
	{
	int i;

	for(i=0; i<nelem(months); i++)
	if(cistrcmp(months[i], s) == 0)
	return i;
	return -1;
	}

	/* Fri Jul 23 14:05:14 EDT 1999 */
	ulong
	parsedatev(char **a)
	{
	char *p;
	Tm tm;

	memset(&tm, 0, sizeof tm);
	if((tm.mon=getmon(a[1])) == -1)
	goto Err;
	tm.mday = strtol(a[2], &p, 10);
	if(*p != '\0')
	goto Err;
	tm.hour = strtol(a[3], &p, 10);
	if(*p != ':')
	goto Err;
	tm.min = strtol(p+1, &p, 10);
	if(*p != ':')
	goto Err;
	tm.sec = strtol(p+1, &p, 10);
	if(*p != '\0')
	goto Err;
	if(strlen(a[4]) != 3)
	goto Err;
	strcpy(tm.zone, a[4]);
	if(strlen(a[5]) != 4)
	goto Err;
	tm.year = strtol(a[5], &p, 10);
	if(*p != '\0')
	goto Err;
	tm.year -= 1900;
	return tm2sec(&tm);
	Err:
	return time(0);
	}

	ulong
	parsedate(char *s)
	{
	char *f[10];
	int nf;

	nf = getfields(s, f, nelem(f), 1, " ");
	if(nf < 6)
	return time(0);
	return parsedatev(f);
	}

	/* achille Jul 23 14:05:15 delivered jmk From ms.com!bub Fri Jul 23 14:05:14 EDT 1999 (plan9.bell-labs.com!jmk) 1352 */
	/* achille Oct 26 13:45:42 remote local!rsc From rsc Sat Oct 26 13:45:41 EDT 2002 (rsc) 170 */
	int
	parselog(char s, char sender, ulong xtime)
	{
	char *f[20];
	int nf;

	nf = getfields(s, f, nelem(f), 1, " ");
	if(nf < 14)
	return 0;
	if(strcmp(f[4], "delivered") == 0 && strcmp(f[5], user) == 0)
	goto Found;
	if(strcmp(f[4], "remote") == 0 && strncmp(f[5], "local!", 6) == 0 && strcmp(f[5]+6, user) == 0)
	goto Found;
	return 0;

	Found:
	*sender = estrdup(f[7]);
	*xtime = parsedatev(&f[8]);
	return 1;
	}

	int
	logrecv(char *sender, ulong xtime)
	{
	char buf[4096];

	for(;;){
	if(getline(buf, sizeof buf) == nil)
	return 0;
	if(parselog(buf, sender, xtime))
	return 1;
	}
	return -1;
	}

	char*
	tweakdate(char *d)
	{
	char e[8];

	/* d, date = "Mon Aug 2 23:46:55 EDT 1999" */

	if(strlen(d) < strlen("Mon Aug 2 23:46:55 EDT 1999"))
	return estrdup("");
	if(strncmp(date, d, 4+4+3) == 0)
	snprint(e, sizeof e, "%.5s", d+4+4+3); /* 23:46 */
	else
	snprint(e, sizeof e, "%.6s", d+4); /* Aug 2 */
	return estrdup(e);
	}

	Face*
	nextface(void)
	{
	int i;
	Face *f;
	Plumbmsg *m;
	char t, senderp, showmailp, digestp;
	ulong xtime;

	f = emalloc(sizeof(Face));
	for(;;){
	if(seefd >= 0){
	m = plumbrecv(seefd);
	if(m == nil)
	killall("error on seemail plumb port");
	t = value(m->attr, "mailtype", "");
	if(strcmp(t, "delete") == 0)
	delete(m->data, value(m->attr, "digest", nil));
	else if(strcmp(t, "new") != 0)
	fprint(2, "faces: unknown plumb message type %s\n", t);
	else for(i=0; i<nmaildirs; i++) {
	if(strncmp(m->data,"/mail/fs/",strlen("/mail/fs/")) == 0)
	m->data += strlen("/mail/fs/");
	if(strncmp(m->data, maildirs[i], strlen(maildirs[i])) == 0)
	goto Found;
	}
	plumbfree(m);
	continue;

	Found:
	xtime = parsedate(value(m->attr, "date", date));
	digestp = value(m->attr, "digest", nil);
	if(alreadyseen(digestp)){
	/* duplicate upas/fs can send duplicate messages */
	plumbfree(m);
	continue;
	}
	senderp = estrdup(value(m->attr, "sender", "???"));
	showmailp = estrdup(m->data);
	if(digestp)
	digestp = estrdup(digestp);
	plumbfree(m);
	}else{
	if(logrecv(&senderp, &xtime) <= 0)
	killall("error reading log file");
	showmailp = estrdup("");
	digestp = nil;
	}
	setname(f, senderp);
	f->time = xtime;
	f->tm = *localtime(xtime);
	f->str[Sshow] = showmailp;
	f->str[Sdigest] = digestp;
	return f;
	}
	return nil;
	}

	char*
	iline(char data, char *pp)
	{
	char *p;

	for(p=data; p!='\0' && p!='\n'; p++)
	;
	if(*p == '\n')
	*p++ = '\0';
	*pp = p;
	return data;
	}

	Face*
	dirface(char dir, char num)
	{
	Face *f;
	char from, date;
	char buf[1024], info, p, *digest;
	int n;
	ulong len;
	CFid *fid;

	#if 0
	/*
	* loadmbox leaves us in maildir, so we needn't
	* walk /mail/fs/mbox for each face; this makes startup
	* a fair bit quicker.
	*/
	if(getwd(pwd, sizeof pwd) != nil && strcmp(pwd, dir) == 0)
	sprint(buf, "%s/info", num);
	else
	sprint(buf, "%s/%s/info", dir, num);
	#endif
	sprint(buf, "%s/%s/info", dir, num);
	len = fsdirlen(upasfs, buf);
	if(len <= 0)
	return nil;
	fid = fsopen(upasfs,buf, OREAD);
	if(fid == nil)
	return nil;
	info = emalloc(len+1);
	n = fsreadn(fid, info, len);
	fsclose(fid);
	if(n < 0){
	free(info);
	return nil;
	}
	info[n] = '\0';
	f = emalloc(sizeof(Face));
	from = iline(info, &p); /* from */
	iline(p, &p); /* to */
	iline(p, &p); /* cc */
	iline(p, &p); /* replyto */
	date = iline(p, &p); /* date */
	setname(f, estrdup(from));
	f->time = parsedate(date);
	f->tm = *localtime(f->time);
	sprint(buf, "%s/%s", dir, num);
	f->str[Sshow] = estrdup(buf);
	iline(p, &p); /* subject */
	iline(p, &p); /* mime content type */
	iline(p, &p); /* mime disposition */
	iline(p, &p); /* filename */
	digest = iline(p, &p); /* digest */
	f->str[Sdigest] = estrdup(digest);
	free(info);
	return f;
	}