Wii can now send a POST request to a website

The extractor now uses libfat to read a config file off the root of the
SD card, a model for which is included in the repository. The extractor
is capable of negotiating an HTTPS connection, but it requires a
download of the target site's root CA certificate to be on the SD card
and pointed at by the config file. TLS/SSL functionality is provided by
wolfSSL. I had to make a couple of minor changes to wolfSSL for it to
work properly, and those changes are located in the devkitpro branch of
my fork, hatkirby/wolfssl.

The Wii program now arranges some of the information that the GBA sends
it into a JSON object, which is then sent off (along with some
authentication information from the config file) to the endpoint defined
in the config file. The code used to maintain the network connection is
from the WiiTweet project, which was licensed under the GPL. Some of the
code used to send the HTTP request also comes from said project.

1 files changed, 229 insertions, 15 deletions

@@ -11,10 +11,14 @@
 #include <stdlib.h>
 #include <unistd.h>
 #include <string.h>
+#include <fat.h>
+#include "cJSON.h"
 #include "link.h"
 #include "encoding.h"
 #include "multiboot.h"
 #include "pokemon.h"
+#include "http.h"
+#include "netinf.h"
 
 void printmain()
 {
@@ -65,8 +69,75 @@ void fatalError(char *msg)
   exit(0);
 }
 
+cJSON* getConfig()
+{
+  FILE* fp = fopen("/gen3uploader.cfg", "r");
+  if (!fp)
+  {
+    fatalError("ERROR: Could not find config file!\n");
+  }
+
+  fseek(fp, 0L, SEEK_END);
+  long lSize = ftell(fp);
+  rewind(fp);
+
+  char* buffer = calloc(1, lSize+1);
+  if (!buffer)
+  {
+    fclose(fp);
+    fatalError("ERROR: Could not allocate memory.\n");
+  }
+
+  if (fread(buffer, lSize, 1, fp) != 1)
+  {
+    fclose(fp);
+    free(buffer);
+    fatalError("ERROR: Could not read config file.\n");
+  }
+
+  cJSON* config = cJSON_Parse(buffer);
+
+  fclose(fp);
+  free(buffer);
+
+  return config;
+}
+
 void* extractor(void* userdata)
 {
+  cJSON* config = (cJSON*)userdata;
+
+  WOLFSSL_CTX* sslContext = 0;
+  if (cJSON_HasObjectItem(config, "certificate"))
+  {
+    wolfSSL_Init();
+
+    sslContext = wolfSSL_CTX_new(wolfTLSv1_client_method());
+    if (sslContext == NULL)
+    {
+      fatalError("wolfSSL_CTX_new error.\n");
+    }
+
+    if (wolfSSL_CTX_load_verify_locations(
+      sslContext,
+      cJSON_GetObjectItem(config, "certificate")->valuestring,
+      0) != SSL_SUCCESS)
+    {
+      fatalError("Error loading certificate file.\n");
+    }
+  }
+
+  while (!initializeNetwork())
+  {
+    printf("Could not initialize network.\n");
+    printf("Press A to try again, press B to exit.\n");
+
+    if (waitForButtons(PAD_BUTTON_A | PAD_BUTTON_B) == PAD_BUTTON_B)
+    {
+      endproc();
+    }
+  }
+
   for (;;)
   {
     printmain();
@@ -174,6 +245,17 @@ void* extractor(void* userdata)
 
     sendMsg(1);
 
+    cJSON* root = cJSON_CreateObject();
+
+    cJSON_AddItemToObject(
+      root,
+      "playerName",
+      cJSON_CreateString(d_trainerName));
+
+    cJSON_AddNumberToObject(root, "playerId", trainerId);
+    cJSON_AddNumberToObject(root, "gameId", gameId);
+    cJSON_AddNumberToObject(root, "language", gameLanguage);
+
     // Get Pokédex data
     u32 pokedexSeen[13];
     u32 pokedexCaught[13];
@@ -208,6 +290,8 @@ void* extractor(void* userdata)
 
     u32 partyCount = getMsg();
 
+    cJSON* jParty = cJSON_CreateArray();
+
     for (u32 i = 0; i < partyCount; i++)
     {
       usleep(5000);
@@ -235,22 +319,144 @@ void* extractor(void* userdata)
         pki->key[5],
         pki->key[6]);
 
-      printf("Species: %d\n", __builtin_bswap16(pki->species));
-      printf("Nickname: %s\n", d_pokename);
-      printf("OT: %s\n", d_otName);
-      printf("Level: %d\n", pki->level);
-      printf("HP: %ld\n", __builtin_bswap32(pki->hp));
-      printf("Attack: %ld\n", __builtin_bswap32(pki->attack));
-      printf("Defense: %ld\n", __builtin_bswap32(pki->defense));
-      printf("Special Attack: %ld\n", __builtin_bswap32(pki->spAttack));
-      printf("Special Defense: %ld\n", __builtin_bswap32(pki->spDefense));
-      printf("Speed: %ld\n", __builtin_bswap32(pki->speed));
-      printf("Key: %s\n", d_key);
-
-      printf("\n");
+      cJSON* jPoke = cJSON_CreateObject();
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "species",
+        __builtin_bswap16(pki->species));
+
+      cJSON_AddItemToObject(
+        jPoke,
+        "nickname",
+        cJSON_CreateString(d_pokename));
+
+      cJSON_AddItemToObject(
+        jPoke,
+        "otName",
+        cJSON_CreateString(d_otName));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "otId",
+        __builtin_bswap16(pki->otId));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "level",
+        pki->level);
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "hp",
+        __builtin_bswap32(pki->hp));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "attack",
+        __builtin_bswap32(pki->attack));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "defense",
+        __builtin_bswap32(pki->defense));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "speed",
+        __builtin_bswap32(pki->speed));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "spAttack",
+        __builtin_bswap32(pki->spAttack));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "spDefense",
+        __builtin_bswap32(pki->spDefense));
+
+      cJSON_AddItemToObject(
+        jPoke,
+        "key",
+        cJSON_CreateString(d_key));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "experience",
+        __builtin_bswap32(pki->experience));
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "heldItem",
+        __builtin_bswap16(pki->heldItem));
+
+      cJSON* jMoves = cJSON_CreateArray();
+
+      for (int j=0; j<4; j++)
+      {
+        if (pki->moves[j] != 0)
+        {
+          cJSON* jMove = cJSON_CreateObject();
+
+          cJSON_AddNumberToObject(
+            jMove,
+            "id",
+            __builtin_bswap16(pki->moves[j]));
+
+          cJSON_AddNumberToObject(
+            jMove,
+            "ppBonuses",
+            (pki->ppBonuses >> (2*j)) & 3);
+
+          cJSON_AddItemToArray(jMoves, jMove);
+        } else {
+          break;
+        }
+      }
+
+      cJSON_AddItemToObject(
+        jPoke,
+        "moves",
+        jMoves);
+
+      if (pki->otGender)
+      {
+        cJSON_AddStringToObject(jPoke, "otGender", "female");
+      } else {
+        cJSON_AddStringToObject(jPoke, "otGender", "male");
+      }
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "metLevel",
+        pki->metLevel);
+
+      cJSON_AddNumberToObject(
+        jPoke,
+        "metLocation",
+        pki->metLocation);
+
+      cJSON_AddItemToArray(jParty, jPoke);
     }
 
