lokinet/include/llarp/dnsd.hpp

126 lines
3.6 KiB
C++
Raw Normal View History

2018-07-16 12:48:04 +00:00
#ifndef LIBLLARP_DNSD_HPP
#define LIBLLARP_DNSD_HPP
#include <llarp/ev.h> // for sockaadr
#include <string>
2018-10-03 10:59:49 +00:00
#include <llarp/dns.hpp> // question and dnsc
#include <llarp/dnsc.hpp>
2018-07-16 12:48:04 +00:00
//
// Input structures/functions:
//
2018-08-01 09:04:40 +00:00
// fwd declaration
struct dnsd_context;
2018-08-01 09:04:40 +00:00
/// sendto hook functor
2018-07-21 13:24:47 +00:00
typedef ssize_t (*sendto_dns_hook_func)(void *sock, const struct sockaddr *from,
const void *buffer, size_t length);
2018-09-29 10:22:00 +00:00
// FIXME: llarp::Addr
2018-07-16 12:48:04 +00:00
2018-08-01 09:04:40 +00:00
/// DNS server query request
2018-07-22 03:34:28 +00:00
struct dnsd_question_request
2018-07-16 12:48:04 +00:00
{
/// sock type
void *user;
// raw or llarp subsystem (is this used? does this matter?)
2018-07-22 03:34:28 +00:00
bool llarp;
2018-07-16 12:48:04 +00:00
/// request id
unsigned int id;
2018-07-22 03:34:28 +00:00
/// question being asked
dns_msg_question question;
// request source socket
2018-09-29 10:22:00 +00:00
struct sockaddr *from; // FIXME: convert to llarp::Addr
sendto_dns_hook_func sendto_hook; // sendto hook tbh
2018-07-22 03:34:28 +00:00
// maybe a reference to dnsd_context incase of multiple
dnsd_context *context; // or you can access it via user (udp)
2018-07-16 12:48:04 +00:00
};
2018-08-08 12:40:54 +00:00
// FIXME: made better as a two way structure, collapse the request and response
// together
struct dnsd_query_hook_response
{
/// turn off communication
bool dontSendResponse;
/// turn off recursion
bool dontLookUp;
/// potential address
2018-11-03 13:19:18 +00:00
llarp::huint32_t *returnThis;
};
/// builds and fires a request based based on llarp_udp_io udp event
/// called by the llarp_handle_dns_recvfrom generic (dnsd/dnsc) handler in dns
void
llarp_handle_dnsd_recvfrom(struct llarp_udp_io *udp,
const struct sockaddr *addr, const void *buf,
ssize_t sz);
//
// output structures/functions:
//
// we may want to pass dnsd_question_request to these,
// incase we need to send an error back up through the pipeline
/// NXDOMAIN not found
void
write404_dnss_response(const struct sockaddr *from,
dnsd_question_request *request);
/// for hook functions to use
void
writecname_dnss_response(std::string cname, const struct sockaddr *from,
dnsd_question_request *request);
// FIXME: llarp::Addr
/// send an A record found response
void
2018-11-03 13:19:18 +00:00
writesend_dnss_response(llarp::huint32_t *hostRes, const struct sockaddr *from,
dnsd_question_request *request);
// FIXME: llarp::Addr
/// send an PTR record found response
void
writesend_dnss_revresponse(std::string reverse, const struct sockaddr *from,
dnsd_question_request *request);
// FIXME: llarp::Addr
//
// setup/teardown functions/structure:
//
2018-08-01 09:04:40 +00:00
/// intercept query hook functor
2018-08-08 12:40:54 +00:00
typedef dnsd_query_hook_response *(*intercept_query_hook)(
std::string name, const struct sockaddr *from,
struct dnsd_question_request *request);
2018-09-29 10:22:00 +00:00
// FIXME: llarp::Addr
2018-07-22 03:34:28 +00:00
2018-08-01 09:04:40 +00:00
/// DNS Server context
2018-07-22 03:34:28 +00:00
struct dnsd_context
{
/// DNS daemon socket to listen on
struct llarp_udp_io udp;
2018-09-22 10:22:18 +00:00
/// udp tracker
2018-08-01 09:04:40 +00:00
struct dns_tracker *tracker;
/// upstream DNS client context to use
2018-07-22 03:34:28 +00:00
dnsc_context client;
2018-09-22 10:22:18 +00:00
/// custom data for intercept query hook (used for configuration of hook)
2018-07-22 03:34:28 +00:00
void *user;
/// hook function for intercepting dns requests
intercept_query_hook intercept;
};
/// initialize dns subsystem and bind socket
/// returns true on bind success otherwise returns false
bool
2018-10-03 10:59:49 +00:00
llarp_dnsd_init(struct dnsd_context *const dnsd,
struct llarp_logic *const logic,
struct llarp_ev_loop *const netloop,
const llarp::Addr &dnsd_sockaddr,
2018-09-29 10:22:00 +00:00
const llarp::Addr &dnsc_sockaddr);
2018-07-22 03:34:28 +00:00
2018-08-01 09:04:40 +00:00
/// shutdowns any events, and deallocates for this context
2018-07-22 03:34:28 +00:00
bool
2018-10-03 10:59:49 +00:00
llarp_dnsd_stop(struct dnsd_context *const dnsd);
2018-07-22 03:34:28 +00:00
2018-07-16 12:48:04 +00:00
#endif