Merge pull request #63843 from rxwei/fix-macro-decl-module-crash

[Macros] Fix `MacroDecl` serialization crash with custom parameter attributes

Author: rxwei

include/swift/AST/Decl.h

@@ -8418,6 +8418,9 @@ class MacroDecl : public GenericContext, public ValueDecl {
   /// Retrieve the definition of this macro.
   MacroDefinition getDefinition() const;
 
+  /// Retrieve the parameter list of this macro.
+  ParameterList *getParameterList() const { return parameterList; }
+
   /// Retrieve the builtin macro kind for this macro, or \c None if it is a
   /// user-defined macro with no special semantics.
   Optional<BuiltinMacroKind> getBuiltinKind() const;

lib/Sema/TypeCheckAccess.cpp

@@ -1174,7 +1174,6 @@ class UsableFromInlineChecker : public AccessControlCheckerBase,
   UNINTERESTING(Destructor) // Always correct.
   UNINTERESTING(Accessor) // Handled by the Var or Subscript.
   UNINTERESTING(OpaqueType) // Handled by the Var or Subscript.
-  UNINTERESTING(Macro)
 
   /// If \p VD's layout is exposed by a @frozen struct or class, return said
   /// struct or class.
@@ -1569,6 +1568,10 @@ class UsableFromInlineChecker : public AccessControlCheckerBase,
     }
   }
 
+  void visitMacroDecl(MacroDecl *MD) {
+    // FIXME: Check access of macro generic parameters, parameters and result
+  }
+
   void visitEnumElementDecl(EnumElementDecl *EED) {
     if (!EED->hasAssociatedValues())
       return;

lib/Sema/TypeCheckDeclPrimary.cpp

@@ -2019,6 +2019,8 @@ class DeclChecker : public DeclVisitor<DeclChecker> {
     if (!MD->getAttrs().hasAttribute<MacroRoleAttr>(/*AllowInvalid*/ true))
       MD->diagnose(diag::macro_without_role, MD->getName());
 
+    TypeChecker::checkParameterList(MD->getParameterList(), MD);
+
     // Check the macro definition.
     switch (auto macroDef = MD->getDefinition()) {
     case MacroDefinition::Kind::Undefined:

test/Serialization/Inputs/def_macros.swift

@@ -24,3 +24,10 @@ public struct Base {
 public struct TestMacroArgTypechecking {
   public var value: Int
 }
+
+@resultBuilder
+public struct Builder {
+  static func buildBlock(_: Int...) -> Void {}
+}
+@freestanding(expression)
+public macro macroWithBuilderArgs(@Builder _: () -> Void) = #externalMacro(module: "A", type: "B")

test/Serialization/macros.swift

@@ -5,6 +5,8 @@
 // RUN: %target-build-swift -I %swift-host-lib-dir -L %swift-host-lib-dir -emit-library -o %t/%target-library-name(MacroDefinition) -module-name=MacroDefinition %S/Inputs/def_macro_plugin.swift -g -no-toolchain-stdlib-rpath
 // RUN: %target-swift-frontend -emit-module -o %t/def_macros.swiftmodule %S/Inputs/def_macros.swift -module-name def_macros -enable-experimental-feature Macros -load-plugin-library %t/%target-library-name(MacroDefinition) -I %swift-host-lib-dir
 // RUN: %target-swift-frontend -typecheck -I%t -verify %s -verify-ignore-unknown  -enable-experimental-feature Macros -load-plugin-library %t/%target-library-name(MacroDefinition) -I %swift-host-lib-dir
+// RUN: llvm-bcanalyzer %t/def_macros.swiftmodule | %FileCheck %s
+// RUN: %target-swift-ide-test(mock-sdk: %clang-importer-sdk) -print-module -module-to-print=def_macros -I %t -source-filename=%s | %FileCheck -check-prefix=CHECK-PRINT %s
 // REQUIRES: OS=macosx
 
 import def_macros
@@ -19,3 +21,9 @@ func test(a: Int, b: Int) {
 struct TestStruct {
   @myWrapper var x: Int
 }
+
+// CHECK: MACRO_DECL
+
+// CHECK-NOT: UnknownCode
+
+// CHECK-PRINT-DAG: macro macroWithBuilderArgs(@Builder _: () -> Void)