[linux-2.6-block.git] / drivers / lightnvm / pblk-write.c

/*
 * Copyright (C) 2016 CNEX Labs
 * Initial release: Javier Gonzalez <javier@cnexlabs.com>
 *                  Matias Bjorling <matias@cnexlabs.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * pblk-write.c - pblk's write path from write buffer to media
 */

#include "pblk.h"

static void pblk_sync_line(struct pblk *pblk, struct pblk_line *line)
{
#ifdef CONFIG_NVM_DEBUG
	atomic_long_inc(&pblk->sync_writes);
#endif

	/* Counter protected by rb sync lock */
	line->left_ssecs--;
	if (!line->left_ssecs)
		pblk_line_run_ws(pblk, line, NULL, pblk_line_close_ws);
}

static unsigned long pblk_end_w_bio(struct pblk *pblk, struct nvm_rq *rqd,
				    struct pblk_c_ctx *c_ctx)
{
	struct nvm_tgt_dev *dev = pblk->dev;
	struct bio *original_bio;
	unsigned long ret;
	int i;

	for (i = 0; i < c_ctx->nr_valid; i++) {
		struct pblk_w_ctx *w_ctx;
		struct ppa_addr p;
		struct pblk_line *line;

		w_ctx = pblk_rb_w_ctx(&pblk->rwb, c_ctx->sentry + i);

		p = rqd->ppa_list[i];
		line = &pblk->lines[pblk_dev_ppa_to_line(p)];
		pblk_sync_line(pblk, line);

		while ((original_bio = bio_list_pop(&w_ctx->bios)))
			bio_endio(original_bio);
	}

#ifdef CONFIG_NVM_DEBUG
	atomic_long_add(c_ctx->nr_valid, &pblk->compl_writes);
#endif

	ret = pblk_rb_sync_advance(&pblk->rwb, c_ctx->nr_valid);

	if (rqd->meta_list)
		nvm_dev_dma_free(dev->parent, rqd->meta_list,
							rqd->dma_meta_list);

	bio_put(rqd->bio);
	pblk_free_rqd(pblk, rqd, WRITE);

	return ret;
}

static unsigned long pblk_end_queued_w_bio(struct pblk *pblk,
					   struct nvm_rq *rqd,
					   struct pblk_c_ctx *c_ctx)
{
	list_del(&c_ctx->list);
	return pblk_end_w_bio(pblk, rqd, c_ctx);
}

static void pblk_complete_write(struct pblk *pblk, struct nvm_rq *rqd,
				struct pblk_c_ctx *c_ctx)
{
	struct pblk_c_ctx *c, *r;
	unsigned long flags;
	unsigned long pos;

#ifdef CONFIG_NVM_DEBUG
	atomic_long_sub(c_ctx->nr_valid, &pblk->inflight_writes);
#endif

	pblk_up_rq(pblk, rqd->ppa_list, rqd->nr_ppas, c_ctx->lun_bitmap);

	pos = pblk_rb_sync_init(&pblk->rwb, &flags);
	if (pos == c_ctx->sentry) {
		pos = pblk_end_w_bio(pblk, rqd, c_ctx);

retry:
		list_for_each_entry_safe(c, r, &pblk->compl_list, list) {
			rqd = nvm_rq_from_c_ctx(c);
			if (c->sentry == pos) {
				pos = pblk_end_queued_w_bio(pblk, rqd, c);
				goto retry;
			}
		}
	} else {
		WARN_ON(nvm_rq_from_c_ctx(c_ctx) != rqd);
		list_add_tail(&c_ctx->list, &pblk->compl_list);
	}
	pblk_rb_sync_end(&pblk->rwb, &flags);
}

/* When a write fails, we are not sure whether the block has grown bad or a page
 * range is more susceptible to write errors. If a high number of pages fail, we
 * assume that the block is bad and we mark it accordingly. In all cases, we
 * remap and resubmit the failed entries as fast as possible; if a flush is
 * waiting on a completion, the whole stack would stall otherwise.
 */
