X-Git-Url: https://git.kernel.dk/?a=blobdiff_plain;f=profiles%2Ftiobench.c;h=f86a33787d9ca4e5f25fe2d096089c04b902154d;hb=b6cf38f0aed8b12db1c8fd520c09e5a74066ebc9;hp=d239468f9f3272269089e08590c8ac69d52c4ddc;hpb=79d16311c8d0c7188d73df77838fb1b4b6ff58db;p=fio.git

diff --git a/profiles/tiobench.c b/profiles/tiobench.c
index d239468f..f86a3378 100644
--- a/profiles/tiobench.c
+++ b/profiles/tiobench.c
@@ -1,9 +1,18 @@
 #include "../fio.h"
 #include "../profile.h"
+#include "../parse.h"
+
+static unsigned long long size;
+static unsigned int loops = 1;
+static unsigned int bs = 4096;
+static unsigned int nthreads = 1;
+static char *dir;
+
+char sz_idx[80], bs_idx[80], loop_idx[80], dir_idx[80], t_idx[80];
 
 static const char *tb_opts[] = {
-	"buffered=0", "size=4*1024*$mb_memory", "bs=4k", "timeout=600",
-	"numjobs=4", "group_reporting", "thread", "overwrite=1",
+	"buffered=0", sz_idx, bs_idx, loop_idx, dir_idx, t_idx,
+	"timeout=600", "group_reporting", "thread", "overwrite=1",
 	"filename=.fio.tio.1:.fio.tio.2:.fio.tio.3:.fio.tio.4",
 	"name=seqwrite", "rw=write", "end_fsync=1",
 	"name=randwrite", "stonewall", "rw=randwrite", "end_fsync=1",
@@ -11,15 +20,81 @@ static const char *tb_opts[] = {
 	"name=randread", "stonewall", "rw=randread", NULL,
 };
 
+static struct fio_option options[] = {
+	{
+		.name	= "size",
+		.type	= FIO_OPT_STR_VAL,
+		.roff1	= &size,
+		.help	= "Size in MB",
+	},
+	{
+		.name	= "block",
+		.type	= FIO_OPT_INT,
+		.roff1	= &bs,
+		.help	= "Block size in bytes",
+		.def	= "4k",
+	},
+	{
+		.name	= "numruns",
+		.type	= FIO_OPT_INT,
+		.roff1	= &loops,
+		.help	= "Number of runs",
+	},
+	{
+		.name	= "dir",
+		.type	= FIO_OPT_STR_STORE,
+		.roff1	= &dir,
+		.help	= "Test directory",
+	},
+	{
+		.name	= "threads",
+		.type	= FIO_OPT_INT,
+		.roff1	= &nthreads,
+		.help	= "Number of Threads",
+	},
+	{
+		.name	= NULL,
+	},
+};
+
+/*
+ * Fill our private options into the command line
+ */
+static void tb_prep_cmdline(void)
+{
+
+	/*
+	 * tiobench uses size as MB, so multiply up
+	 */
+	size *= 1024 * 1024ULL;
+	if (size)
+		sprintf(sz_idx, "size=%llu", size);
+	else
+		strcpy(sz_idx, "size=4*1024*$mb_memory");
+
+	sprintf(bs_idx, "bs=%u", bs);
+	sprintf(loop_idx, "loops=%u", loops);
+
+	if (dir)
+		sprintf(dir_idx, "directory=%s", dir);
+	else
+		sprintf(dir_idx, "directory=./");
+
+	sprintf(t_idx, "numjobs=%u", nthreads);
+}
+
 static struct profile_ops tiobench_profile = {
 	.name		= "tiobench",
-	.version	= FIO_PROFILE_VERSION,
-	.def_ops	= tb_opts,
+	.desc		= "tiotest/tiobench benchmark",
+	.options	= options,
+	.prep_cmd	= tb_prep_cmdline,
+	.cmdline	= tb_opts,
 };
 
 static void fio_init tiobench_register(void)
 {
-	register_profile(&tiobench_profile);
+	if (register_profile(&tiobench_profile))
+		log_err("fio: failed to register profile 'tiobench'\n");
 }
 
 static void fio_exit tiobench_unregister(void)