Merge pull request #94 from CESNET/bsd-support

BSD support

Author: Lukas955

CMakeModules/FindLibEpollShim.cmake

@@ -0,0 +1,40 @@
+#  LIBEPOLLSHIM_FOUND - System has libepoll-shim
+#  LIBEPOLLSHIM_INCLUDE_DIRS - The libepoll-shim include directories
+#  LIBEPOLLSHIM_LIBRARIES - The libraries needed to use libepoll-shim
+#  LIBEPOLLSHIM_DEFINITIONS - Compiler switches required for using libepoll-shim
+
+# use pkg-config to get the directories and then use these values
+# in the find_path() and find_library() calls
+find_package(PkgConfig)
+if (PKG_CONFIG_FOUND)
+    pkg_check_modules(PC_EPOLLSHIM QUIET epoll-shim)
+    set(LIBEPOLLSHIM_DEFINITIONS ${PC_EPOLLSHIM_CFLAGS_OTHER})
+endif()
+
+find_path(
+    EPOLLSHIM_INCLUDE_DIR libepoll-shim/sys/epoll.h
+    HINTS ${PC_EPOLLSHIM_INCLUDEDIR} ${PC_EPOLLSHIM_INCLUDE_DIRS}
+    PATH_SUFFIXES include
+)
+find_library(
+    EPOLLSHIM_LIBRARY NAMES epoll-shim libepoll-shim
+    HINTS ${PC_EPOLLSHIM_LIBDIR} ${PC_EPOLLSHIM_LIBRARY_DIRS}
+    PATH_SUFFIXES lib lib64
+)
+
+if (PC_EPOLLSHIM_VERSION)
+    # Version extracted from pkg-config
+    set(EPOLLSHIM_VERSION_STRING ${PC_EPOLLSHIM_VERSION})
+endif()
+
+# handle the QUIETLY and REQUIRED arguments and set LIBEPOLLSHIM_FOUND to TRUE
+# if all listed variables are TRUE
+include(FindPackageHandleStandardArgs)
+find_package_handle_standard_args(LibEpollShim
+    REQUIRED_VARS EPOLLSHIM_LIBRARY EPOLLSHIM_INCLUDE_DIR
+    VERSION_VAR EPOLLSHIM_VERSION_STRING
+)
+
+set(LIBEPOLLSHIM_LIBRARIES ${EPOLLSHIM_LIBRARY})
+set(LIBEPOLLSHIM_INCLUDE_DIRS ${EPOLLSHIM_INCLUDE_DIR}/libepoll-shim)
+mark_as_advanced(EPOLLSHIM_INCLUDE_DIR EPOLLSHIM_LIBRARY)

src/core/CMakeLists.txt

@@ -79,6 +79,10 @@ target_link_libraries(ipfixcol2base
 
 # Build IPFIXCOL2 exacutable with all symbols from the base library
 set(BASE_LIB -Wl,--whole-archive ipfixcol2base -Wl,--no-whole-archive)
+if (CMAKE_HOST_SYSTEM_NAME STREQUAL "FreeBSD" OR CMAKE_HOST_SYSTEM_NAME STREQUAL "OpenBSD")
+    set(BASE_LIB ${BASE_LIB} -rdynamic)
+endif()
+
 add_executable(ipfixcol2 main.cpp)
 target_link_libraries(ipfixcol2 ${BASE_LIB})
 set_target_properties(ipfixcol2 PROPERTIES  # by default, hide all symbols

src/core/configurator/cpipe.c

@@ -9,6 +9,7 @@
  */
 
 
+#include <assert.h>
 #include <unistd.h>
 #include <fcntl.h>
 #include <errno.h>

src/core/context.c

@@ -45,7 +45,11 @@
 #include <pthread.h>
 #include <errno.h>
 #include <signal.h>
+#if defined(__OpenBSD__) || defined(__FreeBSD__)
+#include <pthread_np.h>
+#else
 #include <sys/prctl.h>
+#endif
 
 #include "context.h"
 #include "extension.h"
@@ -640,13 +644,17 @@ thread_set_name(const char *ident)
     strncpy(name, ident, size - 1);
     name[size - 1] = '\0';
 
+#if defined(__OpenBSD__) || defined(__FreeBSD__)
+    pthread_set_name_np(pthread_self(), name);
+#else
     int rc = prctl(PR_SET_NAME, name, 0, 0, 0);
     if (rc == -1) {
         const char *err_str;
         ipx_strerror(errno, err_str);
         IPX_WARNING(comp_str, "Failed to set the name of a thread. prctl() failed: %s",
             err_str);
     }
+#endif
 }
 
 /**
@@ -656,6 +664,9 @@ thread_set_name(const char *ident)
 static inline void
 thread_get_name(char ident[16])
 {
+#if defined(__OpenBSD__) || defined(__FreeBSD__)
+    pthread_get_name_np(pthread_self(), ident, 16);
+#else
     int rc = prctl(PR_GET_NAME, ident, 0, 0, 0);
     if (rc == -1) {
         const char *err_str;
@@ -664,6 +675,7 @@ thread_get_name(char ident[16])
             err_str);
         ident[0] = '\0';
     }
+#endif
 }
 
 int

src/core/message_ipfix.c

@@ -39,6 +39,7 @@
  *
  */
 
+#include <assert.h>
 #include <ipfixcol2.h>
 #include <libfds.h>
 #include "message_base.h"
@@ -190,4 +191,4 @@ ipx_msg_ipfix_add_drec_ref(struct ipx_msg_ipfix **msg_ref)
     const size_t offset = msg->rec_info.cnt_valid * msg->rec_info.rec_size;
     msg->rec_info.cnt_valid++;
     return ((struct ipx_ipfix_record *) (((uint8_t *) msg->recs) + offset));
-}
+}

