futex.c 49.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10
/*
 *  Fast Userspace Mutexes (which I call "Futexes!").
 *  (C) Rusty Russell, IBM 2002
 *
 *  Generalized futexes, futex requeueing, misc fixes by Ingo Molnar
 *  (C) Copyright 2003 Red Hat Inc, All Rights Reserved
 *
 *  Removed page pinning, fix privately mapped COW pages and other cleanups
 *  (C) Copyright 2003, 2004 Jamie Lokier
 *
11 12 13 14
 *  Robust futex support started by Ingo Molnar
 *  (C) Copyright 2006 Red Hat Inc, All Rights Reserved
 *  Thanks to Thomas Gleixner for suggestions, analysis and fixes.
 *
15 16 17 18
 *  PI-futex support started by Ingo Molnar and Thomas Gleixner
 *  Copyright (C) 2006 Red Hat, Inc., Ingo Molnar <mingo@redhat.com>
 *  Copyright (C) 2006 Timesys Corp., Thomas Gleixner <tglx@timesys.com>
 *
E
Eric Dumazet 已提交
19 20 21
 *  PRIVATE futexes by Eric Dumazet
 *  Copyright (C) 2007 Eric Dumazet <dada1@cosmosbay.com>
 *
L
Linus Torvalds 已提交
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
 *  Thanks to Ben LaHaise for yelling "hashed waitqueues" loudly
 *  enough at me, Linus for the original (flawed) idea, Matthew
 *  Kirkwood for proof-of-concept implementation.
 *
 *  "The futexes are also cursed."
 *  "But they come in a choice of three flavours!"
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
#include <linux/slab.h>
#include <linux/poll.h>
#include <linux/fs.h>
#include <linux/file.h>
#include <linux/jhash.h>
#include <linux/init.h>
#include <linux/futex.h>
#include <linux/mount.h>
#include <linux/pagemap.h>
#include <linux/syscalls.h>
53
#include <linux/signal.h>
54
#include <linux/module.h>
55
#include <linux/magic.h>
56 57 58
#include <linux/pid.h>
#include <linux/nsproxy.h>

59
#include <asm/futex.h>
L
Linus Torvalds 已提交
60

61 62
#include "rtmutex_common.h"

63 64
int __read_mostly futex_cmpxchg_enabled;

L
Linus Torvalds 已提交
65 66
#define FUTEX_HASHBITS (CONFIG_BASE_SMALL ? 4 : 8)

67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
/*
 * Priority Inheritance state:
 */
struct futex_pi_state {
	/*
	 * list of 'owned' pi_state instances - these have to be
	 * cleaned up in do_exit() if the task exits prematurely:
	 */
	struct list_head list;

	/*
	 * The PI object:
	 */
	struct rt_mutex pi_mutex;

	struct task_struct *owner;
	atomic_t refcount;

	union futex_key key;
};

L
Linus Torvalds 已提交
88 89 90 91 92
/*
 * We use this hashed waitqueue instead of a normal wait_queue_t, so
 * we can wake only the relevant ones (hashed queues may be shared).
 *
 * A futex_q has a woken state, just like tasks have TASK_RUNNING.
P
Pierre Peiffer 已提交
93
 * It is considered woken when plist_node_empty(&q->list) || q->lock_ptr == 0.
L
Linus Torvalds 已提交
94
 * The order of wakup is always to make the first condition true, then
95
 * wake up q->waiter, then make the second condition true.
L
Linus Torvalds 已提交
96 97
 */
struct futex_q {
P
Pierre Peiffer 已提交
98
	struct plist_node list;
99 100
	/* There can only be a single waiter */
	wait_queue_head_t waiter;
L
Linus Torvalds 已提交
101

102
	/* Which hash list lock to use: */
L
Linus Torvalds 已提交
103 104
	spinlock_t *lock_ptr;

105
	/* Key which the futex is hashed on: */
L
Linus Torvalds 已提交
106 107
	union futex_key key;

108 109 110
	/* Optional priority inheritance state: */
	struct futex_pi_state *pi_state;
	struct task_struct *task;
111 112 113

	/* Bitset for the optional bitmasked wakeup */
	u32 bitset;
L
Linus Torvalds 已提交
114 115 116
};

/*
D
Darren Hart 已提交
117 118 119
 * Hash buckets are shared by all the futex_keys that hash to the same
 * location.  Each key may have multiple futex_q structures, one for each task
 * waiting on a futex.
L
Linus Torvalds 已提交
120 121
 */
struct futex_hash_bucket {
P
Pierre Peiffer 已提交
122 123
	spinlock_t lock;
	struct plist_head chain;
L
Linus Torvalds 已提交
124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
};

static struct futex_hash_bucket futex_queues[1<<FUTEX_HASHBITS];

/*
 * We hash on the keys returned from get_futex_key (see below).
 */
static struct futex_hash_bucket *hash_futex(union futex_key *key)
{
	u32 hash = jhash2((u32*)&key->both.word,
			  (sizeof(key->both.word)+sizeof(key->both.ptr))/4,
			  key->both.offset);
	return &futex_queues[hash & ((1 << FUTEX_HASHBITS)-1)];
}

/*
 * Return 1 if two futex_keys are equal, 0 otherwise.
 */
static inline int match_futex(union futex_key *key1, union futex_key *key2)
{
	return (key1->both.word == key2->both.word
		&& key1->both.ptr == key2->both.ptr
		&& key1->both.offset == key2->both.offset);
}

149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
/*
 * Take a reference to the resource addressed by a key.
 * Can be called while holding spinlocks.
 *
 */
static void get_futex_key_refs(union futex_key *key)
{
	if (!key->both.ptr)
		return;

	switch (key->both.offset & (FUT_OFF_INODE|FUT_OFF_MMSHARED)) {
	case FUT_OFF_INODE:
		atomic_inc(&key->shared.inode->i_count);
		break;
	case FUT_OFF_MMSHARED:
		atomic_inc(&key->private.mm->mm_count);
		break;
	}
}

/*
 * Drop a reference to the resource addressed by a key.
 * The hash bucket spinlock must not be held.
 */
static void drop_futex_key_refs(union futex_key *key)
{
175 176 177
	if (!key->both.ptr) {
		/* If we're here then we tried to put a key we failed to get */
		WARN_ON_ONCE(1);
178
		return;
179
	}
180 181 182 183 184 185 186 187 188 189 190

	switch (key->both.offset & (FUT_OFF_INODE|FUT_OFF_MMSHARED)) {
	case FUT_OFF_INODE:
		iput(key->shared.inode);
		break;
	case FUT_OFF_MMSHARED:
		mmdrop(key->private.mm);
		break;
	}
}

E
Eric Dumazet 已提交
191 192 193
/**
 * get_futex_key - Get parameters which are the keys for a futex.
 * @uaddr: virtual address of the futex
D
Darren Hart 已提交
194
 * @fshared: 0 for a PROCESS_PRIVATE futex, 1 for PROCESS_SHARED
E
Eric Dumazet 已提交
195 196 197 198
 * @key: address where result is stored.
 *
 * Returns a negative error code or 0
 * The key words are stored in *key on success.
L
Linus Torvalds 已提交
199
 *
200
 * For shared mappings, it's (page->index, vma->vm_file->f_path.dentry->d_inode,
L
Linus Torvalds 已提交
201 202 203
 * offset_within_page).  For private mappings, it's (uaddr, current->mm).
 * We can usually work out the index without swapping in the page.
 *
D
Darren Hart 已提交
204
 * lock_page() might sleep, the caller should not hold a spinlock.
L
Linus Torvalds 已提交
205
 */
P
Peter Zijlstra 已提交
206
static int get_futex_key(u32 __user *uaddr, int fshared, union futex_key *key)
L
Linus Torvalds 已提交
207
{
208
	unsigned long address = (unsigned long)uaddr;
L
Linus Torvalds 已提交
209 210 211 212 213 214 215
	struct mm_struct *mm = current->mm;
	struct page *page;
	int err;

	/*
	 * The futex address must be "naturally" aligned.
	 */
216
	key->both.offset = address % PAGE_SIZE;
E
Eric Dumazet 已提交
217
	if (unlikely((address % sizeof(u32)) != 0))
L
Linus Torvalds 已提交
218
		return -EINVAL;
219
	address -= key->both.offset;
L
Linus Torvalds 已提交
220

E
Eric Dumazet 已提交
221 222 223 224 225 226 227 228 229 230 231 232
	/*
	 * PROCESS_PRIVATE futexes are fast.
	 * As the mm cannot disappear under us and the 'key' only needs
	 * virtual address, we dont even have to find the underlying vma.
	 * Note : We do have to check 'uaddr' is a valid user address,
	 *        but access_ok() should be faster than find_vma()
	 */
	if (!fshared) {
		if (unlikely(!access_ok(VERIFY_WRITE, uaddr, sizeof(u32))))
			return -EFAULT;
		key->private.mm = mm;
		key->private.address = address;
233
		get_futex_key_refs(key);
E
Eric Dumazet 已提交
234 235
		return 0;
	}
L
Linus Torvalds 已提交
236

237
again:
P
Peter Zijlstra 已提交
238
	err = get_user_pages_fast(address, 1, 0, &page);
239 240 241 242 243 244 245 246 247
	if (err < 0)
		return err;

	lock_page(page);
	if (!page->mapping) {
		unlock_page(page);
		put_page(page);
		goto again;
	}
L
Linus Torvalds 已提交
248 249 250 251 252 253

	/*
	 * Private mappings are handled in a simple way.
	 *
	 * NOTE: When userspace waits on a MAP_SHARED mapping, even if
	 * it's a read-only handle, it's expected that futexes attach to
254
	 * the object not the particular process.
L
Linus Torvalds 已提交
255
	 */
256 257
	if (PageAnon(page)) {
		key->both.offset |= FUT_OFF_MMSHARED; /* ref taken on mm */
L
Linus Torvalds 已提交
258
		key->private.mm = mm;
259
		key->private.address = address;
260 261 262 263
	} else {
		key->both.offset |= FUT_OFF_INODE; /* inode-based key */
		key->shared.inode = page->mapping->host;
		key->shared.pgoff = page->index;
L
Linus Torvalds 已提交
264 265
	}

266
	get_futex_key_refs(key);
L
Linus Torvalds 已提交
267

268 269 270
	unlock_page(page);
	put_page(page);
	return 0;
L
Linus Torvalds 已提交
271 272
}

