[fio.git] / verify.c

/*
 * IO verification helpers
 */
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <assert.h>

#include "fio.h"

#include "md5.h"
#include "crc64.h"
#include "crc32.h"
#include "crc16.h"
#include "crc7.h"
#include "sha256.h"
#include "sha512.h"

static void fill_random_bytes(struct thread_data *td, void *p, unsigned int len)
{
	unsigned int todo;
	int r;

	while (len) {
		r = os_random_long(&td->verify_state);

		/*
		 * lrand48_r seems to be broken and only fill the bottom
		 * 32-bits, even on 64-bit archs with 64-bit longs
		 */
		todo = sizeof(r);
		if (todo > len)
			todo = len;

		memcpy(p, &r, todo);

		len -= todo;
		p += todo;
	}
}

static void fill_pattern(struct thread_data *td, void *p, unsigned int len)
{
	switch (td->o.verify_pattern_bytes) {
	case 0:
		fill_random_bytes(td, p, len);
		break;
	case 1:
		memset(p, td->o.verify_pattern, len);
		break;
	case 2:
	case 3:
	case 4: {
		unsigned int pattern = td->o.verify_pattern;
		unsigned int i = 0;
		unsigned char c1, c2, c3, c4;
		unsigned char *b = p;

		c1 = pattern & 0xff;
		pattern >>= 8;
		c2 = pattern & 0xff;
		pattern >>= 8;
		c3 = pattern & 0xff;
		pattern >>= 8;
		c4 = pattern & 0xff;

		while (i < len) {
			b[i++] = c1;
			if (i == len)
				break;
			b[i++] = c2;
			if (td->o.verify_pattern_bytes == 2 || i == len)
				continue;
			b[i++] = c3;
			if (td->o.verify_pattern_bytes == 3 || i == len)
				continue;
			b[i++] = c4;
		}
		break;
		}
	}
}

static void memswp(void* buf1, void* buf2, unsigned int len)
{
	struct verify_header swap;

	memcpy(&swap, buf1, len);
	memcpy(buf1, buf2, len);
	memcpy(buf2, &swap, len);
}

static void hexdump(void *buffer, int len)
{
	unsigned char *p = buffer;
	int i;

	for (i = 0; i < len; i++)
		log_info("%02x", p[i]);
	log_info("\n");
}

/*
 * Prepare for seperation of verify_header and checksum header
 */
static inline unsigned int __hdr_size(int fio_unused verify_type)
{
	return sizeof(struct verify_header);

}

static inline unsigned int hdr_size(struct verify_header *hdr)
{
	return sizeof(*hdr);
}

/*
 * Return data area 'header_num'
 */
static inline void *io_u_verify_off(struct verify_header *hdr,
				    struct io_u *io_u,
				    unsigned char header_num)
{
	return io_u->buf + header_num * hdr->len + hdr_size(hdr);
}

static int verify_io_u_sha512(struct verify_header *hdr, struct io_u *io_u,
			      unsigned int header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	uint8_t sha512[128];
	struct sha512_ctx sha512_ctx = {
		.buf = sha512,
	};

	sha512_init(&sha512_ctx);
	sha512_update(&sha512_ctx, p, hdr->len - hdr_size(hdr));

	if (memcmp(hdr->sha512, sha512_ctx.buf, sizeof(sha512))) {
		log_err("sha512: verify failed at %llu/%u\n",
		              io_u->offset + header_num * hdr->len,
		              hdr->len);
		hexdump(hdr->sha512, sizeof(hdr->sha512));
		hexdump(sha512_ctx.buf, sizeof(sha512));
		return 1;
	}

	return 0;
}

static int verify_io_u_sha256(struct verify_header *hdr, struct io_u *io_u,
			      unsigned int header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	uint8_t sha256[128];
	struct sha256_ctx sha256_ctx = {
		.buf = sha256,
	};

	sha256_init(&sha256_ctx);
	sha256_update(&sha256_ctx, p, hdr->len - hdr_size(hdr));

	if (memcmp(hdr->sha256, sha256_ctx.buf, sizeof(sha256))) {
		log_err("sha256: verify failed at %llu/%u\n",
		              io_u->offset + header_num * hdr->len,
		              hdr->len);
		hexdump(hdr->sha256, sizeof(hdr->sha256));
		hexdump(sha256_ctx.buf, sizeof(sha256));
		return 1;
	}

	return 0;
}

