/*
 * PROJECT:         ReactOS api tests
 * LICENSE:         GPL - See COPYING in the top level directory
 * PURPOSE:         Test for NtGdiCreateDIBSection
 * PROGRAMMERS:
 */

#include <win32nt.h>

/*
HBITMAP
APIENTRY
NtGdiCreateDIBSection(
    IN HDC hDC,
    IN OPTIONAL HANDLE hSection,
    IN DWORD dwOffset,
    IN LPBITMAPINFO pbmi,
    IN DWORD iUsage,
    IN UINT cjHeader,
    IN FLONG fl,
    IN ULONG_PTR dwColorSpace,
    OUT PVOID *ppvBits)
*/

ULONG
GetBitmapSize(BITMAPINFOHEADER *pbih)
{
    ULONG WidthBits, WidthBytes;

    WidthBits = pbih->biWidth * pbih->biBitCount * pbih->biPlanes;
    WidthBytes = ((WidthBits + 31) & ~ 31) >> 3;

    return pbih->biHeight * WidthBytes;
}


START_TEST(NtGdiCreateDIBSection)
{
    HBITMAP hbmp;
    HDC hDC;
    ULONG cjHeader;
    PVOID pvBits = NULL;
    ULONG cEntries;
    DIBSECTION dibsection;

    struct
    {
        BITMAPINFOHEADER bmiHeader;
        RGBQUAD          bmiColors[100];
    } bmi;
    PBITMAPINFO pbmi = (PBITMAPINFO)&bmi;
    PBITMAPINFOHEADER pbih = (PBITMAPINFOHEADER)&bmi.bmiHeader;
    PBITMAPV4HEADER pbV4h = (PBITMAPV4HEADER)&bmi.bmiHeader;
    PBITMAPV5HEADER pbV5h = (PBITMAPV5HEADER)&bmi.bmiHeader;

    HANDLE hSection;
    NTSTATUS Status;
    LARGE_INTEGER MaximumSize;

    hDC = GetDC(0);
    pbih->biSize = sizeof(BITMAPINFOHEADER);
    pbih->biWidth = 2;
    pbih->biHeight = 2;
    pbih->biPlanes = 1;
    pbih->biBitCount = 1;
    pbih->biCompression = BI_RGB;
    pbih->biSizeImage = 0;
    pbih->biXPelsPerMeter = 100;
    pbih->biYPelsPerMeter = 100;
    pbih->biClrUsed = 2;
    pbih->biClrImportant = 2;

    cEntries = 0;

/** iUsage = 0 (DIB_RGB_COLORS) ***********************************************/

    cjHeader = bmi.bmiHeader.biSize + cEntries * 4 + 8;

    /* Test something simple */
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
//    TEST(GetLastError() == 0);
    TEST(GetObject(hbmp, sizeof(DIBSECTION), &dibsection) == sizeof(DIBSECTION));
    TEST(dibsection.dsBitfields[0] == 0);
    TEST(dibsection.dsBitfields[1] == 0);
    TEST(dibsection.dsBitfields[2] == 0);
    TEST(dibsection.dshSection == 0);
    TEST(dibsection.dsOffset == 0);
    if (hbmp) DeleteObject(hbmp);


    /* Test a 0 HDC */
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(0, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
    TEST(GetLastError() == ERROR_NOT_ENOUGH_MEMORY);
    if (hbmp) DeleteObject(hbmp);

    /* Test a wrong HDC */
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection((HDC)(LONG_PTR)0xdeadbeef, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != 0);
    TEST(hbmp != 0);
    TEST(GetLastError() == 8);
    if (hbmp) DeleteObject(hbmp);

    /* Test pbmi = NULL */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, NULL, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test invalid pbmi */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, (PVOID)(LONG_PTR)0x80001234, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
    if (hbmp) DeleteObject(hbmp);

    /* Test invalid pbmi */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, (PVOID)(LONG_PTR)1, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
    if (hbmp) DeleteObject(hbmp);

    /* Test ppvBits = NULL */
    SetLastError(0);
    hbmp = NtGdiCreateDIBSection(0, NULL, 0, pbmi, 0, cjHeader, 0, 0, NULL);
    TEST(hbmp == 0);
    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
    if (hbmp) DeleteObject(hbmp);

    /* Test ppvBits = NULL and pbmi == 0*/
    SetLastError(0);
    hbmp = NtGdiCreateDIBSection(0, NULL, 0, NULL, 0, cjHeader, 0, 0, NULL);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test ppvBits = NULL and wrong cjHeader */
    SetLastError(0);
    hbmp = NtGdiCreateDIBSection(0, NULL, 0, pbmi, 0, cjHeader+4, 0, 0, NULL);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test ppvBits = NULL and cjHeader = 0 */
    SetLastError(0);
    hbmp = NtGdiCreateDIBSection(0, NULL, 0, pbmi, 0, 0, 0, 0, NULL);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test wrong cjHeader */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader+4, 0, 0, &pvBits);
    pvBits = (PVOID)-1;
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test different bitcount */
    pbih->biBitCount = 4;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(hbmp != 0);
    if (hbmp) DeleteObject(hbmp);

    pbih->biBitCount = 8;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(hbmp != 0);
    if (hbmp) DeleteObject(hbmp);

    cjHeader = pbih->biSize;
    pbih->biBitCount = 16;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(hbmp != 0);
    if (hbmp) DeleteObject(hbmp);

    pbih->biBitCount = 24;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(hbmp != 0);
    if (hbmp) DeleteObject(hbmp);

    pbih->biBitCount = 32;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(hbmp != 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test BI_BITFIELDS */
    cEntries = 3;
    cjHeader = pbih->biSize + cEntries * sizeof(DWORD);
    pbih->biBitCount = 16;
    pbih->biCompression = BI_BITFIELDS;
    ((DWORD*)pbmi->bmiColors)[0] = 0x0007;
    ((DWORD*)pbmi->bmiColors)[1] = 0x0038;
    ((DWORD*)pbmi->bmiColors)[2] = 0x01C0;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(hbmp != 0);
    TEST(GetObject(hbmp, sizeof(DIBSECTION), &dibsection) == sizeof(DIBSECTION));
    TEST(dibsection.dsBm.bmType == 0);
    TEST(dibsection.dsBm.bmWidth == 2);
    TEST(dibsection.dsBm.bmHeight == 2);
    TEST(dibsection.dsBm.bmWidthBytes == 4);
    TEST(dibsection.dsBm.bmPlanes == 1);
    TEST(dibsection.dsBm.bmBitsPixel == 16);
    TEST(dibsection.dsBm.bmBits == pvBits);
    TEST(dibsection.dsBmih.biSize == sizeof(BITMAPINFOHEADER));
    TEST(dibsection.dsBmih.biWidth == 2);
    TEST(dibsection.dsBmih.biHeight == 2);
    TEST(dibsection.dsBmih.biPlanes == 1);
    TEST(dibsection.dsBmih.biBitCount == 16);
    TEST(dibsection.dsBmih.biCompression == BI_BITFIELDS);
    TEST(dibsection.dsBmih.biSizeImage == 8);
    TEST(dibsection.dsBmih.biXPelsPerMeter == 0);
    TEST(dibsection.dsBmih.biYPelsPerMeter == 0);
    TEST(dibsection.dsBmih.biClrUsed == 0);
    TEST(dibsection.dsBmih.biClrImportant == 0);
    TEST(dibsection.dsBitfields[0] == 0x0007);
    TEST(dibsection.dsBitfields[1] == 0x0038);
    TEST(dibsection.dsBitfields[2] == 0x01C0);
    TEST(dibsection.dshSection == 0);
    TEST(dibsection.dsOffset == 0);

printf("dib with bitfileds: %p\n", hbmp);
//system("PAUSE");

    if (hbmp) DeleteObject(hbmp);


    /* Test BI_BITFIELDS */
    SetLastError(0);
    pvBits = 0;

    pbih->biSize = sizeof(BITMAPINFOHEADER);
    pbih->biWidth = 2;
    pbih->biHeight = 2;
    pbih->biPlanes = 1;
    pbih->biBitCount = 4;
    pbih->biCompression = BI_RGB;
    pbih->biSizeImage = 0;
    pbih->biXPelsPerMeter = 100;
    pbih->biYPelsPerMeter = 100;
    pbih->biClrUsed = 0;
    pbih->biClrImportant = 0;
    ((DWORD*)pbmi->bmiColors)[0] = 0xF800;
    ((DWORD*)pbmi->bmiColors)[1] = 0x00ff00;
    ((DWORD*)pbmi->bmiColors)[2] = 0x0000ff;
    cEntries = 0;
    cjHeader = bmi.bmiHeader.biSize + cEntries * 4 + 20;


/** iUsage = 1 (DIB_PAL_COLORS) ***********************************************/

    pbmi->bmiHeader.biClrUsed = 2;
    pbmi->bmiHeader.biClrImportant = 2;

    cEntries = 2;
    cjHeader = bmi.bmiHeader.biSize + cEntries * 4 + 8;

    /* Test iUsage = 1 */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 1, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);



/** iUsage = 2 (???) **********************************************************/

    /* Test iUsage = 2 */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 2, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);


