source: trunk/main/xc_utils.c @ 983

Last change on this file since 983 was 983, checked in by moo, 21 months ago

kill warnings

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