cms/conient/ConnectionHandler.java

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org
 * 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.conient;

//---IMPORTS---
import uk.org.iscream.cms.util.*;
import java.io.*;
import java.net.*;
import javax.swing.JOptionPane;

/**
 * This is the main thread for the client.
 * Once started i continually checks it actionQueue
 * for actions that other areas of the system have placed 
 * there, it then performs those actions.
 *
 * Currently this is the main thread where all non-GUI events
 * are dispatched to.
 *
 * @author  $Author: tdb $
 * @version $Id: ConnectionHandler.java,v 1.36 2003/02/05 19:35:04 tdb Exp $
 */
public class ConnectionHandler extends Thread {

//---FINAL ATTRIBUTES---

    /**
     * The current CVS revision of this class
     */
    public final String REVISION = "$Revision: 1.36 $";


    /**
     * The hardcoded protocol version that we are using.
     * Used when handshaking with the server
     */
    public final double PROTOCOL_VERSION = 1.1;

    /**
     * Thread action DONOTHING.
     * This is an invalid action, but here for completeness
     * if anything sends this action it will warn appropriately
     * but do what it says, ie, nothing ;)
     */
    public static final int DONOTHING = 0;
    
    /**
     * Thread action CONNECT.
     * Opens the control link to the i-scream server.
     */
    public static final int CONNECT = 1;
    
    /**
     * Thread action STARTDATA.
     * Opens the data link to the i-scream server
     * and prepares all gui data components for data
     * it then starts all relavant threads going.
     */
    public static final int STARTDATA = 2;
    
    /**
     * Thread action STOPDATA.
     * Closes the data link and shuts down all components
     * that update gui for various things.
     */
    public static final int STOPDATA = 3;
    
    /**
     * Thread action DISCONNECT.
     * Checks to see if STOPDATA has been called if not,
     * it will add STOPDATA and then DISCONNECT to
     * the action queue and return.  If STOPDATA has been
     * called it closes down the Control Link and tidies up
     * all relevant threads
     */
    public static final int DISCONNECT = 4;
    
    /**
     * Thread action QUIT.
     * Checks the status of the two links, if either
     * are still up, it queues the appropriate commands
     * to close them down.  It then System.exit(0)'s!
     */
    public static final int QUIT = 5;

    /**
     * Thread action GETCONFIGURATION
     * Starts the command to obtain the configuration
     * from the server. 
     * It then passes the IO to the Configuration object
     * so it can obtain any specific configuration
     */
    public static final int GETCONFIGURATION = 6;

    /**
     * This is the time in seconds that this class
     * should wait on the DataReader class to fully
     * shutdown after calling shutdown() before
     * forcing a shutdown
     */
    public static final int DATAREADER_SHUTDOWN_TIMEOUT = 5;

    /**
     * The default local server to connect to when a 
     * firewall is in use if one is not specified in the server
     */
    public static final String DEFAULT_FIREWALL_SERVER = "localhost";

    /**
     * The default time in seconds to wait for the
     * firewall setup command to execute
     */
    public static final int DEFAULT_FIREWALL_COMMANDWAIT = 5;
    
    /**
     * The default maximum queue size in use for the data queue
     */
    public static final int DEFAULT_QUEUE_LIMIT = 500;
    
    /**
     * The algorithm we use to remove items from the queue
     * if it reaches its maximum size.
     *
     * We use a FIFO algorithm, see the uk.org.iscream.cms.util.Queue
     * class for other options.
     */
    public static final int QUEUE_ALGORITHM = Queue.FIRST;

//---STATIC METHODS---

//---CONSTRUCTORS---

    /**
     * Constructs new data handler.
     * Needs a reference to the data panel so that
     * it can set it processing inBound data
     * Also gets a reference to the queue that will be
     * used by other areas of the system to send it actions
     *
     * @param data the DataPanel in use
     * @param actionQueue the actionQueue for this class
     */ 
    public ConnectionHandler(DataPanel data, Queue actionQueue) {
        _data = data;
        _actionQueue = actionQueue;
        _myQueue = _actionQueue.getQueue();
        _configuration.setConnectionHandler(this);
    }
    
//---PUBLIC METHODS---


