[IA64] Fix race in mm-context wrap-around logic.
[linux-2.6.git] / include / asm-ia64 / mmu_context.h
blob0680d163be9733ee6fd45522d6e9dcf61de228d9
1 #ifndef _ASM_IA64_MMU_CONTEXT_H
2 #define _ASM_IA64_MMU_CONTEXT_H
4 /*
5 * Copyright (C) 1998-2002 Hewlett-Packard Co
6 * David Mosberger-Tang <davidm@hpl.hp.com>
7 */
9 /*
10 * Routines to manage the allocation of task context numbers. Task context numbers are
11 * used to reduce or eliminate the need to perform TLB flushes due to context switches.
12 * Context numbers are implemented using ia-64 region ids. Since the IA-64 TLB does not
13 * consider the region number when performing a TLB lookup, we need to assign a unique
14 * region id to each region in a process. We use the least significant three bits in a
15 * region id for this purpose.
18 #define IA64_REGION_ID_KERNEL 0 /* the kernel's region id (tlb.c depends on this being 0) */
20 #define ia64_rid(ctx,addr) (((ctx) << 3) | (addr >> 61))
22 # ifndef __ASSEMBLY__
24 #include <linux/compiler.h>
25 #include <linux/percpu.h>
26 #include <linux/sched.h>
27 #include <linux/spinlock.h>
29 #include <asm/processor.h>
31 struct ia64_ctx {
32 spinlock_t lock;
33 unsigned int next; /* next context number to use */
34 unsigned int limit; /* next >= limit => must call wrap_mmu_context() */
35 unsigned int max_ctx; /* max. context value supported by all CPUs */
38 extern struct ia64_ctx ia64_ctx;
39 DECLARE_PER_CPU(u8, ia64_need_tlb_flush);
41 extern void wrap_mmu_context (struct mm_struct *mm);
43 static inline void
44 enter_lazy_tlb (struct mm_struct *mm, struct task_struct *tsk)
49 * When the context counter wraps around all TLBs need to be flushed because an old
50 * context number might have been reused. This is signalled by the ia64_need_tlb_flush
51 * per-CPU variable, which is checked in the routine below. Called by activate_mm().
52 * <efocht@ess.nec.de>
54 static inline void
55 delayed_tlb_flush (void)
57 extern void local_flush_tlb_all (void);
58 unsigned long flags;
60 if (unlikely(__ia64_per_cpu_var(ia64_need_tlb_flush))) {
61 spin_lock_irqsave(&ia64_ctx.lock, flags);
63 if (__ia64_per_cpu_var(ia64_need_tlb_flush)) {
64 local_flush_tlb_all();
65 __ia64_per_cpu_var(ia64_need_tlb_flush) = 0;
68 spin_unlock_irqrestore(&ia64_ctx.lock, flags);
72 static inline nv_mm_context_t
73 get_mmu_context (struct mm_struct *mm)
75 unsigned long flags;
76 nv_mm_context_t context = mm->context;
78 if (unlikely(!context)) {
79 spin_lock_irqsave(&ia64_ctx.lock, flags);
81 /* re-check, now that we've got the lock: */
82 context = mm->context;
83 if (context == 0) {
84 cpus_clear(mm->cpu_vm_mask);
85 if (ia64_ctx.next >= ia64_ctx.limit)
86 wrap_mmu_context(mm);
87 mm->context = context = ia64_ctx.next++;
90 spin_unlock_irqrestore(&ia64_ctx.lock, flags);
93 * Ensure we're not starting to use "context" before any old
94 * uses of it are gone from our TLB.
96 delayed_tlb_flush();
98 return context;
102 * Initialize context number to some sane value. MM is guaranteed to be a brand-new
103 * address-space, so no TLB flushing is needed, ever.
105 static inline int
106 init_new_context (struct task_struct *p, struct mm_struct *mm)
108 mm->context = 0;
109 return 0;
112 static inline void
113 destroy_context (struct mm_struct *mm)
115 /* Nothing to do. */
118 static inline void
119 reload_context (nv_mm_context_t context)
121 unsigned long rid;
122 unsigned long rid_incr = 0;
123 unsigned long rr0, rr1, rr2, rr3, rr4, old_rr4;
125 old_rr4 = ia64_get_rr(0x8000000000000000UL);
126 rid = context << 3; /* make space for encoding the region number */
127 rid_incr = 1 << 8;
129 /* encode the region id, preferred page size, and VHPT enable bit: */
130 rr0 = (rid << 8) | (PAGE_SHIFT << 2) | 1;
131 rr1 = rr0 + 1*rid_incr;
132 rr2 = rr0 + 2*rid_incr;
133 rr3 = rr0 + 3*rid_incr;
134 rr4 = rr0 + 4*rid_incr;
135 #ifdef CONFIG_HUGETLB_PAGE
136 rr4 = (rr4 & (~(0xfcUL))) | (old_rr4 & 0xfc);
137 #endif
139 ia64_set_rr(0x0000000000000000UL, rr0);
140 ia64_set_rr(0x2000000000000000UL, rr1);
141 ia64_set_rr(0x4000000000000000UL, rr2);
142 ia64_set_rr(0x6000000000000000UL, rr3);
143 ia64_set_rr(0x8000000000000000UL, rr4);
144 ia64_srlz_i(); /* srlz.i implies srlz.d */
148 * Must be called with preemption off
150 static inline void
151 activate_context (struct mm_struct *mm)
153 nv_mm_context_t context;
155 do {
156 context = get_mmu_context(mm);
157 if (!cpu_isset(smp_processor_id(), mm->cpu_vm_mask))
158 cpu_set(smp_processor_id(), mm->cpu_vm_mask);
159 reload_context(context);
160 /* in the unlikely event of a TLB-flush by another thread, redo the load: */
161 } while (unlikely(context != mm->context));
164 #define deactivate_mm(tsk,mm) do { } while (0)
167 * Switch from address space PREV to address space NEXT.
169 static inline void
170 activate_mm (struct mm_struct *prev, struct mm_struct *next)
173 * We may get interrupts here, but that's OK because interrupt handlers cannot
174 * touch user-space.
176 ia64_set_kr(IA64_KR_PT_BASE, __pa(next->pgd));
177 activate_context(next);
180 #define switch_mm(prev_mm,next_mm,next_task) activate_mm(prev_mm, next_mm)
182 # endif /* ! __ASSEMBLY__ */
183 #endif /* _ASM_IA64_MMU_CONTEXT_H */