initial import of NetBSD tree

1 files changed, 327 insertions, 0 deletions

diff --git a/usr.bin/vi/ex/ex_write.c b/usr.bin/vi/ex/ex_write.c
new file mode 100644
index 00000000000..e5273e4e675
--- /dev/null
+++ b/usr.bin/vi/ex/ex_write.c
@@ -0,0 +1,327 @@
+/*-
+ * Copyright (c) 1992, 1993, 1994
+ *	The Regents of the University of California.  All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ *    must display the following acknowledgement:
+ *	This product includes software developed by the University of
+ *	California, Berkeley and its contributors.
+ * 4. Neither the name of the University nor the names of its contributors
+ *    may be used to endorse or promote products derived from this software
+ *    without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#ifndef lint
+static char sccsid[] = "@(#)ex_write.c	8.38 (Berkeley) 8/17/94";
+#endif /* not lint */
+
+#include <sys/types.h>
+#include <sys/queue.h>
+#include <sys/stat.h>
+#include <sys/time.h>
+
+#include <bitstring.h>
+#include <ctype.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <limits.h>
+#include <signal.h>
+#include <stdio.h>
+#include <string.h>
+#include <termios.h>
+#include <unistd.h>
+
+#include "compat.h"
+#include <db.h>
+#include <regex.h>
+
+#include "vi.h"
+#include "excmd.h"
+
+enum which {WN, WQ, WRITE, XIT};
+
+static int exwr __P((SCR *, EXF *, EXCMDARG *, enum which));
+
+/*
+ * ex_wn --	:wn[!] [>>] [file]
+ *	Write to a file and switch to the next one.
+ */
+int
+ex_wn(sp, ep, cmdp)
+	SCR *sp;
+	EXF *ep;
+	EXCMDARG *cmdp;
+{
+	if (exwr(sp, ep, cmdp, WN))
+		return (1);
+	if (file_m3(sp, ep, 0))
+		return (1);
+
+	/* The file name isn't a new file to edit. */
+	cmdp->argc = 0;
+
+	return (ex_next(sp, ep, cmdp));
+}
+
+/*
+ * ex_wq --	:wq[!] [>>] [file]
+ *	Write to a file and quit.
+ */
+int
+ex_wq(sp, ep, cmdp)
+	SCR *sp;
+	EXF *ep;
+	EXCMDARG *cmdp;
+{
+	int force;
+
+	if (exwr(sp, ep, cmdp, WQ))
+		return (1);
+	if (file_m3(sp, ep, 0))
+		return (1);
+
+	force = F_ISSET(cmdp, E_FORCE);
+
+	if (ex_ncheck(sp, force))
+		return (1);
+
+	F_SET(sp, force ? S_EXIT_FORCE : S_EXIT);
+	return (0);
+}
+
+/*
+ * ex_write --	:write[!] [>>] [file]
+ *		:write [!] [cmd]
+ *	Write to a file.
+ */
+int
+ex_write(sp, ep, cmdp)
+	SCR *sp;
+	EXF *ep;
+	EXCMDARG *cmdp;
+{
+	return (exwr(sp, ep, cmdp, WRITE));
+}
+
+
+/*
+ * ex_xit -- :x[it]! [file]
+ *
+ *	Write out any modifications and quit.
+ */
+int
+ex_xit(sp, ep, cmdp)
+	SCR *sp;
+	EXF *ep;
+	EXCMDARG *cmdp;
+{
+	int force;
+
+	if (F_ISSET((ep), F_MODIFIED) && exwr(sp, ep, cmdp, XIT))
+		return (1);
+	if (file_m3(sp, ep, 0))
+		return (1);
+
+	force = F_ISSET(cmdp, E_FORCE);
+
+	if (ex_ncheck(sp, force))
+		return (1);
+
+	F_SET(sp, force ? S_EXIT_FORCE : S_EXIT);
+	return (0);
+}
+
+/*
+ * exwr --
+ *	The guts of the ex write commands.
+ */
+static int
+exwr(sp, ep, cmdp, cmd)
+	SCR *sp;
+	EXF *ep;
+	EXCMDARG *cmdp;
+	enum which cmd;
+{
+	EX_PRIVATE *exp;
+	MARK rm;
+	int flags;
+	char *name, *p;
+
+	/* All write commands can have an associated '!'. */
+	LF_INIT(FS_POSSIBLE);
+	if (F_ISSET(cmdp, E_FORCE))
+		LF_SET(FS_FORCE);
+
+	/* Skip any leading whitespace. */
+	if (cmdp->argc != 0)
+		for (p = cmdp->argv[0]->bp; *p && isblank(*p); ++p);
+
+	/* If no arguments, just write the file back. */
+	if (cmdp->argc == 0 || *p == '\0') {
+		if (F_ISSET(cmdp, E_ADDR2_ALL))
+			LF_SET(FS_ALL);
+		return (file_write(sp, ep,
+		    &cmdp->addr1, &cmdp->addr2, NULL, flags));
+	}
+
+	/* If "write !" it's a pipe to a utility. */
+	exp = EXP(sp);
+	if (cmd == WRITE && *p == '!') {
+		for (++p; *p && isblank(*p); ++p);
+		if (*p == '\0') {
+			msgq(sp, M_ERR, "Usage: %s", cmdp->cmd->usage);
+			return (1);
+		}
+		/* Expand the argument. */
+		if (argv_exp1(sp, ep, cmdp, p, strlen(p), 0))
+			return (1);
+		if (filtercmd(sp, ep, &cmdp->addr1, &cmdp->addr2,
+		    &rm, cmdp->argv[1]->bp, FILTER_WRITE))
+			return (1);
+		sp->lno = rm.lno;
+		return (0);
+	}
+
+	/* If "write >>" it's an append to a file. */
+	if (cmd != XIT && p[0] == '>' && p[1] == '>') {
+		LF_SET(FS_APPEND);
+
+		/* Skip ">>" and whitespace. */
+		for (p += 2; *p && isblank(*p); ++p);
+	}
+
+	/* Build an argv so we get an argument count and file expansion. */
+	if (argv_exp2(sp, ep, cmdp, p, strlen(p), 0))
+		return (1);
+
+	switch (cmdp->argc) {
+	case 1:
+		/*
+		 * Nothing to expand, write the current file.
+		 * XXX
+		 * Should never happen, already checked this case.
+		 */
+		name = NULL;
+		break;
+	case 2:
+		/* One new argument, write it. */
+		name = cmdp->argv[exp->argsoff - 1]->bp;
+		set_alt_name(sp, name);
+		break;
+	default:
+		/* If expanded to more than one argument, object. */
+		msgq(sp, M_ERR, "%s expanded into too many file names",
+		    cmdp->argv[0]->bp);
+		msgq(sp, M_ERR, "Usage: %s", cmdp->cmd->usage);
+		return (1);
+	}
+
+	if (F_ISSET(cmdp, E_ADDR2_ALL))
+		LF_SET(FS_ALL);
+	return (file_write(sp, ep, &cmdp->addr1, &cmdp->addr2, name, flags));
+}
+
+/*
+ * ex_writefp --
+ *	Write a range of lines to a FILE *.
+ */
+int
+ex_writefp(sp, ep, name, fp, fm, tm, nlno, nch)
+	SCR *sp;
+	EXF *ep;
+	char *name;
+	FILE *fp;
+	MARK *fm, *tm;
+	u_long *nlno, *nch;
+{
+	struct stat sb;
+	u_long ccnt;			/* XXX: can't print off_t portably. */
+	recno_t fline, tline, lcnt;
+	size_t len;
+	int sv_errno;
+	char *p;
+
+	fline = fm->lno;
+	tline = tm->lno;
+
+	if (nlno != NULL) {
+		*nch = 0;
+		*nlno = 0;
+	}
+
+	/*
+	 * The vi filter code has multiple processes running simultaneously,
+	 * and one of them calls ex_writefp().  The "unsafe" function calls
+	 * in this code are to file_gline() and msgq().  File_gline() is safe,
+	 * see the comment in filter.c:filtercmd() for details.  We don't call
+	 * msgq if the multiple process bit in the EXF is set.
+	 *
+	 * !!!
+	 * Historic vi permitted files of 0 length to be written.  However,
+	 * since the way vi got around dealing with "empty" files was to
+	 * always have a line in the file no matter what, it wrote them as
+	 * files of a single, empty line.  We write empty files.
+	 *
+	 * "Alex, I'll take vi trivia for $1000."
+	 */
+	ccnt = 0;
+	lcnt = 0;
+	if (tline != 0) {
+		for (; fline <= tline; ++fline, ++lcnt) {
+			/* Caller has to provide any interrupt message. */
+			if (INTERRUPTED(sp))
+				break;
+			if ((p = file_gline(sp, ep, fline, &len)) == NULL)
+				break;
+			if (fwrite(p, 1, len, fp) != len) {
+				msgq(sp, M_SYSERR, name);
+				(void)fclose(fp);
+				return (1);
+			}
+			ccnt += len;
+			if (putc('\n', fp) != '\n')
+				break;
+			++ccnt;
+		}
+	}
+
+	/* If it's a regular file, sync it so that NFS is forced to flush. */
+	if (!fstat(fileno(fp), &sb) &&
+	    S_ISREG(sb.st_mode) && fsync(fileno(fp))) {
+		sv_errno = errno;
+		(void)fclose(fp);
+		errno = sv_errno;
+		goto err;
+	}
+	if (fclose(fp))
+		goto err;
+	if (nlno != NULL) {
+		*nch = ccnt;
+		*nlno = lcnt;
+	}
+	return (0);
+
+err:	if (!F_ISSET(ep, F_MULTILOCK))
+		msgq(sp, M_SYSERR, name);
+	return (1);
+}