Coverage Report

Created: 2022-10-14 11:23

/src/php-src/Zend/zend_weakrefs.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
   +----------------------------------------------------------------------+
3
   | Copyright (c) The PHP Group                                          |
4
   +----------------------------------------------------------------------+
5
   | This source file is subject to version 2.00 of the Zend 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.zend.com/license/2_00.txt.                                |
9
   | If you did not receive a copy of the Zend license and are unable to  |
10
   | obtain it through the world-wide-web, please send a note to          |
11
   | license@zend.com so we can mail you a copy immediately.              |
12
   +----------------------------------------------------------------------+
13
   | Authors: krakjoe@php.net                                             |
14
   +----------------------------------------------------------------------+
15
*/
16
17
#include "zend.h"
18
#include "zend_interfaces.h"
19
#include "zend_objects_API.h"
20
#include "zend_weakrefs.h"
21
#include "zend_weakrefs_arginfo.h"
22
23
typedef struct _zend_weakref {
24
  zend_object *referent;
25
  zend_object std;
26
} zend_weakref;
27
28
typedef struct _zend_weakmap {
29
  HashTable ht;
30
  zend_object std;
31
} zend_weakmap;
32
33
typedef struct _zend_weakmap_iterator {
34
  zend_object_iterator it;
35
  uint32_t ht_iter;
36
} zend_weakmap_iterator;
37
38
/* The EG(weakrefs) ht is a map from object address a tagged pointer, that may be one of
39
 * zend_weakref*, zend_weakmap* or HashTable*. */
40
0
#define ZEND_WEAKREF_TAG_REF 0
41
#define ZEND_WEAKREF_TAG_MAP 1
42
0
#define ZEND_WEAKREF_TAG_HT  2
43
0
#define ZEND_WEAKREF_GET_TAG(p) (((uintptr_t) (p)) & 3)
44
0
#define ZEND_WEAKREF_GET_PTR(p) ((void *) (((uintptr_t) (p)) & ~3))
45
0
#define ZEND_WEAKREF_ENCODE(p, t) ((void *) (((uintptr_t) (p)) | (t)))
46
47
zend_class_entry *zend_ce_weakref;
48
zend_class_entry *zend_ce_weakmap;
49
static zend_object_handlers zend_weakref_handlers;
50
static zend_object_handlers zend_weakmap_handlers;
51
52
0
#define zend_weakref_from(o) ((zend_weakref*)(((char*) o) - XtOffsetOf(zend_weakref, std)))
53
0
#define zend_weakref_fetch(z) zend_weakref_from(Z_OBJ_P(z))
54
55
0
#define zend_weakmap_from(o) ((zend_weakmap*)(((char*) o) - XtOffsetOf(zend_weakmap, std)))
56
0
#define zend_weakmap_fetch(z) zend_weakmap_from(Z_OBJ_P(z))
57
58
static inline void zend_weakref_unref_single(
59
    void *ptr, uintptr_t tag, zend_ulong obj_addr)