static void pblk_end_w_fail(struct pblk *pblk, struct nvm_rq *rqd)
{
	void *comp_bits = &rqd->ppa_status;
	struct pblk_c_ctx *c_ctx = nvm_rq_to_pdu(rqd);
	struct pblk_rec_ctx *recovery;
	struct ppa_addr *ppa_list = rqd->ppa_list;
	int nr_ppas = rqd->nr_ppas;
	unsigned int c_entries;
	int bit, ret;

	if (unlikely(nr_ppas == 1))
		ppa_list = &rqd->ppa_addr;

	recovery = mempool_alloc(pblk->rec_pool, GFP_ATOMIC);
	if (!recovery) {
		pr_err("pblk: could not allocate recovery context\n");
		return;
	}
	INIT_LIST_HEAD(&recovery->failed);

	bit = -1;
	while ((bit = find_next_bit(comp_bits, nr_ppas, bit + 1)) < nr_ppas) {
		struct pblk_rb_entry *entry;
		struct ppa_addr ppa;

		/* Logic error */
		if (bit > c_ctx->nr_valid) {
			WARN_ONCE(1, "pblk: corrupted write request\n");
			mempool_free(recovery, pblk->rec_pool);
			goto out;
		}

		ppa = ppa_list[bit];
		entry = pblk_rb_sync_scan_entry(&pblk->rwb, &ppa);
		if (!entry) {
			pr_err("pblk: could not scan entry on write failure\n");
			mempool_free(recovery, pblk->rec_pool);
			goto out;
		}

		/* The list is filled first and emptied afterwards. No need for
		 * protecting it with a lock
		 */
		list_add_tail(&entry->index, &recovery->failed);
	}

	c_entries = find_first_bit(comp_bits, nr_ppas);
	ret = pblk_recov_setup_rq(pblk, c_ctx, recovery, comp_bits, c_entries);
	if (ret) {
		pr_err("pblk: could not recover from write failure\n");
		mempool_free(recovery, pblk->rec_pool);
		goto out;
	}

	INIT_WORK(&recovery->ws_rec, pblk_submit_rec);
	queue_work(pblk->kw_wq, &recovery->ws_rec);

out:
	pblk_complete_write(pblk, rqd, c_ctx);
}

static void pblk_end_io_write(struct nvm_rq *rqd)
{
	struct pblk *pblk = rqd->private;
	struct pblk_c_ctx *c_ctx = nvm_rq_to_pdu(rqd);

	if (rqd->error) {
		pblk_log_write_err(pblk, rqd);
		return pblk_end_w_fail(pblk, rqd);
	}
#ifdef CONFIG_NVM_DEBUG
	else
		WARN_ONCE(rqd->bio->bi_error, "pblk: corrupted write error\n");
#endif

	pblk_complete_write(pblk, rqd, c_ctx);
}

static int pblk_alloc_w_rq(struct pblk *pblk, struct nvm_rq *rqd,
			   unsigned int nr_secs)
{
	struct nvm_tgt_dev *dev = pblk->dev;

	/* Setup write request */
	rqd->opcode = NVM_OP_PWRITE;
	rqd->nr_ppas = nr_secs;
	rqd->flags = pblk_set_progr_mode(pblk, WRITE);
	rqd->private = pblk;
	rqd->end_io = pblk_end_io_write;

	rqd->meta_list = nvm_dev_dma_alloc(dev->parent, GFP_KERNEL,
							&rqd->dma_meta_list);
	if (!rqd->meta_list)
		return -ENOMEM;

	if (unlikely(nr_secs == 1))
		return 0;

	rqd->ppa_list = rqd->meta_list + pblk_dma_meta_size;
	rqd->dma_ppa_list = rqd->dma_meta_list + pblk_dma_meta_size;

	return 0;
}

