e34182f396
This patch adds support for converting normalized mach-o to and from binary mach-o. It also changes WriterMachO (which previously directly wrote a mach-o binary given a set of Atoms) to instead do it in two steps. The first step uses normalizedFromAtoms() to convert Atoms to normalized mach-o, and the second step uses writeBinary() which to generate the mach-o binary file. llvm-svn: 194167
68 lines
1.9 KiB
C++
68 lines
1.9 KiB
C++
//===- lib/ReaderWriter/MachO/WriterMachO.cpp -----------------------------===//
|
|
//
|
|
// The LLVM Linker
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lld/ReaderWriter/Writer.h"
|
|
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/FileOutputBuffer.h"
|
|
#include "llvm/Support/MachO.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/system_error.h"
|
|
|
|
#include "lld/Core/File.h"
|
|
#include "lld/ReaderWriter/MachOLinkingContext.h"
|
|
|
|
|
|
#include "MachONormalizedFile.h"
|
|
#include "ExecutableAtoms.hpp"
|
|
|
|
using lld::mach_o::normalized::NormalizedFile;
|
|
|
|
namespace lld {
|
|
namespace mach_o {
|
|
|
|
class MachOWriter : public Writer {
|
|
public:
|
|
MachOWriter(const MachOLinkingContext &ctxt) : _context(ctxt) { }
|
|
|
|
virtual error_code writeFile(const lld::File &file, StringRef path) {
|
|
// Construct empty normalized file from atoms.
|
|
ErrorOr<std::unique_ptr<NormalizedFile>> nFile =
|
|
normalized::normalizedFromAtoms(file, _context);
|
|
if (!nFile)
|
|
return nFile;
|
|
|
|
// For debugging, write out yaml form of normalized file.
|
|
//writeYaml(*nFile->get(), llvm::errs());
|
|
|
|
// Write normalized file as mach-o binary.
|
|
return writeBinary(*nFile->get(), path);
|
|
}
|
|
|
|
virtual bool createImplicitFiles(std::vector<std::unique_ptr<File> > &r) {
|
|
if (_context.outputFileType() == llvm::MachO::MH_EXECUTE) {
|
|
// When building main executables, add _main as required entry point.
|
|
r.emplace_back(new CRuntimeFile(_context));
|
|
}
|
|
return true;
|
|
}
|
|
private:
|
|
const MachOLinkingContext &_context;
|
|
};
|
|
|
|
|
|
} // namespace mach_o
|
|
|
|
std::unique_ptr<Writer> createWriterMachO(const MachOLinkingContext &context) {
|
|
return std::unique_ptr<Writer>(new lld::mach_o::MachOWriter(context));
|
|
}
|
|
|
|
} // namespace lld
|