work a bit on vmmap reader and add a demo

Author: fijal

vmprof/reader.py

@@ -76,16 +76,28 @@ def read_ranges(data):
         data = data.decode('latin1')
     ranges = []
     lines = data.splitlines()
-    if lines[0].endswith('PATH'):
+    if 'Virtual Memory Map' in lines[0]:
+        mode = 'vmmap'
+        end = 3
+        while lines[end]:
+            end += 1
+        lines = lines[4:end]
+    elif lines[0].endswith('PATH'):
         lines = lines[1:]
-        procstat = True
+        mode = 'procstat'
     else:
-        procstat = False                                               
+        mode = 'proc'                           
     for line in lines:
         parts = re.split("\s+", line)
         name = parts[-1]
-        if procstat:
+        if mode == 'procstat':
             start, end = parts[1], parts[2]
+        elif mode == 'vmmap':
+            k = 1
+            while not parts[k].startswith('000'):
+                k += 1
+                continue
+            start, end = parts[k].split('-')
         else:
             start, end = parts[0].split('-')
         start = int(start, 16)

vmprofdemo.py

@@ -0,0 +1,95 @@
+
+import random, sys
+
+class Node(object):
+	def __init__(self, right, left):
+		self.left = left
+		self.right = right
+
+class Digit(object):
+	def __init__(self, v):
+		self.v = v
+
+	def eval(self):
+		return ord(self.v) - ord('0')
+
+class Plus(Node):
+	def eval(self):
+		return self.left.eval() + self.right.eval()
+
+class Minus(Node):
+	def eval(self):
+		return self.left.eval() - self.right.eval()
+
+def parse_pn(text):
+	stack = []
+	for c in text:
+		if c == '+':
+			stack.append(Plus(stack.pop(), stack.pop()))
+		elif c == '-':
+			stack.append(Minus(stack.pop(), stack.pop()))
+		elif ord('0') <= ord(c) <= ord('9'):
+			stack.append(Digit(c))
+		else:
+			assert False
+	assert len(stack) == 1
+	return stack[0]
+
+def oracle(expr):
+	return parse_pn(expr).eval() > 0
+
+def find(expr):
+	cur_expr = expr
+	if oracle(expr):
+		cur = 0
+		while oracle(cur_expr):
+			cur += 1
+			cur_expr = cur_expr + "1-"
+	else:
+		cur = 0
+		while not oracle(cur_expr):
+			cur += 1
+			cur_expr = cur_expr + "1+"
+		cur = -cur + 1
+	return cur
+
+DIGITS = [chr(ord('0') + i) for i in range(10)]
+
+def gen_exp(lgt):
+	stack_depth = 0
+	exp = ''
+	for i in range(lgt):
+		if stack_depth > 1:
+			c = random.choice(DIGITS + ['-', '+'] * 4)
+		else:
+			c = random.choice(DIGITS)
+		if c in DIGITS:
+			stack_depth += 1
+		else:
+			stack_depth -= 1
+		exp += c
+	while stack_depth > 1:
+		exp += random.choice(['-', '+'])
+		stack_depth -= 1
+	return exp
+
+def fuzzer(count):
+	for i in range(count):
+		exp = gen_exp(10)
+		assert parse_pn(exp).eval() == find(exp)
+
+if __name__ == '__main__':
+	if len(sys.argv) == 2 and sys.argv[1] == 'demo':
+		import time
+		for k in range(30):
+			t0 = time.time()
+			fuzzer(100)
+			print "%.1f ms" % ((time.time() - t0) * 1000)
+		sys.exit(0)
+	if len(sys.argv) > 1:
+		count = int(sys.argv[1])
+	else:
+		count = 10000
+	if len(sys.argv) > 2:
+		random.seed(int(sys.argv[2]))
+	fuzzer(count)