client: Fix a memory leak in an error path
[fio.git] / file.h
... / ...
CommitLineData
1#ifndef FIO_FILE_H
2#define FIO_FILE_H
3
4#include <string.h>
5#include "compiler/compiler.h"
6#include "io_ddir.h"
7#include "flist.h"
8#include "lib/zipf.h"
9#include "lib/axmap.h"
10#include "lib/lfsr.h"
11#include "lib/gauss.h"
12
13/* Forward declarations */
14struct zoned_block_device_info;
15
16/*
17 * The type of object we are working on
18 */
19enum fio_filetype {
20 FIO_TYPE_FILE = 1, /* plain file */
21 FIO_TYPE_BLOCK, /* block device */
22 FIO_TYPE_CHAR, /* character device */
23 FIO_TYPE_PIPE, /* pipe */
24};
25
26enum fio_file_flags {
27 FIO_FILE_open = 1 << 0, /* file is open */
28 FIO_FILE_closing = 1 << 1, /* file being closed */
29 FIO_FILE_extend = 1 << 2, /* needs extend */
30 FIO_FILE_done = 1 << 3, /* io completed to this file */
31 FIO_FILE_size_known = 1 << 4, /* size has been set */
32 FIO_FILE_hashed = 1 << 5, /* file is on hash */
33 FIO_FILE_partial_mmap = 1 << 6, /* can't do full mmap */
34 FIO_FILE_axmap = 1 << 7, /* uses axmap */
35 FIO_FILE_lfsr = 1 << 8, /* lfsr is used */
36};
37
38enum file_lock_mode {
39 FILE_LOCK_NONE,
40 FILE_LOCK_EXCLUSIVE,
41 FILE_LOCK_READWRITE,
42};
43
44/*
45 * How fio chooses what file to service next. Choice of uniformly random, or
46 * some skewed random variants, or just sequentially go through them or
47 * roundrobing.
48 */
49enum {
50 FIO_FSERVICE_RANDOM = 1,
51 FIO_FSERVICE_RR = 2,
52 FIO_FSERVICE_SEQ = 3,
53 __FIO_FSERVICE_NONUNIFORM = 0x100,
54 FIO_FSERVICE_ZIPF = __FIO_FSERVICE_NONUNIFORM | 4,
55 FIO_FSERVICE_PARETO = __FIO_FSERVICE_NONUNIFORM | 5,
56 FIO_FSERVICE_GAUSS = __FIO_FSERVICE_NONUNIFORM | 6,
57
58 FIO_FSERVICE_SHIFT = 10,
59};
60
61/*
62 * No pre-allocation when laying down files, or call posix_fallocate(), or
63 * call fallocate() with FALLOC_FL_KEEP_SIZE set.
64 */
65enum fio_fallocate_mode {
66 FIO_FALLOCATE_NONE = 1,
67 FIO_FALLOCATE_POSIX = 2,
68 FIO_FALLOCATE_KEEP_SIZE = 3,
69 FIO_FALLOCATE_NATIVE = 4,
70 FIO_FALLOCATE_TRUNCATE = 5,
71};
72
73/*
74 * Each thread_data structure has a number of files associated with it,
75 * this structure holds state information for a single file.
76 */
77struct fio_file {
78 struct flist_head hash_list;
79 enum fio_filetype filetype;
80
81 int fd;
82 int shadow_fd;
83#ifdef WIN32
84 HANDLE hFile;
85 HANDLE ioCP;
86#endif
87
88 /*
89 * filename and possible memory mapping
90 */
91 unsigned int major, minor;
92 int fileno;
93 char *file_name;
94
95 /*
96 * size of the file, offset into file, and io size from that offset
97 * (be aware io_size is different from thread_options::io_size)
98 */
99 uint64_t real_file_size;
100 uint64_t file_offset;
101 uint64_t io_size;
102
103 /*
104 * Zoned block device information. See also zonemode=zbd.
105 */
106 struct zoned_block_device_info *zbd_info;
107 /* zonemode=zbd working area */
108 uint32_t min_zone; /* inclusive */
109 uint32_t max_zone; /* exclusive */
110
111 /*
112 * Track last end and last start of IO for a given data direction
113 */
114 uint64_t last_pos[DDIR_RWDIR_CNT];
115 uint64_t last_start[DDIR_RWDIR_CNT];
116
117 uint64_t first_write;
118 uint64_t last_write;
119
120 /*
121 * Tracks the last iodepth number of completed writes, if data
122 * verification is enabled
123 */
124 uint64_t *last_write_comp;
125 unsigned int last_write_idx;
126
127 /*
128 * For use by the io engine for offset or private data storage
129 */
130 union {
131 uint64_t engine_pos;
132 void *engine_data;
133 };
134
135 /*
136 * if io is protected by a semaphore, this is set
137 */
138 union {
139 struct fio_sem *lock;
140 struct fio_rwlock *rwlock;
141 };
142
143 /*
144 * block map or LFSR for random io
145 */
146 union {
147 struct axmap *io_axmap;
148 struct fio_lfsr lfsr;
149 };
150
151 /*
152 * Used for zipf random distribution
153 */
154 union {
155 struct zipf_state zipf;
156 struct gauss_state gauss;
157 };
158
159 int references;
160 enum fio_file_flags flags;
161
162 struct disk_util *du;
163};
164
165#define FILE_ENG_DATA(f) ((f)->engine_data)
166#define FILE_SET_ENG_DATA(f, data) ((f)->engine_data = (data))
167
168#define FILE_FLAG_FNS(name) \
169static inline void fio_file_set_##name(struct fio_file *f) \
170{ \
171 (f)->flags = (enum fio_file_flags) ((f)->flags | FIO_FILE_##name); \
172} \
173static inline void fio_file_clear_##name(struct fio_file *f) \
174{ \
175 (f)->flags = (enum fio_file_flags) ((f)->flags & ~FIO_FILE_##name); \
176} \
177static inline int fio_file_##name(struct fio_file *f) \
178{ \
179 return ((f)->flags & FIO_FILE_##name) != 0; \
180}
181
182FILE_FLAG_FNS(open);
183FILE_FLAG_FNS(closing);
184FILE_FLAG_FNS(extend);
185FILE_FLAG_FNS(done);
186FILE_FLAG_FNS(size_known);
187FILE_FLAG_FNS(hashed);
188FILE_FLAG_FNS(partial_mmap);
189FILE_FLAG_FNS(axmap);
190FILE_FLAG_FNS(lfsr);
191#undef FILE_FLAG_FNS
192
193/*
194 * File setup/shutdown
195 */
196struct thread_data;
197extern void close_files(struct thread_data *);
198extern void close_and_free_files(struct thread_data *);
199extern uint64_t get_start_offset(struct thread_data *, struct fio_file *);
200extern int __must_check setup_files(struct thread_data *);
201extern int __must_check file_invalidate_cache(struct thread_data *, struct fio_file *);
202#ifdef __cplusplus
203extern "C" {
204#endif
205extern int __must_check generic_open_file(struct thread_data *, struct fio_file *);
206extern int __must_check generic_close_file(struct thread_data *, struct fio_file *);
207extern int __must_check generic_get_file_size(struct thread_data *, struct fio_file *);
208#ifdef __cplusplus
209}
210#endif
211extern int __must_check file_lookup_open(struct fio_file *f, int flags);
212extern bool __must_check pre_read_files(struct thread_data *);
213extern unsigned long long get_rand_file_size(struct thread_data *td);
214extern int add_file(struct thread_data *, const char *, int, int);
215extern int add_file_exclusive(struct thread_data *, const char *);
216extern void get_file(struct fio_file *);
217extern int __must_check put_file(struct thread_data *, struct fio_file *);
218extern void put_file_log(struct thread_data *, struct fio_file *);
219extern void lock_file(struct thread_data *, struct fio_file *, enum fio_ddir);
220extern void unlock_file(struct thread_data *, struct fio_file *);
221extern void unlock_file_all(struct thread_data *, struct fio_file *);
222extern int add_dir_files(struct thread_data *, const char *);
223extern bool init_random_map(struct thread_data *);
224extern void dup_files(struct thread_data *, struct thread_data *);
225extern int get_fileno(struct thread_data *, const char *);
226extern void free_release_files(struct thread_data *);
227extern void filesetup_mem_free(void);
228extern void fio_file_reset(struct thread_data *, struct fio_file *);
229extern bool fio_files_done(struct thread_data *);
230extern bool exists_and_not_regfile(const char *);
231extern int fio_set_directio(struct thread_data *, struct fio_file *);
232
233#endif