static int verify_io_u_crc7(struct verify_header *hdr, struct io_u *io_u,
                            unsigned char header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	unsigned char c;

	c = crc7(p, hdr->len - hdr_size(hdr));

	if (c != hdr->crc7) {
		log_err("crc7: verify failed at %llu/%u\n",
		                io_u->offset + header_num * hdr->len,
				hdr->len);
		log_err("crc7: wanted %x, got %x\n", hdr->crc7, c);
		return 1;
	}

	return 0;
}

static int verify_io_u_crc16(struct verify_header *hdr, struct io_u *io_u,
                             unsigned int header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	unsigned short c;

	c = crc16(p, hdr->len - hdr_size(hdr));

	if (c != hdr->crc16) {
		log_err("crc16: verify failed at %llu/%u\n",
		                io_u->offset + header_num * hdr->len,
				hdr->len);
		log_err("crc16: wanted %x, got %x\n", hdr->crc16, c);
		return 1;
	}

	return 0;
}

static int verify_io_u_crc64(struct verify_header *hdr, struct io_u *io_u,
                             unsigned int header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	unsigned long long c;

	c = crc64(p, hdr->len - hdr_size(hdr));

	if (c != hdr->crc64) {
		log_err("crc64: verify failed at %llu/%u\n",
				io_u->offset + header_num * hdr->len,
				hdr->len);
		log_err("crc64: wanted %llx, got %llx\n", hdr->crc64, c);
		return 1;
	}

	return 0;
}

static int verify_io_u_crc32(struct verify_header *hdr, struct io_u *io_u,
		             unsigned int header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	unsigned long c;

	c = crc32(p, hdr->len - hdr_size(hdr));

	if (c != hdr->crc32) {
		log_err("crc32: verify failed at %llu/%u\n",
		                io_u->offset + header_num * hdr->len,
				hdr->len);
		log_err("crc32: wanted %lx, got %lx\n", hdr->crc32, c);
		return 1;
	}

	return 0;
}

static int verify_io_u_md5(struct verify_header *hdr, struct io_u *io_u,
		           unsigned int header_num)
{
	void *p = io_u_verify_off(hdr, io_u, header_num);
	uint32_t hash[MD5_HASH_WORDS];
	struct md5_ctx md5_ctx = {
		.hash = hash,
	};

	md5_init(&md5_ctx);
	md5_update(&md5_ctx, p, hdr->len - hdr_size(hdr));

	if (memcmp(hdr->md5_digest, md5_ctx.hash, sizeof(hash))) {
		log_err("md5: verify failed at %llu/%u\n",
		              io_u->offset + header_num * hdr->len,
		              hdr->len);
		hexdump(hdr->md5_digest, sizeof(hdr->md5_digest));
		hexdump(md5_ctx.hash, sizeof(hash));
		return 1;
	}

	return 0;
}

int verify_io_u(struct thread_data *td, struct io_u *io_u)
{
	struct verify_header *hdr;
	unsigned int hdr_inc, hdr_num = 0;
	void *p;
	int ret;

	if (td->o.verify == VERIFY_NULL || io_u->ddir != DDIR_READ)
		return 0;

	hdr_inc = io_u->buflen;
	if (td->o.verify_interval)
		hdr_inc = td->o.verify_interval;

	for (p = io_u->buf; p < io_u->buf + io_u->buflen; p += hdr_inc) {
		if (td->o.verify_offset)
			memswp(p, p + td->o.verify_offset, __hdr_size(hdr->verify_type));

		hdr = p;

		if (hdr->fio_magic != FIO_HDR_MAGIC) {
			log_err("Bad verify header %x\n", hdr->fio_magic);
			return EIO;
		}

		switch (hdr->verify_type) {
		case VERIFY_MD5:
			ret = verify_io_u_md5(hdr, io_u, hdr_num);
			break;
		case VERIFY_CRC64:
			ret = verify_io_u_crc64(hdr, io_u, hdr_num);
			break;
		case VERIFY_CRC32:
			ret = verify_io_u_crc32(hdr, io_u, hdr_num);
			break;
		case VERIFY_CRC16:
			ret = verify_io_u_crc16(hdr, io_u, hdr_num);
			break;
		case VERIFY_CRC7:
			ret = verify_io_u_crc7(hdr, io_u, hdr_num);
			break;
		case VERIFY_SHA256:
			ret = verify_io_u_sha256(hdr, io_u, hdr_num);
			break;
		case VERIFY_SHA512:
			ret = verify_io_u_sha512(hdr, io_u, hdr_num);
			break;
		default:
			log_err("Bad verify type %u\n", hdr->verify_type);
			ret = 1;
		}
		hdr_num++;
	}

	return 0;
}

