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/memcpy.h"
11 #include "src/string/string_utils.h"
12 
13 #include "src/__support/common.h"
14 
15 #include <stddef.h>
16 #include <stdlib.h>
17 
18 namespace __llvm_libc {
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   char *dest = reinterpret_cast<char *>(::malloc(len + 1)); // NOLINT
27   if (dest == nullptr)
28     return nullptr;
29   char *result =
30       reinterpret_cast<char *>(__llvm_libc::memcpy(dest, src, len + 1));
31   result[len] = '\0';
32   return result;
33 }
34 
35 } // namespace __llvm_libc
36