Makefile

# Makefile

# Compilation options.

ARCH              ?= arm64
SDK               ?= iphoneos
CORE_DIR          ?= core
CORE_LIB          ?= $(CORE_DIR)/libmemctl_core.a
CORE_ENTITLEMENTS ?= $(CORE_DIR)/entitlements.plist

REPL              ?= YES

TEST_DETAIL       ?= 0

ifneq ($(ARCH),x86_64)
CLANG    := $(shell xcrun --sdk $(SDK) --find clang)
AR       := $(shell xcrun --sdk $(SDK) --find ar)
ifeq ($(CLANG),)
$(error Could not find clang for SDK $(SDK))
endif
SYSROOT  := $(shell xcrun --sdk $(SDK) --show-sdk-path)
CC       := $(CLANG) -isysroot $(SYSROOT) -arch $(ARCH)
endif
CODESIGN := codesign

# Directories.

SRC_DIR = src
INC_DIR = include
OBJ_DIR = obj
BIN_DIR = bin
LIB_DIR = lib
EXTERNAL_HDR_DIR = external
EXTERNAL_SRC_DIR = external
TEST_DIR = test

LIBMEMCTL_DIR     = libmemctl
LIBMEMCTL_INC_DIR = memctl
MEMCTL_DIR        = memctl

# Flags.

ERRFLAGS   = -Wall -Wpedantic -Wno-gnu -Werror
CFLAGS     = -g -O2 $(ERRFLAGS) $(EXTRA_CFLAGS)
LDFLAGS    = -g -lcompression
FRAMEWORKS = -framework Foundation -framework IOKit
ARFLAGS    = r

LIBMEMCTL_CFLAGS = -I$(INC_DIR) -I$(EXTERNAL_HDR_DIR) -I$(SRC_DIR)/$(LIBMEMCTL_DIR)
MEMCTL_CFLAGS    = -I$(INC_DIR) -I$(EXTERNAL_HDR_DIR) -I$(SRC_DIR)/$(MEMCTL_DIR)
EXTERNAL_CFLAGS  = -I$(EXTERNAL_HDR_DIR)

ifeq ($(REPL),YES)
LDFLAGS        += -ledit -lcurses
MEMCTL_DEFINES += -DMEMCTL_REPL=1
endif

# libmemctl arm64/aarch64 sources.

ARCH_arm64_DIR = aarch64

LIBMEMCTL_arm64_SRCS = finder/kauth_cred_setsvuidgid.c \
		       finder/pmap_cache_attributes.c \
		       finder/pthread_callbacks.c \
		       finder/vtables.c \
		       finder/zone_element_size.c \
		       jop/call_strategy_1.c \
		       jop/call_strategy_2.c \
		       jop/call_strategy_3.c \
		       jop/gadgets_static.c \
		       disasm.c \
		       kernel_call_aarch64.c \
		       ksim.c \
		       memory_region.c \
		       sim.c

LIBMEMCTL_arm64_HDRS = finder/kauth_cred_setsvuidgid.h \
		       finder/pmap_cache_attributes.h \
		       finder/pthread_callbacks.h \
		       finder/vtables.h \
		       finder/zone_element_size.h \
		       jop/call_strategy.h \
		       jop/gadgets_static.h \
		       kernel_call_aarch64.h

LIBMEMCTL_arm64_INCS = disasm.h \
		       ksim.h \
		       sim.h

# libmemctl x86_64 sources.

ARCH_x86_64_DIR = x86_64

LIBMEMCTL_x86_64_SRCS = kernel_call_syscall_asm.s \
			kernel_call_syscall_x86_64.c \
			memory_region.c

LIBMEMCTL_x86_64_HDRS = kernel_call_syscall_code.h

LIBMEMCTL_x86_64_INCS = kernel_call_syscall_x86_64.h

# libmemctl sources.

LIBMEMCTL_ARCH_SRCS = $(LIBMEMCTL_$(ARCH)_SRCS:%=$(ARCH_$(ARCH)_DIR)/%)
LIBMEMCTL_ARCH_HDRS = $(LIBMEMCTL_$(ARCH)_HDRS:%=$(ARCH_$(ARCH)_DIR)/%)
LIBMEMCTL_ARCH_INCS = $(LIBMEMCTL_$(ARCH)_INCS:%=$(ARCH_$(ARCH)_DIR)/%)

