/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-â
âvi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :viâ
ââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââ¡
â Copyright 2020 Justine Alexandra Roberts Tunney                              â
â                                                                              â
â Permission to use, copy, modify, and/or distribute this software for         â
â any purpose with or without fee is hereby granted, provided that the         â
â above copyright notice and this permission notice appear in all copies.      â
â                                                                              â
â THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL                â
â WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED                â
â WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE             â
â AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL         â
â DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR        â
â PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER               â
â TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR             â
â PERFORMANCE OF THIS SOFTWARE.                                                â
ââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââââ*/
#include "libc/intrin/bits.h"
#include "libc/macros.internal.h"
#include "libc/mem/mem.h"
#include "libc/stdio/rand.h"
#include "libc/stdio/stdio.h"
#include "libc/str/str.h"
#include "libc/testlib/ezbench.h"
#include "libc/testlib/testlib.h"

char u8[] = "utf-8 â»";
char16_t u16[] = u"utf16 â»";
wchar_t u32[] = L"utf32 â»";

size_t strlen_pure(const char *s) {
  size_t n = 0;
  while (*s++) ++n;
  return n;
}

TEST(strlen, usageExample_c11) {
  _Alignas(16) char ugh[] = "eeeeeeeeeeeeeee\017";
  EXPECT_EQ(1, strlen_pure(ugh + 15));
  EXPECT_EQ(6 + 3, strlen(u8));
  EXPECT_EQ(7, strlen16(u16));
  EXPECT_EQ(7, wcslen(u32));
}

TEST(strlen, usageExample_c99) {
  EXPECT_EQ(6 + 3, strlen(u8));
  EXPECT_EQ(7, strlen16(u16));
  EXPECT_EQ(7, wcslen(u32));
}

TEST(strlen, whatMemoryLooksLike) {
  EXPECT_BINEQ(u"utf-8 ÎÃ¿âÂ ", u8); /* â thompson-pike encoded */
  EXPECT_BINEQ(u"uÂ tÂ fÂ 1Â 6Â  Â ;&Â Â ", u16);
  EXPECT_BINEQ(u"uÂ Â Â tÂ Â Â fÂ Â Â 3Â Â Â 2Â Â Â  Â Â Â ;&Â Â Â Â Â Â ", u32);
}

TEST(strlen, test_const) {
  const char buf[] = "hellothere";
  ASSERT_EQ(0, strlen(""));
  ASSERT_EQ(0, strnlen("e", 0));
  ASSERT_EQ(10, strlen(buf));
}

TEST(strlen, test_nonconst) {
  char buf[256];
  unsigned i;
  for (i = 0; i < 255; ++i) buf[i] = i + 1;
  buf[i] = '\0';
  ASSERT_EQ(255, strlen(buf));
}

TEST(strnlen, testconst) {
  ASSERT_EQ(0, strnlen_s(NULL, 3));
  ASSERT_EQ(0, strnlen("", 3));
  ASSERT_EQ(0, strnlen("a", 0));
  ASSERT_EQ(3, strnlen("123", 3));
  ASSERT_EQ(2, strnlen("123", 2));
  ASSERT_EQ(3, strnlen("123", 4));
}

TEST(strlen, testnonconst) {
  /* this test case is a great example of why we need:
       "m"(*(char(*)[0x7fffffff])StR)
     rather than:
       "m"(*StR) */
  char buf[256];
  unsigned i;
  for (i = 0; i < 250; ++i) buf[i] = i + 1;
  buf[i] = '\0';
  ASSERT_EQ(250, strlen(buf));
}

TEST(strnlen_s, null_ReturnsZero) {
  ASSERT_EQ(0, strnlen_s(NULL, 3));
}

TEST(wcsnlen_s, null_ReturnsZero) {
  ASSERT_EQ(0, wcsnlen_s(NULL, 3));
}

TEST(strnlen, nulNotFound_ReturnsSize) {
  int sizes[] = {1, 2, 7, 8, 15, 16, 31, 32, 33};
  for (unsigned i = 0; i < ARRAYLEN(sizes); ++i) {
    char *buf = malloc(sizes[i]);
    memset(buf, ' ', sizes[i]);
    ASSERT_EQ(sizes[i], strnlen(buf, sizes[i]), "%d", sizes[i]);
    free(buf);
  }
}

