src/basic/strxcpyx.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2013 Kay Sievers
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   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 License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 /*
  21  * Concatenates/copies strings. In any case, terminates in all cases
  22  * with '\0' * and moves the @dest pointer forward to the added '\0'.
  23  * Returns the * remaining size, and 0 if the string was truncated.
  24  */
  25
  26 #include <stdarg.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29
  30 #include "strxcpyx.h"
  31
  32 size_t strpcpy(char **dest, size_t size, const char *src) {
  33         size_t len;
  34
  35         len = strlen(src);
  36         if (len >= size) {
  37                 if (size > 1)
  38                         *dest = mempcpy(*dest, src, size-1);
  39                 size = 0;
  40         } else {
  41                 if (len > 0) {
  42                         *dest = mempcpy(*dest, src, len);
  43                         size -= len;
  44                 }
  45         }
  46         *dest[0] = '\0';
  47         return size;
  48 }
  49
  50 size_t strpcpyf(char **dest, size_t size, const char *src, ...) {
  51         va_list va;
  52         int i;
  53
  54         va_start(va, src);
  55         i = vsnprintf(*dest, size, src, va);
  56         if (i < (int)size) {
  57                 *dest += i;
  58                 size -= i;
  59         } else {
  60                 *dest += size;
  61                 size = 0;
  62         }
  63         va_end(va);
  64         *dest[0] = '\0';
  65         return size;
  66 }
  67
  68 size_t strpcpyl(char **dest, size_t size, const char *src, ...) {
  69         va_list va;
  70
  71         va_start(va, src);
  72         do {
  73                 size = strpcpy(dest, size, src);
  74                 src = va_arg(va, char *);
  75         } while (src != NULL);
  76         va_end(va);
  77         return size;
  78 }
  79
  80 size_t strscpy(char *dest, size_t size, const char *src) {
  81         char *s;
  82
  83         s = dest;
  84         return strpcpy(&s, size, src);
  85 }
  86
  87 size_t strscpyl(char *dest, size_t size, const char *src, ...) {
  88         va_list va;
  89         char *s;
  90
  91         va_start(va, src);
  92         s = dest;
  93         do {
  94                 size = strpcpy(&s, size, src);
  95                 src = va_arg(va, char *);
  96         } while (src != NULL);
  97         va_end(va);
  98
  99         return size;
 100 }