/*

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

*

*/

#include "precompiled.hpp"

#include "classfile/classFileParser.hpp"

#include "classfile/classLoader.hpp"

#include "classfile/classLoaderData.inline.hpp"

#include "classfile/javaAssertions.hpp"

#include "classfile/javaClasses.hpp"

#include "classfile/javaClasses.inline.hpp"

#include "classfile/moduleEntry.hpp"

#include "classfile/modules.hpp"

#include "classfile/packageEntry.hpp"

#include "classfile/stringTable.hpp"

#include "classfile/symbolTable.hpp"

#include "classfile/vmSymbols.hpp"

#include "logging/log.hpp"

#include "memory/resourceArea.hpp"

#include "oops/instanceKlass.hpp"

#include "oops/objArrayKlass.hpp"

#include "oops/objArrayOop.inline.hpp"

#include "runtime/arguments.hpp"

#include "runtime/handles.inline.hpp"

#include "runtime/javaCalls.hpp"

#include "runtime/reflection.hpp"

#include "utilities/utf8.hpp"

static bool verify_module_name(char *module_name) {

  if (module_name == NULL) return false;

  int len = (int)strlen(module_name);

  return (len > 0 && len <= Symbol::max_length() &&

    UTF8::is_legal_utf8((unsigned char *)module_name, len, false) &&

    ClassFileParser::verify_unqualified_name(module_name, len,

    ClassFileParser::LegalModule));

}

bool Modules::verify_package_name(char *package_name) {

  if (package_name == NULL) return false;

  int len = (int)strlen(package_name);

  return (len > 0 && len <= Symbol::max_length() &&

    UTF8::is_legal_utf8((unsigned char *)package_name, len, false) &&

    ClassFileParser::verify_unqualified_name(package_name, len,

    ClassFileParser::LegalClass));

}

static char* get_module_name(oop module, TRAPS) {

  oop name_oop = java_lang_reflect_Module::name(module);

  if (name_oop == NULL) {

    THROW_MSG_NULL(vmSymbols::java_lang_NullPointerException(), "Null module name");

  }

  char* module_name = java_lang_String::as_utf8_string(name_oop);

  if (!verify_module_name(module_name)) {

    THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),

                   err_msg("Invalid module name: %s",

                           module_name != NULL ? module_name : "NULL"));

  }

  return module_name;

}

static const char* get_module_version(jstring version) {

  if (version == NULL) {

    return NULL;

  }

  return java_lang_String::as_utf8_string(JNIHandles::resolve_non_null(version));

}

static ModuleEntryTable* get_module_entry_table(Handle h_loader, TRAPS) {

  // This code can be called during start-up, before the classLoader's classLoader data got

  // created.  So, call register_loader() to make sure the classLoader data gets created.

  ClassLoaderData *loader_cld = SystemDictionary::register_loader(h_loader, CHECK_NULL);

  return loader_cld->modules();

}

static PackageEntryTable* get_package_entry_table(Handle h_loader, TRAPS) {

  // This code can be called during start-up, before the classLoader's classLoader data got

  // created.  So, call register_loader() to make sure the classLoader data gets created.

  ClassLoaderData *loader_cld = SystemDictionary::register_loader(h_loader, CHECK_NULL);

  return loader_cld->packages();

}

static ModuleEntry* get_module_entry(jobject module, TRAPS) {

  Handle module_h(THREAD, JNIHandles::resolve(module));

  if (!java_lang_reflect_Module::is_instance(module_h())) {

    THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),

                   "module is not an instance of type java.lang.reflect.Module");

  }

  return java_lang_reflect_Module::module_entry(module_h(), CHECK_NULL);

}

static PackageEntry* get_package_entry(ModuleEntry* module_entry, jstring package, TRAPS) {

  ResourceMark rm(THREAD);

  if (package == NULL) return NULL;

  const char *package_name = java_lang_String::as_utf8_string(JNIHandles::resolve_non_null(package));

  if (package_name == NULL) return NULL;

  TempNewSymbol pkg_symbol = SymbolTable::new_symbol(package_name, CHECK_NULL);

  PackageEntryTable* package_entry_table = module_entry->loader_data()->packages();

  assert(package_entry_table != NULL, "Unexpected null package entry table");

  return package_entry_table->lookup_only(pkg_symbol);

}

static PackageEntry* get_package_entry_by_name(Symbol* package,

                                               Handle h_loader,

                                               TRAPS) {

  if (package != NULL) {

    ResourceMark rm(THREAD);

    if (Modules::verify_package_name(package->as_C_string())) {

      PackageEntryTable* const package_entry_table =

        get_package_entry_table(h_loader, CHECK_NULL);

      assert(package_entry_table != NULL, "Unexpected null package entry table");

      return package_entry_table->lookup_only(package);

    }

  }

  return NULL;

}

