Some more bits of blktrace support
[fio.git] / verify.c
CommitLineData
e29d1b70
JA
1/*
2 * IO verification helpers
3 */
4#include <unistd.h>
5#include <fcntl.h>
6#include <string.h>
7
8#include "fio.h"
e29d1b70
JA
9
10static void fill_random_bytes(struct thread_data *td,
11 unsigned char *p, unsigned int len)
12{
13 unsigned int todo;
14 double r;
15
16 while (len) {
17 r = os_random_double(&td->verify_state);
18
19 /*
20 * lrand48_r seems to be broken and only fill the bottom
21 * 32-bits, even on 64-bit archs with 64-bit longs
22 */
23 todo = sizeof(r);
24 if (todo > len)
25 todo = len;
26
27 memcpy(p, &r, todo);
28
29 len -= todo;
30 p += todo;
31 }
32}
33
34static void hexdump(void *buffer, int len)
35{
36 unsigned char *p = buffer;
37 int i;
38
39 for (i = 0; i < len; i++)
6d86144d
JA
40 log_info("%02x", p[i]);
41 log_info("\n");
e29d1b70
JA
42}
43
44static int verify_io_u_crc32(struct verify_header *hdr, struct io_u *io_u)
45{
46 unsigned char *p = (unsigned char *) io_u->buf;
47 unsigned long c;
48
49 p += sizeof(*hdr);
50 c = crc32(p, hdr->len - sizeof(*hdr));
51
52 if (c != hdr->crc32) {
a4f4fdd7 53 log_err("crc32: verify failed at %llu/%lu\n", io_u->offset, io_u->buflen);
e29d1b70
JA
54 log_err("crc32: wanted %lx, got %lx\n", hdr->crc32, c);
55 return 1;
56 }
57
58 return 0;
59}
60
61static int verify_io_u_md5(struct verify_header *hdr, struct io_u *io_u)
62{
63 unsigned char *p = (unsigned char *) io_u->buf;
64 struct md5_ctx md5_ctx;
65
66 memset(&md5_ctx, 0, sizeof(md5_ctx));
67 p += sizeof(*hdr);
68 md5_update(&md5_ctx, p, hdr->len - sizeof(*hdr));
69
70 if (memcmp(hdr->md5_digest, md5_ctx.hash, sizeof(md5_ctx.hash))) {
a4f4fdd7 71 log_err("md5: verify failed at %llu/%lu\n", io_u->offset, io_u->buflen);
e29d1b70
JA
72 hexdump(hdr->md5_digest, sizeof(hdr->md5_digest));
73 hexdump(md5_ctx.hash, sizeof(md5_ctx.hash));
74 return 1;
75 }
76
77 return 0;
78}
79
36690c9b 80int verify_io_u(struct thread_data *td, struct io_u *io_u)
e29d1b70
JA
81{
82 struct verify_header *hdr = (struct verify_header *) io_u->buf;
83 int ret;
84
36690c9b
JA
85 if (td->o.verify == VERIFY_NULL)
86 return 0;
87
36167d82
JA
88 if (hdr->fio_magic != FIO_HDR_MAGIC) {
89 log_err("Bad verify header %x\n", hdr->fio_magic);
a7dfe862 90 return EIO;
36167d82 91 }
e29d1b70
JA
92
93 if (hdr->verify_type == VERIFY_MD5)
94 ret = verify_io_u_md5(hdr, io_u);
95 else if (hdr->verify_type == VERIFY_CRC32)
96 ret = verify_io_u_crc32(hdr, io_u);
97 else {
1e97cce9 98 log_err("Bad verify type %u\n", hdr->verify_type);
e29d1b70
JA
99 ret = 1;
100 }
101
a7dfe862
JA
102 if (ret)
103 return EIO;
104
105 return 0;
e29d1b70
JA
106}
107
108static void fill_crc32(struct verify_header *hdr, void *p, unsigned int len)
109{
110 hdr->crc32 = crc32(p, len);
111}
112
113static void fill_md5(struct verify_header *hdr, void *p, unsigned int len)
114{
115 struct md5_ctx md5_ctx;
116
117 memset(&md5_ctx, 0, sizeof(md5_ctx));
118 md5_update(&md5_ctx, p, len);
119 memcpy(hdr->md5_digest, md5_ctx.hash, sizeof(md5_ctx.hash));
120}
121
122/*
123 * fill body of io_u->buf with random data and add a header with the
124 * crc32 or md5 sum of that data.
125 */
126void populate_verify_io_u(struct thread_data *td, struct io_u *io_u)
127{
128 unsigned char *p = (unsigned char *) io_u->buf;
129 struct verify_header hdr;
130
9cc3d150
JA
131 if (td->o.verify == VERIFY_NULL)
132 return;
133
e29d1b70
JA
134 hdr.fio_magic = FIO_HDR_MAGIC;
135 hdr.len = io_u->buflen;
136 p += sizeof(hdr);
137 fill_random_bytes(td, p, io_u->buflen - sizeof(hdr));
138
2dc1bbeb 139 if (td->o.verify == VERIFY_MD5) {
e29d1b70
JA
140 fill_md5(&hdr, p, io_u->buflen - sizeof(hdr));
141 hdr.verify_type = VERIFY_MD5;
36690c9b 142 } else if (td->o.verify == VERIFY_CRC32) {
e29d1b70
JA
143 fill_crc32(&hdr, p, io_u->buflen - sizeof(hdr));
144 hdr.verify_type = VERIFY_CRC32;
145 }
146
147 memcpy(io_u->buf, &hdr, sizeof(hdr));
148}
149
150int get_next_verify(struct thread_data *td, struct io_u *io_u)
151{
8de8f047 152 struct io_piece *ipo = NULL;
e29d1b70 153
d2d7fa53
JA
154 /*
155 * this io_u is from a requeue, we already filled the offsets
156 */
157 if (io_u->file)
158 return 0;
159
8de8f047
JA
160 if (!RB_EMPTY_ROOT(&td->io_hist_tree)) {
161 struct rb_node *n = rb_first(&td->io_hist_tree);
e29d1b70 162
8de8f047 163 ipo = rb_entry(n, struct io_piece, rb_node);
4b87898e 164 rb_erase(n, &td->io_hist_tree);
8de8f047
JA
165 } else if (!list_empty(&td->io_hist_list)) {
166 ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
167 list_del(&ipo->list);
168 }
e29d1b70 169
8de8f047 170 if (ipo) {
e29d1b70
JA
171 io_u->offset = ipo->offset;
172 io_u->buflen = ipo->len;
36167d82 173 io_u->file = ipo->file;
e29d1b70 174 io_u->ddir = DDIR_READ;
36167d82
JA
175 io_u->xfer_buf = io_u->buf;
176 io_u->xfer_buflen = io_u->buflen;
e29d1b70
JA
177 free(ipo);
178 return 0;
179 }
180
181 return 1;
182}