Coverage Report

Created: 2026-07-25 06:39

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/src/php-src/Zend/zend_interfaces.c
Line
Count
Source
1
/*
2
   +----------------------------------------------------------------------+
3
   | Zend Engine                                                          |
4
   +----------------------------------------------------------------------+
5
   | Copyright © Zend Technologies Ltd., a subsidiary company of          |
6
   |     Perforce Software, Inc., and Contributors.                       |
7
   +----------------------------------------------------------------------+
8
   | This source file is subject to the Modified BSD License that is      |
9
   | bundled with this package in the file LICENSE, and is available      |
10
   | through the World Wide Web at <https://www.php.net/license/>.        |
11
   |                                                                      |
12
   | SPDX-License-Identifier: BSD-3-Clause                                |
13
   +----------------------------------------------------------------------+
14
   | Authors: Marcus Boerger <helly@php.net>                              |
15
   +----------------------------------------------------------------------+
16
*/
17
18
#include "zend.h"
19
#include "zend_API.h"
20
#include "zend_interfaces.h"
21
#include "zend_exceptions.h"
22
#include "zend_interfaces_arginfo.h"
23
#include "zend_property_hooks.h"
24
25
ZEND_API zend_class_entry *zend_ce_traversable;
26
ZEND_API zend_class_entry *zend_ce_aggregate;
27
ZEND_API zend_class_entry *zend_ce_iterator;
28
ZEND_API zend_class_entry *zend_ce_arrayaccess;
29
ZEND_API zend_class_entry *zend_ce_serializable;
30
ZEND_API zend_class_entry *zend_ce_countable;
31
ZEND_API zend_class_entry *zend_ce_stringable;
32
ZEND_API zend_class_entry *zend_ce_internal_iterator;
33
34
static zend_object_handlers zend_internal_iterator_handlers;
35
36
/* {{{ zend_call_method
37
 Only returns the returned zval if retval_ptr != NULL */
38
ZEND_API zval* zend_call_method(zend_object *object, zend_class_entry *obj_ce, zend_function **fn_proxy, const char *function_name, size_t function_name_len, zval *retval_ptr, uint32_t param_count, zval* arg1, zval* arg2)
39
155k
{
40
155k
  zend_function *fn;
41
155k
  zend_class_entry *called_scope;
42
155k
  zval params[2];
43
44
155k
  if (param_count > 0) {
45
154k
    ZVAL_COPY_VALUE(&params[0], arg1);
46
154k
  }
47
155k
  if (param_count > 1) {
48
0
    ZVAL_COPY_VALUE(&params[1], arg2);
49
0
  }
50
51
155k
  if (!obj_ce) {
52
0
    obj_ce = object ? object->ce : NULL;
53
0
  }
54
155k
  if (!fn_proxy || !*fn_proxy) {
55
155k
    if (EXPECTED(obj_ce)) {
56
155k
      fn = zend_hash_str_find_ptr_lc(
57
155k
        &obj_ce->function_table, function_name, function_name_len);
58
155k
      if (UNEXPECTED(fn == NULL)) {
59
        /* error at c-level */
60
0
        zend_error_noreturn(E_CORE_ERROR, "Couldn't find implementation for method %s::%s", ZSTR_VAL(obj_ce->name), function_name);
61
0
      }
62
155k
    } else {
63
0
      fn = zend_fetch_function_str(function_name, function_name_len);
64
0
      if (UNEXPECTED(fn == NULL)) {
65
        /* error at c-level */
66
0
        zend_error_noreturn(E_CORE_ERROR, "Couldn't find implementation for function %s", function_name);
67
0
      }
68
0
    }
69
155k
    if (fn_proxy) {
70
29
      *fn_proxy = fn;
71
29
    }
72
155k
  } else {
73
0
    fn = *fn_proxy;
74
0
  }
75
76
155k
  if (object) {
77
155k
    called_scope = object->ce;
78
155k
  } else {
79
0
    called_scope = obj_ce;
80
0
  }
81
82
155k
  zend_call_known_function(fn, object, called_scope, retval_ptr, param_count, params, NULL);
83
155k
  return retval_ptr;
84
155k
}
85
/* }}} */
86
87
/* iterator interface, c-level functions used by engine */
88
89
/* {{{ zend_user_it_new_iterator */
90
ZEND_API void zend_user_it_new_iterator(zend_class_entry *ce, zval *object, zval *retval)
91
131
{
92
131
  zend_call_known_instance_method_with_0_params(
93
131
    ce->iterator_funcs_ptr->zf_new_iterator, Z_OBJ_P(object), retval);
94
131
}
95
/* }}} */
96
97
/* {{{ zend_user_it_invalidate_current */
98
ZEND_API void zend_user_it_invalidate_current(zend_object_iterator *_iter)
99
1.65k
{
100
1.65k
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
101
102
1.65k
  if (!Z_ISUNDEF(iter->value)) {
103
413
    zval_ptr_dtor(&iter->value);
104
413
    ZVAL_UNDEF(&iter->value);
105
413
  }
106
1.65k
}
107
/* }}} */
108
109
/* {{{ zend_user_it_dtor */
110
static void zend_user_it_dtor(zend_object_iterator *_iter)
111
589
{
112
589
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
113
589
  zval *object = &iter->it.data;
114
115
589
  zend_user_it_invalidate_current(_iter);
116
589
  zval_ptr_dtor(object);
117
589
}
118
/* }}} */
119
120
/* {{{ zend_user_it_valid */
121
ZEND_API zend_result zend_user_it_valid(zend_object_iterator *_iter)
122
798
{
123
798
  if (_iter) {
124
798
    zend_user_iterator *iter = (zend_user_iterator*)_iter;
125
798
    zval *object = &iter->it.data;
126
798
    zval more;
127
128
798
    zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_valid, Z_OBJ_P(object), &more);
129
798
    bool result = i_zend_is_true(&more);
130
798
    zval_ptr_dtor(&more);
131
798
    return result ? SUCCESS : FAILURE;
132
798
  }