bool Modules::is_package_defined(Symbol* package, Handle h_loader, TRAPS) {

  PackageEntry* res = get_package_entry_by_name(package, h_loader, CHECK_false);

  return res != NULL;

}

static void define_javabase_module(jobject module, jstring version,

                                   jstring location, jobjectArray packages, TRAPS) {

  ResourceMark rm(THREAD);

  Handle module_handle(THREAD, JNIHandles::resolve(module));

  // Obtain java.base's module version

  const char* module_version = get_module_version(version);

  TempNewSymbol version_symbol;

  if (module_version != NULL) {

    version_symbol = SymbolTable::new_symbol(module_version, CHECK);

  } else {

    version_symbol = NULL;

  }

  // Obtain java.base's location

  const char* module_location = NULL;

  TempNewSymbol location_symbol = NULL;

  if (location != NULL) {

    module_location =

      java_lang_String::as_utf8_string(JNIHandles::resolve_non_null(location));

    if (module_location != NULL) {

      location_symbol = SymbolTable::new_symbol(module_location, CHECK);

    }

  }

  objArrayOop packages_oop = objArrayOop(JNIHandles::resolve(packages));

  objArrayHandle packages_h(THREAD, packages_oop);

  int num_packages = (packages_h == NULL ? 0 : packages_h->length());

  // Check that the list of packages has no duplicates and that the

  // packages are syntactically ok.

  GrowableArray<Symbol*>* pkg_list = new GrowableArray<Symbol*>(num_packages);

  for (int x = 0; x < num_packages; x++) {

    oop string_obj = packages_h->obj_at(x);

    if (string_obj == NULL || !string_obj->is_a(SystemDictionary::String_klass())) {

      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

                "Bad package name for module: java.base");

    }

    char *package_name = java_lang_String::as_utf8_string(string_obj);

    if (!Modules::verify_package_name(package_name)) {

      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

                err_msg("Invalid package name: %s for module: java.base", package_name));

    }

    Symbol* pkg_symbol = SymbolTable::new_symbol(package_name, CHECK);

    // append_if_missing() returns FALSE if entry already exists.

    if (!pkg_list->append_if_missing(pkg_symbol)) {

      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

                err_msg("Duplicate package name: %s for module java.base",

                        package_name));

    }

  }

  // Validate java_base's loader is the boot loader.

  oop loader = java_lang_reflect_Module::loader(module_handle());

  if (loader != NULL) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "Class loader must be the boot class loader");

  }

  Handle h_loader = Handle(THREAD, loader);

  // Ensure the boot loader's PackageEntryTable has been created

  PackageEntryTable* package_table = get_package_entry_table(h_loader, CHECK);

  assert(pkg_list->length() == 0 || package_table != NULL, "Bad package_table");

  // Ensure java.base's ModuleEntry has been created

  assert(ModuleEntryTable::javabase_moduleEntry() != NULL, "No ModuleEntry for java.base");

  bool duplicate_javabase = false;

  {

    MutexLocker m1(Module_lock, THREAD);

    if (ModuleEntryTable::javabase_defined()) {

      duplicate_javabase = true;

    } else {

      // Verify that all java.base packages created during bootstrapping are in

      // pkg_list.  If any are not in pkg_list, than a non-java.base class was

      // loaded erroneously pre java.base module definition.

      package_table->verify_javabase_packages(pkg_list);

      // loop through and add any new packages for java.base

      PackageEntry* pkg;

      for (int x = 0; x < pkg_list->length(); x++) {

        // Some of java.base's packages were added early in bootstrapping, ignore duplicates.

        if (package_table->lookup_only(pkg_list->at(x)) == NULL) {

          pkg = package_table->locked_create_entry_or_null(pkg_list->at(x), ModuleEntryTable::javabase_moduleEntry());

          assert(pkg != NULL, "Unable to create a java.base package entry");

        }

        // Unable to have a GrowableArray of TempNewSymbol.  Must decrement the refcount of

        // the Symbol* that was created above for each package. The refcount was incremented

        // by SymbolTable::new_symbol and as well by the PackageEntry creation.

        pkg_list->at(x)->decrement_refcount();

      }

      // Finish defining java.base's ModuleEntry

      ModuleEntryTable::finalize_javabase(module_handle, version_symbol, location_symbol);

    }

  }

  if (duplicate_javabase) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "Module java.base is already defined");

  }

  // Only the thread that actually defined the base module will get here,

  // so no locking is needed.

  // Patch any previously loaded class's module field with java.base's java.lang.reflect.Module.

  ModuleEntryTable::patch_javabase_entries(module_handle);

  log_debug(modules)("define_javabase_module(): Definition of module: java.base,"

                     " version: %s, location: %s, package #: %d",

                     module_version != NULL ? module_version : "NULL",

                     module_location != NULL ? module_location : "NULL",

                     pkg_list->length());

  // packages defined to java.base

  for (int x = 0; x < pkg_list->length(); x++) {

    log_trace(modules)("define_javabase_module(): creation of package %s for module java.base",

                       (pkg_list->at(x))->as_C_string());

  }

}

