source: trunk/utils.c @ 961

Last change on this file since 961 was 961, checked in by moo, 2 years ago

revert opt out of included_files handling. still required by require()/include(). make it work for new sandbox handling

  • Property svn:eol-style set to native
File size: 30.5 KB
RevLine 
[212]1
[1]2#include "xcache.h"
[344]3#include "stack.h"
4#include "xcache_globals.h"
[1]5#include "utils.h"
6#ifdef ZEND_ENGINE_2_1
7#include "zend_vm.h"
8#endif
9#include "opcode_spec.h"
10#undef NDEBUG
11#include "assert.h"
12
[213]13#ifndef max
14#define max(a, b) ((a) < (b) ? (b) : (a))
15#endif
16
[212]17#ifndef ZEND_VM_SET_OPCODE_HANDLER
18#   define ZEND_VM_SET_OPCODE_HANDLER(opline) do { } while (0)
19#endif
20
[822]21#ifdef ZEND_ENGINE_2_4
22#   define OP_ZVAL_DTOR(op) do { } while(0)
23#else
24#   define OP_ZVAL_DTOR(op) do { \
25        Z_UNSET_ISREF(Z_OP_CONSTANT(op)); \
26        zval_dtor(&Z_OP_CONSTANT(op)); \
27    } while(0)
28#endif
[903]29
30static void (*old_zend_error_cb)(int type, const char *error_filename, const uint error_lineno, const char *format, va_list args) = NULL;
31static void call_old_zend_error_cb(int type, const char *error_filename, const uint error_lineno, const char *format, ...) /* {{{ */
32{
33    va_list args;
34    va_start(args, format);
35    old_zend_error_cb(type, error_filename, error_lineno, format, args);
36}
37/* }}} */
38
[1]39xc_compile_result_t *xc_compile_result_init(xc_compile_result_t *cr, /* {{{ */
40        zend_op_array *op_array,
41        HashTable *function_table,
42        HashTable *class_table)
43{
[935]44    assert(cr);
[1]45    cr->op_array       = op_array;
46    cr->function_table = function_table;
47    cr->class_table    = class_table;
48    return cr;
49}
50/* }}} */
51xc_compile_result_t *xc_compile_result_init_cur(xc_compile_result_t *cr, zend_op_array *op_array TSRMLS_DC) /* {{{ */
52{
[935]53    assert(cr);
[1]54    return xc_compile_result_init(cr, op_array, CG(function_table), CG(class_table));
55}
56/* }}} */
57void xc_compile_result_free(xc_compile_result_t *cr) /* {{{ */
58{
59}
60/* }}} */
61
62int xc_apply_function(zend_function *zf, apply_func_t applyer TSRMLS_DC) /* {{{ */
63{
64    switch (zf->type) {
65    case ZEND_USER_FUNCTION:
66    case ZEND_EVAL_CODE:
67        return applyer(&zf->op_array TSRMLS_CC);
68        break;
69
70    case ZEND_INTERNAL_FUNCTION:
71    case ZEND_OVERLOADED_FUNCTION:
72        break;
73
74    EMPTY_SWITCH_DEFAULT_CASE();
75    }
76    return 0;
77}
78/* }}} */
79typedef struct {
80    apply_func_t applyer;
81    zend_class_entry *ce;
82} xc_apply_method_info;
83int xc_apply_method(zend_function *zf, xc_apply_method_info *mi TSRMLS_DC) /* {{{ */
84{
85    /* avoid duplicate apply for shadowed method */
86#ifdef ZEND_ENGINE_2
87    if (mi->ce != zf->common.scope) {
88        /* fprintf(stderr, "avoided duplicate %s\n", zf->common.function_name); */
89        return 0;
90    }
91#else
92    char *name = zf->common.function_name;
93    int name_s = strlen(name) + 1;
94    zend_class_entry *ce;
95    zend_function *ptr;
96
97    for (ce = mi->ce->parent; ce; ce = ce->parent) {
98        if (zend_hash_find(&ce->function_table, name, name_s, (void **) &ptr) == SUCCESS) {
99            if (ptr->op_array.refcount == zf->op_array.refcount) {
100                return 0;
101            }
102        }
103    }
104#endif
105    return xc_apply_function(zf, mi->applyer TSRMLS_CC);
106}
107/* }}} */
108#if 0
109int xc_apply_class(zend_class_entry *ce, apply_func_t applyer TSRMLS_DC) /* {{{ */
110{
111    xc_apply_method_info mi;
112
113    mi.applyer = applyer;
114    mi.ce      = ce;
115    zend_hash_apply_with_argument(&(ce->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
116    return 0;
117}
118/* }}} */
119#endif
120static int xc_apply_cest(xc_cest_t *cest, apply_func_t applyer TSRMLS_DC) /* {{{ */
121{
122    xc_apply_method_info mi;
123
124    mi.applyer = applyer;
125    mi.ce      = CestToCePtr(*cest);
126    zend_hash_apply_with_argument(&(CestToCePtr(*cest)->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
127    return 0;
128}
129/* }}} */
130int xc_apply_op_array(xc_compile_result_t *cr, apply_func_t applyer TSRMLS_DC) /* {{{ */
131{
[294]132    zend_hash_apply_with_argument(cr->function_table, (apply_func_arg_t) xc_apply_function, (void *) applyer TSRMLS_CC);
133    zend_hash_apply_with_argument(cr->class_table, (apply_func_arg_t) xc_apply_cest, (void *) applyer TSRMLS_CC);
[1]134
135    return applyer(cr->op_array TSRMLS_CC);
136}
137/* }}} */
138int xc_undo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
139{
140    zend_op *opline, *end;
141
[822]142#ifdef ZEND_ENGINE_2_4
143    if (!(op_array->fn_flags & ZEND_ACC_DONE_PASS_TWO)) {
144        return 0;
145    }
146#else
[1]147    if (!op_array->done_pass_two) {
148        return 0;
149    }
[716]150#endif
[1]151
152    opline = op_array->opcodes;
153    end = opline + op_array->last;
154    while (opline < end) {
[822]155#ifdef ZEND_ENGINE_2_4
156        if (opline->op1_type == IS_CONST) {
157            opline->op1.constant = opline->op1.literal - op_array->literals;
158        }
159        if (opline->op2_type == IS_CONST) {
160            opline->op2.constant = opline->op2.literal - op_array->literals;
161        }
162#endif
163
[1]164#ifdef ZEND_ENGINE_2_1
165        switch (opline->opcode) {
[515]166#ifdef ZEND_GOTO
167            case ZEND_GOTO:
168#endif
[1]169            case ZEND_JMP:
[917]170                assert(Z_OP(opline->op1).jmp_addr >= op_array->opcodes && (zend_uint) (Z_OP(opline->op1).jmp_addr - op_array->opcodes) < op_array->last);
[716]171                Z_OP(opline->op1).opline_num = Z_OP(opline->op1).jmp_addr - op_array->opcodes;
[1]172                break;
173            case ZEND_JMPZ:
174            case ZEND_JMPNZ:
175            case ZEND_JMPZ_EX:
176            case ZEND_JMPNZ_EX:
[485]177#ifdef ZEND_JMP_SET
178            case ZEND_JMP_SET:
179#endif
[875]180#ifdef ZEND_JMP_SET_VAR
181            case ZEND_JMP_SET_VAR:
182#endif
[917]183                assert(Z_OP(opline->op2).jmp_addr >= op_array->opcodes && (zend_uint) (Z_OP(opline->op2).jmp_addr - op_array->opcodes) < op_array->last);
[716]184                Z_OP(opline->op2).opline_num = Z_OP(opline->op2).jmp_addr - op_array->opcodes;
[1]185                break;
186        }
187#endif
188        opline++;
189    }
[822]190#ifdef ZEND_ENGINE_2_4
191    op_array->fn_flags &= ~ZEND_ACC_DONE_PASS_TWO;
192#else
[1]193    op_array->done_pass_two = 0;
[716]194#endif
[1]195
196    return 0;
197}
198/* }}} */
199int xc_redo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
200{
201    zend_op *opline, *end;
[822]202#ifdef ZEND_ENGINE_2_4
203    zend_literal *literal = op_array->literals;
204#endif
[1]205
[822]206#ifdef ZEND_ENGINE_2_4
207    if ((op_array->fn_flags & ZEND_ACC_DONE_PASS_TWO)) {
208        return 0;
209    }
210#else
[1]211    if (op_array->done_pass_two) {
212        return 0;
213    }
[716]214#endif
[1]215
216    /*
217    op_array->opcodes = (zend_op *) erealloc(op_array->opcodes, sizeof(zend_op)*op_array->last);
218    op_array->size = op_array->last;
219    */
[822]220#ifdef ZEND_ENGINE_2_4
221    if (literal) {
222        zend_literal *end = literal + op_array->last_literal;
223        while (literal < end) {
224            Z_SET_ISREF(literal->constant);
225            Z_SET_REFCOUNT(literal->constant, 2); /* Make sure is_ref won't be reset */
226            literal++;
227        }
228    }
229#endif
[1]230
231    opline = op_array->opcodes;
232    end = opline + op_array->last;
233    while (opline < end) {
[930]234#ifdef ZEND_ENGINE_2_4
235        if (opline->op1_type == IS_CONST) {
236            opline->op1.literal = op_array->literals + opline->op1.constant;
237        }
238        if (opline->op2_type == IS_CONST) {
239            opline->op2.literal = op_array->literals + opline->op2.constant;
240        }
241#else
[716]242        if (Z_OP_TYPE(opline->op1) == IS_CONST) {
243            Z_SET_ISREF(Z_OP_CONSTANT(opline->op1));
244            Z_SET_REFCOUNT(Z_OP_CONSTANT(opline->op1), 2); /* Make sure is_ref won't be reset */
[1]245        }
[716]246        if (Z_OP_TYPE(opline->op2) == IS_CONST) {
247            Z_SET_ISREF(Z_OP_CONSTANT(opline->op2));
248            Z_SET_REFCOUNT(Z_OP_CONSTANT(opline->op2), 2);
[1]249        }
[822]250#endif
[1]251#ifdef ZEND_ENGINE_2_1
252        switch (opline->opcode) {
[515]253#ifdef ZEND_GOTO
254            case ZEND_GOTO:
255#endif
[1]256            case ZEND_JMP:
[716]257                assert(Z_OP(opline->op1).opline_num < op_array->last);
258                Z_OP(opline->op1).jmp_addr = op_array->opcodes + Z_OP(opline->op1).opline_num;
[1]259                break;
260            case ZEND_JMPZ:
261            case ZEND_JMPNZ:
262            case ZEND_JMPZ_EX:
263            case ZEND_JMPNZ_EX:
[485]264#ifdef ZEND_JMP_SET
265            case ZEND_JMP_SET:
266#endif
[716]267                assert(Z_OP(opline->op2).opline_num < op_array->last);
268                Z_OP(opline->op2).jmp_addr = op_array->opcodes + Z_OP(opline->op2).opline_num;
[1]269                break;
270        }
271        ZEND_VM_SET_OPCODE_HANDLER(opline);
272#endif
273        opline++;
274    }
275
[822]276#ifdef ZEND_ENGINE_2_4
277    op_array->fn_flags |= ZEND_ACC_DONE_PASS_TWO;
278#else
[1]279    op_array->done_pass_two = 1;
[716]280#endif
[1]281    return 0;
282}
283/* }}} */
284
[8]285#ifdef HAVE_XCACHE_OPCODE_SPEC_DEF
[769]286static void xc_fix_opcode_ex_znode(int tofix, xc_op_spec_t spec, Z_OP_TYPEOF_TYPE *op_type, znode_op *op, int type TSRMLS_DC) /* {{{ */
[1]287{
288#ifdef ZEND_ENGINE_2
[716]289    if ((*op_type != IS_UNUSED && (spec == OPSPEC_UCLASS || spec == OPSPEC_CLASS)) ||
[1]290            spec == OPSPEC_FETCH) {
291        if (tofix) {
[716]292            switch (*op_type) {
[1]293            case IS_VAR:
294            case IS_TMP_VAR:
295                break;
296
297            default:
298                /* TODO: data lost, find a way to keep it */
[716]299                /* assert(*op_type == IS_CONST); */
300                *op_type = IS_TMP_VAR;
[1]301            }
302        }
303    }
[716]304    switch (*op_type) {
[1]305    case IS_TMP_VAR:
306    case IS_VAR:
307        if (tofix) {
[716]308            Z_OP(*op).var /= sizeof(temp_variable);
[1]309        }
310        else {
[716]311            Z_OP(*op).var *= sizeof(temp_variable);
[1]312        }
313    }
314#endif
315}
316/* }}} */
317
318static void xc_fix_opcode_ex(zend_op_array *op_array, int tofix TSRMLS_DC) /* {{{ */
319{
320    zend_op *opline;
[11]321    zend_uint i;
[1]322
323    opline = op_array->opcodes;
324    for (i = 0; i < op_array->last; i ++, opline ++) {
325        /* 3rd optimizer may have ... */
326        if (opline->opcode < xc_get_opcode_spec_count()) {
327            const xc_opcode_spec_t *spec;
328            spec = xc_get_opcode_spec(opline->opcode);
329
[716]330            xc_fix_opcode_ex_znode(tofix, spec->op1, &Z_OP_TYPE(opline->op1),    &opline->op1, 0 TSRMLS_CC);
331            xc_fix_opcode_ex_znode(tofix, spec->op2, &Z_OP_TYPE(opline->op2),    &opline->op2, 1 TSRMLS_CC);
332            xc_fix_opcode_ex_znode(tofix, spec->res, &Z_OP_TYPE(opline->result), &opline->result, 2 TSRMLS_CC);
[1]333        }
334    }
335}
336/* }}} */
337int xc_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
338{
339    xc_fix_opcode_ex(op_array, 1 TSRMLS_CC);
340    return 0;
341}
342/* }}} */
343int xc_undo_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
344{
345    xc_fix_opcode_ex(op_array, 0 TSRMLS_CC);
346
347    return 0;
348}
349/* }}} */
[8]350#endif
[1]351
[212]352int 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) /* {{{ */
353{
354    zend_op *opline, *begin, *end, *next = NULL;
355
356    opline = begin = op_array->opcodes;
357    end = opline + op_array->last;
358    while (opline < end) {
359        switch (opline->opcode) {
[515]360#ifdef ZEND_GOTO
361            case ZEND_GOTO:
362#endif
[212]363            case ZEND_JMP:
[716]364                next = begin + Z_OP(opline->op1).opline_num;
[212]365                break;
366
367            case ZEND_JMPZNZ:
[716]368                next = begin + max(Z_OP(opline->op2).opline_num, opline->extended_value);
[212]369                break;
370
371            case ZEND_JMPZ:
372            case ZEND_JMPNZ:
373            case ZEND_JMPZ_EX:
374            case ZEND_JMPNZ_EX:
[485]375#ifdef ZEND_JMP_SET
376            case ZEND_JMP_SET:
377#endif
[716]378                next = begin + Z_OP(opline->op2).opline_num;
[212]379                break;
380
381            case ZEND_RETURN:
382                opline = end;
383                break;
384
[226]385#ifdef ZEND_ENGINE_2
[212]386            case ZEND_DECLARE_INHERITED_CLASS:
387                callback(opline, opline - begin, data TSRMLS_CC);
388                break;
[226]389#else
390            case ZEND_DECLARE_FUNCTION_OR_CLASS:
391                if (opline->extended_value == ZEND_DECLARE_INHERITED_CLASS) {
392                    callback(opline, opline - begin, data TSRMLS_CC);
393                }
394                break;
395#endif
[212]396        }
397
398        if (opline < next) {
399            opline = next;
400        }
401        else {
402            opline ++;
403        }
404    }
[295]405    return SUCCESS;
[212]406}
407/* }}} */
[635]408#ifndef ZEND_COMPILE_DELAYED_BINDING
[214]409static int xc_do_early_binding(zend_op_array *op_array, HashTable *class_table, int oplineno TSRMLS_DC) /* {{{ */
[212]410{
[295]411    zend_op *opline;
[212]412
[305]413    TRACE("binding %d", oplineno);
[212]414    assert(oplineno >= 0);
415
416    /* do early binding */
417    opline = &(op_array->opcodes[oplineno]);
418
419    switch (opline->opcode) {
[226]420#ifdef ZEND_ENGINE_2
[212]421    case ZEND_DECLARE_INHERITED_CLASS:
422        {
423            zval *parent_name;
424            zend_class_entry **pce;
[214]425
426            /* don't early-bind classes that implement interfaces */
427            if ((opline + 1)->opcode == ZEND_FETCH_CLASS && (opline + 2)->opcode == ZEND_ADD_INTERFACE) {
428                return FAILURE;
429            }
430
[716]431            parent_name = &(Z_OP_CONSTANT((opline - 1)->op2));
[305]432            TRACE("binding with parent %s", Z_STRVAL_P(parent_name));
[212]433            if (zend_lookup_class(Z_STRVAL_P(parent_name), Z_STRLEN_P(parent_name), &pce TSRMLS_CC) == FAILURE) {
434                return FAILURE;
435            }
436
437            if (do_bind_inherited_class(opline, class_table, *pce, 1 TSRMLS_CC) == NULL) {
438                return FAILURE;
439            }
440        }
441
442        /* clear unnecessary ZEND_FETCH_CLASS opcode */
443        if (opline > op_array->opcodes
444         && (opline - 1)->opcode == ZEND_FETCH_CLASS) {
445            zend_op *fetch_class_opline = opline - 1;
446
[716]447            TRACE("%s %p", Z_STRVAL(Z_OP_CONSTANT(fetch_class_opline->op2)), Z_STRVAL(Z_OP_CONSTANT(fetch_class_opline->op2)));
[212]448            OP_ZVAL_DTOR(fetch_class_opline->op2);
449            fetch_class_opline->opcode = ZEND_NOP;
450            ZEND_VM_SET_OPCODE_HANDLER(fetch_class_opline);
451            memset(&fetch_class_opline->op1, 0, sizeof(znode));
452            memset(&fetch_class_opline->op2, 0, sizeof(znode));
453            SET_UNUSED(fetch_class_opline->op1);
454            SET_UNUSED(fetch_class_opline->op2);
455            SET_UNUSED(fetch_class_opline->result);
456        }
[214]457
458        /* clear unnecessary ZEND_VERIFY_ABSTRACT_CLASS opcode */
459        if ((opline + 1)->opcode == ZEND_VERIFY_ABSTRACT_CLASS) {
460            zend_op *abstract_op = opline + 1;
461            memset(abstract_op, 0, sizeof(abstract_op[0]));
462            abstract_op->lineno = 0;
463            SET_UNUSED(abstract_op->op1);
464            SET_UNUSED(abstract_op->op2);
465            SET_UNUSED(abstract_op->result);
466            abstract_op->opcode = ZEND_NOP;
467            ZEND_VM_SET_OPCODE_HANDLER(abstract_op);
468        }
469#else
[226]470    case ZEND_DECLARE_FUNCTION_OR_CLASS:
[214]471        if (do_bind_function_or_class(opline, NULL, class_table, 1) == FAILURE) {
472            return FAILURE;
473        }
[212]474#endif
475        break;
476
477    default:
478        return FAILURE;
479    }
480
[716]481    zend_hash_del(class_table, Z_OP_CONSTANT(opline->op1).value.str.val, Z_OP_CONSTANT(opline->op1).value.str.len);
[212]482    OP_ZVAL_DTOR(opline->op1);
483    OP_ZVAL_DTOR(opline->op2);
484    opline->opcode = ZEND_NOP;
485    ZEND_VM_SET_OPCODE_HANDLER(opline);
486    memset(&opline->op1, 0, sizeof(znode));
487    memset(&opline->op2, 0, sizeof(znode));
488    SET_UNUSED(opline->op1);
489    SET_UNUSED(opline->op2);
490    return SUCCESS;
491}
492/* }}} */
[635]493#endif
[212]494
[95]495#ifdef HAVE_XCACHE_CONSTANT
[860]496void xc_install_constant(ZEND_24(NOTHING, const) char *filename, zend_constant *constant, zend_uchar type, const24_zstr key, uint len, ulong h TSRMLS_DC) /* {{{ */
[95]497{
498    if (zend_u_hash_add(EG(zend_constants), type, key, len,
499                constant, sizeof(zend_constant),
500                NULL
501                ) == FAILURE) {
502        CG(zend_lineno) = 0;
[103]503#ifdef IS_UNICODE
504        zend_error(E_NOTICE, "Constant %R already defined", type, key);
505#else
[95]506        zend_error(E_NOTICE, "Constant %s already defined", key);
[103]507#endif
508        free(ZSTR_V(constant->name));
[95]509        if (!(constant->flags & CONST_PERSISTENT)) {
510            zval_dtor(&constant->value);
511        }
512    }
513}
514/* }}} */
515#endif
[860]516void xc_install_function(ZEND_24(NOTHING, const) char *filename, zend_function *func, zend_uchar type, const24_zstr key, uint len, ulong h TSRMLS_DC) /* {{{ */
[1]517{
[103]518    zend_bool istmpkey;
519
[1]520    if (func->type == ZEND_USER_FUNCTION) {
[103]521#ifdef IS_UNICODE
522        istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
523#else
524        istmpkey = ZSTR_S(key)[0] == 0;
525#endif
526        if (istmpkey) {
[88]527            zend_u_hash_update(CG(function_table), type, key, len,
528                        func, sizeof(zend_op_array),
529                        NULL
530                        );
531        }
532        else if (zend_u_hash_add(CG(function_table), type, key, len,
[1]533                    func, sizeof(zend_op_array),
534                    NULL
535                    ) == FAILURE) {
536            CG(zend_lineno) = ZESW(func->op_array.opcodes[0].lineno, func->op_array.line_start);
[103]537#ifdef IS_UNICODE
538            zend_error(E_ERROR, "Cannot redeclare %R()", type, key);
539#else
[1]540            zend_error(E_ERROR, "Cannot redeclare %s()", key);
[103]541#endif
[1]542        }
543    }
544}
545/* }}} */
[860]546ZESW(xc_cest_t *, void) xc_install_class(ZEND_24(NOTHING, const) char *filename, xc_cest_t *cest, int oplineno, zend_uchar type, const24_zstr key, uint len, ulong h TSRMLS_DC) /* {{{ */
[1]547{
[103]548    zend_bool istmpkey;
[11]549    zend_class_entry *cep = CestToCePtr(*cest);
[19]550    ZESW(void *stored_ce_ptr, NOTHING);
[1]551
[103]552#ifdef IS_UNICODE
553    istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
554#else
555    istmpkey = ZSTR_S(key)[0] == 0;
556#endif
557    if (istmpkey) {
[506]558        zend_u_hash_quick_update(CG(class_table), type, key, len, h,
[88]559                    cest, sizeof(xc_cest_t),
560                    ZESW(&stored_ce_ptr, NULL)
561                    );
[548]562#ifndef ZEND_COMPILE_DELAYED_BINDING
[212]563        if (oplineno != -1) {
564            xc_do_early_binding(CG(active_op_array), CG(class_table), oplineno TSRMLS_CC);
565        }
[548]566#endif
[88]567    }
[506]568    else if (zend_u_hash_quick_add(CG(class_table), type, key, len, h,
[1]569                cest, sizeof(xc_cest_t),
570                ZESW(&stored_ce_ptr, NULL)
571                ) == FAILURE) {
[716]572        CG(zend_lineno) = ZESW(0, Z_CLASS_INFO(*cep).line_start);
[103]573#ifdef IS_UNICODE
574        zend_error(E_ERROR, "Cannot redeclare class %R", type, cep->name);
575#else
576        zend_error(E_ERROR, "Cannot redeclare class %s", cep->name);
577#endif
[212]578        assert(oplineno == -1);
[1]579    }
[19]580    ZESW(return (xc_cest_t *) stored_ce_ptr, NOTHING);
[1]581}
582/* }}} */
583
[937]584typedef struct { /* sandbox {{{ */
585    ZEND_24(NOTHING, const) char *filename;
586
587    HashTable orig_included_files;
588    HashTable *tmp_included_files;
589
590#ifdef HAVE_XCACHE_CONSTANT
591    HashTable *orig_zend_constants;
592    HashTable tmp_zend_constants;
593#endif
594    HashTable *orig_function_table;
595    HashTable *orig_class_table;
596    HashTable *orig_auto_globals;
597    HashTable tmp_function_table;
598    HashTable tmp_class_table;
599    HashTable tmp_auto_globals;
600#ifdef HAVE_XCACHE_CONSTANT
601    Bucket    *tmp_internal_constant_tail;
602#endif
603    Bucket    *tmp_internal_function_tail;
604    Bucket    *tmp_internal_class_tail;
605
606#ifdef XCACHE_ERROR_CACHING
607    int orig_user_error_handler_error_reporting;
608    zend_uint compilererror_cnt;
609    zend_uint compilererror_size;
610    xc_compilererror_t *compilererrors;
611#endif
612
613#ifdef ZEND_COMPILE_IGNORE_INTERNAL_CLASSES
614    zend_uint orig_compiler_options;
615#endif
616} xc_sandbox_t;
617
[1]618#undef TG
619#undef OG
620#define TG(x) (sandbox->tmp_##x)
621#define OG(x) (sandbox->orig_##x)
622/* }}} */
[669]623#ifdef XCACHE_ERROR_CACHING
[496]624static void xc_sandbox_error_cb(int type, const char *error_filename, const uint error_lineno, const char *format, va_list args) /* {{{ */
625{
626    xc_compilererror_t *compilererror;
627    xc_sandbox_t *sandbox;
628    TSRMLS_FETCH();
629
630    sandbox = (xc_sandbox_t *) XG(sandbox);
[903]631    if (!sandbox) {
632        old_zend_error_cb(type, error_filename, error_lineno, format, args);
633        return;
634    }
635
[648]636    switch (type) {
[669]637#ifdef E_STRICT
[648]638    case E_STRICT:
[668]639#endif
[648]640#ifdef E_DEPRECATED
641    case E_DEPRECATED:
642#endif
643        if (sandbox->compilererror_cnt <= sandbox->compilererror_size) {
644            if (sandbox->compilererror_size) {
645                sandbox->compilererror_size += 16;
646                sandbox->compilererrors = erealloc(sandbox->compilererrors, sandbox->compilererror_size * sizeof(sandbox->compilererrors));
647            }
648            else {
649                sandbox->compilererror_size = 16;
650                sandbox->compilererrors = emalloc(sandbox->compilererror_size * sizeof(sandbox->compilererrors));
651            }
652        }
653        compilererror = &sandbox->compilererrors[sandbox->compilererror_cnt++];
654        compilererror->type = type;
655        compilererror->lineno = error_lineno;
656        compilererror->error_len = vspprintf(&compilererror->error, 0, format, args);
657        break;
658
659    default: {
[500]660        /* give up, and user handler is not supported in this case */
[555]661        zend_uint i;
[903]662        zend_uint old_lineno = CG(zend_lineno);
[500]663
664        for (i = 0; i < sandbox->compilererror_cnt; i ++) {
665            compilererror = &sandbox->compilererrors[i];
666            CG(zend_lineno) = compilererror->lineno;
[903]667            call_old_zend_error_cb(compilererror->type, error_filename, error_lineno, "%s", compilererror->error);
668            efree(compilererror->error);
[500]669        }
[903]670        if (sandbox->compilererrors) {
671            efree(sandbox->compilererrors);
672            sandbox->compilererrors = NULL;
673        }
674        sandbox->compilererror_cnt  = 0;
675        sandbox->compilererror_size = 0;
[500]676
[903]677        CG(zend_lineno) = old_lineno;
678        old_zend_error_cb(type, error_filename, error_lineno, format, args);
[648]679        break;
[500]680    }
[496]681    }
682}
683/* }}} */
684#endif
[268]685#ifdef ZEND_ENGINE_2_1
[860]686static zend_bool xc_auto_global_callback(ZEND_24(NOTHING, const) char *name, uint name_len TSRMLS_DC) /* {{{ */
[268]687{
688    return 0;
689}
690/* }}} */
691static int xc_auto_global_arm(zend_auto_global *auto_global TSRMLS_DC) /* {{{ */
692{
693    if (auto_global->auto_global_callback) {
694        auto_global->armed = 1;
695        auto_global->auto_global_callback = xc_auto_global_callback;
696    }
697    else {
698        auto_global->armed = 0;
699    }
700    return ZEND_HASH_APPLY_KEEP;
701}
702/* }}} */
703#endif
[345]704
[640]705void xc_hash_copy_if(HashTable *target, HashTable *source, copy_ctor_func_t pCopyConstructor, void *tmp, uint size, xc_if_func_t checker) /* {{{ */
[588]706{
[640]707    Bucket *p;
708    void *new_entry;
709    zend_bool setTargetPointer;
710
711    setTargetPointer = !target->pInternalPointer;
712    p = source->pListHead;
713    while (p) {
714        if (checker(p->pData)) {
715            if (setTargetPointer && source->pInternalPointer == p) {
716                target->pInternalPointer = NULL;
717            }
718            if (p->nKeyLength) {
[660]719                zend_u_hash_quick_update(target, p->key.type, ZSTR(BUCKET_KEY_S(p)), p->nKeyLength, p->h, p->pData, size, &new_entry);
[640]720            } else {
721                zend_hash_index_update(target, p->h, p->pData, size, &new_entry);
722            }
723            if (pCopyConstructor) {
724                pCopyConstructor(new_entry);
725            }
726        }
727        p = p->pListNext;
728    }
729    if (!target->pInternalPointer) {
730        target->pInternalPointer = target->pListHead;
731    }
732}
733/* }}} */
734#ifdef HAVE_XCACHE_CONSTANT
735static zend_bool xc_is_internal_zend_constant(zend_constant *c) /* {{{ */
736{
737    return (c->flags & CONST_PERSISTENT) ? 1 : 0;
738}
739/* }}} */
740void xc_zend_constant_ctor(zend_constant *c) /* {{{ */
741{
742    assert((c->flags & CONST_PERSISTENT));
[681]743    ZSTR_U(c->name) = UNISW(zend_strndup, zend_ustrndup)(ZSTR_U(c->name), c->name_len - 1);
[588]744}
745/* }}} */
[640]746void xc_zend_constant_dtor(zend_constant *c) /* {{{ */
747{
[681]748    free(ZSTR_V(c->name));
[640]749}
750/* }}} */
[705]751static void xc_free_zend_constant(zend_constant *c) /* {{{ */
[681]752{
753    if (!(c->flags & CONST_PERSISTENT)) {
754        zval_dtor(&c->value);
755    }
756    free(ZSTR_V(c->name));
757}
758/* }}} */
[640]759void xc_copy_internal_zend_constants(HashTable *target, HashTable *source) /* {{{ */
760{
761    zend_constant tmp_const;
762    xc_hash_copy_if(target, source, (copy_ctor_func_t) xc_zend_constant_ctor, (void *) &tmp_const, sizeof(zend_constant), (xc_if_func_t) xc_is_internal_zend_constant);
763}
764/* }}} */
765#endif
[937]766
[938]767static xc_sandbox_t *xc_sandbox_init(xc_sandbox_t *sandbox, ZEND_24(NOTHING, const) char *filename TSRMLS_DC) /* {{{ */
[1]768{
[269]769    HashTable *h;
[344]770
[935]771    assert(sandbox);
772    memset(sandbox, 0, sizeof(sandbox[0]));
[95]773
[1]774    memcpy(&OG(included_files), &EG(included_files), sizeof(EG(included_files)));
775
[95]776#ifdef HAVE_XCACHE_CONSTANT
777    OG(zend_constants) = EG(zend_constants);
778    EG(zend_constants) = &TG(zend_constants);
779#endif
780
[1]781    OG(function_table) = CG(function_table);
782    CG(function_table) = &TG(function_table);
783
784    OG(class_table) = CG(class_table);
785    CG(class_table) = &TG(class_table);
786    EG(class_table) = CG(class_table);
787
[268]788#ifdef ZEND_ENGINE_2_1
789    OG(auto_globals) = CG(auto_globals);
790    CG(auto_globals) = &TG(auto_globals);
791#endif
792
[1]793    TG(included_files) = &EG(included_files);
794
795    zend_hash_init_ex(TG(included_files), 5, NULL, NULL, 0, 1);
[95]796#ifdef HAVE_XCACHE_CONSTANT
[269]797    h = OG(zend_constants);
[681]798    zend_hash_init_ex(&TG(zend_constants),  20, NULL, (dtor_func_t) xc_free_zend_constant, h->persistent, h->bApplyProtection);
[640]799    xc_copy_internal_zend_constants(&TG(zend_constants), &XG(internal_constant_table));
[651]800    TG(internal_constant_tail) = TG(zend_constants).pListTail;
[95]801#endif
[270]802    h = OG(function_table);
[682]803    zend_hash_init_ex(&TG(function_table), 128, NULL, ZEND_FUNCTION_DTOR, h->persistent, h->bApplyProtection);
[346]804    {
805        zend_function tmp_func;
[504]806        zend_hash_copy(&TG(function_table), &XG(internal_function_table), NULL, (void *) &tmp_func, sizeof(tmp_func));
[346]807    }
[434]808    TG(internal_function_tail) = TG(function_table).pListTail;
[344]809
[270]810    h = OG(class_table);
[682]811    zend_hash_init_ex(&TG(class_table),     16, NULL, ZEND_CLASS_DTOR, h->persistent, h->bApplyProtection);
[346]812#if 0 && TODO
813    {
814        xc_cest_t tmp_cest;
[504]815        zend_hash_copy(&TG(class_table), &XG(internal_class_table), NULL, (void *) &tmp_cest, sizeof(tmp_cest));
[346]816    }
817#endif
[344]818    TG(internal_class_tail) = TG(class_table).pListTail;
819
[268]820#ifdef ZEND_ENGINE_2_1
[269]821    /* shallow copy, don't destruct */
[270]822    h = OG(auto_globals);
[269]823    zend_hash_init_ex(&TG(auto_globals),     8, NULL,           NULL, h->persistent, h->bApplyProtection);
[268]824    {
825        zend_auto_global tmp_autoglobal;
[1]826
[268]827        zend_hash_copy(&TG(auto_globals), OG(auto_globals), NULL, (void *) &tmp_autoglobal, sizeof(tmp_autoglobal));
828        zend_hash_apply(&TG(auto_globals), (apply_func_t) xc_auto_global_arm TSRMLS_CC);
829    }
830#endif
831
[1]832    sandbox->filename = filename;
833
[669]834#ifdef XCACHE_ERROR_CACHING
[209]835    sandbox->orig_user_error_handler_error_reporting = EG(user_error_handler_error_reporting);
[496]836    EG(user_error_handler_error_reporting) = 0;
837
838    sandbox->compilererror_cnt  = 0;
839    sandbox->compilererror_size = 0;
[209]840#endif
841
[548]842#ifdef ZEND_COMPILE_IGNORE_INTERNAL_CLASSES
843    sandbox->orig_compiler_options = CG(compiler_options);
844    /* Using ZEND_COMPILE_IGNORE_INTERNAL_CLASSES for ZEND_FETCH_CLASS_RT_NS_CHECK
845     */
[588]846    CG(compiler_options) |= ZEND_COMPILE_IGNORE_INTERNAL_CLASSES | ZEND_COMPILE_NO_CONSTANT_SUBSTITUTION | ZEND_COMPILE_DELAYED_BINDING;
[548]847#endif
848
[496]849    XG(sandbox) = (void *) sandbox;
[1]850    return sandbox;
851}
852/* }}} */
[635]853#ifndef ZEND_COMPILE_DELAYED_BINDING
[212]854static void xc_early_binding_cb(zend_op *opline, int oplineno, void *data TSRMLS_DC) /* {{{ */
855{
856    xc_sandbox_t *sandbox = (xc_sandbox_t *) data;
857    xc_do_early_binding(CG(active_op_array), OG(class_table), oplineno TSRMLS_CC);
858}
859/* }}} */
[635]860#endif
[937]861static void xc_sandbox_install(xc_sandbox_t *sandbox TSRMLS_DC) /* {{{ */
[1]862{
[555]863    zend_uint i;
[1]864    Bucket *b;
865
[95]866#ifdef HAVE_XCACHE_CONSTANT
[705]867    for (b = TG(zend_constants).pListHead; b != NULL && b != TG(internal_constant_tail); b = b->pListNext) {
868        zend_constant *c = (zend_constant*) b->pData;
869        xc_free_zend_constant(c);
870    }
871
[651]872    b = TG(internal_constant_tail) ? TG(internal_constant_tail)->pListNext : TG(zend_constants).pListHead;
[95]873    /* install constants */
874    while (b != NULL) {
875        zend_constant *c = (zend_constant*) b->pData;
876        xc_install_constant(sandbox->filename, c,
[506]877                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength, b->h TSRMLS_CC);
[95]878        b = b->pListNext;
879    }
880#endif
881
[344]882    b = TG(internal_function_tail) ? TG(internal_function_tail)->pListNext : TG(function_table).pListHead;
[1]883    /* install function */
884    while (b != NULL) {
885        zend_function *func = (zend_function*) b->pData;
886        xc_install_function(sandbox->filename, func,
[506]887                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength, b->h TSRMLS_CC);
[1]888        b = b->pListNext;
889    }
890
[344]891    b = TG(internal_class_tail) ? TG(internal_class_tail)->pListNext : TG(class_table).pListHead;
[1]892    /* install class */
893    while (b != NULL) {
[212]894        xc_install_class(sandbox->filename, (xc_cest_t*) b->pData, -1,
[506]895                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength, b->h TSRMLS_CC);
[1]896        b = b->pListNext;
897    }
[268]898
899#ifdef ZEND_ENGINE_2_1
900    /* trigger auto_globals jit */
901    for (b = TG(auto_globals).pListHead; b != NULL; b = b->pListNext) {
902        zend_auto_global *auto_global = (zend_auto_global *) b->pData;
903        /* check if actived */
904        if (auto_global->auto_global_callback && !auto_global->armed) {
905            zend_u_is_auto_global(BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), auto_global->name_len TSRMLS_CC);
906        }
907    }
908#endif
909
[548]910#ifdef ZEND_COMPILE_DELAYED_BINDING
[937]911    zend_do_delayed_early_binding(CG(active_op_array) TSRMLS_CC);
[548]912#else
[937]913    xc_undo_pass_two(CG(active_op_array) TSRMLS_CC);
914    xc_foreach_early_binding_class(CG(active_op_array), xc_early_binding_cb, (void *) sandbox TSRMLS_CC);
915    xc_redo_pass_two(CG(active_op_array) TSRMLS_CC);
[548]916#endif
[1]917
[669]918#ifdef XCACHE_ERROR_CACHING
[496]919    /* restore trigger errors */
920    for (i = 0; i < sandbox->compilererror_cnt; i ++) {
921        xc_compilererror_t *error = &sandbox->compilererrors[i];
922        CG(zend_lineno) = error->lineno;
[648]923        zend_error(error->type, "%s", error->error);
[496]924    }
925    CG(zend_lineno) = 0;
926#endif
927
[1]928    i = 1;
[961]929    /* still needed because in zend_language_scanner.l, require()/include() check file_handle.handle.stream.handle */
[1]930    zend_hash_add(&OG(included_files), sandbox->filename, strlen(sandbox->filename) + 1, (void *)&i, sizeof(int), NULL);
931}
932/* }}} */
[938]933static void xc_sandbox_free(xc_sandbox_t *sandbox, zend_op_array *op_array TSRMLS_DC) /* {{{ */
[1]934{
[496]935    XG(sandbox) = NULL;
[669]936#ifdef XCACHE_ERROR_CACHING
[496]937    EG(user_error_handler_error_reporting) = sandbox->orig_user_error_handler_error_reporting;
938#endif
939
[1]940    /* restore first first install function/class */
[95]941#ifdef HAVE_XCACHE_CONSTANT
942    EG(zend_constants) = OG(zend_constants);
943#endif
[1]944    CG(function_table) = OG(function_table);
945    CG(class_table)    = OG(class_table);
946    EG(class_table)    = CG(class_table);
[268]947#ifdef ZEND_ENGINE_2_1
948    CG(auto_globals)   = OG(auto_globals);
949#endif
[1]950
[937]951    if (op_array) {
952        zend_op_array *old_active_op_array = CG(active_op_array);
[235]953        CG(in_compilation)    = 1;
[860]954        CG(compiled_filename) = ZEND_24(NOTHING, (char *)) sandbox->filename;
[235]955        CG(zend_lineno)       = 0;
[937]956
957        CG(active_op_array) = op_array;
958        xc_sandbox_install(sandbox TSRMLS_CC);
959        CG(active_op_array) = old_active_op_array;
960
[235]961        CG(in_compilation)    = 0;
962        CG(compiled_filename) = NULL;
[1]963
964        /* no free as it's installed */
[95]965#ifdef HAVE_XCACHE_CONSTANT
966        TG(zend_constants).pDestructor = NULL;
967#endif
[1]968        TG(function_table).pDestructor = NULL;
969        TG(class_table).pDestructor = NULL;
970    }
971
972    /* destroy all the tmp */
[95]973#ifdef HAVE_XCACHE_CONSTANT
974    zend_hash_destroy(&TG(zend_constants));
975#endif
[1]976    zend_hash_destroy(&TG(function_table));
977    zend_hash_destroy(&TG(class_table));
[268]978#ifdef ZEND_ENGINE_2_1
979    zend_hash_destroy(&TG(auto_globals));
980#endif
[1]981    zend_hash_destroy(TG(included_files));
982
983    /* restore orig here, as EG/CG holded tmp before */
984    memcpy(&EG(included_files), &OG(included_files), sizeof(EG(included_files)));
985
[669]986#ifdef XCACHE_ERROR_CACHING
[496]987    if (sandbox->compilererrors) {
[555]988        zend_uint i;
[496]989        for (i = 0; i < sandbox->compilererror_cnt; i ++) {
990            efree(sandbox->compilererrors[i].error);
991        }
992        efree(sandbox->compilererrors);
993    }
[508]994#endif
[548]995
996#ifdef ZEND_COMPILE_IGNORE_INTERNAL_CLASSES
997    CG(compiler_options) = sandbox->orig_compiler_options;
998#endif
[1]999}
1000/* }}} */
[938]1001zend_op_array *xc_sandbox(xc_sandboxed_func_t sandboxed_func, void *data, ZEND_24(NOTHING, const) char *filename TSRMLS_DC) /* {{{ */
[937]1002{
[938]1003    xc_sandbox_t sandbox;
1004    zend_op_array *op_array = NULL;
1005    zend_bool catched = 0;
1006
1007    memset(&sandbox, 0, sizeof(sandbox));
1008    zend_try {
1009        xc_sandbox_init(&sandbox, filename TSRMLS_CC);
1010        op_array = sandboxed_func(data TSRMLS_CC);
1011    } zend_catch {
1012        catched = 1;
1013    } zend_end_try();
1014
1015    xc_sandbox_free(&sandbox, op_array TSRMLS_CC);
1016    if (catched) {
1017        zend_bailout();
1018    }
1019    return op_array;
1020}
1021/* {{{ */
1022const Bucket *xc_sandbox_user_function_begin(TSRMLS_D) /* {{{ */
1023{
[937]1024    xc_sandbox_t *sandbox = (xc_sandbox_t *) XG(sandbox);
1025    assert(sandbox);
1026    return TG(internal_function_tail) ? TG(internal_function_tail)->pListNext : TG(function_table).pListHead;
[938]1027}
1028/* {{{ */
1029const Bucket *xc_sandbox_user_class_begin(TSRMLS_D) /* {{{ */
[937]1030{
1031    xc_sandbox_t *sandbox = (xc_sandbox_t *) XG(sandbox);
1032    assert(sandbox);
1033    return TG(internal_class_tail) ? TG(internal_class_tail)->pListNext : TG(class_table).pListHead;
[938]1034}
1035/* {{{ */
[937]1036#ifdef XCACHE_ERROR_CACHING
[938]1037xc_compilererror_t *xc_sandbox_compilererrors(TSRMLS_D) /* {{{ */
[937]1038{
1039    xc_sandbox_t *sandbox = (xc_sandbox_t *) XG(sandbox);
1040    assert(sandbox);
1041    return sandbox->compilererrors;
[938]1042}
1043/* }}} */
1044zend_uint xc_sandbox_compilererror_cnt(TSRMLS_D) /* {{{ */
[937]1045{
1046    xc_sandbox_t *sandbox = (xc_sandbox_t *) XG(sandbox);
1047    assert(sandbox);
1048    return sandbox->compilererror_cnt;
[938]1049}
1050/* }}} */
[937]1051#endif
1052
[305]1053int xc_vtrace(const char *fmt, va_list args) /* {{{ */
1054{
[349]1055    return vfprintf(stderr, fmt, args);
[305]1056}
1057/* }}} */
1058int xc_trace(const char *fmt, ...) /* {{{ */
1059{
1060    va_list args;
[349]1061    int ret;
[305]1062
1063    va_start(args, fmt);
[349]1064    ret = xc_vtrace(fmt, args);
[305]1065    va_end(args);
[349]1066    return ret;
[305]1067}
1068/* }}} */
[671]1069
1070#ifndef ZEND_ENGINE_2_3
[847]1071#include "ext/standard/php_string.h"
1072size_t xc_dirname(char *path, size_t len) /* {{{ */
[671]1073{
[847]1074#ifdef ZEND_ENGINE_2
1075    return php_dirname(path, len);
1076#else
[671]1077    php_dirname(path, len);
1078    return strlen(path);
[847]1079#endif
[671]1080}
1081/* }}} */
1082
[847]1083long xc_atol(const char *str, int str_len) /* {{{ */
[671]1084{
1085    long retval;
1086
1087    if (!str_len) {
1088        str_len = strlen(str);
1089    }
1090
1091    retval = strtol(str, NULL, 0);
1092    if (str_len > 0) {
1093        switch (str[str_len - 1]) {
1094        case 'g':
1095        case 'G':
1096            retval *= 1024;
1097            /* break intentionally missing */
1098        case 'm':
1099        case 'M':
1100            retval *= 1024;
1101            /* break intentionally missing */
1102        case 'k':
1103        case 'K':
1104            retval *= 1024;
1105            break;
1106        }
1107    }
1108
1109    return retval;
1110}
1111/* }}} */
1112
1113#endif
[903]1114
1115/* init/destroy */
1116int xc_util_init(int module_number TSRMLS_DC) /* {{{ */
1117{
1118#ifdef XCACHE_ERROR_CACHING
1119    old_zend_error_cb = zend_error_cb;
1120    zend_error_cb = xc_sandbox_error_cb;
1121#endif
1122
1123    return SUCCESS;
1124}
1125/* }}} */
1126void xc_util_destroy() /* {{{ */
1127{
1128#ifdef XCACHE_ERROR_CACHING
1129    if (zend_error_cb == xc_sandbox_error_cb) {
1130        zend_error_cb = old_zend_error_cb;
1131    }
1132#endif
1133}
1134/* }}} */
1135
Note: See TracBrowser for help on using the repository browser.