lokinet/include/llarp/dns.hpp

117 lines
2.5 KiB
C++
Raw Normal View History

2018-07-22 03:34:28 +00:00
#ifndef LIBLLARP_DNS_HPP
#define LIBLLARP_DNS_HPP
2018-07-27 04:07:22 +00:00
#include <llarp/dns.h>
2018-07-22 03:34:28 +00:00
#include <sys/types.h> // for uint & ssize_t
2018-07-27 04:07:22 +00:00
#include <map>
#include <string>
#include <vector>
2018-07-22 03:34:28 +00:00
2018-09-25 08:31:29 +00:00
#ifndef _WIN32
2018-09-20 10:07:16 +00:00
#include <sys/socket.h>
2018-09-25 08:31:29 +00:00
#endif
2018-09-20 10:07:16 +00:00
#include <llarp/net.hpp> // for llarp::Addr
2018-10-01 09:59:50 +00:00
struct dnsd_context;
2018-08-01 09:04:40 +00:00
// dnsc can work over any UDP socket
// however we can't ignore udp->user
// we need to be able to reference the request (being a request or response)
// bottom line is we can't use udp->user
// so we'll need to track all incoming and outgoing requests
2018-07-27 04:07:22 +00:00
struct dns_tracker
{
// uint c_responses;
uint c_requests;
2018-08-01 09:04:40 +00:00
// request has to be a pointer
std::unordered_map< uint, std::unique_ptr< dnsc_answer_request > >
client_request;
2018-07-27 04:07:22 +00:00
// FIXME: support multiple dns server contexts
dnsd_context *dnsd;
// rn we need 1 tracker per DNSd and each DNSd needs it's own IP
// std::map< llarp::Addr, std::unique_ptr< dnsc_answer_request > > dnsds;
2018-07-27 04:07:22 +00:00
// std::map< uint, dnsd_question_request * > daemon_request;
};
2018-07-22 03:34:28 +00:00
// protocol parsing/writing structures & functions
struct dns_msg_header
{
uint16_t id;
uint8_t qr : 1;
uint8_t opcode : 4;
uint8_t aa : 1;
uint8_t tc : 1;
uint8_t rd : 1;
uint8_t ra : 1;
uint8_t z : 1;
uint8_t ad : 1;
uint8_t cd : 1;
uint8_t rcode : 4;
2018-07-22 03:34:28 +00:00
uint16_t qdCount;
uint16_t anCount;
uint16_t nsCount;
uint16_t arCount;
};
struct dns_msg_question
{
std::string name;
uint16_t type;
uint16_t qClass;
};
struct dns_msg_answer
{
std::string name;
uint16_t type;
uint16_t aClass;
uint32_t ttl;
uint16_t rdLen;
uint8_t *rData;
};
struct dns_packet
{
struct dns_msg_header *header;
std::vector< std::unique_ptr< dns_msg_question > > questions;
std::vector< std::unique_ptr< dns_msg_answer > > answers;
std::vector< std::unique_ptr< dns_msg_answer > > auth_rrs;
std::vector< std::unique_ptr< dns_msg_answer > > additional_rrs;
};
2018-07-27 04:07:22 +00:00
extern "C"
{
uint16_t
get16bits(const char *&buffer) throw();
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
uint32_t
get32bits(const char *&buffer) throw();
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
dns_msg_header *
decode_hdr(const char *buffer);
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
dns_msg_question *
decode_question(const char *buffer);
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
dns_msg_answer *
decode_answer(const char *buffer);
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
void
put16bits(char *&buffer, uint16_t value) throw();
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
void
put32bits(char *&buffer, uint32_t value) throw();
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
void
code_domain(char *&buffer, const std::string &domain) throw();
2018-07-22 03:34:28 +00:00
2018-07-27 04:07:22 +00:00
void
llarp_handle_dns_recvfrom(struct llarp_udp_io *udp,
2018-08-24 00:39:01 +00:00
const struct sockaddr *addr, const void *buf,
2018-07-27 04:07:22 +00:00
ssize_t sz);
}
2018-07-22 03:34:28 +00:00
#endif