From 642771463aa2bfd4fddf82f333447087b1a5652e Mon Sep 17 00:00:00 2001
From: Michael <TheShadowEeveeYT@gmail.com>
Date: Sat, 19 Mar 2022 21:58:35 -0500
Subject: [PATCH] Add NUS checking (#2)

* Add NUS checking

* Remove obsolete variable
---
 .gitignore         |  3 +-
 NusClient.cs       | 93 ++++++++++++++++++++++++++++++++++++----------
 libWiiSharp.csproj |  6 ---
 3 files changed, 75 insertions(+), 27 deletions(-)

diff --git a/.gitignore b/.gitignore
index b803dcb..f28928e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 /.vs
 /bin
-/obj
\ No newline at end of file
+/obj
+/Log.cs
diff --git a/NusClient.cs b/NusClient.cs
index 8ed5ed0..e5197e2 100644
--- a/NusClient.cs
+++ b/NusClient.cs
@@ -35,7 +35,6 @@ namespace libWiiSharp
 
     public class NusClient : IDisposable
     {
-        private const string nusUrl = "http://ccs.cdn.wup.shop.nintendo.net/ccs/download/";
 #pragma warning disable SYSLIB0014 // Type or member is obsolete
         private readonly WebClient wcNus = new WebClient();
 #pragma warning restore SYSLIB0014 // Type or member is obsolete
@@ -148,11 +147,7 @@ namespace libWiiSharp
             contentId = num.ToString("x8");
             FireDebug("Downloading Content (Content ID: {0}) of Title {1} v{2}...", contentId, titleId, string.IsNullOrEmpty(titleVersion) ? "[Latest]" : titleVersion);
             FireDebug("   Checking for Internet connection...");
-            if (!PrivCheckInet())
-            {
-                FireDebug("   Connection not found...");
-                throw new Exception("You're not connected to the internet!");
-            }
+            string nusUrl = PrivNUSUp();
             FireProgress(0);
             string str1 = "tmd" + (string.IsNullOrEmpty(titleVersion) ? string.Empty : "." + titleVersion);
             string str2 = string.Format("{0}{1}/", nusUrl, titleId);
@@ -215,10 +210,7 @@ namespace libWiiSharp
 
         private Ticket PrivDownloadTicket(string titleId)
         {
-            if (!PrivCheckInet())
-            {
-                throw new Exception("You're not connected to the internet!");
-            }
+            string nusUrl = PrivNUSUp();
 
             string titleUrl = string.Format("{0}{1}/", nusUrl, titleId);
             byte[] tikArray = wcNus.DownloadData(titleUrl + "cetk");
@@ -228,10 +220,7 @@ namespace libWiiSharp
 
         private TMD PrivDownloadTmd(string titleId, string titleVersion)
         {
-            if (!PrivCheckInet())
-            {
-                throw new Exception("You're not connected to the internet!");
-            }
+            string nusUrl = PrivNUSUp();
 
             return TMD.Load(wcNus.DownloadData(string.Format("{0}{1}/", nusUrl, titleId) + ("tmd" + (string.IsNullOrEmpty(titleVersion) ? string.Empty : "." + titleVersion))));
         }
@@ -248,6 +237,8 @@ namespace libWiiSharp
                 FireDebug("  No store types were defined...");
                 throw new Exception("You must at least define one store type!");
             }
+            FireDebug("   Checking for Internet connection...");
+            string nusUrl = PrivNUSUp();
             string str1 = string.Format("{0}{1}/", nusUrl, titleId);
             bool flag1 = false;
             bool flag2 = false;
@@ -285,12 +276,6 @@ namespace libWiiSharp
                 flag1 = true;
                 flag3 = false;
             }
-            FireDebug("   Checking for Internet connection...");
-            if (!PrivCheckInet())
-            {
-                FireDebug("   Connection not found...");
-                throw new Exception("You're not connected to the internet!");
-            }
             if (outputDir[outputDir.Length - 1] != Path.DirectorySeparatorChar)
             {
                 outputDir += Path.DirectorySeparatorChar.ToString();
@@ -495,6 +480,74 @@ namespace libWiiSharp
             return buffer;
         }
 
+        private string PrivNUSUp()
+        {
+            if (!PrivCheckInet())
+            {
+                FireDebug("   Connection not found...");
+                throw new Exception("You're not connected to the internet!");
+            }
+
+            const string WiiEndpoint = "http://nus.cdn.shop.wii.com/ccs/download/";
+            const string WiiUEndpoint = "http://ccs.cdn.wup.shop.nintendo.net/ccs/download/";
+            const string DSiEndpoint = "http://nus.cdn.t.shop.nintendowifi.net/ccs/download/";
+            const string RC24Endpoint = "http://ccs.cdn.sho.rc24.xyz/ccs/download/";
+
+            // Wii Endpoint
+            try
+            {
+                wcNus.DownloadData(WiiEndpoint);
+
+            } catch (WebException e)
+            {
+                if (e.Message.Split('(')[1].Split(')')[0] == "401")
+                {
+                    return WiiEndpoint;
+                }
+            }
+
+            // WiiU Endpoint
+            try
+            {
+                wcNus.DownloadData(WiiUEndpoint);
+            }
+            catch (WebException e)
+            {
+                if (e.Message.Split('(')[1].Split(')')[0] == "401")
+                {
+                    return WiiUEndpoint;
+                }
+            }
+
+            // DSi Endpoint
+            try
+            {
+                wcNus.DownloadData(DSiEndpoint);
+            }
+            catch (WebException e)
+            {
+                if (e.Message.Split('(')[1].Split(')')[0] == "401")
+                {
+                    return DSiEndpoint;
+                }
+            }
+
+            // RC24 Endpoint
+            try
+            {
+                wcNus.DownloadData(RC24Endpoint);
+            }
+            catch (WebException e)
+            {
+                if (e.Message.Split('(')[1].Split(')')[0] == "401")
+                {
+                    return RC24Endpoint;
+                }
+            }
+
+            throw new Exception("Unable to verify any online NUS server!");
+        }
+
         private bool PrivCheckInet()
         {
             try
diff --git a/libWiiSharp.csproj b/libWiiSharp.csproj
index fae96e0..a825793 100644
--- a/libWiiSharp.csproj
+++ b/libWiiSharp.csproj
@@ -9,12 +9,6 @@
     <ImportWindowsDesktopTargets>true</ImportWindowsDesktopTargets>
     <Nullable>annotations</Nullable>
   </PropertyGroup>
-  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
-    <RegisterForComInterop>true</RegisterForComInterop>
-  </PropertyGroup>
-  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
-    <RegisterForComInterop>false</RegisterForComInterop>
-  </PropertyGroup>
   <ItemGroup>
     <Content Include=".gitignore" />
   </ItemGroup>