#!/usr/bin/env python3
+from ctftestsuite.stress import MetadataTestAssitant
-# Expects either "prepare" or "clean" argument
-import sys
-import os
-import struct
+class TestAssistant(MetadataTestAssitant):
+ what = '{size} callsites'
-def arg_err():
- print('Please specify "prepare <SIZE>" or "clean <SIZE>" as first argument.')
- sys.exit(-1)
+ def write_metadata(self, f):
+ p1 = \
+'''event {
+ name = myevent;
+};
-if (len(sys.argv) <= 1):
- arg_err()
-mode = sys.argv[1]
-if (len(sys.argv) <= 2):
- arg_err()
-array_len = int(sys.argv[2])
+'''
-tracedir_name = './trace-' + str(array_len) + '/'
-output_metadata = tracedir_name + 'metadata'
-output_stream = tracedir_name + 'stream'
+ callsite_fmt = \
+'''callsite {{
+ name = "myevent";
+ func = "myfunc{i}";
+ file = "myfile{i}";
+ line = {i};
+ ip = 0x{i};
+}};
+'''
-def write_binary(f, arr):
- f.write(bytes(arr))
+ f.write(self.BASIC_PROLOGUE)
+ f.write(p1)
-def generate_metadata(array_len):
- # Generate metadata
- metadata_str = \
- ( '/* CTF 1.8 */\n'
- '\n'
- 'typealias integer {{ size = 8; align = 8; signed = false; base = 10; }} := uint8_t;\n'
- 'typealias integer {{ size = 32; align = 8; signed = false; base = hex; }} := uint32_t;\n'
- '\n'
- 'trace {{\n'
- ' major = 0;\n'
- ' minor = 0;\n'
- ' uuid = "2a6422d0-6cee-11e0-8c08-cb07d7b3a564";\n'
- ' byte_order = le;\n'
- ' packet.header := struct {{\n'
- ' uint32_t magic;\n'
- ' uint8_t uuid[16];\n'
- ' }};\n'
- '}};\n'
- '\n'
- 'event {{\n'
- ' name = myevent;\n'
- '}};\n'
- '\n').format()
+ for i in range(self.size):
+ callsite = callsite_fmt.format(i=i)
+ f.write(callsite)
- metadata_callsite = \
- ('callsite {{\n'
- ' name = "myevent";\n'
- ' func = "myfunc{}";\n'
- ' file = "myfile{}";\n'
- ' line = {};\n'
- ' ip = 0x{};\n'
- '}};\n')
- metadata_f = open(output_metadata, 'w')
- metadata_f.write(metadata_str)
- for i in range(array_len):
- metadata_f.write(metadata_callsite.format(i, i, i, i))
- metadata_f.close()
-
-def test_prepare():
- print('Preparing test for ' + str(array_len) + ' callsites')
- os.mkdir(tracedir_name)
- generate_metadata(array_len)
-
-def test_clean():
- print('Cleaning up test for ' + str(array_len) + ' callsites')
- try:
- os.remove(output_metadata)
- except:
- pass
- try:
- os.rmdir(tracedir_name)
- except:
- pass
-
-if (mode == "prepare"):
- test_clean()
- test_prepare()
-elif (mode == "clean"):
- test_clean()
-else:
- arg_err()
+if __name__ == '__main__':
+ test_assistant = TestAssistant()
+ test_assistant.main()