xnu-4570.1.46.tar.gz

[apple/xnu.git] / tools / lldbmacros / core / kernelcore.py

diff --git a/tools/lldbmacros/core/kernelcore.py b/tools/lldbmacros/core/kernelcore.py
old mode 100644 (file)
new mode 100755 (executable)
index ec6295d..da145a4
--- a/tools/lldbmacros/core/kernelcore.py
+++ b/tools/lldbmacros/core/kernelcore.py
@@ -46,7 +46,25 @@ def IterateLinkedList(element, field_name):
         elt = elt.__getattr__(field_name)
     #end of while loop
 
         elt = elt.__getattr__(field_name)
     #end of while loop
 

-def IterateListEntry(element, element_type, field_name):
+def IterateSListEntry(element, element_type, field_name, slist_prefix=''):
+    """ iterate over a list as defined with SLIST_HEAD in bsd/sys/queue.h
+        params:
+            element      - value : Value object for slh_first
+            element_type - str   : Type of the next element
+            field_name   - str   : Name of the field in next element's structure
+        returns:
+            A generator does not return. It is used for iterating
+            value  : an object thats of type (element_type) head->sle_next. Always a pointer object
+    """
+    elt = element.__getattr__(slist_prefix + 'slh_first')
+    if type(element_type) == str:
+        element_type = gettype(element_type)
+    while unsigned(elt) != 0:
+        yield elt
+        next_el = elt.__getattr__(field_name).__getattr__(slist_prefix + 'sle_next')
+        elt = cast(next_el, element_type)
+
+def IterateListEntry(element, element_type, field_name, list_prefix=''):

     """ iterate over a list as defined with LIST_HEAD in bsd/sys/queue.h
         params:
             element      - value : Value object for lh_first
     """ iterate over a list as defined with LIST_HEAD in bsd/sys/queue.h
         params:
             element      - value : Value object for lh_first


@@ -60,12 +78,12 @@ def IterateListEntry(element, element_type, field_name):
             for pp in IterateListEntry(headp, 'struct proc *', 'p_sibling'):
                 print GetProcInfo(pp)
     """
             for pp in IterateListEntry(headp, 'struct proc *', 'p_sibling'):
                 print GetProcInfo(pp)
     """

-    elt = element.lh_first
+    elt = element.__getattr__(list_prefix + 'lh_first')

     if type(element_type) == str:
         element_type = gettype(element_type)
     while unsigned(elt) != 0:
         yield elt
     if type(element_type) == str:
         element_type = gettype(element_type)
     while unsigned(elt) != 0:
         yield elt

-        next_el = elt.__getattr__(field_name).le_next
+        next_el = elt.__getattr__(field_name).__getattr__(list_prefix + 'le_next')

         elt = cast(next_el, element_type)
 
 def IterateLinkageChain(queue_head, element_type, field_name, field_ofst=0):
         elt = cast(next_el, element_type)
 
 def IterateLinkageChain(queue_head, element_type, field_name, field_ofst=0):


@@ -109,7 +127,7 @@ def IterateLinkageChain(queue_head, element_type, field_name, field_ofst=0):
         link = link.next
 
 
         link = link.next
 
 

-def IterateQueue(queue_head, element_ptr_type, element_field_name, backwards=False):
+def IterateQueue(queue_head, element_ptr_type, element_field_name, backwards=False, unpack_ptr_fn=None):

     """ Iterate over an Element Chain queue in kernel of type queue_head_t. (osfmk/kern/queue.h method 2)
         params:
             queue_head         - value : Value object for queue_head.
     """ Iterate over an Element Chain queue in kernel of type queue_head_t. (osfmk/kern/queue.h method 2)
         params:
             queue_head         - value : Value object for queue_head.


