source: trunk/disassembler.c @ 938

Last change on this file since 938 was 938, checked in by moo, 2 years ago

finished sandbox rewrite

  • Property svn:eol-style set to native
File size: 4.7 KB
Line 
1#include "disassembler.h"
2#include "xcache.h"
3#include "utils.h"
4#include "processor.h"
5
6#ifndef HAVE_XCACHE_OPCODE_SPEC_DEF
7#error disassembler cannot be built without xcache/opcode_spec_def.h
8#endif
9static void xc_dasm(zval *output, zend_op_array *op_array TSRMLS_DC) /* {{{ */
10{
11    const Bucket *b;
12    zval *zv, *list;
13    xc_compile_result_t cr;
14    int bufsize = 2;
15    char *buf;
16    xc_dasm_t dasm;
17
18    xc_compile_result_init_cur(&cr, op_array TSRMLS_CC);
19
20    xc_apply_op_array(&cr, (apply_func_t) xc_undo_pass_two TSRMLS_CC);
21    xc_apply_op_array(&cr, (apply_func_t) xc_fix_opcode TSRMLS_CC);
22
23    /* go */
24    array_init(output);
25
26    ALLOC_INIT_ZVAL(zv);
27    array_init(zv);
28    xc_dasm_zend_op_array(&dasm, zv, op_array TSRMLS_CC);
29    add_assoc_zval_ex(output, ZEND_STRS("op_array"), zv);
30
31    buf = emalloc(bufsize);
32
33    ALLOC_INIT_ZVAL(list);
34    array_init(list);
35    for (b = xc_sandbox_user_function_begin(TSRMLS_C); b; b = b->pListNext) {
36        int keysize, keyLength;
37
38        ALLOC_INIT_ZVAL(zv);
39        array_init(zv);
40        xc_dasm_zend_function(&dasm, zv, b->pData TSRMLS_CC);
41
42        keysize = BUCKET_KEY_SIZE(b) + 2;
43        if (keysize > bufsize) {
44            do {
45                bufsize *= 2;
46            } while (keysize > bufsize);
47            buf = erealloc(buf, bufsize);
48        }
49        memcpy(buf, BUCKET_KEY_S(b), keysize);
50        buf[keysize - 2] = buf[keysize - 1] = ""[0];
51        keyLength = b->nKeyLength;
52#ifdef IS_UNICODE
53        if (BUCKET_KEY_TYPE(b) == IS_UNICODE) {
54            if (buf[0] == ""[0] && buf[1] == ""[0]) {
55                keyLength ++;
56            }
57        } else
58#endif
59        {
60            if (buf[0] == ""[0]) {
61                keyLength ++;
62            }
63        }
64
65        add_u_assoc_zval_ex(list, BUCKET_KEY_TYPE(b), ZSTR(buf), keyLength, zv);
66    }
67    add_assoc_zval_ex(output, ZEND_STRS("function_table"), list);
68   
69    ALLOC_INIT_ZVAL(list);
70    array_init(list);
71    for (b = xc_sandbox_user_class_begin(TSRMLS_C); b; b = b->pListNext) {
72        int keysize, keyLength;
73
74        ALLOC_INIT_ZVAL(zv);
75        array_init(zv);
76        xc_dasm_zend_class_entry(&dasm, zv, CestToCePtr(*(xc_cest_t *)b->pData) TSRMLS_CC);
77
78        keysize = BUCKET_KEY_SIZE(b) + 2;
79        if (keysize > bufsize) {
80            do {
81                bufsize *= 2;
82            } while (keysize > bufsize);
83            buf = erealloc(buf, bufsize);
84        }
85        memcpy(buf, BUCKET_KEY_S(b), keysize);
86        buf[keysize - 2] = buf[keysize - 1] = ""[0];
87        keyLength = b->nKeyLength;
88#ifdef IS_UNICODE
89        if (BUCKET_KEY_TYPE(b) == IS_UNICODE) {
90            if (buf[0] == ""[0] && buf[1] == ""[0]) {
91                keyLength ++;
92            }
93        } else
94#endif
95        {
96            if (buf[0] == ""[0]) {
97                keyLength ++;
98            }
99        }
100        add_u_assoc_zval_ex(list, BUCKET_KEY_TYPE(b), ZSTR(buf), keyLength, zv);
101    }
102    efree(buf);
103    add_assoc_zval_ex(output, ZEND_STRS("class_table"), list);
104
105    /*xc_apply_op_array(&cr, (apply_func_t) xc_redo_pass_two TSRMLS_CC);*/
106    xc_compile_result_free(&cr);
107}
108/* }}} */
109typedef struct xc_dasm_sandboxed_t { /* {{{ */
110    enum Type {
111        xc_dasm_file_t
112        , xc_dasm_string_t
113    } type;
114    union {
115        zval *zfilename;
116        struct {
117            zval *source;
118            char *eval_name;
119        } compile_string;
120    } input;
121
122    zval *output;
123} xc_dasm_sandboxed_t; /* {{{ */
124
125zend_op_array *xc_dasm_sandboxed(void *data TSRMLS_DC)
126{
127    zend_bool catched = 0;
128    zend_op_array *op_array = NULL;
129    xc_dasm_sandboxed_t *sandboxed_dasm = (xc_dasm_sandboxed_t *) data;
130
131    zend_try {
132        if (sandboxed_dasm->type == xc_dasm_file_t) {
133            op_array = compile_filename(ZEND_REQUIRE, sandboxed_dasm->input.zfilename TSRMLS_CC);
134        }
135        else {
136            op_array = compile_string(sandboxed_dasm->input.compile_string.source, sandboxed_dasm->input.compile_string.eval_name TSRMLS_CC);
137        }
138    } zend_catch {
139        catched = 1;
140    } zend_end_try();
141
142    if (catched || !op_array) {
143#define return_value sandboxed_dasm->output
144        RETVAL_FALSE;
145#undef return_value
146        return NULL;
147    }
148
149    xc_dasm(sandboxed_dasm->output, op_array TSRMLS_CC);
150
151    /* free */
152#ifdef ZEND_ENGINE_2
153    destroy_op_array(op_array TSRMLS_CC);
154#else
155    destroy_op_array(op_array);
156#endif
157    efree(op_array);
158
159    return NULL;
160} /* }}} */
161void xc_dasm_string(zval *output, zval *source TSRMLS_DC) /* {{{ */
162{
163    xc_dasm_sandboxed_t sandboxed_dasm;
164    char *eval_name = zend_make_compiled_string_description("runtime-created function" TSRMLS_CC);
165
166    sandboxed_dasm.output = output;
167    sandboxed_dasm.type = xc_dasm_string_t;
168    sandboxed_dasm.input.compile_string.source = source;
169    sandboxed_dasm.input.compile_string.eval_name = eval_name;
170    xc_sandbox(&xc_dasm_sandboxed, (void *) &sandboxed_dasm, eval_name TSRMLS_CC);
171    efree(eval_name);
172}
173/* }}} */
174void xc_dasm_file(zval *output, const char *filename TSRMLS_DC) /* {{{ */
175{
176    zval *zfilename;
177    xc_dasm_sandboxed_t sandboxed_dasm;
178
179    MAKE_STD_ZVAL(zfilename);
180    zfilename->value.str.val = estrdup(filename);
181    zfilename->value.str.len = strlen(filename);
182    zfilename->type = IS_STRING;
183
184    sandboxed_dasm.output = output;
185    sandboxed_dasm.type = xc_dasm_file_t;
186    sandboxed_dasm.input.zfilename = zfilename;
187    xc_sandbox(&xc_dasm_sandboxed, (void *) &sandboxed_dasm, zfilename->value.str.val TSRMLS_CC);
188
189    zval_dtor(zfilename);
190    FREE_ZVAL(zfilename);
191}
192/* }}} */
Note: See TracBrowser for help on using the repository browser.