blkback.c 21.0 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 173
/*
 * Give back a reference count on the underlaying storage.
 * It is OK to make multiple calls in this function as it
 * resets the plug to NULL when it is done on the first call.
 */
174
static void unplug_queue(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
175 176 177 178 179 180 181 182 183
{
	if (blkif->plug == NULL)
		return;
	if (blkif->plug->unplug_fn)
		blkif->plug->unplug_fn(blkif->plug);
	blk_put_queue(blkif->plug);
	blkif->plug = NULL;
}

184 185 186 187 188 189
/*
 * Take a reference count on the underlaying storage.
 * It is OK to call this multiple times as we check to make sure
 * not to double reference. We also give back a reference count
 * if it corresponds to another queue.
 */
190
static void plug_queue(struct blkif_st *blkif, struct block_device *bdev)
K
Konrad Rzeszutek Wilk 已提交
191
{
J
Jeremy Fitzhardinge 已提交
192
	struct request_queue *q = bdev_get_queue(bdev);
K
Konrad Rzeszutek Wilk 已提交
193 194 195 196 197 198 199 200

	if (q == blkif->plug)
		return;
	unplug_queue(blkif);
	blk_get_queue(q);
	blkif->plug = q;
}

201 202 203 204
/*
 * Unmap the grant references, and also remove the M2P over-rides
 * used in the 'pending_req'.
*/
205
static void fast_flush_area(struct pending_req *req)
K
Konrad Rzeszutek Wilk 已提交
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
{
	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);
225
	/* Note, we use invcount, so nr->pages, so we can't index
226 227
	 * using vaddr(req, i).
	 */
228 229 230 231 232 233 234 235 236
	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;
		}
	}
K
Konrad Rzeszutek Wilk 已提交
237 238
}

239
/*
K
Konrad Rzeszutek Wilk 已提交
240 241 242
 * SCHEDULER FUNCTIONS
 */

243
static void print_stats(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
244 245 246 247 248 249 250 251 252 253 254 255
{
	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)
{
256
	struct blkif_st *blkif = arg;
257
	struct vbd *vbd = &blkif->vbd;
K
Konrad Rzeszutek Wilk 已提交
258 259 260 261 262 263 264 265 266

	blkif_get(blkif);

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

	while (!kthread_should_stop()) {
		if (try_to_freeze())
			continue;
267 268
		if (unlikely(vbd->size != vbd_size(vbd)))
			vbd_resize(blkif);
K
Konrad Rzeszutek Wilk 已提交
269 270 271 272 273

		wait_event_interruptible(
			blkif->wq,
			blkif->waiting_reqs || kthread_should_stop());
		wait_event_interruptible(
274
			blkbk->pending_free_wq,
275 276
			!list_empty(&blkbk->pending_free) ||
			kthread_should_stop());
K
Konrad Rzeszutek Wilk 已提交
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299

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

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

		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;
}

300 301
/*
 * Completion callback on the bio's. Called as bh->b_end_io()
K
Konrad Rzeszutek Wilk 已提交
302 303
 */

304
static void __end_block_io_op(struct pending_req *pending_req, int error)
K
Konrad Rzeszutek Wilk 已提交
305 306 307 308 309 310 311 312 313 314 315 316 317
{
	/* 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;
	}

318 319
	/* If all of the bio's have completed it is time to unmap
	 * the grant references associated with 'request' and provide
320 321
	 * the proper response on the ring.
	 */
K
Konrad Rzeszutek Wilk 已提交
322 323 324 325 326 327 328 329 330
	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);
	}
}

331 332 333
/*
 * bio callback.
 */
J
Jeremy Fitzhardinge 已提交
334
static void end_block_io_op(struct bio *bio, int error)
K
Konrad Rzeszutek Wilk 已提交
335 336 337 338 339 340
{
	__end_block_io_op(bio->bi_private, error);
	bio_put(bio);
}


341 342
/*
 * Notification from the guest OS.
K
Konrad Rzeszutek Wilk 已提交
343 344
 */

