Coverage Report

Created: 2025-12-14 06:09

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
61
{
41
61
  zend_function *fn;
42
61
  zend_class_entry *called_scope;
43
61
  zval params[2];
44
45
61
  if (param_count > 0) {
46
15
    ZVAL_COPY_VALUE(&params[0], arg1);
47
15
  }
48
61
  if (param_count > 1) {
49
0
    ZVAL_COPY_VALUE(&params[1], arg2);
50
0
  }
51
52
61
  if (!obj_ce) {
53
0
    obj_ce = object ? object->ce : NULL;
54
0
  }
55
61
  if (!fn_proxy || !*fn_proxy) {
56
60
    if (EXPECTED(obj_ce)) {
57
60
      fn = zend_hash_str_find_ptr_lc(
58
60
        &obj_ce->function_table, function_name, function_name_len);
59
60
      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
60
    } 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
60
    if (fn_proxy) {
71
9
      *fn_proxy = fn;
72
9
    }
73
60
  } else {
74
1
    fn = *fn_proxy;
75
1
  }
76
77
61
  if (object) {
78
61
    called_scope = object->ce;
79
61
  } else {
80
0
    called_scope = obj_ce;
81
0
  }
82
83
61
  zend_call_known_function(fn, object, called_scope, retval_ptr, param_count, params, NULL);
84
61
  return retval_ptr;
85
61
}
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
84
{
93
84
  zend_call_known_instance_method_with_0_params(
94
84
    ce->iterator_funcs_ptr->zf_new_iterator, Z_OBJ_P(object), retval);
95
84
}
96
/* }}} */
97
98
/* {{{ zend_user_it_invalidate_current */
99
ZEND_API void zend_user_it_invalidate_current(zend_object_iterator *_iter)
100
627
{
101
627
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
102
103
627
  if (!Z_ISUNDEF(iter->value)) {
104
135
    zval_ptr_dtor(&iter->value);
105
135
    ZVAL_UNDEF(&iter->value);
106
135
  }
107
627
}
108
/* }}} */
109
110
/* {{{ zend_user_it_dtor */
111
static void zend_user_it_dtor(zend_object_iterator *_iter)
112
234
{
113
234
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
114
234
  zval *object = &iter->it.data;
115
116
234
  zend_user_it_invalidate_current(_iter);
117
234
  zval_ptr_dtor(object);
118
234
}
119
/* }}} */
120
121
/* {{{ zend_user_it_valid */
122
ZEND_API zend_result zend_user_it_valid(zend_object_iterator *_iter)
123
246
{
124
246
  if (_iter) {
125
246
    zend_user_iterator *iter = (zend_user_iterator*)_iter;
126
246
    zval *object = &iter->it.data;
127
246
    zval more;
128
129
246
    zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_valid, Z_OBJ_P(object), &more);
130
246
    bool result = i_zend_is_true(&more);
131
246
    zval_ptr_dtor(&more);
132
246
    return result ? SUCCESS : FAILURE;
133
246
  }
134
0
  return FAILURE;
135
246
}
136
/* }}} */
137
138
/* {{{ zend_user_it_get_current_data */
139
ZEND_API zval *zend_user_it_get_current_data(zend_object_iterator *_iter)
140
165
{
141
165
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
142
165
  zval *object = &iter->it.data;
143
144
165
  if (Z_ISUNDEF(iter->value)) {
145
165
    zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_current, Z_OBJ_P(object), &iter->value);
146
165
  }
147
165
  return &iter->value;
148
165
}
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
93
{
154
93
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
155
93
  zval *object = &iter->it.data;
156
93
  zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_key, Z_OBJ_P(object), key);
157
93
  if (UNEXPECTED(Z_ISREF_P(key))) {
158
3
    zend_unwrap_reference(key);
159
3
  }
160
93
}
161
/* }}} */
162
163
/* {{{ zend_user_it_move_forward */
164
ZEND_API void zend_user_it_move_forward(zend_object_iterator *_iter)
165
114
{
166
114
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
167
114
  zval *object = &iter->it.data;
168
169
114
  zend_user_it_invalidate_current(_iter);
170
114
  zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_next, Z_OBJ_P(object), NULL);
