/*
* 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 org.jboss.modules;
import java.net.URLStreamHandler;
import java.net.URLStreamHandlerFactory;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.ArrayList;
import java.util.List;
import java.util.ServiceLoader;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArrayList;
/**
* The root URL stream handler factory for the module system.
*
* @author <a href="mailto:david.lloyd@redhat.com">David M. Lloyd</a>
*/
final class ModularURLStreamHandlerFactory implements URLStreamHandlerFactory {
private static final PrivilegedAction<String> URL_MODULES_LIST_ACTION = new PropertyReadAction("jboss.protocol.handler.modules");
private static final List<Module> modules;
private static final ThreadLocal<Set<String>> reentered = new ThreadLocal<Set<String>>() {
protected Set<String> initialValue() {
return new FastCopyHashSet<String>();
}
};
static {
CopyOnWriteArrayList<Module> list = new CopyOnWriteArrayList<Module>();
final SecurityManager sm = System.getSecurityManager();
final String urlModulesList;
if (sm != null) {
urlModulesList = AccessController.doPrivileged(URL_MODULES_LIST_ACTION);
} else {
urlModulesList = URL_MODULES_LIST_ACTION.run();
}
if (urlModulesList != null) {
final List<Module> moduleList = new ArrayList<Module>();
int f = 0;
int i;
do {
i = urlModulesList.indexOf('|', f);
final String moduleId = (i == -1 ? urlModulesList.substring(f) : urlModulesList.substring(f, i)).trim();
if (moduleId.length() > 0) {
try {
Module module = Module.getBootModuleLoader().loadModule(moduleId);
moduleList.add(module);
} catch (RuntimeException | ModuleLoadException e) {
// skip it
}
}
f = i + 1;
} while (i != -1);
list.addAll(moduleList);
}
modules = list;
}
static final ModularURLStreamHandlerFactory INSTANCE = new ModularURLStreamHandlerFactory();
static void addHandlerModule(Module module) {
modules.add(module);
}
private ModularURLStreamHandlerFactory() {
}
private URLStreamHandler locateHandler(final String protocol) {
for (Module module : modules) {
ServiceLoader<URLStreamHandlerFactory> loader = module.loadService(URLStreamHandlerFactory.class);
for (URLStreamHandlerFactory factory : loader) {
try {
final URLStreamHandler handler = factory.createURLStreamHandler(protocol);
if (handler != null) {
return handler;
}
} catch (RuntimeException e) {
// ignored
}
}
}
return null;
}
public URLStreamHandler createURLStreamHandler(final String protocol) {
final Set<String> set = reentered.get();
if (set.add(protocol)) {
try {
if (System.getSecurityManager() == null) {
return locateHandler(protocol);
}
return AccessController.doPrivileged(new PrivilegedAction<URLStreamHandler>() {
public URLStreamHandler run() {
return locateHandler(protocol);
}
});
} finally {
set.remove(protocol);
}
}
return null;
}
}