1 //===-- Implementation of strndup -----------------------------------------===// 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-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "src/string/strndup.h" 10 #include "src/string/memory_utils/inline_memcpy.h" 11 #include "src/string/string_utils.h" 12 13 #include "src/__support/CPP/new.h" 14 #include "src/__support/common.h" 15 16 #include <stddef.h> 17 18 namespace LIBC_NAMESPACE { 19 20 LLVM_LIBC_FUNCTION(char *, strndup, (const char *src, size_t size)) { 21 if (src == nullptr) 22 return nullptr; 23 size_t len = internal::string_length(src); 24 if (len > size) 25 len = size; 26 AllocChecker ac; 27 char *dest = new (ac) char[len + 1]; 28 if (!ac) 29 return nullptr; 30 inline_memcpy(dest, src, len + 1); 31 dest[len] = '\0'; 32 return dest; 33 } 34 35 } // namespace LIBC_NAMESPACE 36