cmake_minimum_required(VERSION 3.15)

# C++ standard
set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

# C standard
set(CMAKE_C_STANDARD 99)
set(CMAKE_C_STANDARD_REQUIRED ON)

#
# Project details
#
project(
  "runtime"
  VERSION 0.1.0
  LANGUAGES C CXX
)

#
# Set project options
#
include(${CMAKE_CURRENT_SOURCE_DIR}/../cmake/StandardSettings.cmake)

if(NOT CMAKE_BUILD_TYPE)
  set(CMAKE_BUILD_TYPE "Debug")
endif()

message(STATUS "Started CMake for ${PROJECT_NAME} v${PROJECT_VERSION}...\n")

if(UNIX)
  add_compile_options("$<$<CONFIG:DEBUG>:-D_DEBUG>") # this will allow to use same _DEBUG macro available in both Linux as well as Windows - MSCV environment. Easy to put Debug specific code.
endif(UNIX)

#
# Prevent building in the source directory
#
if(PROJECT_SOURCE_DIR STREQUAL PROJECT_BINARY_DIR)
  message(FATAL_ERROR "In-source builds not allowed. Please make a new directory (called a build directory) and run CMake from there.\n")
endif()

#
# Create library, setup header and source files
#
find_package(Boost REQUIRED)

# Find all headers and implementation files
if(NOT DEFINED ARCH)
  set(ARCH ${CMAKE_SYSTEM_PROCESSOR})
endif()

message(STATUS "Building for architecture: ${ARCH}")

set(sources
  src/attach/bpf_attach_ctx.cpp

  src/handler/handler_manager.cpp
  src/handler/map_handler.cpp
  src/handler/perf_event_handler.cpp
  src/handler/prog_handler.cpp
  src/handler/epoll_handler.cpp
  src/handler/memfd_handler.cpp
  
  src/bpftime_shm.cpp
  src/bpftime_shm_internal.cpp
  src/bpftime_shm_json.cpp
  src/bpftime_prog.cpp
  src/bpftime_config.cpp
  src/ufunc.cpp
  src/bpf_helper.cpp
  src/platform_utils.cpp

  src/bpf_map/userspace/array_map.cpp
  src/bpf_map/userspace/fix_hash_map.cpp
  src/bpf_map/userspace/var_hash_map.cpp
  src/bpf_map/userspace/ringbuf_map.cpp
  src/bpf_map/userspace/perf_event_array_map.cpp
  src/bpf_map/userspace/per_cpu_array_map.cpp
  src/bpf_map/userspace/per_cpu_hash_map.cpp
  src/bpf_map/userspace/prog_array.cpp
  src/bpf_map/userspace/stack_trace_map.cpp
  src/bpf_map/userspace/lru_var_hash_map.cpp

  src/bpf_map/userspace/queue.cpp
  src/bpf_map/userspace/stack.cpp
  src/bpf_map/userspace/bloom_filter.cpp
  src/bpf_map/userspace/lpm_trie_map.cpp
  extension/extension_helper.cpp
)

if(${BPFTIME_ENABLE_CUDA_ATTACH})
  list(APPEND sources src/attach/bpf_attach_ctx_cuda.cpp)
  list(APPEND sources src/bpf_map/gpu/nv_gpu_array_map.cpp)
  list(APPEND sources src/bpf_map/gpu/nv_gpu_shared_array_map.cpp)
  list(APPEND sources src/bpf_map/gpu/nv_gpu_ringbuf_map.cpp)    
endif()

if(UNIX AND NOT APPLE AND BPFTIME_BUILD_WITH_LIBBPF)
  list(APPEND sources
    src/bpf_map/shared/array_map_kernel_user.cpp
    src/bpf_map/shared/hash_map_kernel_user.cpp
    src/bpf_map/shared/percpu_array_map_kernel_user.cpp
    src/bpf_map/shared/perf_event_array_kernel_user.cpp
  )
endif()

set(headers
  include/
)
message(INFO " Headers: ${headers}")
message(INFO " Found the following sources: ${sources}")

