WIP: KeyPath Calling Convention

include/swift/AST/ExtInfo.h

@@ -172,6 +172,13 @@ enum class SILFunctionTypeRepresentation : uint8_t {
   /// constructor). Except for
   /// handling the "this" argument, has the same behavior as "CFunctionPointer".
   CXXMethod,
+
+  /// A KeyPath accessor function, which is thin and also uses the variadic length
+  /// generic components serialization in trailing buffer.
+  KeyPathAccessorGetter,
+  KeyPathAccessorSetter,
+  KeyPathAccessorEquals,
+  KeyPathAccessorHash,
 };
 
 /// Returns true if the function with this convention doesn't carry a context.
@@ -202,6 +209,10 @@ isThinRepresentation(SILFunctionTypeRepresentation rep) {
   case SILFunctionTypeRepresentation::CFunctionPointer:
   case SILFunctionTypeRepresentation::Closure:
   case SILFunctionTypeRepresentation::CXXMethod:
+  case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+  case SILFunctionTypeRepresentation::KeyPathAccessorHash:
     return true;
   }
   llvm_unreachable("Unhandled SILFunctionTypeRepresentation in switch.");
@@ -214,6 +225,30 @@ isThickRepresentation(Repr repr) {
   return !isThinRepresentation(repr);
 }
 
