Serialize printing during panic()
[akaros.git] / kern / src / kthread.c
index 518f37a..e1010ca 100644 (file)
@@ -126,7 +126,7 @@ void restart_kthread(struct kthread *kthread)
 {
        struct per_cpu_info *pcpui = &per_cpu_info[core_id()];
        uintptr_t current_stacktop;
-       struct kthread *current_kthread;
+       struct kthread *cur_kth;
        /* Avoid messy complications.  The kthread will enable_irqsave() when it
         * comes back up. */
        disable_irq();
@@ -138,11 +138,11 @@ void restart_kthread(struct kthread *kthread)
                put_kstack(pcpui->spare->stacktop);
                kmem_cache_free(kthread_kcache, pcpui->spare);
        }
-       current_kthread = pcpui->cur_kthread;
-       current_stacktop = current_kthread->stacktop;
-       assert(!current_kthread->sysc); /* catch bugs, prev user should clear */
+       cur_kth = pcpui->cur_kthread;
+       current_stacktop = cur_kth->stacktop;
+       assert(!cur_kth->sysc); /* catch bugs, prev user should clear */
        /* Set the spare stuff (current kthread, which includes its stacktop) */
-       pcpui->spare = current_kthread;
+       pcpui->spare = cur_kth;
        /* When a kthread runs, its stack is the default kernel stack */
        set_stack_top(kthread->stacktop);
        pcpui->cur_kthread = kthread;
@@ -154,7 +154,13 @@ void restart_kthread(struct kthread *kthread)
                        proc_decref(kthread->proc);
                        kthread->proc = 0;
                } else {
-                       /* Load our page tables before potentially decreffing cur_proc */
+                       /* Load our page tables before potentially decreffing cur_proc.
+                        *
+                        * We don't need to do an EPT flush here.  The EPT is flushed and
+                        * managed in sync with the VMCS.  We won't run a different VM (and
+                        * thus *need* a different EPT) without first removing the old GPC,
+                        * which ultimately will result in a flushed EPT (on x86, this
+                        * actually happens when we clear_owning_proc()). */
                        lcr3(kthread->proc->env_cr3);
                        /* Might have to clear out an existing current.  If they need to be
                         * set later (like in restartcore), it'll be done on demand. */
@@ -323,15 +329,16 @@ void sem_init_irqsave(struct semaphore *sem, int signals)
        sem->irq_okay = TRUE;
 }
 
