staging: otus: check kmalloc() return value

kmalloc() may fail, if so return error from zfwUsbSubmitControl().

Signed-off-by: Kulikov Vasiliy <segooon@gmail.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@suse.de>

1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/staging/otus/wrap_usb.c b/drivers/staging/otus/wrap_usb.c
index 93459cadc47..9f04047bf5a 100644
--- a/drivers/staging/otus/wrap_usb.c
+++ b/drivers/staging/otus/wrap_usb.c
@@ -104,6 +104,11 @@ u32_t zfwUsbSubmitControl(zdev_t *dev, u8_t req, u16_t value, u16_t index,
 
         if (size > 0) {
                 buf = kmalloc(size, GFP_KERNEL);
+                if (buf == NULL) {
+                        pr_err("zfwUsbSubmitControl() failed, "
+                                  "kmalloc() returned NULL\n");
+                        return 1;
+                }
                 memcpy(buf, (u8_t *)data, size);
         } else
                 buf = NULL;