Commit | Line | Data |
---|---|---|
ebac4655 JA |
1 | #ifndef FIO_H |
2 | #define FIO_H | |
3 | ||
4 | #include <sched.h> | |
5 | #include <limits.h> | |
6 | #include <pthread.h> | |
7 | #include <sys/time.h> | |
8 | #include <sys/resource.h> | |
3c39a379 JA |
9 | #include <errno.h> |
10 | #include <stdlib.h> | |
11 | #include <stdio.h> | |
6d6f031f | 12 | #include <unistd.h> |
34cfcdaf | 13 | #include <string.h> |
ebac4655 JA |
14 | |
15 | #include "list.h" | |
16 | #include "md5.h" | |
17 | #include "crc32.h" | |
18 | #include "arch.h" | |
19 | #include "os.h" | |
20 | ||
b6959b55 | 21 | #ifdef FIO_HAVE_SYSLET |
a4f4fdd7 | 22 | #include "syslet.h" |
b6959b55 | 23 | #endif |
a4f4fdd7 | 24 | |
1e97cce9 JA |
25 | enum fio_ddir { |
26 | DDIR_READ = 0, | |
27 | DDIR_WRITE, | |
28 | DDIR_SYNC, | |
29 | }; | |
30 | ||
413dd459 | 31 | enum td_ddir { |
b1ec1da6 JA |
32 | TD_DDIR_READ = 1 << 0, |
33 | TD_DDIR_WRITE = 1 << 1, | |
34 | TD_DDIR_RAND = 1 << 2, | |
35 | TD_DDIR_RW = TD_DDIR_READ | TD_DDIR_WRITE, | |
36 | TD_DDIR_RANDREAD = TD_DDIR_READ | TD_DDIR_RAND, | |
37 | TD_DDIR_RANDWRITE = TD_DDIR_WRITE | TD_DDIR_READ, | |
38 | TD_DDIR_RANDRW = TD_DDIR_RW | TD_DDIR_RAND, | |
413dd459 JA |
39 | }; |
40 | ||
1c9e06ee JA |
41 | /* |
42 | * Use for maintaining statistics | |
43 | */ | |
ebac4655 | 44 | struct io_stat { |
ebac4655 JA |
45 | unsigned long max_val; |
46 | unsigned long min_val; | |
47 | unsigned long samples; | |
68704084 JA |
48 | |
49 | double mean; | |
50 | double S; | |
ebac4655 JA |
51 | }; |
52 | ||
1c9e06ee JA |
53 | /* |
54 | * A single data sample | |
55 | */ | |
ebac4655 JA |
56 | struct io_sample { |
57 | unsigned long time; | |
58 | unsigned long val; | |
1e97cce9 | 59 | enum fio_ddir ddir; |
ebac4655 JA |
60 | }; |
61 | ||
1c9e06ee JA |
62 | /* |
63 | * Dynamically growing data sample log | |
64 | */ | |
ebac4655 JA |
65 | struct io_log { |
66 | unsigned long nr_samples; | |
67 | unsigned long max_samples; | |
68 | struct io_sample *log; | |
69 | }; | |
70 | ||
1c9e06ee JA |
71 | /* |
72 | * When logging io actions, this matches a single sent io_u | |
73 | */ | |
ebac4655 JA |
74 | struct io_piece { |
75 | struct list_head list; | |
53cdc686 | 76 | struct fio_file *file; |
ebac4655 | 77 | unsigned long long offset; |
a4f4fdd7 | 78 | unsigned long len; |
1e97cce9 | 79 | enum fio_ddir ddir; |
ebac4655 JA |
80 | }; |
81 | ||
a4f4fdd7 JA |
82 | #ifdef FIO_HAVE_SYSLET |
83 | struct syslet_req { | |
84 | struct syslet_uatom atom; | |
a4f4fdd7 JA |
85 | long ret; |
86 | }; | |
87 | #endif | |
88 | ||
0c6e7517 JA |
89 | enum { |
90 | IO_U_F_FREE = 1 << 0, | |
91 | IO_U_F_FLIGHT = 1 << 1, | |
92 | }; | |
93 | ||
ebac4655 JA |
94 | /* |
95 | * The io unit | |
96 | */ | |
97 | struct io_u { | |
98 | union { | |
99 | #ifdef FIO_HAVE_LIBAIO | |
100 | struct iocb iocb; | |
101 | #endif | |
102 | #ifdef FIO_HAVE_POSIXAIO | |
103 | struct aiocb aiocb; | |
104 | #endif | |
105 | #ifdef FIO_HAVE_SGIO | |
106 | struct sg_io_hdr hdr; | |
a4f4fdd7 JA |
107 | #endif |
108 | #ifdef FIO_HAVE_SYSLET | |
a2e1b08a | 109 | struct syslet_req req; |
ebac4655 JA |
110 | #endif |
111 | }; | |
112 | struct timeval start_time; | |
113 | struct timeval issue_time; | |
114 | ||
1c9e06ee JA |
115 | /* |
116 | * Allocated/set buffer and length | |
117 | */ | |
6b9cea23 | 118 | void *buf; |
a4f4fdd7 | 119 | unsigned long buflen; |
ebac4655 JA |
120 | unsigned long long offset; |
121 | ||
1c9e06ee JA |
122 | /* |
123 | * IO engine state, may be different from above when we get | |
124 | * partial transfers / residual data counts | |
125 | */ | |
cec6b55d | 126 | void *xfer_buf; |
a4f4fdd7 | 127 | unsigned long xfer_buflen; |
cec6b55d | 128 | |
ebac4655 JA |
129 | unsigned int resid; |
130 | unsigned int error; | |
131 | ||
1e97cce9 | 132 | enum fio_ddir ddir; |
ebac4655 | 133 | |
dfd7bc2c JA |
134 | /* |
135 | * io engine private data | |
136 | */ | |
137 | union { | |
138 | unsigned int index; | |
139 | unsigned int seen; | |
140 | }; | |
141 | ||
0c6e7517 JA |
142 | unsigned int flags; |
143 | ||
53cdc686 JA |
144 | struct fio_file *file; |
145 | ||
ebac4655 | 146 | struct list_head list; |
d7762cf8 JA |
147 | |
148 | /* | |
149 | * Callback for io completion | |
150 | */ | |
151 | int (*end_io)(struct io_u *); | |
ebac4655 JA |
152 | }; |
153 | ||
36167d82 JA |
154 | /* |
155 | * io_ops->queue() return values | |
156 | */ | |
157 | enum { | |
158 | FIO_Q_COMPLETED = 0, /* completed sync */ | |
159 | FIO_Q_QUEUED = 1, /* queued, will complete async */ | |
755200a3 | 160 | FIO_Q_BUSY = 2, /* no more room, call ->commit() */ |
36167d82 JA |
161 | }; |
162 | ||
ebac4655 JA |
163 | #define FIO_HDR_MAGIC 0xf00baaef |
164 | ||
165 | enum { | |
1c9e06ee JA |
166 | VERIFY_NONE = 0, /* no verification */ |
167 | VERIFY_MD5, /* md5 sum data blocks */ | |
168 | VERIFY_CRC32, /* crc32 sum data blocks */ | |
ebac4655 JA |
169 | }; |
170 | ||
1c9e06ee JA |
171 | /* |
172 | * A header structure associated with each checksummed data block | |
173 | */ | |
ebac4655 JA |
174 | struct verify_header { |
175 | unsigned int fio_magic; | |
176 | unsigned int len; | |
177 | unsigned int verify_type; | |
178 | union { | |
179 | char md5_digest[MD5_HASH_WORDS * 4]; | |
180 | unsigned long crc32; | |
181 | }; | |
182 | }; | |
183 | ||
184 | struct group_run_stats { | |
9104f874 JA |
185 | unsigned long long max_run[2], min_run[2]; |
186 | unsigned long long max_bw[2], min_bw[2]; | |
e9b2a3fa | 187 | unsigned long long io_kb[2]; |
9104f874 | 188 | unsigned long long agg[2]; |
ebac4655 JA |
189 | }; |
190 | ||
e9c047a0 JA |
191 | /* |
192 | * What type of allocation to use for io buffers | |
193 | */ | |
194 | enum fio_memtype { | |
195 | MEM_MALLOC = 0, /* ordinary malloc */ | |
196 | MEM_SHM, /* use shared memory segments */ | |
74b025b0 | 197 | MEM_SHMHUGE, /* use shared memory segments with huge pages */ |
e9c047a0 | 198 | MEM_MMAP, /* use anonynomous mmap */ |
d0bdaf49 | 199 | MEM_MMAPHUGE, /* memory mapped huge file */ |
e9c047a0 JA |
200 | }; |
201 | ||
202 | /* | |
203 | * The type of object we are working on | |
204 | */ | |
205 | enum fio_filetype { | |
1c9e06ee JA |
206 | FIO_TYPE_FILE = 1, /* plain file */ |
207 | FIO_TYPE_BD, /* block device */ | |
208 | FIO_TYPE_CHAR, /* character device */ | |
e9c047a0 JA |
209 | }; |
210 | ||
2866c82d | 211 | enum fio_ioengine_flags { |
1c9e06ee JA |
212 | FIO_SYNCIO = 1 << 0, /* io engine has synchronous ->queue */ |
213 | FIO_CPUIO = 1 << 1, /* cpu burner, doesn't do real io */ | |
214 | FIO_MMAPIO = 1 << 2, /* uses memory mapped io */ | |
215 | FIO_RAWIO = 1 << 3, /* some sort of direct/raw io */ | |
216 | FIO_NETIO = 1 << 4, /* networked io */ | |
217 | FIO_NULLIO = 1 << 5, /* no real data transfer (cpu/null) */ | |
e9c047a0 JA |
218 | }; |
219 | ||
1c9e06ee JA |
220 | /* |
221 | * Each thread_data structure has a number of files associated with it, | |
222 | * this structure holds state information for a single file. | |
223 | */ | |
53cdc686 JA |
224 | struct fio_file { |
225 | /* | |
226 | * A file may not be a file descriptor, let the io engine decide | |
227 | */ | |
228 | union { | |
229 | unsigned long file_data; | |
e26f8f7a | 230 | int fd; |
53cdc686 JA |
231 | }; |
232 | char *file_name; | |
233 | void *mmap; | |
234 | unsigned long long file_size; | |
235 | unsigned long long real_file_size; | |
236 | unsigned long long file_offset; | |
237 | unsigned long long last_pos; | |
02bcaa8c | 238 | unsigned long long last_completed_pos; |
53cdc686 | 239 | |
1c9e06ee JA |
240 | /* |
241 | * block map for random io | |
242 | */ | |
53cdc686 JA |
243 | unsigned long *file_map; |
244 | unsigned int num_maps; | |
1c9e06ee | 245 | unsigned int last_free_lookup; |
9b031dc8 JA |
246 | |
247 | unsigned int unlink; | |
53cdc686 JA |
248 | }; |
249 | ||
079ad09b JA |
250 | struct thread_stat { |
251 | struct io_log *slat_log; | |
252 | struct io_log *clat_log; | |
253 | struct io_log *bw_log; | |
254 | ||
255 | /* | |
256 | * bandwidth and latency stats | |
257 | */ | |
258 | struct io_stat clat_stat[2]; /* completion latency */ | |
259 | struct io_stat slat_stat[2]; /* submission latency */ | |
260 | struct io_stat bw_stat[2]; /* bandwidth stats */ | |
261 | ||
262 | unsigned long long stat_io_bytes[2]; | |
263 | struct timeval stat_sample_time[2]; | |
264 | ||
265 | /* | |
266 | * fio system usage accounting | |
267 | */ | |
268 | struct rusage ru_start; | |
269 | struct rusage ru_end; | |
270 | unsigned long usr_time; | |
271 | unsigned long sys_time; | |
272 | unsigned long ctx; | |
273 | }; | |
274 | ||
71619dc2 JA |
275 | /* |
276 | * How many depth levels to log | |
277 | */ | |
278 | #define FIO_IO_U_MAP_NR 8 | |
ec118304 | 279 | #define FIO_IO_U_LAT_NR 12 |
71619dc2 | 280 | |
e9c047a0 JA |
281 | /* |
282 | * This describes a single thread/process executing a fio job. | |
283 | */ | |
ebac4655 | 284 | struct thread_data { |
b1ec1da6 | 285 | int pad; |
61697c37 | 286 | char *description; |
b4692828 | 287 | char *name; |
ef899b63 | 288 | char *directory; |
13f8e2d2 | 289 | char *filename; |
ebac4655 JA |
290 | char verror[80]; |
291 | pthread_t thread; | |
292 | int thread_number; | |
293 | int groupid; | |
079ad09b | 294 | struct thread_stat ts; |
e9c047a0 | 295 | enum fio_filetype filetype; |
53cdc686 JA |
296 | struct fio_file *files; |
297 | unsigned int nr_files; | |
13f8e2d2 | 298 | unsigned int nr_uniq_files; |
0aabe160 JA |
299 | union { |
300 | unsigned int next_file; | |
301 | os_random_state_t next_file_state; | |
302 | }; | |
ebac4655 | 303 | int error; |
ebac4655 JA |
304 | pid_t pid; |
305 | char *orig_buffer; | |
306 | size_t orig_buffer_size; | |
307 | volatile int terminate; | |
308 | volatile int runstate; | |
413dd459 | 309 | enum td_ddir td_ddir; |
ebac4655 | 310 | unsigned int ioprio; |
87dc1ab1 | 311 | unsigned int last_was_sync; |
e9c047a0 | 312 | |
9158d2f7 JA |
313 | unsigned int odirect; |
314 | unsigned int invalidate_cache; | |
315 | unsigned int create_serialize; | |
316 | unsigned int create_fsync; | |
317 | unsigned int end_fsync; | |
318 | unsigned int sync_io; | |
319 | unsigned int verify; | |
320 | unsigned int use_thread; | |
321 | unsigned int unlink; | |
322 | unsigned int do_disk_util; | |
323 | unsigned int override_sync; | |
324 | unsigned int rand_repeatable; | |
325 | unsigned int write_lat_log; | |
326 | unsigned int write_bw_log; | |
bb8895e0 | 327 | unsigned int norandommap; |
690adba3 | 328 | unsigned int bs_unaligned; |
e9c047a0 | 329 | |
a00735e6 JA |
330 | unsigned int bs[2]; |
331 | unsigned int min_bs[2]; | |
332 | unsigned int max_bs[2]; | |
56bb17f2 | 333 | unsigned int hugepage_size; |
a00735e6 | 334 | unsigned int rw_min_bs; |
ebac4655 | 335 | unsigned int thinktime; |
48097d5c | 336 | unsigned int thinktime_spin; |
9c1f7434 | 337 | unsigned int thinktime_blocks; |
ebac4655 JA |
338 | unsigned int fsync_blocks; |
339 | unsigned int start_delay; | |
906c8d75 | 340 | unsigned long timeout; |
ebac4655 | 341 | unsigned int overwrite; |
ebac4655 | 342 | unsigned int bw_avg_time; |
ebac4655 | 343 | unsigned int loops; |
20dc95c4 JA |
344 | unsigned long long zone_size; |
345 | unsigned long long zone_skip; | |
e9c047a0 | 346 | enum fio_memtype mem_type; |
313cb206 JA |
347 | char *mmapfile; |
348 | int mmapfd; | |
ebac4655 JA |
349 | unsigned int stonewall; |
350 | unsigned int numjobs; | |
ebac4655 | 351 | unsigned int iodepth; |
e916b390 | 352 | unsigned int iodepth_low; |
ebac4655 | 353 | os_cpu_mask_t cpumask; |
aea47d44 | 354 | unsigned int iolog; |
843a7413 | 355 | unsigned int read_iolog; |
a6ccc7be JA |
356 | unsigned int rwmixcycle; |
357 | unsigned int rwmixread; | |
e1f36503 | 358 | unsigned int rwmixwrite; |
b6f4d880 | 359 | unsigned int nice; |
0aabe160 | 360 | unsigned int file_service_type; |
aea47d44 | 361 | |
076efc7c JA |
362 | char *read_iolog_file; |
363 | char *write_iolog_file; | |
843a7413 JA |
364 | void *iolog_buf; |
365 | FILE *iolog_f; | |
ebac4655 | 366 | |
da86774e | 367 | char *sysfs_root; |
da86774e JA |
368 | char *ioscheduler; |
369 | ||
6dfd46b9 JA |
370 | os_random_state_t bsrange_state; |
371 | os_random_state_t verify_state; | |
ebac4655 JA |
372 | |
373 | int shm_id; | |
374 | ||
e9c047a0 JA |
375 | /* |
376 | * IO engine hooks, contains everything needed to submit an io_u | |
377 | * to any of the available IO engines. | |
378 | */ | |
2866c82d | 379 | struct ioengine_ops *io_ops; |
ebac4655 | 380 | |
e9c047a0 JA |
381 | /* |
382 | * Current IO depth and list of free and busy io_u's. | |
383 | */ | |
ebac4655 | 384 | unsigned int cur_depth; |
71619dc2 | 385 | unsigned int io_u_map[FIO_IO_U_MAP_NR]; |
ec118304 | 386 | unsigned int io_u_lat[FIO_IO_U_LAT_NR]; |
71619dc2 | 387 | unsigned long total_io_u; |
ebac4655 JA |
388 | struct list_head io_u_freelist; |
389 | struct list_head io_u_busylist; | |
755200a3 | 390 | struct list_head io_u_requeues; |
ebac4655 | 391 | |
e9c047a0 JA |
392 | /* |
393 | * Rate state | |
394 | */ | |
ebac4655 JA |
395 | unsigned int rate; |
396 | unsigned int ratemin; | |
397 | unsigned int ratecycle; | |
398 | unsigned long rate_usec_cycle; | |
399 | long rate_pending_usleep; | |
400 | unsigned long rate_bytes; | |
401 | struct timeval lastrate; | |
402 | ||
403 | unsigned long runtime[2]; /* msec */ | |
404 | unsigned long long io_size; | |
53cdc686 JA |
405 | unsigned long long total_file_size; |
406 | unsigned long long start_offset; | |
ebac4655 JA |
407 | unsigned long long total_io_size; |
408 | ||
755200a3 | 409 | unsigned long io_issues[2]; |
9104f874 JA |
410 | unsigned long long io_blocks[2]; |
411 | unsigned long long io_bytes[2]; | |
9104f874 | 412 | unsigned long long this_io_bytes[2]; |
079ad09b | 413 | unsigned long long zone_bytes; |
bbfd6b00 | 414 | volatile int mutex; |
ebac4655 | 415 | |
e9c047a0 JA |
416 | /* |
417 | * State for random io, a bitmap of blocks done vs not done | |
418 | */ | |
6dfd46b9 | 419 | os_random_state_t random_state; |
ebac4655 | 420 | |
b990b5c0 JA |
421 | /* |
422 | * CPU "io" cycle burner | |
423 | */ | |
424 | unsigned int cpuload; | |
425 | unsigned int cpucycle; | |
426 | ||
ebac4655 JA |
427 | struct timeval start; /* start of this loop */ |
428 | struct timeval epoch; /* time job was started */ | |
69008999 | 429 | struct timeval end_time;/* time job ended */ |
ebac4655 | 430 | |
e9c047a0 JA |
431 | /* |
432 | * read/write mixed workload state | |
433 | */ | |
6dfd46b9 | 434 | os_random_state_t rwmix_state; |
a6ccc7be | 435 | struct timeval rwmix_switch; |
e9c047a0 | 436 | enum fio_ddir rwmix_ddir; |
a6ccc7be | 437 | |
4e0ba8af JA |
438 | /* |
439 | * Pre-run and post-run shell | |
440 | */ | |
441 | char *exec_prerun; | |
442 | char *exec_postrun; | |
443 | ||
e9c047a0 JA |
444 | /* |
445 | * IO historic logs | |
446 | */ | |
ebac4655 | 447 | struct list_head io_hist_list; |
aea47d44 | 448 | struct list_head io_log_list; |
433afcb4 JA |
449 | |
450 | /* | |
451 | * timeout handling | |
452 | */ | |
453 | struct timeval timeout_end; | |
454 | struct itimerval timer; | |
ebac4655 JA |
455 | }; |
456 | ||
0aabe160 JA |
457 | /* |
458 | * roundrobin available files, or choose one at random. | |
459 | */ | |
460 | enum { | |
461 | FIO_FSERVICE_RANDOM = 1, | |
462 | FIO_FSERVICE_RR = 2, | |
463 | }; | |
464 | ||
433afcb4 JA |
465 | /* |
466 | * 30 second per-io_u timeout, with 5 second intervals to avoid resetting | |
467 | * the timer on each queue operation. | |
468 | */ | |
469 | #define IO_U_TIMEOUT_INC 5 | |
470 | #define IO_U_TIMEOUT 30 | |
471 | ||
e1161c32 | 472 | #define __td_verror(td, err, msg, func) \ |
ebac4655 | 473 | do { \ |
19abcd3d JA |
474 | if ((td)->error) \ |
475 | break; \ | |
ebac4655 JA |
476 | int e = (err); \ |
477 | (td)->error = e; \ | |
e1161c32 | 478 | snprintf(td->verror, sizeof(td->verror) - 1, "file:%s:%d, func=%s, error=%s", __FILE__, __LINE__, (func), (msg)); \ |
ebac4655 JA |
479 | } while (0) |
480 | ||
b990b5c0 | 481 | |
e1161c32 JA |
482 | #define td_verror(td, err, func) \ |
483 | __td_verror((td), (err), strerror((err)), (func)) | |
484 | #define td_vmsg(td, err, msg, func) \ | |
485 | __td_verror((td), (err), (msg), (func)) | |
b990b5c0 | 486 | |
ebac4655 JA |
487 | extern int exitall_on_terminate; |
488 | extern int thread_number; | |
489 | extern int shm_id; | |
490 | extern int groupid; | |
c6ae0a5b | 491 | extern int terse_output; |
eb8bbf48 JA |
492 | extern FILE *f_out; |
493 | extern FILE *f_err; | |
53cdc686 | 494 | extern int temp_stall_ts; |
1e97cce9 | 495 | extern unsigned long long mlock_size; |
ebac4655 JA |
496 | |
497 | extern struct thread_data *threads; | |
498 | ||
413dd459 JA |
499 | #define td_read(td) ((td)->td_ddir & TD_DDIR_READ) |
500 | #define td_write(td) ((td)->td_ddir & TD_DDIR_WRITE) | |
501 | #define td_rw(td) (((td)->td_ddir & TD_DDIR_RW) == TD_DDIR_RW) | |
502 | #define td_random(td) ((td)->td_ddir & TD_DDIR_RAND) | |
ebac4655 JA |
503 | |
504 | #define BLOCKS_PER_MAP (8 * sizeof(long)) | |
a00735e6 | 505 | #define TO_MAP_BLOCK(td, f, b) ((b) - ((f)->file_offset / (td)->rw_min_bs)) |
53cdc686 JA |
506 | #define RAND_MAP_IDX(td, f, b) (TO_MAP_BLOCK(td, f, b) / BLOCKS_PER_MAP) |
507 | #define RAND_MAP_BIT(td, f, b) (TO_MAP_BLOCK(td, f, b) & (BLOCKS_PER_MAP - 1)) | |
ebac4655 JA |
508 | |
509 | #define MAX_JOBS (1024) | |
510 | ||
87dc1ab1 JA |
511 | static inline int should_fsync(struct thread_data *td) |
512 | { | |
513 | if (td->last_was_sync) | |
514 | return 0; | |
515 | if (td->odirect) | |
516 | return 0; | |
517 | if (td_write(td) || td_rw(td) || td->override_sync) | |
518 | return 1; | |
519 | ||
520 | return 0; | |
521 | } | |
522 | ||
1c9e06ee JA |
523 | /* |
524 | * Disk utils as read in /sys/block/<dev>/stat | |
525 | */ | |
ebac4655 JA |
526 | struct disk_util_stat { |
527 | unsigned ios[2]; | |
528 | unsigned merges[2]; | |
529 | unsigned long long sectors[2]; | |
530 | unsigned ticks[2]; | |
531 | unsigned io_ticks; | |
532 | unsigned time_in_queue; | |
533 | }; | |
534 | ||
1c9e06ee JA |
535 | /* |
536 | * Per-device disk util management | |
537 | */ | |
ebac4655 JA |
538 | struct disk_util { |
539 | struct list_head list; | |
540 | ||
541 | char *name; | |
542 | char path[256]; | |
543 | dev_t dev; | |
544 | ||
545 | struct disk_util_stat dus; | |
546 | struct disk_util_stat last_dus; | |
547 | ||
548 | unsigned long msec; | |
549 | struct timeval time; | |
550 | }; | |
551 | ||
ebac4655 JA |
552 | #define DISK_UTIL_MSEC (250) |
553 | ||
6a0106a0 JA |
554 | #ifndef min |
555 | #define min(a, b) ((a) < (b) ? (a) : (b)) | |
556 | #endif | |
a00735e6 JA |
557 | #ifndef max |
558 | #define max(a, b) ((a) > (b) ? (a) : (b)) | |
559 | #endif | |
6a0106a0 | 560 | |
6796209a JA |
561 | /* |
562 | * Log exports | |
563 | */ | |
72cb971b | 564 | extern int __must_check read_iolog_get(struct thread_data *, struct io_u *); |
6796209a | 565 | extern void write_iolog_put(struct thread_data *, struct io_u *); |
72cb971b | 566 | extern int __must_check init_iolog(struct thread_data *td); |
6796209a JA |
567 | extern void log_io_piece(struct thread_data *, struct io_u *); |
568 | extern void prune_io_piece_log(struct thread_data *); | |
569 | extern void write_iolog_close(struct thread_data *); | |
570 | ||
571 | /* | |
572 | * Logging | |
573 | */ | |
1e97cce9 JA |
574 | extern void add_clat_sample(struct thread_data *, enum fio_ddir, unsigned long); |
575 | extern void add_slat_sample(struct thread_data *, enum fio_ddir, unsigned long); | |
576 | extern void add_bw_sample(struct thread_data *, enum fio_ddir, struct timeval *); | |
6796209a JA |
577 | extern void show_run_stats(void); |
578 | extern void init_disk_util(struct thread_data *); | |
579 | extern void update_rusage_stat(struct thread_data *); | |
580 | extern void update_io_ticks(void); | |
581 | extern void disk_util_timer_arm(void); | |
8914a9d8 JA |
582 | extern void setup_log(struct io_log **); |
583 | extern void finish_log(struct thread_data *, struct io_log *, const char *); | |
bb3884d8 | 584 | extern void __finish_log(struct io_log *, const char *); |
8914a9d8 | 585 | extern int setup_rate(struct thread_data *); |
bb3884d8 JA |
586 | extern struct io_log *agg_io_log[2]; |
587 | extern int write_bw_log; | |
588 | extern void add_agg_sample(unsigned long, enum fio_ddir); | |
6796209a JA |
589 | |
590 | /* | |
591 | * Time functions | |
592 | */ | |
593 | extern unsigned long utime_since(struct timeval *, struct timeval *); | |
69008999 | 594 | extern unsigned long utime_since_now(struct timeval *); |
6796209a JA |
595 | extern unsigned long mtime_since(struct timeval *, struct timeval *); |
596 | extern unsigned long mtime_since_now(struct timeval *); | |
597 | extern unsigned long time_since_now(struct timeval *); | |
263e529f | 598 | extern unsigned long mtime_since_genesis(void); |
b990b5c0 | 599 | extern void __usec_sleep(unsigned int); |
6796209a | 600 | extern void usec_sleep(struct thread_data *, unsigned long); |
413dd459 | 601 | extern void rate_throttle(struct thread_data *, unsigned long, unsigned int); |
6043c579 | 602 | extern void fill_start_time(struct timeval *); |
02bcaa8c | 603 | extern void fio_gettime(struct timeval *, void *); |
a2f77c9f | 604 | extern void set_genesis_time(void); |
6796209a | 605 | |
8914a9d8 JA |
606 | /* |
607 | * Init functions | |
608 | */ | |
72cb971b JA |
609 | extern int __must_check parse_options(int, char **); |
610 | extern int __must_check init_random_state(struct thread_data *); | |
8914a9d8 | 611 | |
53cdc686 JA |
612 | /* |
613 | * File setup/shutdown | |
614 | */ | |
615 | extern void close_files(struct thread_data *); | |
b2fdda43 JA |
616 | extern int __must_check setup_files(struct thread_data *); |
617 | extern int __must_check open_files(struct thread_data *); | |
618 | extern int __must_check file_invalidate_cache(struct thread_data *, struct fio_file *); | |
53cdc686 | 619 | |
263e529f JA |
620 | /* |
621 | * ETA/status stuff | |
622 | */ | |
623 | extern void print_thread_status(void); | |
624 | extern void print_status_init(int); | |
625 | ||
626 | /* | |
627 | * Thread life cycle. Once a thread has a runstate beyond TD_INITIALIZED, it | |
628 | * will never back again. It may cycle between running/verififying/fsyncing. | |
629 | * Once the thread reaches TD_EXITED, it is just waiting for the core to | |
630 | * reap it. | |
631 | */ | |
632 | enum { | |
633 | TD_NOT_CREATED = 0, | |
634 | TD_CREATED, | |
635 | TD_INITIALIZED, | |
636 | TD_RUNNING, | |
637 | TD_VERIFYING, | |
638 | TD_FSYNCING, | |
639 | TD_EXITED, | |
640 | TD_REAPED, | |
641 | }; | |
642 | ||
e29d1b70 JA |
643 | /* |
644 | * Verify helpers | |
645 | */ | |
646 | extern void populate_verify_io_u(struct thread_data *, struct io_u *); | |
b2fdda43 JA |
647 | extern int __must_check get_next_verify(struct thread_data *td, struct io_u *); |
648 | extern int __must_check verify_io_u(struct io_u *); | |
e29d1b70 | 649 | |
2f9ade3c JA |
650 | /* |
651 | * Memory helpers | |
652 | */ | |
b2fdda43 | 653 | extern int __must_check fio_pin_memory(void); |
2f9ade3c | 654 | extern void fio_unpin_memory(void); |
b2fdda43 | 655 | extern int __must_check allocate_io_mem(struct thread_data *); |
2f9ade3c JA |
656 | extern void free_io_mem(struct thread_data *); |
657 | ||
10ba535a JA |
658 | /* |
659 | * io unit handling | |
660 | */ | |
661 | #define queue_full(td) list_empty(&(td)->io_u_freelist) | |
662 | extern struct io_u *__get_io_u(struct thread_data *); | |
3d7c391d | 663 | extern struct io_u *get_io_u(struct thread_data *); |
10ba535a | 664 | extern void put_io_u(struct thread_data *, struct io_u *); |
755200a3 | 665 | extern void requeue_io_u(struct thread_data *, struct io_u **); |
d7762cf8 JA |
666 | extern long __must_check io_u_sync_complete(struct thread_data *, struct io_u *); |
667 | extern long __must_check io_u_queued_complete(struct thread_data *, int); | |
7e77dd02 | 668 | extern void io_u_queued(struct thread_data *, struct io_u *); |
433afcb4 JA |
669 | extern void io_u_init_timeout(void); |
670 | extern void io_u_set_timeout(struct thread_data *); | |
10ba535a JA |
671 | |
672 | /* | |
673 | * io engine entry points | |
674 | */ | |
b2fdda43 JA |
675 | extern int __must_check td_io_init(struct thread_data *); |
676 | extern int __must_check td_io_prep(struct thread_data *, struct io_u *); | |
677 | extern int __must_check td_io_queue(struct thread_data *, struct io_u *); | |
678 | extern int __must_check td_io_sync(struct thread_data *, struct fio_file *); | |
679 | extern int __must_check td_io_getevents(struct thread_data *, int, int, struct timespec *); | |
680 | extern int __must_check td_io_commit(struct thread_data *); | |
10ba535a | 681 | |
bbfd6b00 JA |
682 | /* |
683 | * This is a pretty crappy semaphore implementation, but with the use that fio | |
684 | * has (just signalling start/go conditions), it doesn't have to be better. | |
685 | * Naturally this would not work for any type of contended semaphore or | |
686 | * for real locking. | |
687 | */ | |
1056eaad | 688 | static inline void fio_sem_init(volatile int *sem, int val) |
bbfd6b00 JA |
689 | { |
690 | *sem = val; | |
691 | } | |
692 | ||
1056eaad | 693 | static inline void fio_sem_down(volatile int *sem) |
bbfd6b00 JA |
694 | { |
695 | while (*sem == 0) | |
696 | usleep(10000); | |
697 | ||
698 | (*sem)--; | |
699 | } | |
700 | ||
1056eaad | 701 | static inline void fio_sem_up(volatile int *sem) |
bbfd6b00 JA |
702 | { |
703 | (*sem)++; | |
704 | } | |
705 | ||
3b70d7e5 JA |
706 | /* |
707 | * If logging output to a file, stderr should go to both stderr and f_err | |
708 | */ | |
709 | #define log_err(args...) do { \ | |
710 | fprintf(f_err, ##args); \ | |
711 | if (f_err != stderr) \ | |
712 | fprintf(stderr, ##args); \ | |
713 | } while (0) | |
714 | ||
2866c82d | 715 | struct ioengine_ops { |
5f350952 | 716 | struct list_head list; |
2866c82d JA |
717 | char name[16]; |
718 | int version; | |
719 | int flags; | |
ea2877a4 | 720 | int (*setup)(struct thread_data *); |
2866c82d JA |
721 | int (*init)(struct thread_data *); |
722 | int (*prep)(struct thread_data *, struct io_u *); | |
723 | int (*queue)(struct thread_data *, struct io_u *); | |
755200a3 | 724 | int (*commit)(struct thread_data *); |
2866c82d JA |
725 | int (*getevents)(struct thread_data *, int, int, struct timespec *); |
726 | struct io_u *(*event)(struct thread_data *, int); | |
727 | int (*cancel)(struct thread_data *, struct io_u *); | |
728 | void (*cleanup)(struct thread_data *); | |
2866c82d JA |
729 | void *data; |
730 | void *dlhandle; | |
36167d82 | 731 | unsigned long priv; |
2866c82d JA |
732 | }; |
733 | ||
755200a3 | 734 | #define FIO_IOOPS_VERSION 5 |
2866c82d | 735 | |
b4692828 | 736 | extern struct ioengine_ops *load_ioengine(struct thread_data *, const char *); |
b2fdda43 | 737 | extern void register_ioengine(struct ioengine_ops *); |
5f350952 | 738 | extern void unregister_ioengine(struct ioengine_ops *); |
2866c82d JA |
739 | extern void close_ioengine(struct thread_data *); |
740 | ||
741 | /* | |
742 | * Mark unused variables passed to ops functions as unused, to silence gcc | |
743 | */ | |
744 | #define fio_unused __attribute((__unused__)) | |
5f350952 JA |
745 | #define fio_init __attribute__((constructor)) |
746 | #define fio_exit __attribute__((destructor)) | |
2866c82d | 747 | |
34572e28 JA |
748 | #define for_each_td(td, i) \ |
749 | for ((i) = 0, (td) = &threads[0]; (i) < (int) thread_number; (i)++, (td)++) | |
53cdc686 | 750 | #define for_each_file(td, f, i) \ |
34572e28 | 751 | for ((i) = 0, (f) = &(td)->files[0]; (i) < (int) (td)->nr_files; (i)++, (f)++) |
53cdc686 | 752 | |
0032bf9f JA |
753 | #define fio_assert(td, cond) do { \ |
754 | if (!(cond)) { \ | |
340fd243 | 755 | int *__foo = NULL; \ |
0032bf9f JA |
756 | fprintf(stderr, "file:%s:%d, assert %s failed\n", __FILE__, __LINE__, #cond); \ |
757 | (td)->runstate = TD_EXITED; \ | |
437c9b71 | 758 | (td)->error = EFAULT; \ |
340fd243 | 759 | *__foo = 0; \ |
0032bf9f JA |
760 | } \ |
761 | } while (0) | |
762 | ||
ebac4655 | 763 | #endif |