src/common/logger.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2019 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include <iostream>
  22
  23 #ifdef HAVE_SYSLOG
  24 #    include <syslog.h>
  25 #endif
  26
  27 #include <QByteArray>
  28 #include <QDateTime>
  29 #include <QDebug>
  30 #include <QFile>
  31
  32 #include "logger.h"
  33 #include "quassel.h"
  34 #include "types.h"
  35
  36 namespace {
  37
  38 QByteArray msgWithTime(const Logger::LogEntry& msg)
  39 {
  40     return (msg.timeStamp.toString("yyyy-MM-dd hh:mm:ss ") + msg.message + "\n").toUtf8();
  41 }
  42
  43 }  // namespace
  44
  45 Logger::Logger(QObject* parent)
  46     : QObject(parent)
  47 {
  48     static bool registered = []() {
  49         qRegisterMetaType<LogEntry>();
  50         return true;
  51     }();
  52     Q_UNUSED(registered)
  53
  54     connect(this, &Logger::messageLogged, this, &Logger::onMessageLogged);
  55
  56     qInstallMessageHandler(Logger::messageHandler);
  57 }
  58
  59 Logger::~Logger()
  60 {
  61     // If we're not initialized yet, output pending messages so they don't get lost
  62     if (!_initialized) {
  63         for (auto&& message : _messages) {
  64             std::cerr << msgWithTime(message).constData();
  65         }
  66     }
  67 }
  68
  69 std::vector<Logger::LogEntry> Logger::messages() const
  70 {
  71     return _messages;
  72 }
  73
  74 void Logger::setup(bool keepMessages)
  75 {
  76     _keepMessages = keepMessages;
  77
  78     // Set maximum level for output (we still store/announce all messages for client-side filtering)
  79     if (Quassel::isOptionSet("loglevel")) {
  80         QString level = Quassel::optionValue("loglevel").toLower();
  81         if (level == "debug")
  82             _outputLevel = LogLevel::Debug;
  83         else if (level == "info")
  84             _outputLevel = LogLevel::Info;
  85         else if (level == "warning")
  86             _outputLevel = LogLevel::Warning;
  87         else if (level == "error")
  88             _outputLevel = LogLevel::Error;
  89         else {
  90             throw ExitException{EXIT_FAILURE, qPrintable(tr("Invalid log level %1; supported are Debug|Info|Warning|Error").arg(level))};
  91         }
  92     }
  93
  94     QString logfilename = Quassel::optionValue("logfile");
  95     if (!logfilename.isEmpty()) {
  96         _logFile.setFileName(logfilename);
  97         if (!_logFile.open(QFile::Append | QFile::Unbuffered | QFile::Text)) {
  98             qCritical() << qPrintable(tr("Could not open log file \"%1\": %2").arg(logfilename, _logFile.errorString()));
  99         }
 100     }
 101     if (!_logFile.isOpen()) {
 102         if (!_logFile.open(stderr, QFile::WriteOnly | QFile::Unbuffered | QFile::Text)) {
 103             qCritical() << qPrintable(tr("Cannot write to stderr: %1").arg(_logFile.errorString()));
 104         }
 105     }
 106
 107 #ifdef HAVE_SYSLOG
 108     _syslogEnabled = Quassel::isOptionSet("syslog");
 109 #endif
 110
 111     _initialized = true;
 112
 113     // Now that we've setup our logging backends, output pending messages
 114     for (auto&& message : _messages) {
 115         outputMessage(message);
 116     }
 117     if (!_keepMessages) {
 118         _messages.clear();
 119     }
 120 }
 121
 122 void Logger::messageHandler(QtMsgType type, const QMessageLogContext&, const QString& message)
 123 {
 124     Quassel::instance()->logger()->handleMessage(type, message);
 125 }
 126
 127 void Logger::handleMessage(QtMsgType type, const QString& msg)
 128 {
 129     switch (type) {
 130     case QtDebugMsg:
 131         handleMessage(LogLevel::Debug, msg);
 132         break;
 133     case QtInfoMsg:
 134         handleMessage(LogLevel::Info, msg);
 135         break;
 136     case QtWarningMsg:
 137         handleMessage(LogLevel::Warning, msg);
 138         break;
 139     case QtCriticalMsg:
 140         handleMessage(LogLevel::Error, msg);
 141         break;
 142     case QtFatalMsg:
 143         handleMessage(LogLevel::Fatal, msg);
 144         break;
 145     }
 146 }
 147
 148 void Logger::handleMessage(LogLevel level, const QString& msg)
 149 {
 150     QString logString;
 151
 152     switch (level) {
 153     case LogLevel::Debug:
 154         logString = "[Debug] ";
 155         break;
 156     case LogLevel::Info:
 157         logString = "[Info ] ";
 158         break;
 159     case LogLevel::Warning:
 160         logString = "[Warn ] ";
 161         break;
 162     case LogLevel::Error:
 163         logString = "[Error] ";
 164         break;
 165     case LogLevel::Fatal:
 166         logString = "[FATAL] ";
 167         break;
 168     }
 169
 170     // Use signal connection to make this method thread-safe
 171     emit messageLogged({QDateTime::currentDateTime(), level, logString += msg});
 172 }
 173
 174 void Logger::onMessageLogged(const LogEntry& message)
 175 {
 176     if (_keepMessages) {
 177         _messages.push_back(message);
 178     }
 179
 180     // If setup() wasn't called yet, just store the message - will be output later
 181     if (_initialized) {
 182         outputMessage(message);
 183     }
 184 }
 185
 186 void Logger::outputMessage(const LogEntry& message)
 187 {
 188     if (message.logLevel < _outputLevel) {
 189         return;
 190     }
 191
 192 #ifdef HAVE_SYSLOG
 193     if (_syslogEnabled) {
 194         int prio{LOG_INFO};
 195         switch (message.logLevel) {
 196         case LogLevel::Debug:
 197             prio = LOG_DEBUG;
 198             break;
 199         case LogLevel::Info:
 200             prio = LOG_INFO;
 201             break;
 202         case LogLevel::Warning:
 203             prio = LOG_WARNING;
 204             break;
 205         case LogLevel::Error:
 206             prio = LOG_ERR;
 207             break;
 208         case LogLevel::Fatal:
 209             prio = LOG_CRIT;
 210         }
 211         syslog(prio | LOG_USER, "%s", qPrintable(message.message));
 212     }
 213 #endif
 214
 215     if (!_logFile.fileName().isEmpty() || !_syslogEnabled) {
 216         _logFile.write(msgWithTime(message));
 217     }
 218
 219 #ifndef Q_OS_MAC
 220     // For fatal messages, write log to dump file
 221     if (message.logLevel == LogLevel::Fatal) {
 222         QFile dumpFile{Quassel::instance()->coreDumpFileName()};
 223         if (dumpFile.open(QIODevice::Append)) {
 224             dumpFile.write(msgWithTime(message));
 225             dumpFile.close();
 226         }
 227     }
 228 #endif
 229 }