| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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.apache.felix.framework; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.File; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.util.ArrayList; |
| import java.util.Arrays; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.concurrent.Callable; |
| import java.util.jar.JarOutputStream; |
| import java.util.jar.Manifest; |
| import java.util.zip.ZipEntry; |
| |
| import org.junit.Assert; |
| import org.junit.Assume; |
| import org.osgi.framework.Bundle; |
| import org.osgi.framework.BundleActivator; |
| import org.osgi.framework.BundleContext; |
| import org.osgi.framework.BundleReference; |
| import org.osgi.framework.Constants; |
| import org.osgi.framework.ServiceFactory; |
| import org.osgi.framework.ServiceRegistration; |
| import org.osgi.framework.hooks.weaving.WeavingHook; |
| import org.osgi.framework.hooks.weaving.WovenClass; |
| |
| import junit.framework.TestCase; |
| |
| public class CycleDetectionWithWovenClassTest extends TestCase { |
| |
| public void testDoesBootdelegateForClassloaderClassload() throws Exception{ |
| withFelixDo(new ThrowingConsumer<Felix>() { |
| @Override |
| public void accept(Felix felix) throws Exception { |
| BundleImpl bundle = (BundleImpl) felix.getBundleContext().installBundle(createBundle( |
| WeavingActivator.class, CycleDetectionWithWovenClassTest.class, Hook.class, getClass().getClassLoader().loadClass("org.apache.felix.framework.CycleDetectionWithWovenClassTest$WeavingActivator$1") |
| ).toURI().toURL().toString()); |
| |
| bundle.start(); |
| |
| Runnable testClass = felix.getBundleContext().getService( |
| felix.getBundleContext().getServiceReference(Runnable.class)); |
| |
| testClass.run(); |
| } |
| }); |
| } |
| |
| |
| public static class WeavingActivator implements BundleActivator, Runnable { |
| private volatile BundleContext context; |
| private volatile boolean wasNull = false; |
| @Override |
| public void start(final BundleContext context) throws Exception { |
| this.context = context; |
| context.registerService(WeavingHook.class.getName(), new ServiceFactory<WeavingHook>() |
| { |
| |
| @Override |
| public WeavingHook getService(Bundle bundle, ServiceRegistration<WeavingHook> registration) |
| { |
| // TODO Auto-generated method stub |
| try |
| { |
| Class hook = getClass().getClassLoader().loadClass("org.apache.felix.framework.CycleDetectionWithWovenClassTest$Hook"); |
| if (hook == null) |
| { |
| wasNull = true; |
| return null; |
| } |
| else { |
| return (WeavingHook) hook.newInstance(); |
| } |
| |
| } |
| catch (ClassNotFoundException e) |
| { |
| // This is what we expect. |
| } |
| catch (Exception ex) { |
| wasNull = true; |
| } |
| return null; |
| } |
| |
| @Override |
| public void ungetService(Bundle bundle, ServiceRegistration<WeavingHook> registration, |
| WeavingHook service) |
| { |
| // TODO Auto-generated method stub |
| |
| } |
| }, null); |
| |
| context.registerService(Runnable.class, this, null); |
| } |
| |
| @Override |
| public void stop(BundleContext context) throws Exception { |
| // TODO Auto-generated method stub |
| |
| } |
| |
| public void run() { |
| try |
| { |
| ((Callable<Boolean>) context.getBundle().loadClass("org.apache.felix.framework.CycleDetectionWithWovenClassTest$Hook").newInstance()).call(); |
| |
| if (wasNull) |
| { |
| throw new IllegalStateException("Returned null from nested classload"); |
| } |
| } catch (Exception e) |
| { |
| throw new RuntimeException(e); |
| } |
| } |
| |
| } |
| public static class Hook implements WeavingHook, Callable<Boolean> { |
| |
| private static boolean woven = false; |
| @Override |
| public void weave(WovenClass wovenClass) |
| { |
| woven = true; |
| } |
| |
| @Override |
| public Boolean call() throws Exception |
| { |
| return woven; |
| } |
| |
| } |
| |
| private void withFelixDo(ThrowingConsumer<Felix> consumer) throws Exception { |
| File cacheDir = File.createTempFile("felix-cache", ".dir"); |
| try |
| { |
| Felix felix = getFramework(cacheDir); |
| try |
| { |
| felix.init(); |
| felix.start(); |
| consumer.accept(felix); |
| } |
| finally |
| { |
| felix.stop(); |
| felix.waitForStop(1000); |
| } |
| } |
| finally |
| { |
| delete(cacheDir); |
| } |
| } |
| |
| @FunctionalInterface |
| private static interface ThrowingConsumer<T> { |
| public void accept(T t) throws Exception; |
| } |
| |
| |
| private Felix getFramework(File cacheDir) { |
| Map params = new HashMap(); |
| params.put(Constants.FRAMEWORK_SYSTEMPACKAGES, |
| "org.osgi.framework; version=1.4.0," |
| + "org.osgi.service.packageadmin; version=1.2.0," |
| + "org.osgi.service.startlevel; version=1.1.0," |
| + "org.osgi.util.tracker; version=1.3.3," |
| + "org.osgi.service.url; version=1.0.0," |
| + "org.osgi.framework.hooks.weaving"); |
| cacheDir.delete(); |
| cacheDir.mkdirs(); |
| String cache = cacheDir.getPath(); |
| params.put("felix.cache.profiledir", cache); |
| params.put("felix.cache.dir", cache); |
| params.put(Constants.FRAMEWORK_STORAGE, cache); |
| |
| return new Felix(params); |
| } |
| |
| private static File createBundle(Class activator, Class...classes) throws IOException |
| { |
| String mf = "Bundle-SymbolicName: " + activator.getName() +"\n" |
| + "Bundle-Version: 1.0.0\n" |
| + "Bundle-ManifestVersion: 2\n" |
| + "Import-Package: org.osgi.framework,org.osgi.framework.hooks.weaving\n" |
| + "Manifest-Version: 1.0\n" |
| + "Bundle-Activator: " + activator.getName() + "\n\n"; |
| |
| Class[] classesCombined; |
| |
| if (classes.length > 0) { |
| List<Class> list = new ArrayList<Class>(Arrays.asList(classes)); |
| list.add(activator); |
| classesCombined = list.toArray(new Class[0]); |
| } |
| else |
| { |
| classesCombined = new Class[]{activator}; |
| } |
| return createBundle(mf,classesCombined); |
| |
| } |
| |
| private static File createBundle(String manifest, Class... classes) throws IOException |
| { |
| File f = File.createTempFile("felix-bundle", ".jar"); |
| f.deleteOnExit(); |
| |
| Manifest mf = new Manifest(new ByteArrayInputStream(manifest.getBytes("utf-8"))); |
| JarOutputStream os = new JarOutputStream(new FileOutputStream(f), mf); |
| |
| for (Class clazz : classes) |
| { |
| String path = clazz.getName().replace('.', '/') + ".class"; |
| os.putNextEntry(new ZipEntry(path)); |
| |
| InputStream is = clazz.getClassLoader().getResourceAsStream(path); |
| byte[] buffer = new byte[8 * 1024]; |
| for (int i = is.read(buffer); i != -1; i = is.read(buffer)) |
| { |
| os.write(buffer, 0, i); |
| } |
| is.close(); |
| os.closeEntry(); |
| } |
| os.close(); |
| return f; |
| } |
| |
| private static void delete(File file) throws IOException |
| { |
| if (file.isDirectory()) |
| { |
| for (File child : file.listFiles()) |
| { |
| delete(child); |
| } |
| } |
| file.delete(); |
| } |
| } |