testNestedTLV() — netty Function Reference

Source Code

codec-haproxy/src/test/java/io/netty/handler/codec/haproxy/HAProxyMessageDecoderTest.java lines 1199–1258

    @Test
    public void testNestedTLV() throws Exception {
        ByteArrayOutputStream headerWriter = new ByteArrayOutputStream();
        //src_ip = "AAAA", dst_ip = "BBBB", src_port = "CC", dst_port = "DD"
        headerWriter.write(new byte[] {'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'C', 'C', 'D', 'D'});
        //write TLVs
        int countOfTLVs = 8100;
        ByteBuffer tlvLengthBuf = ByteBuffer.allocate(2);
        tlvLengthBuf.order(ByteOrder.BIG_ENDIAN);
        short totalLength = (short) (countOfTLVs * (1 + 2 + 1 + 4));
        for (int i = 0; i < countOfTLVs; i++) {
            //write PP2_TYPE_SSL TLV
            headerWriter.write(0x20); //PP2_TYPE_SSL
            //notice that the TLV length cannot be bigger than 0xffff
            totalLength -= 1 + 2; //exclude type and length themselves
            tlvLengthBuf.clear();
            tlvLengthBuf.putShort(totalLength);
            //add to the header
            headerWriter.write(tlvLengthBuf.array());
            //write client field
            headerWriter.write(1);
            //write verify field
            headerWriter.write(new byte[] {'V', 'V', 'V', 'V'});
            //subtract the client and verify fields
            totalLength -= 1 + 4;
        }
        byte[] header = headerWriter.toByteArray();
        ByteBuffer numsWrite = ByteBuffer.allocate(2);
        numsWrite.order(ByteOrder.BIG_ENDIAN);
        numsWrite.putShort((short) header.length);

        final  ByteBuf data = Unpooled.buffer();
        data.writeBytes(new byte[] {
                (byte) 0x0D,
                (byte) 0x0A,
                (byte) 0x0D,
                (byte) 0x0A,
                (byte) 0x00,
                (byte) 0x0D,
                (byte) 0x0A,
                (byte) 0x51,
                (byte) 0x55,
                (byte) 0x49,
                (byte) 0x54,
                (byte) 0x0A
        });
        //verCmd = 32
        byte versionCmd = 0x20 | 1; //V2 | ProxyCmd
        data.writeByte(versionCmd);
        data.writeByte(17); //TPAF_TCP4_BYTE
        data.writeBytes(numsWrite.array());
        data.writeBytes(header);

        assertThrows(HAProxyProtocolException.class, new Executable() {
            @Override
            public void execute() {
                ch.writeInbound(data);
            }
        });
    }