Input: soc_button_array - add missing memory allocation check

Signed-off-by: Pramod Gurav <pramod.gurav@smartplayin.com>
Signed-off-by: Dmitry Torokhov <dmitry.torokhov@gmail.com>

1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/input/misc/soc_button_array.c b/drivers/input/misc/soc_button_array.c
index 5a6334be30b8..e34dfc29beb3 100644
--- a/drivers/input/misc/soc_button_array.c
+++ b/drivers/input/misc/soc_button_array.c
@@ -83,6 +83,9 @@ soc_button_device_create(struct pnp_dev *pdev,
                                        sizeof(*gpio_keys_pdata) +
                                         sizeof(*gpio_keys) * MAX_NBUTTONS,
                                        GFP_KERNEL);
+        if (!gpio_keys_pdata)
+                return ERR_PTR(-ENOMEM);
+
         gpio_keys = (void *)(gpio_keys_pdata + 1);
 
         for (info = button_info; info->name; info++) {