171
114
}
172
/* }}} */
173
174
/* {{{ zend_user_it_rewind */
175
ZEND_API void zend_user_it_rewind(zend_object_iterator *_iter)
176
237
{
177
237
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
178
237
  zval *object = &iter->it.data;
179
180
237
  zend_user_it_invalidate_current(_iter);
181
237
  zend_call_known_instance_method_with_0_params(iter->ce->iterator_funcs_ptr->zf_rewind, Z_OBJ_P(object), NULL);
182
237
}
183
/* }}} */
184
185
ZEND_API HashTable *zend_user_it_get_gc(zend_object_iterator *_iter, zval **table, int *n)
186
36
{
187
36
  zend_user_iterator *iter = (zend_user_iterator*)_iter;
188
36
  if (Z_ISUNDEF(iter->value)) {
189
36
    *table = &iter->it.data;
190
36
    *n = 1;
191
36
  } else {
192
0
    zend_get_gc_buffer *gc_buffer = zend_get_gc_buffer_create();
193
0
    zend_get_gc_buffer_add_zval(gc_buffer, &iter->it.data);
194
0
    zend_get_gc_buffer_add_zval(gc_buffer, &iter->value);
195
0
    zend_get_gc_buffer_use(gc_buffer, table, n);
196
0
  }
197
36
  return NULL;
198
36
}
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
234
{
215
234
  zend_user_iterator *iterator;
216
217
234
  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
234
  iterator = emalloc(sizeof(zend_user_iterator));
223
224
234
  zend_iterator_init((zend_object_iterator*)iterator);
225
226
234
  ZVAL_OBJ_COPY(&iterator->it.data, Z_OBJ_P(object));
227
234
  iterator->it.funcs = &zend_interface_iterator_funcs_iterator;
228
234
  iterator->ce = Z_OBJCE_P(object);
229
234
  ZVAL_UNDEF(&iterator->value);
230
234
  return (zend_object_iterator*)iterator;
231
234
}
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
84
{
238
84
  zval iterator;
239
84
  zend_object_iterator *new_iterator;
240
84
  zend_class_entry *ce_it;
241
242
84
  zend_user_it_new_iterator(ce, object, &iterator);
243
84
  ce_it = (Z_TYPE(iterator) == IS_OBJECT) ? Z_OBJCE(iterator) : NULL;
244
245
84
  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
12
    if (!EG(exception)) {
247
3
      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
3
    }
249
12
    zval_ptr_dtor(&iterator);
250
12
    return NULL;
251
12
  }
252
253
72
  new_iterator = ce_it->get_iterator(ce_it, &iterator, by_ref);
254
72
  zval_ptr_dtor(&iterator);
255
72
  return new_iterator;
