Added Download Enabler

DownloadEnabler/CMakeLists.txt

@@ -0,0 +1,29 @@
+cmake_minimum_required(VERSION 2.8)
+
+if(NOT DEFINED CMAKE_TOOLCHAIN_FILE)
+  if(DEFINED ENV{VITASDK})
+    set(CMAKE_TOOLCHAIN_FILE "$ENV{VITASDK}/share/vita.toolchain.cmake" CACHE PATH "toolchain file")
+  else()
+    message(FATAL_ERROR "Please define VITASDK to point to your SDK path!")
+  endif()
+endif()
+
+project(download_enabler)
+include("${VITASDK}/share/vita.cmake" REQUIRED)
+
+set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Wl,-q -Wall -O3 -nostdlib")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fno-rtti -fno-exceptions")
+
+add_executable(download_enabler
+  main.c
+)
+
+target_link_libraries(download_enabler
+  SceIofilemgr_stub
+  SceLibKernel_stub
+  taiHEN_stub
+)
+
+vita_create_self(download_enabler.suprx download_enabler CONFIG exports.yml UNSAFE)
+
+vita_create_stubs(stubs download_enabler ${CMAKE_SOURCE_DIR}/exports.yml)

DownloadEnabler/exports.yml

@@ -0,0 +1,8 @@
+DownloadEnabler:
+  attributes: 0
+  version:
+    major: 1
+    minor: 0
+  main:
+    start: module_start
+    stop: module_stop

DownloadEnabler/main.c

