X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/3903760236c30e3b5ace7a4eefac3a269d68957c..d26ffc64f583ab2d29df48f13518685602bc8832:/tools/lldbmacros/core/kernelcore.py diff --git a/tools/lldbmacros/core/kernelcore.py b/tools/lldbmacros/core/kernelcore.py old mode 100644 new mode 100755 index 3c6e5802b..580887584 --- a/tools/lldbmacros/core/kernelcore.py +++ b/tools/lldbmacros/core/kernelcore.py @@ -46,12 +46,13 @@ def IterateLinkedList(element, field_name): elt = elt.__getattr__(field_name) #end of while loop -def IterateListEntry(element, element_type, field_name): +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 element_type - str : Type of the next element field_name - str : Name of the field in next element's structure + list_prefix - str : use 's' here to iterate SLIST_HEAD instead returns: A generator does not return. It is used for iterating value : an object thats of type (element_type) head->le_next. Always a pointer object @@ -60,12 +61,12 @@ def IterateListEntry(element, element_type, field_name): 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 - 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): @@ -160,6 +161,67 @@ def IterateQueue(queue_head, element_ptr_type, element_field_name, backwards=Fal cur_elt = unpack_ptr_and_recast(elt.GetChildMemberWithName(element_field_name).GetChildMemberWithName('next')) +def IterateRBTreeEntry(element, element_type, field_name): + """ iterate over a rbtree as defined with RB_HEAD in libkern/tree.h + element - value : Value object for rbh_root + element_type - str : Type of the link element + field_name - str : Name of the field in link 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__('rbh_root') + if type(element_type) == str: + element_type = gettype(element_type) + + # Walk to find min + parent = elt + while unsigned(elt) != 0: + parent = elt + elt = cast(elt.__getattr__(field_name).__getattr__('rbe_left'), element_type) + elt = parent + + # Now elt is min + while unsigned(elt) != 0: + yield elt + # implementation cribbed from RB_NEXT in libkern/tree.h + right = cast(elt.__getattr__(field_name).__getattr__('rbe_right'), element_type) + if unsigned(right) != 0: + elt = right + left = cast(elt.__getattr__(field_name).__getattr__('rbe_left'), element_type) + while unsigned(left) != 0: + elt = left + left = cast(elt.__getattr__(field_name).__getattr__('rbe_left'), element_type) + else: + + # avoid using GetValueFromAddress + addr = elt.__getattr__(field_name).__getattr__('rbe_parent')&~1 + parent = value(elt.GetSBValue().CreateValueFromExpression(None,'(void *)'+str(addr))) + parent = cast(parent, element_type) + + if unsigned(parent) != 0: + left = cast(parent.__getattr__(field_name).__getattr__('rbe_left'), element_type) + if (unsigned(parent) != 0) and (unsigned(elt) == unsigned(left)): + elt = parent + else: + if unsigned(parent) != 0: + right = cast(parent.__getattr__(field_name).__getattr__('rbe_right'), element_type) + while unsigned(parent) != 0 and (unsigned(elt) == unsigned(right)): + elt = parent + + # avoid using GetValueFromAddress + addr = elt.__getattr__(field_name).__getattr__('rbe_parent')&~1 + parent = value(elt.GetSBValue().CreateValueFromExpression(None,'(void *)'+str(addr))) + parent = cast(parent, element_type) + + right = cast(parent.__getattr__(field_name).__getattr__('rbe_right'), element_type) + + # avoid using GetValueFromAddress + addr = elt.__getattr__(field_name).__getattr__('rbe_parent')&~1 + elt = value(elt.GetSBValue().CreateValueFromExpression(None,'(void *)'+str(addr))) + elt = cast(elt, element_type) + + class KernelTarget(object): """ A common kernel object that provides access to kernel objects and information. The class holds global lists for task, terminated_tasks, procs, zones, zombroc etc. @@ -173,6 +235,7 @@ class KernelTarget(object): self._threads_list = [] self._tasks_list = [] self._coalitions_list = [] + self._thread_groups = [] self._allproc = [] self._terminated_tasks_list = [] self._zones_list = [] @@ -339,10 +402,10 @@ class KernelTarget(object): 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: - 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. @@ -408,6 +471,17 @@ class KernelTarget(object): 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 @@ -481,4 +555,18 @@ class KernelTarget(object): caching.SaveStaticCacheData("kern.ptrsize", self._ptrsize) return self._ptrsize + 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)