git.m6w6.name Git - m6w6/ext-pq/blob - src/php_pq_object.c

   1 /*
   2     +--------------------------------------------------------------------+
   3     | PECL :: pq                                                         |
   4     +--------------------------------------------------------------------+
   5     | Redistribution and use in source and binary forms, with or without |
   6     | modification, are permitted provided that the conditions mentioned |
   7     | in the accompanying LICENSE file are met.                          |
   8     +--------------------------------------------------------------------+
   9     | Copyright (c) 2013, Michael Wallner <mike@php.net>                 |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 #ifdef HAVE_CONFIG_H
  14 #       include "config.h"
  15 #endif
  16
  17 #include <php.h>
  18
  19 #include "php_pq_object.h"
  20 #include "php_pq_misc.h"
  21
  22 void *php_pq_object_create(zend_class_entry *ce, void *intern, size_t obj_size, zend_object_handlers *oh, HashTable *ph)
  23 {
  24         php_pq_object_t *o = ecalloc(1, obj_size + zend_object_properties_size(ce));
  25
  26         zend_object_std_init(&o->zo, ce);
  27         object_properties_init(&o->zo, ce);
  28         o->zo.handlers = oh;
  29         o->intern = intern;
  30         o->prophandler = ph;
  31
  32         zend_hash_init(&o->gc, 0, NULL, NULL, 0);
  33
  34         return o;
  35 }
  36
  37 void php_pq_object_dtor(zend_object *o)
  38 {
  39         php_pq_object_t *obj = PHP_PQ_OBJ(NULL, o);
  40
  41         zend_hash_destroy(&obj->gc);
  42         zend_object_std_dtor(o);
  43 }
  44
  45 void php_pq_object_to_zval(void *o, zval *zv)
  46 {
  47         php_pq_object_t *obj = o;
  48
  49         ZVAL_OBJ(zv, &obj->zo);
  50         Z_ADDREF_P(zv);
  51 }
  52
  53 void php_pq_object_to_zval_no_addref(void *o, zval *zv)
  54 {
  55         php_pq_object_t *obj = o;
  56
  57         ZVAL_OBJ(zv, &obj->zo);
  58 }
  59
  60 void php_pq_object_addref(void *o)
  61 {
  62         php_pq_object_t *obj = o;
  63 #ifdef GC_ADDREF
  64         GC_ADDREF(&obj->zo);
  65 #else
  66         ++GC_REFCOUNT(&obj->zo);
  67 #endif
  68 }
  69
  70 void php_pq_object_delref(void *o)
  71 {
  72         php_pq_object_t *obj = o;
  73         zval tmp;
  74
  75         /* this should gc immediately */
  76         ZVAL_OBJ(&tmp, &obj->zo);
  77         zval_ptr_dtor(&tmp);
  78 }
  79
  80 struct apply_pi_to_ht_arg {
  81         HashTable *ht;
  82         php_pq_object_t *pq_obj;
  83         unsigned gc:1;
  84 };
  85
  86 static int apply_pi_to_ht(zval *p, void *a)
  87 {
  88         zend_property_info *pi = Z_PTR_P(p);
  89         struct apply_pi_to_ht_arg *arg = a;
  90
  91         if (arg->gc) {
  92                 php_pq_object_prophandler_t *handler;
  93
  94                 if ((handler = zend_hash_find_ptr(arg->pq_obj->prophandler, pi->name)) && handler->gc) {
  95                         zval return_value;
  96
  97                         ZVAL_ARR(&return_value, arg->ht);
  98                         handler->gc(arg->pq_obj, &return_value);
  99                 }
 100         } else {
 101                 zval tmp_prop, *property;
 102 #if PHP_VERSION_ID < 80000
 103                 zval zobj;
 104
 105                 ZVAL_OBJ(&zobj, &arg->pq_obj->zo);
 106                 property = zend_read_property_ex(arg->pq_obj->zo.ce, &zobj, pi->name, 0, &tmp_prop);
 107 #else
 108                 property = zend_read_property_ex(arg->pq_obj->zo.ce, &arg->pq_obj->zo, pi->name, 0, &tmp_prop);
 109 #endif
 110                 zend_hash_update(arg->ht, pi->name, property);
 111         }
 112
 113         return ZEND_HASH_APPLY_KEEP;
 114 }
 115
 116 static inline HashTable *php_pq_object_debug_info_ex(zend_object *object, int *temp)
 117 {
 118         struct apply_pi_to_ht_arg arg = {NULL};
 119
 120         *temp = 1;
 121         ALLOC_HASHTABLE(arg.ht);
 122         ZEND_INIT_SYMTABLE(arg.ht);
 123
 124         arg.pq_obj = PHP_PQ_OBJ(NULL, object);
 125         arg.gc = 0;
 126
 127         zend_hash_apply_with_argument(&arg.pq_obj->zo.ce->properties_info, apply_pi_to_ht, &arg);
 128
 129         return arg.ht;
 130 }
 131 #if PHP_VERSION_ID >= 80000
 132 HashTable *php_pq_object_debug_info(zend_object *object, int *temp)
 133 {
 134         return php_pq_object_debug_info_ex(object, temp);
 135 }
 136 #else
 137 HashTable *php_pq_object_debug_info(zval *object, int *temp)
 138 {
 139         return php_pq_object_debug_info_ex(Z_OBJ_P(object), temp);
 140 }
 141 #endif
 142
 143 static inline HashTable *php_pq_object_properties_ex(zend_object *object, HashTable *props)
 144 {
 145         struct apply_pi_to_ht_arg arg = {NULL};
 146
 147         arg.ht = props;
 148         arg.pq_obj = PHP_PQ_OBJ(NULL, object);
 149         arg.gc = 0;
 150
 151         zend_hash_apply_with_argument(&arg.pq_obj->zo.ce->properties_info, apply_pi_to_ht, &arg);
 152
 153         return arg.ht;
 154 }
 155 #if PHP_VERSION_ID >= 80000
 156 HashTable *php_pq_object_properties(zend_object *object)
 157 {
 158         return php_pq_object_properties_ex(object, zend_std_get_properties(object));
 159 }
 160 #else
 161 HashTable *php_pq_object_properties(zval *object)
 162 {
 163         return php_pq_object_properties_ex(Z_OBJ_P(object), zend_std_get_properties(object));
 164 }
 165 #endif
 166
 167 static inline HashTable *php_pq_object_get_gc_ex(zend_object *object, HashTable *props, zval **table, int *n)
 168 {
 169         struct apply_pi_to_ht_arg arg = {NULL};
 170
 171         arg.pq_obj = PHP_PQ_OBJ(NULL, object);
 172         arg.ht = &arg.pq_obj->gc;
 173         arg.gc = 1;
 174
 175         zend_hash_clean(arg.ht);
 176         zend_hash_copy(arg.ht, props, NULL);
 177         zend_hash_apply_with_argument(&arg.pq_obj->zo.ce->properties_info, apply_pi_to_ht, &arg);
 178
 179         *table = NULL;
 180         *n = 0;
 181
 182         return arg.ht;
 183 }
 184 #if PHP_VERSION_ID >= 80000
 185 HashTable *php_pq_object_get_gc(zend_object *object, zval **table, int *n)
 186 {
 187         return php_pq_object_get_gc_ex(object, zend_std_get_properties(object), table, n);
 188 }
 189 #else
 190 HashTable *php_pq_object_get_gc(zval *object, zval **table, int *n)
 191 {
 192         return php_pq_object_get_gc_ex(Z_OBJ_P(object), zend_std_get_properties(object), table, n);
 193 }
 194 #endif
 195
 196 zend_class_entry *ancestor(zend_class_entry *ce)
 197 {
 198         while (ce->parent) {
 199                 ce = ce->parent;
 200         }
 201         return ce;
 202 }
 203
 204 static inline int php_pq_object_read_prop_ex(zend_object *object, zend_string *member, int type, zval *return_value)
 205 {
 206         php_pq_object_t *obj = PHP_PQ_OBJ(NULL, object);
 207         php_pq_object_prophandler_t *handler;
 208
 209         if (!obj->intern) {
 210                 php_error(E_RECOVERABLE_ERROR, "%s not initialized", ancestor(obj->zo.ce)->name->val);
 211         } else if (!(handler = zend_hash_find_ptr(obj->prophandler, member)) || !handler->read) {
 212                 /* default handler */
 213         } else if (type != BP_VAR_R) {
 214                 php_error(E_WARNING, "Cannot access %s properties by reference or array key/index", ancestor(obj->zo.ce)->name->val);
 215         } else {
 216                 handler->read(obj, return_value);
 217                 return SUCCESS;
 218         }
 219
 220         return FAILURE;
 221 }
 222 #if PHP_VERSION_ID >= 80000
 223 zval *php_pq_object_read_prop(zend_object *object, zend_string *member, int type, void **cache_slot, zval *tmp)
 224 {
 225         if (SUCCESS != php_pq_object_read_prop_ex(object, member, type, tmp)) {
 226                 return zend_std_read_property(object, member, type, cache_slot, tmp);
 227         }
 228
 229         zend_std_write_property(object, member, tmp, cache_slot);
 230
 231         if (cache_slot) {
 232                 *cache_slot = NULL;
 233         }
 234         return tmp;
 235 }
 236 #else
 237 zval *php_pq_object_read_prop(zval *object, zval *member, int type, void **cache_slot, zval *tmp)
 238 {
 239         zend_string *member_str = zval_get_string(member);
 240
 241         if (SUCCESS != php_pq_object_read_prop_ex(Z_OBJ_P(object), member_str, type, tmp)) {
 242                 zend_string_release(member_str);
 243                 return zend_std_read_property(object, member, type, cache_slot, tmp);
 244         }
 245         zend_string_release(member_str);
 246
 247         zend_std_write_property(object, member, tmp, cache_slot);
 248
 249         if (cache_slot) {
 250                 *cache_slot = NULL;
 251         }
 252         return tmp;
 253 }
 254 #endif
 255
 256 static inline int php_pq_object_write_prop_ex(zend_object *object, zend_string *member, zval *value)
 257 {
 258         php_pq_object_t *obj = PHP_PQ_OBJ(NULL, object);
 259         php_pq_object_prophandler_t *handler;
 260
 261         if (!obj->intern) {
 262                 php_error(E_RECOVERABLE_ERROR, "%s not initialized", ancestor(obj->zo.ce)->name->val);
 263         } else if (!(handler = zend_hash_find_ptr(obj->prophandler, member))) {
 264                 /* default handler */
 265         } else {
 266                 if (handler->write) {
 267                         handler->write(obj, value);
 268                 }
 269                 return SUCCESS;
 270         }
 271         return FAILURE;
 272 }
 273 #if PHP_VERSION_ID >= 80000
 274 zval *php_pq_object_write_prop(zend_object *object, zend_string *member, zval *value, void **cache_slot)
 275 {
 276         if (SUCCESS != php_pq_object_write_prop_ex(object, member, value)) {
 277                 return zend_std_write_property(object, member, value, cache_slot);
 278         }
 279         return value;
 280 }
 281 #elif PHP_VERSION_ID >= 70400
 282 zval *php_pq_object_write_prop(zval *object, zval *member, zval *value, void **cache_slot)
 283 {
 284         zend_string *member_str = zval_get_string(member);
 285         if (SUCCESS != php_pq_object_write_prop_ex(Z_OBJ_P(object), member_str, value)) {
 286                 value = zend_std_write_property(object, member, value, cache_slot);
 287         }
 288         zend_string_release(member_str);
 289         return value;
 290 }
 291 #else
 292 void php_pq_object_write_prop(zval *object, zval *member, zval *value, void **cache_slot)
 293 {
 294         zend_string *member_str = zval_get_string(member);
 295         if (SUCCESS != php_pq_object_write_prop_ex(Z_OBJ_P(object), member_str, value)) {
 296                 zend_std_write_property(object, member, value, cache_slot);
 297         }
 298         zend_string_release(member_str);
 299 }
 300 #endif
 301
 302 #if PHP_VERSION_ID >= 80000
 303 zval *php_pq_object_get_prop_ptr_null(zend_object *object, zend_string *member, int type, void **cache_slot)
 304 {
 305         return NULL;
 306 }
 307 #else
 308 zval *php_pq_object_get_prop_ptr_null(zval *object, zval *member, int type, void **cache_slot)
 309 {
 310         return NULL;
 311 }
 312 #endif
 313
 314 void php_pq_object_prophandler_dtor(zval *zv) {
 315         pefree(Z_PTR_P(zv), 1);
 316 }
 317