[unix-history] / usr / src / usr.bin / mail / quit.c

#

#include "rcv.h"
#include <sys/stat.h>

/*
 * Rcv -- receive mail rationally.
 *
 * Termination processing.
 */

static char *SccsId = "@(#)quit.c	1.4 %G%";

/*
 * Save all of the undetermined messages at the top of "mbox"
 * Save all untouched messages back in the system mailbox.
 * Remove the system mailbox, if none saved there.
 */

quit()
{
	int mcount, p, modify, autohold, anystat;
	FILE *ibuf, *obuf, *fbuf, *rbuf;
	register struct message *mp;
	register int c;
	extern char tempQuit[], tempResid[];
	struct stat minfo;

	/*
	 * If we are read only, we can't do anything,
	 * so just return quickly.
	 */

	if (readonly)
		return;
	/*
	 * See if there any messages to save in mbox.  If no, we
	 * can save copying mbox to /tmp and back.
	 *
	 * Check also to see if any files need to be preserved.
	 * Delete all untouched messages to keep them out of mbox.
	 * If all the messages are to be preserved, just exit with
	 * a message.
	 *
	 * If the luser has sent mail to himself, refuse to do
	 * anything with the mailbox, unless mail locking works.
	 */

	lock(mailname);
#ifndef CANLOCK
	if (selfsent) {
		printf("You have new mail.\n");
		unlock();
		return;
	}
#endif
	rbuf = NULL;
	if (stat(mailname, &minfo) >= 0 && minfo.st_size > mailsize) {
		printf("New mail has arrived.\n");
		rbuf = fopen(tempResid, "w");
		fbuf = fopen(mailname, "r");
		if (rbuf == NULL || fbuf == NULL)
			goto newmail;
#ifdef APPEND
		fseek(fbuf, mailsize, 0);
		while ((c = getc(fbuf)) != EOF)
			putc(c, rbuf);
#else
		p = minfo.st_size - mailsize;
		while (p-- > 0) {
			c = getc(fbuf);
			if (c == EOF)
				goto newmail;
			putc(c, rbuf);
		}
#endif
		fclose(fbuf);
		fclose(rbuf);
		if ((rbuf = fopen(tempResid, "r")) == NULL)
			goto newmail;
		remove(tempResid);
	}

	/*
	 * Adjust the message flags in each message.
	 */

	anystat = 0;
	for (mp = &message[0]; mp < &message[msgCount]; mp++) {
		if (mp->m_flag & MNEW) {
			mp->m_flag &= ~MNEW;
			mp->m_flag |= MSTATUS;
		}
		if (mp->m_flag & MSTATUS)
			anystat++;
		if (mp->m_flag & MDELETED)
			mp->m_flag = MDELETED|MTOUCH;
		if ((mp->m_flag & MTOUCH) == 0)
			mp->m_flag |= MDELETED;
	}
	autohold = value("hold") != NOSTR;
	modify = 0;
	for (c = 0, p = 0, mp = &message[0]; mp < &message[msgCount]; mp++) {
		if ((mp->m_flag & (MSAVED|MDELETED)) == 0 && autohold)
			mp->m_flag |= MPRESERVE;
		if ((mp->m_flag & (MSAVED|MDELETED|MPRESERVE)) == 0)
			c++;
		if ((mp->m_flag & MPRESERVE) || (mp->m_flag & MTOUCH) == 0)
			p++;
		if (mp->m_flag & MODIFY)
			modify++;
	}
	if (p == msgCount && !modify && !anystat) {
		if (p == 1)
			printf("Held 1 message in %s\n", mailname);
		else
			printf("Held %2d messages in %s\n", p, mailname);
		unlock();
		return;
	}
	if (c == 0) {
		if (p != 0) {
			writeback(rbuf);
			unlock();
			return;
		}
		goto cream;
	}

	/*
	 * Create another temporary file and copy user's mbox file
	 * darin.  If there is no mbox, copy nothing.
	 * If he has specified "append" don't copy his mailbox,
	 * just copy saveable entries at the end.
	 */

	mcount = c;
	if (value("append") == NOSTR) {
		if ((obuf = fopen(tempQuit, "w")) == NULL) {
			perror(tempQuit);
			unlock();
			return;
		}
		if ((ibuf = fopen(tempQuit, "r")) == NULL) {
			perror(tempQuit);
			remove(tempQuit);
			fclose(obuf);
			unlock();
			return;
		}
		remove(tempQuit);
		if ((fbuf = fopen(mbox, "r")) != NULL) {
			while ((c = getc(fbuf)) != EOF)
				putc(c, obuf);
			fclose(fbuf);
		}
		if (ferror(obuf)) {
			perror(tempQuit);
			fclose(ibuf);
			fclose(obuf);
			unlock();
			return;
		}
		fclose(obuf);
		close(creat(mbox, 0600));
		if ((obuf = fopen(mbox, "w")) == NULL) {
			perror(mbox);
			fclose(ibuf);
			unlock();
			return;
		}
	}
	if (value("append") != NOSTR)
		if ((obuf = fopen(mbox, "a")) == NULL) {
			perror(mbox);
			unlock();
			return;
		}
	for (mp = &message[0]; mp < &message[msgCount]; mp++)
		if ((mp->m_flag & (MDELETED|MSAVED|MPRESERVE)) == 0)
			if (send(mp, obuf) < 0) {
				perror(mbox);
				fclose(ibuf);
				fclose(obuf);
				unlock();
				return;
			}

	/*
	 * Copy the user's old mbox contents back
	 * to the end of the stuff we just saved.
	 * If we are appending, this is unnecessary.
	 */

	if (value("append") == NOSTR) {
		rewind(ibuf);
		c = getc(ibuf);
		while (c != EOF) {
			putc(c, obuf);
			if (ferror(obuf))
				break;
			c = getc(ibuf);
		}
		fclose(ibuf);
		fflush(obuf);
	}
	if (ferror(obuf)) {
		perror(mbox);
		fclose(obuf);
		unlock();
		return;
	}
	fclose(obuf);
	if (mcount == 1)
		printf("Saved 1 message in mbox\n");
	else
		printf("Saved %d messages in mbox\n", mcount);

	/*
	 * Now we are ready to copy back preserved files to
	 * the system mailbox, if any were requested.
	 */

	if (p != 0) {
		writeback(rbuf);
		unlock();
		return;
	}

	/*
	 * Finally, remove his /usr/mail file.
	 * If new mail has arrived, copy it back.
	 */

cream:
	if (rbuf != NULL) {
		fbuf = fopen(mailname, "w");
		if (fbuf == NULL)
			goto newmail;
		while ((c = getc(rbuf)) != EOF)
			putc(c, fbuf);
		fclose(rbuf);
		fclose(fbuf);
		alter(mailname);
		unlock();
		return;
	}
	demail();
	unlock();
	return;

newmail:
	printf("Thou hast new mail.\n");
	unlock();
}

