[+] added: dump_stlp_map.py (prints STLPort map)

git-svn-id: https://pykd.svn.codeplex.com/svn@63334 9b283d60-5439-405e-af05-b73fd8c4d996

snippets/dump_stlp_map.py

@@ -0,0 +1,109 @@
+"""Dump STLPort map"""
+
+import sys
+from pykd import *
+
+
+def println(msg):
+    if runningAsWinDbgExtension:
+        dprintln(msg)
+    else:
+        print msg
+
+
+class StlpNodeWrapper:
+    def __init__(self, addr):
+        # Wrapper specific field
+        self.addr = addr
+        self.color = ptrDWord(addr)
+        # By default, fields in structure aligned by 4 Bytes on x86 and by 8 Bytes on x64
+        addr += ptrSize()
+        self.parent = ptrPtr(addr)
+        addr += ptrSize()
+        self.left = ptrPtr(addr)
+        addr += ptrSize()
+        self.right = ptrPtr(addr)
+        # Wrapper specific field
+        self.sizeOf = addr + ptrSize() - self.addr
+
+
+class StlpMapWrapper:
+    def __init__(self, addr):
+        # Wrapper specific field
+        self.addr = addr
+        self.rootNode = StlpNodeWrapper(addr)
+        self.size = ptrPtr(addr + self.rootNode.sizeOf)
+
+
+def stlpMapIncrement(addr):
+    node = StlpNodeWrapper(addr)
+
+    if (node.right != 0):
+        node = StlpNodeWrapper(node.right)
+        while (node.left != 0):
+            node = StlpNodeWrapper(node.left)
+    else:
+        ynode = StlpNodeWrapper(node.parent)
+        while (node.addr == ynode.right):
+            node = ynode
+            ynode = StlpNodeWrapper(ynode.parent)
+        # check special case: This is necessary if _M_node is the
+        # _M_head and the tree contains only a single node __y. In
+        # that case parent, left and right all point to __y!
+        if (node.right != ynode.addr):
+          node = ynode
+
+    return node.addr
+
+
+def dumpStlportMap(addr):
+    """Returns the list of addresses of pair<key, value>"""
+    addrList = list()
+    #println("Map address: 0x%x" % addr)
+    map = StlpMapWrapper(addr)
+    #println("Map node count: %u" % map.size)
+
+    count = 0
+    begin = map.rootNode.left
+    end = addr
+    it = begin
+    while (it and it != end):
+        addrList.append(it + map.rootNode.sizeOf)
+        it = stlpMapIncrement(it)
+        count += 1
+
+    if (count != map.size):
+        println("Error: map was dumped incorrectly.")
+
+    return addrList
+
+
+def printUsage():
+    println("Usage:")
+    println("!py dump_stlp_map map_address [\"EXACT MAP PAIR TYPE\"]")
+    println("To obtain exact map pair type use:")
+    println("dt -r ModuleName!stlp_std::pair*")
+    println("Find required type in the list and copy paste it as parameter.")
+
+if __name__ == "__main__":
+    global runningAsWinDbgExtension
+
+    runningAsWinDbgExtension = not isSessionStart()
+    mapAddr = 0
+
+    argc = len(sys.argv)
+    if (argc < 2):
+        printUsage()
+        quit(0)
+    else:
+        mapAddr = int(sys.argv[1], 16)
+    
+    addrList = dumpStlportMap(mapAddr)
+    for addr in addrList:
+        if (argc == 2):
+            println("0x%x" % addr)
+        else:
+            s = "dt -r " + sys.argv[2] + " 0x%x" % addr
+            #println(s)
+            println("------------------------------------------------")
+            println(dbgCommand(s))