server/filter/TCPReaderInit.java

/* 
 * i-scream central monitoring system
 * 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.
 */

//---PACKAGE DECLARATION---
package uk.org.iscream.cms.server.filter;

//---IMPORTS---
import uk.org.iscream.cms.server.core.*;
import uk.org.iscream.cms.server.filter.*;
import uk.org.iscream.cms.server.componentmanager.*;
import java.net.Socket;
import java.io.*;
import java.util.Random;
import uk.org.iscream.cms.server.util.*;

/**
 * This provides Host heartbeat functionality
 *
 * @author  $Author: tdb $
 * @version $Id: TCPReaderInit.java,v 1.29 2002/05/12 15:49:50 tdb Exp $
 */
class TCPReaderInit extends Thread {

//---FINAL ATTRIBUTES---

    /**
     * The current CVS revision of this class
     */
    public final String REVISION = "$Revision: 1.29 $";
    
//---STATIC METHODS---

//---CONSTRUCTORS---
    
    /**
     * Construct a new TCPReaderInit.
     *
     * @param socket the Socket to which the host is connected
     * @param queue the Queue to which we'll add data
     * @throws IOException if something goes badly wrong
     */
    public TCPReaderInit(Socket socket, Queue queue) throws IOException {
        // set the Thread name
        setName("filter.TCPReaderInit");
        
        _socket = socket;
        _queue = queue;
        // setup the reader & writer
        _socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
        _socketOut = new PrintWriter(_socket.getOutputStream(), true);
        _logger.write(toString(), Logger.SYSINIT, "created");
    }
    
//---PUBLIC METHODS---
    
    /**
     * Main run method. Will communicate with the host, inform it
     * if any updates to it's configuration are needed, and send
     * a heartbeat packet into the system.
     */
    public void run() {
        try {
            // get an instance of the KeyManager
            KeyManager keyman = KeyManager.getInstance();
            
            // get some information about the host
            String hostname = _socket.getInetAddress().getHostName().toLowerCase();
            String ipadd = _socket.getInetAddress().getHostAddress();
            
            // try for HEARTBEAT
            getInBound("HEARTBEAT");
            _socketOut.println("OK");
            
            // look for a command:
            // CONFIG - to check config
            // KEY - to get the key
            // ENDHEARTBEAT - to finish            
            String cmd = getInBound();
            while(!cmd.equals("ENDHEARTBEAT")) {
                if(cmd.equals("CONFIG")) {
                    // respond to CONFIG
                    _socketOut.println("OK");
                        
                    // try for {filelist}
                    String filelist = getInBound();
                    _socketOut.println("OK");
                    
                    // try for {lastModified}
                    String lastModified = getInBound();
                    // check to see if a config update has happen
                    boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
                    if(newConfig) {
                        // new config !
                        _socketOut.println("ERROR");
                    }
                    else {
                        // nothing has changed
                        _socketOut.println("OK");
                    }
                }
                else if(cmd.equals("KEY")) {
                    // repsond to KEY
                    // generate a key
                    String key = keyman.genKey();
                    // send key to host
                    _socketOut.println(key);
                    // add it to the key manager
                    keyman.addKey(hostname, key);
                }
                else {
                    _socketOut.println("ERROR");
                }
                // get the next command
                cmd = getInBound();
            }
            
            // respond to ENDHEARTBEAT
            _socketOut.println("OK");
            
            // work out some information for our heartbeat packet
            String date = new Long(System.currentTimeMillis()/((long) 1000)).toString(); //seconds
            
            // run the service checks for this host
            _logger.write(toString(), Logger.DEBUG, "Running service checks");
            String checks = PluginServiceCheckManager.getInstance().runServiceChecks(hostname);
            
            // build the heartbeat packet
            String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\">" + checks + "</packet>";
            
            // get it to be sent on
            _queue.add(xml);
            
        } catch (Exception e) {
            _logger.write(toString(), Logger.ERROR, "ERROR: " + e);
        }
        
        // Disconnect streams & socket
        try {
            _socketIn.close();
            _socketOut.close();
            _socket.close();
        } catch (IOException e) {
            _logger.write(toString(), Logger.ERROR, "exception on socket close");
        }
        _logger.write(toString(), Logger.DEBUG, "finished");
    }
    
    /**
     * Overrides the {@link java.lang.Object#toString() Object.toString()}
     * method to provide clean logging (every class should have this).
     *
     * This uses the uk.org.iscream.cms.server.util.NameFormat class
     * to format the toString()
     *
     * @return the name of this class and its CVS revision
     */
    public String toString() {
        return FormatName.getName(
            _name,
            getClass().getName(),
            REVISION);
    }

//---PRIVATE METHODS---

