[libc] Templatize str{,n}cmp

[abrachet]

Summary

This will be used to implement the case insensitive str{,n}casecmp

[lntue]

libc/src/string/memory_utils/strcmp_implementations.h

21

Will there be a problem with signed-ness difference between char and unsigned char in lines 20 vs 22?

[abrachet]

libc/src/string/memory_utils/strcmp_implementations.h

21

I don't really know, this is just how it was before and I copied verbatim other than using the comparator function

libc/src/__support/CPP/functional.h

 
 public:
   constexpr function() = default;
   template <typename F> constexpr function(F &&f) : func(f) {}
 
   constexpr R operator()(Args... params) { return func(params...); }
 };
 
+template <class T> struct minus {
+  constexpr T operator()(const T &lhs, const T &rhs) const { return lhs - rhs; }
+};
+
 } // namespace cpp
 } // namespace __llvm_libc
 
 #endif // LLVM_LIBC_SRC_SUPPORT_CPP_FUNCTIONAL_H

libc/src/string/CMakeLists.txt

 )
 
 add_entrypoint_object(
   strcmp
   SRCS
     strcmp.cpp
   HDRS
     strcmp.h
+  DEPENDS
+    .memory_utils.strcmp_implementation
+    libc.src.__support.CPP.functional
 )
 
 add_entrypoint_object(
   strcoll
   SRCS
     strcoll.cpp
   HDRS
     strcoll.h

 )
 
 add_entrypoint_object(
   strncmp
   SRCS
     strncmp.cpp
   HDRS
     strncmp.h
+  DEPENDS
+    .memory_utils.strcmp_implementation
+    libc.src.__support.CPP.functional
 )
 
 add_entrypoint_object(
   strncpy
   SRCS
     strncpy.cpp
   HDRS
     strncpy.h

libc/src/string/memory_utils/CMakeLists.txt

 
 add_header_library(
   bzero_implementation
   HDRS
     bzero_implementations.h
   DEPS
     .memset_implementation
 )
+
+add_header_library(
+  strcmp_implementation
+  HDRS
+    strcmp_implementations.h
+)

libc/src/string/memory_utils/strcmp_implementations.h

+//===-- str{,case}cmp implementation ----------------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_LIBC_SRC_STRING_MEMORY_UTILS_STRCMP_IMPLEMENTATIONS_H
+#define LLVM_LIBC_SRC_STRING_MEMORY_UTILS_STRCMP_IMPLEMENTATIONS_H
+
+#include <stddef.h>
+
+namespace __llvm_libc {
+
+template <typename Comp>
+constexpr static int strcmp_implementation(const char *left, const char *right,
+                                           Comp &&comp) {
+  // TODO: Look at benefits for comparing words at a time.
+  for (; *left && !comp(*left, *right); ++left, ++right)
+    ;
+  return comp(*reinterpret_cast<const unsigned char *>(left),
+              *reinterpret_cast<const unsigned char *>(right));
+}
+
+template <typename Comp>
+constexpr static int strncmp_implementation(const char *left, const char *right,
+                                            size_t n, Comp &&comp) {
+  if (n == 0)
+    return 0;
+
+  // TODO: Look at benefits for comparing words at a time.
+  for (; n > 1; --n, ++left, ++right) {
+    char lc = *left;
+    if (!comp(lc, '\0') || comp(lc, *right))
+      break;
+  }
+  return comp(*reinterpret_cast<const unsigned char *>(left),
+              *reinterpret_cast<const unsigned char *>(right));
+}
+
+} // namespace __llvm_libc
+
+#endif // LLVM_LIBC_SRC_STRING_MEMORY_UTILS_STRCMP_IMPLEMENTATIONS_H

libc/src/string/strcmp.cpp

 //===-- Implementation of strcmp ------------------------------------------===//
 //
 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 //
 //===----------------------------------------------------------------------===//
 
 #include "src/string/strcmp.h"
 
+#include "src/__support/CPP/functional.h"
 #include "src/__support/common.h"
+#include "src/string/memory_utils/strcmp_implementations.h"
 
 namespace __llvm_libc {
 
-// TODO: Look at benefits for comparing words at a time.
 LLVM_LIBC_FUNCTION(int, strcmp, (const char *left, const char *right)) {
-  for (; *left && *left == *right; ++left, ++right)
-    ;
-  return *reinterpret_cast<const unsigned char *>(left) -
-         *reinterpret_cast<const unsigned char *>(right);
+  return strcmp_implementation(left, right, cpp::minus<char>{});
 }
 
 } // namespace __llvm_libc

libc/src/string/strncmp.cpp

 //===-- Implementation of strncmp -----------------------------------------===//
 //
 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 //
 //===----------------------------------------------------------------------===//
 
 #include "src/string/strncmp.h"
 
+#include "src/__support/CPP/functional.h"
 #include "src/__support/common.h"
+#include "src/string/memory_utils/strcmp_implementations.h"
+
 #include <stddef.h>
 
 namespace __llvm_libc {
 
-// TODO: Look at benefits for comparing words at a time.
 LLVM_LIBC_FUNCTION(int, strncmp,
                    (const char *left, const char *right, size_t n)) {
-
-  if (n == 0)
-    return 0;
-
-  for (; n > 1; --n, ++left, ++right) {
-    char lc = *left;
-    if (lc == '\0' || lc != *right)
-      break;
-  }
-  return *reinterpret_cast<const unsigned char *>(left) -
-         *reinterpret_cast<const unsigned char *>(right);
+  return strncmp_implementation(left, right, n, cpp::minus<char>{});
 }
 
 } // namespace __llvm_libc

utils/bazel/llvm-project-overlay/libc/BUILD.bazel

libc_support_library(
     ],
     textual_hdrs = [
         "src/string/memory_utils/bcmp_implementations.h",
         "src/string/memory_utils/bzero_implementations.h",
         "src/string/memory_utils/memcmp_implementations.h",
         "src/string/memory_utils/memcpy_implementations.h",
         "src/string/memory_utils/memmove_implementations.h",
         "src/string/memory_utils/memset_implementations.h",
+        "src/string/memory_utils/strcmp_implementations.h",
     ],
     deps = [
         ":__support_common",
         ":__support_compiler_features",
         ":__support_cpp_array",
         ":__support_cpp_bit",
         ":__support_cpp_cstddef",
         ":__support_cpp_type_traits",

 )
 
 libc_function(
     name = "strcmp",
     srcs = ["src/string/strcmp.cpp"],
     hdrs = ["src/string/strcmp.h"],
     deps = [
         ":__support_common",
+        ":__support_cpp_functional",
+        ":string_memory_utils",
         ":string_utils",
     ],
 )
 
 libc_function(
     name = "strchr",
     srcs = ["src/string/strchr.cpp"],
     hdrs = ["src/string/strchr.h"],