cmake: add option ORC_SWAPRUN_NET for stability from network execution

DFIR-ORC · Feb 11, 2021 · 241711d · 241711d
1 parent 5828703
commit 241711d
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -15,6 +15,7 @@ option(ORC_BUILD_JSON       "Build with JSON StructuredOutput enabled" ON)
 option(ORC_BUILD_BOOST_STACKTRACE  "Build with stack backtrace enabled" ON)
 option(ORC_DOWNLOADS_ONLY   "Do not build ORC but only download vcpkg third parties" OFF)
 option(ORC_DISABLE_PRECOMPILED_HEADERS "Disable precompiled headers" OFF)
+option(ORC_SWAPRUN_NET      "Enable PE option to for stable execution from network" ON)
 
 if(NOT ORC_BUILD_COMMAND)
     set(ORC_BUILD_FASTFIND OFF)

diff --git a/README.md b/README.md
@@ -62,6 +62,7 @@ cmake --build . --config MinSizeRel -- -maxcpucount
 | ORC_USE_STATIC_CRT   | ON                    | Use static runtime               |
 | ORC_VCPKG_ROOT       | ${ORC}/external/vcpkg | VCPKG root directory             |
 | ORC_XMLLITE_PATH     |                       | XmlLite.dll path (xp sp2)        |
+| ORC_SWAPRUN_NET      | ON                    | Enable PE flag for network use   |
 | VCPKG_TARGET_TRIPLET | Autodetect            | VCPKG triplet to use             |
 | CMAKE_TOOLCHAIN_FILE | Autodetect            | VCPKG's toolchain file           |
 

diff --git a/src/FastFind/CMakeLists.txt b/src/FastFind/CMakeLists.txt
@@ -84,6 +84,14 @@ if("${CMAKE_GENERATOR_TOOLSET}" STREQUAL "v141_xp"
     endif()
 endif()
 
+if(${ORC_SWAPRUN_NET})
+    add_custom_command(
+            TARGET FastFind
+            POST_BUILD
+            COMMAND editbin /swaprun:net $<TARGET_FILE:Orc>
+    )
+endif()
+
 foreach(CONFIG Debug MinSizeRel Release RelWithDebInfo)
     install(TARGETS FastFind
         CONFIGURATIONS ${CONFIG}

diff --git a/src/Orc/CMakeLists.txt b/src/Orc/CMakeLists.txt
@@ -91,6 +91,14 @@ if("${CMAKE_GENERATOR_TOOLSET}" STREQUAL "v141_xp"
     endif()
 endif()
 
+if(${ORC_SWAPRUN_NET})
+    add_custom_command(
+            TARGET Orc
+            POST_BUILD
+            COMMAND editbin /swaprun:net $<TARGET_FILE:Orc>
+    )
+endif()
+
 if(${ORC_BUILD_PARQUET})
 
     if("${TARGET_ARCH}" STREQUAL "x64")