@@ -117,6 +135,7 @@ def IterateQueue(queue_head, element_ptr_type, element_field_name, backwards=Fal
                                - str         : OR a string describing the type. ex. 'task *'
             element_field_name - str : name of the field in target struct.
             backwards          - backwards : traverse the queue backwards
                                - str         : OR a string describing the type. ex. 'task *'
             element_field_name - str : name of the field in target struct.
             backwards          - backwards : traverse the queue backwards

+            unpack_ptr_fn      - function : a function ptr of signature def unpack_ptr(long v) which returns long.

         returns:
             A generator does not return. It is used for iterating.
             value  : an object thats of type (element_type) queue_head->next. Always a pointer object
         returns:
             A generator does not return. It is used for iterating.
             value  : an object thats of type (element_type) queue_head->next. Always a pointer object


@@ -133,10 +152,19 @@ def IterateQueue(queue_head, element_ptr_type, element_field_name, backwards=Fal
         queue_head_addr = queue_head.GetValueAsUnsigned()
     else:
         queue_head_addr = queue_head.GetAddress().GetLoadAddress(LazyTarget.GetTarget())
         queue_head_addr = queue_head.GetValueAsUnsigned()
     else:
         queue_head_addr = queue_head.GetAddress().GetLoadAddress(LazyTarget.GetTarget())

+        
+    def unpack_ptr_and_recast(v):
+        if unpack_ptr_fn is None:
+            return v
+        v_unpacked = unpack_ptr_fn(v.GetValueAsUnsigned())
+        obj = v.CreateValueFromExpression(None,'(void *)'+str(v_unpacked))
+        obj.Cast(element_ptr_type)
+        return obj
+

     if backwards:
     if backwards:

-        cur_elt = queue_head.GetChildMemberWithName('prev')
+        cur_elt = unpack_ptr_and_recast(queue_head.GetChildMemberWithName('prev'))

     else:
     else:

-        cur_elt = queue_head.GetChildMemberWithName('next')
+        cur_elt = unpack_ptr_and_recast(queue_head.GetChildMemberWithName('next'))

 
     while True:
 
 
     while True:
 


@@ -145,9 +173,10 @@ def IterateQueue(queue_head, element_ptr_type, element_field_name, backwards=Fal
         elt = cur_elt.Cast(element_ptr_type)
         yield value(elt)
         if backwards:
         elt = cur_elt.Cast(element_ptr_type)
         yield value(elt)
         if backwards:

-            cur_elt = elt.GetChildMemberWithName(element_field_name).GetChildMemberWithName('prev')
+            cur_elt = unpack_ptr_and_recast(elt.GetChildMemberWithName(element_field_name).GetChildMemberWithName('prev'))

         else:
         else:

-            cur_elt = elt.GetChildMemberWithName(element_field_name).GetChildMemberWithName('next')
+            cur_elt = unpack_ptr_and_recast(elt.GetChildMemberWithName(element_field_name).GetChildMemberWithName('next'))
+

 
 class KernelTarget(object):
     """ A common kernel object that provides access to kernel objects and information.
 
 class KernelTarget(object):
     """ A common kernel object that provides access to kernel objects and information.


@@ -162,6 +191,7 @@ class KernelTarget(object):
         self._threads_list = []
         self._tasks_list = []
         self._coalitions_list = []
         self._threads_list = []
         self._tasks_list = []
         self._coalitions_list = []

+        self._thread_groups = []

         self._allproc = []
         self._terminated_tasks_list = []
         self._zones_list = []
         self._allproc = []
         self._terminated_tasks_list = []
         self._zones_list = []


@@ -328,10 +358,10 @@ class KernelTarget(object):
     def PhysToKernelVirt(self, addr):
         if self.arch == 'x86_64':
             return (addr + unsigned(self.GetGlobalVariable('physmap_base')))
     def PhysToKernelVirt(self, addr):
         if self.arch == 'x86_64':
             return (addr + unsigned(self.GetGlobalVariable('physmap_base')))

-        elif self.arch == 'arm' or self.arch == 'arm64':
+        elif self.arch.startswith('arm'):

             return (addr - unsigned(self.GetGlobalVariable("gPhysBase")) + unsigned(self.GetGlobalVariable("gVirtBase")))
         else:
             return (addr - unsigned(self.GetGlobalVariable("gPhysBase")) + unsigned(self.GetGlobalVariable("gVirtBase")))
         else:

-            raise ValueError("PhysToVirt does not support {0}".format(arch))
+            raise ValueError("PhysToVirt does not support {0}".format(self.arch))

 
     def GetNanotimeFromAbstime(self, abstime):
         """ convert absolute time (which is in MATUs) to nano seconds.
 
     def GetNanotimeFromAbstime(self, abstime):
         """ convert absolute time (which is in MATUs) to nano seconds.


@@ -358,9 +388,9 @@ class KernelTarget(object):
         if name == 'zones' :
             self._zones_list = caching.GetDynamicCacheData("kern._zones_list", [])
             if len(self._zones_list) > 0: return self._zones_list
         if name == 'zones' :
             self._zones_list = caching.GetDynamicCacheData("kern._zones_list", [])
             if len(self._zones_list) > 0: return self._zones_list

-            first_zone = self.GetGlobalVariable('first_zone')
-            for z in IterateLinkedList(first_zone, 'next_zone'):
-                self._zones_list.append(z)
+            zone_array = self.GetGlobalVariable('zone_array')
+            for i in range(0, self.GetGlobalVariable('num_zones')):
+                self._zones_list.append(addressof(zone_array[i]))

             caching.SaveDynamicCacheData("kern._zones_list", self._zones_list)
             return self._zones_list
 
             caching.SaveDynamicCacheData("kern._zones_list", self._zones_list)
             return self._zones_list
 


@@ -397,6 +427,17 @@ class KernelTarget(object):
             caching.SaveDynamicCacheData("kern._coalitions_list", self._coalitions_list)
             return self._coalitions_list
 
             caching.SaveDynamicCacheData("kern._coalitions_list", self._coalitions_list)
             return self._coalitions_list
 

+        if name == 'thread_groups' :
+            self._thread_groups_list = caching.GetDynamicCacheData("kern._thread_groups_list", [])
+            if len(self._thread_groups_list) > 0 : return self._thread_groups_list
+            thread_groups_queue_head = self.GetGlobalVariable('tg_queue')
+            thread_group_type = LazyTarget.GetTarget().FindFirstType('thread_group')
+            thread_groups_ptr_type = thread_group_type.GetPointerType()
+            for coal in IterateLinkageChain(addressof(thread_groups_queue_head), thread_groups_ptr_type, 'tg_queue_chain'):
+                self._thread_groups_list.append(coal)
+            caching.SaveDynamicCacheData("kern._thread_groups_list", self._thread_groups_list)
+            return self._thread_groups_list
+

         if name == 'terminated_tasks' :
             self._terminated_tasks_list = caching.GetDynamicCacheData("kern._terminated_tasks_list", [])
             if len(self._terminated_tasks_list) > 0 : return self._terminated_tasks_list
         if name == 'terminated_tasks' :
             self._terminated_tasks_list = caching.GetDynamicCacheData("kern._terminated_tasks_list", [])
             if len(self._terminated_tasks_list) > 0 : return self._terminated_tasks_list


@@ -470,5 +511,18 @@ class KernelTarget(object):
             caching.SaveStaticCacheData("kern.ptrsize", self._ptrsize)
             return self._ptrsize
 
             caching.SaveStaticCacheData("kern.ptrsize", self._ptrsize)
             return self._ptrsize
 

-        return object.__getattribute__(self, name)
+        if name == 'VM_MIN_KERNEL_ADDRESS':
+            if self.arch == 'x86_64':
+                return unsigned(0xFFFFFF8000000000)
+            elif self.arch == 'arm64':
+                return unsigned(0xffffffe000000000)
+            else:
+                return unsigned(0x80000000)

 
 

+        if name == 'VM_MIN_KERNEL_AND_KEXT_ADDRESS':
+            if self.arch == 'x86_64':
+                return self.VM_MIN_KERNEL_ADDRESS - 0x80000000
+            else:
+                return self.VM_MIN_KERNEL_ADDRESS
+
+        return object.__getattribute__(self, name)