Implements NFLOG runmode

diff --git a/src/Makefile.am b/src/Makefile.am
index 27a02dc961d7f71296994e99b507c7a7706ac138..fd143269a1dcb7ee658ca95fd95cf21c00193a13 100644 (file)
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -237,6 +237,7 @@ runmode-erf-file.c runmode-erf-file.h \
 runmode-ipfw.c runmode-ipfw.h \
 runmode-napatech.c runmode-napatech.h \
 runmode-nfq.c runmode-nfq.h \
+runmode-nflog.c runmode-nflog.h \
 runmode-pcap.c runmode-pcap.h \
 runmode-pcap-file.c runmode-pcap-file.h \
 runmode-pfring.c runmode-pfring.h \
@@ -251,6 +252,7 @@ source-ipfw.c source-ipfw.h \
 source-mpipe.c source-mpipe.h \
 source-napatech.c source-napatech.h \
 source-nfq.c source-nfq.h \
+source-nflog.c source-nflog.h \
 source-pcap.c source-pcap.h \
 source-pcap-file.c source-pcap-file.h \
 source-pfring.c source-pfring.h \

diff --git a/src/runmode-nflog.c b/src/runmode-nflog.c
new file mode 100644 (file)
index 0000000..15628f1
--- /dev/null
+++ b/src/runmode-nflog.c
@@ -0,0 +1,251 @@
+/* Copyright (C) 2014 Open Information Security Foundation
+ *
+ * You can copy, redistribute or modify this Program under the terms of
+ * the GNU General Public License version 2 as published by the Free
+ * Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * version 2 along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301, USA.
+ */
+
+/**
+ * \file
+ *
+ * \author Giuseppe Longo <giuseppelng@gmail.com>
+ */
+#include "suricata-common.h"
+#include "config.h"
+#include "tm-threads.h"
+#include "conf.h"
+#include "runmodes.h"
+#include "runmode-nflog.h"
+
+#include "util-debug.h"
+#include "util-device.h"
+#include "util-runmodes.h"
+#include "util-misc.h"
+
+#include "source-nflog.h"
+
+static const char *default_mode = NULL;
+
+const char *RunModeIdsNflogGetDefaultMode(void)
+{
+    return default_mode;
+}
+
+void RunModeIdsNflogRegister(void)
+{
+    default_mode = "autofp";
+    RunModeRegisterNewRunMode(RUNMODE_NFLOG, "autofp",
+                              "Multi threaded nflog mode",
+                              RunModeIdsNflogAutoFp);
+    RunModeRegisterNewRunMode(RUNMODE_NFLOG, "single",
+                              "Single threaded nflog mode",
+                              RunModeIdsNflogSingle);
+    RunModeRegisterNewRunMode(RUNMODE_NFLOG, "workers",
+                              "Workers nflog mode",
+                              RunModeIdsNflogWorkers);
+    return;
+}
+
+
+static void NflogDerefConfig(void *data)
+{
+    NflogGroupConfig *nflogconf = (NflogGroupConfig *)data;
+    SCFree(nflogconf);
+}
+
+void *ParseNflogConfig(const char *group)
+{
+    ConfNode *group_root;
+    ConfNode *group_default = NULL;
+    ConfNode *nflog_node;
+    NflogGroupConfig *nflogconf = SCMalloc(sizeof(*nflogconf));
+    intmax_t bufsize;
+    intmax_t bufsize_max;
+    intmax_t qthreshold;
+    intmax_t qtimeout;
+    int boolval;
+
+    if (unlikely(nflogconf == NULL))
+        return NULL;
+
+    if (group == NULL) {
+        SCFree(nflogconf);
+        return NULL;
+    }
+
+    nflogconf->DerefFunc = NflogDerefConfig;
+    nflog_node = ConfGetNode("nflog");
+
+    if (nflog_node == NULL) {
+        SCLogInfo("Unable to find nflog config using default value");
+        return nflogconf;
+    }
+
+    group_root = ConfNodeLookupKeyValue(nflog_node, "group", group);
+
+    group_default = ConfNodeLookupKeyValue(nflog_node, "group", "default");
+
+    if (group_root == NULL && group_default == NULL) {
+        SCLogInfo("Unable to find nflog config for "
+                  "group \"%s\" or \"default\", using default value",
+                  group);
+        return nflogconf;
+    }
+
+    nflogconf->nful_overrun_warned = 0;
+    strlcpy(nflogconf->numgroup, group, sizeof(nflogconf->numgroup));
+
+    if (ParseSizeStringU16(group, &nflogconf->group) < 0) {
+        SCLogError(SC_ERR_NFLOG_GROUP, "NFLOG's group number invalid.");
+        exit(EXIT_FAILURE);
+    }
+
+    boolval = ConfGetChildValueIntWithDefault(group_root, group_default,
+                                              "buffer-size", &bufsize);
+
+    if (boolval)
+        nflogconf->nlbufsiz = bufsize;
+    else {
+        SCLogError(SC_ERR_INVALID_ARGUMENT, "Invalid buffer-size value");
+        SCFree(nflogconf);
+        return NULL;
+    }
+
+    boolval = ConfGetChildValueIntWithDefault(group_root, group_default,
+                                              "max-size", &bufsize_max);
+
+    if (boolval)
+        nflogconf->nlbufsiz_max = bufsize_max;
+    else {
+        SCLogError(SC_ERR_INVALID_ARGUMENT, "Invalid max-size value");
+        SCFree(nflogconf);
+        return NULL;
+    }
+
+    boolval = ConfGetChildValueIntWithDefault(group_root, group_default,
+                                              "qthreshold", &qthreshold);
+
+    if (boolval)
+        nflogconf->qthreshold = qthreshold;
+    else {
+        SCLogError(SC_ERR_INVALID_ARGUMENT, "Invalid qthreshold value");
+        SCFree(nflogconf);
+        return NULL;
+    }
+
+    boolval = ConfGetChildValueIntWithDefault(group_root, group_default,
+                                              "qtimeout", &qtimeout);
+
+    if (boolval)
+        nflogconf->qtimeout = qtimeout;
+    else {
+        SCLogError(SC_ERR_INVALID_ARGUMENT, "Invalid qtimeout value");
+        SCFree(nflogconf);
+        return NULL;
+    }
+
+    return nflogconf;
+}
+
+int NflogConfigGeThreadsCount(void *conf)
+{
+    /* for each nflog group there is no reason to use more than 1 thread */
+    return 1;
+}
+
+int RunModeIdsNflogAutoFp(DetectEngineCtx *de_ctx)
+{
+    SCEnter();
+
+#ifdef HAVE_NFLOG
+    int ret = 0;
+    char *live_dev = NULL;
+
+    RunModeInitialize();
+    TimeModeSetLive();
+
+    ret = RunModeSetLiveCaptureAutoFp(de_ctx,
+                                      ParseNflogConfig,
+                                      NflogConfigGeThreadsCount,
+                                      "ReceiveNFLOG",
+                                      "DecodeNFLOG",
+                                      "RecvNFLOG",
+                                      live_dev);
+    if (ret != 0) {
+        SCLogError(SC_ERR_RUNMODE, "Unable to start runmode");
+        exit(EXIT_FAILURE);
+    }
+
+    SCLogInfo("RunModeIdsNflogAutoFp initialised");
+#endif /* HAVE_NFLOG */
+
+    SCReturnInt(0);
+}
+
+int RunModeIdsNflogSingle(DetectEngineCtx *de_ctx)
+{
+    SCEnter();
+
+#ifdef HAVE_NFLOG
+    int ret = 0;
+    char *live_dev = NULL;
+
+    RunModeInitialize();
+    TimeModeSetLive();
+
+    ret = RunModeSetLiveCaptureSingle(de_ctx,
+                                      ParseNflogConfig,
+                                      NflogConfigGeThreadsCount,
+                                      "ReceiveNFLOG",
+                                      "DecodeNFLOG",
+                                      "RecvNFLOG",
+                                      live_dev);
+    if (ret != 0) {
+        SCLogError(SC_ERR_RUNMODE, "Unable to start runmode");
+        exit(EXIT_FAILURE);
+    }
+
+    SCLogInfo("RunModeIdsNflogSingle initialised");
+#endif /* HAVE_NFLOG */
+
+    SCReturnInt(0);
+}
+
+int RunModeIdsNflogWorkers(DetectEngineCtx *de_ctx)
+{
+    SCEnter();
+
+#ifdef HAVE_NFLOG
+    int ret = 0;
+    char *live_dev = NULL;
+
+    RunModeInitialize();
+    TimeModeSetLive();
+
+    ret = RunModeSetLiveCaptureWorkers(de_ctx,
+                                       ParseNflogConfig,
+                                       NflogConfigGeThreadsCount,
+                                       "ReceiveNFLOG",
+                                       "DecodeNFLOG",
+                                       "RecvNFLOG",
+                                       live_dev);
+    if (ret != 0) {
+        SCLogError(SC_ERR_RUNMODE, "Unable to start runmode");
+        exit(EXIT_FAILURE);
+    }
+
+    SCLogInfo("RunModeIdsNflogWorkers initialised");
+#endif /* HAVE_NFLOG */
+
+    SCReturnInt(0);
+}

