BSD 2 development

[unix-history] / src / Mail / quit.c

/* Copyright (c) 1979 Regents of the University of California */
#

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

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

/*
* 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;
       FILE *ibuf, *obuf, *fbuf, *rbuf;
       register struct message *mp;
       register int c;
       extern char tempQuit[], tempResid[];
       struct stat minfo;

       /*
        * 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.
        */

       if (selfsent) {
newmail:
               printf("You have new mail.\n");
               unlock();
               return;
       }
       rbuf = NULL;
       lock(mailname);
       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 APPENDS
               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;
               unlink(tempResid);
       }
       for (mp = &message[0]; mp < &message[msgCount]; mp++) {
               if (mp->m_flag & MDELETED)
                       mp->m_flag = MDELETED|MTOUCH;
               if ((mp->m_flag & MTOUCH) == 0)
                       mp->m_flag |= MDELETED;
       }
       modify = 0;
       for (c = 0, p = 0, mp = &message[0]; mp < &message[msgCount]; mp++) {
               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 && rbuf == NULL) {
               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 remove;
       }

       /*
        * 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);
                       unlink(tempQuit);
                       fclose(obuf);
                       unlock();
                       return;
               }
               unlink(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.
        */

remove:
       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();
}

/*
* 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);
}