llvm/unittests/Object/OffloadingTest.cpp

   1 #include "llvm/Object/OffloadBinary.h"
   2
   3 #include "llvm/Testing/Support/Error.h"
   4 #include "gtest/gtest.h"
   5 #include <random>
   6
   7 using namespace llvm;
   8 using namespace llvm::object;
   9
  10 TEST(OffloadingTest, checkOffloadingBinary) {
  11   // Create random data to fill the image.
  12   std::mt19937 Rng(std::random_device{}());
  13   std::uniform_int_distribution<uint64_t> SizeDist(0, 256);
  14   std::uniform_int_distribution<uint16_t> KindDist(0);
  15   std::uniform_int_distribution<uint16_t> BinaryDist(
  16       std::numeric_limits<uint8_t>::min(), std::numeric_limits<uint8_t>::max());
  17   std::uniform_int_distribution<int16_t> StringDist('!', '~');
  18   std::vector<uint8_t> Image(SizeDist(Rng));
  19   std::generate(Image.begin(), Image.end(), [&]() { return BinaryDist(Rng); });
  20   std::vector<std::pair<std::string, std::string>> Strings(SizeDist(Rng));
  21   for (auto &KeyAndValue : Strings) {
  22     std::string Key(SizeDist(Rng), '\0');
  23     std::string Value(SizeDist(Rng), '\0');
  24
  25     std::generate(Key.begin(), Key.end(), [&]() { return StringDist(Rng); });
  26     std::generate(Value.begin(), Value.end(),
  27                   [&]() { return StringDist(Rng); });
  28
  29     KeyAndValue = std::make_pair(Key, Value);
  30   }
  31
  32   // Create the image.
  33   MapVector<StringRef, StringRef> StringData;
  34   for (auto &KeyAndValue : Strings)
  35     StringData[KeyAndValue.first] = KeyAndValue.second;
  36   std::unique_ptr<MemoryBuffer> ImageData = MemoryBuffer::getMemBuffer(
  37       {reinterpret_cast<char *>(Image.data()), Image.size()}, "", false);
  38
  39   OffloadBinary::OffloadingImage Data;
  40   Data.TheImageKind = static_cast<ImageKind>(KindDist(Rng));
  41   Data.TheOffloadKind = static_cast<OffloadKind>(KindDist(Rng));
  42   Data.Flags = KindDist(Rng);
  43   Data.StringData = StringData;
  44   Data.Image = std::move(ImageData);
  45
  46   auto BinaryBuffer =
  47       MemoryBuffer::getMemBufferCopy(OffloadBinary::write(Data));
  48   auto BinaryOrErr = OffloadBinary::create(*BinaryBuffer);
  49   if (!BinaryOrErr)
  50     FAIL();
  51
  52   // Make sure we get the same data out.
  53   auto &Binary = **BinaryOrErr;
  54   ASSERT_EQ(Data.TheImageKind, Binary.getImageKind());
  55   ASSERT_EQ(Data.TheOffloadKind, Binary.getOffloadKind());
  56   ASSERT_EQ(Data.Flags, Binary.getFlags());
  57
  58   for (auto &KeyAndValue : Strings)
  59     ASSERT_TRUE(StringData[KeyAndValue.first] ==
  60                 Binary.getString(KeyAndValue.first));
  61
  62   EXPECT_TRUE(Data.Image->getBuffer() == Binary.getImage());
  63
  64   // Ensure the size and alignment of the data is correct.
  65   EXPECT_TRUE(Binary.getSize() % OffloadBinary::getAlignment() == 0);
  66   EXPECT_TRUE(Binary.getSize() == BinaryBuffer->getBuffer().size());
  67 }