static void fill_sha512(struct verify_header *hdr, void *p, unsigned int len)
{
	struct sha512_ctx sha512_ctx = {
		.buf = hdr->sha512,
	};

	sha512_init(&sha512_ctx);
	sha512_update(&sha512_ctx, p, len);
}

static void fill_sha256(struct verify_header *hdr, void *p, unsigned int len)
{
	struct sha256_ctx sha256_ctx = {
		.buf = hdr->sha256,
	};

	sha256_init(&sha256_ctx);
	sha256_update(&sha256_ctx, p, len);
}

static void fill_crc7(struct verify_header *hdr, void *p, unsigned int len)
{
	hdr->crc7 = crc7(p, len);
}

static void fill_crc16(struct verify_header *hdr, void *p, unsigned int len)
{
	hdr->crc16 = crc16(p, len);
}

static void fill_crc32(struct verify_header *hdr, void *p, unsigned int len)
{
	hdr->crc32 = crc32(p, len);
}

static void fill_crc64(struct verify_header *hdr, void *p, unsigned int len)
{
	hdr->crc64 = crc64(p, len);
}

static void fill_md5(struct verify_header *hdr, void *p, unsigned int len)
{
	struct md5_ctx md5_ctx = {
		.hash = (uint32_t *) hdr->md5_digest,
	};

	md5_init(&md5_ctx);
	md5_update(&md5_ctx, p, len);
}

/*
 * fill body of io_u->buf with random data and add a header with the
 * crc32 or md5 sum of that data.
 */
void populate_verify_io_u(struct thread_data *td, struct io_u *io_u)
{
	struct verify_header *hdr;
	void *p = io_u->buf, *data;
	unsigned int hdr_inc, data_len;

	if (td->o.verify == VERIFY_NULL)
		return;

	fill_pattern(td, p, io_u->buflen);

	hdr_inc = io_u->buflen;
	if (td->o.verify_interval)
		hdr_inc = td->o.verify_interval;

	for (;p < io_u->buf + io_u->buflen; p += hdr_inc) {
		hdr = p;

		hdr->fio_magic = FIO_HDR_MAGIC;
		hdr->verify_type = td->o.verify;
		hdr->len = hdr_inc;
		data_len = hdr_inc - hdr_size(hdr);

		data = p + hdr_size(hdr);
		switch (td->o.verify) {
		case VERIFY_MD5:
			fill_md5(hdr, data, data_len);
			break;
		case VERIFY_CRC64:
			fill_crc64(hdr, data, data_len);
			break;
		case VERIFY_CRC32:
			fill_crc32(hdr, data, data_len);
			break;
		case VERIFY_CRC16:
			fill_crc16(hdr, data, data_len);
			break;
		case VERIFY_CRC7:
			fill_crc7(hdr, data, data_len);
			break;
		case VERIFY_SHA256:
			fill_sha256(hdr, data, data_len);
			break;
		case VERIFY_SHA512:
			fill_sha512(hdr, data, data_len);
			break;
		default:
			log_err("fio: bad verify type: %d\n", td->o.verify);
			assert(0);
		}
		if (td->o.verify_offset)
			memswp(p, p + td->o.verify_offset, hdr_size(hdr));
	}
}

