diff options
Diffstat (limited to 'lib/CodeGen/CodeGen.cc')
| -rw-r--r-- | lib/CodeGen/CodeGen.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/lib/CodeGen/CodeGen.cc b/lib/CodeGen/CodeGen.cc index f7c4de6..c9d8c1a 100644 --- a/lib/CodeGen/CodeGen.cc +++ b/lib/CodeGen/CodeGen.cc @@ -161,6 +161,11 @@ void CodeGen::selectUnit(size_t Unit) { CurrentUnit = Unit; DiagnosticSource = Units->size() > 1 ? (*Units)[Unit].Path : ""; FuncGen->setCurrentUnit(Unit, (*Units)[Unit].Path); + RuntimeChecker->setLibraryMode((*Units)[Unit].IsLibrary); + FuncGen->setCallHook([this](const FuncSig &Sig) { + if (!(*Units)[CurrentUnit].IsLibrary && (*Units)[Sig.OwnerUnit].IsLibrary) + RuntimeChecker->setUserLine(DiagnosticLine); + }); } void CodeGen::importSymbols() { |