support all socket_* functions

Author: staabm

conf/config.neon

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

src/Php/PhpVersions.php

@@ -28,7 +28,7 @@ public function producesWarningForFinalPrivateMethods(): TrinaryLogic
 		return IntegerRangeType::fromInterval(80000, null)->isSuperTypeOf($this->phpVersions)->result;
 	}
 
-	public function socketCreateReturnsObject(): TrinaryLogic
+	public function socketFunctionsUseObject(): TrinaryLogic
 	{
 		return IntegerRangeType::fromInterval(80000, null)->isSuperTypeOf($this->phpVersions)->result;
 	}

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

@@ -11,27 +11,33 @@
 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()) {
+		if ($scope->getPhpVersion()->socketFunctionsUseObject()->yes()) {
 			return new UnionType([new ConstantBooleanType(false), new ObjectType('\\Socket')]);
 		}
 
-		if ($scope->getPhpVersion()->socketCreateReturnsObject()->no()) {
+		if ($scope->getPhpVersion()->socketFunctionsUseObject()->no()) {
 			return new UnionType([new ConstantBooleanType(false), new ResourceType()]);
 		}