1 #include <u.h>
2 #include <libc.h>
3 #include <bio.h>
4 #include <thread.h>
5 #include <9pclient.h>
6 #include <plumb.h>
7 #include <ctype.h>
8 #include "dat.h"
9 
10 char	*maildir = "Mail/";			/* mountpoint of mail file system */
11 char *mboxname = "mbox";			/* mailboxdir/mboxname is mail spool file */
12 char	*mailboxdir = nil;				/* nil == /mail/box/$user */
13 char *fsname;						/* filesystem for mailboxdir/mboxname is at maildir/fsname */
14 char	*user;
15 char	*outgoing;
16 char *srvname;
17 
18 Window	*wbox;
19 Message	mbox;
20 Message	replies;
21 char		*home;
22 CFid		*plumbsendfd;
23 CFid		*plumbseemailfd;
24 CFid		*plumbshowmailfd;
25 CFid		*plumbsendmailfd;
26 Channel	*cplumb;
27 Channel	*cplumbshow;
28 Channel	*cplumbsend;
29 int		wctlfd;
30 void		mainctl(void*);
31 void		plumbproc(void*);
32 void		plumbshowproc(void*);
33 void		plumbsendproc(void*);
34 void		plumbthread(void);
35 void		plumbshowthread(void*);
36 void		plumbsendthread(void*);
37 
38 int			shortmenu;
39 
40 CFsys *mailfs;
41 CFsys *acmefs;
42 
43 void
44 usage(void)
45 {
46 	fprint(2, "usage: Mail [-sS] [-n srvname] [-o outgoing] [mailboxname [directoryname]]\n");
47 	threadexitsall("usage");
48 }
49 
50 void
51 removeupasfs(void)
52 {
53 	char buf[256];
54 
55 	if(strcmp(mboxname, "mbox") == 0)
56 		return;
57 	snprint(buf, sizeof buf, "close %s", mboxname);
58 	fswrite(mbox.ctlfd, buf, strlen(buf));
59 }
60 
61 int
62 ismaildir(char *s)
63 {
64 	Dir *d;
65 	int ret;
66 
67 	d = fsdirstat(mailfs, s);
68 	if(d == nil)
69 		return 0;
70 	ret = d->qid.type & QTDIR;
71 	free(d);
72 	return ret;
73 }
74 
75 void
76 threadmain(int argc, char *argv[])
77 {
78 	char *s, *name;
79 	char err[ERRMAX], *cmd;
80 	int i, newdir;
81 	Fmt fmt;
82 
83 	doquote = needsrcquote;
84 	quotefmtinstall();
85 
86 	/* open these early so we won't miss notification of new mail messages while we read mbox */
87 	if((plumbsendfd = plumbopenfid("send", OWRITE|OCEXEC)) == nil)
88 		fprint(2, "warning: open plumb/send: %r\n");
89 	if((plumbseemailfd = plumbopenfid("seemail", OREAD|OCEXEC)) == nil)
90 		fprint(2, "warning: open plumb/seemail: %r\n");
91 	if((plumbshowmailfd = plumbopenfid("showmail", OREAD|OCEXEC)) == nil)
92 		fprint(2, "warning: open plumb/showmail: %r\n");
93 
94 	shortmenu = 0;
95 	srvname = "mail";
96 	ARGBEGIN{
97 	case 's':
98 		shortmenu = 1;
99 		break;
100 	case 'S':
101 		shortmenu = 2;
102 		break;
103 	case 'o':
104 		outgoing = EARGF(usage());
105 		break;
106 	case 'm':
107 		smprint(maildir, "%s/", EARGF(usage()));
108 		break;
109 	case 'n':
110 		srvname = EARGF(usage());
111 		break;
112 	default:
113 		usage();
114 	}ARGEND
115 
116 	acmefs = nsmount("acme",nil);
117 	if(acmefs == nil)
118 		error("cannot mount acme: %r");
119 	mailfs = nsmount(srvname, nil);
120 	if(mailfs == nil)
121 		error("cannot mount %s: %r", srvname);
122 
123 	name = "mbox";
124 
125 	newdir = 1;
126 	if(argc > 0){
127 		i = strlen(argv[0]);
128 		if(argc>2 || i==0)
129 			usage();
130 		/* see if the name is that of an existing /mail/fs directory */
131 		if(argc==1 && argv[0][0] != '/' && ismaildir(argv[0])){
132 			name = argv[0];
133 			mboxname = estrdup(name);
134 			newdir = 0;
135 		}else{
136 			if(argv[0][i-1] == '/')
137 				argv[0][i-1] = '\0';
138 			s = strrchr(argv[0], '/');
139 			if(s == nil)
140 				mboxname = estrdup(argv[0]);
141 			else{
142 				*s++ = '\0';
143 				if(*s == '\0')
144 					usage();
145 				mailboxdir = argv[0];
146 				mboxname = estrdup(s);
147 			}
148 			if(argc > 1)
149 				name = argv[1];
150 			else
151 				name = mboxname;
152 		}
153 	}
154 
155 	user = getenv("user");
156 	if(user == nil)
157 		user = "none";
158 	home = getenv("home");
159 	if(home == nil)
160 		home = getenv("HOME");
161 	if(home == nil)
162 		error("can't find $home");
163 	if(mailboxdir == nil)
164 		mailboxdir = estrstrdup(home, "/mail");
165 	if(outgoing == nil)
166 		outgoing = estrstrdup(mailboxdir, "/outgoing");
167 
168 	mbox.ctlfd = fsopen(mailfs, estrstrdup(mboxname, "/ctl"), OWRITE);
169 	if(mbox.ctlfd == nil)
170 		error("can't open %s: %r", estrstrdup(mboxname, "/ctl"));
171 
172 	fsname = estrdup(name);
173 	if(newdir && argc > 0){
174 		s = emalloc(5+strlen(mailboxdir)+strlen(mboxname)+strlen(name)+10+1);
175 		for(i=0; i<10; i++){
176 			sprint(s, "open %s/%s %s", mailboxdir, mboxname, fsname);
177 			if(fswrite(mbox.ctlfd, s, strlen(s)) >= 0)
178 				break;
179 			err[0] = '\0';
180 			errstr(err, sizeof err);
181 			if(strstr(err, "mbox name in use") == nil)
182 				error("can't create directory %s for mail: %s", name, err);
183 			free(fsname);
184 			fsname = emalloc(strlen(name)+10);
185 			sprint(fsname, "%s-%d", name, i);
186 		}
187 		if(i == 10)
188 			error("can't open %s/%s: %r", mailboxdir, mboxname);
189 		free(s);
190 	}
191 
192 	s = estrstrdup(fsname, "/");
193 	mbox.name = estrstrdup(maildir, s);
194 	mbox.level= 0;
195 	readmbox(&mbox, maildir, s);
196 	home = getenv("home");
197 	if(home == nil)
198 		home = "/";
199 
200 	wbox = newwindow();
201 	winname(wbox, mbox.name);
202 	wintagwrite(wbox, "Put Mail Delmesg ", 3+1+4+1+7+1);
203 	threadcreate(mainctl, wbox, STACK);
204 
205 	fmtstrinit(&fmt);
206 	fmtprint(&fmt, "Mail");
207 	if(shortmenu)
208 		fmtprint(&fmt, " -%c", "sS"[shortmenu-1]);
209 	if(outgoing)
210 		fmtprint(&fmt, " -o %s", outgoing);
211 	fmtprint(&fmt, " %s", name);
212 	cmd = fmtstrflush(&fmt);
213 	if(cmd == nil)
214 		sysfatal("out of memory");
215 	winsetdump(wbox, "/acme/mail", cmd);
216 	mbox.w = wbox;
217 
218 	mesgmenu(wbox, &mbox);
219 	winclean(wbox);
220 
221 /*	wctlfd = open("/dev/wctl", OWRITE|OCEXEC);	/* for acme window */
222 	wctlfd = -1;
223 	cplumb = chancreate(sizeof(Plumbmsg*), 0);
224 	cplumbshow = chancreate(sizeof(Plumbmsg*), 0);
225 	if(strcmp(name, "mbox") == 0){
226 		/*
227 		 * Avoid creating multiple windows to send mail by only accepting
228 		 * sendmail plumb messages if we're reading the main mailbox.
229 		 */
230 		plumbsendmailfd = plumbopenfid("sendmail", OREAD|OCEXEC);
231 		cplumbsend = chancreate(sizeof(Plumbmsg*), 0);
232 		proccreate(plumbsendproc, nil, STACK);
233 		threadcreate(plumbsendthread, nil, STACK);
234 	}
235 	/* start plumb reader as separate proc ... */
236 	proccreate(plumbproc, nil, STACK);
237 	proccreate(plumbshowproc, nil, STACK);
238 	threadcreate(plumbshowthread, nil, STACK);
239 	fswrite(mbox.ctlfd, "refresh", 7);
240 	/* ... and use this thread to read the messages */
241 	plumbthread();
242 }
243 
244 void
245 plumbproc(void* v)
246 {
247 	Plumbmsg *m;
248 
249 	threadsetname("plumbproc");
250 	for(;;){
251 		m = plumbrecvfid(plumbseemailfd);
252 		sendp(cplumb, m);
253 		if(m == nil)
254 			threadexits(nil);
255 	}
256 }
257 
258 void
259 plumbshowproc(void* v)
260 {
261 	Plumbmsg *m;
262 
263 	threadsetname("plumbshowproc");
264 	for(;;){
265 		m = plumbrecvfid(plumbshowmailfd);
266 		sendp(cplumbshow, m);
267 		if(m == nil)
268 			threadexits(nil);
269 	}
270 }
271 
272 void
273 plumbsendproc(void* v)
274 {
275 	Plumbmsg *m;
276 
277 	threadsetname("plumbsendproc");
278 	for(;;){
279 		m = plumbrecvfid(plumbsendmailfd);
280 		sendp(cplumbsend, m);
281 		if(m == nil)
282 			threadexits(nil);
283 	}
284 }
285 
286 void
287 newmesg(char *name, char *digest)
288 {
289 	Dir *d;
290 
291 	if(strncmp(name, mbox.name, strlen(mbox.name)) != 0)
292 		return;	/* message is about another mailbox */
293 	if(mesglookupfile(&mbox, name, digest) != nil)
294 		return;
295 	if(strncmp(name, "Mail/", 5) == 0)
296 		name += 5;
297 	d = fsdirstat(mailfs, name);
298 	if(d == nil)
299 		return;
300 	if(mesgadd(&mbox, mbox.name, d, digest))
301 		mesgmenunew(wbox, &mbox);
302 	free(d);
303 }
304 
305 void
306 showmesg(char *name, char *digest)
307 {
308 	char *n;
309 	char *mb;
310 
311 	mb = mbox.name;
312 	if(strncmp(name, mb, strlen(mb)) != 0)
313 		return;	/* message is about another mailbox */
314 	n = estrdup(name+strlen(mb));
315 	if(n[strlen(n)-1] != '/')
316 		n = egrow(n, "/", nil);
317 	mesgopen(&mbox, mbox.name, name+strlen(mb), nil, 1, digest);
318 	free(n);
319 }
320 
321 void
322 delmesg(char *name, char *digest, int dodel, char *save)
323 {
324 	Message *m;
325 
326 	m = mesglookupfile(&mbox, name, digest);
327 	if(m != nil){
328 		if(save)
329 			mesgcommand(m, estrstrdup("Save ", save));
330 		if(dodel)
331 			mesgmenumarkdel(wbox, &mbox, m, 1);
332 		else{
333 			/* notification came from plumber - message is gone */
334 			mesgmenudel(wbox, &mbox, m);
335 			if(!m->opened)
336 				mesgdel(&mbox, m);
337 		}
338 	}
339 }
340 
341 void
342 plumbthread(void)
343 {
344 	Plumbmsg *m;
345 	Plumbattr *a;
346 	char *type, *digest;
347 
348 	threadsetname("plumbthread");
349 	while((m = recvp(cplumb)) != nil){
350 		a = m->attr;
351 		digest = plumblookup(a, "digest");
352 		type = plumblookup(a, "mailtype");
353 		if(type == nil)
354 			fprint(2, "Mail: plumb message with no mailtype attribute\n");
355 		else if(strcmp(type, "new") == 0)
356 			newmesg(m->data, digest);
357 		else if(strcmp(type, "delete") == 0)
358 			delmesg(m->data, digest, 0, nil);
359 		else
360 			fprint(2, "Mail: unknown plumb attribute %s\n", type);
361 		plumbfree(m);
362 	}
363 	threadexits(nil);
364 }
365 
366 void
367 plumbshowthread(void *v)
368 {
369 	Plumbmsg *m;
370 
371 	USED(v);
372 	threadsetname("plumbshowthread");
373 	while((m = recvp(cplumbshow)) != nil){
374 		showmesg(m->data, plumblookup(m->attr, "digest"));
375 		plumbfree(m);
376 	}
377 	threadexits(nil);
378 }
379 
380 void
381 plumbsendthread(void *v)
382 {
383 	Plumbmsg *m;
384 
385 	USED(v);
386 	threadsetname("plumbsendthread");
387 	while((m = recvp(cplumbsend)) != nil){
388 		mkreply(nil, "Mail", m->data, m->attr, nil);
389 		plumbfree(m);
390 	}
391 	threadexits(nil);
392 }
393 
394 int
395 mboxcommand(Window *w, char *s)
396 {
397 	char *args[10], **targs, *save;
398 	Message *m, *next;
399 	int ok, nargs, i, j;
400 	char buf[128];
401 
402 	nargs = tokenize(s, args, nelem(args));
403 	if(nargs == 0)
404 		return 0;
405 	if(strcmp(args[0], "Mail") == 0){
406 		if(nargs == 1)
407 			mkreply(nil, "Mail", "", nil, nil);
408 		else
409 			mkreply(nil, "Mail", args[1], nil, nil);
410 		return 1;
411 	}
412 	if(strcmp(s, "Del") == 0){
413 		if(mbox.dirty){
414 			mbox.dirty = 0;
415 			fprint(2, "mail: mailbox not written\n");
416 			return 1;
417 		}
418 		ok = 1;
419 		for(m=mbox.head; m!=nil; m=next){
420 			next = m->next;
421 			if(m->w){
422 				if(windel(m->w, 0))
423 					m->w = nil;
424 				else
425 					ok = 0;
426 			}
427 		}
428 		for(m=replies.head; m!=nil; m=next){
429 			next = m->next;
430 			if(m->w){
431 				if(windel(m->w, 0))
432 					m->w = nil;
433 				else
434 					ok = 0;
435 			}
436 		}
437 		if(ok){
438 			windel(w, 1);
439 			removeupasfs();
440 			threadexitsall(nil);
441 		}
442 		return 1;
443 	}
444 	if(strcmp(s, "Put") == 0){
445 		rewritembox(wbox, &mbox);
446 		return 1;
447 	}
448 	if(strcmp(s, "Get") == 0){
449 		fswrite(mbox.ctlfd, "refresh", 7);
450 		return 1;
451 	}
452 	if(strcmp(s, "Delmesg") == 0){
453 		save = nil;
454 		if(nargs > 1)
455 			save = args[1];
456 		s = winselection(w);
457 		if(s == nil)
458 			return 1;
459 		nargs = 1;
460 		for(i=0; s[i]; i++)
461 			if(s[i] == '\n')
462 				nargs++;
463 		targs = emalloc(nargs*sizeof(char*));	/* could be too many for a local array */
464 		nargs = getfields(s, targs, nargs, 1, "\n");
465 		for(i=0; i<nargs; i++){
466 			if(!isdigit(targs[i][0]))
467 				continue;
468 			j = atoi(targs[i]);	/* easy way to parse the number! */
469 			if(j == 0)
470 				continue;
471 			snprint(buf, sizeof buf, "%s%d", mbox.name, j);
472 			delmesg(buf, nil, 1, save);
473 		}
474 		free(s);
475 		free(targs);
476 		return 1;
477 	}
478 	return 0;
479 }
480 
481 void
482 mainctl(void *v)
483 {
484 	Window *w;
485 	Event *e, *e2, *eq, *ea;
486 	int na, nopen;
487 	char *s, *t, *buf;
488 
489 	w = v;
490 	winincref(w);
491 	proccreate(wineventproc, w, STACK);
492 
493 	for(;;){
494 		e = recvp(w->cevent);
495 		switch(e->c1){
496 		default:
497 		Unknown:
498 			print("unknown message %c%c\n", e->c1, e->c2);
499 			break;
500 	
501 		case 'E':	/* write to body; can't affect us */
502 			break;
503 	
504 		case 'F':	/* generated by our actions; ignore */
505 			break;
506 	
507 		case 'K':	/* type away; we don't care */
508 			break;
509 	
510 		case 'M':
511 			switch(e->c2){
512 			case 'x':
513 			case 'X':
514 				ea = nil;
515 				e2 = nil;
516 				if(e->flag & 2)
517 					e2 = recvp(w->cevent);
518 				if(e->flag & 8){
519 					ea = recvp(w->cevent);
520 					na = ea->nb;
521 					recvp(w->cevent);
522 				}else
523 					na = 0;
524 				s = e->b;
525 				/* if it's a known command, do it */
526 				if((e->flag&2) && e->nb==0)
527 					s = e2->b;
528 				if(na){
529 					t = emalloc(strlen(s)+1+na+1);
530 					sprint(t, "%s %s", s, ea->b);
531 					s = t;
532 				}
533 				/* if it's a long message, it can't be for us anyway */
534 				if(!mboxcommand(w, s))	/* send it back */
535 					winwriteevent(w, e);
536 				if(na)
537 					free(s);
538 				break;
539 	
540 			case 'l':
541 			case 'L':
542 				buf = nil;
543 				eq = e;
544 				if(e->flag & 2){
545 					e2 = recvp(w->cevent);
546 					eq = e2;
547 				}
548 				s = eq->b;
549 				if(eq->q1>eq->q0 && eq->nb==0){
550 					buf = emalloc((eq->q1-eq->q0)*UTFmax+1);
551 					winread(w, eq->q0, eq->q1, buf);
552 					s = buf;
553 				}
554 				nopen = 0;
555 				do{
556 					/* skip 'deleted' string if present' */
557 					if(strncmp(s, deleted, strlen(deleted)) == 0)
558 						s += strlen(deleted);
559 					/* skip mail box name if present */
560 					if(strncmp(s, mbox.name, strlen(mbox.name)) == 0)
561 						s += strlen(mbox.name);
562 					nopen += mesgopen(&mbox, mbox.name, s, nil, 0, nil);
563 					while(*s!='\0' && *s++!='\n')
564 						;
565 				}while(*s);
566 				if(nopen == 0)	/* send it back */
567 					winwriteevent(w, e);
568 				free(buf);
569 				break;
570 	
571 			case 'I':	/* modify away; we don't care */
572 			case 'D':
573 			case 'd':
574 			case 'i':
575 				break;
576 	
577 			default:
578 				goto Unknown;
579 			}
580 		}
581 	}
582 }
583