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
(Generate patch)

Comparing projects/cms/source/server/uk/org/iscream/cms/server/filter/TCPReaderInit.java (file contents):
Revision 1.16 by ajm, Sun Feb 11 18:01:43 2001 UTC vs.
Revision 1.29 by tdb, Sun May 12 15:49:50 2002 UTC

# Line 1 | Line 1
1   //---PACKAGE DECLARATION---
2 < package uk.ac.ukc.iscream.filter;
2 > package uk.org.iscream.cms.server.filter;
3  
4   //---IMPORTS---
5 < import uk.ac.ukc.iscream.core.*;
6 < import uk.ac.ukc.iscream.filter.*;
7 < import uk.ac.ukc.iscream.componentmanager.*;
5 > import uk.org.iscream.cms.server.core.*;
6 > import uk.org.iscream.cms.server.filter.*;
7 > import uk.org.iscream.cms.server.componentmanager.*;
8   import java.net.Socket;
9 – import java.io.InputStream;
10 – import java.io.OutputStream;
11 – import java.io.IOException;
9   import java.io.*;
10 < import uk.ac.ukc.iscream.util.*;
10 > import java.util.Random;
11 > import uk.org.iscream.cms.server.util.*;
12  
13   /**
14   * This provides Host heartbeat functionality
# Line 39 | Line 37 | class TCPReaderInit extends Thread {
37       * @throws IOException if something goes badly wrong
38       */
39      public TCPReaderInit(Socket socket, Queue queue) throws IOException {
40 +        // set the Thread name
41 +        setName("filter.TCPReaderInit");
42 +        
43          _socket = socket;
44          _queue = queue;
45          // setup the reader & writer
# Line 56 | Line 57 | class TCPReaderInit extends Thread {
57       */
58      public void run() {
59          try {
60 <            //variables
61 <            String filelist = "";
61 <            String lastModified = "";
62 <            String inBound = "";
60 >            // get an instance of the KeyManager
61 >            KeyManager keyman = KeyManager.getInstance();
62              
63 <            inBound = _socketIn.readLine();
64 <            if(!inBound.equals("HEARTBEAT")) {
65 <                _socketOut.println("ERROR");
67 <                throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
68 <            } else {
69 <                _socketOut.println("OK");
70 <            }
63 >            // get some information about the host
64 >            String hostname = _socket.getInetAddress().getHostName().toLowerCase();
65 >            String ipadd = _socket.getInetAddress().getHostAddress();
66              
67 <            inBound = _socketIn.readLine();
68 <            if(!inBound.equals("CONFIG")) {
74 <                _socketOut.println("ERROR");
75 <                throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
76 <            } else {
77 <                _socketOut.println("OK");
78 <            }
79 <            
80 <            inBound = _socketIn.readLine();
81 <            filelist = inBound;
67 >            // try for HEARTBEAT
68 >            getInBound("HEARTBEAT");
69              _socketOut.println("OK");
70              
71 <            inBound = _socketIn.readLine();
72 <            lastModified = inBound;
73 <            
74 <            boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
75 <            
76 <            if(newConfig) {
77 <                _socketOut.println("ERROR");
71 >            // look for a command:
72 >            // CONFIG - to check config
73 >            // KEY - to get the key
74 >            // ENDHEARTBEAT - to finish            
75 >            String cmd = getInBound();
76 >            while(!cmd.equals("ENDHEARTBEAT")) {
77 >                if(cmd.equals("CONFIG")) {
78 >                    // respond to CONFIG
79 >                    _socketOut.println("OK");
80 >                        
81 >                    // try for {filelist}
82 >                    String filelist = getInBound();
83 >                    _socketOut.println("OK");
84 >                    
85 >                    // try for {lastModified}
86 >                    String lastModified = getInBound();
87 >                    // check to see if a config update has happen
88 >                    boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
89 >                    if(newConfig) {
90 >                        // new config !
91 >                        _socketOut.println("ERROR");
92 >                    }
93 >                    else {
94 >                        // nothing has changed
95 >                        _socketOut.println("OK");
96 >                    }
97 >                }
98 >                else if(cmd.equals("KEY")) {
99 >                    // repsond to KEY
100 >                    // generate a key
101 >                    String key = keyman.genKey();
102 >                    // send key to host
103 >                    _socketOut.println(key);
104 >                    // add it to the key manager
105 >                    keyman.addKey(hostname, key);
106 >                }
107 >                else {
108 >                    _socketOut.println("ERROR");
109 >                }
110 >                // get the next command
111 >                cmd = getInBound();
112              }
92 –            else {
93 –                _socketOut.println("OK");
94 –            }
113              
114 <            inBound = _socketIn.readLine();
115 <            if(!inBound.equals("ENDHEARTBEAT")) {
98 <                _socketOut.println("ERROR");
99 <                throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
100 <            } else {
101 <                _socketOut.println("OK");
102 <            }
103 <
104 <            String date = new Long(System.currentTimeMillis()).toString();
105 <            String hostname = _socket.getInetAddress().getHostName();
106 <            String ipadd = _socket.getInetAddress().getHostAddress();
114 >            // respond to ENDHEARTBEAT
115 >            _socketOut.println("OK");
116              
117 +            // work out some information for our heartbeat packet
118 +            String date = new Long(System.currentTimeMillis()/((long) 1000)).toString(); //seconds
119 +            
120              // run the service checks for this host
121 +            _logger.write(toString(), Logger.DEBUG, "Running service checks");
122              String checks = PluginServiceCheckManager.getInstance().runServiceChecks(hostname);
123              
124              // build the heartbeat packet
# Line 115 | Line 128 | class TCPReaderInit extends Thread {
128              _queue.add(xml);
129              
130          } catch (Exception e) {
131 <            _logger.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
131 >            _logger.write(toString(), Logger.ERROR, "ERROR: " + e);
132          }
133          
134          // Disconnect streams & socket
# Line 133 | Line 146 | class TCPReaderInit extends Thread {
146       * Overrides the {@link java.lang.Object#toString() Object.toString()}
147       * method to provide clean logging (every class should have this).
148       *
149 <     * This uses the uk.ac.ukc.iscream.util.NameFormat class
149 >     * This uses the uk.org.iscream.cms.server.util.NameFormat class
150       * to format the toString()
151       *
152       * @return the name of this class and its CVS revision
# Line 147 | Line 160 | class TCPReaderInit extends Thread {
160  
161   //---PRIVATE METHODS---
162  
163 +    private String getInBound(String expected) throws IOException {
164 +        // grab the input
165 +        String inBound = getInBound();
166 +        // check if it's what we're expecting
167 +        if(!inBound.equals(expected)) {
168 +            throw new IOException("protocol error from "+_socket.getInetAddress().getHostName()+" - expected:"+expected+" got:" + inBound);
169 +        }
170 +        // it should be ok then
171 +        return inBound;
172 +    }
173 +    
174 +    private String getInBound() throws IOException {
175 +        // grab the input
176 +        String inBound = _socketIn.readLine();
177 +        // check for null's, likely disconnection
178 +        if(inBound == null) {
179 +            throw new IOException("got null from host, maybe it died");
180 +        }
181 +        // it's a valid message it seems
182 +        return inBound;
183 +    }
184 +    
185   //---ACCESSOR/MUTATOR METHODS---
186  
187   //---ATTRIBUTES---
# Line 192 | Line 227 | class TCPReaderInit extends Thread {
227       * A reference to our Queue
228       */
229      Queue _queue;
230 +    
231   //---STATIC ATTRIBUTES---
232  
233   }

Diff Legend

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