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 = OffloadBinary::write(Data); 47 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 } 68