<div dir="ltr"><div style="font-size:12.8px">Previously, the COFF driver would call exit(0) when called as</div><div style="font-size:12.8px">a library. Now there is a global function setLibraryMode()</div><div style="font-size:12.8px">which causes link() to return on success.</div><div style="font-size:12.8px"><br></div><div style="font-size:12.8px">Furthermore, link() calls freeArena() before returning, to</div><div style="font-size:12.8px">clean up resources.</div><div style="font-size:12.8px">---</div><div style="font-size:12.8px"> lld/COFF/Driver.cpp | 27 +++++++++++++++++++++++----</div><div style="font-size:12.8px"> lld/COFF/Driver.h   |  2 ++</div><div style="font-size:12.8px"> 2 files changed, 25 insertions(+), 4 deletions(-)</div><div style="font-size:12.8px"><br></div><div style="font-size:12.8px"><br></div><div style="font-size:12.8px"><div>diff --git a/lld/COFF/Driver.cpp b/lld/COFF/Driver.cpp</div><div>index c039740715c..a769196e8b8 100644</div><div>--- a/lld/COFF/Driver.cpp</div><div>+++ b/lld/COFF/Driver.cpp</div><div>@@ -46,6 +46,12 @@ using llvm::sys::Process;</div><div> namespace lld {</div><div> namespace coff {</div><div> </div><div>+</div><div>+static bool IsLibraryMode = false;</div><div>+void setLibraryMode(void) {</div><div>+  IsLibraryMode = true;</div><div>+}</div><div>+</div><div> Configuration *Config;</div><div> LinkerDriver *Driver;</div><div> </div><div>@@ -65,6 +71,7 @@ bool link(ArrayRef<const char *> Args, raw_ostream &Diag) {</div><div> </div><div>   Driver = make<LinkerDriver>();</div><div>   Driver->link(Args);</div><div>+  freeArena();</div><div>   return !ErrorCount;</div><div> }</div><div> </div><div>@@ -1077,7 +1084,11 @@ void LinkerDriver::link(ArrayRef<<wbr>const char *> ArgsArr) {</div><div>   if (!Args.hasArg(OPT_INPUT)) {</div><div>     fixupExports();</div><div>     createImportLibrary(/*AsLib=*<wbr>/true);</div><div>-    exit(0);</div><div>+    if (IsLibraryMode) {</div><div>+      return;</div><div>+    } else {</div><div>+      exit(0);</div><div>+    }</div><div>   }</div><div> </div><div>   // Handle /delayload</div><div>@@ -1169,7 +1180,11 @@ void LinkerDriver::link(ArrayRef<<wbr>const char *> ArgsArr) {</div><div>   // This is useful because MSVC link.exe can generate complete PDBs.</div><div>   if (Args.hasArg(OPT_msvclto)) {</div><div>     invokeMSVC(Args);</div><div>-    exit(0);</div><div>+    if (IsLibraryMode) {</div><div>+      return;</div><div>+    } else {</div><div>+      exit(0);</div><div>+    }</div><div>   }</div><div> </div><div>   // Do LTO by compiling bitcode input files to a set of native COFF files then</div><div>@@ -1267,8 +1282,12 @@ void LinkerDriver::link(ArrayRef<<wbr>const char *> ArgsArr) {</div><div>   if (ErrorCount)</div><div>     return;</div><div> </div><div>-  // Call exit to avoid calling destructors.</div><div>-  exit(0);</div><div>+  if (IsLibraryMode) {</div><div>+    return</div><div>+  } else {</div><div>+    // Call exit to avoid calling destructors.</div><div>+    exit(0);</div><div>+  }</div><div> }</div><div> </div><div> } // namespace coff</div><div>diff --git a/lld/COFF/Driver.h b/lld/COFF/Driver.h</div><div>index df54fc1e45e..c1c0f36b011 100644</div><div>--- a/lld/COFF/Driver.h</div><div>+++ b/lld/COFF/Driver.h</div><div>@@ -35,6 +35,8 @@ using llvm::COFF::MachineTypes;</div><div> using llvm::COFF::WindowsSubsystem;</div><div> using llvm::Optional;</div><div> </div><div>+void setLibraryMode(void);</div><div>+</div><div> // Implemented in MarkLive.cpp.</div><div> void markLive(const std::vector<Chunk *> &Chunks);</div></div></div>