diff options
Diffstat (limited to 'Zend/zend_weakrefs.c')
| -rw-r--r-- | Zend/zend_weakrefs.c | 213 |
1 files changed, 213 insertions, 0 deletions
diff --git a/Zend/zend_weakrefs.c b/Zend/zend_weakrefs.c new file mode 100644 index 0000000000..b4eaf437c8 --- /dev/null +++ b/Zend/zend_weakrefs.c @@ -0,0 +1,213 @@ +/* + +----------------------------------------------------------------------+ + | PHP Version 7 | + +----------------------------------------------------------------------+ + | Copyright (c) The PHP Group | + +----------------------------------------------------------------------+ + | This source file is subject to version 2.00 of the Zend license, | + | that is bundled with this package in the file LICENSE, and is | + | available through the world-wide-web at the following url: | + | http://www.zend.com/license/2_00.txt. | + | If you did not receive a copy of the Zend license and are unable to | + | obtain it through the world-wide-web, please send a note to | + | license@zend.com so we can mail you a copy immediately. | + +----------------------------------------------------------------------+ + | Authors: krakjoe@php.net | + +----------------------------------------------------------------------+ +*/ + +#include "zend.h" +#include "zend_interfaces.h" +#include "zend_objects_API.h" +#include "zend_weakrefs.h" + +typedef struct _zend_weakref { + zend_object *referent; + zend_object std; +} zend_weakref; + +zend_class_entry *zend_ce_weakref; +static zend_object_handlers zend_weakref_handlers; + +#define zend_weakref_from(o) ((zend_weakref*)(((char*) o) - XtOffsetOf(zend_weakref, std))) +#define zend_weakref_fetch(z) zend_weakref_from(Z_OBJ_P(z)) + +static void zend_weakref_unref(zval *zv) { + zend_weakref *wr = (zend_weakref*) Z_PTR_P(zv); + + GC_DEL_FLAGS(wr->referent, IS_OBJ_WEAKLY_REFERENCED); + + wr->referent = NULL; +} + +void zend_weakrefs_init() { + zend_hash_init(&EG(weakrefs), 8, NULL, zend_weakref_unref, 0); +} + +void zend_weakrefs_notify(zend_object *object) { + zend_hash_index_del(&EG(weakrefs), (zend_ulong) object); +} + +void zend_weakrefs_shutdown() { + zend_hash_destroy(&EG(weakrefs)); +} + +static zend_object* zend_weakref_new(zend_class_entry *ce) { + zend_weakref *wr = zend_object_alloc(sizeof(zend_weakref), zend_ce_weakref); + + zend_object_std_init(&wr->std, zend_ce_weakref); + + wr->std.handlers = &zend_weakref_handlers; + + return &wr->std; +} + +static zend_always_inline zend_bool zend_weakref_find(zval *referent, zval *return_value) { + zend_weakref *wr = zend_hash_index_find_ptr(&EG(weakrefs), (zend_ulong) Z_OBJ_P(referent)); + + if (!wr) { + return 0; + } + + GC_ADDREF(&wr->std); + ZVAL_OBJ(return_value, &wr->std); + + return 1; +} + +static zend_always_inline void zend_weakref_create(zval *referent, zval *return_value) { + zend_weakref *wr; + + object_init_ex(return_value, zend_ce_weakref); + + wr = zend_weakref_fetch(return_value); + + wr->referent = Z_OBJ_P(referent); + + zend_hash_index_add_ptr(&EG(weakrefs), (zend_ulong) wr->referent, wr); + + GC_ADD_FLAGS(wr->referent, IS_OBJ_WEAKLY_REFERENCED); +} + +static zend_always_inline void zend_weakref_get(zval *weakref, zval *return_value) { + zend_weakref *wr = zend_weakref_fetch(weakref); + + if (wr->referent) { + ZVAL_OBJ(return_value, wr->referent); + Z_ADDREF_P(return_value); + } +} + +static void zend_weakref_free(zend_object *zo) { + zend_weakref *wr = zend_weakref_from(zo); + + if (wr->referent) { + zend_hash_index_del( + &EG(weakrefs), (zend_ulong) wr->referent); + } + + zend_object_std_dtor(&wr->std); +} + +#define zend_weakref_unsupported(thing) \ + zend_throw_error(NULL, "WeakReference objects do not support " thing); + +static zval* zend_weakref_no_write(zval *object, zval *member, zval *value, void **rtc) { + zend_weakref_unsupported("properties"); + + return &EG(uninitialized_zval); +} + +static zval* zend_weakref_no_read(zval *object, zval *member, int type, void **rtc, zval *rv) { + if (!EG(exception)) { + zend_weakref_unsupported("properties"); + } + + return &EG(uninitialized_zval); +} + +static zval *zend_weakref_no_read_ptr(zval *object, zval *member, int type, void **rtc) { + zend_weakref_unsupported("property references"); + return NULL; +} + +static int zend_weakref_no_isset(zval *object, zval *member, int hse, void **rtc) { + if (hse != 2) { + zend_weakref_unsupported("properties"); + } + return 0; +} + +static void zend_weakref_no_unset(zval *object, zval *member, void **rtc) { + zend_weakref_unsupported("properties"); +} + +ZEND_BEGIN_ARG_WITH_RETURN_OBJ_INFO_EX(zend_weakref_create_arginfo, 0, 1, WeakReference, 0) + ZEND_ARG_TYPE_INFO(0, referent, IS_OBJECT, 0) +ZEND_END_ARG_INFO() + +ZEND_BEGIN_ARG_WITH_RETURN_TYPE_INFO_EX(zend_weakref_get_arginfo, 0, 0, IS_OBJECT, 1) +ZEND_END_ARG_INFO() + +ZEND_COLD ZEND_METHOD(WeakReference, __construct) +{ + zend_throw_error(NULL, + "Direct instantiation of 'WeakReference' is not allowed, " + "use WeakReference::create instead"); +} + +ZEND_METHOD(WeakReference, create) +{ + zval *referent; + + ZEND_PARSE_PARAMETERS_START_EX(ZEND_PARSE_PARAMS_THROW, 1,1) + Z_PARAM_OBJECT(referent) + ZEND_PARSE_PARAMETERS_END(); + + if (zend_weakref_find(referent, return_value)) { + return; + } + + zend_weakref_create(referent, return_value); +} + +ZEND_METHOD(WeakReference, get) +{ + ZEND_PARSE_PARAMETERS_START_EX(ZEND_PARSE_PARAMS_THROW, 0, 0) + ZEND_PARSE_PARAMETERS_END(); + + zend_weakref_get(getThis(), return_value); +} + +static const zend_function_entry zend_weakref_methods[] = { + ZEND_ME(WeakReference, __construct, NULL, ZEND_ACC_PUBLIC) + ZEND_ME(WeakReference, create, zend_weakref_create_arginfo, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC) + ZEND_ME(WeakReference, get, zend_weakref_get_arginfo, ZEND_ACC_PUBLIC) + ZEND_FE_END +}; + +void zend_register_weakref_ce(void) /* {{{ */ +{ + zend_class_entry ce; + + INIT_CLASS_ENTRY(ce, "WeakReference", zend_weakref_methods); + zend_ce_weakref = zend_register_internal_class(&ce); + zend_ce_weakref->ce_flags |= ZEND_ACC_FINAL; + + zend_ce_weakref->create_object = zend_weakref_new; + zend_ce_weakref->serialize = zend_class_serialize_deny; + zend_ce_weakref->unserialize = zend_class_unserialize_deny; + + memcpy(&zend_weakref_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers)); + zend_weakref_handlers.offset = XtOffsetOf(zend_weakref, std); + + zend_weakref_handlers.free_obj = zend_weakref_free; + zend_weakref_handlers.read_property = zend_weakref_no_read; + zend_weakref_handlers.write_property = zend_weakref_no_write; + zend_weakref_handlers.has_property = zend_weakref_no_isset; + zend_weakref_handlers.unset_property = zend_weakref_no_unset; + zend_weakref_handlers.get_property_ptr_ptr = zend_weakref_no_read_ptr; + zend_weakref_handlers.clone_obj = NULL; +} +/* }}} */ + |