editors/sed.c

   1 /*
   2  * sed.c - very minimalist version of sed
   3  *
   4  * Copyright (C) 1999,2000,2001 by Lineo, inc. and Mark Whitley
   5  * Copyright (C) 1999,2000,2001 by Mark Whitley <markw@codepoet.org>
   6  * Copyright (C) 2002  Matt Kraai
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  21  *
  22  */
  23
  24 /*
  25         Supported features and commands in this version of sed:
  26
  27          - comments ('#')
  28          - address matching: num|/matchstr/[,num|/matchstr/|$]command
  29          - commands: (p)rint, (d)elete, (s)ubstitue (with g & I flags)
  30          - edit commands: (a)ppend, (i)nsert, (c)hange
  31          - file commands: (r)ead
  32          - backreferences in substitution expressions (\1, \2...\9)
  33
  34          (Note: Specifying an address (range) to match is *optional*; commands
  35          default to the whole pattern space if no specific address match was
  36          requested.)
  37
  38         Unsupported features:
  39
  40          - transliteration (y/source-chars/dest-chars/) (use 'tr')
  41          - no pattern space hold space storing / swapping (x, etc.)
  42          - no labels / branching (: label, b, t, and friends)
  43          - and lots, lots more.
  44
  45         Reference http://www.opengroup.org/onlinepubs/007904975/utilities/sed.html
  46 */
  47
  48 #include <stdio.h>
  49 #include <unistd.h> /* for getopt() */
  50 #include <regex.h>
  51 #include <string.h> /* for strdup() */
  52 #include <errno.h>
  53 #include <ctype.h> /* for isspace() */
  54 #include <stdlib.h>
  55 #include "busybox.h"
  56
  57 /* externs */
  58 extern void xregcomp(regex_t *preg, const char *regex, int cflags);
  59 extern int optind; /* in unistd.h */
  60 extern char *optarg; /* ditto */
  61
  62 /* options */
  63 static int be_quiet = 0;
  64
  65
  66 typedef struct sed_cmd_s {
  67         /* Order by alignment requirements */
  68
  69         /* address storage */
  70         regex_t *beg_match; /* sed -e '/match/cmd' */
  71         regex_t *end_match; /* sed -e '/match/,/end_match/cmd' */
  72
  73         /* SUBSTITUTION COMMAND SPECIFIC FIELDS */
  74
  75         /* sed -e 's/sub_match/replace/' */
  76         regex_t *sub_match;
  77         char *replace;
  78
  79         /* EDIT COMMAND (a,i,c) SPECIFIC FIELDS */
  80         char *editline;
  81
  82         /* FILE COMMAND (r) SPECIFIC FIELDS */
  83         char *filename;
  84
  85         /* address storage */
  86         int beg_line; /* 'sed 1p'   0 == no begining line, apply commands to all lines */
  87         int end_line; /* 'sed 1,3p' 0 == no end line, use only beginning. -1 == $ */
  88         /* SUBSTITUTION COMMAND SPECIFIC FIELDS */
  89
  90         unsigned int num_backrefs:4; /* how many back references (\1..\9) */
  91                         /* Note:  GNU/POSIX sed does not save more than nine backrefs, so
  92                          * we only use 4 bits to hold the number */
  93         unsigned int sub_g:1; /* sed -e 's/foo/bar/g' (global) */
  94         unsigned int sub_p:2; /* sed -e 's/foo/bar/p' (print substitution) */
  95
  96         /* GENERAL FIELDS */
  97         char delimiter;     /* The delimiter used to separate regexps */
  98
  99         /* the command */
 100         char cmd; /* p,d,s (add more at your leisure :-) */
 101
 102         /* inversion flag */
 103         int invert;         /* the '!' after the address */
 104 } sed_cmd_t;
 105
 106 /* globals */
 107 static sed_cmd_t *sed_cmds = NULL; /* growable arrary holding a sequence of sed cmds */
 108 static int ncmds = 0; /* number of sed commands */
 109
 110 /*static char *cur_file = NULL;*/ /* file currently being processed XXX: do I need this? */
 111
 112 const char * const semicolon_whitespace = "; \n\r\t\v\0";
 113
 114 #ifdef CONFIG_FEATURE_CLEAN_UP
 115 static void destroy_cmd_strs(void)
 116 {
 117         if (sed_cmds == NULL)
 118                 return;
 119
 120         /* destroy all the elements in the array */
 121         while (--ncmds >= 0) {
 122
 123                 if (sed_cmds[ncmds].beg_match) {
 124                         regfree(sed_cmds[ncmds].beg_match);
 125                         free(sed_cmds[ncmds].beg_match);
 126                 }
 127                 if (sed_cmds[ncmds].end_match) {
 128                         regfree(sed_cmds[ncmds].end_match);
 129                         free(sed_cmds[ncmds].end_match);
 130                 }
 131                 if (sed_cmds[ncmds].sub_match) {
 132                         regfree(sed_cmds[ncmds].sub_match);
 133                         free(sed_cmds[ncmds].sub_match);
 134                 }
 135                 free(sed_cmds[ncmds].replace);
 136         }
 137
 138         /* destroy the array */
 139         free(sed_cmds);
 140         sed_cmds = NULL;
 141 }
 142 #endif
 143
 144
 145 /*
 146  * index_of_next_unescaped_regexp_delim - walks left to right through a string
 147  * beginning at a specified index and returns the index of the next regular
 148  * expression delimiter (typically a forward * slash ('/')) not preceeded by
 149  * a backslash ('\').
 150  */
 151 static int index_of_next_unescaped_regexp_delim(const char delimiter, const char *str, int idx)
 152 {
 153         int bracket = -1;
 154         int escaped = 0;
 155         char ch;
 156
 157         for ( ; (ch = str[idx]); idx++) {
 158                 if (bracket != -1) {
 159                         if (ch == ']' && !(bracket == idx - 1 ||
 160                                                                          (bracket == idx - 2 && str[idx-1] == '^')))
 161                                 bracket = -1;
 162                 } else if (escaped)
 163                         escaped = 0;
 164                 else if (ch == '\\')
 165                         escaped = 1;
 166                 else if (ch == '[')
 167                         bracket = idx;
 168                 else if (ch == delimiter)
 169                         return idx;
 170         }
 171
 172         /* if we make it to here, we've hit the end of the string */
 173         return -1;
 174 }
 175
 176 /*
 177  * returns the index in the string just past where the address ends.
 178  */
 179 static int get_address(char *delimiter, char *my_str, int *linenum, regex_t **regex)
 180 {
 181         int idx = 0;
 182         if (isdigit(my_str[idx])) {
 183                 char *endstr;
 184                 *linenum = strtol(my_str, &endstr, 10);
 185                 /* endstr shouldnt ever equal NULL */
 186                 idx = endstr - my_str;
 187         }
 188         else if (my_str[idx] == '$') {
 189                 *linenum = -1;
 190                 idx++;
 191         }
 192         else if (my_str[idx] == '/' || my_str[idx] == '\\') {
 193                 int idx_start = 1;
 194
 195                 if (my_str[idx] == '\\') {
 196                         idx_start++;
 197                         *delimiter = my_str[++idx];
 198                 }
 199                 idx = index_of_next_unescaped_regexp_delim(*delimiter, my_str, ++idx);
 200                 if (idx == -1) {
 201                         error_msg_and_die("unterminated match expression");
 202                 }
 203                 my_str[idx] = '\0';
 204                 *regex = (regex_t *)xmalloc(sizeof(regex_t));
 205                 xregcomp(*regex, my_str+idx_start, REG_NEWLINE);
 206                 idx++; /* so it points to the next character after the last '/' */
 207         }
 208         return idx;
 209 }
 210
 211 static int parse_subst_cmd(sed_cmd_t * const sed_cmd, const char *substr)
 212 {
 213         int oldidx, cflags = REG_NEWLINE;
 214         char *match;
 215         int idx = 0;
 216         int j;
 217
 218         /*
 219          * the string that gets passed to this function should look like this:
 220          *    s/match/replace/gIp
 221          *    ||     |        |||
 222          *    mandatory       optional
 223          *
 224          *    (all three of the '/' slashes are mandatory)
 225          */
 226
 227         /* verify that the 's' is followed by something.  That something
 228          * (typically a 'slash') is now our regexp delimiter... */
 229         if (!substr[++idx])
 230                 error_msg_and_die("bad format in substitution expression");
 231         else
 232             sed_cmd->delimiter=substr[idx];
 233
 234         /* save the match string */
 235         oldidx = idx+1;
 236         idx = index_of_next_unescaped_regexp_delim(sed_cmd->delimiter, substr, ++idx);
 237         if (idx == -1)
 238                 error_msg_and_die("bad format in substitution expression");
 239         match = xstrndup(substr + oldidx, idx - oldidx);
 240
 241         /* determine the number of back references in the match string */
 242         /* Note: we compute this here rather than in the do_subst_command()
 243          * function to save processor time, at the expense of a little more memory
 244          * (4 bits) per sed_cmd */
 245
 246         /* sed_cmd->num_backrefs = 0; */ /* XXX: not needed? --apparently not */
 247         for (j = 0; match[j]; j++) {
 248                 /* GNU/POSIX sed does not save more than nine backrefs */
 249                 if (match[j] == '\\' && match[j+1] == '(' && sed_cmd->num_backrefs <= 9)
 250                         sed_cmd->num_backrefs++;
 251         }
 252
 253         /* save the replacement string */
 254         oldidx = idx+1;
 255         idx = index_of_next_unescaped_regexp_delim(sed_cmd->delimiter, substr, ++idx);
 256         if (idx == -1)
 257                 error_msg_and_die("bad format in substitution expression");
 258         sed_cmd->replace = xstrndup(substr + oldidx, idx - oldidx);
 259
 260         /* process the flags */
 261         while (substr[++idx]) {
 262                 switch (substr[idx]) {
 263                         case 'g':
 264                                 sed_cmd->sub_g = 1;
 265                                 break;
 266                         case 'I':
 267                                 cflags |= REG_ICASE;
 268                                 break;
 269                         case 'p':
 270                                 sed_cmd->sub_p = 1;
 271                                 break;
 272                         default:
 273                                 /* any whitespace or semicolon trailing after a s/// is ok */
 274                                 if (strchr(semicolon_whitespace, substr[idx]))
 275                                         goto out;
 276                                 /* else */
 277                                 error_msg_and_die("bad option in substitution expression");
 278                 }
 279         }
 280
 281 out:
 282         /* compile the match string into a regex */
 283         sed_cmd->sub_match = (regex_t *)xmalloc(sizeof(regex_t));
 284         xregcomp(sed_cmd->sub_match, match, cflags);
 285         free(match);
 286
 287         return idx;
 288 }
 289
 290 static void move_back(char *str, int offset)
 291 {
 292         memmove(str, str + offset, strlen(str + offset) + 1);
 293 }
 294
 295 static int parse_edit_cmd(sed_cmd_t *sed_cmd, const char *editstr)
 296 {
 297         int i, j;
 298
 299         /*
 300          * the string that gets passed to this function should look like this:
 301          *
 302          *    need one of these
 303          *    |
 304          *    |    this backslash (immediately following the edit command) is mandatory
 305          *    |    |
 306          *    [aic]\
 307          *    TEXT1\
 308          *    TEXT2\
 309          *    TEXTN
 310          *
 311          * as soon as we hit a TEXT line that has no trailing '\', we're done.
 312          * this means a command like:
 313          *
 314          * i\
 315          * INSERTME
 316          *
 317          * is a-ok.
 318          *
 319          */
 320
 321         if (editstr[1] != '\\' || (editstr[2] != '\n' && editstr[2] != '\r'))
 322                 error_msg_and_die("bad format in edit expression");
 323
 324         /* store the edit line text */
 325         sed_cmd->editline = xmalloc(strlen(&editstr[3]) + 2);
 326         for (i = 3, j = 0; editstr[i] != '\0' && strchr("\r\n", editstr[i]) == NULL;
 327                         i++, j++) {
 328                 if (editstr[i] == '\\' && strchr("\n\r", editstr[i+1]) != NULL) {
 329                         sed_cmd->editline[j] = '\n';
 330                         i++;
 331                 } else
 332                         sed_cmd->editline[j] = editstr[i];
 333         }
 334
 335         /* figure out if we need to add a newline */
 336         if (sed_cmd->editline[j-1] != '\n')
 337                 sed_cmd->editline[j++] = '\n';
 338
 339         /* terminate string */
 340         sed_cmd->editline[j] = '\0';
 341
 342         return i;
 343 }
 344
 345
 346 static int parse_file_cmd(sed_cmd_t *sed_cmd, const char *filecmdstr)
 347 {
 348         int idx = 0;
 349         int filenamelen = 0;
 350
 351         /*
 352          * the string that gets passed to this function should look like this:
 353          *    '[ ]filename'
 354          *      |  |
 355          *      |  a filename
 356          *      |
 357          *     optional whitespace
 358
 359          *   re: the file to be read, the GNU manual says the following: "Note that
 360          *   if filename cannot be read, it is treated as if it were an empty file,
 361          *   without any error indication." Thus, all of the following commands are
 362          *   perfectly leagal:
 363          *
 364          *   sed -e '1r noexist'
 365          *   sed -e '1r ;'
 366          *   sed -e '1r'
 367          */
 368
 369         /* the file command may be followed by whitespace; move past it. */
 370         while (isspace(filecmdstr[++idx]))
 371                 { ; }
 372
 373         /* the first non-whitespace we get is a filename. the filename ends when we
 374          * hit a normal sed command terminator or end of string */
 375         filenamelen = strcspn(&filecmdstr[idx], semicolon_whitespace);
 376         sed_cmd->filename = xmalloc(filenamelen + 1);
 377         safe_strncpy(sed_cmd->filename, &filecmdstr[idx], filenamelen + 1);
 378
 379         return idx + filenamelen;
 380 }
 381
 382
 383 static char *parse_cmd_str(sed_cmd_t * const sed_cmd, char *cmdstr)
 384 {
 385         int idx = 0;
 386
 387         /* parse the command
 388          * format is: [addr][,addr]cmd
 389          *            |----||-----||-|
 390          *            part1 part2  part3
 391          */
 392
 393         /* first part (if present) is an address: either a '$', a number or a /regex/ */
 394         idx = get_address(&sed_cmd->delimiter, cmdstr, &sed_cmd->beg_line, &sed_cmd->beg_match);
 395
 396         /* second part (if present) will begin with a comma */
 397         if (cmdstr[idx] == ',') {
 398                 int tmp_idx;
 399                 idx++;
 400                 tmp_idx = get_address(&sed_cmd->delimiter, &cmdstr[idx], &sed_cmd->end_line, &sed_cmd->end_match);
 401                 if (tmp_idx == 0) {
 402                         error_msg_and_die("get_address: no address found in string\n"
 403                                 "\t(you probably didn't check the string you passed me)");
 404                 }
 405                 idx += tmp_idx;
 406         }
 407
 408         /* skip whitespace before the command */
 409         while (isspace(cmdstr[idx]))
 410                 idx++;
 411
 412         /* there my be the inversion flag between part2 and part3 */
 413         sed_cmd->invert = 0;
 414         if (cmdstr[idx] == '!') {
 415                 sed_cmd->invert = 1;
 416                 idx++;
 417
 418 #ifdef SED_FEATURE_STRICT_CHECKING
 419                 /* According to the spec
 420                  * It is unspecified whether <blank>s can follow a '!' character,
 421                  * and conforming applications shall not follow a '!' character
 422                  * with <blank>s.
 423                  */
 424                 if (isblank(cmdstr[idx]) {
 425                         error_msg_and_die("blank follows '!'");
 426                 }
 427 #else
 428                 /* skip whitespace before the command */
 429                 while (isspace(cmdstr[idx]))
 430                         idx++;
 431 #endif
 432         }
 433
 434         /* last part (mandatory) will be a command */
 435         if (cmdstr[idx] == '\0')
 436                 error_msg_and_die("missing command");
 437         sed_cmd->cmd = cmdstr[idx];
 438
 439         /* if it was a single-letter command that takes no arguments (such as 'p'
 440          * or 'd') all we need to do is increment the index past that command */
 441         if (strchr("pd=", sed_cmd->cmd)) {
 442                 idx++;
 443         }
 444         /* handle (s)ubstitution command */
 445         else if (sed_cmd->cmd == 's') {
 446                 idx += parse_subst_cmd(sed_cmd, &cmdstr[idx]);
 447         }
 448         /* handle edit cmds: (a)ppend, (i)nsert, and (c)hange */
 449         else if (strchr("aic", sed_cmd->cmd)) {
 450                 if ((sed_cmd->end_line || sed_cmd->end_match) && sed_cmd->cmd != 'c')
 451                         error_msg_and_die("only a beginning address can be specified for edit commands");
 452                 idx += parse_edit_cmd(sed_cmd, &cmdstr[idx]);
 453         }
 454         /* handle file cmds: (r)ead */
 455         else if (sed_cmd->cmd == 'r') {
 456                 if (sed_cmd->end_line || sed_cmd->end_match)
 457                         error_msg_and_die("Command only uses one address");
 458                 idx += parse_file_cmd(sed_cmd, &cmdstr[idx]);
 459         }
 460         else {
 461                 error_msg_and_die("Unsupported command %c", sed_cmd->cmd);
 462         }
 463
 464         /* give back whatever's left over */
 465         return (char *)&cmdstr[idx];
 466 }
 467
 468 static void add_cmd_str(const char * const cmdstr)
 469 {
 470         char *mystr = (char *)cmdstr;
 471
 472         do {
 473
 474                 /* trim leading whitespace and semicolons */
 475                 move_back(mystr, strspn(mystr, semicolon_whitespace));
 476                 /* if we ate the whole thing, that means there was just trailing
 477                  * whitespace or a final / no-op semicolon. either way, get out */
 478                 if (strlen(mystr) == 0)
 479                         return;
 480                 /* if this is a comment, jump past it and keep going */
 481                 if (mystr[0] == '#') {
 482                         mystr = strpbrk(mystr, "\n\r");
 483                         continue;
 484                 }
 485                 /* grow the array */
 486                 sed_cmds = xrealloc(sed_cmds, sizeof(sed_cmd_t) * (++ncmds));
 487                 /* zero new element */
 488                 memset(&sed_cmds[ncmds-1], 0, sizeof(sed_cmd_t));
 489                 /* load command string into new array element, get remainder */
 490                 mystr = parse_cmd_str(&sed_cmds[ncmds-1], mystr);
 491
 492         } while (mystr && strlen(mystr));
 493 }
 494
 495
 496 static void load_cmd_file(char *filename)
 497 {
 498         FILE *cmdfile;
 499         char *line;
 500         char *nextline;
 501
 502         cmdfile = xfopen(filename, "r");
 503
 504         while ((line = get_line_from_file(cmdfile)) != NULL) {
 505                 /* if a line ends with '\' it needs the next line appended to it */
 506                 while (line[strlen(line)-2] == '\\' &&
 507                                 (nextline = get_line_from_file(cmdfile)) != NULL) {
 508                         line = xrealloc(line, strlen(line) + strlen(nextline) + 1);
 509                         strcat(line, nextline);
 510                         free(nextline);
 511                 }
 512                 /* eat trailing newline (if any) --if I don't do this, edit commands
 513                  * (aic) will print an extra newline */
 514                 chomp(line);
 515                 add_cmd_str(line);
 516                 free(line);
 517         }
 518 }
 519
 520 struct pipeline {
 521         char *buf;
 522         int idx;
 523         int len;
 524 };
 525
 526 #define PIPE_MAGIC 0x7f
 527 #define PIPE_GROW 64
 528
 529 void pipe_putc(struct pipeline *const pipeline, char c)
 530 {
 531         if (pipeline->buf[pipeline->idx] == PIPE_MAGIC) {
 532                 pipeline->buf =
 533                         xrealloc(pipeline->buf, pipeline->len + PIPE_GROW);
 534                 memset(pipeline->buf + pipeline->len, 0, PIPE_GROW);
 535                 pipeline->len += PIPE_GROW;
 536                 pipeline->buf[pipeline->len - 1] = PIPE_MAGIC;
 537         }
 538         pipeline->buf[pipeline->idx++] = (c);
 539 }
 540
 541 #define pipeputc(c)     pipe_putc(pipeline, c)
 542
 543 #if 0
 544 { if (pipeline[pipeline_idx] == PIPE_MAGIC) { \
 545         pipeline = xrealloc(pipeline, pipeline_len+PIPE_GROW); \
 546         memset(pipeline+pipeline_len, 0, PIPE_GROW); \
 547         pipeline_len += PIPE_GROW; \
 548         pipeline[pipeline_len-1] = PIPE_MAGIC; } \
 549         pipeline[pipeline_idx++] = (c); }
 550 #endif
 551
 552 static void print_subst_w_backrefs(const char *line, const char *replace,
 553         regmatch_t *regmatch, struct pipeline *const pipeline, int matches)
 554 {
 555         int i;
 556
 557         /* go through the replacement string */
 558         for (i = 0; replace[i]; i++) {
 559                 /* if we find a backreference (\1, \2, etc.) print the backref'ed * text */
 560                 if (replace[i] == '\\' && isdigit(replace[i+1])) {
 561                         int j;
 562                         char tmpstr[2];
 563                         int backref;
 564                         ++i; /* i now indexes the backref number, instead of the leading slash */
 565                         tmpstr[0] = replace[i];
 566                         tmpstr[1] = 0;
 567                         backref = atoi(tmpstr);
 568                         /* print out the text held in regmatch[backref] */
 569                         if (backref <= matches && regmatch[backref].rm_so != -1)
 570                                 for (j = regmatch[backref].rm_so; j < regmatch[backref].rm_eo; j++)
 571                                         pipeputc(line[j]);
 572                 }
 573
 574                 /* if we find a backslash escaped character, print the character */
 575                 else if (replace[i] == '\\') {
 576                         ++i;
 577                         pipeputc(replace[i]);
 578                 }
 579
 580                 /* if we find an unescaped '&' print out the whole matched text.
 581                  * fortunately, regmatch[0] contains the indicies to the whole matched
 582                  * expression (kinda seems like it was designed for just such a
 583                  * purpose...) */
 584                 else if (replace[i] == '&' && replace[i-1] != '\\') {
 585                         int j;
 586                         for (j = regmatch[0].rm_so; j < regmatch[0].rm_eo; j++)
 587                                 pipeputc(line[j]);
 588                 }
 589                 /* nothing special, just print this char of the replacement string to stdout */
 590                 else
 591                         pipeputc(replace[i]);
 592         }
 593 }
 594
 595 static int do_subst_command(const sed_cmd_t *sed_cmd, char **line)
 596 {
 597         char *hackline = *line;
 598         struct pipeline thepipe = { NULL, 0 , 0};
 599         struct pipeline *const pipeline = &thepipe;
 600         int altered = 0;
 601         regmatch_t *regmatch = NULL;
 602
 603         /* we only proceed if the substitution 'search' expression matches */
 604         if (regexec(sed_cmd->sub_match, hackline, 0, NULL, 0) == REG_NOMATCH)
 605                 return 0;
 606
 607         /* whaddaya know, it matched. get the number of back references */
 608         regmatch = xmalloc(sizeof(regmatch_t) * (sed_cmd->num_backrefs+1));
 609
 610         /* allocate more PIPE_GROW bytes
 611            if replaced string is larger than original */
 612         thepipe.len = strlen(hackline)+PIPE_GROW;
 613         thepipe.buf = xcalloc(1, thepipe.len);
 614         /* buffer magic */
 615         thepipe.buf[thepipe.len-1] = PIPE_MAGIC;
 616
 617         /* and now, as long as we've got a line to try matching and if we can match
 618          * the search string, we make substitutions */
 619         while ((*hackline || !altered) && (regexec(sed_cmd->sub_match, hackline,
 620                                         sed_cmd->num_backrefs+1, regmatch, 0) != REG_NOMATCH) ) {
 621                 int i;
 622
 623                 /* print everything before the match */
 624                 for (i = 0; i < regmatch[0].rm_so; i++)
 625                         pipeputc(hackline[i]);
 626
 627                 /* then print the substitution string */
 628                 print_subst_w_backrefs(hackline, sed_cmd->replace, regmatch,
 629                                 pipeline, sed_cmd->num_backrefs);
 630
 631                 /* advance past the match */
 632                 hackline += regmatch[0].rm_eo;
 633                 /* flag that something has changed */
 634                 altered++;
 635
 636                 /* if we're not doing this globally, get out now */
 637                 if (!sed_cmd->sub_g)
 638                         break;
 639         }
 640
 641         for (; *hackline; hackline++) pipeputc(*hackline);
 642         if (thepipe.buf[thepipe.idx] == PIPE_MAGIC) thepipe.buf[thepipe.idx] = 0;
 643
 644         /* cleanup */
 645         free(regmatch);
 646
 647         free(*line);
 648         *line = thepipe.buf;
 649         return altered;
 650 }
 651
 652
 653 static void process_file(FILE *file)
 654 {
 655         char *line;
 656         static int linenum = 0; /* GNU sed does not restart counting lines at EOF */
 657         unsigned int still_in_range = 0;
 658         int altered;
 659         int i;
 660
 661         line = get_line_from_file(file);
 662         if (line == NULL) {
 663                 return;
 664         }
 665
 666         /* go through every line in the file */
 667         do {
 668                 char *next_line;
 669
 670                 /* Read one line in advance so we can act on the last line, the '$' address */
 671                 next_line = get_line_from_file(file);
 672
 673                 chomp(line);
 674                 linenum++;
 675                 altered = 0;
 676
 677                 /* for every line, go through all the commands */
 678                 for (i = 0; i < ncmds; i++) {
 679                         sed_cmd_t *sed_cmd = &sed_cmds[i];
 680                         int deleted = 0;
 681
 682                         /*
 683                          * entry point into sedding...
 684                          */
 685                         int matched = (
 686                                         /* no range necessary */
 687                                         (sed_cmd->beg_line == 0 && sed_cmd->end_line == 0 &&
 688                                          sed_cmd->beg_match == NULL &&
 689                                          sed_cmd->end_match == NULL) ||
 690                                         /* this line number is the first address we're looking for */
 691                                         (sed_cmd->beg_line && (sed_cmd->beg_line == linenum)) ||
 692                                         /* this line matches our first address regex */
 693                                         (sed_cmd->beg_match && (regexec(sed_cmd->beg_match, line, 0, NULL, 0) == 0)) ||
 694                                         /* we are currently within the beginning & ending address range */
 695                                         still_in_range || ((sed_cmd->beg_line == -1) && (next_line == NULL))
 696                            );
 697
 698                         if (sed_cmd->invert ^ matched) {
 699
 700                                 /*
 701                                  * actual sedding
 702                                  */
 703                                 switch (sed_cmd->cmd) {
 704                                         case '=':
 705                                                 printf("%d\n", linenum);
 706                                                 break;
 707                                         case 'p':
 708                                                 puts(line);
 709                                                 break;
 710
 711                                         case 'd':
 712                                                 altered++;
 713                                                 deleted = 1;
 714                                                 break;
 715
 716                                         case 's':
 717
 718                                                 /*
 719                                                  * Some special cases for 's' printing to make it compliant with
 720                                                  * GNU sed printing behavior (aka "The -n | s///p Matrix"):
 721                                                  *
 722                                                  *    -n ONLY = never print anything regardless of any successful
 723                                                  *    substitution
 724                                                  *
 725                                                  *    s///p ONLY = always print successful substitutions, even if
 726                                                  *    the line is going to be printed anyway (line will be printed
 727                                                  *    twice).
 728                                                  *
 729                                                  *    -n AND s///p = print ONLY a successful substitution ONE TIME;
 730                                                  *    no other lines are printed - this is the reason why the 'p'
 731                                                  *    flag exists in the first place.
 732                                                  */
 733
 734                                                 /* if the user specified that they didn't want anything printed (i.e., a -n
 735                                                  * flag and no 'p' flag after the s///), then there's really no point doing
 736                                                  * anything here. */
 737                                                 if (be_quiet && !sed_cmd->sub_p)
 738                                                         break;
 739
 740                                                 /* we print the line once, unless we were told to be quiet */
 741                                                 if (!be_quiet)
 742                                                         altered |= do_subst_command(sed_cmd, &line);
 743
 744                                                 /* we also print the line if we were given the 'p' flag
 745                                                  * (this is quite possibly the second printing) */
 746                                                 if (sed_cmd->sub_p)
 747                                                         altered |= do_subst_command(sed_cmd, &line);
 748                                                 if (altered && (i+1 >= ncmds || sed_cmds[i+1].cmd != 's'))
 749                                                         puts(line);
 750
 751                                                 break;
 752
 753                                         case 'a':
 754                                                 puts(line);
 755                                                 fputs(sed_cmd->editline, stdout);
 756                                                 altered++;
 757                                                 break;
 758
 759                                         case 'i':
 760                                                 fputs(sed_cmd->editline, stdout);
 761                                                 break;
 762
 763                                         case 'c':
 764                                                 /* single-address case */
 765                                                 if ((sed_cmd->end_match == NULL && sed_cmd->end_line == 0)
 766                                                 /* multi-address case */
 767                                                 /* - matching text */
 768                                                 || (sed_cmd->end_match && (regexec(sed_cmd->end_match, line, 0, NULL, 0) == 0))
 769                                                 /* - matching line numbers */
 770                                                 || (sed_cmd->end_line > 0 && sed_cmd->end_line == linenum))
 771                                                 {
 772                                                         fputs(sed_cmd->editline, stdout);
 773                                                 }
 774                                                 altered++;
 775
 776                                                 break;
 777
 778                                         case 'r': {
 779                                                                   FILE *outfile;
 780                                                                   puts(line);
 781                                                                   outfile = fopen(sed_cmd->filename, "r");
 782                                                                   if (outfile)
 783                                                                           print_file(outfile);
 784                                                                   /* else if we couldn't open the output file,
 785                                                                    * no biggie, just don't print anything */
 786                                                                   altered++;
 787                                                   }
 788                                                           break;
 789                                 }
 790                         }
 791
 792                         /*
 793                          * exit point from sedding...
 794                          */
 795                         if (matched) {
 796                                 if (
 797                                         /* this is a single-address command or... */
 798                                         (sed_cmd->end_line == 0 && sed_cmd->end_match == NULL) || (
 799                                                 /* we were in the middle of our address range (this
 800                                                  * isn't the first time through) and.. */
 801                                                 (still_in_range == 1) && (
 802                                                         /* this line number is the last address we're looking for or... */
 803                                                         (sed_cmd->end_line && (sed_cmd->end_line == linenum)) ||
 804                                                         /* this line matches our last address regex */
 805                                                         (sed_cmd->end_match && (regexec(sed_cmd->end_match, line, 0, NULL, 0) == 0))
 806                                                 )
 807                                         )
 808                                 ) {
 809                                         /* we're out of our address range */
 810                                         still_in_range = 0;
 811                                 }
 812
 813                                 /* didn't hit the exit? then we're still in the middle of an address range */
 814                                 else {
 815                                         still_in_range = 1;
 816                                 }
 817                         }
 818
 819                         if (deleted)
 820                                 break;
 821                 }
 822
 823                 /* we will print the line unless we were told to be quiet or if the
 824                  * line was altered (via a 'd'elete or 's'ubstitution), in which case
 825                  * the altered line was already printed */
 826                 if (!be_quiet && !altered)
 827                         puts(line);
 828
 829                 free(line);
 830                 line = next_line;
 831         } while (line);
 832 }
 833
 834 extern int sed_main(int argc, char **argv)
 835 {
 836         int opt, status = EXIT_SUCCESS;
 837
 838 #ifdef CONFIG_FEATURE_CLEAN_UP
 839         /* destroy command strings on exit */
 840         if (atexit(destroy_cmd_strs) == -1)
 841                 perror_msg_and_die("atexit");
 842 #endif
 843
 844         /* do normal option parsing */
 845         while ((opt = getopt(argc, argv, "ne:f:")) > 0) {
 846                 switch (opt) {
 847                         case 'n':
 848                                 be_quiet++;
 849                                 break;
 850                         case 'e':
 851                                 add_cmd_str(optarg);
 852                                 break;
 853                         case 'f':
 854                                 load_cmd_file(optarg);
 855                                 break;
 856                         default:
 857                                 show_usage();
 858                 }
 859         }
 860
 861         /* if we didn't get a pattern from a -e and no command file was specified,
 862          * argv[optind] should be the pattern. no pattern, no worky */
 863         if (ncmds == 0) {
 864                 if (argv[optind] == NULL)
 865                         show_usage();
 866                 else {
 867                         add_cmd_str(argv[optind]);
 868                         optind++;
 869                 }
 870         }
 871
 872
 873         /* argv[(optind)..(argc-1)] should be names of file to process. If no
 874          * files were specified or '-' was specified, take input from stdin.
 875          * Otherwise, we process all the files specified. */
 876         if (argv[optind] == NULL || (strcmp(argv[optind], "-") == 0)) {
 877                 process_file(stdin);
 878         }
 879         else {
 880                 int i;
 881                 FILE *file;
 882                 for (i = optind; i < argc; i++) {
 883                         file = wfopen(argv[i], "r");
 884                         if (file) {
 885                                 process_file(file);
 886                                 fclose(file);
 887                         } else
 888                                 status = EXIT_FAILURE;
 889                 }
 890         }
 891
 892         return status;
 893 }