Code clean-up.

frontend/MCrypt.vcxproj

@@ -127,7 +127,7 @@
       <RuntimeLibrary>MultiThreaded</RuntimeLibrary>
       <BufferSecurityCheck>false</BufferSecurityCheck>
       <FloatingPointModel>Fast</FloatingPointModel>
-      <EnableEnhancedInstructionSet>StreamingSIMDExtensions2</EnableEnhancedInstructionSet>
+      <EnableEnhancedInstructionSet>NoExtensions</EnableEnhancedInstructionSet>
     </ClCompile>
     <Link>
       <SubSystem>Console</SubSystem>
@@ -165,7 +165,6 @@
       <RuntimeLibrary>MultiThreaded</RuntimeLibrary>
       <BufferSecurityCheck>false</BufferSecurityCheck>
       <FloatingPointModel>Fast</FloatingPointModel>
-      <EnableEnhancedInstructionSet>StreamingSIMDExtensions2</EnableEnhancedInstructionSet>
     </ClCompile>
     <Link>
       <SubSystem>Console</SubSystem>

frontend/src/main.c

@@ -146,6 +146,14 @@ static const uint64_t CRC64_TAB[256] =
 	0xd80c07cd676f8394ULL,  0x9afce626ce85b507ULL,
 };
 
+static void erase(void *const ptr, const size_t length)
+{
+	volatile uint8_t* buffer = ptr;
+	for (size_t i = 0U; i < length; ++i)
+	{
+		buffer[i] = 0U;
+	}
+}
 
 static char* utf16_to_bytes(const wchar_t* const input, const UINT code_page)
 {
@@ -184,59 +192,73 @@ static void print_string(const char *const text, const size_t length)
 	fputs("\"\n\n", stderr);
 }
 
