fix: do not leave uninitialized bytes in datatype input function

Signed-off-by: usamoi <[email protected]>

Author: usamoi

src/datatype/vecf16.rs

@@ -40,6 +40,7 @@ CREATE TYPE vecf16 (
 pub struct Vecf16 {
     varlena: u32,
     kind: u8,
+    pad: u8,
     len: u16,
     phantom: [F16; 0],
 }
@@ -62,6 +63,7 @@ impl Vecf16 {
             let ptr = std::alloc::alloc(layout) as *mut Vecf16;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf16::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(16);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(slice.len() as u16);
             std::ptr::copy_nonoverlapping(slice.as_ptr(), (*ptr).phantom.as_mut_ptr(), slice.len());
             Box::from_raw(ptr)
@@ -74,6 +76,7 @@ impl Vecf16 {
             let ptr = pgrx::pg_sys::palloc(layout.size()) as *mut Vecf16;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf16::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(16);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(slice.len() as u16);
             std::ptr::copy_nonoverlapping(slice.as_ptr(), (*ptr).phantom.as_mut_ptr(), slice.len());
             Vecf16Output(NonNull::new(ptr).unwrap())
@@ -86,6 +89,7 @@ impl Vecf16 {
             let ptr = std::alloc::alloc_zeroed(layout) as *mut Vecf16;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf16::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(16);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(len as u16);
             Box::from_raw(ptr)
         }
@@ -98,6 +102,7 @@ impl Vecf16 {
             let ptr = pgrx::pg_sys::palloc0(layout.size()) as *mut Vecf16;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf16::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(16);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(len as u16);
             Vecf16Output(NonNull::new(ptr).unwrap())
         }

src/datatype/vecf32.rs

@@ -40,6 +40,7 @@ CREATE TYPE vector (
 pub struct Vecf32 {
     varlena: u32,
     kind: u8,
+    pad: u8,
     len: u16,
     phantom: [F32; 0],
 }
@@ -62,6 +63,7 @@ impl Vecf32 {
             let ptr = std::alloc::alloc(layout) as *mut Vecf32;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf32::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(32);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(slice.len() as u16);
             std::ptr::copy_nonoverlapping(slice.as_ptr(), (*ptr).phantom.as_mut_ptr(), slice.len());
             Box::from_raw(ptr)
@@ -74,6 +76,7 @@ impl Vecf32 {
             let ptr = pgrx::pg_sys::palloc(layout.size()) as *mut Vecf32;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf32::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(32);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(slice.len() as u16);
             std::ptr::copy_nonoverlapping(slice.as_ptr(), (*ptr).phantom.as_mut_ptr(), slice.len());
             Vecf32Output(NonNull::new(ptr).unwrap())
@@ -86,6 +89,7 @@ impl Vecf32 {
             let ptr = std::alloc::alloc_zeroed(layout) as *mut Vecf32;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf32::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(32);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(len as u16);
             Box::from_raw(ptr)
         }
@@ -98,6 +102,7 @@ impl Vecf32 {
             let ptr = pgrx::pg_sys::palloc0(layout.size()) as *mut Vecf32;
             std::ptr::addr_of_mut!((*ptr).varlena).write(Vecf32::varlena(layout.size()));
             std::ptr::addr_of_mut!((*ptr).kind).write(32);
+            std::ptr::addr_of_mut!((*ptr).pad).write(0);
             std::ptr::addr_of_mut!((*ptr).len).write(len as u16);
             Vecf32Output(NonNull::new(ptr).unwrap())
         }

src/index/utils.rs

@@ -6,6 +6,7 @@ use service::prelude::DynamicVector;
 struct Header {
     varlena: u32,
     kind: u8,
+    pad: u8,
     len: u16,
 }