brain-hackers_pepepper-mirror/brainlilo
1
0
Fork 0
mirror of https://github.com/brain-hackers/brainlilo synced 2025-01-02 17:40:05 +09:00
Code Issues Projects Releases Wiki Activity

apply clang-format

Browse Source
This commit is contained in:
Chiharu Shirasaka 2022-01-24 01:52:45 +09:00
parent 3689ee66f7
commit d67326dc8f
6 changed files with 468 additions and 467 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.clang-format Normal file
View File

@ -0,0 +1 @@
BasedOnStyle: Microsoft

82
BrainLILO.cpp
View File

@ -40,14 +40,16 @@ static bool g_initialized=false;
static HANDLE g_helperHandle = NULL;
static HINSTANCE g_hInstance;
static void initialize(){
static void initialize()
{
if (g_initialized)
return;
g_initialized = true;
}
static void getThisDllDirectoryPath(LPWSTR buffer){
static void getThisDllDirectoryPath(LPWSTR buffer)
{
// retrive the path of the application.
GetModuleFileName(g_hInstance, buffer, 512);
@ -55,8 +57,10 @@ static void getThisDllDirectoryPath(LPWSTR buffer){
size_t i = 0;
size_t j = notFound;
while(buffer[i]){
if(buffer[i]==L'/' || buffer[i]==L'\\'){
while (buffer[i])
{
if (buffer[i] == L'/' || buffer[i] == L'\\')
{
j = i;
}
i++;
@ -66,26 +70,29 @@ static void getThisDllDirectoryPath(LPWSTR buffer){
return;
buffer[j] = 0;
}
static bool isDriverLoaded(){
static bool isDriverLoaded()
{
HANDLE handle;
handle=CreateFile(L"LIN0:", GENERIC_READ|GENERIC_WRITE|FILE_WRITE_ATTRIBUTES, 0, NULL, OPEN_EXISTING, 0, NULL);
handle =
CreateFile(L"LIN0:", GENERIC_READ | GENERIC_WRITE | FILE_WRITE_ATTRIBUTES, 0, NULL, OPEN_EXISTING, 0, NULL);
if (handle == INVALID_HANDLE_VALUE)
handle = NULL;
if(handle){
if (handle)
{
CloseHandle(handle);
return true;
}
return false;
}
static bool addDriverToRegistry(){
static bool addDriverToRegistry()
{
HKEY hKey;
DWORD dw;
RegCreateKeyEx(HKEY_LOCAL_MACHINE, (L"Drivers\\BuiltIn\\LILODrv"), 0, NULL, REG_OPTION_NON_VOLATILE, 0, NULL, &hKey, &dw);
RegCreateKeyEx(HKEY_LOCAL_MACHINE, (L"Drivers\\BuiltIn\\LILODrv"), 0, NULL, REG_OPTION_NON_VOLATILE, 0, NULL, &hKey,
&dw);
dw = 0x0000;
if (RegSetValueEx(hKey, (L"Index"), 0, REG_DWORD, (BYTE *)&dw, sizeof(DWORD)) != ERROR_SUCCESS)
@ -101,13 +108,15 @@ static bool addDriverToRegistry(){
return false;
}
if(RegSetValueEx(hKey, (L"Dll"), 0, REG_SZ, (BYTE*)(L"\\Windows\\BrainLILODrv.dll"), sizeof(TCHAR) * ((DWORD)wcslen((L"\\Windows\\BrainLILODrv.dll")) + 1)) != ERROR_SUCCESS)
if (RegSetValueEx(hKey, (L"Dll"), 0, REG_SZ, (BYTE *)(L"\\Windows\\BrainLILODrv.dll"),
sizeof(TCHAR) * ((DWORD)wcslen((L"\\Windows\\BrainLILODrv.dll")) + 1)) != ERROR_SUCCESS)
{
RegCloseKey(hKey);
return false;
}
if(RegSetValueEx(hKey, (L"Prefix"), 0, REG_SZ, (BYTE*)(L"LIN"), sizeof(TCHAR) * ((DWORD)wcslen((L"LIN")) + 1)) != ERROR_SUCCESS)
if (RegSetValueEx(hKey, (L"Prefix"), 0, REG_SZ, (BYTE *)(L"LIN"), sizeof(TCHAR) * ((DWORD)wcslen((L"LIN")) + 1)) !=
ERROR_SUCCESS)
{
RegCloseKey(hKey);
return false;
@ -116,7 +125,8 @@ static bool addDriverToRegistry(){
return true;
}
static void loadDriverIfNeeded(){
static void loadDriverIfNeeded()
{
if (isDriverLoaded())
return;
@ -137,61 +147,62 @@ static void loadDriverIfNeeded(){
{
wchar_t buf[1024];
swprintf(buf, L"BrainLILO: copying \"%ls\" to \"%ls\"",
driverPath, newDriverPath);
swprintf(buf, L"BrainLILO: copying \"%ls\" to \"%ls\"", driverPath, newDriverPath);
OutputDebugString(buf);
}
if(!CopyFile(driverPath,newDriverPath, FALSE)){
if (!CopyFile(driverPath, newDriverPath, FALSE))
{
if(GetFileAttributes(newDriverPath)==(DWORD)-1){
if (GetFileAttributes(newDriverPath) == (DWORD)-1)
{
OutputDebugString(L"BrainLILO: failed to copy");
return;
}
}
OutputDebugString(L"BrainLILO: registering BrainLILODriver...");
try{
try
{
driver = RegisterDevice(L"LIN", 0, L"\\Windows\\BrainLILODrv.dll", 0);
if (driver == INVALID_HANDLE_VALUE)
driver = NULL;
if(!driver){
if (!driver)
{
driver = RegisterDevice(L"LIN", 0, L"BrainLILODrv.dll", 0);
if (driver == INVALID_HANDLE_VALUE)
driver = NULL;
}
err = GetLastError();
}catch(DWORD e){
}
catch (DWORD e)
{
err = e;
}
if(!driver && (err!=0x964)){
if (!driver && (err != 0x964))
{
OutputDebugString(L"BrainLILO: failed to install...");
return;
}
}
static void openDriver(){
static void openDriver()
{
if (g_helperHandle)
return;
g_helperHandle=CreateFile(L"LIN0:", GENERIC_READ|GENERIC_WRITE|FILE_WRITE_ATTRIBUTES, 0, NULL, OPEN_EXISTING, 0, NULL);
g_helperHandle =
CreateFile(L"LIN0:", GENERIC_READ | GENERIC_WRITE | FILE_WRITE_ATTRIBUTES, 0, NULL, OPEN_EXISTING, 0, NULL);
if (g_helperHandle == INVALID_HANDLE_VALUE)
g_helperHandle = NULL;
}
extern "C" BOOL APIENTRY DllMain(HINSTANCE hInstance,
DWORD ul_reason_for_call,
LPVOID lpReserved
)
extern "C" BOOL APIENTRY DllMain(HINSTANCE hInstance, DWORD ul_reason_for_call, LPVOID lpReserved)
{
switch (ul_reason_for_call)
{
@ -211,13 +222,12 @@ extern "C" BOOL APIENTRY DllMain(HINSTANCE hInstance,
return TRUE;
}
BRAINLILO_API BOOL RKDoSoftReset(){
BRAINLILO_API BOOL RKDoSoftReset()
{
initialize();
loadDriverIfNeeded();
openDriver();
return DeviceIoControl(g_helperHandle, IOCTL_LIN_DO_LINUX,
NULL, 0, NULL, 0,
NULL, NULL);
return DeviceIoControl(g_helperHandle, IOCTL_LIN_DO_LINUX, NULL, 0, NULL, 0, NULL, NULL);
}

4
BrainLILO.h
View File

@ -9,12 +9,12 @@
#endif
#ifdef __cplusplus
extern "C"{
extern "C"
{
#endif
BRAINLILO_API BOOL RKDoSoftReset();
#ifdef __cplusplus
};
#endif

135
BrainLILODrv.cpp
View File

@ -30,34 +30,27 @@
* THE SOFTWARE.
*
*/
#include <windows.h>
#include <stdint.h>
#include <windows.h>
#define FSNOTIFY_POWER_OFF 1
#define FSNOTIFY_POWER_ON 0
#define BRAINLILODRV_API __declspec(dllexport)
#include "BrainLILODrv.h"
#define FILE_DEVICE_POWER FILE_DEVICE_ACPI
#define IOCTL_POWER_CAPABILITIES \
CTL_CODE(FILE_DEVICE_POWER, 0x400, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_GET \
CTL_CODE(FILE_DEVICE_POWER, 0x401, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_SET \
CTL_CODE(FILE_DEVICE_POWER, 0x402, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_QUERY \
CTL_CODE(FILE_DEVICE_POWER, 0x403, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_CAPABILITIES CTL_CODE(FILE_DEVICE_POWER, 0x400, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_GET CTL_CODE(FILE_DEVICE_POWER, 0x401, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_SET CTL_CODE(FILE_DEVICE_POWER, 0x402, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_POWER_QUERY CTL_CODE(FILE_DEVICE_POWER, 0x403, METHOD_BUFFERED, FILE_ANY_ACCESS)
typedef BOOL (*KernelIoControlProc)(DWORD dwIoControlCode, LPVOID lpInBuf,
DWORD nInBufSize, LPVOID lpOutBuf, DWORD nOutBufSize,LPDWORD lpBytesReturned);
typedef BOOL (*KernelIoControlProc)(DWORD dwIoControlCode, LPVOID lpInBuf, DWORD nInBufSize, LPVOID lpOutBuf,
DWORD nOutBufSize, LPDWORD lpBytesReturned);
static KernelIoControlProc KernelIoControl;
typedef LARGE_INTEGER PHYSICAL_ADDRESS, *PPHYSICAL_ADDRESS;
typedef PVOID (*MmMapIoSpaceProc)(PHYSICAL_ADDRESS, ULONG, BOOL);
@ -70,14 +63,17 @@ typedef LPVOID (*AllocPhysMemProc)(DWORD,DWORD,DWORD,DWORD,PULONG);
DWORD FileSize;
static void disableInterrupts(){
static void disableInterrupts()
{
asm volatile("mrs r0, cpsr\n"
"orr r0,r0,#0x80\n"
"msr cpsr_c,r0\n"
"mov r0,#1":::"r0");
"mov r0,#1" ::
: "r0");
}
static void EDNA2_physicalInvoker(){
static void EDNA2_physicalInvoker()
{
// r0-r7=params
// r8=proc address
asm volatile("nop\n" // who cares interrupt vectors?
@ -103,24 +99,22 @@ static void EDNA2_physicalInvoker(){
"mcr p15,0,r9,c8,c7,0\n" // invalidate tlb
"mov pc, r8\n"
"nop\n"
"nop\n"
);
"nop\n");
}
static void EDNA2_installPhysicalInvoker(){
static void EDNA2_installPhysicalInvoker()
{
void *ptr = (void *)0xa8000000;
wchar_t buf[256];
swprintf(buf, L"BrainLILO: copying PhysicalInvoker to 0x%08x from 0x%08x\n",
(int)(ptr), (int)(&EDNA2_physicalInvoker));
swprintf(buf, L"BrainLILO: copying PhysicalInvoker to 0x%08x from 0x%08x\n", (int)(ptr),
(int)(&EDNA2_physicalInvoker));
OutputDebugString(buf);
memcpy(ptr, (const void *)&EDNA2_physicalInvoker, 64 * 4);
// clearCache();
}
__attribute__((noreturn))
static void EDNA2_runPhysicalInvoker(){
__attribute__((noreturn)) static void EDNA2_runPhysicalInvoker()
{
// r0=info
asm volatile("msr cpsr_c, #211\n" // to supervisor mode
"mrc p15,0,r0,c1,c0,0\n" // read ctrl regs
@ -128,7 +122,8 @@ static void EDNA2_runPhysicalInvoker(){
"mcr p15,0,r0,c1,c0,0\n" // write ctrl regs
);
for(DWORD i=0;i<FileSize;i++)*((char *)(0xa0200000+i))=*((char *)(0xa0000000+i));
for (DWORD i = 0; i < FileSize; i++)
*((char *)(0xa0200000 + i)) = *((char *)(0xa0000000 + i));
asm volatile("ldr r0, =0x0000\n"
"ldr r1, =0x0000\n"
@ -148,11 +143,12 @@ static void EDNA2_runPhysicalInvoker(){
);
// never reach here
while(true);
while (true)
;
}
__attribute__((noreturn))
static DWORD EDNA2_callKernelEntryPoint(){
__attribute__((noreturn)) static DWORD EDNA2_callKernelEntryPoint()
{
OutputDebugString(L"BrainLILO: disabling interrupts");
disableInterrupts();
OutputDebugString(L"BrainLILO: injecting code to internal ram");
@ -161,7 +157,8 @@ static DWORD EDNA2_callKernelEntryPoint(){
EDNA2_runPhysicalInvoker();
}
static bool doLinux(){
static bool doLinux()
{
TCHAR bootloaderFileName[128] = TEXT("\\Storage Card\\loader\\u-boot.bin");
HANDLE hFile;
LPVOID bootloaderptr;
@ -171,7 +168,8 @@ static bool doLinux(){
OutputDebugString(L"BrainLILO: Opening Bootloader file...");
hFile = CreateFile(bootloaderFileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE) {
if (hFile == INVALID_HANDLE_VALUE)
{
OutputDebugString(L"Cant open bootloader");
return false;
}
@ -182,9 +180,9 @@ static bool doLinux(){
swprintf(buf, L"BrainLILO: Bootloader file size %d Byte\n", FileSize);
OutputDebugString(buf);
OutputDebugString(L"BrainLILO: Preloading bootloader to 0xa0000000...");
if (!ReadFile(hFile , (void *)0xa0000000 , FileSize , &wReadSize , NULL)) {
if (!ReadFile(hFile, (void *)0xa0000000, FileSize, &wReadSize, NULL))
{
OutputDebugString(L"Cant read bootloader");
return false;
}
@ -192,20 +190,24 @@ static bool doLinux(){
CloseHandle(hFile);
OutputDebugString(L"BrainLILO: Notifying power off to filesystems...");
if(FileSystemPowerFunction) FileSystemPowerFunction(FSNOTIFY_POWER_OFF);
if (FileSystemPowerFunction)
FileSystemPowerFunction(FSNOTIFY_POWER_OFF);
OutputDebugString(L"BrainLILO: Starting bootloader call sequence...");
EDNA2_callKernelEntryPoint();
return true;
}
extern "C" BRAINLILODRV_API BOOL LIN_IOControl(DWORD handle, DWORD dwIoControlCode, DWORD *pInBuf, DWORD nInBufSize, DWORD * pOutBuf, DWORD nOutBufSize,
PDWORD pBytesReturned){
extern "C" BRAINLILODRV_API BOOL LIN_IOControl(DWORD handle, DWORD dwIoControlCode, DWORD *pInBuf, DWORD nInBufSize,
DWORD *pOutBuf, DWORD nOutBufSize, PDWORD pBytesReturned)
{
SetLastError(0);
switch(dwIoControlCode){
switch (dwIoControlCode)
{
case IOCTL_LIN_DO_LINUX:
if(!doLinux()){
if (!doLinux())
{
if (FileSystemPowerFunction)
FileSystemPowerFunction(FSNOTIFY_POWER_ON);
return FALSE;
@ -216,83 +218,75 @@ extern "C" BRAINLILODRV_API BOOL LIN_IOControl(DWORD handle, DWORD dwIoControlCo
return FALSE;
}
extern "C" BRAINLILODRV_API BOOL LIN_Read(DWORD handle, LPVOID pBuffer, DWORD dwNumBytes){
extern "C" BRAINLILODRV_API BOOL LIN_Read(DWORD handle, LPVOID pBuffer, DWORD dwNumBytes)
{
SetLastError(ERROR_INVALID_FUNCTION);
return FALSE;
}
extern "C" BRAINLILODRV_API BOOL LIN_Write(DWORD handle, LPVOID pBuffer, DWORD dwNumBytes){
extern "C" BRAINLILODRV_API BOOL LIN_Write(DWORD handle, LPVOID pBuffer, DWORD dwNumBytes)
{
SetLastError(ERROR_INVALID_FUNCTION);
return FALSE;
}
extern "C" BRAINLILODRV_API DWORD LIN_Seek(DWORD handle, long lDistance, DWORD dwMoveMethod){
extern "C" BRAINLILODRV_API DWORD LIN_Seek(DWORD handle, long lDistance, DWORD dwMoveMethod)
{
SetLastError(ERROR_INVALID_FUNCTION);
return FALSE;
}
extern "C" BRAINLILODRV_API void LIN_PowerUp(void){
extern "C" BRAINLILODRV_API void LIN_PowerUp(void)
{
OutputDebugString(L"BrainLILO: resuming.");
}
extern "C" BRAINLILODRV_API void LIN_PowerDown(void){
extern "C" BRAINLILODRV_API void LIN_PowerDown(void)
{
}
extern "C" BRAINLILODRV_API DWORD LIN_Init(LPCTSTR pContext,
DWORD dwBusContext){
extern "C" BRAINLILODRV_API DWORD LIN_Init(LPCTSTR pContext, DWORD dwBusContext)
{
void *ctx;
ctx = (void *)LocalAlloc(LPTR, sizeof(4));
return (DWORD)ctx;
}
extern "C" BRAINLILODRV_API DWORD LIN_Open(DWORD dwData, DWORD dwAccess, DWORD dwShareMode){
extern "C" BRAINLILODRV_API DWORD LIN_Open(DWORD dwData, DWORD dwAccess, DWORD dwShareMode)
{
void *hnd = (void *)LocalAlloc(LPTR, 4);
return (DWORD)hnd;
}
extern "C" BRAINLILODRV_API BOOL LIN_Close(DWORD handle){
extern "C" BRAINLILODRV_API BOOL LIN_Close(DWORD handle)
{
LocalFree((void *)handle);
return TRUE;
}
extern "C" BRAINLILODRV_API BOOL LIN_Deinit(DWORD dwContext){
extern "C" BRAINLILODRV_API BOOL LIN_Deinit(DWORD dwContext)
{
LocalFree((void *)dwContext);
return TRUE;
}
extern "C" BOOL APIENTRY DllMainCRTStartup( HANDLE hModule,
DWORD ul_reason_for_call,
LPVOID lpReserved
)
extern "C" BOOL APIENTRY DllMainCRTStartup(HANDLE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
switch (ul_reason_for_call)
{
case DLL_PROCESS_ATTACH:
case DLL_THREAD_ATTACH:
KernelIoControl=(KernelIoControlProc)
GetProcAddress(LoadLibrary(L"COREDLL"),
L"KernelIoControl");
KernelIoControl = (KernelIoControlProc)GetProcAddress(LoadLibrary(L"COREDLL"), L"KernelIoControl");
MmMapIoSpace=(MmMapIoSpaceProc)
GetProcAddress(LoadLibrary(L"CEDDK"),
L"MmMapIoSpace");
MmMapIoSpace = (MmMapIoSpaceProc)GetProcAddress(LoadLibrary(L"CEDDK"), L"MmMapIoSpace");
FileSystemPowerFunction=(FileSystemPowerFunctionProc)
GetProcAddress(LoadLibrary(L"COREDLL"),
L"FileSystemPowerFunction");
FileSystemPowerFunction =
(FileSystemPowerFunctionProc)GetProcAddress(LoadLibrary(L"COREDLL"), L"FileSystemPowerFunction");
case DLL_THREAD_DETACH:
case DLL_PROCESS_DETACH:
@ -300,4 +294,3 @@ extern "C" BOOL APIENTRY DllMainCRTStartup( HANDLE hModule,
}
return TRUE;
}

3
BrainLILODrv.h
View File

@ -3,7 +3,6 @@
#pragma once
#include <winioctl.h>
#define _LIN_ACCESS_CTL_CODE(_Function) \
CTL_CODE(FILE_DEVICE_UNKNOWN, _Function, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define _LIN_ACCESS_CTL_CODE(_Function) CTL_CODE(FILE_DEVICE_UNKNOWN, _Function, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define IOCTL_LIN_DO_LINUX _LIN_ACCESS_CTL_CODE(0x804)

48
bootloader.cpp
View File

@ -1,9 +1,9 @@
// This file is in public domain.
#include <windows.h>
#include <vector>
#include <algorithm>
#include <tchar.h>
#include <vector>
#include <windows.h>
#define MainWindowClassName L"SelectorMainWindowClass"
@ -11,16 +11,15 @@
#define DataTypeAlertInformation 0x1001
#define DataTypeAlertWarning 0x1002
static void showAlertWarning(LPCWSTR message,
LPCWSTR title){
static void showAlertWarning(LPCWSTR message, LPCWSTR title)
{
HWND selectorMainWindow = FindWindow(MainWindowClassName, NULL);
if(!selectorMainWindow){
MessageBox(NULL, message, title,
MB_ICONWARNING);
if (!selectorMainWindow)
{
MessageBox(NULL, message, title, MB_ICONWARNING);
return;
}
wchar_t data[2048];
@ -39,42 +38,43 @@ static void showAlertWarning(LPCWSTR message,
info.lpData = (LPVOID)global;
SendMessage(selectorMainWindow, WM_COPYDATA, NULL,
(LPARAM)&info);
SendMessage(selectorMainWindow, WM_COPYDATA, NULL, (LPARAM)&info);
GlobalFree(global);
}
int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPTSTR lpCmd, int nShow){
int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPTSTR lpCmd, int nShow)
{
HINSTANCE lib = LoadLibrary(L"BrainLILO");
if(!lib){
if (!lib)
{
wchar_t buf[256];
swprintf(buf, L"Cannot perform a soft reset.\n"
swprintf(buf,
L"Cannot perform a soft reset.\n"
L"BrainLILO was not loaded (0x%08x).",
GetLastError());
showAlertWarning(buf, L"Error");
return 1;
}
typedef BOOL (*RKDoSoftResetProc)();
RKDoSoftResetProc RKDoSoftReset=
(RKDoSoftResetProc)GetProcAddress(lib, L"RKDoSoftReset");
if(!RKDoSoftReset){
RKDoSoftResetProc RKDoSoftReset = (RKDoSoftResetProc)GetProcAddress(lib, L"RKDoSoftReset");
if (!RKDoSoftReset)
{
wchar_t buf[256];
swprintf(buf, L"Cannot perform a soft reset.\n"
swprintf(buf,
L"Cannot perform a soft reset.\n"
L"RKDoSoftReset not found (0x%08x).",
GetLastError());
showAlertWarning(buf, L"Error");
return 1;
}
if(!RKDoSoftReset()){
if (!RKDoSoftReset())
{
wchar_t buf[256];
swprintf(buf, L"Cannot perform a soft reset.\n"
swprintf(buf,
L"Cannot perform a soft reset.\n"
L"Operation failed (0x%08x).",
GetLastError());
showAlertWarning(buf, L"Error");
@ -82,5 +82,3 @@ int APIENTRY WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPTSTR lpCmd, int nShow){
}
return 0;
}