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 * Receive end of subbuffer TSC as parameter. It has been read in the
370 * space reservation loop of either reserve or switch, which ensures it
371 * progresses monotonically with event records in the buffer. Therefore,
372 * it ensures that the end timestamp of a subbuffer is <= begin
373 * timestamp of the following subbuffers.
376 void lib_ring_buffer_check_deliver(const struct lttng_ust_lib_ring_buffer_config
*config
,
377 struct lttng_ust_lib_ring_buffer
*buf
,
378 struct channel
*chan
,
379 unsigned long offset
,
380 unsigned long commit_count
,
382 struct lttng_ust_shm_handle
*handle
,
385 unsigned long old_commit_count
= commit_count
386 - chan
->backend
.subbuf_size
;
388 /* Check if all commits have been done */
389 if (caa_unlikely((buf_trunc(offset
, chan
) >> chan
->backend
.num_subbuf_order
)
390 - (old_commit_count
& chan
->commit_count_mask
) == 0)) {
392 * If we succeeded at updating cc_sb below, we are the subbuffer
393 * writer delivering the subbuffer. Deals with concurrent
394 * updates of the "cc" value without adding a add_return atomic
395 * operation to the fast path.
397 * We are doing the delivery in two steps:
398 * - First, we cmpxchg() cc_sb to the new value
399 * old_commit_count + 1. This ensures that we are the only
400 * subbuffer user successfully filling the subbuffer, but we
401 * do _not_ set the cc_sb value to "commit_count" yet.
402 * Therefore, other writers that would wrap around the ring
403 * buffer and try to start writing to our subbuffer would
404 * have to drop records, because it would appear as
406 * We therefore have exclusive access to the subbuffer control
407 * structures. This mutual exclusion with other writers is
408 * crucially important to perform record overruns count in
409 * flight recorder mode locklessly.
410 * - When we are ready to release the subbuffer (either for
411 * reading or for overrun by other writers), we simply set the
412 * cc_sb value to "commit_count" and perform delivery.
414 * The subbuffer size is least 2 bytes (minimum size: 1 page).
415 * This guarantees that old_commit_count + 1 != commit_count.
419 * Order prior updates to reserve count prior to the
420 * commit_cold cc_sb update.
423 if (caa_likely(v_cmpxchg(config
, &shmp_index(handle
, buf
->commit_cold
, idx
)->cc_sb
,
424 old_commit_count
, old_commit_count
+ 1)
425 == old_commit_count
)) {
427 * Start of exclusive subbuffer access. We are
428 * guaranteed to be the last writer in this subbuffer
429 * and any other writer trying to access this subbuffer
430 * in this state is required to drop records.
433 subbuffer_get_records_count(config
,
436 &buf
->records_count
);
438 subbuffer_count_records_overrun(config
,
441 &buf
->records_overrun
);
442 config
->cb
.buffer_end(buf
, tsc
, idx
,
443 lib_ring_buffer_get_data_size(config
,
450 * Set noref flag and offset for this subbuffer id.
451 * Contains a memory barrier that ensures counter stores
452 * are ordered before set noref and offset.
454 lib_ring_buffer_set_noref_offset(config
, &buf
->backend
, idx
,
455 buf_trunc_val(offset
, chan
), handle
);
458 * Order set_noref and record counter updates before the
459 * end of subbuffer exclusive access. Orders with
460 * respect to writers coming into the subbuffer after
461 * wrap around, and also order wrt concurrent readers.
464 /* End of exclusive subbuffer access */
465 v_set(config
, &shmp_index(handle
, buf
->commit_cold
, idx
)->cc_sb
,
468 * Order later updates to reserve count after
469 * the commit cold cc_sb update.
472 lib_ring_buffer_vmcore_check_deliver(config
, buf
,
473 commit_count
, idx
, handle
);
476 * RING_BUFFER_WAKEUP_BY_WRITER wakeup is not lock-free.
478 if (config
->wakeup
== RING_BUFFER_WAKEUP_BY_WRITER
479 && uatomic_read(&buf
->active_readers
)
480 && lib_ring_buffer_poll_deliver(config
, buf
, chan
, handle
)) {
481 lib_ring_buffer_wakeup(buf
, handle
);
488 * lib_ring_buffer_write_commit_counter
490 * For flight recording. must be called after commit.
491 * This function increments the subbuffer's commit_seq counter each time the
492 * commit count reaches back the reserve offset (modulo subbuffer size). It is
493 * useful for crash dump.
496 void lib_ring_buffer_write_commit_counter(const struct lttng_ust_lib_ring_buffer_config
*config
,
497 struct lttng_ust_lib_ring_buffer
*buf
,
498 struct channel
*chan
,
500 unsigned long buf_offset
,
501 unsigned long commit_count
,
502 struct lttng_ust_shm_handle
*handle
)
504 unsigned long commit_seq_old
;
506 if (config
->oops
!= RING_BUFFER_OOPS_CONSISTENCY
)
510 * subbuf_offset includes commit_count_mask. We can simply
511 * compare the offsets within the subbuffer without caring about
512 * buffer full/empty mismatch because offset is never zero here
513 * (subbuffer header and record headers have non-zero length).
515 if (caa_unlikely(subbuf_offset(buf_offset
- commit_count
, chan
)))
518 commit_seq_old
= v_read(config
, &shmp_index(handle
, buf
->commit_hot
, idx
)->seq
);
519 while ((long) (commit_seq_old
- commit_count
) < 0)
520 commit_seq_old
= v_cmpxchg(config
, &shmp_index(handle
, buf
->commit_hot
, idx
)->seq
,
521 commit_seq_old
, commit_count
);
524 extern int lib_ring_buffer_create(struct lttng_ust_lib_ring_buffer
*buf
,
525 struct channel_backend
*chanb
, int cpu
,
526 struct lttng_ust_shm_handle
*handle
,
527 struct shm_object
*shmobj
);
528 extern void lib_ring_buffer_free(struct lttng_ust_lib_ring_buffer
*buf
,
529 struct lttng_ust_shm_handle
*handle
);
531 /* Keep track of trap nesting inside ring buffer code */
532 extern DECLARE_URCU_TLS(unsigned int, lib_ring_buffer_nesting
);
534 #endif /* _LTTNG_RING_BUFFER_FRONTEND_INTERNAL_H */