Add tests to check ZPP handles intersection types for internal functions

Author: Girgias

ext/zend_test/test.c

@@ -206,6 +206,51 @@ static ZEND_FUNCTION(zend_string_or_stdclass)
 	}
 }
 
+/* Tests Z_PARAM_OBJ_OF_CLASS_OR_STR_OR_NULL */
+static ZEND_FUNCTION(zend_string_or_stdclass_or_null)
+{
+	zend_string *str;
+	zend_object *object;
+
+	ZEND_PARSE_PARAMETERS_START(1, 1)
+		Z_PARAM_OBJ_OF_CLASS_OR_STR_OR_NULL(object, zend_standard_class_def, str)
+	ZEND_PARSE_PARAMETERS_END();
+
+	if (str) {
+		RETURN_STR_COPY(str);
+	} else if (object) {
+		RETURN_OBJ_COPY(object);
+	} else {
+		RETURN_NULL();
+	}
+}
+
+static ZEND_FUNCTION(zend_intersection_type)
+{
+	zend_object *object;
+
+	ZEND_PARSE_PARAMETERS_START(1, 1)
+		Z_PARAM_OBJ(object)
+	ZEND_PARSE_PARAMETERS_END();
+
+	if (!(instanceof_function(object->ce, zend_ce_countable) && instanceof_function(object->ce, zend_ce_traversable))) {
+		zend_argument_type_error(1, "must be of type Traversable&Countable, %s given", ZSTR_VAL(object->ce->name));
+	}
+
+	RETURN_OBJ_COPY(object);
+}
+
+static ZEND_FUNCTION(zend_intersection_type_return)
+{
+	zend_object *object;
+
+	ZEND_PARSE_PARAMETERS_START(1, 1)
+		Z_PARAM_OBJ(object)
+	ZEND_PARSE_PARAMETERS_END();
+
+	RETURN_OBJ_COPY(object);
+}
+
 static ZEND_FUNCTION(zend_test_compile_string)
 {
 	zend_string *source_string = NULL;
@@ -241,25 +286,6 @@ static ZEND_FUNCTION(zend_test_compile_string)
 	return;
 }
 
-/* Tests Z_PARAM_OBJ_OF_CLASS_OR_STR_OR_NULL */
-static ZEND_FUNCTION(zend_string_or_stdclass_or_null)
-{
-	zend_string *str;
-	zend_object *object;
-
-	ZEND_PARSE_PARAMETERS_START(1, 1)
-		Z_PARAM_OBJ_OF_CLASS_OR_STR_OR_NULL(object, zend_standard_class_def, str)
-	ZEND_PARSE_PARAMETERS_END();
-
-	if (str) {
-		RETURN_STR_COPY(str);
-	} else if (object) {
-		RETURN_OBJ_COPY(object);
-	} else {
-		RETURN_NULL();
-	}
-}
-
 static ZEND_FUNCTION(zend_weakmap_attach)
 {
 	zval *value;

ext/zend_test/test.stub.php

@@ -119,6 +119,10 @@ function zend_string_or_stdclass($param): stdClass|string {}
     /** @param stdClass|string|null $param */
     function zend_string_or_stdclass_or_null($param): stdClass|string|null {}
 
+    function zend_intersection_type(Traversable&Countable $v): Traversable&Countable {}
+
+    function zend_intersection_type_return(object $v): Traversable&Countable {}
+
     function zend_iterable(iterable $arg1, ?iterable $arg2 = null): void {}
 
     function zend_weakmap_attach(object $object, mixed $value): bool {}

ext/zend_test/test_arginfo.h

@@ -0,0 +1,40 @@
+--TEST--
+Test that internal functions can accept intersection types via ZPP
+--EXTENSIONS--
+zend_test
+spl
+--FILE--
+<?php
+
+class C implements Countable {
+    public function count(): int {
+        return 1;
+    }
+}
+
+class I extends EmptyIterator implements Countable {
+    public function count(): int {
+        return 1;
+    }
+}
+
+
+try {
+    zend_intersection_type(new EmptyIterator());
+} catch (TypeError $e) {
+    echo $e->getMessage(), PHP_EOL;
+}
+try {
+    zend_intersection_type(new C());
+} catch (TypeError $e) {
+    echo $e->getMessage(), PHP_EOL;
+}
+
+zend_intersection_type(new I());
+
+?>
+==DONE==
+--EXPECT--
+zend_intersection_type(): Argument #1 ($v) must be of type Traversable&Countable, EmptyIterator given
+zend_intersection_type(): Argument #1 ($v) must be of type Traversable&Countable, C given
+==DONE==

ext/zend_test/tests/zend_internal_arg_intersection.phpt

@@ -0,0 +1,40 @@
+--TEST--
+Test that internal functions warn on improper intersection types return
+--EXTENSIONS--
+zend_test
+spl
+--FILE--
+<?php
+
+class C implements Countable {
+    public function count(): int {
+        return 1;
+    }
+}
+
+class I extends EmptyIterator implements Countable {
+    public function count(): int {
+        return 1;
+    }
+}
+
+try {
+    var_dump(zend_intersection_type(new EmptyIterator()));
+} catch (TypeError $e) {
+    echo $e->getMessage(), PHP_EOL;
+}
+try {
+    var_dump(zend_intersection_type(new C()));
+} catch (TypeError $e) {
+    echo $e->getMessage(), PHP_EOL;
+}
+var_dump(zend_intersection_type(new I()));
+
+?>
+==DONE==
+--EXPECT--
+zend_intersection_type(): Argument #1 ($v) must be of type Traversable&Countable, EmptyIterator given
+zend_intersection_type(): Argument #1 ($v) must be of type Traversable&Countable, C given
+object(I)#2 (0) {
+}
+==DONE==