aio.c 37.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
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
172
173
174
175
176
177
178
179
180
181
182
183
184
185
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
186
187
static void aio_free_ring(struct kioctx *ctx)
{
188
	int i;
Linus Torvalds's avatar
Linus Torvalds committed
189

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

196
197
	put_aio_ring_file(ctx);

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

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

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

237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
	/* 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);
254
255

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

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

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

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

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

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

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

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

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

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

	/* 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.
	 */
340
341
	ctx->nr_pages = get_user_pages(current, mm, ctx->mmap_base, nr_pages,
				       1, 0, ctx->ring_pages, NULL);
342
343
344
345
346

	/* 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.
	 */
347
348
	for (i = 0; i < ctx->nr_pages; i++)
		put_page(ctx->ring_pages[i]);
Linus Torvalds's avatar
Linus Torvalds committed
349

350
351
	up_write(&mm->mmap_sem);

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

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

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

	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)

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

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

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

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

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

Kent Overstreet's avatar
Kent Overstreet committed
415
416
417
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
418
419

	free_percpu(ctx->cpu);
Kent Overstreet's avatar
Kent Overstreet committed
420
421
422
423
424
425
426
427
	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
428
static void free_ioctx(struct work_struct *work)
Kent Overstreet's avatar
Kent Overstreet committed
429
{
Kent Overstreet's avatar
Kent Overstreet committed
430
	struct kioctx *ctx = container_of(work, struct kioctx, free_work);
431
	struct aio_ring *ring;
Kent Overstreet's avatar
Kent Overstreet committed
432
	struct kiocb *req;
433
434
	unsigned cpu, avail;
	DEFINE_WAIT(wait);
Kent Overstreet's avatar
Kent Overstreet committed
435
436
437
438
439
440
441
442

	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);
443
		kiocb_cancel(ctx, req);
Kent Overstreet's avatar
Kent Overstreet committed
444
445
446
447
	}

	spin_unlock_irq(&ctx->ctx_lock);

Kent Overstreet's avatar
Kent Overstreet committed
448
449
450
451
452
453
454
	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;
	}

455
456
	while (1) {
		prepare_to_wait(&ctx->wait, &wait, TASK_UNINTERRUPTIBLE);
457

458
459
460
461
		ring = kmap_atomic(ctx->ring_pages[0]);
		avail = (ring->head <= ring->tail)
			 ? ring->tail - ring->head
			 : ctx->nr_events - ring->head + ring->tail;
462

463
		atomic_add(avail, &ctx->reqs_available);
464
465
466
467
468
469
470
		ring->head = ring->tail;
		kunmap_atomic(ring);

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

		schedule();
471
	}
472
	finish_wait(&ctx->wait, &wait);
473

474
	WARN_ON(atomic_read(&ctx->reqs_available) > ctx->nr_events - 1);
Kent Overstreet's avatar
Kent Overstreet committed
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489

	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
490
static void free_ioctx_ref(struct percpu_ref *ref)
Kent Overstreet's avatar
Kent Overstreet committed
491
{
Kent Overstreet's avatar
Kent Overstreet committed
492
493
494
495
	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
496
497
}

498
499
500
501
502
503
504
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);
505
	rcu_read_lock();
506
	table = rcu_dereference(mm->ioctx_table);
