nvme-core.c 52.8 KB
Newer Older
M
Matthew Wilcox 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
/*
 * NVM Express device driver
 * Copyright (c) 2011, Intel Corporation.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope 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.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <linux/nvme.h>
#include <linux/bio.h>
21
#include <linux/bitops.h>
M
Matthew Wilcox 已提交
22
#include <linux/blkdev.h>
23
#include <linux/delay.h>
M
Matthew Wilcox 已提交
24 25 26
#include <linux/errno.h>
#include <linux/fs.h>
#include <linux/genhd.h>
27
#include <linux/idr.h>
M
Matthew Wilcox 已提交
28 29 30 31
#include <linux/init.h>
#include <linux/interrupt.h>
#include <linux/io.h>
#include <linux/kdev_t.h>
32
#include <linux/kthread.h>
M
Matthew Wilcox 已提交
33 34 35 36 37
#include <linux/kernel.h>
#include <linux/mm.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/pci.h>
38
#include <linux/poison.h>
M
Matthew Wilcox 已提交
39 40 41
#include <linux/sched.h>
#include <linux/slab.h>
#include <linux/types.h>
V
Vishal Verma 已提交
42
#include <scsi/sg.h>
43 44
#include <asm-generic/io-64-nonatomic-lo-hi.h>

M
Matthew Wilcox 已提交
45 46 47 48
#define NVME_Q_DEPTH 1024
#define SQ_SIZE(depth)		(depth * sizeof(struct nvme_command))
#define CQ_SIZE(depth)		(depth * sizeof(struct nvme_completion))
#define NVME_MINORS 64
49
#define ADMIN_TIMEOUT	(60 * HZ)
M
Matthew Wilcox 已提交
50 51 52 53

static int nvme_major;
module_param(nvme_major, int, 0);

54 55 56
static int use_threaded_interrupts;
module_param(use_threaded_interrupts, int, 0);

57 58 59 60
static DEFINE_SPINLOCK(dev_list_lock);
static LIST_HEAD(dev_list);
static struct task_struct *nvme_thread;

M
Matthew Wilcox 已提交
61 62 63 64 65 66
/*
 * An NVM Express queue.  Each device has at least two (one for admin
 * commands and one for I/O commands).
 */
struct nvme_queue {
	struct device *q_dmadev;
M
Matthew Wilcox 已提交
67
	struct nvme_dev *dev;
M
Matthew Wilcox 已提交
68 69 70 71 72 73
	spinlock_t q_lock;
	struct nvme_command *sq_cmds;
	volatile struct nvme_completion *cqes;
	dma_addr_t sq_dma_addr;
	dma_addr_t cq_dma_addr;
	wait_queue_head_t sq_full;
74
	wait_queue_t sq_cong_wait;
M
Matthew Wilcox 已提交
75 76 77 78 79 80 81
	struct bio_list sq_cong;
	u32 __iomem *q_db;
	u16 q_depth;
	u16 cq_vector;
	u16 sq_head;
	u16 sq_tail;
	u16 cq_head;
M
Matthew Wilcox 已提交
82
	u16 cq_phase;
M
Matthew Wilcox 已提交
83 84 85 86 87 88 89 90 91 92 93 94 95
	unsigned long cmdid_data[];
};

/*
 * Check we didin't inadvertently grow the command struct
 */
static inline void _nvme_check_size(void)
{
	BUILD_BUG_ON(sizeof(struct nvme_rw_command) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_create_cq) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_create_sq) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_delete_queue) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_features) != 64);
96
	BUILD_BUG_ON(sizeof(struct nvme_format_cmd) != 64);
M
Matthew Wilcox 已提交
97 98 99 100
	BUILD_BUG_ON(sizeof(struct nvme_command) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_id_ctrl) != 4096);
	BUILD_BUG_ON(sizeof(struct nvme_id_ns) != 4096);
	BUILD_BUG_ON(sizeof(struct nvme_lba_range_type) != 64);
K
Keith Busch 已提交
101
	BUILD_BUG_ON(sizeof(struct nvme_smart_log) != 512);
M
Matthew Wilcox 已提交
102 103
}

104
typedef void (*nvme_completion_fn)(struct nvme_dev *, void *,
105 106
						struct nvme_completion *);

107
struct nvme_cmd_info {
108 109
	nvme_completion_fn fn;
	void *ctx;
110 111 112 113 114 115 116 117
	unsigned long timeout;
};

static struct nvme_cmd_info *nvme_cmd_info(struct nvme_queue *nvmeq)
{
	return (void *)&nvmeq->cmdid_data[BITS_TO_LONGS(nvmeq->q_depth)];
}

M
Matthew Wilcox 已提交
118
/**
119 120 121
 * alloc_cmdid() - Allocate a Command ID
 * @nvmeq: The queue that will be used for this command
 * @ctx: A pointer that will be passed to the handler
122
 * @handler: The function to call on completion
M
Matthew Wilcox 已提交
123 124 125 126 127 128
 *
 * Allocate a Command ID for a queue.  The data passed in will
 * be passed to the completion handler.  This is implemented by using
 * the bottom two bits of the ctx pointer to store the handler ID.
 * Passing in a pointer that's not 4-byte aligned will cause a BUG.
 * We can change this if it becomes a problem.
129 130 131
 *
 * May be called with local interrupts disabled and the q_lock held,
 * or with interrupts enabled and no locks held.
M
Matthew Wilcox 已提交
132
 */
133 134
static int alloc_cmdid(struct nvme_queue *nvmeq, void *ctx,
				nvme_completion_fn handler, unsigned timeout)
M
Matthew Wilcox 已提交
135
{
136
	int depth = nvmeq->q_depth - 1;
137
	struct nvme_cmd_info *info = nvme_cmd_info(nvmeq);
M
Matthew Wilcox 已提交
138 139 140 141 142 143 144 145
	int cmdid;

	do {
		cmdid = find_first_zero_bit(nvmeq->cmdid_data, depth);
		if (cmdid >= depth)
			return -EBUSY;
	} while (test_and_set_bit(cmdid, nvmeq->cmdid_data));

146 147
	info[cmdid].fn = handler;
	info[cmdid].ctx = ctx;
148
	info[cmdid].timeout = jiffies + timeout;
M
Matthew Wilcox 已提交
149 150 151 152
	return cmdid;
}

static int alloc_cmdid_killable(struct nvme_queue *nvmeq, void *ctx,
153
				nvme_completion_fn handler, unsigned timeout)
M
Matthew Wilcox 已提交
154 155 156
{
	int cmdid;
	wait_event_killable(nvmeq->sq_full,
157
		(cmdid = alloc_cmdid(nvmeq, ctx, handler, timeout)) >= 0);
M
Matthew Wilcox 已提交
158 159 160
	return (cmdid < 0) ? -EINTR : cmdid;
}

161 162
/* Special values must be less than 0x1000 */
#define CMD_CTX_BASE		((void *)POISON_POINTER_DELTA)
163 164 165
#define CMD_CTX_CANCELLED	(0x30C + CMD_CTX_BASE)
#define CMD_CTX_COMPLETED	(0x310 + CMD_CTX_BASE)
#define CMD_CTX_INVALID		(0x314 + CMD_CTX_BASE)
M
Matthew Wilcox 已提交
166
#define CMD_CTX_FLUSH		(0x318 + CMD_CTX_BASE)
167

168
static void special_completion(struct nvme_dev *dev, void *ctx,
169 170 171 172 173 174 175
						struct nvme_completion *cqe)
{
	if (ctx == CMD_CTX_CANCELLED)
		return;
	if (ctx == CMD_CTX_FLUSH)
		return;
	if (ctx == CMD_CTX_COMPLETED) {
176
		dev_warn(&dev->pci_dev->dev,
177 178 179 180 181
				"completed id %d twice on queue %d\n",
				cqe->command_id, le16_to_cpup(&cqe->sq_id));
		return;
	}
	if (ctx == CMD_CTX_INVALID) {
182
		dev_warn(&dev->pci_dev->dev,
183 184 185 186 187
				"invalid id %d completed on queue %d\n",
				cqe->command_id, le16_to_cpup(&cqe->sq_id));
		return;
	}

188
	dev_warn(&dev->pci_dev->dev, "Unknown special completion %p\n", ctx);
189 190
}

191 192 193
/*
 * Called with local interrupts disabled and the q_lock held.  May not sleep.
 */
194 195
static void *free_cmdid(struct nvme_queue *nvmeq, int cmdid,
						nvme_completion_fn *fn)
M
Matthew Wilcox 已提交
196
{
197
	void *ctx;
198
	struct nvme_cmd_info *info = nvme_cmd_info(nvmeq);
M
Matthew Wilcox 已提交
199

200 201
	if (cmdid >= nvmeq->q_depth) {
		*fn = special_completion;
202
		return CMD_CTX_INVALID;
203
	}
204 205
	if (fn)
		*fn = info[cmdid].fn;
206 207
	ctx = info[cmdid].ctx;
	info[cmdid].fn = special_completion;
208
	info[cmdid].ctx = CMD_CTX_COMPLETED;
M
Matthew Wilcox 已提交
209 210
	clear_bit(cmdid, nvmeq->cmdid_data);
	wake_up(&nvmeq->sq_full);
211
	return ctx;
M
Matthew Wilcox 已提交
212 213
}

214 215
static void *cancel_cmdid(struct nvme_queue *nvmeq, int cmdid,
						nvme_completion_fn *fn)
216
{
217
	void *ctx;
218
	struct nvme_cmd_info *info = nvme_cmd_info(nvmeq);
219 220 221 222
	if (fn)
		*fn = info[cmdid].fn;
	ctx = info[cmdid].ctx;
	info[cmdid].fn = special_completion;
223
	info[cmdid].ctx = CMD_CTX_CANCELLED;
224
	return ctx;
225 226
}

V
Vishal Verma 已提交
227
struct nvme_queue *get_nvmeq(struct nvme_dev *dev)
M
Matthew Wilcox 已提交
228
{
229
	return dev->queues[get_cpu() + 1];
M
Matthew Wilcox 已提交
230 231
}

V
Vishal Verma 已提交
232
void put_nvmeq(struct nvme_queue *nvmeq)
M
Matthew Wilcox 已提交
233
{
M
Matthew Wilcox 已提交
234
	put_cpu();
M
Matthew Wilcox 已提交
235 236 237
}

/**
238
 * nvme_submit_cmd() - Copy a command into a queue and ring the doorbell
M
Matthew Wilcox 已提交
239 240 241 242 243 244 245 246 247 248 249 250 251 252
 * @nvmeq: The queue to use
 * @cmd: The command to send
 *
 * Safe to use from interrupt context
 */
static int nvme_submit_cmd(struct nvme_queue *nvmeq, struct nvme_command *cmd)
{
	unsigned long flags;
	u16 tail;
	spin_lock_irqsave(&nvmeq->q_lock, flags);
	tail = nvmeq->sq_tail;
	memcpy(&nvmeq->sq_cmds[tail], cmd, sizeof(*cmd));
	if (++tail == nvmeq->q_depth)
		tail = 0;
253
	writel(tail, nvmeq->q_db);
M
Matthew Wilcox 已提交
254 255 256 257 258 259
	nvmeq->sq_tail = tail;
	spin_unlock_irqrestore(&nvmeq->q_lock, flags);

	return 0;
}

260
static __le64 **iod_list(struct nvme_iod *iod)
261
{
262
	return ((void *)iod) + iod->offset;
263 264
}

265 266 267 268 269 270 271 272 273 274
/*
 * Will slightly overestimate the number of pages needed.  This is OK
 * as it only leads to a small amount of wasted memory for the lifetime of
 * the I/O.
 */
static int nvme_npages(unsigned size)
{
	unsigned nprps = DIV_ROUND_UP(size + PAGE_SIZE, PAGE_SIZE);
	return DIV_ROUND_UP(8 * nprps, PAGE_SIZE - 8);
}
M
Matthew Wilcox 已提交
275

