projects/smsdaemon/common.cpp

#include "common.h"

#include <string>
#include <iostream>
#include <iomanip>
#include <fstream>
#include <sstream>
#include <time.h>

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

Common* Common::instance()
{
        static Common store;
        return &store;
}


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

        ostringstream out;
        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 (this->isDaemon && _logFile != "")
        {
                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( this->uid);
        }
        else
        {
                cout << out.str();
                cout.flush();
        }
}

void Common::printUsage()
{
        cout << VERSION << endl;
        cout << "Usage --daemon|--debug [arguments]" << endl;
        cout << "Commandline arguments :" << endl;
        cout << "  --daemon : Run in daemon (background) mode" << endl;
        cout << "  --debug :  Run as normal (frontground) process," << endl;
        cout << "             with all messages written to std out" << endl;
        cout << "  --help :    Show this help screen" << endl;

        exit(0);
}

void Common::loadConfig(int argc, char* argv[])
{
        isDaemon = false;
        isDebug = false;

        for (int i = 1; i< argc; i++)
        {
                string current(argv[i]);
        if (current == "--daemon")
                {
                isDaemon = true;
                        isDebug = false;
                        
                }
                else if (current == "--debug")
                {
                        isDaemon = false;
                        isDebug = true;
                }
                else if (current == "-h" || current == "--help")
                {
                        printUsage();
                }
                else
                {
                        cout << VERSION;
                        cout << "Unknown argument : " << current << endl; 
                }
        }

        if (! ( isDaemon || isDebug))
                printUsage();
}


std::string Common::getStatusMessage()
{
        long int now = time(0);

        long int diff = now - this->daemonStart;


        std::ostringstream out;
        out << "Uptime " << diff << " seconds. ";
        out << "Messages received:" << this->smsCounter.incomming << ". ";
        out << "Messages sent:" << this->smsCounter.outgoing << ".";
        
        return out.str();
}
1	#include "common.h"
2
3	#include <string>
4	#include <iostream>
5	#include <iomanip>
6	#include <fstream>
7	#include <sstream>
8	#include <time.h>
9
10	using namespace std;
11	char* Months[] = { "", "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov","Dec"};
12
13	Common* Common::instance()
14	{
15	static Common store;
16	return &store;
17	}
18
19
20	void Common::logMessage(string msg)
21	{
22	time_t t = time(0);
23	tm now;
24	localtime_r(&t, &now);
25
26	ostringstream out;
27	out << Months[ now.tm_mon ] << " " << setw(2) << setfill('0') << now.tm_mday;
28	out << " " << setw(2) << setfill('0') << now.tm_hour;
29	out << ":" << setw(2) << setfill('0') << now.tm_min;
30	out << ":" << setw(2) << setfill('0') << now.tm_sec;
31	out << " " << msg << endl;
32
33	if (this->isDaemon && _logFile != "")
34	{
35	seteuid(0);
36	ofstream of( _logFile.c_str(), ios_base::out \| ios_base::app \| ios_base::ate);//append mode
37	if (of)
38	{
39	of << out.str();
40	of.flush();
41	of.close();
42	}
43	seteuid( this->uid);
44	}
45	else
46	{
47	cout << out.str();
48	cout.flush();
49	}
50	}
51
52	void Common::printUsage()
53	{
54	cout << VERSION << endl;
55	cout << "Usage --daemon\|--debug [arguments]" << endl;
56	cout << "Commandline arguments :" << endl;
57	cout << " --daemon : Run in daemon (background) mode" << endl;
58	cout << " --debug : Run as normal (frontground) process," << endl;
59	cout << " with all messages written to std out" << endl;
60	cout << " --help : Show this help screen" << endl;
61
62	exit(0);
63	}
64
65	void Common::loadConfig(int argc, char* argv[])
66	{
67	isDaemon = false;
68	isDebug = false;
69
70	for (int i = 1; i< argc; i++)
71	{
72	string current(argv[i]);
73	if (current == "--daemon")
74	{
75	isDaemon = true;
76	isDebug = false;
77
78	}
79	else if (current == "--debug")
80	{
81	isDaemon = false;
82	isDebug = true;
83	}
84	else if (current == "-h" \|\| current == "--help")
85	{
86	printUsage();
87	}
88	else
89	{
90	cout << VERSION;
91	cout << "Unknown argument : " << current << endl;
92	}
93	}
94
95	if (! ( isDaemon \|\| isDebug))
96	printUsage();
97	}
98
99
100	std::string Common::getStatusMessage()
101	{
102	long int now = time(0);
103
104	long int diff = now - this->daemonStart;
105
106
107	std::ostringstream out;
108	out << "Uptime " << diff << " seconds. ";
109	out << "Messages received:" << this->smsCounter.incomming << ". ";
110	out << "Messages sent:" << this->smsCounter.outgoing << ".";
111
112	return out.str();
113	}