Commit | Line | Data |
---|---|---|
f3bc08c5 MD |
1 | /* |
2 | * ring_buffer_vfs.c | |
3 | * | |
f3bc08c5 MD |
4 | * Ring Buffer VFS file operations. |
5 | * | |
886d51a3 MD |
6 | * Copyright (C) 2010-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com> |
7 | * | |
8 | * This library is free software; you can redistribute it and/or | |
9 | * modify it under the terms of the GNU Lesser General Public | |
10 | * License as published by the Free Software Foundation; only | |
11 | * version 2.1 of the License. | |
12 | * | |
13 | * This library is distributed in the hope that it will be useful, | |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 | * Lesser General Public License for more details. | |
17 | * | |
18 | * You should have received a copy of the GNU Lesser General Public | |
19 | * License along with this library; if not, write to the Free Software | |
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
f3bc08c5 MD |
21 | */ |
22 | ||
23 | #include <linux/module.h> | |
24 | #include <linux/fs.h> | |
25 | #include <linux/compat.h> | |
26 | ||
c075712b MD |
27 | #include <wrapper/ringbuffer/backend.h> |
28 | #include <wrapper/ringbuffer/frontend.h> | |
29 | #include <wrapper/ringbuffer/vfs.h> | |
30 | #include <wrapper/poll.h> | |
31 | #include <lttng-tracer.h> | |
f3bc08c5 MD |
32 | |
33 | static int put_ulong(unsigned long val, unsigned long arg) | |
34 | { | |
35 | return put_user(val, (unsigned long __user *)arg); | |
36 | } | |
37 | ||
38 | #ifdef CONFIG_COMPAT | |
39 | static int compat_put_ulong(compat_ulong_t val, unsigned long arg) | |
40 | { | |
41 | return put_user(val, (compat_ulong_t __user *)compat_ptr(arg)); | |
42 | } | |
43 | #endif | |
44 | ||
b3b8072b MD |
45 | /* |
46 | * This is not used by anonymous file descriptors. This code is left | |
47 | * there if we ever want to implement an inode with open() operation. | |
48 | */ | |
d83004aa JD |
49 | int lib_ring_buffer_open(struct inode *inode, struct file *file, |
50 | struct lib_ring_buffer *buf) | |
f3bc08c5 | 51 | { |
f3bc08c5 MD |
52 | int ret; |
53 | ||
733ce41d MD |
54 | if (!buf) |
55 | return -EINVAL; | |
56 | ||
f3bc08c5 MD |
57 | ret = lib_ring_buffer_open_read(buf); |
58 | if (ret) | |
59 | return ret; | |
60 | ||
f3bc08c5 MD |
61 | ret = nonseekable_open(inode, file); |
62 | if (ret) | |
63 | goto release_read; | |
64 | return 0; | |
65 | ||
66 | release_read: | |
67 | lib_ring_buffer_release_read(buf); | |
68 | return ret; | |
69 | } | |
d83004aa | 70 | EXPORT_SYMBOL_GPL(lib_ring_buffer_open); |
f3bc08c5 MD |
71 | |
72 | /** | |
d83004aa | 73 | * vfs_lib_ring_buffer_open - ring buffer open file operation |
f3bc08c5 MD |
74 | * @inode: opened inode |
75 | * @file: opened file | |
76 | * | |
d83004aa JD |
77 | * Open implementation. Makes sure only one open instance of a buffer is |
78 | * done at a given moment. | |
f3bc08c5 | 79 | */ |
d83004aa JD |
80 | static |
81 | int vfs_lib_ring_buffer_open(struct inode *inode, struct file *file) | |
f3bc08c5 | 82 | { |
d83004aa JD |
83 | struct lib_ring_buffer *buf = inode->i_private; |
84 | ||
85 | file->private_data = buf; | |
86 | return lib_ring_buffer_open(inode, file, buf); | |
87 | } | |
f3bc08c5 | 88 | |
d83004aa JD |
89 | int lib_ring_buffer_release(struct inode *inode, struct file *file, |
90 | struct lib_ring_buffer *buf) | |
91 | { | |
f3bc08c5 MD |
92 | lib_ring_buffer_release_read(buf); |
93 | ||
94 | return 0; | |
95 | } | |
d83004aa | 96 | EXPORT_SYMBOL_GPL(lib_ring_buffer_release); |
f3bc08c5 MD |
97 | |
98 | /** | |
d83004aa JD |
99 | * vfs_lib_ring_buffer_release - ring buffer release file operation |
100 | * @inode: opened inode | |
101 | * @file: opened file | |
f3bc08c5 | 102 | * |
d83004aa | 103 | * Release implementation. |
f3bc08c5 | 104 | */ |
d83004aa JD |
105 | static |
106 | int vfs_lib_ring_buffer_release(struct inode *inode, struct file *file) | |
107 | { | |
108 | struct lib_ring_buffer *buf = file->private_data; | |
109 | ||
110 | return lib_ring_buffer_release(inode, file, buf); | |
111 | } | |
112 | ||
113 | unsigned int lib_ring_buffer_poll(struct file *filp, poll_table *wait, | |
114 | struct lib_ring_buffer *buf) | |
f3bc08c5 MD |
115 | { |
116 | unsigned int mask = 0; | |
f3bc08c5 | 117 | struct channel *chan = buf->backend.chan; |
5a8fd222 | 118 | const struct lib_ring_buffer_config *config = &chan->backend.config; |
254ec7bc | 119 | int finalized, disabled; |
f3bc08c5 MD |
120 | |
121 | if (filp->f_mode & FMODE_READ) { | |
a33e44a6 | 122 | poll_wait_set_exclusive(wait); |
f3bc08c5 MD |
123 | poll_wait(filp, &buf->read_wait, wait); |
124 | ||
125 | finalized = lib_ring_buffer_is_finalized(config, buf); | |
254ec7bc MD |
126 | disabled = lib_ring_buffer_channel_is_disabled(chan); |
127 | ||
f3bc08c5 MD |
128 | /* |
129 | * lib_ring_buffer_is_finalized() contains a smp_rmb() ordering | |
130 | * finalized load before offsets loads. | |
131 | */ | |
132 | WARN_ON(atomic_long_read(&buf->active_readers) != 1); | |
133 | retry: | |
254ec7bc MD |
134 | if (disabled) |
135 | return POLLERR; | |
136 | ||
f3bc08c5 MD |
137 | if (subbuf_trunc(lib_ring_buffer_get_offset(config, buf), chan) |
138 | - subbuf_trunc(lib_ring_buffer_get_consumed(config, buf), chan) | |
139 | == 0) { | |
140 | if (finalized) | |
141 | return POLLHUP; | |
142 | else { | |
143 | /* | |
144 | * The memory barriers | |
145 | * __wait_event()/wake_up_interruptible() take | |
146 | * care of "raw_spin_is_locked" memory ordering. | |
147 | */ | |
148 | if (raw_spin_is_locked(&buf->raw_tick_nohz_spinlock)) | |
149 | goto retry; | |
150 | else | |
151 | return 0; | |
152 | } | |
153 | } else { | |
154 | if (subbuf_trunc(lib_ring_buffer_get_offset(config, buf), | |
155 | chan) | |
156 | - subbuf_trunc(lib_ring_buffer_get_consumed(config, buf), | |
157 | chan) | |
158 | >= chan->backend.buf_size) | |
159 | return POLLPRI | POLLRDBAND; | |
160 | else | |
161 | return POLLIN | POLLRDNORM; | |
162 | } | |
163 | } | |
164 | return mask; | |
165 | } | |
d83004aa | 166 | EXPORT_SYMBOL_GPL(lib_ring_buffer_poll); |
f3bc08c5 MD |
167 | |
168 | /** | |
d83004aa | 169 | * vfs_lib_ring_buffer_poll - ring buffer poll file operation |
f3bc08c5 | 170 | * @filp: the file |
d83004aa | 171 | * @wait: poll table |
f3bc08c5 | 172 | * |
d83004aa | 173 | * Poll implementation. |
f3bc08c5 | 174 | */ |
d83004aa JD |
175 | static |
176 | unsigned int vfs_lib_ring_buffer_poll(struct file *filp, poll_table *wait) | |
f3bc08c5 MD |
177 | { |
178 | struct lib_ring_buffer *buf = filp->private_data; | |
d83004aa JD |
179 | |
180 | return lib_ring_buffer_poll(filp, wait, buf); | |
181 | } | |
182 | ||
183 | long lib_ring_buffer_ioctl(struct file *filp, unsigned int cmd, | |
184 | unsigned long arg, struct lib_ring_buffer *buf) | |
185 | { | |
f3bc08c5 | 186 | struct channel *chan = buf->backend.chan; |
5a8fd222 | 187 | const struct lib_ring_buffer_config *config = &chan->backend.config; |
f3bc08c5 | 188 | |
254ec7bc MD |
189 | if (lib_ring_buffer_channel_is_disabled(chan)) |
190 | return -EIO; | |
191 | ||
f3bc08c5 MD |
192 | switch (cmd) { |
193 | case RING_BUFFER_SNAPSHOT: | |
194 | return lib_ring_buffer_snapshot(buf, &buf->cons_snapshot, | |
195 | &buf->prod_snapshot); | |
4dce5a48 JG |
196 | case RING_BUFFER_SNAPSHOT_SAMPLE_POSITIONS: |
197 | return lib_ring_buffer_snapshot_sample_positions(buf, | |
198 | &buf->cons_snapshot, &buf->prod_snapshot); | |
f3bc08c5 MD |
199 | case RING_BUFFER_SNAPSHOT_GET_CONSUMED: |
200 | return put_ulong(buf->cons_snapshot, arg); | |
201 | case RING_BUFFER_SNAPSHOT_GET_PRODUCED: | |
202 | return put_ulong(buf->prod_snapshot, arg); | |
203 | case RING_BUFFER_GET_SUBBUF: | |
204 | { | |
205 | unsigned long uconsume; | |
206 | long ret; | |
207 | ||
208 | ret = get_user(uconsume, (unsigned long __user *) arg); | |
209 | if (ret) | |
210 | return ret; /* will return -EFAULT */ | |
211 | ret = lib_ring_buffer_get_subbuf(buf, uconsume); | |
212 | if (!ret) { | |
213 | /* Set file position to zero at each successful "get" */ | |
214 | filp->f_pos = 0; | |
215 | } | |
216 | return ret; | |
217 | } | |
218 | case RING_BUFFER_PUT_SUBBUF: | |
219 | lib_ring_buffer_put_subbuf(buf); | |
220 | return 0; | |
221 | ||
222 | case RING_BUFFER_GET_NEXT_SUBBUF: | |
223 | { | |
224 | long ret; | |
225 | ||
226 | ret = lib_ring_buffer_get_next_subbuf(buf); | |
227 | if (!ret) { | |
228 | /* Set file position to zero at each successful "get" */ | |
229 | filp->f_pos = 0; | |
230 | } | |
231 | return ret; | |
232 | } | |
233 | case RING_BUFFER_PUT_NEXT_SUBBUF: | |
234 | lib_ring_buffer_put_next_subbuf(buf); | |
235 | return 0; | |
236 | case RING_BUFFER_GET_SUBBUF_SIZE: | |
237 | return put_ulong(lib_ring_buffer_get_read_data_size(config, buf), | |
238 | arg); | |
239 | case RING_BUFFER_GET_PADDED_SUBBUF_SIZE: | |
240 | { | |
241 | unsigned long size; | |
242 | ||
243 | size = lib_ring_buffer_get_read_data_size(config, buf); | |
244 | size = PAGE_ALIGN(size); | |
245 | return put_ulong(size, arg); | |
246 | } | |
247 | case RING_BUFFER_GET_MAX_SUBBUF_SIZE: | |
248 | return put_ulong(chan->backend.subbuf_size, arg); | |
249 | case RING_BUFFER_GET_MMAP_LEN: | |
250 | { | |
251 | unsigned long mmap_buf_len; | |
252 | ||
253 | if (config->output != RING_BUFFER_MMAP) | |
254 | return -EINVAL; | |
255 | mmap_buf_len = chan->backend.buf_size; | |
256 | if (chan->backend.extra_reader_sb) | |
257 | mmap_buf_len += chan->backend.subbuf_size; | |
258 | if (mmap_buf_len > INT_MAX) | |
259 | return -EFBIG; | |
260 | return put_ulong(mmap_buf_len, arg); | |
261 | } | |
262 | case RING_BUFFER_GET_MMAP_READ_OFFSET: | |
263 | { | |
264 | unsigned long sb_bindex; | |
265 | ||
266 | if (config->output != RING_BUFFER_MMAP) | |
267 | return -EINVAL; | |
268 | sb_bindex = subbuffer_id_get_index(config, | |
269 | buf->backend.buf_rsb.id); | |
270 | return put_ulong(buf->backend.array[sb_bindex]->mmap_offset, | |
271 | arg); | |
272 | } | |
5c055266 | 273 | case RING_BUFFER_FLUSH: |
5e391252 | 274 | lib_ring_buffer_switch_remote(buf); |
5c055266 | 275 | return 0; |
c6f05468 MD |
276 | case RING_BUFFER_FLUSH_EMPTY: |
277 | lib_ring_buffer_switch_remote_empty(buf); | |
278 | return 0; | |
f3bc08c5 MD |
279 | default: |
280 | return -ENOIOCTLCMD; | |
281 | } | |
282 | } | |
d83004aa JD |
283 | EXPORT_SYMBOL_GPL(lib_ring_buffer_ioctl); |
284 | ||
285 | /** | |
286 | * vfs_lib_ring_buffer_ioctl - control ring buffer reader synchronization | |
287 | * | |
288 | * @filp: the file | |
289 | * @cmd: the command | |
290 | * @arg: command arg | |
291 | * | |
292 | * This ioctl implements commands necessary for producer/consumer | |
293 | * and flight recorder reader interaction : | |
294 | * RING_BUFFER_GET_NEXT_SUBBUF | |
295 | * Get the next sub-buffer that can be read. It never blocks. | |
296 | * RING_BUFFER_PUT_NEXT_SUBBUF | |
297 | * Release the currently read sub-buffer. | |
298 | * RING_BUFFER_GET_SUBBUF_SIZE | |
299 | * returns the size of the current sub-buffer. | |
300 | * RING_BUFFER_GET_MAX_SUBBUF_SIZE | |
301 | * returns the maximum size for sub-buffers. | |
302 | * RING_BUFFER_GET_NUM_SUBBUF | |
303 | * returns the number of reader-visible sub-buffers in the per cpu | |
304 | * channel (for mmap). | |
305 | * RING_BUFFER_GET_MMAP_READ_OFFSET | |
306 | * returns the offset of the subbuffer belonging to the reader. | |
307 | * Should only be used for mmap clients. | |
308 | */ | |
309 | static | |
310 | long vfs_lib_ring_buffer_ioctl(struct file *filp, unsigned int cmd, unsigned long arg) | |
311 | { | |
312 | struct lib_ring_buffer *buf = filp->private_data; | |
313 | ||
314 | return lib_ring_buffer_ioctl(filp, cmd, arg, buf); | |
315 | } | |
f3bc08c5 MD |
316 | |
317 | #ifdef CONFIG_COMPAT | |
318 | long lib_ring_buffer_compat_ioctl(struct file *filp, unsigned int cmd, | |
d83004aa | 319 | unsigned long arg, struct lib_ring_buffer *buf) |
f3bc08c5 | 320 | { |
f3bc08c5 | 321 | struct channel *chan = buf->backend.chan; |
5a8fd222 | 322 | const struct lib_ring_buffer_config *config = &chan->backend.config; |
254ec7bc MD |
323 | |
324 | if (lib_ring_buffer_channel_is_disabled(chan)) | |
325 | return -EIO; | |
f3bc08c5 MD |
326 | |
327 | switch (cmd) { | |
216f7feb | 328 | case RING_BUFFER_COMPAT_SNAPSHOT: |
f3bc08c5 MD |
329 | return lib_ring_buffer_snapshot(buf, &buf->cons_snapshot, |
330 | &buf->prod_snapshot); | |
4dce5a48 JG |
331 | case RING_BUFFER_COMPAT_SNAPSHOT_SAMPLE_POSITIONS: |
332 | return lib_ring_buffer_snapshot_sample_positions(buf, | |
333 | &buf->cons_snapshot, &buf->prod_snapshot); | |
216f7feb | 334 | case RING_BUFFER_COMPAT_SNAPSHOT_GET_CONSUMED: |
f3bc08c5 | 335 | return compat_put_ulong(buf->cons_snapshot, arg); |
216f7feb | 336 | case RING_BUFFER_COMPAT_SNAPSHOT_GET_PRODUCED: |
f3bc08c5 | 337 | return compat_put_ulong(buf->prod_snapshot, arg); |
216f7feb | 338 | case RING_BUFFER_COMPAT_GET_SUBBUF: |
f3bc08c5 MD |
339 | { |
340 | __u32 uconsume; | |
341 | unsigned long consume; | |
342 | long ret; | |
343 | ||
344 | ret = get_user(uconsume, (__u32 __user *) arg); | |
345 | if (ret) | |
346 | return ret; /* will return -EFAULT */ | |
347 | consume = buf->cons_snapshot; | |
348 | consume &= ~0xFFFFFFFFL; | |
349 | consume |= uconsume; | |
350 | ret = lib_ring_buffer_get_subbuf(buf, consume); | |
351 | if (!ret) { | |
352 | /* Set file position to zero at each successful "get" */ | |
353 | filp->f_pos = 0; | |
354 | } | |
355 | return ret; | |
356 | } | |
216f7feb | 357 | case RING_BUFFER_COMPAT_PUT_SUBBUF: |
f3bc08c5 MD |
358 | lib_ring_buffer_put_subbuf(buf); |
359 | return 0; | |
360 | ||
216f7feb | 361 | case RING_BUFFER_COMPAT_GET_NEXT_SUBBUF: |
f3bc08c5 MD |
362 | { |
363 | long ret; | |
364 | ||
365 | ret = lib_ring_buffer_get_next_subbuf(buf); | |
366 | if (!ret) { | |
367 | /* Set file position to zero at each successful "get" */ | |
368 | filp->f_pos = 0; | |
369 | } | |
370 | return ret; | |
371 | } | |
216f7feb | 372 | case RING_BUFFER_COMPAT_PUT_NEXT_SUBBUF: |
f3bc08c5 MD |
373 | lib_ring_buffer_put_next_subbuf(buf); |
374 | return 0; | |
216f7feb | 375 | case RING_BUFFER_COMPAT_GET_SUBBUF_SIZE: |
f3bc08c5 MD |
376 | { |
377 | unsigned long data_size; | |
378 | ||
379 | data_size = lib_ring_buffer_get_read_data_size(config, buf); | |
380 | if (data_size > UINT_MAX) | |
381 | return -EFBIG; | |
4343ec8e | 382 | return compat_put_ulong(data_size, arg); |
f3bc08c5 | 383 | } |
216f7feb | 384 | case RING_BUFFER_COMPAT_GET_PADDED_SUBBUF_SIZE: |
f3bc08c5 MD |
385 | { |
386 | unsigned long size; | |
387 | ||
388 | size = lib_ring_buffer_get_read_data_size(config, buf); | |
389 | size = PAGE_ALIGN(size); | |
390 | if (size > UINT_MAX) | |
391 | return -EFBIG; | |
4343ec8e | 392 | return compat_put_ulong(size, arg); |
f3bc08c5 | 393 | } |
216f7feb | 394 | case RING_BUFFER_COMPAT_GET_MAX_SUBBUF_SIZE: |
f3bc08c5 MD |
395 | if (chan->backend.subbuf_size > UINT_MAX) |
396 | return -EFBIG; | |
4343ec8e | 397 | return compat_put_ulong(chan->backend.subbuf_size, arg); |
216f7feb | 398 | case RING_BUFFER_COMPAT_GET_MMAP_LEN: |
f3bc08c5 MD |
399 | { |
400 | unsigned long mmap_buf_len; | |
401 | ||
402 | if (config->output != RING_BUFFER_MMAP) | |
403 | return -EINVAL; | |
404 | mmap_buf_len = chan->backend.buf_size; | |
405 | if (chan->backend.extra_reader_sb) | |
406 | mmap_buf_len += chan->backend.subbuf_size; | |
407 | if (mmap_buf_len > UINT_MAX) | |
408 | return -EFBIG; | |
4343ec8e | 409 | return compat_put_ulong(mmap_buf_len, arg); |
f3bc08c5 | 410 | } |
216f7feb | 411 | case RING_BUFFER_COMPAT_GET_MMAP_READ_OFFSET: |
f3bc08c5 MD |
412 | { |
413 | unsigned long sb_bindex, read_offset; | |
414 | ||
415 | if (config->output != RING_BUFFER_MMAP) | |
416 | return -EINVAL; | |
417 | sb_bindex = subbuffer_id_get_index(config, | |
418 | buf->backend.buf_rsb.id); | |
419 | read_offset = buf->backend.array[sb_bindex]->mmap_offset; | |
420 | if (read_offset > UINT_MAX) | |
421 | return -EINVAL; | |
4343ec8e | 422 | return compat_put_ulong(read_offset, arg); |
f3bc08c5 | 423 | } |
216f7feb | 424 | case RING_BUFFER_COMPAT_FLUSH: |
5e391252 | 425 | lib_ring_buffer_switch_remote(buf); |
5c055266 | 426 | return 0; |
c6f05468 MD |
427 | case RING_BUFFER_COMPAT_FLUSH_EMPTY: |
428 | lib_ring_buffer_switch_remote_empty(buf); | |
429 | return 0; | |
f3bc08c5 MD |
430 | default: |
431 | return -ENOIOCTLCMD; | |
432 | } | |
433 | } | |
d83004aa JD |
434 | EXPORT_SYMBOL_GPL(lib_ring_buffer_compat_ioctl); |
435 | ||
436 | static | |
437 | long vfs_lib_ring_buffer_compat_ioctl(struct file *filp, unsigned int cmd, | |
438 | unsigned long arg) | |
439 | { | |
440 | struct lib_ring_buffer *buf = filp->private_data; | |
441 | ||
442 | return lib_ring_buffer_compat_ioctl(filp, cmd, arg, buf); | |
443 | } | |
f3bc08c5 MD |
444 | #endif |
445 | ||
446 | const struct file_operations lib_ring_buffer_file_operations = { | |
a33c9927 | 447 | .owner = THIS_MODULE, |
d83004aa JD |
448 | .open = vfs_lib_ring_buffer_open, |
449 | .release = vfs_lib_ring_buffer_release, | |
450 | .poll = vfs_lib_ring_buffer_poll, | |
451 | .splice_read = vfs_lib_ring_buffer_splice_read, | |
452 | .mmap = vfs_lib_ring_buffer_mmap, | |
453 | .unlocked_ioctl = vfs_lib_ring_buffer_ioctl, | |
454 | .llseek = vfs_lib_ring_buffer_no_llseek, | |
f3bc08c5 | 455 | #ifdef CONFIG_COMPAT |
d83004aa | 456 | .compat_ioctl = vfs_lib_ring_buffer_compat_ioctl, |
f3bc08c5 MD |
457 | #endif |
458 | }; | |
459 | EXPORT_SYMBOL_GPL(lib_ring_buffer_file_operations); | |
460 | ||
461 | MODULE_LICENSE("GPL and additional rights"); | |
1c124020 MJ |
462 | MODULE_AUTHOR("Mathieu Desnoyers <mathieu.desnoyers@efficios.com>"); |
463 | MODULE_DESCRIPTION("LTTng ring buffer library"); | |
13ab8b0a MD |
464 | MODULE_VERSION(__stringify(LTTNG_MODULES_MAJOR_VERSION) "." |
465 | __stringify(LTTNG_MODULES_MINOR_VERSION) "." | |
466 | __stringify(LTTNG_MODULES_PATCHLEVEL_VERSION) | |
467 | LTTNG_MODULES_EXTRAVERSION); |