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>

diff --git a/drivers/staging/otus/wrap_usb.c b/drivers/staging/otus/wrap_usb.c
index 93459ca..9f04047 100644 (file)
--- 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;