Define max nesting count constant
[lttng-modules.git] / lib / ringbuffer / frontend_api.h
CommitLineData
886d51a3
MD
1#ifndef _LIB_RING_BUFFER_FRONTEND_API_H
2#define _LIB_RING_BUFFER_FRONTEND_API_H
f3bc08c5
MD
3
4/*
886d51a3 5 * lib/ringbuffer/frontend_api.h
f3bc08c5
MD
6 *
7 * Ring Buffer Library Synchronization Header (buffer write API).
8 *
886d51a3
MD
9 * Copyright (C) 2005-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
10 *
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.
15 *
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.
20 *
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
24 *
f3bc08c5
MD
25 * Author:
26 * Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
27 *
28 * See ring_buffer_frontend.c for more information on wait-free algorithms.
29 * See linux/ringbuffer/frontend.h for channel allocation and read-side API.
f3bc08c5
MD
30 */
31
5671a661
MD
32#include <wrapper/ringbuffer/frontend.h>
33#include <wrapper/percpu-defs.h>
f3bc08c5 34#include <linux/errno.h>
dfbc2ec7 35#include <linux/prefetch.h>
f3bc08c5
MD
36
37/**
38 * lib_ring_buffer_get_cpu - Precedes ring buffer reserve/commit.
39 *
40 * Disables preemption (acts as a RCU read-side critical section) and keeps a
41 * ring buffer nesting count as supplementary safety net to ensure tracer client
42 * code will never trigger an endless recursion. Returns the processor ID on
43 * success, -EPERM on failure (nesting count too high).
44 *
45 * asm volatile and "memory" clobber prevent the compiler from moving
46 * instructions out of the ring buffer nesting count. This is required to ensure
47 * that probe side-effects which can cause recursion (e.g. unforeseen traps,
48 * divisions by 0, ...) are triggered within the incremented nesting count
49 * section.
50 */
51static inline
52int lib_ring_buffer_get_cpu(const struct lib_ring_buffer_config *config)
53{
54 int cpu, nesting;
55
56 rcu_read_lock_sched_notrace();
57 cpu = smp_processor_id();
58 nesting = ++per_cpu(lib_ring_buffer_nesting, cpu);
59 barrier();
60
b055caba 61 if (unlikely(nesting > RING_BUFFER_MAX_NESTING)) {
f3bc08c5
MD
62 WARN_ON_ONCE(1);
63 per_cpu(lib_ring_buffer_nesting, cpu)--;
64 rcu_read_unlock_sched_notrace();
65 return -EPERM;
66 } else
67 return cpu;
68}
69
70/**
71 * lib_ring_buffer_put_cpu - Follows ring buffer reserve/commit.
72 */
73static inline
74void lib_ring_buffer_put_cpu(const struct lib_ring_buffer_config *config)
75{
76 barrier();
e6b06d7d 77 (*lttng_this_cpu_ptr(&lib_ring_buffer_nesting))--;
f3bc08c5
MD
78 rcu_read_unlock_sched_notrace();
79}
80
81/*
82 * lib_ring_buffer_try_reserve is called by lib_ring_buffer_reserve(). It is not
83 * part of the API per se.
84 *
85 * returns 0 if reserve ok, or 1 if the slow path must be taken.
86 */
87static inline
88int lib_ring_buffer_try_reserve(const struct lib_ring_buffer_config *config,
89 struct lib_ring_buffer_ctx *ctx,
cc62f29e 90 void *client_ctx,
f3bc08c5
MD
91 unsigned long *o_begin, unsigned long *o_end,
92 unsigned long *o_old, size_t *before_hdr_pad)
93{
94 struct channel *chan = ctx->chan;
95 struct lib_ring_buffer *buf = ctx->buf;
96 *o_begin = v_read(config, &buf->offset);
97 *o_old = *o_begin;
98
99 ctx->tsc = lib_ring_buffer_clock_read(chan);
97ca2c54
MD
100 if ((int64_t) ctx->tsc == -EIO)
101 return 1;
f3bc08c5
MD
102
103 /*
104 * Prefetch cacheline for read because we have to read the previous
105 * commit counter to increment it and commit seq value to compare it to
106 * the commit counter.
107 */
108 prefetch(&buf->commit_hot[subbuf_index(*o_begin, chan)]);
109
110 if (last_tsc_overflow(config, buf, ctx->tsc))
64c796d8 111 ctx->rflags |= RING_BUFFER_RFLAG_FULL_TSC;
f3bc08c5
MD
112
113 if (unlikely(subbuf_offset(*o_begin, chan) == 0))
114 return 1;
115
116 ctx->slot_size = record_header_size(config, chan, *o_begin,
cc62f29e 117 before_hdr_pad, ctx, client_ctx);
f3bc08c5
MD
118 ctx->slot_size +=
119 lib_ring_buffer_align(*o_begin + ctx->slot_size,
120 ctx->largest_align) + ctx->data_size;
121 if (unlikely((subbuf_offset(*o_begin, chan) + ctx->slot_size)
122 > chan->backend.subbuf_size))
123 return 1;
124
125 /*
126 * Record fits in the current buffer and we are not on a switch
127 * boundary. It's safe to write.
128 */
129 *o_end = *o_begin + ctx->slot_size;
f5ea5800
MD
130
131 if (unlikely((subbuf_offset(*o_end, chan)) == 0))
132 /*
133 * The offset_end will fall at the very beginning of the next
134 * subbuffer.
135 */
136 return 1;
137
f3bc08c5
MD
138 return 0;
139}
140
141/**
142 * lib_ring_buffer_reserve - Reserve space in a ring buffer.
143 * @config: ring buffer instance configuration.
144 * @ctx: ring buffer context. (input and output) Must be already initialized.
145 *
146 * Atomic wait-free slot reservation. The reserved space starts at the context
147 * "pre_offset". Its length is "slot_size". The associated time-stamp is "tsc".
148 *
97ca2c54
MD
149 * Return :
150 * 0 on success.
151 * -EAGAIN if channel is disabled.
152 * -ENOSPC if event size is too large for packet.
153 * -ENOBUFS if there is currently not enough space in buffer for the event.
154 * -EIO if data cannot be written into the buffer for any other reason.
f3bc08c5
MD
155 */
156
157static inline
158int lib_ring_buffer_reserve(const struct lib_ring_buffer_config *config,
cc62f29e
MD
159 struct lib_ring_buffer_ctx *ctx,
160 void *client_ctx)
f3bc08c5
MD
161{
162 struct channel *chan = ctx->chan;
163 struct lib_ring_buffer *buf;
164 unsigned long o_begin, o_end, o_old;
165 size_t before_hdr_pad = 0;
166
e650bcc1 167 if (unlikely(atomic_read(&chan->record_disabled)))
f3bc08c5
MD
168 return -EAGAIN;
169
170 if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
171 buf = per_cpu_ptr(chan->backend.buf, ctx->cpu);
172 else
173 buf = chan->backend.buf;
e650bcc1 174 if (unlikely(atomic_read(&buf->record_disabled)))
f3bc08c5
MD
175 return -EAGAIN;
176 ctx->buf = buf;
177
178 /*
179 * Perform retryable operations.
180 */
cc62f29e 181 if (unlikely(lib_ring_buffer_try_reserve(config, ctx, client_ctx, &o_begin,
f3bc08c5
MD
182 &o_end, &o_old, &before_hdr_pad)))
183 goto slow_path;
184
185 if (unlikely(v_cmpxchg(config, &ctx->buf->offset, o_old, o_end)
186 != o_old))
187 goto slow_path;
188
189 /*
190 * Atomically update last_tsc. This update races against concurrent
191 * atomic updates, but the race will always cause supplementary full TSC
192 * record headers, never the opposite (missing a full TSC record header
193 * when it would be needed).
194 */
195 save_last_tsc(config, ctx->buf, ctx->tsc);
196
197 /*
198 * Push the reader if necessary
199 */
200 lib_ring_buffer_reserve_push_reader(ctx->buf, chan, o_end - 1);
201
202 /*
203 * Clear noref flag for this subbuffer.
204 */
205 lib_ring_buffer_clear_noref(config, &ctx->buf->backend,
206 subbuf_index(o_end - 1, chan));
207
208 ctx->pre_offset = o_begin;
209 ctx->buf_offset = o_begin + before_hdr_pad;
210 return 0;
211slow_path:
cc62f29e 212 return lib_ring_buffer_reserve_slow(ctx, client_ctx);
f3bc08c5
MD
213}
214
215/**
216 * lib_ring_buffer_switch - Perform a sub-buffer switch for a per-cpu buffer.
217 * @config: ring buffer instance configuration.
218 * @buf: buffer
219 * @mode: buffer switch mode (SWITCH_ACTIVE or SWITCH_FLUSH)
220 *
221 * This operation is completely reentrant : can be called while tracing is
222 * active with absolutely no lock held.
223 *
224 * Note, however, that as a v_cmpxchg is used for some atomic operations and
225 * requires to be executed locally for per-CPU buffers, this function must be
226 * called from the CPU which owns the buffer for a ACTIVE flush, with preemption
227 * disabled, for RING_BUFFER_SYNC_PER_CPU configuration.
228 */
229static inline
230void lib_ring_buffer_switch(const struct lib_ring_buffer_config *config,
231 struct lib_ring_buffer *buf, enum switch_mode mode)
232{
233 lib_ring_buffer_switch_slow(buf, mode);
234}
235
236/* See ring_buffer_frontend_api.h for lib_ring_buffer_reserve(). */
237
238/**
239 * lib_ring_buffer_commit - Commit an record.
240 * @config: ring buffer instance configuration.
241 * @ctx: ring buffer context. (input arguments only)
242 *
243 * Atomic unordered slot commit. Increments the commit count in the
244 * specified sub-buffer, and delivers it if necessary.
245 */
246static inline
247void lib_ring_buffer_commit(const struct lib_ring_buffer_config *config,
248 const struct lib_ring_buffer_ctx *ctx)
249{
250 struct channel *chan = ctx->chan;
251 struct lib_ring_buffer *buf = ctx->buf;
252 unsigned long offset_end = ctx->buf_offset;
253 unsigned long endidx = subbuf_index(offset_end - 1, chan);
254 unsigned long commit_count;
8ec496cf 255 struct commit_counters_hot *cc_hot = &buf->commit_hot[endidx];
f3bc08c5
MD
256
257 /*
258 * Must count record before incrementing the commit count.
259 */
260 subbuffer_count_record(config, &buf->backend, endidx);
261
262 /*
263 * Order all writes to buffer before the commit count update that will
264 * determine that the subbuffer is full.
265 */
266 if (config->ipi == RING_BUFFER_IPI_BARRIER) {
267 /*
268 * Must write slot data before incrementing commit count. This
269 * compiler barrier is upgraded into a smp_mb() by the IPI sent
270 * by get_subbuf().
271 */
272 barrier();
273 } else
274 smp_wmb();
275
8ec496cf 276 v_add(config, ctx->slot_size, &cc_hot->cc);
f3bc08c5
MD
277
278 /*
279 * commit count read can race with concurrent OOO commit count updates.
280 * This is only needed for lib_ring_buffer_check_deliver (for
281 * non-polling delivery only) and for
282 * lib_ring_buffer_write_commit_counter. The race can only cause the
283 * counter to be read with the same value more than once, which could
284 * cause :
285 * - Multiple delivery for the same sub-buffer (which is handled
286 * gracefully by the reader code) if the value is for a full
287 * sub-buffer. It's important that we can never miss a sub-buffer
288 * delivery. Re-reading the value after the v_add ensures this.
289 * - Reading a commit_count with a higher value that what was actually
290 * added to it for the lib_ring_buffer_write_commit_counter call
291 * (again caused by a concurrent committer). It does not matter,
292 * because this function is interested in the fact that the commit
293 * count reaches back the reserve offset for a specific sub-buffer,
294 * which is completely independent of the order.
295 */
8ec496cf 296 commit_count = v_read(config, &cc_hot->cc);
f3bc08c5
MD
297
298 lib_ring_buffer_check_deliver(config, buf, chan, offset_end - 1,
635e457c 299 commit_count, endidx, ctx->tsc);
f3bc08c5
MD
300 /*
301 * Update used size at each commit. It's needed only for extracting
302 * ring_buffer buffers from vmcore, after crash.
303 */
8ec496cf
MD
304 lib_ring_buffer_write_commit_counter(config, buf, chan,
305 offset_end, commit_count, cc_hot);
f3bc08c5
MD
306}
307
308/**
309 * lib_ring_buffer_try_discard_reserve - Try discarding a record.
310 * @config: ring buffer instance configuration.
311 * @ctx: ring buffer context. (input arguments only)
312 *
313 * Only succeeds if no other record has been written after the record to
314 * discard. If discard fails, the record must be committed to the buffer.
315 *
316 * Returns 0 upon success, -EPERM if the record cannot be discarded.
317 */
318static inline
319int lib_ring_buffer_try_discard_reserve(const struct lib_ring_buffer_config *config,
320 const struct lib_ring_buffer_ctx *ctx)
321{
322 struct lib_ring_buffer *buf = ctx->buf;
323 unsigned long end_offset = ctx->pre_offset + ctx->slot_size;
324
325 /*
326 * We need to ensure that if the cmpxchg succeeds and discards the
327 * record, the next record will record a full TSC, because it cannot
328 * rely on the last_tsc associated with the discarded record to detect
329 * overflows. The only way to ensure this is to set the last_tsc to 0
330 * (assuming no 64-bit TSC overflow), which forces to write a 64-bit
331 * timestamp in the next record.
332 *
333 * Note: if discard fails, we must leave the TSC in the record header.
334 * It is needed to keep track of TSC overflows for the following
335 * records.
336 */
337 save_last_tsc(config, buf, 0ULL);
338
339 if (likely(v_cmpxchg(config, &buf->offset, end_offset, ctx->pre_offset)
340 != end_offset))
341 return -EPERM;
342 else
343 return 0;
344}
345
346static inline
347void channel_record_disable(const struct lib_ring_buffer_config *config,
348 struct channel *chan)
349{
350 atomic_inc(&chan->record_disabled);
351}
352
353static inline
354void channel_record_enable(const struct lib_ring_buffer_config *config,
355 struct channel *chan)
356{
357 atomic_dec(&chan->record_disabled);
358}
359
360static inline
361void lib_ring_buffer_record_disable(const struct lib_ring_buffer_config *config,
362 struct lib_ring_buffer *buf)
363{
364 atomic_inc(&buf->record_disabled);
365}
366
367static inline
368void lib_ring_buffer_record_enable(const struct lib_ring_buffer_config *config,
369 struct lib_ring_buffer *buf)
370{
371 atomic_dec(&buf->record_disabled);
372}
373
886d51a3 374#endif /* _LIB_RING_BUFFER_FRONTEND_API_H */
This page took 0.046391 seconds and 4 git commands to generate.