CC ?= cc

CPPFLAGS += -DPGLC_RESP_STANDALONE -I../../src
COMMON_CFLAGS = -std=c11 -O2 -g -Wall -Wextra -Werror -Wpedantic
PG_CONFIG ?= pg_config
PG_SERVER_INCLUDE = $(shell $(PG_CONFIG) --includedir-server)

# PostgreSQL server headers require the same feature-test macros as server code.
PG_SERVER_CPPFLAGS = $(shell $(PG_CONFIG) --cppflags)
ROW_PAYLOAD_CPPFLAGS = $(PG_SERVER_CPPFLAGS) -DPGLC_BUILD_ID='"unit-test"' \
	-I$(PG_SERVER_INCLUDE) -I../../src
ROW_PAYLOAD_CFLAGS = -std=c11 -O2 -g -Wall -Wextra -Werror \
	-Wno-unused-parameter
SOURCES = resp_test.c ../../src/resp.c
FUZZ_SOURCE = ../../fuzz/resp_parse_fuzzer.c
FUZZ_CORPUS = ../../fuzz/corpus/resp_parse
ROW_PAYLOAD_SOURCES = row_payload_test.c ../../src/row_payload.c
CACHE_STORAGE_SOURCES = cache_arena_test.c ../../src/cache_arena.c ../../src/cache_index.c
CACHE_INDEX_SOURCES = cache_index_test.c ../../src/cache_index.c
CACHE_CODEC_SOURCES = key_codec_test.c ../../src/key_codec.c
DIRTY_KEY_LIMIT_SOURCES = dirty_key_limit_test.c
TARGET = resp_test
SANITIZED_TARGET = resp_test_sanitized
TARGETS = $(TARGET) row_payload_test cache_arena_test cache_index_test key_codec_test dirty_key_limit_test
SANITIZED_TARGETS = $(SANITIZED_TARGET) row_payload_test_sanitized \
	cache_arena_test_sanitized cache_index_test_sanitized key_codec_test_sanitized \
	dirty_key_limit_test_sanitized
FUZZ_REPLAY_TARGET = resp_parse_replay
FUZZ_REPLAY_SANITIZED_TARGET = resp_parse_replay_sanitized

.PHONY: all check sanitize clean

all: $(TARGETS)

$(TARGET): $(SOURCES) ../../src/resp.h ../../src/resp_limits.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) $(CFLAGS) $(SOURCES) $(LDFLAGS) -o $@

row_payload_test: $(ROW_PAYLOAD_SOURCES) ../../src/row_payload.h ../../src/pg_local_cache.h
	$(CC) $(ROW_PAYLOAD_CPPFLAGS) $(ROW_PAYLOAD_CFLAGS) $(CFLAGS) \
		$(ROW_PAYLOAD_SOURCES) $(LDFLAGS) -o $@

cache_arena_test: $(CACHE_STORAGE_SOURCES) ../../src/cache_arena.h ../../src/cache_index.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) $(CFLAGS) $(CACHE_STORAGE_SOURCES) $(LDFLAGS) -o $@

cache_arena_test_sanitized: $(CACHE_STORAGE_SOURCES) ../../src/cache_arena.h ../../src/cache_index.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) -O1 -fsanitize=address,undefined \
		-fno-omit-frame-pointer $(CFLAGS) $(CACHE_STORAGE_SOURCES) $(LDFLAGS) \
		-fsanitize=address,undefined -o $@

cache_index_test: $(CACHE_INDEX_SOURCES) ../../src/cache_index.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) $(CFLAGS) $(CACHE_INDEX_SOURCES) $(LDFLAGS) -o $@

key_codec_test: $(CACHE_CODEC_SOURCES) ../../src/key_codec.h
	$(CC) $(CPPFLAGS) -DPGLC_KEY_CODEC_STANDALONE $(COMMON_CFLAGS) $(CFLAGS) \
		$(CACHE_CODEC_SOURCES) $(LDFLAGS) -o $@

dirty_key_limit_test: $(DIRTY_KEY_LIMIT_SOURCES) ../../src/dirty_key_limit.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) $(CFLAGS) \
		$(DIRTY_KEY_LIMIT_SOURCES) $(LDFLAGS) -o $@

