blob: eb31f7d2e19630b1e105001b8eb80c673182c1ac [file] [log] [blame]
Glenn L McGrathaa5a6022003-10-01 03:06:16 +00001/* vi: set sw=4 ts=4: */
Eric Andersen6b6b3f61999-10-28 16:06:25 +00002/*
Mark Whitley6315ce62000-07-10 22:55:51 +00003 * sed.c - very minimalist version of sed
Eric Andersen6b6b3f61999-10-28 16:06:25 +00004 *
Eric Andersenbdfd0d72001-10-24 05:00:29 +00005 * Copyright (C) 1999,2000,2001 by Lineo, inc. and Mark Whitley
6 * Copyright (C) 1999,2000,2001 by Mark Whitley <markw@codepoet.org>
Matt Kraai5ed78ad2002-01-03 21:12:34 +00007 * Copyright (C) 2002 Matt Kraai
Denis Vlasenko0beaff82007-09-21 13:16:32 +00008 * Copyright (C) 2003 by Glenn McGrath
Rob Landley25d82392004-04-01 09:23:30 +00009 * Copyright (C) 2003,2004 by Rob Landley <rob@landley.net>
Erik Andersen1266a131999-12-29 22:19:46 +000010 *
Rob Landley2b26fd52006-02-24 02:30:39 +000011 * MAINTAINER: Rob Landley <rob@landley.net>
Bernhard Reutner-Fischerb7f39732006-03-01 20:14:16 +000012 *
Rob Landleye9a7a622006-09-22 02:52:41 +000013 * Licensed under GPL version 2, see file LICENSE in this tarball for details.
Eric Andersen6b6b3f61999-10-28 16:06:25 +000014 */
15
Glenn L McGrathaa5a6022003-10-01 03:06:16 +000016/* Code overview.
17
18 Files are laid out to avoid unnecessary function declarations. So for
19 example, every function add_cmd calls occurs before add_cmd in this file.
20
21 add_cmd() is called on each line of sed command text (from a file or from
22 the command line). It calls get_address() and parse_cmd_args(). The
23 resulting sed_cmd_t structures are appended to a linked list
Denis Vlasenko17a15262007-03-26 20:48:46 +000024 (G.sed_cmd_head/G.sed_cmd_tail).
Glenn L McGrathaa5a6022003-10-01 03:06:16 +000025
Denis Vlasenkodefc1ea2008-06-27 02:52:20 +000026 add_input_file() adds a FILE* to the list of input files. We need to
Rob Landleyfae1dc82005-11-20 07:44:35 +000027 know all input sources ahead of time to find the last line for the $ match.
Rob Landleydcc28662004-11-25 07:21:47 +000028
29 process_files() does actual sedding, reading data lines from each input FILE *
Glenn L McGrathaa5a6022003-10-01 03:06:16 +000030 (which could be stdin) and applying the sed command list (sed_cmd_head) to
31 each of the resulting lines.
32
33 sed_main() is where external code calls into this, with a command line.
34*/
35
36
Mark Whitley6315ce62000-07-10 22:55:51 +000037/*
38 Supported features and commands in this version of sed:
39
40 - comments ('#')
Mark Whitley94074a92000-07-14 00:00:15 +000041 - address matching: num|/matchstr/[,num|/matchstr/|$]command
42 - commands: (p)rint, (d)elete, (s)ubstitue (with g & I flags)
43 - edit commands: (a)ppend, (i)nsert, (c)hange
Mark Whitley1f3b9f22001-05-11 22:27:13 +000044 - file commands: (r)ead
Rob Landleyc63fe912005-10-30 10:08:13 +000045 - backreferences in substitution expressions (\0, \1, \2...\9)
Glenn L McGrathf50ce312003-03-09 15:12:24 +000046 - grouped commands: {cmd1;cmd2}
Glenn L McGrathbd9b32b2003-04-09 01:43:54 +000047 - transliteration (y/source-chars/dest-chars/)
48 - pattern space hold space storing / swapping (g, h, x)
Rob Landley93850a52005-05-18 06:34:37 +000049 - labels / branching (: label, b, t, T)
Glenn L McGrathf50ce312003-03-09 15:12:24 +000050
Mark Whitley6315ce62000-07-10 22:55:51 +000051 (Note: Specifying an address (range) to match is *optional*; commands
52 default to the whole pattern space if no specific address match was
53 requested.)
54
Glenn L McGrath2570b432003-09-16 05:25:43 +000055 Todo:
Glenn L McGrath2570b432003-09-16 05:25:43 +000056 - Create a wrapper around regex to make libc's regex conform with sed
Glenn L McGrathbd9b32b2003-04-09 01:43:54 +000057
Glenn L McGrathd5eadea2003-03-09 02:39:29 +000058 Reference http://www.opengroup.org/onlinepubs/007904975/utilities/sed.html
Mark Whitley6315ce62000-07-10 22:55:51 +000059*/
60
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000061#include "libbb.h"
"Vladimir N. Oleynik"23f62fc2005-09-14 16:59:11 +000062#include "xregex.h"
Mark Whitley6315ce62000-07-10 22:55:51 +000063
Rob Landleye3f5a3f2006-05-09 03:53:55 +000064/* Each sed command turns into one of these structures. */
Glenn L McGrathe7a8bc92003-03-09 10:23:57 +000065typedef struct sed_cmd_s {
Denis Vlasenkod18a3a22006-10-25 12:46:03 +000066 /* Ordered by alignment requirements: currently 36 bytes on x86 */
Denis Vlasenko34c4e5f2007-01-30 22:25:16 +000067 struct sed_cmd_s *next; /* Next command (linked list, NULL terminated) */
Mark Whitley2dc192f2000-11-03 19:47:00 +000068
Denis Vlasenkod18a3a22006-10-25 12:46:03 +000069 /* address storage */
70 regex_t *beg_match; /* sed -e '/match/cmd' */
71 regex_t *end_match; /* sed -e '/match/,/end_match/cmd' */
72 regex_t *sub_match; /* For 's/sub_match/string/' */
73 int beg_line; /* 'sed 1p' 0 == apply commands to all lines */
74 int end_line; /* 'sed 1,3p' 0 == one line only. -1 = last line ($) */
Bernhard Reutner-Fischerb7f39732006-03-01 20:14:16 +000075
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +000076 FILE *sw_file; /* File (sw) command writes to, -1 for none. */
Denis Vlasenkod18a3a22006-10-25 12:46:03 +000077 char *string; /* Data string for (saicytb) commands. */
Glenn L McGrathfc4cb4d2003-04-12 16:10:42 +000078
Denis Vlasenko284d0fa2008-02-16 13:18:17 +000079 unsigned which_match; /* (s) Which match to replace (0 for all) */
Glenn L McGrathfc4cb4d2003-04-12 16:10:42 +000080
Denis Vlasenkod18a3a22006-10-25 12:46:03 +000081 /* Bitfields (gcc won't group them if we don't) */
Denis Vlasenko34c4e5f2007-01-30 22:25:16 +000082 unsigned invert:1; /* the '!' after the address */
83 unsigned in_match:1; /* Next line also included in match? */
84 unsigned sub_p:1; /* (s) print option */
Denis Vlasenko1375bc72006-12-02 20:12:12 +000085
Denis Vlasenko34c4e5f2007-01-30 22:25:16 +000086 char sw_last_char; /* Last line written by (sw) had no '\n' */
Glenn L McGrathc18ce372003-09-13 06:57:39 +000087
Denis Vlasenkod18a3a22006-10-25 12:46:03 +000088 /* GENERAL FIELDS */
89 char cmd; /* The command char: abcdDgGhHilnNpPqrstwxy:={} */
Glenn L McGrathe7a8bc92003-03-09 10:23:57 +000090} sed_cmd_t;
Mark Whitley6315ce62000-07-10 22:55:51 +000091
Denis Vlasenko6ca409e2007-08-12 20:58:27 +000092static const char semicolon_whitespace[] ALIGN1 = "; \n\r\t\v";
Glenn L McGrathaa5a6022003-10-01 03:06:16 +000093
Denis Vlasenko17a15262007-03-26 20:48:46 +000094struct globals {
Rob Landleye3f5a3f2006-05-09 03:53:55 +000095 /* options */
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +000096 int be_quiet, regex_type;
Rob Landleye3f5a3f2006-05-09 03:53:55 +000097 FILE *nonstdout;
98 char *outname, *hold_space;
Glenn L McGrathbd9b32b2003-04-09 01:43:54 +000099
Rob Landleye3f5a3f2006-05-09 03:53:55 +0000100 /* List of input files */
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000101 int input_file_count, current_input_file;
Rob Landleye3f5a3f2006-05-09 03:53:55 +0000102 FILE **input_file_list;
Eric Andersen6b6b3f61999-10-28 16:06:25 +0000103
Rob Landleye3f5a3f2006-05-09 03:53:55 +0000104 regmatch_t regmatch[10];
105 regex_t *previous_regex_ptr;
Denis Vlasenko9213a9e2006-09-17 16:28:10 +0000106
Rob Landleye3f5a3f2006-05-09 03:53:55 +0000107 /* linked list of sed commands */
108 sed_cmd_t sed_cmd_head, *sed_cmd_tail;
109
110 /* Linked list of append lines */
111 llist_t *append_head;
112
113 char *add_cmd_line;
114
115 struct pipeline {
116 char *buf; /* Space to hold string */
117 int idx; /* Space used */
118 int len; /* Space allocated */
119 } pipeline;
Denis Vlasenko17a15262007-03-26 20:48:46 +0000120};
Denis Vlasenko17a15262007-03-26 20:48:46 +0000121#define G (*(struct globals*)&bb_common_bufsiz1)
Denis Vlasenko74324c82007-06-04 10:16:52 +0000122void BUG_sed_globals_too_big(void);
123#define INIT_G() do { \
124 if (sizeof(struct globals) > COMMON_BUFSIZE) \
125 BUG_sed_globals_too_big(); \
126 G.sed_cmd_tail = &G.sed_cmd_head; \
127} while (0)
128
Glenn L McGrathbd9b32b2003-04-09 01:43:54 +0000129
Bernhard Reutner-Fischerb7f39732006-03-01 20:14:16 +0000130#if ENABLE_FEATURE_CLEAN_UP
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000131static void sed_free_and_close_stuff(void)
Mark Whitley6315ce62000-07-10 22:55:51 +0000132{
Denis Vlasenko17a15262007-03-26 20:48:46 +0000133 sed_cmd_t *sed_cmd = G.sed_cmd_head.next;
Mark Whitley6315ce62000-07-10 22:55:51 +0000134
Denis Vlasenko17a15262007-03-26 20:48:46 +0000135 llist_free(G.append_head, free);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000136
Glenn L McGrath56c633c2003-03-28 04:23:23 +0000137 while (sed_cmd) {
Glenn L McGrathbd9b32b2003-04-09 01:43:54 +0000138 sed_cmd_t *sed_cmd_next = sed_cmd->next;
Mark Whitley6315ce62000-07-10 22:55:51 +0000139
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000140 if (sed_cmd->sw_file)
141 xprint_and_close_file(sed_cmd->sw_file);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000142
Glenn L McGrath56c633c2003-03-28 04:23:23 +0000143 if (sed_cmd->beg_match) {
144 regfree(sed_cmd->beg_match);
145 free(sed_cmd->beg_match);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000146 }
Glenn L McGrath56c633c2003-03-28 04:23:23 +0000147 if (sed_cmd->end_match) {
148 regfree(sed_cmd->end_match);
149 free(sed_cmd->end_match);
Mark Whitley6315ce62000-07-10 22:55:51 +0000150 }
Glenn L McGrath56c633c2003-03-28 04:23:23 +0000151 if (sed_cmd->sub_match) {
152 regfree(sed_cmd->sub_match);
153 free(sed_cmd->sub_match);
Mark Whitley6315ce62000-07-10 22:55:51 +0000154 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000155 free(sed_cmd->string);
Glenn L McGrath56c633c2003-03-28 04:23:23 +0000156 free(sed_cmd);
157 sed_cmd = sed_cmd_next;
Mark Whitley6315ce62000-07-10 22:55:51 +0000158 }
Rob Landleyce4f0e92004-10-30 06:54:19 +0000159
Denis Vlasenko60818682007-09-28 22:07:23 +0000160 free(G.hold_space);
Rob Landleydcc28662004-11-25 07:21:47 +0000161
Denis Vlasenko17a15262007-03-26 20:48:46 +0000162 while (G.current_input_file < G.input_file_count)
163 fclose(G.input_file_list[G.current_input_file++]);
Mark Whitley6315ce62000-07-10 22:55:51 +0000164}
Denis Vlasenko666da5e2006-12-26 18:17:42 +0000165#else
166void sed_free_and_close_stuff(void);
Mark Whitleyc41e8c82000-07-12 23:35:21 +0000167#endif
Mark Whitley6315ce62000-07-10 22:55:51 +0000168
Rob Landley53302f82004-02-18 09:54:15 +0000169/* If something bad happens during -i operation, delete temp file */
170
171static void cleanup_outname(void)
172{
Denis Vlasenko17a15262007-03-26 20:48:46 +0000173 if (G.outname) unlink(G.outname);
Rob Landley53302f82004-02-18 09:54:15 +0000174}
175
Denis Vlasenko40276642007-11-13 16:48:10 +0000176/* strcpy, replacing "\from" with 'to'. If to is NUL, replacing "\any" with 'any' */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000177
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000178static void parse_escapes(char *dest, const char *string, int len, char from, char to)
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000179{
Denis Vlasenko80778502006-10-25 12:46:46 +0000180 int i = 0;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000181
Denis Vlasenko80778502006-10-25 12:46:46 +0000182 while (i < len) {
183 if (string[i] == '\\') {
184 if (!to || string[i+1] == from) {
Denis Vlasenko4ebaf102007-01-19 21:33:19 +0000185 *dest++ = to ? to : string[i+1];
Denis Vlasenko80778502006-10-25 12:46:46 +0000186 i += 2;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000187 continue;
Denis Vlasenko4ebaf102007-01-19 21:33:19 +0000188 }
189 *dest++ = string[i++];
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000190 }
Denis Vlasenko40276642007-11-13 16:48:10 +0000191 /* TODO: is it safe wrt a string with trailing '\\' ? */
Denis Vlasenko4ebaf102007-01-19 21:33:19 +0000192 *dest++ = string[i++];
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000193 }
Denis Vlasenko40276642007-11-13 16:48:10 +0000194 *dest = '\0';
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000195}
196
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000197static char *copy_parsing_escapes(const char *string, int len)
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000198{
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000199 char *dest = xmalloc(len + 1);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000200
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000201 parse_escapes(dest, string, len, 'n', '\n');
Denis Vlasenko40276642007-11-13 16:48:10 +0000202 /* GNU sed also recognizes \t */
203 parse_escapes(dest, dest, strlen(dest), 't', '\t');
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000204 return dest;
205}
206
207
Mark Whitley6315ce62000-07-10 22:55:51 +0000208/*
Eric Andersen28b3c532001-01-02 11:01:31 +0000209 * index_of_next_unescaped_regexp_delim - walks left to right through a string
210 * beginning at a specified index and returns the index of the next regular
Denis Vlasenko40276642007-11-13 16:48:10 +0000211 * expression delimiter (typically a forward slash ('/')) not preceded by
Rob Landley4795e4e2006-07-26 17:25:08 +0000212 * a backslash ('\'). A negative delimiter disables square bracket checking.
Mark Whitley6315ce62000-07-10 22:55:51 +0000213 */
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000214static int index_of_next_unescaped_regexp_delim(int delimiter, const char *str)
Mark Whitley6315ce62000-07-10 22:55:51 +0000215{
Matt Kraaia0065d52001-08-24 14:45:50 +0000216 int bracket = -1;
217 int escaped = 0;
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000218 int idx = 0;
Eric Andersenc52a6b02001-11-10 10:49:42 +0000219 char ch;
Matt Kraaia0065d52001-08-24 14:45:50 +0000220
Rob Landley4795e4e2006-07-26 17:25:08 +0000221 if (delimiter < 0) {
222 bracket--;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000223 delimiter = -delimiter;
Rob Landley4795e4e2006-07-26 17:25:08 +0000224 }
225
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000226 for (; (ch = str[idx]); idx++) {
Rob Landley4795e4e2006-07-26 17:25:08 +0000227 if (bracket >= 0) {
Glenn L McGrathd4185b02003-04-11 17:10:23 +0000228 if (ch == ']' && !(bracket == idx - 1 || (bracket == idx - 2
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000229 && str[idx - 1] == '^')))
Matt Kraaia0065d52001-08-24 14:45:50 +0000230 bracket = -1;
231 } else if (escaped)
232 escaped = 0;
Eric Andersenc52a6b02001-11-10 10:49:42 +0000233 else if (ch == '\\')
Matt Kraaia0065d52001-08-24 14:45:50 +0000234 escaped = 1;
Rob Landley4795e4e2006-07-26 17:25:08 +0000235 else if (bracket == -1 && ch == '[')
Matt Kraaia0065d52001-08-24 14:45:50 +0000236 bracket = idx;
Glenn L McGrath1fb44672003-03-09 08:44:49 +0000237 else if (ch == delimiter)
Mark Whitley97562bd2000-07-17 20:06:42 +0000238 return idx;
239 }
Mark Whitley6315ce62000-07-10 22:55:51 +0000240
Mark Whitley97562bd2000-07-17 20:06:42 +0000241 /* if we make it to here, we've hit the end of the string */
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000242 bb_error_msg_and_die("unmatched '%c'", delimiter);
Mark Whitley6315ce62000-07-10 22:55:51 +0000243}
244
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000245/*
246 * Returns the index of the third delimiter
247 */
Denis Vlasenko9356b502007-01-29 23:44:38 +0000248static int parse_regex_delim(const char *cmdstr, char **match, char **replace)
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000249{
Denis Vlasenko9356b502007-01-29 23:44:38 +0000250 const char *cmdstr_ptr = cmdstr;
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000251 char delimiter;
252 int idx = 0;
253
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000254 /* verify that the 's' or 'y' is followed by something. That something
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000255 * (typically a 'slash') is now our regexp delimiter... */
Rob Landley4795e4e2006-07-26 17:25:08 +0000256 if (*cmdstr == '\0')
257 bb_error_msg_and_die("bad format in substitution expression");
Denis Vlasenko80778502006-10-25 12:46:46 +0000258 delimiter = *cmdstr_ptr++;
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000259
260 /* save the match string */
261 idx = index_of_next_unescaped_regexp_delim(delimiter, cmdstr_ptr);
Rob Landleyfae1dc82005-11-20 07:44:35 +0000262 *match = copy_parsing_escapes(cmdstr_ptr, idx);
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000263
264 /* save the replacement string */
265 cmdstr_ptr += idx + 1;
Rob Landley4795e4e2006-07-26 17:25:08 +0000266 idx = index_of_next_unescaped_regexp_delim(-delimiter, cmdstr_ptr);
Rob Landleyfae1dc82005-11-20 07:44:35 +0000267 *replace = copy_parsing_escapes(cmdstr_ptr, idx);
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000268
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000269 return ((cmdstr_ptr - cmdstr) + idx);
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000270}
271
Mark Whitley6315ce62000-07-10 22:55:51 +0000272/*
273 * returns the index in the string just past where the address ends.
274 */
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000275static int get_address(const char *my_str, int *linenum, regex_t ** regex)
Mark Whitley6315ce62000-07-10 22:55:51 +0000276{
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000277 const char *pos = my_str;
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000278
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000279 if (isdigit(*my_str)) {
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000280 *linenum = strtol(my_str, (char**)&pos, 10);
Glenn L McGrath1fb44672003-03-09 08:44:49 +0000281 /* endstr shouldnt ever equal NULL */
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000282 } else if (*my_str == '$') {
Mark Whitley0915c4b2001-06-11 23:50:06 +0000283 *linenum = -1;
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000284 pos++;
285 } else if (*my_str == '/' || *my_str == '\\') {
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000286 int next;
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000287 char delimiter;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000288 char *temp;
Glenn L McGrath1fb44672003-03-09 08:44:49 +0000289
Denis Vlasenko80778502006-10-25 12:46:46 +0000290 delimiter = '/';
291 if (*my_str == '\\') delimiter = *++pos;
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000292 next = index_of_next_unescaped_regexp_delim(delimiter, ++pos);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000293 temp = copy_parsing_escapes(pos, next);
Denis Vlasenko80778502006-10-25 12:46:46 +0000294 *regex = xmalloc(sizeof(regex_t));
Denis Vlasenko17a15262007-03-26 20:48:46 +0000295 xregcomp(*regex, temp, G.regex_type|REG_NEWLINE);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000296 free(temp);
297 /* Move position to next character after last delimiter */
Rob Landley4795e4e2006-07-26 17:25:08 +0000298 pos += (next+1);
Mark Whitley6315ce62000-07-10 22:55:51 +0000299 }
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000300 return pos - my_str;
Mark Whitley6315ce62000-07-10 22:55:51 +0000301}
Erik Andersene49d5ec2000-02-08 19:58:47 +0000302
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000303/* Grab a filename. Whitespace at start is skipped, then goes to EOL. */
Denis Vlasenko68404f12008-03-17 09:00:54 +0000304static int parse_file_cmd(/*sed_cmd_t *sed_cmd,*/ const char *filecmdstr, char **retval)
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000305{
Denis Vlasenko80778502006-10-25 12:46:46 +0000306 int start = 0, idx, hack = 0;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000307
308 /* Skip whitespace, then grab filename to end of line */
Denis Vlasenko9356b502007-01-29 23:44:38 +0000309 while (isspace(filecmdstr[start]))
310 start++;
Denis Vlasenko80778502006-10-25 12:46:46 +0000311 idx = start;
Denis Vlasenko9356b502007-01-29 23:44:38 +0000312 while (filecmdstr[idx] && filecmdstr[idx] != '\n')
313 idx++;
Denis Vlasenko80778502006-10-25 12:46:46 +0000314
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000315 /* If lines glued together, put backslash back. */
Denis Vlasenko9356b502007-01-29 23:44:38 +0000316 if (filecmdstr[idx] == '\n')
317 hack = 1;
Denis Vlasenko80778502006-10-25 12:46:46 +0000318 if (idx == start)
319 bb_error_msg_and_die("empty filename");
Rob Landleyd921b2e2006-08-03 15:41:12 +0000320 *retval = xstrndup(filecmdstr+start, idx-start+hack+1);
Denis Vlasenko9356b502007-01-29 23:44:38 +0000321 if (hack)
322 (*retval)[idx] = '\\';
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000323
324 return idx;
325}
326
Denis Vlasenko9356b502007-01-29 23:44:38 +0000327static int parse_subst_cmd(sed_cmd_t *sed_cmd, const char *substr)
Mark Whitley06f35292000-07-13 19:58:04 +0000328{
Denis Vlasenko17a15262007-03-26 20:48:46 +0000329 int cflags = G.regex_type;
Mark Whitley06f35292000-07-13 19:58:04 +0000330 char *match;
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000331 int idx;
Mark Whitley06f35292000-07-13 19:58:04 +0000332
333 /*
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000334 * A substitution command should look something like this:
335 * s/match/replace/ #gIpw
Mark Whitley0e4cec02000-08-21 21:29:20 +0000336 * || | |||
Mark Whitley06f35292000-07-13 19:58:04 +0000337 * mandatory optional
Mark Whitley06f35292000-07-13 19:58:04 +0000338 */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000339 idx = parse_regex_delim(substr, &match, &sed_cmd->string);
Mark Whitley06f35292000-07-13 19:58:04 +0000340
Mark Whitley97562bd2000-07-17 20:06:42 +0000341 /* determine the number of back references in the match string */
342 /* Note: we compute this here rather than in the do_subst_command()
343 * function to save processor time, at the expense of a little more memory
344 * (4 bits) per sed_cmd */
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000345
Mark Whitley06f35292000-07-13 19:58:04 +0000346 /* process the flags */
Mark Whitley70705d72000-07-14 19:06:30 +0000347
Denis Vlasenko80778502006-10-25 12:46:46 +0000348 sed_cmd->which_match = 1;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000349 while (substr[++idx]) {
350 /* Parse match number */
Denis Vlasenko80778502006-10-25 12:46:46 +0000351 if (isdigit(substr[idx])) {
352 if (match[0] != '^') {
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000353 /* Match 0 treated as all, multiple matches we take the last one. */
Denis Vlasenko9356b502007-01-29 23:44:38 +0000354 const char *pos = substr + idx;
355/* FIXME: error check? */
Denis Vlasenko284d0fa2008-02-16 13:18:17 +0000356 sed_cmd->which_match = (unsigned)strtol(substr+idx, (char**) &pos, 10);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000357 idx = pos - substr;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000358 }
359 continue;
360 }
Rob Landley40ec4ae2004-01-04 06:42:14 +0000361 /* Skip spaces */
Denis Vlasenko80778502006-10-25 12:46:46 +0000362 if (isspace(substr[idx])) continue;
Rob Landley40ec4ae2004-01-04 06:42:14 +0000363
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000364 switch (substr[idx]) {
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000365 /* Replace all occurrences */
366 case 'g':
Denis Vlasenko284d0fa2008-02-16 13:18:17 +0000367 if (match[0] != '^')
368 sed_cmd->which_match = 0;
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000369 break;
370 /* Print pattern space */
371 case 'p':
372 sed_cmd->sub_p = 1;
373 break;
374 /* Write to file */
375 case 'w':
376 {
377 char *temp;
Denis Vlasenko68404f12008-03-17 09:00:54 +0000378 idx += parse_file_cmd(/*sed_cmd,*/ substr+idx, &temp);
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000379 break;
380 }
381 /* Ignore case (gnu exension) */
382 case 'I':
383 cflags |= REG_ICASE;
384 break;
385 /* Comment */
386 case '#':
Denis Vlasenko80778502006-10-25 12:46:46 +0000387 while (substr[++idx]) /*skip all*/;
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000388 /* Fall through */
389 /* End of command */
390 case ';':
391 case '}':
392 goto out;
393 default:
394 bb_error_msg_and_die("bad option in substitution expression");
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000395 }
396 }
Glenn L McGrath2570b432003-09-16 05:25:43 +0000397out:
Mark Whitley97562bd2000-07-17 20:06:42 +0000398 /* compile the match string into a regex */
Glenn L McGrathc1d95072003-04-08 06:42:45 +0000399 if (*match != '\0') {
400 /* If match is empty, we use last regex used at runtime */
Denis Vlasenkob95636c2006-12-19 23:36:04 +0000401 sed_cmd->sub_match = xmalloc(sizeof(regex_t));
Glenn L McGrathc1d95072003-04-08 06:42:45 +0000402 xregcomp(sed_cmd->sub_match, match, cflags);
403 }
Mark Whitley06f35292000-07-13 19:58:04 +0000404 free(match);
Mark Whitley70705d72000-07-14 19:06:30 +0000405
406 return idx;
Mark Whitley06f35292000-07-13 19:58:04 +0000407}
408
Glenn L McGrath2971ef12003-03-18 01:19:23 +0000409/*
410 * Process the commands arguments
411 */
Denis Vlasenko9356b502007-01-29 23:44:38 +0000412static const char *parse_cmd_args(sed_cmd_t *sed_cmd, const char *cmdstr)
Mark Whitley6315ce62000-07-10 22:55:51 +0000413{
Glenn L McGrath2f8a4012003-03-09 02:44:49 +0000414 /* handle (s)ubstitution command */
Denis Vlasenko80778502006-10-25 12:46:46 +0000415 if (sed_cmd->cmd == 's')
416 cmdstr += parse_subst_cmd(sed_cmd, cmdstr);
Glenn L McGrath2f8a4012003-03-09 02:44:49 +0000417 /* handle edit cmds: (a)ppend, (i)nsert, and (c)hange */
418 else if (strchr("aic", sed_cmd->cmd)) {
419 if ((sed_cmd->end_line || sed_cmd->end_match) && sed_cmd->cmd != 'c')
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000420 bb_error_msg_and_die
421 ("only a beginning address can be specified for edit commands");
Denis Vlasenko80778502006-10-25 12:46:46 +0000422 for (;;) {
423 if (*cmdstr == '\n' || *cmdstr == '\\') {
Rob Landley25d82392004-04-01 09:23:30 +0000424 cmdstr++;
425 break;
Denis Vlasenko80778502006-10-25 12:46:46 +0000426 } else if (isspace(*cmdstr))
427 cmdstr++;
428 else
429 break;
Rob Landley25d82392004-04-01 09:23:30 +0000430 }
Rob Landleyd921b2e2006-08-03 15:41:12 +0000431 sed_cmd->string = xstrdup(cmdstr);
Denis Vlasenko40276642007-11-13 16:48:10 +0000432 /* "\anychar" -> "anychar" */
433 parse_escapes(sed_cmd->string, sed_cmd->string, strlen(cmdstr), '\0', '\0');
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000434 cmdstr += strlen(cmdstr);
Glenn L McGrath2f8a4012003-03-09 02:44:49 +0000435 /* handle file cmds: (r)ead */
Denis Vlasenko80778502006-10-25 12:46:46 +0000436 } else if (strchr("rw", sed_cmd->cmd)) {
Glenn L McGrath2f8a4012003-03-09 02:44:49 +0000437 if (sed_cmd->end_line || sed_cmd->end_match)
Denis Vlasenko80778502006-10-25 12:46:46 +0000438 bb_error_msg_and_die("command only uses one address");
Denis Vlasenko68404f12008-03-17 09:00:54 +0000439 cmdstr += parse_file_cmd(/*sed_cmd,*/ cmdstr, &sed_cmd->string);
Denis Vlasenkoc562bb72007-01-29 17:08:51 +0000440 if (sed_cmd->cmd == 'w') {
Denis Vlasenko5415c852008-07-21 23:05:26 +0000441 sed_cmd->sw_file = xfopen_for_write(sed_cmd->string);
Denis Vlasenkoc562bb72007-01-29 17:08:51 +0000442 sed_cmd->sw_last_char = '\n';
443 }
Glenn L McGrath961c6c12003-03-28 04:43:39 +0000444 /* handle branch commands */
Rob Landley93850a52005-05-18 06:34:37 +0000445 } else if (strchr(":btT", sed_cmd->cmd)) {
Glenn L McGrath961c6c12003-03-28 04:43:39 +0000446 int length;
447
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000448 cmdstr = skip_whitespace(cmdstr);
Glenn L McGrathf4523562003-09-14 06:01:14 +0000449 length = strcspn(cmdstr, semicolon_whitespace);
450 if (length) {
Rob Landleyd921b2e2006-08-03 15:41:12 +0000451 sed_cmd->string = xstrndup(cmdstr, length);
Glenn L McGrathf4523562003-09-14 06:01:14 +0000452 cmdstr += length;
453 }
Glenn L McGrath961c6c12003-03-28 04:43:39 +0000454 }
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000455 /* translation command */
456 else if (sed_cmd->cmd == 'y') {
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000457 char *match, *replace;
Denis Vlasenko80778502006-10-25 12:46:46 +0000458 int i = cmdstr[0];
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000459
Denis Vlasenko80778502006-10-25 12:46:46 +0000460 cmdstr += parse_regex_delim(cmdstr, &match, &replace)+1;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000461 /* \n already parsed, but \delimiter needs unescaping. */
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000462 parse_escapes(match, match, strlen(match), i, i);
463 parse_escapes(replace, replace, strlen(replace), i, i);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000464
Rob Landley9ffd4232006-05-21 18:30:35 +0000465 sed_cmd->string = xzalloc((strlen(match) + 1) * 2);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000466 for (i = 0; match[i] && replace[i]; i++) {
Denis Vlasenko80778502006-10-25 12:46:46 +0000467 sed_cmd->string[i*2] = match[i];
468 sed_cmd->string[i*2+1] = replace[i];
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000469 }
470 free(match);
471 free(replace);
Glenn L McGrathf01b46d2003-03-30 08:02:18 +0000472 }
Glenn L McGrath2971ef12003-03-18 01:19:23 +0000473 /* if it wasnt a single-letter command that takes no arguments
474 * then it must be an invalid command.
475 */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000476 else if (strchr("dDgGhHlnNpPqx={}", sed_cmd->cmd) == 0) {
Denis Vlasenko80778502006-10-25 12:46:46 +0000477 bb_error_msg_and_die("unsupported command %c", sed_cmd->cmd);
Mark Whitley1f3b9f22001-05-11 22:27:13 +0000478 }
Mark Whitley70705d72000-07-14 19:06:30 +0000479
480 /* give back whatever's left over */
Denis Vlasenko80778502006-10-25 12:46:46 +0000481 return cmdstr;
Eric Andersen50d63601999-11-09 01:47:36 +0000482}
Eric Andersen6b6b3f61999-10-28 16:06:25 +0000483
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000484
485/* Parse address+command sets, skipping comment lines. */
486
Denis Vlasenkob6aae0f2007-01-29 22:51:25 +0000487static void add_cmd(const char *cmdstr)
Erik Andersen1266a131999-12-29 22:19:46 +0000488{
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000489 sed_cmd_t *sed_cmd;
Glenn L McGrath586d86c2003-10-09 07:22:59 +0000490 int temp;
Glenn L McGrath8aac05b2003-09-14 04:06:12 +0000491
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000492 /* Append this line to any unfinished line from last time. */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000493 if (G.add_cmd_line) {
494 char *tp = xasprintf("%s\n%s", G.add_cmd_line, cmdstr);
495 free(G.add_cmd_line);
Denis Vlasenko945bd3d2007-04-12 21:20:25 +0000496 cmdstr = G.add_cmd_line = tp;
Rob Landley12d87552006-06-05 17:32:44 +0000497 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000498
499 /* If this line ends with backslash, request next line. */
Denis Vlasenko80778502006-10-25 12:46:46 +0000500 temp = strlen(cmdstr);
Denis Vlasenko945bd3d2007-04-12 21:20:25 +0000501 if (temp && cmdstr[--temp] == '\\') {
Denis Vlasenko17a15262007-03-26 20:48:46 +0000502 if (!G.add_cmd_line)
503 G.add_cmd_line = xstrdup(cmdstr);
Denis Vlasenko945bd3d2007-04-12 21:20:25 +0000504 G.add_cmd_line[temp] = '\0';
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000505 return;
Glenn L McGrathf50ce312003-03-09 15:12:24 +0000506 }
Mark Whitley6315ce62000-07-10 22:55:51 +0000507
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000508 /* Loop parsing all commands in this line. */
Denis Vlasenko80778502006-10-25 12:46:46 +0000509 while (*cmdstr) {
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000510 /* Skip leading whitespace and semicolons */
511 cmdstr += strspn(cmdstr, semicolon_whitespace);
512
513 /* If no more commands, exit. */
Denis Vlasenko80778502006-10-25 12:46:46 +0000514 if (!*cmdstr) break;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000515
516 /* if this is a comment, jump past it and keep going */
517 if (*cmdstr == '#') {
518 /* "#n" is the same as using -n on the command line */
Denis Vlasenko80778502006-10-25 12:46:46 +0000519 if (cmdstr[1] == 'n')
Denis Vlasenko17a15262007-03-26 20:48:46 +0000520 G.be_quiet++;
Denis Vlasenko80778502006-10-25 12:46:46 +0000521 cmdstr = strpbrk(cmdstr, "\n\r");
522 if (!cmdstr) break;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000523 continue;
Glenn L McGrathc1d95072003-04-08 06:42:45 +0000524 }
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000525
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000526 /* parse the command
527 * format is: [addr][,addr][!]cmd
528 * |----||-----||-|
529 * part1 part2 part3
Glenn L McGrathf50ce312003-03-09 15:12:24 +0000530 */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000531
Rob Landley9ffd4232006-05-21 18:30:35 +0000532 sed_cmd = xzalloc(sizeof(sed_cmd_t));
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000533
534 /* first part (if present) is an address: either a '$', a number or a /regex/ */
535 cmdstr += get_address(cmdstr, &sed_cmd->beg_line, &sed_cmd->beg_match);
536
537 /* second part (if present) will begin with a comma */
538 if (*cmdstr == ',') {
539 int idx;
540
Glenn L McGrathf50ce312003-03-09 15:12:24 +0000541 cmdstr++;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000542 idx = get_address(cmdstr, &sed_cmd->end_line, &sed_cmd->end_match);
Denis Vlasenko80778502006-10-25 12:46:46 +0000543 if (!idx)
544 bb_error_msg_and_die("no address after comma");
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000545 cmdstr += idx;
Glenn L McGrathf50ce312003-03-09 15:12:24 +0000546 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000547
548 /* skip whitespace before the command */
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000549 cmdstr = skip_whitespace(cmdstr);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000550
551 /* Check for inversion flag */
552 if (*cmdstr == '!') {
553 sed_cmd->invert = 1;
554 cmdstr++;
555
556 /* skip whitespace before the command */
Denis Vlasenkod18a3a22006-10-25 12:46:03 +0000557 cmdstr = skip_whitespace(cmdstr);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000558 }
559
560 /* last part (mandatory) will be a command */
Denis Vlasenko80778502006-10-25 12:46:46 +0000561 if (!*cmdstr)
562 bb_error_msg_and_die("missing command");
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000563 sed_cmd->cmd = *(cmdstr++);
564 cmdstr = parse_cmd_args(sed_cmd, cmdstr);
565
566 /* Add the command to the command array */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000567 G.sed_cmd_tail->next = sed_cmd;
568 G.sed_cmd_tail = G.sed_cmd_tail->next;
Glenn L McGrathf50ce312003-03-09 15:12:24 +0000569 }
570
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000571 /* If we glued multiple lines together, free the memory. */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000572 free(G.add_cmd_line);
573 G.add_cmd_line = NULL;
Mark Whitley6315ce62000-07-10 22:55:51 +0000574}
575
Rob Landleydcc28662004-11-25 07:21:47 +0000576/* Append to a string, reallocating memory as necessary. */
577
Glenn L McGrath8d6395d2003-04-08 11:56:11 +0000578#define PIPE_GROW 64
Eric Andersenc52a6b02001-11-10 10:49:42 +0000579
Eric Andersen14f5c8d2005-04-16 19:39:00 +0000580static void pipe_putc(char c)
Eric Andersenc52a6b02001-11-10 10:49:42 +0000581{
Denis Vlasenko17a15262007-03-26 20:48:46 +0000582 if (G.pipeline.idx == G.pipeline.len) {
583 G.pipeline.buf = xrealloc(G.pipeline.buf,
584 G.pipeline.len + PIPE_GROW);
585 G.pipeline.len += PIPE_GROW;
Eric Andersenc52a6b02001-11-10 10:49:42 +0000586 }
Denis Vlasenko17a15262007-03-26 20:48:46 +0000587 G.pipeline.buf[G.pipeline.idx++] = c;
Eric Andersenc52a6b02001-11-10 10:49:42 +0000588}
589
Rob Landley4795e4e2006-07-26 17:25:08 +0000590static void do_subst_w_backrefs(char *line, char *replace)
Mark Whitley97562bd2000-07-17 20:06:42 +0000591{
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000592 int i,j;
Mark Whitley97562bd2000-07-17 20:06:42 +0000593
594 /* go through the replacement string */
595 for (i = 0; replace[i]; i++) {
596 /* if we find a backreference (\1, \2, etc.) print the backref'ed * text */
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000597 if (replace[i] == '\\') {
598 unsigned backref = replace[++i] - '0';
599 if (backref <= 9) {
Denis Vlasenko17a15262007-03-26 20:48:46 +0000600 /* print out the text held in G.regmatch[backref] */
601 if (G.regmatch[backref].rm_so != -1) {
602 j = G.regmatch[backref].rm_so;
603 while (j < G.regmatch[backref].rm_eo)
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000604 pipe_putc(line[j++]);
605 }
606 continue;
Denis Vlasenko80778502006-10-25 12:46:46 +0000607 }
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000608 /* I _think_ it is impossible to get '\' to be
609 * the last char in replace string. Thus we dont check
610 * for replace[i] == NUL. (counterexample anyone?) */
611 /* if we find a backslash escaped character, print the character */
612 pipe_putc(replace[i]);
613 continue;
Mark Whitley97562bd2000-07-17 20:06:42 +0000614 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000615 /* if we find an unescaped '&' print out the whole matched text. */
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000616 if (replace[i] == '&') {
Denis Vlasenko17a15262007-03-26 20:48:46 +0000617 j = G.regmatch[0].rm_so;
618 while (j < G.regmatch[0].rm_eo)
Denis Vlasenko80778502006-10-25 12:46:46 +0000619 pipe_putc(line[j++]);
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000620 continue;
Denis Vlasenko80778502006-10-25 12:46:46 +0000621 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000622 /* Otherwise just output the character. */
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000623 pipe_putc(replace[i]);
Mark Whitley97562bd2000-07-17 20:06:42 +0000624 }
625}
626
Rob Landleye3f5a3f2006-05-09 03:53:55 +0000627static int do_subst_command(sed_cmd_t *sed_cmd, char **line)
Mark Whitley4f7fe772000-07-13 20:01:58 +0000628{
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000629 char *oldline = *line;
Mark Whitley4f7fe772000-07-13 20:01:58 +0000630 int altered = 0;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +0000631 unsigned match_count = 0;
Glenn L McGrathc1d95072003-04-08 06:42:45 +0000632 regex_t *current_regex;
633
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000634 /* Handle empty regex. */
Glenn L McGrathc1d95072003-04-08 06:42:45 +0000635 if (sed_cmd->sub_match == NULL) {
Denis Vlasenko17a15262007-03-26 20:48:46 +0000636 current_regex = G.previous_regex_ptr;
Denis Vlasenko80778502006-10-25 12:46:46 +0000637 if (!current_regex)
Denis Vlasenkod3d004d2006-10-27 09:02:31 +0000638 bb_error_msg_and_die("no previous regexp");
Denis Vlasenko80778502006-10-25 12:46:46 +0000639 } else
Denis Vlasenko17a15262007-03-26 20:48:46 +0000640 G.previous_regex_ptr = current_regex = sed_cmd->sub_match;
Mark Whitley4f7fe772000-07-13 20:01:58 +0000641
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000642 /* Find the first match */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000643 if (REG_NOMATCH == regexec(current_regex, oldline, 10, G.regmatch, 0))
Mark Whitley2dc192f2000-11-03 19:47:00 +0000644 return 0;
645
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000646 /* Initialize temporary output buffer. */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000647 G.pipeline.buf = xmalloc(PIPE_GROW);
648 G.pipeline.len = PIPE_GROW;
649 G.pipeline.idx = 0;
Mark Whitley2dc192f2000-11-03 19:47:00 +0000650
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000651 /* Now loop through, substituting for matches */
652 do {
Mark Whitley4f7fe772000-07-13 20:01:58 +0000653 int i;
Mark Whitley4f7fe772000-07-13 20:01:58 +0000654
Eric Andersenc06f5682004-02-04 10:57:46 +0000655 /* Work around bug in glibc regexec, demonstrated by:
656 echo " a.b" | busybox sed 's [^ .]* x g'
657 The match_count check is so not to break
658 echo "hi" | busybox sed 's/^/!/g' */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000659 if (!G.regmatch[0].rm_so && !G.regmatch[0].rm_eo && match_count) {
Denis Vlasenko80778502006-10-25 12:46:46 +0000660 pipe_putc(*oldline++);
Eric Andersenc06f5682004-02-04 10:57:46 +0000661 continue;
662 }
663
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000664 match_count++;
665
666 /* If we aren't interested in this match, output old line to
667 end of match and continue */
Denis Vlasenko77ad97f2008-05-13 02:27:31 +0000668 if (sed_cmd->which_match
669 && (sed_cmd->which_match != match_count)
670 ) {
Denis Vlasenko17a15262007-03-26 20:48:46 +0000671 for (i = 0; i < G.regmatch[0].rm_eo; i++)
Denis Vlasenko80778502006-10-25 12:46:46 +0000672 pipe_putc(*oldline++);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000673 continue;
674 }
675
Mark Whitley2dc192f2000-11-03 19:47:00 +0000676 /* print everything before the match */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000677 for (i = 0; i < G.regmatch[0].rm_so; i++)
Denis Vlasenko80778502006-10-25 12:46:46 +0000678 pipe_putc(oldline[i]);
Mark Whitley97562bd2000-07-17 20:06:42 +0000679
Mark Whitley2dc192f2000-11-03 19:47:00 +0000680 /* then print the substitution string */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000681 do_subst_w_backrefs(oldline, sed_cmd->string);
Mark Whitley97562bd2000-07-17 20:06:42 +0000682
Mark Whitley2dc192f2000-11-03 19:47:00 +0000683 /* advance past the match */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000684 oldline += G.regmatch[0].rm_eo;
Mark Whitley2dc192f2000-11-03 19:47:00 +0000685 /* flag that something has changed */
686 altered++;
Mark Whitley97562bd2000-07-17 20:06:42 +0000687
Mark Whitley2dc192f2000-11-03 19:47:00 +0000688 /* if we're not doing this globally, get out now */
Denis Vlasenko284d0fa2008-02-16 13:18:17 +0000689 if (sed_cmd->which_match)
690 break;
Denis Vlasenko17a15262007-03-26 20:48:46 +0000691 } while (*oldline && (regexec(current_regex, oldline, 10, G.regmatch, 0) != REG_NOMATCH));
Mark Whitley2dc192f2000-11-03 19:47:00 +0000692
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000693 /* Copy rest of string into output pipeline */
694
Denis Vlasenko80778502006-10-25 12:46:46 +0000695 while (*oldline)
696 pipe_putc(*oldline++);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000697 pipe_putc(0);
Mark Whitley2dc192f2000-11-03 19:47:00 +0000698
Eric Andersenb76cb682001-08-22 05:58:16 +0000699 free(*line);
Denis Vlasenko17a15262007-03-26 20:48:46 +0000700 *line = G.pipeline.buf;
Mark Whitley4f7fe772000-07-13 20:01:58 +0000701 return altered;
702}
Mark Whitley6315ce62000-07-10 22:55:51 +0000703
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000704/* Set command pointer to point to this label. (Does not handle null label.) */
Rob Landley4795e4e2006-07-26 17:25:08 +0000705static sed_cmd_t *branch_to(char *label)
Glenn L McGrath961c6c12003-03-28 04:43:39 +0000706{
707 sed_cmd_t *sed_cmd;
Glenn L McGrathd4185b02003-04-11 17:10:23 +0000708
Denis Vlasenko17a15262007-03-26 20:48:46 +0000709 for (sed_cmd = G.sed_cmd_head.next; sed_cmd; sed_cmd = sed_cmd->next) {
Denis Vlasenko80778502006-10-25 12:46:46 +0000710 if (sed_cmd->cmd == ':' && sed_cmd->string && !strcmp(sed_cmd->string, label)) {
711 return sed_cmd;
Glenn L McGrath961c6c12003-03-28 04:43:39 +0000712 }
713 }
Denis Vlasenko80778502006-10-25 12:46:46 +0000714 bb_error_msg_and_die("can't find label for jump to '%s'", label);
Glenn L McGrath961c6c12003-03-28 04:43:39 +0000715}
Mark Whitley6315ce62000-07-10 22:55:51 +0000716
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000717static void append(char *s)
718{
Denis Vlasenko17a15262007-03-26 20:48:46 +0000719 llist_add_to_end(&G.append_head, xstrdup(s));
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000720}
721
722static void flush_append(void)
723{
Rob Landley0b656282006-05-08 22:17:23 +0000724 char *data;
725
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000726 /* Output appended lines. */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000727 while ((data = (char *)llist_pop(&G.append_head))) {
728 fprintf(G.nonstdout, "%s\n", data);
Rob Landley0b656282006-05-08 22:17:23 +0000729 free(data);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000730 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000731}
732
Eric Andersen14f5c8d2005-04-16 19:39:00 +0000733static void add_input_file(FILE *file)
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000734{
Denis Vlasenkodeeed592008-07-08 05:14:36 +0000735 G.input_file_list = xrealloc_vector(G.input_file_list, 2, G.input_file_count);
Denis Vlasenko17a15262007-03-26 20:48:46 +0000736 G.input_file_list[G.input_file_count++] = file;
Rob Landleydcc28662004-11-25 07:21:47 +0000737}
738
Denis Vlasenko17a15262007-03-26 20:48:46 +0000739/* Get next line of input from G.input_file_list, flushing append buffer and
Rob Landleydcc28662004-11-25 07:21:47 +0000740 * noting if we ran out of files without a newline on the last line we read.
741 */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000742enum {
743 NO_EOL_CHAR = 1,
Denis Vlasenko86811802007-01-29 17:10:19 +0000744 LAST_IS_NUL = 2,
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000745};
746static char *get_next_line(char *gets_char)
Rob Landleydcc28662004-11-25 07:21:47 +0000747{
Denis Vlasenko80778502006-10-25 12:46:46 +0000748 char *temp = NULL;
Denis Vlasenkoc562bb72007-01-29 17:08:51 +0000749 int len;
750 char gc;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000751
752 flush_append();
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000753
754 /* will be returned if last line in the file
755 * doesn't end with either '\n' or '\0' */
756 gc = NO_EOL_CHAR;
Denis Vlasenko17a15262007-03-26 20:48:46 +0000757 while (G.current_input_file < G.input_file_count) {
758 FILE *fp = G.input_file_list[G.current_input_file];
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000759 /* Read line up to a newline or NUL byte, inclusive,
760 * return malloc'ed char[]. length of the chunk read
761 * is stored in len. NULL if EOF/error */
Denis Vlasenko86811802007-01-29 17:10:19 +0000762 temp = bb_get_chunk_from_file(fp, &len);
Rob Landley2b26fd52006-02-24 02:30:39 +0000763 if (temp) {
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000764 /* len > 0 here, it's ok to do temp[len-1] */
765 char c = temp[len-1];
766 if (c == '\n' || c == '\0') {
767 temp[len-1] = '\0';
Denis Vlasenkoc562bb72007-01-29 17:08:51 +0000768 gc = c;
Denis Vlasenko86811802007-01-29 17:10:19 +0000769 if (c == '\0') {
770 int ch = fgetc(fp);
771 if (ch != EOF)
772 ungetc(ch, fp);
773 else
774 gc = LAST_IS_NUL;
775 }
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000776 }
Denis Vlasenkoc562bb72007-01-29 17:08:51 +0000777 /* else we put NO_EOL_CHAR into *gets_char */
778 break;
779
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000780 /* NB: I had the idea of peeking next file(s) and returning
781 * NO_EOL_CHAR only if it is the *last* non-empty
782 * input file. But there is a case where this won't work:
783 * file1: "a woo\nb woo"
784 * file2: "c no\nd no"
785 * sed -ne 's/woo/bang/p' input1 input2 => "a bang\nb bang"
786 * (note: *no* newline after "b bang"!) */
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000787 }
788 /* Close this file and advance to next one */
Denis Vlasenko86811802007-01-29 17:10:19 +0000789 fclose(fp);
Denis Vlasenko17a15262007-03-26 20:48:46 +0000790 G.current_input_file++;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000791 }
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000792 *gets_char = gc;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000793 return temp;
794}
795
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000796/* Output line of text. */
797/* Note:
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000798 * The tricks with NO_EOL_CHAR and last_puts_char are there to emulate gnu sed.
Denis Vlasenko1375bc72006-12-02 20:12:12 +0000799 * Without them, we had this:
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000800 * echo -n thingy >z1
801 * echo -n again >z2
802 * >znull
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000803 * sed "s/i/z/" z1 z2 znull | hexdump -vC
804 * output:
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000805 * gnu sed 4.1.5:
806 * 00000000 74 68 7a 6e 67 79 0a 61 67 61 7a 6e |thzngy.agazn|
807 * bbox:
808 * 00000000 74 68 7a 6e 67 79 61 67 61 7a 6e |thzngyagazn|
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000809 */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000810static void puts_maybe_newline(char *s, FILE *file, char *last_puts_char, char last_gets_char)
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000811{
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000812 char lpc = *last_puts_char;
Denis Vlasenko1375bc72006-12-02 20:12:12 +0000813
Denis Vlasenko86811802007-01-29 17:10:19 +0000814 /* Need to insert a '\n' between two files because first file's
815 * last line wasn't terminated? */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000816 if (lpc != '\n' && lpc != '\0') {
Denis Vlasenko1375bc72006-12-02 20:12:12 +0000817 fputc('\n', file);
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000818 lpc = '\n';
Denis Vlasenko1375bc72006-12-02 20:12:12 +0000819 }
Denis Vlasenko8b22b072006-12-02 17:58:10 +0000820 fputs(s, file);
Denis Vlasenko86811802007-01-29 17:10:19 +0000821
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000822 /* 'x' - just something which is not '\n', '\0' or NO_EOL_CHAR */
Denis Vlasenkoef44d9d2007-01-17 23:16:16 +0000823 if (s[0])
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000824 lpc = 'x';
Denis Vlasenko86811802007-01-29 17:10:19 +0000825
826 /* had trailing '\0' and it was last char of file? */
827 if (last_gets_char == LAST_IS_NUL) {
828 fputc('\0', file);
829 lpc = 'x'; /* */
830 } else
831 /* had trailing '\n' or '\0'? */
832 if (last_gets_char != NO_EOL_CHAR) {
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000833 fputc(last_gets_char, file);
834 lpc = last_gets_char;
Denis Vlasenko1375bc72006-12-02 20:12:12 +0000835 }
Denis Vlasenko86811802007-01-29 17:10:19 +0000836
Denis Vlasenko80778502006-10-25 12:46:46 +0000837 if (ferror(file)) {
Denis Vlasenko40920822006-10-03 20:28:06 +0000838 xfunc_error_retval = 4; /* It's what gnu sed exits with... */
Bernhard Reutner-Fischera3d4bf32006-06-03 21:40:11 +0000839 bb_error_msg_and_die(bb_msg_write_error);
Rob Landley53302f82004-02-18 09:54:15 +0000840 }
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000841 *last_puts_char = lpc;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000842}
843
Denis Vlasenko17a15262007-03-26 20:48:46 +0000844#define sed_puts(s, n) (puts_maybe_newline(s, G.nonstdout, &last_puts_char, n))
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000845
Denis Vlasenko8274e062007-08-06 03:41:08 +0000846static int beg_match(sed_cmd_t *sed_cmd, const char *pattern_space)
847{
848 int retval = sed_cmd->beg_match && !regexec(sed_cmd->beg_match, pattern_space, 0, NULL, 0);
849 if (retval)
850 G.previous_regex_ptr = sed_cmd->beg_match;
851 return retval;
852}
853
Rob Landley2b26fd52006-02-24 02:30:39 +0000854/* Process all the lines in all the files */
855
Rob Landleydcc28662004-11-25 07:21:47 +0000856static void process_files(void)
Mark Whitley6315ce62000-07-10 22:55:51 +0000857{
Rob Landleyce4f0e92004-10-30 06:54:19 +0000858 char *pattern_space, *next_line;
Denis Vlasenko826c85f2007-01-28 23:26:15 +0000859 int linenum = 0;
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000860 char last_puts_char = '\n';
861 char last_gets_char, next_gets_char;
Denis Vlasenko2ea630f2006-12-10 02:52:19 +0000862 sed_cmd_t *sed_cmd;
863 int substituted;
Mark Whitley6315ce62000-07-10 22:55:51 +0000864
Rob Landley2b26fd52006-02-24 02:30:39 +0000865 /* Prime the pump */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000866 next_line = get_next_line(&next_gets_char);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000867
Rob Landleydcc28662004-11-25 07:21:47 +0000868 /* go through every line in each file */
Denis Vlasenkof39c7c02008-02-28 17:59:01 +0000869 again:
Denis Vlasenko2ea630f2006-12-10 02:52:19 +0000870 substituted = 0;
Glenn L McGrath505bd0f2003-03-08 05:21:02 +0000871
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000872 /* Advance to next line. Stop if out of lines. */
873 pattern_space = next_line;
874 if (!pattern_space) return;
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000875 last_gets_char = next_gets_char;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000876
877 /* Read one line in advance so we can act on the last line,
878 * the '$' address */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000879 next_line = get_next_line(&next_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000880 linenum++;
Denis Vlasenkof39c7c02008-02-28 17:59:01 +0000881 restart:
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000882 /* for every line, go through all the commands */
Denis Vlasenko17a15262007-03-26 20:48:46 +0000883 for (sed_cmd = G.sed_cmd_head.next; sed_cmd; sed_cmd = sed_cmd->next) {
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000884 int old_matched, matched;
Eric Andersen52a3c272003-12-23 08:53:51 +0000885
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000886 old_matched = sed_cmd->in_match;
Mark Whitley0915c4b2001-06-11 23:50:06 +0000887
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000888 /* Determine if this command matches this line: */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000889
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000890 /* Are we continuing a previous multi-line match? */
891 sed_cmd->in_match = sed_cmd->in_match
892 /* Or is no range necessary? */
893 || (!sed_cmd->beg_line && !sed_cmd->end_line
894 && !sed_cmd->beg_match && !sed_cmd->end_match)
895 /* Or did we match the start of a numerical range? */
896 || (sed_cmd->beg_line > 0 && (sed_cmd->beg_line == linenum))
897 /* Or does this line match our begin address regex? */
Denis Vlasenko8274e062007-08-06 03:41:08 +0000898 || (beg_match(sed_cmd, pattern_space))
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000899 /* Or did we match last line of input? */
900 || (sed_cmd->beg_line == -1 && next_line == NULL);
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000901
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000902 /* Snapshot the value */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000903
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000904 matched = sed_cmd->in_match;
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000905
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000906 /* Is this line the end of the current match? */
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000907
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000908 if (matched) {
909 sed_cmd->in_match = !(
910 /* has the ending line come, or is this a single address command? */
911 (sed_cmd->end_line ?
912 sed_cmd->end_line == -1 ?
913 !next_line
914 : (sed_cmd->end_line <= linenum)
915 : !sed_cmd->end_match
916 )
917 /* or does this line matches our last address regex */
918 || (sed_cmd->end_match && old_matched
919 && (regexec(sed_cmd->end_match,
Denis Vlasenko8e858e22007-03-07 09:35:43 +0000920 pattern_space, 0, NULL, 0) == 0))
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000921 );
Erik Andersene49d5ec2000-02-08 19:58:47 +0000922 }
Erik Andersen1266a131999-12-29 22:19:46 +0000923
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000924 /* Skip blocks of commands we didn't match. */
925 if (sed_cmd->cmd == '{') {
Denis Vlasenko826c85f2007-01-28 23:26:15 +0000926 if (sed_cmd->invert ? matched : !matched) {
927 while (sed_cmd->cmd != '}') {
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000928 sed_cmd = sed_cmd->next;
Denis Vlasenko826c85f2007-01-28 23:26:15 +0000929 if (!sed_cmd)
930 bb_error_msg_and_die("unterminated {");
931 }
932 }
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000933 continue;
934 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +0000935
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000936 /* Okay, so did this line match? */
937 if (sed_cmd->invert ? !matched : matched) {
938 /* Update last used regex in case a blank substitute BRE is found */
939 if (sed_cmd->beg_match) {
Denis Vlasenko17a15262007-03-26 20:48:46 +0000940 G.previous_regex_ptr = sed_cmd->beg_match;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000941 }
942
943 /* actual sedding */
944 switch (sed_cmd->cmd) {
945
946 /* Print line number */
947 case '=':
Denis Vlasenko17a15262007-03-26 20:48:46 +0000948 fprintf(G.nonstdout, "%d\n", linenum);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000949 break;
950
951 /* Write the current pattern space up to the first newline */
952 case 'P':
953 {
954 char *tmp = strchr(pattern_space, '\n');
955
956 if (tmp) {
957 *tmp = '\0';
Denis Vlasenko826c85f2007-01-28 23:26:15 +0000958 /* TODO: explain why '\n' below */
959 sed_puts(pattern_space, '\n');
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000960 *tmp = '\n';
961 break;
962 }
963 /* Fall Through */
964 }
965
966 /* Write the current pattern space to output */
967 case 'p':
Denis Vlasenko826c85f2007-01-28 23:26:15 +0000968 /* NB: we print this _before_ the last line
969 * (of current file) is printed. Even if
970 * that line is nonterminated, we print
971 * '\n' here (gnu sed does the same) */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000972 sed_puts(pattern_space, '\n');
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000973 break;
974 /* Delete up through first newline */
975 case 'D':
976 {
977 char *tmp = strchr(pattern_space, '\n');
978
979 if (tmp) {
980 tmp = xstrdup(tmp+1);
981 free(pattern_space);
982 pattern_space = tmp;
983 goto restart;
984 }
985 }
986 /* discard this line. */
987 case 'd':
988 goto discard_line;
989
990 /* Substitute with regex */
991 case 's':
992 if (!do_subst_command(sed_cmd, &pattern_space))
993 break;
994 substituted |= 1;
995
996 /* handle p option */
997 if (sed_cmd->sub_p)
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +0000998 sed_puts(pattern_space, last_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +0000999 /* handle w option */
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001000 if (sed_cmd->sw_file)
1001 puts_maybe_newline(
1002 pattern_space, sed_cmd->sw_file,
1003 &sed_cmd->sw_last_char, last_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001004 break;
1005
1006 /* Append line to linked list to be printed later */
1007 case 'a':
1008 append(sed_cmd->string);
1009 break;
1010
1011 /* Insert text before this line */
1012 case 'i':
Denis Vlasenko826c85f2007-01-28 23:26:15 +00001013 sed_puts(sed_cmd->string, '\n');
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001014 break;
1015
1016 /* Cut and paste text (replace) */
1017 case 'c':
1018 /* Only triggers on last line of a matching range. */
1019 if (!sed_cmd->in_match)
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001020 sed_puts(sed_cmd->string, NO_EOL_CHAR);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001021 goto discard_line;
1022
1023 /* Read file, append contents to output */
1024 case 'r':
1025 {
1026 FILE *rfile;
1027
Denis Vlasenko5415c852008-07-21 23:05:26 +00001028 rfile = fopen_for_read(sed_cmd->string);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001029 if (rfile) {
1030 char *line;
1031
Denis Vlasenko8ee649a2008-03-26 20:04:27 +00001032 while ((line = xmalloc_fgetline(rfile))
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001033 != NULL)
1034 append(line);
1035 xprint_and_close_file(rfile);
1036 }
1037
1038 break;
1039 }
1040
1041 /* Write pattern space to file. */
1042 case 'w':
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001043 puts_maybe_newline(
1044 pattern_space, sed_cmd->sw_file,
1045 &sed_cmd->sw_last_char, last_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001046 break;
1047
1048 /* Read next line from input */
1049 case 'n':
Denis Vlasenko17a15262007-03-26 20:48:46 +00001050 if (!G.be_quiet)
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001051 sed_puts(pattern_space, last_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001052 if (next_line) {
1053 free(pattern_space);
1054 pattern_space = next_line;
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001055 last_gets_char = next_gets_char;
1056 next_line = get_next_line(&next_gets_char);
Denis Vlasenkof39c7c02008-02-28 17:59:01 +00001057 substituted = 0;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001058 linenum++;
1059 break;
1060 }
1061 /* fall through */
1062
1063 /* Quit. End of script, end of input. */
1064 case 'q':
1065 /* Exit the outer while loop */
1066 free(next_line);
1067 next_line = NULL;
1068 goto discard_commands;
1069
1070 /* Append the next line to the current line */
1071 case 'N':
1072 {
1073 int len;
1074 /* If no next line, jump to end of script and exit. */
1075 if (next_line == NULL) {
1076 /* Jump to end of script and exit */
1077 free(next_line);
1078 next_line = NULL;
1079 goto discard_line;
1080 /* append next_line, read new next_line. */
1081 }
1082 len = strlen(pattern_space);
1083 pattern_space = realloc(pattern_space, len + strlen(next_line) + 2);
1084 pattern_space[len] = '\n';
1085 strcpy(pattern_space + len+1, next_line);
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001086 last_gets_char = next_gets_char;
1087 next_line = get_next_line(&next_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001088 linenum++;
1089 break;
1090 }
1091
1092 /* Test/branch if substitution occurred */
1093 case 't':
1094 if (!substituted) break;
1095 substituted = 0;
1096 /* Fall through */
1097 /* Test/branch if substitution didn't occur */
1098 case 'T':
1099 if (substituted) break;
1100 /* Fall through */
1101 /* Branch to label */
1102 case 'b':
1103 if (!sed_cmd->string) goto discard_commands;
1104 else sed_cmd = branch_to(sed_cmd->string);
1105 break;
1106 /* Transliterate characters */
1107 case 'y':
1108 {
1109 int i, j;
1110
1111 for (i = 0; pattern_space[i]; i++) {
1112 for (j = 0; sed_cmd->string[j]; j += 2) {
1113 if (pattern_space[i] == sed_cmd->string[j]) {
1114 pattern_space[i] = sed_cmd->string[j + 1];
1115 break;
1116 }
1117 }
1118 }
1119
1120 break;
1121 }
1122 case 'g': /* Replace pattern space with hold space */
1123 free(pattern_space);
Denis Vlasenko17a15262007-03-26 20:48:46 +00001124 pattern_space = xstrdup(G.hold_space ? G.hold_space : "");
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001125 break;
1126 case 'G': /* Append newline and hold space to pattern space */
1127 {
1128 int pattern_space_size = 2;
1129 int hold_space_size = 0;
1130
1131 if (pattern_space)
1132 pattern_space_size += strlen(pattern_space);
Denis Vlasenko17a15262007-03-26 20:48:46 +00001133 if (G.hold_space)
1134 hold_space_size = strlen(G.hold_space);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001135 pattern_space = xrealloc(pattern_space,
1136 pattern_space_size + hold_space_size);
1137 if (pattern_space_size == 2)
1138 pattern_space[0] = 0;
1139 strcat(pattern_space, "\n");
Denis Vlasenko17a15262007-03-26 20:48:46 +00001140 if (G.hold_space)
1141 strcat(pattern_space, G.hold_space);
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001142 last_gets_char = '\n';
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001143
1144 break;
1145 }
1146 case 'h': /* Replace hold space with pattern space */
Denis Vlasenko17a15262007-03-26 20:48:46 +00001147 free(G.hold_space);
1148 G.hold_space = xstrdup(pattern_space);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001149 break;
1150 case 'H': /* Append newline and pattern space to hold space */
1151 {
1152 int hold_space_size = 2;
1153 int pattern_space_size = 0;
1154
Denis Vlasenko17a15262007-03-26 20:48:46 +00001155 if (G.hold_space)
1156 hold_space_size += strlen(G.hold_space);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001157 if (pattern_space)
1158 pattern_space_size = strlen(pattern_space);
Denis Vlasenko17a15262007-03-26 20:48:46 +00001159 G.hold_space = xrealloc(G.hold_space,
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001160 hold_space_size + pattern_space_size);
1161
1162 if (hold_space_size == 2)
Denis Vlasenko17a15262007-03-26 20:48:46 +00001163 *G.hold_space = 0;
1164 strcat(G.hold_space, "\n");
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001165 if (pattern_space)
Denis Vlasenko17a15262007-03-26 20:48:46 +00001166 strcat(G.hold_space, pattern_space);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001167
1168 break;
1169 }
1170 case 'x': /* Exchange hold and pattern space */
1171 {
1172 char *tmp = pattern_space;
Denis Vlasenko17a15262007-03-26 20:48:46 +00001173 pattern_space = G.hold_space ? : xzalloc(1);
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001174 last_gets_char = '\n';
Denis Vlasenko17a15262007-03-26 20:48:46 +00001175 G.hold_space = tmp;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001176 break;
1177 }
1178 }
1179 }
Glenn L McGrathaa5a6022003-10-01 03:06:16 +00001180 }
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001181
1182 /*
1183 * exit point from sedding...
1184 */
Denis Vlasenko826c85f2007-01-28 23:26:15 +00001185 discard_commands:
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001186 /* we will print the line unless we were told to be quiet ('-n')
1187 or if the line was suppressed (ala 'd'elete) */
Denis Vlasenko17a15262007-03-26 20:48:46 +00001188 if (!G.be_quiet)
Denis Vlasenkofe7a9f12007-01-29 14:31:47 +00001189 sed_puts(pattern_space, last_gets_char);
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001190
1191 /* Delete and such jump here. */
Denis Vlasenko826c85f2007-01-28 23:26:15 +00001192 discard_line:
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001193 flush_append();
1194 free(pattern_space);
1195
1196 goto again;
Erik Andersen1266a131999-12-29 22:19:46 +00001197}
1198
Glenn L McGrath42c25732003-10-04 05:27:56 +00001199/* It is possible to have a command line argument with embedded
Denis Vlasenko4b0bb9e2007-03-16 23:36:58 +00001200 * newlines. This counts as multiple command lines.
1201 * However, newline can be escaped: 's/e/z\<newline>z/'
1202 * We check for this.
1203 */
Glenn L McGrath42c25732003-10-04 05:27:56 +00001204
1205static void add_cmd_block(char *cmdstr)
1206{
Denis Vlasenko4b0bb9e2007-03-16 23:36:58 +00001207 char *sv, *eol;
Glenn L McGrath42c25732003-10-04 05:27:56 +00001208
Denis Vlasenko4b0bb9e2007-03-16 23:36:58 +00001209 cmdstr = sv = xstrdup(cmdstr);
1210 do {
1211 eol = strchr(cmdstr, '\n');
1212 next:
1213 if (eol) {
1214 /* Count preceding slashes */
1215 int slashes = 0;
1216 char *sl = eol;
1217
1218 while (sl != cmdstr && *--sl == '\\')
1219 slashes++;
1220 /* Odd number of preceding slashes - newline is escaped */
1221 if (slashes & 1) {
Denis Vlasenko0f293b92008-07-22 20:16:55 +00001222 overlapping_strcpy(eol - 1, eol);
Denis Vlasenko4b0bb9e2007-03-16 23:36:58 +00001223 eol = strchr(eol, '\n');
1224 goto next;
1225 }
1226 *eol = '\0';
1227 }
1228 add_cmd(cmdstr);
1229 cmdstr = eol + 1;
1230 } while (eol);
1231 free(sv);
Glenn L McGrath42c25732003-10-04 05:27:56 +00001232}
1233
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00001234int sed_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +00001235int sed_main(int argc UNUSED_PARAM, char **argv)
Eric Andersen6b6b3f61999-10-28 16:06:25 +00001236{
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001237 enum {
1238 OPT_in_place = 1 << 0,
1239 };
Denis Vlasenko67b23e62006-10-03 21:00:06 +00001240 unsigned opt;
Denis Vlasenkob97c9842006-10-01 21:05:12 +00001241 llist_t *opt_e, *opt_f;
1242 int status = EXIT_SUCCESS;
Eric Andersen6b6b3f61999-10-28 16:06:25 +00001243
Denis Vlasenko74324c82007-06-04 10:16:52 +00001244 INIT_G();
Rob Landleye3f5a3f2006-05-09 03:53:55 +00001245
Mark Whitley858c1ad2000-07-11 21:38:47 +00001246 /* destroy command strings on exit */
Rob Landley0b656282006-05-08 22:17:23 +00001247 if (ENABLE_FEATURE_CLEAN_UP) atexit(sed_free_and_close_stuff);
Mark Whitley858c1ad2000-07-11 21:38:47 +00001248
Rob Landleyfae1dc82005-11-20 07:44:35 +00001249 /* Lie to autoconf when it starts asking stupid questions. */
Denis Vlasenko62a90cd2008-03-17 09:07:36 +00001250 if (argv[1] && !strcmp(argv[1], "--version")) {
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001251 puts("This is not GNU sed version 4.0");
1252 return 0;
Eric Andersenc06f5682004-02-04 10:57:46 +00001253 }
Eric Andersenc06f5682004-02-04 10:57:46 +00001254
Mark Whitley6315ce62000-07-10 22:55:51 +00001255 /* do normal option parsing */
Denis Vlasenkob97c9842006-10-01 21:05:12 +00001256 opt_e = opt_f = NULL;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001257 opt_complementary = "e::f::" /* can occur multiple times */
1258 "nn"; /* count -n */
Denis Vlasenkofe7cd642007-08-18 15:32:12 +00001259 opt = getopt32(argv, "irne:f:", &opt_e, &opt_f,
Denis Vlasenko17a15262007-03-26 20:48:46 +00001260 &G.be_quiet); /* counter for -n */
Denis Vlasenko62a90cd2008-03-17 09:07:36 +00001261 //argc -= optind;
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001262 argv += optind;
1263 if (opt & OPT_in_place) { // -i
Denis Vlasenko750fc6d2006-09-22 08:56:03 +00001264 atexit(cleanup_outname);
1265 }
Denis Vlasenko17a15262007-03-26 20:48:46 +00001266 if (opt & 0x2) G.regex_type |= REG_EXTENDED; // -r
1267 //if (opt & 0x4) G.be_quiet++; // -n
Denis Vlasenko945bd3d2007-04-12 21:20:25 +00001268 while (opt_e) { // -e
Denis Vlasenkod50dda82008-06-15 05:40:56 +00001269 add_cmd_block(llist_pop(&opt_e));
Denis Vlasenko750fc6d2006-09-22 08:56:03 +00001270 }
Denis Vlasenko945bd3d2007-04-12 21:20:25 +00001271 while (opt_f) { // -f
1272 char *line;
1273 FILE *cmdfile;
Denis Vlasenko5415c852008-07-21 23:05:26 +00001274 cmdfile = xfopen_for_read(llist_pop(&opt_f));
Denis Vlasenko8ee649a2008-03-26 20:04:27 +00001275 while ((line = xmalloc_fgetline(cmdfile)) != NULL) {
Denis Vlasenko945bd3d2007-04-12 21:20:25 +00001276 add_cmd(line);
1277 free(line);
1278 }
1279 fclose(cmdfile);
Eric Andersen6b6b3f61999-10-28 16:06:25 +00001280 }
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001281 /* if we didn't get a pattern from -e or -f, use argv[0] */
Denis Vlasenko80778502006-10-25 12:46:46 +00001282 if (!(opt & 0x18)) {
Denis Vlasenko62a90cd2008-03-17 09:07:36 +00001283 if (!*argv)
Manuel Novoa III cad53642003-03-19 09:13:01 +00001284 bb_show_usage();
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001285 add_cmd_block(*argv++);
Mark Whitley6315ce62000-07-10 22:55:51 +00001286 }
Glenn L McGrath42c25732003-10-04 05:27:56 +00001287 /* Flush any unfinished commands. */
1288 add_cmd("");
Mark Whitley6315ce62000-07-10 22:55:51 +00001289
Rob Landley53302f82004-02-18 09:54:15 +00001290 /* By default, we write to stdout */
Denis Vlasenko17a15262007-03-26 20:48:46 +00001291 G.nonstdout = stdout;
Rob Landley53302f82004-02-18 09:54:15 +00001292
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001293 /* argv[0..(argc-1)] should be names of file to process. If no
Mark Whitley6315ce62000-07-10 22:55:51 +00001294 * files were specified or '-' was specified, take input from stdin.
1295 * Otherwise, we process all the files specified. */
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001296 if (argv[0] == NULL) {
1297 if (opt & OPT_in_place)
Denis Vlasenko80778502006-10-25 12:46:46 +00001298 bb_error_msg_and_die(bb_msg_requires_arg, "-i");
Rob Landleydcc28662004-11-25 07:21:47 +00001299 add_input_file(stdin);
1300 process_files();
Glenn L McGrath8d6395d2003-04-08 11:56:11 +00001301 } else {
Mark Whitley6315ce62000-07-10 22:55:51 +00001302 int i;
1303 FILE *file;
Glenn L McGrath8d6395d2003-04-08 11:56:11 +00001304
Denis Vlasenko62a90cd2008-03-17 09:07:36 +00001305 for (i = 0; argv[i]; i++) {
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001306 struct stat statbuf;
1307 int nonstdoutfd;
1308
Denis Vlasenko9f739442006-12-16 23:49:13 +00001309 if (LONE_DASH(argv[i]) && !(opt & OPT_in_place)) {
Rob Landleydcc28662004-11-25 07:21:47 +00001310 add_input_file(stdin);
1311 process_files();
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001312 continue;
Glenn L McGrath8aac05b2003-09-14 04:06:12 +00001313 }
Denis Vlasenkoddec5af2006-10-26 23:25:17 +00001314 file = fopen_or_warn(argv[i], "r");
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001315 if (!file) {
1316 status = EXIT_FAILURE;
1317 continue;
1318 }
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001319 if (!(opt & OPT_in_place)) {
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001320 add_input_file(file);
1321 continue;
1322 }
1323
Denis Vlasenko17a15262007-03-26 20:48:46 +00001324 G.outname = xasprintf("%sXXXXXX", argv[i]);
1325 nonstdoutfd = mkstemp(G.outname);
Denis Vlasenko80778502006-10-25 12:46:46 +00001326 if (-1 == nonstdoutfd)
Denis Vlasenko17a15262007-03-26 20:48:46 +00001327 bb_perror_msg_and_die("cannot create temp file %s", G.outname);
1328 G.nonstdout = fdopen(nonstdoutfd, "w");
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001329
1330 /* Set permissions of output file */
1331
Denis Vlasenko2f8f71b2006-12-10 02:09:12 +00001332 fstat(fileno(file), &statbuf);
1333 fchmod(nonstdoutfd, statbuf.st_mode);
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001334 add_input_file(file);
1335 process_files();
Denis Vlasenko17a15262007-03-26 20:48:46 +00001336 fclose(G.nonstdout);
Denis Vlasenkod18a3a22006-10-25 12:46:03 +00001337
Denis Vlasenko17a15262007-03-26 20:48:46 +00001338 G.nonstdout = stdout;
Denis Vlasenko80778502006-10-25 12:46:46 +00001339 /* unlink(argv[i]); */
Denis Vlasenkocb448fe2008-02-17 14:28:53 +00001340 xrename(G.outname, argv[i]);
Denis Vlasenko17a15262007-03-26 20:48:46 +00001341 free(G.outname);
Denis Vlasenko40276642007-11-13 16:48:10 +00001342 G.outname = NULL;
Mark Whitley6315ce62000-07-10 22:55:51 +00001343 }
Denis Vlasenko17a15262007-03-26 20:48:46 +00001344 if (G.input_file_count > G.current_input_file)
Denis Vlasenko80778502006-10-25 12:46:46 +00001345 process_files();
Mark Whitley6315ce62000-07-10 22:55:51 +00001346 }
Glenn L McGrath8d6395d2003-04-08 11:56:11 +00001347
Matt Kraaia5f09c62001-11-12 16:44:55 +00001348 return status;
Eric Andersen6b6b3f61999-10-28 16:06:25 +00001349}