1# ################################################################
2# Copyright (c) 2016-present, Facebook, Inc.
3# All rights reserved.
4#
5# This source code is licensed under both the BSD-style license (found in the
6# LICENSE file in the root directory of this source tree) and the GPLv2 (found
7# in the COPYING file in the root directory of this source tree).
8# ################################################################
9
10# Optionally user defined flags
11CFLAGS ?= -O3
12CXXFLAGS ?= -O3
13CPPFLAGS ?=
14LDFLAGS ?=
15ARFLAGS ?=
16LIB_FUZZING_ENGINE ?= libregression.a
17PYTHON ?= python
18ifeq ($(shell uname), Darwin)
19	DOWNLOAD?=curl -L -o
20else
21	DOWNLOAD?=wget -O
22endif
23CORPORA_URL_PREFIX:=https://github.com/facebook/zstd/releases/download/fuzz-corpora/
24
25ZSTDDIR = ../../lib
26PRGDIR = ../../programs
27
28FUZZ_CPPFLAGS := -I$(ZSTDDIR) -I$(ZSTDDIR)/common -I$(ZSTDDIR)/compress \
29	-I$(ZSTDDIR)/dictBuilder -I$(ZSTDDIR)/deprecated -I$(PRGDIR) \
30	$(CPPFLAGS)
31FUZZ_EXTRA_FLAGS := -Wall -Wextra -Wcast-qual -Wcast-align -Wshadow \
32	-Wstrict-aliasing=1 -Wswitch-enum -Wdeclaration-after-statement \
33	-Wstrict-prototypes -Wundef -Wformat-security \
34	-Wvla -Wformat=2 -Winit-self -Wfloat-equal -Wwrite-strings \
35	-Wredundant-decls \
36	-g -fno-omit-frame-pointer
37FUZZ_CFLAGS := $(FUZZ_EXTRA_FLAGS) $(CFLAGS)
38FUZZ_CXXFLAGS := $(FUZZ_EXTRA_FLAGS) -std=c++11 $(CXXFLAGS)
39FUZZ_LDFLAGS := $(LDFLAGS)
40FUZZ_ARFLAGS := $(ARFLAGS)
41FUZZ_TARGET_FLAGS = $(FUZZ_CPPFLAGS) $(FUZZ_CXXFLAGS) $(FUZZ_LDFLAGS)
42
43FUZZ_HEADERS := fuzz_helpers.h fuzz.h zstd_helpers.h
44FUZZ_SRC := zstd_helpers.c
45
46ZSTDCOMMON_SRC := $(ZSTDDIR)/common/*.c
47ZSTDCOMP_SRC   := $(ZSTDDIR)/compress/*.c
48ZSTDDECOMP_SRC := $(ZSTDDIR)/decompress/*.c
49FUZZ_SRC       := \
50	$(FUZZ_SRC) \
51	$(ZSTDDECOMP_SRC) \
52	$(ZSTDCOMMON_SRC) \
53	$(ZSTDCOMP_SRC)
54
55FUZZ_OBJ := $(patsubst %.c,%.o, $(wildcard $(FUZZ_SRC)))
56
57
58.PHONY: default all clean cleanall
59
60default: all
61
62FUZZ_TARGETS :=       \
63	simple_round_trip \
64	stream_round_trip \
65	block_round_trip  \
66	simple_decompress \
67	stream_decompress \
68	block_decompress
69
70all: $(FUZZ_TARGETS)
71
72%.o: %.c
73	$(CC) $(FUZZ_CPPFLAGS) $(FUZZ_CFLAGS) $^ -c -o $@
74
75simple_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_round_trip.o
76	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
77
78stream_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) stream_round_trip.o
79	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) stream_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
80
81block_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) block_round_trip.o
82	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) block_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
83
84simple_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_decompress.o
85	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_decompress.o $(LIB_FUZZING_ENGINE) -o $@
86
87stream_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) stream_decompress.o
88	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) stream_decompress.o $(LIB_FUZZING_ENGINE) -o $@
89
90block_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) block_decompress.o
91	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) block_decompress.o $(LIB_FUZZING_ENGINE) -o $@
92
93libregression.a: $(FUZZ_HEADERS) $(PRGDIR)/util.h regression_driver.o
94	$(AR) $(FUZZ_ARFLAGS) $@ regression_driver.o
95
96# Install libfuzzer (not usable for MSAN testing)
97# Provided for convienence. To use this library run make libFuzzer and
98# set LDFLAGS=-L.
99.PHONY: libFuzzer
100libFuzzer:
101	@$(RM) -rf Fuzzer
102	@git clone https://chromium.googlesource.com/chromium/llvm-project/llvm/lib/Fuzzer
103	@cd Fuzzer && ./build.sh
104
105corpora/%_seed_corpus.zip:
106	@mkdir -p corpora
107	$(DOWNLOAD) $@ $(CORPORA_URL_PREFIX)$*_seed_corpus.zip
108
109corpora/%: corpora/%_seed_corpus.zip
110	unzip -q $^ -d $@
111
112.PHONY: corpora
113corpora: $(patsubst %,corpora/%,$(FUZZ_TARGETS))
114
115regressiontest: corpora
116	CC="$(CC)" CXX="$(CXX)" CFLAGS="$(CFLAGS)" CXXFLAGS="$(CXXFLAGS)" LDFLAGS="$(LDFLAGS)" $(PYTHON) ./fuzz.py build all
117	$(PYTHON) ./fuzz.py regression all
118
119clean:
120	@$(MAKE) -C $(ZSTDDIR) clean
121	@$(RM) *.a *.o
122	@$(RM) simple_round_trip stream_round_trip simple_decompress \
123           stream_decompress block_decompress block_round_trip
124
125cleanall:
126	@$(RM) -r Fuzzer
127	@$(RM) -r corpora
128