mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2025-05-17 03:16:35 +00:00
c094d23f6f
Summary: This patch adds the possibility to specify an exit code when calling quit. We accept any int, even though it depends on the user what happens if the int is out of the range of what the operating system supports as exit codes. Fixes rdar://problem/38452312 Reviewers: davide, jingham, clayborg Reviewed By: jingham Subscribers: clayborg, jingham, lldb-commits Differential Revision: https://reviews.llvm.org/D48659 llvm-svn: 336824
122 lines
4.3 KiB
C++
122 lines
4.3 KiB
C++
//===-- CommandObjectQuit.cpp -----------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "CommandObjectQuit.h"
|
|
|
|
// C Includes
|
|
// C++ Includes
|
|
// Other libraries and framework includes
|
|
// Project includes
|
|
#include "lldb/Interpreter/CommandInterpreter.h"
|
|
#include "lldb/Interpreter/CommandReturnObject.h"
|
|
#include "lldb/Target/Process.h"
|
|
#include "lldb/Utility/StreamString.h"
|
|
|
|
using namespace lldb;
|
|
using namespace lldb_private;
|
|
|
|
//-------------------------------------------------------------------------
|
|
// CommandObjectQuit
|
|
//-------------------------------------------------------------------------
|
|
|
|
CommandObjectQuit::CommandObjectQuit(CommandInterpreter &interpreter)
|
|
: CommandObjectParsed(interpreter, "quit", "Quit the LLDB debugger.",
|
|
"quit [exit-code]") {}
|
|
|
|
CommandObjectQuit::~CommandObjectQuit() {}
|
|
|
|
// returns true if there is at least one alive process is_a_detach will be true
|
|
// if all alive processes will be detached when you quit and false if at least
|
|
// one process will be killed instead
|
|
bool CommandObjectQuit::ShouldAskForConfirmation(bool &is_a_detach) {
|
|
if (m_interpreter.GetPromptOnQuit() == false)
|
|
return false;
|
|
bool should_prompt = false;
|
|
is_a_detach = true;
|
|
for (uint32_t debugger_idx = 0; debugger_idx < Debugger::GetNumDebuggers();
|
|
debugger_idx++) {
|
|
DebuggerSP debugger_sp(Debugger::GetDebuggerAtIndex(debugger_idx));
|
|
if (!debugger_sp)
|
|
continue;
|
|
const TargetList &target_list(debugger_sp->GetTargetList());
|
|
for (uint32_t target_idx = 0;
|
|
target_idx < static_cast<uint32_t>(target_list.GetNumTargets());
|
|
target_idx++) {
|
|
TargetSP target_sp(target_list.GetTargetAtIndex(target_idx));
|
|
if (!target_sp)
|
|
continue;
|
|
ProcessSP process_sp(target_sp->GetProcessSP());
|
|
if (process_sp && process_sp->IsValid() && process_sp->IsAlive() &&
|
|
process_sp->WarnBeforeDetach()) {
|
|
should_prompt = true;
|
|
if (process_sp->GetShouldDetach() == false) {
|
|
// if we need to kill at least one process, just say so and return
|
|
is_a_detach = false;
|
|
return should_prompt;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return should_prompt;
|
|
}
|
|
|
|
bool CommandObjectQuit::DoExecute(Args &command, CommandReturnObject &result) {
|
|
bool is_a_detach = true;
|
|
if (ShouldAskForConfirmation(is_a_detach)) {
|
|
StreamString message;
|
|
message.Printf("Quitting LLDB will %s one or more processes. Do you really "
|
|
"want to proceed",
|
|
(is_a_detach ? "detach from" : "kill"));
|
|
if (!m_interpreter.Confirm(message.GetString(), true)) {
|
|
result.SetStatus(eReturnStatusFailed);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (command.GetArgumentCount() > 1) {
|
|
result.AppendError("Too many arguments for 'quit'. Only an optional exit "
|
|
"code is allowed");
|
|
result.SetStatus(eReturnStatusFailed);
|
|
return false;
|
|
}
|
|
|
|
if (command.GetArgumentCount() > 1) {
|
|
result.AppendError("Too many arguments for 'quit'. Only an optional exit "
|
|
"code is allowed");
|
|
result.SetStatus(eReturnStatusFailed);
|
|
return false;
|
|
}
|
|
|
|
// We parse the exit code argument if there is one.
|
|
if (command.GetArgumentCount() == 1) {
|
|
llvm::StringRef arg = command.GetArgumentAtIndex(0);
|
|
int exit_code;
|
|
if (arg.getAsInteger(/*autodetect radix*/ 0, exit_code)) {
|
|
lldb_private::StreamString s;
|
|
std::string arg_str = arg.str();
|
|
s.Printf("Couldn't parse '%s' as integer for exit code.", arg_str.data());
|
|
result.AppendError(s.GetString());
|
|
result.SetStatus(eReturnStatusFailed);
|
|
return false;
|
|
}
|
|
if (!m_interpreter.SetQuitExitCode(exit_code)) {
|
|
result.AppendError("The current driver doesn't allow custom exit codes"
|
|
" for the quit command.");
|
|
result.SetStatus(eReturnStatusFailed);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
const uint32_t event_type =
|
|
CommandInterpreter::eBroadcastBitQuitCommandReceived;
|
|
m_interpreter.BroadcastEvent(event_type);
|
|
result.SetStatus(eReturnStatusQuit);
|
|
return true;
|
|
}
|