Remove CONFIG_KTHREAD_POISON
[akaros.git] / kern / src / kthread.c
index afd7544..518f37a 100644 (file)
 #include <smp.h>
 #include <schedule.h>
 #include <kstack.h>
+#include <kmalloc.h>
 #include <arch/uaccess.h>
 
+#define KSTACK_NR_GUARD_PGS            1
+#define KSTACK_GUARD_SZ                        (KSTACK_NR_GUARD_PGS * PGSIZE)
+static struct kmem_cache *kstack_cache;
+
+/* We allocate KSTKSIZE + PGSIZE vaddrs.  So for one-page stacks, we get two
+ * pages.  blob points to the bottom of this space.  Our job is to allocate the
+ * physical pages for the stack and set up the virtual-to-physical mappings. */
+int kstack_ctor(void *blob, void *priv, int flags)
+{
+       void *stackbot;
+
+       stackbot = kpages_alloc(KSTKSIZE, flags);
+       if (!stackbot)
+               return -1;
+       if (map_vmap_segment((uintptr_t)blob, 0x123456000, KSTACK_NR_GUARD_PGS,
+                                PTE_NONE))
+               goto error;
+       if (map_vmap_segment((uintptr_t)blob + KSTACK_GUARD_SZ, PADDR(stackbot),
+                                KSTKSIZE / PGSIZE, PTE_KERN_RW))
+               goto error;
+       return 0;
+error:
+       /* On failure, we only need to undo what our dtor would do.  The unmaps
+        * happen in the vmap_arena ffunc. */
+       kpages_free(stackbot, KSTKSIZE);
+       return -1;
+}
+
+/* The vmap_arena free will unmap the vaddrs on its own.  We just need to free
+ * the physical memory we allocated in ctor.  Although we still have mappings
+ * and TLB entries pointing to the memory after we free it (and thus it can be
+ * reused), this is no more dangerous than just freeing the stack.  Errant
+ * pointers into an old kstack are still dangerous. */
+void kstack_dtor(void *blob, void *priv)
+{
+       void *stackbot;
+       pte_t pte;
+
+       pte = pgdir_walk(boot_pgdir, blob + KSTACK_GUARD_SZ, 0);
+       assert(pte_walk_okay(pte));
+       stackbot = KADDR(pte_get_paddr(pte));
+       kpages_free(stackbot, KSTKSIZE);
+}
+
 uintptr_t get_kstack(void)
 {
-       uintptr_t stackbot;
-       if (KSTKSIZE == PGSIZE)
-               stackbot = (uintptr_t)kpage_alloc_addr();
-       else
-               stackbot = (uintptr_t)get_cont_pages(KSTKSHIFT - PGSHIFT, 0);
-       assert(stackbot);
-       return stackbot + KSTKSIZE;
+       void *blob;
+
+       blob = kmem_cache_alloc(kstack_cache, MEM_ATOMIC);
+       /* TODO: think about MEM_WAIT within kthread/blocking code. */
+       assert(blob);
+       return (uintptr_t)blob + KSTKSIZE + KSTACK_GUARD_SZ;
 }
 
 void put_kstack(uintptr_t stacktop)
 {
-       uintptr_t stackbot = stacktop - KSTKSIZE;
-       if (KSTKSIZE == PGSIZE)
-               page_decref(kva2page((void*)stackbot));
-       else
-               free_cont_pages((void*)stackbot, KSTKSHIFT - PGSHIFT);
+       kmem_cache_free(kstack_cache, (void*)(stacktop - KSTKSIZE
+                                             - KSTACK_GUARD_SZ));
 }
 
 uintptr_t *kstack_bottom_addr(uintptr_t stacktop)
@@ -46,7 +87,11 @@ struct kmem_cache *kthread_kcache;
 void kthread_init(void)
 {
        kthread_kcache = kmem_cache_create("kthread", sizeof(struct kthread),
-                                          __alignof__(struct kthread), 0, 0, 0);
+                                          __alignof__(struct kthread), 0,
+                                          NULL, 0, 0, NULL);
+       kstack_cache = kmem_cache_create("kstack", KSTKSIZE + KSTACK_GUARD_SZ,
+                                        PGSIZE, 0, vmap_arena, kstack_ctor,
+                                                                        kstack_dtor, NULL);
 }
 
 /* Used by early init routines (smp_boot, etc) */
@@ -59,6 +104,21 @@ struct kthread *__kthread_zalloc(void)
        return kthread;
 }
 
+/* Helper during early boot, where we jump from the bootstack to a real kthread
+ * stack, then run f().  Note that we don't have a kthread yet (done in smp.c).
+ *
+ * After this, our callee (f) can free the bootstack, if we care, by adding it
+ * to the base arena (use the KERNBASE addr, not the KERN_LOAD_ADDR). */
+void __use_real_kstack(void (*f)(void *arg))
+{
+       struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
+       uintptr_t new_stacktop;
+
+       new_stacktop = get_kstack();
+       set_stack_top(new_stacktop);
+       __reset_stack_pointer(0, new_stacktop, f);
+}
+
 /* Starts kthread on the calling core.  This does not return, and will handle
  * the details of cleaning up whatever is currently running (freeing its stack,
  * etc).  Pairs with sem_down(). */
