root/ext/mysqli/mysqli_result_iterator.c

DEFINITIONS

1. php_mysqli_result_get_iterator
2. php_mysqli_result_iterator_dtor
3. php_mysqli_result_iterator_valid
4. php_mysqli_result_iterator_current_data
5. php_mysqli_result_iterator_move_forward
6. php_mysqli_result_iterator_rewind
7. php_mysqli_result_iterator_current_key

   1 /*
   2   +----------------------------------------------------------------------+
   3   | PHP Version 5                                                        |
   4   +----------------------------------------------------------------------+
   5   | Copyright (c) 1997-2016 The PHP Group                                |
   6   +----------------------------------------------------------------------+
   7   | This source file is subject to version 3.01 of the PHP 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.php.net/license/3_01.txt                                  |
  11   | If you did not receive a copy of the PHP license and are unable to   |
  12   | obtain it through the world-wide-web, please send a note to          |
  13   | license@php.net so we can mail you a copy immediately.               |
  14   +----------------------------------------------------------------------+
  15   | Authors: Georg Richter <georg@php.net>                               |
  16   |          Andrey Hristov <andrey@php.net>                             |
  17   |          Ulf Wendel <uw@php.net>                                     |
  18   +----------------------------------------------------------------------+
  19 
  20   $Id: mysqli.c 299335 2010-05-13 11:05:09Z andrey $
  21 */
  22 
  23 #ifdef HAVE_CONFIG_H
  24 #include "config.h"
  25 #endif
  26 
  27 #include <signal.h>
  28 
  29 #include "php.h"
  30 #include "php_ini.h"
  31 #include "php_mysqli_structs.h"
  32 #include "mysqli_priv.h"
  33 #include "zend_interfaces.h"
  34 
  35 
  36 extern zend_object_iterator_funcs php_mysqli_result_iterator_funcs;
  37 
  38 typedef struct {
  39         zend_object_iterator  intern;
  40         mysqli_object *result;
  41         zval *current_row;
  42         my_longlong row_num;
  43 } php_mysqli_result_iterator;
  44 
  45 
  46 /* {{{ */
  47 zend_object_iterator *php_mysqli_result_get_iterator(zend_class_entry *ce, zval *object, int by_ref TSRMLS_DC)
  48 {
  49         php_mysqli_result_iterator *iterator;
  50 
  51         if (by_ref) {
  52                 zend_error(E_ERROR, "An iterator cannot be used with foreach by reference");
  53         }
  54         iterator = ecalloc(1, sizeof(php_mysqli_result_iterator));
  55 
  56         Z_ADDREF_P(object);
  57         iterator->intern.data = (void*)object;
  58         iterator->intern.funcs = &php_mysqli_result_iterator_funcs;
  59         iterator->result = (mysqli_object *) zend_object_store_get_object(object TSRMLS_CC);
  60         iterator->row_num = -1;
  61 
  62         return (zend_object_iterator*)iterator;
  63 }
  64 /* }}} */
  65 
  66 
  67 /* {{{ */
  68 static void php_mysqli_result_iterator_dtor(zend_object_iterator *iter TSRMLS_DC)
  69 {
  70         php_mysqli_result_iterator *iterator = (php_mysqli_result_iterator*) iter;
  71 
  72         /* cleanup handled in sxe_object_dtor as we dont always have an iterator wrapper */
  73         if (iterator->intern.data) {
  74                 zval_ptr_dtor((zval**)&iterator->intern.data);
  75         }
  76         if (iterator->current_row) {
  77                 zval_ptr_dtor(&iterator->current_row);
  78         }
  79         efree(iterator);
  80 }
  81 /* }}} */
  82 
  83 
  84 /* {{{ */
  85 static int php_mysqli_result_iterator_valid(zend_object_iterator *iter TSRMLS_DC)
  86 {
  87         php_mysqli_result_iterator *iterator = (php_mysqli_result_iterator*) iter;
  88 
  89         return iterator->current_row && Z_TYPE_P(iterator->current_row) == IS_ARRAY ? SUCCESS : FAILURE;
  90 }
  91 /* }}} */
  92 
  93 
  94 /* {{{ */
  95 static void php_mysqli_result_iterator_current_data(zend_object_iterator *iter, zval ***data TSRMLS_DC)
  96 {
  97         php_mysqli_result_iterator *iterator = (php_mysqli_result_iterator*) iter;
  98 
  99         *data = &iterator->current_row;
 100 }
 101 /* }}} */
 102 
 103 
 104 /* {{{ */
 105 static void php_mysqli_result_iterator_move_forward(zend_object_iterator *iter TSRMLS_DC)
 106 {
 107 
 108         php_mysqli_result_iterator *iterator = (php_mysqli_result_iterator*) iter;
 109         mysqli_object *intern = iterator->result;
 110         MYSQL_RES       *result;
 111 
 112         MYSQLI_FETCH_RESOURCE_BY_OBJ(result, MYSQL_RES *, intern, "mysqli_result", MYSQLI_STATUS_VALID);
 113         if (iterator->current_row) {
 114                 zval_ptr_dtor(&iterator->current_row);
 115         }
 116         MAKE_STD_ZVAL(iterator->current_row);
 117         php_mysqli_fetch_into_hash_aux(iterator->current_row, result, MYSQLI_ASSOC TSRMLS_CC);
 118         if (Z_TYPE_P(iterator->current_row) == IS_ARRAY) {
 119                 iterator->row_num++;
 120         }
 121 }
 122 /* }}} */
 123 
 124 
 125 /* {{{ */
 126 static void php_mysqli_result_iterator_rewind(zend_object_iterator *iter TSRMLS_DC)
 127 {
 128         php_mysqli_result_iterator *iterator = (php_mysqli_result_iterator*) iter;
 129         mysqli_object *intern = iterator->result;
 130         MYSQL_RES       *result;
 131 
 132         MYSQLI_FETCH_RESOURCE_BY_OBJ(result, MYSQL_RES *, intern, "mysqli_result", MYSQLI_STATUS_VALID);
 133 
 134         if (mysqli_result_is_unbuffered(result)) {
 135 #if MYSQLI_USE_MYSQLND
 136                 if (result->unbuf->eof_reached) {
 137 #else
 138                 if (result->eof) {
 139 #endif
 140                         php_error_docref(NULL TSRMLS_CC, E_WARNING, "Data fetched with MYSQLI_USE_RESULT can be iterated only once");
 141                         return;
 142                 }
 143         } else {
 144                 mysql_data_seek(result, 0);
 145         }
 146         iterator->row_num = -1;
 147         php_mysqli_result_iterator_move_forward(iter TSRMLS_CC);
 148 }
 149 /* }}} */
 150 
 151 
 152 /* {{{ php_mysqli_result_iterator_current_key */
 153 static void php_mysqli_result_iterator_current_key(zend_object_iterator *iter, zval *key TSRMLS_DC)
 154 {
 155         php_mysqli_result_iterator *iterator = (php_mysqli_result_iterator*) iter;
 156 
 157         ZVAL_LONG(key, iterator->row_num);
 158 }
 159 /* }}} */
 160 
 161 
 162 /* {{{ php_mysqli_result_iterator_funcs */
 163 zend_object_iterator_funcs php_mysqli_result_iterator_funcs = {
 164         php_mysqli_result_iterator_dtor,
 165         php_mysqli_result_iterator_valid,
 166         php_mysqli_result_iterator_current_data,
 167         php_mysqli_result_iterator_current_key,
 168         php_mysqli_result_iterator_move_forward,
 169         php_mysqli_result_iterator_rewind,
 170 };
 171 /* }}} */
 172 
 173 
 174 /*
 175  * Local variables:
 176  * tab-width: 4
 177  * c-basic-offset: 4
 178  * End:
 179  * vim600: noet sw=4 ts=4 fdm=marker
 180  * vim<600: noet sw=4 ts=4
 181  */