Home | History | Annotate | Download | only in fio
      1 /*
      2  * TRIM/DISCARD support
      3  */
      4 #include <unistd.h>
      5 #include <fcntl.h>
      6 #include <string.h>
      7 #include <assert.h>
      8 #include <pthread.h>
      9 
     10 #include "fio.h"
     11 #include "trim.h"
     12 
     13 #ifdef FIO_HAVE_TRIM
     14 bool get_next_trim(struct thread_data *td, struct io_u *io_u)
     15 {
     16 	struct io_piece *ipo;
     17 
     18 	/*
     19 	 * this io_u is from a requeue, we already filled the offsets
     20 	 */
     21 	if (io_u->file)
     22 		return true;
     23 	if (flist_empty(&td->trim_list))
     24 		return false;
     25 
     26 	assert(td->trim_entries);
     27 	ipo = flist_first_entry(&td->trim_list, struct io_piece, trim_list);
     28 	remove_trim_entry(td, ipo);
     29 
     30 	io_u->offset = ipo->offset;
     31 	io_u->buflen = ipo->len;
     32 	io_u->file = ipo->file;
     33 
     34 	/*
     35 	 * If not verifying that trimmed ranges return zeroed data,
     36 	 * remove this from the to-read verify lists
     37 	 */
     38 	if (!td->o.trim_zero) {
     39 		if (ipo->flags & IP_F_ONLIST)
     40 			flist_del(&ipo->list);
     41 		else {
     42 			assert(ipo->flags & IP_F_ONRB);
     43 			rb_erase(&ipo->rb_node, &td->io_hist_tree);
     44 		}
     45 		td->io_hist_len--;
     46 		free(ipo);
     47 	} else
     48 		ipo->flags |= IP_F_TRIMMED;
     49 
     50 	if (!fio_file_open(io_u->file)) {
     51 		int r = td_io_open_file(td, io_u->file);
     52 
     53 		if (r) {
     54 			dprint(FD_VERIFY, "failed file %s open\n",
     55 					io_u->file->file_name);
     56 			return false;
     57 		}
     58 	}
     59 
     60 	get_file(io_u->file);
     61 	assert(fio_file_open(io_u->file));
     62 	io_u->ddir = DDIR_TRIM;
     63 	io_u->xfer_buf = NULL;
     64 	io_u->xfer_buflen = io_u->buflen;
     65 
     66 	dprint(FD_VERIFY, "get_next_trim: ret io_u %p\n", io_u);
     67 	return true;
     68 }
     69 
     70 bool io_u_should_trim(struct thread_data *td, struct io_u *io_u)
     71 {
     72 	unsigned long long val;
     73 	uint64_t frand_max;
     74 	unsigned long r;
     75 
     76 	if (!td->o.trim_percentage)
     77 		return false;
     78 
     79 	frand_max = rand_max(&td->trim_state);
     80 	r = __rand(&td->trim_state);
     81 	val = (frand_max / 100ULL);
     82 
     83 	val *= (unsigned long long) td->o.trim_percentage;
     84 	return r <= val;
     85 }
     86 #endif
     87