static int pblk_setup_w_rq(struct pblk *pblk, struct nvm_rq *rqd,
			   struct pblk_c_ctx *c_ctx)
{
	struct pblk_line_meta *lm = &pblk->lm;
	struct pblk_line *e_line = pblk_line_get_data_next(pblk);
	struct ppa_addr erase_ppa;
	unsigned int valid = c_ctx->nr_valid;
	unsigned int padded = c_ctx->nr_padded;
	unsigned int nr_secs = valid + padded;
	unsigned long *lun_bitmap;
	int ret = 0;

	lun_bitmap = kzalloc(lm->lun_bitmap_len, GFP_KERNEL);
	if (!lun_bitmap) {
		ret = -ENOMEM;
		goto out;
	}
	c_ctx->lun_bitmap = lun_bitmap;

	ret = pblk_alloc_w_rq(pblk, rqd, nr_secs);
	if (ret) {
		kfree(lun_bitmap);
		goto out;
	}

	ppa_set_empty(&erase_ppa);
	if (likely(!e_line || !atomic_read(&e_line->left_eblks)))
		pblk_map_rq(pblk, rqd, c_ctx->sentry, lun_bitmap, valid, 0);
	else
		pblk_map_erase_rq(pblk, rqd, c_ctx->sentry, lun_bitmap,
							valid, &erase_ppa);

out:
	if (unlikely(e_line && !ppa_empty(erase_ppa))) {
		if (pblk_blk_erase_async(pblk, erase_ppa)) {
			struct nvm_tgt_dev *dev = pblk->dev;
			struct nvm_geo *geo = &dev->geo;
			int bit;

			atomic_inc(&e_line->left_eblks);
			bit = erase_ppa.g.lun * geo->nr_chnls + erase_ppa.g.ch;
			WARN_ON(!test_and_clear_bit(bit, e_line->erase_bitmap));
			up(&pblk->erase_sem);
		}
	}

	return ret;
}

int pblk_setup_w_rec_rq(struct pblk *pblk, struct nvm_rq *rqd,
			struct pblk_c_ctx *c_ctx)
{
	struct pblk_line_meta *lm = &pblk->lm;
	unsigned long *lun_bitmap;
	int ret;

	lun_bitmap = kzalloc(lm->lun_bitmap_len, GFP_KERNEL);
	if (!lun_bitmap)
		return -ENOMEM;

	c_ctx->lun_bitmap = lun_bitmap;

	ret = pblk_alloc_w_rq(pblk, rqd, rqd->nr_ppas);
	if (ret)
		return ret;

	pblk_map_rq(pblk, rqd, c_ctx->sentry, lun_bitmap, c_ctx->nr_valid, 0);

	rqd->ppa_status = (u64)0;
	rqd->flags = pblk_set_progr_mode(pblk, WRITE);

	return ret;
}

static int pblk_calc_secs_to_sync(struct pblk *pblk, unsigned int secs_avail,
				  unsigned int secs_to_flush)
{
	int secs_to_sync;

	secs_to_sync = pblk_calc_secs(pblk, secs_avail, secs_to_flush);

#ifdef CONFIG_NVM_DEBUG
	if ((!secs_to_sync && secs_to_flush)
			|| (secs_to_sync < 0)
			|| (secs_to_sync > secs_avail && !secs_to_flush)) {
		pr_err("pblk: bad sector calculation (a:%d,s:%d,f:%d)\n",
				secs_avail, secs_to_sync, secs_to_flush);
	}
#endif

	return secs_to_sync;
}

