blkback.c 20.7 KB
Newer Older
K
Konrad Rzeszutek Wilk 已提交
1 2 3 4 5 6
/******************************************************************************
 *
 * Back-end of the driver for virtual block devices. This portion of the
 * driver exports a 'unified' block-device interface that can be accessed
 * by any operating system that implements a compatible front end. A
 * reference front-end implementation can be found in:
7
 *  drivers/block/xen-blkfront.c
K
Konrad Rzeszutek Wilk 已提交
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
 *
 * Copyright (c) 2003-2004, Keir Fraser & Steve Hand
 * Copyright (c) 2005, Christopher Clark
 *
 * 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; or, when distributed
 * separately from the Linux kernel or incorporated into other
 * software packages, subject to the following license:
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this source file (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use, copy, modify,
 * merge, publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 */

#include <linux/spinlock.h>
#include <linux/kthread.h>
#include <linux/list.h>
#include <linux/delay.h>
J
Jeremy Fitzhardinge 已提交
41
#include <linux/freezer.h>
42

J
Jeremy Fitzhardinge 已提交
43 44 45 46
#include <xen/events.h>
#include <xen/page.h>
#include <asm/xen/hypervisor.h>
#include <asm/xen/hypercall.h>
K
Konrad Rzeszutek Wilk 已提交
47 48
#include "common.h"

T
Tom Goetz 已提交
49 50
#define WRITE_BARRIER	(REQ_WRITE | REQ_FLUSH | REQ_FUA)

K
Konrad Rzeszutek Wilk 已提交
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
/*
 * These are rather arbitrary. They are fairly large because adjacent requests
 * pulled from a communication ring are quite likely to end up being part of
 * the same scatter/gather request at the disc.
 *
 * ** TRY INCREASING 'blkif_reqs' IF WRITE SPEEDS SEEM TOO LOW **
 *
 * This will increase the chances of being able to write whole tracks.
 * 64 should be enough to keep us competitive with Linux.
 */
static int blkif_reqs = 64;
module_param_named(reqs, blkif_reqs, int, 0);
MODULE_PARM_DESC(reqs, "Number of blkback requests to allocate");

/* Run-time switchable: /sys/module/blkback/parameters/ */
66 67
static unsigned int log_stats;
static unsigned int debug_lvl;
K
Konrad Rzeszutek Wilk 已提交
68 69 70 71 72 73 74 75 76
module_param(log_stats, int, 0644);
module_param(debug_lvl, int, 0644);

/*
 * Each outstanding request that we've passed to the lower device layers has a
 * 'pending_req' allocated to it. Each buffer_head that completes decrements
 * the pendcnt towards zero. When it hits zero, the specified domain has a
 * response queued for it, with the saved 'id' passed back.
 */
77
struct pending_req {
78
	struct blkif_st       *blkif;
K
Konrad Rzeszutek Wilk 已提交
79 80 81 82 83 84
	u64            id;
	int            nr_pages;
	atomic_t       pendcnt;
	unsigned short operation;
	int            status;
	struct list_head free_list;
85
};
K
Konrad Rzeszutek Wilk 已提交
86 87 88

#define BLKBACK_INVALID_HANDLE (~0)

89
struct xen_blkbk {
90
	struct pending_req	*pending_reqs;
91
	/* List of all 'pending_req' available */
92
	struct list_head	pending_free;
93
	/* And its spinlock. */
94 95
	spinlock_t		pending_free_lock;
	wait_queue_head_t	pending_free_wq;
96
	/* The list of all pages that are available. */
97
	struct page		**pending_pages;
98
	/* And the grant handles that are available. */
99 100 101 102
	grant_handle_t		*pending_grant_handles;
};

static struct xen_blkbk *blkbk;
K
Konrad Rzeszutek Wilk 已提交
103

104 105 106 107 108
/*
 * Little helpful macro to figure out the index and virtual address of the
 * pending_pages[..]. For each 'pending_req' we have have up to
 * BLKIF_MAX_SEGMENTS_PER_REQUEST (11) pages. The seg would be from 0 through
 * 10 and would index in the pending_pages[..]. */