507
508
509
510
511
512
513

	while (1) {
		if (table)
			for (i = 0; i < table->nr; i++)
				if (!table->table[i]) {
					ctx->id = i;
					table->table[i] = ctx;
514
					rcu_read_unlock();
515
516
517
518
519
520
521
522
523
524
					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;

525
		rcu_read_unlock();
526
527
528
529
530
531
532
533
534
535
		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);
536
		rcu_read_lock();
537
		old = rcu_dereference(mm->ioctx_table);
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553

		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
554
555
556
557
558
/* 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
559
	struct mm_struct *mm = current->mm;
Linus Torvalds's avatar
Linus Torvalds committed
560
	struct kioctx *ctx;
561
	int err = -ENOMEM;
Linus Torvalds's avatar
Linus Torvalds committed
562

Kent Overstreet's avatar
Kent Overstreet committed
563
564
565
566
567
568
569
570
571
572
573
574
	/*
	 * 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
575
576
577
578
579
580
581
	/* 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);
	}

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

585
	ctx = kmem_cache_zalloc(kioctx_cachep, GFP_KERNEL);
Linus Torvalds's avatar
Linus Torvalds committed
586
587
588
589
590
	if (!ctx)
		return ERR_PTR(-ENOMEM);

	ctx->max_reqs = nr_events;

Kent Overstreet's avatar
Kent Overstreet committed
591
592
593
	if (percpu_ref_init(&ctx->users, free_ioctx_ref))
		goto out_freectx;

Linus Torvalds's avatar
Linus Torvalds committed
594
	spin_lock_init(&ctx->ctx_lock);
595
	spin_lock_init(&ctx->completion_lock);
596
	mutex_init(&ctx->ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
597
598
599
600
	init_waitqueue_head(&ctx->wait);

	INIT_LIST_HEAD(&ctx->active_reqs);

Kent Overstreet's avatar
Kent Overstreet committed
601
602
	ctx->cpu = alloc_percpu(struct kioctx_cpu);
	if (!ctx->cpu)
Kent Overstreet's avatar
Kent Overstreet committed
603
		goto out_freeref;
Linus Torvalds's avatar
Linus Torvalds committed
604

Kent Overstreet's avatar
Kent Overstreet committed
605
606
607
	if (aio_setup_ring(ctx) < 0)
		goto out_freepcpu;

608
	atomic_set(&ctx->reqs_available, ctx->nr_events - 1);
Kent Overstreet's avatar
Kent Overstreet committed
609
	ctx->req_batch = (ctx->nr_events - 1) / (num_possible_cpus() * 4);
610
611
	if (ctx->req_batch < 1)
		ctx->req_batch = 1;
612

Linus Torvalds's avatar
Linus Torvalds committed
613
	/* limit the number of system wide aios */
614
	spin_lock(&aio_nr_lock);
615
	if (aio_nr + nr_events > (aio_max_nr * 2UL) ||
616
	    aio_nr + nr_events < aio_nr) {
617
		spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
618
		goto out_cleanup;
619
620
	}
	aio_nr += ctx->max_reqs;
621
	spin_unlock(&aio_nr_lock);
Linus Torvalds's avatar
Linus Torvalds committed
622

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

625
626
627
628
	err = ioctx_add_table(ctx, mm);
	if (err)
		goto out_cleanup_put;

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

633
634
out_cleanup_put:
	percpu_ref_put(&ctx->users);
Linus Torvalds's avatar
Linus Torvalds committed
635
out_cleanup:
636
637
	err = -EAGAIN;
	aio_free_ring(ctx);
Kent Overstreet's avatar
Kent Overstreet committed
638
639
out_freepcpu:
	free_percpu(ctx->cpu);
Kent Overstreet's avatar
Kent Overstreet committed
640
641
out_freeref:
	free_percpu(ctx->users.pcpu_count);
Linus Torvalds's avatar
Linus Torvalds committed
642
out_freectx:
643
	put_aio_ring_file(ctx);
Linus Torvalds's avatar
Linus Torvalds committed
644
	kmem_cache_free(kioctx_cachep, ctx);
Kent Overstreet's avatar
Kent Overstreet committed
645
	pr_debug("error allocating ioctx %d\n", err);
646
	return ERR_PTR(err);
Linus Torvalds's avatar
Linus Torvalds committed
647
648
}

Kent Overstreet's avatar
Kent Overstreet committed
649
650
651
652
653
/* 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.
 */
654
static void kill_ioctx(struct mm_struct *mm, struct kioctx *ctx)
Kent Overstreet's avatar
Kent Overstreet committed
655
656
{
	if (!atomic_xchg(&ctx->dead, 1)) {
657
658
659
		struct kioctx_table *table;

		spin_lock(&mm->ioctx_lock);
660
		rcu_read_lock();
661
		table = rcu_dereference(mm->ioctx_table);
662
663
664

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

Kent Overstreet's avatar
Kent Overstreet committed
668
669
		/* percpu_ref_kill() will do the necessary call_rcu() */
		wake_up_all(&ctx->wait);
670

Kent Overstreet's avatar
Kent Overstreet committed
671
		/*
672
673
674
675
676
		 * 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
677
		 */
678
679
680
681
682
683
684
685
		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
686
		percpu_ref_kill(&ctx->users);
Kent Overstreet's avatar
Kent Overstreet committed
687
	}
Linus Torvalds's avatar
Linus Torvalds committed
688
689
690
691
692
}

