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