133
0
  return FAILURE;
134
798
}
135
/* }}} */
136
137
/* {{{ zend_user_it_get_current_data */
138
ZEND_API zval *zend_user_it_get_current_data(zend_object_iterator *_iter)
139
470
{
140
470
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
141
470
  zval *object = &iter->it.data;
142
143
470
  if (Z_ISUNDEF(iter->value)) {
144
470
    zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_current, Z_OBJ_P(object), &iter->value);
145
470
  }
146
470
  return &iter->value;
147
470
}
148
/* }}} */
149
150
/* {{{ zend_user_it_get_current_key */
151
ZEND_API void zend_user_it_get_current_key(zend_object_iterator *_iter, zval *key)
152
235
{
153
235
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
154
235
  zval *object = &iter->it.data;
155
235
  zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_key, Z_OBJ_P(object), key);
156
235
  if (UNEXPECTED(Z_ISREF_P(key))) {
157
8
    zend_unwrap_reference(key);
158
8
  }
159
235
}
160
/* }}} */
161
162
/* {{{ zend_user_it_move_forward */
163
ZEND_API void zend_user_it_move_forward(zend_object_iterator *_iter)
164
362
{
165
362
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
166
362
  zval *object = &iter->it.data;
167
168
362
  zend_user_it_invalidate_current(_iter);
169
362
  zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_next, Z_OBJ_P(object), NULL);
170
362
}
171
/* }}} */
172
173
/* {{{ zend_user_it_rewind */
174
ZEND_API void zend_user_it_rewind(zend_object_iterator *_iter)
175
596
{
176
596
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
177
596
  zval *object = &iter->it.data;
178
179
596
  zend_user_it_invalidate_current(_iter);
180
596
  zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_rewind, Z_OBJ_P(object), NULL);
