[PATCH] x86-64: Calgary IOMMU: Update Jon's contact info
[linux-2.6] / arch / x86_64 / kernel / ldt.c
1 /*
2  * linux/arch/x86_64/kernel/ldt.c
3  *
4  * Copyright (C) 1992 Krishna Balasubramanian and Linus Torvalds
5  * Copyright (C) 1999 Ingo Molnar <mingo@redhat.com>
6  * Copyright (C) 2002 Andi Kleen
7  * 
8  * This handles calls from both 32bit and 64bit mode.
9  */
10
11 #include <linux/errno.h>
12 #include <linux/sched.h>
13 #include <linux/string.h>
14 #include <linux/mm.h>
15 #include <linux/smp.h>
16 #include <linux/smp_lock.h>
17 #include <linux/vmalloc.h>
18 #include <linux/slab.h>
19
20 #include <asm/uaccess.h>
21 #include <asm/system.h>
22 #include <asm/ldt.h>
23 #include <asm/desc.h>
24 #include <asm/proto.h>
25
26 #ifdef CONFIG_SMP /* avoids "defined but not used" warnig */
27 static void flush_ldt(void *null)
28 {
29         if (current->active_mm)
30                load_LDT(&current->active_mm->context);
31 }
32 #endif
33
34 static int alloc_ldt(mm_context_t *pc, unsigned mincount, int reload)
35 {
36         void *oldldt;
37         void *newldt;
38         unsigned oldsize;
39
40         if (mincount <= (unsigned)pc->size)
41                 return 0;
42         oldsize = pc->size;
43         mincount = (mincount+511)&(~511);
44         if (mincount*LDT_ENTRY_SIZE > PAGE_SIZE)
45                 newldt = vmalloc(mincount*LDT_ENTRY_SIZE);
46         else
47                 newldt = kmalloc(mincount*LDT_ENTRY_SIZE, GFP_KERNEL);
48
49         if (!newldt)
50                 return -ENOMEM;
51
52         if (oldsize)
53                 memcpy(newldt, pc->ldt, oldsize*LDT_ENTRY_SIZE);
54         oldldt = pc->ldt;
55         memset(newldt+oldsize*LDT_ENTRY_SIZE, 0, (mincount-oldsize)*LDT_ENTRY_SIZE);
56         wmb();
57         pc->ldt = newldt;
58         wmb();
59         pc->size = mincount;
60         wmb();
61         if (reload) {
62 #ifdef CONFIG_SMP
63                 cpumask_t mask;
64
65                 preempt_disable();
66                 mask = cpumask_of_cpu(smp_processor_id());
67                 load_LDT(pc);
68                 if (!cpus_equal(current->mm->cpu_vm_mask, mask))
69                         smp_call_function(flush_ldt, NULL, 1, 1);
70                 preempt_enable();
71 #else
72                 load_LDT(pc);
73 #endif
74         }
75         if (oldsize) {
76                 if (oldsize*LDT_ENTRY_SIZE > PAGE_SIZE)
77                         vfree(oldldt);
78                 else
79                         kfree(oldldt);
80         }
81         return 0;
82 }
83
84 static inline int copy_ldt(mm_context_t *new, mm_context_t *old)
85 {
86         int err = alloc_ldt(new, old->size, 0);
87         if (err < 0)
88                 return err;
89         memcpy(new->ldt, old->ldt, old->size*LDT_ENTRY_SIZE);
90         return 0;
91 }
92
93 /*
94  * we do not have to muck with descriptors here, that is
95  * done in switch_mm() as needed.
96  */
97 int init_new_context(struct task_struct *tsk, struct mm_struct *mm)
98 {
99         struct mm_struct * old_mm;
100         int retval = 0;
101
102         init_MUTEX(&mm->context.sem);
103         mm->context.size = 0;
104         old_mm = current->mm;
105         if (old_mm && old_mm->context.size > 0) {
106                 down(&old_mm->context.sem);
107                 retval = copy_ldt(&mm->context, &old_mm->context);
108                 up(&old_mm->context.sem);
109         }
110         return retval;
111 }
112
113 /*
114  * 
115  * Don't touch the LDT register - we're already in the next thread.
116  */
117 void destroy_context(struct mm_struct *mm)
118 {
119         if (mm->context.size) {
120                 if ((unsigned)mm->context.size*LDT_ENTRY_SIZE > PAGE_SIZE)
121                         vfree(mm->context.ldt);
122                 else
123                         kfree(mm->context.ldt);
124                 mm->context.size = 0;
125         }
126 }
127
128 static int read_ldt(void __user * ptr, unsigned long bytecount)
129 {
130         int err;
131         unsigned long size;
132         struct mm_struct * mm = current->mm;
133
134         if (!mm->context.size)
135                 return 0;
136         if (bytecount > LDT_ENTRY_SIZE*LDT_ENTRIES)
137                 bytecount = LDT_ENTRY_SIZE*LDT_ENTRIES;
138
139         down(&mm->context.sem);
140         size = mm->context.size*LDT_ENTRY_SIZE;
141         if (size > bytecount)
142                 size = bytecount;
143
144         err = 0;
145         if (copy_to_user(ptr, mm->context.ldt, size))
146                 err = -EFAULT;
147         up(&mm->context.sem);
148         if (err < 0)
149                 goto error_return;
150         if (size != bytecount) {
151                 /* zero-fill the rest */
152                 if (clear_user(ptr+size, bytecount-size) != 0) {
153                         err = -EFAULT;
154                         goto error_return;
155                 }
156         }
157         return bytecount;
158 error_return:
159         return err;
160 }
161
162 static int read_default_ldt(void __user * ptr, unsigned long bytecount)
163 {
164         /* Arbitrary number */ 
165         /* x86-64 default LDT is all zeros */
166         if (bytecount > 128) 
167                 bytecount = 128;        
168         if (clear_user(ptr, bytecount))
169                 return -EFAULT;
170         return bytecount; 
171 }
172
173 static int write_ldt(void __user * ptr, unsigned long bytecount, int oldmode)
174 {
175         struct task_struct *me = current;
176         struct mm_struct * mm = me->mm;
177         __u32 entry_1, entry_2, *lp;
178         int error;
179         struct user_desc ldt_info;
180
181         error = -EINVAL;
182
183         if (bytecount != sizeof(ldt_info))
184                 goto out;
185         error = -EFAULT;        
186         if (copy_from_user(&ldt_info, ptr, bytecount))
187                 goto out;
188
189         error = -EINVAL;
190         if (ldt_info.entry_number >= LDT_ENTRIES)
191                 goto out;
192         if (ldt_info.contents == 3) {
193                 if (oldmode)
194                         goto out;
195                 if (ldt_info.seg_not_present == 0)
196                         goto out;
197         }
198
199         down(&mm->context.sem);
200         if (ldt_info.entry_number >= (unsigned)mm->context.size) {
201                 error = alloc_ldt(&current->mm->context, ldt_info.entry_number+1, 1);
202                 if (error < 0)
203                         goto out_unlock;
204         }
205
206         lp = (__u32 *) ((ldt_info.entry_number << 3) + (char *) mm->context.ldt);
207
208         /* Allow LDTs to be cleared by the user. */
209         if (ldt_info.base_addr == 0 && ldt_info.limit == 0) {
210                 if (oldmode || LDT_empty(&ldt_info)) {
211                         entry_1 = 0;
212                         entry_2 = 0;
213                         goto install;
214                 }
215         }
216
217         entry_1 = LDT_entry_a(&ldt_info);
218         entry_2 = LDT_entry_b(&ldt_info);
219         if (oldmode)
220                 entry_2 &= ~(1 << 20);
221
222         /* Install the new entry ...  */
223 install:
224         *lp     = entry_1;
225         *(lp+1) = entry_2;
226         error = 0;
227
228 out_unlock:
229         up(&mm->context.sem);
230 out:
231         return error;
232 }
233
234 asmlinkage int sys_modify_ldt(int func, void __user *ptr, unsigned long bytecount)
235 {
236         int ret = -ENOSYS;
237
238         switch (func) {
239         case 0:
240                 ret = read_ldt(ptr, bytecount);
241                 break;
242         case 1:
243                 ret = write_ldt(ptr, bytecount, 1);
244                 break;
245         case 2:
246                 ret = read_default_ldt(ptr, bytecount);
247                 break;
248         case 0x11:
249                 ret = write_ldt(ptr, bytecount, 0);
250                 break;
251         }
252         return ret;
253 }