aio.c 36.3 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

Kent Overstreet's avatar
Kent Overstreet committed
83
84
	struct percpu_ref	reqs;

85
86
	unsigned long		user_id;

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

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

103
104
	/* Size of ringbuffer, in units of struct io_event */
	unsigned		nr_events;
105

106
107
108
109
110
111
	unsigned long		mmap_base;
	unsigned long		mmap_size;

	struct page		**ring_pages;
	long			nr_pages;

Kent Overstreet's avatar
Kent Overstreet committed
112
	struct work_struct	free_work;
113
114

	struct {
115
116
117
118
119
		/*
		 * 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
120
121
		 *
		 * We batch accesses to it with a percpu version.
122
123
		 */
		atomic_t	reqs_available;
124
125
126
127
128
129
130
	} ____cacheline_aligned_in_smp;

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

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

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

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

	unsigned		id;
145
146
};

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

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

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

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

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

171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
static void put_aio_ring_file(struct kioctx *ctx)
{
	struct file *aio_ring_file = ctx->aio_ring_file;
	if (aio_ring_file) {
		truncate_setsize(aio_ring_file->f_inode, 0);

		/* Prevent further access to the kioctx from migratepages */
		spin_lock(&aio_ring_file->f_inode->i_mapping->private_lock);
		aio_ring_file->f_inode->i_mapping->private_data = NULL;
		ctx->aio_ring_file = NULL;
		spin_unlock(&aio_ring_file->f_inode->i_mapping->private_lock);

		fput(aio_ring_file);
	}
}

Linus Torvalds's avatar
Linus Torvalds committed
187
188
static void aio_free_ring(struct kioctx *ctx)
{
189
	int i;
Linus Torvalds's avatar
Linus Torvalds committed
190

191
192
193
	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]));
194
		put_page(ctx->ring_pages[i]);
195
	}
Linus Torvalds's avatar
Linus Torvalds committed
196

197
198
	put_aio_ring_file(ctx);

199
200
	if (ctx->ring_pages && ctx->ring_pages != ctx->internal_pages)
		kfree(ctx->ring_pages);
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
}

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

218
#if IS_ENABLED(CONFIG_MIGRATION)
219
220
221
static int aio_migratepage(struct address_space *mapping, struct page *new,
			struct page *old, enum migrate_mode mode)
{
222
	struct kioctx *ctx;
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
	unsigned long flags;
	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);

238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
	/* We can potentially race against kioctx teardown here.  Use the
	 * address_space's private data lock to protect the mapping's
	 * private_data.
	 */
	spin_lock(&mapping->private_lock);
	ctx = mapping->private_data;
	if (ctx) {
		pgoff_t idx;
		spin_lock_irqsave(&ctx->completion_lock, flags);
		migrate_page_copy(new, old);
		idx = old->index;
		if (idx < (pgoff_t)ctx->nr_pages)
			ctx->ring_pages[idx] = new;
		spin_unlock_irqrestore(&ctx->completion_lock, flags);
	} else
		rc = -EBUSY;
	spin_unlock(&mapping->private_lock);
255
256

	return rc;
Linus Torvalds's avatar
Linus Torvalds committed
257
}
258
#endif
Linus Torvalds's avatar
Linus Torvalds committed
259

260
261
static const struct address_space_operations aio_ctx_aops = {
	.set_page_dirty = aio_set_page_dirty,
262
#if IS_ENABLED(CONFIG_MIGRATION)
263
	.migratepage	= aio_migratepage,
264
#endif
265
266
};

Linus Torvalds's avatar
Linus Torvalds committed
267
268
269
270
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
271
	struct mm_struct *mm = current->mm;
272
	unsigned long size, populate;
Linus Torvalds's avatar
Linus Torvalds committed
273
	int nr_pages;
274
275
	int i;
	struct file *file;
Linus Torvalds's avatar
Linus Torvalds committed
276
277
278
279
280
281
282

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

283
	nr_pages = PFN_UP(size);
Linus Torvalds's avatar
Linus Torvalds committed
284
285
286
	if (nr_pages < 0)
		return -EINVAL;

287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
	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
312

313
	ctx->ring_pages = ctx->internal_pages;
Linus Torvalds's avatar
Linus Torvalds committed
314
	if (nr_pages > AIO_RING_PAGES) {
315
316
317
		ctx->ring_pages = kcalloc(nr_pages, sizeof(struct page *),
					  GFP_KERNEL);
		if (!ctx->ring_pages)
Linus Torvalds's avatar
Linus Torvalds committed
318
319
320
			return -ENOMEM;
	}

