16c025e212
Currently we do not de-duplicate library files specified by /defaultlib option. As a result, the same files are added multiple times to the input graph. In particular, some popular files, such as kernel32.lib or oldnames.lib, are added more than 10 times during linking of LLD. That makes the linker slower, as it needs to parse the same file again and again. This patch solves the issue by de-duplicating. The same file will be added only once to the input graph. This patch improved the LLD linking time from 10.5 seconds to 7.7 seconds on my 4-core Core i7 Macbook Pro. llvm-svn: 196504
74 lines
2.2 KiB
C++
74 lines
2.2 KiB
C++
//===- lld/unittest/DriverTest.h ------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "lld/Driver/Driver.h"
|
|
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
namespace {
|
|
|
|
using namespace llvm;
|
|
using namespace lld;
|
|
|
|
template<typename D, typename T>
|
|
class ParserTest : public testing::Test {
|
|
protected:
|
|
|
|
virtual const LinkingContext *linkingContext() = 0;
|
|
|
|
std::string &errorMessage() { return _errorMessage; }
|
|
|
|
// Convenience method for getting number of input files.
|
|
int inputFileCount() { return linkingContext()->inputGraph().size(); }
|
|
|
|
// Convenience method for getting i'th input files name.
|
|
std::string inputFile(int index) {
|
|
const InputElement &inputElement = linkingContext()->inputGraph()[index];
|
|
if (inputElement.kind() == InputElement::Kind::File)
|
|
return *dyn_cast<FileNode>(&inputElement)->getPath(*linkingContext());
|
|
llvm_unreachable("not handling other types of input files");
|
|
}
|
|
|
|
// Convenience method for getting i'th input files name.
|
|
std::string inputFile(int index1, int index2) {
|
|
Group *group = dyn_cast<Group>(&linkingContext()->inputGraph()[index1]);
|
|
if (!group)
|
|
llvm_unreachable("not handling other types of input files");
|
|
FileNode *file = dyn_cast<FileNode>(group->elements()[index2].get());
|
|
if (!file)
|
|
llvm_unreachable("not handling other types of input files");
|
|
return *file->getPath(*linkingContext());
|
|
}
|
|
|
|
// For unit tests to call driver with various command lines.
|
|
bool parse(const char *args, ...) {
|
|
// Construct command line options from varargs.
|
|
std::vector<const char *> vec;
|
|
vec.push_back(args);
|
|
va_list ap;
|
|
va_start(ap, args);
|
|
while (const char *arg = va_arg(ap, const char *))
|
|
vec.push_back(arg);
|
|
va_end(ap);
|
|
|
|
// Call the parser.
|
|
raw_string_ostream os(_errorMessage);
|
|
return D::parse(vec.size(), &vec[0], _context, os);
|
|
}
|
|
|
|
T _context;
|
|
std::string _errorMessage;
|
|
};
|
|
|
|
}
|