direct.c 28.3 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9
/*
 * linux/fs/nfs/direct.c
 *
 * Copyright (C) 2003 by Chuck Lever <cel@netapp.com>
 *
 * High-performance uncached I/O for the Linux NFS client
 *
 * There are important applications whose performance or correctness
 * depends on uncached access to file data.  Database clusters
10
 * (multiple copies of the same instance running on separate hosts)
L
Linus Torvalds 已提交
11
 * implement their own cache coherency protocol that subsumes file
12 13 14
 * system cache protocols.  Applications that process datasets
 * considerably larger than the client's memory do not always benefit
 * from a local cache.  A streaming video server, for instance, has no
L
Linus Torvalds 已提交
15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
 * need to cache the contents of a file.
 *
 * When an application requests uncached I/O, all read and write requests
 * are made directly to the server; data stored or fetched via these
 * requests is not cached in the Linux page cache.  The client does not
 * correct unaligned requests from applications.  All requested bytes are
 * held on permanent storage before a direct write system call returns to
 * an application.
 *
 * Solaris implements an uncached I/O facility called directio() that
 * is used for backups and sequential I/O to very large files.  Solaris
 * also supports uncaching whole NFS partitions with "-o forcedirectio,"
 * an undocumented mount option.
 *
 * Designed by Jeff Kimmel, Chuck Lever, and Trond Myklebust, with
 * help from Andrew Morton.
 *
 * 18 Dec 2001	Initial implementation for 2.4  --cel
 * 08 Jul 2002	Version for 2.4.19, with bug fixes --trondmy
 * 08 Jun 2003	Port to 2.5 APIs  --cel
 * 31 Mar 2004	Handle direct I/O without VFS support  --cel
 * 15 Sep 2004	Parallel async reads  --cel
37
 * 04 May 2005	support O_DIRECT with aio  --cel
L
Linus Torvalds 已提交
38 39 40 41 42 43 44 45 46
 *
 */

#include <linux/errno.h>
#include <linux/sched.h>
#include <linux/kernel.h>
#include <linux/file.h>
#include <linux/pagemap.h>
#include <linux/kref.h>
47
#include <linux/slab.h>
48
#include <linux/task_io_accounting_ops.h>
L
Linus Torvalds 已提交
49 50 51 52 53 54

#include <linux/nfs_fs.h>
#include <linux/nfs_page.h>
#include <linux/sunrpc/clnt.h>

#include <asm/uaccess.h>
A
Arun Sharma 已提交
55
#include <linux/atomic.h>
L
Linus Torvalds 已提交
56

57
#include "internal.h"
C
Chuck Lever 已提交
58 59
#include "iostat.h"

L
Linus Torvalds 已提交
60 61
#define NFSDBG_FACILITY		NFSDBG_VFS

62
static struct kmem_cache *nfs_direct_cachep;
L
Linus Torvalds 已提交
63 64 65 66 67 68

/*
 * This represents a set of asynchronous requests that we're waiting on
 */
struct nfs_direct_req {
	struct kref		kref;		/* release manager */
69 70

	/* I/O parameters */
71
	struct nfs_open_context	*ctx;		/* file open context info */
72
	struct nfs_lock_context *l_ctx;		/* Lock context info */
73
	struct kiocb *		iocb;		/* controlling i/o request */
74
	struct inode *		inode;		/* target file of i/o */
75 76

	/* completion state */
77
	atomic_t		io_count;	/* i/os we're waiting for */
78 79
	spinlock_t		lock;		/* protect completion state */
	ssize_t			count,		/* bytes actually processed */
L
Linus Torvalds 已提交
80
				error;		/* any reported error */
81
	struct completion	completion;	/* wait for i/o completion */
82 83

	/* commit state */
84
	struct list_head	rewrite_list;	/* saved nfs_write_data structs */
85
	struct nfs_commit_data *commit_data;	/* special write_data for commits */
86 87 88 89
	int			flags;
#define NFS_ODIRECT_DO_COMMIT		(1)	/* an unstable reply was received */
#define NFS_ODIRECT_RESCHED_WRITES	(2)	/* write verification failed */
	struct nfs_writeverf	verf;		/* unstable write verifier */
L
Linus Torvalds 已提交
90 91
};

92
static void nfs_direct_write_complete(struct nfs_direct_req *dreq, struct inode *inode);
93 94 95 96 97 98 99 100 101 102 103 104
static const struct rpc_call_ops nfs_write_direct_ops;

static inline void get_dreq(struct nfs_direct_req *dreq)
{
	atomic_inc(&dreq->io_count);
}

static inline int put_dreq(struct nfs_direct_req *dreq)
{
	return atomic_dec_and_test(&dreq->io_count);
}

L
Linus Torvalds 已提交
105
/**
106 107 108 109 110 111 112 113 114 115 116
 * nfs_direct_IO - NFS address space operation for direct I/O
 * @rw: direction (read or write)
 * @iocb: target I/O control block
 * @iov: array of vectors that define I/O buffer
 * @pos: offset in file to begin the operation
 * @nr_segs: size of iovec array
 *
 * The presence of this routine in the address space ops vector means
 * the NFS client supports direct I/O.  However, we shunt off direct
 * read and write requests before the VFS gets them, so this method
 * should never be called.
L
Linus Torvalds 已提交
117
 */
118 119 120
ssize_t nfs_direct_IO(int rw, struct kiocb *iocb, const struct iovec *iov, loff_t pos, unsigned long nr_segs)
{
	dprintk("NFS: nfs_direct_IO (%s) off/no(%Ld/%lu) EINVAL\n",
121
			iocb->ki_filp->f_path.dentry->d_name.name,
122
			(long long) pos, nr_segs);
123 124 125 126

	return -EINVAL;
}

