C2Implant/CMakeLists.txt at master · maxDcb/C2Implant · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
cmake_minimum_required(VERSION 3.24.0 FATAL_ERROR)

project(C2Implant VERSION 0.0.0 LANGUAGES CXX C)

set(CMAKE_BUILD_TYPE Release)
set(CMAKE_CXX_STANDARD 23)

set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)

set(CMAKE_CXX_FLAGS_RELEASE "-MT -O1 -Ob0")
set(CMAKE_C_FLAGS_RELEASE "-MT -O1 -Ob0")

add_compile_definitions(
  NOMINMAX                # <-- disables Windows min/max macros
)

# --- Decide whether a prefix path was provided (var or env) ---
set(_prefixes "")

# Prefer the CMake var if set on the command line/cache.
if(DEFINED CMAKE_PREFIX_PATH AND NOT "${CMAKE_PREFIX_PATH}" STREQUAL "")
  set(_prefixes "${CMAKE_PREFIX_PATH}")
elseif(DEFINED ENV{CMAKE_PREFIX_PATH} AND NOT "$ENV{CMAKE_PREFIX_PATH}" STREQUAL "")
  # fall back to environment variable
  set(_prefixes "$ENV{CMAKE_PREFIX_PATH}")
endif()

# Optional: normalize whitespace
string(STRIP "${_prefixes}" _prefixes)

if (C2CORE_BUILD_TESTS)
    add_definitions(-DC2CORE_BUILD_TESTS)
    # TODO add the right fetch
endif()

if(_prefixes)

    message(STATUS "Using CMAKE_PREFIX_PATH: ${_prefixes}")

    include_directories(thirdParty)

    ##
    ## C2Core dependency and build
    ##

    find_package(C2Core CONFIG REQUIRED)

    include_directories(thirdParty/base64)

    add_subdirectory(beacon/beacon)

else()

    message(STATUS "CMAKE_PREFIX_PATH not provided; using fallback strategy")

    include_directories(thirdParty)

    add_definitions(-DBUILD_IMPLANT)

    ##
    ## Fetch for core
    ##

    include(FetchContent)

    set(BUILD_SHARED_LIBS OFF CACHE BOOL "Build shared libraries" FORCE)
    set(BUILD_STATIC_LIBS ON CACHE BOOL "Build static libraries" FORCE)
    set(BUILD_EXAMPLES OFF CACHE BOOL "" FORCE)
    set(BUILD_TESTING OFF CACHE BOOL "" FORCE)
    add_compile_definitions(LIBSSH2_LIBRARY)    # needed for full static lib

    set(CMAKE_MSVC_RUNTIME_LIBRARY "MultiThreaded")

    FetchContent_Declare(
      libssh2
      GIT_REPOSITORY https://github.com/libssh2/libssh2.git
      GIT_TAG libssh2-1.11.1
    )

    FetchContent_MakeAvailable(libssh2)

    ##
    ## Build
    ##

    add_subdirectory(libs)

    add_subdirectory(thirdParty)
    include_directories(thirdParty/base64)
    include_directories(thirdParty/donut/include)
    include_directories(thirdParty/coffLoader/coffLoader)
    include_directories(thirdParty/MemoryModule)

    set(DONUT_BUILD_DIR "${CMAKE_BINARY_DIR}/thirdParty/donut")
    add_library(Donut STATIC IMPORTED)
    set_target_properties(Donut PROPERTIES
    IMPORTED_LOCATION "${DONUT_BUILD_DIR}/lib/donut.lib"
    )

    add_subdirectory(core/modules)
    add_subdirectory(core/beacon)

    include_directories(core/listener)
    include_directories(core/beacon)
    include_directories(core/modules/ModuleCmd)
    add_subdirectory(beacon/beacon)

endif()