浏览代码

[winusb.c] add checking after allocating memory for variable winusb_device to make sure allocae is successful

ourea1429 2 年之前
父节点
当前提交
085ded8eef
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      components/drivers/usb/usbdevice/class/winusb.c

+ 2 - 0
components/drivers/usb/usbdevice/class/winusb.c

@@ -324,6 +324,8 @@ ufunction_t rt_usbd_function_winusb_create(udevice_t device)
 
     /* allocate memory for cdc vcom data */
     winusb_device = (winusb_device_t)rt_malloc(sizeof(struct winusb_device));
+    if (winusb_device == NULL)
+        return RT_NULL;
     rt_memset((void *)winusb_device, 0, sizeof(struct winusb_device));
     func->user_data = (void*)winusb_device;
     /* create an interface object */