From 10658d615751eb841553703a4862d602db85f990 Mon Sep 17 00:00:00 2001
From: "Prof. 9" <prof9@outlook.com>
Date: Sat, 24 May 2014 18:04:45 +0200
Subject: [PATCH] Abort when no HTTPS URLs are found.

---
 WfcReplay/Program.cs | 55 +++++++++++++++++++++++++++++---------------
 1 file changed, 36 insertions(+), 19 deletions(-)

diff --git a/WfcReplay/Program.cs b/WfcReplay/Program.cs
index c56efed..098ef6f 100644
--- a/WfcReplay/Program.cs
+++ b/WfcReplay/Program.cs
@@ -27,30 +27,38 @@ namespace WfcReplay
 			{
 #if !DEBUG
 				try
-				{
 #endif
+				{
 					string romPath = args[0];
 
 					Stream rom = readFile(romPath);
 					BinaryReader romReader = new BinaryReader(rom);
 					string code = new Program().process(romReader);
-
-					MemoryStream outStream = new MemoryStream(Encoding.UTF8.GetBytes(code));
-					outStream.Position = 0;
-					string fileName = makeGameString(romReader) + ".txt";
-					fileName = makeFileNameSafe(fileName);
-					writeFile(fileName, outStream);
-					Console.WriteLine("Success!");
-					Console.WriteLine("Code written to " + Directory.GetCurrentDirectory() + "\\" + fileName + ".");
-#if !DEBUG
+					Console.WriteLine("Finished analyzing ROM.");
+					
+					if (code != null)
+					{
+						MemoryStream outStream = new MemoryStream(Encoding.UTF8.GetBytes(code));
+						outStream.Position = 0;
+						string fileName = makeGameString(romReader) + ".txt";
+						fileName = makeFileNameSafe(fileName);
+						writeFile(fileName, outStream);
+						Console.WriteLine("Success!");
+						Console.WriteLine("Code written to " + Directory.GetCurrentDirectory() + "\\" + fileName + ".");
+					}
+					else
+					{
+						Console.WriteLine("No HTTPS URLs to patch were found.");
+					}
 				}
+#if !DEBUG
 				catch (Exception ex)
 				{
 					Console.WriteLine("FATAL ERROR: " + ex.Message);
 				}
 #endif
 #if DEBUG
-			Console.ReadKey();
+				Console.ReadKey();
 #endif
 			}
 		}
@@ -198,14 +206,21 @@ namespace WfcReplay
 			urlAddresses = urlAddresses.Distinct().ToList();
 			urlAddresses.Sort();
 
-			// Find largest code cave
-			arm9.Position = 0;
-			List<uint> codeCave = findCodeCave(arm9Reader, (uint)arm9RamAddress);
+			if (urlAddresses.Count > 0)
+			{
+				// Find largest code cave
+				arm9.Position = 0;
+				List<uint> codeCave = findCodeCave(arm9Reader, (uint)arm9RamAddress);
 
-			// Form code
-			string code = "::Bypass HTTPS " + version + " for " + gameString + "\r\n";
-			code += makeCode((uint)hookAddress, urlAddresses, codeCave);
-			return code;
+				// Form code
+				string code = "::Bypass HTTPS " + version + " for " + gameString + "\r\n";
+				code += makeCode((uint)hookAddress, urlAddresses, codeCave);
+				return code;
+			}
+			else
+			{
+				return null;
+			}
 		}
 		void log(string s)
 		{
@@ -501,9 +516,11 @@ namespace WfcReplay
 				{
 					code += " 00000000";
 				}
+
+				code = code.Substring(2);
 			}
 
-			return code.Substring(2);
+			return code;
 		}
 	}
 }
\ No newline at end of file