split code into functions

Author: frewsxcv

run.rb

@@ -19,33 +19,21 @@ def fuzz_function_exists?(file_path)
   reader.getc == '1'
 end
 
-if ARGV.length != 1
-  puts 'USAGE: fuzz.rb <FILE TO FUZZ>'
-  exit(1)
-end
-file_path = ARGV[0].freeze
-
-# TODO: ensure we don't load this file
-if !fuzz_function_exists?(file_path)
-  puts "ERROR: `fuzz` function doesn’t exist in #{file_path}"
-  exit(1)
-end
-
-reader, writer = IO.pipe
-
-fork do
-  seen = {}
-  loop do
-    encoded_bytes, cov_hash = reader.gets.strip.split('_')
+def start_reporting_process(reader)
+  fork do
+    seen = {}
+    loop do
+      encoded_bytes, cov_hash = reader.gets.strip.split('_')
 
-    if !seen.include?(cov_hash)
-      seen[cov_hash] = Base64.strict_decode64(encoded_bytes)
-      puts(seen)
+      if !seen.include?(cov_hash)
+        seen[cov_hash] = Base64.strict_decode64(encoded_bytes)
+        puts(seen)
+      end
     end
   end
 end
 
-loop do
+def start_fuzzing_process(writer)
   fork do
     STDOUT.reopen('/dev/null')
     STDIN.reopen('/dev/null')
@@ -64,5 +52,29 @@ def fuzz_function_exists?(file_path)
 
     writer.puts(Base64.strict_encode64(bytes) + '_' + Coverage.result.hash.to_s)
   end
-  Process.wait
 end
+
+def run
+  if ARGV.length != 1
+    puts 'USAGE: fuzz.rb <FILE TO FUZZ>'
+    exit(1)
+  end
+  file_path = ARGV[0].freeze
+
+  # TODO: ensure we don't load this file
+  if !fuzz_function_exists?(file_path)
+    puts "ERROR: `fuzz` function doesn’t exist in #{file_path}"
+    exit(1)
+  end
+
+  reader, writer = IO.pipe
+
+  start_reporting_process(reader)
+
+  loop do
+    start_fuzzing_process
+    Process.wait
+  end
+end
+
+run