273
static inline
P
Peter Zijlstra 已提交
274
void put_futex_key(int fshared, union futex_key *key)
L
Linus Torvalds 已提交
275
{
276
	drop_futex_key_refs(key);
L
Linus Torvalds 已提交
277 278
}

279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297
/**
 * futex_top_waiter() - Return the highest priority waiter on a futex
 * @hb:     the hash bucket the futex_q's reside in
 * @key:    the futex key (to distinguish it from other futex futex_q's)
 *
 * Must be called with the hb lock held.
 */
static struct futex_q *futex_top_waiter(struct futex_hash_bucket *hb,
					union futex_key *key)
{
	struct futex_q *this;

	plist_for_each_entry(this, &hb->chain, list) {
		if (match_futex(&this->key, key))
			return this;
	}
	return NULL;
}

T
Thomas Gleixner 已提交
298 299 300 301 302 303 304 305 306 307 308 309
static u32 cmpxchg_futex_value_locked(u32 __user *uaddr, u32 uval, u32 newval)
{
	u32 curval;

	pagefault_disable();
	curval = futex_atomic_cmpxchg_inatomic(uaddr, uval, newval);
	pagefault_enable();

	return curval;
}

static int get_futex_value_locked(u32 *dest, u32 __user *from)
L
Linus Torvalds 已提交
310 311 312
{
	int ret;

313
	pagefault_disable();
314
	ret = __copy_from_user_inatomic(dest, from, sizeof(u32));
315
	pagefault_enable();
L
Linus Torvalds 已提交
316 317 318 319

	return ret ? -EFAULT : 0;
}

320 321 322 323 324 325 326 327 328 329 330

/*
 * PI code:
 */
static int refill_pi_state_cache(void)
{
	struct futex_pi_state *pi_state;

	if (likely(current->pi_state_cache))
		return 0;

331
	pi_state = kzalloc(sizeof(*pi_state), GFP_KERNEL);
332 333 334 335 336 337 338 339

	if (!pi_state)
		return -ENOMEM;

	INIT_LIST_HEAD(&pi_state->list);
	/* pi_mutex gets initialized later */
	pi_state->owner = NULL;
	atomic_set(&pi_state->refcount, 1);
340
	pi_state->key = FUTEX_KEY_INIT;
341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394

	current->pi_state_cache = pi_state;

	return 0;
}

static struct futex_pi_state * alloc_pi_state(void)
{
	struct futex_pi_state *pi_state = current->pi_state_cache;

	WARN_ON(!pi_state);
	current->pi_state_cache = NULL;

	return pi_state;
}

static void free_pi_state(struct futex_pi_state *pi_state)
{
	if (!atomic_dec_and_test(&pi_state->refcount))
		return;

	/*
	 * If pi_state->owner is NULL, the owner is most probably dying
	 * and has cleaned up the pi_state already
	 */
	if (pi_state->owner) {
		spin_lock_irq(&pi_state->owner->pi_lock);
		list_del_init(&pi_state->list);
		spin_unlock_irq(&pi_state->owner->pi_lock);

		rt_mutex_proxy_unlock(&pi_state->pi_mutex, pi_state->owner);
	}

	if (current->pi_state_cache)
		kfree(pi_state);
	else {
		/*
		 * pi_state->list is already empty.
		 * clear pi_state->owner.
		 * refcount is at 0 - put it back to 1.
		 */
		pi_state->owner = NULL;
		atomic_set(&pi_state->refcount, 1);
		current->pi_state_cache = pi_state;
	}
}

/*
 * Look up the task based on what TID userspace gave us.
 * We dont trust it.
 */
static struct task_struct * futex_find_get_task(pid_t pid)
{
	struct task_struct *p;
395
	const struct cred *cred = current_cred(), *pcred;
396

397
	rcu_read_lock();
398
	p = find_task_by_vpid(pid);
399
	if (!p) {
400
		p = ERR_PTR(-ESRCH);
401 402 403 404 405 406 407 408
	} else {
		pcred = __task_cred(p);
		if (cred->euid != pcred->euid &&
		    cred->euid != pcred->uid)
			p = ERR_PTR(-ESRCH);
		else
			get_task_struct(p);
	}
409

410
	rcu_read_unlock();
411 412 413 414 415 416 417 418 419 420 421 422 423

	return p;
}

/*
 * This task is holding PI mutexes at exit time => bad.
 * Kernel cleans up PI-state, but userspace is likely hosed.
 * (Robust-futex cleanup is separate and might save the day for userspace.)
 */
void exit_pi_state_list(struct task_struct *curr)
{
	struct list_head *next, *head = &curr->pi_state_list;
	struct futex_pi_state *pi_state;
424
	struct futex_hash_bucket *hb;
425
	union futex_key key = FUTEX_KEY_INIT;
426

427 428
	if (!futex_cmpxchg_enabled)
		return;
429 430 431
	/*
	 * We are a ZOMBIE and nobody can enqueue itself on
	 * pi_state_list anymore, but we have to be careful
432
	 * versus waiters unqueueing themselves:
433 434 435 436 437 438 439
	 */
	spin_lock_irq(&curr->pi_lock);
	while (!list_empty(head)) {

		next = head->next;
		pi_state = list_entry(next, struct futex_pi_state, list);
		key = pi_state->key;
440
		hb = hash_futex(&key);
441 442 443 444 445
		spin_unlock_irq(&curr->pi_lock);

		spin_lock(&hb->lock);

		spin_lock_irq(&curr->pi_lock);
446 447 448 449
		/*
		 * We dropped the pi-lock, so re-check whether this
		 * task still owns the PI-state:
		 */
450 451 452 453 454 455
		if (head->next != next) {
			spin_unlock(&hb->lock);
			continue;
		}

		WARN_ON(pi_state->owner != curr);
456 457
		WARN_ON(list_empty(&pi_state->list));
		list_del_init(&pi_state->list);
458 459 460 461 462 463 464 465 466 467 468 469 470
		pi_state->owner = NULL;
		spin_unlock_irq(&curr->pi_lock);

		rt_mutex_unlock(&pi_state->pi_mutex);

		spin_unlock(&hb->lock);

		spin_lock_irq(&curr->pi_lock);
	}
	spin_unlock_irq(&curr->pi_lock);
}

static int
P
Pierre Peiffer 已提交
471 472
lookup_pi_state(u32 uval, struct futex_hash_bucket *hb,
		union futex_key *key, struct futex_pi_state **ps)
473 474 475
{
	struct futex_pi_state *pi_state = NULL;
	struct futex_q *this, *next;
P
Pierre Peiffer 已提交
476
	struct plist_head *head;
477
	struct task_struct *p;
478
	pid_t pid = uval & FUTEX_TID_MASK;
479 480 481

	head = &hb->chain;

P
Pierre Peiffer 已提交
482
	plist_for_each_entry_safe(this, next, head, list) {
P
Pierre Peiffer 已提交
483
		if (match_futex(&this->key, key)) {
484 485 486 487 488
			/*
			 * Another waiter already exists - bump up
			 * the refcount and return its pi_state:
			 */
			pi_state = this->pi_state;
489 490 491 492 493 494
			/*
			 * Userspace might have messed up non PI and PI futexes
			 */
			if (unlikely(!pi_state))
				return -EINVAL;

495
			WARN_ON(!atomic_read(&pi_state->refcount));
496 497
			WARN_ON(pid && pi_state->owner &&
				pi_state->owner->pid != pid);
498

499
			atomic_inc(&pi_state->refcount);
P
Pierre Peiffer 已提交
500
			*ps = pi_state;
501 502 503 504 505 506

			return 0;
		}
	}

	/*
507
	 * We are the first waiter - try to look up the real owner and attach
508
	 * the new pi_state to it, but bail out when TID = 0
509
	 */
510
	if (!pid)
511
		return -ESRCH;
512
	p = futex_find_get_task(pid);
513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534
	if (IS_ERR(p))
		return PTR_ERR(p);

	/*
	 * We need to look at the task state flags to figure out,
	 * whether the task is exiting. To protect against the do_exit
	 * change of the task flags, we do this protected by
	 * p->pi_lock:
	 */
	spin_lock_irq(&p->pi_lock);
	if (unlikely(p->flags & PF_EXITING)) {
		/*
		 * The task is on the way out. When PF_EXITPIDONE is
		 * set, we know that the task has finished the
		 * cleanup:
		 */
		int ret = (p->flags & PF_EXITPIDONE) ? -ESRCH : -EAGAIN;

		spin_unlock_irq(&p->pi_lock);
		put_task_struct(p);
		return ret;
	}
535 536 537 538 539 540 541 542 543 544

	pi_state = alloc_pi_state();

	/*
	 * Initialize the pi_mutex in locked state and make 'p'
	 * the owner of it:
	 */
	rt_mutex_init_proxy_locked(&pi_state->pi_mutex, p);

	/* Store the key for possible exit cleanups: */
P
Pierre Peiffer 已提交
545
	pi_state->key = *key;
546

547
	WARN_ON(!list_empty(&pi_state->list));
548 549 550 551 552 553
	list_add(&pi_state->list, &p->pi_state_list);
	pi_state->owner = p;
	spin_unlock_irq(&p->pi_lock);

	put_task_struct(p);

P
Pierre Peiffer 已提交
554
	*ps = pi_state;
555 556 557 558

	return 0;
}

559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679
/**
 * futex_lock_pi_atomic() - atomic work required to acquire a pi aware futex
 * @uaddr:	the pi futex user address
 * @hb:		the pi futex hash bucket
 * @key:	the futex key associated with uaddr and hb
 * @ps:		the pi_state pointer where we store the result of the lookup
 * @task:	the task to perform the atomic lock work for.  This will be
 * 		"current" except in the case of requeue pi.
 *
 * Returns:
 *  0 - ready to wait
 *  1 - acquired the lock
 * <0 - error
 *
 * The hb->lock and futex_key refs shall be held by the caller.
 */
