Re: [PATCH] lkdtm: Missing kmalloc check

[Kees Cook]

On Thu, Nov 23, 2017 at 8:00 AM, Vasyl Gomonovych <gomonovych@xxxxxxxxx> wrote:
> Handling a possible memory allocation failure.
>
> Signed-off-by: Vasyl Gomonovych <gomonovych@xxxxxxxxx>

Acked-by: Kees Cook <keescook@xxxxxxxxxxxx>

Greg, can you please take this?

-Kees

> ---
>  drivers/misc/lkdtm_heap.c | 4 ++++
>  1 file changed, 4 insertions(+)
>
> diff --git a/drivers/misc/lkdtm_heap.c b/drivers/misc/lkdtm_heap.c
> index f5494a6d4be5..65026d7de130 100644
> --- a/drivers/misc/lkdtm_heap.c
> +++ b/drivers/misc/lkdtm_heap.c
> @@ -16,6 +16,8 @@ void lkdtm_OVERWRITE_ALLOCATION(void)
>  {
>         size_t len = 1020;
>         u32 *data = kmalloc(len, GFP_KERNEL);
> +       if (!data)
> +               return;
>
>         data[1024 / sizeof(u32)] = 0x12345678;
>         kfree(data);
> @@ -33,6 +35,8 @@ void lkdtm_WRITE_AFTER_FREE(void)
>         size_t offset = (len / sizeof(*base)) / 2;
>
>         base = kmalloc(len, GFP_KERNEL);
> +       if (!base)
> +               return;
>         pr_info("Allocated memory %p-%p\n", base, &base[offset * 2]);
>         pr_info("Attempting bad write to freed memory at %p\n",
>                 &base[offset]);
> --
> 1.9.1
>



-- 
Kees Cook
Pixel Security