Michael S. Tsirkin | 3d2d827 | 2009-09-21 17:03:51 -0700 | [diff] [blame] | 1 | /* Copyright (C) 2009 Red Hat, Inc. |
| 2 | * |
| 3 | * See ../COPYING for licensing terms. |
| 4 | */ |
| 5 | |
| 6 | #include <linux/mm.h> |
| 7 | #include <linux/mmu_context.h> |
| 8 | #include <linux/sched.h> |
| 9 | |
| 10 | #include <asm/mmu_context.h> |
| 11 | |
| 12 | /* |
| 13 | * use_mm |
| 14 | * Makes the calling kernel thread take on the specified |
| 15 | * mm context. |
| 16 | * Called by the retry thread execute retries within the |
| 17 | * iocb issuer's mm context, so that copy_from/to_user |
| 18 | * operations work seamlessly for aio. |
| 19 | * (Note: this routine is intended to be called only |
| 20 | * from a kernel thread context) |
| 21 | */ |
| 22 | void use_mm(struct mm_struct *mm) |
| 23 | { |
| 24 | struct mm_struct *active_mm; |
| 25 | struct task_struct *tsk = current; |
| 26 | |
| 27 | task_lock(tsk); |
| 28 | active_mm = tsk->active_mm; |
Michael S. Tsirkin | f68e148 | 2009-09-21 17:03:52 -0700 | [diff] [blame] | 29 | if (active_mm != mm) { |
| 30 | atomic_inc(&mm->mm_count); |
| 31 | tsk->active_mm = mm; |
| 32 | } |
Michael S. Tsirkin | 3d2d827 | 2009-09-21 17:03:51 -0700 | [diff] [blame] | 33 | tsk->mm = mm; |
Michael S. Tsirkin | 3d2d827 | 2009-09-21 17:03:51 -0700 | [diff] [blame] | 34 | switch_mm(active_mm, mm, tsk); |
| 35 | task_unlock(tsk); |
| 36 | |
Michael S. Tsirkin | f68e148 | 2009-09-21 17:03:52 -0700 | [diff] [blame] | 37 | if (active_mm != mm) |
| 38 | mmdrop(active_mm); |
Michael S. Tsirkin | 3d2d827 | 2009-09-21 17:03:51 -0700 | [diff] [blame] | 39 | } |
| 40 | |
| 41 | /* |
| 42 | * unuse_mm |
| 43 | * Reverses the effect of use_mm, i.e. releases the |
| 44 | * specified mm context which was earlier taken on |
| 45 | * by the calling kernel thread |
| 46 | * (Note: this routine is intended to be called only |
| 47 | * from a kernel thread context) |
| 48 | */ |
| 49 | void unuse_mm(struct mm_struct *mm) |
| 50 | { |
| 51 | struct task_struct *tsk = current; |
| 52 | |
| 53 | task_lock(tsk); |
| 54 | tsk->mm = NULL; |
| 55 | /* active_mm is still 'mm' */ |
| 56 | enter_lazy_tlb(mm, tsk); |
| 57 | task_unlock(tsk); |
| 58 | } |