/*

 * Copyright (c) 2008, 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 "incls/_precompiled.incl"

# include "incls/_disassembler.cpp.incl"

void*       Disassembler::_library               = NULL;

bool        Disassembler::_tried_to_load_library = false;

// This routine is in the shared library:

Disassembler::decode_func Disassembler::_decode_instructions = NULL;

static const char hsdis_library_name[] = "hsdis-"HOTSPOT_LIB_ARCH;

static const char decode_instructions_name[] = "decode_instructions";

#define COMMENT_COLUMN  40 LP64_ONLY(+8) /*could be an option*/

#define BYTES_COMMENT   ";..."  /* funky byte display comment */

bool Disassembler::load_library() {

  if (_decode_instructions != NULL) {

    // Already succeeded.

    return true;

  }

  if (_tried_to_load_library) {

    // Do not try twice.

    // To force retry in debugger: assign _tried_to_load_library=0

    return false;

  }

  // Try to load it.

  char ebuf[1024];

  char buf[JVM_MAXPATHLEN];

  os::jvm_path(buf, sizeof(buf));

  int jvm_offset = -1;

  {

    // Match "jvm[^/]*" in jvm_path.

    const char* base = buf;

    const char* p = strrchr(buf, '/');

    p = strstr(p ? p : base, "jvm");

    if (p != NULL)  jvm_offset = p - base;

  }

  if (jvm_offset >= 0) {

    // Find the disassembler next to libjvm.so.

    strcpy(&buf[jvm_offset], hsdis_library_name);

    strcat(&buf[jvm_offset], os::dll_file_extension());

    _library = hpi::dll_load(buf, ebuf, sizeof ebuf);

  }

  if (_library == NULL) {

    // Try a free-floating lookup.

    strcpy(&buf[0], hsdis_library_name);

    strcat(&buf[0], os::dll_file_extension());

    _library = hpi::dll_load(buf, ebuf, sizeof ebuf);

  }

  if (_library != NULL) {

    _decode_instructions = CAST_TO_FN_PTR(Disassembler::decode_func,

                                          hpi::dll_lookup(_library, decode_instructions_name));

  }

  _tried_to_load_library = true;

  if (_decode_instructions == NULL) {

    tty->print_cr("Could not load %s; %s; %s", buf,

                  ((_library != NULL)

                   ? "entry point is missing"

                   : (WizardMode || PrintMiscellaneous)

                   ? (const char*)ebuf

                   : "library not loadable"),

                  "PrintAssembly is disabled");

    return false;

  }

  // Success.

  tty->print_cr("Loaded disassembler from %s", buf);

  return true;

}

class decode_env {

 private:

  nmethod*      _nm;

  CodeBlob*     _code;

  outputStream* _output;

  address       _start, _end;

  char          _option_buf[512];

  char          _print_raw;

  bool          _print_pc;

  bool          _print_bytes;

  address       _cur_insn;

  int           _total_ticks;

  int           _bytes_per_line; // arch-specific formatting option

  static bool match(const char* event, const char* tag) {

    size_t taglen = strlen(tag);

    if (strncmp(event, tag, taglen) != 0)

      return false;

    char delim = event[taglen];

    return delim == '\0' || delim == ' ' || delim == '/' || delim == '=';

  }

  void collect_options(const char* p) {

    if (p == NULL || p[0] == '\0')  return;

    size_t opt_so_far = strlen(_option_buf);

    if (opt_so_far + 1 + strlen(p) + 1 > sizeof(_option_buf))  return;

    char* fillp = &_option_buf[opt_so_far];

    if (opt_so_far > 0) *fillp++ = ',';

    strcat(fillp, p);

    // replace white space by commas:

    char* q = fillp;

    while ((q = strpbrk(q, " \t\n")) != NULL)

      *q++ = ',';

    // Note that multiple PrintAssemblyOptions flags accumulate with \n,

    // which we want to be changed to a comma...

  }

  void print_insn_labels();

  void print_insn_bytes(address pc0, address pc);

  void print_address(address value);

 public:

  decode_env(CodeBlob* code, outputStream* output);

  address decode_instructions(address start, address end);

  void start_insn(address pc) {

    _cur_insn = pc;

    output()->bol();

    print_insn_labels();

  }

  void end_insn(address pc) {

    address pc0 = cur_insn();

    outputStream* st = output();

    if (_print_bytes && pc > pc0)

      print_insn_bytes(pc0, pc);

    if (_nm != NULL) {

      _nm->print_code_comment_on(st, COMMENT_COLUMN, pc0, pc);

      // this calls reloc_string_for which calls oop::print_value_on

    }

    // Output pc bucket ticks if we have any

    if (total_ticks() != 0) {

      address bucket_pc = FlatProfiler::bucket_start_for(pc);

      if (bucket_pc != NULL && bucket_pc > pc0 && bucket_pc <= pc) {

        int bucket_count = FlatProfiler::bucket_count_for(pc0);

        if (bucket_count != 0) {

          st->bol();

          st->print_cr("%3.1f%% [%d]", bucket_count*100.0/total_ticks(), bucket_count);

        }

      }

    }

  }

