15babbaf5d
This pr implements the boiler plate required to use `llvm-objcopy` for `DXContainer` object files. It defines a minimal structure `object` to represent the `DXContainer` header and the following parts. This structure is a simple representation of the object data to allow for simple modifications at the granularity of each part. It follows similarily to how the respective `object`s are defined for `ELF`, `wasm`, `XCOFF`, etc. This is the first step to implement https://github.com/llvm/llvm-project/issues/150275 and https://github.com/llvm/llvm-project/issues/150277 as compiler actions that invoke `llvm-objcopy` for functionality.
39 lines
1.2 KiB
C++
39 lines
1.2 KiB
C++
//===- DXContainerReader.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 "DXContainerReader.h"
|
|
|
|
namespace llvm {
|
|
namespace objcopy {
|
|
namespace dxbc {
|
|
|
|
using namespace object;
|
|
|
|
Expected<std::unique_ptr<Object>> DXContainerReader::create() const {
|
|
auto Obj = std::make_unique<Object>();
|
|
Obj->Header = DXContainerObj.getHeader();
|
|
for (const SectionRef &Part : DXContainerObj.sections()) {
|
|
DataRefImpl PartDRI = Part.getRawDataRefImpl();
|
|
Expected<StringRef> Name = DXContainerObj.getSectionName(PartDRI);
|
|
if (auto E = Name.takeError())
|
|
return E;
|
|
assert(Name->size() == 4 &&
|
|
"Valid DXIL Part name consists of 4 characters");
|
|
Expected<ArrayRef<uint8_t>> Data =
|
|
DXContainerObj.getSectionContents(PartDRI);
|
|
if (auto E = Data.takeError())
|
|
return E;
|
|
Obj->Parts.push_back({*Name, *Data});
|
|
}
|
|
return std::move(Obj);
|
|
}
|
|
|
|
} // end namespace dxbc
|
|
} // end namespace objcopy
|
|
} // end namespace llvm
|