Fix compilation issues on FreeBSD4

[stg.git] / projects / stargazer / traffcounter_impl.cpp

diff --git a/projects/stargazer/traffcounter_impl.cpp b/projects/stargazer/traffcounter_impl.cpp
index f410b1090b3413e8a039a5a091773fa1ac6105bd..a605e284a3c17c73a3d53d89f283aff1de2be2d9 100644 (file)
--- a/projects/stargazer/traffcounter_impl.cpp
+++ b/projects/stargazer/traffcounter_impl.cpp
@@ -29,6 +29,7 @@
  */
 
 /* inet_aton */
  */
 
 /* inet_aton */

+#include <sys/types.h>

 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>


@@ -37,9 +38,9 @@
 #include <cassert>
 #include <cstdio> // fopen and similar
 
 #include <cassert>
 #include <cstdio> // fopen and similar
 

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

 #include "traffcounter_impl.h"
 #include "traffcounter_impl.h"

-#include "common.h"
-#include "stg_locker.h"

 #include "stg_timer.h"
 
 #define FLUSH_TIME  (10)
 #include "stg_timer.h"
 
 #define FLUSH_TIME  (10)


@@ -89,6 +90,7 @@ if (!stopped)
 
 if (ReadRules())
     {
 
 if (ReadRules())
     {

+    printfd(__FILE__, "TRAFFCOUNTER_IMPL::Start() - Cannot read rules\n");

     WriteServLog("TRAFFCOUNTER: Cannot read rules.");
     return -1;
     }
     WriteServLog("TRAFFCOUNTER: Cannot read rules.");
     return -1;
     }


@@ -98,6 +100,7 @@ int h = users->OpenSearch();
 USER_PTR u;
 if (!h)
     {
 USER_PTR u;
 if (!h)
     {

+    printfd(__FILE__, "TRAFFCOUNTER_IMPL::Start() - Cannot get users\n");

     WriteServLog("TRAFFCOUNTER: Cannot get users.");
     return -1;
     }
     WriteServLog("TRAFFCOUNTER: Cannot get users.");
     return -1;
     }


@@ -111,7 +114,8 @@ users->CloseSearch(h);
 running = true;
 if (pthread_create(&thread, NULL, Run, this))
     {
 running = true;
 if (pthread_create(&thread, NULL, Run, this))
     {

-    WriteServLog("TRAFFCOUNTER: Error: Cannot start thread!");
+    printfd(__FILE__, "TRAFFCOUNTER_IMPL::Start() - Cannot start thread\n");
+    WriteServLog("TRAFFCOUNTER: Error: Cannot start thread.");

     return -1;
     }
 return 0;
     return -1;
     }
 return 0;


@@ -689,7 +693,8 @@ f = fopen(rulesFileName.c_str(), "rt");
 
 if (!f)
     {
 
 if (!f)
     {

-    WriteServLog("File %s cannot be oppened.", rulesFileName.c_str());
+    printfd(__FILE__, "TRAFFCOUNTER_IMPL::ReadRules() - File '%s' cannot be opened.\n", rulesFileName.c_str());
+    WriteServLog("File '%s' cannot be oppened.", rulesFileName.c_str());

     return true;
     }
 
     return true;
     }
 


@@ -704,7 +709,8 @@ while (fgets(str, 1023, f))
     r = sscanf(str,"%s %s %s", tp, ta, td);
     if (r != 3)
         {
     r = sscanf(str,"%s %s %s", tp, ta, td);
     if (r != 3)
         {

-        WriteServLog("Error in file %s. There must be 3 parameters. Line %d.", rulesFileName.c_str(), lineNumber);
+        printfd(__FILE__, "TRAFFCOUNTER_IMPL::ReadRules() - Error in file '%s' at line %d. There must be 3 parameters.\n", rulesFileName.c_str(), lineNumber);
+        WriteServLog("Error in file '%s' at line %d. There must be 3 parameters.", rulesFileName.c_str(), lineNumber);

         fclose(f);
         return true;
         }
         fclose(f);
         return true;
         }


