在Java中快速实现端口转发

发布时间：2020-12-15 00:21:52 所属栏目：Java 来源：网络整理

导读：我构建了一个打开ServerSocket的简单应用程序,在连接时,它将自身连接到远程计算机上的另一个服务器套接字.为了实现端口转发,我使用两个线程,一个从本地输入流读取并流到远程套接字输出流,反之亦然. 实现感觉有点无法实现,所以我问你是否知道更好的实现策略,

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

实现感觉有点无法实现,所以我问你是否知道更好的实现策略,或者甚至有一些代码可以以高效的方式实现.

PS：我知道我可以在Linux上使用IPTables,但这必须在Windows上运行.

PPS：如果你发布这个简单任务的实现,我将创建一个基准来测试所有给定的实现.对于许多小型(~100字节)包和稳定数据流,解决方案应该是快速的.

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

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,b+1);
    }
}

解决方法

几点意见：

>在循环开始时读取的一个字节对提高性能没有任何作用.事实上可能相反.
>不需要调用inputStream.available().您应该尝试读取“缓冲区大小”字符.对Socket流的读取将返回当前可用的字符数,但在缓冲区已满之前不会阻塞. (我在javadocs中找不到任何说明这一点的内容,但我确定情况确实如此.很多事情都会表现不佳……或者中断…如果读取被阻止,直到缓冲区已满.)
>正如@ user479257指出的那样,你应该通过使用java.nio和读写ByteBuffers来获得更好的吞吐量.这将减少JVM中发生的数据复制量.
>如果读取,写入或关闭操作引发异常,则您的方法将泄漏套接字流.您应该使用try …最终如下,以确保无论发生什么情况,流始终都是关闭的.

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,b+1);
        }
    } finally {
        try { inputStream.close();} catch (IOException ex) { /* ignore */ }
        try { outputStream.close();} catch (IOException ex) { /* ignore */ }
    }
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!