使用java实现端口转发(代理)

        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());

}

}

你可能感兴趣的:(使用java实现端口转发(代理))