@@ -34,6 +34,7 @@
#include "gdbsupport/rsp-low.h"
#include "gdbcmd.h"
#include "cli/cli-utils.h"
+#include "extension.h"
#include "gdbarch.h"
/* For maintenance commands. */
@@ -1317,6 +1318,9 @@ ftrace_add_pt (struct btrace_thread_info *btinfo,
uint64_t offset;
int status;
+ /* Register the ptwrite filter. */
+ apply_ext_lang_ptwrite_filter (btinfo);
+
for (;;)
{
struct pt_insn insn;
@@ -352,6 +352,14 @@ struct btrace_thread_info
displaying or stepping through the execution history. */
std::vector<std::string> aux_data;
+ /* Function pointer to the ptwrite callback. Returns the string returned
+ by the ptwrite filter function. */
+ std::string (*ptw_callback_fun) (const uint64_t payload, const uint64_t ip,
+ const void *ptw_context) = nullptr;
+
+ /* Context for the ptw_callback_fun. */
+ void *ptw_context = nullptr;
+
/* The function level offset. When added to each function's LEVEL,
this normalizes the function levels such that the smallest level
becomes zero. */
@@ -75,6 +75,7 @@ PYTHON_FILE_LIST = \
gdb/frames.py \
gdb/printing.py \
gdb/prompt.py \
+ gdb/ptwrite.py \
gdb/styling.py \
gdb/types.py \
gdb/unwinder.py \
@@ -183,6 +183,11 @@ struct extension_language_ops
enum ext_lang_frame_args args_type,
struct ui_out *out, int frame_low, int frame_high);
+ /* Used for registering the ptwrite filter to the current thread. */
+ void (*apply_ptwrite_filter)
+ (const struct extension_language_defn *extlang,
+ struct btrace_thread_info *btinfo);
+
/* Update values held by the extension language when OBJFILE is discarded.
New global types must be created for every such value, which must then be
updated to use the new types.
@@ -550,6 +550,19 @@ apply_ext_lang_frame_filter (frame_info_ptr frame,
return EXT_LANG_BT_NO_FILTERS;
}
+/* Used for registering the ptwrite filter to the current thread. */
+
+void
+apply_ext_lang_ptwrite_filter (btrace_thread_info *btinfo)
+{
+ for (const struct extension_language_defn *extlang : extension_languages)
+ {
+ if (extlang->ops != nullptr
+ && extlang->ops->apply_ptwrite_filter != nullptr)
+ extlang->ops->apply_ptwrite_filter (extlang, btinfo);
+ }
+}
+
/* Update values held by the extension language when OBJFILE is discarded.
New global types must be created for every such value, which must then be
updated to use the new types.
@@ -295,6 +295,9 @@ extern enum ext_lang_bt_status apply_ext_lang_frame_filter
enum ext_lang_frame_args args_type,
struct ui_out *out, int frame_low, int frame_high);
+extern void apply_ext_lang_ptwrite_filter
+ (struct btrace_thread_info *btinfo);
+
extern void preserve_ext_lang_values (struct objfile *, htab_t copied_types);
extern const struct extension_language_defn *get_breakpoint_cond_ext_lang
@@ -124,6 +124,7 @@ static const struct extension_language_ops guile_extension_ops =
gdbscm_apply_val_pretty_printer,
NULL, /* gdbscm_apply_frame_filter, */
+ NULL, /* gdbscm_load_ptwrite_filter, */
gdbscm_preserve_values,
new file mode 100644
@@ -0,0 +1,80 @@
+# Ptwrite utilities.
+# Copyright (C) 2022 Free Software Foundation, Inc.
+
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+"""Utilities for working with ptwrite filters."""
+
+from copy import deepcopy
+import gdb
+
+
+def default_filter(payload, ip):
+ """Default filter that is active upon starting GDB."""
+ return "{:x}".format(payload)
+
+
+# This dict contains the per thread copies of the filter function and the
+# global template filter, from which the copies are created.
+_ptwrite_filter = {"global": default_filter}
+
+
+def ptwrite_exit_handler(event):
+ """Exit handler to prune _ptwrite_filter on inferior exit."""
+ for key in list(_ptwrite_filter.keys()):
+ if key.startswith(f"{event.inferior.pid}."):
+ del _ptwrite_filter[key]
+
+
+gdb.events.exited.connect(ptwrite_exit_handler)
+
+
+def _clear_traces(thread_list):
+ """Helper function to clear the trace of all threads in THREAD_LIST."""
+ current_thread = gdb.selected_thread()
+ recording = gdb.current_recording()
+
+ if recording is not None:
+ for thread in thread_list:
+ thread.switch()
+ recording.clear()
+
+ current_thread.switch()
+
+
+def register_filter(filter_):
+ """Register the ptwrite filter function."""
+ if filter_ is not None and not callable(filter_):
+ raise TypeError("The filter must be callable or 'None'.")
+
+ # Clear the traces of all threads to force re-decoding with
+ # the new filter.
+ thread_list = gdb.selected_inferior().threads()
+ _clear_traces(thread_list)
+
+ _ptwrite_filter.clear()
+ _ptwrite_filter["global"] = filter_
+
+
+def get_filter():
+ """Returns the filters of the current thread."""
+ # The key is of this format to enable an per-inferior cleanup when an
+ # inferior exits.
+ key = f"{gdb.selected_inferior().pid}.{gdb.selected_thread().ptid[1]}"
+
+ # Create a new filter for new threads.
+ if key not in _ptwrite_filter.keys():
+ _ptwrite_filter[key] = deepcopy(_ptwrite_filter["global"])
+
+ return _ptwrite_filter[key]
@@ -763,6 +763,94 @@ recpy_bt_function_call_history (PyObject *self, void *closure)
return btpy_list_new (tinfo, first, last, 1, &recpy_func_type);
}
+/* Helper function that calls PTW_FILTER with PAYLOAD and IP as arguments.
+ Returns the string that will be printed. */
+std::string
+recpy_call_filter (const uint64_t payload, const uint64_t ip,
+ const void *ptw_filter)
+{
+ std::string result;
+
+ if ((PyObject *) ptw_filter == nullptr)
+ error (_("No valid ptwrite filter."));
+ if ((PyObject *) ptw_filter == Py_None)
+ return result;
+
+ gdbpy_enter enter_py;
+
+ gdbpy_ref<> py_payload (PyLong_FromUnsignedLongLong (payload));
+ gdbpy_ref<> py_ip (PyLong_FromUnsignedLongLong (ip));
+
+ if (ip == 0)
+ py_ip = gdbpy_ref<>::new_reference (Py_None);
+
+ gdbpy_ref<> py_result (PyObject_CallFunctionObjArgs ((PyObject *) ptw_filter,
+ py_payload.get (),
+ py_ip.get (),
+ nullptr));
+
+ if (PyErr_Occurred ())
+ {
+ gdbpy_print_stack ();
+ gdbpy_error (_("Couldn't call the ptwrite filter."));
+ }
+
+ /* Py_None is valid and results in no output. */
+ if (py_result == Py_None)
+ return result;
+
+ result = gdbpy_obj_to_string (py_result.get ()).get ();
+
+ if (PyErr_Occurred ())
+ {
+ gdbpy_print_stack ();
+ gdbpy_error (_("The ptwrite filter didn't return a string."));
+ }
+
+ return result;
+}
+
+/* Helper function returning the current ptwrite filter. */
+
+PyObject *
+get_ptwrite_filter ()
+{
+ gdbpy_ref<> module (PyImport_ImportModule ("gdb.ptwrite"));
+
+ if (PyErr_Occurred ())
+ {
+ gdbpy_print_stack ();
+ return nullptr;
+ }
+
+ /* We need to keep the reference count. */
+ gdbpy_ref<> ptw_filter (PyObject_CallMethod (module.get (), "get_filter",
+ nullptr));
+
+ if (PyErr_Occurred ())
+ {
+ gdbpy_print_stack ();
+ gdbpy_error (_("Couldn't get the ptwrite filter."));
+ }
+
+ return ptw_filter.get();
+}
+
+/* Used for registering the default ptwrite filter to the current thread. A
+ pointer to this function is stored in the python extension interface. */
+
+void
+gdbpy_load_ptwrite_filter (const struct extension_language_defn *extlang,
+ struct btrace_thread_info *btinfo)
+{
+ gdb_assert (btinfo != nullptr);
+
+ gdbpy_enter enter_py;
+
+ btinfo->ptw_callback_fun = &recpy_call_filter;
+ btinfo->ptw_context= get_ptwrite_filter ();
+}
+
/* Implementation of BtraceRecord.goto (self, BtraceInstruction) -> None. */
PyObject *
@@ -91,4 +91,12 @@ extern PyObject *recpy_bt_func_prev (PyObject *self, void *closure);
/* Implementation of RecordFunctionSegment.next [RecordFunctionSegment]. */
extern PyObject *recpy_bt_func_next (PyObject *self, void *closure);
+/* Helper function returning the current ptwrite filter. */
+extern PyObject *get_ptwrite_filter ();
+
+/* Helper function to call the ptwrite filter. */
+extern std::string recpy_call_filter (const uint64_t payload,
+ const uint64_t ip,
+ const void *ptw_filter);
+
#endif /* PYTHON_PY_RECORD_BTRACE_H */
@@ -376,6 +376,9 @@ extern enum ext_lang_rc gdbpy_apply_val_pretty_printer
struct ui_file *stream, int recurse,
const struct value_print_options *options,
const struct language_defn *language);
+extern void gdbpy_load_ptwrite_filter
+ (const struct extension_language_defn *extlang,
+ struct btrace_thread_info *btinfo);
extern enum ext_lang_bt_status gdbpy_apply_frame_filter
(const struct extension_language_defn *,
frame_info_ptr frame, frame_filter_flags flags,
@@ -151,6 +151,8 @@ static const struct extension_language_ops python_extension_ops =
gdbpy_apply_frame_filter,
+ gdbpy_load_ptwrite_filter,
+
gdbpy_preserve_values,
gdbpy_breakpoint_has_cond,