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