ifneq (,$(wildcard ./.env))
  include .env
  export
else
  $(warning .env file not found, defaults will be used)
endif

# Default configuration
FLAG_LEN ?= 32
FLAG_FORMAT ?= ctf{FLAG_GOES_HERE}

# Compilation variables
RISCV_CC ?= riscv64-unknown-elf-gcc
RISCV_ARCH_FLAGS ?= -march=rv32im -mabi=ilp32
HOST_CC ?= gcc
HOST_CFLAGS ?= -Wall -Wextra -O3 -g
HOST_LDFLAGS ?= -Wl,-s

EMU_SRC_FILES := $(wildcard src/emulated/*.c)
EMU_OBJ_FILES := $(EMU_SRC_FILES:src/emulated/%.c=build/%.riscv.o)
EMU_OBJ := build/emulated.riscv.elf
EMU_OBJ_LICENSE := build/license
EMU_OBJ_LICENSE_IF_EXISTS := $(wildcard $(EMU_OBJ_LICENSE))

ZIP_FILE := build/challenge.zip
ZIP_FILE_IF_EXISTS := $(wildcard $(ZIP_FILE))


GENERATED := build/include/generated.h

HOST_SRC_EXTRA := src/elf.c src/hwinfo.c src/util.c tiny-AES-c/aes.c
HOST_SRC := src/main.c $(HOST_SRC_EXTRA)
HOST_SRC_ENCRYPTOR := src/encryptor.c $(HOST_SRC_EXTRA)
HOST_INCLUDE := -I src -I tinyriscv/src -I tiny-AES-c
HOST_SRC_SPOOFER := src/spoof/spoofer.c


ifeq ($(OS),Windows_NT)
    HOST_EXE := build/main.exe
else
    HOST_EXE := build/main
endif

ifeq ($(OS),Windows_NT)
    HOST_EXE_ENCRYPTOR := build/encryptor.exe
else
    HOST_EXE_ENCRYPTOR := build/encryptor
endif

HOST_EXE_SPOOFER := build/spoofer


# Disassembly
RISCV_OBJDUMP ?= $(patsubst %-gcc,%-objdump,$(RISCV_CC))
_DISAS_TARGETS := $(filter disas-%,$(MAKECMDGOALS))
_DISAS_FILES   := $(if $(_DISAS_TARGETS),$(filter %.elf %.o,$(MAKECMDGOALS)))



.PHONY: all emulated-riscv host clean disas-risc

all: $(ZIP_FILE) $(HOST_EXE_SPOOFER)

emulated-riscv: $(EMU_OBJ_LICENSE)

host: $(HOST_EXE) $(HOST_EXE_SPOOFER)

# Build the main challenge executable
$(HOST_EXE): $(HOST_SRC) | build
	$(HOST_CC) $(HOST_CFLAGS) $(HOST_LDFLAGS) $(HOST_INCLUDE) $(HOST_SRC) -o $@

# Build the spoofer
$(HOST_EXE_SPOOFER): $(HOST_SRC_SPOOFER) | build
	$(HOST_CC) $(HOST_CFLAGS) $(HOST_LDFLAGS) -shared -fPIC $^ -o $@ -ldl

# Build the encryptor
$(HOST_EXE_ENCRYPTOR): $(HOST_SRC_ENCRYPTOR) | build
	$(HOST_CC) $(HOST_CFLAGS) $(HOST_LDFLAGS) $(HOST_INCLUDE) $(HOST_SRC_ENCRYPTOR) -o $@

# Build emulated object files
$(EMU_OBJ_FILES): $(EMU_SRC_FILES) $(GENERATED) | build
	$(RISCV_CC) $(RISCV_ARCH_FLAGS) -I $(dir $(GENERATED)) -I src/emulated -c $(EMU_SRC_FILES) -o $@


# Encrypt license file
$(EMU_OBJ_LICENSE): $(EMU_OBJ) $(HOST_EXE_ENCRYPTOR) | build
ifneq ($(EMU_OBJ_LICENSE_IF_EXISTS),)
	rm $(EMU_OBJ_LICENSE_IF_EXISTS)
endif
	$(HOST_EXE_ENCRYPTOR) $(EMU_OBJ) $@

# Build (unencrypted) license file
$(EMU_OBJ): $(EMU_OBJ_FILES) | build
	$(RISCV_CC) $(RISCV_ARCH_FLAGS) -nostdlib -Wl,-e,entry -Wl,-Ttext=0 -Wl,--emit-relocs $(EMU_OBJ_FILES) -o $@


# Bundle the challenge
$(ZIP_FILE): $(HOST_EXE) $(EMU_OBJ_LICENSE) | build
ifneq ($(ZIP_FILE_IF_EXISTS),)
	rm $(ZIP_FILE_IF_EXISTS)
endif
	cd build && 7z a -bso0 $(CURDIR)/$@ $(notdir $^)

.FORCE:

$(GENERATED): .FORCE
	mkdir -p $(dir $@)
	python codegen/generate.py $(FLAG_FORMAT) $(FLAG_LEN) > $@


build:
	mkdir -p $@

clean:
	rm -rf build

disas-risc: $(_DISAS_FILES)
	$(RISCV_OBJDUMP) -d $^