[WIN32KNT_APITEST] Test large buffers without relying on stack layout. (#1314)
[reactos.git] / modules / rostests / apitests / win32nt / ntgdi / NtGdiSetBitmapBits.c
index 6924492..88b5720 100644 (file)
  * PROJECT:         ReactOS api tests
  * LICENSE:         GPL - See COPYING in the top level directory
  * PURPOSE:         Test for NtGdiSetBitmapBits
- * PROGRAMMERS:
+ * PROGRAMMERS:     Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
  */
 
 #include <win32nt.h>
 
 START_TEST(NtGdiSetBitmapBits)
 {
-       BYTE Bits[50] = {0,1,2,3,4,5,6,7,8,9};
-       HBITMAP hBitmap;
+    BYTE Bits[100] = {0,1,2,3,4,5,6,7,8,9};
+    HBITMAP hBitmap;
+    HDC hDC;
+    BITMAPINFO bmi;
+    LPVOID pvBits;
+    LPBYTE LargeBits;
 
-       SetLastError(ERROR_SUCCESS);
-       ok_long(NtGdiSetBitmapBits(0, 0, 0), 0);
-       ok_long(GetLastError(), ERROR_SUCCESS);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(0, 0, 0), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test NULL bitnap handle */
-       SetLastError(ERROR_SUCCESS);
-       ok_long(NtGdiSetBitmapBits(0, 5, Bits), 0);
-       ok_long(GetLastError(), ERROR_INVALID_HANDLE);
+    /* Test NULL bitnap handle */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(0, 5, Bits), 0);
+    ok_long(GetLastError(), ERROR_INVALID_HANDLE);
 
-       /* Test invalid bitmap handle */
-       hBitmap = (HBITMAP)CreatePen(PS_SOLID, 1, RGB(1,2,3));
-       SetLastError(ERROR_SUCCESS);
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 0);
-       ok_long(GetLastError(), ERROR_INVALID_HANDLE);
-       DeleteObject(hBitmap);
+    /* Test invalid bitmap handle */
+    hBitmap = (HBITMAP)CreatePen(PS_SOLID, 1, RGB(1,2,3));
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 0);
+    ok_long(GetLastError(), ERROR_INVALID_HANDLE);
+    DeleteObject(hBitmap);
 
-       hBitmap = CreateBitmap(3, 3, 1, 8, NULL);
-       SetLastError(ERROR_SUCCESS);
+    hBitmap = CreateBitmap(3, 3, 1, 8, NULL);
+    ok(hBitmap != NULL, "hBitmap was NULL.\n");
 
-       /* test NULL pointer and count buffer size != 0 */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, NULL), 0);
+    /* test NULL pointer and count buffer size != 0 */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 5, NULL), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* test NULL pointer and buffer size == 0*/
-       ok_long(NtGdiSetBitmapBits(hBitmap, 0, NULL), 0);
+    /* test NULL pointer and buffer size == 0*/
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0, NULL), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* test bad pointer */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, (PBYTE)0x500), 0);
+    /* test bad pointer */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 5, (PBYTE)0x500), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test if we can set a number of bytes between lines */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 5);
+    /* Test if we can set a number of bytes between lines */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 5);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test alignment */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 4, Bits+1), 4);
+    /* Test alignment */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 4, Bits+1), 4);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test 1 byte too much */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    /* Test 1 byte too much */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test one row too much */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 12, Bits), 12);
+    /* Test one row too much */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 12, Bits), 12);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       ok_long(NtGdiSetBitmapBits(hBitmap, 13, Bits), 12);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 13, Bits), 12);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       ok_long(NtGdiSetBitmapBits(hBitmap, 100, Bits), 12);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 100, Bits), 12);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test huge bytes count */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 12345678, Bits), 0);
+    /* Test large byte counts */
+    LargeBits = VirtualAlloc(NULL, 0x100000 + PAGE_SIZE, MEM_RESERVE, PAGE_NOACCESS);
+    VirtualAlloc(LargeBits, 0x100000, MEM_COMMIT, PAGE_READWRITE);
+    CopyMemory(LargeBits, Bits, sizeof(Bits));
 
