1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19 package org.apache.maven.surefire.junit4;
20
21 import java.util.Set;
22
23 /**
24 * Emulate an OSGi classloader which only loads packages that have been imported via Import-Package MANIFEST header.
25 */
26 public class PackageFilteringClassLoader extends ClassLoader {
27
28 private ClassLoader wrapped;
29
30 private Set<String> visiblePackages;
31
32 public PackageFilteringClassLoader(ClassLoader wrapped, Set<String> visiblePackages) {
33 this.wrapped = wrapped;
34 this.visiblePackages = visiblePackages;
35 }
36
37 @Override
38 public Class<?> loadClass(String className) throws ClassNotFoundException {
39 String packageName = "";
40 int lastDot = className.lastIndexOf('.');
41 if (lastDot != -1) {
42 packageName = className.substring(0, lastDot);
43 }
44 if (visiblePackages.contains(packageName)) {
45 return wrapped.loadClass(className);
46 } else {
47 throw new ClassNotFoundException(className);
48 }
49 }
50 }