/*
 * PROJECT:     ReactOS api tests
 * LICENSE:     LGPL-2.0-or-later (https://spdx.org/licenses/LGPL-2.0-or-later)
 * PURPOSE:     Test for NtUserGetThreadState
 * COPYRIGHT:   Copyright 2021-2022 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
 */

#include <win32nt.h>
#include <pseh/pseh2.h>

#define MAX_COUNT 8
#define IGNORED 0xDEADFACE
#define RAISED 0xBADBEEF
#define DO_CHECK(i, value) CheckThreadState(__LINE__, (i), (DWORD_PTR)(value))

#undef DO_PRINT

#ifdef DO_PRINT
static VOID PrintThreadState(INT lineno, HWND hWnd)
{
    INT i;
    HIMC hIMC = ImmGetContext(hWnd);
    HWND hImeWnd = ImmGetDefaultIMEWnd(hWnd);

    trace("---\n");
    trace("__LINE__: %d\n", lineno);
    trace("hWnd: %p\n", (LPVOID)hWnd);
    trace("GetFocus(): %p\n", (LPVOID)GetFocus());
    trace("GetCapture(): %p\n", (LPVOID)GetCapture());
    trace("GetActiveWindow(): %p\n", (LPVOID)GetActiveWindow());
    trace("ImmGetContext(): %p\n", (LPVOID)hIMC);
    trace("ImmGetDefaultIMEWnd(): %p\n", (LPVOID)hImeWnd);

    for (i = 0; i < MAX_COUNT; ++i)
    {
        _SEH2_TRY
        {
            trace("ThreadState[%d]: %p\n", i, (LPVOID)NtUserGetThreadState(i));
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            trace("ThreadState[%d]: exception\n", i);
        }
        _SEH2_END;
    }

    ImmReleaseContext(hWnd, hIMC);
}
#endif

static VOID CheckThreadState(INT lineno, INT i, DWORD_PTR dwState)
{
    DWORD_PTR dwValue;

    _SEH2_TRY
    {
        dwValue = (DWORD_PTR)NtUserGetThreadState(i);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        dwValue = RAISED;
    }
    _SEH2_END;

    if (dwState != IGNORED)
    {
        ok(dwValue == dwState, "Line %d: Mismatch 0x%lX vs. 0x%lX\n",
           lineno, (DWORD)dwValue, (DWORD)dwState);
    }
}

static VOID DoTest_EDIT(VOID)
{
    HWND hWnd, hImeWnd;
    HIMC hIMC;
    HCURSOR hCursor;
    LPARAM ExtraInfo;
    LONG MsgTime;

    hWnd = CreateWindowA("EDIT", "Test", ES_LEFT | ES_MULTILINE | WS_VISIBLE,
                         0, 0, 50, 30,
                         NULL, NULL, GetModuleHandleW(NULL), NULL);
    hImeWnd = ImmGetDefaultIMEWnd(hWnd);
    ok_int(hImeWnd != NULL, TRUE);

    hIMC = ImmGetContext(hWnd);
    ok_int(hIMC != NULL, TRUE);
    ok_int(hIMC == (HIMC)NtUserGetThreadState(4), TRUE);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetCapture(hWnd);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, hWnd);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    ReleaseCapture();
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetFocus(hWnd);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetActiveWindow(hWnd);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetActiveWindow(NULL);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, IGNORED);
    DO_CHECK(1, IGNORED);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    ImmReleaseContext(hWnd, hIMC);
    DestroyWindow(hWnd);
}

static VOID DoTest_BUTTON(VOID)
{
    HWND hWnd, hImeWnd;
    HIMC hIMC;
    HCURSOR hCursor;
    LPARAM ExtraInfo;
    LONG MsgTime;

    hWnd = CreateWindowA("BUTTON", "Test", BS_PUSHBUTTON | WS_VISIBLE,
                         0, 0, 50, 30,
                         NULL, NULL, GetModuleHandleW(NULL), NULL);
    hImeWnd = ImmGetDefaultIMEWnd(hWnd);
    ok_int(hImeWnd != NULL, TRUE);

    hIMC = ImmGetContext(hWnd);
    ok_int(hIMC != NULL, FALSE);

    hIMC = (HIMC)NtUserGetThreadState(4);
    ok_int(hIMC != NULL, TRUE);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetCapture(hWnd);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, hWnd);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    ReleaseCapture();
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetFocus(hWnd);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetActiveWindow(hWnd);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, hWnd);
    DO_CHECK(1, hWnd);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    SetActiveWindow(NULL);
    hCursor = GetCursor();
    ExtraInfo = GetMessageExtraInfo();
    MsgTime = GetMessageTime();

#ifdef DO_PRINT
    PrintThreadState(__LINE__, hWnd);
#endif
    DO_CHECK(0, IGNORED);
    DO_CHECK(1, IGNORED);
    DO_CHECK(2, 0);
    DO_CHECK(3, hImeWnd);
    DO_CHECK(4, hIMC);
    DO_CHECK(6, hCursor);
    DO_CHECK(8, ExtraInfo);
    DO_CHECK(10, MsgTime);

    DestroyWindow(hWnd);
}

START_TEST(NtUserGetThreadState)
{
    DoTest_EDIT();
    DoTest_BUTTON();
}