source: trunk/utils.c @ 749

Last change on this file since 749 was 749, checked in by moo, 3 years ago

decompiler: massive improvements for php 5 and 5.3

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