blob: 5cc4adce77b47178ffefff5ca0d99f684f057524 [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 */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001516static void qrealloc(char **b, int n, int *size)
1517{
Denis Vlasenkodeeed592008-07-08 05:14:36 +00001518 if (!*b || n >= *size) {
1519 *size = n + (n>>1) + 80;
1520 *b = xrealloc(*b, *size);
1521 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001522}
1523
1524/* resize field storage space */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001525static void fsrealloc(int size)
1526{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001527 int i;
1528
1529 if (size >= maxfields) {
1530 i = maxfields;
1531 maxfields = size + 16;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001532 Fields = xrealloc(Fields, maxfields * sizeof(var));
1533 for (; i < maxfields; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001534 Fields[i].type = VF_SPECIAL;
1535 Fields[i].string = NULL;
1536 }
1537 }
1538
1539 if (size < nfields) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001540 for (i = size; i < nfields; i++) {
1541 clrvar(Fields + i);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001542 }
1543 }
1544 nfields = size;
1545}
1546
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001547static int awk_split(const char *s, node *spl, char **slist)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001548{
Denis Vlasenkof782f522007-01-01 23:51:30 +00001549 int l, n = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001550 char c[4];
1551 char *s1;
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001552 regmatch_t pmatch[2]; // TODO: why [2]? [1] is enough...
Glenn L McGrath545106f2002-11-11 06:21:00 +00001553
1554 /* in worst case, each char would be a separate field */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001555 *slist = s1 = xzalloc(strlen(s) * 2 + 3);
1556 strcpy(s1, s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001557
1558 c[0] = c[1] = (char)spl->info;
1559 c[2] = c[3] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001560 if (*getvar_s(intvar[RS]) == '\0')
1561 c[2] = '\n';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001562
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001563 if ((spl->info & OPCLSMASK) == OC_REGEXP) { /* regex split */
1564 if (!*s)
1565 return n; /* "": zero fields */
1566 n++; /* at least one field will be there */
1567 do {
1568 l = strcspn(s, c+2); /* len till next NUL or \n */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001569 if (regexec(icase ? spl->r.ire : spl->l.re, s, 1, pmatch, 0) == 0
1570 && pmatch[0].rm_so <= l
1571 ) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001572 l = pmatch[0].rm_so;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001573 if (pmatch[0].rm_eo == 0) {
1574 l++;
1575 pmatch[0].rm_eo++;
1576 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001577 n++; /* we saw yet another delimiter */
Glenn L McGrath545106f2002-11-11 06:21:00 +00001578 } else {
1579 pmatch[0].rm_eo = l;
Denys Vlasenkoe4244232009-05-18 23:50:03 +02001580 if (s[l])
1581 pmatch[0].rm_eo++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001582 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001583 memcpy(s1, s, l);
Denis Vlasenko67b5eeb2009-04-12 13:54:13 +00001584 /* make sure we remove *all* of the separator chars */
Denys Vlasenkoe4244232009-05-18 23:50:03 +02001585 do {
1586 s1[l] = '\0';
1587 } while (++l < pmatch[0].rm_eo);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001588 nextword(&s1);
1589 s += pmatch[0].rm_eo;
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001590 } while (*s);
1591 return n;
1592 }
1593 if (c[0] == '\0') { /* null split */
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001594 while (*s) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001595 *s1++ = *s++;
1596 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001597 n++;
1598 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001599 return n;
1600 }
1601 if (c[0] != ' ') { /* single-character split */
Glenn L McGrath545106f2002-11-11 06:21:00 +00001602 if (icase) {
1603 c[0] = toupper(c[0]);
1604 c[1] = tolower(c[1]);
1605 }
1606 if (*s1) n++;
1607 while ((s1 = strpbrk(s1, c))) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001608 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001609 n++;
1610 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001611 return n;
1612 }
1613 /* space split */
1614 while (*s) {
1615 s = skip_whitespace(s);
1616 if (!*s) break;
1617 n++;
1618 while (*s && !isspace(*s))
1619 *s1++ = *s++;
1620 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001621 }
1622 return n;
1623}
1624
Mike Frysinger10a11e22005-09-27 02:23:02 +00001625static void split_f0(void)
1626{
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001627/* static char *fstrings; */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001628#define fstrings (G.split_f0__fstrings)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001629
Glenn L McGrath545106f2002-11-11 06:21:00 +00001630 int i, n;
1631 char *s;
1632
1633 if (is_f0_split)
1634 return;
1635
1636 is_f0_split = TRUE;
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00001637 free(fstrings);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001638 fsrealloc(0);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001639 n = awk_split(getvar_s(intvar[F0]), &fsplitter.n, &fstrings);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001640 fsrealloc(n);
1641 s = fstrings;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001642 for (i = 0; i < n; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001643 Fields[i].string = nextword(&s);
1644 Fields[i].type |= (VF_FSTR | VF_USER | VF_DIRTY);
1645 }
1646
1647 /* set NF manually to avoid side effects */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001648 clrvar(intvar[NF]);
1649 intvar[NF]->type = VF_NUMBER | VF_SPECIAL;
1650 intvar[NF]->number = nfields;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001651#undef fstrings
Glenn L McGrath545106f2002-11-11 06:21:00 +00001652}
1653
1654/* perform additional actions when some internal variables changed */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001655static void handle_special(var *v)
1656{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001657 int n;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001658 char *b;
1659 const char *sep, *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001660 int sl, l, len, i, bsize;
1661
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001662 if (!(v->type & VF_SPECIAL))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001663 return;
1664
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001665 if (v == intvar[NF]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001666 n = (int)getvar_i(v);
1667 fsrealloc(n);
1668
1669 /* recalculate $0 */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001670 sep = getvar_s(intvar[OFS]);
Rob Landleya3896512006-05-07 20:20:34 +00001671 sl = strlen(sep);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001672 b = NULL;
1673 len = 0;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001674 for (i = 0; i < n; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001675 s = getvar_s(&Fields[i]);
Rob Landleya3896512006-05-07 20:20:34 +00001676 l = strlen(s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001677 if (b) {
1678 memcpy(b+len, sep, sl);
1679 len += sl;
1680 }
1681 qrealloc(&b, len+l+sl, &bsize);
1682 memcpy(b+len, s, l);
1683 len += l;
1684 }
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001685 if (b)
1686 b[len] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001687 setvar_p(intvar[F0], b);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001688 is_f0_split = TRUE;
1689
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001690 } else if (v == intvar[F0]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001691 is_f0_split = FALSE;
1692
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001693 } else if (v == intvar[FS]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001694 mk_splitter(getvar_s(v), &fsplitter);
1695
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001696 } else if (v == intvar[RS]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001697 mk_splitter(getvar_s(v), &rsplitter);
1698
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001699 } else if (v == intvar[IGNORECASE]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001700 icase = istrue(v);
1701
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001702 } else { /* $n */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001703 n = getvar_i(intvar[NF]);
1704 setvar_i(intvar[NF], n > v-Fields ? n : v-Fields+1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001705 /* right here v is invalid. Just to note... */
1706 }
1707}
1708
1709/* step through func/builtin/etc arguments */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001710static node *nextarg(node **pn)
1711{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001712 node *n;
1713
1714 n = *pn;
1715 if (n && (n->info & OPCLSMASK) == OC_COMMA) {
1716 *pn = n->r.n;
1717 n = n->l.n;
1718 } else {
1719 *pn = NULL;
1720 }
1721 return n;
1722}
1723
Mike Frysinger10a11e22005-09-27 02:23:02 +00001724static void hashwalk_init(var *v, xhash *array)
1725{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001726 char **w;
1727 hash_item *hi;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +00001728 unsigned i;
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001729 char **prev_walker = (v->type & VF_WALK) ? v->x.walker : NULL;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001730
1731 v->type |= VF_WALK;
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001732
1733 /* walker structure is: "[ptr2end][ptr2start][prev]<word1>NUL<word2>NUL" */
1734 w = v->x.walker = xzalloc(2 + 3*sizeof(char *) + array->glen);
1735 //bb_error_msg("walker@%p=%p", &v->x.walker, v->x.walker);
1736 w[0] = w[1] = (char *)(w + 3);
1737 w[2] = (char *)prev_walker;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001738 for (i = 0; i < array->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001739 hi = array->items[i];
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001740 while (hi) {
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001741 strcpy(w[0], hi->name);
1742 nextword(&w[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001743 hi = hi->next;
1744 }
1745 }
1746}
1747
Mike Frysinger10a11e22005-09-27 02:23:02 +00001748static int hashwalk_next(var *v)
1749{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001750 char **w;
1751
1752 w = v->x.walker;
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001753 if (w[1] == w[0]) {
1754 char **prev_walker = (char**)w[2];
Glenn L McGrath545106f2002-11-11 06:21:00 +00001755
Denys Vlasenko3cb60c32010-03-10 19:20:32 +01001756 //bb_error_msg("free(walker@%p:%p) #3, restoring to %p", &v->x.walker, v->x.walker, prev_walker);
1757 free(v->x.walker);
1758 v->x.walker = prev_walker;
1759 return FALSE;
1760 }
1761
1762 setvar_s(v, nextword(&w[1]));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001763 return TRUE;
1764}
1765
1766/* evaluate node, return 1 when result is true, 0 otherwise */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001767static int ptest(node *pattern)
1768{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001769 /* ptest__v is "static": to save stack space? */
1770 return istrue(evaluate(pattern, &G.ptest__v));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001771}
1772
1773/* read next record from stream rsm into a variable v */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001774static int awk_getline(rstream *rsm, var *v)
1775{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001776 char *b;
1777 regmatch_t pmatch[2];
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001778 int a, p, pp=0, size;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001779 int fd, so, eo, r, rp;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001780 char c, *m, *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001781
1782 /* we're using our own buffer since we need access to accumulating
1783 * characters
1784 */
1785 fd = fileno(rsm->F);
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001786 m = rsm->buffer;
1787 a = rsm->adv;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001788 p = rsm->pos;
1789 size = rsm->size;
1790 c = (char) rsplitter.n.info;
1791 rp = 0;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001792
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001793 if (!m) qrealloc(&m, 256, &size);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001794 do {
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001795 b = m + a;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001796 so = eo = p;
1797 r = 1;
1798 if (p > 0) {
1799 if ((rsplitter.n.info & OPCLSMASK) == OC_REGEXP) {
1800 if (regexec(icase ? rsplitter.n.r.ire : rsplitter.n.l.re,
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001801 b, 1, pmatch, 0) == 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001802 so = pmatch[0].rm_so;
1803 eo = pmatch[0].rm_eo;
1804 if (b[eo] != '\0')
1805 break;
1806 }
1807 } else if (c != '\0') {
1808 s = strchr(b+pp, c);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001809 if (!s) s = memchr(b+pp, '\0', p - pp);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001810 if (s) {
1811 so = eo = s-b;
1812 eo++;
1813 break;
1814 }
1815 } else {
1816 while (b[rp] == '\n')
1817 rp++;
1818 s = strstr(b+rp, "\n\n");
1819 if (s) {
1820 so = eo = s-b;
1821 while (b[eo] == '\n') eo++;
1822 if (b[eo] != '\0')
1823 break;
1824 }
1825 }
1826 }
1827
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001828 if (a > 0) {
1829 memmove(m, (const void *)(m+a), p+1);
1830 b = m;
1831 a = 0;
1832 }
1833
1834 qrealloc(&m, a+p+128, &size);
1835 b = m + a;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001836 pp = p;
1837 p += safe_read(fd, b+p, size-p-1);
1838 if (p < pp) {
1839 p = 0;
1840 r = 0;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001841 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001842 }
1843 b[p] = '\0';
1844
1845 } while (p > pp);
1846
1847 if (p == 0) {
1848 r--;
1849 } else {
1850 c = b[so]; b[so] = '\0';
1851 setvar_s(v, b+rp);
1852 v->type |= VF_USER;
1853 b[so] = c;
1854 c = b[eo]; b[eo] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001855 setvar_s(intvar[RT], b+so);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001856 b[eo] = c;
1857 }
1858
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001859 rsm->buffer = m;
1860 rsm->adv = a + eo;
1861 rsm->pos = p - eo;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001862 rsm->size = size;
1863
1864 return r;
1865}
1866
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00001867static int fmt_num(char *b, int size, const char *format, double n, int int_as_int)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001868{
Denis Vlasenkof782f522007-01-01 23:51:30 +00001869 int r = 0;
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00001870 char c;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001871 const char *s = format;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001872
1873 if (int_as_int && n == (int)n) {
1874 r = snprintf(b, size, "%d", (int)n);
1875 } else {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001876 do { c = *s; } while (c && *++s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001877 if (strchr("diouxX", c)) {
1878 r = snprintf(b, size, format, (int)n);
1879 } else if (strchr("eEfgG", c)) {
1880 r = snprintf(b, size, format, n);
1881 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001882 syntax_error(EMSG_INV_FMT);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001883 }
1884 }
1885 return r;
1886}
1887
Glenn L McGrath545106f2002-11-11 06:21:00 +00001888/* formatted output into an allocated buffer, return ptr to buffer */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001889static char *awk_printf(node *n)
1890{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001891 char *b = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001892 char *fmt, *s, *f;
1893 const char *s1;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001894 int i, j, incr, bsize;
1895 char c, c1;
1896 var *v, *arg;
1897
1898 v = nvalloc(1);
Rob Landleyd921b2e2006-08-03 15:41:12 +00001899 fmt = f = xstrdup(getvar_s(evaluate(nextarg(&n), v)));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001900
1901 i = 0;
1902 while (*f) {
1903 s = f;
1904 while (*f && (*f != '%' || *(++f) == '%'))
1905 f++;
Denis Vlasenko389f9d52007-05-09 21:57:23 +00001906 while (*f && !isalpha(*f)) {
1907 if (*f == '*')
1908 syntax_error("%*x formats are not supported");
Glenn L McGrath545106f2002-11-11 06:21:00 +00001909 f++;
Denis Vlasenko389f9d52007-05-09 21:57:23 +00001910 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001911
1912 incr = (f - s) + MAXVARFMT;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001913 qrealloc(&b, incr + i, &bsize);
1914 c = *f;
1915 if (c != '\0') f++;
1916 c1 = *f;
1917 *f = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001918 arg = evaluate(nextarg(&n), v);
1919
1920 j = i;
1921 if (c == 'c' || !c) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001922 i += sprintf(b+i, s, is_numeric(arg) ?
1923 (char)getvar_i(arg) : *getvar_s(arg));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001924 } else if (c == 's') {
Denis Vlasenko92758142006-10-03 19:56:34 +00001925 s1 = getvar_s(arg);
Rob Landleya3896512006-05-07 20:20:34 +00001926 qrealloc(&b, incr+i+strlen(s1), &bsize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001927 i += sprintf(b+i, s, s1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001928 } else {
1929 i += fmt_num(b+i, incr, s, getvar_i(arg), FALSE);
1930 }
1931 *f = c1;
1932
1933 /* if there was an error while sprintf, return value is negative */
1934 if (i < j) i = j;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001935 }
1936
Denis Vlasenkof782f522007-01-01 23:51:30 +00001937 b = xrealloc(b, i + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001938 free(fmt);
1939 nvfree(v);
1940 b[i] = '\0';
1941 return b;
1942}
1943
1944/* common substitution routine
1945 * replace (nm) substring of (src) that match (n) with (repl), store
1946 * result into (dest), return number of substitutions. If nm=0, replace
1947 * all matches. If src or dst is NULL, use $0. If ex=TRUE, enable
1948 * subexpression matching (\1-\9)
1949 */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001950static int awk_sub(node *rn, const char *repl, int nm, var *src, var *dest, int ex)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001951{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001952 char *ds = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001953 const char *s;
1954 const char *sp;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001955 int c, i, j, di, rl, so, eo, nbs, n, dssize;
1956 regmatch_t pmatch[10];
1957 regex_t sreg, *re;
1958
1959 re = as_regex(rn, &sreg);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001960 if (!src) src = intvar[F0];
1961 if (!dest) dest = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00001962
1963 i = di = 0;
1964 sp = getvar_s(src);
Rob Landleya3896512006-05-07 20:20:34 +00001965 rl = strlen(repl);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001966 while (regexec(re, sp, 10, pmatch, sp==getvar_s(src) ? 0 : REG_NOTBOL) == 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001967 so = pmatch[0].rm_so;
1968 eo = pmatch[0].rm_eo;
1969
1970 qrealloc(&ds, di + eo + rl, &dssize);
1971 memcpy(ds + di, sp, eo);
1972 di += eo;
1973 if (++i >= nm) {
1974 /* replace */
1975 di -= (eo - so);
1976 nbs = 0;
1977 for (s = repl; *s; s++) {
1978 ds[di++] = c = *s;
1979 if (c == '\\') {
1980 nbs++;
1981 continue;
1982 }
1983 if (c == '&' || (ex && c >= '0' && c <= '9')) {
1984 di -= ((nbs + 3) >> 1);
1985 j = 0;
1986 if (c != '&') {
1987 j = c - '0';
1988 nbs++;
1989 }
1990 if (nbs % 2) {
1991 ds[di++] = c;
1992 } else {
1993 n = pmatch[j].rm_eo - pmatch[j].rm_so;
1994 qrealloc(&ds, di + rl + n, &dssize);
1995 memcpy(ds + di, sp + pmatch[j].rm_so, n);
1996 di += n;
1997 }
1998 }
1999 nbs = 0;
2000 }
2001 }
2002
2003 sp += eo;
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002004 if (i == nm)
2005 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002006 if (eo == so) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002007 ds[di] = *sp++;
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002008 if (!ds[di++])
2009 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002010 }
2011 }
2012
2013 qrealloc(&ds, di + strlen(sp), &dssize);
2014 strcpy(ds + di, sp);
2015 setvar_p(dest, ds);
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002016 if (re == &sreg)
2017 regfree(re);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002018 return i;
2019}
2020
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002021static NOINLINE int do_mktime(const char *ds)
2022{
2023 struct tm then;
2024 int count;
2025
2026 /*memset(&then, 0, sizeof(then)); - not needed */
2027 then.tm_isdst = -1; /* default is unknown */
2028
2029 /* manpage of mktime says these fields are ints,
2030 * so we can sscanf stuff directly into them */
2031 count = sscanf(ds, "%u %u %u %u %u %u %d",
2032 &then.tm_year, &then.tm_mon, &then.tm_mday,
2033 &then.tm_hour, &then.tm_min, &then.tm_sec,
2034 &then.tm_isdst);
2035
2036 if (count < 6
2037 || (unsigned)then.tm_mon < 1
2038 || (unsigned)then.tm_year < 1900
2039 ) {
2040 return -1;
2041 }
2042
2043 then.tm_mon -= 1;
Denys Vlasenkobc3e9472009-09-21 04:16:00 +02002044 then.tm_year -= 1900;
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002045
2046 return mktime(&then);
2047}
2048
2049static NOINLINE var *exec_builtin(node *op, var *res)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002050{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002051#define tspl (G.exec_builtin__tspl)
2052
Glenn L McGrath545106f2002-11-11 06:21:00 +00002053 var *tv;
2054 node *an[4];
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002055 var *av[4];
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002056 const char *as[4];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002057 regmatch_t pmatch[2];
2058 regex_t sreg, *re;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002059 node *spl;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002060 uint32_t isr, info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002061 int nargs;
2062 time_t tt;
2063 char *s, *s1;
2064 int i, l, ll, n;
2065
2066 tv = nvalloc(4);
2067 isr = info = op->info;
2068 op = op->l.n;
2069
2070 av[2] = av[3] = NULL;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002071 for (i = 0; i < 4 && op; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002072 an[i] = nextarg(&op);
2073 if (isr & 0x09000000) av[i] = evaluate(an[i], &tv[i]);
2074 if (isr & 0x08000000) as[i] = getvar_s(av[i]);
2075 isr >>= 1;
2076 }
2077
2078 nargs = i;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +00002079 if ((uint32_t)nargs < (info >> 30))
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002080 syntax_error(EMSG_TOO_FEW_ARGS);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002081
Denys Vlasenko56b3eec2009-10-23 13:03:59 +02002082 info &= OPNMASK;
2083 switch (info) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002084
Denis Vlasenkof782f522007-01-01 23:51:30 +00002085 case B_a2:
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002086#if ENABLE_FEATURE_AWK_LIBM
Denis Vlasenko37890e22008-10-21 12:59:34 +00002087 setvar_i(res, atan2(getvar_i(av[0]), getvar_i(av[1])));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002088#else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002089 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002090#endif
2091 break;
2092
Denis Vlasenkof782f522007-01-01 23:51:30 +00002093 case B_sp:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002094 if (nargs > 2) {
2095 spl = (an[2]->info & OPCLSMASK) == OC_REGEXP ?
2096 an[2] : mk_splitter(getvar_s(evaluate(an[2], &tv[2])), &tspl);
2097 } else {
2098 spl = &fsplitter.n;
2099 }
2100
2101 n = awk_split(as[0], spl, &s);
2102 s1 = s;
2103 clear_array(iamarray(av[1]));
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002104 for (i = 1; i <= n; i++)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002105 setari_u(av[1], i, nextword(&s1));
2106 free(s);
2107 setvar_i(res, n);
2108 break;
2109
Denis Vlasenkof782f522007-01-01 23:51:30 +00002110 case B_ss:
Rob Landleya3896512006-05-07 20:20:34 +00002111 l = strlen(as[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002112 i = getvar_i(av[1]) - 1;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002113 if (i > l) i = l;
2114 if (i < 0) i = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002115 n = (nargs > 2) ? getvar_i(av[2]) : l-i;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002116 if (n < 0) n = 0;
Denis Vlasenko8ae5b282008-07-02 22:47:49 +00002117 s = xstrndup(as[0]+i, n);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002118 setvar_p(res, s);
2119 break;
Denis Vlasenkof7996f32007-01-11 17:20:00 +00002120
Denis Vlasenko7cbcd1c2008-08-28 23:16:58 +00002121 /* Bitwise ops must assume that operands are unsigned. GNU Awk 3.1.5:
2122 * awk '{ print or(-1,1) }' gives "4.29497e+09", not "-2.xxxe+09" */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002123 case B_an:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002124 setvar_i(res, getvar_i_int(av[0]) & getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002125 break;
Denis Vlasenkof7996f32007-01-11 17:20:00 +00002126
Denis Vlasenkof782f522007-01-01 23:51:30 +00002127 case B_co:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002128 setvar_i(res, ~getvar_i_int(av[0]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002129 break;
2130
Denis Vlasenkof782f522007-01-01 23:51:30 +00002131 case B_ls:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002132 setvar_i(res, getvar_i_int(av[0]) << getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002133 break;
2134
Denis Vlasenkof782f522007-01-01 23:51:30 +00002135 case B_or:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002136 setvar_i(res, getvar_i_int(av[0]) | getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002137 break;
2138
Denis Vlasenkof782f522007-01-01 23:51:30 +00002139 case B_rs:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002140 setvar_i(res, getvar_i_int(av[0]) >> getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002141 break;
2142
Denis Vlasenkof782f522007-01-01 23:51:30 +00002143 case B_xo:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002144 setvar_i(res, getvar_i_int(av[0]) ^ getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002145 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002146
Denis Vlasenkof782f522007-01-01 23:51:30 +00002147 case B_lo:
Denis Vlasenkof782f522007-01-01 23:51:30 +00002148 case B_up:
Rob Landleyd921b2e2006-08-03 15:41:12 +00002149 s1 = s = xstrdup(as[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002150 while (*s1) {
Denys Vlasenko56b3eec2009-10-23 13:03:59 +02002151 //*s1 = (info == B_up) ? toupper(*s1) : tolower(*s1);
2152 if ((unsigned char)((*s1 | 0x20) - 'a') <= ('z' - 'a'))
2153 *s1 = (info == B_up) ? (*s1 & 0xdf) : (*s1 | 0x20);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002154 s1++;
2155 }
2156 setvar_p(res, s);
2157 break;
2158
Denis Vlasenkof782f522007-01-01 23:51:30 +00002159 case B_ix:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002160 n = 0;
Rob Landleya3896512006-05-07 20:20:34 +00002161 ll = strlen(as[1]);
2162 l = strlen(as[0]) - ll;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002163 if (ll > 0 && l >= 0) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002164 if (!icase) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002165 s = strstr(as[0], as[1]);
2166 if (s) n = (s - as[0]) + 1;
2167 } else {
2168 /* this piece of code is terribly slow and
2169 * really should be rewritten
2170 */
2171 for (i=0; i<=l; i++) {
2172 if (strncasecmp(as[0]+i, as[1], ll) == 0) {
2173 n = i+1;
2174 break;
2175 }
2176 }
2177 }
2178 }
2179 setvar_i(res, n);
2180 break;
2181
Denis Vlasenkof782f522007-01-01 23:51:30 +00002182 case B_ti:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002183 if (nargs > 1)
2184 tt = getvar_i(av[1]);
2185 else
2186 time(&tt);
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002187 //s = (nargs > 0) ? as[0] : "%a %b %d %H:%M:%S %Z %Y";
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002188 i = strftime(g_buf, MAXVARFMT,
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002189 ((nargs > 0) ? as[0] : "%a %b %d %H:%M:%S %Z %Y"),
2190 localtime(&tt));
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002191 g_buf[i] = '\0';
2192 setvar_s(res, g_buf);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002193 break;
2194
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002195 case B_mt:
2196 setvar_i(res, do_mktime(as[0]));
2197 break;
2198
Denis Vlasenkof782f522007-01-01 23:51:30 +00002199 case B_ma:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002200 re = as_regex(an[1], &sreg);
2201 n = regexec(re, as[0], 1, pmatch, 0);
2202 if (n == 0) {
2203 pmatch[0].rm_so++;
2204 pmatch[0].rm_eo++;
2205 } else {
2206 pmatch[0].rm_so = 0;
2207 pmatch[0].rm_eo = -1;
2208 }
2209 setvar_i(newvar("RSTART"), pmatch[0].rm_so);
2210 setvar_i(newvar("RLENGTH"), pmatch[0].rm_eo - pmatch[0].rm_so);
2211 setvar_i(res, pmatch[0].rm_so);
2212 if (re == &sreg) regfree(re);
2213 break;
2214
Denis Vlasenkof782f522007-01-01 23:51:30 +00002215 case B_ge:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002216 awk_sub(an[0], as[1], getvar_i(av[2]), av[3], res, TRUE);
2217 break;
2218
Denis Vlasenkof782f522007-01-01 23:51:30 +00002219 case B_gs:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002220 setvar_i(res, awk_sub(an[0], as[1], 0, av[2], av[2], FALSE));
2221 break;
2222
Denis Vlasenkof782f522007-01-01 23:51:30 +00002223 case B_su:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002224 setvar_i(res, awk_sub(an[0], as[1], 1, av[2], av[2], FALSE));
2225 break;
2226 }
2227
2228 nvfree(tv);
2229 return res;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002230#undef tspl
Glenn L McGrath545106f2002-11-11 06:21:00 +00002231}
2232
2233/*
2234 * Evaluate node - the heart of the program. Supplied with subtree
2235 * and place where to store result. returns ptr to result.
2236 */
2237#define XC(n) ((n) >> 8)
2238
Mike Frysinger10a11e22005-09-27 02:23:02 +00002239static var *evaluate(node *op, var *res)
2240{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002241/* This procedure is recursive so we should count every byte */
2242#define fnargs (G.evaluate__fnargs)
2243/* seed is initialized to 1 */
2244#define seed (G.evaluate__seed)
2245#define sreg (G.evaluate__sreg)
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002246
Glenn L McGrath545106f2002-11-11 06:21:00 +00002247 node *op1;
2248 var *v1;
2249 union {
2250 var *v;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002251 const char *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002252 double d;
2253 int i;
2254 } L, R;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002255 uint32_t opinfo;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002256 int opn;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002257 union {
2258 char *s;
2259 rstream *rsm;
2260 FILE *F;
2261 var *v;
2262 regex_t *re;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002263 uint32_t info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002264 } X;
2265
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002266 if (!op)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002267 return setvar_s(res, NULL);
2268
2269 v1 = nvalloc(2);
2270
2271 while (op) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002272 opinfo = op->info;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002273 opn = (opinfo & OPNMASK);
2274 g_lineno = op->lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002275
Mike Frysingerde2b9382005-09-27 03:18:00 +00002276 /* execute inevitable things */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002277 op1 = op->l.n;
2278 if (opinfo & OF_RES1) X.v = L.v = evaluate(op1, v1);
2279 if (opinfo & OF_RES2) R.v = evaluate(op->r.n, v1+1);
2280 if (opinfo & OF_STR1) L.s = getvar_s(L.v);
2281 if (opinfo & OF_STR2) R.s = getvar_s(R.v);
2282 if (opinfo & OF_NUM1) L.d = getvar_i(L.v);
2283
2284 switch (XC(opinfo & OPCLSMASK)) {
2285
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002286 /* -- iterative node type -- */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002287
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002288 /* test pattern */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002289 case XC( OC_TEST ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002290 if ((op1->info & OPCLSMASK) == OC_COMMA) {
2291 /* it's range pattern */
2292 if ((opinfo & OF_CHECKED) || ptest(op1->l.n)) {
2293 op->info |= OF_CHECKED;
2294 if (ptest(op1->r.n))
2295 op->info &= ~OF_CHECKED;
2296
2297 op = op->a.n;
2298 } else {
2299 op = op->r.n;
2300 }
2301 } else {
2302 op = (ptest(op1)) ? op->a.n : op->r.n;
2303 }
2304 break;
2305
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002306 /* just evaluate an expression, also used as unconditional jump */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002307 case XC( OC_EXEC ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002308 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002309
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002310 /* branch, used in if-else and various loops */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002311 case XC( OC_BR ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002312 op = istrue(L.v) ? op->a.n : op->r.n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002313 break;
2314
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002315 /* initialize for-in loop */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002316 case XC( OC_WALKINIT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002317 hashwalk_init(L.v, iamarray(R.v));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002318 break;
2319
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002320 /* get next array item */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002321 case XC( OC_WALKNEXT ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002322 op = hashwalk_next(L.v) ? op->a.n : op->r.n;
2323 break;
2324
Denis Vlasenkof782f522007-01-01 23:51:30 +00002325 case XC( OC_PRINT ):
2326 case XC( OC_PRINTF ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002327 X.F = stdout;
Mike Frysingerde2b9382005-09-27 03:18:00 +00002328 if (op->r.n) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002329 X.rsm = newfile(R.s);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002330 if (!X.rsm->F) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002331 if (opn == '|') {
Denis Vlasenko51742f42007-04-12 00:32:05 +00002332 X.rsm->F = popen(R.s, "w");
2333 if (X.rsm->F == NULL)
Manuel Novoa III cad53642003-03-19 09:13:01 +00002334 bb_perror_msg_and_die("popen");
Glenn L McGrath545106f2002-11-11 06:21:00 +00002335 X.rsm->is_pipe = 1;
2336 } else {
Rob Landleyd921b2e2006-08-03 15:41:12 +00002337 X.rsm->F = xfopen(R.s, opn=='w' ? "w" : "a");
Glenn L McGrath545106f2002-11-11 06:21:00 +00002338 }
2339 }
2340 X.F = X.rsm->F;
2341 }
2342
2343 if ((opinfo & OPCLSMASK) == OC_PRINT) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002344 if (!op1) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002345 fputs(getvar_s(intvar[F0]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002346 } else {
2347 while (op1) {
2348 L.v = evaluate(nextarg(&op1), v1);
2349 if (L.v->type & VF_NUMBER) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002350 fmt_num(g_buf, MAXVARFMT, getvar_s(intvar[OFMT]),
Denis Vlasenkob54b2082006-10-27 09:05:40 +00002351 getvar_i(L.v), TRUE);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002352 fputs(g_buf, X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002353 } else {
2354 fputs(getvar_s(L.v), X.F);
2355 }
2356
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002357 if (op1) fputs(getvar_s(intvar[OFS]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002358 }
2359 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002360 fputs(getvar_s(intvar[ORS]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002361
2362 } else { /* OC_PRINTF */
2363 L.s = awk_printf(op1);
2364 fputs(L.s, X.F);
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002365 free((char*)L.s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002366 }
2367 fflush(X.F);
2368 break;
2369
Denis Vlasenkof782f522007-01-01 23:51:30 +00002370 case XC( OC_DELETE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002371 X.info = op1->info & OPCLSMASK;
2372 if (X.info == OC_VAR) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002373 R.v = op1->l.v;
2374 } else if (X.info == OC_FNARG) {
2375 R.v = &fnargs[op1->l.i];
2376 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002377 syntax_error(EMSG_NOT_ARRAY);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002378 }
2379
Mike Frysingerde2b9382005-09-27 03:18:00 +00002380 if (op1->r.n) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002381 clrvar(L.v);
2382 L.s = getvar_s(evaluate(op1->r.n, v1));
2383 hash_remove(iamarray(R.v), L.s);
2384 } else {
2385 clear_array(iamarray(R.v));
2386 }
2387 break;
2388
Denis Vlasenkof782f522007-01-01 23:51:30 +00002389 case XC( OC_NEWSOURCE ):
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002390 g_progname = op->l.s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002391 break;
2392
Denis Vlasenkof782f522007-01-01 23:51:30 +00002393 case XC( OC_RETURN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002394 copyvar(res, L.v);
2395 break;
2396
Denis Vlasenkof782f522007-01-01 23:51:30 +00002397 case XC( OC_NEXTFILE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002398 nextfile = TRUE;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002399 case XC( OC_NEXT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002400 nextrec = TRUE;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002401 case XC( OC_DONE ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002402 clrvar(res);
2403 break;
2404
Denis Vlasenkof782f522007-01-01 23:51:30 +00002405 case XC( OC_EXIT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002406 awk_exit(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002407
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002408 /* -- recursive node type -- */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002409
Denis Vlasenkof782f522007-01-01 23:51:30 +00002410 case XC( OC_VAR ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002411 L.v = op->l.v;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002412 if (L.v == intvar[NF])
Glenn L McGrath545106f2002-11-11 06:21:00 +00002413 split_f0();
2414 goto v_cont;
2415
Denis Vlasenkof782f522007-01-01 23:51:30 +00002416 case XC( OC_FNARG ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002417 L.v = &fnargs[op->l.i];
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002418 v_cont:
2419 res = op->r.n ? findvar(iamarray(L.v), R.s) : L.v;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002420 break;
2421
Denis Vlasenkof782f522007-01-01 23:51:30 +00002422 case XC( OC_IN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002423 setvar_i(res, hash_search(iamarray(R.v), L.s) ? 1 : 0);
2424 break;
2425
Denis Vlasenkof782f522007-01-01 23:51:30 +00002426 case XC( OC_REGEXP ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002427 op1 = op;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002428 L.s = getvar_s(intvar[F0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002429 goto re_cont;
2430
Denis Vlasenkof782f522007-01-01 23:51:30 +00002431 case XC( OC_MATCH ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002432 op1 = op->r.n;
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002433 re_cont:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002434 X.re = as_regex(op1, &sreg);
2435 R.i = regexec(X.re, L.s, 0, NULL, 0);
2436 if (X.re == &sreg) regfree(X.re);
Denys Vlasenko12847742009-11-30 01:15:04 +01002437 setvar_i(res, (R.i == 0) ^ (opn == '!'));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002438 break;
2439
Denis Vlasenkof782f522007-01-01 23:51:30 +00002440 case XC( OC_MOVE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002441 /* if source is a temporary string, jusk relink it to dest */
Denys Vlasenko12847742009-11-30 01:15:04 +01002442//Disabled: if R.v is numeric but happens to have cached R.v->string,
2443//then L.v ends up being a string, which is wrong
2444// if (R.v == v1+1 && R.v->string) {
2445// res = setvar_p(L.v, R.v->string);
2446// R.v->string = NULL;
2447// } else {
Mike Frysingerde2b9382005-09-27 03:18:00 +00002448 res = copyvar(L.v, R.v);
Denys Vlasenko12847742009-11-30 01:15:04 +01002449// }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002450 break;
2451
Denis Vlasenkof782f522007-01-01 23:51:30 +00002452 case XC( OC_TERNARY ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002453 if ((op->r.n->info & OPCLSMASK) != OC_COLON)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002454 syntax_error(EMSG_POSSIBLE_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002455 res = evaluate(istrue(L.v) ? op->r.n->l.n : op->r.n->r.n, res);
2456 break;
2457
Denis Vlasenkof782f522007-01-01 23:51:30 +00002458 case XC( OC_FUNC ):
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002459 if (!op->r.f->body.first)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002460 syntax_error(EMSG_UNDEF_FUNC);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002461
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002462 X.v = R.v = nvalloc(op->r.f->nargs + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002463 while (op1) {
2464 L.v = evaluate(nextarg(&op1), v1);
2465 copyvar(R.v, L.v);
2466 R.v->type |= VF_CHILD;
2467 R.v->x.parent = L.v;
2468 if (++R.v - X.v >= op->r.f->nargs)
2469 break;
2470 }
2471
2472 R.v = fnargs;
2473 fnargs = X.v;
2474
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002475 L.s = g_progname;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002476 res = evaluate(op->r.f->body.first, res);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002477 g_progname = L.s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002478
2479 nvfree(fnargs);
2480 fnargs = R.v;
2481 break;
2482
Denis Vlasenkof782f522007-01-01 23:51:30 +00002483 case XC( OC_GETLINE ):
2484 case XC( OC_PGETLINE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002485 if (op1) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002486 X.rsm = newfile(L.s);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002487 if (!X.rsm->F) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002488 if ((opinfo & OPCLSMASK) == OC_PGETLINE) {
2489 X.rsm->F = popen(L.s, "r");
2490 X.rsm->is_pipe = TRUE;
2491 } else {
Denis Vlasenko5415c852008-07-21 23:05:26 +00002492 X.rsm->F = fopen_for_read(L.s); /* not xfopen! */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002493 }
2494 }
2495 } else {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002496 if (!iF) iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002497 X.rsm = iF;
2498 }
2499
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002500 if (!X.rsm->F) {
2501 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002502 setvar_i(res, -1);
2503 break;
2504 }
2505
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002506 if (!op->r.n)
2507 R.v = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002508
2509 L.i = awk_getline(X.rsm, R.v);
2510 if (L.i > 0) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002511 if (!op1) {
2512 incvar(intvar[FNR]);
2513 incvar(intvar[NR]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002514 }
2515 }
2516 setvar_i(res, L.i);
2517 break;
2518
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002519 /* simple builtins */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002520 case XC( OC_FBLTIN ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002521 switch (opn) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002522
Denis Vlasenkof782f522007-01-01 23:51:30 +00002523 case F_in:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002524 R.d = (int)L.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002525 break;
2526
Denis Vlasenkof782f522007-01-01 23:51:30 +00002527 case F_rn:
2528 R.d = (double)rand() / (double)RAND_MAX;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002529 break;
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002530#if ENABLE_FEATURE_AWK_LIBM
Denis Vlasenkof782f522007-01-01 23:51:30 +00002531 case F_co:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002532 R.d = cos(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002533 break;
2534
Denis Vlasenkof782f522007-01-01 23:51:30 +00002535 case F_ex:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002536 R.d = exp(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002537 break;
2538
Denis Vlasenkof782f522007-01-01 23:51:30 +00002539 case F_lg:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002540 R.d = log(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002541 break;
2542
Denis Vlasenkof782f522007-01-01 23:51:30 +00002543 case F_si:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002544 R.d = sin(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002545 break;
2546
Denis Vlasenkof782f522007-01-01 23:51:30 +00002547 case F_sq:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002548 R.d = sqrt(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002549 break;
2550#else
Denis Vlasenkof782f522007-01-01 23:51:30 +00002551 case F_co:
2552 case F_ex:
2553 case F_lg:
2554 case F_si:
2555 case F_sq:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002556 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002557 break;
2558#endif
Denis Vlasenkof782f522007-01-01 23:51:30 +00002559 case F_sr:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002560 R.d = (double)seed;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002561 seed = op1 ? (unsigned)L.d : (unsigned)time(NULL);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002562 srand(seed);
2563 break;
2564
Denis Vlasenkof782f522007-01-01 23:51:30 +00002565 case F_ti:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002566 R.d = time(NULL);
2567 break;
2568
Denis Vlasenkof782f522007-01-01 23:51:30 +00002569 case F_le:
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002570 if (!op1)
2571 L.s = getvar_s(intvar[F0]);
Rob Landleya3896512006-05-07 20:20:34 +00002572 R.d = strlen(L.s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002573 break;
2574
Denis Vlasenkof782f522007-01-01 23:51:30 +00002575 case F_sy:
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002576 fflush_all();
Denis Vlasenko249fabf2006-12-19 00:29:22 +00002577 R.d = (ENABLE_FEATURE_ALLOW_EXEC && L.s && *L.s)
2578 ? (system(L.s) >> 8) : 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002579 break;
2580
Denis Vlasenkof782f522007-01-01 23:51:30 +00002581 case F_ff:
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002582 if (!op1)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002583 fflush(stdout);
2584 else {
2585 if (L.s && *L.s) {
2586 X.rsm = newfile(L.s);
2587 fflush(X.rsm->F);
2588 } else {
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002589 fflush_all();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002590 }
2591 }
2592 break;
2593
Denis Vlasenkof782f522007-01-01 23:51:30 +00002594 case F_cl:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002595 X.rsm = (rstream *)hash_search(fdhash, L.s);
2596 if (X.rsm) {
2597 R.i = X.rsm->is_pipe ? pclose(X.rsm->F) : fclose(X.rsm->F);
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00002598 free(X.rsm->buffer);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002599 hash_remove(fdhash, L.s);
2600 }
2601 if (R.i != 0)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002602 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002603 R.d = (double)R.i;
2604 break;
2605 }
2606 setvar_i(res, R.d);
2607 break;
2608
Denis Vlasenkof782f522007-01-01 23:51:30 +00002609 case XC( OC_BUILTIN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002610 res = exec_builtin(op, res);
2611 break;
2612
Denis Vlasenkof782f522007-01-01 23:51:30 +00002613 case XC( OC_SPRINTF ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002614 setvar_p(res, awk_printf(op1));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002615 break;
2616
Denis Vlasenkof782f522007-01-01 23:51:30 +00002617 case XC( OC_UNARY ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002618 X.v = R.v;
2619 L.d = R.d = getvar_i(R.v);
2620 switch (opn) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002621 case 'P':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002622 L.d = ++R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002623 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002624 case 'p':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002625 R.d++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002626 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002627 case 'M':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002628 L.d = --R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002629 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002630 case 'm':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002631 R.d--;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002632 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002633 case '!':
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002634 L.d = !istrue(X.v);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002635 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002636 case '-':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002637 L.d = -R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002638 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002639 r_op_change:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002640 setvar_i(X.v, R.d);
2641 }
2642 setvar_i(res, L.d);
2643 break;
2644
Denis Vlasenkof782f522007-01-01 23:51:30 +00002645 case XC( OC_FIELD ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002646 R.i = (int)getvar_i(R.v);
2647 if (R.i == 0) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002648 res = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002649 } else {
2650 split_f0();
2651 if (R.i > nfields)
2652 fsrealloc(R.i);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002653 res = &Fields[R.i - 1];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002654 }
2655 break;
2656
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002657 /* concatenation (" ") and index joining (",") */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002658 case XC( OC_CONCAT ):
2659 case XC( OC_COMMA ):
Rob Landleya3896512006-05-07 20:20:34 +00002660 opn = strlen(L.s) + strlen(R.s) + 2;
Denis Vlasenkob95636c2006-12-19 23:36:04 +00002661 X.s = xmalloc(opn);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002662 strcpy(X.s, L.s);
2663 if ((opinfo & OPCLSMASK) == OC_COMMA) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002664 L.s = getvar_s(intvar[SUBSEP]);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002665 X.s = xrealloc(X.s, opn + strlen(L.s));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002666 strcat(X.s, L.s);
2667 }
2668 strcat(X.s, R.s);
2669 setvar_p(res, X.s);
2670 break;
2671
Denis Vlasenkof782f522007-01-01 23:51:30 +00002672 case XC( OC_LAND ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002673 setvar_i(res, istrue(L.v) ? ptest(op->r.n) : 0);
2674 break;
2675
Denis Vlasenkof782f522007-01-01 23:51:30 +00002676 case XC( OC_LOR ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002677 setvar_i(res, istrue(L.v) ? 1 : ptest(op->r.n));
2678 break;
2679
Denis Vlasenkof782f522007-01-01 23:51:30 +00002680 case XC( OC_BINARY ):
2681 case XC( OC_REPLACE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002682 R.d = getvar_i(R.v);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002683 switch (opn) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002684 case '+':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002685 L.d += R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002686 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002687 case '-':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002688 L.d -= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002689 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002690 case '*':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002691 L.d *= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002692 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002693 case '/':
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002694 if (R.d == 0)
2695 syntax_error(EMSG_DIV_BY_ZERO);
Mike Frysingerde2b9382005-09-27 03:18:00 +00002696 L.d /= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002697 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002698 case '&':
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002699#if ENABLE_FEATURE_AWK_LIBM
Mike Frysingerde2b9382005-09-27 03:18:00 +00002700 L.d = pow(L.d, R.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002701#else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002702 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002703#endif
2704 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002705 case '%':
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002706 if (R.d == 0)
2707 syntax_error(EMSG_DIV_BY_ZERO);
Mike Frysingerde2b9382005-09-27 03:18:00 +00002708 L.d -= (int)(L.d / R.d) * R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002709 break;
2710 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002711 res = setvar_i(((opinfo & OPCLSMASK) == OC_BINARY) ? res : X.v, L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002712 break;
2713
Denis Vlasenkof782f522007-01-01 23:51:30 +00002714 case XC( OC_COMPARE ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002715 if (is_numeric(L.v) && is_numeric(R.v)) {
2716 L.d = getvar_i(L.v) - getvar_i(R.v);
2717 } else {
2718 L.s = getvar_s(L.v);
2719 R.s = getvar_s(R.v);
2720 L.d = icase ? strcasecmp(L.s, R.s) : strcmp(L.s, R.s);
2721 }
2722 switch (opn & 0xfe) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002723 case 0:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002724 R.i = (L.d > 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002725 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002726 case 2:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002727 R.i = (L.d >= 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002728 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002729 case 4:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002730 R.i = (L.d == 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002731 break;
2732 }
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002733 setvar_i(res, (opn & 1 ? R.i : !R.i) ? 1 : 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002734 break;
2735
Denis Vlasenkof782f522007-01-01 23:51:30 +00002736 default:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002737 syntax_error(EMSG_POSSIBLE_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002738 }
2739 if ((opinfo & OPCLSMASK) <= SHIFT_TIL_THIS)
2740 op = op->a.n;
2741 if ((opinfo & OPCLSMASK) >= RECUR_FROM_THIS)
2742 break;
2743 if (nextrec)
2744 break;
2745 }
2746 nvfree(v1);
2747 return res;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002748#undef fnargs
2749#undef seed
2750#undef sreg
Glenn L McGrath545106f2002-11-11 06:21:00 +00002751}
2752
2753
2754/* -------- main & co. -------- */
2755
Mike Frysinger10a11e22005-09-27 02:23:02 +00002756static int awk_exit(int r)
2757{
Denis Vlasenkof782f522007-01-01 23:51:30 +00002758 var tv;
2759 unsigned i;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002760 hash_item *hi;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002761
Denis Vlasenkof782f522007-01-01 23:51:30 +00002762 zero_out_var(&tv);
2763
2764 if (!exiting) {
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002765 exiting = TRUE;
Glenn L McGrathca29ffc2004-09-24 09:24:27 +00002766 nextrec = FALSE;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002767 evaluate(endseq.first, &tv);
2768 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002769
2770 /* waiting for children */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002771 for (i = 0; i < fdhash->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002772 hi = fdhash->items[i];
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00002773 while (hi) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002774 if (hi->data.rs.F && hi->data.rs.is_pipe)
2775 pclose(hi->data.rs.F);
2776 hi = hi->next;
2777 }
2778 }
2779
2780 exit(r);
2781}
2782
2783/* if expr looks like "var=value", perform assignment and return 1,
2784 * otherwise return 0 */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00002785static int is_assignment(const char *expr)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002786{
Glenn L McGrath545106f2002-11-11 06:21:00 +00002787 char *exprc, *s, *s0, *s1;
2788
Rob Landleyd921b2e2006-08-03 15:41:12 +00002789 exprc = xstrdup(expr);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002790 if (!isalnum_(*exprc) || (s = strchr(exprc, '=')) == NULL) {
2791 free(exprc);
2792 return FALSE;
2793 }
2794
2795 *(s++) = '\0';
2796 s0 = s1 = s;
2797 while (*s)
2798 *(s1++) = nextchar(&s);
2799
2800 *s1 = '\0';
2801 setvar_u(newvar(exprc), s0);
2802 free(exprc);
2803 return TRUE;
2804}
2805
2806/* switch to next input file */
Mike Frysinger10a11e22005-09-27 02:23:02 +00002807static rstream *next_input_file(void)
2808{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002809#define rsm (G.next_input_file__rsm)
2810#define files_happen (G.next_input_file__files_happen)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002811
Glenn L McGrath545106f2002-11-11 06:21:00 +00002812 FILE *F = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002813 const char *fname, *ind;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002814
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002815 if (rsm.F)
2816 fclose(rsm.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002817 rsm.F = NULL;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002818 rsm.pos = rsm.adv = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002819
2820 do {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002821 if (getvar_i(intvar[ARGIND])+1 >= getvar_i(intvar[ARGC])) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002822 if (files_happen)
2823 return NULL;
2824 fname = "-";
2825 F = stdin;
2826 } else {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002827 ind = getvar_s(incvar(intvar[ARGIND]));
2828 fname = getvar_s(findvar(iamarray(intvar[ARGV]), ind));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002829 if (fname && *fname && !is_assignment(fname))
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002830 F = xfopen_stdin(fname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002831 }
2832 } while (!F);
2833
2834 files_happen = TRUE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002835 setvar_s(intvar[FILENAME], fname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002836 rsm.F = F;
2837 return &rsm;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002838#undef rsm
2839#undef files_happen
Glenn L McGrath545106f2002-11-11 06:21:00 +00002840}
2841
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00002842int awk_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Rob Landleydfba7412006-03-06 20:47:33 +00002843int awk_main(int argc, char **argv)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002844{
Denis Vlasenko67b23e62006-10-03 21:00:06 +00002845 unsigned opt;
Denis Vlasenkobe644a82007-03-10 17:22:14 +00002846 char *opt_F, *opt_W;
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002847 llist_t *list_v = NULL;
2848 llist_t *list_f = NULL;
2849 int i, j;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002850 var *v;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002851 var tv;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002852 char **envp;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002853 char *vnames = (char *)vNames; /* cheat */
2854 char *vvalues = (char *)vValues;
2855
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002856 INIT_G();
2857
Denis Vlasenko150f4022007-01-13 21:06:21 +00002858 /* Undo busybox.c, or else strtod may eat ','! This breaks parsing:
Denis Vlasenko6dc6ebb2007-01-01 23:53:12 +00002859 * $1,$2 == '$1,' '$2', NOT '$1' ',' '$2' */
2860 if (ENABLE_LOCALE_SUPPORT)
2861 setlocale(LC_NUMERIC, "C");
2862
Denis Vlasenkof782f522007-01-01 23:51:30 +00002863 zero_out_var(&tv);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002864
2865 /* allocate global buffer */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002866 g_buf = xmalloc(MAXVARFMT + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002867
2868 vhash = hash_init();
2869 ahash = hash_init();
2870 fdhash = hash_init();
2871 fnhash = hash_init();
2872
2873 /* initialize variables */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002874 for (i = 0; *vnames; i++) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002875 intvar[i] = v = newvar(nextword(&vnames));
Denis Vlasenkof782f522007-01-01 23:51:30 +00002876 if (*vvalues != '\377')
2877 setvar_s(v, nextword(&vvalues));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002878 else
2879 setvar_i(v, 0);
2880
Denis Vlasenkof782f522007-01-01 23:51:30 +00002881 if (*vnames == '*') {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002882 v->type |= VF_SPECIAL;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002883 vnames++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002884 }
2885 }
2886
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002887 handle_special(intvar[FS]);
2888 handle_special(intvar[RS]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002889
Denis Vlasenkof782f522007-01-01 23:51:30 +00002890 newfile("/dev/stdin")->F = stdin;
2891 newfile("/dev/stdout")->F = stdout;
2892 newfile("/dev/stderr")->F = stderr;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002893
Denis Vlasenkof71d9162007-05-03 22:57:56 +00002894 /* Huh, people report that sometimes environ is NULL. Oh well. */
2895 if (environ) for (envp = environ; *envp; envp++) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002896 /* environ is writable, thus we don't strdup it needlessly */
2897 char *s = *envp;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002898 char *s1 = strchr(s, '=');
2899 if (s1) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002900 *s1 = '\0';
2901 /* Both findvar and setvar_u take const char*
2902 * as 2nd arg -> environment is not trashed */
2903 setvar_u(findvar(iamarray(intvar[ENVIRON]), s), s1 + 1);
2904 *s1 = '=';
Eric Andersen67776be2004-07-30 23:52:08 +00002905 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002906 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002907 opt_complementary = "v::f::"; /* -v and -f can occur multiple times */
2908 opt = getopt32(argv, "F:v:f:W:", &opt_F, &list_v, &list_f, &opt_W);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002909 argv += optind;
2910 argc -= optind;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002911 if (opt & 0x1)
2912 setvar_s(intvar[FS], opt_F); // -F
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002913 while (list_v) { /* -v */
2914 if (!is_assignment(llist_pop(&list_v)))
Denis Vlasenkobe644a82007-03-10 17:22:14 +00002915 bb_show_usage();
2916 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002917 if (list_f) { /* -f */
2918 do {
2919 char *s = NULL;
2920 FILE *from_file;
2921
2922 g_progname = llist_pop(&list_f);
2923 from_file = xfopen_stdin(g_progname);
2924 /* one byte is reserved for some trick in next_token */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002925 for (i = j = 1; j > 0; i += j) {
2926 s = xrealloc(s, i + 4096);
2927 j = fread(s + i, 1, 4094, from_file);
Denis Vlasenko099efbf2006-09-22 09:02:30 +00002928 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002929 s[i] = '\0';
2930 fclose(from_file);
2931 parse_program(s + 1);
2932 free(s);
2933 } while (list_f);
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +00002934 argc++;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002935 } else { // no -f: take program from 1st parameter
2936 if (!argc)
2937 bb_show_usage();
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002938 g_progname = "cmd. line";
Denis Vlasenkof782f522007-01-01 23:51:30 +00002939 parse_program(*argv++);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002940 }
Denis Vlasenko099efbf2006-09-22 09:02:30 +00002941 if (opt & 0x8) // -W
Denis Vlasenkoe1a0d482006-10-20 13:28:22 +00002942 bb_error_msg("warning: unrecognized option '-W %s' ignored", opt_W);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002943
Glenn L McGrath545106f2002-11-11 06:21:00 +00002944 /* fill in ARGV array */
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +00002945 setvar_i(intvar[ARGC], argc);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002946 setari_u(intvar[ARGV], 0, "awk");
Denis Vlasenkof782f522007-01-01 23:51:30 +00002947 i = 0;
2948 while (*argv)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002949 setari_u(intvar[ARGV], ++i, *argv++);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002950
2951 evaluate(beginseq.first, &tv);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002952 if (!mainseq.first && !endseq.first)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002953 awk_exit(EXIT_SUCCESS);
2954
2955 /* input file could already be opened in BEGIN block */
Denys Vlasenkocdeda162009-11-30 01:14:16 +01002956 if (!iF)
2957 iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002958
2959 /* passing through input files */
2960 while (iF) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002961 nextfile = FALSE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002962 setvar_i(intvar[FNR], 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002963
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002964 while ((i = awk_getline(iF, intvar[F0])) > 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002965 nextrec = FALSE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002966 incvar(intvar[NR]);
2967 incvar(intvar[FNR]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002968 evaluate(mainseq.first, &tv);
2969
2970 if (nextfile)
2971 break;
2972 }
2973
Denis Vlasenkof782f522007-01-01 23:51:30 +00002974 if (i < 0)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002975 syntax_error(strerror(errno));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002976
2977 iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002978 }
2979
Glenn L McGrath545106f2002-11-11 06:21:00 +00002980 awk_exit(EXIT_SUCCESS);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002981 /*return 0;*/
Glenn L McGrath545106f2002-11-11 06:21:00 +00002982}