127
static void nfs_direct_dirty_pages(struct page **pages, unsigned int pgbase, size_t count)
128
{
129
	unsigned int npages;
130
	unsigned int i;
131 132 133 134 135

	if (count == 0)
		return;
	pages += (pgbase >> PAGE_SHIFT);
	npages = (count + (pgbase & ~PAGE_MASK) + PAGE_SIZE - 1) >> PAGE_SHIFT;
136 137
	for (i = 0; i < npages; i++) {
		struct page *page = pages[i];
138
		if (!PageCompound(page))
139
			set_page_dirty(page);
140
	}
141 142
}

143
static void nfs_direct_release_pages(struct page **pages, unsigned int npages)
144
{
145
	unsigned int i;
146 147
	for (i = 0; i < npages; i++)
		page_cache_release(pages[i]);
148 149
}

150
static inline struct nfs_direct_req *nfs_direct_req_alloc(void)
L
Linus Torvalds 已提交
151
{
152 153
	struct nfs_direct_req *dreq;

154
	dreq = kmem_cache_alloc(nfs_direct_cachep, GFP_KERNEL);
155 156 157 158
	if (!dreq)
		return NULL;

	kref_init(&dreq->kref);
159
	kref_get(&dreq->kref);
160
	init_completion(&dreq->completion);
161
	INIT_LIST_HEAD(&dreq->rewrite_list);
162
	dreq->iocb = NULL;
163
	dreq->ctx = NULL;
164
	dreq->l_ctx = NULL;
165
	spin_lock_init(&dreq->lock);
166
	atomic_set(&dreq->io_count, 0);
167 168
	dreq->count = 0;
	dreq->error = 0;
169
	dreq->flags = 0;
170 171

	return dreq;
L
Linus Torvalds 已提交
172 173
}

174
static void nfs_direct_req_free(struct kref *kref)
L
Linus Torvalds 已提交
175 176
{
	struct nfs_direct_req *dreq = container_of(kref, struct nfs_direct_req, kref);
177

178 179
	if (dreq->l_ctx != NULL)
		nfs_put_lock_context(dreq->l_ctx);
180 181
	if (dreq->ctx != NULL)
		put_nfs_open_context(dreq->ctx);
L
Linus Torvalds 已提交
182 183 184
	kmem_cache_free(nfs_direct_cachep, dreq);
}

185 186 187 188 189
static void nfs_direct_req_release(struct nfs_direct_req *dreq)
{
	kref_put(&dreq->kref, nfs_direct_req_free);
}

190 191 192 193 194
/*
 * Collects and returns the final error value/byte-count.
 */
static ssize_t nfs_direct_wait(struct nfs_direct_req *dreq)
{
195
	ssize_t result = -EIOCBQUEUED;
196 197 198 199 200

	/* Async requests don't wait here */
	if (dreq->iocb)
		goto out;

201
	result = wait_for_completion_killable(&dreq->completion);
202 203

	if (!result)
204
		result = dreq->error;
205
	if (!result)
206
		result = dreq->count;
207 208 209 210 211

out:
	return (ssize_t) result;
}

212
/*
213 214
 * Synchronous I/O uses a stack-allocated iocb.  Thus we can't trust
 * the iocb is still valid here if this is a synchronous request.
215 216 217 218
 */
static void nfs_direct_complete(struct nfs_direct_req *dreq)
{
	if (dreq->iocb) {
219
		long res = (long) dreq->error;
220
		if (!res)
221
			res = (long) dreq->count;
222
		aio_complete(dreq->iocb, res, 0);
223 224
	}
	complete_all(&dreq->completion);
225

226
	nfs_direct_req_release(dreq);
227 228
}

229
/*
230 231 232
 * We must hold a reference to all the pages in this direct read request
 * until the RPCs complete.  This could be long *after* we are woken up in
 * nfs_direct_wait (for instance, if someone hits ^C on a slow server).
233
 */
T
Trond Myklebust 已提交
234
static void nfs_direct_read_result(struct rpc_task *task, void *calldata)
L
Linus Torvalds 已提交
235
{
T
Trond Myklebust 已提交
236
	struct nfs_read_data *data = calldata;
L
Linus Torvalds 已提交
237

238 239 240 241 242 243 244
	nfs_readpage_result(task, data);
}

static void nfs_direct_read_release(void *calldata)
{

	struct nfs_read_data *data = calldata;
245
	struct nfs_direct_req *dreq = (struct nfs_direct_req *)data->header->req;
246
	int status = data->task.tk_status;
247 248

	spin_lock(&dreq->lock);
249 250
	if (unlikely(status < 0)) {
		dreq->error = status;
251 252 253 254
		spin_unlock(&dreq->lock);
	} else {
		dreq->count += data->res.count;
		spin_unlock(&dreq->lock);
F
Fred Isaman 已提交
255
		nfs_direct_dirty_pages(data->pages.pagevec,
256 257 258
				data->args.pgbase,
				data->res.count);
	}
F
Fred Isaman 已提交
259
	nfs_direct_release_pages(data->pages.pagevec, data->pages.npages);
260 261 262

	if (put_dreq(dreq))
		nfs_direct_complete(dreq);
263
	nfs_readdata_release(data);
L
Linus Torvalds 已提交
264 265
}

T
Trond Myklebust 已提交
266
static const struct rpc_call_ops nfs_read_direct_ops = {
267
	.rpc_call_prepare = nfs_read_prepare,
T
Trond Myklebust 已提交
268
	.rpc_call_done = nfs_direct_read_result,
269
	.rpc_release = nfs_direct_read_release,
T
Trond Myklebust 已提交
270 271
};

