From: Peter Collingbourne <peter@pcc.me.uk>
Date: Thu, 1 Dec 2016 23:13:11 +0000 (+0000)
Subject: llvm-modextract: Call keep() on the output stream before exiting.
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=a11b564b77a1ddef8f198363ac6ea1d99180839c;p=llvm

llvm-modextract: Call keep() on the output stream before exiting.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@288435 91177308-0d34-0410-b5e6-96231b3b80d8
---

diff --git a/test/tools/llvm-modextract/single.ll b/test/tools/llvm-modextract/single.ll
index 32915ff4dad..1d63b7622eb 100644
--- a/test/tools/llvm-modextract/single.ll
+++ b/test/tools/llvm-modextract/single.ll
@@ -4,6 +4,11 @@
 ; RUN: llvm-modextract -b -n 0 -o - %t | llvm-dis | FileCheck %s
 ; RUN: not llvm-modextract -b -n 1 -o - %t 2>&1 | FileCheck --check-prefix=ERROR %s
 
+; RUN: llvm-modextract -n 0 -o %t0 %t
+; RUN: llvm-dis -o - %t0 | FileCheck %s
+; RUN: llvm-modextract -b -n 0 -o %t1 %t
+; RUN: llvm-dis -o - %t1 | FileCheck %s
+
 ; CHECK: define void @f()
 ; ERROR: llvm-modextract: error: module index out of range; bitcode file contains 1 module(s)
 
diff --git a/tools/llvm-modextract/llvm-modextract.cpp b/tools/llvm-modextract/llvm-modextract.cpp
index b4279f2ab1e..6c2e364be44 100644
--- a/tools/llvm-modextract/llvm-modextract.cpp
+++ b/tools/llvm-modextract/llvm-modextract.cpp
@@ -62,11 +62,13 @@ int main(int argc, char **argv) {
     SmallVector<char, 0> Header;
     BitcodeWriter Writer(Header);
     Out->os() << Header << Ms[ModuleIndex].getBuffer();
+    Out->keep();
     return 0;
   }
 
   std::unique_ptr<Module> M = ExitOnErr(Ms[ModuleIndex].parseModule(Context));
   WriteBitcodeToFile(M.get(), Out->os());
 
+  Out->keep();
   return 0;
 }