VFS_logger.cpp

Go to the documentation of this file.

 
#include <iostream>
 
#include <QtMath>
#include <QDateTime>
#include <QCoreApplication>
 
#include "defines.h"
#include "VFS.h"
#include "VFS_logger.h"
#include "VFS_filetypes/VFS_stream.h"
 
VFS_logger::VFS_logger(int logLevel, int logTime, QString adminEmail)
: VFS_node()
, _logLevel(logLevel)
, _logTime(logTime)
, _adminEmail(adminEmail)
, cp_bold(UNDEF,BOLD)
, cp_warn(YELLOW, BOLD)
, cp_error(RED, BOLD)
{
        _log            = append( "log",        new VFS_stream(655360,true) );
        _critical       = append( "critical",   new VFS_stream(40960,true) );
        _errors         = append( "errors",     new VFS_stream(40960,true) );
        _warnings       = append( "warnings",   new VFS_stream(40960,true) );
        _messages       = append( "messages",   new VFS_stream(655360,true) );
 
        //subscribe _log to the various nodes as a master log
        issueRequest( this, _log->createRequest(VFS_request::subscribe,"messages","logger") );
        issueRequest( this, _log->createRequest(VFS_request::subscribe,"warnings","logger") );
        issueRequest( this, _log->createRequest(VFS_request::subscribe,"errors",  "logger") );
        issueRequest( this, _log->createRequest(VFS_request::subscribe,"critical","logger") );
 
        //subscribe this to the various nodes so they can get to stdout and stderr
        issueRequest( this, createRequest(VFS_request::subscribe,"messages","logger") );
        issueRequest( this, createRequest(VFS_request::subscribe,"warnings","logger") );
        issueRequest( this, createRequest(VFS_request::subscribe,"errors",  "logger") );
        issueRequest( this, createRequest(VFS_request::subscribe,"critical","logger") );
}
 
VFS_logger::~VFS_logger()
{
        //will delete children in the VFS_node destructor
}
 
VFS_node *VFS_logger::find(VFS_request *r)
{
        if (r->_path == "settings")
                return this;
 
        return VFS_node::find(r);
}
 
void VFS_logger::submit(VFS_request *r)
{
        if (r->_path == "settings")
        {
                //printf("Settings: %s\n",qUtf8Printable(r->_data.toJson()));
 
                QDateTime now = QDateTime::currentDateTime();
 
                QJsonObject o = r->_data.object();
 
                if (o.contains("adminemail"))
                {       QString e = o["adminemail"].toString();
                        _adminEmail = e;
 
                        QJsonObject m;
                        m["data"] = QString("%1,LOG,%2,%3,Set admin email to '%4'\n").arg(now.toMSecsSinceEpoch()).arg(r->_user).arg(_logLevel).arg(e);
                        VFS_request *s = createRequest(VFS_request::diff,"",r->_user,QJsonDocument(m));
                        _messages->applyDiff(s);
                        delete s;
                        //VFS::LOG( QString("Set admin email to '%1'").arg(e),0,"logger");
                }
 
                if (o.contains("logtime"))
                {       int l = qBound(0,o["logtime"].toInt(),4);
                        _logTime = l;
 
                        QJsonObject m;
                        m["data"] = QString("%1,LOG,%2,%3,Set log time format to %4\n").arg(now.toMSecsSinceEpoch()).arg(r->_user).arg(_logLevel).arg(l);
                        VFS_request *s = createRequest(VFS_request::diff,"",r->_user,QJsonDocument(m));
                        _messages->applyDiff(s);
                        delete s;
                        //VFS::LOG( QString("Set log time format to %1").arg(l),0,"logger");
                }
 
                if (o.contains("loglevel"))
                {       int l = qBound(0,o["loglevel"].toInt(),9);
                        _logLevel = l;
 
                        QJsonObject m;
                        m["data"] = QString("%1,LOG,%2,%3,Set log level to %4\n").arg(now.toMSecsSinceEpoch()).arg(r->_user).arg(_logLevel).arg(l);
                        VFS_request *s = createRequest(VFS_request::diff,"",r->_user,QJsonDocument(m));
                        _messages->applyDiff(s);
                        delete s;
                        //VFS::LOG( QString("Set log level to %1").arg(l),0,"logger");
                }
 
                r->_success = true;
                return;
        }
 
        r->_reason = "Logger Invalid path: "+r->_path;
        r->_success = false;
}
 