/* wait_on_sync_kiocb:
 *	Waits on the given sync kiocb to complete.
 */
Kent Overstreet's avatar
Kent Overstreet committed
693
ssize_t wait_on_sync_kiocb(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
694
{
Kent Overstreet's avatar
Kent Overstreet committed
695
	while (!req->ki_ctx) {
Linus Torvalds's avatar
Linus Torvalds committed
696
		set_current_state(TASK_UNINTERRUPTIBLE);
Kent Overstreet's avatar
Kent Overstreet committed
697
		if (req->ki_ctx)
Linus Torvalds's avatar
Linus Torvalds committed
698
			break;
699
		io_schedule();
Linus Torvalds's avatar
Linus Torvalds committed
700
701
	}
	__set_current_state(TASK_RUNNING);
Kent Overstreet's avatar
Kent Overstreet committed
702
	return req->ki_user_data;
Linus Torvalds's avatar
Linus Torvalds committed
703
}
704
EXPORT_SYMBOL(wait_on_sync_kiocb);
Linus Torvalds's avatar
Linus Torvalds committed
705

Kent Overstreet's avatar
Kent Overstreet committed
706
707
708
709
710
711
712
/*
 * 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
713
 */
714
void exit_aio(struct mm_struct *mm)
Linus Torvalds's avatar
Linus Torvalds committed
715
{
716
	struct kioctx_table *table;
717
	struct kioctx *ctx;
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
	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();
737

738
739
740
741
742
743
744
745
		/*
		 * 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.
		 */
746
		ctx->mmap_size = 0;
Kent Overstreet's avatar
Kent Overstreet committed
747

748
		kill_ioctx(mm, ctx);
Linus Torvalds's avatar
Linus Torvalds committed
749
750
751
	}
}

Kent Overstreet's avatar
Kent Overstreet committed
752
753
754
755
756
757
758
759
760
761
762
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
793
794
795
796
797
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
798
/* aio_get_req
Kent Overstreet's avatar
Kent Overstreet committed
799
800
 *	Allocate a slot for an aio request.
 * Returns NULL if no requests are free.
Linus Torvalds's avatar
Linus Torvalds committed
801
 */
Kent Overstreet's avatar
Kent Overstreet committed
802
static inline struct kiocb *aio_get_req(struct kioctx *ctx)
Linus Torvalds's avatar
Linus Torvalds committed
803
{
Kent Overstreet's avatar
Kent Overstreet committed
804
805
	struct kiocb *req;

Kent Overstreet's avatar
Kent Overstreet committed
806
	if (!get_reqs_available(ctx))
Kent Overstreet's avatar
Kent Overstreet committed
807
808
		return NULL;

809
	req = kmem_cache_alloc(kiocb_cachep, GFP_KERNEL|__GFP_ZERO);
Linus Torvalds's avatar
Linus Torvalds committed
810
	if (unlikely(!req))
Kent Overstreet's avatar
Kent Overstreet committed
811
		goto out_put;
Linus Torvalds's avatar
Linus Torvalds committed
812
813

	req->ki_ctx = ctx;
Jeff Moyer's avatar
Jeff Moyer committed
814
	return req;
Kent Overstreet's avatar
Kent Overstreet committed
815
out_put:
Kent Overstreet's avatar
Kent Overstreet committed
816
	put_reqs_available(ctx, 1);
Kent Overstreet's avatar
Kent Overstreet committed
817
	return NULL;
Linus Torvalds's avatar
Linus Torvalds committed
818
819
}

