zend_iterators.c 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Zend Engine |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) 1998-2016 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. | Author: Wez Furlong <wez@thebrainroom.com> |
  16. | Marcus Boerger <helly@php.net> |
  17. +----------------------------------------------------------------------+
  18. */
  19. /* $Id$ */
  20. #include "zend.h"
  21. #include "zend_API.h"
  22. static zend_class_entry zend_iterator_class_entry;
  23. static zend_object_handlers iterator_object_handlers = {
  24. ZEND_OBJECTS_STORE_HANDLERS,
  25. NULL, /* prop read */
  26. NULL, /* prop write */
  27. NULL, /* read dim */
  28. NULL, /* write dim */
  29. NULL,
  30. NULL, /* get */
  31. NULL, /* set */
  32. NULL, /* has prop */
  33. NULL, /* unset prop */
  34. NULL, /* has dim */
  35. NULL, /* unset dim */
  36. NULL, /* props get */
  37. NULL, /* method get */
  38. NULL, /* call */
  39. NULL, /* get ctor */
  40. NULL, /* get_ce */
  41. NULL, /* get class name */
  42. NULL, /* compare */
  43. NULL, /* cast */
  44. NULL /* count */
  45. };
  46. ZEND_API void zend_register_iterator_wrapper(TSRMLS_D)
  47. {
  48. INIT_CLASS_ENTRY(zend_iterator_class_entry, "__iterator_wrapper", NULL);
  49. str_free(zend_iterator_class_entry.name);
  50. zend_iterator_class_entry.name = "__iterator_wrapper";
  51. }
  52. static void iter_wrapper_dtor(void *object, zend_object_handle handle TSRMLS_DC)
  53. {
  54. zend_object_iterator *iter = (zend_object_iterator*)object;
  55. iter->funcs->dtor(iter TSRMLS_CC);
  56. }
  57. ZEND_API zval *zend_iterator_wrap(zend_object_iterator *iter TSRMLS_DC)
  58. {
  59. zval *wrapped;
  60. MAKE_STD_ZVAL(wrapped);
  61. Z_TYPE_P(wrapped) = IS_OBJECT;
  62. Z_OBJ_HANDLE_P(wrapped) = zend_objects_store_put(iter, iter_wrapper_dtor, NULL, NULL TSRMLS_CC);
  63. Z_OBJ_HT_P(wrapped) = &iterator_object_handlers;
  64. return wrapped;
  65. }
  66. ZEND_API enum zend_object_iterator_kind zend_iterator_unwrap(
  67. zval *array_ptr, zend_object_iterator **iter TSRMLS_DC)
  68. {
  69. switch (Z_TYPE_P(array_ptr)) {
  70. case IS_OBJECT:
  71. if (Z_OBJ_HT_P(array_ptr) == &iterator_object_handlers) {
  72. *iter = (zend_object_iterator *)zend_object_store_get_object(array_ptr TSRMLS_CC);
  73. return ZEND_ITER_OBJECT;
  74. }
  75. if (Z_OBJPROP_P(array_ptr)) {
  76. return ZEND_ITER_PLAIN_OBJECT;
  77. }
  78. return ZEND_ITER_INVALID;
  79. case IS_ARRAY:
  80. if (Z_ARRVAL_P(array_ptr)) {
  81. return ZEND_ITER_PLAIN_ARRAY;
  82. }
  83. return ZEND_ITER_INVALID;
  84. default:
  85. return ZEND_ITER_INVALID;
  86. }
  87. }
  88. /*
  89. * Local variables:
  90. * tab-width: 4
  91. * c-basic-offset: 4
  92. * indent-tabs-mode: t
  93. * End:
  94. */