GodMode9i/arm9/source/hexEditor.cpp
Pk11 219a0a4325 Fix tiny files and add searching
Also make the formatting of 'if's and 'for's consistent
2021-04-23 01:35:35 -05:00

403 lines
11 KiB
C++

#include "hexEditor.h"
#include "date.h"
#include "tonccpy.h"
#include "file_browse.h"
#include <algorithm>
#include <nds.h>
#include <stdio.h>
extern PrintConsole bottomConsole, bottomConsoleBG, topConsole;
extern void reinitConsoles(void);
u32 jumpToOffset(u32 offset) {
consoleSelect(&bottomConsoleBG);
consoleClear();
consoleSelect(&bottomConsole);
consoleClear();
u8 cursorPosition = 0;
u16 pressed = 0, held = 0;
while(1) {
printf("\x1B[9;6H\x1B[47m-------------------");
printf("\x1B[10;8H\x1B[47mJump to Offset");
printf("\x1B[12;11H\x1B[37m%08lX", offset);
printf("\x1B[12;%dH\x1B[41m%lX", 17 - cursorPosition, (offset >> ((cursorPosition + 1) * 4)) & 0xF);
printf("\x1B[13;6H\x1B[47m-------------------");
consoleSelect(&topConsole);
do {
swiWaitForVBlank();
scanKeys();
pressed = keysDown();
held = keysDownRepeat();
printf("\x1B[30m\x1B[0;26H %s", RetTime().c_str()); // Print time
} while(!held);
consoleSelect(&bottomConsole);
if(held & KEY_UP) {
offset = (offset & ~(0xF0 << cursorPosition * 4)) | ((offset + (0x10 << (cursorPosition * 4))) & (0xF0 << cursorPosition * 4));
} else if(held & KEY_DOWN) {
offset = (offset & ~(0xF0 << cursorPosition * 4)) | ((offset - (0x10 << (cursorPosition * 4))) & (0xF0 << cursorPosition * 4));
} else if(held & KEY_LEFT) {
if(cursorPosition < 6)
cursorPosition++;
} else if(held & KEY_RIGHT) {
if(cursorPosition > 0)
cursorPosition--;
} else if(pressed & (KEY_A | KEY_B)) {
return offset;
}
}
}
u32 search(u32 offset, FILE *file) {
consoleSelect(&bottomConsoleBG);
consoleClear();
consoleSelect(&bottomConsole);
consoleClear();
u8 cursorPosition = 0;
u16 pressed = 0, held = 0;
while(1) {
printf("\x1B[9;4H\x1B[47m-----------------------");
printf("\x1B[10;5H%c Search for String %c", cursorPosition == 0 ? '>' : ' ', cursorPosition == 0 ? '<' : ' ');
printf("\x1B[11;5H%c Search for Data %c", cursorPosition == 1 ? '>' : ' ', cursorPosition == 1 ? '<' : ' ');
printf("\x1B[12;4H-----------------------");
consoleSelect(&topConsole);
do {
swiWaitForVBlank();
scanKeys();
pressed = keysDown();
held = keysDownRepeat();
printf("\x1B[30m\x1B[0;26H %s", RetTime().c_str()); // Print time
} while(!held);
consoleSelect(&bottomConsole);
if(held & (KEY_UP | KEY_DOWN)) {
cursorPosition ^= 1;
} else if(pressed & KEY_A) {
break;
} else if(pressed & KEY_B) {
return offset;
}
}
char str[64] = {0};
size_t strLen = 1;
if(cursorPosition == 0) {
consoleDemoInit();
Keyboard *kbd = keyboardDemoInit();
kbd->OnKeyPressed = OnKeyPressed;
// keyboardShow();
printf("Search for:\n");
fgets(str, sizeof(str), stdin);
keyboardHide();
consoleClear();
reinitConsoles();
consoleSelect(&bottomConsole);
BG_PALETTE_SUB[0x1F] = 0x9CF7;
BG_PALETTE_SUB[0x2F] = 0xB710;
BG_PALETTE_SUB[0x3F] = 0xAE8D;
BG_PALETTE_SUB[0x7F] = 0xEA2D;
strLen = strlen(str) - 1;
if(strLen == 0)
return offset;
str[strLen] = 0; // Remove ending \n that fgets has
} else {
consoleClear();
cursorPosition = 0;
while(1) {
printf("\x1B[9;6H\x1B[47m------------------");
printf("\x1B[10;9HEnter value:");
u8 pos = 15 - strLen;
for(size_t i = 0; i < strLen * 2; i++) {
printf("\x1B[12;%dH\x1B[%dm%X", pos + i, (i == cursorPosition ? 31 : ((i / 2 % 2) ? 33 : 32)), str[i / 2] >> (!(i % 2) * 4) & 0xF);
}
printf("\x1B[13;6H\x1B[47m------------------");
consoleSelect(&topConsole);
do {
swiWaitForVBlank();
scanKeys();
pressed = keysDown();
held = keysDownRepeat();
printf("\x1B[30m\x1B[0;26H %s", RetTime().c_str()); // Print time
} while(!held);
consoleSelect(&bottomConsole);
if(held & KEY_UP) {
char val = str[cursorPosition / 2];
u8 shift = !(cursorPosition % 2) * 4;
str[cursorPosition / 2] = (val & (0xF0 >> shift)) | ((val + (1 << shift)) & (0xF << shift));
} else if(held & KEY_DOWN) {
char val = str[cursorPosition / 2];
u8 shift = !(cursorPosition % 2) * 4;
str[cursorPosition / 2] = (val & (0xF0 >> shift)) | ((val - (1 << shift)) & (0xF << shift));
} else if(held & KEY_LEFT) {
if(cursorPosition > 0)
cursorPosition--;
} else if(held & KEY_RIGHT) {
if(cursorPosition < strLen * 2 - 1) {
cursorPosition++;
} else if(strLen < 8) {
strLen++;
cursorPosition++;
}
} else if(pressed & KEY_A) {
break;
} else if(pressed & KEY_B) {
return offset;
} else if(pressed & KEY_X) {
if(strLen > 1) {
str[strLen - 1] = 0;
strLen--;
if(cursorPosition > strLen * 2 - 1)
cursorPosition -= 2;
consoleClear();
}
}
}
}
consoleClear();
printf("\x1B[9;6H\x1B[47m---------------------");
printf("\x1B[10;12HSearching");
printf("\x1B[14;8HPress B to cancel");
printf("\x1B[15;6H---------------------");
size_t len = 32 << 10, pos = offset;
fseek(file, 0, SEEK_END);
size_t fileLen = ftell(file);
char *buf = new char[len];
do {
scanKeys();
if(keysDown() & KEY_B) {
delete[] buf;
return offset;
}
printf("\x1B[12;6H%10d/%d", pos, fileLen);
if(fseek(file, pos, SEEK_SET) != 0)
break;
len = fread(buf, 1, len, file);
for(size_t i = 0; i < len - strLen && len >= strLen; i++) {
if(memcmp(buf + i, str, strLen) == 0) {
delete[] buf;
return (pos + i) & ~7;
}
}
pos += len;
} while(len == 32 << 10);
delete[] buf;
consoleClear();
printf("\x1B[9;5H\x1B[47m---------------------");
printf("\x1B[10;6HReached end of file");
printf("\x1B[11;8Hwith no results");
printf("\x1B[12;5H---------------------");
do {
swiWaitForVBlank();
scanKeys();
printf("\x1B[30m\x1B[0;26H %s", RetTime().c_str()); // Print time
} while(!keysDown());
return offset;
}
void hexEditor(const char *path, int drive) {
// Custom palettes
BG_PALETTE_SUB[0x1F] = 0x9CF7;
BG_PALETTE_SUB[0x2F] = 0xB710;
BG_PALETTE_SUB[0x3F] = 0xAE8D;
BG_PALETTE_SUB[0x7F] = 0xEA2D;
FILE *file = fopen(path, drive < 4 ? "rb+" : "rb");
if(!file)
return;
consoleClear();
fseek(file, 0, SEEK_END);
u32 fileSize = ftell(file);
fseek(file, 0, SEEK_SET);
u8 maxLines = std::min(23lu, fileSize / 8 + (fileSize % 8 != 0));
u32 maxSize = ((fileSize - 8 * maxLines) & ~7) + (fileSize & 7 ? 8 : 0);
u8 cursorPosition = 0, mode = 0;
u16 pressed = 0, held = 0;
u32 offset = 0;
char data[8 * maxLines];
fseek(file, offset, SEEK_SET);
fread(data, 1, sizeof(data), file);
while(1) {
consoleSelect(&bottomConsoleBG);
printf ("\x1B[0;0H\x1B[46m"); // Blue
for(int i = 0; i < 4; i++)
printf ("\2");
printf ("\x1B[42m"); // Green
for(int i = 0; i < 32 - 4; i++)
printf ("\2");
consoleSelect(&bottomConsole);
printf("\x1B[0;11H\x1B[30mHex Editor");
printf("\x1B[0;0H\x1B[30m%04lX", offset >> 0x10);
if(mode < 2) {
fseek(file, offset, SEEK_SET);
toncset(data, 0, sizeof(data));
fread(data, 1, std::min((u32)sizeof(data), fileSize - offset), file);
}
for(int i = 0; i < maxLines; i++) {
printf("\x1B[%d;0H\x1B[37m%04lX", i + 1, (offset + i * 8) & 0xFFFF);
for(int j = 0; j < 4; j++)
printf("\x1B[%d;%dH\x1B[%dm%02X", i + 1, 5 + (j * 2), (mode > 0 && i * 8 + j == cursorPosition) ? (mode > 1 ? 30 : 31) : (offset + i * 8 + j >= fileSize ? 38 : 32 + (j % 2)), data[i * 8 + j]);
for(int j = 0; j < 4; j++)
printf("\x1B[%d;%dH\x1B[%dm%02X", i + 1, 14 + (j * 2), (mode > 0 && i * 8 + 4 + j == cursorPosition) ? (mode > 1 ? 30 : 31) : (offset + i * 8 + 4 + j >= fileSize ? 38 : 32 + (j % 2)), data[i * 8 + 4 + j]);
char line[9] = {0};
for(int j = 0; j < 8; j++) {
char c = data[i * 8 + j];
if(c < ' ' || c > 127)
line[j] = ' ';
else
line[j] = c;
}
printf("\x1B[%d;23H\x1B[47m%.8s", i + 1, line);
if(mode > 0 && cursorPosition / 8 == i) {
printf("\x1B[%d;%dH\x1B[%dm%c", i + 1, 23 + cursorPosition % 8, mode > 1 ? 30 : 31, line[cursorPosition % 8]);
}
}
consoleSelect(&topConsole);
do {
swiWaitForVBlank();
scanKeys();
pressed = keysDown();
held = keysDownRepeat();
printf("\x1B[30m\x1B[0;26H %s", RetTime().c_str()); // Print time
} while(!held);
consoleSelect(&bottomConsole);
if(mode == 0) {
if(keysHeld() & KEY_R && held & (KEY_UP | KEY_DOWN | KEY_LEFT | KEY_RIGHT)) {
if(held & KEY_UP) {
offset = std::max((s64)offset - 0x1000, 0ll);
} else if(held & KEY_DOWN) {
offset = std::min(offset + 0x1000, maxSize);
} else if(held & KEY_LEFT) {
offset = std::max((s64)offset - 0x10000, 0ll);
} else if(held & KEY_RIGHT) {
offset = std::min(offset + 0x10000, maxSize);
}
} else if(held & KEY_UP) {
if(offset >= 8)
offset -= 8;
} else if(held & KEY_DOWN) {
if(offset < fileSize - 8 * maxLines && fileSize > 8 * maxLines)
offset += 8;
} else if(held & KEY_LEFT) {
offset = std::max((s64)offset - 8 * maxLines, 0ll);
} else if(held & KEY_RIGHT) {
offset = std::min(offset + 8 * maxLines, maxSize);
} else if(pressed & KEY_A) {
mode = 1;
} else if(pressed & KEY_B) {
break;
} else if(pressed & KEY_X) {
offset = std::min(search(offset, file), maxSize);
consoleClear();
} else if(pressed & KEY_Y) {
offset = std::min(jumpToOffset(offset), maxSize);
consoleClear();
}
} else if(mode == 1) {
if(held & KEY_UP) {
if(cursorPosition >= 8)
cursorPosition -= 8;
else if(offset > 8)
offset -= 8;
} else if(held & KEY_DOWN) {
if(cursorPosition < 8 * 22)
cursorPosition += 8;
else if(offset < fileSize - 8 * maxLines && fileSize > 8 * maxLines)
offset += 8;
cursorPosition = std::min(cursorPosition, (u8)(fileSize - offset - 1));
} else if(held & KEY_LEFT) {
if(cursorPosition > 0)
cursorPosition--;
} else if(held & KEY_RIGHT) {
if(cursorPosition < 8 * maxLines - 1)
cursorPosition = std::min((u8)(cursorPosition + 1), (u8)(fileSize - offset - 1));
} else if(pressed & KEY_A) {
if(drive < 4) {
mode = 2;
consoleSelect(&bottomConsoleBG);
printf("\x1B[%d;%dH\x1B[%dm\2\2", 1 + cursorPosition / 8, 5 + (cursorPosition % 8 * 2) + (cursorPosition % 8 / 4), 31);
printf("\x1B[%d;%dH\x1B[%dm\2", 1 + cursorPosition / 8, 23 + cursorPosition % 8, 31);
consoleSelect(&bottomConsole);
}
} else if(pressed & KEY_B) {
mode = 0;
} else if(pressed & KEY_X) {
offset = std::min(search(offset, file), maxSize);
consoleClear();
} else if(pressed & KEY_Y) {
offset = std::min(jumpToOffset(offset), maxSize);
consoleClear();
}
} else if(mode == 2) {
if(held & KEY_UP) {
data[cursorPosition]++;
} else if(held & KEY_DOWN) {
data[cursorPosition]--;
} else if(held & KEY_LEFT) {
data[cursorPosition] -= 0x10;
} else if(held & KEY_RIGHT) {
data[cursorPosition] += 0x10;
} else if(pressed & (KEY_A | KEY_B)) {
mode = 1;
fseek(file, offset + cursorPosition, SEEK_SET);
fwrite(data + cursorPosition, 1, 1, file);
consoleSelect(&bottomConsoleBG);
printf("\x1B[%d;%dH\x1B[%dm\2\2", 1 + cursorPosition / 8, 5 + (cursorPosition % 8 * 2) + (cursorPosition % 8 / 4), 30);
printf("\x1B[%d;%dH\x1B[%dm\2", 1 + cursorPosition / 8, 23 + cursorPosition % 8, 30);
consoleSelect(&bottomConsole);
}
}
}
fclose(file);
// Restore color palette
BG_PALETTE_SUB[0x1F] = 0x000F;
BG_PALETTE_SUB[0x2F] = 0x01E0;
BG_PALETTE_SUB[0x3F] = 0x3339;
BG_PALETTE_SUB[0x7F] = 0x656A;
consoleSelect(&bottomConsoleBG);
consoleClear();
}