+/// Returns true if the function with this convention doesn't carry a context.
+constexpr bool
+isKeyPathAccessorRepresentation(SILFunctionTypeRepresentation rep) {
+  switch (rep) {
+    case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+    case SILFunctionTypeRepresentation::KeyPathAccessorHash:
+      return true;
+    case SILFunctionTypeRepresentation::Thick:
+    case SILFunctionTypeRepresentation::Block:
+    case SILFunctionTypeRepresentation::Thin:
+    case SILFunctionTypeRepresentation::Method:
+    case SILFunctionTypeRepresentation::ObjCMethod:
+    case SILFunctionTypeRepresentation::WitnessMethod:
+    case SILFunctionTypeRepresentation::CFunctionPointer:
+    case SILFunctionTypeRepresentation::Closure:
+    case SILFunctionTypeRepresentation::CXXMethod:
+      return false;
+  }
+  llvm_unreachable("Unhandled SILFunctionTypeRepresentation in switch.");
+}
+
+
 constexpr SILFunctionTypeRepresentation
 convertRepresentation(FunctionTypeRepresentation rep) {
   switch (rep) {
@@ -225,6 +260,7 @@ convertRepresentation(FunctionTypeRepresentation rep) {
     return SILFunctionTypeRepresentation::Thin;
   case FunctionTypeRepresentation::CFunctionPointer:
     return SILFunctionTypeRepresentation::CFunctionPointer;
+    return SILFunctionTypeRepresentation::Thin;
   }
   llvm_unreachable("Unhandled FunctionTypeRepresentation!");
 }
@@ -245,6 +281,10 @@ convertRepresentation(SILFunctionTypeRepresentation rep) {
   case SILFunctionTypeRepresentation::ObjCMethod:
   case SILFunctionTypeRepresentation::WitnessMethod:
   case SILFunctionTypeRepresentation::Closure:
+  case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+  case SILFunctionTypeRepresentation::KeyPathAccessorHash:
     return None;
   }
   llvm_unreachable("Unhandled SILFunctionTypeRepresentation!");
@@ -264,6 +304,10 @@ constexpr bool canBeCalledIndirectly(SILFunctionTypeRepresentation rep) {
   case SILFunctionTypeRepresentation::ObjCMethod:
   case SILFunctionTypeRepresentation::Method:
   case SILFunctionTypeRepresentation::WitnessMethod:
+  case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+  case SILFunctionTypeRepresentation::KeyPathAccessorHash:
     return true;
   }
 
@@ -285,6 +329,10 @@ template <typename Repr> constexpr bool shouldStoreClangType(Repr repr) {
   case SILFunctionTypeRepresentation::Method:
   case SILFunctionTypeRepresentation::WitnessMethod:
   case SILFunctionTypeRepresentation::Closure:
+  case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+  case SILFunctionTypeRepresentation::KeyPathAccessorHash:
     return false;
   }
   llvm_unreachable("Unhandled SILFunctionTypeRepresentation.");
@@ -395,6 +443,10 @@ class ASTExtInfoBuilder {
     case SILFunctionTypeRepresentation::Thin:
     case SILFunctionTypeRepresentation::CFunctionPointer:
     case SILFunctionTypeRepresentation::Closure:
+    case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+    case SILFunctionTypeRepresentation::KeyPathAccessorHash:
       return false;
     case SILFunctionTypeRepresentation::ObjCMethod:
     case SILFunctionTypeRepresentation::Method:
@@ -633,6 +685,10 @@ SILFunctionLanguage getSILFunctionLanguage(SILFunctionTypeRepresentation rep) {
   case SILFunctionTypeRepresentation::Method:
   case SILFunctionTypeRepresentation::WitnessMethod:
   case SILFunctionTypeRepresentation::Closure:
+  case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+  case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+  case SILFunctionTypeRepresentation::KeyPathAccessorHash:
     return SILFunctionLanguage::Swift;
   }
 
@@ -651,17 +707,17 @@ class SILExtInfoBuilder {
   // and NumMaskBits must be updated, and they must match.
 
   //   |representation|pseudogeneric| noescape | concurrent | async |differentiability|
-  //   |    0 .. 3    |      4      |     5    |     6      |   7   |     8 .. 10     |
+  //   |    0 .. 4    |      5      |     6    |     7      |   8   |     9 .. 11     |
   //
   enum : unsigned {
-    RepresentationMask = 0xF << 0,
-    PseudogenericMask = 1 << 4,
-    NoEscapeMask = 1 << 5,
-    SendableMask = 1 << 6,
-    AsyncMask = 1 << 7,
-    DifferentiabilityMaskOffset = 8,
+    RepresentationMask = 0x1F << 0,
+    PseudogenericMask = 1 << 5,
+    NoEscapeMask = 1 << 6,
+    SendableMask = 1 << 7,
+    AsyncMask = 1 << 8,
+    DifferentiabilityMaskOffset = 9,
     DifferentiabilityMask = 0x7 << DifferentiabilityMaskOffset,
-    NumMaskBits = 11
+    NumMaskBits = 12
   };
 
   unsigned bits; // Naturally sized for speed.
@@ -755,6 +811,10 @@ class SILExtInfoBuilder {
     case Representation::Thin:
     case Representation::CFunctionPointer:
     case Representation::Closure:
+    case Representation::KeyPathAccessorGetter:
+    case Representation::KeyPathAccessorSetter:
+    case Representation::KeyPathAccessorEquals:
+    case Representation::KeyPathAccessorHash:
       return false;
     case Representation::ObjCMethod:
     case Representation::Method:
@@ -778,6 +838,10 @@ class SILExtInfoBuilder {
     case Representation::WitnessMethod:
     case Representation::Closure:
     case SILFunctionTypeRepresentation::CXXMethod:
+    case Representation::KeyPathAccessorGetter:
+    case Representation::KeyPathAccessorSetter:
+    case Representation::KeyPathAccessorEquals:
+    case Representation::KeyPathAccessorHash:
       return false;
     }
     llvm_unreachable("Unhandled Representation in switch.");

include/swift/AST/Types.h

@@ -340,7 +340,7 @@ class alignas(1 << TypeAlignInBits) TypeBase
 
 protected:
   enum { NumAFTExtInfoBits = 11 };
-  enum { NumSILExtInfoBits = 11 };
+  enum { NumSILExtInfoBits = 12 };
   union { uint64_t OpaqueBits;
 
   SWIFT_INLINE_BITFIELD_BASE(TypeBase, bitmax(NumTypeKindBits,8) +

include/swift/SIL/ApplySite.h

@@ -21,6 +21,7 @@
 #ifndef SWIFT_SIL_APPLYSITE_H
 #define SWIFT_SIL_APPLYSITE_H
 
+#include "swift/AST/ExtInfo.h"
 #include "swift/Basic/STLExtras.h"
 #include "swift/SIL/SILArgument.h"
 #include "swift/SIL/SILBasicBlock.h"
@@ -253,6 +254,10 @@ class ApplySite {
     case SILFunctionTypeRepresentation::ObjCMethod:
     case SILFunctionTypeRepresentation::WitnessMethod:
     case SILFunctionTypeRepresentation::Closure:
+    case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+    case SILFunctionTypeRepresentation::KeyPathAccessorHash:
       return true;
     case SILFunctionTypeRepresentation::Block:
     case SILFunctionTypeRepresentation::Thick:

lib/AST/ASTDumper.cpp

@@ -136,6 +136,10 @@ getSILFunctionTypeRepresentationString(SILFunctionType::Representation value) {
   case SILFunctionType::Representation::ObjCMethod: return "objc_method";
   case SILFunctionType::Representation::WitnessMethod: return "witness_method";
   case SILFunctionType::Representation::Closure: return "closure";
+  case SILFunctionType::Representation::KeyPathAccessorGetter: return "keypath_accessor_getter";
+  case SILFunctionType::Representation::KeyPathAccessorSetter: return "keypath_accessor_setter";
+  case SILFunctionType::Representation::KeyPathAccessorEquals: return "keypath_accessor_equals";
+  case SILFunctionType::Representation::KeyPathAccessorHash: return "keypath_accessor_hash";
   }
 
   llvm_unreachable("Unhandled SILFunctionTypeRepresentation in switch.");

lib/AST/ASTMangler.cpp

@@ -1872,6 +1872,12 @@ void ASTMangler::appendImplFunctionType(SILFunctionType *fn,
     case SILFunctionTypeRepresentation::WitnessMethod:
       OpArgs.push_back('W');
       break;
+    case SILFunctionTypeRepresentation::KeyPathAccessorGetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorSetter:
+    case SILFunctionTypeRepresentation::KeyPathAccessorEquals:
+    case SILFunctionTypeRepresentation::KeyPathAccessorHash:
+      OpArgs.push_back('P');
+      break;
   }
 
   // Coroutine kind.  This is mangled in all pointer auth modes.

lib/AST/ASTPrinter.cpp

@@ -5785,6 +5785,18 @@ class TypePrinter : public TypeVisitor<TypePrinter> {
       case SILFunctionType::Representation::Closure:
         Printer << "closure";
         break;
+      case SILFunctionType::Representation::KeyPathAccessorGetter:
+        Printer << "keypath_accessor_getter";
+        break;
+      case SILFunctionType::Representation::KeyPathAccessorSetter:
+        Printer << "keypath_accessor_setter";
+        break;
+      case SILFunctionType::Representation::KeyPathAccessorEquals:
+        Printer << "keypath_accessor_equals";
+        break;
+      case SILFunctionType::Representation::KeyPathAccessorHash:
+        Printer << "keypath_accessor_hash";
+        break;
       }
       Printer << ")";
       Printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
@@ -5869,6 +5881,18 @@ class TypePrinter : public TypeVisitor<TypePrinter> {
       case SILFunctionType::Representation::Closure:
         Printer << "closure";
         break;
+      case SILFunctionType::Representation::KeyPathAccessorGetter:
+        Printer << "keypath_accessor_getter";
+        break;
+      case SILFunctionType::Representation::KeyPathAccessorSetter:
+        Printer << "keypath_accessor_setter";
+        break;
+      case SILFunctionType::Representation::KeyPathAccessorEquals:
+        Printer << "keypath_accessor_equals";
+        break;
+      case SILFunctionType::Representation::KeyPathAccessorHash:
+        Printer << "keypath_accessor_hash";
+        break;
       }
       Printer << ")";
       Printer.printStructurePost(PrintStructureKind::BuiltinAttribute);

lib/AST/ClangTypeConverter.cpp

@@ -223,6 +223,10 @@ const clang::Type *ClangTypeConverter::getFunctionType(
   case SILFunctionType::Representation::ObjCMethod:
   case SILFunctionType::Representation::WitnessMethod:
   case SILFunctionType::Representation::Closure:
+  case SILFunctionType::Representation::KeyPathAccessorGetter:
+  case SILFunctionType::Representation::KeyPathAccessorSetter:
+  case SILFunctionType::Representation::KeyPathAccessorEquals:
+  case SILFunctionType::Representation::KeyPathAccessorHash:
     llvm_unreachable("Expected a C-compatible representation.");
   }
   llvm_unreachable("unhandled representation!");

lib/FrontendTool/TBD.cpp

@@ -66,13 +66,22 @@ bool swift::writeTBD(ModuleDecl *M, StringRef OutputFilename,
 ///   should be ignored (instead of potentially producing a diagnostic.)
 static bool isSymbolIgnored(const StringRef& name,
                             const llvm::Module &IRModule) {
-  if (llvm::Triple(IRModule.getTargetTriple()).isOSWindows()) {
+  llvm::Triple triple(IRModule.getTargetTriple());
+  if (triple.isOSWindows()) {
     // (SR-15938) Error when referencing #dsohandle in a Swift test on Windows
     // On Windows, ignore the lack of __ImageBase in the TBD file.
     if (name == "__ImageBase") {
       return true;
     }
   }
+  if (triple.isWasm()) {
+    // `__main_void`, which is called by `_start` in crt1.o, is artificially
+    // aliased in IR module when `main` doesn't take any params.
+    // This alias will be hidden after https://reviews.llvm.org/D75277
+    if (name == "__main_void") {
+      return true;
+    }
+  }
 
   return false;
 }

lib/IRGen/CallEmission.h

@@ -17,6 +17,7 @@
 #ifndef SWIFT_IRGEN_CALLEMISSION_H
 #define SWIFT_IRGEN_CALLEMISSION_H
 
+#include "Address.h"
 #include "Temporary.h"
 #include "Callee.h"
 
@@ -49,6 +50,8 @@ class CallEmission {
   /// Temporaries required by the call.
   TemporarySet Temporaries;
 
+  SmallVector<StackAddress, 8> RawTempraries;
+
   /// The function we're going to call.
   Callee CurCallee;