321
322
	ctx->mmap_size = nr_pages * PAGE_SIZE;
	pr_debug("attempting mmap of %lu bytes\n", ctx->mmap_size);
323

Zach Brown's avatar
Zach Brown committed
324
	down_write(&mm->mmap_sem);
325
326
327
	ctx->mmap_base = do_mmap_pgoff(ctx->aio_ring_file, 0, ctx->mmap_size,
				       PROT_READ | PROT_WRITE,
				       MAP_SHARED | MAP_POPULATE, 0, &populate);
328
	if (IS_ERR((void *)ctx->mmap_base)) {
Zach Brown's avatar
Zach Brown committed
329
		up_write(&mm->mmap_sem);
330
		ctx->mmap_size = 0;
Linus Torvalds's avatar
Linus Torvalds committed
331
332
333
334
		aio_free_ring(ctx);
		return -EAGAIN;
	}

335
	pr_debug("mmap address: 0x%08lx\n", ctx->mmap_base);
336
337
338
339
340

	/* We must do this while still holding mmap_sem for write, as we
	 * need to be protected against userspace attempting to mremap()
	 * or munmap() the ring buffer.
	 */
341
342
	ctx->nr_pages = get_user_pages(current, mm, ctx->mmap_base, nr_pages,
				       1, 0, ctx->ring_pages, NULL);
343
344
345
346
347

	/* Dropping the reference here is safe as the page cache will hold
	 * onto the pages for us.  It is also required so that page migration
	 * can unmap the pages and get the right reference count.
	 */
348
349
	for (i = 0; i < ctx->nr_pages; i++)
		put_page(ctx->ring_pages[i]);
Linus Torvalds's avatar
Linus Torvalds committed
350

351
352
	up_write(&mm->mmap_sem);

353
	if (unlikely(ctx->nr_pages != nr_pages)) {
Linus Torvalds's avatar
Linus Torvalds committed
354
355
356
357
		aio_free_ring(ctx);
		return -EAGAIN;
	}

358
359
	ctx->user_id = ctx->mmap_base;
	ctx->nr_events = nr_events; /* trusted copy */
Linus Torvalds's avatar
Linus Torvalds committed
360

361
	ring = kmap_atomic(ctx->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
362
	ring->nr = nr_events;	/* user copy */
363
	ring->id = ~0U;
Linus Torvalds's avatar
Linus Torvalds committed
364
365
366
367
368
	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);
369
	kunmap_atomic(ring);
370
	flush_dcache_page(ctx->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
371
372
373
374
375
376
377
378

	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)

379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
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);

