PcapWriteHandlerTest Class — netty Architecture

Source Code

handler/src/test/java/io/netty/handler/pcap/PcapWriteHandlerTest.java lines 72–1519

public class PcapWriteHandlerTest {

    @Test
    public void udpV4SharedOutputStreamTest() throws InterruptedException {
        udpV4(true, true);
    }

    @Test
    public void udpV4NonOutputStream() throws InterruptedException {
        udpV4(false, true);
    }

    @Test
    public void udpV4NoGlobalHeaderOutputStream() throws InterruptedException {
        udpV4(false, false);
    }

    private static void udpV4(boolean sharedOutputStream, boolean writeGlobalHeaders)
            throws InterruptedException {
        ByteBuf byteBuf = Unpooled.buffer();
        ByteBuf payload = Unpooled.wrappedBuffer("Meow".getBytes());

        try {
            InetSocketAddress serverAddr = new InetSocketAddress("127.0.0.1", 0);
            InetSocketAddress clientAddr = new InetSocketAddress("127.0.0.1", 0);

            EventLoopGroup eventLoopGroup = new MultiThreadIoEventLoopGroup(1, NioIoHandler.newFactory());

            // We'll bootstrap a UDP Server to avoid "Network Unreachable errors" when sending UDP Packet.
            Bootstrap server = new Bootstrap()
                    .group(eventLoopGroup)
                    .channel(NioDatagramChannel.class)
                    .handler(new SimpleChannelInboundHandler<DatagramPacket>() {
                        @Override
                        protected void channelRead0(ChannelHandlerContext ctx, DatagramPacket msg) {
                            // Discard
                        }
                    });

            ChannelFuture channelFutureServer = server.bind(serverAddr).sync();
            assertTrue(channelFutureServer.isSuccess());

            CloseDetectingByteBufOutputStream outputStream = new CloseDetectingByteBufOutputStream(byteBuf);

            // We'll bootstrap a UDP Client for sending UDP Packets to UDP Server.
            Bootstrap client = new Bootstrap()
                    .group(eventLoopGroup)
                    .channel(NioDatagramChannel.class)
                    .handler(PcapWriteHandler.builder()
                            .sharedOutputStream(sharedOutputStream)
                            .writePcapGlobalHeader(writeGlobalHeaders)
                            .build(outputStream));

            ChannelFuture channelFutureClient =
                    client.bind(clientAddr).sync();
            assertTrue(channelFutureClient.isSuccess());

            Channel clientChannel = channelFutureClient.channel();
            DatagramPacket datagram = new DatagramPacket(payload.copy(),
                    (InetSocketAddress) channelFutureServer.channel().localAddress());
            assertTrue(clientChannel.writeAndFlush(datagram).sync().isSuccess());
            assertTrue(eventLoopGroup.shutdownGracefully().sync().isSuccess());

            // if sharedOutputStream is true or writeGlobalHeaders is false, we don't verify the global headers.
            verifyUdpCapture(!sharedOutputStream && writeGlobalHeaders,
                    byteBuf, payload,
                    (InetSocketAddress) channelFutureServer.channel().localAddress(),
                    (InetSocketAddress) clientChannel.localAddress()
            );

            // If sharedOutputStream is true, we don't close the outputStream.
            // If sharedOutputStream is false, we close the outputStream.
            assertEquals(!sharedOutputStream, outputStream.closeCalled());
        } finally {
            byteBuf.release();
            payload.release();
        }
    }

    @Test
    public void embeddedUdp() {