mirror of
https://github.com/rvtr/GodMode9i.git
synced 2025-06-19 03:05:43 -04:00
4c726266f7
* don't run b4ds anymore, as it is obsolete * fix off by one error when printing SHA1 hash
313 lines
7.6 KiB
C++
313 lines
7.6 KiB
C++
#include "fileOperations.h"
|
|
#include <nds.h>
|
|
#include <fat.h>
|
|
#include <stdio.h>
|
|
#include <dirent.h>
|
|
#include <vector>
|
|
|
|
#include "date.h"
|
|
#include "file_browse.h"
|
|
|
|
#define copyBufSize 0x8000
|
|
#define shaChunkSize 0x10000
|
|
|
|
u32 copyBuf[copyBufSize];
|
|
|
|
extern PrintConsole topConsole, bottomConsole;
|
|
|
|
std::vector<ClipboardFile> clipboard;
|
|
bool clipboardOn = false;
|
|
bool clipboardUsed = false;
|
|
|
|
void printBytes(int bytes)
|
|
{
|
|
if (bytes == 1)
|
|
iprintf("%d Byte", bytes);
|
|
|
|
else if (bytes < 1024)
|
|
iprintf("%d Bytes", bytes);
|
|
|
|
else if (bytes < (1024 * 1024))
|
|
printf("%d KB", bytes / 1024);
|
|
|
|
else if (bytes < (1024 * 1024 * 1024))
|
|
printf("%d MB", bytes / 1024 / 1024);
|
|
|
|
else
|
|
printf("%d GB", bytes / 1024 / 1024 / 1024);
|
|
}
|
|
|
|
void printBytesAlign(int bytes)
|
|
{
|
|
if (bytes == 1)
|
|
iprintf("%4d Byte", bytes);
|
|
|
|
else if (bytes < 1024)
|
|
iprintf("%3d Bytes", bytes);
|
|
|
|
else if (bytes < (1024 * 1024))
|
|
printf("%6d KB", bytes / 1024);
|
|
|
|
else if (bytes < (1024 * 1024 * 1024))
|
|
printf("%6d MB", bytes / 1024 / 1024);
|
|
|
|
else
|
|
printf("%6d GB", bytes / 1024 / 1024 / 1024);
|
|
}
|
|
|
|
off_t getFileSize(const char *fileName)
|
|
{
|
|
FILE* fp = fopen(fileName, "rb");
|
|
off_t fsize = 0;
|
|
if (fp) {
|
|
fseek(fp, 0, SEEK_END);
|
|
fsize = ftell(fp); // Get source file's size
|
|
fseek(fp, 0, SEEK_SET);
|
|
}
|
|
fclose(fp);
|
|
|
|
return fsize;
|
|
}
|
|
|
|
bool calculateSHA1(const char *fileName, u8 *sha1)
|
|
{
|
|
off_t fsize = getFileSize(fileName);
|
|
u8 *buf = (u8*) malloc(shaChunkSize);
|
|
if (!buf) {
|
|
iprintf("Could not allocate buffer\n");
|
|
return false;
|
|
}
|
|
FILE* fp = fopen(fileName, "rb");
|
|
if (!fp) {
|
|
iprintf("Could not open file for reading\n");
|
|
free(buf);
|
|
return false;
|
|
}
|
|
memset(sha1, 0, 20);
|
|
swiSHA1context_t ctx;
|
|
ctx.sha_block=0; //this is weird but it has to be done
|
|
swiSHA1Init(&ctx);
|
|
while (true) {
|
|
size_t ret = fread(buf, 1, shaChunkSize, fp);
|
|
if (!ret) break;
|
|
swiSHA1Update(&ctx, buf, ret);
|
|
scanKeys();
|
|
int keys = keysHeld();
|
|
if (keys & KEY_START) return false;
|
|
iprintf("\x1b[1;A");
|
|
iprintf("%ld/%lld bytes processed\n", ftell(fp), fsize);
|
|
}
|
|
swiSHA1Final(sha1, &ctx);
|
|
free(buf);
|
|
return true;
|
|
}
|
|
|
|
void dirCopy(DirEntry* entry, int i, const char *destinationPath, const char *sourcePath) {
|
|
std::vector<DirEntry> dirContents;
|
|
dirContents.clear();
|
|
if (entry->isDirectory) chdir((sourcePath + ("/" + entry->name)).c_str());
|
|
getDirectoryContents(dirContents);
|
|
if (((int)dirContents.size()) == 1) mkdir((destinationPath + ("/" + entry->name)).c_str(), 0777);
|
|
if (((int)dirContents.size()) != 1) fcopy((sourcePath + ("/" + entry->name)).c_str(), (destinationPath + ("/" + entry->name)).c_str());
|
|
}
|
|
|
|
int fcopy(const char *sourcePath, const char *destinationPath)
|
|
{
|
|
DIR *isDir = opendir (sourcePath);
|
|
|
|
if (isDir != NULL) {
|
|
closedir(isDir);
|
|
|
|
// Source path is a directory
|
|
chdir(sourcePath);
|
|
std::vector<DirEntry> dirContents;
|
|
getDirectoryContents(dirContents);
|
|
DirEntry* entry = NULL;
|
|
|
|
mkdir(destinationPath, 0777);
|
|
for (int i = 1; i < ((int)dirContents.size()); i++) {
|
|
chdir(sourcePath);
|
|
entry = &dirContents.at(i);
|
|
dirCopy(entry, i, destinationPath, sourcePath);
|
|
}
|
|
|
|
chdir (destinationPath);
|
|
chdir ("..");
|
|
return 1;
|
|
} else {
|
|
closedir(isDir);
|
|
|
|
// Source path is a file
|
|
FILE* sourceFile = fopen(sourcePath, "rb");
|
|
off_t fsize = 0;
|
|
if (sourceFile) {
|
|
fseek(sourceFile, 0, SEEK_END);
|
|
fsize = ftell(sourceFile); // Get source file's size
|
|
fseek(sourceFile, 0, SEEK_SET);
|
|
} else {
|
|
fclose(sourceFile);
|
|
return -1;
|
|
}
|
|
|
|
FILE* destinationFile = fopen(destinationPath, "wb");
|
|
//if (destinationFile) {
|
|
fseek(destinationFile, 0, SEEK_SET);
|
|
/*} else {
|
|
fclose(sourceFile);
|
|
fclose(destinationFile);
|
|
return -1;
|
|
}*/
|
|
|
|
off_t offset = 0;
|
|
int numr;
|
|
while (1)
|
|
{
|
|
scanKeys();
|
|
if (keysHeld() & KEY_B) {
|
|
// Cancel copying
|
|
fclose(sourceFile);
|
|
fclose(destinationFile);
|
|
return -1;
|
|
break;
|
|
}
|
|
consoleSelect(&topConsole);
|
|
printf ("\x1B[30m"); // Print black color
|
|
// Move to right side of screen
|
|
printf ("\x1b[0;26H");
|
|
// Print time
|
|
printf (" %s" ,RetTime().c_str());
|
|
|
|
consoleSelect(&bottomConsole);
|
|
printf ("\x1B[47m"); // Print foreground white color
|
|
printf ("\x1b[16;0H");
|
|
printf ("Progress:\n");
|
|
printf ("%i/%i Bytes ", (int)offset, (int)fsize);
|
|
|
|
// Copy file to destination path
|
|
numr = fread(copyBuf, 1, copyBufSize, sourceFile);
|
|
fwrite(copyBuf, 1, numr, destinationFile);
|
|
offset += copyBufSize;
|
|
|
|
if (offset > fsize) {
|
|
fclose(sourceFile);
|
|
fclose(destinationFile);
|
|
|
|
printf ("\x1b[17;0H");
|
|
printf ("%i/%i Bytes ", (int)fsize, (int)fsize);
|
|
for (int i = 0; i < 30; i++) swiWaitForVBlank();
|
|
|
|
return 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
void changeFileAttribs(DirEntry* entry) {
|
|
consoleClear();
|
|
int pressed = 0;
|
|
int cursorScreenPos = 0;
|
|
uint8_t currentAttribs = FAT_getAttr(entry->name.c_str());
|
|
uint8_t newAttribs = currentAttribs;
|
|
|
|
// Position cursor, depending on how long the file name is
|
|
for (int i = 0; i < 256; i++) {
|
|
if (i == 33 || i == 65 || i == 97 || i == 129 || i == 161 || i == 193 || i == 225) {
|
|
cursorScreenPos++;
|
|
}
|
|
if (entry->name.c_str()[i] == '\0') {
|
|
break;
|
|
}
|
|
}
|
|
|
|
printf ("\x1b[0;0H");
|
|
printf (entry->name.c_str());
|
|
if (!entry->isDirectory) {
|
|
printf ("\x1b[%i;0H", 3+cursorScreenPos);
|
|
printf ("filesize: ");
|
|
printBytes(entry->size);
|
|
}
|
|
printf ("\x1b[%i;0H", 5+cursorScreenPos);
|
|
printf ("[ ] U read-only [ ] D hidden");
|
|
printf ("\x1b[%i;0H", 6+cursorScreenPos);
|
|
printf ("[ ] R system [ ] L archive");
|
|
printf ("\x1b[%i;0H", 7+cursorScreenPos);
|
|
printf ("[ ] virtual");
|
|
printf ("\x1b[%i;1H", 7+cursorScreenPos);
|
|
printf ((newAttribs & ATTR_VOLUME) ? "X" : " ");
|
|
printf ("\x1b[%i;0H", 9+cursorScreenPos);
|
|
printf ("(UDRL to change attributes)");
|
|
while (1) {
|
|
consoleSelect(&bottomConsole);
|
|
printf ("\x1B[47m"); // Print foreground white color
|
|
printf ("\x1b[%i;1H", 5+cursorScreenPos);
|
|
printf ((newAttribs & ATTR_READONLY) ? "X" : " ");
|
|
printf ("\x1b[%i;18H", 5+cursorScreenPos);
|
|
printf ((newAttribs & ATTR_HIDDEN) ? "X" : " ");
|
|
printf ("\x1b[%i;1H", 6+cursorScreenPos);
|
|
printf ((newAttribs & ATTR_SYSTEM) ? "X" : " ");
|
|
printf ("\x1b[%i;18H", 6+cursorScreenPos);
|
|
printf ((newAttribs & ATTR_ARCHIVE) ? "X" : " ");
|
|
printf ("\x1b[%i;0H", 11+cursorScreenPos);
|
|
printf ((currentAttribs==newAttribs) ? "(<A> to continue) " : "(<A> to apply, <B> to cancel)");
|
|
|
|
consoleSelect(&topConsole);
|
|
printf ("\x1B[30m"); // Print black color
|
|
// Power saving loop. Only poll the keys once per frame and sleep the CPU if there is nothing else to do
|
|
do {
|
|
// Move to right side of screen
|
|
printf ("\x1b[0;26H");
|
|
// Print time
|
|
printf (" %s" ,RetTime().c_str());
|
|
|
|
scanKeys();
|
|
pressed = keysDown();
|
|
swiWaitForVBlank();
|
|
} while (!(pressed & KEY_UP) && !(pressed & KEY_DOWN) && !(pressed & KEY_RIGHT) && !(pressed & KEY_LEFT)
|
|
&& !(pressed & KEY_A) && !(pressed & KEY_B));
|
|
|
|
if (pressed & KEY_UP) {
|
|
if (newAttribs & ATTR_READONLY) {
|
|
newAttribs -= ATTR_READONLY;
|
|
} else {
|
|
newAttribs += ATTR_READONLY;
|
|
}
|
|
}
|
|
|
|
if (pressed & KEY_DOWN) {
|
|
if (newAttribs & ATTR_HIDDEN) {
|
|
newAttribs -= ATTR_HIDDEN;
|
|
} else {
|
|
newAttribs += ATTR_HIDDEN;
|
|
}
|
|
}
|
|
|
|
if (pressed & KEY_RIGHT) {
|
|
if (newAttribs & ATTR_SYSTEM) {
|
|
newAttribs -= ATTR_SYSTEM;
|
|
} else {
|
|
newAttribs += ATTR_SYSTEM;
|
|
}
|
|
}
|
|
|
|
if (pressed & KEY_LEFT) {
|
|
if (newAttribs & ATTR_ARCHIVE) {
|
|
newAttribs -= ATTR_ARCHIVE;
|
|
} else {
|
|
newAttribs += ATTR_ARCHIVE;
|
|
}
|
|
}
|
|
|
|
if ((pressed & KEY_A) && (currentAttribs!=newAttribs)) {
|
|
FAT_setAttr(entry->name.c_str(), newAttribs);
|
|
break;
|
|
}
|
|
|
|
if ((pressed & KEY_A) || (pressed & KEY_B)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|