1 //===--- DataBufferLLVM.cpp -------------------------------------*- C++ -*-===//
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 "lldb/Utility/DataBufferLLVM.h"
11 
12 #include "llvm/ADT/Twine.h"
13 #include "llvm/Support/FileSystem.h"
14 #include "llvm/Support/MemoryBuffer.h"
15 
16 using namespace lldb_private;
17 
18 DataBufferLLVM::DataBufferLLVM(std::unique_ptr<llvm::MemoryBuffer> MemBuffer)
19     : Buffer(std::move(MemBuffer)) {
20   assert(Buffer != nullptr &&
21          "Cannot construct a DataBufferLLVM with a null buffer");
22 }
23 
24 DataBufferLLVM::~DataBufferLLVM() {}
25 
26 std::shared_ptr<DataBufferLLVM>
27 DataBufferLLVM::CreateSliceFromPath(const llvm::Twine &Path, uint64_t Size,
28                                uint64_t Offset, bool Private) {
29   // If the file resides non-locally, pass the volatile flag so that we don't
30   // mmap it.
31   if (!Private)
32     Private = !llvm::sys::fs::is_local(Path);
33 
34   auto Buffer = llvm::MemoryBuffer::getFileSlice(Path, Size, Offset, Private);
35   if (!Buffer)
36     return nullptr;
37   return std::shared_ptr<DataBufferLLVM>(
38       new DataBufferLLVM(std::move(*Buffer)));
39 }
40 
41 std::shared_ptr<DataBufferLLVM>
42 DataBufferLLVM::CreateFromPath(const llvm::Twine &Path, bool NullTerminate, bool Private) {
43   // If the file resides non-locally, pass the volatile flag so that we don't
44   // mmap it.
45   if (!Private)
46     Private = !llvm::sys::fs::is_local(Path);
47 
48   auto Buffer = llvm::MemoryBuffer::getFile(Path, -1, NullTerminate, Private);
49   if (!Buffer)
50     return nullptr;
51   return std::shared_ptr<DataBufferLLVM>(
52       new DataBufferLLVM(std::move(*Buffer)));
53 }
54 
55 uint8_t *DataBufferLLVM::GetBytes() {
56   return const_cast<uint8_t *>(GetBuffer());
57 }
58 
59 const uint8_t *DataBufferLLVM::GetBytes() const { return GetBuffer(); }
60 
61 lldb::offset_t DataBufferLLVM::GetByteSize() const {
62   return Buffer->getBufferSize();
63 }
64 
65 const uint8_t *DataBufferLLVM::GetBuffer() const {
66   return reinterpret_cast<const uint8_t *>(Buffer->getBufferStart());
67 }
68