    private String getInBound(String expected) throws IOException {
        // grab the input
        String inBound = getInBound();
        // check if it's what we're expecting
        if(!inBound.equals(expected)) {
            throw new IOException("protocol error from "+_socket.getInetAddress().getHostName()+" - expected:"+expected+" got:" + inBound);
        }
        // it should be ok then
        return inBound;
    }
    
    private String getInBound() throws IOException {
        // grab the input
        String inBound = _socketIn.readLine();
        // check for null's, likely disconnection
        if(inBound == null) {
            throw new IOException("got null from host, maybe it died");
        }
        // it's a valid message it seems
        return inBound;
    }
    
//---ACCESSOR/MUTATOR METHODS---

//---ATTRIBUTES---

    /**
     * A reference to the configuration manager
     */
    ConfigurationManager _configManager = ReferenceManager.getInstance().getCM();
    
    /**
     * This is the friendly identifier of the
     * component this class is running in.
     * eg, a Filter may be called "filter1",
     * If this class does not have an owning
     * component,  a name from the configuration
     * can be placed here.  This name could also
     * be changed to null for utility classes.
     */
    private String _name = FilterMain.NAME;

    /**
     * This holds a reference to the
     * system logger that is being used.
     */
    private Logger _logger = ReferenceManager.getInstance().getLogger();
    
    /**
     * The socket we are talking on
     */
    Socket _socket;
    
    /** 
     * The input from the socket
     */
    BufferedReader _socketIn;
    
    /**
     * The output from the socket
     */
    PrintWriter _socketOut;
    
    /**
     * A reference to our Queue
     */
    Queue _queue;
    
//---STATIC ATTRIBUTES---

}
Revision:	1.30
Committed:	Sat May 18 18:16:02 2002 UTC (22 years ago) by tdb
Branch:	MAIN
Changes since 1.29:	+21 -2 lines
Log Message:	i-scream is now licensed under the GPL. I've added the GPL headers to every source file, and put a full copy of the license in the appropriate places. I think I've covered everything. This is going to be a mad commit ;)
#	Content
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.org.iscream.cms.server.filter;
22
23	//---IMPORTS---
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;
28	import java.io.*;
29	import java.util.Random;
30	import uk.org.iscream.cms.server.util.*;
31
32	/**
33	* This provides Host heartbeat functionality
34	*
35	* @author $Author: tdb $
36	* @version $Id: TCPReaderInit.java,v 1.29 2002/05/12 15:49:50 tdb Exp $
37	*/
38	class TCPReaderInit extends Thread {
39
40	//---FINAL ATTRIBUTES---
41
42	/**
43	* The current CVS revision of this class
44	*/
45	public final String REVISION = "$Revision: 1.29 $";
46
47	//---STATIC METHODS---
48
49	//---CONSTRUCTORS---
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	_queue = queue;
64	// setup the reader & writer
65	_socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
66	_socketOut = new PrintWriter(_socket.getOutputStream(), true);
67	_logger.write(toString(), Logger.SYSINIT, "created");
68	}
69
70	//---PUBLIC METHODS---
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	// get an instance of the KeyManager
80	KeyManager keyman = KeyManager.getInstance();
81
82	// get some information about the host
83	String hostname = _socket.getInetAddress().getHostName().toLowerCase();
84	String ipadd = _socket.getInetAddress().getHostAddress();
85
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	// respond to ENDHEARTBEAT
134	_socketOut.println("OK");
135
136	// work out some information for our heartbeat packet
137	String date = new Long(System.currentTimeMillis()/((long) 1000)).toString(); //seconds
138
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	// build the heartbeat packet
144	String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\">" + checks + "</packet>";
145
146	// get it to be sent on
147	_queue.add(xml);
148
149	} catch (Exception e) {
150	_logger.write(toString(), Logger.ERROR, "ERROR: " + e);
151	}
152
153	// Disconnect streams & socket
154	try {
155	_socketIn.close();
156	_socketOut.close();
157	_socket.close();
158	} catch (IOException e) {
159	_logger.write(toString(), Logger.ERROR, "exception on socket close");
160	}
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 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	/**
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
245	/**
246	* A reference to our Queue
247	*/
248	Queue _queue;
249
250	//---STATIC ATTRIBUTES---
251
252	}