diff options
| -rw-r--r-- | lld/lib/Driver/WinLinkDriver.cpp | 15 |
1 files changed, 5 insertions, 10 deletions
diff --git a/lld/lib/Driver/WinLinkDriver.cpp b/lld/lib/Driver/WinLinkDriver.cpp index 644fc0ac9eb..1fc008c22c1 100644 --- a/lld/lib/Driver/WinLinkDriver.cpp +++ b/lld/lib/Driver/WinLinkDriver.cpp @@ -490,6 +490,9 @@ WinLinkDriver::parse(int argc, const char *argv[], PECOFFLinkingContext &ctx, // Use the default entry name if /entry option is not given. if (ctx.entrySymbolName().empty()) ctx.setEntrySymbolName(getDefaultEntrySymbolName(ctx)); + StringRef entry = ctx.entrySymbolName(); + if (!entry.empty()) + ctx.addInitialUndefinedSymbol(entry); // Specifying both /opt:ref and /opt:noref is an error. if (parsedArgs->getLastArg(OPT_ref) && parsedArgs->getLastArg(OPT_ref_no)) { @@ -500,17 +503,9 @@ WinLinkDriver::parse(int argc, const char *argv[], PECOFFLinkingContext &ctx, // If dead-stripping is enabled, we need to add the entry symbol and // symbols given by /include to the dead strip root set, so that it // won't be removed from the output. - if (ctx.deadStrip()) { - StringRef entry = ctx.entrySymbolName(); - if (!entry.empty()) { - ctx.addInitialUndefinedSymbol(entry); - ctx.addDeadStripRoot(entry); - } - for (const StringRef symbolName : ctx.initialUndefinedSymbols()) { - ctx.addInitialUndefinedSymbol(entry); + if (ctx.deadStrip()) + for (const StringRef symbolName : ctx.initialUndefinedSymbols()) ctx.addDeadStripRoot(symbolName); - } - } // Arguments after "--" are interpreted as filenames even if they // start with a hypen or a slash. This is not compatible with link.exe |
