aio.c 36.5 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>
Kent Overstreet's avatar
Kent Overstreet committed
29
#include <linux/percpu.h>
Linus Torvalds's avatar
Linus Torvalds committed
30
31
32
33
34
35
#include <linux/slab.h>
#include <linux/timer.h>
#include <linux/aio.h>
#include <linux/highmem.h>
#include <linux/workqueue.h>
#include <linux/security.h>
36
#include <linux/eventfd.h>
Jeff Moyer's avatar
Jeff Moyer committed
37
#include <linux/blkdev.h>
38
#include <linux/compat.h>
39
40
41
#include <linux/anon_inodes.h>
#include <linux/migrate.h>
#include <linux/ramfs.h>
Kent Overstreet's avatar
Kent Overstreet committed
42
#include <linux/percpu-refcount.h>
Linus Torvalds's avatar
Linus Torvalds committed
43
44
45
46

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

Al Viro's avatar
Al Viro committed
47
48
#include "internal.h"

49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#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

69
70
71
72
73
74
struct kioctx_table {
	struct rcu_head	rcu;
	unsigned	nr;
	struct kioctx	*table[];
};

Kent Overstreet's avatar
Kent Overstreet committed
75
76
77
78
struct kioctx_cpu {
	unsigned		reqs_available;
};

79
struct kioctx {
Kent Overstreet's avatar
Kent Overstreet committed
80
	struct percpu_ref	users;
Kent Overstreet's avatar
Kent Overstreet committed
81
	atomic_t		dead;
82
83
84

	unsigned long		user_id;

Kent Overstreet's avatar
Kent Overstreet committed
85
86
87
88
89
90
91
	struct __percpu kioctx_cpu *cpu;

	/*
	 * For percpu reqs_available, number of slots we move to/from global
	 * counter at a time:
	 */
	unsigned		req_batch;
92
93
94
95
	/*
	 * This is what userspace passed to io_setup(), it's not used for
	 * anything but counting against the global max_reqs quota.
	 *
96
	 * The real limit is nr_events - 1, which will be larger (see
97
98
	 * aio_setup_ring())
	 */
99
100
	unsigned		max_reqs;

101
102
	/* Size of ringbuffer, in units of struct io_event */
	unsigned		nr_events;
103

104
105
106
107
108
109
	unsigned long		mmap_base;
	unsigned long		mmap_size;

	struct page		**ring_pages;
	long			nr_pages;

110
	struct rcu_head		rcu_head;
Kent Overstreet's avatar
Kent Overstreet committed
111
	struct work_struct	free_work;
112
113

	struct {
114
115
116
117
118
		/*
		 * This counts the number of available slots in the ringbuffer,
		 * so we avoid overflowing it: it's decremented (if positive)
		 * when allocating a kiocb and incremented when the resulting
		 * io_event is pulled off the ringbuffer.
Kent Overstreet's avatar
Kent Overstreet committed
119
120
		 *
		 * We batch accesses to it with a percpu version.
121
122
		 */
		atomic_t	reqs_available;
123
124
125
126
127
128
129
	} ____cacheline_aligned_in_smp;

	struct {
		spinlock_t	ctx_lock;
		struct list_head active_reqs;	/* used for cancellation */
	} ____cacheline_aligned_in_smp;

130
131
	struct {
		struct mutex	ring_lock;
132
133
		wait_queue_head_t wait;
	} ____cacheline_aligned_in_smp;
134
135
136
137

	struct {
		unsigned	tail;
		spinlock_t	completion_lock;
138
	} ____cacheline_aligned_in_smp;
139
140

	struct page		*internal_pages[AIO_RING_PAGES];
141
	struct file		*aio_ring_file;
142
143

	unsigned		id;
144
145
};

Linus Torvalds's avatar
Linus Torvalds committed
146
/*------ sysctl variables----*/
147
148
149
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
150
151
/*----end sysctl variables---*/

152
153
static struct kmem_cache	*kiocb_cachep;
static struct kmem_cache	*kioctx_cachep;
Linus Torvalds's avatar
Linus Torvalds committed
154
155
156
157
158
159
160

/* 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)
{
161
162
	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
163

Kent Overstreet's avatar
Kent Overstreet committed
164
	pr_debug("sizeof(struct page) = %zu\n", sizeof(struct page));
Linus Torvalds's avatar
Linus Torvalds committed
165
166
167

	return 0;
}
168
__initcall(aio_setup);
Linus Torvalds's avatar
Linus Torvalds committed
169
170
171

static void aio_free_ring(struct kioctx *ctx)
{
172
173
	int i;
	struct file *aio_ring_file = ctx->aio_ring_file;
Linus Torvalds's avatar
Linus Torvalds committed
174

175
176
177
	for (i = 0; i < ctx->nr_pages; i++) {
		pr_debug("pid(%d) [%d] page->count=%d\n", current->pid, i,
				page_count(ctx->ring_pages[i]));
178
		put_page(ctx->ring_pages[i]);
179
	}
Linus Torvalds's avatar
Linus Torvalds committed
180

181
182
	if (ctx->ring_pages && ctx->ring_pages != ctx->internal_pages)
		kfree(ctx->ring_pages);
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210

	if (aio_ring_file) {
		truncate_setsize(aio_ring_file->f_inode, 0);
		pr_debug("pid(%d) i_nlink=%u d_count=%d d_unhashed=%d i_count=%d\n",
			current->pid, aio_ring_file->f_inode->i_nlink,
			aio_ring_file->f_path.dentry->d_count,
			d_unhashed(aio_ring_file->f_path.dentry),
			atomic_read(&aio_ring_file->f_inode->i_count));
		fput(aio_ring_file);
		ctx->aio_ring_file = NULL;
	}
}

static int aio_ring_mmap(struct file *file, struct vm_area_struct *vma)
{
	vma->vm_ops = &generic_file_vm_ops;
	return 0;
}

static const struct file_operations aio_ring_fops = {
	.mmap = aio_ring_mmap,
};

static int aio_set_page_dirty(struct page *page)
{
	return 0;
}

211
#if IS_ENABLED(CONFIG_MIGRATION)
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
static int aio_migratepage(struct address_space *mapping, struct page *new,
			struct page *old, enum migrate_mode mode)
{
	struct kioctx *ctx = mapping->private_data;
	unsigned long flags;
	unsigned idx = old->index;
	int rc;

	/* Writeback must be complete */
	BUG_ON(PageWriteback(old));
	put_page(old);

	rc = migrate_page_move_mapping(mapping, new, old, NULL, mode);
	if (rc != MIGRATEPAGE_SUCCESS) {
		get_page(old);
		return rc;
	}

	get_page(new);

	spin_lock_irqsave(&ctx->completion_lock, flags);
	migrate_page_copy(new, old);
	ctx->ring_pages[idx] = new;
	spin_unlock_irqrestore(&ctx->completion_lock, flags);

	return rc;