# Keep constructor-based self-registration; do not add centralized registry

add_library(
  ${PROJECT_NAME}
  ${sources}
)
set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 20)
if(${BPFTIME_LLVM_JIT})
  target_compile_definitions(${PROJECT_NAME} PUBLIC BPFTIME_LLVM_JIT=1)
endif()
if(${BPFTIME_UBPF_JIT})
  target_compile_definitions(${PROJECT_NAME} PUBLIC BPFTIME_UBPF_JIT=1)
endif()
add_custom_command(
  OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/syscall_id_list.h
  COMMAND /bin/bash ${CMAKE_CURRENT_SOURCE_DIR}/generate_syscall_id_table.sh "${CMAKE_CURRENT_BINARY_DIR}/syscall_id_list.h"
  USES_TERMINAL
)
add_custom_target(
  syscall_id_table
  DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/syscall_id_list.h
)

message(INFO " Found the following boost include dirs : ${Boost_INCLUDE_DIRS}")


target_include_directories(${PROJECT_NAME}
  PUBLIC
  ${CMAKE_CURRENT_SOURCE_DIR}/../vm/include
  ${CMAKE_CURRENT_SOURCE_DIR}/../runtime/include
  ${CMAKE_CURRENT_SOURCE_DIR}/../runtime
  ${CMAKE_CURRENT_BINARY_DIR}
  ${CMAKE_CURRENT_SOURCE_DIR}/../third_party
  ${CMAKE_CURRENT_SOURCE_DIR}
  ${SPDLOG_INCLUDE}
  ${Boost_INCLUDE_DIRS}
)

target_link_libraries(${PROJECT_NAME}
  PUBLIC
  bpftime_vm
  spdlog::spdlog
  bpftime_base_attach_impl

)


if(${BPFTIME_ENABLE_CUDA_ATTACH})
  include(../cmake/cuda.cmake)
  find_cuda()
  message(STATUS "runtime CUDA libs= ${CUDA_LIBS}")
  target_include_directories(${PROJECT_NAME} PUBLIC ${CUDA_INCLUDE_PATH} ${NV_ATTACH_IMPL_INCLUDE})
  target_link_directories(${PROJECT_NAME} PUBLIC ${CUDA_LIBRARY_PATH})
  target_link_libraries(${PROJECT_NAME} PUBLIC ${CUDA_LIBS} bpftime_nv_attach_impl cuda)
  target_compile_definitions(${PROJECT_NAME} PUBLIC BPFTIME_ENABLE_CUDA_ATTACH=1)
  add_dependencies(${PROJECT_NAME} bpftime_vm_compat llvmbpf_vm bpftime_nv_attach_impl)
  target_link_libraries(${PROJECT_NAME} PUBLIC bpftime_vm_compat llvmbpf_vm bpftime_nv_attach_impl)
  target_include_directories(${PROJECT_NAME} PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}/../vm/llvm-jit/include)
endif()
if(${BPFTIME_BUILD_WITH_LIBBPF})
  add_dependencies(${PROJECT_NAME} bpftime_vm FridaGum spdlog::spdlog libbpf bpftime_base_attach_impl)
else()
  add_dependencies(${PROJECT_NAME} bpftime_vm FridaGum spdlog::spdlog bpftime_base_attach_impl)
endif()

if(BPFTIME_ENABLE_IOURING_EXT)
  target_link_libraries(${PROJECT_NAME}
    PUBLIC
    uring
  )
endif()

if(${ENABLE_EBPF_VERIFIER})
  target_include_directories(${PROJECT_NAME} PRIVATE ${BPFTIME_VERIFIER_INCLUDE})
  target_link_libraries(${PROJECT_NAME} PRIVATE bpftime-verifier)
  add_dependencies(${PROJECT_NAME} bpftime-verifier)
  target_compile_definitions(${PROJECT_NAME} PRIVATE ENABLE_EBPF_VERIFIER ENABLE_BPFTIME_VERIFIER)
