mirror of
https://github.com/brain-hackers/lab
synced 2024-11-14 22:28:00 +09:00
263 lines
7.3 KiB
C++
263 lines
7.3 KiB
C++
/*
|
|
* Tablewalker Kernel Driver
|
|
* Tablewalker for electric dictionary(except PW-x1).
|
|
*
|
|
* Copyright (C) 2021 C. Shirasaka <holly_programmer@outlook.com>
|
|
* based on
|
|
** ResetKitHelper
|
|
** Soft/hard reset the electronic dictionary.
|
|
**
|
|
** Copyright (C) 2012 T. Kawada <tcppjp [ at ] gmail.com>
|
|
*
|
|
* This file is licensed in MIT license.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation the
|
|
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
|
* sell copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include <stdint.h>
|
|
|
|
#define RESETKITHELPER_API __declspec(dllexport)
|
|
|
|
#include "ResetKitHelper.h"
|
|
|
|
#pragma mark - EDNA2 Specific
|
|
|
|
// only for 0x80000000 - 0x80100000
|
|
#define EDNA2Register(addr) (*(volatile uint32_t *)((addr) + 0x30000000))
|
|
|
|
unsigned int pow(unsigned int base, unsigned int exponent) {
|
|
unsigned int ans = 1;
|
|
for (unsigned int i = 0; i < exponent; i++) {
|
|
ans *= base;
|
|
}
|
|
return ans;
|
|
}
|
|
|
|
void print(char *string,unsigned int *addr) {
|
|
do {
|
|
*addr |= *string;
|
|
string++;
|
|
} while (*string != '\0');
|
|
}
|
|
|
|
unsigned int div(unsigned int dividend,unsigned int divisor){
|
|
unsigned int ans;
|
|
ans=0;
|
|
while(dividend>divisor){
|
|
dividend-=divisor;
|
|
ans++;
|
|
}
|
|
return ans;
|
|
}
|
|
|
|
void printhex(unsigned int num,unsigned int *addr) {
|
|
int digit=8;
|
|
print("0x",addr);
|
|
while (digit) {
|
|
unsigned int weight = pow(16, digit - 1);
|
|
unsigned int ans = div(num,weight);
|
|
if (ans < 10)
|
|
*addr |= '0' + ans;
|
|
else
|
|
*addr |= 'A' + ans - 10;
|
|
num -= ans * weight;
|
|
digit--;
|
|
}
|
|
}
|
|
|
|
unsigned int PhysicalRead(unsigned int addr){
|
|
unsigned int ret;
|
|
asm volatile(
|
|
"mrc p15,0,r10,c1,c0,0\n" // read ctrl regs
|
|
"mov r9,r10\n"
|
|
"bic r10, r10, #1\n"
|
|
"mcr p15,0,r10,c1,c0,0\n"// disable MMU
|
|
"ldr %0, [%1]\n"
|
|
"mcr p15,0,r9,c1,c0,0\n" // enable MMU
|
|
:"=r" (ret): "r" (addr):"r10","r9"
|
|
);
|
|
return ret;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API BOOL RKH_IOControl(DWORD handle, DWORD dwIoControlCode, DWORD *pInBuf, DWORD nInBufSize, DWORD *pOutBuf, DWORD nOutBufSize,
|
|
PDWORD pBytesReturned){
|
|
SetLastError(0);
|
|
switch (dwIoControlCode){
|
|
case IOCTL_RKH_DO_SOFT_RESET:
|
|
wchar_t buf[256]={0};
|
|
unsigned int TTBR0,table0,table1,ram=0,ahp=0,descriptor,physaddr,vaddr=0;
|
|
asm volatile(
|
|
"mrs r0, cpsr\n"
|
|
"orr r0,r0,#0x80\n"
|
|
"msr cpsr_c,r0\n" //interrupt disable
|
|
"mrc p15, 0, %0, c2, c0, 0\n"
|
|
:"=r" (TTBR0)::"r0"
|
|
);
|
|
swprintf(buf, L"TTBR=%#x\r\n", TTBR0);
|
|
OutputDebugString(buf);
|
|
do{
|
|
table0=((TTBR0&0xFFFFC000)|((vaddr&0xFFF00000)>>18));
|
|
descriptor = PhysicalRead(table0);
|
|
if((descriptor&0x3)==0) continue; //Fault
|
|
else if((descriptor&0x3)==1){ //Coarse page table
|
|
table1=((descriptor&0xFFFFFC00)|((vaddr&0xFF000)>>9));
|
|
descriptor = PhysicalRead(table1);
|
|
if((descriptor&0x3)==0) continue;//fault
|
|
else if((descriptor&0x3)==1){//large page
|
|
physaddr=((descriptor&0xFFFF0000)|(vaddr&0xFFFF));
|
|
swprintf(buf, L"Virt Addr=%#x : Phys Addr=%#x\r\n",vaddr,physaddr);
|
|
OutputDebugString(buf);
|
|
if(physaddr==0x40000000)ram=vaddr;
|
|
else if(physaddr==0x80070000)ahp=vaddr;
|
|
}else if((descriptor&0x3)==2){//small page
|
|
physaddr=((descriptor&0xFFFFF000)|(vaddr&0xFFF));
|
|
swprintf(buf, L"Virt Addr=%#x : Phys Addr=%#x\r\n",vaddr,physaddr);
|
|
OutputDebugString(buf);
|
|
if(physaddr==0x40000000)ram=vaddr;
|
|
else if(physaddr==0x80070000)ahp=vaddr;
|
|
}
|
|
}else if((descriptor&0x3)==2){ //Section
|
|
physaddr=((descriptor&0xFFF00000)|(vaddr&0xFFFFF));
|
|
swprintf(buf, L"Virt Addr=%#x : Phys Addr=%#x\r\n",vaddr,physaddr);
|
|
OutputDebugString(buf);
|
|
if(physaddr==0x40000000)ram=vaddr;
|
|
else if(physaddr==0x80070000)ahp=vaddr;
|
|
}else if((descriptor&0x3)==3){ //Fine page table
|
|
table1=((descriptor&0xFFFFF000)|((vaddr&0xFC000)>>7));
|
|
swprintf(buf, L"second level page table=%#x\r\n", table1);
|
|
OutputDebugString(buf);
|
|
descriptor = PhysicalRead(table1);
|
|
swprintf(buf, L"second level page table descriptor=%#x\r\n", descriptor);
|
|
OutputDebugString(buf);
|
|
if((descriptor&0x3)==0) continue;//fault
|
|
else {//tiny page
|
|
physaddr=((descriptor&0xFFFFFC00)|(vaddr&0x3FF));
|
|
swprintf(buf, L"Virt Addr=%#x : Phys Addr=%#x\r\n",vaddr,physaddr);
|
|
OutputDebugString(buf);
|
|
if(physaddr==0x40000000)ram=vaddr;
|
|
else if(physaddr==0x80070000)ahp=vaddr;
|
|
}
|
|
}
|
|
}while(vaddr+=0x1000,vaddr<0xffff0000);
|
|
|
|
swprintf(buf, L"RAM=%#x\r\nNear UART=%#x\r\n", ram, ahp);
|
|
OutputDebugString(buf);
|
|
|
|
asm volatile(
|
|
"mrs r0, cpsr\n"
|
|
"bic r0,r0,#0x80\n"
|
|
"msr cpsr_c,r0\n" //interrupt disable
|
|
:::"r0"
|
|
);
|
|
/*
|
|
if(ahp){
|
|
*((unsigned int *)(ahp+0x4000+0x30)) |= 0x101; //UART Tx Enable
|
|
*((unsigned int *)(ahp+0x4000+0x38)) &= 0x0; //UART Intr disable
|
|
print("DUART=",(unsigned int *)ahp+0x4000);
|
|
printhex(ahp+0x2D0000+0x1C,(unsigned int *)ahp+0x4000);
|
|
print("\n",(unsigned int *)ahp+0x4000);
|
|
|
|
if(ram){
|
|
print("DRAM=",(unsigned int *)ahp+0x4000);
|
|
printhex(ram,(unsigned int *)ahp+0x4000);
|
|
print("\n",(unsigned int *)ahp+0x4000);
|
|
}
|
|
|
|
}
|
|
*/
|
|
return TRUE;
|
|
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API BOOL RKH_Read(DWORD handle, LPVOID pBuffer, DWORD dwNumBytes)
|
|
{
|
|
SetLastError(ERROR_INVALID_FUNCTION);
|
|
return FALSE;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API BOOL RKH_Write(DWORD handle, LPVOID pBuffer, DWORD dwNumBytes)
|
|
{
|
|
SetLastError(ERROR_INVALID_FUNCTION);
|
|
return FALSE;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API DWORD RKH_Seek(DWORD handle, long lDistance, DWORD dwMoveMethod)
|
|
{
|
|
SetLastError(ERROR_INVALID_FUNCTION);
|
|
return FALSE;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API void RKH_PowerUp(void)
|
|
{
|
|
OutputDebugString(L"ResetKit: resuming.");
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API void RKH_PowerDown(void)
|
|
{
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API DWORD RKH_Init(LPCTSTR pContext, DWORD dwBusContext)
|
|
{
|
|
|
|
void *ctx;
|
|
ctx = (void *)LocalAlloc(LPTR, sizeof(4));
|
|
|
|
return (DWORD)ctx;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API DWORD RKH_Open(DWORD dwData, DWORD dwAccess, DWORD dwShareMode)
|
|
{
|
|
|
|
//SetProcPermissions(-1);
|
|
|
|
void *hnd = (void *)LocalAlloc(LPTR, 4);
|
|
return (DWORD)hnd;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API BOOL RKH_Close(DWORD handle)
|
|
{
|
|
LocalFree((void *)handle);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
extern "C" RESETKITHELPER_API BOOL RKH_Deinit(DWORD dwContext)
|
|
{
|
|
|
|
LocalFree((void *)dwContext);
|
|
return TRUE;
|
|
}
|
|
|
|
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:
|
|
case DLL_THREAD_DETACH:
|
|
case DLL_PROCESS_DETACH:
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|