debug/stpncpy_chk.c

   1 /* Copyright (C) 1993,1995,1996,1997,2002,2005 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, write to the Free
  16    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  17    02111-1307 USA.  */
  18
  19 /* This is almost copied from strncpy.c, written by Torbjorn Granlund.  */
  20
  21 #include <string.h>
  22
  23
  24 /* Copy no more than N characters of SRC to DEST, returning the address of
  25    the terminating '\0' in DEST, if any, or else DEST + N.  */
  26 char *
  27 __stpncpy_chk (char *dest, const char *src, size_t n, size_t destlen)
  28 {
  29   char c;
  30   char *s = dest;
  31
  32   if (__builtin_expect (destlen < n, 0))
  33     __chk_fail ();
  34
  35   if (n >= 4)
  36     {
  37       size_t n4 = n >> 2;
  38
  39       for (;;)
  40         {
  41           c = *src++;
  42           *dest++ = c;
  43           if (c == '\0')
  44             break;
  45           c = *src++;
  46           *dest++ = c;
  47           if (c == '\0')
  48             break;
  49           c = *src++;
  50           *dest++ = c;
  51           if (c == '\0')
  52             break;
  53           c = *src++;
  54           *dest++ = c;
  55           if (c == '\0')
  56             break;
  57           if (--n4 == 0)
  58             goto last_chars;
  59         }
  60       n -= dest - s;
  61       goto zero_fill;
  62     }
  63
  64  last_chars:
  65   n &= 3;
  66   if (n == 0)
  67     return dest;
  68
  69   for (;;)
  70     {
  71       c = *src++;
  72       --n;
  73       *dest++ = c;
  74       if (c == '\0')
  75         break;
  76       if (n == 0)
  77         return dest;
  78     }
  79
  80  zero_fill:
  81   while (n-- > 0)
  82     dest[n] = '\0';
  83
  84   return dest - 1;
  85 }