mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-07 15:20:31 +00:00
122 lines
2.7 KiB
C++
122 lines
2.7 KiB
C++
#ifndef LIBLLARP_DNS_HPP
|
|
#define LIBLLARP_DNS_HPP
|
|
|
|
#include <llarp/dns.h>
|
|
#include <sys/types.h> // for uint & ssize_t
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#ifndef _WIN32
|
|
#include <sys/socket.h>
|
|
#endif
|
|
|
|
#include <llarp/net.hpp> // for llarp::Addr
|
|
|
|
struct dnsd_context;
|
|
|
|
// 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
|
|
struct dns_tracker
|
|
{
|
|
// uint c_responses;
|
|
uint c_requests;
|
|
// request has to be a pointer
|
|
std::unordered_map< uint, std::unique_ptr< dnsc_answer_request > >
|
|
client_request;
|
|
// FIXME: support multiple dns server contexts
|
|
dnsd_context *dnsd;
|
|
// rn we need 1 tracker per DNSd and each DNSd needs it's own IP
|
|
// actually we can bind once and use the tracker to sort
|
|
// but no way to tell what DNSd they want...
|
|
// std::map< llarp::Addr, std::unique_ptr< dnsc_answer_request > > dnsds;
|
|
// std::map< uint, dnsd_question_request * > daemon_request;
|
|
};
|
|
|
|
// 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;
|
|
|
|
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;
|
|
};
|
|
|
|
std::string
|
|
getDNSstring(const char *buffer);
|
|
|
|
void
|
|
code_domain(char *&buffer, const std::string &domain) throw();
|
|
|
|
extern "C"
|
|
{
|
|
uint16_t
|
|
get16bits(const char *&buffer) throw();
|
|
|
|
uint32_t
|
|
get32bits(const char *&buffer) throw();
|
|
|
|
dns_msg_header *
|
|
decode_hdr(const char *buffer);
|
|
|
|
dns_msg_question *
|
|
decode_question(const char *buffer);
|
|
|
|
dns_msg_answer *
|
|
decode_answer(const char *buffer);
|
|
|
|
void
|
|
put16bits(char *&buffer, uint16_t value) throw();
|
|
|
|
void
|
|
put32bits(char *&buffer, uint32_t value) throw();
|
|
|
|
void
|
|
llarp_handle_dns_recvfrom(struct llarp_udp_io *udp,
|
|
const struct sockaddr *addr, const void *buf,
|
|
ssize_t sz);
|
|
}
|
|
#endif
|