int get_next_verify(struct thread_data *td, struct io_u *io_u)
{
	struct io_piece *ipo = NULL;

	/*
	 * this io_u is from a requeue, we already filled the offsets
	 */
	if (io_u->file)
		return 0;

	if (!RB_EMPTY_ROOT(&td->io_hist_tree)) {
		struct rb_node *n = rb_first(&td->io_hist_tree);

		ipo = rb_entry(n, struct io_piece, rb_node);
		rb_erase(n, &td->io_hist_tree);
	} else if (!list_empty(&td->io_hist_list)) {
		ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
		list_del(&ipo->list);
	}

	if (ipo) {
		io_u->offset = ipo->offset;
		io_u->buflen = ipo->len;
		io_u->file = ipo->file;

		if ((io_u->file->flags & FIO_FILE_OPEN) == 0) {
			int r = td_io_open_file(td, io_u->file);

			if (r)
				return 1;
		}

		get_file(ipo->file);
		assert(io_u->file->flags & FIO_FILE_OPEN);
		io_u->ddir = DDIR_READ;
		io_u->xfer_buf = io_u->buf;
		io_u->xfer_buflen = io_u->buflen;
		free(ipo);
		return 0;
	}

	return 1;
}
Commit	Line	Data
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>
e29d1b70 JA	8
e29d1b70 JA	9	#include "fio.h"
e29d1b70	10
cd14cc10 JA	11	#include "md5.h"
	12	#include "crc64.h"
	13	#include "crc32.h"
	14	#include "crc16.h"
	15	#include "crc7.h"
	16	#include "sha256.h"
	17	#include "sha512.h"
	18
90059d65	19	static void fill_random_bytes(struct thread_data td, void p, unsigned int len)
e29d1b70 JA	20	{
e29d1b70 JA	21	unsigned int todo;
4c5946c6	22	int r;
e29d1b70 JA	23
e29d1b70 JA	24	while (len) {
4c5946c6	25	r = os_random_long(&td->verify_state);
e29d1b70 JA	26
	27	/*
	28	* lrand48_r seems to be broken and only fill the bottom
	29	* 32-bits, even on 64-bit archs with 64-bit longs
	30	*/
	31	todo = sizeof(r);
	32	if (todo > len)
	33	todo = len;
	34
	35	memcpy(p, &r, todo);
	36
	37	len -= todo;
	38	p += todo;
	39	}
	40	}
	41
90059d65 JA	42	static void fill_pattern(struct thread_data td, void p, unsigned int len)
	43	{
	44	switch (td->o.verify_pattern_bytes) {
	45	case 0:
	46	fill_random_bytes(td, p, len);
	47	break;
	48	case 1:
	49	memset(p, td->o.verify_pattern, len);
	50	break;
	51	case 2:
	52	case 3:
	53	case 4: {
	54	unsigned int pattern = td->o.verify_pattern;
	55	unsigned int i = 0;
	56	unsigned char c1, c2, c3, c4;
	57	unsigned char *b = p;
	58
	59	c1 = pattern & 0xff;
	60	pattern >>= 8;
	61	c2 = pattern & 0xff;
	62	pattern >>= 8;
	63	c3 = pattern & 0xff;
	64	pattern >>= 8;
	65	c4 = pattern & 0xff;
	66
	67	while (i < len) {
	68	b[i++] = c1;
	69	if (i == len)
	70	break;
	71	b[i++] = c2;
	72	if (td->o.verify_pattern_bytes == 2 \|\| i == len)
	73	continue;
	74	b[i++] = c3;
	75	if (td->o.verify_pattern_bytes == 3 \|\| i == len)
	76	continue;
	77	b[i++] = c4;
	78	}
	79	break;
	80	}
	81	}
	82	}
	83
	84	static void memswp(void* buf1, void* buf2, unsigned int len)
546a9142 SL	85	{
546a9142 SL	86	struct verify_header swap;
90059d65	87
546a9142 SL	88	memcpy(&swap, buf1, len);
	89	memcpy(buf1, buf2, len);
	90	memcpy(buf2, &swap, len);
	91	}
	92