/*
 * Preserve all the appropriate messages back in the system
 * mailbox, and print a nice message indicated how many were
 * saved.  On any error, just return -1.  Else return 0.
 * Incorporate the any new mail that we found.
 */

writeback(res)
	register FILE *res;
{
	register struct message *mp;
	register int p, c;
	FILE *obuf;

	p = 0;
	if ((obuf = fopen(mailname, "w")) == NULL) {
		perror(mailname);
		return(-1);
	}
#ifndef APPEND
	if (res != NULL)
		while ((c = getc(res)) != EOF)
			putc(c, obuf);
#endif
	for (mp = &message[0]; mp < &message[msgCount]; mp++)
		if ((mp->m_flag&MPRESERVE)||(mp->m_flag&MTOUCH)==0) {
			p++;
			if (send(mp, obuf) < 0) {
				perror(mailname);
				fclose(obuf);
				return(-1);
			}
		}
#ifdef APPEND
	if (res != NULL)
		while ((c = getc(res)) != EOF)
			putc(c, obuf);
#endif
	fflush(obuf);
	if (ferror(obuf)) {
		perror(mailname);
		fclose(obuf);
		return(-1);
	}
	if (res != NULL)
		fclose(res);
	fclose(obuf);
	alter(mailname);
	if (p == 1)
		printf("Held 1 message in %s\n", mailname);
	else
		printf("Held %d messages in %s\n", p, mailname);
	return(0);
}
Commit	Line	Data
865dca9a KS	1	#
	2
	3	#include "rcv.h"
	4	#include <sys/stat.h>
	5
	6	/*
	7	* Rcv -- receive mail rationally.
	8	*
	9	* Termination processing.
	10	*/
	11
