Netty实现TCP代理

发布于:2024-05-07 ⋅ 阅读:(22) ⋅ 点赞:(0)

netty实现TCP代理通信

最近基于netty写了一个通信项目,发现netty还是很强的,就是API用的比较繁琐,当然熟练了之后就好了,实现了一个TCP代理的功能,其实原理比较简单,基本步骤:

  1. netty监听服务端口
  2. 接受来自客户端的连接
  3. 创建客户端连接代理服务器
  4. 将连接、断开、接受消息、发送消息事件绑定
  5. 发送Proxy Protocol消息,告诉目标服务器代理的信息**(解析源IP)**
  6. 目标服务器解析Proxy Protocol消息**(解析源IP)**

当然了这是最简单的实现了,实际情况中还需要考虑异常情况,保证连接正常释放

还有一个关键点就是如果业务内绑定了对方的ip,增加了一层代理会使连接ip发送变化,变成代理服务器的ip,如何解决呢

这里就必须提到一个关键的协议,haproxy的Proxy Protocol代理协议,

这个代理协议的优点是:

  • 它与协议无关(可以与任何7层协议一起使用,即使在加密的情况也可用)

  • 它不需要任何基础架构更改

  • 可以穿透NAT防火墙

  • 它是可扩展的

    Proxy Protocol协议分为俩个版本,V1和V2,具体区别我在这里不详细说了

    具体看官方文档haproxy.org/download/1.8/doc/proxy-protocol.txt

netty代码



import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.haproxy.*;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import lombok.extern.slf4j.Slf4j;
import io.netty.util.ReferenceCountUtil;

import java.net.InetSocketAddress;

@Slf4j
public class TcpForwardServerHandler extends ChannelInboundHandlerAdapter {
   

    protected static final int WAIT_TIMEOUT_MILLIS = 5000;

    private Channel remoteChannel;

    private String proxyHost;

    private int proxyPort;

    private static Bootstrap bootstrap = new Bootstrap();
    private static EventLoopGroup group = new NioEventLoopGroup();

    public TcpForwardServerHandler(String proxyHost, int proxyPort) {
   
        this.proxyHost = proxyHost;
        this.proxyPort = proxyPort;
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) {
   
        In