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 /* {{{ */
php_mysqli_result_get_iterator(zend_class_entry * ce,zval * object,int by_ref TSRMLS_DC)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 /* {{{ */
php_mysqli_result_iterator_dtor(zend_object_iterator * iter TSRMLS_DC)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 /* {{{ */
php_mysqli_result_iterator_valid(zend_object_iterator * iter TSRMLS_DC)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 /* {{{ */
php_mysqli_result_iterator_current_data(zend_object_iterator * iter,zval *** data TSRMLS_DC)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 /* {{{ */
php_mysqli_result_iterator_move_forward(zend_object_iterator * iter TSRMLS_DC)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 /* {{{ */
php_mysqli_result_iterator_rewind(zend_object_iterator * iter TSRMLS_DC)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  */
182