tests/lib/libc/string/t_strlen.c

   1 /* $NetBSD: t_strlen.c,v 1.5 2011/07/14 07:33:20 jruoho Exp $ */
   2
   3 /*
   4  * Written by J.T. Conklin <jtc@acorntoolworks.com>
   5  * Public domain.
   6  */
   7
   8 #include <atf-c.h>
   9 #include <string.h>
  10 #include <unistd.h>
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include <dlfcn.h>
  14 #include <unistd.h>
  15
  16 static void     write_num(int);
  17
  18 static void
  19 write_num(int val)
  20 {
  21         char buf[20];
  22         int i;
  23
  24         for (i = sizeof buf; --i >= 0;) {
  25                 buf[i] = '0' + val % 10;
  26                 val /= 10;
  27                 if (val == 0) {
  28                         write(2, buf + i, sizeof buf - i);
  29                         return;
  30                 }
  31         }
  32         write(2, "overflow", 8);
  33 }
  34
  35 ATF_TC(strlen_basic);
  36 ATF_TC_HEAD(strlen_basic, tc)
  37 {
  38         atf_tc_set_md_var(tc, "descr", "Test strlen(3) results");
  39 }
  40
  41 ATF_TC_BODY(strlen_basic, tc)
  42 {
  43         /* try to trick the compiler */
  44         size_t (*strlen_fn)(const char *);
  45
  46         unsigned int a, t;
  47         size_t len;
  48         char buf[64];
  49
  50         struct tab {
  51                 const char*     val;
  52                 size_t          len;
  53         };
  54
  55         const struct tab tab[] = {
  56                 /*
  57                  * patterns that check for all combinations of leading and
  58                  * trailing unaligned characters (on a 64 bit processor)
  59                  */
  60
  61                 { "",                           0 },
  62                 { "a",                          1 },
  63                 { "ab",                         2 },
  64                 { "abc",                        3 },
  65                 { "abcd",                       4 },
  66                 { "abcde",                      5 },
  67                 { "abcdef",                     6 },
  68                 { "abcdefg",                    7 },
  69                 { "abcdefgh",                   8 },
  70                 { "abcdefghi",                  9 },
  71                 { "abcdefghij",                 10 },
  72                 { "abcdefghijk",                11 },
  73                 { "abcdefghijkl",               12 },
  74                 { "abcdefghijklm",              13 },
  75                 { "abcdefghijklmn",             14 },
  76                 { "abcdefghijklmno",            15 },
  77                 { "abcdefghijklmnop",           16 },
  78                 { "abcdefghijklmnopq",          17 },
  79                 { "abcdefghijklmnopqr",         18 },
  80                 { "abcdefghijklmnopqrs",        19 },
  81                 { "abcdefghijklmnopqrst",       20 },
  82                 { "abcdefghijklmnopqrstu",      21 },
  83                 { "abcdefghijklmnopqrstuv",     22 },
  84                 { "abcdefghijklmnopqrstuvw",    23 },
  85
  86                 /*
  87                  * patterns that check for the cases where the expression:
  88                  *
  89                  *      ((word - 0x7f7f..7f) & 0x8080..80)
  90                  *
  91                  * returns non-zero even though there are no zero bytes in
  92                  * the word.
  93                  */
  94
  95                 { "" "\xff\xff\xff\xff\xff\xff\xff\xff" "abcdefgh",     16 },
  96                 { "a" "\xff\xff\xff\xff\xff\xff\xff\xff" "bcdefgh",     16 },
  97                 { "ab" "\xff\xff\xff\xff\xff\xff\xff\xff" "cdefgh",     16 },
  98                 { "abc" "\xff\xff\xff\xff\xff\xff\xff\xff" "defgh",     16 },
  99                 { "abcd" "\xff\xff\xff\xff\xff\xff\xff\xff" "efgh",     16 },
 100                 { "abcde" "\xff\xff\xff\xff\xff\xff\xff\xff" "fgh",     16 },
 101                 { "abcdef" "\xff\xff\xff\xff\xff\xff\xff\xff" "gh",     16 },
 102                 { "abcdefg" "\xff\xff\xff\xff\xff\xff\xff\xff" "h",     16 },
 103                 { "abcdefgh" "\xff\xff\xff\xff\xff\xff\xff\xff" "",     16 },
 104         };
 105
 106         /*
 107          * During testing it is useful have the rest of the program
 108          * use a known good version!
 109          */
 110         strlen_fn = dlsym(dlopen(NULL, RTLD_LAZY), "test_strlen");
 111         if (!strlen_fn)
 112                 strlen_fn = strlen;
 113
 114         for (a = 0; a < sizeof(long); ++a) {
 115                 for (t = 0; t < (sizeof(tab) / sizeof(tab[0])); ++t) {
 116
 117                         memcpy(&buf[a], tab[t].val, tab[t].len + 1);
 118                         len = strlen_fn(&buf[a]);
 119
 120                         if (len != tab[t].len) {
 121                                 /* Write error without using printf / strlen */
 122                                 write(2, "alignment ", 10);
 123                                 write_num(a);
 124                                 write(2, ", test ", 7);
 125                                 write_num(t);
 126                                 write(2, ", got len ", 10);
 127                                 write_num(len);
 128                                 write(2, ", not ", 6);
 129                                 write_num(tab[t].len);
 130                                 write(2, ", for '", 7);
 131                                 write(2, tab[t].val, tab[t].len);
 132                                 write(2, "'\n", 2);
 133                                 atf_tc_fail("See stderr for details");
 134                         }
 135                 }
 136         }
 137 }
 138
 139 ATF_TC(strlen_huge);
 140 ATF_TC_HEAD(strlen_huge, tc)
 141 {
 142         atf_tc_set_md_var(tc, "descr", "Test strlen(3) with huge strings");
 143 }
 144
 145 ATF_TC_BODY(strlen_huge, tc)
 146 {
 147         long page;
 148         char *str;
 149         size_t i;
 150
 151         page = sysconf(_SC_PAGESIZE);
 152         ATF_REQUIRE(page >= 0);
 153
 154         for (i = 1; i < 1000; i = i + 100) {
 155
 156                 str = malloc(i * page + 1);
 157
 158                 if (str == NULL)
 159                         continue;
 160
 161                 (void)memset(str, 'x', i * page);
 162                 str[i * page] = '\0';
 163
 164                 ATF_REQUIRE(strlen(str) == i * page);
 165                 free(str);
 166         }
 167 }
 168
 169 ATF_TC(strnlen_basic);
 170 ATF_TC_HEAD(strnlen_basic, tc)
 171 {
 172         atf_tc_set_md_var(tc, "descr", "A naive test of strnlen(3)");
 173 }
 174
 175 ATF_TC_BODY(strnlen_basic, tc)
 176 {
 177         char buf[1];
 178
 179         buf[0] = '\0';
 180
 181         ATF_CHECK(strnlen(buf, 000) == 0);
 182         ATF_CHECK(strnlen(buf, 111) == 0);
 183
 184         ATF_CHECK(strnlen("xxx", 0) == 0);
 185         ATF_CHECK(strnlen("xxx", 1) == 1);
 186         ATF_CHECK(strnlen("xxx", 2) == 2);
 187         ATF_CHECK(strnlen("xxx", 3) == 3);
 188         ATF_CHECK(strnlen("xxx", 9) == 3);
 189 }
 190
 191 ATF_TP_ADD_TCS(tp)
 192 {
 193
 194         ATF_TP_ADD_TC(tp, strlen_basic);
 195         ATF_TP_ADD_TC(tp, strlen_huge);
 196         ATF_TP_ADD_TC(tp, strnlen_basic);
 197
 198         return atf_no_error();
 199 }