lkdtm: do not leak free page on kmalloc failure

This frees the allocated page if there is a kmalloc failure.

Signed-off-by: Kees Cook <keescook@chromium.org>

diff --git a/drivers/misc/lkdtm.c b/drivers/misc/lkdtm.c
index 5b3a63c87ec9758820a8d255ea8173a6ef4554fb..0a5cbbe12452da62127af8ced48101bb9fc212b8 100644 (file)
--- a/drivers/misc/lkdtm.c
+++ b/drivers/misc/lkdtm.c
@@ -507,8 +507,10 @@ static void lkdtm_do_action(enum ctype which)
                        break;
 
                val = kmalloc(1024, GFP_KERNEL);
-               if (!val)
+               if (!val) {
+                       free_page(p);
                        break;
+               }
 
                base = (int *)p;