[WIN32KNT_APITEST] Test large buffers without relying on stack layout. (#1314)
[reactos.git] / modules / rostests / apitests / win32nt / ntgdi / NtGdiSetBitmapBits.c
index e13aabd..88b5720 100644 (file)
 
 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(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(0, 0, 0), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    /* 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(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 0);
+    ok_long(GetLastError(), ERROR_INVALID_HANDLE);
+    DeleteObject(hBitmap);
 
-       SetLastError(ERROR_SUCCESS);
-       ok_long(NtGdiSetBitmapBits(0, 0, 0), 0);
-       ok_long(GetLastError(), ERROR_SUCCESS);
+    hBitmap = CreateBitmap(3, 3, 1, 8, NULL);
+    ok(hBitmap != NULL, "hBitmap was NULL.\n");
 
-       /* Test NULL bitnap handle */
-       SetLastError(ERROR_SUCCESS);
-       ok_long(NtGdiSetBitmapBits(0, 5, Bits), 0);
-       ok_long(GetLastError(), ERROR_INVALID_HANDLE);
+    /* 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*/
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0, NULL), 0);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    /* 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 */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 5);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    /* Test alignment */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 4, Bits+1), 4);
+    ok_long(GetLastError(), 0xDEADFACE);
+
+    /* Test 1 byte too much */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* 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 one row too much */
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 12, Bits), 12);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       hBitmap = CreateBitmap(3, 3, 1, 8, NULL);
-       SetLastError(ERROR_SUCCESS);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 13, Bits), 12);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* test NULL pointer and count buffer size != 0 */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, NULL), 0);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 100, Bits), 12);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* test NULL pointer and buffer size == 0*/
-       ok_long(NtGdiSetBitmapBits(hBitmap, 0, NULL), 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 bad pointer */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, (PBYTE)0x500), 0);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x100, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test if we can set a number of bytes between lines */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 5, Bits), 5);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x1000, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test alignment */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 4, Bits+1), 4);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x10000, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test 1 byte too much */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x100000, LargeBits), 0xC);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       /* Test one row too much */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 12, Bits), 12);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 0x100001, LargeBits), 0x0);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       ok_long(NtGdiSetBitmapBits(hBitmap, 13, Bits), 12);
+    VirtualFree(LargeBits, 0, MEM_RELEASE);
+
+    DeleteObject(hBitmap);
 
-       ok_long(NtGdiSetBitmapBits(hBitmap, 100, Bits), 12);
+    /* ------------------------- */
 
-       /* Test huge bytes count */
-       ok_long(NtGdiSetBitmapBits(hBitmap, 12345678, Bits), 0);
+    hBitmap = CreateBitmap(16, 16, 1, 1, NULL);
+    ok(hBitmap != NULL, "hBitmap was NULL.\n");
 
-       /* Test negative bytes count */
-       ok_long(NtGdiSetBitmapBits(hBitmap, -5, Bits), 0);
+    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);
 
-       ok_long(GetLastError(), ERROR_SUCCESS);
+    FillMemory(Bits, sizeof(Bits), 0x33);
+    SetLastError(0xDEADFACE);
+    ok_long(NtGdiSetBitmapBits(hBitmap, 10, Bits), 10);
+    ok_long(GetLastError(), 0xDEADFACE);
 
-       DeleteObject(hBitmap);
+    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);
 
     /* ------------------------- */
 
-    hBitmap = CreateBitmap(16, 16, 1, 1, NULL);
+    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);
@@ -132,6 +295,7 @@ START_TEST(NtGdiSetBitmapBits)
     ok_int(Bits[4], 0x55);
 
     DeleteObject(hBitmap);
+    DeleteDC(hDC);
 
     /* ------------------------- */
 
@@ -140,63 +304,89 @@ START_TEST(NtGdiSetBitmapBits)
     ZeroMemory(&bmi, sizeof(bmi));
     bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
     bmi.bmiHeader.biWidth = 15;
-    bmi.bmiHeader.biHeight = 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);