source: trunk/utils.c @ 506

Last change on this file since 506 was 506, checked in by moo, 6 years ago

optimized function_table/class_table by caching hash value

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