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 // C Includes
13 // C++ Includes
14 // Other libraries and framework includes
15 // Project includes
16 #include "lldb/Interpreter/CommandInterpreter.h"
17 #include "lldb/Interpreter/CommandReturnObject.h"
18 #include "lldb/Target/Process.h"
19 
20 using namespace lldb;
21 using namespace lldb_private;
22 
23 //-------------------------------------------------------------------------
24 // CommandObjectQuit
25 //-------------------------------------------------------------------------
26 
27 CommandObjectQuit::CommandObjectQuit(CommandInterpreter &interpreter)
28     : CommandObjectParsed(interpreter, "quit", "Quit the LLDB debugger.",
29                           "quit") {}
30 
31 CommandObjectQuit::~CommandObjectQuit() {}
32 
33 // returns true if there is at least one alive process is_a_detach will be true
34 // if all alive processes will be detached when you quit and false if at least
35 // one process will be killed instead
36 bool CommandObjectQuit::ShouldAskForConfirmation(bool &is_a_detach) {
37   if (m_interpreter.GetPromptOnQuit() == false)
38     return false;
39   bool should_prompt = false;
40   is_a_detach = true;
41   for (uint32_t debugger_idx = 0; debugger_idx < Debugger::GetNumDebuggers();
42        debugger_idx++) {
43     DebuggerSP debugger_sp(Debugger::GetDebuggerAtIndex(debugger_idx));
44     if (!debugger_sp)
45       continue;
46     const TargetList &target_list(debugger_sp->GetTargetList());
47     for (uint32_t target_idx = 0;
48          target_idx < static_cast<uint32_t>(target_list.GetNumTargets());
49          target_idx++) {
50       TargetSP target_sp(target_list.GetTargetAtIndex(target_idx));
51       if (!target_sp)
52         continue;
53       ProcessSP process_sp(target_sp->GetProcessSP());
54       if (process_sp && process_sp->IsValid() && process_sp->IsAlive() &&
55           process_sp->WarnBeforeDetach()) {
56         should_prompt = true;
57         if (process_sp->GetShouldDetach() == false) {
58           // if we need to kill at least one process, just say so and return
59           is_a_detach = false;
60           return should_prompt;
61         }
62       }
63     }
64   }
65   return should_prompt;
66 }
67 
68 bool CommandObjectQuit::DoExecute(Args &command, CommandReturnObject &result) {
69   bool is_a_detach = true;
70   if (ShouldAskForConfirmation(is_a_detach)) {
71     StreamString message;
72     message.Printf("Quitting LLDB will %s one or more processes. Do you really "
73                    "want to proceed",
74                    (is_a_detach ? "detach from" : "kill"));
75     if (!m_interpreter.Confirm(message.GetString(), true)) {
76       result.SetStatus(eReturnStatusFailed);
77       return false;
78     }
79   }
80   const uint32_t event_type =
81       CommandInterpreter::eBroadcastBitQuitCommandReceived;
82   m_interpreter.BroadcastEvent(event_type);
83   result.SetStatus(eReturnStatusQuit);
84   return true;
85 }
86