22 	atomic_set(&handle->rb->poll, EPOLLIN);  in perf_output_wakeup()
38 	struct perf_buffer *rb = handle->rb;  in perf_output_get_handle()  local
46 	(*(volatile unsigned int *)&rb->nest)++;  in perf_output_get_handle()
47 	handle->wakeup = local_read(&rb->wakeup);  in perf_output_get_handle()
52 	struct perf_buffer *rb = handle->rb;  in perf_output_put_handle()  local
60 	nest = READ_ONCE(rb->nest);  in perf_output_put_handle()
62 		WRITE_ONCE(rb->nest, nest - 1);  in perf_output_put_handle()
76 	head = local_read(&rb->head);  in perf_output_put_handle()
110 	WRITE_ONCE(rb->user_page->data_head, head);  in perf_output_put_handle()
118 	WRITE_ONCE(rb->nest, 0);  in perf_output_put_handle()
125 	if (unlikely(head != local_read(&rb->head))) {  in perf_output_put_handle()
126 		WRITE_ONCE(rb->nest, 1);  in perf_output_put_handle()
130 	if (handle->wakeup != local_read(&rb->wakeup))  in perf_output_put_handle()
154 	struct perf_buffer *rb;  in __perf_output_begin()  local
170 	rb = rcu_dereference(event->rb);  in __perf_output_begin()
171 	if (unlikely(!rb))  in __perf_output_begin()
174 	if (unlikely(rb->paused)) {  in __perf_output_begin()
175 		if (rb->nr_pages) {  in __perf_output_begin()
176 			local_inc(&rb->lost);  in __perf_output_begin()
182 	handle->rb    = rb;  in __perf_output_begin()
185 	have_lost = local_read(&rb->lost);  in __perf_output_begin()
195 		tail = READ_ONCE(rb->user_page->data_tail);  in __perf_output_begin()
196 		offset = head = local_read(&rb->head);  in __perf_output_begin()
197 		if (!rb->overwrite) {  in __perf_output_begin()
199 							    perf_data_size(rb),  in __perf_output_begin()
220 	} while (local_cmpxchg(&rb->head, offset, head) != offset);  in __perf_output_begin()
232 	if (unlikely(head - local_read(&rb->wakeup) > rb->watermark))  in __perf_output_begin()
233 		local_add(rb->watermark, &rb->wakeup);  in __perf_output_begin()
235 	page_shift = PAGE_SHIFT + page_order(rb);  in __perf_output_begin()
237 	handle->page = (offset >> page_shift) & (rb->nr_pages - 1);  in __perf_output_begin()
239 	handle->addr = rb->data_pages[handle->page] + offset;  in __perf_output_begin()
247 		lost_event.lost        = local_xchg(&rb->lost, 0);  in __perf_output_begin()
258 	local_inc(&rb->lost);  in __perf_output_begin()
309 ring_buffer_init(struct perf_buffer *rb, long watermark, int flags)  in ring_buffer_init()  argument
311 	long max_size = perf_data_size(rb);  in ring_buffer_init()
314 		rb->watermark = min(max_size, watermark);  in ring_buffer_init()
316 	if (!rb->watermark)  in ring_buffer_init()
317 		rb->watermark = max_size / 2;  in ring_buffer_init()
320 		rb->overwrite = 0;  in ring_buffer_init()
322 		rb->overwrite = 1;  in ring_buffer_init()
324 	refcount_set(&rb->refcount, 1);  in ring_buffer_init()
326 	INIT_LIST_HEAD(&rb->event_list);  in ring_buffer_init()
327 	spin_lock_init(&rb->event_lock);  in ring_buffer_init()
333 	if (!rb->nr_pages)  in ring_buffer_init()
334 		rb->paused = 1;  in ring_buffer_init()
369 	struct perf_buffer *rb;  in perf_aux_output_begin()  local
380 	rb = ring_buffer_get(output_event);  in perf_aux_output_begin()
381 	if (!rb)  in perf_aux_output_begin()
384 	if (!rb_has_aux(rb))  in perf_aux_output_begin()
395 	if (!atomic_read(&rb->aux_mmap_count))  in perf_aux_output_begin()
398 	if (!refcount_inc_not_zero(&rb->aux_refcount))  in perf_aux_output_begin()
401 	nest = READ_ONCE(rb->aux_nest);  in perf_aux_output_begin()
409 	WRITE_ONCE(rb->aux_nest, nest + 1);  in perf_aux_output_begin()
411 	aux_head = rb->aux_head;  in perf_aux_output_begin()
413 	handle->rb = rb;  in perf_aux_output_begin()
424 	if (!rb->aux_overwrite) {  in perf_aux_output_begin()
425 		aux_tail = READ_ONCE(rb->user_page->aux_tail);  in perf_aux_output_begin()
426 		handle->wakeup = rb->aux_wakeup + rb->aux_watermark;  in perf_aux_output_begin()
427 		if (aux_head - aux_tail < perf_aux_size(rb))  in perf_aux_output_begin()
428 			handle->size = CIRC_SPACE(aux_head, aux_tail, perf_aux_size(rb));  in perf_aux_output_begin()
438 			WRITE_ONCE(rb->aux_nest, 0);  in perf_aux_output_begin()
443 	return handle->rb->aux_priv;  in perf_aux_output_begin()
447 	rb_free_aux(rb);  in perf_aux_output_begin()
450 	ring_buffer_put(rb);  in perf_aux_output_begin()
457 static __always_inline bool rb_need_aux_wakeup(struct perf_buffer *rb)  in rb_need_aux_wakeup()  argument
459 	if (rb->aux_overwrite)  in rb_need_aux_wakeup()
462 	if (rb->aux_head - rb->aux_wakeup >= rb->aux_watermark) {  in rb_need_aux_wakeup()
463 		rb->aux_wakeup = rounddown(rb->aux_head, rb->aux_watermark);  in rb_need_aux_wakeup()
483 	struct perf_buffer *rb = handle->rb;  in perf_aux_output_end()  local
487 	if (rb->aux_overwrite) {  in perf_aux_output_end()
491 		rb->aux_head = aux_head;  in perf_aux_output_end()
495 		aux_head = rb->aux_head;  in perf_aux_output_end()
496 		rb->aux_head += size;  in perf_aux_output_end()
515 	WRITE_ONCE(rb->user_page->aux_head, rb->aux_head);  in perf_aux_output_end()
516 	if (rb_need_aux_wakeup(rb))  in perf_aux_output_end()
527 	WRITE_ONCE(rb->aux_nest, 0);  in perf_aux_output_end()
529 	rb_free_aux(rb);  in perf_aux_output_end()
530 	ring_buffer_put(rb);  in perf_aux_output_end()
540 	struct perf_buffer *rb = handle->rb;  in perf_aux_output_skip()  local
545 	rb->aux_head += size;  in perf_aux_output_skip()
547 	WRITE_ONCE(rb->user_page->aux_head, rb->aux_head);  in perf_aux_output_skip()
548 	if (rb_need_aux_wakeup(rb)) {  in perf_aux_output_skip()
550 		handle->wakeup = rb->aux_wakeup + rb->aux_watermark;  in perf_aux_output_skip()
553 	handle->head = rb->aux_head;  in perf_aux_output_skip()
566 	return handle->rb->aux_priv;  in perf_get_aux()
577 	struct perf_buffer *rb = aux_handle->rb;  in perf_output_copy_aux()  local
581 	from &= (rb->aux_nr_pages << PAGE_SHIFT) - 1;  in perf_output_copy_aux()
582 	to &= (rb->aux_nr_pages << PAGE_SHIFT) - 1;  in perf_output_copy_aux()
591 		addr = rb->aux_pages[from >> PAGE_SHIFT];  in perf_output_copy_aux()
600 		from &= (rb->aux_nr_pages << PAGE_SHIFT) - 1;  in perf_output_copy_aux()
634 static void rb_free_aux_page(struct perf_buffer *rb, int idx)  in rb_free_aux_page()  argument
636 	struct page *page = virt_to_page(rb->aux_pages[idx]);  in rb_free_aux_page()
643 static void __rb_free_aux(struct perf_buffer *rb)  in __rb_free_aux()  argument
655 	if (rb->aux_priv) {  in __rb_free_aux()
656 		rb->free_aux(rb->aux_priv);  in __rb_free_aux()
657 		rb->free_aux = NULL;  in __rb_free_aux()
658 		rb->aux_priv = NULL;  in __rb_free_aux()
661 	if (rb->aux_nr_pages) {  in __rb_free_aux()
662 		for (pg = 0; pg < rb->aux_nr_pages; pg++)  in __rb_free_aux()
663 			rb_free_aux_page(rb, pg);  in __rb_free_aux()
665 		kfree(rb->aux_pages);  in __rb_free_aux()
666 		rb->aux_nr_pages = 0;  in __rb_free_aux()
670 int rb_alloc_aux(struct perf_buffer *rb, struct perf_event *event,  in rb_alloc_aux()  argument
702 	rb->aux_pages = kcalloc_node(nr_pages, sizeof(void *), GFP_KERNEL,  in rb_alloc_aux()
704 	if (!rb->aux_pages)  in rb_alloc_aux()
707 	rb->free_aux = event->pmu->free_aux;  in rb_alloc_aux()
708 	for (rb->aux_nr_pages = 0; rb->aux_nr_pages < nr_pages;) {  in rb_alloc_aux()
712 		order = min(max_order, ilog2(nr_pages - rb->aux_nr_pages));  in rb_alloc_aux()
717 		for (last = rb->aux_nr_pages + (1 << page_private(page));  in rb_alloc_aux()
718 		     last > rb->aux_nr_pages; rb->aux_nr_pages++)  in rb_alloc_aux()
719 			rb->aux_pages[rb->aux_nr_pages] = page_address(page++);  in rb_alloc_aux()
730 		struct page *page = virt_to_page(rb->aux_pages[0]);  in rb_alloc_aux()
736 	rb->aux_priv = event->pmu->setup_aux(event, rb->aux_pages, nr_pages,  in rb_alloc_aux()
738 	if (!rb->aux_priv)  in rb_alloc_aux()
749 	refcount_set(&rb->aux_refcount, 1);  in rb_alloc_aux()
751 	rb->aux_overwrite = overwrite;  in rb_alloc_aux()
752 	rb->aux_watermark = watermark;  in rb_alloc_aux()
756 		rb->aux_pgoff = pgoff;  in rb_alloc_aux()
758 		__rb_free_aux(rb);  in rb_alloc_aux()
763 void rb_free_aux(struct perf_buffer *rb)  in rb_free_aux()  argument
765 	if (refcount_dec_and_test(&rb->aux_refcount))  in rb_free_aux()
766 		__rb_free_aux(rb);  in rb_free_aux()
776 __perf_mmap_to_page(struct perf_buffer *rb, unsigned long pgoff)  in __perf_mmap_to_page()  argument
778 	if (pgoff > rb->nr_pages)  in __perf_mmap_to_page()
782 		return virt_to_page(rb->user_page);  in __perf_mmap_to_page()
784 	return virt_to_page(rb->data_pages[pgoff - 1]);  in __perf_mmap_to_page()
810 	struct perf_buffer *rb;  in rb_alloc()  local
821 	rb = kzalloc_node(size, GFP_KERNEL, node);  in rb_alloc()
822 	if (!rb)  in rb_alloc()
825 	rb->user_page = perf_mmap_alloc_page(cpu);  in rb_alloc()
826 	if (!rb->user_page)  in rb_alloc()
830 		rb->data_pages[i] = perf_mmap_alloc_page(cpu);  in rb_alloc()
831 		if (!rb->data_pages[i])  in rb_alloc()
835 	rb->nr_pages = nr_pages;  in rb_alloc()
837 	ring_buffer_init(rb, watermark, flags);  in rb_alloc()
839 	return rb;  in rb_alloc()
843 		perf_mmap_free_page(rb->data_pages[i]);  in rb_alloc()
845 	perf_mmap_free_page(rb->user_page);  in rb_alloc()
848 	kfree(rb);  in rb_alloc()
854 void rb_free(struct perf_buffer *rb)  in rb_free()  argument
858 	perf_mmap_free_page(rb->user_page);  in rb_free()
859 	for (i = 0; i < rb->nr_pages; i++)  in rb_free()
860 		perf_mmap_free_page(rb->data_pages[i]);  in rb_free()
861 	kfree(rb);  in rb_free()
866 __perf_mmap_to_page(struct perf_buffer *rb, unsigned long pgoff)  in __perf_mmap_to_page()  argument
869 	if (pgoff > data_page_nr(rb))  in __perf_mmap_to_page()
872 	return vmalloc_to_page((void *)rb->user_page + pgoff * PAGE_SIZE);  in __perf_mmap_to_page()
884 	struct perf_buffer *rb;  in rb_free_work()  local
888 	rb = container_of(work, struct perf_buffer, work);  in rb_free_work()
889 	nr = data_page_nr(rb);  in rb_free_work()
891 	base = rb->user_page;  in rb_free_work()
897 	kfree(rb);  in rb_free_work()
900 void rb_free(struct perf_buffer *rb)  in rb_free()  argument
902 	schedule_work(&rb->work);  in rb_free()
907 	struct perf_buffer *rb;  in rb_alloc()  local
916 	rb = kzalloc_node(size, GFP_KERNEL, node);  in rb_alloc()
917 	if (!rb)  in rb_alloc()
920 	INIT_WORK(&rb->work, rb_free_work);  in rb_alloc()
926 	rb->user_page = all_buf;  in rb_alloc()
927 	rb->data_pages[0] = all_buf + PAGE_SIZE;  in rb_alloc()
929 		rb->nr_pages = 1;  in rb_alloc()
930 		rb->page_order = ilog2(nr_pages);  in rb_alloc()
933 	ring_buffer_init(rb, watermark, flags);  in rb_alloc()
935 	return rb;  in rb_alloc()
938 	kfree(rb);  in rb_alloc()
947 perf_mmap_to_page(struct perf_buffer *rb, unsigned long pgoff)  in perf_mmap_to_page()  argument
949 	if (rb->aux_nr_pages) {  in perf_mmap_to_page()
951 		if (pgoff > rb->aux_pgoff + rb->aux_nr_pages)  in perf_mmap_to_page()
955 		if (pgoff >= rb->aux_pgoff) {  in perf_mmap_to_page()
956 			int aux_pgoff = array_index_nospec(pgoff - rb->aux_pgoff, rb->aux_nr_pages);  in perf_mmap_to_page()
957 			return virt_to_page(rb->aux_pages[aux_pgoff]);  in perf_mmap_to_page()
961 	return __perf_mmap_to_page(rb, pgoff);  in perf_mmap_to_page()