60
0
{
61
0
  if (tag == ZEND_WEAKREF_TAG_REF) {
62
0
    zend_weakref *wr = ptr;
63
0
    wr->referent = NULL;
64
0
  } else {
65
0
    ZEND_ASSERT(tag == ZEND_WEAKREF_TAG_MAP);
66
0
    zend_weakmap *wm = ptr;
67
0
    zend_hash_index_del(&wm->ht, obj_addr);
68
0
  }
69
0
}
70
71
0
static void zend_weakref_unref(zend_ulong obj_addr, void *tagged_ptr) {
72
0
  void *ptr = ZEND_WEAKREF_GET_PTR(tagged_ptr);
73
0
  uintptr_t tag = ZEND_WEAKREF_GET_TAG(tagged_ptr);
74
0
  if (tag == ZEND_WEAKREF_TAG_HT) {
75
0
    HashTable *ht = ptr;
76
0
    ZEND_HASH_FOREACH_PTR(ht, tagged_ptr) {
77
0
      zend_weakref_unref_single(
78
0
        ZEND_WEAKREF_GET_PTR(tagged_ptr), ZEND_WEAKREF_GET_TAG(tagged_ptr), obj_addr);
79
0
    } ZEND_HASH_FOREACH_END();
80
0
    zend_hash_destroy(ht);
81
0
    FREE_HASHTABLE(ht);
82
0
  } else {
83
0
    zend_weakref_unref_single(ptr, tag, obj_addr);
84
0
  }
85
0
}
86
87
0
static void zend_weakref_register(zend_object *object, void *payload) {
88
0
  GC_ADD_FLAGS(object, IS_OBJ_WEAKLY_REFERENCED);
89
90
0
  zend_ulong obj_addr = (zend_ulong) object;
91
0
  zval *zv = zend_hash_index_find(&EG(weakrefs), obj_addr);
92
0
  if (!zv) {
93
0
    zend_hash_index_add_new_ptr(&EG(weakrefs), obj_addr, payload);
94
0
    return;
95
0
  }
96
97
0
  void *tagged_ptr = Z_PTR_P(zv);
98
0
  if (ZEND_WEAKREF_GET_TAG(tagged_ptr) == ZEND_WEAKREF_TAG_HT) {
99
0
    HashTable *ht = ZEND_WEAKREF_GET_PTR(tagged_ptr);
100
0
    zend_hash_index_add_new_ptr(ht, (zend_ulong) payload, payload);
101
0
    return;
102
0
  }
103
104
  /* Convert simple pointer to hashtable. */
105
0
  HashTable *ht = emalloc(sizeof(HashTable));
106
0
  zend_hash_init(ht, 0, NULL, NULL, 0);
107
0
  zend_hash_index_add_new_ptr(ht, (zend_ulong) tagged_ptr, tagged_ptr);
108
0
  zend_hash_index_add_new_ptr(ht, (zend_ulong) payload, payload);
109
0
  zend_hash_index_update_ptr(
110
0
    &EG(weakrefs), obj_addr, ZEND_WEAKREF_ENCODE(ht, ZEND_WEAKREF_TAG_HT));
111
0
}
112
113
0
static void zend_weakref_unregister(zend_object *object, void *payload) {
114
0
  zend_ulong obj_addr = (zend_ulong) object;
115
0
  void *tagged_ptr = zend_hash_index_find_ptr(&EG(weakrefs), obj_addr);
116
0
  ZEND_ASSERT(tagged_ptr && "Weakref not registered?");
117
118
0
  void *ptr = ZEND_WEAKREF_GET_PTR(tagged_ptr);
119
0
  uintptr_t tag = ZEND_WEAKREF_GET_TAG(tagged_ptr);
120
0
  if (tag != ZEND_WEAKREF_TAG_HT) {
121
0
    ZEND_ASSERT(tagged_ptr == payload);
122
0
    zend_weakref_unref_single(ptr, tag, obj_addr);
123
0
    zend_hash_index_del(&EG(weakrefs), obj_addr);
124
0
    GC_DEL_FLAGS(object, IS_OBJ_WEAKLY_REFERENCED);
125
0
    return;
126
0
  }
127
128
0
  HashTable *ht = ptr;
129
0
  tagged_ptr = zend_hash_index_find_ptr(ht, (zend_ulong) payload);
130
0
  ZEND_ASSERT(tagged_ptr && "Weakref not registered?");
131
0
  ZEND_ASSERT(tagged_ptr == payload);
132
0
  zend_weakref_unref_single(
133
0
    ZEND_WEAKREF_GET_PTR(payload), ZEND_WEAKREF_GET_TAG(payload), obj_addr);
134
0
  zend_hash_index_del(ht, (zend_ulong) payload);
135
0
  if (zend_hash_num_elements(ht) == 0) {
136
0
    GC_DEL_FLAGS(object, IS_OBJ_WEAKLY_REFERENCED);
137
0
    zend_hash_destroy(ht);
138
0
    FREE_HASHTABLE(ht);
139
0
    zend_hash_index_del(&EG(weakrefs), obj_addr);
140
0
  }
141
0
}
142
143
390k
void zend_weakrefs_init() {
144
390k
  zend_hash_init(&EG(weakrefs), 8, NULL, NULL, 0);
145
390k
}
146
147
0
void zend_weakrefs_notify(zend_object *object) {
148
  /* Annoyingly we can't use the HT destructor here, because we need access to the key (which
149
   * is the object address), which is not provided to the dtor. */
150
0
  zend_ulong obj_addr = (zend_ulong) object;
151
0
  void *tagged_ptr = zend_hash_index_find_ptr(&EG(weakrefs), obj_addr);
152
#if ZEND_DEBUG
153
  ZEND_ASSERT(tagged_ptr && "Tracking of the IS_OBJ_WEAKLY_REFERENCE flag should be precise");
154
#endif
155
0
  if (tagged_ptr) {
156
0
    zend_weakref_unref(obj_addr, tagged_ptr);
157
0
    zend_hash_index_del(&EG(weakrefs), obj_addr);
158
0
  }
159
0
}
160
161
390k
void zend_weakrefs_shutdown() {
162
390k
  zend_ulong obj_addr;
163
390k
  void *tagged_ptr;
164
390k
  ZEND_HASH_FOREACH_NUM_KEY_PTR(&EG(weakrefs), obj_addr, tagged_ptr) {
165
390k
    zend_weakref_unref(obj_addr, tagged_ptr);
166
0
  } ZEND_HASH_FOREACH_END();
167
390k
  zend_hash_destroy(&EG(weakrefs));
168
390k
}
169
170
0
static zend_object* zend_weakref_new(zend_class_entry *ce) {
171
0
  zend_weakref *wr = zend_object_alloc(sizeof(zend_weakref), zend_ce_weakref);
172
173
0
  zend_object_std_init(&wr->std, zend_ce_weakref);
174
175
0
  wr->std.handlers = &zend_weakref_handlers;
176
177
0
  return &wr->std;
178
0
}
179
180
0
static zend_always_inline zend_bool zend_weakref_find(zval *referent, zval *return_value) {
181
0
  void *tagged_ptr = zend_hash_index_find_ptr(&EG(weakrefs), (zend_ulong) Z_OBJ_P(referent));
182
0
  if (!tagged_ptr) {
183
0
    return 0;
184
0
  }
185
186
0
  void *ptr = ZEND_WEAKREF_GET_PTR(tagged_ptr);
187
0
  uintptr_t tag = ZEND_WEAKREF_GET_TAG(tagged_ptr);
188
0
  if (tag == ZEND_WEAKREF_TAG_REF) {
189
0
    zend_weakref *wr;
190
0
found_weakref:
191
0
    wr = ptr;
192
0
    RETVAL_OBJ_COPY(&wr->std);
193
0
    return 1;
194
0
  }
195
196
0
  if (tag == ZEND_WEAKREF_TAG_HT) {
197
0
    ZEND_HASH_FOREACH(ptr, tagged_ptr) {
198
0
      if (ZEND_WEAKREF_GET_TAG(tagged_ptr) == ZEND_WEAKREF_TAG_REF) {
199
0
        ptr = ZEND_WEAKREF_GET_PTR(tagged_ptr);
200
0
        goto found_weakref;
201
0
      }
202
0
    } ZEND_HASH_FOREACH_END();
203
0
  }
204
205
0
  return 0;
206
0
}
207
208
0
static zend_always_inline void zend_weakref_create(zval *referent, zval *return_value) {
209
0
  zend_weakref *wr;
210
211
0
  object_init_ex(return_value, zend_ce_weakref);
212
213
0
  wr = zend_weakref_fetch(return_value);
214
0
  wr->referent = Z_OBJ_P(referent);
215
216
0
  zend_weakref_register(wr->referent, ZEND_WEAKREF_ENCODE(wr, ZEND_WEAKREF_TAG_REF));
217
0
}
218
219
0
static zend_always_inline void zend_weakref_get(zval *weakref, zval *return_value) {
220
0
  zend_weakref *wr = zend_weakref_fetch(weakref);
221
222
0
  if (wr->referent) {
223
0
    RETVAL_OBJ_COPY(wr->referent);
224
0
  }
225
0
}
226
227
0
static void zend_weakref_free(zend_object *zo) {
228
0
  zend_weakref *wr = zend_weakref_from(zo);
229
230
0
  if (wr->referent) {
231
0
    zend_weakref_unregister(wr->referent, ZEND_WEAKREF_ENCODE(wr, ZEND_WEAKREF_TAG_REF));
232
0
  }
233
234
0
  zend_object_std_dtor(&wr->std);
235
0
}
236
237
ZEND_COLD ZEND_METHOD(WeakReference, __construct)
238
0
{
239
0
  zend_throw_error(NULL,
240
0
      "Direct instantiation of 'WeakReference' is not allowed, "
241
0
      "use WeakReference::create instead");
242
0
}
243
244
ZEND_METHOD(WeakReference, create)
245
0
{
246
0
  zval *referent;
247
248
0
  ZEND_PARSE_PARAMETERS_START(1,1)
249
0
    Z_PARAM_OBJECT(referent)
250
0
  ZEND_PARSE_PARAMETERS_END();
251
252
0
  if (zend_weakref_find(referent, return_value)) {
253
0
      return;
254
0
  }
255
256
0
  zend_weakref_create(referent, return_value);
257
0
}
258
259
ZEND_METHOD(WeakReference, get)
260
0
{
261
0
  ZEND_PARSE_PARAMETERS_NONE();
262
263
0
  zend_weakref_get(getThis(), return_value);
264
0
}
265
266
static zend_object *zend_weakmap_create_object(zend_class_entry *ce)
267
0
{
268
0
  zend_weakmap *wm = zend_object_alloc(sizeof(zend_weakmap), ce);
269
0
  zend_object_std_init(&wm->std, ce);
270
0
  wm->std.handlers = &zend_weakmap_handlers;
271
272
0
  zend_hash_init(&wm->ht, 0, NULL, ZVAL_PTR_DTOR, 0);
273
0
  return &wm->std;
274
0
}
275
276
static void zend_weakmap_free_obj(zend_object *object)
277
0
{
278
0
  zend_weakmap *wm = zend_weakmap_from(object);
279
0
  zend_ulong obj_addr;
280
0
  ZEND_HASH_FOREACH_NUM_KEY(&wm->ht, obj_addr) {
281
0
    zend_weakref_unregister(
282
0
      (zend_object *) obj_addr, ZEND_WEAKREF_ENCODE(wm, ZEND_WEAKREF_TAG_MAP));
283
0
  } ZEND_HASH_FOREACH_END();
284
0
  zend_hash_destroy(&wm->ht);
285
0
  zend_object_std_dtor(&wm->std);
286
0
}
287
288
static zval *zend_weakmap_read_dimension(zend_object *object, zval *offset, int type, zval *rv)
289
0
{
290
0
  if (offset == NULL) {
291
0
    zend_throw_error(NULL, "Cannot append to WeakMap");
292
0
    return NULL;
293
0
  }
294
295
0
  if (Z_TYPE_P(offset) != IS_OBJECT) {
296
0
    zend_type_error("WeakMap key must be an object");
297
0
    return NULL;
298
0
  }
299
300
0
  zend_weakmap *wm = zend_weakmap_from(object);
301
0
  zend_object *obj_key = Z_OBJ_P(offset);
302
0
  zval *zv = zend_hash_index_find(&wm->ht, (zend_ulong) obj_key);
303
0
  if (zv == NULL) {
304
0
    if (type != BP_VAR_IS) {
305
0
      zend_throw_error(NULL,
306
0
        "Object %s#%d not contained in WeakMap", ZSTR_VAL(obj_key->ce->name), obj_key->handle);
307
0
      return NULL;
308
0
    }
309
0
    return NULL;
310
0
  }
311
312
0
  if (type == BP_VAR_W || type == BP_VAR_RW) {
313
0
    ZVAL_MAKE_REF(zv);
314
0
  }
315
0
  return zv;
316
0
}
317
318
static void zend_weakmap_write_dimension(zend_object *object, zval *offset, zval *value)
319
0
{
320
0
  if (offset == NULL) {
321
0
    zend_throw_error(NULL, "Cannot append to WeakMap");
322
0
    return;
323
0
  }
324
325
0
  if (Z_TYPE_P(offset) != IS_OBJECT) {
326
0
    zend_type_error("WeakMap key must be an object");
327
0
    return;
328
0
  }
329
330
0
  zend_weakmap *wm = zend_weakmap_from(object);
331
0
  zend_object *obj_key = Z_OBJ_P(offset);
332
0
  Z_TRY_ADDREF_P(value);
333
334
0
  zval *zv = zend_hash_index_find(&wm->ht, (zend_ulong) obj_key);
335
0
  if (zv) {
336
0
    zval_ptr_dtor(zv);
337
0
    ZVAL_COPY_VALUE(zv, value);
338
0
    return;
339
0
  }
340
341
0
  zend_weakref_register(obj_key, ZEND_WEAKREF_ENCODE(wm, ZEND_WEAKREF_TAG_MAP));
342
0
  zend_hash_index_add_new(&wm->ht, (zend_ulong) obj_key, value);
343
0
}
344
345
static int zend_weakmap_has_dimension(zend_object *object, zval *offset, int check_empty)
346
0
{
347
0
  if (Z_TYPE_P(offset) != IS_OBJECT) {
348
0
    zend_type_error("WeakMap key must be an object");
349
0
    return 0;
350
0
  }
351
352
0
  zend_weakmap *wm = zend_weakmap_from(object);
353
0
  zval *zv = zend_hash_index_find(&wm->ht, (zend_ulong) Z_OBJ_P(offset));
354
0
  if (!zv) {
355
0
    return 0;
356
0
  }
357
358
0
  if (check_empty) {
359
0
    return i_zend_is_true(zv);
360
0
  }
361
0
  return Z_TYPE_P(zv) != IS_NULL;
362
0
}
363
364
static void zend_weakmap_unset_dimension(zend_object *object, zval *offset)
365
0
{
366
0
  if (Z_TYPE_P(offset) != IS_OBJECT) {
367
0
    zend_type_error("WeakMap key must be an object");
368
0
    return;
369
0
  }
370
371
0
  zend_weakmap *wm = zend_weakmap_from(object);
372
0
  zend_object *obj_key = Z_OBJ_P(offset);
373
0
  zend_weakref_unregister(obj_key, ZEND_WEAKREF_ENCODE(wm, ZEND_WEAKREF_TAG_MAP));
374
0
}
375
376
static int zend_weakmap_count_elements(zend_object *object, zend_long *count)
377
0
{
378
0
  zend_weakmap *wm = zend_weakmap_from(object);
379
0
  *count = zend_hash_num_elements(&wm->ht);
380
0
  return SUCCESS;
381
0
}
382
383
static HashTable *zend_weakmap_get_properties_for(zend_object *object, zend_prop_purpose purpose)
384
0
{
385
0
  if (purpose != ZEND_PROP_PURPOSE_DEBUG) {
386
0
    return NULL;
387
0
  }
388
389
0
  zend_weakmap *wm = zend_weakmap_from(object);
390
0
  HashTable *ht;
391
0
  ALLOC_HASHTABLE(ht);
392
0
  zend_hash_init(ht, zend_hash_num_elements(&wm->ht), NULL, ZVAL_PTR_DTOR, 0);
393
394
0
  zend_ulong obj_addr;
395
0
  zval *val;
396
0
  ZEND_HASH_FOREACH_NUM_KEY_VAL(&wm->ht, obj_addr, val) {
397
0
    zval pair;
398
0
    zval obj_zv;
399
0
    array_init(&pair);
400
401
0
    ZVAL_OBJ_COPY(&obj_zv, (zend_object *) obj_addr);
402
0
    add_assoc_zval(&pair, "key", &obj_zv);
403
0
    Z_TRY_ADDREF_P(val);
404
0
    add_assoc_zval(&pair, "value", val);
405
406
0
    zend_hash_next_index_insert(ht, &pair);
407
0
  } ZEND_HASH_FOREACH_END();
408
409
0
  return ht;
410
0
}
411
412
static HashTable *zend_weakmap_get_gc(zend_object *object, zval **table, int *n)
413
0
{
414
0
  zend_weakmap *wm = zend_weakmap_from(object);
415
0
  *table = NULL;
416
0
  *n = 0;
417
0
  return &wm->ht;
418
0
}
419
420
static zend_object *zend_weakmap_clone_obj(zend_object *old_object)
421
0
{
422
0
  zend_object *new_object = zend_weakmap_create_object(zend_ce_weakmap);
423
0
  zend_weakmap *old_wm = zend_weakmap_from(old_object);
424
0
  zend_weakmap *new_wm = zend_weakmap_from(new_object);
425
0
  zend_hash_copy(&new_wm->ht, &old_wm->ht, NULL);
426
427
0
  zend_ulong obj_addr;
428
0
  zval *val;
429
0
  ZEND_HASH_FOREACH_NUM_KEY_VAL(&new_wm->ht, obj_addr, val) {
430
0
    zend_weakref_register(
431
0
      (zend_object *) obj_addr, ZEND_WEAKREF_ENCODE(new_wm, ZEND_WEAKREF_TAG_MAP));
432
0
    zval_add_ref(val);
433
0
  } ZEND_HASH_FOREACH_END();
434
0
  return new_object;
435
0
}
436
437
0
static HashPosition *zend_weakmap_iterator_get_pos_ptr(zend_weakmap_iterator *iter) {
438
0
  ZEND_ASSERT(iter->ht_iter != (uint32_t) -1);
439
0
  return &EG(ht_iterators)[iter->ht_iter].pos;
440
0
}
441
442
static void zend_weakmap_iterator_dtor(zend_object_iterator *obj_iter)
443
0
{
444
0
  zend_weakmap_iterator *iter = (zend_weakmap_iterator *) obj_iter;
445
0
  zend_hash_iterator_del(iter->ht_iter);
446
0
  zval_ptr_dtor(&iter->it.data);
447
0
}
448
449
static int zend_weakmap_iterator_valid(zend_object_iterator *obj_iter)
450
0
{
451
0
  zend_weakmap_iterator *iter = (zend_weakmap_iterator *) obj_iter;
452
0
  zend_weakmap *wm = zend_weakmap_fetch(&iter->it.data);
453
0
  HashPosition *pos = zend_weakmap_iterator_get_pos_ptr(iter);
454
0
  return zend_hash_has_more_elements_ex(&wm->ht, pos);
455
0
}
456
457
static zval *zend_weakmap_iterator_get_current_data(zend_object_iterator *obj_iter)
458
0
{
459
0
  zend_weakmap_iterator *iter = (zend_weakmap_iterator *) obj_iter;
460
0
  zend_weakmap *wm = zend_weakmap_fetch(&iter->it.data);
461
0
  HashPosition *pos = zend_weakmap_iterator_get_pos_ptr(iter);
462
0
  return zend_hash_get_current_data_ex(&wm->ht, pos);
463
0
}
464
465
static void zend_weakmap_iterator_get_current_key(zend_object_iterator *obj_iter, zval *key)
466
0
{
467
0
  zend_weakmap_iterator *iter = (zend_weakmap_iterator *) obj_iter;
468
0
  zend_weakmap *wm = zend_weakmap_fetch(&iter->it.data);
469
0
  HashPosition *pos = zend_weakmap_iterator_get_pos_ptr(iter);
470
471
0
  zend_string *string_key;
472
0
  zend_ulong num_key;
473
0
  int key_type = zend_hash_get_current_key_ex(&wm->ht, &string_key, &num_key, pos);
474
0
  if (key_type != HASH_KEY_IS_LONG) {
475
0
    ZEND_ASSERT(0 && "Must have integer key");
476
0
  }
477
478
0
  ZVAL_OBJ_COPY(key, (zend_object *) num_key);
479
0
}
480
481
static void zend_weakmap_iterator_move_forward(zend_object_iterator *obj_iter)
482
0
{
483
0
  zend_weakmap_iterator *iter = (zend_weakmap_iterator *) obj_iter;
484
0
  zend_weakmap *wm = zend_weakmap_fetch(&iter->it.data);
485
0
  HashPosition *pos = zend_weakmap_iterator_get_pos_ptr(iter);
486
0
  zend_hash_move_forward_ex(&wm->ht, pos);
487
0
}
488
489
static void zend_weakmap_iterator_rewind(zend_object_iterator *obj_iter)
490
0
{
491
0
  zend_weakmap_iterator *iter = (zend_weakmap_iterator *) obj_iter;
492
0
  zend_weakmap *wm = zend_weakmap_fetch(&iter->it.data);
493
0
  HashPosition *pos = zend_weakmap_iterator_get_pos_ptr(iter);
494
0
  zend_hash_internal_pointer_reset_ex(&wm->ht, pos);
495
0
}
496
497
static const zend_object_iterator_funcs zend_weakmap_iterator_funcs = {
498
  zend_weakmap_iterator_dtor,
499
  zend_weakmap_iterator_valid,
500
  zend_weakmap_iterator_get_current_data,
501
  zend_weakmap_iterator_get_current_key,
502
  zend_weakmap_iterator_move_forward,
503
  zend_weakmap_iterator_rewind,
504
  NULL
505
};
506
507
static zend_object_iterator *zend_weakmap_get_iterator(
508
    zend_class_entry *ce, zval *object, int by_ref)
