mirror of
https://github.com/DCC-EX/CommandStation-EX.git
synced 2024-12-24 13:21:23 +01:00
137 lines
4.8 KiB
C++
137 lines
4.8 KiB
C++
#include "WifiInterface.h"
|
|
#include "Config.h"
|
|
#include "DIAG.h"
|
|
#include "StringFormatter.h"
|
|
|
|
|
|
const char PROGMEM READY_SEARCH[] ="\r\nready\r\n";
|
|
const char PROGMEM OK_SEARCH[] ="\r\nOK\r\n";
|
|
const char PROGMEM END_DETAIL_SEARCH[] ="@ 1000";
|
|
const char PROGMEM PROMPT_SEARCH[] =">";
|
|
|
|
bool WifiInterface::connected=false;
|
|
DCCEXParser WifiInterface::parser;
|
|
byte WifiInterface::loopstate=0;
|
|
int WifiInterface::datalength=0;
|
|
int WifiInterface::connectionId;
|
|
byte WifiInterface::buffer[MAX_WIFI_BUFFER];
|
|
MemStream WifiInterface::streamer(buffer,sizeof(buffer));
|
|
|
|
void WifiInterface::setup() {
|
|
DIAG(F("\n++++++ Wifi Setup In Progress ++++++++\n"));
|
|
connected=setup2();
|
|
// TODO calloc the buffer and streamer and parser etc
|
|
DIAG(F("\n++++++ Wifi Setup %S ++++++++\n"), connected?F("OK"):F("FAILED"));
|
|
}
|
|
|
|
bool WifiInterface::setup2()
|
|
{
|
|
Serial1.begin(WIFI_BAUD_RATE); // initialize serial for ESP module
|
|
|
|
delay(1000);
|
|
|
|
StringFormatter::send(Serial1,F("AT+RST\r\n")); // reset module
|
|
checkForOK(5000,END_DETAIL_SEARCH,true); // Show startup but ignore unreadable upto ready
|
|
if (!checkForOK(5000,READY_SEARCH,false)) return false;
|
|
|
|
StringFormatter::send(Serial1,F("AT+CWMODE=1\r\n")); // configure as access point
|
|
if (!checkForOK(10000,OK_SEARCH,true)) return false;
|
|
|
|
StringFormatter::send(Serial1,F("AT+CWJAP=\"%S\",\"%S\"\r\n"),WIFI_SSID,WIFI_PASS);
|
|
if (!checkForOK(20000,OK_SEARCH,true)) return false;
|
|
|
|
StringFormatter::send(Serial1,F("AT+CIFSR\r\n")); // get ip address //192.168.4.1
|
|
if (!checkForOK(10000,OK_SEARCH,true)) return false;
|
|
|
|
StringFormatter::send(Serial1,F("AT+CIPMUX=1\r\n")); // configure for multiple connections
|
|
if (!checkForOK(10000,OK_SEARCH,true)) return false;
|
|
|
|
StringFormatter::send(Serial1,F("AT+CIPSERVER=1,%d\r\n"),WIFI_PORT); // turn on server on port 80
|
|
if (!checkForOK(10000,OK_SEARCH,true)) return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool WifiInterface::checkForOK( const int timeout, const char * waitfor, bool echo) {
|
|
long int time = millis()+timeout;
|
|
char *locator=waitfor;
|
|
DIAG(F("\nWifi setup Check: %S\n"),waitfor);
|
|
while( time > millis()) {
|
|
while(Serial1.available()) {
|
|
int ch=Serial1.read();
|
|
if (echo) Serial.write(ch);
|
|
if (ch!=pgm_read_byte_near(locator)) locator=waitfor;
|
|
if (ch==pgm_read_byte_near(locator)) {
|
|
locator++;
|
|
if (!pgm_read_byte_near(locator)) {
|
|
DIAG(F("\nOK after %dms\n"),millis()-time+timeout);
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
DIAG(F("\nTIMEOUT after %dms\n"),timeout);
|
|
return false;
|
|
}
|
|
|
|
|
|
void WifiInterface::loop() {
|
|
if (!connected) return;
|
|
|
|
// read anything into a buffer, collecting info on the way
|
|
while (loopstate!=99 && Serial1.available()) {
|
|
int ch=Serial1.read();
|
|
Serial.write(ch);
|
|
switch (loopstate) {
|
|
case 0: // looking for +
|
|
connectionId=0;
|
|
streamer.flush();
|
|
if (ch=='+') loopstate=1;
|
|
break;
|
|
case 1: // Looking for I
|
|
loopstate= (ch=='I')?2:0;
|
|
break;
|
|
case 2: // Looking for P
|
|
loopstate= (ch=='P')?3:0;
|
|
break;
|
|
case 3: // Looking for D
|
|
loopstate= (ch=='D')?4:0;
|
|
break;
|
|
case 4: // Looking for ,
|
|
loopstate= (ch==',')?5:0;
|
|
break;
|
|
case 5: // reading connection id
|
|
if (ch==',') loopstate=6;
|
|
else connectionId=10*connectionId+(ch-'0');
|
|
break;
|
|
case 6: // reading for length
|
|
if (ch==':') loopstate=(datalength==0)?99:7; // 99 is getout without reading next char
|
|
else datalength=datalength*10 + (ch-'0');
|
|
break;
|
|
case 7: // reading data
|
|
streamer.write(ch);
|
|
datalength--;
|
|
if (datalength==0) loopstate=99;
|
|
break;
|
|
} // switch
|
|
} // while
|
|
if (loopstate!=99) return;
|
|
// TODO remove > in data
|
|
streamer.write('\0');
|
|
|
|
DIAG(F("\nWifiRead:%d:%s\n"),connectionId,buffer);
|
|
streamer.setBufferContentPosition(0,0); // reset write position to start of buffer
|
|
// SIDE EFFECT WARNING:::
|
|
// We know that parser will read the entire buffer before starting to write to it.
|
|
// Otherwise we would have to copy the buffer elsewhere and RAM is in short supply.
|
|
|
|
// TODO ... tell parser that callbacks are diallowed because we dont want to handle the async
|
|
parser.parse(streamer,buffer);
|
|
if (streamer.available()) { // there is a reply to send
|
|
StringFormatter::send(Serial1,F("AT+CIPSEND=%d,%d\r\n"),connectionId,streamer.available());
|
|
streamer.write('\0');
|
|
if (checkForOK(1000,PROMPT_SEARCH,true)) Serial1.print((char *) buffer);
|
|
}
|
|
loopstate=0; // go back to looking for +IPD
|
|
}
|