1 //===-- CommandObjectQuit.cpp -----------------------------------*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "CommandObjectQuit.h" 11 12 #include "lldb/Interpreter/CommandInterpreter.h" 13 #include "lldb/Interpreter/CommandReturnObject.h" 14 #include "lldb/Target/Process.h" 15 #include "lldb/Utility/StreamString.h" 16 17 using namespace lldb; 18 using namespace lldb_private; 19 20 //------------------------------------------------------------------------- 21 // CommandObjectQuit 22 //------------------------------------------------------------------------- 23 24 CommandObjectQuit::CommandObjectQuit(CommandInterpreter &interpreter) 25 : CommandObjectParsed(interpreter, "quit", "Quit the LLDB debugger.", 26 "quit [exit-code]") {} 27 28 CommandObjectQuit::~CommandObjectQuit() {} 29 30 // returns true if there is at least one alive process is_a_detach will be true 31 // if all alive processes will be detached when you quit and false if at least 32 // one process will be killed instead 33 bool CommandObjectQuit::ShouldAskForConfirmation(bool &is_a_detach) { 34 if (!m_interpreter.GetPromptOnQuit()) 35 return false; 36 bool should_prompt = false; 37 is_a_detach = true; 38 for (uint32_t debugger_idx = 0; debugger_idx < Debugger::GetNumDebuggers(); 39 debugger_idx++) { 40 DebuggerSP debugger_sp(Debugger::GetDebuggerAtIndex(debugger_idx)); 41 if (!debugger_sp) 42 continue; 43 const TargetList &target_list(debugger_sp->GetTargetList()); 44 for (uint32_t target_idx = 0; 45 target_idx < static_cast<uint32_t>(target_list.GetNumTargets()); 46 target_idx++) { 47 TargetSP target_sp(target_list.GetTargetAtIndex(target_idx)); 48 if (!target_sp) 49 continue; 50 ProcessSP process_sp(target_sp->GetProcessSP()); 51 if (process_sp && process_sp->IsValid() && process_sp->IsAlive() && 52 process_sp->WarnBeforeDetach()) { 53 should_prompt = true; 54 if (!process_sp->GetShouldDetach()) { 55 // if we need to kill at least one process, just say so and return 56 is_a_detach = false; 57 return should_prompt; 58 } 59 } 60 } 61 } 62 return should_prompt; 63 } 64 65 bool CommandObjectQuit::DoExecute(Args &command, CommandReturnObject &result) { 66 bool is_a_detach = true; 67 if (ShouldAskForConfirmation(is_a_detach)) { 68 StreamString message; 69 message.Printf("Quitting LLDB will %s one or more processes. Do you really " 70 "want to proceed", 71 (is_a_detach ? "detach from" : "kill")); 72 if (!m_interpreter.Confirm(message.GetString(), true)) { 73 result.SetStatus(eReturnStatusFailed); 74 return false; 75 } 76 } 77 78 if (command.GetArgumentCount() > 1) { 79 result.AppendError("Too many arguments for 'quit'. Only an optional exit " 80 "code is allowed"); 81 result.SetStatus(eReturnStatusFailed); 82 return false; 83 } 84 85 // We parse the exit code argument if there is one. 86 if (command.GetArgumentCount() == 1) { 87 llvm::StringRef arg = command.GetArgumentAtIndex(0); 88 int exit_code; 89 if (arg.getAsInteger(/*autodetect radix*/ 0, exit_code)) { 90 lldb_private::StreamString s; 91 std::string arg_str = arg.str(); 92 s.Printf("Couldn't parse '%s' as integer for exit code.", arg_str.data()); 93 result.AppendError(s.GetString()); 94 result.SetStatus(eReturnStatusFailed); 95 return false; 96 } 97 if (!m_interpreter.SetQuitExitCode(exit_code)) { 98 result.AppendError("The current driver doesn't allow custom exit codes" 99 " for the quit command."); 100 result.SetStatus(eReturnStatusFailed); 101 return false; 102 } 103 } 104 105 const uint32_t event_type = 106 CommandInterpreter::eBroadcastBitQuitCommandReceived; 107 m_interpreter.BroadcastEvent(event_type); 108 result.SetStatus(eReturnStatusQuit); 109 return true; 110 } 111