diff options
| -rw-r--r-- | lld/lib/Driver/GnuLdDriver.cpp | 4 | ||||
| -rw-r--r-- | lld/unittests/DriverTests/GnuLdDriverTest.cpp | 13 |
2 files changed, 16 insertions, 1 deletions
diff --git a/lld/lib/Driver/GnuLdDriver.cpp b/lld/lib/Driver/GnuLdDriver.cpp index 244438470c9..bd130ea49f2 100644 --- a/lld/lib/Driver/GnuLdDriver.cpp +++ b/lld/lib/Driver/GnuLdDriver.cpp @@ -613,7 +613,9 @@ bool GnuLdDriver::parse(int argc, const char *argv[], ErrorOr<StringRef> pathOrErr = findFile(*ctx, path, dashL); if (std::error_code ec = pathOrErr.getError()) { auto file = llvm::make_unique<ErrorFile>(path, ec); - ctx->getNodes().push_back(llvm::make_unique<FileNode>(std::move(file))); + auto node = llvm::make_unique<FileNode>(std::move(file)); + node->setAsNeeded(asNeeded); + ctx->getNodes().push_back(std::move(node)); break; } StringRef realpath = pathOrErr.get(); diff --git a/lld/unittests/DriverTests/GnuLdDriverTest.cpp b/lld/unittests/DriverTests/GnuLdDriverTest.cpp index b5a819cf916..5ca62447ba7 100644 --- a/lld/unittests/DriverTests/GnuLdDriverTest.cpp +++ b/lld/unittests/DriverTests/GnuLdDriverTest.cpp @@ -147,6 +147,19 @@ TEST_F(GnuLdParserTest, DefsymMisssingValue) { EXPECT_FALSE(parse("ld", "a.o", "--defsym=sym=", nullptr)); } +// --as-needed + +TEST_F(GnuLdParserTest, AsNeeded) { + EXPECT_TRUE(parse("ld", "a.o", "--as-needed", "b.o", "c.o", + "--no-as-needed", "d.o", nullptr)); + std::vector<std::unique_ptr<Node>> &nodes = _context->getNodes(); + EXPECT_EQ((size_t)4, nodes.size()); + EXPECT_FALSE(cast<FileNode>(nodes[0].get())->asNeeded()); + EXPECT_TRUE(cast<FileNode>(nodes[1].get())->asNeeded()); + EXPECT_TRUE(cast<FileNode>(nodes[2].get())->asNeeded()); + EXPECT_FALSE(cast<FileNode>(nodes[3].get())->asNeeded()); +} + // Linker script TEST_F(GnuLdParserTest, LinkerScriptGroup) { |
