浏览代码

[kservice] add NULL check for rt_free and rt_free_align.

朱天龙 (Armink) 2 年之前
父节点
当前提交
33be34d8ec
共有 1 个文件被更改,包括 4 次插入0 次删除
  1. 4 0
      src/kservice.c

+ 4 - 0
src/kservice.c

@@ -1551,6 +1551,8 @@ RT_WEAK void rt_free(void *rmem)
 
     /* call 'rt_free' hook */
     RT_OBJECT_HOOK_CALL(rt_free_hook, (rmem));
+    /* NULL check */
+    if (rmem == RT_NULL) return;
     /* Enter critical zone */
     level = _heap_lock();
     _MEM_FREE(rmem);
@@ -1672,6 +1674,8 @@ RT_WEAK void rt_free_align(void *ptr)
 {
     void *real_ptr;
 
+    /* NULL check */
+    if (ptr == RT_NULL) return;
     real_ptr = (void *) * (rt_ubase_t *)((rt_ubase_t)ptr - sizeof(void *));
     rt_free(real_ptr);
 }