pdb: convert to memory logging helper, properly free memory in error …

…cases
kamailio · Mar 29, 2023 · ab17b68 · ab17b68
1 parent 3d60451
commit ab17b68
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/modules/pdb/pdb.c b/src/modules/pdb/pdb.c
@@ -541,6 +541,7 @@ static int add_server(char *host, char *port)
 	server->host = pkg_malloc(strlen(host)+1);
 	if (server->host == NULL) {
 		PKG_MEM_ERROR;
+		pkg_free(server);
 		return -1;
 	}
 	strcpy(server->host, host);
@@ -798,9 +799,14 @@ static int mod_init(void)
 		return -1;
 	}
 
-    global_id = (uint16_t*)shm_malloc(sizeof(uint16_t));
-
-    return 0;
+	global_id = (uint16_t*)shm_malloc(sizeof(uint16_t));
+	if(!global_id)
+	{
+		SHM_MEM_ERROR;
+		shm_free(active);
+		return -1;
+	}
+	return 0;
 }
 
 static int child_init (int rank)