staging: btmtk_usb: check for a valid io_buf pointer

[~andy/linux] / drivers / staging / btmtk_usb / btmtk_usb.c

diff --git a/drivers/staging/btmtk_usb/btmtk_usb.c b/drivers/staging/btmtk_usb/btmtk_usb.c
index 85bafacbb2fd343c367a461393063e78dd81f0c4..ab57a4b97e69bb1c9eedd2e67aa56a934de9f23c 100644 (file)
--- a/drivers/staging/btmtk_usb/btmtk_usb.c
+++ b/drivers/staging/btmtk_usb/btmtk_usb.c
@@ -1597,6 +1597,11 @@ static int btmtk_usb_probe(struct usb_interface *intf,
        }
 
        data->io_buf = kmalloc(256, GFP_ATOMIC);
+       if (!data->io_buf) {
+               hci_free_dev(hdev);
+               kfree(data);
+               return -ENOMEM;
+       }
 
        btmtk_usb_switch_iobase(data, WLAN);