X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=src%2Forg%2Fibex%2Fjinetd%2FPort.java;h=ad0331c6088377c8b217f58eac19115e21514005;hb=8c3227dda12abaa856c255d27c3811e9c29167e3;hp=2aded0ad53d0eac06e1e65f90d952d57cf26f3b6;hpb=9219699b69ebb0c743d55862de0e43d43d270f49;p=org.ibex.jinetd.git

diff --git a/src/org/ibex/jinetd/Port.java b/src/org/ibex/jinetd/Port.java
index 2aded0a..ad0331c 100644
--- a/src/org/ibex/jinetd/Port.java
+++ b/src/org/ibex/jinetd/Port.java
@@ -1,3 +1,7 @@
+// Copyright 2000-2005 the Contributors, as shown in the revision logs.
+// Licensed under the Apache Public Source License 2.0 ("the License").
+// You may not use this file except in compliance with the License.
+
 package org.ibex.jinetd;
 import org.ibex.util.*;
 import org.ibex.io.*;
@@ -6,98 +10,70 @@ import java.util.*;
 import java.net.*;
 import java.util.zip.*;
 
-public class Port extends Loader {
+// Feature: port-level redirects
+public class Port extends TreeClassLoader {
 
-    private final NetworkInterface iface;
-    private final int port;
-    private final Thread listener;
+    final ThreadPool tp;
+    final InetAddress bindTo;
+    final int port;
 
-    public Port(String path, NetworkInterface iface, int port) {
-        super(path);
-        this.iface = iface;
-        this.port = port;
-        this.listener = new PortThread();
-        listener.start();
+    private static Hash cache = new Hash();
+    public static Port newPort(String path, InetAddress bindTo, int port, ThreadPool tp) throws IOException {
+        String canonical = new File(path).getCanonicalPath();
+        Port p = (Port)cache.get(canonical);
+        if (p == null) cache.put(canonical, p = new Port(path, bindTo, port, tp));
+        else Log.warn(Port.class, "   sharing " + bindTo+":"+port+" -> "+ (p.bindTo+":"+p.port));
+        p.spawn(bindTo, port);
+        return p;
     }
 
-    public void changed(Watched w) {
-        //Log.warn(this, "Port: noticed change in " + w);
-        super.changed(w);
-    }
+    void spawn(InetAddress bindTo, int port) { new PortThread(bindTo, port).start(); }
+    private static final ClassLoader parentLoader = Port.class.getClassLoader();
+    private Port(String path, InetAddress bindTo, int port, ThreadPool tp) { 
+        super(new File(path), parentLoader);
+        this.bindTo = bindTo; this.port = port; this.tp = tp; }
 
     void dispatch(final Connection conn) throws Exception {
-        getClassLoader();
-        String[] list = list();
-        for(int i=0; i<list.length; i++) {
-            if (!list[i].endsWith(".jar")) continue;
-            //Log.warn(this, "checking " + (this.path + File.separatorChar + list[i]));
-            File f = new File(this.path + File.separatorChar + list[i]);
-            ZipInputStream zis = new ZipInputStream(new FileInputStream(f));
-            for(ZipEntry ze = zis.getNextEntry(); ze != null; ze = zis.getNextEntry()) {
-                String name = ze.getName();
-                if (name.endsWith(".class"))
-                    dispatch(conn, name.substring(0, name.length() - ".class".length()).replace('/', '.'));
-            }
-        }
-        check(conn, new File(getAbsolutePath() + File.separatorChar + "BIN"));
-    }
-
-    void check(Connection conn, File f) throws Exception {
-        //Log.warn(this, "check(" + f.getAbsolutePath() + ")");
-        if (!f.exists()) return;
-        if (!f.isDirectory()) {
-            if (!f.getAbsolutePath().endsWith(".class")) return;
-            String name = f.getAbsolutePath().substring(getAbsolutePath().length() + 5);
-            name = name.substring(0, name.length() - ".class".length()).replace(File.separatorChar, '.');
-            dispatch(conn, name);
-        } else {
-            String[] list = f.list();
-            for(int i=0; i<list.length; i++) check(conn, new File(f.getAbsolutePath() + File.separatorChar + list[i]));
-        }
-    }
-    void dispatch(final Connection conn, String name) throws Exception {
-        //Log.info(this, "attempting class " + name);
-        try {
-            Class c = getClassLoader().loadClass(name);
-            if (c != null) {
-                if (Listener.class.isAssignableFrom(c)) {
-                    Log.error(this, "dispatching connection on port " + port + " to " +
-                              c.getName());
-                    final Listener l = (Listener)c.newInstance();
-                    new Thread() { public void run() {
-                        Log.clearnotes();
-                        try {
-                            l.accept(conn);
-                        } catch (Exception e) {
-                            Log.error(l.getClass(), "Listener threw exception");
-                            Log.error(l.getClass(), e);
-                        } finally {
-                            conn.close();
-                        }
-                    } }.start();
-                    return;
-                }
+        tp.appendTask(new Runnable() { public void run() {
+            String local = conn.getLocalAddress() + ":" + conn.getLocalPort();
+            String remote = conn.getRemoteHostname() + ":" + conn.getRemotePort();
+            try {
+                // FIXME FIXME
+                Class c = loadClass("org.ibex.mail.Main");
+                if (c == null) throw new RuntimeException("couldn't find listener");
+                Log.info("["+local+"]", "connection from " + remote + " => " + c.getName());
+                Log.clearnotes();
+                Thread.currentThread().setContextClassLoader(Port.this);
+                Listener l = (Listener)c.newInstance();
+                l.accept(conn);
+            } catch (org.ibex.io.Stream.EOF eof) {
+                Log.warn(this, "end of stream reached handling connection from " +
+                         conn.getRemoteHostname() + ":" + conn.getRemotePort());
+            } catch (Exception e) {
+                Log.error(this, e);
+                conn.close();
+            } finally {
+                conn.close();
             }
-        } catch (Exception e) {
-            Log.error(this, e);
-        }
+        } });
     }
 
     private class PortThread extends Thread {
+        InetAddress bindTo;
+        int port;
+        public PortThread(InetAddress bindTo, int port) { this.bindTo = bindTo; this.port = port; }
         public void run() {
             try {
-                Log.warn(this, "Now listening on interface " + iface + ", port " + port);
-                ServerSocket ss = new ServerSocket(port);
+                Log.warn(this, "Now listening on address " + (bindTo==null?"all interfaces":bindTo.toString()) + ", port " + port);
+                ServerSocket ss = bindTo == null ? new ServerSocket(port) : new ServerSocket(port, 0, bindTo);
                 for(Socket s = ss.accept(); ; s = ss.accept()) try {
-                    Log.warn(this, "accepted connection on port " + port);
-                    dispatch(new Connection(s, "megacz.com"));
-                    Log.warn(this, "done searching for service on port " + port);
+                    dispatch(new Connection(s, Main.defaultDomain));
                 } catch (Exception e) { Log.warn(Port.class, e); }
             } catch (Exception e) { Log.error(Port.class, e);
             } catch (Throwable t) {
                 Log.error(this, "serious error, aborting VM");
                 Log.error(this, t);
-                Root.reboot();
+                Main.reboot();
             }
         }
     }