strlcpy.h

   1 /*
   2  * Copyright (c) 1998 Todd C. Miller <Todd.Miller@courtesan.com>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16 #ifndef BITCOIN_STRLCPY_H
  17 #define BITCOIN_STRLCPY_H
  18
  19 #include <stdlib.h>
  20 #include <string.h>
  21
  22 /*
  23  * Copy src to string dst of size siz.  At most siz-1 characters
  24  * will be copied.  Always NUL terminates (unless siz == 0).
  25  * Returns strlen(src); if retval >= siz, truncation occurred.
  26  */
  27 inline size_t strlcpy(char *dst, const char *src, size_t siz)
  28 {
  29     char *d = dst;
  30     const char *s = src;
  31     size_t n = siz;
  32
  33     /* Copy as many bytes as will fit */
  34     if (n != 0)
  35     {
  36         while (--n != 0)
  37         {
  38             if ((*d++ = *s++) == '\0')
  39                 break;
  40         }
  41     }
  42
  43     /* Not enough room in dst, add NUL and traverse rest of src */
  44     if (n == 0)
  45     {
  46         if (siz != 0)
  47             *d = '\0';  /* NUL-terminate dst */
  48         while (*s++)
  49             ;
  50     }
  51
  52     return(s - src - 1); /* count does not include NUL */
  53 }
  54
  55 /*
  56  * Appends src to string dst of size siz (unlike strncat, siz is the
  57  * full size of dst, not space left).  At most siz-1 characters
  58  * will be copied.  Always NUL terminates (unless siz <= strlen(dst)).
  59  * Returns strlen(src) + MIN(siz, strlen(initial dst)).
  60  * If retval >= siz, truncation occurred.
  61  */
  62 inline size_t strlcat(char *dst, const char *src, size_t siz)
  63 {
  64     char *d = dst;
  65     const char *s = src;
  66     size_t n = siz;
  67     size_t dlen;
  68
  69     /* Find the end of dst and adjust bytes left but don't go past end */
  70     while (n-- != 0 && *d != '\0')
  71         d++;
  72     dlen = d - dst;
  73     n = siz - dlen;
  74
  75     if (n == 0)
  76         return(dlen + strlen(s));
  77     while (*s != '\0')
  78     {
  79         if (n != 1)
  80         {
  81             *d++ = *s;
  82             n--;
  83         }
  84         s++;
  85     }
  86     *d = '\0';
  87
  88     return(dlen + (s - src)); /* count does not include NUL */
  89 }
  90 #endif