1 /* gdb commands implemented in Python
3 Copyright (C) 2008-2019 Free Software Foundation, Inc.
5 This file is part of GDB.
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>. */
22 #include "arch-utils.h"
24 #include "python-internal.h"
27 #include "cli/cli-decode.h"
28 #include "completer.h"
31 /* Struct representing built-in completion types. */
32 struct cmdpy_completer
34 /* Python symbol name. */
36 /* Completion function. */
37 completer_ftype
*completer
;
40 static const struct cmdpy_completer completers
[] =
42 { "COMPLETE_NONE", noop_completer
},
43 { "COMPLETE_FILENAME", filename_completer
},
44 { "COMPLETE_LOCATION", location_completer
},
45 { "COMPLETE_COMMAND", command_completer
},
46 { "COMPLETE_SYMBOL", symbol_completer
},
47 { "COMPLETE_EXPRESSION", expression_completer
},
50 #define N_COMPLETERS (sizeof (completers) / sizeof (completers[0]))
52 /* A gdb command. For the time being only ordinary commands (not
53 set/show commands) are allowed. */
58 /* The corresponding gdb command object, or NULL if the command is
59 no longer installed. */
60 struct cmd_list_element
*command
;
62 /* A prefix command requires storage for a list of its sub-commands.
63 A pointer to this is passed to add_prefix_command, and to add_cmd
64 for sub-commands of that prefix. If this Command is not a prefix
65 command, then this field is unused. */
66 struct cmd_list_element
*sub_list
;
69 typedef struct cmdpy_object cmdpy_object
;
71 extern PyTypeObject cmdpy_object_type
72 CPYCHECKER_TYPE_OBJECT_FOR_TYPEDEF ("cmdpy_object");
74 /* Constants used by this module. */
75 static PyObject
*invoke_cst
;
76 static PyObject
*complete_cst
;
80 /* Python function which wraps dont_repeat. */
82 cmdpy_dont_repeat (PyObject
*self
, PyObject
*args
)
90 /* Called if the gdb cmd_list_element is destroyed. */
93 cmdpy_destroyer (struct cmd_list_element
*self
, void *context
)
95 gdbpy_enter
enter_py (get_current_arch (), current_language
);
97 /* Release our hold on the command object. */
98 gdbpy_ref
<cmdpy_object
> cmd ((cmdpy_object
*) context
);
101 /* We allocated the name, doc string, and perhaps the prefix
103 xfree ((char *) self
->name
);
104 xfree ((char *) self
->doc
);
105 xfree ((char *) self
->prefixname
);
108 /* Called by gdb to invoke the command. */
111 cmdpy_function (struct cmd_list_element
*command
,
112 const char *args
, int from_tty
)
114 cmdpy_object
*obj
= (cmdpy_object
*) get_cmd_context (command
);
116 gdbpy_enter
enter_py (get_current_arch (), current_language
);
119 error (_("Invalid invocation of Python command object."));
120 if (! PyObject_HasAttr ((PyObject
*) obj
, invoke_cst
))
122 if (obj
->command
->prefixname
)
124 /* A prefix command does not need an invoke method. */
127 error (_("Python command object missing 'invoke' method."));
132 gdbpy_ref
<> argobj (PyUnicode_Decode (args
, strlen (args
), host_charset (),
136 gdbpy_print_stack ();
137 error (_("Could not convert arguments to Python string."));
141 = gdbpy_ref
<>::new_reference (from_tty
? Py_True
: Py_False
);
142 gdbpy_ref
<> result (PyObject_CallMethodObjArgs ((PyObject
*) obj
, invoke_cst
,
143 argobj
.get (), ttyobj
.get (),
147 gdbpy_handle_exception ();
150 /* Helper function for the Python command completers (both "pure"
151 completer and brkchar handler). This function takes COMMAND, TEXT
152 and WORD and tries to call the Python method for completion with
155 This function is usually called twice: once when we are figuring out
156 the break characters to be used, and another to perform the real
157 completion itself. The reason for this two step dance is that we
158 need to know the set of "brkchars" to use early on, before we
159 actually try to perform the completion. But if a Python command
160 supplies a "complete" method then we have to call that method
161 first: it may return as its result the kind of completion to
162 perform and that will in turn specify which brkchars to use. IOW,
163 we need the result of the "complete" method before we actually
164 perform the completion. The only situation when this function is
165 not called twice is when the user uses the "complete" command: in
166 this scenario, there is no call to determine the "brkchars".
168 Ideally, it would be nice to cache the result of the first call (to
169 determine the "brkchars") and return this value directly in the
170 second call (to perform the actual completion). However, due to
171 the peculiarity of the "complete" command mentioned above, it is
172 possible to put GDB in a bad state if you perform a TAB-completion
173 and then a "complete"-completion sequentially. Therefore, we just
174 recalculate everything twice for TAB-completions.
176 This function returns a reference to the PyObject representing the
177 Python method call. */
180 cmdpy_completer_helper (struct cmd_list_element
*command
,
181 const char *text
, const char *word
)
183 cmdpy_object
*obj
= (cmdpy_object
*) get_cmd_context (command
);
186 error (_("Invalid invocation of Python command object."));
187 if (!PyObject_HasAttr ((PyObject
*) obj
, complete_cst
))
189 /* If there is no complete method, don't error. */
193 gdbpy_ref
<> textobj (PyUnicode_Decode (text
, strlen (text
), host_charset (),
196 error (_("Could not convert argument to Python string."));
201 /* "brkchars" phase. */
202 wordobj
= gdbpy_ref
<>::new_reference (Py_None
);
206 wordobj
.reset (PyUnicode_Decode (word
, strlen (word
), host_charset (),
209 error (_("Could not convert argument to Python string."));
212 gdbpy_ref
<> resultobj (PyObject_CallMethodObjArgs ((PyObject
*) obj
,
215 wordobj
.get (), NULL
));
216 if (resultobj
== NULL
)
218 /* Just swallow errors here. */
225 /* Python function called to determine the break characters of a
226 certain completer. We are only interested in knowing if the
227 completer registered by the user will return one of the integer
228 codes (see COMPLETER_* symbols). */
231 cmdpy_completer_handle_brkchars (struct cmd_list_element
*command
,
232 completion_tracker
&tracker
,
233 const char *text
, const char *word
)
235 gdbpy_enter
enter_py (get_current_arch (), current_language
);
237 /* Calling our helper to obtain a reference to the PyObject of the Python
239 gdbpy_ref
<> resultobj
= cmdpy_completer_helper (command
, text
, word
);
241 /* Check if there was an error. */
242 if (resultobj
== NULL
)
245 if (PyInt_Check (resultobj
.get ()))
247 /* User code may also return one of the completion constants,
248 thus requesting that sort of completion. We are only
249 interested in this kind of return. */
252 if (!gdb_py_int_as_long (resultobj
.get (), &value
))
257 else if (value
>= 0 && value
< (long) N_COMPLETERS
)
259 completer_handle_brkchars_ftype
*brkchars_fn
;
261 /* This is the core of this function. Depending on which
262 completer type the Python function returns, we have to
263 adjust the break characters accordingly. */
264 brkchars_fn
= (completer_handle_brkchars_func_for_completer
265 (completers
[value
].completer
));
266 brkchars_fn (command
, tracker
, text
, word
);
271 /* Called by gdb for command completion. */
274 cmdpy_completer (struct cmd_list_element
*command
,
275 completion_tracker
&tracker
,
276 const char *text
, const char *word
)
278 gdbpy_enter
enter_py (get_current_arch (), current_language
);
280 /* Calling our helper to obtain a reference to the PyObject of the Python
282 gdbpy_ref
<> resultobj
= cmdpy_completer_helper (command
, text
, word
);
284 /* If the result object of calling the Python function is NULL, it
285 means that there was an error. In this case, just give up. */
286 if (resultobj
== NULL
)
289 if (PyInt_Check (resultobj
.get ()))
291 /* User code may also return one of the completion constants,
292 thus requesting that sort of completion. */
295 if (! gdb_py_int_as_long (resultobj
.get (), &value
))
300 else if (value
>= 0 && value
< (long) N_COMPLETERS
)
301 completers
[value
].completer (command
, tracker
, text
, word
);
305 gdbpy_ref
<> iter (PyObject_GetIter (resultobj
.get ()));
310 bool got_matches
= false;
313 gdbpy_ref
<> elt (PyIter_Next (iter
.get ()));
317 if (! gdbpy_is_string (elt
.get ()))
319 /* Skip problem elements. */
322 gdb::unique_xmalloc_ptr
<char>
323 item (python_string_to_host_string (elt
.get ()));
326 /* Skip problem elements. */
330 tracker
.add_completion (std::move (item
));
334 /* If we got some results, ignore problems. Otherwise, report
336 if (got_matches
&& PyErr_Occurred ())
341 /* Helper for cmdpy_init which locates the command list to use and
342 pulls out the command name.
344 NAME is the command name list. The final word in the list is the
345 name of the new command. All earlier words must be existing prefix
348 *BASE_LIST is set to the final prefix command's list of
351 START_LIST is the list in which the search starts.
353 This function returns the xmalloc()d name of the new command. On
354 error sets the Python error and returns NULL. */
357 gdbpy_parse_command_name (const char *name
,
358 struct cmd_list_element
***base_list
,
359 struct cmd_list_element
**start_list
)
361 struct cmd_list_element
*elt
;
362 int len
= strlen (name
);
364 const char *prefix_text2
;
367 /* Skip trailing whitespace. */
368 for (i
= len
- 1; i
>= 0 && (name
[i
] == ' ' || name
[i
] == '\t'); --i
)
372 PyErr_SetString (PyExc_RuntimeError
, _("No command name found."));
377 /* Find first character of the final word. */
378 for (; i
> 0 && (isalnum (name
[i
- 1])
379 || name
[i
- 1] == '-'
380 || name
[i
- 1] == '_');
383 result
= (char *) xmalloc (lastchar
- i
+ 2);
384 memcpy (result
, &name
[i
], lastchar
- i
+ 1);
385 result
[lastchar
- i
+ 1] = '\0';
387 /* Skip whitespace again. */
388 for (--i
; i
>= 0 && (name
[i
] == ' ' || name
[i
] == '\t'); --i
)
392 *base_list
= start_list
;
396 std::string
prefix_text (name
, i
+ 1);
398 prefix_text2
= prefix_text
.c_str ();
399 elt
= lookup_cmd_1 (&prefix_text2
, *start_list
, NULL
, 1);
400 if (elt
== NULL
|| elt
== CMD_LIST_AMBIGUOUS
)
402 PyErr_Format (PyExc_RuntimeError
, _("Could not find command prefix %s."),
403 prefix_text
.c_str ());
410 *base_list
= elt
->prefixlist
;
414 PyErr_Format (PyExc_RuntimeError
, _("'%s' is not a prefix command."),
415 prefix_text
.c_str ());
420 /* Object initializer; sets up gdb-side structures for command.
422 Use: __init__(NAME, COMMAND_CLASS [, COMPLETER_CLASS][, PREFIX]]).
424 NAME is the name of the command. It may consist of multiple words,
425 in which case the final word is the name of the new command, and
426 earlier words must be prefix commands.
428 COMMAND_CLASS is the kind of command. It should be one of the COMMAND_*
429 constants defined in the gdb module.
431 COMPLETER_CLASS is the kind of completer. If not given, the
432 "complete" method will be used. Otherwise, it should be one of the
433 COMPLETE_* constants defined in the gdb module.
435 If PREFIX is True, then this command is a prefix command.
437 The documentation for the command is taken from the doc string for
441 cmdpy_init (PyObject
*self
, PyObject
*args
, PyObject
*kw
)
443 cmdpy_object
*obj
= (cmdpy_object
*) self
;
446 int completetype
= -1;
447 char *docstring
= NULL
;
448 struct cmd_list_element
**cmd_list
;
449 char *cmd_name
, *pfx_name
;
450 static const char *keywords
[] = { "name", "command_class", "completer_class",
452 PyObject
*is_prefix
= NULL
;
457 /* Note: this is apparently not documented in Python. We return
458 0 for success, -1 for failure. */
459 PyErr_Format (PyExc_RuntimeError
,
460 _("Command object already initialized."));
464 if (!gdb_PyArg_ParseTupleAndKeywords (args
, kw
, "si|iO",
465 keywords
, &name
, &cmdtype
,
466 &completetype
, &is_prefix
))
469 if (cmdtype
!= no_class
&& cmdtype
!= class_run
470 && cmdtype
!= class_vars
&& cmdtype
!= class_stack
471 && cmdtype
!= class_files
&& cmdtype
!= class_support
472 && cmdtype
!= class_info
&& cmdtype
!= class_breakpoint
473 && cmdtype
!= class_trace
&& cmdtype
!= class_obscure
474 && cmdtype
!= class_maintenance
&& cmdtype
!= class_user
)
476 PyErr_Format (PyExc_RuntimeError
, _("Invalid command class argument."));
480 if (completetype
< -1 || completetype
>= (int) N_COMPLETERS
)
482 PyErr_Format (PyExc_RuntimeError
,
483 _("Invalid completion type argument."));
487 cmd_name
= gdbpy_parse_command_name (name
, &cmd_list
, &cmdlist
);
492 if (is_prefix
!= NULL
)
494 cmp
= PyObject_IsTrue (is_prefix
);
499 /* Make a normalized form of the command name. */
500 pfx_name
= (char *) xmalloc (strlen (name
) + 2);
506 /* Skip whitespace. */
507 while (name
[i
] == ' ' || name
[i
] == '\t')
509 /* Copy non-whitespace characters. */
510 while (name
[i
] && name
[i
] != ' ' && name
[i
] != '\t')
511 pfx_name
[out
++] = name
[i
++];
512 /* Add a single space after each word -- including the final
514 pfx_name
[out
++] = ' ';
516 pfx_name
[out
] = '\0';
524 if (PyObject_HasAttr (self
, gdbpy_doc_cst
))
526 gdbpy_ref
<> ds_obj (PyObject_GetAttr (self
, gdbpy_doc_cst
));
528 if (ds_obj
!= NULL
&& gdbpy_is_string (ds_obj
.get ()))
530 docstring
= python_string_to_host_string (ds_obj
.get ()).release ();
531 if (docstring
== NULL
)
540 docstring
= xstrdup (_("This command is not documented."));
542 gdbpy_ref
<> self_ref
= gdbpy_ref
<>::new_reference (self
);
546 struct cmd_list_element
*cmd
;
552 /* If we have our own "invoke" method, then allow unknown
554 allow_unknown
= PyObject_HasAttr (self
, invoke_cst
);
555 cmd
= add_prefix_cmd (cmd_name
, (enum command_class
) cmdtype
,
556 NULL
, docstring
, &obj
->sub_list
,
557 pfx_name
, allow_unknown
, cmd_list
);
560 cmd
= add_cmd (cmd_name
, (enum command_class
) cmdtype
,
561 docstring
, cmd_list
);
563 /* There appears to be no API to set this. */
564 cmd
->func
= cmdpy_function
;
565 cmd
->destroyer
= cmdpy_destroyer
;
568 set_cmd_context (cmd
, self_ref
.release ());
569 set_cmd_completer (cmd
, ((completetype
== -1) ? cmdpy_completer
570 : completers
[completetype
].completer
));
571 if (completetype
== -1)
572 set_cmd_completer_handle_brkchars (cmd
,
573 cmdpy_completer_handle_brkchars
);
575 catch (const gdb_exception
&except
)
580 gdbpy_convert_exception (except
);
589 /* Initialize the 'commands' code. */
592 gdbpy_initialize_commands (void)
596 cmdpy_object_type
.tp_new
= PyType_GenericNew
;
597 if (PyType_Ready (&cmdpy_object_type
) < 0)
600 /* Note: alias and user are special; pseudo appears to be unused,
601 and there is no reason to expose tui, I think. */
602 if (PyModule_AddIntConstant (gdb_module
, "COMMAND_NONE", no_class
) < 0
603 || PyModule_AddIntConstant (gdb_module
, "COMMAND_RUNNING", class_run
) < 0
604 || PyModule_AddIntConstant (gdb_module
, "COMMAND_DATA", class_vars
) < 0
605 || PyModule_AddIntConstant (gdb_module
, "COMMAND_STACK", class_stack
) < 0
606 || PyModule_AddIntConstant (gdb_module
, "COMMAND_FILES", class_files
) < 0
607 || PyModule_AddIntConstant (gdb_module
, "COMMAND_SUPPORT",
609 || PyModule_AddIntConstant (gdb_module
, "COMMAND_STATUS", class_info
) < 0
610 || PyModule_AddIntConstant (gdb_module
, "COMMAND_BREAKPOINTS",
611 class_breakpoint
) < 0
612 || PyModule_AddIntConstant (gdb_module
, "COMMAND_TRACEPOINTS",
614 || PyModule_AddIntConstant (gdb_module
, "COMMAND_OBSCURE",
616 || PyModule_AddIntConstant (gdb_module
, "COMMAND_MAINTENANCE",
617 class_maintenance
) < 0
618 || PyModule_AddIntConstant (gdb_module
, "COMMAND_USER", class_user
) < 0)
621 for (i
= 0; i
< N_COMPLETERS
; ++i
)
623 if (PyModule_AddIntConstant (gdb_module
, completers
[i
].name
, i
) < 0)
627 if (gdb_pymodule_addobject (gdb_module
, "Command",
628 (PyObject
*) &cmdpy_object_type
) < 0)
631 invoke_cst
= PyString_FromString ("invoke");
632 if (invoke_cst
== NULL
)
634 complete_cst
= PyString_FromString ("complete");
635 if (complete_cst
== NULL
)
643 static PyMethodDef cmdpy_object_methods
[] =
645 { "dont_repeat", cmdpy_dont_repeat
, METH_NOARGS
,
646 "Prevent command repetition when user enters empty line." },
651 PyTypeObject cmdpy_object_type
=
653 PyVarObject_HEAD_INIT (NULL
, 0)
654 "gdb.Command", /*tp_name*/
655 sizeof (cmdpy_object
), /*tp_basicsize*/
664 0, /*tp_as_sequence*/
672 Py_TPFLAGS_DEFAULT
| Py_TPFLAGS_BASETYPE
, /*tp_flags*/
673 "GDB command object", /* tp_doc */
676 0, /* tp_richcompare */
677 0, /* tp_weaklistoffset */
680 cmdpy_object_methods
, /* tp_methods */
685 0, /* tp_descr_get */
686 0, /* tp_descr_set */
687 0, /* tp_dictoffset */
688 cmdpy_init
, /* tp_init */
694 /* Utility to build a buildargv-like result from ARGS.
695 This intentionally parses arguments the way libiberty/argv.c:buildargv
696 does. It splits up arguments in a reasonable way, and we want a standard
697 way of parsing arguments. Several gdb commands use buildargv to parse their
698 arguments. Plus we want to be able to write compatible python
699 implementations of gdb commands. */
702 gdbpy_string_to_argv (PyObject
*self
, PyObject
*args
)
706 if (!PyArg_ParseTuple (args
, "s", &input
))
709 gdbpy_ref
<> py_argv (PyList_New (0));
713 /* buildargv uses NULL to represent an empty argument list, but we can't use
714 that in Python. Instead, if ARGS is "" then return an empty list.
715 This undoes the NULL -> "" conversion that cmdpy_function does. */
719 gdb_argv
c_argv (input
);
721 for (char *arg
: c_argv
)
723 gdbpy_ref
<> argp (PyString_FromString (arg
));
726 || PyList_Append (py_argv
.get (), argp
.get ()) < 0)
731 return py_argv
.release ();