Provide interface to let litmus ask new pages by bank number

1 files changed, 37 insertions, 3 deletions

diff --git a/litmus/bank_proc.c b/litmus/bank_proc.c
index 2c69657b53bd..ec04626b43ec 100644
--- a/litmus/bank_proc.c
+++ b/litmus/bank_proc.c
@@ -131,7 +131,6 @@ out:
  * node = 3 for Level A, B tasks in Cpu 3
  * node = 4 for Level C tasks 
  */
- #if 1 
 struct page *new_alloc_page(struct page *page, unsigned long node, int **x)
 {
         printk("allocate new page node = %d\n", node);  
@@ -194,9 +193,44 @@ out_unlock:
 out:
         do_add_pages();
         return rPage;
-        //return  alloc_page(GFP_HIGHUSER_MOVABLE);
 }
-#endif
+
+struct page *new_alloc_page_banknr(struct page *page, unsigned long banknr, int **x)
+{
+        printk("allocate new page bank = %d\n", banknr);        
+        struct color_group *cgroup;
+        struct page *rPage = NULL;
+        unsigned int color;
+        get_random_bytes(&color, sizeof(unsigned int));
+        
+        if((banknr<= 7) && (banknr>=0)){
+                color = (color%2) * 8 + banknr;
+        }else{
+                goto out;
+        }
+                
+        cgroup = &color_groups[color];
+        spin_lock(&cgroup->lock);
+        if (unlikely(!atomic_read(&cgroup->nr_pages))) {
+                TRACE_CUR("No free %lu colored pages.\n", color);
+                printk(KERN_WARNING "no free %lu colored pages.\n", color);
+                goto out_unlock;
+        }
+        rPage = list_first_entry(&cgroup->list, struct page, lru);
+        BUG_ON(page_count(rPage) > 1);
+        get_page(rPage);
+        list_del(&rPage->lru);
+        atomic_dec(&cgroup->nr_pages);
+//      ClearPageLRU(rPage);
+out_unlock:
+        spin_unlock(&cgroup->lock);
+out:
+        do_add_pages();
+        return rPage;
+
+
+
+}
 
 static int __init init_variables(void)
 {