1 #ifndef FIO_VERIFY_STATE_H
2 #define FIO_VERIFY_STATE_H
3
4 #include <stdint.h>
5 #include <string.h>
6 #include <limits.h>
7
8 struct thread_rand32_state {
9 uint32_t s[4];
10 };
11
12 struct thread_rand64_state {
13 uint64_t s[6];
14 };
15
16 struct thread_rand_state {
17 uint64_t use64;
18 union {
19 struct thread_rand32_state state32;
20 struct thread_rand64_state state64;
21 };
22 };
23
24 /*
25 * For dumping current write state
26 */
27 struct file_comp {
28 uint64_t fileno;
29 uint64_t offset;
30 };
31
32 struct thread_io_list {
33 uint64_t no_comps;
34 uint32_t depth;
35 uint32_t nofiles;
36 uint64_t numberio;
37 uint64_t index;
38 struct thread_rand_state rand;
39 uint8_t name[64];
40 struct file_comp comps[0];
41 };
42
43 struct all_io_list {
44 uint64_t threads;
45 struct thread_io_list state[0];
46 };
47
48 #define VSTATE_HDR_VERSION 0x03
49
50 struct verify_state_hdr {
51 uint64_t version;
52 uint64_t size;
53 uint64_t crc;
54 };
55
56 #define IO_LIST_ALL 0xffffffff
57
58 struct io_u;
59 extern struct all_io_list *get_all_io_list(int, size_t *);
60 extern void __verify_save_state(struct all_io_list *, const char *);
61 extern void verify_save_state(int mask);
62 extern int verify_load_state(struct thread_data *, const char *);
63 extern void verify_free_state(struct thread_data *);
64 extern int verify_state_should_stop(struct thread_data *, struct io_u *);
65 extern void verify_assign_state(struct thread_data *, void *);
66 extern int verify_state_hdr(struct verify_state_hdr *, struct thread_io_list *);
67
__thread_io_list_sz(uint32_t depth,uint32_t nofiles)68 static inline size_t __thread_io_list_sz(uint32_t depth, uint32_t nofiles)
69 {
70 return sizeof(struct thread_io_list) + depth * nofiles * sizeof(struct file_comp);
71 }
72
thread_io_list_sz(struct thread_io_list * s)73 static inline size_t thread_io_list_sz(struct thread_io_list *s)
74 {
75 return __thread_io_list_sz(le32_to_cpu(s->depth), le32_to_cpu(s->nofiles));
76 }
77
io_list_next(struct thread_io_list * s)78 static inline struct thread_io_list *io_list_next(struct thread_io_list *s)
79 {
80 return (void *) s + thread_io_list_sz(s);
81 }
82
verify_state_gen_name(char * out,size_t size,const char * name,const char * prefix,int num)83 static inline void verify_state_gen_name(char *out, size_t size,
84 const char *name, const char *prefix,
85 int num)
86 {
87 char ename[PATH_MAX];
88 char *ptr;
89
90 /*
91 * Escape '/', just turn them into '.'
92 */
93 ptr = ename;
94 do {
95 *ptr = *name;
96 if (*ptr == '\0')
97 break;
98 else if (*ptr == '/')
99 *ptr = '.';
100 ptr++;
101 name++;
102 } while (1);
103
104 snprintf(out, size, "%s-%s-%d-verify.state", prefix, ename, num);
105 out[size - 1] = '\0';
106 }
107
108 #endif
109