gdb/testsuite/gdb.python/py-pp-maint.py

   1 # Copyright (C) 2010-2022 Free Software Foundation, Inc.
   2
   3 # This program is free software; you can redistribute it and/or modify
   4 # it under the terms of the GNU General Public License as published by
   5 # the Free Software Foundation; either version 3 of the License, or
   6 # (at your option) any later version.
   7 #
   8 # This program is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 # GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License
  14 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 # This file is part of the GDB testsuite.  It tests python pretty
  17 # printers.
  18
  19 import re
  20 import gdb.types
  21 import gdb.printing
  22
  23
  24 def lookup_function_lookup_test(val):
  25     class PrintFunctionLookup(object):
  26         def __init__(self, val):
  27             self.val = val
  28
  29         def to_string(self):
  30             return "x=<" + str(self.val["x"]) + "> y=<" + str(self.val["y"]) + ">"
  31
  32     typename = gdb.types.get_basic_type(val.type).tag
  33     # Note: typename could be None.
  34     if typename == "function_lookup_test":
  35         return PrintFunctionLookup(val)
  36     return None
  37
  38
  39 class pp_s(object):
  40     def __init__(self, val):
  41         self.val = val
  42
  43     def to_string(self):
  44         a = self.val["a"]
  45         b = self.val["b"]
  46         if a.address != b:
  47             raise Exception("&a(%s) != b(%s)" % (str(a.address), str(b)))
  48         return "a=<" + str(self.val["a"]) + "> b=<" + str(self.val["b"]) + ">"
  49
  50
  51 class pp_ss(object):
  52     def __init__(self, val):
  53         self.val = val
  54
  55     def to_string(self):
  56         return "a=<" + str(self.val["a"]) + "> b=<" + str(self.val["b"]) + ">"
  57
  58
  59 def build_pretty_printer():
  60     pp = gdb.printing.RegexpCollectionPrettyPrinter("pp-test")
  61
  62     pp.add_printer("struct s", "^struct s$", pp_s)
  63     pp.add_printer("s", "^s$", pp_s)
  64
  65     # Use a lambda this time to exercise doing things this way.
  66     pp.add_printer("struct ss", "^struct ss$", lambda val: pp_ss(val))
  67     pp.add_printer("ss", "^ss$", lambda val: pp_ss(val))
  68
  69     pp.add_printer(
  70         "enum flag_enum",
  71         "^flag_enum$",
  72         gdb.printing.FlagEnumerationPrinter("enum flag_enum"),
  73     )
  74
  75     return pp
  76
  77
  78 gdb.printing.register_pretty_printer(gdb, lookup_function_lookup_test)
  79 my_pretty_printer = build_pretty_printer()
  80 gdb.printing.register_pretty_printer(gdb, my_pretty_printer)