return true;
}
-// Add ".obj" extension if the given path name has no file extension.
-StringRef canonicalizeInputFileName(PECOFFTargetInfo &info, std::string path) {
- if (llvm::sys::path::extension(path).empty())
- path.append(".obj");
- return info.allocateString(path);
-}
-
// Replace a file extension with ".exe". If the given file has no
// extension, just add ".exe".
StringRef getDefaultOutputFileName(PECOFFTargetInfo &info, StringRef path) {
for (int i = doubleDashPosition + 1; i < argc; ++i)
inputPaths.push_back(argv[i]);
- // Add ".obj" extension for those who have no file extension.
for (const StringRef &path : inputPaths)
- info.appendInputFile(canonicalizeInputFileName(info, path));
+ info.appendInputFileOrLibrary(path);
// If -out option was not specified, the default output file name is
- // constructed by replacing an extension with ".exe".
+ // constructed by replacing an extension of the first input file
+ // with ".exe".
if (info.outputPath().empty() && !inputPaths.empty())
info.setOutputPath(getDefaultOutputFileName(info, inputPaths[0]));
#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"
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 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;
}