272 273 274 275
static void nfs_direct_readhdr_release(struct nfs_read_header *rhdr)
{
	struct nfs_read_data *data = &rhdr->rpc_data;

F
Fred Isaman 已提交
276 277
	if (data->pages.pagevec != data->pages.page_array)
		kfree(data->pages.pagevec);
278 279 280
	nfs_readhdr_free(&rhdr->header);
}

281
/*
282 283 284 285 286
 * For each rsize'd chunk of the user's buffer, dispatch an NFS READ
 * operation.  If nfs_readdata_alloc() or get_user_pages() fails,
 * bail and stop sending more reads.  Read length accounting is
 * handled automatically by nfs_direct_read_result().  Otherwise, if
 * no requests have been sent, just return an error.
L
Linus Torvalds 已提交
287
 */
288 289 290
static ssize_t nfs_direct_read_schedule_segment(struct nfs_direct_req *dreq,
						const struct iovec *iov,
						loff_t pos)
L
Linus Torvalds 已提交
291
{
292
	struct nfs_open_context *ctx = dreq->ctx;
293
	struct inode *inode = ctx->dentry->d_inode;
294 295
	unsigned long user_addr = (unsigned long)iov->iov_base;
	size_t count = iov->iov_len;
296
	size_t rsize = NFS_SERVER(inode)->rsize;
297
	struct rpc_task *task;
298 299 300
	struct rpc_message msg = {
		.rpc_cred = ctx->cred,
	};
301 302
	struct rpc_task_setup task_setup_data = {
		.rpc_client = NFS_CLIENT(inode),
303
		.rpc_message = &msg,
304
		.callback_ops = &nfs_read_direct_ops,
305
		.workqueue = nfsiod_workqueue,
306 307
		.flags = RPC_TASK_ASYNC,
	};
308 309 310 311
	unsigned int pgbase;
	int result;
	ssize_t started = 0;

L
Linus Torvalds 已提交
312
	do {
313
		struct nfs_read_header *rhdr;
314
		struct nfs_read_data *data;
F
Fred Isaman 已提交
315
		struct nfs_page_array *pages;
316
		size_t bytes;
L
Linus Torvalds 已提交
317

318 319 320
		pgbase = user_addr & ~PAGE_MASK;
		bytes = min(rsize,count);

321
		result = -ENOMEM;
322 323
		rhdr = nfs_readhdr_alloc(nfs_page_array_len(pgbase, bytes));
		if (unlikely(!rhdr))
324
			break;
325
		data = &rhdr->rpc_data;
F
Fred Isaman 已提交
326
		pages = &data->pages;
327 328 329

		down_read(&current->mm->mmap_sem);
		result = get_user_pages(current, current->mm, user_addr,
F
Fred Isaman 已提交
330
					pages->npages, 1, 0, pages->pagevec, NULL);
331
		up_read(&current->mm->mmap_sem);
332
		if (result < 0) {
333
			nfs_direct_readhdr_release(rhdr);
334 335
			break;
		}
F
Fred Isaman 已提交
336
		if ((unsigned)result < pages->npages) {
337 338
			bytes = result * PAGE_SIZE;
			if (bytes <= pgbase) {
F
Fred Isaman 已提交
339
				nfs_direct_release_pages(pages->pagevec, result);
340
				nfs_direct_readhdr_release(rhdr);
341 342 343
				break;
			}
			bytes -= pgbase;
F
Fred Isaman 已提交
344
			pages->npages = result;
345 346 347
		}

		get_dreq(dreq);
348

349 350 351
		rhdr->header.req = (struct nfs_page *) dreq;
		rhdr->header.inode = inode;
		rhdr->header.cred = msg.rpc_cred;
L
Linus Torvalds 已提交
352
		data->args.fh = NFS_FH(inode);
353
		data->args.context = get_nfs_open_context(ctx);
354
		data->args.lock_context = dreq->l_ctx;
355
		data->args.offset = pos;
L
Linus Torvalds 已提交
356
		data->args.pgbase = pgbase;
F
Fred Isaman 已提交
357
		data->args.pages = pages->pagevec;
L
Linus Torvalds 已提交
358 359 360 361
		data->args.count = bytes;
		data->res.fattr = &data->fattr;
		data->res.eof = 0;
		data->res.count = bytes;
362
		nfs_fattr_init(&data->fattr);
363 364
		msg.rpc_argp = &data->args;
		msg.rpc_resp = &data->res;
L
Linus Torvalds 已提交
365

366
		task_setup_data.task = &data->task;
367
		task_setup_data.callback_data = data;
368
		NFS_PROTO(inode)->read_setup(data, &msg);
L
Linus Torvalds 已提交
369

370
		task = rpc_run_task(&task_setup_data);
371 372
		if (IS_ERR(task))
			break;
L
Linus Torvalds 已提交
373

C
Chuck Lever 已提交
374 375
		dprintk("NFS: %5u initiated direct read call "
			"(req %s/%Ld, %zu bytes @ offset %Lu)\n",
376
				task->tk_pid,
L
Linus Torvalds 已提交
377 378 379 380
				inode->i_sb->s_id,
				(long long)NFS_FILEID(inode),
				bytes,
				(unsigned long long)data->args.offset);
381
		rpc_put_task(task);
L
Linus Torvalds 已提交
382

383 384
		started += bytes;
		user_addr += bytes;
385
		pos += bytes;
386
		/* FIXME: Remove this unnecessary math from final patch */
L
Linus Torvalds 已提交
387 388
		pgbase += bytes;
		pgbase &= ~PAGE_MASK;
389
		BUG_ON(pgbase != (user_addr & ~PAGE_MASK));
L
Linus Torvalds 已提交
390 391 392

		count -= bytes;
	} while (count != 0);
393 394

	if (started)
395
		return started;
396
	return result < 0 ? (ssize_t) result : -EFAULT;
L
Linus Torvalds 已提交
397 398
}

