//===- lib/ReaderWriter/PECOFF/PECOFFTargetInfo.cpp -----------------------===//
//
//                             The LLVM Linker
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "GroupedSectionsPass.h"
#include "IdataPass.h"

#include "llvm/ADT/SmallString.h"
#include "llvm/Support/Path.h"
#include "lld/Core/PassManager.h"
#include "lld/Passes/LayoutPass.h"
#include "lld/ReaderWriter/PECOFFTargetInfo.h"
#include "lld/ReaderWriter/Reader.h"
#include "lld/ReaderWriter/Writer.h"

namespace lld {

namespace {
bool containDirectoryName(StringRef path) {
  SmallString<128> smallStr = StringRef(path);
  llvm::sys::path::remove_filename(smallStr);
  return !smallStr.str().empty();
}
} // anonymous namespace

error_code PECOFFTargetInfo::parseFile(
    std::unique_ptr<MemoryBuffer> &mb,
    std::vector<std::unique_ptr<File>> &result) const {
  return _reader->parseFile(mb, result);
}

bool PECOFFTargetInfo::validateImpl(raw_ostream &diagnostics) {
  if (_stackReserve < _stackCommit) {
    diagnostics << "Invalid stack size: reserve size must be equal to or "
                << "greater than commit size, but got "
                << _stackCommit << " and " << _stackReserve << ".\n";
    return true;
  }

  if (_heapReserve < _heapCommit) {
    diagnostics << "Invalid heap size: reserve size must be equal to or "
                << "greater than commit size, but got "
                << _heapCommit << " and " << _heapReserve << ".\n";
    return true;
  }

  _reader = createReaderPECOFF(*this);
  _writer = createWriterPECOFF(*this);
  return false;
}

/// Append the given file to the input file list. The file must be an object
/// file or an import library file.
bool PECOFFTargetInfo::appendInputFileOrLibrary(std::string path) {
  StringRef ext = llvm::sys::path::extension(path).lower();
  // This is an import library file. Look for the library file in the search
  // paths, unless the path contains a directory name.
  if (ext == ".lib") {
    if (containDirectoryName(path)) {
      appendInputFile(path);
      return true;
    }
    return appendLibraryFile(path);
  }
  // This is an object file otherwise. Add ".obj" extension if the given path
  // name has no file extension.
  if (ext.empty())
    path.append(".obj");
  appendInputFile(allocateString(path));
  return true;
}

/// Try to find the input library file from the search paths and append it to
/// the input file list. Returns true if the library file is found.
bool PECOFFTargetInfo::appendLibraryFile(StringRef filename) {
  // Current directory always takes precedence over the search paths.
  if (llvm::sys::fs::exists(filename)) {
    appendInputFile(filename);
    return true;
  }
  // Iterate over the search paths.
  for (StringRef dir : _inputSearchPaths) {
    SmallString<128> path = dir;
    llvm::sys::path::append(path, filename);
    if (llvm::sys::fs::exists(path.str())) {
      appendInputFile(allocateString(path.str()));
      return true;
    }
  }
  return false;
}

Writer &PECOFFTargetInfo::writer() const {
  return *_writer;
}

ErrorOr<Reference::Kind>
PECOFFTargetInfo::relocKindFromString(StringRef str) const {
  return make_error_code(yaml_reader_error::illegal_value);
}

ErrorOr<std::string>
PECOFFTargetInfo::stringFromRelocKind(Reference::Kind kind) const {
  return make_error_code(yaml_reader_error::illegal_value);
}

void PECOFFTargetInfo::addPasses(PassManager &pm) const {
  pm.add(std::unique_ptr<Pass>(new pecoff::GroupedSectionsPass()));
  pm.add(std::unique_ptr<Pass>(new pecoff::IdataPass()));
  pm.add(std::unique_ptr<Pass>(new LayoutPass()));
}

} // end namespace lld