livepatch: check kzalloc return values

[linux-2.6-microblaze.git] / samples / livepatch / livepatch-shadow-fix1.c

diff --git a/samples/livepatch/livepatch-shadow-fix1.c b/samples/livepatch/livepatch-shadow-fix1.c
index 49b1355..e8f1bd6 100644 (file)
--- a/samples/livepatch/livepatch-shadow-fix1.c
+++ b/samples/livepatch/livepatch-shadow-fix1.c
@@ -89,6 +89,11 @@ struct dummy *livepatch_fix1_dummy_alloc(void)
         * pointer to handle resource release.
         */
        leak = kzalloc(sizeof(int), GFP_KERNEL);
+       if (!leak) {
+               kfree(d);
+               return NULL;
+       }
+
        klp_shadow_alloc(d, SV_LEAK, sizeof(leak), GFP_KERNEL,
                         shadow_leak_ctor, leak);