support all socket_* functions

Author: staabm

conf/config.neon

@@ -1821,7 +1821,7 @@ services:
 			- phpstan.broker.dynamicMethodReturnTypeExtension
 
 	-
-		class: PHPStan\Type\Php\SocketCreateReturnTypeExtension
+		class: PHPStan\Type\Php\SocketObjectReturnTypeFunctionExtension
 		tags:
 			- phpstan.broker.dynamicFunctionReturnTypeExtension
 

src/Type/Php/SocketCreateReturnTypeExtension.php renamed to src/Type/Php/SocketObjectReturnTypeFunctionExtension.php

@@ -11,22 +11,28 @@
 use PHPStan\Type\ResourceType;
 use PHPStan\Type\Type;
 use PHPStan\Type\UnionType;
-use function count;
+use function in_array;
 
-final class SocketCreateReturnTypeExtension implements DynamicFunctionReturnTypeExtension
+final class SocketObjectReturnTypeFunctionExtension implements DynamicFunctionReturnTypeExtension
 {
 
+	private const FUNCTIONS = [
+		'socket_accept',
+		'socket_addrinfo_bind',
+		'socket_addrinfo_connect',
+		'socket_create',
+		'socket_create_listen',
+		'socket_import_stream',
+		'socket_wsaprotocol_info_import',
+	];
+
 	public function isFunctionSupported(FunctionReflection $functionReflection): bool
 	{
-		return $functionReflection->getName() === 'socket_create';
+		return in_array($functionReflection->getName(), self::FUNCTIONS, true);
 	}
 
 	public function getTypeFromFunctionCall(FunctionReflection $functionReflection, FuncCall $functionCall, Scope $scope): ?Type
 	{
-		if (count($functionCall->getArgs()) < 3) {
-			return null;
-		}
-
 		if ($scope->getPhpVersion()->socketCreateReturnsObject()->yes()) {
 			return new UnionType([new ConstantBooleanType(false), new ObjectType('\\Socket')]);
 		}