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