X-Git-Url: https://git.kernel.dk/?a=blobdiff_plain;f=profile.c;h=6a80dec2c7b7479cff92d24c47d2a0d9ac8fa691;hb=ba9c7219eefcc04e84a4fb4970f4b5caeaf48aa6;hp=0e2b97d4a0933896532b4891bd650dcf63c16da0;hpb=15dc1934435dc84d66547c4fc92d936224d7238f;p=fio.git

diff --git a/profile.c b/profile.c
index 0e2b97d4..6a80dec2 100644
--- a/profile.c
+++ b/profile.c
@@ -31,7 +31,7 @@ int load_profile(const char *profile)
 	ops = find_profile(profile);
 	if (ops) {
 		ops->prep_cmd();
-		add_job_opts(ops->cmdline);
+		add_job_opts(ops->cmdline, FIO_CLIENT_TYPE_CLI);
 		return 0;
 	}
 
@@ -42,7 +42,7 @@ int load_profile(const char *profile)
 static int add_profile_options(struct profile_ops *ops)
 {
 	struct fio_option *o;
-	
+
 	if (!ops->options)
 		return 0;
 
@@ -93,6 +93,26 @@ void profile_add_hooks(struct thread_data *td)
 	if (!ops)
 		return;
 
-	td->fill_io_u_off = ops->fill_io_u_off;
-	td->fill_io_u_size = ops->fill_io_u_size;
+	if (ops->io_ops) {
+		td->prof_io_ops = *ops->io_ops;
+		td->flags |= TD_F_PROFILE_OPS;
+	}
+}
+
+int profile_td_init(struct thread_data *td)
+{
+	struct prof_io_ops *ops = &td->prof_io_ops;
+
+	if (ops->td_init)
+		return ops->td_init(td);
+
+	return 0;
+}
+
+void profile_td_exit(struct thread_data *td)
+{
+	struct prof_io_ops *ops = &td->prof_io_ops;
+
+	if (ops->td_exit)
+		ops->td_exit(td);
 }