Fix: filesystem report

[gnucomo.git] / src / gcm_input / gcm_input.cpp

diff --git a/src/gcm_input/gcm_input.cpp b/src/gcm_input/gcm_input.cpp
index bcce2e8..f9e30d9 100644 (file)
--- a/src/gcm_input/gcm_input.cpp
+++ b/src/gcm_input/gcm_input.cpp
@@ -7,7 +7,7 @@
 ***********************
 **      FILE NAME      : gcm_input.cpp
 **      SYSTEM NAME    : Gnucomo - Gnu Computer Monitoring
-**      VERSION NUMBER : $Revision: 1.14 $
+**      VERSION NUMBER : $Revision: 1.16 $
 **
 **  DESCRIPTION      :  Application to store client messages into the database
 **                      The client message contains a log file from one of the
@@ -54,7 +54,14 @@
 
 /*****************************
    $Log: gcm_input.cpp,v $
-   Revision 1.14  2007-11-03 10:26:13  arjen
+   Revision 1.16  2011-03-24 10:20:37  arjen
+   Added some debug info
+
+   Revision 1.15  2007/11/14 16:19:25  arjen
+   Bugfix: Segmentation fault when reading an rpm package list
+   with empty lines.
+
+   Revision 1.14  2007/11/03 10:26:13  arjen
    Added a new filter which can directly read the output
    of the UNIX df command. A brief description is added in
    the user manual.
@@ -118,11 +125,10 @@
 
 *****************************/
 
-static const char *RCSID = "$Id: gcm_input.cpp,v 1.14 2007-11-03 10:26:13 arjen Exp $";
-
 #include <fstream>
 
 #include <getopt.h>
+#include <unistd.h>
 
 #include "message.h"
 #include "log_filter.h"
@@ -140,7 +146,7 @@ bool testmode = false;
 bool incremental = false;
 std::ostream *Log = &std::cerr;
 
-static char *Version = "gcm_input version 0.0.11 - Nov 22, 2007";
+static char Version[] = "gcm_input version 0.0.13 - Sep 22, 2020";
 
 
 /*=========================================================================
@@ -168,7 +174,7 @@ int main(int argc, char *argv[])
                        " [-s service] [-T] [-v] [-V]\n";
 
    gnucomo_config    cfg;
-   char             *config_name = "gnucomo";
+   String           config_name("gnucomo");
    std::ofstream    logfile;
 
 
@@ -245,11 +251,7 @@ int main(int argc, char *argv[])
 
    if (log_method == "file" && log_destination != "")
    {
-#if __GNUC__ == 2
-      logfile.open(log_destination, _IO_APPEND); // for gcc 2
-#else
-      logfile.open(log_destination, std::ios_base::app); // for gcc 3
-#endif
+      logfile.open(log_destination, std::ios_base::app);
       if (!logfile)
       {
          std::cerr << "Can't open logfile " << log_destination << " for writing.\n";
@@ -273,58 +275,72 @@ int main(int argc, char *argv[])
 
    gnucomo_database db(&cfg);
 
-   if (db.is_connected())
-   {
+   int gcm_input_result = 0;
+
+
+   client_message      msg(&std::cin, db);
+
+   double              message_probability;
+
+   message_filter      shortcircuit(hostname, arrival, service);
+   log_filter          lf(hostname, arrival, service);
+   rpm_filter          rf(hostname, arrival, service);
+   df_filter           df(hostname, arrival, service);
+
+   syslog_cooker       slc;
+   irix_syslog_cooker  islc;
+   access_cooker       alc;
+   error_cooker        elc;
+   xml_cooker          xlc;
+   rpm_cooker          rlc;
+   df_cooker           dlc;
+
+   msg.add_cooker(&xlc,  &shortcircuit);
+   msg.add_cooker(&slc,  &lf);
+   msg.add_cooker(&islc, &lf);
+   msg.add_cooker(&alc,  &lf);
+   msg.add_cooker(&elc,  &lf);
+   msg.add_cooker(&rlc,  &rf);
+   msg.add_cooker(&dlc,  &df);
 
-      client_message      msg(&std::cin, db);
-
-      double              message_probability;
-
-      message_filter      shortcircuit(hostname, arrival, service);
-      log_filter          lf(hostname, arrival, service);
-      rpm_filter          rf(hostname, arrival, service);
-      df_filter           df(hostname, arrival, service);
-
-      syslog_cooker       slc;
-      irix_syslog_cooker  islc;
-      access_cooker       alc;
-      error_cooker        elc;
-      xml_cooker          xlc;
-      rpm_cooker          rlc;
-      df_cooker           dlc;
-
-      msg.add_cooker(&xlc,  &shortcircuit);
-      msg.add_cooker(&slc,  &lf);
-      msg.add_cooker(&islc, &lf);
-      msg.add_cooker(&alc,  &lf);
-      msg.add_cooker(&elc,  &lf);
-      msg.add_cooker(&rlc,  &rf);
-      msg.add_cooker(&dlc,  &df);
-
-      message_probability = msg.classify(hostname, arrival, service);
-      *Log << "Message type probability = " << message_probability << "\n";
-      if (message_probability > 0.75)
+   message_probability = msg.classify(hostname, arrival, service);
+   if (message_probability > 0.75)
+   {
+      try
       {
-         try
-         {
-            msg.enter();
-         }
-         catch (std::exception &e)
+         if (msg.enter() < 0)
          {
-            *Log << "Caught an exception: " << e.what() << "\n";
+            // Can not store the input in the database. Dump the message in a file.
+
+            String spool_dir      = cfg.find_parameter("spool", "directory");
+            String pid(getpid());
+            String xmlfilename;
+
+            if (spool_dir)
+            {
+               xmlfilename = spool_dir + "/";
+            }
+            xmlfilename += "gnucomo" + pid + ".xml";
+
+            msg.saveXML(xmlfilename);
+
+            // Report the error to the log and stderr.
+            *Log << "Entering the content into the database failed. XML content stored in " + xmlfilename + "\n";
+            std::cerr << "Entering the content into the database failed. XML content stored in " + xmlfilename + "\n";
+
+            gcm_input_result = 1;
          }
       }
-      else
+      catch (std::exception &e)
       {
-         *Log << "Cannot determine message type with sufficient certainty.\n";
+         *Log << "Caught an exception: " << e.what() << "\n";
       }
-      return 0;
    }
    else
    {
-      *Log << "gcm_input: Can not connect to database.\n";
-      *Log << "Gcm_input finished at " << Now() << ".\n";
-      return 1;
+      *Log << "Cannot determine message type with sufficient certainty.\n";
    }
+
+   return gcm_input_result;
 }