22 	atomic_set(&handle->rb->poll, EPOLLIN | EPOLLRDNORM);  in perf_output_wakeup()
42 	struct perf_buffer *rb = handle->rb;  in perf_output_get_handle()  local
50 	(*(volatile unsigned int *)&rb->nest)++;  in perf_output_get_handle()
51 	handle->wakeup = local_read(&rb->wakeup);  in perf_output_get_handle()
56 	struct perf_buffer *rb = handle->rb;  in perf_output_put_handle()  local
64 	nest = READ_ONCE(rb->nest);  in perf_output_put_handle()
66 		WRITE_ONCE(rb->nest, nest - 1);  in perf_output_put_handle()
80 	head = local_read(&rb->head);  in perf_output_put_handle()
114 	WRITE_ONCE(rb->user_page->data_head, head);  in perf_output_put_handle()
122 	WRITE_ONCE(rb->nest, 0);  in perf_output_put_handle()
129 	if (unlikely(head != local_read(&rb->head))) {  in perf_output_put_handle()
130 		WRITE_ONCE(rb->nest, 1);  in perf_output_put_handle()
134 	if (handle->wakeup != local_read(&rb->wakeup))  in perf_output_put_handle()
158 	struct perf_buffer *rb;  in __perf_output_begin()  local
174 	rb = rcu_dereference(event->rb);  in __perf_output_begin()
175 	if (unlikely(!rb))  in __perf_output_begin()
178 	if (unlikely(rb->paused)) {  in __perf_output_begin()
179 		if (rb->nr_pages) {  in __perf_output_begin()
180 			local_inc(&rb->lost);  in __perf_output_begin()
186 	handle->rb    = rb;  in __perf_output_begin()
190 	have_lost = local_read(&rb->lost);  in __perf_output_begin()
199 	offset = local_read(&rb->head);  in __perf_output_begin()
202 		tail = READ_ONCE(rb->user_page->data_tail);  in __perf_output_begin()
203 		if (!rb->overwrite) {  in __perf_output_begin()
205 							    perf_data_size(rb),  in __perf_output_begin()
226 	} while (!local_try_cmpxchg(&rb->head, &offset, head));  in __perf_output_begin()
238 	if (unlikely(head - local_read(&rb->wakeup) > rb->watermark))  in __perf_output_begin()
239 		local_add(rb->watermark, &rb->wakeup);  in __perf_output_begin()
241 	page_shift = PAGE_SHIFT + page_order(rb);  in __perf_output_begin()
243 	handle->page = (offset >> page_shift) & (rb->nr_pages - 1);  in __perf_output_begin()
245 	handle->addr = rb->data_pages[handle->page] + offset;  in __perf_output_begin()
253 		lost_event.lost        = local_xchg(&rb->lost, 0);  in __perf_output_begin()
264 	local_inc(&rb->lost);  in __perf_output_begin()
315 ring_buffer_init(struct perf_buffer *rb, long watermark, int flags)  in ring_buffer_init()  argument
317 	long max_size = perf_data_size(rb);  in ring_buffer_init()
320 		rb->watermark = min(max_size, watermark);  in ring_buffer_init()
322 	if (!rb->watermark)  in ring_buffer_init()
323 		rb->watermark = max_size / 2;  in ring_buffer_init()
326 		rb->overwrite = 0;  in ring_buffer_init()
328 		rb->overwrite = 1;  in ring_buffer_init()
330 	refcount_set(&rb->refcount, 1);  in ring_buffer_init()
332 	INIT_LIST_HEAD(&rb->event_list);  in ring_buffer_init()
333 	spin_lock_init(&rb->event_lock);  in ring_buffer_init()
339 	if (!rb->nr_pages)  in ring_buffer_init()
340 		rb->paused = 1;  in ring_buffer_init()
342 	mutex_init(&rb->aux_mutex);  in ring_buffer_init()
377 	struct perf_buffer *rb;  in perf_aux_output_begin()  local
388 	rb = ring_buffer_get(output_event);  in perf_aux_output_begin()
389 	if (!rb)  in perf_aux_output_begin()
392 	if (!rb_has_aux(rb))  in perf_aux_output_begin()
403 	if (!atomic_read(&rb->aux_mmap_count))  in perf_aux_output_begin()
406 	if (!refcount_inc_not_zero(&rb->aux_refcount))  in perf_aux_output_begin()
409 	nest = READ_ONCE(rb->aux_nest);  in perf_aux_output_begin()
417 	WRITE_ONCE(rb->aux_nest, nest + 1);  in perf_aux_output_begin()
419 	aux_head = rb->aux_head;  in perf_aux_output_begin()
421 	handle->rb = rb;  in perf_aux_output_begin()
432 	if (!rb->aux_overwrite) {  in perf_aux_output_begin()
433 		aux_tail = READ_ONCE(rb->user_page->aux_tail);  in perf_aux_output_begin()
434 		handle->wakeup = rb->aux_wakeup + rb->aux_watermark;  in perf_aux_output_begin()
435 		if (aux_head - aux_tail < perf_aux_size(rb))  in perf_aux_output_begin()
436 			handle->size = CIRC_SPACE(aux_head, aux_tail, perf_aux_size(rb));  in perf_aux_output_begin()
446 			WRITE_ONCE(rb->aux_nest, 0);  in perf_aux_output_begin()
451 	return handle->rb->aux_priv;  in perf_aux_output_begin()
455 	rb_free_aux(rb);  in perf_aux_output_begin()
458 	ring_buffer_put(rb);  in perf_aux_output_begin()
465 static __always_inline bool rb_need_aux_wakeup(struct perf_buffer *rb)  in rb_need_aux_wakeup()  argument
467 	if (rb->aux_overwrite)  in rb_need_aux_wakeup()
470 	if (rb->aux_head - rb->aux_wakeup >= rb->aux_watermark) {  in rb_need_aux_wakeup()
471 		rb->aux_wakeup = rounddown(rb->aux_head, rb->aux_watermark);  in rb_need_aux_wakeup()
491 	struct perf_buffer *rb = handle->rb;  in perf_aux_output_end()  local
495 	if (rb->aux_overwrite) {  in perf_aux_output_end()
499 		rb->aux_head = aux_head;  in perf_aux_output_end()
503 		aux_head = rb->aux_head;  in perf_aux_output_end()
504 		rb->aux_head += size;  in perf_aux_output_end()
523 	WRITE_ONCE(rb->user_page->aux_head, rb->aux_head);  in perf_aux_output_end()
524 	if (rb_need_aux_wakeup(rb))  in perf_aux_output_end()
535 	WRITE_ONCE(rb->aux_nest, 0);  in perf_aux_output_end()
537 	rb_free_aux(rb);  in perf_aux_output_end()
538 	ring_buffer_put(rb);  in perf_aux_output_end()
548 	struct perf_buffer *rb = handle->rb;  in perf_aux_output_skip()  local
553 	rb->aux_head += size;  in perf_aux_output_skip()
555 	WRITE_ONCE(rb->user_page->aux_head, rb->aux_head);  in perf_aux_output_skip()
556 	if (rb_need_aux_wakeup(rb)) {  in perf_aux_output_skip()
558 		handle->wakeup = rb->aux_wakeup + rb->aux_watermark;  in perf_aux_output_skip()
561 	handle->head = rb->aux_head;  in perf_aux_output_skip()
574 	return handle->rb->aux_priv;  in perf_get_aux()
585 	struct perf_buffer *rb = aux_handle->rb;  in perf_output_copy_aux()  local
589 	from &= (rb->aux_nr_pages << PAGE_SHIFT) - 1;  in perf_output_copy_aux()
590 	to &= (rb->aux_nr_pages << PAGE_SHIFT) - 1;  in perf_output_copy_aux()
599 		addr = rb->aux_pages[from >> PAGE_SHIFT];  in perf_output_copy_aux()
608 		from &= (rb->aux_nr_pages << PAGE_SHIFT) - 1;  in perf_output_copy_aux()
642 static void rb_free_aux_page(struct perf_buffer *rb, int idx)  in rb_free_aux_page()  argument
644 	struct page *page = virt_to_page(rb->aux_pages[idx]);  in rb_free_aux_page()
650 static void __rb_free_aux(struct perf_buffer *rb)  in __rb_free_aux()  argument
662 	if (rb->aux_priv) {  in __rb_free_aux()
663 		rb->free_aux(rb->aux_priv);  in __rb_free_aux()
664 		rb->free_aux = NULL;  in __rb_free_aux()
665 		rb->aux_priv = NULL;  in __rb_free_aux()
668 	if (rb->aux_nr_pages) {  in __rb_free_aux()
669 		for (pg = 0; pg < rb->aux_nr_pages; pg++)  in __rb_free_aux()
670 			rb_free_aux_page(rb, pg);  in __rb_free_aux()
672 		kfree(rb->aux_pages);  in __rb_free_aux()
673 		rb->aux_nr_pages = 0;  in __rb_free_aux()
677 int rb_alloc_aux(struct perf_buffer *rb, struct perf_event *event,  in rb_alloc_aux()  argument
720 	rb->aux_pages = kcalloc_node(nr_pages, sizeof(void *), GFP_KERNEL,  in rb_alloc_aux()
722 	if (!rb->aux_pages)  in rb_alloc_aux()
725 	rb->free_aux = event->pmu->free_aux;  in rb_alloc_aux()
726 	for (rb->aux_nr_pages = 0; rb->aux_nr_pages < nr_pages;) {  in rb_alloc_aux()
730 		order = min(max_order, ilog2(nr_pages - rb->aux_nr_pages));  in rb_alloc_aux()
735 		for (last = rb->aux_nr_pages + (1 << page_private(page));  in rb_alloc_aux()
736 		     last > rb->aux_nr_pages; rb->aux_nr_pages++)  in rb_alloc_aux()
737 			rb->aux_pages[rb->aux_nr_pages] = page_address(page++);  in rb_alloc_aux()
748 		struct page *page = virt_to_page(rb->aux_pages[0]);  in rb_alloc_aux()
754 	rb->aux_priv = event->pmu->setup_aux(event, rb->aux_pages, nr_pages,  in rb_alloc_aux()
756 	if (!rb->aux_priv)  in rb_alloc_aux()
767 	refcount_set(&rb->aux_refcount, 1);  in rb_alloc_aux()
769 	rb->aux_overwrite = overwrite;  in rb_alloc_aux()
770 	rb->aux_watermark = watermark;  in rb_alloc_aux()
774 		rb->aux_pgoff = pgoff;  in rb_alloc_aux()
776 		__rb_free_aux(rb);  in rb_alloc_aux()
781 void rb_free_aux(struct perf_buffer *rb)  in rb_free_aux()  argument
783 	if (refcount_dec_and_test(&rb->aux_refcount))  in rb_free_aux()
784 		__rb_free_aux(rb);  in rb_free_aux()
794 __perf_mmap_to_page(struct perf_buffer *rb, unsigned long pgoff)  in __perf_mmap_to_page()  argument
796 	if (pgoff > rb->nr_pages)  in __perf_mmap_to_page()
800 		return virt_to_page(rb->user_page);  in __perf_mmap_to_page()
802 	return virt_to_page(rb->data_pages[pgoff - 1]);  in __perf_mmap_to_page()
827 	struct perf_buffer *rb;  in rb_alloc()  local
838 	rb = kzalloc_node(size, GFP_KERNEL, node);  in rb_alloc()
839 	if (!rb)  in rb_alloc()
842 	rb->user_page = perf_mmap_alloc_page(cpu);  in rb_alloc()
843 	if (!rb->user_page)  in rb_alloc()
847 		rb->data_pages[i] = perf_mmap_alloc_page(cpu);  in rb_alloc()
848 		if (!rb->data_pages[i])  in rb_alloc()
852 	rb->nr_pages = nr_pages;  in rb_alloc()
854 	ring_buffer_init(rb, watermark, flags);  in rb_alloc()
856 	return rb;  in rb_alloc()
860 		perf_mmap_free_page(rb->data_pages[i]);  in rb_alloc()
862 	perf_mmap_free_page(rb->user_page);  in rb_alloc()
865 	kfree(rb);  in rb_alloc()
871 void rb_free(struct perf_buffer *rb)  in rb_free()  argument
875 	perf_mmap_free_page(rb->user_page);  in rb_free()
876 	for (i = 0; i < rb->nr_pages; i++)  in rb_free()
877 		perf_mmap_free_page(rb->data_pages[i]);  in rb_free()
878 	kfree(rb);  in rb_free()
883 __perf_mmap_to_page(struct perf_buffer *rb, unsigned long pgoff)  in __perf_mmap_to_page()  argument
886 	if (pgoff > data_page_nr(rb))  in __perf_mmap_to_page()
889 	return vmalloc_to_page((void *)rb->user_page + pgoff * PAGE_SIZE);  in __perf_mmap_to_page()
894 	struct perf_buffer *rb;  in rb_free_work()  local
896 	rb = container_of(work, struct perf_buffer, work);  in rb_free_work()
898 	vfree(rb->user_page);  in rb_free_work()
899 	kfree(rb);  in rb_free_work()
902 void rb_free(struct perf_buffer *rb)  in rb_free()  argument
904 	schedule_work(&rb->work);  in rb_free()
909 	struct perf_buffer *rb;  in rb_alloc()  local
918 	rb = kzalloc_node(size, GFP_KERNEL, node);  in rb_alloc()
919 	if (!rb)  in rb_alloc()
922 	INIT_WORK(&rb->work, rb_free_work);  in rb_alloc()
928 	rb->user_page = all_buf;  in rb_alloc()
929 	rb->data_pages[0] = all_buf + PAGE_SIZE;  in rb_alloc()
931 		rb->nr_pages = 1;  in rb_alloc()
932 		rb->page_order = ilog2(nr_pages);  in rb_alloc()
935 	ring_buffer_init(rb, watermark, flags);  in rb_alloc()
937 	return rb;  in rb_alloc()
940 	kfree(rb);  in rb_alloc()
949 perf_mmap_to_page(struct perf_buffer *rb, unsigned long pgoff)  in perf_mmap_to_page()  argument
951 	if (rb->aux_nr_pages) {  in perf_mmap_to_page()
953 		if (pgoff > rb->aux_pgoff + rb->aux_nr_pages)  in perf_mmap_to_page()
957 		if (pgoff >= rb->aux_pgoff) {  in perf_mmap_to_page()
958 			int aux_pgoff = array_index_nospec(pgoff - rb->aux_pgoff, rb->aux_nr_pages);  in perf_mmap_to_page()
959 			return virt_to_page(rb->aux_pages[aux_pgoff]);  in perf_mmap_to_page()
963 	return __perf_mmap_to_page(rb, pgoff);  in perf_mmap_to_page()