host/java/Config.java

//---PACKAGE DECLARATION---

//---IMPORTS---

import java.net.*;
import java.util.*;
import java.io.*;

/**
 * Configurator object for the JavaHost
 * Will connect to the filter manager and collect its specific
 * configuration
 *
 * @author  $Author: ab11 $
 * @version $Id: Config.java,v 1.4 2000/11/30 17:32:18 ab11 Exp $
 */
class Config {

//---FINAL ATTRIBUTES---

//---STATIC METHODS---

//---CONSTRUCTORS---

    public Config( String serverName, int serverPort ){
        // takes in the master config settings and creates a connection with
        // this computer, and downloads all the values listed in config.values.txt
        // which should be in the local directory
        
        // read in from the file.
        try {
            System.out.println("Reading Config file");
            BufferedReader inFile = new BufferedReader(new FileReader("config.values.txt"));
            aList = new ArrayList();
            numProperties = 0;
            String tmpIn = inFile.readLine();
            while ( tmpIn != null ){
                aList.add(numProperties, tmpIn);
                numProperties++;
                tmpIn = inFile.readLine();  
            }
            System.out.println("Added "+numProperties+" properties");
        }
        catch ( FileNotFoundException e ){
            // do something
            System.out.println("Config file not found");
        }
        catch ( IOException e ){
            // do something
        }
    
        myProperties = new HashMap();
        configChanged = false;
        
        // time in seconds before first retry
        filterManagerRetryTime = 10;
    
        // do the funky jibble
        connect(serverName, serverPort);
    }

//---PUBLIC METHODS---

    public InetAddress getFilterName(){
        // will return the most recient IP address (if it is dynamic for whatever reason
        try {
            return InetAddress.getByName(filterName);
        }
        catch ( UnknownHostException e ){
            // do something
            return null;
        }
        
    }
    
    /**
     * Used to retrieve the port to send UDP packets to on the filter
     *
     * @return an integer corrisponding to the UDP port of the filter
     */
    public int getFilterUDPPort(){
        
        return filterUDPPort;
    }
    
    /**
     * Used to retrieve the port to send TCP heartbeats to on the filter
     *
     * @return an integer corrisponding to the TCP of the filter
     */    
    public int getFilterTCPPort(){
        
        return filterTCPPort;
    }

    /**
     * Used to get the hostname of the filter we are to talk to.
     *
     * @return a string representing the hostname of the filter
     */
    public String getProperty( String name ){
        
        if ( myProperties.containsKey(name) ){
            // its in there, may return ERROR
            return (String) myProperties.get(name);
        }
        else
        {
            // can't have been in the config.values.txt file!
            return "ERROR";
        }
    }


//---PRIVATE METHODS---

    private void connect( String serverName, int serverPort ){
        Socket mySocket;
        configChanged = false;
        
        System.out.println("Establishing connection with filter manager");
        
        // might throw a UnknownHostException
        try {
            mySocket = new Socket( serverName, serverPort );
             // ok we have our socket connected ok. grab their input and output streams
            socketIn = new BufferedReader(new InputStreamReader(mySocket.getInputStream()));
            socketOut = new PrintWriter(mySocket.getOutputStream());
 
            if (sendCommand("STARTCONFIG").equals("OK")){
                // everything is fine
                // sendCommand("LASTMODIFIED");
                lastModified = sendCommand("LASTMODIFIED");
                
                fileList = sendCommand("FILELIST");
                // get all the properties
                if ( numProperties > 0 ){
                        // sendCommand("CONFIG");
                        for ( int i = 0; i < numProperties; i++ ){
                            String property = (String) aList.get(i);
                            myProperties.put(property, sendCommand(property));  
                        }
                }
                
                sendCommand("ENDCONFIG");
                String filter_data = sendCommand("FILTER");
                StringTokenizer tok = new StringTokenizer(filter_data,";");
                filterName = tok.nextToken();
                filterUDPPort = Integer.parseInt(tok.nextToken());
                filterTCPPort = Integer.parseInt(tok.nextToken());
                                    
                sendCommand("END");         
                 
            }
           
        
            // close the socket
            mySocket.close();
            System.out.println("Completed communication with filter manager");
        
        }
        catch ( UnknownHostException e ){
            // what to do   
            System.out.println("Host not found");    
        }
        catch ( IOException e ){
            // what to do 
            System.out.println("Unable to read from socket, might not be open");
            System.out.println("Retrying in "+filterManagerRetryTime+" seconds");
            configChanged = true;
            try {
                Thread.sleep(filterManagerRetryTime*1000);
            }
            catch( InterruptedException f ){
                System.out.println("Sleep interrupted");
            }
            filterManagerRetryTime = filterManagerRetryTime * 2;
            // warning this WILL cause a stack overflow after a while..
            // need to fix it.
            connect(serverName, serverPort);
        }
            
    } // connect


