Merge pull request #4 from a2x/main

Prevent potential integer overflow exception
memflow · Mar 30, 2024 · 7fa06a0 · 7fa06a0
2 parents 5b86dbc + e1a3364
commit 7fa06a0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/linux/mem.rs b/src/linux/mem.rs
@@ -165,7 +165,11 @@ impl ProcessVirtualMemory {
                     match vm_err {
                         Some(err) => return Err(Error(ErrorOrigin::OsLayer, err)),
                         _ => {
-                            let mut remaining_written = libcret as usize + 1;
+                            let mut remaining_written = if libcret == -1 {
+                                0
+                            } else {
+                                libcret as usize + 1
+                            };
 
                             for (liof, (_, meta)) in iov_local
                                 .iter()