io.undertow.protocols.alpn.ALPNManager Maven / Gradle / Ivy
/*
* JBoss, Home of Professional Open Source.
* Copyright 2014 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.undertow.protocols.alpn;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.ServiceLoader;
import java.util.function.Function;
import javax.net.ssl.SSLEngine;
/**
* @author Stuart Douglas
*/
public class ALPNManager {
private final ALPNProvider[] alpnProviders;
private final ALPNEngineManager[] alpnEngineManagers;
public static final ALPNManager INSTANCE = new ALPNManager(ALPNManager.class.getClassLoader());
public ALPNManager(ClassLoader classLoader) {
ServiceLoader loader = ServiceLoader.load(ALPNProvider.class, classLoader);
List provider = new ArrayList<>();
for (ALPNProvider prov : loader) {
provider.add(prov);
}
Collections.sort(provider, new Comparator() {
@Override
public int compare(ALPNProvider o1, ALPNProvider o2) {
return Integer.compare(o2.getPriority(), o1.getPriority()); //highest first
}
});
this.alpnProviders = provider.toArray(new ALPNProvider[0]);
ServiceLoader managerLoader = ServiceLoader.load(ALPNEngineManager.class, classLoader);
List managers = new ArrayList<>();
for (ALPNEngineManager manager : managerLoader) {
managers.add(manager);
}
Collections.sort(managers, new Comparator() {
@Override
public int compare(ALPNEngineManager o1, ALPNEngineManager o2) {
return Integer.compare(o2.getPriority(), o1.getPriority()); //highest first
}
});
this.alpnEngineManagers = managers.toArray(new ALPNEngineManager[0]);
}
public ALPNProvider getProvider(SSLEngine engine) {
for (ALPNProvider provider : alpnProviders) {
if (provider.isEnabled(engine)) {
return provider;
}
}
return null;
}
public void registerEngineCallback(SSLEngine original, Function selectionFunction) {
for(ALPNEngineManager manager : alpnEngineManagers) {
if(manager.registerEngine(original, selectionFunction)) {
return;
}
}
}
}