276 277
static struct nvme_iod *
nvme_alloc_iod(unsigned nseg, unsigned nbytes, gfp_t gfp)
M
Matthew Wilcox 已提交
278
{
279 280 281 282 283 284 285 286
	struct nvme_iod *iod = kmalloc(sizeof(struct nvme_iod) +
				sizeof(__le64 *) * nvme_npages(nbytes) +
				sizeof(struct scatterlist) * nseg, gfp);

	if (iod) {
		iod->offset = offsetof(struct nvme_iod, sg[nseg]);
		iod->npages = -1;
		iod->length = nbytes;
K
Keith Busch 已提交
287
		iod->nents = 0;
K
Keith Busch 已提交
288
		iod->start_time = jiffies;
289 290 291
	}

	return iod;
M
Matthew Wilcox 已提交
292 293
}

V
Vishal Verma 已提交
294
void nvme_free_iod(struct nvme_dev *dev, struct nvme_iod *iod)
M
Matthew Wilcox 已提交
295
{
296 297 298 299 300 301 302 303 304 305 306 307 308 309
	const int last_prp = PAGE_SIZE / 8 - 1;
	int i;
	__le64 **list = iod_list(iod);
	dma_addr_t prp_dma = iod->first_dma;

	if (iod->npages == 0)
		dma_pool_free(dev->prp_small_pool, list[0], prp_dma);
	for (i = 0; i < iod->npages; i++) {
		__le64 *prp_list = list[i];
		dma_addr_t next_prp_dma = le64_to_cpu(prp_list[last_prp]);
		dma_pool_free(dev->prp_page_pool, prp_list, prp_dma);
		prp_dma = next_prp_dma;
	}
	kfree(iod);
M
Matthew Wilcox 已提交
310 311
}

K
Keith Busch 已提交
312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335
static void nvme_start_io_acct(struct bio *bio)
{
	struct gendisk *disk = bio->bi_bdev->bd_disk;
	const int rw = bio_data_dir(bio);
	int cpu = part_stat_lock();
	part_round_stats(cpu, &disk->part0);
	part_stat_inc(cpu, &disk->part0, ios[rw]);
	part_stat_add(cpu, &disk->part0, sectors[rw], bio_sectors(bio));
	part_inc_in_flight(&disk->part0, rw);
	part_stat_unlock();
}

static void nvme_end_io_acct(struct bio *bio, unsigned long start_time)
{
	struct gendisk *disk = bio->bi_bdev->bd_disk;
	const int rw = bio_data_dir(bio);
	unsigned long duration = jiffies - start_time;
	int cpu = part_stat_lock();
	part_stat_add(cpu, &disk->part0, ticks[rw], duration);
	part_round_stats(cpu, &disk->part0);
	part_dec_in_flight(&disk->part0, rw);
	part_stat_unlock();
}

336
static void bio_completion(struct nvme_dev *dev, void *ctx,
M
Matthew Wilcox 已提交
337 338
						struct nvme_completion *cqe)
{
339 340
	struct nvme_iod *iod = ctx;
	struct bio *bio = iod->private;
M
Matthew Wilcox 已提交
341 342
	u16 status = le16_to_cpup(&cqe->status) >> 1;

K
Keith Busch 已提交
343 344
	if (iod->nents)
		dma_unmap_sg(&dev->pci_dev->dev, iod->sg, iod->nents,
M
Matthew Wilcox 已提交
345
			bio_data_dir(bio) ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
K
Keith Busch 已提交
346 347

	nvme_end_io_acct(bio, iod->start_time);
348
	nvme_free_iod(dev, iod);
349
	if (status)
350
		bio_endio(bio, -EIO);
351
	else
352
		bio_endio(bio, 0);
M
Matthew Wilcox 已提交
353 354
}

355
/* length is in bytes.  gfp flags indicates whether we may sleep. */
V
Vishal Verma 已提交
356 357
int nvme_setup_prps(struct nvme_dev *dev, struct nvme_common_command *cmd,
			struct nvme_iod *iod, int total_len, gfp_t gfp)
M
Matthew Wilcox 已提交
358
{
359
	struct dma_pool *pool;
360 361
	int length = total_len;
	struct scatterlist *sg = iod->sg;
M
Matthew Wilcox 已提交
362 363 364
	int dma_len = sg_dma_len(sg);
	u64 dma_addr = sg_dma_address(sg);
	int offset = offset_in_page(dma_addr);
365
	__le64 *prp_list;
366
	__le64 **list = iod_list(iod);
367
	dma_addr_t prp_dma;
368
	int nprps, i;
M
Matthew Wilcox 已提交
369 370 371 372

	cmd->prp1 = cpu_to_le64(dma_addr);
	length -= (PAGE_SIZE - offset);
	if (length <= 0)
373
		return total_len;
M
Matthew Wilcox 已提交
374 375 376 377 378 379 380 381 382 383 384 385

	dma_len -= (PAGE_SIZE - offset);
	if (dma_len) {
		dma_addr += (PAGE_SIZE - offset);
	} else {
		sg = sg_next(sg);
		dma_addr = sg_dma_address(sg);
		dma_len = sg_dma_len(sg);
	}

	if (length <= PAGE_SIZE) {
		cmd->prp2 = cpu_to_le64(dma_addr);
386
		return total_len;
387 388 389
	}

	nprps = DIV_ROUND_UP(length, PAGE_SIZE);
390 391
	if (nprps <= (256 / 8)) {
		pool = dev->prp_small_pool;
392
		iod->npages = 0;
393 394
	} else {
		pool = dev->prp_page_pool;
395
		iod->npages = 1;
396 397
	}

398 399 400
	prp_list = dma_pool_alloc(pool, gfp, &prp_dma);
	if (!prp_list) {
		cmd->prp2 = cpu_to_le64(dma_addr);
401 402
		iod->npages = -1;
		return (total_len - length) + PAGE_SIZE;
403
	}
404 405
	list[0] = prp_list;
	iod->first_dma = prp_dma;
406 407 408
	cmd->prp2 = cpu_to_le64(prp_dma);
	i = 0;
	for (;;) {
409
		if (i == PAGE_SIZE / 8) {
410
			__le64 *old_prp_list = prp_list;
411
			prp_list = dma_pool_alloc(pool, gfp, &prp_dma);
412 413 414
			if (!prp_list)
				return total_len - length;
			list[iod->npages++] = prp_list;
415 416 417
			prp_list[0] = old_prp_list[i - 1];
			old_prp_list[i - 1] = cpu_to_le64(prp_dma);
			i = 1;
418 419 420 421 422 423 424 425 426 427 428 429 430
		}
		prp_list[i++] = cpu_to_le64(dma_addr);
		dma_len -= PAGE_SIZE;
		dma_addr += PAGE_SIZE;
		length -= PAGE_SIZE;
		if (length <= 0)
			break;
		if (dma_len > 0)
			continue;
		BUG_ON(dma_len < 0);
		sg = sg_next(sg);
		dma_addr = sg_dma_address(sg);
		dma_len = sg_dma_len(sg);
M
Matthew Wilcox 已提交
431 432
	}

433
	return total_len;
M
Matthew Wilcox 已提交
434 435
}

436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
struct nvme_bio_pair {
	struct bio b1, b2, *parent;
	struct bio_vec *bv1, *bv2;
	int err;
	atomic_t cnt;
};

static void nvme_bio_pair_endio(struct bio *bio, int err)
{
	struct nvme_bio_pair *bp = bio->bi_private;

	if (err)
		bp->err = err;

	if (atomic_dec_and_test(&bp->cnt)) {
		bio_endio(bp->parent, bp->err);
		if (bp->bv1)
			kfree(bp->bv1);
		if (bp->bv2)
			kfree(bp->bv2);
		kfree(bp);
	}
}

static struct nvme_bio_pair *nvme_bio_split(struct bio *bio, int idx,
							int len, int offset)
{
	struct nvme_bio_pair *bp;

	BUG_ON(len > bio->bi_size);
	BUG_ON(idx > bio->bi_vcnt);

	bp = kmalloc(sizeof(*bp), GFP_ATOMIC);
	if (!bp)
		return NULL;
	bp->err = 0;

	bp->b1 = *bio;
	bp->b2 = *bio;

	bp->b1.bi_size = len;
	bp->b2.bi_size -= len;
	bp->b1.bi_vcnt = idx;
	bp->b2.bi_idx = idx;
	bp->b2.bi_sector += len >> 9;

	if (offset) {
		bp->bv1 = kmalloc(bio->bi_max_vecs * sizeof(struct bio_vec),
								GFP_ATOMIC);
		if (!bp->bv1)
			goto split_fail_1;

		bp->bv2 = kmalloc(bio->bi_max_vecs * sizeof(struct bio_vec),
								GFP_ATOMIC);
		if (!bp->bv2)
			goto split_fail_2;

		memcpy(bp->bv1, bio->bi_io_vec,
			bio->bi_max_vecs * sizeof(struct bio_vec));
		memcpy(bp->bv2, bio->bi_io_vec,
			bio->bi_max_vecs * sizeof(struct bio_vec));

		bp->b1.bi_io_vec = bp->bv1;
		bp->b2.bi_io_vec = bp->bv2;
		bp->b2.bi_io_vec[idx].bv_offset += offset;
		bp->b2.bi_io_vec[idx].bv_len -= offset;
		bp->b1.bi_io_vec[idx].bv_len = offset;
		bp->b1.bi_vcnt++;
	} else
		bp->bv1 = bp->bv2 = NULL;

	bp->b1.bi_private = bp;
	bp->b2.bi_private = bp;

	bp->b1.bi_end_io = nvme_bio_pair_endio;
	bp->b2.bi_end_io = nvme_bio_pair_endio;

	bp->parent = bio;
	atomic_set(&bp->cnt, 2);

	return bp;

 split_fail_2:
	kfree(bp->bv1);
 split_fail_1:
	kfree(bp);
	return NULL;
}

static int nvme_split_and_submit(struct bio *bio, struct nvme_queue *nvmeq,
						int idx, int len, int offset)
{
	struct nvme_bio_pair *bp = nvme_bio_split(bio, idx, len, offset);
	if (!bp)
		return -ENOMEM;

	if (bio_list_empty(&nvmeq->sq_cong))
		add_wait_queue(&nvmeq->sq_full, &nvmeq->sq_cong_wait);
	bio_list_add(&nvmeq->sq_cong, &bp->b1);
	bio_list_add(&nvmeq->sq_cong, &bp->b2);

	return 0;
}

540 541 542 543
/* NVMe scatterlists require no holes in the virtual address */
#define BIOVEC_NOT_VIRT_MERGEABLE(vec1, vec2)	((vec2)->bv_offset || \
			(((vec1)->bv_offset + (vec1)->bv_len) % PAGE_SIZE))

544
static int nvme_map_bio(struct nvme_queue *nvmeq, struct nvme_iod *iod,
M
Matthew Wilcox 已提交
545 546
		struct bio *bio, enum dma_data_direction dma_dir, int psegs)
{
547 548
	struct bio_vec *bvec, *bvprv = NULL;
	struct scatterlist *sg = NULL;
549 550 551 552 553
	int i, length = 0, nsegs = 0, split_len = bio->bi_size;