-bool sem_trydown(struct semaphore *sem)
+bool sem_trydown_bulk(struct semaphore *sem, int nr_signals)
 {
        bool ret = FALSE;
+
        /* lockless peek */
-       if (sem->nr_signals <= 0)
+       if (sem->nr_signals - nr_signals < 0)
                return ret;
        debug_lock_semlist();
        spin_lock(&sem->lock);
-       if (sem->nr_signals > 0) {
+       if (sem->nr_signals - nr_signals >= 0) {
                sem->nr_signals--;
                ret = TRUE;
                debug_downed_sem(sem);
@@ -341,6 +348,11 @@ bool sem_trydown(struct semaphore *sem)
        return ret;
 }
 
+bool sem_trydown(struct semaphore *sem)
+{
+       return sem_trydown_bulk(sem, 1);
+}
+
 /* Bottom-half of sem_down.  This is called after we jumped to the new stack. */
 static void __attribute__((noreturn)) __unlock_and_idle(void *arg)
 {
@@ -429,7 +441,8 @@ void sem_down(struct semaphore *sem)
                goto block_return_path;
        debug_lock_semlist();
        spin_lock(&sem->lock);
-       if (sem->nr_signals-- <= 0) {
+       sem->nr_signals -= 1;
+       if (sem->nr_signals < 0) {
                TAILQ_INSERT_TAIL(&sem->waiters, kthread, link);
                debug_downed_sem(sem);  /* need to debug after inserting */
                /* At this point, we know we'll sleep and change stacks.  Once we unlock
@@ -469,6 +482,20 @@ block_return_path:
        return;
 }
 
+void sem_down_bulk(struct semaphore *sem, int nr_signals)
+{
+       /* This is far from ideal.  Our current sem code expects a 1:1 pairing of
+        * signals to waiters.  For instance, if we have 10 waiters of -1 each or 1
+        * waiter of -10, we can't tell from looking at the overall structure.  We'd
+        * need to track the desired number of signals per waiter.
+        *
+        * Note that if there are a bunch of signals available, sem_down will
+        * quickly do a try_down and return, so we won't block repeatedly.  But if
+        * we do block, we could wake up N times. */
+       for (int i = 0; i < nr_signals; i++)
+               sem_down(sem);
+}
+
 /* Ups the semaphore.  If it was < 0, we need to wake up someone, which we do.
  * Returns TRUE if we woke someone, FALSE o/w (used for debugging in some
  * places).  If we need more control, we can implement a version of the old
@@ -500,22 +527,35 @@ bool sem_up(struct semaphore *sem)
        return FALSE;
 }
 
-bool sem_trydown_irqsave(struct semaphore *sem, int8_t *irq_state)
+bool sem_trydown_bulk_irqsave(struct semaphore *sem, int nr_signals,
+                              int8_t *irq_state)
 {
        bool ret;
+
        disable_irqsave(irq_state);
-       ret = sem_trydown(sem);
+       ret = sem_trydown_bulk(sem, nr_signals);
        enable_irqsave(irq_state);
        return ret;
 }
 
-void sem_down_irqsave(struct semaphore *sem, int8_t *irq_state)
+bool sem_trydown_irqsave(struct semaphore *sem, int8_t *irq_state)
+{
+       return sem_trydown_bulk_irqsave(sem, 1, irq_state);
+}
+
+void sem_down_bulk_irqsave(struct semaphore *sem, int nr_signals,
+                           int8_t *irq_state)
 {
        disable_irqsave(irq_state);
-       sem_down(sem);
+       sem_down_bulk(sem, nr_signals);
        enable_irqsave(irq_state);
 }
 
+void sem_down_irqsave(struct semaphore *sem, int8_t *irq_state)
+{
+       sem_down_bulk_irqsave(sem, 1, irq_state);
+}
+
 bool sem_up_irqsave(struct semaphore *sem, int8_t *irq_state)
 {
        bool retval;
@@ -587,11 +627,34 @@ static void debug_upped_sem(struct semaphore *sem)
 
 #endif /* CONFIG_SEMAPHORE_DEBUG */
 
-void print_sem_info(struct semaphore *sem)
+static bool __sem_has_pid(struct semaphore *sem, pid_t pid)
 {
        struct kthread *kth_i;
+
+       if (pid == -1)
+               return TRUE;
+       TAILQ_FOREACH(kth_i, &sem->waiters, link) {
+               if (kth_i->proc) {
+                       if (kth_i->proc->pid == pid)
+                               return TRUE;
+               } else {
+                       if (pid == 0)
+                               return TRUE;
+               }
+       }
+       return FALSE;
+}
+
+static void print_sem_info(struct semaphore *sem, pid_t pid)
+{
+       struct kthread *kth_i;
+
        /* Always safe to irqsave */
        spin_lock_irqsave(&sem->lock);
+       if (!__sem_has_pid(sem, pid)) {
+               spin_unlock_irqsave(&sem->lock);
+               return;
+       }
        printk("Semaphore %p has %d signals (neg = waiters)\n", sem,
               sem->nr_signals);
        TAILQ_FOREACH(kth_i, &sem->waiters, link)
@@ -603,14 +666,14 @@ void print_sem_info(struct semaphore *sem)
        spin_unlock_irqsave(&sem->lock);
 }
 
-void print_all_sem_info(void)
+void print_all_sem_info(pid_t pid)
 {
 #ifdef CONFIG_SEMAPHORE_DEBUG
        struct semaphore *sem_i;
        printk("All sems with waiters:\n");
        spin_lock_irqsave(&sems_with_waiters_lock);
        TAILQ_FOREACH(sem_i, &sems_with_waiters, link)
-               print_sem_info(sem_i);
+               print_sem_info(sem_i, pid);
        spin_unlock_irqsave(&sems_with_waiters_lock);
 #else
        printk("Failed to print all sems: build with CONFIG_SEMAPHORE_DEBUG\n");