static int futex_lock_pi_atomic(u32 __user *uaddr, struct futex_hash_bucket *hb,
				union futex_key *key,
				struct futex_pi_state **ps,
				struct task_struct *task)
{
	int lock_taken, ret, ownerdied = 0;
	u32 uval, newval, curval;

retry:
	ret = lock_taken = 0;

	/*
	 * To avoid races, we attempt to take the lock here again
	 * (by doing a 0 -> TID atomic cmpxchg), while holding all
	 * the locks. It will most likely not succeed.
	 */
	newval = task_pid_vnr(task);

	curval = cmpxchg_futex_value_locked(uaddr, 0, newval);

	if (unlikely(curval == -EFAULT))
		return -EFAULT;

	/*
	 * Detect deadlocks.
	 */
	if ((unlikely((curval & FUTEX_TID_MASK) == task_pid_vnr(task))))
		return -EDEADLK;

	/*
	 * Surprise - we got the lock. Just return to userspace:
	 */
	if (unlikely(!curval))
		return 1;

	uval = curval;

	/*
	 * Set the FUTEX_WAITERS flag, so the owner will know it has someone
	 * to wake at the next unlock.
	 */
	newval = curval | FUTEX_WAITERS;

	/*
	 * There are two cases, where a futex might have no owner (the
	 * owner TID is 0): OWNER_DIED. We take over the futex in this
	 * case. We also do an unconditional take over, when the owner
	 * of the futex died.
	 *
	 * This is safe as we are protected by the hash bucket lock !
	 */
	if (unlikely(ownerdied || !(curval & FUTEX_TID_MASK))) {
		/* Keep the OWNER_DIED bit */
		newval = (curval & ~FUTEX_TID_MASK) | task_pid_vnr(task);
		ownerdied = 0;
		lock_taken = 1;
	}

	curval = cmpxchg_futex_value_locked(uaddr, uval, newval);

	if (unlikely(curval == -EFAULT))
		return -EFAULT;
	if (unlikely(curval != uval))
		goto retry;

	/*
	 * We took the lock due to owner died take over.
	 */
	if (unlikely(lock_taken))
		return 1;

	/*
	 * We dont have the lock. Look up the PI state (or create it if
	 * we are the first waiter):
	 */
	ret = lookup_pi_state(uval, hb, key, ps);

	if (unlikely(ret)) {
		switch (ret) {
		case -ESRCH:
			/*
			 * No owner found for this futex. Check if the
			 * OWNER_DIED bit is set to figure out whether
			 * this is a robust futex or not.
			 */
			if (get_futex_value_locked(&curval, uaddr))
				return -EFAULT;

			/*
			 * We simply start over in case of a robust
			 * futex. The code above will take the futex
			 * and return happy.
			 */
			if (curval & FUTEX_OWNER_DIED) {
				ownerdied = 1;
				goto retry;
			}
		default:
			break;
		}
	}

	return ret;
}

L
Linus Torvalds 已提交
680 681 682 683 684 685
/*
 * The hash bucket lock must be held when this is called.
 * Afterwards, the futex_q must not be accessed.
 */
static void wake_futex(struct futex_q *q)
{
P
Pierre Peiffer 已提交
686
	plist_del(&q->list, &q->list.plist);
L
Linus Torvalds 已提交
687 688
	/*
	 * The lock in wake_up_all() is a crucial memory barrier after the
P
Pierre Peiffer 已提交
689
	 * plist_del() and also before assigning to q->lock_ptr.
L
Linus Torvalds 已提交
690
	 */
691
	wake_up(&q->waiter);
L
Linus Torvalds 已提交
692 693 694
	/*
	 * The waiting task can free the futex_q as soon as this is written,
	 * without taking any locks.  This must come last.
695
	 *
D
Darren Hart 已提交
696 697 698
	 * A memory barrier is required here to prevent the following store to
	 * lock_ptr from getting ahead of the wakeup. Clearing the lock at the
	 * end of wake_up() does not prevent this store from moving.
L
Linus Torvalds 已提交
699
	 */
700
	smp_wmb();
L
Linus Torvalds 已提交
701 702 703
	q->lock_ptr = NULL;
}

704 705 706 707 708 709 710 711 712
static int wake_futex_pi(u32 __user *uaddr, u32 uval, struct futex_q *this)
{
	struct task_struct *new_owner;
	struct futex_pi_state *pi_state = this->pi_state;
	u32 curval, newval;

	if (!pi_state)
		return -EINVAL;

713
	spin_lock(&pi_state->pi_mutex.wait_lock);
714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729
	new_owner = rt_mutex_next_owner(&pi_state->pi_mutex);

	/*
	 * This happens when we have stolen the lock and the original
	 * pending owner did not enqueue itself back on the rt_mutex.
	 * Thats not a tragedy. We know that way, that a lock waiter
	 * is on the fly. We make the futex_q waiter the pending owner.
	 */
	if (!new_owner)
		new_owner = this->task;

	/*
	 * We pass it to the next owner. (The WAITERS bit is always
	 * kept enabled while there is PI state around. We must also
	 * preserve the owner died bit.)
	 */
730
	if (!(uval & FUTEX_OWNER_DIED)) {
731 732
		int ret = 0;

733
		newval = FUTEX_WAITERS | task_pid_vnr(new_owner);
734

T
Thomas Gleixner 已提交
735
		curval = cmpxchg_futex_value_locked(uaddr, uval, newval);
736

737
		if (curval == -EFAULT)
738
			ret = -EFAULT;
739
		else if (curval != uval)
740 741 742 743 744
			ret = -EINVAL;
		if (ret) {
			spin_unlock(&pi_state->pi_mutex.wait_lock);
			return ret;
		}
745
	}
746

747 748 749 750 751 752 753
	spin_lock_irq(&pi_state->owner->pi_lock);
	WARN_ON(list_empty(&pi_state->list));
	list_del_init(&pi_state->list);
	spin_unlock_irq(&pi_state->owner->pi_lock);

	spin_lock_irq(&new_owner->pi_lock);
	WARN_ON(!list_empty(&pi_state->list));
754 755
	list_add(&pi_state->list, &new_owner->pi_state_list);
	pi_state->owner = new_owner;
756 757
	spin_unlock_irq(&new_owner->pi_lock);

758
	spin_unlock(&pi_state->pi_mutex.wait_lock);
759 760 761 762 763 764 765 766 767 768 769 770 771
	rt_mutex_unlock(&pi_state->pi_mutex);

	return 0;
}

static int unlock_futex_pi(u32 __user *uaddr, u32 uval)
{
	u32 oldval;

	/*
	 * There is no waiter, so we unlock the futex. The owner died
	 * bit has not to be preserved here. We are the owner:
	 */
T
Thomas Gleixner 已提交
772
	oldval = cmpxchg_futex_value_locked(uaddr, uval, 0);
773 774 775 776 777 778 779 780 781

	if (oldval == -EFAULT)
		return oldval;
	if (oldval != uval)
		return -EAGAIN;

	return 0;
}

I
Ingo Molnar 已提交
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797
/*
 * Express the locking dependencies for lockdep:
 */
static inline void
double_lock_hb(struct futex_hash_bucket *hb1, struct futex_hash_bucket *hb2)
{
	if (hb1 <= hb2) {
		spin_lock(&hb1->lock);
		if (hb1 < hb2)
			spin_lock_nested(&hb2->lock, SINGLE_DEPTH_NESTING);
	} else { /* hb1 > hb2 */
		spin_lock(&hb2->lock);
		spin_lock_nested(&hb1->lock, SINGLE_DEPTH_NESTING);
	}
}

D
Darren Hart 已提交
798 799 800
static inline void
double_unlock_hb(struct futex_hash_bucket *hb1, struct futex_hash_bucket *hb2)
{
801
	spin_unlock(&hb1->lock);
802 803
	if (hb1 != hb2)
		spin_unlock(&hb2->lock);
D
Darren Hart 已提交
804 805
}

L
Linus Torvalds 已提交
806
/*
D
Darren Hart 已提交
807
 * Wake up waiters matching bitset queued on this futex (uaddr).
L
Linus Torvalds 已提交
808
 */
P
Peter Zijlstra 已提交
809
static int futex_wake(u32 __user *uaddr, int fshared, int nr_wake, u32 bitset)
L
Linus Torvalds 已提交
810
{
811
	struct futex_hash_bucket *hb;
L
Linus Torvalds 已提交
812
	struct futex_q *this, *next;
P
Pierre Peiffer 已提交
813
	struct plist_head *head;
814
	union futex_key key = FUTEX_KEY_INIT;
L
Linus Torvalds 已提交
815 816
	int ret;

817 818 819
	if (!bitset)
		return -EINVAL;

E
Eric Dumazet 已提交
820
	ret = get_futex_key(uaddr, fshared, &key);
L
Linus Torvalds 已提交
821 822 823
	if (unlikely(ret != 0))
		goto out;

824 825 826
	hb = hash_futex(&key);
	spin_lock(&hb->lock);
	head = &hb->chain;
L
Linus Torvalds 已提交
827

P
Pierre Peiffer 已提交
828
	plist_for_each_entry_safe(this, next, head, list) {
L
Linus Torvalds 已提交
829
		if (match_futex (&this->key, &key)) {
830 831 832 833
			if (this->pi_state) {
				ret = -EINVAL;
				break;
			}
834 835 836 837 838

			/* Check if one of the bits is set in both bitsets */
			if (!(this->bitset & bitset))
				continue;

L
Linus Torvalds 已提交
839 840 841 842 843 844
			wake_futex(this);
			if (++ret >= nr_wake)
				break;
		}
	}

845
	spin_unlock(&hb->lock);
846
	put_futex_key(fshared, &key);
847
out:
L
Linus Torvalds 已提交
848 849 850
	return ret;
}

851 852 853 854
/*
 * Wake up all waiters hashed on the physical page that is mapped
 * to this virtual address:
 */
855
static int
P
Peter Zijlstra 已提交
856
futex_wake_op(u32 __user *uaddr1, int fshared, u32 __user *uaddr2,
857
	      int nr_wake, int nr_wake2, int op)
