/*
 * Copyright (c) 2011, 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 "memory/oopFactory.hpp"
#include "runtime/javaCalls.hpp"
#include "runtime/mutexLocker.hpp"
#include "services/diagnosticArgument.hpp"
#include "services/diagnosticFramework.hpp"
#include "services/management.hpp"

CmdLine::CmdLine(const char* line, size_t len, bool no_command_name) {
    assert(line != NULL, "Command line string should not be NULL");
    const char* line_end;
    const char* cmd_end;

    _cmd = line;
    line_end = &line[len];

    // Skip whitespace in the beginning of the line.
    while (_cmd < line_end && isspace((int) _cmd[0])) {
      _cmd++;
    }
    cmd_end = _cmd;

    if(no_command_name) {
      _cmd = NULL;
      _cmd_len = 0;
    } else {
      // Look for end of the command name
      while (cmd_end < line_end && !isspace((int) cmd_end[0])) {
        cmd_end++;
      }
      _cmd_len = cmd_end - line;
    }
    _args = cmd_end;
    _args_len = len - _cmd_len;
}

bool DCmdArgIter::get_next(TRAPS) {
  if(_len == 0) return false;
  // skipping spaces
  while (_cursor < _len - 1 && isspace(_buffer[_cursor])) {
    _cursor++;
  }
  // handling end of command line
  if (_cursor >= _len - 1) {
    _cursor = _len - 1;
    _key_addr = &_buffer[_len - 1];
    _key_len = 0;
    _value_addr = &_buffer[_len - 1];
    _value_len = 0;
    return false;
  }
  // extracting first item, argument or option name
  _key_addr = &_buffer[_cursor];
  while (_cursor <= _len - 1 && _buffer[_cursor] != '=' && _buffer[_cursor] != _delim) {
    // argument can be surrounded by simple or double quotes
    if (_buffer[_cursor] == '\"' || _buffer[_cursor] == '\'') {
      _key_addr++;
      char quote = _buffer[_cursor];
      while (_cursor < _len - 1) {
        _cursor++;
        if (_buffer[_cursor] == quote && _buffer[_cursor - 1] != '\\') {
          break;
        }
      }
      if (_buffer[_cursor] != quote) {
        THROW_MSG_(vmSymbols::java_lang_IllegalArgumentException(),
                "Format error in diagnostic command arguments", false);
      }
      break;
    }
    _cursor++;
  }
  _key_len = &_buffer[_cursor] - _key_addr;
  // check if the argument has the <key>=<value> format
  if (_cursor <= _len -1 && _buffer[_cursor] == '=') {
    _cursor++;
    _value_addr = &_buffer[_cursor];
    // extract the value
    while (_cursor <= _len - 1 && _buffer[_cursor] != _delim) {
      // value can be surrounded by simple or double quotes
      if (_buffer[_cursor] == '\"' || _buffer[_cursor] == '\'') {
        _value_addr++;
        char quote = _buffer[_cursor];
        while (_cursor < _len - 1) {
          _cursor++;
          if (_buffer[_cursor] == quote && _buffer[_cursor - 1] != '\\') {
            break;
          }
        }
        if (_buffer[_cursor] != quote) {
          THROW_MSG_(vmSymbols::java_lang_IllegalArgumentException(),
                  "Format error in diagnostic command arguments", false);
        }
        break;
      }
      _cursor++;
    }
    _value_len = &_buffer[_cursor] - _value_addr;
  } else {
    _value_addr = NULL;
    _value_len = 0;
  }
  return _key_len != 0;
}

bool DCmdInfo::by_name(void* cmd_name, DCmdInfo* info) {
    if(info == NULL)
        return false;

    return strcmp((const char*)cmd_name, info->get_name()) == 0;
}

void DCmdParser::add_dcmd_option(GenDCmdArgument* arg) {
  assert(arg != NULL, "Sanity");
  if (_options == NULL) {
    _options = arg;
  } else {
    GenDCmdArgument* o = _options;
    while (o->get_next() != NULL) {
      o = o->get_next();
    }
    o->set_next(arg);
  }
  arg->set_next(NULL);
  Thread* THREAD = Thread::current();
  arg->init_value(THREAD);
  if (HAS_PENDING_EXCEPTION) {
    fatal("Initialization must be successful");
  }
}

void DCmdParser::add_dcmd_argument(GenDCmdArgument* arg) {
  assert(arg != NULL, "Sanity");
  if (_arguments == NULL) {
    _arguments = arg;
  } else {
    GenDCmdArgument* a = _arguments;
    while (a->get_next() != NULL) {
      a = a->get_next();
    }
    a->set_next(arg);
  }
  arg->set_next(NULL);
  Thread* THREAD = Thread::current();
  arg->init_value(THREAD);
  if (HAS_PENDING_EXCEPTION) {
    fatal("Initialization must be successful");
  }
}

void DCmdParser::parse(CmdLine* line, char delim, TRAPS) {
  GenDCmdArgument* next_argument = _arguments;
  DCmdArgIter iter(line->args_addr(), line->args_len(), delim);
  bool cont = iter.get_next(CHECK);
  while(cont) {
    GenDCmdArgument* arg = lookup_dcmd_option(iter.get_key_addr(),
            iter.get_key_length());
    if(arg != NULL) {
      arg->read_value(iter.get_value_addr(), iter.get_value_length(), CHECK);
    } else {
      if(next_argument != NULL) {
        arg = next_argument;
        arg->read_value(iter.get_key_addr(), iter.get_key_length(), CHECK);
        next_argument = next_argument->get_next();
      } else {
        THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),
          "Unknown argument in diagnostic command");
      }
    }
    cont = iter.get_next(CHECK);
  }
  check(CHECK);
}

GenDCmdArgument* DCmdParser::lookup_dcmd_option(const char* name, size_t len) {
  GenDCmdArgument* arg = _options;
  while (arg != NULL) {
    if (strlen(arg->get_name()) == len &&
      strncmp(name, arg->get_name(), len) == 0) {
      return arg;
    }
    arg = arg->get_next();
  }
  return NULL;
}

void DCmdParser::check(TRAPS) {
  GenDCmdArgument* arg = _arguments;
  while (arg != NULL) {
    if (arg->is_mandatory() && !arg->has_value()) {
      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),
              "Missing argument for diagnostic command");
    }
    arg = arg->get_next();
  }
  arg = _options;
  while (arg != NULL) {
    if (arg->is_mandatory() && !arg->has_value()) {
      THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),
              "Missing option for diagnostic command");
    }
    arg = arg->get_next();
  }
}

void DCmdParser::print_help(outputStream* out, const char* cmd_name) {
  out->print("\nSyntax : %s %s", cmd_name, _options == NULL ? "" : "[options]");
  GenDCmdArgument* arg = _arguments;
  while(arg != NULL) {
    if(arg->is_mandatory()) {
      out->print(" <%s>",arg->get_name());
    } else {
      out->print(" [<%s>]",arg->get_name());
    }
    arg = arg->get_next();
  }
  out->print_cr("");
  if(_arguments != NULL) {
    out->print_cr("\nArguments:");
    arg = _arguments;
    while (arg != NULL) {
      out->print("\t%s : %s %s (%s, ", arg->get_name(),
              arg->is_mandatory() ? "" : "[optional]",
              arg->get_description(), arg->get_type());
      if (arg->has_default()) {
        out->print(arg->get_default_as_string());
      } else {
        out->print("no default value");
      }
      out->print_cr(")");
      arg = arg->get_next();
    }
  }
  if (_options != NULL) {
    out->print_cr("\nOptions: (options must be specified using the <key> or <key>=<value> syntax)");
    arg = _options;
    while (arg != NULL) {
      out->print("\t%s : %s %s (%s, ", arg->get_name(),
              arg->is_mandatory() ? "" : "[optional]",
              arg->get_description(), arg->get_type());
      if (arg->has_default()) {
        out->print(arg->get_default_as_string());
      } else {
        out->print("no default value");
      }
      out->print_cr(")");
      arg = arg->get_next();
    }
  }
}

void DCmdParser::reset(TRAPS) {
  GenDCmdArgument* arg = _arguments;
  while(arg != NULL) {
    arg->reset(CHECK);
    arg = arg->get_next();
  }
  arg = _options;
  while(arg != NULL) {
    arg->reset(CHECK);
    arg = arg->get_next();
  }
}

void DCmdParser::cleanup() {
  GenDCmdArgument* arg = _arguments;
  while(arg != NULL) {
    arg->cleanup();
    arg = arg->get_next();
  }
  arg = _options;
  while(arg != NULL) {
    arg->cleanup();
    arg = arg->get_next();
  }
}

int DCmdParser::get_num_arguments() {
  GenDCmdArgument* arg = _arguments;
  int count = 0;
  while(arg != NULL) {
    count++;
    arg = arg->get_next();
  }
  arg = _options;
  while(arg != NULL) {
    count++;
    arg = arg->get_next();
  }
  return count;
}

GrowableArray<const char *>* DCmdParser::get_argument_name_array() {
  int count = get_num_arguments();
  GrowableArray<const char *>* array = new GrowableArray<const char *>(count);
  int idx = 0;
  GenDCmdArgument* arg = _arguments;
  while(arg != NULL) {
    array->append(arg->get_name());
    idx++;
    arg = arg->get_next();
  }
  arg = _options;
  while(arg != NULL) {
    array->append(arg->get_name());
    idx++;
    arg = arg->get_next();
  }
  return array;
}

GrowableArray<DCmdArgumentInfo*>* DCmdParser::get_argument_info_array() {
  int count = get_num_arguments();
  GrowableArray<DCmdArgumentInfo*>* array = new GrowableArray<DCmdArgumentInfo *>(count);
  int idx = 0;
  GenDCmdArgument* arg = _arguments;
  while(arg != NULL) {
    array->append(new DCmdArgumentInfo(arg->get_name(), arg->get_description(),
            arg->get_type(), arg->get_default_as_string(), arg->is_mandatory(),
            false, idx));
    idx++;
    arg = arg->get_next();
  }
  arg = _options;
  while(arg != NULL) {
    array->append(new DCmdArgumentInfo(arg->get_name(), arg->get_description(),
            arg->get_type(), arg->get_default_as_string(), arg->is_mandatory(),
            true));
    idx++;
    arg = arg->get_next();
  }
  return array;
}

DCmdFactory* DCmdFactory::_DCmdFactoryList = NULL;

void DCmd::parse_and_execute(outputStream* out, const char* cmdline,
        char delim, TRAPS) {

  if (cmdline == NULL) return; // Nothing to do!
  DCmdIter iter(cmdline, '\n');

  while (iter.has_next()) {
    CmdLine line = iter.get_next();
    if(line.is_stop()) {
      break;
    }
    if (line.is_executable()) {
      DCmd* command = DCmdFactory::create_local_DCmd(line,out, CHECK);
      assert(command != NULL,"command error must be handled before this line");
      DCmdMark mark(command);
      command->parse(&line, delim, CHECK);
      command->execute(CHECK);
    }
  }
}

Mutex* DCmdFactory::_dcmdFactory_lock = new Mutex(Mutex::leaf,"DCmdFactory",true);

DCmdFactory* DCmdFactory::get_factory(const char* name, size_t len) {
  MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
  DCmdFactory* factory = _DCmdFactoryList;
  while (factory != NULL) {
    if (strlen(factory->get_name()) == len &&
            strncmp(name,factory->get_name(), len) == 0) {
      return factory;
    }
    factory = factory->_next;
  }
  return NULL;
}

int DCmdFactory::register_DCmdFactory(DCmdFactory* factory) {
  MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
  factory->_next = _DCmdFactoryList;
  _DCmdFactoryList = factory;
  return 0; // Actually, there's no checks for duplicates
}

DCmd* DCmdFactory::create_global_DCmd(CmdLine &line, outputStream* out, TRAPS) {
  {
    DCmdFactory* factory = get_factory(line.cmd_addr(),line.cmd_len());
    if(factory != NULL) {
      if(!factory->is_enabled()) {
        THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
             factory->get_disabled_message());
      }
      return factory->get_cheap_instance(out);
    }
  }
  THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
             "Unknown diagnostic command");
}

DCmd* DCmdFactory::create_local_DCmd(CmdLine &line, outputStream* out, TRAPS) {
  {
    DCmdFactory* factory = get_factory(line.cmd_addr(),line.cmd_len());
    if(factory != NULL) {
      if(!factory->is_enabled()) {
        THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
             factory->get_disabled_message());
      }
      return factory->get_resource_instance(out);
    }
  }
  THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
             "Unknown diagnostic command");
}

GrowableArray<const char*>* DCmdFactory::get_DCmd_list() {
    MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
    GrowableArray<const char*>* array = new GrowableArray<const char*>();
    DCmdFactory* factory = _DCmdFactoryList;
    while (factory != NULL) {
        if (!factory->is_hidden()) {
            array->append(factory->get_name());
        }
        factory = factory->get_next();
    }
    return array;
}

GrowableArray<DCmdInfo*>* DCmdFactory::get_DCmdInfo_list() {
    MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
    GrowableArray<DCmdInfo*>* array = new GrowableArray<DCmdInfo*>();
    DCmdFactory* factory = _DCmdFactoryList;
    while (factory != NULL) {
        if (!factory->is_hidden()) {
            array->append(new DCmdInfo(factory->get_name(),
                    factory->get_description(), factory->get_impact(),
                    factory->get_num_arguments(), factory->is_enabled()));
        }
        factory = factory->get_next();
    }
    return array;
}