---

ConnectionServer.java

/*  Package Web Test Tools 
 *  Copyright (C) 2001 "Artur Hefczyc" <kobit@users.sourceforge.net>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser 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.
 *
 * $Id: ConnectionServer.java,v 1.1 2002/11/29 15:18:30 kobit Exp $
 * $Author: kobit $
 * $Date: 2002/11/29 15:18:30 $
 */

package wttools.remotecons;

import java.net.Socket;
import java.net.SocketException;
import java.io.PrintWriter;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.File;
import java.io.FileReader;
import java.io.ByteArrayOutputStream;
import java.util.List;
import java.util.ArrayList;
import java.util.Collections;
import java.util.StringTokenizer;
import java.util.Date;
import java.text.DateFormat;
import wttools.remotecons.ifc.CommandHandlerIfc;
import wttools.remotecons.ifc.LogIfc;

public class ConnectionServer extends Thread {

  protected static List active_connections =
    Collections.synchronizedList(new ArrayList());
  protected Socket          my_socket = null;
  protected BufferedReader  inst = null;
  protected PrintWriter     out = null;
  protected List            comm_handl = null;
  protected LogIfc          log_impl = null;
  protected boolean         prompt = true;
  protected boolean         echo = true;

  public ConnectionServer(Socket socket, List command_handlers)
    throws IOException
  {
    comm_handl = command_handlers;
    if (comm_handl == null) {
      comm_handl = new ArrayList();
    } // end of if (comm_handl == null)
    comm_handl.add(0, new CommandHandlerImpl());
    comm_handl.add(0, new InternalCommands());
    
    my_socket = socket;
    inst =
      new BufferedReader(new InputStreamReader(my_socket.getInputStream()));
    out = new PrintWriter(my_socket.getOutputStream(), true);
    setDaemon(true);
    start();
    active_connections.add(this);
  }

  public void run() {
    log("Connection server started for client: "+my_socket.getInetAddress().getHostAddress());
    while (inst != null) {
      try {
        if (prompt) {
          out.print(my_socket.getLocalAddress().getHostAddress()+" << "); out.flush();
        } // end of if (prompt)
        String line = "";
        if (echo) {
          char chr = (char)inst.read();
          while (chr != '\n') {
            if (chr != '\r') {
              line += chr;
            } // end of if (chr != '\r')
            out.print(chr); out.flush();
            chr = (char)inst.read();
          } // end of while (chr != '\n')
          out.print(chr); out.flush();
        } // end of if (echo)
        else {
          line = inst.readLine();
        } // end of if (echo)else
        
        log("Received command: "+line);
        if (!processCommand(line) && out != null && !line.equals("")) {
          out.println(getClass().getName()+" >> Command '"+line+"' not understood.");
        } // end of if (!processInternalCommand(line) && !comm_found)
      }
      catch (SocketException e) {
        // Probably someone closed this connection
        out = null;
        inst = null;
        my_socket = null;
      } // end of catch
      catch (IOException e) {
        e.printStackTrace();
        out = null;
        inst = null;
        my_socket = null;
      } // end of try-catch
    } // end of while (inst != null)
    active_connections.remove(this);
  }

  protected void asyncMessage(String msg) {
    out.print(msg); out.flush();
  }

  protected boolean processCommand(String line) {
    boolean comm_found = false;
    for (int i = 0; i < comm_handl.size(); i++) {
      String res = ((CommandHandlerIfc)comm_handl.get(i)).handleCommand(line);
      if (res != null && !res.equals("")) {
        if (out != null) {
          out.print(res); out.flush();
        } // end of if (out != null)
        prompt = res.endsWith("\r\n");
        comm_found = true;
      } // end of if (res != null && !res.equals(""))
    } // end of for (int i = 0; i < comm_handl.size(); i++)
    return comm_found;
  }

  protected void log(String msg) {
    if (log_impl != null) {
      log_impl.log(msg);
    } // end of if (log != null)
  }

  class InternalCommands implements CommandHandlerIfc {

    protected String last_command = null;

