#include "std.h"
#include "resource.h"
#include "emul.h"
#include "vars.h"
#include "gui.h"
#include "dx.h"
#include "util.h"
struct CHEATDLG
{
HWND dlg;
HWND resBox;
HWND resLabel;
void ShowResults();
void SetControls();
void Search();
unsigned char *lastsnap;
unsigned char *bitmask;
unsigned searchSize;
unsigned nFound;
unsigned char wordmode, hex;
CHEATDLG() { lastsnap = 0; bitmask = 0; nFound = -1; mode = S_NEW; }
~CHEATDLG() { free(lastsnap); free(bitmask); }
enum SR_MODE { S_NEW, S_VAL, S_INC, S_DEC } mode;
} CheatDlg;
void CHEATDLG::Search()
{
if (nFound == -1) searchSize = conf.ramsize*1024;
searchSize = min(searchSize, conf.ramsize*1024);
bitmask = (unsigned char*)realloc(bitmask, searchSize/8);
if (nFound == -1) memset(bitmask, 0xFF, searchSize/8);
unsigned i;
switch (mode) {
case S_NEW:
{
memset(bitmask, 0xFF, searchSize/8);
nFound = -1;
break;
}
case S_VAL:
{
char str[16]; GetDlgItemText(dlg, IDE_VALUE, str, sizeof(str));
unsigned val; sscanf(str, hex? "%X" : "%d", &val);
if (wordmode) {
for (i = 0; i < searchSize-1; i++)
if (*(WORD*)(memory+i) != (WORD)val)
bitmask[i/8] &= ~(1 << (i & 7));
} else {
for (i = 0; i < searchSize; i++)
if (memory[i] != (BYTE)val)
bitmask[i/8] &= ~(1 << (i & 7));
}
break;
}
case S_INC:
case S_DEC:
{
unsigned char *ptr1, *ptr2;
if (mode == S_INC) ptr1 = memory, ptr2 = lastsnap;
else ptr2 = memory, ptr1 = lastsnap;
if (wordmode) {
for (i = 0; i < searchSize-1; i++)
if (*(WORD*)(ptr1+i) <= *(WORD*)(ptr2+i))
bitmask[i/8] &= ~(1 << (i & 7));
} else {
for (i = 0; i < searchSize; i++)
if (ptr1[i] <= ptr2[i])
bitmask[i/8] &= ~(1 << (i & 7));
}
break;
}
}
if (wordmode) bitmask[(searchSize-1)/8] &= 0x7F;
if (mode != S_NEW) {
for (nFound = i = 0; i < searchSize/8; i++) {
if (!bitmask[i]) continue;
if (bitmask[i] & 0x01) nFound++;
if (bitmask[i] & 0x02) nFound++;
if (bitmask[i] & 0x04) nFound++;
if (bitmask[i] & 0x08) nFound++;
if (bitmask[i] & 0x10) nFound++;
if (bitmask[i] & 0x20) nFound++;
if (bitmask[i] & 0x40) nFound++;
if (bitmask[i] & 0x80) nFound++;
}
}
lastsnap = (unsigned char*)realloc(lastsnap, searchSize);
memcpy(lastsnap, memory, searchSize);
ShowResults();
}
void CHEATDLG::SetControls()
{
int enabled;
setcheck(IDC_HEX, hex);
setcheck(IDC_BYTE, !wordmode);
setcheck(IDC_WORD, wordmode);
enabled = lastsnap && (nFound > 0);
EnableWindow(GetDlgItem(dlg, IDC_DEC), enabled);
EnableWindow(GetDlgItem(dlg, IDC_INC), enabled);
if (!enabled && (mode == S_DEC || mode == S_INC)) mode = S_NEW;
enabled = (nFound > 0);
EnableWindow(GetDlgItem(dlg, IDC_EXACT), enabled);
if (!enabled && mode == S_VAL) mode = S_NEW;
setcheck(IDC_NEW, (mode == S_NEW));
setcheck(IDC_EXACT, (mode == S_VAL));
setcheck(IDC_INC, (mode == S_INC));
setcheck(IDC_DEC, (mode == S_DEC));
enabled = (nFound == -1);// || (mode == S_NEW);
EnableWindow(GetDlgItem(dlg, IDC_BYTE), enabled);
EnableWindow(GetDlgItem(dlg, IDC_WORD), enabled);
enabled = (mode == S_VAL);
EnableWindow(GetDlgItem(dlg, IDE_VALUE), enabled);
EnableWindow(GetDlgItem(dlg, IDC_HEX), enabled);
}
void CHEATDLG::ShowResults()
{
if (nFound > 0 && nFound <= 100) {
ShowWindow(resLabel, SW_HIDE);
ShowWindow(resBox, SW_SHOW);
SendMessage(resBox, LVM_DELETEALLITEMS, 0, 0);
char fn[10];
LVITEM item; int count = 0;
item.mask = LVIF_TEXT;
item.pszText = fn;
for (unsigned i = 0; i < searchSize; i++) {
if (!(bitmask[i/8] & (1 << (i & 7)))) continue;
unsigned base = 0xC000;
unsigned page = i / PAGE;
if (page == 2) base = 0x8000;
if (page == 5) base = 0x4000;
sprintf(fn, "%04X", base + (i & (PAGE-1)));
item.iItem = count++;
item.iSubItem = 0;
item.iItem = SendMessage(resBox, LVM_INSERTITEM, 0, (LPARAM) &item);
sprintf(fn, "%02X", page);
item.iSubItem = 1;
SendMessage(resBox, LVM_SETITEM, 0, (LPARAM) &item);
sprintf(fn, "%04X", i & (PAGE-1));
item.iSubItem = 2;
SendMessage(resBox, LVM_SETITEM, 0, (LPARAM) &item);
if (wordmode) sprintf(fn, "%04X", *(WORD*)(memory+i));
else sprintf(fn, "%02X", memory[i]);
item.iSubItem = 3;
SendMessage(resBox, LVM_SETITEM, 0, (LPARAM) &item);
}
} else {
ShowWindow(resBox, SW_HIDE);
ShowWindow(resLabel, SW_SHOW);
char str[128];
if (!lastsnap) strcpy(str, "no active search");
else if (nFound == -1) strcpy(str, "new search started");
else if (!nFound) strcpy(str, "found nothing");
else sprintf(str, "result list too large (%d entries)", nFound);
SetWindowText(resLabel, str);
}
}
INT_PTR CALLBACK cheatdlg(HWND dlg, UINT msg, WPARAM wp, LPARAM lp)
{
if (msg == WM_INITDIALOG) {
::dlg = CheatDlg.dlg = dlg;
CheatDlg.resLabel = GetDlgItem(dlg, IDC_STATUS);
CheatDlg.resBox = GetDlgItem(dlg, IDC_RESULTS);
unsigned exflags = LVS_EX_HEADERDRAGDROP | LVS_EX_FULLROWSELECT;
SendMessage(CheatDlg.resBox, LVM_SETEXTENDEDLISTVIEWSTYLE, exflags, exflags);
LVCOLUMN sizeCol;
sizeCol.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
sizeCol.fmt = LVCFMT_LEFT;
sizeCol.cx = 50;
sizeCol.pszText = LPSTR("Address");
SendMessage(CheatDlg.resBox, LVM_INSERTCOLUMN, 0, (LPARAM)&sizeCol);
sizeCol.cx = 40;
sizeCol.pszText = LPSTR("Page");
SendMessage(CheatDlg.resBox, LVM_INSERTCOLUMN, 1, (LPARAM)&sizeCol);
sizeCol.cx = 50;
sizeCol.pszText = LPSTR("Offset");
SendMessage(CheatDlg.resBox, LVM_INSERTCOLUMN, 2, (LPARAM)&sizeCol);
sizeCol.cx = 50;
sizeCol.pszText = LPSTR("Value");
SendMessage(CheatDlg.resBox, LVM_INSERTCOLUMN, 3, (LPARAM)&sizeCol);
// SetFocus(GetDlgItem(dlg, IDE_VALUE));
SendDlgItemMessage(dlg, IDE_VALUE, EM_LIMITTEXT, 5, 0);
CheatDlg.SetControls();
CheatDlg.ShowResults();
return 1;
}
if ((msg == WM_COMMAND && wp == IDCANCEL) ||
(msg == WM_SYSCOMMAND && (wp & 0xFFF0) == SC_CLOSE)) EndDialog(dlg, 0);
if (msg == WM_COMMAND && HIWORD(wp) == BN_CLICKED)
{
DWORD id = LOWORD(wp);
if (id == IDC_NEW) CheatDlg.mode = CHEATDLG::S_NEW;
else if (id == IDC_EXACT) CheatDlg.mode = CHEATDLG::S_VAL;
else if (id == IDC_INC) CheatDlg.mode = CHEATDLG::S_INC;
else if (id == IDC_DEC) CheatDlg.mode = CHEATDLG::S_DEC;
else if (id == IDC_HEX) CheatDlg.hex = getcheck(id);
else if (id == IDC_BYTE) CheatDlg.wordmode = 0;
else if (id == IDC_WORD) CheatDlg.wordmode = 1;
else if (id == IDB_SEARCH) CheatDlg.Search();
else id = 0;
if (id) CheatDlg.SetControls();
}
return 0;
}
void main_cheat()
{
OnEnterGui();
DialogBox(hIn, MAKEINTRESOURCE(IDD_CHEAT), wnd, cheatdlg);
eat();
OnExitGui();
}