mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-22 04:55:50 -04:00
045b8544fd
The first problem was a use-after-free in the tests (detected by asan bots). The temporary array created for the "create" call is guaranteed to live only until the end of the statement. The fix there is to store the test data in a local variable to ensure it has the right lifetime The second issue is broken BUILD_SHARED_LIBS build, which I fix by adding the appropriate BinaryFormat dependency to the Object unit tests. llvm-svn: 356655
252 lines
12 KiB
C++
252 lines
12 KiB
C++
//===- MinidumpTest.cpp - Tests for Minidump.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/Object/Minidump.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Testing/Support/Error.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::object;
|
|
using namespace minidump;
|
|
|
|
static Expected<std::unique_ptr<MinidumpFile>> create(ArrayRef<uint8_t> Data) {
|
|
return MinidumpFile::create(
|
|
MemoryBufferRef(toStringRef(Data), "Test buffer"));
|
|
}
|
|
|
|
TEST(MinidumpFile, BasicInterface) {
|
|
std::vector<uint8_t> Data{ // Header
|
|
'M', 'D', 'M', 'P', // Signature
|
|
0x93, 0xa7, 0, 0, // Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
// A very simple minidump file which contains just a single stream.
|
|
auto ExpectedFile = create(Data);
|
|
ASSERT_THAT_EXPECTED(ExpectedFile, Succeeded());
|
|
const MinidumpFile &File = **ExpectedFile;
|
|
const Header &H = File.header();
|
|
EXPECT_EQ(Header::MagicSignature, H.Signature);
|
|
EXPECT_EQ(Header::MagicVersion, H.Version);
|
|
EXPECT_EQ(1u, H.NumberOfStreams);
|
|
EXPECT_EQ(0x20u, H.StreamDirectoryRVA);
|
|
EXPECT_EQ(0x03020100u, H.Checksum);
|
|
EXPECT_EQ(0x07060504u, H.TimeDateStamp);
|
|
EXPECT_EQ(uint64_t(0x0504030201000908), H.Flags);
|
|
|
|
ASSERT_EQ(1u, File.streams().size());
|
|
const Directory &Stream0 = File.streams()[0];
|
|
EXPECT_EQ(StreamType::LinuxCPUInfo, Stream0.Type);
|
|
EXPECT_EQ(7u, Stream0.Location.DataSize);
|
|
EXPECT_EQ(0x2cu, Stream0.Location.RVA);
|
|
|
|
EXPECT_EQ("CPUINFO", toStringRef(File.getRawStream(Stream0)));
|
|
EXPECT_EQ("CPUINFO",
|
|
toStringRef(*File.getRawStream(StreamType::LinuxCPUInfo)));
|
|
|
|
EXPECT_THAT_EXPECTED(File.getSystemInfo(), Failed<BinaryError>());
|
|
}
|
|
|
|
// Use the input from the previous test, but corrupt it in various ways
|
|
TEST(MinidumpFile, create_ErrorCases) {
|
|
std::vector<uint8_t> FileTooShort{'M', 'D', 'M', 'P'};
|
|
EXPECT_THAT_EXPECTED(create(FileTooShort), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> WrongSignature{
|
|
// Header
|
|
'!', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(WrongSignature), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> WrongVersion{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x39, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(WrongVersion), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> DirectoryAfterEOF{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 1, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(DirectoryAfterEOF), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> TruncatedDirectory{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 1, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(TruncatedDirectory), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> Stream0AfterEOF{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 1, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(Stream0AfterEOF), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> Stream0Truncated{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 8, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(Stream0Truncated), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> DuplicateStream{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
2, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x40, 0, 0, 0, // RVA
|
|
// Stream
|
|
3, 0, 0x67, 0x47, 7, 0, 0, 0, // Type, DataSize,
|
|
0x40, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(DuplicateStream), Failed<BinaryError>());
|
|
|
|
std::vector<uint8_t> DenseMapInfoConflict{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
0xff, 0xff, 0xff, 0xff, 7, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// Stream
|
|
'C', 'P', 'U', 'I', 'N', 'F', 'O'};
|
|
EXPECT_THAT_EXPECTED(create(DenseMapInfoConflict), Failed<BinaryError>());
|
|
}
|
|
|
|
TEST(MinidumpFile, IngoresDummyStreams) {
|
|
std::vector<uint8_t> TwoDummyStreams{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
2, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
0, 0, 0, 0, 0, 0, 0, 0, // Type, DataSize,
|
|
0x20, 0, 0, 0, // RVA
|
|
0, 0, 0, 0, 0, 0, 0, 0, // Type, DataSize,
|
|
0x20, 0, 0, 0, // RVA
|
|
};
|
|
auto ExpectedFile = create(TwoDummyStreams);
|
|
ASSERT_THAT_EXPECTED(ExpectedFile, Succeeded());
|
|
const MinidumpFile &File = **ExpectedFile;
|
|
ASSERT_EQ(2u, File.streams().size());
|
|
EXPECT_EQ(StreamType::Unused, File.streams()[0].Type);
|
|
EXPECT_EQ(StreamType::Unused, File.streams()[1].Type);
|
|
EXPECT_EQ(None, File.getRawStream(StreamType::Unused));
|
|
}
|
|
|
|
TEST(MinidumpFile, getSystemInfo) {
|
|
std::vector<uint8_t> Data{
|
|
// Header
|
|
'M', 'D', 'M', 'P', 0x93, 0xa7, 0, 0, // Signature, Version
|
|
1, 0, 0, 0, // NumberOfStreams,
|
|
0x20, 0, 0, 0, // StreamDirectoryRVA
|
|
0, 1, 2, 3, 4, 5, 6, 7, // Checksum, TimeDateStamp
|
|
8, 9, 0, 1, 2, 3, 4, 5, // Flags
|
|
// Stream Directory
|
|
7, 0, 0, 0, 56, 0, 0, 0, // Type, DataSize,
|
|
0x2c, 0, 0, 0, // RVA
|
|
// SystemInfo
|
|
0, 0, 1, 2, // ProcessorArch, ProcessorLevel
|
|
3, 4, 5, 6, // ProcessorRevision, NumberOfProcessors, ProductType
|
|
7, 8, 9, 0, 1, 2, 3, 4, // MajorVersion, MinorVersion
|
|
5, 6, 7, 8, 2, 0, 0, 0, // BuildNumber, PlatformId
|
|
1, 2, 3, 4, 5, 6, 7, 8, // CSDVersionRVA, SuiteMask, Reserved
|
|
'L', 'L', 'V', 'M', 'L', 'L', 'V', 'M', 'L', 'L', 'V', 'M', // VendorID
|
|
1, 2, 3, 4, 5, 6, 7, 8, // VersionInfo, FeatureInfo
|
|
9, 0, 1, 2, // AMDExtendedFeatures
|
|
};
|
|
auto ExpectedFile = create(Data);
|
|
ASSERT_THAT_EXPECTED(ExpectedFile, Succeeded());
|
|
const MinidumpFile &File = **ExpectedFile;
|
|
|
|
auto ExpectedInfo = File.getSystemInfo();
|
|
ASSERT_THAT_EXPECTED(ExpectedInfo, Succeeded());
|
|
const SystemInfo &Info = *ExpectedInfo;
|
|
EXPECT_EQ(ProcessorArchitecture::X86, Info.ProcessorArch);
|
|
EXPECT_EQ(0x0201, Info.ProcessorLevel);
|
|
EXPECT_EQ(0x0403, Info.ProcessorRevision);
|
|
EXPECT_EQ(5, Info.NumberOfProcessors);
|
|
EXPECT_EQ(6, Info.ProductType);
|
|
EXPECT_EQ(0x00090807u, Info.MajorVersion);
|
|
EXPECT_EQ(0x04030201u, Info.MinorVersion);
|
|
EXPECT_EQ(0x08070605u, Info.BuildNumber);
|
|
EXPECT_EQ(OSPlatform::Win32NT, Info.PlatformId);
|
|
EXPECT_EQ(0x04030201u, Info.CSDVersionRVA);
|
|
EXPECT_EQ(0x0605u, Info.SuiteMask);
|
|
EXPECT_EQ(0x0807u, Info.Reserved);
|
|
EXPECT_EQ("LLVMLLVMLLVM", llvm::StringRef(Info.CPU.X86.VendorID,
|
|
sizeof(Info.CPU.X86.VendorID)));
|
|
EXPECT_EQ(0x04030201u, Info.CPU.X86.VersionInfo);
|
|
EXPECT_EQ(0x08070605u, Info.CPU.X86.FeatureInfo);
|
|
EXPECT_EQ(0x02010009u, Info.CPU.X86.AMDExtendedFeatures);
|
|
}
|