1 /*
   2  * Copyright (c) 2011, 2015, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  *
  23  */
  24 
  25 #include "precompiled.hpp"
  26 #include "memory/oopFactory.hpp"
  27 #include "oops/oop.inline.hpp"
  28 #include "runtime/javaCalls.hpp"
  29 #include "runtime/mutexLocker.hpp"
  30 #include "services/diagnosticArgument.hpp"
  31 #include "services/diagnosticFramework.hpp"
  32 #include "services/management.hpp"
  33 
  34 CmdLine::CmdLine(const char* line, size_t len, bool no_command_name) {
  35   assert(line != NULL, "Command line string should not be NULL");
  36   const char* line_end;
  37   const char* cmd_end;
  38 
  39   _cmd = line;
  40   line_end = &line[len];
  41 
  42   // Skip whitespace in the beginning of the line.
  43   while (_cmd < line_end && isspace((int) _cmd[0])) {
  44     _cmd++;
  45   }
  46   cmd_end = _cmd;
  47 
  48   if (no_command_name) {
  49     _cmd = NULL;
  50     _cmd_len = 0;
  51   } else {
  52     // Look for end of the command name
  53     while (cmd_end < line_end && !isspace((int) cmd_end[0])) {
  54       cmd_end++;
  55     }
  56     _cmd_len = cmd_end - _cmd;
  57   }
  58   _args = cmd_end;
  59   _args_len = line_end - _args;
  60 }
  61 
  62 bool DCmdArgIter::next(TRAPS) {
  63   if (_len == 0) return false;
  64   // skipping delimiters
  65   while (_cursor < _len - 1 && _buffer[_cursor] == _delim) {
  66     _cursor++;
  67   }
  68   // handling end of command line
  69   if (_cursor == _len - 1 && _buffer[_cursor] == _delim) {
  70     _key_addr = &_buffer[_cursor];
  71     _key_len = 0;
  72     _value_addr = &_buffer[_cursor];
  73     _value_len = 0;
  74     return false;
  75   }
  76   // extracting first item, argument or option name
  77   _key_addr = &_buffer[_cursor];
  78   bool arg_had_quotes = false;
  79   while (_cursor <= _len - 1 && _buffer[_cursor] != '=' && _buffer[_cursor] != _delim) {
  80     // argument can be surrounded by single or double quotes
  81     if (_buffer[_cursor] == '\"' || _buffer[_cursor] == '\'') {
  82       _key_addr++;
  83       char quote = _buffer[_cursor];
  84       arg_had_quotes = true;
  85       while (_cursor < _len - 1) {
  86         _cursor++;
  87         if (_buffer[_cursor] == quote && _buffer[_cursor - 1] != '\\') {
  88           break;
  89         }
  90       }
  91       if (_buffer[_cursor] != quote) {
  92         THROW_MSG_(vmSymbols::java_lang_IllegalArgumentException(),
  93                 "Format error in diagnostic command arguments", false);
  94       }
  95       break;
  96     }
  97     _cursor++;
  98   }
  99   _key_len = &_buffer[_cursor] - _key_addr;
 100   if (arg_had_quotes) {
 101     // if the argument was quoted, we need to step past the last quote here
 102     _cursor++;
 103   }
 104   // check if the argument has the <key>=<value> format
 105   if (_cursor <= _len -1 && _buffer[_cursor] == '=') {
 106     _cursor++;
 107     _value_addr = &_buffer[_cursor];
 108     bool value_had_quotes = false;
 109     // extract the value
 110     while (_cursor <= _len - 1 && _buffer[_cursor] != _delim) {
 111       // value can be surrounded by simple or double quotes
 112       if (_buffer[_cursor] == '\"' || _buffer[_cursor] == '\'') {
 113         _value_addr++;
 114         char quote = _buffer[_cursor];
 115         value_had_quotes = true;
 116         while (_cursor < _len - 1) {
 117           _cursor++;
 118           if (_buffer[_cursor] == quote && _buffer[_cursor - 1] != '\\') {
 119             break;
 120           }
 121         }
 122         if (_buffer[_cursor] != quote) {
 123           THROW_MSG_(vmSymbols::java_lang_IllegalArgumentException(),
 124                   "Format error in diagnostic command arguments", false);
 125         }
 126         break;
 127       }
 128       _cursor++;
 129     }
 130     _value_len = &_buffer[_cursor] - _value_addr;
 131     if (value_had_quotes) {
 132       // if the value was quoted, we need to step past the last quote here
 133       _cursor++;
 134     }
 135   } else {
 136     _value_addr = NULL;
 137     _value_len = 0;
 138   }
 139   return _key_len != 0;
 140 }
 141 
 142 bool DCmdInfo::by_name(void* cmd_name, DCmdInfo* info) {
 143   if (info == NULL) return false;
 144   return strcmp((const char*)cmd_name, info->name()) == 0;
 145 }
 146 
 147 void DCmdParser::add_dcmd_option(GenDCmdArgument* arg) {
 148   assert(arg != NULL, "Sanity");
 149   if (_options == NULL) {
 150     _options = arg;
 151   } else {
 152     GenDCmdArgument* o = _options;
 153     while (o->next() != NULL) {
 154       o = o->next();
 155     }
 156     o->set_next(arg);
 157   }
 158   arg->set_next(NULL);
 159   Thread* THREAD = Thread::current();
 160   arg->init_value(THREAD);
 161   if (HAS_PENDING_EXCEPTION) {
 162     fatal("Initialization must be successful");
 163   }
 164 }
 165 
 166 void DCmdParser::add_dcmd_argument(GenDCmdArgument* arg) {
 167   assert(arg != NULL, "Sanity");
 168   if (_arguments_list == NULL) {
 169     _arguments_list = arg;
 170   } else {
 171     GenDCmdArgument* a = _arguments_list;
 172     while (a->next() != NULL) {
 173       a = a->next();
 174     }
 175     a->set_next(arg);
 176   }
 177   arg->set_next(NULL);
 178   Thread* THREAD = Thread::current();
 179   arg->init_value(THREAD);
 180   if (HAS_PENDING_EXCEPTION) {
 181     fatal("Initialization must be successful");
 182   }
 183 }
 184 
 185 void DCmdParser::parse(CmdLine* line, char delim, TRAPS) {
 186   GenDCmdArgument* next_argument = _arguments_list;
 187   DCmdArgIter iter(line->args_addr(), line->args_len(), delim);
 188   bool cont = iter.next(CHECK);
 189   while (cont) {
 190     GenDCmdArgument* arg = lookup_dcmd_option(iter.key_addr(),
 191             iter.key_length());
 192     if (arg != NULL) {
 193       arg->read_value(iter.value_addr(), iter.value_length(), CHECK);
 194     } else {
 195       if (next_argument != NULL) {
 196         arg = next_argument;
 197         arg->read_value(iter.key_addr(), iter.key_length(), CHECK);
 198         next_argument = next_argument->next();
 199       } else {
 200         const size_t buflen    = 120;
 201         const size_t argbuflen = 30;
 202         char buf[buflen];
 203         char argbuf[argbuflen];
 204         size_t len = MIN2<size_t>(iter.key_length(), argbuflen - 1);
 205 
 206         strncpy(argbuf, iter.key_addr(), len);
 207         argbuf[len] = '\0';
 208         jio_snprintf(buf, buflen - 1, "Unknown argument '%s' in diagnostic command.", argbuf);
 209 
 210         THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(), buf);
 211       }
 212     }
 213     cont = iter.next(CHECK);
 214   }
 215   check(CHECK);
 216 }
 217 
 218 GenDCmdArgument* DCmdParser::lookup_dcmd_option(const char* name, size_t len) {
 219   GenDCmdArgument* arg = _options;
 220   while (arg != NULL) {
 221     if (strlen(arg->name()) == len &&
 222       strncmp(name, arg->name(), len) == 0) {
 223       return arg;
 224     }
 225     arg = arg->next();
 226   }
 227   return NULL;
 228 }
 229 
 230 void DCmdParser::check(TRAPS) {
 231   const size_t buflen = 256;
 232   char buf[buflen];
 233   GenDCmdArgument* arg = _arguments_list;
 234   while (arg != NULL) {
 235     if (arg->is_mandatory() && !arg->has_value()) {
 236       jio_snprintf(buf, buflen - 1, "The argument '%s' is mandatory.", arg->name());
 237       THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(), buf);
 238     }
 239     arg = arg->next();
 240   }
 241   arg = _options;
 242   while (arg != NULL) {
 243     if (arg->is_mandatory() && !arg->has_value()) {
 244       jio_snprintf(buf, buflen - 1, "The option '%s' is mandatory.", arg->name());
 245       THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(), buf);
 246     }
 247     arg = arg->next();
 248   }
 249 }
 250 
 251 void DCmdParser::print_help(outputStream* out, const char* cmd_name) {
 252   out->print("Syntax : %s %s", cmd_name, _options == NULL ? "" : "[options]");
 253   GenDCmdArgument* arg = _arguments_list;
 254   while (arg != NULL) {
 255     if (arg->is_mandatory()) {
 256       out->print(" <%s>", arg->name());
 257     } else {
 258       out->print(" [<%s>]", arg->name());
 259     }
 260     arg = arg->next();
 261   }
 262   out->cr();
 263   if (_arguments_list != NULL) {
 264     out->print_cr("\nArguments:");
 265     arg = _arguments_list;
 266     while (arg != NULL) {
 267       out->print("\t%s : %s %s (%s, ", arg->name(),
 268                  arg->is_mandatory() ? "" : "[optional]",
 269                  arg->description(), arg->type());
 270       if (arg->has_default()) {
 271         out->print("%s", arg->default_string());
 272       } else {
 273         out->print("no default value");
 274       }
 275       out->print_cr(")");
 276       arg = arg->next();
 277     }
 278   }
 279   if (_options != NULL) {
 280     out->print_cr("\nOptions: (options must be specified using the <key> or <key>=<value> syntax)");
 281     arg = _options;
 282     while (arg != NULL) {
 283       out->print("\t%s : %s %s (%s, ", arg->name(),
 284                  arg->is_mandatory() ? "" : "[optional]",
 285                  arg->description(), arg->type());
 286       if (arg->has_default()) {
 287         out->print("%s", arg->default_string());
 288       } else {
 289         out->print("no default value");
 290       }
 291       out->print_cr(")");
 292       arg = arg->next();
 293     }
 294   }
 295 }
 296 
 297 void DCmdParser::reset(TRAPS) {
 298   GenDCmdArgument* arg = _arguments_list;
 299   while (arg != NULL) {
 300     arg->reset(CHECK);
 301     arg = arg->next();
 302   }
 303   arg = _options;
 304   while (arg != NULL) {
 305     arg->reset(CHECK);
 306     arg = arg->next();
 307   }
 308 }
 309 
 310 void DCmdParser::cleanup() {
 311   GenDCmdArgument* arg = _arguments_list;
 312   while (arg != NULL) {
 313     arg->cleanup();
 314     arg = arg->next();
 315   }
 316   arg = _options;
 317   while (arg != NULL) {
 318     arg->cleanup();
 319     arg = arg->next();
 320   }
 321 }
 322 
 323 int DCmdParser::num_arguments() {
 324   GenDCmdArgument* arg = _arguments_list;
 325   int count = 0;
 326   while (arg != NULL) {
 327     count++;
 328     arg = arg->next();
 329   }
 330   arg = _options;
 331   while (arg != NULL) {
 332     count++;
 333     arg = arg->next();
 334   }
 335   return count;
 336 }
 337 
 338 GrowableArray<const char *>* DCmdParser::argument_name_array() {
 339   int count = num_arguments();
 340   GrowableArray<const char *>* array = new GrowableArray<const char *>(count);
 341   GenDCmdArgument* arg = _arguments_list;
 342   while (arg != NULL) {
 343     array->append(arg->name());
 344     arg = arg->next();
 345   }
 346   arg = _options;
 347   while (arg != NULL) {
 348     array->append(arg->name());
 349     arg = arg->next();
 350   }
 351   return array;
 352 }
 353 
 354 GrowableArray<DCmdArgumentInfo*>* DCmdParser::argument_info_array() {
 355   int count = num_arguments();
 356   GrowableArray<DCmdArgumentInfo*>* array = new GrowableArray<DCmdArgumentInfo *>(count);
 357   int idx = 0;
 358   GenDCmdArgument* arg = _arguments_list;
 359   while (arg != NULL) {
 360     array->append(new DCmdArgumentInfo(arg->name(), arg->description(),
 361                   arg->type(), arg->default_string(), arg->is_mandatory(),
 362                   false, arg->allow_multiple(), idx));
 363     idx++;
 364     arg = arg->next();
 365   }
 366   arg = _options;
 367   while (arg != NULL) {
 368     array->append(new DCmdArgumentInfo(arg->name(), arg->description(),
 369                   arg->type(), arg->default_string(), arg->is_mandatory(),
 370                   true, arg->allow_multiple()));
 371     arg = arg->next();
 372   }
 373   return array;
 374 }
 375 
 376 DCmdFactory* DCmdFactory::_DCmdFactoryList = NULL;
 377 bool DCmdFactory::_has_pending_jmx_notification = false;
 378 
 379 void DCmd::parse_and_execute(DCmdSource source, outputStream* out,
 380                              const char* cmdline, char delim, TRAPS) {
 381 
 382   if (cmdline == NULL) return; // Nothing to do!
 383   DCmdIter iter(cmdline, '\n');
 384 
 385   int count = 0;
 386   while (iter.has_next()) {
 387     if(source == DCmd_Source_MBean && count > 0) {
 388       // When diagnostic commands are invoked via JMX, each command line
 389       // must contains one and only one command because of the Permission
 390       // checks performed by the DiagnosticCommandMBean
 391       THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),
 392               "Invalid syntax");
 393     }
 394     CmdLine line = iter.next();
 395     if (line.is_stop()) {
 396       break;
 397     }
 398     if (line.is_executable()) {
 399       DCmd* command = DCmdFactory::create_local_DCmd(source, line, out, CHECK);
 400       assert(command != NULL, "command error must be handled before this line");
 401       DCmdMark mark(command);
 402       command->parse(&line, delim, CHECK);
 403       command->execute(source, CHECK);
 404     }
 405     count++;
 406   }
 407 }
 408 
 409 void DCmdWithParser::parse(CmdLine* line, char delim, TRAPS) {
 410   _dcmdparser.parse(line, delim, CHECK);
 411 }
 412 
 413 void DCmdWithParser::print_help(const char* name) {
 414   _dcmdparser.print_help(output(), name);
 415 }
 416 
 417 void DCmdWithParser::reset(TRAPS) {
 418   _dcmdparser.reset(CHECK);
 419 }
 420 
 421 void DCmdWithParser::cleanup() {
 422   _dcmdparser.cleanup();
 423 }
 424 
 425 GrowableArray<const char*>* DCmdWithParser::argument_name_array() {
 426   return _dcmdparser.argument_name_array();
 427 }
 428 
 429 GrowableArray<DCmdArgumentInfo*>* DCmdWithParser::argument_info_array() {
 430   return _dcmdparser.argument_info_array();
 431 }
 432 
 433 void DCmdFactory::push_jmx_notification_request() {
 434   MutexLockerEx ml(Service_lock, Mutex::_no_safepoint_check_flag);
 435   _has_pending_jmx_notification = true;
 436   Service_lock->notify_all();
 437 }
 438 
 439 void DCmdFactory::send_notification(TRAPS) {
 440   DCmdFactory::send_notification_internal(THREAD);
 441   // Clearing pending exception to avoid premature termination of
 442   // the service thread
 443   if (HAS_PENDING_EXCEPTION) {
 444     CLEAR_PENDING_EXCEPTION;
 445   }
 446 }
 447 void DCmdFactory::send_notification_internal(TRAPS) {
 448   ResourceMark rm(THREAD);
 449   HandleMark hm(THREAD);
 450   bool notif = false;
 451   {
 452     MutexLockerEx ml(Service_lock, Mutex::_no_safepoint_check_flag);
 453     notif = _has_pending_jmx_notification;
 454     _has_pending_jmx_notification = false;
 455   }
 456   if (notif) {
 457 
 458     Klass* k = Management::com_sun_management_internal_DiagnosticCommandImpl_klass(CHECK);
 459     instanceKlassHandle dcmd_mbean_klass(THREAD, k);
 460 
 461     JavaValue result(T_OBJECT);
 462     JavaCalls::call_static(&result,
 463             dcmd_mbean_klass,
 464             vmSymbols::getDiagnosticCommandMBean_name(),
 465             vmSymbols::getDiagnosticCommandMBean_signature(),
 466             CHECK);
 467 
 468     instanceOop m = (instanceOop) result.get_jobject();
 469     instanceHandle dcmd_mbean_h(THREAD, m);
 470 
 471     if (!dcmd_mbean_h->is_a(k)) {
 472       THROW_MSG(vmSymbols::java_lang_IllegalArgumentException(),
 473               "DiagnosticCommandImpl.getDiagnosticCommandMBean didn't return a DiagnosticCommandMBean instance");
 474     }
 475 
 476     JavaValue result2(T_VOID);
 477     JavaCallArguments args2(dcmd_mbean_h);
 478 
 479     JavaCalls::call_virtual(&result2,
 480             dcmd_mbean_klass,
 481             vmSymbols::createDiagnosticFrameworkNotification_name(),
 482             vmSymbols::void_method_signature(),
 483             &args2,
 484             CHECK);
 485   }
 486 }
 487 
 488 Mutex* DCmdFactory::_dcmdFactory_lock = new Mutex(Mutex::leaf, "DCmdFactory", true, Monitor::_safepoint_check_never);
 489 bool DCmdFactory::_send_jmx_notification = false;
 490 
 491 DCmdFactory* DCmdFactory::factory(DCmdSource source, const char* name, size_t len) {
 492   MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
 493   DCmdFactory* factory = _DCmdFactoryList;
 494   while (factory != NULL) {
 495     if (strlen(factory->name()) == len &&
 496         strncmp(name, factory->name(), len) == 0) {
 497       if(factory->export_flags() & source) {
 498         return factory;
 499       } else {
 500         return NULL;
 501       }
 502     }
 503     factory = factory->_next;
 504   }
 505   return NULL;
 506 }
 507 
 508 int DCmdFactory::register_DCmdFactory(DCmdFactory* factory) {
 509   MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
 510   factory->_next = _DCmdFactoryList;
 511   _DCmdFactoryList = factory;
 512   if (_send_jmx_notification && !factory->_hidden
 513       && (factory->_export_flags & DCmd_Source_MBean)) {
 514     DCmdFactory::push_jmx_notification_request();
 515   }
 516   return 0; // Actually, there's no checks for duplicates
 517 }
 518 
 519 DCmd* DCmdFactory::create_global_DCmd(DCmdSource source, CmdLine &line,
 520                                       outputStream* out, TRAPS) {
 521   DCmdFactory* f = factory(source, line.cmd_addr(), line.cmd_len());
 522   if (f != NULL) {
 523     if (f->is_enabled()) {
 524       THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
 525                      f->disabled_message());
 526     }
 527     return f->create_Cheap_instance(out);
 528   }
 529   THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
 530              "Unknown diagnostic command");
 531 }
 532 
 533 DCmd* DCmdFactory::create_local_DCmd(DCmdSource source, CmdLine &line,
 534                                      outputStream* out, TRAPS) {
 535   DCmdFactory* f = factory(source, line.cmd_addr(), line.cmd_len());
 536   if (f != NULL) {
 537     if (!f->is_enabled()) {
 538       THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
 539                      f->disabled_message());
 540     }
 541     return f->create_resource_instance(out);
 542   }
 543   THROW_MSG_NULL(vmSymbols::java_lang_IllegalArgumentException(),
 544              "Unknown diagnostic command");
 545 }
 546 
 547 GrowableArray<const char*>* DCmdFactory::DCmd_list(DCmdSource source) {
 548   MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
 549   GrowableArray<const char*>* array = new GrowableArray<const char*>();
 550   DCmdFactory* factory = _DCmdFactoryList;
 551   while (factory != NULL) {
 552     if (!factory->is_hidden() && (factory->export_flags() & source)) {
 553       array->append(factory->name());
 554     }
 555     factory = factory->next();
 556   }
 557   return array;
 558 }
 559 
 560 GrowableArray<DCmdInfo*>* DCmdFactory::DCmdInfo_list(DCmdSource source ) {
 561   MutexLockerEx ml(_dcmdFactory_lock, Mutex::_no_safepoint_check_flag);
 562   GrowableArray<DCmdInfo*>* array = new GrowableArray<DCmdInfo*>();
 563   DCmdFactory* factory = _DCmdFactoryList;
 564   while (factory != NULL) {
 565     if (!factory->is_hidden() && (factory->export_flags() & source)) {
 566       array->append(new DCmdInfo(factory->name(),
 567                     factory->description(), factory->impact(),
 568                     factory->permission(), factory->num_arguments(),
 569                     factory->is_enabled()));
 570     }
 571     factory = factory->next();
 572   }
 573   return array;
 574 }