mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2025-02-01 05:01:59 +01:00
58dc349826
This patch adds basic support for MachO::load_command. Load command types and sizes are encoded in the YAML and expanded back into MachO. The YAML doesn't yet support load command structs, that is coming next. In the meantime as a temporary measure when writing MachO files the load commands are padded with zeros so that the generated binary is valid. llvm-svn: 269442
115 lines
3.2 KiB
C++
115 lines
3.2 KiB
C++
//===- yaml2macho - Convert YAML to a Mach object file --------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// \brief The Mach component of yaml2obj.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "yaml2obj.h"
|
|
#include "llvm/ObjectYAML/MachOYAML.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/MachO.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
class MachOWriter {
|
|
public:
|
|
MachOWriter(MachOYAML::Object &Obj) : Obj(Obj) {
|
|
is64Bit = Obj.Header.magic == MachO::MH_MAGIC_64 ||
|
|
Obj.Header.magic == MachO::MH_CIGAM_64;
|
|
memset(reinterpret_cast<void *>(&Header64), 0,
|
|
sizeof(MachO::mach_header_64));
|
|
assert((is64Bit || Obj.Header.reserved == 0xDEADBEEFu) &&
|
|
"32-bit MachO has reserved in header");
|
|
assert((!is64Bit || Obj.Header.reserved != 0xDEADBEEFu) &&
|
|
"64-bit MachO has missing reserved in header");
|
|
}
|
|
|
|
Error writeMachO(raw_ostream &OS);
|
|
|
|
private:
|
|
Error writeHeader(raw_ostream &OS);
|
|
Error writeLoadCommands(raw_ostream &OS);
|
|
|
|
MachOYAML::Object &Obj;
|
|
bool is64Bit;
|
|
|
|
union {
|
|
MachO::mach_header_64 Header64;
|
|
MachO::mach_header Header;
|
|
};
|
|
};
|
|
|
|
Error MachOWriter::writeMachO(raw_ostream &OS) {
|
|
if (auto Err = writeHeader(OS))
|
|
return Err;
|
|
if (auto Err = writeLoadCommands(OS))
|
|
return Err;
|
|
return Error::success();
|
|
}
|
|
|
|
Error MachOWriter::writeHeader(raw_ostream &OS) {
|
|
Header.magic = Obj.Header.magic;
|
|
Header.cputype = Obj.Header.cputype;
|
|
Header.cpusubtype = Obj.Header.cpusubtype;
|
|
Header.filetype = Obj.Header.filetype;
|
|
Header.ncmds = Obj.Header.ncmds;
|
|
Header.sizeofcmds = Obj.Header.sizeofcmds;
|
|
Header.flags = Obj.Header.flags;
|
|
Header64.reserved = Obj.Header.reserved;
|
|
|
|
if (is64Bit)
|
|
OS.write((const char *)&Header64, sizeof(MachO::mach_header_64));
|
|
else
|
|
OS.write((const char *)&Header, sizeof(MachO::mach_header));
|
|
|
|
return Error::success();
|
|
}
|
|
|
|
Error MachOWriter::writeLoadCommands(raw_ostream &OS) {
|
|
for (auto &LC : Obj.LoadCommands) {
|
|
MachO::load_command LCTemp;
|
|
LCTemp.cmd = LC->cmd;
|
|
LCTemp.cmdsize = LC->cmdsize;
|
|
OS.write(reinterpret_cast<const char *>(&LCTemp),
|
|
sizeof(MachO::load_command));
|
|
auto remaining_size = LC->cmdsize - sizeof(MachO::load_command);
|
|
if (remaining_size > 0) {
|
|
// TODO: Replace all this once the load command data is present in yaml.
|
|
std::vector<char> fill_data;
|
|
fill_data.insert(fill_data.begin(), remaining_size, 0);
|
|
OS.write(fill_data.data(), remaining_size);
|
|
}
|
|
}
|
|
return Error::success();
|
|
}
|
|
|
|
} // end anonymous namespace
|
|
|
|
int yaml2macho(yaml::Input &YIn, raw_ostream &Out) {
|
|
MachOYAML::Object Doc;
|
|
YIn >> Doc;
|
|
if (YIn.error()) {
|
|
errs() << "yaml2obj: Failed to parse YAML file!\n";
|
|
return 1;
|
|
}
|
|
|
|
MachOWriter Writer(Doc);
|
|
if (auto Err = Writer.writeMachO(Out)) {
|
|
errs() << toString(std::move(Err));
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|