    public String handleCommand(String comm) {
      String ret_res = null;
      if (comm.trim().startsWith("echo")) {
        echo = !echo;
        ret_res = "Echo your command is now in '"+echo+"' mode\r\n";
      } // end of if (comm.trim().startsWith("echo"))
      if (comm.equals("quit")) {
        ret_res = "OK";
        log("'quit' received, closing connection...");
        out.println("Bye");
        out.close();
        out = null;
        try {
          inst.close();
          inst = null;
          my_socket.close();
          my_socket = null;
        } catch (Exception e) {
          e.printStackTrace();
          out = null;
          inst = null;
          my_socket = null;
        } // end of try-catch
      } // end of if (line.equals("quit"))
      if (comm.startsWith("help")) {
        ret_res = "";
        if (comm.trim().equals("help")) {
          for (int i = 0; i < comm_handl.size(); i++) {
            String res = ((CommandHandlerIfc)comm_handl.get(i)).help();
            ret_res += comm_handl.get(i).getClass().getName()+" >> "+res;
          } // end of for (int i = 0; i < comm_handl.size(); i++)
        } // end of if (comm.trim().equals("help"))
        else {
          for (int i = 0; i < comm_handl.size(); i++) {
            String res = ((CommandHandlerIfc)comm_handl.get(i)).help(comm.substring(5));
            ret_res += comm_handl.get(i).getClass().getName()+" >> "+res;
          } // end of for (int i = 0; i < comm_handl.size(); i++)
        } // end of if (comm.trim().equals("help"))else
      } // end of if (comm.startsWith("help"))
      if (comm.trim().equals("who")) {
        ret_res = "\r\n";
        for (int i = 0; i < active_connections.size(); i++) {
          ConnectionServer cs = (ConnectionServer)active_connections.get(i);
          if (cs.my_socket == my_socket) {
            ret_res += "this:  ";
          } // end of if (active_connections.get(i) == this)
          else {
            ret_res += "other: ";
          } // end of if (active_connections.get(i) == this)else
          ret_res += ""+i+". "+cs.my_socket+"\r\n";
        } // end of for (int i = 0; i < active_connections.size(); i++)
      } // end of if (comm.trim().equals("who"))
      if (comm.trim().startsWith("ls")) {
        StringTokenizer stt = null;
        if (comm.trim().length() > 2) {
          stt = new StringTokenizer(comm.substring(2));
        } // end of if (comm.trim().length() > 2)
        else {
          stt = new StringTokenizer(".");
        } // end of if (comm.trim().length() > 2)else
        while (stt.hasMoreTokens()) {
          File dir = new File(stt.nextToken());
          File[] files = dir.listFiles();
          if (files != null && files.length > 0) {
            for (int i = 0; i < files.length; i++) {
              out.println(buildFileString(files[i]));
            } // end of for (int i = 0; i < files.length; i++)
            out.println("\r\ntotal "+files.length);
          } // end of if (files != null && files.length > 0)
          else {
            out.println(dir.toString()+" is not a directory or");
            out.println("this directory does not contains any antries.");
          } // end of else
        } // end of while (stt.hasMoreTokens())
        ret_res = "\r\n";
      } // end of if (comm.trim().startsWith("ls"))
      if (comm.trim().startsWith("show")) {
        if (comm.trim().length() < 5) {
          ret_res = "Wrong number of arguments. I need file name here.\r\n";
        } // end of if (comm.trim().length() == 5)
        else {
          StringTokenizer stt = new StringTokenizer(comm.substring(5));
          while (stt.hasMoreTokens()) {
            File ff = new File(stt.nextToken());
            if (ff.exists()) {
              try {
                BufferedReader brin = new BufferedReader(new FileReader(ff));
                String line = brin.readLine();
                while (line != null) {
                  out.println(line);
                  line = brin.readLine();
                } // end of while (line != null)
                brin.close();
                ret_res = "  -----  End of file "+ff.toString()+" -----\r\n";
              } catch (Exception e) {
                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                PrintWriter pw = new PrintWriter(baos, false);
                e.printStackTrace(pw);
                pw.flush();
                ret_res = baos.toString();
                pw.close();
              } // end of try-catch
            } // end of if (ff.exists())
            else {
              ret_res = "File "+ff.toString()+" does not exists!\r\n";
            } // end of if (ff.exists())else
          } // end of while (stt.hasMoreTokens())
        } // end of if (comm.trim().length() == 5)else
      } // end of if (comm.trim().startsWith("show"))
      if (comm.trim().startsWith("close")) {
        int no = -1;
        try {
          no = Integer.parseInt(comm.substring(6).trim());
          if (no >= 0 && no < active_connections.size()) {
            ConnectionServer cs = (ConnectionServer)active_connections.get(no);
            out.print("closing "+cs.my_socket+" ... "); out.flush();
            cs.processCommand("quit");
            active_connections.remove(cs);
            ret_res = "OK\r\n";
          } // end of if (no >= 0 && no < active_connections.size())
          else {
            ret_res = "Argument out of range. Check valid ranges with 'who' command.";
          } // end of if (no >= 0 && no < active_connections.size())else
        }
        catch (StringIndexOutOfBoundsException e) {
          ret_res = "Wrong number of arguments. I need integer number here.";
        } // end of catch
        catch (NumberFormatException e) {
          ret_res = "Bad argument to 'close' command. I need integer number here.";
        } // end of try-catch
      } // end of if (comm.trim().startsWith("close"))
      return ret_res;
    }

