Milestone 5: deliver embedded RDP sessions and lifecycle hardening
This commit is contained in:
Keith Smith
600
third_party/FreeRDP/libfreerdp/gdi/test/TestGdiCreate.c
vendored
Normal file
600
third_party/FreeRDP/libfreerdp/gdi/test/TestGdiCreate.c
vendored
Normal file
@@ -0,0 +1,600 @@
|
||||
|
||||
#include <freerdp/gdi/gdi.h>
|
||||
|
||||
#include <freerdp/gdi/dc.h>
|
||||
#include <freerdp/gdi/pen.h>
|
||||
#include <freerdp/gdi/region.h>
|
||||
#include <freerdp/gdi/bitmap.h>
|
||||
|
||||
#include <winpr/crt.h>
|
||||
#include <winpr/crypto.h>
|
||||
|
||||
#include "line.h"
|
||||
#include "brush.h"
|
||||
#include "drawing.h"
|
||||
|
||||
static const UINT32 colorFormatList[] = {
|
||||
PIXEL_FORMAT_RGB15, PIXEL_FORMAT_BGR15, PIXEL_FORMAT_RGB16, PIXEL_FORMAT_BGR16,
|
||||
PIXEL_FORMAT_RGB24, PIXEL_FORMAT_BGR24, PIXEL_FORMAT_ARGB32, PIXEL_FORMAT_ABGR32,
|
||||
PIXEL_FORMAT_XRGB32, PIXEL_FORMAT_XBGR32, PIXEL_FORMAT_RGBX32, PIXEL_FORMAT_BGRX32
|
||||
|
||||
};
|
||||
static const UINT32 colorFormatCount = sizeof(colorFormatList) / sizeof(colorFormatList[0]);
|
||||
|
||||
static int test_gdi_GetDC(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_DC hdc = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (hdc->format != PIXEL_FORMAT_XRGB32)
|
||||
goto fail;
|
||||
|
||||
if (hdc->drawMode != GDI_R2_BLACK)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
gdi_DeleteDC(hdc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreateCompatibleDC(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_DC hdc = nullptr;
|
||||
HGDI_DC chdc = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
hdc->format = PIXEL_FORMAT_RGB16;
|
||||
hdc->drawMode = GDI_R2_XORPEN;
|
||||
|
||||
if (!(chdc = gdi_CreateCompatibleDC(hdc)))
|
||||
{
|
||||
printf("gdi_CreateCompatibleDC failed\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
if (chdc->format != hdc->format)
|
||||
goto fail;
|
||||
|
||||
if (chdc->drawMode != hdc->drawMode)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
|
||||
if (chdc)
|
||||
gdi_DeleteDC(chdc);
|
||||
|
||||
gdi_DeleteDC(hdc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreateBitmap(void)
|
||||
{
|
||||
int rc = -1;
|
||||
UINT32 format = PIXEL_FORMAT_ARGB32;
|
||||
INT32 width = 0;
|
||||
INT32 height = 0;
|
||||
BYTE* data = nullptr;
|
||||
HGDI_BITMAP hBitmap = nullptr;
|
||||
width = 32;
|
||||
height = 16;
|
||||
|
||||
if (!(data = (BYTE*)winpr_aligned_malloc(4ULL * width * height, 16)))
|
||||
{
|
||||
printf("failed to allocate aligned bitmap data memory\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (!(hBitmap = gdi_CreateBitmap(width, height, format, data)))
|
||||
{
|
||||
printf("gdi_CreateBitmap failed\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
if (hBitmap->objectType != GDIOBJECT_BITMAP)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->format != format)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->width != width)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->height != height)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->data != data)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
|
||||
if (hBitmap)
|
||||
gdi_DeleteObject((HGDIOBJECT)hBitmap);
|
||||
else
|
||||
winpr_aligned_free(data);
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreateCompatibleBitmap(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_DC hdc = nullptr;
|
||||
INT32 width = 0;
|
||||
INT32 height = 0;
|
||||
HGDI_BITMAP hBitmap = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
hdc->format = PIXEL_FORMAT_ARGB32;
|
||||
width = 32;
|
||||
height = 16;
|
||||
hBitmap = gdi_CreateCompatibleBitmap(hdc, width, height);
|
||||
|
||||
if (hBitmap->objectType != GDIOBJECT_BITMAP)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->format != hdc->format)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->width != width)
|
||||
goto fail;
|
||||
|
||||
if (hBitmap->height != height)
|
||||
goto fail;
|
||||
|
||||
if (!hBitmap->data)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
|
||||
if (hBitmap)
|
||||
gdi_DeleteObject((HGDIOBJECT)hBitmap);
|
||||
|
||||
gdi_DeleteDC(hdc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreatePen(void)
|
||||
{
|
||||
int rc = -1;
|
||||
const UINT32 format = PIXEL_FORMAT_RGBA32;
|
||||
HGDI_PEN hPen = gdi_CreatePen(GDI_PS_SOLID, 8, 0xAABBCCDD, format, nullptr);
|
||||
|
||||
if (!hPen)
|
||||
{
|
||||
printf("gdi_CreatePen failed\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (hPen->style != GDI_PS_SOLID)
|
||||
goto fail;
|
||||
|
||||
if (hPen->width != 8)
|
||||
goto fail;
|
||||
|
||||
if (hPen->color != 0xAABBCCDD)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
gdi_DeleteObject((HGDIOBJECT)hPen);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreateSolidBrush(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_BRUSH hBrush = gdi_CreateSolidBrush(0xAABBCCDD);
|
||||
|
||||
if (hBrush->objectType != GDIOBJECT_BRUSH)
|
||||
goto fail;
|
||||
|
||||
if (hBrush->style != GDI_BS_SOLID)
|
||||
goto fail;
|
||||
|
||||
if (hBrush->color != 0xAABBCCDD)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
gdi_DeleteObject((HGDIOBJECT)hBrush);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreatePatternBrush(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_BRUSH hBrush = nullptr;
|
||||
HGDI_BITMAP hBitmap = nullptr;
|
||||
hBitmap = gdi_CreateBitmap(64, 64, 32, nullptr);
|
||||
hBrush = gdi_CreatePatternBrush(hBitmap);
|
||||
|
||||
if (!hBitmap || !hBrush)
|
||||
goto fail;
|
||||
|
||||
if (hBrush->objectType != GDIOBJECT_BRUSH)
|
||||
goto fail;
|
||||
|
||||
if (hBrush->style != GDI_BS_PATTERN)
|
||||
goto fail;
|
||||
|
||||
if (hBrush->pattern != hBitmap)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
|
||||
if (hBitmap)
|
||||
gdi_DeleteObject((HGDIOBJECT)hBitmap);
|
||||
|
||||
if (hBrush)
|
||||
gdi_DeleteObject((HGDIOBJECT)hBrush);
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreateRectRgn(void)
|
||||
{
|
||||
int rc = -1;
|
||||
INT32 x1 = 32;
|
||||
INT32 y1 = 64;
|
||||
INT32 x2 = 128;
|
||||
INT32 y2 = 256;
|
||||
HGDI_RGN hRegion = gdi_CreateRectRgn(x1, y1, x2, y2);
|
||||
|
||||
if (!hRegion)
|
||||
return rc;
|
||||
|
||||
if (hRegion->objectType != GDIOBJECT_REGION)
|
||||
goto fail;
|
||||
|
||||
if (hRegion->x != x1)
|
||||
goto fail;
|
||||
|
||||
if (hRegion->y != y1)
|
||||
goto fail;
|
||||
|
||||
if (hRegion->w != x2 - x1 + 1)
|
||||
goto fail;
|
||||
|
||||
if (hRegion->h != y2 - y1 + 1)
|
||||
goto fail;
|
||||
|
||||
if (hRegion->null)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
gdi_DeleteObject((HGDIOBJECT)hRegion);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_CreateRect(void)
|
||||
{
|
||||
int rc = -1;
|
||||
GDI_RECT* hRect = nullptr;
|
||||
INT32 x1 = 32;
|
||||
INT32 y1 = 64;
|
||||
INT32 x2 = 128;
|
||||
INT32 y2 = 256;
|
||||
|
||||
if (!(hRect = gdi_CreateRect(x1, y1, x2, y2)))
|
||||
{
|
||||
printf("gdi_CreateRect failed\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (hRect->objectType != GDIOBJECT_RECT)
|
||||
goto fail;
|
||||
|
||||
if (hRect->left != x1)
|
||||
goto fail;
|
||||
|
||||
if (hRect->top != y1)
|
||||
goto fail;
|
||||
|
||||
if (hRect->right != x2)
|
||||
goto fail;
|
||||
|
||||
if (hRect->bottom != y2)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
gdi_DeleteObject((HGDIOBJECT)hRect);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static BYTE prand(void)
|
||||
{
|
||||
BYTE tmp = 0;
|
||||
if (winpr_RAND(&tmp, sizeof(tmp)) < 0)
|
||||
{
|
||||
(void)fprintf(stderr, "winpr_RAND faild,retry...\n");
|
||||
// NOLINTNEXTLINE(concurrency-mt-unsafe)
|
||||
exit(-1);
|
||||
}
|
||||
return tmp;
|
||||
}
|
||||
|
||||
static BOOL test_gdi_GetPixel(void)
|
||||
{
|
||||
BOOL rc = TRUE;
|
||||
|
||||
for (UINT32 x = 0; x < colorFormatCount; x++)
|
||||
{
|
||||
UINT32 bpp = 0;
|
||||
HGDI_DC hdc = nullptr;
|
||||
UINT32 width = 128;
|
||||
UINT32 height = 64;
|
||||
HGDI_BITMAP hBitmap = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
hdc->format = colorFormatList[x];
|
||||
hBitmap = gdi_CreateCompatibleBitmap(hdc, width, height);
|
||||
|
||||
if (!hBitmap)
|
||||
{
|
||||
gdi_DeleteDC(hdc);
|
||||
return -1;
|
||||
}
|
||||
|
||||
gdi_SelectObject(hdc, (HGDIOBJECT)hBitmap);
|
||||
bpp = FreeRDPGetBytesPerPixel(hBitmap->format);
|
||||
|
||||
for (UINT32 i = 0; i < height; i++)
|
||||
{
|
||||
for (UINT32 j = 0; j < width; j++)
|
||||
{
|
||||
UINT32 pixel = 0;
|
||||
const UINT32 color =
|
||||
FreeRDPGetColor(hBitmap->format, prand(), prand(), prand(), prand());
|
||||
FreeRDPWriteColor(&hBitmap->data[i * hBitmap->scanline + j * bpp], hBitmap->format,
|
||||
color);
|
||||
pixel = gdi_GetPixel(hdc, j, i);
|
||||
|
||||
if (pixel != color)
|
||||
{
|
||||
rc = FALSE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!rc)
|
||||
break;
|
||||
}
|
||||
|
||||
gdi_DeleteObject((HGDIOBJECT)hBitmap);
|
||||
gdi_DeleteDC(hdc);
|
||||
}
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static BOOL test_gdi_SetPixel(void)
|
||||
{
|
||||
BOOL rc = TRUE;
|
||||
|
||||
for (UINT32 x = 0; x < colorFormatCount; x++)
|
||||
{
|
||||
UINT32 bpp = 0;
|
||||
HGDI_DC hdc = nullptr;
|
||||
UINT32 width = 128;
|
||||
UINT32 height = 64;
|
||||
HGDI_BITMAP hBitmap = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
hdc->format = colorFormatList[x];
|
||||
hBitmap = gdi_CreateCompatibleBitmap(hdc, width, height);
|
||||
gdi_SelectObject(hdc, (HGDIOBJECT)hBitmap);
|
||||
bpp = FreeRDPGetBytesPerPixel(hBitmap->format);
|
||||
|
||||
for (UINT32 i = 0; i < height; i++)
|
||||
{
|
||||
for (UINT32 j = 0; j < width; j++)
|
||||
{
|
||||
UINT32 pixel = 0;
|
||||
const UINT32 color =
|
||||
FreeRDPGetColor(hBitmap->format, prand(), prand(), prand(), prand());
|
||||
gdi_SetPixel(hdc, j, i, color);
|
||||
pixel = FreeRDPReadColor(&hBitmap->data[i * hBitmap->scanline + j * bpp],
|
||||
hBitmap->format);
|
||||
|
||||
if (pixel != color)
|
||||
{
|
||||
rc = FALSE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!rc)
|
||||
break;
|
||||
}
|
||||
|
||||
gdi_DeleteObject((HGDIOBJECT)hBitmap);
|
||||
gdi_DeleteDC(hdc);
|
||||
}
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_SetROP2(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_DC hdc = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
gdi_SetROP2(hdc, GDI_R2_BLACK);
|
||||
|
||||
if (hdc->drawMode != GDI_R2_BLACK)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
gdi_DeleteDC(hdc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int test_gdi_MoveToEx(void)
|
||||
{
|
||||
int rc = -1;
|
||||
HGDI_DC hdc = nullptr;
|
||||
HGDI_PEN hPen = nullptr;
|
||||
HGDI_POINT prevPoint = nullptr;
|
||||
const UINT32 format = PIXEL_FORMAT_RGBA32;
|
||||
gdiPalette* palette = nullptr;
|
||||
|
||||
if (!(hdc = gdi_GetDC()))
|
||||
{
|
||||
printf("failed to get gdi device context\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (!(hPen = gdi_CreatePen(GDI_PS_SOLID, 8, 0xAABBCCDD, format, palette)))
|
||||
{
|
||||
printf("gdi_CreatePen failed\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
gdi_SelectObject(hdc, (HGDIOBJECT)hPen);
|
||||
gdi_MoveToEx(hdc, 128, 256, nullptr);
|
||||
|
||||
if (hdc->pen->posX != 128)
|
||||
goto fail;
|
||||
|
||||
if (hdc->pen->posY != 256)
|
||||
goto fail;
|
||||
|
||||
prevPoint = (HGDI_POINT)malloc(sizeof(GDI_POINT));
|
||||
ZeroMemory(prevPoint, sizeof(GDI_POINT));
|
||||
gdi_MoveToEx(hdc, 64, 128, prevPoint);
|
||||
|
||||
if (prevPoint->x != 128)
|
||||
goto fail;
|
||||
|
||||
if (prevPoint->y != 256)
|
||||
goto fail;
|
||||
|
||||
if (hdc->pen->posX != 64)
|
||||
goto fail;
|
||||
|
||||
if (hdc->pen->posY != 128)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
|
||||
if (hPen)
|
||||
gdi_DeleteObject((HGDIOBJECT)hPen);
|
||||
|
||||
free(prevPoint);
|
||||
gdi_DeleteDC(hdc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
int TestGdiCreate(int argc, char* argv[])
|
||||
{
|
||||
WINPR_UNUSED(argc);
|
||||
WINPR_UNUSED(argv);
|
||||
(void)fprintf(stderr, "test_gdi_GetDC()\n");
|
||||
|
||||
if (test_gdi_GetDC() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreateCompatibleDC()\n");
|
||||
|
||||
if (test_gdi_CreateCompatibleDC() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreateBitmap()\n");
|
||||
|
||||
if (test_gdi_CreateBitmap() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreateCompatibleBitmap()\n");
|
||||
|
||||
if (test_gdi_CreateCompatibleBitmap() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreatePen()\n");
|
||||
|
||||
if (test_gdi_CreatePen() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreateSolidBrush()\n");
|
||||
|
||||
if (test_gdi_CreateSolidBrush() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreatePatternBrush()\n");
|
||||
|
||||
if (test_gdi_CreatePatternBrush() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreateRectRgn()\n");
|
||||
|
||||
if (test_gdi_CreateRectRgn() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_CreateRect()\n");
|
||||
|
||||
if (test_gdi_CreateRect() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_GetPixel()\n");
|
||||
|
||||
if (!test_gdi_GetPixel())
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_SetPixel()\n");
|
||||
|
||||
if (!test_gdi_SetPixel())
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_SetROP2()\n");
|
||||
|
||||
if (test_gdi_SetROP2() < 0)
|
||||
return -1;
|
||||
|
||||
(void)fprintf(stderr, "test_gdi_MoveToEx()\n");
|
||||
|
||||
if (test_gdi_MoveToEx() < 0)
|
||||
return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user