[PATCH] Implement file syncing as data direction
[fio.git] / engines / fio-engine-sg.c
index 0db5ca9..112f027 100644 (file)
@@ -112,8 +112,8 @@ static int fio_sgio_getevents(struct thread_data *td, int min, int max,
        return r;
 }
 
-static int fio_sgio_ioctl_doio(struct thread_data *td, struct fio_file *f,
-                              struct io_u *io_u)
+static int fio_sgio_ioctl_doio(struct thread_data *td,
+                              struct fio_file *f, struct io_u *io_u)
 {
        struct sgio_data *sd = td->io_ops->data;
        struct sg_io_hdr *hdr = &io_u->hdr;
@@ -123,8 +123,7 @@ static int fio_sgio_ioctl_doio(struct thread_data *td, struct fio_file *f,
        return ioctl(f->fd, SG_IO, hdr);
 }
 
-static int fio_sgio_rw_doio(struct thread_data *td, struct fio_file *f,
-                           struct io_u *io_u, int sync)
+static int fio_sgio_rw_doio(struct fio_file *f, struct io_u *io_u, int sync)
 {
        struct sg_io_hdr *hdr = &io_u->hdr;
        int ret;
@@ -149,29 +148,7 @@ static int fio_sgio_doio(struct thread_data *td, struct io_u *io_u, int sync)
        if (td->filetype == FIO_TYPE_BD)
                return fio_sgio_ioctl_doio(td, f, io_u);
 
-       return fio_sgio_rw_doio(td, f, io_u, sync);
-}
-
-static int fio_sgio_sync(struct thread_data *td, struct fio_file *f)
-{
-       struct sgio_data *sd = td->io_ops->data;
-       struct sg_io_hdr *hdr;
-       struct io_u *io_u;
-       int ret;
-
-       io_u = __get_io_u(td);
-       if (!io_u)
-               return ENOMEM;
-
-       hdr = &io_u->hdr;
-       sgio_hdr_init(sd, hdr, io_u, 0);
-       hdr->dxfer_direction = SG_DXFER_NONE;
-
-       hdr->cmdp[0] = 0x35;
-
-       ret = fio_sgio_doio(td, io_u, 1);
-       put_io_u(td, io_u);
-       return ret;
+       return fio_sgio_rw_doio(f, io_u, sync);
 }
 
 static int fio_sgio_prep(struct thread_data *td, struct io_u *io_u)
@@ -185,24 +162,34 @@ static int fio_sgio_prep(struct thread_data *td, struct io_u *io_u)
                return EINVAL;
        }
 
-       sgio_hdr_init(sd, hdr, io_u, 1);
-
        if (io_u->ddir == DDIR_READ) {
+               sgio_hdr_init(sd, hdr, io_u, 1);
+
                hdr->dxfer_direction = SG_DXFER_FROM_DEV;
                hdr->cmdp[0] = 0x28;
-       } else {
+       } else if (io_u->ddir == DDIR_WRITE) {
+               sgio_hdr_init(sd, hdr, io_u, 1);
+
                hdr->dxfer_direction = SG_DXFER_TO_DEV;
                hdr->cmdp[0] = 0x2a;
+       } else {
+               sgio_hdr_init(sd, hdr, io_u, 0);
+
+               hdr->dxfer_direction = SG_DXFER_NONE;
+               hdr->cmdp[0] = 0x35;
+       }
+
+       if (hdr->dxfer_direction != SG_DXFER_NONE) {
+               nr_blocks = io_u->buflen / sd->bs;
+               lba = io_u->offset / sd->bs;
+               hdr->cmdp[2] = (lba >> 24) & 0xff;
+               hdr->cmdp[3] = (lba >> 16) & 0xff;
+               hdr->cmdp[4] = (lba >>  8) & 0xff;
+               hdr->cmdp[5] = lba & 0xff;
+               hdr->cmdp[7] = (nr_blocks >> 8) & 0xff;
+               hdr->cmdp[8] = nr_blocks & 0xff;
        }
 
-       nr_blocks = io_u->buflen / sd->bs;
-       lba = io_u->offset / sd->bs;
-       hdr->cmdp[2] = (lba >> 24) & 0xff;
-       hdr->cmdp[3] = (lba >> 16) & 0xff;
-       hdr->cmdp[4] = (lba >>  8) & 0xff;
-       hdr->cmdp[5] = lba & 0xff;
-       hdr->cmdp[7] = (nr_blocks >> 8) & 0xff;
-       hdr->cmdp[8] = nr_blocks & 0xff;
        return 0;
 }
 
@@ -211,7 +198,7 @@ static int fio_sgio_queue(struct thread_data *td, struct io_u *io_u)
        struct sg_io_hdr *hdr = &io_u->hdr;
        int ret;
 
-       ret = fio_sgio_doio(td, io_u, 0);
+       ret = fio_sgio_doio(td, io_u, io_u->ddir == DDIR_SYNC);
 
        if (ret < 0)
                io_u->error = errno;
@@ -325,6 +312,5 @@ struct ioengine_ops ioengine = {
        .getevents      = fio_sgio_getevents,
        .event          = fio_sgio_event,
        .cleanup        = fio_sgio_cleanup,
-       .sync           = fio_sgio_sync,
-       .flags          = FIO_SYNCIO,
+       .flags          = FIO_SYNCIO | FIO_RAWIO,
 };