Linus Torvalds's avatar
Linus Torvalds committed
238
}
239
#endif
Linus Torvalds's avatar
Linus Torvalds committed
240

241
242
static const struct address_space_operations aio_ctx_aops = {
	.set_page_dirty = aio_set_page_dirty,
243
#if IS_ENABLED(CONFIG_MIGRATION)
244
	.migratepage	= aio_migratepage,
245
#endif
246
247
};

Linus Torvalds's avatar
Linus Torvalds committed
248
249
250
251
static int aio_setup_ring(struct kioctx *ctx)
{
	struct aio_ring *ring;
	unsigned nr_events = ctx->max_reqs;
Zach Brown's avatar
Zach Brown committed
252
	struct mm_struct *mm = current->mm;
253
	unsigned long size, populate;
Linus Torvalds's avatar
Linus Torvalds committed
254
	int nr_pages;
255
256
	int i;
	struct file *file;
Linus Torvalds's avatar
Linus Torvalds committed
257
258
259
260
261
262
263

	/* 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;

264
	nr_pages = PFN_UP(size);
Linus Torvalds's avatar
Linus Torvalds committed
265
266
267
	if (nr_pages < 0)
		return -EINVAL;

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
	file = anon_inode_getfile_private("[aio]", &aio_ring_fops, ctx, O_RDWR);
	if (IS_ERR(file)) {
		ctx->aio_ring_file = NULL;
		return -EAGAIN;
	}

	file->f_inode->i_mapping->a_ops = &aio_ctx_aops;
	file->f_inode->i_mapping->private_data = ctx;
	file->f_inode->i_size = PAGE_SIZE * (loff_t)nr_pages;

	for (i = 0; i < nr_pages; i++) {
		struct page *page;
		page = find_or_create_page(file->f_inode->i_mapping,
					   i, GFP_HIGHUSER | __GFP_ZERO);
		if (!page)
			break;
		pr_debug("pid(%d) page[%d]->count=%d\n",
			 current->pid, i, page_count(page));
		SetPageUptodate(page);
		SetPageDirty(page);
		unlock_page(page);
	}
	ctx->aio_ring_file = file;
	nr_events = (PAGE_SIZE * nr_pages - sizeof(struct aio_ring))
			/ sizeof(struct io_event);
Linus Torvalds's avatar
Linus Torvalds committed
293

294
	ctx->ring_pages = ctx->internal_pages;
Linus Torvalds's avatar
Linus Torvalds committed
295
	if (nr_pages > AIO_RING_PAGES) {
296
297
298
		ctx->ring_pages = kcalloc(nr_pages, sizeof(struct page *),
					  GFP_KERNEL);
		if (!ctx->ring_pages)
Linus Torvalds's avatar
Linus Torvalds committed
299
300
301
			return -ENOMEM;
	}

302
303
	ctx->mmap_size = nr_pages * PAGE_SIZE;
	pr_debug("attempting mmap of %lu bytes\n", ctx->mmap_size);
304

Zach Brown's avatar
Zach Brown committed
305
	down_write(&mm->mmap_sem);
306
307
308
	ctx->mmap_base = do_mmap_pgoff(ctx->aio_ring_file, 0, ctx->mmap_size,
				       PROT_READ | PROT_WRITE,
				       MAP_SHARED | MAP_POPULATE, 0, &populate);
309
	if (IS_ERR((void *)ctx->mmap_base)) {
Zach Brown's avatar
Zach Brown committed
310
		up_write(&mm->mmap_sem);
311
		ctx->mmap_size = 0;
Linus Torvalds's avatar
Linus Torvalds committed
312
313
314
		aio_free_ring(ctx);
		return -EAGAIN;
	}
315
316
317
	up_write(&mm->mmap_sem);

	mm_populate(ctx->mmap_base, populate);
Linus Torvalds's avatar
Linus Torvalds committed
318

319
320
321
	pr_debug("mmap address: 0x%08lx\n", ctx->mmap_base);
	ctx->nr_pages = get_user_pages(current, mm, ctx->mmap_base, nr_pages,
				       1, 0, ctx->ring_pages, NULL);
322
323
	for (i = 0; i < ctx->nr_pages; i++)
		put_page(ctx->ring_pages[i]);
Linus Torvalds's avatar
Linus Torvalds committed
324

325
	if (unlikely(ctx->nr_pages != nr_pages)) {
Linus Torvalds's avatar
Linus Torvalds committed
326
327
328
329
		aio_free_ring(ctx);
		return -EAGAIN;
	}

330
331
	ctx->user_id = ctx->mmap_base;
	ctx->nr_events = nr_events; /* trusted copy */
Linus Torvalds's avatar
Linus Torvalds committed
332

333
	ring = kmap_atomic(ctx->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
334
	ring->nr = nr_events;	/* user copy */
335
	ring->id = ~0U;
Linus Torvalds's avatar
Linus Torvalds committed
336
337
338
339
340
	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);
341
	kunmap_atomic(ring);
342
	flush_dcache_page(ctx->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
343
344
345
346
347
348
349
350

	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)

351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
void kiocb_set_cancel_fn(struct kiocb *req, kiocb_cancel_fn *cancel)
{
	struct kioctx *ctx = req->ki_ctx;
	unsigned long flags;

	spin_lock_irqsave(&ctx->ctx_lock, flags);

	if (!req->ki_list.next)
		list_add(&req->ki_list, &ctx->active_reqs);

	req->ki_cancel = cancel;

	spin_unlock_irqrestore(&ctx->ctx_lock, flags);
}
EXPORT_SYMBOL(kiocb_set_cancel_fn);

367
static int kiocb_cancel(struct kioctx *ctx, struct kiocb *kiocb)
Kent Overstreet's avatar
Kent Overstreet committed
368
{
369
	kiocb_cancel_fn *old, *cancel;
Kent Overstreet's avatar
Kent Overstreet committed
370

371
372
373
374
375
376
377
378
	/*
	 * Don't want to set kiocb->ki_cancel = KIOCB_CANCELLED unless it
	 * actually has a cancel function, hence the cmpxchg()
	 */

	cancel = ACCESS_ONCE(kiocb->ki_cancel);
	do {
		if (!cancel || cancel == KIOCB_CANCELLED)
Kent Overstreet's avatar
Kent Overstreet committed
379
			return -EINVAL;
Kent Overstreet's avatar
Kent Overstreet committed
380

381
382
383
		old = cancel;
		cancel = cmpxchg(&kiocb->ki_cancel, old, KIOCB_CANCELLED);
	} while (cancel != old);
Kent Overstreet's avatar
Kent Overstreet committed
384

Kent Overstreet's avatar
Kent Overstreet committed
385
	return cancel(kiocb);
Kent Overstreet's avatar
Kent Overstreet committed
386
387
}

Kent Overstreet's avatar
Kent Overstreet committed
388
389
390
static void free_ioctx_rcu(struct rcu_head *head)
{
	struct kioctx *ctx = container_of(head, struct kioctx, rcu_head);
Kent Overstreet's avatar
Kent Overstreet committed
391
392

	free_percpu(ctx->cpu);
Kent Overstreet's avatar
Kent Overstreet committed
393
394
395
396
397
398
399
400
	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.
 */
Kent Overstreet's avatar
Kent Overstreet committed
401
static void free_ioctx(struct work_struct *work)
Kent Overstreet's avatar
Kent Overstreet committed
402
{
Kent Overstreet's avatar
Kent Overstreet committed
403
	struct kioctx *ctx = container_of(work, struct kioctx, free_work);
404
	struct aio_ring *ring;
Kent Overstreet's avatar
Kent Overstreet committed
405
	struct kiocb *req;
406
407
	unsigned cpu, avail;
	DEFINE_WAIT(wait);
Kent Overstreet's avatar
Kent Overstreet committed
408
409
410
411
412
413
414
415

	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);
416
		kiocb_cancel(ctx, req);
Kent Overstreet's avatar
Kent Overstreet committed
417
418
419
420
	}

	spin_unlock_irq(&ctx->ctx_lock);

Kent Overstreet's avatar
Kent Overstreet committed
421
422
423
424
425
426
427
	for_each_possible_cpu(cpu) {
		struct kioctx_cpu *kcpu = per_cpu_ptr(ctx->cpu, cpu);

		atomic_add(kcpu->reqs_available, &ctx->reqs_available);
		kcpu->reqs_available = 0;
	}

428
429
	while (1) {
		prepare_to_wait(&ctx->wait, &wait, TASK_UNINTERRUPTIBLE);
430

431
432
433
434
		ring = kmap_atomic(ctx->ring_pages[0]);
		avail = (ring->head <= ring->tail)
			 ? ring->tail - ring->head
			 : ctx->nr_events - ring->head + ring->tail;
435

436
		atomic_add(avail, &ctx->reqs_available);
437
438
439
440
441
442
443
		ring->head = ring->tail;
		kunmap_atomic(ring);

		if (atomic_read(&ctx->reqs_available) >= ctx->nr_events - 1)
			break;

		schedule();
444
	}
445
	finish_wait(&ctx->wait, &wait);
446

447
	WARN_ON(atomic_read(&ctx->reqs_available) > ctx->nr_events - 1);
Kent Overstreet's avatar
Kent Overstreet committed
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462

	aio_free_ring(ctx);

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

Kent Overstreet's avatar
Kent Overstreet committed
463
static void free_ioctx_ref(struct percpu_ref *ref)
Kent Overstreet's avatar
Kent Overstreet committed
464
{
Kent Overstreet's avatar
Kent Overstreet committed
465
466
467
468
	struct kioctx *ctx = container_of(ref, struct kioctx, users);

	INIT_WORK(&ctx->free_work, free_ioctx);
	schedule_work(&ctx->free_work);
Kent Overstreet's avatar
Kent Overstreet committed
469
470
}

471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
static int ioctx_add_table(struct kioctx *ctx, struct mm_struct *mm)
{
	unsigned i, new_nr;
	struct kioctx_table *table, *old;
	struct aio_ring *ring;

	spin_lock(&mm->ioctx_lock);
	table = rcu_dereference(mm->ioctx_table);

	while (1) {
		if (table)
			for (i = 0; i < table->nr; i++)
				if (!table->table[i]) {
					ctx->id = i;
					table->table[i] = ctx;
					spin_unlock(&mm->ioctx_lock);

					ring = kmap_atomic(ctx->ring_pages[0]);
					ring->id = ctx->id;
					kunmap_atomic(ring);
					return 0;
				}

		new_nr = (table ? table->nr : 1) * 4;

		spin_unlock(&mm->ioctx_lock);

		table = kzalloc(sizeof(*table) + sizeof(struct kioctx *) *
				new_nr, GFP_KERNEL);
		if (!table)
			return -ENOMEM;

		table->nr = new_nr;

		spin_lock(&mm->ioctx_lock);
		old = rcu_dereference(mm->ioctx_table);

		if (!old) {
			rcu_assign_pointer(mm->ioctx_table, table);
		} else if (table->nr > old->nr) {
			memcpy(table->table, old->table,
			       old->nr * sizeof(struct kioctx *));

			rcu_assign_pointer(mm->ioctx_table, table);
			kfree_rcu(old, rcu);
		} else {
			kfree(table);
			table = old;
		}
	}
}

Linus Torvalds's avatar
Linus Torvalds committed
523
524
525
526
527
/* 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
528
	struct mm_struct *mm = current->mm;
Linus Torvalds's avatar
Linus Torvalds committed
529
	struct kioctx *ctx;
530
	int err = -ENOMEM;
Linus Torvalds's avatar
Linus Torvalds committed
531

Kent Overstreet's avatar
Kent Overstreet committed
532
533
534
535
536
537
538
539
540
541
542
543
	/*
	 * We keep track of the number of available ringbuffer slots, to prevent
	 * overflow (reqs_available), and we also use percpu counters for this.
	 *
	 * So since up to half the slots might be on other cpu's percpu counters
	 * and unavailable, double nr_events so userspace sees what they
	 * expected: additionally, we move req_batch slots to/from percpu
	 * counters at a time, so make sure that isn't 0:
	 */
	nr_events = max(nr_events, num_possible_cpus() * 4);
	nr_events *= 2;

Linus Torvalds's avatar
Linus Torvalds committed
544
545
546
547
548
549
550
	/* 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);
	}

551
	if (!nr_events || (unsigned long)nr_events > (aio_max_nr * 2UL))
Linus Torvalds's avatar
Linus Torvalds committed
552
553
		return ERR_PTR(-EAGAIN);

554
	ctx = kmem_cache_zalloc(kioctx_cachep, GFP_KERNEL);
Linus Torvalds's avatar
Linus Torvalds committed
555
556
557
558
559
	if (!ctx)
		return ERR_PTR(-ENOMEM);

	ctx->max_reqs = nr_events;

Kent Overstreet's avatar
Kent Overstreet committed
560
561
562
	if (percpu_ref_init(&ctx->users, free_ioctx_ref))
		goto out_freectx;

Linus Torvalds's avatar
Linus Torvalds committed
563
	spin_lock_init(&ctx->ctx_lock);
564
	spin_lock_init(&ctx->completion_lock);
565
	mutex_init(&ctx->ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
566
567
568
569
	init_waitqueue_head(&ctx->wait);

	INIT_LIST_HEAD(&ctx->active_reqs);

Kent Overstreet's avatar
Kent Overstreet committed
570
571
	ctx->cpu = alloc_percpu(struct kioctx_cpu);
	if (!ctx->cpu)
Kent Overstreet's avatar
Kent Overstreet committed
572
		goto out_freeref;
Linus Torvalds's avatar
Linus Torvalds committed
573

Kent Overstreet's avatar
Kent Overstreet committed
574
575
576
	if (aio_setup_ring(ctx) < 0)
		goto out_freepcpu;

577
	atomic_set(&ctx->reqs_available, ctx->nr_events - 1);
Kent Overstreet's avatar
Kent Overstreet committed
578
579
	ctx->req_batch = (ctx->nr_events - 1) / (num_possible_cpus() * 4);
	BUG_ON(!ctx->req_batch);
580

581
582
583
584
	err = ioctx_add_table(ctx, mm);
	if (err)
		goto out_cleanup_noerr;

Linus Torvalds's avatar
Linus Torvalds committed
585
	/* limit the number of system wide aios */
586
	spin_lock(&aio_nr_lock);
587
	if (aio_nr + nr_events > (aio_max_nr * 2UL) ||
588
	    aio_nr + nr_events < aio_nr) {
589
		spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
590
		goto out_cleanup;
591
592
	}
	aio_nr += ctx->max_reqs;
593
	spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
594

Kent Overstreet's avatar
Kent Overstreet committed
595
596
	percpu_ref_get(&ctx->users); /* io_setup() will drop this ref */

Kent Overstreet's avatar
Kent Overstreet committed
597
	pr_debug("allocated ioctx %p[%ld]: mm=%p mask=0x%x\n",
598
		 ctx, ctx->user_id, mm, ctx->nr_events);
Linus Torvalds's avatar
Linus Torvalds committed
599
600
601
	return ctx;

out_cleanup:
602
	err = -EAGAIN;
603
out_cleanup_noerr:
604
	aio_free_ring(ctx);
Kent Overstreet's avatar
Kent Overstreet committed
605
606
out_freepcpu:
	free_percpu(ctx->cpu);
Kent Overstreet's avatar
Kent Overstreet committed
607
608
out_freeref:
	free_percpu(ctx->users.pcpu_count);
Linus Torvalds's avatar
Linus Torvalds committed
609
out_freectx:
610
611
	if (ctx->aio_ring_file)
		fput(ctx->aio_ring_file);
Linus Torvalds's avatar
Linus Torvalds committed
612
	kmem_cache_free(kioctx_cachep, ctx);
Kent Overstreet's avatar
Kent Overstreet committed
613
	pr_debug("error allocating ioctx %d\n", err);
614
	return ERR_PTR(err);
Linus Torvalds's avatar
Linus Torvalds committed
615
616
}

