LXR uboot/test/str

   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Copyright 2020 Google LLC
   4 */
   5
   6#include <common.h>
   7#include <vsprintf.h>
   8#include <test/suites.h>
   9#include <test/test.h>
  10#include <test/ut.h>
  11
  12/* This is large enough for any of the test strings */
  13#define TEST_STR_SIZE   200
  14
  15static const char str1[] = "I'm sorry I'm late.";
  16static const char str2[] = "1099abNo, don't bother apologising.";
  17static const char str3[] = "0xbI'm sorry you're alive.";
  18static const char str4[] = "1234567890123 I lost closer friends";
  19static const char str5[] = "0x9876543210the last time I was deloused";
  20static const char str6[] = "0778octal is seldom used";
  21static const char str7[] = "707it is a piece of computing history";
  22
  23/* Declare a new str test */
  24#define STR_TEST(_name, _flags)         UNIT_TEST(_name, _flags, str_test)
  25
  26static int str_upper(struct unit_test_state *uts)
  27{
  28        char out[TEST_STR_SIZE];
  29
  30        /* Make sure it adds a terminator */
  31        out[strlen(str1)] = 'a';
  32        str_to_upper(str1, out, SIZE_MAX);
  33        ut_asserteq_str("I'M SORRY I'M LATE.", out);
  34
  35        /* In-place operation */
  36        strcpy(out, str2);
  37        str_to_upper(out, out, SIZE_MAX);
  38        ut_asserteq_str("1099ABNO, DON'T BOTHER APOLOGISING.", out);
  39
  40        /* Limited length */
  41        str_to_upper(str1, out, 7);
  42        ut_asserteq_str("I'M SORO, DON'T BOTHER APOLOGISING.", out);
  43
  44        /* In-place with limited length */
  45        strcpy(out, str2);
  46        str_to_upper(out, out, 7);
  47        ut_asserteq_str("1099ABNo, don't bother apologising.", out);
  48
  49        /* Copy an empty string to a buffer with space*/
  50        out[1] = 0x7f;
  51        str_to_upper("", out, SIZE_MAX);
  52        ut_asserteq('\0', *out);
  53        ut_asserteq(0x7f, out[1]);
  54
  55        /* Copy an empty string to a buffer with no space*/
  56        out[0] = 0x7f;
  57        str_to_upper("", out, 0);
  58        ut_asserteq(0x7f, out[0]);
  59
  60        return 0;
  61}
  62STR_TEST(str_upper, 0);
  63
  64static int run_strtoul(struct unit_test_state *uts, const char *str, int base,
  65                       ulong expect_val, int expect_endp_offset, bool upper)
  66{
  67        char out[TEST_STR_SIZE];
  68        char *endp;
  69        ulong val;
  70
  71        strcpy(out, str);
  72        if (upper)
  73                str_to_upper(out, out, -1);
  74
  75        val = simple_strtoul(out, &endp, base);
  76        ut_asserteq(expect_val, val);
  77        ut_asserteq(expect_endp_offset, endp - out);
  78
  79        return 0;
  80}
  81
  82static int str_simple_strtoul(struct unit_test_state *uts)
  83{
  84        int upper;
  85
  86        /* Check that it is case-insentive */
  87        for (upper = 0; upper < 2; upper++) {
  88                /* Base 10 and base 16 */
  89                ut_assertok(run_strtoul(uts, str2, 10, 1099, 4, upper));
  90                ut_assertok(run_strtoul(uts, str2, 16, 0x1099ab, 6, upper));
  91                ut_assertok(run_strtoul(uts, str3, 16, 0xb, 3, upper));
  92                ut_assertok(run_strtoul(uts, str3, 10, 0xb, 3, upper));
  93
  94                /* Octal */
  95                ut_assertok(run_strtoul(uts, str6, 0, 63, 3, upper));
  96                ut_assertok(run_strtoul(uts, str7, 8, 0x1c7, 3, upper));
  97
  98                /* Invalid string */
  99                ut_assertok(run_strtoul(uts, str1, 10, 0, 0, upper));
 100
 101                /* Base 0 */
 102                ut_assertok(run_strtoul(uts, str1, 0, 0, 0, upper));
 103                ut_assertok(run_strtoul(uts, str2, 0, 1099, 4, upper));
 104                ut_assertok(run_strtoul(uts, str3, 0, 0xb, 3, upper));
 105
 106                /* Base 2 */
 107                ut_assertok(run_strtoul(uts, str1, 2, 0, 0, upper));
 108                ut_assertok(run_strtoul(uts, str2, 2, 2, 2, upper));
 109        }
 110
 111        /* Check endp being NULL */
 112        ut_asserteq(1099, simple_strtoul(str2, NULL, 0));
 113
 114        return 0;
 115}
 116STR_TEST(str_simple_strtoul, 0);
 117
 118static int run_strtoull(struct unit_test_state *uts, const char *str, int base,
 119                        unsigned long long expect_val, int expect_endp_offset,
 120                        bool upper)
 121{
 122        char out[TEST_STR_SIZE];
 123        char *endp;
 124        unsigned long long val;
 125
 126        strcpy(out, str);
 127        if (upper)
 128                str_to_upper(out, out, -1);
 129
 130        val = simple_strtoull(out, &endp, base);
 131        ut_asserteq(expect_val, val);
 132        ut_asserteq(expect_endp_offset, endp - out);
 133
 134        return 0;
 135}
 136
 137static int str_simple_strtoull(struct unit_test_state *uts)
 138{
 139        int upper;
 140
 141        /* Check that it is case-insentive */
 142        for (upper = 0; upper < 2; upper++) {
 143                /* Base 10 and base 16 */
 144                ut_assertok(run_strtoull(uts, str2, 10, 1099, 4, upper));
 145                ut_assertok(run_strtoull(uts, str2, 16, 0x1099ab, 6, upper));
 146                ut_assertok(run_strtoull(uts, str3, 16, 0xb, 3, upper));
 147                ut_assertok(run_strtoull(uts, str3, 10, 0xb, 3, upper));
 148
 149                /* Octal */
 150                ut_assertok(run_strtoull(uts, str6, 0, 63, 3, upper));
 151                ut_assertok(run_strtoull(uts, str7, 8, 0x1c7, 3, upper));
 152
 153                /* Large values */
 154                ut_assertok(run_strtoull(uts, str4, 10, 1234567890123, 13,
 155                                         upper));
 156                ut_assertok(run_strtoull(uts, str4, 16, 0x1234567890123, 13,
 157                                         upper));
 158                ut_assertok(run_strtoull(uts, str5, 0, 0x9876543210, 12,
 159                                         upper));
 160
 161                /* Invalid string */
 162                ut_assertok(run_strtoull(uts, str1, 10, 0, 0, upper));
 163
 164                /* Base 0 */
 165                ut_assertok(run_strtoull(uts, str1, 0, 0, 0, upper));
 166                ut_assertok(run_strtoull(uts, str2, 0, 1099, 4, upper));
 167                ut_assertok(run_strtoull(uts, str3, 0, 0xb, 3, upper));
 168
 169                /* Base 2 */
 170                ut_assertok(run_strtoull(uts, str1, 2, 0, 0, upper));
 171                ut_assertok(run_strtoull(uts, str2, 2, 2, 2, upper));
 172        }
 173
 174        /* Check endp being NULL */
 175        ut_asserteq(1099, simple_strtoull(str2, NULL, 0));
 176
 177        return 0;
 178}
 179STR_TEST(str_simple_strtoull, 0);
 180
 181static int str_hextoul(struct unit_test_state *uts)
 182{
 183        char *endp;
 184
 185        /* Just a simple test, since we know this uses simple_strtoul() */
 186        ut_asserteq(0x1099ab, hextoul(str2, &endp));
 187        ut_asserteq(6, endp - str2);
 188
 189        return 0;
 190}
 191STR_TEST(str_hextoul, 0);
 192
 193static int str_dectoul(struct unit_test_state *uts)
 194{
 195        char *endp;
 196
 197        /* Just a simple test, since we know this uses simple_strtoul() */
 198        ut_asserteq(1099, dectoul(str2, &endp));
 199        ut_asserteq(4, endp - str2);
 200
 201        return 0;
 202}
 203STR_TEST(str_dectoul, 0);
 204
 205int do_ut_str(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
 206{
 207        struct unit_test *tests = UNIT_TEST_SUITE_START(str_test);
 208        const int n_ents = UNIT_TEST_SUITE_COUNT(str_test);
 209
 210        return cmd_ut_category("str", "str_", tests, n_ents, argc, argv);
 211}
 212