    /**
     * Starts this ConnectionHandler running.
     * This basically runs until told to stop, it gets "actions"
     * from its actionQueue and switch's on them do determine
     * what it should do it then carries out the action
     *
     * For details on what each action does see the action
     * types.
     */
    public void run() {
        if(_configuration.getProperty("control.onstartconnect").equals("1")) {
            _actionQueue.add(new Integer(CONNECT));
        }
        if(_configuration.getProperty("data.onstartconnect").equals("1")) {
            _actionQueue.add(new Integer(STARTDATA));
        }
        while(_running) {
            // we wait for a call...
            int action = 0;
            try {
                action = ((Integer) _actionQueue.get(_myQueue)).intValue();
            } catch (InvalidQueueException e) {
                // we 're never going to get this
                // but if we do we should do something nasty
                throw new RuntimeException("unable to retrieve events from actionQueue!");
            }
            
            // examine our action...
            // if it was to connect...then we connect...
            switch(action) {
                case CONNECT:
                    if (_controlLink == null) {
                        try {
                            // get the server name from the config
                            _configuredServer = _configuration.getProperty("control.server");
                                                                                    
                            if (_configuredServer == null) {
                                throw new IOException("no i-scream server in current configuration");
                            }
                                                        
                            // open the socket to the server and bind the IO
                            // get the port from the config
                            String portString = _configuration.getProperty("control.port");
                            if (portString == null) {
                                throw new IOException("no i-scream server port in current configuration");
                            }
                            int port = 0;
                            try {
                                port = Integer.parseInt(portString);
                            } catch (NumberFormatException e) {
                                throw new IOException("no valid i-scream server port in current configuration");
                            }
                            
                            // start firewall if needed
                            _server = handleFirewall(_configuredServer, port, _controlFirewallProcess);

                            Conient.setControlStatus("Connecting to - " + _server);                           
                            _controlLink = new Socket(_server, port);
                            _inBound = new BufferedReader(new InputStreamReader(_controlLink.getInputStream()));
                            _outBound = new PrintWriter(_controlLink.getOutputStream());
                            Conient.setControlStatus("Connection Established - " + _server);

                            String response;
                            response = _inBound.readLine();
                            
                            // check the servers Protocol Identity against our own
                            // we SHOULD be backwards compatible, so we can continue if
                            // they are using a newer protocol, anything else then we die
                            double serverProtocolVersion = Double.parseDouble(response.substring(9, response.length()));
                            Conient.addMessage("Protocol Versions: server [" + serverProtocolVersion + "] client [" + PROTOCOL_VERSION + "]");
                            if (serverProtocolVersion > PROTOCOL_VERSION) {
                                Conient.addMessage("WARNING{control link}: server is using a newer protocol (" + response + "), please update your client, continuing with old protocol (PROTOCOL " + PROTOCOL_VERSION + ")" );
                            } else if (serverProtocolVersion < PROTOCOL_VERSION) {
                                // tidy up
                                throw new IOException("incompatible protocol version");
                            }
                            
                            // send the name of the client
                            _outBound.println(_configuration.getProperty("clientname"));
                            _outBound.flush();
                            response = _inBound.readLine();
                            if (!response.equals("OK")) {
                                // tidy up
                                
                                throw new IOException("client name rejected - " + _configuration.getProperty("clientname"));
                            }
                            
                        } catch (IOException e) {
                            // print the error and tidy up what's left
                            Conient.addMessage("ERROR{control link}: " + e);
                            _controlLink = null;
                            // and the firewall handler if there is one
                            closeFirewall(_controlFirewallProcess);
                            _actionQueue.clearQueue(_myQueue);
                            Conient.setControlStatus("Disconnected");
                        }
                    } else {
                        Conient.addMessage("WARNING{control link}: already established");
                    }
                    break;
                case STARTDATA:
                    // as long as the data link hasn't been established
                    // we want to establish it
                    if (_dataLink == null) {
                        // check that the control link is open, if it isn't we
                        // might want to sort that problemo out
                        // we do this by simply queueing the event to occour, then
                        // this event to run again ;-)
                        if(_controlLink == null) {
                            Conient.addMessage("WARNING{data link}: control link not established - queueing start events");
                            _actionQueue.add(new Integer(CONNECT));
                            _actionQueue.add(new Integer(STARTDATA));
                        } else {
                            try {
                                // set our host list if we know we have one we need to set
                                String hostList = _configuration.getProperty("hostList");
                                boolean hostListSet = false;
                                // send our hostList if we have
                                if (_configuration.getProperty("useHostList").equals("1")) {
                                    if (hostList.equals("")) {
                                        Conient.addMessage("WARNING{control link}: your host list is empty, the server will send ALL hosts");
                                    }
                                    hostListSet = setHostList(hostList);
                                // if not, indicate we want the lot                                 
                                } else {
                                    hostListSet = setHostList("");
                                }
                                // warn if there was a problem, it will have already error'd
                                if (!hostListSet) {
                                    Conient.addMessage("WARNING{control link}: unable to set host list");
                                }                       
                                // ask the server to start the data link
                                String response;
                                _outBound.println("STARTDATA");
                                _outBound.flush();
                                response = _inBound.readLine();
                                
                                // see if the server suggested a good port
                                if (response.equals("ERROR")) {
                                    throw new IOException("server unable to start data link at this time");
                                }
                                int port = 0;
                                try {
                                    port = Integer.parseInt(response);
                                } catch (NumberFormatException e) {
                                    throw new IOException("invalid data port suggested by server - " + response);
                                }
                                
                                // start firewall if needed
                                _server = handleFirewall(_configuredServer, port, _dataFirewallProcess);
                                Conient.setDataStatus("Connecting to - " + _server + ":" + response);                                
                                
                                _dataLink = new Socket(_server, port);
                                                                
                                response = _inBound.readLine();
                                if (!response.equals("OK")) {
                                    throw new IOException("server reported error establishing data channel");
                                }
                                
                                // if the socket was ok, then we attack our IO hooks
                                _dataInBound = new BufferedReader(new InputStreamReader(_dataLink.getInputStream()));
                                _dataOutBound = new PrintWriter(_dataLink.getOutputStream());
                                Conient.setDataStatus("Connection Established - " + _server);
                                // now we want to start reading the data in
                                // so we start the appropriate components on their way
                                // we create a queue to give to both the reader and the
                                // displayer
                                
                                String configQueueSize = _configuration.getProperty("dataQueueSize");
                                int queueLimit = DEFAULT_QUEUE_LIMIT;
                                if (configQueueSize != null) {
                                    try {
                                        queueLimit = Integer.parseInt(configQueueSize);
                                    } catch (NumberFormatException e) {
                                        // if it fails it will be default
                                        Conient.addMessage("WARNING{data link}: invalid queue size in configuration, using default of - " + queueLimit);
                                    }
                                }
                               
                                Queue theQueue = new Queue(queueLimit, QUEUE_ALGORITHM);
                                _dataReader = new DataReader(_dataInBound, theQueue, this);
                                _data.setQueue(theQueue);
                                _data.cleanUpTabs();
                                
                                // start the data rocking
                                new Thread(_data).start();
                                _dataReader.start();
                                // finished for us....
                            } catch (IOException e) {
                                // print the error and tidy up what's left
                                Conient.addMessage("ERROR{data link}: " + e);
                                _dataLink = null;
                                // and the firewall handler if there is one
                                closeFirewall(_dataFirewallProcess);
                                _actionQueue.clearQueue(_myQueue);
                                Conient.setDataStatus("Disconnected");
                            }
                        }
                    } else {
                        Conient.addMessage("WARNING{data link}: already established");
                    }
                    break;
                case STOPDATA:
                    if(_dataLink != null) {
                        try {
                            String response;
                            // shut down the data link
                            Conient.setDataStatus("Disconnecting - " + _server);
                            
                            // close the reader
                            _dataReader.shutdown();
                            // wait for it to close
                            boolean dirtyShutdown = true;
                            long startTime = System.currentTimeMillis();
                            while((System.currentTimeMillis() - startTime) < (DATAREADER_SHUTDOWN_TIMEOUT * 1000)) {
                                if (!_dataReader.isAlive()) {
                                    dirtyShutdown = false;
                                    break;
                                }
                            }
                            // warn if it didn't shutdown in time
                            if (dirtyShutdown) {
                                Conient.addMessage("WARNING{data link}: data reader thread did not close within timeout, killing its IO anyway!");
                            }
                            
                            // tell the server
                            _outBound.println("STOPDATA");
                            _outBound.flush();
                            response = _inBound.readLine();
                            
                            // check the server was ok with our request...
                            // even if it wasn't we will go anyway!
                            if (!response.equals("OK")) {
                                throw new IOException("server didn't OK request to stop data channel - stopping anyway");
                            }
                            
                            
                            // close the lot down
                            _dataInBound.close();
                            _dataOutBound.close();
                            _dataLink.close();
                            // get rid of the socket
                            _dataLink = null;
                            
                            // and the firewall handler if there is one
                            closeFirewall(_dataFirewallProcess);
                            
                            Conient.setDataStatus("Disconnected");
                        } catch (IOException e) {
                            // print the error and tidy up what's left
                            Conient.addMessage("ERROR{data link}: " + e);
                            try {
                                _dataOutBound.close();
                                _dataInBound.close();
                                _dataLink.close();
                                // and the firewall handler if there is one
                                closeFirewall(_dataFirewallProcess);
                            } catch (IOException e2) {
                                    Conient.addMessage("CRITICAL{control link}: unable to close socket - " + e2);
                            }
                            _dataLink = null;
                            _actionQueue.clearQueue(_myQueue);
                            Conient.setDataStatus("Disconnected");
                        }
                    } else {
                        Conient.addMessage("WARNING{data link}: already disconnected");
                    }
                    break;
                case DISCONNECT:
                    if (_controlLink != null) {
                        if (_dataLink != null) {
                                // we want to tell ourselves to stop it
                                Conient.addMessage("WARNING{control link}: data link not disconnected - queueing stop events");
                                _actionQueue.add(new Integer(STOPDATA));
                                _actionQueue.add(new Integer(DISCONNECT));
                        } else {
                            try {
                                // request the server to disconnect
                                String response;
                                _outBound.println("DISCONNECT");
                                _outBound.flush();
                                response = _inBound.readLine();
                                
                                // check the server was ok with our request...
                                // even if it wasn't we will go anyway!
                                if (!response.equals("OK")) {
                                    throw new IOException("server didn't OK request to stop control channel - stopping anyway");
                                }
                                
                                // then lets shutdown the link
                                Conient.setControlStatus("Disconnecting - " + _server);
                                _inBound.close();
                                _outBound.close();
                                _controlLink.close();
                                // for good measure
                                _controlLink = null;
                                
                                // and the firewall handler if there is one
                                closeFirewall(_controlFirewallProcess);
                                
                                Conient.setControlStatus("Disconnected");
                            } catch (IOException e) {
                                Conient.addMessage("ERROR{control link}: " + e);
                                try {
                                    _inBound.close();
                                    _outBound.close();
                                    _controlLink.close();
                                    // and the firewall handler if there is one
                                    closeFirewall(_controlFirewallProcess);
                                } catch (IOException e2) {
                                    Conient.addMessage("CRITICAL{control link}: unable to close socket - " + e2);
                                }
                                _controlLink = null;
                                _actionQueue.clearQueue(_myQueue);
                                Conient.setControlStatus("Disconnected");
                            }
                        }
                    } else {
                        Conient.addMessage("WARNING{control link}: already disconnected");
                    }
                    break;
                case QUIT:
                    Conient.addMessage("Exiting.");
                    try {
                        // stop data and control if data up
                        if (_dataLink != null) {
                            _actionQueue.add(new Integer(STOPDATA));
                            _actionQueue.add(new Integer(DISCONNECT));
                            _actionQueue.add(new Integer(QUIT));
                            throw new IOException();
                        }                            
                        // stop control
                        if (_controlLink != null) {
                            _actionQueue.add(new Integer(DISCONNECT));
                            _actionQueue.add(new Integer(QUIT));
                            throw new IOException();
                        }
                        Conient.addMessage("Finished.");
                        // go!
                        System.exit(0);
                    } catch (IOException e) {
                        Conient.addMessage("WARNING: open connections detected - queueing stop events");
                    }
                    break;
            }
        }
    }
 