345
static void blkif_notify_work(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
346 347 348 349 350
{
	blkif->waiting_reqs = 1;
	wake_up(&blkif->wq);
}

J
Jeremy Fitzhardinge 已提交
351
irqreturn_t blkif_be_int(int irq, void *dev_id)
K
Konrad Rzeszutek Wilk 已提交
352 353 354 355 356 357 358
{
	blkif_notify_work(dev_id);
	return IRQ_HANDLED;
}



359 360 361 362
/*
 * 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 已提交
363
 */
364
static int do_block_io_op(struct blkif_st *blkif)
K
Konrad Rzeszutek Wilk 已提交
365
{
J
Jeremy Fitzhardinge 已提交
366 367
	union blkif_back_rings *blk_rings = &blkif->blk_rings;
	struct blkif_request req;
368
	struct pending_req *pending_req;
K
Konrad Rzeszutek Wilk 已提交
369 370 371 372 373 374 375 376 377 378 379 380
	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;

381
		if (kthread_should_stop()) {
K
Konrad Rzeszutek Wilk 已提交
382 383 384 385
			more_to_do = 1;
			break;
		}

386 387 388
		pending_req = alloc_req();
		if (NULL == pending_req) {
			blkif->st_oo_req++;
K
Konrad Rzeszutek Wilk 已提交
389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 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
			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;
}

442 443 444 445
/*
 * Transumation of the 'struct blkif_request' to a proper 'struct bio'
 * and call the 'submit_bio' to pass it to the underlaying storage.
 */
446
static void dispatch_rw_block_io(struct blkif_st *blkif,
J
Jeremy Fitzhardinge 已提交
447
				 struct blkif_request *req,
448
				 struct pending_req *pending_req)
K
Konrad Rzeszutek Wilk 已提交
449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
{
	struct gnttab_map_grant_ref map[BLKIF_MAX_SEGMENTS_PER_REQUEST];
	struct phys_req preq;
	struct {
		unsigned long buf; unsigned int nsec;
	} seg[BLKIF_MAX_SEGMENTS_PER_REQUEST];
	unsigned int nseg;
	struct bio *bio = NULL;
	int ret, i;
	int operation;

	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 已提交
468
		operation = WRITE_BARRIER;
K
Konrad Rzeszutek Wilk 已提交
469 470 471 472 473 474
		break;
	default:
		operation = 0; /* make gcc happy */
		BUG();
	}

475
	/* Check that the number of segments is sane. */
K
Konrad Rzeszutek Wilk 已提交
476
	nseg = req->nr_segments;
T
Tom Goetz 已提交
477
	if (unlikely(nseg == 0 && operation != WRITE_BARRIER) ||
K
Konrad Rzeszutek Wilk 已提交
478 479 480 481 482 483
	    unlikely(nseg > BLKIF_MAX_SEGMENTS_PER_REQUEST)) {
		DPRINTK("Bad number of segments in request (%d)\n", nseg);
		goto fail_response;
	}

	preq.dev           = req->handle;
484
	preq.sector_number = req->u.rw.sector_number;
K
Konrad Rzeszutek Wilk 已提交
485 486 487 488 489 490 491 492
	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;

493 494
	/* 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
495 496
	 * corresponding grant reference for each page.
	 */
K
Konrad Rzeszutek Wilk 已提交
497 498 499
	for (i = 0; i < nseg; i++) {
		uint32_t flags;

500 501 502 503
		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 已提交
504 505 506 507 508 509 510
			goto fail_response;
		preq.nr_sects += seg[i].nsec;

		flags = GNTMAP_host_map;
		if (operation != READ)
			flags |= GNTMAP_readonly;
		gnttab_set_map_op(&map[i], vaddr(pending_req, i), flags,
511
				  req->u.rw.seg[i].gref, blkif->domid);
K
Konrad Rzeszutek Wilk 已提交
512 513 514 515 516
	}

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

517
	/* Now swizzel the MFN in our domain with the MFN from the other domain
518 519 520
	 * so that when we access vaddr(pending_req,i) it has the contents of
	 * the page from the other domain.
	 */
K
Konrad Rzeszutek Wilk 已提交
521 522 523 524 525 526 527 528 529 530 531
	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;
532

533 534 535 536
		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"\
537 538
				" %lx (ret: %d)\n", (unsigned long)
				map[i].dev_bus_addr, ret);
539
			/* We could switch over to GNTTABOP_copy */
540 541
			continue;
		}
K
Konrad Rzeszutek Wilk 已提交
542 543

		seg[i].buf  = map[i].dev_bus_addr |
544
			(req->u.rw.seg[i].first_sect << 9);
K
Konrad Rzeszutek Wilk 已提交
545 546
	}

547 548 549 550 551
	/* 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.
	 */
K
Konrad Rzeszutek Wilk 已提交
552 553 554 555 556 557 558 559 560 561 562
	if (ret)
		goto fail_flush;

	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_flush;
	}

563
	/* Get a reference count for the disk queue and start sending I/O */
K
Konrad Rzeszutek Wilk 已提交
564
	plug_queue(blkif, preq.bdev);
565 566

	/* We set it one so that the last submit_bio does not have to call
567 568
	 * atomic_inc.
	 */
