2 * UCW Library -- FastIO on files with run-time parametrization
4 * (c) 2007 Pavel Charvat <pchar@ucw.cz>
5 * (c) 2007 Martin Mares <mj@ucw.cz>
7 * This software may be freely distributed and used according to the terms
8 * of the GNU Lesser General Public License.
14 #include "ucw/fastbuf.h"
19 struct fb_params fbpar_def = {
26 fbpar_cf_commit(struct fb_params *p UNUSED)
28 #ifndef CONFIG_UCW_THREADS
29 if (p->type == FB_DIRECT)
30 return "Direct I/O is supported only with CONFIG_UCW_THREADS";
33 if (p->type == FB_DIRECT)
34 return "Direct I/O is disabled";
39 struct cf_section fbpar_cf = {
40 # define F(x) PTR_TO(struct fb_params, x)
41 CF_TYPE(struct fb_params),
42 CF_COMMIT(fbpar_cf_commit),
44 CF_LOOKUP("Type", (int *)F(type), ((const char * const []){"std", "direct", "mmap", NULL})),
45 CF_UNS("BufSize", F(buffer_size)),
46 CF_UNS("KeepBackBuf", F(keep_back_buf)),
47 CF_UNS("ReadAhead", F(read_ahead)),
48 CF_UNS("WriteBack", F(write_back)),
54 static struct cf_section fbpar_global_cf = {
56 CF_SECTION("Defaults", &fbpar_def, &fbpar_cf),
61 static void CONSTRUCTOR
62 fbpar_global_init(void)
64 cf_declare_section("FBParam", &fbpar_global_cf, 0);
67 static struct fastbuf *
68 bopen_fd_internal(int fd, struct fb_params *params, uns mode, const char *name)
73 sprintf(buf, "fd%d", fd);
81 fb = fbdir_open_fd_internal(fd, name, params->asio,
82 params->buffer_size ? : fbpar_def.buffer_size,
83 params->read_ahead ? : fbpar_def.read_ahead,
84 params->write_back ? : fbpar_def.write_back);
85 if (!~mode && !fbdir_cheat && ((int)(mode = fcntl(fd, F_GETFL)) < 0 || fcntl(fd, F_SETFL, mode | O_DIRECT)) < 0)
86 msg(L_WARN, "Cannot set O_DIRECT on fd %d: %m", fd);
90 fb = bfdopen_internal(fd, name,
91 params->buffer_size ? : fbpar_def.buffer_size);
92 if (params->keep_back_buf)
93 bconfig(fb, BCONFIG_KEEP_BACK_BUF, 1);
96 if (!~mode && (int)(mode = fcntl(fd, F_GETFL)) < 0)
97 die("Cannot get flags of fd %d: %m", fd);
98 return bfmmopen_internal(fd, name, mode);
104 static struct fastbuf *
105 bopen_file_internal(const char *name, int mode, struct fb_params *params, int try)
110 if (params->type == FB_DIRECT && !fbdir_cheat)
113 if (params->type == FB_MMAP && (mode & O_ACCMODE) == O_WRONLY)
114 mode = (mode & ~O_ACCMODE) | O_RDWR;
115 int fd = ucw_open(name, mode, 0666);
120 die("Unable to %s file %s: %m", (mode & O_CREAT) ? "create" : "open", name);
121 struct fastbuf *fb = bopen_fd_internal(fd, params, mode, name);
124 bseek(fb, 0, SEEK_END);
129 bopen_file(const char *name, int mode, struct fb_params *params)
131 return bopen_file_internal(name, mode, params, 0);
135 bopen_file_try(const char *name, int mode, struct fb_params *params)
137 return bopen_file_internal(name, mode, params, 1);
141 bopen_fd_name(int fd, struct fb_params *params, const char *name)
143 return bopen_fd_internal(fd, params ? : &fbpar_def, ~0U, name);
146 /* Function for use by individual file back-ends */
149 bclose_file_helper(struct fastbuf *f, int fd, int is_temp_file)
151 switch (is_temp_file)
154 if (unlink(f->name) < 0)
155 msg(L_ERROR, "unlink(%s): %m", f->name);
158 die("close(%s): %m", f->name);
162 /* Compatibility wrappers */
165 bopen_try(const char *name, uns mode, uns buflen)
167 return bopen_file_try(name, mode, &(struct fb_params){ .type = FB_STD, .buffer_size = buflen });
171 bopen(const char *name, uns mode, uns buflen)
173 return bopen_file(name, mode, &(struct fb_params){ .type = FB_STD, .buffer_size = buflen });
177 bfdopen(int fd, uns buflen)
179 return bopen_fd(fd, &(struct fb_params){ .type = FB_STD, .buffer_size = buflen });
183 bfdopen_shared(int fd, uns buflen)
185 struct fastbuf *f = bfdopen(fd, buflen);
186 bconfig(f, BCONFIG_IS_TEMP_FILE, 2);