[ELF] Implement semantic action for ENTRY linker script command.

Reviewed by:	shankarke, ruiu

llvm-svn: 227708

diff --git a/lld/lib/Driver/GnuLdDriver.cpp b/lld/lib/Driver/GnuLdDriver.cpp
index 04c9f06..2e00f23 100644 (file)
--- a/lld/lib/Driver/GnuLdDriver.cpp
+++ b/lld/lib/Driver/GnuLdDriver.cpp
@@ -289,6 +289,8 @@ GnuLdDriver::evalLinkerScript(ELFLinkingContext &ctx,
         return ec;
     if (auto *searchDir = dyn_cast<script::SearchDir>(c))
       ctx.addSearchPath(searchDir->getSearchPath());
+    if (auto *entry = dyn_cast<script::Entry>(c))
+      ctx.setEntrySymbolName(entry->getEntryName());
   }
   return std::error_code();
 }

diff --git a/lld/unittests/DriverTests/GnuLdDriverTest.cpp b/lld/unittests/DriverTests/GnuLdDriverTest.cpp
index d943397..261c952 100644 (file)
--- a/lld/unittests/DriverTests/GnuLdDriverTest.cpp
+++ b/lld/unittests/DriverTests/GnuLdDriverTest.cpp
@@ -191,3 +191,16 @@ TEST_F(GnuLdParserTest, LinkerScriptSearchDir) {
   EXPECT_EQ((size_t)2, searchPaths.size());
   EXPECT_EQ("/foo/bar", searchPaths[1]);
 }
+
+TEST_F(GnuLdParserTest, LinkerScriptEntry) {
+  parse("ld", "a.o", nullptr);
+  std::unique_ptr<MemoryBuffer> mb = MemoryBuffer::getMemBuffer(
+    "ENTRY(blah)", "foo.so");
+  std::string s;
+  raw_string_ostream out(s);
+  std::error_code ec = GnuLdDriver::evalLinkerScript(
+    *_context, std::move(mb), out);
+  EXPECT_FALSE(ec);
+  StringRef entrySymbol = _context->entrySymbolName();
+  EXPECT_EQ("blah", entrySymbol);
+}