diff options
Diffstat (limited to 'src/utils.c')
-rw-r--r-- | src/utils.c | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/src/utils.c b/src/utils.c index be1b5d8..4a0bb9c 100644 --- a/src/utils.c +++ b/src/utils.c @@ -14,6 +14,8 @@ #include <limits.h> #include <stdint.h> #include <arpa/inet.h> +#include <errno.h> +#include <inttypes.h> const char *nft_family2str(uint32_t family) { @@ -44,3 +46,74 @@ int nft_str2family(const char *family) return -1; } + +static struct { + int len; + int64_t min; + uint64_t max; +} basetype[] = { + [NFT_TYPE_U8] = { .len = sizeof(uint8_t), .max = UINT8_MAX }, + [NFT_TYPE_U16] = { .len = sizeof(uint16_t), .max = UINT16_MAX }, + [NFT_TYPE_U32] = { .len = sizeof(uint32_t), .max = UINT32_MAX }, + [NFT_TYPE_U64] = { .len = sizeof(uint64_t), .max = UINT64_MAX }, + [NFT_TYPE_S8] = { .len = sizeof(int8_t), .min = INT8_MIN, .max = INT8_MAX }, + [NFT_TYPE_S16] = { .len = sizeof(int16_t), .min = INT16_MIN, .max = INT16_MAX }, + [NFT_TYPE_S32] = { .len = sizeof(int32_t), .min = INT32_MIN, .max = INT32_MAX }, + [NFT_TYPE_S64] = { .len = sizeof(int64_t), .min = INT64_MIN, .max = INT64_MAX }, +}; + +int nft_strtoi(const char *string, int base, void *out, enum nft_type type) +{ + int64_t sval = 0; + uint64_t uval = -1; + char *endptr; + + switch (type) { + case NFT_TYPE_U8: + case NFT_TYPE_U16: + case NFT_TYPE_U32: + case NFT_TYPE_U64: + uval = strtoll(string, &endptr, base); + break; + case NFT_TYPE_S8: + case NFT_TYPE_S16: + case NFT_TYPE_S32: + case NFT_TYPE_S64: + sval = strtoull(string, &endptr, base); + break; + default: + errno = EINVAL; + return -1; + } + + if (*endptr) { + errno = EINVAL; + return -1; + } + + switch (type) { + case NFT_TYPE_U8: + case NFT_TYPE_U16: + case NFT_TYPE_U32: + case NFT_TYPE_U64: + if (uval > basetype[type].max) { + errno = ERANGE; + return -1; + } + memcpy(out, &uval, basetype[type].len); + break; + case NFT_TYPE_S8: + case NFT_TYPE_S16: + case NFT_TYPE_S32: + case NFT_TYPE_S64: + if (sval < basetype[type].min || + sval > (int64_t)basetype[type].max) { + errno = ERANGE; + return -1; + } + memcpy(out, &sval, basetype[type].len); + break; + } + + return 0; +} |