Avoid potential overflow in vvd writer

Author: ftilde

palace-core/src/vec.rs

@@ -766,6 +766,14 @@ impl<D: DynDimension> Vector<D, u32> {
         self.iter().map(|v| *v as usize).sum()
     }
 }
+impl<D: DynDimension> Vector<D, usize> {
+    pub fn hmul(&self) -> usize {
+        self.iter().product()
+    }
+    pub fn hadd(&self) -> usize {
+        self.iter().sum()
+    }
+}
 
 //impl<D: Dimension, T: CoordinateType> Vector<D, Coordinate<T>> {
 //    pub fn to_ndarray_dim(self) -> D::NDArrayDim {

palace-vvd/src/lib.rs

@@ -351,8 +351,8 @@ pub async fn save_embedded_tensor<'cref, 'inv>(
                 .then_req(ctx, |(chunk_handle, chunk_id)| {
                     let chunk_info = md.chunk_info(chunk_id);
 
-                    let begin = chunk_info.begin();
-                    let start_offset = (*begin * stride).hadd();
+                    let begin = chunk_info.begin().map(|v| v.raw as usize);
+                    let start_offset = (begin * stride).hadd();
 
                     let start_ptr =
                         unsafe { out_file_ptr.offset(start_offset.try_into().unwrap()) };