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. 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 } 208