Milestone 5: deliver embedded RDP sessions and lifecycle hardening
This commit is contained in:
296
third_party/FreeRDP/libfreerdp/primitives/test/TestPrimitivesCopy.c
vendored
Normal file
296
third_party/FreeRDP/libfreerdp/primitives/test/TestPrimitivesCopy.c
vendored
Normal file
@@ -0,0 +1,296 @@
|
||||
/* test_copy.c
|
||||
* vi:ts=4 sw=4
|
||||
*
|
||||
* (c) Copyright 2012 Hewlett-Packard Development Company, L.P.
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||||
* not use this file except in compliance with the License. You may obtain
|
||||
* a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
||||
* or implied. See the License for the specific language governing
|
||||
* permissions and limitations under the License.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#include <freerdp/config.h>
|
||||
#include <winpr/crypto.h>
|
||||
|
||||
#include <winpr/sysinfo.h>
|
||||
#include "prim_test.h"
|
||||
|
||||
#define COPY_TESTSIZE (256 * 2 + 16 * 2 + 15 + 15)
|
||||
|
||||
/* ------------------------------------------------------------------------- */
|
||||
static BOOL test_copy8u_func(void)
|
||||
{
|
||||
primitives_t* prims = primitives_get();
|
||||
BYTE data[COPY_TESTSIZE + 15] = WINPR_C_ARRAY_INIT;
|
||||
if (winpr_RAND(data, sizeof(data)) < 0)
|
||||
return FALSE;
|
||||
|
||||
for (int soff = 0; soff < 16; ++soff)
|
||||
{
|
||||
for (int doff = 0; doff < 16; ++doff)
|
||||
{
|
||||
for (int length = 1; length <= COPY_TESTSIZE - doff; ++length)
|
||||
{
|
||||
BYTE dest[COPY_TESTSIZE + 15] = WINPR_C_ARRAY_INIT;
|
||||
|
||||
if (prims->copy_8u(data + soff, dest + doff, length) != PRIMITIVES_SUCCESS)
|
||||
return FALSE;
|
||||
|
||||
for (int i = 0; i < length; ++i)
|
||||
{
|
||||
if (dest[i + doff] != data[i + soff])
|
||||
{
|
||||
printf("COPY8U FAIL: off=%d len=%d, dest[%d]=0x%02" PRIx8 ""
|
||||
"data[%d]=0x%02" PRIx8 "\n",
|
||||
doff, length, i + doff, dest[i + doff], i + soff, data[i + soff]);
|
||||
return FALSE;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
/* ------------------------------------------------------------------------- */
|
||||
static BOOL test_copy8u_speed(void)
|
||||
{
|
||||
BYTE src[MAX_TEST_SIZE + 4] = WINPR_C_ARRAY_INIT;
|
||||
BYTE dst[MAX_TEST_SIZE + 4] = WINPR_C_ARRAY_INIT;
|
||||
|
||||
if (!speed_test("copy_8u", "aligned", g_Iterations, (speed_test_fkt)generic->copy_8u,
|
||||
(speed_test_fkt)optimized->copy_8u, src, dst, MAX_TEST_SIZE))
|
||||
return FALSE;
|
||||
|
||||
if (!speed_test("copy_8u", "unaligned", g_Iterations, (speed_test_fkt)generic->copy_8u,
|
||||
(speed_test_fkt)optimized->copy_8u, src + 1, dst + 1, MAX_TEST_SIZE))
|
||||
return FALSE;
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static BYTE* rand_alloc(size_t w, size_t h, size_t bpp, size_t pad, BYTE** copy)
|
||||
{
|
||||
const size_t s = w * bpp + pad;
|
||||
BYTE* ptr = calloc(s, h);
|
||||
if (!ptr)
|
||||
return nullptr;
|
||||
|
||||
if (winpr_RAND(ptr, s * h) < 0)
|
||||
{
|
||||
free(ptr);
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
if (copy)
|
||||
{
|
||||
BYTE* ptr2 = calloc(s, h);
|
||||
if (!ptr2)
|
||||
{
|
||||
free(ptr);
|
||||
return nullptr;
|
||||
}
|
||||
memcpy(ptr2, ptr, s * h);
|
||||
*copy = ptr2;
|
||||
}
|
||||
return ptr;
|
||||
}
|
||||
|
||||
static size_t runcount = 0;
|
||||
|
||||
static BOOL test_copy_no_overlap_off(BOOL verbose, UINT32 srcFormat, UINT32 dstFormat, UINT32 flags,
|
||||
UINT32 pad, UINT32 w, UINT32 h, UINT32 dxoff, UINT32 dyoff,
|
||||
UINT32 sxoff, UINT32 syoff)
|
||||
{
|
||||
BOOL rc = FALSE;
|
||||
primitives_t* gen = primitives_get_generic();
|
||||
primitives_t* prims = primitives_get();
|
||||
if (!gen || !prims)
|
||||
return FALSE;
|
||||
|
||||
runcount++;
|
||||
|
||||
WINPR_ASSERT(dxoff < w);
|
||||
WINPR_ASSERT(sxoff < w);
|
||||
WINPR_ASSERT(dyoff < h);
|
||||
WINPR_ASSERT(syoff < h);
|
||||
|
||||
const UINT32 sbpp = FreeRDPGetBytesPerPixel(srcFormat);
|
||||
const UINT32 dbpp = FreeRDPGetBytesPerPixel(dstFormat);
|
||||
|
||||
if (verbose)
|
||||
{
|
||||
(void)fprintf(stderr,
|
||||
"run src: %s, dst: %s [flags 0x%08" PRIx32 "] %" PRIu32 "x%" PRIu32
|
||||
", soff=%" PRIu32 "x%" PRIu32 ", doff=%" PRIu32 "x%" PRIu32 ", pad=%" PRIu32
|
||||
"\n",
|
||||
FreeRDPGetColorFormatName(srcFormat), FreeRDPGetColorFormatName(dstFormat),
|
||||
flags, w, h, sxoff, syoff, dxoff, dyoff, pad);
|
||||
}
|
||||
|
||||
const UINT32 sstride = (w + sxoff) * sbpp + pad;
|
||||
const UINT32 dstride = (w + dxoff) * dbpp + pad;
|
||||
BYTE* dst2 = nullptr;
|
||||
BYTE* src2 = nullptr;
|
||||
BYTE* dst1 = rand_alloc(w + dxoff, h + dyoff, dbpp, pad, &dst2);
|
||||
BYTE* src1 = rand_alloc(w + sxoff, h + syoff, sbpp, pad, &src2);
|
||||
if (!dst1 || !dst2 || !src1 || !src2)
|
||||
goto fail;
|
||||
|
||||
if (gen->copy_no_overlap(dst1, dstFormat, dstride, dxoff, dyoff, w, h, src1, srcFormat, sstride,
|
||||
sxoff, syoff, nullptr, flags) != PRIMITIVES_SUCCESS)
|
||||
goto fail;
|
||||
|
||||
if (memcmp(src1, src2, 1ULL * sstride * h) != 0)
|
||||
goto fail;
|
||||
|
||||
if (prims->copy_no_overlap(dst2, dstFormat, dstride, dxoff, dyoff, w, h, src1, srcFormat,
|
||||
sstride, sxoff, syoff, nullptr, flags) != PRIMITIVES_SUCCESS)
|
||||
goto fail;
|
||||
|
||||
if (memcmp(src1, src2, 1ULL * sstride * h) != 0)
|
||||
goto fail;
|
||||
|
||||
if (memcmp(dst1, dst2, 1ULL * dstride * h) != 0)
|
||||
goto fail;
|
||||
|
||||
if (flags == FREERDP_KEEP_DST_ALPHA)
|
||||
{
|
||||
for (size_t y = 0; y < h; y++)
|
||||
{
|
||||
const BYTE* d1 = &dst1[(y + dyoff) * dstride];
|
||||
const BYTE* d2 = &dst2[(y + dyoff) * dstride];
|
||||
for (size_t x = 0; x < w; x++)
|
||||
{
|
||||
const UINT32 c1 = FreeRDPReadColor(&d1[(x + dxoff) * dbpp], dstFormat);
|
||||
const UINT32 c2 = FreeRDPReadColor(&d2[(x + dxoff) * dbpp], dstFormat);
|
||||
BYTE a1 = 0;
|
||||
BYTE a2 = 0;
|
||||
FreeRDPSplitColor(c1, dstFormat, nullptr, nullptr, nullptr, &a1, nullptr);
|
||||
FreeRDPSplitColor(c2, dstFormat, nullptr, nullptr, nullptr, &a2, nullptr);
|
||||
if (a1 != a2)
|
||||
goto fail;
|
||||
}
|
||||
}
|
||||
}
|
||||
rc = TRUE;
|
||||
|
||||
fail:
|
||||
if (!rc)
|
||||
{
|
||||
(void)fprintf(stderr, "failed to compare copy_no_overlap(%s -> %s [0x%08" PRIx32 "])\n",
|
||||
FreeRDPGetColorFormatName(srcFormat), FreeRDPGetColorFormatName(dstFormat),
|
||||
flags);
|
||||
}
|
||||
free(dst1);
|
||||
free(dst2);
|
||||
free(src1);
|
||||
free(src2);
|
||||
return rc;
|
||||
}
|
||||
|
||||
static BOOL test_copy_no_overlap(BOOL verbose, UINT32 srcFormat, UINT32 dstFormat, UINT32 flags,
|
||||
UINT32 width, UINT32 height)
|
||||
{
|
||||
BOOL rc = TRUE;
|
||||
const UINT32 mw = 4;
|
||||
const UINT32 mh = 4;
|
||||
for (UINT32 dxoff = 0; dxoff < mw; dxoff++)
|
||||
{
|
||||
for (UINT32 dyoff = 0; dyoff <= mh; dyoff++)
|
||||
{
|
||||
for (UINT32 sxoff = 0; sxoff <= mw; sxoff++)
|
||||
{
|
||||
for (UINT32 syoff = 0; syoff <= mh; syoff++)
|
||||
{
|
||||
/* We need minimum alignment of 8 bytes.
|
||||
* AVX2 can read 8 pixels (at most 8x4=32 bytes) per step
|
||||
* if we have 24bpp input that is 24 bytes with 8 bytes read
|
||||
* out of bound */
|
||||
for (UINT32 pad = 8; pad <= 12; pad++)
|
||||
{
|
||||
if (!test_copy_no_overlap_off(verbose, srcFormat, dstFormat, flags, pad,
|
||||
width, height, dxoff, dyoff, sxoff, syoff))
|
||||
rc = FALSE;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
int TestPrimitivesCopy(int argc, char* argv[])
|
||||
{
|
||||
WINPR_UNUSED(argc);
|
||||
WINPR_UNUSED(argv);
|
||||
|
||||
const BOOL verbose = argc > 1;
|
||||
|
||||
prim_test_setup(FALSE);
|
||||
|
||||
if (!test_copy8u_func())
|
||||
return 1;
|
||||
|
||||
if (g_TestPrimitivesPerformance)
|
||||
{
|
||||
if (!test_copy8u_speed())
|
||||
return 1;
|
||||
}
|
||||
|
||||
const UINT32 flags[] = {
|
||||
FREERDP_FLIP_NONE,
|
||||
FREERDP_KEEP_DST_ALPHA,
|
||||
FREERDP_FLIP_HORIZONTAL,
|
||||
FREERDP_KEEP_DST_ALPHA | FREERDP_FLIP_HORIZONTAL,
|
||||
#if defined(TEST_ALL_FLAGS)
|
||||
FREERDP_FLIP_VERTICAL,
|
||||
FREERDP_FLIP_VERTICAL | FREERDP_FLIP_HORIZONTAL,
|
||||
FREERDP_KEEP_DST_ALPHA | FREERDP_FLIP_VERTICAL,
|
||||
FREERDP_KEEP_DST_ALPHA | FREERDP_FLIP_VERTICAL | FREERDP_FLIP_HORIZONTAL
|
||||
#endif
|
||||
};
|
||||
const UINT32 formats[] = { PIXEL_FORMAT_BGRA32,
|
||||
PIXEL_FORMAT_BGRX32,
|
||||
PIXEL_FORMAT_BGR24
|
||||
#if defined(TEST_ALL_FLAGS) /* Only the previous 3 have SIMD optimizations, so skip the rest */
|
||||
,
|
||||
PIXEL_FORMAT_RGB24,
|
||||
PIXEL_FORMAT_ABGR32,
|
||||
PIXEL_FORMAT_ARGB32,
|
||||
PIXEL_FORMAT_XBGR32,
|
||||
PIXEL_FORMAT_XRGB32,
|
||||
PIXEL_FORMAT_RGBA32,
|
||||
PIXEL_FORMAT_RGBX32
|
||||
#endif
|
||||
};
|
||||
|
||||
int rc = 0;
|
||||
for (size_t z = 0; z < ARRAYSIZE(flags); z++)
|
||||
{
|
||||
const UINT32 flag = flags[z];
|
||||
for (size_t x = 0; x < ARRAYSIZE(formats); x++)
|
||||
{
|
||||
const UINT32 sformat = formats[x];
|
||||
for (size_t y = 0; y < ARRAYSIZE(formats); y++)
|
||||
{
|
||||
const UINT32 dformat = formats[y];
|
||||
|
||||
if (!test_copy_no_overlap(verbose, sformat, dformat, flag, 21, 17))
|
||||
rc = -1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (verbose)
|
||||
(void)fprintf(stderr, "runcount=%" PRIuz "\n", runcount);
|
||||
|
||||
return rc;
|
||||
}
|
||||
Reference in New Issue
Block a user