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