34 lines · cpp
1//===-- Implementation of strspn ------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "src/string/strspn.h"10 11#include "src/__support/CPP/bitset.h"12#include "src/__support/common.h"13#include "src/__support/macros/config.h"14#include "src/__support/macros/null_check.h"15#include <stddef.h>16 17namespace LIBC_NAMESPACE_DECL {18 19LLVM_LIBC_FUNCTION(size_t, strspn, (const char *src, const char *segment)) {20 LIBC_CRASH_ON_NULLPTR(src);21 LIBC_CRASH_ON_NULLPTR(segment);22 const char *initial = src;23 cpp::bitset<256> bitset;24 25 for (; *segment; ++segment)26 bitset.set(*reinterpret_cast<const unsigned char *>(segment));27 for (; *src && bitset.test(*reinterpret_cast<const unsigned char *>(src));28 ++src)29 ;30 return src - initial;31}32 33} // namespace LIBC_NAMESPACE_DECL34