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$(ZSTDDIR)/legacy \
30	-I$(PRGDIR) -DZSTD_MULTITHREAD -DZSTD_LEGACY_SUPPORT=1 $(CPPFLAGS)
31FUZZ_EXTRA_FLAGS := -Wall -Wextra -Wcast-qual -Wcast-align -Wshadow \
32	-Wstrict-aliasing=1 -Wswitch-enum -Wdeclaration-after-statement \
33	-Wstrict-prototypes -Wundef \
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 := -pthread $(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 := $(PRGDIR)/util.c zstd_helpers.c
45
46ZSTDCOMMON_SRC := $(ZSTDDIR)/common/*.c
47ZSTDCOMP_SRC   := $(ZSTDDIR)/compress/*.c
48ZSTDDECOMP_SRC := $(ZSTDDIR)/decompress/*.c
49ZSTDDICT_SRC := $(ZSTDDIR)/dictBuilder/*.c
50ZSTDLEGACY_SRC := $(ZSTDDIR)/legacy/*.c
51FUZZ_SRC       := \
52	$(FUZZ_SRC) \
53	$(ZSTDDECOMP_SRC) \
54	$(ZSTDCOMMON_SRC) \
55	$(ZSTDCOMP_SRC) \
56	$(ZSTDDICT_SRC) \
57	$(ZSTDLEGACY_SRC)
58
59FUZZ_OBJ := $(patsubst %.c,%.o, $(wildcard $(FUZZ_SRC)))
60
61
62.PHONY: default all clean cleanall
63
64default: all
65
66FUZZ_TARGETS :=       \
67	simple_round_trip \
68	stream_round_trip \
69	block_round_trip  \
70	simple_decompress \
71	stream_decompress \
72	block_decompress  \
73	dictionary_round_trip \
74	dictionary_decompress \
75	zstd_frame_info \
76	simple_compress
77
78all: $(FUZZ_TARGETS)
79
80%.o: %.c
81	$(CC) $(FUZZ_CPPFLAGS) $(FUZZ_CFLAGS) $^ -c -o $@
82
83simple_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_round_trip.o
84	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
85
86stream_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) stream_round_trip.o
87	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) stream_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
88
89block_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) block_round_trip.o
90	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) block_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
91
92simple_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_decompress.o
93	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_decompress.o $(LIB_FUZZING_ENGINE) -o $@
94
95stream_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) stream_decompress.o
96	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) stream_decompress.o $(LIB_FUZZING_ENGINE) -o $@
97
98block_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) block_decompress.o
99	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) block_decompress.o $(LIB_FUZZING_ENGINE) -o $@
100
101dictionary_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) dictionary_round_trip.o
102	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) dictionary_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
103
104dictionary_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) dictionary_decompress.o
105	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) dictionary_decompress.o $(LIB_FUZZING_ENGINE) -o $@
106
107simple_compress: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_compress.o
108	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_compress.o $(LIB_FUZZING_ENGINE) -o $@
109
110zstd_frame_info: $(FUZZ_HEADERS) $(FUZZ_OBJ) zstd_frame_info.o
111	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) zstd_frame_info.o $(LIB_FUZZING_ENGINE) -o $@
112
113libregression.a: $(FUZZ_HEADERS) $(PRGDIR)/util.h $(PRGDIR)/util.c regression_driver.o
114	$(AR) $(FUZZ_ARFLAGS) $@ regression_driver.o
115
116# Install libfuzzer (not usable for MSAN testing)
117# Provided for convenience. To use this library run make libFuzzer and
118# set LDFLAGS=-L.
119.PHONY: libFuzzer
120libFuzzer:
121	@$(RM) -rf Fuzzer
122	@git clone https://chromium.googlesource.com/chromium/llvm-project/compiler-rt/lib/fuzzer Fuzzer
123	@cd Fuzzer && ./build.sh
124
125corpora/%_seed_corpus.zip:
126	@mkdir -p corpora
127	$(DOWNLOAD) $@ $(CORPORA_URL_PREFIX)$*_seed_corpus.zip
128
129corpora/%: corpora/%_seed_corpus.zip
130	unzip -q $^ -d $@
131
132.PHONY: corpora
133corpora: $(patsubst %,corpora/%,$(FUZZ_TARGETS))
134
135.PHONY: seedcorpora
136seedcorpora: $(patsubst %,corpora/%_seed_corpus.zip,$(FUZZ_TARGETS))
137
138regressiontest: corpora
139	CC="$(CC)" CXX="$(CXX)" CFLAGS="$(CFLAGS)" CXXFLAGS="$(CXXFLAGS)" LDFLAGS="$(LDFLAGS)" $(PYTHON) ./fuzz.py build all
140	$(PYTHON) ./fuzz.py regression all
141
142clean:
143	@$(MAKE) -C $(ZSTDDIR) clean
144	@$(RM) *.a *.o
145	@$(RM) simple_round_trip stream_round_trip simple_decompress \
146           stream_decompress block_decompress block_round_trip \
147           simple_compress dictionary_round_trip dictionary_decompress \
148           zstd_frame_info
149
150cleanall:
151	@$(RM) -r Fuzzer
152	@$(RM) -r corpora
153