chore: fix core dump error in AsyncInlinable::read_inlined

Author: zitsen

taos-query/src/common/raw/data.rs

@@ -126,14 +126,20 @@ impl crate::util::AsyncInlinable for RawData {
         reader: &mut R,
     ) -> std::io::Result<Self> {
         use tokio::io::*;
+
         let len = reader.read_u32_le().await?;
         let meta_type = reader.read_u16_le().await?;
-        let mut vec: Vec<u8> = Vec::with_capacity(len as usize);
-        reader.read_exact(&mut vec).await?;
+        let layout = std::alloc::Layout::from_size_align(len as _, 1).map_err(|_| {
+            std::io::Error::new(std::io::ErrorKind::InvalidData, "Invalid raw data length")
+        })?;
+        let ptr = unsafe { std::alloc::alloc(layout) };
+        let buf = unsafe { std::slice::from_raw_parts_mut(ptr, len as _) };
+
+        // let mut vec: Vec<u8> = Vec::with_capacity(len as usize);
+        reader.read_exact(buf).await?;
 
-        let ptr = Box::into_raw(vec.into_boxed_slice());
         let raw = raw_data_t {
-            raw: ptr as _,
+            raw: buf.as_ptr() as _,
             raw_len: len,
             raw_type: meta_type,
         };

taos/examples/raw-restore.rs

@@ -3,6 +3,7 @@ use std::time::Instant;
 
 use clap::Parser;
 use taos::*;
+use taos_query::common::RawData;
 
 #[derive(Debug, Parser)]
 struct Opts {
@@ -96,18 +97,15 @@ async fn main() -> anyhow::Result<()> {
             std::io::stdin().read_line(&mut String::new())?;
         }
         let mut file = std::fs::File::open(&path)?;
-        let mut buf = Vec::new();
-        let size = buf.len();
-        file.read_to_end(&mut buf)?;
-        let raw = RawMeta::new(bytes::Bytes::from(buf));
+        let raw: RawMeta = Inlinable::read_inlined(&mut file)?;
         let instant = Instant::now();
         conn.write_raw_meta(&raw).await?;
         eprintln!("{}: write raw data from {:?} done", idx, path);
         println!(
             "{},{},{},{}",
             idx,
             path.display(),
-            size,
+            raw.raw_len(),
             instant.elapsed().as_millis()
         );
         idx += 1;
@@ -153,10 +151,8 @@ async fn main() -> anyhow::Result<()> {
                 std::io::stdin().read_line(&mut String::new())?;
             }
             let mut file = std::fs::File::open(path)?;
-            let mut buf = Vec::new();
-            file.read_to_end(&mut buf)?;
-            let size = buf.len();
-            let raw = RawMeta::new(bytes::Bytes::from(buf));
+            let raw: RawData = Inlinable::read_inlined(&mut file)?;
+            let size = raw.raw_len() as usize + 6;
             senders[idx % opts.workers]
                 .send((idx, path.clone(), raw, size))
                 .await?;

taos/examples/ts5250.rs

@@ -1,5 +1,3 @@
-use std::io::Read;
-
 use clap::Parser;
 use taos::*;
 
@@ -51,9 +49,8 @@ async fn main() -> anyhow::Result<()> {
         println!("{}: press enter to write raw data from {:?}", idx, path);
         std::io::stdin().read_line(&mut String::new())?;
         let mut file = std::fs::File::open(&path)?;
-        let mut buf = Vec::new();
-        file.read_to_end(&mut buf)?;
-        let raw = RawMeta::new(bytes::Bytes::from(buf));
+
+        let raw: RawMeta = Inlinable::read_inlined(&mut file)?;
         conn.write_raw_meta(&raw).await?;
         println!("{}: write raw data from {:?} done", idx, path);
     }