Merge pull request #4 from brain-hackers/model-detection

Implement model detection
This commit is contained in:
Takumi Sueda 2022-04-19 03:27:08 +09:00 committed by GitHub
commit fd80af3c62
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 65 additions and 76 deletions

4
.editorconfig Normal file
View File

@ -0,0 +1,4 @@
[*.{c,cpp,h}]
indent_style = space
indent_size = 4
tab_width = 4

View File

@ -30,7 +30,10 @@
* THE SOFTWARE.
*
*/
#include <fstream>
#include <regex>
#include <stdint.h>
#include <stdlib.h>
#include <windows.h>
#define FSNOTIFY_POWER_OFF 1
@ -157,35 +160,76 @@ __attribute__((noreturn)) static DWORD EDNA2_callKernelEntryPoint()
EDNA2_runPhysicalInvoker();
}
static void ShowMessage(std::string msg, std::string title, UINT typ)
{
void *bufMsg;
void *bufTitle;
bufMsg = LocalAlloc(LPTR, msg.length() * sizeof(wchar_t));
bufTitle = LocalAlloc(LPTR, title.length() * sizeof(wchar_t));
mbstowcs((wchar_t *)bufMsg, msg.c_str(), msg.length());
mbstowcs((wchar_t *)bufTitle, title.c_str(), title.length());
MessageBox(NULL, (wchar_t *)bufMsg, (wchar_t *)bufTitle, typ);
LocalFree(bufMsg);
LocalFree(bufTitle);
}
static bool doLinux()
{
TCHAR bootloaderFileName[128] = TEXT("\\Storage Card\\loader\\u-boot.bin");
HANDLE hFile;
wchar_t buf[256];
wchar_t wcBuf[256] = {};
std::ifstream iVersion;
std::string line, model;
std::regex modelRe("[A-Z]{2}-[A-Z0-9]+");
std::smatch match;
std::string fn("\\Storage Card\\loader\\");
HANDLE hUBoot;
DWORD wReadSize;
OutputDebugString(L"BrainLILO: Opening Bootloader file...");
hFile = CreateFile(bootloaderFileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE)
iVersion.open("\\NAND\\version.txt");
while (getline(iVersion, line))
{
OutputDebugString(L"Cant open bootloader");
if (regex_search(line, match, modelRe))
{
model = match[0].str();
break;
}
}
if (model.length() == 0)
{
ShowMessage("Failed to match the model name", "BrainLILO", MB_ICONWARNING);
return false;
}
swprintf(buf, L"BrainLILO: Bootloader file handle 0x%08x\n", (int)(hFile));
OutputDebugString(buf);
FileSize = GetFileSize(hFile, NULL);
swprintf(buf, L"BrainLILO: Bootloader file size %d Byte\n", FileSize);
OutputDebugString(buf);
OutputDebugString(L"BrainLILO: Opening Bootloader file...");
fn += model + ".bin";
mbstowcs(wcBuf, fn.c_str(), fn.length());
hUBoot = CreateFile(wcBuf, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hUBoot == INVALID_HANDLE_VALUE)
{
OutputDebugString(L"Could not open the bootloader");
ShowMessage("Could not open the bootloader: " + fn, "BrainLILO", MB_ICONWARNING);
return false;
}
swprintf(wcBuf, L"BrainLILO: Bootloader file handle 0x%08x\n", (int)(hUBoot));
OutputDebugString(wcBuf);
FileSize = GetFileSize(hUBoot, NULL);
swprintf(wcBuf, L"BrainLILO: Bootloader file size %d Byte\n", FileSize);
OutputDebugString(wcBuf);
OutputDebugString(L"BrainLILO: Preloading bootloader to 0xa0000000...");
if (!ReadFile(hFile, (void *)0xa0000000, FileSize, &wReadSize, NULL))
if (!ReadFile(hUBoot, (void *)0xa0000000, FileSize, &wReadSize, NULL))
{
OutputDebugString(L"Cant read bootloader");
OutputDebugString(L"Could not read the bootloader");
ShowMessage("Could not read the bootloader", "BrainLILO", MB_ICONWARNING);
return false;
}
OutputDebugString(L"BrainLILO: Bootloader copied! Closing file handle...");
CloseHandle(hFile);
CloseHandle(hUBoot);
OutputDebugString(L"BrainLILO: Notifying power off to filesystems...");
if (FileSystemPowerFunction)

View File

@ -28,8 +28,8 @@ DLLFLAGS=-DEV_PLATFORM_WIN32 -DUNICODE -D_UNICODE -DEV_UNSAFE_SWPRINTF -mwin32 \
DRVFLAGS= -DEV_PLATFORM_WIN32 -DUNICODE -D_UNICODE -DEV_UNSAFE_SWPRINTF -mwin32 \
-O0 -mcpu=arm926ej-s -D_WIN32_WCE=0x600 -D_LARGEFILE_SOURCE=1 -D_LARGEFILE64_SOURCE=1 \
-D_FILE_OFFSET_BITS=64 -DNDEBUG -Wall -static \
-Wl,--image-base,0x100000 \
-nostdlib -lcoredll -shared
-Wl,--image-base,0x100000,--allow-multiple-definition \
-lcoredll -shared
.PHONY: all clean

View File

@ -1,59 +1,12 @@
// This file is in public domain.
#include <algorithm>
#include <tchar.h>
#include <vector>
#include <windows.h>
#define MainWindowClassName L"SelectorMainWindowClass"
#define DataTypeAlertError 0x1000
#define DataTypeAlertInformation 0x1001
#define DataTypeAlertWarning 0x1002
static void showAlertWarning(LPCWSTR message, LPCWSTR title)
{
HWND selectorMainWindow = FindWindow(MainWindowClassName, NULL);
if (!selectorMainWindow)
{
MessageBox(NULL, message, title, MB_ICONWARNING);
return;
}
wchar_t data[2048];
wcscpy(data, message);
data[wcslen(message)] = 0;
wcscpy(data + wcslen(message) + 1, title);
size_t dataLen = wcslen(message) + wcslen(title) + 1;
COPYDATASTRUCT info;
info.dwData = DataTypeAlertWarning;
info.cbData = dataLen * sizeof(wchar_t);
HGLOBAL global = GlobalAlloc(GPTR, info.cbData);
memcpy((LPVOID)global, data, info.cbData);
info.lpData = (LPVOID)global;
SendMessage(selectorMainWindow, WM_COPYDATA, NULL, (LPARAM)&info);
GlobalFree(global);
}
int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPTSTR lpCmd, int nShow)
{
HINSTANCE lib = LoadLibrary(L"BrainLILO");
if (!lib)
{
wchar_t buf[256];
swprintf(buf,
L"Cannot perform a soft reset.\n"
L"BrainLILO was not loaded (0x%08x).",
GetLastError());
showAlertWarning(buf, L"Error");
return 1;
}
@ -61,23 +14,11 @@ int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPTSTR lpCmd, int nShow)
RKDoSoftResetProc RKDoSoftReset = (RKDoSoftResetProc)GetProcAddress(lib, L"RKDoSoftReset");
if (!RKDoSoftReset)
{
wchar_t buf[256];
swprintf(buf,
L"Cannot perform a soft reset.\n"
L"RKDoSoftReset not found (0x%08x).",
GetLastError());
showAlertWarning(buf, L"Error");
return 1;
}
if (!RKDoSoftReset())
{
wchar_t buf[256];
swprintf(buf,
L"Cannot perform a soft reset.\n"
L"Operation failed (0x%08x).",
GetLastError());
showAlertWarning(buf, L"Error");
return 1;
}
return 0;