129 lines
4.7 KiB
C++
129 lines
4.7 KiB
C++
//===- DXContainerTest.cpp - Tests for DXContainerFile --------------------===//
|
|
//
|
|
// 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/Object/DXContainer.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/BinaryFormat/Magic.h"
|
|
#include "llvm/Support/MemoryBufferRef.h"
|
|
#include "llvm/Testing/Support/Error.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::object;
|
|
|
|
template <std::size_t X> MemoryBufferRef getMemoryBuffer(uint8_t Data[X]) {
|
|
StringRef Obj(reinterpret_cast<char *>(&Data[0]), X);
|
|
return MemoryBufferRef(Obj, "");
|
|
}
|
|
|
|
TEST(DXCFile, IdentifyMagic) {
|
|
{
|
|
StringRef Buffer("DXBC");
|
|
EXPECT_EQ(identify_magic(Buffer), file_magic::dxcontainer_object);
|
|
}
|
|
{
|
|
StringRef Buffer("DXBCBlahBlahBlah");
|
|
EXPECT_EQ(identify_magic(Buffer), file_magic::dxcontainer_object);
|
|
}
|
|
}
|
|
|
|
TEST(DXCFile, ParseHeaderErrors) {
|
|
uint8_t Buffer[] = {0x44, 0x58, 0x42, 0x43};
|
|
EXPECT_THAT_EXPECTED(
|
|
DXContainer::create(getMemoryBuffer<4>(Buffer)),
|
|
FailedWithMessage("Reading structure out of file bounds"));
|
|
}
|
|
|
|
TEST(DXCFile, EmptyFile) {
|
|
EXPECT_THAT_EXPECTED(
|
|
DXContainer::create(MemoryBufferRef(StringRef("", 0), "")),
|
|
FailedWithMessage("Reading structure out of file bounds"));
|
|
}
|
|
|
|
TEST(DXCFile, ParseHeader) {
|
|
uint8_t Buffer[] = {0x44, 0x58, 0x42, 0x43, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
|
|
0x70, 0x0D, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
|
|
DXContainer C =
|
|
llvm::cantFail(DXContainer::create(getMemoryBuffer<32>(Buffer)));
|
|
EXPECT_TRUE(memcmp(C.getHeader().Magic, "DXBC", 4) == 0);
|
|
EXPECT_TRUE(memcmp(C.getHeader().FileHash.Digest,
|
|
"\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", 16) == 0);
|
|
EXPECT_EQ(C.getHeader().Version.Major, 1u);
|
|
EXPECT_EQ(C.getHeader().Version.Minor, 0u);
|
|
}
|
|
|
|
TEST(DXCFile, ParsePartMissingOffsets) {
|
|
uint8_t Buffer[] = {
|
|
0x44, 0x58, 0x42, 0x43, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00,
|
|
0x00, 0x00, 0x70, 0x0D, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
|
|
};
|
|
EXPECT_THAT_EXPECTED(
|
|
DXContainer::create(getMemoryBuffer<32>(Buffer)),
|
|
FailedWithMessage("Reading structure out of file bounds"));
|
|
}
|
|
|
|
TEST(DXCFile, ParsePartInvalidOffsets) {
|
|
uint8_t Buffer[] = {
|
|
0x44, 0x58, 0x42, 0x43, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
|
|
0x70, 0x0D, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0xFF, 0xFF,
|
|
};
|
|
EXPECT_THAT_EXPECTED(
|
|
DXContainer::create(getMemoryBuffer<36>(Buffer)),
|
|
FailedWithMessage("Part offset points beyond boundary of the file"));
|
|
}
|
|
|
|
TEST(DXCFile, ParseEmptyParts) {
|
|
uint8_t Buffer[] = {
|
|
0x44, 0x58, 0x42, 0x43, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
|
|
0x70, 0x0D, 0x00, 0x00, 0x07, 0x00, 0x00, 0x00, 0x3C, 0x00, 0x00, 0x00,
|
|
0x44, 0x00, 0x00, 0x00, 0x4C, 0x00, 0x00, 0x00, 0x54, 0x00, 0x00, 0x00,
|
|
0x5C, 0x00, 0x00, 0x00, 0x64, 0x00, 0x00, 0x00, 0x6C, 0x00, 0x00, 0x00,
|
|
0x53, 0x46, 0x49, 0x30, 0x00, 0x00, 0x00, 0x00, 0x49, 0x53, 0x47, 0x31,
|
|
0x00, 0x00, 0x00, 0x00, 0x4F, 0x53, 0x47, 0x31, 0x00, 0x00, 0x00, 0x00,
|
|
0x50, 0x53, 0x56, 0x30, 0x00, 0x00, 0x00, 0x00, 0x53, 0x54, 0x41, 0x54,
|
|
0x00, 0x00, 0x00, 0x00, 0x43, 0x58, 0x49, 0x4C, 0x00, 0x00, 0x00, 0x00,
|
|
0x44, 0x45, 0x41, 0x44, 0x00, 0x00, 0x00, 0x00,
|
|
};
|
|
DXContainer C =
|
|
llvm::cantFail(DXContainer::create(getMemoryBuffer<116>(Buffer)));
|
|
EXPECT_EQ(C.getHeader().PartCount, 7u);
|
|
|
|
// All the part sizes are 0, which makes a nice test of the range based for
|
|
int ElementsVisited = 0;
|
|
for (auto Part : C) {
|
|
EXPECT_EQ(Part.Part.Size, 0u);
|
|
EXPECT_EQ(Part.Data.size(), 0u);
|
|
++ElementsVisited;
|
|
}
|
|
EXPECT_EQ(ElementsVisited, 7);
|
|
|
|
{
|
|
auto It = C.begin();
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "SFI0", 4) == 0);
|
|
++It;
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "ISG1", 4) == 0);
|
|
++It;
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "OSG1", 4) == 0);
|
|
++It;
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "PSV0", 4) == 0);
|
|
++It;
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "STAT", 4) == 0);
|
|
++It;
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "CXIL", 4) == 0);
|
|
++It;
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "DEAD", 4) == 0);
|
|
++It; // Don't increment past the end
|
|
EXPECT_TRUE(memcmp(It->Part.Name, "DEAD", 4) == 0);
|
|
}
|
|
}
|