xref: /netbsd/bin/ksh/edit.c (revision 822c36bb)
1 /*	$NetBSD: edit.c,v 1.35 2018/06/03 12:18:29 kamil Exp $	*/
2 
3 /*
4  * Command line editing - common code
5  *
6  */
7 #include <sys/cdefs.h>
8 
9 #ifndef lint
10 __RCSID("$NetBSD: edit.c,v 1.35 2018/06/03 12:18:29 kamil Exp $");
11 #endif
12 
13 #include <stdbool.h>
14 
15 #include "config.h"
16 #ifdef EDIT
17 
18 #include "sh.h"
19 #include "tty.h"
20 #define EXTERN
21 #include "edit.h"
22 #undef EXTERN
23 #include <sys/ioctl.h>
24 #include <sys/stat.h>
25 #include <ctype.h>
26 
27 
28 #if defined(TIOCGWINSZ)
29 static RETSIGTYPE x_sigwinch ARGS((int sig));
30 static int got_sigwinch;
31 static void check_sigwinch ARGS((void));
32 #endif /* TIOCGWINSZ */
33 
34 static int	x_file_glob ARGS((int flags, const char *str, int slen,
35 				  char ***wordsp));
36 static int	x_command_glob ARGS((int flags, const char *str, int slen,
37 				     char ***wordsp));
38 static int	x_locate_word ARGS((const char *buf, int buflen, int pos,
39 				    int *startp, int *is_command));
40 
41 static char vdisable_c;
42 
43 
44 /* Called from main */
45 void
x_init()46 x_init()
47 {
48 	/* set to -2 to force initial binding */
49 	edchars.erase = edchars.kill = edchars.intr = edchars.quit
50 		= edchars.eof = -2;
51 	/* default value for deficient systems */
52 	edchars.werase = 027;	/* ^W */
53 
54 #ifdef TIOCGWINSZ
55 # ifdef SIGWINCH
56 	if (setsig(&sigtraps[SIGWINCH], x_sigwinch, SS_RESTORE_ORIG|SS_SHTRAP))
57 		sigtraps[SIGWINCH].flags |= TF_SHELL_USES;
58 # endif /* SIGWINCH */
59 	got_sigwinch = 1; /* force initial check */
60 	check_sigwinch();
61 #endif /* TIOCGWINSZ */
62 
63 #ifdef EMACS
64 	x_init_emacs();
65 #endif /* EMACS */
66 
67 	/* Bizarreness to figure out how to disable
68 	 * a struct termios.c_cc[] char
69 	 */
70 #ifdef _POSIX_VDISABLE
71 	if (_POSIX_VDISABLE >= 0)
72 		vdisable_c = (char) _POSIX_VDISABLE;
73 	else
74 		/* `feature not available' */
75 		vdisable_c = (char) 0377;
76 #else
77 # if defined(HAVE_PATHCONF) && defined(_PC_VDISABLE)
78 	vdisable_c = fpathconf(tty_fd, _PC_VDISABLE);
79 # else
80 	vdisable_c = (char) 0377;	/* default to old BSD value */
81 # endif
82 #endif /* _POSIX_VDISABLE */
83 }
84 
85 #if defined(TIOCGWINSZ)
86 static RETSIGTYPE
x_sigwinch(sig)87 x_sigwinch(sig)
88     	int sig;
89 {
90 	got_sigwinch = 1;
91 	return RETSIGVAL;
92 }
93 
94 static void
check_sigwinch(void)95 check_sigwinch ARGS((void))
96 {
97 	if (got_sigwinch) {
98 		struct winsize ws;
99 
100 		got_sigwinch = 0;
101 		if (procpid == kshpid && ioctl(tty_fd, TIOCGWINSZ, &ws) >= 0) {
102 			struct tbl *vp;
103 
104 			/* Do NOT export COLUMNS/LINES.  Many applications
105 			 * check COLUMNS/LINES before checking ws.ws_col/row,
106 			 * so if the app is started with C/L in the environ
107 			 * and the window is then resized, the app won't
108 			 * see the change cause the environ doesn't change.
109 			 */
110 			if (ws.ws_col) {
111 				x_cols = ws.ws_col < MIN_COLS ? MIN_COLS
112 						: ws.ws_col;
113 
114 				if ((vp = typeset("COLUMNS", 0, 0, 0, 0)))
115 					setint(vp, (long) ws.ws_col);
116 			}
117 			if (ws.ws_row
118 			    && (vp = typeset("LINES", 0, 0, 0, 0)))
119 				setint(vp, (long) ws.ws_row);
120 		}
121 	}
122 }
123 #endif /* TIOCGWINSZ */
124 
125 /*
126  * read an edited command line
127  */
128 int
x_read(buf,len)129 x_read(buf, len)
130 	char *buf;
131 	size_t len;
132 {
133 	int	i;
134 
135 	x_mode(true);
136 #ifdef EMACS
137 	if (Flag(FEMACS) || Flag(FGMACS))
138 		i = x_emacs(buf, len);
139 	else
140 #endif
141 #ifdef VI
142 	if (Flag(FVI))
143 		i = x_vi(buf, len);
144 	else
145 #endif
146 		i = -1;		/* internal error */
147 	x_mode(false);
148 #if defined(TIOCGWINSZ)
149 	if (got_sigwinch)
150 		check_sigwinch();
151 #endif /* TIOCGWINSZ */
152 
153 	return i;
154 }
155 
156 /* tty I/O */
157 
158 int
x_getc()159 x_getc()
160 {
161 	char c;
162 	int n;
163 
164 	while ((n = blocking_read(0, &c, 1)) < 0 && errno == EINTR)
165 		if (trap) {
166 			x_mode(false);
167 			runtraps(0);
168 			x_mode(true);
169 		}
170 	if (n != 1)
171 		return -1;
172 	return (int) (unsigned char) c;
173 }
174 
175 void
x_flush()176 x_flush()
177 {
178 	shf_flush(shl_out);
179 }
180 
181 void
x_putc(c)182 x_putc(c)
183 	int c;
184 {
185 	shf_putc(c, shl_out);
186 }
187 
188 void
x_puts(s)189 x_puts(s)
190 	const char *s;
191 {
192 	while (*s != 0)
193 		shf_putc(*s++, shl_out);
194 }
195 
196 bool
x_mode(bool onoff)197 x_mode(bool onoff)
198 {
199 	static bool	x_cur_mode;
200 	bool		prev;
201 
202 	if (x_cur_mode == onoff)
203 		return x_cur_mode;
204 	prev = x_cur_mode;
205 	x_cur_mode = onoff;
206 
207 	if (onoff) {
208 		TTY_state	cb;
209 		X_chars		oldchars;
210 
211 		oldchars = edchars;
212 		cb = tty_state;
213 
214 #if defined(HAVE_TERMIOS_H) || defined(HAVE_TERMIO_H)
215 		edchars.erase = cb.c_cc[VERASE];
216 		edchars.kill = cb.c_cc[VKILL];
217 		edchars.intr = cb.c_cc[VINTR];
218 		edchars.quit = cb.c_cc[VQUIT];
219 		edchars.eof = cb.c_cc[VEOF];
220 # ifdef VWERASE
221 		edchars.werase = cb.c_cc[VWERASE];
222 # endif
223 # ifdef _CRAY2		/* brain-damaged terminal handler */
224 		cb.c_lflag &= ~(ICANON|ECHO);
225 		/* rely on print routine to map '\n' to CR,LF */
226 # else
227 		cb.c_iflag &= ~(INLCR|ICRNL);
228 #  ifdef _BSD_SYSV	/* need to force CBREAK instead of RAW (need CRMOD on output) */
229 		cb.c_lflag &= ~(ICANON|ECHO);
230 #  else
231 #   ifdef SWTCH	/* need CBREAK to handle swtch char */
232 		cb.c_lflag &= ~(ICANON|ECHO);
233 		cb.c_lflag |= ISIG;
234 		cb.c_cc[VINTR] = vdisable_c;
235 		cb.c_cc[VQUIT] = vdisable_c;
236 #   else
237 		cb.c_lflag &= ~(ISIG|ICANON|ECHO);
238 #   endif
239 #  endif
240 #  ifdef VLNEXT
241 		/* osf/1 processes lnext when ~icanon */
242 		cb.c_cc[VLNEXT] = vdisable_c;
243 #  endif /* VLNEXT */
244 #  ifdef VDISCARD
245 		/* sunos 4.1.x & osf/1 processes discard(flush) when ~icanon */
246 		cb.c_cc[VDISCARD] = vdisable_c;
247 #  endif /* VDISCARD */
248 		cb.c_cc[VTIME] = 0;
249 		cb.c_cc[VMIN] = 1;
250 # endif	/* _CRAY2 */
251 #else
252 	/* Assume BSD tty stuff. */
253 		edchars.erase = cb.sgttyb.sg_erase;
254 		edchars.kill = cb.sgttyb.sg_kill;
255 		cb.sgttyb.sg_flags &= ~ECHO;
256 		cb.sgttyb.sg_flags |= CBREAK;
257 #  ifdef TIOCGATC
258 		edchars.intr = cb.lchars.tc_intrc;
259 		edchars.quit = cb.lchars.tc_quitc;
260 		edchars.eof = cb.lchars.tc_eofc;
261 		edchars.werase = cb.lchars.tc_werasc;
262 		cb.lchars.tc_suspc = -1;
263 		cb.lchars.tc_dsuspc = -1;
264 		cb.lchars.tc_lnextc = -1;
265 		cb.lchars.tc_statc = -1;
266 		cb.lchars.tc_intrc = -1;
267 		cb.lchars.tc_quitc = -1;
268 		cb.lchars.tc_rprntc = -1;
269 #  else
270 		edchars.intr = cb.tchars.t_intrc;
271 		edchars.quit = cb.tchars.t_quitc;
272 		edchars.eof = cb.tchars.t_eofc;
273 		cb.tchars.t_intrc = -1;
274 		cb.tchars.t_quitc = -1;
275 #   ifdef TIOCGLTC
276 		edchars.werase = cb.ltchars.t_werasc;
277 		cb.ltchars.t_suspc = -1;
278 		cb.ltchars.t_dsuspc = -1;
279 		cb.ltchars.t_lnextc = -1;
280 		cb.ltchars.t_rprntc = -1;
281 #   endif
282 #  endif /* TIOCGATC */
283 #endif /* HAVE_TERMIOS_H || HAVE_TERMIO_H */
284 
285 		set_tty(tty_fd, &cb, TF_WAIT);
286 
287 		/* Convert unset values to internal `unset' value */
288 		if (edchars.erase == vdisable_c)
289 			edchars.erase = -1;
290 		if (edchars.kill == vdisable_c)
291 			edchars.kill = -1;
292 		if (edchars.intr == vdisable_c)
293 			edchars.intr = -1;
294 		if (edchars.quit == vdisable_c)
295 			edchars.quit = -1;
296 		if (edchars.eof == vdisable_c)
297 			edchars.eof = -1;
298 		if (edchars.werase == vdisable_c)
299 			edchars.werase = -1;
300 		if (memcmp(&edchars, &oldchars, sizeof(edchars)) != 0) {
301 #ifdef EMACS
302 			x_emacs_keys(&edchars);
303 #endif
304 		}
305 	} else {
306 		/* TF_WAIT doesn't seem to be necessary when leaving xmode */
307 		set_tty(tty_fd, &tty_state, TF_NONE);
308 	}
309 
310 	return prev;
311 }
312 
313 /* NAME:
314  *      promptlen - calculate the length of PS1 etc.
315  *
316  * DESCRIPTION:
317  *      This function is based on a fix from guy@demon.co.uk
318  *      It fixes a bug in that if PS1 contains '!', the length
319  *      given by strlen() is probably wrong.
320  *
321  * RETURN VALUE:
322  *      length
323  */
324 int
promptlen(cp,spp)325 promptlen(cp, spp)
326     const char  *cp;
327     const char **spp;
328 {
329     int count = 0;
330     const char *sp = cp;
331     char delimiter = 0;
332     int indelimit = 0;
333 
334     /* Undocumented AT&T ksh feature:
335      * If the second char in the prompt string is \r then the first char
336      * is taken to be a non-printing delimiter and any chars between two
337      * instances of the delimiter are not considered to be part of the
338      * prompt length
339      */
340     if (*cp && cp[1] == '\r') {
341 	delimiter = *cp;
342 	cp += 2;
343     }
344 
345     for (; *cp; cp++) {
346 	if (indelimit && *cp != delimiter)
347 	    ;
348 	else if (*cp == '\n' || *cp == '\r') {
349 	    count = 0;
350 	    sp = cp + 1;
351 	} else if (*cp == '\t') {
352 	    count = (count | 7) + 1;
353 	} else if (*cp == '\b') {
354 	    if (count > 0)
355 		count--;
356 	} else if (*cp == delimiter)
357 	    indelimit = !indelimit;
358 	else
359 	    count++;
360     }
361     if (spp)
362 	*spp = sp;
363     return count;
364 }
365 
366 void
set_editmode(ed)367 set_editmode(ed)
368 	const char *ed;
369 {
370 	static const enum sh_flag edit_flags[] = {
371 #ifdef EMACS
372 			FEMACS, FGMACS,
373 #endif
374 #ifdef VI
375 			FVI,
376 #endif
377 		    };
378 	char *rcp;
379 	size_t i;
380 
381 	if ((rcp = ksh_strrchr_dirsep(ed)))
382 		ed = ++rcp;
383 	for (i = 0; i < NELEM(edit_flags); i++)
384 		if (strstr(ed, goptions[(int) edit_flags[i]].name)) {
385 			change_flag(edit_flags[i], OF_SPECIAL, 1);
386 			return;
387 		}
388 }
389 
390 /* ------------------------------------------------------------------------- */
391 /*           Misc common code for vi/emacs				     */
392 
393 /* Handle the commenting/uncommenting of a line.
394  * Returns:
395  *	1 if a carriage return is indicated (comment added)
396  *	0 if no return (comment removed)
397  *	-1 if there is an error (not enough room for comment chars)
398  * If successful, *lenp contains the new length.  Note: cursor should be
399  * moved to the start of the line after (un)commenting.
400  */
401 int
x_do_comment(buf,bsize,lenp)402 x_do_comment(buf, bsize, lenp)
403 	char *buf;
404 	int bsize;
405 	int *lenp;
406 {
407 	int i, j;
408 	int len = *lenp;
409 
410 	if (len == 0)
411 		return 1; /* somewhat arbitrary - it's what at&t ksh does */
412 
413 	/* Already commented? */
414 	if (buf[0] == '#') {
415 		int saw_nl = 0;
416 
417 		for (j = 0, i = 1; i < len; i++) {
418 			if (!saw_nl || buf[i] != '#')
419 				buf[j++] = buf[i];
420 			saw_nl = buf[i] == '\n';
421 		}
422 		*lenp = j;
423 		return 0;
424 	} else {
425 		int n = 1;
426 
427 		/* See if there's room for the #'s - 1 per \n */
428 		for (i = 0; i < len; i++)
429 			if (buf[i] == '\n')
430 				n++;
431 		if (len + n >= bsize)
432 			return -1;
433 		/* Now add them... */
434 		for (i = len, j = len + n; --i >= 0; ) {
435 			if (buf[i] == '\n')
436 				buf[--j] = '#';
437 			buf[--j] = buf[i];
438 		}
439 		buf[0] = '#';
440 		*lenp += n;
441 		return 1;
442 	}
443 }
444 
445 /* ------------------------------------------------------------------------- */
446 /*           Common file/command completion code for vi/emacs	             */
447 
448 
449 static char	*add_glob ARGS((const char *, int));
450 static void	glob_table ARGS((const char *, XPtrV *, struct table *));
451 static void	glob_path ARGS((int, const char *, XPtrV *, const char *));
452 
453 void
x_print_expansions(nwords,words,is_command)454 x_print_expansions(nwords, words, is_command)
455 	int nwords;
456 	char *const *words;
457 	int is_command;
458 {
459 	int use_copy = 0;
460 	int prefix_len;
461 	XPtrV l;
462 
463 	l.beg = NULL;
464 
465 	/* Check if all matches are in the same directory (in this
466 	 * case, we want to omit the directory name)
467 	 */
468 	if (!is_command
469 	    && (prefix_len = x_longest_prefix(nwords, words)) > 0)
470 	{
471 		int i;
472 
473 		/* Special case for 1 match (prefix is whole word) */
474 		if (nwords == 1)
475 			prefix_len = x_basename(words[0], (char *) 0);
476 		/* Any (non-trailing) slashes in non-common word suffixes? */
477 		for (i = 0; i < nwords; i++)
478 			if (x_basename(words[i] + prefix_len, (char *) 0)
479 							> prefix_len)
480 				break;
481 		/* All in same directory? */
482 		if (i == nwords) {
483 			while (prefix_len > 0
484 			       && !ISDIRSEP(words[0][prefix_len - 1]))
485 				prefix_len--;
486 			use_copy = 1;
487 			XPinit(l, nwords + 1);
488 			for (i = 0; i < nwords; i++)
489 				XPput(l, words[i] + prefix_len);
490 			XPput(l, (char *) 0);
491 		}
492 	}
493 
494 	/*
495 	 * Enumerate expansions
496 	 */
497 	x_putc('\r');
498 	x_putc('\n');
499 	pr_list(use_copy ? (char **) XPptrv(l) : words);
500 
501 	if (use_copy)
502 		XPfree(l); /* not x_free_words() */
503 }
504 
505 /*
506  *  Do file globbing:
507  *	- appends * to (copy of) str if no globbing chars found
508  *	- does expansion, checks for no match, etc.
509  *	- sets *wordsp to array of matching strings
510  *	- returns number of matching strings
511  */
512 static int
x_file_glob(flags,str,slen,wordsp)513 x_file_glob(flags, str, slen, wordsp)
514 	int flags;
515 	const char *str;
516 	int slen;
517 	char ***wordsp;
518 {
519 	char *toglob;
520 	char **words;
521 	int nwords, i, idx, escaping;
522 	XPtrV w;
523 	struct source *s, *sold;
524 
525 	if (slen < 0)
526 		return 0;
527 
528 	toglob = add_glob(str, slen);
529 
530 	/* remove all escaping backward slashes */
531 	escaping = 0;
532 	for(i = 0, idx = 0; toglob[i]; i++) {
533 		if (toglob[i] == '\\' && !escaping) {
534 			escaping = 1;
535 			continue;
536 		}
537 
538 		toglob[idx] = toglob[i];
539 		idx++;
540 		if (escaping) escaping = 0;
541 	}
542 	toglob[idx] = '\0';
543 
544 	/*
545 	 * Convert "foo*" (toglob) to an array of strings (words)
546 	 */
547 	sold = source;
548 	s = pushs(SWSTR, ATEMP);
549 	s->start = s->str = toglob;
550 	source = s;
551 	if (yylex(ONEWORD) != LWORD) {
552 		source = sold;
553 		internal_errorf(0, "fileglob: substitute error");
554 		return 0;
555 	}
556 	source = sold;
557 	XPinit(w, 32);
558 	expand(yylval.cp, &w, DOGLOB|DOTILDE|DOMARKDIRS);
559 	XPput(w, NULL);
560 	words = (char **) XPclose(w);
561 
562 	for (nwords = 0; words[nwords]; nwords++)
563 		;
564 	if (nwords == 1) {
565 		struct stat statb;
566 
567 		/* Check if globbing failed (returned glob pattern),
568 		 * but be careful (E.g. toglob == "ab*" when the file
569 		 * "ab*" exists is not an error).
570 		 * Also, check for empty result - happens if we tried
571 		 * to glob something which evaluated to an empty
572 		 * string (e.g., "$FOO" when there is no FOO, etc).
573 		 */
574 		if ((strcmp(words[0], toglob) == 0
575 		     && stat(words[0], &statb) < 0)
576 		    || words[0][0] == '\0')
577 		{
578 			x_free_words(nwords, words);
579 			words = NULL;
580 			nwords = 0;
581 		}
582 	}
583 	afree(toglob, ATEMP);
584 
585 	if (nwords) {
586 		*wordsp = words;
587 	} else if (words) {
588 		x_free_words(nwords, words);
589 		*wordsp = NULL;
590 	}
591 	return nwords;
592 }
593 
594 /* Data structure used in x_command_glob() */
595 struct path_order_info {
596 	char *word;
597 	int base;
598 	int path_order;
599 };
600 
601 static int path_order_cmp(const void *aa, const void *bb);
602 
603 /* Compare routine used in x_command_glob() */
604 static int
path_order_cmp(aa,bb)605 path_order_cmp(aa, bb)
606 	const void *aa;
607 	const void *bb;
608 {
609 	const struct path_order_info *a = (const struct path_order_info *) aa;
610 	const struct path_order_info *b = (const struct path_order_info *) bb;
611 	int t;
612 
613 	t = FILECMP(a->word + a->base, b->word + b->base);
614 	return t ? t : a->path_order - b->path_order;
615 }
616 
617 static int
x_command_glob(flags,str,slen,wordsp)618 x_command_glob(flags, str, slen, wordsp)
619 	int flags;
620 	const char *str;
621 	int slen;
622 	char ***wordsp;
623 {
624 	char *toglob;
625 	char *pat;
626 	char *fpath;
627 	int nwords;
628 	XPtrV w;
629 	struct block *l;
630 
631 	if (slen < 0)
632 		return 0;
633 
634 	toglob = add_glob(str, slen);
635 
636 	/* Convert "foo*" (toglob) to a pattern for future use */
637 	pat = evalstr(toglob, DOPAT|DOTILDE);
638 	afree(toglob, ATEMP);
639 
640 	XPinit(w, 32);
641 
642 	glob_table(pat, &w, &keywords);
643 	glob_table(pat, &w, &aliases);
644 	glob_table(pat, &w, &builtins);
645 	for (l = e->loc; l; l = l->next)
646 		glob_table(pat, &w, &l->funs);
647 
648 	glob_path(flags, pat, &w, path);
649 	if ((fpath = str_val(global("FPATH"))) != null)
650 		glob_path(flags, pat, &w, fpath);
651 
652 	nwords = XPsize(w);
653 
654 	if (!nwords) {
655 		*wordsp = (char **) 0;
656 		XPfree(w);
657 		return 0;
658 	}
659 
660 	/* Sort entries */
661 	if (flags & XCF_FULLPATH) {
662 		/* Sort by basename, then path order */
663 		struct path_order_info *info;
664 		struct path_order_info *last_info = 0;
665 		char **words = (char **) XPptrv(w);
666 		int path_order = 0;
667 		int i;
668 
669 		info = (struct path_order_info *)
670 			alloc(sizeof(struct path_order_info) * nwords, ATEMP);
671 		for (i = 0; i < nwords; i++) {
672 			info[i].word = words[i];
673 			info[i].base = x_basename(words[i], (char *) 0);
674 			if (!last_info || info[i].base != last_info->base
675 			    || FILENCMP(words[i],
676 					last_info->word, info[i].base) != 0)
677 			{
678 				last_info = &info[i];
679 				path_order++;
680 			}
681 			info[i].path_order = path_order;
682 		}
683 		qsort(info, nwords, sizeof(struct path_order_info),
684 			path_order_cmp);
685 		for (i = 0; i < nwords; i++)
686 			words[i] = info[i].word;
687 		afree((void *) info, ATEMP);
688 	} else {
689 		/* Sort and remove duplicate entries */
690 		char **words = (char **) XPptrv(w);
691 		int i, j;
692 
693 		qsortp(XPptrv(w), (size_t) nwords, xstrcmp);
694 
695 		for (i = j = 0; i < nwords - 1; i++) {
696 			if (strcmp(words[i], words[i + 1]))
697 				words[j++] = words[i];
698 			else
699 				afree(words[i], ATEMP);
700 		}
701 		words[j++] = words[i];
702 		nwords = j;
703 		w.cur = (void **) &words[j];
704 	}
705 
706 	XPput(w, NULL);
707 	*wordsp = (char **) XPclose(w);
708 
709 	return nwords;
710 }
711 
712 #define IS_WORDC(c)	!( ctype(c, C_LEX1) || (c) == '\'' || (c) == '"'  \
713 			    || (c) == '`' || (c) == '=' || (c) == ':' )
714 
715 static int
x_locate_word(buf,buflen,pos,startp,is_commandp)716 x_locate_word(buf, buflen, pos, startp, is_commandp)
717 	const char *buf;
718 	int buflen;
719 	int pos;
720 	int *startp;
721 	int *is_commandp;
722 {
723 	int p;
724 	int start, end;
725 
726 	/* Bad call?  Probably should report error */
727 	if (pos < 0 || pos > buflen) {
728 		*startp = pos;
729 		*is_commandp = 0;
730 		return 0;
731 	}
732 	/* The case where pos == buflen happens to take care of itself... */
733 
734 	start = pos;
735 	/* Keep going backwards to start of word (has effect of allowing
736 	 * one blank after the end of a word)
737 	 */
738 	for (; (start > 0 && IS_WORDC(buf[start - 1]))
739 		|| (start > 1 && buf[start-2] == '\\'); start--)
740 		;
741 	/* Go forwards to end of word */
742 	for (end = start; end < buflen && IS_WORDC(buf[end]); end++) {
743 		if (buf[end] == '\\' && (end+1) < buflen)
744 			end++;
745 	}
746 
747 	if (is_commandp) {
748 		int iscmd;
749 
750 		/* Figure out if this is a command */
751 		for (p = start - 1; p >= 0 && isspace((unsigned char)buf[p]); p--)
752 			;
753 		iscmd = p < 0 || strchr(";|&()`", buf[p]);
754 		if (iscmd) {
755 			/* If command has a /, path, etc. is not searched;
756 			 * only current directory is searched, which is just
757 			 * like file globbing.
758 			 */
759 			for (p = start; p < end; p++)
760 				if (ISDIRSEP(buf[p]))
761 					break;
762 			iscmd = p == end;
763 		}
764 		*is_commandp = iscmd;
765 	}
766 
767 	*startp = start;
768 
769 	return end - start;
770 }
771 
772 int
x_cf_glob(flags,buf,buflen,pos,startp,endp,wordsp,is_commandp)773 x_cf_glob(flags, buf, buflen, pos, startp, endp, wordsp, is_commandp)
774 	int flags;
775 	const char *buf;
776 	int buflen;
777 	int pos;
778 	int *startp;
779 	int *endp;
780 	char ***wordsp;
781 	int *is_commandp;
782 {
783 	int len;
784 	int nwords;
785 	char **words;
786 	int is_command;
787 
788 	len = x_locate_word(buf, buflen, pos, startp, &is_command);
789 	if (!(flags & XCF_COMMAND))
790 		is_command = 0;
791 	/* Don't do command globing on zero length strings - it takes too
792 	 * long and isn't very useful.  File globs are more likely to be
793 	 * useful, so allow these.
794 	 */
795 	if (len == 0 && is_command)
796 		return 0;
797 
798 	nwords = (is_command ? x_command_glob : x_file_glob)(flags,
799 				    buf + *startp, len, &words);
800 	if (nwords == 0) {
801 		*wordsp = (char **) 0;
802 		return 0;
803 	}
804 
805 	if (is_commandp)
806 		*is_commandp = is_command;
807 	*wordsp = words;
808 	*endp = *startp + len;
809 
810 	return nwords;
811 }
812 
813 /* Given a string, copy it and possibly add a '*' to the end.  The
814  * new string is returned.
815  */
816 static char *
add_glob(str,slen)817 add_glob(str, slen)
818 	const char *str;
819 	int slen;
820 {
821 	char *toglob;
822 	char *s;
823 	bool saw_slash = false;
824 
825 	if (slen < 0)
826 		return (char *) 0;
827 
828 	toglob = str_nsave(str, slen + 1, ATEMP); /* + 1 for "*" */
829 	toglob[slen] = '\0';
830 
831 	/*
832 	 * If the pathname contains a wildcard (an unquoted '*',
833 	 * '?', or '['), or a ~username
834 	 * with no trailing slash, then it is globbed based on that
835 	 * value (i.e., without the appended '*').
836 	 */
837 	for (s = toglob; *s; s++) {
838 		if (*s == '\\' && s[1])
839 			s++;
840 		else if (*s == '*' || *s == '[' || *s == '?'
841 			 || (s[1] == '(' /*)*/ && strchr("*+?@!", *s)))
842 			break;
843 		else if (ISDIRSEP(*s))
844 			saw_slash = true;
845 	}
846 	if (!*s && (*toglob != '~' || saw_slash)) {
847 		toglob[slen] = '*';
848 		toglob[slen + 1] = '\0';
849 	}
850 
851 	return toglob;
852 }
853 
854 /*
855  * Find longest common prefix
856  */
857 int
x_longest_prefix(nwords,words)858 x_longest_prefix(nwords, words)
859 	int nwords;
860 	char *const *words;
861 {
862 	int i, j;
863 	int prefix_len;
864 	char *p;
865 
866 	if (nwords <= 0)
867 		return 0;
868 
869 	prefix_len = strlen(words[0]);
870 	for (i = 1; i < nwords; i++)
871 		for (j = 0, p = words[i]; j < prefix_len; j++)
872 			if (FILECHCONV((unsigned char)p[j])
873 			    != FILECHCONV((unsigned char)words[0][j])) {
874 				prefix_len = j;
875 				break;
876 			}
877 	return prefix_len;
878 }
879 
880 void
x_free_words(nwords,words)881 x_free_words(nwords, words)
882 	int nwords;
883 	char **words;
884 {
885 	int i;
886 
887 	for (i = 0; i < nwords; i++)
888 		if (words[i])
889 			afree(words[i], ATEMP);
890 	afree(words, ATEMP);
891 }
892 
893 /* Return the offset of the basename of string s (which ends at se - need not
894  * be null terminated).  Trailing slashes are ignored.  If s is just a slash,
895  * then the offset is 0 (actually, length - 1).
896  *	s		Return
897  *	/etc		1
898  *	/etc/		1
899  *	/etc//		1
900  *	/etc/fo		5
901  *	foo		0
902  *	///		2
903  *			0
904  */
905 int
x_basename(s,se)906 x_basename(s, se)
907 	const char *s;
908 	const char *se;
909 {
910 	const char *p;
911 
912 	if (se == (char *) 0)
913 		se = s + strlen(s);
914 	if (s == se)
915 		return 0;
916 
917 	/* Skip trailing slashes */
918 	for (p = se - 1; p > s && ISDIRSEP(*p); p--)
919 		;
920 	for (; p > s && !ISDIRSEP(*p); p--)
921 		;
922 	if (ISDIRSEP(*p) && p + 1 < se)
923 		p++;
924 
925 	return p - s;
926 }
927 
928 /*
929  *  Apply pattern matching to a table: all table entries that match a pattern
930  * are added to wp.
931  */
932 static void
glob_table(pat,wp,tp)933 glob_table(pat, wp, tp)
934 	const char *pat;
935 	XPtrV *wp;
936 	struct table *tp;
937 {
938 	struct tstate ts;
939 	struct tbl *te;
940 
941 	for (ksh_twalk(&ts, tp); (te = tnext(&ts)); ) {
942 		if (gmatch(te->name, pat, false))
943 			XPput(*wp, str_save(te->name, ATEMP));
944 	}
945 }
946 
947 static void
glob_path(flags,pat,wp,xpath)948 glob_path(flags, pat, wp, xpath)
949 	int flags;
950 	const char *pat;
951 	XPtrV *wp;
952 	const char *xpath;
953 {
954 	const char *sp, *p;
955 	char *xp;
956 	int staterr;
957 	int pathlen;
958 	int patlen;
959 	int oldsize, newsize, i, j;
960 	char **words;
961 	XString xs;
962 
963 	patlen = strlen(pat) + 1;
964 	sp = xpath;
965 	Xinit(xs, xp, patlen + 128, ATEMP);
966 	while (sp) {
967 		xp = Xstring(xs, xp);
968 		if (!(p = strchr(sp, PATHSEP)))
969 			p = sp + strlen(sp);
970 		pathlen = p - sp;
971 		if (pathlen) {
972 			/* Copy sp into xp, stuffing any MAGIC characters
973 			 * on the way
974 			 */
975 			const char *s = sp;
976 
977 			XcheckN(xs, xp, pathlen * 2);
978 			while (s < p) {
979 				if (ISMAGIC(*s))
980 					*xp++ = MAGIC;
981 				*xp++ = *s++;
982 			}
983 			*xp++ = DIRSEP;
984 			pathlen++;
985 		}
986 		sp = p;
987 		XcheckN(xs, xp, patlen);
988 		memcpy(xp, pat, patlen);
989 
990 		oldsize = XPsize(*wp);
991 		glob_str(Xstring(xs, xp), wp, 1); /* mark dirs */
992 		newsize = XPsize(*wp);
993 
994 		/* Check that each match is executable... */
995 		words = (char **) XPptrv(*wp);
996 		for (i = j = oldsize; i < newsize; i++) {
997 			staterr = 0;
998 			if ((search_access(words[i], X_OK, &staterr) >= 0)
999 			    || (staterr == EISDIR)) {
1000 				words[j] = words[i];
1001 				if (!(flags & XCF_FULLPATH))
1002 					memmove(words[j], words[j] + pathlen,
1003 						strlen(words[j] + pathlen) + 1);
1004 				j++;
1005 			} else
1006 				afree(words[i], ATEMP);
1007 		}
1008 		wp->cur = (void **) &words[j];
1009 
1010 		if (!*sp++)
1011 			break;
1012 	}
1013 	Xfree(xs, xp);
1014 }
1015 
1016 /*
1017  * if argument string contains any special characters, they will
1018  * be escaped and the result will be put into edit buffer by
1019  * keybinding-specific function
1020  */
1021 int
x_escape(s,len,putbuf_func)1022 x_escape(s, len, putbuf_func)
1023 	const char *s;
1024 	size_t len;
1025 	int (*putbuf_func) ARGS((const char *, size_t));
1026 {
1027 	size_t add, wlen;
1028 	const char *ifs = str_val(local("IFS", 0));
1029 	int rval=0;
1030 
1031 	for (add = 0, wlen = len; wlen - add > 0; add++) {
1032 		if (strchr("\\$(){}[]?*&;#|<>\"'`", s[add]) || strchr(ifs, s[add])) {
1033 			if (putbuf_func(s, add) != 0) {
1034 				rval = -1;
1035 				break;
1036 			}
1037 
1038 			putbuf_func("\\", 1);
1039 			putbuf_func(&s[add], 1);
1040 
1041 			add++;
1042 			wlen -= add;
1043 			s += add;
1044 			add = -1; /* after the increment it will go to 0 */
1045 		}
1046 	}
1047 	if (wlen > 0 && rval == 0)
1048 		rval = putbuf_func(s, wlen);
1049 
1050 	return (rval);
1051 }
1052 #endif /* EDIT */
1053