diff --git a/src/cache/fs.rs b/src/cache/fs.rs index 25863dd..c6d47cb 100644 --- a/src/cache/fs.rs +++ b/src/cache/fs.rs @@ -91,14 +91,23 @@ pub(super) async fn read_file( return None; } - Box::pin(EncryptedDiskReader::new( - file, - SecretStream::init_pull( - &Header::from_slice(&header_bytes).expect("failed to get header"), - key, - ) - .expect("Failed to initialize decryption kesy"), - )) + let header = match Header::from_slice(&header_bytes) { + Some(header) => header, + None => { + warn!("Found file, but encrypted header was invalid. Assuming corrupted!"); + return None; + } + }; + + let secret_stream = match SecretStream::init_pull(&header, key) { + Ok(stream) => stream, + Err(_) => { + warn!("Failed to init secret stream with key and header. Assuming corrupted!"); + return None; + } + }; + + Box::pin(EncryptedDiskReader::new(file, secret_stream)) } else { Box::pin(file) }; diff --git a/src/cache/mem.rs b/src/cache/mem.rs index 36b3fc8..5fa57e8 100644 --- a/src/cache/mem.rs +++ b/src/cache/mem.rs @@ -74,22 +74,24 @@ impl InternalMemoryCache for Lru { /// Memory accelerated disk cache. Uses the internal cache implementation in /// memory to speed up reads. -pub struct MemoryCache { - inner: InnerCache, +pub struct MemoryCache { + inner: ColdCache, cur_mem_size: AtomicU64, - mem_cache: Mutex, + mem_cache: Mutex, master_sender: Sender<(CacheKey, Bytes, ImageMetadata, u64)>, } -impl - MemoryCache +impl MemoryCache +where + MemoryCacheImpl: 'static + InternalMemoryCache, + ColdCache: 'static + Cache, { - pub async fn new(inner: InnerCache, max_mem_size: u64) -> Arc { + pub async fn new(inner: ColdCache, max_mem_size: u64) -> Arc { let (tx, mut rx) = channel(100); let new_self = Arc::new(Self { inner, cur_mem_size: AtomicU64::new(0), - mem_cache: Mutex::new(InternalCacheImpl::unbounded()), + mem_cache: Mutex::new(MemoryCacheImpl::unbounded()), master_sender: tx, }); @@ -99,6 +101,7 @@ impl Cache for MemoryCache +impl Cache for MemoryCache where - InternalCacheImpl: InternalMemoryCache, - InnerCache: CallbackCache, + MemoryCacheImpl: InternalMemoryCache, + ColdCache: CallbackCache, { #[inline] async fn get(