Skip to content
This repository was archived by the owner on Mar 17, 2025. It is now read-only.

firebase: telnet console proof of concept #13

Closed
wants to merge 2 commits into from
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
196 changes: 144 additions & 52 deletions firebase/firebase.ino
Original file line number Diff line number Diff line change
Expand Up @@ -26,118 +26,210 @@ struct Credential {
char ssid[32];
char password[32];
} cred;

const int maxWifiConnectAttemps = 20;
WiFiServer telnetServer(23);
WiFiClient telnetClient;
String buffer;

struct MultiSerial {
MultiSerial(HardwareSerial& serial, WiFiClient& telnetClient) : _serial(serial), _telnetClient(telnetClient) {
_current = &serial;
}
HardwareSerial& _serial;
WiFiClient& _telnetClient;
Stream *_current;

int available() {
int serialAvailable = _serial.available();
if (serialAvailable > 0) {
_current = &_serial;
return serialAvailable;
}
if (_telnetClient.connected()) {
int telnetAvailable = _telnetClient.available();
if (telnetAvailable > 0) {
_current = &_telnetClient;
return telnetAvailable;
}
}
return 0;
}
String readStringUntil(char c) {
return _current->readStringUntil(c);
}
void println() {
_current->println();
}
void print(const char* str) {
_current->print(str);
}
template<typename T>
void println(T arg) {
_current->println(arg);
}
} WSerial{Serial, telnetClient};



void setup() {
Serial.begin(115200);

EEPROM.begin(4096);
EEPROM.get(0, cred);
Serial.println();
Serial.println("!I #!/dev/fail");
Serial.print("!I token: ");
Serial.println(cred.token);
WSerial.println();
WSerial.println("!I #!/dev/fail");
WSerial.print("!I token: ");
WSerial.println(cred.token);
bool connected = false;
if (strlen(cred.ssid) > 0) {
wifiConnect();
connected = wifiConnect();
}
if (!connected) {
WSerial.println("!E no wifi connection");
}
softAP("FAIL");
}

