Skip to content

Commit 0b85b0b

Browse files
akhikollaakhikolla
committed
honggfuzz & AFL run
1 parent 25fd2c1 commit 0b85b0b

File tree

3 files changed

+115
-74
lines changed

3 files changed

+115
-74
lines changed

‎R/deepstate_AFL.R

+21
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
##' @title creates afl fuzzer for given functions in package
2+
##' @export
3+
deepstate_make_afl<-function(){
4+
#insts.path <- system.file(package="RcppDeepState")
5+
insts.path <- "~"
6+
deepstate.path <- paste0(insts.path,"/.RcppDeepState")
7+
#dir.create(deepstate.path,showWarnings = FALSE)
8+
master <- file.path(deepstate.path,"deepstate-master")
9+
#afl <- file.path(deepstate.path,"AFL-master")
10+
system(paste0("cd ", insts.path, " ; "," wget http://lcamtuf.coredump.cx/afl/releases/afl-latest.tgz && tar -xzvf afl-latest.tgz && rm -rf afl-latest.tgz && cd afl-2.52b",";", "make"))
11+
build_afl <- file.path(master,"build_afl")
12+
dir.create(build_afl,showWarnings = FALSE)
13+
system("export AFL_HOME=\"~/afl-2.52b\"")
14+
#AFL_HOME <-"\"~/afl-2.52b\""
15+
system(paste0("cd ", build_afl," ; ","CXX=\"$AFL_HOME\afl-clang++\" CC=\"$AFL_HOME/afl-clang\" cmake -DDEEPSTATE_AFL=ON ../"," ; ", "make -j4"))
16+
}
17+
18+
19+
20+
21+

‎R/deepstate_compile_AFL.R

-58
This file was deleted.

‎R/deepstate_compile_tools.R