diff --git a/src/runmode-nflog.h b/src/runmode-nflog.h
new file mode 100644 (file)
index 0000000..2dd4e7a
--- /dev/null
+++ b/src/runmode-nflog.h
@@ -0,0 +1,32 @@
+/* Copyright (C) 2014 Open Information Security Foundation
+ *
+ * You can copy, redistribute or modify this Program under the terms of
+ * the GNU General Public License version 2 as published by the Free
+ * Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * version 2 along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301, USA.
+ */
+
+/**
+ * \file
+ *
+ * \author Giuseppe Longo <giuseppelng@gmail.com>
+ */
+#ifndef __RUNMODE_NFLOG_H__
+#define __RUNMODE_NFLOG_H__
+
+int RunModeIdsNflogAutoFp(DetectEngineCtx *);
+int RunModeIdsNflogSingle(DetectEngineCtx *);
+int RunModeIdsNflogWorkers(DetectEngineCtx *);
+void RunModeIdsNflogRegister(void);
+const char *RunModeIdsNflogGetDefaultMode(void);
+
+#endif /* __RUNMODE_NFLOG_H__ */

diff --git a/src/runmodes.c b/src/runmodes.c
index bca4cc4039a62e7c012cda18084080431dfff256..7a316163d50a4409caa2da49698b5ae8edd82605 100644 (file)
--- a/src/runmodes.c
+++ b/src/runmodes.c
@@ -117,6 +117,8 @@ static const char *RunModeTranslateModeToName(int runmode)
 #endif
         case RUNMODE_NFQ:
             return "NFQ";
