/*

 * Copyright (c) 2014, 2018, 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.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * 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.

 */

package java.lang;

import java.io.IOException;

import java.io.InputStream;

import java.lang.annotation.Annotation;

import java.lang.module.Configuration;

import java.lang.module.ModuleReference;

import java.lang.module.ModuleDescriptor;

import java.lang.module.ModuleDescriptor.Exports;

import java.lang.module.ModuleDescriptor.Opens;

import java.lang.module.ModuleDescriptor.Version;

import java.lang.module.ResolvedModule;

import java.lang.reflect.AnnotatedElement;

import java.net.URI;

import java.net.URL;

import java.security.AccessController;

import java.security.PrivilegedAction;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import java.util.Optional;

import java.util.Set;

import java.util.concurrent.ConcurrentHashMap;

import java.util.function.Function;

import java.util.stream.Collectors;

import java.util.stream.Stream;

import jdk.internal.loader.BuiltinClassLoader;

import jdk.internal.loader.BootLoader;

import jdk.internal.loader.ClassLoaders;

import jdk.internal.module.IllegalAccessLogger;

import jdk.internal.module.ModuleLoaderMap;

import jdk.internal.module.ServicesCatalog;

import jdk.internal.module.Resources;

import jdk.internal.org.objectweb.asm.AnnotationVisitor;

import jdk.internal.org.objectweb.asm.Attribute;

import jdk.internal.org.objectweb.asm.ClassReader;

import jdk.internal.org.objectweb.asm.ClassVisitor;

import jdk.internal.org.objectweb.asm.ClassWriter;

import jdk.internal.org.objectweb.asm.ModuleVisitor;

import jdk.internal.org.objectweb.asm.Opcodes;

import jdk.internal.reflect.CallerSensitive;

import jdk.internal.reflect.Reflection;

import sun.security.util.SecurityConstants;

/**

 * Represents a run-time module, either {@link #isNamed() named} or unnamed.

 *

 * <p> Named modules have a {@link #getName() name} and are constructed by the

 * Java Virtual Machine when a graph of modules is defined to the Java virtual

 * machine to create a {@linkplain ModuleLayer module layer}. </p>

 *

 * <p> An unnamed module does not have a name. There is an unnamed module for

 * each {@link ClassLoader ClassLoader}, obtained by invoking its {@link

 * ClassLoader#getUnnamedModule() getUnnamedModule} method. All types that are

 * not in a named module are members of their defining class loader's unnamed

 * module. </p>

 *

 * <p> The package names that are parameters or returned by methods defined in

 * this class are the fully-qualified names of the packages as defined in

 * section 6.5.3 of <cite>The Java&trade; Language Specification</cite>, for

 * example, {@code "java.lang"}. </p>

 *

 * <p> Unless otherwise specified, passing a {@code null} argument to a method

 * in this class causes a {@link NullPointerException NullPointerException} to

 * be thrown. </p>

 *

 * @since 9

 * @spec JPMS

 * @see Class#getModule()

 */

public final class Module implements AnnotatedElement {

    // the layer that contains this module, can be null

    private final ModuleLayer layer;

    // module name and loader, these fields are read by VM

    private final String name;

    private final ClassLoader loader;

    // the module descriptor

    private final ModuleDescriptor descriptor;

    /**

     * Creates a new named Module. The resulting Module will be defined to the

     * VM but will not read any other modules, will not have any exports setup

     * and will not be registered in the service catalog.

     */

    Module(ModuleLayer layer,

           ClassLoader loader,

           ModuleDescriptor descriptor,

           URI uri)

    {

        this.layer = layer;

        this.name = descriptor.name();

        this.loader = loader;

        this.descriptor = descriptor;

        // define module to VM

        boolean isOpen = descriptor.isOpen() || descriptor.isAutomatic();

        Version version = descriptor.version().orElse(null);

        String vs = Objects.toString(version, null);

        String loc = Objects.toString(uri, null);

        String[] packages = descriptor.packages().toArray(new String[0]);

        defineModule0(this, isOpen, vs, loc, packages);

    }

    /**

     * Create the unnamed Module for the given ClassLoader.

     *

     * @see ClassLoader#getUnnamedModule

     */

    Module(ClassLoader loader) {

        this.layer = null;

        this.name = null;

        this.loader = loader;

        this.descriptor = null;

    }

    /**

     * Creates a named module but without defining the module to the VM.

     *

     * @apiNote This constructor is for VM white-box testing.

     */

    Module(ClassLoader loader, ModuleDescriptor descriptor) {

        this.layer = null;

        this.name = descriptor.name();

        this.loader = loader;

        this.descriptor = descriptor;

    }

    /**

     * Returns {@code true} if this module is a named module.

     *

     * @return {@code true} if this is a named module

     *

     * @see ClassLoader#getUnnamedModule()

     */

    public boolean isNamed() {

        return name != null;

    }

    /**

     * Returns the module name or {@code null} if this module is an unnamed

     * module.

     *

     * @return The module name

     */

