Merge branch 'master' of github.com:astronautlevel2/Anemone3DS

astronautlevel2 · Jun 12, 2020 · d1a821e · d1a821e
2 parents a4532b0 + f819f9d
commit d1a821e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/include/remote.h b/include/remote.h
@@ -29,6 +29,7 @@
 
 #include "common.h"
 #include "draw.h"
+#include <ctype.h>
 
 #define THEMEPLAZA_BASE_URL "https://themeplaza.eu"
 #define THEMEPLAZA_API_URL "/api/anemone/v1"

diff --git a/source/remote.c b/source/remote.c
@@ -807,10 +807,7 @@ u32 http_get(const char *url, char ** filename, char ** buf, InstallType install
             return 0;
         }
 
-        DEBUG(content_disposition);
-        char * tok = strtok(content_disposition, "\"");
-        DEBUG(tok);
-        tok = strtok(NULL, "\"");
+        char * tok = strstr(content_disposition, "filename=");
 
         if(!(tok))
         {
@@ -822,6 +819,17 @@ u32 http_get(const char *url, char ** filename, char ** buf, InstallType install
             return 0;
         }
 
+        tok += sizeof("filename=") - 1;
+        if(ispunct((int)*tok))
+        {
+            tok++;
+        }
+        char* last_char = tok + strlen(tok) - 1;
+        if(ispunct((int)*last_char))
+        {
+            *last_char = '\0';
+        }
+
         char *illegal_characters = "\"?;:/\\+";
         for (size_t i = 0; i < strlen(tok); i++)
         {