+        case RUNMODE_NFLOG:
+            return "NFLOG";
         case RUNMODE_IPFW:
             return "IPFW";
         case RUNMODE_ERF_FILE:
@@ -203,6 +205,7 @@ void RunModeRegisterRunModes(void)
     RunModeErfDagRegister();
     RunModeNapatechRegister();
     RunModeIdsAFPRegister();
+    RunModeIdsNflogRegister();
     RunModeTileMpipeRegister();
     RunModeUnixSocketRegister();
 #ifdef UNITTESTS
@@ -306,6 +309,9 @@ void RunModeDispatch(int runmode, const char *custom_mode, DetectEngineCtx *de_c
             case RUNMODE_UNIX_SOCKET:
                 custom_mode = RunModeUnixSocketGetDefaultMode();
                 break;
+            case RUNMODE_NFLOG:
+                custom_mode = RunModeIdsNflogGetDefaultMode();
+                break;
             default:
                 SCLogError(SC_ERR_UNKNOWN_RUN_MODE, "Unknown runtime mode. Aborting");
                 exit(EXIT_FAILURE);

diff --git a/src/runmodes.h b/src/runmodes.h
index b238902f648b95c95aa41b34459d39979ae0340b..caf09a945747c50395bc428cde572c84ee71cc78 100644 (file)
--- a/src/runmodes.h
+++ b/src/runmodes.h
@@ -30,6 +30,7 @@ enum {
     RUNMODE_PCAP_FILE,
     RUNMODE_PFRING,
     RUNMODE_NFQ,
+    RUNMODE_NFLOG,
     RUNMODE_IPFW,
     RUNMODE_ERF_FILE,
     RUNMODE_DAG,
@@ -81,6 +82,7 @@ void RunModeShutDown(void);
 #include "runmode-erf-dag.h"
 #include "runmode-napatech.h"
 #include "runmode-af-packet.h"
+#include "runmode-nflog.h"
 #include "runmode-unix-socket.h"
 
 int threading_set_cpu_affinity;