Home | History | Annotate | Download | only in test
      1 # Verify that gdb can pretty-print the various PyObject* types
      2 #
      3 # The code for testing gdb was adapted from similar work in Unladen Swallow's
      4 # Lib/test/test_jit_gdb.py
      5 
      6 import os
      7 import re
      8 import subprocess
      9 import sys
     10 import unittest
     11 import sysconfig
     12 
     13 from test.test_support import run_unittest, findfile
     14 
     15 try:
     16     gdb_version, _ = subprocess.Popen(["gdb", "--version"],
     17                                       stdout=subprocess.PIPE).communicate()
     18 except OSError:
     19     # This is what "no gdb" looks like.  There may, however, be other
     20     # errors that manifest this way too.
     21     raise unittest.SkipTest("Couldn't find gdb on the path")
     22 gdb_version_number = re.search("^GNU gdb [^\d]*(\d+)\.(\d)", gdb_version)
     23 gdb_major_version = int(gdb_version_number.group(1))
     24 gdb_minor_version = int(gdb_version_number.group(2))
     25 if gdb_major_version < 7:
     26     raise unittest.SkipTest("gdb versions before 7.0 didn't support python embedding"
     27                             " Saw:\n" + gdb_version)
     28 
     29 # Location of custom hooks file in a repository checkout.
     30 checkout_hook_path = os.path.join(os.path.dirname(sys.executable),
     31                                   'python-gdb.py')
     32 
     33 def run_gdb(*args, **env_vars):
     34     """Runs gdb in --batch mode with the additional arguments given by *args.
     35 
     36     Returns its (stdout, stderr)
     37     """
     38     if env_vars:
     39         env = os.environ.copy()
     40         env.update(env_vars)
     41     else:
     42         env = None
     43     base_cmd = ('gdb', '--batch')
     44     if (gdb_major_version, gdb_minor_version) >= (7, 4):
     45         base_cmd += ('-iex', 'add-auto-load-safe-path ' + checkout_hook_path)
     46     out, err = subprocess.Popen(base_cmd + args,
     47         stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env,
     48         ).communicate()
     49     return out, err
     50 
     51 # Verify that "gdb" was built with the embedded python support enabled:
     52 gdbpy_version, _ = run_gdb("--eval-command=python import sys; print sys.version_info")
     53 if not gdbpy_version:
     54     raise unittest.SkipTest("gdb not built with embedded python support")
     55 
     56 # Verify that "gdb" can load our custom hooks.  In theory this should never
     57 # fail, but we don't handle the case of the hooks file not existing if the
     58 # tests are run from an installed Python (we'll produce failures in that case).
     59 cmd = ['--args', sys.executable]
     60 _, gdbpy_errors = run_gdb('--args', sys.executable)
     61 if "auto-loading has been declined" in gdbpy_errors:
     62     msg = "gdb security settings prevent use of custom hooks: "
     63 
     64 def python_is_optimized():
     65     cflags = sysconfig.get_config_vars()['PY_CFLAGS']
     66     final_opt = ""
     67     for opt in cflags.split():
     68         if opt.startswith('-O'):
     69             final_opt = opt
     70     return (final_opt and final_opt != '-O0')
     71 
     72 def gdb_has_frame_select():
     73     # Does this build of gdb have gdb.Frame.select ?
     74     stdout, _ = run_gdb("--eval-command=python print(dir(gdb.Frame))")
     75     m = re.match(r'.*\[(.*)\].*', stdout)
     76     if not m:
     77         raise unittest.SkipTest("Unable to parse output from gdb.Frame.select test")
     78     gdb_frame_dir = m.group(1).split(', ')
     79     return "'select'" in gdb_frame_dir
     80 
     81 HAS_PYUP_PYDOWN = gdb_has_frame_select()
     82 
     83 class DebuggerTests(unittest.TestCase):
     84 
     85     """Test that the debugger can debug Python."""
     86 
     87     def get_stack_trace(self, source=None, script=None,
     88                         breakpoint='PyObject_Print',
     89                         cmds_after_breakpoint=None,
     90                         import_site=False):
     91         '''
     92         Run 'python -c SOURCE' under gdb with a breakpoint.
     93 
     94         Support injecting commands after the breakpoint is reached
     95 
     96         Returns the stdout from gdb
     97 
     98         cmds_after_breakpoint: if provided, a list of strings: gdb commands
     99         '''
    100         # We use "set breakpoint pending yes" to avoid blocking with a:
    101         #   Function "foo" not defined.
    102         #   Make breakpoint pending on future shared library load? (y or [n])
    103         # error, which typically happens python is dynamically linked (the
    104         # breakpoints of interest are to be found in the shared library)
    105         # When this happens, we still get:
    106         #   Function "PyObject_Print" not defined.
    107         # emitted to stderr each time, alas.
    108 
    109         # Initially I had "--eval-command=continue" here, but removed it to
    110         # avoid repeated print breakpoints when traversing hierarchical data
    111         # structures
    112 
    113         # Generate a list of commands in gdb's language:
    114         commands = ['set breakpoint pending yes',
    115                     'break %s' % breakpoint,
    116                     'run']
    117         if cmds_after_breakpoint:
    118             commands += cmds_after_breakpoint
    119         else:
    120             commands += ['backtrace']
    121 
    122         # print commands
    123 
    124         # Use "commands" to generate the arguments with which to invoke "gdb":
    125         args = ["gdb", "--batch"]
    126         args += ['--eval-command=%s' % cmd for cmd in commands]
    127         args += ["--args",
    128                  sys.executable]
    129 
    130         if not import_site:
    131             # -S suppresses the default 'import site'
    132             args += ["-S"]
    133 
    134         if source:
    135             args += ["-c", source]
    136         elif script:
    137             args += [script]
    138 
    139         # print args
    140         # print ' '.join(args)
    141 
    142         # Use "args" to invoke gdb, capturing stdout, stderr:
    143         out, err = run_gdb(*args, PYTHONHASHSEED='0')
    144 
    145         errlines = err.splitlines()
    146         unexpected_errlines = []
    147 
    148         # Ignore some benign messages on stderr.
    149         ignore_patterns = (
    150             'Function "%s" not defined.' % breakpoint,
    151             "warning: no loadable sections found in added symbol-file"
    152             " system-supplied DSO",
    153             "warning: Unable to find libthread_db matching"
    154             " inferior's thread library, thread debugging will"
    155             " not be available.",
    156             "warning: Cannot initialize thread debugging"
    157             " library: Debugger service failed",
    158             'warning: Could not load shared library symbols for '
    159             'linux-vdso.so',
    160             'warning: Could not load shared library symbols for '
    161             'linux-gate.so',
    162             'Do you need "set solib-search-path" or '
    163             '"set sysroot"?',
    164             )
    165         for line in errlines:
    166             if not line.startswith(ignore_patterns):
    167                 unexpected_errlines.append(line)
    168 
    169         # Ensure no unexpected error messages:
    170         self.assertEqual(unexpected_errlines, [])
    171         return out
    172 
    173     def get_gdb_repr(self, source,
    174                      cmds_after_breakpoint=None,
    175                      import_site=False):
    176         # Given an input python source representation of data,
    177         # run "python -c'print DATA'" under gdb with a breakpoint on
    178         # PyObject_Print and scrape out gdb's representation of the "op"
    179         # parameter, and verify that the gdb displays the same string
    180         #
    181         # For a nested structure, the first time we hit the breakpoint will
    182         # give us the top-level structure
    183         gdb_output = self.get_stack_trace(source, breakpoint='PyObject_Print',
    184                                           cmds_after_breakpoint=cmds_after_breakpoint,
    185                                           import_site=import_site)
    186         # gdb can insert additional '\n' and space characters in various places
    187         # in its output, depending on the width of the terminal it's connected
    188         # to (using its "wrap_here" function)
    189         m = re.match('.*#0\s+PyObject_Print\s+\(\s*op\=\s*(.*?),\s+fp=.*\).*',
    190                      gdb_output, re.DOTALL)
    191         if not m:
    192             self.fail('Unexpected gdb output: %r\n%s' % (gdb_output, gdb_output))
    193         return m.group(1), gdb_output
    194 
    195     def assertEndsWith(self, actual, exp_end):
    196         '''Ensure that the given "actual" string ends with "exp_end"'''
    197         self.assertTrue(actual.endswith(exp_end),
    198                         msg='%r did not end with %r' % (actual, exp_end))
    199 
    200     def assertMultilineMatches(self, actual, pattern):
    201         m = re.match(pattern, actual, re.DOTALL)
    202         self.assertTrue(m, msg='%r did not match %r' % (actual, pattern))
    203 
    204     def get_sample_script(self):
    205         return findfile('gdb_sample.py')
    206 
    207 class PrettyPrintTests(DebuggerTests):
    208     def test_getting_backtrace(self):
    209         gdb_output = self.get_stack_trace('print 42')
    210         self.assertTrue('PyObject_Print' in gdb_output)
    211 
    212     def assertGdbRepr(self, val, cmds_after_breakpoint=None):
    213         # Ensure that gdb's rendering of the value in a debugged process
    214         # matches repr(value) in this process:
    215         gdb_repr, gdb_output = self.get_gdb_repr('print ' + repr(val),
    216                                                  cmds_after_breakpoint)
    217         self.assertEqual(gdb_repr, repr(val), gdb_output)
    218 
    219     def test_int(self):
    220         'Verify the pretty-printing of various "int" values'
    221         self.assertGdbRepr(42)
    222         self.assertGdbRepr(0)
    223         self.assertGdbRepr(-7)
    224         self.assertGdbRepr(sys.maxint)
    225         self.assertGdbRepr(-sys.maxint)
    226 
    227     def test_long(self):
    228         'Verify the pretty-printing of various "long" values'
    229         self.assertGdbRepr(0L)
    230         self.assertGdbRepr(1000000000000L)
    231         self.assertGdbRepr(-1L)
    232         self.assertGdbRepr(-1000000000000000L)
    233 
    234     def test_singletons(self):
    235         'Verify the pretty-printing of True, False and None'
    236         self.assertGdbRepr(True)
    237         self.assertGdbRepr(False)
    238         self.assertGdbRepr(None)
    239 
    240     def test_dicts(self):
    241         'Verify the pretty-printing of dictionaries'
    242         self.assertGdbRepr({})
    243         self.assertGdbRepr({'foo': 'bar'})
    244         self.assertGdbRepr("{'foo': 'bar', 'douglas':42}")
    245 
    246     def test_lists(self):
    247         'Verify the pretty-printing of lists'
    248         self.assertGdbRepr([])
    249         self.assertGdbRepr(range(5))
    250 
    251     def test_strings(self):
    252         'Verify the pretty-printing of strings'
    253         self.assertGdbRepr('')
    254         self.assertGdbRepr('And now for something hopefully the same')
    255         self.assertGdbRepr('string with embedded NUL here \0 and then some more text')
    256         self.assertGdbRepr('this is byte 255:\xff and byte 128:\x80')
    257 
    258     def test_tuples(self):
    259         'Verify the pretty-printing of tuples'
    260         self.assertGdbRepr(tuple())
    261         self.assertGdbRepr((1,))
    262         self.assertGdbRepr(('foo', 'bar', 'baz'))
    263 
    264     def test_unicode(self):
    265         'Verify the pretty-printing of unicode values'
    266         # Test the empty unicode string:
    267         self.assertGdbRepr(u'')
    268 
    269         self.assertGdbRepr(u'hello world')
    270 
    271         # Test printing a single character:
    272         #    U+2620 SKULL AND CROSSBONES
    273         self.assertGdbRepr(u'\u2620')
    274 
    275         # Test printing a Japanese unicode string
    276         # (I believe this reads "mojibake", using 3 characters from the CJK
    277         # Unified Ideographs area, followed by U+3051 HIRAGANA LETTER KE)
    278         self.assertGdbRepr(u'\u6587\u5b57\u5316\u3051')
    279 
    280         # Test a character outside the BMP:
    281         #    U+1D121 MUSICAL SYMBOL C CLEF
    282         # This is:
    283         # UTF-8: 0xF0 0x9D 0x84 0xA1
    284         # UTF-16: 0xD834 0xDD21
    285         # This will only work on wide-unicode builds:
    286         self.assertGdbRepr(u"\U0001D121")
    287 
    288     def test_sets(self):
    289         'Verify the pretty-printing of sets'
    290         self.assertGdbRepr(set())
    291         rep = self.get_gdb_repr("print set(['a', 'b'])")[0]
    292         self.assertTrue(rep.startswith("set(["))
    293         self.assertTrue(rep.endswith("])"))
    294         self.assertEqual(eval(rep), {'a', 'b'})
    295         rep = self.get_gdb_repr("print set([4, 5])")[0]
    296         self.assertTrue(rep.startswith("set(["))
    297         self.assertTrue(rep.endswith("])"))
    298         self.assertEqual(eval(rep), {4, 5})
    299 
    300         # Ensure that we handled sets containing the "dummy" key value,
    301         # which happens on deletion:
    302         gdb_repr, gdb_output = self.get_gdb_repr('''s = set(['a','b'])
    303 s.pop()
    304 print s''')
    305         self.assertEqual(gdb_repr, "set(['b'])")
    306 
    307     def test_frozensets(self):
    308         'Verify the pretty-printing of frozensets'
    309         self.assertGdbRepr(frozenset())
    310         rep = self.get_gdb_repr("print frozenset(['a', 'b'])")[0]
    311         self.assertTrue(rep.startswith("frozenset(["))
    312         self.assertTrue(rep.endswith("])"))
    313         self.assertEqual(eval(rep), {'a', 'b'})
    314         rep = self.get_gdb_repr("print frozenset([4, 5])")[0]
    315         self.assertTrue(rep.startswith("frozenset(["))
    316         self.assertTrue(rep.endswith("])"))
    317         self.assertEqual(eval(rep), {4, 5})
    318 
    319     def test_exceptions(self):
    320         # Test a RuntimeError
    321         gdb_repr, gdb_output = self.get_gdb_repr('''
    322 try:
    323     raise RuntimeError("I am an error")
    324 except RuntimeError, e:
    325     print e
    326 ''')
    327         self.assertEqual(gdb_repr,
    328                          "exceptions.RuntimeError('I am an error',)")
    329 
    330 
    331         # Test division by zero:
    332         gdb_repr, gdb_output = self.get_gdb_repr('''
    333 try:
    334     a = 1 / 0
    335 except ZeroDivisionError, e:
    336     print e
    337 ''')
    338         self.assertEqual(gdb_repr,
    339                          "exceptions.ZeroDivisionError('integer division or modulo by zero',)")
    340 
    341     def test_classic_class(self):
    342         'Verify the pretty-printing of classic class instances'
    343         gdb_repr, gdb_output = self.get_gdb_repr('''
    344 class Foo:
    345     pass
    346 foo = Foo()
    347 foo.an_int = 42
    348 print foo''')
    349         m = re.match(r'<Foo\(an_int=42\) at remote 0x[0-9a-f]+>', gdb_repr)
    350         self.assertTrue(m,
    351                         msg='Unexpected classic-class rendering %r' % gdb_repr)
    352 
    353     def test_modern_class(self):
    354         'Verify the pretty-printing of new-style class instances'
    355         gdb_repr, gdb_output = self.get_gdb_repr('''
    356 class Foo(object):
    357     pass
    358 foo = Foo()
    359 foo.an_int = 42
    360 print foo''')
    361         m = re.match(r'<Foo\(an_int=42\) at remote 0x[0-9a-f]+>', gdb_repr)
    362         self.assertTrue(m,
    363                         msg='Unexpected new-style class rendering %r' % gdb_repr)
    364 
    365     def test_subclassing_list(self):
    366         'Verify the pretty-printing of an instance of a list subclass'
    367         gdb_repr, gdb_output = self.get_gdb_repr('''
    368 class Foo(list):
    369     pass
    370 foo = Foo()
    371 foo += [1, 2, 3]
    372 foo.an_int = 42
    373 print foo''')
    374         m = re.match(r'<Foo\(an_int=42\) at remote 0x[0-9a-f]+>', gdb_repr)
    375         self.assertTrue(m,
    376                         msg='Unexpected new-style class rendering %r' % gdb_repr)
    377 
    378     def test_subclassing_tuple(self):
    379         'Verify the pretty-printing of an instance of a tuple subclass'
    380         # This should exercise the negative tp_dictoffset code in the
    381         # new-style class support
    382         gdb_repr, gdb_output = self.get_gdb_repr('''
    383 class Foo(tuple):
    384     pass
    385 foo = Foo((1, 2, 3))
    386 foo.an_int = 42
    387 print foo''')
    388         m = re.match(r'<Foo\(an_int=42\) at remote 0x[0-9a-f]+>', gdb_repr)
    389         self.assertTrue(m,
    390                         msg='Unexpected new-style class rendering %r' % gdb_repr)
    391 
    392     def assertSane(self, source, corruption, expvalue=None, exptype=None):
    393         '''Run Python under gdb, corrupting variables in the inferior process
    394         immediately before taking a backtrace.
    395 
    396         Verify that the variable's representation is the expected failsafe
    397         representation'''
    398         if corruption:
    399             cmds_after_breakpoint=[corruption, 'backtrace']
    400         else:
    401             cmds_after_breakpoint=['backtrace']
    402 
    403         gdb_repr, gdb_output = \
    404             self.get_gdb_repr(source,
    405                               cmds_after_breakpoint=cmds_after_breakpoint)
    406 
    407         if expvalue:
    408             if gdb_repr == repr(expvalue):
    409                 # gdb managed to print the value in spite of the corruption;
    410                 # this is good (see http://bugs.python.org/issue8330)
    411                 return
    412 
    413         if exptype:
    414             pattern = '<' + exptype + ' at remote 0x[0-9a-f]+>'
    415         else:
    416             # Match anything for the type name; 0xDEADBEEF could point to
    417             # something arbitrary (see  http://bugs.python.org/issue8330)
    418             pattern = '<.* at remote 0x[0-9a-f]+>'
    419 
    420         m = re.match(pattern, gdb_repr)
    421         if not m:
    422             self.fail('Unexpected gdb representation: %r\n%s' % \
    423                           (gdb_repr, gdb_output))
    424 
    425     def test_NULL_ptr(self):
    426         'Ensure that a NULL PyObject* is handled gracefully'
    427         gdb_repr, gdb_output = (
    428             self.get_gdb_repr('print 42',
    429                               cmds_after_breakpoint=['set variable op=0',
    430                                                      'backtrace'])
    431             )
    432 
    433         self.assertEqual(gdb_repr, '0x0')
    434 
    435     def test_NULL_ob_type(self):
    436         'Ensure that a PyObject* with NULL ob_type is handled gracefully'
    437         self.assertSane('print 42',
    438                         'set op->ob_type=0')
    439 
    440     def test_corrupt_ob_type(self):
    441         'Ensure that a PyObject* with a corrupt ob_type is handled gracefully'
    442         self.assertSane('print 42',
    443                         'set op->ob_type=0xDEADBEEF',
    444                         expvalue=42)
    445 
    446     def test_corrupt_tp_flags(self):
    447         'Ensure that a PyObject* with a type with corrupt tp_flags is handled'
    448         self.assertSane('print 42',
    449                         'set op->ob_type->tp_flags=0x0',
    450                         expvalue=42)
    451 
    452     def test_corrupt_tp_name(self):
    453         'Ensure that a PyObject* with a type with corrupt tp_name is handled'
    454         self.assertSane('print 42',
    455                         'set op->ob_type->tp_name=0xDEADBEEF',
    456                         expvalue=42)
    457 
    458     def test_NULL_instance_dict(self):
    459         'Ensure that a PyInstanceObject with with a NULL in_dict is handled'
    460         self.assertSane('''
    461 class Foo:
    462     pass
    463 foo = Foo()
    464 foo.an_int = 42
    465 print foo''',
    466                         'set ((PyInstanceObject*)op)->in_dict = 0',
    467                         exptype='Foo')
    468 
    469     def test_builtins_help(self):
    470         'Ensure that the new-style class _Helper in site.py can be handled'
    471         # (this was the issue causing tracebacks in
    472         #  http://bugs.python.org/issue8032#msg100537 )
    473 
    474         gdb_repr, gdb_output = self.get_gdb_repr('print __builtins__.help', import_site=True)
    475         m = re.match(r'<_Helper at remote 0x[0-9a-f]+>', gdb_repr)
    476         self.assertTrue(m,
    477                         msg='Unexpected rendering %r' % gdb_repr)
    478 
    479     def test_selfreferential_list(self):
    480         '''Ensure that a reference loop involving a list doesn't lead proxyval
    481         into an infinite loop:'''
    482         gdb_repr, gdb_output = \
    483             self.get_gdb_repr("a = [3, 4, 5] ; a.append(a) ; print a")
    484 
    485         self.assertEqual(gdb_repr, '[3, 4, 5, [...]]')
    486 
    487         gdb_repr, gdb_output = \
    488             self.get_gdb_repr("a = [3, 4, 5] ; b = [a] ; a.append(b) ; print a")
    489 
    490         self.assertEqual(gdb_repr, '[3, 4, 5, [[...]]]')
    491 
    492     def test_selfreferential_dict(self):
    493         '''Ensure that a reference loop involving a dict doesn't lead proxyval
    494         into an infinite loop:'''
    495         gdb_repr, gdb_output = \
    496             self.get_gdb_repr("a = {} ; b = {'bar':a} ; a['foo'] = b ; print a")
    497 
    498         self.assertEqual(gdb_repr, "{'foo': {'bar': {...}}}")
    499 
    500     def test_selfreferential_old_style_instance(self):
    501         gdb_repr, gdb_output = \
    502             self.get_gdb_repr('''
    503 class Foo:
    504     pass
    505 foo = Foo()
    506 foo.an_attr = foo
    507 print foo''')
    508         self.assertTrue(re.match('<Foo\(an_attr=<\.\.\.>\) at remote 0x[0-9a-f]+>',
    509                                  gdb_repr),
    510                         'Unexpected gdb representation: %r\n%s' % \
    511                             (gdb_repr, gdb_output))
    512 
    513     def test_selfreferential_new_style_instance(self):
    514         gdb_repr, gdb_output = \
    515             self.get_gdb_repr('''
    516 class Foo(object):
    517     pass
    518 foo = Foo()
    519 foo.an_attr = foo
    520 print foo''')
    521         self.assertTrue(re.match('<Foo\(an_attr=<\.\.\.>\) at remote 0x[0-9a-f]+>',
    522                                  gdb_repr),
    523                         'Unexpected gdb representation: %r\n%s' % \
    524                             (gdb_repr, gdb_output))
    525 
    526         gdb_repr, gdb_output = \
    527             self.get_gdb_repr('''
    528 class Foo(object):
    529     pass
    530 a = Foo()
    531 b = Foo()
    532 a.an_attr = b
    533 b.an_attr = a
    534 print a''')
    535         self.assertTrue(re.match('<Foo\(an_attr=<Foo\(an_attr=<\.\.\.>\) at remote 0x[0-9a-f]+>\) at remote 0x[0-9a-f]+>',
    536                                  gdb_repr),
    537                         'Unexpected gdb representation: %r\n%s' % \
    538                             (gdb_repr, gdb_output))
    539 
    540     def test_truncation(self):
    541         'Verify that very long output is truncated'
    542         gdb_repr, gdb_output = self.get_gdb_repr('print range(1000)')
    543         self.assertEqual(gdb_repr,
    544                          "[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, "
    545                          "14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, "
    546                          "27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, "
    547                          "40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, "
    548                          "53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, "
    549                          "66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, "
    550                          "79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, "
    551                          "92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, "
    552                          "104, 105, 106, 107, 108, 109, 110, 111, 112, 113, "
    553                          "114, 115, 116, 117, 118, 119, 120, 121, 122, 123, "
    554                          "124, 125, 126, 127, 128, 129, 130, 131, 132, 133, "
    555                          "134, 135, 136, 137, 138, 139, 140, 141, 142, 143, "
    556                          "144, 145, 146, 147, 148, 149, 150, 151, 152, 153, "
    557                          "154, 155, 156, 157, 158, 159, 160, 161, 162, 163, "
    558                          "164, 165, 166, 167, 168, 169, 170, 171, 172, 173, "
    559                          "174, 175, 176, 177, 178, 179, 180, 181, 182, 183, "
    560                          "184, 185, 186, 187, 188, 189, 190, 191, 192, 193, "
    561                          "194, 195, 196, 197, 198, 199, 200, 201, 202, 203, "
    562                          "204, 205, 206, 207, 208, 209, 210, 211, 212, 213, "
    563                          "214, 215, 216, 217, 218, 219, 220, 221, 222, 223, "
    564                          "224, 225, 226...(truncated)")
    565         self.assertEqual(len(gdb_repr),
    566                          1024 + len('...(truncated)'))
    567 
    568     def test_builtin_function(self):
    569         gdb_repr, gdb_output = self.get_gdb_repr('print len')
    570         self.assertEqual(gdb_repr, '<built-in function len>')
    571 
    572     def test_builtin_method(self):
    573         gdb_repr, gdb_output = self.get_gdb_repr('import sys; print sys.stdout.readlines')
    574         self.assertTrue(re.match('<built-in method readlines of file object at remote 0x[0-9a-f]+>',
    575                                  gdb_repr),
    576                         'Unexpected gdb representation: %r\n%s' % \
    577                             (gdb_repr, gdb_output))
    578 
    579     def test_frames(self):
    580         gdb_output = self.get_stack_trace('''
    581 def foo(a, b, c):
    582     pass
    583 
    584 foo(3, 4, 5)
    585 print foo.__code__''',
    586                                           breakpoint='PyObject_Print',
    587                                           cmds_after_breakpoint=['print (PyFrameObject*)(((PyCodeObject*)op)->co_zombieframe)']
    588                                           )
    589         self.assertTrue(re.match(r'.*\s+\$1 =\s+Frame 0x[0-9a-f]+, for file <string>, line 3, in foo \(\)\s+.*',
    590                                  gdb_output,
    591                                  re.DOTALL),
    592                         'Unexpected gdb representation: %r\n%s' % (gdb_output, gdb_output))
    593 
    594 @unittest.skipIf(python_is_optimized(),
    595                  "Python was compiled with optimizations")
    596 class PyListTests(DebuggerTests):
    597     def assertListing(self, expected, actual):
    598         self.assertEndsWith(actual, expected)
    599 
    600     def test_basic_command(self):
    601         'Verify that the "py-list" command works'
    602         bt = self.get_stack_trace(script=self.get_sample_script(),
    603                                   cmds_after_breakpoint=['py-list'])
    604 
    605         self.assertListing('   5    \n'
    606                            '   6    def bar(a, b, c):\n'
    607                            '   7        baz(a, b, c)\n'
    608                            '   8    \n'
    609                            '   9    def baz(*args):\n'
    610                            ' >10        print(42)\n'
    611                            '  11    \n'
    612                            '  12    foo(1, 2, 3)\n',
    613                            bt)
    614 
    615     def test_one_abs_arg(self):
    616         'Verify the "py-list" command with one absolute argument'
    617         bt = self.get_stack_trace(script=self.get_sample_script(),
    618                                   cmds_after_breakpoint=['py-list 9'])
    619 
    620         self.assertListing('   9    def baz(*args):\n'
    621                            ' >10        print(42)\n'
    622                            '  11    \n'
    623                            '  12    foo(1, 2, 3)\n',
    624                            bt)
    625 
    626     def test_two_abs_args(self):
    627         'Verify the "py-list" command with two absolute arguments'
    628         bt = self.get_stack_trace(script=self.get_sample_script(),
    629                                   cmds_after_breakpoint=['py-list 1,3'])
    630 
    631         self.assertListing('   1    # Sample script for use by test_gdb.py\n'
    632                            '   2    \n'
    633                            '   3    def foo(a, b, c):\n',
    634                            bt)
    635 
    636 class StackNavigationTests(DebuggerTests):
    637     @unittest.skipUnless(HAS_PYUP_PYDOWN, "test requires py-up/py-down commands")
    638     @unittest.skipIf(python_is_optimized(),
    639                      "Python was compiled with optimizations")
    640     def test_pyup_command(self):
    641         'Verify that the "py-up" command works'
    642         bt = self.get_stack_trace(script=self.get_sample_script(),
    643                                   cmds_after_breakpoint=['py-up'])
    644         self.assertMultilineMatches(bt,
    645                                     r'''^.*
    646 #[0-9]+ Frame 0x[0-9a-f]+, for file .*gdb_sample.py, line 7, in bar \(a=1, b=2, c=3\)
    647     baz\(a, b, c\)
    648 $''')
    649 
    650     @unittest.skipUnless(HAS_PYUP_PYDOWN, "test requires py-up/py-down commands")
    651     def test_down_at_bottom(self):
    652         'Verify handling of "py-down" at the bottom of the stack'
    653         bt = self.get_stack_trace(script=self.get_sample_script(),
    654                                   cmds_after_breakpoint=['py-down'])
    655         self.assertEndsWith(bt,
    656                             'Unable to find a newer python frame\n')
    657 
    658     @unittest.skipUnless(HAS_PYUP_PYDOWN, "test requires py-up/py-down commands")
    659     def test_up_at_top(self):
    660         'Verify handling of "py-up" at the top of the stack'
    661         bt = self.get_stack_trace(script=self.get_sample_script(),
    662                                   cmds_after_breakpoint=['py-up'] * 4)
    663         self.assertEndsWith(bt,
    664                             'Unable to find an older python frame\n')
    665 
    666     @unittest.skipUnless(HAS_PYUP_PYDOWN, "test requires py-up/py-down commands")
    667     @unittest.skipIf(python_is_optimized(),
    668                      "Python was compiled with optimizations")
    669     def test_up_then_down(self):
    670         'Verify "py-up" followed by "py-down"'
    671         bt = self.get_stack_trace(script=self.get_sample_script(),
    672                                   cmds_after_breakpoint=['py-up', 'py-down'])
    673         self.assertMultilineMatches(bt,
    674                                     r'''^.*
    675 #[0-9]+ Frame 0x[0-9a-f]+, for file .*gdb_sample.py, line 7, in bar \(a=1, b=2, c=3\)
    676     baz\(a, b, c\)
    677 #[0-9]+ Frame 0x[0-9a-f]+, for file .*gdb_sample.py, line 10, in baz \(args=\(1, 2, 3\)\)
    678     print\(42\)
    679 $''')
    680 
    681 class PyBtTests(DebuggerTests):
    682     @unittest.skipIf(python_is_optimized(),
    683                      "Python was compiled with optimizations")
    684     def test_basic_command(self):
    685         'Verify that the "py-bt" command works'
    686         bt = self.get_stack_trace(script=self.get_sample_script(),
    687                                   cmds_after_breakpoint=['py-bt'])
    688         self.assertMultilineMatches(bt,
    689                                     r'''^.*
    690 #[0-9]+ Frame 0x[0-9a-f]+, for file .*gdb_sample.py, line 7, in bar \(a=1, b=2, c=3\)
    691     baz\(a, b, c\)
    692 #[0-9]+ Frame 0x[0-9a-f]+, for file .*gdb_sample.py, line 4, in foo \(a=1, b=2, c=3\)
    693     bar\(a, b, c\)
    694 #[0-9]+ Frame 0x[0-9a-f]+, for file .*gdb_sample.py, line 12, in <module> \(\)
    695     foo\(1, 2, 3\)
    696 ''')
    697 
    698 class PyPrintTests(DebuggerTests):
    699     @unittest.skipIf(python_is_optimized(),
    700                      "Python was compiled with optimizations")
    701     def test_basic_command(self):
    702         'Verify that the "py-print" command works'
    703         bt = self.get_stack_trace(script=self.get_sample_script(),
    704                                   cmds_after_breakpoint=['py-print args'])
    705         self.assertMultilineMatches(bt,
    706                                     r".*\nlocal 'args' = \(1, 2, 3\)\n.*")
    707 
    708     @unittest.skipUnless(HAS_PYUP_PYDOWN, "test requires py-up/py-down commands")
    709     @unittest.skipIf(python_is_optimized(),
    710                      "Python was compiled with optimizations")
    711     def test_print_after_up(self):
    712         bt = self.get_stack_trace(script=self.get_sample_script(),
    713                                   cmds_after_breakpoint=['py-up', 'py-print c', 'py-print b', 'py-print a'])
    714         self.assertMultilineMatches(bt,
    715                                     r".*\nlocal 'c' = 3\nlocal 'b' = 2\nlocal 'a' = 1\n.*")
    716 
    717     @unittest.skipIf(python_is_optimized(),
    718                      "Python was compiled with optimizations")
    719     def test_printing_global(self):
    720         bt = self.get_stack_trace(script=self.get_sample_script(),
    721                                   cmds_after_breakpoint=['py-print __name__'])
    722         self.assertMultilineMatches(bt,
    723                                     r".*\nglobal '__name__' = '__main__'\n.*")
    724 
    725     @unittest.skipIf(python_is_optimized(),
    726                      "Python was compiled with optimizations")
    727     def test_printing_builtin(self):
    728         bt = self.get_stack_trace(script=self.get_sample_script(),
    729                                   cmds_after_breakpoint=['py-print len'])
    730         self.assertMultilineMatches(bt,
    731                                     r".*\nbuiltin 'len' = <built-in function len>\n.*")
    732 
    733 class PyLocalsTests(DebuggerTests):
    734     @unittest.skipIf(python_is_optimized(),
    735                      "Python was compiled with optimizations")
    736     def test_basic_command(self):
    737         bt = self.get_stack_trace(script=self.get_sample_script(),
    738                                   cmds_after_breakpoint=['py-locals'])
    739         self.assertMultilineMatches(bt,
    740                                     r".*\nargs = \(1, 2, 3\)\n.*")
    741 
    742     @unittest.skipUnless(HAS_PYUP_PYDOWN, "test requires py-up/py-down commands")
    743     @unittest.skipIf(python_is_optimized(),
    744                      "Python was compiled with optimizations")
    745     def test_locals_after_up(self):
    746         bt = self.get_stack_trace(script=self.get_sample_script(),
    747                                   cmds_after_breakpoint=['py-up', 'py-locals'])
    748         self.assertMultilineMatches(bt,
    749                                     r".*\na = 1\nb = 2\nc = 3\n.*")
    750 
    751 def test_main():
    752     run_unittest(PrettyPrintTests,
    753                  PyListTests,
    754                  StackNavigationTests,
    755                  PyBtTests,
    756                  PyPrintTests,
    757                  PyLocalsTests
    758                  )
    759 
    760 if __name__ == "__main__":
    761     test_main()
    762