858
{
859
	union futex_key key1 = FUTEX_KEY_INIT, key2 = FUTEX_KEY_INIT;
860
	struct futex_hash_bucket *hb1, *hb2;
P
Pierre Peiffer 已提交
861
	struct plist_head *head;
862
	struct futex_q *this, *next;
D
Darren Hart 已提交
863
	int ret, op_ret;
864

D
Darren Hart 已提交
865
retry:
E
Eric Dumazet 已提交
866
	ret = get_futex_key(uaddr1, fshared, &key1);
867 868
	if (unlikely(ret != 0))
		goto out;
E
Eric Dumazet 已提交
869
	ret = get_futex_key(uaddr2, fshared, &key2);
870
	if (unlikely(ret != 0))
871
		goto out_put_key1;
872

873 874
	hb1 = hash_futex(&key1);
	hb2 = hash_futex(&key2);
875

I
Ingo Molnar 已提交
876
	double_lock_hb(hb1, hb2);
D
Darren Hart 已提交
877
retry_private:
878
	op_ret = futex_atomic_op_inuser(op, uaddr2);
879
	if (unlikely(op_ret < 0)) {
880
		u32 dummy;
881

D
Darren Hart 已提交
882
		double_unlock_hb(hb1, hb2);
883

884
#ifndef CONFIG_MMU
885 886 887 888
		/*
		 * we don't get EFAULT from MMU faults if we don't have an MMU,
		 * but we might get them from range checking
		 */
889
		ret = op_ret;
890
		goto out_put_keys;
891 892
#endif

893 894
		if (unlikely(op_ret != -EFAULT)) {
			ret = op_ret;
895
			goto out_put_keys;
896 897
		}

898
		ret = get_user(dummy, uaddr2);
899
		if (ret)
900
			goto out_put_keys;
901

D
Darren Hart 已提交
902 903 904
		if (!fshared)
			goto retry_private;

905 906
		put_futex_key(fshared, &key2);
		put_futex_key(fshared, &key1);
D
Darren Hart 已提交
907
		goto retry;
908 909
	}

910
	head = &hb1->chain;
911

P
Pierre Peiffer 已提交
912
	plist_for_each_entry_safe(this, next, head, list) {
913 914 915 916 917 918 919 920
		if (match_futex (&this->key, &key1)) {
			wake_futex(this);
			if (++ret >= nr_wake)
				break;
		}
	}

	if (op_ret > 0) {
921
		head = &hb2->chain;
922 923

		op_ret = 0;
P
Pierre Peiffer 已提交
924
		plist_for_each_entry_safe(this, next, head, list) {
925 926 927 928 929 930 931 932 933
			if (match_futex (&this->key, &key2)) {
				wake_futex(this);
				if (++op_ret >= nr_wake2)
					break;
			}
		}
		ret += op_ret;
	}

D
Darren Hart 已提交
934
	double_unlock_hb(hb1, hb2);
935
out_put_keys:
936
	put_futex_key(fshared, &key2);
937
out_put_key1:
938
	put_futex_key(fshared, &key1);
939
out:
940 941 942
	return ret;
}

L
Linus Torvalds 已提交
943 944 945 946
/*
 * Requeue all waiters hashed on one physical page to another
 * physical page.
 */
P
Peter Zijlstra 已提交
947
static int futex_requeue(u32 __user *uaddr1, int fshared, u32 __user *uaddr2,
948
			 int nr_wake, int nr_requeue, u32 *cmpval)
L
Linus Torvalds 已提交
949
{
950
	union futex_key key1 = FUTEX_KEY_INIT, key2 = FUTEX_KEY_INIT;
951
	struct futex_hash_bucket *hb1, *hb2;
P
Pierre Peiffer 已提交
952
	struct plist_head *head1;
L
Linus Torvalds 已提交
953 954 955
	struct futex_q *this, *next;
	int ret, drop_count = 0;

956
retry:
E
Eric Dumazet 已提交
957
	ret = get_futex_key(uaddr1, fshared, &key1);
L
Linus Torvalds 已提交
958 959
	if (unlikely(ret != 0))
		goto out;
E
Eric Dumazet 已提交
960
	ret = get_futex_key(uaddr2, fshared, &key2);
L
Linus Torvalds 已提交
961
	if (unlikely(ret != 0))
962
		goto out_put_key1;
L
Linus Torvalds 已提交
963

964 965
	hb1 = hash_futex(&key1);
	hb2 = hash_futex(&key2);
L
Linus Torvalds 已提交
966

D
Darren Hart 已提交
967
retry_private:
I
Ingo Molnar 已提交
968
	double_lock_hb(hb1, hb2);
L
Linus Torvalds 已提交
969

970 971
	if (likely(cmpval != NULL)) {
		u32 curval;
L
Linus Torvalds 已提交
972

973
		ret = get_futex_value_locked(&curval, uaddr1);
L
Linus Torvalds 已提交
974 975

		if (unlikely(ret)) {
D
Darren Hart 已提交
976
			double_unlock_hb(hb1, hb2);
L
Linus Torvalds 已提交
977

978
			ret = get_user(curval, uaddr1);
D
Darren Hart 已提交
979 980
			if (ret)
				goto out_put_keys;
L
Linus Torvalds 已提交
981

D
Darren Hart 已提交
982 983
			if (!fshared)
				goto retry_private;
L
Linus Torvalds 已提交
984

D
Darren Hart 已提交
985 986 987
			put_futex_key(fshared, &key2);
			put_futex_key(fshared, &key1);
			goto retry;
L
Linus Torvalds 已提交
988
		}
989
		if (curval != *cmpval) {
L
Linus Torvalds 已提交
990 991 992 993 994
			ret = -EAGAIN;
			goto out_unlock;
		}
	}

995
	head1 = &hb1->chain;
P
Pierre Peiffer 已提交
996
	plist_for_each_entry_safe(this, next, head1, list) {
L
Linus Torvalds 已提交
997 998 999 1000 1001
		if (!match_futex (&this->key, &key1))
			continue;
		if (++ret <= nr_wake) {
			wake_futex(this);
		} else {
1002 1003 1004 1005 1006
			/*
			 * If key1 and key2 hash to the same bucket, no need to
			 * requeue.
			 */
			if (likely(head1 != &hb2->chain)) {
P
Pierre Peiffer 已提交
1007 1008
				plist_del(&this->list, &hb1->chain);
				plist_add(&this->list, &hb2->chain);
1009
				this->lock_ptr = &hb2->lock;
P
Pierre Peiffer 已提交
1010 1011 1012
#ifdef CONFIG_DEBUG_PI_LIST
				this->list.plist.lock = &hb2->lock;
#endif
1013
			}
L
Linus Torvalds 已提交
1014
			this->key = key2;
1015
			get_futex_key_refs(&key2);
L
Linus Torvalds 已提交
1016 1017 1018 1019 1020 1021 1022 1023
			drop_count++;

			if (ret - nr_wake >= nr_requeue)
				break;
		}
	}

out_unlock:
D
Darren Hart 已提交
1024
	double_unlock_hb(hb1, hb2);
L
Linus Torvalds 已提交
1025

1026
	/* drop_futex_key_refs() must be called outside the spinlocks. */
L
Linus Torvalds 已提交
1027
	while (--drop_count >= 0)
1028
		drop_futex_key_refs(&key1);
L
Linus Torvalds 已提交
1029

1030
out_put_keys:
1031
	put_futex_key(fshared, &key2);
1032
out_put_key1:
1033
	put_futex_key(fshared, &key1);
1034
out:
L
Linus Torvalds 已提交
1035 1036 1037 1038
	return ret;
}

/* The key must be already stored in q->key. */
E
Eric Sesterhenn 已提交
1039
static inline struct futex_hash_bucket *queue_lock(struct futex_q *q)
L
Linus Torvalds 已提交
1040
{
1041
	struct futex_hash_bucket *hb;
L
Linus Torvalds 已提交
1042

1043
	init_waitqueue_head(&q->waiter);
L
Linus Torvalds 已提交
1044

1045
	get_futex_key_refs(&q->key);
1046 1047
	hb = hash_futex(&q->key);
	q->lock_ptr = &hb->lock;
L
Linus Torvalds 已提交
1048

1049 1050
	spin_lock(&hb->lock);
	return hb;
L
Linus Torvalds 已提交
1051 1052
}

E
Eric Sesterhenn 已提交
1053
static inline void queue_me(struct futex_q *q, struct futex_hash_bucket *hb)
L
Linus Torvalds 已提交
1054
{
P
Pierre Peiffer 已提交
1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071
	int prio;

	/*
	 * The priority used to register this element is
	 * - either the real thread-priority for the real-time threads
	 * (i.e. threads with a priority lower than MAX_RT_PRIO)
	 * - or MAX_RT_PRIO for non-RT threads.
	 * Thus, all RT-threads are woken first in priority order, and
	 * the others are woken last, in FIFO order.
	 */
	prio = min(current->normal_prio, MAX_RT_PRIO);

	plist_node_init(&q->list, prio);
#ifdef CONFIG_DEBUG_PI_LIST
	q->list.plist.lock = &hb->lock;
#endif
	plist_add(&q->list, &hb->chain);
1072
	q->task = current;
1073
	spin_unlock(&hb->lock);
L
Linus Torvalds 已提交
1074 1075 1076
}

static inline void
1077
queue_unlock(struct futex_q *q, struct futex_hash_bucket *hb)
L
Linus Torvalds 已提交
1078
{
1079
	spin_unlock(&hb->lock);
1080
	drop_futex_key_refs(&q->key);
L
Linus Torvalds 已提交
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091
}

/*
 * queue_me and unqueue_me must be called as a pair, each
 * exactly once.  They are called with the hashed spinlock held.
 */

