annotate lib/DebugInfo/MSF/MappedBlockStream.cpp @ 124:4fa72497ed5d

fix
author mir3636
date Thu, 30 Nov 2017 20:04:56 +0900
parents 803732b1fca8
children 3a76565eade5
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
1 //===- MappedBlockStream.cpp - Reads stream data from an MSF file ---------===//
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
2 //
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
3 // The LLVM Compiler Infrastructure
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
4 //
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
5 // This file is distributed under the University of Illinois Open Source
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
6 // License. See LICENSE.TXT for details.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
7 //
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
8 //===----------------------------------------------------------------------===//
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
9
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
10 #include "llvm/DebugInfo/MSF/MappedBlockStream.h"
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
11 #include "llvm/ADT/ArrayRef.h"
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
12 #include "llvm/ADT/STLExtras.h"
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
13 #include "llvm/DebugInfo/MSF/MSFCommon.h"
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
14 #include "llvm/Support/BinaryStreamWriter.h"
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
15 #include "llvm/Support/Endian.h"
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
16 #include "llvm/Support/Error.h"
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
17 #include "llvm/Support/MathExtras.h"
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
18 #include <algorithm>
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
19 #include <cassert>
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
20 #include <cstdint>
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
21 #include <cstring>
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
22 #include <utility>
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
23 #include <vector>
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
24
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
25 using namespace llvm;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
26 using namespace llvm::msf;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
27
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
28 namespace {
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
29
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
30 template <typename Base> class MappedBlockStreamImpl : public Base {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
31 public:
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
32 template <typename... Args>
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
33 MappedBlockStreamImpl(Args &&... Params)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
34 : Base(std::forward<Args>(Params)...) {}
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
35 };
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
36
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
37 } // end anonymous namespace
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
38
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
39 using Interval = std::pair<uint32_t, uint32_t>;
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
40
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
41 static Interval intersect(const Interval &I1, const Interval &I2) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
42 return std::make_pair(std::max(I1.first, I2.first),
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
43 std::min(I1.second, I2.second));
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
44 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
45
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
46 MappedBlockStream::MappedBlockStream(uint32_t BlockSize,
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
47 const MSFStreamLayout &Layout,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
48 BinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
49 BumpPtrAllocator &Allocator)
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
50 : BlockSize(BlockSize), StreamLayout(Layout), MsfData(MsfData),
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
51 Allocator(Allocator) {}
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
52
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
53 std::unique_ptr<MappedBlockStream> MappedBlockStream::createStream(
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
54 uint32_t BlockSize, const MSFStreamLayout &Layout, BinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
55 BumpPtrAllocator &Allocator) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
56 return llvm::make_unique<MappedBlockStreamImpl<MappedBlockStream>>(
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
57 BlockSize, Layout, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
58 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
59
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
60 std::unique_ptr<MappedBlockStream> MappedBlockStream::createIndexedStream(
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
61 const MSFLayout &Layout, BinaryStreamRef MsfData, uint32_t StreamIndex,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
62 BumpPtrAllocator &Allocator) {
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
63 assert(StreamIndex < Layout.StreamMap.size() && "Invalid stream index");
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
64 MSFStreamLayout SL;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
65 SL.Blocks = Layout.StreamMap[StreamIndex];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
66 SL.Length = Layout.StreamSizes[StreamIndex];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
67 return llvm::make_unique<MappedBlockStreamImpl<MappedBlockStream>>(
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
68 Layout.SB->BlockSize, SL, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
69 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
70
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
71 std::unique_ptr<MappedBlockStream>
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
72 MappedBlockStream::createDirectoryStream(const MSFLayout &Layout,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
73 BinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
74 BumpPtrAllocator &Allocator) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
75 MSFStreamLayout SL;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
76 SL.Blocks = Layout.DirectoryBlocks;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
77 SL.Length = Layout.SB->NumDirectoryBytes;
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
78 return createStream(Layout.SB->BlockSize, SL, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
79 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
80
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
81 std::unique_ptr<MappedBlockStream>
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
82 MappedBlockStream::createFpmStream(const MSFLayout &Layout,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
83 BinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
84 BumpPtrAllocator &Allocator) {
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
85 MSFStreamLayout SL(getFpmStreamLayout(Layout));
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
86 return createStream(Layout.SB->BlockSize, SL, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
87 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
88
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
89 Error MappedBlockStream::readBytes(uint32_t Offset, uint32_t Size,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
90 ArrayRef<uint8_t> &Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
91 // Make sure we aren't trying to read beyond the end of the stream.
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
92 if (auto EC = checkOffset(Offset, Size))
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
93 return EC;
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
94
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
95 if (tryReadContiguously(Offset, Size, Buffer))
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
96 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
97
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
98 auto CacheIter = CacheMap.find(Offset);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
99 if (CacheIter != CacheMap.end()) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
100 // Try to find an alloc that was large enough for this request.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
101 for (auto &Entry : CacheIter->second) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
102 if (Entry.size() >= Size) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
103 Buffer = Entry.slice(0, Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
104 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
105 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
106 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
107 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
108
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
109 // We couldn't find a buffer that started at the correct offset (the most
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
110 // common scenario). Try to see if there is a buffer that starts at some
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
111 // other offset but overlaps the desired range.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
112 for (auto &CacheItem : CacheMap) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
113 Interval RequestExtent = std::make_pair(Offset, Offset + Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
114
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
115 // We already checked this one on the fast path above.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
116 if (CacheItem.first == Offset)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
117 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
118 // If the initial extent of the cached item is beyond the ending extent
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
119 // of the request, there is no overlap.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
120 if (CacheItem.first >= Offset + Size)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
121 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
122
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
123 // We really only have to check the last item in the list, since we append
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
124 // in order of increasing length.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
125 if (CacheItem.second.empty())
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
126 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
127
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
128 auto CachedAlloc = CacheItem.second.back();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
129 // If the initial extent of the request is beyond the ending extent of
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
130 // the cached item, there is no overlap.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
131 Interval CachedExtent =
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
132 std::make_pair(CacheItem.first, CacheItem.first + CachedAlloc.size());
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
133 if (RequestExtent.first >= CachedExtent.first + CachedExtent.second)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
134 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
135
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
136 Interval Intersection = intersect(CachedExtent, RequestExtent);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
137 // Only use this if the entire request extent is contained in the cached
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
138 // extent.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
139 if (Intersection != RequestExtent)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
140 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
141
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
142 uint32_t CacheRangeOffset =
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
143 AbsoluteDifference(CachedExtent.first, Intersection.first);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
144 Buffer = CachedAlloc.slice(CacheRangeOffset, Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
145 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
146 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
147
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
148 // Otherwise allocate a large enough buffer in the pool, memcpy the data
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
149 // into it, and return an ArrayRef to that. Do not touch existing pool
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
150 // allocations, as existing clients may be holding a pointer which must
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
151 // not be invalidated.
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
152 uint8_t *WriteBuffer = static_cast<uint8_t *>(Allocator.Allocate(Size, 8));
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
153 if (auto EC = readBytes(Offset, MutableArrayRef<uint8_t>(WriteBuffer, Size)))
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
154 return EC;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
155
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
156 if (CacheIter != CacheMap.end()) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
157 CacheIter->second.emplace_back(WriteBuffer, Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
158 } else {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
159 std::vector<CacheEntry> List;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
160 List.emplace_back(WriteBuffer, Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
161 CacheMap.insert(std::make_pair(Offset, List));
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
162 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
163 Buffer = ArrayRef<uint8_t>(WriteBuffer, Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
164 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
165 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
166
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
167 Error MappedBlockStream::readLongestContiguousChunk(uint32_t Offset,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
168 ArrayRef<uint8_t> &Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
169 // Make sure we aren't trying to read beyond the end of the stream.
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
170 if (auto EC = checkOffset(Offset, 1))
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
171 return EC;
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
172
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
173 uint32_t First = Offset / BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
174 uint32_t Last = First;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
175
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
176 while (Last < getNumBlocks() - 1) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
177 if (StreamLayout.Blocks[Last] != StreamLayout.Blocks[Last + 1] - 1)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
178 break;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
179 ++Last;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
180 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
181
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
182 uint32_t OffsetInFirstBlock = Offset % BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
183 uint32_t BytesFromFirstBlock = BlockSize - OffsetInFirstBlock;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
184 uint32_t BlockSpan = Last - First + 1;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
185 uint32_t ByteSpan = BytesFromFirstBlock + (BlockSpan - 1) * BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
186
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
187 ArrayRef<uint8_t> BlockData;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
188 uint32_t MsfOffset = blockToOffset(StreamLayout.Blocks[First], BlockSize);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
189 if (auto EC = MsfData.readBytes(MsfOffset, BlockSize, BlockData))
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
190 return EC;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
191
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
192 BlockData = BlockData.drop_front(OffsetInFirstBlock);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
193 Buffer = ArrayRef<uint8_t>(BlockData.data(), ByteSpan);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
194 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
195 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
196
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
197 uint32_t MappedBlockStream::getLength() { return StreamLayout.Length; }
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
198
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
199 bool MappedBlockStream::tryReadContiguously(uint32_t Offset, uint32_t Size,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
200 ArrayRef<uint8_t> &Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
201 if (Size == 0) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
202 Buffer = ArrayRef<uint8_t>();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
203 return true;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
204 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
205 // Attempt to fulfill the request with a reference directly into the stream.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
206 // This can work even if the request crosses a block boundary, provided that
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
207 // all subsequent blocks are contiguous. For example, a 10k read with a 4k
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
208 // block size can be filled with a reference if, from the starting offset,
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
209 // 3 blocks in a row are contiguous.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
210 uint32_t BlockNum = Offset / BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
211 uint32_t OffsetInBlock = Offset % BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
212 uint32_t BytesFromFirstBlock = std::min(Size, BlockSize - OffsetInBlock);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
213 uint32_t NumAdditionalBlocks =
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
214 alignTo(Size - BytesFromFirstBlock, BlockSize) / BlockSize;
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
215
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
216 uint32_t RequiredContiguousBlocks = NumAdditionalBlocks + 1;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
217 uint32_t E = StreamLayout.Blocks[BlockNum];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
218 for (uint32_t I = 0; I < RequiredContiguousBlocks; ++I, ++E) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
219 if (StreamLayout.Blocks[I + BlockNum] != E)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
220 return false;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
221 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
222
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
223 // Read out the entire block where the requested offset starts. Then drop
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
224 // bytes from the beginning so that the actual starting byte lines up with
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
225 // the requested starting byte. Then, since we know this is a contiguous
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
226 // cross-block span, explicitly resize the ArrayRef to cover the entire
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
227 // request length.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
228 ArrayRef<uint8_t> BlockData;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
229 uint32_t FirstBlockAddr = StreamLayout.Blocks[BlockNum];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
230 uint32_t MsfOffset = blockToOffset(FirstBlockAddr, BlockSize);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
231 if (auto EC = MsfData.readBytes(MsfOffset, BlockSize, BlockData)) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
232 consumeError(std::move(EC));
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
233 return false;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
234 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
235 BlockData = BlockData.drop_front(OffsetInBlock);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
236 Buffer = ArrayRef<uint8_t>(BlockData.data(), Size);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
237 return true;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
238 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
239
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
240 Error MappedBlockStream::readBytes(uint32_t Offset,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
241 MutableArrayRef<uint8_t> Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
242 uint32_t BlockNum = Offset / BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
243 uint32_t OffsetInBlock = Offset % BlockSize;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
244
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
245 // Make sure we aren't trying to read beyond the end of the stream.
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
246 if (auto EC = checkOffset(Offset, Buffer.size()))
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
247 return EC;
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
248
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
249 uint32_t BytesLeft = Buffer.size();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
250 uint32_t BytesWritten = 0;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
251 uint8_t *WriteBuffer = Buffer.data();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
252 while (BytesLeft > 0) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
253 uint32_t StreamBlockAddr = StreamLayout.Blocks[BlockNum];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
254
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
255 ArrayRef<uint8_t> BlockData;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
256 uint32_t Offset = blockToOffset(StreamBlockAddr, BlockSize);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
257 if (auto EC = MsfData.readBytes(Offset, BlockSize, BlockData))
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
258 return EC;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
259
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
260 const uint8_t *ChunkStart = BlockData.data() + OffsetInBlock;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
261 uint32_t BytesInChunk = std::min(BytesLeft, BlockSize - OffsetInBlock);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
262 ::memcpy(WriteBuffer + BytesWritten, ChunkStart, BytesInChunk);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
263
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
264 BytesWritten += BytesInChunk;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
265 BytesLeft -= BytesInChunk;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
266 ++BlockNum;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
267 OffsetInBlock = 0;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
268 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
269
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
270 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
271 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
272
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
273 void MappedBlockStream::invalidateCache() { CacheMap.shrink_and_clear(); }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
274
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
275 void MappedBlockStream::fixCacheAfterWrite(uint32_t Offset,
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
276 ArrayRef<uint8_t> Data) const {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
277 // If this write overlapped a read which previously came from the pool,
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
278 // someone may still be holding a pointer to that alloc which is now invalid.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
279 // Compute the overlapping range and update the cache entry, so any
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
280 // outstanding buffers are automatically updated.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
281 for (const auto &MapEntry : CacheMap) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
282 // If the end of the written extent precedes the beginning of the cached
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
283 // extent, ignore this map entry.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
284 if (Offset + Data.size() < MapEntry.first)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
285 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
286 for (const auto &Alloc : MapEntry.second) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
287 // If the end of the cached extent precedes the beginning of the written
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
288 // extent, ignore this alloc.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
289 if (MapEntry.first + Alloc.size() < Offset)
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
290 continue;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
291
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
292 // If we get here, they are guaranteed to overlap.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
293 Interval WriteInterval = std::make_pair(Offset, Offset + Data.size());
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
294 Interval CachedInterval =
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
295 std::make_pair(MapEntry.first, MapEntry.first + Alloc.size());
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
296 // If they overlap, we need to write the new data into the overlapping
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
297 // range.
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
298 auto Intersection = intersect(WriteInterval, CachedInterval);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
299 assert(Intersection.first <= Intersection.second);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
300
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
301 uint32_t Length = Intersection.second - Intersection.first;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
302 uint32_t SrcOffset =
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
303 AbsoluteDifference(WriteInterval.first, Intersection.first);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
304 uint32_t DestOffset =
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
305 AbsoluteDifference(CachedInterval.first, Intersection.first);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
306 ::memcpy(Alloc.data() + DestOffset, Data.data() + SrcOffset, Length);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
307 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
308 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
309 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
310
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
311 WritableMappedBlockStream::WritableMappedBlockStream(
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
312 uint32_t BlockSize, const MSFStreamLayout &Layout,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
313 WritableBinaryStreamRef MsfData, BumpPtrAllocator &Allocator)
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
314 : ReadInterface(BlockSize, Layout, MsfData, Allocator),
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
315 WriteInterface(MsfData) {}
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
316
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
317 std::unique_ptr<WritableMappedBlockStream>
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
318 WritableMappedBlockStream::createStream(uint32_t BlockSize,
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
319 const MSFStreamLayout &Layout,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
320 WritableBinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
321 BumpPtrAllocator &Allocator) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
322 return llvm::make_unique<MappedBlockStreamImpl<WritableMappedBlockStream>>(
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
323 BlockSize, Layout, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
324 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
325
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
326 std::unique_ptr<WritableMappedBlockStream>
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
327 WritableMappedBlockStream::createIndexedStream(const MSFLayout &Layout,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
328 WritableBinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
329 uint32_t StreamIndex,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
330 BumpPtrAllocator &Allocator) {
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
331 assert(StreamIndex < Layout.StreamMap.size() && "Invalid stream index");
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
332 MSFStreamLayout SL;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
333 SL.Blocks = Layout.StreamMap[StreamIndex];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
334 SL.Length = Layout.StreamSizes[StreamIndex];
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
335 return createStream(Layout.SB->BlockSize, SL, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
336 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
337
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
338 std::unique_ptr<WritableMappedBlockStream>
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
339 WritableMappedBlockStream::createDirectoryStream(
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
340 const MSFLayout &Layout, WritableBinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
341 BumpPtrAllocator &Allocator) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
342 MSFStreamLayout SL;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
343 SL.Blocks = Layout.DirectoryBlocks;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
344 SL.Length = Layout.SB->NumDirectoryBytes;
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
345 return createStream(Layout.SB->BlockSize, SL, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
346 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
347
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
348 std::unique_ptr<WritableMappedBlockStream>
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
349 WritableMappedBlockStream::createFpmStream(const MSFLayout &Layout,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
350 WritableBinaryStreamRef MsfData,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
351 BumpPtrAllocator &Allocator,
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
352 bool AltFpm) {
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
353 // We only want to give the user a stream containing the bytes of the FPM that
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
354 // are actually valid, but we want to initialize all of the bytes, even those
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
355 // that come from reserved FPM blocks where the entire block is unused. To do
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
356 // this, we first create the full layout, which gives us a stream with all
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
357 // bytes and all blocks, and initialize everything to 0xFF (all blocks in the
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
358 // file are unused). Then we create the minimal layout (which contains only a
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
359 // subset of the bytes previously initialized), and return that to the user.
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
360 MSFStreamLayout MinLayout(getFpmStreamLayout(Layout, false, AltFpm));
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
361
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
362 MSFStreamLayout FullLayout(getFpmStreamLayout(Layout, true, AltFpm));
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
363 auto Result =
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
364 createStream(Layout.SB->BlockSize, FullLayout, MsfData, Allocator);
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
365 if (!Result)
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
366 return Result;
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
367 std::vector<uint8_t> InitData(Layout.SB->BlockSize, 0xFF);
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
368 BinaryStreamWriter Initializer(*Result);
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
369 while (Initializer.bytesRemaining() > 0)
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
370 cantFail(Initializer.writeBytes(InitData));
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
371 return createStream(Layout.SB->BlockSize, MinLayout, MsfData, Allocator);
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
372 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
373
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
374 Error WritableMappedBlockStream::readBytes(uint32_t Offset, uint32_t Size,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
375 ArrayRef<uint8_t> &Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
376 return ReadInterface.readBytes(Offset, Size, Buffer);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
377 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
378
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
379 Error WritableMappedBlockStream::readLongestContiguousChunk(
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
380 uint32_t Offset, ArrayRef<uint8_t> &Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
381 return ReadInterface.readLongestContiguousChunk(Offset, Buffer);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
382 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
383
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
384 uint32_t WritableMappedBlockStream::getLength() {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
385 return ReadInterface.getLength();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
386 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
387
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
388 Error WritableMappedBlockStream::writeBytes(uint32_t Offset,
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
389 ArrayRef<uint8_t> Buffer) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
390 // Make sure we aren't trying to write beyond the end of the stream.
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
391 if (auto EC = checkOffset(Offset, Buffer.size()))
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
392 return EC;
120
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
393
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
394 uint32_t BlockNum = Offset / getBlockSize();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
395 uint32_t OffsetInBlock = Offset % getBlockSize();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
396
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
397 uint32_t BytesLeft = Buffer.size();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
398 uint32_t BytesWritten = 0;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
399 while (BytesLeft > 0) {
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
400 uint32_t StreamBlockAddr = getStreamLayout().Blocks[BlockNum];
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
401 uint32_t BytesToWriteInChunk =
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
402 std::min(BytesLeft, getBlockSize() - OffsetInBlock);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
403
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
404 const uint8_t *Chunk = Buffer.data() + BytesWritten;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
405 ArrayRef<uint8_t> ChunkData(Chunk, BytesToWriteInChunk);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
406 uint32_t MsfOffset = blockToOffset(StreamBlockAddr, getBlockSize());
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
407 MsfOffset += OffsetInBlock;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
408 if (auto EC = WriteInterface.writeBytes(MsfOffset, ChunkData))
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
409 return EC;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
410
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
411 BytesLeft -= BytesToWriteInChunk;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
412 BytesWritten += BytesToWriteInChunk;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
413 ++BlockNum;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
414 OffsetInBlock = 0;
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
415 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
416
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
417 ReadInterface.fixCacheAfterWrite(Offset, Buffer);
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
418
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
419 return Error::success();
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
420 }
1172e4bd9c6f update 4.0.0
mir3636
parents:
diff changeset
421
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
422 Error WritableMappedBlockStream::commit() { return WriteInterface.commit(); }