395
static int kiocb_cancel(struct kioctx *ctx, struct kiocb *kiocb)
Kent Overstreet's avatar
Kent Overstreet committed
396
{
397
	kiocb_cancel_fn *old, *cancel;
Kent Overstreet's avatar
Kent Overstreet committed
398

399
400
401
402
403
404
405
406
	/*
	 * 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
407
			return -EINVAL;
Kent Overstreet's avatar
Kent Overstreet committed
408

409
410
411
		old = cancel;
		cancel = cmpxchg(&kiocb->ki_cancel, old, KIOCB_CANCELLED);
	} while (cancel != old);
Kent Overstreet's avatar
Kent Overstreet committed
412

Kent Overstreet's avatar
Kent Overstreet committed
413
	return cancel(kiocb);
Kent Overstreet's avatar
Kent Overstreet committed
414
415
}

Kent Overstreet's avatar
Kent Overstreet committed
416
static void free_ioctx(struct work_struct *work)
Kent Overstreet's avatar
Kent Overstreet committed
417
{
Kent Overstreet's avatar
Kent Overstreet committed
418
419
420
	struct kioctx *ctx = container_of(work, struct kioctx, free_work);

	pr_debug("freeing %p\n", ctx);
Kent Overstreet's avatar
Kent Overstreet committed
421

Kent Overstreet's avatar
Kent Overstreet committed
422
	aio_free_ring(ctx);
Kent Overstreet's avatar
Kent Overstreet committed
423
	free_percpu(ctx->cpu);
Kent Overstreet's avatar
Kent Overstreet committed
424
425
426
	kmem_cache_free(kioctx_cachep, ctx);
}

Kent Overstreet's avatar
Kent Overstreet committed
427
428
429
430
431
432
433
434
static void free_ioctx_reqs(struct percpu_ref *ref)
{
	struct kioctx *ctx = container_of(ref, struct kioctx, reqs);

	INIT_WORK(&ctx->free_work, free_ioctx);
	schedule_work(&ctx->free_work);
}

Kent Overstreet's avatar
Kent Overstreet committed
435
436
437
438
439
/*
 * 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
440
static void free_ioctx_users(struct percpu_ref *ref)
Kent Overstreet's avatar
Kent Overstreet committed
441
{
Kent Overstreet's avatar
Kent Overstreet committed
442
	struct kioctx *ctx = container_of(ref, struct kioctx, users);
Kent Overstreet's avatar
Kent Overstreet committed
443
444
445
446
447
448
449
450
451
	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);
452
		kiocb_cancel(ctx, req);
Kent Overstreet's avatar
Kent Overstreet committed
453
454
455
456
	}

	spin_unlock_irq(&ctx->ctx_lock);

Kent Overstreet's avatar
Kent Overstreet committed
457
458
	percpu_ref_kill(&ctx->reqs);
	percpu_ref_put(&ctx->reqs);
Kent Overstreet's avatar
Kent Overstreet committed
459
460
}

461
462
463
464
465
466
467
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);
468
	rcu_read_lock();
469
	table = rcu_dereference(mm->ioctx_table);
470
471
472
473
474
475
476

	while (1) {
		if (table)
			for (i = 0; i < table->nr; i++)
				if (!table->table[i]) {
					ctx->id = i;
					table->table[i] = ctx;
477
					rcu_read_unlock();
478
479
480
481
482
483
484
485
486
487
					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;

488
		rcu_read_unlock();
489
490
491
492
493
494
495
496
497
498
		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);
499
		rcu_read_lock();
500
		old = rcu_dereference(mm->ioctx_table);
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516

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

Kent Overstreet's avatar
Kent Overstreet committed
517
518
519
520
521
522
523
524
525
526
static void aio_nr_sub(unsigned nr)
{
	spin_lock(&aio_nr_lock);
	if (WARN_ON(aio_nr - nr > aio_nr))
		aio_nr = 0;
	else
		aio_nr -= nr;
	spin_unlock(&aio_nr_lock);
}

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

Kent Overstreet's avatar
Kent Overstreet committed
536
537
538
539
540
541
542
543
544
545
546
547
	/*
	 * 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
548
549
550
551
552
553
554
	/* 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);
	}

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

558
	ctx = kmem_cache_zalloc(kioctx_cachep, GFP_KERNEL);
Linus Torvalds's avatar
Linus Torvalds committed
559
560
561
562
563
	if (!ctx)
		return ERR_PTR(-ENOMEM);

	ctx->max_reqs = nr_events;

Kent Overstreet's avatar
Kent Overstreet committed
564
565
566
567
568
	if (percpu_ref_init(&ctx->users, free_ioctx_users))
		goto err;

	if (percpu_ref_init(&ctx->reqs, free_ioctx_reqs))
		goto err;
Kent Overstreet's avatar
Kent Overstreet committed
569

Linus Torvalds's avatar
Linus Torvalds committed
570
	spin_lock_init(&ctx->ctx_lock);
571
	spin_lock_init(&ctx->completion_lock);
572
	mutex_init(&ctx->ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
573
574
575
576
	init_waitqueue_head(&ctx->wait);

	INIT_LIST_HEAD(&ctx->active_reqs);

Kent Overstreet's avatar
Kent Overstreet committed
577
578
	ctx->cpu = alloc_percpu(struct kioctx_cpu);
	if (!ctx->cpu)
Kent Overstreet's avatar
Kent Overstreet committed
579
		goto err;
Linus Torvalds's avatar
Linus Torvalds committed
580

Kent Overstreet's avatar
Kent Overstreet committed
581
	if (aio_setup_ring(ctx) < 0)
Kent Overstreet's avatar
Kent Overstreet committed
582
		goto err;
Kent Overstreet's avatar
Kent Overstreet committed
583

584
	atomic_set(&ctx->reqs_available, ctx->nr_events - 1);
Kent Overstreet's avatar
Kent Overstreet committed
585
	ctx->req_batch = (ctx->nr_events - 1) / (num_possible_cpus() * 4);
586
587
	if (ctx->req_batch < 1)
		ctx->req_batch = 1;
588

Linus Torvalds's avatar
Linus Torvalds committed
589
	/* limit the number of system wide aios */
590
	spin_lock(&aio_nr_lock);
591
	if (aio_nr + nr_events > (aio_max_nr * 2UL) ||
592
	    aio_nr + nr_events < aio_nr) {
593
		spin_unlock(&aio_nr_lock);
Kent Overstreet's avatar
Kent Overstreet committed
594
595
		err = -EAGAIN;
		goto err;
596
597
	}
	aio_nr += ctx->max_reqs;
598
	spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
599

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

602
603
	err = ioctx_add_table(ctx, mm);
	if (err)
Kent Overstreet's avatar
Kent Overstreet committed
604
		goto err_cleanup;
605

Kent Overstreet's avatar
Kent Overstreet committed
606
	pr_debug("allocated ioctx %p[%ld]: mm=%p mask=0x%x\n",
607
		 ctx, ctx->user_id, mm, ctx->nr_events);
Linus Torvalds's avatar
Linus Torvalds committed
608
609
	return ctx;

Kent Overstreet's avatar
Kent Overstreet committed
610
611
612
err_cleanup:
	aio_nr_sub(ctx->max_reqs);
err:
Kent Overstreet's avatar
Kent Overstreet committed
613
	free_percpu(ctx->cpu);
Kent Overstreet's avatar
Kent Overstreet committed
614
	free_percpu(ctx->reqs.pcpu_count);
Kent Overstreet's avatar
Kent Overstreet committed
615
	free_percpu(ctx->users.pcpu_count);
Linus Torvalds's avatar
Linus Torvalds committed
616
	kmem_cache_free(kioctx_cachep, ctx);
Kent Overstreet's avatar
Kent Overstreet committed
617
	pr_debug("error allocating ioctx %d\n", err);
618
	return ERR_PTR(err);
Linus Torvalds's avatar
Linus Torvalds committed
619
620
}

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

		spin_lock(&mm->ioctx_lock);
