source: trunk/utils.c @ 705

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

fix builtin constant memory leak on compile

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