W7 CrashFix with WindowMode - Page 2

#include <windows.h>
#include <stdio.h>

HANDLE hThread;
BOOL mainThreadRunning = TRUE;
BOOL minimizeBW = FALSE;
LONG prevWindowProc;
DEVMODE lastDevmode;
DEVMODE devmodeBW;


int MainThread();
LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
DEVMODE GetCurrentDeviceMode();


BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) {
    switch (fdwReason) {
    case DLL_PROCESS_ATTACH:
        DWORD threadId;
        hThread = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)MainThread, 0, 0, &threadId);
        break;
    case DLL_PROCESS_DETACH:
        mainThreadRunning = FALSE;
        CloseHandle(hThread);
        ChangeDisplaySettings(&lastDevmode, 0);
        break;
    }

    return TRUE;
}

int MainThread() {
    Sleep(2000); // wait until BW window is created, 2seconds should be enough
    
    DEVMODE d = GetCurrentDeviceMode();
    memcpy(&lastDevmode, &d, sizeof(DEVMODE));
    memcpy(&devmodeBW, &d, sizeof(DEVMODE));
    devmodeBW.dmPelsWidth = 640;
    devmodeBW.dmPelsHeight = 480;

    HWND hWndBW = FindWindow("SWarClass", 0);
    prevWindowProc = SetWindowLong(hWndBW, GWL_WNDPROC, (LONG)WindowProc);
    
    BOOL wasMinimized = TRUE;

    while (mainThreadRunning) {
        WINDOWPLACEMENT wndPl;
        wndPl.length = sizeof(WINDOWPLACEMENT);
        GetWindowPlacement(hWndBW, &wndPl);
        if (wndPl.showCmd == SW_SHOWMINIMIZED) {
            // set bw resolution
            if(!wasMinimized){
                ChangeDisplaySettings(&lastDevmode, 0);
                wasMinimized = TRUE;
            }
        } else {
            // set normal resolution
            if(wasMinimized){
                ChangeDisplaySettings(&devmodeBW, 0);
                SetWindowPos(hWndBW, 0, -5, -31, 0, 0, SWP_NOSIZE);
                wasMinimized = FALSE;
            }
        }

        if (minimizeBW == TRUE) {
            ShowWindow(hWndBW, SW_MINIMIZE);
            minimizeBW = FALSE;
        }
        Sleep(250);
    }

    ChangeDisplaySettings(&lastDevmode, 0);
    return 0;
}

LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    switch(uMsg) {
    case WM_KEYUP:
        if (wParam == VK_F12) {
            // minimize bw
            minimizeBW = TRUE;
        } 
        break;
    }
    return CallWindowProc((WNDPROC)prevWindowProc, hwnd, uMsg, wParam, lParam);
}

DEVMODE GetCurrentDeviceMode() {
    DISPLAY_DEVICE d;
    DEVMODE devMode;
    d.cb = sizeof(DISPLAY_DEVICE); 
    devMode.dmSize = sizeof(DEVMODE);
    int devNum = 0;
    int result = 0;
	
    do {
        result = EnumDisplayDevices(0, devNum, &d, 0);
        // check if it's the main device
        if (result && ((d.StateFlags & DISPLAY_DEVICE_PRIMARY_DEVICE) != 0)) {
            EnumDisplaySettings(d.DeviceName, -1, &devMode);
            return devMode;
        }
        devNum++;
    } while (result);

    // Current Devicemode not found
    devMode.dmSize = 0;
    return devMode;
}

#include <windows.h>
#include <stdio.h>

// structs, vars and flags
DEVMODE lastDevmode;
DEVMODE devmodeBW;
HWND    hWndBW      = NULL;
BOOL    exitThread  = FALSE;
BOOL    WaitFlag    = TRUE;
BOOL    isFS        = FALSE;
BOOL    isMin       = TRUE;

// Constant Values
const int BW_FULL_POS_X = -5;
const int BW_FULL_POS_Y = -31; // lower value causes the bnet ui mess-up
const int BW_WIDTH      = 650;
const int BW_HEIGHT     = 517;

// FUNCTION PROTOTYPES
DEVMODE   GetCurrentDeviceMode();
int       CheckForegroundWindow();
int       KeyHandler();
HWND      FindBW();
BOOL      IsBroodWarWindow();

#define KEYP(x)    (GetKeyState(x) < 0)