e29d1b70 JA	93	static void hexdump(void *buffer, int len)
	94	{
	95	unsigned char *p = buffer;
	96	int i;
	97
	98	for (i = 0; i < len; i++)
6d86144d JA	99	log_info("%02x", p[i]);
6d86144d JA	100	log_info("\n");
e29d1b70 JA	101	}
e29d1b70 JA	102
87677832 JA	103	/*
	104	* Prepare for seperation of verify_header and checksum header
	105	*/
	106	static inline unsigned int __hdr_size(int fio_unused verify_type)
	107	{
	108	return sizeof(struct verify_header);
	109
	110	}
	111
	112	static inline unsigned int hdr_size(struct verify_header *hdr)
	113	{
	114	return sizeof(*hdr);
	115	}
	116
d9f2caf3 JA	117	/*
	118	* Return data area 'header_num'
	119	*/
	120	static inline void io_u_verify_off(struct verify_header hdr,
	121	struct io_u *io_u,
	122	unsigned char header_num)
	123	{
87677832	124	return io_u->buf + header_num * hdr->len + hdr_size(hdr);
d9f2caf3 JA	125	}
d9f2caf3 JA	126
cd14cc10 JA	127	static int verify_io_u_sha512(struct verify_header hdr, struct io_u io_u,
	128	unsigned int header_num)
	129	{
	130	void *p = io_u_verify_off(hdr, io_u, header_num);
	131	uint8_t sha512[128];
	132	struct sha512_ctx sha512_ctx = {
	133	.buf = sha512,
	134	};
	135
	136	sha512_init(&sha512_ctx);
87677832	137	sha512_update(&sha512_ctx, p, hdr->len - hdr_size(hdr));
cd14cc10 JA	138
	139	if (memcmp(hdr->sha512, sha512_ctx.buf, sizeof(sha512))) {
	140	log_err("sha512: verify failed at %llu/%u\n",
	141	io_u->offset + header_num * hdr->len,
	142	hdr->len);
	143	hexdump(hdr->sha512, sizeof(hdr->sha512));
	144	hexdump(sha512_ctx.buf, sizeof(sha512));
	145	return 1;
	146	}
	147
	148	return 0;
	149	}
	150
	151	static int verify_io_u_sha256(struct verify_header hdr, struct io_u io_u,
	152	unsigned int header_num)
	153	{
	154	void *p = io_u_verify_off(hdr, io_u, header_num);
	155	uint8_t sha256[128];
	156	struct sha256_ctx sha256_ctx = {
	157	.buf = sha256,
	158	};
	159
	160	sha256_init(&sha256_ctx);
87677832	161	sha256_update(&sha256_ctx, p, hdr->len - hdr_size(hdr));
cd14cc10 JA	162
	163	if (memcmp(hdr->sha256, sha256_ctx.buf, sizeof(sha256))) {
	164	log_err("sha256: verify failed at %llu/%u\n",
	165	io_u->offset + header_num * hdr->len,
	166	hdr->len);
	167	hexdump(hdr->sha256, sizeof(hdr->sha256));
	168	hexdump(sha256_ctx.buf, sizeof(sha256));
	169	return 1;
	170	}
	171
	172	return 0;
	173	}
	174
3f9f4e26 SL	175	static int verify_io_u_crc7(struct verify_header hdr, struct io_u io_u,
3f9f4e26 SL	176	unsigned char header_num)
1e154bdb	177	{
d9f2caf3	178	void *p = io_u_verify_off(hdr, io_u, header_num);
1e154bdb JA	179	unsigned char c;
1e154bdb JA	180
87677832	181	c = crc7(p, hdr->len - hdr_size(hdr));
1e154bdb JA	182
1e154bdb JA	183	if (c != hdr->crc7) {
3f9f4e26 SL	184	log_err("crc7: verify failed at %llu/%u\n",
	185	io_u->offset + header_num * hdr->len,
	186	hdr->len);
1e154bdb JA	187	log_err("crc7: wanted %x, got %x\n", hdr->crc7, c);
	188	return 1;
	189	}
	190
	191	return 0;
	192	}
	193