/** wrong iUsage **************************************************************/

    cEntries = 0;
    cjHeader = bmi.bmiHeader.biSize + cEntries * 4 + 8;

    /* Test iUsage = 3 */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 3, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
    if (hbmp) DeleteObject(hbmp);

    /* Test iUsage = 3 */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 3, cjHeader+4, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* Test wrong iUsage */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, -55, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
    if (hbmp) DeleteObject(hbmp);

    /* Test wrong iUsage and wrong cjHeader */
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, -55, cjHeader+4, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* increased header size */
    pbih->biSize = sizeof(BITMAPINFOHEADER) + 4;
    cjHeader = pbih->biSize + cEntries * 4 + 8;
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
    TEST(GetLastError() == 8);
    if (hbmp) DeleteObject(hbmp);

    /* increased header size */
    pbih->biSize = sizeof(BITMAPINFOHEADER) + 2;
    cjHeader = pbih->biSize + cEntries * 4 + 8;
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);

    /* decreased header size */
    pbih->biSize = sizeof(BITMAPINFOHEADER) - 4;
    cjHeader = pbih->biSize + cEntries * 4 + 8;
    SetLastError(0);
    pvBits = (PVOID)-1;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits == (PVOID)-1);
    TEST(hbmp == 0);
    TEST(GetLastError() == 0);
    if (hbmp) DeleteObject(hbmp);