LIBMEMCTL_SRCS = $(LIBMEMCTL_ARCH_SRCS) \
		 algorithm.c \
		 class.c \
		 core.c \
		 error.c \
		 kernel.c \
		 kernel_call.c \
		 kernel_memory.c \
		 kernel_slide.c \
		 kernelcache.c \
		 macho.c \
		 mangle.c \
		 mapped_region.c \
		 memctl_common.c \
		 memctl_error.c \
		 memctl_signal.c \
		 oskext.c \
		 platform.c \
		 privilege_escalation.c \
		 process.c \
		 symbol_finders.c \
		 symbol_table.c \
		 task_memory.c

LIBMEMCTL_HDRS = $(LIBMEMCTL_ARCH_HDRS) \
		 algorithm.h \
		 mangle.h \
		 memctl_common.h

LIBMEMCTL_INCS = $(LIBMEMCTL_ARCH_INCS) \
		 class.h \
		 core.h \
		 error.h \
		 kernel.h \
		 kernel_call.h \
		 kernel_memory.h \
		 kernel_slide.h \
		 kernelcache.h \
		 macho.h \
		 mapped_region.h \
		 memctl_error.h \
		 memctl_signal.h \
		 memctl_types.h \
		 memory_region.h \
		 offset.h \
		 oskext.h \
		 platform.h \
		 privilege_escalation.h \
		 process.h \
		 symbol_finders.h \
		 symbol_table.h \
		 task_memory.h \
		 utility.h

LIBMEMCTL_SRCS := $(LIBMEMCTL_SRCS:%=$(SRC_DIR)/$(LIBMEMCTL_DIR)/%)
LIBMEMCTL_HDRS := $(LIBMEMCTL_HDRS:%=$(SRC_DIR)/$(LIBMEMCTL_DIR)/%)
LIBMEMCTL_INCS := $(LIBMEMCTL_INCS:%=$(INC_DIR)/$(LIBMEMCTL_INC_DIR)/%)
LIBMEMCTL_OBJS := $(LIBMEMCTL_SRCS:$(SRC_DIR)/%.c=$(OBJ_DIR)/%.o)
LIBMEMCTL_OBJS := $(LIBMEMCTL_OBJS:$(SRC_DIR)/%.s=$(OBJ_DIR)/%.o)

# External libmemctl sources.

EXTERNAL_LIBMEMCTL_SRCS = lzss.c

EXTERNAL_LIBMEMCTL_SRCS := $(EXTERNAL_LIBMEMCTL_SRCS:%=$(EXTERNAL_SRC_DIR)/%)
EXTERNAL_LIBMEMCTL_OBJS := $(EXTERNAL_LIBMEMCTL_SRCS:%.c=$(OBJ_DIR)/%.o)

LIBMEMCTL_SRCS += $(EXTERNAL_LIBMEMCTL_SRCS)
LIBMEMCTL_OBJS += $(EXTERNAL_LIBMEMCTL_OBJS)

# The libmemctl static library.

MEMCTL_LIB := $(LIB_DIR)/libmemctl.a

# memctl sources.

MEMCTL_arm64_SRCS = disassemble.c

MEMCTL_ARCH_SRCS = $(MEMCTL_$(ARCH)_SRCS:%=$(ARCH_$(ARCH)_DIR)/%)
MEMCTL_ARCH_HDRS = $(MEMCTL_$(ARCH)_HDRS:%=$(ARCH_$(ARCH)_DIR)/%)

MEMCTL_SRCS = $(MEMCTL_ARCH_SRCS) \
	      cli.c \
	      command.c \
	      error.c \
	      find.c \
	      format.c \
	      initialize.c \
	      memctl.c \
	      memory.c \
	      read.c \
	      strparse.c \
	      vmmap.c

MEMCTL_HDRS = $(MEMCTL_ARCH_HDRS) \
	      cli.h \
	      command.h \
	      disassemble.h \
	      error.h \
	      find.h \
	      format.h \
	      initialize.h \
	      memory.h \
	      read.h \
	      strparse.h \
	      vmmap.h

MEMCTL_SRCS := $(MEMCTL_SRCS:%=$(SRC_DIR)/$(MEMCTL_DIR)/%)
MEMCTL_HDRS := $(MEMCTL_HDRS:%=$(SRC_DIR)/$(MEMCTL_DIR)/%)
MEMCTL_OBJS := $(MEMCTL_SRCS:$(SRC_DIR)/%.c=$(OBJ_DIR)/%.o)