TEST(strnlen_s, nulNotFound) {
  char buf[3] = {1, 2, 3};
  ASSERT_EQ(3, strnlen_s(buf, 3));
}

TEST(wcsnlen_s, nulNotFound) {
  wchar_t buf[3] = {1, 2, 3};
  ASSERT_EQ(3, wcsnlen_s(buf, 3));
}

TEST(strlen, fuzz) {
  char *b;
  size_t n, n1, n2;
  for (n = 2; n < 1026; ++n) {
    b = rngset(calloc(1, n), n - 1, _rand64, -1);
    n1 = strlen(b);
    n2 = strlen_pure(b);
    ASSERT_EQ(n1, n2, "%#.*s", n, b);
    n1 = strlen(b + 1);
    n2 = strlen_pure(b + 1);
    ASSERT_EQ(n1, n2);
    free(b);
  }
}

BENCH(strlen, bench) {
  extern size_t strlen_(const char *) asm("strlen");
  extern size_t strlen_pure_(const char *) asm("strlen_pure");
  static char b[2048];
  static char c[512];
  static char d[256];
  memset(b, -1, sizeof(b) - 1);
  memset(c, -1, sizeof(c) - 1);
  memset(d, -1, sizeof(d) - 1);
  EZBENCH2("strlen      1", donothing, strlen_(""));
  EZBENCH2("strlen_pure 1", donothing, strlen_pure_(""));
  EZBENCH2("strlen      2", donothing, strlen_("1"));
  EZBENCH2("strlen_pure 2", donothing, strlen_pure_("1"));
  EZBENCH2("strlen      3", donothing, strlen_("11"));
  EZBENCH2("strlen_pure 3", donothing, strlen_pure_("11"));
  EZBENCH2("strlen      4", donothing, strlen_("113"));
  EZBENCH2("strlen_pure 4", donothing, strlen_pure_("113"));
  EZBENCH2("strlen      7", donothing, strlen_("123456"));
  EZBENCH2("strlen_pure 7", donothing, strlen_pure_("123456"));
  EZBENCH2("strlen      8", donothing, strlen_("1234567"));
  EZBENCH2("strlen_pure 8", donothing, strlen_pure_("1234567"));
  EZBENCH2("strlen      9", donothing, strlen_("12345678"));
  EZBENCH2("strlen_pure 9", donothing, strlen_pure_("12345678"));
  EZBENCH2("strlen      11", donothing, strlen_("12345678aa"));
  EZBENCH2("strlen_pure 11", donothing, strlen_pure_("12345678aa"));
  EZBENCH2("strlen      13", donothing, strlen_("12345678aabb"));
  EZBENCH2("strlen_pure 13", donothing, strlen_pure_("12345678aabb"));
  EZBENCH2("strlen      16", donothing, strlen_("123456781234567"));
  EZBENCH2("strlen_pure 16", donothing, strlen_pure_("123456781234567"));
  EZBENCH2("strlen      17", donothing, strlen_("123456781234567e"));
  EZBENCH2("strlen_pure 17", donothing, strlen_pure_("123456781234567e"));
  EZBENCH2("strlen      34", donothing,
           strlen_("123456781234567e123456781234567ee"));
  EZBENCH2("strlen_pure 34", donothing,
           strlen_pure_("123456781234567e123456781234567ee"));
  EZBENCH2("strlen      68", donothing,
           strlen_("123456781234567e123456781234567ee123456781234567e1234567812"
                   "34567eee"));
  EZBENCH2("strlen_pure 68", donothing,
           strlen_pure_("123456781234567e123456781234567ee123456781234567e12345"
                        "6781234567eee"));
  EZBENCH2("strlen      256", donothing, strlen_(d));
  EZBENCH2("strlen_pure 256", donothing, strlen_pure_(d));
  EZBENCH2("strlen      512", donothing, strlen_(c));
  EZBENCH2("strlen_pure 512", donothing, strlen_pure_(c));
  EZBENCH2("strlen      2048", donothing, strlen_(b));
  EZBENCH2("strlen_pure 2048", donothing, strlen_pure_(b));
}
