suricata: list cuda cards in separate function

diff --git a/src/suricata.c b/src/suricata.c
index 953d114b08a012a913331afa90834b02906f6222..f42d0488758c74338c6051f25bb6a047d1824430 100644 (file)
--- a/src/suricata.c
+++ b/src/suricata.c
@@ -1326,6 +1326,12 @@ int main(int argc, char **argv)
         return SuriListAppLayerProtocols();
     }
 
+#ifdef __SC_CUDA_SUPPORT__
+    if (list_cuda_cards) {
+        return SuriListCudaCards();
+    }
+#endif
+
 #ifdef REVISION
     SCLogInfo("This is %s version %s (rev %s)", PROG_NAME, PROG_VER, xstr(REVISION));
 #elif defined RELEASE
@@ -1341,10 +1347,6 @@ int main(int argc, char **argv)
 #ifdef __SC_CUDA_SUPPORT__
     /* Init the CUDA environment */
     SCCudaInitCudaEnvironment();
-    if (list_cuda_cards) {
-        SCCudaListCards();
-        exit(EXIT_SUCCESS);
-    }
     CudaBufferInit();
 #endif
 

diff --git a/src/util-running-modes.c b/src/util-running-modes.c
index 439d27ffb457febf78a6724e2cca054137bc4272..073199b5eb0da30d02ebab2ca3efe060127ce997 100644 (file)
--- a/src/util-running-modes.c
+++ b/src/util-running-modes.c
@@ -21,7 +21,9 @@
  */
 
 #include "suricata-common.h"
+#include "config.h"
 #include "app-layer-detect-proto.h"
+#include "util-cuda.h"
 
 int SuriListKeywords(const char *keyword_info)
 {
@@ -37,3 +39,12 @@ int SuriListAppLayerProtocols()
     AppLayerListSupportedProtocols();
     exit(EXIT_SUCCESS);
 }
+
+#ifdef __SC_CUDA_SUPPORT__
+int SuriListCudaCards()
+{
+    SCCudaInitCudaEnvironment();
+    SCCudaListCards();
+    exit(EXIT_SUCCESS);
+}
+#endif

diff --git a/src/util-running-modes.h b/src/util-running-modes.h
index e4a421c8002735beceb2693f9071f3d71f03d381..a2a97fedacb90b22f09226254ab233908710400e 100644 (file)
--- a/src/util-running-modes.h
+++ b/src/util-running-modes.h
@@ -26,6 +26,9 @@
 
 int SuriListKeywords(const char *keyword_info);
 int SuriListAppLayerProtocols();
+#ifdef __SC_CUDA_SUPPORT__
+int SuriListCudaCards();
+#endif
 
 
 #endif /* __UTIL_RUNNING_MODES_H__ */