181
596
}
182
/* }}} */
183
184
ZEND_API HashTable *zend_user_it_get_gc(zend_object_iterator *_iter, zval **table, int *n)
185
99
{
186
99
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
187
99
  if (Z_ISUNDEF(iter->value)) {
188
74
    *table = &iter->it.data;
189
74
    *n = 1;
190
74
  } else {
191
25
    zend_get_gc_buffer *gc_buffer = zend_get_gc_buffer_create();
192
25
    zend_get_gc_buffer_add_zval(gc_buffer, &iter->it.data);
193
25
    zend_get_gc_buffer_add_zval(gc_buffer, &iter->value);
194
25
    zend_get_gc_buffer_use(gc_buffer, table, n);
195
25
  }
196
99
  return NULL;
197
99
}
198
199
static const zend_object_iterator_funcs zend_interface_iterator_funcs_iterator = {
200
  zend_user_it_dtor,
201
  zend_user_it_valid,
202
  zend_user_it_get_current_data,
203
  zend_user_it_get_current_key,
204
  zend_user_it_move_forward,
205
  zend_user_it_rewind,
206
  zend_user_it_invalidate_current,
207
  zend_user_it_get_gc,
208
};
209
210
/* {{{ zend_user_it_get_iterator */
211
/* by_ref is int due to Iterator API */
212
static zend_object_iterator *zend_user_it_get_iterator(zend_class_entry *ce, zval *object, int by_ref)
213
589
{
214
589
  zend_user_iterator *iterator;
215
216
589
  if (by_ref) {
217
0
    zend_throw_error(NULL, "An iterator cannot be used with foreach by reference");
218
0
    return NULL;
219
0
  }
220
221
589
  iterator = emalloc(sizeof(zend_user_iterator));
222
223
589
  zend_iterator_init((zend_object_iterator*)iterator);
224
225
589
  ZVAL_OBJ_COPY(&iterator->it.data, Z_OBJ_P(object));
226
589
  iterator->it.funcs = &zend_interface_iterator_funcs_iterator;
227
589
  iterator->ce = Z_OBJCE_P(object);
228
589
  ZVAL_UNDEF(&iterator->value);
229
589
  return (zend_object_iterator*)iterator;
230
589
}
231
/* }}} */
232
233
/* {{{ zend_user_it_get_new_iterator */
234
/* by_ref is int due to Iterator API */
235
ZEND_API zend_object_iterator *zend_user_it_get_new_iterator(zend_class_entry *ce, zval *object, int by_ref)
236
131
{
237
131
  zval iterator;
238
131
  zend_object_iterator *new_iterator;
239
131
  zend_class_entry *ce_it;
240
241
131
  zend_user_it_new_iterator(ce, object, &iterator);
242
131
  ce_it = (Z_TYPE(iterator) == IS_OBJECT) ? Z_OBJCE(iterator) : NULL;
243
244
131
  if (!ce_it || !ce_it->get_iterator || (ce_it->get_iterator == zend_user_it_get_new_iterator && Z_OBJ(iterator) == Z_OBJ_P(object))) {
245
19
    if (!EG(exception)) {
246
5
      zend_throw_exception_ex(NULL, 0, "Objects returned by %s::getIterator() must be traversable or implement interface Iterator", ce ? ZSTR_VAL(ce->name) : ZSTR_VAL(Z_OBJCE_P(object)->name));
247
5
    }
248
19
    zval_ptr_dtor(&iterator);
249
19
    return NULL;
250
19
  }
251
252
112
  new_iterator = ce_it->get_iterator(ce_it, &iterator, by_ref);
253
112
  zval_ptr_dtor(&iterator);
254
112
  return new_iterator;
255
131
}
256
/* }}} */
257
258
/* {{{ zend_implement_traversable */
259
static int zend_implement_traversable(zend_class_entry *interface, zend_class_entry *class_type)
260
1.30k
{
261
  /* Abstract class can implement Traversable only, in which case the extending class must
262
   * implement Iterator or IteratorAggregate. */
263
1.30k
  if (class_type->ce_flags & ZEND_ACC_EXPLICIT_ABSTRACT_CLASS) {
264
75
    return SUCCESS;
265
75
  }
266
267
  /* Check that class_type implements at least one of 'IteratorAggregate' or 'Iterator' */
268
1.23k
  if (class_type->num_interfaces) {
269
1.23k
    ZEND_ASSERT(class_type->ce_flags & ZEND_ACC_RESOLVED_INTERFACES);
270
2.95k
    for (uint32_t i = 0; i < class_type->num_interfaces; i++) {
271
2.94k
      if (class_type->interfaces[i] == zend_ce_aggregate || class_type->interfaces[i] == zend_ce_iterator) {
272
1.22k
        return SUCCESS;
273
1.22k
      }
274
2.94k
    }
275
1.23k
  }
276
8
  zend_error_noreturn(E_CORE_ERROR, "%s %s must implement interface %s as part of either %s or %s",
277
8
    zend_get_object_type_uc(class_type),
278
8
    ZSTR_VAL(class_type->name),
279
8
    ZSTR_VAL(zend_ce_traversable->name),
280
8
    ZSTR_VAL(zend_ce_iterator->name),
281
8
    ZSTR_VAL(zend_ce_aggregate->name));
282
0
  return FAILURE;
283
1.23k
}
284
/* }}} */
285
286
/* {{{ zend_implement_aggregate */
287
static int zend_implement_aggregate(zend_class_entry *interface, zend_class_entry *class_type)
288
301
{
289
301
  if (zend_class_implements_interface(class_type, zend_ce_iterator)) {
290
4
    zend_error_noreturn(E_ERROR,
291
4
      "Class %s cannot implement both Iterator and IteratorAggregate at the same time",
292
4
      ZSTR_VAL(class_type->name));
293
4
  }
294
295
  /* Always initialize iterator_funcs_ptr. */
296
297
  ZEND_ASSERT(!class_type->iterator_funcs_ptr && "Iterator funcs already set?");
297
297
  zend_class_iterator_funcs *funcs_ptr = class_type->type == ZEND_INTERNAL_CLASS
298
297
    ? pemalloc(sizeof(zend_class_iterator_funcs), 1)
299
297
    : zend_arena_alloc(&CG(arena), sizeof(zend_class_iterator_funcs));
300
297
  class_type->iterator_funcs_ptr = funcs_ptr;
301
302
297
  memset(funcs_ptr, 0, sizeof(zend_class_iterator_funcs));
303
297
  funcs_ptr->zf_new_iterator = zend_hash_str_find_ptr(
304
297
    &class_type->function_table, "getiterator", sizeof("getiterator") - 1);
305
306
297
  if (class_type->get_iterator
307
110
   && class_type->get_iterator != zend_user_it_get_new_iterator
308
74
   && class_type->get_iterator != zend_hooked_object_get_iterator) {
309
    /* get_iterator was explicitly assigned for an internal class. */
310
62
    if (!class_type->parent || class_type->parent->get_iterator != class_type->get_iterator) {
311
0
      ZEND_ASSERT(class_type->type == ZEND_INTERNAL_CLASS);
312
0
      return SUCCESS;
313
0
    }
314
315
    /* The getIterator() method has not been overwritten, use inherited get_iterator(). */
316
62
    if (funcs_ptr->zf_new_iterator->common.scope != class_type) {
317
62
      return SUCCESS;
318
62
    }
319
320
    /* getIterator() has been overwritten, switch to zend_user_it_get_new_iterator. */
321
62
  }
322
323
235
  class_type->get_iterator = zend_user_it_get_new_iterator;
324
235
  return SUCCESS;
325
297
}
326
/* }}} */
327
328
/* {{{ zend_implement_iterator */
329
static int zend_implement_iterator(zend_class_entry *interface, zend_class_entry *class_type)
330
980
{
331
980
  if (zend_class_implements_interface(class_type, zend_ce_aggregate)) {
332
4
    zend_error_noreturn(E_ERROR,
333
4
      "Class %s cannot implement both Iterator and IteratorAggregate at the same time",
334
4
      ZSTR_VAL(class_type->name));
335
4
  }
336
337
976
  ZEND_ASSERT(!class_type->iterator_funcs_ptr && "Iterator funcs already set?");
338
976
  zend_class_iterator_funcs *funcs_ptr = class_type->type == ZEND_INTERNAL_CLASS
339
976
    ? pemalloc(sizeof(zend_class_iterator_funcs), 1)
340
976
    : zend_arena_alloc(&CG(arena), sizeof(zend_class_iterator_funcs));
341
976
  class_type->iterator_funcs_ptr = funcs_ptr;
342
343
976
  memset(funcs_ptr, 0, sizeof(zend_class_iterator_funcs));
344
976
  funcs_ptr->zf_rewind = zend_hash_str_find_ptr(
345
976
    &class_type->function_table, "rewind", sizeof("rewind") - 1);
346
976
  funcs_ptr->zf_valid = zend_hash_str_find_ptr(
347
976
    &class_type->function_table, "valid", sizeof("valid") - 1);
348
976
  funcs_ptr->zf_key = zend_hash_find_ptr(
349
976
    &class_type->function_table, ZSTR_KNOWN(ZEND_STR_KEY));
350
976
  funcs_ptr->zf_current = zend_hash_str_find_ptr(
351
976
    &class_type->function_table, "current", sizeof("current") - 1);
352
976
  funcs_ptr->zf_next = zend_hash_str_find_ptr(
353
976
    &class_type->function_table, "next", sizeof("next") - 1);
354
355
976
  if (class_type->get_iterator
356
697
   && class_type->get_iterator != zend_user_it_get_iterator
357
414
   && class_type->get_iterator != zend_hooked_object_get_iterator) {
358
397
    if (!class_type->parent || class_type->parent->get_iterator != class_type->get_iterator) {
359
      /* get_iterator was explicitly assigned for an internal class. */
360
0
      ZEND_ASSERT(class_type->type == ZEND_INTERNAL_CLASS);
361
0
      return SUCCESS;
362
0
    }
363
364
    /* None of the Iterator methods have been overwritten, use inherited get_iterator(). */
365
397
    if (funcs_ptr->zf_rewind->common.scope != class_type &&
366
381
        funcs_ptr->zf_valid->common.scope != class_type &&
367
381
        funcs_ptr->zf_key->common.scope != class_type &&
368
295
        funcs_ptr->zf_current->common.scope != class_type &&
369
287
        funcs_ptr->zf_next->common.scope != class_type) {
370
287
      return SUCCESS;
371
287
    }
372
373
    /* One of the Iterator methods has been overwritten,
374
     * switch to zend_user_it_get_iterator. */
375
397
  }
376
377
689
  class_type->get_iterator = zend_user_it_get_iterator;
378
689
  return SUCCESS;
379
976
}
380
/* }}} */
381
382
/* {{{ zend_implement_arrayaccess */
383
static int zend_implement_arrayaccess(zend_class_entry *interface, zend_class_entry *class_type)
384
854
{
385
854
  ZEND_ASSERT(!class_type->arrayaccess_funcs_ptr && "ArrayAccess funcs already set?");
386
854
  zend_class_arrayaccess_funcs *funcs_ptr = class_type->type == ZEND_INTERNAL_CLASS
387
854
    ? pemalloc(sizeof(zend_class_arrayaccess_funcs), 1)
388
854
    : zend_arena_alloc(&CG(arena), sizeof(zend_class_arrayaccess_funcs));
389
854
  class_type->arrayaccess_funcs_ptr = funcs_ptr;
390
391
854
  funcs_ptr->zf_offsetget = zend_hash_str_find_ptr(
392
854
    &class_type->function_table, "offsetget", sizeof("offsetget") - 1);
393
854
  funcs_ptr->zf_offsetexists = zend_hash_str_find_ptr(
394
854
    &class_type->function_table, "offsetexists", sizeof("offsetexists") - 1);
395
854
  funcs_ptr->zf_offsetset = zend_hash_str_find_ptr(
396
854
    &class_type->function_table, "offsetset", sizeof("offsetset") - 1);
397
854
  funcs_ptr->zf_offsetunset = zend_hash_str_find_ptr(
398
854
    &class_type->function_table, "offsetunset", sizeof("offsetunset") - 1);
399
400
854
  return SUCCESS;
401
854
}
402
/* }}} */
403
404
/* {{{ zend_user_serialize */
405
ZEND_API int zend_user_serialize(zval *object, unsigned char **buffer, size_t *buf_len, zend_serialize_data *data)
406
18
{
407
18
  zend_class_entry * ce = Z_OBJCE_P(object);
408
18
  zval retval;
409
18
  zend_result result;
410
411
18
  zend_call_method_with_0_params(
412
18
    Z_OBJ_P(object), Z_OBJCE_P(object), NULL, "serialize", &retval);
413
414
18
  if (Z_TYPE(retval) == IS_UNDEF) {
415
5
    result = FAILURE;
416
13
  } else {
417
13
    switch(Z_TYPE(retval)) {
418
0
    case IS_NULL:
419
      /* we could also make this '*buf_len = 0' but this allows to skip variables */
420
0
      return FAILURE;
421
13
    case IS_STRING:
422
13
      *buffer = (unsigned char*)estrndup(Z_STRVAL(retval), Z_STRLEN(retval));
423
13
      *buf_len = Z_STRLEN(retval);
424
13
      result = SUCCESS;
425
13
      break;
426
0
    default: /* failure */
427
0
      result = FAILURE;
428
0
      break;
429
13
    }
430
13
    zval_ptr_dtor(&retval);
431
13
  }
432
433
18
  if (result == FAILURE && !EG(exception)) {
434
0
    zend_throw_exception_ex(NULL, 0, "%s::serialize() must return a string or NULL", ZSTR_VAL(ce->name));
435
0
  }
436
18
  return result;
437
18
}
438
/* }}} */
439
440
/* {{{ zend_user_unserialize */
441
ZEND_API int zend_user_unserialize(zval *object, zend_class_entry *ce, const unsigned char *buf, size_t buf_len, zend_unserialize_data *data)
442
154k
{
443
154k
  zval zdata;
444
445
154k
  if (UNEXPECTED(object_init_ex(object, ce) != SUCCESS)) {
446
0
    return FAILURE;
447
0
  }
448
449
154k
  ZVAL_STRINGL(&zdata, (char*)buf, buf_len);
450
154k
  zend_call_method_with_1_params(
451
154k
    Z_OBJ_P(object), Z_OBJCE_P(object), NULL, "unserialize", NULL, &zdata);
452
154k
  zval_ptr_dtor(&zdata);
453
454
154k
  if (EG(exception)) {
455
214
    return FAILURE;
456
154k
  } else {
457
154k
    return SUCCESS;
458
154k
  }
459
154k
}
460
/* }}} */
461
462
/* {{{ zend_implement_serializable */
463
static int zend_implement_serializable(zend_class_entry *interface, zend_class_entry *class_type)
464
495
{
465
495
  if (class_type->parent
466
397
    && (class_type->parent->serialize || class_type->parent->unserialize)
467
397
    && !zend_class_implements_interface(class_type->parent, zend_ce_serializable)) {
468
0
    return FAILURE;
469
0
  }
470
495
  if (!class_type->serialize) {
471
98
    class_type->serialize = zend_user_serialize;
472
98
  }
473
495
  if (!class_type->unserialize) {
474
98
    class_type->unserialize = zend_user_unserialize;
475
98
  }
476
495
  if (!(class_type->ce_flags & ZEND_ACC_EXPLICIT_ABSTRACT_CLASS)
477
487
      && (!class_type->__serialize || !class_type->__unserialize)) {
478
36
    zend_error(E_DEPRECATED, "%s implements the Serializable interface, which is deprecated. Implement __serialize() and __unserialize() instead (or in addition, if support for old PHP versions is necessary)", ZSTR_VAL(class_type->name));
479
36
    if (EG(exception)) {
480
0
      zend_exception_uncaught_error(
481
0
        "During inheritance of %s, while implementing Serializable", ZSTR_VAL(class_type->name));
482
0
    }
483
36
  }
484
495
  return SUCCESS;
485
495
}
486
/* }}}*/
487
488
typedef struct {
489
  zend_object std;
490
  zend_object_iterator *iter;
491
  bool rewind_called;
492
} zend_internal_iterator;
493
494
31
static zend_object *zend_internal_iterator_create(zend_class_entry *ce) {
495
31
  zend_internal_iterator *intern = emalloc(sizeof(zend_internal_iterator));
496
31
  zend_object_std_init(&intern->std, ce);
497
31
  intern->iter = NULL;
498
31
  intern->rewind_called = false;
499
31
  return &intern->std;
500
31
}
501
502
22
ZEND_API zend_result zend_create_internal_iterator_zval(zval *return_value, zval *obj) {
503
22
  zend_class_entry *scope = EG(current_execute_data)->func->common.scope;
504
22
  ZEND_ASSERT(scope->get_iterator != zend_user_it_get_new_iterator);
505
22
  zend_object_iterator *iter = scope->get_iterator(Z_OBJCE_P(obj), obj, /* by_ref */ 0);
506
22
  if (!iter) {
507
0
    return FAILURE;
508
0
  }
509
510
22
  zend_internal_iterator *intern =
511
22
    (zend_internal_iterator *) zend_internal_iterator_create(zend_ce_internal_iterator);
512
22
  intern->iter = iter;
513
22
  intern->iter->index = 0;
514
22
  ZVAL_OBJ(return_value, &intern->std);
515
22
  return SUCCESS;
516
22
}
517
518
31
static void zend_internal_iterator_free(zend_object *obj) {
519
31
  zend_internal_iterator *intern = (zend_internal_iterator *) obj;
520
31
  if (intern->iter) {
521
22
    zend_iterator_dtor(intern->iter);
522
22
  }
523
31
  zend_object_std_dtor(&intern->std);
524
31
}
525
526
65
static zend_internal_iterator *zend_internal_iterator_fetch(zval *This) {
527
65
  zend_internal_iterator *intern = (zend_internal_iterator *) Z_OBJ_P(This);
528
65
  if (!intern->iter) {
529
0
    zend_throw_error(NULL, "The InternalIterator object has not been properly initialized");
530
0
    return NULL;
531
0
  }
532
65
  return intern;
533
65
}
534
535
/* Many iterators will not behave correctly if rewind() is not called, make sure it happens. */
536
65
static zend_result zend_internal_iterator_ensure_rewound(zend_internal_iterator *intern) {
537
65
  if (!intern->rewind_called) {
538
16
    zend_object_iterator *iter = intern->iter;
539
16
    intern->rewind_called = true;
540
16
    if (iter->funcs->rewind) {
541
16
      iter->funcs->rewind(iter);
542
16
      if (UNEXPECTED(EG(exception))) {
543
0
        return FAILURE;
544
0
      }
545
16
    }
546
16
  }
547
65
  return SUCCESS;
548
65
}
549
550
551
0
ZEND_METHOD(InternalIterator, __construct) {
552
0
  zend_throw_error(NULL, "Cannot manually construct InternalIterator");
553
0
}
554
555
21
ZEND_METHOD(InternalIterator, current) {
556
21
  ZEND_PARSE_PARAMETERS_NONE();
557
558
21
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
559
21
  if (!intern) {
560
0
    RETURN_THROWS();
561
0
  }
562
563
21
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
564
0
    RETURN_THROWS();
565
0
  }