void Modules::define_module(jobject module, jstring version,

                            jstring location, jobjectArray packages, TRAPS) {

  ResourceMark rm(THREAD);

  if (module == NULL) {

    THROW_MSG(vmSymbols::java_lang_NullPointerException(), "Null module object");

  }

  Handle module_handle(THREAD, JNIHandles::resolve(module));

  if (!java_lang_reflect_Module::is_instance(module_handle())) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "module is not an instance of type java.lang.reflect.Module");

  }

  char* module_name = get_module_name(module_handle(), CHECK);

  if (module_name == NULL) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "Module name cannot be null");

  }

  // Special handling of java.base definition

  if (strcmp(module_name, "java.base") == 0) {

    define_javabase_module(module, version, location, packages, CHECK);

    return;

  }

  const char* module_version = get_module_version(version);

  objArrayOop packages_oop = objArrayOop(JNIHandles::resolve(packages));

  objArrayHandle packages_h(THREAD, packages_oop);

  int num_packages = (packages_h == NULL ? 0 : packages_h->length());

  // Check that the list of packages has no duplicates and that the

  // packages are syntactically ok.

  GrowableArray<Symbol*>* pkg_list = new GrowableArray<Symbol*>(num_packages);

  for (int x = 0; x < num_packages; x++) {

    oop string_obj = packages_h->obj_at(x);

    if (string_obj == NULL || !string_obj->is_a(SystemDictionary::String_klass())) {

      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

                err_msg("Bad package name for module: %s", module_name));

    }

    char *package_name = java_lang_String::as_utf8_string(string_obj);

    if (!verify_package_name(package_name)) {

      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

                err_msg("Invalid package name: %s for module: %s",

                        package_name, module_name));

    }

    Symbol* pkg_symbol = SymbolTable::new_symbol(package_name, CHECK);

    // append_if_missing() returns FALSE if entry already exists.

    if (!pkg_list->append_if_missing(pkg_symbol)) {

      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

                err_msg("Duplicate package name: %s for module %s",

                        package_name, module_name));

    }

  }

  ModuleEntryTable* module_table = get_module_entry_table(h_loader, CHECK);

  assert(module_table != NULL, "module entry table shouldn't be null");

  // Create symbol* entry for module name.

  TempNewSymbol module_symbol = SymbolTable::new_symbol(module_name, CHECK);

  int dupl_pkg_index = -1;

  bool dupl_modules = false;

  // Create symbol* entry for module version.

  TempNewSymbol version_symbol;

  if (module_version != NULL) {

    version_symbol = SymbolTable::new_symbol(module_version, CHECK);

  } else {

    version_symbol = NULL;

  }

  // Create symbol* entry for module location.

  const char* module_location = NULL;

  TempNewSymbol location_symbol = NULL;

  if (location != NULL) {

    module_location =

      java_lang_String::as_utf8_string(JNIHandles::resolve_non_null(location));

    if (module_location != NULL) {

      location_symbol = SymbolTable::new_symbol(module_location, CHECK);

    }

  }

  ClassLoaderData* loader_data = ClassLoaderData::class_loader_data_or_null(h_loader());

  assert(loader_data != NULL, "class loader data shouldn't be null");

  PackageEntryTable* package_table = NULL;

  {

    MutexLocker ml(Module_lock, THREAD);

    if (num_packages > 0) {

      package_table = get_package_entry_table(h_loader, CHECK);

      assert(package_table != NULL, "Missing package_table");

      // Check that none of the packages exist in the class loader's package table.

      for (int x = 0; x < pkg_list->length(); x++) {

        if (package_table->lookup_only(pkg_list->at(x)) != NULL) {

          // This could be because the module was already defined.  If so,

          // report that error instead of the package error.

          if (module_table->lookup_only(module_symbol) != NULL) {

            dupl_modules = true;

          } else {

            dupl_pkg_index = x;

          }

          break;

        }

      }

    }  // if (num_packages > 0)...

    // Add the module and its packages.

    if (!dupl_modules && dupl_pkg_index == -1) {

      // Create the entry for this module in the class loader's module entry table.

      ModuleEntry* module_entry = module_table->locked_create_entry_or_null(module_handle, module_symbol,

                                    version_symbol, location_symbol, loader_data);

      if (module_entry == NULL) {

        dupl_modules = true;

      } else {

        // Add the packages.

        assert(pkg_list->length() == 0 || package_table != NULL, "Bad package table");

        PackageEntry* pkg;

        for (int y = 0; y < pkg_list->length(); y++) {

          pkg = package_table->locked_create_entry_or_null(pkg_list->at(y), module_entry);

          assert(pkg != NULL, "Unable to create a module's package entry");

          // Unable to have a GrowableArray of TempNewSymbol.  Must decrement the refcount of

          // the Symbol* that was created above for each package. The refcount was incremented

          // by SymbolTable::new_symbol and as well by the PackageEntry creation.

          pkg_list->at(y)->decrement_refcount();

        }

        // Store pointer to ModuleEntry record in java.lang.reflect.Module object.

        java_lang_reflect_Module::set_module_entry(module_handle(), module_entry);

      }

    }

  }  // Release the lock

  // any errors ?

  if (dupl_modules) {

     THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

               err_msg("Module %s is already defined", module_name));

  }

  if (dupl_pkg_index != -1) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              err_msg("Package %s for module %s already exists for class loader",

                      pkg_list->at(dupl_pkg_index)->as_C_string(), module_name));

  }

  if (log_is_enabled(Debug, modules)) {

    outputStream* logst = Log(modules)::debug_stream();

    logst->print("define_module(): creation of module: %s, version: %s, location: %s, ",

                 module_name, module_version != NULL ? module_version : "NULL",

                 module_location != NULL ? module_location : "NULL");

    loader_data->print_value_on(logst);

    logst->print_cr(", package #: %d", pkg_list->length());

    for (int y = 0; y < pkg_list->length(); y++) {

      log_trace(modules)("define_module(): creation of package %s for module %s",

                         (pkg_list->at(y))->as_C_string(), module_name);

    }

  }

  // If the module is defined to the boot loader and an exploded build is being

  // used, prepend <java.home>/modules/modules_name, if it exists, to the system boot class path.

  if (loader == NULL &&

      !Universe::is_module_initialized() &&

      !ClassLoader::has_jrt_entry()) {

    ClassLoader::add_to_exploded_build_list(module_symbol, CHECK);

  }

}

