Merge tag 'v3.14.25' into backport/v3.14.24-ltsi-rc1+v3.14.25/snapshot-merge.wip
[platform/adaptation/renesas_rcar/renesas_kernel.git] / drivers / staging / lttng / lib / ringbuffer / ring_buffer_frontend.c
1 /*
2  * ring_buffer_frontend.c
3  *
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  *
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
52  */
53
54 #include <linux/delay.h>
55 #include <linux/module.h>
56 #include <linux/percpu.h>
57
58 #include "../../wrapper/ringbuffer/config.h"
59 #include "../../wrapper/ringbuffer/backend.h"
60 #include "../../wrapper/ringbuffer/frontend.h"
61 #include "../../wrapper/ringbuffer/iterator.h"
62 #include "../../wrapper/ringbuffer/nohz.h"
63
64 /*
65  * Internal structure representing offsets to use at a sub-buffer switch.
66  */
67 struct switch_offsets {
68         unsigned long begin, end, old;
69         size_t pre_header_padding, size;
70         unsigned int switch_new_start:1, switch_new_end:1, switch_old_start:1,
71                      switch_old_end:1;
72 };
73
74 #ifdef CONFIG_NO_HZ
75 enum tick_nohz_val {
76         TICK_NOHZ_STOP,
77         TICK_NOHZ_FLUSH,
78         TICK_NOHZ_RESTART,
79 };
80
81 static ATOMIC_NOTIFIER_HEAD(tick_nohz_notifier);
82 #endif /* CONFIG_NO_HZ */
83
84 static DEFINE_PER_CPU(spinlock_t, ring_buffer_nohz_lock);
85
86 DEFINE_PER_CPU(unsigned int, lib_ring_buffer_nesting);
87 EXPORT_PER_CPU_SYMBOL(lib_ring_buffer_nesting);
88
89 static
90 void lib_ring_buffer_print_errors(struct channel *chan,
91                                   struct lib_ring_buffer *buf, int cpu);
92
93 /*
94  * Must be called under cpu hotplug protection.
95  */
96 void lib_ring_buffer_free(struct lib_ring_buffer *buf)
97 {
98         struct channel *chan = buf->backend.chan;
99
100         lib_ring_buffer_print_errors(chan, buf, buf->backend.cpu);
101         kfree(buf->commit_hot);
102         kfree(buf->commit_cold);
103
104         lib_ring_buffer_backend_free(&buf->backend);
105 }
106
107 /**
108  * lib_ring_buffer_reset - Reset ring buffer to initial values.
109  * @buf: Ring buffer.
110  *
111  * Effectively empty the ring buffer. Should be called when the buffer is not
112  * used for writing. The ring buffer can be opened for reading, but the reader
113  * should not be using the iterator concurrently with reset. The previous
114  * current iterator record is reset.
115  */
116 void lib_ring_buffer_reset(struct lib_ring_buffer *buf)
117 {
118         struct channel *chan = buf->backend.chan;
119         const struct lib_ring_buffer_config *config = &chan->backend.config;
120         unsigned int i;
121
122         /*
123          * Reset iterator first. It will put the subbuffer if it currently holds
124          * it.
125          */
126         lib_ring_buffer_iterator_reset(buf);
127         v_set(config, &buf->offset, 0);
128         for (i = 0; i < chan->backend.num_subbuf; i++) {
129                 v_set(config, &buf->commit_hot[i].cc, 0);
130                 v_set(config, &buf->commit_hot[i].seq, 0);
131                 v_set(config, &buf->commit_cold[i].cc_sb, 0);
132         }
133         atomic_long_set(&buf->consumed, 0);
134         atomic_set(&buf->record_disabled, 0);
135         v_set(config, &buf->last_tsc, 0);
136         lib_ring_buffer_backend_reset(&buf->backend);
137         /* Don't reset number of active readers */
138         v_set(config, &buf->records_lost_full, 0);
139         v_set(config, &buf->records_lost_wrap, 0);
140         v_set(config, &buf->records_lost_big, 0);
141         v_set(config, &buf->records_count, 0);
142         v_set(config, &buf->records_overrun, 0);
143         buf->finalized = 0;
144 }
145 EXPORT_SYMBOL_GPL(lib_ring_buffer_reset);
146
147 /**
148  * channel_reset - Reset channel to initial values.
149  * @chan: Channel.
150  *
151  * Effectively empty the channel. Should be called when the channel is not used
152  * for writing. The channel can be opened for reading, but the reader should not
153  * be using the iterator concurrently with reset. The previous current iterator
154  * record is reset.
155  */
156 void channel_reset(struct channel *chan)
157 {
158         /*
159          * Reset iterators first. Will put the subbuffer if held for reading.
160          */
161         channel_iterator_reset(chan);
162         atomic_set(&chan->record_disabled, 0);
163         /* Don't reset commit_count_mask, still valid */
164         channel_backend_reset(&chan->backend);
165         /* Don't reset switch/read timer interval */
166         /* Don't reset notifiers and notifier enable bits */
167         /* Don't reset reader reference count */
168 }
169 EXPORT_SYMBOL_GPL(channel_reset);
170
171 /*
172  * Must be called under cpu hotplug protection.
173  */
174 int lib_ring_buffer_create(struct lib_ring_buffer *buf,
175                            struct channel_backend *chanb, int cpu)
176 {
177         const struct lib_ring_buffer_config *config = &chanb->config;
178         struct channel *chan = container_of(chanb, struct channel, backend);
179         void *priv = chanb->priv;
180         size_t subbuf_header_size;
181         u64 tsc;
182         int ret;
183
184         /* Test for cpu hotplug */
185         if (buf->backend.allocated)
186                 return 0;
187
188         /*
189          * Paranoia: per cpu dynamic allocation is not officially documented as
190          * zeroing the memory, so let's do it here too, just in case.
191          */
192         memset(buf, 0, sizeof(*buf));
193
194         ret = lib_ring_buffer_backend_create(&buf->backend, &chan->backend, cpu);
195         if (ret)
196                 return ret;
197
198         buf->commit_hot =
199                 kzalloc_node(ALIGN(sizeof(*buf->commit_hot)
200                                    * chan->backend.num_subbuf,
201                                    1 << INTERNODE_CACHE_SHIFT),
202                         GFP_KERNEL, cpu_to_node(max(cpu, 0)));
203         if (!buf->commit_hot) {
204                 ret = -ENOMEM;
205                 goto free_chanbuf;
206         }
207
208         buf->commit_cold =
209                 kzalloc_node(ALIGN(sizeof(*buf->commit_cold)
210                                    * chan->backend.num_subbuf,
211                                    1 << INTERNODE_CACHE_SHIFT),
212                         GFP_KERNEL, cpu_to_node(max(cpu, 0)));
213         if (!buf->commit_cold) {
214                 ret = -ENOMEM;
215                 goto free_commit;
216         }
217
218         init_waitqueue_head(&buf->read_wait);
219         init_waitqueue_head(&buf->write_wait);
220         raw_spin_lock_init(&buf->raw_tick_nohz_spinlock);
221
222         /*
223          * Write the subbuffer header for first subbuffer so we know the total
224          * duration of data gathering.
225          */
226         subbuf_header_size = config->cb.subbuffer_header_size();
227         v_set(config, &buf->offset, subbuf_header_size);
228         subbuffer_id_clear_noref(config, &buf->backend.buf_wsb[0].id);
229         tsc = config->cb.ring_buffer_clock_read(buf->backend.chan);
230         config->cb.buffer_begin(buf, tsc, 0);
231         v_add(config, subbuf_header_size, &buf->commit_hot[0].cc);
232
233         if (config->cb.buffer_create) {
234                 ret = config->cb.buffer_create(buf, priv, cpu, chanb->name);
235                 if (ret)
236                         goto free_init;
237         }
238
239         /*
240          * Ensure the buffer is ready before setting it to allocated and setting
241          * the cpumask.
242          * Used for cpu hotplug vs cpumask iteration.
243          */
244         smp_wmb();
245         buf->backend.allocated = 1;
246
247         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
248                 CHAN_WARN_ON(chan, cpumask_test_cpu(cpu,
249                              chan->backend.cpumask));
250                 cpumask_set_cpu(cpu, chan->backend.cpumask);
251         }
252
253         return 0;
254
255         /* Error handling */
256 free_init:
257         kfree(buf->commit_cold);
258 free_commit:
259         kfree(buf->commit_hot);
260 free_chanbuf:
261         lib_ring_buffer_backend_free(&buf->backend);
262         return ret;
263 }
264
265 static void switch_buffer_timer(unsigned long data)
266 {
267         struct lib_ring_buffer *buf = (struct lib_ring_buffer *)data;
268         struct channel *chan = buf->backend.chan;
269         const struct lib_ring_buffer_config *config = &chan->backend.config;
270
271         /*
272          * Only flush buffers periodically if readers are active.
273          */
274         if (atomic_long_read(&buf->active_readers))
275                 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
276
277         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
278                 mod_timer_pinned(&buf->switch_timer,
279                                  jiffies + chan->switch_timer_interval);
280         else
281                 mod_timer(&buf->switch_timer,
282                           jiffies + chan->switch_timer_interval);
283 }
284
285 /*
286  * Called with ring_buffer_nohz_lock held for per-cpu buffers.
287  */
288 static void lib_ring_buffer_start_switch_timer(struct lib_ring_buffer *buf)
289 {
290         struct channel *chan = buf->backend.chan;
291         const struct lib_ring_buffer_config *config = &chan->backend.config;
292
293         if (!chan->switch_timer_interval || buf->switch_timer_enabled)
294                 return;
295         init_timer(&buf->switch_timer);
296         buf->switch_timer.function = switch_buffer_timer;
297         buf->switch_timer.expires = jiffies + chan->switch_timer_interval;
298         buf->switch_timer.data = (unsigned long)buf;
299         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
300                 add_timer_on(&buf->switch_timer, buf->backend.cpu);
301         else
302                 add_timer(&buf->switch_timer);
303         buf->switch_timer_enabled = 1;
304 }
305
306 /*
307  * Called with ring_buffer_nohz_lock held for per-cpu buffers.
308  */
309 static void lib_ring_buffer_stop_switch_timer(struct lib_ring_buffer *buf)
310 {
311         struct channel *chan = buf->backend.chan;
312
313         if (!chan->switch_timer_interval || !buf->switch_timer_enabled)
314                 return;
315
316         del_timer_sync(&buf->switch_timer);
317         buf->switch_timer_enabled = 0;
318 }
319
320 /*
321  * Polling timer to check the channels for data.
322  */
323 static void read_buffer_timer(unsigned long data)
324 {
325         struct lib_ring_buffer *buf = (struct lib_ring_buffer *)data;
326         struct channel *chan = buf->backend.chan;
327         const struct lib_ring_buffer_config *config = &chan->backend.config;
328
329         CHAN_WARN_ON(chan, !buf->backend.allocated);
330
331         if (atomic_long_read(&buf->active_readers)
332             && lib_ring_buffer_poll_deliver(config, buf, chan)) {
333                 wake_up_interruptible(&buf->read_wait);
334                 wake_up_interruptible(&chan->read_wait);
335         }
336
337         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
338                 mod_timer_pinned(&buf->read_timer,
339                                  jiffies + chan->read_timer_interval);
340         else
341                 mod_timer(&buf->read_timer,
342                           jiffies + chan->read_timer_interval);
343 }
344
345 /*
346  * Called with ring_buffer_nohz_lock held for per-cpu buffers.
347  */
348 static void lib_ring_buffer_start_read_timer(struct lib_ring_buffer *buf)
349 {
350         struct channel *chan = buf->backend.chan;
351         const struct lib_ring_buffer_config *config = &chan->backend.config;
352
353         if (config->wakeup != RING_BUFFER_WAKEUP_BY_TIMER
354             || !chan->read_timer_interval
355             || buf->read_timer_enabled)
356                 return;
357
358         init_timer(&buf->read_timer);
359         buf->read_timer.function = read_buffer_timer;
360         buf->read_timer.expires = jiffies + chan->read_timer_interval;
361         buf->read_timer.data = (unsigned long)buf;
362
363         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
364                 add_timer_on(&buf->read_timer, buf->backend.cpu);
365         else
366                 add_timer(&buf->read_timer);
367         buf->read_timer_enabled = 1;
368 }
369
370 /*
371  * Called with ring_buffer_nohz_lock held for per-cpu buffers.
372  */
373 static void lib_ring_buffer_stop_read_timer(struct lib_ring_buffer *buf)
374 {
375         struct channel *chan = buf->backend.chan;
376         const struct lib_ring_buffer_config *config = &chan->backend.config;
377
378         if (config->wakeup != RING_BUFFER_WAKEUP_BY_TIMER
379             || !chan->read_timer_interval
380             || !buf->read_timer_enabled)
381                 return;
382
383         del_timer_sync(&buf->read_timer);
384         /*
385          * do one more check to catch data that has been written in the last
386          * timer period.
387          */
388         if (lib_ring_buffer_poll_deliver(config, buf, chan)) {
389                 wake_up_interruptible(&buf->read_wait);
390                 wake_up_interruptible(&chan->read_wait);
391         }
392         buf->read_timer_enabled = 0;
393 }
394
395 #ifdef CONFIG_HOTPLUG_CPU
396 /**
397  *      lib_ring_buffer_cpu_hp_callback - CPU hotplug callback
398  *      @nb: notifier block
399  *      @action: hotplug action to take
400  *      @hcpu: CPU number
401  *
402  *      Returns the success/failure of the operation. (%NOTIFY_OK, %NOTIFY_BAD)
403  */
404 static
405 int lib_ring_buffer_cpu_hp_callback(struct notifier_block *nb,
406                                               unsigned long action,
407                                               void *hcpu)
408 {
409         unsigned int cpu = (unsigned long)hcpu;
410         struct channel *chan = container_of(nb, struct channel,
411                                             cpu_hp_notifier);
412         struct lib_ring_buffer *buf = per_cpu_ptr(chan->backend.buf, cpu);
413         const struct lib_ring_buffer_config *config = &chan->backend.config;
414
415         if (!chan->cpu_hp_enable)
416                 return NOTIFY_DONE;
417
418         CHAN_WARN_ON(chan, config->alloc == RING_BUFFER_ALLOC_GLOBAL);
419
420         switch (action) {
421         case CPU_DOWN_FAILED:
422         case CPU_DOWN_FAILED_FROZEN:
423         case CPU_ONLINE:
424         case CPU_ONLINE_FROZEN:
425                 wake_up_interruptible(&chan->hp_wait);
426                 lib_ring_buffer_start_switch_timer(buf);
427                 lib_ring_buffer_start_read_timer(buf);
428                 return NOTIFY_OK;
429
430         case CPU_DOWN_PREPARE:
431         case CPU_DOWN_PREPARE_FROZEN:
432                 lib_ring_buffer_stop_switch_timer(buf);
433                 lib_ring_buffer_stop_read_timer(buf);
434                 return NOTIFY_OK;
435
436         case CPU_DEAD:
437         case CPU_DEAD_FROZEN:
438                 /*
439                  * Performing a buffer switch on a remote CPU. Performed by
440                  * the CPU responsible for doing the hotunplug after the target
441                  * CPU stopped running completely. Ensures that all data
442                  * from that remote CPU is flushed.
443                  */
444                 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
445                 return NOTIFY_OK;
446
447         default:
448                 return NOTIFY_DONE;
449         }
450 }
451 #endif
452
453 #if defined(CONFIG_NO_HZ) && defined(CONFIG_LIB_RING_BUFFER)
454 /*
455  * For per-cpu buffers, call the reader wakeups before switching the buffer, so
456  * that wake-up-tracing generated events are flushed before going idle (in
457  * tick_nohz). We test if the spinlock is locked to deal with the race where
458  * readers try to sample the ring buffer before we perform the switch. We let
459  * the readers retry in that case. If there is data in the buffer, the wake up
460  * is going to forbid the CPU running the reader thread from going idle.
461  */
462 static int notrace ring_buffer_tick_nohz_callback(struct notifier_block *nb,
463                                                   unsigned long val,
464                                                   void *data)
465 {
466         struct channel *chan = container_of(nb, struct channel,
467                                             tick_nohz_notifier);
468         const struct lib_ring_buffer_config *config = &chan->backend.config;
469         struct lib_ring_buffer *buf;
470         int cpu = smp_processor_id();
471
472         if (config->alloc != RING_BUFFER_ALLOC_PER_CPU) {
473                 /*
474                  * We don't support keeping the system idle with global buffers
475                  * and streaming active. In order to do so, we would need to
476                  * sample a non-nohz-cpumask racelessly with the nohz updates
477                  * without adding synchronization overhead to nohz. Leave this
478                  * use-case out for now.
479                  */
480                 return 0;
481         }
482
483         buf = channel_get_ring_buffer(config, chan, cpu);
484         switch (val) {
485         case TICK_NOHZ_FLUSH:
486                 raw_spin_lock(&buf->raw_tick_nohz_spinlock);
487                 if (config->wakeup == RING_BUFFER_WAKEUP_BY_TIMER
488                     && chan->read_timer_interval
489                     && atomic_long_read(&buf->active_readers)
490                     && (lib_ring_buffer_poll_deliver(config, buf, chan)
491                         || lib_ring_buffer_pending_data(config, buf, chan))) {
492                         wake_up_interruptible(&buf->read_wait);
493                         wake_up_interruptible(&chan->read_wait);
494                 }
495                 if (chan->switch_timer_interval)
496                         lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
497                 raw_spin_unlock(&buf->raw_tick_nohz_spinlock);
498                 break;
499         case TICK_NOHZ_STOP:
500                 spin_lock(&__get_cpu_var(ring_buffer_nohz_lock));
501                 lib_ring_buffer_stop_switch_timer(buf);
502                 lib_ring_buffer_stop_read_timer(buf);
503                 spin_unlock(&__get_cpu_var(ring_buffer_nohz_lock));
504                 break;
505         case TICK_NOHZ_RESTART:
506                 spin_lock(&__get_cpu_var(ring_buffer_nohz_lock));
507                 lib_ring_buffer_start_read_timer(buf);
508                 lib_ring_buffer_start_switch_timer(buf);
509                 spin_unlock(&__get_cpu_var(ring_buffer_nohz_lock));
510                 break;
511         }
512
513         return 0;
514 }
515
516 void notrace lib_ring_buffer_tick_nohz_flush(void)
517 {
518         atomic_notifier_call_chain(&tick_nohz_notifier, TICK_NOHZ_FLUSH,
519                                    NULL);
520 }
521
522 void notrace lib_ring_buffer_tick_nohz_stop(void)
523 {
524         atomic_notifier_call_chain(&tick_nohz_notifier, TICK_NOHZ_STOP,
525                                    NULL);
526 }
527
528 void notrace lib_ring_buffer_tick_nohz_restart(void)
529 {
530         atomic_notifier_call_chain(&tick_nohz_notifier, TICK_NOHZ_RESTART,
531                                    NULL);
532 }
533 #endif /* defined(CONFIG_NO_HZ) && defined(CONFIG_LIB_RING_BUFFER) */
534
535 /*
536  * Holds CPU hotplug.
537  */
538 static void channel_unregister_notifiers(struct channel *chan)
539 {
540         const struct lib_ring_buffer_config *config = &chan->backend.config;
541         int cpu;
542
543         channel_iterator_unregister_notifiers(chan);
544         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
545 #ifdef CONFIG_NO_HZ
546                 /*
547                  * Remove the nohz notifier first, so we are certain we stop
548                  * the timers.
549                  */
550                 atomic_notifier_chain_unregister(&tick_nohz_notifier,
551                                                  &chan->tick_nohz_notifier);
552                 /*
553                  * ring_buffer_nohz_lock will not be needed below, because
554                  * we just removed the notifiers, which were the only source of
555                  * concurrency.
556                  */
557 #endif /* CONFIG_NO_HZ */
558 #ifdef CONFIG_HOTPLUG_CPU
559                 get_online_cpus();
560                 chan->cpu_hp_enable = 0;
561                 for_each_online_cpu(cpu) {
562                         struct lib_ring_buffer *buf = per_cpu_ptr(chan->backend.buf,
563                                                               cpu);
564                         lib_ring_buffer_stop_switch_timer(buf);
565                         lib_ring_buffer_stop_read_timer(buf);
566                 }
567                 put_online_cpus();
568                 unregister_cpu_notifier(&chan->cpu_hp_notifier);
569 #else
570                 for_each_possible_cpu(cpu) {
571                         struct lib_ring_buffer *buf = per_cpu_ptr(chan->backend.buf,
572                                                               cpu);
573                         lib_ring_buffer_stop_switch_timer(buf);
574                         lib_ring_buffer_stop_read_timer(buf);
575                 }
576 #endif
577         } else {
578                 struct lib_ring_buffer *buf = chan->backend.buf;
579
580                 lib_ring_buffer_stop_switch_timer(buf);
581                 lib_ring_buffer_stop_read_timer(buf);
582         }
583         channel_backend_unregister_notifiers(&chan->backend);
584 }
585
586 static void channel_free(struct channel *chan)
587 {
588         channel_iterator_free(chan);
589         channel_backend_free(&chan->backend);
590         kfree(chan);
591 }
592
593 /**
594  * channel_create - Create channel.
595  * @config: ring buffer instance configuration
596  * @name: name of the channel
597  * @priv: ring buffer client private data
598  * @buf_addr: pointer the the beginning of the preallocated buffer contiguous
599  *            address mapping. It is used only by RING_BUFFER_STATIC
600  *            configuration. It can be set to NULL for other backends.
601  * @subbuf_size: subbuffer size
602  * @num_subbuf: number of subbuffers
603  * @switch_timer_interval: Time interval (in us) to fill sub-buffers with
604  *                         padding to let readers get those sub-buffers.
605  *                         Used for live streaming.
606  * @read_timer_interval: Time interval (in us) to wake up pending readers.
607  *
608  * Holds cpu hotplug.
609  * Returns NULL on failure.
610  */
611 struct channel *channel_create(const struct lib_ring_buffer_config *config,
612                    const char *name, void *priv, void *buf_addr,
613                    size_t subbuf_size,
614                    size_t num_subbuf, unsigned int switch_timer_interval,
615                    unsigned int read_timer_interval)
616 {
617         int ret, cpu;
618         struct channel *chan;
619
620         if (lib_ring_buffer_check_config(config, switch_timer_interval,
621                                          read_timer_interval))
622                 return NULL;
623
624         chan = kzalloc(sizeof(struct channel), GFP_KERNEL);
625         if (!chan)
626                 return NULL;
627
628         ret = channel_backend_init(&chan->backend, name, config, priv,
629                                    subbuf_size, num_subbuf);
630         if (ret)
631                 goto error;
632
633         ret = channel_iterator_init(chan);
634         if (ret)
635                 goto error_free_backend;
636
637         chan->commit_count_mask = (~0UL >> chan->backend.num_subbuf_order);
638         chan->switch_timer_interval = usecs_to_jiffies(switch_timer_interval);
639         chan->read_timer_interval = usecs_to_jiffies(read_timer_interval);
640         kref_init(&chan->ref);
641         init_waitqueue_head(&chan->read_wait);
642         init_waitqueue_head(&chan->hp_wait);
643
644         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
645 #if defined(CONFIG_NO_HZ) && defined(CONFIG_LIB_RING_BUFFER)
646                 /* Only benefit from NO_HZ idle with per-cpu buffers for now. */
647                 chan->tick_nohz_notifier.notifier_call =
648                         ring_buffer_tick_nohz_callback;
649                 chan->tick_nohz_notifier.priority = ~0U;
650                 atomic_notifier_chain_register(&tick_nohz_notifier,
651                                        &chan->tick_nohz_notifier);
652 #endif /* defined(CONFIG_NO_HZ) && defined(CONFIG_LIB_RING_BUFFER) */
653
654                 /*
655                  * In case of non-hotplug cpu, if the ring-buffer is allocated
656                  * in early initcall, it will not be notified of secondary cpus.
657                  * In that off case, we need to allocate for all possible cpus.
658                  */
659 #ifdef CONFIG_HOTPLUG_CPU
660                 chan->cpu_hp_notifier.notifier_call =
661                                 lib_ring_buffer_cpu_hp_callback;
662                 chan->cpu_hp_notifier.priority = 6;
663                 register_cpu_notifier(&chan->cpu_hp_notifier);
664
665                 get_online_cpus();
666                 for_each_online_cpu(cpu) {
667                         struct lib_ring_buffer *buf = per_cpu_ptr(chan->backend.buf,
668                                                                cpu);
669                         spin_lock(&per_cpu(ring_buffer_nohz_lock, cpu));
670                         lib_ring_buffer_start_switch_timer(buf);
671                         lib_ring_buffer_start_read_timer(buf);
672                         spin_unlock(&per_cpu(ring_buffer_nohz_lock, cpu));
673                 }
674                 chan->cpu_hp_enable = 1;
675                 put_online_cpus();
676 #else
677                 for_each_possible_cpu(cpu) {
678                         struct lib_ring_buffer *buf = per_cpu_ptr(chan->backend.buf,
679                                                               cpu);
680                         spin_lock(&per_cpu(ring_buffer_nohz_lock, cpu));
681                         lib_ring_buffer_start_switch_timer(buf);
682                         lib_ring_buffer_start_read_timer(buf);
683                         spin_unlock(&per_cpu(ring_buffer_nohz_lock, cpu));
684                 }
685 #endif
686         } else {
687                 struct lib_ring_buffer *buf = chan->backend.buf;
688
689                 lib_ring_buffer_start_switch_timer(buf);
690                 lib_ring_buffer_start_read_timer(buf);
691         }
692
693         return chan;
694
695 error_free_backend:
696         channel_backend_free(&chan->backend);
697 error:
698         kfree(chan);
699         return NULL;
700 }
701 EXPORT_SYMBOL_GPL(channel_create);
702
703 static
704 void channel_release(struct kref *kref)
705 {
706         struct channel *chan = container_of(kref, struct channel, ref);
707         channel_free(chan);
708 }
709
710 /**
711  * channel_destroy - Finalize, wait for q.s. and destroy channel.
712  * @chan: channel to destroy
713  *
714  * Holds cpu hotplug.
715  * Call "destroy" callback, finalize channels, and then decrement the
716  * channel reference count.  Note that when readers have completed data
717  * consumption of finalized channels, get_subbuf() will return -ENODATA.
718  * They should release their handle at that point.  Returns the private
719  * data pointer.
720  */
721 void *channel_destroy(struct channel *chan)
722 {
723         int cpu;
724         const struct lib_ring_buffer_config *config = &chan->backend.config;
725         void *priv;
726
727         channel_unregister_notifiers(chan);
728
729         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
730                 /*
731                  * No need to hold cpu hotplug, because all notifiers have been
732                  * unregistered.
733                  */
734                 for_each_channel_cpu(cpu, chan) {
735                         struct lib_ring_buffer *buf = per_cpu_ptr(chan->backend.buf,
736                                                               cpu);
737
738                         if (config->cb.buffer_finalize)
739                                 config->cb.buffer_finalize(buf,
740                                                            chan->backend.priv,
741                                                            cpu);
742                         if (buf->backend.allocated)
743                                 lib_ring_buffer_switch_slow(buf, SWITCH_FLUSH);
744                         /*
745                          * Perform flush before writing to finalized.
746                          */
747                         smp_wmb();
748                         ACCESS_ONCE(buf->finalized) = 1;
749                         wake_up_interruptible(&buf->read_wait);
750                 }
751         } else {
752                 struct lib_ring_buffer *buf = chan->backend.buf;
753
754                 if (config->cb.buffer_finalize)
755                         config->cb.buffer_finalize(buf, chan->backend.priv, -1);
756                 if (buf->backend.allocated)
757                         lib_ring_buffer_switch_slow(buf, SWITCH_FLUSH);
758                 /*
759                  * Perform flush before writing to finalized.
760                  */
761                 smp_wmb();
762                 ACCESS_ONCE(buf->finalized) = 1;
763                 wake_up_interruptible(&buf->read_wait);
764         }
765         ACCESS_ONCE(chan->finalized) = 1;
766         wake_up_interruptible(&chan->hp_wait);
767         wake_up_interruptible(&chan->read_wait);
768         priv = chan->backend.priv;
769         kref_put(&chan->ref, channel_release);
770         return priv;
771 }
772 EXPORT_SYMBOL_GPL(channel_destroy);
773
774 struct lib_ring_buffer *channel_get_ring_buffer(
775                                         const struct lib_ring_buffer_config *config,
776                                         struct channel *chan, int cpu)
777 {
778         if (config->alloc == RING_BUFFER_ALLOC_GLOBAL)
779                 return chan->backend.buf;
780         else
781                 return per_cpu_ptr(chan->backend.buf, cpu);
782 }
783 EXPORT_SYMBOL_GPL(channel_get_ring_buffer);
784
785 int lib_ring_buffer_open_read(struct lib_ring_buffer *buf)
786 {
787         struct channel *chan = buf->backend.chan;
788
789         if (!atomic_long_add_unless(&buf->active_readers, 1, 1))
790                 return -EBUSY;
791         kref_get(&chan->ref);
792         smp_mb__after_atomic_inc();
793         return 0;
794 }
795 EXPORT_SYMBOL_GPL(lib_ring_buffer_open_read);
796
797 void lib_ring_buffer_release_read(struct lib_ring_buffer *buf)
798 {
799         struct channel *chan = buf->backend.chan;
800
801         CHAN_WARN_ON(chan, atomic_long_read(&buf->active_readers) != 1);
802         smp_mb__before_atomic_dec();
803         atomic_long_dec(&buf->active_readers);
804         kref_put(&chan->ref, channel_release);
805 }
806 EXPORT_SYMBOL_GPL(lib_ring_buffer_release_read);
807
808 /*
809  * Promote compiler barrier to a smp_mb().
810  * For the specific ring buffer case, this IPI call should be removed if the
811  * architecture does not reorder writes.  This should eventually be provided by
812  * a separate architecture-specific infrastructure.
813  */
814 static void remote_mb(void *info)
815 {
816         smp_mb();
817 }
818
819 /**
820  * lib_ring_buffer_snapshot - save subbuffer position snapshot (for read)
821  * @buf: ring buffer
822  * @consumed: consumed count indicating the position where to read
823  * @produced: produced count, indicates position when to stop reading
824  *
825  * Returns -ENODATA if buffer is finalized, -EAGAIN if there is currently no
826  * data to read at consumed position, or 0 if the get operation succeeds.
827  * Busy-loop trying to get data if the tick_nohz sequence lock is held.
828  */
829
830 int lib_ring_buffer_snapshot(struct lib_ring_buffer *buf,
831                              unsigned long *consumed, unsigned long *produced)
832 {
833         struct channel *chan = buf->backend.chan;
834         const struct lib_ring_buffer_config *config = &chan->backend.config;
835         unsigned long consumed_cur, write_offset;
836         int finalized;
837
838 retry:
839         finalized = ACCESS_ONCE(buf->finalized);
840         /*
841          * Read finalized before counters.
842          */
843         smp_rmb();
844         consumed_cur = atomic_long_read(&buf->consumed);
845         /*
846          * No need to issue a memory barrier between consumed count read and
847          * write offset read, because consumed count can only change
848          * concurrently in overwrite mode, and we keep a sequence counter
849          * identifier derived from the write offset to check we are getting
850          * the same sub-buffer we are expecting (the sub-buffers are atomically
851          * "tagged" upon writes, tags are checked upon read).
852          */
853         write_offset = v_read(config, &buf->offset);
854
855         /*
856          * Check that we are not about to read the same subbuffer in
857          * which the writer head is.
858          */
859         if (subbuf_trunc(write_offset, chan) - subbuf_trunc(consumed_cur, chan)
860             == 0)
861                 goto nodata;
862
863         *consumed = consumed_cur;
864         *produced = subbuf_trunc(write_offset, chan);
865
866         return 0;
867
868 nodata:
869         /*
870          * The memory barriers __wait_event()/wake_up_interruptible() take care
871          * of "raw_spin_is_locked" memory ordering.
872          */
873         if (finalized)
874                 return -ENODATA;
875         else if (raw_spin_is_locked(&buf->raw_tick_nohz_spinlock))
876                 goto retry;
877         else
878                 return -EAGAIN;
879 }
880 EXPORT_SYMBOL_GPL(lib_ring_buffer_snapshot);
881
882 /**
883  * lib_ring_buffer_put_snapshot - move consumed counter forward
884  *
885  * Should only be called from consumer context.
886  * @buf: ring buffer
887  * @consumed_new: new consumed count value
888  */
889 void lib_ring_buffer_move_consumer(struct lib_ring_buffer *buf,
890                                    unsigned long consumed_new)
891 {
892         struct lib_ring_buffer_backend *bufb = &buf->backend;
893         struct channel *chan = bufb->chan;
894         unsigned long consumed;
895
896         CHAN_WARN_ON(chan, atomic_long_read(&buf->active_readers) != 1);
897
898         /*
899          * Only push the consumed value forward.
900          * If the consumed cmpxchg fails, this is because we have been pushed by
901          * the writer in flight recorder mode.
902          */
903         consumed = atomic_long_read(&buf->consumed);
904         while ((long) consumed - (long) consumed_new < 0)
905                 consumed = atomic_long_cmpxchg(&buf->consumed, consumed,
906                                                consumed_new);
907         /* Wake-up the metadata producer */
908         wake_up_interruptible(&buf->write_wait);
909 }
910 EXPORT_SYMBOL_GPL(lib_ring_buffer_move_consumer);
911
912 /**
913  * lib_ring_buffer_get_subbuf - get exclusive access to subbuffer for reading
914  * @buf: ring buffer
915  * @consumed: consumed count indicating the position where to read
916  *
917  * Returns -ENODATA if buffer is finalized, -EAGAIN if there is currently no
918  * data to read at consumed position, or 0 if the get operation succeeds.
919  * Busy-loop trying to get data if the tick_nohz sequence lock is held.
920  */
921 int lib_ring_buffer_get_subbuf(struct lib_ring_buffer *buf,
922                                unsigned long consumed)
923 {
924         struct channel *chan = buf->backend.chan;
925         const struct lib_ring_buffer_config *config = &chan->backend.config;
926         unsigned long consumed_cur, consumed_idx, commit_count, write_offset;
927         int ret;
928         int finalized;
929
930         if (buf->get_subbuf) {
931                 /*
932                  * Reader is trying to get a subbuffer twice.
933                  */
934                 CHAN_WARN_ON(chan, 1);
935                 return -EBUSY;
936         }
937 retry:
938         finalized = ACCESS_ONCE(buf->finalized);
939         /*
940          * Read finalized before counters.
941          */
942         smp_rmb();
943         consumed_cur = atomic_long_read(&buf->consumed);
944         consumed_idx = subbuf_index(consumed, chan);
945         commit_count = v_read(config, &buf->commit_cold[consumed_idx].cc_sb);
946         /*
947          * Make sure we read the commit count before reading the buffer
948          * data and the write offset. Correct consumed offset ordering
949          * wrt commit count is insured by the use of cmpxchg to update
950          * the consumed offset.
951          * smp_call_function_single can fail if the remote CPU is offline,
952          * this is OK because then there is no wmb to execute there.
953          * If our thread is executing on the same CPU as the on the buffers
954          * belongs to, we don't have to synchronize it at all. If we are
955          * migrated, the scheduler will take care of the memory barriers.
956          * Normally, smp_call_function_single() should ensure program order when
957          * executing the remote function, which implies that it surrounds the
958          * function execution with :
959          * smp_mb()
960          * send IPI
961          * csd_lock_wait
962          *                recv IPI
963          *                smp_mb()
964          *                exec. function
965          *                smp_mb()
966          *                csd unlock
967          * smp_mb()
968          *
969          * However, smp_call_function_single() does not seem to clearly execute
970          * such barriers. It depends on spinlock semantic to provide the barrier
971          * before executing the IPI and, when busy-looping, csd_lock_wait only
972          * executes smp_mb() when it has to wait for the other CPU.
973          *
974          * I don't trust this code. Therefore, let's add the smp_mb() sequence
975          * required ourself, even if duplicated. It has no performance impact
976          * anyway.
977          *
978          * smp_mb() is needed because smp_rmb() and smp_wmb() only order read vs
979          * read and write vs write. They do not ensure core synchronization. We
980          * really have to ensure total order between the 3 barriers running on
981          * the 2 CPUs.
982          */
983         if (config->ipi == RING_BUFFER_IPI_BARRIER) {
984                 if (config->sync == RING_BUFFER_SYNC_PER_CPU
985                     && config->alloc == RING_BUFFER_ALLOC_PER_CPU) {
986                         if (raw_smp_processor_id() != buf->backend.cpu) {
987                                 /* Total order with IPI handler smp_mb() */
988                                 smp_mb();
989                                 smp_call_function_single(buf->backend.cpu,
990                                                          remote_mb, NULL, 1);
991                                 /* Total order with IPI handler smp_mb() */
992                                 smp_mb();
993                         }
994                 } else {
995                         /* Total order with IPI handler smp_mb() */
996                         smp_mb();
997                         smp_call_function(remote_mb, NULL, 1);
998                         /* Total order with IPI handler smp_mb() */
999                         smp_mb();
1000                 }
1001         } else {
1002                 /*
1003                  * Local rmb to match the remote wmb to read the commit count
1004                  * before the buffer data and the write offset.
1005                  */
1006                 smp_rmb();
1007         }
1008
1009         write_offset = v_read(config, &buf->offset);
1010
1011         /*
1012          * Check that the buffer we are getting is after or at consumed_cur
1013          * position.
1014          */
1015         if ((long) subbuf_trunc(consumed, chan)
1016             - (long) subbuf_trunc(consumed_cur, chan) < 0)
1017                 goto nodata;
1018
1019         /*
1020          * Check that the subbuffer we are trying to consume has been
1021          * already fully committed.
1022          */
1023         if (((commit_count - chan->backend.subbuf_size)
1024              & chan->commit_count_mask)
1025             - (buf_trunc(consumed, chan)
1026                >> chan->backend.num_subbuf_order)
1027             != 0)
1028                 goto nodata;
1029
1030         /*
1031          * Check that we are not about to read the same subbuffer in
1032          * which the writer head is.
1033          */
1034         if (subbuf_trunc(write_offset, chan) - subbuf_trunc(consumed, chan)
1035             == 0)
1036                 goto nodata;
1037
1038         /*
1039          * Failure to get the subbuffer causes a busy-loop retry without going
1040          * to a wait queue. These are caused by short-lived race windows where
1041          * the writer is getting access to a subbuffer we were trying to get
1042          * access to. Also checks that the "consumed" buffer count we are
1043          * looking for matches the one contained in the subbuffer id.
1044          */
1045         ret = update_read_sb_index(config, &buf->backend, &chan->backend,
1046                                    consumed_idx, buf_trunc_val(consumed, chan));
1047         if (ret)
1048                 goto retry;
1049         subbuffer_id_clear_noref(config, &buf->backend.buf_rsb.id);
1050
1051         buf->get_subbuf_consumed = consumed;
1052         buf->get_subbuf = 1;
1053
1054         return 0;
1055
1056 nodata:
1057         /*
1058          * The memory barriers __wait_event()/wake_up_interruptible() take care
1059          * of "raw_spin_is_locked" memory ordering.
1060          */
1061         if (finalized)
1062                 return -ENODATA;
1063         else if (raw_spin_is_locked(&buf->raw_tick_nohz_spinlock))
1064                 goto retry;
1065         else
1066                 return -EAGAIN;
1067 }
1068 EXPORT_SYMBOL_GPL(lib_ring_buffer_get_subbuf);
1069
1070 /**
1071  * lib_ring_buffer_put_subbuf - release exclusive subbuffer access
1072  * @buf: ring buffer
1073  */
1074 void lib_ring_buffer_put_subbuf(struct lib_ring_buffer *buf)
1075 {
1076         struct lib_ring_buffer_backend *bufb = &buf->backend;
1077         struct channel *chan = bufb->chan;
1078         const struct lib_ring_buffer_config *config = &chan->backend.config;
1079         unsigned long read_sb_bindex, consumed_idx, consumed;
1080
1081         CHAN_WARN_ON(chan, atomic_long_read(&buf->active_readers) != 1);
1082
1083         if (!buf->get_subbuf) {
1084                 /*
1085                  * Reader puts a subbuffer it did not get.
1086                  */
1087                 CHAN_WARN_ON(chan, 1);
1088                 return;
1089         }
1090         consumed = buf->get_subbuf_consumed;
1091         buf->get_subbuf = 0;
1092
1093         /*
1094          * Clear the records_unread counter. (overruns counter)
1095          * Can still be non-zero if a file reader simply grabbed the data
1096          * without using iterators.
1097          * Can be below zero if an iterator is used on a snapshot more than
1098          * once.
1099          */
1100         read_sb_bindex = subbuffer_id_get_index(config, bufb->buf_rsb.id);
1101         v_add(config, v_read(config,
1102                              &bufb->array[read_sb_bindex]->records_unread),
1103               &bufb->records_read);
1104         v_set(config, &bufb->array[read_sb_bindex]->records_unread, 0);
1105         CHAN_WARN_ON(chan, config->mode == RING_BUFFER_OVERWRITE
1106                      && subbuffer_id_is_noref(config, bufb->buf_rsb.id));
1107         subbuffer_id_set_noref(config, &bufb->buf_rsb.id);
1108
1109         /*
1110          * Exchange the reader subbuffer with the one we put in its place in the
1111          * writer subbuffer table. Expect the original consumed count. If
1112          * update_read_sb_index fails, this is because the writer updated the
1113          * subbuffer concurrently. We should therefore keep the subbuffer we
1114          * currently have: it has become invalid to try reading this sub-buffer
1115          * consumed count value anyway.
1116          */
1117         consumed_idx = subbuf_index(consumed, chan);
1118         update_read_sb_index(config, &buf->backend, &chan->backend,
1119                              consumed_idx, buf_trunc_val(consumed, chan));
1120         /*
1121          * update_read_sb_index return value ignored. Don't exchange sub-buffer
1122          * if the writer concurrently updated it.
1123          */
1124 }
1125 EXPORT_SYMBOL_GPL(lib_ring_buffer_put_subbuf);
1126
1127 /*
1128  * cons_offset is an iterator on all subbuffer offsets between the reader
1129  * position and the writer position. (inclusive)
1130  */
1131 static
1132 void lib_ring_buffer_print_subbuffer_errors(struct lib_ring_buffer *buf,
1133                                             struct channel *chan,
1134                                             unsigned long cons_offset,
1135                                             int cpu)
1136 {
1137         const struct lib_ring_buffer_config *config = &chan->backend.config;
1138         unsigned long cons_idx, commit_count, commit_count_sb;
1139
1140         cons_idx = subbuf_index(cons_offset, chan);
1141         commit_count = v_read(config, &buf->commit_hot[cons_idx].cc);
1142         commit_count_sb = v_read(config, &buf->commit_cold[cons_idx].cc_sb);
1143
1144         if (subbuf_offset(commit_count, chan) != 0)
1145                 printk(KERN_WARNING
1146                        "ring buffer %s, cpu %d: "
1147                        "commit count in subbuffer %lu,\n"
1148                        "expecting multiples of %lu bytes\n"
1149                        "  [ %lu bytes committed, %lu bytes reader-visible ]\n",
1150                        chan->backend.name, cpu, cons_idx,
1151                        chan->backend.subbuf_size,
1152                        commit_count, commit_count_sb);
1153
1154         printk(KERN_DEBUG "ring buffer: %s, cpu %d: %lu bytes committed\n",
1155                chan->backend.name, cpu, commit_count);
1156 }
1157
1158 static
1159 void lib_ring_buffer_print_buffer_errors(struct lib_ring_buffer *buf,
1160                                          struct channel *chan,
1161                                          void *priv, int cpu)
1162 {
1163         const struct lib_ring_buffer_config *config = &chan->backend.config;
1164         unsigned long write_offset, cons_offset;
1165
1166         /*
1167          * No need to order commit_count, write_offset and cons_offset reads
1168          * because we execute at teardown when no more writer nor reader
1169          * references are left.
1170          */
1171         write_offset = v_read(config, &buf->offset);
1172         cons_offset = atomic_long_read(&buf->consumed);
1173         if (write_offset != cons_offset)
1174                 printk(KERN_DEBUG
1175                        "ring buffer %s, cpu %d: "
1176                        "non-consumed data\n"
1177                        "  [ %lu bytes written, %lu bytes read ]\n",
1178                        chan->backend.name, cpu, write_offset, cons_offset);
1179
1180         for (cons_offset = atomic_long_read(&buf->consumed);
1181              (long) (subbuf_trunc((unsigned long) v_read(config, &buf->offset),
1182                                   chan)
1183                      - cons_offset) > 0;
1184              cons_offset = subbuf_align(cons_offset, chan))
1185                 lib_ring_buffer_print_subbuffer_errors(buf, chan, cons_offset,
1186                                                        cpu);
1187 }
1188
1189 static
1190 void lib_ring_buffer_print_errors(struct channel *chan,
1191                                   struct lib_ring_buffer *buf, int cpu)
1192 {
1193         const struct lib_ring_buffer_config *config = &chan->backend.config;
1194         void *priv = chan->backend.priv;
1195
1196         if (!strcmp(chan->backend.name, "relay-metadata")) {
1197                 printk(KERN_DEBUG "ring buffer %s: %lu records written, "
1198                         "%lu records overrun\n",
1199                         chan->backend.name,
1200                         v_read(config, &buf->records_count),
1201                         v_read(config, &buf->records_overrun));
1202         } else {
1203                 printk(KERN_DEBUG "ring buffer %s, cpu %d: %lu records written, "
1204                         "%lu records overrun\n",
1205                         chan->backend.name, cpu,
1206                         v_read(config, &buf->records_count),
1207                         v_read(config, &buf->records_overrun));
1208
1209                 if (v_read(config, &buf->records_lost_full)
1210                     || v_read(config, &buf->records_lost_wrap)
1211                     || v_read(config, &buf->records_lost_big))
1212                         printk(KERN_WARNING
1213                                 "ring buffer %s, cpu %d: records were lost. Caused by:\n"
1214                                 "  [ %lu buffer full, %lu nest buffer wrap-around, "
1215                                 "%lu event too big ]\n",
1216                                 chan->backend.name, cpu,
1217                                 v_read(config, &buf->records_lost_full),
1218                                 v_read(config, &buf->records_lost_wrap),
1219                                 v_read(config, &buf->records_lost_big));
1220         }
1221         lib_ring_buffer_print_buffer_errors(buf, chan, priv, cpu);
1222 }
1223
1224 /*
1225  * lib_ring_buffer_switch_old_start: Populate old subbuffer header.
1226  *
1227  * Only executed when the buffer is finalized, in SWITCH_FLUSH.
1228  */
1229 static
1230 void lib_ring_buffer_switch_old_start(struct lib_ring_buffer *buf,
1231                                       struct channel *chan,
1232                                       struct switch_offsets *offsets,
1233                                       u64 tsc)
1234 {
1235         const struct lib_ring_buffer_config *config = &chan->backend.config;
1236         unsigned long oldidx = subbuf_index(offsets->old, chan);
1237         unsigned long commit_count;
1238
1239         config->cb.buffer_begin(buf, tsc, oldidx);
1240
1241         /*
1242          * Order all writes to buffer before the commit count update that will
1243          * determine that the subbuffer is full.
1244          */
1245         if (config->ipi == RING_BUFFER_IPI_BARRIER) {
1246                 /*
1247                  * Must write slot data before incrementing commit count.  This
1248                  * compiler barrier is upgraded into a smp_mb() by the IPI sent
1249                  * by get_subbuf().
1250                  */
1251                 barrier();
1252         } else
1253                 smp_wmb();
1254         v_add(config, config->cb.subbuffer_header_size(),
1255               &buf->commit_hot[oldidx].cc);
1256         commit_count = v_read(config, &buf->commit_hot[oldidx].cc);
1257         /* Check if the written buffer has to be delivered */
1258         lib_ring_buffer_check_deliver(config, buf, chan, offsets->old,
1259                                       commit_count, oldidx, tsc);
1260         lib_ring_buffer_write_commit_counter(config, buf, chan, oldidx,
1261                                              offsets->old, commit_count,
1262                                              config->cb.subbuffer_header_size());
1263 }
1264
1265 /*
1266  * lib_ring_buffer_switch_old_end: switch old subbuffer
1267  *
1268  * Note : offset_old should never be 0 here. It is ok, because we never perform
1269  * buffer switch on an empty subbuffer in SWITCH_ACTIVE mode. The caller
1270  * increments the offset_old value when doing a SWITCH_FLUSH on an empty
1271  * subbuffer.
1272  */
1273 static
1274 void lib_ring_buffer_switch_old_end(struct lib_ring_buffer *buf,
1275                                     struct channel *chan,
1276                                     struct switch_offsets *offsets,
1277                                     u64 tsc)
1278 {
1279         const struct lib_ring_buffer_config *config = &chan->backend.config;
1280         unsigned long oldidx = subbuf_index(offsets->old - 1, chan);
1281         unsigned long commit_count, padding_size, data_size;
1282
1283         data_size = subbuf_offset(offsets->old - 1, chan) + 1;
1284         padding_size = chan->backend.subbuf_size - data_size;
1285         subbuffer_set_data_size(config, &buf->backend, oldidx, data_size);
1286
1287         /*
1288          * Order all writes to buffer before the commit count update that will
1289          * determine that the subbuffer is full.
1290          */
1291         if (config->ipi == RING_BUFFER_IPI_BARRIER) {
1292                 /*
1293                  * Must write slot data before incrementing commit count.  This
1294                  * compiler barrier is upgraded into a smp_mb() by the IPI sent
1295                  * by get_subbuf().
1296                  */
1297                 barrier();
1298         } else
1299                 smp_wmb();
1300         v_add(config, padding_size, &buf->commit_hot[oldidx].cc);
1301         commit_count = v_read(config, &buf->commit_hot[oldidx].cc);
1302         lib_ring_buffer_check_deliver(config, buf, chan, offsets->old - 1,
1303                                       commit_count, oldidx, tsc);
1304         lib_ring_buffer_write_commit_counter(config, buf, chan, oldidx,
1305                                              offsets->old, commit_count,
1306                                              padding_size);
1307 }
1308
1309 /*
1310  * lib_ring_buffer_switch_new_start: Populate new subbuffer.
1311  *
1312  * This code can be executed unordered : writers may already have written to the
1313  * sub-buffer before this code gets executed, caution.  The commit makes sure
1314  * that this code is executed before the deliver of this sub-buffer.
1315  */
1316 static
1317 void lib_ring_buffer_switch_new_start(struct lib_ring_buffer *buf,
1318                                       struct channel *chan,
1319                                       struct switch_offsets *offsets,
1320                                       u64 tsc)
1321 {
1322         const struct lib_ring_buffer_config *config = &chan->backend.config;
1323         unsigned long beginidx = subbuf_index(offsets->begin, chan);
1324         unsigned long commit_count;
1325
1326         config->cb.buffer_begin(buf, tsc, beginidx);
1327
1328         /*
1329          * Order all writes to buffer before the commit count update that will
1330          * determine that the subbuffer is full.
1331          */
1332         if (config->ipi == RING_BUFFER_IPI_BARRIER) {
1333                 /*
1334                  * Must write slot data before incrementing commit count.  This
1335                  * compiler barrier is upgraded into a smp_mb() by the IPI sent
1336                  * by get_subbuf().
1337                  */
1338                 barrier();
1339         } else
1340                 smp_wmb();
1341         v_add(config, config->cb.subbuffer_header_size(),
1342               &buf->commit_hot[beginidx].cc);
1343         commit_count = v_read(config, &buf->commit_hot[beginidx].cc);
1344         /* Check if the written buffer has to be delivered */
1345         lib_ring_buffer_check_deliver(config, buf, chan, offsets->begin,
1346                                       commit_count, beginidx, tsc);
1347         lib_ring_buffer_write_commit_counter(config, buf, chan, beginidx,
1348                                              offsets->begin, commit_count,
1349                                              config->cb.subbuffer_header_size());
1350 }
1351
1352 /*
1353  * lib_ring_buffer_switch_new_end: finish switching current subbuffer
1354  *
1355  * Calls subbuffer_set_data_size() to set the data size of the current
1356  * sub-buffer. We do not need to perform check_deliver nor commit here,
1357  * since this task will be done by the "commit" of the event for which
1358  * we are currently doing the space reservation.
1359  */
1360 static
1361 void lib_ring_buffer_switch_new_end(struct lib_ring_buffer *buf,
1362                                             struct channel *chan,
1363                                             struct switch_offsets *offsets,
1364                                             u64 tsc)
1365 {
1366         const struct lib_ring_buffer_config *config = &chan->backend.config;
1367         unsigned long endidx, data_size;
1368
1369         endidx = subbuf_index(offsets->end - 1, chan);
1370         data_size = subbuf_offset(offsets->end - 1, chan) + 1;
1371         subbuffer_set_data_size(config, &buf->backend, endidx, data_size);
1372 }
1373
1374 /*
1375  * Returns :
1376  * 0 if ok
1377  * !0 if execution must be aborted.
1378  */
1379 static
1380 int lib_ring_buffer_try_switch_slow(enum switch_mode mode,
1381                                     struct lib_ring_buffer *buf,
1382                                     struct channel *chan,
1383                                     struct switch_offsets *offsets,
1384                                     u64 *tsc)
1385 {
1386         const struct lib_ring_buffer_config *config = &chan->backend.config;
1387         unsigned long off, reserve_commit_diff;
1388
1389         offsets->begin = v_read(config, &buf->offset);
1390         offsets->old = offsets->begin;
1391         offsets->switch_old_start = 0;
1392         off = subbuf_offset(offsets->begin, chan);
1393
1394         *tsc = config->cb.ring_buffer_clock_read(chan);
1395
1396         /*
1397          * Ensure we flush the header of an empty subbuffer when doing the
1398          * finalize (SWITCH_FLUSH). This ensures that we end up knowing the
1399          * total data gathering duration even if there were no records saved
1400          * after the last buffer switch.
1401          * In SWITCH_ACTIVE mode, switch the buffer when it contains events.
1402          * SWITCH_ACTIVE only flushes the current subbuffer, dealing with end of
1403          * subbuffer header as appropriate.
1404          * The next record that reserves space will be responsible for
1405          * populating the following subbuffer header. We choose not to populate
1406          * the next subbuffer header here because we want to be able to use
1407          * SWITCH_ACTIVE for periodical buffer flush and CPU tick_nohz stop
1408          * buffer flush, which must guarantee that all the buffer content
1409          * (records and header timestamps) are visible to the reader. This is
1410          * required for quiescence guarantees for the fusion merge.
1411          */
1412         if (mode != SWITCH_FLUSH && !off)
1413                 return -1;      /* we do not have to switch : buffer is empty */
1414
1415         if (unlikely(off == 0)) {
1416                 unsigned long sb_index, commit_count;
1417
1418                 /*
1419                  * We are performing a SWITCH_FLUSH. At this stage, there are no
1420                  * concurrent writes into the buffer.
1421                  *
1422                  * The client does not save any header information.  Don't
1423                  * switch empty subbuffer on finalize, because it is invalid to
1424                  * deliver a completely empty subbuffer.
1425                  */
1426                 if (!config->cb.subbuffer_header_size())
1427                         return -1;
1428
1429                 /* Test new buffer integrity */
1430                 sb_index = subbuf_index(offsets->begin, chan);
1431                 commit_count = v_read(config,
1432                                 &buf->commit_cold[sb_index].cc_sb);
1433                 reserve_commit_diff =
1434                   (buf_trunc(offsets->begin, chan)
1435                    >> chan->backend.num_subbuf_order)
1436                   - (commit_count & chan->commit_count_mask);
1437                 if (likely(reserve_commit_diff == 0)) {
1438                         /* Next subbuffer not being written to. */
1439                         if (unlikely(config->mode != RING_BUFFER_OVERWRITE &&
1440                                 subbuf_trunc(offsets->begin, chan)
1441                                  - subbuf_trunc((unsigned long)
1442                                      atomic_long_read(&buf->consumed), chan)
1443                                 >= chan->backend.buf_size)) {
1444                                 /*
1445                                  * We do not overwrite non consumed buffers
1446                                  * and we are full : don't switch.
1447                                  */
1448                                 return -1;
1449                         } else {
1450                                 /*
1451                                  * Next subbuffer not being written to, and we
1452                                  * are either in overwrite mode or the buffer is
1453                                  * not full. It's safe to write in this new
1454                                  * subbuffer.
1455                                  */
1456                         }
1457                 } else {
1458                         /*
1459                          * Next subbuffer reserve offset does not match the
1460                          * commit offset. Don't perform switch in
1461                          * producer-consumer and overwrite mode.  Caused by
1462                          * either a writer OOPS or too many nested writes over a
1463                          * reserve/commit pair.
1464                          */
1465                         return -1;
1466                 }
1467
1468                 /*
1469                  * Need to write the subbuffer start header on finalize.
1470                  */
1471                 offsets->switch_old_start = 1;
1472         }
1473         offsets->begin = subbuf_align(offsets->begin, chan);
1474         /* Note: old points to the next subbuf at offset 0 */
1475         offsets->end = offsets->begin;
1476         return 0;
1477 }
1478
1479 /*
1480  * Force a sub-buffer switch. This operation is completely reentrant : can be
1481  * called while tracing is active with absolutely no lock held.
1482  *
1483  * Note, however, that as a v_cmpxchg is used for some atomic
1484  * operations, this function must be called from the CPU which owns the buffer
1485  * for a ACTIVE flush.
1486  */
1487 void lib_ring_buffer_switch_slow(struct lib_ring_buffer *buf, enum switch_mode mode)
1488 {
1489         struct channel *chan = buf->backend.chan;
1490         const struct lib_ring_buffer_config *config = &chan->backend.config;
1491         struct switch_offsets offsets;
1492         unsigned long oldidx;
1493         u64 tsc;
1494
1495         offsets.size = 0;
1496
1497         /*
1498          * Perform retryable operations.
1499          */
1500         do {
1501                 if (lib_ring_buffer_try_switch_slow(mode, buf, chan, &offsets,
1502                                                     &tsc))
1503                         return; /* Switch not needed */
1504         } while (v_cmpxchg(config, &buf->offset, offsets.old, offsets.end)
1505                  != offsets.old);
1506
1507         /*
1508          * Atomically update last_tsc. This update races against concurrent
1509          * atomic updates, but the race will always cause supplementary full TSC
1510          * records, never the opposite (missing a full TSC record when it would
1511          * be needed).
1512          */
1513         save_last_tsc(config, buf, tsc);
1514
1515         /*
1516          * Push the reader if necessary
1517          */
1518         lib_ring_buffer_reserve_push_reader(buf, chan, offsets.old);
1519
1520         oldidx = subbuf_index(offsets.old, chan);
1521         lib_ring_buffer_clear_noref(config, &buf->backend, oldidx);
1522
1523         /*
1524          * May need to populate header start on SWITCH_FLUSH.
1525          */
1526         if (offsets.switch_old_start) {
1527                 lib_ring_buffer_switch_old_start(buf, chan, &offsets, tsc);
1528                 offsets.old += config->cb.subbuffer_header_size();
1529         }
1530
1531         /*
1532          * Switch old subbuffer.
1533          */
1534         lib_ring_buffer_switch_old_end(buf, chan, &offsets, tsc);
1535 }
1536 EXPORT_SYMBOL_GPL(lib_ring_buffer_switch_slow);
1537
1538 static void remote_switch(void *info)
1539 {
1540         struct lib_ring_buffer *buf = info;
1541
1542         lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
1543 }
1544
1545 void lib_ring_buffer_switch_remote(struct lib_ring_buffer *buf)
1546 {
1547         struct channel *chan = buf->backend.chan;
1548         const struct lib_ring_buffer_config *config = &chan->backend.config;
1549         int ret;
1550
1551         /*
1552          * With global synchronization we don't need to use the IPI scheme.
1553          */
1554         if (config->sync == RING_BUFFER_SYNC_GLOBAL) {
1555                 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
1556                 return;
1557         }
1558
1559         /*
1560          * Taking lock on CPU hotplug to ensure two things: first, that the
1561          * target cpu is not taken concurrently offline while we are within
1562          * smp_call_function_single() (I don't trust that get_cpu() on the
1563          * _local_ CPU actually inhibit CPU hotplug for the _remote_ CPU (to be
1564          * confirmed)). Secondly, if it happens that the CPU is not online, our
1565          * own call to lib_ring_buffer_switch_slow() needs to be protected from
1566          * CPU hotplug handlers, which can also perform a remote subbuffer
1567          * switch.
1568          */
1569         get_online_cpus();
1570         ret = smp_call_function_single(buf->backend.cpu,
1571                                  remote_switch, buf, 1);
1572         if (ret) {
1573                 /* Remote CPU is offline, do it ourself. */
1574                 lib_ring_buffer_switch_slow(buf, SWITCH_ACTIVE);
1575         }
1576         put_online_cpus();
1577 }
1578 EXPORT_SYMBOL_GPL(lib_ring_buffer_switch_remote);
1579
1580 /*
1581  * Returns :
1582  * 0 if ok
1583  * -ENOSPC if event size is too large for packet.
1584  * -ENOBUFS if there is currently not enough space in buffer for the event.
1585  * -EIO if data cannot be written into the buffer for any other reason.
1586  */
1587 static
1588 int lib_ring_buffer_try_reserve_slow(struct lib_ring_buffer *buf,
1589                                      struct channel *chan,
1590                                      struct switch_offsets *offsets,
1591                                      struct lib_ring_buffer_ctx *ctx)
1592 {
1593         const struct lib_ring_buffer_config *config = &chan->backend.config;
1594         unsigned long reserve_commit_diff, offset_cmp;
1595
1596 retry:
1597         offsets->begin = offset_cmp = v_read(config, &buf->offset);
1598         offsets->old = offsets->begin;
1599         offsets->switch_new_start = 0;
1600         offsets->switch_new_end = 0;
1601         offsets->switch_old_end = 0;
1602         offsets->pre_header_padding = 0;
1603
1604         ctx->tsc = config->cb.ring_buffer_clock_read(chan);
1605         if ((int64_t) ctx->tsc == -EIO)
1606                 return -EIO;
1607
1608         if (last_tsc_overflow(config, buf, ctx->tsc))
1609                 ctx->rflags |= RING_BUFFER_RFLAG_FULL_TSC;
1610
1611         if (unlikely(subbuf_offset(offsets->begin, ctx->chan) == 0)) {
1612                 offsets->switch_new_start = 1;          /* For offsets->begin */
1613         } else {
1614                 offsets->size = config->cb.record_header_size(config, chan,
1615                                                 offsets->begin,
1616                                                 &offsets->pre_header_padding,
1617                                                 ctx);
1618                 offsets->size +=
1619                         lib_ring_buffer_align(offsets->begin + offsets->size,
1620                                               ctx->largest_align)
1621                         + ctx->data_size;
1622                 if (unlikely(subbuf_offset(offsets->begin, chan) +
1623                              offsets->size > chan->backend.subbuf_size)) {
1624                         offsets->switch_old_end = 1;    /* For offsets->old */
1625                         offsets->switch_new_start = 1;  /* For offsets->begin */
1626                 }
1627         }
1628         if (unlikely(offsets->switch_new_start)) {
1629                 unsigned long sb_index, commit_count;
1630
1631                 /*
1632                  * We are typically not filling the previous buffer completely.
1633                  */
1634                 if (likely(offsets->switch_old_end))
1635                         offsets->begin = subbuf_align(offsets->begin, chan);
1636                 offsets->begin = offsets->begin
1637                                  + config->cb.subbuffer_header_size();
1638                 /* Test new buffer integrity */
1639                 sb_index = subbuf_index(offsets->begin, chan);
1640                 /*
1641                  * Read buf->offset before buf->commit_cold[sb_index].cc_sb.
1642                  * lib_ring_buffer_check_deliver() has the matching
1643                  * memory barriers required around commit_cold cc_sb
1644                  * updates to ensure reserve and commit counter updates
1645                  * are not seen reordered when updated by another CPU.
1646                  */
1647                 smp_rmb();
1648                 commit_count = v_read(config,
1649                                 &buf->commit_cold[sb_index].cc_sb);
1650                 /* Read buf->commit_cold[sb_index].cc_sb before buf->offset. */
1651                 smp_rmb();
1652                 if (unlikely(offset_cmp != v_read(config, &buf->offset))) {
1653                         /*
1654                          * The reserve counter have been concurrently updated
1655                          * while we read the commit counter. This means the
1656                          * commit counter we read might not match buf->offset
1657                          * due to concurrent update. We therefore need to retry.
1658                          */
1659                         goto retry;
1660                 }
1661                 reserve_commit_diff =
1662                   (buf_trunc(offsets->begin, chan)
1663                    >> chan->backend.num_subbuf_order)
1664                   - (commit_count & chan->commit_count_mask);
1665                 if (likely(reserve_commit_diff == 0)) {
1666                         /* Next subbuffer not being written to. */
1667                         if (unlikely(config->mode != RING_BUFFER_OVERWRITE &&
1668                                 subbuf_trunc(offsets->begin, chan)
1669                                  - subbuf_trunc((unsigned long)
1670                                      atomic_long_read(&buf->consumed), chan)
1671                                 >= chan->backend.buf_size)) {
1672                                 /*
1673                                  * We do not overwrite non consumed buffers
1674                                  * and we are full : record is lost.
1675                                  */
1676                                 v_inc(config, &buf->records_lost_full);
1677                                 return -ENOBUFS;
1678                         } else {
1679                                 /*
1680                                  * Next subbuffer not being written to, and we
1681                                  * are either in overwrite mode or the buffer is
1682                                  * not full. It's safe to write in this new
1683                                  * subbuffer.
1684                                  */
1685                         }
1686                 } else {
1687                         /*
1688                          * Next subbuffer reserve offset does not match the
1689                          * commit offset, and this did not involve update to the
1690                          * reserve counter. Drop record in producer-consumer and
1691                          * overwrite mode.  Caused by either a writer OOPS or
1692                          * too many nested writes over a reserve/commit pair.
1693                          */
1694                         v_inc(config, &buf->records_lost_wrap);
1695                         return -EIO;
1696                 }
1697                 offsets->size =
1698                         config->cb.record_header_size(config, chan,
1699                                                 offsets->begin,
1700                                                 &offsets->pre_header_padding,
1701                                                 ctx);
1702                 offsets->size +=
1703                         lib_ring_buffer_align(offsets->begin + offsets->size,
1704                                               ctx->largest_align)
1705                         + ctx->data_size;
1706                 if (unlikely(subbuf_offset(offsets->begin, chan)
1707                              + offsets->size > chan->backend.subbuf_size)) {
1708                         /*
1709                          * Record too big for subbuffers, report error, don't
1710                          * complete the sub-buffer switch.
1711                          */
1712                         v_inc(config, &buf->records_lost_big);
1713                         return -ENOSPC;
1714                 } else {
1715                         /*
1716                          * We just made a successful buffer switch and the
1717                          * record fits in the new subbuffer. Let's write.
1718                          */
1719                 }
1720         } else {
1721                 /*
1722                  * Record fits in the current buffer and we are not on a switch
1723                  * boundary. It's safe to write.
1724                  */
1725         }
1726         offsets->end = offsets->begin + offsets->size;
1727
1728         if (unlikely(subbuf_offset(offsets->end, chan) == 0)) {
1729                 /*
1730                  * The offset_end will fall at the very beginning of the next
1731                  * subbuffer.
1732                  */
1733                 offsets->switch_new_end = 1;    /* For offsets->begin */
1734         }
1735         return 0;
1736 }
1737
1738 /**
1739  * lib_ring_buffer_reserve_slow - Atomic slot reservation in a buffer.
1740  * @ctx: ring buffer context.
1741  *
1742  * Return : -NOBUFS if not enough space, -ENOSPC if event size too large,
1743  * -EIO for other errors, else returns 0.
1744  * It will take care of sub-buffer switching.
1745  */
1746 int lib_ring_buffer_reserve_slow(struct lib_ring_buffer_ctx *ctx)
1747 {
1748         struct channel *chan = ctx->chan;
1749         const struct lib_ring_buffer_config *config = &chan->backend.config;
1750         struct lib_ring_buffer *buf;
1751         struct switch_offsets offsets;
1752         int ret;
1753
1754         if (config->alloc == RING_BUFFER_ALLOC_PER_CPU)
1755                 buf = per_cpu_ptr(chan->backend.buf, ctx->cpu);
1756         else
1757                 buf = chan->backend.buf;
1758         ctx->buf = buf;
1759
1760         offsets.size = 0;
1761
1762         do {
1763                 ret = lib_ring_buffer_try_reserve_slow(buf, chan, &offsets,
1764                                                        ctx);
1765                 if (unlikely(ret))
1766                         return ret;
1767         } while (unlikely(v_cmpxchg(config, &buf->offset, offsets.old,
1768                                     offsets.end)
1769                           != offsets.old));
1770
1771         /*
1772          * Atomically update last_tsc. This update races against concurrent
1773          * atomic updates, but the race will always cause supplementary full TSC
1774          * records, never the opposite (missing a full TSC record when it would
1775          * be needed).
1776          */
1777         save_last_tsc(config, buf, ctx->tsc);
1778
1779         /*
1780          * Push the reader if necessary
1781          */
1782         lib_ring_buffer_reserve_push_reader(buf, chan, offsets.end - 1);
1783
1784         /*
1785          * Clear noref flag for this subbuffer.
1786          */
1787         lib_ring_buffer_clear_noref(config, &buf->backend,
1788                                     subbuf_index(offsets.end - 1, chan));
1789
1790         /*
1791          * Switch old subbuffer if needed.
1792          */
1793         if (unlikely(offsets.switch_old_end)) {
1794                 lib_ring_buffer_clear_noref(config, &buf->backend,
1795                                             subbuf_index(offsets.old - 1, chan));
1796                 lib_ring_buffer_switch_old_end(buf, chan, &offsets, ctx->tsc);
1797         }
1798
1799         /*
1800          * Populate new subbuffer.
1801          */
1802         if (unlikely(offsets.switch_new_start))
1803                 lib_ring_buffer_switch_new_start(buf, chan, &offsets, ctx->tsc);
1804
1805         if (unlikely(offsets.switch_new_end))
1806                 lib_ring_buffer_switch_new_end(buf, chan, &offsets, ctx->tsc);
1807
1808         ctx->slot_size = offsets.size;
1809         ctx->pre_offset = offsets.begin;
1810         ctx->buf_offset = offsets.begin + offsets.pre_header_padding;
1811         return 0;
1812 }
1813 EXPORT_SYMBOL_GPL(lib_ring_buffer_reserve_slow);
1814
1815 int __init init_lib_ring_buffer_frontend(void)
1816 {
1817         int cpu;
1818
1819         for_each_possible_cpu(cpu)
1820                 spin_lock_init(&per_cpu(ring_buffer_nohz_lock, cpu));
1821         return 0;
1822 }
1823
1824 module_init(init_lib_ring_buffer_frontend);
1825
1826 void __exit exit_lib_ring_buffer_frontend(void)
1827 {
1828 }
1829
1830 module_exit(exit_lib_ring_buffer_frontend);