-       /* Test negative bytes count */
-       ok_long(NtGdiSetBitmapBits(hBitmap, -5, Bits), 0);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x100, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       ok_long(GetLastError(), ERROR_SUCCESS);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x1000, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       DeleteObject(hBitmap);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x10000, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x100000, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x100001, LargeBits), 0x0);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    VirtualFree(LargeBits, 0, MEM_RELEASE);
+
+    DeleteObject(hBitmap);
+
+    /* ------------------------- */
+
+    hBitmap = CreateBitmap(16, 16, 1, 1, NULL);
+    ok(hBitmap != NULL, "hBitmap was NULL.\n");
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 0, Bits), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0);
+    ok_int(Bits[1], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0);
+    ok_int(Bits[1], 0);
+    ok_int(Bits[2], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x33);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0x33);
+    ok_int(Bits[2], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0x33);
+    ok_int(Bits[3], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0xAA);
+    ok_int(Bits[3], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 4, Bits), 4);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0xAA);
+    ok_int(Bits[3], 0x33);
+    ok_int(Bits[4], 0x55);
+
+    DeleteObject(hBitmap);
+
+    /* ------------------------- */
+
+    hDC = CreateCompatibleDC(NULL);
+
+    ZeroMemory(&bmi, sizeof(bmi));
+    bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
+    bmi.bmiHeader.biWidth = 15;
+    bmi.bmiHeader.biHeight = 15;    /* positive */
+    bmi.bmiHeader.biPlanes = 1;
+    bmi.bmiHeader.biBitCount = 24;
+    hBitmap = CreateDIBSection(hDC, &bmi, DIB_RGB_COLORS, &pvBits, NULL, 0);
+    ok(hBitmap != NULL, "hBitmap was NULL.\n");
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 0, Bits), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0);
+    ok_int(Bits[1], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0);
+    ok_int(Bits[1], 0);
+    ok_int(Bits[2], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x33);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0x33);
+    ok_int(Bits[2], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0x33);
+    ok_int(Bits[3], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0xAA);
+    ok_int(Bits[3], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 4, Bits), 4);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0xAA);
+    ok_int(Bits[3], 0x33);
+    ok_int(Bits[4], 0x55);
+
+    DeleteObject(hBitmap);
+    DeleteDC(hDC);
+
+    /* ------------------------- */
+
+    hDC = CreateCompatibleDC(NULL);
+
+    ZeroMemory(&bmi, sizeof(bmi));
+    bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
+    bmi.bmiHeader.biWidth = 15;
+    bmi.bmiHeader.biHeight = -15;   /* negative */
+    bmi.bmiHeader.biPlanes = 1;
+    bmi.bmiHeader.biBitCount = 24;
+    hBitmap = CreateDIBSection(hDC, &bmi, DIB_RGB_COLORS, &pvBits, NULL, 0);
+    ok(hBitmap != NULL, "hBitmap was NULL.\n");
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 0, Bits), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0);
+    ok_int(Bits[1], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0);
+    ok_int(Bits[1], 0);
+    ok_int(Bits[2], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0x33);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 1, Bits), 1);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0x33);
+    ok_int(Bits[2], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 2, Bits), 2);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0x33);
+    ok_int(Bits[3], 0x55);
+
+    FillMemory(Bits, sizeof(Bits), 0xAA);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    FillMemory(Bits, sizeof(Bits), 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 3, Bits), 3);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0xAA);
+    ok_int(Bits[3], 0x55);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiGetBitmapBits(hBitmap, 4, Bits), 4);
+    ok_long(GetLastError(), 0xDEADFACE);
+    ok_int(Bits[0], 0xAA);
+    ok_int(Bits[1], 0xAA);
+    ok_int(Bits[2], 0xAA);
+    ok_int(Bits[3], 0x33);
+    ok_int(Bits[4], 0x55);
+
+    DeleteObject(hBitmap);
+    DeleteDC(hDC);
 }