x86 thread local storage tweaks
[akaros.git] / kern / arch / i386 / env.c
1 /* See COPYRIGHT for copyright information. */
2 #ifdef __SHARC__
3 #pragma nosharc
4 #endif
5
6 #include <arch/trap.h>
7 #include <env.h>
8 #include <assert.h>
9 #include <pmap.h>
10
11 //
12 // This exits the kernel and starts executing some environment's code.
13 // This function does not return.
14 // Uses 'iret' or 'sysexit' depending on CS.
15 //
16 void env_pop_tf(trapframe_t *tf)
17 {
18         /* Load the LDT for this process.  Slightly ghetto doing it here. */
19         segdesc_t *my_gdt = per_cpu_info[core_id()].gdt;
20         /* copy-in and check the LDT location.  the segmentation hardware write the
21          * accessed bit, so we want the memory to be in the user-writeable area. */
22         segdesc_t *ldt = current->env_procdata->ldt;
23         ldt = (segdesc_t*)MIN((uintptr_t)ldt, UTOP - LDT_SIZE);
24         segdesc_t ldt_temp = SEG_SYS(STS_LDT, (uint32_t)ldt, LDT_SIZE, 3);
25         my_gdt[GD_LDT >> 3] = ldt_temp;
26         asm volatile("lldt %%ax" :: "a"(GD_LDT));
27
28         /* In case they are enabled elsewhere.  We can't take an interrupt in these
29          * routines, due to how they play with the kernel stack pointer. */
30         disable_irq();
31         /*
32          * If the process entered the kernel via sysenter, we need to leave via
33          * sysexit.  sysenter trapframes have 0 for a CS, which is pushed in
34          * sysenter_handler.
35          */
36         if(tf->tf_cs) {
37                 /*
38                  * Restores the register values in the Trapframe with the 'iret'
39                  * instruction.  This exits the kernel and starts executing some
40                  * environment's code.  This function does not return.
41                  */
42                 asm volatile ("movl %0,%%esp;           "
43                               "popal;                   "
44                               "popl %%gs;               "
45                               "popl %%fs;               "
46                               "popl %%es;               "
47                               "popl %%ds;               "
48                               "addl $0x8,%%esp;         "
49                               "iret                     "
50                               : : "g" (tf) : "memory");
51                 panic("iret failed");  /* mostly to placate the compiler */
52         } else {
53                 /* Return path of sysexit.  See sysenter_handler's asm for details.
54                  * One difference is that this tf could be somewhere other than a stack
55                  * (like in a struct proc).  We need to make sure esp is valid once
56                  * interrupts are turned on (which would happen on popfl normally), so
57                  * we need to save and restore a decent esp (the current one).  We need
58                  * a place to save it that is accessible after we change the stack
59                  * pointer to the tf *and* that is specific to this core/instance of
60                  * sysexit.  The simplest and nicest is to use the tf_esp, which we
61                  * can just pop.  Incidentally, the value in oesp would work too.
62                  * To prevent popfl from turning interrupts on, we hack the tf's eflags
63                  * so that we have a chance to change esp to a good value before
64                  * interrupts are enabled.  The other option would be to throw away the
65                  * eflags, but that's less desirable. */
66                 tf->tf_eflags &= !FL_IF;
67                 tf->tf_esp = read_esp();
68                 asm volatile ("movl %0,%%esp;           "
69                               "popal;                   "
70                               "popl %%gs;               "
71                               "popl %%fs;               "
72                               "popl %%es;               "
73                               "popl %%ds;               "
74                               "addl $0x10,%%esp;        "
75                               "popfl;                   "
76                               "movl %%ebp,%%ecx;        "
77                               "movl %%esi,%%edx;        "
78                               "popl %%esp;              "
79                               "sti;                     "
80                               "sysexit                  "
81                               : : "g" (tf) : "memory");
82                 panic("sysexit failed");  /* mostly to placate your mom */
83         }
84 }
85
86 // Flush all mapped pages in the user portion of the address space
87 void
88 env_user_mem_free(env_t* e)
89 {
90         pte_t *pt;
91         uint32_t pdeno, pteno;
92         physaddr_t pa;
93
94         static_assert(UTOP % PTSIZE == 0);
95         for (pdeno = 0; pdeno < PDX(UTOP); pdeno++) {
96
97                 // only look at mapped page tables
98                 if (!(e->env_pgdir[pdeno] & PTE_P))
99                         continue;
100
101                 // find the pa and va of the page table
102                 pa = PTE_ADDR(e->env_pgdir[pdeno]);
103                 pt = (pte_t*COUNT(NPTENTRIES)) KADDR(pa);
104
105                 // unmap all PTEs in this page table 
106                 for (pteno = 0; pteno <= PTX(~0); pteno++) {
107                         if (pt[pteno] & PTE_P)
108                                 page_remove(e->env_pgdir, PGADDR(pdeno, pteno, 0));
109                 }
110
111                 // free the page table itself
112                 e->env_pgdir[pdeno] = 0;
113                 page_decref(pa2page(pa));
114         }
115 }