1/*-
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.
6 *
7 * See the LICENSE file for redistribution information.
8 */
9
10#include "config.h"
11
12#ifndef lint
13static const char copyright[] =
14"@(#) Copyright (c) 1992, 1993, 1994\n\
15	The Regents of the University of California.  All rights reserved.\n\
16@(#) Copyright (c) 1992, 1993, 1994, 1995, 1996\n\
17	Keith Bostic.  All rights reserved.\n";
18#endif /* not lint */
19
20#ifndef lint
21static const char sccsid[] = "$Id: main.c,v 11.0 2012/10/17 06:34:37 zy Exp $";
22#endif /* not lint */
23
24#include <sys/types.h>
25#include <sys/queue.h>
26#include <sys/stat.h>
27
28#include <bitstring.h>
29#include <errno.h>
30#include <fcntl.h>
31#include <limits.h>
32#include <stdio.h>
33#include <stdlib.h>
34#include <string.h>
35#include <unistd.h>
36
37#include "common.h"
38#include "../vi/vi.h"
39#include "pathnames.h"
40
41static void	 attach __P((GS *));
42static void	 v_estr __P((char *, int, char *));
43static int	 v_obsolete __P((char *, char *[]));
44
45/*
46 * editor --
47 *	Main editor routine.
48 *
49 * PUBLIC: int editor __P((GS *, int, char *[]));
50 */
51int
52editor(
53	GS *gp,
54	int argc,
55	char *argv[])
56{
57	extern int optind;
58	extern char *optarg;
59	const char *p;
60	EVENT ev;
61	FREF *frp;
62	SCR *sp;
63	size_t len;
64	u_int flags;
65	int ch, flagchk, lflag, secure, startup, readonly, rval, silent;
66	char *tag_f, *wsizearg, path[256];
67	CHAR_T *w;
68	size_t wlen;
69
70	/* Initialize the busy routine, if not defined by the screen. */
71	if (gp->scr_busy == NULL)
72		gp->scr_busy = vs_busy;
73	/* Initialize the message routine, if not defined by the screen. */
74	if (gp->scr_msg == NULL)
75		gp->scr_msg = vs_msg;
76	gp->catd = (nl_catd)-1;
77
78	/* Common global structure initialization. */
79	TAILQ_INIT(gp->dq);
80	TAILQ_INIT(gp->hq);
81	SLIST_INIT(gp->ecq);
82	SLIST_INSERT_HEAD(gp->ecq, &gp->excmd, q);
83	gp->noprint = DEFAULT_NOPRINT;
84
85	/* Structures shared by screens so stored in the GS structure. */
86	TAILQ_INIT(gp->frefq);
87	TAILQ_INIT(gp->dcb_store.textq);
88	SLIST_INIT(gp->cutq);
89	SLIST_INIT(gp->seqq);
90
91	/* Set initial screen type and mode based on the program name. */
92	readonly = 0;
93	if (!strcmp(gp->progname, "ex") || !strcmp(gp->progname, "nex"))
94		LF_INIT(SC_EX);
95	else {
96		/* Nview, view are readonly. */
97		if (!strcmp(gp->progname, "nview") ||
98		    !strcmp(gp->progname, "view"))
99			readonly = 1;
100
101		/* Vi is the default. */
102		LF_INIT(SC_VI);
103	}
104
105	/* Convert old-style arguments into new-style ones. */
106	if (v_obsolete(gp->progname, argv))
107		return (1);
108
109	/* Parse the arguments. */
110	flagchk = '\0';
111	tag_f = wsizearg = NULL;
112	lflag = secure = silent = 0;
113	startup = 1;
114
115	/* Set the file snapshot flag. */
116	F_SET(gp, G_SNAPSHOT);
117
118#ifdef DEBUG
119	while ((ch = getopt(argc, argv, "c:D:eFlRrSsT:t:vw:")) != EOF)
120#else
121	while ((ch = getopt(argc, argv, "c:eFlRrSst:vw:")) != EOF)
122#endif
123		switch (ch) {
124		case 'c':		/* Run the command. */
125			/*
126			 * XXX
127			 * We should support multiple -c options.
128			 */
129			if (gp->c_option != NULL) {
130				v_estr(gp->progname, 0,
131				    "only one -c command may be specified.");
132				return (1);
133			}
134			gp->c_option = optarg;
135			break;
136#ifdef DEBUG
137		case 'D':
138			switch (optarg[0]) {
139			case 's':
140				startup = 0;
141				break;
142			case 'w':
143				attach(gp);
144				break;
145			default:
146				v_estr(gp->progname, 0,
147				    "usage: -D requires s or w argument.");
148				return (1);
149			}
150			break;
151#endif
152		case 'e':		/* Ex mode. */
153			LF_CLR(SC_VI);
154			LF_SET(SC_EX);
155			break;
156		case 'F':		/* No snapshot. */
157			F_CLR(gp, G_SNAPSHOT);
158			break;
159		case 'l':		/* Set lisp, showmatch options. */
160			lflag = 1;
161			break;
162		case 'R':		/* Readonly. */
163			readonly = 1;
164			break;
165		case 'r':		/* Recover. */
166			if (flagchk == 't') {
167				v_estr(gp->progname, 0,
168				    "only one of -r and -t may be specified.");
169				return (1);
170			}
171			flagchk = 'r';
172			break;
173		case 'S':
174			secure = 1;
175			break;
176		case 's':
177			silent = 1;
178			break;
179#ifdef DEBUG
180		case 'T':		/* Trace. */
181			if ((gp->tracefp = fopen(optarg, "w")) == NULL) {
182				v_estr(gp->progname, errno, optarg);
183				goto err;
184			}
185			(void)fprintf(gp->tracefp,
186			    "\n===\ntrace: open %s\n", optarg);
187			break;
188#endif
189		case 't':		/* Tag. */
190			if (flagchk == 'r') {
191				v_estr(gp->progname, 0,
192				    "only one of -r and -t may be specified.");
193				return (1);
194			}
195			if (flagchk == 't') {
196				v_estr(gp->progname, 0,
197				    "only one tag file may be specified.");
198				return (1);
199			}
200			flagchk = 't';
201			tag_f = optarg;
202			break;
203		case 'v':		/* Vi mode. */
204			LF_CLR(SC_EX);
205			LF_SET(SC_VI);
206			break;
207		case 'w':
208			wsizearg = optarg;
209			break;
210		case '?':
211		default:
212			(void)gp->scr_usage();
213			return (1);
214		}
215	argc -= optind;
216	argv += optind;
217
218	/*
219	 * -s option is only meaningful to ex.
220	 *
221	 * If not reading from a terminal, it's like -s was specified.
222	 */
223	if (silent && !LF_ISSET(SC_EX)) {
224		v_estr(gp->progname, 0, "-s option is only applicable to ex.");
225		goto err;
226	}
227	if (LF_ISSET(SC_EX) && F_ISSET(gp, G_SCRIPTED))
228		silent = 1;
229
230	/*
231	 * Build and initialize the first/current screen.  This is a bit
232	 * tricky.  If an error is returned, we may or may not have a
233	 * screen structure.  If we have a screen structure, put it on a
234	 * display queue so that the error messages get displayed.
235	 *
236	 * !!!
237	 * Everything we do until we go interactive is done in ex mode.
238	 */
239	if (screen_init(gp, NULL, &sp)) {
240		if (sp != NULL)
241			TAILQ_INSERT_HEAD(gp->dq, sp, q);
242		goto err;
243	}
244	F_SET(sp, SC_EX);
245	TAILQ_INSERT_HEAD(gp->dq, sp, q);
246
247	if (v_key_init(sp))		/* Special key initialization. */
248		goto err;
249
250	{ int oargs[5], *oargp = oargs;
251	if (lflag) {			/* Command-line options. */
252		*oargp++ = O_LISP;
253		*oargp++ = O_SHOWMATCH;
254	}
255	if (readonly)
256		*oargp++ = O_READONLY;
257	if (secure)
258		*oargp++ = O_SECURE;
259	*oargp = -1;			/* Options initialization. */
260	if (opts_init(sp, oargs))
261		goto err;
262	}
263	if (wsizearg != NULL) {
264		ARGS *av[2], a, b;
265		(void)snprintf(path, sizeof(path), "window=%s", wsizearg);
266		a.bp = (CHAR_T *)path;
267		a.len = strlen(path);
268		b.bp = NULL;
269		b.len = 0;
270		av[0] = &a;
271		av[1] = &b;
272		(void)opts_set(sp, av, NULL);
273	}
274	if (silent) {			/* Ex batch mode option values. */
275		O_CLR(sp, O_AUTOPRINT);
276		O_CLR(sp, O_PROMPT);
277		O_CLR(sp, O_VERBOSE);
278		O_CLR(sp, O_WARN);
279		F_SET(sp, SC_EX_SILENT);
280	}
281
282	sp->rows = O_VAL(sp, O_LINES);	/* Make ex formatting work. */
283	sp->cols = O_VAL(sp, O_COLUMNS);
284
285	if (!silent && startup) {	/* Read EXINIT, exrc files. */
286		if (ex_exrc(sp))
287			goto err;
288		if (F_ISSET(sp, SC_EXIT | SC_EXIT_FORCE)) {
289			if (screen_end(sp))
290				goto err;
291			goto done;
292		}
293	}
294
295	/*
296	 * List recovery files if -r specified without file arguments.
297	 * Note, options must be initialized and startup information
298	 * read before doing this.
299	 */
300	if (flagchk == 'r' && argv[0] == NULL) {
301		if (rcv_list(sp))
302			goto err;
303		if (screen_end(sp))
304			goto err;
305		goto done;
306	}
307
308	/*
309	 * !!!
310	 * Initialize the default ^D, ^U scrolling value here, after the
311	 * user has had every opportunity to set the window option.
312	 *
313	 * It's historic practice that changing the value of the window
314	 * option did not alter the default scrolling value, only giving
315	 * a count to ^D/^U did that.
316	 */
317	sp->defscroll = (O_VAL(sp, O_WINDOW) + 1) / 2;
318
319	/*
320	 * If we don't have a command-line option, switch into the right
321	 * editor now, so that we position default files correctly, and
322	 * so that any tags file file-already-locked messages are in the
323	 * vi screen, not the ex screen.
324	 *
325	 * XXX
326	 * If we have a command-line option, the error message can end
327	 * up in the wrong place, but I think that the combination is
328	 * unlikely.
329	 */
330	if (gp->c_option == NULL) {
331		F_CLR(sp, SC_EX | SC_VI);
332		F_SET(sp, LF_ISSET(SC_EX | SC_VI));
333	}
334
335	/* Open a tag file if specified. */
336	if (tag_f != NULL) {
337		CHAR2INT(sp, tag_f, strlen(tag_f) + 1, w, wlen);
338		if (ex_tag_first(sp, w))
339			goto err;
340	}
341
342	/*
343	 * Append any remaining arguments as file names.  Files are recovery
344	 * files if -r specified.  If the tag option or ex startup commands
345	 * loaded a file, then any file arguments are going to come after it.
346	 */
347	if (*argv != NULL) {
348		if (sp->frp != NULL) {
349			/* Cheat -- we know we have an extra argv slot. */
350			*--argv = strdup(sp->frp->name);
351			if (*argv == NULL) {
352				v_estr(gp->progname, errno, NULL);
353				goto err;
354			}
355		}
356		sp->argv = sp->cargv = argv;
357		F_SET(sp, SC_ARGNOFREE);
358		if (flagchk == 'r')
359			F_SET(sp, SC_ARGRECOVER);
360	}
361
362	/*
363	 * If the ex startup commands and or/the tag option haven't already
364	 * created a file, create one.  If no command-line files were given,
365	 * use a temporary file.
366	 */
367	if (sp->frp == NULL) {
368		if (sp->argv == NULL) {
369			if ((frp = file_add(sp, NULL)) == NULL)
370				goto err;
371		} else  {
372			if ((frp = file_add(sp, sp->argv[0])) == NULL)
373				goto err;
374			if (F_ISSET(sp, SC_ARGRECOVER))
375				F_SET(frp, FR_RECOVER);
376		}
377
378		if (file_init(sp, frp, NULL, 0))
379			goto err;
380		if (EXCMD_RUNNING(gp)) {
381			(void)ex_cmd(sp);
382			if (F_ISSET(sp, SC_EXIT | SC_EXIT_FORCE)) {
383				if (screen_end(sp))
384					goto err;
385				goto done;
386			}
387		}
388	}
389
390	/*
391	 * Check to see if we need to wait for ex.  If SC_SCR_EX is set, ex
392	 * was forced to initialize the screen during startup.  We'd like to
393	 * wait for a single character from the user, but we can't because
394	 * we're not in raw mode.  We can't switch to raw mode because the
395	 * vi initialization will switch to xterm's alternate screen, causing
396	 * us to lose the messages we're pausing to make sure the user read.
397	 * So, wait for a complete line.
398	 */
399	if (F_ISSET(sp, SC_SCR_EX)) {
400		p = msg_cmsg(sp, CMSG_CONT_R, &len);
401		(void)write(STDOUT_FILENO, p, len);
402		for (;;) {
403			if (v_event_get(sp, &ev, 0, 0))
404				goto err;
405			if (ev.e_event == E_INTERRUPT ||
406			    (ev.e_event == E_CHARACTER &&
407			     (ev.e_value == K_CR || ev.e_value == K_NL)))
408				break;
409			(void)gp->scr_bell(sp);
410		}
411	}
412
413	/* Switch into the right editor, regardless. */
414	F_CLR(sp, SC_EX | SC_VI);
415	F_SET(sp, LF_ISSET(SC_EX | SC_VI) | SC_STATUS_CNT);
416
417	/*
418	 * Main edit loop.  Vi handles split screens itself, we only return
419	 * here when switching editor modes or restarting the screen.
420	 */
421	while (sp != NULL)
422		if (F_ISSET(sp, SC_EX) ? ex(&sp) : vi(&sp))
423			goto err;
424
425done:	rval = 0;
426	if (0)
427err:		rval = 1;
428
429	/* Clean out the global structure. */
430	v_end(gp);
431
432	return (rval);
433}
434
435/*
436 * v_end --
437 *	End the program, discarding screens and most of the global area.
438 *
439 * PUBLIC: void v_end __P((GS *));
440 */
441void
442v_end(gp)
443	GS *gp;
444{
445	MSGS *mp;
446	SCR *sp;
447
448	/* If there are any remaining screens, kill them off. */
449	if (gp->ccl_sp != NULL) {
450		(void)file_end(gp->ccl_sp, NULL, 1);
451		(void)screen_end(gp->ccl_sp);
452	}
453	while ((sp = TAILQ_FIRST(gp->dq)) != NULL)
454		(void)screen_end(sp);
455	while ((sp = TAILQ_FIRST(gp->hq)) != NULL)
456		(void)screen_end(sp);
457
458#if defined(DEBUG) || defined(PURIFY) || defined(LIBRARY)
459	{ FREF *frp;
460		/* Free FREF's. */
461		while ((frp = TAILQ_FIRST(gp->frefq)) != NULL) {
462			TAILQ_REMOVE(gp->frefq, frp, q);
463			if (frp->name != NULL)
464				free(frp->name);
465			if (frp->tname != NULL)
466				free(frp->tname);
467			free(frp);
468		}
469	}
470
471	/* Free key input queue. */
472	if (gp->i_event != NULL)
473		free(gp->i_event);
474
475	/* Free cut buffers. */
476	cut_close(gp);
477
478	/* Free map sequences. */
479	seq_close(gp);
480
481	/* Free default buffer storage. */
482	(void)text_lfree(gp->dcb_store.textq);
483
484	/* Close message catalogs. */
485	msg_close(gp);
486#endif
487
488	/* Ring the bell if scheduled. */
489	if (F_ISSET(gp, G_BELLSCHED))
490		(void)fprintf(stderr, "\07");		/* \a */
491
492	/*
493	 * Flush any remaining messages.  If a message is here, it's almost
494	 * certainly the message about the event that killed us (although
495	 * it's possible that the user is sourcing a file that exits from the
496	 * editor).
497	 */
498	while ((mp = SLIST_FIRST(gp->msgq)) != NULL) {
499		(void)fprintf(stderr, "%s%.*s",
500		    mp->mtype == M_ERR ? "ex/vi: " : "", (int)mp->len, mp->buf);
501		SLIST_REMOVE_HEAD(gp->msgq, q);
502#if defined(DEBUG) || defined(PURIFY) || defined(LIBRARY)
503		free(mp->buf);
504		free(mp);
505#endif
506	}
507
508#if defined(DEBUG) || defined(PURIFY) || defined(LIBRARY)
509	/* Free any temporary space. */
510	if (gp->tmp_bp != NULL)
511		free(gp->tmp_bp);
512
513#if defined(DEBUG)
514	/* Close debugging file descriptor. */
515	if (gp->tracefp != NULL)
516		(void)fclose(gp->tracefp);
517#endif
518#endif
519}
520
521/*
522 * v_obsolete --
523 *	Convert historic arguments into something getopt(3) will like.
524 */
525static int
526v_obsolete(
527	char *name,
528	char *argv[])
529{
530	size_t len;
531	char *p;
532
533	/*
534	 * Translate old style arguments into something getopt will like.
535	 * Make sure it's not text space memory, because ex modifies the
536	 * strings.
537	 *	Change "+" into "-c$".
538	 *	Change "+<anything else>" into "-c<anything else>".
539	 *	Change "-" into "-s"
540	 *	The c, T, t and w options take arguments so they can't be
541	 *	    special arguments.
542	 *
543	 * Stop if we find "--" as an argument, the user may want to edit
544	 * a file named "+foo".
545	 */
546	while (*++argv && strcmp(argv[0], "--"))
547		if (argv[0][0] == '+') {
548			if (argv[0][1] == '\0') {
549				argv[0] = strdup("-c$");
550				if (argv[0] == NULL)
551					goto nomem;
552			} else  {
553				p = argv[0];
554				len = strlen(argv[0]);
555				argv[0] = malloc(len + 2);
556				if (argv[0] == NULL)
557					goto nomem;
558				argv[0][0] = '-';
559				argv[0][1] = 'c';
560				(void)strlcpy(argv[0] + 2, p + 1, len);
561			}
562		} else if (argv[0][0] == '-')
563			if (argv[0][1] == '\0') {
564				argv[0] = strdup("-s");
565				if (argv[0] == NULL) {
566nomem:					v_estr(name, errno, NULL);
567					return (1);
568				}
569			} else
570				if ((argv[0][1] == 'c' || argv[0][1] == 'T' ||
571				    argv[0][1] == 't' || argv[0][1] == 'w') &&
572				    argv[0][2] == '\0')
573					++argv;
574	return (0);
575}
576
577#ifdef DEBUG
578static void
579attach(GS *gp)
580{
581	int fd;
582	char ch;
583
584	if ((fd = open(_PATH_TTY, O_RDONLY, 0)) < 0) {
585		v_estr(gp->progname, errno, _PATH_TTY);
586		return;
587	}
588
589	(void)printf("process %lu waiting, enter <CR> to continue: ",
590	    (u_long)getpid());
591	(void)fflush(stdout);
592
593	do {
594		if (read(fd, &ch, 1) != 1) {
595			(void)close(fd);
596			return;
597		}
598	} while (ch != '\n' && ch != '\r');
599	(void)close(fd);
600}
601#endif
602
603static void
604v_estr(
605	char *name,
606	int eno,
607	char *msg)
608{
609	(void)fprintf(stderr, "%s", name);
610	if (msg != NULL)
611		(void)fprintf(stderr, ": %s", msg);
612	if (eno)
613		(void)fprintf(stderr, ": %s", strerror(errno));
614	(void)fprintf(stderr, "\n");
615}
616