/*

 * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/**

 * @test

 * @bug     6936389

 *

 * @summary Test verifes that LogManager shutdown hook does not cause

 *          an application classloader leaks.

 *

 * @run     main/othervm ClassLoaderLeakTest FontManagerTest

 */

import java.awt.Font;

import java.awt.Graphics;

import java.io.File;

import java.lang.ref.WeakReference;

import java.lang.reflect.Constructor;

import java.lang.reflect.Method;

import java.net.MalformedURLException;

import java.net.URL;

import java.net.URLClassLoader;

import java.util.concurrent.CountDownLatch;

public class ClassLoaderLeakTest {

    private static CountDownLatch doneSignal;

    private static CountDownLatch launchSignal;

    private static Throwable launchFailure = null;

    public static void main(String[] args) {

        doneSignal = new CountDownLatch(1);

        launchSignal = new CountDownLatch(1);

        String testcase = "FontManagerTest";

        if (args.length > 0) {

            testcase = args[0];

        }

        /* prepare test  class loader */

        URL pwd = null;

        try {

            pwd = new File(System.getProperty("test.classes", ".")).toURL();

        } catch (MalformedURLException e) {

            throw new RuntimeException("Test failed.", e);

        }

        System.out.println("PWD: " + pwd);

        URL[] urls = new URL[]{pwd};

        MyClassLoader appClassLoader = new MyClassLoader(urls, "test0");

        WeakReference<MyClassLoader> ref =

            new WeakReference<MyClassLoader>(appClassLoader);

        ThreadGroup appsThreadGroup = new ThreadGroup("MyAppsThreadGroup");

        Runnable launcher = new TestLauncher(testcase);

        Thread appThread = new Thread(appsThreadGroup, launcher, "AppThread-0");

        appThread.setContextClassLoader(appClassLoader);

        appThread.start();

        appsThreadGroup = null;

        appClassLoader = null;

        launcher = null;

        appThread = null;

        /* wait for laucnh completion */

        try {

            launchSignal.await();

        } catch (InterruptedException e) {

        }

        /* check if launch failed */

        if (launchFailure != null) {

            throw new RuntimeException("Test failed.", launchFailure);

        }

        /* wait for test app excution completion */

        try {

            doneSignal.await();

        } catch (InterruptedException e) {

        }

        /* give a chance to GC */

        waitAndGC(9);

        if (ref.get() != null) {

            throw new RuntimeException("Test failed: classloader is still alive");

        }

        System.out.println("Test passed.");

    }

    private static class TestLauncher implements Runnable {

        private String className;

        public TestLauncher(String name) {

            className = name;

        }

        public void run() {

            try {

                ClassLoader cl =

                    Thread.currentThread().getContextClassLoader();

                Class appMain = cl.loadClass(className);

                Method launch =

                    appMain.getMethod("launch", doneSignal.getClass());

                Constructor c = appMain.getConstructor();

                Object o = c.newInstance();

                launch.invoke(o, doneSignal);

            } catch (Throwable e) {

                launchFailure = e;

            } finally {

                launchSignal.countDown();

            }

        }

    }

    private static class MyClassLoader extends URLClassLoader {

        private static boolean verbose =

            Boolean.getBoolean("verboseClassLoading");

        private String uniqClassName;

        public MyClassLoader(URL[] urls, String uniq) {

            super(urls);

            uniqClassName = uniq;

        }

        public Class loadClass(String name) throws ClassNotFoundException {

            if (verbose) {

                System.out.printf("%s: load class %s\n", uniqClassName, name);

            }

            if (uniqClassName.equals(name)) {

                return Object.class;

            }

            return super.loadClass(name);

        }

        public String toString() {

            return "MyClassLoader(" + uniqClassName + ")";

        }

    }

    private static void waitAndGC(int sec) {

        int cnt = sec;

        System.out.print("Wait ");

        while (cnt-- > 0) {

            try {

                Thread.sleep(1000);

            } catch (InterruptedException e) {

            }

            // do GC every 3 seconds

            if (cnt % 3 == 2) {

                System.gc();

                System.out.print("+");

            } else {

                System.out.print(".");

            }

            //checkErrors();

        }

        System.out.println("");

    }

}

abstract class AppTest {

    public AppTest() {

    }

    protected abstract void doTest();

    public void launch(CountDownLatch done) {

        System.out.println("Testcase: " + this.getClass().getName());

        try {

            doTest();

        } finally {

            done.countDown();

        }

    }

}

class FontManagerTest extends AppTest {

    public FontManagerTest() {

    }

    protected void doTest() {

        Font f = new Font(Font.SANS_SERIF, Font.ITALIC, 24);

        f.getNumGlyphs();

    }

}