Put the ->real_file_size handling into fio

[fio.git] / engines / cpu.c

diff --git a/engines/cpu.c b/engines/cpu.c
index f65f91d9e0f988e17e7d426bdde81f778a7bf14b..14ad9717aa5fbd024a7e9f286a36fd79542c6365 100644 (file)
--- a/engines/cpu.c
+++ b/engines/cpu.c
@@ -1,30 +1,54 @@
+/*
+ * CPU engine
+ *
+ * Doesn't transfer any data, merely burns CPU cycles according to
+ * the settings.
+ *
+ */
 #include "../fio.h"
-#include "../os.h"
 
-static int fio_cpuio_setup(struct thread_data fio_unused *td)
+static int fio_cpuio_queue(struct thread_data *td, struct io_u fio_unused *io_u)
 {
-       return 0;
+       __usec_sleep(td->o.cpucycle);
+       return FIO_Q_COMPLETED;
 }
 
 static int fio_cpuio_init(struct thread_data *td)
 {
-       if (!td->cpuload) {
-               td_vmsg(td, EINVAL, "cpu thread needs rate");
+       struct thread_options *o = &td->o;
+
+       if (!o->cpuload) {
+               td_vmsg(td, EINVAL, "cpu thread needs rate (cpuload=)","cpuio");
                return 1;
-       } else if (td->cpuload > 100)
-               td->cpuload = 100;
+       }
 
-       td->nr_files = 0;
+       if (o->cpuload > 100)
+               o->cpuload = 100;
 
+       /*
+        * set thinktime_sleep and thinktime_spin appropriately
+        */
+       o->thinktime_blocks = 1;
+       o->thinktime_spin = 0;
+       o->thinktime = (o->cpucycle * (100 - o->cpuload)) / o->cpuload;
+
+       o->nr_files = o->open_files = 1;
+       return 0;
+}
+
+static int fio_cpuio_open(struct thread_data fio_unused *td,
+                         struct fio_file fio_unused *f)
+{
        return 0;
 }
 
 static struct ioengine_ops ioengine = {
        .name           = "cpuio",
        .version        = FIO_IOOPS_VERSION,
+       .queue          = fio_cpuio_queue,
        .init           = fio_cpuio_init,
-       .setup          = fio_cpuio_setup,
-       .flags          = FIO_CPUIO,
+       .open_file      = fio_cpuio_open,
+       .flags          = FIO_SYNCIO | FIO_DISKLESSIO,
 };
 
 static void fio_init fio_cpuio_register(void)