1f22ef01cSRoman Divacky //===-- BitWriter.cpp -----------------------------------------------------===//
2f22ef01cSRoman Divacky //
3f22ef01cSRoman Divacky //                     The LLVM Compiler Infrastructure
4f22ef01cSRoman Divacky //
5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source
6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details.
7f22ef01cSRoman Divacky //
8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
9f22ef01cSRoman Divacky 
10f22ef01cSRoman Divacky #include "llvm-c/BitWriter.h"
11f22ef01cSRoman Divacky #include "llvm/Bitcode/ReaderWriter.h"
12284c1978SDimitry Andric #include "llvm/IR/Module.h"
1391bc56edSDimitry Andric #include "llvm/Support/FileSystem.h"
14f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h"
15f22ef01cSRoman Divacky using namespace llvm;
16f22ef01cSRoman Divacky 
17f22ef01cSRoman Divacky 
18f22ef01cSRoman Divacky /*===-- Operations on modules ---------------------------------------------===*/
19f22ef01cSRoman Divacky 
20f22ef01cSRoman Divacky int LLVMWriteBitcodeToFile(LLVMModuleRef M, const char *Path) {
2139d628a0SDimitry Andric   std::error_code EC;
2239d628a0SDimitry Andric   raw_fd_ostream OS(Path, EC, sys::fs::F_None);
23f22ef01cSRoman Divacky 
2439d628a0SDimitry Andric   if (EC)
25f22ef01cSRoman Divacky     return -1;
26f22ef01cSRoman Divacky 
27f22ef01cSRoman Divacky   WriteBitcodeToFile(unwrap(M), OS);
28f22ef01cSRoman Divacky   return 0;
29f22ef01cSRoman Divacky }
30f22ef01cSRoman Divacky 
31f22ef01cSRoman Divacky int LLVMWriteBitcodeToFD(LLVMModuleRef M, int FD, int ShouldClose,
32f22ef01cSRoman Divacky                          int Unbuffered) {
33f22ef01cSRoman Divacky   raw_fd_ostream OS(FD, ShouldClose, Unbuffered);
34f22ef01cSRoman Divacky 
35f22ef01cSRoman Divacky   WriteBitcodeToFile(unwrap(M), OS);
36f22ef01cSRoman Divacky   return 0;
37f22ef01cSRoman Divacky }
38f22ef01cSRoman Divacky 
39f22ef01cSRoman Divacky int LLVMWriteBitcodeToFileHandle(LLVMModuleRef M, int FileHandle) {
40f22ef01cSRoman Divacky   return LLVMWriteBitcodeToFD(M, FileHandle, true, false);
41f22ef01cSRoman Divacky }
4239d628a0SDimitry Andric 
4339d628a0SDimitry Andric LLVMMemoryBufferRef LLVMWriteBitcodeToMemoryBuffer(LLVMModuleRef M) {
4439d628a0SDimitry Andric   std::string Data;
4539d628a0SDimitry Andric   raw_string_ostream OS(Data);
4639d628a0SDimitry Andric 
4739d628a0SDimitry Andric   WriteBitcodeToFile(unwrap(M), OS);
4839d628a0SDimitry Andric   return wrap(MemoryBuffer::getMemBufferCopy(OS.str()).release());
4939d628a0SDimitry Andric }
50