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 # if PHP_VERSION_ID < 70100
 107                 property = zend_read_property(arg->pq_obj->zo.ce, &zobj, pi->name->val, pi->name->len, 0, &tmp_prop);
 108 # else
 109                 property = zend_read_property_ex(arg->pq_obj->zo.ce, &zobj, pi->name, 0, &tmp_prop);
 110 # endif
 111 #else
 112                 property = zend_read_property_ex(arg->pq_obj->zo.ce, &arg->pq_obj->zo, pi->name, 0, &tmp_prop);
 113 #endif
 114                 zend_hash_update(arg->ht, pi->name, property);
 115         }
 116
 117         return ZEND_HASH_APPLY_KEEP;
 118 }
 119
 120 static inline HashTable *php_pq_object_debug_info_ex(zend_object *object, int *temp)
 121 {
 122         struct apply_pi_to_ht_arg arg = {NULL};
 123
 124         *temp = 1;
 125         ALLOC_HASHTABLE(arg.ht);
 126         ZEND_INIT_SYMTABLE(arg.ht);
 127
 128         arg.pq_obj = PHP_PQ_OBJ(NULL, object);
 129         arg.gc = 0;
 130
 131         zend_hash_apply_with_argument(&arg.pq_obj->zo.ce->properties_info, apply_pi_to_ht, &arg);
 132
 133         return arg.ht;
 134 }
 135 #if PHP_VERSION_ID >= 80000
 136 HashTable *php_pq_object_debug_info(zend_object *object, int *temp)
 137 {
 138         return php_pq_object_debug_info_ex(object, temp);
 139 }
 140 #else
 141 HashTable *php_pq_object_debug_info(zval *object, int *temp)
 142 {
 143         return php_pq_object_debug_info_ex(Z_OBJ_P(object), temp);
 144 }
 145 #endif
 146
 147 static inline HashTable *php_pq_object_properties_ex(zend_object *object, HashTable *props)
 148 {
 149         struct apply_pi_to_ht_arg arg = {NULL};
 150
 151         arg.ht = props;
 152         arg.pq_obj = PHP_PQ_OBJ(NULL, object);
 153         arg.gc = 0;
 154
 155         zend_hash_apply_with_argument(&arg.pq_obj->zo.ce->properties_info, apply_pi_to_ht, &arg);
 156
 157         return arg.ht;
 158 }
 159 #if PHP_VERSION_ID >= 80000
 160 HashTable *php_pq_object_properties(zend_object *object)
 161 {
 162         return php_pq_object_properties_ex(object, zend_std_get_properties(object));
 163 }
 164 #else
 165 HashTable *php_pq_object_properties(zval *object)
 166 {
 167         return php_pq_object_properties_ex(Z_OBJ_P(object), zend_std_get_properties(object));
 168 }
 169 #endif
 170
 171 static inline HashTable *php_pq_object_get_gc_ex(zend_object *object, HashTable *props, zval **table, int *n)
 172 {
 173         struct apply_pi_to_ht_arg arg = {NULL};
 174
 175         arg.pq_obj = PHP_PQ_OBJ(NULL, object);
 176         arg.ht = &arg.pq_obj->gc;
 177         arg.gc = 1;
 178
 179         zend_hash_clean(arg.ht);
 180         zend_hash_copy(arg.ht, props, NULL);
 181         zend_hash_apply_with_argument(&arg.pq_obj->zo.ce->properties_info, apply_pi_to_ht, &arg);
 182
 183         *table = NULL;
 184         *n = 0;
 185
 186         return arg.ht;
 187 }
 188 #if PHP_VERSION_ID >= 80000
 189 HashTable *php_pq_object_get_gc(zend_object *object, zval **table, int *n)
 190 {
 191         return php_pq_object_get_gc_ex(object, zend_std_get_properties(object), table, n);
 192 }
 193 #else
 194 HashTable *php_pq_object_get_gc(zval *object, zval **table, int *n)
 195 {
 196         return php_pq_object_get_gc_ex(Z_OBJ_P(object), zend_std_get_properties(object), table, n);
 197 }
 198 #endif
 199
 200 zend_class_entry *ancestor(zend_class_entry *ce)
 201 {
 202         while (ce->parent) {
 203                 ce = ce->parent;
 204         }
 205         return ce;
 206 }
 207
 208 static inline int php_pq_object_read_prop_ex(zend_object *object, zend_string *member, int type, zval *return_value)
 209 {
 210         php_pq_object_t *obj = PHP_PQ_OBJ(NULL, object);
 211         php_pq_object_prophandler_t *handler;
 212
 213         if (!obj->intern) {
 214                 php_error(E_RECOVERABLE_ERROR, "%s not initialized", ancestor(obj->zo.ce)->name->val);
 215         } else if (!(handler = zend_hash_find_ptr(obj->prophandler, member)) || !handler->read) {
 216                 /* default handler */
 217         } else if (type != BP_VAR_R) {
 218                 php_error(E_WARNING, "Cannot access %s properties by reference or array key/index", ancestor(obj->zo.ce)->name->val);
 219         } else {
 220                 handler->read(obj, return_value);
 221                 return SUCCESS;
 222         }
 223
 224         return FAILURE;
 225 }
 226 #if PHP_VERSION_ID >= 80000
 227 zval *php_pq_object_read_prop(zend_object *object, zend_string *member, int type, void **cache_slot, zval *tmp)
 228 {
 229         if (SUCCESS != php_pq_object_read_prop_ex(object, member, type, tmp)) {
 230                 return zend_std_read_property(object, member, type, cache_slot, tmp);
 231         }
 232
 233         zend_std_write_property(object, member, tmp, cache_slot);
 234
 235         if (cache_slot) {
 236                 *cache_slot = NULL;
 237         }
 238         return tmp;
 239 }
 240 #else
 241 zval *php_pq_object_read_prop(zval *object, zval *member, int type, void **cache_slot, zval *tmp)
 242 {
 243         zend_string *member_str = zval_get_string(member);
 244
 245         if (SUCCESS != php_pq_object_read_prop_ex(Z_OBJ_P(object), member_str, type, tmp)) {
 246                 zend_string_release(member_str);
 247                 return zend_get_std_object_handlers()->read_property(object, member, type, cache_slot, tmp);
 248         }
 249         zend_string_release(member_str);
 250
 251         zend_std_write_property(object, member, tmp, cache_slot);
 252
 253         if (cache_slot) {
 254                 *cache_slot = NULL;
 255         }
 256         return tmp;
 257 }
 258 #endif
 259
 260 static inline int php_pq_object_write_prop_ex(zend_object *object, zend_string *member, zval *value)
 261 {
 262         php_pq_object_t *obj = PHP_PQ_OBJ(NULL, object);
 263         php_pq_object_prophandler_t *handler;
 264
 265         if (!obj->intern) {
 266                 php_error(E_RECOVERABLE_ERROR, "%s not initialized", ancestor(obj->zo.ce)->name->val);
 267         } else if (!(handler = zend_hash_find_ptr(obj->prophandler, member))) {
 268                 /* default handler */
 269         } else {
 270                 if (handler->write) {
 271                         handler->write(obj, value);
 272                 }
 273                 return SUCCESS;
 274         }
 275         return FAILURE;
 276 }
 277 #if PHP_VERSION_ID >= 80000
 278 zval *php_pq_object_write_prop(zend_object *object, zend_string *member, zval *value, void **cache_slot)
 279 {
 280         if (SUCCESS != php_pq_object_write_prop_ex(object, member, value)) {
 281                 return zend_std_write_property(object, member, value, cache_slot);
 282         }
 283         return value;
 284 }
 285 #elif PHP_VERSION_ID >= 70400
 286 zval *php_pq_object_write_prop(zval *object, zval *member, zval *value, void **cache_slot)
 287 {
 288         zend_string *member_str = zval_get_string(member);
 289         if (SUCCESS != php_pq_object_write_prop_ex(Z_OBJ_P(object), member_str, value)) {
 290                 value = zend_std_write_property(object, member, value, cache_slot);
 291         }
 292         zend_string_release(member_str);
 293         return value;
 294 }
 295 #else
 296 void php_pq_object_write_prop(zval *object, zval *member, zval *value, void **cache_slot)
 297 {
 298         zend_string *member_str = zval_get_string(member);
 299         if (SUCCESS != php_pq_object_write_prop_ex(Z_OBJ_P(object), member_str, value)) {
 300                 zend_std_write_property(object, member, value, cache_slot);
 301         }
 302         zend_string_release(member_str);
 303 }
 304 #endif
 305
 306 #if PHP_VERSION_ID >= 80000
 307 zval *php_pq_object_get_prop_ptr_null(zend_object *object, zend_string *member, int type, void **cache_slot)
 308 {
 309         return NULL;
 310 }
 311 #else
 312 zval *php_pq_object_get_prop_ptr_null(zval *object, zval *member, int type, void **cache_slot)
 313 {
 314         return NULL;
 315 }
 316 #endif
 317
 318 void php_pq_object_prophandler_dtor(zval *zv) {
 319         pefree(Z_PTR_P(zv), 1);
 320 }
 321