/* Return 1 if we were still queued (ie. 0 means we were woken) */
static int unqueue_me(struct futex_q *q)
{
	spinlock_t *lock_ptr;
1092
	int ret = 0;
L
Linus Torvalds 已提交
1093 1094

	/* In the common case we don't take the spinlock, which is nice. */
1095
retry:
L
Linus Torvalds 已提交
1096
	lock_ptr = q->lock_ptr;
1097
	barrier();
1098
	if (lock_ptr != NULL) {
L
Linus Torvalds 已提交
1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
		spin_lock(lock_ptr);
		/*
		 * q->lock_ptr can change between reading it and
		 * spin_lock(), causing us to take the wrong lock.  This
		 * corrects the race condition.
		 *
		 * Reasoning goes like this: if we have the wrong lock,
		 * q->lock_ptr must have changed (maybe several times)
		 * between reading it and the spin_lock().  It can
		 * change again after the spin_lock() but only if it was
		 * already changed before the spin_lock().  It cannot,
		 * however, change back to the original value.  Therefore
		 * we can detect whether we acquired the correct lock.
		 */
		if (unlikely(lock_ptr != q->lock_ptr)) {
			spin_unlock(lock_ptr);
			goto retry;
		}
P
Pierre Peiffer 已提交
1117 1118
		WARN_ON(plist_node_empty(&q->list));
		plist_del(&q->list, &q->list.plist);
1119 1120 1121

		BUG_ON(q->pi_state);

L
Linus Torvalds 已提交
1122 1123 1124 1125
		spin_unlock(lock_ptr);
		ret = 1;
	}

1126
	drop_futex_key_refs(&q->key);
L
Linus Torvalds 已提交
1127 1128 1129
	return ret;
}

1130 1131
/*
 * PI futexes can not be requeued and must remove themself from the
P
Pierre Peiffer 已提交
1132 1133
 * hash bucket. The hash bucket lock (i.e. lock_ptr) is held on entry
 * and dropped here.
1134
 */
P
Pierre Peiffer 已提交
1135
static void unqueue_me_pi(struct futex_q *q)
1136
{
P
Pierre Peiffer 已提交
1137 1138
	WARN_ON(plist_node_empty(&q->list));
	plist_del(&q->list, &q->list.plist);
1139 1140 1141 1142 1143

	BUG_ON(!q->pi_state);
	free_pi_state(q->pi_state);
	q->pi_state = NULL;

P
Pierre Peiffer 已提交
1144
	spin_unlock(q->lock_ptr);
1145

1146
	drop_futex_key_refs(&q->key);
1147 1148
}

P
Pierre Peiffer 已提交
1149
/*
1150
 * Fixup the pi_state owner with the new owner.
P
Pierre Peiffer 已提交
1151
 *
1152 1153
 * Must be called with hash bucket lock held and mm->sem held for non
 * private futexes.
P
Pierre Peiffer 已提交
1154
 */
1155
static int fixup_pi_state_owner(u32 __user *uaddr, struct futex_q *q,
P
Peter Zijlstra 已提交
1156
				struct task_struct *newowner, int fshared)
P
Pierre Peiffer 已提交
1157
{
1158
	u32 newtid = task_pid_vnr(newowner) | FUTEX_WAITERS;
P
Pierre Peiffer 已提交
1159
	struct futex_pi_state *pi_state = q->pi_state;
1160
	struct task_struct *oldowner = pi_state->owner;
P
Pierre Peiffer 已提交
1161
	u32 uval, curval, newval;
D
Darren Hart 已提交
1162
	int ret;
P
Pierre Peiffer 已提交
1163 1164

	/* Owner died? */
1165 1166 1167 1168 1169 1170 1171 1172 1173 1174
	if (!pi_state->owner)
		newtid |= FUTEX_OWNER_DIED;

	/*
	 * We are here either because we stole the rtmutex from the
	 * pending owner or we are the pending owner which failed to
	 * get the rtmutex. We have to replace the pending owner TID
	 * in the user space variable. This must be atomic as we have
	 * to preserve the owner died bit here.
	 *
D
Darren Hart 已提交
1175 1176 1177
	 * Note: We write the user space value _before_ changing the pi_state
	 * because we can fault here. Imagine swapped out pages or a fork
	 * that marked all the anonymous memory readonly for cow.
1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204
	 *
	 * Modifying pi_state _before_ the user space value would
	 * leave the pi_state in an inconsistent state when we fault
	 * here, because we need to drop the hash bucket lock to
	 * handle the fault. This might be observed in the PID check
	 * in lookup_pi_state.
	 */
retry:
	if (get_futex_value_locked(&uval, uaddr))
		goto handle_fault;

	while (1) {
		newval = (uval & FUTEX_OWNER_DIED) | newtid;

		curval = cmpxchg_futex_value_locked(uaddr, uval, newval);

		if (curval == -EFAULT)
			goto handle_fault;
		if (curval == uval)
			break;
		uval = curval;
	}

	/*
	 * We fixed up user space. Now we need to fix the pi_state
	 * itself.
	 */
P
Pierre Peiffer 已提交
1205 1206 1207 1208 1209
	if (pi_state->owner != NULL) {
		spin_lock_irq(&pi_state->owner->pi_lock);
		WARN_ON(list_empty(&pi_state->list));
		list_del_init(&pi_state->list);
		spin_unlock_irq(&pi_state->owner->pi_lock);
1210
	}
P
Pierre Peiffer 已提交
1211

1212
	pi_state->owner = newowner;
P
Pierre Peiffer 已提交
1213

1214
	spin_lock_irq(&newowner->pi_lock);
P
Pierre Peiffer 已提交
1215
	WARN_ON(!list_empty(&pi_state->list));
1216 1217
	list_add(&pi_state->list, &newowner->pi_state_list);
	spin_unlock_irq(&newowner->pi_lock);
1218
	return 0;
P
Pierre Peiffer 已提交
1219 1220

	/*
1221 1222 1223 1224 1225 1226 1227 1228
	 * To handle the page fault we need to drop the hash bucket
	 * lock here. That gives the other task (either the pending
	 * owner itself or the task which stole the rtmutex) the
	 * chance to try the fixup of the pi_state. So once we are
	 * back from handling the fault we need to check the pi_state
	 * after reacquiring the hash bucket lock and before trying to
	 * do another fixup. When the fixup has been done already we
	 * simply return.
P
Pierre Peiffer 已提交
1229
	 */
1230 1231
handle_fault:
	spin_unlock(q->lock_ptr);
1232

D
Darren Hart 已提交
1233
	ret = get_user(uval, uaddr);
1234

1235
	spin_lock(q->lock_ptr);
1236

1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
	/*
	 * Check if someone else fixed it for us:
	 */
	if (pi_state->owner != oldowner)
		return 0;

	if (ret)
		return ret;

	goto retry;
P
Pierre Peiffer 已提交
1247 1248
}

E
Eric Dumazet 已提交
1249 1250
/*
 * In case we must use restart_block to restart a futex_wait,
1251
 * we encode in the 'flags' shared capability
E
Eric Dumazet 已提交
1252
 */
1253 1254
#define FLAGS_SHARED		0x01
#define FLAGS_CLOCKRT		0x02
E
Eric Dumazet 已提交
1255

N
Nick Piggin 已提交
1256
static long futex_wait_restart(struct restart_block *restart);
T
Thomas Gleixner 已提交
1257

1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311
/**
 * futex_wait_queue_me() - queue_me() and wait for wakeup, timeout, or signal
 * @hb:		the futex hash bucket, must be locked by the caller
 * @q:		the futex_q to queue up on
 * @timeout:	the prepared hrtimer_sleeper, or null for no timeout
 * @wait:	the wait_queue to add to the futex_q after queueing in the hb
 */
static void futex_wait_queue_me(struct futex_hash_bucket *hb, struct futex_q *q,
				struct hrtimer_sleeper *timeout,
				wait_queue_t *wait)
{
	queue_me(q, hb);

	/*
	 * There might have been scheduling since the queue_me(), as we
	 * cannot hold a spinlock across the get_user() in case it
	 * faults, and we cannot just set TASK_INTERRUPTIBLE state when
	 * queueing ourselves into the futex hash.  This code thus has to
	 * rely on the futex_wake() code removing us from hash when it
	 * wakes us up.
	 */

	/* add_wait_queue is the barrier after __set_current_state. */
	__set_current_state(TASK_INTERRUPTIBLE);

	/*
	 * Add current as the futex_q waiter.  We don't remove ourselves from
	 * the wait_queue because we are the only user of it.
	 */
	add_wait_queue(&q->waiter, wait);

	/* Arm the timer */
	if (timeout) {
		hrtimer_start_expires(&timeout->timer, HRTIMER_MODE_ABS);
		if (!hrtimer_active(&timeout->timer))
			timeout->task = NULL;
	}

	/*
	 * !plist_node_empty() is safe here without any lock.
	 * q.lock_ptr != 0 is not safe, because of ordering against wakeup.
	 */
	if (likely(!plist_node_empty(&q->list))) {
		/*
		 * If the timer has already expired, current will already be
		 * flagged for rescheduling. Only call schedule if there
		 * is no timeout, or if it has yet to expire.
		 */
		if (!timeout || timeout->task)
			schedule();
	}
	__set_current_state(TASK_RUNNING);
}

P
Peter Zijlstra 已提交
1312
static int futex_wait(u32 __user *uaddr, int fshared,
1313
		      u32 val, ktime_t *abs_time, u32 bitset, int clockrt)