-    waitForButtons(PAD_BUTTON_START);
+    cJSON_AddItemToObject(root, "party", jParty);
+
+    char *rendered = cJSON_Print(root);
+    printf("%s\n", rendered);
+
+    waitForButtons(PAD_BUTTON_A);
+
+    int result = submitToApi(
+      cJSON_GetObjectItem(config, "url")->valuestring,
+      sslContext,
+      root,
+      cJSON_GetObjectItem(config, "username")->valuestring,
+      cJSON_GetObjectItem(config, "password")->valuestring);
+
+    printf("Result: %d\n", result);
+
+    waitForButtons(PAD_BUTTON_A);
   }
 
   return NULL;
@@ -276,6 +482,14 @@ int main(int argc, char *argv[])
   VIDEO_ClearFrameBuffer(rmode, xfb, COLOR_BLACK);
   PAD_Init();
 
+  if (!fatInitDefault())
+  {
+    printmain();
+    fatalError("ERROR: Cannot find config file!\n");
+  }
+
+  cJSON* config = getConfig();
+
   initLink();
 
   lwp_t extractorHandle = (lwp_t)NULL;
@@ -283,7 +497,7 @@ int main(int argc, char *argv[])
   LWP_CreateThread(
     &extractorHandle,   // thread handle
     extractor,          // code
-    NULL,               // userdata
+    config,             // userdata
     NULL,               // stack base
     16*1024,            // stack size
     LWP_PRIO_HIGHEST);  // thread priority