source: trunk/utils.c @ 213

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

fix undefined symnbol max

File size: 16.1 KB
RevLine 
[212]1
[1]2#include "xcache.h"
3#include "utils.h"
4#ifdef ZEND_ENGINE_2_1
5#include "zend_vm.h"
6#endif
7#include "opcode_spec.h"
8#undef NDEBUG
9#include "assert.h"
10
[213]11#ifndef max
12#define max(a, b) ((a) < (b) ? (b) : (a))
13#endif
14
[212]15#ifndef ZEND_VM_SET_OPCODE_HANDLER
16#   define ZEND_VM_SET_OPCODE_HANDLER(opline) do { } while (0)
17#endif
18
19#define OP_ZVAL_DTOR(op) do { \
20    (op).u.constant.is_ref = 0; \
21    zval_dtor(&(op).u.constant); \
22} while(0)
[1]23xc_compile_result_t *xc_compile_result_init(xc_compile_result_t *cr, /* {{{ */
24        zend_op_array *op_array,
25        HashTable *function_table,
26        HashTable *class_table)
27{
28    if (cr) {
29        cr->alloc = 0;
30    }
31    else {
32        cr = emalloc(sizeof(xc_compile_result_t));
33        cr->alloc = 1;
34    }
35    cr->op_array       = op_array;
36    cr->function_table = function_table;
37    cr->class_table    = class_table;
38    return cr;
39}
40/* }}} */
41xc_compile_result_t *xc_compile_result_init_cur(xc_compile_result_t *cr, zend_op_array *op_array TSRMLS_DC) /* {{{ */
42{
43    return xc_compile_result_init(cr, op_array, CG(function_table), CG(class_table));
44}
45/* }}} */
46void xc_compile_result_free(xc_compile_result_t *cr) /* {{{ */
47{
48    if (cr->alloc) {
49        efree(cr);
50    }
51}
52/* }}} */
53
54int xc_apply_function(zend_function *zf, apply_func_t applyer TSRMLS_DC) /* {{{ */
55{
56    switch (zf->type) {
57    case ZEND_USER_FUNCTION:
58    case ZEND_EVAL_CODE:
59        return applyer(&zf->op_array TSRMLS_CC);
60        break;
61
62    case ZEND_INTERNAL_FUNCTION:
63    case ZEND_OVERLOADED_FUNCTION:
64        break;
65
66    EMPTY_SWITCH_DEFAULT_CASE();
67    }
68    return 0;
69}
70/* }}} */
71typedef struct {
72    apply_func_t applyer;
73    zend_class_entry *ce;
74} xc_apply_method_info;
75int xc_apply_method(zend_function *zf, xc_apply_method_info *mi TSRMLS_DC) /* {{{ */
76{
77    /* avoid duplicate apply for shadowed method */
78#ifdef ZEND_ENGINE_2
79    if (mi->ce != zf->common.scope) {
80        /* fprintf(stderr, "avoided duplicate %s\n", zf->common.function_name); */
81        return 0;
82    }
83#else
84    char *name = zf->common.function_name;
85    int name_s = strlen(name) + 1;
86    zend_class_entry *ce;
87    zend_function *ptr;
88
89    for (ce = mi->ce->parent; ce; ce = ce->parent) {
90        if (zend_hash_find(&ce->function_table, name, name_s, (void **) &ptr) == SUCCESS) {
91            if (ptr->op_array.refcount == zf->op_array.refcount) {
92                return 0;
93            }
94        }
95    }
96#endif
97    return xc_apply_function(zf, mi->applyer TSRMLS_CC);
98}
99/* }}} */
100#if 0
101int xc_apply_class(zend_class_entry *ce, apply_func_t applyer TSRMLS_DC) /* {{{ */
102{
103    xc_apply_method_info mi;
104
105    mi.applyer = applyer;
106    mi.ce      = ce;
107    zend_hash_apply_with_argument(&(ce->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
108    return 0;
109}
110/* }}} */
111#endif
112static int xc_apply_cest(xc_cest_t *cest, apply_func_t applyer TSRMLS_DC) /* {{{ */
113{
114    xc_apply_method_info mi;
115
116    mi.applyer = applyer;
117    mi.ce      = CestToCePtr(*cest);
118    zend_hash_apply_with_argument(&(CestToCePtr(*cest)->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
119    return 0;
120}
121/* }}} */
122int xc_apply_op_array(xc_compile_result_t *cr, apply_func_t applyer TSRMLS_DC) /* {{{ */
123{
124    zend_hash_apply_with_argument(cr->function_table, (apply_func_arg_t) xc_apply_function, applyer TSRMLS_CC);
125    zend_hash_apply_with_argument(cr->class_table, (apply_func_arg_t) xc_apply_cest, applyer TSRMLS_CC);
126
127    return applyer(cr->op_array TSRMLS_CC);
128}
129/* }}} */
130
131int xc_undo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
132{
133    zend_op *opline, *end;
134
135    if (!op_array->done_pass_two) {
136        return 0;
137    }
138
139    opline = op_array->opcodes;
140    end = opline + op_array->last;
141    while (opline < end) {
142#ifdef ZEND_ENGINE_2_1
143        switch (opline->opcode) {
144            case ZEND_JMP:
145                opline->op1.u.opline_num = opline->op1.u.jmp_addr - op_array->opcodes;
146                assert(opline->op1.u.opline_num < op_array->last);
147                break;
148            case ZEND_JMPZ:
149            case ZEND_JMPNZ:
150            case ZEND_JMPZ_EX:
151            case ZEND_JMPNZ_EX:
152                opline->op2.u.opline_num = opline->op2.u.jmp_addr - op_array->opcodes;
153                assert(opline->op2.u.opline_num < op_array->last);
154                break;
155        }
156#endif
157        opline++;
158    }
159    op_array->done_pass_two = 0;
160
161    return 0;
162}
163/* }}} */
164int xc_redo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
165{
166    zend_op *opline, *end;
167
168    if (op_array->done_pass_two) {
169        return 0;
170    }
171
172    /*
173    op_array->opcodes = (zend_op *) erealloc(op_array->opcodes, sizeof(zend_op)*op_array->last);
174    op_array->size = op_array->last;
175    */
176
177    opline = op_array->opcodes;
178    end = opline + op_array->last;
179    while (opline < end) {
180        if (opline->op1.op_type == IS_CONST) {
181            opline->op1.u.constant.is_ref = 1;
182            opline->op1.u.constant.refcount = 2; /* Make sure is_ref won't be reset */
183        }
184        if (opline->op2.op_type == IS_CONST) {
185            opline->op2.u.constant.is_ref = 1;
186            opline->op2.u.constant.refcount = 2;
187        }
188#ifdef ZEND_ENGINE_2_1
189        switch (opline->opcode) {
190            case ZEND_JMP:
191                assert(opline->op1.u.opline_num < op_array->last);
192                opline->op1.u.jmp_addr = op_array->opcodes + opline->op1.u.opline_num;
193                break;
194            case ZEND_JMPZ:
195            case ZEND_JMPNZ:
196            case ZEND_JMPZ_EX:
197            case ZEND_JMPNZ_EX:
198                assert(opline->op2.u.opline_num < op_array->last);
199                opline->op2.u.jmp_addr = op_array->opcodes + opline->op2.u.opline_num;
200                break;
201        }
202        ZEND_VM_SET_OPCODE_HANDLER(opline);
203#endif
204        opline++;
205    }
206
207    op_array->done_pass_two = 1;
208    return 0;
209}
210/* }}} */
211
[8]212#ifdef HAVE_XCACHE_OPCODE_SPEC_DEF
[1]213static void xc_fix_opcode_ex_znode(int tofix, xc_op_spec_t spec, znode *znode, int type TSRMLS_DC) /* {{{ */
214{
215#ifdef ZEND_ENGINE_2
216    if ((znode->op_type != IS_UNUSED && (spec == OPSPEC_UCLASS || spec == OPSPEC_CLASS)) ||
217            spec == OPSPEC_FETCH) {
218        if (tofix) {
219            switch (znode->op_type) {
220            case IS_VAR:
221            case IS_TMP_VAR:
222                break;
223
224            default:
225                /* TODO: data lost, find a way to keep it */
226                /* assert(znode->op_type == IS_CONST); */
227                znode->op_type = IS_TMP_VAR;
228            }
229        }
230    }
231    switch (znode->op_type) {
232    case IS_TMP_VAR:
233    case IS_VAR:
234        if (tofix) {
235            znode->u.var /= sizeof(temp_variable);
236        }
237        else {
238            znode->u.var *= sizeof(temp_variable);
239        }
240    }
241#endif
242}
243/* }}} */
244
245static void xc_fix_opcode_ex(zend_op_array *op_array, int tofix TSRMLS_DC) /* {{{ */
246{
247    zend_op *opline;
[11]248    zend_uint i;
[1]249
250    opline = op_array->opcodes;
251    for (i = 0; i < op_array->last; i ++, opline ++) {
252        /* 3rd optimizer may have ... */
253        if (opline->opcode < xc_get_opcode_spec_count()) {
254            const xc_opcode_spec_t *spec;
255            spec = xc_get_opcode_spec(opline->opcode);
256
257            xc_fix_opcode_ex_znode(tofix, spec->op1, &opline->op1, 0 TSRMLS_CC);
258            xc_fix_opcode_ex_znode(tofix, spec->op2, &opline->op2, 1 TSRMLS_CC);
259            xc_fix_opcode_ex_znode(tofix, spec->res, &opline->result, 2 TSRMLS_CC);
260        }
261    }
262}
263/* }}} */
264int xc_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
265{
266    xc_fix_opcode_ex(op_array, 1 TSRMLS_CC);
267    return 0;
268}
269/* }}} */
270int xc_undo_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
271{
272    xc_fix_opcode_ex(op_array, 0 TSRMLS_CC);
273
274    return 0;
275}
276/* }}} */
[8]277#endif
[1]278
[212]279int xc_foreach_early_binding_class(zend_op_array *op_array, void (*callback)(zend_op *opline, int oplineno, void *data TSRMLS_DC), void *data TSRMLS_DC) /* {{{ */
280{
281    zend_op *opline, *begin, *end, *next = NULL;
282    xc_cest_t cest;
283
284    opline = begin = op_array->opcodes;
285    end = opline + op_array->last;
286    while (opline < end) {
287        switch (opline->opcode) {
288            case ZEND_JMP:
289                next = begin + opline->op1.u.opline_num;
290                break;
291
292            case ZEND_JMPZNZ:
293                next = begin + max(opline->op2.u.opline_num, opline->extended_value);
294                break;
295
296            case ZEND_JMPZ:
297            case ZEND_JMPNZ:
298            case ZEND_JMPZ_EX:
299            case ZEND_JMPNZ_EX:
300                next = begin + opline->op2.u.opline_num;
301                break;
302
303            case ZEND_RETURN:
304                opline = end;
305                break;
306
307            case ZEND_DECLARE_INHERITED_CLASS:
308                callback(opline, opline - begin, data TSRMLS_CC);
309                break;
310        }
311
312        if (opline < next) {
313            opline = next;
314        }
315        else {
316            opline ++;
317        }
318    }
319}
320/* }}} */
321int xc_do_early_binding(zend_op_array *op_array, HashTable *class_table, int oplineno TSRMLS_DC) /* {{{ */
322{
323    zend_op *opline, *opcodes;
324
325#ifdef DEBUG
326    fprintf(stderr, "binding %d\n", oplineno);
327#endif
328    assert(oplineno >= 0);
329
330    /* do early binding */
331    opline = &(op_array->opcodes[oplineno]);
332
333    switch (opline->opcode) {
334    case ZEND_DECLARE_INHERITED_CLASS:
335#ifdef ZEND_ENGINE_2
336        {
337            zval *parent_name;
338            zend_class_entry **pce;
339            parent_name = &(opline - 1)->op2.u.constant;
340#ifdef DEBUG
341            fprintf(stderr, "binding with parent %s\n", Z_STRVAL_P(parent_name));
342#endif
343            if (zend_lookup_class(Z_STRVAL_P(parent_name), Z_STRLEN_P(parent_name), &pce TSRMLS_CC) == FAILURE) {
344                return FAILURE;
345            }
346
347            if (do_bind_inherited_class(opline, class_table, *pce, 1 TSRMLS_CC) == NULL) {
348                return FAILURE;
349            }
350        }
351#else
352        if (do_bind_function_or_class(opline, NULL, class_table, 1) == FAILURE) {
353            return FAILURE;
354        }
355#endif
356
357#ifdef ZEND_FETCH_CLASS
358        /* clear unnecessary ZEND_FETCH_CLASS opcode */
359        if (opline > op_array->opcodes
360         && (opline - 1)->opcode == ZEND_FETCH_CLASS) {
361            zend_op *fetch_class_opline = opline - 1;
362
363#ifdef DEBUG
364            fprintf(stderr, "%s %p\n", Z_STRVAL(fetch_class_opline->op2.u.constant), Z_STRVAL(fetch_class_opline->op2.u.constant));
365#endif
366            OP_ZVAL_DTOR(fetch_class_opline->op2);
367            fetch_class_opline->opcode = ZEND_NOP;
368            ZEND_VM_SET_OPCODE_HANDLER(fetch_class_opline);
369            memset(&fetch_class_opline->op1, 0, sizeof(znode));
370            memset(&fetch_class_opline->op2, 0, sizeof(znode));
371            SET_UNUSED(fetch_class_opline->op1);
372            SET_UNUSED(fetch_class_opline->op2);
373            SET_UNUSED(fetch_class_opline->result);
374        }
375#endif
376        break;
377
378    default:
379        return FAILURE;
380    }
381
382    zend_hash_del(class_table, opline->op1.u.constant.value.str.val, opline->op1.u.constant.value.str.len);
383    OP_ZVAL_DTOR(opline->op1);
384    OP_ZVAL_DTOR(opline->op2);
385    opline->opcode = ZEND_NOP;
386    ZEND_VM_SET_OPCODE_HANDLER(opline);
387    memset(&opline->op1, 0, sizeof(znode));
388    memset(&opline->op2, 0, sizeof(znode));
389    SET_UNUSED(opline->op1);
390    SET_UNUSED(opline->op2);
391    return SUCCESS;
392}
393/* }}} */
394
[95]395#ifdef HAVE_XCACHE_CONSTANT
[103]396void xc_install_constant(char *filename, zend_constant *constant, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
[95]397{
398    if (zend_u_hash_add(EG(zend_constants), type, key, len,
399                constant, sizeof(zend_constant),
400                NULL
401                ) == FAILURE) {
402        CG(in_compilation) = 1;
403        CG(compiled_filename) = filename;
404        CG(zend_lineno) = 0;
[103]405#ifdef IS_UNICODE
406        zend_error(E_NOTICE, "Constant %R already defined", type, key);
407#else
[95]408        zend_error(E_NOTICE, "Constant %s already defined", key);
[103]409#endif
410        free(ZSTR_V(constant->name));
[95]411        if (!(constant->flags & CONST_PERSISTENT)) {
412            zval_dtor(&constant->value);
413        }
414    }
415}
416/* }}} */
417#endif
[103]418void xc_install_function(char *filename, zend_function *func, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
[1]419{
[103]420    zend_bool istmpkey;
421
[1]422    if (func->type == ZEND_USER_FUNCTION) {
[103]423#ifdef IS_UNICODE
424        istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
425#else
426        istmpkey = ZSTR_S(key)[0] == 0;
427#endif
428        if (istmpkey) {
[88]429            zend_u_hash_update(CG(function_table), type, key, len,
430                        func, sizeof(zend_op_array),
431                        NULL
432                        );
433        }
434        else if (zend_u_hash_add(CG(function_table), type, key, len,
[1]435                    func, sizeof(zend_op_array),
436                    NULL
437                    ) == FAILURE) {
438            CG(in_compilation) = 1;
439            CG(compiled_filename) = filename;
440            CG(zend_lineno) = ZESW(func->op_array.opcodes[0].lineno, func->op_array.line_start);
[103]441#ifdef IS_UNICODE
442            zend_error(E_ERROR, "Cannot redeclare %R()", type, key);
443#else
[1]444            zend_error(E_ERROR, "Cannot redeclare %s()", key);
[103]445#endif
[1]446        }
447    }
448}
449/* }}} */
[212]450ZESW(xc_cest_t *, void) xc_install_class(char *filename, xc_cest_t *cest, int oplineno, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
[1]451{
[103]452    zend_bool istmpkey;
[11]453    zend_class_entry *cep = CestToCePtr(*cest);
[19]454    ZESW(void *stored_ce_ptr, NOTHING);
[1]455
[103]456#ifdef IS_UNICODE
457    istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
458#else
459    istmpkey = ZSTR_S(key)[0] == 0;
460#endif
461    if (istmpkey) {
[88]462        zend_u_hash_update(CG(class_table), type, key, len,
463                    cest, sizeof(xc_cest_t),
464                    ZESW(&stored_ce_ptr, NULL)
465                    );
[212]466        if (oplineno != -1) {
467            xc_do_early_binding(CG(active_op_array), CG(class_table), oplineno TSRMLS_CC);
468        }
[88]469    }
470    else if (zend_u_hash_add(CG(class_table), type, key, len,
[1]471                cest, sizeof(xc_cest_t),
472                ZESW(&stored_ce_ptr, NULL)
473                ) == FAILURE) {
474        CG(in_compilation) = 1;
475        CG(compiled_filename) = filename;
476        CG(zend_lineno) = ZESW(0, cep->line_start);
[103]477#ifdef IS_UNICODE
478        zend_error(E_ERROR, "Cannot redeclare class %R", type, cep->name);
479#else
480        zend_error(E_ERROR, "Cannot redeclare class %s", cep->name);
481#endif
[212]482        assert(oplineno == -1);
[1]483    }
[19]484    ZESW(return (xc_cest_t *) stored_ce_ptr, NOTHING);
[1]485}
486/* }}} */
487
488/* sandbox {{{ */
489#undef TG
490#undef OG
491#define TG(x) (sandbox->tmp_##x)
492#define OG(x) (sandbox->orig_##x)
493/* }}} */
494xc_sandbox_t *xc_sandbox_init(xc_sandbox_t *sandbox, char *filename TSRMLS_DC) /* {{{ */
495{
496    if (sandbox) {
497        memset(sandbox, 0, sizeof(sandbox[0]));
498    }
499    else {
500        ECALLOC_ONE(sandbox);
501        sandbox->alloc = 1;
502    }
[95]503
[1]504    memcpy(&OG(included_files), &EG(included_files), sizeof(EG(included_files)));
505
[95]506#ifdef HAVE_XCACHE_CONSTANT
507    OG(zend_constants) = EG(zend_constants);
508    EG(zend_constants) = &TG(zend_constants);
509#endif
510
[1]511    OG(function_table) = CG(function_table);
512    CG(function_table) = &TG(function_table);
513
514    assert(EG(class_table) == CG(class_table));
515
516    OG(class_table) = CG(class_table);
517    CG(class_table) = &TG(class_table);
518    EG(class_table) = CG(class_table);
519
520    TG(included_files) = &EG(included_files);
521
522    zend_hash_init_ex(TG(included_files), 5, NULL, NULL, 0, 1);
[95]523#ifdef HAVE_XCACHE_CONSTANT
[203]524    zend_hash_init_ex(&TG(zend_constants), 20, NULL, OG(zend_constants)->pDestructor, 1, 0);
[95]525#endif
[203]526    zend_hash_init_ex(&TG(function_table), 128, NULL, OG(function_table)->pDestructor, 0, 0);
527    zend_hash_init_ex(&TG(class_table), 16, NULL, OG(class_table)->pDestructor, 0, 0);
[1]528
529    sandbox->filename = filename;
530
[209]531#ifdef E_STRICT
532    sandbox->orig_user_error_handler_error_reporting = EG(user_error_handler_error_reporting);
533    EG(user_error_handler_error_reporting) &= ~E_STRICT;
534#endif
535
[1]536    return sandbox;
537}
538/* }}} */
[212]539static void xc_early_binding_cb(zend_op *opline, int oplineno, void *data TSRMLS_DC) /* {{{ */
540{
541    xc_sandbox_t *sandbox = (xc_sandbox_t *) data;
542    xc_do_early_binding(CG(active_op_array), OG(class_table), oplineno TSRMLS_CC);
543}
544/* }}} */
[1]545static void xc_sandbox_install(xc_sandbox_t *sandbox TSRMLS_DC) /* {{{ */
546{
547    int i;
548    Bucket *b;
549
[95]550#ifdef HAVE_XCACHE_CONSTANT
551    b = TG(zend_constants).pListHead;
552    /* install constants */
553    while (b != NULL) {
554        zend_constant *c = (zend_constant*) b->pData;
555        xc_install_constant(sandbox->filename, c,
[200]556                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength TSRMLS_CC);
[95]557        b = b->pListNext;
558    }
559#endif
560
[1]561    b = TG(function_table).pListHead;
562    /* install function */
563    while (b != NULL) {
564        zend_function *func = (zend_function*) b->pData;
565        xc_install_function(sandbox->filename, func,
[200]566                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength TSRMLS_CC);
[1]567        b = b->pListNext;
568    }
569
570    b = TG(class_table).pListHead;
571    /* install class */
572    while (b != NULL) {
[212]573        xc_install_class(sandbox->filename, (xc_cest_t*) b->pData, -1,
[200]574                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength TSRMLS_CC);
[1]575        b = b->pListNext;
576    }
[212]577    xc_undo_pass_two(CG(active_op_array) TSRMLS_CC);
578    xc_foreach_early_binding_class(CG(active_op_array), xc_early_binding_cb, (void *) sandbox TSRMLS_CC);
579    xc_redo_pass_two(CG(active_op_array) TSRMLS_CC);
[1]580
581    i = 1;
582    zend_hash_add(&OG(included_files), sandbox->filename, strlen(sandbox->filename) + 1, (void *)&i, sizeof(int), NULL);
583}
584/* }}} */
585void xc_sandbox_free(xc_sandbox_t *sandbox, int install TSRMLS_DC) /* {{{ */
586{
587    /* restore first first install function/class */
[95]588#ifdef HAVE_XCACHE_CONSTANT
589    EG(zend_constants) = OG(zend_constants);
590#endif
[1]591    CG(function_table) = OG(function_table);
592    CG(class_table)    = OG(class_table);
593    EG(class_table)    = CG(class_table);
594
595    if (install) {
596        xc_sandbox_install(sandbox TSRMLS_CC);
597
598        /* no free as it's installed */
[95]599#ifdef HAVE_XCACHE_CONSTANT
600        TG(zend_constants).pDestructor = NULL;
601#endif
[1]602        TG(function_table).pDestructor = NULL;
603        TG(class_table).pDestructor = NULL;
604    }
605
606    /* destroy all the tmp */
[95]607#ifdef HAVE_XCACHE_CONSTANT
608    zend_hash_destroy(&TG(zend_constants));
609#endif
[1]610    zend_hash_destroy(&TG(function_table));
611    zend_hash_destroy(&TG(class_table));
612    zend_hash_destroy(TG(included_files));
613
614    /* restore orig here, as EG/CG holded tmp before */
615    memcpy(&EG(included_files), &OG(included_files), sizeof(EG(included_files)));
616
[209]617#ifdef E_STRICT
618    EG(user_error_handler_error_reporting) = sandbox->orig_user_error_handler_error_reporting;
619#endif
620
[1]621    if (sandbox->alloc) {
622        efree(sandbox);
623    }
624}
625/* }}} */
Note: See TracBrowser for help on using the repository browser.