37 lines
1.2 KiB
C++
37 lines
1.2 KiB
C++
//===-- Implementation of memccpy ----------------------------------------===//
|
|
//
|
|
// 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/memccpy.h"
|
|
|
|
#include "src/__support/common.h"
|
|
#include "src/__support/macros/config.h"
|
|
#include <stddef.h> // For size_t.
|
|
|
|
namespace LIBC_NAMESPACE_DECL {
|
|
|
|
LLVM_LIBC_FUNCTION(void *, memccpy,
|
|
(void *__restrict dest, const void *__restrict src, int c,
|
|
size_t count)) {
|
|
unsigned char end = static_cast<unsigned char>(c);
|
|
const unsigned char *uc_src = static_cast<const unsigned char *>(src);
|
|
unsigned char *uc_dest = static_cast<unsigned char *>(dest);
|
|
size_t i = 0;
|
|
// Copy up until end is found.
|
|
for (; i < count && uc_src[i] != end; ++i)
|
|
uc_dest[i] = uc_src[i];
|
|
// if i < count, then end must have been found, so copy end into dest and
|
|
// return the byte after.
|
|
if (i < count) {
|
|
uc_dest[i] = uc_src[i];
|
|
return uc_dest + i + 1;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
} // namespace LIBC_NAMESPACE_DECL
|