From c0785fdbbb646691ac8fc919c2c6ea0e6f3e1acc Mon Sep 17 00:00:00 2001
From: Dobin <dobin.rutishauser@gmail.com>
Date: Mon, 19 Feb 2024 20:44:27 +0000
Subject: [PATCH] fix: also store payload

---
 phases/assembler.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/phases/assembler.py b/phases/assembler.py
index 06a7de6..f26f000 100644
--- a/phases/assembler.py
+++ b/phases/assembler.py
@@ -35,6 +35,7 @@ def merge_loader_payload(
 ):
     logger.info("--[ Merge stager with payload -> {}".format(
         shellcode_out))
+    observer.add_code("payload_shc", payload_data)
     
     with open(shellcode_in, 'rb') as input1:
         data_stager = input1.read()
@@ -55,3 +56,4 @@ def merge_loader_payload(
         data = data_stager + payload_data
         output.write(data)
         observer.add_code("loader_shc", data) 
+        
\ No newline at end of file