oysteikt/heimdal
0
0
Fork 0
Code Issues 5 Pull Requests Releases Activity

roken: Unparse wider ints

Browse Source
This commit is contained in:
Nicolas Williams
2021-12-17 13:17:59 -06:00
committed by Luke Howard
parent fc76c83ab1
commit 686d5116de
7 changed files with 238 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

252
lib/roken/parse_units.c
View File

@@ -44,15 +44,15 @@
* def_unit defines the default unit.
*/
static int
parse_something (const char *s, const struct units *units,
const char *def_unit,
int (*func)(int res, int val, unsigned mult),
int init,
int accept_no_val_p)
static int64_t
parse_something_signed(const char *s, const struct units *units,
const char *def_unit,
int64_t (*func)(int64_t res, int64_t val, uint64_t mult),
int64_t init,
int accept_no_val_p)
{
const char *p;
int res = init;
int64_t res = init;
unsigned def_mult = 1;
if (def_unit != NULL) {
@@ -70,7 +70,7 @@ parse_something (const char *s, const struct units *units,
p = s;
while (*p) {
int val;
int64_t val;
char *next;
const struct units *u, *partial_unit;
size_t u_len;
@@ -80,7 +80,104 @@ parse_something (const char *s, const struct units *units,
while (isspace((unsigned char)*p) || *p == ',')
++p;
val = strtol(p, &next, 0);
val = strtoll(p, &next, 0);
if (p == next) {
val = 0;
if(!accept_no_val_p)
return -1;
no_val_p = 1;
}
p = next;
while (isspace((unsigned char)*p))
++p;
if (*p == '\0') {
res = (*func)(res, val, def_mult);
if (res < 0)
return res;
break;
} else if (*p == '+') {
++p;
val = 1;
} else if (*p == '-') {
++p;
val = -1;
}
if (no_val_p && val == 0)
val = 1;
u_len = strcspn (p, ", \t");
partial = 0;
partial_unit = NULL;
if (u_len > 1 && p[u_len - 1] == 's')
--u_len;
for (u = units; u->name; ++u) {
if (strncasecmp (p, u->name, u_len) == 0) {
if (u_len == strlen (u->name)) {
p += u_len;
res = (*func)(res, val, u->mult);
if (res < 0)
return res;
break;
} else {
++partial;
partial_unit = u;
}
}
}
if (u->name == NULL) {
if (partial == 1) {
p += u_len;
res = (*func)(res, val, partial_unit->mult);
if (res < 0)
return res;
} else {
return -1;
}
}
if (*p == 's')
++p;
while (isspace((unsigned char)*p))
++p;
}
return res;
}
static uint64_t
parse_something_unsigned(const char *s, const struct units *units,
const char *def_unit,
uint64_t (*func)(uint64_t res, int64_t val, uint64_t mult),
uint64_t init,
int accept_no_val_p)
{
const char *p;
int64_t res = init;
unsigned def_mult = 1;
if (def_unit != NULL) {
const struct units *u;
for (u = units; u->name; ++u) {
if (strcasecmp (u->name, def_unit) == 0) {
def_mult = u->mult;
break;
}
}
if (u->name == NULL)
return -1;
}
p = s;
while (*p) {
int64_t val;
char *next;
const struct units *u, *partial_unit;
size_t u_len;
unsigned partial;
int no_val_p = 0;
while (isspace((unsigned char)*p) || *p == ',')
++p;
val = strtoll(p, &next, 0);
if (p == next) {
val = 0;
if(!accept_no_val_p)
@@ -145,17 +242,17 @@ parse_something (const char *s, const struct units *units,
* The string consists of a sequence of `n unit'
*/
static int
acc_units(int res, int val, unsigned mult)
static int64_t
acc_units(int64_t res, int64_t val, uint64_t mult)
{
return res + val * mult;
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
ROKEN_LIB_FUNCTION int64_t ROKEN_LIB_CALL
parse_units (const char *s, const struct units *units,
const char *def_unit)
{
return parse_something (s, units, def_unit, acc_units, 0, 0);
return parse_something_signed(s, units, def_unit, acc_units, 0, 0);
}
/*
@@ -164,8 +261,8 @@ parse_units (const char *s, const struct units *units,
* the function value.
*/
static int
acc_flags(int res, int val, unsigned mult)
static uint64_t
acc_flags(uint64_t res, int64_t val, uint64_t mult)
{
if(val == 1)
return res | mult;
@@ -177,11 +274,11 @@ acc_flags(int res, int val, unsigned mult)
return -1;
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
ROKEN_LIB_FUNCTION uint64_t ROKEN_LIB_CALL
parse_flags (const char *s, const struct units *units,
int orig)
{
return parse_something (s, units, NULL, acc_flags, orig, 1);
return parse_something_unsigned (s, units, NULL, acc_flags, orig, 1);
}
/*
@@ -190,17 +287,60 @@ parse_flags (const char *s, const struct units *units,
*/
static int
unparse_something (int64_t num, const struct units *units, char *s, size_t len,
long long (*get_divisor) (long long, unsigned long long),
int (*print) (char *, size_t, long long, const char *, int),
int (*update) (long long, unsigned long long),
const char *zero_string)
unparse_something_signed(int64_t num, const struct units *units, char *s,
size_t len,
int64_t (*get_divisor)(int64_t, uint64_t),
int (*print)(char *, size_t, int64_t, const char *, int64_t),
int64_t (*update)(int64_t, uint64_t),
const char *zero_string)
{
const struct units *u;
int ret = 0, tmp;
if (num == 0)
return snprintf (s, len, "%s", zero_string);
if (len)
s[0] = '\0';
if (num < 0)
return -1;
for (u = units; num > 0 && u->name; ++u) {
long long divisor = get_divisor(num, u->mult);
if (divisor) {
num = (*update)(num, u->mult);
tmp = (*print)(s, len, divisor, u->name, num);
if (tmp < 0)
return tmp;
if ((size_t)tmp > len) {
len = 0;
s = NULL;
} else {
len -= tmp;
s += tmp;
}
ret += tmp;
}
}
return ret;
}
static int
unparse_something_unsigned(uint64_t num, const struct units *units, char *s,
size_t len,
uint64_t (*get_divisor)(uint64_t, uint64_t),
int (*print)(char *, size_t, uint64_t, const char *, uint64_t),
uint64_t (*update)(uint64_t, uint64_t),
const char *zero_string)
{
const struct units *u;
int64_t tmp;
int ret = 0;
if (num == 0)
return snprintf (s, len, "%s", zero_string);
if (len)
s[0] = '\0';
for (u = units; num > 0 && u->name; ++u) {
long long divisor = get_divisor(num, u->mult);
@@ -224,28 +364,26 @@ unparse_something (int64_t num, const struct units *units, char *s, size_t len,
}
static int
print_unit(char *s, size_t len, long long divisor, const char *name, int rem)
print_unit(char *s, size_t len, int64_t divisor, const char *name, int64_t rem)
{
return snprintf (s, len, "%lld %s%s%s",
divisor, name,
divisor == 1 ? "" : "s",
rem > 0 ? " " : "");
return snprintf(s, len, "%lld %s%s%s", (long long)divisor, name,
divisor == 1 ? "" : "s", rem > 0 ? " " : "");
}
static long long
get_divisor_unit(long long in, unsigned long long mult)
static int64_t
get_divisor_unit(int64_t in, uint64_t mult)
{
return in / mult;
}
static int
update_unit(long long in, unsigned long long mult)
static int64_t
update_unit(int64_t in, uint64_t mult)
{
return in % mult;
}
static int
update_unit_approx (long long in, unsigned long long mult)
static int64_t
update_unit_approx(int64_t in, uint64_t mult)
{
if (in / mult > 0)
return 0;
@@ -254,23 +392,18 @@ update_unit_approx (long long in, unsigned long long mult)
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
unparse_units (int num, const struct units *units, char *s, size_t len)
unparse_units(int64_t num, const struct units *units, char *s, size_t len)
{
return unparse_something (num, units, s, len,
get_divisor_unit,
print_unit,
update_unit,
"0");
return unparse_something_signed(num, units, s, len,
get_divisor_unit, print_unit, update_unit,
"0");
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
unparse_units_approx (int num, const struct units *units, char *s, size_t len)
unparse_units_approx(int64_t num, const struct units *units, char *s, size_t len)
{
return unparse_something (num, units, s, len,
get_divisor_unit,
print_unit,
update_unit_approx,
"0");
return unparse_something_signed(num, units, s, len, get_divisor_unit,
print_unit, update_unit_approx, "0");
}
ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
@@ -306,32 +439,29 @@ print_units_table (const struct units *units, FILE *f)
}
}
static long long
get_divisor_flag(long long in, unsigned long long mult)
static uint64_t
get_divisor_flag(uint64_t in, uint64_t mult)
{
return in & mult;
}
static int
print_flag (char *s, size_t len, long long divisor, const char *name, int rem)
print_flag(char *s, size_t len, uint64_t divisor, const char *name, uint64_t rem)
{
return snprintf (s, len, "%s%s", name, rem > 0 ? ", " : "");
}
static int
update_flag (long long in, unsigned long long mult)
static uint64_t
update_flag(uint64_t in, uint64_t mult)
{
return in & ~mult;
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
unparse_flags (int num, const struct units *units, char *s, size_t len)
unparse_flags (uint64_t num, const struct units *units, char *s, size_t len)
{
return unparse_something (num, units, s, len,
get_divisor_flag,
print_flag,
update_flag,
"");
return unparse_something_unsigned(num, units, s, len, get_divisor_flag,
print_flag, update_flag, "");
}
ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
@@ -351,7 +481,7 @@ print_flags_table (const struct units *units, FILE *f)
#undef unparse_flags
#undef print_flags_table
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
ROKEN_LIB_FUNCTION int64_t ROKEN_LIB_CALL
parse_units(const char *s, const struct units *units,
const char *def_unit)
{
@@ -359,13 +489,13 @@ parse_units(const char *s, const struct units *units,
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
unparse_units(int num, const struct units *units, char *s, size_t len)
unparse_units(int64_t num, const struct units *units, char *s, size_t len)
{
return rk_unparse_units(num, units, s, len);
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
unparse_units_approx(int num, const struct units *units, char *s, size_t len)
unparse_units_approx(int64_t num, const struct units *units, char *s, size_t len)
{
return rk_unparse_units_approx(num, units, s, len);
}
@@ -376,14 +506,14 @@ print_units_table(const struct units *units, FILE *f)
rk_print_units_table(units, f);
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
ROKEN_LIB_FUNCTION uint64_t ROKEN_LIB_CALL
parse_flags(const char *s, const struct units *units, int orig)
{
return rk_parse_flags(s, units, orig);
}
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
unparse_flags(int num, const struct units *units, char *s, size_t len)
unparse_flags(uint64_t num, const struct units *units, char *s, size_t len)
{
return rk_unparse_flags(num, units, s, len);
}