[CORE] Added memory check (#1638)

open5gs · Jul 1, 2022 · 318fc25 · 318fc25
1 parent 6adf541
commit 318fc25
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/core/ogs-pkbuf.c b/lib/core/ogs-pkbuf.c
@@ -297,8 +297,10 @@ ogs_pkbuf_t *ogs_pkbuf_copy_debug(ogs_pkbuf_t *pkbuf, const char *file_line)
 {
 #if OGS_USE_TALLOC
     ogs_pkbuf_t *newbuf;
-    int size = pkbuf->end - pkbuf->head;
+    int size = 0;
 
+    ogs_assert(pkbuf);
+    size = pkbuf->end - pkbuf->head;
     ogs_assert(size > 0);
     newbuf = ogs_pkbuf_alloc_debug(NULL, size, file_line);
     if (!newbuf) {