write in 30M

hc

2024-02-20 ea08eeccae9297f7aabd2ef7f0c2517ac4549acc

 kernel/lib/test_string.c
..
..
@@ -1,3 +1,4 @@
1
+// SPDX-License-Identifier: GPL-2.0-only
1
2
 #include <linux/module.h>
2
3
 #include <linux/printk.h>
3
4
 #include <linux/slab.h>
..
..
@@ -111,6 +112,73 @@
111
112
 	return 0;
112
113
 }
113
114
 
115
+static __init int strchr_selftest(void)
116
+{
117
+	const char *test_string = "abcdefghijkl";
118
+	const char *empty_string = "";
119
+	char *result;
120
+	int i;
121
+
122
+	for (i = 0; i < strlen(test_string) + 1; i++) {
123
+		result = strchr(test_string, test_string[i]);
124
+		if (result - test_string != i)
125
+			return i + 'a';
126
+	}
127
+
128
+	result = strchr(empty_string, '\0');
129
+	if (result != empty_string)
130
+		return 0x101;
131
+
132
+	result = strchr(empty_string, 'a');
133
+	if (result)
134
+		return 0x102;
135
+
136
+	result = strchr(test_string, 'z');
137
+	if (result)
138
+		return 0x103;
139
+
140
+	return 0;
141
+}
142
+
143
+static __init int strnchr_selftest(void)
144
+{
145
+	const char *test_string = "abcdefghijkl";
146
+	const char *empty_string = "";
147
+	char *result;
148
+	int i, j;
149
+
150
+	for (i = 0; i < strlen(test_string) + 1; i++) {
151
+		for (j = 0; j < strlen(test_string) + 2; j++) {
152
+			result = strnchr(test_string, j, test_string[i]);
153
+			if (j <= i) {
154
+				if (!result)
155
+					continue;
156
+				return ((i + 'a') << 8) | j;
157
+			}
158
+			if (result - test_string != i)
159
+				return ((i + 'a') << 8) | j;
160
+		}
161
+	}
162
+
163
+	result = strnchr(empty_string, 0, '\0');
164
+	if (result)
165
+		return 0x10001;
166
+
167
+	result = strnchr(empty_string, 1, '\0');
168
+	if (result != empty_string)
169
+		return 0x10002;
170
+
171
+	result = strnchr(empty_string, 1, 'a');
172
+	if (result)
173
+		return 0x10003;
174
+
175
+	result = strnchr(NULL, 0, '\0');
176
+	if (result)
177
+		return 0x10004;
178
+
179
+	return 0;
180
+}
181
+
114
182
 static __init int string_selftest_init(void)
115
183
 {
116
184
 	int test, subtest;
..
..
@@ -130,6 +198,16 @@
130
198
 	if (subtest)
131
199
 		goto fail;
132
200
 
201
+	test = 4;
202
+	subtest = strchr_selftest();
203
+	if (subtest)
204
+		goto fail;
205
+
206
+	test = 5;
207
+	subtest = strnchr_selftest();
208
+	if (subtest)
209
+		goto fail;
210
+
133
211
 	pr_info("String selftests succeeded\n");
134
212
 	return 0;
135
213
 fail: