AbstractSslEngineThroughputBenchmark Class — netty Architecture

Source Code

microbench/src/main/java/io/netty/microbench/handler/ssl/AbstractSslEngineThroughputBenchmark.java lines 31–89

public abstract class AbstractSslEngineThroughputBenchmark extends AbstractSslEngineBenchmark {

    @Param({ "64", "128", "512", "1024", "4096" })
    public int messageSize;

    protected CleanableDirectBuffer cleanableWrapSrcBuffer;
    private CleanableDirectBuffer cleanableWrapDstBuffer;
    protected ByteBuffer wrapSrcBuffer;
    private ByteBuffer wrapDstBuffer;

    @Setup(Level.Iteration)
    public final void setup() throws Exception {
        ByteBufAllocator allocator = new PooledByteBufAllocator(true);
        initEngines(allocator);
        initHandshakeBuffers();

        cleanableWrapDstBuffer = allocateBuffer(clientEngine.getSession().getPacketBufferSize() << 2);
        cleanableWrapSrcBuffer = allocateBuffer(messageSize);
        wrapDstBuffer = cleanableWrapDstBuffer.buffer();
        wrapSrcBuffer = cleanableWrapDstBuffer.buffer();

        byte[] bytes = new byte[messageSize];
        ThreadLocalRandom.current().nextBytes(bytes);
        wrapSrcBuffer.put(bytes);
        wrapSrcBuffer.flip();

        // Complete the initial TLS handshake.
        if (!doHandshake()) {
            throw new IllegalStateException();
        }
        doSetup();
    }

    protected void doSetup() throws Exception { }

    @TearDown(Level.Iteration)
    public final void tearDown() throws Exception {
        destroyEngines();
        destroyHandshakeBuffers();
        cleanableWrapSrcBuffer.clean();
        cleanableWrapDstBuffer.clean();
        doTearDown();
    }

    protected void doTearDown() throws Exception { }

    protected final ByteBuffer doWrap(int numWraps) throws SSLException {
        wrapDstBuffer.clear();

        for (int i = 0; i < numWraps; ++i) {
            wrapSrcBuffer.position(0).limit(messageSize);

            SSLEngineResult wrapResult = clientEngine.wrap(wrapSrcBuffer, wrapDstBuffer);

            assert checkSslEngineResult(wrapResult, wrapSrcBuffer, wrapDstBuffer);
        }
        return wrapDstBuffer;
    }
}