399 400 401 402 403 404 405 406 407 408 409 410 411
static ssize_t nfs_direct_read_schedule_iovec(struct nfs_direct_req *dreq,
					      const struct iovec *iov,
					      unsigned long nr_segs,
					      loff_t pos)
{
	ssize_t result = -EINVAL;
	size_t requested_bytes = 0;
	unsigned long seg;

	get_dreq(dreq);

	for (seg = 0; seg < nr_segs; seg++) {
		const struct iovec *vec = &iov[seg];
412
		result = nfs_direct_read_schedule_segment(dreq, vec, pos);
413 414 415 416 417 418 419 420
		if (result < 0)
			break;
		requested_bytes += result;
		if ((size_t)result < vec->iov_len)
			break;
		pos += vec->iov_len;
	}

421 422 423 424 425 426 427 428 429
	/*
	 * If no bytes were started, return the error, and let the
	 * generic layer handle the completion.
	 */
	if (requested_bytes == 0) {
		nfs_direct_req_release(dreq);
		return result < 0 ? result : -EIO;
	}

430 431
	if (put_dreq(dreq))
		nfs_direct_complete(dreq);
432
	return 0;
433 434
}

435 436
static ssize_t nfs_direct_read(struct kiocb *iocb, const struct iovec *iov,
			       unsigned long nr_segs, loff_t pos)
L
Linus Torvalds 已提交
437
{
438
	ssize_t result = -ENOMEM;
439
	struct inode *inode = iocb->ki_filp->f_mapping->host;
L
Linus Torvalds 已提交
440 441
	struct nfs_direct_req *dreq;

442
	dreq = nfs_direct_req_alloc();
443 444
	if (dreq == NULL)
		goto out;
L
Linus Torvalds 已提交
445

C
Chuck Lever 已提交
446
	dreq->inode = inode;
447
	dreq->ctx = get_nfs_open_context(nfs_file_open_context(iocb->ki_filp));
448 449 450
	dreq->l_ctx = nfs_get_lock_context(dreq->ctx);
	if (dreq->l_ctx == NULL)
		goto out_release;
451 452
	if (!is_sync_kiocb(iocb))
		dreq->iocb = iocb;
L
Linus Torvalds 已提交
453

454
	result = nfs_direct_read_schedule_iovec(dreq, iov, nr_segs, pos);
455 456
	if (!result)
		result = nfs_direct_wait(dreq);
457
out_release:
458
	nfs_direct_req_release(dreq);
459
out:
L
Linus Torvalds 已提交
460 461 462
	return result;
}

463 464 465 466
static void nfs_direct_writehdr_release(struct nfs_write_header *whdr)
{
	struct nfs_write_data *data = &whdr->rpc_data;

F
Fred Isaman 已提交
467 468
	if (data->pages.pagevec != data->pages.page_array)
		kfree(data->pages.pagevec);
469 470 471
	nfs_writehdr_free(&whdr->header);
}

472
static void nfs_direct_free_writedata(struct nfs_direct_req *dreq)
L
Linus Torvalds 已提交
473
{
474
	while (!list_empty(&dreq->rewrite_list)) {
475 476
		struct nfs_pgio_header *hdr = list_entry(dreq->rewrite_list.next, struct nfs_pgio_header, pages);
		struct nfs_write_header *whdr = container_of(hdr, struct nfs_write_header, header);
F
Fred Isaman 已提交
477 478
		struct nfs_page_array *p = &whdr->rpc_data.pages;

479
		list_del(&hdr->pages);
F
Fred Isaman 已提交
480
		nfs_direct_release_pages(p->pagevec, p->npages);
481
		nfs_direct_writehdr_release(whdr);
482 483
	}
}
L
Linus Torvalds 已提交
484

485 486 487
#if defined(CONFIG_NFS_V3) || defined(CONFIG_NFS_V4)
static void nfs_direct_write_reschedule(struct nfs_direct_req *dreq)
{
488 489 490
	struct inode *inode = dreq->inode;
	struct list_head *p;
	struct nfs_write_data *data;
491
	struct nfs_pgio_header *hdr;
492
	struct rpc_task *task;
493 494 495
	struct rpc_message msg = {
		.rpc_cred = dreq->ctx->cred,
	};
496 497
	struct rpc_task_setup task_setup_data = {
		.rpc_client = NFS_CLIENT(inode),
T
Terry Loftin 已提交
498
		.rpc_message = &msg,
499
		.callback_ops = &nfs_write_direct_ops,
500
		.workqueue = nfsiod_workqueue,
501 502
		.flags = RPC_TASK_ASYNC,
	};
L
Linus Torvalds 已提交
503

504
	dreq->count = 0;
505 506 507
	get_dreq(dreq);

	list_for_each(p, &dreq->rewrite_list) {
508 509
		hdr = list_entry(p, struct nfs_pgio_header, pages);
		data = &(container_of(hdr, struct nfs_write_header, header))->rpc_data;
510 511 512

		get_dreq(dreq);

513 514 515
		/* Use stable writes */
		data->args.stable = NFS_FILE_SYNC;

516 517 518 519 520 521 522 523 524 525 526
		/*
		 * Reset data->res.
		 */
		nfs_fattr_init(&data->fattr);
		data->res.count = data->args.count;
		memset(&data->verf, 0, sizeof(data->verf));

		/*
		 * Reuse data->task; data->args should not have changed
		 * since the original request was sent.
		 */
527
		task_setup_data.task = &data->task;
528
		task_setup_data.callback_data = data;
529 530 531
		msg.rpc_argp = &data->args;
		msg.rpc_resp = &data->res;
		NFS_PROTO(inode)->write_setup(data, &msg);
532 533 534 535

		/*
		 * We're called via an RPC callback, so BKL is already held.
		 */
536 537 538
		task = rpc_run_task(&task_setup_data);
		if (!IS_ERR(task))
			rpc_put_task(task);
539 540 541 542 543 544 545 546

		dprintk("NFS: %5u rescheduled direct write call (req %s/%Ld, %u bytes @ offset %Lu)\n",
				data->task.tk_pid,
				inode->i_sb->s_id,
				(long long)NFS_FILEID(inode),
				data->args.count,
				(unsigned long long)data->args.offset);
	}
547

548 549
	if (put_dreq(dreq))
		nfs_direct_write_complete(dreq, inode);
550 551 552 553
}

