0ct0pu5/ladybird
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

LibSQL: Prevent writing to free heap blocks

Browse source 
Each block index should have been requested before a write happens to
it. If this is not the case, return an error.
This commit is contained in:
Jelle Raaijmakers 2023-06-13 21:52:10 +02:00 • committed by Tim Flynn
parent d3335d6ef8
commit c05e08decb
Notes: sideshowbarker 2024-07-16 22:11:09 +09:00
2 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Tests/LibSQL/TestSqlHeap.cpp
View file

@ -186,6 +186,7 @@ TEST_CASE(heap_free_storage)
TRY_OR_FAIL(heap->free_storage(storage_block_id));
// Again, write some large storage spanning multiple blocks
storage_block_id = heap->request_new_block_index();
TRY_OR_FAIL(heap->write_storage(storage_block_id, long_string.bytes()));
MUST(heap->flush());
auto new_heap_size = MUST(heap->file_size_in_bytes());

8
Userland/Libraries/LibSQL/Heap.cpp
View file

@ -118,8 +118,12 @@ ErrorOr<ByteBuffer> Heap::read_storage(Block::Index index)
ErrorOr<void> Heap::write_storage(Block::Index index, ReadonlyBytes data)
{
dbgln_if(SQL_DEBUG, "{}({}, {} bytes)", __FUNCTION__, index, data.size());
VERIFY(index > 0);
VERIFY(data.size() > 0);
if (index == 0)
return Error::from_string_view("Writing to zero block is not allowed"sv);
if (data.is_empty())
return Error::from_string_view("Writing empty data is not allowed"sv);
if (m_free_block_indices.contains_slow(index))
return Error::from_string_view("Invalid write to a free block index"sv);
// Split up the storage across multiple blocks if necessary, creating a chain
u32 remaining_size = static_cast<u32>(data.size());