#include <windef.h>
#include <winbase.h>
#include <wingdi.h>
#include <winddi.h>
#include <prntfont.h>
#include <ndk/ntndk.h>
#include <ntgdityp.h>
#include <ntgdi.h>
#include <ntgdihdl.h>
#include "gditools.h"

Include dependency graph for gditools.c:

Macros
#define	WIN32_NO_STATUS

#define	NTOS_MODE_USER

#define	FL_INVERT_COLORS 0x01

#define	FL_RED_BLUE 0x02

Functions
PENTRY	GdiQueryTable (VOID)

BOOL	GdiIsHandleValid (_In_ HGDIOBJ hobj)

BOOL	GdiIsHandleValidEx (_In_ HGDIOBJ hobj, _In_ GDILOOBJTYPE ObjectType)

PVOID	GdiGetHandleUserData (_In_ HGDIOBJ hobj)

BOOL	InitPerBitDepth (_In_ ULONG cBitsPerPixel, _In_ ULONG cx, _In_ ULONG cy, _Out_ HBITMAP phbmp, _Out_ HDC phdcDIB, _Out_ HBITMAP phbmpDIB, _Out_ PVOID ppvBits, _In_ ULONG flags)

BOOL	GdiToolsInit (void)

Variables
HBITMAP	ghbmp1

HBITMAP	ghbmp1_InvCol

HBITMAP	ghbmp1_RB

HBITMAP	ghbmp4

HBITMAP	ghbmp8

HBITMAP	ghbmp16

HBITMAP	ghbmp24

HBITMAP	ghbmp32

HBITMAP	ghbmpDIB1

HBITMAP	ghbmpDIB1_InvCol

HBITMAP	ghbmpDIB1_RB

HBITMAP	ghbmpDIB4

HBITMAP	ghbmpDIB8

HBITMAP	ghbmpDIB16

HBITMAP	ghbmpDIB24

HBITMAP	ghbmpDIB32

HDC	ghdcDIB1

HDC	ghdcDIB1_InvCol

HDC	ghdcDIB1_RB

HDC	ghdcDIB4

HDC	ghdcDIB8

HDC	ghdcDIB16

HDC	ghdcDIB24

HDC	ghdcDIB32

PVOID	gpvDIB1

PVOID	gpvDIB1_InvCol

PVOID	gpvDIB1_RB

PVOID	gpvDIB4

PVOID	gpvDIB8

PVOID	gpvDIB16

PVOID	gpvDIB24

PVOID	gpvDIB32

ULONG(*	gpDIB32 )[8][8]

HPALETTE	ghpal

HDC	ghdcInfo

MYPAL	gpal

Macro Definition Documentation

◆ FL_INVERT_COLORS

#define FL_INVERT_COLORS 0x01

Definition at line 108 of file gditools.c.

◆ FL_RED_BLUE

#define FL_RED_BLUE 0x02

Definition at line 109 of file gditools.c.

◆ NTOS_MODE_USER

#define NTOS_MODE_USER

Definition at line 11 of file gditools.c.

◆ WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 4 of file gditools.c.

Function Documentation

◆ GdiGetHandleUserData()

PVOID GdiGetHandleUserData ( _In_ HGDIOBJ hobj )

Definition at line 93 of file gditools.c.

{
    PENTRY pentHmgr = GdiQueryTable();
    USHORT Index = (ULONG_PTR)hobj;
    PENTRY pentry = &pentHmgr[Index];
 
    if (!GdiIsHandleValid(hobj))
    {
        return NULL;
    }
 
    return pentry->pUser;
}

Referenced by CancelDC(), CreateCompatibleBitmap(), CreateCompatibleDC(), CreatePen(), GdiDeleteBrushOrPen(), GetBrushOrgEx(), GetColorSpace(), IntCreateDICW(), SetBrushOrgEx(), SetDIBitsToDevice(), START_TEST(), and StretchDIBits().

◆ GdiIsHandleValid()

BOOL GdiIsHandleValid ( _In_ HGDIOBJ hobj )

Definition at line 54 of file gditools.c.

