18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
28c2ecf20Sopenharmony_ci/*
38c2ecf20Sopenharmony_ci * Copyright (C) 2016 CNEX Labs
48c2ecf20Sopenharmony_ci * Initial release: Javier Gonzalez <javier@cnexlabs.com>
58c2ecf20Sopenharmony_ci *
68c2ecf20Sopenharmony_ci * Based upon the circular ringbuffer.
78c2ecf20Sopenharmony_ci *
88c2ecf20Sopenharmony_ci * This program is free software; you can redistribute it and/or
98c2ecf20Sopenharmony_ci * modify it under the terms of the GNU General Public License version
108c2ecf20Sopenharmony_ci * 2 as published by the Free Software Foundation.
118c2ecf20Sopenharmony_ci *
128c2ecf20Sopenharmony_ci * This program is distributed in the hope that it will be useful, but
138c2ecf20Sopenharmony_ci * WITHOUT ANY WARRANTY; without even the implied warranty of
148c2ecf20Sopenharmony_ci * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
158c2ecf20Sopenharmony_ci * General Public License for more details.
168c2ecf20Sopenharmony_ci *
178c2ecf20Sopenharmony_ci * pblk-rb.c - pblk's write buffer
188c2ecf20Sopenharmony_ci */
198c2ecf20Sopenharmony_ci
208c2ecf20Sopenharmony_ci#include <linux/circ_buf.h>
218c2ecf20Sopenharmony_ci
228c2ecf20Sopenharmony_ci#include "pblk.h"
238c2ecf20Sopenharmony_ci
248c2ecf20Sopenharmony_cistatic DECLARE_RWSEM(pblk_rb_lock);
258c2ecf20Sopenharmony_ci
268c2ecf20Sopenharmony_cistatic void pblk_rb_data_free(struct pblk_rb *rb)
278c2ecf20Sopenharmony_ci{
288c2ecf20Sopenharmony_ci	struct pblk_rb_pages *p, *t;
298c2ecf20Sopenharmony_ci
308c2ecf20Sopenharmony_ci	down_write(&pblk_rb_lock);
318c2ecf20Sopenharmony_ci	list_for_each_entry_safe(p, t, &rb->pages, list) {
328c2ecf20Sopenharmony_ci		free_pages((unsigned long)page_address(p->pages), p->order);
338c2ecf20Sopenharmony_ci		list_del(&p->list);
348c2ecf20Sopenharmony_ci		kfree(p);
358c2ecf20Sopenharmony_ci	}
368c2ecf20Sopenharmony_ci	up_write(&pblk_rb_lock);
378c2ecf20Sopenharmony_ci}
388c2ecf20Sopenharmony_ci
398c2ecf20Sopenharmony_civoid pblk_rb_free(struct pblk_rb *rb)
408c2ecf20Sopenharmony_ci{
418c2ecf20Sopenharmony_ci	pblk_rb_data_free(rb);
428c2ecf20Sopenharmony_ci	vfree(rb->entries);
438c2ecf20Sopenharmony_ci}
448c2ecf20Sopenharmony_ci
458c2ecf20Sopenharmony_ci/*
468c2ecf20Sopenharmony_ci * pblk_rb_calculate_size -- calculate the size of the write buffer
478c2ecf20Sopenharmony_ci */
488c2ecf20Sopenharmony_cistatic unsigned int pblk_rb_calculate_size(unsigned int nr_entries,
498c2ecf20Sopenharmony_ci					   unsigned int threshold)
508c2ecf20Sopenharmony_ci{
518c2ecf20Sopenharmony_ci	unsigned int thr_sz = 1 << (get_count_order(threshold + NVM_MAX_VLBA));
528c2ecf20Sopenharmony_ci	unsigned int max_sz = max(thr_sz, nr_entries);
538c2ecf20Sopenharmony_ci	unsigned int max_io;
548c2ecf20Sopenharmony_ci
558c2ecf20Sopenharmony_ci	/* Alloc a write buffer that can (i) fit at least two split bios
568c2ecf20Sopenharmony_ci	 * (considering max I/O size NVM_MAX_VLBA, and (ii) guarantee that the
578c2ecf20Sopenharmony_ci	 * threshold will be respected
588c2ecf20Sopenharmony_ci	 */
598c2ecf20Sopenharmony_ci	max_io = (1 << max((int)(get_count_order(max_sz)),
608c2ecf20Sopenharmony_ci				(int)(get_count_order(NVM_MAX_VLBA << 1))));
618c2ecf20Sopenharmony_ci	if ((threshold + NVM_MAX_VLBA) >= max_io)
628c2ecf20Sopenharmony_ci		max_io <<= 1;
638c2ecf20Sopenharmony_ci
648c2ecf20Sopenharmony_ci	return max_io;
658c2ecf20Sopenharmony_ci}
668c2ecf20Sopenharmony_ci
678c2ecf20Sopenharmony_ci/*
688c2ecf20Sopenharmony_ci * Initialize ring buffer. The data and metadata buffers must be previously
698c2ecf20Sopenharmony_ci * allocated and their size must be a power of two
708c2ecf20Sopenharmony_ci * (Documentation/core-api/circular-buffers.rst)
718c2ecf20Sopenharmony_ci */
728c2ecf20Sopenharmony_ciint pblk_rb_init(struct pblk_rb *rb, unsigned int size, unsigned int threshold,
738c2ecf20Sopenharmony_ci		 unsigned int seg_size)
748c2ecf20Sopenharmony_ci{
758c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
768c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entries;
778c2ecf20Sopenharmony_ci	unsigned int init_entry = 0;
788c2ecf20Sopenharmony_ci	unsigned int max_order = MAX_ORDER - 1;
798c2ecf20Sopenharmony_ci	unsigned int power_size, power_seg_sz;
808c2ecf20Sopenharmony_ci	unsigned int alloc_order, order, iter;
818c2ecf20Sopenharmony_ci	unsigned int nr_entries;
828c2ecf20Sopenharmony_ci
838c2ecf20Sopenharmony_ci	nr_entries = pblk_rb_calculate_size(size, threshold);
848c2ecf20Sopenharmony_ci	entries = vzalloc(array_size(nr_entries, sizeof(struct pblk_rb_entry)));
858c2ecf20Sopenharmony_ci	if (!entries)
868c2ecf20Sopenharmony_ci		return -ENOMEM;
878c2ecf20Sopenharmony_ci
888c2ecf20Sopenharmony_ci	power_size = get_count_order(nr_entries);
898c2ecf20Sopenharmony_ci	power_seg_sz = get_count_order(seg_size);
908c2ecf20Sopenharmony_ci
918c2ecf20Sopenharmony_ci	down_write(&pblk_rb_lock);
928c2ecf20Sopenharmony_ci	rb->entries = entries;
938c2ecf20Sopenharmony_ci	rb->seg_size = (1 << power_seg_sz);
948c2ecf20Sopenharmony_ci	rb->nr_entries = (1 << power_size);
958c2ecf20Sopenharmony_ci	rb->mem = rb->subm = rb->sync = rb->l2p_update = 0;
968c2ecf20Sopenharmony_ci	rb->back_thres = threshold;
978c2ecf20Sopenharmony_ci	rb->flush_point = EMPTY_ENTRY;
988c2ecf20Sopenharmony_ci
998c2ecf20Sopenharmony_ci	spin_lock_init(&rb->w_lock);
1008c2ecf20Sopenharmony_ci	spin_lock_init(&rb->s_lock);
1018c2ecf20Sopenharmony_ci
1028c2ecf20Sopenharmony_ci	INIT_LIST_HEAD(&rb->pages);
1038c2ecf20Sopenharmony_ci
1048c2ecf20Sopenharmony_ci	alloc_order = power_size;
1058c2ecf20Sopenharmony_ci	if (alloc_order >= max_order) {
1068c2ecf20Sopenharmony_ci		order = max_order;
1078c2ecf20Sopenharmony_ci		iter = (1 << (alloc_order - max_order));
1088c2ecf20Sopenharmony_ci	} else {
1098c2ecf20Sopenharmony_ci		order = alloc_order;
1108c2ecf20Sopenharmony_ci		iter = 1;
1118c2ecf20Sopenharmony_ci	}
1128c2ecf20Sopenharmony_ci
1138c2ecf20Sopenharmony_ci	do {
1148c2ecf20Sopenharmony_ci		struct pblk_rb_entry *entry;
1158c2ecf20Sopenharmony_ci		struct pblk_rb_pages *page_set;
1168c2ecf20Sopenharmony_ci		void *kaddr;
1178c2ecf20Sopenharmony_ci		unsigned long set_size;
1188c2ecf20Sopenharmony_ci		int i;
1198c2ecf20Sopenharmony_ci
1208c2ecf20Sopenharmony_ci		page_set = kmalloc(sizeof(struct pblk_rb_pages), GFP_KERNEL);
1218c2ecf20Sopenharmony_ci		if (!page_set) {
1228c2ecf20Sopenharmony_ci			up_write(&pblk_rb_lock);
1238c2ecf20Sopenharmony_ci			vfree(entries);
1248c2ecf20Sopenharmony_ci			return -ENOMEM;
1258c2ecf20Sopenharmony_ci		}
1268c2ecf20Sopenharmony_ci
1278c2ecf20Sopenharmony_ci		page_set->order = order;
1288c2ecf20Sopenharmony_ci		page_set->pages = alloc_pages(GFP_KERNEL, order);
1298c2ecf20Sopenharmony_ci		if (!page_set->pages) {
1308c2ecf20Sopenharmony_ci			kfree(page_set);
1318c2ecf20Sopenharmony_ci			pblk_rb_data_free(rb);
1328c2ecf20Sopenharmony_ci			up_write(&pblk_rb_lock);
1338c2ecf20Sopenharmony_ci			vfree(entries);
1348c2ecf20Sopenharmony_ci			return -ENOMEM;
1358c2ecf20Sopenharmony_ci		}
1368c2ecf20Sopenharmony_ci		kaddr = page_address(page_set->pages);
1378c2ecf20Sopenharmony_ci
1388c2ecf20Sopenharmony_ci		entry = &rb->entries[init_entry];
1398c2ecf20Sopenharmony_ci		entry->data = kaddr;
1408c2ecf20Sopenharmony_ci		entry->cacheline = pblk_cacheline_to_addr(init_entry++);
1418c2ecf20Sopenharmony_ci		entry->w_ctx.flags = PBLK_WRITABLE_ENTRY;
1428c2ecf20Sopenharmony_ci
1438c2ecf20Sopenharmony_ci		set_size = (1 << order);
1448c2ecf20Sopenharmony_ci		for (i = 1; i < set_size; i++) {
1458c2ecf20Sopenharmony_ci			entry = &rb->entries[init_entry];
1468c2ecf20Sopenharmony_ci			entry->cacheline = pblk_cacheline_to_addr(init_entry++);
1478c2ecf20Sopenharmony_ci			entry->data = kaddr + (i * rb->seg_size);
1488c2ecf20Sopenharmony_ci			entry->w_ctx.flags = PBLK_WRITABLE_ENTRY;
1498c2ecf20Sopenharmony_ci			bio_list_init(&entry->w_ctx.bios);
1508c2ecf20Sopenharmony_ci		}
1518c2ecf20Sopenharmony_ci
1528c2ecf20Sopenharmony_ci		list_add_tail(&page_set->list, &rb->pages);
1538c2ecf20Sopenharmony_ci		iter--;
1548c2ecf20Sopenharmony_ci	} while (iter > 0);
1558c2ecf20Sopenharmony_ci	up_write(&pblk_rb_lock);
1568c2ecf20Sopenharmony_ci
1578c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
1588c2ecf20Sopenharmony_ci	atomic_set(&rb->inflight_flush_point, 0);
1598c2ecf20Sopenharmony_ci#endif
1608c2ecf20Sopenharmony_ci
1618c2ecf20Sopenharmony_ci	/*
1628c2ecf20Sopenharmony_ci	 * Initialize rate-limiter, which controls access to the write buffer
1638c2ecf20Sopenharmony_ci	 * by user and GC I/O
1648c2ecf20Sopenharmony_ci	 */
1658c2ecf20Sopenharmony_ci	pblk_rl_init(&pblk->rl, rb->nr_entries, threshold);
1668c2ecf20Sopenharmony_ci
1678c2ecf20Sopenharmony_ci	return 0;
1688c2ecf20Sopenharmony_ci}
1698c2ecf20Sopenharmony_ci
1708c2ecf20Sopenharmony_cistatic void clean_wctx(struct pblk_w_ctx *w_ctx)
1718c2ecf20Sopenharmony_ci{
1728c2ecf20Sopenharmony_ci	int flags;
1738c2ecf20Sopenharmony_ci
1748c2ecf20Sopenharmony_ci	flags = READ_ONCE(w_ctx->flags);
1758c2ecf20Sopenharmony_ci	WARN_ONCE(!(flags & PBLK_SUBMITTED_ENTRY),
1768c2ecf20Sopenharmony_ci			"pblk: overwriting unsubmitted data\n");
1778c2ecf20Sopenharmony_ci
1788c2ecf20Sopenharmony_ci	/* Release flags on context. Protect from writes and reads */
1798c2ecf20Sopenharmony_ci	smp_store_release(&w_ctx->flags, PBLK_WRITABLE_ENTRY);
1808c2ecf20Sopenharmony_ci	pblk_ppa_set_empty(&w_ctx->ppa);
1818c2ecf20Sopenharmony_ci	w_ctx->lba = ADDR_EMPTY;
1828c2ecf20Sopenharmony_ci}
1838c2ecf20Sopenharmony_ci
1848c2ecf20Sopenharmony_ci#define pblk_rb_ring_count(head, tail, size) CIRC_CNT(head, tail, size)
1858c2ecf20Sopenharmony_ci#define pblk_rb_ring_space(rb, head, tail, size) \
1868c2ecf20Sopenharmony_ci					(CIRC_SPACE(head, tail, size))
1878c2ecf20Sopenharmony_ci
1888c2ecf20Sopenharmony_ci/*
1898c2ecf20Sopenharmony_ci * Buffer space is calculated with respect to the back pointer signaling
1908c2ecf20Sopenharmony_ci * synchronized entries to the media.
1918c2ecf20Sopenharmony_ci */
1928c2ecf20Sopenharmony_cistatic unsigned int pblk_rb_space(struct pblk_rb *rb)
1938c2ecf20Sopenharmony_ci{
1948c2ecf20Sopenharmony_ci	unsigned int mem = READ_ONCE(rb->mem);
1958c2ecf20Sopenharmony_ci	unsigned int sync = READ_ONCE(rb->sync);
1968c2ecf20Sopenharmony_ci
1978c2ecf20Sopenharmony_ci	return pblk_rb_ring_space(rb, mem, sync, rb->nr_entries);
1988c2ecf20Sopenharmony_ci}
1998c2ecf20Sopenharmony_ci
2008c2ecf20Sopenharmony_ciunsigned int pblk_rb_ptr_wrap(struct pblk_rb *rb, unsigned int p,
2018c2ecf20Sopenharmony_ci			      unsigned int nr_entries)
2028c2ecf20Sopenharmony_ci{
2038c2ecf20Sopenharmony_ci	return (p + nr_entries) & (rb->nr_entries - 1);
2048c2ecf20Sopenharmony_ci}
2058c2ecf20Sopenharmony_ci
2068c2ecf20Sopenharmony_ci/*
2078c2ecf20Sopenharmony_ci * Buffer count is calculated with respect to the submission entry signaling the
2088c2ecf20Sopenharmony_ci * entries that are available to send to the media
2098c2ecf20Sopenharmony_ci */
2108c2ecf20Sopenharmony_ciunsigned int pblk_rb_read_count(struct pblk_rb *rb)
2118c2ecf20Sopenharmony_ci{
2128c2ecf20Sopenharmony_ci	unsigned int mem = READ_ONCE(rb->mem);
2138c2ecf20Sopenharmony_ci	unsigned int subm = READ_ONCE(rb->subm);
2148c2ecf20Sopenharmony_ci
2158c2ecf20Sopenharmony_ci	return pblk_rb_ring_count(mem, subm, rb->nr_entries);
2168c2ecf20Sopenharmony_ci}
2178c2ecf20Sopenharmony_ci
2188c2ecf20Sopenharmony_ciunsigned int pblk_rb_sync_count(struct pblk_rb *rb)
2198c2ecf20Sopenharmony_ci{
2208c2ecf20Sopenharmony_ci	unsigned int mem = READ_ONCE(rb->mem);
2218c2ecf20Sopenharmony_ci	unsigned int sync = READ_ONCE(rb->sync);
2228c2ecf20Sopenharmony_ci
2238c2ecf20Sopenharmony_ci	return pblk_rb_ring_count(mem, sync, rb->nr_entries);
2248c2ecf20Sopenharmony_ci}
2258c2ecf20Sopenharmony_ci
2268c2ecf20Sopenharmony_ciunsigned int pblk_rb_read_commit(struct pblk_rb *rb, unsigned int nr_entries)
2278c2ecf20Sopenharmony_ci{
2288c2ecf20Sopenharmony_ci	unsigned int subm;
2298c2ecf20Sopenharmony_ci
2308c2ecf20Sopenharmony_ci	subm = READ_ONCE(rb->subm);
2318c2ecf20Sopenharmony_ci	/* Commit read means updating submission pointer */
2328c2ecf20Sopenharmony_ci	smp_store_release(&rb->subm, pblk_rb_ptr_wrap(rb, subm, nr_entries));
2338c2ecf20Sopenharmony_ci
2348c2ecf20Sopenharmony_ci	return subm;
2358c2ecf20Sopenharmony_ci}
2368c2ecf20Sopenharmony_ci
2378c2ecf20Sopenharmony_cistatic int __pblk_rb_update_l2p(struct pblk_rb *rb, unsigned int to_update)
2388c2ecf20Sopenharmony_ci{
2398c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
2408c2ecf20Sopenharmony_ci	struct pblk_line *line;
2418c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
2428c2ecf20Sopenharmony_ci	struct pblk_w_ctx *w_ctx;
2438c2ecf20Sopenharmony_ci	unsigned int user_io = 0, gc_io = 0;
2448c2ecf20Sopenharmony_ci	unsigned int i;
2458c2ecf20Sopenharmony_ci	int flags;
2468c2ecf20Sopenharmony_ci
2478c2ecf20Sopenharmony_ci	for (i = 0; i < to_update; i++) {
2488c2ecf20Sopenharmony_ci		entry = &rb->entries[rb->l2p_update];
2498c2ecf20Sopenharmony_ci		w_ctx = &entry->w_ctx;
2508c2ecf20Sopenharmony_ci
2518c2ecf20Sopenharmony_ci		flags = READ_ONCE(entry->w_ctx.flags);
2528c2ecf20Sopenharmony_ci		if (flags & PBLK_IOTYPE_USER)
2538c2ecf20Sopenharmony_ci			user_io++;
2548c2ecf20Sopenharmony_ci		else if (flags & PBLK_IOTYPE_GC)
2558c2ecf20Sopenharmony_ci			gc_io++;
2568c2ecf20Sopenharmony_ci		else
2578c2ecf20Sopenharmony_ci			WARN(1, "pblk: unknown IO type\n");
2588c2ecf20Sopenharmony_ci
2598c2ecf20Sopenharmony_ci		pblk_update_map_dev(pblk, w_ctx->lba, w_ctx->ppa,
2608c2ecf20Sopenharmony_ci							entry->cacheline);
2618c2ecf20Sopenharmony_ci
2628c2ecf20Sopenharmony_ci		line = pblk_ppa_to_line(pblk, w_ctx->ppa);
2638c2ecf20Sopenharmony_ci		atomic_dec(&line->sec_to_update);
2648c2ecf20Sopenharmony_ci		kref_put(&line->ref, pblk_line_put);
2658c2ecf20Sopenharmony_ci		clean_wctx(w_ctx);
2668c2ecf20Sopenharmony_ci		rb->l2p_update = pblk_rb_ptr_wrap(rb, rb->l2p_update, 1);
2678c2ecf20Sopenharmony_ci	}
2688c2ecf20Sopenharmony_ci
2698c2ecf20Sopenharmony_ci	pblk_rl_out(&pblk->rl, user_io, gc_io);
2708c2ecf20Sopenharmony_ci
2718c2ecf20Sopenharmony_ci	return 0;
2728c2ecf20Sopenharmony_ci}
2738c2ecf20Sopenharmony_ci
2748c2ecf20Sopenharmony_ci/*
2758c2ecf20Sopenharmony_ci * When we move the l2p_update pointer, we update the l2p table - lookups will
2768c2ecf20Sopenharmony_ci * point to the physical address instead of to the cacheline in the write buffer
2778c2ecf20Sopenharmony_ci * from this moment on.
2788c2ecf20Sopenharmony_ci */
2798c2ecf20Sopenharmony_cistatic int pblk_rb_update_l2p(struct pblk_rb *rb, unsigned int nr_entries,
2808c2ecf20Sopenharmony_ci			      unsigned int mem, unsigned int sync)
2818c2ecf20Sopenharmony_ci{
2828c2ecf20Sopenharmony_ci	unsigned int space, count;
2838c2ecf20Sopenharmony_ci	int ret = 0;
2848c2ecf20Sopenharmony_ci
2858c2ecf20Sopenharmony_ci	lockdep_assert_held(&rb->w_lock);
2868c2ecf20Sopenharmony_ci
2878c2ecf20Sopenharmony_ci	/* Update l2p only as buffer entries are being overwritten */
2888c2ecf20Sopenharmony_ci	space = pblk_rb_ring_space(rb, mem, rb->l2p_update, rb->nr_entries);
2898c2ecf20Sopenharmony_ci	if (space > nr_entries)
2908c2ecf20Sopenharmony_ci		goto out;
2918c2ecf20Sopenharmony_ci
2928c2ecf20Sopenharmony_ci	count = nr_entries - space;
2938c2ecf20Sopenharmony_ci	/* l2p_update used exclusively under rb->w_lock */
2948c2ecf20Sopenharmony_ci	ret = __pblk_rb_update_l2p(rb, count);
2958c2ecf20Sopenharmony_ci
2968c2ecf20Sopenharmony_ciout:
2978c2ecf20Sopenharmony_ci	return ret;
2988c2ecf20Sopenharmony_ci}
2998c2ecf20Sopenharmony_ci
3008c2ecf20Sopenharmony_ci/*
3018c2ecf20Sopenharmony_ci * Update the l2p entry for all sectors stored on the write buffer. This means
3028c2ecf20Sopenharmony_ci * that all future lookups to the l2p table will point to a device address, not
3038c2ecf20Sopenharmony_ci * to the cacheline in the write buffer.
3048c2ecf20Sopenharmony_ci */
3058c2ecf20Sopenharmony_civoid pblk_rb_sync_l2p(struct pblk_rb *rb)
3068c2ecf20Sopenharmony_ci{
3078c2ecf20Sopenharmony_ci	unsigned int sync;
3088c2ecf20Sopenharmony_ci	unsigned int to_update;
3098c2ecf20Sopenharmony_ci
3108c2ecf20Sopenharmony_ci	spin_lock(&rb->w_lock);
3118c2ecf20Sopenharmony_ci
3128c2ecf20Sopenharmony_ci	/* Protect from reads and writes */
3138c2ecf20Sopenharmony_ci	sync = smp_load_acquire(&rb->sync);
3148c2ecf20Sopenharmony_ci
3158c2ecf20Sopenharmony_ci	to_update = pblk_rb_ring_count(sync, rb->l2p_update, rb->nr_entries);
3168c2ecf20Sopenharmony_ci	__pblk_rb_update_l2p(rb, to_update);
3178c2ecf20Sopenharmony_ci
3188c2ecf20Sopenharmony_ci	spin_unlock(&rb->w_lock);
3198c2ecf20Sopenharmony_ci}
3208c2ecf20Sopenharmony_ci
3218c2ecf20Sopenharmony_ci/*
3228c2ecf20Sopenharmony_ci * Write @nr_entries to ring buffer from @data buffer if there is enough space.
3238c2ecf20Sopenharmony_ci * Typically, 4KB data chunks coming from a bio will be copied to the ring
3248c2ecf20Sopenharmony_ci * buffer, thus the write will fail if not all incoming data can be copied.
3258c2ecf20Sopenharmony_ci *
3268c2ecf20Sopenharmony_ci */
3278c2ecf20Sopenharmony_cistatic void __pblk_rb_write_entry(struct pblk_rb *rb, void *data,
3288c2ecf20Sopenharmony_ci				  struct pblk_w_ctx w_ctx,
3298c2ecf20Sopenharmony_ci				  struct pblk_rb_entry *entry)
3308c2ecf20Sopenharmony_ci{
3318c2ecf20Sopenharmony_ci	memcpy(entry->data, data, rb->seg_size);
3328c2ecf20Sopenharmony_ci
3338c2ecf20Sopenharmony_ci	entry->w_ctx.lba = w_ctx.lba;
3348c2ecf20Sopenharmony_ci	entry->w_ctx.ppa = w_ctx.ppa;
3358c2ecf20Sopenharmony_ci}
3368c2ecf20Sopenharmony_ci
3378c2ecf20Sopenharmony_civoid pblk_rb_write_entry_user(struct pblk_rb *rb, void *data,
3388c2ecf20Sopenharmony_ci			      struct pblk_w_ctx w_ctx, unsigned int ring_pos)
3398c2ecf20Sopenharmony_ci{
3408c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
3418c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
3428c2ecf20Sopenharmony_ci	int flags;
3438c2ecf20Sopenharmony_ci
3448c2ecf20Sopenharmony_ci	entry = &rb->entries[ring_pos];
3458c2ecf20Sopenharmony_ci	flags = READ_ONCE(entry->w_ctx.flags);
3468c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
3478c2ecf20Sopenharmony_ci	/* Caller must guarantee that the entry is free */
3488c2ecf20Sopenharmony_ci	BUG_ON(!(flags & PBLK_WRITABLE_ENTRY));
3498c2ecf20Sopenharmony_ci#endif
3508c2ecf20Sopenharmony_ci
3518c2ecf20Sopenharmony_ci	__pblk_rb_write_entry(rb, data, w_ctx, entry);
3528c2ecf20Sopenharmony_ci
3538c2ecf20Sopenharmony_ci	pblk_update_map_cache(pblk, w_ctx.lba, entry->cacheline);
3548c2ecf20Sopenharmony_ci	flags = w_ctx.flags | PBLK_WRITTEN_DATA;
3558c2ecf20Sopenharmony_ci
3568c2ecf20Sopenharmony_ci	/* Release flags on write context. Protect from writes */
3578c2ecf20Sopenharmony_ci	smp_store_release(&entry->w_ctx.flags, flags);
3588c2ecf20Sopenharmony_ci}
3598c2ecf20Sopenharmony_ci
3608c2ecf20Sopenharmony_civoid pblk_rb_write_entry_gc(struct pblk_rb *rb, void *data,
3618c2ecf20Sopenharmony_ci			    struct pblk_w_ctx w_ctx, struct pblk_line *line,
3628c2ecf20Sopenharmony_ci			    u64 paddr, unsigned int ring_pos)
3638c2ecf20Sopenharmony_ci{
3648c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
3658c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
3668c2ecf20Sopenharmony_ci	int flags;
3678c2ecf20Sopenharmony_ci
3688c2ecf20Sopenharmony_ci	entry = &rb->entries[ring_pos];
3698c2ecf20Sopenharmony_ci	flags = READ_ONCE(entry->w_ctx.flags);
3708c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
3718c2ecf20Sopenharmony_ci	/* Caller must guarantee that the entry is free */
3728c2ecf20Sopenharmony_ci	BUG_ON(!(flags & PBLK_WRITABLE_ENTRY));
3738c2ecf20Sopenharmony_ci#endif
3748c2ecf20Sopenharmony_ci
3758c2ecf20Sopenharmony_ci	__pblk_rb_write_entry(rb, data, w_ctx, entry);
3768c2ecf20Sopenharmony_ci
3778c2ecf20Sopenharmony_ci	if (!pblk_update_map_gc(pblk, w_ctx.lba, entry->cacheline, line, paddr))
3788c2ecf20Sopenharmony_ci		entry->w_ctx.lba = ADDR_EMPTY;
3798c2ecf20Sopenharmony_ci
3808c2ecf20Sopenharmony_ci	flags = w_ctx.flags | PBLK_WRITTEN_DATA;
3818c2ecf20Sopenharmony_ci
3828c2ecf20Sopenharmony_ci	/* Release flags on write context. Protect from writes */
3838c2ecf20Sopenharmony_ci	smp_store_release(&entry->w_ctx.flags, flags);
3848c2ecf20Sopenharmony_ci}
3858c2ecf20Sopenharmony_ci
3868c2ecf20Sopenharmony_cistatic int pblk_rb_flush_point_set(struct pblk_rb *rb, struct bio *bio,
3878c2ecf20Sopenharmony_ci				   unsigned int pos)
3888c2ecf20Sopenharmony_ci{
3898c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
3908c2ecf20Sopenharmony_ci	unsigned int sync, flush_point;
3918c2ecf20Sopenharmony_ci
3928c2ecf20Sopenharmony_ci	pblk_rb_sync_init(rb, NULL);
3938c2ecf20Sopenharmony_ci	sync = READ_ONCE(rb->sync);
3948c2ecf20Sopenharmony_ci
3958c2ecf20Sopenharmony_ci	if (pos == sync) {
3968c2ecf20Sopenharmony_ci		pblk_rb_sync_end(rb, NULL);
3978c2ecf20Sopenharmony_ci		return 0;
3988c2ecf20Sopenharmony_ci	}
3998c2ecf20Sopenharmony_ci
4008c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
4018c2ecf20Sopenharmony_ci	atomic_inc(&rb->inflight_flush_point);
4028c2ecf20Sopenharmony_ci#endif
4038c2ecf20Sopenharmony_ci
4048c2ecf20Sopenharmony_ci	flush_point = (pos == 0) ? (rb->nr_entries - 1) : (pos - 1);
4058c2ecf20Sopenharmony_ci	entry = &rb->entries[flush_point];
4068c2ecf20Sopenharmony_ci
4078c2ecf20Sopenharmony_ci	/* Protect flush points */
4088c2ecf20Sopenharmony_ci	smp_store_release(&rb->flush_point, flush_point);
4098c2ecf20Sopenharmony_ci
4108c2ecf20Sopenharmony_ci	if (bio)
4118c2ecf20Sopenharmony_ci		bio_list_add(&entry->w_ctx.bios, bio);
4128c2ecf20Sopenharmony_ci
4138c2ecf20Sopenharmony_ci	pblk_rb_sync_end(rb, NULL);
4148c2ecf20Sopenharmony_ci
4158c2ecf20Sopenharmony_ci	return bio ? 1 : 0;
4168c2ecf20Sopenharmony_ci}
4178c2ecf20Sopenharmony_ci
4188c2ecf20Sopenharmony_cistatic int __pblk_rb_may_write(struct pblk_rb *rb, unsigned int nr_entries,
4198c2ecf20Sopenharmony_ci			       unsigned int *pos)
4208c2ecf20Sopenharmony_ci{
4218c2ecf20Sopenharmony_ci	unsigned int mem;
4228c2ecf20Sopenharmony_ci	unsigned int sync;
4238c2ecf20Sopenharmony_ci	unsigned int threshold;
4248c2ecf20Sopenharmony_ci
4258c2ecf20Sopenharmony_ci	sync = READ_ONCE(rb->sync);
4268c2ecf20Sopenharmony_ci	mem = READ_ONCE(rb->mem);
4278c2ecf20Sopenharmony_ci
4288c2ecf20Sopenharmony_ci	threshold = nr_entries + rb->back_thres;
4298c2ecf20Sopenharmony_ci
4308c2ecf20Sopenharmony_ci	if (pblk_rb_ring_space(rb, mem, sync, rb->nr_entries) < threshold)
4318c2ecf20Sopenharmony_ci		return 0;
4328c2ecf20Sopenharmony_ci
4338c2ecf20Sopenharmony_ci	if (pblk_rb_update_l2p(rb, nr_entries, mem, sync))
4348c2ecf20Sopenharmony_ci		return 0;
4358c2ecf20Sopenharmony_ci
4368c2ecf20Sopenharmony_ci	*pos = mem;
4378c2ecf20Sopenharmony_ci
4388c2ecf20Sopenharmony_ci	return 1;
4398c2ecf20Sopenharmony_ci}
4408c2ecf20Sopenharmony_ci
4418c2ecf20Sopenharmony_cistatic int pblk_rb_may_write(struct pblk_rb *rb, unsigned int nr_entries,
4428c2ecf20Sopenharmony_ci			     unsigned int *pos)
4438c2ecf20Sopenharmony_ci{
4448c2ecf20Sopenharmony_ci	if (!__pblk_rb_may_write(rb, nr_entries, pos))
4458c2ecf20Sopenharmony_ci		return 0;
4468c2ecf20Sopenharmony_ci
4478c2ecf20Sopenharmony_ci	/* Protect from read count */
4488c2ecf20Sopenharmony_ci	smp_store_release(&rb->mem, pblk_rb_ptr_wrap(rb, *pos, nr_entries));
4498c2ecf20Sopenharmony_ci	return 1;
4508c2ecf20Sopenharmony_ci}
4518c2ecf20Sopenharmony_ci
4528c2ecf20Sopenharmony_civoid pblk_rb_flush(struct pblk_rb *rb)
4538c2ecf20Sopenharmony_ci{
4548c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
4558c2ecf20Sopenharmony_ci	unsigned int mem = READ_ONCE(rb->mem);
4568c2ecf20Sopenharmony_ci
4578c2ecf20Sopenharmony_ci	if (pblk_rb_flush_point_set(rb, NULL, mem))
4588c2ecf20Sopenharmony_ci		return;
4598c2ecf20Sopenharmony_ci
4608c2ecf20Sopenharmony_ci	pblk_write_kick(pblk);
4618c2ecf20Sopenharmony_ci}
4628c2ecf20Sopenharmony_ci
4638c2ecf20Sopenharmony_cistatic int pblk_rb_may_write_flush(struct pblk_rb *rb, unsigned int nr_entries,
4648c2ecf20Sopenharmony_ci				   unsigned int *pos, struct bio *bio,
4658c2ecf20Sopenharmony_ci				   int *io_ret)
4668c2ecf20Sopenharmony_ci{
4678c2ecf20Sopenharmony_ci	unsigned int mem;
4688c2ecf20Sopenharmony_ci
4698c2ecf20Sopenharmony_ci	if (!__pblk_rb_may_write(rb, nr_entries, pos))
4708c2ecf20Sopenharmony_ci		return 0;
4718c2ecf20Sopenharmony_ci
4728c2ecf20Sopenharmony_ci	mem = pblk_rb_ptr_wrap(rb, *pos, nr_entries);
4738c2ecf20Sopenharmony_ci	*io_ret = NVM_IO_DONE;
4748c2ecf20Sopenharmony_ci
4758c2ecf20Sopenharmony_ci	if (bio->bi_opf & REQ_PREFLUSH) {
4768c2ecf20Sopenharmony_ci		struct pblk *pblk = container_of(rb, struct pblk, rwb);
4778c2ecf20Sopenharmony_ci
4788c2ecf20Sopenharmony_ci		atomic64_inc(&pblk->nr_flush);
4798c2ecf20Sopenharmony_ci		if (pblk_rb_flush_point_set(&pblk->rwb, bio, mem))
4808c2ecf20Sopenharmony_ci			*io_ret = NVM_IO_OK;
4818c2ecf20Sopenharmony_ci	}
4828c2ecf20Sopenharmony_ci
4838c2ecf20Sopenharmony_ci	/* Protect from read count */
4848c2ecf20Sopenharmony_ci	smp_store_release(&rb->mem, mem);
4858c2ecf20Sopenharmony_ci
4868c2ecf20Sopenharmony_ci	return 1;
4878c2ecf20Sopenharmony_ci}
4888c2ecf20Sopenharmony_ci
4898c2ecf20Sopenharmony_ci/*
4908c2ecf20Sopenharmony_ci * Atomically check that (i) there is space on the write buffer for the
4918c2ecf20Sopenharmony_ci * incoming I/O, and (ii) the current I/O type has enough budget in the write
4928c2ecf20Sopenharmony_ci * buffer (rate-limiter).
4938c2ecf20Sopenharmony_ci */
4948c2ecf20Sopenharmony_ciint pblk_rb_may_write_user(struct pblk_rb *rb, struct bio *bio,
4958c2ecf20Sopenharmony_ci			   unsigned int nr_entries, unsigned int *pos)
4968c2ecf20Sopenharmony_ci{
4978c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
4988c2ecf20Sopenharmony_ci	int io_ret;
4998c2ecf20Sopenharmony_ci
5008c2ecf20Sopenharmony_ci	spin_lock(&rb->w_lock);
5018c2ecf20Sopenharmony_ci	io_ret = pblk_rl_user_may_insert(&pblk->rl, nr_entries);
5028c2ecf20Sopenharmony_ci	if (io_ret) {
5038c2ecf20Sopenharmony_ci		spin_unlock(&rb->w_lock);
5048c2ecf20Sopenharmony_ci		return io_ret;
5058c2ecf20Sopenharmony_ci	}
5068c2ecf20Sopenharmony_ci
5078c2ecf20Sopenharmony_ci	if (!pblk_rb_may_write_flush(rb, nr_entries, pos, bio, &io_ret)) {
5088c2ecf20Sopenharmony_ci		spin_unlock(&rb->w_lock);
5098c2ecf20Sopenharmony_ci		return NVM_IO_REQUEUE;
5108c2ecf20Sopenharmony_ci	}
5118c2ecf20Sopenharmony_ci
5128c2ecf20Sopenharmony_ci	pblk_rl_user_in(&pblk->rl, nr_entries);
5138c2ecf20Sopenharmony_ci	spin_unlock(&rb->w_lock);
5148c2ecf20Sopenharmony_ci
5158c2ecf20Sopenharmony_ci	return io_ret;
5168c2ecf20Sopenharmony_ci}
5178c2ecf20Sopenharmony_ci
5188c2ecf20Sopenharmony_ci/*
5198c2ecf20Sopenharmony_ci * Look at pblk_rb_may_write_user comment
5208c2ecf20Sopenharmony_ci */
5218c2ecf20Sopenharmony_ciint pblk_rb_may_write_gc(struct pblk_rb *rb, unsigned int nr_entries,
5228c2ecf20Sopenharmony_ci			 unsigned int *pos)
5238c2ecf20Sopenharmony_ci{
5248c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
5258c2ecf20Sopenharmony_ci
5268c2ecf20Sopenharmony_ci	spin_lock(&rb->w_lock);
5278c2ecf20Sopenharmony_ci	if (!pblk_rl_gc_may_insert(&pblk->rl, nr_entries)) {
5288c2ecf20Sopenharmony_ci		spin_unlock(&rb->w_lock);
5298c2ecf20Sopenharmony_ci		return 0;
5308c2ecf20Sopenharmony_ci	}
5318c2ecf20Sopenharmony_ci
5328c2ecf20Sopenharmony_ci	if (!pblk_rb_may_write(rb, nr_entries, pos)) {
5338c2ecf20Sopenharmony_ci		spin_unlock(&rb->w_lock);
5348c2ecf20Sopenharmony_ci		return 0;
5358c2ecf20Sopenharmony_ci	}
5368c2ecf20Sopenharmony_ci
5378c2ecf20Sopenharmony_ci	pblk_rl_gc_in(&pblk->rl, nr_entries);
5388c2ecf20Sopenharmony_ci	spin_unlock(&rb->w_lock);
5398c2ecf20Sopenharmony_ci
5408c2ecf20Sopenharmony_ci	return 1;
5418c2ecf20Sopenharmony_ci}
5428c2ecf20Sopenharmony_ci
5438c2ecf20Sopenharmony_ci/*
5448c2ecf20Sopenharmony_ci * Read available entries on rb and add them to the given bio. To avoid a memory
5458c2ecf20Sopenharmony_ci * copy, a page reference to the write buffer is used to be added to the bio.
5468c2ecf20Sopenharmony_ci *
5478c2ecf20Sopenharmony_ci * This function is used by the write thread to form the write bio that will
5488c2ecf20Sopenharmony_ci * persist data on the write buffer to the media.
5498c2ecf20Sopenharmony_ci */
5508c2ecf20Sopenharmony_ciunsigned int pblk_rb_read_to_bio(struct pblk_rb *rb, struct nvm_rq *rqd,
5518c2ecf20Sopenharmony_ci				 unsigned int pos, unsigned int nr_entries,
5528c2ecf20Sopenharmony_ci				 unsigned int count)
5538c2ecf20Sopenharmony_ci{
5548c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
5558c2ecf20Sopenharmony_ci	struct request_queue *q = pblk->dev->q;
5568c2ecf20Sopenharmony_ci	struct pblk_c_ctx *c_ctx = nvm_rq_to_pdu(rqd);
5578c2ecf20Sopenharmony_ci	struct bio *bio = rqd->bio;
5588c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
5598c2ecf20Sopenharmony_ci	struct page *page;
5608c2ecf20Sopenharmony_ci	unsigned int pad = 0, to_read = nr_entries;
5618c2ecf20Sopenharmony_ci	unsigned int i;
5628c2ecf20Sopenharmony_ci	int flags;
5638c2ecf20Sopenharmony_ci
5648c2ecf20Sopenharmony_ci	if (count < nr_entries) {
5658c2ecf20Sopenharmony_ci		pad = nr_entries - count;
5668c2ecf20Sopenharmony_ci		to_read = count;
5678c2ecf20Sopenharmony_ci	}
5688c2ecf20Sopenharmony_ci
5698c2ecf20Sopenharmony_ci	/* Add space for packed metadata if in use*/
5708c2ecf20Sopenharmony_ci	pad += (pblk->min_write_pgs - pblk->min_write_pgs_data);
5718c2ecf20Sopenharmony_ci
5728c2ecf20Sopenharmony_ci	c_ctx->sentry = pos;
5738c2ecf20Sopenharmony_ci	c_ctx->nr_valid = to_read;
5748c2ecf20Sopenharmony_ci	c_ctx->nr_padded = pad;
5758c2ecf20Sopenharmony_ci
5768c2ecf20Sopenharmony_ci	for (i = 0; i < to_read; i++) {
5778c2ecf20Sopenharmony_ci		entry = &rb->entries[pos];
5788c2ecf20Sopenharmony_ci
5798c2ecf20Sopenharmony_ci		/* A write has been allowed into the buffer, but data is still
5808c2ecf20Sopenharmony_ci		 * being copied to it. It is ok to busy wait.
5818c2ecf20Sopenharmony_ci		 */
5828c2ecf20Sopenharmony_citry:
5838c2ecf20Sopenharmony_ci		flags = READ_ONCE(entry->w_ctx.flags);
5848c2ecf20Sopenharmony_ci		if (!(flags & PBLK_WRITTEN_DATA)) {
5858c2ecf20Sopenharmony_ci			io_schedule();
5868c2ecf20Sopenharmony_ci			goto try;
5878c2ecf20Sopenharmony_ci		}
5888c2ecf20Sopenharmony_ci
5898c2ecf20Sopenharmony_ci		page = virt_to_page(entry->data);
5908c2ecf20Sopenharmony_ci		if (!page) {
5918c2ecf20Sopenharmony_ci			pblk_err(pblk, "could not allocate write bio page\n");
5928c2ecf20Sopenharmony_ci			flags &= ~PBLK_WRITTEN_DATA;
5938c2ecf20Sopenharmony_ci			flags |= PBLK_SUBMITTED_ENTRY;
5948c2ecf20Sopenharmony_ci			/* Release flags on context. Protect from writes */
5958c2ecf20Sopenharmony_ci			smp_store_release(&entry->w_ctx.flags, flags);
5968c2ecf20Sopenharmony_ci			return NVM_IO_ERR;
5978c2ecf20Sopenharmony_ci		}
5988c2ecf20Sopenharmony_ci
5998c2ecf20Sopenharmony_ci		if (bio_add_pc_page(q, bio, page, rb->seg_size, 0) !=
6008c2ecf20Sopenharmony_ci								rb->seg_size) {
6018c2ecf20Sopenharmony_ci			pblk_err(pblk, "could not add page to write bio\n");
6028c2ecf20Sopenharmony_ci			flags &= ~PBLK_WRITTEN_DATA;
6038c2ecf20Sopenharmony_ci			flags |= PBLK_SUBMITTED_ENTRY;
6048c2ecf20Sopenharmony_ci			/* Release flags on context. Protect from writes */
6058c2ecf20Sopenharmony_ci			smp_store_release(&entry->w_ctx.flags, flags);
6068c2ecf20Sopenharmony_ci			return NVM_IO_ERR;
6078c2ecf20Sopenharmony_ci		}
6088c2ecf20Sopenharmony_ci
6098c2ecf20Sopenharmony_ci		flags &= ~PBLK_WRITTEN_DATA;
6108c2ecf20Sopenharmony_ci		flags |= PBLK_SUBMITTED_ENTRY;
6118c2ecf20Sopenharmony_ci
6128c2ecf20Sopenharmony_ci		/* Release flags on context. Protect from writes */
6138c2ecf20Sopenharmony_ci		smp_store_release(&entry->w_ctx.flags, flags);
6148c2ecf20Sopenharmony_ci
6158c2ecf20Sopenharmony_ci		pos = pblk_rb_ptr_wrap(rb, pos, 1);
6168c2ecf20Sopenharmony_ci	}
6178c2ecf20Sopenharmony_ci
6188c2ecf20Sopenharmony_ci	if (pad) {
6198c2ecf20Sopenharmony_ci		if (pblk_bio_add_pages(pblk, bio, GFP_KERNEL, pad)) {
6208c2ecf20Sopenharmony_ci			pblk_err(pblk, "could not pad page in write bio\n");
6218c2ecf20Sopenharmony_ci			return NVM_IO_ERR;
6228c2ecf20Sopenharmony_ci		}
6238c2ecf20Sopenharmony_ci
6248c2ecf20Sopenharmony_ci		if (pad < pblk->min_write_pgs)
6258c2ecf20Sopenharmony_ci			atomic64_inc(&pblk->pad_dist[pad - 1]);
6268c2ecf20Sopenharmony_ci		else
6278c2ecf20Sopenharmony_ci			pblk_warn(pblk, "padding more than min. sectors\n");
6288c2ecf20Sopenharmony_ci
6298c2ecf20Sopenharmony_ci		atomic64_add(pad, &pblk->pad_wa);
6308c2ecf20Sopenharmony_ci	}
6318c2ecf20Sopenharmony_ci
6328c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
6338c2ecf20Sopenharmony_ci	atomic_long_add(pad, &pblk->padded_writes);
6348c2ecf20Sopenharmony_ci#endif
6358c2ecf20Sopenharmony_ci
6368c2ecf20Sopenharmony_ci	return NVM_IO_OK;
6378c2ecf20Sopenharmony_ci}
6388c2ecf20Sopenharmony_ci
6398c2ecf20Sopenharmony_ci/*
6408c2ecf20Sopenharmony_ci * Copy to bio only if the lba matches the one on the given cache entry.
6418c2ecf20Sopenharmony_ci * Otherwise, it means that the entry has been overwritten, and the bio should
6428c2ecf20Sopenharmony_ci * be directed to disk.
6438c2ecf20Sopenharmony_ci */
6448c2ecf20Sopenharmony_ciint pblk_rb_copy_to_bio(struct pblk_rb *rb, struct bio *bio, sector_t lba,
6458c2ecf20Sopenharmony_ci			struct ppa_addr ppa)
6468c2ecf20Sopenharmony_ci{
6478c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
6488c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
6498c2ecf20Sopenharmony_ci	struct pblk_w_ctx *w_ctx;
6508c2ecf20Sopenharmony_ci	struct ppa_addr l2p_ppa;
6518c2ecf20Sopenharmony_ci	u64 pos = pblk_addr_to_cacheline(ppa);
6528c2ecf20Sopenharmony_ci	void *data;
6538c2ecf20Sopenharmony_ci	int flags;
6548c2ecf20Sopenharmony_ci	int ret = 1;
6558c2ecf20Sopenharmony_ci
6568c2ecf20Sopenharmony_ci
6578c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
6588c2ecf20Sopenharmony_ci	/* Caller must ensure that the access will not cause an overflow */
6598c2ecf20Sopenharmony_ci	BUG_ON(pos >= rb->nr_entries);
6608c2ecf20Sopenharmony_ci#endif
6618c2ecf20Sopenharmony_ci	entry = &rb->entries[pos];
6628c2ecf20Sopenharmony_ci	w_ctx = &entry->w_ctx;
6638c2ecf20Sopenharmony_ci	flags = READ_ONCE(w_ctx->flags);
6648c2ecf20Sopenharmony_ci
6658c2ecf20Sopenharmony_ci	spin_lock(&rb->w_lock);
6668c2ecf20Sopenharmony_ci	spin_lock(&pblk->trans_lock);
6678c2ecf20Sopenharmony_ci	l2p_ppa = pblk_trans_map_get(pblk, lba);
6688c2ecf20Sopenharmony_ci	spin_unlock(&pblk->trans_lock);
6698c2ecf20Sopenharmony_ci
6708c2ecf20Sopenharmony_ci	/* Check if the entry has been overwritten or is scheduled to be */
6718c2ecf20Sopenharmony_ci	if (!pblk_ppa_comp(l2p_ppa, ppa) || w_ctx->lba != lba ||
6728c2ecf20Sopenharmony_ci						flags & PBLK_WRITABLE_ENTRY) {
6738c2ecf20Sopenharmony_ci		ret = 0;
6748c2ecf20Sopenharmony_ci		goto out;
6758c2ecf20Sopenharmony_ci	}
6768c2ecf20Sopenharmony_ci	data = bio_data(bio);
6778c2ecf20Sopenharmony_ci	memcpy(data, entry->data, rb->seg_size);
6788c2ecf20Sopenharmony_ci
6798c2ecf20Sopenharmony_ciout:
6808c2ecf20Sopenharmony_ci	spin_unlock(&rb->w_lock);
6818c2ecf20Sopenharmony_ci	return ret;
6828c2ecf20Sopenharmony_ci}
6838c2ecf20Sopenharmony_ci
6848c2ecf20Sopenharmony_cistruct pblk_w_ctx *pblk_rb_w_ctx(struct pblk_rb *rb, unsigned int pos)
6858c2ecf20Sopenharmony_ci{
6868c2ecf20Sopenharmony_ci	unsigned int entry = pblk_rb_ptr_wrap(rb, pos, 0);
6878c2ecf20Sopenharmony_ci
6888c2ecf20Sopenharmony_ci	return &rb->entries[entry].w_ctx;
6898c2ecf20Sopenharmony_ci}
6908c2ecf20Sopenharmony_ci
6918c2ecf20Sopenharmony_ciunsigned int pblk_rb_sync_init(struct pblk_rb *rb, unsigned long *flags)
6928c2ecf20Sopenharmony_ci	__acquires(&rb->s_lock)
6938c2ecf20Sopenharmony_ci{
6948c2ecf20Sopenharmony_ci	if (flags)
6958c2ecf20Sopenharmony_ci		spin_lock_irqsave(&rb->s_lock, *flags);
6968c2ecf20Sopenharmony_ci	else
6978c2ecf20Sopenharmony_ci		spin_lock_irq(&rb->s_lock);
6988c2ecf20Sopenharmony_ci
6998c2ecf20Sopenharmony_ci	return rb->sync;
7008c2ecf20Sopenharmony_ci}
7018c2ecf20Sopenharmony_ci
7028c2ecf20Sopenharmony_civoid pblk_rb_sync_end(struct pblk_rb *rb, unsigned long *flags)
7038c2ecf20Sopenharmony_ci	__releases(&rb->s_lock)
7048c2ecf20Sopenharmony_ci{
7058c2ecf20Sopenharmony_ci	lockdep_assert_held(&rb->s_lock);
7068c2ecf20Sopenharmony_ci
7078c2ecf20Sopenharmony_ci	if (flags)
7088c2ecf20Sopenharmony_ci		spin_unlock_irqrestore(&rb->s_lock, *flags);
7098c2ecf20Sopenharmony_ci	else
7108c2ecf20Sopenharmony_ci		spin_unlock_irq(&rb->s_lock);
7118c2ecf20Sopenharmony_ci}
7128c2ecf20Sopenharmony_ci
7138c2ecf20Sopenharmony_ciunsigned int pblk_rb_sync_advance(struct pblk_rb *rb, unsigned int nr_entries)
7148c2ecf20Sopenharmony_ci{
7158c2ecf20Sopenharmony_ci	unsigned int sync, flush_point;
7168c2ecf20Sopenharmony_ci	lockdep_assert_held(&rb->s_lock);
7178c2ecf20Sopenharmony_ci
7188c2ecf20Sopenharmony_ci	sync = READ_ONCE(rb->sync);
7198c2ecf20Sopenharmony_ci	flush_point = READ_ONCE(rb->flush_point);
7208c2ecf20Sopenharmony_ci
7218c2ecf20Sopenharmony_ci	if (flush_point != EMPTY_ENTRY) {
7228c2ecf20Sopenharmony_ci		unsigned int secs_to_flush;
7238c2ecf20Sopenharmony_ci
7248c2ecf20Sopenharmony_ci		secs_to_flush = pblk_rb_ring_count(flush_point, sync,
7258c2ecf20Sopenharmony_ci					rb->nr_entries);
7268c2ecf20Sopenharmony_ci		if (secs_to_flush < nr_entries) {
7278c2ecf20Sopenharmony_ci			/* Protect flush points */
7288c2ecf20Sopenharmony_ci			smp_store_release(&rb->flush_point, EMPTY_ENTRY);
7298c2ecf20Sopenharmony_ci		}
7308c2ecf20Sopenharmony_ci	}
7318c2ecf20Sopenharmony_ci
7328c2ecf20Sopenharmony_ci	sync = pblk_rb_ptr_wrap(rb, sync, nr_entries);
7338c2ecf20Sopenharmony_ci
7348c2ecf20Sopenharmony_ci	/* Protect from counts */
7358c2ecf20Sopenharmony_ci	smp_store_release(&rb->sync, sync);
7368c2ecf20Sopenharmony_ci
7378c2ecf20Sopenharmony_ci	return sync;
7388c2ecf20Sopenharmony_ci}
7398c2ecf20Sopenharmony_ci
7408c2ecf20Sopenharmony_ci/* Calculate how many sectors to submit up to the current flush point. */
7418c2ecf20Sopenharmony_ciunsigned int pblk_rb_flush_point_count(struct pblk_rb *rb)
7428c2ecf20Sopenharmony_ci{
7438c2ecf20Sopenharmony_ci	unsigned int subm, sync, flush_point;
7448c2ecf20Sopenharmony_ci	unsigned int submitted, to_flush;
7458c2ecf20Sopenharmony_ci
7468c2ecf20Sopenharmony_ci	/* Protect flush points */
7478c2ecf20Sopenharmony_ci	flush_point = smp_load_acquire(&rb->flush_point);
7488c2ecf20Sopenharmony_ci	if (flush_point == EMPTY_ENTRY)
7498c2ecf20Sopenharmony_ci		return 0;
7508c2ecf20Sopenharmony_ci
7518c2ecf20Sopenharmony_ci	/* Protect syncs */
7528c2ecf20Sopenharmony_ci	sync = smp_load_acquire(&rb->sync);
7538c2ecf20Sopenharmony_ci
7548c2ecf20Sopenharmony_ci	subm = READ_ONCE(rb->subm);
7558c2ecf20Sopenharmony_ci	submitted = pblk_rb_ring_count(subm, sync, rb->nr_entries);
7568c2ecf20Sopenharmony_ci
7578c2ecf20Sopenharmony_ci	/* The sync point itself counts as a sector to sync */
7588c2ecf20Sopenharmony_ci	to_flush = pblk_rb_ring_count(flush_point, sync, rb->nr_entries) + 1;
7598c2ecf20Sopenharmony_ci
7608c2ecf20Sopenharmony_ci	return (submitted < to_flush) ? (to_flush - submitted) : 0;
7618c2ecf20Sopenharmony_ci}
7628c2ecf20Sopenharmony_ci
7638c2ecf20Sopenharmony_ciint pblk_rb_tear_down_check(struct pblk_rb *rb)
7648c2ecf20Sopenharmony_ci{
7658c2ecf20Sopenharmony_ci	struct pblk_rb_entry *entry;
7668c2ecf20Sopenharmony_ci	int i;
7678c2ecf20Sopenharmony_ci	int ret = 0;
7688c2ecf20Sopenharmony_ci
7698c2ecf20Sopenharmony_ci	spin_lock(&rb->w_lock);
7708c2ecf20Sopenharmony_ci	spin_lock_irq(&rb->s_lock);
7718c2ecf20Sopenharmony_ci
7728c2ecf20Sopenharmony_ci	if ((rb->mem == rb->subm) && (rb->subm == rb->sync) &&
7738c2ecf20Sopenharmony_ci				(rb->sync == rb->l2p_update) &&
7748c2ecf20Sopenharmony_ci				(rb->flush_point == EMPTY_ENTRY)) {
7758c2ecf20Sopenharmony_ci		goto out;
7768c2ecf20Sopenharmony_ci	}
7778c2ecf20Sopenharmony_ci
7788c2ecf20Sopenharmony_ci	if (!rb->entries) {
7798c2ecf20Sopenharmony_ci		ret = 1;
7808c2ecf20Sopenharmony_ci		goto out;
7818c2ecf20Sopenharmony_ci	}
7828c2ecf20Sopenharmony_ci
7838c2ecf20Sopenharmony_ci	for (i = 0; i < rb->nr_entries; i++) {
7848c2ecf20Sopenharmony_ci		entry = &rb->entries[i];
7858c2ecf20Sopenharmony_ci
7868c2ecf20Sopenharmony_ci		if (!entry->data) {
7878c2ecf20Sopenharmony_ci			ret = 1;
7888c2ecf20Sopenharmony_ci			goto out;
7898c2ecf20Sopenharmony_ci		}
7908c2ecf20Sopenharmony_ci	}
7918c2ecf20Sopenharmony_ci
7928c2ecf20Sopenharmony_ciout:
7938c2ecf20Sopenharmony_ci	spin_unlock_irq(&rb->s_lock);
7948c2ecf20Sopenharmony_ci	spin_unlock(&rb->w_lock);
7958c2ecf20Sopenharmony_ci
7968c2ecf20Sopenharmony_ci	return ret;
7978c2ecf20Sopenharmony_ci}
7988c2ecf20Sopenharmony_ci
7998c2ecf20Sopenharmony_ciunsigned int pblk_rb_wrap_pos(struct pblk_rb *rb, unsigned int pos)
8008c2ecf20Sopenharmony_ci{
8018c2ecf20Sopenharmony_ci	return (pos & (rb->nr_entries - 1));
8028c2ecf20Sopenharmony_ci}
8038c2ecf20Sopenharmony_ci
8048c2ecf20Sopenharmony_ciint pblk_rb_pos_oob(struct pblk_rb *rb, u64 pos)
8058c2ecf20Sopenharmony_ci{
8068c2ecf20Sopenharmony_ci	return (pos >= rb->nr_entries);
8078c2ecf20Sopenharmony_ci}
8088c2ecf20Sopenharmony_ci
8098c2ecf20Sopenharmony_cissize_t pblk_rb_sysfs(struct pblk_rb *rb, char *buf)
8108c2ecf20Sopenharmony_ci{
8118c2ecf20Sopenharmony_ci	struct pblk *pblk = container_of(rb, struct pblk, rwb);
8128c2ecf20Sopenharmony_ci	struct pblk_c_ctx *c;
8138c2ecf20Sopenharmony_ci	ssize_t offset;
8148c2ecf20Sopenharmony_ci	int queued_entries = 0;
8158c2ecf20Sopenharmony_ci
8168c2ecf20Sopenharmony_ci	spin_lock_irq(&rb->s_lock);
8178c2ecf20Sopenharmony_ci	list_for_each_entry(c, &pblk->compl_list, list)
8188c2ecf20Sopenharmony_ci		queued_entries++;
8198c2ecf20Sopenharmony_ci	spin_unlock_irq(&rb->s_lock);
8208c2ecf20Sopenharmony_ci
8218c2ecf20Sopenharmony_ci	if (rb->flush_point != EMPTY_ENTRY)
8228c2ecf20Sopenharmony_ci		offset = scnprintf(buf, PAGE_SIZE,
8238c2ecf20Sopenharmony_ci			"%u\t%u\t%u\t%u\t%u\t%u\t%u - %u/%u/%u - %d\n",
8248c2ecf20Sopenharmony_ci			rb->nr_entries,
8258c2ecf20Sopenharmony_ci			rb->mem,
8268c2ecf20Sopenharmony_ci			rb->subm,
8278c2ecf20Sopenharmony_ci			rb->sync,
8288c2ecf20Sopenharmony_ci			rb->l2p_update,
8298c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
8308c2ecf20Sopenharmony_ci			atomic_read(&rb->inflight_flush_point),
8318c2ecf20Sopenharmony_ci#else
8328c2ecf20Sopenharmony_ci			0,
8338c2ecf20Sopenharmony_ci#endif
8348c2ecf20Sopenharmony_ci			rb->flush_point,
8358c2ecf20Sopenharmony_ci			pblk_rb_read_count(rb),
8368c2ecf20Sopenharmony_ci			pblk_rb_space(rb),
8378c2ecf20Sopenharmony_ci			pblk_rb_flush_point_count(rb),
8388c2ecf20Sopenharmony_ci			queued_entries);
8398c2ecf20Sopenharmony_ci	else
8408c2ecf20Sopenharmony_ci		offset = scnprintf(buf, PAGE_SIZE,
8418c2ecf20Sopenharmony_ci			"%u\t%u\t%u\t%u\t%u\t%u\tNULL - %u/%u/%u - %d\n",
8428c2ecf20Sopenharmony_ci			rb->nr_entries,
8438c2ecf20Sopenharmony_ci			rb->mem,
8448c2ecf20Sopenharmony_ci			rb->subm,
8458c2ecf20Sopenharmony_ci			rb->sync,
8468c2ecf20Sopenharmony_ci			rb->l2p_update,
8478c2ecf20Sopenharmony_ci#ifdef CONFIG_NVM_PBLK_DEBUG
8488c2ecf20Sopenharmony_ci			atomic_read(&rb->inflight_flush_point),
8498c2ecf20Sopenharmony_ci#else
8508c2ecf20Sopenharmony_ci			0,
8518c2ecf20Sopenharmony_ci#endif
8528c2ecf20Sopenharmony_ci			pblk_rb_read_count(rb),
8538c2ecf20Sopenharmony_ci			pblk_rb_space(rb),
8548c2ecf20Sopenharmony_ci			pblk_rb_flush_point_count(rb),
8558c2ecf20Sopenharmony_ci			queued_entries);
8568c2ecf20Sopenharmony_ci
8578c2ecf20Sopenharmony_ci	return offset;
8588c2ecf20Sopenharmony_ci}
859