user
4 years ago
7 changed files with 336 additions and 57 deletions
|
Before Width: | Height: | Size: 6.0 KiB |
@ -0,0 +1,166 @@ |
|||||
|
#include "./DNSServer.h" |
||||
|
#include <lwip/def.h> |
||||
|
#include <Arduino.h> |
||||
|
|
||||
|
#define DEBUG |
||||
|
#define DEBUG_OUTPUT Serial |
||||
|
|
||||
|
DNSServer::DNSServer() |
||||
|
{ |
||||
|
_ttl = htonl(60); |
||||
|
_errorReplyCode = DNSReplyCode::NonExistentDomain; |
||||
|
} |
||||
|
|
||||
|
bool DNSServer::start(const uint16_t &port, const String &domainName, |
||||
|
const IPAddress &resolvedIP) |
||||
|
{ |
||||
|
_port = port; |
||||
|
_domainName = domainName; |
||||
|
_resolvedIP[0] = resolvedIP[0]; |
||||
|
_resolvedIP[1] = resolvedIP[1]; |
||||
|
_resolvedIP[2] = resolvedIP[2]; |
||||
|
_resolvedIP[3] = resolvedIP[3]; |
||||
|
downcaseAndRemoveWwwPrefix(_domainName); |
||||
|
return _udp.begin(_port) == 1; |
||||
|
} |
||||
|
|
||||
|
void DNSServer::setErrorReplyCode(const DNSReplyCode &replyCode) |
||||
|
{ |
||||
|
_errorReplyCode = replyCode; |
||||
|
} |
||||
|
|
||||
|
void DNSServer::setTTL(const uint32_t &ttl) |
||||
|
{ |
||||
|
_ttl = htonl(ttl); |
||||
|
} |
||||
|
|
||||
|
void DNSServer::stop() |
||||
|
{ |
||||
|
_udp.stop(); |
||||
|
} |
||||
|
|
||||
|
void DNSServer::downcaseAndRemoveWwwPrefix(String &domainName) |
||||
|
{ |
||||
|
domainName.toLowerCase(); |
||||
|
domainName.replace("www.", ""); |
||||
|
domainName.replace("https://", ""); |
||||
|
} |
||||
|
|
||||
|
void DNSServer::processNextRequest() |
||||
|
{ |
||||
|
_currentPacketSize = _udp.parsePacket(); |
||||
|
if (_currentPacketSize) |
||||
|
{ |
||||
|
_buffer = (unsigned char*)malloc(_currentPacketSize * sizeof(char)); |
||||
|
_udp.read(_buffer, _currentPacketSize); |
||||
|
_dnsHeader = (DNSHeader*) _buffer; |
||||
|
|
||||
|
if (_dnsHeader->QR == DNS_QR_QUERY && |
||||
|
_dnsHeader->OPCode == DNS_OPCODE_QUERY && |
||||
|
requestIncludesOnlyOneQuestion() && |
||||
|
(_domainName == "*" || getDomainNameWithoutWwwPrefix() == _domainName) |
||||
|
) |
||||
|
|
||||
|
{ |
||||
|
replyWithIP(); |
||||
|
} |
||||
|
else if (_dnsHeader->QR == DNS_QR_QUERY) |
||||
|
{ |
||||
|
replyWithCustomCode(); |
||||
|
} |
||||
|
|
||||
|
free(_buffer); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
bool DNSServer::requestIncludesOnlyOneQuestion() |
||||
|
{ |
||||
|
return ntohs(_dnsHeader->QDCount) == 1 && |
||||
|
_dnsHeader->ANCount == 0 && |
||||
|
_dnsHeader->NSCount == 0 && |
||||
|
_dnsHeader->ARCount == 0; |
||||
|
} |
||||
|
|
||||
|
String DNSServer::getDomainNameWithoutWwwPrefix() |
||||
|
{ |
||||
|
String parsedDomainName = ""; |
||||
|
unsigned char *start = _buffer + 12; |
||||
|
if (*start == 0) |
||||
|
{ |
||||
|
return parsedDomainName; |
||||
|
} |
||||
|
int pos = 0; |
||||
|
while(true) |
||||
|
{ |
||||
|
unsigned char labelLength = *(start + pos); |
||||
|
for(int i = 0; i < labelLength; i++) |
||||
|
{ |
||||
|
pos++; |
||||
|
parsedDomainName += (char)*(start + pos); |
||||
|
} |
||||
|
pos++; |
||||
|
if (*(start + pos) == 0) |
||||
|
{ |
||||
|
downcaseAndRemoveWwwPrefix(parsedDomainName); |
||||
|
return parsedDomainName; |
||||
|
} |
||||
|
else |
||||
|
{ |
||||
|
parsedDomainName += "."; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
void DNSServer::replyWithIP() |
||||
|
{ |
||||
|
_dnsHeader->QR = DNS_QR_RESPONSE; |
||||
|
_dnsHeader->ANCount = _dnsHeader->QDCount; |
||||
|
_dnsHeader->QDCount = _dnsHeader->QDCount; |
||||
|
//_dnsHeader->RA = 1;
|
||||
|
|
||||
|
_udp.beginPacket(_udp.remoteIP(), _udp.remotePort()); |
||||
|
_udp.write(_buffer, _currentPacketSize); |
||||
|
|
||||
|
_udp.write((uint8_t)192); // answer name is a pointer
|
||||
|
_udp.write((uint8_t)12); // pointer to offset at 0x00c
|
||||
|
|
||||
|
_udp.write((uint8_t)0); // 0x0001 answer is type A query (host address)
|
||||
|
_udp.write((uint8_t)1); |
||||
|
|
||||
|
_udp.write((uint8_t)0); //0x0001 answer is class IN (internet address)
|
||||
|
_udp.write((uint8_t)1); |
||||
|
|
||||
|
_udp.write((unsigned char*)&_ttl, 4); |
||||
|
|
||||
|
// Length of RData is 4 bytes (because, in this case, RData is IPv4)
|
||||
|
_udp.write((uint8_t)0); |
||||
|
_udp.write((uint8_t)4); |
||||
|
_udp.write(_resolvedIP, sizeof(_resolvedIP)); |
||||
|
_udp.endPacket(); |
||||
|
|
||||
|
|
||||
|
|
||||
|
#ifdef DEBUG |
||||
|
DEBUG_OUTPUT.print("DNS responds: "); |
||||
|
DEBUG_OUTPUT.print(_resolvedIP[0]); |
||||
|
DEBUG_OUTPUT.print("."); |
||||
|
DEBUG_OUTPUT.print(_resolvedIP[1]); |
||||
|
DEBUG_OUTPUT.print("."); |
||||
|
DEBUG_OUTPUT.print(_resolvedIP[2]); |
||||
|
DEBUG_OUTPUT.print("."); |
||||
|
DEBUG_OUTPUT.print(_resolvedIP[3]); |
||||
|
DEBUG_OUTPUT.print(" for "); |
||||
|
DEBUG_OUTPUT.println(getDomainNameWithoutWwwPrefix()); |
||||
|
#endif |
||||
|
} |
||||
|
|
||||
|
void DNSServer::replyWithCustomCode() |
||||
|
{ |
||||
|
_dnsHeader->QR = DNS_QR_RESPONSE; |
||||
|
_dnsHeader->RCode = (unsigned char)_errorReplyCode; |
||||
|
_dnsHeader->QDCount = 0; |
||||
|
|
||||
|
_udp.beginPacket(_udp.remoteIP(), _udp.remotePort()); |
||||
|
_udp.write(_buffer, sizeof(DNSHeader)); |
||||
|
_udp.endPacket(); |
||||
|
} |
||||
@ -0,0 +1,72 @@ |
|||||
|
#ifndef DNSServer_h |
||||
|
#define DNSServer_h |
||||
|
#include <WiFiUdp.h> |
||||
|
|
||||
|
#define DNS_QR_QUERY 0 |
||||
|
#define DNS_QR_RESPONSE 1 |
||||
|
#define DNS_OPCODE_QUERY 0 |
||||
|
|
||||
|
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(); |
||||
|
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]; |
||||
|
int _currentPacketSize; |
||||
|
unsigned char* _buffer; |
||||
|
DNSHeader* _dnsHeader; |
||||
|
uint32_t _ttl; |
||||
|
DNSReplyCode _errorReplyCode; |
||||
|
|
||||
|
void downcaseAndRemoveWwwPrefix(String &domainName); |
||||
|
String getDomainNameWithoutWwwPrefix(); |
||||
|
bool requestIncludesOnlyOneQuestion(); |
||||
|
void replyWithIP(); |
||||
|
void replyWithCustomCode(); |
||||
|
}; |
||||
|
#endif |
||||
|
|
||||
Binary file not shown.
Binary file not shown.
@ -0,0 +1,98 @@ |
|||||
|
// Captive portal with (arduino) OTA + SPIFFS
|
||||
|
|
||||
|
#include <Arduino.h> |
||||
|
#include <ESP8266WiFi.h> |
||||
|
#include <ESP8266mDNS.h> |
||||
|
#include <WiFiUdp.h> |
||||
|
#include <ArduinoOTA.h> // Over-the-Air updates |
||||
|
#include <ESP8266WebServer.h> |
||||
|
#include "./DNSServer.h" // Dns server |
||||
|
#include <FS.h> // SPIFFS |
||||
|
|
||||
|
DNSServer dnsServer; |
||||
|
const byte DNS_PORT = 53; |
||||
|
|
||||
|
ESP8266WebServer server(80); |
||||
|
|
||||
|
#ifndef STASSID |
||||
|
#define STASSID "Paracity Gallery v041" |
||||
|
//#define STASPSK "mypassword"
|
||||
|
#endif |
||||
|
|
||||
|
IPAddress apIP(192, 168, 4, 1); |
||||
|
const char* ssid = STASSID; |
||||
|
//const char* password = STAPSK;
|
||||
|
|
||||
|
void setup() { |
||||
|
Serial.begin(115200); |
||||
|
Serial.println("Booting"); |
||||
|
|
||||
|
WiFi.mode(WIFI_AP); |
||||
|
WiFi.softAPConfig(apIP, apIP, IPAddress(255, 255, 255, 0)); |
||||
|
WiFi.softAP(ssid); |
||||
|
dnsServer.start(DNS_PORT, "*", apIP); // redirect dns request to AP ip
|
||||
|
|
||||
|
MDNS.begin("opencoil", WiFi.softAPIP()); |
||||
|
Serial.println("Ready"); |
||||
|
Serial.print("IP address: "); |
||||
|
Serial.println(WiFi.softAPIP()); |
||||
|
|
||||
|
//Over-the-Air updates
|
||||
|
ArduinoOTA.setHostname("opencoil"); |
||||
|
//ArduinoOTA.setPassword("change-me"); //disabled to allow data uploads
|
||||
|
ArduinoOTA.begin(); |
||||
|
SPIFFS.begin(); |
||||
|
|
||||
|
//redirect all traffic to index.html
|
||||
|
server.onNotFound([]() { |
||||
|
if(!handleFileRead(server.uri())){ |
||||
|
const char *metaRefreshStr = "<head><meta http-equiv=\"refresh\" content=\"0; url=http://192.168.4.1/index.html\" /></head><body><p>redirecting...</p></body>"; |
||||
|
server.send(200, "text/html", metaRefreshStr); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
server.begin(); |
||||
|
|
||||
|
} |
||||
|
|
||||
|
void loop() { |
||||
|
dnsServer.processNextRequest(); |
||||
|
ArduinoOTA.handle(); |
||||
|
server.handleClient(); |
||||
|
delay(50); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
String getContentType(String filename){ |
||||
|
if(server.hasArg("download")) return "application/octet-stream"; |
||||
|
else if(filename.endsWith(".htm")) return "text/html"; |
||||
|
else if(filename.endsWith(".html")) return "text/html"; |
||||
|
else if(filename.endsWith(".css")) return "text/css"; |
||||
|
else if(filename.endsWith(".js")) return "application/javascript"; |
||||
|
else if(filename.endsWith(".png")) return "image/png"; |
||||
|
else if(filename.endsWith(".gif")) return "image/gif"; |
||||
|
else if(filename.endsWith(".jpg")) return "image/jpeg"; |
||||
|
else if(filename.endsWith(".ico")) return "image/x-icon"; |
||||
|
else if(filename.endsWith(".xml")) return "text/xml"; |
||||
|
else if(filename.endsWith(".mp4")) return "video/mp4"; |
||||
|
else if(filename.endsWith(".pdf")) return "application/x-pdf"; |
||||
|
else if(filename.endsWith(".zip")) return "application/x-zip"; |
||||
|
else if(filename.endsWith(".gz")) return "application/x-gzip"; |
||||
|
return "text/plain"; |
||||
|
} |
||||
|
|
||||
|
//Given a file path, look for it in the SPIFFS file storage. Returns true if found, returns false if not found.
|
||||
|
bool handleFileRead(String path){ |
||||
|
if(path.endsWith("/")) path += "index.html"; |
||||
|
String contentType = getContentType(path); |
||||
|
String pathWithGz = path + ".gz"; |
||||
|
if(SPIFFS.exists(pathWithGz) || SPIFFS.exists(path)){ |
||||
|
if(SPIFFS.exists(pathWithGz)) |
||||
|
path += ".gz"; |
||||
|
File file = SPIFFS.open(path, "r"); |
||||
|
size_t sent = server.streamFile(file, contentType); |
||||
|
file.close(); |
||||
|
return true; |
||||
|
} |
||||
|
return false; |
||||
|
} |
||||
Loading…
Reference in new issue