src/core/message_terminate.c

@@ -39,6 +39,7 @@
  *
  */
 
+#include <assert.h>
 #include <stddef.h>
 #include <stdlib.h>
 #include "message_terminate.h"
@@ -84,4 +85,4 @@ enum ipx_msg_terminate_type
 ipx_msg_terminate_get_type(const ipx_msg_terminate_t *msg)
 {
     return msg->type;
-}
+}

src/core/netflow2ipfix/netflow5.c

@@ -8,6 +8,7 @@
  * SPDX-License-Identifier: BSD-3-Clause
  */
 
+#include <assert.h>
 #include <stdbool.h>
 #include <stdint.h>
 #include <endian.h>
@@ -552,4 +553,4 @@ void
 ipx_nf5_conv_verb(ipx_nf5_conv_t *conv, enum ipx_verb_level v_new)
 {
     conv->conf.vlevel = v_new;
-}
+}

src/core/netflow2ipfix/netflow9.c

@@ -8,6 +8,7 @@
  * SPDX-License-Identifier: BSD-3-Clause
  */
 
+#include <assert.h>
 #include <stdint.h>
 #include <endian.h>
 #include <assert.h>
@@ -1274,4 +1275,4 @@ void
 ipx_nf9_conv_verb(ipx_nf9_conv_t *conv, enum ipx_verb_level v_new)
 {
     conv->vlevel = v_new;
-}
+}

src/core/odid_range.c

@@ -39,6 +39,7 @@
  *
  */
 
+#include <assert.h>
 #include <stdint.h>
 #include <stddef.h>
 #include <stdlib.h>
@@ -433,4 +434,4 @@ ipx_orange_print(const ipx_orange_t *range)
             break;
         }
     }
-}
+}

src/core/session.c

@@ -39,15 +39,36 @@
  *
  */
 
-// Get GNU specific basename() function
-#define _GNU_SOURCE
 #include <string.h>
 
 #include <stdint.h>
 #include <ipfixcol2.h>
 #include <stdlib.h>
 #include <inttypes.h>
 
+/**
+ * \brief Return the name part of a file path, i.e. the part after the last /
+ * \param path  The file path
+ * \return The name part of the path or NULL if path is NULL
+ */
+static const char *get_basename(const char *path)
+{
+    if (path == NULL) {
+        return NULL;
+    }
+
+    const char *res = path;
+    const char *p = path;
+    while (*p != '\0') {
+        if (*p == '/') {
+            res = p + 1;
+        }
+        p++;
+    }
+
+    return res;
+}
+
 /**
  * \brief Create a source description string from a Network Session structure
  *
@@ -166,7 +187,9 @@ ipx_session_new_file(const char *file_path)
         return NULL;
     }
 
-    res->ident = basename(res->file.file_path); // GNU specific basename()
+    res->ident = res->file.file_path;
+
+    res->ident = (char *) get_basename(res->file.file_path);
     if (!res->ident) {
         free(res);
         return NULL;
@@ -190,4 +213,4 @@ ipx_session_destroy(struct ipx_session *session)
     }
 
     free(session);
-}
+}