RC1 optimisation for ucfirst and lcfirst

Author: nielsdos

Zend/zend_types.h

@@ -1512,4 +1512,9 @@ static zend_always_inline bool zend_may_modify_arg_in_place(const zval *arg)
 	return Z_REFCOUNTED_P(arg) && !(GC_FLAGS(Z_COUNTED_P(arg)) & (GC_IMMUTABLE | GC_PERSISTENT)) && Z_REFCOUNT_P(arg) == 1;
 }
 
+static zend_always_inline bool zend_may_modify_string_in_place(const zend_string *arg)
+{
+	return !(GC_FLAGS(arg) & (GC_IMMUTABLE | GC_PERSISTENT)) && GC_REFCOUNT(arg) == 1;
+}
+
 #endif /* ZEND_TYPES_H */

ext/standard/string.c

@@ -2341,7 +2341,13 @@ static zend_string* php_ucfirst(zend_string *str)
 	if (r == ch) {
 		return zend_string_copy(str);
 	} else {
-		zend_string *s = zend_string_init(ZSTR_VAL(str), ZSTR_LEN(str), 0);
+		zend_string *s;
+		if (zend_may_modify_string_in_place(str)) {
+			s = str;
+			zend_string_forget_hash_val(s);
+		} else {
+			s = zend_string_init(ZSTR_VAL(str), ZSTR_LEN(str), false);
+		}
 		ZSTR_VAL(s)[0] = r;
 		return s;
 	}
@@ -2373,7 +2379,13 @@ static zend_string* php_lcfirst(zend_string *str)
 	if (r == ZSTR_VAL(str)[0]) {
 		return zend_string_copy(str);
 	} else {
-		zend_string *s = zend_string_init(ZSTR_VAL(str), ZSTR_LEN(str), 0);
+		zend_string *s;
+		if (zend_may_modify_string_in_place(str)) {
+			s = str;
+			zend_string_forget_hash_val(s);
+		} else {
+			s = zend_string_init(ZSTR_VAL(str), ZSTR_LEN(str), false);
+		}
 		ZSTR_VAL(s)[0] = r;
 		return s;
 	}