Fix malloc error handling

OpenSIPS · Mar 9, 2020 · c58df36 · c58df36
1 parent 31fc3b6
commit c58df36
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cfg_pp.c b/cfg_pp.c
@@ -546,6 +546,11 @@ void cfg_dump_context(const char *file, int line, int colstart, int colend)
 	/* error indicator line */
 	len = strlen(con->lines[i-1]);
 	wsbuf = malloc(len + 1);
+	if (wsbuf == NULL) {
+		LM_ERR("NO MEMORY\n");
+		return;
+	}
+
 	wb = wsbuf;
 	for (p = con->lines[i-1], end = p + len; p < end && is_ws(*p); p++)
 		*wb++ = *p;
@@ -555,6 +560,10 @@ void cfg_dump_context(const char *file, int line, int colstart, int colend)
 		hiline = NULL;
 	} else {
 		hiline = malloc(colend - colstart);
+		if (hiline == NULL) {
+			LM_ERR("NO MEMORY\n");
+			return;
+		}
 		memset(hiline, '~', colend - colstart);
 	}