1// SPDX-License-Identifier: GPL-2.0
  2#include <malloc.h>
  3#include <stdlib.h>
  4#include <string.h>
  5#include <time.h>
  6#include "utils.h"
  7
  8#define SIZE 256
  9#define ITERATIONS 1000
 10#define ITERATIONS_BENCH 100000
 11
 12int test_strlen(const void *s);
 13
 14/* test all offsets and lengths */
 15static void test_one(char *s)
 16{
 17	unsigned long offset;
 18
 19	for (offset = 0; offset < SIZE; offset++) {
 20		int x, y;
 21		unsigned long i;
 22
 23		y = strlen(s + offset);
 24		x = test_strlen(s + offset);
 25
 26		if (x != y) {
 27			printf("strlen() returned %d, should have returned %d (%p offset %ld)\n", x, y, s, offset);
 28
 29			for (i = offset; i < SIZE; i++)
 30				printf("%02x ", s[i]);
 31			printf("\n");
 32		}
 33	}
 34}
 35
 36static void bench_test(char *s)
 37{
 38	struct timespec ts_start, ts_end;
 39	int i;
 40
 41	clock_gettime(CLOCK_MONOTONIC, &ts_start);
 42
 43	for (i = 0; i < ITERATIONS_BENCH; i++)
 44		test_strlen(s);
 45
 46	clock_gettime(CLOCK_MONOTONIC, &ts_end);
 47
 48	printf("len %3.3d : time = %.6f\n", test_strlen(s), ts_end.tv_sec - ts_start.tv_sec + (ts_end.tv_nsec - ts_start.tv_nsec) / 1e9);
 49}
 50
 51static int testcase(void)
 52{
 53	char *s;
 54	unsigned long i;
 55
 56	s = memalign(128, SIZE);
 57	if (!s) {
 58		perror("memalign");
 59		exit(1);
 60	}
 61
 62	srandom(1);
 63
 64	memset(s, 0, SIZE);
 65	for (i = 0; i < SIZE; i++) {
 66		char c;
 67
 68		do {
 69			c = random() & 0x7f;
 70		} while (!c);
 71		s[i] = c;
 72		test_one(s);
 73	}
 74
 75	for (i = 0; i < ITERATIONS; i++) {
 76		unsigned long j;
 77
 78		for (j = 0; j < SIZE; j++) {
 79			char c;
 80
 81			do {
 82				c = random() & 0x7f;
 83			} while (!c);
 84			s[j] = c;
 85		}
 86		for (j = 0; j < sizeof(long); j++) {
 87			s[SIZE - 1 - j] = 0;
 88			test_one(s);
 89		}
 90	}
 91
 92	for (i = 0; i < SIZE; i++) {
 93		char c;
 94
 95		do {
 96			c = random() & 0x7f;
 97		} while (!c);
 98		s[i] = c;
 99	}
100
101	bench_test(s);
102
103	s[16] = 0;
104	bench_test(s);
105
106	s[8] = 0;
107	bench_test(s);
108
109	s[4] = 0;
110	bench_test(s);
111
112	s[3] = 0;
113	bench_test(s);
114
115	s[2] = 0;
116	bench_test(s);
117
118	s[1] = 0;
119	bench_test(s);
120
121	return 0;
122}
123
124int main(void)
125{
126	return test_harness(testcase, "strlen");
127}