Commit | Line | Data |
0d3e774c |
1 | /* $Header: cons.c,v 3.0.1.2 89/11/17 15:08:53 lwall Locked $ |
a687059c |
2 | * |
3 | * Copyright (c) 1989, Larry Wall |
4 | * |
5 | * You may distribute under the terms of the GNU General Public License |
6 | * as specified in the README file that comes with the perl 3.0 kit. |
7 | * |
8 | * $Log: cons.c,v $ |
0d3e774c |
9 | * Revision 3.0.1.2 89/11/17 15:08:53 lwall |
10 | * patch5: nested foreach on same array didn't work |
11 | * |
03a14243 |
12 | * Revision 3.0.1.1 89/10/26 23:09:01 lwall |
13 | * patch1: numeric switch optimization was broken |
14 | * patch1: unless was broken when run under the debugger |
15 | * |
a687059c |
16 | * Revision 3.0 89/10/18 15:10:23 lwall |
17 | * 3.0 baseline |
18 | * |
19 | */ |
20 | |
21 | #include "EXTERN.h" |
22 | #include "perl.h" |
23 | #include "perly.h" |
24 | |
25 | extern char *tokename[]; |
26 | extern int yychar; |
27 | |
28 | static int cmd_tosave(); |
29 | static int arg_tosave(); |
30 | static int spat_tosave(); |
31 | |
32 | static bool saw_return; |
33 | |
34 | SUBR * |
35 | make_sub(name,cmd) |
36 | char *name; |
37 | CMD *cmd; |
38 | { |
39 | register SUBR *sub; |
40 | STAB *stab = stabent(name,TRUE); |
41 | |
42 | Newz(101,sub,1,SUBR); |
43 | if (stab_sub(stab)) { |
44 | if (dowarn) { |
45 | line_t oldline = line; |
46 | |
47 | if (cmd) |
48 | line = cmd->c_line; |
49 | warn("Subroutine %s redefined",name); |
50 | line = oldline; |
51 | } |
52 | cmd_free(stab_sub(stab)->cmd); |
53 | afree(stab_sub(stab)->tosave); |
54 | Safefree(stab_sub(stab)); |
55 | } |
56 | sub->filename = filename; |
57 | saw_return = FALSE; |
58 | tosave = anew(Nullstab); |
59 | tosave->ary_fill = 0; /* make 1 based */ |
60 | (void)cmd_tosave(cmd,FALSE); /* this builds the tosave array */ |
61 | sub->tosave = tosave; |
62 | if (saw_return) { |
63 | struct compcmd mycompblock; |
64 | |
65 | mycompblock.comp_true = cmd; |
66 | mycompblock.comp_alt = Nullcmd; |
67 | cmd = add_label(savestr("SUB"),make_ccmd(C_BLOCK,Nullarg,mycompblock)); |
68 | saw_return = FALSE; |
69 | } |
70 | sub->cmd = cmd; |
71 | stab_sub(stab) = sub; |
72 | if (perldb) { |
73 | STR *str = str_nmake((double)subline); |
74 | |
75 | str_cat(str,"-"); |
76 | sprintf(buf,"%ld",(long)line); |
77 | str_cat(str,buf); |
78 | name = str_get(subname); |
79 | hstore(stab_xhash(DBsub),name,strlen(name),str,0); |
80 | str_set(subname,"main"); |
81 | } |
82 | subline = 0; |
83 | return sub; |
84 | } |
85 | |
86 | CMD * |
87 | block_head(tail) |
88 | register CMD *tail; |
89 | { |
90 | CMD *head; |
91 | register int opt; |
92 | register int last_opt = 0; |
93 | register STAB *last_stab = Nullstab; |
94 | register int count = 0; |
95 | register CMD *switchbeg = Nullcmd; |
96 | |
97 | if (tail == Nullcmd) { |
98 | return tail; |
99 | } |
100 | head = tail->c_head; |
101 | |
102 | for (tail = head; tail; tail = tail->c_next) { |
103 | |
104 | /* save one measly dereference at runtime */ |
105 | if (tail->c_type == C_IF) { |
106 | if (!(tail->ucmd.ccmd.cc_alt = tail->ucmd.ccmd.cc_alt->c_next)) |
107 | tail->c_flags |= CF_TERM; |
108 | } |
109 | else if (tail->c_type == C_EXPR) { |
110 | ARG *arg; |
111 | |
112 | if (tail->ucmd.acmd.ac_expr) |
113 | arg = tail->ucmd.acmd.ac_expr; |
114 | else |
115 | arg = tail->c_expr; |
116 | if (arg) { |
117 | if (arg->arg_type == O_RETURN) |
118 | tail->c_flags |= CF_TERM; |
119 | else if (arg->arg_type == O_ITEM && arg[1].arg_type == A_CMD) |
120 | tail->c_flags |= CF_TERM; |
121 | } |
122 | } |
123 | if (!tail->c_next) |
124 | tail->c_flags |= CF_TERM; |
125 | |
126 | if (tail->c_expr && (tail->c_flags & CF_OPTIMIZE) == CFT_FALSE) |
127 | opt_arg(tail,1, tail->c_type == C_EXPR); |
128 | |
129 | /* now do a little optimization on case-ish structures */ |
130 | switch(tail->c_flags & (CF_OPTIMIZE|CF_FIRSTNEG|CF_INVERT)) { |
131 | case CFT_ANCHOR: |
132 | if (stabent("*",FALSE)) { /* bad assumption here!!! */ |
133 | opt = 0; |
134 | break; |
135 | } |
136 | /* FALL THROUGH */ |
137 | case CFT_STROP: |
138 | opt = (tail->c_flags & CF_NESURE) ? CFT_STROP : 0; |
139 | break; |
140 | case CFT_CCLASS: |
141 | opt = CFT_STROP; |
142 | break; |
143 | case CFT_NUMOP: |
144 | opt = (tail->c_slen == O_NE ? 0 : CFT_NUMOP); |
145 | if ((tail->c_flags&(CF_NESURE|CF_EQSURE)) != (CF_NESURE|CF_EQSURE)) |
146 | opt = 0; |
147 | break; |
148 | default: |
149 | opt = 0; |
150 | } |
151 | if (opt && opt == last_opt && tail->c_stab == last_stab) |
152 | count++; |
153 | else { |
154 | if (count >= 3) { /* is this the breakeven point? */ |
155 | if (last_opt == CFT_NUMOP) |
156 | make_nswitch(switchbeg,count); |
157 | else |
158 | make_cswitch(switchbeg,count); |
159 | } |
160 | if (opt) { |
161 | count = 1; |
162 | switchbeg = tail; |
163 | } |
164 | else |
165 | count = 0; |
166 | } |
167 | last_opt = opt; |
168 | last_stab = tail->c_stab; |
169 | } |
170 | if (count >= 3) { /* is this the breakeven point? */ |
171 | if (last_opt == CFT_NUMOP) |
172 | make_nswitch(switchbeg,count); |
173 | else |
174 | make_cswitch(switchbeg,count); |
175 | } |
176 | return head; |
177 | } |
178 | |
179 | /* We've spotted a sequence of CMDs that all test the value of the same |
180 | * spat. Thus we can insert a SWITCH in front and jump directly |
181 | * to the correct one. |
182 | */ |
183 | make_cswitch(head,count) |
184 | register CMD *head; |
185 | int count; |
186 | { |
187 | register CMD *cur; |
188 | register CMD **loc; |
189 | register int i; |
190 | register int min = 255; |
191 | register int max = 0; |
192 | |
193 | /* make a new head in the exact same spot */ |
194 | New(102,cur, 1, CMD); |
195 | #ifdef STRUCTCOPY |
196 | *cur = *head; |
197 | #else |
198 | Copy(head,cur,1,CMD); |
199 | #endif |
200 | Zero(head,1,CMD); |
201 | head->c_type = C_CSWITCH; |
202 | head->c_next = cur; /* insert new cmd at front of list */ |
203 | head->c_stab = cur->c_stab; |
204 | |
205 | Newz(103,loc,258,CMD*); |
206 | loc++; /* lie a little */ |
207 | while (count--) { |
208 | if ((cur->c_flags & CF_OPTIMIZE) == CFT_CCLASS) { |
209 | for (i = 0; i <= 255; i++) { |
210 | if (!loc[i] && cur->c_short->str_ptr[i>>3] & (1 << (i&7))) { |
211 | loc[i] = cur; |
212 | if (i < min) |
213 | min = i; |
214 | if (i > max) |
215 | max = i; |
216 | } |
217 | } |
218 | } |
219 | else { |
220 | i = *cur->c_short->str_ptr & 255; |
221 | if (!loc[i]) { |
222 | loc[i] = cur; |
223 | if (i < min) |
224 | min = i; |
225 | if (i > max) |
226 | max = i; |
227 | } |
228 | } |
229 | cur = cur->c_next; |
230 | } |
231 | max++; |
232 | if (min > 0) |
233 | Copy(&loc[min],&loc[0], max - min, CMD*); |
234 | loc--; |
235 | min--; |
236 | max -= min; |
237 | for (i = 0; i <= max; i++) |
238 | if (!loc[i]) |
239 | loc[i] = cur; |
240 | Renew(loc,max+1,CMD*); /* chop it down to size */ |
241 | head->ucmd.scmd.sc_offset = min; |
242 | head->ucmd.scmd.sc_max = max; |
243 | head->ucmd.scmd.sc_next = loc; |
244 | } |
245 | |
246 | make_nswitch(head,count) |
247 | register CMD *head; |
248 | int count; |
249 | { |
250 | register CMD *cur = head; |
251 | register CMD **loc; |
252 | register int i; |
253 | register int min = 32767; |
254 | register int max = -32768; |
255 | int origcount = count; |
256 | double value; /* or your money back! */ |
257 | short changed; /* so triple your money back! */ |
258 | |
259 | while (count--) { |
260 | i = (int)str_gnum(cur->c_short); |
261 | value = (double)i; |
262 | if (value != cur->c_short->str_u.str_nval) |
263 | return; /* fractional values--just forget it */ |
264 | changed = i; |
265 | if (changed != i) |
266 | return; /* too big for a short */ |
267 | if (cur->c_slen == O_LE) |
268 | i++; |
269 | else if (cur->c_slen == O_GE) /* we only do < or > here */ |
270 | i--; |
271 | if (i < min) |
272 | min = i; |
273 | if (i > max) |
274 | max = i; |
275 | cur = cur->c_next; |
276 | } |
277 | count = origcount; |
278 | if (max - min > count * 2 + 10) /* too sparse? */ |
279 | return; |
280 | |
281 | /* now make a new head in the exact same spot */ |
282 | New(104,cur, 1, CMD); |
283 | #ifdef STRUCTCOPY |
284 | *cur = *head; |
285 | #else |
286 | Copy(head,cur,1,CMD); |
287 | #endif |
288 | Zero(head,1,CMD); |
289 | head->c_type = C_NSWITCH; |
290 | head->c_next = cur; /* insert new cmd at front of list */ |
291 | head->c_stab = cur->c_stab; |
292 | |
293 | Newz(105,loc, max - min + 3, CMD*); |
294 | loc++; |
03a14243 |
295 | max -= min; |
296 | max++; |
a687059c |
297 | while (count--) { |
298 | i = (int)str_gnum(cur->c_short); |
299 | i -= min; |
a687059c |
300 | switch(cur->c_slen) { |
301 | case O_LE: |
302 | i++; |
303 | case O_LT: |
304 | for (i--; i >= -1; i--) |
305 | if (!loc[i]) |
306 | loc[i] = cur; |
307 | break; |
308 | case O_GE: |
309 | i--; |
310 | case O_GT: |
311 | for (i++; i <= max; i++) |
312 | if (!loc[i]) |
313 | loc[i] = cur; |
314 | break; |
315 | case O_EQ: |
316 | if (!loc[i]) |
317 | loc[i] = cur; |
318 | break; |
319 | } |
320 | cur = cur->c_next; |
321 | } |
322 | loc--; |
323 | min--; |
03a14243 |
324 | max++; |
a687059c |
325 | for (i = 0; i <= max; i++) |
326 | if (!loc[i]) |
327 | loc[i] = cur; |
328 | head->ucmd.scmd.sc_offset = min; |
329 | head->ucmd.scmd.sc_max = max; |
330 | head->ucmd.scmd.sc_next = loc; |
331 | } |
332 | |
333 | CMD * |
334 | append_line(head,tail) |
335 | register CMD *head; |
336 | register CMD *tail; |
337 | { |
338 | if (tail == Nullcmd) |
339 | return head; |
340 | if (!tail->c_head) /* make sure tail is well formed */ |
341 | tail->c_head = tail; |
342 | if (head != Nullcmd) { |
343 | tail = tail->c_head; /* get to start of tail list */ |
344 | if (!head->c_head) |
345 | head->c_head = head; /* start a new head list */ |
346 | while (head->c_next) { |
347 | head->c_next->c_head = head->c_head; |
348 | head = head->c_next; /* get to end of head list */ |
349 | } |
350 | head->c_next = tail; /* link to end of old list */ |
351 | tail->c_head = head->c_head; /* propagate head pointer */ |
352 | } |
353 | while (tail->c_next) { |
354 | tail->c_next->c_head = tail->c_head; |
355 | tail = tail->c_next; |
356 | } |
357 | return tail; |
358 | } |
359 | |
360 | CMD * |
361 | dodb(cur) |
362 | CMD *cur; |
363 | { |
364 | register CMD *cmd; |
365 | register CMD *head = cur->c_head; |
366 | register ARG *arg; |
367 | STR *str; |
368 | |
369 | if (!head) |
370 | head = cur; |
371 | if (!head->c_line) |
372 | return cur; |
373 | str = afetch(lineary,(int)head->c_line,FALSE); |
374 | if (!str || str->str_nok) |
375 | return cur; |
376 | str->str_u.str_nval = (double)head->c_line; |
377 | str->str_nok = 1; |
378 | Newz(106,cmd,1,CMD); |
379 | cmd->c_type = C_EXPR; |
380 | cmd->ucmd.acmd.ac_stab = Nullstab; |
381 | cmd->ucmd.acmd.ac_expr = Nullarg; |
382 | arg = make_op(O_ITEM,1,Nullarg,Nullarg,Nullarg); |
383 | arg[1].arg_type = A_SINGLE; |
384 | arg[1].arg_ptr.arg_str = str_nmake((double)head->c_line); |
385 | cmd->c_expr = make_op(O_SUBR, 2, |
386 | stab2arg(A_WORD,DBstab), |
387 | make_list(arg), |
388 | Nullarg); |
03a14243 |
389 | cmd->c_flags |= CF_COND|CF_DBSUB; |
a687059c |
390 | cmd->c_line = head->c_line; |
391 | cmd->c_label = head->c_label; |
392 | cmd->c_file = filename; |
393 | return append_line(cmd, cur); |
394 | } |
395 | |
396 | CMD * |
397 | make_acmd(type,stab,cond,arg) |
398 | int type; |
399 | STAB *stab; |
400 | ARG *cond; |
401 | ARG *arg; |
402 | { |
403 | register CMD *cmd; |
404 | |
405 | Newz(107,cmd,1,CMD); |
406 | cmd->c_type = type; |
407 | cmd->ucmd.acmd.ac_stab = stab; |
408 | cmd->ucmd.acmd.ac_expr = arg; |
409 | cmd->c_expr = cond; |
410 | if (cond) |
411 | cmd->c_flags |= CF_COND; |
412 | if (cmdline != NOLINE) { |
413 | cmd->c_line = cmdline; |
414 | cmdline = NOLINE; |
415 | } |
416 | cmd->c_file = filename; |
417 | if (perldb) |
418 | cmd = dodb(cmd); |
419 | return cmd; |
420 | } |
421 | |
422 | CMD * |
423 | make_ccmd(type,arg,cblock) |
424 | int type; |
425 | ARG *arg; |
426 | struct compcmd cblock; |
427 | { |
428 | register CMD *cmd; |
429 | |
430 | Newz(108,cmd, 1, CMD); |
431 | cmd->c_type = type; |
432 | cmd->c_expr = arg; |
433 | cmd->ucmd.ccmd.cc_true = cblock.comp_true; |
434 | cmd->ucmd.ccmd.cc_alt = cblock.comp_alt; |
435 | if (arg) |
436 | cmd->c_flags |= CF_COND; |
437 | if (cmdline != NOLINE) { |
438 | cmd->c_line = cmdline; |
439 | cmdline = NOLINE; |
440 | } |
441 | if (perldb) |
442 | cmd = dodb(cmd); |
443 | return cmd; |
444 | } |
445 | |
446 | CMD * |
447 | make_icmd(type,arg,cblock) |
448 | int type; |
449 | ARG *arg; |
450 | struct compcmd cblock; |
451 | { |
452 | register CMD *cmd; |
453 | register CMD *alt; |
454 | register CMD *cur; |
455 | register CMD *head; |
456 | struct compcmd ncblock; |
457 | |
458 | Newz(109,cmd, 1, CMD); |
459 | head = cmd; |
460 | cmd->c_type = type; |
461 | cmd->c_expr = arg; |
462 | cmd->ucmd.ccmd.cc_true = cblock.comp_true; |
463 | cmd->ucmd.ccmd.cc_alt = cblock.comp_alt; |
464 | if (arg) |
465 | cmd->c_flags |= CF_COND; |
466 | if (cmdline != NOLINE) { |
467 | cmd->c_line = cmdline; |
468 | cmdline = NOLINE; |
469 | } |
470 | cur = cmd; |
471 | alt = cblock.comp_alt; |
472 | while (alt && alt->c_type == C_ELSIF) { |
473 | cur = alt; |
474 | alt = alt->ucmd.ccmd.cc_alt; |
475 | } |
476 | if (alt) { /* a real life ELSE at the end? */ |
477 | ncblock.comp_true = alt; |
478 | ncblock.comp_alt = Nullcmd; |
479 | alt = append_line(cur,make_ccmd(C_ELSE,Nullarg,ncblock)); |
480 | cur->ucmd.ccmd.cc_alt = alt; |
481 | } |
482 | else |
483 | alt = cur; /* no ELSE, so cur is proxy ELSE */ |
484 | |
485 | cur = cmd; |
486 | while (cmd) { /* now point everyone at the ELSE */ |
487 | cur = cmd; |
488 | cmd = cur->ucmd.ccmd.cc_alt; |
489 | cur->c_head = head; |
490 | if (cur->c_type == C_ELSIF) |
491 | cur->c_type = C_IF; |
492 | if (cur->c_type == C_IF) |
493 | cur->ucmd.ccmd.cc_alt = alt; |
494 | if (cur == alt) |
495 | break; |
496 | cur->c_next = cmd; |
497 | } |
498 | if (perldb) |
499 | cur = dodb(cur); |
500 | return cur; |
501 | } |
502 | |
503 | void |
504 | opt_arg(cmd,fliporflop,acmd) |
505 | register CMD *cmd; |
506 | int fliporflop; |
507 | int acmd; |
508 | { |
509 | register ARG *arg; |
510 | int opt = CFT_EVAL; |
511 | int sure = 0; |
512 | ARG *arg2; |
513 | int context = 0; /* 0 = normal, 1 = before &&, 2 = before || */ |
514 | int flp = fliporflop; |
515 | |
516 | if (!cmd) |
517 | return; |
518 | if (!(arg = cmd->c_expr)) { |
519 | cmd->c_flags &= ~CF_COND; |
520 | return; |
521 | } |
522 | |
523 | /* Can we turn && and || into if and unless? */ |
524 | |
525 | if (acmd && !cmd->ucmd.acmd.ac_expr && !(cmd->c_flags & CF_TERM) && |
526 | (arg->arg_type == O_AND || arg->arg_type == O_OR) ) { |
527 | dehoist(arg,1); |
528 | arg[2].arg_type &= A_MASK; /* don't suppress eval */ |
529 | dehoist(arg,2); |
530 | cmd->ucmd.acmd.ac_expr = arg[2].arg_ptr.arg_arg; |
531 | cmd->c_expr = arg[1].arg_ptr.arg_arg; |
532 | if (arg->arg_type == O_OR) |
533 | cmd->c_flags ^= CF_INVERT; /* || is like unless */ |
534 | arg->arg_len = 0; |
535 | free_arg(arg); |
536 | arg = cmd->c_expr; |
537 | } |
538 | |
539 | /* Turn "if (!expr)" into "unless (expr)" */ |
540 | |
541 | if (!(cmd->c_flags & CF_TERM)) { /* unless return value wanted */ |
542 | while (arg->arg_type == O_NOT) { |
543 | dehoist(arg,1); |
544 | cmd->c_flags ^= CF_INVERT; /* flip sense of cmd */ |
545 | cmd->c_expr = arg[1].arg_ptr.arg_arg; /* hoist the rest of expr */ |
546 | free_arg(arg); |
547 | arg = cmd->c_expr; /* here we go again */ |
548 | } |
549 | } |
550 | |
551 | if (!arg->arg_len) { /* sanity check */ |
552 | cmd->c_flags |= opt; |
553 | return; |
554 | } |
555 | |
556 | /* for "cond .. cond" we set up for the initial check */ |
557 | |
558 | if (arg->arg_type == O_FLIP) |
559 | context |= 4; |
560 | |
561 | /* for "cond && expr" and "cond || expr" we can ignore expr, sort of */ |
562 | |
563 | morecontext: |
564 | if (arg->arg_type == O_AND) |
565 | context |= 1; |
566 | else if (arg->arg_type == O_OR) |
567 | context |= 2; |
568 | if (context && (arg[flp].arg_type & A_MASK) == A_EXPR) { |
569 | arg = arg[flp].arg_ptr.arg_arg; |
570 | flp = 1; |
571 | if (arg->arg_type == O_AND || arg->arg_type == O_OR) |
572 | goto morecontext; |
573 | } |
574 | if ((context & 3) == 3) |
575 | return; |
576 | |
577 | if (arg[flp].arg_flags & (AF_PRE|AF_POST)) { |
578 | cmd->c_flags |= opt; |
579 | return; /* side effect, can't optimize */ |
580 | } |
581 | |
582 | if (arg->arg_type == O_ITEM || arg->arg_type == O_FLIP || |
583 | arg->arg_type == O_AND || arg->arg_type == O_OR) { |
584 | if ((arg[flp].arg_type & A_MASK) == A_SINGLE) { |
585 | opt = (str_true(arg[flp].arg_ptr.arg_str) ? CFT_TRUE : CFT_FALSE); |
586 | cmd->c_short = str_smake(arg[flp].arg_ptr.arg_str); |
587 | goto literal; |
588 | } |
589 | else if ((arg[flp].arg_type & A_MASK) == A_STAB || |
590 | (arg[flp].arg_type & A_MASK) == A_LVAL) { |
591 | cmd->c_stab = arg[flp].arg_ptr.arg_stab; |
592 | opt = CFT_REG; |
593 | literal: |
594 | if (!context) { /* no && or ||? */ |
595 | free_arg(arg); |
596 | cmd->c_expr = Nullarg; |
597 | } |
598 | if (!(context & 1)) |
599 | cmd->c_flags |= CF_EQSURE; |
600 | if (!(context & 2)) |
601 | cmd->c_flags |= CF_NESURE; |
602 | } |
603 | } |
604 | else if (arg->arg_type == O_MATCH || arg->arg_type == O_SUBST || |
605 | arg->arg_type == O_NMATCH || arg->arg_type == O_NSUBST) { |
606 | if ((arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) && |
607 | (arg[2].arg_type & A_MASK) == A_SPAT && |
608 | arg[2].arg_ptr.arg_spat->spat_short ) { |
609 | cmd->c_stab = arg[1].arg_ptr.arg_stab; |
610 | cmd->c_short = str_smake(arg[2].arg_ptr.arg_spat->spat_short); |
611 | cmd->c_slen = arg[2].arg_ptr.arg_spat->spat_slen; |
612 | if (arg[2].arg_ptr.arg_spat->spat_flags & SPAT_ALL && |
613 | !(arg[2].arg_ptr.arg_spat->spat_flags & SPAT_ONCE) && |
614 | (arg->arg_type == O_MATCH || arg->arg_type == O_NMATCH) ) |
615 | sure |= CF_EQSURE; /* (SUBST must be forced even */ |
616 | /* if we know it will work.) */ |
617 | if (arg->arg_type != O_SUBST) { |
618 | arg[2].arg_ptr.arg_spat->spat_short = Nullstr; |
619 | arg[2].arg_ptr.arg_spat->spat_slen = 0; /* only one chk */ |
620 | } |
621 | sure |= CF_NESURE; /* normally only sure if it fails */ |
622 | if (arg->arg_type == O_NMATCH || arg->arg_type == O_NSUBST) |
623 | cmd->c_flags |= CF_FIRSTNEG; |
624 | if (context & 1) { /* only sure if thing is false */ |
625 | if (cmd->c_flags & CF_FIRSTNEG) |
626 | sure &= ~CF_NESURE; |
627 | else |
628 | sure &= ~CF_EQSURE; |
629 | } |
630 | else if (context & 2) { /* only sure if thing is true */ |
631 | if (cmd->c_flags & CF_FIRSTNEG) |
632 | sure &= ~CF_EQSURE; |
633 | else |
634 | sure &= ~CF_NESURE; |
635 | } |
636 | if (sure & (CF_EQSURE|CF_NESURE)) { /* if we know anything*/ |
637 | if (arg[2].arg_ptr.arg_spat->spat_flags & SPAT_SCANFIRST) |
638 | opt = CFT_SCAN; |
639 | else |
640 | opt = CFT_ANCHOR; |
641 | if (sure == (CF_EQSURE|CF_NESURE) /* really sure? */ |
642 | && arg->arg_type == O_MATCH |
643 | && context & 4 |
644 | && fliporflop == 1) { |
645 | spat_free(arg[2].arg_ptr.arg_spat); |
646 | arg[2].arg_ptr.arg_spat = Nullspat; /* don't do twice */ |
647 | } |
648 | cmd->c_flags |= sure; |
649 | } |
650 | } |
651 | } |
652 | else if (arg->arg_type == O_SEQ || arg->arg_type == O_SNE || |
653 | arg->arg_type == O_SLT || arg->arg_type == O_SGT) { |
654 | if (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) { |
655 | if (arg[2].arg_type == A_SINGLE) { |
656 | cmd->c_stab = arg[1].arg_ptr.arg_stab; |
657 | cmd->c_short = str_smake(arg[2].arg_ptr.arg_str); |
658 | cmd->c_slen = cmd->c_short->str_cur+1; |
659 | switch (arg->arg_type) { |
660 | case O_SLT: case O_SGT: |
661 | sure |= CF_EQSURE; |
662 | cmd->c_flags |= CF_FIRSTNEG; |
663 | break; |
664 | case O_SNE: |
665 | cmd->c_flags |= CF_FIRSTNEG; |
666 | /* FALL THROUGH */ |
667 | case O_SEQ: |
668 | sure |= CF_NESURE|CF_EQSURE; |
669 | break; |
670 | } |
671 | if (context & 1) { /* only sure if thing is false */ |
672 | if (cmd->c_flags & CF_FIRSTNEG) |
673 | sure &= ~CF_NESURE; |
674 | else |
675 | sure &= ~CF_EQSURE; |
676 | } |
677 | else if (context & 2) { /* only sure if thing is true */ |
678 | if (cmd->c_flags & CF_FIRSTNEG) |
679 | sure &= ~CF_EQSURE; |
680 | else |
681 | sure &= ~CF_NESURE; |
682 | } |
683 | if (sure & (CF_EQSURE|CF_NESURE)) { |
684 | opt = CFT_STROP; |
685 | cmd->c_flags |= sure; |
686 | } |
687 | } |
688 | } |
689 | } |
690 | else if (arg->arg_type == O_EQ || arg->arg_type == O_NE || |
691 | arg->arg_type == O_LE || arg->arg_type == O_GE || |
692 | arg->arg_type == O_LT || arg->arg_type == O_GT) { |
693 | if (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) { |
694 | if (arg[2].arg_type == A_SINGLE) { |
695 | cmd->c_stab = arg[1].arg_ptr.arg_stab; |
696 | if (dowarn) { |
697 | STR *str = arg[2].arg_ptr.arg_str; |
698 | |
699 | if ((!str->str_nok && !looks_like_number(str))) |
700 | warn("Possible use of == on string value"); |
701 | } |
702 | cmd->c_short = str_nmake(str_gnum(arg[2].arg_ptr.arg_str)); |
703 | cmd->c_slen = arg->arg_type; |
704 | sure |= CF_NESURE|CF_EQSURE; |
705 | if (context & 1) { /* only sure if thing is false */ |
706 | sure &= ~CF_EQSURE; |
707 | } |
708 | else if (context & 2) { /* only sure if thing is true */ |
709 | sure &= ~CF_NESURE; |
710 | } |
711 | if (sure & (CF_EQSURE|CF_NESURE)) { |
712 | opt = CFT_NUMOP; |
713 | cmd->c_flags |= sure; |
714 | } |
715 | } |
716 | } |
717 | } |
718 | else if (arg->arg_type == O_ASSIGN && |
719 | (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) && |
720 | arg[1].arg_ptr.arg_stab == defstab && |
721 | arg[2].arg_type == A_EXPR ) { |
722 | arg2 = arg[2].arg_ptr.arg_arg; |
723 | if (arg2->arg_type == O_ITEM && arg2[1].arg_type == A_READ) { |
724 | opt = CFT_GETS; |
725 | cmd->c_stab = arg2[1].arg_ptr.arg_stab; |
726 | if (!(stab_io(arg2[1].arg_ptr.arg_stab)->flags & IOF_ARGV)) { |
727 | free_arg(arg2); |
728 | free_arg(arg); |
729 | cmd->c_expr = Nullarg; |
730 | } |
731 | } |
732 | } |
733 | else if (arg->arg_type == O_CHOP && |
734 | (arg[1].arg_type == A_STAB || arg[1].arg_type == A_LVAL) ) { |
735 | opt = CFT_CHOP; |
736 | cmd->c_stab = arg[1].arg_ptr.arg_stab; |
737 | free_arg(arg); |
738 | cmd->c_expr = Nullarg; |
739 | } |
740 | if (context & 4) |
741 | opt |= CF_FLIP; |
742 | cmd->c_flags |= opt; |
743 | |
744 | if (cmd->c_flags & CF_FLIP) { |
745 | if (fliporflop == 1) { |
746 | arg = cmd->c_expr; /* get back to O_FLIP arg */ |
747 | New(110,arg[3].arg_ptr.arg_cmd, 1, CMD); |
748 | Copy(cmd, arg[3].arg_ptr.arg_cmd, 1, CMD); |
749 | New(111,arg[4].arg_ptr.arg_cmd,1,CMD); |
750 | Copy(cmd, arg[4].arg_ptr.arg_cmd, 1, CMD); |
751 | opt_arg(arg[4].arg_ptr.arg_cmd,2,acmd); |
752 | arg->arg_len = 2; /* this is a lie */ |
753 | } |
754 | else { |
755 | if ((opt & CF_OPTIMIZE) == CFT_EVAL) |
756 | cmd->c_flags = (cmd->c_flags & ~CF_OPTIMIZE) | CFT_UNFLIP; |
757 | } |
758 | } |
759 | } |
760 | |
761 | CMD * |
762 | add_label(lbl,cmd) |
763 | char *lbl; |
764 | register CMD *cmd; |
765 | { |
766 | if (cmd) |
767 | cmd->c_label = lbl; |
768 | return cmd; |
769 | } |
770 | |
771 | CMD * |
772 | addcond(cmd, arg) |
773 | register CMD *cmd; |
774 | register ARG *arg; |
775 | { |
776 | cmd->c_expr = arg; |
777 | cmd->c_flags |= CF_COND; |
778 | return cmd; |
779 | } |
780 | |
781 | CMD * |
782 | addloop(cmd, arg) |
783 | register CMD *cmd; |
784 | register ARG *arg; |
785 | { |
786 | void while_io(); |
787 | |
788 | cmd->c_expr = arg; |
789 | cmd->c_flags |= CF_COND|CF_LOOP; |
790 | |
791 | if (!(cmd->c_flags & CF_INVERT)) |
792 | while_io(cmd); /* add $_ =, if necessary */ |
793 | |
794 | if (cmd->c_type == C_BLOCK) |
795 | cmd->c_flags &= ~CF_COND; |
796 | else { |
797 | arg = cmd->ucmd.acmd.ac_expr; |
798 | if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_CMD) |
799 | cmd->c_flags &= ~CF_COND; /* "do {} while" happens at least once */ |
800 | if (arg && arg->arg_type == O_SUBR) |
801 | cmd->c_flags &= ~CF_COND; /* likewise for "do subr() while" */ |
802 | } |
803 | return cmd; |
804 | } |
805 | |
806 | CMD * |
807 | invert(cmd) |
03a14243 |
808 | CMD *cmd; |
a687059c |
809 | { |
03a14243 |
810 | register CMD *targ = cmd; |
811 | if (targ->c_head) |
812 | targ = targ->c_head; |
813 | if (targ->c_flags & CF_DBSUB) |
814 | targ = targ->c_next; |
815 | targ->c_flags ^= CF_INVERT; |
a687059c |
816 | return cmd; |
817 | } |
818 | |
819 | yyerror(s) |
820 | char *s; |
821 | { |
822 | char tmpbuf[258]; |
823 | char tmp2buf[258]; |
824 | char *tname = tmpbuf; |
825 | |
826 | if (bufptr > oldoldbufptr && bufptr - oldoldbufptr < 200 && |
827 | oldoldbufptr != oldbufptr && oldbufptr != bufptr) { |
828 | while (isspace(*oldoldbufptr)) |
829 | oldoldbufptr++; |
830 | strncpy(tmp2buf, oldoldbufptr, bufptr - oldoldbufptr); |
831 | tmp2buf[bufptr - oldoldbufptr] = '\0'; |
832 | sprintf(tname,"next 2 tokens \"%s\"",tmp2buf); |
833 | } |
834 | else if (bufptr > oldbufptr && bufptr - oldbufptr < 200 && |
835 | oldbufptr != bufptr) { |
836 | while (isspace(*oldbufptr)) |
837 | oldbufptr++; |
838 | strncpy(tmp2buf, oldbufptr, bufptr - oldbufptr); |
839 | tmp2buf[bufptr - oldbufptr] = '\0'; |
840 | sprintf(tname,"next token \"%s\"",tmp2buf); |
841 | } |
842 | else if (yychar > 256) |
843 | tname = "next token ???"; |
844 | else if (!yychar) |
845 | (void)strcpy(tname,"at EOF"); |
846 | else if (yychar < 32) |
847 | (void)sprintf(tname,"next char ^%c",yychar+64); |
848 | else if (yychar == 127) |
849 | (void)strcpy(tname,"at EOF"); |
850 | else |
851 | (void)sprintf(tname,"next char %c",yychar); |
852 | (void)sprintf(buf, "%s in file %s at line %d, %s\n", |
853 | s,filename,line,tname); |
854 | if (line == multi_end && multi_start < multi_end) |
855 | sprintf(buf+strlen(buf), |
856 | " (Might be a runaway multi-line %c%c string starting on line %d)\n", |
857 | multi_open,multi_close,multi_start); |
858 | if (in_eval) |
859 | str_cat(stab_val(stabent("@",TRUE)),buf); |
860 | else |
861 | fputs(buf,stderr); |
862 | if (++error_count >= 10) |
863 | fatal("Too many errors\n"); |
864 | } |
865 | |
866 | void |
867 | while_io(cmd) |
868 | register CMD *cmd; |
869 | { |
870 | register ARG *arg = cmd->c_expr; |
871 | STAB *asgnstab; |
872 | |
873 | /* hoist "while (<channel>)" up into command block */ |
874 | |
875 | if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_READ) { |
876 | cmd->c_flags &= ~CF_OPTIMIZE; /* clear optimization type */ |
877 | cmd->c_flags |= CFT_GETS; /* and set it to do the input */ |
878 | cmd->c_stab = arg[1].arg_ptr.arg_stab; |
879 | if (stab_io(arg[1].arg_ptr.arg_stab)->flags & IOF_ARGV) { |
880 | cmd->c_expr = l(make_op(O_ASSIGN, 2, /* fake up "$_ =" */ |
881 | stab2arg(A_LVAL,defstab), arg, Nullarg)); |
882 | } |
883 | else { |
884 | free_arg(arg); |
885 | cmd->c_expr = Nullarg; |
886 | } |
887 | } |
888 | else if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_INDREAD) { |
889 | cmd->c_flags &= ~CF_OPTIMIZE; /* clear optimization type */ |
890 | cmd->c_flags |= CFT_INDGETS; /* and set it to do the input */ |
891 | cmd->c_stab = arg[1].arg_ptr.arg_stab; |
892 | free_arg(arg); |
893 | cmd->c_expr = Nullarg; |
894 | } |
895 | else if (arg && arg->arg_type == O_ITEM && arg[1].arg_type == A_GLOB) { |
896 | if ((cmd->c_flags & CF_OPTIMIZE) == CFT_ARRAY) |
897 | asgnstab = cmd->c_stab; |
898 | else |
899 | asgnstab = defstab; |
900 | cmd->c_expr = l(make_op(O_ASSIGN, 2, /* fake up "$foo =" */ |
901 | stab2arg(A_LVAL,asgnstab), arg, Nullarg)); |
902 | cmd->c_flags &= ~CF_OPTIMIZE; /* clear optimization type */ |
903 | } |
904 | } |
905 | |
906 | CMD * |
907 | wopt(cmd) |
908 | register CMD *cmd; |
909 | { |
910 | register CMD *tail; |
911 | CMD *newtail; |
912 | register int i; |
913 | |
914 | if (cmd->c_expr && (cmd->c_flags & CF_OPTIMIZE) == CFT_FALSE) |
915 | opt_arg(cmd,1, cmd->c_type == C_EXPR); |
916 | |
917 | while_io(cmd); /* add $_ =, if necessary */ |
918 | |
919 | /* First find the end of the true list */ |
920 | |
921 | tail = cmd->ucmd.ccmd.cc_true; |
922 | if (tail == Nullcmd) |
923 | return cmd; |
924 | New(112,newtail, 1, CMD); /* guaranteed continue */ |
925 | for (;;) { |
926 | /* optimize "next" to point directly to continue block */ |
927 | if (tail->c_type == C_EXPR && |
928 | tail->ucmd.acmd.ac_expr && |
929 | tail->ucmd.acmd.ac_expr->arg_type == O_NEXT && |
930 | (tail->ucmd.acmd.ac_expr->arg_len == 0 || |
931 | (cmd->c_label && |
932 | strEQ(cmd->c_label, |
933 | tail->ucmd.acmd.ac_expr[1].arg_ptr.arg_str->str_ptr) ))) |
934 | { |
935 | arg_free(tail->ucmd.acmd.ac_expr); |
936 | tail->c_type = C_NEXT; |
937 | if (cmd->ucmd.ccmd.cc_alt != Nullcmd) |
938 | tail->ucmd.ccmd.cc_alt = cmd->ucmd.ccmd.cc_alt; |
939 | else |
940 | tail->ucmd.ccmd.cc_alt = newtail; |
941 | tail->ucmd.ccmd.cc_true = Nullcmd; |
942 | } |
943 | else if (tail->c_type == C_IF && !tail->ucmd.ccmd.cc_alt) { |
944 | if (cmd->ucmd.ccmd.cc_alt != Nullcmd) |
945 | tail->ucmd.ccmd.cc_alt = cmd->ucmd.ccmd.cc_alt; |
946 | else |
947 | tail->ucmd.ccmd.cc_alt = newtail; |
948 | } |
949 | else if (tail->c_type == C_CSWITCH || tail->c_type == C_NSWITCH) { |
950 | if (cmd->ucmd.ccmd.cc_alt != Nullcmd) { |
951 | for (i = tail->ucmd.scmd.sc_max; i >= 0; i--) |
952 | if (!tail->ucmd.scmd.sc_next[i]) |
953 | tail->ucmd.scmd.sc_next[i] = cmd->ucmd.ccmd.cc_alt; |
954 | } |
955 | else { |
956 | for (i = tail->ucmd.scmd.sc_max; i >= 0; i--) |
957 | if (!tail->ucmd.scmd.sc_next[i]) |
958 | tail->ucmd.scmd.sc_next[i] = newtail; |
959 | } |
960 | } |
961 | |
962 | if (!tail->c_next) |
963 | break; |
964 | tail = tail->c_next; |
965 | } |
966 | |
967 | /* if there's a continue block, link it to true block and find end */ |
968 | |
969 | if (cmd->ucmd.ccmd.cc_alt != Nullcmd) { |
970 | tail->c_next = cmd->ucmd.ccmd.cc_alt; |
971 | tail = tail->c_next; |
972 | for (;;) { |
973 | /* optimize "next" to point directly to continue block */ |
974 | if (tail->c_type == C_EXPR && |
975 | tail->ucmd.acmd.ac_expr && |
976 | tail->ucmd.acmd.ac_expr->arg_type == O_NEXT && |
977 | (tail->ucmd.acmd.ac_expr->arg_len == 0 || |
978 | (cmd->c_label && |
979 | strEQ(cmd->c_label, |
980 | tail->ucmd.acmd.ac_expr[1].arg_ptr.arg_str->str_ptr) ))) |
981 | { |
982 | arg_free(tail->ucmd.acmd.ac_expr); |
983 | tail->c_type = C_NEXT; |
984 | tail->ucmd.ccmd.cc_alt = newtail; |
985 | tail->ucmd.ccmd.cc_true = Nullcmd; |
986 | } |
987 | else if (tail->c_type == C_IF && !tail->ucmd.ccmd.cc_alt) { |
988 | tail->ucmd.ccmd.cc_alt = newtail; |
989 | } |
990 | else if (tail->c_type == C_CSWITCH || tail->c_type == C_NSWITCH) { |
991 | for (i = tail->ucmd.scmd.sc_max; i >= 0; i--) |
992 | if (!tail->ucmd.scmd.sc_next[i]) |
993 | tail->ucmd.scmd.sc_next[i] = newtail; |
994 | } |
995 | |
996 | if (!tail->c_next) |
997 | break; |
998 | tail = tail->c_next; |
999 | } |
1000 | for ( ; tail->c_next; tail = tail->c_next) ; |
1001 | } |
1002 | |
1003 | /* Here's the real trick: link the end of the list back to the beginning, |
1004 | * inserting a "last" block to break out of the loop. This saves one or |
1005 | * two procedure calls every time through the loop, because of how cmd_exec |
1006 | * does tail recursion. |
1007 | */ |
1008 | |
1009 | tail->c_next = newtail; |
1010 | tail = newtail; |
1011 | if (!cmd->ucmd.ccmd.cc_alt) |
1012 | cmd->ucmd.ccmd.cc_alt = tail; /* every loop has a continue now */ |
1013 | |
1014 | #ifndef lint |
1015 | (void)bcopy((char *)cmd, (char *)tail, sizeof(CMD)); |
1016 | #endif |
1017 | tail->c_type = C_EXPR; |
1018 | tail->c_flags ^= CF_INVERT; /* turn into "last unless" */ |
1019 | tail->c_next = tail->ucmd.ccmd.cc_true; /* loop directly back to top */ |
1020 | tail->ucmd.acmd.ac_expr = make_op(O_LAST,0,Nullarg,Nullarg,Nullarg); |
1021 | tail->ucmd.acmd.ac_stab = Nullstab; |
1022 | return cmd; |
1023 | } |
1024 | |
1025 | CMD * |
1026 | over(eachstab,cmd) |
1027 | STAB *eachstab; |
1028 | register CMD *cmd; |
1029 | { |
1030 | /* hoist "for $foo (@bar)" up into command block */ |
1031 | |
1032 | cmd->c_flags &= ~CF_OPTIMIZE; /* clear optimization type */ |
1033 | cmd->c_flags |= CFT_ARRAY; /* and set it to do the iteration */ |
1034 | cmd->c_stab = eachstab; |
0d3e774c |
1035 | cmd->c_short = str_new(0); /* just to save a field in struct cmd */ |
1036 | cmd->c_short->str_u.str_useful = -1; |
a687059c |
1037 | |
1038 | return cmd; |
1039 | } |
1040 | |
1041 | cmd_free(cmd) |
1042 | register CMD *cmd; |
1043 | { |
1044 | register CMD *tofree; |
1045 | register CMD *head = cmd; |
1046 | |
1047 | while (cmd) { |
1048 | if (cmd->c_type != C_WHILE) { /* WHILE block is duplicated */ |
1049 | if (cmd->c_label) |
1050 | Safefree(cmd->c_label); |
1051 | if (cmd->c_short) |
1052 | str_free(cmd->c_short); |
1053 | if (cmd->c_spat) |
1054 | spat_free(cmd->c_spat); |
1055 | if (cmd->c_expr) |
1056 | arg_free(cmd->c_expr); |
1057 | } |
1058 | switch (cmd->c_type) { |
1059 | case C_WHILE: |
1060 | case C_BLOCK: |
1061 | case C_ELSE: |
1062 | case C_IF: |
1063 | if (cmd->ucmd.ccmd.cc_true) |
1064 | cmd_free(cmd->ucmd.ccmd.cc_true); |
1065 | break; |
1066 | case C_EXPR: |
1067 | if (cmd->ucmd.acmd.ac_expr) |
1068 | arg_free(cmd->ucmd.acmd.ac_expr); |
1069 | break; |
1070 | } |
1071 | tofree = cmd; |
1072 | cmd = cmd->c_next; |
1073 | Safefree(tofree); |
1074 | if (cmd && cmd == head) /* reached end of while loop */ |
1075 | break; |
1076 | } |
1077 | } |
1078 | |
1079 | arg_free(arg) |
1080 | register ARG *arg; |
1081 | { |
1082 | register int i; |
1083 | |
1084 | for (i = 1; i <= arg->arg_len; i++) { |
1085 | switch (arg[i].arg_type & A_MASK) { |
1086 | case A_NULL: |
1087 | break; |
1088 | case A_LEXPR: |
1089 | if (arg->arg_type == O_AASSIGN && |
1090 | arg[i].arg_ptr.arg_arg->arg_type == O_LARRAY) { |
1091 | char *name = |
1092 | stab_name(arg[i].arg_ptr.arg_arg[1].arg_ptr.arg_stab); |
1093 | |
1094 | if (strnEQ("_GEN_",name, 5)) /* array for foreach */ |
1095 | hdelete(defstash,name,strlen(name)); |
1096 | } |
1097 | /* FALL THROUGH */ |
1098 | case A_EXPR: |
1099 | arg_free(arg[i].arg_ptr.arg_arg); |
1100 | break; |
1101 | case A_CMD: |
1102 | cmd_free(arg[i].arg_ptr.arg_cmd); |
1103 | break; |
1104 | case A_WORD: |
1105 | case A_STAB: |
1106 | case A_LVAL: |
1107 | case A_READ: |
1108 | case A_GLOB: |
1109 | case A_ARYLEN: |
1110 | case A_LARYLEN: |
1111 | case A_ARYSTAB: |
1112 | case A_LARYSTAB: |
1113 | break; |
1114 | case A_SINGLE: |
1115 | case A_DOUBLE: |
1116 | case A_BACKTICK: |
1117 | str_free(arg[i].arg_ptr.arg_str); |
1118 | break; |
1119 | case A_SPAT: |
1120 | spat_free(arg[i].arg_ptr.arg_spat); |
1121 | break; |
1122 | } |
1123 | } |
1124 | free_arg(arg); |
1125 | } |
1126 | |
1127 | spat_free(spat) |
1128 | register SPAT *spat; |
1129 | { |
1130 | register SPAT *sp; |
1131 | HENT *entry; |
1132 | |
1133 | if (spat->spat_runtime) |
1134 | arg_free(spat->spat_runtime); |
1135 | if (spat->spat_repl) { |
1136 | arg_free(spat->spat_repl); |
1137 | } |
1138 | if (spat->spat_short) { |
1139 | str_free(spat->spat_short); |
1140 | } |
1141 | if (spat->spat_regexp) { |
1142 | regfree(spat->spat_regexp); |
1143 | } |
1144 | |
1145 | /* now unlink from spat list */ |
1146 | |
1147 | for (entry = defstash->tbl_array['_']; entry; entry = entry->hent_next) { |
1148 | register HASH *stash; |
1149 | STAB *stab = (STAB*)entry->hent_val; |
1150 | |
1151 | if (!stab) |
1152 | continue; |
1153 | stash = stab_hash(stab); |
1154 | if (!stash || stash->tbl_spatroot == Null(SPAT*)) |
1155 | continue; |
1156 | if (stash->tbl_spatroot == spat) |
1157 | stash->tbl_spatroot = spat->spat_next; |
1158 | else { |
1159 | for (sp = stash->tbl_spatroot; |
1160 | sp && sp->spat_next != spat; |
1161 | sp = sp->spat_next) |
1162 | ; |
1163 | if (sp) |
1164 | sp->spat_next = spat->spat_next; |
1165 | } |
1166 | } |
1167 | Safefree(spat); |
1168 | } |
1169 | |
1170 | /* Recursively descend a command sequence and push the address of any string |
1171 | * that needs saving on recursion onto the tosave array. |
1172 | */ |
1173 | |
1174 | static int |
1175 | cmd_tosave(cmd,willsave) |
1176 | register CMD *cmd; |
1177 | int willsave; /* willsave passes down the tree */ |
1178 | { |
1179 | register CMD *head = cmd; |
1180 | int shouldsave = FALSE; /* shouldsave passes up the tree */ |
1181 | int tmpsave; |
1182 | register CMD *lastcmd = Nullcmd; |
1183 | |
1184 | while (cmd) { |
1185 | if (cmd->c_spat) |
1186 | shouldsave |= spat_tosave(cmd->c_spat); |
1187 | if (cmd->c_expr) |
1188 | shouldsave |= arg_tosave(cmd->c_expr,willsave); |
1189 | switch (cmd->c_type) { |
1190 | case C_WHILE: |
1191 | if (cmd->ucmd.ccmd.cc_true) { |
1192 | tmpsave = cmd_tosave(cmd->ucmd.ccmd.cc_true,willsave); |
1193 | |
1194 | /* Here we check to see if the temporary array generated for |
1195 | * a foreach needs to be localized because of recursion. |
1196 | */ |
1197 | if (tmpsave && (cmd->c_flags & CF_OPTIMIZE) == CFT_ARRAY && |
1198 | lastcmd && |
1199 | lastcmd->c_type == C_EXPR && |
1200 | lastcmd->ucmd.acmd.ac_expr) { |
1201 | ARG *arg = lastcmd->ucmd.acmd.ac_expr; |
1202 | |
1203 | if (arg->arg_type == O_ASSIGN && |
1204 | arg[1].arg_type == A_LEXPR && |
1205 | arg[1].arg_ptr.arg_arg->arg_type == O_LARRAY && |
1206 | strnEQ("_GEN_", |
1207 | stab_name(arg[1].arg_ptr.arg_arg[1].arg_ptr.arg_stab), |
1208 | 5)) { /* array generated for foreach */ |
1209 | (void)localize(arg[1].arg_ptr.arg_arg); |
1210 | } |
1211 | } |
1212 | shouldsave |= tmpsave; |
1213 | } |
1214 | break; |
1215 | case C_BLOCK: |
1216 | case C_ELSE: |
1217 | case C_IF: |
1218 | if (cmd->ucmd.ccmd.cc_true) |
1219 | shouldsave |= cmd_tosave(cmd->ucmd.ccmd.cc_true,willsave); |
1220 | break; |
1221 | case C_EXPR: |
1222 | if (cmd->ucmd.acmd.ac_expr) |
1223 | shouldsave |= arg_tosave(cmd->ucmd.acmd.ac_expr,willsave); |
1224 | break; |
1225 | } |
1226 | lastcmd = cmd; |
1227 | cmd = cmd->c_next; |
1228 | if (cmd && cmd == head) /* reached end of while loop */ |
1229 | break; |
1230 | } |
1231 | return shouldsave; |
1232 | } |
1233 | |
1234 | static int |
1235 | arg_tosave(arg,willsave) |
1236 | register ARG *arg; |
1237 | int willsave; |
1238 | { |
1239 | register int i; |
1240 | int shouldsave = FALSE; |
1241 | |
1242 | for (i = arg->arg_len; i >= 1; i--) { |
1243 | switch (arg[i].arg_type & A_MASK) { |
1244 | case A_NULL: |
1245 | break; |
1246 | case A_LEXPR: |
1247 | case A_EXPR: |
1248 | shouldsave |= arg_tosave(arg[i].arg_ptr.arg_arg,shouldsave); |
1249 | break; |
1250 | case A_CMD: |
1251 | shouldsave |= cmd_tosave(arg[i].arg_ptr.arg_cmd,shouldsave); |
1252 | break; |
1253 | case A_WORD: |
1254 | case A_STAB: |
1255 | case A_LVAL: |
1256 | case A_READ: |
1257 | case A_GLOB: |
1258 | case A_ARYLEN: |
1259 | case A_SINGLE: |
1260 | case A_DOUBLE: |
1261 | case A_BACKTICK: |
1262 | break; |
1263 | case A_SPAT: |
1264 | shouldsave |= spat_tosave(arg[i].arg_ptr.arg_spat); |
1265 | break; |
1266 | } |
1267 | } |
1268 | switch (arg->arg_type) { |
1269 | case O_RETURN: |
1270 | saw_return = TRUE; |
1271 | break; |
1272 | case O_EVAL: |
1273 | case O_SUBR: |
1274 | shouldsave = TRUE; |
1275 | break; |
1276 | } |
1277 | if (willsave) |
1278 | (void)apush(tosave,arg->arg_ptr.arg_str); |
1279 | return shouldsave; |
1280 | } |
1281 | |
1282 | static int |
1283 | spat_tosave(spat) |
1284 | register SPAT *spat; |
1285 | { |
1286 | int shouldsave = FALSE; |
1287 | |
1288 | if (spat->spat_runtime) |
1289 | shouldsave |= arg_tosave(spat->spat_runtime,FALSE); |
1290 | if (spat->spat_repl) { |
1291 | shouldsave |= arg_tosave(spat->spat_repl,FALSE); |
1292 | } |
1293 | |
1294 | return shouldsave; |
1295 | } |
1296 | |