]> granicus.if.org Git - php/commitdiff
MFB: more SPL leak fixes
authorAntony Dovgal <tony2001@php.net>
Tue, 20 Dec 2005 12:41:38 +0000 (12:41 +0000)
committerAntony Dovgal <tony2001@php.net>
Tue, 20 Dec 2005 12:41:38 +0000 (12:41 +0000)
ext/spl/spl_iterators.c

index 5871edc9b61cca67a2011f5baa028d679652340d..86da1a5fa80bce974295941d2e87a7b929d36b01 100755 (executable)
@@ -372,6 +372,7 @@ SPL_METHOD(RecursiveIteratorIterator, __construct)
        zval                      *iterator;
        zend_class_entry          *ce_iterator;
        long                       mode = RIT_LEAVES_ONLY, flags = 0;
+       int                        inc_refcount = 1;
 
        php_set_error_handling(EH_THROW, U_CLASS_ENTRY(spl_ce_InvalidArgumentException) TSRMLS_CC);
 
@@ -379,6 +380,7 @@ SPL_METHOD(RecursiveIteratorIterator, __construct)
                if (instanceof_function(Z_OBJCE_P(iterator), U_CLASS_ENTRY(zend_ce_aggregate) TSRMLS_CC)) {
                        zval *aggregate = iterator;
                        zend_call_method_with_0_params(&aggregate, Z_OBJCE_P(aggregate), &Z_OBJCE_P(aggregate)->iterator_funcs.zf_new_iterator, "getiterator", &iterator);
+                       inc_refcount = 0;
                }
        } else {
                iterator = NULL;
@@ -428,7 +430,9 @@ SPL_METHOD(RecursiveIteratorIterator, __construct)
        }
        ce_iterator = Z_OBJCE_P(iterator); /* respect inheritance, don't use spl_ce_RecursiveIterator */
        intern->iterators[0].iterator = ce_iterator->get_iterator(ce_iterator, iterator TSRMLS_CC);
-       iterator->refcount++;
+       if (inc_refcount) {
+               iterator->refcount++;
+       }
        intern->iterators[0].zobject = iterator;
        intern->iterators[0].ce = ce_iterator;
        intern->iterators[0].state = RS_START;