    /**
     * This method is called by the Configuration object
     * when a request is made for a server property.
     * The Configuration object is informed of the Connection
     * Handler's whereabouts on construction of the ConnectionHandler.
     *
     * This implements the STARTCONFIG command in the 1.0 protocol.
     */
    public String getConfigFromServer(String configName, String propertyName) {
        String property = "";
        if(_controlLink != null) {
            try {
                Conient.addMessage("Getting configuration from server");
                String response = null;
                _outBound.println("STARTCONFIG");
                _outBound.flush();
                response = _inBound.readLine();
                if (!response.equals("OK")) {
                    throw new IOException("server refused (" + response + ")");
                }
                
                _outBound.println(configName + ";" + propertyName);
                _outBound.flush();
                response = _inBound.readLine();
                if (response.equals("ERROR")) {
                    throw new IOException("server did not returned ERROR");
                }
                property = response;  
                _outBound.println("ENDCONFIG");
                _outBound.flush();
                response = _inBound.readLine();
                if (!response.equals("OK")) {
                    throw new IOException("server reported error when finishing configuration");
                }
            } catch (IOException e) {
                Conient.addMessage("ERROR{control link}: when getting configuration - " + e);
            }
        }
        return property;
    }

    /**
     * This method allows other classes 
     * to shutdown this connection handler.
     */
    public void shutdown() {
        _running = false;
    }
    
