diff src/com/trilead/ssh2/channel/StreamForwarder.java @ 0:0ce5cc452d02

initial version
author Carl Byington <carl@five-ten-sg.com>
date Thu, 22 May 2014 10:41:19 -0700
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/com/trilead/ssh2/channel/StreamForwarder.java	Thu May 22 10:41:19 2014 -0700
@@ -0,0 +1,95 @@
+
+package com.trilead.ssh2.channel;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.Socket;
+
+/**
+ * A StreamForwarder forwards data between two given streams.
+ * If two StreamForwarder threads are used (one for each direction)
+ * then one can be configured to shutdown the underlying channel/socket
+ * if both threads have finished forwarding (EOF).
+ *
+ * @author Christian Plattner, plattner@trilead.com
+ * @version $Id: StreamForwarder.java,v 1.1 2007/10/15 12:49:56 cplattne Exp $
+ */
+public class StreamForwarder extends Thread {
+    final OutputStream os;
+    final InputStream is;
+    final byte[] buffer = new byte[Channel.CHANNEL_BUFFER_SIZE];
+    final Channel c;
+    final StreamForwarder sibling;
+    final Socket s;
+    final String mode;
+
+    StreamForwarder(Channel c, StreamForwarder sibling, Socket s, InputStream is, OutputStream os, String mode)
+    throws IOException {
+        this.is = is;
+        this.os = os;
+        this.mode = mode;
+        this.c = c;
+        this.sibling = sibling;
+        this.s = s;
+    }
+
+    public void run() {
+        try {
+            while (true) {
+                int len = is.read(buffer);
+
+                if (len <= 0)
+                    break;
+
+                os.write(buffer, 0, len);
+                os.flush();
+            }
+        }
+        catch (IOException ignore) {
+            try {
+                c.cm.closeChannel(c, "Closed due to exception in StreamForwarder (" + mode + "): "
+                                  + ignore.getMessage(), true);
+            }
+            catch (IOException e) {
+            }
+        }
+        finally {
+            try {
+                os.close();
+            }
+            catch (IOException e1) {
+            }
+
+            try {
+                is.close();
+            }
+            catch (IOException e2) {
+            }
+
+            if (sibling != null) {
+                while (sibling.isAlive()) {
+                    try {
+                        sibling.join();
+                    }
+                    catch (InterruptedException e) {
+                    }
+                }
+
+                try {
+                    c.cm.closeChannel(c, "StreamForwarder (" + mode + ") is cleaning up the connection", true);
+                }
+                catch (IOException e3) {
+                }
+            }
+
+            if (s != null) {
+                try {
+                    s.close();
+                }
+                catch (IOException e1) {
+                }
+            }
+        }
+    }
+}