Cleanup (most) memory leaks

2 files changed, 114 insertions, 64 deletions

diff --git a/archival/dpkg.c b/archival/dpkg.c
index f207b23..48c3928 100644
--- a/archival/dpkg.c
+++ b/archival/dpkg.c
@@ -161,6 +161,7 @@ int search_name_hashtable(const char *key)
 {
 	unsigned int probe_address = 0;
 	unsigned int probe_decrement = 0;
+//	char *temp;
 
 	make_hash(key, &probe_address, &probe_decrement, NAME_HASH_PRIME);
 	while(name_hashtable[probe_address] != NULL) {
@@ -174,7 +175,6 @@ int search_name_hashtable(const char *key)
 		}
 	}
 	name_hashtable[probe_address] = xstrdup(key);
-
 	return(probe_address);
 }
 
@@ -212,6 +212,7 @@ int version_compare_part(const char *version1, const char *version2)
 	int tmp_int;
 	int ver_num1;
 	int ver_num2;
+	int ret;
 
 	if (version1 == NULL) {
 		version1 = xstrdup("");
@@ -234,7 +235,8 @@ int version_compare_part(const char *version1, const char *version2)
 		free(name1_char);
 		free(name2_char);
 		if (tmp_int != 0) {
-			return(tmp_int);
+			ret = tmp_int;
+			goto cleanup_version_compare_part;
 		}
 
 		/* Compare digits */
@@ -249,13 +251,17 @@ int version_compare_part(const char *version1, const char *version2)
 		free(name1_char);
 		free(name2_char);
 		if (ver_num1 < ver_num2) {
-			return(-1);
+			ret = -1;
+			goto cleanup_version_compare_part;
 		}
 		else if (ver_num1 > ver_num2) {
-			return(1);
+			ret = 1;
+			goto cleanup_version_compare_part;
 		}
 	}
-	return(0);
+	ret = 0;
+cleanup_version_compare_part:
+	return(ret);
 }
 
 /* if ver1 < ver2 return -1,
@@ -499,54 +505,61 @@ unsigned int fill_package_struct(char *control_buffer)
 {
 	common_node_t *new_node = (common_node_t *) xcalloc(1, sizeof(common_node_t));
 
-	char *field_name = xmalloc(sizeof(char *));
-	char *field_value = xmalloc(sizeof(char *));
+	char **field_name = xmalloc(sizeof(char *));
+	char **field_value = xmalloc(sizeof(char *));
 	int field_start = 0;
 	int num = -1;
 	int buffer_length = strlen(control_buffer);
 
 	new_node->version = search_name_hashtable("unknown");
 	while (field_start < buffer_length) {
-		field_start += read_package_field(&control_buffer[field_start], &field_name, &field_value);
+		field_start += read_package_field(&control_buffer[field_start], field_name, field_value);
 
-		if (field_name == NULL) {
+		if (*field_name == NULL) {
 			goto fill_package_struct_cleanup; // Oh no, the dreaded goto statement !!
 		}
 
-		if (strcmp(field_name, "Package") == 0) {
-			new_node->name = search_name_hashtable(field_value);
+		if (strcmp(*field_name, "Package") == 0) {
+			new_node->name = search_name_hashtable(*field_value);
 		}
-		else if (strcmp(field_name, "Version") == 0) {
-			new_node->version = search_name_hashtable(field_value);
+		else if (strcmp(*field_name, "Version") == 0) {
+			new_node->version = search_name_hashtable(*field_value);
 		}
-		else if (strcmp(field_name, "Pre-Depends") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_PRE_DEPENDS);
+		else if (strcmp(*field_name, "Pre-Depends") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_PRE_DEPENDS);
 		}
-		else if (strcmp(field_name, "Depends") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_DEPENDS);
+		else if (strcmp(*field_name, "Depends") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_DEPENDS);
 		}
-		else if (strcmp(field_name, "Replaces") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_REPLACES);
+		else if (strcmp(*field_name, "Replaces") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_REPLACES);
 		}
-		else if (strcmp(field_name, "Provides") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_PROVIDES);
+		else if (strcmp(*field_name, "Provides") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_PROVIDES);
 		}
-		else if (strcmp(field_name, "Conflicts") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_CONFLICTS);
+		else if (strcmp(*field_name, "Conflicts") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_CONFLICTS);
 		}
-		else if (strcmp(field_name, "Suggests") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_SUGGESTS);
+		else if (strcmp(*field_name, "Suggests") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_SUGGESTS);
 		}
-		else if (strcmp(field_name, "Recommends") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_RECOMMENDS);
+		else if (strcmp(*field_name, "Recommends") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_RECOMMENDS);
 		}
-		else if (strcmp(field_name, "Enhances") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_ENHANCES);
+		else if (strcmp(*field_name, "Enhances") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_ENHANCES);
 		}
 fill_package_struct_cleanup:
-		free(field_name);
-		free(field_value);
+		if (*field_name) {
+			free(*field_name);
+		}
+		if (*field_value) {
+			free(*field_value);
+		}
 	}
+	free(field_name);
+	free(field_value);
+
 	if (new_node->version == search_name_hashtable("unknown")) {
 		free_package(new_node);
 		return(-1);
@@ -615,6 +628,7 @@ void set_status(const unsigned int status_node_num, const char *new_value, const
 	new_status = (char *) xmalloc(want_len + flag_len + status_len + 3);
 	sprintf(new_status, "%s %s %s", name_hashtable[want], name_hashtable[flag], name_hashtable[status]);
 	status_hashtable[status_node_num]->status = search_name_hashtable(new_status);
+	free(new_status);
 	return;
 }
 
@@ -819,7 +833,6 @@ void write_status_file(deb_file_t **deb_file)
 				}
 				else if	(strcmp("config-files", name_hashtable[state_status]) == 0) {
 					/* only change the status line */
