java实现端口转发(代理)

https://blog.csdn.net/weixin_36557877/article/details/108580590

1)我们要实现的就是把客户的请求转发到别一服务器.
要实现这个功能就是实现一个中间代理,中间代理,我们使用serverSocket 服务器实现端口帧听,
ServerSocket serverSocket = new ServerSocket(listenPort);
也就是我说我们请求的实际地址为A服务器,但是事实上这个请求是被转发到B服务器去完成的,也就是A和B进行了socket通信,然后把返回的数据再次传回客户端.
在这里我们使用了线程池来运行客户端请求,原因在于如果我们不使用线程的话.我们在使用socket = serverSocket.accept();
后会只有处理完一个连接后才能再次调用accept,因此在此使用线程池来运行连接后的数据处理.
final ExecutorService tpe=Executors.newCachedThreadPool();

public static void main(String[] args) throws Exception {
        SimpleDateFormat sdf=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
        ServerSocket serverSocket = new ServerSocket(listenPort);
        final ExecutorService tpe=Executors.newCachedThreadPool();
        System.out.println("Proxy Server Start At "+sdf.format(new Date()));
        System.out.println("listening port:"+listenPort+"……");
        System.out.println();
        System.out.println();

        while (true) {
            Socket socket = null;
            try {
                socket = serverSocket.accept();
                socket.setKeepAlive(true);
                tpe.execute(new ProxyTask(socket));
                System.out.println("----1");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

运行了代理服务器后,我们关键的在于转发,怎么把客户的请求转发出去,这里我们去掉对头部的处理,我们只是简单的转发,其实转向很简单,只是在代理服务器开启一个新的连接服务器的socket
这里我们不去分析原请求的实现IP和端口,我们写成固定的地址,如:
socketOut = new Socket(“172.29.1.99”, Integer.parseInt(“80”));
这样我们只要把客户端得到的inputStream然后写到指定服务器端的outputStream就行了.然后再把服务端返回的inputStream写到客户端的outputStream中去就行了.
这里要注意的有两点.
1)我们在接收并转发和服务器返回并转发要用线程分开,不能在同一线程中,因为可能两个过程是同步进行的.
2)我们要实现代理,就必须要取得原有请求的实际IP地址和PORT号,因此我们分析请求的头部,要分析头部,我们必须要读取头部的字节来分析.要读取inputstream后再把inputstream转发时肯定会少了这部分数据,因为在头部分析时已读取出来了,因此为了正确的发送我们必须要把头部的数据重新加入inputsteam中去,然头在转发前选把头部数据写入到服务器的连接中去,这样才正常,除非我们不去处理头部.

public void run() {
        System.out.println("----2");
        StringBuilder builder=new StringBuilder();
        try {
            builder.append("\r\n").append("Request Time  ：" + sdf.format(new Date()));

            InputStream isIn = socketIn.getInputStream();
            OutputStream osIn = socketIn.getOutputStream();

            // 查找主机和端口
            socketOut = new Socket("172.29.1.99", Integer.parseInt("80"));
            socketOut.setKeepAlive(true);
            InputStream isOut = socketOut.getInputStream();
            OutputStream osOut = socketOut.getOutputStream();
            //新开一个线程将返回的数据转发给客户端,串行会出问题，尚没搞明白原因
            Thread ot = new DataSendThread(isOut, osIn);
            ot.start();
            //读取客户端请求过来的数据转发给服务器
            readForwardDate(isIn, osOut);
            //等待向客户端转发的线程结束
            ot.join();
        } catch (Exception e) {
            e.printStackTrace();
            if(!socketIn.isOutputShutdown()){
                //如果还可以返回错误状态的话，返回内部错误
                try {
                    socketIn.getOutputStream().write(SERVERERROR.getBytes());
                } catch (IOException e1) {}
            }
        } finally {
            try {
                if (socketIn != null) {
                    socketIn.close();
                }
            } catch (IOException e) {}
            if (socketOut != null) {
                try {
                    socketOut.close();
                } catch (IOException e) {}
            }
            //纪录上下行数据量和最后结束时间并打印
            builder.append("\r\n").append("Up    Bytes  ：" + totalUpload);
            builder.append("\r\n").append("Down  Bytes  ：" + totalDownload);
            builder.append("\r\n").append("Closed Time  ：" + sdf.format(new Date()));
            builder.append("\r\n");
            logRequestMsg(builder.toString());
        }    
    }

AI写代码
java
运行

————————————————

                        版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

原文链接：https://blog.csdn.net/weixin_36557877/article/details/108580590