//===- ReadStageTest.cpp --------------------------------------------------===//
//
//                     The MCLinker Project
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "ReadStageTest.h"

#include <mcld/Module.h>
#include <mcld/LD/LDContext.h>
#include <mcld/LD/LDSection.h>
#include <mcld/LD/SectionData.h>
#include <mcld/Fragment/Fragment.h>
#include <mcld/Fragment/FragmentRef.h>
#include <mcld/LD/LDSymbol.h>
#include <mcld/LD/ResolveInfo.h>
#include <mcld/MC/InputFactory.h>
#include <mcld/Support/FileHandle.h>

#include <sstream>
#include <iostream>

using namespace std;

using namespace mcld;
using namespace mcld::test;


// Constructor can do set-up work for all test here.
ReadStageTest::ReadStageTest()
  : m_pLinker(NULL) {
}

// Destructor can do clean-up work that doesn't throw exceptions here.
ReadStageTest::~ReadStageTest()
{
}

// SetUp() will be called immediately before each test.
void ReadStageTest::SetUp()
{
  m_pLinker = new mcld::test::TestLinker();
  m_pLinker->initialize("arm-none-linux-gnueabi");

  // set up target-dependent constraints of attributes
  m_pLinker->config()->attribute().constraint().enableWholeArchive();
  m_pLinker->config()->attribute().constraint().disableAsNeeded();
  m_pLinker->config()->attribute().constraint().setSharedSystem();

  // set up the predefined attributes
  m_pLinker->config()->attribute().predefined().setWholeArchive();
  m_pLinker->config()->attribute().predefined().setDynamic();

  // set up target dependent options
  mcld::sys::fs::Path path = TOPDIR;
  path.append("test/libs/ARM/Android/android-14");
  m_pLinker->setSysRoot(path);
  m_pLinker->addSearchDir("=/");

  m_pLinker->config()->options().setDyld("/usr/lib/ld.so.1");
  m_pLinker->config()->options().setBsymbolic(true);
}

// TearDown() will be called immediately after each test.
void ReadStageTest::TearDown()
{
  delete m_pLinker;
}

void ReadStageTest::dumpInput(const mcld::Input &pInput, mcld::FileHandle &pFile, size_t pIdent)
{
  stringstream sstream;
  for (size_t i=0; i < pIdent; ++i)
    sstream << " ";
  sstream << "<input name=\"" << pInput.name() << "\">\n";

  LDContext::const_sect_iterator sect, sectEnd = pInput.context()->sectEnd();
  for (sect = pInput.context()->sectBegin(); sect != sectEnd; ++sect) {
    for (size_t i=0; i < (pIdent+1); ++i)
      sstream << " ";
    sstream << "<section name=\"" << (*sect)->name() << "\"/>\n";
  }
  for (size_t i=0; i < pIdent; ++i)
    sstream << " ";
  sstream << "</input>\n";

  size_t org_size = pFile.size();
  pFile.truncate(sstream.str().size() + org_size);
  pFile.write(sstream.str().data(), org_size, sstream.str().size());
}

void ReadStageTest::dumpOutput(const mcld::Module& pModule, mcld::FileHandle &pFile, size_t pIdent)
{
  stringstream sstream;
  for (size_t i=0; i < pIdent; ++i)
    sstream << " ";
  sstream << "<output name=\"" << m_pLinker->module()->name() << "\">\n";

  Module::const_iterator sect, sectEnd = pModule.end();
  for (sect = pModule.begin(); sect != sectEnd; ++sect) {
    for (size_t i=0; i < (pIdent+1); ++i)
      sstream << " ";
    sstream << "<section name=\"" << (*sect)->name() << "\"/>\n";
  }
  for (size_t i=0; i < pIdent; ++i)
    sstream << " ";
  sstream << "</output>\n";

  size_t org_size = pFile.size();
  pFile.truncate(sstream.str().size() + org_size);
  pFile.write(sstream.str().data(), org_size, sstream.str().size());
}
//===----------------------------------------------------------------------===//
// Testcases
//===----------------------------------------------------------------------===//
TEST_F(ReadStageTest, quake) {
  mcld::sys::fs::Path top_level = TOPDIR;

  // set up output
  m_pLinker->config()->setCodeGenType(mcld::LinkerConfig::DynObj);
  m_pLinker->setOutput(top_level + "unittests/plasma.so");


  // set up input
  m_pLinker->addObject(top_level + "test/libs/ARM/Android/android-14/crtbegin_so.o");
  m_pLinker->addObject(top_level + "test/Android/Plasma/ARM/plasma.o");
  m_pLinker->addNameSpec("m");
  m_pLinker->addNameSpec("log");
  m_pLinker->addNameSpec("jnigraphics");
  m_pLinker->addNameSpec("c");
  m_pLinker->addObject(top_level + "test/libs/ARM/Android/android-14/crtend_so.o");

  // dump status
  m_pLinker->getObjLinker()->normalize();

  FileHandle file;
  file.open(top_level + "unittests/read_stage.xml",
     FileHandle::ReadWrite | FileHandle::Create | FileHandle::Truncate, 0644);

  Module::input_iterator input, inEnd = m_pLinker->module()->input_end();
  for (input = m_pLinker->module()->input_begin(); input != inEnd; ++input) {
    dumpInput(**input, file, 1);
  }

  dumpOutput(*m_pLinker->module(), file, 1);
  // dump status
  ASSERT_TRUE(m_pLinker->getObjLinker()->mergeSections());
}