QuicPortReuseTest Class — netty Architecture

Source Code

codec-native-quic/src/test/java/io/netty/handler/codec/quic/QuicPortReuseTest.java lines 38–138

public class QuicPortReuseTest extends AbstractQuicTest {

    @ParameterizedTest
    @MethodSource("newSslTaskExecutors")
    @Timeout(100)
    public void testReusePort(Executor executor) throws Throwable {
        int numBytes = 1000;
        final AtomicInteger byteCounter = new AtomicInteger();

        ChannelOption<Boolean> reusePort = QuicTestUtils.soReusePortOption();
        int numBinds = reusePort == null ? 1 : 4;
        int numConnects = 16;

        List<Channel> serverChannels = new ArrayList<>();
        Bootstrap serverBootstrap = QuicTestUtils.newServerBootstrap()
                .handler(new QuicCodecDispatcher() {
                    @Override
                    protected void initChannel(Channel channel, int localConnectionIdLength,
                                               QuicConnectionIdGenerator idGenerator) {
                        ChannelHandler codec = QuicTestUtils.newQuicServerBuilder(executor)
                                .localConnectionIdLength(localConnectionIdLength)
                                .connectionIdAddressGenerator(idGenerator)
                                .tokenHandler(InsecureQuicTokenHandler.INSTANCE)
                                .handler(QuicTestUtils.NOOP_HANDLER)
                                .streamHandler(new ChannelInboundHandlerAdapter() {

                                    @Override
                                    public boolean isSharable() {
                                        return true;
                                    }

                                    @Override
                                    public void channelRead(ChannelHandlerContext ctx, Object msg) {
                                        byteCounter.addAndGet(((ByteBuf) msg).readableBytes());
                                        ReferenceCountUtil.release(msg);
                                    }
                                }).build();
                        channel.pipeline().addLast(codec);
                    }
                });

        if (reusePort != null) {
            serverBootstrap.option(reusePort, true);
        }

        SocketAddress bindAddress = null;
        for (int i = 0; i < numBinds; i++) {
            Channel bindChannel;
            if (bindAddress == null) {
                bindChannel = serverBootstrap.bind().sync().channel();
            } else {
                bindChannel = serverBootstrap.bind(bindAddress).sync().channel();
            }
            serverChannels.add(bindChannel);
            if (bindAddress == null) {
                bindAddress = bindChannel.localAddress();
            }
        }

        Channel channel = QuicTestUtils.newClient(executor);
        QuicChannelBootstrap cb = QuicTestUtils.newQuicChannelBootstrap(channel)
                .handler(QuicTestUtils.NOOP_HANDLER)
                .streamHandler(QuicTestUtils.NOOP_HANDLER)
                .remoteAddress(bindAddress);
        List<QuicChannel> channels = new ArrayList<>();

        try {
            for (int i = 0; i < numConnects; i++) {
                channels.add(cb
                        .connect()
                        .get());
            }

            for (QuicChannel quicChannel: channels) {
                quicChannel.createStream(QuicStreamType.BIDIRECTIONAL,
                        new ChannelInboundHandlerAdapter() {
                            @Override
                            public void channelActive(ChannelHandlerContext ctx) {
                                ctx.writeAndFlush(Unpooled.directBuffer().writeZero(numBytes))
                                        .addListener(ChannelFutureListener.CLOSE);
                            }