libc/src/string/strndup.cpp

   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/memcpy_implementations.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 __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   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 __llvm_libc