소스 검색

[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 */