Skip to content
2 changes: 1 addition & 1 deletion resources/functionMetadata.php
Original file line number Diff line number Diff line change
Expand Up @@ -1758,4 +1758,4 @@
'zlib_encode' => ['hasSideEffects' => false],
'zlib_get_coding_type' => ['hasSideEffects' => false],

];
];
69 changes: 52 additions & 17 deletions src/Analyser/ExprHandler/AssignHandler.php
Original file line number Diff line number Diff line change
Expand Up @@ -240,6 +240,7 @@ public function processAssignVar(
}
$assignedExpr = $this->unwrapAssign($assignedExpr);
$type = $scopeBeforeAssignEval->getType($assignedExpr);
$isImpure = count($impurePoints) > 0;

$conditionalExpressions = [];
if ($assignedExpr instanceof Ternary) {
Expand All @@ -259,23 +260,25 @@ public function processAssignVar(
$truthyType->isSuperTypeOf($falseyType)->no()
&& $falseyType->isSuperTypeOf($truthyType)->no()
) {
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType, $isImpure);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType, $isImpure);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType, $isImpure);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($condScope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType, $isImpure);
}
}

$truthyType = TypeCombinator::removeFalsey($type);
if ($truthyType !== $type) {
if (
$truthyType !== $type
) {
$truthySpecifiedTypes = $this->typeSpecifier->specifyTypesInCondition($scope, $assignedExpr, TypeSpecifierContext::createTruthy());
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType, $isImpure);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $truthySpecifiedTypes, $truthyType, $isImpure);

$falseyType = TypeCombinator::intersect($type, StaticTypeFactory::falsey());
$falseySpecifiedTypes = $this->typeSpecifier->specifyTypesInCondition($scope, $assignedExpr, TypeSpecifierContext::createFalsey());
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType, $isImpure);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $falseySpecifiedTypes, $falseyType, $isImpure);
}

foreach ([null, false, 0, 0.0, '', '0', []] as $falseyScalar) {
Expand Down Expand Up @@ -304,13 +307,13 @@ public function processAssignVar(

$notIdenticalConditionExpr = new Expr\BinaryOp\NotIdentical($assignedExpr, $astNode);
$notIdenticalSpecifiedTypes = $this->typeSpecifier->specifyTypesInCondition($scope, $notIdenticalConditionExpr, TypeSpecifierContext::createTrue());
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $notIdenticalSpecifiedTypes, $withoutFalseyType);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $notIdenticalSpecifiedTypes, $withoutFalseyType);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $notIdenticalSpecifiedTypes, $withoutFalseyType, $isImpure);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $notIdenticalSpecifiedTypes, $withoutFalseyType, $isImpure);

$identicalConditionExpr = new Expr\BinaryOp\Identical($assignedExpr, $astNode);
$identicalSpecifiedTypes = $this->typeSpecifier->specifyTypesInCondition($scope, $identicalConditionExpr, TypeSpecifierContext::createTrue());
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $identicalSpecifiedTypes, $falseyType);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $identicalSpecifiedTypes, $falseyType);
$conditionalExpressions = $this->processSureTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $identicalSpecifiedTypes, $falseyType, $isImpure);
$conditionalExpressions = $this->processSureNotTypesForConditionalExpressionsAfterAssign($scope, $var->name, $conditionalExpressions, $identicalSpecifiedTypes, $falseyType, $isImpure);
}