bool wifiConnect() {
Serial.print("!I connecting to wifi");
WSerial.println("!I connecting to wifi");
WSerial.print("!I ssid: ");
WSerial.println(cred.ssid);
WSerial.print("!I ");
WiFi.begin(cred.ssid, cred.password);
for (int i = 0; i < 20; i++) {
for (int i = 0; i < maxWifiConnectAttemps; i++) {
delay(500);
Serial.print(".");
WSerial.print(".");
if (WiFi.status() == WL_CONNECTED) {
Serial.println("");
Serial.print("!I wifi connected: ");
Serial.println(WiFi.localIP());
WSerial.println("");
WSerial.println("!I wifi connected");
WSerial.print("!I ip: ");
WSerial.println(WiFi.localIP());
return true;
}
}
Serial.println("");
Serial.println("!I wifi connection failed");
WSerial.println("");
WSerial.println("!E wifi connection failed");
return false;
}

void softAP(String ssid) {
WSerial.println("!I starting soft AP");
WSerial.print("!I ssid: ");
WSerial.println(ssid.c_str());
WiFi.softAP(ssid.c_str());
WSerial.print("!I ip: ");
WSerial.println(WiFi.softAPIP());
telnetServer.begin();
telnetServer.setNoDelay(true);
WSerial.println("!I telnet serial started");
WSerial.println("!I port: 23");
}

String buildRequest(String data) {
return data + "?auth=" + cred.token + " HTTP/1.1\r\n" +
"Host: " + host + "\r\n" +
"User-Agent: Chiplet\r\n" +
"Connection: close\r\n";
}

String unquote(String s) {
if (s[0] == '"' && s[s.length()-1] == '"') {
return s.substring(1, s.length()-1);
}
return s;
}

void loop() {
String request;

if (Serial.available() > 0) {
String data = Serial.readStringUntil('\n');
data.trim();

if (telnetServer.hasClient()) {
if (telnetClient.connected()) {
telnetClient.stop();
}
telnetClient = telnetServer.available();
}
String request;
if (WSerial.available() > 0) {
String data = WSerial.readStringUntil('\n');
data.trim();
if (data.startsWith("GET")) {
Serial.println(String("!I request: ") + data);
WSerial.print("!I request: ");
WSerial.println(data);
request = buildRequest(data) + "\r\n";
} else if (data.startsWith("POST")) {
Serial.println(String("!I request: ") + data);
Serial.println("!I waiting for payload");
WSerial.print("!I request: ");
WSerial.println(data);
WSerial.println("!I waiting for payload");
buffer = buildRequest(data);
} else if (buffer.startsWith("POST")) {
// send POST request when complete
Serial.println(String("!I payload: ") + data);
Serial.println(String("!I content-length: ") + data.length());
WSerial.print("!I payload: ");
WSerial.println(data);
WSerial.print("!I content-length: ");
WSerial.println(data.length());
request = buffer + "Content-Length: " + data.length() + "\r\n\r\n" + data;
buffer = "";
} else if (data.startsWith("TOKEN")) {
data.substring(6).toCharArray(cred.token, sizeof(cred.token));
EEPROM.put(0, cred);
EEPROM.commit();
Serial.println(String("!I token updated: ") + cred.token);
WSerial.print("!I token updated: ");
WSerial.println(cred.token);
} else if (data.startsWith("AT+CWJAP=")) {
String ssidPwd = data.substring(9);
int i = ssidPwd.indexOf(",");
ssidPwd.substring(0, i).toCharArray(cred.ssid, sizeof(cred.token));
ssidPwd.substring(i+1).toCharArray(cred.password, sizeof(cred.password));
Serial.println("!I ssid: " + String(cred.ssid));
Serial.println("!I password: " + String(cred.password));
unquote(ssidPwd.substring(0, i)).toCharArray(cred.ssid, sizeof(cred.token));
if (i != -1) {
unquote(ssidPwd.substring(i+1)).toCharArray(cred.password, sizeof(cred.password));
} else {
cred.password[0] = '\0';
}
if (wifiConnect()) {
EEPROM.put(0, cred);
EEPROM.commit();
EEPROM.commit();
WSerial.println("OK");
}
} else if (data.startsWith("AT+CWSAP=")) {
String ssid = unquote(data.substring(9));
softAP(ssid);
} else {
Serial.println("!E usage:");
Serial.println("!E GET /path");
Serial.println("!E POST /path");
Serial.println("!E payload");
Serial.println("!E TOKEN token");
Serial.println("!E AT+CWJAP=ssid,passwd");
WSerial.println("!E usage:");
WSerial.println("!E GET /path");
WSerial.println("!E POST /path");
WSerial.println("!E payload");
WSerial.println("!E TOKEN token");
WSerial.println("!E AT+CWJAP=ssid,passwd");
WSerial.println("!E AT+CWSAP=ssid");
}
}

if (request.length() > 0) {
WiFiClientSecure client;
if (!client.connect(host, httpsPort)) {
Serial.println("!E connection failed");
WSerial.println("!I request");
WiFiClientSecure httpClient;
if (!httpClient.connect(host, httpsPort)) {
WSerial.println("!E connection failed");
return;
}
if (!client.verify(fingerprint, host)) {
Serial.println("!E certificate doesn't match");
if (!httpClient.verify(fingerprint, host)) {
WSerial.println("!E certificate doesn't match");
return;
}
Serial.println("!I connected");
client.print(request);
WSerial.println("!I connected");
httpClient.print(request);
request.trim();
request.replace("\r\n", "\r\n!> ");
Serial.println(String("!> ") + request);
Serial.println("!I headers");
while (client.connected()) {
String data = client.readStringUntil('\n');
Serial.println("!< " + data);
WSerial.print("!> ");
WSerial.println(request);
WSerial.println("!I headers");
while (httpClient.connected()) {
String data = httpClient.readStringUntil('\n');
WSerial.print("!< ");
WSerial.println(data);
}
Serial.println("!I body");
while(client.available() > 0) {
String data = client.readStringUntil('\n');
Serial.println(data);
WSerial.println("!I body");
while(httpClient.available() > 0) {
String data = httpClient.readStringUntil('\n');
WSerial.println(data);
}
WSerial.println("!I end of request");
}
}