/** BITMAPV4HEADER ************************************************************/

    pbV4h->bV4Size = sizeof(BITMAPV4HEADER);
    pbV4h->bV4Width = 2;
    pbV4h->bV4Height = 3;
    pbV4h->bV4Planes = 1;
    pbV4h->bV4BitCount = 1;
    pbV4h->bV4V4Compression = BI_RGB;
    pbV4h->bV4SizeImage = 0;
    pbV4h->bV4XPelsPerMeter = 100;
    pbV4h->bV4YPelsPerMeter = 100;
    pbV4h->bV4ClrUsed = 0;
    pbV4h->bV4ClrImportant = 0;
    pbV4h->bV4RedMask = 0;
    pbV4h->bV4GreenMask = 0;
    pbV4h->bV4BlueMask = 0;
    pbV4h->bV4AlphaMask = 0;
    pbV4h->bV4CSType = 0;
    memset(&pbV4h->bV4Endpoints, 0, sizeof(CIEXYZTRIPLE));
    pbV4h->bV4GammaRed = 0;
    pbV4h->bV4GammaGreen = 0;
    pbV4h->bV4GammaBlue = 0;

    cEntries = 0;
    cjHeader = bmi.bmiHeader.biSize + cEntries * 4 + 8;

    /* Test something simple */
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
    TEST(GetLastError() == 8);
    if (hbmp) DeleteObject(hbmp);


/** BITMAPV5HEADER ************************************************************/

    pbV5h->bV5Size = sizeof(BITMAPV5HEADER);
    pbV5h->bV5Width = 2;
    pbV5h->bV5Height = 3;
    pbV5h->bV5Planes = 1;
    pbV5h->bV5BitCount = 1;
    pbV5h->bV5Compression = BI_RGB;
    pbV5h->bV5SizeImage = 0;
    pbV5h->bV5XPelsPerMeter = 100;
    pbV5h->bV5YPelsPerMeter = 100;
    pbV5h->bV5ClrUsed = 2;
    pbV5h->bV5ClrImportant = 2;
    pbV5h->bV5RedMask = 0;
    pbV5h->bV5GreenMask = 0;
    pbV5h->bV5BlueMask = 0;
    pbV5h->bV5AlphaMask = 0;
    pbV5h->bV5CSType = 0;
    memset(&pbV5h->bV5Endpoints, 0, sizeof(CIEXYZTRIPLE));
    pbV5h->bV5GammaRed = 0;
    pbV5h->bV5GammaGreen = 0;
    pbV5h->bV5GammaBlue = 0;
    pbV5h->bV5Intent = 0;
    pbV5h->bV5ProfileData = 0;
    pbV5h->bV5ProfileSize = 0;
    pbV5h->bV5Reserved = 0;

    cEntries = 0;
    cjHeader = pbV5h->bV5Size + cEntries * 4 + 8;

    /* Test something simple */
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
    TEST(GetLastError() == 8);
    if (hbmp) DeleteObject(hbmp);

    /* increased header size */
    pbV5h->bV5Size = sizeof(BITMAPV5HEADER) + 64;
    cjHeader = pbV5h->bV5Size + cEntries * 4 + 8;
    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(hDC, NULL, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
    TEST(GetLastError() == 8);
    if (hbmp) DeleteObject(hbmp);

    /* Test section */
    MaximumSize.QuadPart = 4096;
    Status = NtCreateSection(&hSection,
                             SECTION_ALL_ACCESS,
                             NULL,
                             &MaximumSize,
                             PAGE_READWRITE,
                             SEC_COMMIT,
                             NULL);
    ASSERT(NT_SUCCESS(Status));

    SetLastError(0);
    pvBits = 0;
    hbmp = NtGdiCreateDIBSection(hDC, hSection, 0, pbmi, 0, cjHeader, 0, 0, &pvBits);
    TEST(pvBits != NULL);
    TEST(hbmp != 0);
//    TEST(GetLastError() == 0);
    printf("hbmp = %p, pvBits = %p, hSection = %p\n", hbmp, pvBits, hSection);
//system("PAUSE");
    if (hbmp) DeleteObject(hbmp);
}