509
0
{
510
0
  zend_weakmap *wm = zend_weakmap_fetch(object);
511
0
  zend_weakmap_iterator *iter = emalloc(sizeof(zend_weakmap_iterator));
512
0
  zend_iterator_init(&iter->it);
513
0
  iter->it.funcs = &zend_weakmap_iterator_funcs;
514
0
  ZVAL_COPY(&iter->it.data, object);
515
0
  iter->ht_iter = zend_hash_iterator_add(&wm->ht, 0);
516
0
  return &iter->it;
517
0
}
518
519
ZEND_METHOD(WeakMap, offsetGet)
520
0
{
521
0
  zval *key;
522
523
0
  if (zend_parse_parameters(ZEND_NUM_ARGS(), "z", &key) == FAILURE) {
524
0
    return;
525
0
  }
526
527
0
  zval *zv = zend_weakmap_read_dimension(Z_OBJ_P(ZEND_THIS), key, BP_VAR_R, NULL);
528
0
  if (!zv) {
529
0
    return;
530
0
  }
531
532
0
  ZVAL_COPY(return_value, zv);
533
0
}
534
535
ZEND_METHOD(WeakMap, offsetSet)
536
0
{
537
0
  zval *key, *value;
538
539
0
  if (zend_parse_parameters(ZEND_NUM_ARGS(), "zz", &key, &value) == FAILURE) {
540
0
    return;
541
0
  }
542
543
0
  zend_weakmap_write_dimension(Z_OBJ_P(ZEND_THIS), key, value);
544
0
}
545
546
ZEND_METHOD(WeakMap, offsetExists)
547
0
{
548
0
  zval *key;
549
550
0
  if (zend_parse_parameters(ZEND_NUM_ARGS(), "z", &key) == FAILURE) {
551
0
    return;
552
0
  }
553
554
0
  RETURN_BOOL(zend_weakmap_has_dimension(Z_OBJ_P(ZEND_THIS), key, /* check_empty */ 0));
555
0
}
556
557
ZEND_METHOD(WeakMap, offsetUnset)
558
0
{
559
0
  zval *key;
560
561
0
  if (zend_parse_parameters(ZEND_NUM_ARGS(), "z", &key) == FAILURE) {
562
0
    return;
563
0
  }
564
565
0
  zend_weakmap_unset_dimension(Z_OBJ_P(ZEND_THIS), key);
566
0
}
567
568
ZEND_METHOD(WeakMap, count)
569
0
{
570
0
  if (zend_parse_parameters_none() == FAILURE) {
571
0
    return;
572
0
  }
573
574
0
  zend_long count;
575
0
  zend_weakmap_count_elements(Z_OBJ_P(ZEND_THIS), &count);
576
0
  RETURN_LONG(count);
577
0
}
578
579
ZEND_METHOD(WeakMap, getIterator)
580
0
{
581
0
  if (zend_parse_parameters_none() == FAILURE) {
582
0
    return;
583
0
  }
584
585
0
  zend_create_internal_iterator_zval(return_value, ZEND_THIS);
586
0
}
587
588
void zend_register_weakref_ce(void) /* {{{ */
589
3.47k
{
590
3.47k
  zend_class_entry ce;
591
592
3.47k
  INIT_CLASS_ENTRY(ce, "WeakReference", class_WeakReference_methods);
593
3.47k
  zend_ce_weakref = zend_register_internal_class(&ce);
594
3.47k
  zend_ce_weakref->ce_flags |= ZEND_ACC_FINAL | ZEND_ACC_NO_DYNAMIC_PROPERTIES;
595
596
3.47k
  zend_ce_weakref->create_object = zend_weakref_new;
597
3.47k
  zend_ce_weakref->serialize = zend_class_serialize_deny;
598
3.47k
  zend_ce_weakref->unserialize = zend_class_unserialize_deny;
599
600
3.47k
  memcpy(&zend_weakref_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
601
3.47k
  zend_weakref_handlers.offset = XtOffsetOf(zend_weakref, std);
602
603
3.47k
  zend_weakref_handlers.free_obj = zend_weakref_free;
604
3.47k
  zend_weakref_handlers.clone_obj = NULL;
605
606
3.47k
  INIT_CLASS_ENTRY(ce, "WeakMap", class_WeakMap_methods);
607
3.47k
  zend_ce_weakmap = zend_register_internal_class(&ce);
608
3.47k
  zend_ce_weakmap->ce_flags |= ZEND_ACC_FINAL | ZEND_ACC_NO_DYNAMIC_PROPERTIES;
609
3.47k
  zend_class_implements(
610
3.47k
    zend_ce_weakmap, 3, zend_ce_arrayaccess, zend_ce_countable, zend_ce_aggregate);
611
612
3.47k
  zend_ce_weakmap->create_object = zend_weakmap_create_object;
613
3.47k
  zend_ce_weakmap->get_iterator = zend_weakmap_get_iterator;
614
3.47k
  zend_ce_weakmap->serialize = zend_class_serialize_deny;
615
3.47k
  zend_ce_weakmap->unserialize = zend_class_unserialize_deny;
616
617
3.47k
  memcpy(&zend_weakmap_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
618
3.47k
  zend_weakmap_handlers.offset = XtOffsetOf(zend_weakmap, std);
619
3.47k
  zend_weakmap_handlers.free_obj = zend_weakmap_free_obj;
620
3.47k
  zend_weakmap_handlers.read_dimension = zend_weakmap_read_dimension;
621
3.47k
  zend_weakmap_handlers.write_dimension = zend_weakmap_write_dimension;
622
3.47k
  zend_weakmap_handlers.has_dimension = zend_weakmap_has_dimension;
623
3.47k
  zend_weakmap_handlers.unset_dimension = zend_weakmap_unset_dimension;
624
3.47k
  zend_weakmap_handlers.count_elements = zend_weakmap_count_elements;
625
3.47k
  zend_weakmap_handlers.get_properties_for = zend_weakmap_get_properties_for;
626
3.47k
  zend_weakmap_handlers.get_gc = zend_weakmap_get_gc;
627
3.47k
  zend_weakmap_handlers.clone_obj = zend_weakmap_clone_obj;
628
3.47k
}
629
/* }}} */
630