BPF适配不同的编译环境

2 files changed, 35 insertions, 1 deletions

diff --git a/bpf/CMakeLists.txt b/bpf/CMakeLists.txt
index edaf3a1..095e301 100644
--- a/bpf/CMakeLists.txt
+++ b/bpf/CMakeLists.txt
@@ -1 +1,7 @@
-execute_process(COMMAND make WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}/bpf/)
\ No newline at end of file
+add_library(bpf_obj bpf_obj.cpp)
+target_include_directories(bpf_obj PUBLIC ${CMAKE_CURRENT_LIST_DIR})
+if (SUPPORT_BPF)
+    target_link_libraries(bpf_obj PUBLIC bpf)
+    execute_process(COMMAND make WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}/bpf/)
+	install(FILES ./bpf_tun_rss_steering.o DESTINATION ${CMAKE_INSTALL_PREFIX}/resource/bpf/ COMPONENT PROFILE)
+endif()
\ No newline at end of file
diff --git a/bpf/bpf_obj.c b/bpf/bpf_obj.cpp
index 86ce8b2..dde48cd 100644
--- a/bpf/bpf_obj.c
+++ b/bpf/bpf_obj.cpp
@@ -1,9 +1,12 @@
 #include <errno.h>
+#include <stdio.h>
+#include <stdint.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <sys/ioctl.h>
 #include <linux/if_tun.h>
 
+#ifdef SUPPORT_BPF
 #include "bpf_config_user.h"
 
 #define MIN(a, b) ((a) > (b) ? (b) : (a))
@@ -95,3 +98,28 @@ int bpf_obj_attach(struct bpf_obj_ctx *ctx, int fd)
         return -1;
     }
 }
+#else
+
+struct bpf_obj_ctx
+{
+};
+
+#include <sys/types.h>
+
+void bpf_obj_unload(struct bpf_obj_ctx *ctx)
+{
+    printf("ERROR: BPF feature not support on current system\n");
+}
+
+struct bpf_obj_ctx *bpf_obj_load(const char *obj_file, uint32_t queue_num, uint32_t hash_mode, uint32_t debug_log)
+{
+    printf("ERROR: BPF feature not support on current system\n");
+    return NULL;
+}
+
+int bpf_obj_attach(struct bpf_obj_ctx *ctx, int fd)
+{
+    printf("ERROR: BPF feature not support on current system\n");
+    return -1;
+}
+#endif
\ No newline at end of file