usr.bin/mail/edit.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  * 4. Neither the name of the University nor the names of its contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  *
  29  * @(#)edit.c   8.1 (Berkeley) 6/6/93
  30  * $FreeBSD: src/usr.bin/mail/edit.c,v 1.2.6.4 2003/01/06 05:46:03 mikeh Exp $
  31  * $DragonFly: src/usr.bin/mail/edit.c,v 1.4 2004/09/08 03:01:11 joerg Exp $
  32  */
  33
  34 #include "rcv.h"
  35 #include <fcntl.h>
  36 #include "extern.h"
  37
  38 /*
  39  * Mail -- a mail program
  40  *
  41  * Perform message editing functions.
  42  */
  43
  44 /*
  45  * Edit a message list.
  46  */
  47 int
  48 editor(int *msgvec)
  49 {
  50         return (edit1(msgvec, 'e'));
  51 }
  52
  53 /*
  54  * Invoke the visual editor on a message list.
  55  */
  56 int
  57 visual(int *msgvec)
  58 {
  59         return (edit1(msgvec, 'v'));
  60 }
  61
  62 /*
  63  * Edit a message by writing the message into a funnily-named file
  64  * (which should not exist) and forking an editor on it.
  65  * We get the editor from the stuff above.
  66  */
  67 int
  68 edit1(int *msgvec, int type)
  69 {
  70         int c, i;
  71         FILE *fp;
  72         struct message *mp;
  73         off_t size;
  74
  75         /*
  76          * Deal with each message to be edited . . .
  77          */
  78         for (i = 0; msgvec[i] && i < msgCount; i++) {
  79                 sig_t sigint;
  80
  81                 if (i > 0) {
  82                         char buf[100];
  83                         char *p;
  84
  85                         printf("Edit message %d [ynq]? ", msgvec[i]);
  86                         if (fgets(buf, sizeof(buf), stdin) == 0)
  87                                 break;
  88                         for (p = buf; *p == ' ' || *p == '\t'; p++)
  89                                 ;
  90                         if (*p == 'q')
  91                                 break;
  92                         if (*p == 'n')
  93                                 continue;
  94                 }
  95                 dot = mp = &message[msgvec[i] - 1];
  96                 touch(mp);
  97                 sigint = signal(SIGINT, SIG_IGN);
  98                 fp = run_editor(setinput(mp), mp->m_size, type, readonly);
  99                 if (fp != NULL) {
 100                         fseeko(otf, (off_t)0, SEEK_END);
 101                         size = ftello(otf);
 102                         mp->m_block = blockof(size);
 103                         mp->m_offset = boffsetof(size);
 104                         mp->m_size = (long)fsize(fp);
 105                         mp->m_lines = 0;
 106                         mp->m_flag |= MODIFY;
 107                         rewind(fp);
 108                         while ((c = getc(fp)) != EOF) {
 109                                 if (c == '\n')
 110                                         mp->m_lines++;
 111                                 if (putc(c, otf) == EOF)
 112                                         break;
 113                         }
 114                         if (ferror(otf))
 115                                 warnx("/tmp");
 116                         Fclose(fp);
 117                 }
 118                 signal(SIGINT, sigint);
 119         }
 120         return (0);
 121 }
 122
 123 /*
 124  * Run an editor on the file at "fpp" of "size" bytes,
 125  * and return a new file pointer.
 126  * Signals must be handled by the caller.
 127  * "Type" is 'e' for _PATH_EX, 'v' for _PATH_VI.
 128  */
 129 FILE *
 130 run_editor(FILE *fp, off_t size, int type, int readonly)
 131 {
 132         FILE *nf = NULL;
 133         int t;
 134         time_t modtime;
 135         char *edit, tempname[PATHSIZE];
 136         struct stat statb;
 137
 138         snprintf(tempname, sizeof(tempname), "%s/mail.ReXXXXXXXXXX", tmpdir);
 139         if ((t = mkstemp(tempname)) == -1 ||
 140             (nf = Fdopen(t, "w")) == NULL) {
 141                 warn("%s", tempname);
 142                 goto out;
 143         }
 144         if (readonly && fchmod(t, 0400) == -1) {
 145                 warn("%s", tempname);
 146                 rm(tempname);
 147                 goto out;
 148         }
 149         if (size >= 0)
 150                 while (--size >= 0 && (t = getc(fp)) != EOF)
 151                         putc(t, nf);
 152         else
 153                 while ((t = getc(fp)) != EOF)
 154                         putc(t, nf);
 155         fflush(nf);
 156         if (fstat(fileno(nf), &statb) < 0)
 157                 modtime = 0;
 158         else
 159                 modtime = statb.st_mtime;
 160         if (ferror(nf)) {
 161                 Fclose(nf);
 162                 warnx("%s", tempname);
 163                 rm(tempname);
 164                 nf = NULL;
 165                 goto out;
 166         }
 167         if (Fclose(nf) < 0) {
 168                 warn("%s", tempname);
 169                 rm(tempname);
 170                 nf = NULL;
 171                 goto out;
 172         }
 173         nf = NULL;
 174         if ((edit = value(type == 'e' ? "EDITOR" : "VISUAL")) == NULL)
 175                 edit = type == 'e' ? _PATH_EX : _PATH_VI;
 176         if (run_command(edit, 0, -1, -1, tempname, NULL, NULL) < 0) {
 177                 rm(tempname);
 178                 goto out;
 179         }
 180         /*
 181          * If in read only mode or file unchanged, just remove the editor
 182          * temporary and return.
 183          */
 184         if (readonly) {
 185                 rm(tempname);
 186                 goto out;
 187         }
 188         if (stat(tempname, &statb) < 0) {
 189                 warn("%s", tempname);
 190                 goto out;
 191         }
 192         if (modtime == statb.st_mtime) {
 193                 rm(tempname);
 194                 goto out;
 195         }
 196         /*
 197          * Now switch to new file.
 198          */
 199         if ((nf = Fopen(tempname, "a+")) == NULL) {
 200                 warn("%s", tempname);
 201                 rm(tempname);
 202                 goto out;
 203         }
 204         rm(tempname);
 205 out:
 206         return (nf);
 207 }