blockworks-foundation
/
solana
mirror of https://github.com/blockworks-foundation/solana.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

check file size of accounts cache mmap before indexing (#25077)

This commit is contained in:
Jeff Washington (jwash) 2022-05-10 13:33:22 -05:00 committed by GitHub
parent 7e07fc219b
commit 8e5bb0a1e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
runtime/src/cache_hash_data.rs
View File

@ -182,6 +182,10 @@ impl CacheHashData {
let mmap = CacheHashDataFile::load_map(&path)?;
m1.stop();
stats.read_us = m1.as_us();
let header_size = std::mem::size_of::<Header>() as u64;
if file_len < header_size {
return Err(std::io::Error::from(std::io::ErrorKind::UnexpectedEof));
}
let cell_size = std::mem::size_of::<EntryType>() as u64;
let mut cache_file = CacheHashDataFile {
@ -192,7 +196,10 @@ impl CacheHashData {
let header = cache_file.get_header_mut();
let entries = header.count;
let capacity = cell_size * (entries as u64) + std::mem::size_of::<Header>() as u64;
let capacity = cell_size * (entries as u64) + header_size;
if file_len < capacity {
return Err(std::io::Error::from(std::io::ErrorKind::UnexpectedEof));
}
cache_file.capacity = capacity;
assert_eq!(
capacity, file_len,