84 lines
2.2 KiB
Java
84 lines
2.2 KiB
Java
package org.kne.cloud.network;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.net.*;
|
|
import java.nio.channels.SocketChannel;
|
|
|
|
import org.kne.cloud.network.klalb.KLALBVirtualSocket;
|
|
import org.kne.cloud.network.klalb.KLALBVirtualSocketChannel;
|
|
import org.kne.io.Task;
|
|
public class SocketChannelBridge extends Task{
|
|
protected SocketChannel a;
|
|
protected SocketChannel b;
|
|
protected StreamChannelBridge bridgeAB;
|
|
protected StreamChannelBridge bridgeBA;
|
|
public StreamChannelBridge getBridgeAB() {
|
|
return bridgeAB;
|
|
}
|
|
public StreamChannelBridge getBridgeBA() {
|
|
return bridgeBA;
|
|
}
|
|
public SocketChannelBridge(SocketChannel a, SocketChannel b) throws IOException {
|
|
super();
|
|
this.a = a;
|
|
this.b = b;
|
|
createStreamChannelBridge();
|
|
}
|
|
protected void createStreamChannelBridge() throws IOException {
|
|
bridgeAB = new StreamChannelBridge(a, b);
|
|
bridgeBA = new StreamChannelBridge(b, a);
|
|
}
|
|
@Override
|
|
protected void runTask() {
|
|
try {
|
|
if(a instanceof KLALBVirtualSocketChannel) {
|
|
((KLALBVirtualSocketChannel) a).associateSocketChannel(b);
|
|
}else if(b instanceof KLALBVirtualSocketChannel){
|
|
((KLALBVirtualSocketChannel) b).associateSocketChannel(a);
|
|
}else {
|
|
bridgeAB.runAtNewThread("SocketBridge A->B thread");
|
|
bridgeBA.runAtNewThread("SocketBridge B->A thread");
|
|
bridgeAB.waitfortask();
|
|
bridgeBA.waitfortask();
|
|
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}finally {
|
|
//new Exception().printStackTrace();
|
|
try {
|
|
a.close();
|
|
} catch (IOException e) {
|
|
// TODO 自动生成的 catch 块
|
|
e.printStackTrace();
|
|
}
|
|
try {
|
|
b.close();
|
|
} catch (IOException e) {
|
|
// TODO 自动生成的 catch 块
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
public SocketChannel getA() {
|
|
return a;
|
|
}
|
|
public SocketChannel getB() {
|
|
return b;
|
|
}
|
|
/*protected OutputStream getBOUT() throws IOException {
|
|
return b.getOutputStream();
|
|
}
|
|
protected InputStream getBIN() throws IOException {
|
|
return b.getInputStream();
|
|
}
|
|
protected OutputStream getAOUT() throws IOException {
|
|
return a.getOutputStream();
|
|
}
|
|
protected InputStream getAIN() throws IOException {
|
|
return a.getInputStream();
|
|
}*/
|
|
|
|
}
|