From 0016ad7034450bc4b25af1b9f18f69f87beedcc4 Mon Sep 17 00:00:00 2001
From: Anh <zenpoems@gmail.com>
Date: Thu, 13 Sep 2018 00:58:15 +0000
Subject: [PATCH] Fix bug in zafl.sh, more blacklisted funcs

---
 afl_transforms/tools/zafl/zafl.cpp | 6 ++++++
 bin/zafl.sh                        | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/afl_transforms/tools/zafl/zafl.cpp b/afl_transforms/tools/zafl/zafl.cpp
index a1510c2..2b842da 100644
--- a/afl_transforms/tools/zafl/zafl.cpp
+++ b/afl_transforms/tools/zafl/zafl.cpp
@@ -81,6 +81,12 @@ Zafl_t::Zafl_t(libIRDB::pqxxDB_t &p_dbinterface, libIRDB::FileIR_t *p_variantIR,
 	m_blacklist.insert("__do_global_dtors_aux");
 	m_blacklist.insert("__libc_csu_init");
 	m_blacklist.insert("__libc_csu_fini");
+	m_blacklist.insert("start");
+	m_blacklist.insert("__libc_start_main");
+	m_blacklist.insert("__gmon_start__");
+	m_blacklist.insert("__cxa_atexit");
+	m_blacklist.insert("__cxa_finalize");
+	m_blacklist.insert("__assert_fail");
 
 	m_num_flags_saved = 0;
 	m_num_temp_reg_saved = 0;
diff --git a/bin/zafl.sh b/bin/zafl.sh
index 88bbeb3..c49d6cd 100755
--- a/bin/zafl.sh
+++ b/bin/zafl.sh
@@ -108,7 +108,7 @@ fi
 rm $tmp_objdump
 
 echo "Zafl: Transforming input binary $input_binary into $output_zafl_binary"
-cmd="$PSZ $input_binary $output_zafl_binary $ida_or_rida_opt -c move_globals=on -c zafl=on -o move_globals:--elftables -o zipr:--traceplacement:on $stars_opt $options $*"
+cmd="$PSZ $input_binary $output_zafl_binary $ida_or_rida_opt -c move_globals=on -c zafl=on -o move_globals:--elftables-only -o zipr:--traceplacement:on $stars_opt $options $other_args"
 echo "Zafl: Issuing command: $cmd"
 eval $cmd
 if [ $? -eq 0 ]; then
-- 
GitLab