source: trunk/optimizer.c @ 333

Last change on this file since 333 was 333, checked in by moo, 8 years ago

optimizer: better debug print

File size: 12.6 KB
Line 
1#if 1
2#define DEBUG
3#endif
4
5#include "utils.h"
6#include "optimizer.h"
7/* the "vector" stack */
8#include "stack.h"
9
10#ifdef DEBUG
11#   include "processor.h"
12#   include "const_string.h"
13#   include "ext/standard/php_var.h"
14#endif
15
16#ifdef IS_CV
17#   define XCACHE_IS_CV IS_CV
18#else
19#   define XCACHE_IS_CV 16
20#endif
21
22typedef int bbid_t;
23enum {
24    BBID_INVALID = -1,
25};
26/* {{{ basic block */
27typedef struct _bb_t {
28    bbid_t     id;
29    zend_bool  used;
30
31    zend_bool  alloc;
32    zend_op   *opcodes;
33    int        count;
34    int        size;
35
36    bbid_t     fall;
37    bbid_t     catch;
38
39    int        opnum; /* opnum after joining basic block */
40} bb_t;
41/* }}} */
42
43/* basic blocks */
44typedef xc_stack_t bbs_t;
45
46/* op array helper functions */
47static int op_array_convert_switch(zend_op_array *op_array) /* {{{ */
48{
49    int i;
50
51    if (op_array->brk_cont_array == NULL) {
52        return SUCCESS;
53    }
54
55    for (i = 0; i < op_array->last; i ++) {
56        zend_op *opline = &op_array->opcodes[i];
57        zend_brk_cont_element *jmp_to;
58        int array_offset, nest_levels, original_nest_levels;
59
60        if (opline->opcode != ZEND_BRK && opline->opcode != ZEND_CONT) {
61            continue;
62        }
63        if (opline->op2.op_type != IS_CONST
64         || opline->op2.u.constant.type != IS_LONG) {
65            return FAILURE;
66        }
67
68        nest_levels = opline->op2.u.constant.value.lval;
69        original_nest_levels = nest_levels;
70
71        array_offset = opline->op1.u.opline_num;
72        do {
73            if (array_offset == -1) {
74                /* this is a runtime error in ZE
75                zend_error(E_ERROR, "Cannot break/continue %d level%s", original_nest_levels, (original_nest_levels == 1) ? "" : "s");
76                */
77                return FAILURE;
78            }
79            jmp_to = &op_array->brk_cont_array[array_offset];
80            if (nest_levels > 1) {
81                zend_op *brk_opline = &op_array->opcodes[jmp_to->brk];
82
83                switch (brk_opline->opcode) {
84                case ZEND_SWITCH_FREE:
85                    break;
86                case ZEND_FREE:
87                    break;
88                }
89            }
90            array_offset = jmp_to->parent;
91        } while (--nest_levels > 0);
92
93        /* rewrite to jmp */
94        if (opline->opcode == ZEND_BRK) {
95            opline->op1.u.opline_num = jmp_to->brk;
96        }
97        else {
98            opline->op1.u.opline_num = jmp_to->cont;
99        }
100        opline->op2.op_type = IS_UNUSED;
101        opline->opcode = ZEND_JMP;
102    }
103
104    if (op_array->brk_cont_array != NULL) {
105        efree(op_array->brk_cont_array);
106        op_array->brk_cont_array = NULL;
107    }
108    op_array->last_brk_cont = 0;
109    return SUCCESS;
110}
111/* }}} */
112/* {{{ op_flowinfo helper func */
113enum {
114    XC_OPNUM_INVALID = -1,
115};
116typedef struct {
117    int       jmpout_op1;
118    int       jmpout_op2;
119    int       jmpout_ext;
120    zend_bool fall;
121} op_flowinfo_t;
122static void op_flowinfo_init(op_flowinfo_t *fi)
123{
124    fi->jmpout_op1 = fi->jmpout_op2 = fi->jmpout_ext = XC_OPNUM_INVALID;
125    fi->fall = 0;
126}
127/* }}} */
128static int op_get_flowinfo(op_flowinfo_t *fi, zend_op *opline) /* {{{ */
129{
130    op_flowinfo_init(fi);
131
132    /* break=will fall */
133    switch (opline->opcode) {
134    case ZEND_HANDLE_EXCEPTION:
135    case ZEND_RETURN:
136    case ZEND_EXIT:
137        return SUCCESS; /* no fall */
138
139    case ZEND_JMP:
140        fi->jmpout_op1 = opline->op1.u.opline_num;
141        return SUCCESS; /* no fall */
142
143    case ZEND_JMPZNZ:
144        fi->jmpout_op2 = opline->op2.u.opline_num;
145        fi->jmpout_ext = (int) opline->extended_value;
146        return SUCCESS; /* no fall */
147
148    case ZEND_JMPZ:
149    case ZEND_JMPNZ:
150    case ZEND_JMPZ_EX:
151    case ZEND_JMPNZ_EX:
152#ifdef ZEND_JMP_NO_CTOR
153    case ZEND_JMP_NO_CTOR:
154#endif
155#ifdef ZEND_NEW
156    case ZEND_NEW:
157#endif
158#ifdef ZEND_FE_RESET
159    case ZEND_FE_RESET:
160#endif     
161    case ZEND_FE_FETCH:
162        fi->jmpout_op2 = opline->op2.u.opline_num;
163        break;
164
165#ifdef ZEND_CATCH
166    case ZEND_CATCH:
167        fi->jmpout_ext = (int) opline->extended_value;
168        break;
169#endif
170
171    default:
172        return FAILURE;
173    }
174
175    fi->fall = 1;
176    return SUCCESS;
177}
178/* }}} */
179#ifdef DEBUG
180static void op_snprint(char *buf, int size, znode *op) /* {{{ */
181{
182    switch (op->op_type) {
183    case IS_CONST:
184        {
185            zval result;
186            zval *zv = &op->u.constant;
187            TSRMLS_FETCH();
188
189            php_start_ob_buffer(NULL, 0, 1 TSRMLS_CC);
190            php_var_export(&zv, 1 TSRMLS_CC);
191
192            php_ob_get_buffer(&result TSRMLS_CC); 
193            php_end_ob_buffer(0, 0 TSRMLS_CC);
194            snprintf(buf, size, Z_STRVAL(result));
195            zval_dtor(&result);
196        }
197        break;
198
199    case IS_TMP_VAR:
200        snprintf(buf, size, "t@%d", op->u.var);
201        break;
202
203    case XCACHE_IS_CV:
204    case IS_VAR:
205        snprintf(buf, size, "v@%d", op->u.var);
206        break;
207
208    case IS_UNUSED:
209        if (op->u.opline_num) {
210            snprintf(buf, size, "u#%d", op->u.opline_num);
211        }
212        else {
213            snprintf(buf, size, "-");
214        }
215        break;
216
217    default:
218        snprintf(buf, size, "%d %d", op->op_type, op->u.var);
219    }
220}
221/* }}} */
222static void op_print(int line, zend_op *first, zend_op *end) /* {{{ */
223{
224    zend_op *opline;
225    for (opline = first; opline < end; opline ++) {
226        char buf_r[20];
227        char buf_1[20];
228        char buf_2[20];
229        op_snprint(buf_r, sizeof(buf_r), &opline->result);
230        op_snprint(buf_1, sizeof(buf_1), &opline->op1);
231        op_snprint(buf_2, sizeof(buf_2), &opline->op2);
232        fprintf(stderr,
233                "%3d %3d"
234                " %-25s%-5s%-20s%-20s%5lu\r\n"
235                , opline->lineno, opline - first + line
236                , xc_get_opcode(opline->opcode), buf_r, buf_1, buf_2, opline->extended_value);
237    }
238}
239/* }}} */
240#endif
241
242/*
243 * basic block functions
244 */
245
246static bb_t *bb_new_ex(zend_op *opcodes, int count) /* {{{ */
247{
248    bb_t *bb = (bb_t *) ecalloc(sizeof(bb_t), 1);
249
250    bb->fall       = BBID_INVALID;
251    bb->catch      = BBID_INVALID;
252
253    if (opcodes) {
254        bb->alloc   = 0;
255        bb->size    = bb->count = count;
256        bb->opcodes = opcodes;
257    }
258    else {
259        bb->alloc   = 1;
260        bb->size    = bb->count = 8;
261        bb->opcodes = ecalloc(sizeof(zend_op), bb->size);
262    }
263
264    return bb;
265}
266/* }}} */
267#define bb_new() bb_new_ex(NULL, 0)
268static void bb_destroy(bb_t *bb) /* {{{ */
269{
270    if (bb->alloc) {
271        efree(bb->opcodes);
272    }
273    efree(bb);
274}
275/* }}} */
276#ifdef DEBUG
277static void bb_print(bb_t *bb, zend_op *opcodes) /* {{{ */
278{
279    int line = bb->opcodes - opcodes;
280    op_flowinfo_t fi;
281    zend_op *last = bb->opcodes + bb->count - 1;
282
283    op_get_flowinfo(&fi, last);
284
285    fprintf(stderr,
286            "\r\n==== #%-3d cnt:%-3d lno:%-3d"
287            " %c%c"
288            " op1:%-3d op2:%-3d ext:%-3d fal:%-3d cat:%-3d %s ====\r\n"
289            , bb->id, bb->count, bb->alloc ? -1 : line
290            , bb->used ? 'U' : ' ', bb->alloc ? 'A' : ' '
291            , fi.jmpout_op1, fi.jmpout_op2, fi.jmpout_ext, bb->fall, bb->catch, xc_get_opcode(last->opcode)
292            );
293    op_print(line, bb->opcodes, last + 1);
294}
295/* }}} */
296#endif
297
298static bb_t *bbs_get(bbs_t *bbs, int n) /* {{{ */
299{
300    return (bb_t *) xc_stack_get(bbs, n);
301}
302/* }}} */
303static int bbs_count(bbs_t *bbs) /* {{{ */
304{
305    return xc_stack_count(bbs);
306}
307/* }}} */
308static void bbs_destroy(bbs_t *bbs) /* {{{ */
309{
310    bb_t *bb;
311    while (bbs_count(bbs)) {
312        bb = (bb_t *) xc_stack_pop(bbs);
313        bb_destroy(bb);
314    }
315    xc_stack_destroy(bbs);
316}
317/* }}} */
318#ifdef DEBUG
319static void bbs_print(bbs_t *bbs, zend_op *opcodes) /* {{{ */
320{
321    int i;
322    for (i = 0; i < xc_stack_count(bbs); i ++) {
323        bb_print(bbs_get(bbs, i), opcodes);
324    }
325}
326/* }}} */
327#endif
328#define bbs_init(bbs) xc_stack_init_ex(bbs, 16)
329static bb_t *bbs_add_bb(bbs_t *bbs, bb_t *bb) /* {{{ */
330{
331    bb->id = (bbid_t) xc_stack_count(bbs);
332    xc_stack_push(bbs, (void *) bb);
333    return bb;
334}
335/* }}} */
336static bb_t *bbs_new_bb_ex(bbs_t *bbs, zend_op *opcodes, int count) /* {{{ */
337{
338    return bbs_add_bb(bbs, bb_new_ex(opcodes, count));
339}
340/* }}} */
341static int bbs_build_from(bbs_t *bbs, zend_op_array *op_array, int count) /* {{{ */
342{
343    int i, start;
344    bb_t *pbb;
345    bbid_t id;
346    op_flowinfo_t fi;
347    zend_op *opline;
348    bbid_t *bbids          = do_alloca(count * sizeof(bbid_t));
349    bbid_t *catchbbids     = do_alloca(count * sizeof(bbid_t));
350    zend_bool *markbbhead  = do_alloca(count * sizeof(zend_bool));
351
352    /* {{{ mark jmpin/jumpout */
353    memset(markbbhead,  0, count * sizeof(zend_bool));
354
355    markbbhead[0] = 1;
356    for (i = 0; i < count; i ++) {
357        if (op_get_flowinfo(&fi, &op_array->opcodes[i]) == SUCCESS) {
358            if (fi.jmpout_op1 != XC_OPNUM_INVALID) {
359                markbbhead[fi.jmpout_op1] = 1;
360            }
361            if (fi.jmpout_op2 != XC_OPNUM_INVALID) {
362                markbbhead[fi.jmpout_op2] = 1;
363            }
364            if (fi.jmpout_ext != XC_OPNUM_INVALID) {
365                markbbhead[fi.jmpout_ext] = 1;
366            }
367            if (i + 1 < count) {
368                markbbhead[i + 1] = 1;
369            }
370        }
371    }
372    /* mark try start */
373    for (i = 0; i < op_array->last_try_catch; i ++) {
374        markbbhead[op_array->try_catch_array[i].try_op] = 1;
375    }
376    /* }}} */
377    /* {{{ fill op lines with newly allocated id */
378    for (i = 0; i < count; i ++) {
379        bbids[i] = BBID_INVALID;
380    }
381
382    id = -1;
383    for (i = 0; i < count; i ++) {
384        if (markbbhead[i]) {
385            id ++;
386        }
387        bbids[i] = id;
388        TRACE("bbids[%d] = %d", i, id);
389    }
390    /* }}} */
391    /* {{{ fill op lines with catch id */
392    for (i = 0; i < count; i ++) {
393        catchbbids[i] = BBID_INVALID;
394    }
395
396    for (i = 0; i < op_array->last_try_catch; i ++) {
397        int j;
398        zend_try_catch_element *e = &op_array->try_catch_array[i];
399        for (j = e->try_op; j < e->catch_op; j ++) {
400            catchbbids[j] = bbids[e->catch_op];
401        }
402    }
403#ifdef DEBUG
404    for (i = 0; i < count; i ++) {
405        TRACE("catchbbids[%d] = %d", i, catchbbids[i]);
406    }
407#endif
408    /* }}} */
409    /* {{{ create basic blocks */
410    start = 0;
411    id = 0;
412    /* loop over to deal with the last block */
413    for (i = 1; i <= count; i ++) {
414        if (i < count && id == bbids[i]) {
415            continue;
416        }
417
418        opline = op_array->opcodes + start;
419        pbb = bbs_new_bb_ex(bbs, opline, i - start);
420        pbb->catch = catchbbids[start];
421
422        /* last */
423        opline = pbb->opcodes + pbb->count - 1;
424        if (op_get_flowinfo(&fi, opline) == SUCCESS) {
425            if (fi.jmpout_op1 != XC_OPNUM_INVALID) {
426                opline->op1.u.opline_num = bbids[fi.jmpout_op1];
427                assert(opline->op1.u.opline_num != BBID_INVALID);
428            }
429            if (fi.jmpout_op2 != XC_OPNUM_INVALID) {
430                opline->op2.u.opline_num = bbids[fi.jmpout_op2];
431                assert(opline->op2.u.opline_num != BBID_INVALID);
432            }
433            if (fi.jmpout_ext != XC_OPNUM_INVALID) {
434                opline->extended_value = bbids[fi.jmpout_ext];
435                assert(opline->extended_value != BBID_INVALID);
436            }
437            if (fi.fall && i + 1 < count) {
438                pbb->fall = bbids[i + 1];
439                TRACE("fall %d", pbb->fall);
440                assert(pbb->fall != BBID_INVALID);
441            }
442        }
443        if (i >= count) {
444            break;
445        }
446        start = i;
447        id = bbids[i];
448    }
449    /* }}} */
450
451    free_alloca(catchbbids);
452    free_alloca(bbids);
453    free_alloca(markbbhead);
454    return SUCCESS;
455}
456/* }}} */
457static void bbs_restore_opnum(bbs_t *bbs, zend_op_array *op_array) /* {{{ */
458{
459    int i;
460    bbid_t lasttrybbid;
461
462    for (i = 0; i < bbs_count(bbs); i ++) {
463        op_flowinfo_t fi;
464        bb_t *bb = bbs_get(bbs, i);
465        zend_op *last = bb->opcodes + bb->count - 1;
466
467        if (op_get_flowinfo(&fi, last) == SUCCESS) {
468            if (fi.jmpout_op1 != XC_OPNUM_INVALID) {
469                last->op1.u.opline_num = bbs_get(bbs, fi.jmpout_op1)->opnum;
470                assert(last->op1.u.opline_num != BBID_INVALID);
471            }
472            if (fi.jmpout_op2 != XC_OPNUM_INVALID) {
473                last->op2.u.opline_num = bbs_get(bbs, fi.jmpout_op2)->opnum;
474                assert(last->op2.u.opline_num != BBID_INVALID);
475            }
476            if (fi.jmpout_ext != XC_OPNUM_INVALID) {
477                last->extended_value = bbs_get(bbs, fi.jmpout_ext)->opnum;
478                assert(last->extended_value != BBID_INVALID);
479            }
480        }
481    }
482
483    lasttrybbid = BBID_INVALID;
484    op_array->last_try_catch = 0;
485    for (i = 0; i < bbs_count(bbs); i ++) {
486        bb_t *bb = bbs_get(bbs, i);
487
488        if (lasttrybbid != bb->catch) {
489            if (lasttrybbid != BBID_INVALID) {
490                int try_catch_offset = op_array->last_try_catch ++;
491
492                op_array->try_catch_array = erealloc(op_array->try_catch_array, sizeof(zend_try_catch_element) * op_array->last_try_catch);
493                op_array->try_catch_array[try_catch_offset].try_op = bbs_get(bbs, lasttrybbid)->opnum;
494                op_array->try_catch_array[try_catch_offset].catch_op = bbs_get(bbs, bb->id)->opnum;
495            }
496            lasttrybbid = bb->catch;
497        }
498    }
499    /* it is impossible to have last bb catched */
500}
501/* }}} */
502
503/*
504 * optimize
505 */
506static int xc_optimize_op_array(zend_op_array *op_array TSRMLS_DC) /* {{{ */
507{
508    bbs_t bbs;
509
510    if (op_array->type != ZEND_USER_FUNCTION) {
511        return 0;
512    }
513    xc_undo_pass_two(op_array TSRMLS_CC);
514#ifdef DEBUG
515#   if 0
516    TRACE("optimize file: %s", op_array->filename);
517    xc_dprint_zend_op_array(op_array, 0 TSRMLS_CC);
518#   endif
519    op_print(0, op_array->opcodes, op_array->opcodes + op_array->last);
520#endif
521
522    if (op_array_convert_switch(op_array) == SUCCESS) {
523        bbs_init(&bbs);
524        if (bbs_build_from(&bbs, op_array, op_array->last) == SUCCESS) {
525            int i;
526#ifdef DEBUG
527            bbs_print(&bbs, op_array->opcodes);
528#endif
529            /* TODO: calc opnum after basic block move */
530            for (i = 0; i < bbs_count(&bbs); i ++) {
531                bb_t *bb = bbs_get(&bbs, i);
532                bb->opnum = bb->opcodes - op_array->opcodes;
533            }
534            bbs_restore_opnum(&bbs, op_array);
535        }
536        bbs_destroy(&bbs);
537    }
538
539#ifdef DEBUG
540#   if 0
541    TRACE("%s", "after compiles");
542    xc_dprint_zend_op_array(op_array, 0 TSRMLS_CC);
543#   endif
544#endif
545    xc_redo_pass_two(op_array TSRMLS_CC);
546    return 0;
547}
548/* }}} */
549void xc_optimize(zend_op_array *op_array TSRMLS_DC) /* {{{ */
550{
551    xc_compile_result_t cr;
552
553    if (!op_array) {
554        return;
555    }
556
557    xc_compile_result_init_cur(&cr, op_array TSRMLS_CC);
558
559    xc_apply_op_array(&cr, (apply_func_t) xc_undo_pass_two TSRMLS_CC);
560    xc_apply_op_array(&cr, (apply_func_t) xc_optimize_op_array TSRMLS_CC);
561    xc_apply_op_array(&cr, (apply_func_t) xc_redo_pass_two TSRMLS_CC);
562
563    xc_compile_result_free(&cr);
564}
565/* }}} */
Note: See TracBrowser for help on using the repository browser.