mysqli_result_iterator.c

/*
 +----------------------------------------------------------------------+
 | Copyright (c) The PHP Group |
 +----------------------------------------------------------------------+
 | This source file is subject to version 3.01 of the PHP license, |
 | that is bundled with this package in the file LICENSE, and is |
 | available through the world-wide-web at the following url: |
 | https://www.php.net/license/3_01.txt |
 | If you did not receive a copy of the PHP license and are unable to |
 | obtain it through the world-wide-web, please send a note to |
 | license@php.net so we can mail you a copy immediately. |
 +----------------------------------------------------------------------+
 | Authors: Georg Richter <georg@php.net> |
 | Andrey Hristov <andrey@php.net> |
 | Ulf Wendel <uw@php.net> |
 +----------------------------------------------------------------------+
*/

#ifdefHAVE_CONFIG_H
#include<config.h>
#endif

#include<signal.h>

#include"php.h"
#include"php_mysqli_structs.h"
#include"mysqli_priv.h"
#include"zend_interfaces.h"


externconstzend_object_iterator_funcsphp_mysqli_result_iterator_funcs;

typedefstruct {
zend_object_iteratorintern;
mysqli_object*result;
zvalcurrent_row;
my_longlongrow_num;
} php_mysqli_result_iterator;


/* {{{ */
zend_object_iterator*php_mysqli_result_get_iterator(zend_class_entry*ce, zval*object, intby_ref)
{
php_mysqli_result_iterator*iterator;

if (by_ref) {
zend_throw_error(NULL, "An iterator cannot be used with foreach by reference");
returnNULL;
 }

iterator=ecalloc(1, sizeof(php_mysqli_result_iterator));
zend_iterator_init(&iterator->intern);

Z_ADDREF_P(object);
ZVAL_OBJ(&iterator->intern.data, Z_OBJ_P(object));
iterator->intern.funcs=&php_mysqli_result_iterator_funcs;
iterator->result=Z_MYSQLI_P(object);
iterator->row_num=-1;

return&iterator->intern;
}
/* }}} */

/* {{{ */
staticvoidphp_mysqli_result_iterator_dtor(zend_object_iterator*iter)
{
php_mysqli_result_iterator*iterator= (php_mysqli_result_iterator*)iter;

/* cleanup handled in sxe_object_dtor as we don't always have an iterator wrapper */
zval_ptr_dtor(&iterator->intern.data);
zval_ptr_dtor(&iterator->current_row);
}
/* }}} */

/* {{{ */
staticzend_resultphp_mysqli_result_iterator_valid(zend_object_iterator*iter)
{
php_mysqli_result_iterator*iterator= (php_mysqli_result_iterator*) iter;

returnZ_TYPE(iterator->current_row) ==IS_ARRAY ? SUCCESS : FAILURE;
}
/* }}} */

/* {{{ */
staticzval*php_mysqli_result_iterator_current_data(zend_object_iterator*iter)
{
php_mysqli_result_iterator*iterator= (php_mysqli_result_iterator*) iter;

return&iterator->current_row;
}
/* }}} */

/* {{{ */
staticvoidphp_mysqli_result_iterator_move_forward(zend_object_iterator*iter)
{

php_mysqli_result_iterator*iterator= (php_mysqli_result_iterator*) iter;
mysqli_object*intern=iterator->result;
MYSQL_RES*result;

MYSQLI_FETCH_RESOURCE_BY_OBJ(result, MYSQL_RES*, intern, "mysqli_result", MYSQLI_STATUS_VALID);

zval_ptr_dtor(&iterator->current_row);
php_mysqli_fetch_into_hash_aux(&iterator->current_row, result, MYSQLI_ASSOC);
if (Z_TYPE(iterator->current_row) ==IS_ARRAY) {
iterator->row_num++;
 }
}
/* }}} */

/* {{{ */
staticvoidphp_mysqli_result_iterator_rewind(zend_object_iterator*iter)
{
php_mysqli_result_iterator*iterator= (php_mysqli_result_iterator*) iter;
mysqli_object*intern=iterator->result;
MYSQL_RES*result;

MYSQLI_FETCH_RESOURCE_BY_OBJ(result, MYSQL_RES*, intern, "mysqli_result", MYSQLI_STATUS_VALID);

if (mysqli_result_is_unbuffered(result)) {
if (result->unbuf->eof_reached) {
zend_error(E_WARNING, "Data fetched with MYSQLI_USE_RESULT can be iterated only once");
return;
 }
 } else {
mysql_data_seek(result, 0);
 }
iterator->row_num=-1;
php_mysqli_result_iterator_move_forward(iter);
}
/* }}} */

/* {{{ php_mysqli_result_iterator_current_key */
staticvoidphp_mysqli_result_iterator_current_key(zend_object_iterator*iter, zval*key)
{
php_mysqli_result_iterator*iterator= (php_mysqli_result_iterator*) iter;

ZVAL_LONG(key, iterator->row_num);
}
/* }}} */

/* {{{ php_mysqli_result_iterator_funcs */
constzend_object_iterator_funcsphp_mysqli_result_iterator_funcs= {
php_mysqli_result_iterator_dtor,
php_mysqli_result_iterator_valid,
php_mysqli_result_iterator_current_data,
php_mysqli_result_iterator_current_key,
php_mysqli_result_iterator_move_forward,
php_mysqli_result_iterator_rewind,
NULL,
NULL, /* get_gc */
};
/* }}} */