From 5edf83b22fe772507dbc391cf0700631d2bcf72f Mon Sep 17 00:00:00 2001
From: JD <j.davis2344@student.leedsbeckett.ac.uk>
Date: Mon, 24 Jul 2023 11:26:07 +0100
Subject: [PATCH] Fix iterations issue

---
 lib/objects/local_string_encoder.rb | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/lib/objects/local_string_encoder.rb b/lib/objects/local_string_encoder.rb
index 8702631a9..3d9650054 100644
--- a/lib/objects/local_string_encoder.rb
+++ b/lib/objects/local_string_encoder.rb
@@ -17,6 +17,7 @@ class StringEncoder
   attr_accessor :strings_to_encode
   attr_accessor :has_base64_inputs
   attr_accessor :outputs
+  attr_accessor :iterations
 
   # override this
   def initialize
@@ -24,6 +25,7 @@ class StringEncoder
     self.strings_to_encode = []
     self.module_name = 'Null encoder'
     self.has_base64_inputs = false
+    self.iterations = 1
     self.outputs = []
   end
 
@@ -79,7 +81,8 @@ class StringEncoder
   def get_options_array
     [['--help', '-h', GetoptLong::NO_ARGUMENT],
      ['--b64', GetoptLong::OPTIONAL_ARGUMENT],
-     ['--strings_to_encode', '-s', GetoptLong::OPTIONAL_ARGUMENT]]
+     ['--strings_to_encode', '-s', GetoptLong::OPTIONAL_ARGUMENT],
+     ['--iterations', GetoptLong::OPTIONAL_ARGUMENT],]
   end
 
   # Override this when using read_fact's in your module. Always call super first.
@@ -97,6 +100,12 @@ class StringEncoder
         self.strings_to_encode << arg;
       when '--b64'
         # do nothing
+      when '--iterations'
+        if not arg.to_i == 0
+          self.iterations = arg.to_i
+        else
+          self.iterations = 1
+        end
     end
   end