    /**
     * Called by the DataReader if the link
     * is unexpectedly lost.  This closes the
     * data link.
     */
    public void shutdownDataLink() {
        _actionQueue.add(new Integer(STOPDATA));
    }
    
//---PRIVATE METHODS---

    /**
     * This method performs the SETHOSTLIST command.
     * This is run by the CONNECT command.
     * It tells the server which hosts we are interested
     * in, if "" is sent, this indicates we want ALL hosts.
     *
     * @param hostList the list of hosts as gained from the config
     * @return if we succeeded in setting the host list
     */
    private boolean setHostList(String hostList) {
        boolean success = false;
        // must have a control link open
        if (_controlLink != null) {
            // data link must be closed (according to 1.1 PROTOCOL)
            if(_dataLink == null) {
                try {
                    Conient.addMessage("Setting host list to:" + hostList);
                    String response = null;
                    _outBound.println("SETHOSTLIST");
                    _outBound.flush();
                    response = _inBound.readLine();
                    if (!response.equals("OK")) {
                        throw new IOException("server refused - data link possibly still open?");
                    }
                    _outBound.println(hostList);
                    _outBound.flush();
                    response = _inBound.readLine();
                    if (!response.equals("OK")) {
                        throw new IOException("server had trouble with our request");
                    }
                    success = true;
                } catch (IOException e) {
                    Conient.addMessage("ERROR{control link}: when setting hostlist - " + e);
                }
            }
        }
        return success;
    }