3f9f4e26 SL	194	static int verify_io_u_crc16(struct verify_header hdr, struct io_u io_u,
3f9f4e26 SL	195	unsigned int header_num)
969f7ed3	196	{
d9f2caf3	197	void *p = io_u_verify_off(hdr, io_u, header_num);
969f7ed3 JA	198	unsigned short c;
969f7ed3 JA	199
87677832	200	c = crc16(p, hdr->len - hdr_size(hdr));
969f7ed3 JA	201
969f7ed3 JA	202	if (c != hdr->crc16) {
3f9f4e26 SL	203	log_err("crc16: verify failed at %llu/%u\n",
	204	io_u->offset + header_num * hdr->len,
	205	hdr->len);
1f24ea44	206	log_err("crc16: wanted %x, got %x\n", hdr->crc16, c);
969f7ed3 JA	207	return 1;
	208	}
	209
	210	return 0;
	211	}
	212
3f9f4e26 SL	213	static int verify_io_u_crc64(struct verify_header hdr, struct io_u io_u,
3f9f4e26 SL	214	unsigned int header_num)
d77a7af3	215	{
d9f2caf3	216	void *p = io_u_verify_off(hdr, io_u, header_num);
d77a7af3 JA	217	unsigned long long c;
d77a7af3 JA	218
87677832	219	c = crc64(p, hdr->len - hdr_size(hdr));
d77a7af3 JA	220
d77a7af3 JA	221	if (c != hdr->crc64) {
3f9f4e26 SL	222	log_err("crc64: verify failed at %llu/%u\n",
	223	io_u->offset + header_num * hdr->len,
	224	hdr->len);
d77a7af3 JA	225	log_err("crc64: wanted %llx, got %llx\n", hdr->crc64, c);
	226	return 1;
	227	}
	228
	229	return 0;
	230	}
	231
3f9f4e26 SL	232	static int verify_io_u_crc32(struct verify_header hdr, struct io_u io_u,
3f9f4e26 SL	233	unsigned int header_num)
e29d1b70	234	{
d9f2caf3	235	void *p = io_u_verify_off(hdr, io_u, header_num);
e29d1b70 JA	236	unsigned long c;
e29d1b70 JA	237
87677832	238	c = crc32(p, hdr->len - hdr_size(hdr));
e29d1b70 JA	239
e29d1b70 JA	240	if (c != hdr->crc32) {
3f9f4e26 SL	241	log_err("crc32: verify failed at %llu/%u\n",
	242	io_u->offset + header_num * hdr->len,
	243	hdr->len);
e29d1b70 JA	244	log_err("crc32: wanted %lx, got %lx\n", hdr->crc32, c);
	245	return 1;
	246	}
	247
	248	return 0;
	249	}
	250
3f9f4e26 SL	251	static int verify_io_u_md5(struct verify_header hdr, struct io_u io_u,
3f9f4e26 SL	252	unsigned int header_num)
e29d1b70	253	{
d9f2caf3	254	void *p = io_u_verify_off(hdr, io_u, header_num);
8c432325 JA	255	uint32_t hash[MD5_HASH_WORDS];
	256	struct md5_ctx md5_ctx = {
	257	.hash = hash,
	258	};
e29d1b70	259
61f821f1	260	md5_init(&md5_ctx);
87677832	261	md5_update(&md5_ctx, p, hdr->len - hdr_size(hdr));
e29d1b70	262
c9b3fdf3	263	if (memcmp(hdr->md5_digest, md5_ctx.hash, sizeof(hash))) {
3f9f4e26 SL	264	log_err("md5: verify failed at %llu/%u\n",
	265	io_u->offset + header_num * hdr->len,
	266	hdr->len);
e29d1b70	267	hexdump(hdr->md5_digest, sizeof(hdr->md5_digest));
8c432325	268	hexdump(md5_ctx.hash, sizeof(hash));
e29d1b70 JA	269	return 1;
	270	}
	271
	272	return 0;
	273	}
	274
