Add vm_map_ram()/vm_unmap_ram() test case to our stress test-suite. Signed-off-by: Uladzislau Rezki (Sony) <urezki@xxxxxxxxx> --- lib/test_vmalloc.c | 41 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) diff --git a/lib/test_vmalloc.c b/lib/test_vmalloc.c index cd2bdba6d3ed..6633eda4cd4d 100644 --- a/lib/test_vmalloc.c +++ b/lib/test_vmalloc.c @@ -53,6 +53,7 @@ __param(int, run_test_mask, INT_MAX, "\t\tid: 128, name: pcpu_alloc_test\n" "\t\tid: 256, name: kvfree_rcu_1_arg_vmalloc_test\n" "\t\tid: 512, name: kvfree_rcu_2_arg_vmalloc_test\n" + "\t\tid: 1024, name: vm_map_ram_test\n" /* Add a new test case description here. */ ); @@ -358,6 +359,45 @@ kvfree_rcu_2_arg_vmalloc_test(void) return 0; } +static int +vm_map_ram_test(void) +{ + unsigned int map_nr_pages; + unsigned char *v_ptr; + unsigned char *p_ptr; + struct page **pages; + struct page *page; + int i; + + map_nr_pages = nr_pages > 0 ? nr_pages:1; + pages = kmalloc(map_nr_pages * sizeof(*page), GFP_KERNEL); + if (!pages) + return -1; + + for (i = 0; i < map_nr_pages; i++) { + page = alloc_pages(GFP_KERNEL, 1); + if (!page) + return -1; + + pages[i] = page; + } + + /* Run the test loop. */ + for (i = 0; i < test_loop_count; i++) { + v_ptr = vm_map_ram(pages, map_nr_pages, -1); + *v_ptr = 'a'; + vm_unmap_ram(v_ptr, map_nr_pages); + } + + for (i = 0; i < map_nr_pages; i++) { + p_ptr = page_address(pages[i]); + free_pages((unsigned long)p_ptr, 1); + } + + kfree(pages); + return 0; +} + struct test_case_desc { const char *test_name; int (*test_func)(void); @@ -374,6 +414,7 @@ static struct test_case_desc test_case_array[] = { { "pcpu_alloc_test", pcpu_alloc_test }, { "kvfree_rcu_1_arg_vmalloc_test", kvfree_rcu_1_arg_vmalloc_test }, { "kvfree_rcu_2_arg_vmalloc_test", kvfree_rcu_2_arg_vmalloc_test }, + { "vm_map_ram_test", vm_map_ram_test }, /* Add a new test case here. */ }; -- 2.30.2