mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-04 00:56:05 +00:00
85e8b81492
a Stream, and then added GetOutputData & GetErrorData to get the accumulated data. - Added a StreamTee that will tee output to two provided lldb::StreamSP's. - Made the CommandObjectReturn use this so you can Tee the results immediately to the debuggers output file, as well as saving up the results to return when the command is done executing. - HandleCommands now uses this so that if you have a set of commands that continue the target you will see the commands come out as they are processed. - The Driver now uses this to output the command results as you go, which makes the interface more reactive seeming. llvm-svn: 126015
313 lines
10 KiB
C++
313 lines
10 KiB
C++
//===-- CommandObjectMultiword.cpp ------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lldb/Interpreter/CommandObjectMultiword.h"
|
|
// C Includes
|
|
// C++ Includes
|
|
// Other libraries and framework includes
|
|
// Project includes
|
|
#include "lldb/Core/Debugger.h"
|
|
#include "lldb/Interpreter/CommandInterpreter.h"
|
|
#include "lldb/Interpreter/Options.h"
|
|
#include "lldb/Interpreter/CommandReturnObject.h"
|
|
|
|
using namespace lldb;
|
|
using namespace lldb_private;
|
|
|
|
//-------------------------------------------------------------------------
|
|
// CommandObjectMultiword
|
|
//-------------------------------------------------------------------------
|
|
|
|
CommandObjectMultiword::CommandObjectMultiword
|
|
(
|
|
CommandInterpreter &interpreter,
|
|
const char *name,
|
|
const char *help,
|
|
const char *syntax,
|
|
uint32_t flags
|
|
) :
|
|
CommandObject (interpreter, name, help, syntax, flags)
|
|
{
|
|
}
|
|
|
|
CommandObjectMultiword::~CommandObjectMultiword ()
|
|
{
|
|
}
|
|
|
|
CommandObjectSP
|
|
CommandObjectMultiword::GetSubcommandSP (const char *sub_cmd, StringList *matches)
|
|
{
|
|
CommandObjectSP return_cmd_sp;
|
|
CommandObject::CommandMap::iterator pos;
|
|
|
|
if (!m_subcommand_dict.empty())
|
|
{
|
|
pos = m_subcommand_dict.find (sub_cmd);
|
|
if (pos != m_subcommand_dict.end()) {
|
|
// An exact match; append the sub_cmd to the 'matches' string list.
|
|
if (matches)
|
|
matches->AppendString(sub_cmd);
|
|
return_cmd_sp = pos->second;
|
|
}
|
|
else
|
|
{
|
|
|
|
StringList local_matches;
|
|
if (matches == NULL)
|
|
matches = &local_matches;
|
|
int num_matches = CommandObject::AddNamesMatchingPartialString (m_subcommand_dict, sub_cmd, *matches);
|
|
|
|
if (num_matches == 1)
|
|
{
|
|
// Cleaner, but slightly less efficient would be to call back into this function, since I now
|
|
// know I have an exact match...
|
|
|
|
sub_cmd = matches->GetStringAtIndex(0);
|
|
pos = m_subcommand_dict.find(sub_cmd);
|
|
if (pos != m_subcommand_dict.end())
|
|
return_cmd_sp = pos->second;
|
|
}
|
|
}
|
|
}
|
|
return return_cmd_sp;
|
|
}
|
|
|
|
CommandObject *
|
|
CommandObjectMultiword::GetSubcommandObject (const char *sub_cmd, StringList *matches)
|
|
{
|
|
return GetSubcommandSP(sub_cmd, matches).get();
|
|
}
|
|
|
|
bool
|
|
CommandObjectMultiword::LoadSubCommand
|
|
(
|
|
const char *name,
|
|
const CommandObjectSP& cmd_obj
|
|
)
|
|
{
|
|
CommandMap::iterator pos;
|
|
bool success = true;
|
|
|
|
pos = m_subcommand_dict.find(name);
|
|
if (pos == m_subcommand_dict.end())
|
|
{
|
|
m_subcommand_dict[name] = cmd_obj;
|
|
m_interpreter.CrossRegisterCommand (name, GetCommandName());
|
|
}
|
|
else
|
|
success = false;
|
|
|
|
return success;
|
|
}
|
|
|
|
bool
|
|
CommandObjectMultiword::Execute
|
|
(
|
|
Args& args,
|
|
CommandReturnObject &result
|
|
)
|
|
{
|
|
const size_t argc = args.GetArgumentCount();
|
|
if (argc == 0)
|
|
{
|
|
GenerateHelpText (result);
|
|
}
|
|
else
|
|
{
|
|
const char *sub_command = args.GetArgumentAtIndex (0);
|
|
|
|
if (sub_command)
|
|
{
|
|
if (::strcasecmp (sub_command, "help") == 0)
|
|
{
|
|
GenerateHelpText (result);
|
|
}
|
|
else if (!m_subcommand_dict.empty())
|
|
{
|
|
StringList matches;
|
|
CommandObject *sub_cmd_obj = GetSubcommandObject(sub_command, &matches);
|
|
if (sub_cmd_obj != NULL)
|
|
{
|
|
// Now call CommandObject::Execute to process and options in 'rest_of_line'. From there
|
|
// the command-specific version of Execute will be called, with the processed arguments.
|
|
|
|
args.Shift();
|
|
|
|
sub_cmd_obj->ExecuteWithOptions (args, result);
|
|
}
|
|
else
|
|
{
|
|
std::string error_msg;
|
|
int num_subcmd_matches = matches.GetSize();
|
|
if (num_subcmd_matches > 0)
|
|
error_msg.assign ("ambiguous command ");
|
|
else
|
|
error_msg.assign ("invalid command ");
|
|
|
|
error_msg.append ("'");
|
|
error_msg.append (GetCommandName());
|
|
error_msg.append (" ");
|
|
error_msg.append (sub_command);
|
|
error_msg.append ("'");
|
|
|
|
if (num_subcmd_matches > 0)
|
|
{
|
|
error_msg.append (" Possible completions:");
|
|
for (int i = 0; i < num_subcmd_matches; i++)
|
|
{
|
|
error_msg.append ("\n\t");
|
|
error_msg.append (matches.GetStringAtIndex (i));
|
|
}
|
|
}
|
|
error_msg.append ("\n");
|
|
result.AppendRawError (error_msg.c_str(), error_msg.size());
|
|
result.SetStatus (eReturnStatusFailed);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
result.AppendErrorWithFormat ("'%s' does not have any subcommands.\n", GetCommandName());
|
|
result.SetStatus (eReturnStatusFailed);
|
|
}
|
|
}
|
|
}
|
|
|
|
return result.Succeeded();
|
|
}
|
|
|
|
void
|
|
CommandObjectMultiword::GenerateHelpText (CommandReturnObject &result)
|
|
{
|
|
// First time through here, generate the help text for the object and
|
|
// push it to the return result object as well
|
|
|
|
Stream &output_stream = result.GetOutputStream();
|
|
output_stream.PutCString ("The following subcommands are supported:\n\n");
|
|
|
|
CommandMap::iterator pos;
|
|
uint32_t max_len = m_interpreter.FindLongestCommandWord (m_subcommand_dict);
|
|
|
|
if (max_len)
|
|
max_len += 4; // Indent the output by 4 spaces.
|
|
|
|
for (pos = m_subcommand_dict.begin(); pos != m_subcommand_dict.end(); ++pos)
|
|
{
|
|
std::string indented_command (" ");
|
|
indented_command.append (pos->first);
|
|
if (pos->second->WantsRawCommandString ())
|
|
{
|
|
std::string help_text (pos->second->GetHelp());
|
|
help_text.append (" This command takes 'raw' input (no need to quote stuff).");
|
|
m_interpreter.OutputFormattedHelpText (result.GetOutputStream(),
|
|
indented_command.c_str(),
|
|
"--",
|
|
help_text.c_str(),
|
|
max_len);
|
|
}
|
|
else
|
|
m_interpreter.OutputFormattedHelpText (result.GetOutputStream(),
|
|
indented_command.c_str(),
|
|
"--",
|
|
pos->second->GetHelp(),
|
|
max_len);
|
|
}
|
|
|
|
output_stream.PutCString ("\nFor more help on any particular subcommand, type 'help <command> <subcommand>'.\n");
|
|
|
|
result.SetStatus (eReturnStatusSuccessFinishNoResult);
|
|
}
|
|
|
|
int
|
|
CommandObjectMultiword::HandleCompletion
|
|
(
|
|
Args &input,
|
|
int &cursor_index,
|
|
int &cursor_char_position,
|
|
int match_start_point,
|
|
int max_return_elements,
|
|
bool &word_complete,
|
|
StringList &matches
|
|
)
|
|
{
|
|
// Any of the command matches will provide a complete word, otherwise the individual
|
|
// completers will override this.
|
|
word_complete = true;
|
|
|
|
if (cursor_index == 0)
|
|
{
|
|
CommandObject::AddNamesMatchingPartialString (m_subcommand_dict,
|
|
input.GetArgumentAtIndex(0),
|
|
matches);
|
|
|
|
if (matches.GetSize() == 1
|
|
&& matches.GetStringAtIndex(0) != NULL
|
|
&& strcmp (input.GetArgumentAtIndex(0), matches.GetStringAtIndex(0)) == 0)
|
|
{
|
|
StringList temp_matches;
|
|
CommandObject *cmd_obj = GetSubcommandObject (input.GetArgumentAtIndex(0),
|
|
&temp_matches);
|
|
if (cmd_obj != NULL)
|
|
{
|
|
matches.DeleteStringAtIndex (0);
|
|
input.Shift();
|
|
cursor_char_position = 0;
|
|
input.AppendArgument ("");
|
|
return cmd_obj->HandleCompletion (input,
|
|
cursor_index,
|
|
cursor_char_position,
|
|
match_start_point,
|
|
max_return_elements,
|
|
word_complete,
|
|
matches);
|
|
}
|
|
else
|
|
return matches.GetSize();
|
|
}
|
|
else
|
|
return matches.GetSize();
|
|
}
|
|
else
|
|
{
|
|
CommandObject *sub_command_object = GetSubcommandObject (input.GetArgumentAtIndex(0),
|
|
&matches);
|
|
if (sub_command_object == NULL)
|
|
{
|
|
return matches.GetSize();
|
|
}
|
|
else
|
|
{
|
|
// Remove the one match that we got from calling GetSubcommandObject.
|
|
matches.DeleteStringAtIndex(0);
|
|
input.Shift();
|
|
cursor_index--;
|
|
return sub_command_object->HandleCompletion (input,
|
|
cursor_index,
|
|
cursor_char_position,
|
|
match_start_point,
|
|
max_return_elements,
|
|
word_complete,
|
|
matches);
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
const char *
|
|
CommandObjectMultiword::GetRepeatCommand (Args ¤t_command_args, uint32_t index)
|
|
{
|
|
index++;
|
|
if (current_command_args.GetArgumentCount() <= index)
|
|
return NULL;
|
|
CommandObject *sub_command_object = GetSubcommandObject (current_command_args.GetArgumentAtIndex(index));
|
|
if (sub_command_object == NULL)
|
|
return NULL;
|
|
return sub_command_object->GetRepeatCommand(current_command_args, index);
|
|
}
|
|
|