debug/test-stpcpy_chk.c

   1 /* Test and measure stpcpy checking functions.
   2    Copyright (C) 1999-2016 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Written by Jakub Jelinek <jakub@redhat.com>, 1999.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, see
  18    <http://www.gnu.org/licenses/>.  */
  19
  20 #define STRCPY_RESULT(dst, len) ((dst) + (len))
  21 #define TEST_MAIN
  22 #define TEST_NAME "stpcpy_chk"
  23 #include "../string/test-string.h"
  24
  25 extern void __attribute__ ((noreturn)) __chk_fail (void);
  26 char *simple_stpcpy_chk (char *, const char *, size_t);
  27 extern char *normal_stpcpy (char *, const char *, size_t)
  28   __asm ("stpcpy");
  29 extern char *__stpcpy_chk (char *, const char *, size_t);
  30
  31 IMPL (simple_stpcpy_chk, 0)
  32 IMPL (normal_stpcpy, 1)
  33 IMPL (__stpcpy_chk, 2)
  34
  35 char *
  36 simple_stpcpy_chk (char *dst, const char *src, size_t len)
  37 {
  38   if (! len)
  39     __chk_fail ();
  40   while ((*dst++ = *src++) != '\0')
  41     if (--len == 0)
  42       __chk_fail ();
  43   return dst - 1;
  44 }
  45
  46 #include "test-strcpy_chk.c"