static void nfs_direct_commit_result(struct rpc_task *task, void *calldata)
{
554
	struct nfs_commit_data *data = calldata;
555 556

	/* Call the NFS version-specific code */
557 558 559 560 561
	NFS_PROTO(data->inode)->commit_done(task, data);
}

static void nfs_direct_commit_release(void *calldata)
{
562 563
	struct nfs_commit_data *data = calldata;
	struct nfs_direct_req *dreq = data->dreq;
564 565 566
	int status = data->task.tk_status;

	if (status < 0) {
567
		dprintk("NFS: %5u commit failed with error %d.\n",
568
				data->task.tk_pid, status);
569
		dreq->flags = NFS_ODIRECT_RESCHED_WRITES;
570
	} else if (memcmp(&dreq->verf, &data->verf, sizeof(data->verf))) {
571
		dprintk("NFS: %5u commit verify failed\n", data->task.tk_pid);
572
		dreq->flags = NFS_ODIRECT_RESCHED_WRITES;
L
Linus Torvalds 已提交
573 574
	}

575
	dprintk("NFS: %5u commit returned %d\n", data->task.tk_pid, status);
576
	nfs_direct_write_complete(dreq, data->inode);
T
Trond Myklebust 已提交
577
	nfs_commit_free(data);
L
Linus Torvalds 已提交
578 579
}

580
static const struct rpc_call_ops nfs_commit_direct_ops = {
581
	.rpc_call_prepare = nfs_commit_prepare,
582
	.rpc_call_done = nfs_direct_commit_result,
583
	.rpc_release = nfs_direct_commit_release,
584 585 586
};

static void nfs_direct_commit_schedule(struct nfs_direct_req *dreq)
L
Linus Torvalds 已提交
587
{
588
	struct nfs_commit_data *data = dreq->commit_data;
589
	struct rpc_task *task;
590 591 592 593 594
	struct rpc_message msg = {
		.rpc_argp = &data->args,
		.rpc_resp = &data->res,
		.rpc_cred = dreq->ctx->cred,
	};
595
	struct rpc_task_setup task_setup_data = {
596
		.task = &data->task,
597
		.rpc_client = NFS_CLIENT(dreq->inode),
598
		.rpc_message = &msg,
599 600
		.callback_ops = &nfs_commit_direct_ops,
		.callback_data = data,
601
		.workqueue = nfsiod_workqueue,
602 603
		.flags = RPC_TASK_ASYNC,
	};
L
Linus Torvalds 已提交
604

605
	data->inode = dreq->inode;
606
	data->cred = msg.rpc_cred;
L
Linus Torvalds 已提交
607

608
	data->args.fh = NFS_FH(data->inode);
609 610
	data->args.offset = 0;
	data->args.count = 0;
611 612
	data->res.fattr = &data->fattr;
	data->res.verf = &data->verf;
613
	nfs_fattr_init(&data->fattr);
L
Linus Torvalds 已提交
614

615
	NFS_PROTO(data->inode)->commit_setup(data, &msg);
L
Linus Torvalds 已提交
616

617 618
	/* Note: task.tk_ops->rpc_release will free dreq->commit_data */
	dreq->commit_data = NULL;
L
Linus Torvalds 已提交
619

620
	dprintk("NFS: %5u initiated commit call\n", data->task.tk_pid);
L
Linus Torvalds 已提交
621

622 623 624
	task = rpc_run_task(&task_setup_data);
	if (!IS_ERR(task))
		rpc_put_task(task);
625
}
L
Linus Torvalds 已提交
626

627 628 629
static void nfs_direct_write_complete(struct nfs_direct_req *dreq, struct inode *inode)
{
	int flags = dreq->flags;
L
Linus Torvalds 已提交
630

631 632 633 634
	dreq->flags = 0;
	switch (flags) {
		case NFS_ODIRECT_DO_COMMIT:
			nfs_direct_commit_schedule(dreq);
L
Linus Torvalds 已提交
635
			break;
636 637 638 639 640 641 642
		case NFS_ODIRECT_RESCHED_WRITES:
			nfs_direct_write_reschedule(dreq);
			break;
		default:
			if (dreq->commit_data != NULL)
				nfs_commit_free(dreq->commit_data);
			nfs_direct_free_writedata(dreq);
643
			nfs_zap_mapping(inode, inode->i_mapping);
644 645 646
			nfs_direct_complete(dreq);
	}
}
L
Linus Torvalds 已提交
647

648 649
static void nfs_alloc_commit_data(struct nfs_direct_req *dreq)
{
650
	dreq->commit_data = nfs_commitdata_alloc();
651
	if (dreq->commit_data != NULL)
652
		dreq->commit_data->dreq = dreq;
653 654 655 656 657 658
}
#else
static inline void nfs_alloc_commit_data(struct nfs_direct_req *dreq)
{
	dreq->commit_data = NULL;
}
L
Linus Torvalds 已提交
659