256
84
}
257
/* }}} */
258
259
/* {{{ zend_implement_traversable */
260
static int zend_implement_traversable(zend_class_entry *interface, zend_class_entry *class_type)
261
239
{
262
  /* Abstract class can implement Traversable only, in which case the extending class must
263
   * implement Iterator or IteratorAggregate. */
264
239
  if (class_type->ce_flags & ZEND_ACC_EXPLICIT_ABSTRACT_CLASS) {
265
14
    return SUCCESS;
266
14
  }
267
268
  /* Check that class_type implements at least one of 'IteratorAggregate' or 'Iterator' */
269
225
  if (class_type->num_interfaces) {
270
225
    ZEND_ASSERT(class_type->ce_flags & ZEND_ACC_RESOLVED_INTERFACES);
271
418
    for (uint32_t i = 0; i < class_type->num_interfaces; i++) {
272
412
      if (class_type->interfaces[i] == zend_ce_aggregate || class_type->interfaces[i] == zend_ce_iterator) {
273
219
        return SUCCESS;
274
219
      }
275
412
    }
276
225
  }
277
6
  zend_error_noreturn(E_CORE_ERROR, "%s %s must implement interface %s as part of either %s or %s",
278
6
    zend_get_object_type_uc(class_type),
279
6
    ZSTR_VAL(class_type->name),
280
6
    ZSTR_VAL(zend_ce_traversable->name),
281
6
    ZSTR_VAL(zend_ce_iterator->name),
282
6
    ZSTR_VAL(zend_ce_aggregate->name));
283
0
  return FAILURE;
284
225
}
285
/* }}} */
286
287
/* {{{ zend_implement_aggregate */
288
static int zend_implement_aggregate(zend_class_entry *interface, zend_class_entry *class_type)
289
99
{
290
99
  if (zend_class_implements_interface(class_type, zend_ce_iterator)) {
291
3
    zend_error_noreturn(E_ERROR,
292
3
      "Class %s cannot implement both Iterator and IteratorAggregate at the same time",
293
3
      ZSTR_VAL(class_type->name));
294
3
  }
295
296
  /* Always initialize iterator_funcs_ptr. */
297
96
  ZEND_ASSERT(!class_type->iterator_funcs_ptr && "Iterator funcs already set?");
298
96
  zend_class_iterator_funcs *funcs_ptr = class_type->type == ZEND_INTERNAL_CLASS
299
96
    ? pemalloc(sizeof(zend_class_iterator_funcs), 1)
300
96
    : zend_arena_alloc(&CG(arena), sizeof(zend_class_iterator_funcs));
301
96
  class_type->iterator_funcs_ptr = funcs_ptr;
302
303
96
  memset(funcs_ptr, 0, sizeof(zend_class_iterator_funcs));
304
96
  funcs_ptr->zf_new_iterator = zend_hash_str_find_ptr(
305
96
    &class_type->function_table, "getiterator", sizeof("getiterator") - 1);
306
307
96
  if (class_type->get_iterator
308
22
   && class_type->get_iterator != zend_user_it_get_new_iterator
309
12
   && class_type->get_iterator != zend_hooked_object_get_iterator) {
310
    /* get_iterator was explicitly assigned for an internal class. */
311
8
    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
8
    if (funcs_ptr->zf_new_iterator->common.scope != class_type) {
318
8
      return SUCCESS;
319
8
    }
320
321
    /* getIterator() has been overwritten, switch to zend_user_it_get_new_iterator. */
322
8
  }
323
324
88
  class_type->get_iterator = zend_user_it_get_new_iterator;
325
88
  return SUCCESS;
326
96
}
327
/* }}} */
328
329
/* {{{ zend_implement_iterator */
330
static int zend_implement_iterator(zend_class_entry *interface, zend_class_entry *class_type)
331
132
{
332
132
  if (zend_class_implements_interface(class_type, zend_ce_aggregate)) {
333
3
    zend_error_noreturn(E_ERROR,
334
3
      "Class %s cannot implement both Iterator and IteratorAggregate at the same time",
335
3
      ZSTR_VAL(class_type->name));
336
3
  }
337
338
129
  ZEND_ASSERT(!class_type->iterator_funcs_ptr && "Iterator funcs already set?");
339
129
  zend_class_iterator_funcs *funcs_ptr = class_type->type == ZEND_INTERNAL_CLASS
340
129
    ? pemalloc(sizeof(zend_class_iterator_funcs), 1)
341
129
    : zend_arena_alloc(&CG(arena), sizeof(zend_class_iterator_funcs));
342
129
  class_type->iterator_funcs_ptr = funcs_ptr;
343
344
129
  memset(funcs_ptr, 0, sizeof(zend_class_iterator_funcs));
345
129
  funcs_ptr->zf_rewind = zend_hash_str_find_ptr(
346
129
    &class_type->function_table, "rewind", sizeof("rewind") - 1);
347
129
  funcs_ptr->zf_valid = zend_hash_str_find_ptr(
348
129
    &class_type->function_table, "valid", sizeof("valid") - 1);
349
129
  funcs_ptr->zf_key = zend_hash_find_ptr(
350
129
    &class_type->function_table, ZSTR_KNOWN(ZEND_STR_KEY));
351
129
  funcs_ptr->zf_current = zend_hash_str_find_ptr(
352
129
    &class_type->function_table, "current", sizeof("current") - 1);
353
129
  funcs_ptr->zf_next = zend_hash_str_find_ptr(
354
129
    &class_type->function_table, "next", sizeof("next") - 1);
355
356
129
  if (class_type->get_iterator
357
66
   && class_type->get_iterator != zend_user_it_get_iterator
358
26
   && class_type->get_iterator != zend_hooked_object_get_iterator) {
359
22
    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
22
    if (funcs_ptr->zf_rewind->common.scope != class_type &&
367
20
        funcs_ptr->zf_valid->common.scope != class_type &&
368
20
        funcs_ptr->zf_key->common.scope != class_type &&
369
16
        funcs_ptr->zf_current->common.scope != class_type &&
370
14
        funcs_ptr->zf_next->common.scope != class_type) {
371
14
      return SUCCESS;
372
14
    }
373
374
    /* One of the Iterator methods has been overwritten,
375
     * switch to zend_user_it_get_iterator. */
376
22
  }
377
378
115
  class_type->get_iterator = zend_user_it_get_iterator;
379
115
  return SUCCESS;
380
129
}
381
/* }}} */
382
383
/* {{{ zend_implement_arrayaccess */
384
static int zend_implement_arrayaccess(zend_class_entry *interface, zend_class_entry *class_type)
385
175
{
386
175
  ZEND_ASSERT(!class_type->arrayaccess_funcs_ptr && "ArrayAccess funcs already set?");
387
175
  zend_class_arrayaccess_funcs *funcs_ptr = class_type->type == ZEND_INTERNAL_CLASS
388
175
    ? pemalloc(sizeof(zend_class_arrayaccess_funcs), 1)
389
175
    : zend_arena_alloc(&CG(arena), sizeof(zend_class_arrayaccess_funcs));
390
175
  class_type->arrayaccess_funcs_ptr = funcs_ptr;
391
392
175
  funcs_ptr->zf_offsetget = zend_hash_str_find_ptr(
393
175
    &class_type->function_table, "offsetget", sizeof("offsetget") - 1);
394
175
  funcs_ptr->zf_offsetexists = zend_hash_str_find_ptr(
395
175
    &class_type->function_table, "offsetexists", sizeof("offsetexists") - 1);
396
175
  funcs_ptr->zf_offsetset = zend_hash_str_find_ptr(
397
175
    &class_type->function_table, "offsetset", sizeof("offsetset") - 1);
398
175
  funcs_ptr->zf_offsetunset = zend_hash_str_find_ptr(
399
175
    &class_type->function_table, "offsetunset", sizeof("offsetunset") - 1);
400
401
175
  return SUCCESS;
402
175
}
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
6
{
408
6
  zend_class_entry * ce = Z_OBJCE_P(object);
409
6
  zval retval;
410
6
  zend_result result;
411
412
6
  zend_call_method_with_0_params(
413
6
    Z_OBJ_P(object), Z_OBJCE_P(object), NULL, "serialize", &retval);
414
415
6
  if (Z_TYPE(retval) == IS_UNDEF) {
416
3
    result = FAILURE;
417
3
  } else {
418
3
    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
3
    case IS_STRING:
423
3
      *buffer = (unsigned char*)estrndup(Z_STRVAL(retval), Z_STRLEN(retval));
424
3
      *buf_len = Z_STRLEN(retval);
425
3
      result = SUCCESS;
426
3
      break;
427
0
    default: /* failure */
428
0
      result = FAILURE;
429
0
      break;
430
3
    }
431
3
    zval_ptr_dtor(&retval);
432
3
  }