566
567
21
  zval *data = intern->iter->funcs->get_current_data(intern->iter);
568
21
  if (data) {
569
5
    RETURN_COPY_DEREF(data);
570
5
  }
571
21
}
572
573
21
ZEND_METHOD(InternalIterator, key) {
574
21
  ZEND_PARSE_PARAMETERS_NONE();
575
576
21
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
577
21
  if (!intern) {
578
0
    RETURN_THROWS();
579
0
  }
580
581
21
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
582
0
    RETURN_THROWS();
583
0
  }
584
585
21
  if (intern->iter->funcs->get_current_key) {
586
21
    intern->iter->funcs->get_current_key(intern->iter, return_value);
587
21
  } else {
588
0
    RETURN_LONG(intern->iter->index);
589
0
  }
590
21
}
591
592
5
ZEND_METHOD(InternalIterator, next) {
593
5
  ZEND_PARSE_PARAMETERS_NONE();
594
595
5
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
596
5
  if (!intern) {
597
0
    RETURN_THROWS();
598
0
  }
599
600
5
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
601
0
    RETURN_THROWS();
602
0
  }
603
604
  /* Advance index first to match foreach behavior. */
605
5
  intern->iter->index++;
606
5
  intern->iter->funcs->move_forward(intern->iter);
607
5
}
608
609
18
ZEND_METHOD(InternalIterator, valid) {
610
18
  ZEND_PARSE_PARAMETERS_NONE();
611
612
18
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
613
18
  if (!intern) {
614
0
    RETURN_THROWS();
615
0
  }
616
617
18
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
618
0
    RETURN_THROWS();
619
0
  }
