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