Merge branch 'master' of git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux-2.6
[linux-2.6] / arch / um / sys-i386 / tls.c
1 /*
2  * Copyright (C) 2005 Paolo 'Blaisorblade' Giarrusso <blaisorblade@yahoo.it>
3  * Licensed under the GPL
4  */
5
6 #include "linux/kernel.h"
7 #include "linux/sched.h"
8 #include "linux/slab.h"
9 #include "linux/types.h"
10 #include "asm/uaccess.h"
11 #include "asm/ptrace.h"
12 #include "asm/segment.h"
13 #include "asm/smp.h"
14 #include "asm/desc.h"
15 #include "choose-mode.h"
16 #include "kern.h"
17 #include "kern_util.h"
18 #include "mode_kern.h"
19 #include "os.h"
20 #include "mode.h"
21
22 #ifdef CONFIG_MODE_SKAS
23 #include "skas.h"
24 #endif
25
26 /*
27  * If needed we can detect when it's uninitialized.
28  *
29  * These are initialized in an initcall and unchanged thereafter.
30  */
31 static int host_supports_tls = -1;
32 int host_gdt_entry_tls_min;
33
34 #ifdef CONFIG_MODE_SKAS
35 int do_set_thread_area_skas(struct user_desc *info)
36 {
37         int ret;
38         u32 cpu;
39
40         cpu = get_cpu();
41         ret = os_set_thread_area(info, userspace_pid[cpu]);
42         put_cpu();
43         return ret;
44 }
45
46 int do_get_thread_area_skas(struct user_desc *info)
47 {
48         int ret;
49         u32 cpu;
50
51         cpu = get_cpu();
52         ret = os_get_thread_area(info, userspace_pid[cpu]);
53         put_cpu();
54         return ret;
55 }
56 #endif
57
58 /*
59  * sys_get_thread_area: get a yet unused TLS descriptor index.
60  * XXX: Consider leaving one free slot for glibc usage at first place. This must
61  * be done here (and by changing GDT_ENTRY_TLS_* macros) and nowhere else.
62  *
63  * Also, this must be tested when compiling in SKAS mode with dinamic linking
64  * and running against NPTL.
65  */
66 static int get_free_idx(struct task_struct* task)
67 {
68         struct thread_struct *t = &task->thread;
69         int idx;
70
71         if (!t->arch.tls_array)
72                 return GDT_ENTRY_TLS_MIN;
73
74         for (idx = 0; idx < GDT_ENTRY_TLS_ENTRIES; idx++)
75                 if (!t->arch.tls_array[idx].present)
76                         return idx + GDT_ENTRY_TLS_MIN;
77         return -ESRCH;
78 }
79
80 static inline void clear_user_desc(struct user_desc* info)
81 {
82         /* Postcondition: LDT_empty(info) returns true. */
83         memset(info, 0, sizeof(*info));
84
85         /* Check the LDT_empty or the i386 sys_get_thread_area code - we obtain
86          * indeed an empty user_desc.
87          */
88         info->read_exec_only = 1;
89         info->seg_not_present = 1;
90 }
91
92 #define O_FORCE 1
93
94 static int load_TLS(int flags, struct task_struct *to)
95 {
96         int ret = 0;
97         int idx;
98
99         for (idx = GDT_ENTRY_TLS_MIN; idx < GDT_ENTRY_TLS_MAX; idx++) {
100                 struct uml_tls_struct* curr = &to->thread.arch.tls_array[idx - GDT_ENTRY_TLS_MIN];
101
102                 /* Actually, now if it wasn't flushed it gets cleared and
103                  * flushed to the host, which will clear it.*/
104                 if (!curr->present) {
105                         if (!curr->flushed) {
106                                 clear_user_desc(&curr->tls);
107                                 curr->tls.entry_number = idx;
108                         } else {
109                                 WARN_ON(!LDT_empty(&curr->tls));
110                                 continue;
111                         }
112                 }
113
114                 if (!(flags & O_FORCE) && curr->flushed)
115                         continue;
116
117                 ret = do_set_thread_area(&curr->tls);
118                 if (ret)
119                         goto out;
120
121                 curr->flushed = 1;
122         }
123 out:
124         return ret;
125 }
126
127 /* Verify if we need to do a flush for the new process, i.e. if there are any
128  * present desc's, only if they haven't been flushed.
129  */
130 static inline int needs_TLS_update(struct task_struct *task)
131 {
132         int i;
133         int ret = 0;
134
135         for (i = GDT_ENTRY_TLS_MIN; i < GDT_ENTRY_TLS_MAX; i++) {
136                 struct uml_tls_struct* curr = &task->thread.arch.tls_array[i - GDT_ENTRY_TLS_MIN];
137
138                 /* Can't test curr->present, we may need to clear a descriptor
139                  * which had a value. */
140                 if (curr->flushed)
141                         continue;
142                 ret = 1;
143                 break;
144         }
145         return ret;
146 }
147
148 /* On a newly forked process, the TLS descriptors haven't yet been flushed. So
149  * we mark them as such and the first switch_to will do the job.
150  */
151 void clear_flushed_tls(struct task_struct *task)
152 {
153         int i;
154
155         for (i = GDT_ENTRY_TLS_MIN; i < GDT_ENTRY_TLS_MAX; i++) {
156                 struct uml_tls_struct* curr = &task->thread.arch.tls_array[i - GDT_ENTRY_TLS_MIN];
157
158                 /* Still correct to do this, if it wasn't present on the host it
159                  * will remain as flushed as it was. */
160                 if (!curr->present)
161                         continue;
162
163                 curr->flushed = 0;
164         }
165 }
166
167 /* In SKAS0 mode, currently, multiple guest threads sharing the same ->mm have a
168  * common host process. So this is needed in SKAS0 too.
169  *
170  * However, if each thread had a different host process (and this was discussed
171  * for SMP support) this won't be needed.
172  *
173  * And this will not need be used when (and if) we'll add support to the host
174  * SKAS patch. */
175
176 int arch_switch_tls_skas(struct task_struct *from, struct task_struct *to)
177 {
178         if (!host_supports_tls)
179                 return 0;
180
181         /* We have no need whatsoever to switch TLS for kernel threads; beyond
182          * that, that would also result in us calling os_set_thread_area with
183          * userspace_pid[cpu] == 0, which gives an error. */
184         if (likely(to->mm))
185                 return load_TLS(O_FORCE, to);
186
187         return 0;
188 }
189
190 int arch_switch_tls_tt(struct task_struct *from, struct task_struct *to)
191 {
192         if (!host_supports_tls)
193                 return 0;
194
195         if (needs_TLS_update(to))
196                 return load_TLS(0, to);
197
198         return 0;
199 }
200
201 static int set_tls_entry(struct task_struct* task, struct user_desc *info,
202                          int idx, int flushed)
203 {
204         struct thread_struct *t = &task->thread;
205
206         if (idx < GDT_ENTRY_TLS_MIN || idx > GDT_ENTRY_TLS_MAX)
207                 return -EINVAL;
208
209         t->arch.tls_array[idx - GDT_ENTRY_TLS_MIN].tls = *info;
210         t->arch.tls_array[idx - GDT_ENTRY_TLS_MIN].present = 1;
211         t->arch.tls_array[idx - GDT_ENTRY_TLS_MIN].flushed = flushed;
212
213         return 0;
214 }
215
216 int arch_copy_tls(struct task_struct *new)
217 {
218         struct user_desc info;
219         int idx, ret = -EFAULT;
220
221         if (copy_from_user(&info,
222                            (void __user *) UPT_ESI(&new->thread.regs.regs),
223                            sizeof(info)))
224                 goto out;
225
226         ret = -EINVAL;
227         if (LDT_empty(&info))
228                 goto out;
229
230         idx = info.entry_number;
231
232         ret = set_tls_entry(new, &info, idx, 0);
233 out:
234         return ret;
235 }
236
237 /* XXX: use do_get_thread_area to read the host value? I'm not at all sure! */
238 static int get_tls_entry(struct task_struct* task, struct user_desc *info, int idx)
239 {
240         struct thread_struct *t = &task->thread;
241
242         if (!t->arch.tls_array)
243                 goto clear;
244
245         if (idx < GDT_ENTRY_TLS_MIN || idx > GDT_ENTRY_TLS_MAX)
246                 return -EINVAL;
247
248         if (!t->arch.tls_array[idx - GDT_ENTRY_TLS_MIN].present)
249                 goto clear;
250
251         *info = t->arch.tls_array[idx - GDT_ENTRY_TLS_MIN].tls;
252
253 out:
254         /* Temporary debugging check, to make sure that things have been
255          * flushed. This could be triggered if load_TLS() failed.
256          */
257         if (unlikely(task == current && !t->arch.tls_array[idx - GDT_ENTRY_TLS_MIN].flushed)) {
258                 printk(KERN_ERR "get_tls_entry: task with pid %d got here "
259                                 "without flushed TLS.", current->pid);
260         }
261
262         return 0;
263 clear:
264         /* When the TLS entry has not been set, the values read to user in the
265          * tls_array are 0 (because it's cleared at boot, see
266          * arch/i386/kernel/head.S:cpu_gdt_table). Emulate that.
267          */
268         clear_user_desc(info);
269         info->entry_number = idx;
270         goto out;
271 }
272
273 asmlinkage int sys_set_thread_area(struct user_desc __user *user_desc)
274 {
275         struct user_desc info;
276         int idx, ret;
277
278         if (!host_supports_tls)
279                 return -ENOSYS;
280
281         if (copy_from_user(&info, user_desc, sizeof(info)))
282                 return -EFAULT;
283
284         idx = info.entry_number;
285
286         if (idx == -1) {
287                 idx = get_free_idx(current);
288                 if (idx < 0)
289                         return idx;
290                 info.entry_number = idx;
291                 /* Tell the user which slot we chose for him.*/
292                 if (put_user(idx, &user_desc->entry_number))
293                         return -EFAULT;
294         }
295
296         ret = CHOOSE_MODE_PROC(do_set_thread_area_tt, do_set_thread_area_skas, &info);
297         if (ret)
298                 return ret;
299         return set_tls_entry(current, &info, idx, 1);
300 }
301
302 /*
303  * Perform set_thread_area on behalf of the traced child.
304  * Note: error handling is not done on the deferred load, and this differ from
305  * i386. However the only possible error are caused by bugs.
306  */
307 int ptrace_set_thread_area(struct task_struct *child, int idx,
308                 struct user_desc __user *user_desc)
309 {
310         struct user_desc info;
311
312         if (!host_supports_tls)
313                 return -EIO;
314
315         if (copy_from_user(&info, user_desc, sizeof(info)))
316                 return -EFAULT;
317
318         return set_tls_entry(child, &info, idx, 0);
319 }
320
321 asmlinkage int sys_get_thread_area(struct user_desc __user *user_desc)
322 {
323         struct user_desc info;
324         int idx, ret;
325
326         if (!host_supports_tls)
327                 return -ENOSYS;
328
329         if (get_user(idx, &user_desc->entry_number))
330                 return -EFAULT;
331
332         ret = get_tls_entry(current, &info, idx);
333         if (ret < 0)
334                 goto out;
335
336         if (copy_to_user(user_desc, &info, sizeof(info)))
337                 ret = -EFAULT;
338
339 out:
340         return ret;
341 }
342
343 /*
344  * Perform get_thread_area on behalf of the traced child.
345  */
346 int ptrace_get_thread_area(struct task_struct *child, int idx,
347                 struct user_desc __user *user_desc)
348 {
349         struct user_desc info;
350         int ret;
351
352         if (!host_supports_tls)
353                 return -EIO;
354
355         ret = get_tls_entry(child, &info, idx);
356         if (ret < 0)
357                 goto out;
358
359         if (copy_to_user(user_desc, &info, sizeof(info)))
360                 ret = -EFAULT;
361 out:
362         return ret;
363 }
364
365
366 /* XXX: This part is probably common to i386 and x86-64. Don't create a common
367  * file for now, do that when implementing x86-64 support.*/
368 static int __init __setup_host_supports_tls(void)
369 {
370         check_host_supports_tls(&host_supports_tls, &host_gdt_entry_tls_min);
371         if (host_supports_tls) {
372                 printk(KERN_INFO "Host TLS support detected\n");
373                 printk(KERN_INFO "Detected host type: ");
374                 switch (host_gdt_entry_tls_min) {
375                         case GDT_ENTRY_TLS_MIN_I386:
376                                 printk("i386\n");
377                                 break;
378                         case GDT_ENTRY_TLS_MIN_X86_64:
379                                 printk("x86_64\n");
380                                 break;
381                 }
382         } else
383                 printk(KERN_ERR "  Host TLS support NOT detected! "
384                                 "TLS support inside UML will not work\n");
385         return 0;
386 }
387
388 __initcall(__setup_host_supports_tls);