1  /* Test and measure stpncpy functions.
       2     Copyright (C) 1999-2023 Free Software Foundation, Inc.
       3     This file is part of the GNU C Library.
       4  
       5     The GNU C Library is free software; you can redistribute it and/or
       6     modify it under the terms of the GNU Lesser General Public
       7     License as published by the Free Software Foundation; either
       8     version 2.1 of the License, or (at your option) any later version.
       9  
      10     The GNU C Library is distributed in the hope that it will be useful,
      11     but WITHOUT ANY WARRANTY; without even the implied warranty of
      12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      13     Lesser General Public License for more details.
      14  
      15     You should have received a copy of the GNU Lesser General Public
      16     License along with the GNU C Library; if not, see
      17     <https://www.gnu.org/licenses/>.  */
      18  
      19  #define STRNCPY_RESULT(dst, len, n) ((dst) + ((len) > (n) ? (n) : (len)))
      20  #define TEST_MAIN
      21  #ifndef WIDE
      22  # define TEST_NAME "stpncpy"
      23  #else
      24  # define TEST_NAME "wcpncpy"
      25  #endif /* WIDE */
      26  #include "test-string.h"
      27  #ifndef WIDE
      28  # define CHAR char
      29  # define SIMPLE_STPNCPY simple_stpncpy
      30  # define STUPID_STPNCPY stupid_stpncpy
      31  # define STPNCPY stpncpy
      32  # define STRNLEN strnlen
      33  #else
      34  # include <wchar.h>
      35  # define CHAR wchar_t
      36  # define SIMPLE_STPNCPY simple_wcpncpy
      37  # define STUPID_STPNCPY stupid_wcpncpy
      38  # define STPNCPY wcpncpy
      39  # define STRNLEN wcsnlen
      40  #endif /* WIDE */
      41  
      42  CHAR *SIMPLE_STPNCPY (CHAR *, const CHAR *, size_t);
      43  CHAR *STUPID_STPNCPY (CHAR *, const CHAR *, size_t);
      44  
      45  IMPL (STUPID_STPNCPY, 0)
      46  IMPL (SIMPLE_STPNCPY, 0)
      47  IMPL (STPNCPY, 1)
      48  
      49  CHAR *
      50  SIMPLE_STPNCPY (CHAR *dst, const CHAR *src, size_t n)
      51  {
      52    while (n--)
      53      if ((*dst++ = *src++) == '\0')
      54        {
      55  	size_t i;
      56  
      57  	for (i = 0; i < n; ++i)
      58  	  dst[i] = '\0';
      59  	return dst - 1;
      60        }
      61    return dst;
      62  }
      63  
      64  CHAR *
      65  STUPID_STPNCPY (CHAR *dst, const CHAR *src, size_t n)
      66  {
      67    size_t nc = STRNLEN (src, n);
      68    size_t i;
      69  
      70    for (i = 0; i < nc; ++i)
      71      dst[i] = src[i];
      72    for (; i < n; ++i)
      73      dst[i] = '\0';
      74    return dst + nc;
      75  }
      76  
      77  #undef CHAR
      78  #include "test-strncpy.c"