/*
 * PROJECT:     ReactOS API tests
 * LICENSE:     GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
 * PURPOSE:     Test for NtUserCreateAcceleratorTable
 * COPYRIGHT:   Copyright 2025 Max Korostil <mrmks04@yandex.ru>
 */

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

#define MIN_VALID_NUMBER 1
#define MAX_VALID_NUMBER 32767

START_TEST(NtUserCreateAcceleratorTable)
{
    HACCEL hAccel = NULL;
    ACCEL Entries[5] = {0};
    ULONG EntriesCount = 0x80000005;
    BOOL bCrashed = FALSE;
    LPACCEL pEntries;

    /* Try heap overflow */
    SetLastError(0xdeadbeef);
    _SEH2_TRY
    {
        hAccel = NtUserCreateAcceleratorTable(Entries, EntriesCount);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bCrashed = TRUE;
    }
    _SEH2_END;

    ok_long(GetLastError(), 0xdeadbeef);
    ok_bool_false(bCrashed, "bCrashed");
    ok_hdl(hAccel, NULL);

    /* Try NULL Entries argument */
    SetLastError(0xdeadbeef);
    bCrashed = FALSE;
    hAccel = NULL;
    _SEH2_TRY
    {
        hAccel = NtUserCreateAcceleratorTable(NULL, MIN_VALID_NUMBER);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bCrashed = TRUE;
    }
    _SEH2_END;

    ok_long(GetLastError(), 0xdeadbeef);
    ok_bool_false(bCrashed, "bCrashed");
    ok_hdl(hAccel, NULL);

    /* Try invalid Entries argument */
    SetLastError(0xdeadbeef);
    bCrashed = FALSE;
    hAccel = NULL;
    _SEH2_TRY
    {
        hAccel = NtUserCreateAcceleratorTable((LPACCEL)(ULONG_PTR)0xC0FEC0FE, MIN_VALID_NUMBER);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bCrashed = TRUE;
    }
    _SEH2_END;

    ok_long(GetLastError(), ERROR_NOACCESS);
    ok_bool_false(bCrashed, "bCrashed");
    ok_hdl(hAccel, NULL);

    /* Try EntriesCount = 0 */
    SetLastError(0xdeadbeef);
    bCrashed = FALSE;
    hAccel = NULL;
    _SEH2_TRY
    {
        hAccel = NtUserCreateAcceleratorTable(Entries, MIN_VALID_NUMBER - 1);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bCrashed = TRUE;
    }
    _SEH2_END;

    ok_long(GetLastError(), 0xdeadbeef);
    ok_bool_false(bCrashed, "bCrashed");
    ok_hdl(hAccel, NULL);

    /* Try minimum */
    SetLastError(0xdeadbeef);
    bCrashed = FALSE;
    hAccel = NULL;
    _SEH2_TRY
    {
        hAccel = NtUserCreateAcceleratorTable(Entries, MIN_VALID_NUMBER);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bCrashed = TRUE;
    }
    _SEH2_END;

    ok_long(GetLastError(), 0xdeadbeef);
    ok_bool_false(bCrashed, "bCrashed");
    ok(hAccel != NULL, "hAccel is NULL\n");

    if (hAccel != NULL)
        DestroyAcceleratorTable(hAccel);

    /* Try correct parameters */
    SetLastError(0xdeadbeef);
    bCrashed = FALSE;
    hAccel = NULL;
    _SEH2_TRY
    {
        hAccel = NtUserCreateAcceleratorTable(Entries, _countof(Entries));
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bCrashed = TRUE;
    }
    _SEH2_END;

    ok_long(GetLastError(), 0xdeadbeef);
    ok_bool_false(bCrashed, "bCrashed");
    ok(hAccel != NULL, "hAccel is NULL\n");

    if (hAccel != NULL)
        DestroyAcceleratorTable(hAccel);

    /* Try maximum */
    pEntries = HeapAlloc(GetProcessHeap(), 0, MAX_VALID_NUMBER * sizeof(*pEntries));
    if (pEntries == NULL)
    {
        skip("pEntries is NULL\n");
    }
    else
    {
        SetLastError(0xdeadbeef);
        bCrashed = FALSE;
        hAccel = NULL;
        _SEH2_TRY
        {
            hAccel = NtUserCreateAcceleratorTable(pEntries, MAX_VALID_NUMBER);
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            bCrashed = TRUE;
        }
        _SEH2_END;

        ok_long(GetLastError(), 0xdeadbeef);
        ok_bool_false(bCrashed, "bCrashed");
        ok(hAccel != NULL, "hAccel is NULL\n");

        if (hAccel != NULL)
            DestroyAcceleratorTable(hAccel);

        HeapFree(GetProcessHeap(), 0, pEntries);
    }

    /* Try maximum +1 */
    pEntries = HeapAlloc(GetProcessHeap(), 0, (MAX_VALID_NUMBER + 1) * sizeof(*pEntries));
    if (pEntries == NULL)
    {
        skip("pEntries is NULL\n");
    }
    else
    {
        SetLastError(0xdeadbeef);
        bCrashed = FALSE;
        hAccel = NULL;
        _SEH2_TRY
        {
            hAccel = NtUserCreateAcceleratorTable(pEntries, MAX_VALID_NUMBER + 1);
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            bCrashed = TRUE;
        }
        _SEH2_END;

        ok_long(GetLastError(), 0xdeadbeef);
        ok_bool_false(bCrashed, "bCrashed");
        ok(hAccel != NULL, "hAccel is NULL\n");

        if (hAccel != NULL)
            DestroyAcceleratorTable(hAccel);

        HeapFree(GetProcessHeap(), 0, pEntries);
    }
}