From 7b80aa6aaf2cb2af3d8796eb01db5f53afaed278 Mon Sep 17 00:00:00 2001
From: Rodrigo Arias <rodrigo.arias@bsc.es>
Date: Mon, 20 Feb 2023 18:44:54 +0100
Subject: [PATCH] Instrument running thread in sort and flush tests

---
 test/emu/ovni/empty-sort.c | 35 ++++---------------------------
 test/emu/ovni/flush.c      | 42 ++++++--------------------------------
 2 files changed, 10 insertions(+), 67 deletions(-)

diff --git a/test/emu/ovni/empty-sort.c b/test/emu/ovni/empty-sort.c
index 29c67c0..8015f74 100644
--- a/test/emu/ovni/empty-sort.c
+++ b/test/emu/ovni/empty-sort.c
@@ -1,35 +1,9 @@
-/* Copyright (c) 2021-2022 Barcelona Supercomputing Center (BSC)
+/* Copyright (c) 2021-2023 Barcelona Supercomputing Center (BSC)
  * SPDX-License-Identifier: GPL-3.0-or-later */
 
-#define _POSIX_C_SOURCE 200112L
 #define _GNU_SOURCE
 
-#include <limits.h>
-#include <linux/limits.h>
-#include <stddef.h>
-#include <stdint.h>
-#include <stdio.h>
-#include <sys/syscall.h>
-#include <sys/types.h>
-#include <unistd.h>
-
-#include "compat.h"
-#include "ovni.h"
-
-static inline void
-init(void)
-{
-	char hostname[HOST_NAME_MAX];
-
-	if (gethostname(hostname, HOST_NAME_MAX) != 0) {
-		perror("gethostname failed");
-		exit(EXIT_FAILURE);
-	}
-
-	ovni_proc_init(0, hostname, getpid());
-	ovni_thread_init(gettid());
-	ovni_add_cpu(0, 0);
-}
+#include "instr_ovni.h"
 
 static void
 emit(char *mcv, int64_t clock)
@@ -43,7 +17,7 @@ emit(char *mcv, int64_t clock)
 int
 main(void)
 {
-	init();
+	instr_start(0, 1);
 
 	/* Leave some room to prevent clashes */
 	usleep(100); /* 100000 us */
@@ -60,8 +34,7 @@ main(void)
 
 	emit("OU]", ovni_clock_now());
 
-	ovni_flush();
-	ovni_proc_fini();
+	instr_end();
 
 	return 0;
 }
diff --git a/test/emu/ovni/flush.c b/test/emu/ovni/flush.c
index 140bf73..583ca58 100644
--- a/test/emu/ovni/flush.c
+++ b/test/emu/ovni/flush.c
@@ -1,35 +1,9 @@
-/* Copyright (c) 2021-2022 Barcelona Supercomputing Center (BSC)
+/* Copyright (c) 2021-2023 Barcelona Supercomputing Center (BSC)
  * SPDX-License-Identifier: GPL-3.0-or-later */
 
-#define _POSIX_C_SOURCE 200112L
 #define _GNU_SOURCE
 
-#include <limits.h>
-#include <linux/limits.h>
-#include <stddef.h>
-#include <stdint.h>
-#include <stdio.h>
-#include <sys/syscall.h>
-#include <sys/types.h>
-#include <unistd.h>
-
-#include "compat.h"
-#include "ovni.h"
-
-static inline void
-init(void)
-{
-	char hostname[HOST_NAME_MAX];
-
-	if (gethostname(hostname, HOST_NAME_MAX) != 0) {
-		perror("gethostname failed");
-		exit(EXIT_FAILURE);
-	}
-
-	ovni_proc_init(0, hostname, getpid());
-	ovni_thread_init(gettid());
-	ovni_add_cpu(0, 0);
-}
+#include "instr_ovni.h"
 
 static void
 emit(uint8_t *buf, size_t size)
@@ -43,13 +17,10 @@ emit(uint8_t *buf, size_t size)
 int
 main(void)
 {
-	size_t payload_size;
-	uint8_t *payload_buf;
+	instr_start(0, 1);
 
-	init();
-
-	payload_size = (size_t) (0.9 * (double) OVNI_MAX_EV_BUF);
-	payload_buf = calloc(1, payload_size);
+	size_t payload_size = (size_t) (0.9 * (double) OVNI_MAX_EV_BUF);
+	uint8_t *payload_buf = calloc(1, payload_size);
 
 	if (!payload_buf) {
 		perror("calloc failed");
@@ -67,8 +38,7 @@ main(void)
 	emit(payload_buf, payload_size);
 
 	/* Flush the last event to disk manually */
-	ovni_flush();
-	ovni_proc_fini();
+	instr_end();
 
 	free(payload_buf);