用Java快速实现端口转发

Question

我已经构建了一个打开ServerSocket的简单应用程序，并在连接时将其自身连接到远程计算机上的另一个服务器套接字。为了实现端口转发，我使用了两个线程，一个线程从本地输入流读取并向远程套接字输出流读取流，反之亦然。

该实现感觉有点差劲，所以我问您是否知道更好的实现策略，或者是否有一些代码可以以高效的方式实现。

PS：我知道我可以在Linux上使用IPTables，但是必须在Windows上才能使用。

PPS：如果您发布此简单任务的实现，我将创建一个基准测试所有给定的实现。对于许多小型（〜100bytes）封装和稳定的数据流，该解决方案应该是快速的。

我当前的实现是这样（在每个方向的两个线程中的每个线程上执行）：

public static void route(InputStream inputStream, OutputStream outputStream) throws IOException {
    byte[] buffer = new byte[65536];
    while( true ) {
        // Read one byte to block
        int b = inputStream.read();
        if( b == - 1 ) {
            log.info("No data available anymore. Closing stream.");
            inputStream.close();
            outputStream.close();
            return;
        }
        buffer[0] = (byte)b;
        // Read remaining available bytes
        b = inputStream.read(buffer, 1, Math.min(inputStream.available(), 65535));
        if( b == - 1 ) {
            log.info("No data available anymore. Closing stream.");
            inputStream.close();
            outputStream.close();
            return;
        }
        outputStream.write(buffer, 0, b+1);
    }
}

Answer 1

一些观察：

• 在循环开始时读取一个字节不会提高性能。实际上可能相反。

• inputStream.available()不需要调用。您应该只尝试读取“缓冲区大小”字符。read套接字流上的A 将返回与当前可用字符一样多的字符，但是直到缓冲区已满时才会阻塞。（我无法在javadocs中找到任何说明这一点的方法，但是我敢肯定是这种情况。如果read阻塞直到缓冲区已满，很多事情都会表现不佳…或破坏…。）

• 正如@ user479257指出的那样，您应该通过使用java.nio并读写ByteBuffer获得更好的吞吐量。这将减少在JVM中发生的数据复制量。

• 如果读取，写入或关闭操作引发异常，则您的方法将泄漏Socket流。您应该使用try ... finally以下方法来确保无论发生什么情况始终关闭流。

public static void route(InputStream inputStream, OutputStream outputStream) 
throws IOException {
    byte[] buffer = new byte[65536];
    try {
        while( true ) {
            ...
            b = inputStream.read(...);
            if( b == - 1 ) {
                log.info("No data available anymore. Closing stream.");
                return;
            }
            outputStream.write(buffer, 0, b+1);
        }
    } finally {
        try { inputStream.close();} catch (IOException ex) { /* ignore */ }
        try { outputStream.close();} catch (IOException ex) { /* ignore */ }
    }
}