-//					buffer_length = strlen(control_buffer);
 					while (1) {
 						char *field_name;
 						char *field_value;
@@ -1003,6 +1016,7 @@ int check_deps(deb_file_t **deb_file, int deb_start, int dep_max_count)
 		}
 		i++;
 	}
+	free(conflicts);
 	return(TRUE);
 }
 
@@ -1091,8 +1105,10 @@ int run_package_script(const char *package_name, const char *script_type)
 
 	/* If the file doesnt exist is isnt a fatal */
 	if (lstat(script_path, &path_stat) < 0) {
+		free(script_path);
 		return(EXIT_SUCCESS);
 	} else {
+		free(script_path);
 		return(system(script_path));
 	}
 }
@@ -1236,6 +1252,8 @@ void unpack_package(deb_file_t *deb_file)
 	/* change status */
 	set_status(status_num, "install", 1);
 	set_status(status_num, "unpacked", 3);
+
+	free(info_prefix);
 }
 
 void configure_package(deb_file_t *deb_file)
@@ -1399,6 +1417,13 @@ extern int dpkg_main(int argc, char **argv)
 
 	write_status_file(deb_file);
 
+	for (i = 0; i < deb_count; i++) {
+		free(deb_file[i]->control_file);
+		free(deb_file[i]->filename);
+		free(deb_file[i]);
+	}
+	free(deb_file);
+
 	for (i = 0; i < NAME_HASH_PRIME; i++) {
 		if (name_hashtable[i] != NULL) {
 			free(name_hashtable[i]);
diff --git a/dpkg.c b/dpkg.c
index f207b23..48c3928 100644
--- a/dpkg.c
+++ b/dpkg.c
@@ -161,6 +161,7 @@ int search_name_hashtable(const char *key)
 {
 	unsigned int probe_address = 0;
 	unsigned int probe_decrement = 0;
+//	char *temp;
 
 	make_hash(key, &probe_address, &probe_decrement, NAME_HASH_PRIME);
 	while(name_hashtable[probe_address] != NULL) {
@@ -174,7 +175,6 @@ int search_name_hashtable(const char *key)
 		}
 	}
 	name_hashtable[probe_address] = xstrdup(key);
-
 	return(probe_address);
 }
 
@@ -212,6 +212,7 @@ int version_compare_part(const char *version1, const char *version2)
 	int tmp_int;
 	int ver_num1;
 	int ver_num2;
+	int ret;
 
 	if (version1 == NULL) {
 		version1 = xstrdup("");
@@ -234,7 +235,8 @@ int version_compare_part(const char *version1, const char *version2)
 		free(name1_char);
 		free(name2_char);
 		if (tmp_int != 0) {
-			return(tmp_int);
+			ret = tmp_int;
+			goto cleanup_version_compare_part;
 		}
 
 		/* Compare digits */
@@ -249,13 +251,17 @@ int version_compare_part(const char *version1, const char *version2)
 		free(name1_char);
 		free(name2_char);
 		if (ver_num1 < ver_num2) {
-			return(-1);
+			ret = -1;
+			goto cleanup_version_compare_part;
 		}
 		else if (ver_num1 > ver_num2) {
-			return(1);
+			ret = 1;
+			goto cleanup_version_compare_part;
 		}
 	}
-	return(0);
+	ret = 0;
+cleanup_version_compare_part:
+	return(ret);
 }
 
 /* if ver1 < ver2 return -1,
@@ -499,54 +505,61 @@ unsigned int fill_package_struct(char *control_buffer)
 {
 	common_node_t *new_node = (common_node_t *) xcalloc(1, sizeof(common_node_t));
 
-	char *field_name = xmalloc(sizeof(char *));
-	char *field_value = xmalloc(sizeof(char *));
+	char **field_name = xmalloc(sizeof(char *));
+	char **field_value = xmalloc(sizeof(char *));
 	int field_start = 0;
 	int num = -1;
 	int buffer_length = strlen(control_buffer);
 
 	new_node->version = search_name_hashtable("unknown");
 	while (field_start < buffer_length) {
-		field_start += read_package_field(&control_buffer[field_start], &field_name, &field_value);
+		field_start += read_package_field(&control_buffer[field_start], field_name, field_value);
 
-		if (field_name == NULL) {
+		if (*field_name == NULL) {
 			goto fill_package_struct_cleanup; // Oh no, the dreaded goto statement !!
 		}
 
-		if (strcmp(field_name, "Package") == 0) {
-			new_node->name = search_name_hashtable(field_value);
+		if (strcmp(*field_name, "Package") == 0) {
+			new_node->name = search_name_hashtable(*field_value);
 		}
-		else if (strcmp(field_name, "Version") == 0) {
-			new_node->version = search_name_hashtable(field_value);
+		else if (strcmp(*field_name, "Version") == 0) {
+			new_node->version = search_name_hashtable(*field_value);
 		}
-		else if (strcmp(field_name, "Pre-Depends") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_PRE_DEPENDS);
+		else if (strcmp(*field_name, "Pre-Depends") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_PRE_DEPENDS);
 		}
-		else if (strcmp(field_name, "Depends") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_DEPENDS);
+		else if (strcmp(*field_name, "Depends") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_DEPENDS);
 		}
-		else if (strcmp(field_name, "Replaces") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_REPLACES);
+		else if (strcmp(*field_name, "Replaces") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_REPLACES);
 		}
-		else if (strcmp(field_name, "Provides") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_PROVIDES);
+		else if (strcmp(*field_name, "Provides") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_PROVIDES);
 		}
-		else if (strcmp(field_name, "Conflicts") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_CONFLICTS);
+		else if (strcmp(*field_name, "Conflicts") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_CONFLICTS);
 		}
-		else if (strcmp(field_name, "Suggests") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_SUGGESTS);
+		else if (strcmp(*field_name, "Suggests") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_SUGGESTS);
 		}
-		else if (strcmp(field_name, "Recommends") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_RECOMMENDS);
+		else if (strcmp(*field_name, "Recommends") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_RECOMMENDS);
 		}
-		else if (strcmp(field_name, "Enhances") == 0) {
-			add_split_dependencies(new_node, field_value, EDGE_ENHANCES);
+		else if (strcmp(*field_name, "Enhances") == 0) {
+			add_split_dependencies(new_node, *field_value, EDGE_ENHANCES);
 		}
 fill_package_struct_cleanup:
-		free(field_name);
-		free(field_value);
+		if (*field_name) {
+			free(*field_name);
+		}
+		if (*field_value) {
+			free(*field_value);
+		}
 	}
+	free(field_name);
+	free(field_value);
+
 	if (new_node->version == search_name_hashtable("unknown")) {
 		free_package(new_node);
 		return(-1);
@@ -615,6 +628,7 @@ void set_status(const unsigned int status_node_num, const char *new_value, const
 	new_status = (char *) xmalloc(want_len + flag_len + status_len + 3);
 	sprintf(new_status, "%s %s %s", name_hashtable[want], name_hashtable[flag], name_hashtable[status]);
 	status_hashtable[status_node_num]->status = search_name_hashtable(new_status);
+	free(new_status);
 	return;
 }
 
@@ -819,7 +833,6 @@ void write_status_file(deb_file_t **deb_file)
 				}
 				else if	(strcmp("config-files", name_hashtable[state_status]) == 0) {
 					/* only change the status line */
