#include #include #include "ScriptInstance.h" #include "ScriptSystem.h" #include "ScriptHook.h" #include "Breakpoints.h" static BOOL ConnectEx(SOCKET s, const SOCKADDR* name, int namelen, PVOID lpSendBuffer, DWORD dwSendDataLength, LPDWORD lpdwBytesSent, LPOVERLAPPED lpOverlapped); vector CScriptInstance::Cache; void CScriptInstance::CacheInstance(CScriptInstance* _this) { Cache.push_back(_this); } void CScriptInstance::UncacheInstance(CScriptInstance* _this) { for (size_t i = 0; i < Cache.size(); i++) { if (Cache[i] == _this) { Cache.erase(Cache.begin() + i); break; } } } CScriptInstance* CScriptInstance::FetchInstance(duk_context* ctx) { for (size_t i = 0; i < Cache.size(); i++) { if (Cache[i]->DukContext() == ctx) { return Cache[i]; } } return NULL; } CScriptInstance::CScriptInstance(CDebuggerUI* debugger) { m_Debugger = debugger; m_Ctx = duk_create_heap_default(); m_ScriptSystem = m_Debugger->ScriptSystem(); m_NextListenerId = 0; m_hIOCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0); InitializeCriticalSection(&m_CriticalSection); CacheInstance(this); } CScriptInstance::~CScriptInstance() { UncacheInstance(this); DeleteCriticalSection(&m_CriticalSection); duk_destroy_heap(m_Ctx); TerminateThread(m_hThread, 0); CloseHandle(m_hThread); } void CScriptInstance::Start(char* path) { m_TempPath = path; CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)StartThread, this, 0, NULL); } void CScriptInstance::ForceStop() { // Close all files and delete all hooked callbacks EnterCriticalSection(&m_CriticalSection); CleanUp(); LeaveCriticalSection(&m_CriticalSection); SetState(STATE_STOPPED); } duk_context* CScriptInstance::DukContext() { return m_Ctx; } INSTANCE_STATE CScriptInstance::GetState() { return m_State; } void CScriptInstance::SetState(INSTANCE_STATE state) { m_State = state; StateChanged(); } void CScriptInstance::StateChanged() { // todo mutex might be needed here //m_Debugger->Debug_RefreshScriptsWindow(); //m_ScriptSystem->DeleteStoppedInstances(); } DWORD CALLBACK CScriptInstance::StartThread(CScriptInstance* _this) { _this->StartScriptProc(); return 0; } void CScriptInstance::StartScriptProc() { DuplicateHandle(GetCurrentProcess(), GetCurrentThread(), GetCurrentProcess(), &m_hThread, 0, FALSE, DUPLICATE_SAME_ACCESS); SetState(STATE_STARTED); bool bWasUnpaused = false; if (!g_Settings->LoadBool(GameRunning_CPU_Paused) && g_MMU) { // Pause CPU during startup phase bWasUnpaused = true; g_System->ExternalEvent(SysEvent_PauseCPU_AppLostFocus); } duk_context* ctx = m_Ctx; duk_push_object(ctx); duk_put_global_string(ctx, "_native"); duk_get_global_string(ctx, "_native"); duk_put_function_list(ctx, -1, NativeFunctions); duk_pop(ctx); const char* apiScript = m_ScriptSystem->APIScript(); duk_int_t apiresult = duk_peval_string(ctx, apiScript); if (apiresult != 0) { MessageBox(NULL, duk_safe_to_string(ctx, -1), "API Script Error", MB_OK | MB_ICONERROR); return; } if (m_TempPath) { stdstr fullPath = stdstr_f("Scripts/%s", m_TempPath); duk_int_t scriptresult = duk_peval_file(ctx, fullPath.c_str()); m_TempPath = NULL; if (scriptresult != 0) { const char* errorText = duk_safe_to_string(ctx, -1); //MessageBox(NULL, duk_safe_to_string(ctx, -1), "Script error", MB_OK | MB_ICONWARNING); m_Debugger->Debug_LogScriptsWindow(errorText); m_Debugger->Debug_LogScriptsWindow("\r\n"); SetState(STATE_STOPPED); return; } } if (bWasUnpaused) { g_System->ExternalEvent(SysEvent_ResumeCPU_AppGainedFocus); } if (HaveEvents()) { StartEventLoop(); } else { CleanUp(); SetState(STATE_STOPPED); } } void CScriptInstance::CleanUp() { m_ScriptSystem->ClearCallbacksForInstance(this); CloseAllAsyncFiles(); CloseAllFiles(); } void CScriptInstance::StartEventLoop() { SetState(STATE_RUNNING); // Todo interrupt with an apc when an event is removed and event count is 0 while (HaveEvents()) { IOLISTENER* lpListener; EVENT_STATUS status = WaitForEvent(&lpListener); if (status == EVENT_STATUS_INTERRUPTED) { continue; } if (status == EVENT_STATUS_ERROR) { break; } InvokeListenerCallback(lpListener); RemoveListener(lpListener); } CleanUp(); SetState(STATE_STOPPED); } CScriptInstance::EVENT_STATUS CScriptInstance::WaitForEvent(IOLISTENER** lpListener) { DWORD nBytesTransferred; ULONG_PTR completionKey; OVERLAPPED* lpUsedOverlap; BOOL status = GetQueuedCompletionStatus( m_hIOCompletionPort, &nBytesTransferred, &completionKey, &lpUsedOverlap, INFINITE ); if (!status) { *lpListener = NULL; DWORD errorCode = GetLastError(); if (errorCode == STATUS_USER_APC) { return EVENT_STATUS_INTERRUPTED; } return EVENT_STATUS_ERROR; } *lpListener = (IOLISTENER*)lpUsedOverlap; (*lpListener)->dataLen = nBytesTransferred; return EVENT_STATUS_OK; } bool CScriptInstance::HaveEvents() { return (m_Listeners.size() > 0) || m_ScriptSystem->HasCallbacksForInstance(this); } void CScriptInstance::AddAsyncFile(HANDLE fd, bool bSocket) { IOFD iofd; iofd.fd = fd; iofd.iocp = CreateIoCompletionPort(fd, m_hIOCompletionPort, (ULONG_PTR)fd, 0); iofd.bSocket = bSocket; m_AsyncFiles.push_back(iofd); } void CScriptInstance::CloseAsyncFile(HANDLE fd) { // Causes EVT_READ with length 0 // Not safe to remove listeners until then for (uint32_t i = 0; i < m_AsyncFiles.size(); i++) { IOFD iofd = m_AsyncFiles[i]; if (iofd.fd != fd) { continue; } // Close file handle if (iofd.bSocket) { closesocket((SOCKET)iofd.fd); } else { CloseHandle(iofd.fd); } break; } } void CScriptInstance::CloseAllAsyncFiles() { // Causes EVT_READ with length 0 // Not safe to remove listeners until then for (uint32_t i = 0; i < m_AsyncFiles.size(); i++) { IOFD iofd = m_AsyncFiles[i]; // Close file handle if (iofd.bSocket) { closesocket((SOCKET)iofd.fd); } else { CloseHandle(iofd.fd); } } } void CScriptInstance::RemoveAsyncFile(HANDLE fd) { // Stop tracking an fd and remove all of its listeners for (uint32_t i = 0; i < m_AsyncFiles.size(); i++) { IOFD iofd = m_AsyncFiles[i]; if (iofd.fd != fd) { continue; } m_AsyncFiles.erase(m_AsyncFiles.begin() + i); RemoveListenersByFd(fd); break; } } HANDLE CScriptInstance::CreateSocket() { HANDLE fd = (HANDLE)WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED); AddAsyncFile(fd, true); return fd; } CScriptInstance::IOLISTENER* CScriptInstance::AddListener(HANDLE fd, IOEVENTTYPE evt, void* callback, void* data, int dataLen) { IOLISTENER* lpListener = (IOLISTENER*)malloc(sizeof(IOLISTENER)); memset(lpListener, 0x00, sizeof(IOLISTENER)); lpListener->id = m_NextListenerId++; lpListener->eventType = evt; lpListener->fd = fd; lpListener->callback = callback; lpListener->data = data; lpListener->dataLen = dataLen; m_Listeners.push_back(lpListener); return lpListener; } void CScriptInstance::RemoveListenerByIndex(UINT index) { IOLISTENER* lpListener = m_Listeners[index]; if (lpListener->data != NULL) { free(lpListener->data); } CancelIoEx(lpListener->fd, (LPOVERLAPPED)lpListener); free(lpListener); m_Listeners.erase(m_Listeners.begin() + index); } // Free listener & its buffer, remove from list void CScriptInstance::RemoveListener(IOLISTENER* lpListener) { for (UINT i = 0; i < m_Listeners.size(); i++) { if (m_Listeners[i] == lpListener) { RemoveListenerByIndex(i); break; } } } void CScriptInstance::RemoveListenersByFd(HANDLE fd) { for (UINT i = 0; i < m_Listeners.size(); i++) { if (m_Listeners[i]->fd == fd) { RemoveListenerByIndex(i); } } } void CScriptInstance::InvokeListenerCallback(IOLISTENER* lpListener) { if (lpListener->callback == NULL) { return; } EnterCriticalSection(&m_CriticalSection); duk_push_heapptr(m_Ctx, lpListener->callback); int nargs = 0; switch (lpListener->eventType) { case EVENT_READ: nargs = 1; if (lpListener->dataLen > 0) { void* data = duk_push_buffer(m_Ctx, lpListener->dataLen, false); memcpy(data, lpListener->data, lpListener->dataLen); } else { // handle must have closed, safe to untrack fd and remove all associated listeners RemoveAsyncFile(lpListener->fd); // pass null to callback duk_push_null(m_Ctx); } break; case EVENT_WRITE: nargs = 1; duk_push_uint(m_Ctx, lpListener->dataLen); // num bytes written break; case EVENT_ACCEPT: // pass client socket fd to callback nargs = 1; duk_push_uint(m_Ctx, (UINT)lpListener->childFd); break; case EVENT_CONNECT: nargs = 0; break; } duk_int_t status = duk_pcall(m_Ctx, nargs); if (status != DUK_EXEC_SUCCESS) { const char* msg = duk_safe_to_string(m_Ctx, -1); MessageBox(NULL, msg, "Script error", MB_OK | MB_ICONWARNING); } LeaveCriticalSection(&m_CriticalSection); } const char* CScriptInstance::Eval(const char* jsCode) { int result = duk_peval_string(m_Ctx, jsCode); const char* msg = duk_safe_to_string(m_Ctx, -1); if (result != 0) { MessageBox(NULL, msg, "Script error", MB_OK | MB_ICONWARNING); } duk_pop(m_Ctx); return msg; } class PendingEval { char* m_CodeCopy; CScriptInstance* m_ScriptInstance; public: PendingEval(CScriptInstance* instance, const char* jsCode) { m_CodeCopy = (char*)malloc(strlen(jsCode)); m_ScriptInstance = instance; MessageBox(NULL, "pending eval created", "", MB_OK); } ~PendingEval() { MessageBox(NULL, "pending eval deleted", "", MB_OK); free(m_CodeCopy); } void Run() { MessageBox(NULL, "pending eval invoked", "", MB_OK); m_ScriptInstance->Eval(m_CodeCopy); } }; void CScriptInstance::EvalAsync(const char* jsCode) { PendingEval* pendingEval = new PendingEval(this, jsCode); QueueAPC(EvalAsyncCallback, (ULONG_PTR)pendingEval); } void CALLBACK CScriptInstance::EvalAsyncCallback(ULONG_PTR _pendingEval) { PendingEval* evalWait = (PendingEval*)_pendingEval; evalWait->Run(); delete evalWait; } bool CScriptInstance::AddFile(const char* path, const char* mode, int* fd) { FILE* fp = fopen(path, mode); if (fp == NULL) { return false; } FILE_FD filefd; filefd.fp = fp; filefd.fd = _fileno(fp); m_Files.push_back(filefd); *fd = filefd.fd; return true; } void CScriptInstance::CloseFile(int fd) { size_t nFiles = m_Files.size(); for (size_t i = 0; i < nFiles; i++) { FILE_FD* filefd = &m_Files[i]; if (filefd->fd == fd) { fclose(filefd->fp); m_Files.erase(m_Files.begin() + i); return; } } } void CScriptInstance::CloseAllFiles() { size_t nFiles = m_Files.size(); for (size_t i = 0; i < nFiles; i++) { fclose(m_Files[i].fp); m_Files.erase(m_Files.begin() + i); } } FILE* CScriptInstance::GetFilePointer(int fd) { size_t nFiles = m_Files.size(); for (size_t i = 0; i < nFiles; i++) { FILE_FD* filefd = &m_Files[i]; if (filefd->fd == fd) { return filefd->fp; } } return NULL; } const char* CScriptInstance::EvalFile(const char* jsPath) { int result = duk_peval_file(m_Ctx, jsPath); const char* msg = duk_safe_to_string(m_Ctx, -1); if (result != 0) { MessageBox(NULL, msg, jsPath, MB_OK | MB_ICONWARNING); } duk_pop(m_Ctx); return msg; } void CScriptInstance::Invoke(void* heapptr, uint32_t param) { EnterCriticalSection(&m_CriticalSection); duk_push_heapptr(m_Ctx, heapptr); duk_push_uint(m_Ctx, param); duk_int_t status = duk_pcall(m_Ctx, 1); if (status != DUK_EXEC_SUCCESS) { const char* errorText = duk_safe_to_string(m_Ctx, -1); m_Debugger->Debug_LogScriptsWindow(errorText); m_Debugger->Debug_LogScriptsWindow("\r\n"); } duk_pop(m_Ctx); LeaveCriticalSection(&m_CriticalSection); } void CScriptInstance::Invoke2(void* heapptr, uint32_t param, uint32_t param2) { EnterCriticalSection(&m_CriticalSection); duk_push_heapptr(m_Ctx, heapptr); duk_push_uint(m_Ctx, param); duk_push_uint(m_Ctx, param2); duk_int_t status = duk_pcall(m_Ctx, 2); if (status != DUK_EXEC_SUCCESS) { const char* errorText = duk_safe_to_string(m_Ctx, -1); m_Debugger->Debug_LogScriptsWindow(errorText); m_Debugger->Debug_LogScriptsWindow("\r\n"); } duk_pop(m_Ctx); LeaveCriticalSection(&m_CriticalSection); } void CScriptInstance::QueueAPC(PAPCFUNC userProc, ULONG_PTR param) { if (m_hThread != NULL) { MessageBox(NULL, "apc queued", "", MB_OK); QueueUserAPC(userProc, m_hThread, param); } } /****************************/ duk_ret_t CScriptInstance::js_ioSockConnect(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); HANDLE fd = (HANDLE)duk_get_uint(ctx, 0); const char* ipStr = duk_to_string(ctx, 1); USHORT port = (USHORT)duk_get_uint(ctx, 2); void* callback = duk_get_heapptr(ctx, 3); char ipBytes[sizeof(uint32_t)]; sscanf(ipStr, "%hhu.%hhu.%hhu.%hhu", &ipBytes[0], &ipBytes[1], &ipBytes[2], &ipBytes[3]); SOCKADDR_IN serverAddr; serverAddr.sin_addr.s_addr = *(uint32_t*)ipBytes; serverAddr.sin_port = htons(port); serverAddr.sin_family = AF_INET; SOCKADDR_IN clientAddr; clientAddr.sin_addr.s_addr = INADDR_ANY; clientAddr.sin_port = 0; clientAddr.sin_family = AF_INET; ::bind((SOCKET)fd, (SOCKADDR*)&clientAddr, sizeof(SOCKADDR)); IOLISTENER* lpListener = _this->AddListener(fd, EVENT_CONNECT, callback); ConnectEx((SOCKET)fd, (SOCKADDR*)&serverAddr, sizeof(SOCKADDR), NULL, 0, NULL, (LPOVERLAPPED)lpListener); duk_pop_n(ctx, 4); return 1; } duk_ret_t CScriptInstance::js_ioClose(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); HANDLE fd = (HANDLE)duk_get_uint(ctx, 0); duk_pop(ctx); _this->CloseAsyncFile(fd); return 1; } duk_ret_t CScriptInstance::js_ioSockCreate(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); HANDLE fd = _this->CreateSocket(); duk_pop_n(ctx, duk_get_top(ctx)); duk_push_int(ctx, (int)fd); return 1; } duk_ret_t CScriptInstance::js_ioSockListen(duk_context* ctx) { HANDLE fd = (HANDLE)duk_get_uint(ctx, 0); USHORT port = (USHORT)duk_get_uint(ctx, 1); duk_pop_n(ctx, 2); SOCKADDR_IN serverAddr; serverAddr.sin_family = AF_INET; serverAddr.sin_addr.s_addr = INADDR_ANY; serverAddr.sin_port = htons(port); if (::bind((SOCKET)fd, (SOCKADDR*)&serverAddr, sizeof(serverAddr))) { duk_push_boolean(ctx, false); return 1; } bool listenOkay = listen((SOCKET)fd, 3) == 0; duk_push_boolean(ctx, listenOkay); return 1; } duk_ret_t CScriptInstance::js_ioSockAccept(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); HANDLE fd = (HANDLE)duk_get_uint(ctx, 0); void* jsCallback = duk_get_heapptr(ctx, 1); void* data = malloc(sizeof(SOCKADDR) * 4); // issue? IOLISTENER* lpListener = _this->AddListener(fd, EVENT_ACCEPT, jsCallback, data, 0); lpListener->childFd = _this->CreateSocket(); static DWORD unused; int ok = AcceptEx( (SOCKET)fd, (SOCKET)lpListener->childFd, lpListener->data, // local and remote SOCKADDR 0, sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, NULL, (LPOVERLAPPED)lpListener ); duk_pop_n(ctx, 2); if (!ok) { duk_push_boolean(ctx, false); } else { duk_push_boolean(ctx, true); } return 1; } duk_ret_t CScriptInstance::js_ioRead(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); // (fd, bufferSize, callback) HANDLE fd = (HANDLE)duk_get_uint(ctx, 0); size_t bufferSize = duk_get_uint(ctx, 1); void* jsCallback = duk_get_heapptr(ctx, 2); void* data = malloc(bufferSize); // freed after event is fired IOLISTENER* lpListener = _this->AddListener(fd, EVENT_READ, jsCallback, data, bufferSize); BOOL status = ReadFile(fd, lpListener->data, lpListener->dataLen, NULL, (LPOVERLAPPED)lpListener); if (status == false && GetLastError() != ERROR_IO_PENDING) { MessageBox(NULL, "readex error", "", MB_OK); } duk_pop_n(ctx, 3); return 1; } duk_ret_t CScriptInstance::js_ioWrite(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); HANDLE fd = (HANDLE)duk_get_uint(ctx, 0); duk_size_t dataLen; void* jsData = duk_to_buffer(ctx, 1, &dataLen); void* jsCallback = duk_get_heapptr(ctx, 2); char* data = (char*)malloc(dataLen + 1); // freed after event is fired memcpy(data, jsData, dataLen); data[dataLen] = '\0'; IOLISTENER* lpListener = _this->AddListener(fd, EVENT_WRITE, jsCallback, data, dataLen); WriteFile(fd, lpListener->data, lpListener->dataLen, NULL, (LPOVERLAPPED)lpListener); duk_pop_n(ctx, 3); return 1; } duk_ret_t CScriptInstance::js_AddCallback(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); const char* hookId; void* heapptr; uint32_t param = 0; uint32_t param2 = 0; uint32_t param3 = 0; uint32_t param4 = 0; bool bOnce = false; int argc = duk_get_top(ctx); hookId = duk_get_string(ctx, 0); heapptr = duk_get_heapptr(ctx, 1); if (argc >= 3) param = duk_get_uint(ctx, 2); if (argc >= 4) param2 = duk_get_uint(ctx, 3); if (argc >= 5) param3 = duk_get_uint(ctx, 4); if (argc >= 6) param4 = duk_get_uint(ctx, 5); if (argc >= 7) bOnce = (duk_get_boolean(ctx, 6) != 0); int callbackId = -1; CScriptHook* hook = _this->m_ScriptSystem->GetHook(hookId); if (hook != NULL) { callbackId = hook->Add(_this, heapptr, param, param2, param3, param4, bOnce); } duk_pop_n(ctx, argc); duk_push_int(ctx, callbackId); return 1; } duk_ret_t CScriptInstance::js_RemoveCallback(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); int callbackId = duk_get_int(ctx, 0); _this->m_ScriptSystem->RemoveCallbackById(callbackId); duk_pop(ctx); return 1; } duk_ret_t CScriptInstance::js_GetPCVal(duk_context* ctx) { duk_push_uint(ctx, g_Reg->m_PROGRAM_COUNTER); return 1; } duk_ret_t CScriptInstance::js_SetPCVal(duk_context* ctx) { uint32_t val = duk_to_uint32(ctx, 0); g_Reg->m_PROGRAM_COUNTER = val; duk_pop_n(ctx, 1); return 1; } duk_ret_t CScriptInstance::js_GetLOVal(duk_context* ctx) { bool bUpper = duk_get_boolean(ctx, 0) != 0; duk_pop_n(ctx, 1); if (!bUpper) { duk_push_uint(ctx, g_Reg->m_LO.UW[0]); } else { duk_push_uint(ctx, g_Reg->m_LO.UW[1]); } return 1; } duk_ret_t CScriptInstance::js_SetLOVal(duk_context* ctx) { bool bUpper = duk_get_boolean(ctx, 0) != 0; uint32_t val = duk_to_uint32(ctx, 1); duk_pop_n(ctx, 2); if (!bUpper) { g_Reg->m_LO.UW[0] = val; } else { g_Reg->m_LO.UW[1] = val; } return 1; } duk_ret_t CScriptInstance::js_GetHIVal(duk_context* ctx) { bool bUpper = duk_get_boolean(ctx, 0) != 0; duk_pop_n(ctx, 1); if (!bUpper) { duk_push_uint(ctx, g_Reg->m_HI.UW[0]); } else { duk_push_uint(ctx, g_Reg->m_HI.UW[1]); } return 1; } duk_ret_t CScriptInstance::js_SetHIVal(duk_context* ctx) { bool bUpper = duk_get_boolean(ctx, 0) != 0; uint32_t val = duk_to_uint32(ctx, 1); duk_pop_n(ctx, 2); if (!bUpper) { g_Reg->m_HI.UW[0] = val; } else { g_Reg->m_HI.UW[1] = val; } return 1; } duk_ret_t CScriptInstance::js_GetGPRVal(duk_context* ctx) { int nargs = duk_get_top(ctx); int regnum = duk_to_int(ctx, 0); bool bUpper = false; if (nargs > 1) { bUpper = duk_to_boolean(ctx, 1) != 0; } duk_pop_n(ctx, nargs); if (!bUpper) { duk_push_uint(ctx, g_Reg->m_GPR[regnum].UW[0]); } else { duk_push_uint(ctx, g_Reg->m_GPR[regnum].UW[1]); } return 1; } duk_ret_t CScriptInstance::js_SetGPRVal(duk_context* ctx) { int regnum = duk_to_int(ctx, 0); bool bUpper = duk_to_boolean(ctx, 1) != 0; uint32_t val = duk_to_uint32(ctx, 2); if (!bUpper) { g_Reg->m_GPR[regnum].UW[0] = val; } else { g_Reg->m_GPR[regnum].UW[1] = val; } duk_pop_n(ctx, 3); return 1; } duk_ret_t CScriptInstance::js_GetFPRVal(duk_context* ctx) { int nargs = duk_get_top(ctx); int regnum = duk_to_int(ctx, 0); bool bDouble = false; if (nargs > 1) { bDouble = duk_get_boolean(ctx, 1) != 0; } duk_pop_n(ctx, nargs); if (!bDouble) { duk_push_number(ctx, (duk_double_t)*g_Reg->m_FPR_S[regnum]); } else { duk_push_number(ctx, (duk_double_t)*g_Reg->m_FPR_D[regnum & 0x1E]); } return 1; } duk_ret_t CScriptInstance::js_SetFPRVal(duk_context* ctx) { int regnum = duk_to_int(ctx, 0); bool bDouble = duk_to_boolean(ctx, 1) != 0; duk_double_t val = duk_to_number(ctx, 2); if (!bDouble) { *g_Reg->m_FPR_S[regnum] = (float)val; } else { *g_Reg->m_FPR_D[regnum & 0x1E] = val; } duk_pop_n(ctx, 3); return 1; } duk_ret_t CScriptInstance::js_GetCauseVal(duk_context* ctx) { duk_push_uint(ctx, g_Reg->FAKE_CAUSE_REGISTER); return 1; } duk_ret_t CScriptInstance::js_SetCauseVal(duk_context* ctx) { uint32_t val = duk_to_uint32(ctx, 0); g_Reg->FAKE_CAUSE_REGISTER = val; g_Reg->CheckInterrupts(); duk_pop_n(ctx, 1); return 1; } duk_ret_t CScriptInstance::js_GetROMInt(duk_context* ctx) { uint32_t address = duk_to_uint32(ctx, 0) & 0x0FFFFFFF; int bitwidth = duk_to_int(ctx, 1); duk_bool_t bSigned = duk_to_boolean(ctx, 2); duk_pop_n(ctx, 3); if (g_Rom == NULL) { goto return_err; } uint8_t* rom = g_Rom->GetRomAddress(); // little endian uint32_t romSize = g_Rom->GetRomSize(); if (address > romSize) { goto return_err; } DWORD retval; switch (bitwidth) { case 8: { uint8_t val = rom[address ^ 0b11]; retval = bSigned ? (char)val : val; break; } case 16: { uint16_t val = *(uint16_t*)&rom[address ^ 0b10]; retval = bSigned ? (short)val : val; break; } case 32: { uint32_t val = *(uint32_t*)&rom[address]; retval = bSigned ? (int)val : val; break; } default: goto return_err; } if (bSigned) { duk_push_int(ctx, retval); } else { duk_push_uint(ctx, retval); } return 1; return_err: duk_push_boolean(ctx, false); return 1; } duk_ret_t CScriptInstance::js_GetROMFloat(duk_context* ctx) { int argc = duk_get_top(ctx); uint32_t address = duk_to_uint32(ctx, 0); duk_bool_t bDouble = false; if (argc > 1) { bDouble = duk_get_boolean(ctx, 1); } duk_pop_n(ctx, argc); if (g_Rom == NULL) { goto return_err; } uint8_t* rom = g_Rom->GetRomAddress(); // little endian uint32_t romSize = g_Rom->GetRomSize(); if (address > romSize) { goto return_err; } if (!bDouble) { float value = *(float*)&rom[address]; duk_push_number(ctx, (double)value); return 1; } uint8_t raw[8]; memcpy(&raw[0], &rom[address + 4], 4); memcpy(&raw[4], &rom[address], 4); double value; memcpy(&value, raw, 8); duk_push_number(ctx, value); return 1; return_err: duk_push_boolean(ctx, false); return 1; } duk_ret_t CScriptInstance::js_GetRDRAMInt(duk_context* ctx) { uint32_t address = duk_to_uint32(ctx, 0); int bitwidth = duk_to_int(ctx, 1); duk_bool_t bSigned = duk_to_boolean(ctx, 2); duk_pop_n(ctx, 3); if (g_MMU == NULL) { goto return_err; } DWORD retval; switch (bitwidth) { case 8: { uint8_t val; if (!g_MMU->LB_VAddr(address, val)) { goto return_err; } retval = bSigned ? (char)val : val; break; } case 16: { uint16_t val; if (!g_MMU->LH_VAddr(address, val)) { goto return_err; } retval = bSigned ? (short)val : val; break; } case 32: { uint32_t val; if (!g_MMU->LW_VAddr(address, val)) { goto return_err; } retval = bSigned ? (int)val : val; break; } default: goto return_err; } if (bSigned) { duk_push_int(ctx, retval); } else { duk_push_uint(ctx, retval); } return 1; return_err: duk_push_boolean(ctx, false); return 1; } duk_ret_t CScriptInstance::js_SetRDRAMInt(duk_context* ctx) { uint32_t address = duk_to_uint32(ctx, 0); int bitwidth = duk_to_int(ctx, 1); DWORD newValue = duk_to_uint32(ctx, 2); duk_pop_n(ctx, 3); if (g_MMU == NULL) { goto return_err; } switch (bitwidth) { case 8: if (!g_MMU->SB_VAddr(address, (uint8_t)newValue)) { goto return_err; } goto return_ok; case 16: if (!g_MMU->SH_VAddr(address, (uint16_t)newValue)) { goto return_err; } goto return_ok; case 32: if (!g_MMU->SW_VAddr(address, (uint32_t)newValue)) { goto return_err; } goto return_ok; default: goto return_err; } return_ok: duk_push_boolean(ctx, true); return 1; return_err: duk_push_boolean(ctx, false); return 1; } duk_ret_t CScriptInstance::js_GetRDRAMFloat(duk_context* ctx) { int argc = duk_get_top(ctx); uint32_t address = duk_to_uint32(ctx, 0); duk_bool_t bDouble = false; if (argc > 1) { bDouble = duk_get_boolean(ctx, 1); } duk_pop_n(ctx, argc); if (g_MMU == NULL) { goto return_err; } if (!bDouble) { uint32_t rawValue; if (!g_MMU->LW_VAddr(address, rawValue)) { goto return_err; } float floatValue; memcpy(&floatValue, &rawValue, sizeof(float)); duk_push_number(ctx, (double)floatValue); return 1; } uint64_t rawValue; if (!g_MMU->LD_VAddr(address, rawValue)) { goto return_err; } double floatValue; memcpy(&floatValue, &rawValue, sizeof(double)); duk_push_number(ctx, floatValue); return 1; return_err: duk_push_boolean(ctx, false); return 1; } duk_ret_t CScriptInstance::js_SetRDRAMFloat(duk_context* ctx) { int argc = duk_get_top(ctx); uint32_t address = duk_to_uint32(ctx, 0); double value = duk_get_number(ctx, 1); duk_bool_t bDouble = false; if (argc > 2) { bDouble = duk_get_boolean(ctx, 2); } duk_pop_n(ctx, argc); if (g_MMU == NULL) { goto return_err; } if (!bDouble) { float floatValue = (float)value; uint32_t rawValue; memcpy(&rawValue, &floatValue, sizeof(float)); if (!g_MMU->SW_VAddr(address, rawValue)) { goto return_err; } } else { double floatValue = (double)value; uint64_t rawValue; memcpy(&rawValue, &floatValue, sizeof(double)); if (!g_MMU->SD_VAddr(address, rawValue)) { goto return_err; } } duk_push_boolean(ctx, true); return 1; return_err: duk_push_boolean(ctx, false); return 1; } duk_ret_t CScriptInstance::js_GetRDRAMBlock(duk_context* ctx) { uint32_t address = duk_get_uint(ctx, 0); uint32_t size = duk_get_uint(ctx, 1); duk_pop_n(ctx, 2); if (g_MMU == NULL) { duk_push_boolean(ctx, false); return 1; } uint8_t* block = (uint8_t*)duk_push_buffer(ctx, size, false); for (uint32_t i = 0; i < size; i++) { g_MMU->LB_VAddr(address + i, block[i]); } return 1; } duk_ret_t CScriptInstance::js_MsgBox(duk_context* ctx) { int argc = duk_get_top(ctx); const char* msg = duk_to_string(ctx, 0); const char* caption = ""; if (argc > 1) { caption = duk_to_string(ctx, 1); } MessageBox(NULL, msg, caption, MB_OK); duk_pop_n(ctx, argc); duk_push_boolean(ctx, 1); return 1; } // Return zero-terminated string from ram duk_ret_t CScriptInstance::js_GetRDRAMString(duk_context* ctx) { // (address[, maxLen]) int nargs = duk_get_top(ctx); uint32_t address = duk_get_uint(ctx, 0); int maxLen = INT_MAX; if (nargs > 1) { maxLen = duk_get_int(ctx, 1); if (maxLen == 0) { maxLen = INT_MAX; } } uint8_t test = 0xFF; int len = 0; // determine length of string while (len < maxLen && g_MMU->LB_VAddr(address + len, test) && test != 0) // todo protect from ram overrun { if ((address & 0xFFFFFF) + len >= g_MMU->RdramSize()) { break; } len++; } uint8_t* str = (uint8_t*)malloc(len + 1); for (int i = 0; i < len; i++) { g_MMU->LB_VAddr(address + i, str[i]); } str[len] = '\0'; duk_pop_n(ctx, nargs); duk_push_string(ctx, (char*)str); free(str); // duk creates internal copy return 1; } // Return zero-terminated string from rom duk_ret_t CScriptInstance::js_GetROMString(duk_context* ctx) { // (address[, maxLen]) int nargs = duk_get_top(ctx); uint32_t address = duk_get_uint(ctx, 0); int maxLen = INT_MAX; if (nargs > 1) { maxLen = duk_get_int(ctx, 1); if (maxLen == 0) { maxLen = INT_MAX; } } char* rom = (char*)g_Rom->GetRomAddress(); uint32_t romSize = g_Rom->GetRomSize(); int len = 0; while (len < maxLen) { if (address + len >= romSize) { break; } if (rom[address + len] == 0) { break; } len++; } char* str = (char*)malloc(len + 1); for (int i = 0; i < len; i++) { str[i] = rom[(address + i) ^ 3]; } str[len] = '\0'; duk_pop(ctx); duk_push_string(ctx, str); free(str); // duk creates internal copy return 1; } // Return zero-terminated string from rom duk_ret_t CScriptInstance::js_GetROMBlock(duk_context* ctx) { uint32_t address = duk_get_uint(ctx, 0); uint32_t size = duk_get_uint(ctx, 1); duk_pop_n(ctx, 2); uint8_t* block = (uint8_t*)duk_push_buffer(ctx, size, false); uint8_t* rom = g_Rom->GetRomAddress(); memcpy(block, &rom[address], size); return 1; } duk_ret_t CScriptInstance::js_ConsolePrint(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); const char* text = duk_to_string(ctx, 0); _this->m_Debugger->Debug_LogScriptsWindow(text); duk_pop_n(ctx, 1); return 1; } duk_ret_t CScriptInstance::js_ConsoleClear(duk_context* ctx) { FetchInstance(ctx)->m_Debugger->Debug_ClearScriptsWindow(); return 1; } duk_ret_t CScriptInstance::js_BreakHere(duk_context* /*ctx*/) { g_Settings->SaveBool(Debugger_SteppingOps, true); return 1; } duk_ret_t CScriptInstance::js_Pause(duk_context* /*ctx*/) { g_System->Pause(); return 1; } duk_ret_t CScriptInstance::js_ShowCommands(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); int nargs = duk_get_top(ctx); uint32_t address = 0x80000000; if (nargs == 1) { address = duk_get_uint(ctx, 0); } duk_pop_n(ctx, nargs); _this->m_Debugger->Debug_ShowCommandsLocation(address, false); return 1; } duk_ret_t CScriptInstance::js_ScreenPrint(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); HDC hdc = _this->m_ScriptSystem->GetScreenDC(); int nargs = duk_get_top(ctx); if (nargs != 3) { duk_pop_n(ctx, nargs); return 1; } int x = duk_to_int(ctx, 0); int y = duk_to_int(ctx, 1); const char* text = duk_to_string(ctx, 2); int nChars = strlen(text); TextOut(hdc, x, y, text, nChars); duk_pop_n(ctx, nargs); return 1; } duk_ret_t CScriptInstance::js_FSOpen(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); int nargs = duk_get_top(ctx); if (nargs < 2) { duk_pop_n(ctx, nargs); duk_push_false(ctx); return 1; } const char* path = duk_to_string(ctx, 0); const char* mode = duk_to_string(ctx, 1); int fd; bool res = _this->AddFile(path, mode, &fd); duk_pop_n(ctx, nargs); if (res == false) { duk_push_false(ctx); return 1; } duk_push_number(ctx, fd); return 1; } duk_ret_t CScriptInstance::js_FSClose(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); int nargs = duk_get_top(ctx); if (nargs < 1) { duk_pop_n(ctx, nargs); return 1; } int fd = duk_get_int(ctx, 0); duk_pop_n(ctx, nargs); _this->CloseFile(fd); return 1; } duk_ret_t CScriptInstance::js_FSWrite(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); int nargs = duk_get_top(ctx); size_t nBytesWritten = 0; if (nargs < 2) { goto end; } int fd = duk_get_int(ctx, 0); FILE* fp = _this->GetFilePointer(fd); if (fp == NULL) { goto end; } const char* buffer; int offset = 0; duk_size_t length = 0; int position = 0; // fseek if (duk_is_string(ctx, 1)) { // string const char* str = duk_get_string(ctx, 1); length = strlen(str); buffer = str; } else { // buffer buffer = (const char*)duk_get_buffer_data(ctx, 1, &length); if (buffer == NULL) { goto end; } } if (nargs > 2 && duk_get_type(ctx, 2) == DUK_TYPE_NUMBER) { offset = duk_get_int(ctx, 2); length -= offset; } if (nargs > 3 && duk_get_type(ctx, 3) == DUK_TYPE_NUMBER) { size_t lengthArg = duk_get_uint(ctx, 3); if (lengthArg <= length) { length = lengthArg; } } if (nargs > 4 && duk_get_type(ctx, 4) == DUK_TYPE_NUMBER) { position = duk_get_int(ctx, 4); fseek(fp, position, SEEK_SET); } nBytesWritten = fwrite(buffer, 1, length, fp); end: duk_pop_n(ctx, nargs); duk_push_number(ctx, nBytesWritten); return 1; } duk_ret_t CScriptInstance::js_FSRead(duk_context* ctx) { CScriptInstance* _this = FetchInstance(ctx); int nargs = duk_get_top(ctx); size_t nBytesRead = 0; // (fd, buffer, offset, length, position) if (nargs < 5) { goto end; } duk_size_t bufferSize; int fd = duk_get_int(ctx, 0); char* buffer = (char*)duk_get_buffer_data(ctx, 1, &bufferSize); duk_size_t offset = duk_get_uint(ctx, 2); duk_size_t length = duk_get_uint(ctx, 3); duk_size_t position = duk_get_uint(ctx, 4); if (bufferSize == 0) { goto end; } FILE* fp = _this->GetFilePointer(fd); if (fp == NULL) { goto end; } if (offset + length > bufferSize) { length = bufferSize - offset; } fseek(fp, position, SEEK_SET); nBytesRead = fread(&buffer[offset], 1, length, fp); end: duk_pop_n(ctx, nargs); duk_push_number(ctx, nBytesRead); return 1; } duk_ret_t CScriptInstance::js_FSFStat(duk_context* ctx) { int nargs = duk_get_top(ctx); if (nargs < 1) { duk_pop_n(ctx, nargs); duk_push_false(ctx); return 1; } int fd = duk_get_int(ctx, 0); struct stat statbuf; int res = fstat(fd, &statbuf); if (res != 0) { duk_pop_n(ctx, nargs); duk_push_false(ctx); return 1; } duk_pop_n(ctx, nargs); duk_idx_t obj_idx = duk_push_object(ctx); const duk_number_list_entry props[] = { { "dev", (duk_double_t)statbuf.st_dev }, { "ino", (duk_double_t)statbuf.st_ino }, { "mode", (duk_double_t)statbuf.st_mode }, { "nlink", (duk_double_t)statbuf.st_nlink }, { "uid", (duk_double_t)statbuf.st_uid }, { "gid", (duk_double_t)statbuf.st_gid }, { "rdev", (duk_double_t)statbuf.st_rdev }, { "size", (duk_double_t)statbuf.st_size }, { "atimeMs", (duk_double_t)statbuf.st_atime * 1000 }, { "mtimeMs", (duk_double_t)statbuf.st_mtime * 1000 }, { "ctimeMs", (duk_double_t)statbuf.st_ctime * 1000 }, { NULL, 0 } }; duk_put_number_list(ctx, obj_idx, props); return 1; } duk_ret_t CScriptInstance::js_FSStat(duk_context* ctx) { int nargs = duk_get_top(ctx); if (nargs < 1) { duk_pop_n(ctx, nargs); duk_push_false(ctx); return 1; } const char* path = duk_get_string(ctx, 0); struct stat statbuf; int res = stat(path, &statbuf); if (res != 0) { duk_pop_n(ctx, nargs); duk_push_false(ctx); return 1; } duk_pop_n(ctx, nargs); duk_idx_t obj_idx = duk_push_object(ctx); const duk_number_list_entry props[] = { { "dev", (duk_double_t)statbuf.st_dev }, { "ino", (duk_double_t)statbuf.st_ino }, { "mode", (duk_double_t)statbuf.st_mode }, { "nlink", (duk_double_t)statbuf.st_nlink }, { "uid", (duk_double_t)statbuf.st_uid }, { "gid", (duk_double_t)statbuf.st_gid }, { "rdev", (duk_double_t)statbuf.st_rdev }, { "size", (duk_double_t)statbuf.st_size }, { "atimeMs", (duk_double_t)statbuf.st_atime * 1000 }, { "mtimeMs", (duk_double_t)statbuf.st_mtime * 1000 }, { "ctimeMs", (duk_double_t)statbuf.st_ctime * 1000 }, { NULL, 0 } }; duk_put_number_list(ctx, obj_idx, props); return 1; } duk_ret_t CScriptInstance::js_FSMkDir(duk_context* ctx) { int nargs = duk_get_top(ctx); if (nargs < 1) { duk_push_false(ctx); return 1; } const char* path = duk_get_string(ctx, 0); duk_pop_n(ctx, nargs); if (CreateDirectory(path, NULL)) { duk_push_true(ctx); } else { duk_push_false(ctx); } return 1; } duk_ret_t CScriptInstance::js_FSRmDir(duk_context* ctx) { int nargs = duk_get_top(ctx); if (nargs < 1) { duk_push_false(ctx); return 1; } const char* path = duk_get_string(ctx, 0); duk_pop_n(ctx, nargs); if (RemoveDirectory(path)) { duk_push_true(ctx); } else { duk_push_false(ctx); } return 1; } duk_ret_t CScriptInstance::js_FSUnlink(duk_context* ctx) { int nargs = duk_get_top(ctx); if (nargs < 1) { duk_push_false(ctx); return 1; } const char* path = duk_get_string(ctx, 0); duk_pop_n(ctx, nargs); if (!_unlink(path)) { duk_push_true(ctx); } else { duk_push_false(ctx); } return 1; } duk_ret_t CScriptInstance::js_FSReadDir(duk_context* ctx) { int nargs = duk_get_top(ctx); if (nargs < 1) { duk_push_false(ctx); return 1; } const char* path = duk_get_string(ctx, 0); duk_pop_n(ctx, nargs); WIN32_FIND_DATA ffd; HANDLE hFind = FindFirstFile(stdstr_f("%s%s", path, "\\*").c_str(), &ffd); if (hFind == INVALID_HANDLE_VALUE) { duk_push_false(ctx); return 1; } duk_idx_t arr_idx = duk_push_array(ctx); int nfile = 0; do { char filename[MAX_PATH]; strcpy(filename, ffd.cFileName); if (strcmp(filename, ".") == 0 || strcmp(filename, "..") == 0) { continue; } duk_push_string(ctx, filename); duk_put_prop_index(ctx, arr_idx, nfile); nfile++; } while (FindNextFile(hFind, &ffd) != 0); FindClose(hFind); return 1; } //////////// static BOOL ConnectEx(SOCKET s, const SOCKADDR* name, int namelen, PVOID lpSendBuffer, DWORD dwSendDataLength, LPDWORD lpdwBytesSent, LPOVERLAPPED lpOverlapped) { LPFN_CONNECTEX ConnectExPtr = NULL; DWORD nBytes; GUID guid = WSAID_CONNECTEX; int fetched = WSAIoctl( s, SIO_GET_EXTENSION_FUNCTION_POINTER, (void*)&guid, sizeof(GUID), &ConnectExPtr, sizeof(LPFN_CONNECTEX), &nBytes, NULL, NULL ); if (fetched == 0 && ConnectExPtr != NULL) { ConnectExPtr(s, name, namelen, lpSendBuffer, dwSendDataLength, lpdwBytesSent, lpOverlapped); } return false; }