static void disk_util_timer_arm(void);
static void print_thread_status(void);
+extern unsigned long long mlock_size;
+
/*
* thread life cycle
*/
memcpy(io_u->buf, &hdr, sizeof(hdr));
}
-static int td_io_prep(struct thread_data *td, struct io_u *io_u, int read)
+static int td_io_prep(struct thread_data *td, struct io_u *io_u)
{
- if (read)
- io_u->ddir = DDIR_READ;
- else
- io_u->ddir = DDIR_WRITE;
-
if (td->io_prep && td->io_prep(td, io_u))
return 1;
td->cur_depth--;
}
+static int fill_io_u(struct thread_data *td, struct io_u *io_u)
+{
+ /*
+ * If using an iolog, grab next piece if any available.
+ */
+ if (td->iolog) {
+ struct io_piece *ipo;
+
+ if (list_empty(&td->io_log_list))
+ return 1;
+
+ ipo = list_entry(td->io_log_list.next, struct io_piece, list);
+ list_del(&ipo->list);
+ io_u->offset = ipo->offset;
+ io_u->buflen = ipo->len;
+ io_u->ddir = ipo->ddir;
+ free(ipo);
+ return 0;
+ }
+
+ /*
+ * No log, let the seq/rand engine retrieve the next position.
+ */
+ if (!get_next_offset(td, &io_u->offset)) {
+ io_u->buflen = get_next_buflen(td);
+
+ if (io_u->buflen) {
+ io_u->ddir = get_rw_ddir(td);
+ return 0;
+ }
+ }
+
+ return 1;
+}
+
#define queue_full(td) (list_empty(&(td)->io_u_freelist))
struct io_u *__get_io_u(struct thread_data *td)
td->last_pos += td->zone_skip;
}
- if (get_next_offset(td, &io_u->offset)) {
- put_io_u(td, io_u);
- return NULL;
- }
-
- io_u->buflen = get_next_buflen(td);
- if (!io_u->buflen) {
+ if (fill_io_u(td, io_u)) {
put_io_u(td, io_u);
return NULL;
}
return NULL;
}
- if (!td->sequential)
+ if (!td->iolog && !td->sequential)
mark_random_map(td, io_u);
td->last_pos += io_u->buflen;
if (td->verify != VERIFY_NONE)
populate_io_u(td, io_u);
- if (td_io_prep(td, io_u, get_rw_ddir(td))) {
+ if (td_io_prep(td, io_u)) {
put_io_u(td, io_u);
return NULL;
}
td->runstate = runstate;
}
-static int get_next_verify(struct thread_data *td,
- unsigned long long *offset, unsigned int *len)
+static int get_next_verify(struct thread_data *td, struct io_u *io_u)
{
struct io_piece *ipo;
ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
list_del(&ipo->list);
- *offset = ipo->offset;
- *len = ipo->len;
+ io_u->offset = ipo->offset;
+ io_u->buflen = ipo->len;
+ io_u->ddir = DDIR_READ;
free(ipo);
return 0;
}
list_add(&ipo->list, entry);
}
+static int init_iolog(struct thread_data *td)
+{
+ unsigned long long offset;
+ unsigned int bytes;
+ char *str, *p;
+ FILE *f;
+ int rw, i, reads, writes;
+
+ if (!td->iolog)
+ return 0;
+
+ f = fopen(td->iolog_file, "r");
+ if (!f) {
+ perror("fopen iolog");
+ return 1;
+ }
+
+ str = malloc(4096);
+ reads = writes = i = 0;
+ while ((p = fgets(str, 4096, f)) != NULL) {
+ struct io_piece *ipo;
+
+ if (sscanf(p, "%d,%llu,%u", &rw, &offset, &bytes) != 3) {
+ fprintf(stderr, "bad iolog: %s\n", p);
+ continue;
+ }
+ if (rw == DDIR_READ)
+ reads++;
+ else if (rw == DDIR_WRITE)
+ writes++;
+ else {
+ fprintf(stderr, "bad ddir: %d\n", rw);
+ continue;
+ }
+
+ ipo = malloc(sizeof(*ipo));
+ INIT_LIST_HEAD(&ipo->list);
+ ipo->offset = offset;
+ ipo->len = bytes;
+ if (bytes > td->max_bs)
+ td->max_bs = bytes;
+ ipo->ddir = rw;
+ list_add_tail(&ipo->list, &td->io_log_list);
+ i++;
+ }
+
+ free(str);
+ fclose(f);
+
+ if (!i)
+ return 1;
+
+ if (reads && !writes)
+ td->ddir = DDIR_READ;
+ else if (!reads && writes)
+ td->ddir = DDIR_READ;
+ else
+ td->iomix = 1;
+
+ return 0;
+}
+
static int sync_td(struct thread_data *td)
{
if (td->io_sync)
if (!io_u)
break;
- if (get_next_verify(td, &io_u->offset, &io_u->buflen)) {
+ if (get_next_verify(td, io_u)) {
put_io_u(td, io_u);
break;
}
- if (td_io_prep(td, io_u, 1)) {
+ if (td_io_prep(td, io_u)) {
put_io_u(td, io_u);
break;
}
if (td->cur_depth)
cleanup_pending_aio(td);
- if (should_fsync(td))
+ if (should_fsync(td) && td->end_fsync)
sync_td(td);
}
return fio_posixaio_init(td);
else if (td->io_engine == FIO_SGIO)
return fio_sgio_init(td);
+ else if (td->io_engine == FIO_SPLICEIO)
+ return fio_spliceio_init(td);
else {
fprintf(stderr, "bad io_engine %d\n", td->io_engine);
return 1;
}
}
- INIT_LIST_HEAD(&td->io_u_freelist);
- INIT_LIST_HEAD(&td->io_u_busylist);
- INIT_LIST_HEAD(&td->io_hist_list);
-
p = ALIGN(td->orig_buffer);
for (i = 0; i < max_units; i++) {
io_u = malloc(sizeof(*io_u));
td->pid = getpid();
+ INIT_LIST_HEAD(&td->io_u_freelist);
+ INIT_LIST_HEAD(&td->io_u_busylist);
+ INIT_LIST_HEAD(&td->io_hist_list);
+ INIT_LIST_HEAD(&td->io_log_list);
+
if (init_io_u(td))
goto err;
if (init_io(td))
goto err;
+ if (init_iolog(td))
+ goto err;
+
if (td->ioprio) {
if (ioprio_set(IOPRIO_WHO_PROCESS, 0, td->ioprio) == -1) {
td_verror(td, errno);
do_io(td);
td->runtime[td->ddir] += mtime_since_now(&td->start);
- if (td_rw(td))
+ if (td_rw(td) && td->io_bytes[td->ddir ^ 1])
td->runtime[td->ddir ^ 1] = td->runtime[td->ddir];
update_rusage_stat(td);
printf("Client%d (groupid=%d): err=%2d:\n", td->thread_number, td->groupid, td->error);
show_ddir_status(td, rs, td->ddir);
- show_ddir_status(td, rs, td->ddir ^ 1);
+ if (td->io_bytes[td->ddir ^ 1])
+ show_ddir_status(td, rs, td->ddir ^ 1);
if (td->runtime[0] + td->runtime[1]) {
double runt = td->runtime[0] + td->runtime[1];
struct thread_data *td;
unsigned long spent;
int i, todo, nr_running, m_rate, t_rate, nr_started;
+ void *mlocked_mem = NULL;
printf("Starting %d thread%s\n", thread_number, thread_number > 1 ? "s" : "");
+
+ if (mlock_size) {
+ mlocked_mem = malloc(mlock_size);
+ if (!mlocked_mem) {
+ perror("malloc locked mem");
+ return;
+ }
+ if (mlock(mlocked_mem, mlock_size) < 0) {
+ free(mlocked_mem);
+ perror("mlock");
+ return;
+ }
+ }
+
fflush(stdout);
signal(SIGINT, sig_handler);
}
update_io_ticks();
+
+ if (mlocked_mem) {
+ if (munlock(mlocked_mem, mlock_size) < 0)
+ perror("munlock");
+ free(mlocked_mem);
+ }
}
static void show_group_stats(struct group_run_stats *rs, int id)
projects / fio.git / blobdiff