Extend --readonly
[fio.git] / verify.c
index 6972fe3ff59a2a10604ba708c05c2524a562c83f..59a9ddedb6157c0e5a31171069fdb0bab6195bbf 100644 (file)
--- a/verify.c
+++ b/verify.c
@@ -8,8 +8,15 @@
 
 #include "fio.h"
 
-static void fill_random_bytes(struct thread_data *td,
-                             void *p, unsigned int len)
+#include "crc/md5.h"
+#include "crc/crc64.h"
+#include "crc/crc32.h"
+#include "crc/crc16.h"
+#include "crc/crc7.h"
+#include "crc/sha256.h"
+#include "crc/sha512.h"
+
+static void fill_random_bytes(struct thread_data *td, void *p, unsigned int len)
 {
        unsigned int todo;
        int r;
@@ -32,9 +39,54 @@ static void fill_random_bytes(struct thread_data *td,
        }
 }
 
-void memswp(void* buf1, void* buf2, unsigned int len)
+static void fill_pattern(struct thread_data *td, void *p, unsigned int len)
 {
-       struct verify_header swap;
+       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)
+{
+       char swap[200];
+
+       assert(len <= sizeof(swap));
+
        memcpy(&swap, buf1, len);
        memcpy(buf1, buf2, len);
        memcpy(buf2, &swap, len);
@@ -50,6 +102,63 @@ static void hexdump(void *buffer, int len)
        log_info("\n");
 }
 
+/*
+ * Prepare for seperation of verify_header and checksum header
+ */
+static inline unsigned int __hdr_size(int verify_type)
+{
+       unsigned int len;
+
+       switch (verify_type) {
+       case VERIFY_NONE:
+       case VERIFY_NULL:
+       case VERIFY_PATTERN:
+               len = 0;
+               break;
+       case VERIFY_MD5:
+               len = sizeof(struct vhdr_md5);
+               break;
+       case VERIFY_CRC64:
+               len = sizeof(struct vhdr_crc64);
+               break;
+       case VERIFY_CRC32:
+               len = sizeof(struct vhdr_crc32);
+               break;
+       case VERIFY_CRC16:
+               len = sizeof(struct vhdr_crc16);
+               break;
+       case VERIFY_CRC7:
+               len = sizeof(struct vhdr_crc7);
+               break;
+       case VERIFY_SHA256:
+               len = sizeof(struct vhdr_sha256);
+               break;
+       case VERIFY_SHA512:
+               len = sizeof(struct vhdr_sha512);
+               break;
+       case VERIFY_META:
+               len = sizeof(struct vhdr_meta);
+               break;
+       default:
+               log_err("fio: unknown verify header!\n");
+               assert(0);
+       }
+
+       return len + sizeof(struct verify_header);
+}
+
+static inline unsigned int hdr_size(struct verify_header *hdr)
+{
+       return __hdr_size(hdr->verify_type);
+}
+
+static void *hdr_priv(struct verify_header *hdr)
+{
+       void *priv = hdr;
+
+       return priv + sizeof(struct verify_header);
+}
+
 /*
  * Return data area 'header_num'
  */
@@ -57,23 +166,89 @@ static inline void *io_u_verify_off(struct verify_header *hdr,
                                    struct io_u *io_u,
                                    unsigned char header_num)
 {
-       return io_u->buf + sizeof(*hdr) + header_num * hdr->len;
+       return io_u->buf + header_num * hdr->len + hdr_size(hdr);
+}
+
+static int verify_io_u_meta(struct verify_header *hdr, struct thread_data *td,
+                            struct io_u *io_u, unsigned int header_num)
+{
+       struct vhdr_meta *vh = hdr_priv(hdr);
+
+       if (vh->offset != io_u->offset + header_num * td->o.verify_interval) {
+               log_err("meta: verify failed at %llu/%u\n",
+                             io_u->offset + header_num * hdr->len,
+                             hdr->len);
+               return EIO;
+       }
+
+       return 0;
+}
+
+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);
+       struct vhdr_sha512 *vh = hdr_priv(hdr);
+       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(vh->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(vh->sha512, sizeof(vh->sha512));
+               hexdump(sha512_ctx.buf, sizeof(sha512));
+               return EIO;
+       }
+
+       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);
+       struct vhdr_sha256 *vh = hdr_priv(hdr);
+       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(vh->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(vh->sha256, sizeof(vh->sha256));
+               hexdump(sha256_ctx.buf, sizeof(sha256));
+               return EIO;
+       }
+
+       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);
+       struct vhdr_crc7 *vh = hdr_priv(hdr);
        unsigned char c;
 
-       c = crc7(p, hdr->len - sizeof(*hdr));
+       c = crc7(p, hdr->len - hdr_size(hdr));
 
-       if (c != hdr->crc7) {
+       if (c != vh->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;
+               log_err("crc7: wanted %x, got %x\n", vh->crc7, c);
+               return EIO;
        }
 
        return 0;
@@ -83,16 +258,17 @@ 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);
+       struct vhdr_crc16 *vh = hdr_priv(hdr);
        unsigned short c;
 
