]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/logger.cpp
for real
[user/henk/code/inspircd.git] / src / logger.cpp
index 83a65721679ae03cfb6c0eaafe5b008ce7be1377..c8d7e8b398de350e03c1f29e57e8c973f9c2089a 100644 (file)
@@ -15,6 +15,8 @@
 
 #include "inspircd.h"
 
+#include "filelogger.h"
+
 /*
  * Suggested implementation...
  *     class LogManager
  * 
  */
 
-bool LogManager::AddLogType(const std::string &type, LogStream *l)
+void LogManager::SetupNoFork()
+{
+       if (!noforkstream)
+       {
+               FileWriter* fw = new FileWriter(ServerInstance, stdout);
+               noforkstream = new FileLogStream(ServerInstance, ServerInstance->Config->forcedebug ? DEBUG : DEFAULT, fw);
+       }
+       else
+       {
+               noforkstream->ChangeLevel(ServerInstance->Config->forcedebug ? DEBUG : DEFAULT);
+       }
+       AddLogType("*", noforkstream, false);
+}
+
+void LogManager::OpenFileLogs()
+{
+       /* Re-register the nofork stream if necessary. */
+       if (ServerInstance->Config->nofork)
+       {
+               SetupNoFork();
+       }
+       /* Skip rest of logfile opening if we are running -nolog. */
+       if (!ServerInstance->Config->writelog)
+       {
+               return;
+       }
+       ConfigReader* Conf = new ConfigReader(ServerInstance);
+       std::map<std::string, FileWriter*> logmap;
+       std::map<std::string, FileWriter*>::iterator i;
+       for (int index = 0; index < Conf->Enumerate("log"); ++index)
+       {
+               std::string method = Conf->ReadValue("log", "method", index);
+               if (method != "file")
+               {
+                       continue;
+               }
+               std::string type = Conf->ReadValue("log", "type", index);
+               std::string level = Conf->ReadValue("log", "level", index);
+               int loglevel = DEFAULT;
+               if (level == "debug" || ServerInstance->Config->forcedebug)
+               {
+                       loglevel = DEBUG;
+                       ServerInstance->Config->debugging = true;
+               }
+               else if (level == "verbose")
+               {
+                       loglevel = VERBOSE;
+               }
+               else if (level == "default")
+               {
+                       loglevel = DEFAULT;
+               }
+               else if (level == "sparse")
+               {
+                       loglevel = SPARSE;
+               }
+               else if (level == "none")
+               {
+                       loglevel = NONE;
+               }
+               FileWriter* fw;
+               std::string target = Conf->ReadValue("log", "target", index);
+               if ((i = logmap.find(target)) == logmap.end())
+               {
+                       FILE* f = fopen(target.c_str(), "a");
+                       fw = new FileWriter(ServerInstance, f);
+                       logmap.insert(std::make_pair(target, fw));
+               }
+               else
+               {
+                       fw = i->second;
+               }
+               FileLogStream* fls = new FileLogStream(ServerInstance, loglevel, fw);
+               AddLogTypes(type, fls, true);
+       }
+}
+
+void LogManager::CloseLogs()
+{
+       std::map<std::string, std::vector<LogStream*> >().swap(LogStreams); /* Clear it */
+       std::map<LogStream*, std::vector<std::string> >().swap(GlobalLogStreams); /* Clear it */
+       for (std::map<LogStream*, int>::iterator i = AllLogStreams.begin(); i != AllLogStreams.end(); ++i)
+       {
+               delete i->first;
+       }
+       std::map<LogStream*, int>().swap(AllLogStreams); /* And clear it */
+}
+
+void LogManager::AddLogTypes(const std::string &types, LogStream* l, bool autoclose)
+{
+       irc::spacesepstream css(types);
+       std::string tok;
+       std::vector<std::string> excludes;
+       while (css.GetToken(tok))
+       {
+               if (tok.empty())
+               {
+                       continue;
+               }
+               if (tok.at(0) == '-')
+               {
+                       /* Exclude! */
+                       excludes.push_back(tok.substr(1));
+               }
+               else
+               {
+                       AddLogType(tok, l, autoclose);
+               }
+       }
+       // Handle doing things like: USERINPUT USEROUTPUT -USERINPUT should be the same as saying just USEROUTPUT.
+       // (This is so modules could, for example, inject exclusions for logtypes they can't handle.)
+       for (std::vector<std::string>::iterator i = excludes.begin(); i != excludes.end(); ++i)
+       {
+               if (*i == "*")
+               {
+                       /* -* == Exclude all. Why someone would do this, I dunno. */
+                       DelLogStream(l);
+                       return;
+               }
+               DelLogType(*i, l);
+       }
+       // Now if it's registered as a global, add the exclusions there too.
+       std::map<LogStream *, std::vector<std::string> >::iterator gi = GlobalLogStreams.find(l);
+       if (gi != GlobalLogStreams.end())
+       {
+               gi->second.swap(excludes); // Swap with the vector in the hash.
+       }
+}
+
+bool LogManager::AddLogType(const std::string &type, LogStream *l, bool autoclose)
 {
        std::map<std::string, std::vector<LogStream *> >::iterator i = LogStreams.find(type);
 
        if (i != LogStreams.end())
+       {
                i->second.push_back(l);
+       }
        else
        {
                std::vector<LogStream *> v;
@@ -56,78 +189,178 @@ bool LogManager::AddLogType(const std::string &type, LogStream *l)
        }
 
        if (type == "*")
-               GlobalLogStreams.push_back(l);
+       {
+               GlobalLogStreams.insert(std::make_pair(l, std::vector<std::string>()));
+       }
+
+       if (autoclose)
+       {
+               std::map<LogStream*, int>::iterator ai = AllLogStreams.find(l);
+               if (ai == AllLogStreams.end())
+               {
+                       AllLogStreams.insert(std::make_pair(l, 1));
+               }
+               else
+               {
+                       ++ai->second;
+               }
+       }
 
        return true;
 }
 
