ViewVC Help

View File | Revision Log | Show Annotations | Revision Graph | Root Listing

root/i-scream/projects/cms/source/server/uk/org/iscream/cms/server/filter/TCPReaderInit.java

Comparing projects/cms/source/server/uk/org/iscream/cms/server/filter/TCPReaderInit.java (file contents):
Revision 1.5 by tdb, Wed Nov 29 19:26:00 2000 UTC vs.
Revision 1.30 by tdb, Sat May 18 18:16:02 2002 UTC

#	Line 1 | Line 1
1	+	/*
2	+	* i-scream central monitoring system
3	+	* Copyright (C) 2000-2002 i-scream
4	+	*
5	+	* This program is free software; you can redistribute it and/or
6	+	* modify it under the terms of the GNU General Public License
7	+	* as published by the Free Software Foundation; either version 2
8	+	* of the License, or (at your option) any later version.
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 Free Software
17	+	* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
18	+	*/
19	+
20		//---PACKAGE DECLARATION---
21	<	package uk.ac.ukc.iscream.filter;
21	>	package uk.org.iscream.cms.server.filter;
22
23		//---IMPORTS---
24	<	import uk.ac.ukc.iscream.core.*;
25	<	import uk.ac.ukc.iscream.filter.*;
24	>	import uk.org.iscream.cms.server.core.*;
25	>	import uk.org.iscream.cms.server.filter.*;
26	>	import uk.org.iscream.cms.server.componentmanager.*;
27		import java.net.Socket;
8	–	import java.io.InputStream;
9	–	import java.io.OutputStream;
10	–	import java.io.IOException;
28		import java.io.*;
29	<	import org.omg.CORBA.*;
30	<	import org.omg.CosNaming.*;
29	>	import java.util.Random;
30	>	import uk.org.iscream.cms.server.util.*;
31
32		/**
33	<	* <ONE LINE DESCRIPTION>
17	<	* <DETAILED DESCRIPTION>
33	>	* This provides Host heartbeat functionality
34		*
35		* @author  $Author$
36		* @version $Id$
#	Line 31 | Line 47 | class TCPReaderInit extends Thread {
47		//---STATIC METHODS---
48
49		//---CONSTRUCTORS---
50	<
51	<	public TCPReaderInit(Socket socket, ConfigurationManager configManager, Logger logRef, Filter parent) throws IOException {
52	<	_configManager = configManager;
53	<	_logRef = logRef;
50	>
51	>	/**
52	>	* Construct a new TCPReaderInit.
53	>	*
54	>	* @param socket the Socket to which the host is connected
55	>	* @param queue the Queue to which we'll add data
56	>	* @throws IOException if something goes badly wrong
57	>	*/
58	>	public TCPReaderInit(Socket socket, Queue queue) throws IOException {
59	>	// set the Thread name
60	>	setName("filter.TCPReaderInit");
61	>
62		_socket = socket;
63	<	_parent = parent;
63	>	_queue = queue;
64	>	// setup the reader & writer
65		_socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
66	<	_socketOut = new PrintWriter(_socket.getOutputStream());
67	<	_logRef.write(toString(), Logger.SYSINIT, "created");
66	>	_socketOut = new PrintWriter(_socket.getOutputStream(), true);
67	>	_logger.write(toString(), Logger.SYSINIT, "created");
68		}
69
70		//---PUBLIC METHODS---
71	<
72	<	//NEED TO MAJORILY CHANGE THIS BIT !
71	>
72	>	/**
73	>	* Main run method. Will communicate with the host, inform it
74	>	* if any updates to it's configuration are needed, and send
75	>	* a heartbeat packet into the system.
76	>	*/
77		public void run() {
78		try {
79	<	//variables
80	<	String filelist = "";
52	<	String lastModified = "";
53	<	String inBound = "";
79	>	// get an instance of the KeyManager
80	>	KeyManager keyman = KeyManager.getInstance();
81
82	<	inBound = _socketIn.readLine();
83	<	if(!inBound.equals("HEARTBEAT")) {
84	<	_socketOut.println("ERROR");
58	<	_socketOut.flush();
59	<	throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
60	<	} else {
61	<	_socketOut.println("OK");
62	<	_socketOut.flush();
63	<	}
82	>	// get some information about the host
83	>	String hostname = _socket.getInetAddress().getHostName().toLowerCase();
84	>	String ipadd = _socket.getInetAddress().getHostAddress();
85
86	<	inBound = _socketIn.readLine();
87	<	if(!inBound.equals("CONFIG")) {
88	<	_socketOut.println("ERROR");
89	<	_socketOut.flush();
90	<	throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
91	<	} else {
92	<	_socketOut.println("OK");
93	<	_socketOut.flush();
86	>	// try for HEARTBEAT
87	>	getInBound("HEARTBEAT");
88	>	_socketOut.println("OK");
89	>
90	>	// look for a command:
91	>	// CONFIG - to check config
92	>	// KEY - to get the key
93	>	// ENDHEARTBEAT - to finish
94	>	String cmd = getInBound();
95	>	while(!cmd.equals("ENDHEARTBEAT")) {
96	>	if(cmd.equals("CONFIG")) {
97	>	// respond to CONFIG
98	>	_socketOut.println("OK");
99	>
100	>	// try for {filelist}
101	>	String filelist = getInBound();
102	>	_socketOut.println("OK");
103	>
104	>	// try for {lastModified}
105	>	String lastModified = getInBound();
106	>	// check to see if a config update has happen
107	>	boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
108	>	if(newConfig) {
109	>	// new config !
110	>	_socketOut.println("ERROR");
111	>	}
112	>	else {
113	>	// nothing has changed
114	>	_socketOut.println("OK");
115	>	}
116	>	}
117	>	else if(cmd.equals("KEY")) {
118	>	// repsond to KEY
119	>	// generate a key
120	>	String key = keyman.genKey();
121	>	// send key to host
122	>	_socketOut.println(key);
123	>	// add it to the key manager
124	>	keyman.addKey(hostname, key);
125	>	}
126	>	else {
127	>	_socketOut.println("ERROR");
128	>	}
129	>	// get the next command
130	>	cmd = getInBound();
131		}
132
133	<	inBound = _socketIn.readLine();
76	<	filelist = inBound;
133	>	// respond to ENDHEARTBEAT
134		_socketOut.println("OK");
78	–	_socketOut.flush();
135
136	<	inBound = _socketIn.readLine();
137	<	lastModified = inBound;
136	>	// work out some information for our heartbeat packet
137	>	String date = new Long(System.currentTimeMillis()/((long) 1000)).toString(); //seconds
138
139	<	boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
139	>	// run the service checks for this host
140	>	_logger.write(toString(), Logger.DEBUG, "Running service checks");
141	>	String checks = PluginServiceCheckManager.getInstance().runServiceChecks(hostname);
142
143	<	if(newConfig) {
144	<	_socketOut.println("ERROR");
87	<	}
88	<	else {
89	<	_socketOut.println("OK");
90	<	}
91	<	_socketOut.flush();
143	>	// build the heartbeat packet
144	>	String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\">" + checks + "</packet>";
145
146	<	inBound = _socketIn.readLine();
147	<	if(!inBound.equals("ENDHEARTBEAT")) {
95	<	_socketOut.println("ERROR");
96	<	_socketOut.flush();
97	<	throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
98	<	} else {
99	<	_socketOut.println("OK");
100	<	_socketOut.flush();
101	<	}
102	<
103	<	// NEED TO LOG THE HEARTBEAT IN XML HERE.
146	>	// get it to be sent on
147	>	_queue.add(xml);
148
149		} catch (Exception e) {
150	<	_logRef.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
150	>	_logger.write(toString(), Logger.ERROR, "ERROR: " + e);
151		}
152
109	–	_socketOut.flush();
153		// Disconnect streams & socket
154		try {
155		_socketIn.close();
156		_socketOut.close();
157		_socket.close();
158		} catch (IOException e) {
159	<	_logRef.write(toString(), Logger.ERROR, "exception on socket close");
159	>	_logger.write(toString(), Logger.ERROR, "exception on socket close");
160		}
161	<	_logRef.write(toString(), Logger.SYSMSG, "finished");
161	>	_logger.write(toString(), Logger.DEBUG, "finished");
162		}
163
164		/**
165		* Overrides the {@link java.lang.Object#toString() Object.toString()}
166		* method to provide clean logging (every class should have this).
167		*
168	+	* This uses the uk.org.iscream.cms.server.util.NameFormat class
169	+	* to format the toString()
170	+	*
171		* @return the name of this class and its CVS revision
172		*/
173		public String toString() {
174	<	return this.getClass().getName() + "{" + _socket.getInetAddress().getHostName()
175	<	+ "}(" + REVISION.substring(11, REVISION.length() - 2) + ")";
176	<
174	>	return FormatName.getName(
175	>	_name,
176	>	getClass().getName(),
177	>	REVISION);
178		}
179
180		//---PRIVATE METHODS---
181
182	+	private String getInBound(String expected) throws IOException {
183	+	// grab the input
184	+	String inBound = getInBound();
185	+	// check if it's what we're expecting
186	+	if(!inBound.equals(expected)) {
187	+	throw new IOException("protocol error from "+_socket.getInetAddress().getHostName()+" - expected:"+expected+" got:" + inBound);
188	+	}
189	+	// it should be ok then
190	+	return inBound;
191	+	}
192	+
193	+	private String getInBound() throws IOException {
194	+	// grab the input
195	+	String inBound = _socketIn.readLine();
196	+	// check for null's, likely disconnection
197	+	if(inBound == null) {
198	+	throw new IOException("got null from host, maybe it died");
199	+	}
200	+	// it's a valid message it seems
201	+	return inBound;
202	+	}
203	+
204		//---ACCESSOR/MUTATOR METHODS---
205
206		//---ATTRIBUTES---
207
208	<	ConfigurationManager _configManager;
209	<	Logger _logRef;
208	>	/**
209	>	* A reference to the configuration manager
210	>	*/
211	>	ConfigurationManager _configManager = ReferenceManager.getInstance().getCM();
212	>
213	>	/**
214	>	* This is the friendly identifier of the
215	>	* component this class is running in.
216	>	* eg, a Filter may be called "filter1",
217	>	* If this class does not have an owning
218	>	* component,  a name from the configuration
219	>	* can be placed here.  This name could also
220	>	* be changed to null for utility classes.
221	>	*/
222	>	private String _name = FilterMain.NAME;
223	>
224	>	/**
225	>	* This holds a reference to the
226	>	* system logger that is being used.
227	>	*/
228	>	private Logger _logger = ReferenceManager.getInstance().getLogger();
229	>
230	>	/**
231	>	* The socket we are talking on
232	>	*/
233		Socket _socket;
234	+
235	+	/**
236	+	* The input from the socket
237	+	*/
238		BufferedReader _socketIn;
239	+
240	+	/**
241	+	* The output from the socket
242	+	*/
243		PrintWriter _socketOut;
244	<	Filter _parent;
244	>
245	>	/**
246	>	* A reference to our Queue
247	>	*/
248	>	Queue _queue;
249	>
250		//---STATIC ATTRIBUTES---
251
252		}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines