gdb/testsuite/gdb.python/py-framefilter.py

   1 # Copyright (C) 2013-2016 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-based
  17 # frame-filters.
  18 import gdb
  19 import itertools
  20 from gdb.FrameDecorator import FrameDecorator
  21 import copy
  22
  23 class Reverse_Function (FrameDecorator):
  24
  25     def __init__(self, fobj):
  26         super(Reverse_Function, self).__init__(fobj)
  27         self.fobj = fobj
  28
  29     def function (self):
  30         fname = str (self.fobj.function())
  31         if (fname == None or fname == ""):
  32             return None
  33         if fname == 'end_func':
  34             extra = self.fobj.inferior_frame().read_var('str').string()
  35         else:
  36             extra = ''
  37         fname = fname[::-1] + extra
  38         return fname
  39
  40 class Dummy (FrameDecorator):
  41
  42     def __init__(self, fobj):
  43         super(Dummy, self).__init__(fobj)
  44         self.fobj = fobj
  45
  46     def function (self):
  47         return "Dummy function"
  48
  49     def address (self):
  50         return 0x123
  51
  52     def filename (self):
  53         return "Dummy filename"
  54
  55     def frame_args (self):
  56         return [("Foo",gdb.Value(12)),("Bar","Stuff"), ("FooBar",42)]
  57
  58     def frame_locals (self):
  59         return []
  60
  61     def line (self):
  62         return 0
  63
  64     def elided (self):
  65         return None
  66
  67 class FrameFilter ():
  68
  69     def __init__ (self):
  70         self.name = "Reverse"
  71         self.priority = 100
  72         self.enabled = True
  73         gdb.frame_filters [self.name] = self
  74
  75     def filter (self, frame_iter):
  76         # Python 3.x moved the itertools.imap functionality to map(),
  77         # so check if it is available.
  78         if hasattr(itertools, "imap"):
  79             frame_iter = itertools.imap (Reverse_Function,
  80                                          frame_iter)
  81         else:
  82             frame_iter = map(Reverse_Function, frame_iter)
  83
  84         return frame_iter
  85
  86 class ElidingFrameDecorator(FrameDecorator):
  87
  88     def __init__(self, frame, elided_frames):
  89         super(ElidingFrameDecorator, self).__init__(frame)
  90         self.elided_frames = elided_frames
  91
  92     def elided(self):
  93         return iter(self.elided_frames)
  94
  95     def address (self):
  96         # Regression test for an overflow in the python layer.
  97         bitsize = 8 * gdb.lookup_type('void').pointer().sizeof
  98         mask = (1 << bitsize) - 1
  99         return 0xffffffffffffffff & mask
 100
 101 class ElidingIterator:
 102     def __init__(self, ii):
 103         self.input_iterator = ii
 104
 105     def __iter__(self):
 106         return self
 107
 108     def next(self):
 109         frame = next(self.input_iterator)
 110         if str(frame.function()) != 'func1':
 111             return frame
 112
 113         # Suppose we want to return the 'func1' frame but elide the
 114         # next frame.  E.g., if call in our interpreter language takes
 115         # two C frames to implement, and the first one we see is the
 116         # "sentinel".
 117         elided = next(self.input_iterator)
 118         return ElidingFrameDecorator(frame, [elided])
 119
 120     # Python 3.x requires __next__(self) while Python 2.x requires
 121     # next(self).  Define next(self), and for Python 3.x create this
 122     # wrapper.
 123     def __next__(self):
 124         return self.next()
 125
 126 class FrameElider ():
 127
 128     def __init__ (self):
 129         self.name = "Elider"
 130         self.priority = 900
 131         self.enabled = True
 132         gdb.frame_filters [self.name] = self
 133
 134     def filter (self, frame_iter):
 135         return ElidingIterator (frame_iter)
 136
 137 # A simple decorator that gives an error when computing the function.
 138 class ErrorInName(FrameDecorator):
 139     def __init__(self, frame):
 140         FrameDecorator.__init__(self, frame)
 141
 142     def function(self):
 143         raise RuntimeError('whoops')
 144
 145 # A filter that supplies buggy frames.  Disabled by default.
 146 class ErrorFilter():
 147     def __init__ (self):
 148         self.name = "Error"
 149         self.priority = 1
 150         self.enabled = False
 151         gdb.frame_filters [self.name] = self
 152
 153     def filter(self, frame_iter):
 154         # Python 3.x moved the itertools.imap functionality to map(),
 155         # so check if it is available.
 156         if hasattr(itertools, "imap"):
 157             return itertools.imap (ErrorInName, frame_iter)
 158         else:
 159             return map(ErrorInName, frame_iter)
 160
 161 FrameFilter()
 162 FrameElider()
 163 ErrorFilter()