620
621
18
  RETURN_BOOL(intern->iter->funcs->valid(intern->iter) == SUCCESS);
622
18
}
623
624
0
ZEND_METHOD(InternalIterator, rewind) {
625
0
  ZEND_PARSE_PARAMETERS_NONE();
626
627
0
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
628
0
  if (!intern) {
629
0
    RETURN_THROWS();
630
0
  }
631
632
0
  intern->rewind_called = true;
633
0
  if (!intern->iter->funcs->rewind) {
634
    /* Allow calling rewind() if no iteration has happened yet,
635
     * even if the iterator does not support rewinding. */
636
0
    if (intern->iter->index != 0) {
637
0
      zend_throw_error(NULL, "Iterator does not support rewinding");
638
0
      RETURN_THROWS();
639
0
    }
640
0
    intern->iter->index = 0;
641
0
    return;
642
0
  }
643
644
0
  intern->iter->funcs->rewind(intern->iter);
645
0
  intern->iter->index = 0;
646
0
}
647
648
/* {{{ zend_register_interfaces */
649
ZEND_API void zend_register_interfaces(void)
650
16
{
651
16
  zend_ce_traversable = register_class_Traversable();
652
16
  zend_ce_traversable->interface_gets_implemented = zend_implement_traversable;
653
654
16
  zend_ce_aggregate = register_class_IteratorAggregate(zend_ce_traversable);
655
16
  zend_ce_aggregate->interface_gets_implemented = zend_implement_aggregate;
656
657
16
  zend_ce_iterator = register_class_Iterator(zend_ce_traversable);
658
16
  zend_ce_iterator->interface_gets_implemented = zend_implement_iterator;
659
660
16
  zend_ce_serializable = register_class_Serializable();
661
16
  zend_ce_serializable->interface_gets_implemented = zend_implement_serializable;
662
663
16
  zend_ce_arrayaccess = register_class_ArrayAccess();
664
16
  zend_ce_arrayaccess->interface_gets_implemented = zend_implement_arrayaccess;
665
666
16
  zend_ce_countable = register_class_Countable();
667
668
16
  zend_ce_stringable = register_class_Stringable();
669
670
16
  zend_ce_internal_iterator = register_class_InternalIterator(zend_ce_iterator);
671
16
  zend_ce_internal_iterator->create_object = zend_internal_iterator_create;
672
16
  zend_ce_internal_iterator->default_object_handlers = &zend_internal_iterator_handlers;
673
674
16
  memcpy(&zend_internal_iterator_handlers, zend_get_std_object_handlers(),
675
16
    sizeof(zend_object_handlers));
676
  zend_internal_iterator_handlers.clone_obj = NULL;
677
16
  zend_internal_iterator_handlers.free_obj = zend_internal_iterator_free;
678
16
}
679
/* }}} */