def test_trace(self):
"""
This checks the sequence of pushes and pops that take place during an example execution.
That sequence isn't meant to be invariant, so if you change the implementation and break
this test, by all means, replace it with the new sequence.
:return:
"""
def arg_gen(_state):
return [I32(1337)]
m = ManticoreWASM(wasm_file)
tracker_plugin = StackTrackerPlugin()
m.register_plugin(tracker_plugin)
m.invoke("collatz", arg_gen)
m.run()
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [44])
push_pop_seq = "".join(tracker_plugin.context.get("push_pop_seq"))
self.assertEqual(push_pop_seq, "+-" * 892)
def test_getchar(self):
m = ManticoreWASM(if_check_file, env={"getchar": getchar2})
m.main()
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [-1, 0])
def test_getchar(self):
m = ManticoreWASM(wasm_file, env={"getchar": getchar})
m.invoke("main")
m.run()
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [0, 1, 2, 5, 7, 8, 16])
def test_implicit_call(self):
m = ManticoreWASM(collatz_file)
counter_plugin = CallCounterPlugin()
m.register_plugin(counter_plugin)
m.collatz(lambda s: [I32(1337)])
counts = counter_plugin.context.get("counter")
self.assertEqual(counts["br_if"], 45)
self.assertEqual(counts["loop"], 44)
self.assertEqual(counts["i32.add"], 88)
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [44])
m.collatz(lambda s: [I32(1338)])
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [70])
def test_symbolic_args(self):
m = ManticoreWASM(wasm_file, env={})
m.invoke("collatz", arg_gen)
m.run()
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [2, 3, 8])
m.finalize()
inputs = []
for fn in glob.glob(m.workspace + "/*.input"):
with open(fn, "r") as f:
raw = f.read().strip()
inputs.append(int(raw.replace("collatz_arg: ", "")))
self.assertEqual(sorted(inputs), [4, 6, 8])
def test_plugin(self):
def arg_gen(_state):
return [I32(1337)]
m = ManticoreWASM(wasm_file)
counter_plugin = CallCounterPlugin()
m.register_plugin(counter_plugin)
m.invoke("collatz", arg_gen)
m.run()
# counts = m.context.get("<class 'test_examples.CallCounterPlugin'>").get("counter")
counts = counter_plugin.context.get("counter")
self.assertEqual(counts["br_if"], 45)
self.assertEqual(counts["loop"], 44)
self.assertEqual(counts["i32.add"], 88)
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [44])
def test_resume(self):
m = ManticoreWASM(collatz_file)
plugin = CallCounterPlugin()
m.register_plugin(plugin)
m.collatz(lambda s: [I32(1337)])
m.run()
counts_canonical = plugin.context.get("counter")
m = ManticoreWASM(collatz_file)
plugin = SerializerPlugin()
m.register_plugin(plugin)
m.collatz(lambda s: [I32(1337)])
m.run()
counts_save = plugin.context.get("counter")
m = ManticoreWASM.from_saved_state("/tmp/collatz_checkpoint.pkl")
plugin = CallCounterPlugin()
m.register_plugin(plugin)
m.run()
counts_resume = plugin.context.get("counter")
for k in counts_canonical:
with self.subTest(k):
self.assertEqual(
counts_save.get(k, 0) + counts_resume.get(k, 0),
counts_canonical[k],
f"Mismatched {k} count",
)
results = []
for idx, val_list in enumerate(m.collect_returns()):
results.append(val_list[0][0])
self.assertEqual(sorted(results), [44])
from manticore.wasm import ManticoreWASM
from manticore.wasm.structure import MemInst
# from manticore.core.plugin import Plugin
def args(state):
arg1 = state.new_symbolic_value(32, "arg1")
arg2 = state.new_symbolic_value(32, "arg2")
state.constrain(arg1 == 256)
state.constrain(arg2 == 256)
return [arg1, arg2]
mem = MemInst([0x0] * 1024 * 1024)
m = ManticoreWASM("tweetNacl.wasm", sup_env={"js": {"memory": mem}})
# m.register_plugin(PrintRetPlugin())
m.crypto_verify_16(args)
for idx, val_list in enumerate(m.collect_returns()):
print("State", idx, "::", val_list[0])
m.finalize()