{
    PENTRY pentHmgr = GdiQueryTable();
    USHORT Index = (ULONG_PTR)hobj & 0xFFFF;
    PENTRY pentry = &pentHmgr[Index];
 
    if ((pentry->einfo.pobj == NULL) ||
        ((LONG_PTR)pentry->einfo.pobj > 0) ||
        (pentry->FullUnique != (USHORT)((ULONG_PTR)hobj >> 16)))
    {
        return FALSE;
    }
 
    return TRUE;
}

Referenced by GdiGetHandleUserData(), and START_TEST().

◆ GdiIsHandleValidEx()

BOOL GdiIsHandleValidEx	(	_In_ HGDIOBJ	hobj,
		_In_ GDILOOBJTYPE	ObjectType
	)

Definition at line 72 of file gditools.c.

{
    PENTRY pentHmgr = GdiQueryTable();
    USHORT Index = (ULONG_PTR)hobj & 0xFFFF;
    PENTRY pentry = &pentHmgr[Index];
 
    if ((pentry->einfo.pobj == NULL) ||
        ((LONG_PTR)pentry->einfo.pobj > 0) ||
        (pentry->FullUnique != (USHORT)((ULONG_PTR)hobj >> 16)) ||
        (pentry->Objt != (UCHAR)(ObjectType >> 16)) ||
        (pentry->Flags != (UCHAR)(ObjectType >> 24)))
    {
        return FALSE;
    }
 
    return TRUE;
}

◆ GdiQueryTable()

PENTRY GdiQueryTable ( VOID )

Definition at line 45 of file gditools.c.

{
    PTEB pTeb = NtCurrentTeb();
    PPEB pPeb = pTeb->ProcessEnvironmentBlock;
    return pPeb->GdiSharedHandleTable;
}

Referenced by GdiGetHandleUserData(), GdiIsHandleValid(), and GdiIsHandleValidEx().

◆ GdiToolsInit()

BOOL GdiToolsInit ( void )

Definition at line 192 of file gditools.c.

{
 
    /* Initialize a logical palette */
    ghpal = CreatePalette((LOGPALETTE*)&gpal);
    if (!ghpal)
    {
        printf("failed to create a palette\n");
        return FALSE;
    }
 
    if (!InitPerBitDepth(1, 9, 9, &ghbmp1, &ghdcDIB1, &ghbmpDIB1, &gpvDIB1, 0) ||
        !InitPerBitDepth(1, 9, 9, &ghbmp1_InvCol, &ghdcDIB1_InvCol, &ghbmpDIB1_InvCol, &gpvDIB1_InvCol, FL_INVERT_COLORS) ||
        !InitPerBitDepth(1, 9, 9, &ghbmp1_RB, &ghdcDIB1_RB, &ghbmpDIB1_RB, &gpvDIB1_RB, FL_RED_BLUE) ||
        !InitPerBitDepth(4, 5, 5, &ghbmp4, &ghdcDIB4, &ghbmpDIB4, &gpvDIB4, 0) ||
        !InitPerBitDepth(8, 5, 5, &ghbmp8, &ghdcDIB8, &ghbmpDIB8, &gpvDIB8, 0) ||
        !InitPerBitDepth(16, 8, 8, &ghbmp16, &ghdcDIB16, &ghbmpDIB16, &gpvDIB16, 0) ||
        !InitPerBitDepth(24, 8, 8, &ghbmp24, &ghdcDIB24, &ghbmpDIB24, &gpvDIB24, 0) ||
        !InitPerBitDepth(32, 8, 8, &ghbmp32, &ghdcDIB32, &ghbmpDIB32, &gpvDIB32, 0))
    {
        printf("failed to create objects\n");
        return FALSE;
    }
 
    gpDIB32 = gpvDIB32;
 
    /* Create an Info-DC */
    ghdcInfo = CreateDCW(L"DISPLAY", NULL, NULL, NULL);
    if (!ghdcInfo)
    {
        printf("failed to create info DC\n");
        return FALSE;
    }
 
    return TRUE;
}

Referenced by START_TEST().

◆ InitPerBitDepth()

BOOL InitPerBitDepth	(	_In_ ULONG	cBitsPerPixel,
		_In_ ULONG	cx,
		_In_ ULONG	cy,
		_Out_ HBITMAP *	phbmp,
		_Out_ HDC *	phdcDIB,
		_Out_ HBITMAP *	phbmpDIB,
		_Out_ PVOID *	ppvBits,
		_In_ ULONG	flags
	)

