File: [local] / src / usr.bin / mail / quit.c (download)
Revision 1.23, Tue Jul 19 06:43:27 2016 UTC (7 years, 10 months ago) by deraadt
Branch: MAIN
CVS Tags: OPENBSD_7_5_BASE, OPENBSD_7_5, OPENBSD_7_4_BASE, OPENBSD_7_4, OPENBSD_7_3_BASE, OPENBSD_7_3, OPENBSD_7_2_BASE, OPENBSD_7_2, OPENBSD_7_1_BASE, OPENBSD_7_1, OPENBSD_7_0_BASE, OPENBSD_7_0, OPENBSD_6_9_BASE, OPENBSD_6_9, OPENBSD_6_8_BASE, OPENBSD_6_8, OPENBSD_6_7_BASE, OPENBSD_6_7, OPENBSD_6_6_BASE, OPENBSD_6_6, OPENBSD_6_5_BASE, OPENBSD_6_5, OPENBSD_6_4_BASE, OPENBSD_6_4, OPENBSD_6_3_BASE, OPENBSD_6_3, OPENBSD_6_2_BASE, OPENBSD_6_2, OPENBSD_6_1_BASE, OPENBSD_6_1, OPENBSD_6_0_BASE, OPENBSD_6_0, HEAD Changes since 1.22: +5 -2 lines
Cleanup close(open idioms.
ok krw
|
/* $OpenBSD: quit.c,v 1.23 2016/07/19 06:43:27 deraadt Exp $ */
/* $NetBSD: quit.c,v 1.6 1996/12/28 07:11:07 tls Exp $ */
/*
* Copyright (c) 1980, 1993
* The Regents of the University of California. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include "rcv.h"
#include <fcntl.h>
#include "extern.h"
/*
* Rcv -- receive mail rationally.
*
* Termination processing.
*/
/*
* The "quit" command.
*/
int
quitcmd(void *v)
{
/*
* If we are sourcing, then return 1 so execute() can handle it.
* Otherwise, return -1 to abort command loop.
*/
if (sourcing)
return(1);
return(-1);
}
/*
* 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.
*/
int
quit(void)
{
int mcount, p, modify, autohold, anystat, holdbit, nohold;
FILE *ibuf = NULL, *obuf, *fbuf, *rbuf, *abuf;
struct message *mp;
int c, fd;
struct stat minfo;
char *mbox, tempname[PATHSIZE];
/*
* If we are read only, we can't do anything,
* so just return quickly.
*/
if (readonly)
return(0);
/*
* If editing (not reading system mail box), then do the work
* in edstop()
*/
if (edit)
return(edstop());
/*
* 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.
*/
fbuf = Fopen(mailname, "r+");
if (fbuf == NULL)
goto newmail;
if (flock(fileno(fbuf), LOCK_EX) == -1) {
warn("Unable to lock mailbox");
(void)Fclose(fbuf);
return(-1);
}
if (!spool_lock()) {
(void)Fclose(fbuf);
return(-1); /* lockspool printed the error for us */
}
rbuf = NULL;
if (fstat(fileno(fbuf), &minfo) >= 0 && minfo.st_size > mailsize) {
puts("New mail has arrived.");
(void)snprintf(tempname, sizeof(tempname),
"%s/mail.RqXXXXXXXXXX", tmpdir);
if ((fd = mkstemp(tempname)) == -1 ||
(rbuf = Fdopen(fd, "w")) == NULL)
goto newmail;
#ifdef APPEND
fseek(fbuf, (long)mailsize, SEEK_SET);
while ((c = getc(fbuf)) != EOF)
(void)putc(c, rbuf);
#else
p = minfo.st_size - mailsize;
while (p-- > 0) {
c = getc(fbuf);
if (c == EOF)
goto newmail;
(void)putc(c, rbuf);
}
#endif
(void)Fclose(rbuf);
if ((rbuf = Fopen(tempname, "r")) == NULL)
goto newmail;
(void)rm(tempname);
}
/*
* Adjust the message flags in each message.
*/
anystat = 0;
autohold = value("hold") != NULL;
holdbit = autohold ? MPRESERVE : MBOX;
nohold = MBOX|MSAVED|MDELETED|MPRESERVE;
if (value("keepsave") != NULL)
nohold &= ~MSAVED;
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 & MTOUCH) == 0)
mp->m_flag |= MPRESERVE;
if ((mp->m_flag & nohold) == 0)
mp->m_flag |= holdbit;
}
modify = 0;
for (c = 0, p = 0, mp = &message[0]; mp < &message[msgCount]; mp++) {
if (mp->m_flag & MBOX)
c++;
if (mp->m_flag & MPRESERVE)
p++;
if (mp->m_flag & MODIFY)
modify++;
}
if (p == msgCount && !modify && !anystat) {
printf("Held %d message%s in %s\n",
p, p == 1 ? "" : "s", mailname);
(void)Fclose(fbuf);
spool_unlock();
return(0);
}
if (c == 0) {
if (p != 0) {
writeback(rbuf);
(void)Fclose(fbuf);
spool_unlock();
return(0);
}
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.
*/
mbox = expand("&");
mcount = c;
if (value("append") == NULL) {
int fdx;
(void)snprintf(tempname, sizeof(tempname),
"%s/mail.RmXXXXXXXXXX", tmpdir);
if ((fd = mkstemp(tempname)) == -1 ||
(obuf = Fdopen(fd, "w")) == NULL) {
warn("%s", tempname);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
if ((ibuf = Fopen(tempname, "r")) == NULL) {
warn("%s", tempname);
(void)rm(tempname);
(void)Fclose(obuf);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
(void)rm(tempname);
if ((abuf = Fopen(mbox, "r")) != NULL) {
while ((c = getc(abuf)) != EOF)
(void)putc(c, obuf);
(void)Fclose(abuf);
}
if (ferror(obuf)) {
warn("%s", tempname);
(void)Fclose(ibuf);
(void)Fclose(obuf);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
(void)Fclose(obuf);
if ((fdx = open(mbox, O_CREAT | O_TRUNC | O_WRONLY, 0600)) != -1)
close(fdx);
if ((obuf = Fopen(mbox, "r+")) == NULL) {
warn("%s", mbox);
(void)Fclose(ibuf);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
} else {
if ((obuf = Fopen(mbox, "a")) == NULL) {
warn("%s", mbox);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
fchmod(fileno(obuf), 0600);
}
for (mp = &message[0]; mp < &message[msgCount]; mp++)
if (mp->m_flag & MBOX)
if (sendmessage(mp, obuf, saveignore, NULL) < 0) {
warn("%s", mbox);
(void)Fclose(ibuf);
(void)Fclose(obuf);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
/*
* 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") == NULL) {
rewind(ibuf);
c = getc(ibuf);
while (c != EOF) {
(void)putc(c, obuf);
if (ferror(obuf))
break;
c = getc(ibuf);
}
(void)Fclose(ibuf);
fflush(obuf);
}
trunc(obuf);
if (ferror(obuf)) {
warn("%s", mbox);
(void)Fclose(obuf);
(void)Fclose(fbuf);
spool_unlock();
return(-1);
}
(void)Fclose(obuf);
if (mcount == 1)
puts("Saved 1 message in mbox");
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);
(void)Fclose(fbuf);
spool_unlock();
return(0);
}
/*
* Finally, remove his /var/mail file.
* If new mail has arrived, copy it back.
*/
cream:
if (rbuf != NULL) {
abuf = Fopen(mailname, "r+");
if (abuf == NULL)
goto newmail;
while ((c = getc(rbuf)) != EOF)
(void)putc(c, abuf);
(void)Fclose(rbuf);
trunc(abuf);
(void)Fclose(abuf);
alter(mailname);
(void)Fclose(fbuf);
spool_unlock();
return(0);
}
demail();
(void)Fclose(fbuf);
spool_unlock();
return(0);
newmail:
puts("Thou hast new mail.");
if (fbuf != NULL) {
(void)Fclose(fbuf);
spool_unlock();
}
return(0);
}
/*
* 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.
*/
int
writeback(FILE *res)
{
struct message *mp;
int p, c;
FILE *obuf;
p = 0;
if ((obuf = Fopen(mailname, "r+")) == NULL) {
warn("%s", mailname);
return(-1);
}
#ifndef APPEND
if (res != NULL)
while ((c = getc(res)) != EOF)
(void)putc(c, obuf);
#endif
for (mp = &message[0]; mp < &message[msgCount]; mp++)
if ((mp->m_flag&MPRESERVE)||(mp->m_flag&MTOUCH)==0) {
p++;
if (sendmessage(mp, obuf, NULL, NULL) < 0) {
warn("%s", mailname);
(void)Fclose(obuf);
return(-1);
}
}
#ifdef APPEND
if (res != NULL)
while ((c = getc(res)) != EOF)
(void)putc(c, obuf);
#endif
fflush(obuf);
trunc(obuf);
if (ferror(obuf)) {
warn("%s", mailname);
(void)Fclose(obuf);
return(-1);
}
if (res != NULL)
(void)Fclose(res);
(void)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);
}
/*
* Terminate an editing session by attempting to write out the user's
* file from the temporary. Save any new stuff appended to the file.
*/
int
edstop(void)
{
int gotcha, c;
struct message *mp;
FILE *obuf, *ibuf;
struct stat statb;
char tempname[PATHSIZE];
if (readonly)
return(0);
holdsigs();
for (mp = &message[0], gotcha = 0; mp < &message[msgCount]; mp++) {
if (mp->m_flag & MNEW) {
mp->m_flag &= ~MNEW;
mp->m_flag |= MSTATUS;
}
if (mp->m_flag & (MODIFY|MDELETED|MSTATUS))
gotcha++;
}
if (!gotcha)
goto done;
ibuf = NULL;
if (stat(mailname, &statb) >= 0 && statb.st_size > mailsize) {
int fd;
(void)snprintf(tempname, sizeof(tempname), "%s/mbox.XXXXXXXXXX",
tmpdir);
if ((fd = mkstemp(tempname)) == -1 ||
(obuf = Fdopen(fd, "w")) == NULL) {
warn("%s", tempname);
if (fd != -1)
close(fd);
relsesigs();
return(-1);
}
if ((ibuf = Fopen(mailname, "r")) == NULL) {
warn("%s", mailname);
(void)Fclose(obuf);
(void)rm(tempname);
relsesigs();
return(-1);
}
fseek(ibuf, (long)mailsize, SEEK_SET);
while ((c = getc(ibuf)) != EOF)
(void)putc(c, obuf);
(void)Fclose(ibuf);
(void)Fclose(obuf);
if ((ibuf = Fopen(tempname, "r")) == NULL) {
warn("%s", tempname);
(void)rm(tempname);
relsesigs();
return(-1);
}
(void)rm(tempname);
}
printf("\"%s\" ", mailname);
fflush(stdout);
if ((obuf = Fopen(mailname, "r+")) == NULL) {
warn("%s", mailname);
relsesigs();
return(-1);
}
trunc(obuf);
c = 0;
for (mp = &message[0]; mp < &message[msgCount]; mp++) {
if ((mp->m_flag & MDELETED) != 0)
continue;
c++;
if (sendmessage(mp, obuf, NULL, NULL) < 0) {
warn("%s", mailname);
relsesigs();
return(-1);
}
}
gotcha = (c == 0 && ibuf == NULL);
if (ibuf != NULL) {
while ((c = getc(ibuf)) != EOF)
(void)putc(c, obuf);
(void)Fclose(ibuf);
}
fflush(obuf);
if (ferror(obuf)) {
warn("%s", mailname);
relsesigs();
return(-1);
}
(void)Fclose(obuf);
if (gotcha) {
(void)rm(mailname);
puts("removed");
} else
puts("complete");
fflush(stdout);
done:
relsesigs();
return(0);
}