paraspace: paraspace/dexfile.py comparison

comparison paraspace/dexfile.py @ 124:8e42b2816893

Fixing compute_size() and sizeof() for DEX types. - Prevent compute_size() and sizeof() of depend_* to include size of depend-on.

author	Thinker K.F. Li <thinker@codemud.net>
date	Sun, 07 Aug 2011 22:07:08 +0800
parents	78357afb4a9d
children	b9688a3badaa

comparison

equal deleted inserted replaced

-:78357afb4a9d
+:8e42b2816893
 return len(_to_leb128(v))
 def _compute_sz(o, _type):
 if hasattr(o, 'compute_size'):
-o.compute_size()
+_type.compute_size(o)
-return o.data_size
+pass
 return _type.sizeof(o)
 class _dex_type(object):
 pass
 class uint32(_dex_type):
 @staticmethod
 def parse(parent, data, off):
 v = _to_uint(data[off:off + 4])
 return v
+@staticmethod
+def compute_size(v):
+pass
 @staticmethod
 def sizeof(v):
 return 4
 def parse(parent, data, off):
 v = _to_uint(data[off:off + 2])
 return v
 @staticmethod
+def compute_size(v):
+pass
+@staticmethod
 def sizeof(v):
 return 2
 @staticmethod
 def to_str(v):
 def parse(parent, data, off):
 v = _to_uint(data[off:off + 1])
 return v
 @staticmethod
+def compute_size(v):
+pass
+@staticmethod
 def sizeof(v):
 return 1
 @staticmethod
 def to_str(v):
 def parse(parent, data, off):
 v = _to_int(data[off:off + 4])
 return v
 @staticmethod
+def compute_size(v):
+pass
+@staticmethod
 def sizeof(v):
 return 4
 @staticmethod
 def to_str(v):
 def parse(parent, data, off):
 v = _to_int(data[off:off + 2])
 return v
 @staticmethod
+def compute_size(v):
+pass
+@staticmethod
 def sizeof(v):
 return 2
 @staticmethod
 def to_str(v):
 def parse(parent, data, off):
 v, sh = _uleb128(data[off:off + 5])
 return v
 @staticmethod
+def compute_size(v):
+pass
+@staticmethod
 def sizeof(v):
 return _uleb128_sz(v)
 @staticmethod
 def to_str(v):
 @staticmethod
 def parse(parent, data, off):
 v, sh = _leb128(data[off:off + 5])
 return v
+@staticmethod
+def compute_size(v):
+pass
 @staticmethod
 def sizeof(v):
 return _leb128_sz(v)
 @staticmethod
 return padding_sz
 def parse(self, parent, data, off):
 return self.recompute_align(off)
+@staticmethod
+def compute_size(v):
+pass
 @staticmethod
 def sizeof(v):
 return v
 @staticmethod
 return 0
 def to_str(self):
 return ''
+@staticmethod
 def compute_size(self):
 pass
 def children(self):
 return []
 @staticmethod
 def sizeof(v):
 return 0
 @staticmethod
-def compute_size():
+def compute_size(self):
 pass
 @staticmethod
 def to_str(child):
 return ''
 obj.items = items
 obj.data_size = moff() - off
 return obj
+@staticmethod
 def compute_size(self):
 sizes = [_compute_sz(item, self.child_type)
 for item in self.items]
 size = sum(sizes)
 self.data_size = size
 def sizeof(self, v):
 if v.value is None:
 return 0
 return self.child_type.sizeof(v.value)
+@staticmethod
 def compute_size(self):
 if self.is_true:
 self.data_size = _compute_sz(self.value, self.child_type)
 else:
 self.data_size = 0
 @staticmethod
 def sizeof(v):
 return v.child_type.sizeof(v.value)
+@staticmethod
 def compute_size(self):
 self.data_size = _compute_sz(self.value, self.child_type)
 pass
 @staticmethod
 def children(self):
 return ()
 pass
+## \brief Make a dependency to a depend-on for back type.
+#
+# Depend-on is the object that the back type is supposed to point to.
+# Back type of a depend must be not a composite type while depend-on
+# must be.
+#
 class depend(null_relocatable):
 depend_on = None
 def __init__(self, depend_on):
 self.depend_on = depend_on
 pass
 def __call__(self, back_type):
+assert type(back_type) != type or not issubclass(back_type, composite)
 self.back_type = back_type
 return self
 def parse(self, parent, data, off):
 v = self.back_type.parse(parent, data, off)
 setattr(parent, name, obj)
 pass
 class depend_off(depend):
+def compute_size(self, child):
+pass
+def sizeof(self, child):
+if isinstance(child, composite):
+return self.back_type.sizeof(child.data_offset)
+return self.back_type.sizeof(child)
 pass
 class depend_off_rel(depend):
 relative_to = None
+_depon2_log = {}
 def __init__(self, relative_to, depend_on):
 super(depend_off_rel, self).__init__(depend_on)
 self.relative_to = relative_to
 pass
+def parse(self, parent, data, off):
+v = super(depend_off_rel, self).parse(parent, data, off)
+return v
+def compute_size(self, child):
+pass
+def sizeof(self, child):
+if isinstance(child, composite):
+pivot = self._depon2_log[child] # depon2
+off_diff = child.data_offset - pivot.data_offset
+return self.back_type.sizeof(off_diff)
+return self.back_type.sizeof(child)
 pass
 class depend_idx(depend):
 def sizeof(self, v):
 isinstance(v, do_child_clazz.__class__):
 v = v.data_idx
 pass
 v = self.back_type.sizeof(v)
 return v
-pass
+def compute_size(self, child):
+pass
+def sizeof(self, child):
+if isinstance(child, composite):
+return self.back_type.sizeof(child.data_idx)
+return self.back_type.sizeof(child)
 pass
 class _DEX_header(composite):
 magic = rawstr(8)
 child_names = 'size elements'.split()
 pass
-class _DEX_DebugCodeBlock(relocatable):
+class _DEX_DebugCodeBlock(_dex_type):
 DBG_END_SEQUENCE = 0x00
 DBG_ADVANCE_PC = 0x01
 DBG_ADVANCE_LINE = 0x02
 DBG_START_LOCAL = 0x03
 DBG_START_LOCAL_EXTENDED = 0x04
 opcode_sizes = [i for i in opcode_sizes]
 opcodes_size = sum(opcode_sizes)
 self.data_size = opcodes_size
 pass
+@staticmethod
+def sizeof(obj):
+return obj.data_size
 @staticmethod
 def to_str(self):
 #
 # Parse debug opcodes

Mercurial > paraspace

comparison paraspace/dexfile.py @ 124:8e42b2816893