fix memory leaks in azalea-protocol

2 files changed, 15 insertions, 4 deletions

diff --git a/azalea-protocol/src/packets/game/c_merchant_offers.rs b/azalea-protocol/src/packets/game/c_merchant_offers.rs
index 7bbf92c4..957060b4 100644
--- a/azalea-protocol/src/packets/game/c_merchant_offers.rs
+++ b/azalea-protocol/src/packets/game/c_merchant_offers.rs
@@ -2,6 +2,7 @@ use std::{
     any::Any,
     fmt::{self, Debug},
     io::{self, Cursor, Write},
+    mem::ManuallyDrop,
 };
 
 use azalea_buf::{AzBuf, AzaleaRead, AzaleaWrite, BufReadError};
@@ -54,8 +55,13 @@ impl ItemCost {
     pub fn into_item_stack(self) -> ItemStackData {
         let mut component_patch = DataComponentPatch::default();
         for component in self.components.expected {
+            let component = ManuallyDrop::new(component);
+            // SAFETY: DataComponentUnion does not run any destructors unless it's dropped
+            // through drop_as, so since TypedDataComponent is now ManuallyDrop, the value
+            // will stay in memory.
+            let value = unsafe { std::ptr::read(&component.value) };
             unsafe {
-                component_patch.unchecked_insert_component(component.kind, Some(component.value));
+                component_patch.unchecked_insert_component(component.kind, Some(value));
             }
         }
         // TODO: add a fast way to iterate over default components, and insert the ones
@@ -129,6 +135,11 @@ impl Clone for TypedDataComponent {
         }
     }
 }
+impl Drop for TypedDataComponent {
+    fn drop(&mut self) {
+        unsafe { self.value.drop_as(self.kind) };
+    }
+}
 impl PartialEq for TypedDataComponent {
     fn eq(&self, other: &Self) -> bool {
         if self.kind != other.kind {
diff --git a/azalea-protocol/src/packets/game/c_update_advancements.rs b/azalea-protocol/src/packets/game/c_update_advancements.rs
index 62881b9d..24baebc5 100644
--- a/azalea-protocol/src/packets/game/c_update_advancements.rs
+++ b/azalea-protocol/src/packets/game/c_update_advancements.rs
@@ -22,7 +22,7 @@ pub struct ClientboundUpdateAdvancements {
 #[derive(AzBuf, Clone, Debug, PartialEq)]
 pub struct Advancement {
     pub parent_id: Option<Identifier>,
-    pub display: Option<DisplayInfo>,
+    pub display: Option<Box<DisplayInfo>>,
     pub requirements: Vec<Vec<String>>,
     pub sends_telemetry_event: bool,
 }
@@ -134,7 +134,7 @@ mod tests {
                 id: Identifier::new("minecraft:test"),
                 value: Advancement {
                     parent_id: None,
-                    display: Some(DisplayInfo {
+                    display: Some(Box::new(DisplayInfo {
                         title: FormattedText::from("title".to_owned()),
                         description: FormattedText::from("description".to_owned()),
                         icon: ItemStack::Empty,
@@ -144,7 +144,7 @@ mod tests {
                         background: None,
                         x: 0.0,
                         y: 0.0,
-                    }),
+                    })),
                     requirements: Vec::new(),
                     sends_telemetry_event: false,
                 },