-//					buffer_length = strlen(control_buffer);
 					while (1) {
 						char *field_name;
 						char *field_value;
@@ -1003,6 +1016,7 @@ int check_deps(deb_file_t **deb_file, int deb_start, int dep_max_count)
 		}
 		i++;
 	}
+	free(conflicts);
 	return(TRUE);
 }
 
@@ -1091,8 +1105,10 @@ int run_package_script(const char *package_name, const char *script_type)
 
 	/* If the file doesnt exist is isnt a fatal */
 	if (lstat(script_path, &path_stat) < 0) {
+		free(script_path);
 		return(EXIT_SUCCESS);
 	} else {
+		free(script_path);
 		return(system(script_path));
 	}
 }
@@ -1236,6 +1252,8 @@ void unpack_package(deb_file_t *deb_file)
 	/* change status */
 	set_status(status_num, "install", 1);
 	set_status(status_num, "unpacked", 3);
+
+	free(info_prefix);
 }
 
 void configure_package(deb_file_t *deb_file)
@@ -1399,6 +1417,13 @@ extern int dpkg_main(int argc, char **argv)
 
 	write_status_file(deb_file);
 
+	for (i = 0; i < deb_count; i++) {
+		free(deb_file[i]->control_file);
+		free(deb_file[i]->filename);
+		free(deb_file[i]);
+	}
+	free(deb_file);
+
 	for (i = 0; i < NAME_HASH_PRIME; i++) {
 		if (name_hashtable[i] != NULL) {
 			free(name_hashtable[i]);