emehta-contrib/os_resolve.c

80 lines
1.8 KiB
C

/*
* Copyright (c) 2011 William Pitcock <nenolod@dereferenced.org>
* Rights to this code are as documented in doc/LICENSE.
*
* Does an A record lookup.
*/
#include "atheme-compat.h"
DECLARE_MODULE_V1
(
"contrib/os_resolve", false, _modinit, _moddeinit,
PACKAGE_STRING,
"Atheme Development Group <http://www.atheme.org>"
);
static void os_cmd_resolve(sourceinfo_t *si, int parc, char *parv[]);
command_t cmd_resolve = { "RESOLVE", N_("Perform DNS lookup on hostname"), PRIV_ADMIN, 1, os_cmd_resolve, { .path = "contrib/os_resolve" } };
void _modinit(module_t *m)
{
service_named_bind_command("operserv", &cmd_resolve);
}
void _moddeinit(module_unload_intent_t intent)
{
service_named_unbind_command("operserv", &cmd_resolve);
}
typedef struct {
dns_query_t dns_query;
sourceinfo_t *si;
} resolve_req_t;
mowgli_heap_t *request_heap = NULL;
static void resolve_cb(void *vptr, dns_reply_t *reply)
{
resolve_req_t *req = vptr;
char buf[BUFSIZE];
return_if_fail(vptr != NULL);
return_if_fail(reply != NULL);
if (reply->addr.saddr.sa.sa_family != AF_INET)
return;
inet_ntop(reply->addr.saddr.sa.sa_family, &reply->addr.saddr.sin.sin_addr, buf, reply->addr.saddr_len);
command_success_nodata(req->si, "Result is %s", buf);
mowgli_heap_free(request_heap, req);
object_unref(req->si);
}
static void os_cmd_resolve(sourceinfo_t *si, int parc, char *parv[])
{
resolve_req_t *req;
if (request_heap == NULL)
request_heap = mowgli_heap_create(sizeof(resolve_req_t), 32, BH_LAZY);
if (!parv[0])
{
command_fail(si, fault_needmoreparams, STR_INSUFFICIENT_PARAMS, "RESOLVE");
return;
}
req = mowgli_heap_alloc(request_heap);
req->si = si;
req->dns_query.ptr = req;
req->dns_query.callback = resolve_cb;
gethost_byname_type(parv[0], &req->dns_query, T_A);
object_ref(req->si);
}