aio.c 35.4 KB
Newer Older
Linus Torvalds's avatar
Linus Torvalds committed
1
2
3
4
5
6
7
8
9
10
/*
 *	An async IO implementation for Linux
 *	Written by Benjamin LaHaise <bcrl@kvack.org>
 *
 *	Implements an efficient asynchronous io interface.
 *
 *	Copyright 2000, 2001, 2002 Red Hat, Inc.  All Rights Reserved.
 *
 *	See ../COPYING for licensing terms.
 */
Kent Overstreet's avatar
Kent Overstreet committed
11
12
#define pr_fmt(fmt) "%s: " fmt, __func__

Linus Torvalds's avatar
Linus Torvalds committed
13
14
15
16
17
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/errno.h>
#include <linux/time.h>
#include <linux/aio_abi.h>
18
#include <linux/export.h>
Linus Torvalds's avatar
Linus Torvalds committed
19
#include <linux/syscalls.h>
20
#include <linux/backing-dev.h>
21
#include <linux/uio.h>
Linus Torvalds's avatar
Linus Torvalds committed
22
23
24
25
26
27

#include <linux/sched.h>
#include <linux/fs.h>
#include <linux/file.h>
#include <linux/mm.h>
#include <linux/mman.h>
28
#include <linux/mmu_context.h>
Linus Torvalds's avatar
Linus Torvalds committed
29
30
31
32
33
34
#include <linux/slab.h>
#include <linux/timer.h>
#include <linux/aio.h>
#include <linux/highmem.h>
#include <linux/workqueue.h>
#include <linux/security.h>
35
#include <linux/eventfd.h>
Jeff Moyer's avatar
Jeff Moyer committed
36
#include <linux/blkdev.h>
37
#include <linux/compat.h>
Linus Torvalds's avatar
Linus Torvalds committed
38
39
40
41

#include <asm/kmap_types.h>
#include <asm/uaccess.h>

42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#define AIO_RING_MAGIC			0xa10a10a1
#define AIO_RING_COMPAT_FEATURES	1
#define AIO_RING_INCOMPAT_FEATURES	0
struct aio_ring {
	unsigned	id;	/* kernel internal index number */
	unsigned	nr;	/* number of io_events */
	unsigned	head;
	unsigned	tail;

	unsigned	magic;
	unsigned	compat_features;
	unsigned	incompat_features;
	unsigned	header_length;	/* size of aio_ring */


	struct io_event		io_events[0];
}; /* 128 bytes + ring size */

#define AIO_RING_PAGES	8
struct aio_ring_info {
	unsigned long		mmap_base;
	unsigned long		mmap_size;

	struct page		**ring_pages;
66
	struct mutex		ring_lock;
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
	long			nr_pages;

	unsigned		nr, tail;

	struct page		*internal_pages[AIO_RING_PAGES];
};

static inline unsigned aio_ring_avail(struct aio_ring_info *info,
					struct aio_ring *ring)
{
	return (ring->head + info->nr - 1 - ring->tail) % info->nr;
}

struct kioctx {
	atomic_t		users;
Kent Overstreet's avatar
Kent Overstreet committed
82
	atomic_t		dead;
83
84
85
86
87
88
89
90
91

	/* This needs improving */
	unsigned long		user_id;
	struct hlist_node	list;

	wait_queue_head_t	wait;

	spinlock_t		ctx_lock;

92
	atomic_t		reqs_active;
93
94
95
96
97
98
99
100
	struct list_head	active_reqs;	/* used for cancellation */

	/* sys_io_setup currently limits this to an unsigned int */
	unsigned		max_reqs;

	struct aio_ring_info	ring_info;

	struct rcu_head		rcu_head;
Kent Overstreet's avatar
Kent Overstreet committed
101
	struct work_struct	rcu_work;
102
103
};

Linus Torvalds's avatar
Linus Torvalds committed
104
/*------ sysctl variables----*/
105
106
107
static DEFINE_SPINLOCK(aio_nr_lock);
unsigned long aio_nr;		/* current system wide number of aio requests */
unsigned long aio_max_nr = 0x10000; /* system wide maximum number of aio requests */
Linus Torvalds's avatar
Linus Torvalds committed
108
109
/*----end sysctl variables---*/

110
111
static struct kmem_cache	*kiocb_cachep;
static struct kmem_cache	*kioctx_cachep;
Linus Torvalds's avatar
Linus Torvalds committed
112
113
114
115
116
117
118

/* aio_setup
 *	Creates the slab caches used by the aio routines, panic on
 *	failure as this is done early during the boot sequence.
 */
static int __init aio_setup(void)
{
119
120
	kiocb_cachep = KMEM_CACHE(kiocb, SLAB_HWCACHE_ALIGN|SLAB_PANIC);
	kioctx_cachep = KMEM_CACHE(kioctx,SLAB_HWCACHE_ALIGN|SLAB_PANIC);
Linus Torvalds's avatar
Linus Torvalds committed
121

Kent Overstreet's avatar
Kent Overstreet committed
122
	pr_debug("sizeof(struct page) = %zu\n", sizeof(struct page));
Linus Torvalds's avatar
Linus Torvalds committed
123
124
125

	return 0;
}
126
__initcall(aio_setup);
Linus Torvalds's avatar
Linus Torvalds committed
127
128
129
130
131
132
133
134
135

static void aio_free_ring(struct kioctx *ctx)
{
	struct aio_ring_info *info = &ctx->ring_info;
	long i;

	for (i=0; i<info->nr_pages; i++)
		put_page(info->ring_pages[i]);

136
	if (info->mmap_size) {
Al Viro's avatar
Al Viro committed
137
		vm_munmap(info->mmap_base, info->mmap_size);
138
	}
Linus Torvalds's avatar
Linus Torvalds committed
139
140
141
142
143
144
145
146
147
148
149
150

	if (info->ring_pages && info->ring_pages != info->internal_pages)
		kfree(info->ring_pages);
	info->ring_pages = NULL;
	info->nr = 0;
}

