1 files changed, 105 insertions, 0 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 144e3bfdcfc1..cf599d1a6b3a 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -631,6 +631,111 @@ asmlinkage long sys_test_call(unsigned int param)
         return ret;
 }
 
+asmlinkage long sys_recolor_mem(void* mem, int n_pages, int cpu)
+{
+        long ret = 0;
+        struct vm_area_struct *vma_itr = NULL;
+        int nr_pages = 0, nr_failed = 0, nr_not_migrated = 0;
+        unsigned long node;
+        enum crit_level lv;
+        struct mm_struct *mm;
+        
+        LIST_HEAD(pagelist);
+        
+        printk(KERN_INFO "mem addr = %d\n", (unsigned long)mem);
+        return 0;
+        
+        migrate_prep();
+        
+        /* Find the current mm_struct */
+        rcu_read_lock();
+        get_task_struct(current);
+        rcu_read_unlock();
+        mm = get_task_mm(current);
+        put_task_struct(current);
+
+        down_read(&mm->mmap_sem);
+
+        vma_itr = mm->mmap;
+        /* Iterate all vm_area_struct */
+        while (vma_itr != NULL) {
+                unsigned int num_pages = 0, i;
+                struct page *old_page = NULL;
+                int pages_in_vma = 0;
+                
+                num_pages = (vma_itr->vm_end - vma_itr->vm_start) / PAGE_SIZE;
+                if (num_pages != n_pages)
+                        continue;
+                
+                /* Traverse all pages in vm_area_struct */
+                for (i = 0; i < num_pages; i++) {
+                        old_page = follow_page(vma_itr, vma_itr->vm_start + PAGE_SIZE*i, FOLL_GET|FOLL_SPLIT);
+                        
+                        if (IS_ERR(old_page))
+                                continue;
+                        if (!old_page)
+                                continue;
+
+                        if (PageReserved(old_page)) {
+                                TRACE("Reserved Page!\n");
+                                put_page(old_page);
+                                continue;
+                        }
+                        /*
+                        if (PageDirty(old_page)) {
+                                TRACE("Dirty Page!\n");
+                                put_page(old_page);
+                                continue;
+                        }
+                        */
+                        TRACE_TASK(current, "addr: %08x, pfn: %05lx, _mapcount: %d, _count: %d flags: %s%s%s\n", vma_itr->vm_start + PAGE_SIZE*i, page_to_pfn(old_page), page_mapcount(old_page), page_count(old_page), vma_itr->vm_flags&VM_READ?"r":"-", vma_itr->vm_flags&VM_WRITE?"w":"-", vma_itr->vm_flags&VM_EXEC?"x":"-");
+                        pages_in_vma++;
+
+                        /* Conditions for replicable pages */
+                        if (page_count(old_page) == 1) {
+                                ret = isolate_lru_page(old_page);
+                                if (!ret) {
+                                        list_add_tail(&old_page->lru, &pagelist);
+                                        inc_zone_page_state(old_page, NR_ISOLATED_ANON + !PageSwapBacked(old_page));
+                                        nr_pages++;
+                                } else if (!is_in_correct_bank(old_page, cpu)) {
+                                        TRACE_TASK(current, "isolate_lru_page for a private page failed\n");
+                                        nr_failed++;
+                                } else {
+                                        TRACE_TASK(current, "page is already in the correct bank\n");
+                                }
+                                put_page(old_page);
+                        }
+                }
+                TRACE_TASK(current, "PAGES_IN_VMA = %d size = %d KB\n", pages_in_vma, pages_in_vma*4);
+                vma_itr = vma_itr->vm_next;
+        }
+        
+        ret = 0;
+        lv = tsk_rt(current)->mc2_data->crit;
+        if (cpu == -1)
+                node = 8;
+        else
+                node = cpu*2 + lv;
+
+        /* Migrate private pages */
+        if (!list_empty(&pagelist)) {
+                ret = migrate_pages(&pagelist, alloc_colored_page, NULL, node, MIGRATE_SYNC, MR_SYSCALL);
+                TRACE_TASK(current, "%ld pages not migrated.\n", ret);
+                nr_not_migrated = ret;
+                if (ret) {
+                        putback_movable_pages(&pagelist);
+                }
+        }
+        
+        up_read(&mm->mmap_sem);
+
+        TRACE_TASK(current, "nr_pages = %d nr_failed = %d nr_not_migrated = %d\n", nr_pages, nr_failed, nr_not_migrated);
+        printk(KERN_INFO "node = %ld, nr_private_pages = %d, nr_failed_to_isolate_lru = %d, nr_not_migrated = %d\n", node, nr_pages, nr_failed, nr_not_migrated);
+        
+        return nr_not_migrated;
+}
+
 /* p is a real-time task. Re-init its state as a best-effort task. */
 static void reinit_litmus_state(struct task_struct* p, int restore)
 {