ServerSecureDecoder.java

package com.firefly.server.http2;

import com.firefly.codec.http2.stream.HTTPConnection;
import com.firefly.net.DecoderChain;
import com.firefly.net.Session;
import com.firefly.net.tcp.ssl.SSLSession;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.nio.ByteBuffer;

public class ServerSecureDecoder extends DecoderChain {

    private static Logger log = LoggerFactory.getLogger("firefly-system");

    public ServerSecureDecoder(DecoderChain next) {
        super(next);
    }

    @Override
    public void decode(ByteBuffer buf, Session session) throws Throwable {
        if (session.getAttachment() instanceof HTTPConnection) {
            HTTPConnection connection = (HTTPConnection) session.getAttachment();

            ByteBuffer plaintext;
            switch (connection.getHttpVersion()) {
                case HTTP_2:
                    plaintext = ((HTTP2ServerConnection) connection).getSSLSession().read(buf);
                    break;
                case HTTP_1_1:
                    plaintext = ((HTTP1ServerConnection) connection).getSSLSession().read(buf);
                    break;
                default:
                    throw new IllegalStateException("server does not support the http version " + connection.getHttpVersion());
            }

            if (plaintext != null && next != null) {
                next.decode(plaintext, session);
            }
        } else if (session.getAttachment() instanceof SSLSession) {
            SSLSession sslSession = (SSLSession) session.getAttachment();
            ByteBuffer plaintext = sslSession.read(buf);

            if (plaintext != null && plaintext.hasRemaining()) {
                log.debug("server session {} handshake finished and received cleartext size {}", session.getSessionId(), plaintext.remaining());
                if (session.getAttachment() instanceof HTTPConnection) {
                    if (next != null) {
                        next.decode(plaintext, session);
                    }
                } else {
                    throw new IllegalStateException("the server http connection has not been created");
                }
            } else {
                log.debug("server ssl session {} is shaking hands", session.getSessionId());
            }
        }
    }

}