string/test-stpncpy.c

   1 /* Test and measure stpncpy functions.
   2    Copyright (C) 1999-2022 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"