Fix typo (READ/DDIR_READ).
[fio.git] / io_u.c
diff --git a/io_u.c b/io_u.c
index b049b61853fdfbd305ad01d857046be88dc68671..3de04ef0eb91fe25d6561562f931ebfa7fb311f7 100644 (file)
--- a/io_u.c
+++ b/io_u.c
@@ -10,6 +10,7 @@
 #include "verify.h"
 #include "trim.h"
 #include "lib/rand.h"
+#include "lib/axmap.h"
 
 struct io_completion_data {
        int nr;                         /* input */
@@ -20,17 +21,12 @@ struct io_completion_data {
 };
 
 /*
- * The ->file_map[] contains a map of blocks we have or have not done io
+ * The ->io_axmap contains a map of blocks we have or have not done io
  * to yet. Used to make sure we cover the entire range in a fair fashion.
  */
-static int random_map_free(struct fio_file *f, const unsigned long long block)
+static int random_map_free(struct fio_file *f, const uint64_t block)
 {
-       unsigned int idx = RAND_MAP_IDX(f, block);
-       unsigned int bit = RAND_MAP_BIT(f, block);
-
-       dprint(FD_RANDOM, "free: b=%llu, idx=%u, bit=%u\n", block, idx, bit);
-
-       return (f->file_map[idx] & (1UL << bit)) == 0;
+       return !axmap_isset(f->io_axmap, block);
 }
 
 /*
@@ -40,69 +36,24 @@ static void mark_random_map(struct thread_data *td, struct io_u *io_u)
 {
        unsigned int min_bs = td->o.rw_min_bs;
        struct fio_file *f = io_u->file;
-       unsigned long long block;
-       unsigned int blocks, nr_blocks;
-       int busy_check;
+       unsigned int nr_blocks;
+       uint64_t block;
 
-       block = (io_u->offset - f->file_offset) / (unsigned long long) min_bs;
+       block = (io_u->offset - f->file_offset) / (uint64_t) min_bs;
        nr_blocks = (io_u->buflen + min_bs - 1) / min_bs;
-       blocks = 0;
-       busy_check = !(io_u->flags & IO_U_F_BUSY_OK);
-
-       while (nr_blocks) {
-               unsigned int idx, bit;
-               unsigned long mask, this_blocks;
-
-               /*
-                * If we have a mixed random workload, we may
-                * encounter blocks we already did IO to.
-                */
-               if (!busy_check) {
-                       blocks = nr_blocks;
-                       break;
-               }
-               if ((td->o.ddir_seq_nr == 1) && !random_map_free(f, block))
-                       break;
-
-               idx = RAND_MAP_IDX(f, block);
-               bit = RAND_MAP_BIT(f, block);
-
-               fio_assert(td, idx < f->num_maps);
-
-               this_blocks = nr_blocks;
-               if (this_blocks + bit > BLOCKS_PER_MAP)
-                       this_blocks = BLOCKS_PER_MAP - bit;
-
-               do {
-                       if (this_blocks == BLOCKS_PER_MAP)
-                               mask = -1UL;
-                       else
-                               mask = ((1UL << this_blocks) - 1) << bit;
-       
-                       if (!(f->file_map[idx] & mask))
-                               break;
-
-                       this_blocks--;
-               } while (this_blocks);
 
-               if (!this_blocks)
-                       break;
+       if (!(io_u->flags & IO_U_F_BUSY_OK))
+               nr_blocks = axmap_set_nr(f->io_axmap, block, nr_blocks);
 
-               f->file_map[idx] |= mask;
-               nr_blocks -= this_blocks;
-               blocks += this_blocks;
-               block += this_blocks;
-       }
-
-       if ((blocks * min_bs) < io_u->buflen)
-               io_u->buflen = blocks * min_bs;
+       if ((nr_blocks * min_bs) < io_u->buflen)
+               io_u->buflen = nr_blocks * min_bs;
 }
 