Definition at line 112 of file gditools.c.

{
    struct
    {
        BITMAPINFOHEADER bmiHeader;
        ULONG bmiColors[256];
    } bmiBuffer;
    LPBITMAPINFO pbmi = (LPBITMAPINFO)&bmiBuffer;
 
    /* Create a bitmap */
    *phbmp = CreateBitmap(cx, cy, 1, cBitsPerPixel, NULL);
    if (*phbmp == NULL)
    {
        printf("CreateBitmap failed %lu\n", cBitsPerPixel);
        return FALSE;
    }
 
    /* Setup bitmap info */
    memset(&bmiBuffer, 0, sizeof(bmiBuffer));
    pbmi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    pbmi->bmiHeader.biWidth = cx;
    pbmi->bmiHeader.biHeight = -(LONG)cy;
    pbmi->bmiHeader.biPlanes = 1;
    pbmi->bmiHeader.biBitCount = cBitsPerPixel;
    pbmi->bmiHeader.biCompression = BI_RGB;
    pbmi->bmiHeader.biSizeImage = 0;
    pbmi->bmiHeader.biXPelsPerMeter = 0;
    pbmi->bmiHeader.biYPelsPerMeter = 0;
    pbmi->bmiHeader.biClrUsed = 0;
    pbmi->bmiHeader.biClrImportant = 0;
 
    if (cBitsPerPixel == 1)
    {
        if (flags & FL_RED_BLUE)
        {
            bmiBuffer.bmiColors[0] = RGB(0xFF, 0x00, 0x00);
            bmiBuffer.bmiColors[1] = RGB(0x00, 0x00, 0xFF);
        }
        else if (flags & FL_INVERT_COLORS)
        {
            bmiBuffer.bmiColors[0] = 0xFFFFFF;
            bmiBuffer.bmiColors[1] = 0;
        }
        else
        {
            bmiBuffer.bmiColors[0] = 0;
            bmiBuffer.bmiColors[1] = 0xFFFFFF;
        }
        pbmi->bmiHeader.biClrUsed = 2;
    }
 
    /* Create a compatible DC for the DIB */
    *phdcDIB = CreateCompatibleDC(0);
    if (*phdcDIB == NULL)
    {
        printf("CreateCompatibleDC failed %lu\n", cBitsPerPixel);
        return FALSE;
    }
 
    /* Create the DIB section with the same values */
    *phbmpDIB = CreateDIBSection(*phdcDIB, pbmi, DIB_RGB_COLORS, ppvBits, 0, 0 );
    if (*phbmpDIB == NULL)
    {
        printf("CreateDIBSection failed. %lu\n", cBitsPerPixel);
        return FALSE;
    }
 
    SelectObject(*phdcDIB, *phbmpDIB);
 
    return TRUE;
}

Referenced by GdiToolsInit().

Variable Documentation

◆ ghbmp1

HBITMAP ghbmp1

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmp16

HBITMAP ghbmp16

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmp1_InvCol

HBITMAP ghbmp1_InvCol

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmp1_RB

HBITMAP ghbmp1_RB

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmp24

HBITMAP ghbmp24

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit(), and Test_MaskBlt_1bpp().

◆ ghbmp32

HBITMAP ghbmp32

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmp4

HBITMAP ghbmp4

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmp8

HBITMAP ghbmp8

Definition at line 21 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmpDIB1

HBITMAP ghbmpDIB1

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmpDIB16

HBITMAP ghbmpDIB16

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmpDIB1_InvCol

HBITMAP ghbmpDIB1_InvCol

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmpDIB1_RB

HBITMAP ghbmpDIB1_RB

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmpDIB24

HBITMAP ghbmpDIB24

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit().

◆ ghbmpDIB32

HBITMAP ghbmpDIB32

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit(), and Test_NtGdiSetPixel_generic().

◆ ghbmpDIB4

HBITMAP ghbmpDIB4

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit(), and Test_GetDIBits().

◆ ghbmpDIB8

HBITMAP ghbmpDIB8

Definition at line 22 of file gditools.c.

Referenced by GdiToolsInit().