1
0
mirror of https://github.com/danog/psalm.git synced 2024-11-26 20:34:47 +01:00

Move LoopChecker into its own file

This commit is contained in:
Matthew Brown 2017-11-30 01:07:20 -05:00
parent 2b34fa0cd0
commit 888b5dae00
5 changed files with 369 additions and 346 deletions

View File

@ -29,7 +29,7 @@ class ForChecker
$for_context = clone $context;
$for_context->inside_loop = true;
$statements_checker->analyzeLoop($stmt->stmts, $stmt->cond, $stmt->loop, $for_context, $context);
LoopChecker::analyze($statements_checker, $stmt->stmts, $stmt->cond, $stmt->loop, $for_context, $context);
$context->vars_possibly_in_scope = array_merge(
$for_context->vars_possibly_in_scope,

View File

@ -295,7 +295,7 @@ class ForeachChecker
}
}
$statements_checker->analyzeLoop($stmt->stmts, [], [], $foreach_context, $context);
LoopChecker::analyze($statements_checker, $stmt->stmts, [], [], $foreach_context, $context);
$context->vars_possibly_in_scope = array_merge(
$foreach_context->vars_possibly_in_scope,

View File

@ -0,0 +1,363 @@
<?php
namespace Psalm\Checker\Statements\Block;
use PhpParser;
use Psalm\Checker\AlgebraChecker;
use Psalm\Checker\ScopeChecker;
use Psalm\Checker\Statements\ExpressionChecker;
use Psalm\Checker\StatementsChecker;
use Psalm\Checker\TypeChecker;
use Psalm\Clause;
use Psalm\CodeLocation;
use Psalm\Context;
use Psalm\IssueBuffer;
use Psalm\Type;
class LoopChecker
{
/**
* Checks an array of statements in a loop
*
* @param array<PhpParser\Node\Stmt|PhpParser\Node\Expr> $stmts
* @param PhpParser\Node\Expr[] $pre_conditions
* @param PhpParser\Node\Expr[] $post_conditions
* @param array<int, string> $asserted_vars
* @param Context $loop_context
* @param Context $outer_context
*
* @return false|null
*/
public static function analyze(
StatementsChecker $statements_checker,
array $stmts,
array $pre_conditions,
array $post_conditions,
Context $loop_context,
Context $outer_context
) {
$traverser = new PhpParser\NodeTraverser;
$assignment_mapper = new \Psalm\Visitor\AssignmentMapVisitor($loop_context->self);
$traverser->addVisitor($assignment_mapper);
$traverser->traverse($stmts);
$assignment_map = $assignment_mapper->getAssignmentMap();
$assignment_depth = 0;
$asserted_vars = [];
$pre_condition_clauses = [];
if ($pre_conditions) {
foreach ($pre_conditions as $pre_condition) {
$pre_condition_clauses = array_merge(
$pre_condition_clauses,
AlgebraChecker::getFormula(
$pre_condition,
$loop_context->self,
$statements_checker
)
);
}
} else {
$asserted_vars = Context::getNewOrUpdatedVarIds($outer_context, $loop_context);
}
if ($assignment_map) {
$first_var_id = array_keys($assignment_map)[0];
$assignment_depth = self::getAssignmentMapDepth($first_var_id, $assignment_map);
}
$loop_context->parent_context = $outer_context;
if ($assignment_depth === 0) {
foreach ($pre_conditions as $pre_condition) {
if (self::applyPreConditionToLoopContext(
$statements_checker,
$pre_condition,
$pre_condition_clauses,
$loop_context,
$outer_context
) === false) {
return false;
}
}
$statements_checker->analyze($stmts, $loop_context, $outer_context);
foreach ($post_conditions as $post_condition) {
if (ExpressionChecker::analyze($statements_checker, $post_condition, $loop_context) === false) {
return false;
}
}
} else {
// record all the vars that existed before we did the first pass through the loop
$pre_loop_context = clone $loop_context;
$pre_outer_context = clone $outer_context;
IssueBuffer::startRecording();
foreach ($pre_conditions as $pre_condition) {
if (self::applyPreConditionToLoopContext(
$statements_checker,
$pre_condition,
$pre_condition_clauses,
$loop_context,
$outer_context
) === false) {
return false;
}
}
$statements_checker->analyze($stmts, $loop_context, $outer_context);
foreach ($post_conditions as $post_condition) {
if (ExpressionChecker::analyze($statements_checker, $post_condition, $loop_context) === false) {
return false;
}
}
$recorded_issues = IssueBuffer::clearRecordingLevel();
IssueBuffer::stopRecording();
for ($i = 0; $i < $assignment_depth; ++$i) {
$vars_to_remove = [];
$has_changes = false;
foreach ($loop_context->vars_in_scope as $var_id => $type) {
if (in_array($var_id, $asserted_vars, true)) {
// set the vars to whatever the while/foreach loop expects them to be
if ((string)$type !== (string)$pre_loop_context->vars_in_scope[$var_id]) {
$loop_context->vars_in_scope[$var_id] = $pre_loop_context->vars_in_scope[$var_id];
$has_changes = true;
}
} elseif (isset($pre_outer_context->vars_in_scope[$var_id])) {
$pre_outer = (string)$pre_outer_context->vars_in_scope[$var_id];
if ((string)$type !== $pre_outer ||
(string)$outer_context->vars_in_scope[$var_id] !== $pre_outer
) {
$has_changes = true;
// widen the foreach context type with the initial context type
$loop_context->vars_in_scope[$var_id] = Type::combineUnionTypes(
$loop_context->vars_in_scope[$var_id],
$outer_context->vars_in_scope[$var_id]
);
// if there's a change, invalidate related clauses
$pre_loop_context->removeVarFromConflictingClauses($var_id);
}
} else {
$vars_to_remove[] = $var_id;
}
}
foreach ($asserted_vars as $var_id) {
if (!isset($loop_context->vars_in_scope[$var_id])) {
$loop_context->vars_in_scope[$var_id] = $pre_loop_context->vars_in_scope[$var_id];
}
}
// if there are no changes to the types, no need to re-examine
if (!$has_changes) {
break;
}
// remove vars that were defined in the foreach
foreach ($vars_to_remove as $var_id) {
unset($loop_context->vars_in_scope[$var_id]);
}
$loop_context->clauses = $pre_loop_context->clauses;
IssueBuffer::startRecording();
foreach ($pre_conditions as $pre_condition) {
if (self::applyPreConditionToLoopContext(
$statements_checker,
$pre_condition,
$pre_condition_clauses,
$loop_context,
$outer_context
) === false) {
return false;
}
}
$statements_checker->analyze($stmts, $loop_context, $outer_context);
foreach ($post_conditions as $post_condition) {
if (ExpressionChecker::analyze($statements_checker, $post_condition, $loop_context) === false) {
return false;
}
}
$recorded_issues = IssueBuffer::clearRecordingLevel();
IssueBuffer::stopRecording();
}
if ($recorded_issues) {
foreach ($recorded_issues as $recorded_issue) {
// if we're not in any loops then this will just result in the issue being emitted
IssueBuffer::bubbleUp($recorded_issue);
}
}
}
foreach ($outer_context->vars_in_scope as $var_id => $type) {
if ($type->isMixed()) {
continue;
}
if (!isset($loop_context->vars_in_scope[$var_id])) {
unset($outer_context->vars_in_scope[$var_id]);
continue;
}
if ($loop_context->vars_in_scope[$var_id]->isMixed()) {
$outer_context->vars_in_scope[$var_id] = $loop_context->vars_in_scope[$var_id];
}
if ((string) $loop_context->vars_in_scope[$var_id] !== (string) $type) {
$outer_context->vars_in_scope[$var_id] = Type::combineUnionTypes(
$outer_context->vars_in_scope[$var_id],
$loop_context->vars_in_scope[$var_id]
);
$outer_context->removeVarFromConflictingClauses($var_id);
}
}
if ($pre_conditions && $pre_condition_clauses && !ScopeChecker::doesEverBreak($stmts)) {
// if the loop contains an assertion and there are no break statements, we can negate that assertion
// and apply it to the current context
$negated_pre_condition_types = AlgebraChecker::getTruthsFromFormula(
AlgebraChecker::negateFormula($pre_condition_clauses)
);
if ($negated_pre_condition_types) {
$changed_var_ids = [];
$vars_in_scope_reconciled = TypeChecker::reconcileKeyedTypes(
$negated_pre_condition_types,
$loop_context->vars_in_scope,
$changed_var_ids,
[],
$statements_checker,
new CodeLocation($statements_checker->getSource(), $pre_conditions[0]),
$statements_checker->getSuppressedIssues()
);
if ($vars_in_scope_reconciled === false) {
return false;
}
foreach ($outer_context->vars_in_scope as $var_id => $type) {
if (isset($vars_in_scope_reconciled[$var_id])) {
$outer_context->vars_in_scope[$var_id] = $vars_in_scope_reconciled[$var_id];
}
}
foreach ($changed_var_ids as $changed_var_id) {
$outer_context->removeVarFromConflictingClauses($changed_var_id);
}
}
}
}
/**
* @param PhpParser\Node\Expr $pre_condition
* @param array<int, Clause> $pre_condition_clauses
* @param Context $loop_context
* @param Context $outer_context
*
* @return false|null
*/
private static function applyPreConditionToLoopContext(
StatementsChecker $statements_checker,
PhpParser\Node\Expr $pre_condition,
array $pre_condition_clauses,
Context $loop_context,
Context $outer_context
) {
$pre_referenced_var_ids = $loop_context->referenced_var_ids;
$loop_context->referenced_var_ids = [];
$loop_context->inside_conditional = true;
if (ExpressionChecker::analyze($statements_checker, $pre_condition, $loop_context) === false) {
return false;
}
$loop_context->inside_conditional = false;
$new_referenced_var_ids = $loop_context->referenced_var_ids;
$loop_context->referenced_var_ids = array_merge($pre_referenced_var_ids, $new_referenced_var_ids);
$asserted_vars = array_keys(AlgebraChecker::getTruthsFromFormula($pre_condition_clauses));
$loop_context->clauses = AlgebraChecker::simplifyCNF(
array_merge($outer_context->clauses, $pre_condition_clauses)
);
$reconcilable_while_types = AlgebraChecker::getTruthsFromFormula($loop_context->clauses);
// if the while has an or as the main component, we cannot safely reason about it
if ($pre_condition instanceof PhpParser\Node\Expr\BinaryOp &&
($pre_condition instanceof PhpParser\Node\Expr\BinaryOp\BooleanOr ||
$pre_condition instanceof PhpParser\Node\Expr\BinaryOp\LogicalOr)
) {
// do nothing
} else {
$changed_var_ids = [];
$pre_condition_vars_in_scope_reconciled = TypeChecker::reconcileKeyedTypes(
$reconcilable_while_types,
$loop_context->vars_in_scope,
$changed_var_ids,
$new_referenced_var_ids,
$statements_checker,
new CodeLocation($statements_checker->getSource(), $pre_condition),
$statements_checker->getSuppressedIssues()
);
if ($pre_condition_vars_in_scope_reconciled === false) {
return false;
}
$loop_context->vars_in_scope = $pre_condition_vars_in_scope_reconciled;
}
}
/**
* @param string $first_var_id
* @param array<string, array<string, bool>> $assignment_map
*
* @return int
*/
private static function getAssignmentMapDepth($first_var_id, array $assignment_map)
{
$max_depth = 0;
$assignment_var_ids = $assignment_map[$first_var_id];
unset($assignment_map[$first_var_id]);
foreach ($assignment_var_ids as $assignment_var_id => $_) {
$depth = 1;
if (isset($assignment_map[$assignment_var_id])) {
$depth = 1 + self::getAssignmentMapDepth($assignment_var_id, $assignment_map);
}
if ($depth > $max_depth) {
$max_depth = $depth;
}
}
return $max_depth;
}
}

View File

@ -21,7 +21,8 @@ class WhileChecker
) {
$while_context = clone $context;
$statements_checker->analyzeLoop(
LoopChecker::analyze(
$statements_checker,
$stmt->stmts,
$stmt->cond ? [$stmt->cond] : [],
[],

View File

@ -5,13 +5,13 @@ use PhpParser;
use Psalm\Checker\Statements\Block\ForChecker;
use Psalm\Checker\Statements\Block\ForeachChecker;
use Psalm\Checker\Statements\Block\IfChecker;
use Psalm\Checker\Statements\Block\LoopChecker;
use Psalm\Checker\Statements\Block\SwitchChecker;
use Psalm\Checker\Statements\Block\TryChecker;
use Psalm\Checker\Statements\Block\WhileChecker;
use Psalm\Checker\Statements\Expression\AssignmentChecker;
use Psalm\Checker\Statements\Expression\CallChecker;
use Psalm\Checker\Statements\ExpressionChecker;
use Psalm\Clause;
use Psalm\CodeLocation;
use Psalm\Config;
use Psalm\Context;
@ -407,347 +407,6 @@ class StatementsChecker extends SourceChecker implements StatementsSource
return null;
}
/**
* Checks an array of statements in a loop
*
* @param array<PhpParser\Node\Stmt|PhpParser\Node\Expr> $stmts
* @param PhpParser\Node\Expr[] $pre_conditions
* @param PhpParser\Node\Expr[] $post_conditions
* @param array<int, string> $asserted_vars
* @param Context $loop_context
* @param Context $outer_context
*
* @return false|null
*/
public function analyzeLoop(
array $stmts,
array $pre_conditions,
array $post_conditions,
Context $loop_context,
Context $outer_context
) {
$traverser = new PhpParser\NodeTraverser;
$assignment_mapper = new \Psalm\Visitor\AssignmentMapVisitor($loop_context->self);
$traverser->addVisitor($assignment_mapper);
$traverser->traverse($stmts);
$assignment_map = $assignment_mapper->getAssignmentMap();
$assignment_depth = 0;
$asserted_vars = [];
$pre_condition_clauses = [];
if ($pre_conditions) {
foreach ($pre_conditions as $pre_condition) {
$pre_condition_clauses = array_merge(
$pre_condition_clauses,
AlgebraChecker::getFormula(
$pre_condition,
$loop_context->self,
$this
)
);
}
} else {
$asserted_vars = Context::getNewOrUpdatedVarIds($outer_context, $loop_context);
}
if ($assignment_map) {
$first_var_id = array_keys($assignment_map)[0];
$assignment_depth = self::getAssignmentMapDepth($first_var_id, $assignment_map);
}
$loop_context->parent_context = $outer_context;
if ($assignment_depth === 0) {
foreach ($pre_conditions as $pre_condition) {
if ($this->applyPreConditionToLoopContext(
$pre_condition,
$pre_condition_clauses,
$loop_context,
$outer_context
) === false) {
return false;
}
}
$this->analyze($stmts, $loop_context, $outer_context);
foreach ($post_conditions as $post_condition) {
if (ExpressionChecker::analyze($this, $post_condition, $loop_context) === false) {
return false;
}
}
} else {
// record all the vars that existed before we did the first pass through the loop
$pre_loop_context = clone $loop_context;
$pre_outer_context = clone $outer_context;
IssueBuffer::startRecording();
foreach ($pre_conditions as $pre_condition) {
if ($this->applyPreConditionToLoopContext(
$pre_condition,
$pre_condition_clauses,
$loop_context,
$outer_context
) === false) {
return false;
}
}
$this->analyze($stmts, $loop_context, $outer_context);
foreach ($post_conditions as $post_condition) {
if (ExpressionChecker::analyze($this, $post_condition, $loop_context) === false) {
return false;
}
}
$recorded_issues = IssueBuffer::clearRecordingLevel();
IssueBuffer::stopRecording();
for ($i = 0; $i < $assignment_depth; ++$i) {
$vars_to_remove = [];
$has_changes = false;
foreach ($loop_context->vars_in_scope as $var_id => $type) {
if (in_array($var_id, $asserted_vars, true)) {
// set the vars to whatever the while/foreach loop expects them to be
if ((string)$type !== (string)$pre_loop_context->vars_in_scope[$var_id]) {
$loop_context->vars_in_scope[$var_id] = $pre_loop_context->vars_in_scope[$var_id];
$has_changes = true;
}
} elseif (isset($pre_outer_context->vars_in_scope[$var_id])) {
$pre_outer = (string)$pre_outer_context->vars_in_scope[$var_id];
if ((string)$type !== $pre_outer ||
(string)$outer_context->vars_in_scope[$var_id] !== $pre_outer
) {
$has_changes = true;
// widen the foreach context type with the initial context type
$loop_context->vars_in_scope[$var_id] = Type::combineUnionTypes(
$loop_context->vars_in_scope[$var_id],
$outer_context->vars_in_scope[$var_id]
);
// if there's a change, invalidate related clauses
$pre_loop_context->removeVarFromConflictingClauses($var_id);
}
} else {
$vars_to_remove[] = $var_id;
}
}
foreach ($asserted_vars as $var_id) {
if (!isset($loop_context->vars_in_scope[$var_id])) {
$loop_context->vars_in_scope[$var_id] = $pre_loop_context->vars_in_scope[$var_id];
}
}
// if there are no changes to the types, no need to re-examine
if (!$has_changes) {
break;
}
// remove vars that were defined in the foreach
foreach ($vars_to_remove as $var_id) {
unset($loop_context->vars_in_scope[$var_id]);
}
$loop_context->clauses = $pre_loop_context->clauses;
IssueBuffer::startRecording();
foreach ($pre_conditions as $pre_condition) {
if ($this->applyPreConditionToLoopContext(
$pre_condition,
$pre_condition_clauses,
$loop_context,
$outer_context
) === false) {
return false;
}
}
$this->analyze($stmts, $loop_context, $outer_context);
foreach ($post_conditions as $post_condition) {
if (ExpressionChecker::analyze($this, $post_condition, $loop_context) === false) {
return false;
}
}
$recorded_issues = IssueBuffer::clearRecordingLevel();
IssueBuffer::stopRecording();
}
if ($recorded_issues) {
foreach ($recorded_issues as $recorded_issue) {
// if we're not in any loops then this will just result in the issue being emitted
IssueBuffer::bubbleUp($recorded_issue);
}
}
}
foreach ($outer_context->vars_in_scope as $var_id => $type) {
if ($type->isMixed()) {
continue;
}
if (!isset($loop_context->vars_in_scope[$var_id])) {
unset($outer_context->vars_in_scope[$var_id]);
continue;
}
if ($loop_context->vars_in_scope[$var_id]->isMixed()) {
$outer_context->vars_in_scope[$var_id] = $loop_context->vars_in_scope[$var_id];
}
if ((string) $loop_context->vars_in_scope[$var_id] !== (string) $type) {
$outer_context->vars_in_scope[$var_id] = Type::combineUnionTypes(
$outer_context->vars_in_scope[$var_id],
$loop_context->vars_in_scope[$var_id]
);
$outer_context->removeVarFromConflictingClauses($var_id);
}
}
if ($pre_conditions && $pre_condition_clauses && !ScopeChecker::doesEverBreak($stmts)) {
// if the loop contains an assertion and there are no break statements, we can negate that assertion
// and apply it to the current context
$negated_pre_condition_types = AlgebraChecker::getTruthsFromFormula(
AlgebraChecker::negateFormula($pre_condition_clauses)
);
if ($negated_pre_condition_types) {
$changed_var_ids = [];
$vars_in_scope_reconciled = TypeChecker::reconcileKeyedTypes(
$negated_pre_condition_types,
$loop_context->vars_in_scope,
$changed_var_ids,
[],
$this,
new CodeLocation($this->getSource(), $pre_conditions[0]),
$this->getSuppressedIssues()
);
if ($vars_in_scope_reconciled === false) {
return false;
}
foreach ($outer_context->vars_in_scope as $var_id => $type) {
if (isset($vars_in_scope_reconciled[$var_id])) {
$outer_context->vars_in_scope[$var_id] = $vars_in_scope_reconciled[$var_id];
}
}
foreach ($changed_var_ids as $changed_var_id) {
$outer_context->removeVarFromConflictingClauses($changed_var_id);
}
}
}
}
/**
* @param PhpParser\Node\Expr $pre_condition
* @param array<int, Clause> $pre_condition_clauses
* @param Context $loop_context
* @param Context $outer_context
*
* @return false|null
*/
private function applyPreConditionToLoopContext(
PhpParser\Node\Expr $pre_condition,
array $pre_condition_clauses,
Context $loop_context,
Context $outer_context
) {
$pre_referenced_var_ids = $loop_context->referenced_var_ids;
$loop_context->referenced_var_ids = [];
$loop_context->inside_conditional = true;
if (ExpressionChecker::analyze($this, $pre_condition, $loop_context) === false) {
return false;
}
$loop_context->inside_conditional = false;
$new_referenced_var_ids = $loop_context->referenced_var_ids;
$loop_context->referenced_var_ids = array_merge($pre_referenced_var_ids, $new_referenced_var_ids);
$asserted_vars = array_keys(AlgebraChecker::getTruthsFromFormula($pre_condition_clauses));
$loop_context->clauses = AlgebraChecker::simplifyCNF(
array_merge($outer_context->clauses, $pre_condition_clauses)
);
$reconcilable_while_types = AlgebraChecker::getTruthsFromFormula($loop_context->clauses);
// if the while has an or as the main component, we cannot safely reason about it
if ($pre_condition instanceof PhpParser\Node\Expr\BinaryOp &&
($pre_condition instanceof PhpParser\Node\Expr\BinaryOp\BooleanOr ||
$pre_condition instanceof PhpParser\Node\Expr\BinaryOp\LogicalOr)
) {
// do nothing
} else {
$changed_var_ids = [];
$pre_condition_vars_in_scope_reconciled = TypeChecker::reconcileKeyedTypes(
$reconcilable_while_types,
$loop_context->vars_in_scope,
$changed_var_ids,
$new_referenced_var_ids,
$this,
new CodeLocation($this->getSource(), $pre_condition),
$this->getSuppressedIssues()
);
if ($pre_condition_vars_in_scope_reconciled === false) {
return false;
}
$loop_context->vars_in_scope = $pre_condition_vars_in_scope_reconciled;
}
}
/**
* @param string $first_var_id
* @param array<string, array<string, bool>> $assignment_map
*
* @return int
*/
private static function getAssignmentMapDepth($first_var_id, array $assignment_map)
{
$max_depth = 0;
$assignment_var_ids = $assignment_map[$first_var_id];
unset($assignment_map[$first_var_id]);
foreach ($assignment_var_ids as $assignment_var_id => $_) {
$depth = 1;
if (isset($assignment_map[$assignment_var_id])) {
$depth = 1 + self::getAssignmentMapDepth($assignment_var_id, $assignment_map);
}
if ($depth > $max_depth) {
$max_depth = $depth;
}
}
return $max_depth;
}
/**
* @param PhpParser\Node\Stmt\Static_ $stmt
* @param Context $context
@ -946,7 +605,7 @@ class StatementsChecker extends SourceChecker implements StatementsSource
{
$do_context = clone $context;
$this->analyzeLoop($stmt->stmts, [], [], $do_context, $context);
LoopChecker::analyze($this, $stmt->stmts, [], [], $do_context, $context);
foreach ($context->vars_in_scope as $var => $type) {
if ($type->isMixed()) {