From c6dc5dfc4b55f56abd8ee0d146728cccf1c45388 Mon Sep 17 00:00:00 2001
From: Arisotura <thetotalworm@gmail.com>
Date: Mon, 15 Jun 2020 14:35:02 +0200
Subject: [PATCH] get it to atleast finish a scan without shitting itself

---
 src/DSi_NWifi.cpp | 77 ++++++++++++++++++++++++++++++++++++++++-------
 src/DSi_NWifi.h   |  3 +-
 src/Wifi.cpp      |  3 +-
 3 files changed, 69 insertions(+), 14 deletions(-)

diff --git a/src/DSi_NWifi.cpp b/src/DSi_NWifi.cpp
index 4843cfb2..a0778bbf 100644
--- a/src/DSi_NWifi.cpp
+++ b/src/DSi_NWifi.cpp
@@ -21,7 +21,6 @@
 #include "DSi.h"
 #include "DSi_NWifi.h"
 #include "SPI.h"
-#include "WifiAP.h"
 
 
 const u8 CIS0[256] =
@@ -178,6 +177,7 @@ void DSi_NWifi::Reset()
     BootPhase = 0;
 
     ErrorMask = 0;
+    ScanTimer = 0;
 
     NDS::CancelEvent(NDS::Event_DSi_NWifi);
 }
@@ -826,6 +826,29 @@ void DSi_NWifi::WMI_Command()
         }
         break;
 
+    case 0x0007: // start scan
+        {
+            u32 forcefg = MB_Read32(0);
+            u32 legacy = MB_Read32(0);
+            u32 scantime = MB_Read32(0);
+            u32 forceinterval = MB_Read32(0);
+            u8 scantype = Mailbox[0]->Read();
+            u8 nchannels = Mailbox[0]->Read();
+
+            printf("WMI: start scan, forceFG=%d, legacy=%d, scanTime=%d, interval=%d, scanType=%d, chan=%d\n",
+                   forcefg, legacy, scantime, forceinterval, scantype, nchannels);
+
+            if (ScanTimer > 0)
+            {
+                printf("!! CHECKME: START SCAN BUT WAS ALREADY SCANNING (%d)\n", ScanTimer);
+            }
+
+            ScanTimer = scantime;
+
+            SendWMIAck();
+        }
+        break;
+
     case 0x0008: // set scan params
         {
             // TODO: do something with the params!!
@@ -913,6 +936,33 @@ void DSi_NWifi::WMI_Command()
         }
         break;
 
+    case 0x002E: // extension shit
+        {
+            u16 extcmd = MB_Read16(0);
+            switch (extcmd)
+            {
+            case 0x2008: // 'heartbeat'??
+                {
+                    u32 cookie = MB_Read32(0);
+                    u32 source = MB_Read32(0);
+
+                    u8 reply[10];
+                    *(u16*)&reply[0] = 0x3007;
+                    *(u32*)&reply[2] = cookie;
+                    *(u32*)&reply[6] = source;
+                    SendWMIEvent(1, 0x1010, reply, 10);
+                }
+                break;
+
+            default:
+                printf("WMI: unknown ext cmd 002E:%04X\n", extcmd);
+                break;
+            }
+
+            SendWMIAck();
+        }
+        break;
+
     case 0x0047: // cmd47 -- timer shenanigans??
         {
             //
@@ -1053,12 +1103,20 @@ void DSi_NWifi::WindowWrite(u32 addr, u32 val)
 }
 
 
-void DSi_NWifi::CheckRX()
-{return;
-    u16 framelen;
-    u16 framectl;
+void DSi_NWifi::_MSTimer()
+{
+    if (ScanTimer > 0)
+    {
+        ScanTimer--;
 
-    for (;;)
+        if (ScanTimer == 0)
+        {
+            u32 status = 0;
+            SendWMIEvent(1, 0x100A, (u8*)&status, 4);
+        }
+    }
+
+    /*for (;;)
     {
         int rxlen = WifiAP::RecvPacket(RXBuffer);
         if (rxlen == 0) return;
@@ -1092,14 +1150,11 @@ void DSi_NWifi::CheckRX()
             memcpy(&beacon_evt[16], &RXBuffer[12+24], bodylen); // frame body
             //SendWMIFrame(beacon_evt, 16+bodylen, 1, 0, 0);
         }
-    }
+    }*/
 }
 
 void DSi_NWifi::MSTimer(u32 param)
 {
-    WifiAP::MSTimer();
-
-    Ctx->CheckRX();
-
+    Ctx->_MSTimer();
     NDS::ScheduleEvent(NDS::Event_DSi_NWifi, true, 33611, MSTimer, 0);
 }
diff --git a/src/DSi_NWifi.h b/src/DSi_NWifi.h
index 8693b189..68c37957 100644
--- a/src/DSi_NWifi.h
+++ b/src/DSi_NWifi.h
@@ -37,7 +37,7 @@ public:
 
     void SetIRQ_F1_Counter(u32 n);
 
-    void CheckRX();
+    void _MSTimer();
 
     static void MSTimer(u32 param);
 
@@ -125,6 +125,7 @@ private:
     u32 BootPhase;
 
     u32 ErrorMask;
+    u32 ScanTimer;
 
     u8 RXBuffer[2048];
 };
diff --git a/src/Wifi.cpp b/src/Wifi.cpp
index 54fbceb2..ab827244 100644
--- a/src/Wifi.cpp
+++ b/src/Wifi.cpp
@@ -966,8 +966,7 @@ void MSTimer()
 
 void USTimer(u32 param)
 {
-    // FIXME!!!
-    //WifiAP::USTimer();
+    WifiAP::USTimer();
 
     if (IOPORT(W_USCountCnt))
     {