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