mirror of
https://github.com/github/codeql.git
synced 2025-12-22 11:46:32 +01:00
19506dae74
This transfers the current state of `StmtVisitor` in the PoC, plus some changes required for the update to swift 5.6. Also `getLabel` in `SwiftDispatcher` got renamed to `createLabel`, and is now correctly outputting the label assignment to the trap file.
41 lines
1.4 KiB
C++
41 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include "swift/extractor/SwiftDispatcher.h"
|
|
#include "swift/extractor/visitors/DeclVisitor.h"
|
|
#include "swift/extractor/visitors/ExprVisitor.h"
|
|
#include "swift/extractor/visitors/StmtVisitor.h"
|
|
#include "swift/extractor/visitors/TypeVisitor.h"
|
|
#include "swift/extractor/visitors/TypeReprVisitor.h"
|
|
#include "swift/extractor/visitors/PatternVisitor.h"
|
|
|
|
namespace codeql {
|
|
|
|
class SwiftVisitor : private SwiftDispatcher {
|
|
public:
|
|
using SwiftDispatcher::SwiftDispatcher;
|
|
|
|
template <typename T>
|
|
void extract(T* entity) {
|
|
fetchLabel(entity);
|
|
}
|
|
|
|
private:
|
|
void visit(swift::Decl* decl) override { declVisitor.visit(decl); }
|
|
void visit(swift::Stmt* stmt) override { stmtVisitor.visit(stmt); }
|
|
void visit(swift::StmtCondition* cond) override { stmtVisitor.visitStmtCondition(cond); }
|
|
void visit(swift::CaseLabelItem* item) override { stmtVisitor.visitCaseLabelItem(item); }
|
|
void visit(swift::Expr* expr) override { exprVisitor.visit(expr); }
|
|
void visit(swift::Pattern* pattern) override { patternVisitor.visit(pattern); }
|
|
void visit(swift::TypeRepr* type) override { typeReprVisitor.visit(type); }
|
|
void visit(swift::TypeBase* type) override { typeVisitor.visit(type); }
|
|
|
|
DeclVisitor declVisitor{*this};
|
|
ExprVisitor exprVisitor{*this};
|
|
StmtVisitor stmtVisitor{*this};
|
|
TypeReprVisitor typeReprVisitor{*this};
|
|
TypeVisitor typeVisitor{*this};
|
|
PatternVisitor patternVisitor{*this};
|
|
};
|
|
|
|
} // namespace codeql
|