HuntnGather – Diff between revs 33 and 37

Subversion Repositories:

?pathlinks?

Rev 33 – Rev 38 – Go to most recent revision – Details – Blame – Last modification – View Log

Only display areas with differences – Ignore whitespace

 ///////////////////////////////////////////////////////////////////////////
 //    Copyright (C) 2021 Wizardry and Steamworks - License: MIT          //
 ///////////////////////////////////////////////////////////////////////////
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #if !defined ___AmigaOS___
 #include <dirent.h>
 #include <sys/stat.h>
 #include <signal.h>
 #endif
 #include <sys/types.h>
 #include <sys/syslimits.h>
 #if defined ___AmigaOS___
 #include <proto/dos.h>
 #include <proto/exec.h>
 #include <proto/locale.h>
 #include <clib/utility_protos.h>
 #endif
 #if defined ___AsyncIO___
 #include <asyncio.h>
 #endif
 #if !defined ___HAVE_GETOPT___
 #include "/shared/getopt.h"
 #endif
 #include "StringStack.h"
 #include "/shared/utilities.h"
 #define PROGRAM_VERSION "1.7.4"
 #if defined ___AmigaOS___
 /*************************************************************************/
 /*        Version string used for querrying the program version.         */
 /*************************************************************************/
 TEXT version_string[] =
 	"\0$VER: Gather " PROGRAM_VERSION " "__DATE__" by Wizardry and Steamworks";
 #endif
 int PROGRAM_RUN = TRUE;
 int PROGRAM_VERBOSE = TRUE;
 int maxmem = MAX_MEM;
 // Define global locale for string compare.
 #if defined ___AmigaOS___
 struct Locale *locale;
 #endif
 void SignalHandler(int sig) {
 	/* Toggle the run flag to stop execution. */
 	PROGRAM_RUN = FALSE;
+}
 /*
+	*
 	* Used for sorting database lines.
 	*/
 int QsortCompare(const void *a, const void *b) {
 #if defined ___AmigaOS___
 	return StrnCmp(
 		locale,
 		(STRPTR)(*(const char **)a),
 		(STRPTR)*((const char **)b),
 		-1,
 		SC_ASCII
 	);
 #else
 	return strcmp(*(const char **)a, *(const char **)b);
 #endif
+}
 /*
+	*
 	* Sorts a database file lexicographically.
 	*/
 void SortDatabase(char *dbFile, int lines) {
 #if defined ___AsyncIO___
 	struct AsyncFile *fp;
 #else
 	FILE *fp;
 #endif
 	char **database;
 	dbEntry *entry;
 	char *line = NULL;
 	char *rem;
 	int count;
 	int i;
 	// Open database file for reading.
 #if defined ___AsyncIO___
 	if((fp = OpenAsync(dbFile, MODE_READ, ASYNC_BUF)) == NULL) {
 #else
 	if((fp = fopen(dbFile, "r")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for reading.\n", dbFile);
 		return;
+	}
 	if((database = malloc(lines * sizeof(*database))) == NULL) {
 		fprintf(stderr, "Memory allocation failure.\n");
 #if defined ___AsyncIO___
 		CloseAsync(fp);
 #else
 		fclose(fp);
 #endif
 		return;
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Reading lines from file '%s' to array...\n", dbFile);
+	}
 	count = 0;
 	while(PROGRAM_RUN && (line = ReadLine(fp)) != NULL) {
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
    	if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		if((entry = CreateDatabaseEntry(line)) == NULL) {
 			fprintf(stderr, "Unable to create database entry.\n");
 			free(line);
 #if defined ___AsyncIO___
 			CloseAsync(fp);
 #else
 			fclose(fp);
 #endif
 			return;
+		}
 		if((database[count] = malloc((strlen(entry->name) + strlen(entry->path) + 1 + 1) * sizeof(*database[count]))) == NULL) {
 			fprintf(stderr, "Memory allocation failure.\n");
 			free(entry->name);
 			free(entry->path);
 			free(entry);
 			free(line);
 #if defined ___AsyncIO___
 			CloseAsync(fp);
 #else
 			fclose(fp);
 #endif
 			return;
+		}
 		sprintf(database[count], "%s\t%s", entry->name, entry->path);
 		++count;
 		// Free the database entry.
 		free(entry->name);
 		free(entry->path);
 		free(entry);
 		free(line);
+	}
 	if(line != NULL) {
 		free(line);
+	}
 #if defined ___AsyncIO___
 	CloseAsync(fp);
 #else
 	fclose(fp);
 #endif
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Sorting %d lines in '%s'...\n", count, dbFile);
+	}
 	// Sort the database.
 	qsort(database, (unsigned int)count, sizeof(char *), QsortCompare);
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Writing %d sorted lines to '%s'...\n", count, dbFile);
+	}
 	// Write the database lines back to the database.
 #if defined ___AsyncIO___
 	if((fp = OpenAsync(dbFile, MODE_WRITE, ASYNC_BUF)) == NULL) {
 #else
 	if((fp = fopen(dbFile, "w")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for writing.\n", dbFile);
 		return;
+	}
 	rem = NULL;
 	for(i = 0; PROGRAM_RUN && i < count; ++i) {
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
    		if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		if(rem != NULL) {
 #if defined ___AmigaOS___
 			if(StrnCmp(locale, database[i], rem, -1, SC_ASCII) == 0) {
 #else
 			if(strcmp(database[i], rem) == 0) {
 #endif
 				continue;
+			}
+		}
 #if defined ___AsyncIO___
 		WriteAsync(fp, database[i], (LONG)strlen(database[i]));
 		WriteAsync(fp, "\n", 1);
 #else
 		fprintf(fp, "%s\n", database[i]);
 #endif
 		if(rem != NULL) {
 			free(rem);
+		}
 		rem = malloc((strlen(database[i]) + 1) * sizeof(*rem));
 		sprintf(rem, "%s", database[i]);
+	}
 	if(rem != NULL) {
 		free(rem);
+	}
 #if defined ___AsyncIO___
 	CloseAsync(fp);
 #else
 	fclose(fp);
 #endif
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Disposing %d lines of file '%s'...\n", count, dbFile);
+	}
 	// Free up database.
 	for(i = 0; i < count; ++i) {
 		free(database[i]);
+	}
     free(database);
+}
 /*
+	*
 	* Updates a database file "dbFile".
 	*/
 dbStats *CollectFiles(char *dbFile, char **paths, int count) {
 #if defined ___AsyncIO___
 	struct AsyncFile *fp;
 #else
 	FILE *fp;
 #endif
 #if defined ___AmigaOS___
 	struct FileInfoBlock *FIB;
 	BPTR lock;
 #else
 	DIR *dir;
 	struct dirent *entry;
 	struct stat dirStat;
 #endif
 	stringStack *stack;
 	dbStats *stats = NULL;
 	int i;
 	char *path;
 	char *sub;
 #if defined ___AsyncIO___
 	if((fp = OpenAsync(dbFile, MODE_APPEND, ASYNC_BUF)) == NULL) {
 #else
 	if((fp = fopen(dbFile, "a")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for writing.\n", dbFile);
 		return NULL;
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Collecting files...\r");
+	}
 	// Initialize metrics.
  	if((stats = malloc(sizeof(*stats))) == NULL) {
  		fprintf(stderr, "Memory allocation failure.\n");
 #if defined ___AsyncIO___
 		CloseAsync(fp);
 #else
 		fclose(fp);
 #endif
  		return NULL;
+ 	}
  	stats->dirs = 0;
  	stats->files = 0;
  	stats->lines = 0;
  	stats->size = 0;
 	// Push the first path onto the stack.
 	stack = stringStackCreate((unsigned int)count);
 	for(i = 0; PROGRAM_RUN && i < count; ++i) {
 		stringStackPush(stack, paths[i]);
+	}
 	while(PROGRAM_RUN && !stringStackIsEmpty(stack)) {
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
 		if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		if((path = stringStackPop(stack)) == NULL) {
 			break;
+		}
 #if defined ___AmigaOS___
 		if((lock = Lock(path, ACCESS_READ)) == NULL) {
 			fprintf(stderr, "Could not lock path '%s' for reading.\n", path);
 			free(path);
 			continue;
+		}
 		if((FIB = AllocDosObject(DOS_FIB, NULL)) == NULL) {
 			fprintf(stderr, "File information block for path '%s' could not be allocated.\n", path);
 			UnLock(lock);
 			free(path);
 			continue;
+		}
 		if(Examine(lock, FIB) == FALSE) {
 			fprintf(stderr, "Path '%s' could not be examined.\n", path);
 			FreeDosObject(DOS_FIB, FIB);
 			UnLock(lock);
 			free(path);
 			continue;
+		}
 		while(PROGRAM_RUN && ExNext(lock, FIB)) {
 			// Check if CTRL+C was pressed and abort the program.
 			if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 				PROGRAM_RUN = FALSE;
 				continue;
+			}
 #else
 		if((dir = opendir(path)) == NULL) {
 			fprintf(stderr, "Directory '%s' could not be opened.\n", path);
 			free(path);
 			continue;
+		}
 		while(PROGRAM_RUN && (entry = readdir(dir)) != NULL) {
 #endif
 			switch(path[strlen(path) - 1]) {
 				case '/':
 	 			case ':': // This is a drive path.
 #if defined ___AmigaOS___
 					if((sub = malloc(strlen(path) + strlen(FIB->fib_FileName) + 1)) == NULL) {
 #else
 	 				if((sub = malloc(strlen(path) + strlen(entry->d_name) + 1)) == NULL) {
 #endif
 	 					fprintf(stderr, "Memory allocation failure.\n");
 #if defined ___AmigaOS___
 						FreeDosObject(DOS_FIB, FIB);
 						UnLock(lock);
 #else
 	 					closedir(dir);
 #endif
 	 					free(path);
 	 					stringStackDestroy(stack);
 #if defined ___AsyncIO___
 						CloseAsync(fp);
 #else
 						fclose(fp);
 #endif
 	 					return NULL;
+	 				}
 #if defined ___AmigaOS___
 					sprintf(sub, "%s%s", path, FIB->fib_FileName);
 #else
 	 				sprintf(sub, "%s%s", path, entry->d_name);
 #endif
 	 				break;
 	 			default:
 #if defined ___AmigaOS___
 					if((sub = malloc(strlen(path) + strlen(FIB->fib_FileName) + 1 + 1)) == NULL) {
 #else
 	 				if((sub = malloc(strlen(path) + strlen(entry->d_name) + 1 + 1)) == NULL) {
 #endif
 	 					fprintf(stderr, "Memory allocation failure.\n");
 #if defined ___AmigaOS___
 						FreeDosObject(DOS_FIB, FIB);
 						UnLock(lock);
 #else
 	 					closedir(dir);
 #endif
 	 					free(path);
 	 					stringStackDestroy(stack);
 #if defined ___AsyncIO___
 						CloseAsync(fp);
 #else
 						fclose(fp);
 #endif
 	 					return NULL;
+	 				}
 #if defined ___AmigaOS___
 					sprintf(sub, "%s/%s", path, FIB->fib_FileName);
 #else
 	 				sprintf(sub, "%s/%s", path, entry->d_name);
 #endif
 	 				break;
+			}
 			switch(GetFsType(sub)) {
 				case UNKNOWN:
 					free(sub);
 					continue;
 				case REGULAR:
 					++stats->files;
 	 				if(PROGRAM_VERBOSE) {
 	 					fprintf(stdout,
 	 					"Gathered %d directories and %d files.\r",
 	 					stats->dirs,
 	 					stats->files);
+	 				}
 					break;
 				case DIRECTORY:
 	 				stringStackPush(stack, sub);
 	 				++stats->dirs;
 	 				if(PROGRAM_VERBOSE) {
 	 					fprintf(stdout,
 	 						"Gathered %d directories and %d files.\r",
 	 						stats->dirs,
 	 						stats->files);
+	 				}
 	 				free(sub);
 	 				continue;
+			}
 #if defined ___NOCASE_FS___
 #if defined ___AmigaOS___
 			StrUpr(FIB->fib_FileName);
 #else
 			StrUpr(entry->d_name);
 #endif
 #endif
 	 		// Write to database file.
 #if defined ___AsyncIO___
 #if defined ___AmigaOS___
 			WriteAsync(fp, FIB->fib_FileName, (LONG)strlen(FIB->fib_FileName));
 			stats->size = stats->size + strlen(FIB->fib_FileName);
 #else
 			WriteAsync(fp, entry->d_name, (LONG)strlen(entry->d_name));
 			stats->size = stats->size + strlen(entry->d_name);
 #endif
 			WriteAsync(fp, "\t", 1);
 			++stats->size;
 			WriteAsync(fp, sub, (LONG)strlen(sub));
 			stats->size = stats->size + strlen(sub);
 			WriteAsync(fp, "\n", 1);
 			++stats->size;
 #else
 #if defined ___AmigaOS___
 			fprintf(fp, "%s\t%s\n", FIB->fib_FileName, sub);
 			stats->size = stats->size + strlen(FIB->fib_FileName) + strlen(sub) + 1 + 1 + 1;
 #else
 	 		fprintf(fp, "%s\t%s\n", entry->d_name, sub);
 	 		stats->size = stats->size + strlen(entry->d_name) + strlen(sub) + 1 + 1 + 1;
 #endif
 #endif
 			++stats->lines;
 	 		free(sub);
+	 	}
 #if defined ___AmigaOS___
 		FreeDosObject(DOS_FIB, FIB);
 		UnLock(lock);
 #else
 	 	closedir(dir);
 #endif
 		free(path);
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "\n");
+	}
 	stringStackDestroy(stack);
 #if defined ___AsyncIO___
 	CloseAsync(fp);
 #else
 	fclose(fp);
 #endif
 	return stats;
+}
 /*
+	*
 	* Writes lines from the database "dbFile" to temporary filenames "tmpNames".
 	*/
 void WriteTemporaryFiles(char *dbFile, char **tmpNames, int tmpFiles, int tmpLines, int total) {
 #if defined ___AsyncIO___
 	struct AsyncFile *fp, *tp;
 	LONG c;
 #else
 	FILE *fp, *tp;
 	char c;
 #endif
 	int lines;
 	int write;
 #if defined ___AsyncIO___
 	if((fp = OpenAsync(dbFile, MODE_READ, ASYNC_BUF)) == NULL) {
 #else
 	if((fp = fopen(dbFile, "r")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for reading.\n", dbFile);
 		return;
+	}
 #if defined ___AsyncIO___
 	if((tp = OpenAsync(tmpNames[--tmpFiles], MODE_WRITE, ASYNC_BUF)) == NULL) {
 #else
 	if((tp = fopen(tmpNames[--tmpFiles], "w")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for writing.\n", tmpNames[tmpFiles]);
 #if defined ___AsyncIO___
 		CloseAsync(fp);
 #else
 		fclose(fp);
 #endif
 		return;
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Writing to temporary files...\r");
+	}
 	write = 0;
 	lines = 0;
 #if defined ___AsyncIO___
 	while(PROGRAM_RUN && (c = ReadCharAsync(fp)) != -1) {
 #else
 	while(PROGRAM_RUN && fscanf(fp, "%c", &c) == 1) {
 #endif
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
 		if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		switch(c) {
 			case '\n':
 				// Increment the total written lines.
 				++write;
 				if(PROGRAM_VERBOSE) {
 					fprintf(stdout, "Writing to temporary files: %d%%.\r", (int)(((float)write / total) * 100.0));
+				}
 				// Write the newline character back.
 #if defined ___AsyncIO___
 				if(WriteCharAsync(tp, (UBYTE)c) != 1) {
 #else
 				if(fprintf(tp, "%c", c) != 1) {
 #endif
 					fprintf(stderr, "Unable to write to '%s'.\n", tmpNames[tmpFiles]);
 #if defined ___AsyncIO___
 					CloseAsync(tp);
 					CloseAsync(fp);
 #else
 					fclose(tp);
 					fclose(fp);
 #endif
 					return;
+				}
 				// Switch to the next temporary file.
 				if(++lines >= tmpLines) {
 					// If there are no temporary files left then run till the end.
 					if(tmpFiles - 1 < 0) {
 						break;
+					}
 					// Close the previous temporary file and write to the next temporary file.
 #if defined ___AsyncIO___
 					CloseAsync(tp);
 					if((tp = OpenAsync(tmpNames[--tmpFiles], MODE_WRITE, ASYNC_BUF)) == NULL) {
 #else
 					fclose(tp);
 					if((tp = fopen(tmpNames[--tmpFiles], "w")) == NULL) {
 #endif
 						fprintf(stderr, "Could not open '%s' for writing.\n", tmpNames[tmpFiles]);
 #if defined ___AsyncIO___
 						CloseAsync(fp);
 #else
 						fclose(fp);
 #endif
 						return;
+					}
 					lines = 0;
 					break;
+				}
 				break;
 			default:
 #if defined ___AsyncIO___
 				if(WriteCharAsync(tp, (UBYTE)c) != 1) {
 #else
 				if(fprintf(tp, "%c", c) != 1) {
 #endif
 					fprintf(stderr, "Could not write to file '%s'.\n", tmpNames[tmpFiles]);
 #if defined ___AsyncIO___
 					CloseAsync(tp);
 					CloseAsync(fp);
 #else
 					fclose(tp);
 					fclose(fp);
 #endif
 					return;
+				}
 				break;
+		}
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "\n");
+	}
 #if defined ___AsyncIO___
 	CloseAsync(tp);
 	CloseAsync(fp);
 #else
 	fclose(tp);
 	fclose(fp);
 #endif
+}
 /*
+	*
 	* Merges temporary files "tmpNames" into a database "dbFile".
 	*/
 void MergeTemporaryFiles(char *dbFile, char **tmpNames, int files, int lines) {
 #if defined ___AsyncIO___
 	struct AsyncFile *fp;
 	struct AsyncFile **tp;
 #else
 	FILE *fp;
 	FILE **tp;
 #endif
 	int i;
 	int j;
 	char *tmp;
 	char *rem;
 	char *min;
 	int count;
 #if defined ___AsyncIO___
 	if((fp = OpenAsync(dbFile, MODE_WRITE, ASYNC_BUF)) == NULL) {
 #else
 	if((fp = fopen(dbFile, "w")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for writing.\n", dbFile);
 		return;
+	}
 	// Allocate as many file pointers as temporary files.
 	if((tp = malloc(files * sizeof(*tp))) == NULL) {
 		fprintf(stderr, "Memory allocation failure.\n");
 #if defined ___AsyncIO___
 		CloseAsync(fp);
 #else
 		fclose(fp);
 #endif
 		return;
+	}
 	// Open all temporary files for reading.
 	for(i = 0; i < files; ++i) {
 #if defined ___AsyncIO___
 		if((tp[i] = OpenAsync(tmpNames[i], MODE_READ, ASYNC_BUF)) == NULL) {
 #else
 		if((tp[i] = fopen(tmpNames[i], "r")) == NULL) {
 #endif
 			fprintf(stderr, "Could not open file '%s' for reading.\n", tmpNames[i]);
 			// Close all temporary files.
 			while(--i > -1) {
 #if defined ___AsyncIO___
 				CloseAsync(tp[i]);
 #else
 				fclose(tp[i]);
 #endif
+			}
 #if defined ___AsyncIO___
 			CloseAsync(fp);
 #else
 			fclose(fp);
 #endif
 			return;
+		}
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Merging all files...\r");
+	}
 	rem = NULL;
 	count = lines;
 	j = 0;
 	while(PROGRAM_RUN && --count > -1) {
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
 		if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		// Find the smallest line in all temporary files.
 		if(PROGRAM_VERBOSE) {
 			fprintf(stdout, "Merging all files: %d%%.\r", 100 - (int)(((float)count / lines) * 100.0));
+		}
 		min = NULL;
 		for(i = 0; i < files; ++i) {
 			tmp = PeekLine(tp[i]);
 			if(tmp == NULL) {
 				continue;
+			}
 #if defined ___AmigaOS___
 			if(min == NULL || StrnCmp(locale, tmp, min, -1, SC_ASCII) < 0) {
 #else
 			if(min == NULL || strcmp(tmp, min) < 0) {
 #endif
 				if(min != NULL) {
 					// Free previous instance.
 					free(min);
+				}
 				if((min = malloc((strlen(tmp) + 1) * sizeof(*min))) == NULL) {
 					fprintf(stderr, "Memory allocation failure.\n");
 					free(tmp);
 					if(min != NULL) {
 						free(min);
+					}
 					if(rem != NULL) {
 						free(rem);
+					}
 #if defined ___AsyncIO___
 					CloseAsync(fp);
 #else
 					fclose(fp);
 #endif
 					return;
+				}
 				sprintf(min, "%s", tmp);
 				// Remember the index of the file where the smallest entry has been found.
 				j = i;
+			}
 			free(tmp);
+		}
 		// Forward the file where the smallest line was found.
 		SkipLine(tp[j]);
 		// Write the smallest line.
 		if(min != NULL) {
 			// If current minimum line is identical to previous minimum line then skip to remove duplicates.
 			if(rem != NULL) {
 #if defined ___AmigaOS___
 				if(StrnCmp(locale, min, rem, -1, SC_ASCII) == 0) {
 #else
 				if(strcmp(min, rem) == 0) {
 #endif
 					free(min);
 					continue;
+				}
+			}
 #if defined ___AsyncIO___
 			WriteAsync(fp, min, (LONG)strlen(min));
 			WriteAsync(fp, "\n", 1);
 #else
 			fprintf(fp, "%s\n", min);
 #endif
 			if(rem != NULL) {
 				free(rem);
+			}
 			if((rem = malloc((strlen(min) + 1) * sizeof(*rem))) == NULL) {
 				fprintf(stderr, "Memory allocation failure.\n");
 				free(min);
 #if defined ___AsyncIO___
 				CloseAsync(fp);
 #else
 				fclose(fp);
 #endif
 				return;
+			}
 			sprintf(rem, "%s", min);
 			free(min);
+		}
+	}
 	if(rem != NULL) {
 		free(rem);
+	}
 	// Write out any remaining contents from the temporary files.
 	for(i = 0; PROGRAM_RUN && i < files; ++i) {
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
 		if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		tmp = ReadLine(tp[i]);
 		if(tmp == NULL) {
 			continue;
+		}
 #if defined ___AsyncIO___
 		WriteAsync(fp, tmp, (LONG)strlen(tmp));
 		WriteAsync(fp, "\n", 1);
 #else
 		fprintf(fp, "%s\n", tmp);
 #endif
 		free(tmp);
+	}
 	// Close all temporary files.
 	for(i = 0; i < files; ++i) {
 #if defined ___AsyncIO___
 		CloseAsync(tp[i]);
 #else
 		fclose(tp[i]);
 #endif
+	}
 #if defined ___AsyncIO___
 	CloseAsync(fp);
 #else
 	fclose(fp);
 #endif
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "\n");
+	}
+}
 /*
+  *
   * Filter the paths inside the database with provided paths.
   */
 void FilterDatabasePaths(char *dbFile, char *tmpName, char **paths, int count) {
 #if defined ___AsyncIO___
 	struct AsyncFile *fp;
 	struct AsyncFile *tp;
 #else
 	FILE *fp;
 	FILE *tp;
 #endif
 	char *line;
 	int lines;
 	int i;
 	dbEntry *entry;
 	// Open database file for reading.
 #if defined ___AsyncIO___
 	if((fp = OpenAsync(dbFile, MODE_READ, ASYNC_BUF)) == NULL) {
 #else
 	if((fp = fopen(dbFile, "r")) == NULL) {
 #endif
 		fprintf(stderr, "Could not open file '%s' for reading.\n", dbFile);
 		return;
+	}
 	// Open temporary file for writing.
 #if defined ___AsyncIO___
 	if((tp = OpenAsync(tmpName, MODE_WRITE, ASYNC_BUF)) == NULL) {
 #else
 	if((tp = fopen(tmpName, "w")) == NULL) {
 #endif
 		fprintf(stderr, "Copuld not open file '%s' for writing.\n", tmpName);
 		// Close database file.
 #if defined ___AsyncIO___
 		CloseAsync(fp);
 #else
 		fclose(fp);
 #endif
 		return;
+	}
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "Removing lines...\r");
+	}
 	lines = 0;
 	while(PROGRAM_RUN && (line = ReadLine(fp)) != NULL) {
 #if defined ___AmigaOS___
 		// Check if CTRL+C was pressed and abort the program.
    	if(SetSignal(0L, SIGBREAKF_CTRL_C) & SIGBREAKF_CTRL_C) {
 			PROGRAM_RUN = FALSE;
 			continue;
+		}
 #endif
 		if((entry = CreateDatabaseEntry(line)) == NULL) {
 			fprintf(stderr, "Unable to create database entry.\n");
 			free(line);
 			continue;
+		}
 		for(i = 0; i < count; ++i) {
 			if(PathCompare(entry->path, paths[i]) == TRUE) {
 				++lines;
 				if(PROGRAM_VERBOSE) {
 					fprintf(stdout, "Removing lines: %d.\r", lines);
+				}
 				continue;
+			}
 #if defined ___AsyncIO___
 			WriteAsync(tp, line, (LONG)strlen(line));
 			WriteAsync(tp, "\n", 1);
 #else
 			fprintf(tp, "%s\n", line);
 #endif
 			break;
+		}
 		free(entry->name);
 		free(entry->path);
 		free(entry);
 		free(line);
+	}
 #if defined ___AsyncIO___
 	CloseAsync(fp);
 	CloseAsync(tp);
 #else
 	fclose(fp);
 	fclose(tp);
 #endif
 	if(PROGRAM_VERBOSE) {
 		fprintf(stdout, "\n");
+	}
+}
 /*
+	*
 	* Indexes a "path" by creating a database "dbFile".
 	*/
 void GatherDatabaseFiles(char *dbFile, char **paths, int count) {
 	dbStats *stats;
 	char **tmpNames;
 	int tmpFiles;
 	int tmpLines;
 	int i;
 	// Generate the database file from the supplied paths.
 	if((stats = CollectFiles(dbFile, paths, count)) == NULL) {
 		fprintf(stderr, "Collecting files failed.\n");
 		return;
+	}
 	// Calculate the total number of temporary files required.
 	tmpFiles = stats->size / maxmem;
 	/* In case no temporary files are required,
 	 * just sort the database and terminate.
 	 */
 	if(tmpFiles < 2) {
 		SortDatabase(dbFile, stats->lines);
 		return;
+	}
 	// Calculate the number of lines per temporary file.
 	tmpLines = stats->lines / tmpFiles;
 	// Create temporary files.
 	if((tmpNames = CreateTemporaryFiles(tmpFiles)) == NULL) {
 		fprintf(stderr, "Unable to create temporary files.\n");
 		return;
+	}
 	// Write "tmpLines" to temporary files in "tmpNames" from "dbFile".
 	WriteTemporaryFiles(dbFile, tmpNames, tmpFiles, tmpLines, stats->lines);
 	// Sort the temporary files.
 	for(i = 0; i < tmpFiles; ++i) {
-		tmpLines = CountFileLines(tmpNames[i]);
+		// Twice the number of computed lines required should cover the last file.
-		SortDatabase(tmpNames[i], tmpLines);
+		SortDatabase(tmpNames[i], 2 * tmpLines);
+	}
 	// Merge all the temporary files to the database file.
 	MergeTemporaryFiles(dbFile, tmpNames, tmpFiles, stats->lines);
 	// Remove all temporary files.
 	RemoveFiles(tmpNames, tmpFiles);
 	// Free temporary file names.
 	free(tmpNames);
 	// Free statistics.
 	free(stats);
+}
 void RemoveDatabaseFiles(char *dbFile, char **paths, int count) {
 	char *tmpName;
 	// Create a temporary file to hold the changes.
 	if((tmpName = CreateTemporaryFile()) == NULL) {
 		fprintf(stderr, "Unable to create temporary file.\n");
 		return;
+	}
 	// Filter the database of the provided paths.
 	FilterDatabasePaths(dbFile, tmpName, paths, count);
 	// Overwrite the database file with the filtered paths.
 	CopyFile(tmpName, dbFile);
 	// Remove temporary file.
 	if(RemoveFile(tmpName) == FALSE) {
 		fprintf(stderr, "Temporary file could not be removed.\n");
 		return;
+	}
+}
 void usage(char *name) {
 	fprintf(stdout, "Hunt & Gather - %s, a file index generating tool.        \n", name);
 	fprintf(stdout, "Version: %s                                              \n", PROGRAM_VERSION);
 	fprintf(stdout, "                                                         \n");
 	fprintf(stdout, "SYNTAX: %s [-q] <-a|-r|-c> <PATH PATH PATH...>           \n", name);
 	fprintf(stdout, "                                                         \n");
 	fprintf(stdout, "Required:                                                \n");
 	fprintf(stdout, "    -a [PATH...]   Add files.                            \n");
 	fprintf(stdout, "    -c [PATH...]   Create from scratch.                  \n");
 	fprintf(stdout, "    -r [PATH...]   Remove files.                         \n");
 	fprintf(stdout, "                                                         \n");
 	fprintf(stdout, "Optional:                                                \n");
 	fprintf(stdout, "    -d [FIILE]     Where to store the database.          \n");
 	fprintf(stdout, "    -m BYTES       Memory to use (default: %d).          \n", maxmem);
 	fprintf(stdout, "    -q             Do not print out any messages.        \n");
 	fprintf(stdout, "                                                         \n");
 	fprintf(stdout, "DATABASE is a path to where the indexed results will be  \n");
 	fprintf(stdout, "stored for searching with the Hunt tool.                 \n");
 	fprintf(stdout, "                                                         \n");
 	fprintf(stdout, "(c) 2021 Wizardry and Steamworks, MIT.                   \n");
+}
 /*
+	*
 	* Main entry point.
 	*/
 int main(int argc, char **argv) {
 	int option;
 	int i;
 	int count;
 	char *dbFile;
 	char *path;
 	char **paths;
 	OPERATION operation = NONE;
 	// Bind handler to SIGINT.
 #if !defined ___AmigaOS___
  	signal(SIGINT, SignalHandler);
 #endif
 	dbFile = DEFAULT_DATABASE_FILE;
 	while((option = getopt(argc, argv, "hqdm:arc")) != -1) {
 		switch(option) {
 			case 'a':
 				operation = GATHER;
 				break;
 			case 'r':
 				operation = REMOVE;
 				break;
 			case 'c':
 				operation = CREATE;
 				break;
 			case 'm':
 				maxmem = strtoul(optarg, NULL, 10);
 				break;
 			case 'd':
 				dbFile = optarg;
 				break;
 			case 'q':
 				PROGRAM_VERBOSE = FALSE;
 				break;
 			case 'h':
 				usage(argv[0]);
 				return 0;
 			case '?':
 				fprintf(stderr, "Invalid option %ct.\n", optopt);
-				return 1;
+				return 5;
+		}
+	}
 	if(operation == NONE) {
 		usage(argv[0]);
-		return 1;
+		return 5;
+	}
 	if(optind >= argc) {
 		usage(argv[0]);
-		return 1;
+		return 5;
+	}
 	// Go through all supplied arguments and add paths to search.
 	if((paths = malloc((argc - optind) * sizeof(*paths))) == NULL) {
 		fprintf(stderr, "Memory allocation failure.\n");
-		return 1;
+		return 20;
+	}
 	count = 0;
 	for(i = optind, count = 0; i < argc; ++i) {
 		if((path = PathToAbsolute(argv[i])) == NULL) {
 			fprintf(stderr, "Absolute path for '%s' failed to resolve.\n", argv[optind]);
 			continue;
+		}
 		switch(GetFsType(path)) {
 			case UNKNOWN:
 			case REGULAR:
 				fprintf(stderr, "Path '%s' is not a directory.\n", path);
 				free(path);
 				continue;
 				break;
 			case DIRECTORY:
 				break;
+		}
 		if(PROGRAM_VERBOSE) {
 			fprintf(stdout, "Will process path: '%s'\n", path);
+		}
 		// Add the path to the array of paths.
 		if((paths[count] = malloc((strlen(path) + 1) * sizeof(*paths[count]))) == NULL) {
 			fprintf(stderr, "Memory allocation failure.");
-			return 1;
+			return 20;
+		}
 		sprintf(paths[count], "%s", path);
 		++count;
 		free(path);
+	}
 	if(count == 0) {
 		fprintf(stderr, "No valid paths are available.\n");
 		free(paths);
-		return 1;
+		return 5;
+	}
 	if(PROGRAM_VERBOSE) {
    		fprintf(stdout, "Gathering to: '%s'\n", dbFile);
+	}
 #if defined ___AmigaOS___
 	locale = OpenLocale(NULL);
 #endif
  	switch(operation) {
  		case CREATE:
  			if(PROGRAM_VERBOSE) {
  				fprintf(stdout, "Removing '%s' and creating a new database.\n", dbFile);
+ 			}
- 			if(RemoveFile(dbFile) == FALSE) {
+ 			RemoveFile(dbFile);
- 				fprintf(stderr, "File '%s' could not be removed.\n", dbFile);
- 				break;
- 			}
  		case GATHER:
  			if(PROGRAM_VERBOSE) {
  				fprintf(stdout, "Gathering files to database...\n");
+ 			}
  			GatherDatabaseFiles(dbFile, paths, count);
  			break;
  		case REMOVE:
  			if(PROGRAM_VERBOSE) {
  				fprintf(stdout, "Removing files from database...\n");
+ 			}
  			RemoveDatabaseFiles(dbFile, paths, count);
  			break;
  		default:
  			fprintf(stderr, "Unknown operation.\n");
+			#if defined ___AmigaOS___
+				CloseLocale(locale);
+			#endif
+ 			free(paths);
- 			break;
+ 			return 5;
+ 	}
 #if defined ___AmigaOS___
 	CloseLocale(locale);
 #endif
  	free(paths);
  	return 0;
+}