$nodeScopeResolver->callNodeCallback($nodeCallback, new VariableAssignNode($var, $assignedExpr), $scopeBeforeAssignEval, $storage);
Expand Down Expand Up @@ -850,7 +853,14 @@ private function unwrapAssign(Expr $expr): Expr
* @param array<string, ConditionalExpressionHolder[]> $conditionalExpressions
* @return array<string, ConditionalExpressionHolder[]>
*/
private function processSureTypesForConditionalExpressionsAfterAssign(Scope $scope, string $variableName, array $conditionalExpressions, SpecifiedTypes $specifiedTypes, Type $variableType): array
private function processSureTypesForConditionalExpressionsAfterAssign(
Scope $scope,
string $variableName,
array $conditionalExpressions,
SpecifiedTypes $specifiedTypes,
Type $variableType,
bool $isImpure,
): array
{
foreach ($specifiedTypes->getSureTypes() as $exprString => [$expr, $exprType]) {
if ($expr instanceof Variable) {
Expand All @@ -861,10 +871,19 @@ private function processSureTypesForConditionalExpressionsAfterAssign(Scope $sco
if ($expr->name === $variableName) {
continue;
}
} elseif (
(
$expr instanceof FuncCall
|| $expr instanceof MethodCall
|| $expr instanceof Expr\StaticCall
)
) {
if ($isImpure) {
continue;
}
} elseif (
!$expr instanceof PropertyFetch
&& !$expr instanceof ArrayDimFetch
&& !$expr instanceof FuncCall
) {
continue;
}
Expand All @@ -889,7 +908,14 @@ private function processSureTypesForConditionalExpressionsAfterAssign(Scope $sco
* @param array<string, ConditionalExpressionHolder[]> $conditionalExpressions
* @return array<string, ConditionalExpressionHolder[]>
*/
private function processSureNotTypesForConditionalExpressionsAfterAssign(Scope $scope, string $variableName, array $conditionalExpressions, SpecifiedTypes $specifiedTypes, Type $variableType): array
private function processSureNotTypesForConditionalExpressionsAfterAssign(
Scope $scope,
string $variableName,
array $conditionalExpressions,
SpecifiedTypes $specifiedTypes,
Type $variableType,
bool $isImpure,
): array
{
foreach ($specifiedTypes->getSureNotTypes() as $exprString => [$expr, $exprType]) {
if ($expr instanceof Variable) {
Expand All @@ -900,10 +926,19 @@ private function processSureNotTypesForConditionalExpressionsAfterAssign(Scope $
if ($expr->name === $variableName) {
continue;
}
} elseif (
(
$expr instanceof FuncCall
|| $expr instanceof MethodCall
|| $expr instanceof Expr\StaticCall
)
) {
if ($isImpure) {
continue;
}
} elseif (
!$expr instanceof PropertyFetch
&& !$expr instanceof ArrayDimFetch
&& !$expr instanceof FuncCall
) {
continue;
}
Expand Down
1 change: 1 addition & 0 deletions tests/PHPStan/Analyser/nsrt/bug-3190.php
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

interface Server
{
/** @pure */
public function isDedicated(): bool;

public function getSize(): int;
Expand Down
83 changes: 83 additions & 0 deletions tests/PHPStan/Analyser/nsrt/bug-5207.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,83 @@
<?php // lint >= 8.0

declare(strict_types = 1);

namespace Bug5207;

use function PHPStan\Testing\assertType;

abstract class HelloWorld {
/** @phpstan-pure */
abstract public function getChild(): ?HelloWorld;

public function sayHello(): void {
$foo = null !== $this->getChild();
if ($foo) {
assertType('Bug5207\HelloWorld', $this->getChild());
}
}

public function sayHelloInline(): void {
if (null !== $this->getChild()) {
assertType('Bug5207\HelloWorld', $this->getChild());
}
}
}

abstract class StaticWorld {
/** @phpstan-pure */
abstract public static function getChild(): ?StaticWorld;

public static function sayHello(): void {
$foo = null !== static::getChild();
if ($foo) {
assertType('Bug5207\StaticWorld', static::getChild());
}
}

public static function sayHelloInline(): void {
if (null !== static::getChild()) {
assertType('Bug5207\StaticWorld', static::getChild());
}
}
}

abstract class ImpureStaticWorld {
/**
* @phpstan-impure
*/
abstract public static function getChild(): ?ImpureStaticWorld;

public static function sayHello(): void {
$foo = null !== static::getChild();
if ($foo) {
assertType('Bug5207\ImpureStaticWorld|null', static::getChild());
}
}

public static function sayHelloInline(): void {
if (null !== static::getChild()) {
assertType('Bug5207\ImpureStaticWorld|null', static::getChild());
}
}
}

abstract class ImpureWorld {
/**
* @phpstan-impure
*/
abstract public function getChild(): ?ImpureWorld;

public function sayHello(): void {
$foo = null !== $this->getChild();
if ($foo) {
assertType('Bug5207\ImpureWorld|null', $this->getChild());
}
}

public function sayHelloInline(): void {
if (null !== $this->getChild()) {
assertType('Bug5207\ImpureWorld|null', $this->getChild());
}
}
}
87 changes: 87 additions & 0 deletions tests/PHPStan/Analyser/nsrt/bug-9455.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,87 @@
<?php // lint >= 8.0

declare(strict_types = 1);

namespace Bug9455;

use function PHPStan\Testing\assertType;

class A {
public function __construct(private int $id){}

public function getId(): int {
return $this->id;
}
}

class B {
public function __construct(private int $id, private ?A $a = null){}

public function getId(): int {
return $this->id;
}

/**
* @phpstan-pure
*/
public function getA(): ?A {
return $this->a;
}
}

class HelloWorld
{
public function testFails(): void
{
$a = new A(1);
$b = new B(1, $a);

$hasA = $b->getA() !== null;

if($hasA) {
assertType('Bug9455\A', $b->getA());
}
}

public function testSucceeds(): void
{
$a = new A(1);
$b = new B(1, $a);

if($b->getA() !== null) {
assertType('Bug9455\A', $b->getA());
}
}
}

class C {
Comment thread
staabm marked this conversation as resolved.
/**
* @phpstan-impure
*/
public function getA(): ?A {
return rand(0, 1) ? new A(1) : null;
}
}

class ImpureTest
{
public function testImpureMethodNotNarrowed(): void
{
$c = new C();

$hasA = $c->getA() !== null;

if($hasA) {
assertType('Bug9455\A|null', $c->getA());
}
}

public function testImpureMethodInline(): void
{
$c = new C();

if($c->getA() !== null) {
assertType('Bug9455\A|null', $c->getA());
}
}
}
Loading