noit_console_state_t *
noit_console_state_alloc(void) {
noit_console_state_t *s;
s = calloc(1, sizeof(*s));
noit_skiplist_init(&s->cmds);
noit_skiplist_set_compare(&s->cmds, cmd_info_compare, cmd_info_comparek);
noit_console_state_add_cmd(s,
NCSCMD("apply", noit_console_generic_apply, NULL, NULL, NULL));
noit_console_state_add_cmd(s, &console_command_help);
return s;
}
static void
register_console_check_commands() {
noit_console_state_t *tl;
cmd_info_t *showcmd;
tl = noit_console_state_initial();
showcmd = noit_console_state_get_cmd(tl, "show");
assert(showcmd && showcmd->dstate);
noit_console_state_add_cmd(showcmd->dstate,
NCSCMD("checks", noit_console_show_checks, NULL, NULL, NULL));
noit_console_state_add_cmd(showcmd->dstate,
NCSCMD("watches", noit_console_show_watchlist, NULL, NULL, NULL));
}
noit_console_state_t *
noit_console_state_initial() {
static noit_console_state_t *_top_level_state = NULL;
if(!_top_level_state) {
static noit_console_state_t *no_state, *show_state, *evdeb;
_top_level_state = noit_console_state_alloc();
noit_console_state_add_cmd(_top_level_state, &console_command_exit);
show_state = noit_console_mksubdelegate(_top_level_state, "show");
no_state = noit_console_mksubdelegate(_top_level_state, "no");
noit_console_state_add_cmd(_top_level_state, &console_command_shutdown);
noit_console_state_add_cmd(_top_level_state, &console_command_restart);
noit_console_state_add_cmd(show_state, &console_command_version);
(void)no_state;
evdeb = noit_console_mksubdelegate(
noit_console_mksubdelegate(show_state,
"eventer"),
"debug");
noit_console_state_add_cmd(evdeb, &console_command_eventer_timers);
noit_console_state_add_cmd(evdeb, &console_command_eventer_sockets);
noit_console_state_add_cmd(evdeb, &console_command_eventer_jobq);
}
return _top_level_state;
}
void
noit_console_add_help(const char *topic, console_cmd_func_t topic_func,
console_opt_func_t ac) {
noit_console_state_t *s = console_command_help.dstate;
if(!s) {
console_command_help.dstate = s = calloc(1, sizeof(*s));
noit_skiplist_init(&s->cmds);
noit_skiplist_set_compare(&s->cmds, cmd_info_compare, cmd_info_comparek);
}
noit_console_state_add_cmd(s, NCSCMD(topic, topic_func, ac, NULL, NULL));
}
static void
register_console_fq_commands() {
noit_console_state_t *tl;
cmd_info_t *showcmd;
tl = noit_console_state_initial();
showcmd = noit_console_state_get_cmd(tl, "show");
assert(showcmd && showcmd->dstate);
noit_console_state_add_cmd(showcmd->dstate,
NCSCMD("fq", noit_console_show_fq, NULL, NULL, NULL));
}
noit_console_state_t *
noit_console_mksubdelegate(noit_console_state_t *parent, const char *cmd) {
noit_console_state_t *child;
cmd_info_t *existing;
existing = noit_console_state_get_cmd(parent, cmd);
if(existing) return existing->dstate;
child = noit_console_state_alloc();
noit_console_state_add_cmd(parent,
NCSCMD(cmd, noit_console_state_delegate,
noit_console_opt_delegate, child, NULL));
return child;
}
static void
register_console_dns_cache_commands() {
noit_console_state_t *tl;
cmd_info_t *showcmd, *nocmd;
tl = noit_console_state_initial();
showcmd = noit_console_state_get_cmd(tl, "show");
assert(showcmd && showcmd->dstate);
nocmd = noit_console_state_get_cmd(tl, "no");
assert(nocmd && nocmd->dstate);
noit_console_state_add_cmd(showcmd->dstate,
NCSCMD("dns_cache", noit_console_show_dns_cache, NULL, NULL, NULL));
noit_console_state_add_cmd(tl,
NCSCMD("dns_cache", noit_console_manip_dns_cache, NULL, NULL, NULL));
noit_console_state_add_cmd(nocmd->dstate,
NCSCMD("dns_cache", noit_console_manip_dns_cache, NULL, NULL, (void *)0x1));
}
void
noit_console_state_add_check_attrs(noit_console_state_t *state,
console_cmd_func_t f,
const char *scope) {
int i;
for(i = 0;
i < sizeof(valid_attrs)/sizeof(valid_attrs[0]);
i++) {
if(strcmp(valid_attrs[i].scope, scope)) continue;
noit_console_state_add_cmd(state,
NCSCMD(valid_attrs[i].name, f, NULL,
NULL, &valid_attrs[i]));
}
}
static void
register_console_filter_commands() {
noit_console_state_t *tl, *filterset_state, *nostate;
cmd_info_t *confcmd, *conf_t_cmd, *no_cmd;
tl = noit_console_state_initial();
confcmd = noit_console_state_get_cmd(tl, "configure");
assert(confcmd && confcmd->dstate);
conf_t_cmd = noit_console_state_get_cmd(confcmd->dstate, "terminal");
assert(conf_t_cmd && conf_t_cmd->dstate);
nostate = noit_console_state_alloc();
noit_console_state_add_cmd(nostate,
NCSCMD("rule", noit_console_rule_configure, NULL, NULL, (void *)1));
filterset_state = noit_console_state_alloc();
noit_console_state_add_cmd(filterset_state,
NCSCMD("exit", noit_console_config_cd, NULL, NULL, ".."));
noit_console_state_add_cmd(filterset_state,
NCSCMD("status", noit_console_filter_show, NULL, NULL, NULL));
noit_console_state_add_cmd(filterset_state,
NCSCMD("rule", noit_console_rule_configure, NULL, NULL, NULL));
noit_console_state_add_cmd(filterset_state,
NCSCMD("no", noit_console_state_delegate, noit_console_opt_delegate,
nostate, NULL));
filterset_state->console_prompt_function = conf_t_filterset_prompt;
noit_console_state_add_cmd(conf_t_cmd->dstate,
NCSCMD("filterset", noit_console_filter_configure,
NULL, filterset_state, NULL));
no_cmd = noit_console_state_get_cmd(conf_t_cmd->dstate, "no");
assert(no_cmd && no_cmd->dstate);
noit_console_state_add_cmd(no_cmd->dstate,
NCSCMD("filterset", noit_console_filter_configure, NULL, NULL, (void *)1));
}
