LXR linux/mm/mmu

   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/module.h>
   9#include <linux/sched.h>
  10
  11#include <asm/mmu_context.h>
  12
  13/*
  14 * use_mm
  15 *      Makes the calling kernel thread take on the specified
  16 *      mm context.
  17 *      Called by the retry thread execute retries within the
  18 *      iocb issuer's mm context, so that copy_from/to_user
  19 *      operations work seamlessly for aio.
  20 *      (Note: this routine is intended to be called only
  21 *      from a kernel thread context)
  22 */
  23void use_mm(struct mm_struct *mm)
  24{
  25        struct mm_struct *active_mm;
  26        struct task_struct *tsk = current;
  27
  28        task_lock(tsk);
  29        active_mm = tsk->active_mm;
  30        if (active_mm != mm) {
  31                atomic_inc(&mm->mm_count);
  32                tsk->active_mm = mm;
  33        }
  34        tsk->mm = mm;
  35        switch_mm(active_mm, mm, tsk);
  36        task_unlock(tsk);
  37
  38        if (active_mm != mm)
  39                mmdrop(active_mm);
  40}
  41EXPORT_SYMBOL_GPL(use_mm);
  42
  43/*
  44 * unuse_mm
  45 *      Reverses the effect of use_mm, i.e. releases the
  46 *      specified mm context which was earlier taken on
  47 *      by the calling kernel thread
  48 *      (Note: this routine is intended to be called only
  49 *      from a kernel thread context)
  50 */
  51void unuse_mm(struct mm_struct *mm)
  52{
  53        struct task_struct *tsk = current;
  54
  55        task_lock(tsk);
  56        sync_mm_rss(tsk, mm);
  57        tsk->mm = NULL;
  58        /* active_mm is still 'mm' */
  59        enter_lazy_tlb(mm, tsk);
  60        task_unlock(tsk);
  61}
  62EXPORT_SYMBOL_GPL(unuse_mm);
  63