void Modules::set_bootloader_unnamed_module(jobject module, TRAPS) {

  ResourceMark rm(THREAD);

  if (module == NULL) {

    THROW_MSG(vmSymbols::java_lang_NullPointerException(), "Null module object");

  }

  Handle module_handle(THREAD, JNIHandles::resolve(module));

  if (!java_lang_reflect_Module::is_instance(module_handle())) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "module is not an instance of type java.lang.reflect.Module");

  }

  // Ensure that this is an unnamed module

  oop name = java_lang_reflect_Module::name(module_handle());

  if (name != NULL) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "boot loader's unnamed module's java.lang.reflect.Module has a name");

  }

  // Validate java_base's loader is the boot loader.

  oop loader = java_lang_reflect_Module::loader(module_handle());

  if (loader != NULL) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "Class loader must be the boot class loader");

  }

  Handle h_loader = Handle(THREAD, loader);

  log_debug(modules)("set_bootloader_unnamed_module(): recording unnamed module for boot loader");

  // Ensure the boot loader's PackageEntryTable has been created

  ModuleEntryTable* module_table = get_module_entry_table(h_loader, CHECK);

  // Set java.lang.reflect.Module for the boot loader's unnamed module

  ModuleEntry* unnamed_module = module_table->unnamed_module();

  assert(unnamed_module != NULL, "boot loader's unnamed ModuleEntry not defined");

  unnamed_module->set_module(ClassLoaderData::the_null_class_loader_data()->add_handle(module_handle));

  // Store pointer to the ModuleEntry in the unnamed module's java.lang.reflect.Module object.

  java_lang_reflect_Module::set_module_entry(module_handle(), unnamed_module);

}

void Modules::add_module_exports(jobject from_module, jstring package, jobject to_module, TRAPS) {

  if (package == NULL) {

    THROW_MSG(vmSymbols::java_lang_NullPointerException(),

              "package is null");

  }

  if (from_module == NULL) {

    THROW_MSG(vmSymbols::java_lang_NullPointerException(),

              "from_module is null");

  }

  ModuleEntry* from_module_entry = get_module_entry(from_module, CHECK);

  if (from_module_entry == NULL) {

    THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),

              "from_module cannot be found");

  }

  // All packages in unnamed are exported by default.

  if (!from_module_entry->is_named()) return;