/*

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

 *

 */

#ifndef SHARE_CLASSFILE_SYMBOLTABLE_HPP

#define SHARE_CLASSFILE_SYMBOLTABLE_HPP

#include "memory/allocation.hpp"

#include "memory/padded.hpp"

#include "oops/symbol.hpp"

#include "utilities/tableStatistics.hpp"

class JavaThread;

// TempNewSymbol acts as a handle class in a handle/body idiom and is

// responsible for proper resource management of the body (which is a Symbol*).

// The body is resource managed by a reference counting scheme.

// TempNewSymbol can therefore be used to properly hold a newly created or referenced

// Symbol* temporarily in scope.

//

// Routines in SymbolTable will initialize the reference count of a Symbol* before

// it becomes "managed" by TempNewSymbol instances. As a handle class, TempNewSymbol

// needs to maintain proper reference counting in context of copy semantics.

//

// In SymbolTable, new_symbol() will create a Symbol* if not already in the

// symbol table and add to the symbol's reference count.

// probe() and lookup_only() will increment the refcount if symbol is found.

class TempNewSymbol : public StackObj {

  Symbol* _temp;

public:

  TempNewSymbol() : _temp(NULL) {}

  // Conversion from a Symbol* to a TempNewSymbol.

  // Does not increment the current reference count.

  TempNewSymbol(Symbol *s) : _temp(s) {}

  // Copy constructor increments reference count.

  TempNewSymbol(const TempNewSymbol& rhs) : _temp(rhs._temp) {

    if (_temp != NULL) {

      _temp->increment_refcount();

    }

  }

  // Assignment operator uses a c++ trick called copy and swap idiom.

  // rhs is passed by value so within the scope of this method it is a copy.

  // At method exit it contains the former value of _temp, triggering the correct refcount

  // decrement upon destruction.

  void operator=(TempNewSymbol rhs) {

    Symbol* tmp = rhs._temp;

    rhs._temp = _temp;

    _temp = tmp;

  }

  // Decrement reference counter so it can go away if it's unused

  ~TempNewSymbol() {

    if (_temp != NULL) {

      _temp->decrement_refcount();

    }

  }

  // Symbol* conversion operators

  Symbol* operator -> () const                   { return _temp; }

  bool    operator == (Symbol* o) const          { return _temp == o; }

  operator Symbol*()                             { return _temp; }

};

class CompactHashtableWriter;

class SerializeClosure;

class SymbolTableConfig;

class SymbolTableCreateEntry;

class constantPoolHandle;

class SymbolClosure;

class SymbolTable : public AllStatic {

  friend class VMStructs;

  friend class Symbol;

  friend class ClassFileParser;

  friend class SymbolTableConfig;

  friend class SymbolTableCreateEntry;

 private:

  static volatile bool _has_work;

  // Set if one bucket is out of balance due to hash algorithm deficiency

  static volatile bool _needs_rehashing;

  static void delete_symbol(Symbol* sym);

  static void grow(JavaThread* jt);

  static void clean_dead_entries(JavaThread* jt);

  static double get_load_factor();

  static void check_concurrent_work();

  static void item_added();

  static void item_removed();

  // For cleaning

  static void reset_has_items_to_clean();

  static void mark_has_items_to_clean();

  static bool has_items_to_clean();

  static Symbol* allocate_symbol(const char* name, int len, bool c_heap); // Assumes no characters larger than 0x7F

  static Symbol* do_lookup(const char* name, int len, uintx hash);

  static Symbol* do_add_if_needed(const char* name, int len, uintx hash, bool heap);

  // lookup only, won't add. Also calculate hash. Used by the ClassfileParser.

  static Symbol* lookup_only(const char* name, int len, unsigned int& hash);

  static Symbol* lookup_only_unicode(const jchar* name, int len, unsigned int& hash);

  // Adding elements

  static void new_symbols(ClassLoaderData* loader_data,

                          const constantPoolHandle& cp, int names_count,

                          const char** name, int* lengths,

                          int* cp_indices, unsigned int* hashValues);

  static Symbol* lookup_shared(const char* name, int len, unsigned int hash) NOT_CDS_RETURN_(NULL);

  static Symbol* lookup_dynamic(const char* name, int len, unsigned int hash);

  static Symbol* lookup_common(const char* name, int len, unsigned int hash);

  // Arena for permanent symbols (null class loader) that are never unloaded

  static Arena*  _arena;

  static Arena* arena() { return _arena; }  // called for statistics

  static void print_table_statistics(outputStream* st, const char* table_name);

  static void try_rehash_table();

  static bool do_rehash();

public:

  // The symbol table

  static size_t table_size();

  static TableStatistics get_table_statistics();

  enum {

    symbol_alloc_batch_size = 8,

    // Pick initial size based on java -version size measurements

    symbol_alloc_arena_size = 360*K // TODO (revisit)

  };

  static void create_table();

  static void do_concurrent_work(JavaThread* jt);

  static bool has_work() { return _has_work; }

  static void trigger_cleanup();

  // Probing

  // Needed for preloading classes in signatures when compiling.

  // Returns the symbol is already present in symbol table, otherwise

  // NULL.  NO ALLOCATION IS GUARANTEED!

  static Symbol* probe(const char* name, int len) {

    unsigned int ignore_hash;

    return lookup_only(name, len, ignore_hash);

  }

  static Symbol* probe_unicode(const jchar* name, int len) {

    unsigned int ignore_hash;

    return lookup_only_unicode(name, len, ignore_hash);

  }

  // Symbol lookup and create if not found.

  // jchar (UTF16) version of lookup

  static Symbol* new_symbol(const jchar* name, int len);

  // char (UTF8) versions

  static Symbol* new_symbol(const Symbol* sym, int begin, int end);

  static Symbol* new_symbol(const char* utf8_buffer, int length);

  static Symbol* new_symbol(const char* name) {

    return new_symbol(name, (int)strlen(name));

  }

  // Create a symbol in the arena for symbols that are not deleted

  static Symbol* new_permanent_symbol(const char* name);

  // Rehash the string table if it gets out of balance

  static void rehash_table();

  static bool needs_rehashing() { return _needs_rehashing; }

  static inline void update_needs_rehash(bool rehash) {

    if (rehash) {

      _needs_rehashing = true;

    }

  }

  // Heap dumper and CDS

  static void symbols_do(SymbolClosure *cl);

  // Sharing

private:

  static void copy_shared_symbol_table(CompactHashtableWriter* ch_table);

public:

  static size_t estimate_size_for_archive() NOT_CDS_RETURN_(0);

  static void write_to_archive(bool is_static_archive = true) NOT_CDS_RETURN;

  static void serialize_shared_table_header(SerializeClosure* soc,

                                            bool is_static_archive = true) NOT_CDS_RETURN;

  static void metaspace_pointers_do(MetaspaceClosure* it);

  // Jcmd

  static void dump(outputStream* st, bool verbose=false);

  // Debugging

  static void verify();

  // Histogram

  static void print_histogram() PRODUCT_RETURN;

};

#endif // SHARE_CLASSFILE_SYMBOLTABLE_HPP