mirror of
https://github.com/esp8266/Arduino.git
synced 2025-04-29 12:16:48 +03:00
88 lines
2.2 KiB
C++
88 lines
2.2 KiB
C++
#ifndef DNSServer_h
|
|
#define DNSServer_h
|
|
#include <WiFiUdp.h>
|
|
|
|
#define DNS_QR_QUERY 0
|
|
#define DNS_QR_RESPONSE 1
|
|
#define DNS_OPCODE_QUERY 0
|
|
|
|
#define DNS_QCLASS_IN 1
|
|
#define DNS_QCLASS_ANY 255
|
|
|
|
#define DNS_QTYPE_A 1
|
|
#define DNS_QTYPE_ANY 255
|
|
|
|
#define MAX_DNSNAME_LENGTH 253
|
|
#define MAX_DNS_PACKETSIZE 512
|
|
|
|
enum class DNSReplyCode
|
|
{
|
|
NoError = 0,
|
|
FormError = 1,
|
|
ServerFailure = 2,
|
|
NonExistentDomain = 3,
|
|
NotImplemented = 4,
|
|
Refused = 5,
|
|
YXDomain = 6,
|
|
YXRRSet = 7,
|
|
NXRRSet = 8
|
|
};
|
|
|
|
struct DNSHeader
|
|
{
|
|
uint16_t ID; // identification number
|
|
unsigned char RD : 1; // recursion desired
|
|
unsigned char TC : 1; // truncated message
|
|
unsigned char AA : 1; // authoritive answer
|
|
unsigned char OPCode : 4; // message_type
|
|
unsigned char QR : 1; // query/response flag
|
|
unsigned char RCode : 4; // response code
|
|
unsigned char Z : 3; // its z! reserved
|
|
unsigned char RA : 1; // recursion available
|
|
uint16_t QDCount; // number of question entries
|
|
uint16_t ANCount; // number of answer entries
|
|
uint16_t NSCount; // number of authority entries
|
|
uint16_t ARCount; // number of resource entries
|
|
};
|
|
|
|
class DNSServer
|
|
{
|
|
public:
|
|
DNSServer();
|
|
~DNSServer() {
|
|
stop();
|
|
};
|
|
void processNextRequest();
|
|
void setErrorReplyCode(const DNSReplyCode &replyCode);
|
|
void setTTL(const uint32_t &ttl);
|
|
|
|
// Returns true if successful, false if there are no sockets available
|
|
bool start(const uint16_t &port,
|
|
const String &domainName,
|
|
const IPAddress &resolvedIP);
|
|
// stops the DNS server
|
|
void stop();
|
|
|
|
private:
|
|
WiFiUDP _udp;
|
|
uint16_t _port;
|
|
String _domainName;
|
|
unsigned char _resolvedIP[4];
|
|
uint32_t _ttl;
|
|
DNSReplyCode _errorReplyCode;
|
|
|
|
void downcaseAndRemoveWwwPrefix(String &domainName);
|
|
void replyWithIP(DNSHeader *dnsHeader,
|
|
unsigned char * query,
|
|
size_t queryLength);
|
|
void replyWithError(DNSHeader *dnsHeader,
|
|
DNSReplyCode rcode,
|
|
unsigned char *query,
|
|
size_t queryLength);
|
|
void replyWithError(DNSHeader *dnsHeader,
|
|
DNSReplyCode rcode);
|
|
void respondToRequest(uint8_t *buffer, size_t length);
|
|
void writeNBOShort(uint16_t value);
|
|
};
|
|
#endif
|