static int rdpgfx_on_data_received(IWTSVirtualChannelCallback* pChannelCallback, wStream* data) { wStream* s; int status = 0; UINT32 DstSize = 0; BYTE* pDstData = NULL; RDPGFX_CHANNEL_CALLBACK* callback = (RDPGFX_CHANNEL_CALLBACK*) pChannelCallback; RDPGFX_PLUGIN* gfx = (RDPGFX_PLUGIN*) callback->plugin; status = zgfx_decompress(gfx->zgfx, Stream_Pointer(data), Stream_GetRemainingLength(data), &pDstData, &DstSize, 0); if (status < 0) { WLog_DBG(TAG, "zgfx_decompress failure! status: %d", status); return 0; } s = Stream_New(pDstData, DstSize); while (((size_t) Stream_GetPosition(s)) < Stream_Length(s)) { status = rdpgfx_recv_pdu(callback, s); if (status < 0) break; } Stream_Free(s, TRUE); //free(Stream_Buffer(data)); //Stream_Free(data,TRUE); return status; }
static int test_ZGfxDecompressFoxSingle(void) { int rc = -1; int status; UINT32 Flags; BYTE* pSrcData; UINT32 SrcSize; UINT32 DstSize; BYTE* pDstData = NULL; ZGFX_CONTEXT* zgfx; UINT32 expectedSize; zgfx = zgfx_context_new(TRUE); if (!zgfx) return -1; 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); if (status < 0) goto fail; printf("flags: 0x%08"PRIX32" size: %"PRIu32"\n", Flags, DstSize); if (DstSize != expectedSize) { printf("test_ZGfxDecompressFoxSingle: output size mismatch: Actual: %"PRIu32", Expected: %"PRIu32"\n", DstSize, expectedSize); goto fail; } 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); goto fail; } rc = 0; fail: free(pDstData); zgfx_context_free(zgfx); return rc; }
/** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT rdpgfx_on_data_received(IWTSVirtualChannelCallback* pChannelCallback, wStream* data) { wStream* s; int status = 0; UINT32 DstSize = 0; BYTE* pDstData = NULL; RDPGFX_CHANNEL_CALLBACK* callback = (RDPGFX_CHANNEL_CALLBACK*) pChannelCallback; RDPGFX_PLUGIN* gfx = (RDPGFX_PLUGIN*) callback->plugin; UINT error = CHANNEL_RC_OK; status = zgfx_decompress(gfx->zgfx, Stream_Pointer(data), Stream_GetRemainingLength(data), &pDstData, &DstSize, 0); if (status < 0) { WLog_ERR(TAG, "zgfx_decompress failure! status: %d", status); return ERROR_INTERNAL_ERROR; } s = Stream_New(pDstData, DstSize); if (!s) { WLog_ERR(TAG, "calloc failed!"); return CHANNEL_RC_NO_MEMORY; } while (((size_t) Stream_GetPosition(s)) < Stream_Length(s)) { if ((error = rdpgfx_recv_pdu(callback, s))) { WLog_ERR(TAG, "rdpgfx_recv_pdu failed with error %u!", error); break; } } Stream_Free(s, TRUE); return error; }
static int test_ZGfxCompressConsistent(void) { int rc = -1; int status; UINT32 Flags; BYTE* pSrcData; UINT32 SrcSize; UINT32 DstSize; BYTE* pDstData = NULL; UINT32 DstSize2; BYTE* pDstData2 = NULL; 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); if (!zgfx) return -1; /* Compress */ expectedSize = SrcSize = sizeof(BigBuffer); pSrcData = (BYTE*) BigBuffer; Flags = 0; status = zgfx_compress(zgfx, pSrcData, SrcSize, &pDstData2, &DstSize2, &Flags); if (status < 0) goto fail; printf("Compress: flags: 0x%08"PRIX32" size: %"PRIu32"\n", Flags, DstSize2); /* Decompress */ status = zgfx_decompress(zgfx, pDstData2, DstSize2, &pDstData, &DstSize, Flags); if (status < 0) goto fail; printf("Decompress: flags: 0x%08"PRIX32" size: %"PRIu32"\n", Flags, DstSize); if (DstSize != expectedSize) { printf("test_ZGfxDecompressFoxSingle: output size mismatch: Actual: %"PRIu32", Expected: %"PRIu32"\n", DstSize, expectedSize); goto fail; } 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); goto fail; } rc = 0; fail: free(pDstData); free(pDstData2); zgfx_context_free(zgfx); return rc; }