fix(rust): properly check the BMI2 uleb128 (#17191)

pola-rs · Jun 25, 2024 · 332e40a · 332e40a
1 parent 7a1be3b
commit 332e40a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/crates/polars-parquet/src/parquet/encoding/uleb128.rs b/crates/polars-parquet/src/parquet/encoding/uleb128.rs
@@ -27,7 +27,10 @@ pub fn decode(values: &[u8]) -> (u64, usize) {
     #[cfg(target_feature = "bmi2")]
     {
         if polars_utils::cpuid::has_fast_bmi2() && values.len() >= 8 {
-            return unsafe { decode_uleb_bmi2(values) };
+            let (result, consumed) = unsafe { decode_uleb_bmi2(values) };
+            if consumed <= 8 {
+                return (result, consumed);
+            }
         }
     }