From aff598c9c6364059fb3e698f955ed524ee4e081d Mon Sep 17 00:00:00 2001
From: jestin
Date: Thu, 3 Aug 2006 15:54:54 +0000
Subject: a few changes towards the tunnel manager

---
 src/Puppet.java | 121 ++++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 105 insertions(+), 16 deletions(-)

(limited to 'src/Puppet.java')

diff --git a/src/Puppet.java b/src/Puppet.java
index d122d52..e2a90ee 100644
--- a/src/Puppet.java
+++ b/src/Puppet.java
@@ -5,35 +5,124 @@ import java.io.*;
 import java.util.*;
 import org.apache.regexp.*;
 
-class Puppet extends Ivy {
+class Puppet {
 
-  Hashtable bound = new Hashtable(); // clef: l'ID de la regexp choisi par le ghost, valeur: l'Int de l'abonnement local
-  String myDomain;
+  // the mapping between Ghost regexp and local bus regexp numbers
+  Hashtable bound = new Hashtable(); 	// ghostID localID
+  Hashtable regexps = new Hashtable(); 	// ghostID textRegexp
+  String domain;
+  String appName;
   ProxyClient pc;
+  String id;
+  boolean started;
+  PuppetIvy bus;
 
-  Puppet(ProxyClient pc,String domain) {
-    super("noname","noname not ready",null);
-    myDomain=domain;
+  Puppet(ProxyClient pc,String id,String domain) {
+    this.domain=domain;
     this.pc=pc;
+    this.id=id;
   }
 
-  void parse(String s){
-    System.out.println("the puppet must parse "+s);
+  void sendGhost(String s) { pc.send("ForwardGhost id="+id+" buffer="+s); }
+
+  class ForwardMessenger implements IvyMessageListener {
+    String localId,ghostId;
+    public ForwardMessenger(String ghostId,String re) throws IvyException {
+      this.ghostId=ghostId;
+      this.localId = (new Integer(bus.bindMsg(re,ForwardMessenger.this))).toString();
+      bound.put(ghostId,localId);
+    }
+    public void receive(IvyClient ic,String args[]) {
+      String tosend = IvyClient.Msg+" "+ghostId+IvyClient.StartArg;
+      for (int i=0;i<args.length;i++) tosend+=args[i]+IvyClient.EndArg;
+      sendGhost(tosend);
+    }
+  } // ForwardMessenger
+
+  void addRegexp(String ghostId,String re) {
+    regexps.put(ghostId,re);
+    try {
+      if (started) new ForwardMessenger(ghostId,re);
+    } catch( IvyException ie) { ie.printStackTrace(); }
+  }
+
+  void removeRegexp(String ghostId) {
+    try {
+      bus.unBindMsg(Integer.parseInt((String)bound.remove(ghostId)));
+    } catch( IvyException ie) { ie.printStackTrace(); }
+  }
+
+  void stop() {
+    if (started) bus.stop();
   }
 
-  protected IvyClient createIvyClient(Socket s,int port, boolean domachin) throws IOException {
-    return new PuppetClient();
+  // ivy forwarded protocol message
+  static final RE ivyProto = new RE("(\\d+) (\\d+)\\02(.*)");
+  void parse(String s) throws IvyException {
+    if (!ivyProto.match(s)) { System.out.println("Puppet error, can't parse "+s); return; } 
+    int pcode=Integer.parseInt(ivyProto.getParen(1));
+    String pid=ivyProto.getParen(2);
+    String args=ivyProto.getParen(3);
+    trace("must parse code:"+pcode+" id:"+pid+" args:"+args);
+    switch (pcode) {
+      case IvyClient.AddRegexp: // the Ghost's peer subscribes to something
+	addRegexp(pid,args);
+      	break;
+      case IvyClient.DelRegexp: // the Ghost's peer unsubscribes to something
+	removeRegexp(pid);
+        break;
+      case IvyClient.Bye:	// the Ghost's peer disconnects gracefully
+      	bus.stop();
+	// TODO end of the puppet ?
+      	break;
+      case IvyClient.Die:
+      	// the Ghost's peer wants to ... kill ProxyClient ?
+      	break;
+      case IvyClient.Msg:
+      	// the Ghost's peer sends a message to ProxyClient, with regard to one
+	// of our subscriptions
+	// TODO à qui le faire passer ?
+      	break;
+      case IvyClient.SchizoToken:
+      	appName = args;
+	bus = new PuppetIvy(appName,appName+" fakeready",null);
+	for (Enumeration e = regexps.keys();e.hasMoreElements();) {
+	  String ghostId=(String)e.nextElement();
+	  String re=(String)regexps.get(ghostId);
+	  new ForwardMessenger(ghostId,re);
+	}
+	started=true;
+	trace("starting the bus on "+domain);
+	bus.start(domain);
+        break;
+      case IvyClient.Error:
+      case IvyClient.EndRegexp:
+      case IvyClient.DirectMsg:
+      case IvyClient.Ping:
+      case IvyClient.Pong:
+      default:
+    	trace("unused Ivy protocol code "+pcode);
+    }
   }
 
-  class PuppetClient extends IvyClient {
-    protected synchronized void sendBuffer( String buffer ) throws IvyException {
-      super.sendBuffer(buffer); // and to all the agents on the Ghost bus ? I'm not sure
+  class PuppetIvy extends Ivy {
+    PuppetIvy(String name,String ready,IvyApplicationListener ial){super(name,ready,ial);}
+    protected IvyClient createIvyClient(Socket s,int port, boolean domachin) throws IOException {
+      return new PuppetIvyClient(PuppetIvy.this,s,port,domachin);
     }
+    int getAP() {return applicationPort;}
+  }
 
+  class PuppetIvyClient extends IvyClient {
+    PuppetIvyClient(Ivy bus,Socket s,int port,boolean b) throws IOException  { super(bus,s,port,b); }
+    protected synchronized void sendBuffer( String s ) throws IvyException {
+      super.sendBuffer(s); // and to all the agents on the Ghost bus ? I'm not sure
+    }
     protected boolean newParseMsg(String s) throws IvyException {
-      return super.newParseMsg(s);
+      return super.newParseMsg(s); // I'm a normal Ivy citizen
     }
-  } // class PuppetClient
+  }
 
-}
+  void trace(String s) { System.out.println("Puppet["+id+"] "+s);}
 
+}
-- 
cgit v1.1