    public void sendHeartBeat(){
        
        Socket mySocket;
        try {
                 
                mySocket = new Socket( filterName, filterTCPPort );
                // ok we have our socket connected ok. grab their input and output streams
                socketIn = new BufferedReader(new InputStreamReader(mySocket.getInputStream()));
                socketOut = new PrintWriter(mySocket.getOutputStream());
         
                        
                if ( sendCommand("HEARTBEAT").equals("OK") ){
                        if ( sendCommand("CONFIG").equals("OK") ){
                                sendCommand(fileList);
                                if ( sendCommand(lastModified).equals("ERROR") ){
                                        // config has changed
                                        configChanged = true;
                                }
                                sendCommand("ENDHEARTBEAT");
                        }
                }       
        }
        catch ( UnknownHostException e ){
            // what to do   
            System.out.println("Host not found");    
        }
        catch ( IOException e ){
            // what to do 
            System.out.println("Unable to read from socket, might not be open");
            System.out.println("Re-establishing contact with filter manager");
            configChanged = true;
        }
    }

    private String sendCommand(String command){
        
        // System.out.println("Writing command: "+command);
        socketOut.println(command);
        socketOut.flush();
        try {
           String response = socketIn.readLine();
           // System.out.println("Got: "+response);
           return response;
        }
        catch ( IOException e ){
            // something went wrong
            System.out.println("Error recieving response");
            return "ERROR";   
        }
    }

//---ACCESSOR/MUTATOR METHODS---

    public boolean reloadConfig(){
        return configChanged;
    } 

//---ATTRIBUTES---

