From 1b761a9e8b665994c5820d85abc2cb332660c4f7 Mon Sep 17 00:00:00 2001
From: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
Date: Mon, 17 Nov 2014 09:15:00 +0100
Subject: [PATCH] stress test: many traces

Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
---
 .../stream/pass/many-traces/param-list.txt    |  16 +++
 .../stress/stream/pass/many-traces/test.py    | 121 ++++++++++++++++++
 2 files changed, 137 insertions(+)
 create mode 100644 tests/1.8/stress/stream/pass/many-traces/param-list.txt
 create mode 100755 tests/1.8/stress/stream/pass/many-traces/test.py

diff --git a/tests/1.8/stress/stream/pass/many-traces/param-list.txt b/tests/1.8/stress/stream/pass/many-traces/param-list.txt
new file mode 100644
index 0000000..500823d
--- /dev/null
+++ b/tests/1.8/stress/stream/pass/many-traces/param-list.txt
@@ -0,0 +1,16 @@
+16
+32
+64
+128
+256
+512
+1024
+2048
+4096
+8192
+16384
+32768
+65536
+131072
+262144
+524288
diff --git a/tests/1.8/stress/stream/pass/many-traces/test.py b/tests/1.8/stress/stream/pass/many-traces/test.py
new file mode 100755
index 0000000..cb414cf
--- /dev/null
+++ b/tests/1.8/stress/stream/pass/many-traces/test.py
@@ -0,0 +1,121 @@
+#!/usr/bin/env python3
+
+# Expects either "prepare" or "clean" argument
+
+import sys
+import os
+import struct
+
+def arg_err():
+	print('Please specify "prepare <SIZE>" or "clean <SIZE>" as first argument.')
+	sys.exit(-1)
+
+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) + '/'
+
+def write_binary(f, arr):
+	f.write(bytes(arr))
+
+def generate_metadata(trace_nr):
+	# Generate metadata
+	metadata_str = \
+	('/* CTF 1.8 */\n'
+	'\n'
+	'typealias integer {{ size = 8; align = 8; signed = false; base = hex; }} := uint8_t;\n'
+	'typealias integer {{ size = 32; align = 8; signed = false; base = hex; }} := uint32_t;\n'
+	'typealias integer {{ size = 64; align = 8; signed = false; base = hex; }} := uint64_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'
+	'clock {{\n'
+	'	name = monotonic;\n'
+	'	uuid = "e016a9b9-1058-40d5-9074-6ce5e2bb59c6";\n'
+	'	description = "Monotonic Clock";\n'
+	'	freq = 1000000000;\n'
+	'	offset = 1415075600471492540;\n'
+	'}};\n'
+	'\n'
+	'event {{\n'
+	'	name = myevent;\n'
+	'	fields := struct {{\n'
+	'		uint8_t f;\n'
+	'	}};\n'
+	'}};\n').format()
+
+	metadata_f = open(tracedir_name + '/' + str(trace_nr) + '/metadata', 'w')
+	metadata_f.write(metadata_str)
+	metadata_f.close()
+
+def generate_stream(trace_nr):
+	# Generate stream
+	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
+	]
+	event_payload = [
+	  0x42,
+	]
+	stream_f = open(tracedir_name + '/' + str(trace_nr) + '/stream', 'wb')
+	write_binary(stream_f, stream_packet_header)
+	# generate 8-bit per event
+	write_binary(stream_f, event_payload)
+	stream_f.close()
+
+def generate_traces(array_len):
+	for i in range(array_len):
+		os.mkdir(tracedir_name + '/' + str(i))
+		generate_metadata(i)
+		generate_stream(i)
+
+def test_prepare():
+	print('Preparing test for ' + str(array_len) + ' traces')
+	os.mkdir(tracedir_name)
+	generate_traces(array_len)
+
+def test_clean():
+	print('Cleaning up test for ' + str(array_len) + ' traces')
+	try:
+		os.remove(output_metadata)
+	except:
+		pass
+	for i in range(array_len):
+		try:
+			os.remove(tracedir_name + '/' + str(i) + '/metadata')
+		except:
+			pass
+		try:
+			os.remove(tracedir_name + '/' + str(i) + '/stream')
+		except:
+			pass
+		try:
+			os.rmdir(tracedir_name + '/' + str(i))
+		except:
+			pass
+	try:
+		os.rmdir(tracedir_name)
+	except:
+		pass
+
+if (mode == "prepare"):
+	test_clean()
+	test_prepare()
+elif (mode == "clean"):
+	test_clean()
+else:
+	arg_err()
-- 
2.34.1