Coverage Report

Created: 2026-04-01 06:49

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