aboutsummaryrefslogtreecommitdiff
path: root/libc/src/wchar/wcspbrk.cpp
blob: f329b73f9ee701e4f54936e443edabe0304f039d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
//===-- Implementation of wcspbrk -----------------------------------------===//
//
// 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/wchar/wcspbrk.h"

#include "hdr/types/wchar_t.h"
#include "src/__support/common.h"
#include "src/__support/macros/null_check.h"
#include "src/wchar/wchar_utils.h"

namespace LIBC_NAMESPACE_DECL {

LLVM_LIBC_FUNCTION(const wchar_t *, wcspbrk,
                   (const wchar_t *src, const wchar_t *breakset)) {
  LIBC_CRASH_ON_NULLPTR(src);
  LIBC_CRASH_ON_NULLPTR(breakset);

  // currently O(n * m), can be further optimized to O(n + m) with a hash set
  for (int src_idx = 0; src[src_idx] != 0; src_idx++)
    if (internal::wcschr(breakset, src[src_idx]))
      return src + src_idx;

  return nullptr;
}

} // namespace LIBC_NAMESPACE_DECL