+94-16
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,98 @@
11
##' @title creates afl fuzzer for given functions in package
2+
##' @param path to the package to test
23
##' @export
3-
deepstate_make_afl<-function(){
4-
#insts.path <- system.file(package="RcppDeepState")
4+
deepstate_compile_tools<-function(path){
5+
option=readline(prompt="Please choose an option to select the Fuzzer:\n1 - AFL\n2 - LibFuzzer\n3 - Eclipser\n4 - Angora\n5 - HonggFuzz")
56
insts.path <- "~"
6-
deepstate.path <- paste0(insts.path,"/.RcppDeepState")
7-
#dir.create(deepstate.path,showWarnings = FALSE)
8-
master <- file.path(deepstate.path,"deepstate-master")
9-
#afl <- file.path(deepstate.path,"AFL-master")
10-
system(paste0("cd ", deepstate.path, " ; "," wget http://lcamtuf.coredump.cx/afl/releases/afl-latest.tgz && tar -xzvf afl-latest.tgz && rm -rf afl-latest.tgz && cd afl-2.52b",";", "make"))
11-
build_afl <- file.path(master,"build_afl")
12-
dir.create(build_afl,showWarnings = FALSE)
13-
system("export AFL_HOME=\"~/.RcppDeepState/afl-2.52b\"")
14-
system(paste0("cd ", build_afl," ; ","CXX=\"afl-clang++\" CC=\"$AFL_HOME/afl-clang\" cmake -DDEEPSTATE_AFL=ON ../"," ; ", "make -j4"))
7+
inst_path <- file.path(path, "inst")
8+
test_path <- file.path(inst_path,"testfiles")
9+
if((file.exists("~/.RcppDeepState/deepstate-master/build/libdeepstate32.a") &&
10+
file.exists("~/.RcppDeepState/deepstate-master/build/libdeepstate.a"))){
11+
RcppDeepState::deepstate_make_run()
12+
}
13+
if(option == 1){
14+
deepstate.path <- paste0(insts.path,"/.RcppDeepState")
15+
if(!file.exists("~/.RcppDeepState/deepstate-master/build_afl/libdeepstate_AFL.a")){
16+
deepstate_make_afl()
17+
}
18+
inst_path <- file.path(path, "inst")
19+
test_path <- file.path(inst_path,"testfiles")
20+
functions.list <- RcppDeepState::deepstate_get_function_body(path)
21+
fun_names <- unique(functions.list$funName)
22+
for(f in fun_names){
23+
function.path <- file.path(test_path,f)
24+
harness.path <- file.path(function.path,paste0(f,"_DeepState_TestHarness.cpp"))
25+
makefile.path <- file.path(function.path,"Makefile")
26+
if(file.exists(harness.path) && file.exists(makefile.path) ){
27+
executable <- gsub(".cpp$","",harness.path)
28+
object <- gsub(".cpp$",".o",harness.path)
29+
o.logfile <- paste0(function.path,"/",f,"_log")
30+
logfile <- paste0(function.path,"/afl_",f,"_log")
31+
output_dir <- paste0(function.path,"/afl_",f,"_output")
32+
makefile_lines <- readLines(makefile.path,warn=FALSE)
33+
makefile_lines <- gsub(paste0("clang++ -g -o",executable),paste0("clang++ -g -o",executable,".afl"),makefile_lines,fixed=TRUE)
34+
makefile_lines <- gsub("clang++","$(CXX)",makefile_lines,fixed=TRUE)
35+
makefile_lines <- gsub("-ldeepstate","-ldeepstate_AFL",makefile_lines,fixed=TRUE)
36+
makefile_lines <- gsub("deepstate-master/build","deepstate-master/build_afl",makefile_lines,fixed=TRUE)
37+
makefile_lines <- gsub("R_HOME=","export AFL_HOME=~/afl-2.52b\nCXX = ${AFL_HOME}/afl-clang++\nAFL_FUZZ=${AFL_HOME}/afl-fuzz\nR_HOME=",makefile_lines,fixed=TRUE)
38+
makefile_lines <- gsub(o.logfile,logfile,makefile_lines,fixed=TRUE)
39+
makefile_lines <- gsub(executable,paste0(executable,".afl"),makefile_lines,fixed=TRUE)
40+
makefile_lines <- gsub(paste0("./",basename(executable)," --fuzz"),paste0("${AFL_HOME}/afl-fuzz -o ",output_dir," -m 150 -t 2000 -i ~/.RcppDeepState/deepstate-master/build_afl/ -- ",executable,".afl"),makefile_lines,fixed=TRUE)
41+
makefile_lines <- gsub("--output_test_dir.*> /dev/null","",makefile_lines)
42+
makefile_lines <- gsub(paste0("./",executable),paste0("./",executable,".afl"),makefile_lines,fixed=TRUE)
43+
makefile.afl <- file.path(dirname(makefile.path),"AFL.Makefile")
44+
file.create(makefile.afl,recursive=TRUE)
45+
cat(makefile_lines, file=makefile.afl, sep="\n")
46+
file.remove(object)
47+
file.remove(executable)
48+
compile_line <-paste0("rm -f *.o && make -f ",makefile.afl)
49+
print(compile_line)
50+
system(compile_line)
51+
}
52+
}
53+
}else if(option == 2){
54+
print("LibFuzzer")
55+
}else if(option == 3){
56+
print("Eclipser")
57+
}else if(option == 4){
58+
print("Angora")
59+
}else if(option == 5){
60+
deepstate.path <- paste0(insts.path,"/.RcppDeepState")
61+
if(!file.exists("~/.RcppDeepState/deepstate-master/build_afl/libdeepstate_HFUZZ.a")){
62+
deepstate_make__hongg()
63+
}
64+
functions.list <- RcppDeepState::deepstate_get_function_body(path)
65+
fun_names <- unique(functions.list$funName)
66+
for(f in fun_names){
67+
function.path <- file.path(test_path,f)
68+
harness.path <- file.path(function.path,paste0(f,"_DeepState_TestHarness.cpp"))
69+
makefile.path <- file.path(function.path,"Makefile")
70+
if(file.exists(harness.path) && file.exists(makefile.path) ){
71+
executable <- gsub(".cpp$","",harness.path)
72+
object <- gsub(".cpp$",".o",harness.path)
73+
o.logfile <- paste0(function.path,"/",f,"_log")
74+
logfile <- paste0(function.path,"/hongg_",f,"_log")
75+
output_dir <- paste0(function.path,"/hongg_",f,"_output")
76+
makefile_lines <- readLines(makefile.path,warn=FALSE)
77+
makefile_lines <- gsub(paste0("clang++ -g -o",executable),paste0("clang++ -g -o",executable,".hongg"),makefile_lines,fixed=TRUE)
78+
makefile_lines <- gsub("clang++","$(CXX)",makefile_lines,fixed=TRUE)
79+
makefile_lines <- gsub("-ldeepstate","-ldeepstate_HFUZZ",makefile_lines,fixed=TRUE)
80+
makefile_lines <- gsub("deepstate-master/build","deepstate-master/build_honggfuzz",makefile_lines,fixed=TRUE)
81+
makefile_lines <- gsub("R_HOME=","export HONGGFUZZ_HOME=~/honggfuzz\nCXX = ${HONGGFUZZ_HOME}/hfuzz_cc/hfuzz-clang++\nHONGG_FUZZ=${HONGGFUZZ_HOME}/hfuzz-8bitcnt-clang++ \nR_HOME=",makefile_lines,fixed=TRUE)
82+
makefile_lines <- gsub(o.logfile,logfile,makefile_lines,fixed=TRUE)
83+
makefile_lines <- gsub(executable,paste0(executable,".hongg"),makefile_lines,fixed=TRUE)
84+
makefile_lines <- gsub(paste0("./",basename(executable)," --fuzz"),paste0("${HONGGFUZZ_HOME}/hfuzz-8bitcnt-clang++ -o ",output_dir," -m 150 -t 2000 -i ~/.RcppDeepState/deepstate-master/build_hongg/ -- ",executable,".hongg"),makefile_lines,fixed=TRUE)
85+
makefile_lines <- gsub("--output_test_dir.*> /dev/null","",makefile_lines)
86+
makefile_lines <- gsub(paste0("./",executable),paste0("./",executable,".hongg"),makefile_lines,fixed=TRUE)
87+
makefile.hongg <- file.path(dirname(makefile.path),"Hongg.Makefile")
88+
file.create(makefile.hongg,recursive=TRUE)
89+
cat(makefile_lines, file=makefile.hongg, sep="\n")
90+
file.remove(object)
91+
file.remove(executable)
92+
compile_line <-paste0("rm -f *.o && make -f ",makefile.hongg)
93+
#print(compile_line)
94+
system(compile_line)
95+
}
96+
}
97+
}
1598
}
16-
17-
18-
19-
20-

0 commit comments

Comments
 (0)