Prechádzať zdrojové kódy

Refactor AppFrameworkTainter to make it more readable

Signed-off-by: jld3103 <jld3103yt@gmail.com>
jld3103 1 rok pred
rodič
commit
e7926b9644
1 zmenil súbory, kde vykonal 23 pridanie a 20 odobranie
  1. 23 20
      build/psalm/AppFrameworkTainter.php

+ 23 - 20
build/psalm/AppFrameworkTainter.php

@@ -22,6 +22,7 @@
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  * SOFTWARE.
  */
+
 use Psalm\CodeLocation;
 use Psalm\Plugin\EventHandler\AfterFunctionLikeAnalysisInterface;
 use Psalm\Plugin\EventHandler\Event\AfterFunctionLikeAnalysisEvent;
@@ -29,27 +30,29 @@ use Psalm\Type\TaintKindGroup;
 
 class AppFrameworkTainter implements AfterFunctionLikeAnalysisInterface {
 	public static function afterStatementAnalysis(AfterFunctionLikeAnalysisEvent $event): ?bool {
-		if ($event->getStatementsSource()->getFQCLN() !== null) {
-			if ($event->getCodebase()->classExtendsOrImplements($event->getStatementsSource()->getFQCLN(), \OCP\AppFramework\Controller::class)) {
-				if ($event->getStmt() instanceof PhpParser\Node\Stmt\ClassMethod) {
-					if ($event->getStmt()->isPublic() && !$event->getStmt()->isMagic()) {
-						foreach ($event->getStmt()->params as $i => $param) {
-							$expr_type = new Psalm\Type\Union([new Psalm\Type\Atomic\TString()]);
-							$expr_identifier = (strtolower($event->getStatementsSource()->getFQCLN()) . '::' . strtolower($event->getFunctionlikeStorage()->cased_name) . '#' . ($i + 1));
-
-							if ($expr_type) {
-								$event->getCodebase()->addTaintSource(
-									$expr_type,
-									$expr_identifier,
-									TaintKindGroup::ALL_INPUT,
-									new CodeLocation($event->getStatementsSource(), $param)
-								);
-							}
-						}
-					}
-				}
-			}
+		if ($event->getStatementsSource()->getFQCLN() === null) {
+			return null;
+		}
+		if (!$event->getCodebase()->classExtendsOrImplements($event->getStatementsSource()->getFQCLN(), \OCP\AppFramework\Controller::class)) {
+			return null;
+		}
+		if (!($event->getStmt() instanceof PhpParser\Node\Stmt\ClassMethod)) {
+			return null;
 		}
+		if (!$event->getStmt()->isPublic() || $event->getStmt()->isMagic()) {
+			return null;
+		}
+		foreach ($event->getStmt()->params as $i => $param) {
+			$expr_type = new Psalm\Type\Union([new Psalm\Type\Atomic\TString()]);
+			$expr_identifier = (strtolower($event->getStatementsSource()->getFQCLN()) . '::' . strtolower($event->getFunctionlikeStorage()->cased_name) . '#' . ($i + 1));
+			$event->getCodebase()->addTaintSource(
+				$expr_type,
+				$expr_identifier,
+				TaintKindGroup::ALL_INPUT,
+				new CodeLocation($event->getStatementsSource(), $param)
+			);
+		}
+
 		return null;
 	}
 }