/*

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

 *

 * 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/altHashing.hpp"

#include "classfile/compactHashtable.inline.hpp"

#include "classfile/javaClasses.inline.hpp"

#include "classfile/stringTable.hpp"

#include "classfile/systemDictionary.hpp"

#include "gc/shared/collectedHeap.hpp"

#include "gc/shared/oopStorage.inline.hpp"

#include "gc/shared/oopStorageParState.inline.hpp"

#include "logging/log.hpp"

#include "logging/logStream.hpp"

#include "memory/allocation.inline.hpp"

#include "memory/filemap.hpp"

#include "memory/metaspaceShared.hpp"

#include "memory/resourceArea.hpp"

#include "memory/universe.hpp"

#include "oops/access.inline.hpp"

#include "oops/oop.inline.hpp"

#include "oops/typeArrayOop.inline.hpp"

#include "oops/weakHandle.inline.hpp"

#include "runtime/atomic.hpp"

#include "runtime/handles.inline.hpp"

#include "runtime/mutexLocker.hpp"

#include "runtime/safepointVerifiers.hpp"

#include "runtime/timerTrace.hpp"

#include "runtime/interfaceSupport.inline.hpp"

#include "services/diagnosticCommand.hpp"

#include "utilities/concurrentHashTable.inline.hpp"

#include "utilities/concurrentHashTableTasks.inline.hpp"

#include "utilities/macros.hpp"

// We prefer short chains of avg 2

#define PREF_AVG_LIST_LEN   2

// 2^24 is max size

#define END_SIZE           24

// If a chain gets to 32 something might be wrong

#define REHASH_LEN         32

// If we have as many dead items as 50% of the number of bucket

#define CLEAN_DEAD_HIGH_WATER_MARK 0.5

// --------------------------------------------------------------------------

StringTable* StringTable::_the_table = NULL;

bool StringTable::_shared_string_mapped = false;

CompactHashtable<oop, char> StringTable::_shared_table;

bool StringTable::_alt_hash = false;

static juint murmur_seed = 0;

uintx hash_string(const jchar* s, int len, bool useAlt) {

  return  useAlt ?

    AltHashing::murmur3_32(murmur_seed, s, len) :

    java_lang_String::hash_code(s, len);

}

class StringTableConfig : public StringTableHash::BaseConfig {

 private:

 public:

  static uintx get_hash(WeakHandle<vm_string_table_data> const& value,

                        bool* is_dead) {

    EXCEPTION_MARK;

    oop val_oop = value.peek();

    if (val_oop == NULL) {

      *is_dead = true;

      return 0;

    }

    *is_dead = false;

    ResourceMark rm(THREAD);

    // All String oops are hashed as unicode

    int length;

    jchar* chars = java_lang_String::as_unicode_string(val_oop, length, THREAD);

    if (chars != NULL) {

      return hash_string(chars, length, StringTable::_alt_hash);

    }

    vm_exit_out_of_memory(length, OOM_MALLOC_ERROR, "get hash from oop");

    return 0;

  }

  // We use default allocation/deallocation but counted

  static void* allocate_node(size_t size,

                             WeakHandle<vm_string_table_data> const& value) {

    StringTable::item_added();

    return StringTableHash::BaseConfig::allocate_node(size, value);

  }

  static void free_node(void* memory,

                        WeakHandle<vm_string_table_data> const& value) {

    value.release();

    StringTableHash::BaseConfig::free_node(memory, value);

    StringTable::item_removed();

  }

};

class StringTableLookupJchar : StackObj {

 private:

  Thread* _thread;

  uintx _hash;

  int _len;

  const jchar* _str;

  Handle _found;

 public:

  StringTableLookupJchar(Thread* thread, uintx hash, const jchar* key, int len)

    : _thread(thread), _hash(hash), _str(key), _len(len) {

  }

  uintx get_hash() const {

    return _hash;

  }

  bool equals(WeakHandle<vm_string_table_data>* value, bool* is_dead) {

    oop val_oop = value->peek();

    if (val_oop == NULL) {

      // dead oop, mark this hash dead for cleaning

      *is_dead = true;

      return false;

    }

    bool equals = java_lang_String::equals(val_oop, (jchar*)_str, _len);

    if (!equals) {

      return false;

    }

    // Need to resolve weak handle and Handleize through possible safepoint.

     _found = Handle(_thread, value->resolve());

    return true;

  }

};

class StringTableLookupOop : public StackObj {

 private:

  Thread* _thread;

  uintx _hash;

  Handle _find;

  Handle _found;  // Might be a different oop with the same value that's already

                  // in the table, which is the point.

 public:

  StringTableLookupOop(Thread* thread, uintx hash, Handle handle)

    : _thread(thread), _hash(hash), _find(handle) { }

  uintx get_hash() const {

    return _hash;

  }

  bool equals(WeakHandle<vm_string_table_data>* value, bool* is_dead) {

    oop val_oop = value->peek();

    if (val_oop == NULL) {

      // dead oop, mark this hash dead for cleaning

      *is_dead = true;

      return false;

    }

    bool equals = java_lang_String::equals(_find(), val_oop);

    if (!equals) {

      return false;

    }

    // Need to resolve weak handle and Handleize through possible safepoint.

    _found = Handle(_thread, value->resolve());

    return true;

  }

};

static size_t ceil_pow_2(uintx val) {

  size_t ret;

  for (ret = 1; ((size_t)1 << ret) < val; ++ret);

  return ret;

}

StringTable::StringTable() : _local_table(NULL), _current_size(0), _has_work(0),

  _needs_rehashing(false), _weak_handles(NULL), _items(0), _uncleaned_items(0) {

  _weak_handles = new OopStorage("StringTable weak",

                                 StringTableWeakAlloc_lock,

                                 StringTableWeakActive_lock);

  size_t start_size_log_2 = ceil_pow_2(StringTableSize);

  _current_size = ((size_t)1) << start_size_log_2;

  log_trace(stringtable)("Start size: " SIZE_FORMAT " (" SIZE_FORMAT ")",

                         _current_size, start_size_log_2);

  _local_table = new StringTableHash(start_size_log_2, END_SIZE, REHASH_LEN);

}

size_t StringTable::item_added() {

  return Atomic::add((size_t)1, &(the_table()->_items));

}

  log_trace(stringtable)(

     "Uncleaned items:" SIZE_FORMAT " added: " SIZE_FORMAT " total:" SIZE_FORMAT,

  return total;

}

void StringTable::item_removed() {

  Atomic::add((size_t)-1, &(the_table()->_items));

}

double StringTable::get_load_factor() {

  return (_items*1.0)/_current_size;

}

double StringTable::get_dead_factor() {

  return (_uncleaned_items*1.0)/_current_size;

}

size_t StringTable::table_size(Thread* thread) {

  return ((size_t)(1)) << _local_table->get_size_log2(thread != NULL ? thread

                                                      : Thread::current());

}

void StringTable::trigger_concurrent_work() {

  MutexLockerEx ml(Service_lock, Mutex::_no_safepoint_check_flag);

  the_table()->_has_work = true;

  Service_lock->notify_all();

}

// Probing

oop StringTable::lookup(Symbol* symbol) {

  ResourceMark rm;

  int length;

  jchar* chars = symbol->as_unicode(length);

  return lookup(chars, length);

}

oop StringTable::lookup(jchar* name, int len) {

  unsigned int hash = java_lang_String::hash_code(name, len);

  oop string = StringTable::the_table()->lookup_shared(name, len, hash);

  if (string != NULL) {

    return string;

  }

  if (StringTable::_alt_hash) {

    hash = hash_string(name, len, true);

  }

  return StringTable::the_table()->do_lookup(name, len, hash);

}

class StringTableGet : public StackObj {

  Thread* _thread;

  Handle  _return;

 public:

  StringTableGet(Thread* thread) : _thread(thread) {}

  void operator()(WeakHandle<vm_string_table_data>* val) {

    oop result = val->resolve();

    assert(result != NULL, "Result should be reachable");

    _return = Handle(_thread, result);

  }

  oop get_res_oop() {

    return _return();

  }

};

oop StringTable::do_lookup(jchar* name, int len, uintx hash) {

  Thread* thread = Thread::current();

  StringTableLookupJchar lookup(thread, hash, name, len);

  StringTableGet stg(thread);

  bool rehash_warning;

  _local_table->get(thread, lookup, stg, &rehash_warning);

  if (rehash_warning) {

    _needs_rehashing = true;

  }

  return stg.get_res_oop();

}

// Interning

oop StringTable::intern(Symbol* symbol, TRAPS) {

  if (symbol == NULL) return NULL;

  ResourceMark rm(THREAD);

  int length;

  jchar* chars = symbol->as_unicode(length);

  Handle string;

  oop result = intern(string, chars, length, CHECK_NULL);

  return result;

}

oop StringTable::intern(oop string, TRAPS) {

  if (string == NULL) return NULL;

  ResourceMark rm(THREAD);

  int length;

  Handle h_string (THREAD, string);

  jchar* chars = java_lang_String::as_unicode_string(string, length,

                                                     CHECK_NULL);

  oop result = intern(h_string, chars, length, CHECK_NULL);

  return result;

}

oop StringTable::intern(const char* utf8_string, TRAPS) {

  if (utf8_string == NULL) return NULL;

  ResourceMark rm(THREAD);

  int length = UTF8::unicode_length(utf8_string);

  jchar* chars = NEW_RESOURCE_ARRAY(jchar, length);

  UTF8::convert_to_unicode(utf8_string, chars, length);

  Handle string;

  oop result = intern(string, chars, length, CHECK_NULL);

  return result;

}

oop StringTable::intern(Handle string_or_null_h, jchar* name, int len, TRAPS) {

  // shared table always uses java_lang_String::hash_code

  unsigned int hash = java_lang_String::hash_code(name, len);

  oop found_string = StringTable::the_table()->lookup_shared(name, len, hash);

  if (found_string != NULL) {

    return found_string;

  }

  if (StringTable::_alt_hash) {

    hash = hash_string(name, len, true);

  }

  return StringTable::the_table()->do_intern(string_or_null_h, name, len,

                                             hash, CHECK_NULL);

}

class StringTableCreateEntry : public StackObj {

 private:

   Thread* _thread;

   Handle  _return;

   Handle  _store;

 public:

  StringTableCreateEntry(Thread* thread, Handle store)

    : _thread(thread), _store(store) {}

  WeakHandle<vm_string_table_data> operator()() { // No dups found

    WeakHandle<vm_string_table_data> wh =

      WeakHandle<vm_string_table_data>::create(_store);

    return wh;

  }

  void operator()(bool inserted, WeakHandle<vm_string_table_data>* val) {

    oop result = val->resolve();

    assert(result != NULL, "Result should be reachable");

    _return = Handle(_thread, result);

  }

  oop get_return() const {

    return _return();

  }

};

oop StringTable::do_intern(Handle string_or_null_h, jchar* name,

                           int len, uintx hash, TRAPS) {

  HandleMark hm(THREAD);  // cleanup strings created

  Handle string_h;

  if (!string_or_null_h.is_null()) {

    string_h = string_or_null_h;

  } else {

    string_h = java_lang_String::create_from_unicode(name, len, CHECK_NULL);

  }

  // Deduplicate the string before it is interned. Note that we should never

  // deduplicate a string after it has been interned. Doing so will counteract

  // compiler optimizations done on e.g. interned string literals.

  Universe::heap()->deduplicate_string(string_h());

  assert(java_lang_String::equals(string_h(), name, len),

         "string must be properly initialized");

  assert(len == java_lang_String::length(string_h()), "Must be same length");

  StringTableLookupOop lookup(THREAD, hash, string_h);

  StringTableCreateEntry stc(THREAD, string_h);

  bool rehash_warning;

  _local_table->get_insert_lazy(THREAD, lookup, stc, stc, &rehash_warning);

  if (rehash_warning) {

    _needs_rehashing = true;

  }

  return stc.get_return();

}

// GC support

class StringTableIsAliveCounter : public BoolObjectClosure {

  BoolObjectClosure* _real_boc;

 public:

  size_t _count;

  size_t _count_total;

  StringTableIsAliveCounter(BoolObjectClosure* boc) : _real_boc(boc), _count(0),

                                                      _count_total(0) {}

  bool do_object_b(oop obj) {

    bool ret = _real_boc->do_object_b(obj);

    if (!ret) {

      ++_count;

    }

    ++_count_total;

    return ret;

  }

};

void StringTable::unlink_or_oops_do(BoolObjectClosure* is_alive, OopClosure* f,

                                    int* processed, int* removed) {

  DoNothingClosure dnc;

  assert(is_alive != NULL, "No closure");

  StringTableIsAliveCounter stiac(is_alive);

  OopClosure* tmp = f != NULL ? f : &dnc;

  StringTable::the_table()->_weak_handles->weak_oops_do(&stiac, tmp);

  StringTable::the_table()->check_concurrent_work();

  if (processed != NULL) {

    *processed = (int) stiac._count_total;

  }

  if (removed != NULL) {

    *removed = (int) stiac._count;

  }

}

void StringTable::oops_do(OopClosure* f) {

  assert(f != NULL, "No closure");

  StringTable::the_table()->_weak_handles->oops_do(f);

}

void StringTable::possibly_parallel_unlink(

   OopStorage::ParState<false, false>* _par_state_string, BoolObjectClosure* cl,

   int* processed, int* removed)

{

  DoNothingClosure dnc;

  assert(cl != NULL, "No closure");

  StringTableIsAliveCounter stiac(cl);

  _par_state_string->weak_oops_do(&stiac, &dnc);

  *processed = (int) stiac._count_total;

  *removed = (int) stiac._count;

}

void StringTable::possibly_parallel_oops_do(

   OopStorage::ParState<false /* concurrent */, false /* const */>*

   _par_state_string, OopClosure* f)

