1ebc09c36SJim Ingham //===-- CommandObjectSource.cpp ---------------------------------*- C++ -*-===// 2ebc09c36SJim Ingham // 3ebc09c36SJim Ingham // The LLVM Compiler Infrastructure 4ebc09c36SJim Ingham // 5ebc09c36SJim Ingham // This file is distributed under the University of Illinois Open Source 6ebc09c36SJim Ingham // License. See LICENSE.TXT for details. 7ebc09c36SJim Ingham // 8ebc09c36SJim Ingham //===----------------------------------------------------------------------===// 9ebc09c36SJim Ingham 10ebc09c36SJim Ingham #include "CommandObjectCommands.h" 11ebc09c36SJim Ingham 12ebc09c36SJim Ingham // C Includes 13ebc09c36SJim Ingham // C++ Includes 14ebc09c36SJim Ingham // Other libraries and framework includes 150e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h" 160e5e5a79SGreg Clayton 17ebc09c36SJim Ingham // Project includes 18ebc09c36SJim Ingham #include "lldb/Core/Debugger.h" 19de164aaaSGreg Clayton #include "lldb/Core/InputReader.h" 20be93a35aSEnrico Granata #include "lldb/Core/InputReaderEZ.h" 21be93a35aSEnrico Granata #include "lldb/Core/StringList.h" 22de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h" 23ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h" 24de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h" 25ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h" 26ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h" 2799f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h" 2899f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreterPython.h" 29ebc09c36SJim Ingham 30ebc09c36SJim Ingham using namespace lldb; 31ebc09c36SJim Ingham using namespace lldb_private; 32ebc09c36SJim Ingham 33ebc09c36SJim Ingham //------------------------------------------------------------------------- 34ebc09c36SJim Ingham // CommandObjectCommandsSource 35ebc09c36SJim Ingham //------------------------------------------------------------------------- 36ebc09c36SJim Ingham 37a5a97ebeSJim Ingham class CommandObjectCommandsHistory : public CommandObject 38a5a97ebeSJim Ingham { 39a5a97ebeSJim Ingham private: 40a5a97ebeSJim Ingham 41a5a97ebeSJim Ingham class CommandOptions : public Options 42a5a97ebeSJim Ingham { 43a5a97ebeSJim Ingham public: 44a5a97ebeSJim Ingham 45a5a97ebeSJim Ingham CommandOptions (CommandInterpreter &interpreter) : 46a5a97ebeSJim Ingham Options (interpreter) 47a5a97ebeSJim Ingham { 48a5a97ebeSJim Ingham } 49a5a97ebeSJim Ingham 50a5a97ebeSJim Ingham virtual 51a5a97ebeSJim Ingham ~CommandOptions (){} 52a5a97ebeSJim Ingham 53a5a97ebeSJim Ingham virtual Error 54a5a97ebeSJim Ingham SetOptionValue (uint32_t option_idx, const char *option_arg) 55a5a97ebeSJim Ingham { 56a5a97ebeSJim Ingham Error error; 57a5a97ebeSJim Ingham char short_option = (char) m_getopt_table[option_idx].val; 58a5a97ebeSJim Ingham bool success; 59a5a97ebeSJim Ingham 60a5a97ebeSJim Ingham switch (short_option) 61a5a97ebeSJim Ingham { 62a5a97ebeSJim Ingham case 'c': 63a5a97ebeSJim Ingham m_end_idx = Args::StringToUInt32(option_arg, UINT_MAX, 0, &success); 64a5a97ebeSJim Ingham if (!success) 6586edbf41SGreg Clayton error.SetErrorStringWithFormat("invalid value for count: %s", option_arg); 66a5a97ebeSJim Ingham if (m_end_idx != 0) 67a5a97ebeSJim Ingham m_end_idx--; 68a5a97ebeSJim Ingham m_start_idx = 0; 69a5a97ebeSJim Ingham break; 70a5a97ebeSJim Ingham case 'e': 71a5a97ebeSJim Ingham m_end_idx = Args::StringToUInt32(option_arg, 0, 0, &success); 72a5a97ebeSJim Ingham if (!success) 7386edbf41SGreg Clayton error.SetErrorStringWithFormat("invalid value for end index: %s", option_arg); 74a5a97ebeSJim Ingham break; 75a5a97ebeSJim Ingham case 's': 76a5a97ebeSJim Ingham m_start_idx = Args::StringToUInt32(option_arg, 0, 0, &success); 77a5a97ebeSJim Ingham if (!success) 7886edbf41SGreg Clayton error.SetErrorStringWithFormat("invalid value for start index: %s", option_arg); 79a5a97ebeSJim Ingham break; 80a5a97ebeSJim Ingham default: 8186edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 82a5a97ebeSJim Ingham break; 83a5a97ebeSJim Ingham } 84a5a97ebeSJim Ingham 85a5a97ebeSJim Ingham return error; 86a5a97ebeSJim Ingham } 87a5a97ebeSJim Ingham 88a5a97ebeSJim Ingham void 89a5a97ebeSJim Ingham OptionParsingStarting () 90a5a97ebeSJim Ingham { 91a5a97ebeSJim Ingham m_start_idx = 0; 92a5a97ebeSJim Ingham m_end_idx = UINT_MAX; 93a5a97ebeSJim Ingham } 94a5a97ebeSJim Ingham 95a5a97ebeSJim Ingham const OptionDefinition* 96a5a97ebeSJim Ingham GetDefinitions () 97a5a97ebeSJim Ingham { 98a5a97ebeSJim Ingham return g_option_table; 99a5a97ebeSJim Ingham } 100a5a97ebeSJim Ingham 101a5a97ebeSJim Ingham // Options table: Required for subclasses of Options. 102a5a97ebeSJim Ingham 103a5a97ebeSJim Ingham static OptionDefinition g_option_table[]; 104a5a97ebeSJim Ingham 105a5a97ebeSJim Ingham // Instance variables to hold the values for command options. 106a5a97ebeSJim Ingham 107a5a97ebeSJim Ingham uint32_t m_start_idx; 108a5a97ebeSJim Ingham uint32_t m_end_idx; 109a5a97ebeSJim Ingham }; 110a5a97ebeSJim Ingham 111a5a97ebeSJim Ingham CommandOptions m_options; 112a5a97ebeSJim Ingham 113a5a97ebeSJim Ingham virtual Options * 114a5a97ebeSJim Ingham GetOptions () 115a5a97ebeSJim Ingham { 116a5a97ebeSJim Ingham return &m_options; 117a5a97ebeSJim Ingham } 118a5a97ebeSJim Ingham 119a5a97ebeSJim Ingham public: 120a5a97ebeSJim Ingham CommandObjectCommandsHistory(CommandInterpreter &interpreter) : 121a5a97ebeSJim Ingham CommandObject (interpreter, 122a5a97ebeSJim Ingham "command history", 123a5a97ebeSJim Ingham "Dump the history of commands in this session.", 124a5a97ebeSJim Ingham NULL), 125a5a97ebeSJim Ingham m_options (interpreter) 126a5a97ebeSJim Ingham { 127a5a97ebeSJim Ingham } 128a5a97ebeSJim Ingham 129a5a97ebeSJim Ingham ~CommandObjectCommandsHistory () 130a5a97ebeSJim Ingham { 131a5a97ebeSJim Ingham } 132a5a97ebeSJim Ingham 133a5a97ebeSJim Ingham bool 134a5a97ebeSJim Ingham Execute 135a5a97ebeSJim Ingham ( 136a5a97ebeSJim Ingham Args& args, 137a5a97ebeSJim Ingham CommandReturnObject &result 138a5a97ebeSJim Ingham ) 139a5a97ebeSJim Ingham { 140a5a97ebeSJim Ingham 141a5a97ebeSJim Ingham m_interpreter.DumpHistory (result.GetOutputStream(), 142a5a97ebeSJim Ingham m_options.m_start_idx, 143a5a97ebeSJim Ingham m_options.m_end_idx); 144a5a97ebeSJim Ingham return result.Succeeded(); 145a5a97ebeSJim Ingham 146a5a97ebeSJim Ingham } 147a5a97ebeSJim Ingham }; 148a5a97ebeSJim Ingham 149a5a97ebeSJim Ingham OptionDefinition 150a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] = 151a5a97ebeSJim Ingham { 152a5a97ebeSJim Ingham { LLDB_OPT_SET_1, false, "count", 'c', required_argument, NULL, 0, eArgTypeUnsignedInteger, "How many history commands to print."}, 153a5a97ebeSJim Ingham { LLDB_OPT_SET_1, false, "start-index", 's', required_argument, NULL, 0, eArgTypeUnsignedInteger, "Index at which to start printing history commands."}, 154a5a97ebeSJim Ingham { LLDB_OPT_SET_1, false, "end-index", 'e', required_argument, NULL, 0, eArgTypeUnsignedInteger, "Index at which to stop printing history commands."}, 155a5a97ebeSJim Ingham { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 156a5a97ebeSJim Ingham }; 157a5a97ebeSJim Ingham 158a5a97ebeSJim Ingham 159a5a97ebeSJim Ingham //------------------------------------------------------------------------- 160a5a97ebeSJim Ingham // CommandObjectCommandsSource 161a5a97ebeSJim Ingham //------------------------------------------------------------------------- 162a5a97ebeSJim Ingham 163ebc09c36SJim Ingham class CommandObjectCommandsSource : public CommandObject 164ebc09c36SJim Ingham { 165e16c50a1SJim Ingham private: 166e16c50a1SJim Ingham 167e16c50a1SJim Ingham class CommandOptions : public Options 168e16c50a1SJim Ingham { 169e16c50a1SJim Ingham public: 170e16c50a1SJim Ingham 171eb0103f2SGreg Clayton CommandOptions (CommandInterpreter &interpreter) : 172eb0103f2SGreg Clayton Options (interpreter) 173eb0103f2SGreg Clayton { 174eb0103f2SGreg Clayton } 175e16c50a1SJim Ingham 176e16c50a1SJim Ingham virtual 177e16c50a1SJim Ingham ~CommandOptions (){} 178e16c50a1SJim Ingham 179e16c50a1SJim Ingham virtual Error 180f6b8b581SGreg Clayton SetOptionValue (uint32_t option_idx, const char *option_arg) 181e16c50a1SJim Ingham { 182e16c50a1SJim Ingham Error error; 183e16c50a1SJim Ingham char short_option = (char) m_getopt_table[option_idx].val; 184e16c50a1SJim Ingham bool success; 185e16c50a1SJim Ingham 186e16c50a1SJim Ingham switch (short_option) 187e16c50a1SJim Ingham { 188e16c50a1SJim Ingham case 'e': 189e16c50a1SJim Ingham m_stop_on_error = Args::StringToBoolean(option_arg, true, &success); 190e16c50a1SJim Ingham if (!success) 19186edbf41SGreg Clayton error.SetErrorStringWithFormat("invalid value for stop-on-error: %s", option_arg); 192e16c50a1SJim Ingham break; 193e16c50a1SJim Ingham case 'c': 194e16c50a1SJim Ingham m_stop_on_continue = Args::StringToBoolean(option_arg, true, &success); 195e16c50a1SJim Ingham if (!success) 19686edbf41SGreg Clayton error.SetErrorStringWithFormat("invalid value for stop-on-continue: %s", option_arg); 197e16c50a1SJim Ingham break; 198e16c50a1SJim Ingham default: 19986edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 200e16c50a1SJim Ingham break; 201e16c50a1SJim Ingham } 202e16c50a1SJim Ingham 203e16c50a1SJim Ingham return error; 204e16c50a1SJim Ingham } 205e16c50a1SJim Ingham 206e16c50a1SJim Ingham void 207f6b8b581SGreg Clayton OptionParsingStarting () 208e16c50a1SJim Ingham { 209e16c50a1SJim Ingham m_stop_on_error = true; 210e16c50a1SJim Ingham m_stop_on_continue = true; 211e16c50a1SJim Ingham } 212e16c50a1SJim Ingham 213e0d378b3SGreg Clayton const OptionDefinition* 214e16c50a1SJim Ingham GetDefinitions () 215e16c50a1SJim Ingham { 216e16c50a1SJim Ingham return g_option_table; 217e16c50a1SJim Ingham } 218e16c50a1SJim Ingham 219e16c50a1SJim Ingham // Options table: Required for subclasses of Options. 220e16c50a1SJim Ingham 221e0d378b3SGreg Clayton static OptionDefinition g_option_table[]; 222e16c50a1SJim Ingham 223e16c50a1SJim Ingham // Instance variables to hold the values for command options. 224e16c50a1SJim Ingham 225e16c50a1SJim Ingham bool m_stop_on_error; 226e16c50a1SJim Ingham bool m_stop_on_continue; 227e16c50a1SJim Ingham }; 228e16c50a1SJim Ingham 229e16c50a1SJim Ingham CommandOptions m_options; 230e16c50a1SJim Ingham 231e16c50a1SJim Ingham virtual Options * 232e16c50a1SJim Ingham GetOptions () 233e16c50a1SJim Ingham { 234e16c50a1SJim Ingham return &m_options; 235e16c50a1SJim Ingham } 236e16c50a1SJim Ingham 237ebc09c36SJim Ingham public: 238a7015092SGreg Clayton CommandObjectCommandsSource(CommandInterpreter &interpreter) : 239a7015092SGreg Clayton CommandObject (interpreter, 2400e5e5a79SGreg Clayton "command source", 241e3d26315SCaroline Tice "Read in debugger commands from the file <filename> and execute them.", 242eb0103f2SGreg Clayton NULL), 243eb0103f2SGreg Clayton m_options (interpreter) 244ebc09c36SJim Ingham { 245405fe67fSCaroline Tice CommandArgumentEntry arg; 246405fe67fSCaroline Tice CommandArgumentData file_arg; 247405fe67fSCaroline Tice 248405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 249405fe67fSCaroline Tice file_arg.arg_type = eArgTypeFilename; 250405fe67fSCaroline Tice file_arg.arg_repetition = eArgRepeatPlain; 251405fe67fSCaroline Tice 252405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 253405fe67fSCaroline Tice arg.push_back (file_arg); 254405fe67fSCaroline Tice 255405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 256405fe67fSCaroline Tice m_arguments.push_back (arg); 257ebc09c36SJim Ingham } 258ebc09c36SJim Ingham 259ebc09c36SJim Ingham ~CommandObjectCommandsSource () 260ebc09c36SJim Ingham { 261ebc09c36SJim Ingham } 262ebc09c36SJim Ingham 263ebc09c36SJim Ingham bool 264ebc09c36SJim Ingham Execute 265ebc09c36SJim Ingham ( 266ebc09c36SJim Ingham Args& args, 267ebc09c36SJim Ingham CommandReturnObject &result 268ebc09c36SJim Ingham ) 269ebc09c36SJim Ingham { 270ebc09c36SJim Ingham const int argc = args.GetArgumentCount(); 271ebc09c36SJim Ingham if (argc == 1) 272ebc09c36SJim Ingham { 273ebc09c36SJim Ingham const char *filename = args.GetArgumentAtIndex(0); 274ebc09c36SJim Ingham 275ebc09c36SJim Ingham result.AppendMessageWithFormat ("Executing commands in '%s'.\n", filename); 276ebc09c36SJim Ingham 2771ee3853fSJohnny Chen FileSpec cmd_file (filename, true); 278e16c50a1SJim Ingham ExecutionContext *exe_ctx = NULL; // Just use the default context. 279e16c50a1SJim Ingham bool echo_commands = true; 280e16c50a1SJim Ingham bool print_results = true; 281ebc09c36SJim Ingham 282e16c50a1SJim Ingham m_interpreter.HandleCommandsFromFile (cmd_file, 283e16c50a1SJim Ingham exe_ctx, 284e16c50a1SJim Ingham m_options.m_stop_on_continue, 285e16c50a1SJim Ingham m_options.m_stop_on_error, 286e16c50a1SJim Ingham echo_commands, 287e16c50a1SJim Ingham print_results, 288e16c50a1SJim Ingham result); 289ebc09c36SJim Ingham } 290ebc09c36SJim Ingham else 291ebc09c36SJim Ingham { 292ebc09c36SJim Ingham result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName()); 293ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 294ebc09c36SJim Ingham } 295ebc09c36SJim Ingham return result.Succeeded(); 296ebc09c36SJim Ingham 297ebc09c36SJim Ingham } 298ebc09c36SJim Ingham }; 299ebc09c36SJim Ingham 300e0d378b3SGreg Clayton OptionDefinition 301e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] = 302e16c50a1SJim Ingham { 303e16c50a1SJim Ingham { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', required_argument, NULL, 0, eArgTypeBoolean, "If true, stop executing commands on error."}, 304e16c50a1SJim Ingham { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', required_argument, NULL, 0, eArgTypeBoolean, "If true, stop executing commands on continue."}, 305e16c50a1SJim Ingham { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 306e16c50a1SJim Ingham }; 307e16c50a1SJim Ingham 308ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias 309ebc09c36SJim Ingham //------------------------------------------------------------------------- 310ebc09c36SJim Ingham // CommandObjectCommandsAlias 311ebc09c36SJim Ingham //------------------------------------------------------------------------- 312ebc09c36SJim Ingham 313be93a35aSEnrico Granata static const char *g_python_command_instructions = "Enter your Python command(s). Type 'DONE' to end.\n" 314be93a35aSEnrico Granata "You must define a Python function with this signature:\n" 315223383edSEnrico Granata "def my_command_impl(debugger, args, result, dict):"; 316be93a35aSEnrico Granata 317be93a35aSEnrico Granata 318ebc09c36SJim Ingham class CommandObjectCommandsAlias : public CommandObject 319ebc09c36SJim Ingham { 320be93a35aSEnrico Granata 321be93a35aSEnrico Granata 322ebc09c36SJim Ingham public: 323a7015092SGreg Clayton CommandObjectCommandsAlias (CommandInterpreter &interpreter) : 324a7015092SGreg Clayton CommandObject (interpreter, 3250e5e5a79SGreg Clayton "command alias", 326e3d26315SCaroline Tice "Allow users to define their own debugger command abbreviations.", 327405fe67fSCaroline Tice NULL) 328ebc09c36SJim Ingham { 329ebc09c36SJim Ingham SetHelpLong( 330ebc09c36SJim Ingham "'alias' allows the user to create a short-cut or abbreviation for long \n\ 331ebc09c36SJim Ingham commands, multi-word commands, and commands that take particular options. \n\ 332ebc09c36SJim Ingham Below are some simple examples of how one might use the 'alias' command: \n\ 33369c12ccbSJason Molenda \n 'command alias sc script' // Creates the abbreviation 'sc' for the 'script' \n\ 334ebc09c36SJim Ingham // command. \n\ 33569c12ccbSJason Molenda 'command alias bp breakpoint' // Creates the abbreviation 'bp' for the 'breakpoint' \n\ 336ebc09c36SJim Ingham // command. Since breakpoint commands are two-word \n\ 337ebc09c36SJim Ingham // commands, the user will still need to enter the \n\ 338ebc09c36SJim Ingham // second word after 'bp', e.g. 'bp enable' or \n\ 339ebc09c36SJim Ingham // 'bp delete'. \n\ 34069c12ccbSJason Molenda 'command alias bpl breakpoint list' // Creates the abbreviation 'bpl' for the \n\ 341ebc09c36SJim Ingham // two-word command 'breakpoint list'. \n\ 342ebc09c36SJim Ingham \nAn alias can include some options for the command, with the values either \n\ 343ebc09c36SJim Ingham filled in at the time the alias is created, or specified as positional \n\ 344ebc09c36SJim Ingham arguments, to be filled in when the alias is invoked. The following example \n\ 345ebc09c36SJim Ingham shows how to create aliases with options: \n\ 346ebc09c36SJim Ingham \n\ 34769c12ccbSJason Molenda 'command alias bfl breakpoint set -f %1 -l %2' \n\ 348ebc09c36SJim Ingham \nThis creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \n\ 349ebc09c36SJim Ingham options already part of the alias. So if the user wants to set a breakpoint \n\ 350ebc09c36SJim Ingham by file and line without explicitly having to use the -f and -l options, the \n\ 351ebc09c36SJim Ingham user can now use 'bfl' instead. The '%1' and '%2' are positional placeholders \n\ 352ebc09c36SJim Ingham for the actual arguments that will be passed when the alias command is used. \n\ 353ebc09c36SJim Ingham The number in the placeholder refers to the position/order the actual value \n\ 35481ded935SJim Ingham occupies when the alias is used. All the occurrences of '%1' in the alias \n\ 355ebc09c36SJim Ingham will be replaced with the first argument, all the occurrences of '%2' in the \n\ 356ebc09c36SJim Ingham alias will be replaced with the second argument, and so on. This also allows \n\ 357ebc09c36SJim Ingham actual arguments to be used multiple times within an alias (see 'process \n\ 35881ded935SJim Ingham launch' example below). \n\ 35981ded935SJim Ingham Note: the positional arguments must substitute as whole words in the resultant\n\ 36081ded935SJim Ingham command, so you can't at present do something like:\n\ 36181ded935SJim Ingham \n\ 36269c12ccbSJason Molenda command alias bcppfl breakpoint set -f %1.cpp -l %2\n\ 36381ded935SJim Ingham \n\ 36481ded935SJim Ingham to get the file extension \".cpp\" automatically appended. For more complex\n\ 36581ded935SJim Ingham aliasing, use the \"command regex\" command instead.\n\ 36681ded935SJim Ingham \nSo in the 'bfl' case, the actual file value will be \n\ 367ebc09c36SJim Ingham filled in with the first argument following 'bfl' and the actual line number \n\ 368ebc09c36SJim Ingham value will be filled in with the second argument. The user would use this \n\ 369ebc09c36SJim Ingham alias as follows: \n\ 37069c12ccbSJason Molenda \n (lldb) command alias bfl breakpoint set -f %1 -l %2 \n\ 371ebc09c36SJim Ingham <... some time later ...> \n\ 37209799af6SCaroline Tice (lldb) bfl my-file.c 137 \n\ 373ebc09c36SJim Ingham \nThis would be the same as if the user had entered \n\ 374ebc09c36SJim Ingham 'breakpoint set -f my-file.c -l 137'. \n\ 375ebc09c36SJim Ingham \nAnother example: \n\ 37669c12ccbSJason Molenda \n (lldb) command alias pltty process launch -s -o %1 -e %1 \n\ 37709799af6SCaroline Tice (lldb) pltty /dev/tty0 \n\ 378ebc09c36SJim Ingham // becomes 'process launch -s -o /dev/tty0 -e /dev/tty0' \n\ 379ebc09c36SJim Ingham \nIf the user always wanted to pass the same value to a particular option, the \n\ 380ebc09c36SJim Ingham alias could be defined with that value directly in the alias as a constant, \n\ 381ebc09c36SJim Ingham rather than using a positional placeholder: \n\ 38269c12ccbSJason Molenda \n command alias bl3 breakpoint set -f %1 -l 3 // Always sets a breakpoint on line \n\ 383ebc09c36SJim Ingham // 3 of whatever file is indicated. \n"); 384ebc09c36SJim Ingham 385405fe67fSCaroline Tice CommandArgumentEntry arg1; 386405fe67fSCaroline Tice CommandArgumentEntry arg2; 387405fe67fSCaroline Tice CommandArgumentEntry arg3; 388405fe67fSCaroline Tice CommandArgumentData alias_arg; 389405fe67fSCaroline Tice CommandArgumentData cmd_arg; 390405fe67fSCaroline Tice CommandArgumentData options_arg; 391405fe67fSCaroline Tice 392405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 393405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 394405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 395405fe67fSCaroline Tice 396405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 397405fe67fSCaroline Tice arg1.push_back (alias_arg); 398405fe67fSCaroline Tice 399405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 400405fe67fSCaroline Tice cmd_arg.arg_type = eArgTypeCommandName; 401405fe67fSCaroline Tice cmd_arg.arg_repetition = eArgRepeatPlain; 402405fe67fSCaroline Tice 403405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 404405fe67fSCaroline Tice arg2.push_back (cmd_arg); 405405fe67fSCaroline Tice 406405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 407405fe67fSCaroline Tice options_arg.arg_type = eArgTypeAliasOptions; 408405fe67fSCaroline Tice options_arg.arg_repetition = eArgRepeatOptional; 409405fe67fSCaroline Tice 410405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 411405fe67fSCaroline Tice arg3.push_back (options_arg); 412405fe67fSCaroline Tice 413405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 414405fe67fSCaroline Tice m_arguments.push_back (arg1); 415405fe67fSCaroline Tice m_arguments.push_back (arg2); 416405fe67fSCaroline Tice m_arguments.push_back (arg3); 417ebc09c36SJim Ingham } 418ebc09c36SJim Ingham 419ebc09c36SJim Ingham ~CommandObjectCommandsAlias () 420ebc09c36SJim Ingham { 421ebc09c36SJim Ingham } 422ebc09c36SJim Ingham 423844d2303SCaroline Tice bool 424844d2303SCaroline Tice WantsRawCommandString () 425844d2303SCaroline Tice { 426844d2303SCaroline Tice return true; 427844d2303SCaroline Tice } 428844d2303SCaroline Tice 429844d2303SCaroline Tice bool 430844d2303SCaroline Tice ExecuteRawCommandString (const char *raw_command_line, CommandReturnObject &result) 431844d2303SCaroline Tice { 432844d2303SCaroline Tice Args args (raw_command_line); 433844d2303SCaroline Tice std::string raw_command_string (raw_command_line); 434844d2303SCaroline Tice 435844d2303SCaroline Tice size_t argc = args.GetArgumentCount(); 436844d2303SCaroline Tice 437844d2303SCaroline Tice if (argc < 2) 438844d2303SCaroline Tice { 439844d2303SCaroline Tice result.AppendError ("'alias' requires at least two arguments"); 440844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 441844d2303SCaroline Tice return false; 442844d2303SCaroline Tice } 443844d2303SCaroline Tice 444844d2303SCaroline Tice // Get the alias command. 445844d2303SCaroline Tice 446844d2303SCaroline Tice const std::string alias_command = args.GetArgumentAtIndex (0); 447844d2303SCaroline Tice 448844d2303SCaroline Tice // Strip the new alias name off 'raw_command_string' (leave it on args, which gets passed to 'Execute', which 449844d2303SCaroline Tice // does the stripping itself. 450844d2303SCaroline Tice size_t pos = raw_command_string.find (alias_command); 451844d2303SCaroline Tice if (pos == 0) 452844d2303SCaroline Tice { 453844d2303SCaroline Tice raw_command_string = raw_command_string.substr (alias_command.size()); 454844d2303SCaroline Tice pos = raw_command_string.find_first_not_of (' '); 455844d2303SCaroline Tice if ((pos != std::string::npos) && (pos > 0)) 456844d2303SCaroline Tice raw_command_string = raw_command_string.substr (pos); 457844d2303SCaroline Tice } 458844d2303SCaroline Tice else 459844d2303SCaroline Tice { 460844d2303SCaroline Tice result.AppendError ("Error parsing command string. No alias created."); 461844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 462844d2303SCaroline Tice return false; 463844d2303SCaroline Tice } 464844d2303SCaroline Tice 465844d2303SCaroline Tice 466844d2303SCaroline Tice // Verify that the command is alias-able. 467844d2303SCaroline Tice if (m_interpreter.CommandExists (alias_command.c_str())) 468844d2303SCaroline Tice { 469844d2303SCaroline Tice result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n", 470844d2303SCaroline Tice alias_command.c_str()); 471844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 472844d2303SCaroline Tice return false; 473844d2303SCaroline Tice } 474844d2303SCaroline Tice 475844d2303SCaroline Tice // Get CommandObject that is being aliased. The command name is read from the front of raw_command_string. 476844d2303SCaroline Tice // raw_command_string is returned with the name of the command object stripped off the front. 477844d2303SCaroline Tice CommandObject *cmd_obj = m_interpreter.GetCommandObjectForCommand (raw_command_string); 478844d2303SCaroline Tice 479844d2303SCaroline Tice if (!cmd_obj) 480844d2303SCaroline Tice { 48186edbf41SGreg Clayton result.AppendErrorWithFormat ("invalid command given to 'alias'. '%s' does not begin with a valid command." 482844d2303SCaroline Tice " No alias created.", raw_command_string.c_str()); 483844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 484844d2303SCaroline Tice return false; 485844d2303SCaroline Tice } 486844d2303SCaroline Tice else if (!cmd_obj->WantsRawCommandString ()) 487844d2303SCaroline Tice { 488844d2303SCaroline Tice // Note that args was initialized with the original command, and has not been updated to this point. 489844d2303SCaroline Tice // Therefore can we pass it to the version of Execute that does not need/expect raw input in the alias. 490844d2303SCaroline Tice return Execute (args, result); 491844d2303SCaroline Tice } 492844d2303SCaroline Tice else 493844d2303SCaroline Tice { 494844d2303SCaroline Tice // Verify & handle any options/arguments passed to the alias command 495844d2303SCaroline Tice 496844d2303SCaroline Tice OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector); 497844d2303SCaroline Tice OptionArgVector *option_arg_vector = option_arg_vector_sp.get(); 498844d2303SCaroline Tice 499ca90c47eSCaroline Tice CommandObjectSP cmd_obj_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false); 500844d2303SCaroline Tice 501ca90c47eSCaroline Tice if (!m_interpreter.ProcessAliasOptionsArgs (cmd_obj_sp, raw_command_string.c_str(), option_arg_vector_sp)) 502844d2303SCaroline Tice { 503844d2303SCaroline Tice result.AppendError ("Unable to create requested alias.\n"); 504ca90c47eSCaroline Tice result.SetStatus (eReturnStatusFailed); 505844d2303SCaroline Tice return false; 506844d2303SCaroline Tice } 507844d2303SCaroline Tice 508844d2303SCaroline Tice // Create the alias 509844d2303SCaroline Tice if (m_interpreter.AliasExists (alias_command.c_str()) 510844d2303SCaroline Tice || m_interpreter.UserCommandExists (alias_command.c_str())) 511844d2303SCaroline Tice { 512844d2303SCaroline Tice OptionArgVectorSP temp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str())); 513844d2303SCaroline Tice if (temp_option_arg_sp.get()) 514844d2303SCaroline Tice { 515844d2303SCaroline Tice if (option_arg_vector->size() == 0) 516844d2303SCaroline Tice m_interpreter.RemoveAliasOptions (alias_command.c_str()); 517844d2303SCaroline Tice } 518844d2303SCaroline Tice result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n", 519844d2303SCaroline Tice alias_command.c_str()); 520844d2303SCaroline Tice } 521844d2303SCaroline Tice 522472362e6SCaroline Tice if (cmd_obj_sp) 523472362e6SCaroline Tice { 524844d2303SCaroline Tice m_interpreter.AddAlias (alias_command.c_str(), cmd_obj_sp); 525844d2303SCaroline Tice if (option_arg_vector->size() > 0) 526844d2303SCaroline Tice m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp); 527844d2303SCaroline Tice result.SetStatus (eReturnStatusSuccessFinishNoResult); 528844d2303SCaroline Tice } 529472362e6SCaroline Tice else 530472362e6SCaroline Tice { 531472362e6SCaroline Tice result.AppendError ("Unable to create requested alias.\n"); 532472362e6SCaroline Tice result.SetStatus (eReturnStatusFailed); 533472362e6SCaroline Tice } 534472362e6SCaroline Tice } 535844d2303SCaroline Tice return result.Succeeded(); 536844d2303SCaroline Tice } 537ebc09c36SJim Ingham 538ebc09c36SJim Ingham bool 539ebc09c36SJim Ingham Execute 540ebc09c36SJim Ingham ( 541ebc09c36SJim Ingham Args& args, 542ebc09c36SJim Ingham CommandReturnObject &result 543ebc09c36SJim Ingham ) 544ebc09c36SJim Ingham { 545867b185dSCaroline Tice size_t argc = args.GetArgumentCount(); 546ebc09c36SJim Ingham 547ebc09c36SJim Ingham if (argc < 2) 548ebc09c36SJim Ingham { 549ebc09c36SJim Ingham result.AppendError ("'alias' requires at least two arguments"); 550ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 551ebc09c36SJim Ingham return false; 552ebc09c36SJim Ingham } 553ebc09c36SJim Ingham 554ebc09c36SJim Ingham const std::string alias_command = args.GetArgumentAtIndex(0); 555ebc09c36SJim Ingham const std::string actual_command = args.GetArgumentAtIndex(1); 556ebc09c36SJim Ingham 557ebc09c36SJim Ingham args.Shift(); // Shift the alias command word off the argument vector. 558ebc09c36SJim Ingham args.Shift(); // Shift the old command word off the argument vector. 559ebc09c36SJim Ingham 560ebc09c36SJim Ingham // Verify that the command is alias'able, and get the appropriate command object. 561ebc09c36SJim Ingham 562a7015092SGreg Clayton if (m_interpreter.CommandExists (alias_command.c_str())) 563ebc09c36SJim Ingham { 564ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n", 565ebc09c36SJim Ingham alias_command.c_str()); 566ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 567ebc09c36SJim Ingham } 568ebc09c36SJim Ingham else 569ebc09c36SJim Ingham { 570a7015092SGreg Clayton CommandObjectSP command_obj_sp(m_interpreter.GetCommandSPExact (actual_command.c_str(), true)); 571ebc09c36SJim Ingham CommandObjectSP subcommand_obj_sp; 572ebc09c36SJim Ingham bool use_subcommand = false; 573ebc09c36SJim Ingham if (command_obj_sp.get()) 574ebc09c36SJim Ingham { 575ebc09c36SJim Ingham CommandObject *cmd_obj = command_obj_sp.get(); 576c982c768SGreg Clayton CommandObject *sub_cmd_obj = NULL; 577ebc09c36SJim Ingham OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector); 578ebc09c36SJim Ingham OptionArgVector *option_arg_vector = option_arg_vector_sp.get(); 579ebc09c36SJim Ingham 580844d2303SCaroline Tice while (cmd_obj->IsMultiwordObject() && args.GetArgumentCount() > 0) 581ebc09c36SJim Ingham { 582ebc09c36SJim Ingham if (argc >= 3) 583ebc09c36SJim Ingham { 584ebc09c36SJim Ingham const std::string sub_command = args.GetArgumentAtIndex(0); 585ebc09c36SJim Ingham assert (sub_command.length() != 0); 586ebc09c36SJim Ingham subcommand_obj_sp = 587ebc09c36SJim Ingham (((CommandObjectMultiword *) cmd_obj)->GetSubcommandSP (sub_command.c_str())); 588ebc09c36SJim Ingham if (subcommand_obj_sp.get()) 589ebc09c36SJim Ingham { 590ebc09c36SJim Ingham sub_cmd_obj = subcommand_obj_sp.get(); 591ebc09c36SJim Ingham use_subcommand = true; 592ebc09c36SJim Ingham args.Shift(); // Shift the sub_command word off the argument vector. 593844d2303SCaroline Tice cmd_obj = sub_cmd_obj; 594ebc09c36SJim Ingham } 595ebc09c36SJim Ingham else 596ebc09c36SJim Ingham { 597f415eeb4SCaroline Tice result.AppendErrorWithFormat("'%s' is not a valid sub-command of '%s'. " 598f415eeb4SCaroline Tice "Unable to create alias.\n", 599f415eeb4SCaroline Tice sub_command.c_str(), actual_command.c_str()); 600ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 601ebc09c36SJim Ingham return false; 602ebc09c36SJim Ingham } 603ebc09c36SJim Ingham } 604ebc09c36SJim Ingham } 605ebc09c36SJim Ingham 606ebc09c36SJim Ingham // Verify & handle any options/arguments passed to the alias command 607ebc09c36SJim Ingham 608ebc09c36SJim Ingham if (args.GetArgumentCount () > 0) 609ebc09c36SJim Ingham { 610ca90c47eSCaroline Tice CommandObjectSP tmp_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false); 611ebc09c36SJim Ingham if (use_subcommand) 612ca90c47eSCaroline Tice tmp_sp = m_interpreter.GetCommandSPExact (sub_cmd_obj->GetCommandName(), false); 613ca90c47eSCaroline Tice 614ca90c47eSCaroline Tice std::string args_string; 615ca90c47eSCaroline Tice args.GetCommandString (args_string); 616ca90c47eSCaroline Tice 617ca90c47eSCaroline Tice if (!m_interpreter.ProcessAliasOptionsArgs (tmp_sp, args_string.c_str(), option_arg_vector_sp)) 618ebc09c36SJim Ingham { 619ca90c47eSCaroline Tice result.AppendError ("Unable to create requested alias.\n"); 620ca90c47eSCaroline Tice result.SetStatus (eReturnStatusFailed); 621e7941795SCaroline Tice return false; 622867b185dSCaroline Tice } 623867b185dSCaroline Tice } 624867b185dSCaroline Tice 625ebc09c36SJim Ingham // Create the alias. 626ebc09c36SJim Ingham 627a7015092SGreg Clayton if (m_interpreter.AliasExists (alias_command.c_str()) 628a7015092SGreg Clayton || m_interpreter.UserCommandExists (alias_command.c_str())) 629ebc09c36SJim Ingham { 630a7015092SGreg Clayton OptionArgVectorSP tmp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str())); 631ebc09c36SJim Ingham if (tmp_option_arg_sp.get()) 632ebc09c36SJim Ingham { 633ebc09c36SJim Ingham if (option_arg_vector->size() == 0) 634a7015092SGreg Clayton m_interpreter.RemoveAliasOptions (alias_command.c_str()); 635ebc09c36SJim Ingham } 636ebc09c36SJim Ingham result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n", 637ebc09c36SJim Ingham alias_command.c_str()); 638ebc09c36SJim Ingham } 639ebc09c36SJim Ingham 640ebc09c36SJim Ingham if (use_subcommand) 641a7015092SGreg Clayton m_interpreter.AddAlias (alias_command.c_str(), subcommand_obj_sp); 642ebc09c36SJim Ingham else 643a7015092SGreg Clayton m_interpreter.AddAlias (alias_command.c_str(), command_obj_sp); 644ebc09c36SJim Ingham if (option_arg_vector->size() > 0) 645a7015092SGreg Clayton m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp); 646ebc09c36SJim Ingham result.SetStatus (eReturnStatusSuccessFinishNoResult); 647ebc09c36SJim Ingham } 648ebc09c36SJim Ingham else 649ebc09c36SJim Ingham { 650ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is not an existing command.\n", actual_command.c_str()); 651ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 652e7941795SCaroline Tice return false; 653ebc09c36SJim Ingham } 654ebc09c36SJim Ingham } 655ebc09c36SJim Ingham 656ebc09c36SJim Ingham return result.Succeeded(); 657ebc09c36SJim Ingham } 658ebc09c36SJim Ingham }; 659ebc09c36SJim Ingham 660ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias 661ebc09c36SJim Ingham //------------------------------------------------------------------------- 662ebc09c36SJim Ingham // CommandObjectCommandsUnalias 663ebc09c36SJim Ingham //------------------------------------------------------------------------- 664ebc09c36SJim Ingham 665ebc09c36SJim Ingham class CommandObjectCommandsUnalias : public CommandObject 666ebc09c36SJim Ingham { 667ebc09c36SJim Ingham public: 668a7015092SGreg Clayton CommandObjectCommandsUnalias (CommandInterpreter &interpreter) : 669a7015092SGreg Clayton CommandObject (interpreter, 6700e5e5a79SGreg Clayton "command unalias", 67186ddae50SCaroline Tice "Allow the user to remove/delete a user-defined command abbreviation.", 672405fe67fSCaroline Tice NULL) 673ebc09c36SJim Ingham { 674405fe67fSCaroline Tice CommandArgumentEntry arg; 675405fe67fSCaroline Tice CommandArgumentData alias_arg; 676405fe67fSCaroline Tice 677405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 678405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 679405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 680405fe67fSCaroline Tice 681405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 682405fe67fSCaroline Tice arg.push_back (alias_arg); 683405fe67fSCaroline Tice 684405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 685405fe67fSCaroline Tice m_arguments.push_back (arg); 686ebc09c36SJim Ingham } 687ebc09c36SJim Ingham 688ebc09c36SJim Ingham ~CommandObjectCommandsUnalias() 689ebc09c36SJim Ingham { 690ebc09c36SJim Ingham } 691ebc09c36SJim Ingham 692ebc09c36SJim Ingham 693ebc09c36SJim Ingham bool 694ebc09c36SJim Ingham Execute 695ebc09c36SJim Ingham ( 696ebc09c36SJim Ingham Args& args, 697ebc09c36SJim Ingham CommandReturnObject &result 698ebc09c36SJim Ingham ) 699ebc09c36SJim Ingham { 700ebc09c36SJim Ingham CommandObject::CommandMap::iterator pos; 701ebc09c36SJim Ingham CommandObject *cmd_obj; 702ebc09c36SJim Ingham 703ebc09c36SJim Ingham if (args.GetArgumentCount() != 0) 704ebc09c36SJim Ingham { 705ebc09c36SJim Ingham const char *command_name = args.GetArgumentAtIndex(0); 706a7015092SGreg Clayton cmd_obj = m_interpreter.GetCommandObject(command_name); 707ebc09c36SJim Ingham if (cmd_obj) 708ebc09c36SJim Ingham { 709a7015092SGreg Clayton if (m_interpreter.CommandExists (command_name)) 710ebc09c36SJim Ingham { 711ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n", 712ebc09c36SJim Ingham command_name); 713ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 714ebc09c36SJim Ingham } 715ebc09c36SJim Ingham else 716ebc09c36SJim Ingham { 717ebc09c36SJim Ingham 718a7015092SGreg Clayton if (m_interpreter.RemoveAlias (command_name) == false) 719ebc09c36SJim Ingham { 720a7015092SGreg Clayton if (m_interpreter.AliasExists (command_name)) 721ebc09c36SJim Ingham result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n", 722ebc09c36SJim Ingham command_name); 723ebc09c36SJim Ingham else 724ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name); 725ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 726ebc09c36SJim Ingham } 727ebc09c36SJim Ingham else 728ebc09c36SJim Ingham result.SetStatus (eReturnStatusSuccessFinishNoResult); 729ebc09c36SJim Ingham } 730ebc09c36SJim Ingham } 731ebc09c36SJim Ingham else 732ebc09c36SJim Ingham { 733ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a " 734ebc09c36SJim Ingham "current list of commands.\n", 735ebc09c36SJim Ingham command_name); 736ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 737ebc09c36SJim Ingham } 738ebc09c36SJim Ingham } 739ebc09c36SJim Ingham else 740ebc09c36SJim Ingham { 741ebc09c36SJim Ingham result.AppendError ("must call 'unalias' with a valid alias"); 742ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 743ebc09c36SJim Ingham } 744ebc09c36SJim Ingham 745ebc09c36SJim Ingham return result.Succeeded(); 746ebc09c36SJim Ingham } 747ebc09c36SJim Ingham }; 748ebc09c36SJim Ingham 749de164aaaSGreg Clayton #pragma mark CommandObjectCommandsAddRegex 750de164aaaSGreg Clayton //------------------------------------------------------------------------- 751de164aaaSGreg Clayton // CommandObjectCommandsAddRegex 752de164aaaSGreg Clayton //------------------------------------------------------------------------- 753de164aaaSGreg Clayton 754de164aaaSGreg Clayton class CommandObjectCommandsAddRegex : public CommandObject 755de164aaaSGreg Clayton { 756de164aaaSGreg Clayton public: 757de164aaaSGreg Clayton CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) : 758de164aaaSGreg Clayton CommandObject (interpreter, 7590e5e5a79SGreg Clayton "command regex", 760de164aaaSGreg Clayton "Allow the user to create a regular expression command.", 7610e5e5a79SGreg Clayton "command regex <cmd-name> [s/<regex>/<subst>/ ...]"), 762de164aaaSGreg Clayton m_options (interpreter) 763de164aaaSGreg Clayton { 7640e5e5a79SGreg Clayton SetHelpLong( 7650e5e5a79SGreg Clayton "This command allows the user to create powerful regular expression commands\n" 7660e5e5a79SGreg Clayton "with substitutions. The regular expressions and substitutions are specified\n" 7670e5e5a79SGreg Clayton "using the regular exression substitution format of:\n" 7680e5e5a79SGreg Clayton "\n" 7690e5e5a79SGreg Clayton " s/<regex>/<subst>/\n" 7700e5e5a79SGreg Clayton "\n" 7710e5e5a79SGreg Clayton "<regex> is a regular expression that can use parenthesis to capture regular\n" 7720e5e5a79SGreg Clayton "expression input and substitute the captured matches in the output using %1\n" 7730e5e5a79SGreg Clayton "for the first match, %2 for the second, and so on.\n" 7740e5e5a79SGreg Clayton "\n" 7750e5e5a79SGreg Clayton "The regular expressions can all be specified on the command line if more than\n" 7760e5e5a79SGreg Clayton "one argument is provided. If just the command name is provided on the command\n" 7770e5e5a79SGreg Clayton "line, then the regular expressions and substitutions can be entered on separate\n" 7780e5e5a79SGreg Clayton " lines, followed by an empty line to terminate the command definition.\n" 7790e5e5a79SGreg Clayton "\n" 7800e5e5a79SGreg Clayton "EXAMPLES\n" 7810e5e5a79SGreg Clayton "\n" 7820e5e5a79SGreg Clayton "The following example with define a regular expression command named 'f' that\n" 7830e5e5a79SGreg Clayton "will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if\n" 7840e5e5a79SGreg Clayton "a number follows 'f':\n" 7850e5e5a79SGreg Clayton "(lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/'\n" 7860e5e5a79SGreg Clayton ); 787de164aaaSGreg Clayton } 788de164aaaSGreg Clayton 789de164aaaSGreg Clayton ~CommandObjectCommandsAddRegex() 790de164aaaSGreg Clayton { 791de164aaaSGreg Clayton } 792de164aaaSGreg Clayton 793de164aaaSGreg Clayton 794de164aaaSGreg Clayton bool 795de164aaaSGreg Clayton Execute (Args& args, CommandReturnObject &result) 796de164aaaSGreg Clayton { 7970e5e5a79SGreg Clayton const size_t argc = args.GetArgumentCount(); 7980e5e5a79SGreg Clayton if (argc == 0) 799de164aaaSGreg Clayton { 80069c12ccbSJason Molenda result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n"); 8010e5e5a79SGreg Clayton result.SetStatus (eReturnStatusFailed); 8020e5e5a79SGreg Clayton } 8030e5e5a79SGreg Clayton else 8040e5e5a79SGreg Clayton { 8050e5e5a79SGreg Clayton Error error; 806de164aaaSGreg Clayton const char *name = args.GetArgumentAtIndex(0); 807de164aaaSGreg Clayton m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter, 808de164aaaSGreg Clayton name, 809de164aaaSGreg Clayton m_options.GetHelp (), 810de164aaaSGreg Clayton m_options.GetSyntax (), 811de164aaaSGreg Clayton 10)); 8120e5e5a79SGreg Clayton 8130e5e5a79SGreg Clayton if (argc == 1) 8140e5e5a79SGreg Clayton { 8150e5e5a79SGreg Clayton InputReaderSP reader_sp (new InputReader(m_interpreter.GetDebugger())); 816de164aaaSGreg Clayton if (reader_sp) 817de164aaaSGreg Clayton { 8180e5e5a79SGreg Clayton error =reader_sp->Initialize (CommandObjectCommandsAddRegex::InputReaderCallback, 819de164aaaSGreg Clayton this, // baton 820de164aaaSGreg Clayton eInputReaderGranularityLine, // token size, to pass to callback function 8210e5e5a79SGreg Clayton NULL, // end token 822de164aaaSGreg Clayton "> ", // prompt 8230e5e5a79SGreg Clayton true); // echo input 8240e5e5a79SGreg Clayton if (error.Success()) 825de164aaaSGreg Clayton { 826de164aaaSGreg Clayton m_interpreter.GetDebugger().PushInputReader (reader_sp); 827de164aaaSGreg Clayton result.SetStatus (eReturnStatusSuccessFinishNoResult); 8280e5e5a79SGreg Clayton return true; 829de164aaaSGreg Clayton } 830de164aaaSGreg Clayton } 831de164aaaSGreg Clayton } 832de164aaaSGreg Clayton else 833de164aaaSGreg Clayton { 8340e5e5a79SGreg Clayton for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx) 8350e5e5a79SGreg Clayton { 8360e5e5a79SGreg Clayton llvm::StringRef arg_strref (args.GetArgumentAtIndex(arg_idx)); 8370e5e5a79SGreg Clayton error = AppendRegexSubstitution (arg_strref); 8380e5e5a79SGreg Clayton if (error.Fail()) 8390e5e5a79SGreg Clayton break; 8400e5e5a79SGreg Clayton } 8410e5e5a79SGreg Clayton 8420e5e5a79SGreg Clayton if (error.Success()) 8430e5e5a79SGreg Clayton { 8440e5e5a79SGreg Clayton AddRegexCommandToInterpreter(); 8450e5e5a79SGreg Clayton } 8460e5e5a79SGreg Clayton } 8470e5e5a79SGreg Clayton if (error.Fail()) 8480e5e5a79SGreg Clayton { 8490e5e5a79SGreg Clayton result.AppendError (error.AsCString()); 850de164aaaSGreg Clayton result.SetStatus (eReturnStatusFailed); 851de164aaaSGreg Clayton } 8520e5e5a79SGreg Clayton } 8530e5e5a79SGreg Clayton 854de164aaaSGreg Clayton return result.Succeeded(); 855de164aaaSGreg Clayton } 856de164aaaSGreg Clayton 8570e5e5a79SGreg Clayton Error 8580e5e5a79SGreg Clayton AppendRegexSubstitution (const llvm::StringRef ®ex_sed) 859de164aaaSGreg Clayton { 8600e5e5a79SGreg Clayton Error error; 8610e5e5a79SGreg Clayton 8620e5e5a79SGreg Clayton if (m_regex_cmd_ap.get() == NULL) 863de164aaaSGreg Clayton { 8640e5e5a79SGreg Clayton error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'", 8650e5e5a79SGreg Clayton (int)regex_sed.size(), 8660e5e5a79SGreg Clayton regex_sed.data()); 8670e5e5a79SGreg Clayton return error; 868de164aaaSGreg Clayton } 8690e5e5a79SGreg Clayton 8700e5e5a79SGreg Clayton size_t regex_sed_size = regex_sed.size(); 8710e5e5a79SGreg Clayton 8720e5e5a79SGreg Clayton if (regex_sed_size <= 1) 8730e5e5a79SGreg Clayton { 8740e5e5a79SGreg Clayton error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'", 8750e5e5a79SGreg Clayton (int)regex_sed.size(), 8760e5e5a79SGreg Clayton regex_sed.data()); 8770e5e5a79SGreg Clayton return error; 8780e5e5a79SGreg Clayton } 8790e5e5a79SGreg Clayton 8800e5e5a79SGreg Clayton if (regex_sed[0] != 's') 8810e5e5a79SGreg Clayton { 8820e5e5a79SGreg Clayton error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'", 8830e5e5a79SGreg Clayton (int)regex_sed.size(), 8840e5e5a79SGreg Clayton regex_sed.data()); 8850e5e5a79SGreg Clayton return error; 8860e5e5a79SGreg Clayton } 8870e5e5a79SGreg Clayton const size_t first_separator_char_pos = 1; 8880e5e5a79SGreg Clayton // use the char that follows 's' as the regex separator character 8890e5e5a79SGreg Clayton // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|" 8900e5e5a79SGreg Clayton const char separator_char = regex_sed[first_separator_char_pos]; 8910e5e5a79SGreg Clayton const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1); 8920e5e5a79SGreg Clayton 8930e5e5a79SGreg Clayton if (second_separator_char_pos == std::string::npos) 8940e5e5a79SGreg Clayton { 8950e5e5a79SGreg Clayton error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s'", 8960e5e5a79SGreg Clayton separator_char, 8970e5e5a79SGreg Clayton (int)(regex_sed.size() - first_separator_char_pos - 1), 8980e5e5a79SGreg Clayton regex_sed.data() + (first_separator_char_pos + 1)); 8990e5e5a79SGreg Clayton return error; 9000e5e5a79SGreg Clayton } 9010e5e5a79SGreg Clayton 9020e5e5a79SGreg Clayton const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1); 9030e5e5a79SGreg Clayton 9040e5e5a79SGreg Clayton if (third_separator_char_pos == std::string::npos) 9050e5e5a79SGreg Clayton { 9060e5e5a79SGreg Clayton error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s'", 9070e5e5a79SGreg Clayton separator_char, 9080e5e5a79SGreg Clayton (int)(regex_sed.size() - second_separator_char_pos - 1), 9090e5e5a79SGreg Clayton regex_sed.data() + (second_separator_char_pos + 1)); 9100e5e5a79SGreg Clayton return error; 9110e5e5a79SGreg Clayton } 9120e5e5a79SGreg Clayton 9130e5e5a79SGreg Clayton if (third_separator_char_pos != regex_sed_size - 1) 9140e5e5a79SGreg Clayton { 9150e5e5a79SGreg Clayton // Make sure that everything that follows the last regex 9160e5e5a79SGreg Clayton // separator char 9170e5e5a79SGreg Clayton if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos) 9180e5e5a79SGreg Clayton { 9190e5e5a79SGreg Clayton error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'", 9200e5e5a79SGreg Clayton (int)third_separator_char_pos + 1, 9210e5e5a79SGreg Clayton regex_sed.data(), 9220e5e5a79SGreg Clayton (int)(regex_sed.size() - third_separator_char_pos - 1), 9230e5e5a79SGreg Clayton regex_sed.data() + (third_separator_char_pos + 1)); 9240e5e5a79SGreg Clayton return error; 9250e5e5a79SGreg Clayton } 9260e5e5a79SGreg Clayton 9270e5e5a79SGreg Clayton } 9280e5e5a79SGreg Clayton else if (first_separator_char_pos + 1 == second_separator_char_pos) 9290e5e5a79SGreg Clayton { 9300e5e5a79SGreg Clayton error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 9310e5e5a79SGreg Clayton separator_char, 9320e5e5a79SGreg Clayton separator_char, 9330e5e5a79SGreg Clayton separator_char, 9340e5e5a79SGreg Clayton (int)regex_sed.size(), 9350e5e5a79SGreg Clayton regex_sed.data()); 9360e5e5a79SGreg Clayton return error; 9370e5e5a79SGreg Clayton } 9380e5e5a79SGreg Clayton else if (second_separator_char_pos + 1 == third_separator_char_pos) 9390e5e5a79SGreg Clayton { 9400e5e5a79SGreg Clayton error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 9410e5e5a79SGreg Clayton separator_char, 9420e5e5a79SGreg Clayton separator_char, 9430e5e5a79SGreg Clayton separator_char, 9440e5e5a79SGreg Clayton (int)regex_sed.size(), 9450e5e5a79SGreg Clayton regex_sed.data()); 9460e5e5a79SGreg Clayton return error; 9470e5e5a79SGreg Clayton } 9480e5e5a79SGreg Clayton std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1)); 9490e5e5a79SGreg Clayton std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1)); 9500e5e5a79SGreg Clayton m_regex_cmd_ap->AddRegexCommand (regex.c_str(), 9510e5e5a79SGreg Clayton subst.c_str()); 9520e5e5a79SGreg Clayton return error; 953de164aaaSGreg Clayton } 954de164aaaSGreg Clayton 955de164aaaSGreg Clayton void 9560e5e5a79SGreg Clayton AddRegexCommandToInterpreter() 957de164aaaSGreg Clayton { 958de164aaaSGreg Clayton if (m_regex_cmd_ap.get()) 959de164aaaSGreg Clayton { 960de164aaaSGreg Clayton if (m_regex_cmd_ap->HasRegexEntries()) 961de164aaaSGreg Clayton { 962de164aaaSGreg Clayton CommandObjectSP cmd_sp (m_regex_cmd_ap.release()); 963de164aaaSGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 964de164aaaSGreg Clayton } 965de164aaaSGreg Clayton } 966de164aaaSGreg Clayton } 967de164aaaSGreg Clayton 9680e5e5a79SGreg Clayton void 9690e5e5a79SGreg Clayton InputReaderDidCancel() 9700e5e5a79SGreg Clayton { 9710e5e5a79SGreg Clayton m_regex_cmd_ap.reset(); 9720e5e5a79SGreg Clayton } 9730e5e5a79SGreg Clayton 974de164aaaSGreg Clayton static size_t 975de164aaaSGreg Clayton InputReaderCallback (void *baton, 976de164aaaSGreg Clayton InputReader &reader, 977de164aaaSGreg Clayton lldb::InputReaderAction notification, 978de164aaaSGreg Clayton const char *bytes, 979de164aaaSGreg Clayton size_t bytes_len); 980de164aaaSGreg Clayton private: 981de164aaaSGreg Clayton std::auto_ptr<CommandObjectRegexCommand> m_regex_cmd_ap; 982de164aaaSGreg Clayton 983de164aaaSGreg Clayton class CommandOptions : public Options 984de164aaaSGreg Clayton { 985de164aaaSGreg Clayton public: 986de164aaaSGreg Clayton 987de164aaaSGreg Clayton CommandOptions (CommandInterpreter &interpreter) : 988de164aaaSGreg Clayton Options (interpreter) 989de164aaaSGreg Clayton { 990de164aaaSGreg Clayton } 991de164aaaSGreg Clayton 992de164aaaSGreg Clayton virtual 993de164aaaSGreg Clayton ~CommandOptions (){} 994de164aaaSGreg Clayton 995de164aaaSGreg Clayton virtual Error 996de164aaaSGreg Clayton SetOptionValue (uint32_t option_idx, const char *option_arg) 997de164aaaSGreg Clayton { 998de164aaaSGreg Clayton Error error; 999de164aaaSGreg Clayton char short_option = (char) m_getopt_table[option_idx].val; 1000de164aaaSGreg Clayton 1001de164aaaSGreg Clayton switch (short_option) 1002de164aaaSGreg Clayton { 1003de164aaaSGreg Clayton case 'h': 1004de164aaaSGreg Clayton m_help.assign (option_arg); 1005de164aaaSGreg Clayton break; 1006de164aaaSGreg Clayton case 's': 1007de164aaaSGreg Clayton m_syntax.assign (option_arg); 1008de164aaaSGreg Clayton break; 1009de164aaaSGreg Clayton 1010de164aaaSGreg Clayton default: 101186edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 1012de164aaaSGreg Clayton break; 1013de164aaaSGreg Clayton } 1014de164aaaSGreg Clayton 1015de164aaaSGreg Clayton return error; 1016de164aaaSGreg Clayton } 1017de164aaaSGreg Clayton 1018de164aaaSGreg Clayton void 1019de164aaaSGreg Clayton OptionParsingStarting () 1020de164aaaSGreg Clayton { 1021de164aaaSGreg Clayton m_help.clear(); 1022de164aaaSGreg Clayton m_syntax.clear(); 1023de164aaaSGreg Clayton } 1024de164aaaSGreg Clayton 1025de164aaaSGreg Clayton const OptionDefinition* 1026de164aaaSGreg Clayton GetDefinitions () 1027de164aaaSGreg Clayton { 1028de164aaaSGreg Clayton return g_option_table; 1029de164aaaSGreg Clayton } 1030de164aaaSGreg Clayton 1031de164aaaSGreg Clayton // Options table: Required for subclasses of Options. 1032de164aaaSGreg Clayton 1033de164aaaSGreg Clayton static OptionDefinition g_option_table[]; 1034de164aaaSGreg Clayton 1035de164aaaSGreg Clayton const char * 1036de164aaaSGreg Clayton GetHelp () 1037de164aaaSGreg Clayton { 1038de164aaaSGreg Clayton if (m_help.empty()) 1039de164aaaSGreg Clayton return NULL; 1040de164aaaSGreg Clayton return m_help.c_str(); 1041de164aaaSGreg Clayton } 1042de164aaaSGreg Clayton const char * 1043de164aaaSGreg Clayton GetSyntax () 1044de164aaaSGreg Clayton { 1045de164aaaSGreg Clayton if (m_syntax.empty()) 1046de164aaaSGreg Clayton return NULL; 1047de164aaaSGreg Clayton return m_syntax.c_str(); 1048de164aaaSGreg Clayton } 1049de164aaaSGreg Clayton // Instance variables to hold the values for command options. 1050de164aaaSGreg Clayton protected: 1051de164aaaSGreg Clayton std::string m_help; 1052de164aaaSGreg Clayton std::string m_syntax; 1053de164aaaSGreg Clayton }; 1054de164aaaSGreg Clayton 1055de164aaaSGreg Clayton CommandOptions m_options; 1056de164aaaSGreg Clayton 1057de164aaaSGreg Clayton virtual Options * 1058de164aaaSGreg Clayton GetOptions () 1059de164aaaSGreg Clayton { 1060de164aaaSGreg Clayton return &m_options; 1061de164aaaSGreg Clayton } 1062de164aaaSGreg Clayton 1063de164aaaSGreg Clayton }; 1064de164aaaSGreg Clayton 1065de164aaaSGreg Clayton size_t 1066de164aaaSGreg Clayton CommandObjectCommandsAddRegex::InputReaderCallback (void *baton, 1067de164aaaSGreg Clayton InputReader &reader, 1068de164aaaSGreg Clayton lldb::InputReaderAction notification, 1069de164aaaSGreg Clayton const char *bytes, 1070de164aaaSGreg Clayton size_t bytes_len) 1071de164aaaSGreg Clayton { 1072de164aaaSGreg Clayton CommandObjectCommandsAddRegex *add_regex_cmd = (CommandObjectCommandsAddRegex *) baton; 1073d61c10bcSCaroline Tice bool batch_mode = reader.GetDebugger().GetCommandInterpreter().GetBatchCommandMode(); 1074de164aaaSGreg Clayton 1075de164aaaSGreg Clayton switch (notification) 1076de164aaaSGreg Clayton { 1077de164aaaSGreg Clayton case eInputReaderActivate: 1078d61c10bcSCaroline Tice if (!batch_mode) 107915356e7fSCaroline Tice { 108015356e7fSCaroline Tice StreamSP out_stream = reader.GetDebugger().GetAsyncOutputStream (); 108115356e7fSCaroline Tice out_stream->Printf("%s\n", "Enter regular expressions in the form 's/<regex>/<subst>/' and terminate with an empty line:"); 108215356e7fSCaroline Tice out_stream->Flush(); 108315356e7fSCaroline Tice } 1084de164aaaSGreg Clayton break; 1085de164aaaSGreg Clayton case eInputReaderReactivate: 1086de164aaaSGreg Clayton break; 1087de164aaaSGreg Clayton 1088de164aaaSGreg Clayton case eInputReaderDeactivate: 1089de164aaaSGreg Clayton break; 1090de164aaaSGreg Clayton 1091969ed3d1SCaroline Tice case eInputReaderAsynchronousOutputWritten: 1092969ed3d1SCaroline Tice break; 1093969ed3d1SCaroline Tice 1094de164aaaSGreg Clayton case eInputReaderGotToken: 10950e5e5a79SGreg Clayton while (bytes_len > 0 && (bytes[bytes_len-1] == '\r' || bytes[bytes_len-1] == '\n')) 10960e5e5a79SGreg Clayton --bytes_len; 1097de164aaaSGreg Clayton if (bytes_len == 0) 1098de164aaaSGreg Clayton reader.SetIsDone(true); 1099de164aaaSGreg Clayton else if (bytes) 1100de164aaaSGreg Clayton { 11010e5e5a79SGreg Clayton llvm::StringRef bytes_strref (bytes, bytes_len); 11020e5e5a79SGreg Clayton Error error (add_regex_cmd->AppendRegexSubstitution (bytes_strref)); 11030e5e5a79SGreg Clayton if (error.Fail()) 1104de164aaaSGreg Clayton { 1105d61c10bcSCaroline Tice if (!batch_mode) 1106d61c10bcSCaroline Tice { 110715356e7fSCaroline Tice StreamSP out_stream = reader.GetDebugger().GetAsyncOutputStream(); 110815356e7fSCaroline Tice out_stream->Printf("error: %s\n", error.AsCString()); 110915356e7fSCaroline Tice out_stream->Flush(); 1110d61c10bcSCaroline Tice } 11110e5e5a79SGreg Clayton add_regex_cmd->InputReaderDidCancel (); 11120e5e5a79SGreg Clayton reader.SetIsDone (true); 1113de164aaaSGreg Clayton } 1114de164aaaSGreg Clayton } 1115de164aaaSGreg Clayton break; 1116de164aaaSGreg Clayton 1117de164aaaSGreg Clayton case eInputReaderInterrupt: 111815356e7fSCaroline Tice { 1119de164aaaSGreg Clayton reader.SetIsDone (true); 1120d61c10bcSCaroline Tice if (!batch_mode) 1121d61c10bcSCaroline Tice { 112215356e7fSCaroline Tice StreamSP out_stream = reader.GetDebugger().GetAsyncOutputStream(); 112315356e7fSCaroline Tice out_stream->PutCString("Regular expression command creations was cancelled.\n"); 112415356e7fSCaroline Tice out_stream->Flush(); 1125d61c10bcSCaroline Tice } 11260e5e5a79SGreg Clayton add_regex_cmd->InputReaderDidCancel (); 112715356e7fSCaroline Tice } 1128de164aaaSGreg Clayton break; 1129de164aaaSGreg Clayton 1130de164aaaSGreg Clayton case eInputReaderEndOfFile: 1131de164aaaSGreg Clayton reader.SetIsDone (true); 1132de164aaaSGreg Clayton break; 1133de164aaaSGreg Clayton 1134de164aaaSGreg Clayton case eInputReaderDone: 11350e5e5a79SGreg Clayton add_regex_cmd->AddRegexCommandToInterpreter(); 1136de164aaaSGreg Clayton break; 1137de164aaaSGreg Clayton } 1138de164aaaSGreg Clayton 1139de164aaaSGreg Clayton return bytes_len; 1140de164aaaSGreg Clayton } 1141de164aaaSGreg Clayton 1142de164aaaSGreg Clayton 1143de164aaaSGreg Clayton OptionDefinition 1144de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] = 1145de164aaaSGreg Clayton { 1146de164aaaSGreg Clayton { LLDB_OPT_SET_1, false, "help" , 'h', required_argument, NULL, 0, eArgTypeNone, "The help text to display for this command."}, 1147de164aaaSGreg Clayton { LLDB_OPT_SET_1, false, "syntax", 's', required_argument, NULL, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."}, 1148de164aaaSGreg Clayton { 0 , false, NULL , 0 , 0 , NULL, 0, eArgTypeNone, NULL } 1149de164aaaSGreg Clayton }; 1150de164aaaSGreg Clayton 1151de164aaaSGreg Clayton 1152223383edSEnrico Granata class CommandObjectPythonFunction : public CommandObject 1153223383edSEnrico Granata { 1154223383edSEnrico Granata private: 1155223383edSEnrico Granata std::string m_function_name; 11560a305db7SEnrico Granata ScriptedCommandSynchronicity m_synchro; 1157223383edSEnrico Granata 1158223383edSEnrico Granata public: 1159223383edSEnrico Granata 1160223383edSEnrico Granata CommandObjectPythonFunction (CommandInterpreter &interpreter, 1161223383edSEnrico Granata std::string name, 11620a305db7SEnrico Granata std::string funct, 11630a305db7SEnrico Granata ScriptedCommandSynchronicity synch) : 1164223383edSEnrico Granata CommandObject (interpreter, 1165223383edSEnrico Granata name.c_str(), 1166223383edSEnrico Granata (std::string("Run Python function ") + funct).c_str(), 1167223383edSEnrico Granata NULL), 11680a305db7SEnrico Granata m_function_name(funct), 11690a305db7SEnrico Granata m_synchro(synch) 1170223383edSEnrico Granata { 117199f0b8f9SEnrico Granata ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter(); 117299f0b8f9SEnrico Granata if (scripter) 117399f0b8f9SEnrico Granata { 117499f0b8f9SEnrico Granata std::string docstring = scripter->GetDocumentationForItem(funct.c_str()); 117599f0b8f9SEnrico Granata if (!docstring.empty()) 117699f0b8f9SEnrico Granata SetHelpLong(docstring); 117799f0b8f9SEnrico Granata } 1178223383edSEnrico Granata } 1179223383edSEnrico Granata 1180223383edSEnrico Granata virtual 1181223383edSEnrico Granata ~CommandObjectPythonFunction () 1182223383edSEnrico Granata { 1183223383edSEnrico Granata } 1184223383edSEnrico Granata 1185223383edSEnrico Granata virtual bool 1186223383edSEnrico Granata ExecuteRawCommandString (const char *raw_command_line, CommandReturnObject &result) 1187223383edSEnrico Granata { 1188223383edSEnrico Granata ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter(); 1189223383edSEnrico Granata 1190223383edSEnrico Granata Error error; 1191223383edSEnrico Granata 1192223383edSEnrico Granata if (!scripter || scripter->RunScriptBasedCommand(m_function_name.c_str(), 1193223383edSEnrico Granata raw_command_line, 11940a305db7SEnrico Granata m_synchro, 1195223383edSEnrico Granata result, 1196223383edSEnrico Granata error) == false) 1197223383edSEnrico Granata { 1198223383edSEnrico Granata result.AppendError(error.AsCString()); 1199223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1200223383edSEnrico Granata } 1201223383edSEnrico Granata else 1202223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1203223383edSEnrico Granata 1204223383edSEnrico Granata return result.Succeeded(); 1205223383edSEnrico Granata } 1206223383edSEnrico Granata 1207223383edSEnrico Granata virtual bool 1208223383edSEnrico Granata WantsRawCommandString () 1209223383edSEnrico Granata { 1210223383edSEnrico Granata return true; 1211223383edSEnrico Granata } 1212223383edSEnrico Granata 1213223383edSEnrico Granata bool 1214223383edSEnrico Granata Execute (Args& command, 1215223383edSEnrico Granata CommandReturnObject &result) 1216223383edSEnrico Granata { 1217223383edSEnrico Granata std::string cmd_string; 1218223383edSEnrico Granata command.GetCommandString(cmd_string); 1219223383edSEnrico Granata return ExecuteRawCommandString(cmd_string.c_str(), result); 1220223383edSEnrico Granata } 1221223383edSEnrico Granata 1222223383edSEnrico Granata virtual bool 12230a305db7SEnrico Granata IsRemovable () 12240a305db7SEnrico Granata { 12250a305db7SEnrico Granata return true; 12260a305db7SEnrico Granata } 12270a305db7SEnrico Granata 12280a305db7SEnrico Granata const std::string& 12290a305db7SEnrico Granata GetFunctionName () 12300a305db7SEnrico Granata { 12310a305db7SEnrico Granata return m_function_name; 12320a305db7SEnrico Granata } 12330a305db7SEnrico Granata 12340a305db7SEnrico Granata ScriptedCommandSynchronicity 12350a305db7SEnrico Granata GetSynchronicity () 12360a305db7SEnrico Granata { 12370a305db7SEnrico Granata return m_synchro; 12380a305db7SEnrico Granata } 1239223383edSEnrico Granata 1240223383edSEnrico Granata }; 1241223383edSEnrico Granata 1242a9dbf432SEnrico Granata //------------------------------------------------------------------------- 1243a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport 1244a9dbf432SEnrico Granata //------------------------------------------------------------------------- 1245a9dbf432SEnrico Granata 1246a9dbf432SEnrico Granata class CommandObjectCommandsScriptImport : public CommandObject 1247a9dbf432SEnrico Granata { 12480a305db7SEnrico Granata private: 12490a305db7SEnrico Granata 12500a305db7SEnrico Granata class CommandOptions : public Options 12510a305db7SEnrico Granata { 12520a305db7SEnrico Granata public: 12530a305db7SEnrico Granata 12540a305db7SEnrico Granata CommandOptions (CommandInterpreter &interpreter) : 12550a305db7SEnrico Granata Options (interpreter) 12560a305db7SEnrico Granata { 12570a305db7SEnrico Granata } 12580a305db7SEnrico Granata 12590a305db7SEnrico Granata virtual 12600a305db7SEnrico Granata ~CommandOptions (){} 12610a305db7SEnrico Granata 12620a305db7SEnrico Granata virtual Error 12630a305db7SEnrico Granata SetOptionValue (uint32_t option_idx, const char *option_arg) 12640a305db7SEnrico Granata { 12650a305db7SEnrico Granata Error error; 12660a305db7SEnrico Granata char short_option = (char) m_getopt_table[option_idx].val; 12670a305db7SEnrico Granata 12680a305db7SEnrico Granata switch (short_option) 12690a305db7SEnrico Granata { 12700a305db7SEnrico Granata case 'r': 12710a305db7SEnrico Granata m_allow_reload = true; 12720a305db7SEnrico Granata break; 12730a305db7SEnrico Granata default: 12740a305db7SEnrico Granata error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 12750a305db7SEnrico Granata break; 12760a305db7SEnrico Granata } 12770a305db7SEnrico Granata 12780a305db7SEnrico Granata return error; 12790a305db7SEnrico Granata } 12800a305db7SEnrico Granata 12810a305db7SEnrico Granata void 12820a305db7SEnrico Granata OptionParsingStarting () 12830a305db7SEnrico Granata { 12840a305db7SEnrico Granata m_allow_reload = false; 12850a305db7SEnrico Granata } 12860a305db7SEnrico Granata 12870a305db7SEnrico Granata const OptionDefinition* 12880a305db7SEnrico Granata GetDefinitions () 12890a305db7SEnrico Granata { 12900a305db7SEnrico Granata return g_option_table; 12910a305db7SEnrico Granata } 12920a305db7SEnrico Granata 12930a305db7SEnrico Granata // Options table: Required for subclasses of Options. 12940a305db7SEnrico Granata 12950a305db7SEnrico Granata static OptionDefinition g_option_table[]; 12960a305db7SEnrico Granata 12970a305db7SEnrico Granata // Instance variables to hold the values for command options. 12980a305db7SEnrico Granata 12990a305db7SEnrico Granata bool m_allow_reload; 13000a305db7SEnrico Granata }; 13010a305db7SEnrico Granata 13020a305db7SEnrico Granata CommandOptions m_options; 13030a305db7SEnrico Granata 13040a305db7SEnrico Granata virtual Options * 13050a305db7SEnrico Granata GetOptions () 13060a305db7SEnrico Granata { 13070a305db7SEnrico Granata return &m_options; 13080a305db7SEnrico Granata } 13090a305db7SEnrico Granata 1310a9dbf432SEnrico Granata public: 1311a9dbf432SEnrico Granata CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) : 1312a9dbf432SEnrico Granata CommandObject (interpreter, 1313a9dbf432SEnrico Granata "command script import", 1314a9dbf432SEnrico Granata "Import a scripting module in LLDB.", 13150a305db7SEnrico Granata NULL), 13160a305db7SEnrico Granata m_options(interpreter) 1317a9dbf432SEnrico Granata { 1318a9dbf432SEnrico Granata CommandArgumentEntry arg1; 1319a9dbf432SEnrico Granata CommandArgumentData cmd_arg; 1320a9dbf432SEnrico Granata 1321a9dbf432SEnrico Granata // Define the first (and only) variant of this arg. 13220a305db7SEnrico Granata cmd_arg.arg_type = eArgTypeFilename; 1323a9dbf432SEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1324a9dbf432SEnrico Granata 1325a9dbf432SEnrico Granata // There is only one variant this argument could be; put it into the argument entry. 1326a9dbf432SEnrico Granata arg1.push_back (cmd_arg); 1327a9dbf432SEnrico Granata 1328a9dbf432SEnrico Granata // Push the data for the first argument into the m_arguments vector. 1329a9dbf432SEnrico Granata m_arguments.push_back (arg1); 1330a9dbf432SEnrico Granata } 1331a9dbf432SEnrico Granata 1332a9dbf432SEnrico Granata ~CommandObjectCommandsScriptImport () 1333a9dbf432SEnrico Granata { 1334a9dbf432SEnrico Granata } 1335a9dbf432SEnrico Granata 1336a9dbf432SEnrico Granata bool 1337a9dbf432SEnrico Granata Execute 1338a9dbf432SEnrico Granata ( 1339a9dbf432SEnrico Granata Args& args, 1340a9dbf432SEnrico Granata CommandReturnObject &result 1341a9dbf432SEnrico Granata ) 1342a9dbf432SEnrico Granata { 1343a9dbf432SEnrico Granata 1344a9dbf432SEnrico Granata if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) 1345a9dbf432SEnrico Granata { 1346a9dbf432SEnrico Granata result.AppendError ("only scripting language supported for module importing is currently Python"); 1347a9dbf432SEnrico Granata result.SetStatus (eReturnStatusFailed); 1348a9dbf432SEnrico Granata return false; 1349a9dbf432SEnrico Granata } 1350a9dbf432SEnrico Granata 1351a9dbf432SEnrico Granata size_t argc = args.GetArgumentCount(); 1352a9dbf432SEnrico Granata 1353a9dbf432SEnrico Granata if (argc != 1) 1354a9dbf432SEnrico Granata { 1355a9dbf432SEnrico Granata result.AppendError ("'command script import' requires one argument"); 1356a9dbf432SEnrico Granata result.SetStatus (eReturnStatusFailed); 1357a9dbf432SEnrico Granata return false; 1358a9dbf432SEnrico Granata } 1359a9dbf432SEnrico Granata 1360a9dbf432SEnrico Granata std::string path = args.GetArgumentAtIndex(0); 1361a9dbf432SEnrico Granata Error error; 1362a9dbf432SEnrico Granata 1363a9dbf432SEnrico Granata if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(), 13640a305db7SEnrico Granata m_options.m_allow_reload, 1365a9dbf432SEnrico Granata error)) 1366a9dbf432SEnrico Granata { 1367a9dbf432SEnrico Granata result.SetStatus (eReturnStatusSuccessFinishNoResult); 1368a9dbf432SEnrico Granata } 1369a9dbf432SEnrico Granata else 1370a9dbf432SEnrico Granata { 1371a9dbf432SEnrico Granata result.AppendErrorWithFormat("module importing failed: %s", error.AsCString()); 1372a9dbf432SEnrico Granata result.SetStatus (eReturnStatusFailed); 1373a9dbf432SEnrico Granata } 1374a9dbf432SEnrico Granata 1375a9dbf432SEnrico Granata return result.Succeeded(); 1376a9dbf432SEnrico Granata } 13770a305db7SEnrico Granata 13780a305db7SEnrico Granata int 13790a305db7SEnrico Granata HandleArgumentCompletion (Args &input, 13800a305db7SEnrico Granata int &cursor_index, 13810a305db7SEnrico Granata int &cursor_char_position, 13820a305db7SEnrico Granata OptionElementVector &opt_element_vector, 13830a305db7SEnrico Granata int match_start_point, 13840a305db7SEnrico Granata int max_return_elements, 13850a305db7SEnrico Granata bool &word_complete, 13860a305db7SEnrico Granata StringList &matches) 13870a305db7SEnrico Granata { 13880a305db7SEnrico Granata std::string completion_str (input.GetArgumentAtIndex(cursor_index)); 13890a305db7SEnrico Granata completion_str.erase (cursor_char_position); 13900a305db7SEnrico Granata 13910a305db7SEnrico Granata CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter, 13920a305db7SEnrico Granata CommandCompletions::eDiskFileCompletion, 13930a305db7SEnrico Granata completion_str.c_str(), 13940a305db7SEnrico Granata match_start_point, 13950a305db7SEnrico Granata max_return_elements, 13960a305db7SEnrico Granata NULL, 13970a305db7SEnrico Granata word_complete, 13980a305db7SEnrico Granata matches); 13990a305db7SEnrico Granata return matches.GetSize(); 14000a305db7SEnrico Granata } 1401a9dbf432SEnrico Granata }; 1402223383edSEnrico Granata 14030a305db7SEnrico Granata OptionDefinition 14040a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] = 14050a305db7SEnrico Granata { 14060a305db7SEnrico Granata { LLDB_OPT_SET_1, false, "allow-reload", 'r', no_argument, NULL, 0, eArgTypeNone, "Allow the script to be loaded even if it was already loaded before (for Python, the __lldb_init_module function will be called again, but the module will not be reloaded from disk)."}, 14070a305db7SEnrico Granata { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 14080a305db7SEnrico Granata }; 14090a305db7SEnrico Granata 14100a305db7SEnrico Granata 1411223383edSEnrico Granata //------------------------------------------------------------------------- 1412223383edSEnrico Granata // CommandObjectCommandsScriptAdd 1413223383edSEnrico Granata //------------------------------------------------------------------------- 1414223383edSEnrico Granata 1415223383edSEnrico Granata class CommandObjectCommandsScriptAdd : public CommandObject 1416223383edSEnrico Granata { 1417223383edSEnrico Granata private: 1418223383edSEnrico Granata 1419223383edSEnrico Granata class CommandOptions : public Options 1420223383edSEnrico Granata { 1421223383edSEnrico Granata public: 1422223383edSEnrico Granata 1423223383edSEnrico Granata CommandOptions (CommandInterpreter &interpreter) : 1424223383edSEnrico Granata Options (interpreter) 1425223383edSEnrico Granata { 1426223383edSEnrico Granata } 1427223383edSEnrico Granata 1428223383edSEnrico Granata virtual 1429223383edSEnrico Granata ~CommandOptions (){} 1430223383edSEnrico Granata 1431223383edSEnrico Granata virtual Error 1432223383edSEnrico Granata SetOptionValue (uint32_t option_idx, const char *option_arg) 1433223383edSEnrico Granata { 1434223383edSEnrico Granata Error error; 1435223383edSEnrico Granata char short_option = (char) m_getopt_table[option_idx].val; 1436223383edSEnrico Granata 1437223383edSEnrico Granata switch (short_option) 1438223383edSEnrico Granata { 1439223383edSEnrico Granata case 'f': 1440223383edSEnrico Granata m_funct_name = std::string(option_arg); 1441223383edSEnrico Granata break; 14420a305db7SEnrico Granata case 's': 14430a305db7SEnrico Granata m_synchronous = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error); 14440a305db7SEnrico Granata if (!error.Success()) 14450a305db7SEnrico Granata error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg); 14460a305db7SEnrico Granata break; 1447223383edSEnrico Granata default: 144886edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 1449223383edSEnrico Granata break; 1450223383edSEnrico Granata } 1451223383edSEnrico Granata 1452223383edSEnrico Granata return error; 1453223383edSEnrico Granata } 1454223383edSEnrico Granata 1455223383edSEnrico Granata void 1456223383edSEnrico Granata OptionParsingStarting () 1457223383edSEnrico Granata { 1458223383edSEnrico Granata m_funct_name = ""; 14590a305db7SEnrico Granata m_synchronous = eScriptedCommandSynchronicitySynchronous; 1460223383edSEnrico Granata } 1461223383edSEnrico Granata 1462223383edSEnrico Granata const OptionDefinition* 1463223383edSEnrico Granata GetDefinitions () 1464223383edSEnrico Granata { 1465223383edSEnrico Granata return g_option_table; 1466223383edSEnrico Granata } 1467223383edSEnrico Granata 1468223383edSEnrico Granata // Options table: Required for subclasses of Options. 1469223383edSEnrico Granata 1470223383edSEnrico Granata static OptionDefinition g_option_table[]; 1471223383edSEnrico Granata 1472223383edSEnrico Granata // Instance variables to hold the values for command options. 1473223383edSEnrico Granata 1474223383edSEnrico Granata std::string m_funct_name; 14750a305db7SEnrico Granata ScriptedCommandSynchronicity m_synchronous; 1476223383edSEnrico Granata }; 1477223383edSEnrico Granata 1478223383edSEnrico Granata CommandOptions m_options; 1479223383edSEnrico Granata 1480223383edSEnrico Granata virtual Options * 1481223383edSEnrico Granata GetOptions () 1482223383edSEnrico Granata { 1483223383edSEnrico Granata return &m_options; 1484223383edSEnrico Granata } 1485223383edSEnrico Granata 1486223383edSEnrico Granata class PythonAliasReader : public InputReaderEZ 1487223383edSEnrico Granata { 1488223383edSEnrico Granata private: 1489223383edSEnrico Granata CommandInterpreter& m_interpreter; 1490223383edSEnrico Granata std::string m_cmd_name; 14910a305db7SEnrico Granata ScriptedCommandSynchronicity m_synchronous; 1492223383edSEnrico Granata StringList m_user_input; 1493223383edSEnrico Granata DISALLOW_COPY_AND_ASSIGN (PythonAliasReader); 1494223383edSEnrico Granata public: 1495223383edSEnrico Granata PythonAliasReader(Debugger& debugger, 1496223383edSEnrico Granata CommandInterpreter& interpreter, 14970a305db7SEnrico Granata std::string cmd_name, 14980a305db7SEnrico Granata ScriptedCommandSynchronicity synch) : 1499223383edSEnrico Granata InputReaderEZ(debugger), 1500223383edSEnrico Granata m_interpreter(interpreter), 1501223383edSEnrico Granata m_cmd_name(cmd_name), 15020a305db7SEnrico Granata m_synchronous(synch), 1503223383edSEnrico Granata m_user_input() 1504223383edSEnrico Granata {} 1505223383edSEnrico Granata 1506223383edSEnrico Granata virtual 1507223383edSEnrico Granata ~PythonAliasReader() 1508223383edSEnrico Granata { 1509223383edSEnrico Granata } 1510223383edSEnrico Granata 1511223383edSEnrico Granata virtual void ActivateHandler(HandlerData& data) 1512223383edSEnrico Granata { 1513223383edSEnrico Granata StreamSP out_stream = data.GetOutStream(); 1514223383edSEnrico Granata bool batch_mode = data.GetBatchMode(); 1515223383edSEnrico Granata if (!batch_mode) 1516223383edSEnrico Granata { 1517223383edSEnrico Granata out_stream->Printf ("%s\n", g_python_command_instructions); 1518223383edSEnrico Granata if (data.reader.GetPrompt()) 1519223383edSEnrico Granata out_stream->Printf ("%s", data.reader.GetPrompt()); 1520223383edSEnrico Granata out_stream->Flush(); 1521223383edSEnrico Granata } 1522223383edSEnrico Granata } 1523223383edSEnrico Granata 1524223383edSEnrico Granata virtual void ReactivateHandler(HandlerData& data) 1525223383edSEnrico Granata { 1526223383edSEnrico Granata StreamSP out_stream = data.GetOutStream(); 1527223383edSEnrico Granata bool batch_mode = data.GetBatchMode(); 1528223383edSEnrico Granata if (data.reader.GetPrompt() && !batch_mode) 1529223383edSEnrico Granata { 1530223383edSEnrico Granata out_stream->Printf ("%s", data.reader.GetPrompt()); 1531223383edSEnrico Granata out_stream->Flush(); 1532223383edSEnrico Granata } 1533223383edSEnrico Granata } 1534223383edSEnrico Granata virtual void GotTokenHandler(HandlerData& data) 1535223383edSEnrico Granata { 1536223383edSEnrico Granata StreamSP out_stream = data.GetOutStream(); 1537223383edSEnrico Granata bool batch_mode = data.GetBatchMode(); 1538223383edSEnrico Granata if (data.bytes && data.bytes_len) 1539223383edSEnrico Granata { 1540223383edSEnrico Granata m_user_input.AppendString(data.bytes, data.bytes_len); 1541223383edSEnrico Granata } 1542223383edSEnrico Granata if (!data.reader.IsDone() && data.reader.GetPrompt() && !batch_mode) 1543223383edSEnrico Granata { 1544223383edSEnrico Granata out_stream->Printf ("%s", data.reader.GetPrompt()); 1545223383edSEnrico Granata out_stream->Flush(); 1546223383edSEnrico Granata } 1547223383edSEnrico Granata } 1548223383edSEnrico Granata virtual void InterruptHandler(HandlerData& data) 1549223383edSEnrico Granata { 1550223383edSEnrico Granata StreamSP out_stream = data.GetOutStream(); 1551223383edSEnrico Granata bool batch_mode = data.GetBatchMode(); 1552223383edSEnrico Granata data.reader.SetIsDone (true); 1553223383edSEnrico Granata if (!batch_mode) 1554223383edSEnrico Granata { 15550a305db7SEnrico Granata out_stream->Printf ("Warning: No script attached.\n"); 1556223383edSEnrico Granata out_stream->Flush(); 1557223383edSEnrico Granata } 1558223383edSEnrico Granata } 1559223383edSEnrico Granata virtual void EOFHandler(HandlerData& data) 1560223383edSEnrico Granata { 1561223383edSEnrico Granata data.reader.SetIsDone (true); 1562223383edSEnrico Granata } 1563223383edSEnrico Granata virtual void DoneHandler(HandlerData& data) 1564223383edSEnrico Granata { 1565223383edSEnrico Granata StreamSP out_stream = data.GetOutStream(); 1566223383edSEnrico Granata 1567223383edSEnrico Granata ScriptInterpreter *interpreter = data.reader.GetDebugger().GetCommandInterpreter().GetScriptInterpreter(); 1568223383edSEnrico Granata if (!interpreter) 1569223383edSEnrico Granata { 15700a305db7SEnrico Granata out_stream->Printf ("Script interpreter missing: no script attached.\n"); 1571223383edSEnrico Granata out_stream->Flush(); 1572223383edSEnrico Granata return; 1573223383edSEnrico Granata } 1574*a73b7df7SEnrico Granata std::string funct_name_str; 1575223383edSEnrico Granata if (!interpreter->GenerateScriptAliasFunction (m_user_input, 1576*a73b7df7SEnrico Granata funct_name_str)) 1577223383edSEnrico Granata { 15780a305db7SEnrico Granata out_stream->Printf ("Unable to create function: no script attached.\n"); 1579223383edSEnrico Granata out_stream->Flush(); 1580223383edSEnrico Granata return; 1581223383edSEnrico Granata } 1582*a73b7df7SEnrico Granata if (funct_name_str.empty()) 1583223383edSEnrico Granata { 15840a305db7SEnrico Granata out_stream->Printf ("Unable to obtain a function name: no script attached.\n"); 1585223383edSEnrico Granata out_stream->Flush(); 1586223383edSEnrico Granata return; 1587223383edSEnrico Granata } 1588223383edSEnrico Granata // everything should be fine now, let's add this alias 1589223383edSEnrico Granata 1590223383edSEnrico Granata CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(m_interpreter, 1591223383edSEnrico Granata m_cmd_name, 1592*a73b7df7SEnrico Granata funct_name_str.c_str(), 15930a305db7SEnrico Granata m_synchronous)); 1594223383edSEnrico Granata 15950a305db7SEnrico Granata if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true)) 1596223383edSEnrico Granata { 15970a305db7SEnrico Granata out_stream->Printf ("Unable to add selected command: no script attached.\n"); 1598223383edSEnrico Granata out_stream->Flush(); 1599223383edSEnrico Granata return; 1600223383edSEnrico Granata } 1601223383edSEnrico Granata } 1602223383edSEnrico Granata }; 1603223383edSEnrico Granata 1604223383edSEnrico Granata public: 1605223383edSEnrico Granata CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) : 1606223383edSEnrico Granata CommandObject (interpreter, 1607223383edSEnrico Granata "command script add", 1608223383edSEnrico Granata "Add a scripted function as an LLDB command.", 1609223383edSEnrico Granata NULL), 1610223383edSEnrico Granata m_options (interpreter) 1611223383edSEnrico Granata { 1612223383edSEnrico Granata CommandArgumentEntry arg1; 1613223383edSEnrico Granata CommandArgumentData cmd_arg; 1614223383edSEnrico Granata 1615223383edSEnrico Granata // Define the first (and only) variant of this arg. 1616223383edSEnrico Granata cmd_arg.arg_type = eArgTypeCommandName; 1617223383edSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1618223383edSEnrico Granata 1619223383edSEnrico Granata // There is only one variant this argument could be; put it into the argument entry. 1620223383edSEnrico Granata arg1.push_back (cmd_arg); 1621223383edSEnrico Granata 1622223383edSEnrico Granata // Push the data for the first argument into the m_arguments vector. 1623223383edSEnrico Granata m_arguments.push_back (arg1); 1624223383edSEnrico Granata } 1625223383edSEnrico Granata 1626223383edSEnrico Granata ~CommandObjectCommandsScriptAdd () 1627223383edSEnrico Granata { 1628223383edSEnrico Granata } 1629223383edSEnrico Granata 1630223383edSEnrico Granata bool 1631223383edSEnrico Granata Execute 1632223383edSEnrico Granata ( 1633223383edSEnrico Granata Args& args, 1634223383edSEnrico Granata CommandReturnObject &result 1635223383edSEnrico Granata ) 1636223383edSEnrico Granata { 163799f0b8f9SEnrico Granata 163899f0b8f9SEnrico Granata if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) 163999f0b8f9SEnrico Granata { 164099f0b8f9SEnrico Granata result.AppendError ("only scripting language supported for scripted commands is currently Python"); 164199f0b8f9SEnrico Granata result.SetStatus (eReturnStatusFailed); 164299f0b8f9SEnrico Granata return false; 164399f0b8f9SEnrico Granata } 164499f0b8f9SEnrico Granata 1645223383edSEnrico Granata size_t argc = args.GetArgumentCount(); 1646223383edSEnrico Granata 1647223383edSEnrico Granata if (argc != 1) 1648223383edSEnrico Granata { 1649223383edSEnrico Granata result.AppendError ("'command script add' requires one argument"); 1650223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1651223383edSEnrico Granata return false; 1652223383edSEnrico Granata } 1653223383edSEnrico Granata 1654223383edSEnrico Granata std::string cmd_name = args.GetArgumentAtIndex(0); 1655223383edSEnrico Granata 1656223383edSEnrico Granata if (m_options.m_funct_name.empty()) 1657223383edSEnrico Granata { 1658223383edSEnrico Granata InputReaderSP reader_sp (new PythonAliasReader (m_interpreter.GetDebugger(), 1659223383edSEnrico Granata m_interpreter, 16600a305db7SEnrico Granata cmd_name, 16610a305db7SEnrico Granata m_options.m_synchronous)); 1662223383edSEnrico Granata 1663223383edSEnrico Granata if (reader_sp) 1664223383edSEnrico Granata { 1665223383edSEnrico Granata 1666223383edSEnrico Granata InputReaderEZ::InitializationParameters ipr; 1667223383edSEnrico Granata 1668223383edSEnrico Granata Error err (reader_sp->Initialize (ipr.SetBaton(NULL).SetPrompt(" "))); 1669223383edSEnrico Granata if (err.Success()) 1670223383edSEnrico Granata { 1671223383edSEnrico Granata m_interpreter.GetDebugger().PushInputReader (reader_sp); 1672223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishNoResult); 1673223383edSEnrico Granata } 1674223383edSEnrico Granata else 1675223383edSEnrico Granata { 1676223383edSEnrico Granata result.AppendError (err.AsCString()); 1677223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1678223383edSEnrico Granata } 1679223383edSEnrico Granata } 1680223383edSEnrico Granata else 1681223383edSEnrico Granata { 1682223383edSEnrico Granata result.AppendError("out of memory"); 1683223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1684223383edSEnrico Granata } 1685223383edSEnrico Granata } 1686223383edSEnrico Granata else 1687223383edSEnrico Granata { 16880a305db7SEnrico Granata CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter, 16890a305db7SEnrico Granata cmd_name, 16900a305db7SEnrico Granata m_options.m_funct_name, 16910a305db7SEnrico Granata m_options.m_synchronous)); 16920a305db7SEnrico Granata if (m_interpreter.AddUserCommand(cmd_name, new_cmd, true)) 1693223383edSEnrico Granata { 1694223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishNoResult); 1695223383edSEnrico Granata } 1696223383edSEnrico Granata else 1697223383edSEnrico Granata { 1698223383edSEnrico Granata result.AppendError("cannot add command"); 1699223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1700223383edSEnrico Granata } 1701223383edSEnrico Granata } 1702223383edSEnrico Granata 1703223383edSEnrico Granata return result.Succeeded(); 1704223383edSEnrico Granata 1705223383edSEnrico Granata } 1706223383edSEnrico Granata }; 1707223383edSEnrico Granata 17080a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] = 17090a305db7SEnrico Granata { 17100a305db7SEnrico Granata { eScriptedCommandSynchronicitySynchronous, "synchronous", "Run synchronous"}, 17110a305db7SEnrico Granata { eScriptedCommandSynchronicityAsynchronous, "asynchronous", "Run asynchronous"}, 17120a305db7SEnrico Granata { eScriptedCommandSynchronicityCurrentValue, "current", "Do not alter current setting"}, 17130a305db7SEnrico Granata { 0, NULL, NULL } 17140a305db7SEnrico Granata }; 17150a305db7SEnrico Granata 1716223383edSEnrico Granata OptionDefinition 1717223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] = 1718223383edSEnrico Granata { 17199128ee2fSEnrico Granata { LLDB_OPT_SET_1, false, "function", 'f', required_argument, NULL, 0, eArgTypePythonFunction, "Name of the Python function to bind to this command name."}, 17200a305db7SEnrico Granata { LLDB_OPT_SET_1, false, "synchronicity", 's', required_argument, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity, "Set the synchronicity of this command's executions with regard to LLDB event system."}, 1721223383edSEnrico Granata { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 1722223383edSEnrico Granata }; 1723223383edSEnrico Granata 1724223383edSEnrico Granata //------------------------------------------------------------------------- 1725223383edSEnrico Granata // CommandObjectCommandsScriptList 1726223383edSEnrico Granata //------------------------------------------------------------------------- 1727223383edSEnrico Granata 1728223383edSEnrico Granata class CommandObjectCommandsScriptList : public CommandObject 1729223383edSEnrico Granata { 1730223383edSEnrico Granata private: 1731223383edSEnrico Granata 1732223383edSEnrico Granata public: 1733223383edSEnrico Granata CommandObjectCommandsScriptList(CommandInterpreter &interpreter) : 1734223383edSEnrico Granata CommandObject (interpreter, 1735223383edSEnrico Granata "command script list", 1736223383edSEnrico Granata "List defined scripted commands.", 1737223383edSEnrico Granata NULL) 1738223383edSEnrico Granata { 1739223383edSEnrico Granata } 1740223383edSEnrico Granata 1741223383edSEnrico Granata ~CommandObjectCommandsScriptList () 1742223383edSEnrico Granata { 1743223383edSEnrico Granata } 1744223383edSEnrico Granata 1745223383edSEnrico Granata bool 1746223383edSEnrico Granata Execute 1747223383edSEnrico Granata ( 1748223383edSEnrico Granata Args& args, 1749223383edSEnrico Granata CommandReturnObject &result 1750223383edSEnrico Granata ) 1751223383edSEnrico Granata { 1752223383edSEnrico Granata 1753223383edSEnrico Granata m_interpreter.GetHelp(result, 1754223383edSEnrico Granata CommandInterpreter::eCommandTypesUserDef); 1755223383edSEnrico Granata 1756223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishResult); 1757223383edSEnrico Granata 1758223383edSEnrico Granata return true; 1759223383edSEnrico Granata 1760223383edSEnrico Granata 1761223383edSEnrico Granata } 1762223383edSEnrico Granata }; 1763223383edSEnrico Granata 1764223383edSEnrico Granata //------------------------------------------------------------------------- 1765223383edSEnrico Granata // CommandObjectCommandsScriptClear 1766223383edSEnrico Granata //------------------------------------------------------------------------- 1767223383edSEnrico Granata 1768223383edSEnrico Granata class CommandObjectCommandsScriptClear : public CommandObject 1769223383edSEnrico Granata { 1770223383edSEnrico Granata private: 1771223383edSEnrico Granata 1772223383edSEnrico Granata public: 1773223383edSEnrico Granata CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) : 1774223383edSEnrico Granata CommandObject (interpreter, 1775223383edSEnrico Granata "command script clear", 1776223383edSEnrico Granata "Delete all scripted commands.", 1777223383edSEnrico Granata NULL) 1778223383edSEnrico Granata { 1779223383edSEnrico Granata } 1780223383edSEnrico Granata 1781223383edSEnrico Granata ~CommandObjectCommandsScriptClear () 1782223383edSEnrico Granata { 1783223383edSEnrico Granata } 1784223383edSEnrico Granata 1785223383edSEnrico Granata bool 1786223383edSEnrico Granata Execute 1787223383edSEnrico Granata ( 1788223383edSEnrico Granata Args& args, 1789223383edSEnrico Granata CommandReturnObject &result 1790223383edSEnrico Granata ) 1791223383edSEnrico Granata { 1792223383edSEnrico Granata 1793223383edSEnrico Granata m_interpreter.RemoveAllUser(); 1794223383edSEnrico Granata 1795223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishResult); 1796223383edSEnrico Granata 1797223383edSEnrico Granata return true; 1798223383edSEnrico Granata 1799223383edSEnrico Granata 1800223383edSEnrico Granata } 1801223383edSEnrico Granata }; 1802223383edSEnrico Granata 1803223383edSEnrico Granata //------------------------------------------------------------------------- 1804223383edSEnrico Granata // CommandObjectCommandsScriptDelete 1805223383edSEnrico Granata //------------------------------------------------------------------------- 1806223383edSEnrico Granata 1807223383edSEnrico Granata class CommandObjectCommandsScriptDelete : public CommandObject 1808223383edSEnrico Granata { 1809223383edSEnrico Granata private: 1810223383edSEnrico Granata 1811223383edSEnrico Granata public: 1812223383edSEnrico Granata CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) : 1813223383edSEnrico Granata CommandObject (interpreter, 1814223383edSEnrico Granata "command script delete", 1815223383edSEnrico Granata "Delete a scripted command.", 1816223383edSEnrico Granata NULL) 1817223383edSEnrico Granata { 1818223383edSEnrico Granata CommandArgumentEntry arg1; 1819223383edSEnrico Granata CommandArgumentData cmd_arg; 1820223383edSEnrico Granata 1821223383edSEnrico Granata // Define the first (and only) variant of this arg. 1822223383edSEnrico Granata cmd_arg.arg_type = eArgTypeCommandName; 1823223383edSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1824223383edSEnrico Granata 1825223383edSEnrico Granata // There is only one variant this argument could be; put it into the argument entry. 1826223383edSEnrico Granata arg1.push_back (cmd_arg); 1827223383edSEnrico Granata 1828223383edSEnrico Granata // Push the data for the first argument into the m_arguments vector. 1829223383edSEnrico Granata m_arguments.push_back (arg1); 1830223383edSEnrico Granata } 1831223383edSEnrico Granata 1832223383edSEnrico Granata ~CommandObjectCommandsScriptDelete () 1833223383edSEnrico Granata { 1834223383edSEnrico Granata } 1835223383edSEnrico Granata 1836223383edSEnrico Granata bool 1837223383edSEnrico Granata Execute 1838223383edSEnrico Granata ( 1839223383edSEnrico Granata Args& args, 1840223383edSEnrico Granata CommandReturnObject &result 1841223383edSEnrico Granata ) 1842223383edSEnrico Granata { 1843223383edSEnrico Granata 1844223383edSEnrico Granata size_t argc = args.GetArgumentCount(); 1845223383edSEnrico Granata 1846223383edSEnrico Granata if (argc != 1) 1847223383edSEnrico Granata { 1848223383edSEnrico Granata result.AppendError ("'command script delete' requires one argument"); 1849223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1850223383edSEnrico Granata return false; 1851223383edSEnrico Granata } 1852223383edSEnrico Granata 1853223383edSEnrico Granata const char* cmd_name = args.GetArgumentAtIndex(0); 1854223383edSEnrico Granata 1855223383edSEnrico Granata if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name)) 1856223383edSEnrico Granata { 1857223383edSEnrico Granata m_interpreter.RemoveUser(cmd_name); 1858223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishResult); 1859223383edSEnrico Granata } 1860223383edSEnrico Granata else 1861223383edSEnrico Granata { 1862223383edSEnrico Granata result.AppendErrorWithFormat ("command %s not found", cmd_name); 1863223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1864223383edSEnrico Granata } 1865223383edSEnrico Granata 1866223383edSEnrico Granata return result.Succeeded(); 1867223383edSEnrico Granata 1868223383edSEnrico Granata } 1869223383edSEnrico Granata }; 1870223383edSEnrico Granata 1871223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript 1872223383edSEnrico Granata 1873223383edSEnrico Granata //------------------------------------------------------------------------- 1874223383edSEnrico Granata // CommandObjectMultiwordCommandsScript 1875223383edSEnrico Granata //------------------------------------------------------------------------- 1876223383edSEnrico Granata 1877223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword 1878223383edSEnrico Granata { 1879223383edSEnrico Granata public: 1880223383edSEnrico Granata CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) : 1881223383edSEnrico Granata CommandObjectMultiword (interpreter, 1882223383edSEnrico Granata "command script", 1883223383edSEnrico Granata "A set of commands for managing or customizing script commands.", 1884223383edSEnrico Granata "command script <subcommand> [<subcommand-options>]") 1885223383edSEnrico Granata { 1886223383edSEnrico Granata LoadSubCommand ("add", CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter))); 1887223383edSEnrico Granata LoadSubCommand ("delete", CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter))); 1888223383edSEnrico Granata LoadSubCommand ("clear", CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter))); 1889223383edSEnrico Granata LoadSubCommand ("list", CommandObjectSP (new CommandObjectCommandsScriptList (interpreter))); 1890a9dbf432SEnrico Granata LoadSubCommand ("import", CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter))); 1891223383edSEnrico Granata } 1892223383edSEnrico Granata 1893223383edSEnrico Granata ~CommandObjectMultiwordCommandsScript () 1894223383edSEnrico Granata { 1895223383edSEnrico Granata } 1896223383edSEnrico Granata 1897223383edSEnrico Granata }; 1898223383edSEnrico Granata 1899223383edSEnrico Granata 1900ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands 1901ebc09c36SJim Ingham 1902ebc09c36SJim Ingham //------------------------------------------------------------------------- 1903ebc09c36SJim Ingham // CommandObjectMultiwordCommands 1904ebc09c36SJim Ingham //------------------------------------------------------------------------- 1905ebc09c36SJim Ingham 1906ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) : 1907a7015092SGreg Clayton CommandObjectMultiword (interpreter, 19080e5e5a79SGreg Clayton "command", 19093f4c09c1SCaroline Tice "A set of commands for managing or customizing the debugger commands.", 19100e5e5a79SGreg Clayton "command <subcommand> [<subcommand-options>]") 1911ebc09c36SJim Ingham { 1912a7015092SGreg Clayton LoadSubCommand ("source", CommandObjectSP (new CommandObjectCommandsSource (interpreter))); 1913a7015092SGreg Clayton LoadSubCommand ("alias", CommandObjectSP (new CommandObjectCommandsAlias (interpreter))); 1914a7015092SGreg Clayton LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter))); 1915de164aaaSGreg Clayton LoadSubCommand ("regex", CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter))); 1916a5a97ebeSJim Ingham LoadSubCommand ("history", CommandObjectSP (new CommandObjectCommandsHistory (interpreter))); 1917223383edSEnrico Granata LoadSubCommand ("script", CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter))); 1918ebc09c36SJim Ingham } 1919ebc09c36SJim Ingham 1920ebc09c36SJim Ingham CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands () 1921ebc09c36SJim Ingham { 1922ebc09c36SJim Ingham } 1923ebc09c36SJim Ingham 1924