mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
d4ac04d25e
Removes duplicated logic from the implementations of FileUnit::lookupValue, and simplifies the interface to ModuleDecl::lookupValue, where everyone was passing an empty (non-filtering) access path anyway /except/ during actual lookup from source code. No functionality change.
136 lines
4.2 KiB
C++
136 lines
4.2 KiB
C++
//===--- InstrumenterSupport.cpp - Instrumenter Support -------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the supporting functions for writing instrumenters of
|
|
// the Swift AST.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "InstrumenterSupport.h"
|
|
#include "swift/AST/DiagnosticSuppression.h"
|
|
#include "swift/Demangling/Punycode.h"
|
|
#include "llvm/Support/Path.h"
|
|
|
|
using namespace swift;
|
|
using namespace swift::instrumenter_support;
|
|
|
|
namespace {
|
|
|
|
class ErrorGatherer : public DiagnosticConsumer {
|
|
private:
|
|
bool error = false;
|
|
DiagnosticEngine &diags;
|
|
|
|
public:
|
|
ErrorGatherer(DiagnosticEngine &diags) : diags(diags) {
|
|
diags.addConsumer(*this);
|
|
}
|
|
~ErrorGatherer() override { diags.takeConsumers(); }
|
|
void
|
|
handleDiagnostic(SourceManager &SM, SourceLoc Loc, DiagnosticKind Kind,
|
|
StringRef FormatString,
|
|
ArrayRef<DiagnosticArgument> FormatArgs,
|
|
const DiagnosticInfo &Info,
|
|
const SourceLoc bufferIndirectlyCausingDiagnostic) override {
|
|
if (Kind == swift::DiagnosticKind::Error) {
|
|
error = true;
|
|
}
|
|
DiagnosticEngine::formatDiagnosticText(llvm::errs(), FormatString,
|
|
FormatArgs);
|
|
llvm::errs() << "\n";
|
|
}
|
|
bool hadError() { return error; }
|
|
};
|
|
|
|
|
|
class ErrorFinder : public ASTWalker {
|
|
bool error = false;
|
|
|
|
public:
|
|
ErrorFinder() {}
|
|
std::pair<bool, Expr *> walkToExprPre(Expr *E) override {
|
|
if (isa<ErrorExpr>(E) || !E->getType() || E->getType()->hasError()) {
|
|
error = true;
|
|
return {false, E};
|
|
}
|
|
return {true, E};
|
|
}
|
|
bool walkToDeclPre(Decl *D) override {
|
|
if (auto *VD = dyn_cast<ValueDecl>(D)) {
|
|
if (!VD->hasInterfaceType() || VD->getInterfaceType()->hasError()) {
|
|
error = true;
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
bool hadError() { return error; }
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
InstrumenterBase::InstrumenterBase(ASTContext &C, DeclContext *DC)
|
|
: Context(C), TypeCheckDC(DC), CF(*this) {
|
|
// Prefixes for module and file vars
|
|
const std::string builtinPrefix = "__builtin";
|
|
const std::string modulePrefix = "_pg_module_";
|
|
const std::string filePrefix = "_pg_file_";
|
|
|
|
// Setup Module identifier
|
|
std::string moduleName = TypeCheckDC->getParentModule()->getName().str();
|
|
Identifier moduleIdentifier =
|
|
Context.getIdentifier(builtinPrefix + modulePrefix + moduleName);
|
|
|
|
SmallVector<ValueDecl *, 1> results;
|
|
TypeCheckDC->getParentModule()->lookupValue(
|
|
moduleIdentifier, NLKind::UnqualifiedLookup, results);
|
|
|
|
ModuleIdentifier = (results.size() == 1) ? moduleIdentifier : Identifier();
|
|
|
|
// Setup File identifier
|
|
StringRef filePath = TypeCheckDC->getParentSourceFile()->getFilename();
|
|
StringRef fileName = llvm::sys::path::stem(filePath);
|
|
|
|
std::string filePunycodeName;
|
|
Punycode::encodePunycodeUTF8(fileName, filePunycodeName, true);
|
|
Identifier fileIdentifier =
|
|
Context.getIdentifier(builtinPrefix + modulePrefix + moduleName +
|
|
filePrefix + filePunycodeName);
|
|
|
|
results.clear();
|
|
TypeCheckDC->getParentModule()->lookupValue(
|
|
fileIdentifier, NLKind::UnqualifiedLookup, results);
|
|
|
|
FileIdentifier = (results.size() == 1) ? fileIdentifier : Identifier();
|
|
}
|
|
|
|
void InstrumenterBase::anchor() {}
|
|
|
|
bool InstrumenterBase::doTypeCheckImpl(ASTContext &Ctx, DeclContext *DC,
|
|
Expr * &parsedExpr) {
|
|
DiagnosticSuppression suppression(Ctx.Diags);
|
|
ErrorGatherer errorGatherer(Ctx.Diags);
|
|
|
|
TypeChecker &TC = TypeChecker::createForContext(Ctx);
|
|
|
|
TC.typeCheckExpression(parsedExpr, DC);
|
|
|
|
if (parsedExpr) {
|
|
ErrorFinder errorFinder;
|
|
parsedExpr->walk(errorFinder);
|
|
if (!errorFinder.hadError() && !errorGatherer.hadError()) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|