使用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());
		}	
	}