L
Linus Torvalds 已提交
1314
{
1315 1316
	struct hrtimer_sleeper timeout, *to = NULL;
	DECLARE_WAITQUEUE(wait, current);
P
Peter Zijlstra 已提交
1317
	struct restart_block *restart;
1318
	struct futex_hash_bucket *hb;
L
Linus Torvalds 已提交
1319
	struct futex_q q;
1320 1321
	u32 uval;
	int ret;
L
Linus Torvalds 已提交
1322

1323 1324 1325
	if (!bitset)
		return -EINVAL;

1326
	q.pi_state = NULL;
1327
	q.bitset = bitset;
1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338

	if (abs_time) {
		to = &timeout;

		hrtimer_init_on_stack(&to->timer, clockrt ? CLOCK_REALTIME :
				      CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
		hrtimer_init_sleeper(to, current);
		hrtimer_set_expires_range_ns(&to->timer, *abs_time,
					     current->timer_slack_ns);
	}

1339
retry:
1340
	q.key = FUTEX_KEY_INIT;
E
Eric Dumazet 已提交
1341
	ret = get_futex_key(uaddr, fshared, &q.key);
L
Linus Torvalds 已提交
1342
	if (unlikely(ret != 0))
1343
		goto out;
L
Linus Torvalds 已提交
1344

D
Darren Hart 已提交
1345
retry_private:
E
Eric Sesterhenn 已提交
1346
	hb = queue_lock(&q);
L
Linus Torvalds 已提交
1347 1348

	/*
D
Darren Hart 已提交
1349
	 * Access the page AFTER the hash-bucket is locked.
L
Linus Torvalds 已提交
1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364
	 * Order is important:
	 *
	 *   Userspace waiter: val = var; if (cond(val)) futex_wait(&var, val);
	 *   Userspace waker:  if (cond(var)) { var = new; futex_wake(&var); }
	 *
	 * The basic logical guarantee of a futex is that it blocks ONLY
	 * if cond(var) is known to be true at the time of blocking, for
	 * any cond.  If we queued after testing *uaddr, that would open
	 * a race condition where we could block indefinitely with
	 * cond(var) false, which would violate the guarantee.
	 *
	 * A consequence is that futex_wait() can return zero and absorb
	 * a wakeup when *uaddr != val on entry to the syscall.  This is
	 * rare, but normal.
	 *
D
Darren Hart 已提交
1365
	 * For shared futexes, we hold the mmap semaphore, so the mapping
E
Eric Dumazet 已提交
1366
	 * cannot have changed since we looked it up in get_futex_key.
L
Linus Torvalds 已提交
1367
	 */
1368
	ret = get_futex_value_locked(&uval, uaddr);
L
Linus Torvalds 已提交
1369 1370

	if (unlikely(ret)) {
1371
		queue_unlock(&q, hb);
L
Linus Torvalds 已提交
1372

1373
		ret = get_user(uval, uaddr);
D
Darren Hart 已提交
1374 1375
		if (ret)
			goto out_put_key;
L
Linus Torvalds 已提交
1376

D
Darren Hart 已提交
1377 1378 1379 1380 1381
		if (!fshared)
			goto retry_private;

		put_futex_key(fshared, &q.key);
		goto retry;
L
Linus Torvalds 已提交
1382
	}
1383
	ret = -EWOULDBLOCK;
1384 1385

	/* Only actually queue if *uaddr contained val.  */
P
Peter Zijlstra 已提交
1386 1387 1388 1389
	if (unlikely(uval != val)) {
		queue_unlock(&q, hb);
		goto out_put_key;
	}
L
Linus Torvalds 已提交
1390

1391 1392
	/* queue_me and wait for wakeup, timeout, or a signal. */
	futex_wait_queue_me(hb, &q, to, &wait);
L
Linus Torvalds 已提交
1393 1394

	/* If we were woken (and unqueued), we succeeded, whatever. */
P
Peter Zijlstra 已提交
1395
	ret = 0;
L
Linus Torvalds 已提交
1396
	if (!unqueue_me(&q))
P
Peter Zijlstra 已提交
1397 1398
		goto out_put_key;
	ret = -ETIMEDOUT;
1399
	if (to && !to->task)
P
Peter Zijlstra 已提交
1400
		goto out_put_key;
N
Nick Piggin 已提交
1401

1402 1403 1404 1405
	/*
	 * We expect signal_pending(current), but another thread may
	 * have handled it for us already.
	 */
P
Peter Zijlstra 已提交
1406
	ret = -ERESTARTSYS;
1407
	if (!abs_time)
P
Peter Zijlstra 已提交
1408
		goto out_put_key;
L
Linus Torvalds 已提交
1409

P
Peter Zijlstra 已提交
1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421
	restart = &current_thread_info()->restart_block;
	restart->fn = futex_wait_restart;
	restart->futex.uaddr = (u32 *)uaddr;
	restart->futex.val = val;
	restart->futex.time = abs_time->tv64;
	restart->futex.bitset = bitset;
	restart->futex.flags = 0;

	if (fshared)
		restart->futex.flags |= FLAGS_SHARED;
	if (clockrt)
		restart->futex.flags |= FLAGS_CLOCKRT;
1422

P
Peter Zijlstra 已提交
1423 1424 1425 1426
	ret = -ERESTART_RESTARTBLOCK;

out_put_key:
	put_futex_key(fshared, &q.key);
1427
out:
1428 1429 1430 1431
	if (to) {
		hrtimer_cancel(&to->timer);
		destroy_hrtimer_on_stack(&to->timer);
	}
1432 1433 1434
	return ret;
}

N
Nick Piggin 已提交
1435 1436 1437

static long futex_wait_restart(struct restart_block *restart)
{
1438
	u32 __user *uaddr = (u32 __user *)restart->futex.uaddr;
P
Peter Zijlstra 已提交
1439
	int fshared = 0;
1440
	ktime_t t;
N
Nick Piggin 已提交
1441

1442
	t.tv64 = restart->futex.time;
N
Nick Piggin 已提交
1443
	restart->fn = do_no_restart_syscall;
1444
	if (restart->futex.flags & FLAGS_SHARED)
P
Peter Zijlstra 已提交
1445
		fshared = 1;
1446
	return (long)futex_wait(uaddr, fshared, restart->futex.val, &t,
1447 1448
				restart->futex.bitset,
				restart->futex.flags & FLAGS_CLOCKRT);
N
Nick Piggin 已提交
1449 1450 1451
}


1452 1453 1454 1455 1456 1457
/*
 * Userspace tried a 0 -> TID atomic transition of the futex value
 * and failed. The kernel side here does the whole locking operation:
 * if there are waiters then it will block, it does PI, etc. (Due to
 * races the kernel might see a 0 value of the futex too.)
 */
P
Peter Zijlstra 已提交
1458
static int futex_lock_pi(u32 __user *uaddr, int fshared,
E
Eric Dumazet 已提交
1459
			 int detect, ktime_t *time, int trylock)
1460
{
1461
	struct hrtimer_sleeper timeout, *to = NULL;
1462 1463
	struct task_struct *curr = current;
	struct futex_hash_bucket *hb;
1464
	u32 uval;
1465
	struct futex_q q;
1466
	int ret;
1467 1468 1469 1470

	if (refill_pi_state_cache())
		return -ENOMEM;

1471
	if (time) {
1472
		to = &timeout;
1473 1474
		hrtimer_init_on_stack(&to->timer, CLOCK_REALTIME,
				      HRTIMER_MODE_ABS);
1475
		hrtimer_init_sleeper(to, current);
1476
		hrtimer_set_expires(&to->timer, *time);
1477 1478
	}

1479
	q.pi_state = NULL;
1480
retry:
1481
	q.key = FUTEX_KEY_INIT;
E
Eric Dumazet 已提交
1482
	ret = get_futex_key(uaddr, fshared, &q.key);
1483
	if (unlikely(ret != 0))
1484
		goto out;
1485

D
Darren Hart 已提交
1486
retry_private:
E
Eric Sesterhenn 已提交
1487
	hb = queue_lock(&q);
1488

1489
	ret = futex_lock_pi_atomic(uaddr, hb, &q.key, &q.pi_state, current);
1490
	if (unlikely(ret)) {
1491
		switch (ret) {
1492 1493 1494 1495 1496 1497
		case 1:
			/* We got the lock. */
			ret = 0;
			goto out_unlock_put_key;
		case -EFAULT:
			goto uaddr_faulted;
1498 1499 1500 1501 1502 1503
		case -EAGAIN:
			/*
			 * Task is exiting and we just wait for the
			 * exit to complete.
			 */
			queue_unlock(&q, hb);
1504
			put_futex_key(fshared, &q.key);
1505 1506 1507
			cond_resched();
			goto retry;
		default:
1508
			goto out_unlock_put_key;
1509 1510 1511 1512 1513 1514
		}
	}

	/*
	 * Only actually queue now that the atomic ops are done:
	 */
E
Eric Sesterhenn 已提交
1515
	queue_me(&q, hb);
1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528

	WARN_ON(!q.pi_state);
	/*
	 * Block on the PI mutex:
	 */
	if (!trylock)
		ret = rt_mutex_timed_lock(&q.pi_state->pi_mutex, to, 1);
	else {
		ret = rt_mutex_trylock(&q.pi_state->pi_mutex);
		/* Fixup the trylock return value: */
		ret = ret ? 0 : -EWOULDBLOCK;
	}

1529
	spin_lock(q.lock_ptr);
1530

1531 1532 1533 1534 1535 1536 1537
	if (!ret) {
		/*
		 * Got the lock. We might not be the anticipated owner
		 * if we did a lock-steal - fix up the PI-state in
		 * that case:
		 */
		if (q.pi_state->owner != curr)
1538
			ret = fixup_pi_state_owner(uaddr, &q, curr, fshared);
1539
	} else {
1540 1541
		/*
		 * Catch the rare case, where the lock was released
1542 1543
		 * when we were on the way back before we locked the
		 * hash bucket.
1544
		 */
1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569
		if (q.pi_state->owner == curr) {
			/*
			 * Try to get the rt_mutex now. This might
			 * fail as some other task acquired the
			 * rt_mutex after we removed ourself from the
			 * rt_mutex waiters list.
			 */
			if (rt_mutex_trylock(&q.pi_state->pi_mutex))
				ret = 0;
			else {
				/*
				 * pi_state is incorrect, some other
				 * task did a lock steal and we
				 * returned due to timeout or signal
				 * without taking the rt_mutex. Too
				 * late. We can access the
				 * rt_mutex_owner without locking, as
				 * the other task is now blocked on
				 * the hash bucket lock. Fix the state
				 * up.
				 */
				struct task_struct *owner;
				int res;

				owner = rt_mutex_owner(&q.pi_state->pi_mutex);
1570 1571
				res = fixup_pi_state_owner(uaddr, &q, owner,
							   fshared);
1572 1573 1574 1575 1576

				/* propagate -EFAULT, if the fixup failed */
				if (res)
					ret = res;
			}
1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
		} else {
			/*
			 * Paranoia check. If we did not take the lock
			 * in the trylock above, then we should not be
			 * the owner of the rtmutex, neither the real
			 * nor the pending one:
			 */
			if (rt_mutex_owner(&q.pi_state->pi_mutex) == curr)
				printk(KERN_ERR "futex_lock_pi: ret = %d "
				       "pi-mutex: %p pi-state %p\n", ret,
				       q.pi_state->pi_mutex.owner,
				       q.pi_state->owner);
1589 1590 1591
		}
	}

1592 1593 1594 1595 1596 1597 1598
	/*
	 * If fixup_pi_state_owner() faulted and was unable to handle the
	 * fault, unlock it and return the fault to userspace.
	 */
	if (ret && (rt_mutex_owner(&q.pi_state->pi_mutex) == current))
		rt_mutex_unlock(&q.pi_state->pi_mutex);

1599 1600
	/* Unqueue and drop the lock */
	unqueue_me_pi(&q);
1601

1602 1603
	if (to)
		destroy_hrtimer_on_stack(&to->timer);
1604
	return ret != -EINTR ? ret : -ERESTARTNOINTR;
1605

1606
out_unlock_put_key:
1607 1608
	queue_unlock(&q, hb);

1609
out_put_key:
1610
	put_futex_key(fshared, &q.key);
1611
out:
1612 1613
	if (to)
		destroy_hrtimer_on_stack(&to->timer);
1614 1615
	return ret;

1616
uaddr_faulted:
1617
	/*
1618 1619 1620 1621 1622
	 * We have to r/w  *(int __user *)uaddr, and we have to modify it
	 * atomically.  Therefore, if we continue to fault after get_user()
	 * below, we need to handle the fault ourselves, while still holding
	 * the mmap_sem.  This can occur if the uaddr is under contention as
	 * we have to drop the mmap_sem in order to call get_user().
1623
	 */
1624 1625
	queue_unlock(&q, hb);

1626
	ret = get_user(uval, uaddr);
D
Darren Hart 已提交
1627 1628
	if (ret)
		goto out_put_key;
1629

D
Darren Hart 已提交
1630 1631 1632 1633 1634
	if (!fshared)
		goto retry_private;

	put_futex_key(fshared, &q.key);
	goto retry;
1635 1636
}

1637

1638 1639 1640 1641 1642
/*
 * Userspace attempted a TID -> 0 atomic transition, and failed.
 * This is the in-kernel slowpath: we look up the PI state (if any),
 * and do the rt-mutex unlock.
 */
P
Peter Zijlstra 已提交
1643
static int futex_unlock_pi(u32 __user *uaddr, int fshared)
1644 1645 1646 1647
{
	struct futex_hash_bucket *hb;
	struct futex_q *this, *next;
	u32 uval;
P
Pierre Peiffer 已提交
1648
	struct plist_head *head;
1649
	union futex_key key = FUTEX_KEY_INIT;
D
Darren Hart 已提交
1650
	int ret;
1651 1652 1653 1654 1655 1656 1657

retry:
	if (get_user(uval, uaddr))
		return -EFAULT;
	/*
	 * We release only a lock we actually own:
	 */
1658
	if ((uval & FUTEX_TID_MASK) != task_pid_vnr(current))
1659 1660
		return -EPERM;

E
Eric Dumazet 已提交
1661
	ret = get_futex_key(uaddr, fshared, &key);
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672
	if (unlikely(ret != 0))
		goto out;

	hb = hash_futex(&key);
	spin_lock(&hb->lock);

	/*
	 * To avoid races, try to do the TID -> 0 atomic transition
	 * again. If it succeeds then we can return without waking
	 * anyone else up:
	 */
T
Thomas Gleixner 已提交
1673
	if (!(uval & FUTEX_OWNER_DIED))
1674
		uval = cmpxchg_futex_value_locked(uaddr, task_pid_vnr(current), 0);
T
Thomas Gleixner 已提交
1675

1676 1677 1678 1679 1680 1681 1682

	if (unlikely(uval == -EFAULT))
		goto pi_faulted;
	/*
	 * Rare case: we managed to release the lock atomically,
	 * no need to wake anyone else up:
	 */
1683
	if (unlikely(uval == task_pid_vnr(current)))
1684 1685 1686 1687 1688 1689 1690 1691
		goto out_unlock;

	/*
	 * Ok, other tasks may need to be woken up - check waiters
	 * and do the wakeup if necessary:
	 */
	head = &hb->chain;

P
Pierre Peiffer 已提交
1692
	plist_for_each_entry_safe(this, next, head, list) {
1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707
		if (!match_futex (&this->key, &key))
			continue;
		ret = wake_futex_pi(uaddr, uval, this);
		/*
		 * The atomic access to the futex value
		 * generated a pagefault, so retry the
		 * user-access and the wakeup:
		 */
		if (ret == -EFAULT)
			goto pi_faulted;
		goto out_unlock;
	}
	/*
	 * No waiters - kernel unlocks the futex:
	 */
1708 1709 1710 1711 1712
	if (!(uval & FUTEX_OWNER_DIED)) {
		ret = unlock_futex_pi(uaddr, uval);
		if (ret == -EFAULT)
			goto pi_faulted;
	}
1713 1714 1715

out_unlock:
	spin_unlock(&hb->lock);
1716
	put_futex_key(fshared, &key);
1717

1718
out:
1719 1720 1721 1722
	return ret;

pi_faulted:
	/*
1723 1724 1725 1726 1727
	 * We have to r/w  *(int __user *)uaddr, and we have to modify it
	 * atomically.  Therefore, if we continue to fault after get_user()
	 * below, we need to handle the fault ourselves, while still holding
	 * the mmap_sem.  This can occur if the uaddr is under contention as
	 * we have to drop the mmap_sem in order to call get_user().
1728
	 */
1729
	spin_unlock(&hb->lock);
D
Darren Hart 已提交
1730
	put_futex_key(fshared, &key);
1731 1732

	ret = get_user(uval, uaddr);
1733
	if (!ret)
1734 1735
		goto retry;

L
Linus Torvalds 已提交
1736 1737 1738
	return ret;
}

1739 1740 1741 1742 1743 1744 1745
/*
 * Support for robust futexes: the kernel cleans up held futexes at
 * thread exit time.
 *
 * Implementation: user-space maintains a per-thread list of locks it
 * is holding. Upon do_exit(), the kernel carefully walks this list,
 * and marks all locks that are owned by this thread with the
1746
 * FUTEX_OWNER_DIED bit, and wakes up a waiter (if any). The list is
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758
 * always manipulated with the lock held, so the list is private and
 * per-thread. Userspace also maintains a per-thread 'list_op_pending'
 * field, to allow the kernel to clean up if the thread dies after
 * acquiring the lock, but just before it could have added itself to
 * the list. There can only be one such pending lock.
 */

/**
 * sys_set_robust_list - set the robust-futex list head of a task
 * @head: pointer to the list-head
 * @len: length of the list-head, as userspace expects
 */
1759 1760
SYSCALL_DEFINE2(set_robust_list, struct robust_list_head __user *, head,
		size_t, len)
1761
{
1762 1763
	if (!futex_cmpxchg_enabled)
		return -ENOSYS;
1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780
	/*
	 * The kernel knows only one size for now:
	 */
	if (unlikely(len != sizeof(*head)))
		return -EINVAL;

	current->robust_list = head;

	return 0;
}

/**
 * sys_get_robust_list - get the robust-futex list head of a task
 * @pid: pid of the process [zero for current task]
 * @head_ptr: pointer to a list-head pointer, the kernel fills it in
 * @len_ptr: pointer to a length field, the kernel fills in the header size
 */
1781 1782 1783
SYSCALL_DEFINE3(get_robust_list, int, pid,
		struct robust_list_head __user * __user *, head_ptr,
		size_t __user *, len_ptr)
1784
{
A
Al Viro 已提交
1785
	struct robust_list_head __user *head;
1786
	unsigned long ret;
1787
	const struct cred *cred = current_cred(), *pcred;
1788

1789 1790 1791
	if (!futex_cmpxchg_enabled)
		return -ENOSYS;

1792 1793 1794 1795 1796 1797
	if (!pid)
		head = current->robust_list;
	else {
		struct task_struct *p;

		ret = -ESRCH;
1798
		rcu_read_lock();
1799
		p = find_task_by_vpid(pid);
1800 1801 1802
		if (!p)
			goto err_unlock;
		ret = -EPERM;
1803 1804 1805
		pcred = __task_cred(p);
		if (cred->euid != pcred->euid &&
		    cred->euid != pcred->uid &&
1806
		    !capable(CAP_SYS_PTRACE))
1807 1808
			goto err_unlock;
		head = p->robust_list;
1809
		rcu_read_unlock();
1810 1811 1812 1813 1814 1815 1816
	}

	if (put_user(sizeof(*head), len_ptr))
		return -EFAULT;
	return put_user(head, head_ptr);

err_unlock:
1817
	rcu_read_unlock();
1818 1819 1820 1821 1822 1823 1824 1825

	return ret;
}

/*
 * Process a futex-list entry, check whether it's owned by the
 * dying task, and do notification if so:
 */
1826
int handle_futex_death(u32 __user *uaddr, struct task_struct *curr, int pi)
1827
{
1828
	u32 uval, nval, mval;
1829

1830 1831
retry:
	if (get_user(uval, uaddr))
1832 1833
		return -1;

1834
	if ((uval & FUTEX_TID_MASK) == task_pid_vnr(curr)) {
1835 1836 1837 1838 1839 1840 1841 1842 1843 1844
		/*
		 * Ok, this dying thread is truly holding a futex
		 * of interest. Set the OWNER_DIED bit atomically
		 * via cmpxchg, and if the value had FUTEX_WAITERS
		 * set, wake up a waiter (if any). (We have to do a
		 * futex_wake() even if OWNER_DIED is already set -
		 * to handle the rare but possible case of recursive
		 * thread-death.) The rest of the cleanup is done in
		 * userspace.
		 */
1845 1846 1847
		mval = (uval & FUTEX_WAITERS) | FUTEX_OWNER_DIED;
		nval = futex_atomic_cmpxchg_inatomic(uaddr, uval, mval);

1848 1849 1850 1851
		if (nval == -EFAULT)
			return -1;

		if (nval != uval)
1852
			goto retry;
1853

1854 1855 1856 1857
		/*
		 * Wake robust non-PI futexes here. The wakeup of
		 * PI futexes happens in exit_pi_state():
		 */
T
Thomas Gleixner 已提交
1858
		if (!pi && (uval & FUTEX_WAITERS))
P
Peter Zijlstra 已提交
1859
			futex_wake(uaddr, 1, 1, FUTEX_BITSET_MATCH_ANY);
1860 1861 1862 1863
	}
	return 0;
}

1864 1865 1866 1867
/*
 * Fetch a robust-list pointer. Bit 0 signals PI futexes:
 */
static inline int fetch_robust_entry(struct robust_list __user **entry,
A
Al Viro 已提交
1868 1869
				     struct robust_list __user * __user *head,
				     int *pi)
1870 1871 1872
{
	unsigned long uentry;

A
Al Viro 已提交
1873
	if (get_user(uentry, (unsigned long __user *)head))
1874 1875
		return -EFAULT;

A
Al Viro 已提交
1876
	*entry = (void __user *)(uentry & ~1UL);
1877 1878 1879 1880 1881
	*pi = uentry & 1;

	return 0;
}

1882 1883 1884 1885 1886 1887 1888 1889 1890
/*
 * Walk curr->robust_list (very carefully, it's a userspace list!)
 * and mark any locks found there dead, and notify any waiters.
 *
 * We silently return on any sign of list-walking problem.
 */
void exit_robust_list(struct task_struct *curr)
{
	struct robust_list_head __user *head = curr->robust_list;
M
Martin Schwidefsky 已提交
1891 1892
	struct robust_list __user *entry, *next_entry, *pending;
	unsigned int limit = ROBUST_LIST_LIMIT, pi, next_pi, pip;
1893
	unsigned long futex_offset;
M
Martin Schwidefsky 已提交
1894
	int rc;
1895

1896 1897 1898
	if (!futex_cmpxchg_enabled)
		return;

1899 1900 1901 1902
	/*
	 * Fetch the list head (which was registered earlier, via
	 * sys_set_robust_list()):
	 */
1903
	if (fetch_robust_entry(&entry, &head->list.next, &pi))
1904 1905 1906 1907 1908 1909 1910 1911 1912 1913
		return;
	/*
	 * Fetch the relative futex offset:
	 */
	if (get_user(futex_offset, &head->futex_offset))
		return;
	/*
	 * Fetch any possibly pending lock-add first, and handle it
	 * if it exists:
	 */
1914
	if (fetch_robust_entry(&pending, &head->list_op_pending, &pip))
1915
		return;
1916

M
Martin Schwidefsky 已提交
1917
	next_entry = NULL;	/* avoid warning with gcc */
1918
	while (entry != &head->list) {
M
Martin Schwidefsky 已提交
1919 1920 1921 1922 1923
		/*
		 * Fetch the next entry in the list before calling
		 * handle_futex_death:
		 */
		rc = fetch_robust_entry(&next_entry, &entry->next, &next_pi);
1924 1925
		/*
		 * A pending lock might already be on the list, so
1926
		 * don't process it twice:
1927 1928
		 */
		if (entry != pending)
A
Al Viro 已提交
1929
			if (handle_futex_death((void __user *)entry + futex_offset,
1930
						curr, pi))
1931
				return;
M
Martin Schwidefsky 已提交
1932
		if (rc)
1933
			return;
M
Martin Schwidefsky 已提交
1934 1935
		entry = next_entry;
		pi = next_pi;
1936 1937 1938 1939 1940 1941 1942 1943
		/*
		 * Avoid excessively long or circular lists:
		 */
		if (!--limit)
			break;

		cond_resched();
	}
M
Martin Schwidefsky 已提交
1944 1945 1946 1947

	if (pending)
		handle_futex_death((void __user *)pending + futex_offset,
				   curr, pip);
1948 1949
}

1950
long do_futex(u32 __user *uaddr, int op, u32 val, ktime_t *timeout,
1951
		u32 __user *uaddr2, u32 val2, u32 val3)
L
Linus Torvalds 已提交
1952
{
1953
	int clockrt, ret = -ENOSYS;
E
Eric Dumazet 已提交
1954
	int cmd = op & FUTEX_CMD_MASK;
P
Peter Zijlstra 已提交
1955
	int fshared = 0;
E
Eric Dumazet 已提交
1956 1957

	if (!(op & FUTEX_PRIVATE_FLAG))
P
Peter Zijlstra 已提交
1958
		fshared = 1;
L
Linus Torvalds 已提交
1959

1960 1961 1962
	clockrt = op & FUTEX_CLOCK_REALTIME;
	if (clockrt && cmd != FUTEX_WAIT_BITSET)
		return -ENOSYS;
L
Linus Torvalds 已提交
1963

E
Eric Dumazet 已提交
1964
	switch (cmd) {
L
Linus Torvalds 已提交
1965
	case FUTEX_WAIT:
1966 1967
		val3 = FUTEX_BITSET_MATCH_ANY;
	case FUTEX_WAIT_BITSET:
1968
		ret = futex_wait(uaddr, fshared, val, timeout, val3, clockrt);
L
Linus Torvalds 已提交
1969 1970
		break;
	case FUTEX_WAKE:
1971 1972 1973
		val3 = FUTEX_BITSET_MATCH_ANY;
	case FUTEX_WAKE_BITSET:
		ret = futex_wake(uaddr, fshared, val, val3);
L
Linus Torvalds 已提交
1974 1975
		break;
	case FUTEX_REQUEUE:
E
Eric Dumazet 已提交
1976
		ret = futex_requeue(uaddr, fshared, uaddr2, val, val2, NULL);
L
Linus Torvalds 已提交
1977 1978
		break;
	case FUTEX_CMP_REQUEUE:
E
Eric Dumazet 已提交
1979
		ret = futex_requeue(uaddr, fshared, uaddr2, val, val2, &val3);
L
Linus Torvalds 已提交
1980
		break;
1981
	case FUTEX_WAKE_OP:
E
Eric Dumazet 已提交
1982
		ret = futex_wake_op(uaddr, fshared, uaddr2, val, val2, val3);
1983
		break;
1984
	case FUTEX_LOCK_PI:
1985 1986
		if (futex_cmpxchg_enabled)
			ret = futex_lock_pi(uaddr, fshared, val, timeout, 0);
1987 1988
		break;
	case FUTEX_UNLOCK_PI:
1989 1990
		if (futex_cmpxchg_enabled)
			ret = futex_unlock_pi(uaddr, fshared);
1991 1992
		break;
	case FUTEX_TRYLOCK_PI:
1993 1994
		if (futex_cmpxchg_enabled)
			ret = futex_lock_pi(uaddr, fshared, 0, timeout, 1);
1995
		break;
L
Linus Torvalds 已提交
1996 1997 1998 1999 2000 2001 2002
	default:
		ret = -ENOSYS;
	}
	return ret;
}


2003 2004 2005
SYSCALL_DEFINE6(futex, u32 __user *, uaddr, int, op, u32, val,
		struct timespec __user *, utime, u32 __user *, uaddr2,
		u32, val3)
L
Linus Torvalds 已提交
2006
{
2007 2008
	struct timespec ts;
	ktime_t t, *tp = NULL;
2009
	u32 val2 = 0;
E
Eric Dumazet 已提交
2010
	int cmd = op & FUTEX_CMD_MASK;
L
Linus Torvalds 已提交
2011

2012 2013
	if (utime && (cmd == FUTEX_WAIT || cmd == FUTEX_LOCK_PI ||
		      cmd == FUTEX_WAIT_BITSET)) {
2014
		if (copy_from_user(&ts, utime, sizeof(ts)) != 0)
L
Linus Torvalds 已提交
2015
			return -EFAULT;
2016
		if (!timespec_valid(&ts))
2017
			return -EINVAL;
2018 2019

		t = timespec_to_ktime(ts);
E
Eric Dumazet 已提交
2020
		if (cmd == FUTEX_WAIT)
2021
			t = ktime_add_safe(ktime_get(), t);
2022
		tp = &t;
L
Linus Torvalds 已提交
2023 2024
	}
	/*
E
Eric Dumazet 已提交
2025
	 * requeue parameter in 'utime' if cmd == FUTEX_REQUEUE.
2026
	 * number of waiters to wake in 'utime' if cmd == FUTEX_WAKE_OP.
L
Linus Torvalds 已提交
2027
	 */
2028 2029
	if (cmd == FUTEX_REQUEUE || cmd == FUTEX_CMP_REQUEUE ||
	    cmd == FUTEX_WAKE_OP)
2030
		val2 = (u32) (unsigned long) utime;
L
Linus Torvalds 已提交
2031

2032
	return do_futex(uaddr, op, val, tp, uaddr2, val2, val3);
L
Linus Torvalds 已提交
2033 2034
}

2035
static int __init futex_init(void)
L
Linus Torvalds 已提交
2036
{
2037
	u32 curval;
T
Thomas Gleixner 已提交
2038
	int i;
A
Akinobu Mita 已提交
2039

2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053
	/*
	 * This will fail and we want it. Some arch implementations do
	 * runtime detection of the futex_atomic_cmpxchg_inatomic()
	 * functionality. We want to know that before we call in any
	 * of the complex code paths. Also we want to prevent
	 * registration of robust lists in that case. NULL is
	 * guaranteed to fault and we get -EFAULT on functional
	 * implementation, the non functional ones will return
	 * -ENOSYS.
	 */
	curval = cmpxchg_futex_value_locked(NULL, 0, 0);
	if (curval == -EFAULT)
		futex_cmpxchg_enabled = 1;

T
Thomas Gleixner 已提交
2054 2055 2056 2057 2058
	for (i = 0; i < ARRAY_SIZE(futex_queues); i++) {
		plist_head_init(&futex_queues[i].chain, &futex_queues[i].lock);
		spin_lock_init(&futex_queues[i].lock);
	}

L
Linus Torvalds 已提交
2059 2060
	return 0;
}
2061
__initcall(futex_init);