writePcapGreaterThan4Gb() — netty Function Reference

Source Code

handler/src/test/java/io/netty/handler/pcap/PcapWriteHandlerTest.java lines 882–978

    @Test
    public void writePcapGreaterThan4Gb() {
        InetSocketAddress serverAddr = new InetSocketAddress("1.1.1.1", 1234);
        InetSocketAddress clientAddr = new InetSocketAddress("2.2.2.2", 3456);

        class RecordingOutputStream extends OutputStream {
            private long bytesWritten;
            private ByteArrayOutputStream out;
            @Override
            public void write(int b) {
                bytesWritten++;
                if (out != null) {
                    out.write(b);
                }
            }

            @Override
            public void write(byte[] b, int off, int len) {
                bytesWritten += len;
                if (out != null) {
                    out.write(b, off, len);
                }
            }

            void record() {
                out = new ByteArrayOutputStream();
            }

            void reset() {
                out = null;
            }

            byte[] recordedBytes() {
                return out.toByteArray();
            }

            long bytesWritten() {
                return bytesWritten;
            }
        }

        RecordingOutputStream outputStream = new RecordingOutputStream();
        EmbeddedChannel embeddedChannel = new EmbeddedChannel(
                new DiscardingWritesAndFlushesHandler(), //Discard writes/flushes
                PcapWriteHandler.builder()
                                .forceTcpChannel(serverAddr, clientAddr, true)
                                .build(outputStream),
                new DiscardingReadsHandler() //Discard reads
        );

        int chunkSize = 0xFFFF - 40; // 40 bytes for header data
        String payloadString = new String(new char[chunkSize]).replace('\0', 'X');
        final ByteBuf payload = Unpooled.wrappedBuffer(payloadString.getBytes());

        try {
            long fourGB = 0xFFFFFFFFL;

            // Let's send 4 GiB inbound, ...
            long msgCount = (fourGB / chunkSize) + 2;
            for (int i = 0; i < msgCount; i++) {
                //Only store the last data/ack
                if (i == msgCount - 1) {
                    outputStream.record();
                }
                embeddedChannel.writeInbound(payload.retainedDuplicate());
            }

            //Validate the wrapped number
            long sequenceNumber = ((msgCount - 1) * chunkSize) - fourGB;
            long ackNumber = 1;
            ByteBuf buf = Unpooled.wrappedBuffer(outputStream.recordedBytes());
            verifyTcpCapture(buf, payload, serverAddr, clientAddr, sequenceNumber, ackNumber);
            buf.release();
            outputStream.reset();

            // ... and 4 GiB outbound.
            for (int i = 0; i < msgCount; i++) {
                //Only store the last data/ack
                if (i == msgCount - 1) {
                    outputStream.record();
                }