cf42625b20f26c51f95af6e3d45ba29f838543c5
[fio.git] / trim.c
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 int 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 0;
23         if (flist_empty(&td->trim_list))
24                 return 0;
25
26         assert(td->trim_entries);
27         ipo = flist_entry(td->trim_list.next, struct io_piece, trim_list);
28         remove_trim_entry(td, ipo);
29         ipo->flags |= IP_F_TRIMMED;
30
31         /*
32          * If not verifying that trimmed ranges return zeroed data,
33          * remove this from the to-read verify lists
34          */
35         if (!td->o.trim_zero) {
36                 if (ipo->flags & IP_F_ONLIST)
37                         flist_del(&ipo->list);
38                 else {
39                         assert(ipo->flags & IP_F_ONRB);
40                         rb_erase(&ipo->rb_node, &td->io_hist_tree);
41                 }
42                 td->io_hist_len--;
43         }
44
45         io_u->offset = ipo->offset;
46         io_u->buflen = ipo->len;
47         io_u->file = ipo->file;
48
49         if (!fio_file_open(io_u->file)) {
50                 int r = td_io_open_file(td, io_u->file);
51
52                 if (r) {
53                         dprint(FD_VERIFY, "failed file %s open\n",
54                                         io_u->file->file_name);
55                         return 1;
56                 }
57         }
58
59         get_file(ipo->file);
60         assert(fio_file_open(io_u->file));
61         io_u->ddir = DDIR_TRIM;
62         io_u->xfer_buf = NULL;
63         io_u->xfer_buflen = io_u->buflen;
64
65         free(ipo);
66         dprint(FD_VERIFY, "get_next_trim: ret io_u %p\n", io_u);
67         return 0;
68 }
69
70 int io_u_should_trim(struct thread_data *td, struct io_u *io_u)
71 {
72         unsigned long long val;
73         long r;
74
75         if (!td->o.trim_percentage)
76                 return 0;
77
78         r = os_random_long(&td->trim_state);
79         val = (OS_RAND_MAX / 100ULL);
80         val *= (unsigned long long) td->o.trim_percentage;
81
82         return r <= val;
83 }
84 #endif