632
		rcu_read_lock();
633
		table = rcu_dereference(mm->ioctx_table);
634
635
636

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

Kent Overstreet's avatar
Kent Overstreet committed
640
641
		/* percpu_ref_kill() will do the necessary call_rcu() */
		wake_up_all(&ctx->wait);
642

Kent Overstreet's avatar
Kent Overstreet committed
643
		/*
644
645
646
647
648
		 * 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
649
		 */
Kent Overstreet's avatar
Kent Overstreet committed
650
		aio_nr_sub(ctx->max_reqs);
651
652
653
654

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

Kent Overstreet's avatar
Kent Overstreet committed
655
		percpu_ref_kill(&ctx->users);
Kent Overstreet's avatar
Kent Overstreet committed
656
	}
Linus Torvalds's avatar
Linus Torvalds committed
657
658
659
660
661
}

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

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

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

717
		kill_ioctx(mm, ctx);
Linus Torvalds's avatar
Linus Torvalds committed
718
719
720
	}
}

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

Kent Overstreet's avatar
Kent Overstreet committed
775
	if (!get_reqs_available(ctx))
Kent Overstreet's avatar
Kent Overstreet committed
776
777
		return NULL;

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

Kent Overstreet's avatar
Kent Overstreet committed
782
783
	percpu_ref_get(&ctx->reqs);

Linus Torvalds's avatar
Linus Torvalds committed
784
	req->ki_ctx = ctx;
Jeff Moyer's avatar
Jeff Moyer committed
785
	return req;
Kent Overstreet's avatar
Kent Overstreet committed
786
out_put:
Kent Overstreet's avatar
Kent Overstreet committed
787
	put_reqs_available(ctx, 1);
Kent Overstreet's avatar
Kent Overstreet committed
788
	return NULL;
Linus Torvalds's avatar
Linus Torvalds committed
789
790
}

791
static void kiocb_free(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
792
{
793
794
	if (req->ki_filp)
		fput(req->ki_filp);
795
796
	if (req->ki_eventfd != NULL)
		eventfd_ctx_put(req->ki_eventfd);
Linus Torvalds's avatar
Linus Torvalds committed
797
798
799
	kmem_cache_free(kiocb_cachep, req);
}

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

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

811
	rcu_read_lock();
812
	table = rcu_dereference(mm->ioctx_table);
813

814
815
	if (!table || id >= table->nr)
		goto out;
Linus Torvalds's avatar
Linus Torvalds committed
816

817
	ctx = table->table[id];
818
	if (ctx && ctx->user_id == ctx_id) {
819
820
821
822
		percpu_ref_get(&ctx->users);
		ret = ctx;
	}
out:
823
	rcu_read_unlock();
824
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
825
826
827
828
829
}

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

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

853
854
855
856
857
858
859
	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);
	}
860

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Kent Overstreet's avatar
Kent Overstreet committed
927
	percpu_ref_put(&ctx->reqs);
Linus Torvalds's avatar
Linus Torvalds committed
928
}
929
EXPORT_SYMBOL(aio_complete);
Linus Torvalds's avatar
Linus Torvalds committed
930

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	if (timeout) {
		struct timespec	ts;
1027

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Linus Torvalds's avatar
Linus Torvalds committed
1163
1164
1165
1166
1167
/*
 * aio_setup_iocb:
 *	Performs the initial checks and aio retry method
 *	setup for the kiocb at the time of io submission.
 */
<