ELFObjectWriter.cpp revision 0dea6bc96bb52346737966839ac68644f7939f58
15460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===- ELFObjectWriter.cpp ------------------------------------------------===//
25460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//
35460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//                     The MCLinker Project
45460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//
55460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// This file is distributed under the University of Illinois Open Source
65460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// License. See LICENSE.TXT for details.
75460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//
85460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===----------------------------------------------------------------------===//
922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <mcld/LD/ELFObjectWriter.h>
105460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao
1122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <mcld/Module.h>
12d0fbbb227051be16931a1aa9b4a7722ac039c698Shih-wei Liao#include <mcld/LinkerConfig.h>
1387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines#include <mcld/LinkerScript.h>
1422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <mcld/Target/GNULDBackend.h>
156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/Support/MsgHandling.h>
166f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/ADT/SizeTraits.h>
176f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/Fragment/AlignFragment.h>
186f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/Fragment/FillFragment.h>
196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/Fragment/RegionFragment.h>
206f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/Fragment/Stub.h>
216f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/Fragment/NullFragment.h>
2287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines#include <mcld/LD/LDSymbol.h>
236f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/LD/LDSection.h>
246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/LD/SectionData.h>
256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/LD/ELFSegment.h>
266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/LD/ELFSegmentFactory.h>
276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/LD/RelocData.h>
286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <mcld/LD/EhFrame.h>
2987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines#include <mcld/LD/ELFFileFormat.h>
3087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines#include <mcld/Target/GNUInfo.h>
3122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
320dea6bc96bb52346737966839ac68644f7939f58Stephen Hines#include <llvm/Support/Errc.h>
336f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <llvm/Support/ErrorHandling.h>
346f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <llvm/Support/ELF.h>
356f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines#include <llvm/Support/Casting.h>
36d0fbbb227051be16931a1aa9b4a7722ac039c698Shih-wei Liao
3722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liaousing namespace llvm;
386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesusing namespace llvm::ELF;
395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaousing namespace mcld;
405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao
4122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao//===----------------------------------------------------------------------===//
425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// ELFObjectWriter
4322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao//===----------------------------------------------------------------------===//
4422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei LiaoELFObjectWriter::ELFObjectWriter(GNULDBackend& pBackend,
45d0fbbb227051be16931a1aa9b4a7722ac039c698Shih-wei Liao                                 const LinkerConfig& pConfig)
466f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  : ObjectWriter(), m_Backend(pBackend), m_Config(pConfig)
476f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}
495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao
505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoELFObjectWriter::~ELFObjectWriter()
515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{
525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}
535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao
5487f34658dec9097d987d254a990ea7f311bfc95fStephen Hinesvoid ELFObjectWriter::writeSection(Module& pModule,
5587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                   FileOutputBuffer& pOutput, LDSection *section)
566f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
5787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  MemoryRegion region;
586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // Request output region
596f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  switch (section->kind()) {
606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Note:
616f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (section->getSectionData() == NULL)
626f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      return;
636f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Fall through
640dea6bc96bb52346737966839ac68644f7939f58Stephen Hines  case LDFileFormat::TEXT:
650dea6bc96bb52346737966839ac68644f7939f58Stephen Hines  case LDFileFormat::DATA:
666f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Relocation:
676f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Target:
686f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Debug:
696f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::GCCExceptTable:
706f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::EhFrame: {
716f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    region = pOutput.request(section->offset(), section->size());
7287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    if (region.size() == 0) {
7387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      return;
746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    break;
766f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
776f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Null:
786f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::NamePool:
796f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::BSS:
806f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::MetaData:
816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Version:
826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::EhFrameHdr:
836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::StackNote:
846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Ignore these sections
856f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return;
866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  default:
876f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    llvm::errs() << "WARNING: unsupported section kind: "
886f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                 << section->kind()
896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                 << " of section "
906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                 << section->name()
916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                 << ".\n";
926f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return;
936f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
946f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
956f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // Write out sections with data
966f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  switch(section->kind()) {
976f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::GCCExceptTable:
980dea6bc96bb52346737966839ac68644f7939f58Stephen Hines  case LDFileFormat::TEXT:
990dea6bc96bb52346737966839ac68644f7939f58Stephen Hines  case LDFileFormat::DATA:
1006f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Debug:
1016f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Note:
10287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    emitSectionData(*section, region);
10387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    break;
10487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  case LDFileFormat::EhFrame:
10587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    emitEhFrame(pModule, *section->getEhFrame(), region);
1066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    break;
1076f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Relocation:
10887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    // sort relocation for the benefit of the dynamic linker.
10987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    target().sortRelocation(*section);
11087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines
11187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    emitRelocation(m_Config, *section, region);
1126f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    break;
1136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  case LDFileFormat::Target:
11487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    target().emitSectionData(*section, region);
1156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    break;
1166f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  default:
1176f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    llvm_unreachable("invalid section kind");
1186f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
1196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
1206f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1210dea6bc96bb52346737966839ac68644f7939f58Stephen Hinesstd::error_code ELFObjectWriter::writeObject(Module& pModule,
1220dea6bc96bb52346737966839ac68644f7939f58Stephen Hines                                             FileOutputBuffer& pOutput)
12322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao{
1246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  bool is_dynobj = m_Config.codeGenType() == LinkerConfig::DynObj;
1256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  bool is_exec = m_Config.codeGenType() == LinkerConfig::Exec;
1266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  bool is_binary = m_Config.codeGenType() == LinkerConfig::Binary;
1276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  bool is_object = m_Config.codeGenType() == LinkerConfig::Object;
1286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  assert(is_dynobj || is_exec || is_binary || is_object);
1306f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1316f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (is_dynobj || is_exec) {
132f7ac0f19a1c8d0ad14bcf6456ce368b830fea886Stephen Hines    // Allow backend to sort symbols before emitting
133f7ac0f19a1c8d0ad14bcf6456ce368b830fea886Stephen Hines    target().orderSymbolTable(pModule);
134f7ac0f19a1c8d0ad14bcf6456ce368b830fea886Stephen Hines
1356f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Write out the interpreter section: .interp
1366f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    target().emitInterp(pOutput);
1376f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Write out name pool sections: .dynsym, .dynstr, .hash
1396f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    target().emitDynNamePools(pModule, pOutput);
1406f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
1416f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1426f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (is_object || is_dynobj || is_exec) {
1436f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Write out name pool sections: .symtab, .strtab
1446f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    target().emitRegNamePools(pModule, pOutput);
1456f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
1466f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1476f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (is_binary) {
1486f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Iterate over the loadable segments and write the corresponding sections
1496f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    ELFSegmentFactory::iterator seg, segEnd = target().elfSegmentTable().end();
1506f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1516f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    for (seg = target().elfSegmentTable().begin(); seg != segEnd; ++seg) {
15287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      if (llvm::ELF::PT_LOAD == (*seg)->type()) {
15387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        ELFSegment::iterator sect, sectEnd = (*seg)->end();
15487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        for (sect = (*seg)->begin(); sect != sectEnd; ++sect)
15587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines          writeSection(pModule, pOutput, *sect);
15622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao      }
15722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao    }
1586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  } else {
1596f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // Write out regular ELF sections
1606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    Module::iterator sect, sectEnd = pModule.end();
1616f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    for (sect = pModule.begin(); sect != sectEnd; ++sect)
16287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      writeSection(pModule, pOutput, *sect);
16322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
1646f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    emitShStrTab(target().getOutputFormat()->getShStrTab(), pModule, pOutput);
1656f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1666f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (m_Config.targets().is32Bits()) {
1676f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      // Write out ELF header
1686f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      // Write out section header table
1696f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      writeELFHeader<32>(m_Config, pModule, pOutput);
1706f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      if (is_dynobj || is_exec)
1716f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        emitProgramHeader<32>(pOutput);
1726f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1736f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      emitSectionHeader<32>(pModule, m_Config, pOutput);
1746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
1756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else if (m_Config.targets().is64Bits()) {
1766f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      // Write out ELF header
1776f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      // Write out section header table
1786f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      writeELFHeader<64>(m_Config, pModule, pOutput);
1796f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      if (is_dynobj || is_exec)
1806f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        emitProgramHeader<64>(pOutput);
1816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      emitSectionHeader<64>(pModule, m_Config, pOutput);
1836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
1846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else
1850dea6bc96bb52346737966839ac68644f7939f58Stephen Hines      return llvm::make_error_code(llvm::errc::function_not_supported);
1866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
1876f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
1880dea6bc96bb52346737966839ac68644f7939f58Stephen Hines  return std::error_code();
1896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
1906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
19187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines// getOutputSize - count the final output size
19287f34658dec9097d987d254a990ea7f311bfc95fStephen Hinessize_t ELFObjectWriter::getOutputSize(const Module& pModule) const
19387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines{
19487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  if (m_Config.targets().is32Bits()) {
19587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    return getLastStartOffset<32>(pModule) +
19687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines           sizeof(ELFSizeTraits<32>::Shdr) * pModule.size();
19787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  } else if (m_Config.targets().is64Bits()) {
19887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    return getLastStartOffset<64>(pModule) +
19987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines           sizeof(ELFSizeTraits<64>::Shdr) * pModule.size();
20087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  } else {
20187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    assert(0 && "Invalid ELF Class");
20287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    return 0;
20387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  }
20487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines}
20587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines
2066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines// writeELFHeader - emit ElfXX_Ehdr
2076f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<size_t SIZE>
2086f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid ELFObjectWriter::writeELFHeader(const LinkerConfig& pConfig,
2096f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                     const Module& pModule,
21087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                     FileOutputBuffer& pOutput) const
2116f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
2126f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Ehdr ElfXX_Ehdr;
2136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Shdr ElfXX_Shdr;
2146f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Phdr ElfXX_Phdr;
2156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2166f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // ELF header must start from 0x0
21787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  MemoryRegion region = pOutput.request(0, sizeof(ElfXX_Ehdr));
21887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  ElfXX_Ehdr* header = (ElfXX_Ehdr*)region.begin();
2196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2206f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  memcpy(header->e_ident, ElfMagic, EI_MAG3+1);
2216f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2226f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_ident[EI_CLASS]      = (SIZE == 32) ? ELFCLASS32 : ELFCLASS64;
2236f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_ident[EI_DATA]       = pConfig.targets().isLittleEndian()?
2246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                       ELFDATA2LSB : ELFDATA2MSB;
2256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_ident[EI_VERSION]    = target().getInfo().ELFVersion();
2266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_ident[EI_OSABI]      = target().getInfo().OSABI();
2276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_ident[EI_ABIVERSION] = target().getInfo().ABIVersion();
2286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // FIXME: add processor-specific and core file types.
2306f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  switch(pConfig.codeGenType()) {
2316f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    case LinkerConfig::Object:
2326f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      header->e_type = ET_REL;
2336f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      break;
2346f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    case LinkerConfig::DynObj:
2356f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      header->e_type = ET_DYN;
2366f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      break;
2376f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    case LinkerConfig::Exec:
2386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      header->e_type = ET_EXEC;
2396f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      break;
2406f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    default:
2416f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::errs() << "unspported output file type: " << pConfig.codeGenType() << ".\n";
2426f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      header->e_type = ET_NONE;
2436f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
2446f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_machine   = target().getInfo().machine();
2456f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_version   = header->e_ident[EI_VERSION];
2466f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_entry     = getEntryPoint(pConfig, pModule);
2476f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2486f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (LinkerConfig::Object != pConfig.codeGenType())
2496f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    header->e_phoff   = sizeof(ElfXX_Ehdr);
2506f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  else
2516f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    header->e_phoff   = 0x0;
2526f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2536f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_shoff     = getLastStartOffset<SIZE>(pModule);
2546f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_flags     = target().getInfo().flags();
2556f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_ehsize    = sizeof(ElfXX_Ehdr);
2566f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_phentsize = sizeof(ElfXX_Phdr);
25787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  header->e_phnum     = target().elfSegmentTable().size();
2586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_shentsize = sizeof(ElfXX_Shdr);
2596f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_shnum     = pModule.size();
2606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  header->e_shstrndx  = pModule.getSection(".shstrtab")->index();
2616f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
2626f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2636f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// getEntryPoint
2646f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesuint64_t ELFObjectWriter::getEntryPoint(const LinkerConfig& pConfig,
2656f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                        const Module& pModule) const
2666f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
26787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  llvm::StringRef entry_name = target().getEntry(pModule);
2686f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  uint64_t result = 0x0;
2696f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
27087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  bool issue_warning = (pModule.getScript().hasEntry() &&
2716f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                        LinkerConfig::Object != pConfig.codeGenType() &&
2726f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                        LinkerConfig::DynObj != pConfig.codeGenType());
2736f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const LDSymbol* entry_symbol = pModule.getNamePool().findSymbol(entry_name);
2756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
2766f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // found the symbol
2776f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (NULL != entry_symbol) {
2786f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (entry_symbol->desc() != ResolveInfo::Define && issue_warning) {
2796f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::errs() << "WARNING: entry symbol '"
2806f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                   << entry_symbol->name()
2816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                   << "' exists but is not defined.\n";
2826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
2836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    result = entry_symbol->value();
2846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
2856f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // not in the symbol pool
2866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  else {
2876f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // We should parse entry as a number.
2886f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // @ref GNU ld manual, Options -e. e.g., -e 0x1000.
2896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    char* endptr;
2906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    result = strtoull(entry_name.data(), &endptr, 0);
2916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (*endptr != '\0') {
2926f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      if (issue_warning) {
2936f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        llvm::errs() << "cannot find entry symbol '"
2946f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                     << entry_name.data()
2956f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                     << "'.\n";
29622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao      }
2976f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      result = 0x0;
29822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao    }
2996f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
3006f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  return result;
3016f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
30222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
3036f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines// emitSectionHeader - emit ElfXX_Shdr
3046f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<size_t SIZE>
3056f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid ELFObjectWriter::emitSectionHeader(const Module& pModule,
3066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                        const LinkerConfig& pConfig,
30787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                        FileOutputBuffer& pOutput) const
3086f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
3096f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Shdr ElfXX_Shdr;
31022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
3116f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // emit section header
3126f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  unsigned int sectNum = pModule.size();
3136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  unsigned int header_size = sizeof(ElfXX_Shdr) * sectNum;
31487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  MemoryRegion region = pOutput.request(getLastStartOffset<SIZE>(pModule),
31587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                        header_size);
31687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  ElfXX_Shdr* shdr = (ElfXX_Shdr*)region.begin();
31722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
3186f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // Iterate the SectionTable in LDContext
3196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  unsigned int sectIdx = 0;
3206f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  unsigned int shstridx = 0; // NULL section has empty name
3216f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  for (; sectIdx < sectNum; ++sectIdx) {
3226f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    const LDSection *ld_sect   = pModule.getSectionTable().at(sectIdx);
3236f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_name      = shstridx;
3246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_type      = ld_sect->type();
3256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_flags     = ld_sect->flag();
3266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_addr      = ld_sect->addr();
3276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_offset    = ld_sect->offset();
3286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_size      = ld_sect->size();
3296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_addralign = ld_sect->align();
3306f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_entsize   = getSectEntrySize<SIZE>(*ld_sect);
3316f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_link      = getSectLink(*ld_sect, pConfig);
3326f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shdr[sectIdx].sh_info      = getSectInfo(*ld_sect);
3336f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
3346f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    // adjust strshidx
3356f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shstridx += ld_sect->name().size() + 1;
33622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao  }
3376f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
3386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
3396f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines// emitProgramHeader - emit ElfXX_Phdr
3406f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<size_t SIZE>
34187f34658dec9097d987d254a990ea7f311bfc95fStephen Hinesvoid ELFObjectWriter::emitProgramHeader(FileOutputBuffer& pOutput) const
3426f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
3436f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Ehdr ElfXX_Ehdr;
3446f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Phdr ElfXX_Phdr;
3456f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
3466f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  uint64_t start_offset, phdr_size;
34722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
3486f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  start_offset = sizeof(ElfXX_Ehdr);
3496f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  phdr_size = sizeof(ElfXX_Phdr);
3506f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // Program header must start directly after ELF header
35187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  MemoryRegion region = pOutput.request(start_offset,
35287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      target().elfSegmentTable().size() * phdr_size);
3536f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
35487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  ElfXX_Phdr* phdr = (ElfXX_Phdr*)region.begin();
3556f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
3566f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // Iterate the elf segment table in GNULDBackend
3576f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  size_t index = 0;
3586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  ELFSegmentFactory::const_iterator seg = target().elfSegmentTable().begin(),
3596f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                 segEnd = target().elfSegmentTable().end();
3606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  for (; seg != segEnd; ++seg, ++index) {
36187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_type   = (*seg)->type();
36287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_flags  = (*seg)->flag();
36387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_offset = (*seg)->offset();
36487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_vaddr  = (*seg)->vaddr();
36587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_paddr  = (*seg)->paddr();
36687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_filesz = (*seg)->filesz();
36787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_memsz  = (*seg)->memsz();
36887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    phdr[index].p_align  = (*seg)->align();
36922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao  }
3706f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
37122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao
3726f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// emitShStrTab - emit section string table
3736f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid
3746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen HinesELFObjectWriter::emitShStrTab(const LDSection& pShStrTab,
3756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                              const Module& pModule,
37687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                              FileOutputBuffer& pOutput)
3776f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
3786f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // write out data
37987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  MemoryRegion region = pOutput.request(pShStrTab.offset(), pShStrTab.size());
38087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  char* data = (char*)region.begin();
3816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  size_t shstrsize = 0;
3826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  Module::const_iterator section, sectEnd = pModule.end();
3836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  for (section = pModule.begin(); section != sectEnd; ++section) {
3846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    strcpy((char*)(data + shstrsize), (*section)->name().data());
3856f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    shstrsize += (*section)->name().size() + 1;
3866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
3876f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
3886f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
3896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// emitSectionData
3906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid
3916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen HinesELFObjectWriter::emitSectionData(const LDSection& pSection,
3926f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                 MemoryRegion& pRegion) const
3936f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
3946f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const SectionData* sd = NULL;
3956f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  switch (pSection.kind()) {
3966f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    case LDFileFormat::Relocation:
3976f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      assert(pSection.hasRelocData());
3986f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      return;
3996f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    case LDFileFormat::EhFrame:
4006f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      assert(pSection.hasEhFrame());
401f7ac0f19a1c8d0ad14bcf6456ce368b830fea886Stephen Hines      sd = pSection.getEhFrame()->getSectionData();
4026f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      break;
4036f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    default:
4046f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      assert(pSection.hasSectionData());
4056f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      sd = pSection.getSectionData();
4066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      break;
4076f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
4086f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  emitSectionData(*sd, pRegion);
4096f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
4106f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
41187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines/// emitEhFrame
41287f34658dec9097d987d254a990ea7f311bfc95fStephen Hinesvoid ELFObjectWriter::emitEhFrame(Module& pModule,
41387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                  EhFrame& pFrame, MemoryRegion& pRegion) const
41487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines{
41587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  emitSectionData(*pFrame.getSectionData(), pRegion);
41687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines
41787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  // Patch FDE field (offset to CIE)
41887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  for (EhFrame::cie_iterator i = pFrame.cie_begin(), e = pFrame.cie_end();
41987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines       i != e; ++i) {
42087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    EhFrame::CIE& cie = **i;
42187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    for (EhFrame::fde_iterator fi = cie.begin(), fe = cie.end();
42287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines         fi != fe; ++fi) {
42387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      EhFrame::FDE& fde = **fi;
42487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      if (fde.getRecordType() == EhFrame::RECORD_GENERATED) {
42587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        // Patch PLT offset
42687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        LDSection* plt_sect = pModule.getSection(".plt");
42787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        assert (plt_sect && "We have no plt but have corresponding eh_frame?");
42887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        uint64_t plt_offset = plt_sect->offset();
42987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        // FDE entry for PLT is always 32-bit
43087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        uint64_t fde_offset = pFrame.getSection().offset() + fde.getOffset() +
43187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                              EhFrame::getDataStartOffset<32>();
43287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        int32_t offset = fde_offset - plt_offset;
43387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        if (plt_offset < fde_offset)
43487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines          offset = -offset;
43587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        memcpy(pRegion.begin() + fde.getOffset() +
43687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                 EhFrame::getDataStartOffset<32>(),
43787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                 &offset, 4);
43887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        uint32_t size = plt_sect->size();
43987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        memcpy(pRegion.begin() + fde.getOffset() +
44087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                 EhFrame::getDataStartOffset<32>() + 4,
44187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                 &size, 4);
44287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      }
44387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      uint64_t fde_cie_ptr_offset = fde.getOffset() +
44487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                    EhFrame::getDataStartOffset<32>() -
44587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                                    /*ID*/4;
44687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      uint64_t cie_start_offset = cie.getOffset();
44787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      int32_t offset = fde_cie_ptr_offset - cie_start_offset;
44887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      if (fde_cie_ptr_offset < cie_start_offset)
44987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        offset = -offset;
45087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      memcpy(pRegion.begin() + fde_cie_ptr_offset, &offset, 4);
45187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    } // for loop fde_iterator
45287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  } // for loop cie_iterator
45387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines}
45487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines
4556f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// emitRelocation
4566f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid ELFObjectWriter::emitRelocation(const LinkerConfig& pConfig,
4576f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                     const LDSection& pSection,
4586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                     MemoryRegion& pRegion) const
4596f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
4606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const RelocData* sect_data = pSection.getRelocData();
4616f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  assert(NULL != sect_data && "SectionData is NULL in emitRelocation!");
4626f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
4636f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (pSection.type() == SHT_REL) {
4646f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (pConfig.targets().is32Bits())
4656f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      emitRel<32>(pConfig, *sect_data, pRegion);
4666f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else if (pConfig.targets().is64Bits())
4676f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      emitRel<64>(pConfig, *sect_data, pRegion);
4686f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else {
4696f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      fatal(diag::unsupported_bitclass) << pConfig.targets().triple().str()
4706f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                        << pConfig.targets().bitclass();
4716f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
4726f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  } else if (pSection.type() == SHT_RELA) {
4736f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (pConfig.targets().is32Bits())
4746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      emitRela<32>(pConfig, *sect_data, pRegion);
4756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else if (pConfig.targets().is64Bits())
4766f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      emitRela<64>(pConfig, *sect_data, pRegion);
4776f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else {
4786f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      fatal(diag::unsupported_bitclass) << pConfig.targets().triple().str()
4796f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                        << pConfig.targets().bitclass();
4806f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
4816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  } else
4826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    llvm::report_fatal_error("unsupported relocation section type!");
4836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
4846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
4856f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
4866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines// emitRel - emit ElfXX_Rel
4876f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<size_t SIZE>
4886f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid ELFObjectWriter::emitRel(const LinkerConfig& pConfig,
4896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                              const RelocData& pRelocData,
4906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                              MemoryRegion& pRegion) const
4916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
4926f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Rel  ElfXX_Rel;
4936f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Addr ElfXX_Addr;
4946f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Word ElfXX_Word;
4956f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
49687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  ElfXX_Rel* rel = reinterpret_cast<ElfXX_Rel*>(pRegion.begin());
4976f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
4986f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const Relocation* relocation = 0;
4996f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const FragmentRef* frag_ref = 0;
5006f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5016f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  for (RelocData::const_iterator it = pRelocData.begin(),
5026f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines       ie = pRelocData.end(); it != ie; ++it, ++rel) {
50387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    ElfXX_Addr r_offset = 0;
50487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    ElfXX_Word r_sym = 0;
5056f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    relocation = &(llvm::cast<Relocation>(*it));
5076f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    frag_ref = &(relocation->targetRef());
5086f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5096f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if(LinkerConfig::DynObj == pConfig.codeGenType() ||
5106f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines       LinkerConfig::Exec == pConfig.codeGenType()) {
51187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_offset = static_cast<ElfXX_Addr>(
5126f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                      frag_ref->frag()->getParent()->getSection().addr() +
5136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                      frag_ref->getOutputOffset());
5146f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
5156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else {
51687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_offset = static_cast<ElfXX_Addr>(frag_ref->getOutputOffset());
5176f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
51887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines
5196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if( relocation->symInfo() == NULL )
52087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_sym = 0;
5216f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else
52287f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_sym = static_cast<ElfXX_Word>(
5236f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines              target().getSymbolIdx(relocation->symInfo()->outSymbol()));
5246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
52587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    target().emitRelocation(*rel, relocation->type(), r_sym, r_offset);
5266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
5276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
5286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines// emitRela - emit ElfXX_Rela
5306f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<size_t SIZE>
5316f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid ELFObjectWriter::emitRela(const LinkerConfig& pConfig,
5326f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                               const RelocData& pRelocData,
5336f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                               MemoryRegion& pRegion) const
5346f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
5356f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Rela ElfXX_Rela;
5366f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Addr ElfXX_Addr;
5376f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Word ElfXX_Word;
5386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
53987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  ElfXX_Rela* rel = reinterpret_cast<ElfXX_Rela*>(pRegion.begin());
5406f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5416f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const Relocation* relocation = 0;
5426f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const FragmentRef* frag_ref = 0;
5436f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5446f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  for (RelocData::const_iterator it = pRelocData.begin(),
5456f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines       ie = pRelocData.end(); it != ie; ++it, ++rel) {
54687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    ElfXX_Addr r_offset = 0;
54787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    ElfXX_Word r_sym = 0;
5486f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5496f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    relocation = &(llvm::cast<Relocation>(*it));
5506f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    frag_ref = &(relocation->targetRef());
5516f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5526f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if(LinkerConfig::DynObj == pConfig.codeGenType() ||
5536f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines       LinkerConfig::Exec == pConfig.codeGenType()) {
55487f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_offset = static_cast<ElfXX_Addr>(
5556f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                      frag_ref->frag()->getParent()->getSection().addr() +
5566f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                      frag_ref->getOutputOffset());
5576f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
5586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else {
55987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_offset = static_cast<ElfXX_Addr>(frag_ref->getOutputOffset());
5606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
5616f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5626f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if( relocation->symInfo() == NULL )
56387f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_sym = 0;
5646f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else
56587f34658dec9097d987d254a990ea7f311bfc95fStephen Hines      r_sym = static_cast<ElfXX_Word>(
5666f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines              target().getSymbolIdx(relocation->symInfo()->outSymbol()));
5676f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
56887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    target().emitRelocation(*rel, relocation->type(),
56987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines                            r_sym, r_offset, relocation->addend());
5706f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
5716f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
5726f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5736f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// getSectEntrySize - compute ElfXX_Shdr::sh_entsize
5756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<size_t SIZE>
5766f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesuint64_t ELFObjectWriter::getSectEntrySize(const LDSection& pSection) const
5776f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
5786f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Word ElfXX_Word;
5796f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Sym  ElfXX_Sym;
5806f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Rel  ElfXX_Rel;
5816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Rela ElfXX_Rela;
5826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  typedef typename ELFSizeTraits<SIZE>::Dyn  ElfXX_Dyn;
5836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
5846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_DYNSYM == pSection.type() ||
5856f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::ELF::SHT_SYMTAB == pSection.type())
5866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return sizeof(ElfXX_Sym);
5876f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_REL == pSection.type())
5886f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return sizeof(ElfXX_Rel);
5896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_RELA == pSection.type())
5906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return sizeof(ElfXX_Rela);
5916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_HASH     == pSection.type() ||
5926f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::ELF::SHT_GNU_HASH == pSection.type())
5936f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return sizeof(ElfXX_Word);
5946f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_DYNAMIC == pSection.type())
5956f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return sizeof(ElfXX_Dyn);
59687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  // FIXME: We should get the entsize from input since the size of each
59787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  // character is specified in the section header's sh_entsize field.
59887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  // For example, traditional string is 0x1, UCS-2 is 0x2, ... and so on.
59987f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  // Ref: http://www.sco.com/developers/gabi/2003-12-17/ch4.sheader.html
60087f34658dec9097d987d254a990ea7f311bfc95fStephen Hines  if (pSection.flag() & llvm::ELF::SHF_STRINGS)
60187f34658dec9097d987d254a990ea7f311bfc95fStephen Hines    return 0x1;
6026f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  return 0x0;
6036f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
6046f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6056f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// getSectLink - compute ElfXX_Shdr::sh_link
6066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesuint64_t ELFObjectWriter::getSectLink(const LDSection& pSection,
6076f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                      const LinkerConfig& pConfig) const
6086f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
6096f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_SYMTAB == pSection.type())
6106f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return target().getOutputFormat()->getStrTab().index();
6116f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_DYNSYM == pSection.type())
6126f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return target().getOutputFormat()->getDynStrTab().index();
6136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_DYNAMIC == pSection.type())
6146f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return target().getOutputFormat()->getDynStrTab().index();
6156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_HASH     == pSection.type() ||
6166f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::ELF::SHT_GNU_HASH == pSection.type())
6176f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return target().getOutputFormat()->getDynSymTab().index();
6186f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_REL == pSection.type() ||
6196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::ELF::SHT_RELA == pSection.type()) {
6206f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (LinkerConfig::Object == pConfig.codeGenType())
6216f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      return target().getOutputFormat()->getSymTab().index();
6226f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    else
6236f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      return target().getOutputFormat()->getDynSymTab().index();
6246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
6256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  // FIXME: currently we link ARM_EXIDX section to output text section here
6266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_ARM_EXIDX == pSection.type())
6276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return target().getOutputFormat()->getText().index();
6286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  return llvm::ELF::SHN_UNDEF;
6296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
6306f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6316f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// getSectInfo - compute ElfXX_Shdr::sh_info
6326f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesuint64_t ELFObjectWriter::getSectInfo(const LDSection& pSection) const
6336f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
6346f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_SYMTAB == pSection.type() ||
6356f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::ELF::SHT_DYNSYM == pSection.type())
6366f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    return pSection.getInfo();
6376f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  if (llvm::ELF::SHT_REL == pSection.type() ||
6396f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      llvm::ELF::SHT_RELA == pSection.type()) {
6406f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    const LDSection* info_link = pSection.getLink();
6416f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    if (NULL != info_link)
6426f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      return info_link->index();
6436f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
6446f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6456f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  return 0x0;
6466f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
6476f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6486f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// getLastStartOffset
6496f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<>
6506f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesuint64_t ELFObjectWriter::getLastStartOffset<32>(const Module& pModule) const
6516f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
6526f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const LDSection* lastSect = pModule.back();
6536f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  assert(lastSect != NULL);
6546f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  return Align<32>(lastSect->offset() + lastSect->size());
6556f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
6566f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6576f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// getLastStartOffset
6586f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinestemplate<>
6596f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesuint64_t ELFObjectWriter::getLastStartOffset<64>(const Module& pModule) const
6606f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
6616f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  const LDSection* lastSect = pModule.back();
6626f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  assert(lastSect != NULL);
6636f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  return Align<64>(lastSect->offset() + lastSect->size());
6646f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines}
6656f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
6666f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines/// emitSectionData
6676f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesvoid ELFObjectWriter::emitSectionData(const SectionData& pSD,
6686f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                                      MemoryRegion& pRegion) const
6696f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines{
6706f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  SectionData::const_iterator fragIter, fragEnd = pSD.end();
6716f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  size_t cur_offset = 0;
6726f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  for (fragIter = pSD.begin(); fragIter != fragEnd; ++fragIter) {
6736f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    size_t size = fragIter->size();
6746f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    switch(fragIter->getKind()) {
6756f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      case Fragment::Region: {
6766f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        const RegionFragment& region_frag = llvm::cast<RegionFragment>(*fragIter);
67787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        const char* from = region_frag.getRegion().begin();
67887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        memcpy(pRegion.begin() + cur_offset, from, size);
6796f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
6806f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      }
6816f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      case Fragment::Alignment: {
6826f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        // TODO: emit values with different sizes (> 1 byte), and emit nops
6836f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        const AlignFragment& align_frag = llvm::cast<AlignFragment>(*fragIter);
6846f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        uint64_t count = size / align_frag.getValueSize();
6856f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        switch (align_frag.getValueSize()) {
6866f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines          case 1u:
68787f34658dec9097d987d254a990ea7f311bfc95fStephen Hines            std::memset(pRegion.begin() + cur_offset,
6886f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                        align_frag.getValue(),
6896f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                        count);
6906f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines            break;
6916f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines          default:
6926f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines            llvm::report_fatal_error("unsupported value size for align fragment emission yet.\n");
6936f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines            break;
6946f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        }
6956f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
6966f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      }
6976f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      case Fragment::Fillment: {
6986f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        const FillFragment& fill_frag = llvm::cast<FillFragment>(*fragIter);
6996f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        if (0 == size ||
7006f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines            0 == fill_frag.getValueSize() ||
7016f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines            0 == fill_frag.size()) {
7026f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines          // ignore virtual fillment
7036f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines          break;
7046f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        }
7056f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines
7066f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        uint64_t num_tiles = fill_frag.size() / fill_frag.getValueSize();
7076f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        for (uint64_t i = 0; i != num_tiles; ++i) {
70887f34658dec9097d987d254a990ea7f311bfc95fStephen Hines          std::memset(pRegion.begin() + cur_offset,
7096f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                      fill_frag.getValue(),
7106f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines                      fill_frag.getValueSize());
7116f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        }
7126f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
7136f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      }
7146f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      case Fragment::Stub: {
7156f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        const Stub& stub_frag = llvm::cast<Stub>(*fragIter);
71687f34658dec9097d987d254a990ea7f311bfc95fStephen Hines        memcpy(pRegion.begin() + cur_offset, stub_frag.getContent(), size);
7176f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
7186f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      }
7196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      case Fragment::Null: {
7206f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        assert(0x0 == size);
7216f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
7226f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      }
7236f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      case Fragment::Target:
7246f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        llvm::report_fatal_error("Target fragment should not be in a regular section.\n");
7256f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
7266f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines      default:
7276f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        llvm::report_fatal_error("invalid fragment should not be in a regular section.\n");
7286f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines        break;
7296f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    }
7306f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines    cur_offset += size;
7316f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines  }
73222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao}
7335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao
734