source/Commands/CommandObjectQuit.cpp

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