1 files changed, 161 insertions, 0 deletions

@@ -0,0 +1,161 @@
+/*
+ * Copyright (C) 2008 Tantric
+ * Copyright (C) 2012 Pedro Aguiar
+ * Copyright (C) 2017 hatkirby
+ *
+ * Originally part of the WiiTweet project, which was distributed under the
+ * GPL.
+ */
+#include "netinf.h"
+#include <gccore.h>
+#include <network.h>
+#include <ogc/lwp_watchdog.h>
+#include <sys/errno.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdio.h>
+
+static bool networkInit = false;
+static lwp_t networkThread = LWP_THREAD_NULL;
+static u8 netStack[32768] ATTRIBUTE_ALIGN(32);
+static char wiiIP[16] = {0};
+static bool netHalt = false;
+
+static void* interface(void* arg)
+{
+  static bool prevInit = false;
+
+  while (!netHalt)
+  {
+    int retry = 5;
+    int result = -1;
+
+    while ((retry > 0) && (!netHalt))
+    {
+      if (prevInit)
+      {
+        net_deinit();
+
+        for (int i=0; (i<400) && !netHalt; i++)
+        {
+          if (net_get_status() != -EBUSY)
+          {
+            usleep(2000);
+            net_wc24cleanup();
+            prevInit = false;
+            usleep(20000);
+
+            break;
+          }
+
+          usleep(20000);
+        }
+      }
+
+      usleep(2000);
+
+      if (net_init_async(NULL, NULL))
+      {
+        sleep(1);
+
+        retry--;
+
+        continue;
+      }
+
+      result = net_get_status();
+      for (int wait = 400; (wait > 0) && (result == -EBUSY) && !netHalt; wait++)
+      {
+        usleep(20000);
+
+        result = net_get_status();
+      }
+
+      if (result == 0)
+      {
+        break;
+      }
+
+      retry--;
+
+      usleep(2000);
+    }
+
+    if (result == 0)
+    {
+      struct in_addr hostip;
+      hostip.s_addr = net_gethostip();
+
+      if (hostip.s_addr)
+      {
+        strcpy(wiiIP, inet_ntoa(hostip));
+        networkInit = true;
+        prevInit = true;
+      }
+    }
+
+    if (!netHalt)
+    {
+      LWP_SuspendThread(networkThread);
+    }
+  }
+
+  return NULL;
+}
+
+static void startNetworkThread()
+{
+  netHalt = false;
+
+  if (networkThread == LWP_THREAD_NULL)
+  {
+    LWP_CreateThread(&networkThread, interface, NULL, netStack, 8192, 40);
+  } else {
+    LWP_ResumeThread(networkThread);
+  }
+}
+
+static void stopNetworkThread()
+{
+  if ((networkThread == LWP_THREAD_NULL)
+    || (!LWP_ThreadIsSuspended(networkThread)))
+  {
+    return;
+  }
+
+  netHalt = true;
+
+  LWP_ResumeThread(networkThread);
+  LWP_JoinThread(networkThread, NULL);
+
+  networkThread = LWP_THREAD_NULL;
+}
+
+bool initializeNetwork()
+{
+  stopNetworkThread();
+
+  if (networkInit && (net_gethostip() > 0))
+  {
+    return true;
+  }
+
+  networkInit = false;
+
+  printf("Initializing network...\n");
+
+  u64 start = gettime();
+  startNetworkThread();
+
+  while (!LWP_ThreadIsSuspended(networkThread))
+  {
+    usleep(50 * 1000);
+
+    if (diff_sec(start, gettime()) > 10)
+    {
+      break;
+    }
+  }
+
+  return networkInit;
+}