X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=editors%2Fsed.c;h=45574e81dd5138039159a8032e8a3649ee37a44b;hb=21d7d61de164b25bc6a70e49b12567e80caed953;hp=f0a3c028ff2858385eccc74c169a9144ad0b4004;hpb=8417c8c38bdcab79989ceab50382eb6314505c93;p=oweals%2Fbusybox.git diff --git a/editors/sed.c b/editors/sed.c index f0a3c028f..45574e81d 100644 --- a/editors/sed.c +++ b/editors/sed.c @@ -1,10 +1,12 @@ +/* vi: set sw=4 ts=4: */ /* * sed.c - very minimalist version of sed * * Copyright (C) 1999,2000,2001 by Lineo, inc. and Mark Whitley * Copyright (C) 1999,2000,2001 by Mark Whitley * Copyright (C) 2002 Matt Kraai - * Copyright (C) 2003 by Glenn McGrath + * Copyright (C) 2003 by Glenn McGrath + * Copyright (C) 2003,2004 by Rob Landley * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -22,6 +24,24 @@ * */ +/* Code overview. + + Files are laid out to avoid unnecessary function declarations. So for + example, every function add_cmd calls occurs before add_cmd in this file. + + add_cmd() is called on each line of sed command text (from a file or from + the command line). It calls get_address() and parse_cmd_args(). The + resulting sed_cmd_t structures are appended to a linked list + (sed_cmd_head/sed_cmd_tail). + + process_file() does actual sedding, reading data lines from an input FILE * + (which could be stdin) and applying the sed command list (sed_cmd_head) to + each of the resulting lines. + + sed_main() is where external code calls into this, with a command line. +*/ + + /* Supported features and commands in this version of sed: @@ -45,9 +65,11 @@ - GNU extensions - and more. - Bugs: - - - lots + Todo: + + - Create a wrapper around regex to make libc's regex conform with sed + - Fix bugs + Reference http://www.opengroup.org/onlinepubs/007904975/utilities/sed.html */ @@ -62,84 +84,75 @@ #include "busybox.h" typedef struct sed_cmd_s { - /* Order by alignment requirements */ - - /* address storage */ - regex_t *beg_match; /* sed -e '/match/cmd' */ - regex_t *end_match; /* sed -e '/match/,/end_match/cmd' */ - - int beg_line; /* 'sed 1p' 0 == no begining line, apply commands to all lines */ - int end_line; /* 'sed 1,3p' 0 == no end line, use only beginning. -1 == $ */ - - /* inversion flag */ - int invert; /* the '!' after the address */ - - /* Runtime flag no not if the current command match's */ - int still_in_range; - - /* SUBSTITUTION COMMAND SPECIFIC FIELDS */ - - /* sed -e 's/sub_match/replace/' */ - regex_t *sub_match; - char *replace; - - /* EDIT COMMAND (a,i,c) SPECIFIC FIELDS */ - char *editline; - - /* FILE COMMAND (r) SPECIFIC FIELDS */ - char *filename; - - /* SUBSTITUTION COMMAND SPECIFIC FIELDS */ + /* Ordered by alignment requirements: currently 36 bytes on x86 */ - unsigned int num_backrefs:4; /* how many back references (\1..\9) */ - /* Note: GNU/POSIX sed does not save more than nine backrefs, so - * we only use 4 bits to hold the number */ - unsigned int sub_g:1; /* sed -e 's/foo/bar/g' (global) */ - unsigned int sub_p:1; /* sed -e 's/foo/bar/p' (print substitution) */ + /* address storage */ + regex_t *beg_match; /* sed -e '/match/cmd' */ + regex_t *end_match; /* sed -e '/match/,/end_match/cmd' */ + regex_t *sub_match; /* For 's/sub_match/string/' */ + int beg_line; /* 'sed 1p' 0 == apply commands to all lines */ + int end_line; /* 'sed 1,3p' 0 == one line only. -1 = last line ($) */ - /* TRANSLATE COMMAND */ - char *translate; + FILE *file; /* File (sr) command writes to, -1 for none. */ + char *string; /* Data string for (saicytb) commands. */ - /* GENERAL FIELDS */ - /* the command */ - char cmd; /* p,d,s (add more at your leisure :-) */ + unsigned short which_match; /* (s) Which match to replace (0 for all) */ - /* Branch commands */ - char *label; + /* Bitfields (gcc won't group them if we don't) */ + unsigned int invert:1; /* the '!' after the address */ + unsigned int in_match:1; /* Next line also included in match? */ + unsigned int no_newline:1; /* Last line written by (sr) had no '\n' */ + unsigned int sub_p:1; /* (s) print option */ - /* next command in list (sequential list of specified commands) */ - struct sed_cmd_s *next; + /* GENERAL FIELDS */ + char cmd; /* The command char: abcdDgGhHilnNpPqrstwxy:={} */ + struct sed_cmd_s *next; /* Next command (linked list, NULL terminated) */ } sed_cmd_t; - -/* externs */ -extern void xregcomp(regex_t * preg, const char *regex, int cflags); -extern int optind; /* in unistd.h */ -extern char *optarg; /* ditto */ - /* globals */ /* options */ -static int be_quiet = 0; +static int be_quiet = 0, in_place=0; +FILE *nonstdout; +char *outname; + + static const char bad_format_in_subst[] = "bad format in substitution expression"; +const char *const semicolon_whitespace = "; \n\r\t\v"; + +regmatch_t regmatch[10]; +static regex_t *previous_regex_ptr = NULL; /* linked list of sed commands */ static sed_cmd_t sed_cmd_head; static sed_cmd_t *sed_cmd_tail = &sed_cmd_head; -const char *const semicolon_whitespace = "; \n\r\t\v\0"; -static regex_t *previous_regex_ptr = NULL; - +/* Linked list of append lines */ +struct append_list { + char *string; + struct append_list *next; +}; +struct append_list *append_head=NULL, *append_tail=NULL; #ifdef CONFIG_FEATURE_CLEAN_UP -static void destroy_cmd_strs(void) +static void free_and_close_stuff(void) { sed_cmd_t *sed_cmd = sed_cmd_head.next; + while(append_head) { + append_tail=append_head->next; + free(append_head->string); + free(append_head); + append_head=append_tail; + } + while (sed_cmd) { sed_cmd_t *sed_cmd_next = sed_cmd->next; + if(sed_cmd->file) + bb_xprint_and_close_file(sed_cmd->file); + if (sed_cmd->beg_match) { regfree(sed_cmd->beg_match); free(sed_cmd->beg_match); @@ -152,21 +165,52 @@ static void destroy_cmd_strs(void) regfree(sed_cmd->sub_match); free(sed_cmd->sub_match); } - free(sed_cmd->replace); - free(sed_cmd->editline); - free(sed_cmd->filename); - free(sed_cmd->translate); - free(sed_cmd->label); + free(sed_cmd->string); free(sed_cmd); sed_cmd = sed_cmd_next; } } #endif +/* If something bad happens during -i operation, delete temp file */ + +static void cleanup_outname(void) +{ + if(outname) unlink(outname); +} + +/* strdup, replacing "\n" with '\n', and "\delimiter" with 'delimiter' */ + +static void parse_escapes(char *dest, const char *string, int len, char from, char to) +{ + int i=0; + + while(ireplace); + idx = parse_regex_delim(substr, &match, &sed_cmd->string); /* determine the number of back references in the match string */ /* Note: we compute this here rather than in the do_subst_command() * function to save processor time, at the expense of a little more memory * (4 bits) per sed_cmd */ - for (j = 0; match[j]; j++) { - /* GNU/POSIX sed does not save more than nine backrefs */ - if (match[j] == '\\' && match[j + 1] == '(' - && sed_cmd->num_backrefs <= 9) - sed_cmd->num_backrefs++; - } - /* process the flags */ + + sed_cmd->which_match=1; while (substr[++idx]) { + /* Parse match number */ + if(isdigit(substr[idx])) { + if(match[0]!='^') { + /* Match 0 treated as all, multiple matches we take the last one. */ + char *pos=substr+idx; + sed_cmd->which_match=(unsigned short)strtol(substr+idx,&pos,10); + idx=pos-substr; + } + continue; + } + /* Skip spaces */ + if(isspace(substr[idx])) continue; + switch (substr[idx]) { - case 'g': - if (match[0] != '^') { - sed_cmd->sub_g = 1; + /* Replace all occurrences */ + case 'g': + if (match[0] != '^') sed_cmd->which_match = 0; + break; + /* Print pattern space */ + case 'p': + sed_cmd->sub_p = 1; + break; + case 'w': + { + char *temp; + idx+=parse_file_cmd(sed_cmd,substr+idx,&temp); + + break; } - break; - /* Hmm, i dont see the I option mentioned in the standard */ - case 'I': - cflags |= REG_ICASE; - break; - case 'p': - sed_cmd->sub_p = 1; - break; - default: - /* any whitespace or semicolon trailing after a s/// is ok */ - if (strchr(semicolon_whitespace, substr[idx])) + /* Ignore case (gnu exension) */ + case 'I': + cflags |= REG_ICASE; + break; + case ';': + case '}': goto out; - /* else */ - bb_error_msg_and_die("bad option in substitution expression"); + default: + bb_error_msg_and_die("bad option in substitution expression"); } } - - out: +out: /* compile the match string into a regex */ if (*match != '\0') { /* If match is empty, we use last regex used at runtime */ @@ -333,166 +399,68 @@ static int parse_subst_cmd(sed_cmd_t * const sed_cmd, const char *substr) return idx; } -static void replace_slash_n(char *string) -{ - char *dest; - - for (dest = string; *string; string++, dest++) { - if ((string[0] == '\\') && (string[1] == 'n')) { - *dest = '\n'; - string++; - } else { - *dest = *string; - } - } - *dest=0; -} - -static int parse_translate_cmd(sed_cmd_t * const sed_cmd, const char *cmdstr) -{ - char *match; - char *replace; - int idx; - int i; - - idx = parse_regex_delim(cmdstr, &match, &replace); - replace_slash_n(match); - replace_slash_n(replace); - sed_cmd->translate = xcalloc(1, (strlen(match) + 1) * 2); - for (i = 0; (match[i] != 0) && (replace[i] != 0); i++) { - sed_cmd->translate[i * 2] = match[i]; - sed_cmd->translate[(i * 2) + 1] = replace[i]; - } - return (idx + 1); -} - -static int parse_edit_cmd(sed_cmd_t * sed_cmd, const char *editstr) -{ - int i, j; - - /* - * the string that gets passed to this function should look like this: - * - * need one of these - * | - * | this backslash (immediately following the edit command) is mandatory - * | | - * [aic]\ - * TEXT1\ - * TEXT2\ - * TEXTN - * - * as soon as we hit a TEXT line that has no trailing '\', we're done. - * this means a command like: - * - * i\ - * INSERTME - * - * is a-ok. - * - */ - if ((*editstr != '\\') || ((editstr[1] != '\n') && (editstr[1] != '\r'))) { - bb_error_msg_and_die("bad format in edit expression"); - } - - /* store the edit line text */ - sed_cmd->editline = xmalloc(strlen(&editstr[2]) + 2); - for (i = 2, j = 0; - editstr[i] != '\0' && strchr("\r\n", editstr[i]) == NULL; i++, j++) { - if ((editstr[i] == '\\') && strchr("\n\r", editstr[i + 1]) != NULL) { - sed_cmd->editline[j] = '\n'; - i++; - } else - sed_cmd->editline[j] = editstr[i]; - } - - /* figure out if we need to add a newline */ - if (sed_cmd->editline[j - 1] != '\n') - sed_cmd->editline[j++] = '\n'; - - /* terminate string */ - sed_cmd->editline[j] = '\0'; - - return i; -} - - -static int parse_file_cmd(sed_cmd_t * sed_cmd, const char *filecmdstr) -{ - int idx = 0; - int filenamelen = 0; - - /* - * the string that gets passed to this function should look like this: - * '[ ]filename' - * | | - * | a filename - * | - * optional whitespace - - * re: the file to be read, the GNU manual says the following: "Note that - * if filename cannot be read, it is treated as if it were an empty file, - * without any error indication." Thus, all of the following commands are - * perfectly legal: - * - * sed -e '1r noexist' - * sed -e '1r ;' - * sed -e '1r' - */ - - /* the file command may be followed by whitespace; move past it. */ - while (isspace(filecmdstr[++idx])) {; - } - - /* the first non-whitespace we get is a filename. the filename ends when we - * hit a normal sed command terminator or end of string */ - filenamelen = strcspn(&filecmdstr[idx], semicolon_whitespace); - sed_cmd->filename = xmalloc(filenamelen + 1); - safe_strncpy(sed_cmd->filename, &filecmdstr[idx], filenamelen + 1); - return idx + filenamelen; -} - /* * Process the commands arguments */ -static char *parse_cmd_str(sed_cmd_t * sed_cmd, char *cmdstr) +static char *parse_cmd_args(sed_cmd_t *sed_cmd, char *cmdstr) { /* handle (s)ubstitution command */ - if (sed_cmd->cmd == 's') { - cmdstr += parse_subst_cmd(sed_cmd, cmdstr); - } + if (sed_cmd->cmd == 's') cmdstr += parse_subst_cmd(sed_cmd, cmdstr); /* handle edit cmds: (a)ppend, (i)nsert, and (c)hange */ else if (strchr("aic", sed_cmd->cmd)) { if ((sed_cmd->end_line || sed_cmd->end_match) && sed_cmd->cmd != 'c') bb_error_msg_and_die ("only a beginning address can be specified for edit commands"); - cmdstr += parse_edit_cmd(sed_cmd, cmdstr); - } + for(;;) { + if(*cmdstr=='\n' || *cmdstr=='\\') { + cmdstr++; + break; + } else if(isspace(*cmdstr)) cmdstr++; + else break; + } + sed_cmd->string = bb_xstrdup(cmdstr); + parse_escapes(sed_cmd->string,sed_cmd->string,strlen(cmdstr),0,0); + cmdstr += strlen(cmdstr); /* handle file cmds: (r)ead */ - else if (sed_cmd->cmd == 'r') { + } else if(strchr("rw", sed_cmd->cmd)) { if (sed_cmd->end_line || sed_cmd->end_match) bb_error_msg_and_die("Command only uses one address"); - cmdstr += parse_file_cmd(sed_cmd, cmdstr); - } + cmdstr += parse_file_cmd(sed_cmd, cmdstr, &sed_cmd->string); + if(sed_cmd->cmd=='w') + sed_cmd->file=bb_xfopen(sed_cmd->string,"w"); /* handle branch commands */ - else if (strchr(":bt", sed_cmd->cmd)) { + } else if (strchr(":bt", sed_cmd->cmd)) { int length; - cmdstr += strspn(cmdstr, " "); + while(isspace(*cmdstr)) cmdstr++; length = strcspn(cmdstr, semicolon_whitespace); if (length) { - sed_cmd->label = strndup(cmdstr, length); + sed_cmd->string = strndup(cmdstr, length); cmdstr += length; } } /* translation command */ else if (sed_cmd->cmd == 'y') { - cmdstr += parse_translate_cmd(sed_cmd, cmdstr); + char *match, *replace; + int i=cmdstr[0]; + + cmdstr+=parse_regex_delim(cmdstr, &match, &replace)+1; + /* \n already parsed, but \delimiter needs unescaping. */ + parse_escapes(match,match,strlen(match),i,i); + parse_escapes(replace,replace,strlen(replace),i,i); + + sed_cmd->string = xcalloc(1, (strlen(match) + 1) * 2); + for (i = 0; match[i] && replace[i]; i++) { + sed_cmd->string[i * 2] = match[i]; + sed_cmd->string[(i * 2) + 1] = replace[i]; + } + free(match); + free(replace); } /* if it wasnt a single-letter command that takes no arguments * then it must be an invalid command. */ - else if (strchr("dgGhHnNpPqx={}", sed_cmd->cmd) == 0) { + else if (strchr("dDgGhHlnNpPqx={}", sed_cmd->cmd) == 0) { bb_error_msg_and_die("Unsupported command %c", sed_cmd->cmd); } @@ -500,656 +468,683 @@ static char *parse_cmd_str(sed_cmd_t * sed_cmd, char *cmdstr) return (cmdstr); } -static char *add_cmd(char *cmdstr) -{ - sed_cmd_t *sed_cmd; - /* Skip over leading whitespace and semicolons */ - cmdstr += strspn(cmdstr, semicolon_whitespace); +/* Parse address+command sets, skipping comment lines. */ - /* if we ate the whole thing, that means there was just trailing - * whitespace or a final / no-op semicolon. either way, get out */ - if (*cmdstr == '\0') { - return (NULL); - } - - /* if this is a comment, jump past it and keep going */ - if (*cmdstr == '#') { - /* "#n" is the same as using -n on the command line */ - if (cmdstr[1] == 'n') { - be_quiet++; - } - return (strpbrk(cmdstr, "\n\r")); +void add_cmd(char *cmdstr) +{ + static char *add_cmd_line=NULL; + sed_cmd_t *sed_cmd; + int temp; + + /* Append this line to any unfinished line from last time. */ + if(add_cmd_line) { + int lastlen=strlen(add_cmd_line); + char *tmp=xmalloc(lastlen+strlen(cmdstr)+2); + + memcpy(tmp,add_cmd_line,lastlen); + tmp[lastlen]='\n'; + strcpy(tmp+lastlen+1,cmdstr); + free(add_cmd_line); + cmdstr=add_cmd_line=tmp; + } else add_cmd_line=NULL; + + /* If this line ends with backslash, request next line. */ + temp=strlen(cmdstr); + if(temp && cmdstr[temp-1]=='\\') { + if(!add_cmd_line) add_cmd_line=strdup(cmdstr); + add_cmd_line[temp-1]=0; + return; } - /* parse the command - * format is: [addr][,addr]cmd - * |----||-----||-| - * part1 part2 part3 - */ + /* Loop parsing all commands in this line. */ + while(*cmdstr) { + /* Skip leading whitespace and semicolons */ + cmdstr += strspn(cmdstr, semicolon_whitespace); - sed_cmd = xcalloc(1, sizeof(sed_cmd_t)); + /* If no more commands, exit. */ + if(!*cmdstr) break; - /* first part (if present) is an address: either a '$', a number or a /regex/ */ - cmdstr += get_address(cmdstr, &sed_cmd->beg_line, &sed_cmd->beg_match); + /* if this is a comment, jump past it and keep going */ + if (*cmdstr == '#') { + /* "#n" is the same as using -n on the command line */ + if (cmdstr[1] == 'n') be_quiet++; + if(!(cmdstr=strpbrk(cmdstr, "\n\r"))) break; + continue; + } - /* second part (if present) will begin with a comma */ - if (*cmdstr == ',') { - int idx; + /* parse the command + * format is: [addr][,addr][!]cmd + * |----||-----||-| + * part1 part2 part3 + */ - cmdstr++; - idx = get_address(cmdstr, &sed_cmd->end_line, &sed_cmd->end_match); - if (idx == 0) { - bb_error_msg_and_die("get_address: no address found in string\n" - "\t(you probably didn't check the string you passed me)"); - } - cmdstr += idx; - } + sed_cmd = xcalloc(1, sizeof(sed_cmd_t)); - /* skip whitespace before the command */ - while (isspace(*cmdstr)) { - cmdstr++; - } + /* first part (if present) is an address: either a '$', a number or a /regex/ */ + cmdstr += get_address(cmdstr, &sed_cmd->beg_line, &sed_cmd->beg_match); - /* there my be the inversion flag between part2 and part3 */ - if (*cmdstr == '!') { - sed_cmd->invert = 1; - cmdstr++; + /* second part (if present) will begin with a comma */ + if (*cmdstr == ',') { + int idx; -#ifdef SED_FEATURE_STRICT_CHECKING - /* According to the spec - * It is unspecified whether s can follow a '!' character, - * and conforming applications shall not follow a '!' character - * with s. - */ - if (isblank(cmdstr[idx]) { - bb_error_msg_and_die("blank follows '!'");} -#else - /* skip whitespace before the command */ - while (isspace(*cmdstr)) { cmdstr++; + idx = get_address(cmdstr, &sed_cmd->end_line, &sed_cmd->end_match); + if (!idx) bb_error_msg_and_die("get_address: no address found in string\n"); + cmdstr += idx; } -#endif - } - /* last part (mandatory) will be a command */ - if (*cmdstr == '\0') - bb_error_msg_and_die("missing command"); + /* skip whitespace before the command */ + while (isspace(*cmdstr)) cmdstr++; - sed_cmd->cmd = *cmdstr; - cmdstr++; + /* Check for inversion flag */ + if (*cmdstr == '!') { + sed_cmd->invert = 1; + cmdstr++; - cmdstr = parse_cmd_str(sed_cmd, cmdstr); + /* skip whitespace before the command */ + while (isspace(*cmdstr)) cmdstr++; + } - /* Add the command to the command array */ - sed_cmd_tail->next = sed_cmd; - sed_cmd_tail = sed_cmd_tail->next; + /* last part (mandatory) will be a command */ + if (!*cmdstr) bb_error_msg_and_die("missing command"); + sed_cmd->cmd = *(cmdstr++); + cmdstr = parse_cmd_args(sed_cmd, cmdstr); - return (cmdstr); -} - -static void add_cmd_str(char *cmdstr) -{ -#ifdef CONFIG_FEATURE_SED_EMBEDED_NEWLINE - char *cmdstr_ptr = cmdstr; - - /* HACK: convert "\n" to match tranlated '\n' string */ - while ((cmdstr_ptr = strstr(cmdstr_ptr, "\\n")) != NULL) { - cmdstr = xrealloc(cmdstr, strlen(cmdstr) + 2); - cmdstr_ptr = strstr(cmdstr, "\\n"); - memmove(cmdstr_ptr + 1, cmdstr_ptr, strlen(cmdstr_ptr) + 1); - cmdstr_ptr[0] = '\\'; - cmdstr_ptr += 3; + /* Add the command to the command array */ + sed_cmd_tail->next = sed_cmd; + sed_cmd_tail = sed_cmd_tail->next; } -#endif - do { - cmdstr = add_cmd(cmdstr); - } while (cmdstr && strlen(cmdstr)); -} - -static void load_cmd_file(char *filename) -{ - FILE *cmdfile; - char *line; - char *nextline; - char *e; - - cmdfile = bb_xfopen(filename, "r"); - - while ((line = bb_get_line_from_file(cmdfile)) != NULL) { - /* if a line ends with '\' it needs the next line appended to it */ - while (((e = last_char_is(line, '\n')) != NULL) - && (e > line) && (e[-1] == '\\') - && ((nextline = bb_get_line_from_file(cmdfile)) != NULL)) { - line = xrealloc(line, (e - line) + 1 + strlen(nextline) + 1); - strcat(line, nextline); - free(nextline); - } - /* eat trailing newline (if any) --if I don't do this, edit commands - * (aic) will print an extra newline */ - chomp(line); - add_cmd_str(line); - free(line); + /* If we glued multiple lines together, free the memory. */ + if(add_cmd_line) { + free(add_cmd_line); + add_cmd_line=NULL; } } struct pipeline { - char *buf; - int idx; - int len; -}; + char *buf; /* Space to hold string */ + int idx; /* Space used */ + int len; /* Space allocated */ +} pipeline; -#define PIPE_MAGIC 0x7f #define PIPE_GROW 64 -void pipe_putc(struct pipeline *const pipeline, char c) +void pipe_putc(char c) { - if (pipeline->buf[pipeline->idx] == PIPE_MAGIC) { - pipeline->buf = xrealloc(pipeline->buf, pipeline->len + PIPE_GROW); - memset(pipeline->buf + pipeline->len, 0, PIPE_GROW); - pipeline->len += PIPE_GROW; - pipeline->buf[pipeline->len - 1] = PIPE_MAGIC; + if(pipeline.idx==pipeline.len) { + pipeline.buf = xrealloc(pipeline.buf, pipeline.len + PIPE_GROW); + pipeline.len+=PIPE_GROW; } - pipeline->buf[pipeline->idx++] = (c); + pipeline.buf[pipeline.idx++] = (c); } -#define pipeputc(c) pipe_putc(pipeline, c) - -static void print_subst_w_backrefs(const char *line, const char *replace, - regmatch_t * regmatch, struct pipeline *const pipeline, int matches) +static void do_subst_w_backrefs(const char *line, const char *replace) { - int i; + int i,j; /* go through the replacement string */ for (i = 0; replace[i]; i++) { /* if we find a backreference (\1, \2, etc.) print the backref'ed * text */ - if (replace[i] == '\\' && isdigit(replace[i + 1])) { - int j; - char tmpstr[2]; - int backref; - - ++i; /* i now indexes the backref number, instead of the leading slash */ - tmpstr[0] = replace[i]; - tmpstr[1] = 0; - backref = atoi(tmpstr); + if (replace[i] == '\\' && replace[i+1]>'0' && replace[i+1]<='9') { + int backref=replace[++i]-'0'; + /* print out the text held in regmatch[backref] */ - if (backref <= matches && regmatch[backref].rm_so != -1) - for (j = regmatch[backref].rm_so; j < regmatch[backref].rm_eo; - j++) - pipeputc(line[j]); + if(regmatch[backref].rm_so != -1) + for (j = regmatch[backref].rm_so; j < regmatch[backref].rm_eo; j++) + pipe_putc(line[j]); } /* if we find a backslash escaped character, print the character */ - else if (replace[i] == '\\') { - ++i; - pipeputc(replace[i]); - } - - /* if we find an unescaped '&' print out the whole matched text. - * fortunately, regmatch[0] contains the indicies to the whole matched - * expression (kinda seems like it was designed for just such a - * purpose...) */ - else if (replace[i] == '&') { - int j; + else if (replace[i] == '\\') pipe_putc(replace[++i]); + /* if we find an unescaped '&' print out the whole matched text. */ + else if (replace[i] == '&') for (j = regmatch[0].rm_so; j < regmatch[0].rm_eo; j++) - pipeputc(line[j]); - } - /* nothing special, just print this char of the replacement string to stdout */ - else - pipeputc(replace[i]); + pipe_putc(line[j]); + /* Otherwise just output the character. */ + else pipe_putc(replace[i]); } } static int do_subst_command(sed_cmd_t * sed_cmd, char **line) { - char *hackline = *line; - struct pipeline thepipe = { NULL, 0, 0 }; - struct pipeline *const pipeline = &thepipe; + char *oldline = *line; int altered = 0; - int result; - regmatch_t *regmatch = NULL; + int match_count=0; regex_t *current_regex; + /* Handle empty regex. */ if (sed_cmd->sub_match == NULL) { current_regex = previous_regex_ptr; - } else { - previous_regex_ptr = current_regex = sed_cmd->sub_match; - } - result = regexec(current_regex, hackline, 0, NULL, 0); + if(!current_regex) + bb_error_msg_and_die("No previous regexp."); + } else previous_regex_ptr = current_regex = sed_cmd->sub_match; - /* we only proceed if the substitution 'search' expression matches */ - if (result == REG_NOMATCH) { + /* Find the first match */ + if(REG_NOMATCH==regexec(current_regex, oldline, 10, regmatch, 0)) return 0; - } - /* whaddaya know, it matched. get the number of back references */ - regmatch = xmalloc(sizeof(regmatch_t) * (sed_cmd->num_backrefs + 1)); - - /* allocate more PIPE_GROW bytes - if replaced string is larger than original */ - thepipe.len = strlen(hackline) + PIPE_GROW; - thepipe.buf = xcalloc(1, thepipe.len); - /* buffer magic */ - thepipe.buf[thepipe.len - 1] = PIPE_MAGIC; - - /* and now, as long as we've got a line to try matching and if we can match - * the search string, we make substitutions */ - while ((*hackline || !altered) - && (regexec(current_regex, hackline, sed_cmd->num_backrefs + 1, - regmatch, 0) != REG_NOMATCH)) { + /* Initialize temporary output buffer. */ + pipeline.buf=xmalloc(PIPE_GROW); + pipeline.len=PIPE_GROW; + pipeline.idx=0; + + /* Now loop through, substituting for matches */ + do { int i; + /* Work around bug in glibc regexec, demonstrated by: + echo " a.b" | busybox sed 's [^ .]* x g' + The match_count check is so not to break + echo "hi" | busybox sed 's/^/!/g' */ + if(!regmatch[0].rm_so && !regmatch[0].rm_eo && match_count) { + pipe_putc(*(oldline++)); + continue; + } + + match_count++; + + /* If we aren't interested in this match, output old line to + end of match and continue */ + if(sed_cmd->which_match && sed_cmd->which_match!=match_count) { + for(i=0;ireplace, regmatch, pipeline, - sed_cmd->num_backrefs); + do_subst_w_backrefs(oldline, sed_cmd->string); /* advance past the match */ - hackline += regmatch[0].rm_eo; + oldline += regmatch[0].rm_eo; /* flag that something has changed */ altered++; /* if we're not doing this globally, get out now */ - if (!sed_cmd->sub_g) { - break; - } - } - for (; *hackline; hackline++) - pipeputc(*hackline); - if (thepipe.buf[thepipe.idx] == PIPE_MAGIC) - thepipe.buf[thepipe.idx] = 0; + if (sed_cmd->which_match) break; + } while (*oldline && (regexec(current_regex, oldline, 10, regmatch, 0) != REG_NOMATCH)); - /* cleanup */ - free(regmatch); + /* Copy rest of string into output pipeline */ + + while(*oldline) pipe_putc(*(oldline++)); + pipe_putc(0); free(*line); - *line = thepipe.buf; + *line = pipeline.buf; return altered; } +/* Set command pointer to point to this label. (Does not handle null label.) */ static sed_cmd_t *branch_to(const char *label) { sed_cmd_t *sed_cmd; for (sed_cmd = sed_cmd_head.next; sed_cmd; sed_cmd = sed_cmd->next) { - if ((sed_cmd->cmd == ':') && (sed_cmd->label) && (strcmp(sed_cmd->label, label) == 0)) { + if ((sed_cmd->cmd == ':') && (sed_cmd->string) && (strcmp(sed_cmd->string, label) == 0)) { return (sed_cmd); } } bb_error_msg_and_die("Can't find label for jump to `%s'", label); } -static void process_file(FILE * file) +/* Append copy of string to append buffer */ +static void append(char *s) { - char *pattern_space; /* Posix requires it be able to hold at least 8192 bytes */ - char *hold_space = NULL; /* Posix requires it be able to hold at least 8192 bytes */ - static int linenum = 0; /* GNU sed does not restart counting lines at EOF */ - int altered; - int force_print; - - pattern_space = bb_get_chomped_line_from_file(file); - if (pattern_space == NULL) { - return; + struct append_list *temp=calloc(1,sizeof(struct append_list)); + + if(append_head) + append_tail=(append_tail->next=temp); + else append_head=append_tail=temp; + temp->string=strdup(s); +} + +static void flush_append(void) +{ + /* Output appended lines. */ + while(append_head) { + fprintf(nonstdout,"%s\n",append_head->string); + append_tail=append_head->next; + free(append_head->string); + free(append_head); + append_head=append_tail; } - + append_head=append_tail=NULL; +} + +/* Get next line of input, flushing append buffer and noting if we hit EOF + * without a newline on the last line. + */ +static char *get_next_line(FILE * file, int *no_newline) +{ + char *temp; + int len; + + flush_append(); + temp=bb_get_line_from_file(file); + if(temp) { + len=strlen(temp); + if(len && temp[len-1]=='\n') temp[len-1]=0; + else *no_newline=1; + } + + return temp; +} + +/* Output line of text. missing_newline means the last line output did not + end with a newline. no_newline means this line does not end with a + newline. */ + +static int puts_maybe_newline(char *s, FILE *file, int missing_newline, int no_newline) +{ + if(missing_newline) fputc('\n',file); + fputs(s,file); + if(!no_newline) fputc('\n',file); + + if(ferror(file)) { + fprintf(stderr,"Write failed.\n"); + exit(4); /* It's what gnu sed exits with... */ + } + + return no_newline; +} + +#define sed_puts(s,n) missing_newline=puts_maybe_newline(s,nonstdout,missing_newline,n) + +static void process_file(FILE *file) +{ + char *pattern_space, *next_line, *hold_space=NULL; + static int linenum = 0, missing_newline=0; + int no_newline,next_no_newline=0; + + next_line = get_next_line(file,&next_no_newline); + /* go through every line in the file */ - do { - char *next_line; + for(;;) { sed_cmd_t *sed_cmd; - int substituted = 0; - /* This enables whole blocks of commands to be mask'ed out if the lead address doesnt match */ - int block_mask = 1; + int substituted=0; + + /* Advance to next line. Stop if out of lines. */ + if(!(pattern_space=next_line)) break; + no_newline=next_no_newline; /* Read one line in advance so we can act on the last line, the '$' address */ - next_line = bb_get_chomped_line_from_file(file); + next_line = get_next_line(file,&next_no_newline); linenum++; - altered = 0; - force_print = 0; - +restart: /* for every line, go through all the commands */ for (sed_cmd = sed_cmd_head.next; sed_cmd; sed_cmd = sed_cmd->next) { - int deleted = 0; - - /* - * entry point into sedding... - */ - int matched = ( - /* no range necessary */ - (sed_cmd->beg_line == 0 && sed_cmd->end_line == 0 - && sed_cmd->beg_match == NULL - && sed_cmd->end_match == NULL) || - /* this line number is the first address we're looking for */ - (sed_cmd->beg_line > 0 && (sed_cmd->beg_line == linenum)) || - /* this line matches our first address regex */ - (sed_cmd->beg_match - && (regexec(sed_cmd->beg_match, pattern_space, 0, NULL, - 0) == 0)) || - /* we are currently within the beginning & ending address range */ - sed_cmd->still_in_range || ((sed_cmd->beg_line == -1) - && (next_line == NULL)) + int old_matched, matched; + + old_matched = sed_cmd->in_match; + + /* Determine if this command matches this line: */ + + /* Are we continuing a previous multi-line match? */ + + sed_cmd->in_match = sed_cmd->in_match + + /* Or is no range necessary? */ + || (!sed_cmd->beg_line && !sed_cmd->end_line + && !sed_cmd->beg_match && !sed_cmd->end_match) + + /* Or did we match the start of a numerical range? */ + || (sed_cmd->beg_line > 0 && (sed_cmd->beg_line == linenum)) + + /* Or does this line match our begin address regex? */ + || (sed_cmd->beg_match && + !regexec(sed_cmd->beg_match, pattern_space, 0, NULL, 0)) + + /* Or did we match last line of input? */ + || (sed_cmd->beg_line == -1 && next_line == NULL); + + /* Snapshot the value */ + + matched = sed_cmd->in_match; + + /* Is this line the end of the current match? */ + + if(matched) { + sed_cmd->in_match = !( + /* has the ending line come, or is this a single address command? */ + (sed_cmd->end_line ? + sed_cmd->end_line==-1 ? + !next_line + : sed_cmd->end_line<=linenum + : !sed_cmd->end_match) + /* or does this line matches our last address regex */ + || (sed_cmd->end_match && old_matched && (regexec(sed_cmd->end_match, pattern_space, 0, NULL, 0) == 0)) ); + } + + /* Skip blocks of commands we didn't match. */ if (sed_cmd->cmd == '{') { - block_mask = block_mask & matched; + if(sed_cmd->invert ? matched : !matched) + while(sed_cmd && sed_cmd->cmd!='}') sed_cmd=sed_cmd->next; + if(!sed_cmd) bb_error_msg_and_die("Unterminated {"); + continue; } -// matched &= block_mask; - if (sed_cmd->invert ^ (matched & block_mask)) { - /* Update last used regex incase a blank substitute BRE is found */ + /* Okay, so did this line match? */ + if (sed_cmd->invert ? !matched : matched) { + /* Update last used regex in case a blank substitute BRE is found */ if (sed_cmd->beg_match) { previous_regex_ptr = sed_cmd->beg_match; } - /* - * actual sedding - */ + /* actual sedding */ switch (sed_cmd->cmd) { - case '=': - printf("%d\n", linenum); - break; - case 'P':{ - /* Write the current pattern space upto the first newline */ - char *tmp = strchr(pattern_space, '\n'); - - if (tmp) { - *tmp = '\0'; - } - } - case 'p': /* Write the current pattern space to output */ - puts(pattern_space); - break; - case 'd': - altered++; - deleted = 1; - break; - - case 's': - - /* - * Some special cases for 's' printing to make it compliant with - * GNU sed printing behavior (aka "The -n | s///p Matrix"): - * - * -n ONLY = never print anything regardless of any successful - * substitution - * - * s///p ONLY = always print successful substitutions, even if - * the pattern_space is going to be printed anyway (pattern_space - * will be printed twice). - * - * -n AND s///p = print ONLY a successful substitution ONE TIME; - * no other lines are printed - this is the reason why the 'p' - * flag exists in the first place. - */ - -#ifdef CONFIG_FEATURE_SED_EMBEDED_NEWLINE - /* HACK: escape newlines twice so regex can match them */ - { - int offset = 0; - - while (strchr(pattern_space + offset, '\n') != NULL) { - char *tmp; - - pattern_space = - xrealloc(pattern_space, - strlen(pattern_space) + 2); - tmp = strchr(pattern_space + offset, '\n'); - memmove(tmp + 1, tmp, strlen(tmp) + 1); - tmp[0] = '\\'; - tmp[1] = 'n'; - offset = tmp - pattern_space + 2; - } - } -#endif - /* we print the pattern_space once, unless we were told to be quiet */ - substituted |= do_subst_command(sed_cmd, &pattern_space); -#ifdef CONFIG_FEATURE_SED_EMBEDED_NEWLINE - /* undo HACK: escape newlines twice so regex can match them */ + /* Print line number */ + case '=': + fprintf(nonstdout,"%d\n", linenum); + break; + + /* Write the current pattern space up to the first newline */ + case 'P': { - char *tmp = pattern_space; + char *tmp = strchr(pattern_space, '\n'); - while ((tmp = strstr(tmp, "\\n")) != NULL) { - memmove(tmp, tmp + 1, strlen(tmp + 1) + 1); - tmp[0] = '\n'; + if (tmp) { + *tmp = '\0'; + sed_puts(pattern_space,1); + *tmp = '\n'; + break; } + /* Fall Through */ } -#endif - if (!be_quiet && substituted && ((sed_cmd->next == NULL) - || (sed_cmd->next->cmd != 's'))) { - force_print = 1; - } - /* we also print the line if we were given the 'p' flag - * (this is quite possibly the second printing) */ - if ((sed_cmd->sub_p) && altered) { - puts(pattern_space); + /* Write the current pattern space to output */ + case 'p': + sed_puts(pattern_space,no_newline); + break; + /* Delete up through first newline */ + case 'D': + { + char *tmp = strchr(pattern_space,'\n'); + + if(tmp) { + tmp=bb_xstrdup(tmp+1); + free(pattern_space); + pattern_space=tmp; + goto restart; + } } - break; - case 'a': - puts(pattern_space); - fputs(sed_cmd->editline, stdout); - altered++; - break; - - case 'i': - fputs(sed_cmd->editline, stdout); - break; - - case 'c': - /* single-address case */ - if ((sed_cmd->end_match == NULL && sed_cmd->end_line == 0) - /* multi-address case */ - /* - matching text */ - || (sed_cmd->end_match - && (regexec(sed_cmd->end_match, pattern_space, 0, - NULL, 0) == 0)) - /* - matching line numbers */ - || (sed_cmd->end_line > 0 - && sed_cmd->end_line == linenum)) { - fputs(sed_cmd->editline, stdout); + /* discard this line. */ + case 'd': + goto discard_line; + + /* Substitute with regex */ + case 's': + if(do_subst_command(sed_cmd, &pattern_space)) { + substituted|=1; + + /* handle p option */ + if(sed_cmd->sub_p) + sed_puts(pattern_space,no_newline); + /* handle w option */ + if(sed_cmd->file) + sed_cmd->no_newline=puts_maybe_newline(pattern_space, sed_cmd->file, sed_cmd->no_newline, no_newline); + + } + break; + + /* Append line to linked list to be printed later */ + case 'a': + { + append(sed_cmd->string); + break; } - altered++; - break; + /* Insert text before this line */ + case 'i': + sed_puts(sed_cmd->string,1); + break; + + /* Cut and paste text (replace) */ + case 'c': + /* Only triggers on last line of a matching range. */ + if (!sed_cmd->in_match) sed_puts(sed_cmd->string,1); + goto discard_line; - case 'r':{ - FILE *outfile; + /* Read file, append contents to output */ + case 'r': + { + FILE *outfile; - outfile = fopen(sed_cmd->filename, "r"); - if (outfile) { - char *line; + outfile = fopen(sed_cmd->string, "r"); + if (outfile) { + char *line; - while ((line = - bb_get_chomped_line_from_file(outfile)) != - NULL) { - pattern_space = - xrealloc(pattern_space, - strlen(line) + strlen(pattern_space) + 2); - strcat(pattern_space, "\n"); - strcat(pattern_space, line); + while ((line = bb_get_chomped_line_from_file(outfile)) + != NULL) + append(line); + bb_xprint_and_close_file(outfile); } - bb_xprint_and_close_file(outfile); - } - } - break; - case 'q': /* Branch to end of script and quit */ - deleted = 1; - /* Exit the outer while loop */ - free(next_line); - next_line = NULL; - break; - case 'n': /* Read next line from input */ - if (next_line) { - free(pattern_space); - pattern_space = next_line; - next_line = bb_get_chomped_line_from_file(file); - linenum++; - } - break; - case 'N': /* Append the next line to the current line */ - if (next_line) { - pattern_space = - realloc(pattern_space, - strlen(pattern_space) + strlen(next_line) + 2); - strcat(pattern_space, "\n"); - strcat(pattern_space, next_line); - next_line = bb_get_chomped_line_from_file(file); - linenum++; + break; } - break; - case 't': - if (substituted) - /* Fall through */ - case 'b': + + /* Write pattern space to file. */ + case 'w': + sed_cmd->no_newline=puts_maybe_newline(pattern_space,sed_cmd->file, sed_cmd->no_newline,no_newline); + break; + + /* Read next line from input */ + case 'n': + if (!be_quiet) + sed_puts(pattern_space,no_newline); + if (next_line) { + free(pattern_space); + pattern_space = next_line; + no_newline=next_no_newline; + next_line = get_next_line(file,&next_no_newline); + linenum++; + break; + } + /* fall through */ + + /* Quit. End of script, end of input. */ + case 'q': + /* Exit the outer while loop */ + free(next_line); + next_line = NULL; + goto discard_commands; + + /* Append the next line to the current line */ + case 'N': { - if (sed_cmd->label == NULL) { - /* Jump to end of script */ - deleted = 1; + /* If no next line, jump to end of script and exit. */ + if (next_line == NULL) { + /* Jump to end of script and exit */ + free(next_line); + next_line = NULL; + goto discard_line; + /* append next_line, read new next_line. */ } else { - sed_cmd = branch_to(sed_cmd->label); + int len=strlen(pattern_space); + + pattern_space = realloc(pattern_space, len + strlen(next_line) + 2); + pattern_space[len]='\n'; + strcpy(pattern_space+len+1, next_line); + no_newline=next_no_newline; + next_line = get_next_line(file,&next_no_newline); + linenum++; } + break; } - break; - case 'y':{ - int i; - for (i = 0; pattern_space[i] != 0; i++) { - int j; + /* Test if substition worked, branch if so. */ + case 't': + if (!substituted) break; + substituted=0; + /* Fall through */ + /* Branch to label */ + case 'b': + if (!sed_cmd->string) goto discard_commands; + else sed_cmd = branch_to(sed_cmd->string); + break; + /* Transliterate characters */ + case 'y': + { + int i; + + for (i = 0; pattern_space[i]; i++) { + int j; - for (j = 0; sed_cmd->translate[j]; j += 2) { - if (pattern_space[i] == sed_cmd->translate[j]) { - pattern_space[i] = sed_cmd->translate[j + 1]; + for (j = 0; sed_cmd->string[j]; j += 2) { + if (pattern_space[i] == sed_cmd->string[j]) { + pattern_space[i] = sed_cmd->string[j + 1]; + } } } + + break; } - } - break; - case 'g': /* Replace pattern space with hold space */ - free(pattern_space); - pattern_space = strdup(hold_space); - break; - case 'G': { /* Append newline and hold space to pattern space */ - int pattern_space_size = 0; - if (pattern_space) { - pattern_space_size = strlen(pattern_space); - } - pattern_space = xrealloc(pattern_space, pattern_space_size + strlen(hold_space) + 2); - strcat(pattern_space, "\n"); - strcat(pattern_space, hold_space); - break; - } - case 'h': /* Replace hold space with pattern space */ - free(hold_space); - hold_space = strdup(pattern_space); - break; - case 'H': { /* Append newline and pattern space to hold space */ - int hold_space_size = 2; - int pattern_space_size = 0; - - if (hold_space) { - hold_space_size += strlen(hold_space); - } - if (pattern_space) { - pattern_space_size = strlen(pattern_space); + case 'g': /* Replace pattern space with hold space */ + free(pattern_space); + if (hold_space) { + pattern_space = strdup(hold_space); + no_newline=0; + } + break; + case 'G': /* Append newline and hold space to pattern space */ + { + int pattern_space_size = 2; + int hold_space_size = 0; + + if (pattern_space) + pattern_space_size += strlen(pattern_space); + if (hold_space) hold_space_size = strlen(hold_space); + pattern_space = xrealloc(pattern_space, pattern_space_size + hold_space_size); + if (pattern_space_size == 2) pattern_space[0]=0; + strcat(pattern_space, "\n"); + if (hold_space) strcat(pattern_space, hold_space); + no_newline=0; + + break; } - hold_space = xrealloc(hold_space, hold_space_size + pattern_space_size); + case 'h': /* Replace hold space with pattern space */ + free(hold_space); + hold_space = strdup(pattern_space); + break; + case 'H': /* Append newline and pattern space to hold space */ + { + int hold_space_size = 2; + int pattern_space_size = 0; - if (hold_space_size == 2) { - strcpy(hold_space, "\n"); - } else { + if (hold_space) hold_space_size += strlen(hold_space); + if (pattern_space) + pattern_space_size = strlen(pattern_space); + hold_space = xrealloc(hold_space, + hold_space_size + pattern_space_size); + + if (hold_space_size == 2) hold_space[0]=0; strcat(hold_space, "\n"); + if (pattern_space) strcat(hold_space, pattern_space); + + break; } - if (pattern_space) { - strcat(hold_space, pattern_space); + case 'x': /* Exchange hold and pattern space */ + { + char *tmp = pattern_space; + pattern_space = hold_space; + no_newline=0; + hold_space = tmp; + break; } - break; - } - case 'x':{ - /* Swap hold and pattern space */ - char *tmp = pattern_space; - pattern_space = hold_space; - hold_space = tmp; - break; } - } - } - - /* - * exit point from sedding... - */ - if (matched) { - if ( - /* this is a single-address command or... */ - (sed_cmd->end_line == 0 && sed_cmd->end_match == NULL) - /* If only one address */ - /* we were in the middle of our address range (this - * isn't the first time through) and.. */ - || ((sed_cmd->still_in_range == 1) - /* this line number is the last address we're looking for or... */ - && ((sed_cmd->end_line > 0 - && (sed_cmd->end_line == linenum)) - /* this line matches our last address regex */ - || (sed_cmd->end_match - && (regexec(sed_cmd->end_match, pattern_space, - 0, NULL, 0) == 0))))) { - /* we're out of our address range */ - sed_cmd->still_in_range = 0; - } else { - /* didn't hit the exit? then we're still in the middle of an address range */ - sed_cmd->still_in_range = 1; - } - } - - if (sed_cmd->cmd == '}') { - block_mask = 1; } - - if (deleted) - break; - } - /* we will print the line unless we were told to be quiet or if the - * line was altered (via a 'd'elete or 's'ubstitution), in which case - * the altered line was already printed */ - if ((!be_quiet && !altered && !substituted) || force_print) { - puts(pattern_space); - } + /* + * exit point from sedding... + */ +discard_commands: + /* we will print the line unless we were told to be quiet ('-n') + or if the line was suppressed (ala 'd'elete) */ + if (!be_quiet) sed_puts(pattern_space,no_newline); + + /* Delete and such jump here. */ +discard_line: + flush_append(); free(pattern_space); - pattern_space = next_line; - } while (pattern_space); + } +} + +/* It is possible to have a command line argument with embedded + newlines. This counts as multiple command lines. */ + +static void add_cmd_block(char *cmdstr) +{ + int go=1; + char *temp=bb_xstrdup(cmdstr),*temp2=temp; + + while(go) { + int len=strcspn(temp2,"\n"); + if(!temp2[len]) go=0; + else temp2[len]=0; + add_cmd(temp2); + temp2+=len+1; + } + free(temp); } extern int sed_main(int argc, char **argv) { - int opt, status = EXIT_SUCCESS; + int status = EXIT_SUCCESS; + int opt; + uint8_t getpat = 1; -#if 0 /* This doesnt seem to be working */ #ifdef CONFIG_FEATURE_CLEAN_UP /* destroy command strings on exit */ - if (atexit(destroy_cmd_strs) == -1) + if (atexit(free_and_close_stuff) == -1) bb_perror_msg_and_die("atexit"); #endif + +#define LIE_TO_AUTOCONF +#ifdef LIE_TO_AUTOCONF + if(argc==2 && !strcmp(argv[1],"--version")) { + printf("This is not GNU sed version 4.0\n"); + exit(0); + } #endif /* do normal option parsing */ - while ((opt = getopt(argc, argv, "ne:f:")) > 0) { + while ((opt = getopt(argc, argv, "ine:f:")) > 0) { switch (opt) { + case 'i': + in_place++; + atexit(cleanup_outname); + break; case 'n': be_quiet++; break; - case 'e':{ - add_cmd_str(optarg); + case 'e': + add_cmd_block(optarg); + getpat=0; break; - } case 'f': - load_cmd_file(optarg); + { + FILE *cmdfile; + char *line; + + cmdfile = bb_xfopen(optarg, "r"); + + while ((line = bb_get_chomped_line_from_file(cmdfile)) + != NULL) { + add_cmd(line); + getpat=0; + free(line); + } + bb_xprint_and_close_file(cmdfile); + break; + } default: bb_show_usage(); } @@ -1157,30 +1152,60 @@ extern int sed_main(int argc, char **argv) /* if we didn't get a pattern from a -e and no command file was specified, * argv[optind] should be the pattern. no pattern, no worky */ - if (sed_cmd_head.next == NULL) { + if(getpat) { if (argv[optind] == NULL) bb_show_usage(); else - add_cmd_str(argv[optind++]); + add_cmd_block(argv[optind++]); } + /* Flush any unfinished commands. */ + add_cmd(""); + + /* By default, we write to stdout */ + nonstdout=stdout; /* argv[(optind)..(argc-1)] should be names of file to process. If no * files were specified or '-' was specified, take input from stdin. * Otherwise, we process all the files specified. */ if (argv[optind] == NULL) { + if(in_place) { + fprintf(stderr,"sed: Filename required for -i\n"); + exit(1); + } process_file(stdin); } else { int i; FILE *file; for (i = optind; i < argc; i++) { - if(!strcmp(argv[i], "-")) { + if(!strcmp(argv[i], "-") && !in_place) { process_file(stdin); } else { file = bb_wfopen(argv[i], "r"); if (file) { + if(in_place) { + struct stat statbuf; + outname=bb_xstrndup(argv[i],strlen(argv[i])+6); + strcat(outname,"XXXXXX"); + /* Set permissions of output file */ + fstat(fileno(file),&statbuf); + mkstemp(outname); + nonstdout=bb_wfopen(outname,"w"); + /* Set permissions of output file */ + fstat(fileno(file),&statbuf); + fchmod(fileno(nonstdout),statbuf.st_mode); + atexit(cleanup_outname); + } process_file(file); fclose(file); + if(in_place) { + fclose(nonstdout); + nonstdout=stdout; + unlink(argv[i]); + rename(outname,argv[i]); + free(outname); + outname=0; + } } else { status = EXIT_FAILURE; } @@ -1188,8 +1213,5 @@ extern int sed_main(int argc, char **argv) } } -#ifdef CONFIG_FEATURE_CLEAN_UP - destroy_cmd_strs(); -#endif return status; }