FreeRDP/libfreerdp/codec/test/TestFreeRDPCodecZGfx.c

#include <winpr/crt.h>
#include <winpr/print.h>
#include <winpr/bitstream.h>

#include <freerdp/freerdp.h>
#include <freerdp/codec/zgfx.h>
#include <freerdp/log.h>

/* Sample from [MS-RDPEGFX] */
static const BYTE TEST_FOX_DATA[] =
	"The quick brown "
	"fox jumps over t"
  	"he lazy dog";

static const BYTE TEST_FOX_DATA_SINGLE[] =
	"\xE0\x04\x54\x68\x65\x20\x71\x75\x69\x63\x6B\x20\x62\x72\x6F\x77"
	"\x6E\x20\x66\x6F\x78\x20\x6A\x75\x6D\x70\x73\x20\x6F\x76\x65\x72"
	"\x20\x74\x68\x65\x20\x6C\x61\x7A\x79\x20\x64\x6F\x67";

static const BYTE TEST_FOX_DATA_MULTIPART[] =
	"\xE1\x03\x00\x2B\x00\x00\x00\x11\x00\x00\x00\x04\x54\x68\x65\x20"
	"\x71\x75\x69\x63\x6B\x20\x62\x72\x6F\x77\x6E\x20\x0E\x00\x00\x00"
	"\x04\x66\x6F\x78\x20\x6A\x75\x6D\x70\x73\x20\x6F\x76\x65\x10\x00"
	"\x00\x00\x24\x39\x08\x0E\x91\xF8\xD8\x61\x3D\x1E\x44\x06\x43\x79"
	"\x9C\x02";