+static int open_files(FILE** const fin, FILE** const fout, const wchar_t* const input, const wchar_t* const output)
+{
+	*fin = _wfopen(input, L"rb");
+	if (!(*fin))
+	{
+		fputws(L"Error: Failed to open input file for reading!\n\n", stderr);
+		return 1;
+	}
+
+	*fout = _wfopen(output, L"wb");
+	if (!(*fout))
+	{
+		fputws(L"Error: Failed to open output file for writing!\n\n", stderr);
+		fclose(*fin);
+		return 1;
+	}
+
+	return 0;
+}
+
+static uint64_t get_file_size(FILE *const file)
+{
+	struct _stati64 stat;
+	if (_fstati64(_fileno(file), &stat) != 0)
+	{
+		fputws(L"I/O error: Failed to determine size of input file!\n\n", stderr);
+		return UINT64_MAX;
+	}
+	return (stat.st_size >= 0) ? ((uint64_t)stat.st_size) : 0U;
+}
+
 static int encrypt(const wchar_t* const passphrase, const wchar_t* const input, const wchar_t* const output)
 {
-	char* passphrase_utf8 = utf16_to_bytes(passphrase, CP_UTF8);
+	mcrypt_t ctx = NULL;
+	FILE *fin = NULL, *fout = NULL;
+	int result = -1;
+
+	char *const passphrase_utf8 = utf16_to_bytes(passphrase, CP_UTF8);
 	if (!passphrase_utf8)
 	{
 		fputws(L"Error: Failed to convert passphrase to UTF-8 format!\n\n", stderr);
-		return 1;
+		goto clean_up;
 	}
 
-	FILE *fin = _wfopen(input, L"rb");
+	if (open_files(&fin, &fout, input, output) != 0)
-	if (!fin)
 	{
-		fputws(L"Error: Failed to open input file!\n\n", stderr);
+		goto clean_up;;
-		free(passphrase_utf8);
-		return 1;
-	}
-
-	FILE* fout = _wfopen(output, L"wb");
-	if (!fout)
-	{
-		fputws(L"Error: Failed to open output file!\n\n", stderr);
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
 	uint64_t seed;
 	if (mcrypt_generate_seed(&seed) != 0)
 	{
 		fputws(L"MCrypt error: Failed to generate seed!\n\n", stderr);
-		fclose(fout);
+		goto  clean_up;
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
 	if (fwrite(&seed, sizeof(uint64_t), 1U, fout) < 1U)
 	{
-		fputws(L"I/O error: Failed to write encrypted data!\n\n", stderr);
+		fputws(L"I/O error: Failed to write seed value!\n\n", stderr);
-		fclose(fout);
+		goto clean_up;
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	const mcrypt_t ctx = mcrypt_init(seed, passphrase_utf8);
+	ctx = mcrypt_alloc(seed, passphrase_utf8);
 	if (!ctx)
 	{
 		fputws(L"MCrypt error: Failed to initialize encryption!\n\n", stderr);
-		fclose(fout);
+		goto clean_up;
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
 	fputws(L"Encrypting, please be patient... ", stderr);
@@ -250,11 +272,7 @@ static int encrypt(const wchar_t* const passphrase, const wchar_t* const input,
 		if (ferror(fin))
 		{
 			fputws(L"failed!\n\nI/O error: Failed to read input data!\n\n", stderr);
-			mcrypt_free(ctx);
+			goto clean_up;
-			fclose(fout);
-			fclose(fin);
-			free(passphrase_utf8);
-			return 1;
 		}
 		if (count > 0U)
 		{
@@ -265,20 +283,12 @@ static int encrypt(const wchar_t* const passphrase, const wchar_t* const input,
 			if (mcrypt_enc_process_inplace(ctx, buffer, count) != 0)
 			{
 				fputws(L"failed!\n\nMCrypt error: Failed to encrypt data!\n\n", stderr);
-				mcrypt_free(ctx);
+				goto  clean_up;
-				fclose(fout);
-				fclose(fin);
-				free(passphrase_utf8);
-				return 1;
 			}
 			if (fwrite(buffer, sizeof(uint8_t), count, fout) < count)
 			{
 				fputws(L"failed!\n\nI/O error: Failed to write encrypted data!\n\n", stderr);
-				mcrypt_free(ctx);
+				goto clean_up;
-				fclose(fout);
-				fclose(fin);
-				free(passphrase_utf8);
-				return 1;
 			}
 		}
 	}
@@ -288,85 +298,82 @@ static int encrypt(const wchar_t* const passphrase, const wchar_t* const input,
 	if (fwrite(&crc_actual, sizeof(uint64_t), 1U, fout) < 1U)
 	{
 		fputws(L"failed!\n\nI/O error: Failed to write CRC checksum!\n\n", stderr);
-		mcrypt_free(ctx);
+		goto clean_up;
-		fclose(fout);
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	fputws(L"done.\n\n", stderr);
+	result = 0;
 
+	fputws(L"done.\n\n", stderr);
+	fflush(stderr);
+
+clean_up:
+
+	if (ctx)
+	{
 		mcrypt_free(ctx);
+	}
+
+	if (fout)
+	{
 		fclose(fout);
+	}
+
+	if (fin)
+	{
 		fclose(fin);
+	}
+
+	if (passphrase_utf8)
+	{
+		erase(passphrase_utf8, strlen(passphrase_utf8));
 		free(passphrase_utf8);
-	return 0;
+	}
+
+	return result;
 }
 
 static int decrypt(const wchar_t* const passphrase, const wchar_t* const input, const wchar_t* const output)
 {
-	char* passphrase_utf8 = utf16_to_bytes(passphrase, CP_UTF8);
+	mcrypt_t ctx = NULL;
+	FILE *fin = NULL, *fout = NULL;
+	int result = -1;
+
+	char *const passphrase_utf8 = utf16_to_bytes(passphrase, CP_UTF8);
 	if (!passphrase_utf8)
 	{
 		fputws(L"Error: Failed to convert passphrase to UTF-8 format!\n\n", stderr);
-		return 1;
+		goto clean_up;
 	}
 
-	FILE* fin = _wfopen(input, L"rb");
+	if (open_files(&fin, &fout, input, output) != 0)
-	if (!fin)
 	{
-		fputws(L"Error: Failed to open input file!\n\n", stderr);
+		goto clean_up;
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	FILE* fout = _wfopen(output, L"wb");
+	const uint64_t file_size = get_file_size(fin);
-	if (!fout)
+	if (file_size == UINT64_MAX)
 	{
-		fputws(L"Error: Failed to open output file!\n\n", stderr);
+		goto clean_up;
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	struct _stati64 stat;
+	if (file_size < 16LL)
-	if (_fstati64(_fileno(fin), &stat) != 0)
-	{
-		fputws(L"I/O error: Failed to determine size of input file!\n\n", stderr);
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
-	}
-
-	if (stat.st_size < 12LL)
 	{
 		fputws(L"Error: Input file is too small!\n\n", stderr);
-		fclose(fin);
+		goto clean_up;
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	const int64_t limit = stat.st_size - sizeof(uint64_t);
-
 	uint64_t seed;
 	if (fread(&seed, sizeof(uint64_t), 1U, fin) < 1U)
 	{
-		fputws(L"I/O error: Failed to read encrypted data!\n\n", stderr);
+		fputws(L"I/O error: Failed to read seed value!\n\n", stderr);
-		fclose(fout);
+		goto clean_up;
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	const mcrypt_t ctx = mcrypt_init(seed, passphrase_utf8);
+	ctx = mcrypt_alloc(seed, passphrase_utf8);
 	if (!ctx)
 	{
 		fputws(L"MCrypt error: Failed to initialize decryption!\n\n", stderr);
-		fclose(fout);
+		goto clean_up;
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
 	fputws(L"Decrypting, please be patient... ", stderr);
@@ -375,19 +382,16 @@ static int decrypt(const wchar_t* const passphrase, const wchar_t* const input,
 	int64_t bytes_read = sizeof(uint64_t);
 	uint8_t buffer[1024U];
 
-	while ((!feof(fin)) && (bytes_read < limit))
+	const int64_t read_limit = file_size - sizeof(uint64_t);
+	while ((!feof(fin)) && (bytes_read < read_limit))
 	{
-		const int64_t bytes_remaining = limit - bytes_read;
+		const int64_t bytes_remaining = read_limit - bytes_read;
 		const size_t read_len = (bytes_remaining < 1024U) ? ((size_t)bytes_remaining) : 1024U;
 		const size_t count = fread(buffer, sizeof(uint8_t), read_len, fin);
 		if (ferror(fin))
 		{
 			fputws(L"failed!\n\nI/O error: Failed to read encrypted data!\n\n", stderr);
-			mcrypt_free(ctx);
+			goto clean_up;
-			fclose(fout);
-			fclose(fin);
-			free(passphrase_utf8);
-			return 1;
 		}
 		if (count > 0U)
 		{
@@ -395,11 +399,7 @@ static int decrypt(const wchar_t* const passphrase, const wchar_t* const input,
 			if (mcrypt_dec_process_inplace(ctx, buffer, count) != 0)
 			{
 				fputws(L"failed!\n\nMCrypt error: Failed to decrypt data!\n\n", stderr);
-				mcrypt_free(ctx);
+				goto clean_up;
-				fclose(fout);
-				fclose(fin);
-				free(passphrase_utf8);
-				return 1;
 			}
 			for (size_t i = 0U; i < count; ++i)
 			{
@@ -408,55 +408,61 @@ static int decrypt(const wchar_t* const passphrase, const wchar_t* const input,
 			if (fwrite(buffer, sizeof(uint8_t), count, fout) < count)
 			{
 				fputws(L"failed!\n\nI/O error: Failed to write decrypted data!\n\n", stderr);
-				mcrypt_free(ctx);
+				goto clean_up;
-				fclose(fout);
-				fclose(fin);
-				free(passphrase_utf8);
-				return 1;
 			}
 		}
 	}
 
 	crc_actual ^= ~0U;
 
-	if (bytes_read < limit)
+	if (bytes_read < read_limit)
 	{
 		fputws(L"failed!\n\nI/O error: Input file could not be fully read!\n\n", stderr);
-		mcrypt_free(ctx);
+		goto clean_up;
-		fclose(fout);
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
 	uint64_t crc_expected;
 	if (fread(&crc_expected, sizeof(uint64_t), 1U, fin) < 1U)
 	{
 		fputws(L"failed!\n\nI/O error: Failed to read CRC checksum!\n\n", stderr);
-		mcrypt_free(ctx);
+		goto clean_up;
-		fclose(fout);
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
 	if (crc_actual != crc_expected)
 	{
 		fwprintf(stderr, L"failed!\n\nCRC error: Checksum mismatch detected! [0x%016llX vs. 0x%016llX]\n\nWrong passphrase?\n\n", crc_actual, crc_expected);
-		mcrypt_free(ctx);
+		goto clean_up;
-		fclose(fout);
-		fclose(fin);
-		free(passphrase_utf8);
-		return 1;
 	}
 
-	fputws(L"done.\n\nCRC checksum is correct.\n\n", stderr);
+	result = 0;
 
+	fputws(L"done.\n\nCRC checksum is correct.\n\n", stderr);
+	fflush(stderr);
+
+clean_up:
+
+	if (ctx)
+	{
 		mcrypt_free(ctx);
+	}
+
+	if (fout)
+	{
 		fclose(fout);
+	}
+
+	if (fin)
+	{
 		fclose(fin);
+	}
+
+	if (passphrase_utf8)
+	{
+		erase(passphrase_utf8, strlen(passphrase_utf8));
 		free(passphrase_utf8);
-	return 0;
+	}
+
+	return result;
 }
 
 int wmain(int argc, wchar_t* argv[])
@@ -474,15 +480,20 @@ int wmain(int argc, wchar_t* argv[])
 	int result = -1;
 	const clock_t clk_start = clock();
 
-	if ((argc > 4) && (_wcsicmp(argv[1U], L"--decrypt") == 0))
+	if ((argc > 4) && (!_wcsicmp(argv[1U], L"--decrypt")))
 	{
 		result = decrypt(argv[2U], argv[3U], argv[4U]);
+		erase(argv[2U], wcslen(argv[2U]) * sizeof(wchar_t));
 	}
 	else
 	{
 		result = encrypt(argv[1U], argv[2U], argv[3U]);
+		erase(argv[1U], wcslen(argv[1U]) * sizeof(wchar_t));
 	}
 
+	fputws(L"--------\n\n", stderr);
+	fflush(stderr);
+
 	const clock_t clk_end = clock();
-	fwprintf(stderr, L"--------\n\nOperation completed after %.1f seconds.\n\n", ((double)(clk_end - clk_start)) / ((double)CLOCKS_PER_SEC));
+	fwprintf(stderr, L"Operation completed after %.1f seconds.\n\n", (clk_end - clk_start) / ((double)CLOCKS_PER_SEC));
 }

libMCrypt/include/mcrypt.h

@@ -12,11 +12,11 @@
 extern const char *const LIBMCRYPT_VERSION;
 extern const char* const LIBMCRYPT_BUILDNO;
 
-typedef uintptr_t mcrypt_t;
+typedef void* mcrypt_t;
 
 int mcrypt_generate_seed(uint64_t* const seed);
 
-mcrypt_t mcrypt_init(const uint64_t salt, const char* const passphrase);
+mcrypt_t mcrypt_alloc(const uint64_t salt, const char* const passphrase);
 void mcrypt_free(const mcrypt_t context);
 
 int mcrypt_enc_process(const mcrypt_t context, const uint8_t* const input, uint8_t* const output, size_t length);

libMCrypt/libMCrypt.vcxproj

@@ -128,7 +128,7 @@
       <RuntimeLibrary>MultiThreaded</RuntimeLibrary>
       <BufferSecurityCheck>false</BufferSecurityCheck>
       <FloatingPointModel>Fast</FloatingPointModel>
-      <EnableEnhancedInstructionSet>StreamingSIMDExtensions2</EnableEnhancedInstructionSet>
+      <EnableEnhancedInstructionSet>NoExtensions</EnableEnhancedInstructionSet>
     </ClCompile>
     <Link>
       <SubSystem>
@@ -170,7 +170,6 @@
       <RuntimeLibrary>MultiThreaded</RuntimeLibrary>
       <BufferSecurityCheck>false</BufferSecurityCheck>
       <FloatingPointModel>Fast</FloatingPointModel>
-      <EnableEnhancedInstructionSet>StreamingSIMDExtensions2</EnableEnhancedInstructionSet>
     </ClCompile>
     <Link>
       <SubSystem>

libMCrypt/src/mcrypt.c

@@ -76,7 +76,7 @@ static int random_bytes(uint8_t* const buffer, const size_t length)
 #endif
 }
 
-static void erase(void* ptr, const size_t length)
+static void erase(void *const ptr, const size_t length)
 {
 	volatile uint8_t* buffer = ptr;
 	for (size_t i = 0U; i < length; ++i)
@@ -214,7 +214,7 @@ int mcrypt_generate_seed(uint64_t* const seed)
 	return -1;
 }
 
-mcrypt_t mcrypt_init(const uint64_t salt, const char* const passphrase)
+mcrypt_t mcrypt_alloc(const uint64_t salt, const char* const passphrase)
 {
 	if (!passphrase)
 	{
@@ -292,6 +292,6 @@ void mcrypt_free(const mcrypt_t context)
 	if (context)
 	{
 		erase((void*)context, sizeof(crypt_state_t));
-		free((void*)context);
+		free(context);
 	}
 }