[Coverage] Avoid profiling functions with body macros

These replace the body of the function, and as such
shouldn't be profiled.

Author: hamishknight

lib/SIL/IR/SILProfiler.cpp

@@ -114,12 +114,23 @@ static bool shouldProfile(SILDeclRef Constant) {
     }
   }
 
-  // Do not profile AST nodes in unavailable contexts.
   if (auto *D = DC->getInnermostDeclarationDeclContext()) {
+    // Do not profile AST nodes in unavailable contexts.
     if (D->getSemanticUnavailableAttr()) {
       LLVM_DEBUG(llvm::dbgs() << "Skipping ASTNode: unavailable context\n");
       return false;
     }
+
+    // Do not profile functions that have had their bodies replaced (e.g
+    // function body macros).
+    // TODO: If/when preamble macros become an official feature, we'll
+    // need to be more nuanced here.
+    if (auto *AFD = dyn_cast<AbstractFunctionDecl>(D)) {
+      if (AFD->getOriginalBodySourceRange() != AFD->getBodySourceRange()) {
+        LLVM_DEBUG(llvm::dbgs() << "Skipping function: body replaced\n");
+        return false;
+      }
+    }
   }
 
   // Do not profile code that hasn't been written by the user.

test/Macros/Inputs/syntax_macro_definitions.swift

@@ -2350,6 +2350,20 @@ public struct RemoteBodyMacro: BodyMacro {
   }
 }
 
+@_spi(ExperimentalLanguageFeature)
+public struct BodyMacroWithControlFlow: BodyMacro {
+  public static func expansion(
+    of node: AttributeSyntax,
+    providingBodyFor declaration: some DeclSyntaxProtocol & WithOptionalCodeBlockSyntax,
+    in context: some MacroExpansionContext
+  ) throws -> [CodeBlockItemSyntax] {
+    [
+      "guard .random() else { return }",
+      "_ = try throwingFn()"
+    ]
+  }
+}
+
 @_spi(ExperimentalLanguageFeature)
 public struct TracedPreambleMacro: PreambleMacro {
   public static func expansion(

test/Profiler/coverage_macros.swift

@@ -28,6 +28,9 @@ macro fnCall<T>(_: () throws -> T, _: () throws -> T) -> (T, T) = #externalMacro
 @freestanding(expression)
 macro id<T>(_: T) -> T = #externalMacro(module: "MacroDefinition", type: "TupleMacro")
 
+@attached(body)
+macro BodyMacroWithControlFlow() = #externalMacro(module: "MacroDefinition", type: "BodyMacroWithControlFlow")
+
 // This needs to be matched up here due to the sorting of the SIL; just make
 // sure we emit the counter increments for the error branches.
 // CHECK-LABEL: sil hidden @$s15coverage_macros5test2Si_SityKF
@@ -104,3 +107,15 @@ func test5() throws -> (Int, Int, Int) {        // CHECK-NEXT: [[@LINE]]:40 -> [
                         : try throwingFn())     // CHECK-NEXT: [[@LINE]]:44 -> [[@LINE+1]]:19 : ((((0 - 2) - 4) - 6) - 7)
   return (x, y, z)                              // CHECK-NEXT: }
 }
+
+// Not profiled.
+@BodyMacroWithControlFlow
+func test6() throws
+
+
+// Not profiled.
+@BodyMacroWithControlFlow
+func test7() throws {
+  guard .random() else { return }
+  print("hello")
+}