Use common functions in admin tests and service tests.

[stg.git] / projects / stargazer / plugins / capture / cap_nf / cap_nf.cpp

diff --git a/projects/stargazer/plugins/capture/cap_nf/cap_nf.cpp b/projects/stargazer/plugins/capture/cap_nf/cap_nf.cpp
index 51ba49546d26fb7a4d8c9ee3be18b9be1b9750ac..8c4fdc2dd350cf9dbc9306d59273cf39446d2366 100644 (file)
--- a/projects/stargazer/plugins/capture/cap_nf/cap_nf.cpp
+++ b/projects/stargazer/plugins/capture/cap_nf/cap_nf.cpp
@@ -39,13 +39,18 @@ $Author: faust $
 
 #include <vector>
 
 
 #include <vector>
 

-#include "stg/common.h" 
+#include "stg/common.h"

 #include "stg/raw_ip_packet.h"
 #include "stg/traffcounter.h"
 #include "stg/plugin_creator.h"
 #include "cap_nf.h"
 
 #include "stg/raw_ip_packet.h"
 #include "stg/traffcounter.h"
 #include "stg/plugin_creator.h"
 #include "cap_nf.h"
 

+namespace
+{

 PLUGIN_CREATOR<NF_CAP> cnc;
 PLUGIN_CREATOR<NF_CAP> cnc;

+}
+
+extern "C" PLUGIN * GetPlugin();

 
 PLUGIN * GetPlugin()
 {
 
 PLUGIN * GetPlugin()
 {


@@ -65,7 +70,8 @@ NF_CAP::NF_CAP()
       portU(0),
       sockTCP(-1),
       sockUDP(-1),
       portU(0),
       sockTCP(-1),
       sockUDP(-1),

-      errorStr()
+      errorStr(),
+      logger(GetPluginLogger(GetStgLogger(), "cap_nf"))

 {
 }
 
 {
 }
 


@@ -117,6 +123,7 @@ if (portU > 0)
         runningUDP = false;
         CloseUDP();
         errorStr = "Cannot create UDP thread";
         runningUDP = false;
         CloseUDP();
         errorStr = "Cannot create UDP thread";

+        logger("Cannot create UDP thread.");

         printfd(__FILE__, "Error: Cannot create UDP thread\n");
         return -1;
         }
         printfd(__FILE__, "Error: Cannot create UDP thread\n");
         return -1;
         }


@@ -132,6 +139,7 @@ if (portT > 0)
         {
         runningTCP = false;
         CloseTCP();
         {
         runningTCP = false;
         CloseTCP();

+        logger("Cannot create TCP thread.");

         errorStr = "Cannot create TCP thread";
         printfd(__FILE__, "Error: Cannot create TCP thread\n");
         return -1;
         errorStr = "Cannot create TCP thread";
         printfd(__FILE__, "Error: Cannot create TCP thread\n");
         return -1;


@@ -160,10 +168,12 @@ if (portU && !stoppedUDP)
         if (pthread_kill(tidUDP, SIGUSR1))
             {
             errorStr = "Error sending signal to UDP thread";
         if (pthread_kill(tidUDP, SIGUSR1))
             {
             errorStr = "Error sending signal to UDP thread";

+            logger("Error sending sugnal to UDP thread.");

             printfd(__FILE__, "Error: Error sending signal to UDP thread\n");
             return -1;
             }
         printfd(__FILE__, "UDP thread NOT stopped\n");
             printfd(__FILE__, "Error: Error sending signal to UDP thread\n");
             return -1;
             }
         printfd(__FILE__, "UDP thread NOT stopped\n");

+        logger("Cannot stop UDP thread.");

         }
     }
 if (portT && !stoppedTCP)
         }
     }
 if (portT && !stoppedTCP)