static int pblk_submit_write(struct pblk *pblk)
{
	struct bio *bio;
	struct nvm_rq *rqd;
	struct pblk_c_ctx *c_ctx;
	unsigned int pgs_read;
	unsigned int secs_avail, secs_to_sync, secs_to_com;
	unsigned int secs_to_flush;
	unsigned long pos;
	int err;

	/* If there are no sectors in the cache, flushes (bios without data)
	 * will be cleared on the cache threads
	 */
	secs_avail = pblk_rb_read_count(&pblk->rwb);
	if (!secs_avail)
		return 1;

	secs_to_flush = pblk_rb_sync_point_count(&pblk->rwb);
	if (!secs_to_flush && secs_avail < pblk->min_write_pgs)
		return 1;

	rqd = pblk_alloc_rqd(pblk, WRITE);
	if (IS_ERR(rqd)) {
		pr_err("pblk: cannot allocate write req.\n");
		return 1;
	}
	c_ctx = nvm_rq_to_pdu(rqd);

	bio = bio_alloc(GFP_KERNEL, pblk->max_write_pgs);
	if (!bio) {
		pr_err("pblk: cannot allocate write bio\n");
		goto fail_free_rqd;
	}
	bio->bi_iter.bi_sector = 0; /* internal bio */
	bio_set_op_attrs(bio, REQ_OP_WRITE, 0);
	rqd->bio = bio;

	secs_to_sync = pblk_calc_secs_to_sync(pblk, secs_avail, secs_to_flush);
	if (secs_to_sync > pblk->max_write_pgs) {
		pr_err("pblk: bad buffer sync calculation\n");
		goto fail_put_bio;
	}

	secs_to_com = (secs_to_sync > secs_avail) ? secs_avail : secs_to_sync;
	pos = pblk_rb_read_commit(&pblk->rwb, secs_to_com);

	pgs_read = pblk_rb_read_to_bio(&pblk->rwb, bio, c_ctx, pos,
						secs_to_sync, secs_avail);
	if (!pgs_read) {
		pr_err("pblk: corrupted write bio\n");
		goto fail_put_bio;
	}

	if (c_ctx->nr_padded)
		if (pblk_bio_add_pages(pblk, bio, GFP_KERNEL, c_ctx->nr_padded))
			goto fail_put_bio;

	/* Assign lbas to ppas and populate request structure */
	err = pblk_setup_w_rq(pblk, rqd, c_ctx);
	if (err) {
		pr_err("pblk: could not setup write request\n");
		goto fail_free_bio;
	}

	err = pblk_submit_io(pblk, rqd);
	if (err) {
		pr_err("pblk: I/O submission failed: %d\n", err);
		goto fail_free_bio;
	}

#ifdef CONFIG_NVM_DEBUG
	atomic_long_add(secs_to_sync, &pblk->sub_writes);
#endif

	return 0;

fail_free_bio:
	if (c_ctx->nr_padded)
		pblk_bio_free_pages(pblk, bio, secs_to_sync, c_ctx->nr_padded);
fail_put_bio:
	bio_put(bio);
fail_free_rqd:
	pblk_free_rqd(pblk, rqd, WRITE);

	return 1;
}

