Search in sources :

Example 1 with HeaderGenerator

use of org.eclipse.jetty.http2.generator.HeaderGenerator in project jetty.project by eclipse.

the class DataGenerateParseTest method testGenerateParse.

private List<DataFrame> testGenerateParse(ByteBuffer data) {
    DataGenerator generator = new DataGenerator(new HeaderGenerator());
    final List<DataFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onData(DataFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    // Iterate a few times to be sure generator and parser are properly reset.
    for (int i = 0; i < 2; ++i) {
        ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
        ByteBuffer slice = data.slice();
        int generated = 0;
        while (true) {
            generated += generator.generateData(lease, 13, slice, true, slice.remaining());
            generated -= Frame.HEADER_LENGTH;
            if (generated == data.remaining())
                break;
        }
        frames.clear();
        for (ByteBuffer buffer : lease.getByteBuffers()) {
            parser.parse(buffer);
        }
    }
    return frames;
}
Also used : MappedByteBufferPool(org.eclipse.jetty.io.MappedByteBufferPool) ByteBufferPool(org.eclipse.jetty.io.ByteBufferPool) ArrayList(java.util.ArrayList) ByteBuffer(java.nio.ByteBuffer) Parser(org.eclipse.jetty.http2.parser.Parser) DataGenerator(org.eclipse.jetty.http2.generator.DataGenerator) HeaderGenerator(org.eclipse.jetty.http2.generator.HeaderGenerator)

Example 2 with HeaderGenerator

use of org.eclipse.jetty.http2.generator.HeaderGenerator in project jetty.project by eclipse.

the class DataGenerateParseTest method testGenerateParseOneByteAtATime.

@Test
public void testGenerateParseOneByteAtATime() {
    DataGenerator generator = new DataGenerator(new HeaderGenerator());
    final List<DataFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onData(DataFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    // Iterate a few times to be sure generator and parser are properly reset.
    for (int i = 0; i < 2; ++i) {
        ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
        ByteBuffer data = ByteBuffer.wrap(largeContent);
        ByteBuffer slice = data.slice();
        int generated = 0;
        while (true) {
            generated += generator.generateData(lease, 13, slice, true, slice.remaining());
            generated -= Frame.HEADER_LENGTH;
            if (generated == data.remaining())
                break;
        }
        frames.clear();
        for (ByteBuffer buffer : lease.getByteBuffers()) {
            while (buffer.hasRemaining()) {
                parser.parse(ByteBuffer.wrap(new byte[] { buffer.get() }));
            }
        }
        Assert.assertEquals(largeContent.length, frames.size());
    }
}
Also used : MappedByteBufferPool(org.eclipse.jetty.io.MappedByteBufferPool) ByteBufferPool(org.eclipse.jetty.io.ByteBufferPool) ArrayList(java.util.ArrayList) ByteBuffer(java.nio.ByteBuffer) Parser(org.eclipse.jetty.http2.parser.Parser) DataGenerator(org.eclipse.jetty.http2.generator.DataGenerator) HeaderGenerator(org.eclipse.jetty.http2.generator.HeaderGenerator) Test(org.junit.Test)

Example 3 with HeaderGenerator

use of org.eclipse.jetty.http2.generator.HeaderGenerator in project jetty.project by eclipse.

the class PingGenerateParseTest method testPayloadAsLong.

@Test
public void testPayloadAsLong() throws Exception {
    PingGenerator generator = new PingGenerator(new HeaderGenerator());
    final List<PingFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onPing(PingFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
    PingFrame ping = new PingFrame(System.nanoTime(), true);
    generator.generate(lease, ping);
    for (ByteBuffer buffer : lease.getByteBuffers()) {
        while (buffer.hasRemaining()) {
            parser.parse(buffer);
        }
    }
    Assert.assertEquals(1, frames.size());
    PingFrame pong = frames.get(0);
    Assert.assertEquals(ping.getPayloadAsLong(), pong.getPayloadAsLong());
    Assert.assertTrue(pong.isReply());
}
Also used : MappedByteBufferPool(org.eclipse.jetty.io.MappedByteBufferPool) ByteBufferPool(org.eclipse.jetty.io.ByteBufferPool) ArrayList(java.util.ArrayList) PingGenerator(org.eclipse.jetty.http2.generator.PingGenerator) ByteBuffer(java.nio.ByteBuffer) Parser(org.eclipse.jetty.http2.parser.Parser) HeaderGenerator(org.eclipse.jetty.http2.generator.HeaderGenerator) Test(org.junit.Test)

Example 4 with HeaderGenerator

use of org.eclipse.jetty.http2.generator.HeaderGenerator in project jetty.project by eclipse.

the class PingGenerateParseTest method testGenerateParseOneByteAtATime.

@Test
public void testGenerateParseOneByteAtATime() throws Exception {
    PingGenerator generator = new PingGenerator(new HeaderGenerator());
    final List<PingFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onPing(PingFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    byte[] payload = new byte[8];
    new Random().nextBytes(payload);
    // Iterate a few times to be sure generator and parser are properly reset.
    for (int i = 0; i < 2; ++i) {
        ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
        generator.generatePing(lease, payload, true);
        frames.clear();
        for (ByteBuffer buffer : lease.getByteBuffers()) {
            while (buffer.hasRemaining()) {
                parser.parse(ByteBuffer.wrap(new byte[] { buffer.get() }));
            }
        }
        Assert.assertEquals(1, frames.size());
        PingFrame frame = frames.get(0);
        Assert.assertArrayEquals(payload, frame.getPayload());
        Assert.assertTrue(frame.isReply());
    }
}
Also used : MappedByteBufferPool(org.eclipse.jetty.io.MappedByteBufferPool) ByteBufferPool(org.eclipse.jetty.io.ByteBufferPool) ArrayList(java.util.ArrayList) PingGenerator(org.eclipse.jetty.http2.generator.PingGenerator) ByteBuffer(java.nio.ByteBuffer) Parser(org.eclipse.jetty.http2.parser.Parser) Random(java.util.Random) HeaderGenerator(org.eclipse.jetty.http2.generator.HeaderGenerator) Test(org.junit.Test)

Example 5 with HeaderGenerator

use of org.eclipse.jetty.http2.generator.HeaderGenerator in project jetty.project by eclipse.

the class PriorityGenerateParseTest method testGenerateParseOneByteAtATime.

@Test
public void testGenerateParseOneByteAtATime() throws Exception {
    PriorityGenerator generator = new PriorityGenerator(new HeaderGenerator());
    final List<PriorityFrame> frames = new ArrayList<>();
    Parser parser = new Parser(byteBufferPool, new Parser.Listener.Adapter() {

        @Override
        public void onPriority(PriorityFrame frame) {
            frames.add(frame);
        }
    }, 4096, 8192);
    int streamId = 13;
    int parentStreamId = 17;
    int weight = 3;
    boolean exclusive = true;
    // Iterate a few times to be sure generator and parser are properly reset.
    for (int i = 0; i < 2; ++i) {
        ByteBufferPool.Lease lease = new ByteBufferPool.Lease(byteBufferPool);
        generator.generatePriority(lease, streamId, parentStreamId, weight, exclusive);
        frames.clear();
        for (ByteBuffer buffer : lease.getByteBuffers()) {
            while (buffer.hasRemaining()) {
                parser.parse(ByteBuffer.wrap(new byte[] { buffer.get() }));
            }
        }
        Assert.assertEquals(1, frames.size());
        PriorityFrame frame = frames.get(0);
        Assert.assertEquals(streamId, frame.getStreamId());
        Assert.assertEquals(parentStreamId, frame.getParentStreamId());
        Assert.assertEquals(weight, frame.getWeight());
        Assert.assertEquals(exclusive, frame.isExclusive());
    }
}
Also used : MappedByteBufferPool(org.eclipse.jetty.io.MappedByteBufferPool) ByteBufferPool(org.eclipse.jetty.io.ByteBufferPool) PriorityGenerator(org.eclipse.jetty.http2.generator.PriorityGenerator) ArrayList(java.util.ArrayList) ByteBuffer(java.nio.ByteBuffer) Parser(org.eclipse.jetty.http2.parser.Parser) HeaderGenerator(org.eclipse.jetty.http2.generator.HeaderGenerator) Test(org.junit.Test)

Aggregations

ByteBuffer (java.nio.ByteBuffer)21 HeaderGenerator (org.eclipse.jetty.http2.generator.HeaderGenerator)21 Parser (org.eclipse.jetty.http2.parser.Parser)21 ByteBufferPool (org.eclipse.jetty.io.ByteBufferPool)21 MappedByteBufferPool (org.eclipse.jetty.io.MappedByteBufferPool)21 ArrayList (java.util.ArrayList)20 Test (org.junit.Test)19 HostPortHttpField (org.eclipse.jetty.http.HostPortHttpField)5 HttpField (org.eclipse.jetty.http.HttpField)5 HttpFields (org.eclipse.jetty.http.HttpFields)5 MetaData (org.eclipse.jetty.http.MetaData)5 HpackEncoder (org.eclipse.jetty.http2.hpack.HpackEncoder)5 Random (java.util.Random)3 HeadersGenerator (org.eclipse.jetty.http2.generator.HeadersGenerator)3 PingGenerator (org.eclipse.jetty.http2.generator.PingGenerator)3 SettingsGenerator (org.eclipse.jetty.http2.generator.SettingsGenerator)3 HashMap (java.util.HashMap)2 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)2 DataGenerator (org.eclipse.jetty.http2.generator.DataGenerator)2 GoAwayGenerator (org.eclipse.jetty.http2.generator.GoAwayGenerator)2