#ifndef MACHOWRITER_H
#define MACHOWRITER_H
-#include "MachO.h"
-#include "llvm/Constants.h"
-#include "llvm/DerivedTypes.h"
#include "llvm/CodeGen/MachineFunctionPass.h"
-#include "llvm/CodeGen/ObjectCodeEmitter.h"
-#include "llvm/Target/TargetData.h"
-#include "llvm/Target/TargetMachine.h"
-#include "llvm/Target/TargetMachOWriterInfo.h"
#include <vector>
#include <map>
namespace llvm {
+ class Constant;
class GlobalVariable;
class Mangler;
+ class MachineBasicBlock;
class MachineRelocation;
- class ObjectCodeEmitter;
class MachOCodeEmitter;
+ struct MachODySymTab;
+ struct MachOHeader;
+ struct MachOSection;
+ struct MachOSym;
class TargetData;
class TargetMachine;
+ class MCAsmInfo;
+ class ObjectCodeEmitter;
class OutputBuffer;
class raw_ostream;
-
/// MachOWriter - This class implements the common target-independent code for
/// writing Mach-O files. Targets should derive a class from this to
/// parameterize the output format.
/// Mang - The object used to perform name mangling for this module.
///
Mangler *Mang;
-
- /// MachOCE - The MachineCodeEmitter object that we are exposing to emit machine
- /// code for functions to the .o file.
+ /// MachOCE - The MachineCodeEmitter object that we are exposing to emit
+ /// machine code for functions to the .o file.
MachOCodeEmitter *MachOCE;
/// is64Bit/isLittleEndian - This information is inferred from the target
/// machine directly, indicating what header values and flags to set.
-
bool is64Bit, isLittleEndian;
// Target Asm Info
-
- const TargetAsmInfo *TAI;
+ const MCAsmInfo *MAI;
/// Header - An instance of MachOHeader that we will update while we build
/// the file, and then emit during finalization.
-
MachOHeader Header;
/// doInitialization - Emit the file header and all of the global variables
/// for the module to the Mach-O file.
-
bool doInitialization(Module &M);
bool runOnMachineFunction(MachineFunction &MF);
/// doFinalization - Now that the module has been completely processed, emit
/// the Mach-O file to 'O'.
-
bool doFinalization(Module &M);
private:
/// SectionList - This is the list of sections that we have emitted to the
/// file. Once the file has been completely built, the segment load command
/// SectionCommands are constructed from this info.
-
std::vector<MachOSection*> SectionList;
/// SectionLookup - This is a mapping from section name to SectionList entry
-
std::map<std::string, MachOSection*> SectionLookup;
-
+
/// GVSection - This is a mapping from a GlobalValue to a MachOSection,
/// to aid in emitting relocations.
-
std::map<GlobalValue*, MachOSection*> GVSection;
- /// GVOffset - This is a mapping from a GlobalValue to an offset from the
+ /// GVOffset - This is a mapping from a GlobalValue to an offset from the
/// start of the section in which the GV resides, to aid in emitting
/// relocations.
-
std::map<GlobalValue*, intptr_t> GVOffset;
/// getSection - Return the section with the specified name, creating a new
/// section if one does not already exist.
-
MachOSection *getSection(const std::string &seg, const std::string §,
- unsigned Flags = 0) {
- MachOSection *MOS = SectionLookup[seg+sect];
- if (MOS) return MOS;
-
- MOS = new MachOSection(seg, sect);
- SectionList.push_back(MOS);
- MOS->Index = SectionList.size();
- MOS->flags = MachOSection::S_REGULAR | Flags;
- SectionLookup[seg+sect] = MOS;
- return MOS;
- }
- MachOSection *getTextSection(bool isCode = true) {
- if (isCode)
- return getSection("__TEXT", "__text",
- MachOSection::S_ATTR_PURE_INSTRUCTIONS |
- MachOSection::S_ATTR_SOME_INSTRUCTIONS);
- else
- return getSection("__TEXT", "__text");
- }
- MachOSection *getBSSSection() {
- return getSection("__DATA", "__bss", MachOSection::S_ZEROFILL);
- }
+ unsigned Flags = 0);
+
+ /// getTextSection - Return text section with different flags for code/data
+ MachOSection *getTextSection(bool isCode = true);
+
MachOSection *getDataSection() {
return getSection("__DATA", "__data");
}
- MachOSection *getConstSection(Constant *C) {
- const ConstantArray *CVA = dyn_cast<ConstantArray>(C);
- if (CVA && CVA->isCString())
- return getSection("__TEXT", "__cstring",
- MachOSection::S_CSTRING_LITERALS);
-
- const Type *Ty = C->getType();
- if (Ty->isPrimitiveType() || Ty->isInteger()) {
- unsigned Size = TM.getTargetData()->getTypeAllocSize(Ty);
- switch(Size) {
- default: break; // Fall through to __TEXT,__const
- case 4:
- return getSection("__TEXT", "__literal4",
- MachOSection::S_4BYTE_LITERALS);
- case 8:
- return getSection("__TEXT", "__literal8",
- MachOSection::S_8BYTE_LITERALS);
- case 16:
- return getSection("__TEXT", "__literal16",
- MachOSection::S_16BYTE_LITERALS);
- }
- }
- return getSection("__TEXT", "__const");
- }
- MachOSection *getJumpTableSection() {
- if (TM.getRelocationModel() == Reloc::PIC_)
- return getTextSection(false);
- else
- return getSection("__TEXT", "__const");
- }
-
- /// MachOSymTab - This struct contains information about the offsets and
+
+ MachOSection *getBSSSection();
+ MachOSection *getConstSection(Constant *C);
+ MachOSection *getJumpTableSection();
+
+ /// MachOSymTab - This struct contains information about the offsets and
/// size of symbol table information.
/// segment.
struct MachOSymTab {
// see <mach-o/loader.h>
enum { LC_SYMTAB = 0x02 // link-edit stab symbol table info
};
-
+
MachOSymTab() : cmd(LC_SYMTAB), cmdsize(6 * sizeof(uint32_t)), symoff(0),
nsyms(0), stroff(0), strsize(0) { }
};
-
+
/// SymTab - The "stab" style symbol table information
- MachOSymTab SymTab;
+ MachOSymTab SymTab;
/// DySymTab - symbol table info for the dynamic link editor
MachODySymTab DySymTab;
protected:
-
+
/// SymbolTable - This is the list of symbols we have emitted to the file.
/// This actually gets rearranged before emission to the file (to put the
/// local symbols first in the list).
std::vector<MachOSym> SymbolTable;
-
+
/// SymT - A buffer to hold the symbol table before we write it out at the
/// appropriate location in the file.
- DataBuffer SymT;
-
+ std::vector<unsigned char> SymT;
+
/// StrT - A buffer to hold the string table before we write it out at the
/// appropriate location in the file.
- DataBuffer StrT;
-
+ std::vector<unsigned char> StrT;
+
/// PendingSyms - This is a list of externally defined symbols that we have
/// been asked to emit, but have not seen a reference to. When a reference
/// is seen, the symbol will move from this list to the SymbolTable.
std::vector<GlobalValue*> PendingGlobals;
-
+
/// DynamicSymbolTable - This is just a vector of indices into
/// SymbolTable to aid in emitting the DYSYMTAB load command.
std::vector<unsigned> DynamicSymbolTable;
-
- static void InitMem(const Constant *C,
- uintptr_t Offset,
- const TargetData *TD,
- MachOSection* mos);
+
+ static void InitMem(const Constant *C, uintptr_t Offset,
+ const TargetData *TD, MachOSection* mos);
private:
void AddSymbolToSection(MachOSection *MOS, GlobalVariable *GV);
void BufferSymbolAndStringTable();
void CalculateRelocations(MachOSection &MOS);
+ // GetJTRelocation - Get a relocation a new BB relocation based
+ // on target information.
MachineRelocation GetJTRelocation(unsigned Offset,
- MachineBasicBlock *MBB) const {
- return TM.getMachOWriterInfo()->GetJTRelocation(Offset, MBB);
- }
+ MachineBasicBlock *MBB) const;
/// GetTargetRelocation - Returns the number of relocations.
- unsigned GetTargetRelocation(MachineRelocation &MR,
- unsigned FromIdx,
- unsigned ToAddr,
- unsigned ToIndex,
- OutputBuffer &RelocOut,
- OutputBuffer &SecOut,
- bool Scattered,
- bool Extern) {
- return TM.getMachOWriterInfo()->GetTargetRelocation(MR, FromIdx, ToAddr,
- ToIndex, RelocOut,
- SecOut, Scattered,
- Extern);
- }
+ unsigned GetTargetRelocation(MachineRelocation &MR, unsigned FromIdx,
+ unsigned ToAddr, unsigned ToIndex,
+ OutputBuffer &RelocOut, OutputBuffer &SecOut,
+ bool Scattered, bool Extern);
};
}