@@ -183,10 +193,12 @@ if (portT && !stoppedTCP)
         if (pthread_kill(tidTCP, SIGUSR1))
             {
             errorStr = "Error sending signal to TCP thread";
         if (pthread_kill(tidTCP, SIGUSR1))
             {
             errorStr = "Error sending signal to TCP thread";

+            logger("Error sending signal to TCP thread.");

             printfd(__FILE__, "Error: Error sending signal to TCP thread\n");
             return -1;
             }
         printfd(__FILE__, "TCP thread NOT stopped\n");
             printfd(__FILE__, "Error: Error sending signal to TCP thread\n");
             return -1;
             }
         printfd(__FILE__, "TCP thread NOT stopped\n");

+        logger("Cannot stop TCP thread.");

         }
     }
 return 0;
         }
     }
 return 0;


@@ -199,6 +211,7 @@ sockUDP = socket(PF_INET, SOCK_DGRAM, 0);
 if (sockUDP <= 0)
     {
     errorStr = "Error opening UDP socket";
 if (sockUDP <= 0)
     {
     errorStr = "Error opening UDP socket";

+    logger("Cannot create UDP socket: %s", strerror(errno));

     printfd(__FILE__, "Error: Error opening UDP socket\n");
     return true;
     }
     printfd(__FILE__, "Error: Error opening UDP socket\n");
     return true;
     }


@@ -208,6 +221,7 @@ sin.sin_addr.s_addr = inet_addr("0.0.0.0");
 if (bind(sockUDP, (struct sockaddr *)&sin, sizeof(sin)))
     {
     errorStr = "Error binding UDP socket";
 if (bind(sockUDP, (struct sockaddr *)&sin, sizeof(sin)))
     {
     errorStr = "Error binding UDP socket";

+    logger("Cannot bind UDP socket: %s", strerror(errno));

     printfd(__FILE__, "Error: Error binding UDP socket\n");
     return true;
     }
     printfd(__FILE__, "Error: Error binding UDP socket\n");
     return true;
     }


@@ -221,6 +235,7 @@ sockTCP = socket(PF_INET, SOCK_STREAM, 0);
 if (sockTCP <= 0)
     {
     errorStr = "Error opening TCP socket";
 if (sockTCP <= 0)
     {
     errorStr = "Error opening TCP socket";

+    logger("Cannot create TCP socket: %s", strerror(errno));

     printfd(__FILE__, "Error: Error opening TCP socket\n");
     return true;
     }
     printfd(__FILE__, "Error: Error opening TCP socket\n");
     return true;
     }


@@ -230,12 +245,14 @@ sin.sin_addr.s_addr = inet_addr("0.0.0.0");
 if (bind(sockTCP, (struct sockaddr *)&sin, sizeof(sin)))
     {
     errorStr = "Error binding TCP socket";
 if (bind(sockTCP, (struct sockaddr *)&sin, sizeof(sin)))
     {
     errorStr = "Error binding TCP socket";

+    logger("Cannot bind TCP socket: %s", strerror(errno));

     printfd(__FILE__, "Error: Error binding TCP socket\n");
     return true;
     }
 if (listen(sockTCP, 1))
     {
     errorStr = "Error listening on TCP socket";
     printfd(__FILE__, "Error: Error binding TCP socket\n");
     return true;
     }
 if (listen(sockTCP, 1))
     {
     errorStr = "Error listening on TCP socket";

+    logger("Cannot listen on TCP socket: %s", strerror(errno));

     printfd(__FILE__, "Error: Error listening TCP socket\n");
     return true;
     }
     printfd(__FILE__, "Error: Error listening TCP socket\n");
     return true;
     }


@@ -249,10 +266,6 @@ sigfillset(&signalSet);
 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
 
 NF_CAP * cap = static_cast<NF_CAP *>(c);
 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
 
 NF_CAP * cap = static_cast<NF_CAP *>(c);

-uint8_t buf[BUF_SIZE];
-int res;
-struct sockaddr_in sin;
-socklen_t slen;

 cap->stoppedUDP = false;
 while (cap->runningUDP)
     {
 cap->stoppedUDP = false;
 while (cap->runningUDP)
     {


@@ -262,11 +275,19 @@ while (cap->runningUDP)
         }
 
     // Data
         }
 
     // Data