660 661 662
static void nfs_direct_write_complete(struct nfs_direct_req *dreq, struct inode *inode)
{
	nfs_direct_free_writedata(dreq);
663
	nfs_zap_mapping(inode, inode->i_mapping);
664 665 666
	nfs_direct_complete(dreq);
}
#endif
L
Linus Torvalds 已提交
667

668
static void nfs_direct_write_result(struct rpc_task *task, void *calldata)
L
Linus Torvalds 已提交
669
{
670 671
	struct nfs_write_data *data = calldata;

672
	nfs_writeback_done(task, data);
673 674 675 676 677 678 679 680 681
}

/*
 * NB: Return the value of the first error return code.  Subsequent
 *     errors after the first one are ignored.
 */
static void nfs_direct_write_release(void *calldata)
{
	struct nfs_write_data *data = calldata;
682 683
	struct nfs_pgio_header *hdr = data->header;
	struct nfs_direct_req *dreq = (struct nfs_direct_req *) hdr->req;
684
	int status = data->task.tk_status;
685

686
	spin_lock(&dreq->lock);
L
Linus Torvalds 已提交
687

688
	if (unlikely(status < 0)) {
689
		/* An error has occurred, so we should not commit */
690
		dreq->flags = 0;
691 692
		dreq->error = status;
	}
693 694
	if (unlikely(dreq->error != 0))
		goto out_unlock;
695 696

	dreq->count += data->res.count;
L
Linus Torvalds 已提交
697

698 699 700 701 702
	if (data->res.verf->committed != NFS_FILE_SYNC) {
		switch (dreq->flags) {
			case 0:
				memcpy(&dreq->verf, &data->verf, sizeof(dreq->verf));
				dreq->flags = NFS_ODIRECT_DO_COMMIT;
L
Linus Torvalds 已提交
703
				break;
704 705
			case NFS_ODIRECT_DO_COMMIT:
				if (memcmp(&dreq->verf, &data->verf, sizeof(dreq->verf))) {
706
					dprintk("NFS: %5u write verify failed\n", data->task.tk_pid);
707 708
					dreq->flags = NFS_ODIRECT_RESCHED_WRITES;
				}
L
Linus Torvalds 已提交
709 710
		}
	}
711
out_unlock:
712
	spin_unlock(&dreq->lock);
L
Linus Torvalds 已提交
713

714
	if (put_dreq(dreq))
715
		nfs_direct_write_complete(dreq, hdr->inode);
716 717 718
}

static const struct rpc_call_ops nfs_write_direct_ops = {
719
	.rpc_call_prepare = nfs_write_prepare,
720
	.rpc_call_done = nfs_direct_write_result,
721
	.rpc_release = nfs_direct_write_release,
722 723 724
};

/*
725 726 727 728 729
 * For each wsize'd chunk of the user's buffer, dispatch an NFS WRITE
 * operation.  If nfs_writedata_alloc() or get_user_pages() fails,
 * bail and stop sending more writes.  Write length accounting is
 * handled automatically by nfs_direct_write_result().  Otherwise, if
 * no requests have been sent, just return an error.
730
 */
731 732 733
static ssize_t nfs_direct_write_schedule_segment(struct nfs_direct_req *dreq,
						 const struct iovec *iov,
						 loff_t pos, int sync)
734
{
735
	struct nfs_open_context *ctx = dreq->ctx;
736
	struct inode *inode = ctx->dentry->d_inode;
737 738
	unsigned long user_addr = (unsigned long)iov->iov_base;
	size_t count = iov->iov_len;
739
	struct rpc_task *task;
740 741 742
	struct rpc_message msg = {
		.rpc_cred = ctx->cred,
	};
743 744
	struct rpc_task_setup task_setup_data = {
		.rpc_client = NFS_CLIENT(inode),
745
		.rpc_message = &msg,
746
		.callback_ops = &nfs_write_direct_ops,
747
		.workqueue = nfsiod_workqueue,
748 749
		.flags = RPC_TASK_ASYNC,
	};
750
	size_t wsize = NFS_SERVER(inode)->wsize;
751 752 753
	unsigned int pgbase;
	int result;
	ssize_t started = 0;
754

L
Linus Torvalds 已提交
755
	do {
756
		struct nfs_write_header *whdr;
757
		struct nfs_write_data *data;
F
Fred Isaman 已提交
758
		struct nfs_page_array *pages;
759 760
		size_t bytes;

761 762 763
		pgbase = user_addr & ~PAGE_MASK;
		bytes = min(wsize,count);

764
		result = -ENOMEM;
765 766
		whdr = nfs_writehdr_alloc(nfs_page_array_len(pgbase, bytes));
		if (unlikely(!whdr))
767 768
			break;

769
		data = &whdr->rpc_data;
F
Fred Isaman 已提交
770
		pages = &data->pages;
771

772 773
		down_read(&current->mm->mmap_sem);
		result = get_user_pages(current, current->mm, user_addr,
F
Fred Isaman 已提交
774
					pages->npages, 0, 0, pages->pagevec, NULL);
775
		up_read(&current->mm->mmap_sem);
776
		if (result < 0) {
777
			nfs_direct_writehdr_release(whdr);
778 779
			break;
		}
F
Fred Isaman 已提交
780
		if ((unsigned)result < pages->npages) {
781 782
			bytes = result * PAGE_SIZE;
			if (bytes <= pgbase) {
F
Fred Isaman 已提交
783
				nfs_direct_release_pages(pages->pagevec, result);
784
				nfs_direct_writehdr_release(whdr);
785 786 787
				break;
			}
			bytes -= pgbase;
F
Fred Isaman 已提交
788
			pages->npages = result;
789 790 791 792
		}

		get_dreq(dreq);

793
		list_move_tail(&whdr->header.pages, &dreq->rewrite_list);
794

795 796 797
		whdr->header.req = (struct nfs_page *) dreq;
		whdr->header.inode = inode;
		whdr->header.cred = msg.rpc_cred;
798
		data->args.fh = NFS_FH(inode);
T
Trond Myklebust 已提交
799
		data->args.context = ctx;
800
		data->args.lock_context = dreq->l_ctx;
801
		data->args.offset = pos;
802
		data->args.pgbase = pgbase;
F
Fred Isaman 已提交
803
		data->args.pages = pages->pagevec;
804
		data->args.count = bytes;
805
		data->args.stable = sync;
806 807
		data->res.fattr = &data->fattr;
		data->res.count = bytes;
808
		data->res.verf = &data->verf;
809
		nfs_fattr_init(&data->fattr);
810

811
		task_setup_data.task = &data->task;
812
		task_setup_data.callback_data = data;
813 814 815
		msg.rpc_argp = &data->args;
		msg.rpc_resp = &data->res;
		NFS_PROTO(inode)->write_setup(data, &msg);
L
Linus Torvalds 已提交
816

817
		task = rpc_run_task(&task_setup_data);
818 819
		if (IS_ERR(task))
			break;
L
Linus Torvalds 已提交
820

C
Chuck Lever 已提交
821 822
		dprintk("NFS: %5u initiated direct write call "
			"(req %s/%Ld, %zu bytes @ offset %Lu)\n",
823
				task->tk_pid,
824 825 826 827
				inode->i_sb->s_id,
				(long long)NFS_FILEID(inode),
				bytes,
				(unsigned long long)data->args.offset);
828
		rpc_put_task(task);
L
Linus Torvalds 已提交
829

830 831
		started += bytes;
		user_addr += bytes;
832
		pos += bytes;
833 834

		/* FIXME: Remove this useless math from the final patch */
835 836
		pgbase += bytes;
		pgbase &= ~PAGE_MASK;
837
		BUG_ON(pgbase != (user_addr & ~PAGE_MASK));
L
Linus Torvalds 已提交
838

839 840
		count -= bytes;
	} while (count != 0);
841 842

	if (started)
843
		return started;
844
	return result < 0 ? (ssize_t) result : -EFAULT;
845
}
L
Linus Torvalds 已提交
846

