Examples with Socks5ProxyHandler org.rx.net.socks.upstream.Socks5ProxyHandler used on opensource projects

Example 1 with Socks5ProxyHandler

use of org.rx.net.socks.upstream.Socks5ProxyHandler in project rxlib by RockyLOMO.

the class Socks5CommandRequestHandler method connect.

private void connect(Channel inbound, Socks5AddressType dstAddrType, RouteEventArgs e) {
    SocksProxyServer server = SocksContext.server(inbound);
    UnresolvedEndpoint realEp = SocksContext.realDestination(inbound);
    Sockets.bootstrap(inbound.eventLoop(), server.getConfig(), outbound -> {
        SocksContext.server(outbound, server);
        e.getValue().initChannel(outbound);
    }).connect(e.getValue().getDestination().socketAddress()).addListener((ChannelFutureListener) f -> {
        UnresolvedEndpoint dstEp = e.getValue().getDestination();
        if (!f.isSuccess()) {
            if (server.onReconnecting != null) {
                server.raiseEvent(server.onReconnecting, e);
                if (!e.isCancel() && e.isUpstreamChanged()) {
                    e.reset();
                    connect(inbound, dstAddrType, e);
                    return;
                }
            }
            ExceptionHandler.INSTANCE.log("socks5[{}] connect {}[{}] fail", server.getConfig().getListenPort(), dstEp, realEp, f.cause());
            inbound.writeAndFlush(new DefaultSocks5CommandResponse(Socks5CommandStatus.FAILURE, dstAddrType)).addListener(ChannelFutureListener.CLOSE);
            return;
        }
        Channel outbound = f.channel();
        StringBuilder aesMsg = new StringBuilder();
        Socks5ProxyHandler proxyHandler;
        SocksConfig config = server.getConfig();
        if (server.aesRouter(realEp) && (proxyHandler = outbound.pipeline().get(Socks5ProxyHandler.class)) != null) {
            proxyHandler.setHandshakeCallback(() -> {
                if (config.getTransportFlags().has(TransportFlags.BACKEND_COMPRESS)) {
                    ChannelHandler[] handlers = new AESCodec(config.getAesKey()).channelHandlers();
                    for (int i = handlers.length - 1; i > -1; i--) {
                        ChannelHandler handler = handlers[i];
                        outbound.pipeline().addAfter(TransportUtil.ZIP_DECODER, handler.getClass().getSimpleName(), handler);
                    }
                    aesMsg.append("[BACKEND_AES]");
                }
                relay(inbound, outbound, dstAddrType, dstEp, aesMsg);
            });
            return;
        }
        relay(inbound, outbound, dstAddrType, dstEp, aesMsg);
    });
}