Commit | Line | Data |
---|---|---|
f3bc08c5 MD |
1 | /* |
2 | * ring_buffer_vfs.c | |
3 | * | |
4 | * Copyright (C) 2009-2010 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com> | |
5 | * | |
6 | * Ring Buffer VFS file operations. | |
7 | * | |
8 | * Dual LGPL v2.1/GPL v2 license. | |
9 | */ | |
10 | ||
11 | #include <linux/module.h> | |
12 | #include <linux/fs.h> | |
13 | #include <linux/compat.h> | |
14 | ||
15 | #include "../../wrapper/ringbuffer/backend.h" | |
16 | #include "../../wrapper/ringbuffer/frontend.h" | |
17 | #include "../../wrapper/ringbuffer/vfs.h" | |
18 | #include "../../wrapper/poll.h" | |
19 | ||
20 | static int put_ulong(unsigned long val, unsigned long arg) | |
21 | { | |
22 | return put_user(val, (unsigned long __user *)arg); | |
23 | } | |
24 | ||
25 | #ifdef CONFIG_COMPAT | |
26 | static int compat_put_ulong(compat_ulong_t val, unsigned long arg) | |
27 | { | |
28 | return put_user(val, (compat_ulong_t __user *)compat_ptr(arg)); | |
29 | } | |
30 | #endif | |
31 | ||
32 | /** | |
33 | * lib_ring_buffer_open - ring buffer open file operation | |
34 | * @inode: opened inode | |
35 | * @file: opened file | |
36 | * | |
37 | * Open implementation. Makes sure only one open instance of a buffer is | |
38 | * done at a given moment. | |
39 | */ | |
40 | int lib_ring_buffer_open(struct inode *inode, struct file *file) | |
41 | { | |
42 | struct lib_ring_buffer *buf = inode->i_private; | |
43 | int ret; | |
44 | ||
45 | ret = lib_ring_buffer_open_read(buf); | |
46 | if (ret) | |
47 | return ret; | |
48 | ||
49 | file->private_data = buf; | |
50 | ret = nonseekable_open(inode, file); | |
51 | if (ret) | |
52 | goto release_read; | |
53 | return 0; | |
54 | ||
55 | release_read: | |
56 | lib_ring_buffer_release_read(buf); | |
57 | return ret; | |
58 | } | |
59 | ||
60 | /** | |
61 | * lib_ring_buffer_release - ring buffer release file operation | |
62 | * @inode: opened inode | |
63 | * @file: opened file | |
64 | * | |
65 | * Release implementation. | |
66 | */ | |
67 | int lib_ring_buffer_release(struct inode *inode, struct file *file) | |
68 | { | |
69 | struct lib_ring_buffer *buf = file->private_data; | |
70 | ||
71 | lib_ring_buffer_release_read(buf); | |
72 | ||
73 | return 0; | |
74 | } | |
75 | ||
76 | /** | |
77 | * lib_ring_buffer_poll - ring buffer poll file operation | |
78 | * @filp: the file | |
79 | * @wait: poll table | |
80 | * | |
81 | * Poll implementation. | |
82 | */ | |
83 | unsigned int lib_ring_buffer_poll(struct file *filp, poll_table *wait) | |
84 | { | |
85 | unsigned int mask = 0; | |
86 | struct lib_ring_buffer *buf = filp->private_data; | |
87 | struct channel *chan = buf->backend.chan; | |
88 | const struct lib_ring_buffer_config *config = chan->backend.config; | |
254ec7bc | 89 | int finalized, disabled; |
f3bc08c5 MD |
90 | |
91 | if (filp->f_mode & FMODE_READ) { | |
a33e44a6 | 92 | poll_wait_set_exclusive(wait); |
f3bc08c5 MD |
93 | poll_wait(filp, &buf->read_wait, wait); |
94 | ||
95 | finalized = lib_ring_buffer_is_finalized(config, buf); | |
254ec7bc MD |
96 | disabled = lib_ring_buffer_channel_is_disabled(chan); |
97 | ||
f3bc08c5 MD |
98 | /* |
99 | * lib_ring_buffer_is_finalized() contains a smp_rmb() ordering | |
100 | * finalized load before offsets loads. | |
101 | */ | |
102 | WARN_ON(atomic_long_read(&buf->active_readers) != 1); | |
103 | retry: | |
254ec7bc MD |
104 | if (disabled) |
105 | return POLLERR; | |
106 | ||
f3bc08c5 MD |
107 | if (subbuf_trunc(lib_ring_buffer_get_offset(config, buf), chan) |
108 | - subbuf_trunc(lib_ring_buffer_get_consumed(config, buf), chan) | |
109 | == 0) { | |
110 | if (finalized) | |
111 | return POLLHUP; | |
112 | else { | |
113 | /* | |
114 | * The memory barriers | |
115 | * __wait_event()/wake_up_interruptible() take | |
116 | * care of "raw_spin_is_locked" memory ordering. | |
117 | */ | |
118 | if (raw_spin_is_locked(&buf->raw_tick_nohz_spinlock)) | |
119 | goto retry; | |
120 | else | |
121 | return 0; | |
122 | } | |
123 | } else { | |
124 | if (subbuf_trunc(lib_ring_buffer_get_offset(config, buf), | |
125 | chan) | |
126 | - subbuf_trunc(lib_ring_buffer_get_consumed(config, buf), | |
127 | chan) | |
128 | >= chan->backend.buf_size) | |
129 | return POLLPRI | POLLRDBAND; | |
130 | else | |
131 | return POLLIN | POLLRDNORM; | |
132 | } | |
133 | } | |
134 | return mask; | |
135 | } | |
136 | ||
137 | /** | |
138 | * lib_ring_buffer_ioctl - control ring buffer reader synchronization | |
139 | * | |
140 | * @filp: the file | |
141 | * @cmd: the command | |
142 | * @arg: command arg | |
143 | * | |
144 | * This ioctl implements commands necessary for producer/consumer | |
145 | * and flight recorder reader interaction : | |
146 | * RING_BUFFER_GET_NEXT_SUBBUF | |
147 | * Get the next sub-buffer that can be read. It never blocks. | |
148 | * RING_BUFFER_PUT_NEXT_SUBBUF | |
149 | * Release the currently read sub-buffer. | |
150 | * RING_BUFFER_GET_SUBBUF_SIZE | |
151 | * returns the size of the current sub-buffer. | |
152 | * RING_BUFFER_GET_MAX_SUBBUF_SIZE | |
153 | * returns the maximum size for sub-buffers. | |
154 | * RING_BUFFER_GET_NUM_SUBBUF | |
155 | * returns the number of reader-visible sub-buffers in the per cpu | |
156 | * channel (for mmap). | |
157 | * RING_BUFFER_GET_MMAP_READ_OFFSET | |
158 | * returns the offset of the subbuffer belonging to the reader. | |
159 | * Should only be used for mmap clients. | |
160 | */ | |
161 | long lib_ring_buffer_ioctl(struct file *filp, unsigned int cmd, unsigned long arg) | |
162 | { | |
163 | struct lib_ring_buffer *buf = filp->private_data; | |
164 | struct channel *chan = buf->backend.chan; | |
165 | const struct lib_ring_buffer_config *config = chan->backend.config; | |
166 | ||
254ec7bc MD |
167 | if (lib_ring_buffer_channel_is_disabled(chan)) |
168 | return -EIO; | |
169 | ||
f3bc08c5 MD |
170 | switch (cmd) { |
171 | case RING_BUFFER_SNAPSHOT: | |
172 | return lib_ring_buffer_snapshot(buf, &buf->cons_snapshot, | |
173 | &buf->prod_snapshot); | |
174 | case RING_BUFFER_SNAPSHOT_GET_CONSUMED: | |
175 | return put_ulong(buf->cons_snapshot, arg); | |
176 | case RING_BUFFER_SNAPSHOT_GET_PRODUCED: | |
177 | return put_ulong(buf->prod_snapshot, arg); | |
178 | case RING_BUFFER_GET_SUBBUF: | |
179 | { | |
180 | unsigned long uconsume; | |
181 | long ret; | |
182 | ||
183 | ret = get_user(uconsume, (unsigned long __user *) arg); | |
184 | if (ret) | |
185 | return ret; /* will return -EFAULT */ | |
186 | ret = lib_ring_buffer_get_subbuf(buf, uconsume); | |
187 | if (!ret) { | |
188 | /* Set file position to zero at each successful "get" */ | |
189 | filp->f_pos = 0; | |
190 | } | |
191 | return ret; | |
192 | } | |
193 | case RING_BUFFER_PUT_SUBBUF: | |
194 | lib_ring_buffer_put_subbuf(buf); | |
195 | return 0; | |
196 | ||
197 | case RING_BUFFER_GET_NEXT_SUBBUF: | |
198 | { | |
199 | long ret; | |
200 | ||
201 | ret = lib_ring_buffer_get_next_subbuf(buf); | |
202 | if (!ret) { | |
203 | /* Set file position to zero at each successful "get" */ | |
204 | filp->f_pos = 0; | |
205 | } | |
206 | return ret; | |
207 | } | |
208 | case RING_BUFFER_PUT_NEXT_SUBBUF: | |
209 | lib_ring_buffer_put_next_subbuf(buf); | |
210 | return 0; | |
211 | case RING_BUFFER_GET_SUBBUF_SIZE: | |
212 | return put_ulong(lib_ring_buffer_get_read_data_size(config, buf), | |
213 | arg); | |
214 | case RING_BUFFER_GET_PADDED_SUBBUF_SIZE: | |
215 | { | |
216 | unsigned long size; | |
217 | ||
218 | size = lib_ring_buffer_get_read_data_size(config, buf); | |
219 | size = PAGE_ALIGN(size); | |
220 | return put_ulong(size, arg); | |
221 | } | |
222 | case RING_BUFFER_GET_MAX_SUBBUF_SIZE: | |
223 | return put_ulong(chan->backend.subbuf_size, arg); | |
224 | case RING_BUFFER_GET_MMAP_LEN: | |
225 | { | |
226 | unsigned long mmap_buf_len; | |
227 | ||
228 | if (config->output != RING_BUFFER_MMAP) | |
229 | return -EINVAL; | |
230 | mmap_buf_len = chan->backend.buf_size; | |
231 | if (chan->backend.extra_reader_sb) | |
232 | mmap_buf_len += chan->backend.subbuf_size; | |
233 | if (mmap_buf_len > INT_MAX) | |
234 | return -EFBIG; | |
235 | return put_ulong(mmap_buf_len, arg); | |
236 | } | |
237 | case RING_BUFFER_GET_MMAP_READ_OFFSET: | |
238 | { | |
239 | unsigned long sb_bindex; | |
240 | ||
241 | if (config->output != RING_BUFFER_MMAP) | |
242 | return -EINVAL; | |
243 | sb_bindex = subbuffer_id_get_index(config, | |
244 | buf->backend.buf_rsb.id); | |
245 | return put_ulong(buf->backend.array[sb_bindex]->mmap_offset, | |
246 | arg); | |
247 | } | |
5c055266 MD |
248 | case RING_BUFFER_FLUSH: |
249 | lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE); | |
250 | return 0; | |
f3bc08c5 MD |
251 | default: |
252 | return -ENOIOCTLCMD; | |
253 | } | |
254 | } | |
255 | ||
256 | #ifdef CONFIG_COMPAT | |
257 | long lib_ring_buffer_compat_ioctl(struct file *filp, unsigned int cmd, | |
258 | unsigned long arg) | |
259 | { | |
260 | struct lib_ring_buffer *buf = filp->private_data; | |
261 | struct channel *chan = buf->backend.chan; | |
262 | const struct lib_ring_buffer_config *config = chan->backend.config; | |
254ec7bc MD |
263 | |
264 | if (lib_ring_buffer_channel_is_disabled(chan)) | |
265 | return -EIO; | |
f3bc08c5 MD |
266 | |
267 | switch (cmd) { | |
268 | case RING_BUFFER_SNAPSHOT: | |
269 | return lib_ring_buffer_snapshot(buf, &buf->cons_snapshot, | |
270 | &buf->prod_snapshot); | |
271 | case RING_BUFFER_SNAPSHOT_GET_CONSUMED: | |
272 | return compat_put_ulong(buf->cons_snapshot, arg); | |
273 | case RING_BUFFER_SNAPSHOT_GET_PRODUCED: | |
274 | return compat_put_ulong(buf->prod_snapshot, arg); | |
275 | case RING_BUFFER_GET_SUBBUF: | |
276 | { | |
277 | __u32 uconsume; | |
278 | unsigned long consume; | |
279 | long ret; | |
280 | ||
281 | ret = get_user(uconsume, (__u32 __user *) arg); | |
282 | if (ret) | |
283 | return ret; /* will return -EFAULT */ | |
284 | consume = buf->cons_snapshot; | |
285 | consume &= ~0xFFFFFFFFL; | |
286 | consume |= uconsume; | |
287 | ret = lib_ring_buffer_get_subbuf(buf, consume); | |
288 | if (!ret) { | |
289 | /* Set file position to zero at each successful "get" */ | |
290 | filp->f_pos = 0; | |
291 | } | |
292 | return ret; | |
293 | } | |
294 | case RING_BUFFER_PUT_SUBBUF: | |
295 | lib_ring_buffer_put_subbuf(buf); | |
296 | return 0; | |
297 | ||
298 | case RING_BUFFER_GET_NEXT_SUBBUF: | |
299 | { | |
300 | long ret; | |
301 | ||
302 | ret = lib_ring_buffer_get_next_subbuf(buf); | |
303 | if (!ret) { | |
304 | /* Set file position to zero at each successful "get" */ | |
305 | filp->f_pos = 0; | |
306 | } | |
307 | return ret; | |
308 | } | |
309 | case RING_BUFFER_PUT_NEXT_SUBBUF: | |
310 | lib_ring_buffer_put_next_subbuf(buf); | |
311 | return 0; | |
312 | case RING_BUFFER_GET_SUBBUF_SIZE: | |
313 | { | |
314 | unsigned long data_size; | |
315 | ||
316 | data_size = lib_ring_buffer_get_read_data_size(config, buf); | |
317 | if (data_size > UINT_MAX) | |
318 | return -EFBIG; | |
319 | return put_ulong(data_size, arg); | |
320 | } | |
321 | case RING_BUFFER_GET_PADDED_SUBBUF_SIZE: | |
322 | { | |
323 | unsigned long size; | |
324 | ||
325 | size = lib_ring_buffer_get_read_data_size(config, buf); | |
326 | size = PAGE_ALIGN(size); | |
327 | if (size > UINT_MAX) | |
328 | return -EFBIG; | |
329 | return put_ulong(size, arg); | |
330 | } | |
331 | case RING_BUFFER_GET_MAX_SUBBUF_SIZE: | |
332 | if (chan->backend.subbuf_size > UINT_MAX) | |
333 | return -EFBIG; | |
334 | return put_ulong(chan->backend.subbuf_size, arg); | |
335 | case RING_BUFFER_GET_MMAP_LEN: | |
336 | { | |
337 | unsigned long mmap_buf_len; | |
338 | ||
339 | if (config->output != RING_BUFFER_MMAP) | |
340 | return -EINVAL; | |
341 | mmap_buf_len = chan->backend.buf_size; | |
342 | if (chan->backend.extra_reader_sb) | |
343 | mmap_buf_len += chan->backend.subbuf_size; | |
344 | if (mmap_buf_len > UINT_MAX) | |
345 | return -EFBIG; | |
346 | return put_ulong(mmap_buf_len, arg); | |
347 | } | |
348 | case RING_BUFFER_GET_MMAP_READ_OFFSET: | |
349 | { | |
350 | unsigned long sb_bindex, read_offset; | |
351 | ||
352 | if (config->output != RING_BUFFER_MMAP) | |
353 | return -EINVAL; | |
354 | sb_bindex = subbuffer_id_get_index(config, | |
355 | buf->backend.buf_rsb.id); | |
356 | read_offset = buf->backend.array[sb_bindex]->mmap_offset; | |
357 | if (read_offset > UINT_MAX) | |
358 | return -EINVAL; | |
359 | return put_ulong(read_offset, arg); | |
360 | } | |
5c055266 MD |
361 | case RING_BUFFER_FLUSH: |
362 | lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE); | |
363 | return 0; | |
f3bc08c5 MD |
364 | default: |
365 | return -ENOIOCTLCMD; | |
366 | } | |
367 | } | |
368 | #endif | |
369 | ||
370 | const struct file_operations lib_ring_buffer_file_operations = { | |
371 | .open = lib_ring_buffer_open, | |
372 | .release = lib_ring_buffer_release, | |
373 | .poll = lib_ring_buffer_poll, | |
374 | .splice_read = lib_ring_buffer_splice_read, | |
375 | .mmap = lib_ring_buffer_mmap, | |
376 | .unlocked_ioctl = lib_ring_buffer_ioctl, | |
377 | .llseek = lib_ring_buffer_no_llseek, | |
378 | #ifdef CONFIG_COMPAT | |
379 | .compat_ioctl = lib_ring_buffer_compat_ioctl, | |
380 | #endif | |
381 | }; | |
382 | EXPORT_SYMBOL_GPL(lib_ring_buffer_file_operations); | |
383 | ||
384 | MODULE_LICENSE("GPL and additional rights"); | |
385 | MODULE_AUTHOR("Mathieu Desnoyers"); | |
386 | MODULE_DESCRIPTION("Ring Buffer Library VFS"); |