/*
* 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.net.InetSocketAddress;
import java.net.Proxy.Type;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
import java.util.jar.JarOutputStream;
import java.util.jar.Manifest;
import java.util.zip.ZipEntry;
import org.osgi.framework.Bundle;
import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.BundleEvent;
import org.osgi.framework.BundleException;
import org.osgi.framework.Constants;
import org.osgi.framework.SynchronousBundleListener;
import junit.framework.TestCase;
public class ConcurrentBundleUpdateTest extends TestCase
{
public void testConcurrentBundleUpdate() throws Exception
{
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");
File cacheDir = File.createTempFile("felix-cache", ".dir");
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);
try
{
Felix felix = new Felix(params);
try
{
felix.init();
felix.start();
String mf = "Bundle-SymbolicName: test.concurrent.bundleupdate\n"
+ "Bundle-Version: 1.0.0\n"
+ "Bundle-ManifestVersion: 2\n"
+ "Import-Package: org.osgi.framework\n"
+ "Manifest-Version: 1.0\n"
+ "Bundle-Activator: " + ConcurrentBundleUpdaterActivator.class.getName() + "\n\n";
final BundleImpl updater = (BundleImpl) felix.getBundleContext().installBundle(createBundle(mf, ConcurrentBundleUpdaterActivator.class).toURI().toURL().toString());
final Semaphore step = new Semaphore(0);
SynchronousBundleListener listenerStarting = new SynchronousBundleListener()
{
@Override
public void bundleChanged(BundleEvent event)
{
if (event.getBundle().equals(updater) && event.getType() == BundleEvent.STARTING)
{
step.release();
}
}
};
felix.getBundleContext().addBundleListener(listenerStarting);
new Thread()
{
public void run()
{
try
{
updater.start();
}
catch (Exception ex)
{
}
}
}.start();
assertTrue(step.tryAcquire(1, TimeUnit.SECONDS));
felix.getBundleContext().removeBundleListener(listenerStarting);
assertEquals(Bundle.STARTING, updater.getState());
assertEquals(0, step.availablePermits());
new Thread()
{
public void run()
{
try
{
step.release();
updater.update();
step.release();
}
catch (Exception ex)
{
}
}
}.start();
assertTrue(step.tryAcquire(1, TimeUnit.SECONDS));
SynchronousBundleListener listenerStarted = new SynchronousBundleListener()
{
@Override
public void bundleChanged(BundleEvent event)
{
if (event.getBundle().equals(updater) && event.getType() == BundleEvent.STARTED)
{
step.release();
}
if (event.getBundle().equals(updater) && event.getType() == BundleEvent.STOPPING)
{
step.release();
}
}
};
felix.getBundleContext().addBundleListener(listenerStarted);
((Runnable) updater.getActivator()).run();
assertTrue(step.tryAcquire(2, 1, TimeUnit.SECONDS));
felix.getBundleContext().removeBundleListener(listenerStarted);
assertEquals(0, step.availablePermits());
assertEquals(Bundle.STOPPING, updater.getState());
felix.getBundleContext().addBundleListener(listenerStarting);
((Runnable) updater.getActivator()).run();
assertTrue(step.tryAcquire(1, TimeUnit.SECONDS));
felix.getBundleContext().removeBundleListener(listenerStarting);
assertEquals(Bundle.STARTING, updater.getState());
((Runnable) updater.getActivator()).run();
assertTrue(step.tryAcquire(1, TimeUnit.SECONDS));
assertEquals(Bundle.ACTIVE, updater.getState());
((Runnable) updater.getActivator()).run();
updater.uninstall();
assertEquals(Bundle.UNINSTALLED, updater.getState());
try
{
updater.update();
fail("Expected exception on update of uninstalled bundle");
}
catch (IllegalStateException expected) {
}
}
finally
{
felix.stop();
felix.waitForStop(1000);
}
}
finally
{
delete(cacheDir);
}
}
public void testConcurrentBundleCycleUpdate() throws Exception
{
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");
File cacheDir = File.createTempFile("felix-cache", ".dir");
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);
try
{
Felix felix = new Felix(params);
try
{
felix.init();
felix.start();
String mf = "Bundle-SymbolicName: test.concurrent.bundleupdate\n"
+ "Bundle-Version: 1.0.0\n"
+ "Bundle-ManifestVersion: 2\n"
+ "Import-Package: org.osgi.framework\n"
+ "Manifest-Version: 1.0\n"
+ "Bundle-Activator: " + ConcurrentBundleUpdaterCycleActivator.class.getName() + "\n\n";
final BundleImpl updater = (BundleImpl) felix.getBundleContext().installBundle(createBundle(mf, ConcurrentBundleUpdaterCycleActivator.class).toURI().toURL().toString());
final Semaphore step = new Semaphore(0);
SynchronousBundleListener listenerStarting = new SynchronousBundleListener()
{
@Override
public void bundleChanged(BundleEvent event)
{
if (event.getBundle().equals(updater) && event.getType() == BundleEvent.STARTING)
{
step.release();
}
}
};
felix.getBundleContext().addBundleListener(listenerStarting);
new Thread()
{
public void run()
{
try
{
updater.start();
}
catch (Exception ex)
{
step.release();
}
}
}.start();
assertTrue(step.tryAcquire(1, TimeUnit.SECONDS));
felix.getBundleContext().removeBundleListener(listenerStarting);
assertEquals(Bundle.STARTING, updater.getState());
assertEquals(0, step.availablePermits());
((Runnable) updater.getActivator()).run();
assertTrue(step.tryAcquire(1, TimeUnit.SECONDS));
assertEquals(Bundle.RESOLVED, updater.getState());
updater.uninstall();
assertEquals(Bundle.UNINSTALLED, updater.getState());
try
{
updater.update();
fail("Expected exception on update of uninstalled bundle");
}
catch (IllegalStateException expected) {
}
}
finally
{
felix.stop();
felix.waitForStop(1000);
}
}
finally
{
delete(cacheDir);
}
}
public static final class ConcurrentBundleUpdaterActivator implements BundleActivator, Runnable
{
Semaphore semaphore = new Semaphore(0);
private BundleContext context;
@Override
public void start(BundleContext context) throws Exception
{
this.context = context;
if (!semaphore.tryAcquire(1, TimeUnit.SECONDS))
{
throw new BundleException("Timeout");
}
}
@Override
public void stop(BundleContext context) throws Exception
{
this.context = context;
if (!semaphore.tryAcquire(1, TimeUnit.SECONDS))
{
throw new BundleException("Timeout");
}
}
@Override
public void run()
{
semaphore.release();
}
}
public static final class ConcurrentBundleUpdaterCycleActivator implements BundleActivator, Runnable
{
Semaphore semaphore = new Semaphore(0);
private BundleContext context;
@Override
public void start(BundleContext context) throws Exception
{
this.context = context;
if (!semaphore.tryAcquire(1, TimeUnit.SECONDS))
{
throw new BundleException("Timeout");
}
context.getBundle().update();
}
@Override
public void stop(BundleContext context) throws Exception {
this.context = context;
if (!semaphore.tryAcquire(1, TimeUnit.SECONDS))
{
throw new BundleException("Timeout");
}
}
@Override
public void run()
{
semaphore.release();
}
}
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();
}
}