Cleanup: comment mismatch with code
[lttng-ust.git] / libringbuffer / ring_buffer_frontend.c
CommitLineData
852c2936
MD
1/*
2 * ring_buffer_frontend.c
3 *
e92f3e28
MD
4 * Copyright (C) 2005-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; only
9 * version 2.1 of the License.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 *
852c2936
MD
20 *
21 * Ring buffer wait-free buffer synchronization. Producer-consumer and flight
22 * recorder (overwrite) modes. See thesis:
23 *
24 * Desnoyers, Mathieu (2009), "Low-Impact Operating System Tracing", Ph.D.
25 * dissertation, Ecole Polytechnique de Montreal.
26 * http://www.lttng.org/pub/thesis/desnoyers-dissertation-2009-12.pdf
27 *
28 * - Algorithm presentation in Chapter 5:
29 * "Lockless Multi-Core High-Throughput Buffering".
30 * - Algorithm formal verification in Section 8.6:
31 * "Formal verification of LTTng"
32 *
33 * Author:
34 * Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
35 *
36 * Inspired from LTT and RelayFS:
37 * Karim Yaghmour <karim@opersys.com>
38 * Tom Zanussi <zanussi@us.ibm.com>
39 * Bob Wisniewski <bob@watson.ibm.com>
40 * And from K42 :
41 * Bob Wisniewski <bob@watson.ibm.com>
42 *
43 * Buffer reader semantic :
44 *
45 * - get_subbuf_size
46 * while buffer is not finalized and empty
47 * - get_subbuf
48 * - if return value != 0, continue
49 * - splice one subbuffer worth of data to a pipe
50 * - splice the data from pipe to disk/network
51 * - put_subbuf
852c2936
MD
52 */
53
5ad63a16 54#define _GNU_SOURCE
a6352fd4 55#include <sys/types.h>
431d5cf0
MD
56#include <sys/mman.h>
57#include <sys/stat.h>
03d2d293 58#include <unistd.h>
431d5cf0 59#include <fcntl.h>
03d2d293
MD
60#include <signal.h>
61#include <time.h>
14641deb 62#include <urcu/compiler.h>
a6352fd4 63#include <urcu/ref.h>
8c90a710 64#include <urcu/tls-compat.h>
35897f8b 65#include <helper.h>
14641deb 66
a6352fd4 67#include "smp.h"
4318ae1b 68#include <lttng/ringbuffer-config.h>
2fed87ae 69#include "vatomic.h"
4931a13e
MD
70#include "backend.h"
71#include "frontend.h"
a6352fd4 72#include "shm.h"
f645cfa7 73#include "tlsfixup.h"
bdcf8d82 74#include "../liblttng-ust/compat.h" /* For ENODATA */
852c2936 75
431d5cf0
MD
76#ifndef max
77#define max(a, b) ((a) > (b) ? (a) : (b))
78#endif
79
64493e4f
MD
80/* Print DBG() messages about events lost only every 1048576 hits */
81#define DBG_PRINT_NR_LOST (1UL << 20)
82
34a91bdb
MD
83#define LTTNG_UST_RB_SIG_FLUSH SIGRTMIN
84#define LTTNG_UST_RB_SIG_READ SIGRTMIN + 1
85#define LTTNG_UST_RB_SIG_TEARDOWN SIGRTMIN + 2
03d2d293
MD
86#define CLOCKID CLOCK_MONOTONIC
87
2432c3c9
MD
88/*
89 * Use POSIX SHM: shm_open(3) and shm_unlink(3).
90 * close(2) to close the fd returned by shm_open.
91 * shm_unlink releases the shared memory object name.
92 * ftruncate(2) sets the size of the memory object.
93 * mmap/munmap maps the shared memory obj to a virtual address in the
94 * calling proceess (should be done both in libust and consumer).
95 * See shm_overview(7) for details.
96 * Pass file descriptor returned by shm_open(3) to ltt-sessiond through
97 * a UNIX socket.
98 *
99 * Since we don't need to access the object using its name, we can
100 * immediately shm_unlink(3) it, and only keep the handle with its file
101 * descriptor.
102 */
103
852c2936
MD
104/*
105 * Internal structure representing offsets to use at a sub-buffer switch.
106 */
107struct switch_offsets {
108 unsigned long begin, end, old;
109 size_t pre_header_padding, size;
110 unsigned int switch_new_start:1, switch_new_end:1, switch_old_start:1,
111 switch_old_end:1;
112};
113
8c90a710 114DEFINE_URCU_TLS(unsigned int, lib_ring_buffer_nesting);
852c2936 115
cb7378b3
MD
116/*
117 * wakeup_fd_mutex protects wakeup fd use by timer from concurrent
118 * close.
119 */
120static pthread_mutex_t wakeup_fd_mutex = PTHREAD_MUTEX_INITIALIZER;
121
852c2936
MD
122static
123void lib_ring_buffer_print_errors(struct channel *chan,
009769ca
MD
124 struct lttng_ust_lib_ring_buffer *buf, int cpu,
125 struct lttng_ust_shm_handle *handle);
852c2936 126
03d2d293
MD
127/*
128 * Handle timer teardown race wrt memory free of private data by
129 * ring buffer signals are handled by a single thread, which permits
130 * a synchronization point between handling of each signal.
131 * Protected by the ust mutex.
132 */
133struct timer_signal_data {
134 pthread_t tid; /* thread id managing signals */
135 int setup_done;
136 int qs_done;
137};
138
139static struct timer_signal_data timer_signal;
140
852c2936
MD
141/**
142 * lib_ring_buffer_reset - Reset ring buffer to initial values.
143 * @buf: Ring buffer.
144 *
145 * Effectively empty the ring buffer. Should be called when the buffer is not
146 * used for writing. The ring buffer can be opened for reading, but the reader
147 * should not be using the iterator concurrently with reset. The previous
148 * current iterator record is reset.
149 */
4cfec15c 150void lib_ring_buffer_reset(struct lttng_ust_lib_ring_buffer *buf,
38fae1d3 151 struct lttng_ust_shm_handle *handle)
852c2936 152{
1d498196 153 struct channel *chan = shmp(handle, buf->backend.chan);
4cfec15c 154 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
155 unsigned int i;
156
157 /*
158 * Reset iterator first. It will put the subbuffer if it currently holds
159 * it.
160 */
852c2936
MD
161 v_set(config, &buf->offset, 0);
162 for (i = 0; i < chan->backend.num_subbuf; i++) {
4746ae29
MD
163 v_set(config, &shmp_index(handle, buf->commit_hot, i)->cc, 0);
164 v_set(config, &shmp_index(handle, buf->commit_hot, i)->seq, 0);
165 v_set(config, &shmp_index(handle, buf->commit_cold, i)->cc_sb, 0);
852c2936 166 }
a6352fd4
MD
167 uatomic_set(&buf->consumed, 0);
168 uatomic_set(&buf->record_disabled, 0);
852c2936 169 v_set(config, &buf->last_tsc, 0);
1d498196 170 lib_ring_buffer_backend_reset(&buf->backend, handle);
852c2936
MD
171 /* Don't reset number of active readers */
172 v_set(config, &buf->records_lost_full, 0);
173 v_set(config, &buf->records_lost_wrap, 0);
174 v_set(config, &buf->records_lost_big, 0);
175 v_set(config, &buf->records_count, 0);
176 v_set(config, &buf->records_overrun, 0);
177 buf->finalized = 0;
178}
852c2936
MD
179
180/**
181 * channel_reset - Reset channel to initial values.
182 * @chan: Channel.
183 *
184 * Effectively empty the channel. Should be called when the channel is not used
185 * for writing. The channel can be opened for reading, but the reader should not
186 * be using the iterator concurrently with reset. The previous current iterator
187 * record is reset.
188 */
189void channel_reset(struct channel *chan)
190{
191 /*
192 * Reset iterators first. Will put the subbuffer if held for reading.
193 */
a6352fd4 194 uatomic_set(&chan->record_disabled, 0);
852c2936
MD
195 /* Don't reset commit_count_mask, still valid */
196 channel_backend_reset(&chan->backend);
197 /* Don't reset switch/read timer interval */
198 /* Don't reset notifiers and notifier enable bits */
199 /* Don't reset reader reference count */
200}
852c2936
MD
201
202/*
203 * Must be called under cpu hotplug protection.
204 */
4cfec15c 205int lib_ring_buffer_create(struct lttng_ust_lib_ring_buffer *buf,
a6352fd4 206 struct channel_backend *chanb, int cpu,
38fae1d3 207 struct lttng_ust_shm_handle *handle,
1d498196 208 struct shm_object *shmobj)
852c2936 209{
4cfec15c 210 const struct lttng_ust_lib_ring_buffer_config *config = &chanb->config;
14641deb 211 struct channel *chan = caa_container_of(chanb, struct channel, backend);
a3f61e7f 212 void *priv = channel_get_private(chan);
852c2936 213 size_t subbuf_header_size;
2fed87ae 214 uint64_t tsc;
852c2936
MD
215 int ret;
216
217 /* Test for cpu hotplug */
218 if (buf->backend.allocated)
219 return 0;
220
a6352fd4 221 ret = lib_ring_buffer_backend_create(&buf->backend, &chan->backend,
1d498196 222 cpu, handle, shmobj);
852c2936
MD
223 if (ret)
224 return ret;
225
1d498196
MD
226 align_shm(shmobj, __alignof__(struct commit_counters_hot));
227 set_shmp(buf->commit_hot,
228 zalloc_shm(shmobj,
229 sizeof(struct commit_counters_hot) * chan->backend.num_subbuf));
230 if (!shmp(handle, buf->commit_hot)) {
852c2936
MD
231 ret = -ENOMEM;
232 goto free_chanbuf;
233 }
234
1d498196
MD
235 align_shm(shmobj, __alignof__(struct commit_counters_cold));
236 set_shmp(buf->commit_cold,
237 zalloc_shm(shmobj,
238 sizeof(struct commit_counters_cold) * chan->backend.num_subbuf));
239 if (!shmp(handle, buf->commit_cold)) {
852c2936
MD
240 ret = -ENOMEM;
241 goto free_commit;
242 }
243
852c2936
MD
244 /*
245 * Write the subbuffer header for first subbuffer so we know the total
246 * duration of data gathering.
247 */
248 subbuf_header_size = config->cb.subbuffer_header_size();
249 v_set(config, &buf->offset, subbuf_header_size);
4746ae29 250 subbuffer_id_clear_noref(config, &shmp_index(handle, buf->backend.buf_wsb, 0)->id);
1d498196
MD
251 tsc = config->cb.ring_buffer_clock_read(shmp(handle, buf->backend.chan));
252 config->cb.buffer_begin(buf, tsc, 0, handle);
4746ae29 253 v_add(config, subbuf_header_size, &shmp_index(handle, buf->commit_hot, 0)->cc);
852c2936
MD
254
255 if (config->cb.buffer_create) {
1d498196 256 ret = config->cb.buffer_create(buf, priv, cpu, chanb->name, handle);
852c2936
MD
257 if (ret)
258 goto free_init;
259 }
852c2936 260 buf->backend.allocated = 1;
852c2936
MD
261 return 0;
262
263 /* Error handling */
264free_init:
a6352fd4 265 /* commit_cold will be freed by shm teardown */
852c2936 266free_commit:
a6352fd4 267 /* commit_hot will be freed by shm teardown */
852c2936 268free_chanbuf:
852c2936
MD
269 return ret;
270}
271
03d2d293
MD
272static
273void lib_ring_buffer_channel_switch_timer(int sig, siginfo_t *si, void *uc)
852c2936 274{
03d2d293
MD
275 const struct lttng_ust_lib_ring_buffer_config *config;
276 struct lttng_ust_shm_handle *handle;
277 struct channel *chan;
278 int cpu;
279
280 assert(CMM_LOAD_SHARED(timer_signal.tid) == pthread_self());
281
282 chan = si->si_value.sival_ptr;
283 handle = chan->handle;
284 config = &chan->backend.config;
852c2936 285
34a91bdb 286 DBG("Switch timer for channel %p\n", chan);
03d2d293 287
4dcd7e80
MD
288 /*
289 * Only flush buffers periodically if readers are active.
290 */
cb7378b3 291 pthread_mutex_lock(&wakeup_fd_mutex);
03d2d293
MD
292 if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
293 for_each_possible_cpu(cpu) {
294 struct lttng_ust_lib_ring_buffer *buf =
295 shmp(handle, chan->backend.buf[cpu].shmp);
4dcd7e80
MD
296 if (uatomic_read(&buf->active_readers))
297 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE,
298 chan->handle);
03d2d293
MD
299 }
300 } else {
301 struct lttng_ust_lib_ring_buffer *buf =
302 shmp(handle, chan->backend.buf[0].shmp);
303
34a91bdb
MD
304 if (uatomic_read(&buf->active_readers))
305 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE,
306 chan->handle);
307 }
308 pthread_mutex_unlock(&wakeup_fd_mutex);
309 return;
310}
311
312static
313void lib_ring_buffer_channel_do_read(struct channel *chan)
314{
315 const struct lttng_ust_lib_ring_buffer_config *config;
316 struct lttng_ust_shm_handle *handle;
317 int cpu;
318
319 handle = chan->handle;
320 config = &chan->backend.config;
321
322 /*
323 * Only flush buffers periodically if readers are active.
324 */
325 pthread_mutex_lock(&wakeup_fd_mutex);
326 if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
327 for_each_possible_cpu(cpu) {
328 struct lttng_ust_lib_ring_buffer *buf =
329 shmp(handle, chan->backend.buf[cpu].shmp);
330
331 if (uatomic_read(&buf->active_readers)
332 && lib_ring_buffer_poll_deliver(config, buf,
333 chan, handle)) {
334 lib_ring_buffer_wakeup(buf, handle);
335 }
336 }
337 } else {
338 struct lttng_ust_lib_ring_buffer *buf =
339 shmp(handle, chan->backend.buf[0].shmp);
340
341 if (uatomic_read(&buf->active_readers)
342 && lib_ring_buffer_poll_deliver(config, buf,
343 chan, handle)) {
344 lib_ring_buffer_wakeup(buf, handle);
345 }
03d2d293 346 }
cb7378b3 347 pthread_mutex_unlock(&wakeup_fd_mutex);
34a91bdb
MD
348}
349
350static
351void lib_ring_buffer_channel_read_timer(int sig, siginfo_t *si, void *uc)
352{
353 struct channel *chan;
354
355 assert(CMM_LOAD_SHARED(timer_signal.tid) == pthread_self());
356 chan = si->si_value.sival_ptr;
357 DBG("Read timer for channel %p\n", chan);
358 lib_ring_buffer_channel_do_read(chan);
03d2d293
MD
359 return;
360}
361
362static
363void rb_setmask(sigset_t *mask)
364{
365 int ret;
366
367 ret = sigemptyset(mask);
368 if (ret) {
369 PERROR("sigemptyset");
370 }
34a91bdb
MD
371 ret = sigaddset(mask, LTTNG_UST_RB_SIG_FLUSH);
372 if (ret) {
373 PERROR("sigaddset");
374 }
375 ret = sigaddset(mask, LTTNG_UST_RB_SIG_READ);
03d2d293
MD
376 if (ret) {
377 PERROR("sigaddset");
378 }
379 ret = sigaddset(mask, LTTNG_UST_RB_SIG_TEARDOWN);
380 if (ret) {
381 PERROR("sigaddset");
382 }
383}
384
385static
386void *sig_thread(void *arg)
387{
388 sigset_t mask;
389 siginfo_t info;
390 int signr;
391
392 /* Only self thread will receive signal mask. */
393 rb_setmask(&mask);
394 CMM_STORE_SHARED(timer_signal.tid, pthread_self());
395
396 for (;;) {
397 signr = sigwaitinfo(&mask, &info);
398 if (signr == -1) {
34a91bdb
MD
399 if (errno != EINTR)
400 PERROR("sigwaitinfo");
03d2d293
MD
401 continue;
402 }
34a91bdb 403 if (signr == LTTNG_UST_RB_SIG_FLUSH) {
03d2d293
MD
404 lib_ring_buffer_channel_switch_timer(info.si_signo,
405 &info, NULL);
34a91bdb
MD
406 } else if (signr == LTTNG_UST_RB_SIG_READ) {
407 lib_ring_buffer_channel_read_timer(info.si_signo,
408 &info, NULL);
03d2d293
MD
409 } else if (signr == LTTNG_UST_RB_SIG_TEARDOWN) {
410 cmm_smp_mb();
411 CMM_STORE_SHARED(timer_signal.qs_done, 1);
412 cmm_smp_mb();
413 } else {
414 ERR("Unexptected signal %d\n", info.si_signo);
415 }
416 }
417 return NULL;
418}
419
420/*
421 * Called with ust_lock() held.
422 * Ensure only a single thread listens on the timer signal.
423 */
424static
425void lib_ring_buffer_setup_timer_thread(void)
426{
427 pthread_t thread;
428 int ret;
429
430 if (timer_signal.setup_done)
852c2936 431 return;
03d2d293
MD
432
433 ret = pthread_create(&thread, NULL, &sig_thread, NULL);
434 if (ret) {
435 errno = ret;
436 PERROR("pthread_create");
437 }
438 ret = pthread_detach(thread);
439 if (ret) {
440 errno = ret;
441 PERROR("pthread_detach");
442 }
443 timer_signal.setup_done = 1;
852c2936
MD
444}
445
03d2d293
MD
446/*
447 * Called with ust_lock() held.
448 */
449static
450void lib_ring_buffer_channel_switch_timer_start(struct channel *chan)
852c2936 451{
03d2d293
MD
452 struct sigevent sev;
453 struct itimerspec its;
454 int ret;
852c2936 455
03d2d293 456 if (!chan->switch_timer_interval || chan->switch_timer_enabled)
852c2936
MD
457 return;
458
03d2d293
MD
459 chan->switch_timer_enabled = 1;
460
461 lib_ring_buffer_setup_timer_thread();
462
463 sev.sigev_notify = SIGEV_SIGNAL;
34a91bdb 464 sev.sigev_signo = LTTNG_UST_RB_SIG_FLUSH;
03d2d293
MD
465 sev.sigev_value.sival_ptr = chan;
466 ret = timer_create(CLOCKID, &sev, &chan->switch_timer);
467 if (ret == -1) {
468 PERROR("timer_create");
469 }
470
471 its.it_value.tv_sec = chan->switch_timer_interval / 1000000;
472 its.it_value.tv_nsec = chan->switch_timer_interval % 1000000;
473 its.it_interval.tv_sec = its.it_value.tv_sec;
474 its.it_interval.tv_nsec = its.it_value.tv_nsec;
475
476 ret = timer_settime(chan->switch_timer, 0, &its, NULL);
477 if (ret == -1) {
478 PERROR("timer_settime");
479 }
480}
481
482/*
483 * Called with ust_lock() held.
484 */
485static
486void lib_ring_buffer_channel_switch_timer_stop(struct channel *chan)
487{
488 sigset_t pending_set;
34a91bdb 489 int ret;
03d2d293
MD
490
491 if (!chan->switch_timer_interval || !chan->switch_timer_enabled)
492 return;
493
494 ret = timer_delete(chan->switch_timer);
495 if (ret == -1) {
496 PERROR("timer_delete");
497 }
498
499 /*
500 * Ensure we don't have any signal queued for this channel.
501 */
502 for (;;) {
503 ret = sigemptyset(&pending_set);
504 if (ret == -1) {
505 PERROR("sigemptyset");
506 }
507 ret = sigpending(&pending_set);
508 if (ret == -1) {
509 PERROR("sigpending");
510 }
34a91bdb 511 if (!sigismember(&pending_set, LTTNG_UST_RB_SIG_FLUSH))
03d2d293
MD
512 break;
513 caa_cpu_relax();
514 }
515
516 /*
517 * From this point, no new signal handler will be fired that
518 * would try to access "chan". However, we still need to wait
519 * for any currently executing handler to complete.
520 */
521 cmm_smp_mb();
522 CMM_STORE_SHARED(timer_signal.qs_done, 0);
523 cmm_smp_mb();
524
525 /*
526 * Kill with LTTNG_UST_RB_SIG_TEARDOWN, so signal management
527 * thread wakes up.
528 */
529 kill(getpid(), LTTNG_UST_RB_SIG_TEARDOWN);
530
531 while (!CMM_LOAD_SHARED(timer_signal.qs_done))
532 caa_cpu_relax();
533 cmm_smp_mb();
534
535 chan->switch_timer = 0;
536 chan->switch_timer_enabled = 0;
852c2936
MD
537}
538
539/*
34a91bdb 540 * Called with ust_lock() held.
852c2936 541 */
34a91bdb
MD
542static
543void lib_ring_buffer_channel_read_timer_start(struct channel *chan)
852c2936 544{
4cfec15c 545 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
34a91bdb
MD
546 struct sigevent sev;
547 struct itimerspec its;
548 int ret;
852c2936 549
34a91bdb
MD
550 if (config->wakeup != RING_BUFFER_WAKEUP_BY_TIMER
551 || !chan->read_timer_interval || chan->read_timer_enabled)
552 return;
852c2936 553
34a91bdb 554 chan->read_timer_enabled = 1;
852c2936 555
34a91bdb 556 lib_ring_buffer_setup_timer_thread();
852c2936 557
34a91bdb
MD
558 sev.sigev_notify = SIGEV_SIGNAL;
559 sev.sigev_signo = LTTNG_UST_RB_SIG_READ;
560 sev.sigev_value.sival_ptr = chan;
561 ret = timer_create(CLOCKID, &sev, &chan->read_timer);
562 if (ret == -1) {
563 PERROR("timer_create");
564 }
852c2936 565
34a91bdb
MD
566 its.it_value.tv_sec = chan->read_timer_interval / 1000000;
567 its.it_value.tv_nsec = chan->read_timer_interval % 1000000;
568 its.it_interval.tv_sec = its.it_value.tv_sec;
569 its.it_interval.tv_nsec = its.it_value.tv_nsec;
852c2936 570
34a91bdb
MD
571 ret = timer_settime(chan->read_timer, 0, &its, NULL);
572 if (ret == -1) {
573 PERROR("timer_settime");
574 }
852c2936
MD
575}
576
34a91bdb
MD
577/*
578 * Called with ust_lock() held.
579 */
580static
581void lib_ring_buffer_channel_read_timer_stop(struct channel *chan)
852c2936 582{
4cfec15c 583 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
34a91bdb
MD
584 sigset_t pending_set;
585 int ret;
852c2936
MD
586
587 if (config->wakeup != RING_BUFFER_WAKEUP_BY_TIMER
34a91bdb 588 || !chan->read_timer_interval || !chan->read_timer_enabled)
852c2936
MD
589 return;
590
34a91bdb
MD
591 ret = timer_delete(chan->read_timer);
592 if (ret == -1) {
593 PERROR("timer_delete");
594 }
595
852c2936
MD
596 /*
597 * do one more check to catch data that has been written in the last
598 * timer period.
599 */
34a91bdb
MD
600 lib_ring_buffer_channel_do_read(chan);
601
602
603 /*
604 * Ensure we don't have any signal queued for this channel.
605 */
606 for (;;) {
607 ret = sigemptyset(&pending_set);
608 if (ret == -1) {
609 PERROR("sigemptyset");
610 }
611 ret = sigpending(&pending_set);
612 if (ret == -1) {
613 PERROR("sigpending");
614 }
615 if (!sigismember(&pending_set, LTTNG_UST_RB_SIG_READ))
616 break;
617 caa_cpu_relax();
852c2936 618 }
34a91bdb
MD
619
620 /*
621 * From this point, no new signal handler will be fired that
622 * would try to access "chan". However, we still need to wait
623 * for any currently executing handler to complete.
624 */
625 cmm_smp_mb();
626 CMM_STORE_SHARED(timer_signal.qs_done, 0);
627 cmm_smp_mb();
628
629 /*
630 * Kill with LTTNG_UST_RB_SIG_TEARDOWN, so signal management
631 * thread wakes up.
632 */
633 kill(getpid(), LTTNG_UST_RB_SIG_TEARDOWN);
634
635 while (!CMM_LOAD_SHARED(timer_signal.qs_done))
636 caa_cpu_relax();
637 cmm_smp_mb();
638
639 chan->read_timer = 0;
640 chan->read_timer_enabled = 0;
852c2936
MD
641}
642
1d498196 643static void channel_unregister_notifiers(struct channel *chan,
38fae1d3 644 struct lttng_ust_shm_handle *handle)
852c2936 645{
03d2d293 646 lib_ring_buffer_channel_switch_timer_stop(chan);
34a91bdb 647 lib_ring_buffer_channel_read_timer_stop(chan);
852c2936
MD
648}
649
009769ca
MD
650static void channel_print_errors(struct channel *chan,
651 struct lttng_ust_shm_handle *handle)
652{
653 const struct lttng_ust_lib_ring_buffer_config *config =
654 &chan->backend.config;
655 int cpu;
656
657 if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
658 for_each_possible_cpu(cpu) {
659 struct lttng_ust_lib_ring_buffer *buf =
660 shmp(handle, chan->backend.buf[cpu].shmp);
661 lib_ring_buffer_print_errors(chan, buf, cpu, handle);
662 }
663 } else {
664 struct lttng_ust_lib_ring_buffer *buf =
665 shmp(handle, chan->backend.buf[0].shmp);
666
667 lib_ring_buffer_print_errors(chan, buf, -1, handle);
668 }
669}
670
671static void channel_free(struct channel *chan,
672 struct lttng_ust_shm_handle *handle)
852c2936 673{
74d81a6c 674 channel_backend_free(&chan->backend, handle);
431d5cf0 675 /* chan is freed by shm teardown */
1d498196
MD
676 shm_object_table_destroy(handle->table);
677 free(handle);
852c2936
MD
678}
679
680/**
681 * channel_create - Create channel.
682 * @config: ring buffer instance configuration
683 * @name: name of the channel
a3f61e7f
MD
684 * @priv_data: ring buffer client private data area pointer (output)
685 * @priv_data_size: length, in bytes, of the private data area.
d028eddb 686 * @priv_data_init: initialization data for private data.
852c2936
MD
687 * @buf_addr: pointer the the beginning of the preallocated buffer contiguous
688 * address mapping. It is used only by RING_BUFFER_STATIC
689 * configuration. It can be set to NULL for other backends.
690 * @subbuf_size: subbuffer size
691 * @num_subbuf: number of subbuffers
692 * @switch_timer_interval: Time interval (in us) to fill sub-buffers with
693 * padding to let readers get those sub-buffers.
694 * Used for live streaming.
695 * @read_timer_interval: Time interval (in us) to wake up pending readers.
696 *
697 * Holds cpu hotplug.
698 * Returns NULL on failure.
699 */
4cfec15c 700struct lttng_ust_shm_handle *channel_create(const struct lttng_ust_lib_ring_buffer_config *config,
a3f61e7f
MD
701 const char *name,
702 void **priv_data,
703 size_t priv_data_align,
704 size_t priv_data_size,
d028eddb 705 void *priv_data_init,
a3f61e7f 706 void *buf_addr, size_t subbuf_size,
852c2936 707 size_t num_subbuf, unsigned int switch_timer_interval,
74d81a6c 708 unsigned int read_timer_interval)
852c2936 709{
24622edc 710 int ret;
a3f61e7f 711 size_t shmsize, chansize;
852c2936 712 struct channel *chan;
38fae1d3 713 struct lttng_ust_shm_handle *handle;
1d498196 714 struct shm_object *shmobj;
74d81a6c
MD
715 unsigned int nr_streams;
716
717 if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
718 nr_streams = num_possible_cpus();
719 else
720 nr_streams = 1;
852c2936
MD
721
722 if (lib_ring_buffer_check_config(config, switch_timer_interval,
723 read_timer_interval))
724 return NULL;
725
38fae1d3 726 handle = zmalloc(sizeof(struct lttng_ust_shm_handle));
431d5cf0
MD
727 if (!handle)
728 return NULL;
729
1d498196
MD
730 /* Allocate table for channel + per-cpu buffers */
731 handle->table = shm_object_table_create(1 + num_possible_cpus());
732 if (!handle->table)
733 goto error_table_alloc;
852c2936 734
1d498196
MD
735 /* Calculate the shm allocation layout */
736 shmsize = sizeof(struct channel);
c1fca457 737 shmsize += offset_align(shmsize, __alignof__(struct lttng_ust_lib_ring_buffer_shmp));
74d81a6c 738 shmsize += sizeof(struct lttng_ust_lib_ring_buffer_shmp) * nr_streams;
a3f61e7f 739 chansize = shmsize;
74d81a6c
MD
740 if (priv_data_align)
741 shmsize += offset_align(shmsize, priv_data_align);
a3f61e7f 742 shmsize += priv_data_size;
a6352fd4 743
74d81a6c
MD
744 /* Allocate normal memory for channel (not shared) */
745 shmobj = shm_object_table_alloc(handle->table, shmsize, SHM_OBJECT_MEM);
b5a14697
MD
746 if (!shmobj)
747 goto error_append;
57773204 748 /* struct channel is at object 0, offset 0 (hardcoded) */
a3f61e7f 749 set_shmp(handle->chan, zalloc_shm(shmobj, chansize));
57773204
MD
750 assert(handle->chan._ref.index == 0);
751 assert(handle->chan._ref.offset == 0);
1d498196 752 chan = shmp(handle, handle->chan);
a6352fd4 753 if (!chan)
1d498196 754 goto error_append;
74d81a6c 755 chan->nr_streams = nr_streams;
a6352fd4 756
a3f61e7f
MD
757 /* space for private data */
758 if (priv_data_size) {
759 DECLARE_SHMP(void, priv_data_alloc);
760
761 align_shm(shmobj, priv_data_align);
762 chan->priv_data_offset = shmobj->allocated_len;
763 set_shmp(priv_data_alloc, zalloc_shm(shmobj, priv_data_size));
764 if (!shmp(handle, priv_data_alloc))
765 goto error_append;
766 *priv_data = channel_get_private(chan);
d028eddb 767 memcpy(*priv_data, priv_data_init, priv_data_size);
a3f61e7f
MD
768 } else {
769 chan->priv_data_offset = -1;
74d81a6c
MD
770 if (priv_data)
771 *priv_data = NULL;
a3f61e7f
MD
772 }
773
774 ret = channel_backend_init(&chan->backend, name, config,
1d498196 775 subbuf_size, num_subbuf, handle);
852c2936 776 if (ret)
1d498196 777 goto error_backend_init;
852c2936 778
03d2d293 779 chan->handle = handle;
852c2936 780 chan->commit_count_mask = (~0UL >> chan->backend.num_subbuf_order);
852c2936 781
34a91bdb
MD
782 chan->switch_timer_interval = switch_timer_interval;
783 chan->read_timer_interval = read_timer_interval;
03d2d293 784 lib_ring_buffer_channel_switch_timer_start(chan);
34a91bdb 785 lib_ring_buffer_channel_read_timer_start(chan);
852c2936 786
431d5cf0 787 return handle;
852c2936 788
1d498196
MD
789error_backend_init:
790error_append:
791 shm_object_table_destroy(handle->table);
792error_table_alloc:
431d5cf0 793 free(handle);
852c2936
MD
794 return NULL;
795}
852c2936 796
74d81a6c 797struct lttng_ust_shm_handle *channel_handle_create(void *data,
ff0f5728
MD
798 uint64_t memory_map_size,
799 int wakeup_fd)
193183fb 800{
38fae1d3 801 struct lttng_ust_shm_handle *handle;
193183fb
MD
802 struct shm_object *object;
803
38fae1d3 804 handle = zmalloc(sizeof(struct lttng_ust_shm_handle));
193183fb
MD
805 if (!handle)
806 return NULL;
807
808 /* Allocate table for channel + per-cpu buffers */
809 handle->table = shm_object_table_create(1 + num_possible_cpus());
810 if (!handle->table)
811 goto error_table_alloc;
812 /* Add channel object */
74d81a6c 813 object = shm_object_table_append_mem(handle->table, data,
ff0f5728 814 memory_map_size, wakeup_fd);
193183fb
MD
815 if (!object)
816 goto error_table_object;
57773204
MD
817 /* struct channel is at object 0, offset 0 (hardcoded) */
818 handle->chan._ref.index = 0;
819 handle->chan._ref.offset = 0;
193183fb
MD
820 return handle;
821
822error_table_object:
823 shm_object_table_destroy(handle->table);
824error_table_alloc:
825 free(handle);
826 return NULL;
827}
828
38fae1d3 829int channel_handle_add_stream(struct lttng_ust_shm_handle *handle,
74d81a6c
MD
830 int shm_fd, int wakeup_fd, uint32_t stream_nr,
831 uint64_t memory_map_size)
193183fb
MD
832{
833 struct shm_object *object;
834
835 /* Add stream object */
74d81a6c
MD
836 object = shm_object_table_append_shm(handle->table,
837 shm_fd, wakeup_fd, stream_nr,
838 memory_map_size);
193183fb 839 if (!object)
74d81a6c 840 return -EINVAL;
193183fb
MD
841 return 0;
842}
843
74d81a6c
MD
844unsigned int channel_handle_get_nr_streams(struct lttng_ust_shm_handle *handle)
845{
846 assert(handle->table);
847 return handle->table->allocated_len - 1;
848}
849
852c2936 850static
74d81a6c 851void channel_release(struct channel *chan, struct lttng_ust_shm_handle *handle)
852c2936 852{
74d81a6c 853 channel_free(chan, handle);
852c2936
MD
854}
855
856/**
857 * channel_destroy - Finalize, wait for q.s. and destroy channel.
858 * @chan: channel to destroy
859 *
860 * Holds cpu hotplug.
431d5cf0
MD
861 * Call "destroy" callback, finalize channels, decrement the channel
862 * reference count. Note that when readers have completed data
863 * consumption of finalized channels, get_subbuf() will return -ENODATA.
a3f61e7f 864 * They should release their handle at that point.
852c2936 865 */
a3f61e7f 866void channel_destroy(struct channel *chan, struct lttng_ust_shm_handle *handle,
74d81a6c 867 int consumer)
852c2936 868{
74d81a6c
MD
869 if (consumer) {
870 /*
871 * Note: the consumer takes care of finalizing and
872 * switching the buffers.
873 */
874 channel_unregister_notifiers(chan, handle);
3d0ef9f6
MD
875 /*
876 * The consumer prints errors.
877 */
878 channel_print_errors(chan, handle);
824f40b8
MD
879 }
880
431d5cf0
MD
881 /*
882 * sessiond/consumer are keeping a reference on the shm file
883 * descriptor directly. No need to refcount.
884 */
74d81a6c 885 channel_release(chan, handle);
a3f61e7f 886 return;
852c2936 887}
852c2936 888
4cfec15c
MD
889struct lttng_ust_lib_ring_buffer *channel_get_ring_buffer(
890 const struct lttng_ust_lib_ring_buffer_config *config,
1d498196 891 struct channel *chan, int cpu,
38fae1d3 892 struct lttng_ust_shm_handle *handle,
74d81a6c
MD
893 int *shm_fd, int *wait_fd,
894 int *wakeup_fd,
895 uint64_t *memory_map_size)
852c2936 896{
381c0f1e
MD
897 struct shm_ref *ref;
898
899 if (config->alloc == RING_BUFFER_ALLOC_GLOBAL) {
74d81a6c 900 cpu = 0;
381c0f1e 901 } else {
e095d803
MD
902 if (cpu >= num_possible_cpus())
903 return NULL;
381c0f1e 904 }
74d81a6c
MD
905 ref = &chan->backend.buf[cpu].shmp._ref;
906 *shm_fd = shm_get_shm_fd(handle, ref);
907 *wait_fd = shm_get_wait_fd(handle, ref);
908 *wakeup_fd = shm_get_wakeup_fd(handle, ref);
909 if (shm_get_shm_size(handle, ref, memory_map_size))
910 return NULL;
911 return shmp(handle, chan->backend.buf[cpu].shmp);
852c2936 912}
852c2936 913
ff0f5728
MD
914int ring_buffer_channel_close_wait_fd(const struct lttng_ust_lib_ring_buffer_config *config,
915 struct channel *chan,
916 struct lttng_ust_shm_handle *handle)
917{
918 struct shm_ref *ref;
919
920 ref = &handle->chan._ref;
921 return shm_close_wait_fd(handle, ref);
922}
923
924int ring_buffer_channel_close_wakeup_fd(const struct lttng_ust_lib_ring_buffer_config *config,
925 struct channel *chan,
926 struct lttng_ust_shm_handle *handle)
927{
928 struct shm_ref *ref;
929
930 ref = &handle->chan._ref;
931 return shm_close_wakeup_fd(handle, ref);
932}
933
934int ring_buffer_stream_close_wait_fd(const struct lttng_ust_lib_ring_buffer_config *config,
74d81a6c
MD
935 struct channel *chan,
936 struct lttng_ust_shm_handle *handle,
937 int cpu)
852c2936 938{
74d81a6c
MD
939 struct shm_ref *ref;
940
941 if (config->alloc == RING_BUFFER_ALLOC_GLOBAL) {
942 cpu = 0;
943 } else {
944 if (cpu >= num_possible_cpus())
945 return -EINVAL;
824f40b8 946 }
74d81a6c
MD
947 ref = &chan->backend.buf[cpu].shmp._ref;
948 return shm_close_wait_fd(handle, ref);
949}
950
ff0f5728 951int ring_buffer_stream_close_wakeup_fd(const struct lttng_ust_lib_ring_buffer_config *config,
74d81a6c
MD
952 struct channel *chan,
953 struct lttng_ust_shm_handle *handle,
954 int cpu)
955{
956 struct shm_ref *ref;
cb7378b3 957 int ret;
74d81a6c
MD
958
959 if (config->alloc == RING_BUFFER_ALLOC_GLOBAL) {
960 cpu = 0;
961 } else {
962 if (cpu >= num_possible_cpus())
963 return -EINVAL;
964 }
965 ref = &chan->backend.buf[cpu].shmp._ref;
cb7378b3
MD
966 pthread_mutex_lock(&wakeup_fd_mutex);
967 ret = shm_close_wakeup_fd(handle, ref);
968 pthread_mutex_unlock(&wakeup_fd_mutex);
969 return ret;
74d81a6c
MD
970}
971
972int lib_ring_buffer_open_read(struct lttng_ust_lib_ring_buffer *buf,
973 struct lttng_ust_shm_handle *handle)
974{
a6352fd4 975 if (uatomic_cmpxchg(&buf->active_readers, 0, 1) != 0)
852c2936 976 return -EBUSY;
a6352fd4 977 cmm_smp_mb();
852c2936
MD
978 return 0;
979}
852c2936 980
4cfec15c 981void lib_ring_buffer_release_read(struct lttng_ust_lib_ring_buffer *buf,
74d81a6c 982 struct lttng_ust_shm_handle *handle)
852c2936 983{
1d498196 984 struct channel *chan = shmp(handle, buf->backend.chan);
852c2936 985
a6352fd4
MD
986 CHAN_WARN_ON(chan, uatomic_read(&buf->active_readers) != 1);
987 cmm_smp_mb();
988 uatomic_dec(&buf->active_readers);
852c2936
MD
989}
990
991/**
992 * lib_ring_buffer_snapshot - save subbuffer position snapshot (for read)
993 * @buf: ring buffer
994 * @consumed: consumed count indicating the position where to read
995 * @produced: produced count, indicates position when to stop reading
996 *
997 * Returns -ENODATA if buffer is finalized, -EAGAIN if there is currently no
998 * data to read at consumed position, or 0 if the get operation succeeds.
852c2936
MD
999 */
1000
4cfec15c 1001int lib_ring_buffer_snapshot(struct lttng_ust_lib_ring_buffer *buf,
1d498196 1002 unsigned long *consumed, unsigned long *produced,
38fae1d3 1003 struct lttng_ust_shm_handle *handle)
852c2936 1004{
1d498196 1005 struct channel *chan = shmp(handle, buf->backend.chan);
4cfec15c 1006 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1007 unsigned long consumed_cur, write_offset;
1008 int finalized;
1009
14641deb 1010 finalized = CMM_ACCESS_ONCE(buf->finalized);
852c2936
MD
1011 /*
1012 * Read finalized before counters.
1013 */
a6352fd4
MD
1014 cmm_smp_rmb();
1015 consumed_cur = uatomic_read(&buf->consumed);
852c2936
MD
1016 /*
1017 * No need to issue a memory barrier between consumed count read and
1018 * write offset read, because consumed count can only change
1019 * concurrently in overwrite mode, and we keep a sequence counter
1020 * identifier derived from the write offset to check we are getting
1021 * the same sub-buffer we are expecting (the sub-buffers are atomically
1022 * "tagged" upon writes, tags are checked upon read).
1023 */
1024 write_offset = v_read(config, &buf->offset);
1025
1026 /*
1027 * Check that we are not about to read the same subbuffer in
1028 * which the writer head is.
1029 */
1030 if (subbuf_trunc(write_offset, chan) - subbuf_trunc(consumed_cur, chan)
1031 == 0)
1032 goto nodata;
1033
1034 *consumed = consumed_cur;
1035 *produced = subbuf_trunc(write_offset, chan);
1036
1037 return 0;
1038
1039nodata:
1040 /*
1041 * The memory barriers __wait_event()/wake_up_interruptible() take care
1042 * of "raw_spin_is_locked" memory ordering.
1043 */
1044 if (finalized)
1045 return -ENODATA;
852c2936
MD
1046 else
1047 return -EAGAIN;
1048}
852c2936
MD
1049
1050/**
d1a996f5 1051 * lib_ring_buffer_move_consumer - move consumed counter forward
852c2936
MD
1052 * @buf: ring buffer
1053 * @consumed_new: new consumed count value
1054 */
4cfec15c 1055void lib_ring_buffer_move_consumer(struct lttng_ust_lib_ring_buffer *buf,
1d498196 1056 unsigned long consumed_new,
38fae1d3 1057 struct lttng_ust_shm_handle *handle)
852c2936 1058{
4cfec15c 1059 struct lttng_ust_lib_ring_buffer_backend *bufb = &buf->backend;
1d498196 1060 struct channel *chan = shmp(handle, bufb->chan);
852c2936
MD
1061 unsigned long consumed;
1062
74d81a6c 1063 CHAN_WARN_ON(chan, uatomic_read(&buf->active_readers) != 1);
852c2936
MD
1064
1065 /*
1066 * Only push the consumed value forward.
1067 * If the consumed cmpxchg fails, this is because we have been pushed by
1068 * the writer in flight recorder mode.
1069 */
a6352fd4 1070 consumed = uatomic_read(&buf->consumed);
852c2936 1071 while ((long) consumed - (long) consumed_new < 0)
a6352fd4
MD
1072 consumed = uatomic_cmpxchg(&buf->consumed, consumed,
1073 consumed_new);
852c2936 1074}
852c2936
MD
1075
1076/**
1077 * lib_ring_buffer_get_subbuf - get exclusive access to subbuffer for reading
1078 * @buf: ring buffer
1079 * @consumed: consumed count indicating the position where to read
1080 *
1081 * Returns -ENODATA if buffer is finalized, -EAGAIN if there is currently no
1082 * data to read at consumed position, or 0 if the get operation succeeds.
852c2936 1083 */
4cfec15c 1084int lib_ring_buffer_get_subbuf(struct lttng_ust_lib_ring_buffer *buf,
1d498196 1085 unsigned long consumed,
38fae1d3 1086 struct lttng_ust_shm_handle *handle)
852c2936 1087{
1d498196 1088 struct channel *chan = shmp(handle, buf->backend.chan);
4cfec15c 1089 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1090 unsigned long consumed_cur, consumed_idx, commit_count, write_offset;
1091 int ret;
1092 int finalized;
1093
1094retry:
14641deb 1095 finalized = CMM_ACCESS_ONCE(buf->finalized);
852c2936
MD
1096 /*
1097 * Read finalized before counters.
1098 */
a6352fd4
MD
1099 cmm_smp_rmb();
1100 consumed_cur = uatomic_read(&buf->consumed);
852c2936 1101 consumed_idx = subbuf_index(consumed, chan);
4746ae29 1102 commit_count = v_read(config, &shmp_index(handle, buf->commit_cold, consumed_idx)->cc_sb);
852c2936
MD
1103 /*
1104 * Make sure we read the commit count before reading the buffer
1105 * data and the write offset. Correct consumed offset ordering
1106 * wrt commit count is insured by the use of cmpxchg to update
1107 * the consumed offset.
852c2936 1108 */
a6352fd4
MD
1109 /*
1110 * Local rmb to match the remote wmb to read the commit count
1111 * before the buffer data and the write offset.
1112 */
1113 cmm_smp_rmb();
852c2936
MD
1114
1115 write_offset = v_read(config, &buf->offset);
1116
1117 /*
1118 * Check that the buffer we are getting is after or at consumed_cur
1119 * position.
1120 */
1121 if ((long) subbuf_trunc(consumed, chan)
1122 - (long) subbuf_trunc(consumed_cur, chan) < 0)
1123 goto nodata;
1124
1125 /*
1126 * Check that the subbuffer we are trying to consume has been
1127 * already fully committed.
1128 */
1129 if (((commit_count - chan->backend.subbuf_size)
1130 & chan->commit_count_mask)
1131 - (buf_trunc(consumed_cur, chan)
1132 >> chan->backend.num_subbuf_order)
1133 != 0)
1134 goto nodata;
1135
1136 /*
1137 * Check that we are not about to read the same subbuffer in
1138 * which the writer head is.
1139 */
1140 if (subbuf_trunc(write_offset, chan) - subbuf_trunc(consumed_cur, chan)
1141 == 0)
1142 goto nodata;
1143
1144 /*
1145 * Failure to get the subbuffer causes a busy-loop retry without going
1146 * to a wait queue. These are caused by short-lived race windows where
1147 * the writer is getting access to a subbuffer we were trying to get
1148 * access to. Also checks that the "consumed" buffer count we are
1149 * looking for matches the one contained in the subbuffer id.
1150 */
1151 ret = update_read_sb_index(config, &buf->backend, &chan->backend,
1d498196
MD
1152 consumed_idx, buf_trunc_val(consumed, chan),
1153 handle);
852c2936
MD
1154 if (ret)
1155 goto retry;
1156 subbuffer_id_clear_noref(config, &buf->backend.buf_rsb.id);
1157
1158 buf->get_subbuf_consumed = consumed;
1159 buf->get_subbuf = 1;
1160
1161 return 0;
1162
1163nodata:
1164 /*
1165 * The memory barriers __wait_event()/wake_up_interruptible() take care
1166 * of "raw_spin_is_locked" memory ordering.
1167 */
1168 if (finalized)
1169 return -ENODATA;
852c2936
MD
1170 else
1171 return -EAGAIN;
1172}
852c2936
MD
1173
1174/**
1175 * lib_ring_buffer_put_subbuf - release exclusive subbuffer access
1176 * @buf: ring buffer
1177 */
4cfec15c 1178void lib_ring_buffer_put_subbuf(struct lttng_ust_lib_ring_buffer *buf,
38fae1d3 1179 struct lttng_ust_shm_handle *handle)
852c2936 1180{
4cfec15c 1181 struct lttng_ust_lib_ring_buffer_backend *bufb = &buf->backend;
1d498196 1182 struct channel *chan = shmp(handle, bufb->chan);
4cfec15c 1183 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1184 unsigned long read_sb_bindex, consumed_idx, consumed;
1185
74d81a6c 1186 CHAN_WARN_ON(chan, uatomic_read(&buf->active_readers) != 1);
852c2936
MD
1187
1188 if (!buf->get_subbuf) {
1189 /*
1190 * Reader puts a subbuffer it did not get.
1191 */
1192 CHAN_WARN_ON(chan, 1);
1193 return;
1194 }
1195 consumed = buf->get_subbuf_consumed;
1196 buf->get_subbuf = 0;
1197
1198 /*
1199 * Clear the records_unread counter. (overruns counter)
1200 * Can still be non-zero if a file reader simply grabbed the data
1201 * without using iterators.
1202 * Can be below zero if an iterator is used on a snapshot more than
1203 * once.
1204 */
1205 read_sb_bindex = subbuffer_id_get_index(config, bufb->buf_rsb.id);
1206 v_add(config, v_read(config,
4746ae29 1207 &shmp(handle, shmp_index(handle, bufb->array, read_sb_bindex)->shmp)->records_unread),
852c2936 1208 &bufb->records_read);
4746ae29 1209 v_set(config, &shmp(handle, shmp_index(handle, bufb->array, read_sb_bindex)->shmp)->records_unread, 0);
852c2936
MD
1210 CHAN_WARN_ON(chan, config->mode == RING_BUFFER_OVERWRITE
1211 && subbuffer_id_is_noref(config, bufb->buf_rsb.id));
1212 subbuffer_id_set_noref(config, &bufb->buf_rsb.id);
1213
1214 /*
1215 * Exchange the reader subbuffer with the one we put in its place in the
1216 * writer subbuffer table. Expect the original consumed count. If
1217 * update_read_sb_index fails, this is because the writer updated the
1218 * subbuffer concurrently. We should therefore keep the subbuffer we
1219 * currently have: it has become invalid to try reading this sub-buffer
1220 * consumed count value anyway.
1221 */
1222 consumed_idx = subbuf_index(consumed, chan);
1223 update_read_sb_index(config, &buf->backend, &chan->backend,
1d498196
MD
1224 consumed_idx, buf_trunc_val(consumed, chan),
1225 handle);
852c2936
MD
1226 /*
1227 * update_read_sb_index return value ignored. Don't exchange sub-buffer
1228 * if the writer concurrently updated it.
1229 */
1230}
852c2936
MD
1231
1232/*
1233 * cons_offset is an iterator on all subbuffer offsets between the reader
1234 * position and the writer position. (inclusive)
1235 */
1236static
4cfec15c 1237void lib_ring_buffer_print_subbuffer_errors(struct lttng_ust_lib_ring_buffer *buf,
852c2936
MD
1238 struct channel *chan,
1239 unsigned long cons_offset,
1d498196 1240 int cpu,
38fae1d3 1241 struct lttng_ust_shm_handle *handle)
852c2936 1242{
4cfec15c 1243 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1244 unsigned long cons_idx, commit_count, commit_count_sb;
1245
1246 cons_idx = subbuf_index(cons_offset, chan);
4746ae29
MD
1247 commit_count = v_read(config, &shmp_index(handle, buf->commit_hot, cons_idx)->cc);
1248 commit_count_sb = v_read(config, &shmp_index(handle, buf->commit_cold, cons_idx)->cc_sb);
852c2936
MD
1249
1250 if (subbuf_offset(commit_count, chan) != 0)
4d3c9523 1251 DBG("ring buffer %s, cpu %d: "
852c2936
MD
1252 "commit count in subbuffer %lu,\n"
1253 "expecting multiples of %lu bytes\n"
1254 " [ %lu bytes committed, %lu bytes reader-visible ]\n",
1255 chan->backend.name, cpu, cons_idx,
1256 chan->backend.subbuf_size,
1257 commit_count, commit_count_sb);
1258
4d3c9523 1259 DBG("ring buffer: %s, cpu %d: %lu bytes committed\n",
852c2936
MD
1260 chan->backend.name, cpu, commit_count);
1261}
1262
1263static
4cfec15c 1264void lib_ring_buffer_print_buffer_errors(struct lttng_ust_lib_ring_buffer *buf,
852c2936 1265 struct channel *chan,
1d498196 1266 void *priv, int cpu,
38fae1d3 1267 struct lttng_ust_shm_handle *handle)
852c2936 1268{
4cfec15c 1269 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1270 unsigned long write_offset, cons_offset;
1271
852c2936
MD
1272 /*
1273 * No need to order commit_count, write_offset and cons_offset reads
1274 * because we execute at teardown when no more writer nor reader
1275 * references are left.
1276 */
1277 write_offset = v_read(config, &buf->offset);
a6352fd4 1278 cons_offset = uatomic_read(&buf->consumed);
852c2936 1279 if (write_offset != cons_offset)
4d3c9523 1280 DBG("ring buffer %s, cpu %d: "
852c2936
MD
1281 "non-consumed data\n"
1282 " [ %lu bytes written, %lu bytes read ]\n",
1283 chan->backend.name, cpu, write_offset, cons_offset);
1284
a6352fd4 1285 for (cons_offset = uatomic_read(&buf->consumed);
852c2936
MD
1286 (long) (subbuf_trunc((unsigned long) v_read(config, &buf->offset),
1287 chan)
1288 - cons_offset) > 0;
1289 cons_offset = subbuf_align(cons_offset, chan))
1290 lib_ring_buffer_print_subbuffer_errors(buf, chan, cons_offset,
1d498196 1291 cpu, handle);
852c2936
MD
1292}
1293
1294static
1295void lib_ring_buffer_print_errors(struct channel *chan,
009769ca
MD
1296 struct lttng_ust_lib_ring_buffer *buf, int cpu,
1297 struct lttng_ust_shm_handle *handle)
852c2936 1298{
4cfec15c 1299 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
a3f61e7f 1300 void *priv = channel_get_private(chan);
852c2936 1301
a1360615
MD
1302 if (!strcmp(chan->backend.name, "relay-metadata-mmap")) {
1303 DBG("ring buffer %s: %lu records written, "
1304 "%lu records overrun\n",
1305 chan->backend.name,
1306 v_read(config, &buf->records_count),
1307 v_read(config, &buf->records_overrun));
1308 } else {
1309 DBG("ring buffer %s, cpu %d: %lu records written, "
1310 "%lu records overrun\n",
1311 chan->backend.name, cpu,
1312 v_read(config, &buf->records_count),
1313 v_read(config, &buf->records_overrun));
1314
1315 if (v_read(config, &buf->records_lost_full)
1316 || v_read(config, &buf->records_lost_wrap)
1317 || v_read(config, &buf->records_lost_big))
1318 DBG("ring buffer %s, cpu %d: records were lost. Caused by:\n"
1319 " [ %lu buffer full, %lu nest buffer wrap-around, "
1320 "%lu event too big ]\n",
1321 chan->backend.name, cpu,
1322 v_read(config, &buf->records_lost_full),
1323 v_read(config, &buf->records_lost_wrap),
1324 v_read(config, &buf->records_lost_big));
1325 }
1d498196 1326 lib_ring_buffer_print_buffer_errors(buf, chan, priv, cpu, handle);
852c2936
MD
1327}
1328
1329/*
1330 * lib_ring_buffer_switch_old_start: Populate old subbuffer header.
1331 *
1332 * Only executed when the buffer is finalized, in SWITCH_FLUSH.
1333 */
1334static
4cfec15c 1335void lib_ring_buffer_switch_old_start(struct lttng_ust_lib_ring_buffer *buf,
852c2936
MD
1336 struct channel *chan,
1337 struct switch_offsets *offsets,
2fed87ae 1338 uint64_t tsc,
38fae1d3 1339 struct lttng_ust_shm_handle *handle)
852c2936 1340{
4cfec15c 1341 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1342 unsigned long oldidx = subbuf_index(offsets->old, chan);
1343 unsigned long commit_count;
1344
1d498196 1345 config->cb.buffer_begin(buf, tsc, oldidx, handle);
852c2936
MD
1346
1347 /*
1348 * Order all writes to buffer before the commit count update that will
1349 * determine that the subbuffer is full.
1350 */
a6352fd4 1351 cmm_smp_wmb();
852c2936 1352 v_add(config, config->cb.subbuffer_header_size(),
4746ae29
MD
1353 &shmp_index(handle, buf->commit_hot, oldidx)->cc);
1354 commit_count = v_read(config, &shmp_index(handle, buf->commit_hot, oldidx)->cc);
852c2936
MD
1355 /* Check if the written buffer has to be delivered */
1356 lib_ring_buffer_check_deliver(config, buf, chan, offsets->old,
1d498196 1357 commit_count, oldidx, handle);
852c2936
MD
1358 lib_ring_buffer_write_commit_counter(config, buf, chan, oldidx,
1359 offsets->old, commit_count,
1d498196
MD
1360 config->cb.subbuffer_header_size(),
1361 handle);
852c2936
MD
1362}
1363
1364/*
1365 * lib_ring_buffer_switch_old_end: switch old subbuffer
1366 *
1367 * Note : offset_old should never be 0 here. It is ok, because we never perform
1368 * buffer switch on an empty subbuffer in SWITCH_ACTIVE mode. The caller
1369 * increments the offset_old value when doing a SWITCH_FLUSH on an empty
1370 * subbuffer.
1371 */
1372static
4cfec15c 1373void lib_ring_buffer_switch_old_end(struct lttng_ust_lib_ring_buffer *buf,
852c2936
MD
1374 struct channel *chan,
1375 struct switch_offsets *offsets,
2fed87ae 1376 uint64_t tsc,
38fae1d3 1377 struct lttng_ust_shm_handle *handle)
852c2936 1378{
4cfec15c 1379 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1380 unsigned long oldidx = subbuf_index(offsets->old - 1, chan);
1381 unsigned long commit_count, padding_size, data_size;
1382
1383 data_size = subbuf_offset(offsets->old - 1, chan) + 1;
1384 padding_size = chan->backend.subbuf_size - data_size;
1d498196
MD
1385 subbuffer_set_data_size(config, &buf->backend, oldidx, data_size,
1386 handle);
852c2936
MD
1387
1388 /*
1389 * Order all writes to buffer before the commit count update that will
1390 * determine that the subbuffer is full.
1391 */
a6352fd4 1392 cmm_smp_wmb();
4746ae29
MD
1393 v_add(config, padding_size, &shmp_index(handle, buf->commit_hot, oldidx)->cc);
1394 commit_count = v_read(config, &shmp_index(handle, buf->commit_hot, oldidx)->cc);
852c2936 1395 lib_ring_buffer_check_deliver(config, buf, chan, offsets->old - 1,
1d498196 1396 commit_count, oldidx, handle);
852c2936
MD
1397 lib_ring_buffer_write_commit_counter(config, buf, chan, oldidx,
1398 offsets->old, commit_count,
1d498196 1399 padding_size, handle);
852c2936
MD
1400}
1401
1402/*
1403 * lib_ring_buffer_switch_new_start: Populate new subbuffer.
1404 *
1405 * This code can be executed unordered : writers may already have written to the
1406 * sub-buffer before this code gets executed, caution. The commit makes sure
1407 * that this code is executed before the deliver of this sub-buffer.
1408 */
1409static
4cfec15c 1410void lib_ring_buffer_switch_new_start(struct lttng_ust_lib_ring_buffer *buf,
852c2936
MD
1411 struct channel *chan,
1412 struct switch_offsets *offsets,
2fed87ae 1413 uint64_t tsc,
38fae1d3 1414 struct lttng_ust_shm_handle *handle)
852c2936 1415{
4cfec15c 1416 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1417 unsigned long beginidx = subbuf_index(offsets->begin, chan);
1418 unsigned long commit_count;
1419
1d498196 1420 config->cb.buffer_begin(buf, tsc, beginidx, handle);
852c2936
MD
1421
1422 /*
1423 * Order all writes to buffer before the commit count update that will
1424 * determine that the subbuffer is full.
1425 */
a6352fd4 1426 cmm_smp_wmb();
852c2936 1427 v_add(config, config->cb.subbuffer_header_size(),
4746ae29
MD
1428 &shmp_index(handle, buf->commit_hot, beginidx)->cc);
1429 commit_count = v_read(config, &shmp_index(handle, buf->commit_hot, beginidx)->cc);
852c2936
MD
1430 /* Check if the written buffer has to be delivered */
1431 lib_ring_buffer_check_deliver(config, buf, chan, offsets->begin,
1d498196 1432 commit_count, beginidx, handle);
852c2936
MD
1433 lib_ring_buffer_write_commit_counter(config, buf, chan, beginidx,
1434 offsets->begin, commit_count,
1d498196
MD
1435 config->cb.subbuffer_header_size(),
1436 handle);
852c2936
MD
1437}
1438
1439/*
1440 * lib_ring_buffer_switch_new_end: finish switching current subbuffer
1441 *
1442 * The only remaining threads could be the ones with pending commits. They will
1443 * have to do the deliver themselves.
1444 */
1445static
4cfec15c 1446void lib_ring_buffer_switch_new_end(struct lttng_ust_lib_ring_buffer *buf,
1d498196
MD
1447 struct channel *chan,
1448 struct switch_offsets *offsets,
2fed87ae 1449 uint64_t tsc,
38fae1d3 1450 struct lttng_ust_shm_handle *handle)
852c2936 1451{
4cfec15c 1452 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1453 unsigned long endidx = subbuf_index(offsets->end - 1, chan);
1454 unsigned long commit_count, padding_size, data_size;
1455
1456 data_size = subbuf_offset(offsets->end - 1, chan) + 1;
1457 padding_size = chan->backend.subbuf_size - data_size;
1d498196
MD
1458 subbuffer_set_data_size(config, &buf->backend, endidx, data_size,
1459 handle);
852c2936
MD
1460
1461 /*
1462 * Order all writes to buffer before the commit count update that will
1463 * determine that the subbuffer is full.
1464 */
a6352fd4 1465 cmm_smp_wmb();
4746ae29
MD
1466 v_add(config, padding_size, &shmp_index(handle, buf->commit_hot, endidx)->cc);
1467 commit_count = v_read(config, &shmp_index(handle, buf->commit_hot, endidx)->cc);
852c2936 1468 lib_ring_buffer_check_deliver(config, buf, chan, offsets->end - 1,
1d498196 1469 commit_count, endidx, handle);
852c2936
MD
1470 lib_ring_buffer_write_commit_counter(config, buf, chan, endidx,
1471 offsets->end, commit_count,
1d498196 1472 padding_size, handle);
852c2936
MD
1473}
1474
1475/*
1476 * Returns :
1477 * 0 if ok
1478 * !0 if execution must be aborted.
1479 */
1480static
1481int lib_ring_buffer_try_switch_slow(enum switch_mode mode,
4cfec15c 1482 struct lttng_ust_lib_ring_buffer *buf,
852c2936
MD
1483 struct channel *chan,
1484 struct switch_offsets *offsets,
2fed87ae 1485 uint64_t *tsc)
852c2936 1486{
4cfec15c 1487 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1488 unsigned long off;
1489
1490 offsets->begin = v_read(config, &buf->offset);
1491 offsets->old = offsets->begin;
1492 offsets->switch_old_start = 0;
1493 off = subbuf_offset(offsets->begin, chan);
1494
1495 *tsc = config->cb.ring_buffer_clock_read(chan);
1496
1497 /*
1498 * Ensure we flush the header of an empty subbuffer when doing the
1499 * finalize (SWITCH_FLUSH). This ensures that we end up knowing the
1500 * total data gathering duration even if there were no records saved
1501 * after the last buffer switch.
1502 * In SWITCH_ACTIVE mode, switch the buffer when it contains events.
1503 * SWITCH_ACTIVE only flushes the current subbuffer, dealing with end of
1504 * subbuffer header as appropriate.
1505 * The next record that reserves space will be responsible for
1506 * populating the following subbuffer header. We choose not to populate
1507 * the next subbuffer header here because we want to be able to use
a6352fd4
MD
1508 * SWITCH_ACTIVE for periodical buffer flush, which must
1509 * guarantee that all the buffer content (records and header
1510 * timestamps) are visible to the reader. This is required for
1511 * quiescence guarantees for the fusion merge.
852c2936
MD
1512 */
1513 if (mode == SWITCH_FLUSH || off > 0) {
b5a3dfa5 1514 if (caa_unlikely(off == 0)) {
c352d3a4
MD
1515 /*
1516 * A final flush that encounters an empty
1517 * sub-buffer cannot switch buffer if a
1518 * reader is located within this sub-buffer.
1519 * Anyway, the purpose of final flushing of a
1520 * sub-buffer at offset 0 is to handle the case
1521 * of entirely empty stream.
1522 */
1523 if (caa_unlikely(subbuf_trunc(offsets->begin, chan)
1524 - subbuf_trunc((unsigned long)
1525 uatomic_read(&buf->consumed), chan)
1526 >= chan->backend.buf_size))
1527 return -1;
852c2936
MD
1528 /*
1529 * The client does not save any header information.
1530 * Don't switch empty subbuffer on finalize, because it
1531 * is invalid to deliver a completely empty subbuffer.
1532 */
1533 if (!config->cb.subbuffer_header_size())
1534 return -1;
1535 /*
1536 * Need to write the subbuffer start header on finalize.
1537 */
1538 offsets->switch_old_start = 1;
1539 }
1540 offsets->begin = subbuf_align(offsets->begin, chan);
1541 } else
1542 return -1; /* we do not have to switch : buffer is empty */
1543 /* Note: old points to the next subbuf at offset 0 */
1544 offsets->end = offsets->begin;
1545 return 0;
1546}
1547
1548/*
1549 * Force a sub-buffer switch. This operation is completely reentrant : can be
1550 * called while tracing is active with absolutely no lock held.
1551 *
1552 * Note, however, that as a v_cmpxchg is used for some atomic
1553 * operations, this function must be called from the CPU which owns the buffer
1554 * for a ACTIVE flush.
1555 */
4cfec15c 1556void lib_ring_buffer_switch_slow(struct lttng_ust_lib_ring_buffer *buf, enum switch_mode mode,
38fae1d3 1557 struct lttng_ust_shm_handle *handle)
852c2936 1558{
1d498196 1559 struct channel *chan = shmp(handle, buf->backend.chan);
4cfec15c 1560 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
852c2936
MD
1561 struct switch_offsets offsets;
1562 unsigned long oldidx;
2fed87ae 1563 uint64_t tsc;
852c2936
MD
1564
1565 offsets.size = 0;
1566
1567 /*
1568 * Perform retryable operations.
1569 */
1570 do {
1571 if (lib_ring_buffer_try_switch_slow(mode, buf, chan, &offsets,
1572 &tsc))
1573 return; /* Switch not needed */
1574 } while (v_cmpxchg(config, &buf->offset, offsets.old, offsets.end)
1575 != offsets.old);
1576
1577 /*
1578 * Atomically update last_tsc. This update races against concurrent
1579 * atomic updates, but the race will always cause supplementary full TSC
1580 * records, never the opposite (missing a full TSC record when it would
1581 * be needed).
1582 */
1583 save_last_tsc(config, buf, tsc);
1584
1585 /*
1586 * Push the reader if necessary
1587 */
1588 lib_ring_buffer_reserve_push_reader(buf, chan, offsets.old);
1589
1590 oldidx = subbuf_index(offsets.old, chan);
1d498196 1591 lib_ring_buffer_clear_noref(config, &buf->backend, oldidx, handle);
852c2936
MD
1592
1593 /*
1594 * May need to populate header start on SWITCH_FLUSH.
1595 */
1596 if (offsets.switch_old_start) {
1d498196 1597 lib_ring_buffer_switch_old_start(buf, chan, &offsets, tsc, handle);
852c2936
MD
1598 offsets.old += config->cb.subbuffer_header_size();
1599 }
1600
1601 /*
1602 * Switch old subbuffer.
1603 */
1d498196 1604 lib_ring_buffer_switch_old_end(buf, chan, &offsets, tsc, handle);
852c2936 1605}
852c2936
MD
1606
1607/*
1608 * Returns :
1609 * 0 if ok
1610 * -ENOSPC if event size is too large for packet.
1611 * -ENOBUFS if there is currently not enough space in buffer for the event.
1612 * -EIO if data cannot be written into the buffer for any other reason.
1613 */
1614static
4cfec15c 1615int lib_ring_buffer_try_reserve_slow(struct lttng_ust_lib_ring_buffer *buf,
852c2936
MD
1616 struct channel *chan,
1617 struct switch_offsets *offsets,
4cfec15c 1618 struct lttng_ust_lib_ring_buffer_ctx *ctx)
852c2936 1619{
4cfec15c 1620 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
38fae1d3 1621 struct lttng_ust_shm_handle *handle = ctx->handle;
852c2936
MD
1622 unsigned long reserve_commit_diff;
1623
1624 offsets->begin = v_read(config, &buf->offset);
1625 offsets->old = offsets->begin;
1626 offsets->switch_new_start = 0;
1627 offsets->switch_new_end = 0;
1628 offsets->switch_old_end = 0;
1629 offsets->pre_header_padding = 0;
1630
1631 ctx->tsc = config->cb.ring_buffer_clock_read(chan);
1632 if ((int64_t) ctx->tsc == -EIO)
1633 return -EIO;
1634
1635 if (last_tsc_overflow(config, buf, ctx->tsc))
1636 ctx->rflags |= RING_BUFFER_RFLAG_FULL_TSC;
1637
b5a3dfa5 1638 if (caa_unlikely(subbuf_offset(offsets->begin, ctx->chan) == 0)) {
852c2936
MD
1639 offsets->switch_new_start = 1; /* For offsets->begin */
1640 } else {
1641 offsets->size = config->cb.record_header_size(config, chan,
1642 offsets->begin,
1643 &offsets->pre_header_padding,
1644 ctx);
1645 offsets->size +=
1646 lib_ring_buffer_align(offsets->begin + offsets->size,
1647 ctx->largest_align)
1648 + ctx->data_size;
b5a3dfa5 1649 if (caa_unlikely(subbuf_offset(offsets->begin, chan) +
852c2936
MD
1650 offsets->size > chan->backend.subbuf_size)) {
1651 offsets->switch_old_end = 1; /* For offsets->old */
1652 offsets->switch_new_start = 1; /* For offsets->begin */
1653 }
1654 }
b5a3dfa5 1655 if (caa_unlikely(offsets->switch_new_start)) {
852c2936
MD
1656 unsigned long sb_index;
1657
1658 /*
1659 * We are typically not filling the previous buffer completely.
1660 */
b5a3dfa5 1661 if (caa_likely(offsets->switch_old_end))
852c2936
MD
1662 offsets->begin = subbuf_align(offsets->begin, chan);
1663 offsets->begin = offsets->begin
1664 + config->cb.subbuffer_header_size();
1665 /* Test new buffer integrity */
1666 sb_index = subbuf_index(offsets->begin, chan);
1667 reserve_commit_diff =
1668 (buf_trunc(offsets->begin, chan)
1669 >> chan->backend.num_subbuf_order)
1670 - ((unsigned long) v_read(config,
4746ae29 1671 &shmp_index(handle, buf->commit_cold, sb_index)->cc_sb)
852c2936 1672 & chan->commit_count_mask);
b5a3dfa5 1673 if (caa_likely(reserve_commit_diff == 0)) {
852c2936 1674 /* Next subbuffer not being written to. */
b5a3dfa5 1675 if (caa_unlikely(config->mode != RING_BUFFER_OVERWRITE &&
852c2936
MD
1676 subbuf_trunc(offsets->begin, chan)
1677 - subbuf_trunc((unsigned long)
a6352fd4 1678 uatomic_read(&buf->consumed), chan)
852c2936 1679 >= chan->backend.buf_size)) {
64493e4f
MD
1680 unsigned long nr_lost;
1681
852c2936
MD
1682 /*
1683 * We do not overwrite non consumed buffers
1684 * and we are full : record is lost.
1685 */
64493e4f 1686 nr_lost = v_read(config, &buf->records_lost_full);
852c2936 1687 v_inc(config, &buf->records_lost_full);
64493e4f
MD
1688 if ((nr_lost & (DBG_PRINT_NR_LOST - 1)) == 0) {
1689 DBG("%lu or more records lost in (%s:%d) (buffer full)\n",
1690 nr_lost + 1, chan->backend.name,
1691 buf->backend.cpu);
1692 }
852c2936
MD
1693 return -ENOBUFS;
1694 } else {
1695 /*
1696 * Next subbuffer not being written to, and we
1697 * are either in overwrite mode or the buffer is
1698 * not full. It's safe to write in this new
1699 * subbuffer.
1700 */
1701 }
1702 } else {
64493e4f
MD
1703 unsigned long nr_lost;
1704
852c2936
MD
1705 /*
1706 * Next subbuffer reserve offset does not match the
1707 * commit offset. Drop record in producer-consumer and
1708 * overwrite mode. Caused by either a writer OOPS or too
1709 * many nested writes over a reserve/commit pair.
1710 */
64493e4f 1711 nr_lost = v_read(config, &buf->records_lost_wrap);
852c2936 1712 v_inc(config, &buf->records_lost_wrap);
64493e4f
MD
1713 if ((nr_lost & (DBG_PRINT_NR_LOST - 1)) == 0) {
1714 DBG("%lu or more records lost in (%s:%d) (wrap-around)\n",
1715 nr_lost + 1, chan->backend.name,
1716 buf->backend.cpu);
1717 }
852c2936
MD
1718 return -EIO;
1719 }
1720 offsets->size =
1721 config->cb.record_header_size(config, chan,
1722 offsets->begin,
1723 &offsets->pre_header_padding,
1724 ctx);
1725 offsets->size +=
1726 lib_ring_buffer_align(offsets->begin + offsets->size,
1727 ctx->largest_align)
1728 + ctx->data_size;
b5a3dfa5 1729 if (caa_unlikely(subbuf_offset(offsets->begin, chan)
852c2936 1730 + offsets->size > chan->backend.subbuf_size)) {
64493e4f
MD
1731 unsigned long nr_lost;
1732
852c2936
MD
1733 /*
1734 * Record too big for subbuffers, report error, don't
1735 * complete the sub-buffer switch.
1736 */
64493e4f 1737 nr_lost = v_read(config, &buf->records_lost_big);
852c2936 1738 v_inc(config, &buf->records_lost_big);
64493e4f
MD
1739 if ((nr_lost & (DBG_PRINT_NR_LOST - 1)) == 0) {
1740 DBG("%lu or more records lost in (%s:%d) record size "
1741 " of %zu bytes is too large for buffer\n",
1742 nr_lost + 1, chan->backend.name,
1743 buf->backend.cpu, offsets->size);
1744 }
852c2936
MD
1745 return -ENOSPC;
1746 } else {
1747 /*
1748 * We just made a successful buffer switch and the
1749 * record fits in the new subbuffer. Let's write.
1750 */
1751 }
1752 } else {
1753 /*
1754 * Record fits in the current buffer and we are not on a switch
1755 * boundary. It's safe to write.
1756 */
1757 }
1758 offsets->end = offsets->begin + offsets->size;
1759
b5a3dfa5 1760 if (caa_unlikely(subbuf_offset(offsets->end, chan) == 0)) {
852c2936
MD
1761 /*
1762 * The offset_end will fall at the very beginning of the next
1763 * subbuffer.
1764 */
1765 offsets->switch_new_end = 1; /* For offsets->begin */
1766 }
1767 return 0;
1768}
1769
1770/**
1771 * lib_ring_buffer_reserve_slow - Atomic slot reservation in a buffer.
1772 * @ctx: ring buffer context.
1773 *
1774 * Return : -NOBUFS if not enough space, -ENOSPC if event size too large,
1775 * -EIO for other errors, else returns 0.
1776 * It will take care of sub-buffer switching.
1777 */
4cfec15c 1778int lib_ring_buffer_reserve_slow(struct lttng_ust_lib_ring_buffer_ctx *ctx)
852c2936
MD
1779{
1780 struct channel *chan = ctx->chan;
38fae1d3 1781 struct lttng_ust_shm_handle *handle = ctx->handle;
4cfec15c
MD
1782 const struct lttng_ust_lib_ring_buffer_config *config = &chan->backend.config;
1783 struct lttng_ust_lib_ring_buffer *buf;
852c2936
MD
1784 struct switch_offsets offsets;
1785 int ret;
1786
1787 if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
1d498196 1788 buf = shmp(handle, chan->backend.buf[ctx->cpu].shmp);
852c2936 1789 else
1d498196 1790 buf = shmp(handle, chan->backend.buf[0].shmp);
852c2936
MD
1791 ctx->buf = buf;
1792
1793 offsets.size = 0;
1794
1795 do {
1796 ret = lib_ring_buffer_try_reserve_slow(buf, chan, &offsets,
1797 ctx);
b5a3dfa5 1798 if (caa_unlikely(ret))
852c2936 1799 return ret;
b5a3dfa5 1800 } while (caa_unlikely(v_cmpxchg(config, &buf->offset, offsets.old,
852c2936
MD
1801 offsets.end)
1802 != offsets.old));
1803
1804 /*
1805 * Atomically update last_tsc. This update races against concurrent
1806 * atomic updates, but the race will always cause supplementary full TSC
1807 * records, never the opposite (missing a full TSC record when it would
1808 * be needed).
1809 */
1810 save_last_tsc(config, buf, ctx->tsc);
1811
1812 /*
1813 * Push the reader if necessary
1814 */
1815 lib_ring_buffer_reserve_push_reader(buf, chan, offsets.end - 1);
1816
1817 /*
1818 * Clear noref flag for this subbuffer.
1819 */
1820 lib_ring_buffer_clear_noref(config, &buf->backend,
1d498196
MD
1821 subbuf_index(offsets.end - 1, chan),
1822 handle);
852c2936
MD
1823
1824 /*
1825 * Switch old subbuffer if needed.
1826 */
b5a3dfa5 1827 if (caa_unlikely(offsets.switch_old_end)) {
852c2936 1828 lib_ring_buffer_clear_noref(config, &buf->backend,
1d498196
MD
1829 subbuf_index(offsets.old - 1, chan),
1830 handle);
1831 lib_ring_buffer_switch_old_end(buf, chan, &offsets, ctx->tsc, handle);
852c2936
MD
1832 }
1833
1834 /*
1835 * Populate new subbuffer.
1836 */
b5a3dfa5 1837 if (caa_unlikely(offsets.switch_new_start))
1d498196 1838 lib_ring_buffer_switch_new_start(buf, chan, &offsets, ctx->tsc, handle);
852c2936 1839
b5a3dfa5 1840 if (caa_unlikely(offsets.switch_new_end))
1d498196 1841 lib_ring_buffer_switch_new_end(buf, chan, &offsets, ctx->tsc, handle);
852c2936
MD
1842
1843 ctx->slot_size = offsets.size;
1844 ctx->pre_offset = offsets.begin;
1845 ctx->buf_offset = offsets.begin + offsets.pre_header_padding;
1846 return 0;
1847}
f645cfa7
MD
1848
1849/*
1850 * Force a read (imply TLS fixup for dlopen) of TLS variables.
1851 */
1852void lttng_fixup_ringbuffer_tls(void)
1853{
8c90a710 1854 asm volatile ("" : : "m" (URCU_TLS(lib_ring_buffer_nesting)));
f645cfa7 1855}
03d2d293
MD
1856
1857void lib_ringbuffer_signal_init(void)
1858{
1859 sigset_t mask;
1860 int ret;
1861
1862 /*
1863 * Block signal for entire process, so only our thread processes
1864 * it.
1865 */
1866 rb_setmask(&mask);
1867 ret = pthread_sigmask(SIG_BLOCK, &mask, NULL);
1868 if (ret) {
1869 errno = ret;
1870 PERROR("pthread_sigmask");
1871 }
1872}
This page took 0.115853 seconds and 4 git commands to generate.