Refactor AppFrameworkTainter to make it more readable

Signed-off-by: jld3103 <jld3103yt@gmail.com>
pull/37504/head
jld3103 1 year ago
parent 14283001a5
commit e7926b9644
No known key found for this signature in database
GPG Key ID: 9062417B9E8EB7B3

@ -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;
}
}

Loading…
Cancel
Save