endif()

message(DEBUG "Found the following sources: ${sources}")

message(DEBUG "Found the following headers: ${headers}")

# set the -static flag for static linking
if(NOT BPFTIME_ENABLE_ASAN)
  # set the -static flag for static linking
  # set_target_properties(${test_name}_Tests PROPERTIES LINK_FLAGS "-static")
  # need on qemu-user
endif()

message(STATUS "Added all header and implementation files.\n")

#
# Set the project standard and warnings
#
set_project_warnings(runtime)

message(DEBUG "Applied compiler warnings. Using standard ${CMAKE_CXX_STANDARD}.")

if(APPLE)
  find_library(COCOA_LIBRARY Cocoa)
  find_library(FOUNDATION_LIBRARY Foundation)
  find_library(APPKIT_LIBRARY AppKit)
  find_library(COREF_LIBRARY CoreFoundation)
  find_library(RESOLV_LIBRARY resolv)
  mark_as_advanced(COCOA_LIBRARY FOUNDATION_LIBRARY APPKIT_LIBRARY COREF_LIBRARY RESOLV_LIBRARY)
  set(EXTRA_LIBS ${COCOA_LIBRARY} ${FOUNDATION_LIBRARY} ${APPKIT_LIBRARY} ${COREF_LIBRARY} ${RESOLV_LIBRARY})
endif()

if(${BPFTIME_BUILD_WITH_LIBBPF})
  target_link_libraries(${PROJECT_NAME}
    PUBLIC
    ${LIBBPF_LIBRARIES}
    ${FRIDA_GUM_INSTALL_DIR}/libfrida-gum.a
    -lpthread
    -lm
    -ldl
    -lz
    -lelf
    bpftime_base_attach_impl
    ${EXTRA_LIBS}
  )
else()
  target_link_libraries(${PROJECT_NAME}
    PUBLIC
    ${FRIDA_GUM_INSTALL_DIR}/libfrida-gum.a
    -lpthread
    -lm
    -ldl
    -lz
    bpftime_base_attach_impl
    ${EXTRA_LIBS}
  )
endif()

if(BPFTIME_BUILD_WITH_LIBBPF)
  target_include_directories(${PROJECT_NAME} PUBLIC
    ${LIBBPF_INCLUDE_DIRS}/uapi
    ${LIBBPF_INCLUDE_DIRS}
    ${FRIDA_GUM_INSTALL_DIR}
    $<INSTALL_INTERFACE:runtime>
    $<INSTALL_INTERFACE:runtime/src>
    $<INSTALL_INTERFACE:include>
    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/src>
  )
else()
  target_include_directories(${PROJECT_NAME} PUBLIC
    ${FRIDA_GUM_INSTALL_DIR}
    $<INSTALL_INTERFACE:runtime>
    $<INSTALL_INTERFACE:runtime/src>
    $<INSTALL_INTERFACE:include>
    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/src>
  )
endif()

message(DEBUG "Successfully added all dependencies and linked against them.")

set(BPFTIME_RUNTIME_INCLUDE ${CMAKE_CURRENT_SOURCE_DIR}/include ${CMAKE_CURRENT_SOURCE_DIR}/src)

if(BPFTIME_BUILD_WITH_LIBBPF)
  add_subdirectory(object)
endif()

add_subdirectory(agent)
add_subdirectory(syscall-server)

#
# Unit testing setup
#
if(BPFTIME_ENABLE_UNIT_TESTING AND BPFTIME_BUILD_WITH_LIBBPF)
  enable_testing()
  message(STATUS "Build unit tests for the runtime. Tests should always be found in the test folder\n")
  add_subdirectory(test)
  add_subdirectory(unit-test)
endif()

if(${TEST_LCOV})
  target_compile_options(${PROJECT_NAME} PUBLIC -fprofile-arcs -ftest-coverage -fprofile-update=atomic)
  target_link_options(${PROJECT_NAME} PUBLIC -fprofile-arcs)
endif()