source: trunk/xcache/xc_utils.c @ 1314

Last change on this file since 1314 was 1314, checked in by moo, 13 months ago

Decompiler, disassembler: update PHP 5.0 support

  • Property svn:eol-style set to native
File size: 15.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#include "util/xc_trace.h"
10
11#ifndef max
12#define max(a, b) ((a) < (b) ? (b) : (a))
13#endif
14
15#ifndef ZEND_VM_SET_OPCODE_HANDLER
16#   define ZEND_VM_SET_OPCODE_HANDLER(opline) do { } while (0)
17#endif
18
19#ifdef ZEND_ENGINE_2_4
20#   define OP_ZVAL_DTOR(op) do { } while(0)
21#else
22#   define OP_ZVAL_DTOR(op) do { \
23        Z_UNSET_ISREF(Z_OP_CONSTANT(op)); \
24        zval_dtor(&Z_OP_CONSTANT(op)); \
25    } while(0)
26#endif
27
28/* }}} */
29
30xc_compile_result_t *xc_compile_result_init(xc_compile_result_t *cr, /* {{{ */
31        zend_op_array *op_array,
32        HashTable *function_table,
33        HashTable *class_table)
34{
35    assert(cr);
36    cr->op_array       = op_array;
37    cr->function_table = function_table;
38    cr->class_table    = class_table;
39    return cr;
40}
41/* }}} */
42xc_compile_result_t *xc_compile_result_init_cur(xc_compile_result_t *cr, zend_op_array *op_array TSRMLS_DC) /* {{{ */
43{
44    assert(cr);
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}
51/* }}} */
52
53typedef struct {
54    apply_func_t applyer;
55} xc_apply_func_info;
56static int xc_apply_function(zend_function *zf, xc_apply_func_info *fi TSRMLS_DC) /* {{{ */
57{
58    switch (zf->type) {
59    case ZEND_USER_FUNCTION:
60    case ZEND_EVAL_CODE:
61        return fi->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    xc_apply_func_info fi;
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    if (mi->ce != zf->common.scope) {
81        /* fprintf(stderr, "avoided duplicate %s\n", zf->common.function_name); */
82        return 0;
83    }
84    return xc_apply_function(zf, &mi->fi TSRMLS_CC);
85}
86/* }}} */
87static int xc_apply_class(zend_class_entry **ce, xc_apply_func_info *fi TSRMLS_DC) /* {{{ */
88{
89    xc_apply_method_info mi;
90
91    mi.fi = *fi;
92    mi.ce = *ce;
93    zend_hash_apply_with_argument(&((*ce)->function_table), (apply_func_arg_t) xc_apply_method, &mi TSRMLS_CC);
94    return 0;
95}
96/* }}} */
97int xc_apply_op_array(xc_compile_result_t *cr, apply_func_t applyer TSRMLS_DC) /* {{{ */
98{
99    xc_apply_func_info fi;
100    fi.applyer = applyer;
101    zend_hash_apply_with_argument(cr->function_table, (apply_func_arg_t) xc_apply_function, &fi TSRMLS_CC);
102    zend_hash_apply_with_argument(cr->class_table, (apply_func_arg_t) xc_apply_class, &fi TSRMLS_CC);
103
104    return applyer(cr->op_array TSRMLS_CC);
105}
106/* }}} */
107int xc_undo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
108{
109    zend_op *opline, *opline_end;
110
111#ifdef ZEND_ENGINE_2_4
112    if (!(op_array->fn_flags & ZEND_ACC_DONE_PASS_TWO)) {
113        return 0;
114    }
115#else
116    if (!op_array->done_pass_two) {
117        return 0;
118    }
119#endif
120
121    opline = op_array->opcodes;
122    opline_end = opline + op_array->last;
123    while (opline < opline_end) {
124#ifdef ZEND_ENGINE_2_4
125        if (opline->op1_type == IS_CONST) {
126            opline->op1.constant = opline->op1.literal - op_array->literals;
127        }
128        if (opline->op2_type == IS_CONST) {
129            opline->op2.constant = opline->op2.literal - op_array->literals;
130        }
131#endif
132
133        switch (opline->opcode) {
134#ifdef ZEND_GOTO
135            case ZEND_GOTO:
136#endif
137            case ZEND_JMP:
138#ifdef ZEND_FAST_CALL
139            case ZEND_FAST_CALL:
140#endif
141                assert(Z_OP(opline->op1).jmp_addr >= op_array->opcodes && (zend_uint) (Z_OP(opline->op1).jmp_addr - op_array->opcodes) < op_array->last);
142                Z_OP(opline->op1).opline_num = Z_OP(opline->op1).jmp_addr - op_array->opcodes;
143                break;
144            case ZEND_JMPZ:
145            case ZEND_JMPNZ:
146            case ZEND_JMPZ_EX:
147            case ZEND_JMPNZ_EX:
148#ifdef ZEND_JMP_SET
149            case ZEND_JMP_SET:
150#endif
151#ifdef ZEND_JMP_SET_VAR
152            case ZEND_JMP_SET_VAR:
153#endif
154                assert(Z_OP(opline->op2).jmp_addr >= op_array->opcodes && (zend_uint) (Z_OP(opline->op2).jmp_addr - op_array->opcodes) < op_array->last);
155                Z_OP(opline->op2).opline_num = Z_OP(opline->op2).jmp_addr - op_array->opcodes;
156                break;
157        }
158        opline++;
159    }
160#ifdef ZEND_ENGINE_2_4
161    op_array->fn_flags &= ~ZEND_ACC_DONE_PASS_TWO;
162#else
163    op_array->done_pass_two = 0;
164#endif
165
166    return 0;
167}
168/* }}} */
169int xc_redo_pass_two(zend_op_array *op_array TSRMLS_DC) /* {{{ */
170{
171    zend_op *opline, *opline_end;
172#ifdef ZEND_ENGINE_2_4
173    zend_literal *literal = op_array->literals;
174#endif
175
176#ifdef ZEND_ENGINE_2_4
177    if ((op_array->fn_flags & ZEND_ACC_DONE_PASS_TWO)) {
178        return 0;
179    }
180#else
181    if (op_array->done_pass_two) {
182        return 0;
183    }
184#endif
185
186    /*
187    op_array->opcodes = (zend_op *) erealloc(op_array->opcodes, sizeof(zend_op)*op_array->last);
188    op_array->size = op_array->last;
189    */
190#ifdef ZEND_ENGINE_2_4
191    if (literal) {
192        zend_literal *literal_end = literal + op_array->last_literal;
193        while (literal < literal_end) {
194            Z_SET_ISREF(literal->constant);
195            Z_SET_REFCOUNT(literal->constant, 2); /* Make sure is_ref won't be reset */
196            literal++;
197        }
198    }
199#endif
200
201    opline = op_array->opcodes;
202    opline_end = opline + op_array->last;
203    while (opline < opline_end) {
204#ifdef ZEND_ENGINE_2_4
205        if (opline->op1_type == IS_CONST) {
206            opline->op1.literal = op_array->literals + opline->op1.constant;
207        }
208        if (opline->op2_type == IS_CONST) {
209            opline->op2.literal = op_array->literals + opline->op2.constant;
210        }
211#else
212        if (Z_OP_TYPE(opline->op1) == IS_CONST) {
213            Z_SET_ISREF(Z_OP_CONSTANT(opline->op1));
214            Z_SET_REFCOUNT(Z_OP_CONSTANT(opline->op1), 2); /* Make sure is_ref won't be reset */
215        }
216        if (Z_OP_TYPE(opline->op2) == IS_CONST) {
217            Z_SET_ISREF(Z_OP_CONSTANT(opline->op2));
218            Z_SET_REFCOUNT(Z_OP_CONSTANT(opline->op2), 2);
219        }
220#endif
221        switch (opline->opcode) {
222#ifdef ZEND_GOTO
223            case ZEND_GOTO:
224#endif
225            case ZEND_JMP:
226#ifdef ZEND_FAST_CALL
227            case ZEND_FAST_CALL:
228#endif
229                assert(Z_OP(opline->op1).opline_num < op_array->last);
230                Z_OP(opline->op1).jmp_addr = op_array->opcodes + Z_OP(opline->op1).opline_num;
231                break;
232            case ZEND_JMPZ:
233            case ZEND_JMPNZ:
234            case ZEND_JMPZ_EX:
235            case ZEND_JMPNZ_EX:
236#ifdef ZEND_JMP_SET
237            case ZEND_JMP_SET:
238#endif
239#ifdef ZEND_JMP_SET_VAR
240            case ZEND_JMP_SET_VAR:
241#endif
242                assert(Z_OP(opline->op2).opline_num < op_array->last);
243                Z_OP(opline->op2).jmp_addr = op_array->opcodes + Z_OP(opline->op2).opline_num;
244                break;
245        }
246        /* ZEND_VM_SET_OPCODE_HANDLER(opline); this is not undone, don't redo. only do this for loader */
247        opline++;
248    }
249
250#ifdef ZEND_ENGINE_2_4
251    op_array->fn_flags |= ZEND_ACC_DONE_PASS_TWO;
252#else
253    op_array->done_pass_two = 1;
254#endif
255    return 0;
256}
257/* }}} */
258
259static 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) /* {{{ */
260{
261    if ((*op_type != IS_UNUSED && (spec == OPSPEC_UCLASS || spec == OPSPEC_CLASS)) ||
262            spec == OPSPEC_FETCH) {
263        if (tofix) {
264            switch (*op_type) {
265            case IS_VAR:
266            case IS_TMP_VAR:
267                break;
268
269            default:
270                /* TODO: data lost, find a way to keep it */
271                /* assert(*op_type == IS_CONST); */
272                *op_type = IS_TMP_VAR;
273            }
274        }
275    }
276    switch (*op_type) {
277    case IS_TMP_VAR:
278    case IS_VAR:
279        if (tofix) {
280            Z_OP(*op).var /= sizeof(temp_variable);
281        }
282        else {
283            Z_OP(*op).var *= sizeof(temp_variable);
284        }
285    }
286}
287/* }}} */
288
289static void xc_fix_opcode_ex(zend_op_array *op_array, int tofix TSRMLS_DC) /* {{{ */
290{
291    zend_op *opline;
292    zend_uint i;
293
294    opline = op_array->opcodes;
295    for (i = 0; i < op_array->last; i ++, opline ++) {
296        /* 3rd optimizer may have ... */
297        if (opline->opcode < xc_get_opcode_spec_count()) {
298            const xc_opcode_spec_t *spec;
299            spec = xc_get_opcode_spec(opline->opcode);
300
301            xc_fix_opcode_ex_znode(tofix, spec->op1, &Z_OP_TYPE(opline->op1),    &opline->op1, 0 TSRMLS_CC);
302            xc_fix_opcode_ex_znode(tofix, spec->op2, &Z_OP_TYPE(opline->op2),    &opline->op2, 1 TSRMLS_CC);
303            xc_fix_opcode_ex_znode(tofix, spec->res, &Z_OP_TYPE(opline->result), &opline->result, 2 TSRMLS_CC);
304        }
305    }
306}
307/* }}} */
308int xc_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
309{
310    xc_fix_opcode_ex(op_array, 1 TSRMLS_CC);
311    return 0;
312}
313/* }}} */
314int xc_undo_fix_opcode(zend_op_array *op_array TSRMLS_DC) /* {{{ */
315{
316    xc_fix_opcode_ex(op_array, 0 TSRMLS_CC);
317
318    return 0;
319}
320/* }}} */
321
322#ifndef ZEND_COMPILE_DELAYED_BINDING
323int xc_foreach_early_binding_class(zend_op_array *op_array, xc_foreach_early_binding_class_cb callback, void *data TSRMLS_DC) /* {{{ */
324{
325    zend_op *opline, *begin, *opline_end, *next = NULL;
326
327    opline = begin = op_array->opcodes;
328    opline_end = opline + op_array->last;
329    while (opline < opline_end) {
330        switch (opline->opcode) {
331#ifdef ZEND_GOTO
332            case ZEND_GOTO:
333#endif
334            case ZEND_JMP:
335#ifdef ZEND_FAST_CALL
336            case ZEND_FAST_CALL:
337#endif
338                next = begin + Z_OP(opline->op1).opline_num;
339                break;
340
341            case ZEND_JMPZNZ:
342                next = begin + max(Z_OP(opline->op2).opline_num, opline->extended_value);
343                break;
344
345            case ZEND_JMPZ:
346            case ZEND_JMPNZ:
347            case ZEND_JMPZ_EX:
348            case ZEND_JMPNZ_EX:
349#ifdef ZEND_JMP_SET
350            case ZEND_JMP_SET:
351#endif
352#ifdef ZEND_JMP_SET_VAR
353            case ZEND_JMP_SET_VAR:
354#endif
355                next = begin + Z_OP(opline->op2).opline_num;
356                break;
357
358            case ZEND_RETURN:
359                opline = opline_end;
360                break;
361
362            case ZEND_DECLARE_INHERITED_CLASS:
363                callback(opline, opline - begin, data TSRMLS_CC);
364                break;
365        }
366
367        if (opline < next) {
368            opline = next;
369        }
370        else {
371            opline ++;
372        }
373    }
374    return SUCCESS;
375}
376/* }}} */
377int xc_do_early_binding(zend_op_array *op_array, HashTable *class_table, int oplineno TSRMLS_DC) /* {{{ */
378{
379    zend_op *opline;
380
381    TRACE("binding %d", oplineno);
382    assert(oplineno >= 0);
383
384    /* do early binding */
385    opline = &(op_array->opcodes[oplineno]);
386
387    switch (opline->opcode) {
388    case ZEND_DECLARE_INHERITED_CLASS:
389        {
390            zval *parent_name;
391            zend_class_entry **pce;
392
393            /* don't early-bind classes that implement interfaces */
394            if ((opline + 1)->opcode == ZEND_FETCH_CLASS && (opline + 2)->opcode == ZEND_ADD_INTERFACE) {
395                return FAILURE;
396            }
397
398            parent_name = &(Z_OP_CONSTANT((opline - 1)->op2));
399            TRACE("binding with parent %s", Z_STRVAL_P(parent_name));
400            if (zend_lookup_class(Z_STRVAL_P(parent_name), Z_STRLEN_P(parent_name), &pce TSRMLS_CC) == FAILURE) {
401                return FAILURE;
402            }
403
404            if (do_bind_inherited_class(opline, class_table, *pce, 1 TSRMLS_CC) == NULL) {
405                return FAILURE;
406            }
407        }
408
409        /* clear unnecessary ZEND_FETCH_CLASS opcode */
410        if (opline > op_array->opcodes
411         && (opline - 1)->opcode == ZEND_FETCH_CLASS) {
412            zend_op *fetch_class_opline = opline - 1;
413
414            TRACE("%s %p", Z_STRVAL(Z_OP_CONSTANT(fetch_class_opline->op2)), Z_STRVAL(Z_OP_CONSTANT(fetch_class_opline->op2)));
415            OP_ZVAL_DTOR(fetch_class_opline->op2);
416            fetch_class_opline->opcode = ZEND_NOP;
417            ZEND_VM_SET_OPCODE_HANDLER(fetch_class_opline);
418            memset(&fetch_class_opline->op1, 0, sizeof(znode));
419            memset(&fetch_class_opline->op2, 0, sizeof(znode));
420            SET_UNUSED(fetch_class_opline->op1);
421            SET_UNUSED(fetch_class_opline->op2);
422            SET_UNUSED(fetch_class_opline->result);
423        }
424
425        /* clear unnecessary ZEND_VERIFY_ABSTRACT_CLASS opcode */
426        if ((opline + 1)->opcode == ZEND_VERIFY_ABSTRACT_CLASS) {
427            zend_op *abstract_op = opline + 1;
428            memset(abstract_op, 0, sizeof(abstract_op[0]));
429            abstract_op->lineno = 0;
430            SET_UNUSED(abstract_op->op1);
431            SET_UNUSED(abstract_op->op2);
432            SET_UNUSED(abstract_op->result);
433            abstract_op->opcode = ZEND_NOP;
434            ZEND_VM_SET_OPCODE_HANDLER(abstract_op);
435        }
436        break;
437
438    default:
439        return FAILURE;
440    }
441
442    zend_hash_del(class_table, Z_OP_CONSTANT(opline->op1).value.str.val, Z_OP_CONSTANT(opline->op1).value.str.len);
443    OP_ZVAL_DTOR(opline->op1);
444    OP_ZVAL_DTOR(opline->op2);
445    opline->opcode = ZEND_NOP;
446    ZEND_VM_SET_OPCODE_HANDLER(opline);
447    memset(&opline->op1, 0, sizeof(znode));
448    memset(&opline->op2, 0, sizeof(znode));
449    SET_UNUSED(opline->op1);
450    SET_UNUSED(opline->op2);
451    return SUCCESS;
452}
453/* }}} */
454#endif
455
456#ifdef HAVE_XCACHE_CONSTANT
457void xc_install_constant(ZEND_24(NOTHING, const) char *filename, zend_constant *constant, zend_uchar type, const24_zstr key, uint len, ulong h TSRMLS_DC) /* {{{ */
458{
459    if (zend_u_hash_add(EG(zend_constants), type, key, len,
460                constant, sizeof(zend_constant),
461                NULL
462                ) == FAILURE) {
463        CG(zend_lineno) = 0;
464#ifdef IS_UNICODE
465        zend_error(E_NOTICE, "Constant %R already defined", type, key);
466#else
467        zend_error(E_NOTICE, "Constant %s already defined", key);
468#endif
469        free(ZSTR_V(constant->name));
470        if (!(constant->flags & CONST_PERSISTENT)) {
471            zval_dtor(&constant->value);
472        }
473    }
474}
475/* }}} */
476#endif
477void xc_install_function(ZEND_24(NOTHING, const) char *filename, zend_function *func, zend_uchar type, const24_zstr key, uint len, ulong h TSRMLS_DC) /* {{{ */
478{
479    zend_bool istmpkey;
480
481    if (func->type == ZEND_USER_FUNCTION) {
482#ifdef IS_UNICODE
483        istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
484#else
485        istmpkey = ZSTR_S(key)[0] == 0;
486#endif
487        if (istmpkey) {
488            zend_u_hash_update(CG(function_table), type, key, len,
489                        func, sizeof(zend_op_array),
490                        NULL
491                        );
492        }
493        else if (zend_u_hash_add(CG(function_table), type, key, len,
494                    func, sizeof(zend_op_array),
495                    NULL
496                    ) == FAILURE) {
497            CG(zend_lineno) = func->op_array.line_start;
498#ifdef IS_UNICODE
499            zend_error(E_ERROR, "Cannot redeclare %R()", type, key);
500#else
501            zend_error(E_ERROR, "Cannot redeclare %s()", key);
502#endif
503        }
504    }
505}
506/* }}} */
507void xc_install_class(ZEND_24(NOTHING, const) char *filename, zend_class_entry *ce, int oplineno, zend_uchar type, const24_zstr key, uint len, ulong h TSRMLS_DC) /* {{{ */
508{
509    zend_bool istmpkey;
510
511#ifdef IS_UNICODE
512    istmpkey = (type == IS_STRING && ZSTR_S(key)[0] == 0) || ZSTR_U(key)[0] == 0;
513#else
514    istmpkey = ZSTR_S(key)[0] == 0;
515#endif
516    if (istmpkey) {
517        zend_u_hash_quick_update(CG(class_table), type, key, len, h,
518                    &ce, sizeof(zend_class_entry *),
519                    NULL
520                    );
521#ifndef ZEND_COMPILE_DELAYED_BINDING
522        if (oplineno != -1) {
523            xc_do_early_binding(CG(active_op_array), CG(class_table), oplineno TSRMLS_CC);
524        }
525#endif
526    }
527    else if (zend_u_hash_quick_add(CG(class_table), type, key, len, h,
528                &ce, sizeof(zend_class_entry *),
529                NULL
530                ) == FAILURE) {
531        CG(zend_lineno) = Z_CLASS_INFO(*ce).line_start;
532#ifdef IS_UNICODE
533        zend_error(E_ERROR, "Cannot redeclare class %R", type, ce->name);
534#else
535        zend_error(E_ERROR, "Cannot redeclare class %s", ce->name);
536#endif
537        assert(oplineno == -1);
538    }
539}
540/* }}} */
541
542void xc_hash_copy_if(HashTable *target, HashTable *source, copy_ctor_func_t pCopyConstructor, void *tmp, uint size, xc_if_func_t checker) /* {{{ */
543{
544    Bucket *p;
545    void *new_entry;
546    zend_bool setTargetPointer;
547
548    setTargetPointer = !target->pInternalPointer;
549    p = source->pListHead;
550    while (p) {
551        if (checker(p->pData)) {
552            if (p->nKeyLength) {
553                zend_u_hash_quick_update(target, p->key.type, ZSTR(BUCKET_KEY_S(p)), p->nKeyLength, p->h, p->pData, size, &new_entry);
554            } else {
555                zend_hash_index_update(target, p->h, p->pData, size, &new_entry);
556            }
557            if (pCopyConstructor) {
558                pCopyConstructor(new_entry);
559            }
560            if (setTargetPointer && source->pInternalPointer == p) {
561                target->pInternalPointer = new_entry;
562            }
563        }
564        p = p->pListNext;
565    }
566    if (!target->pInternalPointer) {
567        target->pInternalPointer = target->pListHead;
568    }
569}
570/* }}} */
571#ifdef HAVE_XCACHE_CONSTANT
572static zend_bool xc_is_internal_zend_constant(zend_constant *c) /* {{{ */
573{
574    return (c->flags & CONST_PERSISTENT) ? 1 : 0;
575}
576/* }}} */
577void xc_zend_constant_ctor(zend_constant *c) /* {{{ */
578{
579    assert((c->flags & CONST_PERSISTENT));
580    ZSTR_U(c->name) = UNISW(zend_strndup, zend_ustrndup)(ZSTR_U(c->name), c->name_len - 1);
581}
582/* }}} */
583void xc_zend_constant_dtor(zend_constant *c) /* {{{ */
584{
585    free(ZSTR_V(c->name));
586}
587/* }}} */
588void xc_copy_internal_zend_constants(HashTable *target, HashTable *source) /* {{{ */
589{
590    zend_constant tmp_const;
591    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);
592}
593/* }}} */
594#endif
Note: See TracBrowser for help on using the repository browser.