    /**
     * Handles opening pipes through firewalls.
     * It checks the configuration for various entries
     * to set up the link or not.
     *
     * If a link is setup it will return the name of the 
     * new server to connect to, if a link is NOT setup,
     * it will simply return the original server name.
     *
     * The name of the machine that the pipe is setup on
     * defaults to "localhost", unless the configuration
     * specifies otherwise.
     *
     * Once it has run the firewall command it then waits
     * a set period according to the config for the firewall
     * pipe to be set up.
     *
     * The firewall process should be destroyed when the 
     * link is finished with.
     *
     * @param server the name of the server to open up the pipe to
     * @param port the port number to open up the pipe to
     * @param firewallProcess the holder for the new firewall process
     * @return the server to connect to, as determined by this routine
     */
    private String handleFirewall(String server, int port, TunnelProcess firewallProcess) {
        String firewallCommand = _configuration.getProperty("firewall.command");
        String useFirewall = _configuration.getProperty("useFirewall");
        String firewallCommandWait = _configuration.getProperty("firewall.commandwait");
        String firewallServer = _configuration.getProperty("firewall.server");
        // if we are running firewall support...then lets start it
        if (useFirewall.equals("1")) {
            // clean up the command with what we want
            firewallCommand = StringUtils.replaceText(firewallCommand, "%PORT%", new Integer(port).toString());
            firewallCommand = StringUtils.replaceText(firewallCommand, "%SERVER%", server);
            Conient.addMessage("WARNING{firewall}: firewall pipes requested, running pipe setup command \"" + firewallCommand + "\"");
            try {
                // run the command
                firewallProcess.setProcess(Runtime.getRuntime().exec(firewallCommand));
                
                // work out how long we should wait before carrying on                
                int time = 0;
                try {
                    time = Integer.parseInt(firewallCommandWait);
                } catch (NumberFormatException e) {
                    time = 0;
                }
                if (time == 0) {
                    time = DEFAULT_FIREWALL_COMMANDWAIT;
                }
                Conient.addMessage("WARNING{firewall}: waiting " + time + " seconds for command to complete!");
                // wait for the command to finish
                try {
                    Thread.sleep(time * 1000);
                } catch (InterruptedException e) {
                }
            
                // set the server we want to return
                server = firewallServer;
                if (server.equals("")) {
                    server = DEFAULT_FIREWALL_SERVER;
                }
            } catch (IOException e) {
                Conient.addMessage("ERROR{firewall}: unable to start pipe to i-scream server");
            }
        }        
        return server;
    }