  address handle_event(const char* event, address arg);

  outputStream* output() { return _output; }

  address cur_insn() { return _cur_insn; }

  int total_ticks() { return _total_ticks; }

  void set_total_ticks(int n) { _total_ticks = n; }

  const char* options() { return _option_buf; }

};

decode_env::decode_env(CodeBlob* code, outputStream* output) {

  memset(this, 0, sizeof(*this));

  _output = output ? output : tty;

  _code = code;

  if (code != NULL && code->is_nmethod())

    _nm = (nmethod*) code;

  // by default, output pc but not bytes:

  _print_pc       = true;

  _print_bytes    = false;

  _bytes_per_line = Disassembler::pd_instruction_alignment();

  // parse the global option string:

  collect_options(Disassembler::pd_cpu_opts());

  collect_options(PrintAssemblyOptions);

  if (strstr(options(), "hsdis-")) {

    if (strstr(options(), "hsdis-print-raw"))

      _print_raw = (strstr(options(), "xml") ? 2 : 1);

    if (strstr(options(), "hsdis-print-pc"))

      _print_pc = !_print_pc;

    if (strstr(options(), "hsdis-print-bytes"))

      _print_bytes = !_print_bytes;

  }

  if (strstr(options(), "help")) {

    tty->print_cr("PrintAssemblyOptions help:");

    tty->print_cr("  hsdis-print-raw       test plugin by requesting raw output");

    tty->print_cr("  hsdis-print-raw-xml   test plugin by requesting raw xml");

    tty->print_cr("  hsdis-print-pc        turn off PC printing (on by default)");

    tty->print_cr("  hsdis-print-bytes     turn on instruction byte output");

    tty->print_cr("combined options: %s", options());

  }

}

address decode_env::handle_event(const char* event, address arg) {

  if (match(event, "insn")) {

    start_insn(arg);

  } else if (match(event, "/insn")) {

    end_insn(arg);

  } else if (match(event, "addr")) {

    if (arg != NULL) {

      print_address(arg);

      return arg;

    }

  } else if (match(event, "mach")) {

   output()->print_cr("[Disassembling for mach='%s']", arg);

  } else if (match(event, "format bytes-per-line")) {

    _bytes_per_line = (int) (intptr_t) arg;

  } else {

    // ignore unrecognized markup

  }

  return NULL;

}

// called by the disassembler to print out jump targets and data addresses

void decode_env::print_address(address adr) {

  outputStream* st = _output;

  if (adr == NULL) {

    st->print("NULL");

    return;

  }

  int small_num = (int)(intptr_t)adr;

  if ((intptr_t)adr == (intptr_t)small_num

      && -1 <= small_num && small_num <= 9) {

    st->print("%d", small_num);

    return;

  }

  if (Universe::is_fully_initialized()) {

    if (StubRoutines::contains(adr)) {

      StubCodeDesc* desc = StubCodeDesc::desc_for(adr);

      if (desc == NULL)

        desc = StubCodeDesc::desc_for(adr + frame::pc_return_offset);

      if (desc != NULL) {

        st->print("Stub::%s", desc->name());

        if (desc->begin() != adr)

          st->print("%+d 0x%p",adr - desc->begin(), adr);

        else if (WizardMode) st->print(" " INTPTR_FORMAT, adr);

        return;

      }

      st->print("Stub::<unknown> " INTPTR_FORMAT, adr);

      return;

    }

    BarrierSet* bs = Universe::heap()->barrier_set();

    if (bs->kind() == BarrierSet::CardTableModRef &&

        adr == (address)((CardTableModRefBS*)(bs))->byte_map_base) {

      st->print("word_map_base");

      if (WizardMode) st->print(" " INTPTR_FORMAT, (intptr_t)adr);

      return;

    }

    oop obj;

    if (_nm != NULL

        && (obj = _nm->embeddedOop_at(cur_insn())) != NULL

        && (address) obj == adr

        && Universe::heap()->is_in(obj)

        && Universe::heap()->is_in(obj->klass())) {

      julong c = st->count();

      obj->print_value_on(st);

      if (st->count() == c) {

        // No output.  (Can happen in product builds.)

        st->print("(a %s)", Klass::cast(obj->klass())->external_name());

      }

      return;

    }

  }

  // Fall through to a simple numeral.

  st->print(INTPTR_FORMAT, (intptr_t)adr);

}

void decode_env::print_insn_labels() {

  address p = cur_insn();

  outputStream* st = output();

  CodeBlob* cb = _code;

  if (cb != NULL) {

    cb->print_block_comment(st, p);

  }

  if (_print_pc) {

    st->print("  " INTPTR_FORMAT ": ", (intptr_t) p);

  }

}

