[trace-dump] Fix memory trace file writing for mem_profile (#1051)

- The trace-dump utility tool expects only a pointer when a free
operation is reported.

Signed-off-by: Doru Blânzeanu <[email protected]>

Author: dblnz

src/hyperlight_host/src/sandbox/trace/mem_profile.rs

@@ -155,11 +155,22 @@ impl MemTraceInfo {
         let amt = regs.rax;
         let ptr = regs.rcx;
 
-        self.record_trace_frame(self.epoch, trace_identifier as u64, |f| {
-            let _ = f.write_all(&ptr.to_ne_bytes());
-            let _ = f.write_all(&amt.to_ne_bytes());
-            self.write_stack(f, &stack);
-        })
+        match trace_identifier {
+            TraceFrameType::MemAlloc => {
+                self.record_trace_frame(self.epoch, trace_identifier as u64, |f| {
+                    let _ = f.write_all(&ptr.to_ne_bytes());
+                    let _ = f.write_all(&amt.to_ne_bytes());
+                    self.write_stack(f, &stack);
+                })
+            }
+            // The MemFree case does not expect an amount, only a pointer
+            TraceFrameType::MemFree => {
+                self.record_trace_frame(self.epoch, trace_identifier as u64, |f| {
+                    let _ = f.write_all(&ptr.to_ne_bytes());
+                    self.write_stack(f, &stack);
+                })
+            }
+        }
     }
 
     #[inline(always)]