Gordon Henriksen | 2b0eed2 | 2007-12-11 00:20:48 +0000 | [diff] [blame] | 1 | //===-- BitReader.cpp -----------------------------------------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | f3ebc3f | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Gordon Henriksen | 2b0eed2 | 2007-12-11 00:20:48 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | #include "llvm-c/BitReader.h" |
Eric Christopher | a6b9600 | 2015-12-18 01:46:52 +0000 | [diff] [blame] | 11 | #include "llvm-c/Core.h" |
Teresa Johnson | ad17679 | 2016-11-11 05:34:58 +0000 | [diff] [blame] | 12 | #include "llvm/Bitcode/BitcodeReader.h" |
Rafael Espindola | 3ee23a9 | 2015-02-03 00:49:57 +0000 | [diff] [blame] | 13 | #include "llvm/IR/DiagnosticPrinter.h" |
Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 14 | #include "llvm/IR/LLVMContext.h" |
Filip Pizlo | dec20e4 | 2013-05-01 20:59:00 +0000 | [diff] [blame] | 15 | #include "llvm/IR/Module.h" |
Gordon Henriksen | 2b0eed2 | 2007-12-11 00:20:48 +0000 | [diff] [blame] | 16 | #include "llvm/Support/MemoryBuffer.h" |
Rafael Espindola | 3ee23a9 | 2015-02-03 00:49:57 +0000 | [diff] [blame] | 17 | #include "llvm/Support/raw_ostream.h" |
Anton Korobeynikov | 579f071 | 2008-02-20 11:08:44 +0000 | [diff] [blame] | 18 | #include <cstring> |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 19 | #include <string> |
Gordon Henriksen | 2b0eed2 | 2007-12-11 00:20:48 +0000 | [diff] [blame] | 20 | |
| 21 | using namespace llvm; |
| 22 | |
Gordon Henriksen | 34eb6d8 | 2007-12-19 22:30:40 +0000 | [diff] [blame] | 23 | /* Builds a module from the bitcode in the specified memory buffer, returning a |
| 24 | reference to the module via the OutModule parameter. Returns 0 on success. |
Chris Lattner | 25963c6 | 2010-01-09 22:27:07 +0000 | [diff] [blame] | 25 | Optionally returns a human-readable error message via OutMessage. */ |
Rafael Espindola | d7f9c25 | 2015-12-18 14:06:34 +0000 | [diff] [blame] | 26 | LLVMBool LLVMParseBitcode(LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutModule, |
| 27 | char **OutMessage) { |
Mehdi Amini | 03b42e4 | 2016-04-14 21:59:01 +0000 | [diff] [blame] | 28 | return LLVMParseBitcodeInContext(LLVMGetGlobalContext(), MemBuf, OutModule, |
Daniel Dunbar | 754946c | 2010-02-15 21:08:22 +0000 | [diff] [blame] | 29 | OutMessage); |
Owen Anderson | 31d44e4 | 2009-07-02 07:17:57 +0000 | [diff] [blame] | 30 | } |
| 31 | |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 32 | LLVMBool LLVMParseBitcode2(LLVMMemoryBufferRef MemBuf, |
| 33 | LLVMModuleRef *OutModule) { |
Mehdi Amini | 03b42e4 | 2016-04-14 21:59:01 +0000 | [diff] [blame] | 34 | return LLVMParseBitcodeInContext2(LLVMGetGlobalContext(), MemBuf, OutModule); |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 35 | } |
| 36 | |
Chris Lattner | 25963c6 | 2010-01-09 22:27:07 +0000 | [diff] [blame] | 37 | LLVMBool LLVMParseBitcodeInContext(LLVMContextRef ContextRef, |
| 38 | LLVMMemoryBufferRef MemBuf, |
| 39 | LLVMModuleRef *OutModule, |
| 40 | char **OutMessage) { |
Rafael Espindola | 3ee23a9 | 2015-02-03 00:49:57 +0000 | [diff] [blame] | 41 | MemoryBufferRef Buf = unwrap(MemBuf)->getMemBufferRef(); |
| 42 | LLVMContext &Ctx = *unwrap(ContextRef); |
| 43 | |
Peter Collingbourne | d9445c4 | 2016-11-13 07:00:17 +0000 | [diff] [blame] | 44 | Expected<std::unique_ptr<Module>> ModuleOrErr = parseBitcodeFile(Buf, Ctx); |
| 45 | if (Error Err = ModuleOrErr.takeError()) { |
| 46 | std::string Message; |
| 47 | handleAllErrors(std::move(Err), [&](ErrorInfoBase &EIB) { |
| 48 | Message = EIB.message(); |
| 49 | }); |
Rafael Espindola | 9d2bfc4 | 2015-12-14 23:17:03 +0000 | [diff] [blame] | 50 | if (OutMessage) |
Rafael Espindola | dcfd6ed | 2015-02-03 01:53:03 +0000 | [diff] [blame] | 51 | *OutMessage = strdup(Message.c_str()); |
Rafael Espindola | d7f9c25 | 2015-12-18 14:06:34 +0000 | [diff] [blame] | 52 | *OutModule = wrap((Module *)nullptr); |
Gordon Henriksen | 2b0eed2 | 2007-12-11 00:20:48 +0000 | [diff] [blame] | 53 | return 1; |
| 54 | } |
Joe Abbey | 2ad8df2 | 2012-11-25 15:23:39 +0000 | [diff] [blame] | 55 | |
Rafael Espindola | dcd1dca | 2015-06-16 22:27:55 +0000 | [diff] [blame] | 56 | *OutModule = wrap(ModuleOrErr.get().release()); |
Gordon Henriksen | 2b0eed2 | 2007-12-11 00:20:48 +0000 | [diff] [blame] | 57 | return 0; |
| 58 | } |
| 59 | |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 60 | LLVMBool LLVMParseBitcodeInContext2(LLVMContextRef ContextRef, |
| 61 | LLVMMemoryBufferRef MemBuf, |
| 62 | LLVMModuleRef *OutModule) { |
| 63 | MemoryBufferRef Buf = unwrap(MemBuf)->getMemBufferRef(); |
| 64 | LLVMContext &Ctx = *unwrap(ContextRef); |
| 65 | |
Peter Collingbourne | d9445c4 | 2016-11-13 07:00:17 +0000 | [diff] [blame] | 66 | ErrorOr<std::unique_ptr<Module>> ModuleOrErr = |
| 67 | expectedToErrorOrAndEmitErrors(Ctx, parseBitcodeFile(Buf, Ctx)); |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 68 | if (ModuleOrErr.getError()) { |
| 69 | *OutModule = wrap((Module *)nullptr); |
| 70 | return 1; |
| 71 | } |
| 72 | |
| 73 | *OutModule = wrap(ModuleOrErr.get().release()); |
| 74 | return 0; |
| 75 | } |
| 76 | |
Gordon Henriksen | 34eb6d8 | 2007-12-19 22:30:40 +0000 | [diff] [blame] | 77 | /* Reads a module from the specified path, returning via the OutModule parameter |
| 78 | a module provider which performs lazy deserialization. Returns 0 on success. |
Joe Abbey | 2ad8df2 | 2012-11-25 15:23:39 +0000 | [diff] [blame] | 79 | Optionally returns a human-readable error message via OutMessage. */ |
Erick Tryzelaar | ad0e0cb | 2010-03-02 23:58:54 +0000 | [diff] [blame] | 80 | LLVMBool LLVMGetBitcodeModuleInContext(LLVMContextRef ContextRef, |
| 81 | LLVMMemoryBufferRef MemBuf, |
Rafael Espindola | d7f9c25 | 2015-12-18 14:06:34 +0000 | [diff] [blame] | 82 | LLVMModuleRef *OutM, char **OutMessage) { |
Rafael Espindola | f382b88 | 2015-12-18 13:58:05 +0000 | [diff] [blame] | 83 | LLVMContext &Ctx = *unwrap(ContextRef); |
Rafael Espindola | e2c1d77 | 2014-08-26 22:00:09 +0000 | [diff] [blame] | 84 | std::unique_ptr<MemoryBuffer> Owner(unwrap(MemBuf)); |
Peter Collingbourne | d9445c4 | 2016-11-13 07:00:17 +0000 | [diff] [blame] | 85 | Expected<std::unique_ptr<Module>> ModuleOrErr = |
Peter Collingbourne | e2dcf7c | 2016-11-08 06:03:43 +0000 | [diff] [blame] | 86 | getOwningLazyBitcodeModule(std::move(Owner), Ctx); |
Eric Fiselier | c79c9d8 | 2016-11-14 07:26:17 +0000 | [diff] [blame] | 87 | // Release the buffer if we didn't take ownership of it since we never owned |
| 88 | // it anyway. |
| 89 | (void)Owner.release(); |
Joe Abbey | 2ad8df2 | 2012-11-25 15:23:39 +0000 | [diff] [blame] | 90 | |
Peter Collingbourne | d9445c4 | 2016-11-13 07:00:17 +0000 | [diff] [blame] | 91 | if (Error Err = ModuleOrErr.takeError()) { |
| 92 | std::string Message; |
| 93 | handleAllErrors(std::move(Err), [&](ErrorInfoBase &EIB) { |
| 94 | Message = EIB.message(); |
| 95 | }); |
Gordon Henriksen | 34eb6d8 | 2007-12-19 22:30:40 +0000 | [diff] [blame] | 96 | if (OutMessage) |
Rafael Espindola | f382b88 | 2015-12-18 13:58:05 +0000 | [diff] [blame] | 97 | *OutMessage = strdup(Message.c_str()); |
Peter Collingbourne | d9445c4 | 2016-11-13 07:00:17 +0000 | [diff] [blame] | 98 | *OutM = wrap((Module *)nullptr); |
Gordon Henriksen | 34eb6d8 | 2007-12-19 22:30:40 +0000 | [diff] [blame] | 99 | return 1; |
| 100 | } |
Joe Abbey | 2ad8df2 | 2012-11-25 15:23:39 +0000 | [diff] [blame] | 101 | |
Rafael Espindola | dcd1dca | 2015-06-16 22:27:55 +0000 | [diff] [blame] | 102 | *OutM = wrap(ModuleOrErr.get().release()); |
Rafael Espindola | 5b6c1e8 | 2014-01-13 18:31:04 +0000 | [diff] [blame] | 103 | |
Gordon Henriksen | 34eb6d8 | 2007-12-19 22:30:40 +0000 | [diff] [blame] | 104 | return 0; |
Erick Tryzelaar | ad0e0cb | 2010-03-02 23:58:54 +0000 | [diff] [blame] | 105 | } |
| 106 | |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 107 | LLVMBool LLVMGetBitcodeModuleInContext2(LLVMContextRef ContextRef, |
| 108 | LLVMMemoryBufferRef MemBuf, |
| 109 | LLVMModuleRef *OutM) { |
| 110 | LLVMContext &Ctx = *unwrap(ContextRef); |
| 111 | std::unique_ptr<MemoryBuffer> Owner(unwrap(MemBuf)); |
| 112 | |
Peter Collingbourne | d9445c4 | 2016-11-13 07:00:17 +0000 | [diff] [blame] | 113 | ErrorOr<std::unique_ptr<Module>> ModuleOrErr = expectedToErrorOrAndEmitErrors( |
| 114 | Ctx, getOwningLazyBitcodeModule(std::move(Owner), Ctx)); |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 115 | Owner.release(); |
| 116 | |
| 117 | if (ModuleOrErr.getError()) { |
| 118 | *OutM = wrap((Module *)nullptr); |
| 119 | return 1; |
| 120 | } |
| 121 | |
| 122 | *OutM = wrap(ModuleOrErr.get().release()); |
| 123 | return 0; |
| 124 | } |
| 125 | |
Erick Tryzelaar | ad0e0cb | 2010-03-02 23:58:54 +0000 | [diff] [blame] | 126 | LLVMBool LLVMGetBitcodeModule(LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutM, |
| 127 | char **OutMessage) { |
| 128 | return LLVMGetBitcodeModuleInContext(LLVMGetGlobalContext(), MemBuf, OutM, |
| 129 | OutMessage); |
| 130 | } |
Rafael Espindola | 2339ffe | 2015-12-18 23:46:42 +0000 | [diff] [blame] | 131 | |
| 132 | LLVMBool LLVMGetBitcodeModule2(LLVMMemoryBufferRef MemBuf, |
| 133 | LLVMModuleRef *OutM) { |
| 134 | return LLVMGetBitcodeModuleInContext2(LLVMGetGlobalContext(), MemBuf, OutM); |
| 135 | } |