mir/bind
2
0
Fork 0
mirror of https://gitlab.isc.org/isc-projects/bind9 synced 2025-08-30 14:07:59 +00:00
Code Issues Releases Wiki Activity

hack statschannel to access cfgmgr data

Browse Source
This commit is contained in:
Colin Vidal 2025-06-02 13:53:53 +02:00
parent 921ba8e61d
commit 497166a7eb
1 changed files with 191 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

191
bin/named/statschannel.c
View File

@ -17,6 +17,7 @@
#include <stdbool.h>
#include <isc/buffer.h>
#include <isc/cfgmgr.h>
#include <isc/httpd.h>
#include <isc/mem.h>
#include <isc/once.h>
@ -3494,6 +3495,192 @@ render_json_traffic(const isc_httpd_t *httpd, const isc_httpdurl_t *urlinfo,
b, freecb, freecb_args);
}
typedef struct cfgmgr_dumpctx cfgmgr_dumpctx_t;
struct cfgmgr_dumpctx {
size_t indent;
size_t lvl;
isc_buffer_t *buffer;
};
static void
statschannel__cfgmgr_dumpprop(void *state, const char *name,
const isc_cfgmgr_val_t *val) {
cfgmgr_dumpctx_t *ctx = state;
for (size_t i = 0; i < ctx->lvl * ctx->indent; i++) {
isc_buffer_putstr(ctx->buffer, " ");
}
isc_buffer_printf(ctx->buffer, "%s ", name);
switch (val->type) {
case ISC_CFGMGR_UINT32:
isc_buffer_printf(ctx->buffer, "%u", val->uint32);
break;
case ISC_CFGMGR_STRING:
isc_buffer_printf(ctx->buffer, "%s", val->string);
break;
case ISC_CFGMGR_REF:
isc_buffer_printf(ctx->buffer, "%p", val->ptr);
break;
default:
UNREACHABLE();
}
isc_buffer_putstr(ctx->buffer, ";\n");
}
static void
statschannel__cfgmgr_dumplabeldown(void *state, const char *name) {
cfgmgr_dumpctx_t *ctx = state;
for (size_t i = 0; i < ctx->lvl * ctx->indent; i++) {
isc_buffer_putstr(ctx->buffer, " ");
}
isc_buffer_printf(ctx->buffer, "%s {\n", name);
ctx->lvl++;
}
static void
statschannel__cfgmgr_dumplabelup(void *state) {
cfgmgr_dumpctx_t *dump = state;
dump->lvl--;
for (size_t i = 0; i < dump->lvl * dump->indent; i++) {
isc_buffer_putstr(dump->buffer, " ");
}
isc_buffer_printf(dump->buffer, "};\n");
}
static void
statschannel__cfgmgr_dump(isc_cfgmgr_mode_t mode, isc_buffer_t *buffer) {
cfgmgr_dumpctx_t ctx = { .lvl = 0, .indent = 8, .buffer = buffer };
isc_cfgmgr_mode(mode);
isc_cfgmgr_txn();
isc_cfgmgr_foreach("/", 0, &ctx, statschannel__cfgmgr_dumpprop,
statschannel__cfgmgr_dumplabeldown,
statschannel__cfgmgr_dumplabelup);
isc_cfgmgr_closetxn();
}
static void
statschannel__cfgmgr_dumpfree(isc_buffer_t *buffer, void *arg) {
isc_mem_t *mctx = arg;
void *data = isc_buffer_base(buffer);
isc_buffer_initnull(buffer);
isc_mem_free(mctx, data);
}
static isc_result_t
render_cfgmgr_dump(const isc_httpd_t *httpd, const isc_httpdurl_t *urlinfo,
void *arg, unsigned int *retcode, const char **retmsg,
const char **mimetype, isc_buffer_t *b,
isc_httpdfree_t **freecb, void **freecb_args) {
const size_t bufsz = 16360008;
named_server_t *server = arg;
char *bufdata = isc_mem_allocate(server->mctx, bufsz);
UNUSED(httpd);
UNUSED(urlinfo);
UNUSED(retmsg);
*retcode = 200;
*retmsg = "OK";
*mimetype = "text/html";
*freecb = statschannel__cfgmgr_dumpfree;
*freecb_args = server->mctx;
isc_buffer_reinit(b, bufdata, bufsz);
isc_buffer_putstr(b, "<pre>");
isc_buffer_putstr(b, "# Builtin config dump\n\n");
statschannel__cfgmgr_dump(ISC_CFGMGR_MODEBUILTIN, b);
isc_buffer_putstr(b, "\n\n\n\n# User/named.conf config dump\n\n");
statschannel__cfgmgr_dump(ISC_CFGMGR_MODEUSER, b);
isc_buffer_putstr(b, "\n\n\n\n# Running config dump\n\n");
statschannel__cfgmgr_dump(ISC_CFGMGR_MODERUNNING, b);
isc_buffer_putstr(b, "</pre>");
return ISC_R_SUCCESS;
}
static isc_result_t
cfgmgr_write(const isc_httpd_t *httpd, const isc_httpdurl_t *urlinfo, void *arg,
unsigned int *retcode, const char **retmsg, const char **mimetype,
isc_buffer_t *b, isc_httpdfree_t **freecb, void **freecb_args) {
isc_cfgmgr_val_t val;
const char *base;
uint16_t len;
UNUSED(httpd);
UNUSED(urlinfo);
UNUSED(retmsg);
UNUSED(retcode);
UNUSED(b);
UNUSED(freecb);
UNUSED(freecb_args);
UNUSED(mimetype);
UNUSED(arg);
*retcode = 200;
*retmsg = "OK";
base = NULL;
if (isc_httpd_geturlfield(httpd, ISC_UF_QUERY, &base, &len) ==
ISC_R_SUCCESS)
{
char query[512];
char *name;
char *value;
REQUIRE(len < 511);
memset(query, 0, sizeof(query));
strncpy(query, base, len);
/*
* if there is several queries options, let's pick only the
* first one and ignore the other
*/
name = strtok(query, "=");
value = strtok(NULL, "&");
REQUIRE(name != NULL);
REQUIRE(isc_cfgmgr_rwtxn() == ISC_R_SUCCESS);
if (value != NULL) {
val.type = ISC_CFGMGR_STRING;
val.string = value;
REQUIRE(isc_cfgmgr_write(name, &val) == ISC_R_SUCCESS);
} else {
isc_cfgmgr_write(name, NULL);
}
if (strcmp(name, "/options/allow-query") == 0) {
dns_view_t *view = NULL;
dns_acl_t *newacl = NULL;
REQUIRE(dns_viewlist_find(&named_g_server->viewlist,
"_default", dns_rdataclass_in,
&view) == ISC_R_SUCCESS);
REQUIRE(dns_acl_configure(view->mctx, name,
&newacl) == ISC_R_SUCCESS);
dns_view_setacl(view, "allow-query", newacl);
}
REQUIRE(isc_cfgmgr_commit() == ISC_R_SUCCESS);
}
return ISC_R_SUCCESS;
}
#endif /* HAVE_JSON_C */
#if HAVE_LIBXML2
@ -3728,6 +3915,10 @@ add_listener(named_server_t *server, named_statschannel_t **listenerp,
isc_httpdmgr_addurl(listener->httpdmgr,
"/json/v" STATS_JSON_VERSION_MAJOR "/traffic",
false, render_json_traffic, server);
isc_httpdmgr_addurl(listener->httpdmgr, "/cfgmgrdump", false,
render_cfgmgr_dump, server);
isc_httpdmgr_addurl(listener->httpdmgr, "/cfgmgrwrite", false,
cfgmgr_write, server);
#endif /* ifdef HAVE_JSON_C */
*listenerp = listener;