1 files changed, 118 insertions, 0 deletions
diff --git a/llvm/unittests/BinaryFormat/SFrameTest.cpp b/llvm/unittests/BinaryFormat/SFrameTest.cpp
new file mode 100644
index 000000000000..394e382e041e
--- /dev/null
+++ b/llvm/unittests/BinaryFormat/SFrameTest.cpp
@@ -0,0 +1,118 @@
+//===- SFrameTest.cpp -----------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/BinaryFormat/SFrame.h"
+#include "gtest/gtest.h"
+#include <type_traits>
+
+using namespace llvm;
+using namespace llvm::sframe;
+
+namespace {
+
+template <typename EndianT> class SFrameTest : public testing::Test {
+protected:
+  static constexpr endianness Endian = EndianT::value;
+
+  // Test structure sizes and triviality.
+  static_assert(std::is_trivial_v<Preamble<Endian>>);
+  static_assert(sizeof(Preamble<Endian>) == 4);
+
+  static_assert(std::is_trivial_v<Header<Endian>>);
+  static_assert(sizeof(Header<Endian>) == 28);
+
+  static_assert(std::is_trivial_v<FuncDescEntry<Endian>>);
+  static_assert(sizeof(FuncDescEntry<Endian>) == 20);
+
+  static_assert(std::is_trivial_v<FrameRowEntryAddr1<Endian>>);
+  static_assert(sizeof(FrameRowEntryAddr1<Endian>) == 2);
+
+  static_assert(std::is_trivial_v<FrameRowEntryAddr2<Endian>>);
+  static_assert(sizeof(FrameRowEntryAddr2<Endian>) == 3);
+
+  static_assert(std::is_trivial_v<FrameRowEntryAddr4<Endian>>);
+  static_assert(sizeof(FrameRowEntryAddr4<Endian>) == 5);
+};
+
+struct NameGenerator {
+  template <typename T> static constexpr const char *GetName(int) {
+    if constexpr (T::value == endianness::little)
+      return "little";
+    if constexpr (T::value == endianness::big)
+      return "big";
+  }
+};
+using Types =
+    testing::Types<std::integral_constant<endianness, endianness::little>,
+                   std::integral_constant<endianness, endianness::big>>;
+TYPED_TEST_SUITE(SFrameTest, Types, NameGenerator);
+
+TYPED_TEST(SFrameTest, FDEFlags) {
+  FuncDescEntry<TestFixture::Endian> FDE = {};
+  EXPECT_EQ(FDE.Info, 0u);
+  EXPECT_EQ(FDE.getPAuthKey(), 0);
+  EXPECT_EQ(FDE.getFDEType(), FDEType::PCInc);
+  EXPECT_EQ(FDE.getFREType(), FREType::Addr1);
+
+  FDE.setPAuthKey(1);
+  EXPECT_EQ(FDE.Info, 0x20u);
+  EXPECT_EQ(FDE.getPAuthKey(), 1);
+  EXPECT_EQ(FDE.getFDEType(), FDEType::PCInc);
+  EXPECT_EQ(FDE.getFREType(), FREType::Addr1);
+
+  FDE.setFDEType(FDEType::PCMask);
+  EXPECT_EQ(FDE.Info, 0x30u);
+  EXPECT_EQ(FDE.getPAuthKey(), 1);
+  EXPECT_EQ(FDE.getFDEType(), FDEType::PCMask);
+  EXPECT_EQ(FDE.getFREType(), FREType::Addr1);
+
+  FDE.setFREType(FREType::Addr4);
+  EXPECT_EQ(FDE.Info, 0x32u);
+  EXPECT_EQ(FDE.getPAuthKey(), 1);
+  EXPECT_EQ(FDE.getFDEType(), FDEType::PCMask);
+  EXPECT_EQ(FDE.getFREType(), FREType::Addr4);
+}
+
+TYPED_TEST(SFrameTest, FREFlags) {
+  FREInfo<TestFixture::Endian> Info = {};
+  EXPECT_EQ(Info.Info, 0u);
+  EXPECT_FALSE(Info.isReturnAddressSigned());
+  EXPECT_EQ(Info.getOffsetSize(), FREOffset::B1);
+  EXPECT_EQ(Info.getOffsetCount(), 0u);
+  EXPECT_EQ(Info.getBaseRegister(), BaseReg::FP);
+
+  Info.setReturnAddressSigned(true);
+  EXPECT_EQ(Info.Info, 0x80u);
+  EXPECT_TRUE(Info.isReturnAddressSigned());
+  EXPECT_EQ(Info.getOffsetSize(), FREOffset::B1);
+  EXPECT_EQ(Info.getOffsetCount(), 0u);
+  EXPECT_EQ(Info.getBaseRegister(), BaseReg::FP);
+
+  Info.setOffsetSize(FREOffset::B4);
+  EXPECT_EQ(Info.Info, 0xc0u);
+  EXPECT_TRUE(Info.isReturnAddressSigned());
+  EXPECT_EQ(Info.getOffsetSize(), FREOffset::B4);
+  EXPECT_EQ(Info.getOffsetCount(), 0u);
+  EXPECT_EQ(Info.getBaseRegister(), BaseReg::FP);
+
+  Info.setOffsetCount(3);
+  EXPECT_EQ(Info.Info, 0xc6u);
+  EXPECT_TRUE(Info.isReturnAddressSigned());
+  EXPECT_EQ(Info.getOffsetSize(), FREOffset::B4);
+  EXPECT_EQ(Info.getOffsetCount(), 3u);
+  EXPECT_EQ(Info.getBaseRegister(), BaseReg::FP);
+
+  Info.setBaseRegister(BaseReg::SP);
+  EXPECT_EQ(Info.Info, 0xc7u);
+  EXPECT_TRUE(Info.isReturnAddressSigned());
+  EXPECT_EQ(Info.getOffsetSize(), FREOffset::B4);
+  EXPECT_EQ(Info.getOffsetCount(), 3u);
+  EXPECT_EQ(Info.getBaseRegister(), BaseReg::SP);
+}
+
+} // namespace