	if (nvmeq->dev->stripe_size)
		split_len = nvmeq->dev->stripe_size -
			((bio->bi_sector << 9) & (nvmeq->dev->stripe_size - 1));
M
Matthew Wilcox 已提交
554

555
	sg_init_table(iod->sg, psegs);
M
Matthew Wilcox 已提交
556
	bio_for_each_segment(bvec, bio, i) {
557 558 559
		if (bvprv && BIOVEC_PHYS_MERGEABLE(bvprv, bvec)) {
			sg->length += bvec->bv_len;
		} else {
560
			if (bvprv && BIOVEC_NOT_VIRT_MERGEABLE(bvprv, bvec))
561 562 563
				return nvme_split_and_submit(bio, nvmeq, i,
								length, 0);

564
			sg = sg ? sg + 1 : iod->sg;
565 566 567 568
			sg_set_page(sg, bvec->bv_page, bvec->bv_len,
							bvec->bv_offset);
			nsegs++;
		}
569 570 571 572

		if (split_len - length < bvec->bv_len)
			return nvme_split_and_submit(bio, nvmeq, i, split_len,
							split_len - length);
573
		length += bvec->bv_len;
574
		bvprv = bvec;
M
Matthew Wilcox 已提交
575
	}
576
	iod->nents = nsegs;
577
	sg_mark_end(sg);
578
	if (dma_map_sg(nvmeq->q_dmadev, iod->sg, iod->nents, dma_dir) == 0)
579
		return -ENOMEM;
580

581
	BUG_ON(length != bio->bi_size);
582
	return length;
M
Matthew Wilcox 已提交
583 584
}

585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605
/*
 * We reuse the small pool to allocate the 16-byte range here as it is not
 * worth having a special pool for these or additional cases to handle freeing
 * the iod.
 */
static int nvme_submit_discard(struct nvme_queue *nvmeq, struct nvme_ns *ns,
		struct bio *bio, struct nvme_iod *iod, int cmdid)
{
	struct nvme_dsm_range *range;
	struct nvme_command *cmnd = &nvmeq->sq_cmds[nvmeq->sq_tail];

	range = dma_pool_alloc(nvmeq->dev->prp_small_pool, GFP_ATOMIC,
							&iod->first_dma);
	if (!range)
		return -ENOMEM;

	iod_list(iod)[0] = (__le64 *)range;
	iod->npages = 0;

	range->cattr = cpu_to_le32(0);
	range->nlb = cpu_to_le32(bio->bi_size >> ns->lba_shift);
606
	range->slba = cpu_to_le64(nvme_block_nr(ns, bio->bi_sector));
607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622

	memset(cmnd, 0, sizeof(*cmnd));
	cmnd->dsm.opcode = nvme_cmd_dsm;
	cmnd->dsm.command_id = cmdid;
	cmnd->dsm.nsid = cpu_to_le32(ns->ns_id);
	cmnd->dsm.prp1 = cpu_to_le64(iod->first_dma);
	cmnd->dsm.nr = 0;
	cmnd->dsm.attributes = cpu_to_le32(NVME_DSMGMT_AD);

	if (++nvmeq->sq_tail == nvmeq->q_depth)
		nvmeq->sq_tail = 0;
	writel(nvmeq->sq_tail, nvmeq->q_db);

	return 0;
}

M
Matthew Wilcox 已提交
623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
static int nvme_submit_flush(struct nvme_queue *nvmeq, struct nvme_ns *ns,
								int cmdid)
{
	struct nvme_command *cmnd = &nvmeq->sq_cmds[nvmeq->sq_tail];

	memset(cmnd, 0, sizeof(*cmnd));
	cmnd->common.opcode = nvme_cmd_flush;
	cmnd->common.command_id = cmdid;
	cmnd->common.nsid = cpu_to_le32(ns->ns_id);

	if (++nvmeq->sq_tail == nvmeq->q_depth)
		nvmeq->sq_tail = 0;
	writel(nvmeq->sq_tail, nvmeq->q_db);

	return 0;
}

V
Vishal Verma 已提交
640
int nvme_submit_flush_data(struct nvme_queue *nvmeq, struct nvme_ns *ns)
M
Matthew Wilcox 已提交
641 642
{
	int cmdid = alloc_cmdid(nvmeq, (void *)CMD_CTX_FLUSH,
643
					special_completion, NVME_IO_TIMEOUT);
M
Matthew Wilcox 已提交
644 645 646 647 648 649
	if (unlikely(cmdid < 0))
		return cmdid;

	return nvme_submit_flush(nvmeq, ns, cmdid);
}

650 651 652
/*
 * Called with local interrupts disabled and the q_lock held.  May not sleep.
 */
M
Matthew Wilcox 已提交
653 654 655
static int nvme_submit_bio_queue(struct nvme_queue *nvmeq, struct nvme_ns *ns,
								struct bio *bio)
{
M
Matthew Wilcox 已提交
656
	struct nvme_command *cmnd;
657
	struct nvme_iod *iod;
M
Matthew Wilcox 已提交
658
	enum dma_data_direction dma_dir;
659
	int cmdid, length, result;
M
Matthew Wilcox 已提交
660 661 662 663
	u16 control;
	u32 dsmgmt;
	int psegs = bio_phys_segments(ns->queue, bio);

M
Matthew Wilcox 已提交
664 665 666 667 668 669
	if ((bio->bi_rw & REQ_FLUSH) && psegs) {
		result = nvme_submit_flush_data(nvmeq, ns);
		if (result)
			return result;
	}

670
	result = -ENOMEM;
671 672
	iod = nvme_alloc_iod(psegs, bio->bi_size, GFP_ATOMIC);
	if (!iod)
673
		goto nomem;
674
	iod->private = bio;
M
Matthew Wilcox 已提交
675

676
	result = -EBUSY;
677
	cmdid = alloc_cmdid(nvmeq, iod, bio_completion, NVME_IO_TIMEOUT);
M
Matthew Wilcox 已提交
678
	if (unlikely(cmdid < 0))
679
		goto free_iod;
M
Matthew Wilcox 已提交
680

681 682 683 684 685 686
	if (bio->bi_rw & REQ_DISCARD) {
		result = nvme_submit_discard(nvmeq, ns, bio, iod, cmdid);
		if (result)
			goto free_cmdid;
		return result;
	}
M
Matthew Wilcox 已提交
687 688 689
	if ((bio->bi_rw & REQ_FLUSH) && !psegs)
		return nvme_submit_flush(nvmeq, ns, cmdid);

M
Matthew Wilcox 已提交
690 691 692 693 694 695 696 697 698 699
	control = 0;
	if (bio->bi_rw & REQ_FUA)
		control |= NVME_RW_FUA;
	if (bio->bi_rw & (REQ_FAILFAST_DEV | REQ_RAHEAD))
		control |= NVME_RW_LR;

	dsmgmt = 0;
	if (bio->bi_rw & REQ_RAHEAD)
		dsmgmt |= NVME_RW_DSM_FREQ_PREFETCH;

M
Matthew Wilcox 已提交
700
	cmnd = &nvmeq->sq_cmds[nvmeq->sq_tail];
M
Matthew Wilcox 已提交
701

702
	memset(cmnd, 0, sizeof(*cmnd));
M
Matthew Wilcox 已提交
703
	if (bio_data_dir(bio)) {
M
Matthew Wilcox 已提交
704
		cmnd->rw.opcode = nvme_cmd_write;
M
Matthew Wilcox 已提交
705 706
		dma_dir = DMA_TO_DEVICE;
	} else {
M
Matthew Wilcox 已提交
707
		cmnd->rw.opcode = nvme_cmd_read;
M
Matthew Wilcox 已提交
708 709 710
		dma_dir = DMA_FROM_DEVICE;
	}

711 712
	result = nvme_map_bio(nvmeq, iod, bio, dma_dir, psegs);
	if (result <= 0)
713
		goto free_cmdid;
714
	length = result;
M
Matthew Wilcox 已提交
715

M
Matthew Wilcox 已提交
716 717
	cmnd->rw.command_id = cmdid;
	cmnd->rw.nsid = cpu_to_le32(ns->ns_id);
718 719
	length = nvme_setup_prps(nvmeq->dev, &cmnd->common, iod, length,
								GFP_ATOMIC);
720
	cmnd->rw.slba = cpu_to_le64(nvme_block_nr(ns, bio->bi_sector));
721
	cmnd->rw.length = cpu_to_le16((length >> ns->lba_shift) - 1);
M
Matthew Wilcox 已提交
722 723
	cmnd->rw.control = cpu_to_le16(control);
	cmnd->rw.dsmgmt = cpu_to_le32(dsmgmt);
M
Matthew Wilcox 已提交
724

K
Keith Busch 已提交
725
	nvme_start_io_acct(bio);
M
Matthew Wilcox 已提交
726 727
	if (++nvmeq->sq_tail == nvmeq->q_depth)
		nvmeq->sq_tail = 0;
728
	writel(nvmeq->sq_tail, nvmeq->q_db);
M
Matthew Wilcox 已提交
729

730 731
	return 0;

732 733
 free_cmdid:
	free_cmdid(nvmeq, cmdid, NULL);
734 735
 free_iod:
	nvme_free_iod(nvmeq->dev, iod);
736 737
 nomem:
	return result;
M
Matthew Wilcox 已提交
738 739
}

740
static void nvme_make_request(struct request_queue *q, struct bio *bio)
M
Matthew Wilcox 已提交
741 742
{
	struct nvme_ns *ns = q->queuedata;
743
	struct nvme_queue *nvmeq = get_nvmeq(ns->dev);
744 745 746 747 748 749 750 751
	int result = -EBUSY;

	spin_lock_irq(&nvmeq->q_lock);
	if (bio_list_empty(&nvmeq->sq_cong))
		result = nvme_submit_bio_queue(nvmeq, ns, bio);
	if (unlikely(result)) {
		if (bio_list_empty(&nvmeq->sq_cong))
			add_wait_queue(&nvmeq->sq_full, &nvmeq->sq_cong_wait);
M
Matthew Wilcox 已提交
752 753
		bio_list_add(&nvmeq->sq_cong, bio);
	}
754 755

	spin_unlock_irq(&nvmeq->q_lock);
M
Matthew Wilcox 已提交
756 757 758 759 760
	put_nvmeq(nvmeq);
}

static irqreturn_t nvme_process_cq(struct nvme_queue *nvmeq)
{
M
Matthew Wilcox 已提交
761
	u16 head, phase;
M
Matthew Wilcox 已提交
762 763

	head = nvmeq->cq_head;
M
Matthew Wilcox 已提交
764
	phase = nvmeq->cq_phase;
M
Matthew Wilcox 已提交
765 766

	for (;;) {
767 768
		void *ctx;
		nvme_completion_fn fn;
M
Matthew Wilcox 已提交
769
		struct nvme_completion cqe = nvmeq->cqes[head];
M
Matthew Wilcox 已提交
770
		if ((le16_to_cpu(cqe.status) & 1) != phase)
M
Matthew Wilcox 已提交
771 772 773 774
			break;
		nvmeq->sq_head = le16_to_cpu(cqe.sq_head);
		if (++head == nvmeq->q_depth) {
			head = 0;
M
Matthew Wilcox 已提交
775
			phase = !phase;
M
Matthew Wilcox 已提交
776 777
		}

778
		ctx = free_cmdid(nvmeq, cqe.command_id, &fn);
779
		fn(nvmeq->dev, ctx, &cqe);
M
Matthew Wilcox 已提交
780 781 782 783 784 785 786 787
	}

	/* If the controller ignores the cq head doorbell and continuously
	 * writes to the queue, it is theoretically possible to wrap around
	 * the queue twice and mistakenly return IRQ_NONE.  Linux only
	 * requires that 0.1% of your interrupts are handled, so this isn't
	 * a big problem.
	 */
M
Matthew Wilcox 已提交
788
	if (head == nvmeq->cq_head && phase == nvmeq->cq_phase)
M
Matthew Wilcox 已提交
789 790
		return IRQ_NONE;

791
	writel(head, nvmeq->q_db + (1 << nvmeq->dev->db_stride));
M
Matthew Wilcox 已提交
792
	nvmeq->cq_head = head;
M
Matthew Wilcox 已提交
793
	nvmeq->cq_phase = phase;
M
Matthew Wilcox 已提交
794 795 796 797 798

	return IRQ_HANDLED;
}

static irqreturn_t nvme_irq(int irq, void *data)
799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816
{
	irqreturn_t result;
	struct nvme_queue *nvmeq = data;
	spin_lock(&nvmeq->q_lock);
	result = nvme_process_cq(nvmeq);
	spin_unlock(&nvmeq->q_lock);
	return result;
}