MEMCTL_BIN := $(BIN_DIR)/memctl

# Tests.

aarch64_disasm_SRCS = $(SRC_DIR)/$(LIBMEMCTL_DIR)/aarch64/disasm.c \
		      $(SRC_DIR)/$(LIBMEMCTL_DIR)/macho.c \
		      $(SRC_DIR)/$(MEMCTL_DIR)/aarch64/disassemble.c \
		      $(TEST_DIR)/aarch64_disasm/aarch64_disasm.c

aarch64_disasm_CFLAGS = -I$(INC_DIR) \
			-I$(SRC_DIR)/$(MEMCTL_DIR) \
			-DMEMCTL_DISASSEMBLY=1

TESTS = aarch64_disasm

define make_test_rules

$(1)_OBJS := $$($(1)_SRCS:$$(SRC_DIR)/%.c=$$(OBJ_DIR)/%.o)
$(1)_OBJS := $$($(1)_OBJS:$$(TEST_DIR)/%.c=$$(OBJ_DIR)/%.o)

$$(OBJ_DIR)/$(1)/%.o: $$(TEST_DIR)/$(1)/%.c $$(LIBMEMCTL_INCS)
	@mkdir -p $$(@D)
	$$(CC) $$(CFLAGS) $$($(1)_CFLAGS) -c $$< -o $$@

$$(BIN_DIR)/$(1): $$($(1)_OBJS)
	@mkdir -p $$(@D)
	$$(CC) $$(LDFLAGS) $$(FRAMEWORKS) $$^ -o $$@
	$$(CODESIGN) $$(CODESIGN_FLAGS) -s - $$@

test_$(1): $$(BIN_DIR)/$(1)
	$$(TEST_DIR)/$(1)/test.sh "$$(BIN_DIR)/$(1)" "$$(TEST_DIR)/$(1)" $$(TEST_DETAIL)

endef

# Targets.

.PHONY: all clean test

all: $(MEMCTL_BIN)

ifneq ($(wildcard $(CORE_ENTITLEMENTS)),)
CODESIGN_FLAGS = --entitlements "$(CORE_ENTITLEMENTS)"
endif

$(OBJ_DIR)/$(MEMCTL_DIR)/%.o: $(SRC_DIR)/$(MEMCTL_DIR)/%.c $(MEMCTL_HDRS) $(LIBMEMCTL_INCS)
	@mkdir -p $(@D)
	$(CC) $(CFLAGS) $(MEMCTL_CFLAGS) $(MEMCTL_DEFINES) -c $< -o $@

$(OBJ_DIR)/$(LIBMEMCTL_DIR)/%.o: $(SRC_DIR)/$(LIBMEMCTL_DIR)/%.c $(LIBMEMCTL_INCS) $(LIBMEMCTL_HDRS)
	@mkdir -p $(@D)
	$(CC) $(CFLAGS) $(LIBMEMCTL_CFLAGS) -c $< -o $@

$(OBJ_DIR)/$(LIBMEMCTL_DIR)/%.o: $(SRC_DIR)/$(LIBMEMCTL_DIR)/%.s $(LIBMEMCTL_INCS) $(LIBMEMCTL_HDRS)
	@mkdir -p $(@D)
	$(CC) $(CFLAGS) $(LIBMEMCTL_CFLAGS) -c $< -o $@

$(OBJ_DIR)/$(EXTERNAL_SRC_DIR)/%.o: $(EXTERNAL_SRC_DIR)/%.c
	@mkdir -p $(@D)
	$(CC) $(CFLAGS) $(EXTERNAL_CFLAGS) -c $< -o $@

$(MEMCTL_BIN): $(MEMCTL_LIB) $(CORE_LIB) $(MEMCTL_OBJS)
	@mkdir -p $(@D)
	$(CC) $(LDFLAGS) $(FRAMEWORKS) $(MEMCTL_OBJS) "$(CORE_LIB)" -force_load $(MEMCTL_LIB) -o $@
	$(CODESIGN) $(CODESIGN_FLAGS) -s - $@

$(MEMCTL_LIB): $(LIBMEMCTL_OBJS)
	@mkdir -p $(@D)
	$(AR) $(ARFLAGS) $@ $^

test: $(TESTS:%=test_%)

$(foreach test,$(TESTS),$(eval $(call make_test_rules,$(test))))

clean:
	rm -rf -- $(OBJ_DIR) $(BIN_DIR) $(LIB_DIR)