static int aio_setup_ring(struct kioctx *ctx)
{
	struct aio_ring *ring;
	struct aio_ring_info *info = &ctx->ring_info;
	unsigned nr_events = ctx->max_reqs;
Zach Brown's avatar
Zach Brown committed
151
	struct mm_struct *mm = current->mm;
152
	unsigned long size, populate;
Linus Torvalds's avatar
Linus Torvalds committed
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
	int nr_pages;

	/* Compensate for the ring buffer's head/tail overlap entry */
	nr_events += 2;	/* 1 is required, 2 for good luck */

	size = sizeof(struct aio_ring);
	size += sizeof(struct io_event) * nr_events;
	nr_pages = (size + PAGE_SIZE-1) >> PAGE_SHIFT;

	if (nr_pages < 0)
		return -EINVAL;

	nr_events = (PAGE_SIZE * nr_pages - sizeof(struct aio_ring)) / sizeof(struct io_event);

	info->nr = 0;
	info->ring_pages = info->internal_pages;
	if (nr_pages > AIO_RING_PAGES) {
170
		info->ring_pages = kcalloc(nr_pages, sizeof(struct page *), GFP_KERNEL);
Linus Torvalds's avatar
Linus Torvalds committed
171
172
173
174
175
		if (!info->ring_pages)
			return -ENOMEM;
	}

	info->mmap_size = nr_pages * PAGE_SIZE;
Kent Overstreet's avatar
Kent Overstreet committed
176
	pr_debug("attempting mmap of %lu bytes\n", info->mmap_size);
Zach Brown's avatar
Zach Brown committed
177
	down_write(&mm->mmap_sem);
178
179
	info->mmap_base = do_mmap_pgoff(NULL, 0, info->mmap_size, 
					PROT_READ|PROT_WRITE,
180
181
					MAP_ANONYMOUS|MAP_PRIVATE, 0,
					&populate);
Linus Torvalds's avatar
Linus Torvalds committed
182
	if (IS_ERR((void *)info->mmap_base)) {
Zach Brown's avatar
Zach Brown committed
183
		up_write(&mm->mmap_sem);
Linus Torvalds's avatar
Linus Torvalds committed
184
185
186
187
188
		info->mmap_size = 0;
		aio_free_ring(ctx);
		return -EAGAIN;
	}

Kent Overstreet's avatar
Kent Overstreet committed
189
	pr_debug("mmap address: 0x%08lx\n", info->mmap_base);
Zach Brown's avatar
Zach Brown committed
190
	info->nr_pages = get_user_pages(current, mm, info->mmap_base, nr_pages,
Linus Torvalds's avatar
Linus Torvalds committed
191
					1, 0, info->ring_pages, NULL);
Zach Brown's avatar
Zach Brown committed
192
	up_write(&mm->mmap_sem);
Linus Torvalds's avatar
Linus Torvalds committed
193
194
195
196
197

	if (unlikely(info->nr_pages != nr_pages)) {
		aio_free_ring(ctx);
		return -EAGAIN;
	}
198
	if (populate)
199
		mm_populate(info->mmap_base, populate);
Linus Torvalds's avatar
Linus Torvalds committed
200
201
202
203
204

	ctx->user_id = info->mmap_base;

	info->nr = nr_events;		/* trusted copy */

205
	ring = kmap_atomic(info->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
206
207
208
209
210
211
212
	ring->nr = nr_events;	/* user copy */
	ring->id = ctx->user_id;
	ring->head = ring->tail = 0;
	ring->magic = AIO_RING_MAGIC;
	ring->compat_features = AIO_RING_COMPAT_FEATURES;
	ring->incompat_features = AIO_RING_INCOMPAT_FEATURES;
	ring->header_length = sizeof(struct aio_ring);
213
	kunmap_atomic(ring);
Kent Overstreet's avatar
Kent Overstreet committed
214
	flush_dcache_page(info->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
215
216
217
218
219
220
221
222

	return 0;
}

#define AIO_EVENTS_PER_PAGE	(PAGE_SIZE / sizeof(struct io_event))
#define AIO_EVENTS_FIRST_PAGE	((PAGE_SIZE - sizeof(struct aio_ring)) / sizeof(struct io_event))
#define AIO_EVENTS_OFFSET	(AIO_EVENTS_PER_PAGE - AIO_EVENTS_FIRST_PAGE)

Kent Overstreet's avatar
Kent Overstreet committed
223
224
225
226
227
228
229
230
231
static int kiocb_cancel(struct kioctx *ctx, struct kiocb *kiocb,
			struct io_event *res)
{
	int (*cancel)(struct kiocb *, struct io_event *);
	int ret = -EINVAL;

	cancel = kiocb->ki_cancel;
	kiocbSetCancelled(kiocb);
	if (cancel) {
232
		atomic_inc(&kiocb->ki_users);
Kent Overstreet's avatar
Kent Overstreet committed
233
234
235
236
237
238
239
240
241
242
243
244
245
		spin_unlock_irq(&ctx->ctx_lock);

		memset(res, 0, sizeof(*res));
		res->obj = (u64)(unsigned long)kiocb->ki_obj.user;
		res->data = kiocb->ki_user_data;
		ret = cancel(kiocb, res);

		spin_lock_irq(&ctx->ctx_lock);
	}

	return ret;
}

Kent Overstreet's avatar
Kent Overstreet committed
246
247
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
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
static void free_ioctx_rcu(struct rcu_head *head)
{
	struct kioctx *ctx = container_of(head, struct kioctx, rcu_head);
	kmem_cache_free(kioctx_cachep, ctx);
}

/*
 * When this function runs, the kioctx has been removed from the "hash table"
 * and ctx->users has dropped to 0, so we know no more kiocbs can be submitted -
 * now it's safe to cancel any that need to be.
 */
static void free_ioctx(struct kioctx *ctx)
{
	struct io_event res;
	struct kiocb *req;

	spin_lock_irq(&ctx->ctx_lock);

	while (!list_empty(&ctx->active_reqs)) {
		req = list_first_entry(&ctx->active_reqs,
				       struct kiocb, ki_list);

		list_del_init(&req->ki_list);
		kiocb_cancel(ctx, req, &res);
	}

	spin_unlock_irq(&ctx->ctx_lock);

	wait_event(ctx->wait, !atomic_read(&ctx->reqs_active));

	aio_free_ring(ctx);

	spin_lock(&aio_nr_lock);
	BUG_ON(aio_nr - ctx->max_reqs > aio_nr);
	aio_nr -= ctx->max_reqs;
	spin_unlock(&aio_nr_lock);

	pr_debug("freeing %p\n", ctx);

	/*
	 * Here the call_rcu() is between the wait_event() for reqs_active to
	 * hit 0, and freeing the ioctx.
	 *
	 * aio_complete() decrements reqs_active, but it has to touch the ioctx
	 * after to issue a wakeup so we use rcu.
	 */
	call_rcu(&ctx->rcu_head, free_ioctx_rcu);
}

static void put_ioctx(struct kioctx *ctx)
{
	if (unlikely(atomic_dec_and_test(&ctx->users)))
		free_ioctx(ctx);
}

Linus Torvalds's avatar
Linus Torvalds committed
301
302
303
304
305
/* ioctx_alloc
 *	Allocates and initializes an ioctx.  Returns an ERR_PTR if it failed.
 */
static struct kioctx *ioctx_alloc(unsigned nr_events)
{
Zach Brown's avatar
Zach Brown committed
306
	struct mm_struct *mm = current->mm;
Linus Torvalds's avatar
Linus Torvalds committed
307
	struct kioctx *ctx;
308
	int err = -ENOMEM;
Linus Torvalds's avatar
Linus Torvalds committed
309
310
311
312
313
314
315
316

	/* Prevent overflows */
	if ((nr_events > (0x10000000U / sizeof(struct io_event))) ||
	    (nr_events > (0x10000000U / sizeof(struct kiocb)))) {
		pr_debug("ENOMEM: nr_events too high\n");
		return ERR_PTR(-EINVAL);
	}

317
	if (!nr_events || (unsigned long)nr_events > aio_max_nr)
Linus Torvalds's avatar
Linus Torvalds committed
318
319
		return ERR_PTR(-EAGAIN);

320
	ctx = kmem_cache_zalloc(kioctx_cachep, GFP_KERNEL);
Linus Torvalds's avatar
Linus Torvalds committed
321
322
323
324
325
	if (!ctx)
		return ERR_PTR(-ENOMEM);

	ctx->max_reqs = nr_events;

Al Viro's avatar
Al Viro committed
326
	atomic_set(&ctx->users, 2);
Kent Overstreet's avatar
Kent Overstreet committed
327
	atomic_set(&ctx->dead, 0);
Linus Torvalds's avatar
Linus Torvalds committed
328
	spin_lock_init(&ctx->ctx_lock);
329
	mutex_init(&ctx->ring_info.ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
330
331
332
333
334
335
336
337
	init_waitqueue_head(&ctx->wait);

	INIT_LIST_HEAD(&ctx->active_reqs);

	if (aio_setup_ring(ctx) < 0)
		goto out_freectx;

	/* limit the number of system wide aios */
338
	spin_lock(&aio_nr_lock);
339
340
	if (aio_nr + nr_events > aio_max_nr ||
	    aio_nr + nr_events < aio_nr) {
341
		spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
342
		goto out_cleanup;
343
344
	}
	aio_nr += ctx->max_reqs;
345
	spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
346

Jeff Moyer's avatar
Jeff Moyer committed
347
	/* now link into global list. */
348
349
350
	spin_lock(&mm->ioctx_lock);
	hlist_add_head_rcu(&ctx->list, &mm->ioctx_list);
	spin_unlock(&mm->ioctx_lock);
Linus Torvalds's avatar
Linus Torvalds committed
351

Kent Overstreet's avatar
Kent Overstreet committed
352
	pr_debug("allocated ioctx %p[%ld]: mm=%p mask=0x%x\n",
Zach Brown's avatar
Zach Brown committed
353
		ctx, ctx->user_id, mm, ctx->ring_info.nr);
Linus Torvalds's avatar
Linus Torvalds committed
354
355
356
	return ctx;

out_cleanup:
357
358
	err = -EAGAIN;
	aio_free_ring(ctx);
Linus Torvalds's avatar
Linus Torvalds committed
359
360
out_freectx:
	kmem_cache_free(kioctx_cachep, ctx);
Kent Overstreet's avatar
Kent Overstreet committed
361
	pr_debug("error allocating ioctx %d\n", err);
362
	return ERR_PTR(err);
Linus Torvalds's avatar
Linus Torvalds committed
363
364
}

Kent Overstreet's avatar
Kent Overstreet committed
365
static void kill_ioctx_work(struct work_struct *work)
Linus Torvalds's avatar
Linus Torvalds committed
366
{
Kent Overstreet's avatar
Kent Overstreet committed
367
	struct kioctx *ctx = container_of(work, struct kioctx, rcu_work);
368

Kent Overstreet's avatar
Kent Overstreet committed
369
370
371
	wake_up_all(&ctx->wait);
	put_ioctx(ctx);
}
Kent Overstreet's avatar
Kent Overstreet committed
372

Kent Overstreet's avatar
Kent Overstreet committed
373
374
375
static void kill_ioctx_rcu(struct rcu_head *head)
{
	struct kioctx *ctx = container_of(head, struct kioctx, rcu_head);
Linus Torvalds's avatar
Linus Torvalds committed
376

Kent Overstreet's avatar
Kent Overstreet committed
377
378
379
	INIT_WORK(&ctx->rcu_work, kill_ioctx_work);
	schedule_work(&ctx->rcu_work);
}
Linus Torvalds's avatar
Linus Torvalds committed
380

Kent Overstreet's avatar
Kent Overstreet committed
381
382
383
384
385
386
387
388
389
390
391
/* kill_ioctx
 *	Cancels all outstanding aio requests on an aio context.  Used
 *	when the processes owning a context have all exited to encourage
 *	the rapid destruction of the kioctx.
 */
static void kill_ioctx(struct kioctx *ctx)
{
	if (!atomic_xchg(&ctx->dead, 1)) {
		hlist_del_rcu(&ctx->list);
		/* Between hlist_del_rcu() and dropping the initial ref */
		synchronize_rcu();
392

Kent Overstreet's avatar
Kent Overstreet committed
393
394
395
396
397
398
399
400
401
		/*
		 * We can't punt to workqueue here because put_ioctx() ->
		 * free_ioctx() will unmap the ringbuffer, and that has to be
		 * done in the original process's context. kill_ioctx_rcu/work()
		 * exist for exit_aio(), as in that path free_ioctx() won't do
		 * the unmap.
		 */
		kill_ioctx_work(&ctx->rcu_work);
	}
Linus Torvalds's avatar
Linus Torvalds committed
402
403
404
405
406
}

/* wait_on_sync_kiocb:
 *	Waits on the given sync kiocb to complete.
 */
407
ssize_t wait_on_sync_kiocb(struct kiocb *iocb)
Linus Torvalds's avatar
Linus Torvalds committed
408
{
409
	while (atomic_read(&iocb->ki_users)) {
Linus Torvalds's avatar
Linus Torvalds committed
410
		set_current_state(TASK_UNINTERRUPTIBLE);
411
		if (!atomic_read(&iocb->ki_users))
Linus Torvalds's avatar
Linus Torvalds committed
412
			break;
413
		io_schedule();
Linus Torvalds's avatar
Linus Torvalds committed
414
415
416
417
	}
	__set_current_state(TASK_RUNNING);
	return iocb->ki_user_data;
}
418
EXPORT_SYMBOL(wait_on_sync_kiocb);
Linus Torvalds's avatar
Linus Torvalds committed
419

Kent Overstreet's avatar
Kent Overstreet committed
420
421
422
423
424
425
426
/*
 * exit_aio: called when the last user of mm goes away.  At this point, there is
 * no way for any new requests to be submited or any of the io_* syscalls to be
 * called on the context.
 *
 * There may be outstanding kiocbs, but free_ioctx() will explicitly wait on
 * them.
Linus Torvalds's avatar
Linus Torvalds committed
427
 */
428
void exit_aio(struct mm_struct *mm)
Linus Torvalds's avatar
Linus Torvalds committed
429
{
430
	struct kioctx *ctx;
Kent Overstreet's avatar
Kent Overstreet committed
431
	struct hlist_node *n;
432

Kent Overstreet's avatar
Kent Overstreet committed
433
	hlist_for_each_entry_safe(ctx, n, &mm->ioctx_list, list) {
Linus Torvalds's avatar
Linus Torvalds committed
434
435
436
		if (1 != atomic_read(&ctx->users))
			printk(KERN_DEBUG
				"exit_aio:ioctx still alive: %d %d %d\n",
Kent Overstreet's avatar
Kent Overstreet committed
437
438
				atomic_read(&ctx->users),
				atomic_read(&ctx->dead),
439
				atomic_read(&ctx->reqs_active));
440
441
442
443
444
445
446
447
448
		/*
		 * We don't need to bother with munmap() here -
		 * exit_mmap(mm) is coming and it'll unmap everything.
		 * Since aio_free_ring() uses non-zero ->mmap_size
		 * as indicator that it needs to unmap the area,
		 * just set it to 0; aio_free_ring() is the only
		 * place that uses ->mmap_size, so it's safe.
		 */
		ctx->ring_info.mmap_size = 0;
Kent Overstreet's avatar
Kent Overstreet committed
449
450
451
452
453

		if (!atomic_xchg(&ctx->dead, 1)) {
			hlist_del_rcu(&ctx->list);
			call_rcu(&ctx->rcu_head, kill_ioctx_rcu);
		}
Linus Torvalds's avatar
Linus Torvalds committed
454
455
456
457
	}
}

/* aio_get_req
458
 *	Allocate a slot for an aio request.  Increments the ki_users count
Linus Torvalds's avatar
Linus Torvalds committed
459
460
461
 * of the kioctx so that the kioctx stays around until all requests are
 * complete.  Returns NULL if no requests are free.
 *
462
 * Returns with kiocb->ki_users set to 2.  The io submit code path holds
Linus Torvalds's avatar
Linus Torvalds committed
463
464
465
466
 * an extra reference while submitting the i/o.
 * This prevents races between the aio code path referencing the
 * req (after submitting it) and aio_complete() freeing the req.
 */
467
static struct kiocb *__aio_get_req(struct kioctx *ctx)
Linus Torvalds's avatar
Linus Torvalds committed
468
469
470
471
472
473
474
{
	struct kiocb *req = NULL;

	req = kmem_cache_alloc(kiocb_cachep, GFP_KERNEL);
	if (unlikely(!req))
		return NULL;

Zach Brown's avatar
Zach Brown committed
475
	req->ki_flags = 0;
476
	atomic_set(&req->ki_users, 2);
Linus Torvalds's avatar
Linus Torvalds committed
477
478
479
480
481
482
	req->ki_key = 0;
	req->ki_ctx = ctx;
	req->ki_cancel = NULL;
	req->ki_retry = NULL;
	req->ki_dtor = NULL;
	req->private = NULL;
483
	req->ki_iovec = NULL;
484
	req->ki_eventfd = NULL;
Linus Torvalds's avatar
Linus Torvalds committed
485

Jeff Moyer's avatar
Jeff Moyer committed
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
	return req;
}

/*
 * struct kiocb's are allocated in batches to reduce the number of
 * times the ctx lock is acquired and released.
 */
#define KIOCB_BATCH_SIZE	32L
struct kiocb_batch {
	struct list_head head;
	long count; /* number of requests left to allocate */
};

static void kiocb_batch_init(struct kiocb_batch *batch, long total)
{
	INIT_LIST_HEAD(&batch->head);
	batch->count = total;
}

505
static void kiocb_batch_free(struct kioctx *ctx, struct kiocb_batch *batch)
Jeff Moyer's avatar
Jeff Moyer committed
506
507
508
{
	struct kiocb *req, *n;

509
510
511
512
	if (list_empty(&batch->head))
		return;

	spin_lock_irq(&ctx->ctx_lock);
Jeff Moyer's avatar
Jeff Moyer committed
513
514
	list_for_each_entry_safe(req, n, &batch->head, ki_batch) {
		list_del(&req->ki_batch);
515
		list_del(&req->ki_list);
Jeff Moyer's avatar
Jeff Moyer committed
516
		kmem_cache_free(kiocb_cachep, req);
517
		atomic_dec(&ctx->reqs_active);
Jeff Moyer's avatar
Jeff Moyer committed
518
	}
519
	spin_unlock_irq(&ctx->ctx_lock);
Jeff Moyer's avatar
Jeff Moyer committed
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
}

/*
 * Allocate a batch of kiocbs.  This avoids taking and dropping the
 * context lock a lot during setup.
 */
static int kiocb_batch_refill(struct kioctx *ctx, struct kiocb_batch *batch)
{
	unsigned short allocated, to_alloc;
	long avail;
	struct kiocb *req, *n;
	struct aio_ring *ring;

	to_alloc = min(batch->count, KIOCB_BATCH_SIZE);
	for (allocated = 0; allocated < to_alloc; allocated++) {
		req = __aio_get_req(ctx);
		if (!req)
			/* allocation failed, go with what we've got */
			break;
		list_add(&req->ki_batch, &batch->head);
	}

	if (allocated == 0)
		goto out;

Linus Torvalds's avatar
Linus Torvalds committed
545
	spin_lock_irq(&ctx->ctx_lock);
Jeff Moyer's avatar
Jeff Moyer committed
546
547
	ring = kmap_atomic(ctx->ring_info.ring_pages[0]);

548
549
	avail = aio_ring_avail(&ctx->ring_info, ring) -
				atomic_read(&ctx->reqs_active);
Jeff Moyer's avatar
Jeff Moyer committed
550
551
552
553
554
555
556
557
558
559
560
561
562
	BUG_ON(avail < 0);
	if (avail < allocated) {
		/* Trim back the number of requests. */
		list_for_each_entry_safe(req, n, &batch->head, ki_batch) {
			list_del(&req->ki_batch);
			kmem_cache_free(kiocb_cachep, req);
			if (--allocated <= avail)
				break;
		}
	}

	batch->count -= allocated;
	list_for_each_entry(req, &batch->head, ki_batch) {
Linus Torvalds's avatar
Linus Torvalds committed
563
		list_add(&req->ki_list, &ctx->active_reqs);
564
		atomic_inc(&ctx->reqs_active);
Linus Torvalds's avatar
Linus Torvalds committed
565
566
	}

Jeff Moyer's avatar
Jeff Moyer committed
567
568
	kunmap_atomic(ring);
	spin_unlock_irq(&ctx->ctx_lock);
Linus Torvalds's avatar
Linus Torvalds committed
569

Jeff Moyer's avatar
Jeff Moyer committed
570
571
out:
	return allocated;
Linus Torvalds's avatar
Linus Torvalds committed
572
573
}

Jeff Moyer's avatar
Jeff Moyer committed
574
575
static inline struct kiocb *aio_get_req(struct kioctx *ctx,
					struct kiocb_batch *batch)
Linus Torvalds's avatar
Linus Torvalds committed
576
577
{
	struct kiocb *req;
Jeff Moyer's avatar
Jeff Moyer committed
578
579
580
581
582
583

	if (list_empty(&batch->head))
		if (kiocb_batch_refill(ctx, batch) == 0)
			return NULL;
	req = list_first_entry(&batch->head, struct kiocb, ki_batch);
	list_del(&req->ki_batch);
Linus Torvalds's avatar
Linus Torvalds committed
584
585
586
	return req;
}

587
static void kiocb_free(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
588
{
589
590
	if (req->ki_filp)
		fput(req->ki_filp);
591
592
	if (req->ki_eventfd != NULL)
		eventfd_ctx_put(req->ki_eventfd);
Linus Torvalds's avatar
Linus Torvalds committed
593
594
	if (req->ki_dtor)
		req->ki_dtor(req);
595
596
	if (req->ki_iovec != &req->ki_inline_vec)
		kfree(req->ki_iovec);
Linus Torvalds's avatar
Linus Torvalds committed
597
598
599
	kmem_cache_free(kiocb_cachep, req);
}

600
void aio_put_req(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
601
{
602
603
	if (atomic_dec_and_test(&req->ki_users))
		kiocb_free(req);
Linus Torvalds's avatar
Linus Torvalds committed
604
}
605
EXPORT_SYMBOL(aio_put_req);
Linus Torvalds's avatar
Linus Torvalds committed
606

607
static struct kioctx *lookup_ioctx(unsigned long ctx_id)
Linus Torvalds's avatar
Linus Torvalds committed
608
{
609
	struct mm_struct *mm = current->mm;
610
	struct kioctx *ctx, *ret = NULL;
Linus Torvalds's avatar
Linus Torvalds committed
611

612
613
	rcu_read_lock();

614
	hlist_for_each_entry_rcu(ctx, &mm->ioctx_list, list) {
Kent Overstreet's avatar
Kent Overstreet committed
615
616
		if (ctx->user_id == ctx_id) {
			atomic_inc(&ctx->users);
617
			ret = ctx;
Linus Torvalds's avatar
Linus Torvalds committed
618
619
			break;
		}
620
	}
Linus Torvalds's avatar
Linus Torvalds committed
621

622
	rcu_read_unlock();
623
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
624
625
626
627
628
}

/* aio_complete
 *	Called when the io request on the given iocb is complete.
 */
629
void aio_complete(struct kiocb *iocb, long res, long res2)
Linus Torvalds's avatar
Linus Torvalds committed
630
631
632
633
{
	struct kioctx	*ctx = iocb->ki_ctx;
	struct aio_ring_info	*info;
	struct aio_ring	*ring;
Kent Overstreet's avatar
Kent Overstreet committed
634
	struct io_event	*ev_page, *event;
Linus Torvalds's avatar
Linus Torvalds committed
635
	unsigned long	flags;
Kent Overstreet's avatar
Kent Overstreet committed
636
	unsigned tail, pos;
Linus Torvalds's avatar
Linus Torvalds committed
637

638
639
640
641
642
643
	/*
	 * Special case handling for sync iocbs:
	 *  - events go directly into the iocb for fast handling
	 *  - the sync task with the iocb in its stack holds the single iocb
	 *    ref, no other paths have a way to get another ref
	 *  - the sync task helpfully left a reference to itself in the iocb
Linus Torvalds's avatar
Linus Torvalds committed
644
645
	 */
	if (is_sync_kiocb(iocb)) {
646
		BUG_ON(atomic_read(&iocb->ki_users) != 1);
Linus Torvalds's avatar
Linus Torvalds committed
647
		iocb->ki_user_data = res;
648
		atomic_set(&iocb->ki_users, 0);
Linus Torvalds's avatar
Linus Torvalds committed
649
		wake_up_process(iocb->ki_obj.tsk);
650
		return;
Linus Torvalds's avatar
Linus Torvalds committed
651
652
653
654
	}

	info = &ctx->ring_info;

Kent Overstreet's avatar
Kent Overstreet committed
655
656
657
658
659
660
661
	/*
	 * Add a completion event to the ring buffer. Must be done holding
	 * ctx->ctx_lock to prevent other code from messing with the tail
	 * pointer since we might be called from irq context.
	 *
	 * Take rcu_read_lock() in case the kioctx is being destroyed, as we
	 * need to issue a wakeup after decrementing reqs_active.
Linus Torvalds's avatar
Linus Torvalds committed
662
	 */
Kent Overstreet's avatar
Kent Overstreet committed
663
	rcu_read_lock();
Linus Torvalds's avatar
Linus Torvalds committed
664
665
	spin_lock_irqsave(&ctx->ctx_lock, flags);

666
667
	list_del(&iocb->ki_list); /* remove from active_reqs */

Linus Torvalds's avatar
Linus Torvalds committed
668
669
670
671
672
673
674
675
	/*
	 * cancelled requests don't get events, userland was given one
	 * when the event got cancelled.
	 */
	if (kiocbIsCancelled(iocb))
		goto put_rq;

	tail = info->tail;
Kent Overstreet's avatar
Kent Overstreet committed
676
677
	pos = tail + AIO_EVENTS_OFFSET;

678
679
	if (++tail >= info->nr)
		tail = 0;
Linus Torvalds's avatar
Linus Torvalds committed
680

Kent Overstreet's avatar
Kent Overstreet committed
681
682
683
	ev_page = kmap_atomic(info->ring_pages[pos / AIO_EVENTS_PER_PAGE]);
	event = ev_page + pos % AIO_EVENTS_PER_PAGE;

Linus Torvalds's avatar
Linus Torvalds committed
684
685
686
687
688
	event->obj = (u64)(unsigned long)iocb->ki_obj.user;
	event->data = iocb->ki_user_data;
	event->res = res;
	event->res2 = res2;

Kent Overstreet's avatar
Kent Overstreet committed
689
690
691
692
	kunmap_atomic(ev_page);
	flush_dcache_page(info->ring_pages[pos / AIO_EVENTS_PER_PAGE]);

	pr_debug("%p[%u]: %p: %p %Lx %lx %lx\n",
Kent Overstreet's avatar
Kent Overstreet committed
693
694
		 ctx, tail, iocb, iocb->ki_obj.user, iocb->ki_user_data,
		 res, res2);
Linus Torvalds's avatar
Linus Torvalds committed
695
696
697
698
699
700
701
702

	/* after flagging the request as done, we
	 * must never even look at it again
	 */
	smp_wmb();	/* make event visible before updating tail */

	info->tail = tail;

Kent Overstreet's avatar
Kent Overstreet committed
703
704
	ring = kmap_atomic(info->ring_pages[0]);
	ring->tail = tail;
705
	kunmap_atomic(ring);
Kent Overstreet's avatar
Kent Overstreet committed
706
	flush_dcache_page(info->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
707

Kent Overstreet's avatar
Kent Overstreet committed
708
	pr_debug("added to ring %p at [%u]\n", iocb, tail);
Davide Libenzi's avatar
Davide Libenzi committed
709
710
711
712
713
714

	/*
	 * Check if the user asked us to deliver the result through an
	 * eventfd. The eventfd_signal() function is safe to be called
	 * from IRQ context.
	 */
715
	if (iocb->ki_eventfd != NULL)
Davide Libenzi's avatar
Davide Libenzi committed
716
717
		eventfd_signal(iocb->ki_eventfd, 1);

Linus Torvalds's avatar
Linus Torvalds committed
718
719
put_rq:
	/* everything turned out well, dispose of the aiocb. */
720
721
	aio_put_req(iocb);
	atomic_dec(&ctx->reqs_active);
Linus Torvalds's avatar
Linus Torvalds committed
722

723
724
725
726
727
728
729
730
	/*
	 * We have to order our ring_info tail store above and test
	 * of the wait list below outside the wait lock.  This is
	 * like in wake_up_bit() where clearing a bit has to be
	 * ordered with the unlocked test.
	 */
	smp_mb();

Linus Torvalds's avatar
Linus Torvalds committed
731
732
733
	if (waitqueue_active(&ctx->wait))
		wake_up(&ctx->wait);

734
	spin_unlock_irqrestore(&ctx->ctx_lock, flags);
Kent Overstreet's avatar
Kent Overstreet committed
735
	rcu_read_unlock();
Linus Torvalds's avatar
Linus Torvalds committed
736
}
737
EXPORT_SYMBOL(aio_complete);
Linus Torvalds's avatar
Linus Torvalds committed
738

739
740
741
/* aio_read_events
 *	Pull an event off of the ioctx's event ring.  Returns the number of
 *	events fetched
Linus Torvalds's avatar
Linus Torvalds committed
742
 */
743
744
static long aio_read_events_ring(struct kioctx *ctx,
				 struct io_event __user *event, long nr)
Linus Torvalds's avatar
Linus Torvalds committed
745
{
746
	struct aio_ring_info *info = &ctx->ring_info;
Linus Torvalds's avatar
Linus Torvalds committed
747
	struct aio_ring *ring;
748
749
750
751
752
	unsigned head, pos;
	long ret = 0;
	int copy_ret;

	mutex_lock(&info->ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
753

754
	ring = kmap_atomic(info->ring_pages[0]);
755
756
757
758
	head = ring->head;
	kunmap_atomic(ring);

	pr_debug("h%u t%u m%u\n", head, info->tail, info->nr);
Linus Torvalds's avatar
Linus Torvalds committed
759

760
	if (head == info->tail)
Linus Torvalds's avatar
Linus Torvalds committed
761
762
		goto out;

763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
	while (ret < nr) {
		long avail;
		struct io_event *ev;
		struct page *page;

		avail = (head <= info->tail ? info->tail : info->nr) - head;
		if (head == info->tail)
			break;

		avail = min(avail, nr - ret);
		avail = min_t(long, avail, AIO_EVENTS_PER_PAGE -
			    ((head + AIO_EVENTS_OFFSET) % AIO_EVENTS_PER_PAGE));

		pos = head + AIO_EVENTS_OFFSET;
		page = info->ring_pages[pos / AIO_EVENTS_PER_PAGE];
		pos %= AIO_EVENTS_PER_PAGE;

		ev = kmap(page);
		copy_ret = copy_to_user(event + ret, ev + pos,
					sizeof(*ev) * avail);
		kunmap(page);

		if (unlikely(copy_ret)) {
			ret = -EFAULT;
			goto out;
		}

		ret += avail;
		head += avail;
		head %= info->nr;
Linus Torvalds's avatar
Linus Torvalds committed
793
794
	}

795
796
	ring = kmap_atomic(info->ring_pages[0]);
	ring->head = head;
797
	kunmap_atomic(ring);
Kent Overstreet's avatar
Kent Overstreet committed
798
	flush_dcache_page(info->ring_pages[0]);
799
800
801
802
803

	pr_debug("%li  h%u t%u\n", ret, head, info->tail);
out:
	mutex_unlock(&info->ring_lock);

Linus Torvalds's avatar
Linus Torvalds committed
804
805
806
	return ret;
}

807
808
static bool aio_read_events(struct kioctx *ctx, long min_nr, long nr,
			    struct io_event __user *event, long *i)
Linus Torvalds's avatar
Linus Torvalds committed
809
{
810
	long ret = aio_read_events_ring(ctx, event + *i, nr - *i);
Linus Torvalds's avatar
Linus Torvalds committed
811

812
813
	if (ret > 0)
		*i += ret;
Linus Torvalds's avatar
Linus Torvalds committed
814

815
816
	if (unlikely(atomic_read(&ctx->dead)))
		ret = -EINVAL;
Linus Torvalds's avatar
Linus Torvalds committed
817

818
819
	if (!*i)
		*i = ret;
Linus Torvalds's avatar
Linus Torvalds committed
820

821
	return ret < 0 || *i >= min_nr;
Linus Torvalds's avatar
Linus Torvalds committed
822
823
}

824
static long read_events(struct kioctx *ctx, long min_nr, long nr,
Linus Torvalds's avatar
Linus Torvalds committed
825
826
827
			struct io_event __user *event,
			struct timespec __user *timeout)
{
828
829
	ktime_t until = { .tv64 = KTIME_MAX };
	long ret = 0;
Linus Torvalds's avatar
Linus Torvalds committed
830
831
832

	if (timeout) {
		struct timespec	ts;
833

Linus Torvalds's avatar
Linus Torvalds committed
834
		if (unlikely(copy_from_user(&ts, timeout, sizeof(ts))))
835
			return -EFAULT;
Linus Torvalds's avatar
Linus Torvalds committed
836

837
		until = timespec_to_ktime(ts);
Linus Torvalds's avatar
Linus Torvalds committed
838
839
	}

840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
	/*
	 * Note that aio_read_events() is being called as the conditional - i.e.
	 * we're calling it after prepare_to_wait() has set task state to
	 * TASK_INTERRUPTIBLE.
	 *
	 * But aio_read_events() can block, and if it blocks it's going to flip
	 * the task state back to TASK_RUNNING.
	 *
	 * This should be ok, provided it doesn't flip the state back to
	 * TASK_RUNNING and return 0 too much - that causes us to spin. That
	 * will only happen if the mutex_lock() call blocks, and we then find
	 * the ringbuffer empty. So in practice we should be ok, but it's
	 * something to be aware of when touching this code.
	 */
	wait_event_interruptible_hrtimeout(ctx->wait,
			aio_read_events(ctx, min_nr, nr, event, &ret), until);
Linus Torvalds's avatar
Linus Torvalds committed
856

857
858
	if (!ret && signal_pending(current))
		ret = -EINTR;
Linus Torvalds's avatar
Linus Torvalds committed
859

860
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
}

/* sys_io_setup:
 *	Create an aio_context capable of receiving at least nr_events.
 *	ctxp must not point to an aio_context that already exists, and
 *	must be initialized to 0 prior to the call.  On successful
 *	creation of the aio_context, *ctxp is filled in with the resulting 
 *	handle.  May fail with -EINVAL if *ctxp is not initialized,
 *	if the specified nr_events exceeds internal limits.  May fail 
 *	with -EAGAIN if the specified nr_events exceeds the user's limit 
 *	of available events.  May fail with -ENOMEM if insufficient kernel
 *	resources are available.  May fail with -EFAULT if an invalid
 *	pointer is passed for ctxp.  Will fail with -ENOSYS if not
 *	implemented.
 */
876
SYSCALL_DEFINE2(io_setup, unsigned, nr_events, aio_context_t __user *, ctxp)
Linus Torvalds's avatar
Linus Torvalds committed
877
878
879
880
881
882
883
884
885
886
{
	struct kioctx *ioctx = NULL;
	unsigned long ctx;
	long ret;

	ret = get_user(ctx, ctxp);
	if (unlikely(ret))
		goto out;

	ret = -EINVAL;
887
888
889
	if (unlikely(ctx || nr_events == 0)) {
		pr_debug("EINVAL: io_setup: ctx %lu nr_events %u\n",
		         ctx, nr_events);
Linus Torvalds's avatar
Linus Torvalds committed
890
891
892
893
894
895
896
		goto out;
	}

	ioctx = ioctx_alloc(nr_events);
	ret = PTR_ERR(ioctx);
	if (!IS_ERR(ioctx)) {
		ret = put_user(ioctx->user_id, ctxp);
897
		if (ret)
Kent Overstreet's avatar
Kent Overstreet committed
898
			kill_ioctx(ioctx);
899
		put_ioctx(ioctx);
Linus Torvalds's avatar
Linus Torvalds committed
900
901
902
903
904
905
906
907
908
	}

out:
	return ret;
}

/* sys_io_destroy:
 *	Destroy the aio_context specified.  May cancel any outstanding 
 *	AIOs and block on completion.  Will fail with -ENOSYS if not
909
 *	implemented.  May fail with -EINVAL if the context pointed to
Linus Torvalds's avatar
Linus Torvalds committed
910
911
 *	is invalid.
 */
912
SYSCALL_DEFINE1(io_destroy, aio_context_t, ctx)
Linus Torvalds's avatar
Linus Torvalds committed
913
914
915
{
	struct kioctx *ioctx = lookup_ioctx(ctx);
	if (likely(NULL != ioctx)) {
Kent Overstreet's avatar
Kent Overstreet committed
916
		kill_ioctx(ioctx);
917
		put_ioctx(ioctx);
Linus Torvalds's avatar
Linus Torvalds committed
918
919
920
921
922
923
		return 0;
	}
	pr_debug("EINVAL: io_destroy: invalid context id\n");
	return -EINVAL;
}

924
static void aio_advance_iovec(struct kiocb *iocb, ssize_t ret)
Linus Torvalds's avatar
Linus Torvalds committed
925
{
926
927
928
929
930
931
932
933
934
935
936
937
938
	struct iovec *iov = &iocb->ki_iovec[iocb->ki_cur_seg];

	BUG_ON(ret <= 0);

	while (iocb->ki_cur_seg < iocb->ki_nr_segs && ret > 0) {
		ssize_t this = min((ssize_t)iov->iov_len, ret);
		iov->iov_base += this;
		iov->iov_len -= this;
		iocb->ki_left -= this;
		ret -= this;
		if (iov->iov_len == 0) {
			iocb->ki_cur_seg++;
			iov++;
939
		}
940
	}
Linus Torvalds's avatar
Linus Torvalds committed
941

942
943
944
	/* the caller should not have done more io than what fit in
	 * the remaining iovecs */
	BUG_ON(ret > 0 && iocb->ki_left == 0);
Linus Torvalds's avatar
Linus Torvalds committed
945
946
}

947
static ssize_t aio_rw_vect_retry(struct kiocb *iocb)
Linus Torvalds's avatar
Linus Torvalds committed
948
949
{
	struct file *file = iocb->ki_filp;
950
951
952
953
	struct address_space *mapping = file->f_mapping;
	struct inode *inode = mapping->host;
	ssize_t (*rw_op)(struct kiocb *, const struct iovec *,
			 unsigned long, loff_t);
Linus Torvalds's avatar
Linus Torvalds committed
954
	ssize_t ret = 0;
955
956
957
958
959
960
961
962
963
964
	unsigned short opcode;

	if ((iocb->ki_opcode == IOCB_CMD_PREADV) ||
		(iocb->ki_opcode == IOCB_CMD_PREAD)) {
		rw_op = file->f_op->aio_read;
		opcode = IOCB_CMD_PREADV;
	} else {
		rw_op = file->f_op->aio_write;
		opcode = IOCB_CMD_PWRITEV;
	}
Linus Torvalds's avatar
Linus Torvalds committed
965

966
967
968
969
	/* This matches the pread()/pwrite() logic */
	if (iocb->ki_pos < 0)
		return -EINVAL;

970
971
	if (opcode == IOCB_CMD_PWRITEV)
		file_start_write(file);
972
	do {
973
974
975
976
977
978
979
980
981
982
983
		ret = rw_op(iocb, &iocb->ki_iovec[iocb->ki_cur_seg],
			    iocb->ki_nr_segs - iocb->ki_cur_seg,
			    iocb->ki_pos);
		if (ret > 0)
			aio_advance_iovec(iocb, ret);

	/* retry all partial writes.  retry partial reads as long as its a
	 * regular file. */
	} while (ret > 0 && iocb->ki_left > 0 &&
		 (opcode == IOCB_CMD_PWRITEV ||
		  (!S_ISFIFO(inode->i_mode) && !S_ISSOCK(inode->i_mode))));
984
985
	if (opcode == IOCB_CMD_PWRITEV)
		file_end_write(file);
Linus Torvalds's avatar
Linus Torvalds committed
986

987
988
	/* This means we must have transferred all that we could */
	/* No need to retry anymore */
Linus Torvalds's avatar
Linus Torvalds committed
989
990
991
	if ((ret == 0) || (iocb->ki_left == 0))
		ret = iocb->ki_nbytes - iocb->ki_left;

992
993
994
	/* If we managed to write some out we return that, rather than
	 * the eventual error. */
	if (opcode == IOCB_CMD_PWRITEV
Zach Brown's avatar
Zach Brown committed
995
	    && ret < 0 && ret != -EIOCBQUEUED
996
997
998
	    && iocb->ki_nbytes - iocb->ki_left)
		ret = iocb->ki_nbytes - iocb->ki_left;

Linus Torvalds's avatar
Linus Torvalds committed
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
	return ret;
}

static ssize_t aio_fdsync(struct kiocb *iocb)
{
	struct file *file = iocb->ki_filp;
	ssize_t ret = -EINVAL;

	if (file->f_op->aio_fsync)
		ret = file->f_op->aio_fsync(iocb, 1);
	return ret;
}

static ssize_t aio_fsync(struct kiocb *iocb)
{
	struct file *file = iocb->ki_filp;
	ssize_t ret = -EINVAL;

	if (file->f_op->aio_fsync)
		ret = file->f_op->aio_fsync(iocb, 0);
	return ret;
}

1022
static ssize_t aio_setup_vectored_rw(int type, struct kiocb *kiocb, bool compat)
1023
1024
1025
{
	ssize_t ret;

1026
1027
1028
1029
1030
#ifdef CONFIG_COMPAT
	if (compat)
		ret = compat_rw_copy_check_uvector(type,
				(struct compat_iovec __user *)kiocb->ki_buf,
				kiocb->ki_nbytes, 1, &kiocb->ki_inline_vec,
1031
				&kiocb->ki_iovec);
1032
1033
1034
1035
1036
	else
#endif
		ret = rw_copy_check_uvector(type,
				(struct iovec __user *)kiocb->ki_buf,
				kiocb->ki_nbytes, 1, &kiocb->ki_inline_vec,
1037
				&kiocb->ki_iovec);
1038
1039
1040
	if (ret < 0)
		goto out;

1041
1042
1043
1044
	ret = rw_verify_area(type, kiocb->ki_filp, &kiocb->ki_pos, ret);
	if (ret < 0)
		goto out;

1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
	kiocb->ki_nr_segs = kiocb->ki_nbytes;
	kiocb->ki_cur_seg = 0;
	/* ki_nbytes/left now reflect bytes instead of segs */
	kiocb->ki_nbytes = ret;
	kiocb->ki_left = ret;

	ret = 0;
out:
	return ret;
}

1056
static ssize_t aio_setup_single_vector(int type, struct file * file, struct kiocb *kiocb)
1057
{
1058
1059
1060
1061
1062
1063
	int bytes;

	bytes = rw_verify_area(type, file, &kiocb->ki_pos, kiocb->ki_left);
	if (bytes < 0)
		return bytes;

1064
1065
	kiocb->ki_iovec = &kiocb->ki_inline_vec;
	kiocb->ki_iovec->iov_base = kiocb->ki_buf;
1066
	kiocb->ki_iovec->iov_len = bytes;
1067
1068
1069
1070
1071
	kiocb->ki_nr_segs = 1;
	kiocb->ki_cur_seg = 0;
	return 0;
}

Linus Torvalds's avatar
Linus Torvalds committed
1072
1073
1074
1075
1076
/*
 * aio_setup_iocb:
 *	Performs the initial checks and aio retry method
 *	setup for the kiocb at the time of io submission.
 */
1077
static ssize_t aio_setup_iocb(struct kiocb *kiocb, bool compat)
Linus Torvalds's avatar
Linus Torvalds committed
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
{
	struct file *file = kiocb->ki_filp;
	ssize_t ret = 0;

	switch (kiocb->ki_opcode) {
	case IOCB_CMD_PREAD:
		ret = -EBADF;
		if (unlikely(!(file->f_mode & FMODE_READ)))
			break;
		ret = -EFAULT;
		if (unlikely(!access_ok(VERIFY_WRITE, kiocb->ki_buf,
			kiocb->ki_left)))
			break;
1091
		ret = aio_setup_single_vector(READ, file, kiocb);
1092
1093
		if (ret)
			break;
Linus Torvalds's avatar
Linus Torvalds committed
1094
1095
		ret = -EINVAL;
		if (file->f_op->aio_read)
1096
			kiocb->ki_retry = aio_rw_vect_retry;
Linus Torvalds's avatar
Linus Torvalds committed
1097
1098
1099
1100
1101
1102
1103
1104
1105
		break;
	case IOCB_CMD_PWRITE:
		ret = -EBADF;
		if (unlikely(!(file->f_mode & FMODE_WRITE)))
			break;
		ret = -EFAULT;
		if (unlikely(!access_ok(VERIFY_READ, kiocb->ki_buf,
			kiocb->ki_left)))
			break;
1106
		ret = aio_setup_single_vector(WRITE, file, kiocb);
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
		if (ret)
			break;
		ret = -EINVAL;
		if (file->f_op->aio_write)
			kiocb->ki_retry = aio_rw_vect_retry;
		break;
	case IOCB_CMD_PREADV:
		ret = -EBADF;
		if (unlikely(!(file->f_mode & FMODE_READ)))
			break;
1117
		ret = aio_setup_vectored_rw(READ, kiocb, compat);
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
		if (ret)
			break;
		ret = -EINVAL;
		if (file->f_op->aio_read)
			kiocb->ki_retry = aio_rw_vect_retry;
		break;
	case IOCB_CMD_PWRITEV:
		ret = -EBADF;
		if (unlikely(!(file->f_mode & FMODE_WRITE)))
			break;
1128
		ret = aio_setup_vectored_rw(WRITE, kiocb, compat);
1129
1130
		if (ret)
			break;
Linus Torvalds's avatar
Linus Torvalds committed
1131
1132
		ret = -EINVAL;
		if (file->f_op->aio_write)
1133
			kiocb->ki_retry = aio_rw_vect_retry;
Linus Torvalds's avatar
Linus Torvalds committed
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
		break;
	case IOCB_CMD_FDSYNC:
		ret = -EINVAL;
		if (file->f_op->aio_fsync)
			kiocb->ki_retry = aio_fdsync;
		break;
	case IOCB_CMD_FSYNC:
		ret = -EINVAL;
		if (file->f_op->aio_fsync)
			kiocb->ki_retry = aio_fsync;
		break;
	default:
Kent Overstreet's avatar
Kent Overstreet committed
1146
		pr_debug("EINVAL: no operation provided\n");
Linus Torvalds's avatar
Linus Torvalds committed
1147
1148
1149
1150
1151
1152
1153
1154
1155
		ret = -EINVAL;
	}

	if (!kiocb->ki_retry)
		return ret;

	return 0;
}

1156
static int io_submit_one(struct kioctx *ctx, struct iocb __user *user_iocb,
Jeff Moyer's avatar
Jeff Moyer committed
1157
1158
			 struct iocb *iocb, struct kiocb_batch *batch,
			 bool compat)
Linus Torvalds's avatar
Linus Torvalds committed
1159
1160
1161
1162
1163
{
	struct kiocb *req;
	ssize_t ret;

	/* enforce forwards compatibility on users */
1164
	if (unlikely(iocb->aio_reserved1 || iocb->aio_reserved2)) {
Kent Overstreet's avatar
Kent Overstreet committed
1165
		pr_debug("EINVAL: reserve field set\n");
Linus Torvalds's avatar
Linus Torvalds committed
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
		return -EINVAL;
	}

	/* prevent overflows */
	if (unlikely(
	    (iocb->aio_buf != (unsigned long)iocb->aio_buf) ||
	    (iocb->aio_nbytes != (size_t)iocb->aio_nbytes) ||
	    ((ssize_t)iocb->aio_nbytes < 0)
	   )) {
		pr_debug("EINVAL: io_submit: overflow check\n");
		return -EINVAL;
	}

Jeff Moyer's avatar
Jeff Moyer committed
1179
	req = aio_get_req(ctx, batch);  /* returns with 2 references to req */
1180
	if (unlikely(!req))
Linus Torvalds's avatar
Linus Torvalds committed
1181
		return -EAGAIN;
1182
1183
1184
1185
1186

	req->ki_filp = fget(iocb->aio_fildes);
	if (unlikely(!req->ki_filp)) {
		ret = -EBADF;
		goto out_put_req;
Linus Torvalds's avatar
Linus Torvalds committed
1187
	}
1188

1189
1190
1191
1192
1193
1194
1195
	if (iocb->aio_flags & IOCB_FLAG_RESFD) {
		/*
		 * If the IOCB_FLAG_RESFD flag of aio_flags is set, get an
		 * instance of the file* now. The file descriptor must be
		 * an eventfd() fd, and will be signaled for each completed
		 * event using the eventfd_signal() function.
		 */
1196
		req->ki_eventfd = eventfd_ctx_fdget((int) iocb->aio_resfd);
1197
		if (IS_ERR(req->ki_eventfd)) {
1198
			ret = PTR_ERR(req->ki_eventfd);
1199
			req->ki_eventfd = NULL;
1200
1201
1202
			goto out_put_req;
		}
	}
Linus Torvalds's avatar
Linus Torvalds committed
1203

1204
	ret = put_user(req->ki_key, &user_iocb->aio_key);
Linus Torvalds's avatar
Linus Torvalds committed
1205
	if (unlikely(ret)) {
Kent Overstreet's avatar
Kent Overstreet committed
1206
		pr_debug("EFAULT: aio_key\n");
Linus Torvalds's avatar
Linus Torvalds committed
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
		goto out_put_req;
	}

	req->ki_obj.user = user_iocb;
	req->ki_user_data = iocb->aio_data;
	req->ki_pos = iocb->aio_offset;

	req->ki_buf = (char __user *)(unsigned long)iocb->aio_buf;
	req->ki_left = req->ki_nbytes = iocb->aio_nbytes;
	req->ki_opcode = iocb->aio_lio_opcode;

1218
	ret = aio_setup_iocb(req, compat);
Linus Torvalds's avatar
Linus Torvalds committed
1219

Zach Brown's avatar
Zach Brown committed
1220
	if (ret)
1221
		goto out_put_req;
Zach Brown's avatar
Zach Brown committed
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237

	if (unlikely(kiocbIsCancelled(req)))
		ret = -EINTR;
	else
		ret = req->ki_retry(req);

	if (ret != -EIOCBQUEUED) {
		/*
		 * There's no easy way to restart the syscall since other AIO's
		 * may be already running. Just fail this IO with EINTR.
		 */
		if (unlikely(ret == -ERESTARTSYS || ret == -ERESTARTNOINTR ||
			     ret == -ERESTARTNOHAND ||
			     ret == -ERESTART_RESTARTBLOCK))
			ret = -EINTR;
		aio_complete(req, ret, 0);
1238
	}
Jeff Moyer's avatar
Jeff Moyer committed
1239

Linus Torvalds's avatar
Linus Torvalds committed
1240
1241
1242
1243
	aio_put_req(req);	/* drop extra ref to req */
	return 0;

out_put_req:
1244
1245
1246
1247
1248
	spin_lock_irq(&ctx->ctx_lock);
	list_del(&req->ki_list);
	spin_unlock_irq(&ctx->ctx_lock);

	atomic_dec(&ctx->reqs_active);
Linus Torvalds's avatar
Linus Torvalds committed
1249
1250
1251
1252
1253
	aio_put_req(req);	/* drop extra ref to req */
	aio_put_req(req);	/* drop i/o ref to req */
	return ret;
}

1254
1255
long do_io_submit(aio_context_t ctx_id, long nr,
		  struct iocb __user *__user *iocbpp, bool compat)
Linus Torvalds's avatar
Linus Torvalds committed
1256
1257
1258
{
	struct kioctx *ctx;
	long ret = 0;
Jeff Moyer's avatar
Jeff Moyer committed
1259
	int i = 0;
Shaohua Li's avatar
Shaohua Li committed
1260
	struct blk_plug plug;
Jeff Moyer's avatar
Jeff Moyer committed
1261
	struct kiocb_batch batch;
Linus Torvalds's avatar
Linus Torvalds committed
1262
1263
1264
1265

	if (unlikely(nr < 0))
		return -EINVAL;

1266
1267
1268
	if (unlikely(nr > LONG_MAX/sizeof(*iocbpp)))
		nr = LONG_MAX/sizeof(*iocbpp);

Linus Torvalds's avatar
Linus Torvalds committed
1269
1270
1271
1272
1273
	if (unlikely(!access_ok(VERIFY_READ, iocbpp, (nr*sizeof(*iocbpp)))))
		return -EFAULT;

	ctx = lookup_ioctx(ctx_id);
	if (unlikely(!ctx)) {
Kent Overstreet's avatar
Kent Overstreet committed
1274
		pr_debug("EINVAL: invalid context id\n");
Linus Torvalds's avatar
Linus Torvalds committed
1275
1276
1277
		return -EINVAL;
	}

Jeff Moyer's avatar
Jeff Moyer committed
1278
1279
	kiocb_batch_init(&batch, nr);

Shaohua Li's avatar
Shaohua Li committed
1280
1281
	blk_start_plug(&plug);

Linus Torvalds's avatar
Linus Torvalds committed
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
	/*
	 * AKPM: should this return a partial result if some of the IOs were
	 * successfully submitted?
	 */
	for (i=0; i<nr; i++) {
		struct iocb __user *user_iocb;
		struct iocb tmp;

		if (unlikely(__get_user(user_iocb, iocbpp + i))) {
			ret = -EFAULT;
			break;
		}

		if (unlikely(copy_from_user(&tmp, user_iocb, sizeof(tmp)))) {
			ret = -EFAULT;
			break;
		}

Jeff Moyer's avatar
Jeff Moyer committed
1300
		ret = io_submit_one(ctx, user_iocb, &tmp, &batch, compat);
Linus Torvalds's avatar
Linus Torvalds committed
1301
1302
1303
		if (ret)
			break;
	}
Shaohua Li's avatar
Shaohua Li committed
1304
	blk_finish_plug(&plug);
Linus Torvalds's avatar
Linus Torvalds committed
1305

1306
	kiocb_batch_free(ctx, &batch);
Linus Torvalds's avatar
Linus Torvalds committed
1307
1308
1309
1310
	put_ioctx(ctx);
	return i ? i : ret;
}

1311
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
/* sys_io_submit:
 *	Queue the nr iocbs pointed to by iocbpp for processing.  Returns
 *	the number of iocbs queued.  May return -EINVAL if the aio_context
 *	specified by ctx_id is invalid, if nr is < 0, if the iocb at
 *	*iocbpp[0] is not properly initialized, if the operation specified
 *	is invali