    private boolean configChanged;
    private String lastModified;
    private String fileList;
    private int numProperties;
    private HashMap myProperties;
    private String filterName;
    private int filterUDPPort;
    private int filterTCPPort;
    private ArrayList aList;
    private BufferedReader socketIn;
    private PrintWriter socketOut;
    private int filterManagerRetryTime;
    

//---STATIC ATTRIBUTES---

} // class
Revision:	1.5
Committed:	Thu Dec 7 23:18:28 2000 UTC (23 years, 11 months ago) by tdb
Branch:	MAIN
Changes since 1.4:	+4 -4 lines
Log Message:	Just a terminology change :)
#	Content
1	//---PACKAGE DECLARATION---
2
3	//---IMPORTS---
4
5	import java.net.*;
6	import java.util.*;
7	import java.io.*;
8
9	/**
10	* Configurator object for the JavaHost
11	* Will connect to the filter manager and collect its specific
12	* configuration
13	*
14	* @author $Author: ab11 $
15	* @version $Id: Config.java,v 1.4 2000/11/30 17:32:18 ab11 Exp $
16	*/
17	class Config {
18
19	//---FINAL ATTRIBUTES---
20
21	//---STATIC METHODS---
22
23	//---CONSTRUCTORS---
24
25	public Config( String serverName, int serverPort ){
26	// takes in the master config settings and creates a connection with
27	// this computer, and downloads all the values listed in config.values.txt
28	// which should be in the local directory
29
30	// read in from the file.
31	try {
32	System.out.println("Reading Config file");
33	BufferedReader inFile = new BufferedReader(new FileReader("config.values.txt"));
34	aList = new ArrayList();
35	numProperties = 0;
36	String tmpIn = inFile.readLine();
37	while ( tmpIn != null ){
38	aList.add(numProperties, tmpIn);
39	numProperties++;
40	tmpIn = inFile.readLine();
41	}
42	System.out.println("Added "+numProperties+" properties");
43	}
44	catch ( FileNotFoundException e ){
45	// do something
46	System.out.println("Config file not found");
47	}
48	catch ( IOException e ){
49	// do something
50	}
51
52	myProperties = new HashMap();
53	configChanged = false;
54
55	// time in seconds before first retry
56	filterManagerRetryTime = 10;
57
58	// do the funky jibble
59	connect(serverName, serverPort);
60	}
61
62	//---PUBLIC METHODS---
63
64	public InetAddress getFilterName(){
65	// will return the most recient IP address (if it is dynamic for whatever reason
66	try {
67	return InetAddress.getByName(filterName);
68	}
69	catch ( UnknownHostException e ){
70	// do something
71	return null;
72	}
73
74	}
75
76	/**
77	* Used to retrieve the port to send UDP packets to on the filter
78	*
79	* @return an integer corrisponding to the UDP port of the filter
80	*/
81	public int getFilterUDPPort(){
82
83	return filterUDPPort;
84	}
85
86	/**
87	* Used to retrieve the port to send TCP heartbeats to on the filter
88	*
89	* @return an integer corrisponding to the TCP of the filter
90	*/
91	public int getFilterTCPPort(){
92
93	return filterTCPPort;
94	}
95
96	/**
97	* Used to get the hostname of the filter we are to talk to.
98	*
99	* @return a string representing the hostname of the filter
100	*/
101	public String getProperty( String name ){
102
103	if ( myProperties.containsKey(name) ){
104	// its in there, may return ERROR
105	return (String) myProperties.get(name);
106	}
107	else
108	{
109	// can't have been in the config.values.txt file!
110	return "ERROR";
111	}
112	}
113
114
115	//---PRIVATE METHODS---
116
117	private void connect( String serverName, int serverPort ){
118	Socket mySocket;
119	configChanged = false;
120
121	System.out.println("Establishing connection with filter manager");
122
123	// might throw a UnknownHostException
124	try {
125	mySocket = new Socket( serverName, serverPort );
126	// ok we have our socket connected ok. grab their input and output streams
127	socketIn = new BufferedReader(new InputStreamReader(mySocket.getInputStream()));
128	socketOut = new PrintWriter(mySocket.getOutputStream());
129
130	if (sendCommand("STARTCONFIG").equals("OK")){
131	// everything is fine
132	// sendCommand("LASTMODIFIED");
133	lastModified = sendCommand("LASTMODIFIED");
134
135	fileList = sendCommand("FILELIST");
136	// get all the properties
137	if ( numProperties > 0 ){
138	// sendCommand("CONFIG");
139	for ( int i = 0; i < numProperties; i++ ){
140	String property = (String) aList.get(i);
141	myProperties.put(property, sendCommand(property));
142	}
143	}
144
145	sendCommand("ENDCONFIG");
146	String filter_data = sendCommand("FILTER");
147	StringTokenizer tok = new StringTokenizer(filter_data,";");
148	filterName = tok.nextToken();
149	filterUDPPort = Integer.parseInt(tok.nextToken());
150	filterTCPPort = Integer.parseInt(tok.nextToken());
151
152	sendCommand("END");
153
154	}
155
156
157	// close the socket
158	mySocket.close();
159	System.out.println("Completed communication with filter manager");
160
161	}
162	catch ( UnknownHostException e ){
163	// what to do
164	System.out.println("Host not found");
165	}
166	catch ( IOException e ){
167	// what to do
168	System.out.println("Unable to read from socket, might not be open");
169	System.out.println("Retrying in "+filterManagerRetryTime+" seconds");
170	configChanged = true;
171	try {
172	Thread.sleep(filterManagerRetryTime*1000);
173	}
174	catch( InterruptedException f ){
175	System.out.println("Sleep interrupted");
176	}
177	filterManagerRetryTime = filterManagerRetryTime * 2;
178	// warning this WILL cause a stack overflow after a while..
179	// need to fix it.
180	connect(serverName, serverPort);
181	}
182
183	} // connect
184
185
186	public void sendHeartBeat(){
187
188	Socket mySocket;
189	try {
190
191	mySocket = new Socket( filterName, filterTCPPort );
192	// ok we have our socket connected ok. grab their input and output streams
193	socketIn = new BufferedReader(new InputStreamReader(mySocket.getInputStream()));
194	socketOut = new PrintWriter(mySocket.getOutputStream());
195
196
197	if ( sendCommand("HEARTBEAT").equals("OK") ){
198	if ( sendCommand("CONFIG").equals("OK") ){
199	sendCommand(fileList);
200	if ( sendCommand(lastModified).equals("ERROR") ){
201	// config has changed
202	configChanged = true;
203	}
204	sendCommand("ENDHEARTBEAT");
205	}
206	}
207	}
208	catch ( UnknownHostException e ){
209	// what to do
210	System.out.println("Host not found");
211	}
212	catch ( IOException e ){
213	// what to do
214	System.out.println("Unable to read from socket, might not be open");
215	System.out.println("Re-establishing contact with filter manager");
216	configChanged = true;
217	}
218	}
219
220	private String sendCommand(String command){
221
222	// System.out.println("Writing command: "+command);
223	socketOut.println(command);
224	socketOut.flush();
225	try {
226	String response = socketIn.readLine();
227	// System.out.println("Got: "+response);
228	return response;
229	}
230	catch ( IOException e ){
231	// something went wrong
232	System.out.println("Error recieving response");
233	return "ERROR";
234	}
235	}
236
237	//---ACCESSOR/MUTATOR METHODS---
238
239	public boolean reloadConfig(){
240	return configChanged;
241	}
242
243	//---ATTRIBUTES---
244
245	private boolean configChanged;
246	private String lastModified;
247	private String fileList;
248	private int numProperties;
249	private HashMap myProperties;
250	private String filterName;
251	private int filterUDPPort;
252	private int filterTCPPort;
253	private ArrayList aList;
254	private BufferedReader socketIn;
255	private PrintWriter socketOut;
256	private int filterManagerRetryTime;
257
258
259
260	//---STATIC ATTRIBUTES---
261
262	} // class