asp/keep
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: handle buffer capacity check correctly

Browse Source
This commit is contained in:
Andrew Phillips
2025-08-19 12:56:45 -03:00
committed by Andrew Phillips (aider)
parent 9aa76857b0
commit 1659bf20d4
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/meta_plugin/binary.rs
View File

@@ -60,16 +60,14 @@ impl MetaPlugin for BinaryMetaPlugin {
// Calculate how much data we can still accept // Calculate how much data we can still accept
let remaining_capacity = self.max_buffer_size.saturating_sub(self.buffer.len()); let remaining_capacity = self.max_buffer_size.saturating_sub(self.buffer.len());
if remaining_capacity == 0 { if remaining_capacity > 0 {
return; // Determine how much data to copy
let bytes_to_take = std::cmp::min(data.len(), remaining_capacity);
// Add data to our buffer
self.buffer.extend_from_slice(&data[..bytes_to_take]);
} }
// Determine how much data to copy
let bytes_to_take = std::cmp::min(data.len(), remaining_capacity);
// Add data to our buffer
self.buffer.extend_from_slice(&data[..bytes_to_take]);
// If we've reached our buffer limit, save the metadata immediately // If we've reached our buffer limit, save the metadata immediately
if self.buffer.len() >= self.max_buffer_size { if self.buffer.len() >= self.max_buffer_size {
let _ = self.save_metadata(conn); let _ = self.save_metadata(conn);