-    slen = sizeof(sin);
-    res = recvfrom(cap->sockUDP, buf, BUF_SIZE, 0, reinterpret_cast<struct sockaddr *>(&sin), &slen);
+    struct sockaddr_in sin;
+    socklen_t slen = sizeof(sin);
+    uint8_t buf[BUF_SIZE];
+    ssize_t res = recvfrom(cap->sockUDP, buf, BUF_SIZE, 0, reinterpret_cast<struct sockaddr *>(&sin), &slen);

     if (!cap->runningUDP)
         break;
 
     if (!cap->runningUDP)
         break;
 

+    if (res < 0)
+        {
+        cap->logger("recvfrom error: %s", strerror(errno));
+        continue;
+        }
+

     if (res == 0) // EOF
         {
         continue;
     if (res == 0) // EOF
         {
         continue;


@@ -295,11 +316,6 @@ sigfillset(&signalSet);
 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
 
 NF_CAP * cap = static_cast<NF_CAP *>(c);
 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
 
 NF_CAP * cap = static_cast<NF_CAP *>(c);

-uint8_t buf[BUF_SIZE];
-int res;
-int sd;
-struct sockaddr_in sin;
-socklen_t slen;

 cap->stoppedTCP = false;
 while (cap->runningTCP)
     {
 cap->stoppedTCP = false;
 while (cap->runningTCP)
     {


@@ -309,18 +325,16 @@ while (cap->runningTCP)
         }
 
     // Data
         }
 
     // Data

-    slen = sizeof(sin);
-    sd = accept(cap->sockTCP, reinterpret_cast<struct sockaddr *>(&sin), &slen);
+    struct sockaddr_in sin;
+    socklen_t slen = sizeof(sin);
+    int sd = accept(cap->sockTCP, reinterpret_cast<struct sockaddr *>(&sin), &slen);

     if (!cap->runningTCP)
         break;
 
     if (sd <= 0)
         {
     if (!cap->runningTCP)
         break;
 
     if (sd <= 0)
         {

-        if (errno != EINTR)
-            {
-            cap->errorStr = "Error accepting connection";
-            printfd(__FILE__, "Error: Error accepting connection\n");
-            }
+        if (sd < 0)
+            cap->logger("accept error: %s", strerror(errno));

         continue;
         }
 
         continue;
         }
 


@@ -330,7 +344,12 @@ while (cap->runningTCP)
         continue;
         }
 
         continue;
         }
 

-    res = recv(sd, buf, BUF_SIZE, MSG_WAITALL);
+    uint8_t buf[BUF_SIZE];
+    ssize_t res = recv(sd, buf, BUF_SIZE, MSG_WAITALL);
+
+    if (res < 0)
+        cap->logger("recv error: %s", strerror(errno));
+

     close(sd);
 
     if (!cap->runningTCP)
     close(sd);
 
     if (!cap->runningTCP)


@@ -345,11 +364,6 @@ while (cap->runningTCP)
     // Need to check actual data length and wait all data to receive
     if (res < 24)
         {
     // Need to check actual data length and wait all data to receive
     if (res < 24)
         {

-        if (errno != EINTR)
-            {
-            cap->errorStr = "Invalid data received";
-            printfd(__FILE__, "Error: Invalid data received through TCP\n");
-            }

         continue;
         }
 
         continue;
         }
 


@@ -359,7 +373,7 @@ cap->stoppedTCP = true;
 return NULL;
 }
 
 return NULL;
 }
 

-void NF_CAP::ParseBuffer(uint8_t * buf, int size)
+void NF_CAP::ParseBuffer(uint8_t * buf, ssize_t size)

 {
 RAW_PACKET ip;
 NF_HEADER * hdr = reinterpret_cast<NF_HEADER *>(buf);
 {
 RAW_PACKET ip;
 NF_HEADER * hdr = reinterpret_cast<NF_HEADER *>(buf);