433
434
6
  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
6
  return result;
438
6
}
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
15
{
444
15
  zval zdata;
445
446
15
  if (UNEXPECTED(object_init_ex(object, ce) != SUCCESS)) {
447
0
    return FAILURE;
448
0
  }
449
450
15
  ZVAL_STRINGL(&zdata, (char*)buf, buf_len);
451
15
  zend_call_method_with_1_params(
452
15
    Z_OBJ_P(object), Z_OBJCE_P(object), NULL, "unserialize", NULL, &zdata);
453
15
  zval_ptr_dtor(&zdata);
454
455
15
  if (EG(exception)) {
456
6
    return FAILURE;
457
9
  } else {
458
9
    return SUCCESS;
459
9
  }
460
15
}
461
/* }}} */
462
463
/* {{{ zend_implement_serializable */
464
static int zend_implement_serializable(zend_class_entry *interface, zend_class_entry *class_type)
465
47
{
466
47
  if (class_type->parent
467
30
    && (class_type->parent->serialize || class_type->parent->unserialize)
468
30
    && !zend_class_implements_interface(class_type->parent, zend_ce_serializable)) {
469
0
    return FAILURE;
470
0
  }
471
47
  if (!class_type->serialize) {
472
17
    class_type->serialize = zend_user_serialize;
473
17
  }
474
47
  if (!class_type->unserialize) {
475
17
    class_type->unserialize = zend_user_unserialize;
476
17
  }
477
47
  if (!(class_type->ce_flags & ZEND_ACC_EXPLICIT_ABSTRACT_CLASS)
478
45
      && (!class_type->__serialize || !class_type->__unserialize)) {
479
9
    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
9
    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
9
  }
485
47
  return SUCCESS;
486
47
}
487
/* }}}*/
488
489
typedef struct {
490
  zend_object std;
491
  zend_object_iterator *iter;
492
  bool rewind_called;
493
} zend_internal_iterator;
494
495
10
static zend_object *zend_internal_iterator_create(zend_class_entry *ce) {
496
10
  zend_internal_iterator *intern = emalloc(sizeof(zend_internal_iterator));
497
10
  zend_object_std_init(&intern->std, ce);
498
10
  intern->iter = NULL;
499
10
  intern->rewind_called = false;
500
10
  return &intern->std;
501
10
}
502
503
6
ZEND_API zend_result zend_create_internal_iterator_zval(zval *return_value, zval *obj) {
504
6
  zend_class_entry *scope = EG(current_execute_data)->func->common.scope;
505
6
  ZEND_ASSERT(scope->get_iterator != zend_user_it_get_new_iterator);
506
6
  zend_object_iterator *iter = scope->get_iterator(Z_OBJCE_P(obj), obj, /* by_ref */ 0);
507
6
  if (!iter) {
508
0
    return FAILURE;
509
0
  }
510
511
6
  zend_internal_iterator *intern =
512
6
    (zend_internal_iterator *) zend_internal_iterator_create(zend_ce_internal_iterator);
513
6
  intern->iter = iter;
514
6
  intern->iter->index = 0;
515
6
  ZVAL_OBJ(return_value, &intern->std);
516
6
  return SUCCESS;
517
6
}
518
519
10
static void zend_internal_iterator_free(zend_object *obj) {
520
10
  zend_internal_iterator *intern = (zend_internal_iterator *) obj;
521
10
  if (intern->iter) {
522
6
    zend_iterator_dtor(intern->iter);
523
6
  }
524
10
  zend_object_std_dtor(&intern->std);
525
10
}
526
527
30
static zend_internal_iterator *zend_internal_iterator_fetch(zval *This) {
528
30
  zend_internal_iterator *intern = (zend_internal_iterator *) Z_OBJ_P(This);
529
30
  if (!intern->iter) {
530
0
    zend_throw_error(NULL, "The InternalIterator object has not been properly initialized");
531
0
    return NULL;
532
0
  }
533
30
  return intern;
534
30
}
535
536
/* Many iterators will not behave correctly if rewind() is not called, make sure it happens. */
537
30
static zend_result zend_internal_iterator_ensure_rewound(zend_internal_iterator *intern) {
538
30
  if (!intern->rewind_called) {
539
6
    zend_object_iterator *iter = intern->iter;
540
6
    intern->rewind_called = true;
541
6
    if (iter->funcs->rewind) {
542
6
      iter->funcs->rewind(iter);
543
6
      if (UNEXPECTED(EG(exception))) {
544
0
        return FAILURE;
545
0
      }
546
6
    }
547
6
  }
548
30
  return SUCCESS;
549
30
}
550
551
552
0
ZEND_METHOD(InternalIterator, __construct) {
553
0
  zend_throw_error(NULL, "Cannot manually construct InternalIterator");
554
0
}
555
556
9
ZEND_METHOD(InternalIterator, current) {
557
9
  ZEND_PARSE_PARAMETERS_NONE();
558
559
9
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
560
9
  if (!intern) {
561
0
    RETURN_THROWS();
562
0
  }
563
564
9
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
565
0
    RETURN_THROWS();
566
0
  }