static irqreturn_t nvme_irq_check(int irq, void *data)
{
	struct nvme_queue *nvmeq = data;
	struct nvme_completion cqe = nvmeq->cqes[nvmeq->cq_head];
	if ((le16_to_cpu(cqe.status) & 1) != nvmeq->cq_phase)
		return IRQ_NONE;
	return IRQ_WAKE_THREAD;
}

817 818 819
static void nvme_abort_command(struct nvme_queue *nvmeq, int cmdid)
{
	spin_lock_irq(&nvmeq->q_lock);
820
	cancel_cmdid(nvmeq, cmdid, NULL);
821 822 823
	spin_unlock_irq(&nvmeq->q_lock);
}

824 825 826 827 828 829
struct sync_cmd_info {
	struct task_struct *task;
	u32 result;
	int status;
};

830
static void sync_completion(struct nvme_dev *dev, void *ctx,
831 832 833 834 835 836 837 838
						struct nvme_completion *cqe)
{
	struct sync_cmd_info *cmdinfo = ctx;
	cmdinfo->result = le32_to_cpup(&cqe->result);
	cmdinfo->status = le16_to_cpup(&cqe->status) >> 1;
	wake_up_process(cmdinfo->task);
}

M
Matthew Wilcox 已提交
839 840 841 842
/*
 * Returns 0 on success.  If the result is negative, it's a Linux error code;
 * if the result is positive, it's an NVM Express status code
 */
V
Vishal Verma 已提交
843 844
int nvme_submit_sync_cmd(struct nvme_queue *nvmeq, struct nvme_command *cmd,
						u32 *result, unsigned timeout)
M
Matthew Wilcox 已提交
845 846 847 848 849 850 851
{
	int cmdid;
	struct sync_cmd_info cmdinfo;

	cmdinfo.task = current;
	cmdinfo.status = -EINTR;

852
	cmdid = alloc_cmdid_killable(nvmeq, &cmdinfo, sync_completion,
853
								timeout);
M
Matthew Wilcox 已提交
854 855 856 857
	if (cmdid < 0)
		return cmdid;
	cmd->common.command_id = cmdid;

858 859
	set_current_state(TASK_KILLABLE);
	nvme_submit_cmd(nvmeq, cmd);
860
	schedule_timeout(timeout);
M
Matthew Wilcox 已提交
861

862 863 864 865 866
	if (cmdinfo.status == -EINTR) {
		nvme_abort_command(nvmeq, cmdid);
		return -EINTR;
	}

M
Matthew Wilcox 已提交
867 868 869 870 871 872
	if (result)
		*result = cmdinfo.result;

	return cmdinfo.status;
}

V
Vishal Verma 已提交
873
int nvme_submit_admin_cmd(struct nvme_dev *dev, struct nvme_command *cmd,
M
Matthew Wilcox 已提交
874 875
								u32 *result)
{
876
	return nvme_submit_sync_cmd(dev->queues[0], cmd, result, ADMIN_TIMEOUT);
M
Matthew Wilcox 已提交
877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945
}

static int adapter_delete_queue(struct nvme_dev *dev, u8 opcode, u16 id)
{
	int status;
	struct nvme_command c;

	memset(&c, 0, sizeof(c));
	c.delete_queue.opcode = opcode;
	c.delete_queue.qid = cpu_to_le16(id);

	status = nvme_submit_admin_cmd(dev, &c, NULL);
	if (status)
		return -EIO;
	return 0;
}

static int adapter_alloc_cq(struct nvme_dev *dev, u16 qid,
						struct nvme_queue *nvmeq)
{
	int status;
	struct nvme_command c;
	int flags = NVME_QUEUE_PHYS_CONTIG | NVME_CQ_IRQ_ENABLED;

	memset(&c, 0, sizeof(c));
	c.create_cq.opcode = nvme_admin_create_cq;
	c.create_cq.prp1 = cpu_to_le64(nvmeq->cq_dma_addr);
	c.create_cq.cqid = cpu_to_le16(qid);
	c.create_cq.qsize = cpu_to_le16(nvmeq->q_depth - 1);
	c.create_cq.cq_flags = cpu_to_le16(flags);
	c.create_cq.irq_vector = cpu_to_le16(nvmeq->cq_vector);

	status = nvme_submit_admin_cmd(dev, &c, NULL);
	if (status)
		return -EIO;
	return 0;
}

static int adapter_alloc_sq(struct nvme_dev *dev, u16 qid,
						struct nvme_queue *nvmeq)
{
	int status;
	struct nvme_command c;
	int flags = NVME_QUEUE_PHYS_CONTIG | NVME_SQ_PRIO_MEDIUM;

	memset(&c, 0, sizeof(c));
	c.create_sq.opcode = nvme_admin_create_sq;
	c.create_sq.prp1 = cpu_to_le64(nvmeq->sq_dma_addr);
	c.create_sq.sqid = cpu_to_le16(qid);
	c.create_sq.qsize = cpu_to_le16(nvmeq->q_depth - 1);
	c.create_sq.sq_flags = cpu_to_le16(flags);
	c.create_sq.cqid = cpu_to_le16(qid);

	status = nvme_submit_admin_cmd(dev, &c, NULL);
	if (status)
		return -EIO;
	return 0;
}

static int adapter_delete_cq(struct nvme_dev *dev, u16 cqid)
{
	return adapter_delete_queue(dev, nvme_admin_delete_cq, cqid);
}

static int adapter_delete_sq(struct nvme_dev *dev, u16 sqid)
{
	return adapter_delete_queue(dev, nvme_admin_delete_sq, sqid);
}

V
Vishal Verma 已提交
946
int nvme_identify(struct nvme_dev *dev, unsigned nsid, unsigned cns,
947 948 949 950 951 952 953 954 955 956 957 958 959
							dma_addr_t dma_addr)
{
	struct nvme_command c;

	memset(&c, 0, sizeof(c));
	c.identify.opcode = nvme_admin_identify;
	c.identify.nsid = cpu_to_le32(nsid);
	c.identify.prp1 = cpu_to_le64(dma_addr);
	c.identify.cns = cpu_to_le32(cns);

	return nvme_submit_admin_cmd(dev, &c, NULL);
}

V
Vishal Verma 已提交
960
int nvme_get_features(struct nvme_dev *dev, unsigned fid, unsigned nsid,
961
					dma_addr_t dma_addr, u32 *result)
962 963 964 965 966
{
	struct nvme_command c;

	memset(&c, 0, sizeof(c));
	c.features.opcode = nvme_admin_get_features;
967
	c.features.nsid = cpu_to_le32(nsid);
968 969 970
	c.features.prp1 = cpu_to_le64(dma_addr);
	c.features.fid = cpu_to_le32(fid);

971
	return nvme_submit_admin_cmd(dev, &c, result);
972 973
}

V
Vishal Verma 已提交
974 975
int nvme_set_features(struct nvme_dev *dev, unsigned fid, unsigned dword11,
					dma_addr_t dma_addr, u32 *result)
976 977 978 979 980 981 982 983 984
{
	struct nvme_command c;

	memset(&c, 0, sizeof(c));
	c.features.opcode = nvme_admin_set_features;
	c.features.prp1 = cpu_to_le64(dma_addr);
	c.features.fid = cpu_to_le32(fid);
	c.features.dword11 = cpu_to_le32(dword11);

985 986 987
	return nvme_submit_admin_cmd(dev, &c, result);
}

988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003
/**
 * nvme_cancel_ios - Cancel outstanding I/Os
 * @queue: The queue to cancel I/Os on
 * @timeout: True to only cancel I/Os which have timed out
 */
static void nvme_cancel_ios(struct nvme_queue *nvmeq, bool timeout)
{
	int depth = nvmeq->q_depth - 1;
	struct nvme_cmd_info *info = nvme_cmd_info(nvmeq);
	unsigned long now = jiffies;
	int cmdid;

	for_each_set_bit(cmdid, nvmeq->cmdid_data, depth) {
		void *ctx;
		nvme_completion_fn fn;
		static struct nvme_completion cqe = {
1004
			.status = cpu_to_le16(NVME_SC_ABORT_REQ << 1),
1005 1006 1007 1008
		};

		if (timeout && !time_after(now, info[cmdid].timeout))
			continue;
1009 1010
		if (info[cmdid].ctx == CMD_CTX_CANCELLED)
			continue;
1011 1012 1013 1014 1015 1016
		dev_warn(nvmeq->q_dmadev, "Cancelling I/O %d\n", cmdid);
		ctx = cancel_cmdid(nvmeq, cmdid, &fn);
		fn(nvmeq->dev, ctx, &cqe);
	}
}

1017 1018 1019 1020 1021 1022 1023 1024 1025
static void nvme_free_queue_mem(struct nvme_queue *nvmeq)
{
	dma_free_coherent(nvmeq->q_dmadev, CQ_SIZE(nvmeq->q_depth),
				(void *)nvmeq->cqes, nvmeq->cq_dma_addr);
	dma_free_coherent(nvmeq->q_dmadev, SQ_SIZE(nvmeq->q_depth),
					nvmeq->sq_cmds, nvmeq->sq_dma_addr);
	kfree(nvmeq);
}

M
Matthew Wilcox 已提交
1026 1027 1028
static void nvme_free_queue(struct nvme_dev *dev, int qid)
{
	struct nvme_queue *nvmeq = dev->queues[qid];
M
Matthew Wilcox 已提交
1029
	int vector = dev->entry[nvmeq->cq_vector].vector;
M
Matthew Wilcox 已提交
1030

1031 1032
	spin_lock_irq(&nvmeq->q_lock);
	nvme_cancel_ios(nvmeq, false);
1033 1034 1035 1036
	while (bio_list_peek(&nvmeq->sq_cong)) {
		struct bio *bio = bio_list_pop(&nvmeq->sq_cong);
		bio_endio(bio, -EIO);
	}
1037 1038
	spin_unlock_irq(&nvmeq->q_lock);

M
Matthew Wilcox 已提交
1039 1040
	irq_set_affinity_hint(vector, NULL);
	free_irq(vector, nvmeq);
M
Matthew Wilcox 已提交
1041 1042 1043 1044 1045 1046 1047

	/* Don't tell the adapter to delete the admin queue */
	if (qid) {
		adapter_delete_sq(dev, qid);
		adapter_delete_cq(dev, qid);
	}

1048
	nvme_free_queue_mem(nvmeq);
M
Matthew Wilcox 已提交
1049 1050 1051 1052 1053 1054
}

static struct nvme_queue *nvme_alloc_queue(struct nvme_dev *dev, int qid,
							int depth, int vector)
{
	struct device *dmadev = &dev->pci_dev->dev;
1055 1056
	unsigned extra = DIV_ROUND_UP(depth, 8) + (depth *
						sizeof(struct nvme_cmd_info));
M
Matthew Wilcox 已提交
1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072
	struct nvme_queue *nvmeq = kzalloc(sizeof(*nvmeq) + extra, GFP_KERNEL);
	if (!nvmeq)
		return NULL;

	nvmeq->cqes = dma_alloc_coherent(dmadev, CQ_SIZE(depth),
					&nvmeq->cq_dma_addr, GFP_KERNEL);
	if (!nvmeq->cqes)
		goto free_nvmeq;
	memset((void *)nvmeq->cqes, 0, CQ_SIZE(depth));

	nvmeq->sq_cmds = dma_alloc_coherent(dmadev, SQ_SIZE(depth),
					&nvmeq->sq_dma_addr, GFP_KERNEL);
	if (!nvmeq->sq_cmds)
		goto free_cqdma;

	nvmeq->q_dmadev = dmadev;
M
Matthew Wilcox 已提交
1073
	nvmeq->dev = dev;
M
Matthew Wilcox 已提交
1074 1075
	spin_lock_init(&nvmeq->q_lock);
	nvmeq->cq_head = 0;
M
Matthew Wilcox 已提交
1076
	nvmeq->cq_phase = 1;
M
Matthew Wilcox 已提交
1077
	init_waitqueue_head(&nvmeq->sq_full);
1078
	init_waitqueue_entry(&nvmeq->sq_cong_wait, nvme_thread);
M
Matthew Wilcox 已提交
1079
	bio_list_init(&nvmeq->sq_cong);
1080
	nvmeq->q_db = &dev->dbs[qid << (dev->db_stride + 1)];
M
Matthew Wilcox 已提交
1081 1082 1083 1084 1085 1086
	nvmeq->q_depth = depth;
	nvmeq->cq_vector = vector;

	return nvmeq;

 free_cqdma:
1087
	dma_free_coherent(dmadev, CQ_SIZE(depth), (void *)nvmeq->cqes,
M
Matthew Wilcox 已提交
1088 1089 1090 1091 1092 1093
							nvmeq->cq_dma_addr);
 free_nvmeq:
	kfree(nvmeq);
	return NULL;
}

