1 PHP_FUNCTION(range) 2 { 3 zval *zlow, *zhigh, *zstep = NULL, tmp; 4 int err = 0, is_step_double = 0; 5 double step = 1.0; // 步长默认为1 6 7 if (zend_parse_parameters(ZEND_NUM_ARGS(), "zz|z", &zlow, &zhigh, &zstep) == FAILURE) { 8 RETURN_FALSE; 9 } 10 11 // 步长 12 if (zstep) { 13 if (Z_TYPE_P(zstep) == IS_DOUBLE || 14 (Z_TYPE_P(zstep) == IS_STRING && is_numeric_string(Z_STRVAL_P(zstep), Z_STRLEN_P(zstep), NULL, NULL, 0) == IS_DOUBLE) 15 ) { 16 is_step_double = 1; 17 } 18 19 step = zval_get_double(zstep); 20 21 // 步长可以为负数。取绝对值 22 /* We only want positive step values. */ 23 if (step < 0.0) { 24 step *= -1; 25 } 26 } 27 28 /* If the range is given as strings, generate an array of characters. */ 29 /* start、end是字符串 */ 30 if (Z_TYPE_P(zlow) == IS_STRING && Z_TYPE_P(zhigh) == IS_STRING && Z_STRLEN_P(zlow) >= 1 && Z_STRLEN_P(zhigh) >= 1) { 31 int type1, type2; 32 unsigned char low, high; 33 zend_long lstep = (zend_long) step; 34 35 type1 = is_numeric_string(Z_STRVAL_P(zlow), Z_STRLEN_P(zlow), NULL, NULL, 0); 36 type2 = is_numeric_string(Z_STRVAL_P(zhigh), Z_STRLEN_P(zhigh), NULL, NULL, 0); 37 38 /* 填充数字 */ 39 if (type1 == IS_DOUBLE || type2 == IS_DOUBLE || is_step_double) { 40 goto double_str; 41 } else if (type1 == IS_LONG || type2 == IS_LONG) { 42 goto long_str; 43 } 44 45 /* 填充字符，只取字符串start和字符串end的第一个字符 */ 46 low = (unsigned char)Z_STRVAL_P(zlow)[0]; 47 high = (unsigned char)Z_STRVAL_P(zhigh)[0]; 48 49 if (low > high) { /* Negative Steps */ 50 if (lstep <= 0) { 51 err = 1; 52 goto err; 53 } 54 /* Initialize the return_value as an array. */ 55 array_init_size(return_value, (uint32_t)(((low - high) / lstep) + 1)); 56 zend_hash_real_init(Z_ARRVAL_P(return_value), 1); 57 ZEND_HASH_FILL_PACKED(Z_ARRVAL_P(return_value)) { 58 for (; low >= high; low -= (unsigned int)lstep) { 59 if (CG(one_char_string)[low]) { 60 ZVAL_INTERNED_STR(&tmp, CG(one_char_string)[low]); 61 } else { 62 ZVAL_STRINGL(&tmp, (char*)&low, 1); 63 } 64 ZEND_HASH_FILL_ADD(&tmp); 65 if (((signed int)low - lstep) < 0) { 66 break; 67 } 68 } 69 } ZEND_HASH_FILL_END(); 70 } else if (high > low) { /* Positive Steps */ 71 if (lstep <= 0) { 72 err = 1; 73 goto err; 74 } 75 // 初始化返回的数组 76 array_init_size(return_value, (uint32_t)(((high - low) / lstep) + 1)); 77 zend_hash_real_init(Z_ARRVAL_P(return_value), 1); 78 ZEND_HASH_FILL_PACKED(Z_ARRVAL_P(return_value)) { 79 // 将字符插入返回的数组 80 for (; low <= high; low += (unsigned int)lstep) { 81 if (CG(one_char_string)[low]) { 82 ZVAL_INTERNED_STR(&tmp, CG(one_char_string)[low]); 83 } else { 84 ZVAL_STRINGL(&tmp, (char*)&low, 1); 85 } 86 ZEND_HASH_FILL_ADD(&tmp); 87 if (((signed int)low + lstep) > 255) { 88 break; 89 } 90 } 91 } ZEND_HASH_FILL_END(); 92 } else { 93 array_init(return_value); 94 if (CG(one_char_string)[low]) { 95 ZVAL_INTERNED_STR(&tmp, CG(one_char_string)[low]); 96 } else { 97 ZVAL_STRINGL(&tmp, (char*)&low, 1); 98 } 99 zend_hash_next_index_insert_new(Z_ARRVAL_P(return_value), &tmp); 100 } 101 } else if (Z_TYPE_P(zlow) == IS_DOUBLE || Z_TYPE_P(zhigh) == IS_DOUBLE || is_step_double) { // start、end是浮点数 102 double low, high, value; 103 zend_long i; 104 double_str: 105 low = zval_get_double(zlow); 106 high = zval_get_double(zhigh); 107 i = 0; 108 109 // 浮点数范围检查 110 if (zend_isinf(high) || zend_isinf(low)) { 111 php_error_docref(NULL, E_WARNING, "Invalid range supplied: start=%0.0f end=%0.0f", low, high); 112 RETURN_FALSE; 113 } 114 115 Z_TYPE_INFO(tmp) = IS_DOUBLE; 116 if (low > high) { /* Negative steps 第一个参数比第二大*/ 117 if (low - high < step || step <= 0) { // 范围差小于步长或步长取绝对值之后仍小于0。则报WARNING错误。 118 err = 1; 119 goto err; 120 } 121 122 // 检查并初始化返回的数组 123 RANGE_CHECK_INIT_ARRAY(low, high); 124 // 向返回的数组中填充值，降序 125 ZEND_HASH_FILL_PACKED(Z_ARRVAL_P(return_value)) { 126 for (value = low; value >= (high - DOUBLE_DRIFT_FIX); value = low - (++i * step)) { 127 Z_DVAL(tmp) = value; 128 ZEND_HASH_FILL_ADD(&tmp); 129 } 130 } ZEND_HASH_FILL_END(); 131 } else if (high > low) { /* Positive steps 第一个参数比第二个小*/ 132 if (high - low < step || step <= 0) { 133 err = 1; 134 goto err; 135 } 136 137 // 检查并初始化返回的数组 138 RANGE_CHECK_INIT_ARRAY(high, low); 139 // 向返回的数组中填充值，升序 140 ZEND_HASH_FILL_PACKED(Z_ARRVAL_P(return_value)) { 141 for (value = low; value <= (high + DOUBLE_DRIFT_FIX); value = low + (++i * step)) { 142 Z_DVAL(tmp) = value; 143 ZEND_HASH_FILL_ADD(&tmp); 144 } 145 } ZEND_HASH_FILL_END(); 146 } else { // low == high。则返回的数组只用一个元素：array(0=>low) 147 array_init(return_value); 148 Z_DVAL(tmp) = low; 149 zend_hash_next_index_insert_new(Z_ARRVAL_P(return_value), &tmp); 150 } 151 } else { // start、end是整数 152 double low, high; 153 zend_long lstep; 154 long_str: 155 low = zval_get_double(zlow); 156 high = zval_get_double(zhigh); 157 lstep = (zend_long) step; 158 159 Z_TYPE_INFO(tmp) = IS_LONG; 160 if (low > high) { /* Negative steps */ 161 if (low - high < lstep || lstep <= 0) { 162 err = 1; 163 goto err; 164 } 165 166 RANGE_CHECK_INIT_ARRAY(low, high); 167 ZEND_HASH_FILL_PACKED(Z_ARRVAL_P(return_value)) { 168 for (; low >= high; low -= lstep) { 169 Z_LVAL(tmp) = (zend_long)low; 170 ZEND_HASH_FILL_ADD(&tmp); 171 } 172 } ZEND_HASH_FILL_END(); 173 } else if (high > low) { /* Positive steps */ 174 if (high - low < lstep || lstep <= 0) { 175 err = 1; 176 goto err; 177 } 178 179 RANGE_CHECK_INIT_ARRAY(high, low); 180 ZEND_HASH_FILL_PACKED(Z_ARRVAL_P(return_value)) { 181 for (; low <= high; low += lstep) { 182 Z_LVAL(tmp) = (zend_long)low; 183 ZEND_HASH_FILL_ADD(&tmp); 184 } 185 } ZEND_HASH_FILL_END(); 186 } else { 187 array_init(return_value); 188 Z_LVAL(tmp) = (zend_long)low; 189 zend_hash_next_index_insert_new(Z_ARRVAL_P(return_value), &tmp); 190 } 191 } 192 err: 193 if (err) { 194 php_error_docref(NULL, E_WARNING, "step exceeds the specified range"); 195 RETURN_FALSE; 196 } 197 }