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 fuzz_data_producer.h
44FUZZ_SRC := $(PRGDIR)/util.c zstd_helpers.c fuzz_data_producer.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	dictionary_loader
78
79all: $(FUZZ_TARGETS)
80
81%.o: %.c
82	$(CC) $(FUZZ_CPPFLAGS) $(FUZZ_CFLAGS) $^ -c -o $@
83
84simple_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_round_trip.o
85	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
86
87stream_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) stream_round_trip.o
88	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) stream_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
89
90block_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) block_round_trip.o
91	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) block_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
92
93simple_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_decompress.o
94	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_decompress.o $(LIB_FUZZING_ENGINE) -o $@
95
96stream_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) stream_decompress.o
97	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) stream_decompress.o $(LIB_FUZZING_ENGINE) -o $@
98
99block_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) block_decompress.o
100	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) block_decompress.o $(LIB_FUZZING_ENGINE) -o $@
101
102dictionary_round_trip: $(FUZZ_HEADERS) $(FUZZ_OBJ) dictionary_round_trip.o
103	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) dictionary_round_trip.o $(LIB_FUZZING_ENGINE) -o $@
104
105dictionary_decompress: $(FUZZ_HEADERS) $(FUZZ_OBJ) dictionary_decompress.o
106	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) dictionary_decompress.o $(LIB_FUZZING_ENGINE) -o $@
107
108simple_compress: $(FUZZ_HEADERS) $(FUZZ_OBJ) simple_compress.o
109	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) simple_compress.o $(LIB_FUZZING_ENGINE) -o $@
110
111zstd_frame_info: $(FUZZ_HEADERS) $(FUZZ_OBJ) zstd_frame_info.o
112	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) zstd_frame_info.o $(LIB_FUZZING_ENGINE) -o $@
113
114dictionary_loader: $(FUZZ_HEADERS) $(FUZZ_OBJ) dictionary_loader.o
115	$(CXX) $(FUZZ_TARGET_FLAGS) $(FUZZ_OBJ) dictionary_loader.o $(LIB_FUZZING_ENGINE) -o $@
116
117libregression.a: $(FUZZ_HEADERS) $(PRGDIR)/util.h $(PRGDIR)/util.c regression_driver.o
118	$(AR) $(FUZZ_ARFLAGS) $@ regression_driver.o
119
120corpora/%_seed_corpus.zip:
121	@mkdir -p corpora
122	$(DOWNLOAD) $@ $(CORPORA_URL_PREFIX)$*_seed_corpus.zip
123
124corpora/%: corpora/%_seed_corpus.zip
125	unzip -q $^ -d $@
126
127.PHONY: corpora
128corpora: $(patsubst %,corpora/%,$(FUZZ_TARGETS))
129
130.PHONY: seedcorpora
131seedcorpora: $(patsubst %,corpora/%_seed_corpus.zip,$(FUZZ_TARGETS))
132
133regressiontest: corpora
134	CC="$(CC)" CXX="$(CXX)" CFLAGS="$(CFLAGS)" CXXFLAGS="$(CXXFLAGS)" LDFLAGS="$(LDFLAGS)" $(PYTHON) ./fuzz.py build all
135	$(PYTHON) ./fuzz.py regression all
136
137clean:
138	@$(MAKE) -C $(ZSTDDIR) clean
139	@$(RM) *.a *.o
140	@$(RM) simple_round_trip stream_round_trip simple_decompress \
141           stream_decompress block_decompress block_round_trip \
142           simple_compress dictionary_round_trip dictionary_decompress \
143           zstd_frame_info
144
145cleanall:
146	@$(RM) -r Fuzzer
147	@$(RM) -r corpora
148