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

   1 # Copyright (C) 2013 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         else:
  34             fname = fname[::-1]
  35         return fname
  36
  37 class Dummy (FrameDecorator):
  38
  39     def __init__(self, fobj):
  40         super(Dummy, self).__init__(fobj)
  41         self.fobj = fobj
  42
  43     def function (self):
  44         return "Dummy function"
  45
  46     def address (self):
  47         return 0x123
  48
  49     def filename (self):
  50         return "Dummy filename"
  51
  52     def frame_args (self):
  53         return [("Foo",gdb.Value(12)),("Bar","Stuff"), ("FooBar",42)]
  54
  55     def frame_locals (self):
  56         return []
  57
  58     def line (self):
  59         return 0
  60
  61     def elided (self):
  62         return None
  63
  64 class FrameFilter ():
  65
  66     def __init__ (self):
  67         self.name = "Reverse"
  68         self.priority = 100
  69         self.enabled = True
  70         gdb.frame_filters [self.name] = self
  71
  72     def filter (self, frame_iter):
  73         frame_iter = itertools.imap (Reverse_Function,
  74                                      frame_iter)
  75         return frame_iter
  76
  77 class ElidingFrameDecorator(FrameDecorator):
  78
  79     def __init__(self, frame, elided_frames):
  80         super(ElidingFrameDecorator, self).__init__(frame)
  81         self.elided_frames = elided_frames
  82
  83     def elided(self):
  84         return iter(self.elided_frames)
  85
  86 class ElidingIterator:
  87     def __init__(self, ii):
  88         self.input_iterator = ii
  89
  90     def __iter__(self):
  91         return self
  92
  93     def next(self):
  94         frame = next(self.input_iterator)
  95         if str(frame.function()) != 'func1':
  96             return frame
  97
  98         # Suppose we want to return the 'func1' frame but elide the
  99         # next frame.  E.g., if call in our interpreter language takes
 100         # two C frames to implement, and the first one we see is the
 101         # "sentinel".
 102         elided = next(self.input_iterator)
 103         return ElidingFrameDecorator(frame, [elided])
 104
 105 class FrameElider ():
 106
 107     def __init__ (self):
 108         self.name = "Elider"
 109         self.priority = 900
 110         self.enabled = True
 111         gdb.frame_filters [self.name] = self
 112
 113     def filter (self, frame_iter):
 114         return ElidingIterator (frame_iter)
 115
 116 FrameFilter()
 117 FrameElider()