/*

 * Copyright (c) 2013, 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.

 */

import boot.GetCallerClass;

import java.lang.reflect.*;

import sun.reflect.CallerSensitive;

import sun.reflect.Reflection;

public class GetCallerClassTest {

    private final GetCallerClass gcc;   // boot.GetCallerClass is in bootclasspath

    GetCallerClassTest() {

        this.gcc = new GetCallerClass();

    }

    public static void main(String[] args) throws Exception {

        GetCallerClassTest gcct = new GetCallerClassTest();

        // ensure methods are annotated with @CallerSensitive

        ensureAnnotationPresent(boot.GetCallerClass.class, "getCallerLoader", true);

        ensureAnnotationPresent(GetCallerClassTest.class, "testNonSystemMethod", false);

        // call Reflection.getCallerClass from bootclasspath with and without @CS

        gcct.testCallerSensitiveMethods();

        // call Reflection.getCallerClass from classpath with @CS

        gcct.testNonSystemMethod();

    }

    private static void ensureAnnotationPresent(Class<?> c, String name, boolean cs)

        throws NoSuchMethodException

    {

        Method m = c.getDeclaredMethod(name);

        if (!m.isAnnotationPresent(CallerSensitive.class)) {

            throw new RuntimeException("@CallerSensitive not present in method " + m);

        }

        if (Reflection.isCallerSensitive(m) != cs) {

            throw new RuntimeException("Unexpected: isCallerSensitive returns " +

                Reflection.isCallerSensitive(m));

        }

    }

    private void testCallerSensitiveMethods() {

        try {

            ClassLoader cl = gcc.getCallerLoader();

            if (cl != GetCallerClassTest.class.getClassLoader()) {

                throw new RuntimeException("mismatched class loader");

            }

            gcc.missingCallerSensitiveAnnotation();

            throw new RuntimeException("getCallerLoader not marked with @CallerSensitive");

        } catch (InternalError e) {

            StackTraceElement[] stackTrace = e.getStackTrace();

            checkStackTrace(stackTrace, e);

            if (!stackTrace[1].getClassName().equals(GetCallerClass.class.getName()) ||

                !stackTrace[1].getMethodName().equals("missingCallerSensitiveAnnotation")) {

                throw new RuntimeException("Unexpected error: " + e.getMessage(), e);

            }

            if (!stackTrace[2].getClassName().equals(GetCallerClassTest.class.getName()) ||

                !stackTrace[2].getMethodName().equals("testCallerSensitiveMethods")) {

                throw new RuntimeException("Unexpected error: " + e.getMessage(), e);

            }

            System.out.println("Expected error: " + e.getMessage());

        }

    }

    @CallerSensitive

    private void testNonSystemMethod() {

        try {

            Class<?> c = Reflection.getCallerClass();

            throw new RuntimeException("@CallerSensitive testNonSystemMethods not supported");

        } catch (InternalError e) {

            StackTraceElement[] stackTrace = e.getStackTrace();

            checkStackTrace(stackTrace, e);

            if (!stackTrace[1].getClassName().equals(GetCallerClassTest.class.getName()) ||

                !stackTrace[1].getMethodName().equals("testNonSystemMethod")) {

                throw new RuntimeException("Unexpected error: " + e.getMessage(), e);

            }

            if (!stackTrace[2].getClassName().equals(GetCallerClassTest.class.getName()) ||

                !stackTrace[2].getMethodName().equals("main")) {

                throw new RuntimeException("Unexpected error: " + e.getMessage(), e);

            }

            System.out.println("Expected error: " + e.getMessage());

        }

    }

    private void checkStackTrace(StackTraceElement[] stackTrace, Error e) {

        if (stackTrace.length < 3) {

            throw new RuntimeException("Unexpected error: " + e.getMessage(), e);

        }

        if (!stackTrace[0].getClassName().equals("sun.reflect.Reflection") ||

            !stackTrace[0].getMethodName().equals("getCallerClass")) {

            throw new RuntimeException("Unexpected error: " + e.getMessage(), e);

        }

    }

}