projects/smsdaemon/Logger.cpp

#include "Logger.h"

#include "Common.h"
#include "ConfigFile.h"
#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>

#include <stdlib.h>
#include <syslog.h>

using namespace std;

const char* Months[] = {  "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov","Dec"};

namespace Logger
{
        string _logFile;

        enum LogDest 
        {
                LogFile,
                LogSyslog,
                LogStdout
        };

        LogDest logger;

        void initLog()
        {
                Common* cmn = Common::instance();
                string logmethod = cmn->GetConfigfile()->GetValue("smsdaemon","log");
                if (logmethod == "syslog") 
                {
                        logger = LogSyslog;

                        openlog("smsdaemon", 0 , LOG_DAEMON);
                } 
                else if (logmethod == "logfile")
                {
                        logger = LogFile;
                } 
                else
                {
                        cerr << "Invalid config file setting for 'log' : " << logmethod << endl;
                        exit(1);
                }

                if (cmn->isDebug) 
                {
                        logger = LogStdout;
                }
        }

        void setLogfile(std::string file)
        {
                _logFile = file;
        }

        void logMessage(string msg)
        {
                Common* cmn = Common::instance();
                time_t t = time(0);
                tm now;
                localtime_r(&t, &now);

                ostringstream out;
                if (logger == LogFile || logger == LogStdout) 
                {
                        out << Months[ now.tm_mon ] << " " << setw(2) << setfill('0') << now.tm_mday;
                        out << " " << setw(2) << setfill('0') << now.tm_hour;
                        out << ":" << setw(2) << setfill('0') << now.tm_min;
                        out << ":" << setw(2) << setfill('0') << now.tm_sec;
                        out << "  " << msg << endl;

                        if (cmn->isDaemon && _logFile != "" && cmn->daemonized)
                        {
                                seteuid(0);
                                ofstream of( _logFile.c_str(), ios_base::out | ios_base::app | ios_base::ate);//append mode
                                if (of)
                                {
                                        of << out.str();
                                        of.flush();
                                        of.close();
                                }
                                seteuid( cmn->uid);
                        }
                        else
                        {
                                cout << out.str();
                                cout.flush();
                        }
                } 
                else  // logSyslog
                {
                        syslog(LOG_INFO, "%s", msg.c_str());
                }
        }
}
1	#include "Logger.h"
2
3	#include "Common.h"
4	#include "ConfigFile.h"
5	#include <iostream>
6	#include <sstream>
7	#include <fstream>
8	#include <iomanip>
9
10	#include <stdlib.h>
11	#include <syslog.h>
12
13	using namespace std;
14
15	const char* Months[] = { "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov","Dec"};
16
17	namespace Logger
18	{
19	string _logFile;
20
21	enum LogDest
22	{
23	LogFile,
24	LogSyslog,
25	LogStdout
26	};
27
28	LogDest logger;
29
30	void initLog()
31	{
32	Common* cmn = Common::instance();
33	string logmethod = cmn->GetConfigfile()->GetValue("smsdaemon","log");
34	if (logmethod == "syslog")
35	{
36	logger = LogSyslog;
37
38	openlog("smsdaemon", 0 , LOG_DAEMON);
39	}
40	else if (logmethod == "logfile")
41	{
42	logger = LogFile;
43	}
44	else
45	{
46	cerr << "Invalid config file setting for 'log' : " << logmethod << endl;
47	exit(1);
48	}
49
50	if (cmn->isDebug)
51	{
52	logger = LogStdout;
53	}
54	}
55
56	void setLogfile(std::string file)
57	{
58	_logFile = file;
59	}
60
61	void logMessage(string msg)
62	{
63	Common* cmn = Common::instance();
64	time_t t = time(0);
65	tm now;
66	localtime_r(&t, &now);
67
68	ostringstream out;
69	if (logger == LogFile \|\| logger == LogStdout)
70	{
71	out << Months[ now.tm_mon ] << " " << setw(2) << setfill('0') << now.tm_mday;
72	out << " " << setw(2) << setfill('0') << now.tm_hour;
73	out << ":" << setw(2) << setfill('0') << now.tm_min;
74	out << ":" << setw(2) << setfill('0') << now.tm_sec;
75	out << " " << msg << endl;
76
77	if (cmn->isDaemon && _logFile != "" && cmn->daemonized)
78	{
79	seteuid(0);
80	ofstream of( _logFile.c_str(), ios_base::out \| ios_base::app \| ios_base::ate);//append mode
81	if (of)
82	{
83	of << out.str();
84	of.flush();
85	of.close();
86	}
87	seteuid( cmn->uid);
88	}
89	else
90	{
91	cout << out.str();
92	cout.flush();
93	}
94	}
95	else // logSyslog
96	{
97	syslog(LOG_INFO, "%s", msg.c_str());
98	}
99	}
100	}