[PATCH] Get closer to FreeBSD compile
[fio.git] / fio.c
diff --git a/fio.c b/fio.c
index 01da9085d738f20e6652b625b9c0c2b1111b316e..1e94cc6fe984515cb026e52a0a4d8fd40bfe221e 100644 (file)
--- a/fio.c
+++ b/fio.c
  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  *
  */
-#include <stdio.h>
-#include <stdlib.h>
 #include <unistd.h>
 #include <fcntl.h>
 #include <string.h>
-#include <errno.h>
 #include <signal.h>
 #include <time.h>
 #include <assert.h>
 #include "fio.h"
 #include "os.h"
 
-#include "fio-time.h"
-#include "fio-stat.h"
-#include "fio-log.h"
-
 #define MASK   (4095)
 
 #define ALIGN(buf)     (char *) (((unsigned long) (buf) + MASK) & ~(MASK))
@@ -228,7 +221,7 @@ static int check_min_rate(struct thread_data *td, struct timeval *now)
 
                rate = (td->this_io_bytes[ddir] - td->rate_bytes) / spent;
                if (rate < td->ratemin) {
-                       printf("Client%d: min rate %d not met, got %ldKiB/sec\n", td->thread_number, td->ratemin, rate);
+                       printf("%s: min rate %d not met, got %ldKiB/sec\n", td->name, td->ratemin, rate);
                        if (rate_quit)
                                terminate_threads(td->groupid);
                        return 1;
@@ -288,38 +281,36 @@ static int verify_io_u_crc32(struct verify_header *hdr, struct io_u *io_u)
 {
        unsigned char *p = (unsigned char *) io_u->buf;
        unsigned long c;
-       int ret;
 
        p += sizeof(*hdr);
        c = crc32(p, hdr->len - sizeof(*hdr));
-       ret = c != hdr->crc32;
 
-       if (ret) {
+       if (c != hdr->crc32) {
                fprintf(stderr, "crc32: verify failed at %llu/%u\n", io_u->offset, io_u->buflen);
                fprintf(stderr, "crc32: wanted %lx, got %lx\n", hdr->crc32, c);
+               return 1;
        }
 
-       return ret;
+       return 0;
 }
 
 static int verify_io_u_md5(struct verify_header *hdr, struct io_u *io_u)
 {
        unsigned char *p = (unsigned char *) io_u->buf;
        struct md5_ctx md5_ctx;
-       int ret;
 
        memset(&md5_ctx, 0, sizeof(md5_ctx));
        p += sizeof(*hdr);
        md5_update(&md5_ctx, p, hdr->len - sizeof(*hdr));
 
-       ret = memcmp(hdr->md5_digest, md5_ctx.hash, sizeof(md5_ctx.hash));
-       if (ret) {
+       if (memcmp(hdr->md5_digest, md5_ctx.hash, sizeof(md5_ctx.hash))) {
                fprintf(stderr, "md5: verify failed at %llu/%u\n", io_u->offset, io_u->buflen);
                hexdump(hdr->md5_digest, sizeof(hdr->md5_digest));
                hexdump(md5_ctx.hash, sizeof(md5_ctx.hash));
+               return 1;
        }
 
-       return ret;
+       return 0;
 }
 
 static int verify_io_u(struct io_u *io_u)
@@ -389,7 +380,7 @@ static int get_rw_ddir(struct thread_data *td)
 
 /*
  * fill body of io_u->buf with random data and add a header with the
- * (eg) sha1sum of that data.
+ * crc32 or md5 sum of that data.
  */
 static void populate_io_u(struct thread_data *td, struct io_u *io_u)
 {
@@ -457,21 +448,22 @@ static int fill_io_u(struct thread_data *td, struct io_u *io_u)
        return 1;
 }
 
-#define queue_full(td) (list_empty(&(td)->io_u_freelist))
+#define queue_full(td) list_empty(&(td)->io_u_freelist)
 
 struct io_u *__get_io_u(struct thread_data *td)
 {
-       struct io_u *io_u;
+       struct io_u *io_u = NULL;
 
-       if (queue_full(td))
-               return NULL;
+       if (!queue_full(td)) {
+               io_u = list_entry(td->io_u_freelist.next, struct io_u, list);
+
+               io_u->error = 0;
+               io_u->resid = 0;
+               list_del(&io_u->list);
+               list_add(&io_u->list, &td->io_u_busylist);
+               td->cur_depth++;
+       }
 
-       io_u = list_entry(td->io_u_freelist.next, struct io_u, list);
-       io_u->error = 0;
-       io_u->resid = 0;
-       list_del(&io_u->list);
-       list_add(&io_u->list, &td->io_u_busylist);
-       td->cur_depth++;
        return io_u;
 }
 
@@ -528,17 +520,19 @@ static int get_next_verify(struct thread_data *td, struct io_u *io_u)
 {
        struct io_piece *ipo;
 
-       if (list_empty(&td->io_hist_list))
-               return 1;
+       if (!list_empty(&td->io_hist_list)) {
+               ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
 
-       ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
-       list_del(&ipo->list);
+               list_del(&ipo->list);
 
-       io_u->offset = ipo->offset;
-       io_u->buflen = ipo->len;
-       io_u->ddir = DDIR_READ;
-       free(ipo);
-       return 0;
+               io_u->offset = ipo->offset;
+               io_u->buflen = ipo->len;
+               io_u->ddir = DDIR_READ;
+               free(ipo);
+               return 0;
+       }
+
+       return 1;
 }
 
 static int sync_td(struct thread_data *td)
@@ -743,6 +737,10 @@ static void do_verify(struct thread_data *td)
        td_set_runstate(td, TD_RUNNING);
 }
 
+/*
+ * Main IO worker functions. It retrieves io_u's to process and queues
+ * and reaps them, checking for rate and errors along the way.
+ */
 static void do_io(struct thread_data *td)
 {
        struct io_completion_data icd;
@@ -973,10 +971,10 @@ static int create_file(struct thread_data *td, unsigned long long size,
 
        if (!extend) {
                oflags = O_CREAT | O_TRUNC;
-               printf("Client%d: Laying out IO file (%LuMiB)\n", td->thread_number, size >> 20);
+               printf("%s: Laying out IO file (%LuMiB)\n", td->name, size >> 20);
        } else {
                oflags = O_APPEND;
-               printf("Client%d: Extending IO file (%Lu -> %LuMiB)\n", td->thread_number, (td->file_size - size) >> 20, td->file_size >> 20);
+               printf("%s: Extending IO file (%Lu -> %LuMiB)\n", td->name, (td->file_size - size) >> 20, td->file_size >> 20);
        }
 
        td->fd = open(td->file_name, O_WRONLY | oflags, 0644);
@@ -1082,13 +1080,13 @@ static int get_file_size(struct thread_data *td)
                return ret;
 
        if (td->file_offset > td->real_file_size) {
-               fprintf(stderr, "Client%d: offset extends end (%Lu > %Lu)\n", td->thread_number, td->file_offset, td->real_file_size);
+               fprintf(stderr, "%s: offset extends end (%Lu > %Lu)\n", td->name, td->file_offset, td->real_file_size);
                return 1;
        }
 
        td->io_size = td->file_size;
        if (td->io_size == 0) {
-               fprintf(stderr, "Client%d: no io blocks\n", td->thread_number);
+               fprintf(stderr, "%s: no io blocks\n", td->name);
                td_verror(td, EINVAL);
                return 1;
        }
@@ -1676,7 +1674,7 @@ static void fio_unpin_memory(void *pinned)
 
 static void *fio_pin_memory(void)
 {
-       long pagesize, pages;
+       unsigned long long phys_mem;
        void *ptr;
 
        if (!mlock_size)
@@ -1685,13 +1683,10 @@ static void *fio_pin_memory(void)
        /*
         * Don't allow mlock of more than real_mem-128MB
         */
-       pagesize = sysconf(_SC_PAGESIZE);
-       pages = sysconf(_SC_PHYS_PAGES);
-       if (pages != -1 && pagesize != -1) {
-               unsigned long long real_mem = pages * pagesize;
-
-               if ((mlock_size + 128 * 1024 * 1024) > real_mem) {
-                       mlock_size = real_mem - 128 * 1024 * 1024;
+       phys_mem = os_phys_mem();
+       if (phys_mem) {
+               if ((mlock_size + 128 * 1024 * 1024) > phys_mem) {
+                       mlock_size = phys_mem - 128 * 1024 * 1024;
                        printf("fio: limiting mlocked memory to %lluMiB\n",
                                                        mlock_size >> 20);
                }