diff options
| author | Per Larsson <[email protected]> | 2021-11-16 11:29:07 +0100 |
|---|---|---|
| committer | Per Larsson <[email protected]> | 2021-11-16 11:29:07 +0100 |
| commit | 2a3f095377247450bf8cb38a79a0218aa498d4c0 (patch) | |
| tree | 3ea4cdc1d9ee6529e2450d26f3de489c7c315b92 /zencore/compress.cpp | |
| parent | Handle 'partial on error' cache policy. (diff) | |
| download | zen-2a3f095377247450bf8cb38a79a0218aa498d4c0.tar.xz zen-2a3f095377247450bf8cb38a79a0218aa498d4c0.zip | |
Check input range when copy compressed range.
Diffstat (limited to 'zencore/compress.cpp')
| -rw-r--r-- | zencore/compress.cpp | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/zencore/compress.cpp b/zencore/compress.cpp index 4b6d83f6f..61f1effe4 100644 --- a/zencore/compress.cpp +++ b/zencore/compress.cpp @@ -693,6 +693,11 @@ ValidBufferOrEmpty(BufferType&& CompressedData) CompositeBuffer CopyCompressedRange(const BufferHeader& Header, const CompositeBuffer& CompressedData, uint64_t RawOffset, uint64_t RawSize) { + if (Header.TotalRawSize < RawOffset + RawSize) + { + return CompositeBuffer(); + } + if (Header.Method == CompressionMethod::None) { UniqueBuffer NewCompressedData = UniqueBuffer::Alloc(RawSize); @@ -862,9 +867,11 @@ CompressedBuffer CompressedBuffer::CopyRange(uint64_t RawOffset, uint64_t RawSize) const { using namespace detail; - const BufferHeader Header = BufferHeader::Read(CompressedData); - CompressedBuffer Range; - Range.CompressedData = CopyCompressedRange(Header, CompressedData, RawOffset, RawSize); + const BufferHeader Header = BufferHeader::Read(CompressedData); + const uint64_t TotalRawSize = RawSize < ~uint64_t(0) ? RawSize : Header.TotalRawSize - RawOffset; + + CompressedBuffer Range; + Range.CompressedData = CopyCompressedRange(Header, CompressedData, RawOffset, TotalRawSize); return Range; } |