From 29a9d8983a40ed62ee11a6acf2acda6c5cfe2748 Mon Sep 17 00:00:00 2001 From: Alexey Samsonov Date: Wed, 22 Oct 2014 19:34:25 +0000 Subject: [PATCH] Fixup for r220403: Use getFileLoc() instead of getSpellingLoc() in SanitizerBlacklist. This also handles the case where function name (not its body) is obtained from macro expansion. git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@220407 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Basic/SanitizerBlacklist.cpp | 2 +- test/CodeGen/address-safety-attr.cpp | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/lib/Basic/SanitizerBlacklist.cpp b/lib/Basic/SanitizerBlacklist.cpp index c894129e2a..ea5b8d0da8 100644 --- a/lib/Basic/SanitizerBlacklist.cpp +++ b/lib/Basic/SanitizerBlacklist.cpp @@ -41,6 +41,6 @@ bool SanitizerBlacklist::isBlacklistedFile(StringRef FileName, bool SanitizerBlacklist::isBlacklistedLocation(SourceLocation Loc, StringRef Category) const { return !Loc.isInvalid() && - isBlacklistedFile(SM.getFilename(SM.getSpellingLoc(Loc)), Category); + isBlacklistedFile(SM.getFilename(SM.getFileLoc(Loc)), Category); } diff --git a/test/CodeGen/address-safety-attr.cpp b/test/CodeGen/address-safety-attr.cpp index 0d585c7449..031d013a9b 100644 --- a/test/CodeGen/address-safety-attr.cpp +++ b/test/CodeGen/address-safety-attr.cpp @@ -66,13 +66,19 @@ int BlacklistedFunction(int *a) { return *a; } #define GENERATE_FUNC(name) \ int name(int *a) { return *a; } - // WITHOUT: GeneratedFunction{{.*}}) [[NOATTR]] // BLFILE: GeneratedFunction{{.*}}) [[NOATTR]] // BLFUNC: GeneratedFunction{{.*}}) [[WITH]] // ASAN: GeneratedFunction{{.*}}) [[WITH]] GENERATE_FUNC(GeneratedFunction) +#define GENERATE_NAME(name) name##_generated +// WITHOUT: Function_generated{{.*}}) [[NOATTR]] +// BLFILE: Function_generated{{.*}}) [[NOATTR]] +// BLFUNC: Function_generated{{.*}}) [[WITH]] +// ASAN: Function_generated{{.*}}) [[WITH]] +int GENERATE_NAME(Function)(int *a) { return *a; } + // WITHOUT: TemplateAddressSafetyOk{{.*}}) [[NOATTR]] // BLFILE: TemplateAddressSafetyOk{{.*}}) [[NOATTR]] // BLFUNC: TemplateAddressSafetyOk{{.*}}) [[WITH]] -- 2.40.0