Make it possible to limit the number of open zones
[fio.git] / fio.h
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>
9 #include <errno.h>
10 #include <stdlib.h>
11 #include <stdio.h>
12 #include <unistd.h>
13 #include <string.h>
14 #include <inttypes.h>
15 #include <assert.h>
16
17 #include "compiler/compiler.h"
18 #include "thread_options.h"
19 #include "flist.h"
20 #include "fifo.h"
21 #include "arch/arch.h"
22 #include "os/os.h"
23 #include "log.h"
24 #include "debug.h"
25 #include "file.h"
26 #include "io_ddir.h"
27 #include "ioengines.h"
28 #include "iolog.h"
29 #include "helpers.h"
30 #include "minmax.h"
31 #include "options.h"
32 #include "profile.h"
33 #include "fio_time.h"
34 #include "gettime.h"
35 #include "oslib/getopt.h"
36 #include "lib/rand.h"
37 #include "lib/rbtree.h"
38 #include "lib/num2str.h"
39 #include "client.h"
40 #include "server.h"
41 #include "stat.h"
42 #include "flow.h"
43 #include "io_u.h"
44 #include "io_u_queue.h"
45 #include "workqueue.h"
46 #include "steadystate.h"
47 #include "lib/nowarn_snprintf.h"
48
49 #ifdef CONFIG_SOLARISAIO
50 #include <sys/asynch.h>
51 #endif
52
53 #ifdef CONFIG_LIBNUMA
54 #include <linux/mempolicy.h>
55 #include <numa.h>
56
57 /*
58  * "local" is pseudo-policy
59  */
60 #define MPOL_LOCAL MPOL_MAX
61 #endif
62
63 #ifdef CONFIG_CUDA
64 #include <cuda.h>
65 #endif
66
67 struct fio_sem;
68
69 /*
70  * offset generator types
71  */
72 enum {
73         RW_SEQ_SEQ      = 0,
74         RW_SEQ_IDENT,
75 };
76
77 enum {
78         __TD_F_VER_BACKLOG      = 0,
79         __TD_F_TRIM_BACKLOG,
80         __TD_F_READ_IOLOG,
81         __TD_F_REFILL_BUFFERS,
82         __TD_F_SCRAMBLE_BUFFERS,
83         __TD_F_DO_VERIFY,
84         __TD_F_PROFILE_OPS,
85         __TD_F_COMPRESS,
86         __TD_F_COMPRESS_LOG,
87         __TD_F_VSTATE_SAVED,
88         __TD_F_NEED_LOCK,
89         __TD_F_CHILD,
90         __TD_F_NO_PROGRESS,
91         __TD_F_REGROW_LOGS,
92         __TD_F_MMAP_KEEP,
93         __TD_F_DIRS_CREATED,
94         __TD_F_CHECK_RATE,
95         __TD_F_LAST,            /* not a real bit, keep last */
96 };
97
98 enum {
99         TD_F_VER_BACKLOG        = 1U << __TD_F_VER_BACKLOG,
100         TD_F_TRIM_BACKLOG       = 1U << __TD_F_TRIM_BACKLOG,
101         TD_F_READ_IOLOG         = 1U << __TD_F_READ_IOLOG,
102         TD_F_REFILL_BUFFERS     = 1U << __TD_F_REFILL_BUFFERS,
103         TD_F_SCRAMBLE_BUFFERS   = 1U << __TD_F_SCRAMBLE_BUFFERS,
104         TD_F_DO_VERIFY          = 1U << __TD_F_DO_VERIFY,
105         TD_F_PROFILE_OPS        = 1U << __TD_F_PROFILE_OPS,
106         TD_F_COMPRESS           = 1U << __TD_F_COMPRESS,
107         TD_F_COMPRESS_LOG       = 1U << __TD_F_COMPRESS_LOG,
108         TD_F_VSTATE_SAVED       = 1U << __TD_F_VSTATE_SAVED,
109         TD_F_NEED_LOCK          = 1U << __TD_F_NEED_LOCK,
110         TD_F_CHILD              = 1U << __TD_F_CHILD,
111         TD_F_NO_PROGRESS        = 1U << __TD_F_NO_PROGRESS,
112         TD_F_REGROW_LOGS        = 1U << __TD_F_REGROW_LOGS,
113         TD_F_MMAP_KEEP          = 1U << __TD_F_MMAP_KEEP,
114         TD_F_DIRS_CREATED       = 1U << __TD_F_DIRS_CREATED,
115         TD_F_CHECK_RATE         = 1U << __TD_F_CHECK_RATE,
116 };
117
118 enum {
119         FIO_RAND_BS_OFF         = 0,
120         FIO_RAND_BS1_OFF,
121         FIO_RAND_BS2_OFF,
122         FIO_RAND_VER_OFF,
123         FIO_RAND_MIX_OFF,
124         FIO_RAND_FILE_OFF,
125         FIO_RAND_BLOCK_OFF,
126         FIO_RAND_FILE_SIZE_OFF,
127         FIO_RAND_TRIM_OFF,
128         FIO_RAND_BUF_OFF,
129         FIO_RAND_SEQ_RAND_READ_OFF,
130         FIO_RAND_SEQ_RAND_WRITE_OFF,
131         FIO_RAND_SEQ_RAND_TRIM_OFF,
132         FIO_RAND_START_DELAY,
133         FIO_DEDUPE_OFF,
134         FIO_RAND_POISSON_OFF,
135         FIO_RAND_ZONE_OFF,
136         FIO_RAND_POISSON2_OFF,
137         FIO_RAND_POISSON3_OFF,
138         FIO_RAND_NR_OFFS,
139 };
140
141 enum {
142         IO_MODE_INLINE = 0,
143         IO_MODE_OFFLOAD = 1,
144
145         RATE_PROCESS_LINEAR = 0,
146         RATE_PROCESS_POISSON = 1,
147 };
148
149 enum {
150         F_ADV_NONE = 0,
151         F_ADV_TYPE,
152         F_ADV_RANDOM,
153         F_ADV_SEQUENTIAL,
154 };
155
156 /*
157  * Per-thread/process specific data. Only used for the network client
158  * for now.
159  */
160 void sk_out_assign(struct sk_out *);
161 void sk_out_drop(void);
162
163 struct zone_split_index {
164         uint8_t size_perc;
165         uint8_t size_perc_prev;
166         uint64_t size;
167         uint64_t size_prev;
168 };
169
170 #define FIO_MAX_OPEN_ZBD_ZONES 128
171
172 /*
173  * This describes a single thread/process executing a fio job.
174  */
175 struct thread_data {
176         struct flist_head opt_list;
177         unsigned long flags;
178         struct thread_options o;
179         void *eo;
180         pthread_t thread;
181         unsigned int thread_number;
182         unsigned int subjob_number;
183         unsigned int groupid;
184         struct thread_stat ts __attribute__ ((aligned(8)));
185
186         int client_type;
187
188         struct io_log *slat_log;
189         struct io_log *clat_log;
190         struct io_log *clat_hist_log;
191         struct io_log *lat_log;
192         struct io_log *bw_log;
193         struct io_log *iops_log;
194
195         struct workqueue log_compress_wq;
196
197         struct thread_data *parent;
198
199         uint64_t stat_io_bytes[DDIR_RWDIR_CNT];
200         struct timespec bw_sample_time;
201
202         uint64_t stat_io_blocks[DDIR_RWDIR_CNT];
203         struct timespec iops_sample_time;
204
205         volatile int update_rusage;
206         struct fio_sem *rusage_sem;
207         struct rusage ru_start;
208         struct rusage ru_end;
209
210         struct fio_file **files;
211         unsigned char *file_locks;
212         unsigned int files_size;
213         unsigned int files_index;
214         unsigned int nr_open_files;
215         unsigned int nr_done_files;
216         union {
217                 unsigned int next_file;
218                 struct frand_state next_file_state;
219         };
220         union {
221                 struct zipf_state next_file_zipf;
222                 struct gauss_state next_file_gauss;
223         };
224         union {
225                 double zipf_theta;
226                 double pareto_h;
227                 double gauss_dev;
228         };
229         int error;
230         int sig;
231         int done;
232         int stop_io;
233         pid_t pid;
234         char *orig_buffer;
235         size_t orig_buffer_size;
236         volatile int runstate;
237         volatile bool terminate;
238         bool last_was_sync;
239         enum fio_ddir last_ddir;
240
241         int mmapfd;
242
243         void *iolog_buf;
244         FILE *iolog_f;
245
246         unsigned long rand_seeds[FIO_RAND_NR_OFFS];
247
248         struct frand_state bsrange_state[DDIR_RWDIR_CNT];
249         struct frand_state verify_state;
250         struct frand_state trim_state;
251         struct frand_state delay_state;
252
253         struct frand_state buf_state;
254         struct frand_state buf_state_prev;
255         struct frand_state dedupe_state;
256         struct frand_state zone_state;
257
258         struct zone_split_index **zone_state_index;
259
260         unsigned int verify_batch;
261         unsigned int trim_batch;
262
263         struct thread_io_list *vstate;
264
265         int shm_id;
266
267         /*
268          * IO engine hooks, contains everything needed to submit an io_u
269          * to any of the available IO engines.
270          */
271         struct ioengine_ops *io_ops;
272         int io_ops_init;
273
274         /*
275          * IO engine private data and dlhandle.
276          */
277         void *io_ops_data;
278         void *io_ops_dlhandle;
279
280         /*
281          * Queue depth of io_u's that fio MIGHT do
282          */
283         unsigned int cur_depth;
284
285         /*
286          * io_u's about to be committed
287          */
288         unsigned int io_u_queued;
289
290         /*
291          * io_u's submitted but not completed yet
292          */
293         unsigned int io_u_in_flight;
294
295         /*
296          * List of free and busy io_u's
297          */
298         struct io_u_ring io_u_requeues;
299         struct io_u_queue io_u_freelist;
300         struct io_u_queue io_u_all;
301         pthread_mutex_t io_u_lock;
302         pthread_cond_t free_cond;
303
304         /*
305          * async verify offload
306          */
307         struct flist_head verify_list;
308         pthread_t *verify_threads;
309         unsigned int nr_verify_threads;
310         pthread_cond_t verify_cond;
311         int verify_thread_exit;
312
313         /*
314          * Rate state
315          */
316         uint64_t rate_bps[DDIR_RWDIR_CNT];
317         uint64_t rate_next_io_time[DDIR_RWDIR_CNT];
318         unsigned long rate_bytes[DDIR_RWDIR_CNT];
319         unsigned long rate_blocks[DDIR_RWDIR_CNT];
320         unsigned long long rate_io_issue_bytes[DDIR_RWDIR_CNT];
321         struct timespec lastrate[DDIR_RWDIR_CNT];
322         int64_t last_usec[DDIR_RWDIR_CNT];
323         struct frand_state poisson_state[DDIR_RWDIR_CNT];
324
325         /*
326          * Enforced rate submission/completion workqueue
327          */
328         struct workqueue io_wq;
329
330         uint64_t total_io_size;
331         uint64_t fill_device_size;
332
333         /*
334          * Issue side
335          */
336         uint64_t io_issues[DDIR_RWDIR_CNT];
337         uint64_t io_issue_bytes[DDIR_RWDIR_CNT];
338         uint64_t loops;
339
340         /*
341          * Completions
342          */
343         uint64_t io_blocks[DDIR_RWDIR_CNT];
344         uint64_t this_io_blocks[DDIR_RWDIR_CNT];
345         uint64_t io_bytes[DDIR_RWDIR_CNT];
346         uint64_t this_io_bytes[DDIR_RWDIR_CNT];
347         uint64_t io_skip_bytes;
348         uint64_t zone_bytes;
349         struct fio_sem *sem;
350         uint64_t bytes_done[DDIR_RWDIR_CNT];
351
352         /*
353          * State for random io, a bitmap of blocks done vs not done
354          */
355         struct frand_state random_state;
356
357         struct timespec start;  /* start of this loop */
358         struct timespec epoch;  /* time job was started */
359         unsigned long long unix_epoch; /* Time job was started, unix epoch based. */
360         struct timespec last_issue;
361         long time_offset;
362         struct timespec ts_cache;
363         struct timespec terminate_time;
364         unsigned int ts_cache_nr;
365         unsigned int ts_cache_mask;
366         bool ramp_time_over;
367
368         /*
369          * Time since last latency_window was started
370          */
371         struct timespec latency_ts;
372         unsigned int latency_qd;
373         unsigned int latency_qd_high;
374         unsigned int latency_qd_low;
375         unsigned int latency_failed;
376         uint64_t latency_ios;
377         int latency_end_run;
378
379         /*
380          * read/write mixed workload state
381          */
382         struct frand_state rwmix_state;
383         unsigned long rwmix_issues;
384         enum fio_ddir rwmix_ddir;
385         unsigned int ddir_seq_nr;
386
387         /*
388          * rand/seq mixed workload state
389          */
390         struct frand_state seq_rand_state[DDIR_RWDIR_CNT];
391
392         /*
393          * IO history logs for verification. We use a tree for sorting,
394          * if we are overwriting. Otherwise just use a fifo.
395          */
396         struct rb_root io_hist_tree;
397         struct flist_head io_hist_list;
398         unsigned long io_hist_len;
399
400         /*
401          * For IO replaying
402          */
403         struct flist_head io_log_list;
404         FILE *io_log_rfile;
405         unsigned int io_log_current;
406         unsigned int io_log_checkmark;
407         unsigned int io_log_highmark;
408         struct timespec io_log_highmark_time;
409
410         /*
411          * For tracking/handling discards
412          */
413         struct flist_head trim_list;
414         unsigned long trim_entries;
415
416         /*
417          * for fileservice, how often to switch to a new file
418          */
419         unsigned int file_service_nr;
420         unsigned int file_service_left;
421         struct fio_file *file_service_file;
422
423         unsigned int sync_file_range_nr;
424
425         /*
426          * For generating file sizes
427          */
428         struct frand_state file_size_state;
429
430         /*
431          * Error counts
432          */
433         unsigned int total_err_count;
434         int first_error;
435
436         struct fio_flow *flow;
437
438         /*
439          * Can be overloaded by profiles
440          */
441         struct prof_io_ops prof_io_ops;
442         void *prof_data;
443
444         void *pinned_mem;
445
446         struct steadystate_data ss;
447
448         char verror[FIO_VERROR_SIZE];
449
450 #ifdef CONFIG_CUDA
451         /*
452          * for GPU memory management
453          */
454         int gpu_dev_cnt;
455         int gpu_dev_id;
456         CUdevice  cu_dev;
457         CUcontext cu_ctx;
458         CUdeviceptr dev_mem_ptr;
459 #endif  
460
461 };
462
463 /*
464  * when should interactive ETA output be generated
465  */
466 enum {
467         FIO_ETA_AUTO,
468         FIO_ETA_ALWAYS,
469         FIO_ETA_NEVER,
470 };
471
472 #define __td_verror(td, err, msg, func)                                 \
473         do {                                                            \
474                 unsigned int ____e = (err);                             \
475                 if ((td)->error)                                        \
476                         break;                                          \
477                 (td)->error = ____e;                                    \
478                 if (!(td)->first_error)                                 \
479                         nowarn_snprintf(td->verror, sizeof(td->verror), \
480                                         "file:%s:%d, func=%s, error=%s", \
481                                         __FILE__, __LINE__, (func), (msg)); \
482         } while (0)
483
484
485 #define td_clear_error(td)              do {            \
486         (td)->error = 0;                                \
487         if ((td)->parent)                               \
488                 (td)->parent->error = 0;                \
489 } while (0)
490
491 #define td_verror(td, err, func)        do {                    \
492         __td_verror((td), (err), strerror((err)), (func));      \
493         if ((td)->parent)                                       \
494                 __td_verror((td)->parent, (err), strerror((err)), (func)); \
495 } while (0)
496
497 #define td_vmsg(td, err, msg, func)     do {                    \
498         __td_verror((td), (err), (msg), (func));                \
499         if ((td)->parent)                                       \
500                 __td_verror((td)->parent, (err), (msg), (func));        \
501 } while (0)
502
503 #define __fio_stringify_1(x)    #x
504 #define __fio_stringify(x)      __fio_stringify_1(x)
505
506 extern int exitall_on_terminate;
507 extern unsigned int thread_number;
508 extern unsigned int stat_number;
509 extern int shm_id;
510 extern int groupid;
511 extern int output_format;
512 extern int append_terse_output;
513 extern int temp_stall_ts;
514 extern uintptr_t page_mask, page_size;
515 extern int read_only;
516 extern int eta_print;
517 extern int eta_new_line;
518 extern unsigned int eta_interval_msec;
519 extern unsigned long done_secs;
520 extern int fio_gtod_offload;
521 extern int fio_gtod_cpu;
522 extern enum fio_cs fio_clock_source;
523 extern int fio_clock_source_set;
524 extern int warnings_fatal;
525 extern int terse_version;
526 extern int is_backend;
527 extern int is_local_backend;
528 extern int nr_clients;
529 extern int log_syslog;
530 extern int status_interval;
531 extern const char fio_version_string[];
532 extern char *trigger_file;
533 extern char *trigger_cmd;
534 extern char *trigger_remote_cmd;
535 extern long long trigger_timeout;
536 extern char *aux_path;
537
538 extern struct thread_data *threads;
539
540 static inline bool is_running_backend(void)
541 {
542         return is_backend || is_local_backend;
543 }
544
545 extern bool eta_time_within_slack(unsigned int time);
546
547 static inline void fio_ro_check(const struct thread_data *td, struct io_u *io_u)
548 {
549         assert(!(io_u->ddir == DDIR_WRITE && !td_write(td)) &&
550                !(io_u->ddir == DDIR_TRIM && !td_trim(td)));
551 }
552
553 #define REAL_MAX_JOBS           4096
554
555 static inline bool should_fsync(struct thread_data *td)
556 {
557         if (td->last_was_sync)
558                 return false;
559         if (td_write(td) || td->o.override_sync)
560                 return true;
561
562         return false;
563 }
564
565 /*
566  * Init/option functions
567  */
568 extern int __must_check fio_init_options(void);
569 extern int __must_check parse_options(int, char **);
570 extern int parse_jobs_ini(char *, int, int, int);
571 extern int parse_cmd_line(int, char **, int);
572 extern int fio_backend(struct sk_out *);
573 extern void reset_fio_state(void);
574 extern void clear_io_state(struct thread_data *, int);
575 extern int fio_options_parse(struct thread_data *, char **, int);
576 extern void fio_keywords_init(void);
577 extern void fio_keywords_exit(void);
578 extern int fio_cmd_option_parse(struct thread_data *, const char *, char *);
579 extern int fio_cmd_ioengine_option_parse(struct thread_data *, const char *, char *);
580 extern void fio_fill_default_options(struct thread_data *);
581 extern int fio_show_option_help(const char *);
582 extern void fio_options_set_ioengine_opts(struct option *long_options, struct thread_data *td);
583 extern void fio_options_dup_and_init(struct option *);
584 extern char *fio_option_dup_subs(const char *);
585 extern void fio_options_mem_dupe(struct thread_data *);
586 extern void td_fill_rand_seeds(struct thread_data *);
587 extern void td_fill_verify_state_seed(struct thread_data *);
588 extern void add_job_opts(const char **, int);
589 extern int ioengine_load(struct thread_data *);
590 extern bool parse_dryrun(void);
591 extern int fio_running_or_pending_io_threads(void);
592 extern int fio_set_fd_nonblocking(int, const char *);
593 extern void sig_show_status(int sig);
594 extern struct thread_data *get_global_options(void);
595
596 extern uintptr_t page_mask;
597 extern uintptr_t page_size;
598 extern int initialize_fio(char *envp[]);
599 extern void deinitialize_fio(void);
600
601 #define FIO_GETOPT_JOB          0x89000000
602 #define FIO_GETOPT_IOENGINE     0x98000000
603 #define FIO_NR_OPTIONS          (FIO_MAX_OPTS + 128)
604
605 /*
606  * ETA/status stuff
607  */
608 extern void print_thread_status(void);
609 extern void print_status_init(int);
610 extern char *fio_uint_to_kmg(unsigned int val);
611
612 /*
613  * Thread life cycle. Once a thread has a runstate beyond TD_INITIALIZED, it
614  * will never back again. It may cycle between running/verififying/fsyncing.
615  * Once the thread reaches TD_EXITED, it is just waiting for the core to
616  * reap it.
617  */
618 enum {
619         TD_NOT_CREATED = 0,
620         TD_CREATED,
621         TD_INITIALIZED,
622         TD_RAMP,
623         TD_SETTING_UP,
624         TD_RUNNING,
625         TD_PRE_READING,
626         TD_VERIFYING,
627         TD_FSYNCING,
628         TD_FINISHING,
629         TD_EXITED,
630         TD_REAPED,
631         TD_LAST,
632         TD_NR,
633 };
634
635 #define TD_ENG_FLAG_SHIFT       17
636 #define TD_ENG_FLAG_MASK        ((1U << 17) - 1)
637
638 static inline void td_set_ioengine_flags(struct thread_data *td)
639 {
640         td->flags = (~(TD_ENG_FLAG_MASK << TD_ENG_FLAG_SHIFT) & td->flags) |
641                     (td->io_ops->flags << TD_ENG_FLAG_SHIFT);
642 }
643
644 static inline bool td_ioengine_flagged(struct thread_data *td,
645                                        enum fio_ioengine_flags flags)
646 {
647         return ((td->flags >> TD_ENG_FLAG_SHIFT) & flags) != 0;
648 }
649
650 extern void td_set_runstate(struct thread_data *, int);
651 extern int td_bump_runstate(struct thread_data *, int);
652 extern void td_restore_runstate(struct thread_data *, int);
653 extern const char *runstate_to_name(int runstate);
654
655 /*
656  * Allow 60 seconds for a job to quit on its own, otherwise reap with
657  * a vengeance.
658  */
659 #define FIO_REAP_TIMEOUT        300
660
661 #define TERMINATE_ALL           (-1U)
662 extern void fio_terminate_threads(unsigned int);
663 extern void fio_mark_td_terminate(struct thread_data *);
664
665 /*
666  * Memory helpers
667  */
668 extern int __must_check fio_pin_memory(struct thread_data *);
669 extern void fio_unpin_memory(struct thread_data *);
670 extern int __must_check allocate_io_mem(struct thread_data *);
671 extern void free_io_mem(struct thread_data *);
672 extern void free_threads_shm(void);
673
674 #ifdef FIO_INTERNAL
675 #define PTR_ALIGN(ptr, mask)    \
676         (char *) (((uintptr_t) (ptr) + (mask)) & ~(mask))
677 #endif
678
679 /*
680  * Reset stats after ramp time completes
681  */
682 extern void reset_all_stats(struct thread_data *);
683
684 extern int io_queue_event(struct thread_data *td, struct io_u *io_u, int *ret,
685                    enum fio_ddir ddir, uint64_t *bytes_issued, int from_verify,
686                    struct timespec *comp_time);
687
688 /*
689  * Latency target helpers
690  */
691 extern void lat_target_check(struct thread_data *);
692 extern void lat_target_init(struct thread_data *);
693 extern void lat_target_reset(struct thread_data *);
694
695 /*
696  * Iterates all threads/processes within all the defined jobs
697  */
698 #define for_each_td(td, i)      \
699         for ((i) = 0, (td) = &threads[0]; (i) < (int) thread_number; (i)++, (td)++)
700 #define for_each_file(td, f, i) \
701         if ((td)->files_index)                                          \
702                 for ((i) = 0, (f) = (td)->files[0];                     \
703                  (i) < (td)->o.nr_files && ((f) = (td)->files[i]) != NULL; \
704                  (i)++)
705
706 #define fio_assert(td, cond)    do {    \
707         if (!(cond)) {                  \
708                 int *__foo = NULL;      \
709                 fprintf(stderr, "file:%s:%d, assert %s failed\n", __FILE__, __LINE__, #cond);   \
710                 td_set_runstate((td), TD_EXITED);       \
711                 (td)->error = EFAULT;           \
712                 *__foo = 0;                     \
713         }       \
714 } while (0)
715
716 static inline bool fio_fill_issue_time(struct thread_data *td)
717 {
718         if (td->o.read_iolog_file ||
719             !td->o.disable_clat || !td->o.disable_slat || !td->o.disable_bw)
720                 return true;
721
722         return false;
723 }
724
725 static inline bool option_check_rate(struct thread_data *td, enum fio_ddir ddir)
726 {
727         struct thread_options *o = &td->o;
728
729         /*
730          * If some rate setting was given, we need to check it
731          */
732         if (o->rate[ddir] || o->ratemin[ddir] || o->rate_iops[ddir] ||
733             o->rate_iops_min[ddir])
734                 return true;
735
736         return false;
737 }
738
739 static inline bool __should_check_rate(struct thread_data *td)
740 {
741         return (td->flags & TD_F_CHECK_RATE) != 0;
742 }
743
744 static inline bool should_check_rate(struct thread_data *td)
745 {
746         if (!__should_check_rate(td))
747                 return false;
748
749         return ddir_rw_sum(td->bytes_done) != 0;
750 }
751
752 static inline unsigned long long td_max_bs(struct thread_data *td)
753 {
754         unsigned long long max_bs;
755
756         max_bs = max(td->o.max_bs[DDIR_READ], td->o.max_bs[DDIR_WRITE]);
757         return max(td->o.max_bs[DDIR_TRIM], max_bs);
758 }
759
760 static inline unsigned long long td_min_bs(struct thread_data *td)
761 {
762         unsigned long long min_bs;
763
764         min_bs = min(td->o.min_bs[DDIR_READ], td->o.min_bs[DDIR_WRITE]);
765         return min(td->o.min_bs[DDIR_TRIM], min_bs);
766 }
767
768 static inline bool td_async_processing(struct thread_data *td)
769 {
770         return (td->flags & TD_F_NEED_LOCK) != 0;
771 }
772
773 /*
774  * We currently only need to do locking if we have verifier threads
775  * accessing our internal structures too
776  */
777 static inline void td_io_u_lock(struct thread_data *td)
778 {
779         if (td_async_processing(td))
780                 pthread_mutex_lock(&td->io_u_lock);
781 }
782
783 static inline void td_io_u_unlock(struct thread_data *td)
784 {
785         if (td_async_processing(td))
786                 pthread_mutex_unlock(&td->io_u_lock);
787 }
788
789 static inline void td_io_u_free_notify(struct thread_data *td)
790 {
791         if (td_async_processing(td))
792                 pthread_cond_signal(&td->free_cond);
793 }
794
795 static inline void td_flags_clear(struct thread_data *td, unsigned int *flags,
796                                   unsigned int value)
797 {
798         if (!td_async_processing(td))
799                 *flags &= ~value;
800         else
801                 __sync_fetch_and_and(flags, ~value);
802 }
803
804 static inline void td_flags_set(struct thread_data *td, unsigned int *flags,
805                                 unsigned int value)
806 {
807         if (!td_async_processing(td))
808                 *flags |= value;
809         else
810                 __sync_fetch_and_or(flags, value);
811 }
812
813 extern const char *fio_get_arch_string(int);
814 extern const char *fio_get_os_string(int);
815
816 enum {
817         __FIO_OUTPUT_TERSE      = 0,
818         __FIO_OUTPUT_JSON       = 1,
819         __FIO_OUTPUT_NORMAL     = 2,
820         __FIO_OUTPUT_JSON_PLUS  = 3,
821         FIO_OUTPUT_NR           = 4,
822
823         FIO_OUTPUT_TERSE        = 1U << __FIO_OUTPUT_TERSE,
824         FIO_OUTPUT_JSON         = 1U << __FIO_OUTPUT_JSON,
825         FIO_OUTPUT_NORMAL       = 1U << __FIO_OUTPUT_NORMAL,
826         FIO_OUTPUT_JSON_PLUS    = 1U << __FIO_OUTPUT_JSON_PLUS,
827 };
828
829 enum {
830         FIO_RAND_DIST_RANDOM    = 0,
831         FIO_RAND_DIST_ZIPF,
832         FIO_RAND_DIST_PARETO,
833         FIO_RAND_DIST_GAUSS,
834         FIO_RAND_DIST_ZONED,
835         FIO_RAND_DIST_ZONED_ABS,
836 };
837
838 #define FIO_DEF_ZIPF            1.1
839 #define FIO_DEF_PARETO          0.2
840
841 enum {
842         FIO_RAND_GEN_TAUSWORTHE = 0,
843         FIO_RAND_GEN_LFSR,
844         FIO_RAND_GEN_TAUSWORTHE64,
845 };
846
847 enum {
848         FIO_CPUS_SHARED         = 0,
849         FIO_CPUS_SPLIT,
850 };
851
852 extern void exec_trigger(const char *);
853 extern void check_trigger_file(void);
854
855 #endif