tests/1.8/stress/stream/pass/variant-many-tags/test.py

   1 #!/usr/bin/env python3
   2
   3 # Expects either "prepare" or "clean" argument
   4
   5 import sys
   6 import os
   7
   8 def arg_err():
   9         print('Please specify "prepare <SIZE>" or "clean <SIZE>" as first argument.')
  10         sys.exit(-1)
  11
  12 if (len(sys.argv) <= 1):
  13         arg_err()
  14 mode = sys.argv[1]
  15 if (len(sys.argv) <= 2):
  16         arg_err()
  17 array_len = int(sys.argv[2])
  18
  19 tracedir_name = './trace-' + str(array_len) + '/'
  20 output_metadata = tracedir_name + 'metadata'
  21 output_stream = tracedir_name + 'stream'
  22
  23
  24 def write_binary(f, arr):
  25         f.write(bytes(arr))
  26
  27 def generate_metadata(array_len):
  28         # Generate metadata
  29         metadata_str1 = \
  30         ( '/* CTF 1.8 */\n'
  31         '\n'
  32         'typealias integer {{ size = 8; align = 8; signed = false; base = 10; }} := uint8_t;\n'
  33         'typealias integer {{ size = 32; align = 8; signed = false; base = hex; }} := uint32_t;\n'
  34         'typealias integer {{ size = 64; align = 8; signed = false; base = hex; }} := uint64_t;\n'
  35         '\n'
  36         'trace {{\n'
  37         '       major = 0;\n'
  38         '       minor = 0;\n'
  39         '       uuid = "2a6422d0-6cee-11e0-8c08-cb07d7b3a564";\n'
  40         '       byte_order = le;\n'
  41         '       packet.header := struct {{\n'
  42         '               uint32_t magic;\n'
  43         '               uint8_t uuid[16];\n'
  44         '       }};\n'
  45         '}};\n'
  46         '\n'
  47         'event {{\n'
  48         '       name = myevent;\n'
  49         '       fields := struct {{\n'
  50         '               enum : uint64_t {{\n').format()
  51
  52         metadata_enum = \
  53         '                       t{},\n'
  54
  55         metadata_str2 = \
  56         ( '             }} select;\n'
  57         '               variant <select> {{\n').format()
  58
  59         metadata_variant = \
  60         '                       uint8_t t{};\n'
  61
  62         metadata_str3 = \
  63         ( '             }} myvar;\n'
  64         '       }};\n'
  65         '}};\n').format()
  66         metadata_f = open(output_metadata, 'w')
  67         metadata_f.write(metadata_str1)
  68         for i in range(array_len):
  69                 metadata_f.write(metadata_enum.format(i))
  70         metadata_f.write(metadata_str2)
  71         for i in range(array_len):
  72                 metadata_f.write(metadata_variant.format(i))
  73         metadata_f.write(metadata_str3)
  74         metadata_f.close()
  75
  76 def generate_stream(array_len):
  77         # Generate stream
  78         stream_packet_header = [
  79           0xC1, 0x1F, 0xFC, 0xC1,       # magic
  80           0x2A, 0x64, 0x22, 0xD0, 0x6C, 0xEE, 0x11, 0xE0,
  81           0x8C, 0x08, 0xCB, 0x07, 0xD7, 0xB3, 0xA5, 0x64, # uuid
  82         ]
  83         stream_f = open(output_stream, 'wb')
  84         write_binary(stream_f, stream_packet_header)
  85         # generate 8 bytes (enum) + 1 byte (variant) as content
  86         os.ftruncate(stream_f.fileno(), len(stream_packet_header) + 8 + 1)
  87         stream_f.close()
  88
  89 def test_prepare():
  90         print('Preparing test for many tags variant length ' + str(array_len) + ' tags')
  91         os.mkdir(tracedir_name)
  92         generate_metadata(array_len)
  93         generate_stream(array_len)
  94
  95 def test_clean():
  96         print('Cleaning up test for many tags variant length ' + str(array_len) + ' tags')
  97         try:
  98                 os.remove(output_metadata)
  99         except:
 100                 pass
 101         try:
 102                 os.remove(output_stream)
 103         except:
 104                 pass
 105         try:
 106                 os.rmdir(tracedir_name)
 107         except:
 108                 pass
 109
 110 if (mode == "prepare"):
 111         test_clean()
 112         test_prepare()
 113 elif (mode == "clean"):
 114         test_clean()
 115 else:
 116         arg_err()