1 /*
2 +----------------------------------------------------------------------+
3 | PHP Version 7 |
4 +----------------------------------------------------------------------+
5 | Copyright (c) 1997-2018 The PHP Group |
6 +----------------------------------------------------------------------+
7 | This source file is subject to version 3.01 of the PHP license, |
8 | that is bundled with this package in the file LICENSE, and is |
9 | available through the world-wide-web at the following url: |
10 | http://www.php.net/license/3_01.txt |
11 | If you did not receive a copy of the PHP license and are unable to |
12 | obtain it through the world-wide-web, please send a note to |
13 | license@php.net so we can mail you a copy immediately. |
14 +----------------------------------------------------------------------+
15 | Author: Thies C. Arntzen <thies@thieso.net> |
16 +----------------------------------------------------------------------+
17 */
18
19 /* $Id$ */
20
21 /* {{{ includes */
22 #include "php.h"
23 #include "php_assert.h"
24 #include "php_ini.h"
25 #include "zend_exceptions.h"
26 /* }}} */
27
28 ZEND_BEGIN_MODULE_GLOBALS(assert)
29 zval callback;
30 char *cb;
31 zend_bool active;
32 zend_bool bail;
33 zend_bool warning;
34 zend_bool quiet_eval;
35 zend_bool exception;
36 ZEND_END_MODULE_GLOBALS(assert)
37
38 ZEND_DECLARE_MODULE_GLOBALS(assert)
39
40 static zend_class_entry *assertion_error_ce;
41
42 #define ASSERTG(v) ZEND_MODULE_GLOBALS_ACCESSOR(assert, v)
43
44 #define SAFE_STRING(s) ((s)?(s):"")
45
46 enum {
47 ASSERT_ACTIVE=1,
48 ASSERT_CALLBACK,
49 ASSERT_BAIL,
50 ASSERT_WARNING,
51 ASSERT_QUIET_EVAL,
52 ASSERT_EXCEPTION
53 };
54
PHP_INI_MH(OnChangeCallback)55 static PHP_INI_MH(OnChangeCallback) /* {{{ */
56 {
57 if (EG(current_execute_data)) {
58 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
59 zval_ptr_dtor(&ASSERTG(callback));
60 ZVAL_UNDEF(&ASSERTG(callback));
61 }
62 if (new_value && (Z_TYPE(ASSERTG(callback)) != IS_UNDEF || ZSTR_LEN(new_value))) {
63 ZVAL_STR_COPY(&ASSERTG(callback), new_value);
64 }
65 } else {
66 if (ASSERTG(cb)) {
67 pefree(ASSERTG(cb), 1);
68 }
69 if (new_value && ZSTR_LEN(new_value)) {
70 ASSERTG(cb) = pemalloc(ZSTR_LEN(new_value) + 1, 1);
71 memcpy(ASSERTG(cb), ZSTR_VAL(new_value), ZSTR_LEN(new_value));
72 ASSERTG(cb)[ZSTR_LEN(new_value)] = '\0';
73 } else {
74 ASSERTG(cb) = NULL;
75 }
76 }
77 return SUCCESS;
78 }
79 /* }}} */
80
81 PHP_INI_BEGIN()
82 STD_PHP_INI_ENTRY("assert.active", "1", PHP_INI_ALL, OnUpdateBool, active, zend_assert_globals, assert_globals)
83 STD_PHP_INI_ENTRY("assert.bail", "0", PHP_INI_ALL, OnUpdateBool, bail, zend_assert_globals, assert_globals)
84 STD_PHP_INI_ENTRY("assert.warning", "1", PHP_INI_ALL, OnUpdateBool, warning, zend_assert_globals, assert_globals)
85 PHP_INI_ENTRY("assert.callback", NULL, PHP_INI_ALL, OnChangeCallback)
86 STD_PHP_INI_ENTRY("assert.quiet_eval", "0", PHP_INI_ALL, OnUpdateBool, quiet_eval, zend_assert_globals, assert_globals)
87 STD_PHP_INI_ENTRY("assert.exception", "0", PHP_INI_ALL, OnUpdateBool, exception, zend_assert_globals, assert_globals)
PHP_INI_END()88 PHP_INI_END()
89
90 static void php_assert_init_globals(zend_assert_globals *assert_globals_p) /* {{{ */
91 {
92 ZVAL_UNDEF(&assert_globals_p->callback);
93 assert_globals_p->cb = NULL;
94 }
95 /* }}} */
96
PHP_MINIT_FUNCTION(assert)97 PHP_MINIT_FUNCTION(assert) /* {{{ */
98 {
99 zend_class_entry ce;
100
101 ZEND_INIT_MODULE_GLOBALS(assert, php_assert_init_globals, NULL);
102
103 REGISTER_INI_ENTRIES();
104
105 REGISTER_LONG_CONSTANT("ASSERT_ACTIVE", ASSERT_ACTIVE, CONST_CS|CONST_PERSISTENT);
106 REGISTER_LONG_CONSTANT("ASSERT_CALLBACK", ASSERT_CALLBACK, CONST_CS|CONST_PERSISTENT);
107 REGISTER_LONG_CONSTANT("ASSERT_BAIL", ASSERT_BAIL, CONST_CS|CONST_PERSISTENT);
108 REGISTER_LONG_CONSTANT("ASSERT_WARNING", ASSERT_WARNING, CONST_CS|CONST_PERSISTENT);
109 REGISTER_LONG_CONSTANT("ASSERT_QUIET_EVAL", ASSERT_QUIET_EVAL, CONST_CS|CONST_PERSISTENT);
110 REGISTER_LONG_CONSTANT("ASSERT_EXCEPTION", ASSERT_EXCEPTION, CONST_CS|CONST_PERSISTENT);
111
112 INIT_CLASS_ENTRY(ce, "AssertionError", NULL);
113 assertion_error_ce = zend_register_internal_class_ex(&ce, zend_ce_error);
114
115 return SUCCESS;
116 }
117 /* }}} */
118
PHP_MSHUTDOWN_FUNCTION(assert)119 PHP_MSHUTDOWN_FUNCTION(assert) /* {{{ */
120 {
121 if (ASSERTG(cb)) {
122 pefree(ASSERTG(cb), 1);
123 ASSERTG(cb) = NULL;
124 }
125 return SUCCESS;
126 }
127 /* }}} */
128
PHP_RSHUTDOWN_FUNCTION(assert)129 PHP_RSHUTDOWN_FUNCTION(assert) /* {{{ */
130 {
131 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
132 zval_ptr_dtor(&ASSERTG(callback));
133 ZVAL_UNDEF(&ASSERTG(callback));
134 }
135
136 return SUCCESS;
137 }
138 /* }}} */
139
PHP_MINFO_FUNCTION(assert)140 PHP_MINFO_FUNCTION(assert) /* {{{ */
141 {
142 DISPLAY_INI_ENTRIES();
143 }
144 /* }}} */
145
146 /* {{{ proto int assert(string|bool assertion[, mixed description])
147 Checks if assertion is false */
PHP_FUNCTION(assert)148 PHP_FUNCTION(assert)
149 {
150 zval *assertion;
151 zval *description = NULL;
152 int val;
153 char *myeval = NULL;
154 char *compiled_string_description;
155
156 if (! ASSERTG(active)) {
157 RETURN_TRUE;
158 }
159
160 if (zend_parse_parameters(ZEND_NUM_ARGS(), "z|z", &assertion, &description) == FAILURE) {
161 return;
162 }
163
164 if (Z_TYPE_P(assertion) == IS_STRING) {
165 zval retval;
166 int old_error_reporting = 0; /* shut up gcc! */
167
168 if (zend_forbid_dynamic_call("assert() with string argument") == FAILURE) {
169 RETURN_FALSE;
170 }
171
172 myeval = Z_STRVAL_P(assertion);
173
174 if (ASSERTG(quiet_eval)) {
175 old_error_reporting = EG(error_reporting);
176 EG(error_reporting) = 0;
177 }
178
179 compiled_string_description = zend_make_compiled_string_description("assert code");
180 if (zend_eval_stringl(myeval, Z_STRLEN_P(assertion), &retval, compiled_string_description) == FAILURE) {
181 efree(compiled_string_description);
182 if (!description) {
183 zend_throw_error(NULL, "Failure evaluating code: %s%s", PHP_EOL, myeval);
184 } else {
185 zend_string *str = zval_get_string(description);
186 zend_throw_error(NULL, "Failure evaluating code: %s%s:\"%s\"", PHP_EOL, ZSTR_VAL(str), myeval);
187 zend_string_release(str);
188 }
189 if (ASSERTG(bail)) {
190 zend_bailout();
191 }
192 RETURN_FALSE;
193 }
194 efree(compiled_string_description);
195
196 if (ASSERTG(quiet_eval)) {
197 EG(error_reporting) = old_error_reporting;
198 }
199
200 convert_to_boolean(&retval);
201 val = Z_TYPE(retval) == IS_TRUE;
202 } else {
203 val = zend_is_true(assertion);
204 }
205
206 if (val) {
207 RETURN_TRUE;
208 }
209
210 if (Z_TYPE(ASSERTG(callback)) == IS_UNDEF && ASSERTG(cb)) {
211 ZVAL_STRING(&ASSERTG(callback), ASSERTG(cb));
212 }
213
214 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
215 zval *args = safe_emalloc(!description ? 3 : 4, sizeof(zval), 0);
216 zval retval;
217 int i;
218 uint lineno = zend_get_executed_lineno();
219 const char *filename = zend_get_executed_filename();
220
221 ZVAL_STRING(&args[0], SAFE_STRING(filename));
222 ZVAL_LONG (&args[1], lineno);
223 ZVAL_STRING(&args[2], SAFE_STRING(myeval));
224
225 ZVAL_FALSE(&retval);
226
227 /* XXX do we want to check for error here? */
228 if (!description) {
229 call_user_function(CG(function_table), NULL, &ASSERTG(callback), &retval, 3, args);
230 for (i = 0; i <= 2; i++) {
231 zval_ptr_dtor(&(args[i]));
232 }
233 } else {
234 ZVAL_STR(&args[3], zval_get_string(description));
235 call_user_function(CG(function_table), NULL, &ASSERTG(callback), &retval, 4, args);
236 for (i = 0; i <= 3; i++) {
237 zval_ptr_dtor(&(args[i]));
238 }
239 }
240
241 efree(args);
242 zval_ptr_dtor(&retval);
243 }
244
245 if (ASSERTG(exception)) {
246 if (!description) {
247 zend_throw_exception(assertion_error_ce, NULL, E_ERROR);
248 } else if (Z_TYPE_P(description) == IS_OBJECT &&
249 instanceof_function(Z_OBJCE_P(description), zend_ce_throwable)) {
250 Z_ADDREF_P(description);
251 zend_throw_exception_object(description);
252 } else {
253 zend_string *str = zval_get_string(description);
254 zend_throw_exception(assertion_error_ce, ZSTR_VAL(str), E_ERROR);
255 zend_string_release(str);
256 }
257 } else if (ASSERTG(warning)) {
258 if (!description) {
259 if (myeval) {
260 php_error_docref(NULL, E_WARNING, "Assertion \"%s\" failed", myeval);
261 } else {
262 php_error_docref(NULL, E_WARNING, "Assertion failed");
263 }
264 } else {
265 zend_string *str = zval_get_string(description);
266 if (myeval) {
267 php_error_docref(NULL, E_WARNING, "%s: \"%s\" failed", ZSTR_VAL(str), myeval);
268 } else {
269 php_error_docref(NULL, E_WARNING, "%s failed", ZSTR_VAL(str));
270 }
271 zend_string_release(str);
272 }
273 }
274
275 if (ASSERTG(bail)) {
276 zend_bailout();
277 }
278
279 RETURN_FALSE;
280 }
281 /* }}} */
282
283 /* {{{ proto mixed assert_options(int what [, mixed value])
284 Set/get the various assert flags */
PHP_FUNCTION(assert_options)285 PHP_FUNCTION(assert_options)
286 {
287 zval *value = NULL;
288 zend_long what;
289 zend_bool oldint;
290 int ac = ZEND_NUM_ARGS();
291 zend_string *key;
292
293 if (zend_parse_parameters(ac, "l|z", &what, &value) == FAILURE) {
294 return;
295 }
296
297 switch (what) {
298 case ASSERT_ACTIVE:
299 oldint = ASSERTG(active);
300 if (ac == 2) {
301 zend_string *value_str = zval_get_string(value);
302 key = zend_string_init("assert.active", sizeof("assert.active")-1, 0);
303 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
304 zend_string_release(key);
305 zend_string_release(value_str);
306 }
307 RETURN_LONG(oldint);
308 break;
309
310 case ASSERT_BAIL:
311 oldint = ASSERTG(bail);
312 if (ac == 2) {
313 zend_string *value_str = zval_get_string(value);
314 key = zend_string_init("assert.bail", sizeof("assert.bail")-1, 0);
315 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
316 zend_string_release(key);
317 zend_string_release(value_str);
318 }
319 RETURN_LONG(oldint);
320 break;
321
322 case ASSERT_QUIET_EVAL:
323 oldint = ASSERTG(quiet_eval);
324 if (ac == 2) {
325 zend_string *value_str = zval_get_string(value);
326 key = zend_string_init("assert.quiet_eval", sizeof("assert.quiet_eval")-1, 0);
327 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
328 zend_string_release(key);
329 zend_string_release(value_str);
330 }
331 RETURN_LONG(oldint);
332 break;
333
334 case ASSERT_WARNING:
335 oldint = ASSERTG(warning);
336 if (ac == 2) {
337 zend_string *value_str = zval_get_string(value);
338 key = zend_string_init("assert.warning", sizeof("assert.warning")-1, 0);
339 zend_alter_ini_entry_ex(key, value_str, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
340 zend_string_release(key);
341 zend_string_release(value_str);
342 }
343 RETURN_LONG(oldint);
344 break;
345
346 case ASSERT_CALLBACK:
347 if (Z_TYPE(ASSERTG(callback)) != IS_UNDEF) {
348 ZVAL_COPY(return_value, &ASSERTG(callback));
349 } else if (ASSERTG(cb)) {
350 RETVAL_STRING(ASSERTG(cb));
351 } else {
352 RETVAL_NULL();
353 }
354 if (ac == 2) {
355 zval_ptr_dtor(&ASSERTG(callback));
356 ZVAL_COPY(&ASSERTG(callback), value);
357 }
358 return;
359
360 case ASSERT_EXCEPTION:
361 oldint = ASSERTG(exception);
362 if (ac == 2) {
363 zend_string *key = zend_string_init("assert.exception", sizeof("assert.exception")-1, 0);
364 zend_string *val = zval_get_string(value);
365 zend_alter_ini_entry_ex(key, val, PHP_INI_USER, PHP_INI_STAGE_RUNTIME, 0);
366 zend_string_release(val);
367 zend_string_release(key);
368 }
369 RETURN_LONG(oldint);
370 break;
371
372 default:
373 php_error_docref(NULL, E_WARNING, "Unknown value " ZEND_LONG_FMT, what);
374 break;
375 }
376
377 RETURN_FALSE;
378 }
379 /* }}} */
380
381 /*
382 * Local variables:
383 * tab-width: 4
384 * c-basic-offset: 4
385 * End:
386 * vim600: sw=4 ts=4 fdm=marker
387 * vim<600: sw=4 ts=4
388 */
389
