Mercurial > 510Connectbot

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

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial version
author Carl Byington <carl@five-ten-sg.com>
date Thu, 22 May 2014 10:41:19 -0700
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:0ce5cc452d02
1
2 package com.trilead.ssh2.channel;
3
4 import java.io.IOException;
5 import java.io.InputStream;
6 import java.io.OutputStream;
7 import java.net.Socket;
8
9 /**
10 * A StreamForwarder forwards data between two given streams.
11 * If two StreamForwarder threads are used (one for each direction)
12 * then one can be configured to shutdown the underlying channel/socket
13 * if both threads have finished forwarding (EOF).
14 *
15 * @author Christian Plattner, plattner@trilead.com
16 * @version $Id: StreamForwarder.java,v 1.1 2007/10/15 12:49:56 cplattne Exp $
17 */
18 public class StreamForwarder extends Thread {
19 final OutputStream os;
20 final InputStream is;
21 final byte[] buffer = new byte[Channel.CHANNEL_BUFFER_SIZE];
22 final Channel c;
23 final StreamForwarder sibling;
24 final Socket s;
25 final String mode;
26
27 StreamForwarder(Channel c, StreamForwarder sibling, Socket s, InputStream is, OutputStream os, String mode)
28 throws IOException {
29 this.is = is;
30 this.os = os;
31 this.mode = mode;
32 this.c = c;
33 this.sibling = sibling;
34 this.s = s;
35 }
36
37 public void run() {
38 try {
39 while (true) {
40 int len = is.read(buffer);
41
42 if (len <= 0)
43 break;
44
45 os.write(buffer, 0, len);
46 os.flush();
47 }
48 }
49 catch (IOException ignore) {
50 try {
51 c.cm.closeChannel(c, "Closed due to exception in StreamForwarder (" + mode + "): "
52 + ignore.getMessage(), true);
53 }
54 catch (IOException e) {
55 }
56 }
57 finally {
58 try {
59 os.close();
60 }
61 catch (IOException e1) {
62 }
63
64 try {
65 is.close();
66 }
67 catch (IOException e2) {
68 }
69
70 if (sibling != null) {
71 while (sibling.isAlive()) {
72 try {
73 sibling.join();
74 }
75 catch (InterruptedException e) {
76 }
77 }
78
79 try {
80 c.cm.closeChannel(c, "StreamForwarder (" + mode + ") is cleaning up the connection", true);
81 }
82 catch (IOException e3) {
83 }
84 }
85
86 if (s != null) {
87 try {
88 s.close();
89 }
90 catch (IOException e1) {
91 }
92 }
93 }
94 }
95 }