source: trunk/utils.c @ 716

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

PHP 6: initial catch up with php 6 literals and trait

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