include/logger.h

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #ifndef __LOGMANAGER_H
  15 #define __LOGMANAGER_H
  16
  17 /** This class implements a nonblocking writer.
  18  * Most people writing an ircd give little thought to their disk
  19  * i/o. On a congested system, disk writes can block for long
  20  * periods of time (e.g. if the system is busy and/or swapping
  21  * a lot). If we just use a blocking fprintf() call, this could
  22  * block for undesirable amounts of time (half of a second through
  23  * to whole seconds). We DO NOT want this, so we make our logfile
  24  * nonblocking and hook it into the SocketEngine.
  25  * NB: If the operating system does not support nonblocking file
  26  * I/O (linux seems to, as does freebsd) this will default to
  27  * blocking behaviour.
  28  */
  29 class CoreExport FileWriter : public EventHandler
  30 {
  31  protected:
  32         /** The creator/owner of this object
  33          */
  34         InspIRCd* ServerInstance;
  35
  36         /** The log file (fd is inside this somewhere,
  37          * we get it out with fileno())
  38          */
  39         FILE* log;
  40
  41         /** Number of write operations that have occured
  42          */
  43         int writeops;
  44
  45  public:
  46         /** The constructor takes an already opened logfile.
  47          */
  48         FileWriter(InspIRCd* Instance, FILE* logfile);
  49
  50         /** Handle pending write events.
  51          * This will flush any waiting data to disk.
  52          * If any data remains after the fprintf call,
  53          * another write event is scheduled to write
  54          * the rest of the data when possible.
  55          */
  56         virtual void HandleEvent(EventType et, int errornum = 0);
  57
  58         /** Write one or more preformatted log lines.
  59          * If the data cannot be written immediately,
  60          * this class will insert itself into the
  61          * SocketEngine, and register a write event,
  62          * and when the write event occurs it will
  63          * attempt again to write the data.
  64          */
  65         void WriteLogLine(const std::string &line);
  66
  67         /** Close the log file and cancel any events.
  68          */
  69         virtual void Close();
  70
  71         /** Close the log file and cancel any events.
  72          */
  73         virtual ~FileWriter();
  74 };
  75
  76
  77
  78 /*
  79  * New world logging!
  80  * The brief summary:
  81  *  Logging used to be a simple affair, a FILE * handled by a nonblocking logging class inheriting from EventHandler, that was inserted
  82  *  into the socket engine, and wrote lines. If nofork was on, it was printf()'d.
  83  *
  84  *  We decided to horribly overcomplicate matters, and create vastly customisable logging. LogManager and LogStream form the visible basis
  85  *  of the new interface. Basically, a LogStream can be inherited to do different things with logging output. We inherit from it once in core
  86  *  to create a FileLogStream, that writes to a file, for example. Different LogStreams can hook different types of log messages, and different
  87  *  levels of output too, for extreme customisation. Multiple LogStreams can hook the same message/levels of output, meaning that e.g. output
  88  *  can go to a channel as well as a file.
  89  *
  90  *  HOW THIS WORKS
  91  *   LogManager handles all instances of LogStreams, classes derived from LogStream are instantiated and passed to it.
  92  */
  93
  94 /** LogStream base class. Modules (and other stuff) inherit from this to decide what logging they are interested in, and what to do with it.
  95  */
  96 class CoreExport LogStream : public classbase
  97 {
  98  protected:
  99         InspIRCd *ServerInstance;
 100         int loglvl;
 101  public:
 102         LogStream(InspIRCd *Instance, int loglevel) : ServerInstance(Instance), loglvl(loglevel)
 103         {
 104         }
 105
 106         /* A LogStream's destructor should do whatever it needs to close any resources it was using (or indicate that it is no longer using a resource
 107          * in the event that the resource is shared, see for example FileLogStream).
 108          */
 109         virtual ~LogStream() { }
 110
 111         /** Changes the loglevel for this LogStream on-the-fly.
 112          * This is needed for -nofork. But other LogStreams could use it to change loglevels.
 113          */
 114         void ChangeLevel(int lvl) { this->loglvl = lvl; }
 115
 116         /** Called when there is stuff to log for this particular logstream. The derived class may take no action with it, or do what it
 117          * wants with the output, basically. loglevel and type are primarily for informational purposes (the level and type of the event triggered)
 118          * and msg is, of course, the actual message to log.
 119          */
 120         virtual void OnLog(int loglevel, const std::string &type, const std::string &msg) = 0;
 121 };
 122
 123 typedef std::map<FileWriter*, int> FileLogMap;
 124
 125 class CoreExport LogManager : public classbase
 126 {
 127  private:
 128         /** Lock variable, set to true when a log is in progress, which prevents further loggging from happening and creating a loop.
 129          */
 130         bool Logging;
 131
 132         /** LogStream for -nofork, logs to STDOUT when it's active.
 133          */
 134         LogStream* noforkstream;
 135
 136         InspIRCd *ServerInstance;
 137
 138         /** Map of active log types and what LogStreams will receive them.
 139          */
 140         std::map<std::string, std::vector<LogStream *> > LogStreams;
 141
 142         /** Refcount map of all LogStreams managed by LogManager.
 143          * If a logstream is not listed here, it won't be automatically closed by LogManager, even if it's loaded in one of the other lists.
 144          */
 145         std::map<LogStream *, int> AllLogStreams;
 146
 147         /** LogStreams with type * (which means everything), and a list a logtypes they are excluded from (eg for "* -USERINPUT -USEROUTPUT").
 148          */
 149         std::map<LogStream *, std::vector<std::string> > GlobalLogStreams;
 150
 151         /** Refcounted map of all FileWriters in use by FileLogStreams, for file stream sharing.
 152          */
 153         FileLogMap FileLogs;
 154
 155  public:
 156
 157         LogManager(InspIRCd *Instance)
 158         {
 159                 noforkstream = NULL;
 160                 ServerInstance = Instance;
 161                 Logging = false;
 162         }
 163
 164         ~LogManager()
 165         {
 166                 if (noforkstream)
 167                         delete noforkstream;
 168         }
 169
 170         /** Sets up the logstream for -nofork. Called by InspIRCd::OpenLog() and LogManager::OpenFileLogs().
 171          * First time called it creates the nofork stream and stores it in noforkstream. Each call thereafter just readds it to GlobalLogStreams
 172          * and updates the loglevel.
 173          */
 174         void SetupNoFork();
 175
 176         /** Adds a FileWriter instance to LogManager, or increments the reference count of an existing instance.
 177          * Used for file-stream sharing for FileLogStreams.
 178          */
 179         void AddLoggerRef(FileWriter* fw)
 180         {
 181                 FileLogMap::iterator i = FileLogs.find(fw);
 182                 if (i == FileLogs.end())
 183                 {
 184                         FileLogs.insert(std::make_pair(fw, 1));
 185                 }
 186                 else
 187                 {
 188                         ++i->second;
 189                 }
 190         }
 191
 192         /** Indicates that a FileWriter reference has been removed. Reference count is decreased, and if zeroed, the FileWriter is closed.
 193          */
 194         void DelLoggerRef(FileWriter* fw)
 195         {
 196                 FileLogMap::iterator i = FileLogs.find(fw);
 197                 if (i == FileLogs.end()) return; /* Maybe should log this? */
 198                 if (--i->second < 1)
 199                 {
 200                         delete i->first;
 201                         FileLogs.erase(i);
 202                 }
 203         }
 204
 205         /** Opens all logfiles defined in the configuration file using <log method="file">.
 206          */
 207         void OpenFileLogs();
 208
 209         /** Removes all LogStreams, meaning they have to be readded for logging to continue.
 210          * Only LogStreams that were listed in AllLogStreams are actually closed.
 211          */
 212         void CloseLogs();
 213
 214         /** Adds a single LogStream to multiple logtypes.
 215          * This automatically handles things like "* -USERINPUT -USEROUTPUT" to mean all but USERINPUT and USEROUTPUT types.
 216          * It is not a good idea to mix values of autoclose for the same LogStream.
 217          * @param type The type string (from configuration, or whatever) to parse.
 218          * @param l The LogStream to add.
 219          * @param autoclose True to have the LogStream automatically closed when all references to it are removed from LogManager. False to leave it open.
 220          */
 221         void AddLogTypes(const std::string &type, LogStream *l, bool autoclose);
 222
 223         /** Registers a new logstream into the logging core, so it can be called for future events
 224          * It is not a good idea to mix values of autoclose for the same LogStream.
 225          * @param type The type to add this LogStream to.
 226          * @param l The LogStream to add.
 227          * @param autoclose True to have the LogStream automatically closed when all references to it are removed from LogManager. False to leave it open.
 228          * @return True if the LogStream was added successfully, False otherwise.
 229          */
 230         bool AddLogType(const std::string &type, LogStream *l, bool autoclose);
 231
 232         /** Removes a logstream from the core. After removal, it will not recieve further events.
 233          * If the LogStream was ever added with autoclose, it will be closed after this call (this means the pointer won't be valid anymore).
 234          */
 235         void DelLogStream(LogStream* l);
 236
 237         /** Removes a LogStream from a single type. If the LogStream has been registered for "*" it will still receive the type unless you remove it from "*" specifically.
 238          * If the LogStream was added with autoclose set to true, then when the last occurrence of the stream is removed it will automatically be closed (freed).
 239          */
 240         bool DelLogType(const std::string &type, LogStream *l);
 241
 242         /** Logs an event, sending it to all LogStreams registered for the type.
 243          * @param type Log message type (ex: "USERINPUT", "MODULE", ...)
 244          * @param loglevel Log message level (DEBUG, VERBOSE, DEFAULT, SPARSE, NONE)
 245          * @param msg The message to be logged (literal).
 246          */
 247         void Log(const std::string &type, int loglevel, const std::string &msg);
 248
 249         /** Logs an event, sending it to all LogStreams registered for the type.
 250          * @param type Log message type (ex: "USERINPUT", "MODULE", ...)
 251          * @param loglevel Log message level (DEBUG, VERBOSE, DEFAULT, SPARSE, NONE)
 252          * @param msg The format of the message to be logged. See your C manual on printf() for details.
 253          */
 254         void Log(const std::string &type, int loglevel, const char *fmt, ...) CUSTOM_PRINTF(4, 5);
 255 };
 256
 257 #endif