mirror of
https://github.com/esp8266/Arduino.git
synced 2025-07-27 18:02:17 +03:00
656bf146bc
Clean up the DNSServer class. Removed member variables that are not required outside a member call lifetime, and add destructor/checks. Fixes #5179
73 lines
2.0 KiB
C++
73 lines
2.0 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 MAX_DNSNAME_LENGTH 253
|
|
|
|
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);
|
|
String getDomainNameWithoutWwwPrefix(const uint8_t* buffer, size_t packetSize);
|
|
bool requestIncludesOnlyOneQuestion(const DNSHeader* dnsHeader);
|
|
void replyWithIP(uint8_t* buffer, size_t packetSize);
|
|
void replyWithCustomCode(uint8_t* buffer, size_t packetSize);
|
|
};
|
|
#endif |