host/c%2B%2B/Host.cpp

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org.uk
 * Copyright (C) 2000-2002 i-scream
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#include <iostream.h>
#include <unistd.h> // linux header for the sleep function
#include <stdlib.h> // itoa()
#include <time.h>
#include "SysMon.h"
// #include "Config.h" - redefinition error - turns out i don't need it
#ifndef  XMLFORMATTER_H         /* Has the file been INCLUDE'd already? */
    #define  XMLFORMATTER_H  yes
    #include "XMLFormatter.h"
#endif

using std::string;

int main(){
        
        // list of local variables
        int noErrors = 1;
        int reloadConfig = 0;
        
        // print some extra info..
        int debug = 0;
        
        // create a new connection with the Config Host
        // values hard coded for the moment
        std::cout << "C Host starting..\n";
        
        std::cout << "Attemping to load config\n";
        Config config("raptor.ukc.ac.uk",4567,debug);           // host, port, debug (0/1)
        
        std::cout << "Creating data collector\n";
        SysMon sysMon(config, debug);
        
        std::cout << "Configuring UDPUpdateTime\n";
        int UDPUpdateTime = config.getIntProperty("UDPUpdateTime");
        if (  UDPUpdateTime == -1 ){
                if ( debug == 1 ){
                        std::cout << "Using Default time!\n";
                        UDPUpdateTime = 5;
                } // if
        } // if
                        
                        
        std::cout << "Configuring TCPUpdateTime\n";
        int TCPUpdateTime = config.getIntProperty("TCPUpdateTime");
        if ( TCPUpdateTime  == -1){
                if ( debug == 1 ){
                        std::cout << "Using Default time!\n";
                        TCPUpdateTime = 10;
                } // if
        } // if
        
        std::cout << "Configuring AveragerUpdateTime\n";
        int AveragerUpdateTime = config.getIntProperty("AVERAGERUpdateTime");
        if ( AveragerUpdateTime == -1){
                if ( debug == 1 ){
                        std::cout << "Using Default time!\n";
                } // if
                AveragerUpdateTime = 2;
        } // if
        
        std::cout << "Configuring Filter Retry Tolerances\n";
        int MaxFilterRetrys = config.getIntProperty("MAXTCPFilterRetries");
        if ( MaxFilterRetrys == -1){
                if ( debug == 1 ){
                        std::cout << "Using Default Tolerance\n";
                } // if
                MaxFilterRetrys = 1;
        } // if
        
        
        // std::cout << "Configuring Alerter\n";
        // Set the values at which the Alerter should kick in. and how many packets it should
        // send
        
        // now do the data collection
        
        std::cout << "Starting Data Collection\n";
        // work out which is closer, the next heartbeat or the next data collection
        int localTime = time(NULL);
        // std::cout << "#";
        int nextUDPTime = localTime + UDPUpdateTime;
        int nextAveragerTime = localTime + AveragerUpdateTime;
        int nextTCPTime = localTime + TCPUpdateTime;
        int sequence = 0;
                
        int nextAction = 0;
        while ( noErrors == 1){
                // firstly check we haven't got to reload our config
                if ( reloadConfig == 1 ){
                        // reload the config by starting a new
                        std::cout << "Forcing reload of Configuration\n";
                        config = Config("raptor.ukc.ac.uk",4567,debug);
                        
                        std::cout << "Configuring UDPUpdateTime\n";
                        int UDPUpdateTime = config.getIntProperty("UDPUpdateTime");
                        if (  UDPUpdateTime == -1 ){
                                if ( debug == 1 ){
                                        std::cout << "Using Default time!\n";
                                        UDPUpdateTime = 5;
                                } // if
                        } // if
                                        
                                        
                        std::cout << "Configuring TCPUpdateTime\n";
                        int TCPUpdateTime = config.getIntProperty("TCPUpdateTime");
                        if ( TCPUpdateTime  == -1){
                                if ( debug == 1 ){
                                        std::cout << "Using Default time!\n";
                                        TCPUpdateTime = 10;
                                } // if
                        } // if
                        
                        std::cout << "Configuring AveragerUpdateTime\n";
                        int AveragerUpdateTime = config.getIntProperty("AVERAGERUpdateTime");
                        if ( AveragerUpdateTime == -1){
                                if ( debug == 1 ){
                                        std::cout << "Using Default time!\n";
                                } // if
                                AveragerUpdateTime = 2;
                        } // if
                        
                        std::cout << "Configuring Filter Retry Tolerances\n";
                        int MaxFilterRetrys = config.getIntProperty("MAXTCPFilterRetries");
                        if ( MaxFilterRetrys == -1){
                                if ( debug == 1 ){
                                        std::cout << "Using Default Tolerance\n";
                                } // if
                                MaxFilterRetrys = 5;
                        } // if
                        // reloadConfig is done now
                        reloadConfig = 0;
                } // if reloadconfig
                
                
                
                localTime = time(NULL);
                // keep going while no non-fatal errors have occurred
                int waitTime = 0;
                
                if ( nextTCPTime  <= nextAveragerTime ){
                        if ( nextTCPTime >= nextUDPTime ){
                                nextAction = 3;
                                waitTime = nextUDPTime - localTime;
                        } else {
                                // set the heartbeat to be the next item
                                nextAction = 1;
                                waitTime = nextTCPTime - localTime;
                        } // if
                } else {
                        if ( nextAveragerTime >=  nextUDPTime ){
                                nextAction = 3;
                                waitTime = nextUDPTime - localTime;
                        } else {
                                // set grabstats to be in the next action
                                nextAction = 2;
                                waitTime = nextAveragerTime - localTime;
                        } // if
                } // if
                
                // sleep until this time only if it is greater than 1!
                if ( waitTime >= 1 ){
                        sleep(waitTime);
                        
                } 
                
                
                localTime = time(NULL);
                int con = -1;
                int conCount = 0;
                
                // do the action
                switch (nextAction){
                        case 1: std::cout << "Scheduled TCP Heartbeat Connection" << endl; 
                                nextTCPTime = localTime + TCPUpdateTime; 
                                // connect to the filter - should do all the talking within config.
                                con = (config.getSmallNet())->connectHeartBeat(config.getUDPFilterName(), config.getFilterPort());                             
                                conCount = 0;
                                while (( con != 0 ) && ( conCount < MaxFilterRetrys )){
                                        int con = (config.getSmallNet())->connectHeartBeat(config.getUDPFilterName(), config.getFilterPort());                          
                                        conCount++;
                                }
                                // if con != 0, then we must reconnect to the main filter manager and 
                                // reload our config!
                                if ( con == 0 ){
                                        // run the chat script
                                        int result = config.chatHeartBeat();
                                        if ( result != 0 ){
                                                // gotta reload our config
                                                reloadConfig = 1;
                                        } // if
                                } else {
                                        // set the config to be reloaded next time
                                        reloadConfig = 1;
                                } // if con =!= 0
                                
                                break;
                        case 2: nextAveragerTime = localTime + AveragerUpdateTime; 
                                /*
                                if ( sysMon.collect() > 0 ){
                                        // suggests that there should be an alert
                                        std::cout << "Alert Status" << endl;                                    
                                    }*/
                                break;
                        default: std::cout << "Scheduled UDP Connection" << endl; 
                             sysMon.collect();
                                 nextUDPTime = localTime + UDPUpdateTime; 
                                 // package up the system data
                                 string host = config.getFQDN();
                                 // string ip = (config.getSmallNet())->getHostIP();
                                 
                                 string seq = "";
                                 // char * buffer;
                                 char buffer[32];
                                 sprintf(buffer,"%u", sequence); 
                                 seq = buffer;
                                 string date = "";
                                 sprintf(buffer,"%u", localTime); 
                                 date = buffer;
                         string attributes = "type=\"data\" machine_name=\""+host+"\" ip=\"0.0.0.0\" date=\""+date+"\" seq_no=\""+seq+"\"";

                                 XMLFormatter xml("packet",attributes);
                                 // put in the data
                                 xml.addString(sysMon.getData());
                                 (config.getSmallNet())->sendUPDPacket(config.getUDPFilterName(), config.getUDPFilterPort(), xml.returnXML());
                                 // now clear the averaged data
                                 sysMon.clearData();
                                 sequence++;
                                 break;
                } // switch
                
        } // while
        
        std::cout << "A fatal error occurred which caused this program to quit\n";
        std::cout << "It is suggested that all belly button lint should be removed\n";
        std::cout << "prior to use of this program\n";
        std::cout << "Oh and BTW, All you BASE are belong to us!\n";
        std::cout << "Error Code: " << noErrors << "\n";
        
        return 0;
        
} // main
Revision:	1.7
Committed:	Tue May 21 16:47:11 2002 UTC (22 years, 4 months ago) by tdb
Branch:	MAIN
Changes since 1.6:	+1 -0 lines
Log Message:	Added URL to GPL headers.
#	User	Rev	Content
1	tdb	1.6	/*
2			* i-scream central monitoring system
3	tdb	1.7	* http://www.i-scream.org.uk
4	tdb	1.6	* Copyright (C) 2000-2002 i-scream
5			*
6			* This program is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU General Public License
8			* as published by the Free Software Foundation; either version 2
9			* of the License, or (at your option) any later version.
10			*
11			* This program is distributed in the hope that it will be useful,
12			* but WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU General Public License for more details.
15			*
16			* You should have received a copy of the GNU General Public License
17			* along with this program; if not, write to the Free Software
18			* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19			*/
20
21	ab11	1.1	#include <iostream.h>
22			#include <unistd.h> // linux header for the sleep function
23	ab11	1.2	#include <stdlib.h> // itoa()
24	ab11	1.1	#include <time.h>
25			#include "SysMon.h"
26			// #include "Config.h" - redefinition error - turns out i don't need it
27			#ifndef XMLFORMATTER_H /* Has the file been INCLUDE'd already? */
28			#define XMLFORMATTER_H yes
29			#include "XMLFormatter.h"
30			#endif
31
32			using std::string;
33
34			int main(){
35
36			// list of local variables
37			int noErrors = 1;
38	ab11	1.2	int reloadConfig = 0;
39	ab11	1.1
40			// print some extra info..
41			int debug = 0;
42
43			// create a new connection with the Config Host
44			// values hard coded for the moment
45			std::cout << "C Host starting..\n";
46
47			std::cout << "Attemping to load config\n";
48	ab11	1.2	Config config("raptor.ukc.ac.uk",4567,debug); // host, port, debug (0/1)
49	ab11	1.1
50			std::cout << "Creating data collector\n";
51			SysMon sysMon(config, debug);
52
53			std::cout << "Configuring UDPUpdateTime\n";
54			int UDPUpdateTime = config.getIntProperty("UDPUpdateTime");
55			if ( UDPUpdateTime == -1 ){
56			if ( debug == 1 ){
57			std::cout << "Using Default time!\n";
58			UDPUpdateTime = 5;
59			} // if
60			} // if
61
62	ab11	1.2
63	ab11	1.1	std::cout << "Configuring TCPUpdateTime\n";
64			int TCPUpdateTime = config.getIntProperty("TCPUpdateTime");
65			if ( TCPUpdateTime == -1){
66			if ( debug == 1 ){
67			std::cout << "Using Default time!\n";
68			TCPUpdateTime = 10;
69			} // if
70			} // if
71
72			std::cout << "Configuring AveragerUpdateTime\n";
73			int AveragerUpdateTime = config.getIntProperty("AVERAGERUpdateTime");
74			if ( AveragerUpdateTime == -1){
75			if ( debug == 1 ){
76			std::cout << "Using Default time!\n";
77			} // if
78			AveragerUpdateTime = 2;
79			} // if
80
81	ab11	1.2	std::cout << "Configuring Filter Retry Tolerances\n";
82			int MaxFilterRetrys = config.getIntProperty("MAXTCPFilterRetries");
83			if ( MaxFilterRetrys == -1){
84			if ( debug == 1 ){
85			std::cout << "Using Default Tolerance\n";
86			} // if
87	ab11	1.5	MaxFilterRetrys = 1;
88	ab11	1.2	} // if
89
90
91	ab11	1.5	// std::cout << "Configuring Alerter\n";
92	ab11	1.1	// Set the values at which the Alerter should kick in. and how many packets it should
93			// send
94
95			// now do the data collection
96
97	ab11	1.2	std::cout << "Starting Data Collection\n";
98	ab11	1.1	// work out which is closer, the next heartbeat or the next data collection
99	ab11	1.3	int localTime = time(NULL);
100	ab11	1.2	// std::cout << "#";
101	ab11	1.1	int nextUDPTime = localTime + UDPUpdateTime;
102			int nextAveragerTime = localTime + AveragerUpdateTime;
103			int nextTCPTime = localTime + TCPUpdateTime;
104	ab11	1.3	int sequence = 0;
105	ab11	1.2
106	ab11	1.1	int nextAction = 0;
107			while ( noErrors == 1){
108	ab11	1.2	// firstly check we haven't got to reload our config
109			if ( reloadConfig == 1 ){
110			// reload the config by starting a new
111			std::cout << "Forcing reload of Configuration\n";
112			config = Config("raptor.ukc.ac.uk",4567,debug);
113
114			std::cout << "Configuring UDPUpdateTime\n";
115			int UDPUpdateTime = config.getIntProperty("UDPUpdateTime");
116			if ( UDPUpdateTime == -1 ){
117			if ( debug == 1 ){
118			std::cout << "Using Default time!\n";
119			UDPUpdateTime = 5;
120			} // if
121			} // if
122
123
124			std::cout << "Configuring TCPUpdateTime\n";
125			int TCPUpdateTime = config.getIntProperty("TCPUpdateTime");
126			if ( TCPUpdateTime == -1){
127			if ( debug == 1 ){
128			std::cout << "Using Default time!\n";
129			TCPUpdateTime = 10;
130			} // if
131			} // if
132
133			std::cout << "Configuring AveragerUpdateTime\n";
134			int AveragerUpdateTime = config.getIntProperty("AVERAGERUpdateTime");
135			if ( AveragerUpdateTime == -1){
136			if ( debug == 1 ){
137			std::cout << "Using Default time!\n";
138			} // if
139			AveragerUpdateTime = 2;
140			} // if
141
142			std::cout << "Configuring Filter Retry Tolerances\n";
143			int MaxFilterRetrys = config.getIntProperty("MAXTCPFilterRetries");
144			if ( MaxFilterRetrys == -1){
145			if ( debug == 1 ){
146			std::cout << "Using Default Tolerance\n";
147			} // if
148			MaxFilterRetrys = 5;
149			} // if
150	ab11	1.3	// reloadConfig is done now
151			reloadConfig = 0;
152	ab11	1.2	} // if reloadconfig
153
154
155	ab11	1.3
156	ab11	1.2	localTime = time(NULL);
157	ab11	1.1	// keep going while no non-fatal errors have occurred
158			int waitTime = 0;
159
160			if ( nextTCPTime <= nextAveragerTime ){
161			if ( nextTCPTime >= nextUDPTime ){
162			nextAction = 3;
163	ab11	1.2	waitTime = nextUDPTime - localTime;
164	ab11	1.1	} else {
165			// set the heartbeat to be the next item
166			nextAction = 1;
167	ab11	1.2	waitTime = nextTCPTime - localTime;
168	ab11	1.1	} // if
169			} else {
170			if ( nextAveragerTime >= nextUDPTime ){
171			nextAction = 3;
172	ab11	1.2	waitTime = nextUDPTime - localTime;
173	ab11	1.1	} else {
174			// set grabstats to be in the next action
175			nextAction = 2;
176	ab11	1.2	waitTime = nextAveragerTime - localTime;
177	ab11	1.1	} // if
178			} // if
179
180			// sleep until this time only if it is greater than 1!
181			if ( waitTime >= 1 ){
182	ab11	1.4	sleep(waitTime);
183	ab11	1.2
184	ab11	1.1	}
185
186	ab11	1.2
187	ab11	1.1	localTime = time(NULL);
188	ab11	1.2	int con = -1;
189			int conCount = 0;
190
191	ab11	1.1	// do the action
192			switch (nextAction){
193	ab11	1.2	case 1: std::cout << "Scheduled TCP Heartbeat Connection" << endl;
194	ab11	1.1	nextTCPTime = localTime + TCPUpdateTime;
195	ab11	1.2	// connect to the filter - should do all the talking within config.
196			con = (config.getSmallNet())->connectHeartBeat(config.getUDPFilterName(), config.getFilterPort());
197			conCount = 0;
198			while (( con != 0 ) && ( conCount < MaxFilterRetrys )){
199			int con = (config.getSmallNet())->connectHeartBeat(config.getUDPFilterName(), config.getFilterPort());
200			conCount++;
201			}
202			// if con != 0, then we must reconnect to the main filter manager and
203			// reload our config!
204			if ( con == 0 ){
205			// run the chat script
206			int result = config.chatHeartBeat();
207			if ( result != 0 ){
208			// gotta reload our config
209			reloadConfig = 1;
210			} // if
211			} else {
212			// set the config to be reloaded next time
213			reloadConfig = 1;
214			} // if con =!= 0
215
216	ab11	1.1	break;
217	ab11	1.2	case 2: nextAveragerTime = localTime + AveragerUpdateTime;
218	ab11	1.4	/*
219	ab11	1.1	if ( sysMon.collect() > 0 ){
220			// suggests that there should be an alert
221	ab11	1.2	std::cout << "Alert Status" << endl;
222	ab11	1.4	}*/
223	ab11	1.1	break;
224	ab11	1.2	default: std::cout << "Scheduled UDP Connection" << endl;
225	ab11	1.4	sysMon.collect();
226	ab11	1.1	nextUDPTime = localTime + UDPUpdateTime;
227			// package up the system data
228	ab11	1.4	string host = config.getFQDN();
229	ab11	1.5	// string ip = (config.getSmallNet())->getHostIP();
230	ab11	1.2
231	ab11	1.3	string seq = "";
232			// char * buffer;
233			char buffer[32];
234			sprintf(buffer,"%u", sequence);
235			seq = buffer;
236			string date = "";
237			sprintf(buffer,"%u", localTime);
238			date = buffer;
239	ab11	1.5	string attributes = "type=\"data\" machine_name=\""+host+"\" ip=\"0.0.0.0\" date=\""+date+"\" seq_no=\""+seq+"\"";
240	ab11	1.1
241			XMLFormatter xml("packet",attributes);
242			// put in the data
243			xml.addString(sysMon.getData());
244			(config.getSmallNet())->sendUPDPacket(config.getUDPFilterName(), config.getUDPFilterPort(), xml.returnXML());
245	ab11	1.4	// now clear the averaged data
246			sysMon.clearData();
247	ab11	1.3	sequence++;
248	ab11	1.1	break;
249			} // switch
250
251			} // while
252
253			std::cout << "A fatal error occurred which caused this program to quit\n";
254			std::cout << "It is suggested that all belly button lint should be removed\n";
255	ab11	1.2	std::cout << "prior to use of this program\n";
256			std::cout << "Oh and BTW, All you BASE are belong to us!\n";
257	ab11	1.1	std::cout << "Error Code: " << noErrors << "\n";
258
259			return 0;
260
261	ab11	1.3	} // main