dirty_key_limit_test_sanitized: $(DIRTY_KEY_LIMIT_SOURCES) ../../src/dirty_key_limit.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) -O1 -fsanitize=address,undefined \
		-fno-omit-frame-pointer $(CFLAGS) $(DIRTY_KEY_LIMIT_SOURCES) $(LDFLAGS) \
		-fsanitize=address,undefined -o $@

key_codec_test_sanitized: $(CACHE_CODEC_SOURCES) ../../src/key_codec.h
	$(CC) $(CPPFLAGS) -DPGLC_KEY_CODEC_STANDALONE $(COMMON_CFLAGS) -O1 \
		-fsanitize=address,undefined -fno-omit-frame-pointer $(CFLAGS) \
		$(CACHE_CODEC_SOURCES) $(LDFLAGS) -fsanitize=address,undefined -o $@

cache_index_test_sanitized: $(CACHE_INDEX_SOURCES) ../../src/cache_index.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) -O1 -fsanitize=address,undefined \
		-fno-omit-frame-pointer $(CFLAGS) $(CACHE_INDEX_SOURCES) $(LDFLAGS) \
		-fsanitize=address,undefined -o $@

$(SANITIZED_TARGET): $(SOURCES) ../../src/resp.h ../../src/resp_limits.h
	$(CC) $(CPPFLAGS) $(COMMON_CFLAGS) -O1 \
		-fsanitize=address,undefined -fno-omit-frame-pointer \
		$(CFLAGS) $(SOURCES) $(LDFLAGS) \
		-fsanitize=address,undefined -o $@

row_payload_test_sanitized: $(ROW_PAYLOAD_SOURCES) ../../src/row_payload.h ../../src/pg_local_cache.h
	$(CC) $(ROW_PAYLOAD_CPPFLAGS) $(ROW_PAYLOAD_CFLAGS) -O1 \
		-fsanitize=address,undefined -fno-omit-frame-pointer $(CFLAGS) \
		$(ROW_PAYLOAD_SOURCES) $(LDFLAGS) \
		-fsanitize=address,undefined -o $@

$(FUZZ_REPLAY_TARGET): $(FUZZ_SOURCE) ../../src/resp.c ../../src/resp.h ../../src/resp_limits.h ../../src/key_codec.c ../../src/key_codec.h
	$(CC) $(CPPFLAGS) -DPGLC_KEY_CODEC_STANDALONE $(COMMON_CFLAGS) \
		-DPGLC_RESP_FUZZ_REPLAY $(FUZZ_SOURCE) ../../src/resp.c \
		../../src/key_codec.c $(LDFLAGS) -o $@

$(FUZZ_REPLAY_SANITIZED_TARGET): $(FUZZ_SOURCE) ../../src/resp.c ../../src/resp.h ../../src/resp_limits.h ../../src/key_codec.c ../../src/key_codec.h
	$(CC) $(CPPFLAGS) -DPGLC_KEY_CODEC_STANDALONE $(COMMON_CFLAGS) -O1 \
		-fsanitize=address,undefined -fno-omit-frame-pointer \
		-DPGLC_RESP_FUZZ_REPLAY $(FUZZ_SOURCE) ../../src/resp.c \
		../../src/key_codec.c $(LDFLAGS) \
		-fsanitize=address,undefined -o $@

check: $(TARGETS) $(FUZZ_REPLAY_TARGET)
	./$(TARGET)
	./row_payload_test
	./cache_arena_test
	./cache_index_test
	./key_codec_test
	./dirty_key_limit_test
	./$(FUZZ_REPLAY_TARGET) $(FUZZ_CORPUS)

sanitize: $(SANITIZED_TARGETS) $(FUZZ_REPLAY_SANITIZED_TARGET)
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./$(SANITIZED_TARGET)
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./row_payload_test_sanitized
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./cache_arena_test_sanitized
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./cache_index_test_sanitized
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./key_codec_test_sanitized
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./dirty_key_limit_test_sanitized
	ASAN_OPTIONS=$${ASAN_OPTIONS:-detect_leaks=0:strict_string_checks=1} \
	UBSAN_OPTIONS=$${UBSAN_OPTIONS:-print_stacktrace=1:halt_on_error=1} \
	./$(FUZZ_REPLAY_SANITIZED_TARGET) $(FUZZ_CORPUS)

clean:
	rm -f $(TARGETS) $(SANITIZED_TARGETS) $(FUZZ_REPLAY_TARGET) $(FUZZ_REPLAY_SANITIZED_TARGET)