//routes data written from subscribed files as diffs out to stdout, stderr, etc.
void VFS_logger::applyDiff(VFS_request *t)
{
        //QMutexLocker l(&_lock);
 
        VFS_node::applyDiff(t);                         //distribute the diff to subscribers regardless of log level.
 
        VFS_node *origin = t->_origin;
 
        QJsonDocument diffdoc = t->_data;
        QString diff = diffdoc.object()["data"].toString();
 
        QString time = diff.section(",",0,0);
        QByteArray which = diff.section(",",1,1).toUtf8();
        //QByteArray user = diff.section(",",2,2).toUtf8();
        QByteArray level = diff.section(",",3,3).toUtf8();
        QByteArray message = diff.section(",",4).toUtf8();
 
        int ll = qBound(0,level.toInt(),9);
        if (ll > _logLevel)
        {       t->_success = true;
                return;
        }
 
        std::string timeString = "";
        std::string timeOffset = "";
        //QDateTime now = QDateTime::currentDateTime();
        QDateTime now = QDateTime::fromMSecsSinceEpoch( time.toLongLong() );
        switch (_logTime)
        {
                case 0:         timeString = "";                                                        break;
                case 1:         timeString = now.toString("hh:mm:ss.zzz ").toStdString();               break;
                case 2:         timeString = now.toString("yyyy/MM/dd hh:mm:ss:zzz t ").toStdString();  break;
                case 3:         timeString = now.toString("yyyy/MM/dd hh:mm:ss:zzz ").toStdString();
                                {
                                        int offh = qFloor(now.offsetFromUtc()/(60.0*60.0));             //seconds resolved to hours
                                        int offm = qFloor(now.offsetFromUtc()/(60.0)) % 60;             //seconds resolved to minutes, mod-ed by an hour
                                        timeOffset = QString("%1%2:%3 GMT ").arg(offh >= 0 ? "+" : "").arg(offh).arg(offm,2,10,QChar('0')).toStdString();
                                }
                                                                                                        break;
                case 4:
                default:        //timeString = QString("%1").arg(now.toSecsSinceEpoch()).toStdString();
                                timeString = time.toStdString()+" ";
                                                                                                        break;
        }
 
        if (origin == _messages)
                std::cout << cp_bold.cp_s_init() << timeString << timeOffset << "[ LOG  " << level.data() << " ] " << qUtf8Printable(t->_user) << " " << cp_bold.cp_s_rst() << message.data();
        else if (origin == _warnings)
                std::cout << cp_warn.cp_s_init() << timeString << timeOffset << "[ WARN " << level.data() << " ] " << qUtf8Printable(t->_user) << " " << message.data() << cp_warn.cp_s_rst();
        else if (origin == _errors)
                std::cerr << cp_error.cp_s_init() << timeString << timeOffset << "[ ERROR  ] " << qUtf8Printable(t->_user) << " " << message.data() << cp_error.cp_s_rst();
        else if (origin == _critical)
        {
                std::cerr << cp_error.cp_s_init() << timeString << timeOffset << "[ CRITICAL ] " << diff.toUtf8().data() << cp_error.cp_s_rst();
                email(_adminEmail,APPLICATION_NAME" CRITICAL",diff.toUtf8().data());
                std::cerr << cp_error.cp_s_init() << timeString << timeOffset << "             notification email sent to: " << qUtf8Printable(_adminEmail) << std::endl << cp_error.cp_s_rst();
        }
        else
        //      std::cerr << cp_error.cp_s_init() << "[ (" << which.data() << ")   ] " << qUtf8Printable(t->_user) << cp_error.cp_s_rst() << " " << diff.toUtf8().data() << (diff.right(1)!="\n"? "\n" : "");
                std::cerr << cp_error.cp_s_init() << timeString << timeOffset << "[ UNKNOWN (" << which.data() << ") ] " << qUtf8Printable(t->_user) << cp_error.cp_s_rst() << " " << diffdoc.toJson().data() << (diff.right(1)!="\n"? "\n" : "");
 
        std::cout << std::flush;        //to make the output grep-able, we need to flush the streams.
        std::cerr << std::flush;
}
 
void VFS_logger::email(QString address, QString subject, QString message)
{
        //system("mail -s "This is the subject" somebody@example.com <<< 'This is the message'");
        //system("echo "This is the body" | mail -s "Subject" -aFrom:Harry<harry@gmail.com> someone@example.com");
 
        //FIXME: potential security issue here with raw system commands
        //http://stackoverflow.com/questions/288038/how-to-safely-escape-a-string-from-c
        address = address.replace("'","\"'\"").trimmed();
        subject = subject.replace("'","\"'\"").trimmed();
        message = message.replace("'","\"'\"").trimmed();
 
        if (address == "")
        {       VFS::ERROR("Cannot send CRITICAL email when address is empty.  Provide a space-separated list of adminemail addresses in the root VFS env config.\n");
                return;
        }
 
        //QString command = QString("echo '%3' | mail -s '%1' '%2'").arg(subject).arg(address).arg(message);
        QString command = QString("mail -s '%1' '%2' <<< '%3'").arg(subject).arg(address).arg(message);
        printf("%s\n",qUtf8Printable(command));
 
        //int s = QProcess::execute( command );
        int s = system(qUtf8Printable(command));
        if (s != 0)
                VFS::ERROR( QString("Could not send email with command:\n%1").arg(command));
}