Bläddra i källkod

component/bt: add protection in GKI_getbuf to protect against memory allocation failure;

wangmengyang 9 år sedan
förälder
incheckning
dae40afffb
1 ändrade filer med 11 tillägg och 6 borttagningar
  1. 11 6
      components/bt/bluedroid/gki/gki_buffer.c

+ 11 - 6
components/bt/bluedroid/gki/gki_buffer.c

@@ -196,12 +196,17 @@ void GKI_init_q (BUFFER_Q *p_q)
 void *GKI_getbuf (UINT16 size)
 {
   BUFFER_HDR_T *header = osi_malloc(size + BUFFER_HDR_SIZE);
-  header->status  = BUF_STATUS_UNLINKED;
-  header->p_next  = NULL;
-  header->Type    = 0;
-  header->size = size;
-
-  return header + 1;
+  assert(header != NULL);
+  if (header != NULL) {
+    header->status  = BUF_STATUS_UNLINKED;
+    header->p_next  = NULL;
+    header->Type    = 0;
+    header->size = size;
+
+    return header + 1;
+  } else {
+      return NULL;
+  }
 }