fix: audit sqlx_postgres::types::array for bad casts

launchbadge · Aug 17, 2024 · 2cb6217 · 2cb6217
1 parent 8a17e46
commit 2cb6217
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/sqlx-postgres/src/types/array.rs b/sqlx-postgres/src/types/array.rs
@@ -174,7 +174,14 @@ where
             }
         }
 
-        buf.extend(&(self.len() as i32).to_be_bytes()); // len
+        let array_len = i32::try_from(self.len()).map_err(|_| {
+            format!(
+                "encoded array length is too large for Postgres: {}",
+                self.len()
+            )
+        })?;
+
+        buf.extend(array_len.to_be_bytes()); // len
         buf.extend(&1_i32.to_be_bytes()); // lower bound
 
         for element in self.iter() {