1204431Sraj/* 2204431Sraj * strchrnul test. 3204431Sraj * 4204431Sraj * Copyright (c) 2019-2020, Arm Limited. 5204431Sraj * SPDX-License-Identifier: MIT OR Apache-2.0 WITH LLVM-exception 6204431Sraj */ 7204431Sraj 8204431Sraj#ifndef _GNU_SOURCE 9204431Sraj#define _GNU_SOURCE 10204431Sraj#endif 11204431Sraj 12204431Sraj#include <stdint.h> 13204431Sraj#include <stdio.h> 14204431Sraj#include <stdlib.h> 15204431Sraj#include <string.h> 16204431Sraj#include <limits.h> 17204431Sraj#include "mte.h" 18204431Sraj#include "stringlib.h" 19204431Sraj#include "stringtest.h" 20204431Sraj 21204431Sraj#define F(x, mte) {#x, x, mte}, 22204431Sraj 23204431Srajstatic const struct fun 24204431Sraj{ 25204431Sraj const char *name; 26204431Sraj char *(*fun) (const char *s, int c); 27204431Sraj int test_mte; 28204431Sraj} funtab[] = { 29204431Sraj // clang-format off 30204431Sraj F(strchrnul, 0) 31204431Sraj#if __aarch64__ 32204431Sraj F(__strchrnul_aarch64, 0) 33204431Sraj F(__strchrnul_aarch64_mte, 1) 34204431Sraj# if __ARM_FEATURE_SVE 35204431Sraj F(__strchrnul_aarch64_sve, 1) 36204431Sraj# endif 37204431Sraj#endif 38204431Sraj {0, 0, 0} 39204431Sraj // clang-format on 40204431Sraj}; 41204431Sraj#undef F 42204431Sraj 43204431Sraj#define ALIGN 32 44204431Sraj#define LEN 512 45204431Srajstatic char *sbuf; 46204431Sraj 47204431Srajstatic void * 48204431Srajalignup (void *p) 49204431Sraj{ 50204431Sraj return (void *) (((uintptr_t) p + ALIGN - 1) & -ALIGN); 51204431Sraj} 52204431Sraj 53204431Srajstatic void 54204431Srajtest (const struct fun *fun, int align, int seekpos, int len) 55204431Sraj{ 56204431Sraj char *src = alignup (sbuf); 57204431Sraj char *s = src + align; 58204431Sraj char *f = seekpos != -1 ? s + seekpos : s + len; 59204431Sraj int seekchar = 0x1; 60204431Sraj void *p; 61204431Sraj 62204431Sraj if (err_count >= ERR_LIMIT) 63204431Sraj return; 64204431Sraj if (len > LEN || seekpos >= len || align >= ALIGN) 65204431Sraj abort (); 66204431Sraj 67204431Sraj for (int i = 0; src + i < s; i++) 68204431Sraj src[i] = (i + len) & 1 ? seekchar : 0; 69204431Sraj for (int i = 1; i <= ALIGN; i++) 70204431Sraj s[len + i] = (i + len) & 1 ? seekchar : 0; 71204431Sraj for (int i = 0; i < len; i++) 72204431Sraj s[i] = 'a' + (i & 31); 73204431Sraj if (seekpos != -1) 74204431Sraj s[seekpos] = seekchar; 75204431Sraj if (seekpos != -1 && (len + align) & 1) 76204431Sraj s[seekpos + 1] = seekchar; 77204431Sraj s[len] = '\0'; 78204431Sraj 79204431Sraj int mte_len = seekpos != -1 ? seekpos + 1 : len + 1; 80204431Sraj s = tag_buffer (s, mte_len, fun->test_mte); 81204431Sraj p = fun->fun (s, seekchar); 82204431Sraj untag_buffer (s, mte_len, fun->test_mte); 83204431Sraj p = untag_pointer (p); 84204431Sraj 85204431Sraj if (p != f) 86204431Sraj { 87204431Sraj ERR ("%s (%p, 0x%02x) len %d returned %p, expected %p pos %d\n", 88204431Sraj fun->name, s, seekchar, len, p, f, seekpos); 89204431Sraj quote ("input", s, len); 90204431Sraj } 91204431Sraj 92204431Sraj s = tag_buffer (s, len + 1, fun->test_mte); 93204431Sraj p = fun->fun (s, 0); 94204431Sraj untag_buffer (s, len + 1, fun->test_mte); 95204431Sraj 96204431Sraj if (p != s + len) 97204431Sraj { 98204431Sraj ERR ("%s (%p, 0x%02x) len %d returned %p, expected %p pos %d\n", 99204431Sraj fun->name, s, 0, len, p, f, len); 100204431Sraj quote ("input", s, len); 101204431Sraj } 102204431Sraj} 103204431Sraj 104204431Srajint 105204431Srajmain (void) 106204431Sraj{ 107204431Sraj sbuf = mte_mmap (LEN + 3 * ALIGN); 108204431Sraj int r = 0; 109204431Sraj for (int i = 0; funtab[i].name; i++) 110204431Sraj { 111204431Sraj err_count = 0; 112204431Sraj for (int a = 0; a < ALIGN; a++) 113204431Sraj for (int n = 0; n < LEN; n++) 114204431Sraj { 115204431Sraj for (int sp = 0; sp < n; sp++) 116204431Sraj test (funtab + i, a, sp, n); 117204431Sraj test (funtab + i, a, -1, n); 118204431Sraj } 119204431Sraj 120204431Sraj char *pass = funtab[i].test_mte && mte_enabled () ? "MTE PASS" : "PASS"; 121204431Sraj printf ("%s %s\n", err_count ? "FAIL" : pass, funtab[i].name); 122204431Sraj if (err_count) 123204431Sraj r = -1; 124204431Sraj } 125204431Sraj return r; 126204431Sraj} 127204431Sraj