    public String getName() {

        return name;

    }

    /**

     * Returns the {@code ClassLoader} for this module.

     *

     * <p> If there is a security manager then its {@code checkPermission}

     * method if first called with a {@code RuntimePermission("getClassLoader")}

     * permission to check that the caller is allowed to get access to the

     * class loader. </p>

     *

     * @return The class loader for this module

     *

     * @throws SecurityException

     *         If denied by the security manager

     */

    public ClassLoader getClassLoader() {

        SecurityManager sm = System.getSecurityManager();

        if (sm != null) {

            sm.checkPermission(SecurityConstants.GET_CLASSLOADER_PERMISSION);

        }

        return loader;

    }

    /**

     * Returns the module descriptor for this module or {@code null} if this

     * module is an unnamed module.

     *

     * @return The module descriptor for this module

     */

    public ModuleDescriptor getDescriptor() {

        return descriptor;

    }

    /**

     * Returns the module layer that contains this module or {@code null} if

     * this module is not in a module layer.

     *

     * A module layer contains named modules and therefore this method always

     * returns {@code null} when invoked on an unnamed module.

     *

     * <p> <a href="reflect/Proxy.html#dynamicmodule">Dynamic modules</a> are

     * named modules that are generated at runtime. A dynamic module may or may

     * not be in a module layer. </p>

     *

     * @return The module layer that contains this module

     *

     * @see java.lang.reflect.Proxy

     */

    public ModuleLayer getLayer() {

        if (isNamed()) {

            ModuleLayer layer = this.layer;

            if (layer != null)

                return layer;

            // special-case java.base as it is created before the boot layer

            if (loader == null && name.equals("java.base")) {

                return ModuleLayer.boot();

            }

        }

        return null;

    }

    // --

    // special Module to mean "all unnamed modules"

    private static final Module ALL_UNNAMED_MODULE = new Module(null);

    private static final Set<Module> ALL_UNNAMED_MODULE_SET = Set.of(ALL_UNNAMED_MODULE);

    // special Module to mean "everyone"

    private static final Module EVERYONE_MODULE = new Module(null);

    private static final Set<Module> EVERYONE_SET = Set.of(EVERYONE_MODULE);

    /**

     * The holder of data structures to support readability, exports, and

     * service use added at runtime with the reflective APIs.

     */

    private static class ReflectionData {

        /**

         * A module (1st key) reads another module (2nd key)

         */

        static final WeakPairMap<Module, Module, Boolean> reads =

            new WeakPairMap<>();

        /**

         * A module (1st key) exports or opens a package to another module

         * (2nd key). The map value is a map of package name to a boolean

         * that indicates if the package is opened.

         */

        static final WeakPairMap<Module, Module, Map<String, Boolean>> exports =

            new WeakPairMap<>();

        /**

         * A module (1st key) uses a service (2nd key)

         */

        static final WeakPairMap<Module, Class<?>, Boolean> uses =

            new WeakPairMap<>();

    }

    // -- readability --

    // the modules that this module reads

    private volatile Set<Module> reads;

    /**

     * Indicates if this module reads the given module. This method returns

     * {@code true} if invoked to test if this module reads itself. It also

     * returns {@code true} if invoked on an unnamed module (as unnamed

     * modules read all modules).

     *

     * @param  other

     *         The other module

     *

     * @return {@code true} if this module reads {@code other}

     *

     * @see #addReads(Module)

     */

    public boolean canRead(Module other) {

        Objects.requireNonNull(other);

        // an unnamed module reads all modules

        if (!this.isNamed())

            return true;

        // all modules read themselves

        if (other == this)

            return true;

        // check if this module reads other

        if (other.isNamed()) {

            Set<Module> reads = this.reads; // volatile read

            if (reads != null && reads.contains(other))

                return true;

        }

        // check if this module reads the other module reflectively

        if (ReflectionData.reads.containsKeyPair(this, other))

            return true;

        // if other is an unnamed module then check if this module reads

        // all unnamed modules

        if (!other.isNamed()

            && ReflectionData.reads.containsKeyPair(this, ALL_UNNAMED_MODULE))

            return true;

        return false;

    }

    /**

     * If the caller's module is this module then update this module to read

     * the given module.

     *

     * This method is a no-op if {@code other} is this module (all modules read

     * themselves), this module is an unnamed module (as unnamed modules read

     * all modules), or this module already reads {@code other}.

     *

     * @implNote <em>Read edges</em> added by this method are <em>weak</em> and

     * do not prevent {@code other} from being GC'ed when this module is

     * strongly reachable.

     *

     * @param  other

     *         The other module

     *

     * @return this module

     *

     * @throws IllegalCallerException

     *         If this is a named module and the caller's module is not this

     *         module

     *

     * @see #canRead

     */

    @CallerSensitive