K
Konrad Rzeszutek Wilk 已提交
569 570 571 572 573
	atomic_set(&pending_req->pendcnt, 1);
	blkif_get(blkif);

	for (i = 0; i < nseg; i++) {
		if (((int)preq.sector_number|(int)seg[i].nsec) &
574
		    ((bdev_logical_block_size(preq.bdev) >> 9) - 1)) {
K
Konrad Rzeszutek Wilk 已提交
575 576 577 578 579 580 581
			DPRINTK("Misaligned I/O request from domain %d",
				blkif->domid);
			goto fail_put_bio;
		}

		while ((bio == NULL) ||
		       (bio_add_page(bio,
582
				     blkbk->pending_page(pending_req, i),
K
Konrad Rzeszutek Wilk 已提交
583 584 585 586 587 588
				     seg[i].nsec << 9,
				     seg[i].buf & ~PAGE_MASK) == 0)) {
			if (bio) {
				atomic_inc(&pending_req->pendcnt);
				submit_bio(operation, bio);
			}
589

K
Konrad Rzeszutek Wilk 已提交
590 591 592 593 594 595 596 597 598 599 600 601 602
			bio = bio_alloc(GFP_KERNEL, nseg-i);
			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;
	}

603
	/* This will be hit if the operation was a barrier. */
K
Konrad Rzeszutek Wilk 已提交
604
	if (!bio) {
T
Tom Goetz 已提交
605
		BUG_ON(operation != WRITE_BARRIER);
K
Konrad Rzeszutek Wilk 已提交
606 607 608 609 610 611 612 613 614 615 616 617 618 619
		bio = bio_alloc(GFP_KERNEL, 0);
		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;
	}

	submit_bio(operation, bio);

	if (operation == READ)
		blkif->st_rd_sect += preq.nr_sects;
T
Tom Goetz 已提交
620
	else if (operation == WRITE || operation == WRITE_BARRIER)
K
Konrad Rzeszutek Wilk 已提交
621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643
		blkif->st_wr_sect += preq.nr_sects;

	return;

 fail_flush:
	fast_flush_area(pending_req);
 fail_response:
	make_response(blkif, req->id, req->operation, BLKIF_RSP_ERROR);
	free_req(pending_req);
	msleep(1); /* back off a bit */
	return;

 fail_put_bio:
	__end_block_io_op(pending_req, -EINVAL);
	if (bio)
		bio_put(bio);
	unplug_queue(blkif);
	msleep(1); /* back off a bit */
	return;
}



644 645
/*
 * Put a response on the ring on how the operation fared.
K
Konrad Rzeszutek Wilk 已提交
646
 */
647
static void make_response(struct blkif_st *blkif, u64 id,
K
Konrad Rzeszutek Wilk 已提交
648 649
			  unsigned short op, int st)
{
J
Jeremy Fitzhardinge 已提交
650
	struct blkif_response  resp;
K
Konrad Rzeszutek Wilk 已提交
651
	unsigned long     flags;
J
Jeremy Fitzhardinge 已提交
652
	union blkif_back_rings *blk_rings = &blkif->blk_rings;
K
Konrad Rzeszutek Wilk 已提交
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 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702
	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;
703
	int rc = 0;
K
Konrad Rzeszutek Wilk 已提交
704

J
Jeremy Fitzhardinge 已提交
705
	if (!xen_pv_domain())
K
Konrad Rzeszutek Wilk 已提交
706 707
		return -ENODEV;

708
	blkbk = kzalloc(sizeof(struct xen_blkbk), GFP_KERNEL);
709 710 711 712 713
	if (!blkbk) {
		printk(KERN_ALERT "%s: out of memory!\n", __func__);
		return -ENOMEM;
	}

K
Konrad Rzeszutek Wilk 已提交
714 715
	mmap_pages = blkif_reqs * BLKIF_MAX_SEGMENTS_PER_REQUEST;

716
	blkbk->pending_reqs          = kmalloc(sizeof(blkbk->pending_reqs[0]) *
K
Konrad Rzeszutek Wilk 已提交
717
					blkif_reqs, GFP_KERNEL);
718 719 720 721
	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 已提交
722

723 724
	if (!blkbk->pending_reqs || !blkbk->pending_grant_handles ||
	    !blkbk->pending_pages) {
725
		rc = -ENOMEM;
K
Konrad Rzeszutek Wilk 已提交
726
		goto out_of_memory;
727
	}
K
Konrad Rzeszutek Wilk 已提交
728

729
	for (i = 0; i < mmap_pages; i++) {
730
		blkbk->pending_grant_handles[i] = BLKBACK_INVALID_HANDLE;
731
		blkbk->pending_pages[i] = alloc_page(GFP_KERNEL);
732 733 734 735 736
		if (blkbk->pending_pages[i] == NULL) {
			rc = -ENOMEM;
			goto out_of_memory;
		}
	}
737 738 739
	rc = blkif_interface_init();
	if (rc)
		goto failed_init;
K
Konrad Rzeszutek Wilk 已提交
740

741 742 743 744 745
	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 已提交
746 747

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

751 752 753
	rc = blkif_xenbus_init();
	if (rc)
		goto failed_init;
K
Konrad Rzeszutek Wilk 已提交
754 755 756 757

	return 0;

 out_of_memory:
758 759
	printk(KERN_ERR "%s: out of memory\n", __func__);
 failed_init:
760
	kfree(blkbk->pending_reqs);
761
	kfree(blkbk->pending_grant_handles);
762 763 764 765
	for (i = 0; i < mmap_pages; i++) {
		if (blkbk->pending_pages[i])
			__free_page(blkbk->pending_pages[i]);
	}
766 767
	kfree(blkbk->pending_pages);
	kfree(blkbk);
768
	blkbk = NULL;
769
	return rc;
K
Konrad Rzeszutek Wilk 已提交
770 771 772 773 774
}

module_init(blkif_init);

MODULE_LICENSE("Dual BSD/GPL");