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.
346 lines
11 KiB
C
346 lines
11 KiB
C
#include <arpa/inet.h>
|
|
#include <sys/unistd.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <malloc.h>
|
|
#include <stdio.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"
|
|
|
|
typedef struct {
|
|
int serverSocket;
|
|
int clientSocket;
|
|
|
|
u64 currTitleId;
|
|
bool ticket;
|
|
|
|
data_op_info installInfo;
|
|
Handle cancelEvent;
|
|
} network_install_data;
|
|
|
|
static int recvwait(int sockfd, void* buf, size_t len, int flags) {
|
|
errno = 0;
|
|
|
|
int ret = 0;
|
|
size_t read = 0;
|
|
while(((ret = recv(sockfd, buf + read, len - read, flags)) >= 0 && (read += ret) < len) || errno == EAGAIN) {
|
|
errno = 0;
|
|
}
|
|
|
|
return ret < 0 ? ret : (int) read;
|
|
}
|
|
|
|
static int sendwait(int sockfd, void* buf, size_t len, int flags) {
|
|
errno = 0;
|
|
|
|
int ret = 0;
|
|
size_t written = 0;
|
|
while(((ret = send(sockfd, buf + written, len - written, flags)) >= 0 && (written += ret) < len) || errno == EAGAIN) {
|
|
errno = 0;
|
|
}
|
|
|
|
return ret < 0 ? ret : (int) written;
|
|
}
|
|
|
|
static Result networkinstall_is_src_directory(void* data, u32 index, bool* isDirectory) {
|
|
*isDirectory = false;
|
|
return 0;
|
|
}
|
|
|
|
static Result networkinstall_make_dst_directory(void* data, u32 index) {
|
|
return 0;
|
|
}
|
|
|
|
static Result networkinstall_open_src(void* data, u32 index, u32* handle) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
u8 ack = 1;
|
|
if(sendwait(networkInstallData->clientSocket, &ack, sizeof(ack), 0) < 0) {
|
|
return R_FBI_ERRNO;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static Result networkinstall_close_src(void* data, u32 index, bool succeeded, u32 handle) {
|
|
return 0;
|
|
}
|
|
|
|
static Result networkinstall_get_src_size(void* data, u32 handle, u64* size) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
u64 netSize = 0;
|
|
if(recvwait(networkInstallData->clientSocket, &netSize, sizeof(netSize), 0) < 0) {
|
|
return R_FBI_ERRNO;
|
|
}
|
|
|
|
*size = __builtin_bswap64(netSize);
|
|
return 0;
|
|
}
|
|
|
|
static Result networkinstall_read_src(void* data, u32 handle, u32* bytesRead, void* buffer, u64 offset, u32 size) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
int ret = 0;
|
|
if((ret = recvwait(networkInstallData->clientSocket, buffer, size, 0)) < 0) {
|
|
return R_FBI_ERRNO;
|
|
}
|
|
|
|
*bytesRead = (u32) ret;
|
|
return 0;
|
|
}
|
|
|
|
static Result networkinstall_open_dst(void* data, u32 index, void* initialReadBlock, u32* handle) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
networkInstallData->ticket = *(u16*) initialReadBlock == 0x0100;
|
|
|
|
Result res = 0;
|
|
|
|
if(networkInstallData->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))) {
|
|
networkInstallData->currTitleId = titleId;
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
static Result networkinstall_close_dst(void* data, u32 index, bool succeeded, u32 handle) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
if(succeeded) {
|
|
Result res = 0;
|
|
|
|
if(networkInstallData->ticket) {
|
|
res = AM_InstallTicketFinish(handle);
|
|
} else {
|
|
if(R_SUCCEEDED(res = AM_FinishCiaInstall(handle))) {
|
|
if(networkInstallData->currTitleId == 0x0004013800000002 || networkInstallData->currTitleId == 0x0004013820000002) {
|
|
res = AM_InstallFirm(networkInstallData->currTitleId);
|
|
}
|
|
}
|
|
}
|
|
|
|
return res;
|
|
} else {
|
|
if(networkInstallData->ticket) {
|
|
return AM_InstallTicketAbort(handle);
|
|
} else {
|
|
return AM_CancelCIAInstall(handle);
|
|
}
|
|
}
|
|
}
|
|
|
|
static Result networkinstall_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 networkinstall_error(void* data, u32 index, Result res) {
|
|
if(res == R_FBI_CANCELLED) {
|
|
prompt_display("Failure", "Install cancelled.", COLOR_TEXT, false, NULL, NULL, NULL, NULL);
|
|
} else if(res == R_FBI_ERRNO) {
|
|
error_display_errno(NULL, NULL, NULL, errno, "Failed to install over the network.");
|
|
} else if(res == R_FBI_WRONG_SYSTEM) {
|
|
error_display(NULL, NULL, NULL, "Failed to install over the network.\nAttempted to install N3DS title to O3DS.");
|
|
} else {
|
|
error_display_res(NULL, NULL, NULL, res, "Failed to install over the network.");
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
static void networkinstall_close_client(network_install_data* data) {
|
|
u8 ack = 0;
|
|
sendwait(data->clientSocket, &ack, sizeof(ack), 0);
|
|
|
|
close(data->clientSocket);
|
|
|
|
data->currTitleId = 0;
|
|
data->cancelEvent = 0;
|
|
}
|
|
|
|
static void networkinstall_install_update(ui_view* view, void* data, float* progress, char* text) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
if(networkInstallData->installInfo.finished) {
|
|
networkinstall_close_client(networkInstallData);
|
|
|
|
ui_pop();
|
|
info_destroy(view);
|
|
|
|
if(!networkInstallData->installInfo.premature) {
|
|
prompt_display("Success", "Install finished.", COLOR_TEXT, false, data, NULL, NULL, NULL);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
if(hidKeysDown() & KEY_B) {
|
|
svcSignalEvent(networkInstallData->cancelEvent);
|
|
}
|
|
|
|
*progress = networkInstallData->installInfo.currTotal != 0 ? (float) ((double) networkInstallData->installInfo.currProcessed / (double) networkInstallData->installInfo.currTotal) : 0;
|
|
snprintf(text, PROGRESS_TEXT_MAX, "%lu / %lu\n%.2f MB / %.2f MB", networkInstallData->installInfo.processed, networkInstallData->installInfo.total, networkInstallData->installInfo.currProcessed / 1024.0 / 1024.0, networkInstallData->installInfo.currTotal / 1024.0 / 1024.0);
|
|
}
|
|
|
|
static void networkinstall_confirm_onresponse(ui_view* view, void* data, bool response) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
if(response) {
|
|
networkInstallData->cancelEvent = task_data_op(&networkInstallData->installInfo);
|
|
if(networkInstallData->cancelEvent != 0) {
|
|
info_display("Installing Over Network", "Press B to cancel.", true, data, networkinstall_install_update, NULL);
|
|
} else {
|
|
error_display(NULL, NULL, NULL, "Failed to initiate installation.");
|
|
|
|
networkinstall_close_client(networkInstallData);
|
|
}
|
|
} else {
|
|
networkinstall_close_client(networkInstallData);
|
|
}
|
|
}
|
|
|
|
static void networkinstall_wait_update(ui_view* view, void* data, float* progress, char* text) {
|
|
network_install_data* networkInstallData = (network_install_data*) data;
|
|
|
|
if(hidKeysDown() & KEY_B) {
|
|
ui_pop();
|
|
info_destroy(view);
|
|
|
|
close(networkInstallData->serverSocket);
|
|
free(networkInstallData);
|
|
|
|
return;
|
|
}
|
|
|
|
struct sockaddr_in client;
|
|
socklen_t clientLen = sizeof(client);
|
|
|
|
int sock = accept(networkInstallData->serverSocket, (struct sockaddr*) &client, &clientLen);
|
|
if(sock >= 0) {
|
|
if(recvwait(sock, &networkInstallData->installInfo.total, sizeof(networkInstallData->installInfo.total), 0) < 0) {
|
|
close(sock);
|
|
|
|
error_display_errno(NULL, NULL, NULL, errno, "Failed to read file count.");
|
|
return;
|
|
}
|
|
|
|
networkInstallData->installInfo.total = ntohl(networkInstallData->installInfo.total);
|
|
|
|
networkInstallData->clientSocket = sock;
|
|
prompt_display("Confirmation", "Install the received file(s)?", COLOR_TEXT, true, data, NULL, NULL, networkinstall_confirm_onresponse);
|
|
} else if(errno != EAGAIN) {
|
|
error_display_errno(NULL, NULL, NULL, errno, "Failed to open socket.");
|
|
}
|
|
|
|
struct in_addr addr = {(in_addr_t) gethostid()};
|
|
snprintf(text, PROGRESS_TEXT_MAX, "Waiting for connection...\nIP: %s\nPort: 5000", inet_ntoa(addr));
|
|
}
|
|
|
|
void networkinstall_open() {
|
|
network_install_data* data = (network_install_data*) calloc(1, sizeof(network_install_data));
|
|
if(data == NULL) {
|
|
error_display(NULL, NULL, NULL, "Failed to allocate network install data.");
|
|
|
|
return;
|
|
}
|
|
|
|
int sock = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
|
|
if(sock < 0) {
|
|
error_display_errno(NULL, NULL, NULL, errno, "Failed to open server socket.");
|
|
|
|
free(data);
|
|
return;
|
|
}
|
|
|
|
int bufSize = 1024 * 32;
|
|
setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &bufSize, sizeof(bufSize));
|
|
|
|
struct sockaddr_in server;
|
|
server.sin_family = AF_INET;
|
|
server.sin_port = htons(5000);
|
|
server.sin_addr.s_addr = (in_addr_t) gethostid();
|
|
|
|
if(bind(sock, (struct sockaddr*) &server, sizeof(server)) < 0) {
|
|
error_display_errno(NULL, NULL, NULL, errno, "Failed to bind server socket.");
|
|
|
|
close(sock);
|
|
free(data);
|
|
return;
|
|
}
|
|
|
|
fcntl(sock, F_SETFL, fcntl(sock, F_GETFL, 0) | O_NONBLOCK);
|
|
|
|
if(listen(sock, 5) < 0) {
|
|
error_display_errno(NULL, NULL, NULL, errno, "Failed to listen on server socket.");
|
|
|
|
close(sock);
|
|
free(data);
|
|
return;
|
|
}
|
|
|
|
data->serverSocket = sock;
|
|
data->clientSocket = 0;
|
|
|
|
data->currTitleId = 0;
|
|
data->ticket = false;
|
|
|
|
data->installInfo.data = data;
|
|
|
|
data->installInfo.op = DATAOP_COPY;
|
|
|
|
data->installInfo.copyEmpty = false;
|
|
|
|
data->installInfo.isSrcDirectory = networkinstall_is_src_directory;
|
|
data->installInfo.makeDstDirectory = networkinstall_make_dst_directory;
|
|
|
|
data->installInfo.openSrc = networkinstall_open_src;
|
|
data->installInfo.closeSrc = networkinstall_close_src;
|
|
data->installInfo.getSrcSize = networkinstall_get_src_size;
|
|
data->installInfo.readSrc = networkinstall_read_src;
|
|
|
|
data->installInfo.openDst = networkinstall_open_dst;
|
|
data->installInfo.closeDst = networkinstall_close_dst;
|
|
data->installInfo.writeDst = networkinstall_write_dst;
|
|
|
|
data->installInfo.error = networkinstall_error;
|
|
|
|
data->cancelEvent = 0;
|
|
|
|
info_display("Network Install", "B: Return", false, data, networkinstall_wait_update, NULL);
|
|
}
|