847 848 849 850 851 852 853 854 855 856 857 858 859
static ssize_t nfs_direct_write_schedule_iovec(struct nfs_direct_req *dreq,
					       const struct iovec *iov,
					       unsigned long nr_segs,
					       loff_t pos, int sync)
{
	ssize_t result = 0;
	size_t requested_bytes = 0;
	unsigned long seg;

	get_dreq(dreq);

	for (seg = 0; seg < nr_segs; seg++) {
		const struct iovec *vec = &iov[seg];
860 861
		result = nfs_direct_write_schedule_segment(dreq, vec,
							   pos, sync);
862 863 864 865 866 867 868 869
		if (result < 0)
			break;
		requested_bytes += result;
		if ((size_t)result < vec->iov_len)
			break;
		pos += vec->iov_len;
	}

870 871 872 873 874 875 876 877 878
	/*
	 * If no bytes were started, return the error, and let the
	 * generic layer handle the completion.
	 */
	if (requested_bytes == 0) {
		nfs_direct_req_release(dreq);
		return result < 0 ? result : -EIO;
	}

879 880
	if (put_dreq(dreq))
		nfs_direct_write_complete(dreq, dreq->inode);
881
	return 0;
882 883
}

884 885 886
static ssize_t nfs_direct_write(struct kiocb *iocb, const struct iovec *iov,
				unsigned long nr_segs, loff_t pos,
				size_t count)
887
{
888
	ssize_t result = -ENOMEM;
889
	struct inode *inode = iocb->ki_filp->f_mapping->host;
890
	struct nfs_direct_req *dreq;
891
	size_t wsize = NFS_SERVER(inode)->wsize;
892
	int sync = NFS_UNSTABLE;
L
Linus Torvalds 已提交
893

894
	dreq = nfs_direct_req_alloc();
895
	if (!dreq)
896
		goto out;
897 898
	nfs_alloc_commit_data(dreq);

899
	if (dreq->commit_data == NULL || count <= wsize)
900
		sync = NFS_FILE_SYNC;
L
Linus Torvalds 已提交
901

902
	dreq->inode = inode;
903
	dreq->ctx = get_nfs_open_context(nfs_file_open_context(iocb->ki_filp));
904
	dreq->l_ctx = nfs_get_lock_context(dreq->ctx);
905
	if (dreq->l_ctx == NULL)
906
		goto out_release;
907 908
	if (!is_sync_kiocb(iocb))
		dreq->iocb = iocb;
L
Linus Torvalds 已提交
909

910
	result = nfs_direct_write_schedule_iovec(dreq, iov, nr_segs, pos, sync);
911 912
	if (!result)
		result = nfs_direct_wait(dreq);
913
out_release:
914
	nfs_direct_req_release(dreq);
915
out:
L
Linus Torvalds 已提交
916 917 918 919 920 921
	return result;
}

/**
 * nfs_file_direct_read - file direct read operation for NFS files
 * @iocb: target I/O control block
922 923
 * @iov: vector of user buffers into which to read data
 * @nr_segs: size of iov vector
924
 * @pos: byte offset in file where reading starts
L
Linus Torvalds 已提交
925 926 927 928 929 930
 *
 * We use this function for direct reads instead of calling
 * generic_file_aio_read() in order to avoid gfar's check to see if
 * the request starts before the end of the file.  For that check
 * to work, we must generate a GETATTR before each direct read, and
 * even then there is a window between the GETATTR and the subsequent
931
 * READ where the file size could change.  Our preference is simply
L
Linus Torvalds 已提交
932 933
 * to do all reads the application wants, and the server will take
 * care of managing the end of file boundary.
934
 *
L
Linus Torvalds 已提交
935 936 937 938 939
 * This function also eliminates unnecessarily updating the file's
 * atime locally, as the NFS server sets the file's atime, and this
 * client must read the updated atime from the server back into its
 * cache.
 */
