mirror of
https://github.com/suchmememanyskill/TegraExplorer.git
synced 2024-11-26 22:02:08 +00:00
603 lines
No EOL
12 KiB
C
603 lines
No EOL
12 KiB
C
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include "../../mem/heap.h"
|
|
#include "../gfx/gfxutils.h"
|
|
#include "../emmc/emmc.h"
|
|
#include "../../utils/types.h"
|
|
#include "../../libs/fatfs/ff.h"
|
|
#include "../../utils/sprintf.h"
|
|
#include "../../hid/hid.h"
|
|
#include "../../gfx/gfx.h"
|
|
#include "../../utils/util.h"
|
|
#include "../../storage/emummc.h"
|
|
#include "parser.h"
|
|
#include "../common/common.h"
|
|
#include "../fs/fsactions.h"
|
|
#include "variables.h"
|
|
#include "../utils/utils.h"
|
|
#include "functions.h"
|
|
#include "../fs/fsutils.h"
|
|
#include "../../utils/sprintf.h"
|
|
#include "../fs/fsactions.h"
|
|
#include "../emmc/emmcoperations.h"
|
|
#include "../emmc/emmcmenu.h"
|
|
|
|
extern FIL scriptin;
|
|
extern char **argv;
|
|
extern u32 argc;
|
|
extern int forceExit;
|
|
extern short currentlyMounted;
|
|
|
|
int parseIntInput(char *in, int *out){
|
|
if (in[0] == '@'){
|
|
if (str_int_find(in, out))
|
|
return -1;
|
|
}
|
|
else
|
|
*out = atoi(in);
|
|
|
|
return 0;
|
|
}
|
|
/*
|
|
int parseJmpInput(char *in, u64 *out){
|
|
if (in[0] == '?'){
|
|
if (str_jmp_find(in, out))
|
|
return -1;
|
|
else
|
|
return 0;
|
|
}
|
|
else
|
|
return -1;
|
|
}
|
|
*/
|
|
|
|
int parseStringInput(char *in, char **out){
|
|
if (in[0] == '$'){
|
|
if (str_str_find(in, out))
|
|
return -1;
|
|
else
|
|
return 0;
|
|
}
|
|
else{
|
|
*out = in;
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
#pragma GCC push_options
|
|
#pragma GCC optimize ("Os")
|
|
|
|
u32 currentcolor = COLOR_WHITE;
|
|
int part_printf(){
|
|
SWAPCOLOR(currentcolor);
|
|
for (int i = 0; i < argc; i++){
|
|
if (argv[i][0] == '@'){
|
|
int toprintint;
|
|
if (parseIntInput(argv[i], &toprintint))
|
|
return -1;
|
|
|
|
gfx_printf("%d", toprintint);
|
|
}
|
|
else {
|
|
char *toprintstring;
|
|
if (parseStringInput(argv[i], &toprintstring))
|
|
return -1;
|
|
|
|
gfx_printf(toprintstring);
|
|
}
|
|
}
|
|
|
|
gfx_printf("\n");
|
|
return 0;
|
|
}
|
|
|
|
int part_print_int(){
|
|
int toprint;
|
|
if (parseIntInput(argv[0], &toprint))
|
|
return -1;
|
|
|
|
SWAPCOLOR(currentcolor);
|
|
gfx_printf("%s: %d\n", argv[0], toprint);
|
|
return 0;
|
|
}
|
|
|
|
int part_Wait(){
|
|
int arg;
|
|
u32 begintime;
|
|
SWAPCOLOR(currentcolor);
|
|
|
|
if (parseIntInput(argv[0], &arg))
|
|
return -1;
|
|
|
|
begintime = get_tmr_s();
|
|
|
|
while (begintime + arg > get_tmr_s()){
|
|
gfx_printf("\r<Wait %d seconds> ", (begintime + arg) - get_tmr_s());
|
|
}
|
|
|
|
gfx_printf("\r \r");
|
|
return 0;
|
|
}
|
|
|
|
int part_Check(){
|
|
int left, right;
|
|
if (parseIntInput(argv[0], &left))
|
|
return -1;
|
|
if (parseIntInput(argv[2], &right))
|
|
return -1;
|
|
|
|
if (!strcmp(argv[1], "=="))
|
|
return (left == right);
|
|
else if (!strcmp(argv[1], "!="))
|
|
return (left != right);
|
|
else if (!strcmp(argv[1], ">="))
|
|
return (left >= right);
|
|
else if (!strcmp(argv[1], "<="))
|
|
return (left <= right);
|
|
else if (!strcmp(argv[1], ">"))
|
|
return (left > right);
|
|
else if (!strcmp(argv[1], "<"))
|
|
return (left < right);
|
|
else
|
|
return -1;
|
|
}
|
|
|
|
int part_if(){
|
|
int condition;
|
|
if (parseIntInput(argv[0], &condition))
|
|
return -1;
|
|
|
|
getfollowingchar('{');
|
|
|
|
if (!condition)
|
|
skipbrackets();
|
|
|
|
return 0;
|
|
|
|
/*
|
|
if (condition)
|
|
return 0;
|
|
else {
|
|
skipbrackets();
|
|
return 0;
|
|
}
|
|
*/
|
|
}
|
|
|
|
int part_if_args(){
|
|
int condition;
|
|
if ((condition = part_Check()) < 0)
|
|
return -1;
|
|
|
|
getfollowingchar('{');
|
|
|
|
if (!condition)
|
|
skipbrackets();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int part_Math(){
|
|
int left, right;
|
|
if (parseIntInput(argv[0], &left))
|
|
return -1;
|
|
if (parseIntInput(argv[2], &right))
|
|
return -1;
|
|
|
|
switch (argv[1][0]){
|
|
case '+':
|
|
return left + right;
|
|
case '-':
|
|
return left - right;
|
|
case '*':
|
|
return left * right;
|
|
case '/':
|
|
return left / right;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int part_SetInt(){
|
|
int out;
|
|
parseIntInput(argv[0], &out);
|
|
return out;
|
|
}
|
|
|
|
int part_SetString(){
|
|
char *arg0;
|
|
if (parseStringInput(argv[0], &arg0))
|
|
return -1;
|
|
if (argv[1][0] != '$')
|
|
return -1;
|
|
|
|
str_str_add(argv[1], arg0);
|
|
return 0;
|
|
}
|
|
|
|
int part_SetStringIndex(){
|
|
int index;
|
|
char *out;
|
|
if (parseIntInput(argv[0], &index))
|
|
return -1;
|
|
if (argv[1][0] != '$')
|
|
return -1;
|
|
if (str_str_index(index, &out))
|
|
return -1;
|
|
|
|
str_str_add(argv[1], out);
|
|
return 0;
|
|
}
|
|
|
|
int part_goto(){
|
|
int target = 0;
|
|
if (parseIntInput(argv[0], &target))
|
|
return -1;
|
|
f_lseek(&scriptin, target);
|
|
return 0;
|
|
}
|
|
|
|
int part_invert(){
|
|
int arg;
|
|
if (parseIntInput(argv[0], &arg))
|
|
return -1;
|
|
return (arg) ? 0 : 1;
|
|
}
|
|
|
|
int part_fs_exists(){
|
|
char *path;
|
|
if (parseStringInput(argv[0], &path))
|
|
return -1;
|
|
|
|
return fsutil_checkfile(path);
|
|
}
|
|
|
|
int part_ConnectMMC(){
|
|
char *arg;
|
|
parseStringInput(argv[0], &arg);
|
|
|
|
if (!strcmp(arg, "SYSMMC"))
|
|
connect_mmc(SYSMMC);
|
|
else if (!strcmp(arg, "EMUMMC"))
|
|
connect_mmc(EMUMMC);
|
|
else
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int part_MountMMC(){
|
|
char *arg;
|
|
parseStringInput(argv[0], &arg);
|
|
return mount_mmc(arg, 2);
|
|
}
|
|
|
|
int part_Pause(){
|
|
Inputs *input = hidWaitMask(KEY_A | KEY_B | KEY_X | KEY_Y | KEY_POW | KEY_VOLP | KEY_VOLM | KEY_LUP | KEY_LDOWN | KEY_LLEFT | KEY_LRIGHT);
|
|
|
|
str_int_add("@BTN_POWER", input->pow);
|
|
str_int_add("@BTN_VOL+", input->volp);
|
|
str_int_add("@BTN_VOL-", input->volm);
|
|
str_int_add("@BTN_A", input->a);
|
|
str_int_add("@BTN_B", input->b);
|
|
str_int_add("@BTN_X", input->x);
|
|
str_int_add("@BTN_Y", input->y);
|
|
str_int_add("@BTN_UP", input->Lup);
|
|
str_int_add("@BTN_DOWN", input->Ldown);
|
|
str_int_add("@BTN_LEFT", input->Lleft);
|
|
str_int_add("@BTN_RIGHT", input->Lright);
|
|
|
|
return input->buttons;
|
|
}
|
|
|
|
int part_addstrings(){
|
|
char *combined, *left, *middle;
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &middle))
|
|
return -1;
|
|
if (argv[2][0] != '$')
|
|
return -1;
|
|
|
|
combined = calloc(strlen(left) + strlen(middle) + 1, sizeof(char));
|
|
sprintf(combined, "%s%s", left, middle);
|
|
|
|
str_str_add(argv[2], combined);
|
|
free(combined);
|
|
return 0;
|
|
}
|
|
|
|
int part_setColor(){
|
|
char *arg;
|
|
if (parseStringInput(argv[0], &arg))
|
|
return -1;
|
|
|
|
if (!strcmp(arg, "RED"))
|
|
currentcolor = COLOR_RED;
|
|
else if (!strcmp(arg, "ORANGE"))
|
|
currentcolor = COLOR_ORANGE;
|
|
else if (!strcmp(arg, "YELLOW"))
|
|
currentcolor = COLOR_YELLOW;
|
|
else if (!strcmp(arg, "GREEN"))
|
|
currentcolor = COLOR_GREEN;
|
|
else if (!strcmp(arg, "BLUE"))
|
|
currentcolor = COLOR_BLUE;
|
|
else if (!strcmp(arg, "VIOLET"))
|
|
currentcolor = COLOR_VIOLET;
|
|
else if (!strcmp(arg, "WHITE"))
|
|
currentcolor = COLOR_WHITE;
|
|
else
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int part_Exit(){
|
|
forceExit = true;
|
|
return 0;
|
|
}
|
|
|
|
int part_fs_Move(){
|
|
char *left, *right;
|
|
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &right))
|
|
return -1;
|
|
|
|
int res;
|
|
res = f_rename(left, right);
|
|
if (res)
|
|
res = f_rename(left, right);
|
|
|
|
return res;
|
|
}
|
|
|
|
int part_fs_Delete(){
|
|
char *arg;
|
|
|
|
if (parseStringInput(argv[0], &arg))
|
|
return -1;
|
|
|
|
int res;
|
|
res = f_unlink(arg);
|
|
if (res)
|
|
res = f_unlink(arg);
|
|
|
|
return res;
|
|
}
|
|
|
|
int part_fs_DeleteRecursive(){
|
|
char *arg;
|
|
|
|
if (parseStringInput(argv[0], &arg))
|
|
return -1;
|
|
|
|
return fsact_del_recursive(arg);
|
|
}
|
|
|
|
int part_fs_Copy(){
|
|
char *left, *right;
|
|
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &right))
|
|
return -1;
|
|
|
|
return fsact_copy(left, right, COPY_MODE_PRINT);
|
|
}
|
|
|
|
int part_fs_CopyRecursive(){
|
|
char *left, *right;
|
|
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &right))
|
|
return -1;
|
|
|
|
return fsact_copy_recursive(left, right);
|
|
}
|
|
|
|
int part_fs_MakeDir(){
|
|
char *arg;
|
|
|
|
if (parseStringInput(argv[0], &arg))
|
|
return -1;
|
|
|
|
int res;
|
|
res = f_mkdir(arg);
|
|
if (res)
|
|
res = f_mkdir(arg);
|
|
|
|
return res;
|
|
}
|
|
|
|
DIR dir;
|
|
FILINFO fno;
|
|
int isdirvalid = false;
|
|
int part_fs_OpenDir(){
|
|
char *path;
|
|
|
|
if (parseStringInput(argv[0], &path))
|
|
return -1;
|
|
|
|
if (f_opendir(&dir, path))
|
|
return -1;
|
|
|
|
isdirvalid = true;
|
|
str_int_add("@ISDIRVALID", isdirvalid);
|
|
return 0;
|
|
}
|
|
|
|
int part_fs_CloseDir(){
|
|
if (!isdirvalid)
|
|
return 0;
|
|
|
|
f_closedir(&dir);
|
|
isdirvalid = false;
|
|
str_int_add("@ISDIRVALID", isdirvalid);
|
|
return 0;
|
|
}
|
|
|
|
int part_fs_ReadDir(){
|
|
if (!isdirvalid)
|
|
return -1;
|
|
|
|
if (!f_readdir(&dir, &fno) && fno.fname[0]){
|
|
str_str_add("$FILENAME", fno.fname);
|
|
str_int_add("@ISDIR", (fno.fattrib & AM_DIR) ? 1 : 0);
|
|
}
|
|
else {
|
|
part_fs_CloseDir();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int part_setPrintPos(){
|
|
int left, right;
|
|
|
|
if (parseIntInput(argv[0], &left))
|
|
return -1;
|
|
|
|
if (parseIntInput(argv[1], &right))
|
|
return -1;
|
|
|
|
if (left > 78)
|
|
return -1;
|
|
|
|
if (right > 42)
|
|
return -1;
|
|
|
|
gfx_con_setpos(left * 16, right * 16);
|
|
return 0;
|
|
}
|
|
|
|
int part_stringcompare(){
|
|
char *left, *right;
|
|
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &right))
|
|
return -1;
|
|
|
|
return (strcmp(left, right)) ? 0 : 1;
|
|
}
|
|
|
|
int part_fs_combinePath(){
|
|
char *combined, *left, *middle;
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &middle))
|
|
return -1;
|
|
if (argv[2][0] != '$')
|
|
return -1;
|
|
|
|
combined = fsutil_getnextloc(left, middle);
|
|
|
|
str_str_add(argv[2], combined);
|
|
free(combined);
|
|
return 0;
|
|
}
|
|
|
|
int part_mmc_dumpPart(){
|
|
char *left, *right;
|
|
|
|
if (parseStringInput(argv[0], &left))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &right))
|
|
return -1;
|
|
|
|
if (!strcmp(left, "BOOT")){
|
|
return emmcDumpBoot(right);
|
|
}
|
|
else {
|
|
return emmcDumpSpecific(left, right);
|
|
}
|
|
}
|
|
|
|
int part_mmc_restorePart(){
|
|
char *path;
|
|
|
|
if (parseStringInput(argv[0], &path))
|
|
return -1;
|
|
|
|
if (currentlyMounted < 0)
|
|
return -1;
|
|
|
|
return mmcFlashFile(path, currentlyMounted, false);
|
|
}
|
|
|
|
int part_fs_extractBisFile(){
|
|
char *path, *outfolder;
|
|
|
|
if (parseStringInput(argv[0], &path))
|
|
return -1;
|
|
if (parseStringInput(argv[1], &outfolder))
|
|
return -1;
|
|
|
|
return extract_bis_file(path, outfolder);
|
|
}
|
|
|
|
int part_clearscreen(){
|
|
gfx_clearscreen();
|
|
return 0;
|
|
}
|
|
|
|
int part_getPos(){
|
|
return (int)f_tell(&scriptin);
|
|
}
|
|
|
|
str_fnc_struct functions[] = {
|
|
{"printf", part_printf, 255},
|
|
{"printInt", part_print_int, 1},
|
|
{"setPrintPos", part_setPrintPos, 2},
|
|
{"clearscreen", part_clearscreen, 0},
|
|
{"if", part_if, 1},
|
|
{"if", part_if_args, 3}, // function overloading
|
|
{"math", part_Math, 3},
|
|
{"check", part_Check, 3},
|
|
{"setInt", part_SetInt, 1},
|
|
{"goto", part_goto, 1},
|
|
{"setString", part_SetString, 2},
|
|
{"setStringIndex", part_SetStringIndex, 2},
|
|
{"setColor", part_setColor, 1},
|
|
{"combineStrings", part_addstrings, 3},
|
|
{"compareStrings", part_stringcompare, 2},
|
|
{"invert", part_invert, 1},
|
|
{"fs_exists", part_fs_exists, 1},
|
|
{"fs_move", part_fs_Move, 2},
|
|
{"fs_mkdir", part_fs_MakeDir, 1},
|
|
{"fs_del", part_fs_Delete, 1},
|
|
{"fs_delRecursive", part_fs_DeleteRecursive, 1},
|
|
{"fs_copy", part_fs_Copy, 2},
|
|
{"fs_copyRecursive", part_fs_CopyRecursive, 2},
|
|
{"fs_openDir", part_fs_OpenDir, 1},
|
|
{"fs_closeDir", part_fs_CloseDir, 0},
|
|
{"fs_readDir", part_fs_ReadDir, 0},
|
|
{"fs_combinePath", part_fs_combinePath, 3},
|
|
{"fs_extractBisFile", part_fs_extractBisFile, 2},
|
|
{"mmc_connect", part_ConnectMMC, 1},
|
|
{"mmc_mount", part_MountMMC, 1},
|
|
{"mmc_dumpPart", part_mmc_dumpPart, 2},
|
|
{"mmc_restorePart", part_mmc_restorePart, 1},
|
|
{"getPosition", part_getPos, 0},
|
|
{"pause", part_Pause, 0},
|
|
{"wait", part_Wait, 1},
|
|
{"exit", part_Exit, 0},
|
|
{NULL, NULL, 0}
|
|
};
|
|
|
|
#pragma GCC pop_options
|
|
|
|
int run_function(char *func_name, int *out){
|
|
for (u32 i = 0; functions[i].key != NULL; i++){
|
|
if (!strcmp(functions[i].key, func_name)){
|
|
if (argc != functions[i].arg_count && functions[i].arg_count != 255)
|
|
continue;
|
|
|
|
*out = functions[i].value();
|
|
return (*out < 0) ? -1 : 0;
|
|
}
|
|
}
|
|
return -1;
|
|
} |