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.14 by tdb, Thu Feb 1 00:18:42 2001 UTC vs.
Revision 1.28 by tdb, Thu Mar 21 22:09:13 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
46          _socketIn = new BufferedReader(new InputStreamReader(_socket.getInputStream()));
47 <        _socketOut = new PrintWriter(_socket.getOutputStream());
47 >        _socketOut = new PrintWriter(_socket.getOutputStream(), true);
48          _logger.write(toString(), Logger.SYSINIT, "created");
49      }
50      
# 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 <                _socketOut.flush();
68 <                throw new IOException("protocol error - expecting:HEARTBEAT got:" + inBound);
69 <            } else {
70 <                _socketOut.println("OK");
71 <                _socketOut.flush();
72 <            }
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")) {
69 <                _socketOut.println("ERROR");
70 <                _socketOut.flush();
71 <                throw new IOException("protocol error - expecting:CONFIG got:" + inBound);
72 <            } else {
73 <                _socketOut.println("OK");
74 <                _socketOut.flush();
67 >            // try for HEARTBEAT
68 >            getInBound("HEARTBEAT");
69 >            _socketOut.println("OK");
70 >            
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 >                    // add it to the key manager
103 >                    keyman.addKey(hostname, key);
104 >                    _socketOut.println(key);
105 >                }
106 >                else {
107 >                    _socketOut.println("ERROR");
108 >                }
109 >                // get the next command
110 >                cmd = getInBound();
111              }
112              
113 <            inBound = _socketIn.readLine();
85 <            filelist = inBound;
113 >            // respond to ENDHEARTBEAT
114              _socketOut.println("OK");
87 –            _socketOut.flush();
115              
116 <            inBound = _socketIn.readLine();
117 <            lastModified = inBound;
116 >            // work out some information for our heartbeat packet
117 >            String date = new Long(System.currentTimeMillis()/((long) 1000)).toString(); //seconds
118              
119 <            boolean newConfig = _configManager.isModified(filelist, Long.parseLong(lastModified));
119 >            // run the service checks for this host
120 >            _logger.write(toString(), Logger.DEBUG, "Running service checks");
121 >            String checks = PluginServiceCheckManager.getInstance().runServiceChecks(hostname);
122              
123 <            if(newConfig) {
124 <                _socketOut.println("ERROR");
96 <            }
97 <            else {
98 <                _socketOut.println("OK");
99 <            }
100 <            _socketOut.flush();
123 >            // build the heartbeat packet
124 >            String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\">" + checks + "</packet>";
125              
126 <            inBound = _socketIn.readLine();
103 <            if(!inBound.equals("ENDHEARTBEAT")) {
104 <                _socketOut.println("ERROR");
105 <                _socketOut.flush();
106 <                throw new IOException("protocol error - expecting:ENDHEARTBEAT got:" + inBound);
107 <            } else {
108 <                _socketOut.println("OK");
109 <                _socketOut.flush();
110 <            }
111 <
112 <            String date = new Long(System.currentTimeMillis()).toString();
113 <            String hostname = _socket.getInetAddress().getHostName();
114 <            String ipadd = _socket.getInetAddress().getHostAddress();
115 <            String xml = "<packet type=\"heartbeat\" machine_name=\""+hostname+"\" date=\""+date+"\" ip=\""+ipadd+"\"></packet>";
116 <            
126 >            // get it to be sent on
127              _queue.add(xml);
128              
129          } catch (Exception e) {
130 <            _logger.write(toString(), Logger.ERROR, "ERROR: " + e.getMessage());
130 >            _logger.write(toString(), Logger.ERROR, "ERROR: " + e);
131          }
132          
123 –        _socketOut.flush();
133          // Disconnect streams & socket
134          try {
135              _socketIn.close();
# Line 136 | Line 145 | class TCPReaderInit extends Thread {
145       * Overrides the {@link java.lang.Object#toString() Object.toString()}
146       * method to provide clean logging (every class should have this).
147       *
148 <     * This uses the uk.ac.ukc.iscream.util.NameFormat class
148 >     * This uses the uk.org.iscream.cms.server.util.NameFormat class
149       * to format the toString()
150       *
151       * @return the name of this class and its CVS revision
# Line 150 | Line 159 | class TCPReaderInit extends Thread {
159  
160   //---PRIVATE METHODS---
161  
162 +    private String getInBound(String expected) throws IOException {
163 +        // grab the input
164 +        String inBound = getInBound();
165 +        // check if it's what we're expecting
166 +        if(!inBound.equals(expected)) {
167 +            throw new IOException("protocol error from "+_socket.getInetAddress().getHostName()+" - expected:"+expected+" got:" + inBound);
168 +        }
169 +        // it should be ok then
170 +        return inBound;
171 +    }
172 +    
173 +    private String getInBound() throws IOException {
174 +        // grab the input
175 +        String inBound = _socketIn.readLine();
176 +        // check for null's, likely disconnection
177 +        if(inBound == null) {
178 +            throw new IOException("got null from host, maybe it died");
179 +        }
180 +        // it's a valid message it seems
181 +        return inBound;
182 +    }
183 +    
184   //---ACCESSOR/MUTATOR METHODS---
185  
186   //---ATTRIBUTES---
# Line 195 | Line 226 | class TCPReaderInit extends Thread {
226       * A reference to our Queue
227       */
228      Queue _queue;
229 +    
230   //---STATIC ATTRIBUTES---
231  
232   }

Diff Legend

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