100 lines · c
1/*2 * memcpy test.3 *4 * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.5 * See https://llvm.org/LICENSE.txt for license information.6 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception7 */8 9#include <stdint.h>10#include <stdio.h>11#include <stdlib.h>12#include <string.h>13#include "stringlib.h"14 15static const struct fun16{17 const char *name;18 void *(*fun)(void *, const void *, size_t);19} funtab[] = {20#define F(x) {#x, x},21F(memcpy)22#if __aarch64__23F(__memcpy_aarch64)24# if __ARM_NEON25F(__memcpy_aarch64_simd)26# endif27#elif __arm__28F(__memcpy_arm)29#endif30#undef F31 {0, 0}32};33 34static int test_status;35#define ERR(...) (test_status=1, printf(__VA_ARGS__))36 37#define A 3238#define LEN 25000039static unsigned char dbuf[LEN+2*A];40static unsigned char sbuf[LEN+2*A];41static unsigned char wbuf[LEN+2*A];42 43static void *alignup(void *p)44{45 return (void*)(((uintptr_t)p + A-1) & -A);46}47 48static void test(const struct fun *fun, int dalign, int salign, int len)49{50 unsigned char *src = alignup(sbuf);51 unsigned char *dst = alignup(dbuf);52 unsigned char *want = wbuf;53 unsigned char *s = src + salign;54 unsigned char *d = dst + dalign;55 unsigned char *w = want + dalign;56 void *p;57 int i;58 59 if (len > LEN || dalign >= A || salign >= A)60 abort();61 for (i = 0; i < len+A; i++) {62 src[i] = '?';63 want[i] = dst[i] = '*';64 }65 for (i = 0; i < len; i++)66 s[i] = w[i] = 'a' + i%23;67 68 p = fun->fun(d, s, len);69 if (p != d)70 ERR("%s(%p,..) returned %p\n", fun->name, d, p);71 for (i = 0; i < len+A; i++) {72 if (dst[i] != want[i]) {73 ERR("%s(align %d, align %d, %d) failed\n", fun->name, dalign, salign, len);74 ERR("got : %.*s\n", dalign+len+1, dst);75 ERR("want: %.*s\n", dalign+len+1, want);76 break;77 }78 }79}80 81int main()82{83 int r = 0;84 for (int i=0; funtab[i].name; i++) {85 test_status = 0;86 for (int d = 0; d < A; d++)87 for (int s = 0; s < A; s++) {88 int n;89 for (n = 0; n < 100; n++)90 test(funtab+i, d, s, n);91 for (; n < LEN; n *= 2)92 test(funtab+i, d, s, n);93 }94 printf("%s %s\n", test_status ? "FAIL" : "PASS", funtab[i].name);95 if (test_status)96 r = -1;97 }98 return r;99}100