diff options
| -rw-r--r-- | lldb/scripts/Python/modify-python-lldb.py | 61 | ||||
| -rw-r--r-- | lldb/test/python_api/value/linked_list/Makefile | 5 | ||||
| -rw-r--r-- | lldb/test/python_api/value/linked_list/TestValueAPILinkedList.py | 96 | ||||
| -rw-r--r-- | lldb/test/python_api/value/linked_list/main.cpp | 45 |
4 files changed, 207 insertions, 0 deletions
diff --git a/lldb/scripts/Python/modify-python-lldb.py b/lldb/scripts/Python/modify-python-lldb.py index 789e2e73337..6b44f264132 100644 --- a/lldb/scripts/Python/modify-python-lldb.py +++ b/lldb/scripts/Python/modify-python-lldb.py @@ -82,6 +82,63 @@ def lldb_iter(obj, getsize, getelem): # ============================================================================== ''' +# +# linked_list_iter() is a special purpose iterator to treat the SBValue as a +# list data structure, where you specify the child member name which points to +# the next item on the list and you specify the end-of-list function which takes +# an SBValue and returns True if EOL is reached and False if not. +# +linked_list_iter_def = ''' + # ================================================== + # Iterator for lldb.SBValue treated as a linked list + # ================================================== + def linked_list_iter(self, next_item_name, end_of_list): + """A generator adaptor to support iteration for SBValue as a linked list. + + For example, + + # Test function to determine end of list. + def eol(val): + if not val: + return True + try: + # Test the semantics of the item we got. + id = val.GetChildMemberWithName("id") + if int(id.GetValue()) > 0: + return False + except: + pass + + # If we fall through to here. It could be that exception + # occurred or the "id" child member does not qualify as a + # valid item. Return True for EOL. + return True + + # Get Frame #0. + ... + + # Get variable 'task_head'. + task_head = frame0.FindVariable('task_head') + ... + + for t in task_head.linked_list_iter('next', eol): + print t + """ + try: + item = self.GetChildMemberWithName(next_item_name) + while item: + yield item + # Prepare for the next iteration. + item = item.GetChildMemberWithName(next_item_name) + if end_of_list(item): + break + except: + # Exception occurred. Stop the generator. + pass + + return +''' + # This supports the iteration protocol. iter_def = " def __iter__(self): return lldb_iter(self, '%s', '%s')" module_iter = " def module_iter(self): return lldb_iter(self, '%s', '%s')" @@ -266,6 +323,10 @@ for line in content.splitlines(): new_content.add_line(eq_def % (cls, list_to_frag(e[cls]))) new_content.add_line(ne_def) + # This special purpose iterator is for SBValue only!!! + if cls == "SBValue": + new_content.add_line(linked_list_iter_def) + # Next state will be NORMAL. state = NORMAL diff --git a/lldb/test/python_api/value/linked_list/Makefile b/lldb/test/python_api/value/linked_list/Makefile new file mode 100644 index 00000000000..314f1cb2f07 --- /dev/null +++ b/lldb/test/python_api/value/linked_list/Makefile @@ -0,0 +1,5 @@ +LEVEL = ../../../make + +CXX_SOURCES := main.cpp + +include $(LEVEL)/Makefile.rules diff --git a/lldb/test/python_api/value/linked_list/TestValueAPILinkedList.py b/lldb/test/python_api/value/linked_list/TestValueAPILinkedList.py new file mode 100644 index 00000000000..2720b2cef23 --- /dev/null +++ b/lldb/test/python_api/value/linked_list/TestValueAPILinkedList.py @@ -0,0 +1,96 @@ +""" +Test SBValue API linked_list_iter which treats the SBValue as a linked list and +supports iteration till the end of list is reached. +""" + +import os, time +import re +import unittest2 +import lldb, lldbutil +from lldbtest import * + +class ValueAsLinkedListTestCase(TestBase): + + mydir = os.path.join("python_api", "value", "linked_list") + + @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin") + @python_api_test + def test_with_dsym(self): + """Exercise SBValue API linked_list_iter.""" + d = {'EXE': self.exe_name} + self.buildDsym(dictionary=d) + self.setTearDownCleanup(dictionary=d) + self.linked_list_api(self.exe_name) + + @python_api_test + def test_with_dwarf(self): + """Exercise SBValue API linked_list_iter.""" + d = {'EXE': self.exe_name} + self.buildDwarf(dictionary=d) + self.setTearDownCleanup(dictionary=d) + self.linked_list_api(self.exe_name) + + def setUp(self): + # Call super's setUp(). + TestBase.setUp(self) + # We'll use the test method name as the exe_name. + self.exe_name = self.testMethodName + # Find the line number to break at. + self.line = line_number('main.cpp', '// Break at this line') + + def linked_list_api(self, exe_name): + """Exercise SBValue API linked_list-iter.""" + exe = os.path.join(os.getcwd(), exe_name) + + # Create a target by the debugger. + target = self.dbg.CreateTarget(exe) + self.assertTrue(target, VALID_TARGET) + + # Create the breakpoint inside function 'main'. + breakpoint = target.BreakpointCreateByLocation('main.cpp', self.line) + self.assertTrue(breakpoint, VALID_BREAKPOINT) + + # Now launch the process, and do not stop at entry point. + process = target.LaunchSimple(None, None, os.getcwd()) + self.assertTrue(process, PROCESS_IS_VALID) + + def eol(val): + """Test to determine end of list.""" + if not val: + return True + try: + id = val.GetChildMemberWithName("id") + if int(id.GetValue()) > 0: + return False + except: + #exc_type, exc_value, exc_tb = sys.exc_info() + #traceback.print_exception(exc_type, exc_value, exc_tb) + pass + + # If we fall through to here. It could be that exception + # occurred or the "id" child member does not qualify as a + # valid item. Return True for EOL. + return True + + # Get Frame #0. + self.assertTrue(process.GetState() == lldb.eStateStopped) + thread = lldbutil.get_stopped_thread(process, lldb.eStopReasonBreakpoint) + self.assertTrue(thread != None, "There should be a thread stopped due to breakpoint condition") + frame0 = thread.GetFrameAtIndex(0) + + # Get variable 'task_head'. + task_head = frame0.FindVariable('task_head') + self.assertTrue(task_head, VALID_VARIABLE) + self.DebugSBValue(task_head) + + cvf = lldbutil.ChildVisitingFormatter(indent_child=2) + for t in task_head.linked_list_iter('next', eol): + self.assertTrue(t, VALID_VARIABLE) + if self.TraceOn(): + print cvf.format(t) + +if __name__ == '__main__': + import atexit + lldb.SBDebugger.Initialize() + atexit.register(lambda: lldb.SBDebugger.Terminate()) + unittest2.main() diff --git a/lldb/test/python_api/value/linked_list/main.cpp b/lldb/test/python_api/value/linked_list/main.cpp new file mode 100644 index 00000000000..772d5ed5d1b --- /dev/null +++ b/lldb/test/python_api/value/linked_list/main.cpp @@ -0,0 +1,45 @@ +//===-- main.c --------------------------------------------------*- C++ -*-===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +#include <stdio.h> + +class Task { +public: + int id; + Task *next; + Task(int i, Task *n): + id(i), + next(n) + {} +}; + + +int main (int argc, char const *argv[]) +{ + Task *task_head = new Task(-1, NULL); + Task *task1 = new Task(1, NULL); + Task *task2 = new Task(2, NULL); + Task *task3 = new Task(3, NULL); // Orphaned. + Task *task4 = new Task(4, NULL); + Task *task5 = new Task(5, NULL); + + task_head->next = task1; + task1->next = task2; + task2->next = task4; + task4->next = task5; + + int total = 0; // Break at this line + Task *t = task_head; + while (t != NULL) { + if (t->id >= 0) + ++total; + t = t->next; + } + printf("We have a total number of %d tasks\n", total); + return 0; +} |
