debug/test-stpcpy_chk.c

   1 /* Test and measure stpcpy checking 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 STRCPY_RESULT(dst, len) ((dst) + (len))
  20 #define TEST_MAIN
  21 #define TEST_NAME "stpcpy_chk"
  22 #include "../string/test-string.h"
  23
  24 extern void __attribute__ ((noreturn)) __chk_fail (void);
  25 char *simple_stpcpy_chk (char *, const char *, size_t);
  26 extern char *normal_stpcpy (char *, const char *, size_t)
  27   __asm ("stpcpy");
  28 extern char *__stpcpy_chk (char *, const char *, size_t);
  29
  30 IMPL (simple_stpcpy_chk, 0)
  31 IMPL (normal_stpcpy, 1)
  32 IMPL (__stpcpy_chk, 2)
  33
  34 char *
  35 simple_stpcpy_chk (char *dst, const char *src, size_t len)
  36 {
  37   if (! len)
  38     __chk_fail ();
  39   while ((*dst++ = *src++) != '\0')
  40     if (--len == 0)
  41       __chk_fail ();
  42   return dst - 1;
  43 }
  44
  45 #include "test-strcpy_chk.c"