1 #ifndef _LTTNG_RING_BUFFER_FRONTEND_INTERNAL_H
2 #define _LTTNG_RING_BUFFER_FRONTEND_INTERNAL_H
5 * libringbuffer/frontend_internal.h
7 * Ring Buffer Library Synchronization Header (internal helpers).
9 * Copyright (C) 2005-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
11 * This library is free software; you can redistribute it and/or
12 * modify it under the terms of the GNU Lesser General Public
13 * License as published by the Free Software Foundation; only
14 * version 2.1 of the License.
16 * This library is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19 * Lesser General Public License for more details.
21 * You should have received a copy of the GNU Lesser General Public
22 * License along with this library; if not, write to the Free Software
23 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27 * Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
29 * See ring_buffer_frontend.c for more information on wait-free algorithms.
31 * Dual LGPL v2.1/GPL v2 license.
34 #include <urcu/compiler.h>
35 #include <urcu/tls-compat.h>
39 #include <lttng/ringbuffer-config.h>
40 #include "backend_types.h"
41 #include "frontend_types.h"
44 /* Buffer offset macros */
46 /* buf_trunc mask selects only the buffer number. */
48 unsigned long buf_trunc(unsigned long offset
, struct channel
*chan
)
50 return offset
& ~(chan
->backend
.buf_size
- 1);
54 /* Select the buffer number value (counter). */
56 unsigned long buf_trunc_val(unsigned long offset
, struct channel
*chan
)
58 return buf_trunc(offset
, chan
) >> chan
->backend
.buf_size_order
;
61 /* buf_offset mask selects only the offset within the current buffer. */
63 unsigned long buf_offset(unsigned long offset
, struct channel
*chan
)
65 return offset
& (chan
->backend
.buf_size
- 1);
68 /* subbuf_offset mask selects the offset within the current subbuffer. */
70 unsigned long subbuf_offset(unsigned long offset
, struct channel
*chan
)
72 return offset
& (chan
->backend
.subbuf_size
- 1);
75 /* subbuf_trunc mask selects the subbuffer number. */
77 unsigned long subbuf_trunc(unsigned long offset
, struct channel
*chan
)
79 return offset
& ~(chan
->backend
.subbuf_size
- 1);
82 /* subbuf_align aligns the offset to the next subbuffer. */
84 unsigned long subbuf_align(unsigned long offset
, struct channel
*chan
)
86 return (offset
+ chan
->backend
.subbuf_size
)
87 & ~(chan
->backend
.subbuf_size
- 1);
90 /* subbuf_index returns the index of the current subbuffer within the buffer. */
92 unsigned long subbuf_index(unsigned long offset
, struct channel
*chan
)
94 return buf_offset(offset
, chan
) >> chan
->backend
.subbuf_size_order
;
98 * Last TSC comparison functions. Check if the current TSC overflows tsc_bits
99 * bits from the last TSC read. When overflows are detected, the full 64-bit
100 * timestamp counter should be written in the record header. Reads and writes
101 * last_tsc atomically.
104 #if (CAA_BITS_PER_LONG == 32)
106 void save_last_tsc(const struct lttng_ust_lib_ring_buffer_config
*config
,
107 struct lttng_ust_lib_ring_buffer
*buf
, uint64_t tsc
)
109 if (config
->tsc_bits
== 0 || config
->tsc_bits
== 64)
113 * Ensure the compiler performs this update in a single instruction.
115 v_set(config
, &buf
->last_tsc
, (unsigned long)(tsc
>> config
->tsc_bits
));
119 int last_tsc_overflow(const struct lttng_ust_lib_ring_buffer_config
*config
,
120 struct lttng_ust_lib_ring_buffer
*buf
, uint64_t tsc
)
122 unsigned long tsc_shifted
;
124 if (config
->tsc_bits
== 0 || config
->tsc_bits
== 64)
127 tsc_shifted
= (unsigned long)(tsc
>> config
->tsc_bits
);
128 if (caa_unlikely(tsc_shifted
129 - (unsigned long)v_read(config
, &buf
->last_tsc
)))
136 void save_last_tsc(const struct lttng_ust_lib_ring_buffer_config
*config
,
137 struct lttng_ust_lib_ring_buffer
*buf
, uint64_t tsc
)
139 if (config
->tsc_bits
== 0 || config
->tsc_bits
== 64)
142 v_set(config
, &buf
->last_tsc
, (unsigned long)tsc
);
146 int last_tsc_overflow(const struct lttng_ust_lib_ring_buffer_config
*config
,
147 struct lttng_ust_lib_ring_buffer
*buf
, uint64_t tsc
)
149 if (config
->tsc_bits
== 0 || config
->tsc_bits
== 64)
152 if (caa_unlikely((tsc
- v_read(config
, &buf
->last_tsc
))
153 >> config
->tsc_bits
))
161 int lib_ring_buffer_reserve_slow(struct lttng_ust_lib_ring_buffer_ctx
*ctx
);
164 void lib_ring_buffer_switch_slow(struct lttng_ust_lib_ring_buffer
*buf
,
165 enum switch_mode mode
,
166 struct lttng_ust_shm_handle
*handle
);
168 /* Buffer write helpers */
171 void lib_ring_buffer_reserve_push_reader(struct lttng_ust_lib_ring_buffer
*buf
,
172 struct channel
*chan
,
173 unsigned long offset
)
175 unsigned long consumed_old
, consumed_new
;
178 consumed_old
= uatomic_read(&buf
->consumed
);
180 * If buffer is in overwrite mode, push the reader consumed
181 * count if the write position has reached it and we are not
182 * at the first iteration (don't push the reader farther than
183 * the writer). This operation can be done concurrently by many
184 * writers in the same buffer, the writer being at the farthest
185 * write position sub-buffer index in the buffer being the one
186 * which will win this loop.
188 if (caa_unlikely(subbuf_trunc(offset
, chan
)
189 - subbuf_trunc(consumed_old
, chan
)
190 >= chan
->backend
.buf_size
))
191 consumed_new
= subbuf_align(consumed_old
, chan
);
194 } while (caa_unlikely(uatomic_cmpxchg(&buf
->consumed
, consumed_old
,
195 consumed_new
) != consumed_old
));
199 void lib_ring_buffer_vmcore_check_deliver(const struct lttng_ust_lib_ring_buffer_config
*config
,
200 struct lttng_ust_lib_ring_buffer
*buf
,
201 unsigned long commit_count
,
203 struct lttng_ust_shm_handle
*handle
)
205 if (config
->oops
== RING_BUFFER_OOPS_CONSISTENCY
)
206 v_set(config
, &shmp_index(handle
, buf
->commit_hot
, idx
)->seq
, commit_count
);
210 int lib_ring_buffer_poll_deliver(const struct lttng_ust_lib_ring_buffer_config
*config
,
211 struct lttng_ust_lib_ring_buffer
*buf
,
212 struct channel
*chan
,
213 struct lttng_ust_shm_handle
*handle
)
215 unsigned long consumed_old
, consumed_idx
, commit_count
, write_offset
;
217 consumed_old
= uatomic_read(&buf
->consumed
);
218 consumed_idx
= subbuf_index(consumed_old
, chan
);
219 commit_count
= v_read(config
, &shmp_index(handle
, buf
->commit_cold
, consumed_idx
)->cc_sb
);
221 * No memory barrier here, since we are only interested
222 * in a statistically correct polling result. The next poll will
223 * get the data is we are racing. The mb() that ensures correct
224 * memory order is in get_subbuf.
226 write_offset
= v_read(config
, &buf
->offset
);
229 * Check that the subbuffer we are trying to consume has been
230 * already fully committed.
233 if (((commit_count
- chan
->backend
.subbuf_size
)
234 & chan
->commit_count_mask
)
235 - (buf_trunc(consumed_old
, chan
)
236 >> chan
->backend
.num_subbuf_order
)
241 * Check that we are not about to read the same subbuffer in
242 * which the writer head is.
244 if (subbuf_trunc(write_offset
, chan
) - subbuf_trunc(consumed_old
, chan
)
253 int lib_ring_buffer_pending_data(const struct lttng_ust_lib_ring_buffer_config
*config
,
254 struct lttng_ust_lib_ring_buffer
*buf
,
255 struct channel
*chan
)
257 return !!subbuf_offset(v_read(config
, &buf
->offset
), chan
);
261 unsigned long lib_ring_buffer_get_data_size(const struct lttng_ust_lib_ring_buffer_config
*config
,
262 struct lttng_ust_lib_ring_buffer
*buf
,
264 struct lttng_ust_shm_handle
*handle
)
266 return subbuffer_get_data_size(config
, &buf
->backend
, idx
, handle
);
270 * Check if all space reservation in a buffer have been committed. This helps
271 * knowing if an execution context is nested (for per-cpu buffers only).
272 * This is a very specific ftrace use-case, so we keep this as "internal" API.
275 int lib_ring_buffer_reserve_committed(const struct lttng_ust_lib_ring_buffer_config
*config
,
276 struct lttng_ust_lib_ring_buffer
*buf
,
277 struct channel
*chan
,
278 struct lttng_ust_shm_handle
*handle
)
280 unsigned long offset
, idx
, commit_count
;
282 CHAN_WARN_ON(chan
, config
->alloc
!= RING_BUFFER_ALLOC_PER_CPU
);
283 CHAN_WARN_ON(chan
, config
->sync
!= RING_BUFFER_SYNC_PER_CPU
);
286 * Read offset and commit count in a loop so they are both read
287 * atomically wrt interrupts. By deal with interrupt concurrency by
288 * restarting both reads if the offset has been pushed. Note that given
289 * we only have to deal with interrupt concurrency here, an interrupt
290 * modifying the commit count will also modify "offset", so it is safe
291 * to only check for offset modifications.
294 offset
= v_read(config
, &buf
->offset
);
295 idx
= subbuf_index(offset
, chan
);
296 commit_count
= v_read(config
, &shmp_index(handle
, buf
->commit_hot
, idx
)->cc
);
297 } while (offset
!= v_read(config
, &buf
->offset
));
299 return ((buf_trunc(offset
, chan
) >> chan
->backend
.num_subbuf_order
)
300 - (commit_count
& chan
->commit_count_mask
) == 0);
304 void lib_ring_buffer_wakeup(struct lttng_ust_lib_ring_buffer
*buf
,
305 struct lttng_ust_shm_handle
*handle
)
307 int wakeup_fd
= shm_get_wakeup_fd(handle
, &buf
->self
._ref
);
308 sigset_t sigpipe_set
, pending_set
, old_set
;
309 int ret
, sigpipe_was_pending
= 0;
315 * Wake-up the other end by writing a null byte in the pipe
316 * (non-blocking). Important note: Because writing into the
317 * pipe is non-blocking (and therefore we allow dropping wakeup
318 * data, as long as there is wakeup data present in the pipe
319 * buffer to wake up the consumer), the consumer should perform
320 * the following sequence for waiting:
321 * 1) empty the pipe (reads).
322 * 2) check if there is data in the buffer.
323 * 3) wait on the pipe (poll).
325 * Discard the SIGPIPE from write(), not disturbing any SIGPIPE
326 * that might be already pending. If a bogus SIGPIPE is sent to
327 * the entire process concurrently by a malicious user, it may
328 * be simply discarded.
330 ret
= sigemptyset(&pending_set
);
333 * sigpending returns the mask of signals that are _both_
334 * blocked for the thread _and_ pending for either the thread or
335 * the entire process.
337 ret
= sigpending(&pending_set
);
339 sigpipe_was_pending
= sigismember(&pending_set
, SIGPIPE
);
341 * If sigpipe was pending, it means it was already blocked, so
342 * no need to block it.
344 if (!sigpipe_was_pending
) {
345 ret
= sigemptyset(&sigpipe_set
);
347 ret
= sigaddset(&sigpipe_set
, SIGPIPE
);
349 ret
= pthread_sigmask(SIG_BLOCK
, &sigpipe_set
, &old_set
);
353 ret
= write(wakeup_fd
, "", 1);
354 } while (ret
== -1L && errno
== EINTR
);
355 if (ret
== -1L && errno
== EPIPE
&& !sigpipe_was_pending
) {
356 struct timespec timeout
= { 0, 0 };
358 ret
= sigtimedwait(&sigpipe_set
, NULL
,
360 } while (ret
== -1L && errno
== EINTR
);
362 if (!sigpipe_was_pending
) {
363 ret
= pthread_sigmask(SIG_SETMASK
, &old_set
, NULL
);
369 void lib_ring_buffer_check_deliver(const struct lttng_ust_lib_ring_buffer_config
*config
,
370 struct lttng_ust_lib_ring_buffer
*buf
,
371 struct channel
*chan
,
372 unsigned long offset
,
373 unsigned long commit_count
,
375 struct lttng_ust_shm_handle
*handle
)
377 unsigned long old_commit_count
= commit_count
378 - chan
->backend
.subbuf_size
;
381 /* Check if all commits have been done */
382 if (caa_unlikely((buf_trunc(offset
, chan
) >> chan
->backend
.num_subbuf_order
)
383 - (old_commit_count
& chan
->commit_count_mask
) == 0)) {
385 * If we succeeded at updating cc_sb below, we are the subbuffer
386 * writer delivering the subbuffer. Deals with concurrent
387 * updates of the "cc" value without adding a add_return atomic
388 * operation to the fast path.
390 * We are doing the delivery in two steps:
391 * - First, we cmpxchg() cc_sb to the new value
392 * old_commit_count + 1. This ensures that we are the only
393 * subbuffer user successfully filling the subbuffer, but we
394 * do _not_ set the cc_sb value to "commit_count" yet.
395 * Therefore, other writers that would wrap around the ring
396 * buffer and try to start writing to our subbuffer would
397 * have to drop records, because it would appear as
399 * We therefore have exclusive access to the subbuffer control
400 * structures. This mutual exclusion with other writers is
401 * crucially important to perform record overruns count in
402 * flight recorder mode locklessly.
403 * - When we are ready to release the subbuffer (either for
404 * reading or for overrun by other writers), we simply set the
405 * cc_sb value to "commit_count" and perform delivery.
407 * The subbuffer size is least 2 bytes (minimum size: 1 page).
408 * This guarantees that old_commit_count + 1 != commit_count.
410 if (caa_likely(v_cmpxchg(config
, &shmp_index(handle
, buf
->commit_cold
, idx
)->cc_sb
,
411 old_commit_count
, old_commit_count
+ 1)
412 == old_commit_count
)) {
414 * Start of exclusive subbuffer access. We are
415 * guaranteed to be the last writer in this subbuffer
416 * and any other writer trying to access this subbuffer
417 * in this state is required to drop records.
419 tsc
= config
->cb
.ring_buffer_clock_read(chan
);
421 subbuffer_get_records_count(config
,
424 &buf
->records_count
);
426 subbuffer_count_records_overrun(config
,
429 &buf
->records_overrun
);
430 config
->cb
.buffer_end(buf
, tsc
, idx
,
431 lib_ring_buffer_get_data_size(config
,
438 * Set noref flag and offset for this subbuffer id.
439 * Contains a memory barrier that ensures counter stores
440 * are ordered before set noref and offset.
442 lib_ring_buffer_set_noref_offset(config
, &buf
->backend
, idx
,
443 buf_trunc_val(offset
, chan
), handle
);
446 * Order set_noref and record counter updates before the
447 * end of subbuffer exclusive access. Orders with
448 * respect to writers coming into the subbuffer after
449 * wrap around, and also order wrt concurrent readers.
452 /* End of exclusive subbuffer access */
453 v_set(config
, &shmp_index(handle
, buf
->commit_cold
, idx
)->cc_sb
,
455 lib_ring_buffer_vmcore_check_deliver(config
, buf
,
456 commit_count
, idx
, handle
);
459 * RING_BUFFER_WAKEUP_BY_WRITER wakeup is not lock-free.
461 if (config
->wakeup
== RING_BUFFER_WAKEUP_BY_WRITER
462 && uatomic_read(&buf
->active_readers
)
463 && lib_ring_buffer_poll_deliver(config
, buf
, chan
, handle
)) {
464 lib_ring_buffer_wakeup(buf
, handle
);
471 * lib_ring_buffer_write_commit_counter
473 * For flight recording. must be called after commit.
474 * This function increments the subbuffer's commit_seq counter each time the
475 * commit count reaches back the reserve offset (modulo subbuffer size). It is
476 * useful for crash dump.
479 void lib_ring_buffer_write_commit_counter(const struct lttng_ust_lib_ring_buffer_config
*config
,
480 struct lttng_ust_lib_ring_buffer
*buf
,
481 struct channel
*chan
,
483 unsigned long buf_offset
,
484 unsigned long commit_count
,
486 struct lttng_ust_shm_handle
*handle
)
488 unsigned long offset
, commit_seq_old
;
490 if (config
->oops
!= RING_BUFFER_OOPS_CONSISTENCY
)
493 offset
= buf_offset
+ slot_size
;
496 * subbuf_offset includes commit_count_mask. We can simply
497 * compare the offsets within the subbuffer without caring about
498 * buffer full/empty mismatch because offset is never zero here
499 * (subbuffer header and record headers have non-zero length).
501 if (caa_unlikely(subbuf_offset(offset
- commit_count
, chan
)))
504 commit_seq_old
= v_read(config
, &shmp_index(handle
, buf
->commit_hot
, idx
)->seq
);
505 while ((long) (commit_seq_old
- commit_count
) < 0)
506 commit_seq_old
= v_cmpxchg(config
, &shmp_index(handle
, buf
->commit_hot
, idx
)->seq
,
507 commit_seq_old
, commit_count
);
510 extern int lib_ring_buffer_create(struct lttng_ust_lib_ring_buffer
*buf
,
511 struct channel_backend
*chanb
, int cpu
,
512 struct lttng_ust_shm_handle
*handle
,
513 struct shm_object
*shmobj
);
514 extern void lib_ring_buffer_free(struct lttng_ust_lib_ring_buffer
*buf
,
515 struct lttng_ust_shm_handle
*handle
);
517 /* Keep track of trap nesting inside ring buffer code */
518 extern DECLARE_URCU_TLS(unsigned int, lib_ring_buffer_nesting
);
520 #endif /* _LTTNG_RING_BUFFER_FRONTEND_INTERNAL_H */