SOBLIV: Drain budget while task is in top-m only.

Also fixed numerous bugs...

1 files changed, 23 insertions, 0 deletions

diff --git a/litmus/bheap.c b/litmus/bheap.c
index 45e1db36fa36..403c09cc9e81 100644
--- a/litmus/bheap.c
+++ b/litmus/bheap.c
@@ -21,6 +21,29 @@ void bheap_node_init(struct bheap_node** _h, void* value)
 }
 
 
+static void __bheap_for_all(struct bheap_node *h, bheap_for_all_t fn, void* args)
+{
+        /* pre-order */
+        fn(h, args);
+
+        /* depth-first */
+        if (h->child)
+                __bheap_for_all(h->child, fn, args);
+        if (h->next)
+                __bheap_for_all(h->next, fn, args);
+}
+
+void bheap_for_all(struct bheap* heap, bheap_for_all_t fn, void* args)
+{
+        struct bheap_node *head;
+
+        BUG_ON(!heap);
+        BUG_ON(!fn);
+
+        head = heap->head;
+        __bheap_for_all(head, fn, args);
+}
+
 /* make child a subtree of root */
 static void __bheap_link(struct bheap_node* root,
                         struct bheap_node* child)