source: branches/1.2/utils.c @ 394

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

merged [393] from trunk: svn:eol-style

  • Property svn:eol-style set to native
File size: 19.5 KB
Line 
1
2#include "xcache.h"
3#include "stack.h"
4#include "xcache_globals.h"
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
13#ifndef max
14#define max(a, b) ((a) < (b) ? (b) : (a))
15#endif
16
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 { \
22    (op).u.constant.is_ref = 0; \
23    zval_dtor(&(op).u.constant); \
24} while(0)
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{
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);
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) {
146            case ZEND_JMP:
147                opline->op1.u.opline_num = opline->op1.u.jmp_addr - op_array->opcodes;
148                assert(opline->op1.u.opline_num < op_array->last);
149                break;
150            case ZEND_JMPZ:
151            case ZEND_JMPNZ:
152            case ZEND_JMPZ_EX:
153            case ZEND_JMPNZ_EX:
154                opline->op2.u.opline_num = opline->op2.u.jmp_addr - op_array->opcodes;
155                assert(opline->op2.u.opline_num < op_array->last);
156                break;
157        }
158#endif
159        opline++;
160    }
161    op_array->done_pass_two = 0;
162
163    return 0;
164}
165/* }}} */
166int xc_redo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
167{
168    zend_op *opline, *end;
169
170    if (op_array->done_pass_two) {
171        return 0;
172    }
173
174    /*
175    op_array->opcodes = (zend_op *) erealloc(op_array->opcodes, sizeof(zend_op)*op_array->last);
176    op_array->size = op_array->last;
177    */
178
179    opline = op_array->opcodes;
180    end = opline + op_array->last;
181    while (opline < end) {
182        if (opline->op1.op_type == IS_CONST) {
183            opline->op1.u.constant.is_ref = 1;
184            opline->op1.u.constant.refcount = 2; /* Make sure is_ref won't be reset */
185        }
186        if (opline->op2.op_type == IS_CONST) {
187            opline->op2.u.constant.is_ref = 1;
188            opline->op2.u.constant.refcount = 2;
189        }
190#ifdef ZEND_ENGINE_2_1
191        switch (opline->opcode) {
192            case ZEND_JMP:
193                assert(opline->op1.u.opline_num < op_array->last);
194                opline->op1.u.jmp_addr = op_array->opcodes + opline->op1.u.opline_num;
195                break;
196            case ZEND_JMPZ:
197            case ZEND_JMPNZ:
198            case ZEND_JMPZ_EX:
199            case ZEND_JMPNZ_EX:
200                assert(opline->op2.u.opline_num < op_array->last);
201                opline->op2.u.jmp_addr = op_array->opcodes + opline->op2.u.opline_num;
202                break;
203        }
204        ZEND_VM_SET_OPCODE_HANDLER(opline);
205#endif
206        opline++;
207    }
208
209    op_array->done_pass_two = 1;
210    return 0;
211}
212/* }}} */
213
214#ifdef HAVE_XCACHE_OPCODE_SPEC_DEF
215static void xc_fix_opcode_ex_znode(int tofix, xc_op_spec_t spec, znode *znode, int type TSRMLS_DC) /* {{{ */
216{
217#ifdef ZEND_ENGINE_2
218    if ((znode->op_type != IS_UNUSED && (spec == OPSPEC_UCLASS || spec == OPSPEC_CLASS)) ||
219            spec == OPSPEC_FETCH) {
220        if (tofix) {
221            switch (znode->op_type) {
222            case IS_VAR:
223            case IS_TMP_VAR:
224                break;
225
226            default:
227                /* TODO: data lost, find a way to keep it */
228                /* assert(znode->op_type == IS_CONST); */
229                znode->op_type = IS_TMP_VAR;
230            }
231        }
232    }
233    switch (znode->op_type) {
234    case IS_TMP_VAR:
235    case IS_VAR:
236        if (tofix) {
237            znode->u.var /= sizeof(temp_variable);
238        }
239        else {
240            znode->u.var *= sizeof(temp_variable);
241        }
242    }
243#endif
244}
245/* }}} */
246
247static void xc_fix_opcode_ex(zend_op_array *op_array, int tofix TSRMLS_DC) /* {{{ */
248{
249    zend_op *opline;
250    zend_uint i;
251
252    opline = op_array->opcodes;
253    for (i = 0; i < op_array->last; i ++, opline ++) {
254        /* 3rd optimizer may have ... */
255        if (opline->opcode < xc_get_opcode_spec_count()) {
256            const xc_opcode_spec_t *spec;
257            spec = xc_get_opcode_spec(opline->opcode);
258
259            xc_fix_opcode_ex_znode(tofix, spec->op1, &opline->op1, 0 TSRMLS_CC);
260            xc_fix_opcode_ex_znode(tofix, spec->op2, &opline->op2, 1 TSRMLS_CC);
261            xc_fix_opcode_ex_znode(tofix, spec->res, &opline->result, 2 TSRMLS_CC);
262        }
263    }
264}
265/* }}} */
266int xc_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
267{
268    xc_fix_opcode_ex(op_array, 1 TSRMLS_CC);
269    return 0;
270}
271/* }}} */
272int xc_undo_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
273{
274    xc_fix_opcode_ex(op_array, 0 TSRMLS_CC);
275
276    return 0;
277}
278/* }}} */
279#endif
280
281int 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) /* {{{ */
282{
283    zend_op *opline, *begin, *end, *next = NULL;
284
285    opline = begin = op_array->opcodes;
286    end = opline + op_array->last;
287    while (opline < end) {
288        switch (opline->opcode) {
289            case ZEND_JMP:
290                next = begin + opline->op1.u.opline_num;
291                break;
292
293            case ZEND_JMPZNZ:
294                next = begin + max(opline->op2.u.opline_num, opline->extended_value);
295                break;
296
297            case ZEND_JMPZ:
298            case ZEND_JMPNZ:
299            case ZEND_JMPZ_EX:
300            case ZEND_JMPNZ_EX:
301                next = begin + opline->op2.u.opline_num;
302                break;
303
304            case ZEND_RETURN:
305                opline = end;
306                break;
307
308#ifdef ZEND_ENGINE_2
309            case ZEND_DECLARE_INHERITED_CLASS:
310                callback(opline, opline - begin, data TSRMLS_CC);
311                break;
312#else
313            case ZEND_DECLARE_FUNCTION_OR_CLASS:
314                if (opline->extended_value == ZEND_DECLARE_INHERITED_CLASS) {
315                    callback(opline, opline - begin, data TSRMLS_CC);
316                }
317                break;
318#endif
319        }
320
321        if (opline < next) {
322            opline = next;
323        }
324        else {
325            opline ++;
326        }
327    }
328    return SUCCESS;
329}
330/* }}} */
331static int xc_do_early_binding(zend_op_array *op_array, HashTable *class_table, int oplineno TSRMLS_DC) /* {{{ */
332{
333    zend_op *opline;
334
335#ifdef DEBUG
336    fprintf(stderr, "binding %d\n", oplineno);
337#endif
338    assert(oplineno >= 0);
339
340    /* do early binding */
341    opline = &(op_array->opcodes[oplineno]);
342
343    switch (opline->opcode) {
344#ifdef ZEND_ENGINE_2
345    case ZEND_DECLARE_INHERITED_CLASS:
346        {
347            zval *parent_name;
348            zend_class_entry **pce;
349
350            /* don't early-bind classes that implement interfaces */
351            if ((opline + 1)->opcode == ZEND_FETCH_CLASS && (opline + 2)->opcode == ZEND_ADD_INTERFACE) {
352                return FAILURE;
353            }
354
355            parent_name = &(opline - 1)->op2.u.constant;
356#   ifdef DEBUG
357            fprintf(stderr, "binding with parent %s\n", Z_STRVAL_P(parent_name));
358#   endif
359            if (zend_lookup_class(Z_STRVAL_P(parent_name), Z_STRLEN_P(parent_name), &pce TSRMLS_CC) == FAILURE) {
360                return FAILURE;
361            }
362
363            if (do_bind_inherited_class(opline, class_table, *pce, 1 TSRMLS_CC) == NULL) {
364                return FAILURE;
365            }
366        }
367
368        /* clear unnecessary ZEND_FETCH_CLASS opcode */
369        if (opline > op_array->opcodes
370         && (opline - 1)->opcode == ZEND_FETCH_CLASS) {
371            zend_op *fetch_class_opline = opline - 1;
372
373#   ifdef DEBUG
374            fprintf(stderr, "%s %p\n", Z_STRVAL(fetch_class_opline->op2.u.constant), Z_STRVAL(fetch_class_opline->op2.u.constant));
375#   endif
376            OP_ZVAL_DTOR(fetch_class_opline->op2);
377            fetch_class_opline->opcode = ZEND_NOP;
378            ZEND_VM_SET_OPCODE_HANDLER(fetch_class_opline);
379            memset(&fetch_class_opline->op1, 0, sizeof(znode));
380            memset(&fetch_class_opline->op2, 0, sizeof(znode));
381            SET_UNUSED(fetch_class_opline->op1);
382            SET_UNUSED(fetch_class_opline->op2);
383            SET_UNUSED(fetch_class_opline->result);
384        }
385
386        /* clear unnecessary ZEND_VERIFY_ABSTRACT_CLASS opcode */
387        if ((opline + 1)->opcode == ZEND_VERIFY_ABSTRACT_CLASS) {
388            zend_op *abstract_op = opline + 1;
389            memset(abstract_op, 0, sizeof(abstract_op[0]));
390            abstract_op->lineno = 0;
391            SET_UNUSED(abstract_op->op1);
392            SET_UNUSED(abstract_op->op2);
393            SET_UNUSED(abstract_op->result);
394            abstract_op->opcode = ZEND_NOP;
395            ZEND_VM_SET_OPCODE_HANDLER(abstract_op);
396        }
397#else
398    case ZEND_DECLARE_FUNCTION_OR_CLASS:
399        if (do_bind_function_or_class(opline, NULL, class_table, 1) == FAILURE) {
400            return FAILURE;
401        }
402#endif
403        break;
404
405    default:
406        return FAILURE;
407    }
408
409    zend_hash_del(class_table, opline->op1.u.constant.value.str.val, opline->op1.u.constant.value.str.len);
410    OP_ZVAL_DTOR(opline->op1);
411    OP_ZVAL_DTOR(opline->op2);
412    opline->opcode = ZEND_NOP;
413    ZEND_VM_SET_OPCODE_HANDLER(opline);
414    memset(&opline->op1, 0, sizeof(znode));
415    memset(&opline->op2, 0, sizeof(znode));
416    SET_UNUSED(opline->op1);
417    SET_UNUSED(opline->op2);
418    return SUCCESS;
419}
420/* }}} */
421
422#ifdef HAVE_XCACHE_CONSTANT
423void xc_install_constant(char *filename, zend_constant *constant, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
424{
425    if (zend_u_hash_add(EG(zend_constants), type, key, len,
426                constant, sizeof(zend_constant),
427                NULL
428                ) == FAILURE) {
429        CG(zend_lineno) = 0;
430#ifdef IS_UNICODE
431        zend_error(E_NOTICE, "Constant %R already defined", type, key);
432#else
433        zend_error(E_NOTICE, "Constant %s already defined", key);
434#endif
435        free(ZSTR_V(constant->name));
436        if (!(constant->flags & CONST_PERSISTENT)) {
437            zval_dtor(&constant->value);
438        }
439    }
440}
441/* }}} */
442#endif
443void xc_install_function(char *filename, zend_function *func, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
444{
445    zend_bool istmpkey;
446
447    if (func->type == ZEND_USER_FUNCTION) {
448#ifdef IS_UNICODE
449        istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
450#else
451        istmpkey = ZSTR_S(key)[0] == 0;
452#endif
453        if (istmpkey) {
454            zend_u_hash_update(CG(function_table), type, key, len,
455                        func, sizeof(zend_op_array),
456                        NULL
457                        );
458        }
459        else if (zend_u_hash_add(CG(function_table), type, key, len,
460                    func, sizeof(zend_op_array),
461                    NULL
462                    ) == FAILURE) {
463            CG(zend_lineno) = ZESW(func->op_array.opcodes[0].lineno, func->op_array.line_start);
464#ifdef IS_UNICODE
465            zend_error(E_ERROR, "Cannot redeclare %R()", type, key);
466#else
467            zend_error(E_ERROR, "Cannot redeclare %s()", key);
468#endif
469        }
470    }
471}
472/* }}} */
473ZESW(xc_cest_t *, void) xc_install_class(char *filename, xc_cest_t *cest, int oplineno, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
474{
475    zend_bool istmpkey;
476    zend_class_entry *cep = CestToCePtr(*cest);
477    ZESW(void *stored_ce_ptr, NOTHING);
478
479#ifdef IS_UNICODE
480    istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
481#else
482    istmpkey = ZSTR_S(key)[0] == 0;
483#endif
484    if (istmpkey) {
485        zend_u_hash_update(CG(class_table), type, key, len,
486                    cest, sizeof(xc_cest_t),
487                    ZESW(&stored_ce_ptr, NULL)
488                    );
489        if (oplineno != -1) {
490            xc_do_early_binding(CG(active_op_array), CG(class_table), oplineno TSRMLS_CC);
491        }
492    }
493    else if (zend_u_hash_add(CG(class_table), type, key, len,
494                cest, sizeof(xc_cest_t),
495                ZESW(&stored_ce_ptr, NULL)
496                ) == FAILURE) {
497        CG(zend_lineno) = ZESW(0, cep->line_start);
498#ifdef IS_UNICODE
499        zend_error(E_ERROR, "Cannot redeclare class %R", type, cep->name);
500#else
501        zend_error(E_ERROR, "Cannot redeclare class %s", cep->name);
502#endif
503        assert(oplineno == -1);
504    }
505    ZESW(return (xc_cest_t *) stored_ce_ptr, NOTHING);
506}
507/* }}} */
508
509/* sandbox {{{ */
510#undef TG
511#undef OG
512#define TG(x) (sandbox->tmp_##x)
513#define OG(x) (sandbox->orig_##x)
514/* }}} */
515#ifdef ZEND_ENGINE_2_1
516static zend_bool xc_auto_global_callback(char *name, uint name_len TSRMLS_DC) /* {{{ */
517{
518    zend_auto_global *auto_global;
519    if (zend_u_hash_find(CG(auto_globals), UG(unicode) ? IS_UNICODE : IS_STRING, ZSTR(name), name_len + 1, (void **) &auto_global) == FAILURE) {
520        return 1;
521    }
522    return 0;
523}
524/* }}} */
525static int xc_auto_global_arm(zend_auto_global *auto_global TSRMLS_DC) /* {{{ */
526{
527    if (auto_global->auto_global_callback) {
528        auto_global->armed = 1;
529        auto_global->auto_global_callback = xc_auto_global_callback;
530    }
531    else {
532        auto_global->armed = 0;
533    }
534    return ZEND_HASH_APPLY_KEEP;
535}
536/* }}} */
537#endif
538
539void xc_zend_class_add_ref(zend_class_entry ZESW(*ce, **ce))
540{
541#ifdef ZEND_ENGINE_2
542    (*ce)->refcount++;
543#else
544    (*ce->refcount)++;
545#endif
546}
547
548xc_sandbox_t *xc_sandbox_init(xc_sandbox_t *sandbox, char *filename TSRMLS_DC) /* {{{ */
549{
550    HashTable *h;
551
552    if (sandbox) {
553        memset(sandbox, 0, sizeof(sandbox[0]));
554    }
555    else {
556        ECALLOC_ONE(sandbox);
557        sandbox->alloc = 1;
558    }
559
560    memcpy(&OG(included_files), &EG(included_files), sizeof(EG(included_files)));
561
562#ifdef HAVE_XCACHE_CONSTANT
563    OG(zend_constants) = EG(zend_constants);
564    EG(zend_constants) = &TG(zend_constants);
565#endif
566
567    OG(function_table) = CG(function_table);
568    CG(function_table) = &TG(function_table);
569
570    OG(class_table) = CG(class_table);
571    CG(class_table) = &TG(class_table);
572    EG(class_table) = CG(class_table);
573
574#ifdef ZEND_ENGINE_2_1
575    OG(auto_globals) = CG(auto_globals);
576    CG(auto_globals) = &TG(auto_globals);
577#endif
578
579    TG(included_files) = &EG(included_files);
580
581    zend_hash_init_ex(TG(included_files), 5, NULL, NULL, 0, 1);
582#ifdef HAVE_XCACHE_CONSTANT
583    h = OG(zend_constants);
584    zend_hash_init_ex(&TG(zend_constants),  20, NULL, h->pDestructor, h->persistent, h->bApplyProtection);
585#endif
586    h = OG(function_table);
587    zend_hash_init_ex(&TG(function_table), 128, NULL, h->pDestructor, h->persistent, h->bApplyProtection);
588    {
589        zend_function tmp_func;
590        zend_hash_copy(&TG(function_table), &XG(internal_function_table), (copy_ctor_func_t) function_add_ref, (void *) &tmp_func, sizeof(tmp_func));
591    }
592    TG(internal_class_tail) = TG(function_table).pListTail;
593
594    h = OG(class_table);
595    zend_hash_init_ex(&TG(class_table),     16, NULL, h->pDestructor, h->persistent, h->bApplyProtection);
596#if 0 && TODO
597    {
598        xc_cest_t tmp_cest;
599        zend_hash_copy(&TG(class_table), &XG(internal_class_table), (copy_ctor_func_t) xc_zend_class_add_ref, (void *) &tmp_cest, sizeof(tmp_cest));
600    }
601#endif
602    TG(internal_class_tail) = TG(class_table).pListTail;
603
604#ifdef ZEND_ENGINE_2_1
605    /* shallow copy, don't destruct */
606    h = OG(auto_globals);
607    zend_hash_init_ex(&TG(auto_globals),     8, NULL,           NULL, h->persistent, h->bApplyProtection);
608    {
609        zend_auto_global tmp_autoglobal;
610
611        zend_hash_copy(&TG(auto_globals), OG(auto_globals), NULL, (void *) &tmp_autoglobal, sizeof(tmp_autoglobal));
612        zend_hash_apply(&TG(auto_globals), (apply_func_t) xc_auto_global_arm TSRMLS_CC);
613    }
614#endif
615
616    sandbox->filename = filename;
617
618#ifdef E_STRICT
619    sandbox->orig_user_error_handler_error_reporting = EG(user_error_handler_error_reporting);
620    EG(user_error_handler_error_reporting) &= ~E_STRICT;
621#endif
622
623    return sandbox;
624}
625/* }}} */
626static void xc_early_binding_cb(zend_op *opline, int oplineno, void *data TSRMLS_DC) /* {{{ */
627{
628    xc_sandbox_t *sandbox = (xc_sandbox_t *) data;
629    xc_do_early_binding(CG(active_op_array), OG(class_table), oplineno TSRMLS_CC);
630}
631/* }}} */
632static void xc_sandbox_install(xc_sandbox_t *sandbox TSRMLS_DC) /* {{{ */
633{
634    int i;
635    Bucket *b;
636
637#ifdef HAVE_XCACHE_CONSTANT
638    b = TG(zend_constants).pListHead;
639    /* install constants */
640    while (b != NULL) {
641        zend_constant *c = (zend_constant*) b->pData;
642        xc_install_constant(sandbox->filename, c,
643                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength TSRMLS_CC);
644        b = b->pListNext;
645    }
646#endif
647
648    b = TG(internal_function_tail) ? TG(internal_function_tail)->pListNext : TG(function_table).pListHead;
649    /* install function */
650    while (b != NULL) {
651        zend_function *func = (zend_function*) b->pData;
652        xc_install_function(sandbox->filename, func,
653                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength TSRMLS_CC);
654        b = b->pListNext;
655    }
656
657    b = TG(internal_class_tail) ? TG(internal_class_tail)->pListNext : TG(class_table).pListHead;
658    /* install class */
659    while (b != NULL) {
660        xc_install_class(sandbox->filename, (xc_cest_t*) b->pData, -1,
661                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), b->nKeyLength TSRMLS_CC);
662        b = b->pListNext;
663    }
664
665#ifdef ZEND_ENGINE_2_1
666    /* trigger auto_globals jit */
667    for (b = TG(auto_globals).pListHead; b != NULL; b = b->pListNext) {
668        zend_auto_global *auto_global = (zend_auto_global *) b->pData;
669        /* check if actived */
670        if (auto_global->auto_global_callback && !auto_global->armed) {
671            zend_u_is_auto_global(BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY_S(b)), auto_global->name_len TSRMLS_CC);
672        }
673    }
674#endif
675
676    xc_undo_pass_two(CG(active_op_array) TSRMLS_CC);
677    xc_foreach_early_binding_class(CG(active_op_array), xc_early_binding_cb, (void *) sandbox TSRMLS_CC);
678    xc_redo_pass_two(CG(active_op_array) TSRMLS_CC);
679
680    i = 1;
681    zend_hash_add(&OG(included_files), sandbox->filename, strlen(sandbox->filename) + 1, (void *)&i, sizeof(int), NULL);
682}
683/* }}} */
684void xc_sandbox_free(xc_sandbox_t *sandbox, int install TSRMLS_DC) /* {{{ */
685{
686    /* restore first first install function/class */
687#ifdef HAVE_XCACHE_CONSTANT
688    EG(zend_constants) = OG(zend_constants);
689#endif
690    CG(function_table) = OG(function_table);
691    CG(class_table)    = OG(class_table);
692    EG(class_table)    = CG(class_table);
693#ifdef ZEND_ENGINE_2_1
694    CG(auto_globals)   = OG(auto_globals);
695#endif
696
697    if (install) {
698        CG(in_compilation)    = 1;
699        CG(compiled_filename) = sandbox->filename;
700        CG(zend_lineno)       = 0;
701        xc_sandbox_install(sandbox TSRMLS_CC);
702        CG(in_compilation)    = 0;
703        CG(compiled_filename) = NULL;
704
705        /* no free as it's installed */
706#ifdef HAVE_XCACHE_CONSTANT
707        TG(zend_constants).pDestructor = NULL;
708#endif
709        TG(function_table).pDestructor = NULL;
710        TG(class_table).pDestructor = NULL;
711    }
712
713    /* destroy all the tmp */
714#ifdef HAVE_XCACHE_CONSTANT
715    zend_hash_destroy(&TG(zend_constants));
716#endif
717    zend_hash_destroy(&TG(function_table));
718    zend_hash_destroy(&TG(class_table));
719#ifdef ZEND_ENGINE_2_1
720    zend_hash_destroy(&TG(auto_globals));
721#endif
722    zend_hash_destroy(TG(included_files));
723
724    /* restore orig here, as EG/CG holded tmp before */
725    memcpy(&EG(included_files), &OG(included_files), sizeof(EG(included_files)));
726
727#ifdef E_STRICT
728    EG(user_error_handler_error_reporting) = sandbox->orig_user_error_handler_error_reporting;
729#endif
730
731    if (sandbox->alloc) {
732        efree(sandbox);
733    }
734}
735/* }}} */
Note: See TracBrowser for help on using the repository browser.