@@ -0,0 +1,173 @@
+/*
+  VitaTweaks: Download Enabler
+  Copyright (C) 2018, TheFloW
+
+  This program is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  This program is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <psp2/io/dirent.h>
+#include <psp2/io/fcntl.h>
+#include <psp2/kernel/clib.h>
+#include <psp2/kernel/modulemgr.h>
+
+#include <taihen.h>
+
+static tai_hook_ref_t ExportFileRef;
+static tai_hook_ref_t GetFileTypeRef;
+
+static SceUID hooks[3];
+
+static int ExportFilePatched(uint32_t *data) {
+  int res = TAI_CONTINUE(int, ExportFileRef, data);
+
+  if (res == 0x80101A09) {
+    char download_path[1024];
+    char bgdl_path[1024];
+    char file_name[256];
+    char short_name[256];
+    uint16_t url_length = 0;
+    uint32_t count = 0;
+
+    uint32_t num = *(uint32_t *)data[0];
+
+    sceClibSnprintf(bgdl_path, sizeof(bgdl_path), "ux0:bgdl/t/%08x/d0.pdb", num);
+
+    SceUID fd = sceIoOpen(bgdl_path, SCE_O_RDONLY, 0);
+    if (fd < 0)
+      return fd;
+
+    sceIoPread(fd, &url_length, sizeof(uint16_t), 0xD6);
+    sceIoPread(fd, file_name, sizeof(file_name), 0xF7 + url_length);
+    sceIoClose(fd);
+
+    sceClibSnprintf(bgdl_path, sizeof(bgdl_path), "ux0:bgdl/t/%08x/%s", num, file_name);
+
+    char *ext = sceClibStrrchr(file_name, '.');
+    if (ext) {
+      int len = ext-file_name;
+      if (len > sizeof(short_name) - 1)
+        len = sizeof(short_name) - 1;
+      sceClibStrncpy(short_name, file_name, len);
+      short_name[len] = '\0';
+    } else {
+      sceClibStrncpy(short_name, file_name, sizeof(short_name));
+      ext = "";
+    }
+
+    while (1) {
+      if (count == 0)
+        sceClibSnprintf(download_path, sizeof(download_path), "ux0:download/%s", file_name);
+      else
+        sceClibSnprintf(download_path, sizeof(download_path), "ux0:download/%s (%d)%s", short_name, count, ext);
+
+      SceIoStat stat;
+      sceClibMemset(&stat, 0, sizeof(SceIoStat));
+      if (sceIoGetstat(download_path, &stat) < 0)
+        break;
+
+      count++;
+    }
+
+    res = sceIoMkdir("ux0:download", 0006);
+    if (res < 0 && res != 0x80010011)
+      return res;
+
+    return sceIoRename(bgdl_path, download_path);
+  }
+
+  return res;
+}
+
+static int GetFileTypePatched(int unk, int *type, char **filename, char **mime_type) {
+  int res = TAI_CONTINUE(int, GetFileTypeRef, unk, type, filename, mime_type);
+
+  if (res == 0x80103A21) {
+    *type = 1; // Type photo
+    return 0;
+  }
+
+  return res;
+}
+
+void _start() __attribute__ ((weak, alias("module_start")));
+int module_start(SceSize args, void *argp) {
+  tai_module_info_t info;
+  info.size = sizeof(info);
+  if (taiGetModuleInfo("SceShell", &info) >= 0) {
+    switch (info.module_nid) {
+      case 0x0552F692: // 3.60 retail
+      {
+        hooks[0] = taiInjectData(info.modid, 0, 0x50A4A8, "GET", 4);
+        hooks[1] = taiHookFunctionOffset(&ExportFileRef, info.modid, 0, 0x1163F6, 1, ExportFilePatched);
+        hooks[2] = taiHookFunctionOffset(&GetFileTypeRef, info.modid, 0, 0x11B5E4, 1, GetFileTypePatched);
+        break;
+      }
+
+      case 0x6CB01295: // 3.60 PDEL
+      {
+        hooks[0] = taiInjectData(info.modid, 0, 0x4F9A18, "GET", 4);
+        hooks[1] = taiHookFunctionOffset(&ExportFileRef, info.modid, 0, 0x111D5A, 1, ExportFilePatched);
+        hooks[2] = taiHookFunctionOffset(&GetFileTypeRef, info.modid, 0, 0x116F48, 1, GetFileTypePatched);
+        break;
+      }
+
+      case 0xEAB89D5C: // 3.60 PTEL
+      {
+        hooks[0] = taiInjectData(info.modid, 0, 0x4FEBF8, "GET", 4);
+        hooks[1] = taiHookFunctionOffset(&ExportFileRef, info.modid, 0, 0x112756, 1, ExportFilePatched);
+        hooks[2] = taiHookFunctionOffset(&GetFileTypeRef, info.modid, 0, 0x117944, 1, GetFileTypePatched);
+        break;
+      }
+
+      case 0x5549BF1F: // 3.65 retail
+      case 0x34B4D82E: // 3.67 retail
+      case 0x12DAC0F3: // 3.68 retail
+      {
+        hooks[0] = taiInjectData(info.modid, 0, 0x50A9E8, "GET", 4);
+        hooks[1] = taiHookFunctionOffset(&ExportFileRef, info.modid, 0, 0x11644E, 1, ExportFilePatched);
+        hooks[2] = taiHookFunctionOffset(&GetFileTypeRef, info.modid, 0, 0x11B63C, 1, GetFileTypePatched);
+        break;
+      }
+
+      case 0xE6A02F2B: // 3.65 PDEL
+      {
+        hooks[0] = taiInjectData(info.modid, 0, 0x4F9F58, "GET", 4);
+        hooks[1] = taiHookFunctionOffset(&ExportFileRef, info.modid, 0, 0x111DB2, 1, ExportFilePatched);
+        hooks[2] = taiHookFunctionOffset(&GetFileTypeRef, info.modid, 0, 0x116FA0, 1, GetFileTypePatched);
+        break;
+      }
+
+      case 0x587F9CED: // 3.65 PTEL
+      {
+        hooks[0] = taiInjectData(info.modid, 0, 0x4FF0F8, "GET", 4);
+        hooks[1] = taiHookFunctionOffset(&ExportFileRef, info.modid, 0, 0x1127AE, 1, ExportFilePatched);
+        hooks[2] = taiHookFunctionOffset(&GetFileTypeRef, info.modid, 0, 0x11799C, 1, GetFileTypePatched);
+        break;
+      }
+    }
+  }
+
+  return SCE_KERNEL_START_SUCCESS;
+}
+
+int module_stop(SceSize args, void *argp) {
+  if (hooks[2] >= 0)
+    taiHookRelease(hooks[2], GetFileTypeRef);
+  if (hooks[1] >= 0)
+    taiHookRelease(hooks[1], ExportFileRef);
+  if (hooks[0] >= 0)
+    taiInjectRelease(hooks[0]);
+
+  return SCE_KERNEL_STOP_SUCCESS;
+}

README.md

@@ -11,3 +11,12 @@ Download [nolockscreen.suprx](https://github.com/TheOfficialFloW/VitaTweaks/rele
 ux0:tai/nolockscreen.suprx
 ```
 
+## 2. Download Enabler
+
+Download [download_enabler.suprx](https://github.com/TheOfficialFloW/VitaTweaks/releases/tag/DownloadEnabler) and enable it in `*main` as follows:
+
+```
+*main
+ux0:tai/download_enabler.suprx
+```
+