From 3a1658d1db8558b19a933802251e88e6dfd905f7 Mon Sep 17 00:00:00 2001 From: Corey Farwell Date: Fri, 31 Aug 2018 21:35:46 -0500 Subject: [PATCH] error if there is no fuzz function --- run.rb | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/run.rb b/run.rb index 1d02de6..488732f 100644 --- a/run.rb +++ b/run.rb @@ -2,6 +2,24 @@ require 'coverage' +# Is the `fuzz` function defined in the fuzz target? +def fuzz_function_exists?(file_path) + reader, writer = IO.pipe + fork do + load(file_path) + begin + method(:fuzz) + rescue NameError + writer.puts('false') + else + writer.puts('true') + end + end + Process.wait + writer.close + reader.gets == "true\n" +end + if ARGV.length != 1 puts 'USAGE: fuzz.rb ' exit -- 2.51.2