1094 1095 1096
static int queue_request_irq(struct nvme_dev *dev, struct nvme_queue *nvmeq,
							const char *name)
{
1097 1098
	if (use_threaded_interrupts)
		return request_threaded_irq(dev->entry[nvmeq->cq_vector].vector,
1099
					nvme_irq_check, nvme_irq,
1100 1101
					IRQF_DISABLED | IRQF_SHARED,
					name, nvmeq);
1102 1103 1104 1105
	return request_irq(dev->entry[nvmeq->cq_vector].vector, nvme_irq,
				IRQF_DISABLED | IRQF_SHARED, name, nvmeq);
}

1106 1107
static struct nvme_queue *nvme_create_queue(struct nvme_dev *dev, int qid,
					    int cq_size, int vector)
M
Matthew Wilcox 已提交
1108 1109 1110 1111
{
	int result;
	struct nvme_queue *nvmeq = nvme_alloc_queue(dev, qid, cq_size, vector);

1112
	if (!nvmeq)
1113
		return ERR_PTR(-ENOMEM);
1114

M
Matthew Wilcox 已提交
1115 1116 1117 1118 1119 1120 1121 1122
	result = adapter_alloc_cq(dev, qid, nvmeq);
	if (result < 0)
		goto free_nvmeq;

	result = adapter_alloc_sq(dev, qid, nvmeq);
	if (result < 0)
		goto release_cq;

1123
	result = queue_request_irq(dev, nvmeq, "nvme");
M
Matthew Wilcox 已提交
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
	if (result < 0)
		goto release_sq;

	return nvmeq;

 release_sq:
	adapter_delete_sq(dev, qid);
 release_cq:
	adapter_delete_cq(dev, qid);
 free_nvmeq:
	dma_free_coherent(nvmeq->q_dmadev, CQ_SIZE(nvmeq->q_depth),
				(void *)nvmeq->cqes, nvmeq->cq_dma_addr);
	dma_free_coherent(nvmeq->q_dmadev, SQ_SIZE(nvmeq->q_depth),
					nvmeq->sq_cmds, nvmeq->sq_dma_addr);
	kfree(nvmeq);
1139
	return ERR_PTR(result);
M
Matthew Wilcox 已提交
1140 1141
}

1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170
static int nvme_wait_ready(struct nvme_dev *dev, u64 cap, bool enabled)
{
	unsigned long timeout;
	u32 bit = enabled ? NVME_CSTS_RDY : 0;

	timeout = ((NVME_CAP_TIMEOUT(cap) + 1) * HZ / 2) + jiffies;

	while ((readl(&dev->bar->csts) & NVME_CSTS_RDY) != bit) {
		msleep(100);
		if (fatal_signal_pending(current))
			return -EINTR;
		if (time_after(jiffies, timeout)) {
			dev_err(&dev->pci_dev->dev,
				"Device not ready; aborting initialisation\n");
			return -ENODEV;
		}
	}

	return 0;
}

/*
 * If the device has been passed off to us in an enabled state, just clear
 * the enabled bit.  The spec says we should set the 'shutdown notification
 * bits', but doing so may cause the device to complete commands to the
 * admin queue ... and we don't know what memory that might be pointing at!
 */
static int nvme_disable_ctrl(struct nvme_dev *dev, u64 cap)
{
1171 1172 1173 1174
	u32 cc = readl(&dev->bar->cc);

	if (cc & NVME_CC_ENABLE)
		writel(cc & ~NVME_CC_ENABLE, &dev->bar->cc);
1175 1176 1177 1178 1179 1180 1181 1182
	return nvme_wait_ready(dev, cap, false);
}

static int nvme_enable_ctrl(struct nvme_dev *dev, u64 cap)
{
	return nvme_wait_ready(dev, cap, true);
}

1183
static int nvme_configure_admin_queue(struct nvme_dev *dev)
M
Matthew Wilcox 已提交
1184
{
1185
	int result;
M
Matthew Wilcox 已提交
1186
	u32 aqa;
1187
	u64 cap = readq(&dev->bar->cap);
M
Matthew Wilcox 已提交
1188 1189 1190
	struct nvme_queue *nvmeq;

	dev->dbs = ((void __iomem *)dev->bar) + 4096;
1191 1192 1193 1194 1195
	dev->db_stride = NVME_CAP_STRIDE(cap);

	result = nvme_disable_ctrl(dev, cap);
	if (result < 0)
		return result;
M
Matthew Wilcox 已提交
1196 1197

	nvmeq = nvme_alloc_queue(dev, 0, 64, 0);
1198 1199
	if (!nvmeq)
		return -ENOMEM;
M
Matthew Wilcox 已提交
1200 1201 1202 1203 1204 1205 1206

	aqa = nvmeq->q_depth - 1;
	aqa |= aqa << 16;

	dev->ctrl_config = NVME_CC_ENABLE | NVME_CC_CSS_NVM;
	dev->ctrl_config |= (PAGE_SHIFT - 12) << NVME_CC_MPS_SHIFT;
	dev->ctrl_config |= NVME_CC_ARB_RR | NVME_CC_SHN_NONE;
1207
	dev->ctrl_config |= NVME_CC_IOSQES | NVME_CC_IOCQES;
M
Matthew Wilcox 已提交
1208 1209 1210 1211 1212 1213

	writel(aqa, &dev->bar->aqa);
	writeq(nvmeq->sq_dma_addr, &dev->bar->asq);
	writeq(nvmeq->cq_dma_addr, &dev->bar->acq);
	writel(dev->ctrl_config, &dev->bar->cc);

1214
	result = nvme_enable_ctrl(dev, cap);
1215 1216
	if (result)
		goto free_q;
1217

1218
	result = queue_request_irq(dev, nvmeq, "nvme admin");
1219 1220 1221
	if (result)
		goto free_q;

M
Matthew Wilcox 已提交
1222 1223
	dev->queues[0] = nvmeq;
	return result;
1224 1225 1226 1227

 free_q:
	nvme_free_queue_mem(nvmeq);
	return result;
M
Matthew Wilcox 已提交
1228 1229
}

V
Vishal Verma 已提交
1230
struct nvme_iod *nvme_map_user_pages(struct nvme_dev *dev, int write,
1231
				unsigned long addr, unsigned length)
