blob: 1d704e87c8496cb8787fe62f8aa5794a2a78b300 [file] [log] [blame]
Glenn L McGrath545106f2002-11-11 06:21:00 +00001/* vi: set sw=4 ts=4: */
2/*
3 * awk implementation for busybox
4 *
5 * Copyright (C) 2002 by Dmitry Zakharov <dmit@crp.bank.gov.ua>
6 *
Bernhard Reutner-Fischer86f5c992006-01-22 22:55:11 +00007 * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
Glenn L McGrath545106f2002-11-11 06:21:00 +00008 */
9
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000010#include "libbb.h"
Rob Landleyd921b2e2006-08-03 15:41:12 +000011#include "xregex.h"
12#include <math.h>
Glenn L McGrath545106f2002-11-11 06:21:00 +000013
Denis Vlasenko99912ca2007-04-10 15:43:37 +000014/* This is a NOEXEC applet. Be very careful! */
15
Glenn L McGrath545106f2002-11-11 06:21:00 +000016
Denis Vlasenko629563b2007-02-24 17:05:52 +000017#define MAXVARFMT 240
18#define MINNVBLOCK 64
Glenn L McGrath545106f2002-11-11 06:21:00 +000019
20/* variable flags */
Denis Vlasenko629563b2007-02-24 17:05:52 +000021#define VF_NUMBER 0x0001 /* 1 = primary type is number */
22#define VF_ARRAY 0x0002 /* 1 = it's an array */
Glenn L McGrath545106f2002-11-11 06:21:00 +000023
Denis Vlasenko629563b2007-02-24 17:05:52 +000024#define VF_CACHED 0x0100 /* 1 = num/str value has cached str/num eq */
25#define VF_USER 0x0200 /* 1 = user input (may be numeric string) */
26#define VF_SPECIAL 0x0400 /* 1 = requires extra handling when changed */
27#define VF_WALK 0x0800 /* 1 = variable has alloc'd x.walker list */
28#define VF_FSTR 0x1000 /* 1 = var::string points to fstring buffer */
29#define VF_CHILD 0x2000 /* 1 = function arg; x.parent points to source */
30#define VF_DIRTY 0x4000 /* 1 = variable was set explicitly */
Glenn L McGrath545106f2002-11-11 06:21:00 +000031
32/* these flags are static, don't change them when value is changed */
Denis Vlasenko629563b2007-02-24 17:05:52 +000033#define VF_DONTTOUCH (VF_ARRAY | VF_SPECIAL | VF_WALK | VF_CHILD | VF_DIRTY)
Glenn L McGrath545106f2002-11-11 06:21:00 +000034
35/* Variable */
36typedef struct var_s {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +000037 unsigned type; /* flags */
Glenn L McGrath545106f2002-11-11 06:21:00 +000038 double number;
39 char *string;
40 union {
Denis Vlasenko629563b2007-02-24 17:05:52 +000041 int aidx; /* func arg idx (for compilation stage) */
42 struct xhash_s *array; /* array ptr */
43 struct var_s *parent; /* for func args, ptr to actual parameter */
44 char **walker; /* list of array elements (for..in) */
Glenn L McGrath545106f2002-11-11 06:21:00 +000045 } x;
46} var;
47
48/* Node chain (pattern-action chain, BEGIN, END, function bodies) */
49typedef struct chain_s {
50 struct node_s *first;
51 struct node_s *last;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +000052 const char *programname;
Glenn L McGrath545106f2002-11-11 06:21:00 +000053} chain;
54
55/* Function */
56typedef struct func_s {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +000057 unsigned nargs;
Glenn L McGrath545106f2002-11-11 06:21:00 +000058 struct chain_s body;
59} func;
60
61/* I/O stream */
62typedef struct rstream_s {
63 FILE *F;
64 char *buffer;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +000065 int adv;
Glenn L McGrath545106f2002-11-11 06:21:00 +000066 int size;
67 int pos;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +000068 smallint is_pipe;
Glenn L McGrath545106f2002-11-11 06:21:00 +000069} rstream;
70
71typedef struct hash_item_s {
72 union {
Denis Vlasenkoffba9412007-05-17 23:03:35 +000073 struct var_s v; /* variable/array hash */
74 struct rstream_s rs; /* redirect streams hash */
75 struct func_s f; /* functions hash */
Glenn L McGrath545106f2002-11-11 06:21:00 +000076 } data;
Denis Vlasenkoffba9412007-05-17 23:03:35 +000077 struct hash_item_s *next; /* next in chain */
78 char name[1]; /* really it's longer */
Glenn L McGrath545106f2002-11-11 06:21:00 +000079} hash_item;
80
81typedef struct xhash_s {
Denis Vlasenkoffba9412007-05-17 23:03:35 +000082 unsigned nel; /* num of elements */
83 unsigned csize; /* current hash size */
84 unsigned nprime; /* next hash size in PRIMES[] */
85 unsigned glen; /* summary length of item names */
Glenn L McGrath545106f2002-11-11 06:21:00 +000086 struct hash_item_s **items;
87} xhash;
88
89/* Tree node */
90typedef struct node_s {
Mike Frysingerf87b3e32005-09-27 04:16:22 +000091 uint32_t info;
Denis Vlasenkocd5c7862007-05-17 16:37:22 +000092 unsigned lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +000093 union {
94 struct node_s *n;
95 var *v;
96 int i;
97 char *s;
98 regex_t *re;
99 } l;
100 union {
101 struct node_s *n;
102 regex_t *ire;
103 func *f;
104 int argno;
105 } r;
106 union {
107 struct node_s *n;
108 } a;
109} node;
110
111/* Block of temporary variables */
112typedef struct nvblock_s {
113 int size;
114 var *pos;
115 struct nvblock_s *prev;
116 struct nvblock_s *next;
Denys Vlasenkod069e532009-09-09 23:12:10 +0200117 var nv[];
Glenn L McGrath545106f2002-11-11 06:21:00 +0000118} nvblock;
119
120typedef struct tsplitter_s {
121 node n;
122 regex_t re[2];
123} tsplitter;
124
125/* simple token classes */
126/* Order and hex values are very important!!! See next_token() */
127#define TC_SEQSTART 1 /* ( */
128#define TC_SEQTERM (1 << 1) /* ) */
129#define TC_REGEXP (1 << 2) /* /.../ */
130#define TC_OUTRDR (1 << 3) /* | > >> */
131#define TC_UOPPOST (1 << 4) /* unary postfix operator */
132#define TC_UOPPRE1 (1 << 5) /* unary prefix operator */
133#define TC_BINOPX (1 << 6) /* two-opnd operator */
134#define TC_IN (1 << 7)
135#define TC_COMMA (1 << 8)
136#define TC_PIPE (1 << 9) /* input redirection pipe */
137#define TC_UOPPRE2 (1 << 10) /* unary prefix operator */
138#define TC_ARRTERM (1 << 11) /* ] */
139#define TC_GRPSTART (1 << 12) /* { */
140#define TC_GRPTERM (1 << 13) /* } */
141#define TC_SEMICOL (1 << 14)
142#define TC_NEWLINE (1 << 15)
143#define TC_STATX (1 << 16) /* ctl statement (for, next...) */
144#define TC_WHILE (1 << 17)
145#define TC_ELSE (1 << 18)
146#define TC_BUILTIN (1 << 19)
147#define TC_GETLINE (1 << 20)
148#define TC_FUNCDECL (1 << 21) /* `function' `func' */
149#define TC_BEGIN (1 << 22)
150#define TC_END (1 << 23)
151#define TC_EOF (1 << 24)
152#define TC_VARIABLE (1 << 25)
153#define TC_ARRAY (1 << 26)
154#define TC_FUNCTION (1 << 27)
155#define TC_STRING (1 << 28)
156#define TC_NUMBER (1 << 29)
157
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000158#define TC_UOPPRE (TC_UOPPRE1 | TC_UOPPRE2)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000159
160/* combined token classes */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000161#define TC_BINOP (TC_BINOPX | TC_COMMA | TC_PIPE | TC_IN)
162#define TC_UNARYOP (TC_UOPPRE | TC_UOPPOST)
163#define TC_OPERAND (TC_VARIABLE | TC_ARRAY | TC_FUNCTION \
164 | TC_BUILTIN | TC_GETLINE | TC_SEQSTART | TC_STRING | TC_NUMBER)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000165
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000166#define TC_STATEMNT (TC_STATX | TC_WHILE)
167#define TC_OPTERM (TC_SEMICOL | TC_NEWLINE)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000168
169/* word tokens, cannot mean something else if not expected */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000170#define TC_WORD (TC_IN | TC_STATEMNT | TC_ELSE | TC_BUILTIN \
171 | TC_GETLINE | TC_FUNCDECL | TC_BEGIN | TC_END)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000172
173/* discard newlines after these */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000174#define TC_NOTERM (TC_COMMA | TC_GRPSTART | TC_GRPTERM \
175 | TC_BINOP | TC_OPTERM)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000176
177/* what can expression begin with */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000178#define TC_OPSEQ (TC_OPERAND | TC_UOPPRE | TC_REGEXP)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000179/* what can group begin with */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000180#define TC_GRPSEQ (TC_OPSEQ | TC_OPTERM | TC_STATEMNT | TC_GRPSTART)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000181
182/* if previous token class is CONCAT1 and next is CONCAT2, concatenation */
183/* operator is inserted between them */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000184#define TC_CONCAT1 (TC_VARIABLE | TC_ARRTERM | TC_SEQTERM \
185 | TC_STRING | TC_NUMBER | TC_UOPPOST)
186#define TC_CONCAT2 (TC_OPERAND | TC_UOPPRE)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000187
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000188#define OF_RES1 0x010000
189#define OF_RES2 0x020000
190#define OF_STR1 0x040000
191#define OF_STR2 0x080000
192#define OF_NUM1 0x100000
193#define OF_CHECKED 0x200000
Glenn L McGrath545106f2002-11-11 06:21:00 +0000194
195/* combined operator flags */
196#define xx 0
197#define xV OF_RES2
198#define xS (OF_RES2 | OF_STR2)
199#define Vx OF_RES1
200#define VV (OF_RES1 | OF_RES2)
201#define Nx (OF_RES1 | OF_NUM1)
202#define NV (OF_RES1 | OF_NUM1 | OF_RES2)
203#define Sx (OF_RES1 | OF_STR1)
204#define SV (OF_RES1 | OF_STR1 | OF_RES2)
205#define SS (OF_RES1 | OF_STR1 | OF_RES2 | OF_STR2)
206
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000207#define OPCLSMASK 0xFF00
208#define OPNMASK 0x007F
Glenn L McGrath545106f2002-11-11 06:21:00 +0000209
210/* operator priority is a highest byte (even: r->l, odd: l->r grouping)
211 * For builtins it has different meaning: n n s3 s2 s1 v3 v2 v1,
212 * n - min. number of args, vN - resolve Nth arg to var, sN - resolve to string
213 */
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000214#define P(x) (x << 24)
215#define PRIMASK 0x7F000000
216#define PRIMASK2 0x7E000000
Glenn L McGrath545106f2002-11-11 06:21:00 +0000217
218/* Operation classes */
219
220#define SHIFT_TIL_THIS 0x0600
221#define RECUR_FROM_THIS 0x1000
222
223enum {
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000224 OC_DELETE = 0x0100, OC_EXEC = 0x0200, OC_NEWSOURCE = 0x0300,
225 OC_PRINT = 0x0400, OC_PRINTF = 0x0500, OC_WALKINIT = 0x0600,
Glenn L McGrath545106f2002-11-11 06:21:00 +0000226
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000227 OC_BR = 0x0700, OC_BREAK = 0x0800, OC_CONTINUE = 0x0900,
228 OC_EXIT = 0x0a00, OC_NEXT = 0x0b00, OC_NEXTFILE = 0x0c00,
229 OC_TEST = 0x0d00, OC_WALKNEXT = 0x0e00,
Glenn L McGrath545106f2002-11-11 06:21:00 +0000230
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000231 OC_BINARY = 0x1000, OC_BUILTIN = 0x1100, OC_COLON = 0x1200,
232 OC_COMMA = 0x1300, OC_COMPARE = 0x1400, OC_CONCAT = 0x1500,
233 OC_FBLTIN = 0x1600, OC_FIELD = 0x1700, OC_FNARG = 0x1800,
234 OC_FUNC = 0x1900, OC_GETLINE = 0x1a00, OC_IN = 0x1b00,
235 OC_LAND = 0x1c00, OC_LOR = 0x1d00, OC_MATCH = 0x1e00,
236 OC_MOVE = 0x1f00, OC_PGETLINE = 0x2000, OC_REGEXP = 0x2100,
237 OC_REPLACE = 0x2200, OC_RETURN = 0x2300, OC_SPRINTF = 0x2400,
238 OC_TERNARY = 0x2500, OC_UNARY = 0x2600, OC_VAR = 0x2700,
239 OC_DONE = 0x2800,
Glenn L McGrath545106f2002-11-11 06:21:00 +0000240
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000241 ST_IF = 0x3000, ST_DO = 0x3100, ST_FOR = 0x3200,
242 ST_WHILE = 0x3300
Glenn L McGrath545106f2002-11-11 06:21:00 +0000243};
244
245/* simple builtins */
246enum {
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000247 F_in, F_rn, F_co, F_ex, F_lg, F_si, F_sq, F_sr,
Glenn L McGrath545106f2002-11-11 06:21:00 +0000248 F_ti, F_le, F_sy, F_ff, F_cl
249};
250
251/* builtins */
252enum {
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +0200253 B_a2, B_ix, B_ma, B_sp, B_ss, B_ti, B_mt, B_lo, B_up,
Denis Vlasenkoe175ff22006-09-26 17:41:00 +0000254 B_ge, B_gs, B_su,
255 B_an, B_co, B_ls, B_or, B_rs, B_xo,
Glenn L McGrath545106f2002-11-11 06:21:00 +0000256};
257
258/* tokens and their corresponding info values */
259
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000260#define NTC "\377" /* switch to next token class (tc<<1) */
261#define NTCC '\377'
Glenn L McGrath545106f2002-11-11 06:21:00 +0000262
263#define OC_B OC_BUILTIN
264
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000265static const char tokenlist[] ALIGN1 =
Denis Vlasenkof782f522007-01-01 23:51:30 +0000266 "\1(" NTC
267 "\1)" NTC
268 "\1/" NTC /* REGEXP */
269 "\2>>" "\1>" "\1|" NTC /* OUTRDR */
270 "\2++" "\2--" NTC /* UOPPOST */
271 "\2++" "\2--" "\1$" NTC /* UOPPRE1 */
272 "\2==" "\1=" "\2+=" "\2-=" /* BINOPX */
273 "\2*=" "\2/=" "\2%=" "\2^="
274 "\1+" "\1-" "\3**=" "\2**"
275 "\1/" "\1%" "\1^" "\1*"
276 "\2!=" "\2>=" "\2<=" "\1>"
277 "\1<" "\2!~" "\1~" "\2&&"
278 "\2||" "\1?" "\1:" NTC
279 "\2in" NTC
280 "\1," NTC
281 "\1|" NTC
282 "\1+" "\1-" "\1!" NTC /* UOPPRE2 */
283 "\1]" NTC
284 "\1{" NTC
285 "\1}" NTC
286 "\1;" NTC
287 "\1\n" NTC
288 "\2if" "\2do" "\3for" "\5break" /* STATX */
289 "\10continue" "\6delete" "\5print"
290 "\6printf" "\4next" "\10nextfile"
291 "\6return" "\4exit" NTC
292 "\5while" NTC
293 "\4else" NTC
Glenn L McGrath545106f2002-11-11 06:21:00 +0000294
Denis Vlasenkof782f522007-01-01 23:51:30 +0000295 "\3and" "\5compl" "\6lshift" "\2or"
296 "\6rshift" "\3xor"
297 "\5close" "\6system" "\6fflush" "\5atan2" /* BUILTIN */
298 "\3cos" "\3exp" "\3int" "\3log"
299 "\4rand" "\3sin" "\4sqrt" "\5srand"
300 "\6gensub" "\4gsub" "\5index" "\6length"
301 "\5match" "\5split" "\7sprintf" "\3sub"
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +0200302 "\6substr" "\7systime" "\10strftime" "\6mktime"
Denis Vlasenkof782f522007-01-01 23:51:30 +0000303 "\7tolower" "\7toupper" NTC
304 "\7getline" NTC
305 "\4func" "\10function" NTC
306 "\5BEGIN" NTC
307 "\3END" "\0"
Glenn L McGrath545106f2002-11-11 06:21:00 +0000308 ;
309
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000310static const uint32_t tokeninfo[] = {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000311 0,
312 0,
313 OC_REGEXP,
Denis Vlasenkof782f522007-01-01 23:51:30 +0000314 xS|'a', xS|'w', xS|'|',
315 OC_UNARY|xV|P(9)|'p', OC_UNARY|xV|P(9)|'m',
316 OC_UNARY|xV|P(9)|'P', OC_UNARY|xV|P(9)|'M',
317 OC_FIELD|xV|P(5),
318 OC_COMPARE|VV|P(39)|5, OC_MOVE|VV|P(74),
319 OC_REPLACE|NV|P(74)|'+', OC_REPLACE|NV|P(74)|'-',
320 OC_REPLACE|NV|P(74)|'*', OC_REPLACE|NV|P(74)|'/',
321 OC_REPLACE|NV|P(74)|'%', OC_REPLACE|NV|P(74)|'&',
322 OC_BINARY|NV|P(29)|'+', OC_BINARY|NV|P(29)|'-',
323 OC_REPLACE|NV|P(74)|'&', OC_BINARY|NV|P(15)|'&',
324 OC_BINARY|NV|P(25)|'/', OC_BINARY|NV|P(25)|'%',
325 OC_BINARY|NV|P(15)|'&', OC_BINARY|NV|P(25)|'*',
326 OC_COMPARE|VV|P(39)|4, OC_COMPARE|VV|P(39)|3,
327 OC_COMPARE|VV|P(39)|0, OC_COMPARE|VV|P(39)|1,
328 OC_COMPARE|VV|P(39)|2, OC_MATCH|Sx|P(45)|'!',
329 OC_MATCH|Sx|P(45)|'~', OC_LAND|Vx|P(55),
330 OC_LOR|Vx|P(59), OC_TERNARY|Vx|P(64)|'?',
331 OC_COLON|xx|P(67)|':',
Glenn L McGrath545106f2002-11-11 06:21:00 +0000332 OC_IN|SV|P(49),
333 OC_COMMA|SS|P(80),
334 OC_PGETLINE|SV|P(37),
Denis Vlasenkof782f522007-01-01 23:51:30 +0000335 OC_UNARY|xV|P(19)|'+', OC_UNARY|xV|P(19)|'-',
336 OC_UNARY|xV|P(19)|'!',
Glenn L McGrath545106f2002-11-11 06:21:00 +0000337 0,
338 0,
339 0,
340 0,
341 0,
Denis Vlasenkof782f522007-01-01 23:51:30 +0000342 ST_IF, ST_DO, ST_FOR, OC_BREAK,
343 OC_CONTINUE, OC_DELETE|Vx, OC_PRINT,
344 OC_PRINTF, OC_NEXT, OC_NEXTFILE,
345 OC_RETURN|Vx, OC_EXIT|Nx,
Glenn L McGrath545106f2002-11-11 06:21:00 +0000346 ST_WHILE,
347 0,
348
Denis Vlasenkoe175ff22006-09-26 17:41:00 +0000349 OC_B|B_an|P(0x83), OC_B|B_co|P(0x41), OC_B|B_ls|P(0x83), OC_B|B_or|P(0x83),
350 OC_B|B_rs|P(0x83), OC_B|B_xo|P(0x83),
Glenn L McGrath545106f2002-11-11 06:21:00 +0000351 OC_FBLTIN|Sx|F_cl, OC_FBLTIN|Sx|F_sy, OC_FBLTIN|Sx|F_ff, OC_B|B_a2|P(0x83),
352 OC_FBLTIN|Nx|F_co, OC_FBLTIN|Nx|F_ex, OC_FBLTIN|Nx|F_in, OC_FBLTIN|Nx|F_lg,
353 OC_FBLTIN|F_rn, OC_FBLTIN|Nx|F_si, OC_FBLTIN|Nx|F_sq, OC_FBLTIN|Nx|F_sr,
354 OC_B|B_ge|P(0xd6), OC_B|B_gs|P(0xb6), OC_B|B_ix|P(0x9b), OC_FBLTIN|Sx|F_le,
355 OC_B|B_ma|P(0x89), OC_B|B_sp|P(0x8b), OC_SPRINTF, OC_B|B_su|P(0xb6),
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +0200356 OC_B|B_ss|P(0x8f), OC_FBLTIN|F_ti, OC_B|B_ti|P(0x0b), OC_B|B_mt|P(0x0b),
Glenn L McGrath545106f2002-11-11 06:21:00 +0000357 OC_B|B_lo|P(0x49), OC_B|B_up|P(0x49),
358 OC_GETLINE|SV|P(0),
359 0, 0,
360 0,
361 0
362};
363
364/* internal variable names and their initial values */
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000365/* asterisk marks SPECIAL vars; $ is just no-named Field0 */
Glenn L McGrath545106f2002-11-11 06:21:00 +0000366enum {
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000367 CONVFMT, OFMT, FS, OFS,
Denis Vlasenkof782f522007-01-01 23:51:30 +0000368 ORS, RS, RT, FILENAME,
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +0000369 SUBSEP, F0, ARGIND, ARGC,
370 ARGV, ERRNO, FNR, NR,
371 NF, IGNORECASE, ENVIRON, NUM_INTERNAL_VARS
Glenn L McGrath545106f2002-11-11 06:21:00 +0000372};
373
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000374static const char vNames[] ALIGN1 =
Denis Vlasenkof782f522007-01-01 23:51:30 +0000375 "CONVFMT\0" "OFMT\0" "FS\0*" "OFS\0"
376 "ORS\0" "RS\0*" "RT\0" "FILENAME\0"
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +0000377 "SUBSEP\0" "$\0*" "ARGIND\0" "ARGC\0"
378 "ARGV\0" "ERRNO\0" "FNR\0" "NR\0"
379 "NF\0*" "IGNORECASE\0*" "ENVIRON\0" "\0";
Glenn L McGrath545106f2002-11-11 06:21:00 +0000380
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000381static const char vValues[] ALIGN1 =
Denis Vlasenkof782f522007-01-01 23:51:30 +0000382 "%.6g\0" "%.6g\0" " \0" " \0"
383 "\n\0" "\n\0" "\0" "\0"
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +0000384 "\034\0" "\0" "\377";
Glenn L McGrath545106f2002-11-11 06:21:00 +0000385
386/* hash size may grow to these values */
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000387#define FIRST_PRIME 61
388static const uint16_t PRIMES[] ALIGN2 = { 251, 1021, 4093, 16381, 65521 };
Glenn L McGrath545106f2002-11-11 06:21:00 +0000389
Glenn L McGrath545106f2002-11-11 06:21:00 +0000390
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000391/* Globals. Split in two parts so that first one is addressed
Denis Vlasenko9aa5c652009-02-26 11:21:04 +0000392 * with (mostly short) negative offsets.
393 * NB: it's unsafe to put members of type "double"
394 * into globals2 (gcc may fail to align them).
395 */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000396struct globals {
Denis Vlasenko9aa5c652009-02-26 11:21:04 +0000397 double t_double;
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000398 chain beginseq, mainseq, endseq;
399 chain *seq;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000400 node *break_ptr, *continue_ptr;
401 rstream *iF;
402 xhash *vhash, *ahash, *fdhash, *fnhash;
403 const char *g_progname;
404 int g_lineno;
405 int nfields;
406 int maxfields; /* used in fsrealloc() only */
407 var *Fields;
408 nvblock *g_cb;
409 char *g_pos;
410 char *g_buf;
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000411 smallint icase;
412 smallint exiting;
413 smallint nextrec;
414 smallint nextfile;
415 smallint is_f0_split;
416};
417struct globals2 {
418 uint32_t t_info; /* often used */
419 uint32_t t_tclass;
420 char *t_string;
421 int t_lineno;
422 int t_rollback;
423
424 var *intvar[NUM_INTERNAL_VARS]; /* often used */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000425
426 /* former statics from various functions */
427 char *split_f0__fstrings;
428
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000429 uint32_t next_token__save_tclass;
430 uint32_t next_token__save_info;
431 uint32_t next_token__ltclass;
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000432 smallint next_token__concat_inserted;
433
434 smallint next_input_file__files_happen;
435 rstream next_input_file__rsm;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000436
437 var *evaluate__fnargs;
438 unsigned evaluate__seed;
439 regex_t evaluate__sreg;
440
441 var ptest__v;
442
443 tsplitter exec_builtin__tspl;
444
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000445 /* biggest and least used members go last */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000446 tsplitter fsplitter, rsplitter;
Denys Vlasenko3dbc5a92010-02-05 14:54:22 +0100447};
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000448#define G1 (ptr_to_globals[-1])
Denis Vlasenko574f2f42008-02-27 18:41:59 +0000449#define G (*(struct globals2 *)ptr_to_globals)
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000450/* For debug. nm --size-sort awk.o | grep -vi ' [tr] ' */
Denis Vlasenko9aa5c652009-02-26 11:21:04 +0000451/*char G1size[sizeof(G1)]; - 0x74 */
452/*char Gsize[sizeof(G)]; - 0x1c4 */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000453/* Trying to keep most of members accessible with short offsets: */
Denis Vlasenko9aa5c652009-02-26 11:21:04 +0000454/*char Gofs_seed[offsetof(struct globals2, evaluate__seed)]; - 0x90 */
455#define t_double (G1.t_double )
Denis Vlasenkoe16e6e72007-06-07 13:14:53 +0000456#define beginseq (G1.beginseq )
457#define mainseq (G1.mainseq )
458#define endseq (G1.endseq )
459#define seq (G1.seq )
460#define break_ptr (G1.break_ptr )
461#define continue_ptr (G1.continue_ptr)
462#define iF (G1.iF )
463#define vhash (G1.vhash )
464#define ahash (G1.ahash )
465#define fdhash (G1.fdhash )
466#define fnhash (G1.fnhash )
467#define g_progname (G1.g_progname )
468#define g_lineno (G1.g_lineno )
469#define nfields (G1.nfields )
470#define maxfields (G1.maxfields )
471#define Fields (G1.Fields )
472#define g_cb (G1.g_cb )
473#define g_pos (G1.g_pos )
474#define g_buf (G1.g_buf )
475#define icase (G1.icase )
476#define exiting (G1.exiting )
477#define nextrec (G1.nextrec )
478#define nextfile (G1.nextfile )
479#define is_f0_split (G1.is_f0_split )
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000480#define t_info (G.t_info )
481#define t_tclass (G.t_tclass )
482#define t_string (G.t_string )
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000483#define t_lineno (G.t_lineno )
484#define t_rollback (G.t_rollback )
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000485#define intvar (G.intvar )
486#define fsplitter (G.fsplitter )
487#define rsplitter (G.rsplitter )
488#define INIT_G() do { \
Denys Vlasenko90a99042009-09-06 02:36:23 +0200489 SET_PTR_TO_GLOBALS((char*)xzalloc(sizeof(G1)+sizeof(G)) + sizeof(G1)); \
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000490 G.next_token__ltclass = TC_OPTERM; \
491 G.evaluate__seed = 1; \
492} while (0)
493
Glenn L McGrath545106f2002-11-11 06:21:00 +0000494
495/* function prototypes */
Glenn L McGrath545106f2002-11-11 06:21:00 +0000496static void handle_special(var *);
Mike Frysingerf87b3e32005-09-27 04:16:22 +0000497static node *parse_expr(uint32_t);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000498static void chain_group(void);
499static var *evaluate(node *, var *);
500static rstream *next_input_file(void);
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000501static int fmt_num(char *, int, const char *, double, int);
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000502static int awk_exit(int) NORETURN;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000503
504/* ---- error handling ---- */
505
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000506static const char EMSG_INTERNAL_ERROR[] ALIGN1 = "Internal error";
507static const char EMSG_UNEXP_EOS[] ALIGN1 = "Unexpected end of string";
508static const char EMSG_UNEXP_TOKEN[] ALIGN1 = "Unexpected token";
509static const char EMSG_DIV_BY_ZERO[] ALIGN1 = "Division by zero";
510static const char EMSG_INV_FMT[] ALIGN1 = "Invalid format specifier";
511static const char EMSG_TOO_FEW_ARGS[] ALIGN1 = "Too few arguments for builtin";
512static const char EMSG_NOT_ARRAY[] ALIGN1 = "Not an array";
513static const char EMSG_POSSIBLE_ERROR[] ALIGN1 = "Possible syntax error";
514static const char EMSG_UNDEF_FUNC[] ALIGN1 = "Call to undefined function";
Denis Vlasenko2d5bd802008-10-24 10:49:49 +0000515#if !ENABLE_FEATURE_AWK_LIBM
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000516static const char EMSG_NO_MATH[] ALIGN1 = "Math support is not compiled in";
Glenn L McGrath545106f2002-11-11 06:21:00 +0000517#endif
518
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100519static void zero_out_var(var *vp)
Denis Vlasenkof782f522007-01-01 23:51:30 +0000520{
521 memset(vp, 0, sizeof(*vp));
522}
523
Denis Vlasenkoc7cc5a92009-04-19 01:27:20 +0000524static void syntax_error(const char *message) NORETURN;
525static void syntax_error(const char *message)
Glenn L McGrathd4036f82002-11-28 09:30:40 +0000526{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000527 bb_error_msg_and_die("%s:%i: %s", g_progname, g_lineno, message);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000528}
529
Glenn L McGrath545106f2002-11-11 06:21:00 +0000530/* ---- hash stuff ---- */
531
Denis Vlasenkof782f522007-01-01 23:51:30 +0000532static unsigned hashidx(const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000533{
Denis Vlasenkof782f522007-01-01 23:51:30 +0000534 unsigned idx = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000535
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100536 while (*name)
537 idx = *name++ + (idx << 6) - idx;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000538 return idx;
539}
540
541/* create new hash */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000542static xhash *hash_init(void)
543{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000544 xhash *newhash;
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000545
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100546 newhash = xzalloc(sizeof(*newhash));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000547 newhash->csize = FIRST_PRIME;
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100548 newhash->items = xzalloc(FIRST_PRIME * sizeof(newhash->items[0]));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000549
550 return newhash;
551}
552
553/* find item in hash, return ptr to data, NULL if not found */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000554static void *hash_search(xhash *hash, const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000555{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000556 hash_item *hi;
557
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100558 hi = hash->items[hashidx(name) % hash->csize];
Glenn L McGrath545106f2002-11-11 06:21:00 +0000559 while (hi) {
560 if (strcmp(hi->name, name) == 0)
561 return &(hi->data);
562 hi = hi->next;
563 }
564 return NULL;
565}
566
567/* grow hash if it becomes too big */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000568static void hash_rebuild(xhash *hash)
569{
Denis Vlasenkof782f522007-01-01 23:51:30 +0000570 unsigned newsize, i, idx;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000571 hash_item **newitems, *hi, *thi;
572
Denis Vlasenko80b8b392007-06-25 10:55:35 +0000573 if (hash->nprime == ARRAY_SIZE(PRIMES))
Glenn L McGrath545106f2002-11-11 06:21:00 +0000574 return;
575
576 newsize = PRIMES[hash->nprime++];
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100577 newitems = xzalloc(newsize * sizeof(newitems[0]));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000578
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000579 for (i = 0; i < hash->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000580 hi = hash->items[i];
581 while (hi) {
582 thi = hi;
583 hi = thi->next;
584 idx = hashidx(thi->name) % newsize;
585 thi->next = newitems[idx];
586 newitems[idx] = thi;
587 }
588 }
589
590 free(hash->items);
591 hash->csize = newsize;
592 hash->items = newitems;
593}
594
595/* find item in hash, add it if necessary. Return ptr to data */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000596static void *hash_find(xhash *hash, const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000597{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000598 hash_item *hi;
Denis Vlasenkof782f522007-01-01 23:51:30 +0000599 unsigned idx;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000600 int l;
601
602 hi = hash_search(hash, name);
Denis Vlasenkob78c7822007-07-18 18:31:11 +0000603 if (!hi) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000604 if (++hash->nel / hash->csize > 10)
605 hash_rebuild(hash);
606
Rob Landleya3896512006-05-07 20:20:34 +0000607 l = strlen(name) + 1;
Denis Vlasenko7a676642009-03-15 22:20:31 +0000608 hi = xzalloc(sizeof(*hi) + l);
609 strcpy(hi->name, name);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000610
611 idx = hashidx(name) % hash->csize;
612 hi->next = hash->items[idx];
613 hash->items[idx] = hi;
614 hash->glen += l;
615 }
616 return &(hi->data);
617}
618
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000619#define findvar(hash, name) ((var*) hash_find((hash), (name)))
620#define newvar(name) ((var*) hash_find(vhash, (name)))
621#define newfile(name) ((rstream*)hash_find(fdhash, (name)))
622#define newfunc(name) ((func*) hash_find(fnhash, (name)))
Glenn L McGrath545106f2002-11-11 06:21:00 +0000623
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000624static void hash_remove(xhash *hash, const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000625{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000626 hash_item *hi, **phi;
627
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000628 phi = &(hash->items[hashidx(name) % hash->csize]);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000629 while (*phi) {
630 hi = *phi;
631 if (strcmp(hi->name, name) == 0) {
Rob Landleya3896512006-05-07 20:20:34 +0000632 hash->glen -= (strlen(name) + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000633 hash->nel--;
634 *phi = hi->next;
635 free(hi);
636 break;
637 }
638 phi = &(hi->next);
639 }
640}
641
642/* ------ some useful functions ------ */
643
Mike Frysinger10a11e22005-09-27 02:23:02 +0000644static void skip_spaces(char **s)
645{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000646 char *p = *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000647
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000648 while (1) {
649 if (*p == '\\' && p[1] == '\n') {
650 p++;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000651 t_lineno++;
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000652 } else if (*p != ' ' && *p != '\t') {
653 break;
654 }
Mike Frysingerde2b9382005-09-27 03:18:00 +0000655 p++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000656 }
657 *s = p;
658}
659
Mike Frysinger10a11e22005-09-27 02:23:02 +0000660static char *nextword(char **s)
661{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000662 char *p = *s;
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100663 while (*(*s)++)
664 continue;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000665 return p;
666}
667
Mike Frysinger10a11e22005-09-27 02:23:02 +0000668static char nextchar(char **s)
669{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000670 char c, *pps;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000671
672 c = *((*s)++);
673 pps = *s;
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100674 if (c == '\\')
675 c = bb_process_escape_sequence((const char**)s);
676 if (c == '\\' && *s == pps)
677 c = *((*s)++);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000678 return c;
679}
680
Denis Vlasenko77ad97f2008-05-13 02:27:31 +0000681static ALWAYS_INLINE int isalnum_(int c)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000682{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000683 return (isalnum(c) || c == '_');
684}
685
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000686static double my_strtod(char **pp)
687{
688#if ENABLE_DESKTOP
689 if ((*pp)[0] == '0'
690 && ((((*pp)[1] | 0x20) == 'x') || isdigit((*pp)[1]))
691 ) {
692 return strtoull(*pp, pp, 0);
693 }
694#endif
695 return strtod(*pp, pp);
696}
697
Glenn L McGrath545106f2002-11-11 06:21:00 +0000698/* -------- working with variables (set/get/copy/etc) -------- */
699
Mike Frysinger10a11e22005-09-27 02:23:02 +0000700static xhash *iamarray(var *v)
701{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000702 var *a = v;
703
704 while (a->type & VF_CHILD)
705 a = a->x.parent;
706
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000707 if (!(a->type & VF_ARRAY)) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000708 a->type |= VF_ARRAY;
709 a->x.array = hash_init();
710 }
711 return a->x.array;
712}
713
Mike Frysinger10a11e22005-09-27 02:23:02 +0000714static void clear_array(xhash *array)
715{
Denis Vlasenkof782f522007-01-01 23:51:30 +0000716 unsigned i;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000717 hash_item *hi, *thi;
718
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000719 for (i = 0; i < array->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000720 hi = array->items[i];
721 while (hi) {
722 thi = hi;
723 hi = hi->next;
Aaron Lehmanna170e1c2002-11-28 11:27:31 +0000724 free(thi->data.v.string);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000725 free(thi);
726 }
727 array->items[i] = NULL;
728 }
729 array->glen = array->nel = 0;
730}
731
732/* clear a variable */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000733static var *clrvar(var *v)
734{
Aaron Lehmanna170e1c2002-11-28 11:27:31 +0000735 if (!(v->type & VF_FSTR))
Glenn L McGrath545106f2002-11-11 06:21:00 +0000736 free(v->string);
737
738 v->type &= VF_DONTTOUCH;
739 v->type |= VF_DIRTY;
740 v->string = NULL;
741 return v;
742}
743
744/* assign string value to variable */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000745static var *setvar_p(var *v, char *value)
746{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000747 clrvar(v);
748 v->string = value;
749 handle_special(v);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000750 return v;
751}
752
753/* same as setvar_p but make a copy of string */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000754static var *setvar_s(var *v, const char *value)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000755{
Rob Landleyd921b2e2006-08-03 15:41:12 +0000756 return setvar_p(v, (value && *value) ? xstrdup(value) : NULL);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000757}
758
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100759/* same as setvar_s but sets USER flag */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000760static var *setvar_u(var *v, const char *value)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000761{
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100762 v = setvar_s(v, value);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000763 v->type |= VF_USER;
764 return v;
765}
766
767/* set array element to user string */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000768static void setari_u(var *a, int idx, const char *s)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000769{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000770 var *v;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000771
Denys Vlasenko7bb346f2009-10-06 22:09:50 +0200772 v = findvar(iamarray(a), itoa(idx));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000773 setvar_u(v, s);
774}
775
776/* assign numeric value to variable */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000777static var *setvar_i(var *v, double value)
778{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000779 clrvar(v);
780 v->type |= VF_NUMBER;
781 v->number = value;
782 handle_special(v);
783 return v;
784}
785
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +0000786static const char *getvar_s(var *v)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000787{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000788 /* if v is numeric and has no cached string, convert it to string */
789 if ((v->type & (VF_NUMBER | VF_CACHED)) == VF_NUMBER) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000790 fmt_num(g_buf, MAXVARFMT, getvar_s(intvar[CONVFMT]), v->number, TRUE);
791 v->string = xstrdup(g_buf);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000792 v->type |= VF_CACHED;
793 }
794 return (v->string == NULL) ? "" : v->string;
795}
796
Mike Frysinger10a11e22005-09-27 02:23:02 +0000797static double getvar_i(var *v)
798{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000799 char *s;
800
801 if ((v->type & (VF_NUMBER | VF_CACHED)) == 0) {
802 v->number = 0;
803 s = v->string;
804 if (s && *s) {
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000805 v->number = my_strtod(&s);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000806 if (v->type & VF_USER) {
807 skip_spaces(&s);
808 if (*s != '\0')
809 v->type &= ~VF_USER;
810 }
811 } else {
812 v->type &= ~VF_USER;
813 }
814 v->type |= VF_CACHED;
815 }
816 return v->number;
817}
818
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000819/* Used for operands of bitwise ops */
820static unsigned long getvar_i_int(var *v)
821{
822 double d = getvar_i(v);
823
824 /* Casting doubles to longs is undefined for values outside
825 * of target type range. Try to widen it as much as possible */
826 if (d >= 0)
827 return (unsigned long)d;
Denis Vlasenko665eaff2008-09-05 04:59:02 +0000828 /* Why? Think about d == -4294967295.0 (assuming 32bit longs) */
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000829 return - (long) (unsigned long) (-d);
830}
831
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000832static var *copyvar(var *dest, const var *src)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000833{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000834 if (dest != src) {
835 clrvar(dest);
Denis Vlasenko629563b2007-02-24 17:05:52 +0000836 dest->type |= (src->type & ~(VF_DONTTOUCH | VF_FSTR));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000837 dest->number = src->number;
838 if (src->string)
Rob Landleyd921b2e2006-08-03 15:41:12 +0000839 dest->string = xstrdup(src->string);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000840 }
841 handle_special(dest);
842 return dest;
843}
844
Mike Frysinger10a11e22005-09-27 02:23:02 +0000845static var *incvar(var *v)
846{
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100847 return setvar_i(v, getvar_i(v) + 1.0);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000848}
849
850/* return true if v is number or numeric string */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000851static int is_numeric(var *v)
852{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000853 getvar_i(v);
854 return ((v->type ^ VF_DIRTY) & (VF_NUMBER | VF_USER | VF_DIRTY));
855}
856
857/* return 1 when value of v corresponds to true, 0 otherwise */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000858static int istrue(var *v)
859{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000860 if (is_numeric(v))
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100861 return (v->number != 0);
862 return (v->string && v->string[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000863}
864
Eric Andersenaff114c2004-04-14 17:51:38 +0000865/* temporary variables allocator. Last allocated should be first freed */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000866static var *nvalloc(int n)
867{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000868 nvblock *pb = NULL;
869 var *v, *r;
870 int size;
871
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000872 while (g_cb) {
873 pb = g_cb;
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100874 if ((g_cb->pos - g_cb->nv) + n <= g_cb->size)
875 break;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000876 g_cb = g_cb->next;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000877 }
878
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000879 if (!g_cb) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000880 size = (n <= MINNVBLOCK) ? MINNVBLOCK : n;
Denis Vlasenkoe0a7fc52008-07-02 11:14:59 +0000881 g_cb = xzalloc(sizeof(nvblock) + size * sizeof(var));
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000882 g_cb->size = size;
883 g_cb->pos = g_cb->nv;
884 g_cb->prev = pb;
Denis Vlasenkoe0a7fc52008-07-02 11:14:59 +0000885 /*g_cb->next = NULL; - xzalloc did it */
Denys Vlasenkocdeda162009-11-30 01:14:16 +0100886 if (pb)
887 pb->next = g_cb;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000888 }
889
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000890 v = r = g_cb->pos;
891 g_cb->pos += n;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000892
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000893 while (v < g_cb->pos) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000894 v->type = 0;
895 v->string = NULL;
896 v++;
897 }
898
899 return r;
900}
901
Mike Frysinger10a11e22005-09-27 02:23:02 +0000902static void nvfree(var *v)
903{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000904 var *p;
905
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000906 if (v < g_cb->nv || v >= g_cb->pos)
907 syntax_error(EMSG_INTERNAL_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000908
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000909 for (p = v; p < g_cb->pos; p++) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000910 if ((p->type & (VF_ARRAY | VF_CHILD)) == VF_ARRAY) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000911 clear_array(iamarray(p));
912 free(p->x.array->items);
913 free(p->x.array);
914 }
Denys Vlasenko3cb60c32010-03-10 19:20:32 +0100915 if (p->type & VF_WALK) {
916 //bb_error_msg("free(walker@%p:%p) #1", &p->x.walker, p->x.walker);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000917 free(p->x.walker);
Denys Vlasenko3cb60c32010-03-10 19:20:32 +0100918 }
Glenn L McGrath545106f2002-11-11 06:21:00 +0000919
920 clrvar(p);
921 }
922
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000923 g_cb->pos = v;
924 while (g_cb->prev && g_cb->pos == g_cb->nv) {
925 g_cb = g_cb->prev;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000926 }
927}
928
929/* ------- awk program text parsing ------- */
930
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000931/* Parse next token pointed by global pos, place results into global ttt.
Glenn L McGrath545106f2002-11-11 06:21:00 +0000932 * If token isn't expected, give away. Return token class
933 */
Mike Frysingerf87b3e32005-09-27 04:16:22 +0000934static uint32_t next_token(uint32_t expected)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000935{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000936#define concat_inserted (G.next_token__concat_inserted)
937#define save_tclass (G.next_token__save_tclass)
938#define save_info (G.next_token__save_info)
939/* Initialized to TC_OPTERM: */
940#define ltclass (G.next_token__ltclass)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000941
Denis Vlasenkof782f522007-01-01 23:51:30 +0000942 char *p, *pp, *s;
943 const char *tl;
944 uint32_t tc;
945 const uint32_t *ti;
946 int l;
947
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000948 if (t_rollback) {
949 t_rollback = FALSE;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000950
951 } else if (concat_inserted) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000952 concat_inserted = FALSE;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000953 t_tclass = save_tclass;
954 t_info = save_info;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000955
956 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000957 p = g_pos;
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +0000958 readnext:
Glenn L McGrath545106f2002-11-11 06:21:00 +0000959 skip_spaces(&p);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000960 g_lineno = t_lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000961 if (*p == '#')
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000962 while (*p != '\n' && *p != '\0')
963 p++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000964
965 if (*p == '\n')
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000966 t_lineno++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000967
968 if (*p == '\0') {
969 tc = TC_EOF;
970
971 } else if (*p == '\"') {
972 /* it's a string */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000973 t_string = s = ++p;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000974 while (*p != '\"') {
975 if (*p == '\0' || *p == '\n')
976 syntax_error(EMSG_UNEXP_EOS);
977 *(s++) = nextchar(&p);
978 }
979 p++;
980 *s = '\0';
981 tc = TC_STRING;
982
983 } else if ((expected & TC_REGEXP) && *p == '/') {
984 /* it's regexp */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000985 t_string = s = ++p;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000986 while (*p != '/') {
987 if (*p == '\0' || *p == '\n')
988 syntax_error(EMSG_UNEXP_EOS);
Denis Vlasenkod9b5ab82007-05-18 07:30:43 +0000989 *s = *p++;
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000990 if (*s++ == '\\') {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000991 pp = p;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000992 *(s-1) = bb_process_escape_sequence((const char **)&p);
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000993 if (*pp == '\\')
994 *s++ = '\\';
995 if (p == pp)
996 *s++ = *p++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000997 }
998 }
999 p++;
1000 *s = '\0';
1001 tc = TC_REGEXP;
1002
1003 } else if (*p == '.' || isdigit(*p)) {
1004 /* it's a number */
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00001005 t_double = my_strtod(&p);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001006 if (*p == '.')
1007 syntax_error(EMSG_UNEXP_TOKEN);
1008 tc = TC_NUMBER;
1009
1010 } else {
1011 /* search for something known */
1012 tl = tokenlist;
1013 tc = 0x00000001;
1014 ti = tokeninfo;
1015 while (*tl) {
1016 l = *(tl++);
1017 if (l == NTCC) {
1018 tc <<= 1;
1019 continue;
1020 }
1021 /* if token class is expected, token
1022 * matches and it's not a longer word,
1023 * then this is what we are looking for
1024 */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001025 if ((tc & (expected | TC_WORD | TC_NEWLINE))
1026 && *tl == *p && strncmp(p, tl, l) == 0
1027 && !((tc & TC_WORD) && isalnum_(p[l]))
1028 ) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001029 t_info = *ti;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001030 p += l;
1031 break;
1032 }
1033 ti++;
1034 tl += l;
1035 }
1036
Denis Vlasenkof782f522007-01-01 23:51:30 +00001037 if (!*tl) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001038 /* it's a name (var/array/function),
1039 * otherwise it's something wrong
1040 */
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001041 if (!isalnum_(*p))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001042 syntax_error(EMSG_UNEXP_TOKEN);
1043
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001044 t_string = --p;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001045 while (isalnum_(*(++p))) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001046 *(p-1) = *p;
1047 }
1048 *(p-1) = '\0';
1049 tc = TC_VARIABLE;
Bernhard Reutner-Fischerbb204622005-10-17 14:21:06 +00001050 /* also consume whitespace between functionname and bracket */
Alexander Shishkind03cd3b2010-02-25 17:55:40 +02001051 if (!(expected & TC_VARIABLE) || (expected & TC_ARRAY))
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001052 skip_spaces(&p);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001053 if (*p == '(') {
1054 tc = TC_FUNCTION;
1055 } else {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001056 if (*p == '[') {
1057 p++;
1058 tc = TC_ARRAY;
1059 }
1060 }
1061 }
1062 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001063 g_pos = p;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001064
1065 /* skipping newlines in some cases */
1066 if ((ltclass & TC_NOTERM) && (tc & TC_NEWLINE))
1067 goto readnext;
1068
1069 /* insert concatenation operator when needed */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001070 if ((ltclass & TC_CONCAT1) && (tc & TC_CONCAT2) && (expected & TC_BINOP)) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001071 concat_inserted = TRUE;
1072 save_tclass = tc;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001073 save_info = t_info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001074 tc = TC_BINOP;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001075 t_info = OC_CONCAT | SS | P(35);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001076 }
1077
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001078 t_tclass = tc;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001079 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001080 ltclass = t_tclass;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001081
1082 /* Are we ready for this? */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001083 if (!(ltclass & expected))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001084 syntax_error((ltclass & (TC_NEWLINE | TC_EOF)) ?
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001085 EMSG_UNEXP_EOS : EMSG_UNEXP_TOKEN);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001086
1087 return ltclass;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001088#undef concat_inserted
1089#undef save_tclass
1090#undef save_info
1091#undef ltclass
Glenn L McGrath545106f2002-11-11 06:21:00 +00001092}
1093
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001094static void rollback_token(void)
1095{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001096 t_rollback = TRUE;
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001097}
Glenn L McGrath545106f2002-11-11 06:21:00 +00001098
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001099static node *new_node(uint32_t info)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001100{
"Robert P. J. Day"68229832006-07-01 13:08:46 +00001101 node *n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001102
Denis Vlasenko4cccc032006-12-22 18:37:07 +00001103 n = xzalloc(sizeof(node));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001104 n->info = info;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001105 n->lineno = g_lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001106 return n;
1107}
1108
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001109static node *mk_re_node(const char *s, node *n, regex_t *re)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001110{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001111 n->info = OC_REGEXP;
1112 n->l.re = re;
1113 n->r.ire = re + 1;
1114 xregcomp(re, s, REG_EXTENDED);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001115 xregcomp(re + 1, s, REG_EXTENDED | REG_ICASE);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001116
1117 return n;
1118}
1119
Mike Frysinger10a11e22005-09-27 02:23:02 +00001120static node *condition(void)
1121{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001122 next_token(TC_SEQSTART);
1123 return parse_expr(TC_SEQTERM);
1124}
1125
1126/* parse expression terminated by given argument, return ptr
1127 * to built subtree. Terminator is eaten by parse_expr */
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001128static node *parse_expr(uint32_t iexp)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001129{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001130 node sn;
1131 node *cn = &sn;
1132 node *vn, *glptr;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001133 uint32_t tc, xtc;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001134 var *v;
1135
1136 sn.info = PRIMASK;
1137 sn.r.n = glptr = NULL;
1138 xtc = TC_OPERAND | TC_UOPPRE | TC_REGEXP | iexp;
1139
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001140 while (!((tc = next_token(xtc)) & iexp)) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001141 if (glptr && (t_info == (OC_COMPARE | VV | P(39) | 2))) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001142 /* input redirection (<) attached to glptr node */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001143 cn = glptr->l.n = new_node(OC_CONCAT | SS | P(37));
Glenn L McGrath4bded582004-02-22 11:55:09 +00001144 cn->a.n = glptr;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001145 xtc = TC_OPERAND | TC_UOPPRE;
1146 glptr = NULL;
1147
1148 } else if (tc & (TC_BINOP | TC_UOPPOST)) {
1149 /* for binary and postfix-unary operators, jump back over
1150 * previous operators with higher priority */
1151 vn = cn;
Denys Vlasenkocdeda162009-11-30 01:14:16 +01001152 while (((t_info & PRIMASK) > (vn->a.n->info & PRIMASK2))
1153 || ((t_info == vn->info) && ((t_info & OPCLSMASK) == OC_COLON))
1154 ) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001155 vn = vn->a.n;
Denys Vlasenkocdeda162009-11-30 01:14:16 +01001156 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001157 if ((t_info & OPCLSMASK) == OC_TERNARY)
1158 t_info += P(6);
1159 cn = vn->a.n->r.n = new_node(t_info);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001160 cn->a.n = vn->a.n;
1161 if (tc & TC_BINOP) {
1162 cn->l.n = vn;
1163 xtc = TC_OPERAND | TC_UOPPRE | TC_REGEXP;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001164 if ((t_info & OPCLSMASK) == OC_PGETLINE) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001165 /* it's a pipe */
1166 next_token(TC_GETLINE);
1167 /* give maximum priority to this pipe */
1168 cn->info &= ~PRIMASK;
1169 xtc = TC_OPERAND | TC_UOPPRE | TC_BINOP | iexp;
1170 }
1171 } else {
1172 cn->r.n = vn;
1173 xtc = TC_OPERAND | TC_UOPPRE | TC_BINOP | iexp;
1174 }
1175 vn->a.n = cn;
1176
1177 } else {
1178 /* for operands and prefix-unary operators, attach them
1179 * to last node */
1180 vn = cn;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001181 cn = vn->r.n = new_node(t_info);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001182 cn->a.n = vn;
1183 xtc = TC_OPERAND | TC_UOPPRE | TC_REGEXP;
1184 if (tc & (TC_OPERAND | TC_REGEXP)) {
Rob Landleyed830e82005-06-07 02:43:52 +00001185 xtc = TC_UOPPRE | TC_UOPPOST | TC_BINOP | TC_OPERAND | iexp;
Eric Andersenc7bda1c2004-03-15 08:29:22 +00001186 /* one should be very careful with switch on tclass -
Glenn L McGrath545106f2002-11-11 06:21:00 +00001187 * only simple tclasses should be used! */
1188 switch (tc) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001189 case TC_VARIABLE:
1190 case TC_ARRAY:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001191 cn->info = OC_VAR;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001192 v = hash_search(ahash, t_string);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001193 if (v != NULL) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001194 cn->info = OC_FNARG;
1195 cn->l.i = v->x.aidx;
1196 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001197 cn->l.v = newvar(t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001198 }
1199 if (tc & TC_ARRAY) {
1200 cn->info |= xS;
1201 cn->r.n = parse_expr(TC_ARRTERM);
1202 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001203 break;
Mike Frysingerde2b9382005-09-27 03:18:00 +00001204
Denis Vlasenkof782f522007-01-01 23:51:30 +00001205 case TC_NUMBER:
1206 case TC_STRING:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001207 cn->info = OC_VAR;
Rob Landley9ffd4232006-05-21 18:30:35 +00001208 v = cn->l.v = xzalloc(sizeof(var));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001209 if (tc & TC_NUMBER)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001210 setvar_i(v, t_double);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001211 else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001212 setvar_s(v, t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001213 break;
1214
Denis Vlasenkof782f522007-01-01 23:51:30 +00001215 case TC_REGEXP:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001216 mk_re_node(t_string, cn, xzalloc(sizeof(regex_t)*2));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001217 break;
1218
Denis Vlasenkof782f522007-01-01 23:51:30 +00001219 case TC_FUNCTION:
Mike Frysingerde2b9382005-09-27 03:18:00 +00001220 cn->info = OC_FUNC;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001221 cn->r.f = newfunc(t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001222 cn->l.n = condition();
1223 break;
1224
Denis Vlasenkof782f522007-01-01 23:51:30 +00001225 case TC_SEQSTART:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001226 cn = vn->r.n = parse_expr(TC_SEQTERM);
1227 cn->a.n = vn;
1228 break;
1229
Denis Vlasenkof782f522007-01-01 23:51:30 +00001230 case TC_GETLINE:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001231 glptr = cn;
1232 xtc = TC_OPERAND | TC_UOPPRE | TC_BINOP | iexp;
1233 break;
1234
Denis Vlasenkof782f522007-01-01 23:51:30 +00001235 case TC_BUILTIN:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001236 cn->l.n = condition();
1237 break;
1238 }
1239 }
1240 }
1241 }
1242 return sn.r.n;
1243}
1244
1245/* add node to chain. Return ptr to alloc'd node */
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001246static node *chain_node(uint32_t info)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001247{
"Robert P. J. Day"68229832006-07-01 13:08:46 +00001248 node *n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001249
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001250 if (!seq->first)
Glenn L McGrath545106f2002-11-11 06:21:00 +00001251 seq->first = seq->last = new_node(0);
1252
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001253 if (seq->programname != g_progname) {
1254 seq->programname = g_progname;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001255 n = chain_node(OC_NEWSOURCE);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001256 n->l.s = xstrdup(g_progname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001257 }
1258
1259 n = seq->last;
1260 n->info = info;
1261 seq->last = n->a.n = new_node(OC_DONE);
1262
1263 return n;
1264}
1265
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001266static void chain_expr(uint32_t info)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001267{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001268 node *n;
1269
1270 n = chain_node(info);
1271 n->l.n = parse_expr(TC_OPTERM | TC_GRPTERM);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001272 if (t_tclass & TC_GRPTERM)
Glenn L McGrath545106f2002-11-11 06:21:00 +00001273 rollback_token();
1274}
1275
Mike Frysinger10a11e22005-09-27 02:23:02 +00001276static node *chain_loop(node *nn)
1277{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001278 node *n, *n2, *save_brk, *save_cont;
1279
1280 save_brk = break_ptr;
1281 save_cont = continue_ptr;
1282
1283 n = chain_node(OC_BR | Vx);
1284 continue_ptr = new_node(OC_EXEC);
1285 break_ptr = new_node(OC_EXEC);
1286 chain_group();
1287 n2 = chain_node(OC_EXEC | Vx);
1288 n2->l.n = nn;
1289 n2->a.n = n;
1290 continue_ptr->a.n = n2;
1291 break_ptr->a.n = n->r.n = seq->last;
1292
1293 continue_ptr = save_cont;
1294 break_ptr = save_brk;
1295
1296 return n;
1297}
1298
1299/* parse group and attach it to chain */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001300static void chain_group(void)
1301{
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001302 uint32_t c;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001303 node *n, *n2, *n3;
1304
1305 do {
1306 c = next_token(TC_GRPSEQ);
1307 } while (c & TC_NEWLINE);
1308
1309 if (c & TC_GRPSTART) {
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001310 while (next_token(TC_GRPSEQ | TC_GRPTERM) != TC_GRPTERM) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001311 if (t_tclass & TC_NEWLINE) continue;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001312 rollback_token();
1313 chain_group();
1314 }
1315 } else if (c & (TC_OPSEQ | TC_OPTERM)) {
1316 rollback_token();
1317 chain_expr(OC_EXEC | Vx);
1318 } else { /* TC_STATEMNT */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001319 switch (t_info & OPCLSMASK) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001320 case ST_IF:
1321 n = chain_node(OC_BR | Vx);
1322 n->l.n = condition();
1323 chain_group();
1324 n2 = chain_node(OC_EXEC);
1325 n->r.n = seq->last;
1326 if (next_token(TC_GRPSEQ | TC_GRPTERM | TC_ELSE) == TC_ELSE) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001327 chain_group();
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001328 n2->a.n = seq->last;
1329 } else {
1330 rollback_token();
1331 }
1332 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001333
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001334 case ST_WHILE:
1335 n2 = condition();
1336 n = chain_loop(NULL);
1337 n->l.n = n2;
1338 break;
1339
1340 case ST_DO:
1341 n2 = chain_node(OC_EXEC);
1342 n = chain_loop(NULL);
1343 n2->a.n = n->a.n;
1344 next_token(TC_WHILE);
1345 n->l.n = condition();
1346 break;
1347
1348 case ST_FOR:
1349 next_token(TC_SEQSTART);
1350 n2 = parse_expr(TC_SEMICOL | TC_SEQTERM);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001351 if (t_tclass & TC_SEQTERM) { /* for-in */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001352 if ((n2->info & OPCLSMASK) != OC_IN)
1353 syntax_error(EMSG_UNEXP_TOKEN);
1354 n = chain_node(OC_WALKINIT | VV);
1355 n->l.n = n2->l.n;
1356 n->r.n = n2->r.n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001357 n = chain_loop(NULL);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001358 n->info = OC_WALKNEXT | Vx;
1359 n->l.n = n2->l.n;
1360 } else { /* for (;;) */
1361 n = chain_node(OC_EXEC | Vx);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001362 n->l.n = n2;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001363 n2 = parse_expr(TC_SEMICOL);
1364 n3 = parse_expr(TC_SEQTERM);
1365 n = chain_loop(n3);
1366 n->l.n = n2;
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001367 if (!n2)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001368 n->info = OC_EXEC;
1369 }
1370 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001371
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001372 case OC_PRINT:
1373 case OC_PRINTF:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001374 n = chain_node(t_info);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001375 n->l.n = parse_expr(TC_OPTERM | TC_OUTRDR | TC_GRPTERM);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001376 if (t_tclass & TC_OUTRDR) {
1377 n->info |= t_info;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001378 n->r.n = parse_expr(TC_OPTERM | TC_GRPTERM);
1379 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001380 if (t_tclass & TC_GRPTERM)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001381 rollback_token();
1382 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001383
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001384 case OC_BREAK:
1385 n = chain_node(OC_EXEC);
1386 n->a.n = break_ptr;
1387 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001388
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001389 case OC_CONTINUE:
1390 n = chain_node(OC_EXEC);
1391 n->a.n = continue_ptr;
1392 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001393
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001394 /* delete, next, nextfile, return, exit */
1395 default:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001396 chain_expr(t_info);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001397 }
1398 }
1399}
1400
Mike Frysinger10a11e22005-09-27 02:23:02 +00001401static void parse_program(char *p)
1402{
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001403 uint32_t tclass;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001404 node *cn;
1405 func *f;
1406 var *v;
1407
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001408 g_pos = p;
1409 t_lineno = 1;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001410 while ((tclass = next_token(TC_EOF | TC_OPSEQ | TC_GRPSTART |
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001411 TC_OPTERM | TC_BEGIN | TC_END | TC_FUNCDECL)) != TC_EOF) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001412
1413 if (tclass & TC_OPTERM)
1414 continue;
1415
1416 seq = &mainseq;
1417 if (tclass & TC_BEGIN) {
1418 seq = &beginseq;
1419 chain_group();
1420
1421 } else if (tclass & TC_END) {
1422 seq = &endseq;
1423 chain_group();
1424
1425 } else if (tclass & TC_FUNCDECL) {
1426 next_token(TC_FUNCTION);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001427 g_pos++;
1428 f = newfunc(t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001429 f->body.first = NULL;
1430 f->nargs = 0;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001431 while (next_token(TC_VARIABLE | TC_SEQTERM) & TC_VARIABLE) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001432 v = findvar(ahash, t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001433 v->x.aidx = (f->nargs)++;
1434
1435 if (next_token(TC_COMMA | TC_SEQTERM) & TC_SEQTERM)
1436 break;
1437 }
1438 seq = &(f->body);
1439 chain_group();
1440 clear_array(ahash);
1441
1442 } else if (tclass & TC_OPSEQ) {
1443 rollback_token();
1444 cn = chain_node(OC_TEST);
1445 cn->l.n = parse_expr(TC_OPTERM | TC_EOF | TC_GRPSTART);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001446 if (t_tclass & TC_GRPSTART) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001447 rollback_token();
1448 chain_group();
1449 } else {
1450 chain_node(OC_PRINT);
1451 }
1452 cn->r.n = mainseq.last;
1453
1454 } else /* if (tclass & TC_GRPSTART) */ {
1455 rollback_token();
1456 chain_group();
1457 }
1458 }
1459}
1460
1461
1462/* -------- program execution part -------- */
1463
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001464static node *mk_splitter(const char *s, tsplitter *spl)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001465{
"Robert P. J. Day"68229832006-07-01 13:08:46 +00001466 regex_t *re, *ire;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001467 node *n;
1468
1469 re = &spl->re[0];
1470 ire = &spl->re[1];
1471 n = &spl->n;
Denis Vlasenko890ac9d2006-10-07 15:16:19 +00001472 if ((n->info & OPCLSMASK) == OC_REGEXP) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001473 regfree(re);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001474 regfree(ire); // TODO: nuke ire, use re+1?
Glenn L McGrath545106f2002-11-11 06:21:00 +00001475 }
Rob Landleya3896512006-05-07 20:20:34 +00001476 if (strlen(s) > 1) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001477 mk_re_node(s, n, re);
1478 } else {
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001479 n->info = (uint32_t) *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001480 }
1481
1482 return n;
1483}
1484
1485/* use node as a regular expression. Supplied with node ptr and regex_t
Eric Andersenaff114c2004-04-14 17:51:38 +00001486 * storage space. Return ptr to regex (if result points to preg, it should
Glenn L McGrath545106f2002-11-11 06:21:00 +00001487 * be later regfree'd manually
1488 */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001489static regex_t *as_regex(node *op, regex_t *preg)
1490{
Denis Vlasenko7a676642009-03-15 22:20:31 +00001491 int cflags;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001492 var *v;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001493 const char *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001494
1495 if ((op->info & OPCLSMASK) == OC_REGEXP) {
1496 return icase ? op->r.ire : op->l.re;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001497 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001498 v = nvalloc(1);
1499 s = getvar_s(evaluate(op, v));
Denis Vlasenko7a676642009-03-15 22:20:31 +00001500
1501 cflags = icase ? REG_EXTENDED | REG_ICASE : REG_EXTENDED;
1502 /* Testcase where REG_EXTENDED fails (unpaired '{'):
1503 * echo Hi | awk 'gsub("@(samp|code|file)\{","");'
1504 * gawk 3.1.5 eats this. We revert to ~REG_EXTENDED
1505 * (maybe gsub is not supposed to use REG_EXTENDED?).
1506 */
1507 if (regcomp(preg, s, cflags)) {
1508 cflags &= ~REG_EXTENDED;
1509 xregcomp(preg, s, cflags);
1510 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001511 nvfree(v);
1512 return preg;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001513}
1514
1515/* gradually increasing buffer */
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001516static char* qrealloc(char *b, int n, int *size)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001517{
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001518 if (!b || n >= *size) {
Denis Vlasenkodeeed592008-07-08 05:14:36 +00001519 *size = n + (n>>1) + 80;
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001520 b = xrealloc(b, *size);
Denis Vlasenkodeeed592008-07-08 05:14:36 +00001521 }
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001522 return b;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001523}
1524
1525/* resize field storage space */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001526static void fsrealloc(int size)
1527{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001528 int i;
1529
1530 if (size >= maxfields) {
1531 i = maxfields;
1532 maxfields = size + 16;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001533 Fields = xrealloc(Fields, maxfields * sizeof(var));
1534 for (; i < maxfields; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001535 Fields[i].type = VF_SPECIAL;
1536 Fields[i].string = NULL;
1537 }
1538 }
1539
1540 if (size < nfields) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001541 for (i = size; i < nfields; i++) {
1542 clrvar(Fields + i);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001543 }
1544 }
1545 nfields = size;
1546}
1547
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001548static int awk_split(const char *s, node *spl, char **slist)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001549{
Denis Vlasenkof782f522007-01-01 23:51:30 +00001550 int l, n = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001551 char c[4];
1552 char *s1;
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001553 regmatch_t pmatch[2]; // TODO: why [2]? [1] is enough...
Glenn L McGrath545106f2002-11-11 06:21:00 +00001554
1555 /* in worst case, each char would be a separate field */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001556 *slist = s1 = xzalloc(strlen(s) * 2 + 3);
1557 strcpy(s1, s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001558
1559 c[0] = c[1] = (char)spl->info;
1560 c[2] = c[3] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001561 if (*getvar_s(intvar[RS]) == '\0')
1562 c[2] = '\n';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001563
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001564 if ((spl->info & OPCLSMASK) == OC_REGEXP) { /* regex split */
1565 if (!*s)
1566 return n; /* "": zero fields */
1567 n++; /* at least one field will be there */
1568 do {
1569 l = strcspn(s, c+2); /* len till next NUL or \n */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001570 if (regexec(icase ? spl->r.ire : spl->l.re, s, 1, pmatch, 0) == 0
1571 && pmatch[0].rm_so <= l
1572 ) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001573 l = pmatch[0].rm_so;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001574 if (pmatch[0].rm_eo == 0) {
1575 l++;
1576 pmatch[0].rm_eo++;
1577 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001578 n++; /* we saw yet another delimiter */
Glenn L McGrath545106f2002-11-11 06:21:00 +00001579 } else {
1580 pmatch[0].rm_eo = l;
Denys Vlasenkoe4244232009-05-18 23:50:03 +02001581 if (s[l])
1582 pmatch[0].rm_eo++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001583 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001584 memcpy(s1, s, l);
Denis Vlasenko67b5eeb2009-04-12 13:54:13 +00001585 /* make sure we remove *all* of the separator chars */
Denys Vlasenkoe4244232009-05-18 23:50:03 +02001586 do {
1587 s1[l] = '\0';
1588 } while (++l < pmatch[0].rm_eo);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001589 nextword(&s1);
1590 s += pmatch[0].rm_eo;
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001591 } while (*s);
1592 return n;
1593 }
1594 if (c[0] == '\0') { /* null split */
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001595 while (*s) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001596 *s1++ = *s++;
1597 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001598 n++;
1599 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001600 return n;
1601 }
1602 if (c[0] != ' ') { /* single-character split */
Glenn L McGrath545106f2002-11-11 06:21:00 +00001603 if (icase) {
1604 c[0] = toupper(c[0]);
1605 c[1] = tolower(c[1]);
1606 }
1607 if (*s1) n++;
1608 while ((s1 = strpbrk(s1, c))) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001609 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001610 n++;
1611 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001612 return n;
1613 }
1614 /* space split */
1615 while (*s) {
1616 s = skip_whitespace(s);
1617 if (!*s) break;
1618 n++;
1619 while (*s && !isspace(*s))
1620 *s1++ = *s++;
1621 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001622 }
1623 return n;
1624}
1625
Mike Frysinger10a11e22005-09-27 02:23:02 +00001626static void split_f0(void)
1627{
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001628/* static char *fstrings; */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001629#define fstrings (G.split_f0__fstrings)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001630
Glenn L McGrath545106f2002-11-11 06:21:00 +00001631 int i, n;
1632 char *s;
1633
1634 if (is_f0_split)
1635 return;
1636
1637 is_f0_split = TRUE;
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00001638 free(fstrings);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001639 fsrealloc(0);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001640 n = awk_split(getvar_s(intvar[F0]), &fsplitter.n, &fstrings);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001641 fsrealloc(n);
1642 s = fstrings;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001643 for (i = 0; i < n; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001644 Fields[i].string = nextword(&s);
1645 Fields[i].type |= (VF_FSTR | VF_USER | VF_DIRTY);
1646 }
1647
1648 /* set NF manually to avoid side effects */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001649 clrvar(intvar[NF]);
1650 intvar[NF]->type = VF_NUMBER | VF_SPECIAL;
1651 intvar[NF]->number = nfields;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001652#undef fstrings
Glenn L McGrath545106f2002-11-11 06:21:00 +00001653}
1654
1655/* perform additional actions when some internal variables changed */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001656static void handle_special(var *v)
1657{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001658 int n;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001659 char *b;
1660 const char *sep, *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001661 int sl, l, len, i, bsize;
1662
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001663 if (!(v->type & VF_SPECIAL))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001664 return;
1665
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001666 if (v == intvar[NF]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001667 n = (int)getvar_i(v);
1668 fsrealloc(n);
1669
1670 /* recalculate $0 */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001671 sep = getvar_s(intvar[OFS]);
Rob Landleya3896512006-05-07 20:20:34 +00001672 sl = strlen(sep);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001673 b = NULL;
1674 len = 0;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001675 for (i = 0; i < n; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001676 s = getvar_s(&Fields[i]);
Rob Landleya3896512006-05-07 20:20:34 +00001677 l = strlen(s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001678 if (b) {
1679 memcpy(b+len, sep, sl);
1680 len += sl;
1681 }
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001682 b = qrealloc(b, len+l+sl, &bsize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001683 memcpy(b+len, s, l);
1684 len += l;
1685 }
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001686 if (b)
1687 b[len] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001688 setvar_p(intvar[F0], b);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001689 is_f0_split = TRUE;
1690
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001691 } else if (v == intvar[F0]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001692 is_f0_split = FALSE;
1693
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001694 } else if (v == intvar[FS]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001695 mk_splitter(getvar_s(v), &fsplitter);
1696
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001697 } else if (v == intvar[RS]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001698 mk_splitter(getvar_s(v), &rsplitter);
1699
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001700 } else if (v == intvar[IGNORECASE]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001701 icase = istrue(v);
1702
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001703 } else { /* $n */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001704 n = getvar_i(intvar[NF]);
1705 setvar_i(intvar[NF], n > v-Fields ? n : v-Fields+1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001706 /* right here v is invalid. Just to note... */
1707 }
1708}
1709
1710/* step through func/builtin/etc arguments */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001711static node *nextarg(node **pn)
1712{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001713 node *n;
1714
1715 n = *pn;
1716 if (n && (n->info & OPCLSMASK) == OC_COMMA) {
1717 *pn = n->r.n;
1718 n = n->l.n;
1719 } else {
1720 *pn = NULL;
1721 }
1722 return n;
1723}
1724
Mike Frysinger10a11e22005-09-27 02:23:02 +00001725static void hashwalk_init(var *v, xhash *array)
1726{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001727 char **w;
1728 hash_item *hi;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +00001729 unsigned i;
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001730 char **prev_walker = (v->type & VF_WALK) ? v->x.walker : NULL;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001731
1732 v->type |= VF_WALK;
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001733
1734 /* walker structure is: "[ptr2end][ptr2start][prev]<word1>NUL<word2>NUL" */
1735 w = v->x.walker = xzalloc(2 + 3*sizeof(char *) + array->glen);
1736 //bb_error_msg("walker@%p=%p", &v->x.walker, v->x.walker);
1737 w[0] = w[1] = (char *)(w + 3);
1738 w[2] = (char *)prev_walker;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001739 for (i = 0; i < array->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001740 hi = array->items[i];
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001741 while (hi) {
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001742 strcpy(w[0], hi->name);
1743 nextword(&w[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001744 hi = hi->next;
1745 }
1746 }
1747}
1748
Mike Frysinger10a11e22005-09-27 02:23:02 +00001749static int hashwalk_next(var *v)
1750{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001751 char **w;
1752
1753 w = v->x.walker;
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001754 if (w[1] == w[0]) {
1755 char **prev_walker = (char**)w[2];
Glenn L McGrath545106f2002-11-11 06:21:00 +00001756
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001757 //bb_error_msg("free(walker@%p:%p) #3, restoring to %p", &v->x.walker, v->x.walker, prev_walker);
1758 free(v->x.walker);
1759 v->x.walker = prev_walker;
1760 return FALSE;
1761 }
1762
1763 setvar_s(v, nextword(&w[1]));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001764 return TRUE;
1765}
1766
1767/* evaluate node, return 1 when result is true, 0 otherwise */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001768static int ptest(node *pattern)
1769{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001770 /* ptest__v is "static": to save stack space? */
1771 return istrue(evaluate(pattern, &G.ptest__v));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001772}
1773
1774/* read next record from stream rsm into a variable v */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001775static int awk_getline(rstream *rsm, var *v)
1776{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001777 char *b;
1778 regmatch_t pmatch[2];
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001779 int a, p, pp=0, size;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001780 int fd, so, eo, r, rp;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001781 char c, *m, *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001782
1783 /* we're using our own buffer since we need access to accumulating
1784 * characters
1785 */
1786 fd = fileno(rsm->F);
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001787 m = rsm->buffer;
1788 a = rsm->adv;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001789 p = rsm->pos;
1790 size = rsm->size;
1791 c = (char) rsplitter.n.info;
1792 rp = 0;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001793
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001794 if (!m)
1795 m = qrealloc(m, 256, &size);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001796 do {
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001797 b = m + a;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001798 so = eo = p;
1799 r = 1;
1800 if (p > 0) {
1801 if ((rsplitter.n.info & OPCLSMASK) == OC_REGEXP) {
1802 if (regexec(icase ? rsplitter.n.r.ire : rsplitter.n.l.re,
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001803 b, 1, pmatch, 0) == 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001804 so = pmatch[0].rm_so;
1805 eo = pmatch[0].rm_eo;
1806 if (b[eo] != '\0')
1807 break;
1808 }
1809 } else if (c != '\0') {
1810 s = strchr(b+pp, c);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001811 if (!s) s = memchr(b+pp, '\0', p - pp);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001812 if (s) {
1813 so = eo = s-b;
1814 eo++;
1815 break;
1816 }
1817 } else {
1818 while (b[rp] == '\n')
1819 rp++;
1820 s = strstr(b+rp, "\n\n");
1821 if (s) {
1822 so = eo = s-b;
1823 while (b[eo] == '\n') eo++;
1824 if (b[eo] != '\0')
1825 break;
1826 }
1827 }
1828 }
1829
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001830 if (a > 0) {
1831 memmove(m, (const void *)(m+a), p+1);
1832 b = m;
1833 a = 0;
1834 }
1835
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001836 m = qrealloc(m, a+p+128, &size);
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001837 b = m + a;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001838 pp = p;
1839 p += safe_read(fd, b+p, size-p-1);
1840 if (p < pp) {
1841 p = 0;
1842 r = 0;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001843 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001844 }
1845 b[p] = '\0';
1846
1847 } while (p > pp);
1848
1849 if (p == 0) {
1850 r--;
1851 } else {
1852 c = b[so]; b[so] = '\0';
1853 setvar_s(v, b+rp);
1854 v->type |= VF_USER;
1855 b[so] = c;
1856 c = b[eo]; b[eo] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001857 setvar_s(intvar[RT], b+so);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001858 b[eo] = c;
1859 }
1860
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001861 rsm->buffer = m;
1862 rsm->adv = a + eo;
1863 rsm->pos = p - eo;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001864 rsm->size = size;
1865
1866 return r;
1867}
1868
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00001869static int fmt_num(char *b, int size, const char *format, double n, int int_as_int)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001870{
Denis Vlasenkof782f522007-01-01 23:51:30 +00001871 int r = 0;
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00001872 char c;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001873 const char *s = format;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001874
1875 if (int_as_int && n == (int)n) {
1876 r = snprintf(b, size, "%d", (int)n);
1877 } else {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001878 do { c = *s; } while (c && *++s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001879 if (strchr("diouxX", c)) {
1880 r = snprintf(b, size, format, (int)n);
1881 } else if (strchr("eEfgG", c)) {
1882 r = snprintf(b, size, format, n);
1883 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001884 syntax_error(EMSG_INV_FMT);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001885 }
1886 }
1887 return r;
1888}
1889
Glenn L McGrath545106f2002-11-11 06:21:00 +00001890/* formatted output into an allocated buffer, return ptr to buffer */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001891static char *awk_printf(node *n)
1892{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001893 char *b = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001894 char *fmt, *s, *f;
1895 const char *s1;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001896 int i, j, incr, bsize;
1897 char c, c1;
1898 var *v, *arg;
1899
1900 v = nvalloc(1);
Rob Landleyd921b2e2006-08-03 15:41:12 +00001901 fmt = f = xstrdup(getvar_s(evaluate(nextarg(&n), v)));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001902
1903 i = 0;
1904 while (*f) {
1905 s = f;
1906 while (*f && (*f != '%' || *(++f) == '%'))
1907 f++;
Denis Vlasenko389f9d52007-05-09 21:57:23 +00001908 while (*f && !isalpha(*f)) {
1909 if (*f == '*')
1910 syntax_error("%*x formats are not supported");
Glenn L McGrath545106f2002-11-11 06:21:00 +00001911 f++;
Denis Vlasenko389f9d52007-05-09 21:57:23 +00001912 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001913
1914 incr = (f - s) + MAXVARFMT;
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001915 b = qrealloc(b, incr + i, &bsize);
Denis Vlasenkof782f522007-01-01 23:51:30 +00001916 c = *f;
1917 if (c != '\0') f++;
1918 c1 = *f;
1919 *f = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001920 arg = evaluate(nextarg(&n), v);
1921
1922 j = i;
1923 if (c == 'c' || !c) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001924 i += sprintf(b+i, s, is_numeric(arg) ?
1925 (char)getvar_i(arg) : *getvar_s(arg));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001926 } else if (c == 's') {
Denis Vlasenko92758142006-10-03 19:56:34 +00001927 s1 = getvar_s(arg);
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001928 b = qrealloc(b, incr+i+strlen(s1), &bsize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001929 i += sprintf(b+i, s, s1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001930 } else {
1931 i += fmt_num(b+i, incr, s, getvar_i(arg), FALSE);
1932 }
1933 *f = c1;
1934
1935 /* if there was an error while sprintf, return value is negative */
1936 if (i < j) i = j;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001937 }
1938
Denis Vlasenkof782f522007-01-01 23:51:30 +00001939 b = xrealloc(b, i + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001940 free(fmt);
1941 nvfree(v);
1942 b[i] = '\0';
1943 return b;
1944}
1945
1946/* common substitution routine
1947 * replace (nm) substring of (src) that match (n) with (repl), store
1948 * result into (dest), return number of substitutions. If nm=0, replace
1949 * all matches. If src or dst is NULL, use $0. If ex=TRUE, enable
1950 * subexpression matching (\1-\9)
1951 */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001952static int awk_sub(node *rn, const char *repl, int nm, var *src, var *dest, int ex)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001953{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001954 char *ds = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001955 const char *s;
1956 const char *sp;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001957 int c, i, j, di, rl, so, eo, nbs, n, dssize;
1958 regmatch_t pmatch[10];
1959 regex_t sreg, *re;
1960
1961 re = as_regex(rn, &sreg);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001962 if (!src) src = intvar[F0];
1963 if (!dest) dest = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00001964
1965 i = di = 0;
1966 sp = getvar_s(src);
Rob Landleya3896512006-05-07 20:20:34 +00001967 rl = strlen(repl);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001968 while (regexec(re, sp, 10, pmatch, sp==getvar_s(src) ? 0 : REG_NOTBOL) == 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001969 so = pmatch[0].rm_so;
1970 eo = pmatch[0].rm_eo;
1971
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001972 ds = qrealloc(ds, di + eo + rl, &dssize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001973 memcpy(ds + di, sp, eo);
1974 di += eo;
1975 if (++i >= nm) {
1976 /* replace */
1977 di -= (eo - so);
1978 nbs = 0;
1979 for (s = repl; *s; s++) {
1980 ds[di++] = c = *s;
1981 if (c == '\\') {
1982 nbs++;
1983 continue;
1984 }
1985 if (c == '&' || (ex && c >= '0' && c <= '9')) {
1986 di -= ((nbs + 3) >> 1);
1987 j = 0;
1988 if (c != '&') {
1989 j = c - '0';
1990 nbs++;
1991 }
1992 if (nbs % 2) {
1993 ds[di++] = c;
1994 } else {
1995 n = pmatch[j].rm_eo - pmatch[j].rm_so;
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01001996 ds = qrealloc(ds, di + rl + n, &dssize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001997 memcpy(ds + di, sp + pmatch[j].rm_so, n);
1998 di += n;
1999 }
2000 }
2001 nbs = 0;
2002 }
2003 }
2004
2005 sp += eo;
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002006 if (i == nm)
2007 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002008 if (eo == so) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002009 ds[di] = *sp++;
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002010 if (!ds[di++])
2011 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002012 }
2013 }
2014
Denys Vlasenkoc9955f22010-03-10 19:21:54 +01002015 ds = qrealloc(ds, di + strlen(sp), &dssize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002016 strcpy(ds + di, sp);
2017 setvar_p(dest, ds);
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002018 if (re == &sreg)
2019 regfree(re);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002020 return i;
2021}
2022
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002023static NOINLINE int do_mktime(const char *ds)
2024{
2025 struct tm then;
2026 int count;
2027
2028 /*memset(&then, 0, sizeof(then)); - not needed */
2029 then.tm_isdst = -1; /* default is unknown */
2030
2031 /* manpage of mktime says these fields are ints,
2032 * so we can sscanf stuff directly into them */
2033 count = sscanf(ds, "%u %u %u %u %u %u %d",
2034 &then.tm_year, &then.tm_mon, &then.tm_mday,
2035 &then.tm_hour, &then.tm_min, &then.tm_sec,
2036 &then.tm_isdst);
2037
2038 if (count < 6
2039 || (unsigned)then.tm_mon < 1
2040 || (unsigned)then.tm_year < 1900
2041 ) {
2042 return -1;
2043 }
2044
2045 then.tm_mon -= 1;
Denys Vlasenkobc3e9472009-09-21 04:16:00 +02002046 then.tm_year -= 1900;
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002047
2048 return mktime(&then);
2049}
2050
2051static NOINLINE var *exec_builtin(node *op, var *res)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002052{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002053#define tspl (G.exec_builtin__tspl)
2054
Glenn L McGrath545106f2002-11-11 06:21:00 +00002055 var *tv;
2056 node *an[4];
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002057 var *av[4];
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002058 const char *as[4];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002059 regmatch_t pmatch[2];
2060 regex_t sreg, *re;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002061 node *spl;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002062 uint32_t isr, info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002063 int nargs;
2064 time_t tt;
2065 char *s, *s1;
2066 int i, l, ll, n;
2067
2068 tv = nvalloc(4);
2069 isr = info = op->info;
2070 op = op->l.n;
2071
2072 av[2] = av[3] = NULL;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002073 for (i = 0; i < 4 && op; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002074 an[i] = nextarg(&op);
2075 if (isr & 0x09000000) av[i] = evaluate(an[i], &tv[i]);
2076 if (isr & 0x08000000) as[i] = getvar_s(av[i]);
2077 isr >>= 1;
2078 }
2079
2080 nargs = i;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +00002081 if ((uint32_t)nargs < (info >> 30))
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002082 syntax_error(EMSG_TOO_FEW_ARGS);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002083
Denys Vlasenko56b3eec2009-10-23 13:03:59 +02002084 info &= OPNMASK;
2085 switch (info) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002086
Denis Vlasenkof782f522007-01-01 23:51:30 +00002087 case B_a2:
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002088#if ENABLE_FEATURE_AWK_LIBM
Denis Vlasenko37890e22008-10-21 12:59:34 +00002089 setvar_i(res, atan2(getvar_i(av[0]), getvar_i(av[1])));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002090#else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002091 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002092#endif
2093 break;
2094
Denis Vlasenkof782f522007-01-01 23:51:30 +00002095 case B_sp:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002096 if (nargs > 2) {
2097 spl = (an[2]->info & OPCLSMASK) == OC_REGEXP ?
2098 an[2] : mk_splitter(getvar_s(evaluate(an[2], &tv[2])), &tspl);
2099 } else {
2100 spl = &fsplitter.n;
2101 }
2102
2103 n = awk_split(as[0], spl, &s);
2104 s1 = s;
2105 clear_array(iamarray(av[1]));
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002106 for (i = 1; i <= n; i++)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002107 setari_u(av[1], i, nextword(&s1));
2108 free(s);
2109 setvar_i(res, n);
2110 break;
2111
Denis Vlasenkof782f522007-01-01 23:51:30 +00002112 case B_ss:
Rob Landleya3896512006-05-07 20:20:34 +00002113 l = strlen(as[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002114 i = getvar_i(av[1]) - 1;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002115 if (i > l) i = l;
2116 if (i < 0) i = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002117 n = (nargs > 2) ? getvar_i(av[2]) : l-i;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002118 if (n < 0) n = 0;
Denis Vlasenko8ae5b282008-07-02 22:47:49 +00002119 s = xstrndup(as[0]+i, n);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002120 setvar_p(res, s);
2121 break;
Denis Vlasenkof7996f32007-01-11 17:20:00 +00002122
Denis Vlasenko7cbcd1c2008-08-28 23:16:58 +00002123 /* Bitwise ops must assume that operands are unsigned. GNU Awk 3.1.5:
2124 * awk '{ print or(-1,1) }' gives "4.29497e+09", not "-2.xxxe+09" */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002125 case B_an:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002126 setvar_i(res, getvar_i_int(av[0]) & getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002127 break;
Denis Vlasenkof7996f32007-01-11 17:20:00 +00002128
Denis Vlasenkof782f522007-01-01 23:51:30 +00002129 case B_co:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002130 setvar_i(res, ~getvar_i_int(av[0]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002131 break;
2132
Denis Vlasenkof782f522007-01-01 23:51:30 +00002133 case B_ls:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002134 setvar_i(res, getvar_i_int(av[0]) << getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002135 break;
2136
Denis Vlasenkof782f522007-01-01 23:51:30 +00002137 case B_or:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002138 setvar_i(res, getvar_i_int(av[0]) | getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002139 break;
2140
Denis Vlasenkof782f522007-01-01 23:51:30 +00002141 case B_rs:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002142 setvar_i(res, getvar_i_int(av[0]) >> getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002143 break;
2144
Denis Vlasenkof782f522007-01-01 23:51:30 +00002145 case B_xo:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002146 setvar_i(res, getvar_i_int(av[0]) ^ getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002147 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002148
Denis Vlasenkof782f522007-01-01 23:51:30 +00002149 case B_lo:
Denis Vlasenkof782f522007-01-01 23:51:30 +00002150 case B_up:
Rob Landleyd921b2e2006-08-03 15:41:12 +00002151 s1 = s = xstrdup(as[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002152 while (*s1) {
Denys Vlasenko56b3eec2009-10-23 13:03:59 +02002153 //*s1 = (info == B_up) ? toupper(*s1) : tolower(*s1);
2154 if ((unsigned char)((*s1 | 0x20) - 'a') <= ('z' - 'a'))
2155 *s1 = (info == B_up) ? (*s1 & 0xdf) : (*s1 | 0x20);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002156 s1++;
2157 }
2158 setvar_p(res, s);
2159 break;
2160
Denis Vlasenkof782f522007-01-01 23:51:30 +00002161 case B_ix:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002162 n = 0;
Rob Landleya3896512006-05-07 20:20:34 +00002163 ll = strlen(as[1]);
2164 l = strlen(as[0]) - ll;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002165 if (ll > 0 && l >= 0) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002166 if (!icase) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002167 s = strstr(as[0], as[1]);
2168 if (s) n = (s - as[0]) + 1;
2169 } else {
2170 /* this piece of code is terribly slow and
2171 * really should be rewritten
2172 */
2173 for (i=0; i<=l; i++) {
2174 if (strncasecmp(as[0]+i, as[1], ll) == 0) {
2175 n = i+1;
2176 break;
2177 }
2178 }
2179 }
2180 }
2181 setvar_i(res, n);
2182 break;
2183
Denis Vlasenkof782f522007-01-01 23:51:30 +00002184 case B_ti:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002185 if (nargs > 1)
2186 tt = getvar_i(av[1]);
2187 else
2188 time(&tt);
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002189 //s = (nargs > 0) ? as[0] : "%a %b %d %H:%M:%S %Z %Y";
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002190 i = strftime(g_buf, MAXVARFMT,
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002191 ((nargs > 0) ? as[0] : "%a %b %d %H:%M:%S %Z %Y"),
2192 localtime(&tt));
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002193 g_buf[i] = '\0';
2194 setvar_s(res, g_buf);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002195 break;
2196
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002197 case B_mt:
2198 setvar_i(res, do_mktime(as[0]));
2199 break;
2200
Denis Vlasenkof782f522007-01-01 23:51:30 +00002201 case B_ma:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002202 re = as_regex(an[1], &sreg);
2203 n = regexec(re, as[0], 1, pmatch, 0);
2204 if (n == 0) {
2205 pmatch[0].rm_so++;
2206 pmatch[0].rm_eo++;
2207 } else {
2208 pmatch[0].rm_so = 0;
2209 pmatch[0].rm_eo = -1;
2210 }
2211 setvar_i(newvar("RSTART"), pmatch[0].rm_so);
2212 setvar_i(newvar("RLENGTH"), pmatch[0].rm_eo - pmatch[0].rm_so);
2213 setvar_i(res, pmatch[0].rm_so);
2214 if (re == &sreg) regfree(re);
2215 break;
2216
Denis Vlasenkof782f522007-01-01 23:51:30 +00002217 case B_ge:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002218 awk_sub(an[0], as[1], getvar_i(av[2]), av[3], res, TRUE);
2219 break;
2220
Denis Vlasenkof782f522007-01-01 23:51:30 +00002221 case B_gs:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002222 setvar_i(res, awk_sub(an[0], as[1], 0, av[2], av[2], FALSE));
2223 break;
2224
Denis Vlasenkof782f522007-01-01 23:51:30 +00002225 case B_su:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002226 setvar_i(res, awk_sub(an[0], as[1], 1, av[2], av[2], FALSE));
2227 break;
2228 }
2229
2230 nvfree(tv);
2231 return res;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002232#undef tspl
Glenn L McGrath545106f2002-11-11 06:21:00 +00002233}
2234
2235/*
2236 * Evaluate node - the heart of the program. Supplied with subtree
2237 * and place where to store result. returns ptr to result.
2238 */
2239#define XC(n) ((n) >> 8)
2240
Mike Frysinger10a11e22005-09-27 02:23:02 +00002241static var *evaluate(node *op, var *res)
2242{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002243/* This procedure is recursive so we should count every byte */
2244#define fnargs (G.evaluate__fnargs)
2245/* seed is initialized to 1 */
2246#define seed (G.evaluate__seed)
2247#define sreg (G.evaluate__sreg)
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002248
Glenn L McGrath545106f2002-11-11 06:21:00 +00002249 node *op1;
2250 var *v1;
2251 union {
2252 var *v;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002253 const char *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002254 double d;
2255 int i;
2256 } L, R;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002257 uint32_t opinfo;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002258 int opn;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002259 union {
2260 char *s;
2261 rstream *rsm;
2262 FILE *F;
2263 var *v;
2264 regex_t *re;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002265 uint32_t info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002266 } X;
2267
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002268 if (!op)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002269 return setvar_s(res, NULL);
2270
2271 v1 = nvalloc(2);
2272
2273 while (op) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002274 opinfo = op->info;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002275 opn = (opinfo & OPNMASK);
2276 g_lineno = op->lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002277
Mike Frysingerde2b9382005-09-27 03:18:00 +00002278 /* execute inevitable things */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002279 op1 = op->l.n;
2280 if (opinfo & OF_RES1) X.v = L.v = evaluate(op1, v1);
2281 if (opinfo & OF_RES2) R.v = evaluate(op->r.n, v1+1);
2282 if (opinfo & OF_STR1) L.s = getvar_s(L.v);
2283 if (opinfo & OF_STR2) R.s = getvar_s(R.v);
2284 if (opinfo & OF_NUM1) L.d = getvar_i(L.v);
2285
2286 switch (XC(opinfo & OPCLSMASK)) {
2287
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002288 /* -- iterative node type -- */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002289
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002290 /* test pattern */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002291 case XC( OC_TEST ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002292 if ((op1->info & OPCLSMASK) == OC_COMMA) {
2293 /* it's range pattern */
2294 if ((opinfo & OF_CHECKED) || ptest(op1->l.n)) {
2295 op->info |= OF_CHECKED;
2296 if (ptest(op1->r.n))
2297 op->info &= ~OF_CHECKED;
2298
2299 op = op->a.n;
2300 } else {
2301 op = op->r.n;
2302 }
2303 } else {
2304 op = (ptest(op1)) ? op->a.n : op->r.n;
2305 }
2306 break;
2307
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002308 /* just evaluate an expression, also used as unconditional jump */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002309 case XC( OC_EXEC ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002310 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002311
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002312 /* branch, used in if-else and various loops */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002313 case XC( OC_BR ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002314 op = istrue(L.v) ? op->a.n : op->r.n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002315 break;
2316
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002317 /* initialize for-in loop */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002318 case XC( OC_WALKINIT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002319 hashwalk_init(L.v, iamarray(R.v));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002320 break;
2321
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002322 /* get next array item */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002323 case XC( OC_WALKNEXT ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002324 op = hashwalk_next(L.v) ? op->a.n : op->r.n;
2325 break;
2326
Denis Vlasenkof782f522007-01-01 23:51:30 +00002327 case XC( OC_PRINT ):
2328 case XC( OC_PRINTF ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002329 X.F = stdout;
Mike Frysingerde2b9382005-09-27 03:18:00 +00002330 if (op->r.n) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002331 X.rsm = newfile(R.s);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002332 if (!X.rsm->F) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002333 if (opn == '|') {
Denis Vlasenko51742f42007-04-12 00:32:05 +00002334 X.rsm->F = popen(R.s, "w");
2335 if (X.rsm->F == NULL)
Manuel Novoa III cad53642003-03-19 09:13:01 +00002336 bb_perror_msg_and_die("popen");
Glenn L McGrath545106f2002-11-11 06:21:00 +00002337 X.rsm->is_pipe = 1;
2338 } else {
Rob Landleyd921b2e2006-08-03 15:41:12 +00002339 X.rsm->F = xfopen(R.s, opn=='w' ? "w" : "a");
Glenn L McGrath545106f2002-11-11 06:21:00 +00002340 }
2341 }
2342 X.F = X.rsm->F;
2343 }
2344
2345 if ((opinfo & OPCLSMASK) == OC_PRINT) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002346 if (!op1) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002347 fputs(getvar_s(intvar[F0]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002348 } else {
2349 while (op1) {
2350 L.v = evaluate(nextarg(&op1), v1);
2351 if (L.v->type & VF_NUMBER) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002352 fmt_num(g_buf, MAXVARFMT, getvar_s(intvar[OFMT]),
Denis Vlasenkob54b2082006-10-27 09:05:40 +00002353 getvar_i(L.v), TRUE);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002354 fputs(g_buf, X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002355 } else {
2356 fputs(getvar_s(L.v), X.F);
2357 }
2358
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002359 if (op1) fputs(getvar_s(intvar[OFS]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002360 }
2361 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002362 fputs(getvar_s(intvar[ORS]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002363
2364 } else { /* OC_PRINTF */
2365 L.s = awk_printf(op1);
2366 fputs(L.s, X.F);
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002367 free((char*)L.s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002368 }
2369 fflush(X.F);
2370 break;
2371
Denis Vlasenkof782f522007-01-01 23:51:30 +00002372 case XC( OC_DELETE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002373 X.info = op1->info & OPCLSMASK;
2374 if (X.info == OC_VAR) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002375 R.v = op1->l.v;
2376 } else if (X.info == OC_FNARG) {
2377 R.v = &fnargs[op1->l.i];
2378 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002379 syntax_error(EMSG_NOT_ARRAY);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002380 }
2381
Mike Frysingerde2b9382005-09-27 03:18:00 +00002382 if (op1->r.n) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002383 clrvar(L.v);
2384 L.s = getvar_s(evaluate(op1->r.n, v1));
2385 hash_remove(iamarray(R.v), L.s);
2386 } else {
2387 clear_array(iamarray(R.v));
2388 }
2389 break;
2390
Denis Vlasenkof782f522007-01-01 23:51:30 +00002391 case XC( OC_NEWSOURCE ):
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002392 g_progname = op->l.s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002393 break;
2394
Denis Vlasenkof782f522007-01-01 23:51:30 +00002395 case XC( OC_RETURN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002396 copyvar(res, L.v);
2397 break;
2398
Denis Vlasenkof782f522007-01-01 23:51:30 +00002399 case XC( OC_NEXTFILE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002400 nextfile = TRUE;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002401 case XC( OC_NEXT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002402 nextrec = TRUE;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002403 case XC( OC_DONE ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002404 clrvar(res);
2405 break;
2406
Denis Vlasenkof782f522007-01-01 23:51:30 +00002407 case XC( OC_EXIT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002408 awk_exit(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002409
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002410 /* -- recursive node type -- */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002411
Denis Vlasenkof782f522007-01-01 23:51:30 +00002412 case XC( OC_VAR ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002413 L.v = op->l.v;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002414 if (L.v == intvar[NF])
Glenn L McGrath545106f2002-11-11 06:21:00 +00002415 split_f0();
2416 goto v_cont;
2417
Denis Vlasenkof782f522007-01-01 23:51:30 +00002418 case XC( OC_FNARG ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002419 L.v = &fnargs[op->l.i];
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002420 v_cont:
2421 res = op->r.n ? findvar(iamarray(L.v), R.s) : L.v;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002422 break;
2423
Denis Vlasenkof782f522007-01-01 23:51:30 +00002424 case XC( OC_IN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002425 setvar_i(res, hash_search(iamarray(R.v), L.s) ? 1 : 0);
2426 break;
2427
Denis Vlasenkof782f522007-01-01 23:51:30 +00002428 case XC( OC_REGEXP ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002429 op1 = op;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002430 L.s = getvar_s(intvar[F0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002431 goto re_cont;
2432
Denis Vlasenkof782f522007-01-01 23:51:30 +00002433 case XC( OC_MATCH ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002434 op1 = op->r.n;
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002435 re_cont:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002436 X.re = as_regex(op1, &sreg);
2437 R.i = regexec(X.re, L.s, 0, NULL, 0);
2438 if (X.re == &sreg) regfree(X.re);
Denys Vlasenko12847742009-11-30 01:15:04 +01002439 setvar_i(res, (R.i == 0) ^ (opn == '!'));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002440 break;
2441
Denis Vlasenkof782f522007-01-01 23:51:30 +00002442 case XC( OC_MOVE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002443 /* if source is a temporary string, jusk relink it to dest */
Denys Vlasenko12847742009-11-30 01:15:04 +01002444//Disabled: if R.v is numeric but happens to have cached R.v->string,
2445//then L.v ends up being a string, which is wrong
2446// if (R.v == v1+1 && R.v->string) {
2447// res = setvar_p(L.v, R.v->string);
2448// R.v->string = NULL;
2449// } else {
Mike Frysingerde2b9382005-09-27 03:18:00 +00002450 res = copyvar(L.v, R.v);
Denys Vlasenko12847742009-11-30 01:15:04 +01002451// }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002452 break;
2453
Denis Vlasenkof782f522007-01-01 23:51:30 +00002454 case XC( OC_TERNARY ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002455 if ((op->r.n->info & OPCLSMASK) != OC_COLON)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002456 syntax_error(EMSG_POSSIBLE_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002457 res = evaluate(istrue(L.v) ? op->r.n->l.n : op->r.n->r.n, res);
2458 break;
2459
Denis Vlasenkof782f522007-01-01 23:51:30 +00002460 case XC( OC_FUNC ):
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002461 if (!op->r.f->body.first)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002462 syntax_error(EMSG_UNDEF_FUNC);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002463
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002464 X.v = R.v = nvalloc(op->r.f->nargs + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002465 while (op1) {
2466 L.v = evaluate(nextarg(&op1), v1);
2467 copyvar(R.v, L.v);
2468 R.v->type |= VF_CHILD;
2469 R.v->x.parent = L.v;
2470 if (++R.v - X.v >= op->r.f->nargs)
2471 break;
2472 }
2473
2474 R.v = fnargs;
2475 fnargs = X.v;
2476
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002477 L.s = g_progname;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002478 res = evaluate(op->r.f->body.first, res);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002479 g_progname = L.s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002480
2481 nvfree(fnargs);
2482 fnargs = R.v;
2483 break;
2484
Denis Vlasenkof782f522007-01-01 23:51:30 +00002485 case XC( OC_GETLINE ):
2486 case XC( OC_PGETLINE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002487 if (op1) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002488 X.rsm = newfile(L.s);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002489 if (!X.rsm->F) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002490 if ((opinfo & OPCLSMASK) == OC_PGETLINE) {
2491 X.rsm->F = popen(L.s, "r");
2492 X.rsm->is_pipe = TRUE;
2493 } else {
Denis Vlasenko5415c852008-07-21 23:05:26 +00002494 X.rsm->F = fopen_for_read(L.s); /* not xfopen! */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002495 }
2496 }
2497 } else {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002498 if (!iF) iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002499 X.rsm = iF;
2500 }
2501
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002502 if (!X.rsm->F) {
2503 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002504 setvar_i(res, -1);
2505 break;
2506 }
2507
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002508 if (!op->r.n)
2509 R.v = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002510
2511 L.i = awk_getline(X.rsm, R.v);
2512 if (L.i > 0) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002513 if (!op1) {
2514 incvar(intvar[FNR]);
2515 incvar(intvar[NR]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002516 }
2517 }
2518 setvar_i(res, L.i);
2519 break;
2520
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002521 /* simple builtins */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002522 case XC( OC_FBLTIN ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002523 switch (opn) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002524
Denis Vlasenkof782f522007-01-01 23:51:30 +00002525 case F_in:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002526 R.d = (int)L.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002527 break;
2528
Denis Vlasenkof782f522007-01-01 23:51:30 +00002529 case F_rn:
2530 R.d = (double)rand() / (double)RAND_MAX;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002531 break;
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002532#if ENABLE_FEATURE_AWK_LIBM
Denis Vlasenkof782f522007-01-01 23:51:30 +00002533 case F_co:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002534 R.d = cos(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002535 break;
2536
Denis Vlasenkof782f522007-01-01 23:51:30 +00002537 case F_ex:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002538 R.d = exp(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002539 break;
2540
Denis Vlasenkof782f522007-01-01 23:51:30 +00002541 case F_lg:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002542 R.d = log(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002543 break;
2544
Denis Vlasenkof782f522007-01-01 23:51:30 +00002545 case F_si:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002546 R.d = sin(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002547 break;
2548
Denis Vlasenkof782f522007-01-01 23:51:30 +00002549 case F_sq:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002550 R.d = sqrt(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002551 break;
2552#else
Denis Vlasenkof782f522007-01-01 23:51:30 +00002553 case F_co:
2554 case F_ex:
2555 case F_lg:
2556 case F_si:
2557 case F_sq:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002558 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002559 break;
2560#endif
Denis Vlasenkof782f522007-01-01 23:51:30 +00002561 case F_sr:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002562 R.d = (double)seed;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002563 seed = op1 ? (unsigned)L.d : (unsigned)time(NULL);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002564 srand(seed);
2565 break;
2566
Denis Vlasenkof782f522007-01-01 23:51:30 +00002567 case F_ti:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002568 R.d = time(NULL);
2569 break;
2570
Denis Vlasenkof782f522007-01-01 23:51:30 +00002571 case F_le:
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002572 if (!op1)
2573 L.s = getvar_s(intvar[F0]);
Rob Landleya3896512006-05-07 20:20:34 +00002574 R.d = strlen(L.s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002575 break;
2576
Denis Vlasenkof782f522007-01-01 23:51:30 +00002577 case F_sy:
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002578 fflush_all();
Denis Vlasenko249fabf2006-12-19 00:29:22 +00002579 R.d = (ENABLE_FEATURE_ALLOW_EXEC && L.s && *L.s)
2580 ? (system(L.s) >> 8) : 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002581 break;
2582
Denis Vlasenkof782f522007-01-01 23:51:30 +00002583 case F_ff:
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002584 if (!op1)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002585 fflush(stdout);
2586 else {
2587 if (L.s && *L.s) {
2588 X.rsm = newfile(L.s);
2589 fflush(X.rsm->F);
2590 } else {
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002591 fflush_all();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002592 }
2593 }
2594 break;
2595
Denis Vlasenkof782f522007-01-01 23:51:30 +00002596 case F_cl:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002597 X.rsm = (rstream *)hash_search(fdhash, L.s);
2598 if (X.rsm) {
2599 R.i = X.rsm->is_pipe ? pclose(X.rsm->F) : fclose(X.rsm->F);
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00002600 free(X.rsm->buffer);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002601 hash_remove(fdhash, L.s);
2602 }
2603 if (R.i != 0)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002604 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002605 R.d = (double)R.i;
2606 break;
2607 }
2608 setvar_i(res, R.d);
2609 break;
2610
Denis Vlasenkof782f522007-01-01 23:51:30 +00002611 case XC( OC_BUILTIN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002612 res = exec_builtin(op, res);
2613 break;
2614
Denis Vlasenkof782f522007-01-01 23:51:30 +00002615 case XC( OC_SPRINTF ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002616 setvar_p(res, awk_printf(op1));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002617 break;
2618
Denis Vlasenkof782f522007-01-01 23:51:30 +00002619 case XC( OC_UNARY ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002620 X.v = R.v;
2621 L.d = R.d = getvar_i(R.v);
2622 switch (opn) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002623 case 'P':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002624 L.d = ++R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002625 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002626 case 'p':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002627 R.d++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002628 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002629 case 'M':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002630 L.d = --R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002631 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002632 case 'm':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002633 R.d--;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002634 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002635 case '!':
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002636 L.d = !istrue(X.v);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002637 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002638 case '-':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002639 L.d = -R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002640 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002641 r_op_change:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002642 setvar_i(X.v, R.d);
2643 }
2644 setvar_i(res, L.d);
2645 break;
2646
Denis Vlasenkof782f522007-01-01 23:51:30 +00002647 case XC( OC_FIELD ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002648 R.i = (int)getvar_i(R.v);
2649 if (R.i == 0) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002650 res = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002651 } else {
2652 split_f0();
2653 if (R.i > nfields)
2654 fsrealloc(R.i);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002655 res = &Fields[R.i - 1];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002656 }
2657 break;
2658
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002659 /* concatenation (" ") and index joining (",") */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002660 case XC( OC_CONCAT ):
2661 case XC( OC_COMMA ):
Rob Landleya3896512006-05-07 20:20:34 +00002662 opn = strlen(L.s) + strlen(R.s) + 2;
Denis Vlasenkob95636c2006-12-19 23:36:04 +00002663 X.s = xmalloc(opn);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002664 strcpy(X.s, L.s);
2665 if ((opinfo & OPCLSMASK) == OC_COMMA) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002666 L.s = getvar_s(intvar[SUBSEP]);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002667 X.s = xrealloc(X.s, opn + strlen(L.s));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002668 strcat(X.s, L.s);
2669 }
2670 strcat(X.s, R.s);
2671 setvar_p(res, X.s);
2672 break;
2673
Denis Vlasenkof782f522007-01-01 23:51:30 +00002674 case XC( OC_LAND ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002675 setvar_i(res, istrue(L.v) ? ptest(op->r.n) : 0);
2676 break;
2677
Denis Vlasenkof782f522007-01-01 23:51:30 +00002678 case XC( OC_LOR ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002679 setvar_i(res, istrue(L.v) ? 1 : ptest(op->r.n));
2680 break;
2681
Denis Vlasenkof782f522007-01-01 23:51:30 +00002682 case XC( OC_BINARY ):
2683 case XC( OC_REPLACE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002684 R.d = getvar_i(R.v);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002685 switch (opn) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002686 case '+':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002687 L.d += R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002688 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002689 case '-':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002690 L.d -= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002691 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002692 case '*':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002693 L.d *= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002694 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002695 case '/':
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002696 if (R.d == 0)
2697 syntax_error(EMSG_DIV_BY_ZERO);
Mike Frysingerde2b9382005-09-27 03:18:00 +00002698 L.d /= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002699 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002700 case '&':
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002701#if ENABLE_FEATURE_AWK_LIBM
Mike Frysingerde2b9382005-09-27 03:18:00 +00002702 L.d = pow(L.d, R.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002703#else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002704 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002705#endif
2706 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002707 case '%':
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002708 if (R.d == 0)
2709 syntax_error(EMSG_DIV_BY_ZERO);
Mike Frysingerde2b9382005-09-27 03:18:00 +00002710 L.d -= (int)(L.d / R.d) * R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002711 break;
2712 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002713 res = setvar_i(((opinfo & OPCLSMASK) == OC_BINARY) ? res : X.v, L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002714 break;
2715
Denis Vlasenkof782f522007-01-01 23:51:30 +00002716 case XC( OC_COMPARE ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002717 if (is_numeric(L.v) && is_numeric(R.v)) {
2718 L.d = getvar_i(L.v) - getvar_i(R.v);
2719 } else {
2720 L.s = getvar_s(L.v);
2721 R.s = getvar_s(R.v);
2722 L.d = icase ? strcasecmp(L.s, R.s) : strcmp(L.s, R.s);
2723 }
2724 switch (opn & 0xfe) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002725 case 0:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002726 R.i = (L.d > 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002727 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002728 case 2:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002729 R.i = (L.d >= 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002730 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002731 case 4:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002732 R.i = (L.d == 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002733 break;
2734 }
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002735 setvar_i(res, (opn & 1 ? R.i : !R.i) ? 1 : 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002736 break;
2737
Denis Vlasenkof782f522007-01-01 23:51:30 +00002738 default:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002739 syntax_error(EMSG_POSSIBLE_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002740 }
2741 if ((opinfo & OPCLSMASK) <= SHIFT_TIL_THIS)
2742 op = op->a.n;
2743 if ((opinfo & OPCLSMASK) >= RECUR_FROM_THIS)
2744 break;
2745 if (nextrec)
2746 break;
2747 }
2748 nvfree(v1);
2749 return res;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002750#undef fnargs
2751#undef seed
2752#undef sreg
Glenn L McGrath545106f2002-11-11 06:21:00 +00002753}
2754
2755
2756/* -------- main & co. -------- */
2757
Mike Frysinger10a11e22005-09-27 02:23:02 +00002758static int awk_exit(int r)
2759{
Denis Vlasenkof782f522007-01-01 23:51:30 +00002760 var tv;
2761 unsigned i;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002762 hash_item *hi;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002763
Denis Vlasenkof782f522007-01-01 23:51:30 +00002764 zero_out_var(&tv);
2765
2766 if (!exiting) {
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002767 exiting = TRUE;
Glenn L McGrathca29ffc2004-09-24 09:24:27 +00002768 nextrec = FALSE;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002769 evaluate(endseq.first, &tv);
2770 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002771
2772 /* waiting for children */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002773 for (i = 0; i < fdhash->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002774 hi = fdhash->items[i];
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00002775 while (hi) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002776 if (hi->data.rs.F && hi->data.rs.is_pipe)
2777 pclose(hi->data.rs.F);
2778 hi = hi->next;
2779 }
2780 }
2781
2782 exit(r);
2783}
2784
2785/* if expr looks like "var=value", perform assignment and return 1,
2786 * otherwise return 0 */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00002787static int is_assignment(const char *expr)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002788{
Glenn L McGrath545106f2002-11-11 06:21:00 +00002789 char *exprc, *s, *s0, *s1;
2790
Rob Landleyd921b2e2006-08-03 15:41:12 +00002791 exprc = xstrdup(expr);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002792 if (!isalnum_(*exprc) || (s = strchr(exprc, '=')) == NULL) {
2793 free(exprc);
2794 return FALSE;
2795 }
2796
2797 *(s++) = '\0';
2798 s0 = s1 = s;
2799 while (*s)
2800 *(s1++) = nextchar(&s);
2801
2802 *s1 = '\0';
2803 setvar_u(newvar(exprc), s0);
2804 free(exprc);
2805 return TRUE;
2806}
2807
2808/* switch to next input file */
Mike Frysinger10a11e22005-09-27 02:23:02 +00002809static rstream *next_input_file(void)
2810{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002811#define rsm (G.next_input_file__rsm)
2812#define files_happen (G.next_input_file__files_happen)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002813
Glenn L McGrath545106f2002-11-11 06:21:00 +00002814 FILE *F = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002815 const char *fname, *ind;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002816
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002817 if (rsm.F)
2818 fclose(rsm.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002819 rsm.F = NULL;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002820 rsm.pos = rsm.adv = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002821
2822 do {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002823 if (getvar_i(intvar[ARGIND])+1 >= getvar_i(intvar[ARGC])) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002824 if (files_happen)
2825 return NULL;
2826 fname = "-";
2827 F = stdin;
2828 } else {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002829 ind = getvar_s(incvar(intvar[ARGIND]));
2830 fname = getvar_s(findvar(iamarray(intvar[ARGV]), ind));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002831 if (fname && *fname && !is_assignment(fname))
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002832 F = xfopen_stdin(fname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002833 }
2834 } while (!F);
2835
2836 files_happen = TRUE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002837 setvar_s(intvar[FILENAME], fname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002838 rsm.F = F;
2839 return &rsm;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002840#undef rsm
2841#undef files_happen
Glenn L McGrath545106f2002-11-11 06:21:00 +00002842}
2843
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00002844int awk_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Rob Landleydfba7412006-03-06 20:47:33 +00002845int awk_main(int argc, char **argv)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002846{
Denis Vlasenko67b23e62006-10-03 21:00:06 +00002847 unsigned opt;
Denis Vlasenkobe644a82007-03-10 17:22:14 +00002848 char *opt_F, *opt_W;
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002849 llist_t *list_v = NULL;
2850 llist_t *list_f = NULL;
2851 int i, j;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002852 var *v;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002853 var tv;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002854 char **envp;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002855 char *vnames = (char *)vNames; /* cheat */
2856 char *vvalues = (char *)vValues;
2857
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002858 INIT_G();
2859
Denis Vlasenko150f4022007-01-13 21:06:21 +00002860 /* Undo busybox.c, or else strtod may eat ','! This breaks parsing:
Denis Vlasenko6dc6ebb2007-01-01 23:53:12 +00002861 * $1,$2 == '$1,' '$2', NOT '$1' ',' '$2' */
2862 if (ENABLE_LOCALE_SUPPORT)
2863 setlocale(LC_NUMERIC, "C");
2864
Denis Vlasenkof782f522007-01-01 23:51:30 +00002865 zero_out_var(&tv);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002866
2867 /* allocate global buffer */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002868 g_buf = xmalloc(MAXVARFMT + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002869
2870 vhash = hash_init();
2871 ahash = hash_init();
2872 fdhash = hash_init();
2873 fnhash = hash_init();
2874
2875 /* initialize variables */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002876 for (i = 0; *vnames; i++) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002877 intvar[i] = v = newvar(nextword(&vnames));
Denis Vlasenkof782f522007-01-01 23:51:30 +00002878 if (*vvalues != '\377')
2879 setvar_s(v, nextword(&vvalues));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002880 else
2881 setvar_i(v, 0);
2882
Denis Vlasenkof782f522007-01-01 23:51:30 +00002883 if (*vnames == '*') {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002884 v->type |= VF_SPECIAL;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002885 vnames++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002886 }
2887 }
2888
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002889 handle_special(intvar[FS]);
2890 handle_special(intvar[RS]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002891
Denis Vlasenkof782f522007-01-01 23:51:30 +00002892 newfile("/dev/stdin")->F = stdin;
2893 newfile("/dev/stdout")->F = stdout;
2894 newfile("/dev/stderr")->F = stderr;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002895
Denis Vlasenkof71d9162007-05-03 22:57:56 +00002896 /* Huh, people report that sometimes environ is NULL. Oh well. */
2897 if (environ) for (envp = environ; *envp; envp++) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002898 /* environ is writable, thus we don't strdup it needlessly */
2899 char *s = *envp;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002900 char *s1 = strchr(s, '=');
2901 if (s1) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002902 *s1 = '\0';
2903 /* Both findvar and setvar_u take const char*
2904 * as 2nd arg -> environment is not trashed */
2905 setvar_u(findvar(iamarray(intvar[ENVIRON]), s), s1 + 1);
2906 *s1 = '=';
Eric Andersen67776be2004-07-30 23:52:08 +00002907 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002908 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002909 opt_complementary = "v::f::"; /* -v and -f can occur multiple times */
2910 opt = getopt32(argv, "F:v:f:W:", &opt_F, &list_v, &list_f, &opt_W);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002911 argv += optind;
2912 argc -= optind;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002913 if (opt & 0x1)
2914 setvar_s(intvar[FS], opt_F); // -F
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002915 while (list_v) { /* -v */
2916 if (!is_assignment(llist_pop(&list_v)))
Denis Vlasenkobe644a82007-03-10 17:22:14 +00002917 bb_show_usage();
2918 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002919 if (list_f) { /* -f */
2920 do {
2921 char *s = NULL;
2922 FILE *from_file;
2923
2924 g_progname = llist_pop(&list_f);
2925 from_file = xfopen_stdin(g_progname);
2926 /* one byte is reserved for some trick in next_token */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002927 for (i = j = 1; j > 0; i += j) {
2928 s = xrealloc(s, i + 4096);
2929 j = fread(s + i, 1, 4094, from_file);
Denis Vlasenko099efbf2006-09-22 09:02:30 +00002930 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002931 s[i] = '\0';
2932 fclose(from_file);
2933 parse_program(s + 1);
2934 free(s);
2935 } while (list_f);
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +00002936 argc++;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002937 } else { // no -f: take program from 1st parameter
2938 if (!argc)
2939 bb_show_usage();
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002940 g_progname = "cmd. line";
Denis Vlasenkof782f522007-01-01 23:51:30 +00002941 parse_program(*argv++);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002942 }
Denis Vlasenko099efbf2006-09-22 09:02:30 +00002943 if (opt & 0x8) // -W
Denis Vlasenkoe1a0d482006-10-20 13:28:22 +00002944 bb_error_msg("warning: unrecognized option '-W %s' ignored", opt_W);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002945
Glenn L McGrath545106f2002-11-11 06:21:00 +00002946 /* fill in ARGV array */
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +00002947 setvar_i(intvar[ARGC], argc);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002948 setari_u(intvar[ARGV], 0, "awk");
Denis Vlasenkof782f522007-01-01 23:51:30 +00002949 i = 0;
2950 while (*argv)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002951 setari_u(intvar[ARGV], ++i, *argv++);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002952
2953 evaluate(beginseq.first, &tv);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002954 if (!mainseq.first && !endseq.first)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002955 awk_exit(EXIT_SUCCESS);
2956
2957 /* input file could already be opened in BEGIN block */
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002958 if (!iF)
2959 iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002960
2961 /* passing through input files */
2962 while (iF) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002963 nextfile = FALSE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002964 setvar_i(intvar[FNR], 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002965
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002966 while ((i = awk_getline(iF, intvar[F0])) > 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002967 nextrec = FALSE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002968 incvar(intvar[NR]);
2969 incvar(intvar[FNR]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002970 evaluate(mainseq.first, &tv);
2971
2972 if (nextfile)
2973 break;
2974 }
2975
Denis Vlasenkof782f522007-01-01 23:51:30 +00002976 if (i < 0)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002977 syntax_error(strerror(errno));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002978
2979 iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002980 }
2981
Glenn L McGrath545106f2002-11-11 06:21:00 +00002982 awk_exit(EXIT_SUCCESS);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002983 /*return 0;*/
Glenn L McGrath545106f2002-11-11 06:21:00 +00002984}