1*31914882SAlex Richardson /* 2*31914882SAlex Richardson * memset test. 3*31914882SAlex Richardson * 4*31914882SAlex Richardson * Copyright (c) 2019-2020, Arm Limited. 5*31914882SAlex Richardson * SPDX-License-Identifier: MIT 6*31914882SAlex Richardson */ 7*31914882SAlex Richardson 8*31914882SAlex Richardson #include <stdint.h> 9*31914882SAlex Richardson #include <stdio.h> 10*31914882SAlex Richardson #include <stdlib.h> 11*31914882SAlex Richardson #include <string.h> 12*31914882SAlex Richardson #include "mte.h" 13*31914882SAlex Richardson #include "stringlib.h" 14*31914882SAlex Richardson #include "stringtest.h" 15*31914882SAlex Richardson 16*31914882SAlex Richardson #define F(x, mte) {#x, x, mte}, 17*31914882SAlex Richardson 18*31914882SAlex Richardson static const struct fun 19*31914882SAlex Richardson { 20*31914882SAlex Richardson const char *name; 21*31914882SAlex Richardson void *(*fun) (void *s, int c, size_t n); 22*31914882SAlex Richardson int test_mte; 23*31914882SAlex Richardson } funtab[] = { 24*31914882SAlex Richardson // clang-format off 25*31914882SAlex Richardson F(memset, 0) 26*31914882SAlex Richardson #if __aarch64__ 27*31914882SAlex Richardson F(__memset_aarch64, 1) 28*31914882SAlex Richardson #elif __arm__ 29*31914882SAlex Richardson F(__memset_arm, 0) 30*31914882SAlex Richardson #endif 31*31914882SAlex Richardson {0, 0, 0} 32*31914882SAlex Richardson // clang-format on 33*31914882SAlex Richardson }; 34*31914882SAlex Richardson #undef F 35*31914882SAlex Richardson 36*31914882SAlex Richardson #define A 32 37*31914882SAlex Richardson #define LEN 250000 38*31914882SAlex Richardson static unsigned char *sbuf; 39*31914882SAlex Richardson 40*31914882SAlex Richardson static void * 41*31914882SAlex Richardson alignup (void *p) 42*31914882SAlex Richardson { 43*31914882SAlex Richardson return (void *) (((uintptr_t) p + A - 1) & -A); 44*31914882SAlex Richardson } 45*31914882SAlex Richardson 46*31914882SAlex Richardson static void 47*31914882SAlex Richardson test (const struct fun *fun, int salign, int c, int len) 48*31914882SAlex Richardson { 49*31914882SAlex Richardson unsigned char *src = alignup (sbuf); 50*31914882SAlex Richardson unsigned char *s = src + salign; 51*31914882SAlex Richardson void *p; 52*31914882SAlex Richardson int i; 53*31914882SAlex Richardson 54*31914882SAlex Richardson if (err_count >= ERR_LIMIT) 55*31914882SAlex Richardson return; 56*31914882SAlex Richardson if (len > LEN || salign >= A) 57*31914882SAlex Richardson abort (); 58*31914882SAlex Richardson for (i = 0; i < len + A; i++) 59*31914882SAlex Richardson src[i] = '?'; 60*31914882SAlex Richardson for (i = 0; i < len; i++) 61*31914882SAlex Richardson s[i] = 'a' + i % 23; 62*31914882SAlex Richardson 63*31914882SAlex Richardson s = tag_buffer (s, len, fun->test_mte); 64*31914882SAlex Richardson p = fun->fun (s, c, len); 65*31914882SAlex Richardson untag_buffer (s, len, fun->test_mte); 66*31914882SAlex Richardson 67*31914882SAlex Richardson if (p != s) 68*31914882SAlex Richardson ERR ("%s(%p,..) returned %p\n", fun->name, s, p); 69*31914882SAlex Richardson 70*31914882SAlex Richardson for (i = 0; i < salign; i++) 71*31914882SAlex Richardson { 72*31914882SAlex Richardson if (src[i] != '?') 73*31914882SAlex Richardson { 74*31914882SAlex Richardson ERR ("%s(align %d, %d, %d) failed\n", fun->name, salign, c, len); 75*31914882SAlex Richardson quoteat ("got", src, len + A, i); 76*31914882SAlex Richardson return; 77*31914882SAlex Richardson } 78*31914882SAlex Richardson } 79*31914882SAlex Richardson for (; i < salign + len; i++) 80*31914882SAlex Richardson { 81*31914882SAlex Richardson if (src[i] != (unsigned char) c) 82*31914882SAlex Richardson { 83*31914882SAlex Richardson ERR ("%s(align %d, %d, %d) failed\n", fun->name, salign, c, len); 84*31914882SAlex Richardson quoteat ("got", src, len + A, i); 85*31914882SAlex Richardson return; 86*31914882SAlex Richardson } 87*31914882SAlex Richardson } 88*31914882SAlex Richardson for (; i < len + A; i++) 89*31914882SAlex Richardson { 90*31914882SAlex Richardson if (src[i] != '?') 91*31914882SAlex Richardson { 92*31914882SAlex Richardson ERR ("%s(align %d, %d, %d) failed\n", fun->name, salign, c, len); 93*31914882SAlex Richardson quoteat ("got", src, len + A, i); 94*31914882SAlex Richardson return; 95*31914882SAlex Richardson } 96*31914882SAlex Richardson } 97*31914882SAlex Richardson } 98*31914882SAlex Richardson 99*31914882SAlex Richardson int 100*31914882SAlex Richardson main () 101*31914882SAlex Richardson { 102*31914882SAlex Richardson sbuf = mte_mmap (LEN + 2 * A); 103*31914882SAlex Richardson int r = 0; 104*31914882SAlex Richardson for (int i = 0; funtab[i].name; i++) 105*31914882SAlex Richardson { 106*31914882SAlex Richardson err_count = 0; 107*31914882SAlex Richardson for (int s = 0; s < A; s++) 108*31914882SAlex Richardson { 109*31914882SAlex Richardson int n; 110*31914882SAlex Richardson for (n = 0; n < 100; n++) 111*31914882SAlex Richardson { 112*31914882SAlex Richardson test (funtab + i, s, 0, n); 113*31914882SAlex Richardson test (funtab + i, s, 0x25, n); 114*31914882SAlex Richardson test (funtab + i, s, 0xaa25, n); 115*31914882SAlex Richardson } 116*31914882SAlex Richardson for (; n < LEN; n *= 2) 117*31914882SAlex Richardson { 118*31914882SAlex Richardson test (funtab + i, s, 0, n); 119*31914882SAlex Richardson test (funtab + i, s, 0x25, n); 120*31914882SAlex Richardson test (funtab + i, s, 0xaa25, n); 121*31914882SAlex Richardson } 122*31914882SAlex Richardson } 123*31914882SAlex Richardson char *pass = funtab[i].test_mte && mte_enabled () ? "MTE PASS" : "PASS"; 124*31914882SAlex Richardson printf ("%s %s\n", err_count ? "FAIL" : pass, funtab[i].name); 125*31914882SAlex Richardson if (err_count) 126*31914882SAlex Richardson r = -1; 127*31914882SAlex Richardson } 128*31914882SAlex Richardson return r; 129*31914882SAlex Richardson } 130