X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=src%2Forg%2Fibex%2Fjinetd%2FPort.java;h=200ce2c34e9fccbc7e775795588b9b088489386d;hb=HEAD;hp=d7f850d42f191506841b8d9b2d6393c5e4cd403f;hpb=99f8fd894be14226a64f1889657edf863edf8afb;p=org.ibex.jinetd.git

diff --git a/src/org/ibex/jinetd/Port.java b/src/org/ibex/jinetd/Port.java
index d7f850d..200ce2c 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,106 +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 InetAddress bindTo;
-    private final int port;
-    private final Thread listener;
+    final ThreadPool tp;
+    final InetAddress bindTo;
+    final int port;
 
-    public Port(String path, InetAddress bindTo, int port) {
-        super(path);
-        this.bindTo = bindTo;
-        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 = Main.getRootClassLoader();
+    private Port(String path, InetAddress bindTo, int port, ThreadPool tp) { 
+        super(new File(path), parentLoader);
+        this.bindTo = bindTo; this.port = port; this.tp = tp; }
 
-    boolean 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]);
-            FileInputStream fis = null;
+    void dispatch(final Connection conn) throws Exception {
+        tp.appendTask(new Runnable() { public void run() {
+            String local = conn.getLocalAddress() + ":" + conn.getLocalPort();
+            String remote = conn.getRemoteHostname() + ":" + conn.getRemotePort();
             try {
-                fis = new FileInputStream(f);
-                ZipInputStream zis = new ZipInputStream(fis);
-                for(ZipEntry ze = zis.getNextEntry(); ze != null; ze = zis.getNextEntry()) {
-                    String name = ze.getName();
-                    if (name.endsWith(".class"))
-                        if (dispatch(conn, name.substring(0, name.length() - ".class".length()).replace('/', '.')))
-                            return true;
-                }
-            } finally { if (fis != null) fis.close(); }
-        }
-        if (check(conn, new File(getAbsolutePath() + File.separatorChar + "BIN"))) return true;
-        return false;
-    }
-
-    boolean check(Connection conn, File f) throws Exception {
-        if (!f.exists()) return false;
-        if (!f.isDirectory()) {
-            if (!f.getAbsolutePath().endsWith(".class")) return false;
-            String name = f.getAbsolutePath().substring(getAbsolutePath().length() + 5);
-            name = name.substring(0, name.length() - ".class".length()).replace(File.separatorChar, '.');
-            if (dispatch(conn, name)) return true;
-        } else {
-            String[] list = f.list();
-            for(int i=0; i<list.length; i++) 
-                if (check(conn, new File(f.getAbsolutePath() + File.separatorChar + list[i])))
-                    return true;
-        }
-        return false;
-    }
-
-    boolean dispatch(final Connection conn, String name) throws Exception {
-        try {
-            final ClassLoader cl = getClassLoader();
-            final Class c = cl.loadClass(name);
-            if (c == null) return false;
-            if (!(Listener.class.isAssignableFrom(c) && c != Listener.class)) return false;
-            Log.info(this, "dispatching connection on port " + port + " to " + c.getName());
-            new Thread(tg, new Runnable() { public void run() {
+                // 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() + ":"+conn.getLocalPort());
                 Log.clearnotes();
-                Thread.currentThread().setContextClassLoader(cl);
-                try {
-                    Listener l = (Listener)c.newInstance();
-                    l.accept(conn);
-                } catch (Exception e) {
-                    Log.error(c, "Listener threw exception");
-                    Log.error(c, e);
-                } finally {
-                    conn.close();
-                }
-            } }).start();
-            return true;
-        } catch (Exception e) { Log.error(this, e); }
-        return false;
+                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();
+            }
+        } });
     }
 
     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 address " + (bindTo == null ? "all interfaces" : bindTo.toString()) +
-                         ", port " + 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 {
-                    if (!dispatch(new Connection(s, "megacz.com"))) {
-                        Log.warn(this, "no handler for connection on port " + port);
-                        s.close();
-                    }
+                    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();
             }
         }
     }