[Instrumentation] Do not request sanitizers for naked functions

Sanitizers instrumentation may be incompatible with naked functions,
which lack of standard prologue/epilogue.

Author: antoniofrighetto

llvm/lib/Transforms/Instrumentation/AddressSanitizer.cpp

@@ -2960,6 +2960,10 @@ bool AddressSanitizer::instrumentFunction(Function &F,
 
   bool FunctionModified = false;
 
+  // Do not apply any instrumentation for naked functions.
+  if (F.hasFnAttribute(Attribute::Naked))
+    return FunctionModified;
+
   // If needed, insert __asan_init before checking for SanitizeAddress attr.
   // This function needs to be called even if the function body is not
   // instrumented.

llvm/lib/Transforms/Instrumentation/HWAddressSanitizer.cpp

@@ -1598,6 +1598,10 @@ void HWAddressSanitizer::sanitizeFunction(Function &F,
   if (&F == HwasanCtorFunction)
     return;
 
+  // Do not apply any instrumentation for naked functions.
+  if (F.hasFnAttribute(Attribute::Naked))
+    return;
+
   if (!F.hasFnAttribute(Attribute::SanitizeHWAddress))
     return;
 

llvm/lib/Transforms/Instrumentation/SanitizerBinaryMetadata.cpp

@@ -258,6 +258,9 @@ bool SanitizerBinaryMetadata::run() {
 void SanitizerBinaryMetadata::runOn(Function &F, MetadataInfoSet &MIS) {
   if (F.empty())
     return;
+  // Do not apply any instrumentation for naked functions.
+  if (F.hasFnAttribute(Attribute::Naked))
+    return;
   if (F.hasFnAttribute(Attribute::DisableSanitizerInstrumentation))
     return;
   if (Ignorelist && Ignorelist->inSection("metadata", "fun", F.getName()))

llvm/lib/Transforms/Instrumentation/SanitizerCoverage.cpp

@@ -629,6 +629,9 @@ void ModuleSanitizerCoverage::instrumentFunction(Function &F) {
     return;
   if (Blocklist && Blocklist->inSection("coverage", "fun", F.getName()))
     return;
+  // Do not apply any instrumentation for naked functions.
+  if (F.hasFnAttribute(Attribute::Naked))
+    return;
   if (F.hasFnAttribute(Attribute::NoSanitizeCoverage))
     return;
   if (F.hasFnAttribute(Attribute::DisableSanitizerInstrumentation))

llvm/test/Instrumentation/sanitizers-naked.ll

@@ -38,7 +38,7 @@ define void @naked_function() naked {
 ; CHECK-SANCOV-NEXT:    unreachable
 ;
 ; CHECK-SANMD-LABEL: define void @naked_function(
-; CHECK-SANMD-SAME: ) #[[ATTR0:[0-9]+]] !pcsections [[META0:![0-9]+]] {
+; CHECK-SANMD-SAME: ) #[[ATTR0:[0-9]+]] {
 ; CHECK-SANMD-NEXT:    call void asm sideeffect "nop", ""()
 ; CHECK-SANMD-NEXT:    unreachable
 ;
@@ -79,16 +79,12 @@ define void @naked_function_with_msan() sanitize_memory naked {
 
 define void @naked_function_with_hwasan() sanitize_hwaddress naked {
 ; CHECK-HWASAN-LABEL: define void @naked_function_with_hwasan(
-; CHECK-HWASAN-SAME: ) #[[ATTR4:[0-9]+]] {
+; CHECK-HWASAN-SAME: ) #[[ATTR4:[0-9]+]] personality ptr @__hwasan_personality_thunk {
 ; CHECK-HWASAN-NEXT:    call void asm sideeffect "nop", ""()
 ; CHECK-HWASAN-NEXT:    unreachable
 ;
   call void asm sideeffect "nop", ""()
   unreachable
 }
-;.
-; CHECK-SANMD: [[META0]] = !{!"sanmd_covered2!C", [[META1:![0-9]+]]}
-; CHECK-SANMD: [[META1]] = !{i64 1}
-;.
 ;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line:
 ; CHECK-DFSAN: {{.*}}