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