M
Matthew Wilcox 已提交
1232
{
1233
	int i, err, count, nents, offset;
1234 1235
	struct scatterlist *sg;
	struct page **pages;
1236
	struct nvme_iod *iod;
1237 1238

	if (addr & 3)
1239
		return ERR_PTR(-EINVAL);
1240
	if (!length || length > INT_MAX - PAGE_SIZE)
1241
		return ERR_PTR(-EINVAL);
1242

1243
	offset = offset_in_page(addr);
1244 1245
	count = DIV_ROUND_UP(offset + length, PAGE_SIZE);
	pages = kcalloc(count, sizeof(*pages), GFP_KERNEL);
1246 1247
	if (!pages)
		return ERR_PTR(-ENOMEM);
1248 1249 1250 1251 1252 1253 1254

	err = get_user_pages_fast(addr, count, 1, pages);
	if (err < count) {
		count = err;
		err = -EFAULT;
		goto put_pages;
	}
1255

1256 1257
	iod = nvme_alloc_iod(count, length, GFP_KERNEL);
	sg = iod->sg;
1258
	sg_init_table(sg, count);
1259 1260
	for (i = 0; i < count; i++) {
		sg_set_page(&sg[i], pages[i],
1261 1262
			    min_t(unsigned, length, PAGE_SIZE - offset),
			    offset);
1263 1264
		length -= (PAGE_SIZE - offset);
		offset = 0;
1265
	}
1266
	sg_mark_end(&sg[i - 1]);
1267
	iod->nents = count;
1268 1269 1270 1271

	err = -ENOMEM;
	nents = dma_map_sg(&dev->pci_dev->dev, sg, count,
				write ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
1272
	if (!nents)
1273
		goto free_iod;
M
Matthew Wilcox 已提交
1274

1275
	kfree(pages);
1276
	return iod;
M
Matthew Wilcox 已提交
1277

1278 1279
 free_iod:
	kfree(iod);
1280 1281 1282 1283
 put_pages:
	for (i = 0; i < count; i++)
		put_page(pages[i]);
	kfree(pages);
1284
	return ERR_PTR(err);
1285
}
M
Matthew Wilcox 已提交
1286

V
Vishal Verma 已提交
1287
void nvme_unmap_user_pages(struct nvme_dev *dev, int write,
1288
			struct nvme_iod *iod)
1289
{
1290
	int i;
M
Matthew Wilcox 已提交
1291

1292 1293
	dma_unmap_sg(&dev->pci_dev->dev, iod->sg, iod->nents,
				write ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
1294

1295 1296
	for (i = 0; i < iod->nents; i++)
		put_page(sg_page(&iod->sg[i]));
1297
}
M
Matthew Wilcox 已提交
1298

M
Matthew Wilcox 已提交
1299 1300 1301 1302 1303 1304
static int nvme_submit_io(struct nvme_ns *ns, struct nvme_user_io __user *uio)
{
	struct nvme_dev *dev = ns->dev;
	struct nvme_queue *nvmeq;
	struct nvme_user_io io;
	struct nvme_command c;
1305 1306 1307 1308 1309
	unsigned length, meta_len;
	int status, i;
	struct nvme_iod *iod, *meta_iod = NULL;
	dma_addr_t meta_dma_addr;
	void *meta, *uninitialized_var(meta_mem);
M
Matthew Wilcox 已提交
1310 1311 1312

	if (copy_from_user(&io, uio, sizeof(io)))
		return -EFAULT;
1313
	length = (io.nblocks + 1) << ns->lba_shift;
1314 1315 1316 1317
	meta_len = (io.nblocks + 1) * ns->ms;

	if (meta_len && ((io.metadata & 3) || !io.metadata))
		return -EINVAL;
1318 1319 1320 1321

	switch (io.opcode) {
	case nvme_cmd_write:
	case nvme_cmd_read:
M
Matthew Wilcox 已提交
1322
	case nvme_cmd_compare:
1323
		iod = nvme_map_user_pages(dev, io.opcode & 1, io.addr, length);
M
Matthew Wilcox 已提交
1324
		break;
1325
	default:
M
Matthew Wilcox 已提交
1326
		return -EINVAL;
1327 1328
	}

1329 1330
	if (IS_ERR(iod))
		return PTR_ERR(iod);
M
Matthew Wilcox 已提交
1331 1332 1333 1334

	memset(&c, 0, sizeof(c));
	c.rw.opcode = io.opcode;
	c.rw.flags = io.flags;
1335
	c.rw.nsid = cpu_to_le32(ns->ns_id);
M
Matthew Wilcox 已提交
1336
	c.rw.slba = cpu_to_le64(io.slba);
1337
	c.rw.length = cpu_to_le16(io.nblocks);
M
Matthew Wilcox 已提交
1338
	c.rw.control = cpu_to_le16(io.control);
1339 1340 1341 1342
	c.rw.dsmgmt = cpu_to_le32(io.dsmgmt);
	c.rw.reftag = cpu_to_le32(io.reftag);
	c.rw.apptag = cpu_to_le16(io.apptag);
	c.rw.appmask = cpu_to_le16(io.appmask);
1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374

	if (meta_len) {
		meta_iod = nvme_map_user_pages(dev, io.opcode & 1, io.metadata, meta_len);
		if (IS_ERR(meta_iod)) {
			status = PTR_ERR(meta_iod);
			meta_iod = NULL;
			goto unmap;
		}

		meta_mem = dma_alloc_coherent(&dev->pci_dev->dev, meta_len,
						&meta_dma_addr, GFP_KERNEL);
		if (!meta_mem) {
			status = -ENOMEM;
			goto unmap;
		}

		if (io.opcode & 1) {
			int meta_offset = 0;

			for (i = 0; i < meta_iod->nents; i++) {
				meta = kmap_atomic(sg_page(&meta_iod->sg[i])) +
						meta_iod->sg[i].offset;
				memcpy(meta_mem + meta_offset, meta,
						meta_iod->sg[i].length);
				kunmap_atomic(meta);
				meta_offset += meta_iod->sg[i].length;
			}
		}

		c.rw.metadata = cpu_to_le64(meta_dma_addr);
	}

1375
	length = nvme_setup_prps(dev, &c.common, iod, length, GFP_KERNEL);
M
Matthew Wilcox 已提交
1376

1377
	nvmeq = get_nvmeq(dev);
M
Matthew Wilcox 已提交
1378 1379
	/*
	 * Since nvme_submit_sync_cmd sleeps, we can't keep preemption
1380 1381 1382 1383
	 * disabled.  We may be preempted at any point, and be rescheduled
	 * to a different CPU.  That will cause cacheline bouncing, but no
	 * additional races since q_lock already protects against other CPUs.
	 */
M
Matthew Wilcox 已提交
1384
	put_nvmeq(nvmeq);
1385 1386 1387
	if (length != (io.nblocks + 1) << ns->lba_shift)
		status = -ENOMEM;
	else
1388
		status = nvme_submit_sync_cmd(nvmeq, &c, NULL, NVME_IO_TIMEOUT);
M
Matthew Wilcox 已提交
1389

1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408
	if (meta_len) {
		if (status == NVME_SC_SUCCESS && !(io.opcode & 1)) {
			int meta_offset = 0;

			for (i = 0; i < meta_iod->nents; i++) {
				meta = kmap_atomic(sg_page(&meta_iod->sg[i])) +
						meta_iod->sg[i].offset;
				memcpy(meta, meta_mem + meta_offset,
						meta_iod->sg[i].length);
				kunmap_atomic(meta);
				meta_offset += meta_iod->sg[i].length;
			}
		}

		dma_free_coherent(&dev->pci_dev->dev, meta_len, meta_mem,
								meta_dma_addr);
	}

 unmap:
1409
	nvme_unmap_user_pages(dev, io.opcode & 1, iod);
1410
	nvme_free_iod(dev, iod);
1411 1412 1413 1414 1415 1416

	if (meta_iod) {
		nvme_unmap_user_pages(dev, io.opcode & 1, meta_iod);
		nvme_free_iod(dev, meta_iod);
	}

M
Matthew Wilcox 已提交
1417 1418 1419
	return status;
}

1420
static int nvme_user_admin_cmd(struct nvme_dev *dev,
M
Matthew Wilcox 已提交
1421
					struct nvme_admin_cmd __user *ucmd)
1422
{
M
Matthew Wilcox 已提交
1423
	struct nvme_admin_cmd cmd;
1424
	struct nvme_command c;
1425
	int status, length;
1426
	struct nvme_iod *uninitialized_var(iod);
1427
	unsigned timeout;
1428

M
Matthew Wilcox 已提交
1429 1430 1431
	if (!capable(CAP_SYS_ADMIN))
		return -EACCES;
	if (copy_from_user(&cmd, ucmd, sizeof(cmd)))
1432 1433 1434
		return -EFAULT;

	memset(&c, 0, sizeof(c));
M
Matthew Wilcox 已提交
1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448
	c.common.opcode = cmd.opcode;
	c.common.flags = cmd.flags;
	c.common.nsid = cpu_to_le32(cmd.nsid);
	c.common.cdw2[0] = cpu_to_le32(cmd.cdw2);
	c.common.cdw2[1] = cpu_to_le32(cmd.cdw3);
	c.common.cdw10[0] = cpu_to_le32(cmd.cdw10);
	c.common.cdw10[1] = cpu_to_le32(cmd.cdw11);
	c.common.cdw10[2] = cpu_to_le32(cmd.cdw12);
	c.common.cdw10[3] = cpu_to_le32(cmd.cdw13);
	c.common.cdw10[4] = cpu_to_le32(cmd.cdw14);
	c.common.cdw10[5] = cpu_to_le32(cmd.cdw15);

	length = cmd.data_len;
	if (cmd.data_len) {
1449 1450
		iod = nvme_map_user_pages(dev, cmd.opcode & 1, cmd.addr,
								length);
1451 1452 1453 1454
		if (IS_ERR(iod))
			return PTR_ERR(iod);
		length = nvme_setup_prps(dev, &c.common, iod, length,
								GFP_KERNEL);
M
Matthew Wilcox 已提交
1455 1456
	}

1457 1458
	timeout = cmd.timeout_ms ? msecs_to_jiffies(cmd.timeout_ms) :
								ADMIN_TIMEOUT;
M
Matthew Wilcox 已提交
1459
	if (length != cmd.data_len)
1460 1461
		status = -ENOMEM;
	else
1462 1463
		status = nvme_submit_sync_cmd(dev->queues[0], &c, &cmd.result,
								timeout);
1464

M
Matthew Wilcox 已提交
1465
	if (cmd.data_len) {
1466
		nvme_unmap_user_pages(dev, cmd.opcode & 1, iod);
1467
		nvme_free_iod(dev, iod);
M
Matthew Wilcox 已提交
1468
	}
1469

1470
	if ((status >= 0) && copy_to_user(&ucmd->result, &cmd.result,
1471 1472 1473
							sizeof(cmd.result)))
		status = -EFAULT;

1474 1475 1476
	return status;
}

M
Matthew Wilcox 已提交
1477 1478 1479 1480 1481 1482
static int nvme_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd,
							unsigned long arg)
{
	struct nvme_ns *ns = bdev->bd_disk->private_data;

	switch (cmd) {
M
Matthew Wilcox 已提交
1483 1484 1485
	case NVME_IOCTL_ID:
		return ns->ns_id;
	case NVME_IOCTL_ADMIN_CMD:
1486
		return nvme_user_admin_cmd(ns->dev, (void __user *)arg);
M
Matthew Wilcox 已提交
1487 1488
	case NVME_IOCTL_SUBMIT_IO:
		return nvme_submit_io(ns, (void __user *)arg);
V
Vishal Verma 已提交
1489 1490 1491 1492
	case SG_GET_VERSION_NUM:
		return nvme_sg_get_version_num((void __user *)arg);
	case SG_IO:
		return nvme_sg_io(ns, (void __user *)arg);
M
Matthew Wilcox 已提交
1493 1494 1495 1496 1497 1498 1499 1500
	default:
		return -ENOTTY;
	}
}

static const struct block_device_operations nvme_fops = {
	.owner		= THIS_MODULE,
	.ioctl		= nvme_ioctl,
M
Matthew Wilcox 已提交
1501
	.compat_ioctl	= nvme_ioctl,
M
Matthew Wilcox 已提交
1502 1503
};

1504 1505 1506 1507 1508
static void nvme_resubmit_bios(struct nvme_queue *nvmeq)
{
	while (bio_list_peek(&nvmeq->sq_cong)) {
		struct bio *bio = bio_list_pop(&nvmeq->sq_cong);
		struct nvme_ns *ns = bio->bi_bdev->bd_disk->private_data;
1509 1510 1511 1512

		if (bio_list_empty(&nvmeq->sq_cong))
			remove_wait_queue(&nvmeq->sq_full,
							&nvmeq->sq_cong_wait);
1513
		if (nvme_submit_bio_queue(nvmeq, ns, bio)) {
1514 1515 1516
			if (bio_list_empty(&nvmeq->sq_cong))
				add_wait_queue(&nvmeq->sq_full,
							&nvmeq->sq_cong_wait);
1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527
			bio_list_add_head(&nvmeq->sq_cong, bio);
			break;
		}
	}
}

static int nvme_kthread(void *data)
{
	struct nvme_dev *dev;

	while (!kthread_should_stop()) {
1528
		set_current_state(TASK_INTERRUPTIBLE);
1529 1530 1531 1532 1533
		spin_lock(&dev_list_lock);
		list_for_each_entry(dev, &dev_list, node) {
			int i;
			for (i = 0; i < dev->queue_count; i++) {
				struct nvme_queue *nvmeq = dev->queues[i];
1534 1535
				if (!nvmeq)
					continue;
1536 1537 1538
				spin_lock_irq(&nvmeq->q_lock);
				if (nvme_process_cq(nvmeq))
					printk("process_cq did something\n");
1539
				nvme_cancel_ios(nvmeq, true);
1540 1541 1542 1543 1544
				nvme_resubmit_bios(nvmeq);
				spin_unlock_irq(&nvmeq->q_lock);
			}
		}
		spin_unlock(&dev_list_lock);
1545
		schedule_timeout(round_jiffies_relative(HZ));
1546 1547 1548 1549
	}
	return 0;
}

1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576
static DEFINE_IDA(nvme_index_ida);

static int nvme_get_ns_idx(void)
{
	int index, error;

	do {
		if (!ida_pre_get(&nvme_index_ida, GFP_KERNEL))
			return -1;

		spin_lock(&dev_list_lock);
		error = ida_get_new(&nvme_index_ida, &index);
		spin_unlock(&dev_list_lock);
	} while (error == -EAGAIN);

	if (error)
		index = -1;
	return index;
}

static void nvme_put_ns_idx(int index)
{
	spin_lock(&dev_list_lock);
	ida_remove(&nvme_index_ida, index);
	spin_unlock(&dev_list_lock);
}

1577 1578 1579 1580 1581 1582 1583 1584 1585 1586
static void nvme_config_discard(struct nvme_ns *ns)
{
	u32 logical_block_size = queue_logical_block_size(ns->queue);
	ns->queue->limits.discard_zeroes_data = 0;
	ns->queue->limits.discard_alignment = logical_block_size;
	ns->queue->limits.discard_granularity = logical_block_size;
	ns->queue->limits.max_discard_sectors = 0xffffffff;
	queue_flag_set_unlocked(QUEUE_FLAG_DISCARD, ns->queue);
}

1587
static struct nvme_ns *nvme_alloc_ns(struct nvme_dev *dev, int nsid,
M
Matthew Wilcox 已提交
1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602
			struct nvme_id_ns *id, struct nvme_lba_range_type *rt)
{
	struct nvme_ns *ns;
	struct gendisk *disk;
	int lbaf;

	if (rt->attributes & NVME_LBART_ATTRIB_HIDE)
		return NULL;

	ns = kzalloc(sizeof(*ns), GFP_KERNEL);
	if (!ns)
		return NULL;
	ns->queue = blk_alloc_queue(GFP_KERNEL);
	if (!ns->queue)
		goto out_free_ns;
M
Matthew Wilcox 已提交
1603 1604 1605
	ns->queue->queue_flags = QUEUE_FLAG_DEFAULT;
	queue_flag_set_unlocked(QUEUE_FLAG_NOMERGES, ns->queue);
	queue_flag_set_unlocked(QUEUE_FLAG_NONROT, ns->queue);
M
Matthew Wilcox 已提交
1606 1607 1608 1609 1610 1611 1612
	blk_queue_make_request(ns->queue, nvme_make_request);
	ns->dev = dev;
	ns->queue->queuedata = ns;

	disk = alloc_disk(NVME_MINORS);
	if (!disk)
		goto out_free_queue;
1613
	ns->ns_id = nsid;
M
Matthew Wilcox 已提交
1614 1615 1616
	ns->disk = disk;
	lbaf = id->flbas & 0xf;
	ns->lba_shift = id->lbaf[lbaf].ds;
1617
	ns->ms = le16_to_cpu(id->lbaf[lbaf].ms);
1618
	blk_queue_logical_block_size(ns->queue, 1 << ns->lba_shift);
K
Keith Busch 已提交
1619 1620
	if (dev->max_hw_sectors)
		blk_queue_max_hw_sectors(ns->queue, dev->max_hw_sectors);
M
Matthew Wilcox 已提交
1621 1622 1623

	disk->major = nvme_major;
	disk->minors = NVME_MINORS;
1624
	disk->first_minor = NVME_MINORS * nvme_get_ns_idx();
M
Matthew Wilcox 已提交
1625 1626 1627
	disk->fops = &nvme_fops;
	disk->private_data = ns;
	disk->queue = ns->queue;
1628
	disk->driverfs_dev = &dev->pci_dev->dev;
1629
	sprintf(disk->disk_name, "nvme%dn%d", dev->instance, nsid);
M
Matthew Wilcox 已提交
1630 1631
	set_capacity(disk, le64_to_cpup(&id->nsze) << (ns->lba_shift - 9));

1632 1633 1634
	if (dev->oncs & NVME_CTRL_ONCS_DSM)
		nvme_config_discard(ns);

M
Matthew Wilcox 已提交
1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645
	return ns;

 out_free_queue:
	blk_cleanup_queue(ns->queue);
 out_free_ns:
	kfree(ns);
	return NULL;
}

static void nvme_ns_free(struct nvme_ns *ns)
{
1646
	int index = ns->disk->first_minor / NVME_MINORS;
M
Matthew Wilcox 已提交
1647
	put_disk(ns->disk);
1648
	nvme_put_ns_idx(index);
M
Matthew Wilcox 已提交
1649 1650 1651 1652
	blk_cleanup_queue(ns->queue);
	kfree(ns);
}

1653
static int set_queue_count(struct nvme_dev *dev, int count)
M
Matthew Wilcox 已提交
1654 1655 1656
{
	int status;
	u32 result;
1657
	u32 q_count = (count - 1) | ((count - 1) << 16);
M
Matthew Wilcox 已提交
1658

1659
	status = nvme_set_features(dev, NVME_FEAT_NUM_QUEUES, q_count, 0,
1660
								&result);
M
Matthew Wilcox 已提交
1661 1662 1663 1664 1665
	if (status)
		return -EIO;
	return min(result & 0xffff, result >> 16) + 1;
}

1666
static int nvme_setup_io_queues(struct nvme_dev *dev)
M
Matthew Wilcox 已提交
1667
{
R
Ramachandra Rao Gajula 已提交
1668
	struct pci_dev *pdev = dev->pci_dev;
1669
	int result, cpu, i, vecs, nr_io_queues, db_bar_size, q_depth;
M
Matthew Wilcox 已提交
1670

1671 1672
	nr_io_queues = num_online_cpus();
	result = set_queue_count(dev, nr_io_queues);
M
Matthew Wilcox 已提交
1673 1674
	if (result < 0)
		return result;
1675 1676
	if (result < nr_io_queues)
		nr_io_queues = result;
M
Matthew Wilcox 已提交
1677

M
Matthew Wilcox 已提交
1678 1679 1680
	/* Deregister the admin queue's interrupt */
	free_irq(dev->entry[0].vector, dev->queues[0]);

1681 1682 1683
	db_bar_size = 4096 + ((nr_io_queues + 1) << (dev->db_stride + 3));
	if (db_bar_size > 8192) {
		iounmap(dev->bar);
R
Ramachandra Rao Gajula 已提交
1684
		dev->bar = ioremap(pci_resource_start(pdev, 0), db_bar_size);
1685 1686 1687 1688
		dev->dbs = ((void __iomem *)dev->bar) + 4096;
		dev->queues[0]->q_db = dev->dbs;
	}

1689 1690
	vecs = nr_io_queues;
	for (i = 0; i < vecs; i++)
M
Matthew Wilcox 已提交
1691 1692
		dev->entry[i].entry = i;
	for (;;) {
1693 1694
		result = pci_enable_msix(pdev, dev->entry, vecs);
		if (result <= 0)
M
Matthew Wilcox 已提交
1695
			break;
1696
		vecs = result;
M
Matthew Wilcox 已提交
1697 1698
	}

1699 1700 1701 1702
	if (result < 0) {
		vecs = nr_io_queues;
		if (vecs > 32)
			vecs = 32;
R
Ramachandra Rao Gajula 已提交
1703
		for (;;) {
1704
			result = pci_enable_msi_block(pdev, vecs);
R
Ramachandra Rao Gajula 已提交
1705
			if (result == 0) {
1706
				for (i = 0; i < vecs; i++)
R
Ramachandra Rao Gajula 已提交
1707 1708
					dev->entry[i].vector = i + pdev->irq;
				break;
1709 1710
			} else if (result < 0) {
				vecs = 1;
R
Ramachandra Rao Gajula 已提交
1711 1712
				break;
			}
1713
			vecs = result;
R
Ramachandra Rao Gajula 已提交
1714 1715 1716
		}
	}

1717 1718 1719 1720 1721 1722 1723 1724
	/*
	 * Should investigate if there's a performance win from allocating
	 * more queues than interrupt vectors; it might allow the submission
	 * path to scale better, even if the receive path is limited by the
	 * number of interrupts.
	 */
	nr_io_queues = vecs;

M
Matthew Wilcox 已提交
1725 1726 1727 1728
	result = queue_request_irq(dev, dev->queues[0], "nvme admin");
	/* XXX: handle failure here */

	cpu = cpumask_first(cpu_online_mask);
1729
	for (i = 0; i < nr_io_queues; i++) {
M
Matthew Wilcox 已提交
1730 1731 1732 1733
		irq_set_affinity_hint(dev->entry[i].vector, get_cpu_mask(cpu));
		cpu = cpumask_next(cpu, cpu_online_mask);
	}

1734 1735
	q_depth = min_t(int, NVME_CAP_MQES(readq(&dev->bar->cap)) + 1,
								NVME_Q_DEPTH);
1736
	for (i = 0; i < nr_io_queues; i++) {
1737
		dev->queues[i + 1] = nvme_create_queue(dev, i + 1, q_depth, i);
1738 1739
		if (IS_ERR(dev->queues[i + 1]))
			return PTR_ERR(dev->queues[i + 1]);
M
Matthew Wilcox 已提交
1740 1741
		dev->queue_count++;
	}
M
Matthew Wilcox 已提交
1742

M
Matthew Wilcox 已提交
1743 1744 1745 1746 1747
	for (; i < num_possible_cpus(); i++) {
		int target = i % rounddown_pow_of_two(dev->queue_count - 1);
		dev->queues[i + 1] = dev->queues[target + 1];
	}

M
Matthew Wilcox 已提交
1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758
	return 0;
}

static void nvme_free_queues(struct nvme_dev *dev)
{
	int i;

	for (i = dev->queue_count - 1; i >= 0; i--)
		nvme_free_queue(dev, i);
}

1759 1760 1761 1762 1763 1764
/*
 * Return: error value if an error occurred setting up the queues or calling
 * Identify Device.  0 if these succeeded, even if adding some of the
 * namespaces failed.  At the moment, these failures are silent.  TBD which
 * failures should be reported.
 */
1765
static int nvme_dev_add(struct nvme_dev *dev)
M
Matthew Wilcox 已提交
1766 1767
{
	int res, nn, i;
1768
	struct nvme_ns *ns;
1769
	struct nvme_id_ctrl *ctrl;
1770 1771
	struct nvme_id_ns *id_ns;
	void *mem;
M
Matthew Wilcox 已提交
1772
	dma_addr_t dma_addr;
1773
	int shift = NVME_CAP_MPSMIN(readq(&dev->bar->cap)) + 12;
M
Matthew Wilcox 已提交
1774 1775 1776 1777 1778

	res = nvme_setup_io_queues(dev);
	if (res)
		return res;

1779
	mem = dma_alloc_coherent(&dev->pci_dev->dev, 8192, &dma_addr,
M
Matthew Wilcox 已提交
1780
								GFP_KERNEL);
1781 1782
	if (!mem)
		return -ENOMEM;
M
Matthew Wilcox 已提交
1783

1784
	res = nvme_identify(dev, 0, 1, dma_addr);
M
Matthew Wilcox 已提交
1785 1786
	if (res) {
		res = -EIO;
1787
		goto out;
M
Matthew Wilcox 已提交
1788 1789
	}

1790
	ctrl = mem;
1791
	nn = le32_to_cpup(&ctrl->nn);
1792
	dev->oncs = le16_to_cpup(&ctrl->oncs);
1793 1794 1795
	memcpy(dev->serial, ctrl->sn, sizeof(ctrl->sn));
	memcpy(dev->model, ctrl->mn, sizeof(ctrl->mn));
	memcpy(dev->firmware_rev, ctrl->fr, sizeof(ctrl->fr));
1796
	if (ctrl->mdts)
K
Keith Busch 已提交
1797
		dev->max_hw_sectors = 1 << (ctrl->mdts + shift - 9);
1798 1799 1800
	if ((dev->pci_dev->vendor == PCI_VENDOR_ID_INTEL) &&
			(dev->pci_dev->device == 0x0953) && ctrl->vs[3])
		dev->stripe_size = 1 << (ctrl->vs[3] + shift);
M
Matthew Wilcox 已提交
1801

1802
	id_ns = mem;
M
Matthew Wilcox 已提交
1803
	for (i = 1; i <= nn; i++) {
1804
		res = nvme_identify(dev, i, 0, dma_addr);
M
Matthew Wilcox 已提交
1805 1806 1807
		if (res)
			continue;

1808
		if (id_ns->ncap == 0)
M
Matthew Wilcox 已提交
1809 1810
			continue;

1811
		res = nvme_get_features(dev, NVME_FEAT_LBA_RANGE, i,
1812
							dma_addr + 4096, NULL);
M
Matthew Wilcox 已提交
1813
		if (res)
1814
			memset(mem + 4096, 0, 4096);
M
Matthew Wilcox 已提交
1815

1816
		ns = nvme_alloc_ns(dev, i, mem, mem + 4096);
M
Matthew Wilcox 已提交
1817 1818 1819 1820 1821
		if (ns)
			list_add_tail(&ns->list, &dev->namespaces);
	}
	list_for_each_entry(ns, &dev->namespaces, list)
		add_disk(ns->disk);
1822
	res = 0;
M
Matthew Wilcox 已提交
1823

1824
 out:
1825
	dma_free_coherent(&dev->pci_dev->dev, 8192, mem, dma_addr);
M
Matthew Wilcox 已提交
1826 1827 1828 1829 1830 1831 1832
	return res;
}

static int nvme_dev_remove(struct nvme_dev *dev)
{
	struct nvme_ns *ns, *next;

1833 1834 1835 1836
	spin_lock(&dev_list_lock);
	list_del(&dev->node);
	spin_unlock(&dev_list_lock);

M
Matthew Wilcox 已提交
1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847
	list_for_each_entry_safe(ns, next, &dev->namespaces, list) {
		list_del(&ns->list);
		del_gendisk(ns->disk);
		nvme_ns_free(ns);
	}

	nvme_free_queues(dev);

	return 0;
}

M
Matthew Wilcox 已提交
1848 1849 1850 1851 1852 1853 1854 1855
static int nvme_setup_prp_pools(struct nvme_dev *dev)
{
	struct device *dmadev = &dev->pci_dev->dev;
	dev->prp_page_pool = dma_pool_create("prp list page", dmadev,
						PAGE_SIZE, PAGE_SIZE, 0);
	if (!dev->prp_page_pool)
		return -ENOMEM;

1856 1857 1858 1859 1860 1861 1862
	/* Optimisation for I/Os between 4k and 128k */
	dev->prp_small_pool = dma_pool_create("prp list 256", dmadev,
						256, 256, 0);
	if (!dev->prp_small_pool) {
		dma_pool_destroy(dev->prp_page_pool);
		return -ENOMEM;
	}
M
Matthew Wilcox 已提交
1863 1864 1865 1866 1867 1868
	return 0;
}

static void nvme_release_prp_pools(struct nvme_dev *dev)
{
	dma_pool_destroy(dev->prp_page_pool);
1869
	dma_pool_destroy(dev->prp_small_pool);
M
Matthew Wilcox 已提交
1870 1871
}

1872 1873 1874
static DEFINE_IDA(nvme_instance_ida);

static int nvme_set_instance(struct nvme_dev *dev)
M
Matthew Wilcox 已提交
1875
{
1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891
	int instance, error;

	do {
		if (!ida_pre_get(&nvme_instance_ida, GFP_KERNEL))
			return -ENODEV;

		spin_lock(&dev_list_lock);
		error = ida_get_new(&nvme_instance_ida, &instance);
		spin_unlock(&dev_list_lock);
	} while (error == -EAGAIN);

	if (error)
		return -ENODEV;

	dev->instance = instance;
	return 0;
M
Matthew Wilcox 已提交
1892 1893 1894 1895
}

static void nvme_release_instance(struct nvme_dev *dev)
{
1896 1897 1898
	spin_lock(&dev_list_lock);
	ida_remove(&nvme_instance_ida, dev->instance);
	spin_unlock(&dev_list_lock);
M
Matthew Wilcox 已提交
1899 1900
}

1901 1902 1903 1904
static void nvme_free_dev(struct kref *kref)
{
	struct nvme_dev *dev = container_of(kref, struct nvme_dev, kref);
	nvme_dev_remove(dev);
R
Ramachandra Rao Gajula 已提交
1905 1906 1907 1908
	if (dev->pci_dev->msi_enabled)
		pci_disable_msi(dev->pci_dev);
	else if (dev->pci_dev->msix_enabled)
		pci_disable_msix(dev->pci_dev);
1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953
	iounmap(dev->bar);
	nvme_release_instance(dev);
	nvme_release_prp_pools(dev);
	pci_disable_device(dev->pci_dev);
	pci_release_regions(dev->pci_dev);
	kfree(dev->queues);
	kfree(dev->entry);
	kfree(dev);
}

static int nvme_dev_open(struct inode *inode, struct file *f)
{
	struct nvme_dev *dev = container_of(f->private_data, struct nvme_dev,
								miscdev);
	kref_get(&dev->kref);
	f->private_data = dev;
	return 0;
}

static int nvme_dev_release(struct inode *inode, struct file *f)
{
	struct nvme_dev *dev = f->private_data;
	kref_put(&dev->kref, nvme_free_dev);
	return 0;
}

static long nvme_dev_ioctl(struct file *f, unsigned int cmd, unsigned long arg)
{
	struct nvme_dev *dev = f->private_data;
	switch (cmd) {
	case NVME_IOCTL_ADMIN_CMD:
		return nvme_user_admin_cmd(dev, (void __user *)arg);
	default:
		return -ENOTTY;
	}
}

static const struct file_operations nvme_dev_fops = {
	.owner		= THIS_MODULE,
	.open		= nvme_dev_open,
	.release	= nvme_dev_release,
	.unlocked_ioctl	= nvme_dev_ioctl,
	.compat_ioctl	= nvme_dev_ioctl,
};

1954
static int nvme_probe(struct pci_dev *pdev, const struct pci_device_id *id)
M
Matthew Wilcox 已提交
1955
{
M
Matthew Wilcox 已提交
1956
	int bars, result = -ENOMEM;
M
Matthew Wilcox 已提交
1957 1958 1959 1960 1961 1962 1963 1964 1965
	struct nvme_dev *dev;

	dev = kzalloc(sizeof(*dev), GFP_KERNEL);
	if (!dev)
		return -ENOMEM;
	dev->entry = kcalloc(num_possible_cpus(), sizeof(*dev->entry),
								GFP_KERNEL);
	if (!dev->entry)
		goto free;
M
Matthew Wilcox 已提交
1966 1967
	dev->queues = kcalloc(num_possible_cpus() + 1, sizeof(void *),
								GFP_KERNEL);
M
Matthew Wilcox 已提交
1968 1969 1970
	if (!dev->queues)
		goto free;

1971 1972
	if (pci_enable_device_mem(pdev))
		goto free;
M
Matthew Wilcox 已提交
1973
	pci_set_master(pdev);
M
Matthew Wilcox 已提交
1974 1975 1976
	bars = pci_select_bars(pdev, IORESOURCE_MEM);
	if (pci_request_selected_regions(pdev, bars, "nvme"))
		goto disable;
1977

M
Matthew Wilcox 已提交
1978 1979 1980
	INIT_LIST_HEAD(&dev->namespaces);
	dev->pci_dev = pdev;
	pci_set_drvdata(pdev, dev);
1981 1982 1983 1984 1985 1986 1987 1988

	if (!dma_set_mask(&pdev->dev, DMA_BIT_MASK(64)))
		dma_set_coherent_mask(&pdev->dev, DMA_BIT_MASK(64));
	else if (!dma_set_mask(&pdev->dev, DMA_BIT_MASK(32)))
		dma_set_coherent_mask(&pdev->dev, DMA_BIT_MASK(32));
	else
		goto disable;

1989 1990 1991 1992
	result = nvme_set_instance(dev);
	if (result)
		goto disable;

1993
	dev->entry[0].vector = pdev->irq;
M
Matthew Wilcox 已提交
1994

M
Matthew Wilcox 已提交
1995 1996 1997 1998
	result = nvme_setup_prp_pools(dev);
	if (result)
		goto disable_msix;

M
Matthew Wilcox 已提交
1999 2000 2001
	dev->bar = ioremap(pci_resource_start(pdev, 0), 8192);
	if (!dev->bar) {
		result = -ENOMEM;
M
Matthew Wilcox 已提交
2002
		goto disable_msix;
M
Matthew Wilcox 已提交
2003 2004 2005 2006 2007 2008 2009
	}

	result = nvme_configure_admin_queue(dev);
	if (result)
		goto unmap;
	dev->queue_count++;

2010 2011 2012 2013
	spin_lock(&dev_list_lock);
	list_add(&dev->node, &dev_list);
	spin_unlock(&dev_list_lock);

2014 2015 2016 2017
	result = nvme_dev_add(dev);
	if (result)
		goto delete;

2018 2019 2020 2021 2022 2023 2024 2025 2026 2027
	scnprintf(dev->name, sizeof(dev->name), "nvme%d", dev->instance);
	dev->miscdev.minor = MISC_DYNAMIC_MINOR;
	dev->miscdev.parent = &pdev->dev;
	dev->miscdev.name = dev->name;
	dev->miscdev.fops = &nvme_dev_fops;
	result = misc_register(&dev->miscdev);
	if (result)
		goto remove;

	kref_init(&dev->kref);
M
Matthew Wilcox 已提交
2028 2029
	return 0;

2030 2031
 remove:
	nvme_dev_remove(dev);
M
Matthew Wilcox 已提交
2032
 delete:
2033 2034 2035 2036
	spin_lock(&dev_list_lock);
	list_del(&dev->node);
	spin_unlock(&dev_list_lock);

M
Matthew Wilcox 已提交
2037 2038 2039
	nvme_free_queues(dev);
 unmap:
	iounmap(dev->bar);
M
Matthew Wilcox 已提交
2040
 disable_msix:
R
Ramachandra Rao Gajula 已提交
2041 2042 2043 2044
	if (dev->pci_dev->msi_enabled)
		pci_disable_msi(dev->pci_dev);
	else if (dev->pci_dev->msix_enabled)
		pci_disable_msix(dev->pci_dev);
M
Matthew Wilcox 已提交
2045
	nvme_release_instance(dev);
M
Matthew Wilcox 已提交
2046
	nvme_release_prp_pools(dev);
M
Matthew Wilcox 已提交
2047
 disable:
2048
	pci_disable_device(pdev);
M
Matthew Wilcox 已提交
2049
	pci_release_regions(pdev);
M
Matthew Wilcox 已提交
2050 2051 2052 2053 2054 2055 2056
 free:
	kfree(dev->queues);
	kfree(dev->entry);
	kfree(dev);
	return result;
}

2057
static void nvme_remove(struct pci_dev *pdev)
M
Matthew Wilcox 已提交
2058 2059
{
	struct nvme_dev *dev = pci_get_drvdata(pdev);
2060 2061
	misc_deregister(&dev->miscdev);
	kref_put(&dev->kref, nvme_free_dev);
M
Matthew Wilcox 已提交
2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072
}

/* These functions are yet to be implemented */
#define nvme_error_detected NULL
#define nvme_dump_registers NULL
#define nvme_link_reset NULL
#define nvme_slot_reset NULL
#define nvme_error_resume NULL
#define nvme_suspend NULL
#define nvme_resume NULL

2073
static const struct pci_error_handlers nvme_err_handler = {
M
Matthew Wilcox 已提交
2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093
	.error_detected	= nvme_error_detected,
	.mmio_enabled	= nvme_dump_registers,
	.link_reset	= nvme_link_reset,
	.slot_reset	= nvme_slot_reset,
	.resume		= nvme_error_resume,
};

/* Move to pci_ids.h later */
#define PCI_CLASS_STORAGE_EXPRESS	0x010802

static DEFINE_PCI_DEVICE_TABLE(nvme_id_table) = {
	{ PCI_DEVICE_CLASS(PCI_CLASS_STORAGE_EXPRESS, 0xffffff) },
	{ 0, }
};
MODULE_DEVICE_TABLE(pci, nvme_id_table);

static struct pci_driver nvme_driver = {
	.name		= "nvme",
	.id_table	= nvme_id_table,
	.probe		= nvme_probe,
2094
	.remove		= nvme_remove,
M
Matthew Wilcox 已提交
2095 2096 2097 2098 2099 2100 2101
	.suspend	= nvme_suspend,
	.resume		= nvme_resume,
	.err_handler	= &nvme_err_handler,
};

static int __init nvme_init(void)
{
2102
	int result;
2103 2104 2105 2106

	nvme_thread = kthread_run(nvme_kthread, NULL, "nvme");
	if (IS_ERR(nvme_thread))
		return PTR_ERR(nvme_thread);
M
Matthew Wilcox 已提交
2107

2108 2109
	result = register_blkdev(nvme_major, "nvme");
	if (result < 0)
2110
		goto kill_kthread;
2111
	else if (result > 0)
2112
		nvme_major = result;
M
Matthew Wilcox 已提交
2113 2114

	result = pci_register_driver(&nvme_driver);
2115 2116 2117
	if (result)
		goto unregister_blkdev;
	return 0;
M
Matthew Wilcox 已提交
2118

2119
 unregister_blkdev:
M
Matthew Wilcox 已提交
2120
	unregister_blkdev(nvme_major, "nvme");
2121 2122
 kill_kthread:
	kthread_stop(nvme_thread);
M
Matthew Wilcox 已提交
2123 2124 2125 2126 2127 2128 2129
	return result;
}

static void __exit nvme_exit(void)
{
	pci_unregister_driver(&nvme_driver);
	unregister_blkdev(nvme_major, "nvme");
2130
	kthread_stop(nvme_thread);
M
Matthew Wilcox 已提交
2131 2132 2133 2134
}

MODULE_AUTHOR("Matthew Wilcox <willy@linux.intel.com>");
MODULE_LICENSE("GPL");
M
Matthew Wilcox 已提交
2135
MODULE_VERSION("0.8");
M
Matthew Wilcox 已提交
2136 2137
module_init(nvme_init);
module_exit(nvme_exit);