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.
59 lines
1.7 KiB
C++
59 lines
1.7 KiB
C++
//===- DXContainerWriter.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 "DXContainerWriter.h"
|
|
|
|
namespace llvm {
|
|
namespace objcopy {
|
|
namespace dxbc {
|
|
|
|
using namespace object;
|
|
|
|
size_t DXContainerWriter::finalize() {
|
|
assert(Offsets.empty() &&
|
|
"Attempted to finalize writer with already computed offsets");
|
|
Offsets.reserve(Obj.Parts.size());
|
|
size_t Offset = Obj.headerSize();
|
|
for (const Part &P : Obj.Parts) {
|
|
Offsets.push_back(Offset);
|
|
Offset += P.size();
|
|
}
|
|
return Obj.Header.FileSize;
|
|
}
|
|
|
|
Error DXContainerWriter::write() {
|
|
size_t TotalSize = finalize();
|
|
Out.reserveExtraSpace(TotalSize);
|
|
|
|
llvm::dxbc::Header Header = Obj.Header;
|
|
if (sys::IsBigEndianHost)
|
|
Header.swapBytes();
|
|
Out.write(reinterpret_cast<const char *>(&Header),
|
|
sizeof(::llvm::dxbc::Header));
|
|
if (sys::IsBigEndianHost)
|
|
for (auto &O : Offsets)
|
|
sys::swapByteOrder(O);
|
|
Out.write(reinterpret_cast<const char *>(Offsets.data()),
|
|
Offsets.size() * sizeof(uint32_t));
|
|
|
|
for (const Part &P : Obj.Parts) {
|
|
Out.write(reinterpret_cast<const char *>(P.Name.data()), 4);
|
|
uint32_t Size = P.Data.size();
|
|
if (sys::IsBigEndianHost)
|
|
sys::swapByteOrder(Size);
|
|
Out.write(reinterpret_cast<const char *>(&Size), sizeof(uint32_t));
|
|
Out.write(reinterpret_cast<const char *>(P.Data.data()), P.Data.size());
|
|
}
|
|
|
|
return Error::success();
|
|
}
|
|
|
|
} // end namespace dxbc
|
|
} // end namespace objcopy
|
|
} // end namespace llvm
|