63e83a9b	12	static char *SccsId = "@(#)quit.c 1.4 %G%";
865dca9a KS	13
	14	/*
	15	* Save all of the undetermined messages at the top of "mbox"
	16	* Save all untouched messages back in the system mailbox.
	17	* Remove the system mailbox, if none saved there.
	18	*/
	19
	20	quit()
	21	{
63e83a9b	22	int mcount, p, modify, autohold, anystat;
865dca9a KS	23	FILE ibuf, obuf, fbuf, rbuf;
	24	register struct message *mp;
	25	register int c;
	26	extern char tempQuit[], tempResid[];
	27	struct stat minfo;
	28
3ecebc0f KS	29	/*
	30	* If we are read only, we can't do anything,
	31	* so just return quickly.
	32	*/
	33
	34	if (readonly)
	35	return;
865dca9a KS	36	/*
	37	* See if there any messages to save in mbox. If no, we
	38	* can save copying mbox to /tmp and back.
	39	*
	40	* Check also to see if any files need to be preserved.
	41	* Delete all untouched messages to keep them out of mbox.
	42	* If all the messages are to be preserved, just exit with
	43	* a message.
	44	*
	45	* If the luser has sent mail to himself, refuse to do
	46	* anything with the mailbox, unless mail locking works.
	47	*/
	48
	49	lock(mailname);
	50	#ifndef CANLOCK
	51	if (selfsent) {
	52	printf("You have new mail.\n");
	53	unlock();
	54	return;
	55	}
	56	#endif
	57	rbuf = NULL;
	58	if (stat(mailname, &minfo) >= 0 && minfo.st_size > mailsize) {
	59	printf("New mail has arrived.\n");
	60	rbuf = fopen(tempResid, "w");
	61	fbuf = fopen(mailname, "r");
	62	if (rbuf == NULL \|\| fbuf == NULL)
	63	goto newmail;
	64	#ifdef APPEND
	65	fseek(fbuf, mailsize, 0);
	66	while ((c = getc(fbuf)) != EOF)
	67	putc(c, rbuf);
	68	#else
	69	p = minfo.st_size - mailsize;
	70	while (p-- > 0) {
	71	c = getc(fbuf);
	72	if (c == EOF)
	73	goto newmail;
	74	putc(c, rbuf);
	75	}
	76	#endif
	77	fclose(fbuf);
	78	fclose(rbuf);
	79	if ((rbuf = fopen(tempResid, "r")) == NULL)
	80	goto newmail;
	81	remove(tempResid);
	82	}
63e83a9b KS	83
	84	/*
	85	* Adjust the message flags in each message.
	86	*/
	87
	88	anystat = 0;
865dca9a	89	for (mp = &message[0]; mp < &message[msgCount]; mp++) {
63e83a9b KS	90	if (mp->m_flag & MNEW) {
	91	mp->m_flag &= ~MNEW;
	92	mp->m_flag \|= MSTATUS;
	93	}
	94	if (mp->m_flag & MSTATUS)
	95	anystat++;
865dca9a KS	96	if (mp->m_flag & MDELETED)
	97	mp->m_flag = MDELETED\|MTOUCH;
	98	if ((mp->m_flag & MTOUCH) == 0)
	99	mp->m_flag \|= MDELETED;
	100	}
a333d502	101	autohold = value("hold") != NOSTR;
865dca9a KS	102	modify = 0;
865dca9a KS	103	for (c = 0, p = 0, mp = &message[0]; mp < &message[msgCount]; mp++) {
a333d502 KS	104	if ((mp->m_flag & (MSAVED\|MDELETED)) == 0 && autohold)
a333d502 KS	105	mp->m_flag \|= MPRESERVE;
865dca9a KS	106	if ((mp->m_flag & (MSAVED\|MDELETED\|MPRESERVE)) == 0)
	107	c++;
	108	if ((mp->m_flag & MPRESERVE) \|\| (mp->m_flag & MTOUCH) == 0)
	109	p++;
	110	if (mp->m_flag & MODIFY)
	111	modify++;
	112	}
63e83a9b	113	if (p == msgCount && !modify && !anystat) {
865dca9a KS	114	if (p == 1)
	115	printf("Held 1 message in %s\n", mailname);
	116	else
	117	printf("Held %2d messages in %s\n", p, mailname);
	118	unlock();
	119	return;
	120	}
	121	if (c == 0) {
	122	if (p != 0) {
	123	writeback(rbuf);
	124	unlock();
	125	return;
	126	}
	127	goto cream;
	128	}
	129
	130	/*
	131	* Create another temporary file and copy user's mbox file
	132	* darin. If there is no mbox, copy nothing.
	133	* If he has specified "append" don't copy his mailbox,
	134	* just copy saveable entries at the end.
	135	*/
	136
	137	mcount = c;
	138	if (value("append") == NOSTR) {
	139	if ((obuf = fopen(tempQuit, "w")) == NULL) {
	140	perror(tempQuit);
	141	unlock();
	142	return;
	143	}
	144	if ((ibuf = fopen(tempQuit, "r")) == NULL) {
	145	perror(tempQuit);
	146	remove(tempQuit);
	147	fclose(obuf);
	148	unlock();
	149	return;
	150	}
	151	remove(tempQuit);
	152	if ((fbuf = fopen(mbox, "r")) != NULL) {
	153	while ((c = getc(fbuf)) != EOF)
	154	putc(c, obuf);
	155	fclose(fbuf);
	156	}
	157	if (ferror(obuf)) {
	158	perror(tempQuit);
	159	fclose(ibuf);
	160	fclose(obuf);
	161	unlock();
	162	return;
	163	}
	164	fclose(obuf);
	165	close(creat(mbox, 0600));
	166	if ((obuf = fopen(mbox, "w")) == NULL) {
	167	perror(mbox);
	168	fclose(ibuf);
	169	unlock();
	170	return;
	171	}
	172	}
	173	if (value("append") != NOSTR)
	174	if ((obuf = fopen(mbox, "a")) == NULL) {
	175	perror(mbox);
	176	unlock();
	177	return;
178	}
179	for (mp = &message[0]; mp < &message[msgCount]; mp++)
180	if ((mp->m_flag & (MDELETED\|MSAVED\|MPRESERVE)) == 0)
181	if (send(mp, obuf) < 0) {
182	perror(mbox);
183	fclose(ibuf);
184	fclose(obuf);
185	unlock();
186	return;
187	}
188
189	/*
190	* Copy the user's old mbox contents back
191	* to the end of the stuff we just saved.
192	* If we are appending, this is unnecessary.
193	*/
194
195	if (value("append") == NOSTR) {
196	rewind(ibuf);
197	c = getc(ibuf);
198	while (c != EOF) {
199	putc(c, obuf);
200	if (ferror(obuf))
201	break;
202	c = getc(ibuf);
203	}
204	fclose(ibuf);
205	fflush(obuf);
206	}
207	if (ferror(obuf)) {
208	perror(mbox);
209	fclose(obuf);
210	unlock();
211	return;
212	}
213	fclose(obuf);
214	if (mcount == 1)
215	printf("Saved 1 message in mbox\n");
216	else
217	printf("Saved %d messages in mbox\n", mcount);
218
219	/*
220	* Now we are ready to copy back preserved files to
221	* the system mailbox, if any were requested.
222	*/
223
224	if (p != 0) {
225	writeback(rbuf);
226	unlock();
227	return;
228	}
229
230	/*
231	* Finally, remove his /usr/mail file.
232	* If new mail has arrived, copy it back.
233	*/
234
235	cream:
236	if (rbuf != NULL) {
237	fbuf = fopen(mailname, "w");
238	if (fbuf == NULL)
239	goto newmail;
240	while ((c = getc(rbuf)) != EOF)
241	putc(c, fbuf);
242	fclose(rbuf);
243	fclose(fbuf);
244	alter(mailname);
245	unlock();
246	return;
247	}
248	demail();
249	unlock();
250	return;
251
252	newmail:
253	printf("Thou hast new mail.\n");
254	unlock();
255	}
256
257	/*
258	* Preserve all the appropriate messages back in the system
259	* mailbox, and print a nice message indicated how many were
260	* saved. On any error, just return -1. Else return 0.
261	* Incorporate the any new mail that we found.
262	*/
263
264	writeback(res)
265	register FILE *res;
266	{
267	register struct message *mp;
268	register int p, c;
269	FILE *obuf;
270
271	p = 0;
272	if ((obuf = fopen(mailname, "w")) == NULL) {
273	perror(mailname);
274	return(-1);
275	}
276	#ifndef APPEND
277	if (res != NULL)
278	while ((c = getc(res)) != EOF)
279	putc(c, obuf);
280	#endif
281	for (mp = &message[0]; mp < &message[msgCount]; mp++)
282	if ((mp->m_flag&MPRESERVE)\|\|(mp->m_flag&MTOUCH)==0) {
283	p++;
284	if (send(mp, obuf) < 0) {
285	perror(mailname);
286	fclose(obuf);
287	return(-1);
288	}
289	}
290	#ifdef APPEND
291	if (res != NULL)
292	while ((c = getc(res)) != EOF)
293	putc(c, obuf);
294	#endif
295	fflush(obuf);
296	if (ferror(obuf)) {
297	perror(mailname);
298	fclose(obuf);
299	return(-1);
300	}
301	if (res != NULL)
302	fclose(res);
303	fclose(obuf);
304	alter(mailname);
305	if (p == 1)
306	printf("Held 1 message in %s\n", mailname);
307	else
308	printf("Held %d messages in %s\n", p, mailname);
309	return(0);
310	}