socketcontroller.java

移动Agent编程工具Naplet

package naplet.nsocket;

import java.io.*;
import java.net.*;
import java.util.*;
import java.math.BigInteger;
import java.security.*;
import javax.security.auth.*;
import javax.security.auth.callback.*;
import javax.security.auth.login.*;
import naplet.NapletID;
import naplet.server.*;
import naplet.tracing.*;
import naplet.*;

/**
 * Controller of this all NapletSocket connection in this machine.
 * It stores all connection information and provides a channel for
 * communiction.
 *
 * Only one controller for a machine.
 * <p>Title: </p>
 * <p>Description: </p>
 * <p>Copyright: Copyright (c) 2003</p>
 * <p>Company: </p>
 * @version 1.0
 */
public class SocketController
    implements Runnable
{
    /** single instance for  controller*/
    private static SocketController instance = null;

    /**channel for exchange message*/
    public static ControlChannel channel = null;
    /** table for NapletSocket*/
    private static SocketTable socketTable = new SocketTable();
    /**table for server socket, key is socket ID*/
    private static SocketTable serverSocketTable = new SocketTable();
    /**table for server socket, key is naplet ID*/
    private static SocketTable nidServerTable = new SocketTable();
    /**running status for current control socket*/
    private static boolean bStop;
    /**serversocket of this control channel, use to read msg from client*/
    private static ServerSocket controlServer;
    /** client part of this control channel, create here and keep a copy in
     * every napletsocket */
    private static Socket controlClient;
    /** whether to output some debug info */
    static final boolean debug = false;
    /** whether check security by changing subject */
    private static boolean CHECKRIGHT = false;
    /** whether consider security when setup a connection */
    static boolean GENKEY = false;
    static int ControlPort;
    static int UDPPort;
    private static ServerProperty property;
    public SocketController( ServerProperty property )
        throws IOException
    {
        this.property = property;

        ControlPort = property.getControlPort();
        UDPPort = property.getControlChannelPort();

        channel = new ControlChannel();

        // start control channel thread
        ( new Thread( channel, "control channel" ) ).start();

    }

    public SocketController()
    {
    }

    public static SocketController getInstance( ServerProperty property )
        throws IOException
    {
        if ( instance == null )
        {
            instance = new SocketController( property );
        }

        return instance;
    }

    /**
     * Called from NapletSocket constructor. Creates a data socket
     * and generates
     * some control info.
     *
     * @param nsocket
     * @param sid
     * @param nid
     * @return
     * @throws IOException
     */
    public static Socket registerSocket( NapletSocket nsocket, String sid,
                                         NapletID nid )
        throws IOException, NoSocketControllerException
    {
        if ( instance == null )
        {
	   throw new NoSocketControllerException(
	      "\n\tSocketController is not installed.\n"
	      + "\n\tPlease provide the SocketControllerPort \n"
	      + "\tand ControlChannelPort in server configuration file.\n" );
        }

        if ( nsocket == null )
        {
            throw new NullPointerException( "nsocket" );
        }

        if ( sid == null )
        {
            throw new NullPointerException( "socket id" );
        }

        if ( nid == null )
        {
            throw new NullPointerException( "naplet id" );
        }

        if ( CHECKRIGHT )
        {
            authorize();
        }

        String server = locateNaplet( nid );

        // socket for data, connect to remote server socket
        Socket sock = new Socket( server, ControlPort );

        socketTable.put( sid, nsocket );

        // should add to nidSocketTable, similiar to server socket,
        // for suspendAll(nid);
        return sock;
    }

    /**
     * Call from socket handoff in napletsocket(socket)
     * Already has data socket under nsocket because of socket handoff.
     * No need to create a new one.
     * @param nsocket
     * @param id
     */
    public static void registerSocket( NapletSocket nsocket, String sid )
        throws IOException, NoSocketControllerException
    {
        if ( instance == null )
        {
	   throw new NoSocketControllerException(
	      "\n\tSocketController is not installed.\n"
	      + "\n\tPlease provide the SocketControllerPort \n"
	      + "\tand ControlChannelPort in server configuration file.\n" );
         }

        if ( nsocket == null )
        {
            throw new NullPointerException( "nsocket" );
        }
        if ( sid == null )
        {
            throw new NullPointerException( "sid" );
        }

        // if suspend/resume in local machine, then may already
        // contain the key
        if ( !socketTable.containsKey( sid ) )
        {
            socketTable.put( sid, nsocket );
            // should add to nidSocketTable, similiar to server socket
        }
    }

    /**
     * No use in this version. just for reference. Called from NapletSocket
     * @param nsocket
     * @param inetaddress
     * @param i
     * @param id
     * @return
     * @throws IOException
     */
    public static Socket registerSocket( NapletSocket nsocket,
                                         InetAddress inetaddress,
                                         int i, String id )
        throws IOException, NoSocketControllerException
    {

       if ( instance == null )
        {
             throw new NoSocketControllerException(
	      "\n\tSocketController is not installed.\n"
	      + "\n\tPlease provide the SocketControllerPort \n"
	      + "\tand ControlChannelPort in server configuration file.\n" );
        }


        if ( nsocket == null )
        {
            throw new NullPointerException( "nsocket" );
        }

        if ( id == null )
        {
            throw new NullPointerException( "id" );
        }

        if ( inetaddress == null )
        {
            throw new NullPointerException( "remoteAddress" );
        }

        if ( i <= 0 )
        {
            throw new IllegalArgumentException(
                "iRemotePort must be greater than 0" );
        }

        socketTable.put( id, nsocket );
        try
        {
            Socket socket1 = new Socket( inetaddress, i );
            return socket1;
        }
        catch ( IOException ioexception )
        {
            socketTable.remove( id );
            throw ioexception;
        }
    }

    /**
     * Create a server socket and generate some control info.
     * @param serversocket
     * @param s socket ID
     */
    public static ServerSocket registerServerSocket(
        NapletServerSocket serverNSocket,
        NapletID nid, String sid )
        throws IOException, NoSocketControllerException
    {
        if ( instance == null )
        {
              throw new NoSocketControllerException(
	      "\n\tSocketController is not installed.\n"
	      + "\n\tPlease provide the SocketControllerPort \n"
	      + "\tand ControlChannelPort in server configuration file.\n" );
        }

        if ( serverNSocket == null )
        {
            throw new NullPointerException( "serverSocket" );
        }
        if ( sid == null )
        {
            throw new NullPointerException( "id" );
        }

        if ( CHECKRIGHT )
        {
            authorize();
        }

        // port management , return a free port upon request.
        ServerSocket ss = null;

        while ( ss == null )
        {
            try
            {
                ss = new ServerSocket( Global.TCPPORT++ );
            }
            catch ( BindException be )
            {
                System.out.println( be.getMessage()
                                    + ",try to get a new one.." );
            }
        }

        serverSocketTable.put( sid, serverNSocket );

        log( "---put nid to table:" + nid );
        nidServerTable.put( nid.toString(), serverNSocket );

        return ss;
    }

    /**
     * create socket client for napletsocket
     * @param host
     * @param port
     * @return
     * @throws IOException
     */
    public static synchronized Socket createSocketChannel(
        InetAddress host, int port )
        throws IOException, NoSocketControllerException
    {
        if ( instance == null )
        {
	   throw new NoSocketControllerException(
	      "\n\tSocketController is not installed.\n"
	      + "\n\tPlease provide the SocketControllerPort \n"
	      + "\tand ControlChannelPort in server configuration file.\n" );
        }

        if ( controlClient == null )
        {
            controlClient = new Socket( host, port );
        }

        return controlClient;
    }

    /**
     * Used to create socket for socket controller
     * @param host
     * @param port
     * @return
     * @throws IOException
     */
    public static Socket createSocket( InetAddress host, int port )
        throws IOException, NoSocketControllerException
    {
        if ( instance == null )
        {
	   throw new NoSocketControllerException(
	      "\n\tSocketController is not installed.\n"
	      + "\n\tPlease provide the SocketControllerPort \n"
	      + "\tand ControlChannelPort in server configuration file.\n" );
        }

        Socket sock = new Socket( host, port );
        return sock;
    }

    /**
     * For use in navigator.
     * Gets all napletsocket in this machine.
     * @return
     */
    public static Hashtable getSocketTable()
    {
        return socketTable.getTable();
    }

    public static void setSocketTable( Hashtable table )
    {
        if ( table == null )
        {
            return;
        }

        for ( Enumeration enum = table.keys();
              enum.hasMoreElements(); )
        {

            String key = ( String ) enum.nextElement();
            if ( socketTable.containsKey( key ) )
            {
                log( "wron update sk tbl aft land:already has the key" );
                socketTable.remove( key );
            }
            socketTable.put( key, table.get( key ) );
        }
    }

    /**
     * Get all NapletServerSocket in this host.
     * @return
     */
    public static Hashtable getServerSocketTable()
    {
        return serverSocketTable.getTable();
    }

    public static void setServerSocketTable( Hashtable table )
    {
        if ( table == null )
        {