[Lldb-commits] [lldb] r135938 - in /lldb/trunk: scripts/Python/modify-python-lldb.py test/python_api/value/linked_list/ test/python_api/value/linked_list/Makefile test/python_api/value/linked_list/TestValueAPILinkedList.py test/python_api/value/linked_list/main.cpp
Johnny Chen
johnny.chen at apple.com
Mon Jul 25 12:32:35 PDT 2011
Author: johnny
Date: Mon Jul 25 14:32:35 2011
New Revision: 135938
URL: http://llvm.org/viewvc/llvm-project?rev=135938&view=rev
Log:
Provide an add-on API to SBValue class by post-processing to provide a way
to iterate through an SBValue instance by treating it as the head of a linked
list. API program must provide two args to the linked_list_iter() method:
the first being the child member name which points to the next item on the list
and the second being a Python function which an SBValue (for the next item) and
returns True if end of list is reached, otherwise it returns False.
For example, suppose we have the following sample program.
#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;
}
The test program produces the following output while exercising the linked_list_iter() SBVAlue API:
task_head:
TypeName -> Task *
ByteSize -> 8
NumChildren -> 2
Value -> 0x0000000106400380
ValueType -> local_variable
Summary -> None
IsPointerType -> True
Location -> 0x00007fff65f06e60
(Task *) next = 0x0000000106400390
(int) id = 1
(Task *) next = 0x00000001064003a0
(Task *) next = 0x00000001064003a0
(int) id = 2
(Task *) next = 0x00000001064003c0
(Task *) next = 0x00000001064003c0
(int) id = 4
(Task *) next = 0x00000001064003d0
(Task *) next = 0x00000001064003d0
(int) id = 5
(Task *) next = 0x0000000000000000
Added:
lldb/trunk/test/python_api/value/linked_list/
lldb/trunk/test/python_api/value/linked_list/Makefile
lldb/trunk/test/python_api/value/linked_list/TestValueAPILinkedList.py
lldb/trunk/test/python_api/value/linked_list/main.cpp
Modified:
lldb/trunk/scripts/Python/modify-python-lldb.py
Modified: lldb/trunk/scripts/Python/modify-python-lldb.py
URL: http://llvm.org/viewvc/llvm-project/lldb/trunk/scripts/Python/modify-python-lldb.py?rev=135938&r1=135937&r2=135938&view=diff
==============================================================================
--- lldb/trunk/scripts/Python/modify-python-lldb.py (original)
+++ lldb/trunk/scripts/Python/modify-python-lldb.py Mon Jul 25 14:32:35 2011
@@ -82,6 +82,63 @@
# ==============================================================================
'''
+#
+# 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 @@
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
Added: lldb/trunk/test/python_api/value/linked_list/Makefile
URL: http://llvm.org/viewvc/llvm-project/lldb/trunk/test/python_api/value/linked_list/Makefile?rev=135938&view=auto
==============================================================================
--- lldb/trunk/test/python_api/value/linked_list/Makefile (added)
+++ lldb/trunk/test/python_api/value/linked_list/Makefile Mon Jul 25 14:32:35 2011
@@ -0,0 +1,5 @@
+LEVEL = ../../../make
+
+CXX_SOURCES := main.cpp
+
+include $(LEVEL)/Makefile.rules
Added: lldb/trunk/test/python_api/value/linked_list/TestValueAPILinkedList.py
URL: http://llvm.org/viewvc/llvm-project/lldb/trunk/test/python_api/value/linked_list/TestValueAPILinkedList.py?rev=135938&view=auto
==============================================================================
--- lldb/trunk/test/python_api/value/linked_list/TestValueAPILinkedList.py (added)
+++ lldb/trunk/test/python_api/value/linked_list/TestValueAPILinkedList.py Mon Jul 25 14:32:35 2011
@@ -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()
Added: lldb/trunk/test/python_api/value/linked_list/main.cpp
URL: http://llvm.org/viewvc/llvm-project/lldb/trunk/test/python_api/value/linked_list/main.cpp?rev=135938&view=auto
==============================================================================
--- lldb/trunk/test/python_api/value/linked_list/main.cpp (added)
+++ lldb/trunk/test/python_api/value/linked_list/main.cpp Mon Jul 25 14:32:35 2011
@@ -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;
+}
More information about the lldb-commits
mailing list