bin/ed/main.c

   1 /* main.c: This file contains the main control and user-interface routines
   2    for the ed line editor. */
   3 /*-
   4  * Copyright (c) 1993 Andrew Moore, Talke Studio.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  *
  28  * @(#) Copyright (c) 1993 Andrew Moore, Talke Studio. All rights reserved.
  29  * @(#)main.c,v 1.1 1994/02/01 00:34:42 alm Exp
  30  * $FreeBSD: head/bin/ed/main.c 292455 2015-12-18 23:05:36Z pfg $
  31  */
  32
  33 /*
  34  * CREDITS
  35  *
  36  *      This program is based on the editor algorithm described in
  37  *      Brian W. Kernighan and P. J. Plauger's book "Software Tools
  38  *      in Pascal," Addison-Wesley, 1981.
  39  *
  40  *      The buffering algorithm is attributed to Rodney Ruddock of
  41  *      the University of Guelph, Guelph, Ontario.
  42  *
  43  *      The cbc.c encryption code is adapted from
  44  *      the bdes program by Matt Bishop of Dartmouth College,
  45  *      Hanover, NH.
  46  *
  47  */
  48
  49 #include <sys/types.h>
  50
  51 #include <sys/ioctl.h>
  52 #include <sys/wait.h>
  53 #include <ctype.h>
  54 #include <locale.h>
  55 #include <pwd.h>
  56 #include <setjmp.h>
  57
  58 #include "ed.h"
  59
  60
  61 static sigjmp_buf env;
  62
  63 /* static buffers */
  64 char stdinbuf[1];               /* stdin buffer */
  65 static char *shcmd;             /* shell command buffer */
  66 static int shcmdsz;             /* shell command buffer size */
  67 static int shcmdi;              /* shell command buffer index */
  68 char *ibuf;                     /* ed command-line buffer */
  69 int ibufsz;                     /* ed command-line buffer size */
  70 char *ibufp;                    /* pointer to ed command-line buffer */
  71
  72 /* global flags */
  73 int des = 0;                    /* if set, use crypt(3) for i/o */
  74 static int garrulous = 0;       /* if set, print all error messages */
  75 int isbinary;                   /* if set, buffer contains ASCII NULs */
  76 int isglobal;                   /* if set, doing a global command */
  77 int modified;                   /* if set, buffer modified since last write */
  78 int mutex = 0;                  /* if set, signals set "sigflags" */
  79 static int red = 0;             /* if set, restrict shell/directory access */
  80 int scripted = 0;               /* if set, suppress diagnostics */
  81 int sigflags = 0;               /* if set, signals received while mutex set */
  82 static int sigactive = 0;       /* if set, signal handlers are enabled */
  83
  84 static char old_filename[PATH_MAX] = ""; /* default filename */
  85 long current_addr;              /* current address in editor buffer */
  86 long addr_last;                 /* last address in editor buffer */
  87 int lineno;                     /* script line number */
  88 static const char *prompt;      /* command-line prompt */
  89 static const char *dps = "*";   /* default command-line prompt */
  90
  91 static const char *usage = "usage: %s [-] [-sx] [-p string] [file]\n";
  92
  93 /* ed: line editor */
  94 int
  95 main(volatile int argc, char ** volatile argv)
  96 {
  97         int c, n;
  98         long status = 0;
  99
 100         setlocale(LC_ALL, "");
 101
 102         red = (n = strlen(argv[0])) > 2 && argv[0][n - 3] == 'r';
 103 top:
 104         while ((c = getopt(argc, argv, "p:sx")) != -1)
 105                 switch(c) {
 106                 case 'p':                               /* set prompt */
 107                         prompt = optarg;
 108                         break;
 109                 case 's':                               /* run script */
 110                         scripted = 1;
 111                         break;
 112                 case 'x':                               /* use crypt */
 113 #ifdef DES
 114                         des = get_keyword();
 115 #else
 116                         fprintf(stderr, "crypt unavailable\n?\n");
 117 #endif
 118                         break;
 119
 120                 default:
 121                         fprintf(stderr, usage, red ? "red" : "ed");
 122                         exit(1);
 123                 }
 124         argv += optind;
 125         argc -= optind;
 126         if (argc && **argv == '-') {
 127                 scripted = 1;
 128                 if (argc > 1) {
 129                         optind = 1;
 130                         goto top;
 131                 }
 132                 argv++;
 133                 argc--;
 134         }
 135         /* assert: reliable signals! */
 136 #ifdef SIGWINCH
 137         handle_winch(SIGWINCH);
 138         if (isatty(0)) signal(SIGWINCH, handle_winch);
 139 #endif
 140         signal(SIGHUP, signal_hup);
 141         signal(SIGQUIT, SIG_IGN);
 142         signal(SIGINT, signal_int);
 143         if ((status = sigsetjmp(env, 1))) {
 144                 fputs("\n?\n", stderr);
 145                 errmsg = "interrupt";
 146         } else {
 147                 init_buffers();
 148                 sigactive = 1;                  /* enable signal handlers */
 149                 if (argc && **argv && is_legal_filename(*argv)) {
 150                         if (read_file(*argv, 0) < 0 && !isatty(0))
 151                                 quit(2);
 152                         else if (**argv != '!')
 153                                 if (strlcpy(old_filename, *argv, sizeof(old_filename))
 154                                     >= sizeof(old_filename))
 155                                         quit(2);
 156                 } else if (argc) {
 157                         fputs("?\n", stderr);
 158                         if (**argv == '\0')
 159                                 errmsg = "invalid filename";
 160                         if (!isatty(0))
 161                                 quit(2);
 162                 }
 163         }
 164         for (;;) {
 165                 if (status < 0 && garrulous)
 166                         fprintf(stderr, "%s\n", errmsg);
 167                 if (prompt) {
 168                         printf("%s", prompt);
 169                         fflush(stdout);
 170                 }
 171                 if ((n = get_tty_line()) < 0) {
 172                         status = ERR;
 173                         continue;
 174                 } else if (n == 0) {
 175                         if (modified && !scripted) {
 176                                 fputs("?\n", stderr);
 177                                 errmsg = "warning: file modified";
 178                                 if (!isatty(0)) {
 179                                         if (garrulous)
 180                                                 fprintf(stderr,
 181                                                     "script, line %d: %s\n",
 182                                                     lineno, errmsg);
 183                                         quit(2);
 184                                 }
 185                                 clearerr(stdin);
 186                                 modified = 0;
 187                                 status = EMOD;
 188                                 continue;
 189                         } else
 190                                 quit(0);
 191                 } else if (ibuf[n - 1] != '\n') {
 192                         /* discard line */
 193                         errmsg = "unexpected end-of-file";
 194                         clearerr(stdin);
 195                         status = ERR;
 196                         continue;
 197                 }
 198                 isglobal = 0;
 199                 if ((status = extract_addr_range()) >= 0 &&
 200                     (status = exec_command()) >= 0)
 201                         if (!status ||
 202                             (status = display_lines(current_addr, current_addr,
 203                                 status)) >= 0)
 204                                 continue;
 205                 switch (status) {
 206                 case EOF:
 207                         quit(0);
 208                 case EMOD:
 209                         modified = 0;
 210                         fputs("?\n", stderr);           /* give warning */
 211                         errmsg = "warning: file modified";
 212                         if (!isatty(0)) {
 213                                 if (garrulous)
 214                                         fprintf(stderr, "script, line %d: %s\n",
 215                                             lineno, errmsg);
 216                                 quit(2);
 217                         }
 218                         break;
 219                 case FATAL:
 220                         if (!isatty(0)) {
 221                                 if (garrulous)
 222                                         fprintf(stderr, "script, line %d: %s\n",
 223                                             lineno, errmsg);
 224                         } else if (garrulous)
 225                                 fprintf(stderr, "%s\n", errmsg);
 226                         quit(3);
 227                 default:
 228                         fputs("?\n", stderr);
 229                         if (!isatty(0)) {
 230                                 if (garrulous)
 231                                         fprintf(stderr, "script, line %d: %s\n",
 232                                             lineno, errmsg);
 233                                 quit(2);
 234                         }
 235                         break;
 236                 }
 237         }
 238         /*NOTREACHED*/
 239 }
 240
 241 long first_addr, second_addr;
 242 static long addr_cnt;
 243
 244 /* extract_addr_range: get line addresses from the command buffer until an
 245    illegal address is seen; return status */
 246 int
 247 extract_addr_range(void)
 248 {
 249         long addr;
 250
 251         addr_cnt = 0;
 252         first_addr = second_addr = current_addr;
 253         while ((addr = next_addr()) >= 0) {
 254                 addr_cnt++;
 255                 first_addr = second_addr;
 256                 second_addr = addr;
 257                 if (*ibufp != ',' && *ibufp != ';')
 258                         break;
 259                 else if (*ibufp++ == ';')
 260                         current_addr = addr;
 261         }
 262         if ((addr_cnt = min(addr_cnt, 2)) == 1 || second_addr != addr)
 263                 first_addr = second_addr;
 264         return (addr == ERR) ? ERR : 0;
 265 }
 266
 267
 268 #define SKIP_BLANKS() while (isspace((unsigned char)*ibufp) && *ibufp != '\n') ibufp++
 269
 270 #define MUST_BE_FIRST() do {                                    \
 271         if (!first) {                                           \
 272                 errmsg = "invalid address";                     \
 273                 return ERR;                                     \
 274         }                                                       \
 275 } while (0)
 276
 277 /*  next_addr: return the next line address in the command buffer */
 278 long
 279 next_addr(void)
 280 {
 281         const char *hd;
 282         long addr = current_addr;
 283         long n;
 284         int first = 1;
 285         int c;
 286
 287         SKIP_BLANKS();
 288         for (hd = ibufp;; first = 0)
 289                 switch (c = *ibufp) {
 290                 case '+':
 291                 case '\t':
 292                 case ' ':
 293                 case '-':
 294                 case '^':
 295                         ibufp++;
 296                         SKIP_BLANKS();
 297                         if (isdigit((unsigned char)*ibufp)) {
 298                                 STRTOL(n, ibufp);
 299                                 addr += (c == '-' || c == '^') ? -n : n;
 300                         } else if (!isspace((unsigned char)c))
 301                                 addr += (c == '-' || c == '^') ? -1 : 1;
 302                         break;
 303                 case '0': case '1': case '2':
 304                 case '3': case '4': case '5':
 305                 case '6': case '7': case '8': case '9':
 306                         MUST_BE_FIRST();
 307                         STRTOL(addr, ibufp);
 308                         break;
 309                 case '.':
 310                 case '$':
 311                         MUST_BE_FIRST();
 312                         ibufp++;
 313                         addr = (c == '.') ? current_addr : addr_last;
 314                         break;
 315                 case '/':
 316                 case '?':
 317                         MUST_BE_FIRST();
 318                         if ((addr = get_matching_node_addr(
 319                             get_compiled_pattern(), c == '/')) < 0)
 320                                 return ERR;
 321                         else if (c == *ibufp)
 322                                 ibufp++;
 323                         break;
 324                 case '\'':
 325                         MUST_BE_FIRST();
 326                         ibufp++;
 327                         if ((addr = get_marked_node_addr(*ibufp++)) < 0)
 328                                 return ERR;
 329                         break;
 330                 case '%':
 331                 case ',':
 332                 case ';':
 333                         if (first) {
 334                                 ibufp++;
 335                                 addr_cnt++;
 336                                 second_addr = (c == ';') ? current_addr : 1;
 337                                 addr = addr_last;
 338                                 break;
 339                         }
 340                         /* FALLTHROUGH */
 341                 default:
 342                         if (ibufp == hd)
 343                                 return EOF;
 344                         else if (addr < 0 || addr_last < addr) {
 345                                 errmsg = "invalid address";
 346                                 return ERR;
 347                         } else
 348                                 return addr;
 349                 }
 350         /* NOTREACHED */
 351 }
 352
 353
 354 #ifdef BACKWARDS
 355 /* GET_THIRD_ADDR: get a legal address from the command buffer */
 356 #define GET_THIRD_ADDR(addr) \
 357 { \
 358         long ol1, ol2; \
 359 \
 360         ol1 = first_addr, ol2 = second_addr; \
 361         if (extract_addr_range() < 0) \
 362                 return ERR; \
 363         else if (addr_cnt == 0) { \
 364                 errmsg = "destination expected"; \
 365                 return ERR; \
 366         } else if (second_addr < 0 || addr_last < second_addr) { \
 367                 errmsg = "invalid address"; \
 368                 return ERR; \
 369         } \
 370         addr = second_addr; \
 371         first_addr = ol1, second_addr = ol2; \
 372 }
 373 #else   /* BACKWARDS */
 374 /* GET_THIRD_ADDR: get a legal address from the command buffer */
 375 #define GET_THIRD_ADDR(addr) \
 376 { \
 377         long ol1, ol2; \
 378 \
 379         ol1 = first_addr, ol2 = second_addr; \
 380         if (extract_addr_range() < 0) \
 381                 return ERR; \
 382         if (second_addr < 0 || addr_last < second_addr) { \
 383                 errmsg = "invalid address"; \
 384                 return ERR; \
 385         } \
 386         addr = second_addr; \
 387         first_addr = ol1, second_addr = ol2; \
 388 }
 389 #endif
 390
 391
 392 /* GET_COMMAND_SUFFIX: verify the command suffix in the command buffer */
 393 #define GET_COMMAND_SUFFIX() { \
 394         int done = 0; \
 395         do { \
 396                 switch(*ibufp) { \
 397                 case 'p': \
 398                         gflag |= GPR, ibufp++; \
 399                         break; \
 400                 case 'l': \
 401                         gflag |= GLS, ibufp++; \
 402                         break; \
 403                 case 'n': \
 404                         gflag |= GNP, ibufp++; \
 405                         break; \
 406                 default: \
 407                         done++; \
 408                 } \
 409         } while (!done); \
 410         if (*ibufp++ != '\n') { \
 411                 errmsg = "invalid command suffix"; \
 412                 return ERR; \
 413         } \
 414 }
 415
 416
 417 /* sflags */
 418 #define SGG 001         /* complement previous global substitute suffix */
 419 #define SGP 002         /* complement previous print suffix */
 420 #define SGR 004         /* use last regex instead of last pat */
 421 #define SGF 010         /* repeat last substitution */
 422
 423 int patlock = 0;        /* if set, pattern not freed by get_compiled_pattern() */
 424
 425 long rows = 22;         /* scroll length: ws_row - 2 */
 426
 427 /* exec_command: execute the next command in command buffer; return print
 428    request, if any */
 429 int
 430 exec_command(void)
 431 {
 432         static pattern_t *pat = NULL;
 433         static int sgflag = 0;
 434         static long sgnum = 0;
 435
 436         pattern_t *tpat;
 437         char *fnp;
 438         int gflag = 0;
 439         int sflags = 0;
 440         long addr = 0;
 441         int n = 0;
 442         int c;
 443
 444         SKIP_BLANKS();
 445         switch(c = *ibufp++) {
 446         case 'a':
 447                 GET_COMMAND_SUFFIX();
 448                 if (!isglobal) clear_undo_stack();
 449                 if (append_lines(second_addr) < 0)
 450                         return ERR;
 451                 break;
 452         case 'c':
 453                 if (check_addr_range(current_addr, current_addr) < 0)
 454                         return ERR;
 455                 GET_COMMAND_SUFFIX();
 456                 if (!isglobal) clear_undo_stack();
 457                 if (delete_lines(first_addr, second_addr) < 0 ||
 458                     append_lines(current_addr) < 0)
 459                         return ERR;
 460                 break;
 461         case 'd':
 462                 if (check_addr_range(current_addr, current_addr) < 0)
 463                         return ERR;
 464                 GET_COMMAND_SUFFIX();
 465                 if (!isglobal) clear_undo_stack();
 466                 if (delete_lines(first_addr, second_addr) < 0)
 467                         return ERR;
 468                 else if ((addr = INC_MOD(current_addr, addr_last)) != 0)
 469                         current_addr = addr;
 470                 break;
 471         case 'e':
 472                 if (modified && !scripted)
 473                         return EMOD;
 474                 /* FALLTHROUGH */
 475         case 'E':
 476                 if (addr_cnt > 0) {
 477                         errmsg = "unexpected address";
 478                         return ERR;
 479                 } else if (!isspace((unsigned char)*ibufp)) {
 480                         errmsg = "unexpected command suffix";
 481                         return ERR;
 482                 } else if ((fnp = get_filename()) == NULL)
 483                         return ERR;
 484                 GET_COMMAND_SUFFIX();
 485                 if (delete_lines(1, addr_last) < 0)
 486                         return ERR;
 487                 clear_undo_stack();
 488                 if (close_sbuf() < 0)
 489                         return ERR;
 490                 else if (open_sbuf() < 0)
 491                         return FATAL;
 492                 if (*fnp && *fnp != '!')
 493                          strlcpy(old_filename, fnp, PATH_MAX);
 494 #ifdef BACKWARDS
 495                 if (*fnp == '\0' && *old_filename == '\0') {
 496                         errmsg = "no current filename";
 497                         return ERR;
 498                 }
 499 #endif
 500                 if (read_file(*fnp ? fnp : old_filename, 0) < 0)
 501                         return ERR;
 502                 clear_undo_stack();
 503                 modified = 0;
 504                 u_current_addr = u_addr_last = -1;
 505                 break;
 506         case 'f':
 507                 if (addr_cnt > 0) {
 508                         errmsg = "unexpected address";
 509                         return ERR;
 510                 } else if (!isspace((unsigned char)*ibufp)) {
 511                         errmsg = "unexpected command suffix";
 512                         return ERR;
 513                 } else if ((fnp = get_filename()) == NULL)
 514                         return ERR;
 515                 else if (*fnp == '!') {
 516                         errmsg = "invalid redirection";
 517                         return ERR;
 518                 }
 519                 GET_COMMAND_SUFFIX();
 520                 if (*fnp)
 521                         strlcpy(old_filename, fnp, PATH_MAX);
 522                 printf("%s\n", strip_escapes(old_filename));
 523                 break;
 524         case 'g':
 525         case 'v':
 526         case 'G':
 527         case 'V':
 528                 if (isglobal) {
 529                         errmsg = "cannot nest global commands";
 530                         return ERR;
 531                 } else if (check_addr_range(1, addr_last) < 0)
 532                         return ERR;
 533                 else if (build_active_list(c == 'g' || c == 'G') < 0)
 534                         return ERR;
 535                 else if ((n = (c == 'G' || c == 'V')))
 536                         GET_COMMAND_SUFFIX();
 537                 isglobal++;
 538                 if (exec_global(n, gflag) < 0)
 539                         return ERR;
 540                 break;
 541         case 'h':
 542                 if (addr_cnt > 0) {
 543                         errmsg = "unexpected address";
 544                         return ERR;
 545                 }
 546                 GET_COMMAND_SUFFIX();
 547                 if (*errmsg) fprintf(stderr, "%s\n", errmsg);
 548                 break;
 549         case 'H':
 550                 if (addr_cnt > 0) {
 551                         errmsg = "unexpected address";
 552                         return ERR;
 553                 }
 554                 GET_COMMAND_SUFFIX();
 555                 if ((garrulous = 1 - garrulous) && *errmsg)
 556                         fprintf(stderr, "%s\n", errmsg);
 557                 break;
 558         case 'i':
 559                 if (second_addr == 0) {
 560                         errmsg = "invalid address";
 561                         return ERR;
 562                 }
 563                 GET_COMMAND_SUFFIX();
 564                 if (!isglobal) clear_undo_stack();
 565                 if (append_lines(second_addr - 1) < 0)
 566                         return ERR;
 567                 break;
 568         case 'j':
 569                 if (check_addr_range(current_addr, current_addr + 1) < 0)
 570                         return ERR;
 571                 GET_COMMAND_SUFFIX();
 572                 if (!isglobal) clear_undo_stack();
 573                 if (first_addr != second_addr &&
 574                     join_lines(first_addr, second_addr) < 0)
 575                         return ERR;
 576                 break;
 577         case 'k':
 578                 c = *ibufp++;
 579                 if (second_addr == 0) {
 580                         errmsg = "invalid address";
 581                         return ERR;
 582                 }
 583                 GET_COMMAND_SUFFIX();
 584                 if (mark_line_node(get_addressed_line_node(second_addr), c) < 0)
 585                         return ERR;
 586                 break;
 587         case 'l':
 588                 if (check_addr_range(current_addr, current_addr) < 0)
 589                         return ERR;
 590                 GET_COMMAND_SUFFIX();
 591                 if (display_lines(first_addr, second_addr, gflag | GLS) < 0)
 592                         return ERR;
 593                 gflag = 0;
 594                 break;
 595         case 'm':
 596                 if (check_addr_range(current_addr, current_addr) < 0)
 597                         return ERR;
 598                 GET_THIRD_ADDR(addr);
 599                 if (first_addr <= addr && addr < second_addr) {
 600                         errmsg = "invalid destination";
 601                         return ERR;
 602                 }
 603                 GET_COMMAND_SUFFIX();
 604                 if (!isglobal) clear_undo_stack();
 605                 if (move_lines(addr) < 0)
 606                         return ERR;
 607                 break;
 608         case 'n':
 609                 if (check_addr_range(current_addr, current_addr) < 0)
 610                         return ERR;
 611                 GET_COMMAND_SUFFIX();
 612                 if (display_lines(first_addr, second_addr, gflag | GNP) < 0)
 613                         return ERR;
 614                 gflag = 0;
 615                 break;
 616         case 'p':
 617                 if (check_addr_range(current_addr, current_addr) < 0)
 618                         return ERR;
 619                 GET_COMMAND_SUFFIX();
 620                 if (display_lines(first_addr, second_addr, gflag | GPR) < 0)
 621                         return ERR;
 622                 gflag = 0;
 623                 break;
 624         case 'P':
 625                 if (addr_cnt > 0) {
 626                         errmsg = "unexpected address";
 627                         return ERR;
 628                 }
 629                 GET_COMMAND_SUFFIX();
 630                 prompt = prompt ? NULL : optarg ? optarg : dps;
 631                 break;
 632         case 'q':
 633         case 'Q':
 634                 if (addr_cnt > 0) {
 635                         errmsg = "unexpected address";
 636                         return ERR;
 637                 }
 638                 GET_COMMAND_SUFFIX();
 639                 gflag =  (modified && !scripted && c == 'q') ? EMOD : EOF;
 640                 break;
 641         case 'r':
 642                 if (!isspace((unsigned char)*ibufp)) {
 643                         errmsg = "unexpected command suffix";
 644                         return ERR;
 645                 } else if (addr_cnt == 0)
 646                         second_addr = addr_last;
 647                 if ((fnp = get_filename()) == NULL)
 648                         return ERR;
 649                 GET_COMMAND_SUFFIX();
 650                 if (!isglobal) clear_undo_stack();
 651                 if (*old_filename == '\0' && *fnp != '!')
 652                         strlcpy(old_filename, fnp, PATH_MAX);
 653 #ifdef BACKWARDS
 654                 if (*fnp == '\0' && *old_filename == '\0') {
 655                         errmsg = "no current filename";
 656                         return ERR;
 657                 }
 658 #endif
 659                 if ((addr = read_file(*fnp ? fnp : old_filename, second_addr)) < 0)
 660                         return ERR;
 661                 else if (addr && addr != addr_last)
 662                         modified = 1;
 663                 break;
 664         case 's':
 665                 do {
 666                         switch(*ibufp) {
 667                         case '\n':
 668                                 sflags |=SGF;
 669                                 break;
 670                         case 'g':
 671                                 sflags |= SGG;
 672                                 ibufp++;
 673                                 break;
 674                         case 'p':
 675                                 sflags |= SGP;
 676                                 ibufp++;
 677                                 break;
 678                         case 'r':
 679                                 sflags |= SGR;
 680                                 ibufp++;
 681                                 break;
 682                         case '0': case '1': case '2': case '3': case '4':
 683                         case '5': case '6': case '7': case '8': case '9':
 684                                 STRTOL(sgnum, ibufp);
 685                                 sflags |= SGF;
 686                                 sgflag &= ~GSG;         /* override GSG */
 687                                 break;
 688                         default:
 689                                 if (sflags) {
 690                                         errmsg = "invalid command suffix";
 691                                         return ERR;
 692                                 }
 693                         }
 694                 } while (sflags && *ibufp != '\n');
 695                 if (sflags && !pat) {
 696                         errmsg = "no previous substitution";
 697                         return ERR;
 698                 } else if (sflags & SGG)
 699                         sgnum = 0;              /* override numeric arg */
 700                 if (*ibufp != '\n' && *(ibufp + 1) == '\n') {
 701                         errmsg = "invalid pattern delimiter";
 702                         return ERR;
 703                 }
 704                 tpat = pat;
 705                 SPL1();
 706                 if ((!sflags || (sflags & SGR)) &&
 707                     (tpat = get_compiled_pattern()) == NULL) {
 708                         SPL0();
 709                         return ERR;
 710                 } else if (tpat != pat) {
 711                         if (pat) {
 712                                 regfree(pat);
 713                                 free(pat);
 714                         }
 715                         pat = tpat;
 716                         patlock = 1;            /* reserve pattern */
 717                 }
 718                 SPL0();
 719                 if (!sflags && extract_subst_tail(&sgflag, &sgnum) < 0)
 720                         return ERR;
 721                 else if (isglobal)
 722                         sgflag |= GLB;
 723                 else
 724                         sgflag &= ~GLB;
 725                 if (sflags & SGG)
 726                         sgflag ^= GSG;
 727                 if (sflags & SGP)
 728                         sgflag ^= GPR, sgflag &= ~(GLS | GNP);
 729                 do {
 730                         switch(*ibufp) {
 731                         case 'p':
 732                                 sgflag |= GPR, ibufp++;
 733                                 break;
 734                         case 'l':
 735                                 sgflag |= GLS, ibufp++;
 736                                 break;
 737                         case 'n':
 738                                 sgflag |= GNP, ibufp++;
 739                                 break;
 740                         default:
 741                                 n++;
 742                         }
 743                 } while (!n);
 744                 if (check_addr_range(current_addr, current_addr) < 0)
 745                         return ERR;
 746                 GET_COMMAND_SUFFIX();
 747                 if (!isglobal) clear_undo_stack();
 748                 if (search_and_replace(pat, sgflag, sgnum) < 0)
 749                         return ERR;
 750                 break;
 751         case 't':
 752                 if (check_addr_range(current_addr, current_addr) < 0)
 753                         return ERR;
 754                 GET_THIRD_ADDR(addr);
 755                 GET_COMMAND_SUFFIX();
 756                 if (!isglobal) clear_undo_stack();
 757                 if (copy_lines(addr) < 0)
 758                         return ERR;
 759                 break;
 760         case 'u':
 761                 if (addr_cnt > 0) {
 762                         errmsg = "unexpected address";
 763                         return ERR;
 764                 }
 765                 GET_COMMAND_SUFFIX();
 766                 if (pop_undo_stack() < 0)
 767                         return ERR;
 768                 break;
 769         case 'w':
 770         case 'W':
 771                 if ((n = *ibufp) == 'q' || n == 'Q') {
 772                         gflag = EOF;
 773                         ibufp++;
 774                 }
 775                 if (!isspace((unsigned char)*ibufp)) {
 776                         errmsg = "unexpected command suffix";
 777                         return ERR;
 778                 } else if ((fnp = get_filename()) == NULL)
 779                         return ERR;
 780                 if (addr_cnt == 0 && !addr_last)
 781                         first_addr = second_addr = 0;
 782                 else if (check_addr_range(1, addr_last) < 0)
 783                         return ERR;
 784                 GET_COMMAND_SUFFIX();
 785                 if (*old_filename == '\0' && *fnp != '!')
 786                         strlcpy(old_filename, fnp, PATH_MAX);
 787 #ifdef BACKWARDS
 788                 if (*fnp == '\0' && *old_filename == '\0') {
 789                         errmsg = "no current filename";
 790                         return ERR;
 791                 }
 792 #endif
 793                 if ((addr = write_file(*fnp ? fnp : old_filename,
 794                     (c == 'W') ? "a" : "w", first_addr, second_addr)) < 0)
 795                         return ERR;
 796                 else if (addr == addr_last)
 797                         modified = 0;
 798                 else if (modified && !scripted && n == 'q')
 799                         gflag = EMOD;
 800                 break;
 801         case 'x':
 802                 if (addr_cnt > 0) {
 803                         errmsg = "unexpected address";
 804                         return ERR;
 805                 }
 806                 GET_COMMAND_SUFFIX();
 807 #ifdef DES
 808                 des = get_keyword();
 809                 break;
 810 #else
 811                 errmsg = "crypt unavailable";
 812                 return ERR;
 813 #endif
 814         case 'z':
 815 #ifdef BACKWARDS
 816                 if (check_addr_range(first_addr = 1, current_addr + 1) < 0)
 817 #else
 818                 if (check_addr_range(first_addr = 1, current_addr + !isglobal) < 0)
 819 #endif
 820                         return ERR;
 821                 else if ('0' < *ibufp && *ibufp <= '9')
 822                         STRTOL(rows, ibufp);
 823                 GET_COMMAND_SUFFIX();
 824                 if (display_lines(second_addr, min(addr_last,
 825                     second_addr + rows), gflag) < 0)
 826                         return ERR;
 827                 gflag = 0;
 828                 break;
 829         case '=':
 830                 GET_COMMAND_SUFFIX();
 831                 printf("%ld\n", addr_cnt ? second_addr : addr_last);
 832                 break;
 833         case '!':
 834                 if (addr_cnt > 0) {
 835                         errmsg = "unexpected address";
 836                         return ERR;
 837                 } else if ((sflags = get_shell_command()) < 0)
 838                         return ERR;
 839                 GET_COMMAND_SUFFIX();
 840                 if (sflags) printf("%s\n", shcmd + 1);
 841                 system(shcmd + 1);
 842                 if (!scripted) printf("!\n");
 843                 break;
 844         case '\n':
 845 #ifdef BACKWARDS
 846                 if (check_addr_range(first_addr = 1, current_addr + 1) < 0
 847 #else
 848                 if (check_addr_range(first_addr = 1, current_addr + !isglobal) < 0
 849 #endif
 850                  || display_lines(second_addr, second_addr, 0) < 0)
 851                         return ERR;
 852                 break;
 853         default:
 854                 errmsg = "unknown command";
 855                 return ERR;
 856         }
 857         return gflag;
 858 }
 859
 860
 861 /* check_addr_range: return status of address range check */
 862 int
 863 check_addr_range(long n, long m)
 864 {
 865         if (addr_cnt == 0) {
 866                 first_addr = n;
 867                 second_addr = m;
 868         }
 869         if (first_addr > second_addr || 1 > first_addr ||
 870             second_addr > addr_last) {
 871                 errmsg = "invalid address";
 872                 return ERR;
 873         }
 874         return 0;
 875 }
 876
 877
 878 /* get_matching_node_addr: return the address of the next line matching a
 879    pattern in a given direction.  wrap around begin/end of editor buffer if
 880    necessary */
 881 long
 882 get_matching_node_addr(pattern_t *pat, int dir)
 883 {
 884         char *s;
 885         long n = current_addr;
 886         line_t *lp;
 887
 888         if (!pat) return ERR;
 889         do {
 890                if ((n = dir ? INC_MOD(n, addr_last) : DEC_MOD(n, addr_last))) {
 891                         lp = get_addressed_line_node(n);
 892                         if ((s = get_sbuf_line(lp)) == NULL)
 893                                 return ERR;
 894                         if (isbinary)
 895                                 NUL_TO_NEWLINE(s, lp->len);
 896                         if (!regexec(pat, s, 0, NULL, 0))
 897                                 return n;
 898                }
 899         } while (n != current_addr);
 900         errmsg = "no match";
 901         return  ERR;
 902 }
 903
 904
 905 /* get_filename: return pointer to copy of filename in the command buffer */
 906 char *
 907 get_filename(void)
 908 {
 909         static char *file = NULL;
 910         static int filesz = 0;
 911
 912         int n;
 913
 914         if (*ibufp != '\n') {
 915                 SKIP_BLANKS();
 916                 if (*ibufp == '\n') {
 917                         errmsg = "invalid filename";
 918                         return NULL;
 919                 } else if ((ibufp = get_extended_line(&n, 1)) == NULL)
 920                         return NULL;
 921                 else if (*ibufp == '!') {
 922                         ibufp++;
 923                         if ((n = get_shell_command()) < 0)
 924                                 return NULL;
 925                         if (n)
 926                                 printf("%s\n", shcmd + 1);
 927                         return shcmd;
 928                 } else if (n > PATH_MAX - 1) {
 929                         errmsg = "filename too long";
 930                         return  NULL;
 931                 }
 932         }
 933 #ifndef BACKWARDS
 934         else if (*old_filename == '\0') {
 935                 errmsg = "no current filename";
 936                 return  NULL;
 937         }
 938 #endif
 939         REALLOC(file, filesz, PATH_MAX, NULL);
 940         for (n = 0; *ibufp != '\n';)
 941                 file[n++] = *ibufp++;
 942         file[n] = '\0';
 943         return is_legal_filename(file) ? file : NULL;
 944 }
 945
 946
 947 /* get_shell_command: read a shell command from stdin; return substitution
 948    status */
 949 int
 950 get_shell_command(void)
 951 {
 952         static char *buf = NULL;
 953         static int n = 0;
 954
 955         char *s;                        /* substitution char pointer */
 956         int i = 0;
 957         int j = 0;
 958
 959         if (red) {
 960                 errmsg = "shell access restricted";
 961                 return ERR;
 962         } else if ((s = ibufp = get_extended_line(&j, 1)) == NULL)
 963                 return ERR;
 964         REALLOC(buf, n, j + 1, ERR);
 965         buf[i++] = '!';                 /* prefix command w/ bang */
 966         while (*ibufp != '\n')
 967                 switch (*ibufp) {
 968                 default:
 969                         REALLOC(buf, n, i + 2, ERR);
 970                         buf[i++] = *ibufp;
 971                         if (*ibufp++ == '\\')
 972                                 buf[i++] = *ibufp++;
 973                         break;
 974                 case '!':
 975                         if (s != ibufp) {
 976                                 REALLOC(buf, n, i + 1, ERR);
 977                                 buf[i++] = *ibufp++;
 978                         }
 979 #ifdef BACKWARDS
 980                         else if (shcmd == NULL || *(shcmd + 1) == '\0')
 981 #else
 982                         else if (shcmd == NULL)
 983 #endif
 984                         {
 985                                 errmsg = "no previous command";
 986                                 return ERR;
 987                         } else {
 988                                 REALLOC(buf, n, i + shcmdi, ERR);
 989                                 for (s = shcmd + 1; s < shcmd + shcmdi;)
 990                                         buf[i++] = *s++;
 991                                 s = ibufp++;
 992                         }
 993                         break;
 994                 case '%':
 995                         if (*old_filename  == '\0') {
 996                                 errmsg = "no current filename";
 997                                 return ERR;
 998                         }
 999                         j = strlen(s = strip_escapes(old_filename));
1000                         REALLOC(buf, n, i + j, ERR);
1001                         while (j--)
1002                                 buf[i++] = *s++;
1003                         s = ibufp++;
1004                         break;
1005                 }
1006         REALLOC(shcmd, shcmdsz, i + 1, ERR);
1007         memcpy(shcmd, buf, i);
1008         shcmd[shcmdi = i] = '\0';
1009         return *s == '!' || *s == '%';
1010 }
1011
1012
1013 /* append_lines: insert text from stdin to after line n; stop when either a
1014    single period is read or EOF; return status */
1015 int
1016 append_lines(long n)
1017 {
1018         int l;
1019         const char *lp = ibuf;
1020         const char *eot;
1021         undo_t *up = NULL;
1022
1023         for (current_addr = n;;) {
1024                 if (!isglobal) {
1025                         if ((l = get_tty_line()) < 0)
1026                                 return ERR;
1027                         else if (l == 0 || ibuf[l - 1] != '\n') {
1028                                 clearerr(stdin);
1029                                 return  l ? EOF : 0;
1030                         }
1031                         lp = ibuf;
1032                 } else if (*(lp = ibufp) == '\0')
1033                         return 0;
1034                 else {
1035                         while (*ibufp++ != '\n')
1036                                 ;
1037                         l = ibufp - lp;
1038                 }
1039                 if (l == 2 && lp[0] == '.' && lp[1] == '\n') {
1040                         return 0;
1041                 }
1042                 eot = lp + l;
1043                 SPL1();
1044                 do {
1045                         if ((lp = put_sbuf_line(lp)) == NULL) {
1046                                 SPL0();
1047                                 return ERR;
1048                         } else if (up)
1049                                 up->t = get_addressed_line_node(current_addr);
1050                         else if ((up = push_undo_stack(UADD, current_addr,
1051                             current_addr)) == NULL) {
1052                                 SPL0();
1053                                 return ERR;
1054                         }
1055                 } while (lp != eot);
1056                 modified = 1;
1057                 SPL0();
1058         }
1059         /* NOTREACHED */
1060 }
1061
1062
1063 /* join_lines: replace a range of lines with the joined text of those lines */
1064 int
1065 join_lines(long from, long to)
1066 {
1067         static char *buf = NULL;
1068         static int n;
1069
1070         char *s;
1071         int size = 0;
1072         line_t *bp, *ep;
1073
1074         ep = get_addressed_line_node(INC_MOD(to, addr_last));
1075         bp = get_addressed_line_node(from);
1076         for (; bp != ep; bp = bp->q_forw) {
1077                 if ((s = get_sbuf_line(bp)) == NULL)
1078                         return ERR;
1079                 REALLOC(buf, n, size + bp->len, ERR);
1080                 memcpy(buf + size, s, bp->len);
1081                 size += bp->len;
1082         }
1083         REALLOC(buf, n, size + 2, ERR);
1084         memcpy(buf + size, "\n", 2);
1085         if (delete_lines(from, to) < 0)
1086                 return ERR;
1087         current_addr = from - 1;
1088         SPL1();
1089         if (put_sbuf_line(buf) == NULL ||
1090             push_undo_stack(UADD, current_addr, current_addr) == NULL) {
1091                 SPL0();
1092                 return ERR;
1093         }
1094         modified = 1;
1095         SPL0();
1096         return 0;
1097 }
1098
1099
1100 /* move_lines: move a range of lines */
1101 int
1102 move_lines(long addr)
1103 {
1104         line_t *b1, *a1, *b2, *a2;
1105         long n = INC_MOD(second_addr, addr_last);
1106         long p = first_addr - 1;
1107         int done = (addr == first_addr - 1 || addr == second_addr);
1108
1109         SPL1();
1110         if (done) {
1111                 a2 = get_addressed_line_node(n);
1112                 b2 = get_addressed_line_node(p);
1113                 current_addr = second_addr;
1114         } else if (push_undo_stack(UMOV, p, n) == NULL ||
1115             push_undo_stack(UMOV, addr, INC_MOD(addr, addr_last)) == NULL) {
1116                 SPL0();
1117                 return ERR;
1118         } else {
1119                 a1 = get_addressed_line_node(n);
1120                 if (addr < first_addr) {
1121                         b1 = get_addressed_line_node(p);
1122                         b2 = get_addressed_line_node(addr);
1123                                         /* this get_addressed_line_node last! */
1124                 } else {
1125                         b2 = get_addressed_line_node(addr);
1126                         b1 = get_addressed_line_node(p);
1127                                         /* this get_addressed_line_node last! */
1128                 }
1129                 a2 = b2->q_forw;
1130                 REQUE(b2, b1->q_forw);
1131                 REQUE(a1->q_back, a2);
1132                 REQUE(b1, a1);
1133                 current_addr = addr + ((addr < first_addr) ?
1134                     second_addr - first_addr + 1 : 0);
1135         }
1136         if (isglobal)
1137                 unset_active_nodes(b2->q_forw, a2);
1138         modified = 1;
1139         SPL0();
1140         return 0;
1141 }
1142
1143
1144 /* copy_lines: copy a range of lines; return status */
1145 int
1146 copy_lines(long addr)
1147 {
1148         line_t *lp, *np = get_addressed_line_node(first_addr);
1149         undo_t *up = NULL;
1150         long n = second_addr - first_addr + 1;
1151         long m = 0;
1152
1153         current_addr = addr;
1154         if (first_addr <= addr && addr < second_addr) {
1155                 n =  addr - first_addr + 1;
1156                 m = second_addr - addr;
1157         }
1158         for (; n > 0; n=m, m=0, np = get_addressed_line_node(current_addr + 1))
1159                 for (; n-- > 0; np = np->q_forw) {
1160                         SPL1();
1161                         if ((lp = dup_line_node(np)) == NULL) {
1162                                 SPL0();
1163                                 return ERR;
1164                         }
1165                         add_line_node(lp);
1166                         if (up)
1167                                 up->t = lp;
1168                         else if ((up = push_undo_stack(UADD, current_addr,
1169                             current_addr)) == NULL) {
1170                                 SPL0();
1171                                 return ERR;
1172                         }
1173                         modified = 1;
1174                         SPL0();
1175                 }
1176         return 0;
1177 }
1178
1179
1180 /* delete_lines: delete a range of lines */
1181 int
1182 delete_lines(long from, long to)
1183 {
1184         line_t *n, *p;
1185
1186         SPL1();
1187         if (push_undo_stack(UDEL, from, to) == NULL) {
1188                 SPL0();
1189                 return ERR;
1190         }
1191         n = get_addressed_line_node(INC_MOD(to, addr_last));
1192         p = get_addressed_line_node(from - 1);
1193                                         /* this get_addressed_line_node last! */
1194         if (isglobal)
1195                 unset_active_nodes(p->q_forw, n);
1196         REQUE(p, n);
1197         addr_last -= to - from + 1;
1198         current_addr = from - 1;
1199         modified = 1;
1200         SPL0();
1201         return 0;
1202 }
1203
1204
1205 /* display_lines: print a range of lines to stdout */
1206 int
1207 display_lines(long from, long to, int gflag)
1208 {
1209         line_t *bp;
1210         line_t *ep;
1211         char *s;
1212
1213         if (!from) {
1214                 errmsg = "invalid address";
1215                 return ERR;
1216         }
1217         ep = get_addressed_line_node(INC_MOD(to, addr_last));
1218         bp = get_addressed_line_node(from);
1219         for (; bp != ep; bp = bp->q_forw) {
1220                 if ((s = get_sbuf_line(bp)) == NULL)
1221                         return ERR;
1222                 if (put_tty_line(s, bp->len, current_addr = from++, gflag) < 0)
1223                         return ERR;
1224         }
1225         return 0;
1226 }
1227
1228
1229 #define MAXMARK 26                      /* max number of marks */
1230
1231 static line_t *mark[MAXMARK];           /* line markers */
1232 static int markno;                      /* line marker count */
1233
1234 /* mark_line_node: set a line node mark */
1235 int
1236 mark_line_node(line_t *lp, int n)
1237 {
1238         if (!islower((unsigned char)n)) {
1239                 errmsg = "invalid mark character";
1240                 return ERR;
1241         } else if (mark[n - 'a'] == NULL)
1242                 markno++;
1243         mark[n - 'a'] = lp;
1244         return 0;
1245 }
1246
1247
1248 /* get_marked_node_addr: return address of a marked line */
1249 long
1250 get_marked_node_addr(int n)
1251 {
1252         if (!islower((unsigned char)n)) {
1253                 errmsg = "invalid mark character";
1254                 return ERR;
1255         }
1256         return get_line_node_addr(mark[n - 'a']);
1257 }
1258
1259
1260 /* unmark_line_node: clear line node mark */
1261 void
1262 unmark_line_node(line_t *lp)
1263 {
1264         int i;
1265
1266         for (i = 0; markno && i < MAXMARK; i++)
1267                 if (mark[i] == lp) {
1268                         mark[i] = NULL;
1269                         markno--;
1270                 }
1271 }
1272
1273
1274 /* dup_line_node: return a pointer to a copy of a line node */
1275 line_t *
1276 dup_line_node(line_t *lp)
1277 {
1278         line_t *np;
1279
1280         if ((np = (line_t *) malloc(sizeof(line_t))) == NULL) {
1281                 fprintf(stderr, "%s\n", strerror(errno));
1282                 errmsg = "out of memory";
1283                 return NULL;
1284         }
1285         np->seek = lp->seek;
1286         np->len = lp->len;
1287         return np;
1288 }
1289
1290
1291 /* has_trailing_escape:  return the parity of escapes preceding a character
1292    in a string */
1293 int
1294 has_trailing_escape(char *s, char *t)
1295 {
1296     return (s == t || *(t - 1) != '\\') ? 0 : !has_trailing_escape(s, t - 1);
1297 }
1298
1299
1300 /* strip_escapes: return copy of escaped string of at most length PATH_MAX */
1301 char *
1302 strip_escapes(char *s)
1303 {
1304         static char *file = NULL;
1305         static int filesz = 0;
1306
1307         int i = 0;
1308
1309         REALLOC(file, filesz, PATH_MAX, NULL);
1310         while (i < filesz - 1   /* Worry about a possible trailing escape */
1311                && (file[i++] = (*s == '\\') ? *++s : *s))
1312                 s++;
1313         return file;
1314 }
1315
1316
1317 void
1318 signal_hup(int signo)
1319 {
1320         if (mutex)
1321                 sigflags |= (1 << (signo - 1));
1322         else
1323                 handle_hup(signo);
1324 }
1325
1326
1327 void
1328 signal_int(int signo)
1329 {
1330         if (mutex)
1331                 sigflags |= (1 << (signo - 1));
1332         else
1333                 handle_int(signo);
1334 }
1335
1336
1337 void
1338 handle_hup(int signo)
1339 {
1340         char *hup = NULL;               /* hup filename */
1341         char *s;
1342         char ed_hup[] = "ed.hup";
1343         int n;
1344
1345         if (!sigactive)
1346                 quit(1);
1347         sigflags &= ~(1 << (signo - 1));
1348         if (addr_last && write_file(ed_hup, "w", 1, addr_last) < 0 &&
1349             (s = getenv("HOME")) != NULL &&
1350             (n = strlen(s)) + 8 <= PATH_MAX &&  /* "ed.hup" + '/' */
1351             (hup = (char *) malloc(n + 10)) != NULL) {
1352                 strcpy(hup, s);
1353                 if (hup[n - 1] != '/')
1354                         hup[n] = '/', hup[n+1] = '\0';
1355                 strcat(hup, "ed.hup");
1356                 write_file(hup, "w", 1, addr_last);
1357         }
1358         quit(2);
1359 }
1360
1361
1362 void
1363 handle_int(int signo)
1364 {
1365         if (!sigactive)
1366                 quit(1);
1367         sigflags &= ~(1 << (signo - 1));
1368         siglongjmp(env, -1);
1369 }
1370
1371
1372 int cols = 72;                          /* wrap column */
1373
1374 void
1375 handle_winch(int signo)
1376 {
1377         int save_errno = errno;
1378
1379         struct winsize ws;              /* window size structure */
1380
1381         sigflags &= ~(1 << (signo - 1));
1382         if (ioctl(0, TIOCGWINSZ, (char *) &ws) >= 0) {
1383                 if (ws.ws_row > 2) rows = ws.ws_row - 2;
1384                 if (ws.ws_col > 8) cols = ws.ws_col - 8;
1385         }
1386         errno = save_errno;
1387 }
1388
1389
1390 /* is_legal_filename: return a legal filename */
1391 int
1392 is_legal_filename(char *s)
1393 {
1394         if (red && (*s == '!' || !strcmp(s, "..") || strchr(s, '/'))) {
1395                 errmsg = "shell access restricted";
1396                 return 0;
1397         }
1398         return 1;
1399 }