test-strncpy.c 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323
  1. /* Test strncpy functions.
  2. Copyright (C) 1999-2019 Free Software Foundation, Inc.
  3. This file is part of the GNU C Library.
  4. Written by Jakub Jelinek <jakub@redhat.com>, 1999.
  5. The GNU C Library is free software; you can redistribute it and/or
  6. modify it under the terms of the GNU Lesser General Public
  7. License as published by the Free Software Foundation; either
  8. version 2.1 of the License, or (at your option) any later version.
  9. The GNU C Library is distributed in the hope that it will be useful,
  10. but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. Lesser General Public License for more details.
  13. You should have received a copy of the GNU Lesser General Public
  14. License along with the GNU C Library; if not, see
  15. <http://www.gnu.org/licenses/>. */
  16. #ifdef WIDE
  17. # include <wchar.h>
  18. # define CHAR wchar_t
  19. # define UCHAR wchar_t
  20. # define BIG_CHAR WCHAR_MAX
  21. # define SMALL_CHAR 1273
  22. # define MEMCMP wmemcmp
  23. # define MEMSET wmemset
  24. # define STRNLEN wcsnlen
  25. #else
  26. # define CHAR char
  27. # define UCHAR unsigned char
  28. # define BIG_CHAR CHAR_MAX
  29. # define SMALL_CHAR 127
  30. # define MEMCMP memcmp
  31. # define MEMSET memset
  32. # define STRNLEN strnlen
  33. #endif /* !WIDE */
  34. #ifndef STRNCPY_RESULT
  35. # define STRNCPY_RESULT(dst, len, n) dst
  36. # define TEST_MAIN
  37. # ifndef WIDE
  38. # define TEST_NAME "strncpy"
  39. # else
  40. # define TEST_NAME "wcsncpy"
  41. # endif /* WIDE */
  42. # include "test-string.h"
  43. # ifndef WIDE
  44. # define SIMPLE_STRNCPY simple_strncpy
  45. # define STUPID_STRNCPY stupid_strncpy
  46. # define STRNCPY strncpy
  47. # else
  48. # define SIMPLE_STRNCPY simple_wcsncpy
  49. # define STUPID_STRNCPY stupid_wcsncpy
  50. # define STRNCPY wcsncpy
  51. # endif /* WIDE */
  52. CHAR *SIMPLE_STRNCPY (CHAR *, const CHAR *, size_t);
  53. CHAR *STUPID_STRNCPY (CHAR *, const CHAR *, size_t);
  54. IMPL (STUPID_STRNCPY, 0)
  55. IMPL (SIMPLE_STRNCPY, 0)
  56. IMPL (STRNCPY, 1)
  57. CHAR *
  58. SIMPLE_STRNCPY (CHAR *dst, const CHAR *src, size_t n)
  59. {
  60. CHAR *ret = dst;
  61. while (n--)
  62. if ((*dst++ = *src++) == '\0')
  63. {
  64. while (n--)
  65. *dst++ = '\0';
  66. return ret;
  67. }
  68. return ret;
  69. }
  70. CHAR *
  71. STUPID_STRNCPY (CHAR *dst, const CHAR *src, size_t n)
  72. {
  73. size_t nc = STRNLEN (src, n);
  74. size_t i;
  75. for (i = 0; i < nc; ++i)
  76. dst[i] = src[i];
  77. for (; i < n; ++i)
  78. dst[i] = '\0';
  79. return dst;
  80. }
  81. #endif /* !STRNCPY_RESULT */
  82. typedef CHAR *(*proto_t) (CHAR *, const CHAR *, size_t);
  83. static void
  84. do_one_test (impl_t *impl, CHAR *dst, const CHAR *src, size_t len, size_t n)
  85. {
  86. if (CALL (impl, dst, src, n) != STRNCPY_RESULT (dst, len, n))
  87. {
  88. error (0, 0, "Wrong result in function %s %p %p", impl->name,
  89. CALL (impl, dst, src, n), dst);
  90. ret = 1;
  91. return;
  92. }
  93. if (memcmp (dst, src, (len > n ? n : len) * sizeof (CHAR)) != 0)
  94. {
  95. error (0, 0, "Wrong result in function %s", impl->name);
  96. ret = 1;
  97. return;
  98. }
  99. if (n > len)
  100. {
  101. size_t i;
  102. for (i = len; i < n; ++i)
  103. if (dst [i] != '\0')
  104. {
  105. error (0, 0, "Wrong result in function %s", impl->name);
  106. ret = 1;
  107. return;
  108. }
  109. }
  110. }
  111. static void
  112. do_test (size_t align1, size_t align2, size_t len, size_t n, int max_char)
  113. {
  114. size_t i;
  115. CHAR *s1, *s2;
  116. /* For wcsncpy: align1 and align2 here mean alignment not in bytes,
  117. but in wchar_ts, in bytes it will equal to align * (sizeof (wchar_t)). */
  118. align1 &= 7;
  119. if ((align1 + len) * sizeof (CHAR) >= page_size)
  120. return;
  121. align2 &= 7;
  122. if ((align2 + len) * sizeof (CHAR) >= page_size)
  123. return;
  124. s1 = (CHAR *) (buf1) + align1;
  125. s2 = (CHAR *) (buf2) + align2;
  126. for (i = 0; i < len; ++i)
  127. s1[i] = 32 + 23 * i % (max_char - 32);
  128. s1[len] = 0;
  129. for (i = len + 1; (i + align1) * sizeof (CHAR) < page_size && i < len + 64;
  130. ++i)
  131. s1[i] = 32 + 32 * i % (max_char - 32);
  132. FOR_EACH_IMPL (impl, 0)
  133. do_one_test (impl, s2, s1, len, n);
  134. }
  135. static void
  136. do_random_tests (void)
  137. {
  138. size_t i, j, n, align1, align2, len, size, mode;
  139. UCHAR *p1 = (UCHAR *) (buf1 + page_size) - 512;
  140. UCHAR *p2 = (UCHAR *) (buf2 + page_size) - 512;
  141. UCHAR *res;
  142. for (n = 0; n < ITERATIONS; n++)
  143. {
  144. /* For wcsncpy: align1 and align2 here mean align not in bytes,
  145. but in wchar_ts, in bytes it will equal to align * (sizeof
  146. (wchar_t)). */
  147. mode = random ();
  148. if (mode & 1)
  149. {
  150. size = random () & 255;
  151. align1 = 512 - size - (random () & 15);
  152. if (mode & 2)
  153. align2 = align1 - (random () & 24);
  154. else
  155. align2 = align1 - (random () & 31);
  156. if (mode & 4)
  157. {
  158. j = align1;
  159. align1 = align2;
  160. align2 = j;
  161. }
  162. if (mode & 8)
  163. len = size - (random () & 31);
  164. else
  165. len = 512;
  166. if (len >= 512)
  167. len = random () & 511;
  168. }
  169. else
  170. {
  171. align1 = random () & 31;
  172. if (mode & 2)
  173. align2 = random () & 31;
  174. else
  175. align2 = align1 + (random () & 24);
  176. len = random () & 511;
  177. j = align1;
  178. if (align2 > j)
  179. j = align2;
  180. if (mode & 4)
  181. {
  182. size = random () & 511;
  183. if (size + j > 512)
  184. size = 512 - j - (random () & 31);
  185. }
  186. else
  187. size = 512 - j;
  188. if ((mode & 8) && len + j >= 512)
  189. len = 512 - j - (random () & 7);
  190. }
  191. j = len + align1 + 64;
  192. if (j > 512)
  193. j = 512;
  194. for (i = 0; i < j; i++)
  195. {
  196. if (i == len + align1)
  197. p1[i] = 0;
  198. else
  199. {
  200. p1[i] = random () & BIG_CHAR;
  201. if (i >= align1 && i < len + align1 && !p1[i])
  202. p1[i] = (random () & SMALL_CHAR) + 3;
  203. }
  204. }
  205. FOR_EACH_IMPL (impl, 1)
  206. {
  207. MEMSET (p2 - 64, '\1', 512 + 64);
  208. res = (UCHAR *) CALL (impl, (CHAR *) (p2 + align2),
  209. (CHAR *) (p1 + align1), size);
  210. if (res != STRNCPY_RESULT (p2 + align2, len, size))
  211. {
  212. error (0, 0, "Iteration %zd - wrong result in function %s (%zd, %zd, %zd) %p != %p",
  213. n, impl->name, align1, align2, len, res,
  214. STRNCPY_RESULT (p2 + align2, len, size));
  215. ret = 1;
  216. }
  217. for (j = 0; j < align2 + 64; ++j)
  218. {
  219. if (p2[j - 64] != '\1')
  220. {
  221. error (0, 0, "Iteration %zd - garbage before, %s (%zd, %zd, %zd)",
  222. n, impl->name, align1, align2, len);
  223. ret = 1;
  224. break;
  225. }
  226. }
  227. j = align2 + len + 1;
  228. if (size + align2 > j)
  229. j = size + align2;
  230. for (; j < 512; ++j)
  231. {
  232. if (p2[j] != '\1')
  233. {
  234. error (0, 0, "Iteration %zd - garbage after, %s (%zd, %zd, %zd)",
  235. n, impl->name, align1, align2, len);
  236. ret = 1;
  237. break;
  238. }
  239. }
  240. for (j = align2 + len + 1; j < align2 + size; ++j)
  241. if (p2[j])
  242. {
  243. error (0, 0, "Iteration %zd - garbage after size, %s (%zd, %zd, %zd)",
  244. n, impl->name, align1, align2, len);
  245. ret = 1;
  246. break;
  247. }
  248. j = len + 1;
  249. if (size < j)
  250. j = size;
  251. if (MEMCMP (p1 + align1, p2 + align2, j))
  252. {
  253. error (0, 0, "Iteration %zd - different strings, %s (%zd, %zd, %zd)",
  254. n, impl->name, align1, align2, len);
  255. ret = 1;
  256. }
  257. }
  258. }
  259. }
  260. int
  261. test_main (void)
  262. {
  263. size_t i;
  264. test_init ();
  265. printf ("%28s", "");
  266. FOR_EACH_IMPL (impl, 0)
  267. printf ("\t%s", impl->name);
  268. putchar ('\n');
  269. for (i = 1; i < 8; ++i)
  270. {
  271. do_test (i, i, 16, 16, SMALL_CHAR);
  272. do_test (i, i, 16, 16, BIG_CHAR);
  273. do_test (i, 2 * i, 16, 16, SMALL_CHAR);
  274. do_test (2 * i, i, 16, 16, BIG_CHAR);
  275. do_test (8 - i, 2 * i, 1 << i, 2 << i, SMALL_CHAR);
  276. do_test (2 * i, 8 - i, 2 << i, 1 << i, SMALL_CHAR);
  277. do_test (8 - i, 2 * i, 1 << i, 2 << i, BIG_CHAR);
  278. do_test (2 * i, 8 - i, 2 << i, 1 << i, BIG_CHAR);
  279. }
  280. for (i = 1; i < 8; ++i)
  281. {
  282. do_test (0, 0, 4 << i, 8 << i, SMALL_CHAR);
  283. do_test (0, 0, 16 << i, 8 << i, SMALL_CHAR);
  284. do_test (8 - i, 2 * i, 4 << i, 8 << i, SMALL_CHAR);
  285. do_test (8 - i, 2 * i, 16 << i, 8 << i, SMALL_CHAR);
  286. }
  287. do_random_tests ();
  288. return ret;
  289. }
  290. #include <support/test-driver.c>