contrib/llvm/tools/llvm-objcopy/Buffer.cpp

   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
  19 Buffer::~Buffer() {}
  20
  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
  30 Error FileBuffer::commit() { return Buf->commit(); }
  31
  32 uint8_t *FileBuffer::getBufferStart() {
  33   return reinterpret_cast<uint8_t *>(Buf->getBufferStart());
  34 }
  35
  36 void MemBuffer::allocate(size_t Size) {
  37   Buf = WritableMemoryBuffer::getNewMemBuffer(Size, getName());
  38 }
  39
  40 Error MemBuffer::commit() { return Error::success(); }
  41
  42 uint8_t *MemBuffer::getBufferStart() {
  43   return reinterpret_cast<uint8_t *>(Buf->getBufferStart());
  44 }
  45
  46 std::unique_ptr<WritableMemoryBuffer> MemBuffer::releaseMemoryBuffer() {
  47   return std::move(Buf);
  48 }
  49
  50 } // end namespace objcopy
  51 } // end namespace llvm