109
static inline int vaddr_pagenr(struct pending_req *req, int seg)
K
Konrad Rzeszutek Wilk 已提交
110
{
111 112
	return (req - blkbk->pending_reqs) *
		BLKIF_MAX_SEGMENTS_PER_REQUEST + seg;
K
Konrad Rzeszutek Wilk 已提交
113 114
}

115 116
#define pending_page(req, seg) pending_pages[vaddr_pagenr(req, seg)]

117
static inline unsigned long vaddr(struct pending_req *req, int seg)
K
Konrad Rzeszutek Wilk 已提交
118
{
119
	unsigned long pfn = page_to_pfn(blkbk->pending_page(req, seg));
K
Konrad Rzeszutek Wilk 已提交
120 121 122 123
	return (unsigned long)pfn_to_kaddr(pfn);
}

#define pending_handle(_req, _seg) \
124
	(blkbk->pending_grant_handles[vaddr_pagenr(_req, _seg)])
K
Konrad Rzeszutek Wilk 已提交
125 126


127 128
static int do_block_io_op(struct blkif_st *blkif);
static void dispatch_rw_block_io(struct blkif_st *blkif,
J
Jeremy Fitzhardinge 已提交
129
				 struct blkif_request *req,
130
				 struct pending_req *pending_req);
131
static void make_response(struct blkif_st *blkif, u64 id,
K
Konrad Rzeszutek Wilk 已提交
132 133
			  unsigned short op, int st);

134 135
/*
 * Retrieve from the 'pending_reqs' a free pending_req structure to be used.
K
Konrad Rzeszutek Wilk 已提交
136
 */
137
static struct pending_req *alloc_req(void)
K
Konrad Rzeszutek Wilk 已提交
138
{
139
	struct pending_req *req = NULL;
K
Konrad Rzeszutek Wilk 已提交
140 141
	unsigned long flags;

142 143
	spin_lock_irqsave(&blkbk->pending_free_lock, flags);
	if (!list_empty(&blkbk->pending_free)) {
144 145
		req = list_entry(blkbk->pending_free.next, struct pending_req,
				 free_list);
K
Konrad Rzeszutek Wilk 已提交
146 147
		list_del(&req->free_list);
	}
148
	spin_unlock_irqrestore(&blkbk->pending_free_lock, flags);
K
Konrad Rzeszutek Wilk 已提交
149 150 151
	return req;
}

152 153 154 155
/*
 * Return the 'pending_req' structure back to the freepool. We also
 * wake up the thread if it was waiting for a free page.
 */
156
static void free_req(struct pending_req *req)
K
Konrad Rzeszutek Wilk 已提交
157 158 159 160
{
	unsigned long flags;
	int was_empty;

161 162 163 164
	spin_lock_irqsave(&blkbk->pending_free_lock, flags);
	was_empty = list_empty(&blkbk->pending_free);
	list_add(&req->free_list, &blkbk->pending_free);
	spin_unlock_irqrestore(&blkbk->pending_free_lock, flags);
K
Konrad Rzeszutek Wilk 已提交
165
	if (was_empty)
166
		wake_up(&blkbk->pending_free_wq);
K
Konrad Rzeszutek Wilk 已提交
167 168
}

169
/*
170 171 172
 * Notification from the guest OS.
 */
