Commit | Line | Data |
---|---|---|
76369139 FW |
1 | /* |
2 | * Performance events ring-buffer code: | |
3 | * | |
4 | * Copyright (C) 2008 Thomas Gleixner <tglx@linutronix.de> | |
5 | * Copyright (C) 2008-2011 Red Hat, Inc., Ingo Molnar | |
6 | * Copyright (C) 2008-2011 Red Hat, Inc., Peter Zijlstra <pzijlstr@redhat.com> | |
d36b6910 | 7 | * Copyright © 2009 Paul Mackerras, IBM Corp. <paulus@au1.ibm.com> |
76369139 FW |
8 | * |
9 | * For licensing details see kernel-base/COPYING | |
10 | */ | |
11 | ||
12 | #include <linux/perf_event.h> | |
13 | #include <linux/vmalloc.h> | |
14 | #include <linux/slab.h> | |
26c86da8 | 15 | #include <linux/circ_buf.h> |
7c60fc0e | 16 | #include <linux/poll.h> |
76369139 FW |
17 | |
18 | #include "internal.h" | |
19 | ||
76369139 FW |
20 | static void perf_output_wakeup(struct perf_output_handle *handle) |
21 | { | |
7c60fc0e | 22 | atomic_set(&handle->rb->poll, POLLIN); |
76369139 | 23 | |
a8b0ca17 PZ |
24 | handle->event->pending_wakeup = 1; |
25 | irq_work_queue(&handle->event->pending); | |
76369139 FW |
26 | } |
27 | ||
28 | /* | |
29 | * We need to ensure a later event_id doesn't publish a head when a former | |
30 | * event isn't done writing. However since we need to deal with NMIs we | |
31 | * cannot fully serialize things. | |
32 | * | |
33 | * We only publish the head (and generate a wakeup) when the outer-most | |
34 | * event completes. | |
35 | */ | |
36 | static void perf_output_get_handle(struct perf_output_handle *handle) | |
37 | { | |
38 | struct ring_buffer *rb = handle->rb; | |
39 | ||
40 | preempt_disable(); | |
41 | local_inc(&rb->nest); | |
42 | handle->wakeup = local_read(&rb->wakeup); | |
43 | } | |
44 | ||
45 | static void perf_output_put_handle(struct perf_output_handle *handle) | |
46 | { | |
47 | struct ring_buffer *rb = handle->rb; | |
48 | unsigned long head; | |
49 | ||
50 | again: | |
51 | head = local_read(&rb->head); | |
52 | ||
53 | /* | |
54 | * IRQ/NMI can happen here, which means we can miss a head update. | |
55 | */ | |
56 | ||
57 | if (!local_dec_and_test(&rb->nest)) | |
58 | goto out; | |
59 | ||
60 | /* | |
bf378d34 PZ |
61 | * Since the mmap() consumer (userspace) can run on a different CPU: |
62 | * | |
63 | * kernel user | |
64 | * | |
c7f2e3cd PZ |
65 | * if (LOAD ->data_tail) { LOAD ->data_head |
66 | * (A) smp_rmb() (C) | |
67 | * STORE $data LOAD $data | |
68 | * smp_wmb() (B) smp_mb() (D) | |
69 | * STORE ->data_head STORE ->data_tail | |
70 | * } | |
bf378d34 PZ |
71 | * |
72 | * Where A pairs with D, and B pairs with C. | |
73 | * | |
c7f2e3cd PZ |
74 | * In our case (A) is a control dependency that separates the load of |
75 | * the ->data_tail and the stores of $data. In case ->data_tail | |
76 | * indicates there is no room in the buffer to store $data we do not. | |
bf378d34 | 77 | * |
c7f2e3cd | 78 | * D needs to be a full barrier since it separates the data READ |
bf378d34 PZ |
79 | * from the tail WRITE. |
80 | * | |
81 | * For B a WMB is sufficient since it separates two WRITEs, and for C | |
82 | * an RMB is sufficient since it separates two READs. | |
83 | * | |
84 | * See perf_output_begin(). | |
76369139 | 85 | */ |
c7f2e3cd | 86 | smp_wmb(); /* B, matches C */ |
76369139 FW |
87 | rb->user_page->data_head = head; |
88 | ||
89 | /* | |
394570b7 PZ |
90 | * Now check if we missed an update -- rely on previous implied |
91 | * compiler barriers to force a re-read. | |
76369139 FW |
92 | */ |
93 | if (unlikely(head != local_read(&rb->head))) { | |
94 | local_inc(&rb->nest); | |
95 | goto again; | |
96 | } | |
97 | ||
98 | if (handle->wakeup != local_read(&rb->wakeup)) | |
99 | perf_output_wakeup(handle); | |
100 | ||
101 | out: | |
102 | preempt_enable(); | |
103 | } | |
104 | ||
105 | int perf_output_begin(struct perf_output_handle *handle, | |
a7ac67ea | 106 | struct perf_event *event, unsigned int size) |
76369139 FW |
107 | { |
108 | struct ring_buffer *rb; | |
109 | unsigned long tail, offset, head; | |
524feca5 | 110 | int have_lost, page_shift; |
76369139 FW |
111 | struct { |
112 | struct perf_event_header header; | |
113 | u64 id; | |
114 | u64 lost; | |
115 | } lost_event; | |
116 | ||
117 | rcu_read_lock(); | |
118 | /* | |
119 | * For inherited events we send all the output towards the parent. | |
120 | */ | |
121 | if (event->parent) | |
122 | event = event->parent; | |
123 | ||
124 | rb = rcu_dereference(event->rb); | |
c72b42a3 | 125 | if (unlikely(!rb)) |
76369139 FW |
126 | goto out; |
127 | ||
c72b42a3 | 128 | if (unlikely(!rb->nr_pages)) |
76369139 FW |
129 | goto out; |
130 | ||
c72b42a3 PZ |
131 | handle->rb = rb; |
132 | handle->event = event; | |
133 | ||
76369139 | 134 | have_lost = local_read(&rb->lost); |
c72b42a3 | 135 | if (unlikely(have_lost)) { |
d20a973f PZ |
136 | size += sizeof(lost_event); |
137 | if (event->attr.sample_id_all) | |
138 | size += event->id_header_size; | |
76369139 FW |
139 | } |
140 | ||
141 | perf_output_get_handle(handle); | |
142 | ||
143 | do { | |
76369139 | 144 | tail = ACCESS_ONCE(rb->user_page->data_tail); |
76369139 | 145 | offset = head = local_read(&rb->head); |
26c86da8 PZ |
146 | if (!rb->overwrite && |
147 | unlikely(CIRC_SPACE(head, tail, perf_data_size(rb)) < size)) | |
76369139 | 148 | goto fail; |
c7f2e3cd PZ |
149 | |
150 | /* | |
151 | * The above forms a control dependency barrier separating the | |
152 | * @tail load above from the data stores below. Since the @tail | |
153 | * load is required to compute the branch to fail below. | |
154 | * | |
155 | * A, matches D; the full memory barrier userspace SHOULD issue | |
156 | * after reading the data and before storing the new tail | |
157 | * position. | |
158 | * | |
159 | * See perf_output_put_handle(). | |
160 | */ | |
161 | ||
26c86da8 | 162 | head += size; |
76369139 FW |
163 | } while (local_cmpxchg(&rb->head, offset, head) != offset); |
164 | ||
85f59edf | 165 | /* |
c7f2e3cd PZ |
166 | * We rely on the implied barrier() by local_cmpxchg() to ensure |
167 | * none of the data stores below can be lifted up by the compiler. | |
85f59edf | 168 | */ |
85f59edf | 169 | |
c72b42a3 | 170 | if (unlikely(head - local_read(&rb->wakeup) > rb->watermark)) |
76369139 FW |
171 | local_add(rb->watermark, &rb->wakeup); |
172 | ||
524feca5 PZ |
173 | page_shift = PAGE_SHIFT + page_order(rb); |
174 | ||
175 | handle->page = (offset >> page_shift) & (rb->nr_pages - 1); | |
176 | offset &= (1UL << page_shift) - 1; | |
177 | handle->addr = rb->data_pages[handle->page] + offset; | |
178 | handle->size = (1UL << page_shift) - offset; | |
76369139 | 179 | |
c72b42a3 | 180 | if (unlikely(have_lost)) { |
d20a973f PZ |
181 | struct perf_sample_data sample_data; |
182 | ||
183 | lost_event.header.size = sizeof(lost_event); | |
76369139 FW |
184 | lost_event.header.type = PERF_RECORD_LOST; |
185 | lost_event.header.misc = 0; | |
186 | lost_event.id = event->id; | |
187 | lost_event.lost = local_xchg(&rb->lost, 0); | |
188 | ||
d20a973f PZ |
189 | perf_event_header__init_id(&lost_event.header, |
190 | &sample_data, event); | |
76369139 FW |
191 | perf_output_put(handle, lost_event); |
192 | perf_event__output_id_sample(event, handle, &sample_data); | |
193 | } | |
194 | ||
195 | return 0; | |
196 | ||
197 | fail: | |
198 | local_inc(&rb->lost); | |
199 | perf_output_put_handle(handle); | |
200 | out: | |
201 | rcu_read_unlock(); | |
202 | ||
203 | return -ENOSPC; | |
204 | } | |
205 | ||
91d7753a | 206 | unsigned int perf_output_copy(struct perf_output_handle *handle, |
76369139 FW |
207 | const void *buf, unsigned int len) |
208 | { | |
91d7753a | 209 | return __output_copy(handle, buf, len); |
76369139 FW |
210 | } |
211 | ||
5685e0ff JO |
212 | unsigned int perf_output_skip(struct perf_output_handle *handle, |
213 | unsigned int len) | |
214 | { | |
215 | return __output_skip(handle, NULL, len); | |
216 | } | |
217 | ||
76369139 FW |
218 | void perf_output_end(struct perf_output_handle *handle) |
219 | { | |
76369139 FW |
220 | perf_output_put_handle(handle); |
221 | rcu_read_unlock(); | |
222 | } | |
223 | ||
224 | static void | |
225 | ring_buffer_init(struct ring_buffer *rb, long watermark, int flags) | |
226 | { | |
227 | long max_size = perf_data_size(rb); | |
228 | ||
229 | if (watermark) | |
230 | rb->watermark = min(max_size, watermark); | |
231 | ||
232 | if (!rb->watermark) | |
233 | rb->watermark = max_size / 2; | |
234 | ||
235 | if (flags & RING_BUFFER_WRITABLE) | |
dd9c086d SE |
236 | rb->overwrite = 0; |
237 | else | |
238 | rb->overwrite = 1; | |
76369139 FW |
239 | |
240 | atomic_set(&rb->refcount, 1); | |
10c6db11 PZ |
241 | |
242 | INIT_LIST_HEAD(&rb->event_list); | |
243 | spin_lock_init(&rb->event_lock); | |
76369139 FW |
244 | } |
245 | ||
fdc26706 AS |
246 | /* |
247 | * This is called before hardware starts writing to the AUX area to | |
248 | * obtain an output handle and make sure there's room in the buffer. | |
249 | * When the capture completes, call perf_aux_output_end() to commit | |
250 | * the recorded data to the buffer. | |
251 | * | |
252 | * The ordering is similar to that of perf_output_{begin,end}, with | |
253 | * the exception of (B), which should be taken care of by the pmu | |
254 | * driver, since ordering rules will differ depending on hardware. | |
255 | */ | |
256 | void *perf_aux_output_begin(struct perf_output_handle *handle, | |
257 | struct perf_event *event) | |
258 | { | |
259 | struct perf_event *output_event = event; | |
260 | unsigned long aux_head, aux_tail; | |
261 | struct ring_buffer *rb; | |
262 | ||
263 | if (output_event->parent) | |
264 | output_event = output_event->parent; | |
265 | ||
266 | /* | |
267 | * Since this will typically be open across pmu::add/pmu::del, we | |
268 | * grab ring_buffer's refcount instead of holding rcu read lock | |
269 | * to make sure it doesn't disappear under us. | |
270 | */ | |
271 | rb = ring_buffer_get(output_event); | |
272 | if (!rb) | |
273 | return NULL; | |
274 | ||
275 | if (!rb_has_aux(rb) || !atomic_inc_not_zero(&rb->aux_refcount)) | |
276 | goto err; | |
277 | ||
278 | /* | |
279 | * Nesting is not supported for AUX area, make sure nested | |
280 | * writers are caught early | |
281 | */ | |
282 | if (WARN_ON_ONCE(local_xchg(&rb->aux_nest, 1))) | |
283 | goto err_put; | |
284 | ||
285 | aux_head = local_read(&rb->aux_head); | |
fdc26706 AS |
286 | |
287 | handle->rb = rb; | |
288 | handle->event = event; | |
289 | handle->head = aux_head; | |
2023a0d2 | 290 | handle->size = 0; |
fdc26706 AS |
291 | |
292 | /* | |
2023a0d2 AS |
293 | * In overwrite mode, AUX data stores do not depend on aux_tail, |
294 | * therefore (A) control dependency barrier does not exist. The | |
295 | * (B) <-> (C) ordering is still observed by the pmu driver. | |
fdc26706 | 296 | */ |
2023a0d2 AS |
297 | if (!rb->aux_overwrite) { |
298 | aux_tail = ACCESS_ONCE(rb->user_page->aux_tail); | |
299 | if (aux_head - aux_tail < perf_aux_size(rb)) | |
300 | handle->size = CIRC_SPACE(aux_head, aux_tail, perf_aux_size(rb)); | |
301 | ||
302 | /* | |
303 | * handle->size computation depends on aux_tail load; this forms a | |
304 | * control dependency barrier separating aux_tail load from aux data | |
305 | * store that will be enabled on successful return | |
306 | */ | |
307 | if (!handle->size) { /* A, matches D */ | |
308 | event->pending_disable = 1; | |
309 | perf_output_wakeup(handle); | |
310 | local_set(&rb->aux_nest, 0); | |
311 | goto err_put; | |
312 | } | |
fdc26706 AS |
313 | } |
314 | ||
315 | return handle->rb->aux_priv; | |
316 | ||
317 | err_put: | |
318 | rb_free_aux(rb); | |
319 | ||
320 | err: | |
321 | ring_buffer_put(rb); | |
322 | handle->event = NULL; | |
323 | ||
324 | return NULL; | |
325 | } | |
326 | ||
327 | /* | |
328 | * Commit the data written by hardware into the ring buffer by adjusting | |
329 | * aux_head and posting a PERF_RECORD_AUX into the perf buffer. It is the | |
330 | * pmu driver's responsibility to observe ordering rules of the hardware, | |
331 | * so that all the data is externally visible before this is called. | |
332 | */ | |
333 | void perf_aux_output_end(struct perf_output_handle *handle, unsigned long size, | |
334 | bool truncated) | |
335 | { | |
336 | struct ring_buffer *rb = handle->rb; | |
2023a0d2 | 337 | unsigned long aux_head; |
fdc26706 AS |
338 | u64 flags = 0; |
339 | ||
340 | if (truncated) | |
341 | flags |= PERF_AUX_FLAG_TRUNCATED; | |
342 | ||
2023a0d2 AS |
343 | /* in overwrite mode, driver provides aux_head via handle */ |
344 | if (rb->aux_overwrite) { | |
345 | flags |= PERF_AUX_FLAG_OVERWRITE; | |
346 | ||
347 | aux_head = handle->head; | |
348 | local_set(&rb->aux_head, aux_head); | |
349 | } else { | |
350 | aux_head = local_read(&rb->aux_head); | |
351 | local_add(size, &rb->aux_head); | |
352 | } | |
fdc26706 AS |
353 | |
354 | if (size || flags) { | |
355 | /* | |
356 | * Only send RECORD_AUX if we have something useful to communicate | |
357 | */ | |
358 | ||
359 | perf_event_aux_event(handle->event, aux_head, size, flags); | |
360 | } | |
361 | ||
362 | rb->user_page->aux_head = local_read(&rb->aux_head); | |
363 | ||
364 | perf_output_wakeup(handle); | |
365 | handle->event = NULL; | |
366 | ||
367 | local_set(&rb->aux_nest, 0); | |
368 | rb_free_aux(rb); | |
369 | ring_buffer_put(rb); | |
370 | } | |
371 | ||
372 | /* | |
373 | * Skip over a given number of bytes in the AUX buffer, due to, for example, | |
374 | * hardware's alignment constraints. | |
375 | */ | |
376 | int perf_aux_output_skip(struct perf_output_handle *handle, unsigned long size) | |
377 | { | |
378 | struct ring_buffer *rb = handle->rb; | |
379 | unsigned long aux_head; | |
380 | ||
381 | if (size > handle->size) | |
382 | return -ENOSPC; | |
383 | ||
384 | local_add(size, &rb->aux_head); | |
385 | ||
386 | handle->head = aux_head; | |
387 | handle->size -= size; | |
388 | ||
389 | return 0; | |
390 | } | |
391 | ||
392 | void *perf_get_aux(struct perf_output_handle *handle) | |
393 | { | |
394 | /* this is only valid between perf_aux_output_begin and *_end */ | |
395 | if (!handle->event) | |
396 | return NULL; | |
397 | ||
398 | return handle->rb->aux_priv; | |
399 | } | |
400 | ||
0a4e38e6 AS |
401 | #define PERF_AUX_GFP (GFP_KERNEL | __GFP_ZERO | __GFP_NOWARN | __GFP_NORETRY) |
402 | ||
403 | static struct page *rb_alloc_aux_page(int node, int order) | |
404 | { | |
405 | struct page *page; | |
406 | ||
407 | if (order > MAX_ORDER) | |
408 | order = MAX_ORDER; | |
409 | ||
410 | do { | |
411 | page = alloc_pages_node(node, PERF_AUX_GFP, order); | |
412 | } while (!page && order--); | |
413 | ||
414 | if (page && order) { | |
415 | /* | |
416 | * Communicate the allocation size to the driver | |
417 | */ | |
418 | split_page(page, order); | |
419 | SetPagePrivate(page); | |
420 | set_page_private(page, order); | |
421 | } | |
422 | ||
423 | return page; | |
424 | } | |
425 | ||
426 | static void rb_free_aux_page(struct ring_buffer *rb, int idx) | |
427 | { | |
428 | struct page *page = virt_to_page(rb->aux_pages[idx]); | |
429 | ||
430 | ClearPagePrivate(page); | |
431 | page->mapping = NULL; | |
432 | __free_page(page); | |
433 | } | |
434 | ||
45bfb2e5 PZ |
435 | int rb_alloc_aux(struct ring_buffer *rb, struct perf_event *event, |
436 | pgoff_t pgoff, int nr_pages, int flags) | |
437 | { | |
438 | bool overwrite = !(flags & RING_BUFFER_WRITABLE); | |
439 | int node = (event->cpu == -1) ? -1 : cpu_to_node(event->cpu); | |
0a4e38e6 | 440 | int ret = -ENOMEM, max_order = 0; |
45bfb2e5 PZ |
441 | |
442 | if (!has_aux(event)) | |
443 | return -ENOTSUPP; | |
444 | ||
6a279230 | 445 | if (event->pmu->capabilities & PERF_PMU_CAP_AUX_NO_SG) { |
0a4e38e6 AS |
446 | /* |
447 | * We need to start with the max_order that fits in nr_pages, | |
448 | * not the other way around, hence ilog2() and not get_order. | |
449 | */ | |
450 | max_order = ilog2(nr_pages); | |
451 | ||
6a279230 AS |
452 | /* |
453 | * PMU requests more than one contiguous chunks of memory | |
454 | * for SW double buffering | |
455 | */ | |
456 | if ((event->pmu->capabilities & PERF_PMU_CAP_AUX_SW_DOUBLEBUF) && | |
457 | !overwrite) { | |
458 | if (!max_order) | |
459 | return -EINVAL; | |
460 | ||
461 | max_order--; | |
462 | } | |
463 | } | |
464 | ||
45bfb2e5 PZ |
465 | rb->aux_pages = kzalloc_node(nr_pages * sizeof(void *), GFP_KERNEL, node); |
466 | if (!rb->aux_pages) | |
467 | return -ENOMEM; | |
468 | ||
469 | rb->free_aux = event->pmu->free_aux; | |
0a4e38e6 | 470 | for (rb->aux_nr_pages = 0; rb->aux_nr_pages < nr_pages;) { |
45bfb2e5 | 471 | struct page *page; |
0a4e38e6 | 472 | int last, order; |
45bfb2e5 | 473 | |
0a4e38e6 AS |
474 | order = min(max_order, ilog2(nr_pages - rb->aux_nr_pages)); |
475 | page = rb_alloc_aux_page(node, order); | |
45bfb2e5 PZ |
476 | if (!page) |
477 | goto out; | |
478 | ||
0a4e38e6 AS |
479 | for (last = rb->aux_nr_pages + (1 << page_private(page)); |
480 | last > rb->aux_nr_pages; rb->aux_nr_pages++) | |
481 | rb->aux_pages[rb->aux_nr_pages] = page_address(page++); | |
45bfb2e5 PZ |
482 | } |
483 | ||
484 | rb->aux_priv = event->pmu->setup_aux(event->cpu, rb->aux_pages, nr_pages, | |
485 | overwrite); | |
486 | if (!rb->aux_priv) | |
487 | goto out; | |
488 | ||
489 | ret = 0; | |
490 | ||
491 | /* | |
492 | * aux_pages (and pmu driver's private data, aux_priv) will be | |
493 | * referenced in both producer's and consumer's contexts, thus | |
494 | * we keep a refcount here to make sure either of the two can | |
495 | * reference them safely. | |
496 | */ | |
497 | atomic_set(&rb->aux_refcount, 1); | |
498 | ||
2023a0d2 AS |
499 | rb->aux_overwrite = overwrite; |
500 | ||
45bfb2e5 PZ |
501 | out: |
502 | if (!ret) | |
503 | rb->aux_pgoff = pgoff; | |
504 | else | |
505 | rb_free_aux(rb); | |
506 | ||
507 | return ret; | |
508 | } | |
509 | ||
510 | static void __rb_free_aux(struct ring_buffer *rb) | |
511 | { | |
512 | int pg; | |
513 | ||
514 | if (rb->aux_priv) { | |
515 | rb->free_aux(rb->aux_priv); | |
516 | rb->free_aux = NULL; | |
517 | rb->aux_priv = NULL; | |
518 | } | |
519 | ||
520 | for (pg = 0; pg < rb->aux_nr_pages; pg++) | |
0a4e38e6 | 521 | rb_free_aux_page(rb, pg); |
45bfb2e5 PZ |
522 | |
523 | kfree(rb->aux_pages); | |
524 | rb->aux_nr_pages = 0; | |
525 | } | |
526 | ||
527 | void rb_free_aux(struct ring_buffer *rb) | |
528 | { | |
529 | if (atomic_dec_and_test(&rb->aux_refcount)) | |
530 | __rb_free_aux(rb); | |
531 | } | |
532 | ||
76369139 FW |
533 | #ifndef CONFIG_PERF_USE_VMALLOC |
534 | ||
535 | /* | |
536 | * Back perf_mmap() with regular GFP_KERNEL-0 pages. | |
537 | */ | |
538 | ||
45bfb2e5 PZ |
539 | static struct page * |
540 | __perf_mmap_to_page(struct ring_buffer *rb, unsigned long pgoff) | |
76369139 FW |
541 | { |
542 | if (pgoff > rb->nr_pages) | |
543 | return NULL; | |
544 | ||
545 | if (pgoff == 0) | |
546 | return virt_to_page(rb->user_page); | |
547 | ||
548 | return virt_to_page(rb->data_pages[pgoff - 1]); | |
549 | } | |
550 | ||
551 | static void *perf_mmap_alloc_page(int cpu) | |
552 | { | |
553 | struct page *page; | |
554 | int node; | |
555 | ||
556 | node = (cpu == -1) ? cpu : cpu_to_node(cpu); | |
557 | page = alloc_pages_node(node, GFP_KERNEL | __GFP_ZERO, 0); | |
558 | if (!page) | |
559 | return NULL; | |
560 | ||
561 | return page_address(page); | |
562 | } | |
563 | ||
564 | struct ring_buffer *rb_alloc(int nr_pages, long watermark, int cpu, int flags) | |
565 | { | |
566 | struct ring_buffer *rb; | |
567 | unsigned long size; | |
568 | int i; | |
569 | ||
570 | size = sizeof(struct ring_buffer); | |
571 | size += nr_pages * sizeof(void *); | |
572 | ||
573 | rb = kzalloc(size, GFP_KERNEL); | |
574 | if (!rb) | |
575 | goto fail; | |
576 | ||
577 | rb->user_page = perf_mmap_alloc_page(cpu); | |
578 | if (!rb->user_page) | |
579 | goto fail_user_page; | |
580 | ||
581 | for (i = 0; i < nr_pages; i++) { | |
582 | rb->data_pages[i] = perf_mmap_alloc_page(cpu); | |
583 | if (!rb->data_pages[i]) | |
584 | goto fail_data_pages; | |
585 | } | |
586 | ||
587 | rb->nr_pages = nr_pages; | |
588 | ||
589 | ring_buffer_init(rb, watermark, flags); | |
590 | ||
591 | return rb; | |
592 | ||
593 | fail_data_pages: | |
594 | for (i--; i >= 0; i--) | |
595 | free_page((unsigned long)rb->data_pages[i]); | |
596 | ||
597 | free_page((unsigned long)rb->user_page); | |
598 | ||
599 | fail_user_page: | |
600 | kfree(rb); | |
601 | ||
602 | fail: | |
603 | return NULL; | |
604 | } | |
605 | ||
606 | static void perf_mmap_free_page(unsigned long addr) | |
607 | { | |
608 | struct page *page = virt_to_page((void *)addr); | |
609 | ||
610 | page->mapping = NULL; | |
611 | __free_page(page); | |
612 | } | |
613 | ||
614 | void rb_free(struct ring_buffer *rb) | |
615 | { | |
616 | int i; | |
617 | ||
618 | perf_mmap_free_page((unsigned long)rb->user_page); | |
619 | for (i = 0; i < rb->nr_pages; i++) | |
620 | perf_mmap_free_page((unsigned long)rb->data_pages[i]); | |
621 | kfree(rb); | |
622 | } | |
623 | ||
624 | #else | |
5919b309 JO |
625 | static int data_page_nr(struct ring_buffer *rb) |
626 | { | |
627 | return rb->nr_pages << page_order(rb); | |
628 | } | |
76369139 | 629 | |
45bfb2e5 PZ |
630 | static struct page * |
631 | __perf_mmap_to_page(struct ring_buffer *rb, unsigned long pgoff) | |
76369139 | 632 | { |
5919b309 JO |
633 | /* The '>' counts in the user page. */ |
634 | if (pgoff > data_page_nr(rb)) | |
76369139 FW |
635 | return NULL; |
636 | ||
637 | return vmalloc_to_page((void *)rb->user_page + pgoff * PAGE_SIZE); | |
638 | } | |
639 | ||
640 | static void perf_mmap_unmark_page(void *addr) | |
641 | { | |
642 | struct page *page = vmalloc_to_page(addr); | |
643 | ||
644 | page->mapping = NULL; | |
645 | } | |
646 | ||
647 | static void rb_free_work(struct work_struct *work) | |
648 | { | |
649 | struct ring_buffer *rb; | |
650 | void *base; | |
651 | int i, nr; | |
652 | ||
653 | rb = container_of(work, struct ring_buffer, work); | |
5919b309 | 654 | nr = data_page_nr(rb); |
76369139 FW |
655 | |
656 | base = rb->user_page; | |
5919b309 JO |
657 | /* The '<=' counts in the user page. */ |
658 | for (i = 0; i <= nr; i++) | |
76369139 FW |
659 | perf_mmap_unmark_page(base + (i * PAGE_SIZE)); |
660 | ||
661 | vfree(base); | |
662 | kfree(rb); | |
663 | } | |
664 | ||
665 | void rb_free(struct ring_buffer *rb) | |
666 | { | |
667 | schedule_work(&rb->work); | |
668 | } | |
669 | ||
670 | struct ring_buffer *rb_alloc(int nr_pages, long watermark, int cpu, int flags) | |
671 | { | |
672 | struct ring_buffer *rb; | |
673 | unsigned long size; | |
674 | void *all_buf; | |
675 | ||
676 | size = sizeof(struct ring_buffer); | |
677 | size += sizeof(void *); | |
678 | ||
679 | rb = kzalloc(size, GFP_KERNEL); | |
680 | if (!rb) | |
681 | goto fail; | |
682 | ||
683 | INIT_WORK(&rb->work, rb_free_work); | |
684 | ||
685 | all_buf = vmalloc_user((nr_pages + 1) * PAGE_SIZE); | |
686 | if (!all_buf) | |
687 | goto fail_all_buf; | |
688 | ||
689 | rb->user_page = all_buf; | |
690 | rb->data_pages[0] = all_buf + PAGE_SIZE; | |
691 | rb->page_order = ilog2(nr_pages); | |
5919b309 | 692 | rb->nr_pages = !!nr_pages; |
76369139 FW |
693 | |
694 | ring_buffer_init(rb, watermark, flags); | |
695 | ||
696 | return rb; | |
697 | ||
698 | fail_all_buf: | |
699 | kfree(rb); | |
700 | ||
701 | fail: | |
702 | return NULL; | |
703 | } | |
704 | ||
705 | #endif | |
45bfb2e5 PZ |
706 | |
707 | struct page * | |
708 | perf_mmap_to_page(struct ring_buffer *rb, unsigned long pgoff) | |
709 | { | |
710 | if (rb->aux_nr_pages) { | |
711 | /* above AUX space */ | |
712 | if (pgoff > rb->aux_pgoff + rb->aux_nr_pages) | |
713 | return NULL; | |
714 | ||
715 | /* AUX space */ | |
716 | if (pgoff >= rb->aux_pgoff) | |
717 | return virt_to_page(rb->aux_pages[pgoff - rb->aux_pgoff]); | |
718 | } | |
719 | ||
720 | return __perf_mmap_to_page(rb, pgoff); | |
721 | } |