@@ -86,16 +146,6 @@ void restart_kthread(struct kthread *kthread)
        /* When a kthread runs, its stack is the default kernel stack */
        set_stack_top(kthread->stacktop);
        pcpui->cur_kthread = kthread;
-#ifdef CONFIG_KTHREAD_POISON
-       /* Assert and switch to cur stack not in use, kthr stack in use */
-       uintptr_t *cur_stack_poison, *kth_stack_poison;
-       cur_stack_poison = kstack_bottom_addr(current_stacktop);
-       assert(*cur_stack_poison == 0xdeadbeef);
-       *cur_stack_poison = 0;
-       kth_stack_poison = kstack_bottom_addr(kthread->stacktop);
-       assert(!*kth_stack_poison);
-       *kth_stack_poison = 0xdeadbeef;
-#endif /* CONFIG_KTHREAD_POISON */
        /* Only change current if we need to (the kthread was in process context) */
        if (kthread->proc) {
                if (kthread->proc == pcpui->cur_proc) {
@@ -244,18 +294,6 @@ void ktask(char *name, void (*fn)(void*), void *arg)
                            (long)name, KMSG_ROUTINE);
 }
 
-void check_poison(char *msg)
-{
-#ifdef CONFIG_KTHREAD_POISON
-       struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
-       if (pcpui->cur_kthread && pcpui->cur_kthread->stacktop &&
-           (*kstack_bottom_addr(pcpui->cur_kthread->stacktop) != 0xdeadbeef)) {
-               printk("\nBad kthread canary, msg: %s\n", msg);
-               panic("");
-       }
-#endif /* CONFIG_KTHREAD_POISON */
-}
-
 /* Semaphores, using kthreads directly */
 static void debug_downed_sem(struct semaphore *sem);
 static void debug_upped_sem(struct semaphore *sem);
@@ -327,7 +365,6 @@ void sem_down(struct semaphore *sem)
        /* Make sure we aren't holding any locks (only works if SPINLOCK_DEBUG) */
        if (pcpui->lock_depth)
                panic("Kthread tried to sleep, with lockdepth %d\n", pcpui->lock_depth);
-       assert(pcpui->cur_kthread);
        /* Try to down the semaphore.  If there is a signal there, we can skip all
         * of the sleep prep and just return. */
 #ifdef CONFIG_SEM_SPINWAIT
@@ -340,6 +377,7 @@ void sem_down(struct semaphore *sem)
        if (sem_trydown(sem))
                goto block_return_path;
 #endif
+       assert(pcpui->cur_kthread);
        /* We're probably going to sleep, so get ready.  We'll check again later. */
        kthread = pcpui->cur_kthread;
        /* We need to have a spare slot for restart, so we also use it when
@@ -364,23 +402,10 @@ void sem_down(struct semaphore *sem)
                new_kthread->flags = KTH_DEFAULT_FLAGS;
                new_stacktop = get_kstack();
                new_kthread->stacktop = new_stacktop;
-#ifdef CONFIG_KTHREAD_POISON
-               *kstack_bottom_addr(new_stacktop) = 0;
-#endif /* CONFIG_KTHREAD_POISON */
        }
        /* Set the core's new default stack and kthread */
        set_stack_top(new_stacktop);
        pcpui->cur_kthread = new_kthread;
-#ifdef CONFIG_KTHREAD_POISON
-       /* Mark the new stack as in-use, and unmark the current kthread */
-       uintptr_t *new_stack_poison, *kth_stack_poison;
-       new_stack_poison = kstack_bottom_addr(new_stacktop);
-       assert(!*new_stack_poison);
-       *new_stack_poison = 0xdeadbeef;
-       kth_stack_poison = kstack_bottom_addr(kthread->stacktop);
-       assert(*kth_stack_poison == 0xdeadbeef);
-       *kth_stack_poison = 0;
-#endif /* CONFIG_KTHREAD_POISON */
        /* Kthreads that are ktasks are not related to any process, and do not need
         * to work in a process's address space.  They can operate in any address
         * space that has the kernel mapped (like boot_pgdir, or any pgdir).  Some
@@ -433,11 +458,6 @@ void sem_down(struct semaphore *sem)
        /* Save the allocs as the spare */
        assert(!pcpui->spare);
        pcpui->spare = new_kthread;
-#ifdef CONFIG_KTHREAD_POISON
-       /* switch back to old stack in use, new one not */
-       *new_stack_poison = 0;
-       *kth_stack_poison = 0xdeadbeef;
-#endif /* CONFIG_KTHREAD_POISON */
 block_return_path:
        printd("[kernel] Returning from being 'blocked'! at %llu\n", read_tsc());
        /* restart_kthread and longjmp did not reenable IRQs.  We need to make sure