940 941
ssize_t nfs_file_direct_read(struct kiocb *iocb, const struct iovec *iov,
				unsigned long nr_segs, loff_t pos)
L
Linus Torvalds 已提交
942 943 944 945
{
	ssize_t retval = -EINVAL;
	struct file *file = iocb->ki_filp;
	struct address_space *mapping = file->f_mapping;
946 947 948 949
	size_t count;

	count = iov_length(iov, nr_segs);
	nfs_add_stats(mapping->host, NFSIOS_DIRECTREADBYTES, count);
L
Linus Torvalds 已提交
950

C
Chuck Lever 已提交
951
	dfprintk(FILE, "NFS: direct read(%s/%s, %zd@%Ld)\n",
952 953
		file->f_path.dentry->d_parent->d_name.name,
		file->f_path.dentry->d_name.name,
954
		count, (long long) pos);
L
Linus Torvalds 已提交
955 956 957 958 959

	retval = 0;
	if (!count)
		goto out;

T
Trond Myklebust 已提交
960 961 962
	retval = nfs_sync_mapping(mapping);
	if (retval)
		goto out;
L
Linus Torvalds 已提交
963

964 965
	task_io_account_read(count);

966
	retval = nfs_direct_read(iocb, iov, nr_segs, pos);
L
Linus Torvalds 已提交
967
	if (retval > 0)
968
		iocb->ki_pos = pos + retval;
L
Linus Torvalds 已提交
969 970 971 972 973 974 975 976

out:
	return retval;
}

/**
 * nfs_file_direct_write - file direct write operation for NFS files
 * @iocb: target I/O control block
977 978
 * @iov: vector of user buffers from which to write data
 * @nr_segs: size of iov vector
979
 * @pos: byte offset in file where writing starts
L
Linus Torvalds 已提交
980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995
 *
 * We use this function for direct writes instead of calling
 * generic_file_aio_write() in order to avoid taking the inode
 * semaphore and updating the i_size.  The NFS server will set
 * the new i_size and this client must read the updated size
 * back into its cache.  We let the server do generic write
 * parameter checking and report problems.
 *
 * We eliminate local atime updates, see direct read above.
 *
 * We avoid unnecessary page cache invalidations for normal cached
 * readers of this file.
 *
 * Note that O_APPEND is not supported for NFS direct writes, as there
 * is no atomic O_APPEND write facility in the NFS protocol.
 */
996 997
ssize_t nfs_file_direct_write(struct kiocb *iocb, const struct iovec *iov,
				unsigned long nr_segs, loff_t pos)
L
Linus Torvalds 已提交
998
{
C
Chuck Lever 已提交
999
	ssize_t retval = -EINVAL;
L
Linus Torvalds 已提交
1000 1001
	struct file *file = iocb->ki_filp;
	struct address_space *mapping = file->f_mapping;
1002
	size_t count;
L
Linus Torvalds 已提交
1003

1004 1005 1006
	count = iov_length(iov, nr_segs);
	nfs_add_stats(mapping->host, NFSIOS_DIRECTWRITTENBYTES, count);

C
Chuck Lever 已提交
1007
	dfprintk(FILE, "NFS: direct write(%s/%s, %zd@%Ld)\n",
1008 1009
		file->f_path.dentry->d_parent->d_name.name,
		file->f_path.dentry->d_name.name,
1010
		count, (long long) pos);
1011

1012 1013
	retval = generic_write_checks(file, &pos, &count, 0);
	if (retval)
L
Linus Torvalds 已提交
1014
		goto out;
1015 1016 1017

	retval = -EINVAL;
	if ((ssize_t) count < 0)
L
Linus Torvalds 已提交
1018 1019 1020 1021
		goto out;
	retval = 0;
	if (!count)
		goto out;
1022

T
Trond Myklebust 已提交
1023 1024 1025
	retval = nfs_sync_mapping(mapping);
	if (retval)
		goto out;
L
Linus Torvalds 已提交
1026

1027 1028
	task_io_account_write(count);

1029
	retval = nfs_direct_write(iocb, iov, nr_segs, pos, count);
1030

L
Linus Torvalds 已提交
1031
	if (retval > 0)
1032
		iocb->ki_pos = pos + retval;
L
Linus Torvalds 已提交
1033 1034 1035 1036 1037

out:
	return retval;
}

1038 1039 1040 1041
/**
 * nfs_init_directcache - create a slab cache for nfs_direct_req structures
 *
 */
D
David Howells 已提交
1042
int __init nfs_init_directcache(void)
L
Linus Torvalds 已提交
1043 1044 1045
{
	nfs_direct_cachep = kmem_cache_create("nfs_direct_cache",
						sizeof(struct nfs_direct_req),
1046 1047
						0, (SLAB_RECLAIM_ACCOUNT|
							SLAB_MEM_SPREAD),
1048
						NULL);
L
Linus Torvalds 已提交
1049 1050 1051 1052 1053 1054
	if (nfs_direct_cachep == NULL)
		return -ENOMEM;

	return 0;
}

1055
/**
D
David Howells 已提交
1056
 * nfs_destroy_directcache - destroy the slab cache for nfs_direct_req structures
1057 1058
 *
 */
1059
void nfs_destroy_directcache(void)
L
Linus Torvalds 已提交
1060
{
1061
	kmem_cache_destroy(nfs_direct_cachep);
L
Linus Torvalds 已提交
1062
}