Refactor AppFrameworkTainter to make it more readable

Signed-off-by: jld3103 <jld3103yt@gmail.com>
This commit is contained in:
jld3103 2023-03-31 17:19:12 +02:00
Родитель 14283001a5
Коммит e7926b9644
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 9062417B9E8EB7B3
1 изменённых файлов: 23 добавлений и 20 удалений

Просмотреть файл

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