From 59eaf62931aa763234e5a2082202e7ab26280629 Mon Sep 17 00:00:00 2001 From: Ben Kurtovic Date: Fri, 20 Apr 2012 14:13:57 -0300 Subject: [PATCH] Adding stuff! --- func_smash.py | 78 +++++++++++++++++++++++++++++------------------------------ 1 file changed, 38 insertions(+), 40 deletions(-) diff --git a/func_smash.py b/func_smash.py index 71d6a8c..d085766 100644 --- a/func_smash.py +++ b/func_smash.py @@ -1,12 +1,47 @@ +import imp import opcode +import os import random +import re +import sys +import traceback import types -############## CODE ########################################################### - MARKOV_START = -1 MARKOV_END = -2 +def run(): + try: + path, name = os.path.split(sys.argv[1]) + name = re.sub("\.pyc?$", "", name) + except IndexError: + raise RuntimeError("Needs a filename as a command-line argument") + f, path, desc = imp.find_module(name, [path]) + try: + module = imp.load_module(name, f, path, desc) + finally: + f.close() + + corpus = module.corpus + chain = make_chain(corpus) + func = make_function(chain, "func") + print "Using {0}-function corpus.".format(len(corpus)) + print "Smashed function disassembly:" + print_function(func) + print + n = 12.0 + print "func(%s) =" % n, func(n) + + if len(sys.argv) > 2 and sys.argv[2] == "-i": # Allow input after the fact + print + while 1: + try: + input(">>> ") + except EOFError: + break + except Exception: + traceback.print_exc() + def make_chain(funcs): chain = {} for func in funcs: @@ -137,42 +172,5 @@ def _int_to_opname(op): return "END" return opcode.opname[op] -############## FUNCTION CORPUS ################################################ - -def f1(a): - b = a + 9.0 - return b - -def f2(a): - b = a - 7.0 - return b - -def f3(a): - b = a * 5.0 - return b - -def f4(a): - b = a / 3.0 - return b - -def f5(a): - b = a ** 2.0 - return b - -def f6(a): - b = a % 10.0 - return b - -corpus = [f1, f2, f3, f4, f5, f6] - -############## DEMO ########################################################### - if __name__ == "__main__": - chain = make_chain(corpus) - func = make_function(chain, "func") - print "Using {0}-function corpus.".format(len(corpus)) - print "Smashed function disassembly:" - print_function(func) - print - n = 12.0 - print "func(%s) =" % n, func(n) + run()