lib/libedit/search.c

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