-bool LogManager::DelLogType(const std::string &type, LogStream *l)
+void LogManager::DelLogStream(LogStream* l)
 {
-       std::map<std::string, std::vector<LogStream *> >::iterator i = LogStreams.find(type);
-       std::vector<LogStream *>::iterator gi = GlobalLogStreams.begin();
-
-       while (gi != GlobalLogStreams.end())
+       for (std::map<std::string, std::vector<LogStream*> >::iterator i = LogStreams.begin(); i != LogStreams.end(); ++i)
        {
-               if ((*gi) == l)
+               std::vector<LogStream*>::iterator it;
+               while ((it = std::find(i->second.begin(), i->second.end(), l)) != i->second.end())
                {
-                       GlobalLogStreams.erase(gi);
-                       break;
+                       if (it == i->second.end())
+                               continue;
+                       i->second.erase(it);
                }
        }
+       std::map<LogStream *, std::vector<std::string> >::iterator gi = GlobalLogStreams.find(l);
+       if (gi != GlobalLogStreams.end())
+       {
+               GlobalLogStreams.erase(gi);
+       }
+       std::map<LogStream*, int>::iterator ai = AllLogStreams.begin();
+       if (ai == AllLogStreams.end())
+       {
+               return; /* Done. */
+       }
+       delete ai->first;
+       AllLogStreams.erase(ai);
+}
+
+bool LogManager::DelLogType(const std::string &type, LogStream *l)
+{
+       std::map<std::string, std::vector<LogStream *> >::iterator i = LogStreams.find(type);
+       if (type == "*")
+       {
+               std::map<LogStream *, std::vector<std::string> >::iterator gi = GlobalLogStreams.find(l);
+               if (gi != GlobalLogStreams.end()) GlobalLogStreams.erase(gi);
+       }
 
        if (i != LogStreams.end())
        {
-               std::vector<LogStream *>::iterator it = i->second.begin();
+               std::vector<LogStream *>::iterator it = std::find(i->second.begin(), i->second.end(), l);
 
-               while (it != i->second.end())
+               if (it != i->second.end())
                {
-                       if (*it == l)
+                       i->second.erase(it);
+                       if (i->second.size() == 0)
                        {
-                               i->second.erase(it);
+                               LogStreams.erase(i);
+                       }
+               }
+               else
+               {
+                       return false;
+               }
+       }
+       else
+       {
+               return false;
+       }
 
-                               if (i->second.size() == 0)
-                               {
-                                       LogStreams.erase(i);
-                               }
+       std::map<LogStream*, int>::iterator ai = AllLogStreams.find(l);
+       if (ai == AllLogStreams.end())
+       {
+               return true;
+       }
 
-                               delete l;
-                               return true;
-                       }
+       if ((--ai->second) < 1)
+       {
+               AllLogStreams.erase(ai);
+               delete l;
+       }
 
-                       it++;
-               }
+       return true;
+}
+
+void LogManager::Log(const std::string &type, int loglevel, const char *fmt, ...)
+{
+       if (Logging)
+       {
+               return;
        }
 
-       return false;
+       va_list a;
+       static char buf[65536];
+
+       va_start(a, fmt);
+       vsnprintf(buf, 65536, fmt, a);
+       va_end(a);
+
+       this->Log(type, loglevel, std::string(buf));
 }
 
 void LogManager::Log(const std::string &type, int loglevel, const std::string &msg)
 {
+       if (Logging)
+       {
+               return;
+       }
+
+       Logging = true;
+
+       for (std::map<LogStream *, std::vector<std::string> >::iterator gi = GlobalLogStreams.begin(); gi != GlobalLogStreams.end(); ++gi)
+       {
+               if (std::find(gi->second.begin(), gi->second.end(), type) != gi->second.end())
+               {
+                       continue;
+               }
+               gi->first->OnLog(loglevel, type, msg);
+       }
+
        std::map<std::string, std::vector<LogStream *> >::iterator i = LogStreams.find(type);
 
        if (i != LogStreams.end())
        {
-               std::vector<LogStream *>::iterator it = i->second.begin();
-
-               while (it != i->second.end())
+               for (std::vector<LogStream *>::iterator it = i->second.begin(); it != i->second.end(); ++it)
                {
-                       (*it)->OnLog(loglevel, msg);
-                       it++;
+                       (*it)->OnLog(loglevel, type, msg);
                }
-
-               return;
        }
 
-       std::vector<LogStream *>::iterator gi = GlobalLogStreams.begin();
+       Logging = false;
+}
 
-       while (gi != GlobalLogStreams.end())
+
+FileWriter::FileWriter(InspIRCd* Instance, FILE* logfile)
+: ServerInstance(Instance), log(logfile), writeops(0)
+{
+}
+
+void FileWriter::HandleEvent(EventType ev, int)
+{
+}
+
+void FileWriter::WriteLogLine(const std::string &line)
+{
+       if (log)
        {
-               (*gi)->OnLog(loglevel, msg);
-               gi++;
+               fprintf(log,"%s",line.c_str());
+               if (writeops++ % 20)
+               {
+                       fflush(log);
+               }
        }
+}
 
-       // blurp, no handler for this type
-       return;
+void FileWriter::Close()
+{
+       if (log)
+       {
+               fflush(log);
+               fclose(log);
+       }
 }
 
+FileWriter::~FileWriter()
+{
+}