2 * Copyright (c) 1992, 1993, 1994
3 * The Regents of the University of California. All rights reserved.
4 * Copyright (c) 1992, 1993, 1994, 1995, 1996
5 * Keith Bostic. All rights reserved.
7 * See the LICENSE file for redistribution information.
13 static const char sccsid
[] = "$Id: ex_write.c,v 10.38 2001/06/25 15:19:22 skimo Exp $ (Berkeley) $Date: 2001/06/25 15:19:22 $";
16 #include <sys/types.h>
17 #include <sys/queue.h>
20 #include <bitstring.h>
30 #include "../common/common.h"
32 enum which
{WN
, WQ
, WRITE
, XIT
};
33 static int exwr
__P((SCR
*, EXCMD
*, enum which
));
36 * ex_wn -- :wn[!] [>>] [file]
37 * Write to a file and switch to the next one.
39 * PUBLIC: int ex_wn __P((SCR *, EXCMD *));
42 ex_wn(SCR
*sp
, EXCMD
*cmdp
)
44 if (exwr(sp
, cmdp
, WN
))
49 /* The file name isn't a new file to edit. */
52 return (ex_next(sp
, cmdp
));
56 * ex_wq -- :wq[!] [>>] [file]
57 * Write to a file and quit.
59 * PUBLIC: int ex_wq __P((SCR *, EXCMD *));
62 ex_wq(SCR
*sp
, EXCMD
*cmdp
)
66 if (exwr(sp
, cmdp
, WQ
))
71 force
= FL_ISSET(cmdp
->iflags
, E_C_FORCE
);
73 if (ex_ncheck(sp
, force
))
76 F_SET(sp
, force
? SC_EXIT_FORCE
: SC_EXIT
);
81 * ex_write -- :write[!] [>>] [file]
85 * PUBLIC: int ex_write __P((SCR *, EXCMD *));
88 ex_write(SCR
*sp
, EXCMD
*cmdp
)
90 return (exwr(sp
, cmdp
, WRITE
));
95 * ex_xit -- :x[it]! [file]
96 * Write out any modifications and quit.
98 * PUBLIC: int ex_xit __P((SCR *, EXCMD *));
101 ex_xit(SCR
*sp
, EXCMD
*cmdp
)
107 if (F_ISSET(sp
->ep
, F_MODIFIED
) && exwr(sp
, cmdp
, XIT
))
112 force
= FL_ISSET(cmdp
->iflags
, E_C_FORCE
);
114 if (ex_ncheck(sp
, force
))
117 F_SET(sp
, force
? SC_EXIT_FORCE
: SC_EXIT
);
123 * The guts of the ex write commands.
126 exwr(SCR
*sp
, EXCMD
*cmdp
, enum which cmd
)
138 /* All write commands can have an associated '!'. */
139 LF_INIT(FS_POSSIBLE
);
140 if (FL_ISSET(cmdp
->iflags
, E_C_FORCE
))
143 /* Skip any leading whitespace. */
145 for (p
= cmdp
->argv
[0]->bp
; *p
!= '\0' && isblank(*p
); ++p
);
147 /* If "write !" it's a pipe to a utility. */
148 if (cmdp
->argc
!= 0 && cmd
== WRITE
&& *p
== '!') {
149 /* Secure means no shell access. */
150 if (O_ISSET(sp
, O_SECURE
)) {
151 ex_wemsg(sp
, cmdp
->cmd
->name
, EXM_SECURE_F
);
155 /* Expand the argument. */
156 for (++p
; *p
&& isblank(*p
); ++p
);
158 ex_emsg(sp
, cmdp
->cmd
->usage
, EXM_USAGE
);
161 if (argv_exp1(sp
, cmdp
, p
, STRLEN(p
), 1))
165 * Historically, vi waited after a write filter even if there
166 * wasn't any output from the command. People complained when
167 * nvi waited only if there was output, wanting the visual cue
168 * that the program hadn't written anything.
170 F_SET(sp
, SC_EX_WAIT_YES
);
174 * Ignore the return cursor position, the cursor doesn't
177 if (ex_filter(sp
, cmdp
, &cmdp
->addr1
,
178 &cmdp
->addr2
, &rm
, cmdp
->argv
[1]->bp
, FILTER_WRITE
))
181 /* Ex terminates with a bang, even if the command fails. */
182 if (!F_ISSET(sp
, SC_VI
) && !F_ISSET(sp
, SC_EX_SILENT
))
183 (void)ex_puts(sp
, "!\n");
188 /* Set the FS_ALL flag if we're writing the entire file. */
189 if (cmdp
->addr1
.lno
<= 1 && !db_exist(sp
, cmdp
->addr2
.lno
+ 1))
192 /* If "write >>" it's an append to a file. */
193 if (cmdp
->argc
!= 0 && cmd
!= XIT
&& p
[0] == '>' && p
[1] == '>') {
196 /* Skip ">>" and whitespace. */
197 for (p
+= 2; *p
&& isblank(*p
); ++p
);
200 /* If no other arguments, just write the file back. */
201 if (cmdp
->argc
== 0 || *p
== '\0')
202 return (file_write(sp
,
203 &cmdp
->addr1
, &cmdp
->addr2
, NULL
, flags
));
205 /* Build an argv so we get an argument count and file expansion. */
206 if (argv_exp2(sp
, cmdp
, p
, STRLEN(p
)))
210 * 0 args: impossible.
211 * 1 args: impossible (I hope).
213 * >2 args: object, too many args.
215 * The 1 args case depends on the argv_sexp() function refusing
216 * to return success without at least one non-blank character.
218 switch (cmdp
->argc
) {
224 INT2CHAR(sp
, cmdp
->argv
[1]->bp
, cmdp
->argv
[1]->len
+1,
226 name
= v_strdup(sp
, n
, nlen
- 1);
230 * Historically, the read and write commands renamed
231 * "unnamed" files, or, if the file had a name, set
232 * the alternate file name.
234 if (F_ISSET(sp
->frp
, FR_TMPFILE
) &&
235 !F_ISSET(sp
->frp
, FR_EXNAMED
)) {
236 if ((n
= v_strdup(sp
, name
, nlen
- 1)) != NULL
) {
241 * The file has a real name, it's no longer a
242 * temporary, clear the temporary file flags.
245 * If we're writing the whole file, FR_NAMECHANGE
246 * will be cleared by the write routine -- this is
249 F_CLR(sp
->frp
, FR_TMPEXIT
| FR_TMPFILE
);
250 F_SET(sp
->frp
, FR_NAMECHANGE
| FR_EXNAMED
);
252 /* Notify the screen. */
253 (void)sp
->gp
->scr_rename(sp
, sp
->frp
->name
, 1);
255 set_alt_name(sp
, name
);
258 INT2CHAR(sp
, p
, STRLEN(p
) + 1, n
, nlen
);
259 ex_emsg(sp
, n
, EXM_FILECOUNT
);
263 rc
= file_write(sp
, &cmdp
->addr1
, &cmdp
->addr2
, name
, flags
);
272 * Write a range of lines to a FILE *.
274 * PUBLIC: int ex_writefp __P((SCR *,
275 * PUBLIC: char *, FILE *, MARK *, MARK *, u_long *, u_long *, int));
278 ex_writefp(SCR
*sp
, char *name
, FILE *fp
, MARK
*fm
, MARK
*tm
, u_long
*nlno
, u_long
*nch
, int silent
)
282 u_long ccnt
; /* XXX: can't print off_t portably. */
283 db_recno_t fline
, tline
, lcnt
;
301 * The vi filter code has multiple processes running simultaneously,
302 * and one of them calls ex_writefp(). The "unsafe" function calls
303 * in this code are to db_get() and msgq(). Db_get() is safe, see
304 * the comment in ex_filter.c:ex_filter() for details. We don't call
305 * msgq if the multiple process bit in the EXF is set.
308 * Historic vi permitted files of 0 length to be written. However,
309 * since the way vi got around dealing with "empty" files was to
310 * always have a line in the file no matter what, it wrote them as
311 * files of a single, empty line. We write empty files.
313 * "Alex, I'll take vi trivia for $1000."
317 msg
= "253|Writing...";
319 for (; fline
<= tline
; ++fline
, ++lcnt
) {
320 /* Caller has to provide any interrupt message. */
321 if ((lcnt
+ 1) % INTERRUPT_CHECK
== 0) {
325 gp
->scr_busy(sp
, msg
, msg
== NULL
?
326 BUSY_UPDATE
: BUSY_ON
);
330 if (db_get(sp
, fline
, DBG_FATAL
, &p
, &len
))
332 INT2FILE(sp
, p
, len
, f
, flen
);
333 if (fwrite(f
, 1, flen
, fp
) != flen
)
336 if (putc('\n', fp
) != '\n')
345 * I don't trust NFS -- check to make sure that we're talking to
346 * a regular file and sync so that NFS is forced to flush.
348 if (!fstat(fileno(fp
), &sb
) &&
349 S_ISREG(sb
.st_mode
) && fsync(fileno(fp
)))
357 err
: if (!F_ISSET(sp
->ep
, F_MULTILOCK
))
358 msgq_str(sp
, M_SYSERR
, name
, "%s");
364 gp
->scr_busy(sp
, NULL
, BUSY_OFF
);
366 /* Report the possibly partial transfer. */