added nullptr handling and made functions static

Author: Sriya Pratipati

libc/src/wchar/wchar_utils.h

@@ -18,7 +18,7 @@ namespace LIBC_NAMESPACE_DECL {
 namespace internal {
 
 // returns true if the character exists in the string
-LIBC_INLINE bool internal_wcschr(wchar_t c, const wchar_t *str) {
+LIBC_INLINE static bool wcschr(wchar_t c, const wchar_t *str) {
   for (int n = 0; str[n]; ++n) {
     if (str[n] == c)
       return true;
@@ -28,11 +28,11 @@ LIBC_INLINE bool internal_wcschr(wchar_t c, const wchar_t *str) {
 
 // bool should be true for wcscspn for complimentary span
 // should be false for wcsspn since we want it to span
-LIBC_INLINE size_t inline_wcsspn(const wchar_t *s1, const wchar_t *s2,
+LIBC_INLINE static size_t wcsspn(const wchar_t *s1, const wchar_t *s2,
                                  bool not_match_set) {
   size_t i = 0;
   for (; s1[i]; ++i) {
-    bool in_set = internal_wcschr(s1[i], s2);
+    bool in_set = wcschr(s1[i], s2);
     if (in_set == not_match_set)
       return i;
   }

libc/src/wchar/wcscspn.cpp

@@ -12,12 +12,15 @@
 #include "hdr/types/wchar_t.h"
 #include "src/__support/common.h"
 #include "src/__support/macros/config.h"
+#include "src/__support/macros/null_check.h"
 #include "wchar_utils.h"
 
 namespace LIBC_NAMESPACE_DECL {
 
 LLVM_LIBC_FUNCTION(size_t, wcscspn, (const wchar_t *s1, const wchar_t *s2)) {
-  return internal::inline_wcsspn(s1, s2, true);
+  LIBC_CRASH_ON_NULLPTR(s1);
+  LIBC_CRASH_ON_NULLPTR(s2);
+  return internal::wcsspn(s1, s2, /*not_match_set=*/true);
 }
 
 } // namespace LIBC_NAMESPACE_DECL

libc/src/wchar/wcsspn.cpp

@@ -12,12 +12,15 @@
 #include "hdr/types/wchar_t.h"
 #include "src/__support/common.h"
 #include "src/__support/macros/config.h"
+#include "src/__support/macros/null_check.h"
 #include "wchar_utils.h"
 
 namespace LIBC_NAMESPACE_DECL {
 
 LLVM_LIBC_FUNCTION(size_t, wcsspn, (const wchar_t *s1, const wchar_t *s2)) {
-  return internal::inline_wcsspn(s1, s2, false);
+  LIBC_CRASH_ON_NULLPTR(s1);
+  LIBC_CRASH_ON_NULLPTR(s2);
+  return internal::wcsspn(s1, s2, /*not_match_set=*/false);
 }
 
 } // namespace LIBC_NAMESPACE_DECL