1 /* $NetBSD: search.c,v 1.30 2011/10/04 15:27:04 christos Exp $ */
2
3 /*-
4 * Copyright (c) 1992, 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * This code is derived from software contributed to Berkeley by
8 * Christos Zoulas of Cornell University.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 * 3. Neither the name of the University nor the names of its contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 */
34
35 #include "config.h"
36 #if !defined(lint) && !defined(SCCSID)
37 #if 0
38 static char sccsid[] = "@(#)search.c 8.1 (Berkeley) 6/4/93";
39 #else
40 __RCSID("$NetBSD: search.c,v 1.30 2011/10/04 15:27:04 christos Exp $");
41 #endif
42 #endif /* not lint && not SCCSID */
43
44 /*
45 * search.c: History and character search functions
46 */
47 #include <stdlib.h>
48 #if defined(REGEX)
49 #include <regex.h>
50 #elif defined(REGEXP)
51 #include <regexp.h>
52 #endif
53 #include "el.h"
54
55 /*
56 * Adjust cursor in vi mode to include the character under it
57 */
58 #define EL_CURSOR(el) \
59 ((el)->el_line.cursor + (((el)->el_map.type == MAP_VI) && \
60 ((el)->el_map.current == (el)->el_map.alt)))
61
62 /* search_init():
63 * Initialize the search stuff
64 */
65 protected int
search_init(EditLine * el)66 search_init(EditLine *el)
67 {
68
69 el->el_search.patbuf = el_malloc(EL_BUFSIZ *
70 sizeof(*el->el_search.patbuf));
71 if (el->el_search.patbuf == NULL)
72 return -1;
73 el->el_search.patlen = 0;
74 el->el_search.patdir = -1;
75 el->el_search.chacha = '\0';
76 el->el_search.chadir = CHAR_FWD;
77 el->el_search.chatflg = 0;
78 return 0;
79 }
80
81
82 /* search_end():
83 * Initialize the search stuff
84 */
85 protected void
search_end(EditLine * el)86 search_end(EditLine *el)
87 {
88
89 el_free(el->el_search.patbuf);
90 el->el_search.patbuf = NULL;
91 }
92
93
94 #ifdef REGEXP
95 /* regerror():
96 * Handle regular expression errors
97 */
98 public void
99 /*ARGSUSED*/
regerror(const char * msg)100 regerror(const char *msg)
101 {
102 }
103 #endif
104
105
106 /* el_match():
107 * Return if string matches pattern
108 */
109 protected int
el_match(const Char * str,const Char * pat)110 el_match(const Char *str, const Char *pat)
111 {
112 #ifdef WIDECHAR
113 static ct_buffer_t conv;
114 #endif
115 #if defined (REGEX)
116 regex_t re;
117 int rv;
118 #elif defined (REGEXP)
119 regexp *rp;
120 int rv;
121 #else
122 extern char *re_comp(const char *);
123 extern int re_exec(const char *);
124 #endif
125
126 if (Strstr(str, pat) != 0)
127 return 1;
128
129 #if defined(REGEX)
130 if (regcomp(&re, ct_encode_string(pat, &conv), 0) == 0) {
131 rv = regexec(&re, ct_encode_string(str, &conv), (size_t)0, NULL,
132 0) == 0;
133 regfree(&re);
134 } else {
135 rv = 0;
136 }
137 return rv;
138 #elif defined(REGEXP)
139 if ((re = regcomp(ct_encode_string(pat, &conv))) != NULL) {
140 rv = regexec(re, ct_encode_string(str, &conv));
141 el_free(re);
142 } else {
143 rv = 0;
144 }
145 return rv;
146 #else
147 if (re_comp(ct_encode_string(pat, &conv)) != NULL)
148 return 0;
149 else
150 return re_exec(ct_encode_string(str, &conv) == 1);
151 #endif
152 }
153
154
155 /* c_hmatch():
156 * return True if the pattern matches the prefix
157 */
158 protected int
c_hmatch(EditLine * el,const Char * str)159 c_hmatch(EditLine *el, const Char *str)
160 {
161 #ifdef SDEBUG
162 (void) fprintf(el->el_errfile, "match `%s' with `%s'\n",
163 el->el_search.patbuf, str);
164 #endif /* SDEBUG */
165
166 return el_match(str, el->el_search.patbuf);
167 }
168
169
170 /* c_setpat():
171 * Set the history seatch pattern
172 */
173 protected void
c_setpat(EditLine * el)174 c_setpat(EditLine *el)
175 {
176 if (el->el_state.lastcmd != ED_SEARCH_PREV_HISTORY &&
177 el->el_state.lastcmd != ED_SEARCH_NEXT_HISTORY) {
178 el->el_search.patlen =
179 (size_t)(EL_CURSOR(el) - el->el_line.buffer);
180 if (el->el_search.patlen >= EL_BUFSIZ)
181 el->el_search.patlen = EL_BUFSIZ - 1;
182 if (el->el_search.patlen != 0) {
183 (void) Strncpy(el->el_search.patbuf, el->el_line.buffer,
184 el->el_search.patlen);
185 el->el_search.patbuf[el->el_search.patlen] = '\0';
186 } else
187 el->el_search.patlen = Strlen(el->el_search.patbuf);
188 }
189 #ifdef SDEBUG
190 (void) fprintf(el->el_errfile, "\neventno = %d\n",
191 el->el_history.eventno);
192 (void) fprintf(el->el_errfile, "patlen = %d\n", el->el_search.patlen);
193 (void) fprintf(el->el_errfile, "patbuf = \"%s\"\n",
194 el->el_search.patbuf);
195 (void) fprintf(el->el_errfile, "cursor %d lastchar %d\n",
196 EL_CURSOR(el) - el->el_line.buffer,
197 el->el_line.lastchar - el->el_line.buffer);
198 #endif
199 }
200
201
202 /* ce_inc_search():
203 * Emacs incremental search
204 */
205 protected el_action_t
ce_inc_search(EditLine * el,int dir)206 ce_inc_search(EditLine *el, int dir)
207 {
208 static const Char STRfwd[] = {'f', 'w', 'd', '\0'},
209 STRbck[] = {'b', 'c', 'k', '\0'};
210 static Char pchar = ':';/* ':' = normal, '?' = failed */
211 static Char endcmd[2] = {'\0', '\0'};
212 Char ch, *ocursor = el->el_line.cursor, oldpchar = pchar;
213 const Char *cp;
214
215 el_action_t ret = CC_NORM;
216
217 int ohisteventno = el->el_history.eventno;
218 size_t oldpatlen = el->el_search.patlen;
219 int newdir = dir;
220 int done, redo;
221
222 if (el->el_line.lastchar + sizeof(STRfwd) /
223 sizeof(*el->el_line.lastchar) + 2 +
224 el->el_search.patlen >= el->el_line.limit)
225 return CC_ERROR;
226
227 for (;;) {
228
229 if (el->el_search.patlen == 0) { /* first round */
230 pchar = ':';
231 #ifdef ANCHOR
232 #define LEN 2
233 el->el_search.patbuf[el->el_search.patlen++] = '.';
234 el->el_search.patbuf[el->el_search.patlen++] = '*';
235 #else
236 #define LEN 0
237 #endif
238 }
239 done = redo = 0;
240 *el->el_line.lastchar++ = '\n';
241 for (cp = (newdir == ED_SEARCH_PREV_HISTORY) ? STRbck : STRfwd;
242 *cp; *el->el_line.lastchar++ = *cp++)
243 continue;
244 *el->el_line.lastchar++ = pchar;
245 for (cp = &el->el_search.patbuf[LEN];
246 cp < &el->el_search.patbuf[el->el_search.patlen];
247 *el->el_line.lastchar++ = *cp++)
248 continue;
249 *el->el_line.lastchar = '\0';
250 re_refresh(el);
251
252 if (FUN(el,getc)(el, &ch) != 1)
253 return ed_end_of_file(el, 0);
254
255 switch (el->el_map.current[(unsigned char) ch]) {
256 case ED_INSERT:
257 case ED_DIGIT:
258 if (el->el_search.patlen >= EL_BUFSIZ - LEN)
259 terminal_beep(el);
260 else {
261 el->el_search.patbuf[el->el_search.patlen++] =
262 ch;
263 *el->el_line.lastchar++ = ch;
264 *el->el_line.lastchar = '\0';
265 re_refresh(el);
266 }
267 break;
268
269 case EM_INC_SEARCH_NEXT:
270 newdir = ED_SEARCH_NEXT_HISTORY;
271 redo++;
272 break;
273
274 case EM_INC_SEARCH_PREV:
275 newdir = ED_SEARCH_PREV_HISTORY;
276 redo++;
277 break;
278
279 case EM_DELETE_PREV_CHAR:
280 case ED_DELETE_PREV_CHAR:
281 if (el->el_search.patlen > LEN)
282 done++;
283 else
284 terminal_beep(el);
285 break;
286
287 default:
288 switch (ch) {
289 case 0007: /* ^G: Abort */
290 ret = CC_ERROR;
291 done++;
292 break;
293
294 case 0027: /* ^W: Append word */
295 /* No can do if globbing characters in pattern */
296 for (cp = &el->el_search.patbuf[LEN];; cp++)
297 if (cp >= &el->el_search.patbuf[
298 el->el_search.patlen]) {
299 el->el_line.cursor +=
300 el->el_search.patlen - LEN - 1;
301 cp = c__next_word(el->el_line.cursor,
302 el->el_line.lastchar, 1,
303 ce__isword);
304 while (el->el_line.cursor < cp &&
305 *el->el_line.cursor != '\n') {
306 if (el->el_search.patlen >=
307 EL_BUFSIZ - LEN) {
308 terminal_beep(el);
309 break;
310 }
311 el->el_search.patbuf[el->el_search.patlen++] =
312 *el->el_line.cursor;
313 *el->el_line.lastchar++ =
314 *el->el_line.cursor++;
315 }
316 el->el_line.cursor = ocursor;
317 *el->el_line.lastchar = '\0';
318 re_refresh(el);
319 break;
320 } else if (isglob(*cp)) {
321 terminal_beep(el);
322 break;
323 }
324 break;
325
326 default: /* Terminate and execute cmd */
327 endcmd[0] = ch;
328 FUN(el,push)(el, endcmd);
329 /* FALLTHROUGH */
330
331 case 0033: /* ESC: Terminate */
332 ret = CC_REFRESH;
333 done++;
334 break;
335 }
336 break;
337 }
338
339 while (el->el_line.lastchar > el->el_line.buffer &&
340 *el->el_line.lastchar != '\n')
341 *el->el_line.lastchar-- = '\0';
342 *el->el_line.lastchar = '\0';
343
344 if (!done) {
345
346 /* Can't search if unmatched '[' */
347 for (cp = &el->el_search.patbuf[el->el_search.patlen-1],
348 ch = ']';
349 cp >= &el->el_search.patbuf[LEN];
350 cp--)
351 if (*cp == '[' || *cp == ']') {
352 ch = *cp;
353 break;
354 }
355 if (el->el_search.patlen > LEN && ch != '[') {
356 if (redo && newdir == dir) {
357 if (pchar == '?') { /* wrap around */
358 el->el_history.eventno =
359 newdir == ED_SEARCH_PREV_HISTORY ? 0 : 0x7fffffff;
360 if (hist_get(el) == CC_ERROR)
361 /* el->el_history.event
362 * no was fixed by
363 * first call */
364 (void) hist_get(el);
365 el->el_line.cursor = newdir ==
366 ED_SEARCH_PREV_HISTORY ?
367 el->el_line.lastchar :
368 el->el_line.buffer;
369 } else
370 el->el_line.cursor +=
371 newdir ==
372 ED_SEARCH_PREV_HISTORY ?
373 -1 : 1;
374 }
375 #ifdef ANCHOR
376 el->el_search.patbuf[el->el_search.patlen++] =
377 '.';
378 el->el_search.patbuf[el->el_search.patlen++] =
379 '*';
380 #endif
381 el->el_search.patbuf[el->el_search.patlen] =
382 '\0';
383 if (el->el_line.cursor < el->el_line.buffer ||
384 el->el_line.cursor > el->el_line.lastchar ||
385 (ret = ce_search_line(el, newdir))
386 == CC_ERROR) {
387 /* avoid c_setpat */
388 el->el_state.lastcmd =
389 (el_action_t) newdir;
390 ret = (el_action_t)
391 (newdir == ED_SEARCH_PREV_HISTORY ?
392 ed_search_prev_history(el, 0) :
393 ed_search_next_history(el, 0));
394 if (ret != CC_ERROR) {
395 el->el_line.cursor = newdir ==
396 ED_SEARCH_PREV_HISTORY ?
397 el->el_line.lastchar :
398 el->el_line.buffer;
399 (void) ce_search_line(el,
400 newdir);
401 }
402 }
403 el->el_search.patlen -= LEN;
404 el->el_search.patbuf[el->el_search.patlen] =
405 '\0';
406 if (ret == CC_ERROR) {
407 terminal_beep(el);
408 if (el->el_history.eventno !=
409 ohisteventno) {
410 el->el_history.eventno =
411 ohisteventno;
412 if (hist_get(el) == CC_ERROR)
413 return CC_ERROR;
414 }
415 el->el_line.cursor = ocursor;
416 pchar = '?';
417 } else {
418 pchar = ':';
419 }
420 }
421 ret = ce_inc_search(el, newdir);
422
423 if (ret == CC_ERROR && pchar == '?' && oldpchar == ':')
424 /*
425 * break abort of failed search at last
426 * non-failed
427 */
428 ret = CC_NORM;
429
430 }
431 if (ret == CC_NORM || (ret == CC_ERROR && oldpatlen == 0)) {
432 /* restore on normal return or error exit */
433 pchar = oldpchar;
434 el->el_search.patlen = oldpatlen;
435 if (el->el_history.eventno != ohisteventno) {
436 el->el_history.eventno = ohisteventno;
437 if (hist_get(el) == CC_ERROR)
438 return CC_ERROR;
439 }
440 el->el_line.cursor = ocursor;
441 if (ret == CC_ERROR)
442 re_refresh(el);
443 }
444 if (done || ret != CC_NORM)
445 return ret;
446 }
447 }
448
449
450 /* cv_search():
451 * Vi search.
452 */
453 protected el_action_t
cv_search(EditLine * el,int dir)454 cv_search(EditLine *el, int dir)
455 {
456 Char ch;
457 Char tmpbuf[EL_BUFSIZ];
458 ssize_t tmplen;
459
460 #ifdef ANCHOR
461 tmpbuf[0] = '.';
462 tmpbuf[1] = '*';
463 #endif
464 tmplen = LEN;
465
466 el->el_search.patdir = dir;
467
468 tmplen = c_gets(el, &tmpbuf[LEN],
469 dir == ED_SEARCH_PREV_HISTORY ? STR("\n/") : STR("\n?") );
470 if (tmplen == -1)
471 return CC_REFRESH;
472
473 tmplen += LEN;
474 ch = tmpbuf[tmplen];
475 tmpbuf[tmplen] = '\0';
476
477 if (tmplen == LEN) {
478 /*
479 * Use the old pattern, but wild-card it.
480 */
481 if (el->el_search.patlen == 0) {
482 re_refresh(el);
483 return CC_ERROR;
484 }
485 #ifdef ANCHOR
486 if (el->el_search.patbuf[0] != '.' &&
487 el->el_search.patbuf[0] != '*') {
488 (void) Strncpy(tmpbuf, el->el_search.patbuf,
489 sizeof(tmpbuf) / sizeof(*tmpbuf) - 1);
490 el->el_search.patbuf[0] = '.';
491 el->el_search.patbuf[1] = '*';
492 (void) Strncpy(&el->el_search.patbuf[2], tmpbuf,
493 EL_BUFSIZ - 3);
494 el->el_search.patlen++;
495 el->el_search.patbuf[el->el_search.patlen++] = '.';
496 el->el_search.patbuf[el->el_search.patlen++] = '*';
497 el->el_search.patbuf[el->el_search.patlen] = '\0';
498 }
499 #endif
500 } else {
501 #ifdef ANCHOR
502 tmpbuf[tmplen++] = '.';
503 tmpbuf[tmplen++] = '*';
504 #endif
505 tmpbuf[tmplen] = '\0';
506 (void) Strncpy(el->el_search.patbuf, tmpbuf, EL_BUFSIZ - 1);
507 el->el_search.patlen = (size_t)tmplen;
508 }
509 el->el_state.lastcmd = (el_action_t) dir; /* avoid c_setpat */
510 el->el_line.cursor = el->el_line.lastchar = el->el_line.buffer;
511 if ((dir == ED_SEARCH_PREV_HISTORY ? ed_search_prev_history(el, 0) :
512 ed_search_next_history(el, 0)) == CC_ERROR) {
513 re_refresh(el);
514 return CC_ERROR;
515 }
516 if (ch == 0033) {
517 re_refresh(el);
518 return ed_newline(el, 0);
519 }
520 return CC_REFRESH;
521 }
522
523
524 /* ce_search_line():
525 * Look for a pattern inside a line
526 */
527 protected el_action_t
ce_search_line(EditLine * el,int dir)528 ce_search_line(EditLine *el, int dir)
529 {
530 Char *cp = el->el_line.cursor;
531 Char *pattern = el->el_search.patbuf;
532 Char oc, *ocp;
533 #ifdef ANCHOR
534 ocp = &pattern[1];
535 oc = *ocp;
536 *ocp = '^';
537 #else
538 ocp = pattern;
539 oc = *ocp;
540 #endif
541
542 if (dir == ED_SEARCH_PREV_HISTORY) {
543 for (; cp >= el->el_line.buffer; cp--) {
544 if (el_match(cp, ocp)) {
545 *ocp = oc;
546 el->el_line.cursor = cp;
547 return CC_NORM;
548 }
549 }
550 *ocp = oc;
551 return CC_ERROR;
552 } else {
553 for (; *cp != '\0' && cp < el->el_line.limit; cp++) {
554 if (el_match(cp, ocp)) {
555 *ocp = oc;
556 el->el_line.cursor = cp;
557 return CC_NORM;
558 }
559 }
560 *ocp = oc;
561 return CC_ERROR;
562 }
563 }
564
565
566 /* cv_repeat_srch():
567 * Vi repeat search
568 */
569 protected el_action_t
cv_repeat_srch(EditLine * el,Int c)570 cv_repeat_srch(EditLine *el, Int c)
571 {
572
573 #ifdef SDEBUG
574 (void) fprintf(el->el_errfile, "dir %d patlen %d patbuf %s\n",
575 c, el->el_search.patlen, ct_encode_string(el->el_search.patbuf));
576 #endif
577
578 el->el_state.lastcmd = (el_action_t) c; /* Hack to stop c_setpat */
579 el->el_line.lastchar = el->el_line.buffer;
580
581 switch (c) {
582 case ED_SEARCH_NEXT_HISTORY:
583 return ed_search_next_history(el, 0);
584 case ED_SEARCH_PREV_HISTORY:
585 return ed_search_prev_history(el, 0);
586 default:
587 return CC_ERROR;
588 }
589 }
590
591
592 /* cv_csearch():
593 * Vi character search
594 */
595 protected el_action_t
cv_csearch(EditLine * el,int direction,Int ch,int count,int tflag)596 cv_csearch(EditLine *el, int direction, Int ch, int count, int tflag)
597 {
598 Char *cp;
599
600 if (ch == 0)
601 return CC_ERROR;
602
603 if (ch == (Int)-1) {
604 Char c;
605 if (FUN(el,getc)(el, &c) != 1)
606 return ed_end_of_file(el, 0);
607 ch = c;
608 }
609
610 /* Save for ';' and ',' commands */
611 el->el_search.chacha = ch;
612 el->el_search.chadir = direction;
613 el->el_search.chatflg = (char)tflag;
614
615 cp = el->el_line.cursor;
616 while (count--) {
617 if ((Int)*cp == ch)
618 cp += direction;
619 for (;;cp += direction) {
620 if (cp >= el->el_line.lastchar)
621 return CC_ERROR;
622 if (cp < el->el_line.buffer)
623 return CC_ERROR;
624 if ((Int)*cp == ch)
625 break;
626 }
627 }
628
629 if (tflag)
630 cp -= direction;
631
632 el->el_line.cursor = cp;
633
634 if (el->el_chared.c_vcmd.action != NOP) {
635 if (direction > 0)
636 el->el_line.cursor++;
637 cv_delfini(el);
638 return CC_REFRESH;
639 }
640 return CC_CURSOR;
641 }
642