1 files changed, 61 insertions, 0 deletions
diff --git a/tests/simple_mem/traffic_gen.c b/tests/simple_mem/traffic_gen.c
new file mode 100644
index 0000000..4d14d12
--- /dev/null
+++ b/tests/simple_mem/traffic_gen.c
@@ -0,0 +1,61 @@
+#include <stdlib.h>
+#include <assert.h>
+
+#include <gran/component.h>
+
+#include "traffic_gen.h"
+
+struct traffic_gen {
+	struct component component;
+	struct component *stress;
+	uintptr_t addr;
+	uintptr_t end;
+
+	size_t counter;
+};
+
+static stat traffic_gen_clock(struct traffic_gen *tg)
+{
+	if (tg->addr == tg->end)
+		return DONE;
+
+	uint8_t c = 0;
+	switch (tg->counter) {
+	case 0:
+		assert(write_u8(tg->stress, tg->addr, 13) == OK);
+		tg->counter = 1;
+		break;
+
+	case 1:
+		c = 0;
+		assert(read_u8(tg->stress, tg->addr, &c) == OK);
+		assert(c == 13);
+		tg->counter = 0;
+		tg->addr++;
+		break;
+	}
+
+	return OK;
+}
+
+void traffic_gen_destroy(struct traffic_gen *tg)
+{
+	destroy(tg->stress);
+	free(tg);
+}
+
+struct component *create_traffic_gen(struct component *stress, uintptr_t start, size_t size)
+{
+	struct traffic_gen *new = calloc(1, sizeof(struct traffic_gen));
+	if (!new)
+		return NULL;
+
+	new->stress = stress;
+	new->addr = start;
+	new->end = start + size;
+	new->counter = 0;
+
+	new->component.clock = (clock_callback)traffic_gen_clock;
+	new->component.destroy = (destroy_callback)traffic_gen_destroy;
+	return (struct component *)new;
+}