com.firefly.net.tcp.secure.jdk.JettyALPNSelector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of firefly-nettool Show documentation
There is a newer version: 5.0.0-dev6
package com.firefly.net.tcp.secure.jdk;

import com.firefly.net.ApplicationProtocolSelector;
import com.firefly.utils.CollectionUtils;
import org.eclipse.jetty.alpn.ALPN;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.net.ssl.SSLEngine;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * @author Pengtao Qiu
 */
public class JettyALPNSelector implements ApplicationProtocolSelector {

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

    private final List supportedProtocols;
    private volatile String applicationProtocol;

    public JettyALPNSelector(SSLEngine sslEngine, List supportedProtocols) {
        if (CollectionUtils.isEmpty(supportedProtocols)) {
            this.supportedProtocols = Collections.unmodifiableList(Arrays.asList("h2", "http/1.1"));
        } else {
            this.supportedProtocols = supportedProtocols;
        }
        if (sslEngine.getUseClientMode()) {
            ALPN.put(sslEngine, new ALPN.ClientProvider() {

                @Override
                public List protocols() {
                    return supportedProtocols;
                }

                @Override
                public void unsupported() {
                    ALPN.remove(sslEngine);
                }

                @Override
                public void selected(String protocol) {
                    ALPN.remove(sslEngine);
                    log.debug("ALPN remote server selected protocol -> {}", protocol);
                    applicationProtocol = protocol;
                }
            });
        } else {
            ALPN.put(sslEngine, new ALPN.ServerProvider() {
                @Override
                public void unsupported() {
                    ALPN.remove(sslEngine);
                }

                @Override
                public String select(List protocols) {
                    log.debug("ALPN remote client supported protocols -> {}", protocols);
                    ALPN.remove(sslEngine);
                    if (!CollectionUtils.isEmpty(protocols)) {
                        for (String p : supportedProtocols) {
                            if (protocols.contains(p)) {
                                applicationProtocol = p;
                                log.debug("ALPN local server selected protocol -> {}", p);
                                return applicationProtocol;
                            }
                        }
                    }
                    return null;
                }
            });
        }
    }

    @Override
    public String getApplicationProtocol() {
        return applicationProtocol;
    }

    @Override
    public List getSupportedApplicationProtocols() {
        return supportedProtocols;
    }
}