source: fedd/abac-src/ttg/test/ClientRTMLService.java @ 53c649f

package com.nailabs.abac.test;

import com.nailabs.abac.process.*;
import com.nailabs.abac.credential.*;
import edu.stanford.peer.rbtm.credential.*;
import java.rmi.*;
import java.rmi.server.*;
import java.rmi.registry.*;
import java.util.*;

import javax.swing.*;
import java.awt.*;
import java.awt.event.*;

public class ClientRTMLService extends UnicastRemoteObject 
    implements NegotiatorFactory
{
    static HashMap configs = new HashMap();

    public static final int PORT = 6666;
    FrontierManager frontiers;
    boolean isDone = false;

    static {
        try {
            LocateRegistry.createRegistry(1099);
            System.out.println("Creating default registry on port 1099...");
        } catch(Exception ex) {
            ex.printStackTrace();
        }
    }

    public ClientRTMLService(int port) 
        throws RemoteException
    {
        super();
    }

    public ClientRTMLService() throws RemoteException 
    {
        this(PORT);
    }

    public Remote getNegotiator(String entity, Negotiator peer) 
        throws RemoteException
    {
        System.out.println("Looking up negotiator for " + entity);
        if(!FrontierManager.getFrontiers().contains(entity))return null;
        Entity oppo = peer.getSelf();
        HashMap config = (HashMap)configs.get(entity);
        Entity self = (Entity)config.get("EntityID");
        config.put("Peer", oppo);
        NegotiationContext context = new NegotiationContext(config);
        TextGraphObserver observer = new TextGraphObserver(entity);
        RMINegotiator agent = null;
        Remote stub = null;
        // grab a set of credential graph engines
        System.out.println("Received request for " + entity + " from " +
                           oppo.toString());
        try {
            agent = new RMINegotiator(context);
            context.getGraph().addObserver(observer);
            //stub =  (Negotiator)exportObject(agent);
            stub = (agent == null)? null: toStub(agent);
        }
        catch(Exception ex) {
            ex.printStackTrace();
        }
        System.out.println("agent = " + agent);
        System.out.println("stub = " + stub);
        // construct a new negotiation context
        return stub;
    }

    public void kill() throws RemoteException { isDone = true; }

    public static void main(String arg[]) 
    {
        //if (System.getSecurityManager() == null) {
        //    System.setSecurityManager(new RMISecurityManager());
        //}
        Vector entities = new Vector(arg.length);
        for(int i = 0; i < arg.length; i++) 
        {
            System.out.print("Loading from ");
            System.out.println(arg[i]);
            //check for int port number (TBD)
            HashMap config = RtmlTest.loadConfiguration(arg[i]);
            RtmlEntity id = (RtmlEntity)config.get("EntityID");
            //System.out.print(" (config = ");
            //System.out.print(config);
            //System.out.println("), ");
            FrontierManager.addFrontier(config);
            configs.put(id.getHash(), config);
            configs.put(id.getName(), config);
            entities.add(id.getName());
            System.out.println("\nLoaded configuration for " + id);
        }
        System.out.println(".");
        try {
            ClientRTMLService service = 
                new ClientRTMLService();
            Naming.rebind("ClientRTMLService", service);
            UIManager.setLookAndFeel
                (UIManager.getCrossPlatformLookAndFeelClassName());
        }
        catch(Exception ex) {
            ex.printStackTrace();
        }
        //...create the components to go into the frame...
        final JFrame frame = new JFrame("TTG Callback Admin");
        JButton discoverButton = new JButton("Rediscover");
        JButton exitButton = new JButton("Exit");
        JPanel buttonBar = new JPanel();
        final JList list = new JList(entities);
        JLabel title = new JLabel("Negotiators Running:");
        JPanel contents = new JPanel();
        JScrollPane listScroller = new JScrollPane(list);
        Dimension dim = new Dimension(300, 80);

        title.setAlignmentX(Component.CENTER_ALIGNMENT);
        list.setSelectedIndex(0);
        listScroller.setPreferredSize(new Dimension(250, 80));
        listScroller.setMinimumSize(new Dimension(250, 80));
        listScroller.setAlignmentX(Component.CENTER_ALIGNMENT);
        exitButton.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent e) {
                    System.exit(0);
                }
            });
        discoverButton.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent e) {
                    String name = (String)list.getSelectedValue();
                    System.out.println("Rediscovering for " + name);
                    HashMap config = (HashMap)configs.get(name);
                    RtmlEntity id = (RtmlEntity)config.get("EntityID");
                    RtmlFrontier frontier = 
                        (RtmlFrontier)RtmlFrontier.getFrontier(id);
                    frontier.init(config);
                }
            });

        contents.setLayout(new BoxLayout(contents, BoxLayout.Y_AXIS));
        buttonBar.add(discoverButton);
        buttonBar.add(exitButton);
        buttonBar.setAlignmentX(Component.CENTER_ALIGNMENT);
                                 
        //...stick them in a container named contents...
        frame.getContentPane().add(contents);
        contents.add(Box.createRigidArea(new Dimension(0,5)));
        contents.add(title);
        contents.add(Box.createRigidArea(new Dimension(0,5)));
        contents.add(listScroller);
        contents.add(buttonBar);
        //contents.setPreferredSize(dim);
        //contents.setMinimumSize(dim);
        
        //Finish setting up the frame, and show it.
        //frame.addWindowListener(...);
        frame.pack();
        frame.setLocation(new Point(100,100));
        frame.setVisible(true);
    }
}