int pblk_write_ts(void *data)
{
	struct pblk *pblk = data;

	while (!kthread_should_stop()) {
		if (!pblk_submit_write(pblk))
			continue;
		set_current_state(TASK_INTERRUPTIBLE);
		io_schedule();
	}

	return 0;
}
Commit	Line	Data
a4bd217b JG	1	/*
	2	* Copyright (C) 2016 CNEX Labs
	3	* Initial release: Javier Gonzalez <javier@cnexlabs.com>
	4	* Matias Bjorling <matias@cnexlabs.com>
	5	*
	6	* This program is free software; you can redistribute it and/or
	7	* modify it under the terms of the GNU General Public License version
	8	* 2 as published by the Free Software Foundation.
	9	*
	10	* This program is distributed in the hope that it will be useful, but
	11	* WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	13	* General Public License for more details.
	14	*
	15	* pblk-write.c - pblk's write path from write buffer to media
	16	*/
	17
	18	#include "pblk.h"
	19
	20	static void pblk_sync_line(struct pblk pblk, struct pblk_line line)
	21	{
	22	#ifdef CONFIG_NVM_DEBUG
	23	atomic_long_inc(&pblk->sync_writes);
	24	#endif
	25
	26	/* Counter protected by rb sync lock */
	27	line->left_ssecs--;
	28	if (!line->left_ssecs)
	29	pblk_line_run_ws(pblk, line, NULL, pblk_line_close_ws);
	30	}
	31
	32	static unsigned long pblk_end_w_bio(struct pblk pblk, struct nvm_rq rqd,
	33	struct pblk_c_ctx *c_ctx)
	34	{
	35	struct nvm_tgt_dev *dev = pblk->dev;
	36	struct bio *original_bio;
	37	unsigned long ret;
	38	int i;
	39
	40	for (i = 0; i < c_ctx->nr_valid; i++) {
	41	struct pblk_w_ctx *w_ctx;
	42	struct ppa_addr p;
	43	struct pblk_line *line;
	44
	45	w_ctx = pblk_rb_w_ctx(&pblk->rwb, c_ctx->sentry + i);
	46
	47	p = rqd->ppa_list[i];
	48	line = &pblk->lines[pblk_dev_ppa_to_line(p)];
	49	pblk_sync_line(pblk, line);
	50
	51	while ((original_bio = bio_list_pop(&w_ctx->bios)))
	52	bio_endio(original_bio);
	53	}
	54
	55	#ifdef CONFIG_NVM_DEBUG
	56	atomic_long_add(c_ctx->nr_valid, &pblk->compl_writes);
	57	#endif
	58
	59	ret = pblk_rb_sync_advance(&pblk->rwb, c_ctx->nr_valid);
	60
	61	if (rqd->meta_list)
	62	nvm_dev_dma_free(dev->parent, rqd->meta_list,
	63	rqd->dma_meta_list);
	64
65	bio_put(rqd->bio);
66	pblk_free_rqd(pblk, rqd, WRITE);
67
68	return ret;
69	}
70
71	static unsigned long pblk_end_queued_w_bio(struct pblk *pblk,
72	struct nvm_rq *rqd,
73	struct pblk_c_ctx *c_ctx)
74	{
75	list_del(&c_ctx->list);
76	return pblk_end_w_bio(pblk, rqd, c_ctx);
77	}
78
79	static void pblk_complete_write(struct pblk pblk, struct nvm_rq rqd,
80	struct pblk_c_ctx *c_ctx)
81	{
82	struct pblk_c_ctx c, r;
83	unsigned long flags;
84	unsigned long pos;
85
86	#ifdef CONFIG_NVM_DEBUG
87	atomic_long_sub(c_ctx->nr_valid, &pblk->inflight_writes);
88	#endif
89
90	pblk_up_rq(pblk, rqd->ppa_list, rqd->nr_ppas, c_ctx->lun_bitmap);
91
92	pos = pblk_rb_sync_init(&pblk->rwb, &flags);
93	if (pos == c_ctx->sentry) {
94	pos = pblk_end_w_bio(pblk, rqd, c_ctx);
95
96	retry:
97	list_for_each_entry_safe(c, r, &pblk->compl_list, list) {
98	rqd = nvm_rq_from_c_ctx(c);
99	if (c->sentry == pos) {
100	pos = pblk_end_queued_w_bio(pblk, rqd, c);
101	goto retry;
102	}
103	}
104	} else {
105	WARN_ON(nvm_rq_from_c_ctx(c_ctx) != rqd);
106	list_add_tail(&c_ctx->list, &pblk->compl_list);
107	}
108	pblk_rb_sync_end(&pblk->rwb, &flags);
109	}
110
111	/* When a write fails, we are not sure whether the block has grown bad or a page
112	* range is more susceptible to write errors. If a high number of pages fail, we
113	* assume that the block is bad and we mark it accordingly. In all cases, we
114	* remap and resubmit the failed entries as fast as possible; if a flush is
115	* waiting on a completion, the whole stack would stall otherwise.
116	*/
117	static void pblk_end_w_fail(struct pblk pblk, struct nvm_rq rqd)
118	{
119	void *comp_bits = &rqd->ppa_status;
120	struct pblk_c_ctx *c_ctx = nvm_rq_to_pdu(rqd);
121	struct pblk_rec_ctx *recovery;
122	struct ppa_addr *ppa_list = rqd->ppa_list;
123	int nr_ppas = rqd->nr_ppas;
124	unsigned int c_entries;
125	int bit, ret;
126
127	if (unlikely(nr_ppas == 1))
128	ppa_list = &rqd->ppa_addr;
129
130	recovery = mempool_alloc(pblk->rec_pool, GFP_ATOMIC);
131	if (!recovery) {
132	pr_err("pblk: could not allocate recovery context\n");
133	return;
134	}
135	INIT_LIST_HEAD(&recovery->failed);
136
137	bit = -1;
138	while ((bit = find_next_bit(comp_bits, nr_ppas, bit + 1)) < nr_ppas) {
139	struct pblk_rb_entry *entry;
140	struct ppa_addr ppa;
141
142	/* Logic error */
143	if (bit > c_ctx->nr_valid) {
2a79efd8	144	WARN_ONCE(1, "pblk: corrupted write request\n");
33db9fd4	145	mempool_free(recovery, pblk->rec_pool);
a4bd217b JG	146	goto out;
	147	}
	148
	149	ppa = ppa_list[bit];
	150	entry = pblk_rb_sync_scan_entry(&pblk->rwb, &ppa);
	151	if (!entry) {
	152	pr_err("pblk: could not scan entry on write failure\n");
33db9fd4	153	mempool_free(recovery, pblk->rec_pool);
a4bd217b JG	154	goto out;
	155	}
	156
	157	/* The list is filled first and emptied afterwards. No need for
	158	* protecting it with a lock
	159	*/
	160	list_add_tail(&entry->index, &recovery->failed);
	161	}
	162
	163	c_entries = find_first_bit(comp_bits, nr_ppas);
	164	ret = pblk_recov_setup_rq(pblk, c_ctx, recovery, comp_bits, c_entries);
	165	if (ret) {
	166	pr_err("pblk: could not recover from write failure\n");
33db9fd4	167	mempool_free(recovery, pblk->rec_pool);
a4bd217b JG	168	goto out;
	169	}
	170
	171	INIT_WORK(&recovery->ws_rec, pblk_submit_rec);
	172	queue_work(pblk->kw_wq, &recovery->ws_rec);
	173
	174	out:
	175	pblk_complete_write(pblk, rqd, c_ctx);
	176	}
	177
	178	static void pblk_end_io_write(struct nvm_rq *rqd)
	179	{
	180	struct pblk *pblk = rqd->private;
	181	struct pblk_c_ctx *c_ctx = nvm_rq_to_pdu(rqd);
	182
	183	if (rqd->error) {
	184	pblk_log_write_err(pblk, rqd);
	185	return pblk_end_w_fail(pblk, rqd);
	186	}
	187	#ifdef CONFIG_NVM_DEBUG
	188	else
	189	WARN_ONCE(rqd->bio->bi_error, "pblk: corrupted write error\n");
	190	#endif
	191
	192	pblk_complete_write(pblk, rqd, c_ctx);
	193	}
	194
	195	static int pblk_alloc_w_rq(struct pblk pblk, struct nvm_rq rqd,
	196	unsigned int nr_secs)
	197	{
	198	struct nvm_tgt_dev *dev = pblk->dev;
	199
	200	/* Setup write request */
	201	rqd->opcode = NVM_OP_PWRITE;
	202	rqd->nr_ppas = nr_secs;
	203	rqd->flags = pblk_set_progr_mode(pblk, WRITE);
	204	rqd->private = pblk;
	205	rqd->end_io = pblk_end_io_write;
	206
	207	rqd->meta_list = nvm_dev_dma_alloc(dev->parent, GFP_KERNEL,
	208	&rqd->dma_meta_list);
	209	if (!rqd->meta_list)
	210	return -ENOMEM;
	211
	212	if (unlikely(nr_secs == 1))
	213	return 0;
	214
	215	rqd->ppa_list = rqd->meta_list + pblk_dma_meta_size;
	216	rqd->dma_ppa_list = rqd->dma_meta_list + pblk_dma_meta_size;
	217
	218	return 0;
	219	}
	220
	221	static int pblk_setup_w_rq(struct pblk pblk, struct nvm_rq rqd,
	222	struct pblk_c_ctx *c_ctx)
	223	{
	224	struct pblk_line_meta *lm = &pblk->lm;
	225	struct pblk_line *e_line = pblk_line_get_data_next(pblk);
	226	struct ppa_addr erase_ppa;
	227	unsigned int valid = c_ctx->nr_valid;
	228	unsigned int padded = c_ctx->nr_padded;
	229	unsigned int nr_secs = valid + padded;
	230	unsigned long *lun_bitmap;
	231	int ret = 0;
232
233	lun_bitmap = kzalloc(lm->lun_bitmap_len, GFP_KERNEL);
234	if (!lun_bitmap) {
235	ret = -ENOMEM;
236	goto out;
237	}
238	c_ctx->lun_bitmap = lun_bitmap;
239
240	ret = pblk_alloc_w_rq(pblk, rqd, nr_secs);
241	if (ret) {
242	kfree(lun_bitmap);
243	goto out;
244	}
245
246	ppa_set_empty(&erase_ppa);
a44f53fa	247	if (likely(!e_line \|\| !atomic_read(&e_line->left_eblks)))
a4bd217b JG	248	pblk_map_rq(pblk, rqd, c_ctx->sentry, lun_bitmap, valid, 0);
	249	else
	250	pblk_map_erase_rq(pblk, rqd, c_ctx->sentry, lun_bitmap,
	251	valid, &erase_ppa);
	252
	253	out:
	254	if (unlikely(e_line && !ppa_empty(erase_ppa))) {
	255	if (pblk_blk_erase_async(pblk, erase_ppa)) {
	256	struct nvm_tgt_dev *dev = pblk->dev;
	257	struct nvm_geo *geo = &dev->geo;
	258	int bit;
	259
a44f53fa	260	atomic_inc(&e_line->left_eblks);
a4bd217b JG	261	bit = erase_ppa.g.lun * geo->nr_chnls + erase_ppa.g.ch;
	262	WARN_ON(!test_and_clear_bit(bit, e_line->erase_bitmap));
	263	up(&pblk->erase_sem);
	264	}
	265	}
	266
	267	return ret;
	268	}
	269
	270	int pblk_setup_w_rec_rq(struct pblk pblk, struct nvm_rq rqd,
	271	struct pblk_c_ctx *c_ctx)
	272	{
	273	struct pblk_line_meta *lm = &pblk->lm;
	274	unsigned long *lun_bitmap;
	275	int ret;
	276
	277	lun_bitmap = kzalloc(lm->lun_bitmap_len, GFP_KERNEL);
	278	if (!lun_bitmap)
	279	return -ENOMEM;
	280
	281	c_ctx->lun_bitmap = lun_bitmap;
	282
	283	ret = pblk_alloc_w_rq(pblk, rqd, rqd->nr_ppas);
	284	if (ret)
	285	return ret;
	286
	287	pblk_map_rq(pblk, rqd, c_ctx->sentry, lun_bitmap, c_ctx->nr_valid, 0);
	288
	289	rqd->ppa_status = (u64)0;
	290	rqd->flags = pblk_set_progr_mode(pblk, WRITE);
	291
	292	return ret;
	293	}
	294
	295	static int pblk_calc_secs_to_sync(struct pblk *pblk, unsigned int secs_avail,
	296	unsigned int secs_to_flush)
	297	{
	298	int secs_to_sync;
	299
	300	secs_to_sync = pblk_calc_secs(pblk, secs_avail, secs_to_flush);
	301
	302	#ifdef CONFIG_NVM_DEBUG
	303	if ((!secs_to_sync && secs_to_flush)
	304	\|\| (secs_to_sync < 0)
	305	\|\| (secs_to_sync > secs_avail && !secs_to_flush)) {
	306	pr_err("pblk: bad sector calculation (a:%d,s:%d,f:%d)\n",
	307	secs_avail, secs_to_sync, secs_to_flush);
	308	}
	309	#endif
	310
	311	return secs_to_sync;
	312	}
	313
	314	static int pblk_submit_write(struct pblk *pblk)
	315	{
	316	struct bio *bio;
	317	struct nvm_rq *rqd;
	318	struct pblk_c_ctx *c_ctx;
	319	unsigned int pgs_read;
	320	unsigned int secs_avail, secs_to_sync, secs_to_com;
	321	unsigned int secs_to_flush;
	322	unsigned long pos;
	323	int err;
	324
325	/* If there are no sectors in the cache, flushes (bios without data)
326	* will be cleared on the cache threads
327	*/
328	secs_avail = pblk_rb_read_count(&pblk->rwb);
329	if (!secs_avail)
330	return 1;
331
332	secs_to_flush = pblk_rb_sync_point_count(&pblk->rwb);
333	if (!secs_to_flush && secs_avail < pblk->min_write_pgs)
334	return 1;
335
336	rqd = pblk_alloc_rqd(pblk, WRITE);
337	if (IS_ERR(rqd)) {
338	pr_err("pblk: cannot allocate write req.\n");
339	return 1;
340	}
341	c_ctx = nvm_rq_to_pdu(rqd);
342
343	bio = bio_alloc(GFP_KERNEL, pblk->max_write_pgs);
344	if (!bio) {
345	pr_err("pblk: cannot allocate write bio\n");
346	goto fail_free_rqd;
347	}
348	bio->bi_iter.bi_sector = 0; /* internal bio */
349	bio_set_op_attrs(bio, REQ_OP_WRITE, 0);
350	rqd->bio = bio;
351
352	secs_to_sync = pblk_calc_secs_to_sync(pblk, secs_avail, secs_to_flush);
353	if (secs_to_sync > pblk->max_write_pgs) {
354	pr_err("pblk: bad buffer sync calculation\n");
355	goto fail_put_bio;
356	}
357
358	secs_to_com = (secs_to_sync > secs_avail) ? secs_avail : secs_to_sync;
359	pos = pblk_rb_read_commit(&pblk->rwb, secs_to_com);
360
361	pgs_read = pblk_rb_read_to_bio(&pblk->rwb, bio, c_ctx, pos,
362	secs_to_sync, secs_avail);
363	if (!pgs_read) {
364	pr_err("pblk: corrupted write bio\n");
365	goto fail_put_bio;
366	}
367
368	if (c_ctx->nr_padded)
369	if (pblk_bio_add_pages(pblk, bio, GFP_KERNEL, c_ctx->nr_padded))
370	goto fail_put_bio;
371
372	/* Assign lbas to ppas and populate request structure */
373	err = pblk_setup_w_rq(pblk, rqd, c_ctx);
374	if (err) {
375	pr_err("pblk: could not setup write request\n");
376	goto fail_free_bio;
377	}
378
379	err = pblk_submit_io(pblk, rqd);
380	if (err) {
381	pr_err("pblk: I/O submission failed: %d\n", err);
382	goto fail_free_bio;
383	}
384
385	#ifdef CONFIG_NVM_DEBUG
386	atomic_long_add(secs_to_sync, &pblk->sub_writes);
387	#endif
388
389	return 0;
390
391	fail_free_bio:
392	if (c_ctx->nr_padded)
393	pblk_bio_free_pages(pblk, bio, secs_to_sync, c_ctx->nr_padded);
394	fail_put_bio:
395	bio_put(bio);
396	fail_free_rqd:
397	pblk_free_rqd(pblk, rqd, WRITE);
398
399	return 1;
400	}
401
402	int pblk_write_ts(void *data)
403	{
404	struct pblk *pblk = data;
405
406	while (!kthread_should_stop()) {
407	if (!pblk_submit_write(pblk))
408	continue;
409	set_current_state(TASK_INTERRUPTIBLE);
410	io_schedule();
411	}
412
413	return 0;
414	}