Kent Overstreet's avatar
Kent Overstreet committed
617
618
619
620
621
/* 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.
 */
622
static void kill_ioctx(struct mm_struct *mm, struct kioctx *ctx)
Kent Overstreet's avatar
Kent Overstreet committed
623
624
{
	if (!atomic_xchg(&ctx->dead, 1)) {
625
626
627
628
629
630
631
632
633
		struct kioctx_table *table;

		spin_lock(&mm->ioctx_lock);
		table = rcu_dereference(mm->ioctx_table);

		WARN_ON(ctx != table->table[ctx->id]);
		table->table[ctx->id] = NULL;
		spin_unlock(&mm->ioctx_lock);

Kent Overstreet's avatar
Kent Overstreet committed
634
635
		/* percpu_ref_kill() will do the necessary call_rcu() */
		wake_up_all(&ctx->wait);
636

Kent Overstreet's avatar
Kent Overstreet committed
637
		/*
638
639
640
641
642
		 * It'd be more correct to do this in free_ioctx(), after all
		 * the outstanding kiocbs have finished - but by then io_destroy
		 * has already returned, so io_setup() could potentially return
		 * -EAGAIN with no ioctxs actually in use (as far as userspace
		 *  could tell).
Kent Overstreet's avatar
Kent Overstreet committed
643
		 */
644
645
646
647
648
649
650
651
		spin_lock(&aio_nr_lock);
		BUG_ON(aio_nr - ctx->max_reqs > aio_nr);
		aio_nr -= ctx->max_reqs;
		spin_unlock(&aio_nr_lock);

		if (ctx->mmap_size)
			vm_munmap(ctx->mmap_base, ctx->mmap_size);

Kent Overstreet's avatar
Kent Overstreet committed
652
		percpu_ref_kill(&ctx->users);
Kent Overstreet's avatar
Kent Overstreet committed
653
	}
Linus Torvalds's avatar
Linus Torvalds committed
654
655
656
657
658
}