@@ -726,6 +732,7 @@ while (fgets(str, 1023, f))
 
     if (rul.dir == 0xff || rul.proto == 0xff)
         {
 
     if (rul.dir == 0xff || rul.proto == 0xff)
         {

+        printfd(__FILE__, "TRAFFCOUNTER_IMPL::ReadRules() - Error in file '%s' at line %d.\n", rulesFileName.c_str(), lineNumber);

         WriteServLog("Error in file %s. Line %d.",
                      rulesFileName.c_str(), lineNumber);
         fclose(f);
         WriteServLog("Error in file %s. Line %d.",
                      rulesFileName.c_str(), lineNumber);
         fclose(f);


@@ -734,6 +741,7 @@ while (fgets(str, 1023, f))
 
     if (ParseAddress(ta, &rul) != 0)
         {
 
     if (ParseAddress(ta, &rul) != 0)
         {

+        printfd(__FILE__, "TRAFFCOUNTER_IMPL::ReadRules() - Error in file '%s' at line %d. Error in adress.\n", rulesFileName.c_str(), lineNumber);

         WriteServLog("Error in file %s. Error in adress. Line %d.",
                      rulesFileName.c_str(), lineNumber);
         fclose(f);
         WriteServLog("Error in file %s. Error in adress. Line %d.",
                      rulesFileName.c_str(), lineNumber);
         fclose(f);


@@ -768,12 +776,14 @@ STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 if (ReadRules(true))
     {
 
 if (ReadRules(true))
     {

+    printfd(__FILE__, "TRAFFCOUNTER_IMPL::Reload() - Failed to reload rules.\n");

     WriteServLog("TRAFFCOUNTER: Cannot reload rules. Errors found.");
     return -1;
     }
 
 FreeRules();
 ReadRules();
     WriteServLog("TRAFFCOUNTER: Cannot reload rules. Errors found.");
     return -1;
     }
 
 FreeRules();
 ReadRules();

+printfd(__FILE__, "TRAFFCOUNTER_IMPL::Reload() -  Reload rules successfull.\n");

 WriteServLog("TRAFFCOUNTER: Reload rules successfull.");
 return 0;
 }
 WriteServLog("TRAFFCOUNTER: Reload rules successfull.");
 return 0;
 }


@@ -828,6 +838,7 @@ if (n == ':')
     // port
     if (!(rule->proto == tcp || rule->proto == udp || rule->proto == tcp_udp))
         {
     // port
     if (!(rule->proto == tcp || rule->proto == udp || rule->proto == tcp_udp))
         {

+        printfd(__FILE__, "TRAFFCOUNTER_IMPL::ParseAddress() - No ports specified for this protocol.\n");

         WriteServLog("No ports specified for this protocol.");
         return true;
         }
         WriteServLog("No ports specified for this protocol.");
         return true;
         }


@@ -886,6 +897,7 @@ rule->mask = CalcMask(msk);
 
 if ((ip & rule->mask) != ip)
     {
 
 if ((ip & rule->mask) != ip)
     {

+    printfd(__FILE__, "TRAFFCOUNTER_IMPL::ParseAddress() - Address does'n match mask.\n");

     WriteServLog("Address does'n match mask.");
     return true;
     }
     WriteServLog("Address does'n match mask.");
     return true;
     }


@@ -938,7 +950,7 @@ return;
 //-----------------------------------------------------------------------------
 void TRAFFCOUNTER_IMPL::SetMonitorDir(const std::string & monitorDir)
 {
 //-----------------------------------------------------------------------------
 void TRAFFCOUNTER_IMPL::SetMonitorDir(const std::string & monitorDir)
 {

-TRAFFCOUNTER::monitorDir = monitorDir;
+TRAFFCOUNTER_IMPL::monitorDir = monitorDir;

 monitoring = (monitorDir != "");
 }
 //-----------------------------------------------------------------------------
 monitoring = (monitorDir != "");
 }
 //-----------------------------------------------------------------------------