| //===- Buffer.cpp ---------------------------------------------------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "Buffer.h" |
| #include "llvm-objcopy.h" |
| #include "llvm/Support/FileOutputBuffer.h" |
| #include "llvm/Support/MemoryBuffer.h" |
| #include <memory> |
| |
| namespace llvm { |
| namespace objcopy { |
| |
| Buffer::~Buffer() {} |
| |
| void FileBuffer::allocate(size_t Size) { |
| Expected<std::unique_ptr<FileOutputBuffer>> BufferOrErr = |
| FileOutputBuffer::create(getName(), Size, FileOutputBuffer::F_executable); |
| handleAllErrors(BufferOrErr.takeError(), [this](const ErrorInfoBase &E) { |
| error("failed to open " + getName() + ": " + E.message()); |
| }); |
| Buf = std::move(*BufferOrErr); |
| } |
| |
| Error FileBuffer::commit() { return Buf->commit(); } |
| |
| uint8_t *FileBuffer::getBufferStart() { |
| return reinterpret_cast<uint8_t *>(Buf->getBufferStart()); |
| } |
| |
| void MemBuffer::allocate(size_t Size) { |
| Buf = WritableMemoryBuffer::getNewMemBuffer(Size, getName()); |
| } |
| |
| Error MemBuffer::commit() { return Error::success(); } |
| |
| uint8_t *MemBuffer::getBufferStart() { |
| return reinterpret_cast<uint8_t *>(Buf->getBufferStart()); |
| } |
| |
| std::unique_ptr<WritableMemoryBuffer> MemBuffer::releaseMemoryBuffer() { |
| return std::move(Buf); |
| } |
| |
| } // end namespace objcopy |
| } // end namespace llvm |