567
568
9
  zval *data = intern->iter->funcs->get_current_data(intern->iter);
569
9
  if (data) {
570
3
    RETURN_COPY_DEREF(data);
571
3
  }
572
9
}
573
574
9
ZEND_METHOD(InternalIterator, key) {
575
9
  ZEND_PARSE_PARAMETERS_NONE();
576
577
9
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
578
9
  if (!intern) {
579
0
    RETURN_THROWS();
580
0
  }
581
582
9
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
583
0
    RETURN_THROWS();
584
0
  }
585
586
9
  if (intern->iter->funcs->get_current_key) {
587
9
    intern->iter->funcs->get_current_key(intern->iter, return_value);
588
9
  } else {
589
0
    RETURN_LONG(intern->iter->index);
590
0
  }
591
9
}
592
593
3
ZEND_METHOD(InternalIterator, next) {
594
3
  ZEND_PARSE_PARAMETERS_NONE();
595
596
3
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
597
3
  if (!intern) {
598
0
    RETURN_THROWS();
599
0
  }
600
601
3
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
602
0
    RETURN_THROWS();
603
0
  }
604
605
  /* Advance index first to match foreach behavior. */
606
3
  intern->iter->index++;
607
3
  intern->iter->funcs->move_forward(intern->iter);
608
3
}
609
610
9
ZEND_METHOD(InternalIterator, valid) {
611
9
  ZEND_PARSE_PARAMETERS_NONE();
612
613
9
  zend_internal_iterator *intern = zend_internal_iterator_fetch(ZEND_THIS);
614
9
  if (!intern) {
615
0
    RETURN_THROWS();
616
0
  }
617
618
9
  if (zend_internal_iterator_ensure_rewound(intern) == FAILURE) {
619
0
    RETURN_THROWS();
620
0
  }
621
622
9
  RETURN_BOOL(intern->iter->funcs->valid(intern->iter) == SUCCESS);
623
9
}
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
2
{
652
2
  zend_ce_traversable = register_class_Traversable();
653
2
  zend_ce_traversable->interface_gets_implemented = zend_implement_traversable;
654
655
2
  zend_ce_aggregate = register_class_IteratorAggregate(zend_ce_traversable);
656
2
  zend_ce_aggregate->interface_gets_implemented = zend_implement_aggregate;
657
658
2
  zend_ce_iterator = register_class_Iterator(zend_ce_traversable);
659
2
  zend_ce_iterator->interface_gets_implemented = zend_implement_iterator;
660
661
2
  zend_ce_serializable = register_class_Serializable();
662
2
  zend_ce_serializable->interface_gets_implemented = zend_implement_serializable;
663
664
2
  zend_ce_arrayaccess = register_class_ArrayAccess();
665
2
  zend_ce_arrayaccess->interface_gets_implemented = zend_implement_arrayaccess;
666
667
2
  zend_ce_countable = register_class_Countable();
668
669
2
  zend_ce_stringable = register_class_Stringable();
670
671
2
  zend_ce_internal_iterator = register_class_InternalIterator(zend_ce_iterator);
672
2
  zend_ce_internal_iterator->create_object = zend_internal_iterator_create;
673
2
  zend_ce_internal_iterator->default_object_handlers = &zend_internal_iterator_handlers;
674
675
2
  memcpy(&zend_internal_iterator_handlers, zend_get_std_object_handlers(),
676
2
    sizeof(zend_object_handlers));
677
  zend_internal_iterator_handlers.clone_obj = NULL;
678
2
  zend_internal_iterator_handlers.free_obj = zend_internal_iterator_free;
679
2
}
680
/* }}} */