    /**
     * Checks to see if the given firewall process 
     * has been created.  If it has it calls destroy()
     * on it.
     *
     * @param firewallProcess the process to check
     */
    private void closeFirewall(TunnelProcess firewallProcess) {
        if (firewallProcess.getProcess() != null) {
            firewallProcess.getProcess().destroy();
            firewallProcess.setProcess(null);
            Conient.addMessage("WARNING{firewall}: firewall process destroyed");
        }
    }

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

//---ATTRIBUTES---       

    /**
     * The state if this thread
     */
    private boolean _running = true;
    
    /**
     * A reference to the displaying class DataPanel
     */
    private DataPanel _data;
    
    /**
     * The queue that actions are added to by other parts of the system
     */
    private Queue _actionQueue;
    
    /**
     * The queue number that we are reading from on the action queue
     */
    private int _myQueue;
    
    /**
     * The control link socket
     */
    private Socket _controlLink;
    
    /**
     * The data link socket
     */
    private Socket _dataLink;
    
    /**
     * The input for the control link
     */
    private BufferedReader _inBound;
    
    /**
     * The output for the control link
     */
    private PrintWriter _outBound;
    
    /**
     * The input for the data link
     */
    private BufferedReader _dataInBound;
    
    /**
     * The output for the data link
     */
    private PrintWriter _dataOutBound;
    
    /**
     * A reference to the DataReader in use
     */
    private DataReader _dataReader;

    /**
     * A reference to the system configuration component
     */
    private Configuration _configuration = Configuration.getInstance();

    /**
     * The server we will be connecting to
     */
    private String _server = null;
    
    /**
     * The server that the config says we should connect to
     */
    private String _configuredServer = null;

    /**
     * The process used to start the firewall pipe
     * for the control link.
     */
    private TunnelProcess _controlFirewallProcess = new TunnelProcess();
    
    /**
     * The process used to start the firewall pipe
     * for the data link.
     */
    private TunnelProcess _dataFirewallProcess = new TunnelProcess();
    
//---STATIC ATTRIBUTES---

}
Revision:	1.37
Committed:	Sun Aug 1 10:40:05 2004 UTC (19 years, 9 months ago) by tdb
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.36:	+3 -3 lines
Log Message:	Catch a lot of old URL's and update them. Also remove a couple of old files that aren't used.