radare2/libr/config/config.c
2011-06-30 00:17:12 +02:00

287 lines
7.1 KiB
C

/* radare - LGPL - Copyright 2006-2011 pancake<nopcode.org> */
#include "r_config.h"
#include "r_util.h" // r_str_hash, r_str_chop, ...
R_API RConfigNode* r_config_node_new(const char *name, const char *value) {
RConfigNode *node = R_NEW (RConfigNode);
node->name = strdup (name);
node->hash = r_str_hash (name);
node->value = strdup (value?value:"");
node->flags = CN_RW | CN_STR;
node->i_value = 0;
node->callback = NULL;
return node;
}
R_API void r_config_list(RConfig *cfg, const char *str, boolt rad) {
RConfigNode *node;
RListIter *iter;
int len = 0;
if (!strnull (str)) {
str = r_str_chop_ro (str);
len = strlen (str);
}
r_list_foreach (cfg->nodes, iter, node) {
if (str) {
if (!strncmp (str, node->name, len))
cfg->printf ("%s%s = %s\n", rad?"e ":"",
node->name, node->value);
} else cfg->printf ("%s%s = %s\n", rad?"e ":"", node->name, node->value);
}
}
R_API RConfigNode *r_config_node_get(RConfig *cfg, const char *name) {
if (strnull (name))
return NULL;
return r_hashtable_lookup (cfg->ht, r_str_hash (name));
}
R_API const char *r_config_get(RConfig *cfg, const char *name) {
RConfigNode *node = r_config_node_get (cfg, name);
if (node) {
cfg->last_notfound = 0;
if (node->flags & CN_BOOL)
return (const char *)
(((!strcmp ("true", node->value))
|| (!strcmp ("1", node->value)))?
(const char *)"true":"false"); // XXX (char*)1 is ugly
return node->value;
}
cfg->last_notfound = 1;
return NULL;
}
R_API int r_config_swap(RConfig *cfg, const char *name) {
RConfigNode *node = r_config_node_get (cfg, name);
if (node && node->flags & CN_BOOL) {
r_config_set_i (cfg, name, !node->i_value);
return R_TRUE;
}
return R_FALSE;
}
R_API ut64 r_config_get_i(RConfig *cfg, const char *name) {
RConfigNode *node = r_config_node_get (cfg, name);
if (node) {
if (node->i_value != 0)
return node->i_value;
return (ut64)r_num_math (NULL, node->value);
}
return (ut64)0LL;
}
R_API RConfigNode *r_config_set_cb(RConfig *cfg, const char *name, const char *value, RConfigCallback cb) {
RConfigNode *node = r_config_set (cfg, name, value);
if ((node->callback = cb))
if (!cb (cfg->user, node))
return NULL;
return node;
}
R_API RConfigNode *r_config_set_i_cb(RConfig *cfg, const char *name, int ivalue, RConfigCallback cb) {
RConfigNode *node = r_config_set_i (cfg, name, ivalue);
if ((node->callback = cb))
if (!node->callback (cfg->user, node))
return NULL;
return node;
}
/* TODO: reduce number of strdups here */
R_API RConfigNode *r_config_set(RConfig *cfg, const char *name, const char *value) {
RConfigNode *node;
char *ov = NULL;
ut64 oi;
if (strnull (name))
return NULL;
node = r_config_node_get (cfg, name);
// TODO: store old value somewhere..
if (node) {
if (node->flags & CN_RO) {
eprintf ("(read only)\n");
return node;
}
oi = node->i_value;
if (node->value)
ov = strdup (node->value);
else node->value = strdup ("");
free (node->value);
if (node->flags & CN_BOOL) {
int b = (!strcmp (value,"true") || !strcmp (value,"1"));
node->i_value = (ut64)(b==0)?0:1;
node->value = strdup (b?"true":"false");
} else {
if (value == NULL) {
node->value = strdup ("");
node->i_value = 0;
} else {
node->value = strdup (value);
if (strchr(value, '/'))
node->i_value = r_num_get (NULL, value);
else node->i_value = r_num_math (NULL, value);
node->flags |= CN_INT;
}
}
} else {
oi = UT64_MAX;
if (!cfg->lock) {
node = r_config_node_new (name, value);
if (value && (!strcmp(value,"true")||!strcmp(value,"false"))) {
node->flags|=CN_BOOL;
node->i_value = (!strcmp(value,"true"))?1:0;
}
r_hashtable_insert (cfg->ht, node->hash, node);
r_list_append (cfg->nodes, node);
cfg->n_nodes++;
} else eprintf ("config is locked: cannot create '%s'\n", name);
}
if (node && node->callback) {
int ret = node->callback (cfg->user, node);
if (ret == R_FALSE) {
if (oi != UT64_MAX)
node->i_value = oi;
free (node->value);
node->value = strdup (ov);
return NULL;
}
}
free (ov);
return node;
}
R_API int r_config_rm(RConfig *cfg, const char *name) {
RConfigNode *node = r_config_node_get (cfg, name);
if (node) {
r_hashtable_remove (cfg->ht, node->hash);
r_list_delete_data (cfg->nodes, node);
cfg->n_nodes--;
return R_TRUE;
}
return R_FALSE;
}
R_API RConfigNode *r_config_set_i(RConfig *cfg, const char *name, const ut64 i) {
char buf[128], *ov = NULL;
RConfigNode *node = r_config_node_get (cfg, name);
if (node) {
if (node->flags & CN_RO)
return NULL;
if (node->value)
ov = strdup (node->value);
else node->value = strdup("");
free (node->value);
if (node->flags & CN_BOOL) {
node->value = strdup(i?"true":"false");
} else {
sprintf (buf, "%"PFMT64d"", i); //0x%08lx", i);
node->value = strdup(buf);
}
//node->flags = CN_RW | CN_INT;
node->i_value = i;
} else {
if (!cfg->lock) {
if (i<1024) sprintf (buf, "%"PFMT64d"", i);
else sprintf (buf, "0x%08"PFMT64x"", i);
node = r_config_node_new (name, buf);
node->flags = CN_RW | CN_OFFT;
node->i_value = i;
r_hashtable_insert (cfg->ht, node->hash, node);
r_list_append (cfg->nodes, node);
cfg->n_nodes++;
} else eprintf ("(locked: no new keys can be created (%s))\n", name);
}
if (node && node->callback) {
ut64 oi = node->i_value;
int ret = node->callback(cfg->user, node);
if (ret == R_FALSE) {
node->i_value = oi;
free(node->value);
node->value = strdup(ov);
}
}
free (ov);
return node;
}
R_API int r_config_eval(RConfig *cfg, const char *str) {
char *ptr, *a, *b, name[1024];
int len = strlen (str)+1;
if (len >=sizeof (name))
return R_FALSE;
memcpy (name, str, len);
str = r_str_chop (name);
if (str == NULL)
return R_FALSE;
if (str[0]=='\0' || !strcmp (str, "help")) {
r_config_list (cfg, NULL, 0);
return R_FALSE;
}
if (str[0]=='-') {
r_config_rm (cfg, str+1);
return R_FALSE;
}
ptr = strchr (str, '=');
if (ptr) {
/* set */
ptr[0]='\0';
a = r_str_chop(name);
b = r_str_chop(ptr+1);
r_config_set(cfg, a, b);
} else {
char *foo = r_str_chop (name);
if (foo[strlen(foo)-1]=='.') {
r_config_list (cfg, name, 0);
return R_FALSE;
} else {
/* get */
const char *str = r_config_get(cfg, foo);
if (cfg->last_notfound)
r_config_list (cfg, name, 0);
else cfg->printf ("%s\n", (((int)(size_t)str)==1)?"true":
(str==0)?"false":str);
}
}
return R_TRUE;
}
R_API void r_config_lock(RConfig *cfg, int l) {
cfg->lock = l;
}
R_API RConfig *r_config_new(void *user) {
RConfig *cfg = R_NEW (RConfig);
if (cfg) {
cfg->ht = r_hashtable_new ();
cfg->nodes = r_list_new ();
cfg->nodes->free = free;
cfg->user = user;
cfg->n_nodes = 0;
cfg->lock = 0;
cfg->printf = (void *)printf;
}
return cfg;
}
R_API int r_config_free(RConfig *cfg) {
// XXX: memory leak ! r_list_destroy (cfg->nodes);
r_list_free (cfg->nodes);
r_hashtable_free (cfg->ht);
free (cfg);
return 0;
}
R_API void r_config_visual_hit_i(RConfig *cfg, const char *name, int delta) {
RConfigNode *node = r_config_node_get (cfg, name);
if (node && (node->flags & CN_INT || node->flags & CN_OFFT))
r_config_set_i(cfg, name, r_config_get_i(cfg, name)+delta);
}