36690c9b	275	int verify_io_u(struct thread_data td, struct io_u io_u)
e29d1b70	276	{
3f9f4e26 SL	277	struct verify_header *hdr;
3f9f4e26 SL	278	unsigned int hdr_inc, hdr_num = 0;
95646108	279	void *p;
e29d1b70 JA	280	int ret;
e29d1b70 JA	281
1dcc0498	282	if (td->o.verify == VERIFY_NULL \|\| io_u->ddir != DDIR_READ)
36690c9b JA	283	return 0;
36690c9b JA	284
3f9f4e26	285	hdr_inc = io_u->buflen;
a59e170d JA	286	if (td->o.verify_interval)
a59e170d JA	287	hdr_inc = td->o.verify_interval;
e29d1b70	288
95646108	289	for (p = io_u->buf; p < io_u->buf + io_u->buflen; p += hdr_inc) {
a59e170d	290	if (td->o.verify_offset)
87677832	291	memswp(p, p + td->o.verify_offset, __hdr_size(hdr->verify_type));
546a9142	292
95646108	293	hdr = p;
e29d1b70	294
3f9f4e26 SL	295	if (hdr->fio_magic != FIO_HDR_MAGIC) {
	296	log_err("Bad verify header %x\n", hdr->fio_magic);
	297	return EIO;
	298	}
	299
	300	switch (hdr->verify_type) {
	301	case VERIFY_MD5:
	302	ret = verify_io_u_md5(hdr, io_u, hdr_num);
	303	break;
	304	case VERIFY_CRC64:
	305	ret = verify_io_u_crc64(hdr, io_u, hdr_num);
	306	break;
	307	case VERIFY_CRC32:
	308	ret = verify_io_u_crc32(hdr, io_u, hdr_num);
	309	break;
	310	case VERIFY_CRC16:
	311	ret = verify_io_u_crc16(hdr, io_u, hdr_num);
	312	break;
	313	case VERIFY_CRC7:
	314	ret = verify_io_u_crc7(hdr, io_u, hdr_num);
	315	break;
cd14cc10 JA	316	case VERIFY_SHA256:
	317	ret = verify_io_u_sha256(hdr, io_u, hdr_num);
	318	break;
	319	case VERIFY_SHA512:
	320	ret = verify_io_u_sha512(hdr, io_u, hdr_num);
	321	break;
3f9f4e26 SL	322	default:
	323	log_err("Bad verify type %u\n", hdr->verify_type);
	324	ret = 1;
	325	}
	326	hdr_num++;
	327	}
a7dfe862 JA	328
a7dfe862 JA	329	return 0;
e29d1b70 JA	330	}
e29d1b70 JA	331
cd14cc10 JA	332	static void fill_sha512(struct verify_header hdr, void p, unsigned int len)
	333	{
	334	struct sha512_ctx sha512_ctx = {
	335	.buf = hdr->sha512,
	336	};
	337
	338	sha512_init(&sha512_ctx);
	339	sha512_update(&sha512_ctx, p, len);
	340	}
	341
	342	static void fill_sha256(struct verify_header hdr, void p, unsigned int len)
	343	{
	344	struct sha256_ctx sha256_ctx = {
	345	.buf = hdr->sha256,
	346	};
	347
	348	sha256_init(&sha256_ctx);
	349	sha256_update(&sha256_ctx, p, len);
	350	}
	351
1e154bdb JA	352	static void fill_crc7(struct verify_header hdr, void p, unsigned int len)
	353	{
	354	hdr->crc7 = crc7(p, len);
	355	}
	356
969f7ed3 JA	357	static void fill_crc16(struct verify_header hdr, void p, unsigned int len)
	358	{
	359	hdr->crc16 = crc16(p, len);
	360	}
	361
e29d1b70 JA	362	static void fill_crc32(struct verify_header hdr, void p, unsigned int len)
	363	{
	364	hdr->crc32 = crc32(p, len);
	365	}
	366
d77a7af3 JA	367	static void fill_crc64(struct verify_header hdr, void p, unsigned int len)
	368	{
	369	hdr->crc64 = crc64(p, len);
	370	}
	371