820
static void kiocb_free(struct kiocb *req)
Linus Torvalds's avatar
Linus Torvalds committed
821
{
822
823
	if (req->ki_filp)
		fput(req->ki_filp);
824
825
	if (req->ki_eventfd != NULL)
		eventfd_ctx_put(req->ki_eventfd);
Linus Torvalds's avatar
Linus Torvalds committed
826
827
828
	kmem_cache_free(kiocb_cachep, req);
}

829
static struct kioctx *lookup_ioctx(unsigned long ctx_id)
Linus Torvalds's avatar
Linus Torvalds committed
830
{
831
	struct aio_ring __user *ring  = (void __user *)ctx_id;
832
	struct mm_struct *mm = current->mm;
833
	struct kioctx *ctx, *ret = NULL;
834
835
836
837
838
	struct kioctx_table *table;
	unsigned id;

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

840
	rcu_read_lock();
841
	table = rcu_dereference(mm->ioctx_table);
842

843
844
	if (!table || id >= table->nr)
		goto out;
Linus Torvalds's avatar
Linus Torvalds committed
845

846
	ctx = table->table[id];
847
	if (ctx && ctx->user_id == ctx_id) {
848
849
850
851
		percpu_ref_get(&ctx->users);
		ret = ctx;
	}
out:
852
	rcu_read_unlock();
853
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
854
855
856
857
858
}

/* aio_complete
 *	Called when the io request on the given iocb is complete.
 */
859
void aio_complete(struct kiocb *iocb, long res, long res2)
Linus Torvalds's avatar
Linus Torvalds committed
860
861
862
{
	struct kioctx	*ctx = iocb->ki_ctx;
	struct aio_ring	*ring;
Kent Overstreet's avatar
Kent Overstreet committed
863
	struct io_event	*ev_page, *event;
Linus Torvalds's avatar
Linus Torvalds committed
864
	unsigned long	flags;
Kent Overstreet's avatar
Kent Overstreet committed
865
	unsigned tail, pos;
Linus Torvalds's avatar
Linus Torvalds committed
866

867
868
869
870
871
872
	/*
	 * 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
873
874
875
	 */
	if (is_sync_kiocb(iocb)) {
		iocb->ki_user_data = res;
Kent Overstreet's avatar
Kent Overstreet committed
876
877
		smp_wmb();
		iocb->ki_ctx = ERR_PTR(-EXDEV);
Linus Torvalds's avatar
Linus Torvalds committed
878
		wake_up_process(iocb->ki_obj.tsk);
879
		return;
Linus Torvalds's avatar
Linus Torvalds committed
880
881
	}

Kent Overstreet's avatar
Kent Overstreet committed
882
883
	/*
	 * Take rcu_read_lock() in case the kioctx is being destroyed, as we
884
	 * need to issue a wakeup after incrementing reqs_available.
Linus Torvalds's avatar
Linus Torvalds committed
885
	 */
Kent Overstreet's avatar
Kent Overstreet committed
886
	rcu_read_lock();
Linus Torvalds's avatar
Linus Torvalds committed
887

888
889
890
891
892
893
894
	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);
	}
895

896
897
	/*
	 * Add a completion event to the ring buffer. Must be done holding
898
	 * ctx->completion_lock to prevent other code from messing with the tail
899
900
901
902
	 * pointer since we might be called from irq context.
	 */
	spin_lock_irqsave(&ctx->completion_lock, flags);

903
	tail = ctx->tail;
Kent Overstreet's avatar
Kent Overstreet committed
904
905
	pos = tail + AIO_EVENTS_OFFSET;

906
	if (++tail >= ctx->nr_events)
907
		tail = 0;
Linus Torvalds's avatar
Linus Torvalds committed
908

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

Linus Torvalds's avatar
Linus Torvalds committed
912
913
914
915
916
	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
917
	kunmap_atomic(ev_page);
918
	flush_dcache_page(ctx->ring_pages[pos / AIO_EVENTS_PER_PAGE]);
Kent Overstreet's avatar
Kent Overstreet committed
919
920

	pr_debug("%p[%u]: %p: %p %Lx %lx %lx\n",
Kent Overstreet's avatar
Kent Overstreet committed
921
922
		 ctx, tail, iocb, iocb->ki_obj.user, iocb->ki_user_data,
		 res, res2);
