1 /*
2 +----------------------------------------------------------------------+
3 | Copyright (c) The PHP Group |
4 +----------------------------------------------------------------------+
5 | This source file is subject to version 3.01 of the PHP license, |
6 | that is bundled with this package in the file LICENSE, and is |
7 | available through the world-wide-web at the following url: |
8 | http://www.php.net/license/3_01.txt |
9 | If you did not receive a copy of the PHP license and are unable to |
10 | obtain it through the world-wide-web, please send a note to |
11 | license@php.net so we can mail you a copy immediately. |
12 +----------------------------------------------------------------------+
13 | Author: Thies C. Arntzen <thies@thieso.net> |
14 +----------------------------------------------------------------------+
15 */
16
17 /* {{{ includes */
18 #include "php.h"
19 #include "php_assert.h"
20 #include "php_ini.h"
21 #include "zend_exceptions.h"
22 /* }}} */
23
24 ZEND_BEGIN_MODULE_GLOBALS(assert)
25 zval callback;
26 char *cb;
27 zend_bool active;
28 zend_bool bail;
29 zend_bool warning;
30 zend_bool exception;
31 ZEND_END_MODULE_GLOBALS(assert)
32
33 ZEND_DECLARE_MODULE_GLOBALS(assert)
34
35 static zend_class_entry *assertion_error_ce;
36
37 #define ASSERTG(v) ZEND_MODULE_GLOBALS_ACCESSOR(assert, v)
38
39 #define SAFE_STRING(s) ((s)?(s):"")
40
41 enum {
42 ASSERT_ACTIVE=1,
43 ASSERT_CALLBACK,
44 ASSERT_BAIL,
45 ASSERT_WARNING,
46 ASSERT_EXCEPTION
47 };
48
PHP_INI_MH(OnChangeCallback)49 static PHP_INI_MH(OnChangeCallback) /* {{{ */
50 {
51 if (EG(current_execute_data)) {
52 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
53 zval_ptr_dtor(&ASSERTG(callback));
54 ZVAL_UNDEF(&ASSERTG(callback));
55 }
56 if (new_value && (Z_TYPE(ASSERTG(callback)) != IS_UNDEF || ZSTR_LEN(new_value))) {
57 ZVAL_STR_COPY(&ASSERTG(callback), new_value);
58 }
59 } else {
60 if (ASSERTG(cb)) {
61 pefree(ASSERTG(cb), 1);
62 }
63 if (new_value && ZSTR_LEN(new_value)) {
64 ASSERTG(cb) = pemalloc(ZSTR_LEN(new_value) + 1, 1);
65 memcpy(ASSERTG(cb), ZSTR_VAL(new_value), ZSTR_LEN(new_value));
66 ASSERTG(cb)[ZSTR_LEN(new_value)] = '\0';
67 } else {
68 ASSERTG(cb) = NULL;
69 }
70 }
71 return SUCCESS;
72 }
73 /* }}} */
74
75 PHP_INI_BEGIN()
76 STD_PHP_INI_BOOLEAN("assert.active", "1", PHP_INI_ALL, OnUpdateBool, active, zend_assert_globals, assert_globals)
77 STD_PHP_INI_BOOLEAN("assert.bail", "0", PHP_INI_ALL, OnUpdateBool, bail, zend_assert_globals, assert_globals)
78 STD_PHP_INI_BOOLEAN("assert.warning", "1", PHP_INI_ALL, OnUpdateBool, warning, zend_assert_globals, assert_globals)
79 PHP_INI_ENTRY("assert.callback", NULL, PHP_INI_ALL, OnChangeCallback)
80 STD_PHP_INI_BOOLEAN("assert.exception", "1", PHP_INI_ALL, OnUpdateBool, exception, zend_assert_globals, assert_globals)
PHP_INI_END()81 PHP_INI_END()
82
83 static void php_assert_init_globals(zend_assert_globals *assert_globals_p) /* {{{ */
84 {
85 ZVAL_UNDEF(&assert_globals_p->callback);
86 assert_globals_p->cb = NULL;
87 }
88 /* }}} */
89
PHP_MINIT_FUNCTION(assert)90 PHP_MINIT_FUNCTION(assert) /* {{{ */
91 {
92 zend_class_entry ce;
93
94 ZEND_INIT_MODULE_GLOBALS(assert, php_assert_init_globals, NULL);
95
96 REGISTER_INI_ENTRIES();
97
98 REGISTER_LONG_CONSTANT("ASSERT_ACTIVE", ASSERT_ACTIVE, CONST_CS|CONST_PERSISTENT);
99 REGISTER_LONG_CONSTANT("ASSERT_CALLBACK", ASSERT_CALLBACK, CONST_CS|CONST_PERSISTENT);
100 REGISTER_LONG_CONSTANT("ASSERT_BAIL", ASSERT_BAIL, CONST_CS|CONST_PERSISTENT);
101 REGISTER_LONG_CONSTANT("ASSERT_WARNING", ASSERT_WARNING, CONST_CS|CONST_PERSISTENT);
102 REGISTER_LONG_CONSTANT("ASSERT_EXCEPTION", ASSERT_EXCEPTION, CONST_CS|CONST_PERSISTENT);
103
104 INIT_CLASS_ENTRY(ce, "AssertionError", NULL);
105 assertion_error_ce = zend_register_internal_class_ex(&ce, zend_ce_error);
106
107 return SUCCESS;
108 }
109 /* }}} */
110
PHP_MSHUTDOWN_FUNCTION(assert)111 PHP_MSHUTDOWN_FUNCTION(assert) /* {{{ */
112 {
113 if (ASSERTG(cb)) {
114 pefree(ASSERTG(cb), 1);
115 ASSERTG(cb) = NULL;
116 }
117 return SUCCESS;
118 }
119 /* }}} */
120
PHP_RSHUTDOWN_FUNCTION(assert)121 PHP_RSHUTDOWN_FUNCTION(assert) /* {{{ */
122 {
123 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
124 zval_ptr_dtor(&ASSERTG(callback));
125 ZVAL_UNDEF(&ASSERTG(callback));
126 }
127
128 return SUCCESS;
129 }
130 /* }}} */
131
PHP_MINFO_FUNCTION(assert)132 PHP_MINFO_FUNCTION(assert) /* {{{ */
133 {
134 DISPLAY_INI_ENTRIES();
135 }
136 /* }}} */
137
138 /* {{{ Checks if assertion is false */
PHP_FUNCTION(assert)139 PHP_FUNCTION(assert)
140 {
141 zval *assertion;
142 zend_string *description_str = NULL;
143 zend_object *description_obj = NULL;
144
145 if (!ASSERTG(active)) {
146 RETURN_TRUE;
147 }
148
149 ZEND_PARSE_PARAMETERS_START(1, 2)
150 Z_PARAM_ZVAL(assertion)
151 Z_PARAM_OPTIONAL
152 Z_PARAM_OBJ_OF_CLASS_OR_STR_OR_NULL(description_obj, zend_ce_throwable, description_str)
153 ZEND_PARSE_PARAMETERS_END();
154
155 if (zend_is_true(assertion)) {
156 RETURN_TRUE;
157 }
158
159 if (description_obj) {
160 GC_ADDREF(description_obj);
161 zend_throw_exception_internal(description_obj);
162 RETURN_THROWS();
163 }
164
165 if (Z_TYPE(ASSERTG(callback)) == IS_UNDEF && ASSERTG(cb)) {
166 ZVAL_STRING(&ASSERTG(callback), ASSERTG(cb));
167 }
168
169 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
170 zval args[4];
171 zval retval;
172 uint32_t lineno = zend_get_executed_lineno();
173 const char *filename = zend_get_executed_filename();
174
175 ZVAL_STRING(&args[0], SAFE_STRING(filename));
176 ZVAL_LONG(&args[1], lineno);
177 ZVAL_NULL(&args[2]);
178
179 ZVAL_FALSE(&retval);
180
181 if (description_str) {
182 ZVAL_STR(&args[3], description_str);
183 call_user_function(NULL, NULL, &ASSERTG(callback), &retval, 4, args);
184 } else {
185 call_user_function(NULL, NULL, &ASSERTG(callback), &retval, 3, args);
186 }
187
188 zval_ptr_dtor(&args[0]);
189 zval_ptr_dtor(&retval);
190 }
191
192 if (ASSERTG(exception)) {
193 zend_throw_exception(assertion_error_ce, description_str ? ZSTR_VAL(description_str) : NULL, E_ERROR);
194 } else if (ASSERTG(warning)) {
195 php_error_docref(NULL, E_WARNING, "%s failed", description_str ? ZSTR_VAL(description_str) : "Assertion failed");
196 }
197
198 if (ASSERTG(bail)) {
199 zend_bailout();
200 }
201
202 RETURN_FALSE;
203 }
204 /* }}} */
205
206 /* {{{ Set/get the various assert flags */
PHP_FUNCTION(assert_options)207 PHP_FUNCTION(assert_options)
208 {
209 zval *value = NULL;
210 zend_long what;
211 zend_bool oldint;
212 int ac = ZEND_NUM_ARGS();
213 zend_string *key;
214
215 ZEND_PARSE_PARAMETERS_START(1, 2)
216 Z_PARAM_LONG(what)
217 Z_PARAM_OPTIONAL
218 Z_PARAM_ZVAL(value)
219 ZEND_PARSE_PARAMETERS_END();
220
221 switch (what) {
222 case ASSERT_ACTIVE:
223 oldint = ASSERTG(active);
224 if (ac == 2) {
225 zend_string *value_str = zval_try_get_string(value);
226 if (UNEXPECTED(!value_str)) {
227 RETURN_THROWS();
228 }
229
230 key = zend_string_init("assert.active", sizeof("assert.active")-1, 0);
231 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
232 zend_string_release_ex(key, 0);
233 zend_string_release_ex(value_str, 0);
234 }
235 RETURN_LONG(oldint);
236 break;
237
238 case ASSERT_BAIL:
239 oldint = ASSERTG(bail);
240 if (ac == 2) {
241 zend_string *value_str = zval_try_get_string(value);
242 if (UNEXPECTED(!value_str)) {
243 RETURN_THROWS();
244 }
245
246 key = zend_string_init("assert.bail", sizeof("assert.bail")-1, 0);
247 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
248 zend_string_release_ex(key, 0);
249 zend_string_release_ex(value_str, 0);
250 }
251 RETURN_LONG(oldint);
252 break;
253
254 case ASSERT_WARNING:
255 oldint = ASSERTG(warning);
256 if (ac == 2) {
257 zend_string *value_str = zval_try_get_string(value);
258 if (UNEXPECTED(!value_str)) {
259 RETURN_THROWS();
260 }
261
262 key = zend_string_init("assert.warning", sizeof("assert.warning")-1, 0);
263 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
264 zend_string_release_ex(key, 0);
265 zend_string_release_ex(value_str, 0);
266 }
267 RETURN_LONG(oldint);
268 break;
269
270 case ASSERT_CALLBACK:
271 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
272 ZVAL_COPY(return_value, &ASSERTG(callback));
273 } else if (ASSERTG(cb)) {
274 RETVAL_STRING(ASSERTG(cb));
275 } else {
276 RETVAL_NULL();
277 }
278
279 if (ac == 2) {
280 zval_ptr_dtor(&ASSERTG(callback));
281 if (Z_TYPE_P(value) == IS_NULL) {
282 ZVAL_UNDEF(&ASSERTG(callback));
283 } else {
284 ZVAL_COPY(&ASSERTG(callback), value);
285 }
286 }
287 return;
288
289 case ASSERT_EXCEPTION:
290 oldint = ASSERTG(exception);
291 if (ac == 2) {
292 zend_string *val = zval_try_get_string(value);
293 if (UNEXPECTED(!val)) {
294 RETURN_THROWS();
295 }
296
297 key = zend_string_init("assert.exception", sizeof("assert.exception")-1, 0);
298 zend_alter_ini_entry_ex(key, val, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
299 zend_string_release_ex(val, 0);
300 zend_string_release_ex(key, 0);
301 }
302 RETURN_LONG(oldint);
303 break;
304
305 default:
306 zend_argument_value_error(1, "must be an ASSERT_* constant");
307 RETURN_THROWS();
308 }
309 }
310 /* }}} */
311