/* wait_on_sync_kiocb:
 *	Waits on the given sync kiocb to complete.
 */
Kent Overstreet's avatar
Kent Overstreet committed
659
ssize_t wait_on_sync_kiocb(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
660
{
Kent Overstreet's avatar
Kent Overstreet committed
661
	while (!req->ki_ctx) {
Linus Torvalds's avatar
Linus Torvalds committed
662
		set_current_state(TASK_UNINTERRUPTIBLE);
Kent Overstreet's avatar
Kent Overstreet committed
663
		if (req->ki_ctx)
Linus Torvalds's avatar
Linus Torvalds committed
664
			break;
665
		io_schedule();
Linus Torvalds's avatar
Linus Torvalds committed
666
667
	}
	__set_current_state(TASK_RUNNING);
Kent Overstreet's avatar
Kent Overstreet committed
668
	return req->ki_user_data;
Linus Torvalds's avatar
Linus Torvalds committed
669
}
670
EXPORT_SYMBOL(wait_on_sync_kiocb);
Linus Torvalds's avatar
Linus Torvalds committed
671

Kent Overstreet's avatar
Kent Overstreet committed
672
673
674
675
676
677
678
/*
 * 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
679
 */
680
void exit_aio(struct mm_struct *mm)
Linus Torvalds's avatar
Linus Torvalds committed
681
{
682
	struct kioctx_table *table;
683
	struct kioctx *ctx;
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
	unsigned i = 0;

	while (1) {
		rcu_read_lock();
		table = rcu_dereference(mm->ioctx_table);

		do {
			if (!table || i >= table->nr) {
				rcu_read_unlock();
				rcu_assign_pointer(mm->ioctx_table, NULL);
				if (table)
					kfree(table);
				return;
			}

			ctx = table->table[i++];
		} while (!ctx);

		rcu_read_unlock();
703

704
705
706
707
708
709
710
711
		/*
		 * 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.
		 */
712
		ctx->mmap_size = 0;
Kent Overstreet's avatar
Kent Overstreet committed
713

714
		kill_ioctx(mm, ctx);
Linus Torvalds's avatar
Linus Torvalds committed
715
716
717
	}
}

Kent Overstreet's avatar
Kent Overstreet committed
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
static void put_reqs_available(struct kioctx *ctx, unsigned nr)
{
	struct kioctx_cpu *kcpu;

	preempt_disable();
	kcpu = this_cpu_ptr(ctx->cpu);

	kcpu->reqs_available += nr;
	while (kcpu->reqs_available >= ctx->req_batch * 2) {
		kcpu->reqs_available -= ctx->req_batch;
		atomic_add(ctx->req_batch, &ctx->reqs_available);
	}

	preempt_enable();
}

static bool get_reqs_available(struct kioctx *ctx)
{
	struct kioctx_cpu *kcpu;
	bool ret = false;

	preempt_disable();
	kcpu = this_cpu_ptr(ctx->cpu);

	if (!kcpu->reqs_available) {
		int old, avail = atomic_read(&ctx->reqs_available);

		do {
			if (avail < ctx->req_batch)
				goto out;

			old = avail;
			avail = atomic_cmpxchg(&ctx->reqs_available,
					       avail, avail - ctx->req_batch);
		} while (avail != old);

		kcpu->reqs_available += ctx->req_batch;
	}

	ret = true;
	kcpu->reqs_available--;
out:
	preempt_enable();
	return ret;
}

Linus Torvalds's avatar
Linus Torvalds committed
764
/* aio_get_req
Kent Overstreet's avatar
Kent Overstreet committed
765
766
 *	Allocate a slot for an aio request.
 * Returns NULL if no requests are free.
Linus Torvalds's avatar
Linus Torvalds committed
767
 */
Kent Overstreet's avatar
Kent Overstreet committed
768
static inline struct kiocb *aio_get_req(struct kioctx *ctx)
Linus Torvalds's avatar
Linus Torvalds committed
769
{
Kent Overstreet's avatar
Kent Overstreet committed
770
771
	struct kiocb *req;

Kent Overstreet's avatar
Kent Overstreet committed
772
	if (!get_reqs_available(ctx))
Kent Overstreet's avatar
Kent Overstreet committed
773
774
		return NULL;

775
	req = kmem_cache_alloc(kiocb_cachep, GFP_KERNEL|__GFP_ZERO);
Linus Torvalds's avatar
Linus Torvalds committed
776
	if (unlikely(!req))
Kent Overstreet's avatar
Kent Overstreet committed
777
		goto out_put;
Linus Torvalds's avatar
Linus Torvalds committed
778
779

	req->ki_ctx = ctx;
Jeff Moyer's avatar
Jeff Moyer committed
780
	return req;
Kent Overstreet's avatar
Kent Overstreet committed
781
out_put:
Kent Overstreet's avatar
Kent Overstreet committed
782
	put_reqs_available(ctx, 1);
Kent Overstreet's avatar
Kent Overstreet committed
783
	return NULL;
Linus Torvalds's avatar
Linus Torvalds committed
784
785
}

786
static void kiocb_free(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
787
{
788
789
	if (req->ki_filp)
		fput(req->ki_filp);
790
791
	if (req->ki_eventfd != NULL)
		eventfd_ctx_put(req->ki_eventfd);
Linus Torvalds's avatar
Linus Torvalds committed
792
793
794
	kmem_cache_free(kiocb_cachep, req);
}

795
static struct kioctx *lookup_ioctx(unsigned long ctx_id)
Linus Torvalds's avatar
Linus Torvalds committed
796
{
797
	struct aio_ring __user *ring  = (void __user *)ctx_id;
798
	struct mm_struct *mm = current->mm;
799
	struct kioctx *ctx, *ret = NULL;
800
801
802
803
804
	struct kioctx_table *table;
	unsigned id;

	if (get_user(id, &ring->id))
		return NULL;
Linus Torvalds's avatar
Linus Torvalds committed
805

806
	rcu_read_lock();
807
	table = rcu_dereference(mm->ioctx_table);
808

809
810
	if (!table || id >= table->nr)
		goto out;
Linus Torvalds's avatar
Linus Torvalds committed
811

812
813
814
815
816
817
	ctx = table->table[id];
	if (ctx->user_id == ctx_id) {
		percpu_ref_get(&ctx->users);
		ret = ctx;
	}
out:
818
	rcu_read_unlock();
819
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
820
821
822
823
824
}

/* aio_complete
 *	Called when the io request on the given iocb is complete.
 */
825
void aio_complete(struct kiocb *iocb, long res, long res2)
Linus Torvalds's avatar
Linus Torvalds committed
826
827
828
{
	struct kioctx	*ctx = iocb->ki_ctx;
	struct aio_ring	*ring;
Kent Overstreet's avatar
Kent Overstreet committed
829
	struct io_event	*ev_page, *event;
Linus Torvalds's avatar
Linus Torvalds committed
830
	unsigned long	flags;
Kent Overstreet's avatar
Kent Overstreet committed
831
	unsigned tail, pos;
Linus Torvalds's avatar
Linus Torvalds committed
832

833
834
835
836
837
838
	/*
	 * 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
839
840
841
	 */
	if (is_sync_kiocb(iocb)) {
		iocb->ki_user_data = res;
Kent Overstreet's avatar
Kent Overstreet committed
842
843
		smp_wmb();
		iocb->ki_ctx = ERR_PTR(-EXDEV);
Linus Torvalds's avatar
Linus Torvalds committed
844
		wake_up_process(iocb->ki_obj.tsk);
845
		return;
Linus Torvalds's avatar
Linus Torvalds committed
846
847
	}

Kent Overstreet's avatar
Kent Overstreet committed
848
849
	/*
	 * Take rcu_read_lock() in case the kioctx is being destroyed, as we
850
	 * need to issue a wakeup after incrementing reqs_available.
Linus Torvalds's avatar
Linus Torvalds committed
851
	 */
Kent Overstreet's avatar
Kent Overstreet committed
852
	rcu_read_lock();
Linus Torvalds's avatar
Linus Torvalds committed
853

854
855
856
857
858
859
860
	if (iocb->ki_list.next) {
		unsigned long flags;

		spin_lock_irqsave(&ctx->ctx_lock, flags);
		list_del(&iocb->ki_list);
		spin_unlock_irqrestore(&ctx->ctx_lock, flags);
	}
861

862
863
	/*
	 * Add a completion event to the ring buffer. Must be done holding
864
	 * ctx->completion_lock to prevent other code from messing with the tail
865
866
867
868
	 * pointer since we might be called from irq context.
	 */
	spin_lock_irqsave(&ctx->completion_lock, flags);

869
	tail = ctx->tail;
Kent Overstreet's avatar
Kent Overstreet committed
870
871
	pos = tail + AIO_EVENTS_OFFSET;

872
	if (++tail >= ctx->nr_events)
873
		tail = 0;
Linus Torvalds's avatar
Linus Torvalds committed
874

875
	ev_page = kmap_atomic(ctx->ring_pages[pos / AIO_EVENTS_PER_PAGE]);
Kent Overstreet's avatar
Kent Overstreet committed
876
877
	event = ev_page + pos % AIO_EVENTS_PER_PAGE;

Linus Torvalds's avatar
Linus Torvalds committed
878
879
880
881
882
	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
883
	kunmap_atomic(ev_page);
884
	flush_dcache_page(ctx->ring_pages[pos / AIO_EVENTS_PER_PAGE]);
Kent Overstreet's avatar
Kent Overstreet committed
885
886

	pr_debug("%p[%u]: %p: %p %Lx %lx %lx\n",
Kent Overstreet's avatar
Kent Overstreet committed
887
888
		 ctx, tail, iocb, iocb->ki_obj.user, iocb->ki_user_data,
		 res, res2);
Linus Torvalds's avatar
Linus Torvalds committed
889
890
891
892
893
894

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

895
	ctx->tail = tail;
Linus Torvalds's avatar
Linus Torvalds committed
896

897
	ring = kmap_atomic(ctx->ring_pages[0]);
Kent Overstreet's avatar
Kent Overstreet committed
898
	ring->tail = tail;
899
	kunmap_atomic(ring);
900
	flush_dcache_page(ctx->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
901

902
903
	spin_unlock_irqrestore(&ctx->completion_lock, flags);

Kent Overstreet's avatar
Kent Overstreet committed
904
	pr_debug("added to ring %p at [%u]\n", iocb, tail);
Davide Libenzi's avatar
Davide Libenzi committed
905
906
907
908
909
910

	/*
	 * 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.
	 */
911
	if (iocb->ki_eventfd != NULL)
Davide Libenzi's avatar
Davide Libenzi committed
912
913
		eventfd_signal(iocb->ki_eventfd, 1);

Linus Torvalds's avatar
Linus Torvalds committed
914
	/* everything turned out well, dispose of the aiocb. */
Kent Overstreet's avatar
Kent Overstreet committed
915
	kiocb_free(iocb);
Linus Torvalds's avatar
Linus Torvalds committed
916

917
918
919
920
921
922
923
924
	/*
	 * 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
925
926
927
	if (waitqueue_active(&ctx->wait))
		wake_up(&ctx->wait);

Kent Overstreet's avatar
Kent Overstreet committed
928
	rcu_read_unlock();
Linus Torvalds's avatar
Linus Torvalds committed
929
}
930
EXPORT_SYMBOL(aio_complete);
Linus Torvalds's avatar
Linus Torvalds committed
931

932
933
934
/* 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
935
 */
936
937
static long aio_read_events_ring(struct kioctx *ctx,
				 struct io_event __user *event, long nr)
Linus Torvalds's avatar
Linus Torvalds committed
938
939
{
	struct aio_ring *ring;
940
	unsigned head, tail, pos;
941
942
943
	long ret = 0;
	int copy_ret;

944
	mutex_lock(&ctx->ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
945

946
	ring = kmap_atomic(ctx->ring_pages[0]);
947
	head = ring->head;
948
	tail = ring->tail;
949
950
	kunmap_atomic(ring);

951
	pr_debug("h%u t%u m%u\n", head, tail, ctx->nr_events);
Linus Torvalds's avatar
Linus Torvalds committed
952

953
	if (head == tail)
Linus Torvalds's avatar
Linus Torvalds committed
954
955
		goto out;

956
957
958
959
960
	while (ret < nr) {
		long avail;
		struct io_event *ev;
		struct page *page;

961
962
		avail = (head <= tail ?  tail : ctx->nr_events) - head;
		if (head == tail)
963
964
965
966
967
968
969
			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;
970
		page = ctx->ring_pages[pos / AIO_EVENTS_PER_PAGE];
971
972
973
974
975
976
977
978
979
980
981
982
983
984
		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;
985
		head %= ctx->nr_events;
Linus Torvalds's avatar
Linus Torvalds committed
986
987
	}

988
	ring = kmap_atomic(ctx->ring_pages[0]);
989
	ring->head = head;
990
	kunmap_atomic(ring);
991
	flush_dcache_page(ctx->ring_pages[0]);
992

993
	pr_debug("%li  h%u t%u\n", ret, head, tail);
994

Kent Overstreet's avatar
Kent Overstreet committed
995
	put_reqs_available(ctx, ret);
996
out:
997
	mutex_unlock(&ctx->ring_lock);
998

Linus Torvalds's avatar
Linus Torvalds committed
999
1000
1001
	return ret;
}

1002
1003
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
1004
{
1005
	long ret = aio_read_events_ring(ctx, event + *i, nr - *i);
Linus Torvalds's avatar
Linus Torvalds committed
1006

1007
1008
	if (ret > 0)
		*i += ret;
Linus Torvalds's avatar
Linus Torvalds committed
1009

1010
1011
	if (unlikely(atomic_read(&ctx->dead)))
		ret = -EINVAL;
Linus Torvalds's avatar
Linus Torvalds committed
1012

1013
1014
	if (!*i)
		*i = ret;
Linus Torvalds's avatar
Linus Torvalds committed
1015

1016
	return ret < 0 || *i >= min_nr;
Linus Torvalds's avatar
Linus Torvalds committed
1017
1018
}

1019
static long read_events(struct kioctx *ctx, long min_nr, long nr,
Linus Torvalds's avatar
Linus Torvalds committed
1020
1021
1022
			struct io_event __user *event,
			struct timespec __user *timeout)
{
1023
1024
	ktime_t until = { .tv64 = KTIME_MAX };
	long ret = 0;
Linus Torvalds's avatar
Linus Torvalds committed
1025
1026
1027

	if (timeout) {
		struct timespec	ts;
1028

Linus Torvalds's avatar
Linus Torvalds committed
1029
		if (unlikely(copy_from_user(&ts, timeout, sizeof(ts))))
1030
			return -EFAULT;
Linus Torvalds's avatar
Linus Torvalds committed
1031

1032
		until = timespec_to_ktime(ts);
Linus Torvalds's avatar
Linus Torvalds committed
1033
1034
	}

1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
	/*
	 * 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
1051

1052
1053
	if (!ret && signal_pending(current))
		ret = -EINTR;
Linus Torvalds's avatar
Linus Torvalds committed
1054

1055
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
}

/* 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.
 */
1071
SYSCALL_DEFINE2(io_setup, unsigned, nr_events, aio_context_t __user *, ctxp)
Linus Torvalds's avatar
Linus Torvalds committed
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
{
	struct kioctx *ioctx = NULL;
	unsigned long ctx;
	long ret;

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

	ret = -EINVAL;
1082
1083
1084
	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
1085
1086
1087
1088
1089
1090
1091
		goto out;
	}

	ioctx = ioctx_alloc(nr_events);
	ret = PTR_ERR(ioctx);
	if (!IS_ERR(ioctx)) {
		ret = put_user(ioctx->user_id, ctxp);
1092
		if (ret)
1093
			kill_ioctx(current->mm, ioctx);
Kent Overstreet's avatar
Kent Overstreet committed
1094
		percpu_ref_put(&ioctx->users);
Linus Torvalds's avatar
Linus Torvalds committed
1095
1096
1097
1098
1099
1100
1101
1102
1103
	}

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
1104
 *	implemented.  May fail with -EINVAL if the context pointed to
Linus Torvalds's avatar
Linus Torvalds committed
1105
1106
 *	is invalid.
 */
1107
SYSCALL_DEFINE1(io_destroy, aio_context_t, ctx)
Linus Torvalds's avatar
Linus Torvalds committed
1108
1109
1110
{
	struct kioctx *ioctx = lookup_ioctx(ctx);
	if (likely(NULL != ioctx)) {
1111
		kill_ioctx(current->mm, ioctx);
Kent Overstreet's avatar
Kent Overstreet committed
1112
		percpu_ref_put(&ioctx->users);
Linus Torvalds's avatar
Linus Torvalds committed
1113
1114
1115
1116
1117
1118
		return 0;
	}
	pr_debug("EINVAL: io_destroy: invalid context id\n");
	return -EINVAL;
}

Kent Overstreet's avatar
Kent Overstreet committed
1119
1120
1121
typedef ssize_t (aio_rw_op)(struct kiocb *, const struct iovec *,
			    unsigned long, loff_t);

1122
1123
1124
1125
1126
static ssize_t aio_setup_vectored_rw(struct kiocb *kiocb,
				     int rw, char __user *buf,
				     unsigned long *nr_segs,
				     struct iovec **iovec,
				     bool compat)
1127
1128
1129
{
	ssize_t ret;

1130
	*nr_segs = kiocb->ki_nbytes;
Kent Overstreet's avatar
Kent Overstreet committed
1131

1132
1133
#ifdef CONFIG_COMPAT
	if (compat)
Kent Overstreet's avatar
Kent Overstreet committed
1134
		ret = compat_rw_copy_check_uvector(rw,
1135
1136
				(struct compat_iovec __user *)buf,
				*nr_segs, 1, *iovec, iovec);
1137
1138
	else
#endif
Kent Overstreet's avatar
Kent Overstreet committed
1139
		ret = rw_copy_check_uvector(rw,
1140
1141
				(struct iovec __user *)buf,
				*nr_segs, 1, *iovec, iovec);
1142
	if (ret < 0)
Kent Overstreet's avatar
Kent Overstreet committed
1143
		return ret;
1144

Kent Overstreet's avatar
Kent Overstreet committed
1145
	/* ki_nbytes now reflect bytes instead of segs */
1146
	kiocb->ki_nbytes = ret;
Kent Overstreet's avatar
Kent Overstreet committed
1147
	return 0;
1148
1149
}

1150
1151
1152
1153
static ssize_t aio_setup_single_vector(struct kiocb *kiocb,
				       int rw, char __user *buf,
				       unsigned long *nr_segs,
				       struct iovec *iovec)
1154
{
1155
	if (unlikely(!access_ok(!rw, buf, kiocb->ki_nbytes)))
Kent Overstreet's avatar
Kent Overstreet committed
1156
		return -EFAULT;
1157

1158
1159
1160
	iovec->iov_base = buf;
	iovec->iov_len = kiocb->ki_nbytes;
	*nr_segs = 1;
1161
1162
1163
	return 0;
}

Linus Torvalds's avatar
Linus Torvalds committed
1164
1165
1166
1167
1168
/*
 * aio_setup_iocb:
 *	Performs the initial checks and aio retry method
 *	setup for the kiocb at the time of io submission.
 */
1169
1170
static ssize_t aio_run_iocb(struct kiocb *req, unsigned opcode,
			    char __user *buf, bool compat)
Linus Torvalds's avatar
Linus Torvalds committed
1171
{
Kent Overstreet's avatar
Kent Overstreet committed
1172
1173
	struct file *file = req->ki_filp;
	ssize_t ret;
1174
	unsigned long nr_segs;
Kent Overstreet's avatar
Kent Overstreet committed
1175
1176
1177
	int rw;
	fmode_t mode;
	aio_rw_op *rw_op;
1178
	struct iovec inline_vec, *iovec = &inline_vec;
Linus Torvalds's avatar
Linus Torvalds committed
1179

1180
	switch (opcode) {
Linus Torvalds's avatar
Linus Torvalds committed
1181
	case IOCB_CMD_PREAD:
1182
	case IOCB_CMD_PREADV:
Kent Overstreet's avatar
Kent Overstreet committed
1183
1184
1185
1186
1187
1188
		mode	= FMODE_READ;
		rw	= READ;
		rw_op	= file->f_op->aio_read;
		goto rw_common;

	case IOCB_CMD_PWRITE:
1189
	case IOCB_CMD_PWRITEV:
Kent Overstreet's avatar
Kent Overstreet committed
1190
1191
1192
1193
1194
1195
1196
1197