e29d1b70 JA	372	static void fill_md5(struct verify_header hdr, void p, unsigned int len)
e29d1b70 JA	373	{
8c432325 JA	374	struct md5_ctx md5_ctx = {
	375	.hash = (uint32_t *) hdr->md5_digest,
	376	};
e29d1b70	377
61f821f1	378	md5_init(&md5_ctx);
e29d1b70	379	md5_update(&md5_ctx, p, len);
e29d1b70 JA	380	}
	381
	382	/*
	383	* fill body of io_u->buf with random data and add a header with the
	384	* crc32 or md5 sum of that data.
	385	*/
	386	void populate_verify_io_u(struct thread_data td, struct io_u io_u)
	387	{
baefa9be	388	struct verify_header *hdr;
95646108	389	void p = io_u->buf, data;
546a9142	390	unsigned int hdr_inc, data_len;
e29d1b70	391
9cc3d150 JA	392	if (td->o.verify == VERIFY_NULL)
	393	return;
	394
90059d65	395	fill_pattern(td, p, io_u->buflen);
546a9142 SL	396
546a9142 SL	397	hdr_inc = io_u->buflen;
a59e170d JA	398	if (td->o.verify_interval)
a59e170d JA	399	hdr_inc = td->o.verify_interval;
baefa9be	400
95646108 JA	401	for (;p < io_u->buf + io_u->buflen; p += hdr_inc) {
95646108 JA	402	hdr = p;
3f9f4e26 SL	403
	404	hdr->fio_magic = FIO_HDR_MAGIC;
	405	hdr->verify_type = td->o.verify;
546a9142	406	hdr->len = hdr_inc;
87677832	407	data_len = hdr_inc - hdr_size(hdr);
3f9f4e26	408
87677832	409	data = p + hdr_size(hdr);
3f9f4e26 SL	410	switch (td->o.verify) {
	411	case VERIFY_MD5:
	412	fill_md5(hdr, data, data_len);
	413	break;
	414	case VERIFY_CRC64:
	415	fill_crc64(hdr, data, data_len);
	416	break;
	417	case VERIFY_CRC32:
	418	fill_crc32(hdr, data, data_len);
	419	break;
	420	case VERIFY_CRC16:
	421	fill_crc16(hdr, data, data_len);
	422	break;
	423	case VERIFY_CRC7:
	424	fill_crc7(hdr, data, data_len);
	425	break;
cd14cc10 JA	426	case VERIFY_SHA256:
	427	fill_sha256(hdr, data, data_len);
	428	break;
	429	case VERIFY_SHA512:
	430	fill_sha512(hdr, data, data_len);
	431	break;
3f9f4e26 SL	432	default:
	433	log_err("fio: bad verify type: %d\n", td->o.verify);
	434	assert(0);
	435	}
a59e170d	436	if (td->o.verify_offset)
87677832	437	memswp(p, p + td->o.verify_offset, hdr_size(hdr));
e29d1b70	438	}
e29d1b70 JA	439	}
	440
	441	int get_next_verify(struct thread_data td, struct io_u io_u)
	442	{
8de8f047	443	struct io_piece *ipo = NULL;
e29d1b70	444
d2d7fa53 JA	445	/*
	446	* this io_u is from a requeue, we already filled the offsets
	447	*/
	448	if (io_u->file)
	449	return 0;
	450
8de8f047 JA	451	if (!RB_EMPTY_ROOT(&td->io_hist_tree)) {
8de8f047 JA	452	struct rb_node *n = rb_first(&td->io_hist_tree);
e29d1b70	453
8de8f047	454	ipo = rb_entry(n, struct io_piece, rb_node);
4b87898e	455	rb_erase(n, &td->io_hist_tree);
8de8f047 JA	456	} else if (!list_empty(&td->io_hist_list)) {
	457	ipo = list_entry(td->io_hist_list.next, struct io_piece, list);
	458	list_del(&ipo->list);
	459	}
e29d1b70	460
8de8f047	461	if (ipo) {
e29d1b70 JA	462	io_u->offset = ipo->offset;
e29d1b70 JA	463	io_u->buflen = ipo->len;
36167d82	464	io_u->file = ipo->file;
97af62ce JA	465
	466	if ((io_u->file->flags & FIO_FILE_OPEN) == 0) {
	467	int r = td_io_open_file(td, io_u->file);
	468
	469	if (r)
	470	return 1;
	471	}
	472
	473	get_file(ipo->file);
	474	assert(io_u->file->flags & FIO_FILE_OPEN);
e29d1b70	475	io_u->ddir = DDIR_READ;
36167d82 JA	476	io_u->xfer_buf = io_u->buf;
36167d82 JA	477	io_u->xfer_buflen = io_u->buflen;
e29d1b70 JA	478	free(ipo);
	479	return 0;
	480	}
	481
	482	return 1;
	483	}