[components][rndis.c] add safety checking after allocating memory

1 files changed, 13 insertions, 0 deletions

diff --git a/components/drivers/usb/usbdevice/class/rndis.c b/components/drivers/usb/usbdevice/class/rndis.c
index 1fe48c7e9d..d3ea89ee0a 100644
--- a/components/drivers/usb/usbdevice/class/rndis.c
+++ b/components/drivers/usb/usbdevice/class/rndis.c
@@ -1348,6 +1348,19 @@ ufunction_t rt_usbd_function_rndis_create(udevice_t device)
     intf_comm = rt_usbd_interface_new(device, _interface_handler);
     intf_data = rt_usbd_interface_new(device, _interface_handler);
 
+    if((intf_comm == RT_NULL) || (intf_data == RT_NULL))
+    {
+        LOG_E("%s,%d: no memory!", __func__, __LINE__);
+
+        if(intf_comm != RT_NULL)
+            rt_free(intf_comm);
+
+        if(intf_data != RT_NULL)
+            rt_free(intf_data);
+
+        return RT_NULL;
+    }
+
     /* create a communication alternate setting and a data alternate setting */
     comm_setting = rt_usbd_altsetting_new(sizeof(struct ucdc_comm_descriptor));
     data_setting = rt_usbd_altsetting_new(sizeof(struct ucdc_data_descriptor));