usr.bin/mail/quit.c

   1 /*
   2  * Copyright (c) 1980, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #ifndef lint
  35 static char sccsid[] = "@(#)quit.c      8.1 (Berkeley) 6/6/93";
  36 #endif /* not lint */
  37
  38 #include "rcv.h"
  39 #include <fcntl.h>
  40 #include "extern.h"
  41
  42 /*
  43  * Rcv -- receive mail rationally.
  44  *
  45  * Termination processing.
  46  */
  47
  48 /*
  49  * The "quit" command.
  50  */
  51 int
  52 quitcmd()
  53 {
  54         /*
  55          * If we are sourcing, then return 1 so execute() can handle it.
  56          * Otherwise, return -1 to abort command loop.
  57          */
  58         if (sourcing)
  59                 return 1;
  60         return -1;
  61 }
  62
  63 /*
  64  * Save all of the undetermined messages at the top of "mbox"
  65  * Save all untouched messages back in the system mailbox.
  66  * Remove the system mailbox, if none saved there.
  67  */
  68 void
  69 quit()
  70 {
  71         int mcount, p, modify, autohold, anystat, holdbit, nohold;
  72         FILE *ibuf, *obuf, *fbuf, *rbuf, *readstat, *abuf;
  73         register struct message *mp;
  74         register int c;
  75         extern char *tempQuit, *tempResid;
  76         struct stat minfo;
  77         char *mbox;
  78
  79         /*
  80          * If we are read only, we can't do anything,
  81          * so just return quickly.
  82          */
  83         if (readonly)
  84                 return;
  85         /*
  86          * If editing (not reading system mail box), then do the work
  87          * in edstop()
  88          */
  89         if (edit) {
  90                 edstop();
  91                 return;
  92         }
  93
  94         /*
  95          * See if there any messages to save in mbox.  If no, we
  96          * can save copying mbox to /tmp and back.
  97          *
  98          * Check also to see if any files need to be preserved.
  99          * Delete all untouched messages to keep them out of mbox.
 100          * If all the messages are to be preserved, just exit with
 101          * a message.
 102          */
 103
 104         fbuf = Fopen(mailname, "r");
 105         if (fbuf == NULL)
 106                 goto newmail;
 107         flock(fileno(fbuf), LOCK_EX);
 108         rbuf = NULL;
 109         if (fstat(fileno(fbuf), &minfo) >= 0 && minfo.st_size > mailsize) {
 110                 printf("New mail has arrived.\n");
 111                 rbuf = Fopen(tempResid, "w");
 112                 if (rbuf == NULL || fbuf == NULL)
 113                         goto newmail;
 114 #ifdef APPEND
 115                 fseek(fbuf, (long)mailsize, 0);
 116                 while ((c = getc(fbuf)) != EOF)
 117                         (void) putc(c, rbuf);
 118 #else
 119                 p = minfo.st_size - mailsize;
 120                 while (p-- > 0) {
 121                         c = getc(fbuf);
 122                         if (c == EOF)
 123                                 goto newmail;
 124                         (void) putc(c, rbuf);
 125                 }
 126 #endif
 127                 Fclose(rbuf);
 128                 if ((rbuf = Fopen(tempResid, "r")) == NULL)
 129                         goto newmail;
 130                 rm(tempResid);
 131         }
 132
 133         /*
 134          * Adjust the message flags in each message.
 135          */
 136
 137         anystat = 0;
 138         autohold = value("hold") != NOSTR;
 139         holdbit = autohold ? MPRESERVE : MBOX;
 140         nohold = MBOX|MSAVED|MDELETED|MPRESERVE;
 141         if (value("keepsave") != NOSTR)
 142                 nohold &= ~MSAVED;
 143         for (mp = &message[0]; mp < &message[msgCount]; mp++) {
 144                 if (mp->m_flag & MNEW) {
 145                         mp->m_flag &= ~MNEW;
 146                         mp->m_flag |= MSTATUS;
 147                 }
 148                 if (mp->m_flag & MSTATUS)
 149                         anystat++;
 150                 if ((mp->m_flag & MTOUCH) == 0)
 151                         mp->m_flag |= MPRESERVE;
 152                 if ((mp->m_flag & nohold) == 0)
 153                         mp->m_flag |= holdbit;
 154         }
 155         modify = 0;
 156         if (Tflag != NOSTR) {
 157                 if ((readstat = Fopen(Tflag, "w")) == NULL)
 158                         Tflag = NOSTR;
 159         }
 160         for (c = 0, p = 0, mp = &message[0]; mp < &message[msgCount]; mp++) {
 161                 if (mp->m_flag & MBOX)
 162                         c++;
 163                 if (mp->m_flag & MPRESERVE)
 164                         p++;
 165                 if (mp->m_flag & MODIFY)
 166                         modify++;
 167                 if (Tflag != NOSTR && (mp->m_flag & (MREAD|MDELETED)) != 0) {
 168                         char *id;
 169
 170                         if ((id = hfield("article-id", mp)) != NOSTR)
 171                                 fprintf(readstat, "%s\n", id);
 172                 }
 173         }
 174         if (Tflag != NOSTR)
 175                 Fclose(readstat);
 176         if (p == msgCount && !modify && !anystat) {
 177                 printf("Held %d message%s in %s\n",
 178                         p, p == 1 ? "" : "s", mailname);
 179                 Fclose(fbuf);
 180                 return;
 181         }
 182         if (c == 0) {
 183                 if (p != 0) {
 184                         writeback(rbuf);
 185                         Fclose(fbuf);
 186                         return;
 187                 }
 188                 goto cream;
 189         }
 190
 191         /*
 192          * Create another temporary file and copy user's mbox file
 193          * darin.  If there is no mbox, copy nothing.
 194          * If he has specified "append" don't copy his mailbox,
 195          * just copy saveable entries at the end.
 196          */
 197
 198         mbox = expand("&");
 199         mcount = c;
 200         if (value("append") == NOSTR) {
 201                 if ((obuf = Fopen(tempQuit, "w")) == NULL) {
 202                         perror(tempQuit);
 203                         Fclose(fbuf);
 204                         return;
 205                 }
 206                 if ((ibuf = Fopen(tempQuit, "r")) == NULL) {
 207                         perror(tempQuit);
 208                         rm(tempQuit);
 209                         Fclose(obuf);
 210                         Fclose(fbuf);
 211                         return;
 212                 }
 213                 rm(tempQuit);
 214                 if ((abuf = Fopen(mbox, "r")) != NULL) {
 215                         while ((c = getc(abuf)) != EOF)
 216                                 (void) putc(c, obuf);
 217                         Fclose(abuf);
 218                 }
 219                 if (ferror(obuf)) {
 220                         perror(tempQuit);
 221                         Fclose(ibuf);
 222                         Fclose(obuf);
 223                         Fclose(fbuf);
 224                         return;
 225                 }
 226                 Fclose(obuf);
 227                 close(creat(mbox, 0600));
 228                 if ((obuf = Fopen(mbox, "r+")) == NULL) {
 229                         perror(mbox);
 230                         Fclose(ibuf);
 231                         Fclose(fbuf);
 232                         return;
 233                 }
 234         }
 235         if (value("append") != NOSTR) {
 236                 if ((obuf = Fopen(mbox, "a")) == NULL) {
 237                         perror(mbox);
 238                         Fclose(fbuf);
 239                         return;
 240                 }
 241                 fchmod(fileno(obuf), 0600);
 242         }
 243         for (mp = &message[0]; mp < &message[msgCount]; mp++)
 244                 if (mp->m_flag & MBOX)
 245                         if (send(mp, obuf, saveignore, NOSTR) < 0) {
 246                                 perror(mbox);
 247                                 Fclose(ibuf);
 248                                 Fclose(obuf);
 249                                 Fclose(fbuf);
 250                                 return;
 251                         }
 252
 253         /*
 254          * Copy the user's old mbox contents back
 255          * to the end of the stuff we just saved.
 256          * If we are appending, this is unnecessary.
 257          */
 258
 259         if (value("append") == NOSTR) {
 260                 rewind(ibuf);
 261                 c = getc(ibuf);
 262                 while (c != EOF) {
 263                         (void) putc(c, obuf);
 264                         if (ferror(obuf))
 265                                 break;
 266                         c = getc(ibuf);
 267                 }
 268                 Fclose(ibuf);
 269                 fflush(obuf);
 270         }
 271         trunc(obuf);
 272         if (ferror(obuf)) {
 273                 perror(mbox);
 274                 Fclose(obuf);
 275                 Fclose(fbuf);
 276                 return;
 277         }
 278         Fclose(obuf);
 279         if (mcount == 1)
 280                 printf("Saved 1 message in mbox\n");
 281         else
 282                 printf("Saved %d messages in mbox\n", mcount);
 283
 284         /*
 285          * Now we are ready to copy back preserved files to
 286          * the system mailbox, if any were requested.
 287          */
 288
 289         if (p != 0) {
 290                 writeback(rbuf);
 291                 Fclose(fbuf);
 292                 return;
 293         }
 294
 295         /*
 296          * Finally, remove his /usr/mail file.
 297          * If new mail has arrived, copy it back.
 298          */
 299
 300 cream:
 301         if (rbuf != NULL) {
 302                 abuf = Fopen(mailname, "r+");
 303                 if (abuf == NULL)
 304                         goto newmail;
 305                 while ((c = getc(rbuf)) != EOF)
 306                         (void) putc(c, abuf);
 307                 Fclose(rbuf);
 308                 trunc(abuf);
 309                 Fclose(abuf);
 310                 alter(mailname);
 311                 Fclose(fbuf);
 312                 return;
 313         }
 314         demail();
 315         Fclose(fbuf);
 316         return;
 317
 318 newmail:
 319         printf("Thou hast new mail.\n");
 320         if (fbuf != NULL)
 321                 Fclose(fbuf);
 322 }
 323
 324 /*
 325  * Preserve all the appropriate messages back in the system
 326  * mailbox, and print a nice message indicated how many were
 327  * saved.  On any error, just return -1.  Else return 0.
 328  * Incorporate the any new mail that we found.
 329  */
 330 int
 331 writeback(res)
 332         register FILE *res;
 333 {
 334         register struct message *mp;
 335         register int p, c;
 336         FILE *obuf;
 337
 338         p = 0;
 339         if ((obuf = Fopen(mailname, "r+")) == NULL) {
 340                 perror(mailname);
 341                 return(-1);
 342         }
 343 #ifndef APPEND
 344         if (res != NULL)
 345                 while ((c = getc(res)) != EOF)
 346                         (void) putc(c, obuf);
 347 #endif
 348         for (mp = &message[0]; mp < &message[msgCount]; mp++)
 349                 if ((mp->m_flag&MPRESERVE)||(mp->m_flag&MTOUCH)==0) {
 350                         p++;
 351                         if (send(mp, obuf, (struct ignoretab *)0, NOSTR) < 0) {
 352                                 perror(mailname);
 353                                 Fclose(obuf);
 354                                 return(-1);
 355                         }
 356                 }
 357 #ifdef APPEND
 358         if (res != NULL)
 359                 while ((c = getc(res)) != EOF)
 360                         (void) putc(c, obuf);
 361 #endif
 362         fflush(obuf);
 363         trunc(obuf);
 364         if (ferror(obuf)) {
 365                 perror(mailname);
 366                 Fclose(obuf);
 367                 return(-1);
 368         }
 369         if (res != NULL)
 370                 Fclose(res);
 371         Fclose(obuf);
 372         alter(mailname);
 373         if (p == 1)
 374                 printf("Held 1 message in %s\n", mailname);
 375         else
 376                 printf("Held %d messages in %s\n", p, mailname);
 377         return(0);
 378 }
 379
 380 /*
 381  * Terminate an editing session by attempting to write out the user's
 382  * file from the temporary.  Save any new stuff appended to the file.
 383  */
 384 void
 385 edstop()
 386 {
 387         extern char *tmpdir;
 388         register int gotcha, c;
 389         register struct message *mp;
 390         FILE *obuf, *ibuf, *readstat;
 391         struct stat statb;
 392         char tempname[30];
 393         char *mktemp();
 394
 395         if (readonly)
 396                 return;
 397         holdsigs();
 398         if (Tflag != NOSTR) {
 399                 if ((readstat = Fopen(Tflag, "w")) == NULL)
 400                         Tflag = NOSTR;
 401         }
 402         for (mp = &message[0], gotcha = 0; mp < &message[msgCount]; mp++) {
 403                 if (mp->m_flag & MNEW) {
 404                         mp->m_flag &= ~MNEW;
 405                         mp->m_flag |= MSTATUS;
 406                 }
 407                 if (mp->m_flag & (MODIFY|MDELETED|MSTATUS))
 408                         gotcha++;
 409                 if (Tflag != NOSTR && (mp->m_flag & (MREAD|MDELETED)) != 0) {
 410                         char *id;
 411
 412                         if ((id = hfield("article-id", mp)) != NOSTR)
 413                                 fprintf(readstat, "%s\n", id);
 414                 }
 415         }
 416         if (Tflag != NOSTR)
 417                 Fclose(readstat);
 418         if (!gotcha || Tflag != NOSTR)
 419                 goto done;
 420         ibuf = NULL;
 421         if (stat(mailname, &statb) >= 0 && statb.st_size > mailsize) {
 422                 strcpy(tempname, tmpdir);
 423                 strcat(tempname, "mboxXXXXXX");
 424                 mktemp(tempname);
 425                 if ((obuf = Fopen(tempname, "w")) == NULL) {
 426                         perror(tempname);
 427                         relsesigs();
 428                         reset(0);
 429                 }
 430                 if ((ibuf = Fopen(mailname, "r")) == NULL) {
 431                         perror(mailname);
 432                         Fclose(obuf);
 433                         rm(tempname);
 434                         relsesigs();
 435                         reset(0);
 436                 }
 437                 fseek(ibuf, (long)mailsize, 0);
 438                 while ((c = getc(ibuf)) != EOF)
 439                         (void) putc(c, obuf);
 440                 Fclose(ibuf);
 441                 Fclose(obuf);
 442                 if ((ibuf = Fopen(tempname, "r")) == NULL) {
 443                         perror(tempname);
 444                         rm(tempname);
 445                         relsesigs();
 446                         reset(0);
 447                 }
 448                 rm(tempname);
 449         }
 450         printf("\"%s\" ", mailname);
 451         fflush(stdout);
 452         if ((obuf = Fopen(mailname, "r+")) == NULL) {
 453                 perror(mailname);
 454                 relsesigs();
 455                 reset(0);
 456         }
 457         trunc(obuf);
 458         c = 0;
 459         for (mp = &message[0]; mp < &message[msgCount]; mp++) {
 460                 if ((mp->m_flag & MDELETED) != 0)
 461                         continue;
 462                 c++;
 463                 if (send(mp, obuf, (struct ignoretab *) NULL, NOSTR) < 0) {
 464                         perror(mailname);
 465                         relsesigs();
 466                         reset(0);
 467                 }
 468         }
 469         gotcha = (c == 0 && ibuf == NULL);
 470         if (ibuf != NULL) {
 471                 while ((c = getc(ibuf)) != EOF)
 472                         (void) putc(c, obuf);
 473                 Fclose(ibuf);
 474         }
 475         fflush(obuf);
 476         if (ferror(obuf)) {
 477                 perror(mailname);
 478                 relsesigs();
 479                 reset(0);
 480         }
 481         Fclose(obuf);
 482         if (gotcha) {
 483                 rm(mailname);
 484                 printf("removed\n");
 485         } else
 486                 printf("complete\n");
 487         fflush(stdout);
 488
 489 done:
 490         relsesigs();
 491 }