"""
Objective-C runtime wrapper for use by LLDB Python formatters
Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
See https://llvm.org/LICENSE.txt for license information.
SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""
import lldb
import lldb.formatters.cache
import lldb.formatters.attrib_fromdict
import functools
import lldb.formatters.Logger
class Utilities:
@staticmethod
def read_ascii(process, pointer, max_len=128):
logger = lldb.formatters.Logger.Logger()
error = lldb.SBError()
content = None
try:
content = process.ReadCStringFromMemory(pointer, max_len, error)
except:
pass
if content is None or len(content) == 0 or error.fail:
return None
return content
@staticmethod
def is_valid_pointer(pointer, pointer_size, allow_tagged=0, allow_NULL=0):
logger = lldb.formatters.Logger.Logger()
if pointer is None:
return 0
if pointer == 0:
return allow_NULL
if allow_tagged and (pointer % 2) == 1:
return 1
return (pointer % pointer_size) == 0
@staticmethod
def is_allowed_pointer(pointer):
logger = lldb.formatters.Logger.Logger()
if pointer is None:
return 0
return (pointer & 0xFFFF800000000000) == 0
@staticmethod
def read_child_of(valobj, offset, type):
logger = lldb.formatters.Logger.Logger()
if offset == 0 and type.GetByteSize() == valobj.GetByteSize():
return valobj.GetValueAsUnsigned()
child = valobj.CreateChildAtOffset("childUNK", offset, type)
if child is None or child.IsValid() == 0:
return None
return child.GetValueAsUnsigned()
@staticmethod
def is_valid_identifier(name):
logger = lldb.formatters.Logger.Logger()
if name is None:
return None
if len(name) == 0:
return None
ok_values = dict.fromkeys(
"$%_.-ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz1234567890"
)
return all(c in ok_values for c in name)
@staticmethod
def check_is_osx_lion(target):
logger = lldb.formatters.Logger.Logger()
try:
mod = target.module["Foundation"]
except:
mod = None
if mod is None or mod.IsValid() == 0:
return None
ver = mod.GetVersion()
if ver is None or ver == []:
return None
return ver[0] < 900
@staticmethod
def prepare_class_detection(valobj, statistics):
logger = lldb.formatters.Logger.Logger()
class_data = ObjCRuntime(valobj)
if class_data.is_valid() == 0:
statistics.metric_hit("invalid_pointer", valobj)
wrapper = InvalidPointer_Description(valobj.GetValueAsUnsigned(0) == 0)
return class_data, wrapper
class_data = class_data.read_class_data()
if class_data.is_valid() == 0:
statistics.metric_hit("invalid_isa", valobj)
wrapper = InvalidISA_Description()
return class_data, wrapper
if class_data.is_kvo():
class_data = class_data.get_superclass()
if class_data.class_name() == "_NSZombie_OriginalClass":
wrapper = ThisIsZombie_Description()
return class_data, wrapper
return class_data, None
class RoT_Data:
def __init__(self, rot_pointer, params):
logger = lldb.formatters.Logger.Logger()
if Utilities.is_valid_pointer(
rot_pointer.GetValueAsUnsigned(), params.pointer_size, allow_tagged=0
):
self.sys_params = params
self.valobj = rot_pointer
self.instanceSize = None
offset = 24 if self.sys_params.is_64_bit else 16
self.namePointer = Utilities.read_child_of(
self.valobj, offset, self.sys_params.types_cache.addr_ptr_type
)
self.valid = 1
else:
logger >> "Marking as invalid - rot is invalid"
self.valid = 0
if self.valid:
self.name = Utilities.read_ascii(
self.valobj.GetTarget().GetProcess(), self.namePointer
)
if not (Utilities.is_valid_identifier(self.name)):
logger >> "Marking as invalid - name is invalid"
self.valid = 0
def check_valid(self):
self.valid = 1
def __str__(self):
logger = lldb.formatters.Logger.Logger()
return (
"instanceSize = "
+ hex(self.instance_size())
+ "\n"
+ "namePointer = "
+ hex(self.namePointer)
+ " --> "
+ self.name
)
def is_valid(self):
return self.valid
def instance_size(self, align=0):
logger = lldb.formatters.Logger.Logger()
if self.is_valid() == 0:
return None
if self.instanceSize is None:
self.instanceSize = Utilities.read_child_of(
self.valobj, 8, self.sys_params.types_cache.uint32_t
)
if align:
unalign = self.instance_size(0)
if self.sys_params.is_64_bit:
return ((unalign + 7) & ~7) % 0x100000000
else:
return ((unalign + 3) & ~3) % 0x100000000
else:
return self.instanceSize
class RwT_Data:
def __init__(self, rwt_pointer, params):
logger = lldb.formatters.Logger.Logger()
if Utilities.is_valid_pointer(
rwt_pointer.GetValueAsUnsigned(), params.pointer_size, allow_tagged=0
):
self.sys_params = params
self.valobj = rwt_pointer
self.roPointer = Utilities.read_child_of(
self.valobj, 8, self.sys_params.types_cache.addr_ptr_type
)
self.check_valid()
else:
logger >> "Marking as invalid - rwt is invald"
self.valid = 0
if self.valid:
self.rot = self.valobj.CreateValueFromData(
"rot",
lldb.SBData.CreateDataFromUInt64Array(
self.sys_params.endianness,
self.sys_params.pointer_size,
[self.roPointer],
),
self.sys_params.types_cache.addr_ptr_type,
)
self.data = RoT_Data(self.rot, self.sys_params)
def check_valid(self):
logger = lldb.formatters.Logger.Logger()
self.valid = 1
if not (
Utilities.is_valid_pointer(
self.roPointer, self.sys_params.pointer_size, allow_tagged=0
)
):
logger >> "Marking as invalid - ropointer is invalid"
self.valid = 0
def __str__(self):
logger = lldb.formatters.Logger.Logger()
return "roPointer = " + hex(self.roPointer)
def is_valid(self):
logger = lldb.formatters.Logger.Logger()
if self.valid:
return self.data.is_valid()
return 0
class Class_Data_V2:
def __init__(self, isa_pointer, params):
logger = lldb.formatters.Logger.Logger()
if (isa_pointer is not None) and (
Utilities.is_valid_pointer(
isa_pointer.GetValueAsUnsigned(), params.pointer_size, allow_tagged=0
)
):
self.sys_params = params
self.valobj = isa_pointer
self.check_valid()
else:
logger >> "Marking as invalid - isa is invalid or None"
self.valid = 0
if self.valid:
self.rwt = self.valobj.CreateValueFromData(
"rwt",
lldb.SBData.CreateDataFromUInt64Array(
self.sys_params.endianness,
self.sys_params.pointer_size,
[self.dataPointer],
),
self.sys_params.types_cache.addr_ptr_type,
)
self.data = RwT_Data(self.rwt, self.sys_params)
def check_valid(self):
logger = lldb.formatters.Logger.Logger()
self.valid = 1
self.isaPointer = Utilities.read_child_of(
self.valobj, 0, self.sys_params.types_cache.addr_ptr_type
)
if not (
Utilities.is_valid_pointer(
self.isaPointer, self.sys_params.pointer_size, allow_tagged=0
)
):
logger >> "Marking as invalid - isaPointer is invalid"
self.valid = 0
return
if not (Utilities.is_allowed_pointer(self.isaPointer)):
logger >> "Marking as invalid - isaPointer is not allowed"
self.valid = 0
return
self.cachePointer = Utilities.read_child_of(
self.valobj,
2 * self.sys_params.pointer_size,
self.sys_params.types_cache.addr_ptr_type,
)
if not (
Utilities.is_valid_pointer(
self.cachePointer, self.sys_params.pointer_size, allow_tagged=0
)
):
logger >> "Marking as invalid - cachePointer is invalid"
self.valid = 0
return
if not (Utilities.is_allowed_pointer(self.cachePointer)):
logger >> "Marking as invalid - cachePointer is not allowed"
self.valid = 0
return
self.dataPointer = Utilities.read_child_of(
self.valobj,
4 * self.sys_params.pointer_size,
self.sys_params.types_cache.addr_ptr_type,
)
if not (
Utilities.is_valid_pointer(
self.dataPointer, self.sys_params.pointer_size, allow_tagged=0
)
):
logger >> "Marking as invalid - dataPointer is invalid"
self.valid = 0
return
if not (Utilities.is_allowed_pointer(self.dataPointer)):
logger >> "Marking as invalid - dataPointer is not allowed"
self.valid = 0
return
self.superclassIsaPointer = Utilities.read_child_of(
self.valobj,
1 * self.sys_params.pointer_size,
self.sys_params.types_cache.addr_ptr_type,
)
if not (
Utilities.is_valid_pointer(
self.superclassIsaPointer,
self.sys_params.pointer_size,
allow_tagged=0,
allow_NULL=1,
)
):
logger >> "Marking as invalid - superclassIsa is invalid"
self.valid = 0
return
if not (Utilities.is_allowed_pointer(self.superclassIsaPointer)):
logger >> "Marking as invalid - superclassIsa is not allowed"
self.valid = 0
return
def is_kvo(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
if self.class_name().startswith("NSKVONotifying_"):
return 1
return 0
def is_cftype(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
return self.class_name() == "__NSCFType" or self.class_name() == "NSCFType"
def get_superclass(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
parent_isa_pointer = self.valobj.CreateChildAtOffset(
"parent_isa",
self.sys_params.pointer_size,
self.sys_params.addr_ptr_type,
)
return Class_Data_V2(parent_isa_pointer, self.sys_params)
else:
return None
def class_name(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
return self.data.data.name
else:
return None
def is_valid(self):
logger = lldb.formatters.Logger.Logger()
if self.valid:
return self.data.is_valid()
return 0
def __str__(self):
logger = lldb.formatters.Logger.Logger()
return (
"isaPointer = "
+ hex(self.isaPointer)
+ "\n"
+ "superclassIsaPointer = "
+ hex(self.superclassIsaPointer)
+ "\n"
+ "cachePointer = "
+ hex(self.cachePointer)
+ "\n"
+ "data = "
+ hex(self.dataPointer)
)
def is_tagged(self):
return 0
def instance_size(self, align=0):
logger = lldb.formatters.Logger.Logger()
if self.is_valid() == 0:
return None
return self.rwt.rot.instance_size(align)
class Class_Data_V1:
def __init__(self, isa_pointer, params):
logger = lldb.formatters.Logger.Logger()
if (isa_pointer is not None) and (
Utilities.is_valid_pointer(
isa_pointer.GetValueAsUnsigned(), params.pointer_size, allow_tagged=0
)
):
self.valid = 1
self.sys_params = params
self.valobj = isa_pointer
self.check_valid()
else:
logger >> "Marking as invalid - isaPointer is invalid or None"
self.valid = 0
if self.valid:
self.name = Utilities.read_ascii(
self.valobj.GetTarget().GetProcess(), self.namePointer
)
if not (Utilities.is_valid_identifier(self.name)):
logger >> "Marking as invalid - name is not valid"
self.valid = 0
def check_valid(self):
logger = lldb.formatters.Logger.Logger()
self.valid = 1
self.isaPointer = Utilities.read_child_of(
self.valobj, 0, self.sys_params.types_cache.addr_ptr_type
)
if not (
Utilities.is_valid_pointer(
self.isaPointer, self.sys_params.pointer_size, allow_tagged=0
)
):
logger >> "Marking as invalid - isaPointer is invalid"
self.valid = 0
return
self.superclassIsaPointer = Utilities.read_child_of(
self.valobj,
1 * self.sys_params.pointer_size,
self.sys_params.types_cache.addr_ptr_type,
)
if not (
Utilities.is_valid_pointer(
self.superclassIsaPointer,
self.sys_params.pointer_size,
allow_tagged=0,
allow_NULL=1,
)
):
logger >> "Marking as invalid - superclassIsa is invalid"
self.valid = 0
return
self.namePointer = Utilities.read_child_of(
self.valobj,
2 * self.sys_params.pointer_size,
self.sys_params.types_cache.addr_ptr_type,
)
def is_kvo(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
if self.class_name().startswith("NSKVONotifying_"):
return 1
return 0
def is_cftype(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
return self.class_name() == "__NSCFType" or self.class_name() == "NSCFType"
def get_superclass(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
parent_isa_pointer = self.valobj.CreateChildAtOffset(
"parent_isa",
self.sys_params.pointer_size,
self.sys_params.addr_ptr_type,
)
return Class_Data_V1(parent_isa_pointer, self.sys_params)
else:
return None
def class_name(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
return self.name
else:
return None
def is_valid(self):
return self.valid
def __str__(self):
logger = lldb.formatters.Logger.Logger()
return (
"isaPointer = "
+ hex(self.isaPointer)
+ "\n"
+ "superclassIsaPointer = "
+ hex(self.superclassIsaPointer)
+ "\n"
+ "namePointer = "
+ hex(self.namePointer)
+ " --> "
+ self.name
+ "instanceSize = "
+ hex(self.instanceSize())
+ "\n"
)
def is_tagged(self):
return 0
def instance_size(self, align=0):
logger = lldb.formatters.Logger.Logger()
if self.is_valid() == 0:
return None
if self.instanceSize is None:
self.instanceSize = Utilities.read_child_of(
self.valobj,
5 * self.sys_params.pointer_size,
self.sys_params.types_cache.addr_ptr_type,
)
if align:
unalign = self.instance_size(0)
if self.sys_params.is_64_bit:
return ((unalign + 7) & ~7) % 0x100000000
else:
return ((unalign + 3) & ~3) % 0x100000000
else:
return self.instanceSize
TaggedClass_Values_Lion = {
1: "NSNumber",
5: "NSManagedObject",
6: "NSDate",
7: "NSDateTS",
}
TaggedClass_Values_NMOS = {
0: "NSAtom",
3: "NSNumber",
4: "NSDateTS",
5: "NSManagedObject",
6: "NSDate",
}
class TaggedClass_Data:
def __init__(self, pointer, params):
logger = lldb.formatters.Logger.Logger()
global TaggedClass_Values_Lion, TaggedClass_Values_NMOS
self.valid = 1
self.name = None
self.sys_params = params
self.valobj = pointer
self.val = (pointer & ~0x0000000000000000FF) >> 8
self.class_bits = (pointer & 0xE) >> 1
self.i_bits = (pointer & 0xF0) >> 4
if self.sys_params.is_lion:
if self.class_bits in TaggedClass_Values_Lion:
self.name = TaggedClass_Values_Lion[self.class_bits]
else:
logger >> "Marking as invalid - not a good tagged pointer for Lion"
self.valid = 0
else:
if self.class_bits in TaggedClass_Values_NMOS:
self.name = TaggedClass_Values_NMOS[self.class_bits]
else:
logger >> "Marking as invalid - not a good tagged pointer for NMOS"
self.valid = 0
def is_valid(self):
return self.valid
def class_name(self):
logger = lldb.formatters.Logger.Logger()
if self.is_valid():
return self.name
else:
return 0
def value(self):
return self.val if self.is_valid() else None
def info_bits(self):
return self.i_bits if self.is_valid() else None
def is_kvo(self):
return 0
def is_cftype(self):
return 0
def get_superclass(self):
return None
def is_tagged(self):
return 1
def instance_size(self, align=0):
logger = lldb.formatters.Logger.Logger()
if self.is_valid() == 0:
return None
return self.sys_params.pointer_size
class InvalidClass_Data:
def __init__(self):
pass
def is_valid(self):
return 0
class Version:
def __init__(self, major, minor, release, build_string):
self._major = major
self._minor = minor
self._release = release
self._build_string = build_string
def get_major(self):
return self._major
def get_minor(self):
return self._minor
def get_release(self):
return self._release
def get_build_string(self):
return self._build_string
major = property(get_major, None)
minor = property(get_minor, None)
release = property(get_release, None)
build_string = property(get_build_string, None)
def __lt__(self, other):
if self.major < other.major:
return 1
if self.minor < other.minor:
return 1
if self.release < other.release:
return 1
return 0
def __eq__(self, other):
return (
(self.major == other.major)
and (self.minor == other.minor)
and (self.release == other.release)
and (self.build_string == other.build_string)
)
def __gt__(self, other):
return other < self
def __le__(self, other):
return not other < self
def __ge__(self, other):
return not self < other
runtime_version = lldb.formatters.cache.Cache()
os_version = lldb.formatters.cache.Cache()
types_caches = lldb.formatters.cache.Cache()
isa_caches = lldb.formatters.cache.Cache()
class SystemParameters:
def __init__(self, valobj):
logger = lldb.formatters.Logger.Logger()
self.adjust_for_architecture(valobj)
self.adjust_for_process(valobj)
def adjust_for_process(self, valobj):
logger = lldb.formatters.Logger.Logger()
global runtime_version
global os_version
global types_caches
global isa_caches
process = valobj.GetTarget().GetProcess()
self.pid = process.GetUniqueID()
if runtime_version.look_for_key(self.pid):
self.runtime_version = runtime_version.get_value(self.pid)
else:
self.runtime_version = ObjCRuntime.runtime_version(process)
runtime_version.add_item(self.pid, self.runtime_version)
if os_version.look_for_key(self.pid):
self.is_lion = os_version.get_value(self.pid)
else:
self.is_lion = Utilities.check_is_osx_lion(valobj.GetTarget())
os_version.add_item(self.pid, self.is_lion)
if types_caches.look_for_key(self.pid):
self.types_cache = types_caches.get_value(self.pid)
else:
self.types_cache = lldb.formatters.attrib_fromdict.AttributesDictionary(
allow_reset=0
)
self.types_cache.addr_type = valobj.GetType().GetBasicType(
lldb.eBasicTypeUnsignedLong
)
self.types_cache.addr_ptr_type = self.types_cache.addr_type.GetPointerType()
self.types_cache.uint32_t = valobj.GetType().GetBasicType(
lldb.eBasicTypeUnsignedInt
)
types_caches.add_item(self.pid, self.types_cache)
if isa_caches.look_for_key(self.pid):
self.isa_cache = isa_caches.get_value(self.pid)
else:
self.isa_cache = lldb.formatters.cache.Cache()
isa_caches.add_item(self.pid, self.isa_cache)
def adjust_for_architecture(self, valobj):
process = valobj.GetTarget().GetProcess()
self.pointer_size = process.GetAddressByteSize()
self.is_64_bit = self.pointer_size == 8
self.endianness = process.GetByteOrder()
self.is_little = self.endianness == lldb.eByteOrderLittle
self.cfruntime_size = 16 if self.is_64_bit else 8
def calculate_offset(self, num_pointers=0, bytes_count=0, padding32=0, padding64=0):
value = bytes_count + num_pointers * self.pointer_size
return value + padding64 if self.is_64_bit else value + padding32
class ObjCRuntime:
@staticmethod
def runtime_version(process):
logger = lldb.formatters.Logger.Logger()
if process.IsValid() == 0:
logger >> "No process - bailing out"
return None
target = process.GetTarget()
num_modules = target.GetNumModules()
module_objc = None
for idx in range(num_modules):
module = target.GetModuleAtIndex(idx)
if module.GetFileSpec().GetFilename() == "libobjc.A.dylib":
module_objc = module
break
if module_objc is None or module_objc.IsValid() == 0:
logger >> "no libobjc - bailing out"
return None
num_sections = module.GetNumSections()
section_objc = None
for idx in range(num_sections):
section = module.GetSectionAtIndex(idx)
if section.GetName() == "__OBJC":
section_objc = section
break
if section_objc is not None and section_objc.IsValid():
logger >> "found __OBJC: v1"
return 1
logger >> "no __OBJC: v2"
return 2
@staticmethod
def runtime_from_isa(isa):
logger = lldb.formatters.Logger.Logger()
runtime = ObjCRuntime(isa)
runtime.isa = isa
return runtime
def __init__(self, valobj):
logger = lldb.formatters.Logger.Logger()
self.valobj = valobj
self.adjust_for_architecture()
self.sys_params = SystemParameters(self.valobj)
self.unsigned_value = self.valobj.GetValueAsUnsigned()
self.isa_value = None
def adjust_for_architecture(self):
pass
def is_tagged(self):
logger = lldb.formatters.Logger.Logger()
if self.valobj is None:
return 0
return Utilities.is_valid_pointer(
self.unsigned_value, self.sys_params.pointer_size, allow_tagged=1
) and not (
Utilities.is_valid_pointer(
self.unsigned_value, self.sys_params.pointer_size, allow_tagged=0
)
)
def is_valid(self):
logger = lldb.formatters.Logger.Logger()
if self.valobj is None:
return 0
if self.valobj.IsInScope() == 0:
return 0
return Utilities.is_valid_pointer(
self.unsigned_value, self.sys_params.pointer_size, allow_tagged=1
)
def is_nil(self):
return self.unsigned_value == 0
def read_isa(self):
logger = lldb.formatters.Logger.Logger()
if self.isa_value is not None:
logger >> "using cached isa"
return self.isa_value
self.isa_pointer = self.valobj.CreateChildAtOffset(
"cfisa", 0, self.sys_params.types_cache.addr_ptr_type
)
if self.isa_pointer is None or self.isa_pointer.IsValid() == 0:
logger >> "invalid isa - bailing out"
return None
self.isa_value = self.isa_pointer.GetValueAsUnsigned(1)
if self.isa_value == 1:
logger >> "invalid isa value - bailing out"
return None
return Ellipsis
def read_class_data(self):
logger = lldb.formatters.Logger.Logger()
global isa_cache
if self.is_tagged():
if self.sys_params.runtime_version == 2:
logger >> "on v2 and tagged - maybe"
tentative_tagged = TaggedClass_Data(
self.unsigned_value, self.sys_params
)
if tentative_tagged.is_valid():
logger >> "truly tagged"
return tentative_tagged
else:
logger >> "not tagged - error"
return InvalidClass_Data()
else:
logger >> "on v1 and tagged - error"
return InvalidClass_Data()
if self.is_valid() == 0 or self.read_isa() is None:
return InvalidClass_Data()
data = self.sys_params.isa_cache.get_value(self.isa_value, default=None)
if data is not None:
return data
if self.sys_params.runtime_version == 2:
data = Class_Data_V2(self.isa_pointer, self.sys_params)
else:
data = Class_Data_V1(self.isa_pointer, self.sys_params)
if data is None:
return InvalidClass_Data()
if data.is_valid():
self.sys_params.isa_cache.add_item(self.isa_value, data, ok_to_replace=1)
return data
class SpecialSituation_Description:
def message(self):
return ""
class InvalidPointer_Description(SpecialSituation_Description):
def __init__(self, nil):
self.is_nil = nil
def message(self):
if self.is_nil:
return '@"<nil>"'
else:
return "<invalid pointer>"
class InvalidISA_Description(SpecialSituation_Description):
def __init__(self):
pass
def message(self):
return "<not an Objective-C object>"
class ThisIsZombie_Description(SpecialSituation_Description):
def message(self):
return "<freed object>"