rizin/librz/util/sdb/src/num.c
Riccardo Schirone faac2436b1 Move SDB into RzUtil
* Make SDB code SPDX compliant
* Use some RzUtil functions inside SDB

util: compile natively without any extra dependency
2022-07-19 08:45:20 +02:00

89 lines
2.3 KiB
C

// SPDX-FileCopyrightText: 2011-2016 pancake <pancake@nopcode.org>
// SPDX-License-Identifier: MIT
#include "sdb.h"
#include <rz_types.h>
// check if key exists and if it's a number.. rename?
RZ_API bool sdb_num_exists(Sdb *s, const char *key) {
const char *o = sdb_const_get(s, key, NULL);
return o ? (*o >= '0' && *o <= '9') : false;
}
RZ_API ut64 sdb_num_get(Sdb *s, const char *key, ut32 *cas) {
const char *v = sdb_const_get(s, key, cas);
return (!v || *v == '-') ? 0LL : sdb_atoi(v);
}
RZ_API int sdb_num_add(Sdb *s, const char *key, ut64 v, ut32 cas) {
char *val, b[SDB_NUM_BUFSZ];
int numbase = sdb_num_base(sdb_const_get(s, key, NULL));
val = sdb_itoa(v, b, numbase);
return sdb_add(s, key, val, cas);
}
RZ_API int sdb_num_set(Sdb *s, const char *key, ut64 v, ut32 cas) {
char *val, b[SDB_NUM_BUFSZ];
int numbase = sdb_num_base(sdb_const_get(s, key, NULL));
val = sdb_itoa(v, b, numbase);
return sdb_set(s, key, val, cas);
}
RZ_API ut64 sdb_num_inc(Sdb *s, const char *key, ut64 n2, ut32 cas) {
ut32 c;
ut64 n = sdb_num_get(s, key, &c);
ut64 res = n + n2;
if ((cas && c != cas) || res < n) {
return 0LL;
}
sdb_num_set(s, key, res, cas);
return res;
}
RZ_API ut64 sdb_num_dec(Sdb *s, const char *key, ut64 n2, ut32 cas) {
ut32 c;
ut64 n = sdb_num_get(s, key, &c);
if (cas && c != cas) {
return 0LL;
}
if (n2 > n) {
sdb_set(s, key, "0", cas);
return 0LL; // XXX must be -1LL?
}
n -= n2;
sdb_num_set(s, key, n, cas);
return n;
}
RZ_API int sdb_num_min(Sdb *db, const char *k, ut64 n, ut32 cas) {
const char *a = sdb_const_get(db, k, NULL);
return (!a || n < sdb_atoi(a))
? sdb_num_set(db, k, n, cas)
: 0;
}
RZ_API int sdb_num_max(Sdb *db, const char *k, ut64 n, ut32 cas) {
const char *a = sdb_const_get(db, k, NULL);
return (!a || n > sdb_atoi(a))
? sdb_num_set(db, k, n, cas)
: 0;
}
RZ_API int sdb_bool_set(Sdb *db, const char *str, bool v, ut32 cas) {
return sdb_set(db, str, v ? "true" : "false", cas);
}
RZ_API bool sdb_bool_get(Sdb *db, const char *str, ut32 *cas) {
const char *b = sdb_const_get(db, str, cas);
return b && (!strcmp(b, "1") || !strcmp(b, "true"));
}
/* pointers */
RZ_API int sdb_ptr_set(Sdb *db, const char *key, void *p, ut32 cas) {
return sdb_num_set(db, key, (ut64)(size_t)p, cas);
}
RZ_API void *sdb_ptr_get(Sdb *db, const char *key, ut32 *cas) {
return (void *)(size_t)sdb_num_get(db, key, cas);
}