contrib/llvm/tools/lldb/source/Commands/CommandObjectPlugin.cpp

   1 //===-- CommandObjectPlugin.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 "lldb/lldb-python.h"
  11
  12 #include "CommandObjectPlugin.h"
  13
  14 #include "lldb/API/SBDebugger.h"
  15 #include "lldb/API/SBCommandInterpreter.h"
  16 #include "lldb/API/SBCommandReturnObject.h"
  17
  18 #include "lldb/Host/Host.h"
  19
  20 #include "lldb/Interpreter/CommandInterpreter.h"
  21 #include "lldb/Interpreter/CommandReturnObject.h"
  22
  23 using namespace lldb;
  24 using namespace lldb_private;
  25
  26 class CommandObjectPluginLoad : public CommandObjectParsed
  27 {
  28 private:
  29 public:
  30     CommandObjectPluginLoad (CommandInterpreter &interpreter) :
  31     CommandObjectParsed (interpreter,
  32                          "plugin load",
  33                          "Import a dylib that implements an LLDB plugin.",
  34                          NULL)
  35     {
  36         CommandArgumentEntry arg1;
  37         CommandArgumentData cmd_arg;
  38
  39         // Define the first (and only) variant of this arg.
  40         cmd_arg.arg_type = eArgTypeFilename;
  41         cmd_arg.arg_repetition = eArgRepeatPlain;
  42
  43         // There is only one variant this argument could be; put it into the argument entry.
  44         arg1.push_back (cmd_arg);
  45
  46         // Push the data for the first argument into the m_arguments vector.
  47         m_arguments.push_back (arg1);
  48     }
  49
  50     ~CommandObjectPluginLoad ()
  51     {
  52     }
  53
  54     int
  55     HandleArgumentCompletion (Args &input,
  56                               int &cursor_index,
  57                               int &cursor_char_position,
  58                               OptionElementVector &opt_element_vector,
  59                               int match_start_point,
  60                               int max_return_elements,
  61                               bool &word_complete,
  62                               StringList &matches)
  63     {
  64         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
  65         completion_str.erase (cursor_char_position);
  66
  67         CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
  68                                                              CommandCompletions::eDiskFileCompletion,
  69                                                              completion_str.c_str(),
  70                                                              match_start_point,
  71                                                              max_return_elements,
  72                                                              NULL,
  73                                                              word_complete,
  74                                                              matches);
  75         return matches.GetSize();
  76     }
  77
  78 protected:
  79     bool
  80     DoExecute (Args& command, CommandReturnObject &result)
  81     {
  82         typedef void (*LLDBCommandPluginInit) (lldb::SBDebugger debugger);
  83
  84         size_t argc = command.GetArgumentCount();
  85
  86         if (argc != 1)
  87         {
  88             result.AppendError ("'plugin load' requires one argument");
  89             result.SetStatus (eReturnStatusFailed);
  90             return false;
  91         }
  92
  93         const char* path = command.GetArgumentAtIndex(0);
  94
  95         Error error;
  96
  97         FileSpec dylib_fspec(path,true);
  98
  99         if (m_interpreter.GetDebugger().LoadPlugin(dylib_fspec, error))
 100             result.SetStatus(eReturnStatusSuccessFinishResult);
 101         else
 102         {
 103             result.AppendError(error.AsCString());
 104             result.SetStatus(eReturnStatusFailed);
 105         }
 106
 107         return result.Succeeded();
 108     }
 109 };
 110
 111 CommandObjectPlugin::CommandObjectPlugin (CommandInterpreter &interpreter) :
 112 CommandObjectMultiword (interpreter,
 113                         "plugin",
 114                         "A set of commands for managing or customizing plugin commands.",
 115                         "plugin <subcommand> [<subcommand-options>]")
 116 {
 117     LoadSubCommand ("load",  CommandObjectSP (new CommandObjectPluginLoad (interpreter)));
 118 }
 119
 120 CommandObjectPlugin::~CommandObjectPlugin ()
 121 {
 122 }