GDB command that adds a break with prefix functionality; load in GDB with `source breakprefix.py`

breakprefix.py

"""
Adds a command to GDB that let you specify prefix for a breakpoint.
E.g. you want to invoke `z foo` and make it so that it creates a breakpoint on `packageX.foo`.
With this command you just do `z prefix packageX.` and then `z SOMETHING` will
set a breakpoint at `packageX.SOMETHING`.

For more info on GDB API refer to its docs:
https://sourceware.org/gdb/onlinedocs/gdb/Commands-In-Python.html#Commands-In-Python
"""
import gdb

COMMAND_NAME = 'z'

class BreakWithPrefix(gdb.Command):
    """Breaks with given prefix"""

    def __init__(self):
        super(BreakWithPrefix, self).__init__(COMMAND_NAME, gdb.COMMAND_USER)
        self.prefix = None

    def invoke(self, arg, from_tty):
        prefix_str = 'prefix'

        if arg.startswith(prefix_str):
            self.prefix = arg[len(prefix_str)+1:]
            print('The command %s will break using prefix %s from now' % (COMMAND_NAME, self.prefix))

        elif not self.prefix:
            print('You need to apply a prefix first with "%s %s <prefix>"' % (COMMAND_NAME, prefix_str))

        else:
            cmd = 'break %s%s' % (self.prefix, arg)
            print("Executing '%s' command" % cmd)
            gdb.execute(cmd)


# registering command with GDB
BreakWithPrefix()