Linus Torvalds's avatar
Linus Torvalds committed
923
924
925
926
927
928

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

929
	ctx->tail = tail;
Linus Torvalds's avatar
Linus Torvalds committed
930

931
	ring = kmap_atomic(ctx->ring_pages[0]);
Kent Overstreet's avatar
Kent Overstreet committed
932
	ring->tail = tail;
933
	kunmap_atomic(ring);
934
	flush_dcache_page(ctx->ring_pages[0]);
Linus Torvalds's avatar
Linus Torvalds committed
935

936
937
	spin_unlock_irqrestore(&ctx->completion_lock, flags);

Kent Overstreet's avatar
Kent Overstreet committed
938
	pr_debug("added to ring %p at [%u]\n", iocb, tail);
Davide Libenzi's avatar
Davide Libenzi committed
939
940
941
942
943
944

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

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

951
952
953
954
955
956
957
958
	/*
	 * 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
959
960
961
	if (waitqueue_active(&ctx->wait))
		wake_up(&ctx->wait);

Kent Overstreet's avatar
Kent Overstreet committed
962
	rcu_read_unlock();
Linus Torvalds's avatar
Linus Torvalds committed
963
}
964
EXPORT_SYMBOL(aio_complete);
Linus Torvalds's avatar
Linus Torvalds committed
965

966
967
968
/* 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
969
 */
970
971
static long aio_read_events_ring(struct kioctx *ctx,
				 struct io_event __user *event, long nr)
Linus Torvalds's avatar
Linus Torvalds committed
972
973
{
	struct aio_ring *ring;
974
	unsigned head, tail, pos;
975
976
977
	long ret = 0;
	int copy_ret;

978
	mutex_lock(&ctx->ring_lock);
Linus Torvalds's avatar
Linus Torvalds committed
979

980
	ring = kmap_atomic(ctx->ring_pages[0]);
981
	head = ring->head;
982
	tail = ring->tail;
983
984
	kunmap_atomic(ring);

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

987
	if (head == tail)
Linus Torvalds's avatar
Linus Torvalds committed
988
989
		goto out;

990
991
992
993
994
	while (ret < nr) {
		long avail;
		struct io_event *ev;
		struct page *page;

995
996
		avail = (head <= tail ?  tail : ctx->nr_events) - head;
		if (head == tail)
997
998
999
1000
1001
1002
1003
			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;
1004
		page = ctx->ring_pages[pos / AIO_EVENTS_PER_PAGE];
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
		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;
1019
		head %= ctx->nr_events;
Linus Torvalds's avatar
Linus Torvalds committed
1020
1021
	}

1022
	ring = kmap_atomic(ctx->ring_pages[0]);
1023
	ring->head = head;
1024
	kunmap_atomic(ring);
1025
	flush_dcache_page(ctx->ring_pages[0]);
1026

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

Kent Overstreet's avatar
Kent Overstreet committed
1029
	put_reqs_available(ctx, ret);
1030
out:
1031
	mutex_unlock(&ctx->ring_lock);
1032

Linus Torvalds's avatar
Linus Torvalds committed
1033
1034
1035
	return ret;
}

1036
1037
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
1038
{
1039
	long ret = aio_read_events_ring(ctx, event + *i, nr - *i);
Linus Torvalds's avatar
Linus Torvalds committed
1040

1041
1042
	if (ret > 0)
		*i += ret;
Linus Torvalds's avatar
Linus Torvalds committed
1043

1044
1045
	if (unlikely(atomic_read(&ctx->dead)))
		ret = -EINVAL;
Linus Torvalds's avatar
Linus Torvalds committed
1046

1047
1048
	if (!*i)
		*i = ret;
Linus Torvalds's avatar
Linus Torvalds committed
1049

1050
	return ret < 0 || *i >= min_nr;
Linus Torvalds's avatar
Linus Torvalds committed
1051
1052
}

1053
static long read_events(struct kioctx *ctx, long min_nr, long nr,
Linus Torvalds's avatar
Linus Torvalds committed
1054
1055
1056
			struct io_event __user *event,
			struct timespec __user *timeout)
{
1057
1058
	ktime_t until = { .tv64 = KTIME_MAX };
	long ret = 0;
Linus Torvalds's avatar
Linus Torvalds committed
1059
1060
1061

