IRGen: Use C compatible representation for @cdecl enums

swiftlang#83738

Author: xymus

lib/IRGen/GenEnum.cpp

@@ -6536,8 +6536,9 @@ EnumImplStrategy::get(TypeConverter &TC, SILType type, EnumDecl *theEnum) {
         std::move(elementsWithPayload), std::move(elementsWithNoPayload)));
   }
 
-  // Enums imported from Clang or marked with @objc use C-compatible layout.
-  if (theEnum->hasClangNode() || theEnum->isObjC()) {
+  // Enums imported from Clang or marked with @objc or @cdecl use a
+  // C-compatible layout.
+  if (theEnum->hasClangNode() || theEnum->isCCompatibleEnum()) {
     assert(elementsWithPayload.empty() && "C enum with payload?!");
     assert(alwaysFixedSize == IsFixedSize && "C enum with resilient payload?!");
     return std::unique_ptr<EnumImplStrategy>(
@@ -6982,7 +6983,7 @@ CCompatibleEnumImplStrategy::completeEnumTypeLayout(TypeConverter &TC,
                                                     llvm::StructType *enumTy){
   // The type should have come from Clang or be @objc,
   // and should have a raw type.
-  assert((theEnum->hasClangNode() || theEnum->isObjC())
+  assert((theEnum->hasClangNode() || theEnum->isCCompatibleEnum())
          && "c-compatible enum didn't come from clang!");
   assert(theEnum->hasRawType()
          && "c-compatible enum doesn't have raw type!");

test/Interpreter/cdecl_enum_run.swift

@@ -0,0 +1,45 @@
+// RUN: %target-run-simple-swift(-enable-experimental-feature CDecl) > %t.out
+// RUN: %FileCheck --input-file %t.out %s
+
+// REQUIRES: swift_feature_CDecl
+// REQUIRES: executable_test
+
+@cdecl
+enum CDecl8: UInt8 {
+case a
+case b
+}
+
+@cdecl
+enum CDecl16: UInt16 {
+case a
+case b
+}
+
+@cdecl(SomeName)
+enum CDecl32: UInt32 {
+case a
+case b
+}
+
+@objc
+enum ObjCEnum: UInt32 {
+case a
+case b
+}
+
+enum SwiftEnum: Int32 {
+case a
+case b
+}
+
+print("@cdecl enum 8 is \(MemoryLayout<CDecl8>.size) bytes")
+// CHECK: @cdecl enum 8 is 1 bytes
+print("@cdecl enum 16 is \(MemoryLayout<CDecl16>.size) bytes")
+// CHECK: @cdecl enum 16 is 2 bytes
+print("@cdecl enum 32 is \(MemoryLayout<CDecl32>.size) bytes")
+// CHECK: @cdecl enum 32 is 4 bytes
+print("@objc enum is \(MemoryLayout<ObjCEnum>.size) bytes")
+// CHECK: @objc enum is 4 bytes
+print("Swift enum is \(MemoryLayout<SwiftEnum>.size) bytes")
+// CHECK: Swift enum is 1 bytes

test/Interpreter/cdecl_official_run.swift

@@ -39,6 +39,7 @@
 
 @cdecl func useEnum(e: CEnum) -> CEnum {
     print(e)
+    print(e.rawValue)
     return e
 }
 
@@ -59,8 +60,15 @@ int main() {
     primitiveTypes(1, 2, 3, 'a', 1.0f, 2.0, true);
     // CHECK-NEXT: 1 2 3 97 1.0 2.0 true
 
-    CEnum e = useEnum(CEnumB);
-    // CHECK-NEXT: B
-    printf("%d\n", e);
+    CEnum a = useEnum(CEnumA);
+    // CHECK-NEXT: CEnum
+    // CHECK-NEXT: 0
+    printf("%d\n", a);
+    // CHECK-NEXT: 0
+
+    CEnum b = useEnum(CEnumB);
+    // CHECK-NEXT: CEnum
+    // CHECK-NEXT: 1
+    printf("%d\n", b);
     // CHECK-NEXT: 1
 }