void decode_env::print_insn_bytes(address pc, address pc_limit) {

  outputStream* st = output();

  size_t incr = 1;

  size_t perline = _bytes_per_line;

  if ((size_t) Disassembler::pd_instruction_alignment() >= sizeof(int)

      && !((uintptr_t)pc % sizeof(int))

      && !((uintptr_t)pc_limit % sizeof(int))) {

    incr = sizeof(int);

    if (perline % incr)  perline += incr - (perline % incr);

  }

  while (pc < pc_limit) {

    // tab to the desired column:

    st->move_to(COMMENT_COLUMN);

    address pc0 = pc;

    address pc1 = pc + perline;

    if (pc1 > pc_limit)  pc1 = pc_limit;

    for (; pc < pc1; pc += incr) {

      if (pc == pc0)

        st->print(BYTES_COMMENT);

      else if ((uint)(pc - pc0) % sizeof(int) == 0)

        st->print(" ");         // put out a space on word boundaries

      if (incr == sizeof(int))

            st->print("%08lx", *(int*)pc);

      else  st->print("%02x",   (*pc)&0xFF);

    }

    st->cr();

  }

}

static void* event_to_env(void* env_pv, const char* event, void* arg) {

  decode_env* env = (decode_env*) env_pv;

  return env->handle_event(event, (address) arg);

}

static int printf_to_env(void* env_pv, const char* format, ...) {

  decode_env* env = (decode_env*) env_pv;

  outputStream* st = env->output();

  size_t flen = strlen(format);

  const char* raw = NULL;

  if (flen == 0)  return 0;

  if (flen == 1 && format[0] == '\n') { st->bol(); return 1; }

  if (flen < 2 ||

      strchr(format, '%') == NULL) {

    raw = format;

  } else if (format[0] == '%' && format[1] == '%' &&

             strchr(format+2, '%') == NULL) {

    // happens a lot on machines with names like %foo

    flen--;

    raw = format+1;

  }

  if (raw != NULL) {

    st->print_raw(raw, (int) flen);

    return (int) flen;

  }

  va_list ap;

  va_start(ap, format);

  julong cnt0 = st->count();

  st->vprint(format, ap);

  julong cnt1 = st->count();

  va_end(ap);

  return (int)(cnt1 - cnt0);

}

address decode_env::decode_instructions(address start, address end) {

  _start = start; _end = end;

  const int show_bytes = false; // for disassembler debugging

  //_version = Disassembler::pd_cpu_version();

  if (!Disassembler::can_decode()) {

    return NULL;

  }

  // decode a series of instructions and return the end of the last instruction

  if (_print_raw) {

    // Print whatever the library wants to print, w/o fancy callbacks.

    // This is mainly for debugging the library itself.

    FILE* out = stdout;

    FILE* xmlout = (_print_raw > 1 ? out : NULL);

    return (address)

      (*Disassembler::_decode_instructions)(start, end,

                                            NULL, (void*) xmlout,

                                            NULL, (void*) out,

                                            options());

  }

  return (address)

    (*Disassembler::_decode_instructions)(start, end,

                                          &event_to_env,  (void*) this,

                                          &printf_to_env, (void*) this,

                                          options());

}

void Disassembler::decode(CodeBlob* cb, outputStream* st) {

  if (!load_library())  return;

  decode_env env(cb, st);

  env.output()->print_cr("Decoding CodeBlob " INTPTR_FORMAT, cb);

  env.decode_instructions(cb->instructions_begin(), cb->instructions_end());

}

void Disassembler::decode(address start, address end, outputStream* st) {

  if (!load_library())  return;

  decode_env env(CodeCache::find_blob_unsafe(start), st);

  env.decode_instructions(start, end);

}

void Disassembler::decode(nmethod* nm, outputStream* st) {

  if (!load_library())  return;

  decode_env env(nm, st);

  env.output()->print_cr("Decoding compiled method " INTPTR_FORMAT ":", nm);

  env.output()->print_cr("Code:");

  unsigned char* p = nm->instructions_begin();

  unsigned char* end = nm->instructions_end();

  // If there has been profiling, print the buckets.

  if (FlatProfiler::bucket_start_for(p) != NULL) {

    unsigned char* p1 = p;

    int total_bucket_count = 0;

    while (p1 < end) {

      unsigned char* p0 = p1;

      p1 += pd_instruction_alignment();

      address bucket_pc = FlatProfiler::bucket_start_for(p1);

      if (bucket_pc != NULL && bucket_pc > p0 && bucket_pc <= p1)

        total_bucket_count += FlatProfiler::bucket_count_for(p0);

    }

    env.set_total_ticks(total_bucket_count);

  }

  env.decode_instructions(p, end);

}