	if (timeout) {
		struct timespec	ts;
1062

Linus Torvalds's avatar
Linus Torvalds committed
1063
		if (unlikely(copy_from_user(&ts, timeout, sizeof(ts))))
1064
			return -EFAULT;
Linus Torvalds's avatar
Linus Torvalds committed
1065

1066
		until = timespec_to_ktime(ts);
Linus Torvalds's avatar
Linus Torvalds committed
1067
1068
	}

1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
	/*
	 * 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
1085

1086
1087
	if (!ret && signal_pending(current))
		ret = -EINTR;
Linus Torvalds's avatar
Linus Torvalds committed
1088

1089
	return ret;
Linus Torvalds's avatar
Linus Torvalds committed
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
}

/* 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.
 */
1105
SYSCALL_DEFINE2(io_setup, unsigned, nr_events, aio_context_t __user *, ctxp)
Linus Torvalds's avatar
Linus Torvalds committed
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
{
	struct kioctx *ioctx = NULL;
	unsigned long ctx;
	long ret;

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

	ret = -EINVAL;
1116
1117
1118
	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
1119
1120
1121
1122
1123
1124
1125
		goto out;
	}

	ioctx = ioctx_alloc(nr_events);
	ret = PTR_ERR(ioctx);
	if (!IS_ERR(ioctx)) {
		ret = put_user(ioctx->user_id, ctxp);
1126
		if (ret)
1127
			kill_ioctx(current->mm, ioctx);
Kent Overstreet's avatar
Kent Overstreet committed
1128
		percpu_ref_put(&ioctx->users);
Linus Torvalds's avatar
Linus Torvalds committed
1129
1130
1131
1132
1133
1134
1135
1136
1137
	}

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
1138
 *	implemented.  May fail with -EINVAL if the context pointed to
Linus Torvalds's avatar
Linus Torvalds committed
1139
1140
 *	is invalid.
 */
1141
SYSCALL_DEFINE1(io_destroy, aio_context_t, ctx)
Linus Torvalds's avatar
Linus Torvalds committed
1142
1143
1144
{
	struct kioctx *ioctx = lookup_ioctx(ctx);
	if (likely(NULL != ioctx)) {
1145
		kill_ioctx(current->mm, ioctx);
Kent Overstreet's avatar
Kent Overstreet committed
1146
		percpu_ref_put(&ioctx->users);
Linus Torvalds's avatar
Linus Torvalds committed
1147
1148
1149
1150
1151
1152
		return 0;
	}
	pr_debug("EINVAL: io_destroy: invalid context id\n");
	return -EINVAL;
}

Kent Overstreet's avatar
Kent Overstreet committed
1153
1154
1155
typedef ssize_t (aio_rw_op)(struct kiocb *, const struct iovec *,
			    unsigned long, loff_t);

1156
1157
1158
1159
1160
static ssize_t aio_setup_vectored_rw(struct kiocb *kiocb,
				     int rw, char __user *buf,
				     unsigned long *nr_segs,
				     struct iovec **iovec,
				     bool compat)
1161
1162
1163
{
	ssize_t ret;

1164
	*nr_segs = kiocb->ki_nbytes;
Kent Overstreet's avatar
Kent Overstreet committed
1165

1166
1167
#ifdef CONFIG_COMPAT
	if (compat)
Kent Overstreet's avatar
Kent Overstreet committed
1168
		ret = compat_rw_copy_check_uvector(rw,
1169
1170
				(struct compat_iovec __user *)buf,
				*nr_segs, 1, *iovec, iovec);
1171
1172
	else
#endif
Kent Overstreet's avatar
Kent Overstreet committed
1173
		ret = rw_copy_check_uvector(rw,
1174
1175
				(struct iovec __user *)buf,
				*nr_segs, 1, *iovec, iovec);
1176
	if (ret < 0)
Kent Overstreet's avatar
Kent Overstreet committed
1177
		return ret;
1178

Kent Overstreet's avatar
Kent Overstreet committed