Fix race on io_u->flags
[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>
97af62ce 7#include <assert.h>
e8462bd8 8#include <pthread.h>
e29d1b70
JA
9
10#include "fio.h"
4f5af7b2 11#include "verify.h"
e8462bd8 12#include "smalloc.h"
e29d1b70 13
eef6eea1
JA
14#include "crc/md5.h"
15#include "crc/crc64.h"
16#include "crc/crc32.h"
bac39e0e 17#include "crc/crc32c.h"
eef6eea1
JA
18#include "crc/crc16.h"
19#include "crc/crc7.h"
20#include "crc/sha256.h"
21#include "crc/sha512.h"
7c353ceb 22#include "crc/sha1.h"
cd14cc10 23
90059d65 24static void fill_random_bytes(struct thread_data *td, void *p, unsigned int len)
e29d1b70
JA
25{
26 unsigned int todo;
4c5946c6 27 int r;
e29d1b70
JA
28
29 while (len) {
4c5946c6 30 r = os_random_long(&td->verify_state);
e29d1b70
JA
31
32 /*
33 * lrand48_r seems to be broken and only fill the bottom
34 * 32-bits, even on 64-bit archs with 64-bit longs
35 */
36 todo = sizeof(r);
37 if (todo > len)
38 todo = len;
39
40 memcpy(p, &r, todo);
41
42 len -= todo;
43 p += todo;
44 }
45}
46
90059d65
JA
47static void fill_pattern(struct thread_data *td, void *p, unsigned int len)
48{
49 switch (td->o.verify_pattern_bytes) {
50 case 0:
bd6f78b2 51 dprint(FD_VERIFY, "fill random bytes len=%u\n", len);
90059d65
JA
52 fill_random_bytes(td, p, len);
53 break;
54 case 1:
bd6f78b2 55 dprint(FD_VERIFY, "fill verify pattern b=0 len=%u\n", len);
0e92f873 56 memset(p, td->o.verify_pattern[0], len);
90059d65 57 break;
0e92f873
RR
58 default: {
59 unsigned int i = 0, size = 0;
90059d65
JA
60 unsigned char *b = p;
61
bd6f78b2
JA
62 dprint(FD_VERIFY, "fill verify pattern b=%d len=%u\n",
63 td->o.verify_pattern_bytes, len);
64
90059d65 65 while (i < len) {
0e92f873
RR
66 size = td->o.verify_pattern_bytes;
67 if (size > (len - i))
68 size = len - i;
69 memcpy(b+i, td->o.verify_pattern, size);
70 i += size;
90059d65
JA
71 }
72 break;
73 }
74 }
75}
76
4764aec9 77static void memswp(void *buf1, void *buf2, unsigned int len)
546a9142 78{
dee6de74
SL
79 char swap[200];
80
81 assert(len <= sizeof(swap));
90059d65 82
546a9142
SL
83 memcpy(&swap, buf1, len);
84 memcpy(buf1, buf2, len);
85 memcpy(buf2, &swap, len);
86}
87
e29d1b70
JA
88static void hexdump(void *buffer, int len)
89{
90 unsigned char *p = buffer;
91 int i;
92
93 for (i = 0; i < len; i++)
6d86144d
JA
94 log_info("%02x", p[i]);
95 log_info("\n");
e29d1b70
JA
96}
97
87677832
JA
98/*
99 * Prepare for seperation of verify_header and checksum header
100 */
546dfd9f 101static inline unsigned int __hdr_size(int verify_type)
87677832 102{
5921e80c 103 unsigned int len = len;
87677832 104
546dfd9f
JA
105 switch (verify_type) {
106 case VERIFY_NONE:
107 case VERIFY_NULL:
108 len = 0;
109 break;
110 case VERIFY_MD5:
111 len = sizeof(struct vhdr_md5);
112 break;
113 case VERIFY_CRC64:
114 len = sizeof(struct vhdr_crc64);
115 break;
bac39e0e 116 case VERIFY_CRC32C:
546dfd9f 117 case VERIFY_CRC32:
3845591f 118 case VERIFY_CRC32C_INTEL:
546dfd9f
JA
119 len = sizeof(struct vhdr_crc32);
120 break;
121 case VERIFY_CRC16:
122 len = sizeof(struct vhdr_crc16);
123 break;
124 case VERIFY_CRC7:
125 len = sizeof(struct vhdr_crc7);
126 break;
127 case VERIFY_SHA256:
128 len = sizeof(struct vhdr_sha256);
129 break;
130 case VERIFY_SHA512:
131 len = sizeof(struct vhdr_sha512);
132 break;
7437ee87
SL
133 case VERIFY_META:
134 len = sizeof(struct vhdr_meta);
135 break;
7c353ceb
JA
136 case VERIFY_SHA1:
137 len = sizeof(struct vhdr_sha1);
138 break;
546dfd9f
JA
139 default:
140 log_err("fio: unknown verify header!\n");
141 assert(0);
142 }
143
144 return len + sizeof(struct verify_header);
87677832
JA
145}
146
147static inline unsigned int hdr_size(struct verify_header *hdr)
148{
546dfd9f
JA
149 return __hdr_size(hdr->verify_type);
150}
151
152static void *hdr_priv(struct verify_header *hdr)
153{
154 void *priv = hdr;
155
156 return priv + sizeof(struct verify_header);
87677832
JA
157}
158
d9f2caf3
JA
159/*
160 * Return data area 'header_num'
161 */
162static inline void *io_u_verify_off(struct verify_header *hdr,
5ec10eaa 163 struct io_u *io_u, unsigned char header_num)
d9f2caf3 164{
87677832 165 return io_u->buf + header_num * hdr->len + hdr_size(hdr);
d9f2caf3
JA
166}
167
7437ee87 168static int verify_io_u_meta(struct verify_header *hdr, struct thread_data *td,
5ec10eaa 169 struct io_u *io_u, unsigned int header_num)
7437ee87
SL
170{
171 struct vhdr_meta *vh = hdr_priv(hdr);
172
bd6f78b2
JA
173 dprint(FD_VERIFY, "meta verify io_u %p, len %u\n", io_u, hdr->len);
174
7437ee87
SL
175 if (vh->offset != io_u->offset + header_num * td->o.verify_interval) {
176 log_err("meta: verify failed at %llu/%u\n",
5ec10eaa 177 io_u->offset + header_num * hdr->len, hdr->len);
9fd18969 178 return EILSEQ;
7437ee87
SL
179 }
180
181 return 0;
182}
183
cd14cc10
JA
184static int verify_io_u_sha512(struct verify_header *hdr, struct io_u *io_u,
185 unsigned int header_num)
186{
187 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f 188 struct vhdr_sha512 *vh = hdr_priv(hdr);
cd14cc10
JA
189 uint8_t sha512[128];
190 struct sha512_ctx sha512_ctx = {
191 .buf = sha512,
192 };
193
bd6f78b2
JA
194 dprint(FD_VERIFY, "sha512 verify io_u %p, len %u\n", io_u, hdr->len);
195
cd14cc10 196 sha512_init(&sha512_ctx);
87677832 197 sha512_update(&sha512_ctx, p, hdr->len - hdr_size(hdr));
cd14cc10 198
546dfd9f 199 if (memcmp(vh->sha512, sha512_ctx.buf, sizeof(sha512))) {
cd14cc10 200 log_err("sha512: verify failed at %llu/%u\n",
5ec10eaa 201 io_u->offset + header_num * hdr->len, hdr->len);
546dfd9f 202 hexdump(vh->sha512, sizeof(vh->sha512));
cd14cc10 203 hexdump(sha512_ctx.buf, sizeof(sha512));
9fd18969 204 return EILSEQ;
cd14cc10
JA
205 }
206
207 return 0;
208}
209
210static int verify_io_u_sha256(struct verify_header *hdr, struct io_u *io_u,
211 unsigned int header_num)
212{
213 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f 214 struct vhdr_sha256 *vh = hdr_priv(hdr);
cd14cc10
JA
215 uint8_t sha256[128];
216 struct sha256_ctx sha256_ctx = {
217 .buf = sha256,
218 };
219
bd6f78b2
JA
220 dprint(FD_VERIFY, "sha256 verify io_u %p, len %u\n", io_u, hdr->len);
221
cd14cc10 222 sha256_init(&sha256_ctx);
87677832 223 sha256_update(&sha256_ctx, p, hdr->len - hdr_size(hdr));
cd14cc10 224
546dfd9f 225 if (memcmp(vh->sha256, sha256_ctx.buf, sizeof(sha256))) {
cd14cc10 226 log_err("sha256: verify failed at %llu/%u\n",
5ec10eaa 227 io_u->offset + header_num * hdr->len, hdr->len);
546dfd9f 228 hexdump(vh->sha256, sizeof(vh->sha256));
cd14cc10 229 hexdump(sha256_ctx.buf, sizeof(sha256));
9fd18969 230 return EILSEQ;
cd14cc10
JA
231 }
232
233 return 0;
234}
235
7c353ceb
JA
236static int verify_io_u_sha1(struct verify_header *hdr, struct io_u *io_u,
237 unsigned int header_num)
238{
239 void *p = io_u_verify_off(hdr, io_u, header_num);
240 struct vhdr_sha1 *vh = hdr_priv(hdr);
241 uint32_t sha1[5];
242 struct sha1_ctx sha1_ctx = {
243 .H = sha1,
244 };
245
246 dprint(FD_VERIFY, "sha1 verify io_u %p, len %u\n", io_u, hdr->len);
247
248 sha1_init(&sha1_ctx);
249 sha1_update(&sha1_ctx, p, hdr->len - hdr_size(hdr));
250
251 if (memcmp(vh->sha1, sha1_ctx.H, sizeof(sha1))) {
252 log_err("sha1: verify failed at %llu/%u\n",
253 io_u->offset + header_num * hdr->len, hdr->len);
254 hexdump(vh->sha1, sizeof(vh->sha1));
255 hexdump(sha1_ctx.H, sizeof(sha1));
256 return EILSEQ;
257 }
258
259 return 0;
260}
261
3f9f4e26 262static int verify_io_u_crc7(struct verify_header *hdr, struct io_u *io_u,
5ec10eaa 263 unsigned char header_num)
1e154bdb 264{
d9f2caf3 265 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f 266 struct vhdr_crc7 *vh = hdr_priv(hdr);
1e154bdb
JA
267 unsigned char c;
268
bd6f78b2
JA
269 dprint(FD_VERIFY, "crc7 verify io_u %p, len %u\n", io_u, hdr->len);
270
87677832 271 c = crc7(p, hdr->len - hdr_size(hdr));
1e154bdb 272
546dfd9f 273 if (c != vh->crc7) {
3f9f4e26 274 log_err("crc7: verify failed at %llu/%u\n",
5ec10eaa 275 io_u->offset + header_num * hdr->len, hdr->len);
546dfd9f 276 log_err("crc7: wanted %x, got %x\n", vh->crc7, c);
9fd18969 277 return EILSEQ;
1e154bdb
JA
278 }
279
280 return 0;
281}
282
3f9f4e26 283static int verify_io_u_crc16(struct verify_header *hdr, struct io_u *io_u,
5ec10eaa 284 unsigned int header_num)
969f7ed3 285{
d9f2caf3 286 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f 287 struct vhdr_crc16 *vh = hdr_priv(hdr);
969f7ed3
JA
288 unsigned short c;
289
bd6f78b2
JA
290 dprint(FD_VERIFY, "crc16 verify io_u %p, len %u\n", io_u, hdr->len);
291
87677832 292 c = crc16(p, hdr->len - hdr_size(hdr));
969f7ed3 293
546dfd9f 294 if (c != vh->crc16) {
3f9f4e26 295 log_err("crc16: verify failed at %llu/%u\n",
5ec10eaa 296 io_u->offset + header_num * hdr->len, hdr->len);
546dfd9f 297 log_err("crc16: wanted %x, got %x\n", vh->crc16, c);
9fd18969 298 return EILSEQ;
969f7ed3
JA
299 }
300
301 return 0;
302}
303
3f9f4e26 304static int verify_io_u_crc64(struct verify_header *hdr, struct io_u *io_u,
5ec10eaa 305 unsigned int header_num)
d77a7af3 306{
d9f2caf3 307 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f 308 struct vhdr_crc64 *vh = hdr_priv(hdr);
d77a7af3
JA
309 unsigned long long c;
310
bd6f78b2
JA
311 dprint(FD_VERIFY, "crc64 verify io_u %p, len %u\n", io_u, hdr->len);
312
87677832 313 c = crc64(p, hdr->len - hdr_size(hdr));
d77a7af3 314
546dfd9f 315 if (c != vh->crc64) {
3f9f4e26
SL
316 log_err("crc64: verify failed at %llu/%u\n",
317 io_u->offset + header_num * hdr->len,
318 hdr->len);
5ec10eaa
JA
319 log_err("crc64: wanted %llx, got %llx\n",
320 (unsigned long long) vh->crc64, c);
9fd18969 321 return EILSEQ;
d77a7af3
JA
322 }
323
324 return 0;
325}
326
3f9f4e26 327static int verify_io_u_crc32(struct verify_header *hdr, struct io_u *io_u,
5ec10eaa 328 unsigned int header_num)
e29d1b70 329{
d9f2caf3 330 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f
JA
331 struct vhdr_crc32 *vh = hdr_priv(hdr);
332 uint32_t c;
e29d1b70 333
bd6f78b2
JA
334 dprint(FD_VERIFY, "crc32 verify io_u %p, len %u\n", io_u, hdr->len);
335
87677832 336 c = crc32(p, hdr->len - hdr_size(hdr));
e29d1b70 337
546dfd9f 338 if (c != vh->crc32) {
3f9f4e26 339 log_err("crc32: verify failed at %llu/%u\n",
5ec10eaa 340 io_u->offset + header_num * hdr->len, hdr->len);
546dfd9f 341 log_err("crc32: wanted %x, got %x\n", vh->crc32, c);
9fd18969 342 return EILSEQ;
e29d1b70
JA
343 }
344
345 return 0;
346}
347
bac39e0e
JA
348static int verify_io_u_crc32c(struct verify_header *hdr, struct io_u *io_u,
349 unsigned int header_num)
350{
351 void *p = io_u_verify_off(hdr, io_u, header_num);
352 struct vhdr_crc32 *vh = hdr_priv(hdr);
353 uint32_t c;
354
355 dprint(FD_VERIFY, "crc32c verify io_u %p, len %u\n", io_u, hdr->len);
356
3845591f
JA
357 if (hdr->verify_type == VERIFY_CRC32C_INTEL)
358 c = crc32c_intel(p, hdr->len - hdr_size(hdr));
359 else
360 c = crc32c(p, hdr->len - hdr_size(hdr));
bac39e0e
JA
361
362 if (c != vh->crc32) {
363 log_err("crc32c: verify failed at %llu/%u\n",
364 io_u->offset + header_num * hdr->len, hdr->len);
365 log_err("crc32c: wanted %x, got %x\n", vh->crc32, c);
9fd18969 366 return EILSEQ;
bac39e0e
JA
367 }
368
369 return 0;
370}
371
3f9f4e26 372static int verify_io_u_md5(struct verify_header *hdr, struct io_u *io_u,
5ec10eaa 373 unsigned int header_num)
e29d1b70 374{
d9f2caf3 375 void *p = io_u_verify_off(hdr, io_u, header_num);
546dfd9f 376 struct vhdr_md5 *vh = hdr_priv(hdr);
8c432325
JA
377 uint32_t hash[MD5_HASH_WORDS];
378 struct md5_ctx md5_ctx = {
379 .hash = hash,
380 };
e29d1b70 381
bd6f78b2
JA
382 dprint(FD_VERIFY, "md5 verify io_u %p, len %u\n", io_u, hdr->len);
383
61f821f1 384 md5_init(&md5_ctx);
87677832 385 md5_update(&md5_ctx, p, hdr->len - hdr_size(hdr));
e29d1b70 386
546dfd9f 387 if (memcmp(vh->md5_digest, md5_ctx.hash, sizeof(hash))) {
3f9f4e26 388 log_err("md5: verify failed at %llu/%u\n",
5ec10eaa 389 io_u->offset + header_num * hdr->len, hdr->len);
546dfd9f 390 hexdump(vh->md5_digest, sizeof(vh->md5_digest));
8c432325 391 hexdump(md5_ctx.hash, sizeof(hash));
9fd18969 392 return EILSEQ;
e29d1b70
JA
393 }
394
395 return 0;
396}
397
3f199b01
JA
398static unsigned int hweight8(unsigned int w)
399{
400 unsigned int res = w - ((w >> 1) & 0x55);
401
402 res = (res & 0x33) + ((res >> 2) & 0x33);
403 return (res + (res >> 4)) & 0x0F;
404}
405
0e92f873 406int verify_io_u_pattern(char *pattern, unsigned long pattern_size,
5ec10eaa 407 char *buf, unsigned int len, unsigned int mod)
a944e335
SL
408{
409 unsigned int i;
a944e335
SL
410
411 for (i = 0; i < len; i++) {
0e92f873 412 if (buf[i] != pattern[mod]) {
3f199b01
JA
413 unsigned int bits;
414
0e92f873 415 bits = hweight8(buf[i] ^ pattern[mod]);
3f199b01 416 log_err("fio: got pattern %x, wanted %x. Bad bits %d\n",
0e92f873 417 buf[i], pattern[mod], bits);
3f199b01 418 log_err("fio: bad pattern block offset %u\n", i);
9fd18969 419 return EILSEQ;
3f199b01 420 }
a944e335
SL
421 mod++;
422 if (mod == pattern_size)
423 mod = 0;
424 }
425
426 return 0;
427}
428
e8462bd8
JA
429/*
430 * Push IO verification to a separate thread
431 */
432int verify_io_u_async(struct thread_data *td, struct io_u *io_u)
433{
434 if (io_u->file)
435 put_file_log(td, io_u->file);
436
437 io_u->file = NULL;
438
439 pthread_mutex_lock(&td->io_u_lock);
0c41214f
RR
440
441 if (io_u->flags & IO_U_F_IN_CUR_DEPTH) {
442 td->cur_depth--;
443 io_u->flags &= ~IO_U_F_IN_CUR_DEPTH;
444 }
e8462bd8
JA
445 flist_del(&io_u->list);
446 flist_add_tail(&io_u->list, &td->verify_list);
2ecc1b57 447 io_u->flags |= IO_U_F_FREE_DEF;
e8462bd8
JA
448 pthread_mutex_unlock(&td->io_u_lock);
449
450 pthread_cond_signal(&td->verify_cond);
e8462bd8
JA
451 return 0;
452}
453
36690c9b 454int verify_io_u(struct thread_data *td, struct io_u *io_u)
e29d1b70 455{
3f9f4e26 456 struct verify_header *hdr;
a944e335 457 unsigned int hdr_size, hdr_inc, hdr_num = 0;
95646108 458 void *p;
e29d1b70
JA
459 int ret;
460
1dcc0498 461 if (td->o.verify == VERIFY_NULL || io_u->ddir != DDIR_READ)
36690c9b
JA
462 return 0;
463
3f9f4e26 464 hdr_inc = io_u->buflen;
a59e170d
JA
465 if (td->o.verify_interval)
466 hdr_inc = td->o.verify_interval;
e29d1b70 467
a12a3b4d 468 ret = 0;
5ec10eaa
JA
469 for (p = io_u->buf; p < io_u->buf + io_u->buflen;
470 p += hdr_inc, hdr_num++) {
a12a3b4d
JA
471 if (ret && td->o.verify_fatal) {
472 td->terminate = 1;
473 break;
474 }
a944e335 475 hdr_size = __hdr_size(td->o.verify);
a59e170d 476 if (td->o.verify_offset)
a944e335 477 memswp(p, p + td->o.verify_offset, hdr_size);
95646108 478 hdr = p;
e29d1b70 479
3f199b01
JA
480 if (hdr->fio_magic != FIO_HDR_MAGIC) {
481 log_err("Bad verify header %x\n", hdr->fio_magic);
9fd18969 482 return EILSEQ;
3f199b01
JA
483 }
484
e28218f3 485 if (td->o.verify_pattern_bytes) {
bd6f78b2
JA
486 dprint(FD_VERIFY, "pattern verify io_u %p, len %u\n",
487 io_u, hdr->len);
e28218f3 488 ret = verify_io_u_pattern(td->o.verify_pattern,
0e92f873
RR
489 td->o.verify_pattern_bytes,
490 p + hdr_size,
491 hdr_inc - hdr_size,
492 hdr_size % td->o.verify_pattern_bytes);
e92d3d71
RR
493 /*
494 * Also verify the meta data, if applicable
495 */
496 if (hdr->verify_type == VERIFY_META)
497 ret |= verify_io_u_meta(hdr, td, io_u, hdr_num);
498
e28218f3
SL
499 if (ret)
500 log_err("fio: verify failed at %llu/%u\n",
501 io_u->offset + hdr_num * hdr->len,
502 hdr->len);
503 continue;
504 }
505
3f9f4e26
SL
506 switch (hdr->verify_type) {
507 case VERIFY_MD5:
508 ret = verify_io_u_md5(hdr, io_u, hdr_num);
509 break;
510 case VERIFY_CRC64:
511 ret = verify_io_u_crc64(hdr, io_u, hdr_num);
512 break;
bac39e0e 513 case VERIFY_CRC32C:
3845591f 514 case VERIFY_CRC32C_INTEL:
bac39e0e
JA
515 ret = verify_io_u_crc32c(hdr, io_u, hdr_num);
516 break;
3f9f4e26
SL
517 case VERIFY_CRC32:
518 ret = verify_io_u_crc32(hdr, io_u, hdr_num);
519 break;
520 case VERIFY_CRC16:
521 ret = verify_io_u_crc16(hdr, io_u, hdr_num);
522 break;
523 case VERIFY_CRC7:
524 ret = verify_io_u_crc7(hdr, io_u, hdr_num);
525 break;
cd14cc10
JA
526 case VERIFY_SHA256:
527 ret = verify_io_u_sha256(hdr, io_u, hdr_num);
528 break;
529 case VERIFY_SHA512:
530 ret = verify_io_u_sha512(hdr, io_u, hdr_num);
531 break;
7437ee87
SL
532 case VERIFY_META:
533 ret = verify_io_u_meta(hdr, td, io_u, hdr_num);
534 break;
7c353ceb
JA
535 case VERIFY_SHA1:
536 ret = verify_io_u_sha1(hdr, io_u, hdr_num);
537 break;
3f9f4e26
SL
538 default:
539 log_err("Bad verify type %u\n", hdr->verify_type);
d16d4e09 540 ret = EINVAL;
3f9f4e26 541 }
3f9f4e26 542 }
a7dfe862 543
a12a3b4d 544 return ret;
e29d1b70
JA
545}
546
7437ee87 547static void fill_meta(struct verify_header *hdr, struct thread_data *td,
5ec10eaa 548 struct io_u *io_u, unsigned int header_num)
7437ee87
SL
549{
550 struct vhdr_meta *vh = hdr_priv(hdr);
551
552 vh->thread = td->thread_number;
553
554 vh->time_sec = io_u->start_time.tv_sec;
555 vh->time_usec = io_u->start_time.tv_usec;
556
557 vh->numberio = td->io_issues[DDIR_WRITE];
558
559 vh->offset = io_u->offset + header_num * td->o.verify_interval;
560}
561
cd14cc10
JA
562static void fill_sha512(struct verify_header *hdr, void *p, unsigned int len)
563{
546dfd9f 564 struct vhdr_sha512 *vh = hdr_priv(hdr);
cd14cc10 565 struct sha512_ctx sha512_ctx = {
546dfd9f 566 .buf = vh->sha512,
cd14cc10
JA
567 };
568
569 sha512_init(&sha512_ctx);
570 sha512_update(&sha512_ctx, p, len);
571}
572
573static void fill_sha256(struct verify_header *hdr, void *p, unsigned int len)
574{
546dfd9f 575 struct vhdr_sha256 *vh = hdr_priv(hdr);
cd14cc10 576 struct sha256_ctx sha256_ctx = {
546dfd9f 577 .buf = vh->sha256,
cd14cc10
JA
578 };
579
580 sha256_init(&sha256_ctx);
581 sha256_update(&sha256_ctx, p, len);
582}
583
7c353ceb
JA
584static void fill_sha1(struct verify_header *hdr, void *p, unsigned int len)
585{
586 struct vhdr_sha1 *vh = hdr_priv(hdr);
587 struct sha1_ctx sha1_ctx = {
588 .H = vh->sha1,
589 };
590
591 sha1_init(&sha1_ctx);
592 sha1_update(&sha1_ctx, p, len);
593}
594
1e154bdb
JA
595static void fill_crc7(struct verify_header *hdr, void *p, unsigned int len)
596{
546dfd9f
JA
597 struct vhdr_crc7 *vh = hdr_priv(hdr);
598
599 vh->crc7 = crc7(p, len);
1e154bdb
JA
600}
601
969f7ed3
JA
602static void fill_crc16(struct verify_header *hdr, void *p, unsigned int len)
603{
546dfd9f
JA
604 struct vhdr_crc16 *vh = hdr_priv(hdr);
605
606 vh->crc16 = crc16(p, len);
969f7ed3
JA
607}
608
e29d1b70
JA
609static void fill_crc32(struct verify_header *hdr, void *p, unsigned int len)
610{
546dfd9f
JA
611 struct vhdr_crc32 *vh = hdr_priv(hdr);
612
613 vh->crc32 = crc32(p, len);
e29d1b70
JA
614}
615
bac39e0e
JA
616static void fill_crc32c(struct verify_header *hdr, void *p, unsigned int len)
617{
618 struct vhdr_crc32 *vh = hdr_priv(hdr);
619
3845591f
JA
620 if (hdr->verify_type == VERIFY_CRC32C_INTEL)
621 vh->crc32 = crc32c_intel(p, len);
622 else
623 vh->crc32 = crc32c(p, len);
bac39e0e
JA
624}
625
d77a7af3
JA
626static void fill_crc64(struct verify_header *hdr, void *p, unsigned int len)
627{
546dfd9f
JA
628 struct vhdr_crc64 *vh = hdr_priv(hdr);
629
630 vh->crc64 = crc64(p, len);
d77a7af3
JA
631}
632
e29d1b70
JA
633static void fill_md5(struct verify_header *hdr, void *p, unsigned int len)
634{
546dfd9f 635 struct vhdr_md5 *vh = hdr_priv(hdr);
8c432325 636 struct md5_ctx md5_ctx = {
546dfd9f 637 .hash = (uint32_t *) vh->md5_digest,
8c432325 638 };
e29d1b70 639
61f821f1 640 md5_init(&md5_ctx);
e29d1b70 641 md5_update(&md5_ctx, p, len);
e29d1b70
JA
642}
643
644/*
645 * fill body of io_u->buf with random data and add a header with the
646 * crc32 or md5 sum of that data.
647 */
648void populate_verify_io_u(struct thread_data *td, struct io_u *io_u)
649{
baefa9be 650 struct verify_header *hdr;
95646108 651 void *p = io_u->buf, *data;
7437ee87 652 unsigned int hdr_inc, data_len, header_num = 0;
e29d1b70 653
9cc3d150
JA
654 if (td->o.verify == VERIFY_NULL)
655 return;
656
90059d65 657 fill_pattern(td, p, io_u->buflen);
546a9142
SL
658
659 hdr_inc = io_u->buflen;
a59e170d
JA
660 if (td->o.verify_interval)
661 hdr_inc = td->o.verify_interval;
baefa9be 662
5ec10eaa 663 for (; p < io_u->buf + io_u->buflen; p += hdr_inc) {
95646108 664 hdr = p;
3f9f4e26
SL
665
666 hdr->fio_magic = FIO_HDR_MAGIC;
667 hdr->verify_type = td->o.verify;
546a9142 668 hdr->len = hdr_inc;
87677832 669 data_len = hdr_inc - hdr_size(hdr);
3f9f4e26 670
87677832 671 data = p + hdr_size(hdr);
3f9f4e26
SL
672 switch (td->o.verify) {
673 case VERIFY_MD5:
bd6f78b2
JA
674 dprint(FD_VERIFY, "fill md5 io_u %p, len %u\n",
675 io_u, hdr->len);
3f9f4e26
SL
676 fill_md5(hdr, data, data_len);
677 break;
678 case VERIFY_CRC64:
bd6f78b2
JA
679 dprint(FD_VERIFY, "fill crc64 io_u %p, len %u\n",
680 io_u, hdr->len);
3f9f4e26
SL
681 fill_crc64(hdr, data, data_len);
682 break;
bac39e0e 683 case VERIFY_CRC32C:
3845591f 684 case VERIFY_CRC32C_INTEL:
bac39e0e
JA
685 dprint(FD_VERIFY, "fill crc32c io_u %p, len %u\n",
686 io_u, hdr->len);
687 fill_crc32c(hdr, data, data_len);
688 break;
3f9f4e26 689 case VERIFY_CRC32:
bd6f78b2
JA
690 dprint(FD_VERIFY, "fill crc32 io_u %p, len %u\n",
691 io_u, hdr->len);
3f9f4e26
SL
692 fill_crc32(hdr, data, data_len);
693 break;
694 case VERIFY_CRC16:
bd6f78b2
JA
695 dprint(FD_VERIFY, "fill crc16 io_u %p, len %u\n",
696 io_u, hdr->len);
3f9f4e26
SL
697 fill_crc16(hdr, data, data_len);
698 break;
699 case VERIFY_CRC7:
bd6f78b2
JA
700 dprint(FD_VERIFY, "fill crc7 io_u %p, len %u\n",
701 io_u, hdr->len);
3f9f4e26
SL
702 fill_crc7(hdr, data, data_len);
703 break;
cd14cc10 704 case VERIFY_SHA256:
bd6f78b2
JA
705 dprint(FD_VERIFY, "fill sha256 io_u %p, len %u\n",
706 io_u, hdr->len);
cd14cc10
JA
707 fill_sha256(hdr, data, data_len);
708 break;
709 case VERIFY_SHA512:
bd6f78b2
JA
710 dprint(FD_VERIFY, "fill sha512 io_u %p, len %u\n",
711 io_u, hdr->len);
cd14cc10
JA
712 fill_sha512(hdr, data, data_len);
713 break;
7437ee87 714 case VERIFY_META:
bd6f78b2
JA
715 dprint(FD_VERIFY, "fill meta io_u %p, len %u\n",
716 io_u, hdr->len);
7437ee87
SL
717 fill_meta(hdr, td, io_u, header_num);
718 break;
7c353ceb
JA
719 case VERIFY_SHA1:
720 dprint(FD_VERIFY, "fill sha1 io_u %p, len %u\n",
721 io_u, hdr->len);
722 fill_sha1(hdr, data, data_len);
723 break;
3f9f4e26
SL
724 default:
725 log_err("fio: bad verify type: %d\n", td->o.verify);
726 assert(0);
727 }
a59e170d 728 if (td->o.verify_offset)
87677832 729 memswp(p, p + td->o.verify_offset, hdr_size(hdr));
7437ee87 730 header_num++;
e29d1b70 731 }
e29d1b70
JA
732}
733
734int get_next_verify(struct thread_data *td, struct io_u *io_u)
735{
8de8f047 736 struct io_piece *ipo = NULL;
e29d1b70 737
d2d7fa53
JA
738 /*
739 * this io_u is from a requeue, we already filled the offsets
740 */
741 if (io_u->file)
742 return 0;
743
8de8f047
JA
744 if (!RB_EMPTY_ROOT(&td->io_hist_tree)) {
745 struct rb_node *n = rb_first(&td->io_hist_tree);
e29d1b70 746
8de8f047 747 ipo = rb_entry(n, struct io_piece, rb_node);
4b87898e 748 rb_erase(n, &td->io_hist_tree);
01743ee1
JA
749 } else if (!flist_empty(&td->io_hist_list)) {
750 ipo = flist_entry(td->io_hist_list.next, struct io_piece, list);
751 flist_del(&ipo->list);
8de8f047 752 }
e29d1b70 753
8de8f047 754 if (ipo) {
e29d1b70
JA
755 io_u->offset = ipo->offset;
756 io_u->buflen = ipo->len;
36167d82 757 io_u->file = ipo->file;
97af62ce 758
d6aed795 759 if (!fio_file_open(io_u->file)) {
97af62ce
JA
760 int r = td_io_open_file(td, io_u->file);
761
bd6f78b2
JA
762 if (r) {
763 dprint(FD_VERIFY, "failed file %s open\n",
764 io_u->file->file_name);
97af62ce 765 return 1;
bd6f78b2 766 }
97af62ce
JA
767 }
768
769 get_file(ipo->file);
d6aed795 770 assert(fio_file_open(io_u->file));
e29d1b70 771 io_u->ddir = DDIR_READ;
36167d82
JA
772 io_u->xfer_buf = io_u->buf;
773 io_u->xfer_buflen = io_u->buflen;
e29d1b70 774 free(ipo);
bd6f78b2 775 dprint(FD_VERIFY, "get_next_verify: ret io_u %p\n", io_u);
e29d1b70
JA
776 return 0;
777 }
778
bd6f78b2 779 dprint(FD_VERIFY, "get_next_verify: empty\n");
e29d1b70
JA
780 return 1;
781}
e8462bd8
JA
782
783static void *verify_async_thread(void *data)
784{
785 struct thread_data *td = data;
786 struct io_u *io_u;
787 int ret = 0;
788
789 if (td->o.verify_cpumask_set &&
790 fio_setaffinity(td->pid, td->o.verify_cpumask)) {
791 log_err("fio: failed setting verify thread affinity\n");
792 goto done;
793 }
794
795 do {
e53ab27c
JA
796 FLIST_HEAD(list);
797
e8462bd8
JA
798 read_barrier();
799 if (td->verify_thread_exit)
800 break;
801
802 pthread_mutex_lock(&td->io_u_lock);
803
804 while (flist_empty(&td->verify_list) &&
805 !td->verify_thread_exit) {
b36e298b
JA
806 ret = pthread_cond_wait(&td->verify_cond,
807 &td->io_u_lock);
e8462bd8
JA
808 if (ret) {
809 pthread_mutex_unlock(&td->io_u_lock);
810 break;
811 }
812 }
813
e53ab27c
JA
814 flist_splice_init(&td->verify_list, &list);
815 pthread_mutex_unlock(&td->io_u_lock);
816
817 if (flist_empty(&list))
e8462bd8 818 continue;
e8462bd8 819
e53ab27c
JA
820 while (!flist_empty(&list)) {
821 io_u = flist_entry(list.next, struct io_u, list);
822 flist_del_init(&io_u->list);
e8462bd8 823
d561f2ab 824 ret = verify_io_u(td, io_u);
e53ab27c 825 put_io_u(td, io_u);
d561f2ab
JA
826 if (!ret)
827 continue;
828 if (td->o.continue_on_error &&
829 td_non_fatal_error(ret)) {
830 update_error_count(td, ret);
831 td_clear_error(td);
832 ret = 0;
833 }
e53ab27c 834 }
e8462bd8
JA
835 } while (!ret);
836
d561f2ab
JA
837 if (ret) {
838 td_verror(td, ret, "async_verify");
839 td->terminate = 1;
840 }
841
e8462bd8
JA
842done:
843 pthread_mutex_lock(&td->io_u_lock);
844 td->nr_verify_threads--;
845 pthread_mutex_unlock(&td->io_u_lock);
846
847 pthread_cond_signal(&td->free_cond);
848 return NULL;
849}
850
851int verify_async_init(struct thread_data *td)
852{
853 int i, ret;
854
855 td->verify_thread_exit = 0;
856
857 td->verify_threads = malloc(sizeof(pthread_t) * td->o.verify_async);
858 for (i = 0; i < td->o.verify_async; i++) {
859 ret = pthread_create(&td->verify_threads[i], NULL,
860 verify_async_thread, td);
861 if (ret) {
862 log_err("fio: async verify creation failed: %s\n",
863 strerror(ret));
864 break;
865 }
866 ret = pthread_detach(td->verify_threads[i]);
867 if (ret) {
868 log_err("fio: async verify thread detach failed: %s\n",
869 strerror(ret));
870 break;
871 }
872 td->nr_verify_threads++;
873 }
874
875 if (i != td->o.verify_async) {
e40823b1 876 log_err("fio: only %d verify threads started, exiting\n", i);
e8462bd8
JA
877 td->verify_thread_exit = 1;
878 write_barrier();
879 pthread_cond_broadcast(&td->verify_cond);
880 return 1;
881 }
882
883 return 0;
884}
885
886void verify_async_exit(struct thread_data *td)
887{
888 td->verify_thread_exit = 1;
889 write_barrier();
890 pthread_cond_broadcast(&td->verify_cond);
891
892 pthread_mutex_lock(&td->io_u_lock);
893
894 while (td->nr_verify_threads)
895 pthread_cond_wait(&td->free_cond, &td->io_u_lock);
896
897 pthread_mutex_unlock(&td->io_u_lock);
898 free(td->verify_threads);
899 td->verify_threads = NULL;
900}