119 lines
3.7 KiB
C++
119 lines
3.7 KiB
C++
//===- 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
|