1 //===- Buffer.cpp ---------------------------------------------------------===//
2 //
3 //                      The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "Buffer.h"
11 #include "llvm-objcopy.h"
12 #include "llvm/Support/FileOutputBuffer.h"
13 #include "llvm/Support/MemoryBuffer.h"
14 #include <memory>
15 
16 namespace llvm {
17 namespace objcopy {
18 
~Buffer()19 Buffer::~Buffer() {}
20 
allocate(size_t Size)21 void FileBuffer::allocate(size_t Size) {
22   Expected<std::unique_ptr<FileOutputBuffer>> BufferOrErr =
23       FileOutputBuffer::create(getName(), Size, FileOutputBuffer::F_executable);
24   handleAllErrors(BufferOrErr.takeError(), [this](const ErrorInfoBase &E) {
25     error("failed to open " + getName() + ": " + E.message());
26   });
27   Buf = std::move(*BufferOrErr);
28 }
29 
commit()30 Error FileBuffer::commit() { return Buf->commit(); }
31 
getBufferStart()32 uint8_t *FileBuffer::getBufferStart() {
33   return reinterpret_cast<uint8_t *>(Buf->getBufferStart());
34 }
35 
allocate(size_t Size)36 void MemBuffer::allocate(size_t Size) {
37   Buf = WritableMemoryBuffer::getNewMemBuffer(Size, getName());
38 }
39 
commit()40 Error MemBuffer::commit() { return Error::success(); }
41 
getBufferStart()42 uint8_t *MemBuffer::getBufferStart() {
43   return reinterpret_cast<uint8_t *>(Buf->getBufferStart());
44 }
45 
releaseMemoryBuffer()46 std::unique_ptr<WritableMemoryBuffer> MemBuffer::releaseMemoryBuffer() {
47   return std::move(Buf);
48 }
49 
50 } // end namespace objcopy
51 } // end namespace llvm
52