static void blkif_notify_work(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
173
{
174 175 176
	blkif->waiting_reqs = 1;
	wake_up(&blkif->wq);
}
K
Konrad Rzeszutek Wilk 已提交
177

178 179 180 181
irqreturn_t blkif_be_int(int irq, void *dev_id)
{
	blkif_notify_work(dev_id);
	return IRQ_HANDLED;
K
Konrad Rzeszutek Wilk 已提交
182 183
}

184
/*
K
Konrad Rzeszutek Wilk 已提交
185 186 187
 * SCHEDULER FUNCTIONS
 */

188
static void print_stats(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
189 190 191 192 193 194 195 196 197 198 199 200
{
	printk(KERN_DEBUG "%s: oo %3d  |  rd %4d  |  wr %4d  |  br %4d\n",
	       current->comm, blkif->st_oo_req,
	       blkif->st_rd_req, blkif->st_wr_req, blkif->st_br_req);
	blkif->st_print = jiffies + msecs_to_jiffies(10 * 1000);
	blkif->st_rd_req = 0;
	blkif->st_wr_req = 0;
	blkif->st_oo_req = 0;
}

int blkif_schedule(void *arg)
{
201
	struct blkif_st *blkif = arg;
202
	struct vbd *vbd = &blkif->vbd;
K
Konrad Rzeszutek Wilk 已提交
203 204 205 206 207 208 209 210 211

	blkif_get(blkif);

	if (debug_lvl)
		printk(KERN_DEBUG "%s: started\n", current->comm);

	while (!kthread_should_stop()) {
		if (try_to_freeze())
			continue;
212 213
		if (unlikely(vbd->size != vbd_size(vbd)))
			vbd_resize(blkif);
K
Konrad Rzeszutek Wilk 已提交
214 215 216 217 218

		wait_event_interruptible(
			blkif->wq,
			blkif->waiting_reqs || kthread_should_stop());
		wait_event_interruptible(
219
			blkbk->pending_free_wq,
220 221
			!list_empty(&blkbk->pending_free) ||
			kthread_should_stop());
K
Konrad Rzeszutek Wilk 已提交
222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243

		blkif->waiting_reqs = 0;
		smp_mb(); /* clear flag *before* checking for work */

		if (do_block_io_op(blkif))
			blkif->waiting_reqs = 1;

		if (log_stats && time_after(jiffies, blkif->st_print))
			print_stats(blkif);
	}

	if (log_stats)
		print_stats(blkif);
	if (debug_lvl)
		printk(KERN_DEBUG "%s: exiting\n", current->comm);

	blkif->xenblkd = NULL;
	blkif_put(blkif);

	return 0;
}

244 245 246 247
struct seg_buf {
	unsigned long buf;
	unsigned int nsec;
};
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
/*
 * Unmap the grant references, and also remove the M2P over-rides
 * used in the 'pending_req'.
*/
static void fast_flush_area(struct pending_req *req)
{
	struct gnttab_unmap_grant_ref unmap[BLKIF_MAX_SEGMENTS_PER_REQUEST];
	unsigned int i, invcount = 0;
	grant_handle_t handle;
	int ret;

	for (i = 0; i < req->nr_pages; i++) {
		handle = pending_handle(req, i);
		if (handle == BLKBACK_INVALID_HANDLE)
			continue;
		gnttab_set_unmap_op(&unmap[invcount], vaddr(req, i),
				    GNTMAP_host_map, handle);
		pending_handle(req, i) = BLKBACK_INVALID_HANDLE;
		invcount++;
	}

	ret = HYPERVISOR_grant_table_op(
		GNTTABOP_unmap_grant_ref, unmap, invcount);
	BUG_ON(ret);
	/* Note, we use invcount, so nr->pages, so we can't index
	 * using vaddr(req, i).
	 */
	for (i = 0; i < invcount; i++) {
		ret = m2p_remove_override(
			virt_to_page(unmap[i].host_addr), false);
		if (ret) {
			printk(KERN_ALERT "Failed to remove M2P override for " \
				"%lx\n", (unsigned long)unmap[i].host_addr);
			continue;
		}
	}
}
285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
static int xen_blk_map_buf(struct blkif_request *req, struct pending_req *pending_req,
			   struct seg_buf seg[])
{
	struct gnttab_map_grant_ref map[BLKIF_MAX_SEGMENTS_PER_REQUEST];
	int i;
	int nseg = req->nr_segments;
	int ret = 0;
	/* Fill out preq.nr_sects with proper amount of sectors, and setup
	 * assign map[..] with the PFN of the page in our domain with the
	 * corresponding grant reference for each page.
	 */
	for (i = 0; i < nseg; i++) {
		uint32_t flags;

		flags = GNTMAP_host_map;
		if (pending_req->operation != BLKIF_OP_READ)
			flags |= GNTMAP_readonly;
		gnttab_set_map_op(&map[i], vaddr(pending_req, i), flags,
				  req->u.rw.seg[i].gref, pending_req->blkif->domid);
	}

	ret = HYPERVISOR_grant_table_op(GNTTABOP_map_grant_ref, map, nseg);
	BUG_ON(ret);

	/* Now swizzel the MFN in our domain with the MFN from the other domain
	 * so that when we access vaddr(pending_req,i) it has the contents of
	 * the page from the other domain.
	 */
	for (i = 0; i < nseg; i++) {
		if (unlikely(map[i].status != 0)) {
			DPRINTK("invalid buffer -- could not remap it\n");
			map[i].handle = BLKBACK_INVALID_HANDLE;
			ret |= 1;
		}

		pending_handle(pending_req, i) = map[i].handle;

		if (ret)
			continue;

		ret = m2p_add_override(PFN_DOWN(map[i].dev_bus_addr),
			blkbk->pending_page(pending_req, i), false);
		if (ret) {
			printk(KERN_ALERT "Failed to install M2P override for"\
				" %lx (ret: %d)\n", (unsigned long)
				map[i].dev_bus_addr, ret);
			/* We could switch over to GNTTABOP_copy */
			continue;
		}

		seg[i].buf  = map[i].dev_bus_addr |
			(req->u.rw.seg[i].first_sect << 9);
	}
	return ret;
}

341 342
/*
 * Completion callback on the bio's. Called as bh->b_end_io()
K
Konrad Rzeszutek Wilk 已提交
343 344
 */

345
static void __end_block_io_op(struct pending_req *pending_req, int error)
K
Konrad Rzeszutek Wilk 已提交
346 347 348 349 350 351 352 353 354 355 356 357 358
{
	/* An error fails the entire request. */
	if ((pending_req->operation == BLKIF_OP_WRITE_BARRIER) &&
	    (error == -EOPNOTSUPP)) {
		DPRINTK("blkback: write barrier op failed, not supported\n");
		blkback_barrier(XBT_NIL, pending_req->blkif->be, 0);
		pending_req->status = BLKIF_RSP_EOPNOTSUPP;
	} else if (error) {
		DPRINTK("Buffer not up-to-date at end of operation, "
			"error=%d\n", error);
		pending_req->status = BLKIF_RSP_ERROR;
	}

359 360
	/* If all of the bio's have completed it is time to unmap
	 * the grant references associated with 'request' and provide
361 362
	 * the proper response on the ring.
	 */
K
Konrad Rzeszutek Wilk 已提交
363 364 365 366 367 368 369 370 371
	if (atomic_dec_and_test(&pending_req->pendcnt)) {
		fast_flush_area(pending_req);
		make_response(pending_req->blkif, pending_req->id,
			      pending_req->operation, pending_req->status);
		blkif_put(pending_req->blkif);
		free_req(pending_req);
	}
}

372 373 374
/*
 * bio callback.
 */
J
Jeremy Fitzhardinge 已提交
375
static void end_block_io_op(struct bio *bio, int error)
K
Konrad Rzeszutek Wilk 已提交
376 377 378 379 380 381 382
{
	__end_block_io_op(bio->bi_private, error);
	bio_put(bio);
}



383 384 385 386
/*
 * Function to copy the from the ring buffer the 'struct blkif_request'
 * (which has the sectors we want, number of them, grant references, etc),
 * and transmute  it to the block API to hand it over to the proper block disk.
K
Konrad Rzeszutek Wilk 已提交
387
 */
388
static int do_block_io_op(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
389
{
J
Jeremy Fitzhardinge 已提交
390 391
	union blkif_back_rings *blk_rings = &blkif->blk_rings;
	struct blkif_request req;
392
	struct pending_req *pending_req;
K
Konrad Rzeszutek Wilk 已提交
393 394 395 396 397 398 399 400 401 402 403 404
	RING_IDX rc, rp;
	int more_to_do = 0;

	rc = blk_rings->common.req_cons;
	rp = blk_rings->common.sring->req_prod;
	rmb(); /* Ensure we see queued requests up to 'rp'. */

	while (rc != rp) {

		if (RING_REQUEST_CONS_OVERFLOW(&blk_rings->common, rc))
			break;

405
		if (kthread_should_stop()) {
K
Konrad Rzeszutek Wilk 已提交
406 407 408 409
			more_to_do = 1;
			break;
		}

410 411 412
		pending_req = alloc_req();
		if (NULL == pending_req) {
			blkif->st_oo_req++;
K
Konrad Rzeszutek Wilk 已提交
413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 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
			more_to_do = 1;
			break;
		}

		switch (blkif->blk_protocol) {
		case BLKIF_PROTOCOL_NATIVE:
			memcpy(&req, RING_GET_REQUEST(&blk_rings->native, rc), sizeof(req));
			break;
		case BLKIF_PROTOCOL_X86_32:
			blkif_get_x86_32_req(&req, RING_GET_REQUEST(&blk_rings->x86_32, rc));
			break;
		case BLKIF_PROTOCOL_X86_64:
			blkif_get_x86_64_req(&req, RING_GET_REQUEST(&blk_rings->x86_64, rc));
			break;
		default:
			BUG();
		}
		blk_rings->common.req_cons = ++rc; /* before make_response() */

		/* Apply all sanity checks to /private copy/ of request. */
		barrier();

		switch (req.operation) {
		case BLKIF_OP_READ:
			blkif->st_rd_req++;
			dispatch_rw_block_io(blkif, &req, pending_req);
			break;
		case BLKIF_OP_WRITE_BARRIER:
			blkif->st_br_req++;
			/* fall through */
		case BLKIF_OP_WRITE:
			blkif->st_wr_req++;
			dispatch_rw_block_io(blkif, &req, pending_req);
			break;
		default:
			/* A good sign something is wrong: sleep for a while to
			 * avoid excessive CPU consumption by a bad guest. */
			msleep(1);
			DPRINTK("error: unknown block io operation [%d]\n",
				req.operation);
			make_response(blkif, req.id, req.operation,
				      BLKIF_RSP_ERROR);
			free_req(pending_req);
			break;
		}

		/* Yield point for this unbounded loop. */
		cond_resched();
	}

	return more_to_do;
}

466 467 468 469
/*
 * Transumation of the 'struct blkif_request' to a proper 'struct bio'
 * and call the 'submit_bio' to pass it to the underlaying storage.
 */
470
static void dispatch_rw_block_io(struct blkif_st *blkif,
J
Jeremy Fitzhardinge 已提交
471
				 struct blkif_request *req,
472
				 struct pending_req *pending_req)
K
Konrad Rzeszutek Wilk 已提交
473 474
{
	struct phys_req preq;
475
	struct seg_buf seg[BLKIF_MAX_SEGMENTS_PER_REQUEST];
K
Konrad Rzeszutek Wilk 已提交
476 477
	unsigned int nseg;
	struct bio *bio = NULL;
478
	struct bio *biolist[BLKIF_MAX_SEGMENTS_PER_REQUEST];
479
	int i, nbio = 0;
K
Konrad Rzeszutek Wilk 已提交
480
	int operation;
481 482
	struct blk_plug plug;
	struct request_queue *q;
K
Konrad Rzeszutek Wilk 已提交
483 484 485 486 487 488 489 490 491

	switch (req->operation) {
	case BLKIF_OP_READ:
		operation = READ;
		break;
	case BLKIF_OP_WRITE:
		operation = WRITE;
		break;
	case BLKIF_OP_WRITE_BARRIER:
T
Tom Goetz 已提交
492
		operation = WRITE_BARRIER;
K
Konrad Rzeszutek Wilk 已提交
493 494 495 496 497 498
		break;
	default:
		operation = 0; /* make gcc happy */
		BUG();
	}

499
	/* Check that the number of segments is sane. */
K
Konrad Rzeszutek Wilk 已提交
500
	nseg = req->nr_segments;
T
Tom Goetz 已提交
501
	if (unlikely(nseg == 0 && operation != WRITE_BARRIER) ||
K
Konrad Rzeszutek Wilk 已提交
502 503
	    unlikely(nseg > BLKIF_MAX_SEGMENTS_PER_REQUEST)) {
		DPRINTK("Bad number of segments in request (%d)\n", nseg);
504
		/* Haven't submitted any bio's yet. */
K
Konrad Rzeszutek Wilk 已提交
505 506 507 508
		goto fail_response;
	}

	preq.dev           = req->handle;
509
	preq.sector_number = req->u.rw.sector_number;
K
Konrad Rzeszutek Wilk 已提交
510 511 512 513 514 515 516 517
	preq.nr_sects      = 0;

	pending_req->blkif     = blkif;
	pending_req->id        = req->id;
	pending_req->operation = req->operation;
	pending_req->status    = BLKIF_RSP_OKAY;
	pending_req->nr_pages  = nseg;
	for (i = 0; i < nseg; i++) {
518 519 520 521
		seg[i].nsec = req->u.rw.seg[i].last_sect -
			req->u.rw.seg[i].first_sect + 1;
		if ((req->u.rw.seg[i].last_sect >= (PAGE_SIZE >> 9)) ||
		    (req->u.rw.seg[i].last_sect < req->u.rw.seg[i].first_sect))
K
Konrad Rzeszutek Wilk 已提交
522 523
			goto fail_response;
		preq.nr_sects += seg[i].nsec;
524 525 526 527 528 529 530

		if (((int)preq.sector_number|(int)seg[i].nsec) &
		    ((bdev_logical_block_size(preq.bdev) >> 9) - 1)) {
			DPRINTK("Misaligned I/O request from domain %d",
				blkif->domid);
			goto fail_response;
		}
K
Konrad Rzeszutek Wilk 已提交
531 532
	}

533 534 535 536 537 538
	if (vbd_translate(&preq, blkif, operation) != 0) {
		DPRINTK("access denied: %s of [%llu,%llu] on dev=%04x\n",
			operation == READ ? "read" : "write",
			preq.sector_number,
			preq.sector_number + preq.nr_sects, preq.dev);
		goto fail_response;
K
Konrad Rzeszutek Wilk 已提交
539
	}
540 541 542 543 544
	/* If we have failed at this point, we need to undo the M2P override,
	 * set gnttab_set_unmap_op on all of the grant references and perform
	 * the hypercall to unmap the grants - that is all done in
	 * fast_flush_area.
	 */
545
	if (xen_blk_map_buf(req, pending_req, seg))
K
Konrad Rzeszutek Wilk 已提交
546 547
		goto fail_flush;

548
	/* This corresponding blkif_put is done in __end_block_io_op */
K
Konrad Rzeszutek Wilk 已提交
549 550 551 552 553
	blkif_get(blkif);

	for (i = 0; i < nseg; i++) {
		while ((bio == NULL) ||
		       (bio_add_page(bio,
554
				     blkbk->pending_page(pending_req, i),
K
Konrad Rzeszutek Wilk 已提交
555 556
				     seg[i].nsec << 9,
				     seg[i].buf & ~PAGE_MASK) == 0)) {
557

558
			bio = biolist[nbio++] = bio_alloc(GFP_KERNEL, nseg-i);
K
Konrad Rzeszutek Wilk 已提交
559 560 561 562 563 564 565 566 567 568 569 570
			if (unlikely(bio == NULL))
				goto fail_put_bio;

			bio->bi_bdev    = preq.bdev;
			bio->bi_private = pending_req;
			bio->bi_end_io  = end_block_io_op;
			bio->bi_sector  = preq.sector_number;
		}

		preq.sector_number += seg[i].nsec;
	}

571
	/* This will be hit if the operation was a barrier. */
K
Konrad Rzeszutek Wilk 已提交
572
	if (!bio) {
T
Tom Goetz 已提交
573
		BUG_ON(operation != WRITE_BARRIER);
574
		bio = biolist[nbio++] = bio_alloc(GFP_KERNEL, 0);
K
Konrad Rzeszutek Wilk 已提交
575 576 577 578 579 580 581 582 583
		if (unlikely(bio == NULL))
			goto fail_put_bio;

		bio->bi_bdev    = preq.bdev;
		bio->bi_private = pending_req;
		bio->bi_end_io  = end_block_io_op;
		bio->bi_sector  = -1;
	}

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599

	/* We set it one so that the last submit_bio does not have to call
	 * atomic_inc.
	 */
	atomic_set(&pending_req->pendcnt, nbio);

	/* Get a reference count for the disk queue and start sending I/O */
	blk_get_queue(q);
	blk_start_plug(&plug);

	for (i = 0; i < nbio; i++)
		submit_bio(operation, biolist[i]);

	blk_finish_plug(&plug);
	/* Let the I/Os go.. */
	blk_put_queue(q);
K
Konrad Rzeszutek Wilk 已提交
600 601 602

	if (operation == READ)
		blkif->st_rd_sect += preq.nr_sects;
T
Tom Goetz 已提交
603
	else if (operation == WRITE || operation == WRITE_BARRIER)
K
Konrad Rzeszutek Wilk 已提交
604 605 606 607 608 609 610
		blkif->st_wr_sect += preq.nr_sects;

	return;

 fail_flush:
	fast_flush_area(pending_req);
 fail_response:
611
	/* Haven't submitted any bio's yet. */
K
Konrad Rzeszutek Wilk 已提交
612 613 614 615 616 617
	make_response(blkif, req->id, req->operation, BLKIF_RSP_ERROR);
	free_req(pending_req);
	msleep(1); /* back off a bit */
	return;

 fail_put_bio:
618 619
	for (i = 0; i < (nbio-1); i++)
		bio_put(biolist[i]);
K
Konrad Rzeszutek Wilk 已提交
620 621 622 623 624 625 626
	__end_block_io_op(pending_req, -EINVAL);
	msleep(1); /* back off a bit */
	return;
}



627 628
/*
 * Put a response on the ring on how the operation fared.
K
Konrad Rzeszutek Wilk 已提交
629
 */
630
static void make_response(struct blkif_st *blkif, u64 id,
K
Konrad Rzeszutek Wilk 已提交
631 632
			  unsigned short op, int st)
{
J
Jeremy Fitzhardinge 已提交
633
	struct blkif_response  resp;
K
Konrad Rzeszutek Wilk 已提交
634
	unsigned long     flags;
J
Jeremy Fitzhardinge 已提交
635
	union blkif_back_rings *blk_rings = &blkif->blk_rings;
K
Konrad Rzeszutek Wilk 已提交
636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
	int more_to_do = 0;
	int notify;

	resp.id        = id;
	resp.operation = op;
	resp.status    = st;

	spin_lock_irqsave(&blkif->blk_ring_lock, flags);
	/* Place on the response ring for the relevant domain. */
	switch (blkif->blk_protocol) {
	case BLKIF_PROTOCOL_NATIVE:
		memcpy(RING_GET_RESPONSE(&blk_rings->native, blk_rings->native.rsp_prod_pvt),
		       &resp, sizeof(resp));
		break;
	case BLKIF_PROTOCOL_X86_32:
		memcpy(RING_GET_RESPONSE(&blk_rings->x86_32, blk_rings->x86_32.rsp_prod_pvt),
		       &resp, sizeof(resp));
		break;
	case BLKIF_PROTOCOL_X86_64:
		memcpy(RING_GET_RESPONSE(&blk_rings->x86_64, blk_rings->x86_64.rsp_prod_pvt),
		       &resp, sizeof(resp));
		break;
	default:
		BUG();
	}
	blk_rings->common.rsp_prod_pvt++;
	RING_PUSH_RESPONSES_AND_CHECK_NOTIFY(&blk_rings->common, notify);
	if (blk_rings->common.rsp_prod_pvt == blk_rings->common.req_cons) {
		/*
		 * Tail check for pending requests. Allows frontend to avoid
		 * notifications if requests are already in flight (lower
		 * overheads and promotes batching).
		 */
		RING_FINAL_CHECK_FOR_REQUESTS(&blk_rings->common, more_to_do);

	} else if (RING_HAS_UNCONSUMED_REQUESTS(&blk_rings->common)) {
		more_to_do = 1;
	}

	spin_unlock_irqrestore(&blkif->blk_ring_lock, flags);

	if (more_to_do)
		blkif_notify_work(blkif);
	if (notify)
		notify_remote_via_irq(blkif->irq);
}

static int __init blkif_init(void)
{
	int i, mmap_pages;
686
	int rc = 0;
K
Konrad Rzeszutek Wilk 已提交
687

J
Jeremy Fitzhardinge 已提交
688
	if (!xen_pv_domain())
K
Konrad Rzeszutek Wilk 已提交
689 690
		return -ENODEV;

691
	blkbk = kzalloc(sizeof(struct xen_blkbk), GFP_KERNEL);
692 693 694 695 696
	if (!blkbk) {
		printk(KERN_ALERT "%s: out of memory!\n", __func__);
		return -ENOMEM;
	}

K
Konrad Rzeszutek Wilk 已提交
697 698
	mmap_pages = blkif_reqs * BLKIF_MAX_SEGMENTS_PER_REQUEST;

699
	blkbk->pending_reqs          = kmalloc(sizeof(blkbk->pending_reqs[0]) *
K
Konrad Rzeszutek Wilk 已提交
700
					blkif_reqs, GFP_KERNEL);
701 702 703 704
	blkbk->pending_grant_handles = kzalloc(sizeof(blkbk->pending_grant_handles[0]) *
					mmap_pages, GFP_KERNEL);
	blkbk->pending_pages         = kzalloc(sizeof(blkbk->pending_pages[0]) *
					mmap_pages, GFP_KERNEL);
K
Konrad Rzeszutek Wilk 已提交
705

706 707
	if (!blkbk->pending_reqs || !blkbk->pending_grant_handles ||
	    !blkbk->pending_pages) {
708
		rc = -ENOMEM;
K
Konrad Rzeszutek Wilk 已提交
709
		goto out_of_memory;
710
	}
K
Konrad Rzeszutek Wilk 已提交
711

712
	for (i = 0; i < mmap_pages; i++) {
713
		blkbk->pending_grant_handles[i] = BLKBACK_INVALID_HANDLE;
714
		blkbk->pending_pages[i] = alloc_page(GFP_KERNEL);
715 716 717 718 719
		if (blkbk->pending_pages[i] == NULL) {
			rc = -ENOMEM;
			goto out_of_memory;
		}
	}
720 721 722
	rc = blkif_interface_init();
	if (rc)
		goto failed_init;
K
Konrad Rzeszutek Wilk 已提交
723

724 725 726 727 728
	memset(blkbk->pending_reqs, 0, sizeof(blkbk->pending_reqs));

	INIT_LIST_HEAD(&blkbk->pending_free);
	spin_lock_init(&blkbk->pending_free_lock);
	init_waitqueue_head(&blkbk->pending_free_wq);
K
Konrad Rzeszutek Wilk 已提交
729 730

	for (i = 0; i < blkif_reqs; i++)
731 732
		list_add_tail(&blkbk->pending_reqs[i].free_list,
			      &blkbk->pending_free);
K
Konrad Rzeszutek Wilk 已提交
733

734 735 736
	rc = blkif_xenbus_init();
	if (rc)
		goto failed_init;
K
Konrad Rzeszutek Wilk 已提交
737 738 739 740

	return 0;

 out_of_memory:
741 742
	printk(KERN_ERR "%s: out of memory\n", __func__);
 failed_init:
743
	kfree(blkbk->pending_reqs);
744
	kfree(blkbk->pending_grant_handles);
745 746 747 748
	for (i = 0; i < mmap_pages; i++) {
		if (blkbk->pending_pages[i])
			__free_page(blkbk->pending_pages[i]);
	}
749 750
	kfree(blkbk->pending_pages);
	kfree(blkbk);
751
	blkbk = NULL;
752
	return rc;
K
Konrad Rzeszutek Wilk 已提交
753 754 755 756 757
}

module_init(blkif_init);

MODULE_LICENSE("Dual BSD/GPL");