-       c = crc16(p, hdr->len - sizeof(*hdr));
+       c = crc16(p, hdr->len - hdr_size(hdr));
 
-       if (c != hdr->crc16) {
+       if (c != vh->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;
+               log_err("crc16: wanted %x, got %x\n", vh->crc16, c);
+               return EIO;
        }
 
        return 0;
@@ -102,16 +278,17 @@ 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);
+       struct vhdr_crc64 *vh = hdr_priv(hdr);
        unsigned long long c;
 
-       c = crc64(p, hdr->len - sizeof(*hdr));
+       c = crc64(p, hdr->len - hdr_size(hdr));
 
-       if (c != hdr->crc64) {
+       if (c != vh->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;
+               log_err("crc64: wanted %llx, got %llx\n", (unsigned long long) vh->crc64, c);
+               return EIO;
        }
 
        return 0;
@@ -121,16 +298,17 @@ 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;
+       struct vhdr_crc32 *vh = hdr_priv(hdr);
+       uint32_t c;
 
-       c = crc32(p, hdr->len - sizeof(*hdr));
+       c = crc32(p, hdr->len - hdr_size(hdr));
 
-       if (c != hdr->crc32) {
+       if (c != vh->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;
+               log_err("crc32: wanted %x, got %x\n", vh->crc32, c);
+               return EIO;
        }
 
        return 0;
@@ -140,21 +318,59 @@ 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);
+       struct vhdr_md5 *vh = hdr_priv(hdr);
        uint32_t hash[MD5_HASH_WORDS];
        struct md5_ctx md5_ctx = {
                .hash = hash,
        };
 
-       memset(md5_ctx.hash, 0, sizeof(hdr->md5_digest));
-       md5_update(&md5_ctx, p, hdr->len - sizeof(*hdr));
+       md5_init(&md5_ctx);
+       md5_update(&md5_ctx, p, hdr->len - hdr_size(hdr));
 
-       if (memcmp(hdr->md5_digest, md5_ctx.hash, sizeof(hash))) {
+       if (memcmp(vh->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(vh->md5_digest, sizeof(vh->md5_digest));
                hexdump(md5_ctx.hash, sizeof(hash));
-               return 1;
+               return EIO;
+       }
+
+       return 0;
+}
+
+static unsigned int hweight8(unsigned int w)
+{
+       unsigned int res = w - ((w >> 1) & 0x55);
+
+       res = (res & 0x33) + ((res >> 2) & 0x33);
+       return (res + (res >> 4)) & 0x0F;
+}
+
+int verify_io_u_pattern(unsigned long pattern, unsigned long pattern_size,
+                        char *buf, unsigned int len, unsigned int mod)
+{
+       unsigned int i;
+       char split_pattern[4];
+
+       for (i = 0; i < 4; i++) {
+               split_pattern[i] = pattern & 0xff;
+               pattern >>= 8;
+       }
+
+       for (i = 0; i < len; i++) {
+               if (buf[i] != split_pattern[mod]) {
+                       unsigned int bits;
+
+                       bits = hweight8(buf[i] ^ split_pattern[mod]);
+                       log_err("fio: got pattern %x, wanted %x. Bad bits %d\n",
+                               buf[i], split_pattern[mod], bits);
+                       log_err("fio: bad pattern block offset %u\n", i);
+                       return EIO;
+               }
+               mod++;
+               if (mod == pattern_size)
+                       mod = 0;
        }
 
        return 0;
@@ -163,7 +379,7 @@ static int verify_io_u_md5(struct verify_header *hdr, struct io_u *io_u,
 int verify_io_u(struct thread_data *td, struct io_u *io_u)
 {
        struct verify_header *hdr;
-       unsigned int hdr_inc, hdr_num = 0;
+       unsigned int hdr_size, hdr_inc, hdr_num = 0;
        void *p;
        int ret;
 
@@ -171,13 +387,18 @@ int verify_io_u(struct thread_data *td, struct io_u *io_u)
                return 0;
 
        hdr_inc = io_u->buflen;
-       if (td->o.header_interval)
-               hdr_inc = td->o.header_interval;
-
-       for (p = io_u->buf; p < io_u->buf + io_u->buflen; p += hdr_inc) {
-               if (td->o.header_offset)
-                       memswp(p, p + td->o.header_offset, sizeof(*hdr));
+       if (td->o.verify_interval)
+               hdr_inc = td->o.verify_interval;
 
+       ret = 0;
+       for (p = io_u->buf; p < io_u->buf + io_u->buflen; p += hdr_inc, hdr_num++) {
+               if (ret && td->o.verify_fatal) {
+                       td->terminate = 1;
+                       break;
+               }
+               hdr_size = __hdr_size(td->o.verify);
+               if (td->o.verify_offset)
+                       memswp(p, p + td->o.verify_offset, hdr_size);
                hdr = p;
 
                if (hdr->fio_magic != FIO_HDR_MAGIC) {
@@ -201,43 +422,108 @@ int verify_io_u(struct thread_data *td, struct io_u *io_u)
                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;
+               case VERIFY_META:
+                       ret = verify_io_u_meta(hdr, td, io_u, hdr_num);
+                       break;
+               case VERIFY_PATTERN:
+                       ret = verify_io_u_pattern(td->o.verify_pattern,
+                                                 td->o.verify_pattern_bytes,
+                                                 p + hdr_size,
+                                                 hdr_inc - hdr_size,
+                                                 hdr_size % 4);
+                       if (ret)
+                               log_err("fio: verify failed at %llu/%u\n",
+                                       io_u->offset + hdr_num * hdr->len,
+                                       hdr->len);
+                       break;
                default:
                        log_err("Bad verify type %u\n", hdr->verify_type);
-                       ret = 1;
+                       ret = EINVAL;
                }
-               hdr_num++;
        }
 
-       return 0;
+       return ret;
+}
+
+static void fill_meta(struct verify_header *hdr, struct thread_data *td,
+                      struct io_u *io_u, unsigned int header_num)
+{
+       struct vhdr_meta *vh = hdr_priv(hdr);
+
+       vh->thread = td->thread_number;
+
+       vh->time_sec = io_u->start_time.tv_sec;
+       vh->time_usec = io_u->start_time.tv_usec;
+
+       vh->numberio = td->io_issues[DDIR_WRITE];
+
+       vh->offset = io_u->offset + header_num * td->o.verify_interval;
+}
+
+static void fill_sha512(struct verify_header *hdr, void *p, unsigned int len)
+{
+       struct vhdr_sha512 *vh = hdr_priv(hdr);
+       struct sha512_ctx sha512_ctx = {
+               .buf = vh->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 vhdr_sha256 *vh = hdr_priv(hdr);
+       struct sha256_ctx sha256_ctx = {
+               .buf = vh->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);
+       struct vhdr_crc7 *vh = hdr_priv(hdr);
+
+       vh->crc7 = crc7(p, len);
 }
 
 static void fill_crc16(struct verify_header *hdr, void *p, unsigned int len)
 {
-       hdr->crc16 = crc16(p, len);
+       struct vhdr_crc16 *vh = hdr_priv(hdr);
+
+       vh->crc16 = crc16(p, len);
 }
 
 static void fill_crc32(struct verify_header *hdr, void *p, unsigned int len)
 {
-       hdr->crc32 = crc32(p, len);
+       struct vhdr_crc32 *vh = hdr_priv(hdr);
+
+       vh->crc32 = crc32(p, len);
 }
 
 static void fill_crc64(struct verify_header *hdr, void *p, unsigned int len)
 {
-       hdr->crc64 = crc64(p, len);
+       struct vhdr_crc64 *vh = hdr_priv(hdr);
+
+       vh->crc64 = crc64(p, len);
 }
 
 static void fill_md5(struct verify_header *hdr, void *p, unsigned int len)
 {
+       struct vhdr_md5 *vh = hdr_priv(hdr);
        struct md5_ctx md5_ctx = {
-               .hash = (uint32_t *) hdr->md5_digest,
+               .hash = (uint32_t *) vh->md5_digest,
        };
 
-       memset(md5_ctx.hash, 0, sizeof(hdr->md5_digest));
+       md5_init(&md5_ctx);
        md5_update(&md5_ctx, p, len);
 }
 
@@ -249,17 +535,16 @@ 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;
+       unsigned int hdr_inc, data_len, header_num = 0;
 
        if (td->o.verify == VERIFY_NULL)
                return;
 
-       fill_random_bytes(td, p, io_u->buflen);
+       fill_pattern(td, p, io_u->buflen);
 
        hdr_inc = io_u->buflen;
-       if (td->o.header_interval)
-               hdr_inc = td->o.header_interval;
-       data_len = hdr_inc - sizeof(*hdr);
+       if (td->o.verify_interval)
+               hdr_inc = td->o.verify_interval;
 
        for (;p < io_u->buf + io_u->buflen; p += hdr_inc) {
                hdr = p;
@@ -267,8 +552,9 @@ void populate_verify_io_u(struct thread_data *td, struct io_u *io_u)
                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 + sizeof(*hdr);
+               data = p + hdr_size(hdr);
                switch (td->o.verify) {
                case VERIFY_MD5:
                        fill_md5(hdr, data, data_len);
@@ -285,12 +571,24 @@ void populate_verify_io_u(struct thread_data *td, struct io_u *io_u)
                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;
+               case VERIFY_META:
+                       fill_meta(hdr, td, io_u, header_num);
+                       break;
+               case VERIFY_PATTERN:
+                       break;
                default:
                        log_err("fio: bad verify type: %d\n", td->o.verify);
                        assert(0);
                }
-               if (td->o.header_offset)
-                       memswp(p, p + td->o.header_offset, sizeof(*hdr));
+               if (td->o.verify_offset)
+                       memswp(p, p + td->o.verify_offset, hdr_size(hdr));
+               header_num++;
        }
 }