/*
 * 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();
    }
}