fio: fix dlopen refcounting of dynamic engines
[fio.git] / diskutil.h
1 #ifndef FIO_DISKUTIL_H
2 #define FIO_DISKUTIL_H
3 #define FIO_DU_NAME_SZ          64
4
5 #include "helper_thread.h"
6 #include "fio_sem.h"
7
8 struct disk_util_stats {
9         uint64_t ios[2];
10         uint64_t merges[2];
11         uint64_t sectors[2];
12         uint64_t ticks[2];
13         uint64_t io_ticks;
14         uint64_t time_in_queue;
15         uint64_t msec;
16 };
17
18 /*
19  * Disk utils as read in /sys/block/<dev>/stat
20  */
21 struct disk_util_stat {
22         uint8_t name[FIO_DU_NAME_SZ];
23         struct disk_util_stats s;
24 };
25
26 struct disk_util_agg {
27         uint64_t ios[2];
28         uint64_t merges[2];
29         uint64_t sectors[2];
30         uint64_t ticks[2];
31         uint64_t io_ticks;
32         uint64_t time_in_queue;
33         uint32_t slavecount;
34         uint32_t pad;
35         fio_fp64_t max_util;
36 };
37
38 /*
39  * Per-device disk util management
40  */
41 struct disk_util {
42         struct flist_head list;
43         /* If this disk is a slave, hook it into the master's
44          * list using this head.
45          */
46         struct flist_head slavelist;
47
48         char *sysfs_root;
49         char path[PATH_MAX];
50         int major, minor;
51
52         struct disk_util_stat dus;
53         struct disk_util_stat last_dus;
54
55         struct disk_util_agg agg;
56
57         /* For software raids, this entry maintains pointers to the
58          * entries for the slave devices. The disk_util entries for
59          * the slaves devices should primarily be maintained through
60          * the disk_list list, i.e. for memory allocation and
61          * de-allocation, etc. Whereas this list should be used only
62          * for aggregating a software RAID's disk util figures.
63          */
64         struct flist_head slaves;
65
66         struct timespec time;
67
68         struct fio_sem *lock;
69         unsigned long users;
70 };
71
72 static inline void disk_util_mod(struct disk_util *du, int val)
73 {
74         if (du) {
75                 struct flist_head *n;
76
77                 fio_sem_down(du->lock);
78                 du->users += val;
79
80                 flist_for_each(n, &du->slavelist) {
81                         struct disk_util *slave;
82
83                         slave = flist_entry(n, struct disk_util, slavelist);
84                         slave->users += val;
85                 }
86                 fio_sem_up(du->lock);
87         }
88 }
89 static inline void disk_util_inc(struct disk_util *du)
90 {
91         disk_util_mod(du, 1);
92 }
93
94 static inline void disk_util_dec(struct disk_util *du)
95 {
96         disk_util_mod(du, -1);
97 }
98
99 #define DISK_UTIL_MSEC  (250)
100
101 extern struct flist_head disk_list;
102
103 /*
104  * disk util stuff
105  */
106 #ifdef FIO_HAVE_DISK_UTIL
107 extern void init_disk_util(struct thread_data *);
108 extern int update_io_ticks(void);
109 extern void setup_disk_util(void);
110 extern void disk_util_prune_entries(void);
111 #else
112 /* keep this as a function to avoid a warning in handle_du() */
113 #define disk_util_prune_entries()
114 #define init_disk_util(td)
115 #define setup_disk_util()
116
117 static inline int update_io_ticks(void)
118 {
119         return helper_should_exit();
120 }
121 #endif
122
123 #endif