{

  assert(f != NULL, "No closure");

  _par_state_string->oops_do(f);

}

// Concurrent work

void StringTable::grow(JavaThread* jt) {

  StringTableHash::GrowTask gt(_local_table);

  if (!gt.prepare(jt)) {

    return;

  }

  log_trace(stringtable)("Started to grow");

  {

    TraceTime timer("Grow", TRACETIME_LOG(Debug, stringtable, perf));

    while (gt.doTask(jt)) {

      gt.pause(jt);

      {

        ThreadBlockInVM tbivm(jt);

      }

      gt.cont(jt);

    }

  }

  gt.done(jt);

  _current_size = table_size(jt);

  log_debug(stringtable)("Grown to size:" SIZE_FORMAT, _current_size);

}

struct StringTableDoDelete : StackObj {

  void operator()(WeakHandle<vm_string_table_data>* val) {

  }

};

struct StringTableDeleteCheck : StackObj {

  long _count;

  long _item;

  StringTableDeleteCheck() : _count(0), _item(0) {}

  bool operator()(WeakHandle<vm_string_table_data>* val) {

    ++_item;

    oop tmp = val->peek();

    if (tmp == NULL) {

      ++_count;

      return true;

    } else {

      return false;

    }

  }

};

void StringTable::clean_dead_entries(JavaThread* jt) {

  StringTableHash::BulkDeleteTask bdt(_local_table);