blob: ce6c9b726d80c4cefb74d48dc4e79ecbe69a442f [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;
447};
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
Denis Vlasenkof782f522007-01-01 23:51:30 +0000519static void zero_out_var(var * vp)
520{
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
Denis Vlasenkof782f522007-01-01 23:51:30 +0000536 while (*name) idx = *name++ + (idx << 6) - idx;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000537 return idx;
538}
539
540/* create new hash */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000541static xhash *hash_init(void)
542{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000543 xhash *newhash;
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000544
Denis Vlasenko4cccc032006-12-22 18:37:07 +0000545 newhash = xzalloc(sizeof(xhash));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000546 newhash->csize = FIRST_PRIME;
Denis Vlasenko4cccc032006-12-22 18:37:07 +0000547 newhash->items = xzalloc(newhash->csize * sizeof(hash_item *));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000548
549 return newhash;
550}
551
552/* find item in hash, return ptr to data, NULL if not found */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000553static void *hash_search(xhash *hash, const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000554{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000555 hash_item *hi;
556
557 hi = hash->items [ hashidx(name) % hash->csize ];
558 while (hi) {
559 if (strcmp(hi->name, name) == 0)
560 return &(hi->data);
561 hi = hi->next;
562 }
563 return NULL;
564}
565
566/* grow hash if it becomes too big */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000567static void hash_rebuild(xhash *hash)
568{
Denis Vlasenkof782f522007-01-01 23:51:30 +0000569 unsigned newsize, i, idx;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000570 hash_item **newitems, *hi, *thi;
571
Denis Vlasenko80b8b392007-06-25 10:55:35 +0000572 if (hash->nprime == ARRAY_SIZE(PRIMES))
Glenn L McGrath545106f2002-11-11 06:21:00 +0000573 return;
574
575 newsize = PRIMES[hash->nprime++];
Denis Vlasenko4cccc032006-12-22 18:37:07 +0000576 newitems = xzalloc(newsize * sizeof(hash_item *));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000577
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000578 for (i = 0; i < hash->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000579 hi = hash->items[i];
580 while (hi) {
581 thi = hi;
582 hi = thi->next;
583 idx = hashidx(thi->name) % newsize;
584 thi->next = newitems[idx];
585 newitems[idx] = thi;
586 }
587 }
588
589 free(hash->items);
590 hash->csize = newsize;
591 hash->items = newitems;
592}
593
594/* find item in hash, add it if necessary. Return ptr to data */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000595static void *hash_find(xhash *hash, const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000596{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000597 hash_item *hi;
Denis Vlasenkof782f522007-01-01 23:51:30 +0000598 unsigned idx;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000599 int l;
600
601 hi = hash_search(hash, name);
Denis Vlasenkob78c7822007-07-18 18:31:11 +0000602 if (!hi) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000603 if (++hash->nel / hash->csize > 10)
604 hash_rebuild(hash);
605
Rob Landleya3896512006-05-07 20:20:34 +0000606 l = strlen(name) + 1;
Denis Vlasenko7a676642009-03-15 22:20:31 +0000607 hi = xzalloc(sizeof(*hi) + l);
608 strcpy(hi->name, name);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000609
610 idx = hashidx(name) % hash->csize;
611 hi->next = hash->items[idx];
612 hash->items[idx] = hi;
613 hash->glen += l;
614 }
615 return &(hi->data);
616}
617
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000618#define findvar(hash, name) ((var*) hash_find((hash), (name)))
619#define newvar(name) ((var*) hash_find(vhash, (name)))
620#define newfile(name) ((rstream*)hash_find(fdhash, (name)))
621#define newfunc(name) ((func*) hash_find(fnhash, (name)))
Glenn L McGrath545106f2002-11-11 06:21:00 +0000622
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000623static void hash_remove(xhash *hash, const char *name)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000624{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000625 hash_item *hi, **phi;
626
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000627 phi = &(hash->items[hashidx(name) % hash->csize]);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000628 while (*phi) {
629 hi = *phi;
630 if (strcmp(hi->name, name) == 0) {
Rob Landleya3896512006-05-07 20:20:34 +0000631 hash->glen -= (strlen(name) + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000632 hash->nel--;
633 *phi = hi->next;
634 free(hi);
635 break;
636 }
637 phi = &(hi->next);
638 }
639}
640
641/* ------ some useful functions ------ */
642
Mike Frysinger10a11e22005-09-27 02:23:02 +0000643static void skip_spaces(char **s)
644{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000645 char *p = *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000646
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000647 while (1) {
648 if (*p == '\\' && p[1] == '\n') {
649 p++;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000650 t_lineno++;
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000651 } else if (*p != ' ' && *p != '\t') {
652 break;
653 }
Mike Frysingerde2b9382005-09-27 03:18:00 +0000654 p++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000655 }
656 *s = p;
657}
658
Mike Frysinger10a11e22005-09-27 02:23:02 +0000659static char *nextword(char **s)
660{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000661 char *p = *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000662
Denis Vlasenkof782f522007-01-01 23:51:30 +0000663 while (*(*s)++) /* */;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000664
665 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;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000674 if (c == '\\') c = bb_process_escape_sequence((const char**)s);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000675 if (c == '\\' && *s == pps) c = *((*s)++);
676 return c;
677}
678
Denis Vlasenko77ad97f2008-05-13 02:27:31 +0000679static ALWAYS_INLINE int isalnum_(int c)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000680{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000681 return (isalnum(c) || c == '_');
682}
683
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000684static double my_strtod(char **pp)
685{
686#if ENABLE_DESKTOP
687 if ((*pp)[0] == '0'
688 && ((((*pp)[1] | 0x20) == 'x') || isdigit((*pp)[1]))
689 ) {
690 return strtoull(*pp, pp, 0);
691 }
692#endif
693 return strtod(*pp, pp);
694}
695
Glenn L McGrath545106f2002-11-11 06:21:00 +0000696/* -------- working with variables (set/get/copy/etc) -------- */
697
Mike Frysinger10a11e22005-09-27 02:23:02 +0000698static xhash *iamarray(var *v)
699{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000700 var *a = v;
701
702 while (a->type & VF_CHILD)
703 a = a->x.parent;
704
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000705 if (!(a->type & VF_ARRAY)) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000706 a->type |= VF_ARRAY;
707 a->x.array = hash_init();
708 }
709 return a->x.array;
710}
711
Mike Frysinger10a11e22005-09-27 02:23:02 +0000712static void clear_array(xhash *array)
713{
Denis Vlasenkof782f522007-01-01 23:51:30 +0000714 unsigned i;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000715 hash_item *hi, *thi;
716
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000717 for (i = 0; i < array->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000718 hi = array->items[i];
719 while (hi) {
720 thi = hi;
721 hi = hi->next;
Aaron Lehmanna170e1c2002-11-28 11:27:31 +0000722 free(thi->data.v.string);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000723 free(thi);
724 }
725 array->items[i] = NULL;
726 }
727 array->glen = array->nel = 0;
728}
729
730/* clear a variable */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000731static var *clrvar(var *v)
732{
Aaron Lehmanna170e1c2002-11-28 11:27:31 +0000733 if (!(v->type & VF_FSTR))
Glenn L McGrath545106f2002-11-11 06:21:00 +0000734 free(v->string);
735
736 v->type &= VF_DONTTOUCH;
737 v->type |= VF_DIRTY;
738 v->string = NULL;
739 return v;
740}
741
742/* assign string value to variable */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000743static var *setvar_p(var *v, char *value)
744{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000745 clrvar(v);
746 v->string = value;
747 handle_special(v);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000748 return v;
749}
750
751/* same as setvar_p but make a copy of string */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000752static var *setvar_s(var *v, const char *value)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000753{
Rob Landleyd921b2e2006-08-03 15:41:12 +0000754 return setvar_p(v, (value && *value) ? xstrdup(value) : NULL);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000755}
756
757/* same as setvar_s but set USER flag */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000758static var *setvar_u(var *v, const char *value)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000759{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000760 setvar_s(v, value);
761 v->type |= VF_USER;
762 return v;
763}
764
765/* set array element to user string */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000766static void setari_u(var *a, int idx, const char *s)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000767{
"Robert P. J. Day"68229832006-07-01 13:08:46 +0000768 var *v;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000769
Denys Vlasenko7bb346f2009-10-06 22:09:50 +0200770 v = findvar(iamarray(a), itoa(idx));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000771 setvar_u(v, s);
772}
773
774/* assign numeric value to variable */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000775static var *setvar_i(var *v, double value)
776{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000777 clrvar(v);
778 v->type |= VF_NUMBER;
779 v->number = value;
780 handle_special(v);
781 return v;
782}
783
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +0000784static const char *getvar_s(var *v)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000785{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000786 /* if v is numeric and has no cached string, convert it to string */
787 if ((v->type & (VF_NUMBER | VF_CACHED)) == VF_NUMBER) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000788 fmt_num(g_buf, MAXVARFMT, getvar_s(intvar[CONVFMT]), v->number, TRUE);
789 v->string = xstrdup(g_buf);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000790 v->type |= VF_CACHED;
791 }
792 return (v->string == NULL) ? "" : v->string;
793}
794
Mike Frysinger10a11e22005-09-27 02:23:02 +0000795static double getvar_i(var *v)
796{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000797 char *s;
798
799 if ((v->type & (VF_NUMBER | VF_CACHED)) == 0) {
800 v->number = 0;
801 s = v->string;
802 if (s && *s) {
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000803 v->number = my_strtod(&s);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000804 if (v->type & VF_USER) {
805 skip_spaces(&s);
806 if (*s != '\0')
807 v->type &= ~VF_USER;
808 }
809 } else {
810 v->type &= ~VF_USER;
811 }
812 v->type |= VF_CACHED;
813 }
814 return v->number;
815}
816
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000817/* Used for operands of bitwise ops */
818static unsigned long getvar_i_int(var *v)
819{
820 double d = getvar_i(v);
821
822 /* Casting doubles to longs is undefined for values outside
823 * of target type range. Try to widen it as much as possible */
824 if (d >= 0)
825 return (unsigned long)d;
Denis Vlasenko665eaff2008-09-05 04:59:02 +0000826 /* Why? Think about d == -4294967295.0 (assuming 32bit longs) */
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000827 return - (long) (unsigned long) (-d);
828}
829
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +0000830static var *copyvar(var *dest, const var *src)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000831{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000832 if (dest != src) {
833 clrvar(dest);
Denis Vlasenko629563b2007-02-24 17:05:52 +0000834 dest->type |= (src->type & ~(VF_DONTTOUCH | VF_FSTR));
Glenn L McGrath545106f2002-11-11 06:21:00 +0000835 dest->number = src->number;
836 if (src->string)
Rob Landleyd921b2e2006-08-03 15:41:12 +0000837 dest->string = xstrdup(src->string);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000838 }
839 handle_special(dest);
840 return dest;
841}
842
Mike Frysinger10a11e22005-09-27 02:23:02 +0000843static var *incvar(var *v)
844{
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000845 return setvar_i(v, getvar_i(v) + 1.);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000846}
847
848/* return true if v is number or numeric string */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000849static int is_numeric(var *v)
850{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000851 getvar_i(v);
852 return ((v->type ^ VF_DIRTY) & (VF_NUMBER | VF_USER | VF_DIRTY));
853}
854
855/* return 1 when value of v corresponds to true, 0 otherwise */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000856static int istrue(var *v)
857{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000858 if (is_numeric(v))
859 return (v->number == 0) ? 0 : 1;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000860 return (v->string && *(v->string)) ? 1 : 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000861}
862
Eric Andersenaff114c2004-04-14 17:51:38 +0000863/* temporary variables allocator. Last allocated should be first freed */
Mike Frysinger10a11e22005-09-27 02:23:02 +0000864static var *nvalloc(int n)
865{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000866 nvblock *pb = NULL;
867 var *v, *r;
868 int size;
869
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000870 while (g_cb) {
871 pb = g_cb;
872 if ((g_cb->pos - g_cb->nv) + n <= g_cb->size) break;
873 g_cb = g_cb->next;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000874 }
875
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000876 if (!g_cb) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000877 size = (n <= MINNVBLOCK) ? MINNVBLOCK : n;
Denis Vlasenkoe0a7fc52008-07-02 11:14:59 +0000878 g_cb = xzalloc(sizeof(nvblock) + size * sizeof(var));
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000879 g_cb->size = size;
880 g_cb->pos = g_cb->nv;
881 g_cb->prev = pb;
Denis Vlasenkoe0a7fc52008-07-02 11:14:59 +0000882 /*g_cb->next = NULL; - xzalloc did it */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000883 if (pb) pb->next = g_cb;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000884 }
885
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000886 v = r = g_cb->pos;
887 g_cb->pos += n;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000888
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000889 while (v < g_cb->pos) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000890 v->type = 0;
891 v->string = NULL;
892 v++;
893 }
894
895 return r;
896}
897
Mike Frysinger10a11e22005-09-27 02:23:02 +0000898static void nvfree(var *v)
899{
Glenn L McGrath545106f2002-11-11 06:21:00 +0000900 var *p;
901
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000902 if (v < g_cb->nv || v >= g_cb->pos)
903 syntax_error(EMSG_INTERNAL_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000904
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000905 for (p = v; p < g_cb->pos; p++) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000906 if ((p->type & (VF_ARRAY | VF_CHILD)) == VF_ARRAY) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000907 clear_array(iamarray(p));
908 free(p->x.array->items);
909 free(p->x.array);
910 }
911 if (p->type & VF_WALK)
912 free(p->x.walker);
913
914 clrvar(p);
915 }
916
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000917 g_cb->pos = v;
918 while (g_cb->prev && g_cb->pos == g_cb->nv) {
919 g_cb = g_cb->prev;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000920 }
921}
922
923/* ------- awk program text parsing ------- */
924
Denis Vlasenkocd5c7862007-05-17 16:37:22 +0000925/* Parse next token pointed by global pos, place results into global ttt.
Glenn L McGrath545106f2002-11-11 06:21:00 +0000926 * If token isn't expected, give away. Return token class
927 */
Mike Frysingerf87b3e32005-09-27 04:16:22 +0000928static uint32_t next_token(uint32_t expected)
Mike Frysinger10a11e22005-09-27 02:23:02 +0000929{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000930#define concat_inserted (G.next_token__concat_inserted)
931#define save_tclass (G.next_token__save_tclass)
932#define save_info (G.next_token__save_info)
933/* Initialized to TC_OPTERM: */
934#define ltclass (G.next_token__ltclass)
Glenn L McGrath545106f2002-11-11 06:21:00 +0000935
Denis Vlasenkof782f522007-01-01 23:51:30 +0000936 char *p, *pp, *s;
937 const char *tl;
938 uint32_t tc;
939 const uint32_t *ti;
940 int l;
941
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000942 if (t_rollback) {
943 t_rollback = FALSE;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000944
945 } else if (concat_inserted) {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000946 concat_inserted = FALSE;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000947 t_tclass = save_tclass;
948 t_info = save_info;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000949
950 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000951 p = g_pos;
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +0000952 readnext:
Glenn L McGrath545106f2002-11-11 06:21:00 +0000953 skip_spaces(&p);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000954 g_lineno = t_lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000955 if (*p == '#')
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000956 while (*p != '\n' && *p != '\0')
957 p++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000958
959 if (*p == '\n')
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000960 t_lineno++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000961
962 if (*p == '\0') {
963 tc = TC_EOF;
964
965 } else if (*p == '\"') {
966 /* it's a string */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000967 t_string = s = ++p;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000968 while (*p != '\"') {
969 if (*p == '\0' || *p == '\n')
970 syntax_error(EMSG_UNEXP_EOS);
971 *(s++) = nextchar(&p);
972 }
973 p++;
974 *s = '\0';
975 tc = TC_STRING;
976
977 } else if ((expected & TC_REGEXP) && *p == '/') {
978 /* it's regexp */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +0000979 t_string = s = ++p;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000980 while (*p != '/') {
981 if (*p == '\0' || *p == '\n')
982 syntax_error(EMSG_UNEXP_EOS);
Denis Vlasenkod9b5ab82007-05-18 07:30:43 +0000983 *s = *p++;
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000984 if (*s++ == '\\') {
Glenn L McGrath545106f2002-11-11 06:21:00 +0000985 pp = p;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000986 *(s-1) = bb_process_escape_sequence((const char **)&p);
Denis Vlasenkoffba9412007-05-17 23:03:35 +0000987 if (*pp == '\\')
988 *s++ = '\\';
989 if (p == pp)
990 *s++ = *p++;
Glenn L McGrath545106f2002-11-11 06:21:00 +0000991 }
992 }
993 p++;
994 *s = '\0';
995 tc = TC_REGEXP;
996
997 } else if (*p == '.' || isdigit(*p)) {
998 /* it's a number */
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +0000999 t_double = my_strtod(&p);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001000 if (*p == '.')
1001 syntax_error(EMSG_UNEXP_TOKEN);
1002 tc = TC_NUMBER;
1003
1004 } else {
1005 /* search for something known */
1006 tl = tokenlist;
1007 tc = 0x00000001;
1008 ti = tokeninfo;
1009 while (*tl) {
1010 l = *(tl++);
1011 if (l == NTCC) {
1012 tc <<= 1;
1013 continue;
1014 }
1015 /* if token class is expected, token
1016 * matches and it's not a longer word,
1017 * then this is what we are looking for
1018 */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001019 if ((tc & (expected | TC_WORD | TC_NEWLINE))
1020 && *tl == *p && strncmp(p, tl, l) == 0
1021 && !((tc & TC_WORD) && isalnum_(p[l]))
1022 ) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001023 t_info = *ti;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001024 p += l;
1025 break;
1026 }
1027 ti++;
1028 tl += l;
1029 }
1030
Denis Vlasenkof782f522007-01-01 23:51:30 +00001031 if (!*tl) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001032 /* it's a name (var/array/function),
1033 * otherwise it's something wrong
1034 */
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001035 if (!isalnum_(*p))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001036 syntax_error(EMSG_UNEXP_TOKEN);
1037
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001038 t_string = --p;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001039 while (isalnum_(*(++p))) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001040 *(p-1) = *p;
1041 }
1042 *(p-1) = '\0';
1043 tc = TC_VARIABLE;
Bernhard Reutner-Fischerbb204622005-10-17 14:21:06 +00001044 /* also consume whitespace between functionname and bracket */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001045 if (!(expected & TC_VARIABLE))
1046 skip_spaces(&p);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001047 if (*p == '(') {
1048 tc = TC_FUNCTION;
1049 } else {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001050 if (*p == '[') {
1051 p++;
1052 tc = TC_ARRAY;
1053 }
1054 }
1055 }
1056 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001057 g_pos = p;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001058
1059 /* skipping newlines in some cases */
1060 if ((ltclass & TC_NOTERM) && (tc & TC_NEWLINE))
1061 goto readnext;
1062
1063 /* insert concatenation operator when needed */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001064 if ((ltclass & TC_CONCAT1) && (tc & TC_CONCAT2) && (expected & TC_BINOP)) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001065 concat_inserted = TRUE;
1066 save_tclass = tc;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001067 save_info = t_info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001068 tc = TC_BINOP;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001069 t_info = OC_CONCAT | SS | P(35);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001070 }
1071
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001072 t_tclass = tc;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001073 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001074 ltclass = t_tclass;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001075
1076 /* Are we ready for this? */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001077 if (!(ltclass & expected))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001078 syntax_error((ltclass & (TC_NEWLINE | TC_EOF)) ?
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001079 EMSG_UNEXP_EOS : EMSG_UNEXP_TOKEN);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001080
1081 return ltclass;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001082#undef concat_inserted
1083#undef save_tclass
1084#undef save_info
1085#undef ltclass
Glenn L McGrath545106f2002-11-11 06:21:00 +00001086}
1087
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001088static void rollback_token(void)
1089{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001090 t_rollback = TRUE;
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001091}
Glenn L McGrath545106f2002-11-11 06:21:00 +00001092
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001093static node *new_node(uint32_t info)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001094{
"Robert P. J. Day"68229832006-07-01 13:08:46 +00001095 node *n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001096
Denis Vlasenko4cccc032006-12-22 18:37:07 +00001097 n = xzalloc(sizeof(node));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001098 n->info = info;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001099 n->lineno = g_lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001100 return n;
1101}
1102
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001103static node *mk_re_node(const char *s, node *n, regex_t *re)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001104{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001105 n->info = OC_REGEXP;
1106 n->l.re = re;
1107 n->r.ire = re + 1;
1108 xregcomp(re, s, REG_EXTENDED);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001109 xregcomp(re + 1, s, REG_EXTENDED | REG_ICASE);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001110
1111 return n;
1112}
1113
Mike Frysinger10a11e22005-09-27 02:23:02 +00001114static node *condition(void)
1115{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001116 next_token(TC_SEQSTART);
1117 return parse_expr(TC_SEQTERM);
1118}
1119
1120/* parse expression terminated by given argument, return ptr
1121 * to built subtree. Terminator is eaten by parse_expr */
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001122static node *parse_expr(uint32_t iexp)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001123{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001124 node sn;
1125 node *cn = &sn;
1126 node *vn, *glptr;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001127 uint32_t tc, xtc;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001128 var *v;
1129
1130 sn.info = PRIMASK;
1131 sn.r.n = glptr = NULL;
1132 xtc = TC_OPERAND | TC_UOPPRE | TC_REGEXP | iexp;
1133
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001134 while (!((tc = next_token(xtc)) & iexp)) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001135 if (glptr && (t_info == (OC_COMPARE | VV | P(39) | 2))) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001136 /* input redirection (<) attached to glptr node */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001137 cn = glptr->l.n = new_node(OC_CONCAT | SS | P(37));
Glenn L McGrath4bded582004-02-22 11:55:09 +00001138 cn->a.n = glptr;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001139 xtc = TC_OPERAND | TC_UOPPRE;
1140 glptr = NULL;
1141
1142 } else if (tc & (TC_BINOP | TC_UOPPOST)) {
1143 /* for binary and postfix-unary operators, jump back over
1144 * previous operators with higher priority */
1145 vn = cn;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001146 while ( ((t_info & PRIMASK) > (vn->a.n->info & PRIMASK2))
1147 || ((t_info == vn->info) && ((t_info & OPCLSMASK) == OC_COLON)) )
Glenn L McGrath545106f2002-11-11 06:21:00 +00001148 vn = vn->a.n;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001149 if ((t_info & OPCLSMASK) == OC_TERNARY)
1150 t_info += P(6);
1151 cn = vn->a.n->r.n = new_node(t_info);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001152 cn->a.n = vn->a.n;
1153 if (tc & TC_BINOP) {
1154 cn->l.n = vn;
1155 xtc = TC_OPERAND | TC_UOPPRE | TC_REGEXP;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001156 if ((t_info & OPCLSMASK) == OC_PGETLINE) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001157 /* it's a pipe */
1158 next_token(TC_GETLINE);
1159 /* give maximum priority to this pipe */
1160 cn->info &= ~PRIMASK;
1161 xtc = TC_OPERAND | TC_UOPPRE | TC_BINOP | iexp;
1162 }
1163 } else {
1164 cn->r.n = vn;
1165 xtc = TC_OPERAND | TC_UOPPRE | TC_BINOP | iexp;
1166 }
1167 vn->a.n = cn;
1168
1169 } else {
1170 /* for operands and prefix-unary operators, attach them
1171 * to last node */
1172 vn = cn;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001173 cn = vn->r.n = new_node(t_info);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001174 cn->a.n = vn;
1175 xtc = TC_OPERAND | TC_UOPPRE | TC_REGEXP;
1176 if (tc & (TC_OPERAND | TC_REGEXP)) {
Rob Landleyed830e82005-06-07 02:43:52 +00001177 xtc = TC_UOPPRE | TC_UOPPOST | TC_BINOP | TC_OPERAND | iexp;
Eric Andersenc7bda1c2004-03-15 08:29:22 +00001178 /* one should be very careful with switch on tclass -
Glenn L McGrath545106f2002-11-11 06:21:00 +00001179 * only simple tclasses should be used! */
1180 switch (tc) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001181 case TC_VARIABLE:
1182 case TC_ARRAY:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001183 cn->info = OC_VAR;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001184 v = hash_search(ahash, t_string);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001185 if (v != NULL) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001186 cn->info = OC_FNARG;
1187 cn->l.i = v->x.aidx;
1188 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001189 cn->l.v = newvar(t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001190 }
1191 if (tc & TC_ARRAY) {
1192 cn->info |= xS;
1193 cn->r.n = parse_expr(TC_ARRTERM);
1194 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001195 break;
Mike Frysingerde2b9382005-09-27 03:18:00 +00001196
Denis Vlasenkof782f522007-01-01 23:51:30 +00001197 case TC_NUMBER:
1198 case TC_STRING:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001199 cn->info = OC_VAR;
Rob Landley9ffd4232006-05-21 18:30:35 +00001200 v = cn->l.v = xzalloc(sizeof(var));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001201 if (tc & TC_NUMBER)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001202 setvar_i(v, t_double);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001203 else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001204 setvar_s(v, t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001205 break;
1206
Denis Vlasenkof782f522007-01-01 23:51:30 +00001207 case TC_REGEXP:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001208 mk_re_node(t_string, cn, xzalloc(sizeof(regex_t)*2));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001209 break;
1210
Denis Vlasenkof782f522007-01-01 23:51:30 +00001211 case TC_FUNCTION:
Mike Frysingerde2b9382005-09-27 03:18:00 +00001212 cn->info = OC_FUNC;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001213 cn->r.f = newfunc(t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001214 cn->l.n = condition();
1215 break;
1216
Denis Vlasenkof782f522007-01-01 23:51:30 +00001217 case TC_SEQSTART:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001218 cn = vn->r.n = parse_expr(TC_SEQTERM);
1219 cn->a.n = vn;
1220 break;
1221
Denis Vlasenkof782f522007-01-01 23:51:30 +00001222 case TC_GETLINE:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001223 glptr = cn;
1224 xtc = TC_OPERAND | TC_UOPPRE | TC_BINOP | iexp;
1225 break;
1226
Denis Vlasenkof782f522007-01-01 23:51:30 +00001227 case TC_BUILTIN:
Glenn L McGrath545106f2002-11-11 06:21:00 +00001228 cn->l.n = condition();
1229 break;
1230 }
1231 }
1232 }
1233 }
1234 return sn.r.n;
1235}
1236
1237/* add node to chain. Return ptr to alloc'd node */
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001238static node *chain_node(uint32_t info)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001239{
"Robert P. J. Day"68229832006-07-01 13:08:46 +00001240 node *n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001241
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00001242 if (!seq->first)
Glenn L McGrath545106f2002-11-11 06:21:00 +00001243 seq->first = seq->last = new_node(0);
1244
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001245 if (seq->programname != g_progname) {
1246 seq->programname = g_progname;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001247 n = chain_node(OC_NEWSOURCE);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001248 n->l.s = xstrdup(g_progname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001249 }
1250
1251 n = seq->last;
1252 n->info = info;
1253 seq->last = n->a.n = new_node(OC_DONE);
1254
1255 return n;
1256}
1257
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001258static void chain_expr(uint32_t info)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001259{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001260 node *n;
1261
1262 n = chain_node(info);
1263 n->l.n = parse_expr(TC_OPTERM | TC_GRPTERM);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001264 if (t_tclass & TC_GRPTERM)
Glenn L McGrath545106f2002-11-11 06:21:00 +00001265 rollback_token();
1266}
1267
Mike Frysinger10a11e22005-09-27 02:23:02 +00001268static node *chain_loop(node *nn)
1269{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001270 node *n, *n2, *save_brk, *save_cont;
1271
1272 save_brk = break_ptr;
1273 save_cont = continue_ptr;
1274
1275 n = chain_node(OC_BR | Vx);
1276 continue_ptr = new_node(OC_EXEC);
1277 break_ptr = new_node(OC_EXEC);
1278 chain_group();
1279 n2 = chain_node(OC_EXEC | Vx);
1280 n2->l.n = nn;
1281 n2->a.n = n;
1282 continue_ptr->a.n = n2;
1283 break_ptr->a.n = n->r.n = seq->last;
1284
1285 continue_ptr = save_cont;
1286 break_ptr = save_brk;
1287
1288 return n;
1289}
1290
1291/* parse group and attach it to chain */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001292static void chain_group(void)
1293{
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001294 uint32_t c;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001295 node *n, *n2, *n3;
1296
1297 do {
1298 c = next_token(TC_GRPSEQ);
1299 } while (c & TC_NEWLINE);
1300
1301 if (c & TC_GRPSTART) {
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001302 while (next_token(TC_GRPSEQ | TC_GRPTERM) != TC_GRPTERM) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001303 if (t_tclass & TC_NEWLINE) continue;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001304 rollback_token();
1305 chain_group();
1306 }
1307 } else if (c & (TC_OPSEQ | TC_OPTERM)) {
1308 rollback_token();
1309 chain_expr(OC_EXEC | Vx);
1310 } else { /* TC_STATEMNT */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001311 switch (t_info & OPCLSMASK) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001312 case ST_IF:
1313 n = chain_node(OC_BR | Vx);
1314 n->l.n = condition();
1315 chain_group();
1316 n2 = chain_node(OC_EXEC);
1317 n->r.n = seq->last;
1318 if (next_token(TC_GRPSEQ | TC_GRPTERM | TC_ELSE) == TC_ELSE) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001319 chain_group();
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001320 n2->a.n = seq->last;
1321 } else {
1322 rollback_token();
1323 }
1324 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001325
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001326 case ST_WHILE:
1327 n2 = condition();
1328 n = chain_loop(NULL);
1329 n->l.n = n2;
1330 break;
1331
1332 case ST_DO:
1333 n2 = chain_node(OC_EXEC);
1334 n = chain_loop(NULL);
1335 n2->a.n = n->a.n;
1336 next_token(TC_WHILE);
1337 n->l.n = condition();
1338 break;
1339
1340 case ST_FOR:
1341 next_token(TC_SEQSTART);
1342 n2 = parse_expr(TC_SEMICOL | TC_SEQTERM);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001343 if (t_tclass & TC_SEQTERM) { /* for-in */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001344 if ((n2->info & OPCLSMASK) != OC_IN)
1345 syntax_error(EMSG_UNEXP_TOKEN);
1346 n = chain_node(OC_WALKINIT | VV);
1347 n->l.n = n2->l.n;
1348 n->r.n = n2->r.n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001349 n = chain_loop(NULL);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001350 n->info = OC_WALKNEXT | Vx;
1351 n->l.n = n2->l.n;
1352 } else { /* for (;;) */
1353 n = chain_node(OC_EXEC | Vx);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001354 n->l.n = n2;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001355 n2 = parse_expr(TC_SEMICOL);
1356 n3 = parse_expr(TC_SEQTERM);
1357 n = chain_loop(n3);
1358 n->l.n = n2;
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001359 if (!n2)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001360 n->info = OC_EXEC;
1361 }
1362 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001363
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001364 case OC_PRINT:
1365 case OC_PRINTF:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001366 n = chain_node(t_info);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001367 n->l.n = parse_expr(TC_OPTERM | TC_OUTRDR | TC_GRPTERM);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001368 if (t_tclass & TC_OUTRDR) {
1369 n->info |= t_info;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001370 n->r.n = parse_expr(TC_OPTERM | TC_GRPTERM);
1371 }
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001372 if (t_tclass & TC_GRPTERM)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001373 rollback_token();
1374 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001375
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001376 case OC_BREAK:
1377 n = chain_node(OC_EXEC);
1378 n->a.n = break_ptr;
1379 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001380
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001381 case OC_CONTINUE:
1382 n = chain_node(OC_EXEC);
1383 n->a.n = continue_ptr;
1384 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001385
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001386 /* delete, next, nextfile, return, exit */
1387 default:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001388 chain_expr(t_info);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001389 }
1390 }
1391}
1392
Mike Frysinger10a11e22005-09-27 02:23:02 +00001393static void parse_program(char *p)
1394{
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001395 uint32_t tclass;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001396 node *cn;
1397 func *f;
1398 var *v;
1399
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001400 g_pos = p;
1401 t_lineno = 1;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001402 while ((tclass = next_token(TC_EOF | TC_OPSEQ | TC_GRPSTART |
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001403 TC_OPTERM | TC_BEGIN | TC_END | TC_FUNCDECL)) != TC_EOF) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001404
1405 if (tclass & TC_OPTERM)
1406 continue;
1407
1408 seq = &mainseq;
1409 if (tclass & TC_BEGIN) {
1410 seq = &beginseq;
1411 chain_group();
1412
1413 } else if (tclass & TC_END) {
1414 seq = &endseq;
1415 chain_group();
1416
1417 } else if (tclass & TC_FUNCDECL) {
1418 next_token(TC_FUNCTION);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001419 g_pos++;
1420 f = newfunc(t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001421 f->body.first = NULL;
1422 f->nargs = 0;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001423 while (next_token(TC_VARIABLE | TC_SEQTERM) & TC_VARIABLE) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001424 v = findvar(ahash, t_string);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001425 v->x.aidx = (f->nargs)++;
1426
1427 if (next_token(TC_COMMA | TC_SEQTERM) & TC_SEQTERM)
1428 break;
1429 }
1430 seq = &(f->body);
1431 chain_group();
1432 clear_array(ahash);
1433
1434 } else if (tclass & TC_OPSEQ) {
1435 rollback_token();
1436 cn = chain_node(OC_TEST);
1437 cn->l.n = parse_expr(TC_OPTERM | TC_EOF | TC_GRPSTART);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001438 if (t_tclass & TC_GRPSTART) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001439 rollback_token();
1440 chain_group();
1441 } else {
1442 chain_node(OC_PRINT);
1443 }
1444 cn->r.n = mainseq.last;
1445
1446 } else /* if (tclass & TC_GRPSTART) */ {
1447 rollback_token();
1448 chain_group();
1449 }
1450 }
1451}
1452
1453
1454/* -------- program execution part -------- */
1455
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001456static node *mk_splitter(const char *s, tsplitter *spl)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001457{
"Robert P. J. Day"68229832006-07-01 13:08:46 +00001458 regex_t *re, *ire;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001459 node *n;
1460
1461 re = &spl->re[0];
1462 ire = &spl->re[1];
1463 n = &spl->n;
Denis Vlasenko890ac9d2006-10-07 15:16:19 +00001464 if ((n->info & OPCLSMASK) == OC_REGEXP) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001465 regfree(re);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001466 regfree(ire); // TODO: nuke ire, use re+1?
Glenn L McGrath545106f2002-11-11 06:21:00 +00001467 }
Rob Landleya3896512006-05-07 20:20:34 +00001468 if (strlen(s) > 1) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001469 mk_re_node(s, n, re);
1470 } else {
Mike Frysingerf87b3e32005-09-27 04:16:22 +00001471 n->info = (uint32_t) *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001472 }
1473
1474 return n;
1475}
1476
1477/* use node as a regular expression. Supplied with node ptr and regex_t
Eric Andersenaff114c2004-04-14 17:51:38 +00001478 * storage space. Return ptr to regex (if result points to preg, it should
Glenn L McGrath545106f2002-11-11 06:21:00 +00001479 * be later regfree'd manually
1480 */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001481static regex_t *as_regex(node *op, regex_t *preg)
1482{
Denis Vlasenko7a676642009-03-15 22:20:31 +00001483 int cflags;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001484 var *v;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001485 const char *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001486
1487 if ((op->info & OPCLSMASK) == OC_REGEXP) {
1488 return icase ? op->r.ire : op->l.re;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001489 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001490 v = nvalloc(1);
1491 s = getvar_s(evaluate(op, v));
Denis Vlasenko7a676642009-03-15 22:20:31 +00001492
1493 cflags = icase ? REG_EXTENDED | REG_ICASE : REG_EXTENDED;
1494 /* Testcase where REG_EXTENDED fails (unpaired '{'):
1495 * echo Hi | awk 'gsub("@(samp|code|file)\{","");'
1496 * gawk 3.1.5 eats this. We revert to ~REG_EXTENDED
1497 * (maybe gsub is not supposed to use REG_EXTENDED?).
1498 */
1499 if (regcomp(preg, s, cflags)) {
1500 cflags &= ~REG_EXTENDED;
1501 xregcomp(preg, s, cflags);
1502 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001503 nvfree(v);
1504 return preg;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001505}
1506
1507/* gradually increasing buffer */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001508static void qrealloc(char **b, int n, int *size)
1509{
Denis Vlasenkodeeed592008-07-08 05:14:36 +00001510 if (!*b || n >= *size) {
1511 *size = n + (n>>1) + 80;
1512 *b = xrealloc(*b, *size);
1513 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001514}
1515
1516/* resize field storage space */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001517static void fsrealloc(int size)
1518{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001519 int i;
1520
1521 if (size >= maxfields) {
1522 i = maxfields;
1523 maxfields = size + 16;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001524 Fields = xrealloc(Fields, maxfields * sizeof(var));
1525 for (; i < maxfields; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001526 Fields[i].type = VF_SPECIAL;
1527 Fields[i].string = NULL;
1528 }
1529 }
1530
1531 if (size < nfields) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001532 for (i = size; i < nfields; i++) {
1533 clrvar(Fields + i);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001534 }
1535 }
1536 nfields = size;
1537}
1538
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001539static int awk_split(const char *s, node *spl, char **slist)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001540{
Denis Vlasenkof782f522007-01-01 23:51:30 +00001541 int l, n = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001542 char c[4];
1543 char *s1;
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001544 regmatch_t pmatch[2]; // TODO: why [2]? [1] is enough...
Glenn L McGrath545106f2002-11-11 06:21:00 +00001545
1546 /* in worst case, each char would be a separate field */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001547 *slist = s1 = xzalloc(strlen(s) * 2 + 3);
1548 strcpy(s1, s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001549
1550 c[0] = c[1] = (char)spl->info;
1551 c[2] = c[3] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001552 if (*getvar_s(intvar[RS]) == '\0')
1553 c[2] = '\n';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001554
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001555 if ((spl->info & OPCLSMASK) == OC_REGEXP) { /* regex split */
1556 if (!*s)
1557 return n; /* "": zero fields */
1558 n++; /* at least one field will be there */
1559 do {
1560 l = strcspn(s, c+2); /* len till next NUL or \n */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001561 if (regexec(icase ? spl->r.ire : spl->l.re, s, 1, pmatch, 0) == 0
1562 && pmatch[0].rm_so <= l
1563 ) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001564 l = pmatch[0].rm_so;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001565 if (pmatch[0].rm_eo == 0) {
1566 l++;
1567 pmatch[0].rm_eo++;
1568 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001569 n++; /* we saw yet another delimiter */
Glenn L McGrath545106f2002-11-11 06:21:00 +00001570 } else {
1571 pmatch[0].rm_eo = l;
Denys Vlasenkoe4244232009-05-18 23:50:03 +02001572 if (s[l])
1573 pmatch[0].rm_eo++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001574 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001575 memcpy(s1, s, l);
Denis Vlasenko67b5eeb2009-04-12 13:54:13 +00001576 /* make sure we remove *all* of the separator chars */
Denys Vlasenkoe4244232009-05-18 23:50:03 +02001577 do {
1578 s1[l] = '\0';
1579 } while (++l < pmatch[0].rm_eo);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001580 nextword(&s1);
1581 s += pmatch[0].rm_eo;
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001582 } while (*s);
1583 return n;
1584 }
1585 if (c[0] == '\0') { /* null split */
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001586 while (*s) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001587 *s1++ = *s++;
1588 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001589 n++;
1590 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001591 return n;
1592 }
1593 if (c[0] != ' ') { /* single-character split */
Glenn L McGrath545106f2002-11-11 06:21:00 +00001594 if (icase) {
1595 c[0] = toupper(c[0]);
1596 c[1] = tolower(c[1]);
1597 }
1598 if (*s1) n++;
1599 while ((s1 = strpbrk(s1, c))) {
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001600 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001601 n++;
1602 }
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001603 return n;
1604 }
1605 /* space split */
1606 while (*s) {
1607 s = skip_whitespace(s);
1608 if (!*s) break;
1609 n++;
1610 while (*s && !isspace(*s))
1611 *s1++ = *s++;
1612 *s1++ = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001613 }
1614 return n;
1615}
1616
Mike Frysinger10a11e22005-09-27 02:23:02 +00001617static void split_f0(void)
1618{
Denis Vlasenkoaf1bd092007-07-18 18:32:25 +00001619/* static char *fstrings; */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001620#define fstrings (G.split_f0__fstrings)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001621
Glenn L McGrath545106f2002-11-11 06:21:00 +00001622 int i, n;
1623 char *s;
1624
1625 if (is_f0_split)
1626 return;
1627
1628 is_f0_split = TRUE;
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00001629 free(fstrings);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001630 fsrealloc(0);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001631 n = awk_split(getvar_s(intvar[F0]), &fsplitter.n, &fstrings);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001632 fsrealloc(n);
1633 s = fstrings;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001634 for (i = 0; i < n; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001635 Fields[i].string = nextword(&s);
1636 Fields[i].type |= (VF_FSTR | VF_USER | VF_DIRTY);
1637 }
1638
1639 /* set NF manually to avoid side effects */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001640 clrvar(intvar[NF]);
1641 intvar[NF]->type = VF_NUMBER | VF_SPECIAL;
1642 intvar[NF]->number = nfields;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001643#undef fstrings
Glenn L McGrath545106f2002-11-11 06:21:00 +00001644}
1645
1646/* perform additional actions when some internal variables changed */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001647static void handle_special(var *v)
1648{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001649 int n;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001650 char *b;
1651 const char *sep, *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001652 int sl, l, len, i, bsize;
1653
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001654 if (!(v->type & VF_SPECIAL))
Glenn L McGrath545106f2002-11-11 06:21:00 +00001655 return;
1656
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001657 if (v == intvar[NF]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001658 n = (int)getvar_i(v);
1659 fsrealloc(n);
1660
1661 /* recalculate $0 */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001662 sep = getvar_s(intvar[OFS]);
Rob Landleya3896512006-05-07 20:20:34 +00001663 sl = strlen(sep);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001664 b = NULL;
1665 len = 0;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001666 for (i = 0; i < n; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001667 s = getvar_s(&Fields[i]);
Rob Landleya3896512006-05-07 20:20:34 +00001668 l = strlen(s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001669 if (b) {
1670 memcpy(b+len, sep, sl);
1671 len += sl;
1672 }
1673 qrealloc(&b, len+l+sl, &bsize);
1674 memcpy(b+len, s, l);
1675 len += l;
1676 }
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001677 if (b)
1678 b[len] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001679 setvar_p(intvar[F0], b);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001680 is_f0_split = TRUE;
1681
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001682 } else if (v == intvar[F0]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001683 is_f0_split = FALSE;
1684
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001685 } else if (v == intvar[FS]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001686 mk_splitter(getvar_s(v), &fsplitter);
1687
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001688 } else if (v == intvar[RS]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001689 mk_splitter(getvar_s(v), &rsplitter);
1690
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001691 } else if (v == intvar[IGNORECASE]) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001692 icase = istrue(v);
1693
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00001694 } else { /* $n */
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001695 n = getvar_i(intvar[NF]);
1696 setvar_i(intvar[NF], n > v-Fields ? n : v-Fields+1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001697 /* right here v is invalid. Just to note... */
1698 }
1699}
1700
1701/* step through func/builtin/etc arguments */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001702static node *nextarg(node **pn)
1703{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001704 node *n;
1705
1706 n = *pn;
1707 if (n && (n->info & OPCLSMASK) == OC_COMMA) {
1708 *pn = n->r.n;
1709 n = n->l.n;
1710 } else {
1711 *pn = NULL;
1712 }
1713 return n;
1714}
1715
Mike Frysinger10a11e22005-09-27 02:23:02 +00001716static void hashwalk_init(var *v, xhash *array)
1717{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001718 char **w;
1719 hash_item *hi;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +00001720 unsigned i;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001721
1722 if (v->type & VF_WALK)
1723 free(v->x.walker);
1724
1725 v->type |= VF_WALK;
Denis Vlasenko4cccc032006-12-22 18:37:07 +00001726 w = v->x.walker = xzalloc(2 + 2*sizeof(char *) + array->glen);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001727 w[0] = w[1] = (char *)(w + 2);
1728 for (i = 0; i < array->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001729 hi = array->items[i];
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00001730 while (hi) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001731 strcpy(*w, hi->name);
1732 nextword(w);
1733 hi = hi->next;
1734 }
1735 }
1736}
1737
Mike Frysinger10a11e22005-09-27 02:23:02 +00001738static int hashwalk_next(var *v)
1739{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001740 char **w;
1741
1742 w = v->x.walker;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001743 if (w[1] == w[0])
Glenn L McGrath545106f2002-11-11 06:21:00 +00001744 return FALSE;
1745
1746 setvar_s(v, nextword(w+1));
1747 return TRUE;
1748}
1749
1750/* evaluate node, return 1 when result is true, 0 otherwise */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001751static int ptest(node *pattern)
1752{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001753 /* ptest__v is "static": to save stack space? */
1754 return istrue(evaluate(pattern, &G.ptest__v));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001755}
1756
1757/* read next record from stream rsm into a variable v */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001758static int awk_getline(rstream *rsm, var *v)
1759{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001760 char *b;
1761 regmatch_t pmatch[2];
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001762 int a, p, pp=0, size;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001763 int fd, so, eo, r, rp;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001764 char c, *m, *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001765
1766 /* we're using our own buffer since we need access to accumulating
1767 * characters
1768 */
1769 fd = fileno(rsm->F);
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001770 m = rsm->buffer;
1771 a = rsm->adv;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001772 p = rsm->pos;
1773 size = rsm->size;
1774 c = (char) rsplitter.n.info;
1775 rp = 0;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001776
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001777 if (!m) qrealloc(&m, 256, &size);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001778 do {
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001779 b = m + a;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001780 so = eo = p;
1781 r = 1;
1782 if (p > 0) {
1783 if ((rsplitter.n.info & OPCLSMASK) == OC_REGEXP) {
1784 if (regexec(icase ? rsplitter.n.r.ire : rsplitter.n.l.re,
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001785 b, 1, pmatch, 0) == 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001786 so = pmatch[0].rm_so;
1787 eo = pmatch[0].rm_eo;
1788 if (b[eo] != '\0')
1789 break;
1790 }
1791 } else if (c != '\0') {
1792 s = strchr(b+pp, c);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001793 if (!s) s = memchr(b+pp, '\0', p - pp);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001794 if (s) {
1795 so = eo = s-b;
1796 eo++;
1797 break;
1798 }
1799 } else {
1800 while (b[rp] == '\n')
1801 rp++;
1802 s = strstr(b+rp, "\n\n");
1803 if (s) {
1804 so = eo = s-b;
1805 while (b[eo] == '\n') eo++;
1806 if (b[eo] != '\0')
1807 break;
1808 }
1809 }
1810 }
1811
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001812 if (a > 0) {
1813 memmove(m, (const void *)(m+a), p+1);
1814 b = m;
1815 a = 0;
1816 }
1817
1818 qrealloc(&m, a+p+128, &size);
1819 b = m + a;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001820 pp = p;
1821 p += safe_read(fd, b+p, size-p-1);
1822 if (p < pp) {
1823 p = 0;
1824 r = 0;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001825 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001826 }
1827 b[p] = '\0';
1828
1829 } while (p > pp);
1830
1831 if (p == 0) {
1832 r--;
1833 } else {
1834 c = b[so]; b[so] = '\0';
1835 setvar_s(v, b+rp);
1836 v->type |= VF_USER;
1837 b[so] = c;
1838 c = b[eo]; b[eo] = '\0';
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001839 setvar_s(intvar[RT], b+so);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001840 b[eo] = c;
1841 }
1842
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00001843 rsm->buffer = m;
1844 rsm->adv = a + eo;
1845 rsm->pos = p - eo;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001846 rsm->size = size;
1847
1848 return r;
1849}
1850
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00001851static int fmt_num(char *b, int size, const char *format, double n, int int_as_int)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001852{
Denis Vlasenkof782f522007-01-01 23:51:30 +00001853 int r = 0;
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00001854 char c;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001855 const char *s = format;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001856
1857 if (int_as_int && n == (int)n) {
1858 r = snprintf(b, size, "%d", (int)n);
1859 } else {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001860 do { c = *s; } while (c && *++s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001861 if (strchr("diouxX", c)) {
1862 r = snprintf(b, size, format, (int)n);
1863 } else if (strchr("eEfgG", c)) {
1864 r = snprintf(b, size, format, n);
1865 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00001866 syntax_error(EMSG_INV_FMT);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001867 }
1868 }
1869 return r;
1870}
1871
Glenn L McGrath545106f2002-11-11 06:21:00 +00001872/* formatted output into an allocated buffer, return ptr to buffer */
Mike Frysinger10a11e22005-09-27 02:23:02 +00001873static char *awk_printf(node *n)
1874{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001875 char *b = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001876 char *fmt, *s, *f;
1877 const char *s1;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001878 int i, j, incr, bsize;
1879 char c, c1;
1880 var *v, *arg;
1881
1882 v = nvalloc(1);
Rob Landleyd921b2e2006-08-03 15:41:12 +00001883 fmt = f = xstrdup(getvar_s(evaluate(nextarg(&n), v)));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001884
1885 i = 0;
1886 while (*f) {
1887 s = f;
1888 while (*f && (*f != '%' || *(++f) == '%'))
1889 f++;
Denis Vlasenko389f9d52007-05-09 21:57:23 +00001890 while (*f && !isalpha(*f)) {
1891 if (*f == '*')
1892 syntax_error("%*x formats are not supported");
Glenn L McGrath545106f2002-11-11 06:21:00 +00001893 f++;
Denis Vlasenko389f9d52007-05-09 21:57:23 +00001894 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00001895
1896 incr = (f - s) + MAXVARFMT;
Denis Vlasenkof782f522007-01-01 23:51:30 +00001897 qrealloc(&b, incr + i, &bsize);
1898 c = *f;
1899 if (c != '\0') f++;
1900 c1 = *f;
1901 *f = '\0';
Glenn L McGrath545106f2002-11-11 06:21:00 +00001902 arg = evaluate(nextarg(&n), v);
1903
1904 j = i;
1905 if (c == 'c' || !c) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00001906 i += sprintf(b+i, s, is_numeric(arg) ?
1907 (char)getvar_i(arg) : *getvar_s(arg));
Glenn L McGrath545106f2002-11-11 06:21:00 +00001908 } else if (c == 's') {
Denis Vlasenko92758142006-10-03 19:56:34 +00001909 s1 = getvar_s(arg);
Rob Landleya3896512006-05-07 20:20:34 +00001910 qrealloc(&b, incr+i+strlen(s1), &bsize);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001911 i += sprintf(b+i, s, s1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001912 } else {
1913 i += fmt_num(b+i, incr, s, getvar_i(arg), FALSE);
1914 }
1915 *f = c1;
1916
1917 /* if there was an error while sprintf, return value is negative */
1918 if (i < j) i = j;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001919 }
1920
Denis Vlasenkof782f522007-01-01 23:51:30 +00001921 b = xrealloc(b, i + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00001922 free(fmt);
1923 nvfree(v);
1924 b[i] = '\0';
1925 return b;
1926}
1927
1928/* common substitution routine
1929 * replace (nm) substring of (src) that match (n) with (repl), store
1930 * result into (dest), return number of substitutions. If nm=0, replace
1931 * all matches. If src or dst is NULL, use $0. If ex=TRUE, enable
1932 * subexpression matching (\1-\9)
1933 */
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001934static int awk_sub(node *rn, const char *repl, int nm, var *src, var *dest, int ex)
Mike Frysinger10a11e22005-09-27 02:23:02 +00001935{
Glenn L McGrath545106f2002-11-11 06:21:00 +00001936 char *ds = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00001937 const char *s;
1938 const char *sp;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001939 int c, i, j, di, rl, so, eo, nbs, n, dssize;
1940 regmatch_t pmatch[10];
1941 regex_t sreg, *re;
1942
1943 re = as_regex(rn, &sreg);
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001944 if (!src) src = intvar[F0];
1945 if (!dest) dest = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00001946
1947 i = di = 0;
1948 sp = getvar_s(src);
Rob Landleya3896512006-05-07 20:20:34 +00001949 rl = strlen(repl);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00001950 while (regexec(re, sp, 10, pmatch, sp==getvar_s(src) ? 0 : REG_NOTBOL) == 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00001951 so = pmatch[0].rm_so;
1952 eo = pmatch[0].rm_eo;
1953
1954 qrealloc(&ds, di + eo + rl, &dssize);
1955 memcpy(ds + di, sp, eo);
1956 di += eo;
1957 if (++i >= nm) {
1958 /* replace */
1959 di -= (eo - so);
1960 nbs = 0;
1961 for (s = repl; *s; s++) {
1962 ds[di++] = c = *s;
1963 if (c == '\\') {
1964 nbs++;
1965 continue;
1966 }
1967 if (c == '&' || (ex && c >= '0' && c <= '9')) {
1968 di -= ((nbs + 3) >> 1);
1969 j = 0;
1970 if (c != '&') {
1971 j = c - '0';
1972 nbs++;
1973 }
1974 if (nbs % 2) {
1975 ds[di++] = c;
1976 } else {
1977 n = pmatch[j].rm_eo - pmatch[j].rm_so;
1978 qrealloc(&ds, di + rl + n, &dssize);
1979 memcpy(ds + di, sp + pmatch[j].rm_so, n);
1980 di += n;
1981 }
1982 }
1983 nbs = 0;
1984 }
1985 }
1986
1987 sp += eo;
1988 if (i == nm) break;
1989 if (eo == so) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00001990 ds[di] = *sp++;
1991 if (!ds[di++]) break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00001992 }
1993 }
1994
1995 qrealloc(&ds, di + strlen(sp), &dssize);
1996 strcpy(ds + di, sp);
1997 setvar_p(dest, ds);
1998 if (re == &sreg) regfree(re);
1999 return i;
2000}
2001
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002002static NOINLINE int do_mktime(const char *ds)
2003{
2004 struct tm then;
2005 int count;
2006
2007 /*memset(&then, 0, sizeof(then)); - not needed */
2008 then.tm_isdst = -1; /* default is unknown */
2009
2010 /* manpage of mktime says these fields are ints,
2011 * so we can sscanf stuff directly into them */
2012 count = sscanf(ds, "%u %u %u %u %u %u %d",
2013 &then.tm_year, &then.tm_mon, &then.tm_mday,
2014 &then.tm_hour, &then.tm_min, &then.tm_sec,
2015 &then.tm_isdst);
2016
2017 if (count < 6
2018 || (unsigned)then.tm_mon < 1
2019 || (unsigned)then.tm_year < 1900
2020 ) {
2021 return -1;
2022 }
2023
2024 then.tm_mon -= 1;
Denys Vlasenkobc3e9472009-09-21 04:16:00 +02002025 then.tm_year -= 1900;
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002026
2027 return mktime(&then);
2028}
2029
2030static NOINLINE var *exec_builtin(node *op, var *res)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002031{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002032#define tspl (G.exec_builtin__tspl)
2033
Glenn L McGrath545106f2002-11-11 06:21:00 +00002034 var *tv;
2035 node *an[4];
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002036 var *av[4];
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002037 const char *as[4];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002038 regmatch_t pmatch[2];
2039 regex_t sreg, *re;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002040 node *spl;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002041 uint32_t isr, info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002042 int nargs;
2043 time_t tt;
2044 char *s, *s1;
2045 int i, l, ll, n;
2046
2047 tv = nvalloc(4);
2048 isr = info = op->info;
2049 op = op->l.n;
2050
2051 av[2] = av[3] = NULL;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002052 for (i = 0; i < 4 && op; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002053 an[i] = nextarg(&op);
2054 if (isr & 0x09000000) av[i] = evaluate(an[i], &tv[i]);
2055 if (isr & 0x08000000) as[i] = getvar_s(av[i]);
2056 isr >>= 1;
2057 }
2058
2059 nargs = i;
Denis Vlasenko77ad97f2008-05-13 02:27:31 +00002060 if ((uint32_t)nargs < (info >> 30))
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002061 syntax_error(EMSG_TOO_FEW_ARGS);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002062
Denys Vlasenko56b3eec2009-10-23 13:03:59 +02002063 info &= OPNMASK;
2064 switch (info) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002065
Denis Vlasenkof782f522007-01-01 23:51:30 +00002066 case B_a2:
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002067#if ENABLE_FEATURE_AWK_LIBM
Denis Vlasenko37890e22008-10-21 12:59:34 +00002068 setvar_i(res, atan2(getvar_i(av[0]), getvar_i(av[1])));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002069#else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002070 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002071#endif
2072 break;
2073
Denis Vlasenkof782f522007-01-01 23:51:30 +00002074 case B_sp:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002075 if (nargs > 2) {
2076 spl = (an[2]->info & OPCLSMASK) == OC_REGEXP ?
2077 an[2] : mk_splitter(getvar_s(evaluate(an[2], &tv[2])), &tspl);
2078 } else {
2079 spl = &fsplitter.n;
2080 }
2081
2082 n = awk_split(as[0], spl, &s);
2083 s1 = s;
2084 clear_array(iamarray(av[1]));
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002085 for (i = 1; i <= n; i++)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002086 setari_u(av[1], i, nextword(&s1));
2087 free(s);
2088 setvar_i(res, n);
2089 break;
2090
Denis Vlasenkof782f522007-01-01 23:51:30 +00002091 case B_ss:
Rob Landleya3896512006-05-07 20:20:34 +00002092 l = strlen(as[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002093 i = getvar_i(av[1]) - 1;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002094 if (i > l) i = l;
2095 if (i < 0) i = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002096 n = (nargs > 2) ? getvar_i(av[2]) : l-i;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002097 if (n < 0) n = 0;
Denis Vlasenko8ae5b282008-07-02 22:47:49 +00002098 s = xstrndup(as[0]+i, n);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002099 setvar_p(res, s);
2100 break;
Denis Vlasenkof7996f32007-01-11 17:20:00 +00002101
Denis Vlasenko7cbcd1c2008-08-28 23:16:58 +00002102 /* Bitwise ops must assume that operands are unsigned. GNU Awk 3.1.5:
2103 * awk '{ print or(-1,1) }' gives "4.29497e+09", not "-2.xxxe+09" */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002104 case B_an:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002105 setvar_i(res, getvar_i_int(av[0]) & getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002106 break;
Denis Vlasenkof7996f32007-01-11 17:20:00 +00002107
Denis Vlasenkof782f522007-01-01 23:51:30 +00002108 case B_co:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002109 setvar_i(res, ~getvar_i_int(av[0]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002110 break;
2111
Denis Vlasenkof782f522007-01-01 23:51:30 +00002112 case B_ls:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002113 setvar_i(res, getvar_i_int(av[0]) << getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002114 break;
2115
Denis Vlasenkof782f522007-01-01 23:51:30 +00002116 case B_or:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002117 setvar_i(res, getvar_i_int(av[0]) | getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002118 break;
2119
Denis Vlasenkof782f522007-01-01 23:51:30 +00002120 case B_rs:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002121 setvar_i(res, getvar_i_int(av[0]) >> getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002122 break;
2123
Denis Vlasenkof782f522007-01-01 23:51:30 +00002124 case B_xo:
Denis Vlasenkoa2e1eea2008-09-02 09:00:23 +00002125 setvar_i(res, getvar_i_int(av[0]) ^ getvar_i_int(av[1]));
Denis Vlasenkoe175ff22006-09-26 17:41:00 +00002126 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002127
Denis Vlasenkof782f522007-01-01 23:51:30 +00002128 case B_lo:
Denis Vlasenkof782f522007-01-01 23:51:30 +00002129 case B_up:
Rob Landleyd921b2e2006-08-03 15:41:12 +00002130 s1 = s = xstrdup(as[0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002131 while (*s1) {
Denys Vlasenko56b3eec2009-10-23 13:03:59 +02002132 //*s1 = (info == B_up) ? toupper(*s1) : tolower(*s1);
2133 if ((unsigned char)((*s1 | 0x20) - 'a') <= ('z' - 'a'))
2134 *s1 = (info == B_up) ? (*s1 & 0xdf) : (*s1 | 0x20);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002135 s1++;
2136 }
2137 setvar_p(res, s);
2138 break;
2139
Denis Vlasenkof782f522007-01-01 23:51:30 +00002140 case B_ix:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002141 n = 0;
Rob Landleya3896512006-05-07 20:20:34 +00002142 ll = strlen(as[1]);
2143 l = strlen(as[0]) - ll;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002144 if (ll > 0 && l >= 0) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002145 if (!icase) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002146 s = strstr(as[0], as[1]);
2147 if (s) n = (s - as[0]) + 1;
2148 } else {
2149 /* this piece of code is terribly slow and
2150 * really should be rewritten
2151 */
2152 for (i=0; i<=l; i++) {
2153 if (strncasecmp(as[0]+i, as[1], ll) == 0) {
2154 n = i+1;
2155 break;
2156 }
2157 }
2158 }
2159 }
2160 setvar_i(res, n);
2161 break;
2162
Denis Vlasenkof782f522007-01-01 23:51:30 +00002163 case B_ti:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002164 if (nargs > 1)
2165 tt = getvar_i(av[1]);
2166 else
2167 time(&tt);
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002168 //s = (nargs > 0) ? as[0] : "%a %b %d %H:%M:%S %Z %Y";
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002169 i = strftime(g_buf, MAXVARFMT,
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002170 ((nargs > 0) ? as[0] : "%a %b %d %H:%M:%S %Z %Y"),
2171 localtime(&tt));
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002172 g_buf[i] = '\0';
2173 setvar_s(res, g_buf);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002174 break;
2175
Leonid Lisovskiy46a0be52009-09-21 04:08:08 +02002176 case B_mt:
2177 setvar_i(res, do_mktime(as[0]));
2178 break;
2179
Denis Vlasenkof782f522007-01-01 23:51:30 +00002180 case B_ma:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002181 re = as_regex(an[1], &sreg);
2182 n = regexec(re, as[0], 1, pmatch, 0);
2183 if (n == 0) {
2184 pmatch[0].rm_so++;
2185 pmatch[0].rm_eo++;
2186 } else {
2187 pmatch[0].rm_so = 0;
2188 pmatch[0].rm_eo = -1;
2189 }
2190 setvar_i(newvar("RSTART"), pmatch[0].rm_so);
2191 setvar_i(newvar("RLENGTH"), pmatch[0].rm_eo - pmatch[0].rm_so);
2192 setvar_i(res, pmatch[0].rm_so);
2193 if (re == &sreg) regfree(re);
2194 break;
2195
Denis Vlasenkof782f522007-01-01 23:51:30 +00002196 case B_ge:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002197 awk_sub(an[0], as[1], getvar_i(av[2]), av[3], res, TRUE);
2198 break;
2199
Denis Vlasenkof782f522007-01-01 23:51:30 +00002200 case B_gs:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002201 setvar_i(res, awk_sub(an[0], as[1], 0, av[2], av[2], FALSE));
2202 break;
2203
Denis Vlasenkof782f522007-01-01 23:51:30 +00002204 case B_su:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002205 setvar_i(res, awk_sub(an[0], as[1], 1, av[2], av[2], FALSE));
2206 break;
2207 }
2208
2209 nvfree(tv);
2210 return res;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002211#undef tspl
Glenn L McGrath545106f2002-11-11 06:21:00 +00002212}
2213
2214/*
2215 * Evaluate node - the heart of the program. Supplied with subtree
2216 * and place where to store result. returns ptr to result.
2217 */
2218#define XC(n) ((n) >> 8)
2219
Mike Frysinger10a11e22005-09-27 02:23:02 +00002220static var *evaluate(node *op, var *res)
2221{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002222/* This procedure is recursive so we should count every byte */
2223#define fnargs (G.evaluate__fnargs)
2224/* seed is initialized to 1 */
2225#define seed (G.evaluate__seed)
2226#define sreg (G.evaluate__sreg)
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002227
Glenn L McGrath545106f2002-11-11 06:21:00 +00002228 node *op1;
2229 var *v1;
2230 union {
2231 var *v;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002232 const char *s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002233 double d;
2234 int i;
2235 } L, R;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002236 uint32_t opinfo;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002237 int opn;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002238 union {
2239 char *s;
2240 rstream *rsm;
2241 FILE *F;
2242 var *v;
2243 regex_t *re;
Mike Frysingerf87b3e32005-09-27 04:16:22 +00002244 uint32_t info;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002245 } X;
2246
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002247 if (!op)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002248 return setvar_s(res, NULL);
2249
2250 v1 = nvalloc(2);
2251
2252 while (op) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002253 opinfo = op->info;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002254 opn = (opinfo & OPNMASK);
2255 g_lineno = op->lineno;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002256
Mike Frysingerde2b9382005-09-27 03:18:00 +00002257 /* execute inevitable things */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002258 op1 = op->l.n;
2259 if (opinfo & OF_RES1) X.v = L.v = evaluate(op1, v1);
2260 if (opinfo & OF_RES2) R.v = evaluate(op->r.n, v1+1);
2261 if (opinfo & OF_STR1) L.s = getvar_s(L.v);
2262 if (opinfo & OF_STR2) R.s = getvar_s(R.v);
2263 if (opinfo & OF_NUM1) L.d = getvar_i(L.v);
2264
2265 switch (XC(opinfo & OPCLSMASK)) {
2266
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002267 /* -- iterative node type -- */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002268
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002269 /* test pattern */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002270 case XC( OC_TEST ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002271 if ((op1->info & OPCLSMASK) == OC_COMMA) {
2272 /* it's range pattern */
2273 if ((opinfo & OF_CHECKED) || ptest(op1->l.n)) {
2274 op->info |= OF_CHECKED;
2275 if (ptest(op1->r.n))
2276 op->info &= ~OF_CHECKED;
2277
2278 op = op->a.n;
2279 } else {
2280 op = op->r.n;
2281 }
2282 } else {
2283 op = (ptest(op1)) ? op->a.n : op->r.n;
2284 }
2285 break;
2286
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002287 /* just evaluate an expression, also used as unconditional jump */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002288 case XC( OC_EXEC ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002289 break;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002290
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002291 /* branch, used in if-else and various loops */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002292 case XC( OC_BR ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002293 op = istrue(L.v) ? op->a.n : op->r.n;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002294 break;
2295
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002296 /* initialize for-in loop */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002297 case XC( OC_WALKINIT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002298 hashwalk_init(L.v, iamarray(R.v));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002299 break;
2300
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002301 /* get next array item */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002302 case XC( OC_WALKNEXT ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002303 op = hashwalk_next(L.v) ? op->a.n : op->r.n;
2304 break;
2305
Denis Vlasenkof782f522007-01-01 23:51:30 +00002306 case XC( OC_PRINT ):
2307 case XC( OC_PRINTF ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002308 X.F = stdout;
Mike Frysingerde2b9382005-09-27 03:18:00 +00002309 if (op->r.n) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002310 X.rsm = newfile(R.s);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002311 if (!X.rsm->F) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002312 if (opn == '|') {
Denis Vlasenko51742f42007-04-12 00:32:05 +00002313 X.rsm->F = popen(R.s, "w");
2314 if (X.rsm->F == NULL)
Manuel Novoa III cad53642003-03-19 09:13:01 +00002315 bb_perror_msg_and_die("popen");
Glenn L McGrath545106f2002-11-11 06:21:00 +00002316 X.rsm->is_pipe = 1;
2317 } else {
Rob Landleyd921b2e2006-08-03 15:41:12 +00002318 X.rsm->F = xfopen(R.s, opn=='w' ? "w" : "a");
Glenn L McGrath545106f2002-11-11 06:21:00 +00002319 }
2320 }
2321 X.F = X.rsm->F;
2322 }
2323
2324 if ((opinfo & OPCLSMASK) == OC_PRINT) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002325 if (!op1) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002326 fputs(getvar_s(intvar[F0]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002327 } else {
2328 while (op1) {
2329 L.v = evaluate(nextarg(&op1), v1);
2330 if (L.v->type & VF_NUMBER) {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002331 fmt_num(g_buf, MAXVARFMT, getvar_s(intvar[OFMT]),
Denis Vlasenkob54b2082006-10-27 09:05:40 +00002332 getvar_i(L.v), TRUE);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002333 fputs(g_buf, X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002334 } else {
2335 fputs(getvar_s(L.v), X.F);
2336 }
2337
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002338 if (op1) fputs(getvar_s(intvar[OFS]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002339 }
2340 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002341 fputs(getvar_s(intvar[ORS]), X.F);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002342
2343 } else { /* OC_PRINTF */
2344 L.s = awk_printf(op1);
2345 fputs(L.s, X.F);
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002346 free((char*)L.s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002347 }
2348 fflush(X.F);
2349 break;
2350
Denis Vlasenkof782f522007-01-01 23:51:30 +00002351 case XC( OC_DELETE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002352 X.info = op1->info & OPCLSMASK;
2353 if (X.info == OC_VAR) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002354 R.v = op1->l.v;
2355 } else if (X.info == OC_FNARG) {
2356 R.v = &fnargs[op1->l.i];
2357 } else {
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002358 syntax_error(EMSG_NOT_ARRAY);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002359 }
2360
Mike Frysingerde2b9382005-09-27 03:18:00 +00002361 if (op1->r.n) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002362 clrvar(L.v);
2363 L.s = getvar_s(evaluate(op1->r.n, v1));
2364 hash_remove(iamarray(R.v), L.s);
2365 } else {
2366 clear_array(iamarray(R.v));
2367 }
2368 break;
2369
Denis Vlasenkof782f522007-01-01 23:51:30 +00002370 case XC( OC_NEWSOURCE ):
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002371 g_progname = op->l.s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002372 break;
2373
Denis Vlasenkof782f522007-01-01 23:51:30 +00002374 case XC( OC_RETURN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002375 copyvar(res, L.v);
2376 break;
2377
Denis Vlasenkof782f522007-01-01 23:51:30 +00002378 case XC( OC_NEXTFILE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002379 nextfile = TRUE;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002380 case XC( OC_NEXT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002381 nextrec = TRUE;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002382 case XC( OC_DONE ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002383 clrvar(res);
2384 break;
2385
Denis Vlasenkof782f522007-01-01 23:51:30 +00002386 case XC( OC_EXIT ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002387 awk_exit(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002388
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002389 /* -- recursive node type -- */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002390
Denis Vlasenkof782f522007-01-01 23:51:30 +00002391 case XC( OC_VAR ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002392 L.v = op->l.v;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002393 if (L.v == intvar[NF])
Glenn L McGrath545106f2002-11-11 06:21:00 +00002394 split_f0();
2395 goto v_cont;
2396
Denis Vlasenkof782f522007-01-01 23:51:30 +00002397 case XC( OC_FNARG ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002398 L.v = &fnargs[op->l.i];
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002399 v_cont:
2400 res = op->r.n ? findvar(iamarray(L.v), R.s) : L.v;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002401 break;
2402
Denis Vlasenkof782f522007-01-01 23:51:30 +00002403 case XC( OC_IN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002404 setvar_i(res, hash_search(iamarray(R.v), L.s) ? 1 : 0);
2405 break;
2406
Denis Vlasenkof782f522007-01-01 23:51:30 +00002407 case XC( OC_REGEXP ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002408 op1 = op;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002409 L.s = getvar_s(intvar[F0]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002410 goto re_cont;
2411
Denis Vlasenkof782f522007-01-01 23:51:30 +00002412 case XC( OC_MATCH ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002413 op1 = op->r.n;
Denis Vlasenkoe1d3e032007-01-01 23:53:52 +00002414 re_cont:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002415 X.re = as_regex(op1, &sreg);
2416 R.i = regexec(X.re, L.s, 0, NULL, 0);
2417 if (X.re == &sreg) regfree(X.re);
2418 setvar_i(res, (R.i == 0 ? 1 : 0) ^ (opn == '!' ? 1 : 0));
2419 break;
2420
Denis Vlasenkof782f522007-01-01 23:51:30 +00002421 case XC( OC_MOVE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002422 /* if source is a temporary string, jusk relink it to dest */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002423 if (R.v == v1+1 && R.v->string) {
2424 res = setvar_p(L.v, R.v->string);
2425 R.v->string = NULL;
2426 } else {
Mike Frysingerde2b9382005-09-27 03:18:00 +00002427 res = copyvar(L.v, R.v);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002428 }
2429 break;
2430
Denis Vlasenkof782f522007-01-01 23:51:30 +00002431 case XC( OC_TERNARY ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002432 if ((op->r.n->info & OPCLSMASK) != OC_COLON)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002433 syntax_error(EMSG_POSSIBLE_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002434 res = evaluate(istrue(L.v) ? op->r.n->l.n : op->r.n->r.n, res);
2435 break;
2436
Denis Vlasenkof782f522007-01-01 23:51:30 +00002437 case XC( OC_FUNC ):
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002438 if (!op->r.f->body.first)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002439 syntax_error(EMSG_UNDEF_FUNC);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002440
2441 X.v = R.v = nvalloc(op->r.f->nargs+1);
2442 while (op1) {
2443 L.v = evaluate(nextarg(&op1), v1);
2444 copyvar(R.v, L.v);
2445 R.v->type |= VF_CHILD;
2446 R.v->x.parent = L.v;
2447 if (++R.v - X.v >= op->r.f->nargs)
2448 break;
2449 }
2450
2451 R.v = fnargs;
2452 fnargs = X.v;
2453
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002454 L.s = g_progname;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002455 res = evaluate(op->r.f->body.first, res);
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002456 g_progname = L.s;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002457
2458 nvfree(fnargs);
2459 fnargs = R.v;
2460 break;
2461
Denis Vlasenkof782f522007-01-01 23:51:30 +00002462 case XC( OC_GETLINE ):
2463 case XC( OC_PGETLINE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002464 if (op1) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002465 X.rsm = newfile(L.s);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002466 if (!X.rsm->F) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002467 if ((opinfo & OPCLSMASK) == OC_PGETLINE) {
2468 X.rsm->F = popen(L.s, "r");
2469 X.rsm->is_pipe = TRUE;
2470 } else {
Denis Vlasenko5415c852008-07-21 23:05:26 +00002471 X.rsm->F = fopen_for_read(L.s); /* not xfopen! */
Glenn L McGrath545106f2002-11-11 06:21:00 +00002472 }
2473 }
2474 } else {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002475 if (!iF) iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002476 X.rsm = iF;
2477 }
2478
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002479 if (!X.rsm->F) {
2480 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002481 setvar_i(res, -1);
2482 break;
2483 }
2484
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002485 if (!op->r.n)
2486 R.v = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002487
2488 L.i = awk_getline(X.rsm, R.v);
2489 if (L.i > 0) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002490 if (!op1) {
2491 incvar(intvar[FNR]);
2492 incvar(intvar[NR]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002493 }
2494 }
2495 setvar_i(res, L.i);
2496 break;
2497
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002498 /* simple builtins */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002499 case XC( OC_FBLTIN ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002500 switch (opn) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002501
Denis Vlasenkof782f522007-01-01 23:51:30 +00002502 case F_in:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002503 R.d = (int)L.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002504 break;
2505
Denis Vlasenkof782f522007-01-01 23:51:30 +00002506 case F_rn:
2507 R.d = (double)rand() / (double)RAND_MAX;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002508 break;
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002509#if ENABLE_FEATURE_AWK_LIBM
Denis Vlasenkof782f522007-01-01 23:51:30 +00002510 case F_co:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002511 R.d = cos(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002512 break;
2513
Denis Vlasenkof782f522007-01-01 23:51:30 +00002514 case F_ex:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002515 R.d = exp(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002516 break;
2517
Denis Vlasenkof782f522007-01-01 23:51:30 +00002518 case F_lg:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002519 R.d = log(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002520 break;
2521
Denis Vlasenkof782f522007-01-01 23:51:30 +00002522 case F_si:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002523 R.d = sin(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002524 break;
2525
Denis Vlasenkof782f522007-01-01 23:51:30 +00002526 case F_sq:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002527 R.d = sqrt(L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002528 break;
2529#else
Denis Vlasenkof782f522007-01-01 23:51:30 +00002530 case F_co:
2531 case F_ex:
2532 case F_lg:
2533 case F_si:
2534 case F_sq:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002535 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002536 break;
2537#endif
Denis Vlasenkof782f522007-01-01 23:51:30 +00002538 case F_sr:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002539 R.d = (double)seed;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002540 seed = op1 ? (unsigned)L.d : (unsigned)time(NULL);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002541 srand(seed);
2542 break;
2543
Denis Vlasenkof782f522007-01-01 23:51:30 +00002544 case F_ti:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002545 R.d = time(NULL);
2546 break;
2547
Denis Vlasenkof782f522007-01-01 23:51:30 +00002548 case F_le:
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002549 if (!op1)
2550 L.s = getvar_s(intvar[F0]);
Rob Landleya3896512006-05-07 20:20:34 +00002551 R.d = strlen(L.s);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002552 break;
2553
Denis Vlasenkof782f522007-01-01 23:51:30 +00002554 case F_sy:
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002555 fflush_all();
Denis Vlasenko249fabf2006-12-19 00:29:22 +00002556 R.d = (ENABLE_FEATURE_ALLOW_EXEC && L.s && *L.s)
2557 ? (system(L.s) >> 8) : 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002558 break;
2559
Denis Vlasenkof782f522007-01-01 23:51:30 +00002560 case F_ff:
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002561 if (!op1)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002562 fflush(stdout);
2563 else {
2564 if (L.s && *L.s) {
2565 X.rsm = newfile(L.s);
2566 fflush(X.rsm->F);
2567 } else {
Denys Vlasenko8131eea2009-11-02 14:19:51 +01002568 fflush_all();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002569 }
2570 }
2571 break;
2572
Denis Vlasenkof782f522007-01-01 23:51:30 +00002573 case F_cl:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002574 X.rsm = (rstream *)hash_search(fdhash, L.s);
2575 if (X.rsm) {
2576 R.i = X.rsm->is_pipe ? pclose(X.rsm->F) : fclose(X.rsm->F);
Aaron Lehmanna170e1c2002-11-28 11:27:31 +00002577 free(X.rsm->buffer);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002578 hash_remove(fdhash, L.s);
2579 }
2580 if (R.i != 0)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002581 setvar_i(intvar[ERRNO], errno);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002582 R.d = (double)R.i;
2583 break;
2584 }
2585 setvar_i(res, R.d);
2586 break;
2587
Denis Vlasenkof782f522007-01-01 23:51:30 +00002588 case XC( OC_BUILTIN ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002589 res = exec_builtin(op, res);
2590 break;
2591
Denis Vlasenkof782f522007-01-01 23:51:30 +00002592 case XC( OC_SPRINTF ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002593 setvar_p(res, awk_printf(op1));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002594 break;
2595
Denis Vlasenkof782f522007-01-01 23:51:30 +00002596 case XC( OC_UNARY ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002597 X.v = R.v;
2598 L.d = R.d = getvar_i(R.v);
2599 switch (opn) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002600 case 'P':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002601 L.d = ++R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002602 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002603 case 'p':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002604 R.d++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002605 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002606 case 'M':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002607 L.d = --R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002608 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002609 case 'm':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002610 R.d--;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002611 goto r_op_change;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002612 case '!':
Denis Vlasenko92758142006-10-03 19:56:34 +00002613 L.d = istrue(X.v) ? 0 : 1;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002614 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002615 case '-':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002616 L.d = -R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002617 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002618 r_op_change:
Glenn L McGrath545106f2002-11-11 06:21:00 +00002619 setvar_i(X.v, R.d);
2620 }
2621 setvar_i(res, L.d);
2622 break;
2623
Denis Vlasenkof782f522007-01-01 23:51:30 +00002624 case XC( OC_FIELD ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002625 R.i = (int)getvar_i(R.v);
2626 if (R.i == 0) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002627 res = intvar[F0];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002628 } else {
2629 split_f0();
2630 if (R.i > nfields)
2631 fsrealloc(R.i);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002632 res = &Fields[R.i - 1];
Glenn L McGrath545106f2002-11-11 06:21:00 +00002633 }
2634 break;
2635
Denis Vlasenkocd5c7862007-05-17 16:37:22 +00002636 /* concatenation (" ") and index joining (",") */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002637 case XC( OC_CONCAT ):
2638 case XC( OC_COMMA ):
Rob Landleya3896512006-05-07 20:20:34 +00002639 opn = strlen(L.s) + strlen(R.s) + 2;
Denis Vlasenkob95636c2006-12-19 23:36:04 +00002640 X.s = xmalloc(opn);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002641 strcpy(X.s, L.s);
2642 if ((opinfo & OPCLSMASK) == OC_COMMA) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002643 L.s = getvar_s(intvar[SUBSEP]);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002644 X.s = xrealloc(X.s, opn + strlen(L.s));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002645 strcat(X.s, L.s);
2646 }
2647 strcat(X.s, R.s);
2648 setvar_p(res, X.s);
2649 break;
2650
Denis Vlasenkof782f522007-01-01 23:51:30 +00002651 case XC( OC_LAND ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002652 setvar_i(res, istrue(L.v) ? ptest(op->r.n) : 0);
2653 break;
2654
Denis Vlasenkof782f522007-01-01 23:51:30 +00002655 case XC( OC_LOR ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002656 setvar_i(res, istrue(L.v) ? 1 : ptest(op->r.n));
2657 break;
2658
Denis Vlasenkof782f522007-01-01 23:51:30 +00002659 case XC( OC_BINARY ):
2660 case XC( OC_REPLACE ):
Mike Frysingerde2b9382005-09-27 03:18:00 +00002661 R.d = getvar_i(R.v);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002662 switch (opn) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002663 case '+':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002664 L.d += R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002665 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002666 case '-':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002667 L.d -= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002668 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002669 case '*':
Mike Frysingerde2b9382005-09-27 03:18:00 +00002670 L.d *= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002671 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002672 case '/':
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002673 if (R.d == 0) syntax_error(EMSG_DIV_BY_ZERO);
Mike Frysingerde2b9382005-09-27 03:18:00 +00002674 L.d /= R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002675 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002676 case '&':
Denis Vlasenko2d5bd802008-10-24 10:49:49 +00002677#if ENABLE_FEATURE_AWK_LIBM
Mike Frysingerde2b9382005-09-27 03:18:00 +00002678 L.d = pow(L.d, R.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002679#else
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002680 syntax_error(EMSG_NO_MATH);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002681#endif
2682 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002683 case '%':
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002684 if (R.d == 0) syntax_error(EMSG_DIV_BY_ZERO);
Mike Frysingerde2b9382005-09-27 03:18:00 +00002685 L.d -= (int)(L.d / R.d) * R.d;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002686 break;
2687 }
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002688 res = setvar_i(((opinfo & OPCLSMASK) == OC_BINARY) ? res : X.v, L.d);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002689 break;
2690
Denis Vlasenkof782f522007-01-01 23:51:30 +00002691 case XC( OC_COMPARE ):
Glenn L McGrath545106f2002-11-11 06:21:00 +00002692 if (is_numeric(L.v) && is_numeric(R.v)) {
2693 L.d = getvar_i(L.v) - getvar_i(R.v);
2694 } else {
2695 L.s = getvar_s(L.v);
2696 R.s = getvar_s(R.v);
2697 L.d = icase ? strcasecmp(L.s, R.s) : strcmp(L.s, R.s);
2698 }
2699 switch (opn & 0xfe) {
Denis Vlasenkof782f522007-01-01 23:51:30 +00002700 case 0:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002701 R.i = (L.d > 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002702 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002703 case 2:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002704 R.i = (L.d >= 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002705 break;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002706 case 4:
Mike Frysingerde2b9382005-09-27 03:18:00 +00002707 R.i = (L.d == 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002708 break;
2709 }
2710 setvar_i(res, (opn & 0x1 ? R.i : !R.i) ? 1 : 0);
2711 break;
2712
Denis Vlasenkof782f522007-01-01 23:51:30 +00002713 default:
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002714 syntax_error(EMSG_POSSIBLE_ERROR);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002715 }
2716 if ((opinfo & OPCLSMASK) <= SHIFT_TIL_THIS)
2717 op = op->a.n;
2718 if ((opinfo & OPCLSMASK) >= RECUR_FROM_THIS)
2719 break;
2720 if (nextrec)
2721 break;
2722 }
2723 nvfree(v1);
2724 return res;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002725#undef fnargs
2726#undef seed
2727#undef sreg
Glenn L McGrath545106f2002-11-11 06:21:00 +00002728}
2729
2730
2731/* -------- main & co. -------- */
2732
Mike Frysinger10a11e22005-09-27 02:23:02 +00002733static int awk_exit(int r)
2734{
Denis Vlasenkof782f522007-01-01 23:51:30 +00002735 var tv;
2736 unsigned i;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002737 hash_item *hi;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002738
Denis Vlasenkof782f522007-01-01 23:51:30 +00002739 zero_out_var(&tv);
2740
2741 if (!exiting) {
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002742 exiting = TRUE;
Glenn L McGrathca29ffc2004-09-24 09:24:27 +00002743 nextrec = FALSE;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002744 evaluate(endseq.first, &tv);
2745 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002746
2747 /* waiting for children */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002748 for (i = 0; i < fdhash->csize; i++) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002749 hi = fdhash->items[i];
Denis Vlasenkobf0a2012006-12-26 10:42:51 +00002750 while (hi) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002751 if (hi->data.rs.F && hi->data.rs.is_pipe)
2752 pclose(hi->data.rs.F);
2753 hi = hi->next;
2754 }
2755 }
2756
2757 exit(r);
2758}
2759
2760/* if expr looks like "var=value", perform assignment and return 1,
2761 * otherwise return 0 */
"Vladimir N. Oleynik"5cf9a032005-10-19 09:21:51 +00002762static int is_assignment(const char *expr)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002763{
Glenn L McGrath545106f2002-11-11 06:21:00 +00002764 char *exprc, *s, *s0, *s1;
2765
Rob Landleyd921b2e2006-08-03 15:41:12 +00002766 exprc = xstrdup(expr);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002767 if (!isalnum_(*exprc) || (s = strchr(exprc, '=')) == NULL) {
2768 free(exprc);
2769 return FALSE;
2770 }
2771
2772 *(s++) = '\0';
2773 s0 = s1 = s;
2774 while (*s)
2775 *(s1++) = nextchar(&s);
2776
2777 *s1 = '\0';
2778 setvar_u(newvar(exprc), s0);
2779 free(exprc);
2780 return TRUE;
2781}
2782
2783/* switch to next input file */
Mike Frysinger10a11e22005-09-27 02:23:02 +00002784static rstream *next_input_file(void)
2785{
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002786#define rsm (G.next_input_file__rsm)
2787#define files_happen (G.next_input_file__files_happen)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002788
Glenn L McGrath545106f2002-11-11 06:21:00 +00002789 FILE *F = NULL;
Denis Vlasenkoa41fdf32007-01-29 22:51:00 +00002790 const char *fname, *ind;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002791
2792 if (rsm.F) fclose(rsm.F);
2793 rsm.F = NULL;
Glenn L McGrath00ed36f2003-10-30 13:36:39 +00002794 rsm.pos = rsm.adv = 0;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002795
2796 do {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002797 if (getvar_i(intvar[ARGIND])+1 >= getvar_i(intvar[ARGC])) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002798 if (files_happen)
2799 return NULL;
2800 fname = "-";
2801 F = stdin;
2802 } else {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002803 ind = getvar_s(incvar(intvar[ARGIND]));
2804 fname = getvar_s(findvar(iamarray(intvar[ARGV]), ind));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002805 if (fname && *fname && !is_assignment(fname))
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002806 F = xfopen_stdin(fname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002807 }
2808 } while (!F);
2809
2810 files_happen = TRUE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002811 setvar_s(intvar[FILENAME], fname);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002812 rsm.F = F;
2813 return &rsm;
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002814#undef rsm
2815#undef files_happen
Glenn L McGrath545106f2002-11-11 06:21:00 +00002816}
2817
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +00002818int awk_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Rob Landleydfba7412006-03-06 20:47:33 +00002819int awk_main(int argc, char **argv)
Mike Frysinger10a11e22005-09-27 02:23:02 +00002820{
Denis Vlasenko67b23e62006-10-03 21:00:06 +00002821 unsigned opt;
Denis Vlasenkobe644a82007-03-10 17:22:14 +00002822 char *opt_F, *opt_W;
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002823 llist_t *list_v = NULL;
2824 llist_t *list_f = NULL;
2825 int i, j;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002826 var *v;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002827 var tv;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002828 char **envp;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002829 char *vnames = (char *)vNames; /* cheat */
2830 char *vvalues = (char *)vValues;
2831
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002832 INIT_G();
2833
Denis Vlasenko150f4022007-01-13 21:06:21 +00002834 /* Undo busybox.c, or else strtod may eat ','! This breaks parsing:
Denis Vlasenko6dc6ebb2007-01-01 23:53:12 +00002835 * $1,$2 == '$1,' '$2', NOT '$1' ',' '$2' */
2836 if (ENABLE_LOCALE_SUPPORT)
2837 setlocale(LC_NUMERIC, "C");
2838
Denis Vlasenkof782f522007-01-01 23:51:30 +00002839 zero_out_var(&tv);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002840
2841 /* allocate global buffer */
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002842 g_buf = xmalloc(MAXVARFMT + 1);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002843
2844 vhash = hash_init();
2845 ahash = hash_init();
2846 fdhash = hash_init();
2847 fnhash = hash_init();
2848
2849 /* initialize variables */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002850 for (i = 0; *vnames; i++) {
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002851 intvar[i] = v = newvar(nextword(&vnames));
Denis Vlasenkof782f522007-01-01 23:51:30 +00002852 if (*vvalues != '\377')
2853 setvar_s(v, nextword(&vvalues));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002854 else
2855 setvar_i(v, 0);
2856
Denis Vlasenkof782f522007-01-01 23:51:30 +00002857 if (*vnames == '*') {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002858 v->type |= VF_SPECIAL;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002859 vnames++;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002860 }
2861 }
2862
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002863 handle_special(intvar[FS]);
2864 handle_special(intvar[RS]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002865
Denis Vlasenkof782f522007-01-01 23:51:30 +00002866 newfile("/dev/stdin")->F = stdin;
2867 newfile("/dev/stdout")->F = stdout;
2868 newfile("/dev/stderr")->F = stderr;
Glenn L McGrath545106f2002-11-11 06:21:00 +00002869
Denis Vlasenkof71d9162007-05-03 22:57:56 +00002870 /* Huh, people report that sometimes environ is NULL. Oh well. */
2871 if (environ) for (envp = environ; *envp; envp++) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002872 /* environ is writable, thus we don't strdup it needlessly */
2873 char *s = *envp;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002874 char *s1 = strchr(s, '=');
2875 if (s1) {
Denis Vlasenkob78c7822007-07-18 18:31:11 +00002876 *s1 = '\0';
2877 /* Both findvar and setvar_u take const char*
2878 * as 2nd arg -> environment is not trashed */
2879 setvar_u(findvar(iamarray(intvar[ENVIRON]), s), s1 + 1);
2880 *s1 = '=';
Eric Andersen67776be2004-07-30 23:52:08 +00002881 }
Glenn L McGrath545106f2002-11-11 06:21:00 +00002882 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002883 opt_complementary = "v::f::"; /* -v and -f can occur multiple times */
2884 opt = getopt32(argv, "F:v:f:W:", &opt_F, &list_v, &list_f, &opt_W);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002885 argv += optind;
2886 argc -= optind;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002887 if (opt & 0x1)
2888 setvar_s(intvar[FS], opt_F); // -F
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002889 while (list_v) { /* -v */
2890 if (!is_assignment(llist_pop(&list_v)))
Denis Vlasenkobe644a82007-03-10 17:22:14 +00002891 bb_show_usage();
2892 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002893 if (list_f) { /* -f */
2894 do {
2895 char *s = NULL;
2896 FILE *from_file;
2897
2898 g_progname = llist_pop(&list_f);
2899 from_file = xfopen_stdin(g_progname);
2900 /* one byte is reserved for some trick in next_token */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002901 for (i = j = 1; j > 0; i += j) {
2902 s = xrealloc(s, i + 4096);
2903 j = fread(s + i, 1, 4094, from_file);
Denis Vlasenko099efbf2006-09-22 09:02:30 +00002904 }
Denis Vlasenko3bb2bbd2008-07-01 01:57:36 +00002905 s[i] = '\0';
2906 fclose(from_file);
2907 parse_program(s + 1);
2908 free(s);
2909 } while (list_f);
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +00002910 argc++;
Denis Vlasenkof782f522007-01-01 23:51:30 +00002911 } else { // no -f: take program from 1st parameter
2912 if (!argc)
2913 bb_show_usage();
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002914 g_progname = "cmd. line";
Denis Vlasenkof782f522007-01-01 23:51:30 +00002915 parse_program(*argv++);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002916 }
Denis Vlasenko099efbf2006-09-22 09:02:30 +00002917 if (opt & 0x8) // -W
Denis Vlasenkoe1a0d482006-10-20 13:28:22 +00002918 bb_error_msg("warning: unrecognized option '-W %s' ignored", opt_W);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002919
Glenn L McGrath545106f2002-11-11 06:21:00 +00002920 /* fill in ARGV array */
Denis Vlasenko41d5ebe2009-01-25 01:00:15 +00002921 setvar_i(intvar[ARGC], argc);
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002922 setari_u(intvar[ARGV], 0, "awk");
Denis Vlasenkof782f522007-01-01 23:51:30 +00002923 i = 0;
2924 while (*argv)
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002925 setari_u(intvar[ARGV], ++i, *argv++);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002926
2927 evaluate(beginseq.first, &tv);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002928 if (!mainseq.first && !endseq.first)
Glenn L McGrath545106f2002-11-11 06:21:00 +00002929 awk_exit(EXIT_SUCCESS);
2930
2931 /* input file could already be opened in BEGIN block */
Denis Vlasenkof782f522007-01-01 23:51:30 +00002932 if (!iF) iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002933
2934 /* passing through input files */
2935 while (iF) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002936 nextfile = FALSE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002937 setvar_i(intvar[FNR], 0);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002938
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002939 while ((i = awk_getline(iF, intvar[F0])) > 0) {
Glenn L McGrath545106f2002-11-11 06:21:00 +00002940 nextrec = FALSE;
Denis Vlasenkoffba9412007-05-17 23:03:35 +00002941 incvar(intvar[NR]);
2942 incvar(intvar[FNR]);
Glenn L McGrath545106f2002-11-11 06:21:00 +00002943 evaluate(mainseq.first, &tv);
2944
2945 if (nextfile)
2946 break;
2947 }
2948
Denis Vlasenkof782f522007-01-01 23:51:30 +00002949 if (i < 0)
Denis Vlasenkoae5a8aa2007-06-06 17:01:00 +00002950 syntax_error(strerror(errno));
Glenn L McGrath545106f2002-11-11 06:21:00 +00002951
2952 iF = next_input_file();
Glenn L McGrath545106f2002-11-11 06:21:00 +00002953 }
2954
Glenn L McGrath545106f2002-11-11 06:21:00 +00002955 awk_exit(EXIT_SUCCESS);
Denis Vlasenkof782f522007-01-01 23:51:30 +00002956 /*return 0;*/
Glenn L McGrath545106f2002-11-11 06:21:00 +00002957}