Add 2GB array stress test

[ctf-testsuite.git] / tests / 1.8 / stress / stream / pass / array-2GB / test.py

#!/usr/bin/env python3

# Expects either "prepare" or "clean" argument

import sys
import os

output_metadata = './metadata'
output_stream = './stream'

def arg_err():
	print('Please specify "prepare" or "clean" as first argument.')
	sys.exit(-1)

if (len(sys.argv) <= 1):
	arg_err()
mode = sys.argv[1]

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 = 32; 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 = seq;\n' \
'	fields := struct {\n' \
'		uint8_t field[2147483648]; /* 2GB array */\n' \
'	};\n' \
'};\n'

stream_packet_header = [
  0xC1, 0x1F, 0xFC, 0xC1,	# magic
  0x2A, 0x64, 0x22, 0xD0, 0x6C, 0xEE, 0x11, 0xE0,
  0x8C, 0x08, 0xCB, 0x07, 0xD7, 0xB3, 0xA5, 0x64, # uuid
]

array_byte = [ 0x00, ]

def write_binary(f, arr):
	f.write(bytes(arr))

def generate_metadata():
	# Generate metadata
	metadata_f = open(output_metadata, 'w')
	metadata_f.write(metadata_str)
	metadata_f.close()

def generate_stream():
	# Generate stream
	stream_f = open(output_stream, 'wb')
	write_binary(stream_f, stream_packet_header)
	# generate 2GB array as event content
	os.ftruncate(stream_f.fileno(), 2147483648)
	stream_f.close()

def test_prepare():
	print('Preparing test')
	generate_metadata()
	generate_stream()

def test_clean():
	print('Cleaning up test')
	try:
		os.remove(output_metadata)
	except:
		pass
	try:
		os.remove(output_stream)
	except:
		pass

if (mode == "prepare"):
	test_prepare()
elif (mode == "clean"):
	test_clean()
else:
	arg_err()