diff --git a/libutil/eprintf.c b/libutil/eprintf.c index 407c502..4d8f726 100644 --- a/libutil/eprintf.c +++ b/libutil/eprintf.c @@ -33,9 +33,8 @@ enprintf(int status, const char *fmt, ...) void venprintf(int status, const char *fmt, va_list ap) { -#ifdef DEBUG - fprintf(stderr, "%s: ", argv0); -#endif + if (strncmp(fmt, "usage", strlen("usage"))) + fprintf(stderr, "%s: ", argv0); vfprintf(stderr, fmt, ap); @@ -52,9 +51,8 @@ weprintf(const char *fmt, ...) { va_list ap; -#ifdef DEBUG - fprintf(stderr, "%s: ", argv0); -#endif + if (strncmp(fmt, "usage", strlen("usage"))) + fprintf(stderr, "%s: ", argv0); va_start(ap, fmt); vfprintf(stderr, fmt, ap); diff --git a/libutil/strlcat.c b/libutil/strlcat.c index 9e2d251..bf263b8 100644 --- a/libutil/strlcat.c +++ b/libutil/strlcat.c @@ -50,3 +50,14 @@ strlcat(char *dst, const char *src, size_t siz) *d = '\0'; return(dlen + (s - src)); /* count does not include NUL */ } + +size_t +estrlcat(char *dst, const char *src, size_t siz) +{ + size_t ret; + + if ((ret = strlcat(dst, src, siz)) >= siz) + eprintf("strlcat: input string too long\n"); + + return ret; +} diff --git a/libutil/strlcpy.c b/libutil/strlcpy.c index 388b426..44b618a 100644 --- a/libutil/strlcpy.c +++ b/libutil/strlcpy.c @@ -46,3 +46,14 @@ strlcpy(char *dst, const char *src, size_t siz) } return(s - src - 1); /* count does not include NUL */ } + +size_t +estrlcpy(char *dst, const char *src, size_t siz) +{ + size_t ret; + + if ((ret = strlcpy(dst, src, siz)) >= siz) + eprintf("strlcpy: input string too long\n"); + + return ret; +} diff --git a/util.h b/util.h index d9b5f1a..7b832e7 100644 --- a/util.h +++ b/util.h @@ -40,13 +40,15 @@ void putword(const char *); /* recurse.c */ void recurse(const char *, void (*)(const char *)); -/* strlcpy.c */ +/* strlcat.c */ #undef strlcat size_t strlcat(char *, const char *, size_t); +size_t estrlcat(char *, const char *, size_t); -/* strlcat.c */ +/* strlcpy.c */ #undef strlcpy size_t strlcpy(char *, const char *, size_t); +size_t estrlcpy(char *, const char *, size_t); /* tty.c */ void devtotty(int, int *, int *);