/* +----------------------------------------------------------------------+ | PHP version 4.0 | +----------------------------------------------------------------------+ | Copyright (c) 1997-2001 The PHP Group | +----------------------------------------------------------------------+ | This source file is subject to version 2.02 of the PHP license, | | that is bundled with this package in the file LICENSE, and is | | available at through the world-wide-web at | | http://www.php.net/license/2_02.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: Andi Gutmans | | Zeev Suraski | | Rasmus Lerdorf | | Andrei Zmievski | | Stig Venaas | | Jason Greene | +----------------------------------------------------------------------+ */ /* $Id$ */ #include "php.h" #include "php_ini.h" #include "zend_operators.h" #include #include #include #include #include #if HAVE_STRING_H #include #else #include #endif #ifdef PHP_WIN32 #include "win32/unistd.h" #endif #include "zend_globals.h" #include "php_globals.h" #include "php_array.h" #include "basic_functions.h" #include "php_string.h" #include "php_rand.h" #ifdef ZTS int array_globals_id; #else php_array_globals array_globals; #endif #define EXTR_OVERWRITE 0 #define EXTR_SKIP 1 #define EXTR_PREFIX_SAME 2 #define EXTR_PREFIX_ALL 3 #define EXTR_PREFIX_INVALID 4 #define SORT_REGULAR 0 #define SORT_NUMERIC 1 #define SORT_STRING 2 #define SORT_DESC 3 #define SORT_ASC 4 PHP_MINIT_FUNCTION(array) { #ifdef ZTS ts_allocate_id(&array_globals_id, sizeof(php_array_globals), NULL, NULL); #endif REGISTER_LONG_CONSTANT("EXTR_OVERWRITE", EXTR_OVERWRITE, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("EXTR_SKIP", EXTR_SKIP, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("EXTR_PREFIX_SAME", EXTR_PREFIX_SAME, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("EXTR_PREFIX_ALL", EXTR_PREFIX_ALL, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("EXTR_PREFIX_INVALID", EXTR_PREFIX_INVALID, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("SORT_ASC", SORT_ASC, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("SORT_DESC", SORT_DESC, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("SORT_REGULAR", SORT_REGULAR, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("SORT_NUMERIC", SORT_NUMERIC, CONST_CS | CONST_PERSISTENT); REGISTER_LONG_CONSTANT("SORT_STRING", SORT_STRING, CONST_CS | CONST_PERSISTENT); return SUCCESS; } PHP_MSHUTDOWN_FUNCTION(array) { #ifdef ZTS ts_free_id(array_globals_id); #endif return SUCCESS; } static void set_compare_func(int sort_type) { ARRAYLS_FETCH(); switch (sort_type) { case SORT_NUMERIC: ARRAYG(compare_func) = numeric_compare_function; break; case SORT_STRING: ARRAYG(compare_func) = string_compare_function; break; case SORT_REGULAR: default: ARRAYG(compare_func) = compare_function; break; } } static int array_key_compare(const void *a, const void *b) { Bucket *f; Bucket *s; pval result; pval first; pval second; ARRAYLS_FETCH(); f = *((Bucket **) a); s = *((Bucket **) b); if (f->nKeyLength == 0) { Z_TYPE(first) = IS_LONG; Z_LVAL(first) = f->h; } else { Z_TYPE(first) = IS_STRING; Z_STRVAL(first) = f->arKey; Z_STRLEN(first) = f->nKeyLength; } if (s->nKeyLength == 0) { Z_TYPE(second) = IS_LONG; Z_LVAL(second) = s->h; } else { Z_TYPE(second) = IS_STRING; Z_STRVAL(second) = s->arKey; Z_STRLEN(second) = s->nKeyLength; } if (ARRAYG(compare_func)(&result, &first, &second) == FAILURE) { return 0; } if (Z_TYPE(result) == IS_DOUBLE) { if (Z_DVAL(result) < 0) { return -1; } else if (Z_DVAL(result) > 0) { return 1; } else { return 0; } } convert_to_long(&result); if (Z_LVAL(result) < 0) { return -1; } else if (Z_LVAL(result) > 0) { return 1; } return 0; } static int array_reverse_key_compare(const void *a, const void *b) { return array_key_compare(a,b)*-1; } /* {{{ proto int krsort(array array_arg [, int sort_flags]) Sort an array reverse by key */ PHP_FUNCTION(krsort) { zval **array, **sort_type; int sort_type_val = SORT_REGULAR; HashTable *target_hash; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &array, &sort_type) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in krsort() call"); return; } if (ZEND_NUM_ARGS() == 2) { convert_to_long_ex(sort_type); sort_type_val = Z_LVAL_PP(sort_type); } set_compare_func(sort_type_val); if (zend_hash_sort(target_hash, qsort, array_reverse_key_compare, 0) == FAILURE) { return; } RETURN_TRUE; } /* }}} */ /* {{{ proto int ksort(array array_arg [, int sort_flags]) Sort an array by key */ PHP_FUNCTION(ksort) { zval **array, **sort_type; int sort_type_val = SORT_REGULAR; HashTable *target_hash; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &array, &sort_type) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in ksort() call"); return; } if (ZEND_NUM_ARGS() == 2) { convert_to_long_ex(sort_type); sort_type_val = Z_LVAL_PP(sort_type); } set_compare_func(sort_type_val); if (zend_hash_sort(target_hash, qsort, array_key_compare,0) == FAILURE) { return; } RETURN_TRUE; } /* }}} */ /* {{{ proto int count(mixed var) Count the number of elements in a variable (usually an array) */ PHP_FUNCTION(count) { pval **array; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { if (Z_TYPE_PP(array) == IS_NULL) { RETURN_LONG(0); } else { RETURN_LONG(1); } } RETURN_LONG(zend_hash_num_elements(target_hash)); } /* }}} */ /* Numbers are always smaller than strings int this function as it * anyway doesn't make much sense to compare two different data types. * This keeps it consistant and simple. * * This is not correct any more, depends on what compare_func is set to. */ static int array_data_compare(const void *a, const void *b) { Bucket *f; Bucket *s; pval result; pval *first; pval *second; ARRAYLS_FETCH(); f = *((Bucket **) a); s = *((Bucket **) b); first = *((pval **) f->pData); second = *((pval **) s->pData); if (ARRAYG(compare_func)(&result, first, second) == FAILURE) { return 0; } if (Z_TYPE(result) == IS_DOUBLE) { if (Z_DVAL(result) < 0) { return -1; } else if (Z_DVAL(result) > 0) { return 1; } else { return 0; } } convert_to_long(&result); if (Z_LVAL(result) < 0) { return -1; } else if (Z_LVAL(result) > 0) { return 1; } return 0; } static int array_reverse_data_compare(const void *a, const void *b) { return array_data_compare(a,b)*-1; } static int array_natural_general_compare(const void *a, const void *b, int fold_case) { Bucket *f, *s; zval *fval, *sval; zval first, second; int result; f = *((Bucket **) a); s = *((Bucket **) b); fval = *((pval **) f->pData); sval = *((pval **) s->pData); first = *fval; second = *sval; if (Z_TYPE_P(fval) != IS_STRING) { zval_copy_ctor(&first); convert_to_string(&first); } if (Z_TYPE_P(sval) != IS_STRING) { zval_copy_ctor(&first); convert_to_string(&second); } result = strnatcmp_ex(Z_STRVAL(first), Z_STRLEN(first), Z_STRVAL(second), Z_STRLEN(second), fold_case); if (Z_TYPE_P(fval) != IS_STRING) zval_dtor(&first); if (Z_TYPE_P(sval) != IS_STRING) zval_dtor(&second); return result; } static int array_natural_compare(const void *a, const void *b) { return array_natural_general_compare(a, b, 0); } static int array_natural_case_compare(const void *a, const void *b) { return array_natural_general_compare(a, b, 1); } static void php_natsort(INTERNAL_FUNCTION_PARAMETERS, int fold_case) { zval **array; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in %s() call", get_active_function_name()); return; } if (fold_case) { if (zend_hash_sort(target_hash, qsort, array_natural_case_compare, 0) == FAILURE) { return; } } else { if (zend_hash_sort(target_hash, qsort, array_natural_compare, 0) == FAILURE) { return; } } RETURN_TRUE; } /* {{{ proto void natsort(array array_arg) Sort an array using natural sort */ PHP_FUNCTION(natsort) { php_natsort(INTERNAL_FUNCTION_PARAM_PASSTHRU, 0); } /* }}} */ /* {{{ proto void natcasesort(array array_arg) Sort an array using case-insensitive natural sort */ PHP_FUNCTION(natcasesort) { php_natsort(INTERNAL_FUNCTION_PARAM_PASSTHRU, 1); } /* }}} */ /* {{{ proto void asort(array array_arg [, int sort_flags]) Sort an array and maintain index association */ PHP_FUNCTION(asort) { zval **array, **sort_type; int sort_type_val = SORT_REGULAR; HashTable *target_hash; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &array, &sort_type) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in asort() call"); return; } if (ZEND_NUM_ARGS() == 2) { convert_to_long_ex(sort_type); sort_type_val = Z_LVAL_PP(sort_type); } set_compare_func(sort_type_val); if (zend_hash_sort(target_hash, qsort, array_data_compare,0) == FAILURE) { return; } RETURN_TRUE; } /* }}} */ /* {{{ proto void arsort(array array_arg [, int sort_flags]) Sort an array in reverse order and maintain index association */ PHP_FUNCTION(arsort) { zval **array, **sort_type; int sort_type_val = SORT_REGULAR; HashTable *target_hash; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &array, &sort_type) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in arsort() call"); RETURN_FALSE; } if (ZEND_NUM_ARGS() == 2) { convert_to_long_ex(sort_type); sort_type_val = Z_LVAL_PP(sort_type); } set_compare_func(sort_type_val); if (zend_hash_sort(target_hash, qsort, array_reverse_data_compare,0) == FAILURE) { RETURN_FALSE; } RETURN_TRUE; } /* }}} */ /* {{{ proto void sort(array array_arg [, int sort_flags]) Sort an array */ PHP_FUNCTION(sort) { zval **array, **sort_type; int sort_type_val = SORT_REGULAR; HashTable *target_hash; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &array, &sort_type) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in sort() call"); RETURN_FALSE; } if (ZEND_NUM_ARGS() == 2) { convert_to_long_ex(sort_type); sort_type_val = Z_LVAL_PP(sort_type); } set_compare_func(sort_type_val); if (zend_hash_sort(target_hash, qsort, array_data_compare,1) == FAILURE) { RETURN_FALSE; } RETURN_TRUE; } /* }}} */ /* {{{ proto void rsort(array array_arg [, int sort_flags]) Sort an array in reverse order */ PHP_FUNCTION(rsort) { zval **array, **sort_type; int sort_type_val = SORT_REGULAR; HashTable *target_hash; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &array, &sort_type) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in rsort() call"); RETURN_FALSE; } if (ZEND_NUM_ARGS() == 2) { convert_to_long_ex(sort_type); sort_type_val = Z_LVAL_PP(sort_type); } set_compare_func(sort_type_val); if (zend_hash_sort(target_hash, qsort, array_reverse_data_compare,1) == FAILURE) { RETURN_FALSE; } RETURN_TRUE; } static int array_user_compare(const void *a, const void *b) { Bucket *f; Bucket *s; pval **args[2]; pval *retval_ptr; TSRMLS_FETCH(); BLS_FETCH(); f = *((Bucket **) a); s = *((Bucket **) b); args[0] = (pval **) f->pData; args[1] = (pval **) s->pData; if (call_user_function_ex(EG(function_table), NULL, *BG(user_compare_func_name), &retval_ptr, 2, args, 0, NULL)==SUCCESS && retval_ptr) { long retval; convert_to_long_ex(&retval_ptr); retval = Z_LVAL_P(retval_ptr); zval_ptr_dtor(&retval_ptr); return retval; } else { return 0; } } /* {{{ proto void usort(array array_arg, string cmp_function) Sort an array by values using a user-defined comparison function */ PHP_FUNCTION(usort) { pval **array; pval **old_compare_func; HashTable *target_hash; BLS_FETCH(); old_compare_func = BG(user_compare_func_name); if (ZEND_NUM_ARGS() != 2 || zend_get_parameters_ex(2, &array, &BG(user_compare_func_name)) == FAILURE) { BG(user_compare_func_name) = old_compare_func; WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in usort() call"); BG(user_compare_func_name) = old_compare_func; RETURN_FALSE; } if (zend_hash_sort(target_hash, qsort, array_user_compare, 1) == FAILURE) { BG(user_compare_func_name) = old_compare_func; RETURN_FALSE; } BG(user_compare_func_name) = old_compare_func; RETURN_TRUE; } /* }}} */ /* {{{ proto void uasort(array array_arg, string cmp_function) Sort an array with a user-defined comparison function and maintain index association */ PHP_FUNCTION(uasort) { pval **array; pval **old_compare_func; HashTable *target_hash; BLS_FETCH(); old_compare_func = BG(user_compare_func_name); if (ZEND_NUM_ARGS() != 2 || zend_get_parameters_ex(2, &array, &BG(user_compare_func_name)) == FAILURE) { BG(user_compare_func_name) = old_compare_func; WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in uasort() call"); BG(user_compare_func_name) = old_compare_func; RETURN_FALSE; } if (zend_hash_sort(target_hash, qsort, array_user_compare, 0) == FAILURE) { BG(user_compare_func_name) = old_compare_func; RETURN_FALSE; } BG(user_compare_func_name) = old_compare_func; RETURN_TRUE; } /* }}} */ static int array_user_key_compare(const void *a, const void *b) { Bucket *f; Bucket *s; pval key1, key2; pval *args[2]; pval retval; int status; TSRMLS_FETCH(); BLS_FETCH(); args[0] = &key1; args[1] = &key2; INIT_PZVAL(&key1); INIT_PZVAL(&key2); f = *((Bucket **) a); s = *((Bucket **) b); if (f->nKeyLength) { Z_STRVAL(key1) = estrndup(f->arKey, f->nKeyLength); Z_STRLEN(key1) = f->nKeyLength-1; Z_TYPE(key1) = IS_STRING; } else { Z_LVAL(key1) = f->h; Z_TYPE(key1) = IS_LONG; } if (s->nKeyLength) { Z_STRVAL(key2) = estrndup(s->arKey, s->nKeyLength); Z_STRLEN(key2) = s->nKeyLength-1; Z_TYPE(key2) = IS_STRING; } else { Z_LVAL(key2) = s->h; Z_TYPE(key2) = IS_LONG; } status = call_user_function(EG(function_table), NULL, *BG(user_compare_func_name), &retval, 2, args); zval_dtor(&key1); zval_dtor(&key2); if (status==SUCCESS) { convert_to_long(&retval); return Z_LVAL(retval); } else { return 0; } } /* {{{ proto void uksort(array array_arg, string cmp_function) Sort an array by keys using a user-defined comparison function */ PHP_FUNCTION(uksort) { pval **array; pval **old_compare_func; HashTable *target_hash; BLS_FETCH(); old_compare_func = BG(user_compare_func_name); if (ZEND_NUM_ARGS() != 2 || zend_get_parameters_ex(2, &array, &BG(user_compare_func_name)) == FAILURE) { BG(user_compare_func_name) = old_compare_func; WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in uksort() call"); BG(user_compare_func_name) = old_compare_func; RETURN_FALSE; } if (zend_hash_sort(target_hash, qsort, array_user_key_compare, 0) == FAILURE) { BG(user_compare_func_name) = old_compare_func; RETURN_FALSE; } BG(user_compare_func_name) = old_compare_func; RETURN_TRUE; } /* }}} */ /* {{{ proto mixed end(array array_arg) Advances array argument's internal pointer to the last element and return it */ PHP_FUNCTION(end) { pval **array, **entry; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Variable passed to end() is not an array or object"); RETURN_FALSE; } zend_hash_internal_pointer_end(target_hash); if (return_value_used) { if (zend_hash_get_current_data(target_hash, (void **) &entry) == FAILURE) { RETURN_FALSE; } *return_value = **entry; zval_copy_ctor(return_value); } } /* }}} */ /* {{{ proto mixed prev(array array_arg) Move array argument's internal pointer to the previous element and return it */ PHP_FUNCTION(prev) { pval **array, **entry; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Variable passed to prev() is not an array or object"); RETURN_FALSE; } zend_hash_move_backwards(target_hash); if (return_value_used) { if (zend_hash_get_current_data(target_hash, (void **) &entry) == FAILURE) { RETURN_FALSE; } *return_value = **entry; zval_copy_ctor(return_value); } } /* }}} */ /* {{{ proto mixed next(array array_arg) Move array argument's internal pointer to the next element and return it */ PHP_FUNCTION(next) { pval **array, **entry; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Variable passed to next() is not an array or object"); RETURN_FALSE; } zend_hash_move_forward(target_hash); if (return_value_used) { if (zend_hash_get_current_data(target_hash, (void **) &entry) == FAILURE) { RETURN_FALSE; } *return_value = **entry; zval_copy_ctor(return_value); } } /* }}} */ /* {{{ proto mixed reset(array array_arg) Set array argument's internal pointer to the first element and return it */ PHP_FUNCTION(reset) { pval **array, **entry; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Variable passed to reset() is not an array or object"); RETURN_FALSE; } zend_hash_internal_pointer_reset(target_hash); if (return_value_used) { if (zend_hash_get_current_data(target_hash, (void **) &entry) == FAILURE) { RETURN_FALSE; } *return_value = **entry; zval_copy_ctor(return_value); } } /* }}} */ /* {{{ proto mixed current(array array_arg) Return the element currently pointed to by the internal array pointer */ PHP_FUNCTION(current) { pval **array, **entry; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Variable passed to current() is not an array or object"); RETURN_FALSE; } if (zend_hash_get_current_data(target_hash, (void **) &entry) == FAILURE) { RETURN_FALSE; } *return_value = **entry; zval_copy_ctor(return_value); } /* }}} */ /* {{{ proto mixed key(array array_arg) Return the key of the element currently pointed to by the internal array pointer */ PHP_FUNCTION(key) { pval **array; char *string_key; ulong num_key; HashTable *target_hash; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Variable passed to key() is not an array or object"); RETURN_FALSE; } switch (zend_hash_get_current_key(target_hash, &string_key, &num_key, 1)) { case HASH_KEY_IS_STRING: RETVAL_STRING(string_key, 0); break; case HASH_KEY_IS_LONG: RETVAL_LONG(num_key); break; case HASH_KEY_NON_EXISTANT: return; } } /* }}} */ /* {{{ proto mixed min(mixed arg1 [, mixed arg2 [, mixed ...]]) Return the lowest value in an array or a series of arguments */ PHP_FUNCTION(min) { int argc=ZEND_NUM_ARGS(); pval **result; if (argc<=0) { php_error(E_WARNING, "min: must be passed at least 1 value"); RETURN_NULL(); } set_compare_func(SORT_REGULAR); if (argc == 1) { pval **arr; if (zend_get_parameters_ex(1, &arr) == FAILURE || Z_TYPE_PP(arr) != IS_ARRAY) { WRONG_PARAM_COUNT; } if (zend_hash_minmax(Z_ARRVAL_PP(arr), array_data_compare, 0, (void **) &result)==SUCCESS) { *return_value = **result; zval_copy_ctor(return_value); } else { php_error(E_WARNING, "min: array must contain at least 1 element"); RETURN_FALSE; } } else { pval ***args = (pval ***) emalloc(sizeof(pval **)*ZEND_NUM_ARGS()); pval **min, result; int i; if (zend_get_parameters_array_ex(ZEND_NUM_ARGS(), args)==FAILURE) { efree(args); WRONG_PARAM_COUNT; } min = args[0]; for (i=1; ivalue.str.val); zend_hash_move_forward_ex(target_hash, &pos); } efree(key); return 0; } /* {{{ proto int array_walk(array input, string funcname [, mixed userdata]) Apply a user function to every member of an array */ PHP_FUNCTION(array_walk) { int argc; zval **array, **userdata = NULL, **old_walk_func_name; HashTable *target_hash; BLS_FETCH(); argc = ZEND_NUM_ARGS(); old_walk_func_name = BG(array_walk_func_name); if (argc < 2 || argc > 3 || zend_get_parameters_ex(argc, &array, &BG(array_walk_func_name), &userdata) == FAILURE) { BG(array_walk_func_name) = old_walk_func_name; WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in %s() call", get_active_function_name()); BG(array_walk_func_name) = old_walk_func_name; RETURN_FALSE; } if (Z_TYPE_PP(BG(array_walk_func_name)) != IS_ARRAY && Z_TYPE_PP(BG(array_walk_func_name)) != IS_STRING) { php_error(E_WARNING, "Wrong syntax for function name in %s() call", get_active_function_name()); BG(array_walk_func_name) = old_walk_func_name; RETURN_FALSE; } php_array_walk(target_hash, userdata); BG(array_walk_func_name) = old_walk_func_name; RETURN_TRUE; } /* }}} */ /* void php_search_array(INTERNAL_FUNCTION_PARAMETERS, int behavior) * 0 = return boolean * 1 = return key */ static void php_search_array(INTERNAL_FUNCTION_PARAMETERS, int behavior) { zval **value, /* value to check for */ **array, /* array to check in */ **strict, /* strict comparison or not */ **entry, /* pointer to array entry */ res; /* comparison result */ HashTable *target_hash; /* array hashtable */ HashPosition pos; /* hash iterator */ ulong num_key; char *string_key; int (*compare_func)(zval *, zval *, zval *) = is_equal_function; if (ZEND_NUM_ARGS() < 2 || ZEND_NUM_ARGS() > 3 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &value, &array, &strict) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(value) == IS_ARRAY || Z_TYPE_PP(value) == IS_OBJECT) { php_error(E_WARNING, "Wrong datatype for first argument in call to %s", get_active_function_name()); RETURN_FALSE; } if (Z_TYPE_PP(array) != IS_ARRAY) { php_error(E_WARNING, "Wrong datatype for second argument in call to %s", get_active_function_name()); RETURN_FALSE; } if (ZEND_NUM_ARGS() == 3) { convert_to_boolean_ex(strict); if (Z_LVAL_PP(strict)) { compare_func = is_identical_function; } } target_hash = HASH_OF(*array); zend_hash_internal_pointer_reset_ex(target_hash, &pos); while(zend_hash_get_current_data_ex(target_hash, (void **)&entry, &pos) == SUCCESS) { compare_func(&res, *value, *entry); if (Z_LVAL(res) == 1) { if (behavior==0) { RETURN_TRUE; } else { /* Return current key */ switch (zend_hash_get_current_key_ex(target_hash, &string_key, NULL, &num_key, 1, &pos)) { case HASH_KEY_IS_STRING: RETVAL_STRING(string_key, 0); break; case HASH_KEY_IS_LONG: RETVAL_LONG(num_key); break; } } } zend_hash_move_forward_ex(target_hash, &pos); } if (behavior == 0) { RETURN_FALSE; } else { return; } } /* {{{ proto bool in_array(mixed needle, array haystack [, bool strict]) Checks if the given value exists in the array */ PHP_FUNCTION(in_array) { php_search_array(INTERNAL_FUNCTION_PARAM_PASSTHRU, 0); } /* }}} */ /* {{{ proto mixed array_search(mixed needle, array haystack [, bool strict]) Searches the array for a given value and returns the corresponding key if successful */ PHP_FUNCTION(array_search) { php_search_array(INTERNAL_FUNCTION_PARAM_PASSTHRU, 1); } /* }}} */ static int php_valid_var_name(char *var_name) { int len, i; if (!var_name) return 0; len = strlen(var_name); if (!isalpha((int)var_name[0]) && var_name[0] != '_') return 0; if (len > 1) { for(i=1; i EXTR_SKIP && extract_type <= EXTR_PREFIX_INVALID) { php_error(E_WARNING, "%s() expects a prefix to be specified", get_active_function_name()); return; } break; case 3: if (zend_get_parameters_ex(3, &var_array, &z_extract_type, &prefix) == FAILURE) { WRONG_PARAM_COUNT; } convert_to_long_ex(z_extract_type); extract_type = Z_LVAL_PP(z_extract_type); convert_to_string_ex(prefix); break; default: WRONG_PARAM_COUNT; break; } if (extract_type < EXTR_OVERWRITE || extract_type > EXTR_PREFIX_INVALID) { php_error(E_WARNING, "Unknown extract type in call to %s()", get_active_function_name()); return; } if (Z_TYPE_PP(var_array) != IS_ARRAY) { php_error(E_WARNING, "%s() expects first argument to be an array", get_active_function_name()); return; } zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(var_array), &pos); while(zend_hash_get_current_data_ex(Z_ARRVAL_PP(var_array), (void **)&entry, &pos) == SUCCESS) { key_type = zend_hash_get_current_key_ex(Z_ARRVAL_PP(var_array), &var_name, &var_name_len, &num_key, 0, &pos); final_name = NULL; var_exists = 0; if (key_type == HASH_KEY_IS_STRING) { var_name_len--; var_exists = zend_hash_exists(EG(active_symbol_table), var_name, var_name_len + 1); } else if (extract_type == EXTR_PREFIX_ALL || extract_type == EXTR_PREFIX_INVALID) { final_name = emalloc(MAX_LENGTH_OF_LONG + Z_STRLEN_PP(prefix) + 2); zend_sprintf(final_name, "%s_%ld", Z_STRVAL_PP(prefix), num_key); } else { zend_hash_move_forward_ex(Z_ARRVAL_PP(var_array), &pos); continue; } switch (extract_type) { case EXTR_OVERWRITE: final_name = estrndup(var_name, var_name_len); break; case EXTR_PREFIX_SAME: if (!var_exists) final_name = estrndup(var_name, var_name_len); /* break omitted intentionally */ case EXTR_PREFIX_ALL: if (!final_name) { final_name = emalloc(var_name_len + Z_STRLEN_PP(prefix) + 2); strcpy(final_name, Z_STRVAL_PP(prefix)); strcat(final_name, "_"); strcat(final_name, var_name); } break; case EXTR_PREFIX_INVALID: if (!final_name) { if (!php_valid_var_name(var_name)) { final_name = emalloc(var_name_len + Z_STRLEN_PP(prefix) + 2); strcpy(final_name, Z_STRVAL_PP(prefix)); strcat(final_name, "_"); strcat(final_name, var_name); } else final_name = estrndup(var_name, var_name_len); } break; default: if (!var_exists) final_name = estrndup(var_name, var_name_len); break; } if (final_name) { if (php_valid_var_name(final_name)) { MAKE_STD_ZVAL(data); *data = **entry; zval_copy_ctor(data); ZEND_SET_SYMBOL(EG(active_symbol_table), final_name, data); count++; } efree(final_name); } zend_hash_move_forward_ex(Z_ARRVAL_PP(var_array), &pos); } RETURN_LONG(count); } /* }}} */ static void php_compact_var(HashTable *eg_active_symbol_table, zval *return_value, zval *entry) { zval **value_ptr, *value, *data; if (Z_TYPE_P(entry) == IS_STRING) { if (zend_hash_find(eg_active_symbol_table, Z_STRVAL_P(entry), Z_STRLEN_P(entry)+1, (void **)&value_ptr) != FAILURE) { value = *value_ptr; ALLOC_ZVAL(data); *data = *value; zval_copy_ctor(data); INIT_PZVAL(data); zend_hash_update(Z_ARRVAL_P(return_value), Z_STRVAL_P(entry), Z_STRLEN_P(entry)+1, &data, sizeof(zval *), NULL); } } else if (Z_TYPE_P(entry) == IS_ARRAY) { HashPosition pos; zend_hash_internal_pointer_reset_ex(Z_ARRVAL_P(entry), &pos); while (zend_hash_get_current_data_ex(Z_ARRVAL_P(entry), (void**)&value_ptr, &pos) == SUCCESS) { value = *value_ptr; php_compact_var(eg_active_symbol_table, return_value, value); zend_hash_move_forward_ex(Z_ARRVAL_P(entry), &pos); } } } /* {{{ proto array compact(mixed var_names [, mixed ...]) Creates a hash containing variables and their values */ PHP_FUNCTION(compact) { zval ***args; /* function arguments array */ int i; args = (zval ***)emalloc(ZEND_NUM_ARGS() * sizeof(zval **)); if (zend_get_parameters_array_ex(ZEND_NUM_ARGS(), args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } array_init(return_value); for (i=0; i*high) { for (; *low >= *high; (*low)--) { add_next_index_stringl(return_value, low, 1, 1); } } else { for (; *low <= *high; (*low)++) { add_next_index_stringl(return_value, low, 1, 1); } } } else { int low, high; convert_to_long_ex(zlow); convert_to_long_ex(zhigh); low = Z_LVAL_PP(zlow); high = Z_LVAL_PP(zhigh); if(low>high) { for (; low >= high; low--) { add_next_index_long(return_value, low); } } else { for (; low <= high; low++) { add_next_index_long(return_value, low); } } } } /* }}} */ static int array_data_shuffle(const void *a, const void*b) { return (php_rand() % 2) ? 1 : -1; } /* {{{ proto int shuffle(array array_arg) Randomly shuffle the contents of an array */ PHP_FUNCTION(shuffle) { zval **array; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(array) != IS_ARRAY) { php_error(E_WARNING, "Wrong datatype in shuffle() call"); RETURN_FALSE; } if (zend_hash_sort(Z_ARRVAL_PP(array), (sort_func_t)php_mergesort, array_data_shuffle, 1) == FAILURE) { RETURN_FALSE; } RETURN_TRUE; } /* }}} */ /* HashTable* php_splice(HashTable *in_hash, int offset, int length, zval ***list, int list_count, HashTable **removed) */ HashTable* php_splice(HashTable *in_hash, int offset, int length, zval ***list, int list_count, HashTable **removed) { HashTable *out_hash = NULL; /* Output hashtable */ int num_in, /* Number of entries in the input hashtable */ pos, /* Current position in the hashtable */ i; /* Loop counter */ Bucket *p; /* Pointer to hash bucket */ zval *entry; /* Hash entry */ /* If input hash doesn't exist, we have nothing to do */ if (!in_hash) return NULL; /* Get number of entries in the input hash */ num_in = zend_hash_num_elements(in_hash); /* Clamp the offset.. */ if (offset > num_in) offset = num_in; else if (offset < 0 && (offset=num_in+offset) < 0) offset = 0; /* ..and the length */ if (length < 0) length = num_in-offset+length; else if(offset+length > num_in) length = num_in-offset; /* Create and initialize output hash */ ALLOC_HASHTABLE(out_hash); zend_hash_init(out_hash, 0, NULL, ZVAL_PTR_DTOR, 0); /* Start at the beginning of the input hash and copy entries to output hash until offset is reached */ for (pos=0, p=in_hash->pListHead; pospListNext) { /* Get entry and increase reference count */ entry = *((zval **)p->pData); entry->refcount++; /* Update output hash depending on key type */ if (p->nKeyLength) zend_hash_update(out_hash, p->arKey, p->nKeyLength, &entry, sizeof(zval *), NULL); else zend_hash_next_index_insert(out_hash, &entry, sizeof(zval *), NULL); } /* If hash for removed entries exists, go until offset+length and copy the entries to it */ if (removed != NULL) { for( ; pospListNext) { entry = *((zval **)p->pData); entry->refcount++; if (p->nKeyLength) zend_hash_update(*removed, p->arKey, p->nKeyLength, &entry, sizeof(zval *), NULL); else zend_hash_next_index_insert(*removed, &entry, sizeof(zval *), NULL); } } else /* otherwise just skip those entries */ for( ; pospListNext); /* If there are entries to insert.. */ if (list != NULL) { /* ..for each one, create a new zval, copy entry into it and copy it into the output hash */ for (i=0; irefcount++; zend_hash_next_index_insert(out_hash, &entry, sizeof(zval *), NULL); } } /* Copy the remaining input hash entries to the output hash */ for ( ; p ; p=p->pListNext) { entry = *((zval **)p->pData); entry->refcount++; if (p->nKeyLength) zend_hash_update(out_hash, p->arKey, p->nKeyLength, &entry, sizeof(zval *), NULL); else zend_hash_next_index_insert(out_hash, &entry, sizeof(zval *), NULL); } zend_hash_internal_pointer_reset(out_hash); return out_hash; } /* }}} */ /* {{{ proto int array_push(array stack, mixed var [, mixed ...]) Pushes elements onto the end of the array */ PHP_FUNCTION(array_push) { zval ***args, /* Function arguments array */ *stack, /* Input array */ *new_var; /* Variable to be pushed */ int i, /* Loop counter */ argc; /* Number of function arguments */ /* Get the argument count and check it */ argc = ZEND_NUM_ARGS(); if (argc < 2) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } /* Get first argument and check that it's an array */ stack = *args[0]; if (Z_TYPE_P(stack) != IS_ARRAY) { php_error(E_WARNING, "First argument to array_push() needs to be an array"); efree(args); RETURN_FALSE; } /* For each subsequent argument, make it a reference, increase refcount, and add it to the end of the array */ for (i=1; irefcount++; zend_hash_next_index_insert(Z_ARRVAL_P(stack), &new_var, sizeof(zval *), NULL); } /* Clean up and return the number of values in the stack */ efree(args); RETVAL_LONG(zend_hash_num_elements(Z_ARRVAL_P(stack))); } /* }}} */ /* {{{ void _phpi_pop(INTERNAL_FUNCTION_PARAMETERS, int which_end) */ static void _phpi_pop(INTERNAL_FUNCTION_PARAMETERS, int off_the_end) { zval **stack, /* Input stack */ **val; /* Value to be popped */ HashTable *new_hash; /* New stack */ /* Get the arguments and do error-checking */ if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &stack) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(stack) != IS_ARRAY) { php_error(E_WARNING, "The argument needs to be an array"); return; } if (zend_hash_num_elements(Z_ARRVAL_PP(stack)) == 0) { return; } /* Get the first or last value and copy it into the return value */ if (off_the_end) zend_hash_internal_pointer_end(Z_ARRVAL_PP(stack)); else zend_hash_internal_pointer_reset(Z_ARRVAL_PP(stack)); zend_hash_get_current_data(Z_ARRVAL_PP(stack), (void **)&val); *return_value = **val; zval_copy_ctor(return_value); INIT_PZVAL(return_value); /* Delete the first or last value */ new_hash = php_splice(Z_ARRVAL_PP(stack), (off_the_end) ? -1 : 0, 1, NULL, 0, NULL); zend_hash_destroy(Z_ARRVAL_PP(stack)); efree(Z_ARRVAL_PP(stack)); Z_ARRVAL_PP(stack) = new_hash; } /* }}} */ /* {{{ proto mixed array_pop(array stack) Pops an element off the end of the array */ PHP_FUNCTION(array_pop) { _phpi_pop(INTERNAL_FUNCTION_PARAM_PASSTHRU, 1); } /* }}} */ /* {{{ proto mixed array_shift(array stack) Pops an element off the beginning of the array */ PHP_FUNCTION(array_shift) { _phpi_pop(INTERNAL_FUNCTION_PARAM_PASSTHRU, 0); } /* }}} */ /* {{{ proto int array_unshift(array stack, mixed var [, mixed ...]) Pushes elements onto the beginning of the array */ PHP_FUNCTION(array_unshift) { zval ***args, /* Function arguments array */ *stack; /* Input stack */ HashTable *new_hash; /* New hashtable for the stack */ int argc; /* Number of function arguments */ /* Get the argument count and check it */ argc = ZEND_NUM_ARGS(); if (argc < 2) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } /* Get first argument and check that it's an array */ stack = *args[0]; if (Z_TYPE_P(stack) != IS_ARRAY) { php_error(E_WARNING, "First argument to array_unshift() needs to be an array"); efree(args); RETURN_FALSE; } /* Use splice to insert the elements at the beginning. Destroy old hashtable and replace it with new one */ new_hash = php_splice(Z_ARRVAL_P(stack), 0, 0, &args[1], argc-1, NULL); zend_hash_destroy(Z_ARRVAL_P(stack)); efree(Z_ARRVAL_P(stack)); Z_ARRVAL_P(stack) = new_hash; /* Clean up and return the number of elements in the stack */ efree(args); RETVAL_LONG(zend_hash_num_elements(Z_ARRVAL_P(stack))); } /* }}} */ /* {{{ proto array array_splice(array input, int offset [, int length [, array replacement]]) Removes the elements designated by offset and length and replace them with supplied array */ PHP_FUNCTION(array_splice) { zval ***args, /* Function arguments array */ *array, /* Input array */ ***repl = NULL; /* Replacement elements */ HashTable *new_hash = NULL; /* Output array's hash */ Bucket *p; /* Bucket used for traversing hash */ int argc, /* Number of function arguments */ i, offset, length, repl_num = 0; /* Number of replacement elements */ /* Get the argument count and check it */ argc = ZEND_NUM_ARGS(); if (argc < 2 || argc > 4) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } /* Get first argument and check that it's an array */ array = *args[0]; if (Z_TYPE_P(array) != IS_ARRAY) { php_error(E_WARNING, "First argument to array_splice() should be an array"); efree(args); return; } /* Get the next two arguments. If length is omitted, it's assumed to be until the end of the array */ convert_to_long_ex(args[1]); offset = Z_LVAL_PP(args[1]); if (argc > 2) { convert_to_long_ex(args[2]); length = Z_LVAL_PP(args[2]); } else length = zend_hash_num_elements(Z_ARRVAL_P(array)); if (argc == 4) { /* Make sure the last argument, if passed, is an array */ convert_to_array_ex(args[3]); /* Create the array of replacement elements */ repl_num = zend_hash_num_elements(Z_ARRVAL_PP(args[3])); repl = (zval ***)emalloc(repl_num * sizeof(zval **)); for (p=Z_ARRVAL_PP(args[3])->pListHead, i=0; p; p=p->pListNext, i++) { repl[i] = ((zval **)p->pData); } } /* Initialize return value */ array_init(return_value); /* Perform splice */ new_hash = php_splice(Z_ARRVAL_P(array), offset, length, repl, repl_num, &Z_ARRVAL_P(return_value)); /* Replace input array's hashtable with the new one */ zend_hash_destroy(Z_ARRVAL_P(array)); efree(Z_ARRVAL_P(array)); Z_ARRVAL_P(array) = new_hash; /* Clean up */ if (argc == 4) efree(repl); efree(args); } /* }}} */ /* {{{ proto array array_slice(array input, int offset [, int length]) Returns elements specified by offset and length */ PHP_FUNCTION(array_slice) { zval **input, /* Input array */ **offset, /* Offset to get elements from */ **length, /* How many elements to get */ **entry; /* An array entry */ int offset_val, /* Value of the offset argument */ length_val, /* Value of the length argument */ num_in, /* Number of elements in the input array */ pos, /* Current position in the array */ argc; /* Number of function arguments */ char *string_key; ulong string_key_len; ulong num_key; HashPosition hpos; /* Get the arguments and do error-checking */ argc = ZEND_NUM_ARGS(); if (argc < 2 || argc > 3 || zend_get_parameters_ex(argc, &input, &offset, &length)) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "First argument to array_slice() should be an array"); return; } /* Make sure offset and length are integers and assume we want all entries from offset to the end if length is not passed */ convert_to_long_ex(offset); offset_val = Z_LVAL_PP(offset); if (argc == 3) { convert_to_long_ex(length); length_val = Z_LVAL_PP(length); } else length_val = zend_hash_num_elements(Z_ARRVAL_PP(input)); /* Initialize returned array */ array_init(return_value); /* Get number of entries in the input hash */ num_in = zend_hash_num_elements(Z_ARRVAL_PP(input)); /* Clamp the offset.. */ if (offset_val > num_in) return; else if (offset_val < 0 && (offset_val=num_in+offset_val) < 0) offset_val = 0; /* ..and the length */ if (length_val < 0) length_val = num_in-offset_val+length_val; else if(offset_val+length_val > num_in) length_val = num_in-offset_val; if (length_val == 0) return; /* Start at the beginning and go until we hit offset */ pos = 0; zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &hpos); while(pos < offset_val && zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&entry, &hpos) == SUCCESS) { pos++; zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &hpos); } /* Copy elements from input array to the one that's returned */ while(pos < offset_val+length_val && zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&entry, &hpos) == SUCCESS) { (*entry)->refcount++; switch (zend_hash_get_current_key_ex(Z_ARRVAL_PP(input), &string_key, &string_key_len, &num_key, 0, &hpos)) { case HASH_KEY_IS_STRING: zend_hash_update(Z_ARRVAL_P(return_value), string_key, string_key_len, entry, sizeof(zval *), NULL); break; case HASH_KEY_IS_LONG: zend_hash_next_index_insert(Z_ARRVAL_P(return_value), entry, sizeof(zval *), NULL); break; } pos++; zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &hpos); } } /* }}} */ PHPAPI void php_array_merge(HashTable *dest, HashTable *src, int recursive) { zval **src_entry, **dest_entry; char *string_key; ulong string_key_len; ulong num_key; HashPosition pos; zend_hash_internal_pointer_reset_ex(src, &pos); while(zend_hash_get_current_data_ex(src, (void **)&src_entry, &pos) == SUCCESS) { switch (zend_hash_get_current_key_ex(src, &string_key, &string_key_len, &num_key, 0, &pos)) { case HASH_KEY_IS_STRING: if (recursive && zend_hash_find(dest, string_key, string_key_len, (void **)&dest_entry) == SUCCESS) { convert_to_array_ex(dest_entry); convert_to_array_ex(src_entry); php_array_merge(Z_ARRVAL_PP(dest_entry), Z_ARRVAL_PP(src_entry), recursive); } else { (*src_entry)->refcount++; zend_hash_update(dest, string_key, strlen(string_key)+1, src_entry, sizeof(zval *), NULL); } break; case HASH_KEY_IS_LONG: (*src_entry)->refcount++; zend_hash_next_index_insert(dest, src_entry, sizeof(zval *), NULL); break; } zend_hash_move_forward_ex(src, &pos); } } static void php_array_merge_wrapper(INTERNAL_FUNCTION_PARAMETERS, int recursive) { zval ***args = NULL; int argc, i; /* Get the argument count and check it */ argc = ZEND_NUM_ARGS(); if (argc < 2) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } array_init(return_value); for (i=0; i 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &input, &search_value) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "First argument to array_keys() should be an array"); return; } /* Initialize return array */ array_init(return_value); add_key = 1; /* Go through input array and add keys to the return array */ zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &pos); while(zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&entry, &pos) == SUCCESS) { if (search_value != NULL) { is_equal_function(&res, *search_value, *entry); add_key = zval_is_true(&res); } if (add_key) { MAKE_STD_ZVAL(new_val); switch (zend_hash_get_current_key_ex(Z_ARRVAL_PP(input), &string_key, &string_key_len, &num_key, 1, &pos)) { case HASH_KEY_IS_STRING: Z_TYPE_P(new_val) = IS_STRING; Z_STRVAL_P(new_val) = string_key; Z_STRLEN_P(new_val) = string_key_len-1; zend_hash_next_index_insert(Z_ARRVAL_P(return_value), &new_val, sizeof(zval *), NULL); break; case HASH_KEY_IS_LONG: Z_TYPE_P(new_val) = IS_LONG; Z_LVAL_P(new_val) = num_key; zend_hash_next_index_insert(Z_ARRVAL_P(return_value), &new_val, sizeof(zval *), NULL); break; } } zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &pos); } } /* }}} */ /* {{{ proto array array_values(array input) Return just the values from the input array */ PHP_FUNCTION(array_values) { zval **input, /* Input array */ **entry; /* An entry in the input array */ HashPosition pos; /* Get arguments and do error-checking */ if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &input) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "Argument to array_values() should be an array"); return; } /* Initialize return array */ array_init(return_value); /* Go through input array and add values to the return array */ zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &pos); while(zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&entry, &pos) == SUCCESS) { (*entry)->refcount++; zend_hash_next_index_insert(Z_ARRVAL_P(return_value), entry, sizeof(zval *), NULL); zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &pos); } } /* }}} */ /* {{{ proto array array_count_values(array input) Return the value as key and the frequency of that value in input as value */ PHP_FUNCTION(array_count_values) { zval **input, /* Input array */ **entry; /* An entry in the input array */ zval **tmp; HashTable *myht; HashPosition pos; /* Get arguments and do error-checking */ if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &input) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "Argument to array_count_values() should be an array"); return; } /* Initialize return array */ array_init(return_value); /* Go through input array and add values to the return array */ myht = Z_ARRVAL_PP(input); zend_hash_internal_pointer_reset_ex(myht, &pos); while (zend_hash_get_current_data_ex(myht, (void **)&entry, &pos) == SUCCESS) { if (Z_TYPE_PP(entry) == IS_LONG) { if (zend_hash_index_find(Z_ARRVAL_P(return_value), Z_LVAL_PP(entry), (void**)&tmp) == FAILURE) { zval *data; MAKE_STD_ZVAL(data); Z_TYPE_P(data) = IS_LONG; Z_LVAL_P(data) = 1; zend_hash_index_update(Z_ARRVAL_P(return_value),Z_LVAL_PP(entry), &data, sizeof(data), NULL); } else { Z_LVAL_PP(tmp)++; } } else if (Z_TYPE_PP(entry) == IS_STRING) { if (zend_hash_find(Z_ARRVAL_P(return_value), Z_STRVAL_PP(entry), Z_STRLEN_PP(entry)+1, (void**)&tmp) == FAILURE) { zval *data; MAKE_STD_ZVAL(data); Z_TYPE_P(data) = IS_LONG; Z_LVAL_P(data) = 1; zend_hash_update(Z_ARRVAL_P(return_value),Z_STRVAL_PP(entry),Z_STRLEN_PP(entry) + 1, &data, sizeof(data), NULL); } else { Z_LVAL_PP(tmp)++; } } else { php_error(E_WARNING, "Can only count STRING and INTEGER values!"); } zend_hash_move_forward_ex(myht, &pos); } } /* }}} */ /* {{{ proto array array_reverse(array input [, bool preserve keys]) Return input as a new array with the order of the entries reversed */ PHP_FUNCTION(array_reverse) { zval **input, /* Input array */ **z_preserve_keys, /* Flag: whether to preserve keys */ **entry; /* An entry in the input array */ char *string_key; ulong string_key_len; ulong num_key; zend_bool preserve_keys = 0; HashPosition pos; /* Get arguments and do error-checking */ if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &input, &z_preserve_keys) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "Argument to array_reverse() should be an array"); return; } if (ZEND_NUM_ARGS() > 1) { convert_to_boolean_ex(z_preserve_keys); preserve_keys = Z_BVAL_PP(z_preserve_keys); } /* Initialize return array */ array_init(return_value); zend_hash_internal_pointer_end_ex(Z_ARRVAL_PP(input), &pos); while(zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&entry, &pos) == SUCCESS) { (*entry)->refcount++; switch (zend_hash_get_current_key_ex(Z_ARRVAL_PP(input), &string_key, &string_key_len, &num_key, 0, &pos)) { case HASH_KEY_IS_STRING: zend_hash_update(Z_ARRVAL_P(return_value), string_key, string_key_len, entry, sizeof(zval *), NULL); break; case HASH_KEY_IS_LONG: if (preserve_keys) zend_hash_index_update(Z_ARRVAL_P(return_value), num_key, entry, sizeof(zval *), NULL); else zend_hash_next_index_insert(Z_ARRVAL_P(return_value), entry, sizeof(zval *), NULL); break; } zend_hash_move_backwards_ex(Z_ARRVAL_PP(input), &pos); } } /* }}} */ /* {{{ proto array array_pad(array input, int pad_size, mixed pad_value) Returns a copy of input array padded with pad_value to size pad_size */ PHP_FUNCTION(array_pad) { zval **input; /* Input array */ zval **pad_size; /* Size to pad to */ zval **pad_value; /* Padding value obviously */ zval ***pads; /* Array to pass to splice */ HashTable *new_hash; /* Return value from splice */ int input_size; /* Size of the input array */ int pad_size_abs; /* Absolute value of pad_size */ int num_pads; /* How many pads do we need */ int do_pad; /* Whether we should do padding at all */ int i; /* Get arguments and do error-checking */ if (ZEND_NUM_ARGS() != 3 || zend_get_parameters_ex(3, &input, &pad_size, &pad_value) == FAILURE) { WRONG_PARAM_COUNT; } /* Make sure arguments are of the proper type */ if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "Argument to %s() should be an array", get_active_function_name()); return; } convert_to_long_ex(pad_size); /* Do some initial calculations */ input_size = zend_hash_num_elements(Z_ARRVAL_PP(input)); pad_size_abs = abs(Z_LVAL_PP(pad_size)); do_pad = (input_size >= pad_size_abs) ? 0 : 1; /* Copy the original array */ *return_value = **input; zval_copy_ctor(return_value); /* If no need to pad, no need to continue */ if (!do_pad) return; /* Populate the pads array */ num_pads = pad_size_abs - input_size; pads = (zval ***)emalloc(num_pads * sizeof(zval **)); for (i = 0; i < num_pads; i++) pads[i] = pad_value; /* Pad on the right or on the left */ if (Z_LVAL_PP(pad_size) > 0) new_hash = php_splice(Z_ARRVAL_P(return_value), input_size, 0, pads, num_pads, NULL); else new_hash = php_splice(Z_ARRVAL_P(return_value), 0, 0, pads, num_pads, NULL); /* Copy the result hash into return value */ zend_hash_destroy(Z_ARRVAL_P(return_value)); efree(Z_ARRVAL_P(return_value)); Z_ARRVAL_P(return_value) = new_hash; /* Clean up */ efree(pads); } /* }}} */ /* {{{ proto array array_flip(array input) Return array with key <-> value flipped */ PHP_FUNCTION(array_flip) { zval **array, **entry, *data; HashTable *target_hash; char *string_key; ulong str_key_len; ulong num_key; HashPosition pos; if (ZEND_NUM_ARGS() != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in array_flip() call"); RETURN_FALSE; } array_init(return_value); zend_hash_internal_pointer_reset_ex(target_hash, &pos); while (zend_hash_get_current_data_ex(target_hash, (void **)&entry, &pos) == SUCCESS) { MAKE_STD_ZVAL(data); switch (zend_hash_get_current_key_ex(target_hash, &string_key, &str_key_len, &num_key, 1, &pos)) { case HASH_KEY_IS_STRING: Z_STRVAL_P(data) = string_key; Z_STRLEN_P(data) = str_key_len-1; Z_TYPE_P(data) = IS_STRING; break; case HASH_KEY_IS_LONG: Z_TYPE_P(data) = IS_LONG; Z_LVAL_P(data) = num_key; break; } if (Z_TYPE_PP(entry) == IS_LONG) { zend_hash_index_update(Z_ARRVAL_P(return_value),Z_LVAL_PP(entry), &data, sizeof(data), NULL); } else if (Z_TYPE_PP(entry) == IS_STRING) { zend_hash_update(Z_ARRVAL_P(return_value),Z_STRVAL_PP(entry),Z_STRLEN_PP(entry) + 1, &data, sizeof(data), NULL); } else { zval_ptr_dtor(&data); /* will free also zval structure */ php_error(E_WARNING, "Can only flip STRING and INTEGER values!"); } zend_hash_move_forward_ex(target_hash, &pos); } } /* }}} */ /* {{{ proto array array_unique(array input) Removes duplicate values from array */ PHP_FUNCTION(array_unique) { zval **array; HashTable *target_hash; Bucket **arTmp, **cmpdata, **lastkept; Bucket *p; int i; if (ARG_COUNT(ht) != 1 || zend_get_parameters_ex(1, &array) == FAILURE) { WRONG_PARAM_COUNT; } target_hash = HASH_OF(*array); if (!target_hash) { php_error(E_WARNING, "Wrong datatype in array_unique() call"); RETURN_FALSE; } /* copy the argument array */ *return_value = **array; zval_copy_ctor(return_value); if (target_hash->nNumOfElements <= 1) /* nothing to do */ return; /* create and sort array with pointers to the target_hash buckets */ arTmp = (Bucket **) pemalloc((target_hash->nNumOfElements + 1) * sizeof(Bucket *), target_hash->persistent); if (!arTmp) RETURN_FALSE; for (i = 0, p = target_hash->pListHead; p; i++, p = p->pListNext) arTmp[i] = p; arTmp[i] = NULL; set_compare_func(SORT_STRING); qsort((void *) arTmp, i, sizeof(Bucket *), array_data_compare); /* go through the sorted array and delete duplicates from the copy */ lastkept = arTmp; for (cmpdata = arTmp + 1; *cmpdata; cmpdata++) { if (array_data_compare(lastkept, cmpdata)) { lastkept = cmpdata; } else { p = *cmpdata; if (p->nKeyLength) zend_hash_del(Z_ARRVAL_P(return_value), p->arKey, p->nKeyLength); else zend_hash_index_del(Z_ARRVAL_P(return_value), p->h); } } pefree(arTmp, target_hash->persistent); } /* }}} */ /* {{{ proto array array_intersect(array arr1, array arr2 [, array ...]) Returns the entries of arr1 that have values which are present in all the other arguments */ PHP_FUNCTION(array_intersect) { zval ***args = NULL; HashTable *hash; int argc, i, c = 0; Bucket ***lists, **list, ***ptrs, *p; /* Get the argument count and check it */ argc = ARG_COUNT(ht); if (argc < 2) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } /* for each argument, create and sort list with pointers to the hash buckets */ lists = (Bucket ***)emalloc(argc * sizeof(Bucket **)); ptrs = (Bucket ***)emalloc(argc * sizeof(Bucket **)); set_compare_func(SORT_STRING); for (i=0; inNumOfElements + 1) * sizeof(Bucket *), hash->persistent); if (!list) RETURN_FALSE; lists[i] = list; ptrs[i] = list; for (p = hash->pListHead; p; p = p->pListNext) *list++ = p; *list = NULL; qsort((void *) lists[i], hash->nNumOfElements, sizeof(Bucket *), array_data_compare); } /* copy the argument array */ *return_value = **args[0]; zval_copy_ctor(return_value); /* go through the lists and look for common values */ while (*ptrs[0]) { for (i=1; inKeyLength) zend_hash_del(Z_ARRVAL_P(return_value), p->arKey, p->nKeyLength); else zend_hash_index_del(Z_ARRVAL_P(return_value), p->h); } } if (c) break; ptrs[i]++; } if (c) { /* Value of ptrs[0] not in all arguments, delete all entries */ /* with value < value of ptrs[i] */ for (;;) { p = *ptrs[0]; if (p->nKeyLength) zend_hash_del(Z_ARRVAL_P(return_value), p->arKey, p->nKeyLength); else zend_hash_index_del(Z_ARRVAL_P(return_value), p->h); if (!*++ptrs[0]) goto out; if (0 <= array_data_compare(ptrs[0], ptrs[i])) break; } } else { /* ptrs[0] is present in all the arguments */ /* Skip all entries with same value as ptrs[0] */ for (;;) { if (!*++ptrs[0]) goto out; if (array_data_compare(ptrs[0]-1, ptrs[0])) break; } } } out: for (i=0; ipersistent); } efree(ptrs); efree(lists); efree(args); } /* }}} */ /* {{{ proto array array_diff(array arr1, array arr2 [, array ...]) Returns the entries of arr1 that have values which are not present in any of the others arguments */ PHP_FUNCTION(array_diff) { zval ***args = NULL; HashTable *hash; int argc, i, c; Bucket ***lists, **list, ***ptrs, *p; /* Get the argument count and check it */ argc = ARG_COUNT(ht); if (argc < 2) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } /* for each argument, create and sort list with pointers to the hash buckets */ lists = (Bucket ***)emalloc(argc * sizeof(Bucket **)); ptrs = (Bucket ***)emalloc(argc * sizeof(Bucket **)); set_compare_func(SORT_STRING); for (i=0; inNumOfElements + 1) * sizeof(Bucket *), hash->persistent); if (!list) RETURN_FALSE; lists[i] = list; ptrs[i] = list; for (p = hash->pListHead; p; p = p->pListNext) *list++ = p; *list = NULL; qsort((void *) lists[i], hash->nNumOfElements, sizeof(Bucket *), array_data_compare); } /* copy the argument array */ *return_value = **args[0]; zval_copy_ctor(return_value); /* go through the lists and look for values of ptr[0] that are not in the others */ while (*ptrs[0]) { c = 1; for (i=1; inKeyLength) zend_hash_del(Z_ARRVAL_P(return_value), p->arKey, p->nKeyLength); else zend_hash_index_del(Z_ARRVAL_P(return_value), p->h); if (!*++ptrs[0]) goto out; if (array_data_compare(ptrs[0]-1, ptrs[0])) break; } } else { /* ptrs[0] in none of the other arguments */ /* skip all entries with value as ptrs[0] */ for (;;) { if (!*++ptrs[0]) goto out; if (array_data_compare(ptrs[0]-1, ptrs[0])) break; } } } out: for (i=0; ipersistent); } efree(ptrs); efree(lists); efree(args); } /* }}} */ #define MULTISORT_ORDER 0 #define MULTISORT_TYPE 1 #define MULTISORT_LAST 2 int multisort_compare(const void *a, const void *b) { Bucket** ab = *(Bucket ***)a; Bucket** bb = *(Bucket ***)b; int r; int result = 0; zval temp; ARRAYLS_FETCH(); r = 0; do { set_compare_func(ARRAYG(multisort_flags)[MULTISORT_TYPE][r]); ARRAYG(compare_func)(&temp, *((zval **)ab[r]->pData), *((zval **)bb[r]->pData)); result = ARRAYG(multisort_flags)[MULTISORT_ORDER][r] * Z_LVAL(temp); if (result != 0) return result; r++; } while (ab[r] != NULL); return result; } #define MULTISORT_ABORT \ for (k = 0; k < MULTISORT_LAST; k++) \ efree(ARRAYG(multisort_flags)[k]); \ efree(arrays); \ efree(args); \ RETURN_FALSE; /* {{{ proto bool array_multisort(array ar1 [, SORT_ASC|SORT_DESC [, SORT_REGULAR|SORT_NUMERIC|SORT_STRING]] [, array ar2 [, SORT_ASC|SORT_DESC [, SORT_REGULAR|SORT_NUMERIC|SORT_STRING]], ...]) Sort multiple arrays at once similar to how ORDER BY clause works in SQL */ PHP_FUNCTION(array_multisort) { zval*** args; zval*** arrays; Bucket*** indirect; Bucket* p; HashTable* hash; int argc; int array_size; int num_arrays = 0; int parse_state[MULTISORT_LAST]; /* 0 - flag not allowed 1 - flag allowed */ int sort_order = SORT_ASC; int sort_type = SORT_REGULAR; int i, k; ARRAYLS_FETCH(); /* Get the argument count and check it */ argc = ZEND_NUM_ARGS(); if (argc < 1) { WRONG_PARAM_COUNT; } /* Allocate arguments array and get the arguments, checking for errors. */ args = (zval ***)emalloc(argc * sizeof(zval **)); if (zend_get_parameters_array_ex(argc, args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } /* Allocate space for storing pointers to input arrays and sort flags. */ arrays = (zval ***)ecalloc(argc, sizeof(zval **)); for (i = 0; i < MULTISORT_LAST; i++) { parse_state[i] = 0; ARRAYG(multisort_flags)[i] = (int *)ecalloc(argc, sizeof(int)); } /* Here we go through the input arguments and parse them. Each one can be either an array or a sort flag which follows an array. If not specified, the sort flags defaults to SORT_ASC and SORT_REGULAR accordingly. There can't be two sort flags of the same type after an array, and the very first argument has to be an array. */ for (i = 0; i < argc; i++) { if (Z_TYPE_PP(args[i]) == IS_ARRAY) { /* We see the next array, so we update the sort flags of the previous array and reset the sort flags. */ if (i > 0) { ARRAYG(multisort_flags)[MULTISORT_ORDER][num_arrays-1] = sort_order; ARRAYG(multisort_flags)[MULTISORT_TYPE][num_arrays-1] = sort_type; sort_order = SORT_ASC; sort_type = SORT_REGULAR; } arrays[num_arrays++] = args[i]; /* Next one may be an array or a list of sort flags. */ for (k = 0; k < MULTISORT_LAST; k++) parse_state[k] = 1; } else if (Z_TYPE_PP(args[i]) == IS_LONG) { switch (Z_LVAL_PP(args[i])) { case SORT_ASC: case SORT_DESC: /* flag allowed here */ if (parse_state[MULTISORT_ORDER] == 1) { /* Save the flag and make sure then next arg is not the current flag. */ sort_order = Z_LVAL_PP(args[i]) == SORT_DESC ? -1 : 1; parse_state[MULTISORT_ORDER] = 0; } else { php_error(E_WARNING, "Argument %i to %s() is expected to be an array or sorting flag that has not already been specified", i+1, get_active_function_name()); MULTISORT_ABORT; } break; case SORT_REGULAR: case SORT_NUMERIC: case SORT_STRING: /* flag allowed here */ if (parse_state[MULTISORT_TYPE] == 1) { /* Save the flag and make sure then next arg is not the current flag. */ sort_type = Z_LVAL_PP(args[i]); parse_state[MULTISORT_TYPE] = 0; } else { php_error(E_WARNING, "Argument %i to %s() is expected to be an array or sorting flag that has not already been specified", i+1, get_active_function_name()); MULTISORT_ABORT; } break; default: php_error(E_WARNING, "Argument %i to %s() is an unknown sort flag", i+1, get_active_function_name()); MULTISORT_ABORT; break; } } else { php_error(E_WARNING, "Argument %i to %s() is expected to be an array or a sort flag", i+1, get_active_function_name()); MULTISORT_ABORT; } } /* Take care of the last array sort flags. */ ARRAYG(multisort_flags)[MULTISORT_ORDER][num_arrays-1] = sort_order; ARRAYG(multisort_flags)[MULTISORT_TYPE][num_arrays-1] = sort_type; /* Make sure the arrays are of the same size. */ array_size = zend_hash_num_elements(Z_ARRVAL_PP(arrays[0])); for (i = 0; i < num_arrays; i++) { if (zend_hash_num_elements(Z_ARRVAL_PP(arrays[i])) != array_size) { php_error(E_WARNING, "Array sizes are inconsistent"); MULTISORT_ABORT; } } /* If all arrays are empty or have only one entry, we don't need to do anything. */ if (array_size <= 1) { for (k = 0; k < MULTISORT_LAST; k++) efree(ARRAYG(multisort_flags)[k]); efree(arrays); efree(args); RETURN_TRUE; } /* Create the indirection array. This array is of size MxN, where M is the number of entries in each input array and N is the number of the input arrays + 1. The last column is NULL to indicate the end of the row. */ indirect = (Bucket ***)emalloc(array_size * sizeof(Bucket **)); for (i = 0; i < array_size; i++) indirect[i] = (Bucket **)emalloc((num_arrays+1) * sizeof(Bucket *)); for (i = 0; i < num_arrays; i++) { k = 0; for (p = Z_ARRVAL_PP(arrays[i])->pListHead; p; p = p->pListNext, k++) { indirect[k][i] = p; } } for (k = 0; k < array_size; k++) indirect[k][num_arrays] = NULL; /* Do the actual sort magic - bada-bim, bada-boom. */ qsort(indirect, array_size, sizeof(Bucket **), multisort_compare); /* Restructure the arrays based on sorted indirect - this is mostly taken from zend_hash_sort() function. */ HANDLE_BLOCK_INTERRUPTIONS(); for (i = 0; i < num_arrays; i++) { hash = Z_ARRVAL_PP(arrays[i]); hash->pListHead = indirect[0][i];; hash->pListTail = NULL; hash->pInternalPointer = hash->pListHead; for (k = 0; k < array_size; k++) { if (hash->pListTail) { hash->pListTail->pListNext = indirect[k][i]; } indirect[k][i]->pListLast = hash->pListTail; indirect[k][i]->pListNext = NULL; hash->pListTail = indirect[k][i]; } p = hash->pListHead; k = 0; while (p != NULL) { if (p->nKeyLength == 0) p->h = k++; p = p->pListNext; } hash->nNextFreeElement = array_size; zend_hash_rehash(hash); } HANDLE_UNBLOCK_INTERRUPTIONS(); /* Clean up. */ for (i = 0; i < array_size; i++) efree(indirect[i]); efree(indirect); for (k = 0; k < MULTISORT_LAST; k++) efree(ARRAYG(multisort_flags)[k]); efree(arrays); efree(args); RETURN_TRUE; } /* }}} */ /* {{{ proto mixed array_rand(array input [, int num_req]) Return key/keys for random entry/entries in the array */ PHP_FUNCTION(array_rand) { zval **input, **num_req; long randval; int num_req_val, num_avail, key_type; char *string_key; ulong string_key_len; ulong num_key; HashPosition pos; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &input, &num_req) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { zend_error(E_WARNING, "Argument to %s() has to be an array", get_active_function_name()); return; } num_avail = zend_hash_num_elements(Z_ARRVAL_PP(input)); if (ZEND_NUM_ARGS() > 1) { convert_to_long_ex(num_req); num_req_val = Z_LVAL_PP(num_req); if (num_req_val <= 0 || num_req_val > num_avail) { zend_error(E_WARNING, "Second argument to %s() has to be between 1 and the number of elements in the array", get_active_function_name()); return; } } else num_req_val = 1; /* Make the return value an array only if we need to pass back more than one result. */ if (num_req_val > 1) array_init(return_value); /* We can't use zend_hash_index_find() because the array may have string keys or gaps. */ zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &pos); while (num_req_val && (key_type = zend_hash_get_current_key_ex(Z_ARRVAL_PP(input), &string_key, &string_key_len, &num_key, 0, &pos)) != HASH_KEY_NON_EXISTANT) { #ifdef HAVE_RANDOM randval = random(); #else #ifdef HAVE_LRAND48 randval = lrand48(); #else randval = rand(); #endif #endif if ((double)(randval/(PHP_RAND_MAX+1.0)) < (double)num_req_val/(double)num_avail) { /* If we are returning a single result, just do it. */ if (Z_TYPE_P(return_value) != IS_ARRAY) { if (key_type == HASH_KEY_IS_STRING) { RETURN_STRINGL(string_key, string_key_len-1, 1); } else { RETURN_LONG(num_key); } } else { /* Append the result to the return value. */ if (key_type == HASH_KEY_IS_STRING) add_next_index_stringl(return_value, string_key, string_key_len-1, 1); else add_next_index_long(return_value, num_key); } num_req_val--; } num_avail--; zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &pos); } if (num_req_val == num_avail) { if (zend_hash_sort(Z_ARRVAL_P(return_value), (sort_func_t)php_mergesort, array_data_shuffle, 1) == FAILURE) { zval_dtor(return_value); RETURN_FALSE; } } } /* }}} */ /* {{{ proto mixed array_sum(array input) Returns the sum of the array entries */ PHP_FUNCTION(array_sum) { zval **input, **entry; int argc = ZEND_NUM_ARGS(); HashPosition pos; if (argc != 1 || zend_get_parameters_ex(argc, &input) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "The argument to %s() should be an array", get_active_function_name()); return; } ZVAL_LONG(return_value, 0); for (zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &pos); zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&entry, &pos) == SUCCESS; zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &pos)) { if (Z_TYPE_PP(entry) == IS_ARRAY || Z_TYPE_PP(entry) == IS_OBJECT) continue; SEPARATE_ZVAL(entry); convert_scalar_to_number(*entry); if (Z_TYPE_PP(entry) == IS_LONG && Z_TYPE_P(return_value) == IS_LONG) { Z_LVAL_P(return_value) += Z_LVAL_PP(entry); } else { convert_to_double(return_value); convert_to_double_ex(entry); Z_DVAL_P(return_value) += Z_DVAL_PP(entry); } } } /* }}} */ /* {{{ proto mixed array_reduce(array input, mixed callback [, int initial]) Iteratively reduce the array to a single value via the callback. */ PHP_FUNCTION(array_reduce) { zval **input, **callback, **initial; zval **args[2]; zval **operand; zval *result = NULL; zval *retval; char *callback_name; HashPosition pos; if (ZEND_NUM_ARGS() < 2 || ZEND_NUM_ARGS() > 3 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &input, &callback, &initial) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "%s() expects argument 1 to be an array", get_active_function_name()); return; } if (!zend_is_callable(*callback, 0, &callback_name)) { php_error(E_WARNING, "%s() expects argument 2, '%s', to be a valid callback", get_active_function_name(), callback_name); efree(callback_name); return; } efree(callback_name); if (ZEND_NUM_ARGS() > 2) result = *initial; if (zend_hash_num_elements(Z_ARRVAL_PP(input)) == 0) { if (result) { *return_value = *result; zval_copy_ctor(return_value); } return; } zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &pos); while (zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&operand, &pos) == SUCCESS) { if (result) { args[0] = &result; args[1] = operand; if (call_user_function_ex(EG(function_table), NULL, *callback, &retval, 2, args, 0, NULL) == SUCCESS && retval) { result = retval; } else { php_error(E_WARNING, "%s() had an error invoking the reduction callback", get_active_function_name()); return; } } else result = *operand; zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &pos); } *return_value = *result; } /* }}} */ /* {{{ proto array array_filter(array input [, mixed callback]) Filters elements from the array via the callback. */ PHP_FUNCTION(array_filter) { zval **input, **callback = NULL; zval **operand; zval **args[1]; zval *retval = NULL; char *callback_name; char *string_key; ulong string_key_len; ulong num_key; HashPosition pos; if (ZEND_NUM_ARGS() < 1 || ZEND_NUM_ARGS() > 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &input, &callback) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(input) != IS_ARRAY) { php_error(E_WARNING, "%s() expects argument 1 to be an array", get_active_function_name()); return; } if (ZEND_NUM_ARGS() > 1) { if (!zend_is_callable(*callback, 0, &callback_name)) { php_error(E_WARNING, "%s() expects argument 2, '%s', to be a valid callback", get_active_function_name(), callback_name); efree(callback_name); return; } efree(callback_name); } array_init(return_value); if (zend_hash_num_elements(Z_ARRVAL_PP(input)) == 0) return; for (zend_hash_internal_pointer_reset_ex(Z_ARRVAL_PP(input), &pos); zend_hash_get_current_data_ex(Z_ARRVAL_PP(input), (void **)&operand, &pos) == SUCCESS; zend_hash_move_forward_ex(Z_ARRVAL_PP(input), &pos)) { if (callback) { args[0] = operand; if (call_user_function_ex(EG(function_table), NULL, *callback, &retval, 1, args, 0, NULL) == SUCCESS && retval) { if (!zend_is_true(retval)) { zval_ptr_dtor(&retval); continue; } else zval_ptr_dtor(&retval); } else { php_error(E_WARNING, "%s() had an error invoking the filter callback", get_active_function_name()); return; } } else if (!zend_is_true(*operand)) continue; zval_add_ref(operand); switch (zend_hash_get_current_key_ex(Z_ARRVAL_PP(input), &string_key, &string_key_len, &num_key, 0, &pos)) { case HASH_KEY_IS_STRING: zend_hash_update(Z_ARRVAL_P(return_value), string_key, string_key_len, operand, sizeof(zval *), NULL); break; case HASH_KEY_IS_LONG: zend_hash_index_update(Z_ARRVAL_P(return_value), num_key, operand, sizeof(zval *), NULL); break; } } } /* }}} */ /* {{{ proto array array_map(mixed callback, array input1 [, array input2 ,...]) Applies the callback to the elements in given arrays. */ PHP_FUNCTION(array_map) { zval ***args = NULL; zval ***params; zval *callback; zval *result, *null; char *callback_name; int i, k, maxlen = 0; int *array_len; if (ZEND_NUM_ARGS() < 2) { WRONG_PARAM_COUNT; } args = (zval ***)emalloc(ZEND_NUM_ARGS() * sizeof(zval **)); if (zend_get_parameters_array_ex(ZEND_NUM_ARGS(), args) == FAILURE) { efree(args); WRONG_PARAM_COUNT; } callback = *args[0]; if (Z_TYPE_P(callback) != IS_NULL && !zend_is_callable(callback, 0, &callback_name)) { php_error(E_WARNING, "%s() expects argument 1, '%s', to be either NULL or a valid callback", get_active_function_name(), callback_name); efree(callback_name); efree(args); return; } efree(callback_name); /* Cache array sizes. */ array_len = (int*)emalloc((ZEND_NUM_ARGS()-1) * sizeof(int)); /* Check that arrays are indeed arrays and calculate maximum size. */ for (i = 0; i < ZEND_NUM_ARGS()-1; i++) { if (Z_TYPE_PP(args[i+1]) != IS_ARRAY) { php_error(E_WARNING, "%s() expects argument %d to be an array", get_active_function_name(), i + 2); efree(array_len); efree(args); return; } array_len[i] = zend_hash_num_elements(Z_ARRVAL_PP(args[i+1])); if (array_len[i] > maxlen) maxlen = array_len[i]; } /* Short-circuit: if no callback and only one array, just return it. */ if (Z_TYPE_P(callback) == IS_NULL && ZEND_NUM_ARGS() == 2) { *return_value = **args[1]; zval_copy_ctor(return_value); efree(array_len); efree(args); return; } array_init(return_value); params = (zval ***)emalloc((ZEND_NUM_ARGS()-1) * sizeof(zval **)); MAKE_STD_ZVAL(null); ZVAL_NULL(null); /* We iterate through all the arrays at once. */ for (k = 0; k < maxlen; k++) { /* * If no callback, the result will be an array, consisting of current * entries from all arrays. */ if (Z_TYPE_P(callback) == IS_NULL) { MAKE_STD_ZVAL(result); array_init(result); } for (i = 0; i < ZEND_NUM_ARGS()-1; i++) { /* * If this array still hash elements, add the current one to the * parameter list, otherwise use null value. */ if (k < array_len[i]) { zend_hash_index_find(Z_ARRVAL_PP(args[i+1]), k, (void **)¶ms[i]); } else { if (Z_TYPE_P(callback) == IS_NULL) zval_add_ref(&null); params[i] = &null; } if (Z_TYPE_P(callback) == IS_NULL) add_next_index_zval(result, *params[i]); } if (Z_TYPE_P(callback) != IS_NULL) { if (!call_user_function_ex(EG(function_table), NULL, callback, &result, ZEND_NUM_ARGS()-1, params, 0, NULL) == SUCCESS && result) { php_error(E_WARNING, "%s() had an error invoking the map callback", get_active_function_name()); efree(array_len); efree(args); zval_dtor(return_value); RETURN_NULL(); } } add_next_index_zval(return_value, result); } zval_ptr_dtor(&null); efree(params); efree(array_len); efree(args); } /* }}} */ /* {{{ proto bool key_exists(mixed key, array search) Checks if the given key or index exists in the array */ PHP_FUNCTION(key_exists) { zval **key, /* key to check for */ **array; /* array to check in */ if (ZEND_NUM_ARGS() != 2 || zend_get_parameters_ex(ZEND_NUM_ARGS(), &key, &array) == FAILURE) { WRONG_PARAM_COUNT; } if (Z_TYPE_PP(array) != IS_ARRAY && Z_TYPE_PP(array) != IS_OBJECT) { php_error(E_WARNING, "Wrong datatype for second argument in call to %s", get_active_function_name()); RETURN_FALSE; } switch (Z_TYPE_PP(key)) { case IS_STRING: if (zend_hash_exists(HASH_OF(*array), Z_STRVAL_PP(key), Z_STRLEN_PP(key)+1)) { RETURN_TRUE; } RETURN_FALSE; case IS_LONG: if (zend_hash_index_exists(HASH_OF(*array), Z_LVAL_PP(key))) { RETURN_TRUE; } RETURN_FALSE; default: php_error(E_WARNING, "Wrong datatype for first argument in call to %s", get_active_function_name()); RETURN_FALSE; } } /* }}} */ /* * Local variables: * tab-width: 4 * c-basic-offset: 4 * End: * vim600: sw=4 ts=4 tw=78 fdm=marker * vim<600: sw=4 ts=4 tw=78 */