diff options
Diffstat (limited to 'suite/generator.py')
| -rwxr-xr-x | suite/generator.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/suite/generator.py b/suite/generator.py index 443bb6bd..2fe7e8e0 100755 --- a/suite/generator.py +++ b/suite/generator.py @@ -290,6 +290,15 @@ def generate(test_store, outdir, exclude_binaries): binary_oracle.close() print("{0:.2f}".format(time.time() - binary_start_time)) + # Generate oracle data for rebasing tests + name = oraclefile_rel[len(test_store):].replace(os.path.sep, "_").replace(".", "_")[1:] + if name in ["helloworld", "duff", "partial_register_dataflow", "raw"]: + test_data = testcommon.BinaryViewTestBuilder(oraclefile_rel, imageBase=0xf00000) + binary_oracle = OracleTestFile(oraclefile + "_rebasing") + for method in test_data.methods(): + binary_oracle.add_entry(test_data, method) + binary_oracle.close() + if not os.path.exists(oraclefile + ".zip"): with zipfile.ZipFile(oraclefile + ".zip", "w") as zf: zf.write(oraclefile, os.path.relpath(oraclefile, start=os.path.dirname(__file__))) |