aboutsummaryrefslogtreecommitdiff
path: root/examples/summaries/cocoa/NSData.py
blob: 3aa30b29f54bab31b5f892de0edb82dd153e4a2f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
"""
LLDB AppKit formatters

part of The LLVM Compiler Infrastructure
This file is distributed under the University of Illinois Open Source
License. See LICENSE.TXT for details.
"""
# example summary provider for NSData
# the real summary is now C++ code built into LLDB
import lldb
import ctypes
import lldb.runtime.objc.objc_runtime
import lldb.formatters.metrics
import lldb.formatters.Logger

statistics = lldb.formatters.metrics.Metrics()
statistics.add_metric('invalid_isa')
statistics.add_metric('invalid_pointer')
statistics.add_metric('unknown_class')
statistics.add_metric('code_notrun')

# despite the similary to synthetic children providers, these classes are not
# trying to provide anything but the length for an NSData, so they need not
# obey the interface specification for synthetic children providers
class NSConcreteData_SummaryProvider:
	def adjust_for_architecture(self):
		pass

	def __init__(self, valobj, params):
		logger = lldb.formatters.Logger.Logger()
		logger >> "NSConcreteData_SummaryProvider __init__"
		self.valobj = valobj;
		self.sys_params = params
		if not(self.sys_params.types_cache.NSUInteger):
			if self.sys_params.is_64_bit:
				self.sys_params.types_cache.NSUInteger = self.valobj.GetType().GetBasicType(lldb.eBasicTypeUnsignedLong)
			else:
				self.sys_params.types_cache.NSUInteger = self.valobj.GetType().GetBasicType(lldb.eBasicTypeUnsignedInt)
		self.update();

	def update(self):
		self.adjust_for_architecture();

	# one pointer is the ISA
	# then there are 32 bit worth of flags and other data
	# however, on 64bit systems these are padded to be a full
	# machine word long, which means we actually have two pointers
	# worth of data to skip
	def offset(self):
		return 2 * self.sys_params.pointer_size

	def length(self):
		logger = lldb.formatters.Logger.Logger()
		logger >> "NSConcreteData_SummaryProvider length"
		size = self.valobj.CreateChildAtOffset("count",
							self.offset(),
							self.sys_params.types_cache.NSUInteger)
		logger >> str(size)
		logger >> str(size.GetValueAsUnsigned(0))
		return size.GetValueAsUnsigned(0)


class NSDataUnknown_SummaryProvider:
	def adjust_for_architecture(self):
		pass

	def __init__(self, valobj, params):
		logger = lldb.formatters.Logger.Logger()
		logger >> "NSDataUnknown_SummaryProvider __init__"
		self.valobj = valobj;
		self.sys_params = params
		self.update();

	def update(self):
		self.adjust_for_architecture();

	def length(self):
		logger = lldb.formatters.Logger.Logger()
		logger >> "NSDataUnknown_SummaryProvider length"
		stream = lldb.SBStream()
		self.valobj.GetExpressionPath(stream)
		logger >> stream.GetData()
		num_children_vo = self.valobj.CreateValueFromExpression("count","(int)[" + stream.GetData() + " length]");
		logger >> "still in after expression: " + str(num_children_vo)
		if num_children_vo.IsValid():
			logger >> "wow - expr output is valid: " + str(num_children_vo.GetValueAsUnsigned())
			return num_children_vo.GetValueAsUnsigned(0)
		logger >> "invalid expr output - too bad"
		return '<variable is not NSData>'


def GetSummary_Impl(valobj):
	global statistics
	logger = lldb.formatters.Logger.Logger()
	logger >> "NSData GetSummary_Impl"
	class_data,wrapper =lldb.runtime.objc.objc_runtime.Utilities.prepare_class_detection(valobj,statistics)
	if wrapper:
		logger >> "got a wrapper summary - using it"
		return wrapper
	
	name_string = class_data.class_name()
	logger >> "class name: " + name_string
	if name_string == 'NSConcreteData' or \
	   name_string == 'NSConcreteMutableData' or \
	   name_string == '__NSCFData':
		wrapper = NSConcreteData_SummaryProvider(valobj, class_data.sys_params)
		statistics.metric_hit('code_notrun',valobj)
	else:
		wrapper = NSDataUnknown_SummaryProvider(valobj, class_data.sys_params)
		statistics.metric_hit('unknown_class',valobj.GetName() + " seen as " + name_string)
	return wrapper;

def NSData_SummaryProvider (valobj,dict):
	logger = lldb.formatters.Logger.Logger()
	logger >> "NSData_SummaryProvider"
	provider = GetSummary_Impl(valobj);
	logger >> "found a summary provider, it is: " + str(provider)
	if provider != None:
		try:
			summary = provider.length();
		except:
			summary = None
		logger >> "got a summary: it is " + str(summary)
		if summary == None:
			summary = '<variable is not NSData>'
		elif isinstance(summary,basestring):
			pass
		else:
			if summary == 1:
				summary = '1 byte'
			else:
				summary = str(summary) + ' bytes'
		return summary
	return 'Summary Unavailable'

def NSData_SummaryProvider2 (valobj,dict):
	logger = lldb.formatters.Logger.Logger()
	logger >> "NSData_SummaryProvider2"
	provider = GetSummary_Impl(valobj);
	logger >> "found a summary provider, it is: " + str(provider)
	if provider != None:
		if isinstance(provider,lldb.runtime.objc.objc_runtime.SpecialSituation_Description):
			return provider.message()
		try:
			summary = provider.length();
		except:
			summary = None
		logger >> "got a summary: it is " + str(summary)
		if summary == None:
			summary = '<variable is not CFData>'
		elif isinstance(summary,basestring):
			pass
		else:
			if summary == 1:
				summary = '@"1 byte"'
			else:
				summary = '@"' + str(summary) + ' bytes"'
		return summary
	return 'Summary Unavailable'

def __lldb_init_module(debugger,dict):
	debugger.HandleCommand("type summary add -F NSData.NSData_SummaryProvider NSData")
	debugger.HandleCommand("type summary add -F NSData.NSData_SummaryProvider2 CFDataRef CFMutableDataRef")