mirror of
https://gitlab.com/Theopse/fbi-i18n-zh.git
synced 2025-04-06 03:58:02 +08:00
1b78bebd94
* Maintain list position after operations. * Show current target info in batch operations. * Fix freeze when suspending to the home menu during a threaded task. * Other miscellaneous fixes and clean-ups.
419 lines
13 KiB
C
419 lines
13 KiB
C
#include <malloc.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include <3ds.h>
|
|
|
|
#include "section.h"
|
|
#include "task/task.h"
|
|
#include "../error.h"
|
|
#include "../info.h"
|
|
#include "../prompt.h"
|
|
#include "../ui.h"
|
|
#include "../../core/screen.h"
|
|
#include "../../quirc/quirc_internal.h"
|
|
|
|
#define IMAGE_WIDTH 400
|
|
#define IMAGE_HEIGHT 240
|
|
|
|
#define URL_MAX 1024
|
|
#define URLS_MAX 128
|
|
|
|
typedef struct {
|
|
struct quirc* qrContext;
|
|
char urls[URLS_MAX][URL_MAX];
|
|
|
|
Handle mutex;
|
|
u16* buffer;
|
|
u32 tex;
|
|
Handle camCancelEvent;
|
|
|
|
u32 responseCode;
|
|
u64 currTitleId;
|
|
bool ticket;
|
|
|
|
data_op_info installInfo;
|
|
Handle installCancelEvent;
|
|
} qr_install_data;
|
|
|
|
static Result qrinstall_is_src_directory(void* data, u32 index, bool* isDirectory) {
|
|
*isDirectory = false;
|
|
return 0;
|
|
}
|
|
|
|
static Result qrinstall_make_dst_directory(void* data, u32 index) {
|
|
return 0;
|
|
}
|
|
|
|
static Result qrinstall_open_src(void* data, u32 index, u32* handle) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
Result res = 0;
|
|
|
|
httpcContext* context = (httpcContext*) calloc(1, sizeof(httpcContext));
|
|
if(context != NULL) {
|
|
if(R_SUCCEEDED(res = httpcOpenContext(context, HTTPC_METHOD_GET, qrInstallData->urls[index], 1))) {
|
|
httpcSetSSLOpt(context, SSLCOPT_DisableVerify);
|
|
if(R_SUCCEEDED(res = httpcBeginRequest(context)) && R_SUCCEEDED(res = httpcGetResponseStatusCode(context, &qrInstallData->responseCode, 0))) {
|
|
if(qrInstallData->responseCode == 200) {
|
|
*handle = (u32) context;
|
|
} else {
|
|
res = R_FBI_HTTP_RESPONSE_CODE;
|
|
}
|
|
}
|
|
|
|
if(R_FAILED(res)) {
|
|
httpcCloseContext(context);
|
|
}
|
|
}
|
|
|
|
if(R_FAILED(res)) {
|
|
free(context);
|
|
}
|
|
} else {
|
|
res = R_FBI_OUT_OF_MEMORY;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
static Result qrinstall_close_src(void* data, u32 index, bool succeeded, u32 handle) {
|
|
return httpcCloseContext((httpcContext*) handle);
|
|
}
|
|
|
|
static Result qrinstall_get_src_size(void* data, u32 handle, u64* size) {
|
|
u32 downloadSize = 0;
|
|
Result res = httpcGetDownloadSizeState((httpcContext*) handle, NULL, &downloadSize);
|
|
|
|
*size = downloadSize;
|
|
return res;
|
|
}
|
|
|
|
static Result qrinstall_read_src(void* data, u32 handle, u32* bytesRead, void* buffer, u64 offset, u32 size) {
|
|
Result res = httpcDownloadData((httpcContext*) handle, buffer, size, bytesRead);
|
|
return res != HTTPC_RESULTCODE_DOWNLOADPENDING ? res : 0;
|
|
}
|
|
|
|
static Result qrinstall_open_dst(void* data, u32 index, void* initialReadBlock, u32* handle) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
qrInstallData->ticket = *(u16*) initialReadBlock == 0x0100;
|
|
|
|
Result res = 0;
|
|
|
|
if(qrInstallData->ticket) {
|
|
res = AM_InstallTicketBegin(handle);
|
|
} else {
|
|
u8* cia = (u8*) initialReadBlock;
|
|
|
|
u32 headerSize = *(u32*) &cia[0x00];
|
|
u32 certSize = *(u32*) &cia[0x08];
|
|
u64 titleId = __builtin_bswap64(*(u64*) &cia[((headerSize + 0x3F) & ~0x3F) + ((certSize + 0x3F) & ~0x3F) + 0x1DC]);
|
|
|
|
FS_MediaType dest = ((titleId >> 32) & 0x8010) != 0 ? MEDIATYPE_NAND : MEDIATYPE_SD;
|
|
|
|
u8 n3ds = false;
|
|
if(R_SUCCEEDED(APT_CheckNew3DS(&n3ds)) && !n3ds && ((titleId >> 28) & 0xF) == 2) {
|
|
return R_FBI_WRONG_SYSTEM;
|
|
}
|
|
|
|
// Deleting FBI before it reinstalls itself causes issues.
|
|
if(((titleId >> 8) & 0xFFFFF) != 0xF8001) {
|
|
AM_DeleteTitle(dest, titleId);
|
|
AM_DeleteTicket(titleId);
|
|
|
|
if(dest == MEDIATYPE_SD) {
|
|
AM_QueryAvailableExternalTitleDatabase(NULL);
|
|
}
|
|
}
|
|
|
|
if(R_SUCCEEDED(res = AM_StartCiaInstall(dest, handle))) {
|
|
qrInstallData->currTitleId = titleId;
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
static Result qrinstall_close_dst(void* data, u32 index, bool succeeded, u32 handle) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
if(succeeded) {
|
|
Result res = 0;
|
|
|
|
if(qrInstallData->ticket) {
|
|
res = AM_InstallTicketFinish(handle);
|
|
} else {
|
|
if(R_SUCCEEDED(res = AM_FinishCiaInstall(handle))) {
|
|
if(qrInstallData->currTitleId == 0x0004013800000002 || qrInstallData->currTitleId == 0x0004013820000002) {
|
|
res = AM_InstallFirm(qrInstallData->currTitleId);
|
|
}
|
|
}
|
|
}
|
|
|
|
return res;
|
|
} else {
|
|
if(qrInstallData->ticket) {
|
|
return AM_InstallTicketAbort(handle);
|
|
} else {
|
|
return AM_CancelCIAInstall(handle);
|
|
}
|
|
}
|
|
}
|
|
|
|
static Result qrinstall_write_dst(void* data, u32 handle, u32* bytesWritten, void* buffer, u64 offset, u32 size) {
|
|
return FSFILE_Write(handle, bytesWritten, offset, buffer, size, 0);
|
|
}
|
|
|
|
static bool qrinstall_error(void* data, u32 index, Result res) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
if(res == R_FBI_CANCELLED) {
|
|
prompt_display("Failure", "Install cancelled.", COLOR_TEXT, false, NULL, NULL, NULL, NULL);
|
|
return false;
|
|
} else {
|
|
char* url = qrInstallData->urls[index];
|
|
|
|
volatile bool dismissed = false;
|
|
if(res == R_FBI_WRONG_SYSTEM) {
|
|
if(strlen(url) > 48) {
|
|
error_display(&dismissed, NULL, NULL, "Failed to install from QR code.\n%.45s...\nAttempted to install N3DS title to O3DS.", url);
|
|
} else {
|
|
error_display(&dismissed, NULL, NULL, "Failed to install from QR code.\n%.48s\nAttempted to install N3DS title to O3DS.", url);
|
|
}
|
|
} else if(res == R_FBI_HTTP_RESPONSE_CODE) {
|
|
if(strlen(url) > 48) {
|
|
error_display(&dismissed, NULL, NULL, "Failed to install from QR code.\n%.45s...\nHTTP server returned response code %d", url, qrInstallData->responseCode);
|
|
} else {
|
|
error_display(&dismissed, NULL, NULL, "Failed to install from QR code.\n%.48s\nHTTP server returned response code %d", url, qrInstallData->responseCode);
|
|
}
|
|
} else {
|
|
if(strlen(url) > 48) {
|
|
error_display_res(&dismissed, NULL, NULL, res, "Failed to install from QR code.\n%.45s...", url);
|
|
} else {
|
|
error_display_res(&dismissed, NULL, NULL, res, "Failed to install from QR code.\n%.48s", url);
|
|
}
|
|
}
|
|
|
|
while(!dismissed) {
|
|
svcSleepThread(1000000);
|
|
}
|
|
}
|
|
|
|
return index < qrInstallData->installInfo.total - 1;
|
|
}
|
|
|
|
static void qrinstall_install_update(ui_view* view, void* data, float* progress, char* text) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
if(qrInstallData->installInfo.finished) {
|
|
ui_pop();
|
|
info_destroy(view);
|
|
|
|
if(!qrInstallData->installInfo.premature) {
|
|
prompt_display("Success", "Install finished.", COLOR_TEXT, false, data, NULL, NULL, NULL);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
if(hidKeysDown() & KEY_B) {
|
|
svcSignalEvent(qrInstallData->installCancelEvent);
|
|
}
|
|
|
|
*progress = qrInstallData->installInfo.currTotal != 0 ? (float) ((double) qrInstallData->installInfo.currProcessed / (double) qrInstallData->installInfo.currTotal) : 0;
|
|
snprintf(text, PROGRESS_TEXT_MAX, "%lu / %lu\n%.2f MB / %.2f MB", qrInstallData->installInfo.processed, qrInstallData->installInfo.total, qrInstallData->installInfo.currProcessed / 1024.0 / 1024.0, qrInstallData->installInfo.currTotal / 1024.0 / 1024.0);
|
|
}
|
|
|
|
static void qrinstall_confirm_onresponse(ui_view* view, void* data, bool response) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
if(response) {
|
|
qrInstallData->installCancelEvent = task_data_op(&qrInstallData->installInfo);
|
|
if(qrInstallData->installCancelEvent != 0) {
|
|
info_display("Installing From QR Code", "Press B to cancel.", true, data, qrinstall_install_update, NULL);
|
|
} else {
|
|
error_display(NULL, NULL, NULL, "Failed to initiate installation.");
|
|
}
|
|
}
|
|
}
|
|
|
|
static void qrinstall_free_data(qr_install_data* data) {
|
|
if(data->camCancelEvent != 0) {
|
|
svcSignalEvent(data->camCancelEvent);
|
|
while(svcWaitSynchronization(data->camCancelEvent, 0) == 0) {
|
|
svcSleepThread(1000000);
|
|
}
|
|
|
|
data->camCancelEvent = 0;
|
|
}
|
|
|
|
if(data->qrContext != NULL) {
|
|
quirc_destroy(data->qrContext);
|
|
data->qrContext = NULL;
|
|
}
|
|
|
|
if(data->buffer != NULL) {
|
|
free(data->buffer);
|
|
data->buffer = NULL;
|
|
}
|
|
|
|
if(data->tex != 0) {
|
|
screen_unload_texture(data->tex);
|
|
data->tex = 0;
|
|
}
|
|
|
|
free(data);
|
|
}
|
|
|
|
static void qrinstall_wait_draw_top(ui_view* view, void* data, float x1, float y1, float x2, float y2) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
if(qrInstallData->tex != 0) {
|
|
screen_draw_texture(qrInstallData->tex, 0, 0, IMAGE_WIDTH, IMAGE_HEIGHT);
|
|
}
|
|
}
|
|
|
|
static void qrinstall_wait_update(ui_view* view, void* data, float* progress, char* text) {
|
|
qr_install_data* qrInstallData = (qr_install_data*) data;
|
|
|
|
if(hidKeysDown() & KEY_B) {
|
|
ui_pop();
|
|
info_destroy(view);
|
|
|
|
qrinstall_free_data(qrInstallData);
|
|
|
|
return;
|
|
}
|
|
|
|
if(qrInstallData->tex != 0) {
|
|
screen_unload_texture(qrInstallData->tex);
|
|
qrInstallData->tex = 0;
|
|
}
|
|
|
|
int w = 0;
|
|
int h = 0;
|
|
uint8_t* qrBuf = quirc_begin(qrInstallData->qrContext, &w, &h);
|
|
|
|
svcWaitSynchronization(qrInstallData->mutex, U64_MAX);
|
|
|
|
qrInstallData->tex = screen_load_texture_auto(qrInstallData->buffer, IMAGE_WIDTH * IMAGE_HEIGHT * sizeof(u16), IMAGE_WIDTH, IMAGE_HEIGHT, GPU_RGB565, false);
|
|
|
|
for(int x = 0; x < w; x++) {
|
|
for(int y = 0; y < h; y++) {
|
|
u16 px = qrInstallData->buffer[y * IMAGE_WIDTH + x];
|
|
qrBuf[y * w + x] = (u8) (((((px >> 11) & 0x1F) << 3) + (((px >> 5) & 0x3F) << 2) + ((px & 0x1F) << 3)) / 3);
|
|
}
|
|
}
|
|
|
|
svcReleaseMutex(qrInstallData->mutex);
|
|
|
|
quirc_end(qrInstallData->qrContext);
|
|
|
|
int qrCount = quirc_count(qrInstallData->qrContext);
|
|
for(int i = 0; i < qrCount; i++) {
|
|
struct quirc_code qrCode;
|
|
quirc_extract(qrInstallData->qrContext, i, &qrCode);
|
|
|
|
struct quirc_data qrData;
|
|
quirc_decode_error_t err = quirc_decode(&qrCode, &qrData);
|
|
|
|
if(err == 0) {
|
|
qrInstallData->installInfo.total = 0;
|
|
|
|
char* currStart = (char*) qrData.payload;
|
|
char* currEnd = NULL;
|
|
while((currEnd = strchr(currStart, '\n')) != NULL) {
|
|
u32 len = currEnd - currStart;
|
|
if(len > URL_MAX) {
|
|
len = URL_MAX;
|
|
}
|
|
|
|
strncpy(qrInstallData->urls[qrInstallData->installInfo.total++], currStart, len);
|
|
|
|
currStart = currEnd + 1;
|
|
}
|
|
|
|
if(*currStart != '\0') {
|
|
strncpy(qrInstallData->urls[qrInstallData->installInfo.total++], currStart, URL_MAX);
|
|
}
|
|
|
|
prompt_display("Confirmation", "Install from the scanned URL(s)?", COLOR_TEXT, true, data, NULL, NULL, qrinstall_confirm_onresponse);
|
|
}
|
|
}
|
|
|
|
snprintf(text, PROGRESS_TEXT_MAX, "Waiting for QR code...");
|
|
}
|
|
|
|
void qrinstall_open() {
|
|
qr_install_data* data = (qr_install_data*) calloc(1, sizeof(qr_install_data));
|
|
if(data == NULL) {
|
|
error_display(NULL, NULL, NULL, "Failed to allocate QR install data.");
|
|
|
|
return;
|
|
}
|
|
|
|
data->qrContext = quirc_new();
|
|
if(data->qrContext == NULL) {
|
|
error_display(NULL, NULL, NULL, "Failed to create QR context.");
|
|
|
|
qrinstall_free_data(data);
|
|
return;
|
|
}
|
|
|
|
if(quirc_resize(data->qrContext, IMAGE_WIDTH, IMAGE_HEIGHT) != 0) {
|
|
error_display(NULL, NULL, NULL, "Failed to resize QR context.");
|
|
|
|
qrinstall_free_data(data);
|
|
return;
|
|
}
|
|
|
|
data->buffer = (u16*) calloc(1, IMAGE_WIDTH * IMAGE_HEIGHT * sizeof(u16));
|
|
if(data->buffer == NULL) {
|
|
error_display(NULL, NULL, NULL, "Failed to create image buffer.");
|
|
|
|
qrinstall_free_data(data);
|
|
return;
|
|
}
|
|
|
|
data->camCancelEvent = task_capture_cam(&data->mutex, data->buffer, IMAGE_WIDTH, IMAGE_HEIGHT);
|
|
if(data->camCancelEvent == 0) {
|
|
error_display(NULL, NULL, NULL, "Failed to start camera capture.");
|
|
|
|
qrinstall_free_data(data);
|
|
return;
|
|
}
|
|
|
|
data->tex = 0;
|
|
|
|
data->responseCode = 0;
|
|
data->currTitleId = 0;
|
|
data->ticket = false;
|
|
|
|
data->installInfo.data = data;
|
|
|
|
data->installInfo.op = DATAOP_COPY;
|
|
|
|
data->installInfo.copyEmpty = false;
|
|
|
|
data->installInfo.total = 0;
|
|
|
|
data->installInfo.isSrcDirectory = qrinstall_is_src_directory;
|
|
data->installInfo.makeDstDirectory = qrinstall_make_dst_directory;
|
|
|
|
data->installInfo.openSrc = qrinstall_open_src;
|
|
data->installInfo.closeSrc = qrinstall_close_src;
|
|
data->installInfo.getSrcSize = qrinstall_get_src_size;
|
|
data->installInfo.readSrc = qrinstall_read_src;
|
|
|
|
data->installInfo.openDst = qrinstall_open_dst;
|
|
data->installInfo.closeDst = qrinstall_close_dst;
|
|
data->installInfo.writeDst = qrinstall_write_dst;
|
|
|
|
data->installInfo.error = qrinstall_error;
|
|
|
|
data->installCancelEvent = 0;
|
|
|
|
info_display("QR Code Install", "B: Return", false, data, qrinstall_wait_update, qrinstall_wait_draw_top);
|
|
}
|