Fix underflow in search functions when input is empty

neosmart · Aug 1, 2021 · e2d5a8f · e2d5a8f
1 parent e9371f9
commit e2d5a8f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -307,7 +307,10 @@ fn reverse_file(path: &str, force_flush: bool) -> std::io::Result<()> {
                 use_avx2 = true;
             }
         }
-        if use_avx2 {
+        if bytes.len() == 0 {
+            // Do nothing. This avoids an underflow in the search functions which expect there to
+            // be at least one byte.
+        } else if use_avx2 {
             unsafe {
                 search256(bytes, &mut output)?;
             }