src/xstring.c

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * By Daniel Borkmann <daniel@netsniff-ng.org>
   4  * Copyright 2009, 2010, 2011, 2012 Daniel Borkmann.
   5  * Copyright 2009, 2010, 2011, 2012 Emmanuel Roullit.
   6  * strlcpy, Copyright 1991, 1992  Linus Torvalds.
   7  * Subject to the GPL, version 2.
   8  */
   9
  10 #define _BSD_SOURCE
  11 #include <string.h>
  12 #include <stdio.h>
  13 #include <stdarg.h>
  14 #include <ctype.h>
  15 #include <stdint.h>
  16
  17 #include "xstring.h"
  18 #include "built_in.h"
  19
  20 size_t strlcpy(char *dest, const char *src, size_t size)
  21 {
  22         size_t ret = strlen(src);
  23
  24         if (size) {
  25                 size_t len = (ret >= size) ? size - 1 : ret;
  26
  27                 memcpy(dest, src, len);
  28                 dest[len] = '\0';
  29         }
  30
  31         return ret;
  32 }
  33
  34 int slprintf(char *dst, size_t size, const char *fmt, ...)
  35 {
  36         int ret;
  37         va_list ap;
  38
  39         va_start(ap, fmt);
  40
  41         ret = vsnprintf(dst, size, fmt, ap);
  42         dst[size - 1] = '\0';
  43
  44         va_end(ap);
  45
  46         return ret;
  47 }
  48
  49 noinline void *xmemset(void *s, int c, size_t n)
  50 {
  51         size_t i;
  52         uint8_t *ptr = s;
  53
  54         for (i = 0; i < n; ++i)
  55                 ptr[i] = (uint8_t) c;
  56
  57         return ptr;
  58 }
  59
  60 char *getuint(char *in, uint32_t *out)
  61 {
  62         char *pt = in, tmp;
  63         char *endptr = NULL;
  64
  65         while (*in && (isdigit(*in) || isxdigit(*in) || *in == 'x'))
  66                 in++;
  67         if (!*in)
  68                 panic("Syntax error!\n");
  69         errno = 0;
  70         tmp = *in;
  71         *in = 0;
  72         *out = strtoul(pt, &endptr, 0);
  73         if ((endptr != NULL && *endptr != '\0') || errno != 0) {
  74                 panic("Syntax error!\n");
  75         }
  76         *in = tmp;
  77
  78         return in;
  79 }
  80
  81 char *strtrim_right(register char *p, register char c)
  82 {
  83         register char *end;
  84         register int len;
  85
  86         len = strlen(p);
  87         while (*p && len) {
  88                 end = p + len - 1;
  89                 if (c == *end)
  90                         *end = 0;
  91                 else
  92                         break;
  93                 len = strlen(p);
  94         }
  95
  96         return p;
  97 }
  98
  99 char *strtrim_left(register char *p, register char c)
 100 {
 101         register int len;
 102
 103         len = strlen(p);
 104         while (*p && len--) {
 105                 if (c == *p)
 106                         p++;
 107                 else
 108                         break;
 109         }
 110
 111         return p;
 112 }