From 033b8b3a0da0128a62fc03af1bda55fe2bd255d0 Mon Sep 17 00:00:00 2001
From: ts <thomashaw@localhost.localdomain>
Date: Tue, 9 Oct 2018 12:42:04 +0100
Subject: [PATCH] huffman coding (WIP )

---
 .../encoders/compression/huffman/secgen_local/local.rb    | 8 ++++----
 .../examples/ctf_challenge_examples/huffman_challenge.xml | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/modules/encoders/compression/huffman/secgen_local/local.rb b/modules/encoders/compression/huffman/secgen_local/local.rb
index 20325db02..7ab62090b 100644
--- a/modules/encoders/compression/huffman/secgen_local/local.rb
+++ b/modules/encoders/compression/huffman/secgen_local/local.rb
@@ -13,12 +13,12 @@ class HuffmanEncoder < StringEncoder
     Dir.mkdir '../tmp/' unless Dir.exists? '../tmp/'
   end
 
-  def encode(str)
+  def encode_all
     tree_path = "../tmp/tree_#{index}"
-    result = Huffman.encode_text(str, tree_picture: true, tree_path: tree_path)
+    result = Huffman.encode_text(strings_to_encode[0], tree_picture: true, tree_path: tree_path)
     self.index += 1
-    # Return an array with the compressed string, and the tree png encoded in base64
-    [result.first, Base64.strict_encode64(File.binread("#{tree_path}.png"))]
+    self.outputs << result.first
+    self.outputs << Base64.strict_encode64(File.binread("#{tree_path}.png"))
   end
 end
 
diff --git a/scenarios/examples/ctf_challenge_examples/huffman_challenge.xml b/scenarios/examples/ctf_challenge_examples/huffman_challenge.xml
index acdf6da30..b420b2a84 100644
--- a/scenarios/examples/ctf_challenge_examples/huffman_challenge.xml
+++ b/scenarios/examples/ctf_challenge_examples/huffman_challenge.xml
@@ -21,7 +21,7 @@
 				<datastore access="0">huffman_challenge</datastore>
 			</input>
 			<input into="images_to_leak">
-				<datastore access="0">huffman_challenge</datastore>
+				<datastore access="1">huffman_challenge</datastore>
 			</input>
 		</vulnerability>