/*
* Copyright (c) 2000, 2021, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package javax.imageio.spi;
import java.io.File;
import java.security.AccessControlContext;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Set;
import java.util.ServiceLoader;
/**
* A registry for service provider instances for Image I/O service types.
*
*
Service providers are stored in one or more categories,
* each of which is defined by a class or interface (described by a
* {@code Class} object) that all of its members must implement.
*
*
The set of categories supported by this class is limited
* to the following standard Image I/O service types:
*
*
*
{@link ImageInputStreamSpi}
*
{@link ImageOutputStreamSpi}
*
{@link ImageReaderSpi}
*
{@link ImageTranscoderSpi}
*
{@link ImageWriterSpi}
*
*
*
An attempt to load a provider that is not a subtype of one of the
* above types will result in {@code IllegalArgumentException}.
*
For the general mechanism to load service providers, see
* {@link java.util.ServiceLoader ServiceLoader}, which is
* the underlying standard mechanism used by this class.
*
*
Only a single instance of a given leaf class (that is, the
* actual class returned by {@code getClass()}, as opposed to any
* inherited classes or interfaces) may be registered. That is,
* suppose that the
* {@code com.mycompany.mypkg.GreenImageReaderProvider} class
* is a subclass of {@code javax.imageio.spi.ImageReaderSpi}.
* If a {@code GreenImageReaderProvider} instance is
* registered, it will be stored in the category defined by the
* {@code ImageReaderSpi} class. If a new instance of
* {@code GreenImageReaderProvider} is registered, it will replace
* the previous instance. In practice, service provider objects are
* usually singletons so this behavior is appropriate.
*
*
The service provider classes should be lightweight and
* quick to load. Implementations of these interfaces should avoid
* complex dependencies on other classes and on native code. The usual
* pattern for more complex services is to register a lightweight
* proxy for the heavyweight service.
*
*
An application may customize the contents of a registry as it
* sees fit, so long as it has the appropriate runtime permission.
*
*
For information on how to create and deploy service providers,
* refer to the documentation on {@link java.util.ServiceLoader ServiceLoader}
*
* @see RegisterableService
* @see java.util.ServiceLoader
*/
public class ServiceRegistry {
// Class -> Registry
private Map, SubRegistry> categoryMap = new HashMap<>();
/**
* Constructs a {@code ServiceRegistry} instance with a
* set of categories taken from the {@code categories}
* argument. The categories must all be members of the set
* of service types listed in the class specification.
*
* @param categories an {@code Iterator} containing
* {@code Class} objects to be used to define categories.
*
* @exception IllegalArgumentException if
* {@code categories} is {@code null}, or if
* one of the categories is not an allowed service type.
*/
public ServiceRegistry(Iterator> categories) {
if (categories == null) {
throw new IllegalArgumentException("categories == null!");
}
while (categories.hasNext()) {
Class> category = categories.next();
checkClassAllowed(category);
SubRegistry reg = new SubRegistry(this, category);
categoryMap.put(category, reg);
}
}
/**
* Searches for implementations of a particular service class
* using the given class loader.
*
*
The service class must be one of the service types listed
* in the class specification. If it is not, {@code IllegalArgumentException}
* will be thrown.
*
*
This method transforms the name of the given service class
* into a provider-configuration filename as described in the
* class comment and then uses the {@code getResources}
* method of the given class loader to find all available files
* with that name. These files are then read and parsed to
* produce a list of provider-class names. The iterator that is
* returned uses the given class loader to look up and then
* instantiate each element of the list.
*
*
Because it is possible for extensions to be installed into
* a running Java virtual machine, this method may return
* different results each time it is invoked.
*
* @param providerClass a {@code Class} object indicating the
* class or interface of the service providers being detected.
*
* @param loader the class loader to be used to load
* provider-configuration files and instantiate provider classes,
* or {@code null} if the system class loader (or, failing that
* the bootstrap class loader) is to be used.
*
* @param the type of the providerClass.
*
* @return An {@code Iterator} that yields provider objects
* for the given service, in some arbitrary order. The iterator
* will throw an {@code Error} if a provider-configuration
* file violates the specified format or if a provider class
* cannot be found and instantiated.
*
* @exception IllegalArgumentException if
* {@code providerClass} is {@code null}, or if it is
* not one of the allowed service types.
*/
public static Iterator lookupProviders(Class providerClass,
ClassLoader loader)
{
if (providerClass == null) {
throw new IllegalArgumentException("providerClass == null!");
}
checkClassAllowed(providerClass);
return ServiceLoader.load(providerClass, loader).iterator();
}
/**
* Locates and incrementally instantiates the available providers
* of a given service using the context class loader. This
* convenience method is equivalent to:
*
*
The service class must be one of the service types listed
* in the class specification. If it is not, {@code IllegalArgumentException}
* will be thrown.
*
* @param providerClass a {@code Class} object indicating the
* class or interface of the service providers being detected.
*
* @param the type of the providerClass.
*
* @return An {@code Iterator} that yields provider objects
* for the given service, in some arbitrary order. The iterator
* will throw an {@code Error} if a provider-configuration
* file violates the specified format or if a provider class
* cannot be found and instantiated.
*
* @exception IllegalArgumentException if
* {@code providerClass} is {@code null}, or if it is
* not one of the allowed service types.
*/
public static Iterator lookupProviders(Class providerClass) {
if (providerClass == null) {
throw new IllegalArgumentException("providerClass == null!");
}
checkClassAllowed(providerClass);
return ServiceLoader.load(providerClass).iterator();
}
/**
* Returns an {@code Iterator} of {@code Class} objects
* indicating the current set of categories. The iterator will be
* empty if no categories exist.
*
* @return an {@code Iterator} containing
* {@code Class} objects.
*/
public Iterator> getCategories() {
Set> keySet = categoryMap.keySet();
return keySet.iterator();
}
/**
* Returns an Iterator containing the subregistries to which the
* provider belongs.
*/
private Iterator getSubRegistries(Object provider) {
List l = new ArrayList<>();
for (Class> c : categoryMap.keySet()) {
if (c.isAssignableFrom(provider.getClass())) {
l.add(categoryMap.get(c));
}
}
return l.iterator();
}
/**
* Adds a service provider object to the registry. The provider
* is associated with the given category.
*
*
If {@code provider} implements the
* {@code RegisterableService} interface, its
* {@code onRegistration} method will be called. Its
* {@code onDeregistration} method will be called each time
* it is deregistered from a category, for example if a
* category is removed or the registry is garbage collected.
*
* @param provider the service provide object to be registered.
* @param category the category under which to register the
* provider.
* @param the type of the provider.
*
* @return true if no provider of the same class was previously
* registered in the same category category.
*
* @exception IllegalArgumentException if {@code provider} is
* {@code null}.
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
* @exception ClassCastException if provider does not implement
* the {@code Class} defined by {@code category}.
*/
public boolean registerServiceProvider(T provider,
Class category) {
if (provider == null) {
throw new IllegalArgumentException("provider == null!");
}
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
if (!category.isAssignableFrom(provider.getClass())) {
throw new ClassCastException();
}
return reg.registerServiceProvider(provider);
}
/**
* Adds a service provider object to the registry. The provider
* is associated within each category present in the registry
* whose {@code Class} it implements.
*
*
If {@code provider} implements the
* {@code RegisterableService} interface, its
* {@code onRegistration} method will be called once for each
* category it is registered under. Its
* {@code onDeregistration} method will be called each time
* it is deregistered from a category or when the registry is
* finalized.
*
* @param provider the service provider object to be registered.
*
* @exception IllegalArgumentException if
* {@code provider} is {@code null}.
*/
public void registerServiceProvider(Object provider) {
if (provider == null) {
throw new IllegalArgumentException("provider == null!");
}
Iterator regs = getSubRegistries(provider);
while (regs.hasNext()) {
SubRegistry reg = regs.next();
reg.registerServiceProvider(provider);
}
}
/**
* Adds a set of service provider objects, taken from an
* {@code Iterator} to the registry. Each provider is
* associated within each category present in the registry whose
* {@code Class} it implements.
*
*
For each entry of {@code providers} that implements
* the {@code RegisterableService} interface, its
* {@code onRegistration} method will be called once for each
* category it is registered under. Its
* {@code onDeregistration} method will be called each time
* it is deregistered from a category or when the registry is
* finalized.
*
* @param providers an Iterator containing service provider
* objects to be registered.
*
* @exception IllegalArgumentException if {@code providers}
* is {@code null} or contains a {@code null} entry.
*/
public void registerServiceProviders(Iterator> providers) {
if (providers == null) {
throw new IllegalArgumentException("provider == null!");
}
while (providers.hasNext()) {
registerServiceProvider(providers.next());
}
}
/**
* Removes a service provider object from the given category. If
* the provider was not previously registered, nothing happens and
* {@code false} is returned. Otherwise, {@code true}
* is returned. If an object of the same class as
* {@code provider} but not equal (using {@code ==}) to
* {@code provider} is registered, it will not be
* deregistered.
*
*
If {@code provider} implements the
* {@code RegisterableService} interface, its
* {@code onDeregistration} method will be called.
*
* @param provider the service provider object to be deregistered.
* @param category the category from which to deregister the
* provider.
* @param the type of the provider.
*
* @return {@code true} if the provider was previously
* registered in the same category category,
* {@code false} otherwise.
*
* @exception IllegalArgumentException if {@code provider} is
* {@code null}.
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
* @exception ClassCastException if provider does not implement
* the class defined by {@code category}.
*/
public boolean deregisterServiceProvider(T provider,
Class category) {
if (provider == null) {
throw new IllegalArgumentException("provider == null!");
}
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
if (!category.isAssignableFrom(provider.getClass())) {
throw new ClassCastException();
}
return reg.deregisterServiceProvider(provider);
}
/**
* Removes a service provider object from all categories that
* contain it.
*
* @param provider the service provider object to be deregistered.
*
* @exception IllegalArgumentException if {@code provider} is
* {@code null}.
*/
public void deregisterServiceProvider(Object provider) {
if (provider == null) {
throw new IllegalArgumentException("provider == null!");
}
Iterator regs = getSubRegistries(provider);
while (regs.hasNext()) {
SubRegistry reg = regs.next();
reg.deregisterServiceProvider(provider);
}
}
/**
* Returns {@code true} if {@code provider} is currently
* registered.
*
* @param provider the service provider object to be queried.
*
* @return {@code true} if the given provider has been
* registered.
*
* @exception IllegalArgumentException if {@code provider} is
* {@code null}.
*/
public boolean contains(Object provider) {
if (provider == null) {
throw new IllegalArgumentException("provider == null!");
}
Iterator regs = getSubRegistries(provider);
while (regs.hasNext()) {
SubRegistry reg = regs.next();
if (reg.contains(provider)) {
return true;
}
}
return false;
}
/**
* Returns an {@code Iterator} containing all registered
* service providers in the given category. If
* {@code useOrdering} is {@code false}, the iterator
* will return all of the server provider objects in an arbitrary
* order. Otherwise, the ordering will respect any pairwise
* orderings that have been set. If the graph of pairwise
* orderings contains cycles, any providers that belong to a cycle
* will not be returned.
*
* @param category the category to be retrieved from.
* @param useOrdering {@code true} if pairwise orderings
* should be taken account in ordering the returned objects.
* @param the type of the category.
*
* @return an {@code Iterator} containing service provider
* objects from the given category, possibly in order.
*
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
*/
public Iterator getServiceProviders(Class category,
boolean useOrdering) {
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
@SuppressWarnings("unchecked")
Iterator it = (Iterator)reg.getServiceProviders(useOrdering);
return it;
}
/**
* A simple filter interface used by
* {@code ServiceRegistry.getServiceProviders} to select
* providers matching an arbitrary criterion. Classes that
* implement this interface should be defined in order to make use
* of the {@code getServiceProviders} method of
* {@code ServiceRegistry} that takes a {@code Filter}.
*
* @see ServiceRegistry#getServiceProviders(Class, ServiceRegistry.Filter, boolean)
*/
public interface Filter {
/**
* Returns {@code true} if the given
* {@code provider} object matches the criterion defined
* by this {@code Filter}.
*
* @param provider a service provider {@code Object}.
*
* @return true if the provider matches the criterion.
*/
boolean filter(Object provider);
}
/**
* Returns an {@code Iterator} containing service provider
* objects within a given category that satisfy a criterion
* imposed by the supplied {@code ServiceRegistry.Filter}
* object's {@code filter} method.
*
*
The {@code useOrdering} argument controls the
* ordering of the results using the same rules as
* {@code getServiceProviders(Class, boolean)}.
*
* @param category the category to be retrieved from.
* @param filter an instance of {@code ServiceRegistry.Filter}
* whose {@code filter} method will be invoked.
* @param useOrdering {@code true} if pairwise orderings
* should be taken account in ordering the returned objects.
* @param the type of the category.
*
* @return an {@code Iterator} containing service provider
* objects from the given category, possibly in order.
*
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
*/
public Iterator getServiceProviders(Class category,
Filter filter,
boolean useOrdering) {
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
Iterator iter = getServiceProviders(category, useOrdering);
return new FilterIterator<>(iter, filter);
}
/**
* Returns the currently registered service provider object that
* is of the given class type. At most one object of a given
* class is allowed to be registered at any given time. If no
* registered object has the desired class type, {@code null}
* is returned.
*
* @param providerClass the {@code Class} of the desired
* service provider object.
* @param the type of the provider.
*
* @return a currently registered service provider object with the
* desired {@code Class} type, or {@code null} is none is
* present.
*
* @exception IllegalArgumentException if {@code providerClass} is
* {@code null}.
*/
public T getServiceProviderByClass(Class providerClass) {
if (providerClass == null) {
throw new IllegalArgumentException("providerClass == null!");
}
for (Class> c : categoryMap.keySet()) {
if (c.isAssignableFrom(providerClass)) {
SubRegistry reg = categoryMap.get(c);
T provider = reg.getServiceProviderByClass(providerClass);
if (provider != null) {
return provider;
}
}
}
return null;
}
/**
* Sets a pairwise ordering between two service provider objects
* within a given category. If one or both objects are not
* currently registered within the given category, or if the
* desired ordering is already set, nothing happens and
* {@code false} is returned. If the providers previously
* were ordered in the reverse direction, that ordering is
* removed.
*
*
The ordering will be used by the
* {@code getServiceProviders} methods when their
* {@code useOrdering} argument is {@code true}.
*
* @param category a {@code Class} object indicating the
* category under which the preference is to be established.
* @param firstProvider the preferred provider.
* @param secondProvider the provider to which
* {@code firstProvider} is preferred.
* @param the type of the category.
*
* @return {@code true} if a previously unset ordering
* was established.
*
* @exception IllegalArgumentException if either provider is
* {@code null} or they are the same object.
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
*/
public boolean setOrdering(Class category,
T firstProvider,
T secondProvider) {
if (firstProvider == null || secondProvider == null) {
throw new IllegalArgumentException("provider is null!");
}
if (firstProvider == secondProvider) {
throw new IllegalArgumentException("providers are the same!");
}
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
if (reg.contains(firstProvider) &&
reg.contains(secondProvider)) {
return reg.setOrdering(firstProvider, secondProvider);
}
return false;
}
/**
* Sets a pairwise ordering between two service provider objects
* within a given category. If one or both objects are not
* currently registered within the given category, or if no
* ordering is currently set between them, nothing happens
* and {@code false} is returned.
*
*
The ordering will be used by the
* {@code getServiceProviders} methods when their
* {@code useOrdering} argument is {@code true}.
*
* @param category a {@code Class} object indicating the
* category under which the preference is to be disestablished.
* @param firstProvider the formerly preferred provider.
* @param secondProvider the provider to which
* {@code firstProvider} was formerly preferred.
* @param the type of the category.
*
* @return {@code true} if a previously set ordering was
* disestablished.
*
* @exception IllegalArgumentException if either provider is
* {@code null} or they are the same object.
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
*/
public boolean unsetOrdering(Class category,
T firstProvider,
T secondProvider) {
if (firstProvider == null || secondProvider == null) {
throw new IllegalArgumentException("provider is null!");
}
if (firstProvider == secondProvider) {
throw new IllegalArgumentException("providers are the same!");
}
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
if (reg.contains(firstProvider) &&
reg.contains(secondProvider)) {
return reg.unsetOrdering(firstProvider, secondProvider);
}
return false;
}
/**
* Deregisters all service provider object currently registered
* under the given category.
*
* @param category the category to be emptied.
*
* @exception IllegalArgumentException if there is no category
* corresponding to {@code category}.
*/
public void deregisterAll(Class> category) {
SubRegistry reg = categoryMap.get(category);
if (reg == null) {
throw new IllegalArgumentException("category unknown!");
}
reg.clear();
}
/**
* Deregisters all currently registered service providers from all
* categories.
*/
public void deregisterAll() {
for (SubRegistry reg : categoryMap.values()) {
reg.clear();
}
}
/**
* Finalizes this object prior to garbage collection. The
* {@code deregisterAll} method is called to deregister all
* currently registered service providers. This method should not
* be called from application code.
*
* @exception Throwable if an error occurs during superclass
* finalization.
*
* @deprecated Finalization has been deprecated for removal. See
* {@link java.lang.Object#finalize} for background information and details
* about migration options.
*/
@Deprecated(since="9", forRemoval=true)
@SuppressWarnings("removal")
public void finalize() throws Throwable {
deregisterAll();
super.finalize();
}
/**
* Checks whether the provided class is one of the allowed
* ImageIO service provider classes. If it is, returns normally.
* If it is not, throws IllegalArgumentException.
*
* @param clazz
* @throws IllegalArgumentException if clazz is null or is not one of the allowed set
*/
private static void checkClassAllowed(Class> clazz) {
if (clazz == null) {
throw new IllegalArgumentException("class must not be null");
}
if ( clazz != ImageInputStreamSpi.class
&& clazz != ImageOutputStreamSpi.class
&& clazz != ImageReaderSpi.class
&& clazz != ImageTranscoderSpi.class
&& clazz != ImageWriterSpi.class) {
throw new IllegalArgumentException(clazz.getName() + " is not an ImageIO SPI class");
}
}
}
/**
* A portion of a registry dealing with a single superclass or
* interface.
*/
class SubRegistry {
ServiceRegistry registry;
Class> category;
// Provider Objects organized by partial ordering
final PartiallyOrderedSet