source: trunk/utils.c @ 103

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

PHP_6: builds again with php6. update UChar/void ptr to zstr.

File size: 13.0 KB
Line 
1#include "php.h"
2#include "xcache.h"
3#include "utils.h"
4#ifdef ZEND_ENGINE_2_1
5#include "zend_vm.h"
6#endif
7#include "opcode_spec.h"
8#undef NDEBUG
9#include "assert.h"
10
11xc_compile_result_t *xc_compile_result_init(xc_compile_result_t *cr, /* {{{ */
12        zend_op_array *op_array,
13        HashTable *function_table,
14        HashTable *class_table)
15{
16    if (cr) {
17        cr->alloc = 0;
18    }
19    else {
20        cr = emalloc(sizeof(xc_compile_result_t));
21        cr->alloc = 1;
22    }
23    cr->op_array       = op_array;
24    cr->function_table = function_table;
25    cr->class_table    = class_table;
26    return cr;
27}
28/* }}} */
29xc_compile_result_t *xc_compile_result_init_cur(xc_compile_result_t *cr, zend_op_array *op_array TSRMLS_DC) /* {{{ */
30{
31    return xc_compile_result_init(cr, op_array, CG(function_table), CG(class_table));
32}
33/* }}} */
34void xc_compile_result_free(xc_compile_result_t *cr) /* {{{ */
35{
36    if (cr->alloc) {
37        efree(cr);
38    }
39}
40/* }}} */
41
42int xc_apply_function(zend_function *zf, apply_func_t applyer TSRMLS_DC) /* {{{ */
43{
44    switch (zf->type) {
45    case ZEND_USER_FUNCTION:
46    case ZEND_EVAL_CODE:
47        return applyer(&zf->op_array TSRMLS_CC);
48        break;
49
50    case ZEND_INTERNAL_FUNCTION:
51    case ZEND_OVERLOADED_FUNCTION:
52        break;
53
54    EMPTY_SWITCH_DEFAULT_CASE();
55    }
56    return 0;
57}
58/* }}} */
59typedef struct {
60    apply_func_t applyer;
61    zend_class_entry *ce;
62} xc_apply_method_info;
63int xc_apply_method(zend_function *zf, xc_apply_method_info *mi TSRMLS_DC) /* {{{ */
64{
65    /* avoid duplicate apply for shadowed method */
66#ifdef ZEND_ENGINE_2
67    if (mi->ce != zf->common.scope) {
68        /* fprintf(stderr, "avoided duplicate %s\n", zf->common.function_name); */
69        return 0;
70    }
71#else
72    char *name = zf->common.function_name;
73    int name_s = strlen(name) + 1;
74    zend_class_entry *ce;
75    zend_function *ptr;
76
77    for (ce = mi->ce->parent; ce; ce = ce->parent) {
78        if (zend_hash_find(&ce->function_table, name, name_s, (void **) &ptr) == SUCCESS) {
79            if (ptr->op_array.refcount == zf->op_array.refcount) {
80                return 0;
81            }
82        }
83    }
84#endif
85    return xc_apply_function(zf, mi->applyer TSRMLS_CC);
86}
87/* }}} */
88#if 0
89int xc_apply_class(zend_class_entry *ce, apply_func_t applyer TSRMLS_DC) /* {{{ */
90{
91    xc_apply_method_info mi;
92
93    mi.applyer = applyer;
94    mi.ce      = ce;
95    zend_hash_apply_with_argument(&(ce->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
96    return 0;
97}
98/* }}} */
99#endif
100static int xc_apply_cest(xc_cest_t *cest, apply_func_t applyer TSRMLS_DC) /* {{{ */
101{
102    xc_apply_method_info mi;
103
104    mi.applyer = applyer;
105    mi.ce      = CestToCePtr(*cest);
106    zend_hash_apply_with_argument(&(CestToCePtr(*cest)->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
107    return 0;
108}
109/* }}} */
110int xc_apply_op_array(xc_compile_result_t *cr, apply_func_t applyer TSRMLS_DC) /* {{{ */
111{
112    zend_hash_apply_with_argument(cr->function_table, (apply_func_arg_t) xc_apply_function, applyer TSRMLS_CC);
113    zend_hash_apply_with_argument(cr->class_table, (apply_func_arg_t) xc_apply_cest, applyer TSRMLS_CC);
114
115    return applyer(cr->op_array TSRMLS_CC);
116}
117/* }}} */
118
119int xc_undo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
120{
121    zend_op *opline, *end;
122
123    if (!op_array->done_pass_two) {
124        return 0;
125    }
126
127    opline = op_array->opcodes;
128    end = opline + op_array->last;
129    while (opline < end) {
130#ifdef ZEND_ENGINE_2_1
131        switch (opline->opcode) {
132            case ZEND_JMP:
133                opline->op1.u.opline_num = opline->op1.u.jmp_addr - op_array->opcodes;
134                assert(opline->op1.u.opline_num < op_array->last);
135                break;
136            case ZEND_JMPZ:
137            case ZEND_JMPNZ:
138            case ZEND_JMPZ_EX:
139            case ZEND_JMPNZ_EX:
140                opline->op2.u.opline_num = opline->op2.u.jmp_addr - op_array->opcodes;
141                assert(opline->op2.u.opline_num < op_array->last);
142                break;
143        }
144#endif
145        opline++;
146    }
147    op_array->done_pass_two = 0;
148
149    return 0;
150}
151/* }}} */
152int xc_redo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
153{
154    zend_op *opline, *end;
155
156    if (op_array->done_pass_two) {
157        return 0;
158    }
159
160    /*
161    op_array->opcodes = (zend_op *) erealloc(op_array->opcodes, sizeof(zend_op)*op_array->last);
162    op_array->size = op_array->last;
163    */
164
165    opline = op_array->opcodes;
166    end = opline + op_array->last;
167    while (opline < end) {
168        if (opline->op1.op_type == IS_CONST) {
169            opline->op1.u.constant.is_ref = 1;
170            opline->op1.u.constant.refcount = 2; /* Make sure is_ref won't be reset */
171        }
172        if (opline->op2.op_type == IS_CONST) {
173            opline->op2.u.constant.is_ref = 1;
174            opline->op2.u.constant.refcount = 2;
175        }
176#ifdef ZEND_ENGINE_2_1
177        switch (opline->opcode) {
178            case ZEND_JMP:
179                assert(opline->op1.u.opline_num < op_array->last);
180                opline->op1.u.jmp_addr = op_array->opcodes + opline->op1.u.opline_num;
181                break;
182            case ZEND_JMPZ:
183            case ZEND_JMPNZ:
184            case ZEND_JMPZ_EX:
185            case ZEND_JMPNZ_EX:
186                assert(opline->op2.u.opline_num < op_array->last);
187                opline->op2.u.jmp_addr = op_array->opcodes + opline->op2.u.opline_num;
188                break;
189        }
190        ZEND_VM_SET_OPCODE_HANDLER(opline);
191#endif
192        opline++;
193    }
194
195    op_array->done_pass_two = 1;
196    return 0;
197}
198/* }}} */
199
200#ifdef HAVE_XCACHE_OPCODE_SPEC_DEF
201static void xc_fix_opcode_ex_znode(int tofix, xc_op_spec_t spec, znode *znode, int type TSRMLS_DC) /* {{{ */
202{
203#ifdef ZEND_ENGINE_2
204    if ((znode->op_type != IS_UNUSED && (spec == OPSPEC_UCLASS || spec == OPSPEC_CLASS)) ||
205            spec == OPSPEC_FETCH) {
206        if (tofix) {
207            switch (znode->op_type) {
208            case IS_VAR:
209            case IS_TMP_VAR:
210                break;
211
212            default:
213                /* TODO: data lost, find a way to keep it */
214                /* assert(znode->op_type == IS_CONST); */
215                znode->op_type = IS_TMP_VAR;
216            }
217        }
218    }
219    switch (znode->op_type) {
220    case IS_TMP_VAR:
221    case IS_VAR:
222        if (tofix) {
223            znode->u.var /= sizeof(temp_variable);
224        }
225        else {
226            znode->u.var *= sizeof(temp_variable);
227        }
228    }
229#endif
230}
231/* }}} */
232
233static void xc_fix_opcode_ex(zend_op_array *op_array, int tofix TSRMLS_DC) /* {{{ */
234{
235    zend_op *opline;
236    zend_uint i;
237
238    opline = op_array->opcodes;
239    for (i = 0; i < op_array->last; i ++, opline ++) {
240        /* 3rd optimizer may have ... */
241        if (opline->opcode < xc_get_opcode_spec_count()) {
242            const xc_opcode_spec_t *spec;
243            spec = xc_get_opcode_spec(opline->opcode);
244
245            xc_fix_opcode_ex_znode(tofix, spec->op1, &opline->op1, 0 TSRMLS_CC);
246            xc_fix_opcode_ex_znode(tofix, spec->op2, &opline->op2, 1 TSRMLS_CC);
247            xc_fix_opcode_ex_znode(tofix, spec->res, &opline->result, 2 TSRMLS_CC);
248        }
249    }
250}
251/* }}} */
252int xc_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
253{
254    xc_fix_opcode_ex(op_array, 1 TSRMLS_CC);
255    return 0;
256}
257/* }}} */
258int xc_undo_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
259{
260    xc_fix_opcode_ex(op_array, 0 TSRMLS_CC);
261
262    return 0;
263}
264/* }}} */
265#endif
266
267#ifdef HAVE_XCACHE_CONSTANT
268void xc_install_constant(char *filename, zend_constant *constant, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
269{
270    if (zend_u_hash_add(EG(zend_constants), type, key, len,
271                constant, sizeof(zend_constant),
272                NULL
273                ) == FAILURE) {
274        CG(in_compilation) = 1;
275        CG(compiled_filename) = filename;
276        CG(zend_lineno) = 0;
277#ifdef IS_UNICODE
278        zend_error(E_NOTICE, "Constant %R already defined", type, key);
279#else
280        zend_error(E_NOTICE, "Constant %s already defined", key);
281#endif
282        free(ZSTR_V(constant->name));
283        if (!(constant->flags & CONST_PERSISTENT)) {
284            zval_dtor(&constant->value);
285        }
286    }
287}
288/* }}} */
289#endif
290void xc_install_function(char *filename, zend_function *func, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
291{
292    zend_bool istmpkey;
293
294    if (func->type == ZEND_USER_FUNCTION) {
295#ifdef IS_UNICODE
296        istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
297#else
298        istmpkey = ZSTR_S(key)[0] == 0;
299#endif
300        if (istmpkey) {
301            zend_u_hash_update(CG(function_table), type, key, len,
302                        func, sizeof(zend_op_array),
303                        NULL
304                        );
305        }
306        else if (zend_u_hash_add(CG(function_table), type, key, len,
307                    func, sizeof(zend_op_array),
308                    NULL
309                    ) == FAILURE) {
310            CG(in_compilation) = 1;
311            CG(compiled_filename) = filename;
312            CG(zend_lineno) = ZESW(func->op_array.opcodes[0].lineno, func->op_array.line_start);
313#ifdef IS_UNICODE
314            zend_error(E_ERROR, "Cannot redeclare %R()", type, key);
315#else
316            zend_error(E_ERROR, "Cannot redeclare %s()", key);
317#endif
318        }
319    }
320}
321/* }}} */
322ZESW(xc_cest_t *, void) xc_install_class(char *filename, xc_cest_t *cest, zend_uchar type, zstr key, uint len TSRMLS_DC) /* {{{ */
323{
324    zend_bool istmpkey;
325    zend_class_entry *cep = CestToCePtr(*cest);
326    ZESW(void *stored_ce_ptr, NOTHING);
327
328#ifdef IS_UNICODE
329    istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
330#else
331    istmpkey = ZSTR_S(key)[0] == 0;
332#endif
333    if (istmpkey) {
334        zend_u_hash_update(CG(class_table), type, key, len,
335                    cest, sizeof(xc_cest_t),
336                    ZESW(&stored_ce_ptr, NULL)
337                    );
338    }
339    else if (zend_u_hash_add(CG(class_table), type, key, len,
340                cest, sizeof(xc_cest_t),
341                ZESW(&stored_ce_ptr, NULL)
342                ) == FAILURE) {
343        CG(in_compilation) = 1;
344        CG(compiled_filename) = filename;
345        CG(zend_lineno) = ZESW(0, cep->line_start);
346#ifdef IS_UNICODE
347        zend_error(E_ERROR, "Cannot redeclare class %R", type, cep->name);
348#else
349        zend_error(E_ERROR, "Cannot redeclare class %s", cep->name);
350#endif
351    }
352    ZESW(return (xc_cest_t *) stored_ce_ptr, NOTHING);
353}
354/* }}} */
355
356/* sandbox {{{ */
357#undef TG
358#undef OG
359#define TG(x) (sandbox->tmp_##x)
360#define OG(x) (sandbox->orig_##x)
361/* }}} */
362#ifdef HAVE_XCACHE_CONSTANT
363static void xc_constant_copy_ctor(zend_constant *c) /* {{{ */
364{
365#ifdef IS_UNICODE
366    if (UG(unicode)) {
367        ZSTR_U(c->name) = zend_ustrndup(ZSTR_U(c->name), c->name_len - 1);
368    }
369    else
370#endif
371    {
372        ZSTR_S(c->name) = zend_strndup(ZSTR_S(c->name), c->name_len - 1);
373    }
374    if (!(c->flags & CONST_PERSISTENT)) {
375        zval_copy_ctor(&c->value);
376    }
377}
378/* }}} */
379#endif
380xc_sandbox_t *xc_sandbox_init(xc_sandbox_t *sandbox, char *filename TSRMLS_DC) /* {{{ */
381{
382    if (sandbox) {
383        memset(sandbox, 0, sizeof(sandbox[0]));
384    }
385    else {
386        ECALLOC_ONE(sandbox);
387        sandbox->alloc = 1;
388    }
389
390    memcpy(&OG(included_files), &EG(included_files), sizeof(EG(included_files)));
391    memcpy(&OG(open_files), &CG(open_files), sizeof(CG(open_files)));
392
393#ifdef HAVE_XCACHE_CONSTANT
394    OG(zend_constants) = EG(zend_constants);
395    EG(zend_constants) = &TG(zend_constants);
396#endif
397
398    OG(function_table) = CG(function_table);
399    CG(function_table) = &TG(function_table);
400
401    assert(EG(class_table) == CG(class_table));
402
403    OG(class_table) = CG(class_table);
404    CG(class_table) = &TG(class_table);
405    EG(class_table) = CG(class_table);
406
407    TG(included_files) = &EG(included_files);
408    TG(open_files)     = &CG(open_files);
409
410    zend_llist_init(TG(open_files), sizeof(zend_file_handle), (void (*)(void *)) zend_file_handle_dtor, 0);
411    zend_hash_init_ex(TG(included_files), 5, NULL, NULL, 0, 1);
412#ifdef HAVE_XCACHE_CONSTANT
413    zend_hash_init_ex(&TG(zend_constants), 20, NULL, ZEND_CONSTANT_DTOR, 1, 0);
414#endif
415    zend_hash_init_ex(&TG(function_table), 128, NULL, ZEND_FUNCTION_DTOR, 0, 0);
416    zend_hash_init_ex(&TG(class_table), 16, NULL, ZEND_CLASS_DTOR, 0, 0);
417
418    sandbox->filename = filename;
419
420    return sandbox;
421}
422/* }}} */
423static void xc_sandbox_install(xc_sandbox_t *sandbox TSRMLS_DC) /* {{{ */
424{
425    int i;
426    Bucket *b;
427    zend_llist_position lpos;
428    zend_file_handle *handle;
429
430#ifdef HAVE_XCACHE_CONSTANT
431    b = TG(zend_constants).pListHead;
432    /* install constants */
433    while (b != NULL) {
434        zend_constant *c = (zend_constant*) b->pData;
435        xc_install_constant(sandbox->filename, c,
436                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY(b)), b->nKeyLength TSRMLS_CC);
437        b = b->pListNext;
438    }
439#endif
440
441    b = TG(function_table).pListHead;
442    /* install function */
443    while (b != NULL) {
444        zend_function *func = (zend_function*) b->pData;
445        xc_install_function(sandbox->filename, func,
446                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY(b)), b->nKeyLength TSRMLS_CC);
447        b = b->pListNext;
448    }
449
450    b = TG(class_table).pListHead;
451    /* install class */
452    while (b != NULL) {
453        xc_install_class(sandbox->filename, (xc_cest_t*)b->pData,
454                BUCKET_KEY_TYPE(b), ZSTR(BUCKET_KEY(b)), b->nKeyLength TSRMLS_CC);
455        b = b->pListNext;
456    }
457
458    i = 1;
459    zend_hash_add(&OG(included_files), sandbox->filename, strlen(sandbox->filename) + 1, (void *)&i, sizeof(int), NULL);
460    for (handle = zend_llist_get_first_ex(TG(open_files), &lpos);
461            handle;
462            handle = zend_llist_get_next_ex(TG(open_files), &lpos)) {
463        zend_llist_add_element(&OG(open_files), handle);
464    }
465}
466/* }}} */
467void xc_sandbox_free(xc_sandbox_t *sandbox, int install TSRMLS_DC) /* {{{ */
468{
469    /* restore first first install function/class */
470#ifdef HAVE_XCACHE_CONSTANT
471    EG(zend_constants) = OG(zend_constants);
472#endif
473    CG(function_table) = OG(function_table);
474    CG(class_table)    = OG(class_table);
475    EG(class_table)    = CG(class_table);
476
477    if (install) {
478        xc_sandbox_install(sandbox TSRMLS_CC);
479
480        /* no free as it's installed */
481#ifdef HAVE_XCACHE_CONSTANT
482        TG(zend_constants).pDestructor = NULL;
483#endif
484        TG(function_table).pDestructor = NULL;
485        TG(class_table).pDestructor = NULL;
486        TG(open_files)->dtor = NULL;
487    }
488
489    /* destroy all the tmp */
490#ifdef HAVE_XCACHE_CONSTANT
491    zend_hash_destroy(&TG(zend_constants));
492#endif
493    zend_hash_destroy(&TG(function_table));
494    zend_hash_destroy(&TG(class_table));
495    zend_hash_destroy(TG(included_files));
496    zend_llist_destroy(TG(open_files));
497
498    /* restore orig here, as EG/CG holded tmp before */
499    memcpy(&EG(included_files), &OG(included_files), sizeof(EG(included_files)));
500    memcpy(&CG(open_files),     &OG(open_files),     sizeof(CG(open_files)));
501
502    if (sandbox->alloc) {
503        efree(sandbox);
504    }
505}
506/* }}} */
Note: See TracBrowser for help on using the repository browser.