    public Module addReads(Module other) {

        Objects.requireNonNull(other);

        if (this.isNamed()) {

            Module caller = getCallerModule(Reflection.getCallerClass());

            if (caller != this) {

                throw new IllegalCallerException(caller + " != " + this);

            }

            implAddReads(other, true);

        }

        return this;

    }

    /**

     * Updates this module to read another module.

     *

     * @apiNote Used by the --add-reads command line option.

     */

    void implAddReads(Module other) {

        implAddReads(other, true);

    }

    /**

     * Updates this module to read all unnamed modules.

     *

     * @apiNote Used by the --add-reads command line option.

     */

    void implAddReadsAllUnnamed() {

        implAddReads(Module.ALL_UNNAMED_MODULE, true);

    }

    /**

     * Updates this module to read another module without notifying the VM.

     *

     * @apiNote This method is for VM white-box testing.

     */

    void implAddReadsNoSync(Module other) {

        implAddReads(other, false);

    }

    /**

     * Makes the given {@code Module} readable to this module.

     *

     * If {@code syncVM} is {@code true} then the VM is notified.

     */

    private void implAddReads(Module other, boolean syncVM) {

        Objects.requireNonNull(other);

        if (!canRead(other)) {

            // update VM first, just in case it fails

            if (syncVM) {

                if (other == ALL_UNNAMED_MODULE) {

                    addReads0(this, null);

                } else {

                    addReads0(this, other);

                }

            }

            // add reflective read

            ReflectionData.reads.putIfAbsent(this, other, Boolean.TRUE);

        }

    }

    // -- exported and open packages --

    // the packages are open to other modules, can be null

    // if the value contains EVERYONE_MODULE then the package is open to all

    private volatile Map<String, Set<Module>> openPackages;

    // the packages that are exported, can be null

    // if the value contains EVERYONE_MODULE then the package is exported to all

    private volatile Map<String, Set<Module>> exportedPackages;

    /**

     * Returns {@code true} if this module exports the given package to at

     * least the given module.

     *

     * <p> This method returns {@code true} if invoked to test if a package in

     * this module is exported to itself. It always returns {@code true} when

     * invoked on an unnamed module. A package that is {@link #isOpen open} to

     * the given module is considered exported to that module at run-time and

     * so this method returns {@code true} if the package is open to the given

     * module. </p>

     *

     * <p> This method does not check if the given module reads this module. </p>

     *

     * @param  pn

     *         The package name

     * @param  other

     *         The other module

     *

     * @return {@code true} if this module exports the package to at least the

     *         given module

     *

     * @see ModuleDescriptor#exports()

     * @see #addExports(String,Module)

     */

    public boolean isExported(String pn, Module other) {

        Objects.requireNonNull(pn);

        Objects.requireNonNull(other);

        return implIsExportedOrOpen(pn, other, /*open*/false);

    }

    /**

     * Returns {@code true} if this module has <em>opened</em> a package to at

     * least the given module.

     *

     * <p> This method returns {@code true} if invoked to test if a package in

     * this module is open to itself. It returns {@code true} when invoked on an

     * {@link ModuleDescriptor#isOpen open} module with a package in the module.

     * It always returns {@code true} when invoked on an unnamed module. </p>

     *

     * <p> This method does not check if the given module reads this module. </p>

     *

     * @param  pn

     *         The package name

     * @param  other

     *         The other module

     *

     * @return {@code true} if this module has <em>opened</em> the package

     *         to at least the given module

     *

     * @see ModuleDescriptor#opens()

     * @see #addOpens(String,Module)

     * @see java.lang.reflect.AccessibleObject#setAccessible(boolean)

     * @see java.lang.invoke.MethodHandles#privateLookupIn

     */

    public boolean isOpen(String pn, Module other) {

        Objects.requireNonNull(pn);

        Objects.requireNonNull(other);

        return implIsExportedOrOpen(pn, other, /*open*/true);

    }

    /**

     * Returns {@code true} if this module exports the given package

     * unconditionally.

     *

     * <p> This method always returns {@code true} when invoked on an unnamed

     * module. A package that is {@link #isOpen(String) opened} unconditionally

     * is considered exported unconditionally at run-time and so this method

     * returns {@code true} if the package is opened unconditionally. </p>

     *

     * <p> This method does not check if the given module reads this module. </p>

     *

     * @param  pn

     *         The package name

     *

     * @return {@code true} if this module exports the package unconditionally

     *

     * @see ModuleDescriptor#exports()

     */

    public boolean isExported(String pn) {

        Objects.requireNonNull(pn);

        return implIsExportedOrOpen(pn, EVERYONE_MODULE, /*open*/false);

    }

    /**

     * Returns {@code true} if this module has <em>opened</em> a package

     * unconditionally.

     *

     * <p> This method always returns {@code true} when invoked on an unnamed

     * module. Additionally, it always returns {@code true} when invoked on an

     * {@link ModuleDescriptor#isOpen open} module with a package in the

     * module. </p>

     *

     * <p> This method does not check if the given module reads this module. </p>

     *

     * @param  pn

     *         The package name