-static unsigned long long last_block(struct thread_data *td, struct fio_file *f,
-                                    enum fio_ddir ddir)
+static uint64_t last_block(struct thread_data *td, struct fio_file *f,
+                          enum fio_ddir ddir)
 {
-       unsigned long long max_blocks;
-       unsigned long long max_size;
+       uint64_t max_blocks;
+       uint64_t max_size;
 
        assert(ddir_rw(ddir));
 
@@ -116,126 +67,157 @@ static unsigned long long last_block(struct thread_data *td, struct fio_file *f,
        if (td->o.zone_range)
                max_size = td->o.zone_range;
 
-       max_blocks = max_size / (unsigned long long) td->o.ba[ddir];
+       max_blocks = max_size / (uint64_t) td->o.ba[ddir];
        if (!max_blocks)
                return 0;
 
        return max_blocks;
 }
 
-/*
- * Return the next free block in the map.
- */
-static int get_next_free_block(struct thread_data *td, struct fio_file *f,
-                              enum fio_ddir ddir, unsigned long long *b)
+struct rand_off {
+       struct flist_head list;
+       uint64_t off;
+};
+
+static int __get_next_rand_offset(struct thread_data *td, struct fio_file *f,
+                                 enum fio_ddir ddir, uint64_t *b)
 {
-       unsigned long long block, min_bs = td->o.rw_min_bs, lastb;
-       int i;
+       uint64_t r, lastb;
 
        lastb = last_block(td, f, ddir);
        if (!lastb)
                return 1;
 
-       i = f->last_free_lookup;
-       block = i * BLOCKS_PER_MAP;
-       while (block * min_bs < f->real_file_size &&
-               block * min_bs < f->io_size) {
-               if (f->file_map[i] != -1UL) {
-                       block += ffz(f->file_map[i]);
-                       if (block > lastb)
-                               break;
-                       f->last_free_lookup = i;
-                       *b = block;
-                       return 0;
-               }
+       if (td->o.random_generator == FIO_RAND_GEN_TAUSWORTHE) {
+               uint64_t rmax;
 
-               block += BLOCKS_PER_MAP;
-               i++;
-       }
+               rmax = td->o.use_os_rand ? OS_RAND_MAX : FRAND_MAX;
 
-       dprint(FD_IO, "failed finding a free block\n");
-       return 1;
-}
+               if (td->o.use_os_rand) {
+                       rmax = OS_RAND_MAX;
+                       r = os_random_long(&td->random_state);
+               } else {
+                       rmax = FRAND_MAX;
+                       r = __rand(&td->__random_state);
+               }
 
-static int get_next_rand_offset(struct thread_data *td, struct fio_file *f,
-                               enum fio_ddir ddir, unsigned long long *b)
-{
-       unsigned long long rmax, r, lastb;
-       int loops = 5;
+               dprint(FD_RANDOM, "off rand %llu\n", r);
 
-       lastb = last_block(td, f, ddir);
-       if (!lastb)
-               return 1;
+               *b = (lastb - 1) * (r / ((uint64_t) rmax + 1.0));
+       } else {
+               uint64_t off = 0;
 
-       if (f->failed_rands >= 200)
-               goto ffz;
+               if (lfsr_next(&f->lfsr, &off, lastb))
+                       return 1;
 
-       rmax = td->o.use_os_rand ? OS_RAND_MAX : FRAND_MAX;
-       do {
-               if (td->o.use_os_rand)
-                       r = os_random_long(&td->random_state);
-               else
-                       r = __rand(&td->__random_state);
+               *b = off;
+       }
 
-               *b = (lastb - 1) * (r / ((unsigned long long) rmax + 1.0));
+       /*
+        * if we are not maintaining a random map, we are done.
+        */
+       if (!file_randommap(td, f))
+               goto ret;
 
-               dprint(FD_RANDOM, "off rand %llu\n", r);
+       /*
+        * calculate map offset and check if it's free
+        */
+       if (random_map_free(f, *b))
+               goto ret;
 
+       dprint(FD_RANDOM, "get_next_rand_offset: offset %llu busy\n", *b);
 
-               /*
-                * if we are not maintaining a random map, we are done.
-                */
-               if (!file_randommap(td, f))
-                       goto ret_good;
+       *b = axmap_next_free(f->io_axmap, *b);
+       if (*b == (uint64_t) -1ULL)
+               return 1;
+ret:
+       return 0;
+}
 
-               /*
-                * calculate map offset and check if it's free
-                */
-               if (random_map_free(f, *b))
-                       goto ret_good;
+static int __get_next_rand_offset_zipf(struct thread_data *td,
+                                      struct fio_file *f, enum fio_ddir ddir,
+                                      uint64_t *b)
+{
+       *b = zipf_next(&f->zipf);
+       return 0;
+}
 
-               dprint(FD_RANDOM, "get_next_rand_offset: offset %llu busy\n",
-                                                                       *b);
-       } while (--loops);
+static int __get_next_rand_offset_pareto(struct thread_data *td,
+                                        struct fio_file *f, enum fio_ddir ddir,
+                                        uint64_t *b)
+{
+       *b = pareto_next(&f->zipf);
+       return 0;
+}
 
-       if (!f->failed_rands++)
-               f->last_free_lookup = 0;
+static int flist_cmp(void *data, struct flist_head *a, struct flist_head *b)
+{
+       struct rand_off *r1 = flist_entry(a, struct rand_off, list);
+       struct rand_off *r2 = flist_entry(b, struct rand_off, list);
 
-       /*
-        * we get here, if we didn't suceed in looking up a block. generate
-        * a random start offset into the filemap, and find the first free
-        * block from there.
-        */
-       loops = 10;
-       do {
-               f->last_free_lookup = (f->num_maps - 1) *
-                                       (r / ((unsigned long long) rmax + 1.0));
-               if (!get_next_free_block(td, f, ddir, b))
-                       goto ret;
+       return r1->off - r2->off;
+}
 
-               if (td->o.use_os_rand)
-                       r = os_random_long(&td->random_state);
-               else
-                       r = __rand(&td->__random_state);
-       } while (--loops);
+static int get_off_from_method(struct thread_data *td, struct fio_file *f,
+                              enum fio_ddir ddir, uint64_t *b)
+{
+       if (td->o.random_distribution == FIO_RAND_DIST_RANDOM)
+               return __get_next_rand_offset(td, f, ddir, b);
+       else if (td->o.random_distribution == FIO_RAND_DIST_ZIPF)
+               return __get_next_rand_offset_zipf(td, f, ddir, b);
+       else if (td->o.random_distribution == FIO_RAND_DIST_PARETO)
+               return __get_next_rand_offset_pareto(td, f, ddir, b);
+
+       log_err("fio: unknown random distribution: %d\n", td->o.random_distribution);
+       return 1;
+}
+
+static int get_next_rand_offset(struct thread_data *td, struct fio_file *f,
+                               enum fio_ddir ddir, uint64_t *b)
+{
+       struct rand_off *r;
+       int i, ret = 1;
 
        /*
-        * that didn't work either, try exhaustive search from the start
+        * If sort not enabled, or not a pure random read workload without
+        * any stored write metadata, just return a random offset
         */
-       f->last_free_lookup = 0;
-ffz:
-       if (!get_next_free_block(td, f, ddir, b))
+       if (!td->o.verifysort_nr || !(ddir == DDIR_READ && td->o.do_verify &&
+           td->o.verify != VERIFY_NONE && td_random(td)))
+               return get_off_from_method(td, f, ddir, b);
+
+       if (!flist_empty(&td->next_rand_list)) {
+               struct rand_off *r;
+fetch:
+               r = flist_entry(td->next_rand_list.next, struct rand_off, list);
+               flist_del(&r->list);
+               *b = r->off;
+               free(r);
                return 0;
-       f->last_free_lookup = 0;
-       return get_next_free_block(td, f, ddir, b);
-ret_good:
-       f->failed_rands = 0;
-ret:
-       return 0;
+       }
+
+       for (i = 0; i < td->o.verifysort_nr; i++) {
+               r = malloc(sizeof(*r));
+
+               ret = get_off_from_method(td, f, ddir, &r->off);
+               if (ret) {
+                       free(r);
+                       break;
+               }
+
+               flist_add(&r->list, &td->next_rand_list);
+       }
+
+       if (ret && !i)
+               return ret;
+
+       assert(!flist_empty(&td->next_rand_list));
+       flist_sort(NULL, &td->next_rand_list, flist_cmp);
+       goto fetch;
 }
 
 static int get_next_rand_block(struct thread_data *td, struct fio_file *f,
-                              enum fio_ddir ddir, unsigned long long *b)
+                              enum fio_ddir ddir, uint64_t *b)
 {
        if (!get_next_rand_offset(td, f, ddir, b))
                return 0;
@@ -252,7 +234,7 @@ static int get_next_rand_block(struct thread_data *td, struct fio_file *f,
 }
 
 static int get_next_seq_offset(struct thread_data *td, struct fio_file *f,
-                              enum fio_ddir ddir, unsigned long long *offset)
+                              enum fio_ddir ddir, uint64_t *offset)
 {
        assert(ddir_rw(ddir));
 
@@ -260,7 +242,7 @@ static int get_next_seq_offset(struct thread_data *td, struct fio_file *f,
                f->last_pos = f->last_pos - f->io_size;
 
        if (f->last_pos < f->real_file_size) {
-               unsigned long long pos;
+               uint64_t pos;
 
                if (f->last_pos == f->file_offset && td->o.ddir_seq_add < 0)
                        f->last_pos = f->real_file_size;
@@ -280,7 +262,7 @@ static int get_next_block(struct thread_data *td, struct io_u *io_u,
                          enum fio_ddir ddir, int rw_seq)
 {
        struct fio_file *f = io_u->file;
-       unsigned long long b, offset;
+       uint64_t b, offset;
        int ret;
 
        assert(ddir_rw(ddir));
@@ -310,14 +292,15 @@ static int get_next_block(struct thread_data *td, struct io_u *io_u,
                        ret = 1;
                }
        }
-       
+
        if (!ret) {
                if (offset != -1ULL)
                        io_u->offset = offset;
                else if (b != -1ULL)
                        io_u->offset = b * td->o.ba[ddir];
                else {
-                       log_err("fio: bug in offset generation\n");
+                       log_err("fio: bug in offset generation: offset=%llu, b=%llu\n",
+                                                               offset, b);
                        ret = 1;
                }
        }
@@ -364,10 +347,12 @@ static int __get_next_offset(struct thread_data *td, struct io_u *io_u)
 
 static int get_next_offset(struct thread_data *td, struct io_u *io_u)
 {
-       struct prof_io_ops *ops = &td->prof_io_ops;
+       if (td->flags & TD_F_PROFILE_OPS) {
+               struct prof_io_ops *ops = &td->prof_io_ops;
 
-       if (ops->fill_io_u_off)
-               return ops->fill_io_u_off(td, io_u);
+               if (ops->fill_io_u_off)
+                       return ops->fill_io_u_off(td, io_u);
+       }
 
        return __get_next_offset(td, io_u);
 }
@@ -383,7 +368,7 @@ static inline int io_u_fits(struct thread_data *td, struct io_u *io_u,
 static unsigned int __get_next_buflen(struct thread_data *td, struct io_u *io_u)
 {
        const int ddir = io_u->ddir;
-       unsigned int uninitialized_var(buflen);
+       unsigned int buflen = 0;
        unsigned int minbs, maxbs;
        unsigned long r, rand_max;
 
@@ -442,10 +427,12 @@ static unsigned int __get_next_buflen(struct thread_data *td, struct io_u *io_u)
 
 static unsigned int get_next_buflen(struct thread_data *td, struct io_u *io_u)
 {
-       struct prof_io_ops *ops = &td->prof_io_ops;
+       if (td->flags & TD_F_PROFILE_OPS) {
+               struct prof_io_ops *ops = &td->prof_io_ops;
 
-       if (ops->fill_io_u_size)
-               return ops->fill_io_u_size(td, io_u);
+               if (ops->fill_io_u_size)
+                       return ops->fill_io_u_size(td, io_u);
+       }
 
        return __get_next_buflen(td, io_u);
 }
@@ -1034,10 +1021,12 @@ out:
 
 static struct fio_file *get_next_file(struct thread_data *td)
 {
-       struct prof_io_ops *ops = &td->prof_io_ops;
+       if (!(td->flags & TD_F_PROFILE_OPS)) {
+               struct prof_io_ops *ops = &td->prof_io_ops;
 
-       if (ops->get_next_file)
-               return ops->get_next_file(td);
+               if (ops->get_next_file)
+                       return ops->get_next_file(td);
+       }
 
        return __get_next_file(td);
 }
@@ -1114,7 +1103,10 @@ again:
 
 static int check_get_trim(struct thread_data *td, struct io_u *io_u)
 {
-       if (td->o.trim_backlog && td->trim_entries) {
+       if (!(td->flags & TD_F_TRIM_BACKLOG))
+               return 0;
+
+       if (td->trim_entries) {
                int get_trim = 0;
 
                if (td->trim_batch) {
@@ -1137,7 +1129,10 @@ static int check_get_trim(struct thread_data *td, struct io_u *io_u)
 
 static int check_get_verify(struct thread_data *td, struct io_u *io_u)
 {
-       if (td->o.verify_backlog && td->io_hist_len) {
+       if (!(td->flags & TD_F_VER_BACKLOG))
+               return 0;
+
+       if (td->io_hist_len) {
                int get_verify = 0;
 
                if (td->verify_batch)
@@ -1168,7 +1163,7 @@ static int check_get_verify(struct thread_data *td, struct io_u *io_u)
 static void small_content_scramble(struct io_u *io_u)
 {
        unsigned int i, nr_blocks = io_u->buflen / 512;
-       unsigned long long boffset;
+       uint64_t boffset;
        unsigned int offset;
        void *p, *end;
 
@@ -1186,9 +1181,9 @@ static void small_content_scramble(struct io_u *io_u)
                 * and the actual offset.
                 */
                offset = (io_u->start_time.tv_usec ^ boffset) & 511;
-               offset &= ~(sizeof(unsigned long long) - 1);
-               if (offset >= 512 - sizeof(unsigned long long))
-                       offset -= sizeof(unsigned long long);
+               offset &= ~(sizeof(uint64_t) - 1);
+               if (offset >= 512 - sizeof(uint64_t))
+                       offset -= sizeof(uint64_t);
                memcpy(p + offset, &boffset, sizeof(boffset));
 
                end = p + 512 - sizeof(io_u->start_time);
@@ -1228,7 +1223,7 @@ struct io_u *get_io_u(struct thread_data *td)
        /*
         * If using an iolog, grab next piece if any available.
         */
-       if (td->o.read_iolog_file) {
+       if (td->flags & TD_F_READ_IOLOG) {
                if (read_iolog_get(td, io_u))
                        goto err_put;
        } else if (set_io_u_file(td, io_u)) {
@@ -1249,12 +1244,12 @@ struct io_u *get_io_u(struct thread_data *td)
                f->last_pos = io_u->offset + io_u->buflen;
 
                if (io_u->ddir == DDIR_WRITE) {
-                       if (td->o.refill_buffers) {
+                       if (td->flags & TD_F_REFILL_BUFFERS) {
                                io_u_fill_buffer(td, io_u,
                                        io_u->xfer_buflen, io_u->xfer_buflen);
-                       } else if (td->o.scramble_buffers)
+                       } else if (td->flags & TD_F_SCRAMBLE_BUFFERS)
                                do_scramble = 1;
-                       if (td->o.verify != VERIFY_NONE) {
+                       if (td->flags & TD_F_VER_NONE) {
                                populate_verify_io_u(td, io_u);
                                do_scramble = 0;
                        }
@@ -1315,7 +1310,7 @@ static void account_io_completion(struct thread_data *td, struct io_u *io_u,
                                  struct io_completion_data *icd,
                                  const enum fio_ddir idx, unsigned int bytes)
 {
-       unsigned long uninitialized_var(lusec);
+       unsigned long lusec = 0;
 
        if (!td->o.disable_clat || !td->o.disable_bw)
                lusec = utime_since(&io_u->issue_time, &icd->time);
@@ -1329,8 +1324,8 @@ static void account_io_completion(struct thread_data *td, struct io_u *io_u,
                if (td->o.max_latency && tusec > td->o.max_latency) {
                        if (!td->error)
                                log_err("fio: latency of %lu usec exceeds specified max (%u usec)\n", tusec, td->o.max_latency);
-                       td_verror(td, ETIME, "max latency exceeded");
-                       icd->error = ETIME;
+                       td_verror(td, ETIMEDOUT, "max latency exceeded");
+                       icd->error = ETIMEDOUT;
                }
        }
 
@@ -1347,7 +1342,8 @@ static void account_io_completion(struct thread_data *td, struct io_u *io_u,
 
 static long long usec_for_io(struct thread_data *td, enum fio_ddir ddir)
 {
-       unsigned long long secs, remainder, bps, bytes;
+       uint64_t secs, remainder, bps, bytes;
+
        bytes = td->this_io_bytes[ddir];
        bps = td->rate_bps[ddir];
        secs = bytes / bps;
@@ -1358,11 +1354,6 @@ static long long usec_for_io(struct thread_data *td, enum fio_ddir ddir)
 static void io_completed(struct thread_data *td, struct io_u *io_u,
                         struct io_completion_data *icd)
 {
-       /*
-        * Older gcc's are too dumb to realize that usec is always used
-        * initialized, silence that warning.
-        */
-       unsigned long uninitialized_var(usec);
        struct fio_file *f;
 
        dprint_io_u(io_u, "io complete");