// Entry
int main(void) {
// int main(int argc, char* argv[]) {
    DWORD threadId, khId;
    HANDLE hThread, khThread;
    DEVMODE d;

    d = GetCurrentDeviceMode();
    if (!d.dmSize)
    {
        printf("Current Devicemode not found, exiting process.\n");
        ExitProcess(1);
    }

    memcpy(&lastDevmode, &d, sizeof(DEVMODE));
    memcpy(&devmodeBW, &d, sizeof(DEVMODE));
    devmodeBW.dmPelsWidth = 640;
    devmodeBW.dmPelsHeight = 480;

    printf("\nBroodWar Resolution Changer (by WhuazGoodNjaggah)\n" \
    "modified version by Dosbox\n" \
    "Press:\n" \
    "\t- Alt+F1 if you can't scroll inside BW (See WModeReadme.txt)\n" \
    "\t- Alt+Tab, WinKey or Alt+F8 to minimize BW\n" \
    "\t- Any key in this window to exit\n\n" \
    );

    if (FindBW())
        printf("BW seems to be active on start.\n");

    threadId = khId = 0;
    hThread = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)CheckForegroundWindow, 0, 0, &threadId);
    khThread = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)KeyHandler, 0, 0, &khId);

    getchar();

    printf("Exiting...\n");
    exitThread = TRUE;
    CloseHandle(hThread);
    CloseHandle(khThread);
    ExitProcess(1);
}

// GetCurrentDeviceMode
DEVMODE GetCurrentDeviceMode() {
    int devNum = 0;
    int result = 0;
    DISPLAY_DEVICE d;
    DEVMODE devMode;

    d.cb = sizeof(DISPLAY_DEVICE);
    memset(&devMode, 0, sizeof(DEVMODE));
    do {
        result = EnumDisplayDevices(0, devNum, &d, 0);
        // check if it's the main device
        if (result && ((d.StateFlags & DISPLAY_DEVICE_PRIMARY_DEVICE) != 0)) {
            EnumDisplaySettings(d.DeviceName, (DWORD)-1, &devMode);
            devMode.dmSize = sizeof(DEVMODE);
            return devMode;
        }
        devNum++;
    } while (result);

    // Current Devicemode not found, exiting process
    return devMode;
}

// KeyHandler
int KeyHandler()
{
    while(!exitThread) {
        if (FindBW()) {
            // try minimizing before the actual ALT+TAB or WinKey actions:
            // prevent a possible crash on XP / Vista (/ Win7 ?).
            if (( KEYP(VK_MENU) && (KEYP(VK_F8) || KEYP(VK_TAB))) ||
                  KEYP(VK_LWIN) || KEYP(VK_RWIN) )
            {
                ShowWindow(hWndBW, SW_MINIMIZE);
                if (!isMin) {
                    ChangeDisplaySettings(&lastDevmode, 0);
                    printf("BW Minimized.\n");                    
                    isMin = TRUE;
                }
            } else {
                isMin = FALSE;
            }
        }
        // value should be low
        Sleep(1);
    }
    return 0;
}

// CheckForegroundWindow
int CheckForegroundWindow() {
    while(!exitThread) {
        if (FindBW()) {
            if (!WaitFlag) {
                printf("BW Found.\n");
                WaitFlag = TRUE;
            }
            if (IsBroodWarWindow()) {
                SetWindowPos(hWndBW, 0, BW_FULL_POS_X, BW_FULL_POS_Y, 0, 0, SWP_NOSIZE);
                if (!isFS) {
                    printf("BW Full Screen.\n");
                    ChangeDisplaySettings(&devmodeBW, 0);
                    isFS = TRUE;
                }
            } else {
                SetWindowPos(hWndBW, 0, 0, 0, 0, 0, SWP_NOSIZE);
                if (isFS) {
                    ChangeDisplaySettings(&lastDevmode, 0);
                    isFS = FALSE;
                }
            }
        } else {
            if (WaitFlag) {
                printf("Waiting for BW...\n");
                ChangeDisplaySettings(&lastDevmode, 0);
                WaitFlag = FALSE;
            }
        }
        Sleep(100);
    }
    return 0;
}

// Find BW
HWND FindBW(void)
{
  return hWndBW = FindWindow("SWarClass", 0);
}

// IsBroodWarWindow
BOOL IsBroodWarWindow(void) {
    char szWindowClass[255];
    int cmpSWC;
    int cmpSDD;

    RealGetWindowClass(GetForegroundWindow(), szWindowClass, 255);
    cmpSWC = strncmp(szWindowClass, "SWarClass", 9);
    cmpSDD = strncmp(szWindowClass, "#32770", 6);
    return (cmpSWC == 0) || (cmpSDD == 0);
}