int test_ZGfxCompressFox()
{
	int status;
	UINT32 Flags;
	BYTE* pSrcData;
	UINT32 SrcSize;
	UINT32 DstSize;
	BYTE* pDstData;
	ZGFX_CONTEXT* zgfx;
	UINT32 expectedSize;

	zgfx = zgfx_context_new(TRUE);
	SrcSize = sizeof(TEST_FOX_DATA) - 1;
	pSrcData = (BYTE*) TEST_FOX_DATA;
	Flags = 0;
	expectedSize = sizeof(TEST_FOX_DATA_SINGLE) - 1;
	status = zgfx_compress(zgfx, pSrcData, SrcSize, &pDstData, &DstSize, &Flags);
	printf("flags: 0x%04X size: %d\n", Flags, DstSize);

	if (DstSize != expectedSize)
	{
		printf("test_ZGfxCompressFox: output size mismatch: Actual: %d, Expected: %d\n", DstSize, expectedSize);
		return -1;
	}

	if (memcmp(pDstData, TEST_FOX_DATA_SINGLE, DstSize) != 0)
	{
		printf("test_ZGfxCompressFox: output mismatch\n");
		printf("Actual\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData, DstSize * 8, 0);
		printf("Expected\n");
		BitDump(__FUNCTION__, WLOG_INFO, TEST_FOX_DATA_SINGLE, DstSize * 8, 0);
		return -1;
	}

	zgfx_context_free(zgfx);
	return 0;
}

int test_ZGfxDecompressFoxSingle()
{
	int status;
	UINT32 Flags;
	BYTE* pSrcData;
	UINT32 SrcSize;
	UINT32 DstSize;
	BYTE* pDstData;
	ZGFX_CONTEXT* zgfx;
	UINT32 expectedSize;

	zgfx = zgfx_context_new(TRUE);
	SrcSize = sizeof(TEST_FOX_DATA_SINGLE) - 1;
	pSrcData = (BYTE*) TEST_FOX_DATA_SINGLE;
	Flags = 0;
	expectedSize = sizeof(TEST_FOX_DATA) - 1;
	status = zgfx_decompress(zgfx, pSrcData, SrcSize, &pDstData, &DstSize, Flags);
	printf("flags: 0x%04X size: %d\n", Flags, DstSize);

	if (DstSize != expectedSize)
	{
		printf("test_ZGfxDecompressFoxSingle: output size mismatch: Actual: %d, Expected: %d\n", DstSize, expectedSize);
		return -1;
	}

	if (memcmp(pDstData, TEST_FOX_DATA, DstSize) != 0)
	{
		printf("test_ZGfxDecompressFoxSingle: output mismatch\n");
		printf("Actual\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData, DstSize * 8, 0);
		printf("Expected\n");
		BitDump(__FUNCTION__, WLOG_INFO, TEST_FOX_DATA, DstSize * 8, 0);
		return -1;
	}

	zgfx_context_free(zgfx);
	return 0;
}

int test_ZGfxDecompressFoxMultipart()
{
	int status;
	UINT32 Flags;
	BYTE* pSrcData;
	UINT32 SrcSize;
	UINT32 DstSize;
	BYTE* pDstData;
	ZGFX_CONTEXT* zgfx;
	UINT32 expectedSize;

	zgfx = zgfx_context_new(TRUE);
	SrcSize = sizeof(TEST_FOX_DATA_MULTIPART) - 1;
	pSrcData = (BYTE*) TEST_FOX_DATA_MULTIPART;
	Flags = 0;
	expectedSize = sizeof(TEST_FOX_DATA) - 1;
	status = zgfx_decompress(zgfx, pSrcData, SrcSize, &pDstData, &DstSize, Flags);
	printf("flags: 0x%04X size: %d\n", Flags, DstSize);

	if (DstSize != expectedSize)
	{
		printf("test_ZGfxDecompressFoxSingle: output size mismatch: Actual: %d, Expected: %d\n", DstSize, expectedSize);
		return -1;
	}

	if (memcmp(pDstData, TEST_FOX_DATA, DstSize) != 0)
	{
		printf("test_ZGfxDecompressFoxSingle: output mismatch\n");
		printf("Actual\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData, DstSize * 8, 0);
		printf("Expected\n");
		BitDump(__FUNCTION__, WLOG_INFO, TEST_FOX_DATA, DstSize * 8, 0);
		return -1;
	}

	zgfx_context_free(zgfx);
	return 0;
}

int test_ZGfxCompressConsistent()
{
	int status;
	UINT32 Flags;
	BYTE* pSrcData;
	UINT32 SrcSize;
	UINT32 DstSize;
	BYTE* pDstData;
	UINT32 DstSize2;
	BYTE* pDstData2;
	ZGFX_CONTEXT* zgfx;
	UINT32 expectedSize;
	BYTE BigBuffer[65536];

	memset(BigBuffer, 0xaa, sizeof(BigBuffer));
	memcpy(BigBuffer, TEST_FOX_DATA, sizeof(TEST_FOX_DATA) - 1);
	zgfx = zgfx_context_new(TRUE);
	
	/* Compress */
	expectedSize = SrcSize = sizeof(BigBuffer);
	pSrcData = (BYTE*) BigBuffer;
	Flags = 0;
	status = zgfx_compress(zgfx, pSrcData, SrcSize, &pDstData2, &DstSize2, &Flags);
	printf("Compress: flags: 0x%04X size: %d\n", Flags, DstSize2);

	/* Decompress */
	status = zgfx_decompress(zgfx, pDstData2, DstSize2, &pDstData, &DstSize, Flags);
	printf("Decompress: flags: 0x%04X size: %d\n", Flags, DstSize);

	if (DstSize != expectedSize)
	{
		printf("test_ZGfxDecompressFoxSingle: output size mismatch: Actual: %d, Expected: %d\n", DstSize, expectedSize);
		return -1;
	}

	if (memcmp(pDstData, BigBuffer, DstSize) != 0)
	{
		printf("test_ZGfxDecompressFoxSingle: output mismatch\n");
		printf("Actual\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData, 64 * 8, 0);
		printf("...\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData + DstSize - 64, 64 * 8, 0);
		printf("Expected\n");
		BitDump(__FUNCTION__, WLOG_INFO, BigBuffer, 64 * 8, 0);
		printf("...\n");
		BitDump(__FUNCTION__, WLOG_INFO, BigBuffer + DstSize - 64, 64 * 8, 0);
		printf("Middle Result\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData2, 64 * 8, 0);
		printf("...\n");
		BitDump(__FUNCTION__, WLOG_INFO, pDstData2 + DstSize2 - 64, 64 * 8, 0);
		return -1;
	}

	zgfx_context_free(zgfx);
	return 0;
}

int TestFreeRDPCodecZGfx(int argc, char* argv[])
{
	if (test_ZGfxCompressFox() < 0)
		return -1;

	if (test_ZGfxDecompressFoxSingle() < 0)
		return -1;

	if (test_ZGfxDecompressFoxMultipart() < 0)
		return -1;

	if (test_ZGfxCompressConsistent() < 0)
		return -1;

	return 0;
}