    protected String buildFileString(File ff) {
      StringBuffer res = new StringBuffer(80);
      res.append(setForTrueOrFalse(ff.isDirectory(), "d", "-"));
      res.append(setForTrueOrFalse(ff.canRead(), "r", "-"));
      res.append(setForTrueOrFalse(ff.canWrite(), "w", "-"));
      res.append(setForTrueOrFalse(ff.isHidden(), "h", "-"));
      res.append("   "+ff.length());
      while (res.length() < 17) {
        res.append(' ');
      } // end of while (res.length() < 17)
      DateFormat dt = DateFormat.getDateInstance();
      res.append("   "+dt.format(new Date(ff.lastModified())));
      res.append("   "+ff.getName());
      res.append(setForTrueOrFalse(ff.isDirectory(), "/", ""));
      return res.toString();
    }

    protected String setForTrueOrFalse(boolean cond, String for_true, String for_false) {
      if (cond) {
        return for_true;
      } // end of if (cond)
      else {
        return for_false;
      } // end of if (cond)else
    }

    public String help() {
      String help_text =
        "\r\n"+
        "  echo           - switch 'echo' mode (for windows telnet is really needed)\r\n"+
        "  quit           - close connection to server\r\n"+
        "  who            - list all active connections\r\n"+
        "  close n        - close active connection number 'n'\r\n"+
        "  help [command] - display this message info\r\n"+
        "  ls [dir]*      - list content of given directory\r\n"+
        "  show filename  - display content of given file\r\n";
      return help_text;
    }

    public String help(String command) {
      return "";
    }

    public CommandHandlerIfc getInstance() {
      return new InternalCommands();
    }
    
  }
  
  
}// ConnectionServer
/*
 * Changes in file:
 *
 * $Log: ConnectionServer.java,v $
 * Revision 1.1  2002/11/29 15:18:30  kobit
 * Refactoring packages, thanks to RefactorIT
 *
 * Revision 1.1  2002/03/21 09:57:27  kobit
 * 1. Small but important redesign:
 *    - package 'remotecons.wttools' was removed and all classes
 *      were moved to 'remotecons' package
 * 2. Added 'asyncMessage' to provide sending messages to remote clients
 *    in asynchronous mode
 *
 * Revision 1.14  2002/01/23 15:49:24  kobit
 * Java doc for all classes finishedant clean dist
 *
 * Revision 1.13  2002/01/21 16:30:19  kobit
 * Commenting out code in progress
 *
 * Revision 1.12  2002/01/18 14:18:46  kobit
 * Next part of javadoc documentation
 *
 * Revision 1.11  2002/01/17 22:37:44  kobit
 * Javadoc comments for classes in progress...
 *
 * Revision 1.10  2002/01/14 17:02:02  kobit
 * Changed interface to CommandHandlersIfc
 *
 * Revision 1.9  2002/01/13 18:09:29  kobit
 * Added methods for closing remote console
 *
 * Revision 1.8  2002/01/13 17:27:13  kobit
 * Corrected interface for external command handlers
 *
 * Revision 1.7  2002/01/13 17:14:45  kobit
 * Added commands: ls, show file, echo and support for windows telnet
 *
 * Revision 1.6  2002/01/12 16:19:21  kobit
 * Started code for command 'show filename'
 *
 * Revision 1.5  2002/01/12 12:54:22  kobit
 * Bug in closing connection command fixed
 *
 * Revision 1.4  2002/01/12 12:25:34  kobit
 * Almost finished basic code
 *
 *
 */

---